95-PING 命令实现

学完了 ICMP 回显请求与应答报文,就可以用它来实现我们平时使用的 PING 命令了。

先来看看效果:


这里写图片描述
图1 自带的 ping 命令与我们自己实现的

1. 程序路径

本文使用的程序托管在 gitos 上:http://git.oschina.net/ivan_allen/unp

如果你已经 clone 过这个代码了,请使用 git pull 更新一下。本节程序所使用的程序路径是 unp/program/icmp/icmpecho.

2. 程序设计

ping 命令本质上就是给目标主机发送一个 ICMP 回显请求报文,然后等待对方答复 ICMP 回显应答报文,如果对方不应答,可能就是我们的主机和对方的网路不通。

ping 命令还可以计算出往返时间,其实这很简单,我们只要在发送 ICMP 报文的时候,记录一下当前时间,再接收报文的时候再记录一下时间,两个时间相减就是 RTT 时间。

ping 命令中的 ttl 字段可以从 ip 数据报首部中拿到。

程序代码如下:

#include "common.h"  #define BUF_SIZE 4096 #define ICMP_SIZE 64  char *hostname; // 主机名或 ip char recvbuf[BUF_SIZE]; // 接收缓冲区 char sendbuf[ICMP_SIZE]; // 发送缓冲区  void run(); // ping 主程序 void handler(int sig); // 信号处理函数 int nsend, nrecv; // 用于统计 int64_t start;  int main(int argc, char* argv[]) {   if (argc < 2) {     ERR_QUIT("Usage: %s <hostname or ip>\n", argv[0]);   }   hostname = argv[1];   registSignal(SIGALRM, handler);   registSignal(SIGINT, handler);   run(); }  void run() {   struct ip *ip; // ip 首部   struct icmp_echo *icmp_echo, *icmp_echo_reply; // 请求与应答   struct sockaddr_in to; // 目标主机地址   int i, len, sockfd, ret, nr;   int64_t timerecv, timesend;   double rtt;    nsend = 0; // 已经发送包的个数   nrecv = 0; // 已经接收包的个数    // 随机初始化   for (i = 0; i < ICMP_SIZE; ++i)     sendbuf[i] = "abcdefghijklmnopqrstuvwxyz"[i%26];    // 构造套接字地址,端口并没有什么用,随便填   ret = resolve(hostname, 0, &to);   if (ret < 0) ERR_EXIT("resolve");     // 创建原始套接字, 只接收承载 ICMP 协议的数据报   sockfd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);   if (sockfd < 0) ERR_EXIT("socket");    // 打印信息   WARNING("PING %s (%s) %d bytes of data.\n", hostname, inet_ntoa(to.sin_addr), ICMP_SIZE);    icmp_echo = (struct icmp_echo*)sendbuf;   ip = (struct ip*)recvbuf;    // 用于时间统计   start = now();    // 填充 icmp 回显请求报文, 这三个字段以后都不用再改,放在循环外面   icmp_echo->icmp_type = 8;   icmp_echo->icmp_code = 0;   icmp_echo->icmp_id = getpid() & 0xffff;    while(1) {     icmp_echo->icmp_cksum = 0;     icmp_echo->icmp_seq = nsend + 1; // 序号, 起始序号为 1     *((int64_t*)icmp_echo->icmp_data) = now(); // 当前时间,微秒     icmp_echo->icmp_cksum = cksum((unsigned short*)sendbuf, ICMP_SIZE); // 指定大小为 64      // 发送数据     ret = sendto(sockfd, sendbuf, ICMP_SIZE, 0, (struct sockaddr*)&to, sizeof(to));     if (ret < 0) {       if (errno == EINTR) continue;       ERR_EXIT("sendto");     }     ++nsend; // 序号加 1  again:     // 接收 ip 数据报,超时时间设置为 5 秒。     alarm(5);     nr = recvfrom(sockfd, recvbuf, BUF_SIZE, 0, NULL, NULL);     if (nr < 0) {       if (errno == EINTR) {         ERR_PRINT("TIMEDOUT.\n");         continue; // 超时,丢包       }       ERR_EXIT("recvfrom");     }      // 从 ip 数据报里拿到 icmp 报文     icmp_echo_reply = (struct icmp_echo*)((char*)ip + (ip->ip_hl << 2));     if (icmp_echo_reply->icmp_type != 0 || icmp_echo_reply->icmp_code != 0         || icmp_echo_reply->icmp_id != (getpid() & 0xffff)) {       // 收到的不是我们想要的包,重来继续收,不能使用 continue 是因为我们不能判定包就丢失了       goto again;     }     // 执行到这里说明收到了自己的包,将接收个数递增     ++nrecv;     // 计算往返时间     timerecv = now();     timesend = *((int64_t*)icmp_echo_reply->icmp_data);     rtt = (timerecv - timesend) / 1000.0;     LOG("%d bytes from %s (%s): icmp_seq=%d ttl=%d time=%.1f ms\n",         ICMP_SIZE, hostname, inet_ntoa(to.sin_addr), icmp_echo_reply->icmp_seq, ip->ip_ttl, rtt);     sleep(1);   } }  void handler(int sig) {   int64_t end;   if (sig == SIGINT) {     end = now();     WARNING("\n--- %s ping statistics ---\n", hostname);     WARNING("%d packets transmitted, %d received, %d%% packet loss, time %dms\n",         nsend, nrecv, (nsend - nrecv) * 100 / nsend, (int)(end - start) / 1000);     exit(0);   } }

这段代码没有什么特别的地方,唯一要注意的就是为了防止对端没有回应应答报文而导致 recvfrom 阻塞,我们需要给 recvfrom 设置超时时间。使用 alarm 信号为读写 I/O 设置超时时间是惯用的技俩,这些以前博客里都写过。

代码中使用了 goto 语句,当然你也可以使用 while 循环,但在这里使用 goto 反而能让代码可读性更好。一味的抨击 goto 是不正确的,任何事物都有两面性。

3. 实验

  • ping GitHub 首页

选 GitHub 是因为这个网站打开的速度很慢。


这里写图片描述
图2 ping GitHub

  • ping Google 首页

选它是因为它在国内被墙,所以肯定会超时。


这里写图片描述
图3 ping Google.com

  • ping 百度首页


这里写图片描述
图4 ping 百度首页

ping 国内的主机速度还是相当快的。

4. 总结

  • 掌握 ping 命令原理

说明:本文转自blog.csdn.net,用于学习交流分享,仅代表原文作者观点。如有疑问,请联系我们删除~