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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#include "common.h"
char data[] = "AAAABBBBCCCCDDDDEEEEFFFFGGGGHHH";
ICMP *ICMP::instance = 0;
#define BUFFER_SIZE (16 * (sizeof(ICMP_ECHO_REPLY) + sizeof(data)))
ICMP::ICMP():
timeout(2000),
functions_loaded(false)
{
hDLL = LoadLibrary(_T("IPHLPAPI.DLL"));
if(hDLL) {
pIcmpCreateFile = (pfnHV)GetProcAddress(hDLL, "IcmpCreateFile");
pIcmpCloseHandle = (pfnBH)GetProcAddress(hDLL, "IcmpCloseHandle");
pIcmpSendEcho2 = (pfnDHDPWPipPDD)GetProcAddress(hDLL, "IcmpSendEcho2");
}
if (hDLL == 0 || pIcmpCreateFile == 0 || pIcmpCloseHandle == 0 || pIcmpSendEcho2 == 0) {
hDLL = LoadLibrary(_T("ICMP.DLL"));
if(hDLL) {
pIcmpCreateFile = (pfnHV)GetProcAddress(hDLL, "IcmpCreateFile");
pIcmpCloseHandle = (pfnBH)GetProcAddress(hDLL, "IcmpCloseHandle");
pIcmpSendEcho2 = (pfnDHDPWPipPDD)GetProcAddress(hDLL, "IcmpSendEcho2");
}
if (hDLL == 0 || pIcmpCreateFile == 0 || pIcmpCloseHandle == 0 || pIcmpSendEcho2 == 0)
return;
else
db_set_s(0, PLUG, "PingLib", "ICMP.DLL"); // for debugging
} else
db_set_s(0, PLUG, "PingLib", "IPHLPAPI.DLL"); // for debugging
WSAData wsaData;
if (WSAStartup(MAKEWORD(1, 1), &wsaData) != 0) {
WSACleanup();
FreeLibrary((HMODULE)hDLL);
return;
}
/*
hIP = pIcmpCreateFile();
if (hIP == INVALID_HANDLE_VALUE) {
pIcmpCloseHandle(hIP);
return;
}
*/
buff = new char[BUFFER_SIZE];
functions_loaded = true;
}
void ICMP::stop() {
//if(hIP) {
// pIcmpCloseHandle(hIP);
// hIP = 0;
//}
}
ICMP::~ICMP() {
if(hIP) stop();
WSACleanup();
if(hDLL)
FreeLibrary(hDLL);
delete[] buff;
}
bool ICMP::ping(char *host, ICMP_ECHO_REPLY &reply)
{
if (!functions_loaded)
return false;
HOSTENT *rec;
IP_OPTION_INFORMATION ipoi;
unsigned long address = inet_addr(host);
if (address == INADDR_NONE)
{
rec = gethostbyname(host);
if (rec != NULL)
address = *(unsigned long *)(*rec->h_addr_list);
else
return false;
}
ipoi.Ttl = 255;
ipoi.Tos = 0;
ipoi.Flags = 0;
ipoi.OptionsSize = 0;
ipoi.OptionsData = 0;
reply.Status = 0;
hIP = pIcmpCreateFile();
if (hIP == INVALID_HANDLE_VALUE)
return false;
//pIcmpSendEcho2(hIP, 0, 0, 0, address, data, sizeof(data), &ipoi, buff, sizeof(ICMP_ECHO_REPLY) + sizeof(data), timeout);
DWORD rep_cnt = pIcmpSendEcho2(hIP, 0, 0, 0, address, data, sizeof(data), 0, buff, BUFFER_SIZE, timeout);
if (rep_cnt == 0)
{
DWORD code = GetLastError();
if (code != 11010)
{
char winmsg[512], msg[1024];
FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM, 0, code, 0, winmsg, 512, 0);
mir_snprintf(msg, 1024, "Ping error (%d): %s", code, winmsg);
PUShowMessage(msg, SM_NOTIFY);
return false;
}
}
memcpy(&reply, buff, sizeof(ICMP_ECHO_REPLY));
pIcmpCloseHandle(hIP);
return (reply.Status == 0);
}
ICMP *ICMP::get_instance() {
if(!instance)
instance = new ICMP();
return instance;
}
void ICMP::cleanup() {
if(instance) delete instance;
}
|
