如何从 net/http 请求中读取 HTTP/2 推送帧

Question

我正在尝试编写一个 Go 客户端来测试我们的 http/2 基础架构。我想向https://mydomain.tld/somePage 发出一个http 请求，并希望收到一个html 响应，以及几个推送的资源。我想确保这些推送成功，否则失败。

我不清楚标准库的任何部分是否公开了我需要的这个功能。

我可以查看响应并检查协议版本以检测 http2。

我可以在来自https://http2-push.appspot.com/ 等发送推送的网站的响应中看到Link 标头，但我不太清楚链接标头和实际推送承诺帧之间的关系。您可以通过 http 1.1 获取链接标头，因此我不确定仅此一项能否确保推送会发生。

http2 package 有一个较低级别的Framer 接口，我可以利用它来验证原始帧，但老实说，我不知道如何设置一个并向它发出初始请求。

是否有 go 客户端如何验证 http2 推送资源的正确配置的示例？

Answer 1

使用 golang.org/x/net/http2 中的 Framer 并不难，如果我们可以得到 http.Client 自然读取的字节的副本。我们可以通过实现自己的 net.Conn 来做到这一点。

我在下面的程序中取得了一些进展，但是我没有看到预期的 PUSH_PROMISE 帧。经过一番挖掘，我发现了the Go client explicitly disables Push。在这种情况下，不允许服务器发送这些帧。我看不到更改该设置的明显方法（没有破解标准库）。

以为我仍然分享我的代码。也许我错过了一些简单的东西让它发挥作用。

package main

import (
    "bytes"
    "crypto/tls"
    "io"
    "io/ioutil"
    "log"
    "net"
    "net/http"
    "os"

    "golang.org/x/net/http2"
)

func main() {
    buf := &bytes.Buffer{}
    transport := &http2.Transport{DialTLS: dialT(buf)}
    client := &http.Client{Transport: transport}

    res, err := client.Get("https://http2-push.appspot.com/")
    if err != nil {
            log.Fatal(err)
    }

    res.Body.Close()
    res.Write(os.Stdout)

    framer := http2.NewFramer(ioutil.Discard, buf)
    for {
            f, err := framer.ReadFrame()
            if err == io.EOF || err == io.ErrUnexpectedEOF {
                    break
            }
            switch err.(type) {
            case nil:
                    log.Println(f)
            case http2.ConnectionError:
                    // Ignore. There will be many errors of type "PROTOCOL_ERROR, DATA
                    // frame with stream ID 0". Presumably we are abusing the framer.
            default:
                    log.Println(err, framer.ErrorDetail())
            }
    }
}

// dialT returns a connection that writes everything that is read to w.
func dialT(w io.Writer) func(network, addr string, cfg *tls.Config) (net.Conn, error) {
    return func(network, addr string, cfg *tls.Config) (net.Conn, error) {
            conn, err := tls.Dial(network, addr, cfg)
            return &tConn{conn, w}, err
    }
}

type tConn struct {
    net.Conn
    T io.Writer // receives everything that is read from Conn
}

func (w *tConn) Read(b []byte) (n int, err error) {
    n, err = w.Conn.Read(b)
    w.T.Write(b)
    return
}

Answer 2

patch 已提交审核。

“http2：支持在客户端消费PUSH_PROMISE流”

（github 问题具有里程碑“未计划”，希望不会显着降低其在审查队列中的优先级。）