go语言ping实现

package main import ( "flag" "fmt" "net" "os" "strconv" "time" ) func main() { var count int var timeout int64 var size int var neverstop bool flag.Int64Var(&timeout, "w", 1000, "等待每次回复的超时时间(毫秒)。") flag.IntVar(&count, "n", 4, "要发送的回显请求数。") flag.IntVar(&size, "l", 32, "要发送缓冲区大小。") flag.BoolVar(&neverstop, "t", true, "Ping 指定的主机,直到停止。") flag.Parse() args := flag.Args() if len(args) < 1 { fmt.Println("Usage: ", os.Args[0], "host") flag.PrintDefaults() flag.Usage() os.Exit(1) } ch := make(chan int) argsmap := map[string]interface{}{} argsmap["w"] = timeout argsmap["n"] = count argsmap["l"] = size argsmap["t"] = neverstop for _, host := range args { go ping(host, ch, argsmap) } for i := 0; i < len(args); i++ { <-ch } os.Exit(0) } func ping(host string, c chan int, args map[string]interface{}) { var count int var size int var timeout int64 var neverstop bool count = args["n"].(int) size = args["l"].(int) timeout = args["w"].(int64) neverstop = args["t"].(bool) cname, _ := net.LookupCNAME(host) starttime := time.Now() conn, err := net.DialTimeout("ip4:icmp", host, time.Duration(timeout*1000*1000)) ip := conn.RemoteAddr() fmt.Println("正在 Ping " + cname + " [" + ip.String() + "] 具有 32 字节的数据:") var seq int16 = 1 id0, id1 := genidentifier(host) const ECHO_REQUEST_HEAD_LEN = 8 sendN := 0 recvN := 0 lostN := 0 shortT := -1 longT := -1 sumT := 0 for count > 0 || neverstop { sendN++ var msg []byte = make([]byte, size+ECHO_REQUEST_HEAD_LEN) msg[0] = 8 // echo msg[1] = 0 // code 0 msg[2] = 0 // checksum msg[3] = 0 // checksum msg[4], msg[5] = id0, id1 //identifier[0] identifier[1] msg[6], msg[7] = gensequence(seq) //sequence[0], sequence[1] length := size + ECHO_REQUEST_HEAD_LEN check := checkSum(msg[0:length]) msg[2] = byte(check >> 8) msg[3] = byte(check & 255) conn, err = net.DialTimeout("ip:icmp", host, time.Duration(timeout*1000*1000)) checkError(err) starttime = time.Now() conn.SetDeadline(starttime.Add(time.Duration(timeout * 1000 * 1000))) _, err = conn.Write(msg[0:length]) const ECHO_REPLY_HEAD_LEN = 20 var receive []byte = make([]byte, ECHO_REPLY_HEAD_LEN+length) n, err := conn.Read(receive) _ = n var endduration int = int(int64(time.Since(starttime)) / (1000 * 1000)) sumT += endduration time.Sleep(1000 * 1000 * 1000) if err != nil || receive[ECHO_REPLY_HEAD_LEN+4] != msg[4] || receive[ECHO_REPLY_HEAD_LEN+5] != msg[5] || receive[ECHO_REPLY_HEAD_LEN+6] != msg[6] || receive[ECHO_REPLY_HEAD_LEN+7] != msg[7] || endduration >= int(timeout) || receive[ECHO_REPLY_HEAD_LEN] == 11 { lostN++ fmt.Println("对 " + cname + "[" + ip.String() + "]" + " 的请求超时。") } else { if shortT == -1 { shortT = endduration } else if shortT > endduration { shortT = endduration } if longT == -1 { longT = endduration } else if longT < endduration { longT = endduration } recvN++ ttl := int(receive[8]) // fmt.Println(ttl) fmt.Println("来自 " + cname + "[" + ip.String() + "]" + " 的回复: 字节=32 时间=" + strconv.Itoa(endduration) + "ms TTL=" + strconv.Itoa(ttl)) } seq++ count-- } stat(ip.String(), sendN, lostN, recvN, shortT, longT, sumT) c <- 1 } func checkSum(msg []byte) uint16 { sum := 0 length := len(msg) for i := 0; i < length-1; i += 2 { sum += int(msg[i])*256 + int(msg[i+1]) } if length%2 == 1 { sum += int(msg[length-1]) * 256 // notice here, why *256? } sum = (sum >> 16) + (sum & 0xffff) sum += (sum >> 16) var answer uint16 = uint16(^sum) return answer } func checkError(err error) { if err != nil { fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error()) os.Exit(1) } } func gensequence(v int16) (byte, byte) { ret1 := byte(v >> 8) ret2 := byte(v & 255) return ret1, ret2 } func genidentifier(host string) (byte, byte) { return host[0], host[1] } func stat(ip string, sendN int, lostN int, recvN int, shortT int, longT int, sumT int) { fmt.Println() fmt.Println(ip, " 的 Ping 统计信息:") fmt.Printf(" 数据包: 已发送 = %d,已接收 = %d,丢失 = %d (%d%% 丢失),\n", sendN, recvN, lostN, int(lostN*100/sendN)) fmt.Println("往返行程的估计时间(以毫秒为单位):") if recvN != 0 { fmt.Printf(" 最短 = %dms,最长 = %dms,平均 = %dms\n", shortT, longT, sumT/sendN) } }

本文分享自微信公众号 - Golang语言社区(Golangweb)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2017-09-27

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • Golang实现ping

    ICMP部分的结构 报头 ICMP报头从IP报头的第160位开始,即第20个字节开始(除非使用了IP报头的可选部分)。 Bits 160-167 168-175...

    李海彬
  • Golang实现ping

    在使用Go语言的net.Dial函数时,发送echo request报文时,不用考虑i前20个字节的ip头;但是在接收到echo response消息时,前20...

    李海彬
  • 【Go 语言社区】GO语言练习:网络编程 ICMP 示例

    1、代码 2、编译及运行 ---- 1、Go语言网络编程:ICMP示例代码 icmptest.go 1 package main 2 3 import...

    李海彬
  • Golang实现ping

    ICMP部分的结构 报头 ICMP报头从IP报头的第160位开始,即第20个字节开始(除非使用了IP报头的可选部分)。 Bits 160-167 168-175...

    李海彬
  • Golang实现ping

    在使用Go语言的net.Dial函数时,发送echo request报文时,不用考虑i前20个字节的ip头;但是在接收到echo response消息时,前20...

    李海彬
  • 排序----选择排序

    SuperHeroes
  • 15:Challenge 11(主席树裸题)

    总时间限制: 10000ms单个测试点时间限制: 1000ms内存限制: 262144kB描述 给一个长为N的数列,有M次操作,每次操作是以下两种之一: (1)...

    attack
  • 15(进程间通信)

    管道是Unix系统IPC最古老的方式。管道有下列两种局限性: (1) 历史上,它们是半双工的(即数据只能在一个方向上流动)。 (2) 它们只能在具有公共...

    提莫队长
  • Determining 32 vs 64 bit in C++

    2.size of pointer 通常情况下,在32位平台上一个指针的宽度为4bytes,而在64位平台上位8bytes.

    Rock_Lee
  • 洛谷P4064 [JXOI2017]加法(贪心 差分)

    首先二分一个答案,然后check是否可行。check的时候我们需要对每个位置\(i\),维护出所有左端点在\(i\)左侧,右端点在\(i\)右侧的所有区间。最优...

    attack

扫码关注云+社区

领取腾讯云代金券