设置 Golang 中 TCP 连接的 IP 标头 ToS 字段的方法

在 Go 语言中设置 TCP 连接的 IP 标头 ToS 字段对于确定不同流量的优先级至关重要。然而，通过使用 `ipv4.newconn()` 或自定义拨号器来设置 ToS 字段的方法在某些情况下可能无效。为了解决这个问题，可以利用 `Dial()` 函数的 `Control` 方法，该方法允许在创建连接时控制套接字选项。通过将 `Control` 方法设置为一个函数，可以在该函数中使用 `syscall.SetsockoptInt()` 函数设置 ToS 字段的值。这种方法在 Go 1.15.5 版本中经过验证有效，为设置 ToS 字段提供了可靠且高效的解决方案。

我正在尝试使用 golang 创建 tcp 服务器和客户端，我可以在 ip 标头中设置服务类型字段，以便确定不同流量的优先级。

客户端和服务器能够通信，但我不知道如何设置 tos 字段。

我尝试按照此处描述的方法使用 ipv4 golang 包：https://godoc.org/golang.org/x/net/ipv4#newconn

简化的服务器示例：

func main () {
ln, err := net.listen("tcp4", "192.168.0.20:1024")
if err != nil {
    // error handling
}
defer ln.close()
for {
    c, err := ln.accept()
    if err != nil {
        // error handling
    }
    go func(c net.conn) {
        defer c.close()

        if err := ipv4.newconn(c).settos(0x28); err != nil {
            fmt.println("error: ", err.error())
        }
    }(c)
}

以及对应的客户端（也简化了）

func main () {

    conn, err := net.dial("tcp4", "192.168.0.20:1024")
    if err != nil {
        fmt.println(err)
    }

    for {

            writer := bufio.newwriter(conn)
            // create "packet"
            data := make([]byte, 1200)
            endline := "\r\n"



            //set packetlength
            length := strconv.formatint(int64(1200), 10)
            copy(data[0:], length)

            //set id
            idstring := strconv.formatint(int64(1), 10)
            if strings.contains(idstring, "\r") || strings.contains(idstring, "\n") || strings.contains(idstring, "\r\n") {
                fmt.println("this is gonna result in an error in the id string.")
            }
            idbuf := []byte(idstring)
            copy(data[15:], idbuf)

            //set timestamp
            timestamp0 := time.now().unixnano()
            timestampstring := strconv.formatint(timestamp0, 10)
            if strings.contains(timestampstring, "\r") || strings.contains(timestampstring, "\n") || strings.contains(timestampstring, "\r\n") {
                fmt.println("this is gonna result in an error in the timestamp string.")
            }
            buf := []byte(timestampstring)
            copy(data[50:], buf)

            copy(data[int(1200)-2:], endline)

            if len(data) != int(1200) {
                fmt.println("this is also gonna be an error. length is: ", len(data))
            }
            //send the data and flush the writer
            writer.write(data)
            writer.flush()
        }
        //time.sleep(1*time.nanosecond)
    }

我还尝试使用控制函数创建自己的拨号器，该函数传递系统调用以便设置套接字，如下所示：

dialer := &net.Dialer{
        Timeout:       5 * time.Second,
        Deadline:      time.Time{},
        LocalAddr:     tcpAddr,
        DualStack:     false,
        FallbackDelay: 0,
        KeepAlive:     0,
        Resolver:      nil,
        Control:       highPrio,
    }

    func highPrio(network, address string, c syscall.RawConn) error {
    return c.Control(func(fd uintptr) {
        // set the socket options
        err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.IPPROTO_IP, syscall.IP_TOS, 128)
        if err != nil {
            log.Println("setsocketopt: ", err)
        }
    })

我通过使用 wireshark 检查流量来验证它是否不起作用，并使用 windows 10 pro 作为我的操作系统。

解决方案

我正在尝试使用 golang 1.15.5 在 dial() 上设置 tos 方法，并且它有效：

dialer := net.Dialer{
    Timeout: this.TcpWaitConnectTimeout,
}
dialer.Control = func(network, address string, c syscall.RawConn) error {
    var err error
    c.Control(func(fd uintptr) {
        err = syscall.SetsockoptInt(int(fd), syscall.IPPROTO_IP, syscall.IP_TOS, 0x80)
    })
    return err
}

c, err := dialer.Dial("tcp", this.serverAddr)

以上就是《设置 Golang 中 TCP 连接的 IP 标头 ToS 字段的方法》的详细内容，更多关于的资料请关注golang学习网公众号！