Go net.Dialer.Control 怎样检查底层连接参数

需要在 Go 发起 TCP 连接前检查或设置 socket 参数时，可以给 net.Dialer.Control 配置回调。它发生在网络连接对象创建之后、真正拨号之前，回调会收到网络名、目标地址和一个 syscall.RawConn。通过这个 RawConn 的 Control 方法，可以在文件描述符仍然有效的短暂窗口内读取或设置底层参数。

关键点只有一句：把所有 fd 操作放进 RawConn.Control 提供的函数里，错误从回调中捕获并返回给 Dialer.Control，不要保存 fd，也不要把它当成跨平台的普通整数句柄。

先把回调的位置放对

Dialer.Control 的调用位置决定了它适合做“拨号前”的 socket 配置。Go 官方文档还说明，传入回调的网络名不一定等于调用 Dial 时的原始字符串：使用 TCP 时可能变成 tcp4 或 tcp6，UDP 和 IP 也可能带上地址族后缀。

如果同时设置了 ControlContext，Control 会被忽略。因此需要上下文时直接使用 ControlContext；只关心拨号前参数时，使用 Control 即可。

package main

import (
    "context"
    "fmt"
    "net"
    "syscall"
)

func dialWithControl(ctx context.Context, address string) (net.Conn, error) {
    d := net.Dialer{
        Control: func(network, address string, raw syscall.RawConn) error {
            // network 可能是 tcp4 或 tcp6，不能只按调用方的 tcp 判断。
            fmt.Printf("准备拨号：network=%s address=%s\\n", network, address)
            // 具体 fd 读取或设置必须继续交给 RawConn.Control。
            return nil
        },
    }
    // ctx 只约束拨号阶段；连接建立后应由连接自身管理生命周期。
    return d.DialContext(ctx, "tcp", address)
}

上面的回调没有触碰 fd，只用于说明入口位置。真正检查参数时，把系统调用放进 raw.Control，这样可以明确 fd 的有效范围。

用 RawConn.Control 读取底层参数

syscall.RawConn.Control 会把底层文件描述符或句柄交给一个函数，并保证它只在该函数执行期间有效。下面示例以 Unix 系统的 SO_KEEPALIVE 为例，展示读取 socket 选项的写法。syscall.GetsockoptInt 属于操作系统相关实现，生产代码应把这段逻辑放入对应平台文件。

package main

import (
    "fmt"
    "net"
    "syscall"
)

func inspectKeepAlive(raw syscall.RawConn) error {
    var optionErr error
    controlErr := raw.Control(func(fd uintptr) {
        // fd 只在这个函数内保证有效，不能保存到外层异步使用。
        value, err := syscall.GetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE)
        if err != nil {
            optionErr = fmt.Errorf("读取 SO_KEEPALIVE 失败: %w", err)
            return
        }
        fmt.Printf("SO_KEEPALIVE=%d\\n", value)
    })
    if controlErr != nil {
        return fmt.Errorf("取得底层连接控制权失败: %w", controlErr)
    }
    // RawConn.Control 本身和系统调用各有一条错误路径，都不能丢掉。
    return optionErr
}

func dialAndInspect(address string) (net.Conn, error) {
    d := net.Dialer{
        Control: func(network, address string, raw syscall.RawConn) error {
            // 先检查参数，再决定是否允许继续拨号。
            return inspectKeepAlive(raw)
        },
    }
    return d.Dial("tcp", address)
}

这里要特别区分两类错误：raw.Control 可能因为底层控制失败而返回错误；传入的函数内部也可能从 GetsockoptInt 得到错误。只有两条路径都返回 nil，Dial 才会继续。

需要设置参数时，仍然遵守同一个生命周期

检查只是读取；如果需求是设置拨号前选项，可以在同一个回调里调用对应平台的设置函数。比如 Unix 实现中可在 fd 上设置 SO_KEEPALIVE，但不要在连接建立后再依赖这个 Control 回调修改参数，也不要把系统调用写死成所有操作系统都支持的形式。

func setKeepAlive(raw syscall.RawConn) error {
    var optionErr error
    controlErr := raw.Control(func(fd uintptr) {
        // 这里只演示拨号前设置 socket 选项，数值和常量应按平台需求确认。
        optionErr = syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 1)
    })
    if controlErr != nil {
        return fmt.Errorf("设置底层连接参数失败: %w", controlErr)
    }
    return optionErr
}

跨平台项目可以为不同 GOOS 提供不同的 setKeepAlive 实现，公共的 Dialer.Control 只负责调用它。若需要上下文参与控制，则把同样的逻辑放进 ControlContext，不要同时配置两个回调。

四个容易写错的边界

• 不要把 fd 留到回调外。官方保证只覆盖回调执行期间，回调返回后继续使用属于未定义的生命周期假设。
• 不要只比较 network == "tcp"。地址族解析后可能收到 tcp4 或 tcp6。
• 不要忽略内部错误。给 optionErr 赋值后仍要检查 controlErr，两者都成功才能继续。
• 不要把 Unix 系统调用当成通用 Go API。把平台差异隔离到文件或小函数中，公共拨号代码只保留 RawConn 组合逻辑。

可以按下面的速查表判断实现是否放在正确位置：

相关问题

Control 回调里能不能保存文件描述符？

不建议，也不能依赖其继续有效。应在回调内部完成需要的读取或设置，并把结果或错误传回外层。

Control 和 ControlContext 应该选哪个？

不需要上下文就用 Control；需要根据上下文取消、截止时间或请求元数据做决定时用 ControlContext，但不要同时设置。

为什么回调收到的网络名不是 tcp？

Dialer 可能在解析地址族后把 TCP 细分为 tcp4 或 tcp6，这是官方文档说明的正常行为。