VC++ Ping IP地址

 

// // Ping.h // #pragma pack(1) #define ICMP_ECHOREPLY 0 #define ICMP_ECHOREQ 8 #define REQ_DATASIZE 32 // Echo 请求数据的大小 class CPing { public: DWORD m_dwSendTime; HWND m_hWnd; //窗口句柄 BOOL Ping(UINT nNumIp, CString pstrHost, CString &m_Msg ); int WaitForEchoReply(SOCKET s); //ICMP回应的请求和回答函数 int SendEchoRequest(SOCKET, LPSOCKADDR_IN); DWORD RecvEchoReply(SOCKET, LPSOCKADDR_IN, u_char *); u_short in_cksum(u_short *addr, int len); protected: CString WSAError(LPCSTR pstrFrom); }; // 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; // 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; #pragma pack() // // ping.cpp // #include "stdafx.h" #include "Ping.h" BOOL CPing::Ping(UINT nNumIp, CString pstrHost, CString &m_Msg ) { 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套节字 int ret = WSAStartup(0x202, &wsaData); if( ret != 0) { m_Msg = "端口错误"; return FALSE; } rawSocket = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP); if (rawSocket == SOCKET_ERROR) { nRet = closesocket(rawSocket); ::WSACleanup(); m_Msg = WSAError("socket()"); return FALSE; } //获得主机信息 lpHost = gethostbyname(pstrHost); if (lpHost == NULL) { nRet = closesocket(rawSocket); ::WSACleanup(); m_Msg.Format("Host not found: %s", pstrHost); 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); nRet = WaitForEchoReply(rawSocket); if (nRet == SOCKET_ERROR){ nRet = closesocket(rawSocket); ::WSACleanup(); m_Msg = WSAError("select()"); return FALSE; } if(!nRet){ nRet = closesocket(rawSocket); ::WSACleanup(); m_Msg.Format("Request Timed Out"); 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(); m_Msg.Format("OK!!!"); 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(); m_Msg.Format("OK !!!"); return TRUE; } } nRet = closesocket(rawSocket); nRet = ::WSACleanup(); m_Msg.Format("ERROR !!!"); return FALSE; } } } //发送ICMPECHO数据包请求 int CPing::SendEchoRequest(SOCKET s,LPSOCKADDR_IN lpstToAddr) { static ECHOREQUEST echoReq; static int nSeq = 1; int nRet; //构造回应请求 echoReq.icmpHdr.Type = ICMP_ECHOREQ; echoReq.icmpHdr.Code = 0; echoReq.icmpHdr.Checksum = 0; echoReq.icmpHdr.ID = (USHORT)GetCurrentThreadId(); echoReq.icmpHdr.Seq = nSeq++; for (nRet = 0; nRet < REQ_DATASIZE; nRet++) echoReq.cData[nRet] = ' '+nRet; //保存发送时间 echoReq.dwTime = GetTickCount(); this->m_dwSendTime = echoReq.dwTime; echoReq.icmpHdr.Checksum = in_cksum((u_short *)&echoReq, sizeof(ECHOREQUEST)); //发送请求 nRet = sendto(s, (LPSTR)&echoReq, sizeof(ECHOREQUEST), 0, (LPSOCKADDR)lpstToAddr, sizeof(SOCKADDR_IN)); if (nRet == SOCKET_ERROR){ ::closesocket(s); ::WSACleanup(); } return (nRet); } //接收ICMPECHO数据包回应 DWORD CPing::RecvEchoReply(SOCKET s, LPSOCKADDR_IN lpsaFrom, u_char *pTTL) { ECHOREPLY echoReply; int nRet; int nAddrLen = sizeof(struct sockaddr_in); //接收请求回应 nRet = recvfrom(s, (LPSTR)&echoReply, sizeof(ECHOREPLY), 0, (LPSOCKADDR)lpsaFrom, &nAddrLen); //检查返回值 if (nRet == SOCKET_ERROR){ ::closesocket(s); ::WSACleanup(); return FALSE; } //返回发送的时间 *pTTL = echoReply.ipHdr.TTL; return(echoReply.echoRequest.dwTime); } //等待回应 int CPing::WaitForEchoReply(SOCKET s) { struct timeval Timeout; fd_set readfds; readfds.fd_count = 1; readfds.fd_array[0] = s; Timeout.tv_sec = 0; // 秒 Timeout.tv_usec = 10; // 毫秒 return(select(1, &readfds, NULL, NULL, &Timeout)); } //转换地址 u_short CPing::in_cksum(u_short *addr, int len) { register int nleft = len; register u_short *w = addr; register u_short answer; register 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); sum += (sum >> 16); answer = ~sum; return (answer); } //错误处理 CString CPing::WSAError(LPCSTR lpMsg) { CString strMsg; strMsg.Format("%s - WSAError: %ld",lpMsg,WSAGetLastError()); //返回报错信息 return strMsg; }