利用Go语言实现简单Ping过程的方法

一、准备工作

安装最新的go

1、由于google被墙的原因，如果没有vpn的话，就到这里下载：http://www.golangtc.com/download

2、使用任意文本编辑器，或者liteide会比较方便编译和调试

二、编码

要用到的package：

import (
 "bytes"
 "container/list"
 "encoding/binary"
 "fmt"
 "net"
 "os"
 "time"
)

1、使用golang提供的net包中的相关函数可以快速构造一个ip包并自定义其中一些关键参数，而不需要再自己手动填充ip报文。

2、使用encoding/binary包可以轻松获取结构体struct的内存数据并且可以规定字节序（这里要用网络字节序bigendian），而不需要自己去转换字节序。之前的一片文中使用boost，还要自己去实现转换过程

3、使用container/list包，方便进行结果统计

4、使用time包实现耗时和超时处理

icmp报文struct：

type icmp struct {
 type    uint8
 code    uint8
 checksum  uint16
 identifier uint16
 sequencenum uint16
}

usage提示：

arg_num := len(os.args)
 if arg_num < 2 {
 fmt.print(
  "please runas [super user] in [terminal].\n",
  "usage:\n",
  "\tgoping url\n",
  "\texample: goping www.baidu.com",
 )
 time.sleep(5e9)
 return
 }

注意这个ping程序，包括之前的arp程序都必须使用系统最高权限执行，所以这里先给出提示，使用time.sleep(5e9) ，暂停5秒，是为了使双击执行者看到提示，避免控制台一闪而过。

关键net对象的创建和初始化：

var (
 icmp   icmp
 laddr  = net.ipaddr{ip: net.parseip("0.0.0.0")}
 raddr, _ = net.resolveipaddr("ip", os.args[1])
 )
 conn, err := net.dialip("ip4:icmp", &laddr, raddr)
 if err != nil {
 fmt.println(err.error())
 return
 }
 defer conn.close()

net.dialip表示生成一个ip报文，版本号是v4，协议是icmp（这里字符串ip4:icmp会把ip报文的协议字段设为1表示icmp协议），

源地址laddr可以是0.0.0.0也可以是自己的ip，这个并不影响icmp的工作。

目的地址raddr是一个url，这里使用resolve进行dns解析，注意返回值是一个指针，所以下面的dialip方法中参数表示没有取地址符。

这样一个完整的ip报文就装配好了，我们并没有去操心ip中的其他一些字段，go已经为我们处理好了。

通过返回的conn *net.ipconn对象可以进行后续操作。

defer conn.close() 表示该函数将在return时被执行，确保不会忘记关闭。

下面需要构造icmp报文了：

icmp.type = 8
 icmp.code = 0
 icmp.checksum = 0
 icmp.identifier = 0
 icmp.sequencenum = 0
 var buffer bytes.buffer
 binary.write(&buffer, binary.bigendian, icmp)
 icmp.checksum = checksum(buffer.bytes())
 buffer.reset()
 binary.write(&buffer, binary.bigendian, icmp)

仍然非常简单，利用binary可以把一个结构体数据按照指定的字节序读到缓冲区里面，计算校验和后，再读进去。

检验和算法参考上面给出的url中的实现：

func checksum(data []byte) uint16 {
 var (
 sum  uint32
 length int = len(data)
 index int
 )
 for length > 1 {
 sum += uint32(data[index])<<8 + uint32(data[index+1])
 index += 2
 length -= 2
 }
 if length > 0 {
 sum += uint32(data[index])
 }
 sum += (sum >> 16)
 return uint16(^sum)
}

下面是ping的request过程，这里仿照windows的ping，默认只进行4次：

fmt.printf("\n正在 ping %s 具有 0 字节的数据:\n", raddr.string())
 recv := make([]byte, 1024)
 statistic := list.new()
 sended_packets := 0
 for i := 4; i > 0; i-- {
 if _, err := conn.write(buffer.bytes()); err != nil {
  fmt.println(err.error())
  return
 }
 sended_packets++
 t_start := time.now()
 conn.setreaddeadline((time.now().add(time.second * 5)))
 _, err := conn.read(recv)
 if err != nil {
  fmt.println("请求超时")
  continue
 }
 t_end := time.now()
 dur := t_end.sub(t_start).nanoseconds() / 1e6
 fmt.printf("来自 %s 的回复: 时间 = %dms\n", raddr.string(), dur)
 statistic.pushback(dur)
 //for i := 0; i < recvsize; i++ {
 // if i%16 == 0 {
 // fmt.println("")
 // }
 // fmt.printf("%.2x ", recv[i])
 //}
 //fmt.println("")
 }

"具有0字节的数据"表示icmp报文中没有数据字段，这和windows里面32字节的数据的略有不同。

conn.write方法执行之后也就发送了一条icmp请求，同时进行计时和计次。

conn.setreaddeadline可以在未收到数据的指定时间内停止read等待，并返回错误err，然后判定请求超时。否则，收到回应后，计算来回所用时间，并放入一个list方便后续统计。

注释部分内容是我在探索返回数据时的代码，读者可以试试看read到的数据是哪个数据包的？

统计工作将在循环结束时进行，这里使用了defer其实是希望按了ctrl+c之后能return执行，但是控制台确实不给力，直接给杀掉了。。

defer func() {
 fmt.println("")
 //信息统计
 var min, max, sum int64
 if statistic.len() == 0 {
  min, max, sum = 0, 0, 0
 } else {
  min, max, sum = statistic.front().value.(int64), statistic.front().value.(int64), int64(0)
 }
 for v := statistic.front(); v != nil; v = v.next() {
  val := v.value.(int64)
  switch {
  case val < min:
  min = val
  case val > max:
  max = val
  }
  sum = sum + val
 }
 recved, losted := statistic.len(), sended_packets-statistic.len()
 fmt.printf("%s 的 ping 统计信息：\n 数据包：已发送 = %d，已接收 = %d，丢失 = %d (%.1f%% 丢失)，\n往返行程的估计时间(以毫秒为单位)：\n 最短 = %dms，最长 = %dms，平均 = %.0fms\n",
  raddr.string(),
  sended_packets, recved, losted, float32(losted)/float32(sended_packets)*100,
  min, max, float32(sum)/float32(recved),
 )
 }()

统计过程注意类型的转换和格式化就行了。

全部代码就这些，执行结果大概是这个样子的：

 

注意每次ping后都没有"休息"，不像windows或者linux的会停顿几秒再ping下一轮。

总结

golang实现整个ping比我想象中的还要简单很多，静态编译速度是十分快速，相比c而言，你需要更多得了解底层，甚至要从链路层开始，你需要写更多更复杂的代码来完成相同的工作，但究其根本，c语言仍然是鼻祖，功不可没，很多原理和思想都要继承和发展，这一点golang做的很好。以上就是这篇文章的全部内容，希望对大家的学习或者工作带来一定的帮助，如果有疑问大家可以留言交流。