#include <afxsock.h>
#define ICMP_ECHOREPLY 0
#define ICMP_ECHOREQ 8
// IP Header -- RFC 791
typedef struct tagIPHDR
{
u_char VIHL; // Version and IHL
u_char TOS; // Type Of Service
short TotLen; // Total Length
short ID; // Identification
short FlagOff; // Flags and Fragment Offset
u_char TTL; // Time To Live
u_char Protocol; // Protocol
u_short Checksum; // Checksum
struct in_addr iaSrc; // Internet Address - Source
struct in_addr iaDst; // Internet Address - Destination
}IPHDR, *PIPHDR;
// ICMP Header - RFC 792
typedef struct tagICMPHDR
{
u_char Type; // Type
u_char Code; // Code
u_short Checksum; // Checksum
u_short ID; // Identification
u_short Seq; // Sequence
char Data; // Data
}ICMPHDR, *PICMPHDR;
#define REQ_DATASIZE 32 // Echo Request Data size
// ICMP Echo Request
typedef struct tagECHOREQUEST
{
ICMPHDR icmpHdr;
DWORD dwTime;
char cData[REQ_DATASIZE];
}ECHOREQUEST, *PECHOREQUEST;
// ICMP Echo Reply
typedef struct tagECHOREPLY
{
IPHDR ipHdr;
ECHOREQUEST echoRequest;
char cFiller[256];
}ECHOREPLY, *PECHOREPLY;
u_short in_cksum(u_short *addr, int len)
{
int nleft = len;
u_short *w = addr;
u_short answer;
int sum = 0;
while( nleft > 1 ) {
sum += *w++;
nleft -= 2;
}
if( nleft == 1 ) {
u_short u = 0;
*(u_char *)(&u) = *(u_char *)w ;
sum += u;
}
sum = (sum >> 16) + (sum & 0xffff); /* add hi 16 to low 16 */
sum += (sum >> 16); /* add carry */
answer = ~sum; /* truncate to 16 bits */
return (answer);
}
int WaitForEchoReply(SOCKET s)
{
struct timeval Timeout;
fd_set readfds;
readfds.fd_count = 1;
readfds.fd_array[0] = s;
Timeout.tv_sec = 1;
Timeout.tv_usec = 0;
return(select(1, &readfds, NULL, NULL, &Timeout));
}
unsigned int RecvEchoReply(SOCKET s, LPSOCKADDR_IN lpsaFrom, u_char *pTTL)
{
ECHOREPLY echoReply;
int nRet;
int nAddrLen = sizeof(struct sockaddr_in);
// Receive the echo reply
nRet = recvfrom(s, // socket
(LPSTR)&echoReply, // buffer
sizeof(ECHOREPLY), // size of buffer
0, // flags
(LPSOCKADDR)lpsaFrom, // From address
&nAddrLen); // pointer to address len
// Check return value
// return time sent and IP TTL
*pTTL = echoReply.ipHdr.TTL;
return(echoReply.echoRequest.dwTime);
}
extern "C" _declspec (dllexport)BOOL Ping(CString pstrHost, UINT nNumIp)
{
ECHOREQUEST echoReq;
int nId = 1;
int nSeq = 1;
SOCKET rawSocket;
LPHOSTENT lpHost;
UINT nLoop;
int nRet;
struct sockaddr_in saDest;
struct sockaddr_in saSrc;
DWORD dwTimeSent;
DWORD dwElapsed;
u_char cTTL;
char *destIP = NULL;
WSADATA wsaData;
//创建一个Raw套节字
if(WSAStartup(MAKEWORD(2,1), &wsaData))
{
return false;
}
rawSocket = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);
if (rawSocket == SOCKET_ERROR)
{
nRet = closesocket(rawSocket);
::WSACleanup();
return FALSE;
}
//获得主机信息
lpHost = gethostbyname(pstrHost);
if (lpHost == NULL)
{
nRet = closesocket(rawSocket);
::WSACleanup();
return FALSE;
}
//构造目标套节字地址信息
memset(&saDest, 0, sizeof(sockaddr_in));
saDest.sin_addr.s_addr = *((u_long FAR *) (lpHost->h_addr));
saDest.sin_family = AF_INET;
saDest.sin_port = 0;
//发送ICMP回应请求
// SendEchoRequest(rawSocket,&saDest);
int rnRet;
echoReq.icmpHdr.Type = ICMP_ECHOREQ;
echoReq.icmpHdr.Code = 0;
echoReq.icmpHdr.Checksum = 0;
echoReq.icmpHdr.ID = nId++;
echoReq.icmpHdr.Seq = nSeq++;
// Fill in some data to send
for (rnRet = 0; rnRet < REQ_DATASIZE; rnRet++)
echoReq.cData[rnRet] = ' '+rnRet;
// Save tick count when sent
echoReq.dwTime = GetTickCount();
// Put data in packet and compute checksum
echoReq.icmpHdr.Checksum = in_cksum((u_short *)&echoReq, sizeof(ECHOREQUEST));
// Send the echo request
rnRet = sendto(rawSocket, /* socket */
(LPSTR)&echoReq, /* buffer */
sizeof(ECHOREQUEST),
0, /* flags */
(LPSOCKADDR)&saDest, /* destination */
sizeof(SOCKADDR_IN)); /* address length */
if (rnRet == SOCKET_ERROR)
{
rnRet = closesocket(rawSocket);
::WSACleanup();
return FALSE;
}
nRet = WaitForEchoReply(rawSocket);
if (nRet == SOCKET_ERROR){
nRet = closesocket(rawSocket);
::WSACleanup();
return FALSE;
}
if(!nRet){
nRet = closesocket(rawSocket);
::WSACleanup();
return FALSE;
}
else{
//获得回应
dwTimeSent = RecvEchoReply(rawSocket, &saSrc, &cTTL);
//计算时间
dwElapsed = GetTickCount() - dwTimeSent;
::Sleep(1000);
destIP = inet_ntoa(saSrc.sin_addr);//获得客户IP地址
if(pstrHost == destIP){
nRet = closesocket(rawSocket);
nRet = ::WSACleanup();
return TRUE;
}
else
{
for(nLoop = 0; nLoop < nNumIp; nLoop++)
{
dwTimeSent = RecvEchoReply(rawSocket, &saSrc, &cTTL);
destIP = inet_ntoa(saSrc.sin_addr);
if(pstrHost == destIP)
{
nRet = closesocket(rawSocket);
nRet = ::WSACleanup();
return TRUE;
}
}
nRet = closesocket(rawSocket);
nRet = ::WSACleanup();
return FALSE;
}
}
}
