文档章节

使用Golang实现简单Ping过程

Micooz
 Micooz
发布于 2014/08/11 21:21
字数 1611
阅读 12446
收藏 154

引言

    关于各种语言实现Ping已经是大家喜闻乐见的事情了,网络上利用Golang实现Ping已经有比较详细的代码示例,但大多是仅仅是实现了Request过程,而对Response的回显内容并没有做接收。而Ping程序不仅仅是发送一个ICMP,更重要的是如何接收并进行统计。

    下面是网络上几篇关于Ping的实现代码:

    https://github.com/paulstuart/ping/blob/master/ping.go

    http://blog.csdn.net/gophers/article/details/21481447

    http://blog.csdn.net/laputa73/article/details/17226337

本文借鉴了第二个链接里面的部分代码。

准备

  1. 安装最新的Go

    由于Google被墙的原因,如果没有VPN的话,就到这里下载:

    http://www.golangtc.com/download

  2. 使用任意文本编辑器,或者LiteIDE会比较方便编译和调试,下面是LiteIDE的下载地址

    https://github.com/visualfc/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做的很好。


© 著作权归作者所有

Micooz
粉丝 40
博文 34
码字总数 51094
作品 1
其它
其他
私信 提问
加载中

评论(13)

OSC首席键客
OSC首席键客
ttl怎么搞?
Micooz
Micooz 博主

引用来自“L_world_125”的评论

golang是个什么鬼。

Go语言
L_world_125
L_world_125
golang是个什么鬼。
yangjh_chs
yangjh_chs
谢谢,很清晰
Silvery
Silvery

引用来自“Silvery”的评论

"相比C而言,你需要更多得了解底层,甚至要从链路层开始" -- 无语。

C代码:https://github.com/Silveryfu/Projects-SFU/blob/master/CMPT471/RawSocket.cpp

引用来自“Micooz”的评论

这个代码的确是从链路层开始封装的:ether_header->ip_header ->icmp_header,上述go代码仅涉及icmp_header
哈哈, sorry 我这才意识到自己误会你原文的意思了。
LongRaindy
LongRaindy
对于我这种go新手实用啊,谢了。
yunfound
yunfound

引用来自“Silvery”的评论

"相比C而言,你需要更多得了解底层,甚至要从链路层开始" -- 无语。

C代码:https://github.com/Silveryfu/Projects-SFU/blob/master/CMPT471/RawSocket.cpp
太复杂了,没看懂!!!
Micooz
Micooz 博主

引用来自“Silvery”的评论

"相比C而言,你需要更多得了解底层,甚至要从链路层开始" -- 无语。

C代码:https://github.com/Silveryfu/Projects-SFU/blob/master/CMPT471/RawSocket.cpp
这个代码的确是从链路层开始封装的:ether_header->ip_header ->icmp_header,上述go代码仅涉及icmp_header
Silvery
Silvery
"相比C而言,你需要更多得了解底层,甚至要从链路层开始" -- 无语。

C代码:https://github.com/Silveryfu/Projects-SFU/blob/master/CMPT471/RawSocket.cpp
金茧
金茧

引用来自“包菜兄”的评论

赞,回家跟着实现一下看看
实现好了,分享一下。
golang入门学习笔记(二)

作者: 一字马胡 转载标志 【2017-11-22】 更新日志 日期 更新内容 备注 2017-11-22 新建文章 go语言入门学习笔记(二) golang入门学习笔记系列 golang入门学习笔记(一) interface for go...

一字马胡
2017/11/22
0
0
深入分析golang多值返回以及闭包的实现

一、前言 golang有很多新颖的特性,不知道大家的使用的时候,有没想过,这些特性是如何实现的?当然你可能会说,不了解这些特性好像也不影响自己使用golang,你说的也有道理,但是,多了解底...

万建宁
2018/07/17
0
0
go 数据库相关

golang内部自带了连接池功能,所以只需要全局创建单例的db就可以了 创建连接池 创建db后并不需要自己去使用db.Close()去关闭,虽然提供了这样关闭方法但是完全不需要使用。 只要像下面这样创...

solate
2016/10/31
22
0
Go言语 linux环境 exec Command 执行长命令失败

无论windows还是linux/unix,程序都可以通过命令行执行。 golang提供了os/exec包,来完成调用过程。 构造命令命令的构造是调用Command函数: func Command(name string, arg ...string) *Cm...

海飞飞
2013/05/08
6.4K
3
Golang中使用log(二):Golang 标准库log的实现

前一篇文章我们看到了Golang标准库中log模块的使用,那么它是如何实现的呢?下面我从log.Logger开始逐步分析其实现。 其源码可以参考官方地址 1.Logger结构 首先来看下类型Logger的定义: ty...

gotaly
2018/06/28
0
0

没有更多内容

加载失败,请刷新页面

加载更多

Linux的基本命令

目录的操作命令(增删改查) 增: mkdir 目录名称; 查: ls 可以看到该目录下的所有的目录和文件 ls -a,可以看到该目录下的所有文件和目录,包括隐藏的 ls -l,可以看到该目录下的所有目录和...

凹凸凸
今天
2
0
在古老unix中增加新用户

Installing 4.3 BSD Quasijarus on SIMH 目标:要在4.3BSD中新增加用户dmr,指定目录/home/dmr,uid为10 gid=31(guest组,系统已建立) 4.3BSD还没有adduser或useradd 直接修改/etc/passwd...

wangxuwei
今天
2
0
Bootstrap(六)表单样式

基本样式 所有设置了 .form-control 类的 <input>、<textarea> 和 <select> 元素都将被默认设置宽度属性为 width: 100%;。 将 label 元素和前面提到的控件包裹在 .form-group 中可以获得最好...

ZeroBit
昨天
3
0
SSL 证书格式转换

SSL 证书格式转换 不同服务器情况下,需要不同的证书格式。 比如 pem 转 pfx。 pem在window 平台下可以导入,但是无法正常使用。 需要转换成pfx。 推荐在线转换工具,由中国数字证书网站提供...

DrChenXX
昨天
2
0
HAProxy

xx

Canaan_
昨天
2
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部