主页 / 问题 / 839259
Accepted
Eugene X
Asked:2020-06-08 04:42:24 +0800 CST2020-06-08 04:42:24 +0800 CST

Go：net/http.ListenAndServe 的套接字挂钩

  • 772

您能告诉我如何在 HTTP 服务器本身接收请求之前预先更改请求吗？我正在为微控制器开发。以前，该应用程序在 MQTT 上工作，但随着项目的发展，需要一个新的 - 减少微控制器内存的负载和更稳定的解决方案。决定以HTTP协议为基础，但需要加密一点。也就是说，客户端不会向服务器发送“GET / HTTP/1.1\n\n”，而是更像base64的东西。

所以任务是动态解析请求，转换并发送一个正常的请求到ServeMux，它已经被正常处理了。但这是运气不好，我不知道如何在飞行中做到这一点。有没有人有任何简单的解决方案？

http

1 个回答

  1. Best Answer

    你想要的是一些变态。需要加密 - 使用 HTTPS。但是，您很幸运 Go 标准库可以容纳这种变态。首先，您需要自己的net.Conn：

    type base64TCPConn struct {
    net.Conn
    io.Reader
    io.WriteCloser
}

func (conn base64TCPConn) Read(b []byte) (n int, err error) {
    return conn.Reader.Read(b)
}

func (conn base64TCPConn) Write(b []byte) (n int, err error) {
    n, err = conn.WriteCloser.Write(b)
    if err != nil {
        return n, err
    }

    err = conn.WriteCloser.Close()
    return n, err
}

func (conn base64TCPConn) Close() error {
    err := conn.WriteCloser.Close()
    if err != nil {
        return err
    }

    return conn.Conn.Close()
}

    其次，拥有net.Listener，发行这些net.Conn：

    type base64TCPListener struct {
    net.Listener
}

func (l base64TCPListener) Accept() (conn net.Conn, err error) {
    conn, err = l.Listener.Accept()
    if err != nil {
        return nil, err
    }

    return base64TCPConn{
        Conn:   conn,
        Reader: base64.NewDecoder(base64.StdEncoding, conn),
        WriteCloser: base64.NewEncoder(base64.StdEncoding, conn),
    }, err
}

    第三，将其与net/http.Server：

    mux := http.NewServeMux()
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
    io.WriteString(w, "Hello, World!")
})
srv := &http.Server{Handler: mux}

tcpL, err := net.ListenTCP("tcp4", &net.TCPAddr{
    IP:   net.IPv4(127, 0, 0, 1),
    Port: 8080,
})
// Check err.

srv.Serve(base64TCPListener{tcpL})

    四、运行测试：

    $ echo $'GET / HTTP/1.0\r\n\r\n' | base64 | nc 127.0.0.1 8080 | base64 -d
HTTP/1.0 200 OK
Date: Fri, 08 Jun 2018 10:15:23 GMT
Content-Length: 13
Content-Type: text/plain; charset=utf-8

Hello, World!

    完整示例： https: //play.golang.org/p/RP6OKgSSOhV。

    • 4

相关问题