go - 使用 golang 诊断从 unix 套接字读取非常慢(1 分钟对 netcat 中的 1 秒)

Question

背景我写了几个包来与 OpenVas 漏洞扫描器通信——扫描器使用一些不同的propitiatory协议(protocol)来通信——都是由通过 unix 套接字或 tcp 连接发送的 xml 或文本字符串组成的(我使用 unix 套接字)。

我遇到的问题是 OTP 协议(protocol)(​​没有详细记录的 OpenVas 内部协议(protocol))

我可以使用 netcat 运行以下命令，我会在一秒钟内收到回复:

echo -en '< OTP/2.0 >\nCLIENT <|> NVT_INFO\n' | ncat -U /var/run/openvassd.sock

这会导致相当大的响应，在终端中看起来像这样:

< OTP/2.0 >
SERVER <|> NVT_INFO <|> 201802131248 <|> SERVER
SERVER <|> PREFERENCES <|>
cache_folder <|> /var/cache/openvas
include_folders <|> /var/lib/openvas/plugins
max_hosts <|> 30
//lots more here

例如，我以前有一些这样的代码来读取响应:

func (c Client) read() ([]byte, error) {

    // set up buffer to read in chunks
    bufSize := 8096
    resp := []byte{}
    buf := make([]byte, bufSize)

    for {
        n, err := c.conn.Read(buf)
        resp = append(resp, buf[:n]...)
        if err != nil {
            if err != io.EOF {
                return resp, fmt.Errorf("read error: %s", err)
            }
            break
        }
        fmt.Println("got", n, "bytes.")

    }
    fmt.Println("total response size:", len(resp))

    return resp, nil
}

我得到了完整的结果，但它是小块的(我猜是逐行的)所以我看到的输出是这样的(在显示完整响应之前的一分钟左右的过程中):

got 53 bytes.
got 62 bytes.
got 55 bytes.
got 62 bytes.
got 64 bytes.
got 59 bytes.
got 58 bytes.
got 54 bytes.
got 54 bytes.
got 54 bytes.
got 64 bytes.
got 59 bytes.
... (more)

所以我决定试试 ioutil.ReadAll:

func (c Client) read() ([]byte, error) {
    fmt.Println("read start")
    d, err := ioutil.ReadAll(c.conn)
    fmt.Println("read done")
    return d, err
}

这确实会再次返回完整的响应，但与命令预计花费的 < 1 秒相比，“读取开始”和“读取完成”之间的时间大约为一分钟。

关于为什么与 netcat 相比通过 golang 读取如此慢的任何想法 - 我如何诊断/解决问题？**

Answer 1

服务似乎正在等待更多输入，并最终在一分钟后超时。在您的 CLI 示例中，一旦 echo 命令完成，管道的那一侧将关闭以进行写入，在这种情况下，服务会收到长度为 0 的 recv 通知。

为了在 Go 中执行相同的操作，您需要调用 CloseWrite在您完成发送命令后，在 net.UnixConn 上。

c.conn.(*net.UnixConn).CloseWrite()