blob: 3e78f291352c916434ff254cedd6523b4105f89c [file] [log] [blame]
// Copyright 2018 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package h2c
import (
"context"
"crypto/tls"
"fmt"
"io"
"log"
"net"
"net/http"
"net/http/httptest"
"strings"
"testing"
"golang.org/x/net/http2"
)
func ExampleNewHandler() {
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Fprint(w, "Hello world")
})
h2s := &http2.Server{
// ...
}
h1s := &http.Server{
Addr: ":8080",
Handler: NewHandler(handler, h2s),
}
log.Fatal(h1s.ListenAndServe())
}
func TestContext(t *testing.T) {
baseCtx := context.WithValue(context.Background(), "testkey", "testvalue")
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if r.ProtoMajor != 2 {
t.Errorf("Request wasn't handled by h2c. Got ProtoMajor=%v", r.ProtoMajor)
}
if r.Context().Value("testkey") != "testvalue" {
t.Errorf("Request doesn't have expected base context: %v", r.Context())
}
fmt.Fprint(w, "Hello world")
})
h2s := &http2.Server{}
h1s := httptest.NewUnstartedServer(NewHandler(handler, h2s))
h1s.Config.BaseContext = func(_ net.Listener) context.Context {
return baseCtx
}
h1s.Start()
defer h1s.Close()
client := &http.Client{
Transport: &http2.Transport{
AllowHTTP: true,
DialTLS: func(network, addr string, _ *tls.Config) (net.Conn, error) {
return net.Dial(network, addr)
},
},
}
resp, err := client.Get(h1s.URL)
if err != nil {
t.Fatal(err)
}
_, err = io.ReadAll(resp.Body)
if err != nil {
t.Fatal(err)
}
if err := resp.Body.Close(); err != nil {
t.Fatal(err)
}
}
func TestPropagation(t *testing.T) {
var (
server *http.Server
// double the limit because http2 will compress header
headerSize = 1 << 11
headerLimit = 1 << 10
)
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if r.ProtoMajor != 2 {
t.Errorf("Request wasn't handled by h2c. Got ProtoMajor=%v", r.ProtoMajor)
}
if r.Context().Value(http.ServerContextKey).(*http.Server) != server {
t.Errorf("Request doesn't have expected http server: %v", r.Context())
}
if len(r.Header.Get("Long-Header")) != headerSize {
t.Errorf("Request doesn't have expected http header length: %v", len(r.Header.Get("Long-Header")))
}
fmt.Fprint(w, "Hello world")
})
h2s := &http2.Server{}
h1s := httptest.NewUnstartedServer(NewHandler(handler, h2s))
server = h1s.Config
server.MaxHeaderBytes = headerLimit
server.ConnState = func(conn net.Conn, state http.ConnState) {
t.Logf("server conn state: conn %s -> %s, status changed to %s", conn.RemoteAddr(), conn.LocalAddr(), state)
}
h1s.Start()
defer h1s.Close()
client := &http.Client{
Transport: &http2.Transport{
AllowHTTP: true,
DialTLS: func(network, addr string, _ *tls.Config) (net.Conn, error) {
conn, err := net.Dial(network, addr)
if conn != nil {
t.Logf("client dial tls: %s -> %s", conn.RemoteAddr(), conn.LocalAddr())
}
return conn, err
},
},
}
req, err := http.NewRequest("GET", h1s.URL, nil)
if err != nil {
t.Fatal(err)
}
req.Header.Set("Long-Header", strings.Repeat("A", headerSize))
_, err = client.Do(req)
if err == nil {
t.Fatal("expected server err, got nil")
}
}
func TestMaxBytesHandler(t *testing.T) {
const bodyLimit = 10
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
t.Errorf("got request, expected to be blocked by body limit")
})
h2s := &http2.Server{}
h1s := httptest.NewUnstartedServer(http.MaxBytesHandler(NewHandler(handler, h2s), bodyLimit))
h1s.Start()
defer h1s.Close()
// Wrap the body in a struct{io.Reader} to prevent it being rewound and resent.
body := "0123456789abcdef"
req, err := http.NewRequest("POST", h1s.URL, struct{ io.Reader }{strings.NewReader(body)})
if err != nil {
t.Fatal(err)
}
req.Header.Set("Http2-Settings", "")
req.Header.Set("Upgrade", "h2c")
req.Header.Set("Connection", "Upgrade, HTTP2-Settings")
resp, err := h1s.Client().Do(req)
if err != nil {
t.Fatal(err)
}
defer resp.Body.Close()
_, err = io.ReadAll(resp.Body)
if err != nil {
t.Fatal(err)
}
if got, want := resp.StatusCode, http.StatusInternalServerError; got != want {
t.Errorf("resp.StatusCode = %v, want %v", got, want)
}
}