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
|
#include "stdafx.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)
{
WSAData wsaData;
if (WSAStartup(MAKEWORD(1, 1), &wsaData) != 0) {
WSACleanup();
return;
}
buff = new char[BUFFER_SIZE];
functions_loaded = true;
}
void ICMP::stop()
{
}
ICMP::~ICMP() {
if (hIP) stop();
WSACleanup();
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 = IcmpCreateFile();
if (hIP == INVALID_HANDLE_VALUE)
return false;
DWORD rep_cnt = IcmpSendEcho2(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, _countof(msg), "Ping error (%d): %s", code, winmsg);
PUShowMessage(msg, SM_NOTIFY);
return false;
}
}
memcpy(&reply, buff, sizeof(ICMP_ECHO_REPLY));
IcmpCloseHandle(hIP);
return (reply.Status == 0);
}
ICMP *ICMP::get_instance() {
if (!instance)
instance = new ICMP();
return instance;
}
void ICMP::cleanup() {
if (instance) delete instance;
}
|
