写了一个ping程序，但是只能收到一个reply然后就阻塞了，谁能帮我看看？

svenwang

1. 
   
2. // file: ping.cpp
   
3. #include <sys/types.h>
   
4. #include <sys/socket.h>
   
5. #include <unistd.h>
   
6. #include <arpa/inet.h>
   
7. #include <netinet/in.h>
   
8. #include <stdio.h>
   
9. #include <time.h>
   
10. #include <netinet/in_systm.h>
    
11. #include <netinet/ip.h>
    
12. #include <netinet/ip_icmp.h>
    
13. #include <sys/time.h>
    
14. #include <string.h>
    
15. #include <signal.h>
    
16. #include <netdb.h>
    
17. 
    
18. int fd = 0;
    
19. struct addrinfo *ai = NULL;
    
20. int pid = 0;
    
21. 
    
22. struct addrinfo *host_serv(const char *host, const char *serv, int family, int socktype)
    
23. {
    
24.         struct addrinfo hints, *res = NULL;
    
25.         memset(&hints, 0, sizeof(hints));
    
26.         hints.ai_flags = AI_CANONNAME;
    
27.         hints.ai_family = family;
    
28.         hints.ai_socktype = socktype;
    
29.         int n = getaddrinfo(host, serv, &hints, &res);
    
30.         if (n != 0)
    
31.                 return NULL;
    
32.         return res;
    
33. }
    
34. 
    
35. void tv_sub(struct timeval *out, struct timeval *in)
    
36. {
    
37.         out->tv_usec -= in->tv_usec;
    
38.         if (out->tv_usec < 0) {
    
39.                 --out->tv_sec;
    
40.                 out->tv_usec += 1000000;
    
41.         }
    
42.         out->tv_sec -= in->tv_sec;
    
43. }
    
44. 
    
45. unsigned short in_cksum(unsigned short *addr, int len)
    
46. {
    
47.         int left = len;
    
48.         int sum = 0;
    
49.         unsigned short *ptr = addr;
    
50.         while (left > 1) {
    
51.                 sum += *ptr++;
    
52.                 left -= 2;
    
53.         }
    
54.        
    
55.         unsigned short answer = 0;
    
56.         if (left == 1) {
    
57.                 *(unsigned char *)(&answer) = *(unsigned char *)ptr;
    
58.                 sum += answer;
    
59.         }
    
60.        
    
61.         sum = (sum >> 16) + (sum & 0xffff);
    
62.         sum += (sum >> 16);
    
63.         answer = ~sum;
    
64.         return answer;
    
65. }
    
66. 
    
67. void recv_ping(char *ptr, ssize_t len, struct timeval *tvrecv)
    
68. {
    
69.         if (len < sizeof(struct ip))
    
70.                 return;
    
71.         struct ip *ip_ptr = (struct ip *)ptr;
    
72.         int ip_hlen = (ip_ptr->ip_hl << 2);
    
73.         struct icmp *icmp_ptr = (struct icmp *)(ptr + ip_hlen);
    
74.         int icmp_len = len - ip_hlen;
    
75.         if (icmp_len < 8)
    
76.                 return;
    
77.         if (icmp_ptr->icmp_type == ICMP_ECHOREPLY) {
    
78.                 if (icmp_ptr->icmp_id != pid)
    
79.                         return;
    
80.                 if (icmp_len < 16)
    
81.                         return;
    
82.                 struct timeval *tvsend = (struct timeval *)icmp_ptr->icmp_data;
    
83.                 tv_sub(tvrecv, tvsend);
    
84.                 double rtt = tvrecv->tv_sec * 1000.0 + tvrecv->tv_usec / 1000.0;
    
85.                
    
86.                 printf("%d bytes: seq=%u, ttl=%d, rtt=%.3fms\n",
    
87.                         icmp_len,
    
88.                         icmp_ptr->icmp_seq,
    
89.                         ip_ptr->ip_ttl,
    
90.                         rtt);
    
91.         }
    
92. }
    
93. 
    
94. void send_ping(int fd, struct addrinfo *addr, int pid)
    
95. {
    
96.         static int seq = 0;
    
97.         char buff[1024];
    
98.         struct icmp *icmp_ptr = (struct icmp *)buff;
    
99.         icmp_ptr->icmp_type = ICMP_ECHO;
    
100.         icmp_ptr->icmp_code = 0;
     
101.         icmp_ptr->icmp_id = pid;
     
102.         icmp_ptr->icmp_seq = ++seq;
     
103.         memset(icmp_ptr->icmp_data, 0xa5, 56);
     
104.        
     
105.         gettimeofday((struct timeval *)icmp_ptr->icmp_data, NULL);
     
106.         int len = sizeof(timeval) + 56;
     
107.         icmp_ptr->icmp_cksum = in_cksum((u_short*)icmp_ptr, len);
     
108. 
     
109.         sendto(fd, buff, len, 0, addr->ai_addr, addr->ai_addrlen);
     
110. }
     
111. 
     
112. void sig_alarm(int signo)
     
113. {
     
114.         send_ping(fd, ai, pid);
     
115.         alarm(1);
     
116. }
     
117. 
     
118. int main(int argc, char *argv[])
     
119. {
     
120.         if (argc != 2)
     
121.         {
     
122.                 printf("Usage: %s <ip>\n", argv[0]);
     
123.                 return -1;
     
124.         }
     
125. 
     
126.         fd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);
     
127.         if (fd < 0)
     
128.         {
     
129.                 printf("Failed to create socket!\n");
     
130.                 return -1;
     
131.         }
     
132.         ai = host_serv(argv[1], NULL, 0, 0);
     
133.         if (!ai)
     
134.         {
     
135.                 printf("Failed to get address of %s\n", argv[1]);
     
136.                 return -1;
     
137.         }
     
138.        
     
139.         pid = getpid();
     
140.         signal(SIGALRM, sig_alarm);
     
141.         sig_alarm(SIGALRM);
     
142.        
     
143.         char buff[1024];
     
144.         char addr[1024];
     
145.         for (;;)
     
146.         {
     
147.                 socklen_t len = sizeof(ai->ai_addrlen);
     
148.                 ssize_t n = recvfrom(fd, buff, sizeof(buff), 0, (sockaddr*)addr, &len);
     
149.                 if (n > 0)
     
150.                 {
     
151.                         struct timeval tv;
     
152.                         gettimeofday(&tv, NULL);
     
153.                         recv_ping(buff, n, &tv);
     
154.                 }
     
155.         }
     
156.         return 0;
     
157. }
     

复制代码

编译：#g++ -g -o ping ping.cpp

svenwang

明白了，是校验和的问题。
icmp_ptr->icmp_cksum = 0; // <== 加上这一句
icmp_ptr->icmp_cksum = in_cksum((u_short*)icmp_ptr, len);