如何在 Golang 中包装 net.Conn.Read()

Question

我想包装Read函数 net.Conn.Read()。这样做的目的是读取 SSL 握手消息。https://pkg.go.dev/net#TCPConn.Read

nc, err := net.Dial("tcp", "google.com:443")

if err != nil {

    fmt.Println(err)

}

tls.Client(nc, &tls.Config{})

有什么办法吗？

Answer 1

使用以下代码拦截 read on a net.Conn： type wrap {     // Conn is the wrapped net.Conn.     // Because it's an embedded field, the      // net.Conn methods are automatically promoted     // to wrap.     net.Conn  } // Read calls through to the wrapped read and // prints the bytes that flow through. Replace // the print statement with whatever is appropriate // for your application. func (w wrap) Read(p []byte) (int, error) {     n, err := w.Conn.Read()     fmt.Printf("%x\n", p[:n]) // example     return n, err }像这样包裹： tnc, err :=tls.Client(wrap{nc}, &tls.Config{})

Answer 2

以前的答案确实完成了工作。不过，我会推荐 Liz Rice 的演讲：GopherCon 2018：Liz Rice - The Go Programmer's Guide to Secure Connections浏览她在Github中的代码，您可能会找到一种更优雅的方式来实现您想要的。从第 26 行的客户端代码开始。