一个基本的ping程序C代码

在看Steven的TCP/IP Illustrated, V1, 改了一个非常简单的ping echo的C代码，在linux上编译通,就当学习吧。

 

       1  #include  < stdio.h >
       2  #include  < sys / types.h >
       3  #include  < sys / socket.h >
       4  #include  < arpa / inet.h >
       5  #include  < netdb.h >
       6  #include  < netinet / in .h >
       7  #include  < netinet / ip_icmp.h >
       8  #include  < signal.h >
       9  #include  < errno.h >
      10
      11   static   const   int  DEFDATALEN  =   56 ;
      12   static   const   int  MAXIPLEN  =   60 ;    //  max len of ip header, 15 * 4
      13   static   const   int  MAXICMPLEN  =   76 ;  //  max len of icmp header
      14
      15   static   void  noresp( int );
      16   static   void  ping( const   char   * );
      17  u_short in_cksum(u_short  * ,  int );
      18   void  usage( void );
      19
      20   int  main( int  argc,  char   * argv[])
      21   ... {
     22
     23     if(argc != 2)
     24         usage();
     25     ping(*++argv);
     26 }
      27
      28   static   void  noresp( int  ign)
      29   ... {   // signal fun
     30     printf("No response ");
     31     exit(0);
     32 }
      33
      34   static   void  ping( const   char   * host)
      35   ... {
     36     struct hostent *hp;
     37     struct sockaddr_in target;
     38     struct icmp *packet;
     39     char outpack[DEFDATALEN + MAXIPLEN + MAXICMPLEN];
     40     int sock, c;
     41
     42     if((sock = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP)) < 0)
     43     ...{
     44         perror("socket");
     45         exit(1);
     46     }
     47
     48     memset(&target, 0, sizeof(struct sockaddr_in));
     49     target.sin_family = AF_INET;
     50     if(inet_aton(host, (struct in_addr*)&target.sin_addr.s_addr)== 0)
     51     ...{
     52         hp = gethostbyname(host);
     53         if(!hp)
     54         ...{
     55             fprintf(stderr, "ping: unknown host %s ", host);
     56             exit(1);
     57         }
     58         memcpy(&target.sin_addr, hp->h_addr, 4);
     59     }
     60
     61     memset(outpack, 0, sizeof(outpack));
     62     packet = (struct icmp*)outpack;
     63     packet->icmp_type = ICMP_ECHO;
     64     packet->icmp_id = getpid();
     65
     66     // checksum
     67     packet->icmp_cksum = in_cksum((u_short *)packet, sizeof(outpack));
     68     c = sendto(sock, packet, sizeof(outpack), 0, (struct sockaddr*)&target, sizeof(struct sockaddr_in));
     69     if(c < 0 || c != sizeof(outpack))
     70     ...{
     71         perror("sendto");
     72         exit(1);
     73     }
     74
     75     // set a timer
     76     signal(SIGALRM, noresp);
     77     alarm(5);
     78     while(1)
     79     ...{
     80         struct sockaddr_in from;
     81         size_t fromlen = sizeof(struct sockaddr_in);
     82
     83         if (( c = recvfrom(sock, outpack, sizeof(outpack), 0,
     84                     (struct sockaddr*)&from, &fromlen)) < 0)
     85         ...{
     86             if (errno == EINTR)
     87                 continue;
     88             perror("recvfrom");
     89             continue;
     90         }
     91
     92         if (c >= 76)
     93         ...{   // the request ip header is in the reply pack, skip
     94             struct iphdr *iphdr = (struct iphdr *)outpack;
     95             packet = (struct icmp *)(outpack + (iphdr->ihl<<2));
     96             if (packet->icmp_type == ICMP_ECHOREPLY)
     97                 break;
     98         }
     99     }
    100     printf("%s is alive ", inet_ntoa(target.sin_addr));
    101     return;
    102 }
     103
     104   void  usage( void )
     105   ... {
    106     struct sockaddr_in whereto;
    107     fprintf(stderr, "Usage: prg host ");
    108     exit(1);
    109 }
     110
     111  u_short in_cksum(u_short  * addr,  int  len)
     112   ... {
    113     int sum = 0, nleft = len;
    114     u_short *w = addr, ret;
    115
    116     while(nleft > 1)
    117     ...{
    118         sum += *w++;
    119         nleft -= 2;
    120     }
    121     if(nleft == 1)
    122         sum += htons(*(u_char*)w << 8);
    123
    124     sum = (sum >> 16) + (sum & 0xffff);
    125     sum += sum >> 16;
    126     ret = ~sum;
    127     return ret;
    128 }