golang实现ping命令的完整代码

这篇文章将为大家详细讲解有关golang实现ping命令的完整代码，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。

1. 导入必要的库

import (
    "context"
    "fmt"
    "net"
    "os"
)

2. 定义 ping 函数

func ping(host string) error {
    // 创建一个带有超时设置的上下文
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    // 拨号到主机并检查连接
    conn, err := net.DialTimeout("ip4:icmp", host, 1*time.Second)
    if err != nil {
        return err
    }
    defer conn.Close()

    // 发送 ICMP 回显请求
    req := &net.ICMPv4Message{
        Type: net.ICMPv4EchoRequest,
        Code: 0,
        Body: &net.ICMPv4Echo{
            Data: []byte("ping"),
        },
    }

    // 发送请求并接收响应
    if err := conn.SetDeadline(ctx.Deadline()); err != nil {
        return err
    }
    r := make([]byte, 64)
    if _, err := conn.Write(req.Marshal()); err != nil {
        return err
    }
    if _, err := conn.Read(r); err != nil {
        return err
    }

    // 解析响应
    resp := &net.ICMPv4Message{}
    if err := resp.Unmarshal(r); err != nil {
        return err
    }

    // 检查响应是否成功
    if resp.Type != net.ICMPv4EchoReply {
        return fmt.Errorf("received unexpected ICMP type %d", resp.Type)
    }

    // 打印响应时间
    fmt.Printf("Round trip time: %s
", time.Since(req.Timestamp))

    return nil
}

3. 处理用户输入

func main() {
    if len(os.Args) != 2 {
        fmt.Println("Usage: ping <host>")
        return
    }

    if err := ping(os.Args[1]); err != nil {
        fmt.Println(err)
    }
}

以上就是golang实现ping命令的完整代码的详细内容，更多请关注编程学习网其它相关文章！