设为主页 | 加入收藏 | 繁體中文

微软ping程序源代码完整版

  编写自己的一个ping步伐,可以说是很多人迈出网络编程的第一步吧!!这个ping步伐的源代码颠末我的修改和调试,根本上可以取代windows中自带的ping步伐. 各个模块后都有我的细致注释和修改日志,盼望能够对大家的学习有所帮助!!
  /*  本步伐的重要源代码来自MSDN网站, 笔者只是做了一些改进和注释! 别的需要细致的是在Build之前,必须加入ws2_32.lib库文件,否则会提示"error LNK2001:"的错误!*/
  /纨绔子弟纨绔子弟纨绔子弟纨绔子弟纨绔子弟纨绔子弟******\
  | Version 1.1 修改记载:                                                                                                              |
  |    <1> 解决了socket阻塞的题目,从而能够精确地处置惩罚超时的请求!                                                        |
  |----------------------------------------------------------------------------------------------------|
  | Version 1.2 修改记载:                                                                                                              |
  |    <1> 增长了由用户控制发送ICMP包的数目标功效(即命令的第二个参数).                                              |   
  |    <2> 增长了对ping结果的统计功效.                                                                                           |
  \纨绔子弟纨绔子弟纨绔子弟纨绔子弟纨绔子弟纨绔子弟******/
  #pragma pack(4)
  #include
  #include
  #include
  #define ICMP_ECHO 8
  #define ICMP_ECHOREPLY 0
  #define ICMP_MIN 8 // minimum 8 byte icmp packet (just header)
  /* The IP header */
  typedef struct iphdr {
  unsigned int h_len:4; // length of the header
  unsigned int 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 icmphdr {
  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 DEF_PACKET_NUMBER  4    /* 发送数据报的个数 */
  #define MAX_PACKET 1024
  #define xmalloc(s) HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,(s))
  #define xfree(p) HeapFree (GetProcessHeap(),0,(p))
  void fill_icmp_data(char *, int);
  USHORT checksum(USHORT *, int);
  int decode_resp(char *,int ,struct sockaddr_in *);
  void Usage(char *progname){
  fprintf(stderr,"Usage:\n");
  fprintf(stderr,"%s [number of packets] [data_size]\n",progname);
  fprintf(stderr,"datasize can be up to 1Kb\n");
  ExitProcess(STATUS_FAILED);
  }
  int main(int argc, char **argv){
  WSADATA wsaData;
  SOCKET sockRaw;
  struct sockaddr_in dest,from;
  struct hostent * hp;
  int bread,datasize,times;
  int fromlen = sizeof(from);
  int timeout = 1000;
  int statistic = 0;  /* 用于统计结果 */ 
  char *dest_ip;
  char *icmp_data;
  char *recvbuf;
  unsigned int addr=0;
  USHORT seq_no = 0;
  if (WSAStartup(MAKEWORD(2,1),&wsaData) != 0){
  fprintf(stderr,"WSAStartup failed: %d\n",GetLastError());
  ExitProcess(STATUS_FAILED);
  }
  if (argc <2 ) {
  Usage(argv[0]);
  }
  sockRaw = WSASocket(AF_INET,SOCK_RAW,IPPROTO_ICMP,NULL, 0,WSA_FLAG_OVERLAPPED);
  //
  //注:为了使用发送吸收超时设置(即设置SO_RCVTIMEO, SO_SNDTIMEO),
  //    必须将标志位设为WSA_FLAG_OVERLAPPED !
  //
  if (sockRaw == INVALID_SOCKET) {
  fprintf(stderr,"WSASocket() failed: %d\n",WSAGetLastError());
  ExitProcess(STATUS_FAILED);
  }
  bread = setsockopt(sockRaw,SOL_SOCKET,SO_RCVTIMEO,(char*)&timeout,
  sizeof(timeout));
  if(bread == SOCKET_ERROR) {
  fprintf(stderr,"failed to set recv timeout: %d\n",WSAGetLastError());
  ExitProcess(STATUS_FAILED);
  }
  timeout = 1000;
  bread = setsockopt(sockRaw,SOL_SOCKET,SO_SNDTIMEO,(char*)&timeout,
  sizeof(timeout));
  if(bread == SOCKET_ERROR) {
  fprintf(stderr,"failed to set send timeout: %d\n",WSAGetLastError());
  ExitProcess(STATUS_FAILED);
  }
  memset(&dest,0,sizeof(dest));
  hp = gethostbyname(argv[1]);
  if (!hp){
  addr = inet_addr(argv[1]);
  }
  if ((!hp) && (addr == INADDR_NONE) ) {
  fprintf(stderr,"Unable to resolve %s\n",argv[1]);
  ExitProcess(STATUS_FAILED);
  }
  if (hp != NULL)
  memcpy(&(dest.sin_addr),hp->h_addr,hp->h_length);
  else
  dest.sin_addr.s_addr = addr;
  if (hp)
  dest.sin_family = hp->h_addrtype;
  else
  dest.sin_family = AF_INET;
  dest_ip = inet_ntoa(dest.sin_addr);
  //
  //  atoi函数原型是: int atoi( const char *string );
  //  The return value is 0 if the input cannot be converted to an integer !
  //
  if(argc>2)
  {
  times=atoi(argv[2]);
  if(times == 0)
  times=DEF_PACKET_NUMBER;
  }
  else
  times=DEF_PACKET_NUMBER;
  if (argc >3)
  {
  datasize = atoi(argv[3]);
  if (datasize == 0)
  datasize = DEF_PACKET_SIZE;
  if (datasize >1024)   /* 用户给出的数据包巨细太大 */
  {
  fprintf(stderr,"WARNING : data_size is too large !\n");
  datasize = DEF_PACKET_SIZE;
  }
  }
  else
  datasize = DEF_PACKET_SIZE;
  datasize += sizeof(IcmpHeader);
  icmp_data = (char*)xmalloc(MAX_PACKET);
  recvbuf = (char*)xmalloc(MAX_PACKET);
  if (!icmp_data) {
  fprintf(stderr,"HeapAlloc failed %d\n",GetLastError());
  ExitProcess(STATUS_FAILED);
  }
  memset(icmp_data,0,MAX_PACKET);
  fill_icmp_data(icmp_data,datasize);
  //
  //显示提示信息
  //
  fprintf(stdout,"\nPinging %s ....\n\n",dest_ip);
  for(int i=0;i{
  int bwrote;
  ((IcmpHeader*)icmp_data)->i_cksum = 0;
  ((IcmpHeader*)icmp_data)->timestamp = GetTickCount();
  ((IcmpHeader*)icmp_data)->i_seq = seq_no++;
  ((IcmpHeader*)icmp_data)->i_cksum = checksum((USHORT*)icmp_data,datasize);
  bwrote = sendto(sockRaw,icmp_data,datasize,0,(struct sockaddr*)&dest,sizeof(dest));
  if (bwrote == SOCKET_ERROR){
  if (WSAGetLastError() == WSAETIMEDOUT) {
  printf("Request timed out.\n");
  continue;
  }
  fprintf(stderr,"sendto failed: %d\n",WSAGetLastError());
  ExitProcess(STATUS_FAILED);
  }
  if (bwrote < datasize ) {
  fprintf(stdout,"Wrote %d bytes\n",bwrote);
  }
  bread = recvfrom(sockRaw,recvbuf,MAX_PACKET,0,(struct sockaddr*)&from,&fromlen);
  if (bread == SOCKET_ERROR){
  if (WSAGetLastError() == WSAETIMEDOUT) {
  printf("Request timed out.\n");
  continue;
  }
  fprintf(stderr,"recvfrom failed: %d\n",WSAGetLastError());
  ExitProcess(STATUS_FAILED);
  }
  if(!decode_resp(recvbuf,bread,&from))
  statistic++; /* 乐成吸收的数目++ */
  Sleep(1000);
  }
  /*
  Display the statistic result
  */
  fprintf(stdout,"\nPing statistics for %s \n",dest_ip);
  fprintf(stdout,"    Packets: Sent = %d,Received = %d, Lost = %d (%2.0f%% loss)\n",times,
  statistic,(times-statistic),(float)(times-statistic)/times*100);
  WSACleanup();
  return 0;
  }
  /*
  The response is an IP packet. We must decode the IP header to locate
  the ICMP data
  */
  int decode_resp(char *buf, int bytes,struct sockaddr_in *from) {
  IpHeader *iphdr;
  IcmpHeader *icmphdr;
  unsigned short iphdrlen;
  iphdr = (IpHeader *)buf;
  iphdrlen = (iphdr->h_len) * 4 ; // number of 32-bit words *4 = bytes
  if (bytes < iphdrlen + ICMP_MIN) {
  printf("Too few bytes from %s\n",inet_ntoa(from->sin_addr));
  }
  icmphdr = (IcmpHeader*)(buf + iphdrlen);
  if (icmphdr->i_type != ICMP_ECHOREPLY) {
  fprintf(stderr,"non-echo type %d recvd\n",icmphdr->i_type);
  return 1;
  }
  if (icmphdr->i_id != (USHORT)GetCurrentProcessId()) {
  fprintf(stderr,"someone else's packet!\n");
  return 1;
  }
  printf("%d bytes from %s:",bytes, inet_ntoa(from->sin_addr));
  printf(" icmp_seq = %d. ",icmphdr->i_seq);
  printf(" time: %d ms ",GetTickCount()-icmphdr->timestamp);
  printf("\n");
  return 0;
  }
  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){
  IcmpHeader *icmp_hdr;
  char *datapart;
  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 = 0;
  datapart = icmp_data + sizeof(IcmpHeader);
  //
  // Place some junk in the buffer.
  //
  memset(datapart,'E', datasize - sizeof(IcmpHeader));
  }
  /纨绔子弟******* 附: ping命令执行时显示的画面 纨绔子弟***\
  *  C:\Documents and Settings\houzhijiang>ping 236.56.54.12               *
  *                                                                                                      *
  *  Pinging 236.56.54.12 with 32 bytes of data:                                      *
  *                                                                                                      *
  *  Request timed out.                                                                         *
  *  Request timed out.                                                                         *
  *  Request timed out.                                                                         *
  *  Request timed out.                                                                         *
  *                                                                                                      *
  *  Ping statistics for 236.56.54.12:                                                      *
  *     Packets: Sent = 4, Received = 0, Lost = 4 (100% loss),                  *
  *                                                                                                      *
  \纨绔子弟纨绔子弟纨绔子弟纨绔子弟*********/
  /纨绔子弟纨绔子弟纨绔子弟纨绔子弟*********\
  *  C:\Documents and Settings\houzhijiang>ping 127.0.0.1                    *
  *                                                                                                     *
  *  Pinging 127.0.0.1 with 32 bytes of data:                                          *
  *                                                                                                     *
  *  Reply from 127.0.0.1: bytes=32 time<1ms TTL=128                        *
  *  Reply from 127.0.0.1: bytes=32 time<1ms TTL=128                        *
  *  Reply from 127.0.0.1: bytes=32 time<1ms TTL=128                        *
  *  Reply from 127.0.0.1: bytes=32 time<1ms TTL=128                        *
  *                                                                                                     *
  *  Ping statistics for 127.0.0.1:                                                          *
  *     Packets: Sent = 4, Received = 4, Lost = 0 (0% loss),                    *
  *  Approximate round trip times in milli-seconds:                                 *
  *     Minimum = 0ms, Maximum = 0ms, Average = 0ms                        *
  *                                                                                                    *
  \纨绔子弟纨绔子弟纨绔子弟纨绔子弟********/
 


    文章作者: 福州军威计算机技术有限公司
    军威网络是福州最专业的电脑维修公司,专业承接福州电脑维修、上门维修、IT外包、企业电脑包年维护、局域网网络布线、网吧承包等相关维修服务。
    版权声明:原创作品,允许转载,转载时请务必以超链接形式标明文章原始出处 、作者信息和声明。否则将追究法律责任。

TAG:
评论加载中...
内容:
评论者: 验证码: