Windows系统ping命令的c++实现

// ping.cpp : 定义控制台应用程序的入口点。
//#include <winsock2.h> 
#include <stdio.h> 
#include <stdlib.h> #define ICMP_ECHO 8 
#define ICMP_ECHOREPLY 0 /* The IP header */ 
typedef struct iphdr 
{ unsigned char h_len:4; // length of the header unsigned char version:4; // Version of IP unsigned char tos; // Type of service unsigned short total_len; // total length of the packet unsigned short ident; // unique identifier unsigned short frag_and_flags; // flags unsigned char ttl; unsigned char proto; // protocol (TCP, UDP etc) unsigned short checksum; // IP checksum unsigned int sourceIP; unsigned int destIP; 
}IpHeader; /* ICMP header */
typedef struct _ihdr 
{ BYTE i_type; BYTE i_code; /* type sub code */USHORT i_cksum; USHORT i_id; USHORT i_seq; /* This is not the std header, but we reserve space for time */ ULONG timestamp; 
}IcmpHeader; #define STATUS_FAILED 0xFFFF 
#define DEF_PACKET_SIZE 32 
#define MAX_PACKET 1024 /* The response is an IP packet. We must decode the IP header to locate the ICMP data */ 
void decode_resp(char *buf, int bytes,struct sockaddr_in *from)
{IpHeader *iphdr; IcmpHeader *icmphdr; unsigned short iphdrlen; iphdr=(IpHeader*)buf; iphdrlen=sizeof(IpHeader)+sizeof(IcmpHeader);if(bytes<iphdrlen) { printf("Too few bytes from %s\n",inet_ntoa(from->sin_addr)); } icmphdr=(IcmpHeader*)(buf+sizeof(IpHeader)); if(icmphdr->i_type!=ICMP_ECHOREPLY) { fprintf(stderr,"non-echo type %d recvd\n",icmphdr->i_type); return; }if(icmphdr->i_id!=(USHORT)GetCurrentProcessId()) { fprintf(stderr,"someone else's packet!\n"); return ; }int correct=0;for(int k=0;k<bytes-iphdrlen;k++){if(*(buf+iphdrlen+k)=='E'){correct++;}}printf("来自 %s 的回复: 字节=%d 时间=%dms TTL=%d\n",\inet_ntoa(from->sin_addr),correct,GetTickCount()-icmphdr->timestamp,iphdr->ttl); 
} USHORT checksum(USHORT *buffer, int size) 
{ unsigned long cksum=0; while(size>1) { cksum+=*buffer++; size-=sizeof(USHORT); } if(size) { cksum+=*(UCHAR*)buffer; } cksum=(cksum>>16)+(cksum&0xffff); cksum+=(cksum>>16); return (USHORT)(~cksum); 
} /* Helper function to fill in various stuff in our ICMP request. */ 
void fill_icmp_data(char *icmp_data, int datasize)
{ char *datapart=NULL; IcmpHeader *icmp_hdr=NULL; icmp_hdr=(IcmpHeader*)icmp_data; icmp_hdr->i_type=ICMP_ECHO; icmp_hdr->i_code=0; icmp_hdr->i_id=(USHORT)GetCurrentProcessId(); icmp_hdr->i_cksum=0; icmp_hdr->i_seq++; icmp_hdr->timestamp=GetTickCount(); datapart=icmp_data+sizeof(IcmpHeader); // Place some junk in the buffer. memset(datapart,'E',datasize-sizeof(IcmpHeader)); icmp_hdr->i_cksum=checksum((USHORT*)icmp_data,datasize);
} int main(int argc, char *argv[])
{ int count=1;int error=-1;int datasize=0; int timeout=3000; struct sockaddr_in addrServer; struct hostent *phostent=NULL; int destlen=sizeof(addrServer); char *dest_ip=NULL; char icmp_data[MAX_PACKET]; char recvbuf[MAX_PACKET]; unsigned int addr=0; char srcIP[MAX_PATH]="0.0.0.0";char dstIP[MAX_PATH]="127.0.0.1";for(int n=1;n<argc;){if(!strcmp(argv[n],"-S")){            n++;strcpy_s(srcIP,argv[n++]);}else if(!strcmp(argv[n],"-n")){n++;count=atoi(argv[n++]);}else{strcpy_s(dstIP,argv[n++]);}}WSADATA wsaData={0}; if(WSAStartup(MAKEWORD(1,1),&wsaData)!=0){ printf("WSA startup error\n"); ExitProcess(STATUS_FAILED); } SOCKET sockClient=WSASocket(AF_INET, SOCK_RAW, IPPROTO_ICMP, NULL, 0,0); if(sockClient==INVALID_SOCKET) { printf("WSA socket error\n"); ExitProcess(STATUS_FAILED); } sockaddr_in addrClient;addrClient.sin_addr.S_un.S_addr=inet_addr(srcIP);addrClient.sin_family=AF_INET;addrClient.sin_port=htons(0);error=bind(sockClient,(SOCKADDR *)&addrClient,sizeof(addrClient));if(error){closesocket(sockClient);printf("bind client error\n");return WSACleanup();}unsigned long ul=true;error=ioctlsocket(sockClient,FIONBIO,(unsigned long*)&ul);if(error){closesocket(sockClient);printf("set ioctlsocket error\n");return WSACleanup();}error=setsockopt(sockClient,SOL_SOCKET,SO_RCVTIMEO,(char*)&timeout, sizeof(timeout)); if(error==SOCKET_ERROR) { fprintf(stderr,"failed to set recv timeout: %d\n",WSAGetLastError()); ExitProcess(STATUS_FAILED); } error=setsockopt(sockClient,SOL_SOCKET,SO_SNDTIMEO,(char*)&timeout, sizeof(timeout)); if(error==SOCKET_ERROR) { fprintf(stderr,"failed to set send timeout: %d\n",WSAGetLastError()); ExitProcess(STATUS_FAILED); } memset(&addrServer,0,sizeof(addrServer)); phostent=gethostbyname(dstIP); if(!phostent){ addr=inet_addr(dstIP); } if((!phostent)&&(addr==INADDR_NONE)) { fprintf(stderr,"Unable to resolve host %s\n",dstIP); ExitProcess(STATUS_FAILED); } if(phostent!=NULL) {memcpy(&(addrServer.sin_addr),phostent->h_addr,phostent->h_length);}else{addrServer.sin_addr.s_addr=addr; }if(phostent) {addrServer.sin_family=phostent->h_addrtype;}else{addrServer.sin_family=AF_INET; dest_ip=inet_ntoa(addrServer.sin_addr);} datasize=DEF_PACKET_SIZE; datasize+=sizeof(IcmpHeader); memset(icmp_data,0,MAX_PACKET);for(int i=0;i<count;i++) {         fill_icmp_data(icmp_data,datasize); int num=sendto(sockClient,icmp_data,datasize,0,(struct sockaddr*)&addrServer, sizeof(addrServer)); if(num==SOCKET_ERROR){ error=WSAGetLastError();if(error==WSAETIMEDOUT) { fprintf(stderr,"errorcode=%d\n",error); continue; }} if(num<datasize) { fprintf(stdout,"Write %d bytes\n",num); }timeval tm;int len=sizeof(int);fd_set set;for(int k=0;k<3;k++){num=recvfrom(sockClient,recvbuf,MAX_PACKET,0,(struct sockaddr*)&addrServer, &destlen); if(num!=SOCKET_ERROR){break;}if(num==SOCKET_ERROR){    Sleep(1);tm.tv_sec=0;        tm.tv_usec=1;        FD_ZERO(&set);        FD_SET(sockClient,&set);        if(select(sockClient,NULL,&set,NULL,&tm)>0)        {            getsockopt(sockClient,SOL_SOCKET,SO_ERROR,(char*)&error,&len);            }                }}if(num!=SOCKET_ERROR){decode_resp(recvbuf,num,&addrServer); }else{printf("Can not find host %s\n",dstIP); }if(i>0){Sleep(1000);}} closesocket(sockClient);WSACleanup();return 0; 
}
编译命令如下：
g++ -std=c++2a -o ping.exe ping.cpp -lws2_32