PING程序设计测试网络连通

#include"Init_ping.h" #include <iostream> #define DEF_PACKET_SIZE 32 #define ECHO_REQUEST 8 #define ECHO_REPLY 0 USHORT CPing::s_usPacketSeq = 0; USHORT CPing::Cchecksum(USHORT *buffer,int len)//检验和算法：16位字二进制反码相加的反码 { unsigned long cksum=0; while(len>1) { cksum+= *buffer++; len -=sizeof(USHORT); } if(len) { cksum += *(UCHAR*)buffer; } cksum=(cksum>>16)+(cksum&0xffff); cksum +=(cksum>>16); return (USHORT)(~cksum); } CPing::CPing() :szICMPData(NULL),bIsInitSucc(FALSE) //函数创建 { WSADATA WSAData; //保存WSAStartup返回的SOCKET //WSAStartup(MAKEWORD(2, 2), &WSAData); if (WSAStartup(MAKEWORD(1, 1), &WSAData) != 0) { /*如果初始化不成功则报错，GetLastError()返回发生的错误信息*/ printf("WSAStartup() failed: %d\n", GetLastError()); return; } Event = WSACreateEvent(); //创建一个新对象 usCurrentProcID = (USHORT)GetCurrentProcessId(); //获取当前进程一个唯一的标识符 //setsockopt(m_sockRaw); /*if ((m_sockRaw = WSASocket(AF_INET, SOCK_RAW, IPPROTO_ICMP, NULL, 0, 0)) != SOCKET_ERROR) { WSAEventSelect(m_sockRaw, m_event, FD_READ); m_bIsInitSucc = TRUE; m_szICMPData = (char*)malloc(DEF_PACKET_SIZE + sizeof(ICMPHeader)); if (m_szICMPData == NULL) { m_bIsInitSucc = FALSE; } }*/ soce = WSASocket(AF_INET, SOCK_RAW, IPPROTO_ICMP, NULL, 0, 0); //创建一个原始套接字 if (soce == INVALID_SOCKET) { std::cerr << "WSASocket() failed:" << WSAGetLastError ()<< std::endl; //10013 以一种访问权限不允许的方式做了一个访问套接字的尝试。 } else { WSAEventSelect(soce, Event, FD_READ); bIsInitSucc = TRUE; szICMPData = (char*)malloc(DEF_PACKET_SIZE + sizeof(IcmpHeader)); if (szICMPData == NULL) { bIsInitSucc = FALSE; } } } CPing::~CPing() //析构函数 { WSACleanup(); if (NULL != szICMPData) { free(szICMPData); szICMPData = NULL; } } BOOL CPing::Ping(DWORD wdestIP, PingReply *pPingReply, DWORD wTimeout) { return PingCore(wdestIP, pPingReply, wTimeout); } BOOL CPing::Ping(char *szdestIP, PingReply *pPingReply, DWORD dwTimeout) { if (NULL != szdestIP) { return PingCore(inet_addr(szdestIP), pPingReply, dwTimeout); //IP转换 } return FALSE; } BOOL CPing::PingCore(DWORD wdestIP, PingReply *pPingReply, DWORD wTimeout) { //判断初始化是否成功 if (!bIsInitSucc) { return FALSE; } //配置SOCKET sockaddr_in sockaddrDest; sockaddrDest.sin_family = AF_INET; sockaddrDest.sin_addr.s_addr = wdestIP; int nSockaddrDestSize = sizeof(sockaddrDest); PINGping pping; //构建ICMP包 int nICMPDataSize = DEF_PACKET_SIZE + sizeof(IcmpHeader); ULONG ulSendTimestamp = GetTickCountCalibrate(); USHORT usSeq = ++s_usPacketSeq; memset(szICMPData, 0, nICMPDataSize); IcmpHeader *pIcmpHeader=(IcmpHeader*)szICMPData; pIcmpHeader->IcmpType = ECHO_REQUEST; pIcmpHeader->IcmpCode = 0; pIcmpHeader->IcmpId = usCurrentProcID; pIcmpHeader->IcmpSeq = usSeq; pIcmpHeader->IcmpTimestamp = ulSendTimestamp; pIcmpHeader->IcmpChecksum = Cchecksum((USHORT*)szICMPData, nICMPDataSize); //发送ICMP报文 if (sendto(soce, szICMPData, nICMPDataSize, 0, (struct sockaddr*)&sockaddrDest, nSockaddrDestSize) == SOCKET_ERROR) { return FALSE; } //判断是否需要接收相应报文 if (pPingReply == NULL) { return TRUE; } char recvbuf[256] = { "\0" }; while (TRUE) { //接收响应报文 if (WSAWaitForMultipleEvents(1, &Event, FALSE, 100, FALSE) != WSA_WAIT_TIMEOUT) { WSANETWORKEVENTS netEvent; WSAEnumNetworkEvents(soce, Event, &netEvent); if (netEvent.lNetworkEvents & FD_READ) { ULONG nRecvTimestamp = GetTickCountCalibrate(); int nPacketSize = recvfrom(soce, recvbuf, 256, 0, (struct sockaddr*)&sockaddrDest, &nSockaddrDestSize); if (nPacketSize != SOCKET_ERROR) { IpHeader *pIPHeader = (IpHeader*)recvbuf; USHORT usIPHeaderLen = (USHORT)((pIPHeader->Version_and_HeaderLength & 0x0f) * 4); IcmpHeader *pICMPHeader = (IcmpHeader*)(recvbuf + usIPHeaderLen); if (pICMPHeader->IcmpId == usCurrentProcID //是当前进程发出的报文 && pICMPHeader->IcmpType == ECHO_REPLY //是ICMP响应报文 && pICMPHeader->IcmpSeq == usSeq //是本次请求报文的响应报文 ) { pPingReply->IcmpSeq = usSeq; pPingReply->RoundTripTime = nRecvTimestamp - pICMPHeader->IcmpTimestamp; pPingReply->Bytes = nPacketSize - usIPHeaderLen - sizeof(IcmpHeader); pPingReply->TTL = pIPHeader->TTL; return TRUE; } } } } //超时 if ((GetTickCountCalibrate() - ulSendTimestamp) >= wTimeout) { return FALSE; } } } ULONG CPing::GetTickCountCalibrate() { static ULONG s_ulFirstCallTick = 0; static LONGLONG s_ullFirstCallTickMS = 0; SYSTEMTIME systemtime; FILETIME filetime; GetLocalTime(&systemtime); SystemTimeToFileTime(&systemtime, &filetime); LARGE_INTEGER liCurrentTime; liCurrentTime.HighPart = filetime.dwHighDateTime; liCurrentTime.LowPart = filetime.dwLowDateTime; LONGLONG llCurrentTimeMS = liCurrentTime.QuadPart / 10000; if (s_ulFirstCallTick == 0) { s_ulFirstCallTick = GetTickCount(); } if (s_ullFirstCallTickMS == 0) { s_ullFirstCallTickMS = llCurrentTimeMS; } return s_ulFirstCallTick + (ULONG)(llCurrentTimeMS - s_ullFirstCallTickMS); }