summaryrefslogtreecommitdiff
path: root/ping_protocol/icmp.cpp
blob: a9cd2962c4b6c5ce8ecb34b3cc338b9f3e07c511 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include "stdafx.h"
#include "icmp.h"

ICMP ICMP::instance;

ICMP::ICMP():
	timeout(2000),
	functions_loaded(false)
{
	hDLL = LoadLibrary("ICMP.DLL");
	if(!hDLL) return;

	pIcmpCreateFile = (pfnHV)GetProcAddress(hDLL, "IcmpCreateFile");
	pIcmpCloseHandle = (pfnBH)GetProcAddress(hDLL, "IcmpCloseHandle");
	pIcmpSendEcho = (pfnDHDPWPipPDD)GetProcAddress(hDLL, "IcmpSendEcho");
	if ((pIcmpCreateFile == 0) || (pIcmpCloseHandle == 0) || (pIcmpSendEcho == 0)) {
		return;
	}

    WSAData wsaData;
    if (WSAStartup(MAKEWORD(1, 1), &wsaData) != 0) {
		WSACleanup();
		FreeLibrary((HMODULE)hDLL);
		return;
    }

	functions_loaded = true;
}

void ICMP::stop() {
	if(hIP) {
		pIcmpCloseHandle(hIP);
		hIP = 0;
	}
}

ICMP::~ICMP() {
	if(hIP) pIcmpCloseHandle(hIP);
	WSACleanup();
	if(hDLL)
		FreeLibrary(hDLL);	
}

bool ICMP::ping(char *host, IP_ECHO_REPLY &reply) {
	if(!functions_loaded) return false;

	DWORD address;
	struct in_addr dest;
	HOSTENT *rec;
	IP_OPTION_INFORMATION ipoi;

	hIP = pIcmpCreateFile();
	if (hIP == INVALID_HANDLE_VALUE) {
		return false;
	}
   

	dest.s_addr = inet_addr(host);
	if (dest.s_addr == INADDR_NONE) {
		rec = gethostbyname(host);
		if(rec) address = *(DWORD *)(*rec->h_addr_list);
		else return false;
	} else {
		address = dest.s_addr;
	}

	ipoi.Ttl = 255;
	ipoi.Tos = 0;
	ipoi.Flags = 0;
	ipoi.OptionsSize = 0;
	ipoi.OptionsData = 0;
	
	reply.Status = 0;	

	pIcmpSendEcho(hIP, address, (void *)"PINGPONG", 8, &ipoi, &reply, sizeof(reply), timeout);

	if(hIP) {
		pIcmpCloseHandle(hIP);
		hIP = 0;
	}

	return (reply.Status == 0);
}