DevicePinger.h

#pragma once
#include <winsock2.h>
#include <ws2tcpip.h>
#include <stdio.h>
#include <stdlib.h>
#include<string>;
 
#include "iphdr.h"
 
#ifdef STARFLEETTOOLBOX_EXPORTS
#define ETHDEVICEPINGER __declspec(dllexport)
#else
#define ETHDEVICEPINGER __declspec(dllimport)
#endif
 
#define DEFAULT_DATA_SIZE       32       // default data size
#define DEFAULT_SEND_COUNT      4        // number of ICMP requests to send
#define DEFAULT_RECV_TIMEOUT    1000     // one second
#define DEFAULT_TTL             128
#define MAX_RECV_BUF_LEN        0xFFFF   // Max incoming packet size.
#define PING_SLEEP              250     // sleep between ping
 
using namespace std;
 
class ETHDEVICEPINGER DevicePinger
{
public:
    DevicePinger();
 
    DevicePinger(string address);
 
    DevicePinger(string address, int timeout);
 
    void SetAddress(string address);
        
 
    PING_RESULT Ping();
 
 
    PING_RESULT Ping(int packetAmount);
 
private:
    SOCKET  _s = INVALID_SOCKET;
    WSAOVERLAPPED _recvol;
    char* _icmpbuf = NULL;
    int _packetlen = 0;
    int _receiveTimeOut = DEFAULT_RECV_TIMEOUT;
    struct addrinfo* _dest = NULL;
 
 
    int     _addressFamily = AF_INET,               
            _protocol = IPPROTO_ICMP,               
            _ttl = DEFAULT_TTL;                     
    int     _dataSize = DEFAULT_DATA_SIZE;          
    BOOL    _recordRoute = FALSE;                   
    char*   _destination = NULL,                    
            _recvbuf[MAX_RECV_BUF_LEN];             
    int     _recvbuflen = MAX_RECV_BUF_LEN;         
    char _lastKnownIp[100];                         
    bool Initialize();
 
    void InitIcmpHeader(char* buf, int datasize);
 
    USHORT Checksum(USHORT* buffer, int size);
 
    void  SetIcmpSequence(char* buf);
 
    void ComputeIcmpChecksum(SOCKET s, char* buf, int packetlen, struct addrinfo* dest);
 
    int PostRecvfrom(SOCKET s, char* buf, int buflen, SOCKADDR* from, int* fromlen, WSAOVERLAPPED* ol);
 
    int SetTtl(SOCKET s, int ttl);
 
    struct addrinfo* ResolveAddress(char* addr, char* port, int af, int type, int proto);
};