使用Golang實現簡單Ping過程

jopen 10年前發布 | 32K 次閱讀 Google Go/Golang開發 Golang

Ping的基本原理是發送和接受ICMP請求回顯報文,利用Go語言可以輕松實現這一過程,較之C/C++語言,Go的實現過程十分簡單,效率和安全性也十分完美,本文將一步一步帶著大家來實現Ping

    關于各種語言實現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被墻的原因,如果沒有V*N的話,就到這里下載:

    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做的很好。

來自:http://my.oschina.net/ybusad/blog/300155

 本文由用戶 jopen 自行上傳分享,僅供網友學習交流。所有權歸原作者,若您的權利被侵害,請聯系管理員。
 轉載本站原創文章,請注明出處,并保留原始鏈接、圖片水印。
 本站是一個以用戶分享為主的開源技術平臺,歡迎各類分享!