Unvendor x/net/http2
This commit is contained in:
parent
37c8e780bc
commit
64a4bfbd8a
@ -208,8 +208,8 @@ func serveIndex(w http.ResponseWriter, r *http.Request) {
|
|||||||
|
|
||||||
cookie, err := r.Cookie("push")
|
cookie, err := r.Cookie("push")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = pusher.Push("/"+files[1].Path, options)
|
pusher.Push("/"+files[1].Path, options)
|
||||||
err = pusher.Push("/"+files[0].Path, options)
|
pusher.Push("/"+files[0].Path, options)
|
||||||
setPushCookie(w, r)
|
setPushCookie(w, r)
|
||||||
} else {
|
} else {
|
||||||
pushed := false
|
pushed := false
|
||||||
|
51
vendor/golang.org/x/net/http2/Dockerfile
generated
vendored
51
vendor/golang.org/x/net/http2/Dockerfile
generated
vendored
@ -1,51 +0,0 @@
|
|||||||
#
|
|
||||||
# This Dockerfile builds a recent curl with HTTP/2 client support, using
|
|
||||||
# a recent nghttp2 build.
|
|
||||||
#
|
|
||||||
# See the Makefile for how to tag it. If Docker and that image is found, the
|
|
||||||
# Go tests use this curl binary for integration tests.
|
|
||||||
#
|
|
||||||
|
|
||||||
FROM ubuntu:trusty
|
|
||||||
|
|
||||||
RUN apt-get update && \
|
|
||||||
apt-get upgrade -y && \
|
|
||||||
apt-get install -y git-core build-essential wget
|
|
||||||
|
|
||||||
RUN apt-get install -y --no-install-recommends \
|
|
||||||
autotools-dev libtool pkg-config zlib1g-dev \
|
|
||||||
libcunit1-dev libssl-dev libxml2-dev libevent-dev \
|
|
||||||
automake autoconf
|
|
||||||
|
|
||||||
# The list of packages nghttp2 recommends for h2load:
|
|
||||||
RUN apt-get install -y --no-install-recommends make binutils \
|
|
||||||
autoconf automake autotools-dev \
|
|
||||||
libtool pkg-config zlib1g-dev libcunit1-dev libssl-dev libxml2-dev \
|
|
||||||
libev-dev libevent-dev libjansson-dev libjemalloc-dev \
|
|
||||||
cython python3.4-dev python-setuptools
|
|
||||||
|
|
||||||
# Note: setting NGHTTP2_VER before the git clone, so an old git clone isn't cached:
|
|
||||||
ENV NGHTTP2_VER 895da9a
|
|
||||||
RUN cd /root && git clone https://github.com/tatsuhiro-t/nghttp2.git
|
|
||||||
|
|
||||||
WORKDIR /root/nghttp2
|
|
||||||
RUN git reset --hard $NGHTTP2_VER
|
|
||||||
RUN autoreconf -i
|
|
||||||
RUN automake
|
|
||||||
RUN autoconf
|
|
||||||
RUN ./configure
|
|
||||||
RUN make
|
|
||||||
RUN make install
|
|
||||||
|
|
||||||
WORKDIR /root
|
|
||||||
RUN wget http://curl.haxx.se/download/curl-7.45.0.tar.gz
|
|
||||||
RUN tar -zxvf curl-7.45.0.tar.gz
|
|
||||||
WORKDIR /root/curl-7.45.0
|
|
||||||
RUN ./configure --with-ssl --with-nghttp2=/usr/local
|
|
||||||
RUN make
|
|
||||||
RUN make install
|
|
||||||
RUN ldconfig
|
|
||||||
|
|
||||||
CMD ["-h"]
|
|
||||||
ENTRYPOINT ["/usr/local/bin/curl"]
|
|
||||||
|
|
3
vendor/golang.org/x/net/http2/Makefile
generated
vendored
3
vendor/golang.org/x/net/http2/Makefile
generated
vendored
@ -1,3 +0,0 @@
|
|||||||
curlimage:
|
|
||||||
docker build -t gohttp2/curl .
|
|
||||||
|
|
20
vendor/golang.org/x/net/http2/README
generated
vendored
20
vendor/golang.org/x/net/http2/README
generated
vendored
@ -1,20 +0,0 @@
|
|||||||
This is a work-in-progress HTTP/2 implementation for Go.
|
|
||||||
|
|
||||||
It will eventually live in the Go standard library and won't require
|
|
||||||
any changes to your code to use. It will just be automatic.
|
|
||||||
|
|
||||||
Status:
|
|
||||||
|
|
||||||
* The server support is pretty good. A few things are missing
|
|
||||||
but are being worked on.
|
|
||||||
* The client work has just started but shares a lot of code
|
|
||||||
is coming along much quicker.
|
|
||||||
|
|
||||||
Docs are at https://godoc.org/golang.org/x/net/http2
|
|
||||||
|
|
||||||
Demo test server at https://http2.golang.org/
|
|
||||||
|
|
||||||
Help & bug reports welcome!
|
|
||||||
|
|
||||||
Contributing: https://golang.org/doc/contribute.html
|
|
||||||
Bugs: https://golang.org/issue/new?title=x/net/http2:+
|
|
225
vendor/golang.org/x/net/http2/client_conn_pool.go
generated
vendored
225
vendor/golang.org/x/net/http2/client_conn_pool.go
generated
vendored
@ -1,225 +0,0 @@
|
|||||||
// Copyright 2015 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.
|
|
||||||
|
|
||||||
// Transport code's client connection pooling.
|
|
||||||
|
|
||||||
package http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/tls"
|
|
||||||
"net/http"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
// ClientConnPool manages a pool of HTTP/2 client connections.
|
|
||||||
type ClientConnPool interface {
|
|
||||||
GetClientConn(req *http.Request, addr string) (*ClientConn, error)
|
|
||||||
MarkDead(*ClientConn)
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO: use singleflight for dialing and addConnCalls?
|
|
||||||
type clientConnPool struct {
|
|
||||||
t *Transport
|
|
||||||
|
|
||||||
mu sync.Mutex // TODO: maybe switch to RWMutex
|
|
||||||
// TODO: add support for sharing conns based on cert names
|
|
||||||
// (e.g. share conn for googleapis.com and appspot.com)
|
|
||||||
conns map[string][]*ClientConn // key is host:port
|
|
||||||
dialing map[string]*dialCall // currently in-flight dials
|
|
||||||
keys map[*ClientConn][]string
|
|
||||||
addConnCalls map[string]*addConnCall // in-flight addConnIfNeede calls
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *clientConnPool) GetClientConn(req *http.Request, addr string) (*ClientConn, error) {
|
|
||||||
return p.getClientConn(req, addr, dialOnMiss)
|
|
||||||
}
|
|
||||||
|
|
||||||
const (
|
|
||||||
dialOnMiss = true
|
|
||||||
noDialOnMiss = false
|
|
||||||
)
|
|
||||||
|
|
||||||
func (p *clientConnPool) getClientConn(_ *http.Request, addr string, dialOnMiss bool) (*ClientConn, error) {
|
|
||||||
p.mu.Lock()
|
|
||||||
for _, cc := range p.conns[addr] {
|
|
||||||
if cc.CanTakeNewRequest() {
|
|
||||||
p.mu.Unlock()
|
|
||||||
return cc, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !dialOnMiss {
|
|
||||||
p.mu.Unlock()
|
|
||||||
return nil, ErrNoCachedConn
|
|
||||||
}
|
|
||||||
call := p.getStartDialLocked(addr)
|
|
||||||
p.mu.Unlock()
|
|
||||||
<-call.done
|
|
||||||
return call.res, call.err
|
|
||||||
}
|
|
||||||
|
|
||||||
// dialCall is an in-flight Transport dial call to a host.
|
|
||||||
type dialCall struct {
|
|
||||||
p *clientConnPool
|
|
||||||
done chan struct{} // closed when done
|
|
||||||
res *ClientConn // valid after done is closed
|
|
||||||
err error // valid after done is closed
|
|
||||||
}
|
|
||||||
|
|
||||||
// requires p.mu is held.
|
|
||||||
func (p *clientConnPool) getStartDialLocked(addr string) *dialCall {
|
|
||||||
if call, ok := p.dialing[addr]; ok {
|
|
||||||
// A dial is already in-flight. Don't start another.
|
|
||||||
return call
|
|
||||||
}
|
|
||||||
call := &dialCall{p: p, done: make(chan struct{})}
|
|
||||||
if p.dialing == nil {
|
|
||||||
p.dialing = make(map[string]*dialCall)
|
|
||||||
}
|
|
||||||
p.dialing[addr] = call
|
|
||||||
go call.dial(addr)
|
|
||||||
return call
|
|
||||||
}
|
|
||||||
|
|
||||||
// run in its own goroutine.
|
|
||||||
func (c *dialCall) dial(addr string) {
|
|
||||||
c.res, c.err = c.p.t.dialClientConn(addr)
|
|
||||||
close(c.done)
|
|
||||||
|
|
||||||
c.p.mu.Lock()
|
|
||||||
delete(c.p.dialing, addr)
|
|
||||||
if c.err == nil {
|
|
||||||
c.p.addConnLocked(addr, c.res)
|
|
||||||
}
|
|
||||||
c.p.mu.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
// addConnIfNeeded makes a NewClientConn out of c if a connection for key doesn't
|
|
||||||
// already exist. It coalesces concurrent calls with the same key.
|
|
||||||
// This is used by the http1 Transport code when it creates a new connection. Because
|
|
||||||
// the http1 Transport doesn't de-dup TCP dials to outbound hosts (because it doesn't know
|
|
||||||
// the protocol), it can get into a situation where it has multiple TLS connections.
|
|
||||||
// This code decides which ones live or die.
|
|
||||||
// The return value used is whether c was used.
|
|
||||||
// c is never closed.
|
|
||||||
func (p *clientConnPool) addConnIfNeeded(key string, t *Transport, c *tls.Conn) (used bool, err error) {
|
|
||||||
p.mu.Lock()
|
|
||||||
for _, cc := range p.conns[key] {
|
|
||||||
if cc.CanTakeNewRequest() {
|
|
||||||
p.mu.Unlock()
|
|
||||||
return false, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
call, dup := p.addConnCalls[key]
|
|
||||||
if !dup {
|
|
||||||
if p.addConnCalls == nil {
|
|
||||||
p.addConnCalls = make(map[string]*addConnCall)
|
|
||||||
}
|
|
||||||
call = &addConnCall{
|
|
||||||
p: p,
|
|
||||||
done: make(chan struct{}),
|
|
||||||
}
|
|
||||||
p.addConnCalls[key] = call
|
|
||||||
go call.run(t, key, c)
|
|
||||||
}
|
|
||||||
p.mu.Unlock()
|
|
||||||
|
|
||||||
<-call.done
|
|
||||||
if call.err != nil {
|
|
||||||
return false, call.err
|
|
||||||
}
|
|
||||||
return !dup, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type addConnCall struct {
|
|
||||||
p *clientConnPool
|
|
||||||
done chan struct{} // closed when done
|
|
||||||
err error
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *addConnCall) run(t *Transport, key string, tc *tls.Conn) {
|
|
||||||
cc, err := t.NewClientConn(tc)
|
|
||||||
|
|
||||||
p := c.p
|
|
||||||
p.mu.Lock()
|
|
||||||
if err != nil {
|
|
||||||
c.err = err
|
|
||||||
} else {
|
|
||||||
p.addConnLocked(key, cc)
|
|
||||||
}
|
|
||||||
delete(p.addConnCalls, key)
|
|
||||||
p.mu.Unlock()
|
|
||||||
close(c.done)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *clientConnPool) addConn(key string, cc *ClientConn) {
|
|
||||||
p.mu.Lock()
|
|
||||||
p.addConnLocked(key, cc)
|
|
||||||
p.mu.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
// p.mu must be held
|
|
||||||
func (p *clientConnPool) addConnLocked(key string, cc *ClientConn) {
|
|
||||||
for _, v := range p.conns[key] {
|
|
||||||
if v == cc {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if p.conns == nil {
|
|
||||||
p.conns = make(map[string][]*ClientConn)
|
|
||||||
}
|
|
||||||
if p.keys == nil {
|
|
||||||
p.keys = make(map[*ClientConn][]string)
|
|
||||||
}
|
|
||||||
p.conns[key] = append(p.conns[key], cc)
|
|
||||||
p.keys[cc] = append(p.keys[cc], key)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *clientConnPool) MarkDead(cc *ClientConn) {
|
|
||||||
p.mu.Lock()
|
|
||||||
defer p.mu.Unlock()
|
|
||||||
for _, key := range p.keys[cc] {
|
|
||||||
vv, ok := p.conns[key]
|
|
||||||
if !ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
newList := filterOutClientConn(vv, cc)
|
|
||||||
if len(newList) > 0 {
|
|
||||||
p.conns[key] = newList
|
|
||||||
} else {
|
|
||||||
delete(p.conns, key)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
delete(p.keys, cc)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *clientConnPool) closeIdleConnections() {
|
|
||||||
p.mu.Lock()
|
|
||||||
defer p.mu.Unlock()
|
|
||||||
// TODO: don't close a cc if it was just added to the pool
|
|
||||||
// milliseconds ago and has never been used. There's currently
|
|
||||||
// a small race window with the HTTP/1 Transport's integration
|
|
||||||
// where it can add an idle conn just before using it, and
|
|
||||||
// somebody else can concurrently call CloseIdleConns and
|
|
||||||
// break some caller's RoundTrip.
|
|
||||||
for _, vv := range p.conns {
|
|
||||||
for _, cc := range vv {
|
|
||||||
cc.closeIfIdle()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func filterOutClientConn(in []*ClientConn, exclude *ClientConn) []*ClientConn {
|
|
||||||
out := in[:0]
|
|
||||||
for _, v := range in {
|
|
||||||
if v != exclude {
|
|
||||||
out = append(out, v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// If we filtered it out, zero out the last item to prevent
|
|
||||||
// the GC from seeing it.
|
|
||||||
if len(in) != len(out) {
|
|
||||||
in[len(in)-1] = nil
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
89
vendor/golang.org/x/net/http2/configure_transport.go
generated
vendored
89
vendor/golang.org/x/net/http2/configure_transport.go
generated
vendored
@ -1,89 +0,0 @@
|
|||||||
// Copyright 2015 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.
|
|
||||||
|
|
||||||
// +build go1.6
|
|
||||||
|
|
||||||
package http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/tls"
|
|
||||||
"fmt"
|
|
||||||
"net/http"
|
|
||||||
)
|
|
||||||
|
|
||||||
func configureTransport(t1 *http.Transport) (*Transport, error) {
|
|
||||||
connPool := new(clientConnPool)
|
|
||||||
t2 := &Transport{
|
|
||||||
ConnPool: noDialClientConnPool{connPool},
|
|
||||||
t1: t1,
|
|
||||||
}
|
|
||||||
connPool.t = t2
|
|
||||||
if err := registerHTTPSProtocol(t1, noDialH2RoundTripper{t2}); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if t1.TLSClientConfig == nil {
|
|
||||||
t1.TLSClientConfig = new(tls.Config)
|
|
||||||
}
|
|
||||||
if !strSliceContains(t1.TLSClientConfig.NextProtos, "h2") {
|
|
||||||
t1.TLSClientConfig.NextProtos = append([]string{"h2"}, t1.TLSClientConfig.NextProtos...)
|
|
||||||
}
|
|
||||||
if !strSliceContains(t1.TLSClientConfig.NextProtos, "http/1.1") {
|
|
||||||
t1.TLSClientConfig.NextProtos = append(t1.TLSClientConfig.NextProtos, "http/1.1")
|
|
||||||
}
|
|
||||||
upgradeFn := func(authority string, c *tls.Conn) http.RoundTripper {
|
|
||||||
addr := authorityAddr(authority)
|
|
||||||
if used, err := connPool.addConnIfNeeded(addr, t2, c); err != nil {
|
|
||||||
go c.Close()
|
|
||||||
return erringRoundTripper{err}
|
|
||||||
} else if !used {
|
|
||||||
// Turns out we don't need this c.
|
|
||||||
// For example, two goroutines made requests to the same host
|
|
||||||
// at the same time, both kicking off TCP dials. (since protocol
|
|
||||||
// was unknown)
|
|
||||||
go c.Close()
|
|
||||||
}
|
|
||||||
return t2
|
|
||||||
}
|
|
||||||
if m := t1.TLSNextProto; len(m) == 0 {
|
|
||||||
t1.TLSNextProto = map[string]func(string, *tls.Conn) http.RoundTripper{
|
|
||||||
"h2": upgradeFn,
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
m["h2"] = upgradeFn
|
|
||||||
}
|
|
||||||
return t2, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// registerHTTPSProtocol calls Transport.RegisterProtocol but
|
|
||||||
// convering panics into errors.
|
|
||||||
func registerHTTPSProtocol(t *http.Transport, rt http.RoundTripper) (err error) {
|
|
||||||
defer func() {
|
|
||||||
if e := recover(); e != nil {
|
|
||||||
err = fmt.Errorf("%v", e)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
t.RegisterProtocol("https", rt)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// noDialClientConnPool is an implementation of http2.ClientConnPool
|
|
||||||
// which never dials. We let the HTTP/1.1 client dial and use its TLS
|
|
||||||
// connection instead.
|
|
||||||
type noDialClientConnPool struct{ *clientConnPool }
|
|
||||||
|
|
||||||
func (p noDialClientConnPool) GetClientConn(req *http.Request, addr string) (*ClientConn, error) {
|
|
||||||
return p.getClientConn(req, addr, noDialOnMiss)
|
|
||||||
}
|
|
||||||
|
|
||||||
// noDialH2RoundTripper is a RoundTripper which only tries to complete the request
|
|
||||||
// if there's already has a cached connection to the host.
|
|
||||||
type noDialH2RoundTripper struct{ t *Transport }
|
|
||||||
|
|
||||||
func (rt noDialH2RoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
||||||
res, err := rt.t.RoundTrip(req)
|
|
||||||
if err == ErrNoCachedConn {
|
|
||||||
return nil, http.ErrSkipAltProtocol
|
|
||||||
}
|
|
||||||
return res, err
|
|
||||||
}
|
|
122
vendor/golang.org/x/net/http2/errors.go
generated
vendored
122
vendor/golang.org/x/net/http2/errors.go
generated
vendored
@ -1,122 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
)
|
|
||||||
|
|
||||||
// An ErrCode is an unsigned 32-bit error code as defined in the HTTP/2 spec.
|
|
||||||
type ErrCode uint32
|
|
||||||
|
|
||||||
const (
|
|
||||||
ErrCodeNo ErrCode = 0x0
|
|
||||||
ErrCodeProtocol ErrCode = 0x1
|
|
||||||
ErrCodeInternal ErrCode = 0x2
|
|
||||||
ErrCodeFlowControl ErrCode = 0x3
|
|
||||||
ErrCodeSettingsTimeout ErrCode = 0x4
|
|
||||||
ErrCodeStreamClosed ErrCode = 0x5
|
|
||||||
ErrCodeFrameSize ErrCode = 0x6
|
|
||||||
ErrCodeRefusedStream ErrCode = 0x7
|
|
||||||
ErrCodeCancel ErrCode = 0x8
|
|
||||||
ErrCodeCompression ErrCode = 0x9
|
|
||||||
ErrCodeConnect ErrCode = 0xa
|
|
||||||
ErrCodeEnhanceYourCalm ErrCode = 0xb
|
|
||||||
ErrCodeInadequateSecurity ErrCode = 0xc
|
|
||||||
ErrCodeHTTP11Required ErrCode = 0xd
|
|
||||||
)
|
|
||||||
|
|
||||||
var errCodeName = map[ErrCode]string{
|
|
||||||
ErrCodeNo: "NO_ERROR",
|
|
||||||
ErrCodeProtocol: "PROTOCOL_ERROR",
|
|
||||||
ErrCodeInternal: "INTERNAL_ERROR",
|
|
||||||
ErrCodeFlowControl: "FLOW_CONTROL_ERROR",
|
|
||||||
ErrCodeSettingsTimeout: "SETTINGS_TIMEOUT",
|
|
||||||
ErrCodeStreamClosed: "STREAM_CLOSED",
|
|
||||||
ErrCodeFrameSize: "FRAME_SIZE_ERROR",
|
|
||||||
ErrCodeRefusedStream: "REFUSED_STREAM",
|
|
||||||
ErrCodeCancel: "CANCEL",
|
|
||||||
ErrCodeCompression: "COMPRESSION_ERROR",
|
|
||||||
ErrCodeConnect: "CONNECT_ERROR",
|
|
||||||
ErrCodeEnhanceYourCalm: "ENHANCE_YOUR_CALM",
|
|
||||||
ErrCodeInadequateSecurity: "INADEQUATE_SECURITY",
|
|
||||||
ErrCodeHTTP11Required: "HTTP_1_1_REQUIRED",
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e ErrCode) String() string {
|
|
||||||
if s, ok := errCodeName[e]; ok {
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
return fmt.Sprintf("unknown error code 0x%x", uint32(e))
|
|
||||||
}
|
|
||||||
|
|
||||||
// ConnectionError is an error that results in the termination of the
|
|
||||||
// entire connection.
|
|
||||||
type ConnectionError ErrCode
|
|
||||||
|
|
||||||
func (e ConnectionError) Error() string { return fmt.Sprintf("connection error: %s", ErrCode(e)) }
|
|
||||||
|
|
||||||
// StreamError is an error that only affects one stream within an
|
|
||||||
// HTTP/2 connection.
|
|
||||||
type StreamError struct {
|
|
||||||
StreamID uint32
|
|
||||||
Code ErrCode
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e StreamError) Error() string {
|
|
||||||
return fmt.Sprintf("stream error: stream ID %d; %v", e.StreamID, e.Code)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 6.9.1 The Flow Control Window
|
|
||||||
// "If a sender receives a WINDOW_UPDATE that causes a flow control
|
|
||||||
// window to exceed this maximum it MUST terminate either the stream
|
|
||||||
// or the connection, as appropriate. For streams, [...]; for the
|
|
||||||
// connection, a GOAWAY frame with a FLOW_CONTROL_ERROR code."
|
|
||||||
type goAwayFlowError struct{}
|
|
||||||
|
|
||||||
func (goAwayFlowError) Error() string { return "connection exceeded flow control window size" }
|
|
||||||
|
|
||||||
// connErrorReason wraps a ConnectionError with an informative error about why it occurs.
|
|
||||||
|
|
||||||
// Errors of this type are only returned by the frame parser functions
|
|
||||||
// and converted into ConnectionError(ErrCodeProtocol).
|
|
||||||
type connError struct {
|
|
||||||
Code ErrCode
|
|
||||||
Reason string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e connError) Error() string {
|
|
||||||
return fmt.Sprintf("http2: connection error: %v: %v", e.Code, e.Reason)
|
|
||||||
}
|
|
||||||
|
|
||||||
type pseudoHeaderError string
|
|
||||||
|
|
||||||
func (e pseudoHeaderError) Error() string {
|
|
||||||
return fmt.Sprintf("invalid pseudo-header %q", string(e))
|
|
||||||
}
|
|
||||||
|
|
||||||
type duplicatePseudoHeaderError string
|
|
||||||
|
|
||||||
func (e duplicatePseudoHeaderError) Error() string {
|
|
||||||
return fmt.Sprintf("duplicate pseudo-header %q", string(e))
|
|
||||||
}
|
|
||||||
|
|
||||||
type headerFieldNameError string
|
|
||||||
|
|
||||||
func (e headerFieldNameError) Error() string {
|
|
||||||
return fmt.Sprintf("invalid header field name %q", string(e))
|
|
||||||
}
|
|
||||||
|
|
||||||
type headerFieldValueError string
|
|
||||||
|
|
||||||
func (e headerFieldValueError) Error() string {
|
|
||||||
return fmt.Sprintf("invalid header field value %q", string(e))
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
errMixPseudoHeaderTypes = errors.New("mix of request and response pseudo headers")
|
|
||||||
errPseudoAfterRegular = errors.New("pseudo header field after regular")
|
|
||||||
)
|
|
24
vendor/golang.org/x/net/http2/errors_test.go
generated
vendored
24
vendor/golang.org/x/net/http2/errors_test.go
generated
vendored
@ -1,24 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import "testing"
|
|
||||||
|
|
||||||
func TestErrCodeString(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
err ErrCode
|
|
||||||
want string
|
|
||||||
}{
|
|
||||||
{ErrCodeProtocol, "PROTOCOL_ERROR"},
|
|
||||||
{0xd, "HTTP_1_1_REQUIRED"},
|
|
||||||
{0xf, "unknown error code 0xf"},
|
|
||||||
}
|
|
||||||
for i, tt := range tests {
|
|
||||||
got := tt.err.String()
|
|
||||||
if got != tt.want {
|
|
||||||
t.Errorf("%d. Error = %q; want %q", i, got, tt.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
60
vendor/golang.org/x/net/http2/fixed_buffer.go
generated
vendored
60
vendor/golang.org/x/net/http2/fixed_buffer.go
generated
vendored
@ -1,60 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
)
|
|
||||||
|
|
||||||
// fixedBuffer is an io.ReadWriter backed by a fixed size buffer.
|
|
||||||
// It never allocates, but moves old data as new data is written.
|
|
||||||
type fixedBuffer struct {
|
|
||||||
buf []byte
|
|
||||||
r, w int
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
errReadEmpty = errors.New("read from empty fixedBuffer")
|
|
||||||
errWriteFull = errors.New("write on full fixedBuffer")
|
|
||||||
)
|
|
||||||
|
|
||||||
// Read copies bytes from the buffer into p.
|
|
||||||
// It is an error to read when no data is available.
|
|
||||||
func (b *fixedBuffer) Read(p []byte) (n int, err error) {
|
|
||||||
if b.r == b.w {
|
|
||||||
return 0, errReadEmpty
|
|
||||||
}
|
|
||||||
n = copy(p, b.buf[b.r:b.w])
|
|
||||||
b.r += n
|
|
||||||
if b.r == b.w {
|
|
||||||
b.r = 0
|
|
||||||
b.w = 0
|
|
||||||
}
|
|
||||||
return n, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Len returns the number of bytes of the unread portion of the buffer.
|
|
||||||
func (b *fixedBuffer) Len() int {
|
|
||||||
return b.w - b.r
|
|
||||||
}
|
|
||||||
|
|
||||||
// Write copies bytes from p into the buffer.
|
|
||||||
// It is an error to write more data than the buffer can hold.
|
|
||||||
func (b *fixedBuffer) Write(p []byte) (n int, err error) {
|
|
||||||
// Slide existing data to beginning.
|
|
||||||
if b.r > 0 && len(p) > len(b.buf)-b.w {
|
|
||||||
copy(b.buf, b.buf[b.r:b.w])
|
|
||||||
b.w -= b.r
|
|
||||||
b.r = 0
|
|
||||||
}
|
|
||||||
|
|
||||||
// Write new data.
|
|
||||||
n = copy(b.buf[b.w:], p)
|
|
||||||
b.w += n
|
|
||||||
if n < len(p) {
|
|
||||||
err = errWriteFull
|
|
||||||
}
|
|
||||||
return n, err
|
|
||||||
}
|
|
128
vendor/golang.org/x/net/http2/fixed_buffer_test.go
generated
vendored
128
vendor/golang.org/x/net/http2/fixed_buffer_test.go
generated
vendored
@ -1,128 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"reflect"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
var bufferReadTests = []struct {
|
|
||||||
buf fixedBuffer
|
|
||||||
read, wn int
|
|
||||||
werr error
|
|
||||||
wp []byte
|
|
||||||
wbuf fixedBuffer
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
fixedBuffer{[]byte{'a', 0}, 0, 1},
|
|
||||||
5, 1, nil, []byte{'a'},
|
|
||||||
fixedBuffer{[]byte{'a', 0}, 0, 0},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
fixedBuffer{[]byte{0, 'a'}, 1, 2},
|
|
||||||
5, 1, nil, []byte{'a'},
|
|
||||||
fixedBuffer{[]byte{0, 'a'}, 0, 0},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
fixedBuffer{[]byte{'a', 'b'}, 0, 2},
|
|
||||||
1, 1, nil, []byte{'a'},
|
|
||||||
fixedBuffer{[]byte{'a', 'b'}, 1, 2},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
fixedBuffer{[]byte{}, 0, 0},
|
|
||||||
5, 0, errReadEmpty, []byte{},
|
|
||||||
fixedBuffer{[]byte{}, 0, 0},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBufferRead(t *testing.T) {
|
|
||||||
for i, tt := range bufferReadTests {
|
|
||||||
read := make([]byte, tt.read)
|
|
||||||
n, err := tt.buf.Read(read)
|
|
||||||
if n != tt.wn {
|
|
||||||
t.Errorf("#%d: wn = %d want %d", i, n, tt.wn)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if err != tt.werr {
|
|
||||||
t.Errorf("#%d: werr = %v want %v", i, err, tt.werr)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
read = read[:n]
|
|
||||||
if !reflect.DeepEqual(read, tt.wp) {
|
|
||||||
t.Errorf("#%d: read = %+v want %+v", i, read, tt.wp)
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(tt.buf, tt.wbuf) {
|
|
||||||
t.Errorf("#%d: buf = %+v want %+v", i, tt.buf, tt.wbuf)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var bufferWriteTests = []struct {
|
|
||||||
buf fixedBuffer
|
|
||||||
write, wn int
|
|
||||||
werr error
|
|
||||||
wbuf fixedBuffer
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
buf: fixedBuffer{
|
|
||||||
buf: []byte{},
|
|
||||||
},
|
|
||||||
wbuf: fixedBuffer{
|
|
||||||
buf: []byte{},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
buf: fixedBuffer{
|
|
||||||
buf: []byte{1, 'a'},
|
|
||||||
},
|
|
||||||
write: 1,
|
|
||||||
wn: 1,
|
|
||||||
wbuf: fixedBuffer{
|
|
||||||
buf: []byte{0, 'a'},
|
|
||||||
w: 1,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
buf: fixedBuffer{
|
|
||||||
buf: []byte{'a', 1},
|
|
||||||
r: 1,
|
|
||||||
w: 1,
|
|
||||||
},
|
|
||||||
write: 2,
|
|
||||||
wn: 2,
|
|
||||||
wbuf: fixedBuffer{
|
|
||||||
buf: []byte{0, 0},
|
|
||||||
w: 2,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
buf: fixedBuffer{
|
|
||||||
buf: []byte{},
|
|
||||||
},
|
|
||||||
write: 5,
|
|
||||||
werr: errWriteFull,
|
|
||||||
wbuf: fixedBuffer{
|
|
||||||
buf: []byte{},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBufferWrite(t *testing.T) {
|
|
||||||
for i, tt := range bufferWriteTests {
|
|
||||||
n, err := tt.buf.Write(make([]byte, tt.write))
|
|
||||||
if n != tt.wn {
|
|
||||||
t.Errorf("#%d: wrote %d bytes; want %d", i, n, tt.wn)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if err != tt.werr {
|
|
||||||
t.Errorf("#%d: error = %v; want %v", i, err, tt.werr)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(tt.buf, tt.wbuf) {
|
|
||||||
t.Errorf("#%d: buf = %+v; want %+v", i, tt.buf, tt.wbuf)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
50
vendor/golang.org/x/net/http2/flow.go
generated
vendored
50
vendor/golang.org/x/net/http2/flow.go
generated
vendored
@ -1,50 +0,0 @@
|
|||||||
// Copyright 2014 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.
|
|
||||||
|
|
||||||
// Flow control
|
|
||||||
|
|
||||||
package http2
|
|
||||||
|
|
||||||
// flow is the flow control window's size.
|
|
||||||
type flow struct {
|
|
||||||
// n is the number of DATA bytes we're allowed to send.
|
|
||||||
// A flow is kept both on a conn and a per-stream.
|
|
||||||
n int32
|
|
||||||
|
|
||||||
// conn points to the shared connection-level flow that is
|
|
||||||
// shared by all streams on that conn. It is nil for the flow
|
|
||||||
// that's on the conn directly.
|
|
||||||
conn *flow
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *flow) setConnFlow(cf *flow) { f.conn = cf }
|
|
||||||
|
|
||||||
func (f *flow) available() int32 {
|
|
||||||
n := f.n
|
|
||||||
if f.conn != nil && f.conn.n < n {
|
|
||||||
n = f.conn.n
|
|
||||||
}
|
|
||||||
return n
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *flow) take(n int32) {
|
|
||||||
if n > f.available() {
|
|
||||||
panic("internal error: took too much")
|
|
||||||
}
|
|
||||||
f.n -= n
|
|
||||||
if f.conn != nil {
|
|
||||||
f.conn.n -= n
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// add adds n bytes (positive or negative) to the flow control window.
|
|
||||||
// It returns false if the sum would exceed 2^31-1.
|
|
||||||
func (f *flow) add(n int32) bool {
|
|
||||||
remain := (1<<31 - 1) - f.n
|
|
||||||
if n > remain {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
f.n += n
|
|
||||||
return true
|
|
||||||
}
|
|
53
vendor/golang.org/x/net/http2/flow_test.go
generated
vendored
53
vendor/golang.org/x/net/http2/flow_test.go
generated
vendored
@ -1,53 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import "testing"
|
|
||||||
|
|
||||||
func TestFlow(t *testing.T) {
|
|
||||||
var st flow
|
|
||||||
var conn flow
|
|
||||||
st.add(3)
|
|
||||||
conn.add(2)
|
|
||||||
|
|
||||||
if got, want := st.available(), int32(3); got != want {
|
|
||||||
t.Errorf("available = %d; want %d", got, want)
|
|
||||||
}
|
|
||||||
st.setConnFlow(&conn)
|
|
||||||
if got, want := st.available(), int32(2); got != want {
|
|
||||||
t.Errorf("after parent setup, available = %d; want %d", got, want)
|
|
||||||
}
|
|
||||||
|
|
||||||
st.take(2)
|
|
||||||
if got, want := conn.available(), int32(0); got != want {
|
|
||||||
t.Errorf("after taking 2, conn = %d; want %d", got, want)
|
|
||||||
}
|
|
||||||
if got, want := st.available(), int32(0); got != want {
|
|
||||||
t.Errorf("after taking 2, stream = %d; want %d", got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFlowAdd(t *testing.T) {
|
|
||||||
var f flow
|
|
||||||
if !f.add(1) {
|
|
||||||
t.Fatal("failed to add 1")
|
|
||||||
}
|
|
||||||
if !f.add(-1) {
|
|
||||||
t.Fatal("failed to add -1")
|
|
||||||
}
|
|
||||||
if got, want := f.available(), int32(0); got != want {
|
|
||||||
t.Fatalf("size = %d; want %d", got, want)
|
|
||||||
}
|
|
||||||
if !f.add(1<<31 - 1) {
|
|
||||||
t.Fatal("failed to add 2^31-1")
|
|
||||||
}
|
|
||||||
if got, want := f.available(), int32(1<<31-1); got != want {
|
|
||||||
t.Fatalf("size = %d; want %d", got, want)
|
|
||||||
}
|
|
||||||
if f.add(1) {
|
|
||||||
t.Fatal("adding 1 to max shouldn't be allowed")
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
1496
vendor/golang.org/x/net/http2/frame.go
generated
vendored
1496
vendor/golang.org/x/net/http2/frame.go
generated
vendored
File diff suppressed because it is too large
Load Diff
975
vendor/golang.org/x/net/http2/frame_test.go
generated
vendored
975
vendor/golang.org/x/net/http2/frame_test.go
generated
vendored
@ -1,975 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"reflect"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
"unsafe"
|
|
||||||
|
|
||||||
"golang.org/x/net/http2/hpack"
|
|
||||||
)
|
|
||||||
|
|
||||||
func testFramer() (*Framer, *bytes.Buffer) {
|
|
||||||
buf := new(bytes.Buffer)
|
|
||||||
return NewFramer(buf, buf), buf
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFrameSizes(t *testing.T) {
|
|
||||||
// Catch people rearranging the FrameHeader fields.
|
|
||||||
if got, want := int(unsafe.Sizeof(FrameHeader{})), 12; got != want {
|
|
||||||
t.Errorf("FrameHeader size = %d; want %d", got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFrameTypeString(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
ft FrameType
|
|
||||||
want string
|
|
||||||
}{
|
|
||||||
{FrameData, "DATA"},
|
|
||||||
{FramePing, "PING"},
|
|
||||||
{FrameGoAway, "GOAWAY"},
|
|
||||||
{0xf, "UNKNOWN_FRAME_TYPE_15"},
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, tt := range tests {
|
|
||||||
got := tt.ft.String()
|
|
||||||
if got != tt.want {
|
|
||||||
t.Errorf("%d. String(FrameType %d) = %q; want %q", i, int(tt.ft), got, tt.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWriteRST(t *testing.T) {
|
|
||||||
fr, buf := testFramer()
|
|
||||||
var streamID uint32 = 1<<24 + 2<<16 + 3<<8 + 4
|
|
||||||
var errCode uint32 = 7<<24 + 6<<16 + 5<<8 + 4
|
|
||||||
fr.WriteRSTStream(streamID, ErrCode(errCode))
|
|
||||||
const wantEnc = "\x00\x00\x04\x03\x00\x01\x02\x03\x04\x07\x06\x05\x04"
|
|
||||||
if buf.String() != wantEnc {
|
|
||||||
t.Errorf("encoded as %q; want %q", buf.Bytes(), wantEnc)
|
|
||||||
}
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
want := &RSTStreamFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
Type: 0x3,
|
|
||||||
Flags: 0x0,
|
|
||||||
Length: 0x4,
|
|
||||||
StreamID: 0x1020304,
|
|
||||||
},
|
|
||||||
ErrCode: 0x7060504,
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(f, want) {
|
|
||||||
t.Errorf("parsed back %#v; want %#v", f, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWriteData(t *testing.T) {
|
|
||||||
fr, buf := testFramer()
|
|
||||||
var streamID uint32 = 1<<24 + 2<<16 + 3<<8 + 4
|
|
||||||
data := []byte("ABC")
|
|
||||||
fr.WriteData(streamID, true, data)
|
|
||||||
const wantEnc = "\x00\x00\x03\x00\x01\x01\x02\x03\x04ABC"
|
|
||||||
if buf.String() != wantEnc {
|
|
||||||
t.Errorf("encoded as %q; want %q", buf.Bytes(), wantEnc)
|
|
||||||
}
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
df, ok := f.(*DataFrame)
|
|
||||||
if !ok {
|
|
||||||
t.Fatalf("got %T; want *DataFrame", f)
|
|
||||||
}
|
|
||||||
if !bytes.Equal(df.Data(), data) {
|
|
||||||
t.Errorf("got %q; want %q", df.Data(), data)
|
|
||||||
}
|
|
||||||
if f.Header().Flags&1 == 0 {
|
|
||||||
t.Errorf("didn't see END_STREAM flag")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWriteHeaders(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
name string
|
|
||||||
p HeadersFrameParam
|
|
||||||
wantEnc string
|
|
||||||
wantFrame *HeadersFrame
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
"basic",
|
|
||||||
HeadersFrameParam{
|
|
||||||
StreamID: 42,
|
|
||||||
BlockFragment: []byte("abc"),
|
|
||||||
Priority: PriorityParam{},
|
|
||||||
},
|
|
||||||
"\x00\x00\x03\x01\x00\x00\x00\x00*abc",
|
|
||||||
&HeadersFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
StreamID: 42,
|
|
||||||
Type: FrameHeaders,
|
|
||||||
Length: uint32(len("abc")),
|
|
||||||
},
|
|
||||||
Priority: PriorityParam{},
|
|
||||||
headerFragBuf: []byte("abc"),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"basic + end flags",
|
|
||||||
HeadersFrameParam{
|
|
||||||
StreamID: 42,
|
|
||||||
BlockFragment: []byte("abc"),
|
|
||||||
EndStream: true,
|
|
||||||
EndHeaders: true,
|
|
||||||
Priority: PriorityParam{},
|
|
||||||
},
|
|
||||||
"\x00\x00\x03\x01\x05\x00\x00\x00*abc",
|
|
||||||
&HeadersFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
StreamID: 42,
|
|
||||||
Type: FrameHeaders,
|
|
||||||
Flags: FlagHeadersEndStream | FlagHeadersEndHeaders,
|
|
||||||
Length: uint32(len("abc")),
|
|
||||||
},
|
|
||||||
Priority: PriorityParam{},
|
|
||||||
headerFragBuf: []byte("abc"),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"with padding",
|
|
||||||
HeadersFrameParam{
|
|
||||||
StreamID: 42,
|
|
||||||
BlockFragment: []byte("abc"),
|
|
||||||
EndStream: true,
|
|
||||||
EndHeaders: true,
|
|
||||||
PadLength: 5,
|
|
||||||
Priority: PriorityParam{},
|
|
||||||
},
|
|
||||||
"\x00\x00\t\x01\r\x00\x00\x00*\x05abc\x00\x00\x00\x00\x00",
|
|
||||||
&HeadersFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
StreamID: 42,
|
|
||||||
Type: FrameHeaders,
|
|
||||||
Flags: FlagHeadersEndStream | FlagHeadersEndHeaders | FlagHeadersPadded,
|
|
||||||
Length: uint32(1 + len("abc") + 5), // pad length + contents + padding
|
|
||||||
},
|
|
||||||
Priority: PriorityParam{},
|
|
||||||
headerFragBuf: []byte("abc"),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"with priority",
|
|
||||||
HeadersFrameParam{
|
|
||||||
StreamID: 42,
|
|
||||||
BlockFragment: []byte("abc"),
|
|
||||||
EndStream: true,
|
|
||||||
EndHeaders: true,
|
|
||||||
PadLength: 2,
|
|
||||||
Priority: PriorityParam{
|
|
||||||
StreamDep: 15,
|
|
||||||
Exclusive: true,
|
|
||||||
Weight: 127,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
"\x00\x00\v\x01-\x00\x00\x00*\x02\x80\x00\x00\x0f\u007fabc\x00\x00",
|
|
||||||
&HeadersFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
StreamID: 42,
|
|
||||||
Type: FrameHeaders,
|
|
||||||
Flags: FlagHeadersEndStream | FlagHeadersEndHeaders | FlagHeadersPadded | FlagHeadersPriority,
|
|
||||||
Length: uint32(1 + 5 + len("abc") + 2), // pad length + priority + contents + padding
|
|
||||||
},
|
|
||||||
Priority: PriorityParam{
|
|
||||||
StreamDep: 15,
|
|
||||||
Exclusive: true,
|
|
||||||
Weight: 127,
|
|
||||||
},
|
|
||||||
headerFragBuf: []byte("abc"),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
fr, buf := testFramer()
|
|
||||||
if err := fr.WriteHeaders(tt.p); err != nil {
|
|
||||||
t.Errorf("test %q: %v", tt.name, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if buf.String() != tt.wantEnc {
|
|
||||||
t.Errorf("test %q: encoded %q; want %q", tt.name, buf.Bytes(), tt.wantEnc)
|
|
||||||
}
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("test %q: failed to read the frame back: %v", tt.name, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(f, tt.wantFrame) {
|
|
||||||
t.Errorf("test %q: mismatch.\n got: %#v\nwant: %#v\n", tt.name, f, tt.wantFrame)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWriteContinuation(t *testing.T) {
|
|
||||||
const streamID = 42
|
|
||||||
tests := []struct {
|
|
||||||
name string
|
|
||||||
end bool
|
|
||||||
frag []byte
|
|
||||||
|
|
||||||
wantFrame *ContinuationFrame
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
"not end",
|
|
||||||
false,
|
|
||||||
[]byte("abc"),
|
|
||||||
&ContinuationFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
StreamID: streamID,
|
|
||||||
Type: FrameContinuation,
|
|
||||||
Length: uint32(len("abc")),
|
|
||||||
},
|
|
||||||
headerFragBuf: []byte("abc"),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"end",
|
|
||||||
true,
|
|
||||||
[]byte("def"),
|
|
||||||
&ContinuationFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
StreamID: streamID,
|
|
||||||
Type: FrameContinuation,
|
|
||||||
Flags: FlagContinuationEndHeaders,
|
|
||||||
Length: uint32(len("def")),
|
|
||||||
},
|
|
||||||
headerFragBuf: []byte("def"),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
fr, _ := testFramer()
|
|
||||||
if err := fr.WriteContinuation(streamID, tt.end, tt.frag); err != nil {
|
|
||||||
t.Errorf("test %q: %v", tt.name, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
fr.AllowIllegalReads = true
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("test %q: failed to read the frame back: %v", tt.name, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(f, tt.wantFrame) {
|
|
||||||
t.Errorf("test %q: mismatch.\n got: %#v\nwant: %#v\n", tt.name, f, tt.wantFrame)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWritePriority(t *testing.T) {
|
|
||||||
const streamID = 42
|
|
||||||
tests := []struct {
|
|
||||||
name string
|
|
||||||
priority PriorityParam
|
|
||||||
wantFrame *PriorityFrame
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
"not exclusive",
|
|
||||||
PriorityParam{
|
|
||||||
StreamDep: 2,
|
|
||||||
Exclusive: false,
|
|
||||||
Weight: 127,
|
|
||||||
},
|
|
||||||
&PriorityFrame{
|
|
||||||
FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
StreamID: streamID,
|
|
||||||
Type: FramePriority,
|
|
||||||
Length: 5,
|
|
||||||
},
|
|
||||||
PriorityParam{
|
|
||||||
StreamDep: 2,
|
|
||||||
Exclusive: false,
|
|
||||||
Weight: 127,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
|
|
||||||
{
|
|
||||||
"exclusive",
|
|
||||||
PriorityParam{
|
|
||||||
StreamDep: 3,
|
|
||||||
Exclusive: true,
|
|
||||||
Weight: 77,
|
|
||||||
},
|
|
||||||
&PriorityFrame{
|
|
||||||
FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
StreamID: streamID,
|
|
||||||
Type: FramePriority,
|
|
||||||
Length: 5,
|
|
||||||
},
|
|
||||||
PriorityParam{
|
|
||||||
StreamDep: 3,
|
|
||||||
Exclusive: true,
|
|
||||||
Weight: 77,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
fr, _ := testFramer()
|
|
||||||
if err := fr.WritePriority(streamID, tt.priority); err != nil {
|
|
||||||
t.Errorf("test %q: %v", tt.name, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("test %q: failed to read the frame back: %v", tt.name, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(f, tt.wantFrame) {
|
|
||||||
t.Errorf("test %q: mismatch.\n got: %#v\nwant: %#v\n", tt.name, f, tt.wantFrame)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWriteSettings(t *testing.T) {
|
|
||||||
fr, buf := testFramer()
|
|
||||||
settings := []Setting{{1, 2}, {3, 4}}
|
|
||||||
fr.WriteSettings(settings...)
|
|
||||||
const wantEnc = "\x00\x00\f\x04\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x02\x00\x03\x00\x00\x00\x04"
|
|
||||||
if buf.String() != wantEnc {
|
|
||||||
t.Errorf("encoded as %q; want %q", buf.Bytes(), wantEnc)
|
|
||||||
}
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
sf, ok := f.(*SettingsFrame)
|
|
||||||
if !ok {
|
|
||||||
t.Fatalf("Got a %T; want a SettingsFrame", f)
|
|
||||||
}
|
|
||||||
var got []Setting
|
|
||||||
sf.ForeachSetting(func(s Setting) error {
|
|
||||||
got = append(got, s)
|
|
||||||
valBack, ok := sf.Value(s.ID)
|
|
||||||
if !ok || valBack != s.Val {
|
|
||||||
t.Errorf("Value(%d) = %v, %v; want %v, true", s.ID, valBack, ok, s.Val)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
if !reflect.DeepEqual(settings, got) {
|
|
||||||
t.Errorf("Read settings %+v != written settings %+v", got, settings)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWriteSettingsAck(t *testing.T) {
|
|
||||||
fr, buf := testFramer()
|
|
||||||
fr.WriteSettingsAck()
|
|
||||||
const wantEnc = "\x00\x00\x00\x04\x01\x00\x00\x00\x00"
|
|
||||||
if buf.String() != wantEnc {
|
|
||||||
t.Errorf("encoded as %q; want %q", buf.Bytes(), wantEnc)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWriteWindowUpdate(t *testing.T) {
|
|
||||||
fr, buf := testFramer()
|
|
||||||
const streamID = 1<<24 + 2<<16 + 3<<8 + 4
|
|
||||||
const incr = 7<<24 + 6<<16 + 5<<8 + 4
|
|
||||||
if err := fr.WriteWindowUpdate(streamID, incr); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
const wantEnc = "\x00\x00\x04\x08\x00\x01\x02\x03\x04\x07\x06\x05\x04"
|
|
||||||
if buf.String() != wantEnc {
|
|
||||||
t.Errorf("encoded as %q; want %q", buf.Bytes(), wantEnc)
|
|
||||||
}
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
want := &WindowUpdateFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
Type: 0x8,
|
|
||||||
Flags: 0x0,
|
|
||||||
Length: 0x4,
|
|
||||||
StreamID: 0x1020304,
|
|
||||||
},
|
|
||||||
Increment: 0x7060504,
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(f, want) {
|
|
||||||
t.Errorf("parsed back %#v; want %#v", f, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWritePing(t *testing.T) { testWritePing(t, false) }
|
|
||||||
func TestWritePingAck(t *testing.T) { testWritePing(t, true) }
|
|
||||||
|
|
||||||
func testWritePing(t *testing.T, ack bool) {
|
|
||||||
fr, buf := testFramer()
|
|
||||||
if err := fr.WritePing(ack, [8]byte{1, 2, 3, 4, 5, 6, 7, 8}); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
var wantFlags Flags
|
|
||||||
if ack {
|
|
||||||
wantFlags = FlagPingAck
|
|
||||||
}
|
|
||||||
var wantEnc = "\x00\x00\x08\x06" + string(wantFlags) + "\x00\x00\x00\x00" + "\x01\x02\x03\x04\x05\x06\x07\x08"
|
|
||||||
if buf.String() != wantEnc {
|
|
||||||
t.Errorf("encoded as %q; want %q", buf.Bytes(), wantEnc)
|
|
||||||
}
|
|
||||||
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
want := &PingFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
Type: 0x6,
|
|
||||||
Flags: wantFlags,
|
|
||||||
Length: 0x8,
|
|
||||||
StreamID: 0,
|
|
||||||
},
|
|
||||||
Data: [8]byte{1, 2, 3, 4, 5, 6, 7, 8},
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(f, want) {
|
|
||||||
t.Errorf("parsed back %#v; want %#v", f, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadFrameHeader(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
in string
|
|
||||||
want FrameHeader
|
|
||||||
}{
|
|
||||||
{in: "\x00\x00\x00" + "\x00" + "\x00" + "\x00\x00\x00\x00", want: FrameHeader{}},
|
|
||||||
{in: "\x01\x02\x03" + "\x04" + "\x05" + "\x06\x07\x08\x09", want: FrameHeader{
|
|
||||||
Length: 66051, Type: 4, Flags: 5, StreamID: 101124105,
|
|
||||||
}},
|
|
||||||
// Ignore high bit:
|
|
||||||
{in: "\xff\xff\xff" + "\xff" + "\xff" + "\xff\xff\xff\xff", want: FrameHeader{
|
|
||||||
Length: 16777215, Type: 255, Flags: 255, StreamID: 2147483647}},
|
|
||||||
{in: "\xff\xff\xff" + "\xff" + "\xff" + "\x7f\xff\xff\xff", want: FrameHeader{
|
|
||||||
Length: 16777215, Type: 255, Flags: 255, StreamID: 2147483647}},
|
|
||||||
}
|
|
||||||
for i, tt := range tests {
|
|
||||||
got, err := readFrameHeader(make([]byte, 9), strings.NewReader(tt.in))
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("%d. readFrameHeader(%q) = %v", i, tt.in, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
tt.want.valid = true
|
|
||||||
if got != tt.want {
|
|
||||||
t.Errorf("%d. readFrameHeader(%q) = %+v; want %+v", i, tt.in, got, tt.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadWriteFrameHeader(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
len uint32
|
|
||||||
typ FrameType
|
|
||||||
flags Flags
|
|
||||||
streamID uint32
|
|
||||||
}{
|
|
||||||
{len: 0, typ: 255, flags: 1, streamID: 0},
|
|
||||||
{len: 0, typ: 255, flags: 1, streamID: 1},
|
|
||||||
{len: 0, typ: 255, flags: 1, streamID: 255},
|
|
||||||
{len: 0, typ: 255, flags: 1, streamID: 256},
|
|
||||||
{len: 0, typ: 255, flags: 1, streamID: 65535},
|
|
||||||
{len: 0, typ: 255, flags: 1, streamID: 65536},
|
|
||||||
|
|
||||||
{len: 0, typ: 1, flags: 255, streamID: 1},
|
|
||||||
{len: 255, typ: 1, flags: 255, streamID: 1},
|
|
||||||
{len: 256, typ: 1, flags: 255, streamID: 1},
|
|
||||||
{len: 65535, typ: 1, flags: 255, streamID: 1},
|
|
||||||
{len: 65536, typ: 1, flags: 255, streamID: 1},
|
|
||||||
{len: 16777215, typ: 1, flags: 255, streamID: 1},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
fr, buf := testFramer()
|
|
||||||
fr.startWrite(tt.typ, tt.flags, tt.streamID)
|
|
||||||
fr.writeBytes(make([]byte, tt.len))
|
|
||||||
fr.endWrite()
|
|
||||||
fh, err := ReadFrameHeader(buf)
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("ReadFrameHeader(%+v) = %v", tt, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if fh.Type != tt.typ || fh.Flags != tt.flags || fh.Length != tt.len || fh.StreamID != tt.streamID {
|
|
||||||
t.Errorf("ReadFrameHeader(%+v) = %+v; mismatch", tt, fh)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWriteTooLargeFrame(t *testing.T) {
|
|
||||||
fr, _ := testFramer()
|
|
||||||
fr.startWrite(0, 1, 1)
|
|
||||||
fr.writeBytes(make([]byte, 1<<24))
|
|
||||||
err := fr.endWrite()
|
|
||||||
if err != ErrFrameTooLarge {
|
|
||||||
t.Errorf("endWrite = %v; want errFrameTooLarge", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWriteGoAway(t *testing.T) {
|
|
||||||
const debug = "foo"
|
|
||||||
fr, buf := testFramer()
|
|
||||||
if err := fr.WriteGoAway(0x01020304, 0x05060708, []byte(debug)); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
const wantEnc = "\x00\x00\v\a\x00\x00\x00\x00\x00\x01\x02\x03\x04\x05\x06\x07\x08" + debug
|
|
||||||
if buf.String() != wantEnc {
|
|
||||||
t.Errorf("encoded as %q; want %q", buf.Bytes(), wantEnc)
|
|
||||||
}
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
want := &GoAwayFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
Type: 0x7,
|
|
||||||
Flags: 0,
|
|
||||||
Length: uint32(4 + 4 + len(debug)),
|
|
||||||
StreamID: 0,
|
|
||||||
},
|
|
||||||
LastStreamID: 0x01020304,
|
|
||||||
ErrCode: 0x05060708,
|
|
||||||
debugData: []byte(debug),
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(f, want) {
|
|
||||||
t.Fatalf("parsed back:\n%#v\nwant:\n%#v", f, want)
|
|
||||||
}
|
|
||||||
if got := string(f.(*GoAwayFrame).DebugData()); got != debug {
|
|
||||||
t.Errorf("debug data = %q; want %q", got, debug)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWritePushPromise(t *testing.T) {
|
|
||||||
pp := PushPromiseParam{
|
|
||||||
StreamID: 42,
|
|
||||||
PromiseID: 42,
|
|
||||||
BlockFragment: []byte("abc"),
|
|
||||||
}
|
|
||||||
fr, buf := testFramer()
|
|
||||||
if err := fr.WritePushPromise(pp); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
const wantEnc = "\x00\x00\x07\x05\x00\x00\x00\x00*\x00\x00\x00*abc"
|
|
||||||
if buf.String() != wantEnc {
|
|
||||||
t.Errorf("encoded as %q; want %q", buf.Bytes(), wantEnc)
|
|
||||||
}
|
|
||||||
f, err := fr.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
_, ok := f.(*PushPromiseFrame)
|
|
||||||
if !ok {
|
|
||||||
t.Fatalf("got %T; want *PushPromiseFrame", f)
|
|
||||||
}
|
|
||||||
want := &PushPromiseFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
valid: true,
|
|
||||||
Type: 0x5,
|
|
||||||
Flags: 0x0,
|
|
||||||
Length: 0x7,
|
|
||||||
StreamID: 42,
|
|
||||||
},
|
|
||||||
PromiseID: 42,
|
|
||||||
headerFragBuf: []byte("abc"),
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(f, want) {
|
|
||||||
t.Fatalf("parsed back:\n%#v\nwant:\n%#v", f, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// test checkFrameOrder and that HEADERS and CONTINUATION frames can't be intermingled.
|
|
||||||
func TestReadFrameOrder(t *testing.T) {
|
|
||||||
head := func(f *Framer, id uint32, end bool) {
|
|
||||||
f.WriteHeaders(HeadersFrameParam{
|
|
||||||
StreamID: id,
|
|
||||||
BlockFragment: []byte("foo"), // unused, but non-empty
|
|
||||||
EndHeaders: end,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
cont := func(f *Framer, id uint32, end bool) {
|
|
||||||
f.WriteContinuation(id, end, []byte("foo"))
|
|
||||||
}
|
|
||||||
|
|
||||||
tests := [...]struct {
|
|
||||||
name string
|
|
||||||
w func(*Framer)
|
|
||||||
atLeast int
|
|
||||||
wantErr string
|
|
||||||
}{
|
|
||||||
0: {
|
|
||||||
w: func(f *Framer) {
|
|
||||||
head(f, 1, true)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
1: {
|
|
||||||
w: func(f *Framer) {
|
|
||||||
head(f, 1, true)
|
|
||||||
head(f, 2, true)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
2: {
|
|
||||||
wantErr: "got HEADERS for stream 2; expected CONTINUATION following HEADERS for stream 1",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
head(f, 1, false)
|
|
||||||
head(f, 2, true)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
3: {
|
|
||||||
wantErr: "got DATA for stream 1; expected CONTINUATION following HEADERS for stream 1",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
head(f, 1, false)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
4: {
|
|
||||||
w: func(f *Framer) {
|
|
||||||
head(f, 1, false)
|
|
||||||
cont(f, 1, true)
|
|
||||||
head(f, 2, true)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
5: {
|
|
||||||
wantErr: "got CONTINUATION for stream 2; expected stream 1",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
head(f, 1, false)
|
|
||||||
cont(f, 2, true)
|
|
||||||
head(f, 2, true)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
6: {
|
|
||||||
wantErr: "unexpected CONTINUATION for stream 1",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
cont(f, 1, true)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
7: {
|
|
||||||
wantErr: "unexpected CONTINUATION for stream 1",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
cont(f, 1, false)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
8: {
|
|
||||||
wantErr: "HEADERS frame with stream ID 0",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
head(f, 0, true)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
9: {
|
|
||||||
wantErr: "CONTINUATION frame with stream ID 0",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
cont(f, 0, true)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
10: {
|
|
||||||
wantErr: "unexpected CONTINUATION for stream 1",
|
|
||||||
atLeast: 5,
|
|
||||||
w: func(f *Framer) {
|
|
||||||
head(f, 1, false)
|
|
||||||
cont(f, 1, false)
|
|
||||||
cont(f, 1, false)
|
|
||||||
cont(f, 1, false)
|
|
||||||
cont(f, 1, true)
|
|
||||||
cont(f, 1, false)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
for i, tt := range tests {
|
|
||||||
buf := new(bytes.Buffer)
|
|
||||||
f := NewFramer(buf, buf)
|
|
||||||
f.AllowIllegalWrites = true
|
|
||||||
tt.w(f)
|
|
||||||
f.WriteData(1, true, nil) // to test transition away from last step
|
|
||||||
|
|
||||||
var err error
|
|
||||||
n := 0
|
|
||||||
var log bytes.Buffer
|
|
||||||
for {
|
|
||||||
var got Frame
|
|
||||||
got, err = f.ReadFrame()
|
|
||||||
fmt.Fprintf(&log, " read %v, %v\n", got, err)
|
|
||||||
if err != nil {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
n++
|
|
||||||
}
|
|
||||||
if err == io.EOF {
|
|
||||||
err = nil
|
|
||||||
}
|
|
||||||
ok := tt.wantErr == ""
|
|
||||||
if ok && err != nil {
|
|
||||||
t.Errorf("%d. after %d good frames, ReadFrame = %v; want success\n%s", i, n, err, log.Bytes())
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !ok && err != ConnectionError(ErrCodeProtocol) {
|
|
||||||
t.Errorf("%d. after %d good frames, ReadFrame = %v; want ConnectionError(ErrCodeProtocol)\n%s", i, n, err, log.Bytes())
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !((f.errDetail == nil && tt.wantErr == "") || (fmt.Sprint(f.errDetail) == tt.wantErr)) {
|
|
||||||
t.Errorf("%d. framer eror = %q; want %q\n%s", i, f.errDetail, tt.wantErr, log.Bytes())
|
|
||||||
}
|
|
||||||
if n < tt.atLeast {
|
|
||||||
t.Errorf("%d. framer only read %d frames; want at least %d\n%s", i, n, tt.atLeast, log.Bytes())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestMetaFrameHeader(t *testing.T) {
|
|
||||||
write := func(f *Framer, frags ...[]byte) {
|
|
||||||
for i, frag := range frags {
|
|
||||||
end := (i == len(frags)-1)
|
|
||||||
if i == 0 {
|
|
||||||
f.WriteHeaders(HeadersFrameParam{
|
|
||||||
StreamID: 1,
|
|
||||||
BlockFragment: frag,
|
|
||||||
EndHeaders: end,
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
f.WriteContinuation(1, end, frag)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
want := func(flags Flags, length uint32, pairs ...string) *MetaHeadersFrame {
|
|
||||||
mh := &MetaHeadersFrame{
|
|
||||||
HeadersFrame: &HeadersFrame{
|
|
||||||
FrameHeader: FrameHeader{
|
|
||||||
Type: FrameHeaders,
|
|
||||||
Flags: flags,
|
|
||||||
Length: length,
|
|
||||||
StreamID: 1,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
Fields: []hpack.HeaderField(nil),
|
|
||||||
}
|
|
||||||
for len(pairs) > 0 {
|
|
||||||
mh.Fields = append(mh.Fields, hpack.HeaderField{
|
|
||||||
Name: pairs[0],
|
|
||||||
Value: pairs[1],
|
|
||||||
})
|
|
||||||
pairs = pairs[2:]
|
|
||||||
}
|
|
||||||
return mh
|
|
||||||
}
|
|
||||||
truncated := func(mh *MetaHeadersFrame) *MetaHeadersFrame {
|
|
||||||
mh.Truncated = true
|
|
||||||
return mh
|
|
||||||
}
|
|
||||||
|
|
||||||
const noFlags Flags = 0
|
|
||||||
|
|
||||||
oneKBString := strings.Repeat("a", 1<<10)
|
|
||||||
|
|
||||||
tests := [...]struct {
|
|
||||||
name string
|
|
||||||
w func(*Framer)
|
|
||||||
want interface{} // *MetaHeaderFrame or error
|
|
||||||
wantErrReason string
|
|
||||||
maxHeaderListSize uint32
|
|
||||||
}{
|
|
||||||
0: {
|
|
||||||
name: "single_headers",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
var he hpackEncoder
|
|
||||||
all := he.encodeHeaderRaw(t, ":method", "GET", ":path", "/")
|
|
||||||
write(f, all)
|
|
||||||
},
|
|
||||||
want: want(FlagHeadersEndHeaders, 2, ":method", "GET", ":path", "/"),
|
|
||||||
},
|
|
||||||
1: {
|
|
||||||
name: "with_continuation",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
var he hpackEncoder
|
|
||||||
all := he.encodeHeaderRaw(t, ":method", "GET", ":path", "/", "foo", "bar")
|
|
||||||
write(f, all[:1], all[1:])
|
|
||||||
},
|
|
||||||
want: want(noFlags, 1, ":method", "GET", ":path", "/", "foo", "bar"),
|
|
||||||
},
|
|
||||||
2: {
|
|
||||||
name: "with_two_continuation",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
var he hpackEncoder
|
|
||||||
all := he.encodeHeaderRaw(t, ":method", "GET", ":path", "/", "foo", "bar")
|
|
||||||
write(f, all[:2], all[2:4], all[4:])
|
|
||||||
},
|
|
||||||
want: want(noFlags, 2, ":method", "GET", ":path", "/", "foo", "bar"),
|
|
||||||
},
|
|
||||||
3: {
|
|
||||||
name: "big_string_okay",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
var he hpackEncoder
|
|
||||||
all := he.encodeHeaderRaw(t, ":method", "GET", ":path", "/", "foo", oneKBString)
|
|
||||||
write(f, all[:2], all[2:])
|
|
||||||
},
|
|
||||||
want: want(noFlags, 2, ":method", "GET", ":path", "/", "foo", oneKBString),
|
|
||||||
},
|
|
||||||
4: {
|
|
||||||
name: "big_string_error",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
var he hpackEncoder
|
|
||||||
all := he.encodeHeaderRaw(t, ":method", "GET", ":path", "/", "foo", oneKBString)
|
|
||||||
write(f, all[:2], all[2:])
|
|
||||||
},
|
|
||||||
maxHeaderListSize: (1 << 10) / 2,
|
|
||||||
want: ConnectionError(ErrCodeCompression),
|
|
||||||
},
|
|
||||||
5: {
|
|
||||||
name: "max_header_list_truncated",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
var he hpackEncoder
|
|
||||||
var pairs = []string{":method", "GET", ":path", "/"}
|
|
||||||
for i := 0; i < 100; i++ {
|
|
||||||
pairs = append(pairs, "foo", "bar")
|
|
||||||
}
|
|
||||||
all := he.encodeHeaderRaw(t, pairs...)
|
|
||||||
write(f, all[:2], all[2:])
|
|
||||||
},
|
|
||||||
maxHeaderListSize: (1 << 10) / 2,
|
|
||||||
want: truncated(want(noFlags, 2,
|
|
||||||
":method", "GET",
|
|
||||||
":path", "/",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar",
|
|
||||||
"foo", "bar", // 11
|
|
||||||
)),
|
|
||||||
},
|
|
||||||
6: {
|
|
||||||
name: "pseudo_order",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
write(f, encodeHeaderRaw(t,
|
|
||||||
":method", "GET",
|
|
||||||
"foo", "bar",
|
|
||||||
":path", "/", // bogus
|
|
||||||
))
|
|
||||||
},
|
|
||||||
want: StreamError{1, ErrCodeProtocol},
|
|
||||||
wantErrReason: "pseudo header field after regular",
|
|
||||||
},
|
|
||||||
7: {
|
|
||||||
name: "pseudo_unknown",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
write(f, encodeHeaderRaw(t,
|
|
||||||
":unknown", "foo", // bogus
|
|
||||||
"foo", "bar",
|
|
||||||
))
|
|
||||||
},
|
|
||||||
want: StreamError{1, ErrCodeProtocol},
|
|
||||||
wantErrReason: "invalid pseudo-header \":unknown\"",
|
|
||||||
},
|
|
||||||
8: {
|
|
||||||
name: "pseudo_mix_request_response",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
write(f, encodeHeaderRaw(t,
|
|
||||||
":method", "GET",
|
|
||||||
":status", "100",
|
|
||||||
))
|
|
||||||
},
|
|
||||||
want: StreamError{1, ErrCodeProtocol},
|
|
||||||
wantErrReason: "mix of request and response pseudo headers",
|
|
||||||
},
|
|
||||||
9: {
|
|
||||||
name: "pseudo_dup",
|
|
||||||
w: func(f *Framer) {
|
|
||||||
write(f, encodeHeaderRaw(t,
|
|
||||||
":method", "GET",
|
|
||||||
":method", "POST",
|
|
||||||
))
|
|
||||||
},
|
|
||||||
want: StreamError{1, ErrCodeProtocol},
|
|
||||||
wantErrReason: "duplicate pseudo-header \":method\"",
|
|
||||||
},
|
|
||||||
10: {
|
|
||||||
name: "trailer_okay_no_pseudo",
|
|
||||||
w: func(f *Framer) { write(f, encodeHeaderRaw(t, "foo", "bar")) },
|
|
||||||
want: want(FlagHeadersEndHeaders, 8, "foo", "bar"),
|
|
||||||
},
|
|
||||||
11: {
|
|
||||||
name: "invalid_field_name",
|
|
||||||
w: func(f *Framer) { write(f, encodeHeaderRaw(t, "CapitalBad", "x")) },
|
|
||||||
want: StreamError{1, ErrCodeProtocol},
|
|
||||||
wantErrReason: "invalid header field name \"CapitalBad\"",
|
|
||||||
},
|
|
||||||
12: {
|
|
||||||
name: "invalid_field_value",
|
|
||||||
w: func(f *Framer) { write(f, encodeHeaderRaw(t, "key", "bad_null\x00")) },
|
|
||||||
want: StreamError{1, ErrCodeProtocol},
|
|
||||||
wantErrReason: "invalid header field value \"bad_null\\x00\"",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
for i, tt := range tests {
|
|
||||||
buf := new(bytes.Buffer)
|
|
||||||
f := NewFramer(buf, buf)
|
|
||||||
f.ReadMetaHeaders = hpack.NewDecoder(initialHeaderTableSize, nil)
|
|
||||||
f.MaxHeaderListSize = tt.maxHeaderListSize
|
|
||||||
tt.w(f)
|
|
||||||
|
|
||||||
name := tt.name
|
|
||||||
if name == "" {
|
|
||||||
name = fmt.Sprintf("test index %d", i)
|
|
||||||
}
|
|
||||||
|
|
||||||
var got interface{}
|
|
||||||
var err error
|
|
||||||
got, err = f.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
got = err
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(got, tt.want) {
|
|
||||||
if mhg, ok := got.(*MetaHeadersFrame); ok {
|
|
||||||
if mhw, ok := tt.want.(*MetaHeadersFrame); ok {
|
|
||||||
hg := mhg.HeadersFrame
|
|
||||||
hw := mhw.HeadersFrame
|
|
||||||
if hg != nil && hw != nil && !reflect.DeepEqual(*hg, *hw) {
|
|
||||||
t.Errorf("%s: headers differ:\n got: %+v\nwant: %+v\n", name, *hg, *hw)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
str := func(v interface{}) string {
|
|
||||||
if _, ok := v.(error); ok {
|
|
||||||
return fmt.Sprintf("error %v", v)
|
|
||||||
} else {
|
|
||||||
return fmt.Sprintf("value %#v", v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
t.Errorf("%s:\n got: %v\nwant: %s", name, str(got), str(tt.want))
|
|
||||||
}
|
|
||||||
if tt.wantErrReason != "" && tt.wantErrReason != fmt.Sprint(f.errDetail) {
|
|
||||||
t.Errorf("%s: got error reason %q; want %q", name, f.errDetail, tt.wantErrReason)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func encodeHeaderRaw(t *testing.T, pairs ...string) []byte {
|
|
||||||
var he hpackEncoder
|
|
||||||
return he.encodeHeaderRaw(t, pairs...)
|
|
||||||
}
|
|
11
vendor/golang.org/x/net/http2/go15.go
generated
vendored
11
vendor/golang.org/x/net/http2/go15.go
generated
vendored
@ -1,11 +0,0 @@
|
|||||||
// Copyright 2015 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.
|
|
||||||
|
|
||||||
// +build go1.5
|
|
||||||
|
|
||||||
package http2
|
|
||||||
|
|
||||||
import "net/http"
|
|
||||||
|
|
||||||
func requestCancel(req *http.Request) <-chan struct{} { return req.Cancel }
|
|
170
vendor/golang.org/x/net/http2/gotrack.go
generated
vendored
170
vendor/golang.org/x/net/http2/gotrack.go
generated
vendored
@ -1,170 +0,0 @@
|
|||||||
// Copyright 2014 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.
|
|
||||||
|
|
||||||
// Defensive debug-only utility to track that functions run on the
|
|
||||||
// goroutine that they're supposed to.
|
|
||||||
|
|
||||||
package http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"runtime"
|
|
||||||
"strconv"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
var DebugGoroutines = os.Getenv("DEBUG_HTTP2_GOROUTINES") == "1"
|
|
||||||
|
|
||||||
type goroutineLock uint64
|
|
||||||
|
|
||||||
func newGoroutineLock() goroutineLock {
|
|
||||||
if !DebugGoroutines {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
return goroutineLock(curGoroutineID())
|
|
||||||
}
|
|
||||||
|
|
||||||
func (g goroutineLock) check() {
|
|
||||||
if !DebugGoroutines {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if curGoroutineID() != uint64(g) {
|
|
||||||
panic("running on the wrong goroutine")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (g goroutineLock) checkNotOn() {
|
|
||||||
if !DebugGoroutines {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if curGoroutineID() == uint64(g) {
|
|
||||||
panic("running on the wrong goroutine")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var goroutineSpace = []byte("goroutine ")
|
|
||||||
|
|
||||||
func curGoroutineID() uint64 {
|
|
||||||
bp := littleBuf.Get().(*[]byte)
|
|
||||||
defer littleBuf.Put(bp)
|
|
||||||
b := *bp
|
|
||||||
b = b[:runtime.Stack(b, false)]
|
|
||||||
// Parse the 4707 out of "goroutine 4707 ["
|
|
||||||
b = bytes.TrimPrefix(b, goroutineSpace)
|
|
||||||
i := bytes.IndexByte(b, ' ')
|
|
||||||
if i < 0 {
|
|
||||||
panic(fmt.Sprintf("No space found in %q", b))
|
|
||||||
}
|
|
||||||
b = b[:i]
|
|
||||||
n, err := parseUintBytes(b, 10, 64)
|
|
||||||
if err != nil {
|
|
||||||
panic(fmt.Sprintf("Failed to parse goroutine ID out of %q: %v", b, err))
|
|
||||||
}
|
|
||||||
return n
|
|
||||||
}
|
|
||||||
|
|
||||||
var littleBuf = sync.Pool{
|
|
||||||
New: func() interface{} {
|
|
||||||
buf := make([]byte, 64)
|
|
||||||
return &buf
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
// parseUintBytes is like strconv.ParseUint, but using a []byte.
|
|
||||||
func parseUintBytes(s []byte, base int, bitSize int) (n uint64, err error) {
|
|
||||||
var cutoff, maxVal uint64
|
|
||||||
|
|
||||||
if bitSize == 0 {
|
|
||||||
bitSize = int(strconv.IntSize)
|
|
||||||
}
|
|
||||||
|
|
||||||
s0 := s
|
|
||||||
switch {
|
|
||||||
case len(s) < 1:
|
|
||||||
err = strconv.ErrSyntax
|
|
||||||
goto Error
|
|
||||||
|
|
||||||
case 2 <= base && base <= 36:
|
|
||||||
// valid base; nothing to do
|
|
||||||
|
|
||||||
case base == 0:
|
|
||||||
// Look for octal, hex prefix.
|
|
||||||
switch {
|
|
||||||
case s[0] == '0' && len(s) > 1 && (s[1] == 'x' || s[1] == 'X'):
|
|
||||||
base = 16
|
|
||||||
s = s[2:]
|
|
||||||
if len(s) < 1 {
|
|
||||||
err = strconv.ErrSyntax
|
|
||||||
goto Error
|
|
||||||
}
|
|
||||||
case s[0] == '0':
|
|
||||||
base = 8
|
|
||||||
default:
|
|
||||||
base = 10
|
|
||||||
}
|
|
||||||
|
|
||||||
default:
|
|
||||||
err = errors.New("invalid base " + strconv.Itoa(base))
|
|
||||||
goto Error
|
|
||||||
}
|
|
||||||
|
|
||||||
n = 0
|
|
||||||
cutoff = cutoff64(base)
|
|
||||||
maxVal = 1<<uint(bitSize) - 1
|
|
||||||
|
|
||||||
for i := 0; i < len(s); i++ {
|
|
||||||
var v byte
|
|
||||||
d := s[i]
|
|
||||||
switch {
|
|
||||||
case '0' <= d && d <= '9':
|
|
||||||
v = d - '0'
|
|
||||||
case 'a' <= d && d <= 'z':
|
|
||||||
v = d - 'a' + 10
|
|
||||||
case 'A' <= d && d <= 'Z':
|
|
||||||
v = d - 'A' + 10
|
|
||||||
default:
|
|
||||||
n = 0
|
|
||||||
err = strconv.ErrSyntax
|
|
||||||
goto Error
|
|
||||||
}
|
|
||||||
if int(v) >= base {
|
|
||||||
n = 0
|
|
||||||
err = strconv.ErrSyntax
|
|
||||||
goto Error
|
|
||||||
}
|
|
||||||
|
|
||||||
if n >= cutoff {
|
|
||||||
// n*base overflows
|
|
||||||
n = 1<<64 - 1
|
|
||||||
err = strconv.ErrRange
|
|
||||||
goto Error
|
|
||||||
}
|
|
||||||
n *= uint64(base)
|
|
||||||
|
|
||||||
n1 := n + uint64(v)
|
|
||||||
if n1 < n || n1 > maxVal {
|
|
||||||
// n+v overflows
|
|
||||||
n = 1<<64 - 1
|
|
||||||
err = strconv.ErrRange
|
|
||||||
goto Error
|
|
||||||
}
|
|
||||||
n = n1
|
|
||||||
}
|
|
||||||
|
|
||||||
return n, nil
|
|
||||||
|
|
||||||
Error:
|
|
||||||
return n, &strconv.NumError{Func: "ParseUint", Num: string(s0), Err: err}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the first number n such that n*base >= 1<<64.
|
|
||||||
func cutoff64(base int) uint64 {
|
|
||||||
if base < 2 {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
return (1<<64-1)/uint64(base) + 1
|
|
||||||
}
|
|
33
vendor/golang.org/x/net/http2/gotrack_test.go
generated
vendored
33
vendor/golang.org/x/net/http2/gotrack_test.go
generated
vendored
@ -1,33 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestGoroutineLock(t *testing.T) {
|
|
||||||
oldDebug := DebugGoroutines
|
|
||||||
DebugGoroutines = true
|
|
||||||
defer func() { DebugGoroutines = oldDebug }()
|
|
||||||
|
|
||||||
g := newGoroutineLock()
|
|
||||||
g.check()
|
|
||||||
|
|
||||||
sawPanic := make(chan interface{})
|
|
||||||
go func() {
|
|
||||||
defer func() { sawPanic <- recover() }()
|
|
||||||
g.check() // should panic
|
|
||||||
}()
|
|
||||||
e := <-sawPanic
|
|
||||||
if e == nil {
|
|
||||||
t.Fatal("did not see panic from check in other goroutine")
|
|
||||||
}
|
|
||||||
if !strings.Contains(fmt.Sprint(e), "wrong goroutine") {
|
|
||||||
t.Errorf("expected on see panic about running on the wrong goroutine; got %v", e)
|
|
||||||
}
|
|
||||||
}
|
|
8
vendor/golang.org/x/net/http2/h2demo/Makefile
generated
vendored
8
vendor/golang.org/x/net/http2/h2demo/Makefile
generated
vendored
@ -1,8 +0,0 @@
|
|||||||
h2demo.linux: h2demo.go
|
|
||||||
GOOS=linux go build --tags=h2demo -o h2demo.linux .
|
|
||||||
|
|
||||||
FORCE:
|
|
||||||
|
|
||||||
upload: FORCE
|
|
||||||
go install golang.org/x/build/cmd/upload
|
|
||||||
upload --verbose --osarch=linux-amd64 --tags=h2demo --file=go:golang.org/x/net/http2/h2demo --public http2-demo-server-tls/h2demo
|
|
16
vendor/golang.org/x/net/http2/h2demo/README
generated
vendored
16
vendor/golang.org/x/net/http2/h2demo/README
generated
vendored
@ -1,16 +0,0 @@
|
|||||||
|
|
||||||
Client:
|
|
||||||
-- Firefox nightly with about:config network.http.spdy.enabled.http2draft set true
|
|
||||||
-- Chrome: go to chrome://flags/#enable-spdy4, save and restart (button at bottom)
|
|
||||||
|
|
||||||
Make CA:
|
|
||||||
$ openssl genrsa -out rootCA.key 2048
|
|
||||||
$ openssl req -x509 -new -nodes -key rootCA.key -days 1024 -out rootCA.pem
|
|
||||||
... install that to Firefox
|
|
||||||
|
|
||||||
Make cert:
|
|
||||||
$ openssl genrsa -out server.key 2048
|
|
||||||
$ openssl req -new -key server.key -out server.csr
|
|
||||||
$ openssl x509 -req -in server.csr -CA rootCA.pem -CAkey rootCA.key -CAcreateserial -out server.crt -days 500
|
|
||||||
|
|
||||||
|
|
504
vendor/golang.org/x/net/http2/h2demo/h2demo.go
generated
vendored
504
vendor/golang.org/x/net/http2/h2demo/h2demo.go
generated
vendored
@ -1,504 +0,0 @@
|
|||||||
// Copyright 2014 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.
|
|
||||||
|
|
||||||
// +build h2demo
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"crypto/tls"
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"hash/crc32"
|
|
||||||
"image"
|
|
||||||
"image/jpeg"
|
|
||||||
"io"
|
|
||||||
"io/ioutil"
|
|
||||||
"log"
|
|
||||||
"net"
|
|
||||||
"net/http"
|
|
||||||
"path"
|
|
||||||
"regexp"
|
|
||||||
"runtime"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"camlistore.org/pkg/googlestorage"
|
|
||||||
"go4.org/syncutil/singleflight"
|
|
||||||
"golang.org/x/net/http2"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
prod = flag.Bool("prod", false, "Whether to configure itself to be the production http2.golang.org server.")
|
|
||||||
|
|
||||||
httpsAddr = flag.String("https_addr", "localhost:4430", "TLS address to listen on ('host:port' or ':port'). Required.")
|
|
||||||
httpAddr = flag.String("http_addr", "", "Plain HTTP address to listen on ('host:port', or ':port'). Empty means no HTTP.")
|
|
||||||
|
|
||||||
hostHTTP = flag.String("http_host", "", "Optional host or host:port to use for http:// links to this service. By default, this is implied from -http_addr.")
|
|
||||||
hostHTTPS = flag.String("https_host", "", "Optional host or host:port to use for http:// links to this service. By default, this is implied from -https_addr.")
|
|
||||||
)
|
|
||||||
|
|
||||||
func homeOldHTTP(w http.ResponseWriter, r *http.Request) {
|
|
||||||
io.WriteString(w, `<html>
|
|
||||||
<body>
|
|
||||||
<h1>Go + HTTP/2</h1>
|
|
||||||
<p>Welcome to <a href="https://golang.org/">the Go language</a>'s <a href="https://http2.github.io/">HTTP/2</a> demo & interop server.</p>
|
|
||||||
<p>Unfortunately, you're <b>not</b> using HTTP/2 right now. To do so:</p>
|
|
||||||
<ul>
|
|
||||||
<li>Use Firefox Nightly or go to <b>about:config</b> and enable "network.http.spdy.enabled.http2draft"</li>
|
|
||||||
<li>Use Google Chrome Canary and/or go to <b>chrome://flags/#enable-spdy4</b> to <i>Enable SPDY/4</i> (Chrome's name for HTTP/2)</li>
|
|
||||||
</ul>
|
|
||||||
<p>See code & instructions for connecting at <a href="https://github.com/golang/net/tree/master/http2">https://github.com/golang/net/tree/master/http2</a>.</p>
|
|
||||||
|
|
||||||
</body></html>`)
|
|
||||||
}
|
|
||||||
|
|
||||||
func home(w http.ResponseWriter, r *http.Request) {
|
|
||||||
if r.URL.Path != "/" {
|
|
||||||
http.NotFound(w, r)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
io.WriteString(w, `<html>
|
|
||||||
<body>
|
|
||||||
<h1>Go + HTTP/2</h1>
|
|
||||||
|
|
||||||
<p>Welcome to <a href="https://golang.org/">the Go language</a>'s <a
|
|
||||||
href="https://http2.github.io/">HTTP/2</a> demo & interop server.</p>
|
|
||||||
|
|
||||||
<p>Congratulations, <b>you're using HTTP/2 right now</b>.</p>
|
|
||||||
|
|
||||||
<p>This server exists for others in the HTTP/2 community to test their HTTP/2 client implementations and point out flaws in our server.</p>
|
|
||||||
|
|
||||||
<p>
|
|
||||||
The code is at <a href="https://golang.org/x/net/http2">golang.org/x/net/http2</a> and
|
|
||||||
is used transparently by the Go standard library from Go 1.6 and later.
|
|
||||||
</p>
|
|
||||||
|
|
||||||
<p>Contact info: <i>bradfitz@golang.org</i>, or <a
|
|
||||||
href="https://golang.org/s/http2bug">file a bug</a>.</p>
|
|
||||||
|
|
||||||
<h2>Handlers for testing</h2>
|
|
||||||
<ul>
|
|
||||||
<li>GET <a href="/reqinfo">/reqinfo</a> to dump the request + headers received</li>
|
|
||||||
<li>GET <a href="/clockstream">/clockstream</a> streams the current time every second</li>
|
|
||||||
<li>GET <a href="/gophertiles">/gophertiles</a> to see a page with a bunch of images</li>
|
|
||||||
<li>GET <a href="/file/gopher.png">/file/gopher.png</a> for a small file (does If-Modified-Since, Content-Range, etc)</li>
|
|
||||||
<li>GET <a href="/file/go.src.tar.gz">/file/go.src.tar.gz</a> for a larger file (~10 MB)</li>
|
|
||||||
<li>GET <a href="/redirect">/redirect</a> to redirect back to / (this page)</li>
|
|
||||||
<li>GET <a href="/goroutines">/goroutines</a> to see all active goroutines in this server</li>
|
|
||||||
<li>PUT something to <a href="/crc32">/crc32</a> to get a count of number of bytes and its CRC-32</li>
|
|
||||||
<li>PUT something to <a href="/ECHO">/ECHO</a> and it will be streamed back to you capitalized</li>
|
|
||||||
</ul>
|
|
||||||
|
|
||||||
</body></html>`)
|
|
||||||
}
|
|
||||||
|
|
||||||
func reqInfoHandler(w http.ResponseWriter, r *http.Request) {
|
|
||||||
w.Header().Set("Content-Type", "text/plain")
|
|
||||||
fmt.Fprintf(w, "Method: %s\n", r.Method)
|
|
||||||
fmt.Fprintf(w, "Protocol: %s\n", r.Proto)
|
|
||||||
fmt.Fprintf(w, "Host: %s\n", r.Host)
|
|
||||||
fmt.Fprintf(w, "RemoteAddr: %s\n", r.RemoteAddr)
|
|
||||||
fmt.Fprintf(w, "RequestURI: %q\n", r.RequestURI)
|
|
||||||
fmt.Fprintf(w, "URL: %#v\n", r.URL)
|
|
||||||
fmt.Fprintf(w, "Body.ContentLength: %d (-1 means unknown)\n", r.ContentLength)
|
|
||||||
fmt.Fprintf(w, "Close: %v (relevant for HTTP/1 only)\n", r.Close)
|
|
||||||
fmt.Fprintf(w, "TLS: %#v\n", r.TLS)
|
|
||||||
fmt.Fprintf(w, "\nHeaders:\n")
|
|
||||||
r.Header.Write(w)
|
|
||||||
}
|
|
||||||
|
|
||||||
func crcHandler(w http.ResponseWriter, r *http.Request) {
|
|
||||||
if r.Method != "PUT" {
|
|
||||||
http.Error(w, "PUT required.", 400)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
crc := crc32.NewIEEE()
|
|
||||||
n, err := io.Copy(crc, r.Body)
|
|
||||||
if err == nil {
|
|
||||||
w.Header().Set("Content-Type", "text/plain")
|
|
||||||
fmt.Fprintf(w, "bytes=%d, CRC32=%x", n, crc.Sum(nil))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type capitalizeReader struct {
|
|
||||||
r io.Reader
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cr capitalizeReader) Read(p []byte) (n int, err error) {
|
|
||||||
n, err = cr.r.Read(p)
|
|
||||||
for i, b := range p[:n] {
|
|
||||||
if b >= 'a' && b <= 'z' {
|
|
||||||
p[i] = b - ('a' - 'A')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
type flushWriter struct {
|
|
||||||
w io.Writer
|
|
||||||
}
|
|
||||||
|
|
||||||
func (fw flushWriter) Write(p []byte) (n int, err error) {
|
|
||||||
n, err = fw.w.Write(p)
|
|
||||||
if f, ok := fw.w.(http.Flusher); ok {
|
|
||||||
f.Flush()
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func echoCapitalHandler(w http.ResponseWriter, r *http.Request) {
|
|
||||||
if r.Method != "PUT" {
|
|
||||||
http.Error(w, "PUT required.", 400)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
io.Copy(flushWriter{w}, capitalizeReader{r.Body})
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
fsGrp singleflight.Group
|
|
||||||
fsMu sync.Mutex // guards fsCache
|
|
||||||
fsCache = map[string]http.Handler{}
|
|
||||||
)
|
|
||||||
|
|
||||||
// fileServer returns a file-serving handler that proxies URL.
|
|
||||||
// It lazily fetches URL on the first access and caches its contents forever.
|
|
||||||
func fileServer(url string) http.Handler {
|
|
||||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
hi, err := fsGrp.Do(url, func() (interface{}, error) {
|
|
||||||
fsMu.Lock()
|
|
||||||
if h, ok := fsCache[url]; ok {
|
|
||||||
fsMu.Unlock()
|
|
||||||
return h, nil
|
|
||||||
}
|
|
||||||
fsMu.Unlock()
|
|
||||||
|
|
||||||
res, err := http.Get(url)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer res.Body.Close()
|
|
||||||
slurp, err := ioutil.ReadAll(res.Body)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
modTime := time.Now()
|
|
||||||
var h http.Handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
http.ServeContent(w, r, path.Base(url), modTime, bytes.NewReader(slurp))
|
|
||||||
})
|
|
||||||
fsMu.Lock()
|
|
||||||
fsCache[url] = h
|
|
||||||
fsMu.Unlock()
|
|
||||||
return h, nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
http.Error(w, err.Error(), 500)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
hi.(http.Handler).ServeHTTP(w, r)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func clockStreamHandler(w http.ResponseWriter, r *http.Request) {
|
|
||||||
clientGone := w.(http.CloseNotifier).CloseNotify()
|
|
||||||
w.Header().Set("Content-Type", "text/plain")
|
|
||||||
ticker := time.NewTicker(1 * time.Second)
|
|
||||||
defer ticker.Stop()
|
|
||||||
fmt.Fprintf(w, "# ~1KB of junk to force browsers to start rendering immediately: \n")
|
|
||||||
io.WriteString(w, strings.Repeat("# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n", 13))
|
|
||||||
|
|
||||||
for {
|
|
||||||
fmt.Fprintf(w, "%v\n", time.Now())
|
|
||||||
w.(http.Flusher).Flush()
|
|
||||||
select {
|
|
||||||
case <-ticker.C:
|
|
||||||
case <-clientGone:
|
|
||||||
log.Printf("Client %v disconnected from the clock", r.RemoteAddr)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func registerHandlers() {
|
|
||||||
tiles := newGopherTilesHandler()
|
|
||||||
|
|
||||||
mux2 := http.NewServeMux()
|
|
||||||
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
if r.TLS == nil {
|
|
||||||
if r.URL.Path == "/gophertiles" {
|
|
||||||
tiles.ServeHTTP(w, r)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
http.Redirect(w, r, "https://"+httpsHost()+"/", http.StatusFound)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if r.ProtoMajor == 1 {
|
|
||||||
if r.URL.Path == "/reqinfo" {
|
|
||||||
reqInfoHandler(w, r)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
homeOldHTTP(w, r)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
mux2.ServeHTTP(w, r)
|
|
||||||
})
|
|
||||||
mux2.HandleFunc("/", home)
|
|
||||||
mux2.Handle("/file/gopher.png", fileServer("https://golang.org/doc/gopher/frontpage.png"))
|
|
||||||
mux2.Handle("/file/go.src.tar.gz", fileServer("https://storage.googleapis.com/golang/go1.4.1.src.tar.gz"))
|
|
||||||
mux2.HandleFunc("/reqinfo", reqInfoHandler)
|
|
||||||
mux2.HandleFunc("/crc32", crcHandler)
|
|
||||||
mux2.HandleFunc("/ECHO", echoCapitalHandler)
|
|
||||||
mux2.HandleFunc("/clockstream", clockStreamHandler)
|
|
||||||
mux2.Handle("/gophertiles", tiles)
|
|
||||||
mux2.HandleFunc("/redirect", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
http.Redirect(w, r, "/", http.StatusFound)
|
|
||||||
})
|
|
||||||
stripHomedir := regexp.MustCompile(`/(Users|home)/\w+`)
|
|
||||||
mux2.HandleFunc("/goroutines", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
w.Header().Set("Content-Type", "text/plain; charset=utf-8")
|
|
||||||
buf := make([]byte, 2<<20)
|
|
||||||
w.Write(stripHomedir.ReplaceAll(buf[:runtime.Stack(buf, true)], nil))
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func newGopherTilesHandler() http.Handler {
|
|
||||||
const gopherURL = "https://blog.golang.org/go-programming-language-turns-two_gophers.jpg"
|
|
||||||
res, err := http.Get(gopherURL)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
if res.StatusCode != 200 {
|
|
||||||
log.Fatalf("Error fetching %s: %v", gopherURL, res.Status)
|
|
||||||
}
|
|
||||||
slurp, err := ioutil.ReadAll(res.Body)
|
|
||||||
res.Body.Close()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
im, err := jpeg.Decode(bytes.NewReader(slurp))
|
|
||||||
if err != nil {
|
|
||||||
if len(slurp) > 1024 {
|
|
||||||
slurp = slurp[:1024]
|
|
||||||
}
|
|
||||||
log.Fatalf("Failed to decode gopher image: %v (got %q)", err, slurp)
|
|
||||||
}
|
|
||||||
|
|
||||||
type subImager interface {
|
|
||||||
SubImage(image.Rectangle) image.Image
|
|
||||||
}
|
|
||||||
const tileSize = 32
|
|
||||||
xt := im.Bounds().Max.X / tileSize
|
|
||||||
yt := im.Bounds().Max.Y / tileSize
|
|
||||||
var tile [][][]byte // y -> x -> jpeg bytes
|
|
||||||
for yi := 0; yi < yt; yi++ {
|
|
||||||
var row [][]byte
|
|
||||||
for xi := 0; xi < xt; xi++ {
|
|
||||||
si := im.(subImager).SubImage(image.Rectangle{
|
|
||||||
Min: image.Point{xi * tileSize, yi * tileSize},
|
|
||||||
Max: image.Point{(xi + 1) * tileSize, (yi + 1) * tileSize},
|
|
||||||
})
|
|
||||||
buf := new(bytes.Buffer)
|
|
||||||
if err := jpeg.Encode(buf, si, &jpeg.Options{Quality: 90}); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
row = append(row, buf.Bytes())
|
|
||||||
}
|
|
||||||
tile = append(tile, row)
|
|
||||||
}
|
|
||||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
ms, _ := strconv.Atoi(r.FormValue("latency"))
|
|
||||||
const nanosPerMilli = 1e6
|
|
||||||
if r.FormValue("x") != "" {
|
|
||||||
x, _ := strconv.Atoi(r.FormValue("x"))
|
|
||||||
y, _ := strconv.Atoi(r.FormValue("y"))
|
|
||||||
if ms <= 1000 {
|
|
||||||
time.Sleep(time.Duration(ms) * nanosPerMilli)
|
|
||||||
}
|
|
||||||
if x >= 0 && x < xt && y >= 0 && y < yt {
|
|
||||||
http.ServeContent(w, r, "", time.Time{}, bytes.NewReader(tile[y][x]))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
io.WriteString(w, "<html><body onload='showtimes()'>")
|
|
||||||
fmt.Fprintf(w, "A grid of %d tiled images is below. Compare:<p>", xt*yt)
|
|
||||||
for _, ms := range []int{0, 30, 200, 1000} {
|
|
||||||
d := time.Duration(ms) * nanosPerMilli
|
|
||||||
fmt.Fprintf(w, "[<a href='https://%s/gophertiles?latency=%d'>HTTP/2, %v latency</a>] [<a href='http://%s/gophertiles?latency=%d'>HTTP/1, %v latency</a>]<br>\n",
|
|
||||||
httpsHost(), ms, d,
|
|
||||||
httpHost(), ms, d,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
io.WriteString(w, "<p>\n")
|
|
||||||
cacheBust := time.Now().UnixNano()
|
|
||||||
for y := 0; y < yt; y++ {
|
|
||||||
for x := 0; x < xt; x++ {
|
|
||||||
fmt.Fprintf(w, "<img width=%d height=%d src='/gophertiles?x=%d&y=%d&cachebust=%d&latency=%d'>",
|
|
||||||
tileSize, tileSize, x, y, cacheBust, ms)
|
|
||||||
}
|
|
||||||
io.WriteString(w, "<br/>\n")
|
|
||||||
}
|
|
||||||
io.WriteString(w, `<p><div id='loadtimes'></div></p>
|
|
||||||
<script>
|
|
||||||
function showtimes() {
|
|
||||||
var times = 'Times from connection start:<br>'
|
|
||||||
times += 'DOM loaded: ' + (window.performance.timing.domContentLoadedEventEnd - window.performance.timing.connectStart) + 'ms<br>'
|
|
||||||
times += 'DOM complete (images loaded): ' + (window.performance.timing.domComplete - window.performance.timing.connectStart) + 'ms<br>'
|
|
||||||
document.getElementById('loadtimes').innerHTML = times
|
|
||||||
}
|
|
||||||
</script>
|
|
||||||
<hr><a href='/'><< Back to Go HTTP/2 demo server</a></body></html>`)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func httpsHost() string {
|
|
||||||
if *hostHTTPS != "" {
|
|
||||||
return *hostHTTPS
|
|
||||||
}
|
|
||||||
if v := *httpsAddr; strings.HasPrefix(v, ":") {
|
|
||||||
return "localhost" + v
|
|
||||||
} else {
|
|
||||||
return v
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func httpHost() string {
|
|
||||||
if *hostHTTP != "" {
|
|
||||||
return *hostHTTP
|
|
||||||
}
|
|
||||||
if v := *httpAddr; strings.HasPrefix(v, ":") {
|
|
||||||
return "localhost" + v
|
|
||||||
} else {
|
|
||||||
return v
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func serveProdTLS() error {
|
|
||||||
c, err := googlestorage.NewServiceClient()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
slurp := func(key string) ([]byte, error) {
|
|
||||||
const bucket = "http2-demo-server-tls"
|
|
||||||
rc, _, err := c.GetObject(&googlestorage.Object{
|
|
||||||
Bucket: bucket,
|
|
||||||
Key: key,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("Error fetching GCS object %q in bucket %q: %v", key, bucket, err)
|
|
||||||
}
|
|
||||||
defer rc.Close()
|
|
||||||
return ioutil.ReadAll(rc)
|
|
||||||
}
|
|
||||||
certPem, err := slurp("http2.golang.org.chained.pem")
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
keyPem, err := slurp("http2.golang.org.key")
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
cert, err := tls.X509KeyPair(certPem, keyPem)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
srv := &http.Server{
|
|
||||||
TLSConfig: &tls.Config{
|
|
||||||
Certificates: []tls.Certificate{cert},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
http2.ConfigureServer(srv, &http2.Server{})
|
|
||||||
ln, err := net.Listen("tcp", ":443")
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return srv.Serve(tls.NewListener(tcpKeepAliveListener{ln.(*net.TCPListener)}, srv.TLSConfig))
|
|
||||||
}
|
|
||||||
|
|
||||||
type tcpKeepAliveListener struct {
|
|
||||||
*net.TCPListener
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ln tcpKeepAliveListener) Accept() (c net.Conn, err error) {
|
|
||||||
tc, err := ln.AcceptTCP()
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
tc.SetKeepAlive(true)
|
|
||||||
tc.SetKeepAlivePeriod(3 * time.Minute)
|
|
||||||
return tc, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func serveProd() error {
|
|
||||||
errc := make(chan error, 2)
|
|
||||||
go func() { errc <- http.ListenAndServe(":80", nil) }()
|
|
||||||
go func() { errc <- serveProdTLS() }()
|
|
||||||
return <-errc
|
|
||||||
}
|
|
||||||
|
|
||||||
const idleTimeout = 5 * time.Minute
|
|
||||||
const activeTimeout = 10 * time.Minute
|
|
||||||
|
|
||||||
// TODO: put this into the standard library and actually send
|
|
||||||
// PING frames and GOAWAY, etc: golang.org/issue/14204
|
|
||||||
func idleTimeoutHook() func(net.Conn, http.ConnState) {
|
|
||||||
var mu sync.Mutex
|
|
||||||
m := map[net.Conn]*time.Timer{}
|
|
||||||
return func(c net.Conn, cs http.ConnState) {
|
|
||||||
mu.Lock()
|
|
||||||
defer mu.Unlock()
|
|
||||||
if t, ok := m[c]; ok {
|
|
||||||
delete(m, c)
|
|
||||||
t.Stop()
|
|
||||||
}
|
|
||||||
var d time.Duration
|
|
||||||
switch cs {
|
|
||||||
case http.StateNew, http.StateIdle:
|
|
||||||
d = idleTimeout
|
|
||||||
case http.StateActive:
|
|
||||||
d = activeTimeout
|
|
||||||
default:
|
|
||||||
return
|
|
||||||
}
|
|
||||||
m[c] = time.AfterFunc(d, func() {
|
|
||||||
log.Printf("closing idle conn %v after %v", c.RemoteAddr(), d)
|
|
||||||
go c.Close()
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
var srv http.Server
|
|
||||||
flag.BoolVar(&http2.VerboseLogs, "verbose", false, "Verbose HTTP/2 debugging.")
|
|
||||||
flag.Parse()
|
|
||||||
srv.Addr = *httpsAddr
|
|
||||||
srv.ConnState = idleTimeoutHook()
|
|
||||||
|
|
||||||
registerHandlers()
|
|
||||||
|
|
||||||
if *prod {
|
|
||||||
*hostHTTP = "http2.golang.org"
|
|
||||||
*hostHTTPS = "http2.golang.org"
|
|
||||||
log.Fatal(serveProd())
|
|
||||||
}
|
|
||||||
|
|
||||||
url := "https://" + httpsHost() + "/"
|
|
||||||
log.Printf("Listening on " + url)
|
|
||||||
http2.ConfigureServer(&srv, &http2.Server{})
|
|
||||||
|
|
||||||
if *httpAddr != "" {
|
|
||||||
go func() {
|
|
||||||
log.Printf("Listening on http://" + httpHost() + "/ (for unencrypted HTTP/1)")
|
|
||||||
log.Fatal(http.ListenAndServe(*httpAddr, nil))
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
log.Fatal(srv.ListenAndServeTLS("server.crt", "server.key"))
|
|
||||||
}()
|
|
||||||
select {}
|
|
||||||
}
|
|
302
vendor/golang.org/x/net/http2/h2demo/launch.go
generated
vendored
302
vendor/golang.org/x/net/http2/h2demo/launch.go
generated
vendored
@ -1,302 +0,0 @@
|
|||||||
// Copyright 2014 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.
|
|
||||||
|
|
||||||
// +build ignore
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bufio"
|
|
||||||
"bytes"
|
|
||||||
"encoding/json"
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"io/ioutil"
|
|
||||||
"log"
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"golang.org/x/oauth2"
|
|
||||||
"golang.org/x/oauth2/google"
|
|
||||||
compute "google.golang.org/api/compute/v1"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
proj = flag.String("project", "symbolic-datum-552", "name of Project")
|
|
||||||
zone = flag.String("zone", "us-central1-a", "GCE zone")
|
|
||||||
mach = flag.String("machinetype", "n1-standard-1", "Machine type")
|
|
||||||
instName = flag.String("instance_name", "http2-demo", "Name of VM instance.")
|
|
||||||
sshPub = flag.String("ssh_public_key", "", "ssh public key file to authorize. Can modify later in Google's web UI anyway.")
|
|
||||||
staticIP = flag.String("static_ip", "130.211.116.44", "Static IP to use. If empty, automatic.")
|
|
||||||
|
|
||||||
writeObject = flag.String("write_object", "", "If non-empty, a VM isn't created and the flag value is Google Cloud Storage bucket/object to write. The contents from stdin.")
|
|
||||||
publicObject = flag.Bool("write_object_is_public", false, "Whether the object created by --write_object should be public.")
|
|
||||||
)
|
|
||||||
|
|
||||||
func readFile(v string) string {
|
|
||||||
slurp, err := ioutil.ReadFile(v)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Error reading %s: %v", v, err)
|
|
||||||
}
|
|
||||||
return strings.TrimSpace(string(slurp))
|
|
||||||
}
|
|
||||||
|
|
||||||
var config = &oauth2.Config{
|
|
||||||
// The client-id and secret should be for an "Installed Application" when using
|
|
||||||
// the CLI. Later we'll use a web application with a callback.
|
|
||||||
ClientID: readFile("client-id.dat"),
|
|
||||||
ClientSecret: readFile("client-secret.dat"),
|
|
||||||
Endpoint: google.Endpoint,
|
|
||||||
Scopes: []string{
|
|
||||||
compute.DevstorageFullControlScope,
|
|
||||||
compute.ComputeScope,
|
|
||||||
"https://www.googleapis.com/auth/sqlservice",
|
|
||||||
"https://www.googleapis.com/auth/sqlservice.admin",
|
|
||||||
},
|
|
||||||
RedirectURL: "urn:ietf:wg:oauth:2.0:oob",
|
|
||||||
}
|
|
||||||
|
|
||||||
const baseConfig = `#cloud-config
|
|
||||||
coreos:
|
|
||||||
units:
|
|
||||||
- name: h2demo.service
|
|
||||||
command: start
|
|
||||||
content: |
|
|
||||||
[Unit]
|
|
||||||
Description=HTTP2 Demo
|
|
||||||
|
|
||||||
[Service]
|
|
||||||
ExecStartPre=/bin/bash -c 'mkdir -p /opt/bin && curl -s -o /opt/bin/h2demo http://storage.googleapis.com/http2-demo-server-tls/h2demo && chmod +x /opt/bin/h2demo'
|
|
||||||
ExecStart=/opt/bin/h2demo --prod
|
|
||||||
RestartSec=5s
|
|
||||||
Restart=always
|
|
||||||
Type=simple
|
|
||||||
|
|
||||||
[Install]
|
|
||||||
WantedBy=multi-user.target
|
|
||||||
`
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
flag.Parse()
|
|
||||||
if *proj == "" {
|
|
||||||
log.Fatalf("Missing --project flag")
|
|
||||||
}
|
|
||||||
prefix := "https://www.googleapis.com/compute/v1/projects/" + *proj
|
|
||||||
machType := prefix + "/zones/" + *zone + "/machineTypes/" + *mach
|
|
||||||
|
|
||||||
const tokenFileName = "token.dat"
|
|
||||||
tokenFile := tokenCacheFile(tokenFileName)
|
|
||||||
tokenSource := oauth2.ReuseTokenSource(nil, tokenFile)
|
|
||||||
token, err := tokenSource.Token()
|
|
||||||
if err != nil {
|
|
||||||
if *writeObject != "" {
|
|
||||||
log.Fatalf("Can't use --write_object without a valid token.dat file already cached.")
|
|
||||||
}
|
|
||||||
log.Printf("Error getting token from %s: %v", tokenFileName, err)
|
|
||||||
log.Printf("Get auth code from %v", config.AuthCodeURL("my-state"))
|
|
||||||
fmt.Print("\nEnter auth code: ")
|
|
||||||
sc := bufio.NewScanner(os.Stdin)
|
|
||||||
sc.Scan()
|
|
||||||
authCode := strings.TrimSpace(sc.Text())
|
|
||||||
token, err = config.Exchange(oauth2.NoContext, authCode)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Error exchanging auth code for a token: %v", err)
|
|
||||||
}
|
|
||||||
if err := tokenFile.WriteToken(token); err != nil {
|
|
||||||
log.Fatalf("Error writing to %s: %v", tokenFileName, err)
|
|
||||||
}
|
|
||||||
tokenSource = oauth2.ReuseTokenSource(token, nil)
|
|
||||||
}
|
|
||||||
|
|
||||||
oauthClient := oauth2.NewClient(oauth2.NoContext, tokenSource)
|
|
||||||
|
|
||||||
if *writeObject != "" {
|
|
||||||
writeCloudStorageObject(oauthClient)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
computeService, _ := compute.New(oauthClient)
|
|
||||||
|
|
||||||
natIP := *staticIP
|
|
||||||
if natIP == "" {
|
|
||||||
// Try to find it by name.
|
|
||||||
aggAddrList, err := computeService.Addresses.AggregatedList(*proj).Do()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
// http://godoc.org/code.google.com/p/google-api-go-client/compute/v1#AddressAggregatedList
|
|
||||||
IPLoop:
|
|
||||||
for _, asl := range aggAddrList.Items {
|
|
||||||
for _, addr := range asl.Addresses {
|
|
||||||
if addr.Name == *instName+"-ip" && addr.Status == "RESERVED" {
|
|
||||||
natIP = addr.Address
|
|
||||||
break IPLoop
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
cloudConfig := baseConfig
|
|
||||||
if *sshPub != "" {
|
|
||||||
key := strings.TrimSpace(readFile(*sshPub))
|
|
||||||
cloudConfig += fmt.Sprintf("\nssh_authorized_keys:\n - %s\n", key)
|
|
||||||
}
|
|
||||||
if os.Getenv("USER") == "bradfitz" {
|
|
||||||
cloudConfig += fmt.Sprintf("\nssh_authorized_keys:\n - %s\n", "ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAIEAwks9dwWKlRC+73gRbvYtVg0vdCwDSuIlyt4z6xa/YU/jTDynM4R4W10hm2tPjy8iR1k8XhDv4/qdxe6m07NjG/By1tkmGpm1mGwho4Pr5kbAAy/Qg+NLCSdAYnnE00FQEcFOC15GFVMOW2AzDGKisReohwH9eIzHPzdYQNPRWXE= bradfitz@papag.bradfitz.com")
|
|
||||||
}
|
|
||||||
const maxCloudConfig = 32 << 10 // per compute API docs
|
|
||||||
if len(cloudConfig) > maxCloudConfig {
|
|
||||||
log.Fatalf("cloud config length of %d bytes is over %d byte limit", len(cloudConfig), maxCloudConfig)
|
|
||||||
}
|
|
||||||
|
|
||||||
instance := &compute.Instance{
|
|
||||||
Name: *instName,
|
|
||||||
Description: "Go Builder",
|
|
||||||
MachineType: machType,
|
|
||||||
Disks: []*compute.AttachedDisk{instanceDisk(computeService)},
|
|
||||||
Tags: &compute.Tags{
|
|
||||||
Items: []string{"http-server", "https-server"},
|
|
||||||
},
|
|
||||||
Metadata: &compute.Metadata{
|
|
||||||
Items: []*compute.MetadataItems{
|
|
||||||
{
|
|
||||||
Key: "user-data",
|
|
||||||
Value: &cloudConfig,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
NetworkInterfaces: []*compute.NetworkInterface{
|
|
||||||
&compute.NetworkInterface{
|
|
||||||
AccessConfigs: []*compute.AccessConfig{
|
|
||||||
&compute.AccessConfig{
|
|
||||||
Type: "ONE_TO_ONE_NAT",
|
|
||||||
Name: "External NAT",
|
|
||||||
NatIP: natIP,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
Network: prefix + "/global/networks/default",
|
|
||||||
},
|
|
||||||
},
|
|
||||||
ServiceAccounts: []*compute.ServiceAccount{
|
|
||||||
{
|
|
||||||
Email: "default",
|
|
||||||
Scopes: []string{
|
|
||||||
compute.DevstorageFullControlScope,
|
|
||||||
compute.ComputeScope,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Printf("Creating instance...")
|
|
||||||
op, err := computeService.Instances.Insert(*proj, *zone, instance).Do()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to create instance: %v", err)
|
|
||||||
}
|
|
||||||
opName := op.Name
|
|
||||||
log.Printf("Created. Waiting on operation %v", opName)
|
|
||||||
OpLoop:
|
|
||||||
for {
|
|
||||||
time.Sleep(2 * time.Second)
|
|
||||||
op, err := computeService.ZoneOperations.Get(*proj, *zone, opName).Do()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to get op %s: %v", opName, err)
|
|
||||||
}
|
|
||||||
switch op.Status {
|
|
||||||
case "PENDING", "RUNNING":
|
|
||||||
log.Printf("Waiting on operation %v", opName)
|
|
||||||
continue
|
|
||||||
case "DONE":
|
|
||||||
if op.Error != nil {
|
|
||||||
for _, operr := range op.Error.Errors {
|
|
||||||
log.Printf("Error: %+v", operr)
|
|
||||||
}
|
|
||||||
log.Fatalf("Failed to start.")
|
|
||||||
}
|
|
||||||
log.Printf("Success. %+v", op)
|
|
||||||
break OpLoop
|
|
||||||
default:
|
|
||||||
log.Fatalf("Unknown status %q: %+v", op.Status, op)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
inst, err := computeService.Instances.Get(*proj, *zone, *instName).Do()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Error getting instance after creation: %v", err)
|
|
||||||
}
|
|
||||||
ij, _ := json.MarshalIndent(inst, "", " ")
|
|
||||||
log.Printf("Instance: %s", ij)
|
|
||||||
}
|
|
||||||
|
|
||||||
func instanceDisk(svc *compute.Service) *compute.AttachedDisk {
|
|
||||||
const imageURL = "https://www.googleapis.com/compute/v1/projects/coreos-cloud/global/images/coreos-stable-444-5-0-v20141016"
|
|
||||||
diskName := *instName + "-disk"
|
|
||||||
|
|
||||||
return &compute.AttachedDisk{
|
|
||||||
AutoDelete: true,
|
|
||||||
Boot: true,
|
|
||||||
Type: "PERSISTENT",
|
|
||||||
InitializeParams: &compute.AttachedDiskInitializeParams{
|
|
||||||
DiskName: diskName,
|
|
||||||
SourceImage: imageURL,
|
|
||||||
DiskSizeGb: 50,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func writeCloudStorageObject(httpClient *http.Client) {
|
|
||||||
content := os.Stdin
|
|
||||||
const maxSlurp = 1 << 20
|
|
||||||
var buf bytes.Buffer
|
|
||||||
n, err := io.CopyN(&buf, content, maxSlurp)
|
|
||||||
if err != nil && err != io.EOF {
|
|
||||||
log.Fatalf("Error reading from stdin: %v, %v", n, err)
|
|
||||||
}
|
|
||||||
contentType := http.DetectContentType(buf.Bytes())
|
|
||||||
|
|
||||||
req, err := http.NewRequest("PUT", "https://storage.googleapis.com/"+*writeObject, io.MultiReader(&buf, content))
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
req.Header.Set("x-goog-api-version", "2")
|
|
||||||
if *publicObject {
|
|
||||||
req.Header.Set("x-goog-acl", "public-read")
|
|
||||||
}
|
|
||||||
req.Header.Set("Content-Type", contentType)
|
|
||||||
res, err := httpClient.Do(req)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
if res.StatusCode != 200 {
|
|
||||||
res.Write(os.Stderr)
|
|
||||||
log.Fatalf("Failed.")
|
|
||||||
}
|
|
||||||
log.Printf("Success.")
|
|
||||||
os.Exit(0)
|
|
||||||
}
|
|
||||||
|
|
||||||
type tokenCacheFile string
|
|
||||||
|
|
||||||
func (f tokenCacheFile) Token() (*oauth2.Token, error) {
|
|
||||||
slurp, err := ioutil.ReadFile(string(f))
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
t := new(oauth2.Token)
|
|
||||||
if err := json.Unmarshal(slurp, t); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return t, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f tokenCacheFile) WriteToken(t *oauth2.Token) error {
|
|
||||||
jt, err := json.Marshal(t)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return ioutil.WriteFile(string(f), jt, 0600)
|
|
||||||
}
|
|
27
vendor/golang.org/x/net/http2/h2demo/rootCA.key
generated
vendored
27
vendor/golang.org/x/net/http2/h2demo/rootCA.key
generated
vendored
@ -1,27 +0,0 @@
|
|||||||
-----BEGIN RSA PRIVATE KEY-----
|
|
||||||
MIIEowIBAAKCAQEAt5fAjp4fTcekWUTfzsp0kyih1OYbsGL0KX1eRbSSR8Od0+9Q
|
|
||||||
62Hyny+GFwMTb4A/KU8mssoHvcceSAAbwfbxFK/+s51TobqUnORZrOoTZjkUygby
|
|
||||||
XDSK99YBbcR1Pip8vwMTm4XKuLtCigeBBdjjAQdgUO28LENGlsMnmeYkJfODVGnV
|
|
||||||
mr5Ltb9ANA8IKyTfsnHJ4iOCS/PlPbUj2q7YnoVLposUBMlgUb/CykX3mOoLb4yJ
|
|
||||||
JQyA/iST6ZxiIEj36D4yWZ5lg7YJl+UiiBQHGCnPdGyipqV06ex0heYWcaiW8LWZ
|
|
||||||
SUQ93jQ+WVCH8hT7DQO1dmsvUmXlq/JeAlwQ/QIDAQABAoIBAFFHV7JMAqPWnMYA
|
|
||||||
nezY6J81v9+XN+7xABNWM2Q8uv4WdksbigGLTXR3/680Z2hXqJ7LMeC5XJACFT/e
|
|
||||||
/Gr0vmpgOCygnCPfjGehGKpavtfksXV3edikUlnCXsOP1C//c1bFL+sMYmFCVgTx
|
|
||||||
qYdDK8yKzXNGrKYT6q5YG7IglyRNV1rsQa8lM/5taFYiD1Ck/3tQi3YIq8Lcuser
|
|
||||||
hrxsMABcQ6mi+EIvG6Xr4mfJug0dGJMHG4RG1UGFQn6RXrQq2+q53fC8ZbVUSi0j
|
|
||||||
NQ918aKFzktwv+DouKU0ME4I9toks03gM860bAL7zCbKGmwR3hfgX/TqzVCWpG9E
|
|
||||||
LDVfvekCgYEA8fk9N53jbBRmULUGEf4qWypcLGiZnNU0OeXWpbPV9aa3H0VDytA7
|
|
||||||
8fCN2dPAVDPqlthMDdVe983NCNwp2Yo8ZimDgowyIAKhdC25s1kejuaiH9OAPj3c
|
|
||||||
0f8KbriYX4n8zNHxFwK6Ae3pQ6EqOLJVCUsziUaZX9nyKY5aZlyX6xcCgYEAwjws
|
|
||||||
K62PjC64U5wYddNLp+kNdJ4edx+a7qBb3mEgPvSFT2RO3/xafJyG8kQB30Mfstjd
|
|
||||||
bRxyUV6N0vtX1zA7VQtRUAvfGCecpMo+VQZzcHXKzoRTnQ7eZg4Lmj5fQ9tOAKAo
|
|
||||||
QCVBoSW/DI4PZL26CAMDcAba4Pa22ooLapoRIQsCgYA6pIfkkbxLNkpxpt2YwLtt
|
|
||||||
Kr/590O7UaR9n6k8sW/aQBRDXNsILR1KDl2ifAIxpf9lnXgZJiwE7HiTfCAcW7c1
|
|
||||||
nzwDCI0hWuHcMTS/NYsFYPnLsstyyjVZI3FY0h4DkYKV9Q9z3zJLQ2hz/nwoD3gy
|
|
||||||
b2pHC7giFcTts1VPV4Nt8wKBgHeFn4ihHJweg76vZz3Z78w7VNRWGFklUalVdDK7
|
|
||||||
gaQ7w2y/ROn/146mo0OhJaXFIFRlrpvdzVrU3GDf2YXJYDlM5ZRkObwbZADjksev
|
|
||||||
WInzcgDy3KDg7WnPasRXbTfMU4t/AkW2p1QKbi3DnSVYuokDkbH2Beo45vxDxhKr
|
|
||||||
C69RAoGBAIyo3+OJenoZmoNzNJl2WPW5MeBUzSh8T/bgyjFTdqFHF5WiYRD/lfHj
|
|
||||||
x9Glyw2nutuT4hlOqHvKhgTYdDMsF2oQ72fe3v8Q5FU7FuKndNPEAyvKNXZaShVA
|
|
||||||
hnlhv5DjXKb0wFWnt5PCCiQLtzG0yyHaITrrEme7FikkIcTxaX/Y
|
|
||||||
-----END RSA PRIVATE KEY-----
|
|
26
vendor/golang.org/x/net/http2/h2demo/rootCA.pem
generated
vendored
26
vendor/golang.org/x/net/http2/h2demo/rootCA.pem
generated
vendored
@ -1,26 +0,0 @@
|
|||||||
-----BEGIN CERTIFICATE-----
|
|
||||||
MIIEWjCCA0KgAwIBAgIJALfRlWsI8YQHMA0GCSqGSIb3DQEBBQUAMHsxCzAJBgNV
|
|
||||||
BAYTAlVTMQswCQYDVQQIEwJDQTEWMBQGA1UEBxMNU2FuIEZyYW5jaXNjbzEUMBIG
|
|
||||||
A1UEChMLQnJhZGZpdHppbmMxEjAQBgNVBAMTCWxvY2FsaG9zdDEdMBsGCSqGSIb3
|
|
||||||
DQEJARYOYnJhZEBkYW5nYS5jb20wHhcNMTQwNzE1MjA0NjA1WhcNMTcwNTA0MjA0
|
|
||||||
NjA1WjB7MQswCQYDVQQGEwJVUzELMAkGA1UECBMCQ0ExFjAUBgNVBAcTDVNhbiBG
|
|
||||||
cmFuY2lzY28xFDASBgNVBAoTC0JyYWRmaXR6aW5jMRIwEAYDVQQDEwlsb2NhbGhv
|
|
||||||
c3QxHTAbBgkqhkiG9w0BCQEWDmJyYWRAZGFuZ2EuY29tMIIBIjANBgkqhkiG9w0B
|
|
||||||
AQEFAAOCAQ8AMIIBCgKCAQEAt5fAjp4fTcekWUTfzsp0kyih1OYbsGL0KX1eRbSS
|
|
||||||
R8Od0+9Q62Hyny+GFwMTb4A/KU8mssoHvcceSAAbwfbxFK/+s51TobqUnORZrOoT
|
|
||||||
ZjkUygbyXDSK99YBbcR1Pip8vwMTm4XKuLtCigeBBdjjAQdgUO28LENGlsMnmeYk
|
|
||||||
JfODVGnVmr5Ltb9ANA8IKyTfsnHJ4iOCS/PlPbUj2q7YnoVLposUBMlgUb/CykX3
|
|
||||||
mOoLb4yJJQyA/iST6ZxiIEj36D4yWZ5lg7YJl+UiiBQHGCnPdGyipqV06ex0heYW
|
|
||||||
caiW8LWZSUQ93jQ+WVCH8hT7DQO1dmsvUmXlq/JeAlwQ/QIDAQABo4HgMIHdMB0G
|
|
||||||
A1UdDgQWBBRcAROthS4P4U7vTfjByC569R7E6DCBrQYDVR0jBIGlMIGigBRcAROt
|
|
||||||
hS4P4U7vTfjByC569R7E6KF/pH0wezELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAkNB
|
|
||||||
MRYwFAYDVQQHEw1TYW4gRnJhbmNpc2NvMRQwEgYDVQQKEwtCcmFkZml0emluYzES
|
|
||||||
MBAGA1UEAxMJbG9jYWxob3N0MR0wGwYJKoZIhvcNAQkBFg5icmFkQGRhbmdhLmNv
|
|
||||||
bYIJALfRlWsI8YQHMAwGA1UdEwQFMAMBAf8wDQYJKoZIhvcNAQEFBQADggEBAG6h
|
|
||||||
U9f9sNH0/6oBbGGy2EVU0UgITUQIrFWo9rFkrW5k/XkDjQm+3lzjT0iGR4IxE/Ao
|
|
||||||
eU6sQhua7wrWeFEn47GL98lnCsJdD7oZNhFmQ95Tb/LnDUjs5Yj9brP0NWzXfYU4
|
|
||||||
UK2ZnINJRcJpB8iRCaCxE8DdcUF0XqIEq6pA272snoLmiXLMvNl3kYEdm+je6voD
|
|
||||||
58SNVEUsztzQyXmJEhCpwVI0A6QCjzXj+qvpmw3ZZHi8JwXei8ZZBLTSFBki8Z7n
|
|
||||||
sH9BBH38/SzUmAN4QHSPy1gjqm00OAE8NaYDkh/bzE4d7mLGGMWp/WE3KPSu82HF
|
|
||||||
kPe6XoSbiLm/kxk32T0=
|
|
||||||
-----END CERTIFICATE-----
|
|
1
vendor/golang.org/x/net/http2/h2demo/rootCA.srl
generated
vendored
1
vendor/golang.org/x/net/http2/h2demo/rootCA.srl
generated
vendored
@ -1 +0,0 @@
|
|||||||
E2CE26BF3285059C
|
|
20
vendor/golang.org/x/net/http2/h2demo/server.crt
generated
vendored
20
vendor/golang.org/x/net/http2/h2demo/server.crt
generated
vendored
@ -1,20 +0,0 @@
|
|||||||
-----BEGIN CERTIFICATE-----
|
|
||||||
MIIDPjCCAiYCCQDizia/MoUFnDANBgkqhkiG9w0BAQUFADB7MQswCQYDVQQGEwJV
|
|
||||||
UzELMAkGA1UECBMCQ0ExFjAUBgNVBAcTDVNhbiBGcmFuY2lzY28xFDASBgNVBAoT
|
|
||||||
C0JyYWRmaXR6aW5jMRIwEAYDVQQDEwlsb2NhbGhvc3QxHTAbBgkqhkiG9w0BCQEW
|
|
||||||
DmJyYWRAZGFuZ2EuY29tMB4XDTE0MDcxNTIwNTAyN1oXDTE1MTEyNzIwNTAyN1ow
|
|
||||||
RzELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAkNBMQswCQYDVQQHEwJTRjEeMBwGA1UE
|
|
||||||
ChMVYnJhZGZpdHogaHR0cDIgc2VydmVyMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8A
|
|
||||||
MIIBCgKCAQEAs1Y9CyLFrdL8VQWN1WaifDqaZFnoqjHhCMlc1TfG2zA+InDifx2l
|
|
||||||
gZD3o8FeNnAcfM2sPlk3+ZleOYw9P/CklFVDlvqmpCv9ss/BEp/dDaWvy1LmJ4c2
|
|
||||||
dbQJfmTxn7CV1H3TsVJvKdwFmdoABb41NoBp6+NNO7OtDyhbIMiCI0pL3Nefb3HL
|
|
||||||
A7hIMo3DYbORTtJLTIH9W8YKrEWL0lwHLrYFx/UdutZnv+HjdmO6vCN4na55mjws
|
|
||||||
/vjKQUmc7xeY7Xe20xDEG2oDKVkL2eD7FfyrYMS3rO1ExP2KSqlXYG/1S9I/fz88
|
|
||||||
F0GK7HX55b5WjZCl2J3ERVdnv/0MQv+sYQIDAQABMA0GCSqGSIb3DQEBBQUAA4IB
|
|
||||||
AQC0zL+n/YpRZOdulSu9tS8FxrstXqGWoxfe+vIUgqfMZ5+0MkjJ/vW0FqlLDl2R
|
|
||||||
rn4XaR3e7FmWkwdDVbq/UB6lPmoAaFkCgh9/5oapMaclNVNnfF3fjCJfRr+qj/iD
|
|
||||||
EmJStTIN0ZuUjAlpiACmfnpEU55PafT5Zx+i1yE4FGjw8bJpFoyD4Hnm54nGjX19
|
|
||||||
KeCuvcYFUPnBm3lcL0FalF2AjqV02WTHYNQk7YF/oeO7NKBoEgvGvKG3x+xaOeBI
|
|
||||||
dwvdq175ZsGul30h+QjrRlXhH/twcuaT3GSdoysDl9cCYE8f1Mk8PD6gan3uBCJU
|
|
||||||
90p6/CbU71bGbfpM2PHot2fm
|
|
||||||
-----END CERTIFICATE-----
|
|
27
vendor/golang.org/x/net/http2/h2demo/server.key
generated
vendored
27
vendor/golang.org/x/net/http2/h2demo/server.key
generated
vendored
@ -1,27 +0,0 @@
|
|||||||
-----BEGIN RSA PRIVATE KEY-----
|
|
||||||
MIIEowIBAAKCAQEAs1Y9CyLFrdL8VQWN1WaifDqaZFnoqjHhCMlc1TfG2zA+InDi
|
|
||||||
fx2lgZD3o8FeNnAcfM2sPlk3+ZleOYw9P/CklFVDlvqmpCv9ss/BEp/dDaWvy1Lm
|
|
||||||
J4c2dbQJfmTxn7CV1H3TsVJvKdwFmdoABb41NoBp6+NNO7OtDyhbIMiCI0pL3Nef
|
|
||||||
b3HLA7hIMo3DYbORTtJLTIH9W8YKrEWL0lwHLrYFx/UdutZnv+HjdmO6vCN4na55
|
|
||||||
mjws/vjKQUmc7xeY7Xe20xDEG2oDKVkL2eD7FfyrYMS3rO1ExP2KSqlXYG/1S9I/
|
|
||||||
fz88F0GK7HX55b5WjZCl2J3ERVdnv/0MQv+sYQIDAQABAoIBADQ2spUwbY+bcz4p
|
|
||||||
3M66ECrNQTBggP40gYl2XyHxGGOu2xhZ94f9ELf1hjRWU2DUKWco1rJcdZClV6q3
|
|
||||||
qwmXvcM2Q/SMS8JW0ImkNVl/0/NqPxGatEnj8zY30d/L8hGFb0orzFu/XYA5gCP4
|
|
||||||
NbN2WrXgk3ZLeqwcNxHHtSiJWGJ/fPyeDWAu/apy75u9Xf2GlzBZmV6HYD9EfK80
|
|
||||||
LTlI60f5FO487CrJnboL7ovPJrIHn+k05xRQqwma4orpz932rTXnTjs9Lg6KtbQN
|
|
||||||
a7PrqfAntIISgr11a66Mng3IYH1lYqJsWJJwX/xHT4WLEy0EH4/0+PfYemJekz2+
|
|
||||||
Co62drECgYEA6O9zVJZXrLSDsIi54cfxA7nEZWm5CAtkYWeAHa4EJ+IlZ7gIf9sL
|
|
||||||
W8oFcEfFGpvwVqWZ+AsQ70dsjXAv3zXaG0tmg9FtqWp7pzRSMPidifZcQwWkKeTO
|
|
||||||
gJnFmnVyed8h6GfjTEu4gxo1/S5U0V+mYSha01z5NTnN6ltKx1Or3b0CgYEAxRgm
|
|
||||||
S30nZxnyg/V7ys61AZhst1DG2tkZXEMcA7dYhabMoXPJAP/EfhlWwpWYYUs/u0gS
|
|
||||||
Wwmf5IivX5TlYScgmkvb/NYz0u4ZmOXkLTnLPtdKKFXhjXJcHjUP67jYmOxNlJLp
|
|
||||||
V4vLRnFxTpffAV+OszzRxsXX6fvruwZBANYJeXUCgYBVouLFsFgfWGYp2rpr9XP4
|
|
||||||
KK25kvrBqF6JKOIDB1zjxNJ3pUMKrl8oqccCFoCyXa4oTM2kUX0yWxHfleUjrMq4
|
|
||||||
yimwQKiOZmV7fVLSSjSw6e/VfBd0h3gb82ygcplZkN0IclkwTY5SNKqwn/3y07V5
|
|
||||||
drqdhkrgdJXtmQ6O5YYECQKBgATERcDToQ1USlI4sKrB/wyv1AlG8dg/IebiVJ4e
|
|
||||||
ZAyvcQmClFzq0qS+FiQUnB/WQw9TeeYrwGs1hxBHuJh16srwhLyDrbMvQP06qh8R
|
|
||||||
48F8UXXSRec22dV9MQphaROhu2qZdv1AC0WD3tqov6L33aqmEOi+xi8JgbT/PLk5
|
|
||||||
c/c1AoGBAI1A/02ryksW6/wc7/6SP2M2rTy4m1sD/GnrTc67EHnRcVBdKO6qH2RY
|
|
||||||
nqC8YcveC2ZghgPTDsA3VGuzuBXpwY6wTyV99q6jxQJ6/xcrD9/NUG6Uwv/xfCxl
|
|
||||||
IJLeBYEqQundSSny3VtaAUK8Ul1nxpTvVRNwtcyWTo8RHAAyNPWd
|
|
||||||
-----END RSA PRIVATE KEY-----
|
|
97
vendor/golang.org/x/net/http2/h2i/README.md
generated
vendored
97
vendor/golang.org/x/net/http2/h2i/README.md
generated
vendored
@ -1,97 +0,0 @@
|
|||||||
# h2i
|
|
||||||
|
|
||||||
**h2i** is an interactive HTTP/2 ("h2") console debugger. Miss the good ol'
|
|
||||||
days of telnetting to your HTTP/1.n servers? We're bringing you
|
|
||||||
back.
|
|
||||||
|
|
||||||
Features:
|
|
||||||
- send raw HTTP/2 frames
|
|
||||||
- PING
|
|
||||||
- SETTINGS
|
|
||||||
- HEADERS
|
|
||||||
- etc
|
|
||||||
- type in HTTP/1.n and have it auto-HPACK/frame-ify it for HTTP/2
|
|
||||||
- pretty print all received HTTP/2 frames from the peer (including HPACK decoding)
|
|
||||||
- tab completion of commands, options
|
|
||||||
|
|
||||||
Not yet features, but soon:
|
|
||||||
- unnecessary CONTINUATION frames on short boundaries, to test peer implementations
|
|
||||||
- request bodies (DATA frames)
|
|
||||||
- send invalid frames for testing server implementations (supported by underlying Framer)
|
|
||||||
|
|
||||||
Later:
|
|
||||||
- act like a server
|
|
||||||
|
|
||||||
## Installation
|
|
||||||
|
|
||||||
```
|
|
||||||
$ go get golang.org/x/net/http2/h2i
|
|
||||||
$ h2i <host>
|
|
||||||
```
|
|
||||||
|
|
||||||
## Demo
|
|
||||||
|
|
||||||
```
|
|
||||||
$ h2i
|
|
||||||
Usage: h2i <hostname>
|
|
||||||
|
|
||||||
-insecure
|
|
||||||
Whether to skip TLS cert validation
|
|
||||||
-nextproto string
|
|
||||||
Comma-separated list of NPN/ALPN protocol names to negotiate. (default "h2,h2-14")
|
|
||||||
|
|
||||||
$ h2i google.com
|
|
||||||
Connecting to google.com:443 ...
|
|
||||||
Connected to 74.125.224.41:443
|
|
||||||
Negotiated protocol "h2-14"
|
|
||||||
[FrameHeader SETTINGS len=18]
|
|
||||||
[MAX_CONCURRENT_STREAMS = 100]
|
|
||||||
[INITIAL_WINDOW_SIZE = 1048576]
|
|
||||||
[MAX_FRAME_SIZE = 16384]
|
|
||||||
[FrameHeader WINDOW_UPDATE len=4]
|
|
||||||
Window-Increment = 983041
|
|
||||||
|
|
||||||
h2i> PING h2iSayHI
|
|
||||||
[FrameHeader PING flags=ACK len=8]
|
|
||||||
Data = "h2iSayHI"
|
|
||||||
h2i> headers
|
|
||||||
(as HTTP/1.1)> GET / HTTP/1.1
|
|
||||||
(as HTTP/1.1)> Host: ip.appspot.com
|
|
||||||
(as HTTP/1.1)> User-Agent: h2i/brad-n-blake
|
|
||||||
(as HTTP/1.1)>
|
|
||||||
Opening Stream-ID 1:
|
|
||||||
:authority = ip.appspot.com
|
|
||||||
:method = GET
|
|
||||||
:path = /
|
|
||||||
:scheme = https
|
|
||||||
user-agent = h2i/brad-n-blake
|
|
||||||
[FrameHeader HEADERS flags=END_HEADERS stream=1 len=77]
|
|
||||||
:status = "200"
|
|
||||||
alternate-protocol = "443:quic,p=1"
|
|
||||||
content-length = "15"
|
|
||||||
content-type = "text/html"
|
|
||||||
date = "Fri, 01 May 2015 23:06:56 GMT"
|
|
||||||
server = "Google Frontend"
|
|
||||||
[FrameHeader DATA flags=END_STREAM stream=1 len=15]
|
|
||||||
"173.164.155.78\n"
|
|
||||||
[FrameHeader PING len=8]
|
|
||||||
Data = "\x00\x00\x00\x00\x00\x00\x00\x00"
|
|
||||||
h2i> ping
|
|
||||||
[FrameHeader PING flags=ACK len=8]
|
|
||||||
Data = "h2i_ping"
|
|
||||||
h2i> ping
|
|
||||||
[FrameHeader PING flags=ACK len=8]
|
|
||||||
Data = "h2i_ping"
|
|
||||||
h2i> ping
|
|
||||||
[FrameHeader GOAWAY len=22]
|
|
||||||
Last-Stream-ID = 1; Error-Code = PROTOCOL_ERROR (1)
|
|
||||||
|
|
||||||
ReadFrame: EOF
|
|
||||||
```
|
|
||||||
|
|
||||||
## Status
|
|
||||||
|
|
||||||
Quick few hour hack. So much yet to do. Feel free to file issues for
|
|
||||||
bugs or wishlist items, but [@bmizerany](https://github.com/bmizerany/)
|
|
||||||
and I aren't yet accepting pull requests until things settle down.
|
|
||||||
|
|
501
vendor/golang.org/x/net/http2/h2i/h2i.go
generated
vendored
501
vendor/golang.org/x/net/http2/h2i/h2i.go
generated
vendored
@ -1,501 +0,0 @@
|
|||||||
// Copyright 2015 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.
|
|
||||||
|
|
||||||
// +build !plan9,!solaris
|
|
||||||
|
|
||||||
/*
|
|
||||||
The h2i command is an interactive HTTP/2 console.
|
|
||||||
|
|
||||||
Usage:
|
|
||||||
$ h2i [flags] <hostname>
|
|
||||||
|
|
||||||
Interactive commands in the console: (all parts case-insensitive)
|
|
||||||
|
|
||||||
ping [data]
|
|
||||||
settings ack
|
|
||||||
settings FOO=n BAR=z
|
|
||||||
headers (open a new stream by typing HTTP/1.1)
|
|
||||||
*/
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bufio"
|
|
||||||
"bytes"
|
|
||||||
"crypto/tls"
|
|
||||||
"errors"
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"log"
|
|
||||||
"net"
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"regexp"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"golang.org/x/crypto/ssh/terminal"
|
|
||||||
"golang.org/x/net/http2"
|
|
||||||
"golang.org/x/net/http2/hpack"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Flags
|
|
||||||
var (
|
|
||||||
flagNextProto = flag.String("nextproto", "h2,h2-14", "Comma-separated list of NPN/ALPN protocol names to negotiate.")
|
|
||||||
flagInsecure = flag.Bool("insecure", false, "Whether to skip TLS cert validation")
|
|
||||||
flagSettings = flag.String("settings", "empty", "comma-separated list of KEY=value settings for the initial SETTINGS frame. The magic value 'empty' sends an empty initial settings frame, and the magic value 'omit' causes no initial settings frame to be sent.")
|
|
||||||
)
|
|
||||||
|
|
||||||
type command struct {
|
|
||||||
run func(*h2i, []string) error // required
|
|
||||||
|
|
||||||
// complete optionally specifies tokens (case-insensitive) which are
|
|
||||||
// valid for this subcommand.
|
|
||||||
complete func() []string
|
|
||||||
}
|
|
||||||
|
|
||||||
var commands = map[string]command{
|
|
||||||
"ping": command{run: (*h2i).cmdPing},
|
|
||||||
"settings": command{
|
|
||||||
run: (*h2i).cmdSettings,
|
|
||||||
complete: func() []string {
|
|
||||||
return []string{
|
|
||||||
"ACK",
|
|
||||||
http2.SettingHeaderTableSize.String(),
|
|
||||||
http2.SettingEnablePush.String(),
|
|
||||||
http2.SettingMaxConcurrentStreams.String(),
|
|
||||||
http2.SettingInitialWindowSize.String(),
|
|
||||||
http2.SettingMaxFrameSize.String(),
|
|
||||||
http2.SettingMaxHeaderListSize.String(),
|
|
||||||
}
|
|
||||||
},
|
|
||||||
},
|
|
||||||
"quit": command{run: (*h2i).cmdQuit},
|
|
||||||
"headers": command{run: (*h2i).cmdHeaders},
|
|
||||||
}
|
|
||||||
|
|
||||||
func usage() {
|
|
||||||
fmt.Fprintf(os.Stderr, "Usage: h2i <hostname>\n\n")
|
|
||||||
flag.PrintDefaults()
|
|
||||||
}
|
|
||||||
|
|
||||||
// withPort adds ":443" if another port isn't already present.
|
|
||||||
func withPort(host string) string {
|
|
||||||
if _, _, err := net.SplitHostPort(host); err != nil {
|
|
||||||
return net.JoinHostPort(host, "443")
|
|
||||||
}
|
|
||||||
return host
|
|
||||||
}
|
|
||||||
|
|
||||||
// h2i is the app's state.
|
|
||||||
type h2i struct {
|
|
||||||
host string
|
|
||||||
tc *tls.Conn
|
|
||||||
framer *http2.Framer
|
|
||||||
term *terminal.Terminal
|
|
||||||
|
|
||||||
// owned by the command loop:
|
|
||||||
streamID uint32
|
|
||||||
hbuf bytes.Buffer
|
|
||||||
henc *hpack.Encoder
|
|
||||||
|
|
||||||
// owned by the readFrames loop:
|
|
||||||
peerSetting map[http2.SettingID]uint32
|
|
||||||
hdec *hpack.Decoder
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
flag.Usage = usage
|
|
||||||
flag.Parse()
|
|
||||||
if flag.NArg() != 1 {
|
|
||||||
usage()
|
|
||||||
os.Exit(2)
|
|
||||||
}
|
|
||||||
log.SetFlags(0)
|
|
||||||
|
|
||||||
host := flag.Arg(0)
|
|
||||||
app := &h2i{
|
|
||||||
host: host,
|
|
||||||
peerSetting: make(map[http2.SettingID]uint32),
|
|
||||||
}
|
|
||||||
app.henc = hpack.NewEncoder(&app.hbuf)
|
|
||||||
|
|
||||||
if err := app.Main(); err != nil {
|
|
||||||
if app.term != nil {
|
|
||||||
app.logf("%v\n", err)
|
|
||||||
} else {
|
|
||||||
fmt.Fprintf(os.Stderr, "%v\n", err)
|
|
||||||
}
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
fmt.Fprintf(os.Stdout, "\n")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (app *h2i) Main() error {
|
|
||||||
cfg := &tls.Config{
|
|
||||||
ServerName: app.host,
|
|
||||||
NextProtos: strings.Split(*flagNextProto, ","),
|
|
||||||
InsecureSkipVerify: *flagInsecure,
|
|
||||||
}
|
|
||||||
|
|
||||||
hostAndPort := withPort(app.host)
|
|
||||||
log.Printf("Connecting to %s ...", hostAndPort)
|
|
||||||
tc, err := tls.Dial("tcp", hostAndPort, cfg)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("Error dialing %s: %v", withPort(app.host), err)
|
|
||||||
}
|
|
||||||
log.Printf("Connected to %v", tc.RemoteAddr())
|
|
||||||
defer tc.Close()
|
|
||||||
|
|
||||||
if err := tc.Handshake(); err != nil {
|
|
||||||
return fmt.Errorf("TLS handshake: %v", err)
|
|
||||||
}
|
|
||||||
if !*flagInsecure {
|
|
||||||
if err := tc.VerifyHostname(app.host); err != nil {
|
|
||||||
return fmt.Errorf("VerifyHostname: %v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
state := tc.ConnectionState()
|
|
||||||
log.Printf("Negotiated protocol %q", state.NegotiatedProtocol)
|
|
||||||
if !state.NegotiatedProtocolIsMutual || state.NegotiatedProtocol == "" {
|
|
||||||
return fmt.Errorf("Could not negotiate protocol mutually")
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := io.WriteString(tc, http2.ClientPreface); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
app.framer = http2.NewFramer(tc, tc)
|
|
||||||
|
|
||||||
oldState, err := terminal.MakeRaw(0)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer terminal.Restore(0, oldState)
|
|
||||||
|
|
||||||
var screen = struct {
|
|
||||||
io.Reader
|
|
||||||
io.Writer
|
|
||||||
}{os.Stdin, os.Stdout}
|
|
||||||
|
|
||||||
app.term = terminal.NewTerminal(screen, "h2i> ")
|
|
||||||
lastWord := regexp.MustCompile(`.+\W(\w+)$`)
|
|
||||||
app.term.AutoCompleteCallback = func(line string, pos int, key rune) (newLine string, newPos int, ok bool) {
|
|
||||||
if key != '\t' {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if pos != len(line) {
|
|
||||||
// TODO: we're being lazy for now, only supporting tab completion at the end.
|
|
||||||
return
|
|
||||||
}
|
|
||||||
// Auto-complete for the command itself.
|
|
||||||
if !strings.Contains(line, " ") {
|
|
||||||
var name string
|
|
||||||
name, _, ok = lookupCommand(line)
|
|
||||||
if !ok {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
return name, len(name), true
|
|
||||||
}
|
|
||||||
_, c, ok := lookupCommand(line[:strings.IndexByte(line, ' ')])
|
|
||||||
if !ok || c.complete == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if strings.HasSuffix(line, " ") {
|
|
||||||
app.logf("%s", strings.Join(c.complete(), " "))
|
|
||||||
return line, pos, true
|
|
||||||
}
|
|
||||||
m := lastWord.FindStringSubmatch(line)
|
|
||||||
if m == nil {
|
|
||||||
return line, len(line), true
|
|
||||||
}
|
|
||||||
soFar := m[1]
|
|
||||||
var match []string
|
|
||||||
for _, cand := range c.complete() {
|
|
||||||
if len(soFar) > len(cand) || !strings.EqualFold(cand[:len(soFar)], soFar) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
match = append(match, cand)
|
|
||||||
}
|
|
||||||
if len(match) == 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if len(match) > 1 {
|
|
||||||
// TODO: auto-complete any common prefix
|
|
||||||
app.logf("%s", strings.Join(match, " "))
|
|
||||||
return line, pos, true
|
|
||||||
}
|
|
||||||
newLine = line[:len(line)-len(soFar)] + match[0]
|
|
||||||
return newLine, len(newLine), true
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
errc := make(chan error, 2)
|
|
||||||
go func() { errc <- app.readFrames() }()
|
|
||||||
go func() { errc <- app.readConsole() }()
|
|
||||||
return <-errc
|
|
||||||
}
|
|
||||||
|
|
||||||
func (app *h2i) logf(format string, args ...interface{}) {
|
|
||||||
fmt.Fprintf(app.term, format+"\n", args...)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (app *h2i) readConsole() error {
|
|
||||||
if s := *flagSettings; s != "omit" {
|
|
||||||
var args []string
|
|
||||||
if s != "empty" {
|
|
||||||
args = strings.Split(s, ",")
|
|
||||||
}
|
|
||||||
_, c, ok := lookupCommand("settings")
|
|
||||||
if !ok {
|
|
||||||
panic("settings command not found")
|
|
||||||
}
|
|
||||||
c.run(app, args)
|
|
||||||
}
|
|
||||||
|
|
||||||
for {
|
|
||||||
line, err := app.term.ReadLine()
|
|
||||||
if err == io.EOF {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("terminal.ReadLine: %v", err)
|
|
||||||
}
|
|
||||||
f := strings.Fields(line)
|
|
||||||
if len(f) == 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
cmd, args := f[0], f[1:]
|
|
||||||
if _, c, ok := lookupCommand(cmd); ok {
|
|
||||||
err = c.run(app, args)
|
|
||||||
} else {
|
|
||||||
app.logf("Unknown command %q", line)
|
|
||||||
}
|
|
||||||
if err == errExitApp {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func lookupCommand(prefix string) (name string, c command, ok bool) {
|
|
||||||
prefix = strings.ToLower(prefix)
|
|
||||||
if c, ok = commands[prefix]; ok {
|
|
||||||
return prefix, c, ok
|
|
||||||
}
|
|
||||||
|
|
||||||
for full, candidate := range commands {
|
|
||||||
if strings.HasPrefix(full, prefix) {
|
|
||||||
if c.run != nil {
|
|
||||||
return "", command{}, false // ambiguous
|
|
||||||
}
|
|
||||||
c = candidate
|
|
||||||
name = full
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return name, c, c.run != nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var errExitApp = errors.New("internal sentinel error value to quit the console reading loop")
|
|
||||||
|
|
||||||
func (a *h2i) cmdQuit(args []string) error {
|
|
||||||
if len(args) > 0 {
|
|
||||||
a.logf("the QUIT command takes no argument")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return errExitApp
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *h2i) cmdSettings(args []string) error {
|
|
||||||
if len(args) == 1 && strings.EqualFold(args[0], "ACK") {
|
|
||||||
return a.framer.WriteSettingsAck()
|
|
||||||
}
|
|
||||||
var settings []http2.Setting
|
|
||||||
for _, arg := range args {
|
|
||||||
if strings.EqualFold(arg, "ACK") {
|
|
||||||
a.logf("Error: ACK must be only argument with the SETTINGS command")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
eq := strings.Index(arg, "=")
|
|
||||||
if eq == -1 {
|
|
||||||
a.logf("Error: invalid argument %q (expected SETTING_NAME=nnnn)", arg)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
sid, ok := settingByName(arg[:eq])
|
|
||||||
if !ok {
|
|
||||||
a.logf("Error: unknown setting name %q", arg[:eq])
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
val, err := strconv.ParseUint(arg[eq+1:], 10, 32)
|
|
||||||
if err != nil {
|
|
||||||
a.logf("Error: invalid argument %q (expected SETTING_NAME=nnnn)", arg)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
settings = append(settings, http2.Setting{
|
|
||||||
ID: sid,
|
|
||||||
Val: uint32(val),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
a.logf("Sending: %v", settings)
|
|
||||||
return a.framer.WriteSettings(settings...)
|
|
||||||
}
|
|
||||||
|
|
||||||
func settingByName(name string) (http2.SettingID, bool) {
|
|
||||||
for _, sid := range [...]http2.SettingID{
|
|
||||||
http2.SettingHeaderTableSize,
|
|
||||||
http2.SettingEnablePush,
|
|
||||||
http2.SettingMaxConcurrentStreams,
|
|
||||||
http2.SettingInitialWindowSize,
|
|
||||||
http2.SettingMaxFrameSize,
|
|
||||||
http2.SettingMaxHeaderListSize,
|
|
||||||
} {
|
|
||||||
if strings.EqualFold(sid.String(), name) {
|
|
||||||
return sid, true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return 0, false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (app *h2i) cmdPing(args []string) error {
|
|
||||||
if len(args) > 1 {
|
|
||||||
app.logf("invalid PING usage: only accepts 0 or 1 args")
|
|
||||||
return nil // nil means don't end the program
|
|
||||||
}
|
|
||||||
var data [8]byte
|
|
||||||
if len(args) == 1 {
|
|
||||||
copy(data[:], args[0])
|
|
||||||
} else {
|
|
||||||
copy(data[:], "h2i_ping")
|
|
||||||
}
|
|
||||||
return app.framer.WritePing(false, data)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (app *h2i) cmdHeaders(args []string) error {
|
|
||||||
if len(args) > 0 {
|
|
||||||
app.logf("Error: HEADERS doesn't yet take arguments.")
|
|
||||||
// TODO: flags for restricting window size, to force CONTINUATION
|
|
||||||
// frames.
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
var h1req bytes.Buffer
|
|
||||||
app.term.SetPrompt("(as HTTP/1.1)> ")
|
|
||||||
defer app.term.SetPrompt("h2i> ")
|
|
||||||
for {
|
|
||||||
line, err := app.term.ReadLine()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
h1req.WriteString(line)
|
|
||||||
h1req.WriteString("\r\n")
|
|
||||||
if line == "" {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
req, err := http.ReadRequest(bufio.NewReader(&h1req))
|
|
||||||
if err != nil {
|
|
||||||
app.logf("Invalid HTTP/1.1 request: %v", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if app.streamID == 0 {
|
|
||||||
app.streamID = 1
|
|
||||||
} else {
|
|
||||||
app.streamID += 2
|
|
||||||
}
|
|
||||||
app.logf("Opening Stream-ID %d:", app.streamID)
|
|
||||||
hbf := app.encodeHeaders(req)
|
|
||||||
if len(hbf) > 16<<10 {
|
|
||||||
app.logf("TODO: h2i doesn't yet write CONTINUATION frames. Copy it from transport.go")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return app.framer.WriteHeaders(http2.HeadersFrameParam{
|
|
||||||
StreamID: app.streamID,
|
|
||||||
BlockFragment: hbf,
|
|
||||||
EndStream: req.Method == "GET" || req.Method == "HEAD", // good enough for now
|
|
||||||
EndHeaders: true, // for now
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (app *h2i) readFrames() error {
|
|
||||||
for {
|
|
||||||
f, err := app.framer.ReadFrame()
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("ReadFrame: %v", err)
|
|
||||||
}
|
|
||||||
app.logf("%v", f)
|
|
||||||
switch f := f.(type) {
|
|
||||||
case *http2.PingFrame:
|
|
||||||
app.logf(" Data = %q", f.Data)
|
|
||||||
case *http2.SettingsFrame:
|
|
||||||
f.ForeachSetting(func(s http2.Setting) error {
|
|
||||||
app.logf(" %v", s)
|
|
||||||
app.peerSetting[s.ID] = s.Val
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
case *http2.WindowUpdateFrame:
|
|
||||||
app.logf(" Window-Increment = %v\n", f.Increment)
|
|
||||||
case *http2.GoAwayFrame:
|
|
||||||
app.logf(" Last-Stream-ID = %d; Error-Code = %v (%d)\n", f.LastStreamID, f.ErrCode, f.ErrCode)
|
|
||||||
case *http2.DataFrame:
|
|
||||||
app.logf(" %q", f.Data())
|
|
||||||
case *http2.HeadersFrame:
|
|
||||||
if f.HasPriority() {
|
|
||||||
app.logf(" PRIORITY = %v", f.Priority)
|
|
||||||
}
|
|
||||||
if app.hdec == nil {
|
|
||||||
// TODO: if the user uses h2i to send a SETTINGS frame advertising
|
|
||||||
// something larger, we'll need to respect SETTINGS_HEADER_TABLE_SIZE
|
|
||||||
// and stuff here instead of using the 4k default. But for now:
|
|
||||||
tableSize := uint32(4 << 10)
|
|
||||||
app.hdec = hpack.NewDecoder(tableSize, app.onNewHeaderField)
|
|
||||||
}
|
|
||||||
app.hdec.Write(f.HeaderBlockFragment())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// called from readLoop
|
|
||||||
func (app *h2i) onNewHeaderField(f hpack.HeaderField) {
|
|
||||||
if f.Sensitive {
|
|
||||||
app.logf(" %s = %q (SENSITIVE)", f.Name, f.Value)
|
|
||||||
}
|
|
||||||
app.logf(" %s = %q", f.Name, f.Value)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (app *h2i) encodeHeaders(req *http.Request) []byte {
|
|
||||||
app.hbuf.Reset()
|
|
||||||
|
|
||||||
// TODO(bradfitz): figure out :authority-vs-Host stuff between http2 and Go
|
|
||||||
host := req.Host
|
|
||||||
if host == "" {
|
|
||||||
host = req.URL.Host
|
|
||||||
}
|
|
||||||
|
|
||||||
path := req.URL.Path
|
|
||||||
if path == "" {
|
|
||||||
path = "/"
|
|
||||||
}
|
|
||||||
|
|
||||||
app.writeHeader(":authority", host) // probably not right for all sites
|
|
||||||
app.writeHeader(":method", req.Method)
|
|
||||||
app.writeHeader(":path", path)
|
|
||||||
app.writeHeader(":scheme", "https")
|
|
||||||
|
|
||||||
for k, vv := range req.Header {
|
|
||||||
lowKey := strings.ToLower(k)
|
|
||||||
if lowKey == "host" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
for _, v := range vv {
|
|
||||||
app.writeHeader(lowKey, v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return app.hbuf.Bytes()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (app *h2i) writeHeader(name, value string) {
|
|
||||||
app.henc.WriteField(hpack.HeaderField{Name: name, Value: value})
|
|
||||||
app.logf(" %s = %s", name, value)
|
|
||||||
}
|
|
78
vendor/golang.org/x/net/http2/headermap.go
generated
vendored
78
vendor/golang.org/x/net/http2/headermap.go
generated
vendored
@ -1,78 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net/http"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
commonLowerHeader = map[string]string{} // Go-Canonical-Case -> lower-case
|
|
||||||
commonCanonHeader = map[string]string{} // lower-case -> Go-Canonical-Case
|
|
||||||
)
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
for _, v := range []string{
|
|
||||||
"accept",
|
|
||||||
"accept-charset",
|
|
||||||
"accept-encoding",
|
|
||||||
"accept-language",
|
|
||||||
"accept-ranges",
|
|
||||||
"age",
|
|
||||||
"access-control-allow-origin",
|
|
||||||
"allow",
|
|
||||||
"authorization",
|
|
||||||
"cache-control",
|
|
||||||
"content-disposition",
|
|
||||||
"content-encoding",
|
|
||||||
"content-language",
|
|
||||||
"content-length",
|
|
||||||
"content-location",
|
|
||||||
"content-range",
|
|
||||||
"content-type",
|
|
||||||
"cookie",
|
|
||||||
"date",
|
|
||||||
"etag",
|
|
||||||
"expect",
|
|
||||||
"expires",
|
|
||||||
"from",
|
|
||||||
"host",
|
|
||||||
"if-match",
|
|
||||||
"if-modified-since",
|
|
||||||
"if-none-match",
|
|
||||||
"if-unmodified-since",
|
|
||||||
"last-modified",
|
|
||||||
"link",
|
|
||||||
"location",
|
|
||||||
"max-forwards",
|
|
||||||
"proxy-authenticate",
|
|
||||||
"proxy-authorization",
|
|
||||||
"range",
|
|
||||||
"referer",
|
|
||||||
"refresh",
|
|
||||||
"retry-after",
|
|
||||||
"server",
|
|
||||||
"set-cookie",
|
|
||||||
"strict-transport-security",
|
|
||||||
"trailer",
|
|
||||||
"transfer-encoding",
|
|
||||||
"user-agent",
|
|
||||||
"vary",
|
|
||||||
"via",
|
|
||||||
"www-authenticate",
|
|
||||||
} {
|
|
||||||
chk := http.CanonicalHeaderKey(v)
|
|
||||||
commonLowerHeader[chk] = v
|
|
||||||
commonCanonHeader[v] = chk
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func lowerHeader(v string) string {
|
|
||||||
if s, ok := commonLowerHeader[v]; ok {
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
return strings.ToLower(v)
|
|
||||||
}
|
|
251
vendor/golang.org/x/net/http2/hpack/encode.go
generated
vendored
251
vendor/golang.org/x/net/http2/hpack/encode.go
generated
vendored
@ -1,251 +0,0 @@
|
|||||||
// Copyright 2014 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 hpack
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
uint32Max = ^uint32(0)
|
|
||||||
initialHeaderTableSize = 4096
|
|
||||||
)
|
|
||||||
|
|
||||||
type Encoder struct {
|
|
||||||
dynTab dynamicTable
|
|
||||||
// minSize is the minimum table size set by
|
|
||||||
// SetMaxDynamicTableSize after the previous Header Table Size
|
|
||||||
// Update.
|
|
||||||
minSize uint32
|
|
||||||
// maxSizeLimit is the maximum table size this encoder
|
|
||||||
// supports. This will protect the encoder from too large
|
|
||||||
// size.
|
|
||||||
maxSizeLimit uint32
|
|
||||||
// tableSizeUpdate indicates whether "Header Table Size
|
|
||||||
// Update" is required.
|
|
||||||
tableSizeUpdate bool
|
|
||||||
w io.Writer
|
|
||||||
buf []byte
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewEncoder returns a new Encoder which performs HPACK encoding. An
|
|
||||||
// encoded data is written to w.
|
|
||||||
func NewEncoder(w io.Writer) *Encoder {
|
|
||||||
e := &Encoder{
|
|
||||||
minSize: uint32Max,
|
|
||||||
maxSizeLimit: initialHeaderTableSize,
|
|
||||||
tableSizeUpdate: false,
|
|
||||||
w: w,
|
|
||||||
}
|
|
||||||
e.dynTab.setMaxSize(initialHeaderTableSize)
|
|
||||||
return e
|
|
||||||
}
|
|
||||||
|
|
||||||
// WriteField encodes f into a single Write to e's underlying Writer.
|
|
||||||
// This function may also produce bytes for "Header Table Size Update"
|
|
||||||
// if necessary. If produced, it is done before encoding f.
|
|
||||||
func (e *Encoder) WriteField(f HeaderField) error {
|
|
||||||
e.buf = e.buf[:0]
|
|
||||||
|
|
||||||
if e.tableSizeUpdate {
|
|
||||||
e.tableSizeUpdate = false
|
|
||||||
if e.minSize < e.dynTab.maxSize {
|
|
||||||
e.buf = appendTableSize(e.buf, e.minSize)
|
|
||||||
}
|
|
||||||
e.minSize = uint32Max
|
|
||||||
e.buf = appendTableSize(e.buf, e.dynTab.maxSize)
|
|
||||||
}
|
|
||||||
|
|
||||||
idx, nameValueMatch := e.searchTable(f)
|
|
||||||
if nameValueMatch {
|
|
||||||
e.buf = appendIndexed(e.buf, idx)
|
|
||||||
} else {
|
|
||||||
indexing := e.shouldIndex(f)
|
|
||||||
if indexing {
|
|
||||||
e.dynTab.add(f)
|
|
||||||
}
|
|
||||||
|
|
||||||
if idx == 0 {
|
|
||||||
e.buf = appendNewName(e.buf, f, indexing)
|
|
||||||
} else {
|
|
||||||
e.buf = appendIndexedName(e.buf, f, idx, indexing)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
n, err := e.w.Write(e.buf)
|
|
||||||
if err == nil && n != len(e.buf) {
|
|
||||||
err = io.ErrShortWrite
|
|
||||||
}
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// searchTable searches f in both stable and dynamic header tables.
|
|
||||||
// The static header table is searched first. Only when there is no
|
|
||||||
// exact match for both name and value, the dynamic header table is
|
|
||||||
// then searched. If there is no match, i is 0. If both name and value
|
|
||||||
// match, i is the matched index and nameValueMatch becomes true. If
|
|
||||||
// only name matches, i points to that index and nameValueMatch
|
|
||||||
// becomes false.
|
|
||||||
func (e *Encoder) searchTable(f HeaderField) (i uint64, nameValueMatch bool) {
|
|
||||||
for idx, hf := range staticTable {
|
|
||||||
if !constantTimeStringCompare(hf.Name, f.Name) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if i == 0 {
|
|
||||||
i = uint64(idx + 1)
|
|
||||||
}
|
|
||||||
if f.Sensitive {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !constantTimeStringCompare(hf.Value, f.Value) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
i = uint64(idx + 1)
|
|
||||||
nameValueMatch = true
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
j, nameValueMatch := e.dynTab.search(f)
|
|
||||||
if nameValueMatch || (i == 0 && j != 0) {
|
|
||||||
i = j + uint64(len(staticTable))
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetMaxDynamicTableSize changes the dynamic header table size to v.
|
|
||||||
// The actual size is bounded by the value passed to
|
|
||||||
// SetMaxDynamicTableSizeLimit.
|
|
||||||
func (e *Encoder) SetMaxDynamicTableSize(v uint32) {
|
|
||||||
if v > e.maxSizeLimit {
|
|
||||||
v = e.maxSizeLimit
|
|
||||||
}
|
|
||||||
if v < e.minSize {
|
|
||||||
e.minSize = v
|
|
||||||
}
|
|
||||||
e.tableSizeUpdate = true
|
|
||||||
e.dynTab.setMaxSize(v)
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetMaxDynamicTableSizeLimit changes the maximum value that can be
|
|
||||||
// specified in SetMaxDynamicTableSize to v. By default, it is set to
|
|
||||||
// 4096, which is the same size of the default dynamic header table
|
|
||||||
// size described in HPACK specification. If the current maximum
|
|
||||||
// dynamic header table size is strictly greater than v, "Header Table
|
|
||||||
// Size Update" will be done in the next WriteField call and the
|
|
||||||
// maximum dynamic header table size is truncated to v.
|
|
||||||
func (e *Encoder) SetMaxDynamicTableSizeLimit(v uint32) {
|
|
||||||
e.maxSizeLimit = v
|
|
||||||
if e.dynTab.maxSize > v {
|
|
||||||
e.tableSizeUpdate = true
|
|
||||||
e.dynTab.setMaxSize(v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// shouldIndex reports whether f should be indexed.
|
|
||||||
func (e *Encoder) shouldIndex(f HeaderField) bool {
|
|
||||||
return !f.Sensitive && f.Size() <= e.dynTab.maxSize
|
|
||||||
}
|
|
||||||
|
|
||||||
// appendIndexed appends index i, as encoded in "Indexed Header Field"
|
|
||||||
// representation, to dst and returns the extended buffer.
|
|
||||||
func appendIndexed(dst []byte, i uint64) []byte {
|
|
||||||
first := len(dst)
|
|
||||||
dst = appendVarInt(dst, 7, i)
|
|
||||||
dst[first] |= 0x80
|
|
||||||
return dst
|
|
||||||
}
|
|
||||||
|
|
||||||
// appendNewName appends f, as encoded in one of "Literal Header field
|
|
||||||
// - New Name" representation variants, to dst and returns the
|
|
||||||
// extended buffer.
|
|
||||||
//
|
|
||||||
// If f.Sensitive is true, "Never Indexed" representation is used. If
|
|
||||||
// f.Sensitive is false and indexing is true, "Inremental Indexing"
|
|
||||||
// representation is used.
|
|
||||||
func appendNewName(dst []byte, f HeaderField, indexing bool) []byte {
|
|
||||||
dst = append(dst, encodeTypeByte(indexing, f.Sensitive))
|
|
||||||
dst = appendHpackString(dst, f.Name)
|
|
||||||
return appendHpackString(dst, f.Value)
|
|
||||||
}
|
|
||||||
|
|
||||||
// appendIndexedName appends f and index i referring indexed name
|
|
||||||
// entry, as encoded in one of "Literal Header field - Indexed Name"
|
|
||||||
// representation variants, to dst and returns the extended buffer.
|
|
||||||
//
|
|
||||||
// If f.Sensitive is true, "Never Indexed" representation is used. If
|
|
||||||
// f.Sensitive is false and indexing is true, "Incremental Indexing"
|
|
||||||
// representation is used.
|
|
||||||
func appendIndexedName(dst []byte, f HeaderField, i uint64, indexing bool) []byte {
|
|
||||||
first := len(dst)
|
|
||||||
var n byte
|
|
||||||
if indexing {
|
|
||||||
n = 6
|
|
||||||
} else {
|
|
||||||
n = 4
|
|
||||||
}
|
|
||||||
dst = appendVarInt(dst, n, i)
|
|
||||||
dst[first] |= encodeTypeByte(indexing, f.Sensitive)
|
|
||||||
return appendHpackString(dst, f.Value)
|
|
||||||
}
|
|
||||||
|
|
||||||
// appendTableSize appends v, as encoded in "Header Table Size Update"
|
|
||||||
// representation, to dst and returns the extended buffer.
|
|
||||||
func appendTableSize(dst []byte, v uint32) []byte {
|
|
||||||
first := len(dst)
|
|
||||||
dst = appendVarInt(dst, 5, uint64(v))
|
|
||||||
dst[first] |= 0x20
|
|
||||||
return dst
|
|
||||||
}
|
|
||||||
|
|
||||||
// appendVarInt appends i, as encoded in variable integer form using n
|
|
||||||
// bit prefix, to dst and returns the extended buffer.
|
|
||||||
//
|
|
||||||
// See
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#integer.representation
|
|
||||||
func appendVarInt(dst []byte, n byte, i uint64) []byte {
|
|
||||||
k := uint64((1 << n) - 1)
|
|
||||||
if i < k {
|
|
||||||
return append(dst, byte(i))
|
|
||||||
}
|
|
||||||
dst = append(dst, byte(k))
|
|
||||||
i -= k
|
|
||||||
for ; i >= 128; i >>= 7 {
|
|
||||||
dst = append(dst, byte(0x80|(i&0x7f)))
|
|
||||||
}
|
|
||||||
return append(dst, byte(i))
|
|
||||||
}
|
|
||||||
|
|
||||||
// appendHpackString appends s, as encoded in "String Literal"
|
|
||||||
// representation, to dst and returns the the extended buffer.
|
|
||||||
//
|
|
||||||
// s will be encoded in Huffman codes only when it produces strictly
|
|
||||||
// shorter byte string.
|
|
||||||
func appendHpackString(dst []byte, s string) []byte {
|
|
||||||
huffmanLength := HuffmanEncodeLength(s)
|
|
||||||
if huffmanLength < uint64(len(s)) {
|
|
||||||
first := len(dst)
|
|
||||||
dst = appendVarInt(dst, 7, huffmanLength)
|
|
||||||
dst = AppendHuffmanString(dst, s)
|
|
||||||
dst[first] |= 0x80
|
|
||||||
} else {
|
|
||||||
dst = appendVarInt(dst, 7, uint64(len(s)))
|
|
||||||
dst = append(dst, s...)
|
|
||||||
}
|
|
||||||
return dst
|
|
||||||
}
|
|
||||||
|
|
||||||
// encodeTypeByte returns type byte. If sensitive is true, type byte
|
|
||||||
// for "Never Indexed" representation is returned. If sensitive is
|
|
||||||
// false and indexing is true, type byte for "Incremental Indexing"
|
|
||||||
// representation is returned. Otherwise, type byte for "Without
|
|
||||||
// Indexing" is returned.
|
|
||||||
func encodeTypeByte(indexing, sensitive bool) byte {
|
|
||||||
if sensitive {
|
|
||||||
return 0x10
|
|
||||||
}
|
|
||||||
if indexing {
|
|
||||||
return 0x40
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
330
vendor/golang.org/x/net/http2/hpack/encode_test.go
generated
vendored
330
vendor/golang.org/x/net/http2/hpack/encode_test.go
generated
vendored
@ -1,330 +0,0 @@
|
|||||||
// Copyright 2014 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 hpack
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"encoding/hex"
|
|
||||||
"reflect"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestEncoderTableSizeUpdate(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
size1, size2 uint32
|
|
||||||
wantHex string
|
|
||||||
}{
|
|
||||||
// Should emit 2 table size updates (2048 and 4096)
|
|
||||||
{2048, 4096, "3fe10f 3fe11f 82"},
|
|
||||||
|
|
||||||
// Should emit 1 table size update (2048)
|
|
||||||
{16384, 2048, "3fe10f 82"},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
var buf bytes.Buffer
|
|
||||||
e := NewEncoder(&buf)
|
|
||||||
e.SetMaxDynamicTableSize(tt.size1)
|
|
||||||
e.SetMaxDynamicTableSize(tt.size2)
|
|
||||||
if err := e.WriteField(pair(":method", "GET")); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
want := removeSpace(tt.wantHex)
|
|
||||||
if got := hex.EncodeToString(buf.Bytes()); got != want {
|
|
||||||
t.Errorf("e.SetDynamicTableSize %v, %v = %q; want %q", tt.size1, tt.size2, got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestEncoderWriteField(t *testing.T) {
|
|
||||||
var buf bytes.Buffer
|
|
||||||
e := NewEncoder(&buf)
|
|
||||||
var got []HeaderField
|
|
||||||
d := NewDecoder(4<<10, func(f HeaderField) {
|
|
||||||
got = append(got, f)
|
|
||||||
})
|
|
||||||
|
|
||||||
tests := []struct {
|
|
||||||
hdrs []HeaderField
|
|
||||||
}{
|
|
||||||
{[]HeaderField{
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":scheme", "http"),
|
|
||||||
pair(":path", "/"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
}},
|
|
||||||
{[]HeaderField{
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":scheme", "http"),
|
|
||||||
pair(":path", "/"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
pair("cache-control", "no-cache"),
|
|
||||||
}},
|
|
||||||
{[]HeaderField{
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":scheme", "https"),
|
|
||||||
pair(":path", "/index.html"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
pair("custom-key", "custom-value"),
|
|
||||||
}},
|
|
||||||
}
|
|
||||||
for i, tt := range tests {
|
|
||||||
buf.Reset()
|
|
||||||
got = got[:0]
|
|
||||||
for _, hf := range tt.hdrs {
|
|
||||||
if err := e.WriteField(hf); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_, err := d.Write(buf.Bytes())
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("%d. Decoder Write = %v", i, err)
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(got, tt.hdrs) {
|
|
||||||
t.Errorf("%d. Decoded %+v; want %+v", i, got, tt.hdrs)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestEncoderSearchTable(t *testing.T) {
|
|
||||||
e := NewEncoder(nil)
|
|
||||||
|
|
||||||
e.dynTab.add(pair("foo", "bar"))
|
|
||||||
e.dynTab.add(pair("blake", "miz"))
|
|
||||||
e.dynTab.add(pair(":method", "GET"))
|
|
||||||
|
|
||||||
tests := []struct {
|
|
||||||
hf HeaderField
|
|
||||||
wantI uint64
|
|
||||||
wantMatch bool
|
|
||||||
}{
|
|
||||||
// Name and Value match
|
|
||||||
{pair("foo", "bar"), uint64(len(staticTable) + 3), true},
|
|
||||||
{pair("blake", "miz"), uint64(len(staticTable) + 2), true},
|
|
||||||
{pair(":method", "GET"), 2, true},
|
|
||||||
|
|
||||||
// Only name match because Sensitive == true
|
|
||||||
{HeaderField{":method", "GET", true}, 2, false},
|
|
||||||
|
|
||||||
// Only Name matches
|
|
||||||
{pair("foo", "..."), uint64(len(staticTable) + 3), false},
|
|
||||||
{pair("blake", "..."), uint64(len(staticTable) + 2), false},
|
|
||||||
{pair(":method", "..."), 2, false},
|
|
||||||
|
|
||||||
// None match
|
|
||||||
{pair("foo-", "bar"), 0, false},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
if gotI, gotMatch := e.searchTable(tt.hf); gotI != tt.wantI || gotMatch != tt.wantMatch {
|
|
||||||
t.Errorf("d.search(%+v) = %v, %v; want %v, %v", tt.hf, gotI, gotMatch, tt.wantI, tt.wantMatch)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestAppendVarInt(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
n byte
|
|
||||||
i uint64
|
|
||||||
want []byte
|
|
||||||
}{
|
|
||||||
// Fits in a byte:
|
|
||||||
{1, 0, []byte{0}},
|
|
||||||
{2, 2, []byte{2}},
|
|
||||||
{3, 6, []byte{6}},
|
|
||||||
{4, 14, []byte{14}},
|
|
||||||
{5, 30, []byte{30}},
|
|
||||||
{6, 62, []byte{62}},
|
|
||||||
{7, 126, []byte{126}},
|
|
||||||
{8, 254, []byte{254}},
|
|
||||||
|
|
||||||
// Multiple bytes:
|
|
||||||
{5, 1337, []byte{31, 154, 10}},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
got := appendVarInt(nil, tt.n, tt.i)
|
|
||||||
if !bytes.Equal(got, tt.want) {
|
|
||||||
t.Errorf("appendVarInt(nil, %v, %v) = %v; want %v", tt.n, tt.i, got, tt.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestAppendHpackString(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
s, wantHex string
|
|
||||||
}{
|
|
||||||
// Huffman encoded
|
|
||||||
{"www.example.com", "8c f1e3 c2e5 f23a 6ba0 ab90 f4ff"},
|
|
||||||
|
|
||||||
// Not Huffman encoded
|
|
||||||
{"a", "01 61"},
|
|
||||||
|
|
||||||
// zero length
|
|
||||||
{"", "00"},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
want := removeSpace(tt.wantHex)
|
|
||||||
buf := appendHpackString(nil, tt.s)
|
|
||||||
if got := hex.EncodeToString(buf); want != got {
|
|
||||||
t.Errorf("appendHpackString(nil, %q) = %q; want %q", tt.s, got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestAppendIndexed(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
i uint64
|
|
||||||
wantHex string
|
|
||||||
}{
|
|
||||||
// 1 byte
|
|
||||||
{1, "81"},
|
|
||||||
{126, "fe"},
|
|
||||||
|
|
||||||
// 2 bytes
|
|
||||||
{127, "ff00"},
|
|
||||||
{128, "ff01"},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
want := removeSpace(tt.wantHex)
|
|
||||||
buf := appendIndexed(nil, tt.i)
|
|
||||||
if got := hex.EncodeToString(buf); want != got {
|
|
||||||
t.Errorf("appendIndex(nil, %v) = %q; want %q", tt.i, got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestAppendNewName(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
f HeaderField
|
|
||||||
indexing bool
|
|
||||||
wantHex string
|
|
||||||
}{
|
|
||||||
// Incremental indexing
|
|
||||||
{HeaderField{"custom-key", "custom-value", false}, true, "40 88 25a8 49e9 5ba9 7d7f 89 25a8 49e9 5bb8 e8b4 bf"},
|
|
||||||
|
|
||||||
// Without indexing
|
|
||||||
{HeaderField{"custom-key", "custom-value", false}, false, "00 88 25a8 49e9 5ba9 7d7f 89 25a8 49e9 5bb8 e8b4 bf"},
|
|
||||||
|
|
||||||
// Never indexed
|
|
||||||
{HeaderField{"custom-key", "custom-value", true}, true, "10 88 25a8 49e9 5ba9 7d7f 89 25a8 49e9 5bb8 e8b4 bf"},
|
|
||||||
{HeaderField{"custom-key", "custom-value", true}, false, "10 88 25a8 49e9 5ba9 7d7f 89 25a8 49e9 5bb8 e8b4 bf"},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
want := removeSpace(tt.wantHex)
|
|
||||||
buf := appendNewName(nil, tt.f, tt.indexing)
|
|
||||||
if got := hex.EncodeToString(buf); want != got {
|
|
||||||
t.Errorf("appendNewName(nil, %+v, %v) = %q; want %q", tt.f, tt.indexing, got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestAppendIndexedName(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
f HeaderField
|
|
||||||
i uint64
|
|
||||||
indexing bool
|
|
||||||
wantHex string
|
|
||||||
}{
|
|
||||||
// Incremental indexing
|
|
||||||
{HeaderField{":status", "302", false}, 8, true, "48 82 6402"},
|
|
||||||
|
|
||||||
// Without indexing
|
|
||||||
{HeaderField{":status", "302", false}, 8, false, "08 82 6402"},
|
|
||||||
|
|
||||||
// Never indexed
|
|
||||||
{HeaderField{":status", "302", true}, 8, true, "18 82 6402"},
|
|
||||||
{HeaderField{":status", "302", true}, 8, false, "18 82 6402"},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
want := removeSpace(tt.wantHex)
|
|
||||||
buf := appendIndexedName(nil, tt.f, tt.i, tt.indexing)
|
|
||||||
if got := hex.EncodeToString(buf); want != got {
|
|
||||||
t.Errorf("appendIndexedName(nil, %+v, %v) = %q; want %q", tt.f, tt.indexing, got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestAppendTableSize(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
i uint32
|
|
||||||
wantHex string
|
|
||||||
}{
|
|
||||||
// Fits into 1 byte
|
|
||||||
{30, "3e"},
|
|
||||||
|
|
||||||
// Extra byte
|
|
||||||
{31, "3f00"},
|
|
||||||
{32, "3f01"},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
want := removeSpace(tt.wantHex)
|
|
||||||
buf := appendTableSize(nil, tt.i)
|
|
||||||
if got := hex.EncodeToString(buf); want != got {
|
|
||||||
t.Errorf("appendTableSize(nil, %v) = %q; want %q", tt.i, got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestEncoderSetMaxDynamicTableSize(t *testing.T) {
|
|
||||||
var buf bytes.Buffer
|
|
||||||
e := NewEncoder(&buf)
|
|
||||||
tests := []struct {
|
|
||||||
v uint32
|
|
||||||
wantUpdate bool
|
|
||||||
wantMinSize uint32
|
|
||||||
wantMaxSize uint32
|
|
||||||
}{
|
|
||||||
// Set new table size to 2048
|
|
||||||
{2048, true, 2048, 2048},
|
|
||||||
|
|
||||||
// Set new table size to 16384, but still limited to
|
|
||||||
// 4096
|
|
||||||
{16384, true, 2048, 4096},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
e.SetMaxDynamicTableSize(tt.v)
|
|
||||||
if got := e.tableSizeUpdate; tt.wantUpdate != got {
|
|
||||||
t.Errorf("e.tableSizeUpdate = %v; want %v", got, tt.wantUpdate)
|
|
||||||
}
|
|
||||||
if got := e.minSize; tt.wantMinSize != got {
|
|
||||||
t.Errorf("e.minSize = %v; want %v", got, tt.wantMinSize)
|
|
||||||
}
|
|
||||||
if got := e.dynTab.maxSize; tt.wantMaxSize != got {
|
|
||||||
t.Errorf("e.maxSize = %v; want %v", got, tt.wantMaxSize)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestEncoderSetMaxDynamicTableSizeLimit(t *testing.T) {
|
|
||||||
e := NewEncoder(nil)
|
|
||||||
// 4095 < initialHeaderTableSize means maxSize is truncated to
|
|
||||||
// 4095.
|
|
||||||
e.SetMaxDynamicTableSizeLimit(4095)
|
|
||||||
if got, want := e.dynTab.maxSize, uint32(4095); got != want {
|
|
||||||
t.Errorf("e.dynTab.maxSize = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
if got, want := e.maxSizeLimit, uint32(4095); got != want {
|
|
||||||
t.Errorf("e.maxSizeLimit = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
if got, want := e.tableSizeUpdate, true; got != want {
|
|
||||||
t.Errorf("e.tableSizeUpdate = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
// maxSize will be truncated to maxSizeLimit
|
|
||||||
e.SetMaxDynamicTableSize(16384)
|
|
||||||
if got, want := e.dynTab.maxSize, uint32(4095); got != want {
|
|
||||||
t.Errorf("e.dynTab.maxSize = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
// 8192 > current maxSizeLimit, so maxSize does not change.
|
|
||||||
e.SetMaxDynamicTableSizeLimit(8192)
|
|
||||||
if got, want := e.dynTab.maxSize, uint32(4095); got != want {
|
|
||||||
t.Errorf("e.dynTab.maxSize = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
if got, want := e.maxSizeLimit, uint32(8192); got != want {
|
|
||||||
t.Errorf("e.maxSizeLimit = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func removeSpace(s string) string {
|
|
||||||
return strings.Replace(s, " ", "", -1)
|
|
||||||
}
|
|
542
vendor/golang.org/x/net/http2/hpack/hpack.go
generated
vendored
542
vendor/golang.org/x/net/http2/hpack/hpack.go
generated
vendored
@ -1,542 +0,0 @@
|
|||||||
// Copyright 2014 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 hpack implements HPACK, a compression format for
|
|
||||||
// efficiently representing HTTP header fields in the context of HTTP/2.
|
|
||||||
//
|
|
||||||
// See http://tools.ietf.org/html/draft-ietf-httpbis-header-compression-09
|
|
||||||
package hpack
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
)
|
|
||||||
|
|
||||||
// A DecodingError is something the spec defines as a decoding error.
|
|
||||||
type DecodingError struct {
|
|
||||||
Err error
|
|
||||||
}
|
|
||||||
|
|
||||||
func (de DecodingError) Error() string {
|
|
||||||
return fmt.Sprintf("decoding error: %v", de.Err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// An InvalidIndexError is returned when an encoder references a table
|
|
||||||
// entry before the static table or after the end of the dynamic table.
|
|
||||||
type InvalidIndexError int
|
|
||||||
|
|
||||||
func (e InvalidIndexError) Error() string {
|
|
||||||
return fmt.Sprintf("invalid indexed representation index %d", int(e))
|
|
||||||
}
|
|
||||||
|
|
||||||
// A HeaderField is a name-value pair. Both the name and value are
|
|
||||||
// treated as opaque sequences of octets.
|
|
||||||
type HeaderField struct {
|
|
||||||
Name, Value string
|
|
||||||
|
|
||||||
// Sensitive means that this header field should never be
|
|
||||||
// indexed.
|
|
||||||
Sensitive bool
|
|
||||||
}
|
|
||||||
|
|
||||||
// IsPseudo reports whether the header field is an http2 pseudo header.
|
|
||||||
// That is, it reports whether it starts with a colon.
|
|
||||||
// It is not otherwise guaranteed to be a valid psuedo header field,
|
|
||||||
// though.
|
|
||||||
func (hf HeaderField) IsPseudo() bool {
|
|
||||||
return len(hf.Name) != 0 && hf.Name[0] == ':'
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hf HeaderField) String() string {
|
|
||||||
var suffix string
|
|
||||||
if hf.Sensitive {
|
|
||||||
suffix = " (sensitive)"
|
|
||||||
}
|
|
||||||
return fmt.Sprintf("header field %q = %q%s", hf.Name, hf.Value, suffix)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Size returns the size of an entry per RFC 7540 section 5.2.
|
|
||||||
func (hf HeaderField) Size() uint32 {
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.4.1
|
|
||||||
// "The size of the dynamic table is the sum of the size of
|
|
||||||
// its entries. The size of an entry is the sum of its name's
|
|
||||||
// length in octets (as defined in Section 5.2), its value's
|
|
||||||
// length in octets (see Section 5.2), plus 32. The size of
|
|
||||||
// an entry is calculated using the length of the name and
|
|
||||||
// value without any Huffman encoding applied."
|
|
||||||
|
|
||||||
// This can overflow if somebody makes a large HeaderField
|
|
||||||
// Name and/or Value by hand, but we don't care, because that
|
|
||||||
// won't happen on the wire because the encoding doesn't allow
|
|
||||||
// it.
|
|
||||||
return uint32(len(hf.Name) + len(hf.Value) + 32)
|
|
||||||
}
|
|
||||||
|
|
||||||
// A Decoder is the decoding context for incremental processing of
|
|
||||||
// header blocks.
|
|
||||||
type Decoder struct {
|
|
||||||
dynTab dynamicTable
|
|
||||||
emit func(f HeaderField)
|
|
||||||
|
|
||||||
emitEnabled bool // whether calls to emit are enabled
|
|
||||||
maxStrLen int // 0 means unlimited
|
|
||||||
|
|
||||||
// buf is the unparsed buffer. It's only written to
|
|
||||||
// saveBuf if it was truncated in the middle of a header
|
|
||||||
// block. Because it's usually not owned, we can only
|
|
||||||
// process it under Write.
|
|
||||||
buf []byte // not owned; only valid during Write
|
|
||||||
|
|
||||||
// saveBuf is previous data passed to Write which we weren't able
|
|
||||||
// to fully parse before. Unlike buf, we own this data.
|
|
||||||
saveBuf bytes.Buffer
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewDecoder returns a new decoder with the provided maximum dynamic
|
|
||||||
// table size. The emitFunc will be called for each valid field
|
|
||||||
// parsed, in the same goroutine as calls to Write, before Write returns.
|
|
||||||
func NewDecoder(maxDynamicTableSize uint32, emitFunc func(f HeaderField)) *Decoder {
|
|
||||||
d := &Decoder{
|
|
||||||
emit: emitFunc,
|
|
||||||
emitEnabled: true,
|
|
||||||
}
|
|
||||||
d.dynTab.allowedMaxSize = maxDynamicTableSize
|
|
||||||
d.dynTab.setMaxSize(maxDynamicTableSize)
|
|
||||||
return d
|
|
||||||
}
|
|
||||||
|
|
||||||
// ErrStringLength is returned by Decoder.Write when the max string length
|
|
||||||
// (as configured by Decoder.SetMaxStringLength) would be violated.
|
|
||||||
var ErrStringLength = errors.New("hpack: string too long")
|
|
||||||
|
|
||||||
// SetMaxStringLength sets the maximum size of a HeaderField name or
|
|
||||||
// value string. If a string exceeds this length (even after any
|
|
||||||
// decompression), Write will return ErrStringLength.
|
|
||||||
// A value of 0 means unlimited and is the default from NewDecoder.
|
|
||||||
func (d *Decoder) SetMaxStringLength(n int) {
|
|
||||||
d.maxStrLen = n
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetEmitFunc changes the callback used when new header fields
|
|
||||||
// are decoded.
|
|
||||||
// It must be non-nil. It does not affect EmitEnabled.
|
|
||||||
func (d *Decoder) SetEmitFunc(emitFunc func(f HeaderField)) {
|
|
||||||
d.emit = emitFunc
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetEmitEnabled controls whether the emitFunc provided to NewDecoder
|
|
||||||
// should be called. The default is true.
|
|
||||||
//
|
|
||||||
// This facility exists to let servers enforce MAX_HEADER_LIST_SIZE
|
|
||||||
// while still decoding and keeping in-sync with decoder state, but
|
|
||||||
// without doing unnecessary decompression or generating unnecessary
|
|
||||||
// garbage for header fields past the limit.
|
|
||||||
func (d *Decoder) SetEmitEnabled(v bool) { d.emitEnabled = v }
|
|
||||||
|
|
||||||
// EmitEnabled reports whether calls to the emitFunc provided to NewDecoder
|
|
||||||
// are currently enabled. The default is true.
|
|
||||||
func (d *Decoder) EmitEnabled() bool { return d.emitEnabled }
|
|
||||||
|
|
||||||
// TODO: add method *Decoder.Reset(maxSize, emitFunc) to let callers re-use Decoders and their
|
|
||||||
// underlying buffers for garbage reasons.
|
|
||||||
|
|
||||||
func (d *Decoder) SetMaxDynamicTableSize(v uint32) {
|
|
||||||
d.dynTab.setMaxSize(v)
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetAllowedMaxDynamicTableSize sets the upper bound that the encoded
|
|
||||||
// stream (via dynamic table size updates) may set the maximum size
|
|
||||||
// to.
|
|
||||||
func (d *Decoder) SetAllowedMaxDynamicTableSize(v uint32) {
|
|
||||||
d.dynTab.allowedMaxSize = v
|
|
||||||
}
|
|
||||||
|
|
||||||
type dynamicTable struct {
|
|
||||||
// ents is the FIFO described at
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.2.3.2
|
|
||||||
// The newest (low index) is append at the end, and items are
|
|
||||||
// evicted from the front.
|
|
||||||
ents []HeaderField
|
|
||||||
size uint32
|
|
||||||
maxSize uint32 // current maxSize
|
|
||||||
allowedMaxSize uint32 // maxSize may go up to this, inclusive
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dt *dynamicTable) setMaxSize(v uint32) {
|
|
||||||
dt.maxSize = v
|
|
||||||
dt.evict()
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO: change dynamicTable to be a struct with a slice and a size int field,
|
|
||||||
// per http://http2.github.io/http2-spec/compression.html#rfc.section.4.1:
|
|
||||||
//
|
|
||||||
//
|
|
||||||
// Then make add increment the size. maybe the max size should move from Decoder to
|
|
||||||
// dynamicTable and add should return an ok bool if there was enough space.
|
|
||||||
//
|
|
||||||
// Later we'll need a remove operation on dynamicTable.
|
|
||||||
|
|
||||||
func (dt *dynamicTable) add(f HeaderField) {
|
|
||||||
dt.ents = append(dt.ents, f)
|
|
||||||
dt.size += f.Size()
|
|
||||||
dt.evict()
|
|
||||||
}
|
|
||||||
|
|
||||||
// If we're too big, evict old stuff (front of the slice)
|
|
||||||
func (dt *dynamicTable) evict() {
|
|
||||||
base := dt.ents // keep base pointer of slice
|
|
||||||
for dt.size > dt.maxSize {
|
|
||||||
dt.size -= dt.ents[0].Size()
|
|
||||||
dt.ents = dt.ents[1:]
|
|
||||||
}
|
|
||||||
|
|
||||||
// Shift slice contents down if we evicted things.
|
|
||||||
if len(dt.ents) != len(base) {
|
|
||||||
copy(base, dt.ents)
|
|
||||||
dt.ents = base[:len(dt.ents)]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// constantTimeStringCompare compares string a and b in a constant
|
|
||||||
// time manner.
|
|
||||||
func constantTimeStringCompare(a, b string) bool {
|
|
||||||
if len(a) != len(b) {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
c := byte(0)
|
|
||||||
|
|
||||||
for i := 0; i < len(a); i++ {
|
|
||||||
c |= a[i] ^ b[i]
|
|
||||||
}
|
|
||||||
|
|
||||||
return c == 0
|
|
||||||
}
|
|
||||||
|
|
||||||
// Search searches f in the table. The return value i is 0 if there is
|
|
||||||
// no name match. If there is name match or name/value match, i is the
|
|
||||||
// index of that entry (1-based). If both name and value match,
|
|
||||||
// nameValueMatch becomes true.
|
|
||||||
func (dt *dynamicTable) search(f HeaderField) (i uint64, nameValueMatch bool) {
|
|
||||||
l := len(dt.ents)
|
|
||||||
for j := l - 1; j >= 0; j-- {
|
|
||||||
ent := dt.ents[j]
|
|
||||||
if !constantTimeStringCompare(ent.Name, f.Name) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if i == 0 {
|
|
||||||
i = uint64(l - j)
|
|
||||||
}
|
|
||||||
if f.Sensitive {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !constantTimeStringCompare(ent.Value, f.Value) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
i = uint64(l - j)
|
|
||||||
nameValueMatch = true
|
|
||||||
return
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Decoder) maxTableIndex() int {
|
|
||||||
return len(d.dynTab.ents) + len(staticTable)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Decoder) at(i uint64) (hf HeaderField, ok bool) {
|
|
||||||
if i < 1 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if i > uint64(d.maxTableIndex()) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if i <= uint64(len(staticTable)) {
|
|
||||||
return staticTable[i-1], true
|
|
||||||
}
|
|
||||||
dents := d.dynTab.ents
|
|
||||||
return dents[len(dents)-(int(i)-len(staticTable))], true
|
|
||||||
}
|
|
||||||
|
|
||||||
// Decode decodes an entire block.
|
|
||||||
//
|
|
||||||
// TODO: remove this method and make it incremental later? This is
|
|
||||||
// easier for debugging now.
|
|
||||||
func (d *Decoder) DecodeFull(p []byte) ([]HeaderField, error) {
|
|
||||||
var hf []HeaderField
|
|
||||||
saveFunc := d.emit
|
|
||||||
defer func() { d.emit = saveFunc }()
|
|
||||||
d.emit = func(f HeaderField) { hf = append(hf, f) }
|
|
||||||
if _, err := d.Write(p); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if err := d.Close(); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return hf, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Decoder) Close() error {
|
|
||||||
if d.saveBuf.Len() > 0 {
|
|
||||||
d.saveBuf.Reset()
|
|
||||||
return DecodingError{errors.New("truncated headers")}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Decoder) Write(p []byte) (n int, err error) {
|
|
||||||
if len(p) == 0 {
|
|
||||||
// Prevent state machine CPU attacks (making us redo
|
|
||||||
// work up to the point of finding out we don't have
|
|
||||||
// enough data)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
// Only copy the data if we have to. Optimistically assume
|
|
||||||
// that p will contain a complete header block.
|
|
||||||
if d.saveBuf.Len() == 0 {
|
|
||||||
d.buf = p
|
|
||||||
} else {
|
|
||||||
d.saveBuf.Write(p)
|
|
||||||
d.buf = d.saveBuf.Bytes()
|
|
||||||
d.saveBuf.Reset()
|
|
||||||
}
|
|
||||||
|
|
||||||
for len(d.buf) > 0 {
|
|
||||||
err = d.parseHeaderFieldRepr()
|
|
||||||
if err == errNeedMore {
|
|
||||||
// Extra paranoia, making sure saveBuf won't
|
|
||||||
// get too large. All the varint and string
|
|
||||||
// reading code earlier should already catch
|
|
||||||
// overlong things and return ErrStringLength,
|
|
||||||
// but keep this as a last resort.
|
|
||||||
const varIntOverhead = 8 // conservative
|
|
||||||
if d.maxStrLen != 0 && int64(len(d.buf)) > 2*(int64(d.maxStrLen)+varIntOverhead) {
|
|
||||||
return 0, ErrStringLength
|
|
||||||
}
|
|
||||||
d.saveBuf.Write(d.buf)
|
|
||||||
return len(p), nil
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return len(p), err
|
|
||||||
}
|
|
||||||
|
|
||||||
// errNeedMore is an internal sentinel error value that means the
|
|
||||||
// buffer is truncated and we need to read more data before we can
|
|
||||||
// continue parsing.
|
|
||||||
var errNeedMore = errors.New("need more data")
|
|
||||||
|
|
||||||
type indexType int
|
|
||||||
|
|
||||||
const (
|
|
||||||
indexedTrue indexType = iota
|
|
||||||
indexedFalse
|
|
||||||
indexedNever
|
|
||||||
)
|
|
||||||
|
|
||||||
func (v indexType) indexed() bool { return v == indexedTrue }
|
|
||||||
func (v indexType) sensitive() bool { return v == indexedNever }
|
|
||||||
|
|
||||||
// returns errNeedMore if there isn't enough data available.
|
|
||||||
// any other error is fatal.
|
|
||||||
// consumes d.buf iff it returns nil.
|
|
||||||
// precondition: must be called with len(d.buf) > 0
|
|
||||||
func (d *Decoder) parseHeaderFieldRepr() error {
|
|
||||||
b := d.buf[0]
|
|
||||||
switch {
|
|
||||||
case b&128 != 0:
|
|
||||||
// Indexed representation.
|
|
||||||
// High bit set?
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.6.1
|
|
||||||
return d.parseFieldIndexed()
|
|
||||||
case b&192 == 64:
|
|
||||||
// 6.2.1 Literal Header Field with Incremental Indexing
|
|
||||||
// 0b10xxxxxx: top two bits are 10
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.6.2.1
|
|
||||||
return d.parseFieldLiteral(6, indexedTrue)
|
|
||||||
case b&240 == 0:
|
|
||||||
// 6.2.2 Literal Header Field without Indexing
|
|
||||||
// 0b0000xxxx: top four bits are 0000
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.6.2.2
|
|
||||||
return d.parseFieldLiteral(4, indexedFalse)
|
|
||||||
case b&240 == 16:
|
|
||||||
// 6.2.3 Literal Header Field never Indexed
|
|
||||||
// 0b0001xxxx: top four bits are 0001
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.6.2.3
|
|
||||||
return d.parseFieldLiteral(4, indexedNever)
|
|
||||||
case b&224 == 32:
|
|
||||||
// 6.3 Dynamic Table Size Update
|
|
||||||
// Top three bits are '001'.
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.6.3
|
|
||||||
return d.parseDynamicTableSizeUpdate()
|
|
||||||
}
|
|
||||||
|
|
||||||
return DecodingError{errors.New("invalid encoding")}
|
|
||||||
}
|
|
||||||
|
|
||||||
// (same invariants and behavior as parseHeaderFieldRepr)
|
|
||||||
func (d *Decoder) parseFieldIndexed() error {
|
|
||||||
buf := d.buf
|
|
||||||
idx, buf, err := readVarInt(7, buf)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
hf, ok := d.at(idx)
|
|
||||||
if !ok {
|
|
||||||
return DecodingError{InvalidIndexError(idx)}
|
|
||||||
}
|
|
||||||
d.buf = buf
|
|
||||||
return d.callEmit(HeaderField{Name: hf.Name, Value: hf.Value})
|
|
||||||
}
|
|
||||||
|
|
||||||
// (same invariants and behavior as parseHeaderFieldRepr)
|
|
||||||
func (d *Decoder) parseFieldLiteral(n uint8, it indexType) error {
|
|
||||||
buf := d.buf
|
|
||||||
nameIdx, buf, err := readVarInt(n, buf)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
var hf HeaderField
|
|
||||||
wantStr := d.emitEnabled || it.indexed()
|
|
||||||
if nameIdx > 0 {
|
|
||||||
ihf, ok := d.at(nameIdx)
|
|
||||||
if !ok {
|
|
||||||
return DecodingError{InvalidIndexError(nameIdx)}
|
|
||||||
}
|
|
||||||
hf.Name = ihf.Name
|
|
||||||
} else {
|
|
||||||
hf.Name, buf, err = d.readString(buf, wantStr)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
hf.Value, buf, err = d.readString(buf, wantStr)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
d.buf = buf
|
|
||||||
if it.indexed() {
|
|
||||||
d.dynTab.add(hf)
|
|
||||||
}
|
|
||||||
hf.Sensitive = it.sensitive()
|
|
||||||
return d.callEmit(hf)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Decoder) callEmit(hf HeaderField) error {
|
|
||||||
if d.maxStrLen != 0 {
|
|
||||||
if len(hf.Name) > d.maxStrLen || len(hf.Value) > d.maxStrLen {
|
|
||||||
return ErrStringLength
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if d.emitEnabled {
|
|
||||||
d.emit(hf)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// (same invariants and behavior as parseHeaderFieldRepr)
|
|
||||||
func (d *Decoder) parseDynamicTableSizeUpdate() error {
|
|
||||||
buf := d.buf
|
|
||||||
size, buf, err := readVarInt(5, buf)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if size > uint64(d.dynTab.allowedMaxSize) {
|
|
||||||
return DecodingError{errors.New("dynamic table size update too large")}
|
|
||||||
}
|
|
||||||
d.dynTab.setMaxSize(uint32(size))
|
|
||||||
d.buf = buf
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var errVarintOverflow = DecodingError{errors.New("varint integer overflow")}
|
|
||||||
|
|
||||||
// readVarInt reads an unsigned variable length integer off the
|
|
||||||
// beginning of p. n is the parameter as described in
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.5.1.
|
|
||||||
//
|
|
||||||
// n must always be between 1 and 8.
|
|
||||||
//
|
|
||||||
// The returned remain buffer is either a smaller suffix of p, or err != nil.
|
|
||||||
// The error is errNeedMore if p doesn't contain a complete integer.
|
|
||||||
func readVarInt(n byte, p []byte) (i uint64, remain []byte, err error) {
|
|
||||||
if n < 1 || n > 8 {
|
|
||||||
panic("bad n")
|
|
||||||
}
|
|
||||||
if len(p) == 0 {
|
|
||||||
return 0, p, errNeedMore
|
|
||||||
}
|
|
||||||
i = uint64(p[0])
|
|
||||||
if n < 8 {
|
|
||||||
i &= (1 << uint64(n)) - 1
|
|
||||||
}
|
|
||||||
if i < (1<<uint64(n))-1 {
|
|
||||||
return i, p[1:], nil
|
|
||||||
}
|
|
||||||
|
|
||||||
origP := p
|
|
||||||
p = p[1:]
|
|
||||||
var m uint64
|
|
||||||
for len(p) > 0 {
|
|
||||||
b := p[0]
|
|
||||||
p = p[1:]
|
|
||||||
i += uint64(b&127) << m
|
|
||||||
if b&128 == 0 {
|
|
||||||
return i, p, nil
|
|
||||||
}
|
|
||||||
m += 7
|
|
||||||
if m >= 63 { // TODO: proper overflow check. making this up.
|
|
||||||
return 0, origP, errVarintOverflow
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return 0, origP, errNeedMore
|
|
||||||
}
|
|
||||||
|
|
||||||
// readString decodes an hpack string from p.
|
|
||||||
//
|
|
||||||
// wantStr is whether s will be used. If false, decompression and
|
|
||||||
// []byte->string garbage are skipped if s will be ignored
|
|
||||||
// anyway. This does mean that huffman decoding errors for non-indexed
|
|
||||||
// strings past the MAX_HEADER_LIST_SIZE are ignored, but the server
|
|
||||||
// is returning an error anyway, and because they're not indexed, the error
|
|
||||||
// won't affect the decoding state.
|
|
||||||
func (d *Decoder) readString(p []byte, wantStr bool) (s string, remain []byte, err error) {
|
|
||||||
if len(p) == 0 {
|
|
||||||
return "", p, errNeedMore
|
|
||||||
}
|
|
||||||
isHuff := p[0]&128 != 0
|
|
||||||
strLen, p, err := readVarInt(7, p)
|
|
||||||
if err != nil {
|
|
||||||
return "", p, err
|
|
||||||
}
|
|
||||||
if d.maxStrLen != 0 && strLen > uint64(d.maxStrLen) {
|
|
||||||
return "", nil, ErrStringLength
|
|
||||||
}
|
|
||||||
if uint64(len(p)) < strLen {
|
|
||||||
return "", p, errNeedMore
|
|
||||||
}
|
|
||||||
if !isHuff {
|
|
||||||
if wantStr {
|
|
||||||
s = string(p[:strLen])
|
|
||||||
}
|
|
||||||
return s, p[strLen:], nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if wantStr {
|
|
||||||
buf := bufPool.Get().(*bytes.Buffer)
|
|
||||||
buf.Reset() // don't trust others
|
|
||||||
defer bufPool.Put(buf)
|
|
||||||
if err := huffmanDecode(buf, d.maxStrLen, p[:strLen]); err != nil {
|
|
||||||
buf.Reset()
|
|
||||||
return "", nil, err
|
|
||||||
}
|
|
||||||
s = buf.String()
|
|
||||||
buf.Reset() // be nice to GC
|
|
||||||
}
|
|
||||||
return s, p[strLen:], nil
|
|
||||||
}
|
|
813
vendor/golang.org/x/net/http2/hpack/hpack_test.go
generated
vendored
813
vendor/golang.org/x/net/http2/hpack/hpack_test.go
generated
vendored
@ -1,813 +0,0 @@
|
|||||||
// Copyright 2014 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 hpack
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bufio"
|
|
||||||
"bytes"
|
|
||||||
"encoding/hex"
|
|
||||||
"fmt"
|
|
||||||
"math/rand"
|
|
||||||
"reflect"
|
|
||||||
"regexp"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestStaticTable(t *testing.T) {
|
|
||||||
fromSpec := `
|
|
||||||
+-------+-----------------------------+---------------+
|
|
||||||
| 1 | :authority | |
|
|
||||||
| 2 | :method | GET |
|
|
||||||
| 3 | :method | POST |
|
|
||||||
| 4 | :path | / |
|
|
||||||
| 5 | :path | /index.html |
|
|
||||||
| 6 | :scheme | http |
|
|
||||||
| 7 | :scheme | https |
|
|
||||||
| 8 | :status | 200 |
|
|
||||||
| 9 | :status | 204 |
|
|
||||||
| 10 | :status | 206 |
|
|
||||||
| 11 | :status | 304 |
|
|
||||||
| 12 | :status | 400 |
|
|
||||||
| 13 | :status | 404 |
|
|
||||||
| 14 | :status | 500 |
|
|
||||||
| 15 | accept-charset | |
|
|
||||||
| 16 | accept-encoding | gzip, deflate |
|
|
||||||
| 17 | accept-language | |
|
|
||||||
| 18 | accept-ranges | |
|
|
||||||
| 19 | accept | |
|
|
||||||
| 20 | access-control-allow-origin | |
|
|
||||||
| 21 | age | |
|
|
||||||
| 22 | allow | |
|
|
||||||
| 23 | authorization | |
|
|
||||||
| 24 | cache-control | |
|
|
||||||
| 25 | content-disposition | |
|
|
||||||
| 26 | content-encoding | |
|
|
||||||
| 27 | content-language | |
|
|
||||||
| 28 | content-length | |
|
|
||||||
| 29 | content-location | |
|
|
||||||
| 30 | content-range | |
|
|
||||||
| 31 | content-type | |
|
|
||||||
| 32 | cookie | |
|
|
||||||
| 33 | date | |
|
|
||||||
| 34 | etag | |
|
|
||||||
| 35 | expect | |
|
|
||||||
| 36 | expires | |
|
|
||||||
| 37 | from | |
|
|
||||||
| 38 | host | |
|
|
||||||
| 39 | if-match | |
|
|
||||||
| 40 | if-modified-since | |
|
|
||||||
| 41 | if-none-match | |
|
|
||||||
| 42 | if-range | |
|
|
||||||
| 43 | if-unmodified-since | |
|
|
||||||
| 44 | last-modified | |
|
|
||||||
| 45 | link | |
|
|
||||||
| 46 | location | |
|
|
||||||
| 47 | max-forwards | |
|
|
||||||
| 48 | proxy-authenticate | |
|
|
||||||
| 49 | proxy-authorization | |
|
|
||||||
| 50 | range | |
|
|
||||||
| 51 | referer | |
|
|
||||||
| 52 | refresh | |
|
|
||||||
| 53 | retry-after | |
|
|
||||||
| 54 | server | |
|
|
||||||
| 55 | set-cookie | |
|
|
||||||
| 56 | strict-transport-security | |
|
|
||||||
| 57 | transfer-encoding | |
|
|
||||||
| 58 | user-agent | |
|
|
||||||
| 59 | vary | |
|
|
||||||
| 60 | via | |
|
|
||||||
| 61 | www-authenticate | |
|
|
||||||
+-------+-----------------------------+---------------+
|
|
||||||
`
|
|
||||||
bs := bufio.NewScanner(strings.NewReader(fromSpec))
|
|
||||||
re := regexp.MustCompile(`\| (\d+)\s+\| (\S+)\s*\| (\S(.*\S)?)?\s+\|`)
|
|
||||||
for bs.Scan() {
|
|
||||||
l := bs.Text()
|
|
||||||
if !strings.Contains(l, "|") {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
m := re.FindStringSubmatch(l)
|
|
||||||
if m == nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
i, err := strconv.Atoi(m[1])
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("Bogus integer on line %q", l)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if i < 1 || i > len(staticTable) {
|
|
||||||
t.Errorf("Bogus index %d on line %q", i, l)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if got, want := staticTable[i-1].Name, m[2]; got != want {
|
|
||||||
t.Errorf("header index %d name = %q; want %q", i, got, want)
|
|
||||||
}
|
|
||||||
if got, want := staticTable[i-1].Value, m[3]; got != want {
|
|
||||||
t.Errorf("header index %d value = %q; want %q", i, got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if err := bs.Err(); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Decoder) mustAt(idx int) HeaderField {
|
|
||||||
if hf, ok := d.at(uint64(idx)); !ok {
|
|
||||||
panic(fmt.Sprintf("bogus index %d", idx))
|
|
||||||
} else {
|
|
||||||
return hf
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDynamicTableAt(t *testing.T) {
|
|
||||||
d := NewDecoder(4096, nil)
|
|
||||||
at := d.mustAt
|
|
||||||
if got, want := at(2), (pair(":method", "GET")); got != want {
|
|
||||||
t.Errorf("at(2) = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
d.dynTab.add(pair("foo", "bar"))
|
|
||||||
d.dynTab.add(pair("blake", "miz"))
|
|
||||||
if got, want := at(len(staticTable)+1), (pair("blake", "miz")); got != want {
|
|
||||||
t.Errorf("at(dyn 1) = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
if got, want := at(len(staticTable)+2), (pair("foo", "bar")); got != want {
|
|
||||||
t.Errorf("at(dyn 2) = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
if got, want := at(3), (pair(":method", "POST")); got != want {
|
|
||||||
t.Errorf("at(3) = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDynamicTableSearch(t *testing.T) {
|
|
||||||
dt := dynamicTable{}
|
|
||||||
dt.setMaxSize(4096)
|
|
||||||
|
|
||||||
dt.add(pair("foo", "bar"))
|
|
||||||
dt.add(pair("blake", "miz"))
|
|
||||||
dt.add(pair(":method", "GET"))
|
|
||||||
|
|
||||||
tests := []struct {
|
|
||||||
hf HeaderField
|
|
||||||
wantI uint64
|
|
||||||
wantMatch bool
|
|
||||||
}{
|
|
||||||
// Name and Value match
|
|
||||||
{pair("foo", "bar"), 3, true},
|
|
||||||
{pair(":method", "GET"), 1, true},
|
|
||||||
|
|
||||||
// Only name match because of Sensitive == true
|
|
||||||
{HeaderField{"blake", "miz", true}, 2, false},
|
|
||||||
|
|
||||||
// Only Name matches
|
|
||||||
{pair("foo", "..."), 3, false},
|
|
||||||
{pair("blake", "..."), 2, false},
|
|
||||||
{pair(":method", "..."), 1, false},
|
|
||||||
|
|
||||||
// None match
|
|
||||||
{pair("foo-", "bar"), 0, false},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
if gotI, gotMatch := dt.search(tt.hf); gotI != tt.wantI || gotMatch != tt.wantMatch {
|
|
||||||
t.Errorf("d.search(%+v) = %v, %v; want %v, %v", tt.hf, gotI, gotMatch, tt.wantI, tt.wantMatch)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDynamicTableSizeEvict(t *testing.T) {
|
|
||||||
d := NewDecoder(4096, nil)
|
|
||||||
if want := uint32(0); d.dynTab.size != want {
|
|
||||||
t.Fatalf("size = %d; want %d", d.dynTab.size, want)
|
|
||||||
}
|
|
||||||
add := d.dynTab.add
|
|
||||||
add(pair("blake", "eats pizza"))
|
|
||||||
if want := uint32(15 + 32); d.dynTab.size != want {
|
|
||||||
t.Fatalf("after pizza, size = %d; want %d", d.dynTab.size, want)
|
|
||||||
}
|
|
||||||
add(pair("foo", "bar"))
|
|
||||||
if want := uint32(15 + 32 + 6 + 32); d.dynTab.size != want {
|
|
||||||
t.Fatalf("after foo bar, size = %d; want %d", d.dynTab.size, want)
|
|
||||||
}
|
|
||||||
d.dynTab.setMaxSize(15 + 32 + 1 /* slop */)
|
|
||||||
if want := uint32(6 + 32); d.dynTab.size != want {
|
|
||||||
t.Fatalf("after setMaxSize, size = %d; want %d", d.dynTab.size, want)
|
|
||||||
}
|
|
||||||
if got, want := d.mustAt(len(staticTable)+1), (pair("foo", "bar")); got != want {
|
|
||||||
t.Errorf("at(dyn 1) = %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
add(pair("long", strings.Repeat("x", 500)))
|
|
||||||
if want := uint32(0); d.dynTab.size != want {
|
|
||||||
t.Fatalf("after big one, size = %d; want %d", d.dynTab.size, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDecoderDecode(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
name string
|
|
||||||
in []byte
|
|
||||||
want []HeaderField
|
|
||||||
wantDynTab []HeaderField // newest entry first
|
|
||||||
}{
|
|
||||||
// C.2.1 Literal Header Field with Indexing
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.C.2.1
|
|
||||||
{"C.2.1", dehex("400a 6375 7374 6f6d 2d6b 6579 0d63 7573 746f 6d2d 6865 6164 6572"),
|
|
||||||
[]HeaderField{pair("custom-key", "custom-header")},
|
|
||||||
[]HeaderField{pair("custom-key", "custom-header")},
|
|
||||||
},
|
|
||||||
|
|
||||||
// C.2.2 Literal Header Field without Indexing
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.C.2.2
|
|
||||||
{"C.2.2", dehex("040c 2f73 616d 706c 652f 7061 7468"),
|
|
||||||
[]HeaderField{pair(":path", "/sample/path")},
|
|
||||||
[]HeaderField{}},
|
|
||||||
|
|
||||||
// C.2.3 Literal Header Field never Indexed
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.C.2.3
|
|
||||||
{"C.2.3", dehex("1008 7061 7373 776f 7264 0673 6563 7265 74"),
|
|
||||||
[]HeaderField{{"password", "secret", true}},
|
|
||||||
[]HeaderField{}},
|
|
||||||
|
|
||||||
// C.2.4 Indexed Header Field
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.C.2.4
|
|
||||||
{"C.2.4", []byte("\x82"),
|
|
||||||
[]HeaderField{pair(":method", "GET")},
|
|
||||||
[]HeaderField{}},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
d := NewDecoder(4096, nil)
|
|
||||||
hf, err := d.DecodeFull(tt.in)
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("%s: %v", tt.name, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(hf, tt.want) {
|
|
||||||
t.Errorf("%s: Got %v; want %v", tt.name, hf, tt.want)
|
|
||||||
}
|
|
||||||
gotDynTab := d.dynTab.reverseCopy()
|
|
||||||
if !reflect.DeepEqual(gotDynTab, tt.wantDynTab) {
|
|
||||||
t.Errorf("%s: dynamic table after = %v; want %v", tt.name, gotDynTab, tt.wantDynTab)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dt *dynamicTable) reverseCopy() (hf []HeaderField) {
|
|
||||||
hf = make([]HeaderField, len(dt.ents))
|
|
||||||
for i := range hf {
|
|
||||||
hf[i] = dt.ents[len(dt.ents)-1-i]
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
type encAndWant struct {
|
|
||||||
enc []byte
|
|
||||||
want []HeaderField
|
|
||||||
wantDynTab []HeaderField
|
|
||||||
wantDynSize uint32
|
|
||||||
}
|
|
||||||
|
|
||||||
// C.3 Request Examples without Huffman Coding
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.C.3
|
|
||||||
func TestDecodeC3_NoHuffman(t *testing.T) {
|
|
||||||
testDecodeSeries(t, 4096, []encAndWant{
|
|
||||||
{dehex("8286 8441 0f77 7777 2e65 7861 6d70 6c65 2e63 6f6d"),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":scheme", "http"),
|
|
||||||
pair(":path", "/"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
},
|
|
||||||
57,
|
|
||||||
},
|
|
||||||
{dehex("8286 84be 5808 6e6f 2d63 6163 6865"),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":scheme", "http"),
|
|
||||||
pair(":path", "/"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
pair("cache-control", "no-cache"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair("cache-control", "no-cache"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
},
|
|
||||||
110,
|
|
||||||
},
|
|
||||||
{dehex("8287 85bf 400a 6375 7374 6f6d 2d6b 6579 0c63 7573 746f 6d2d 7661 6c75 65"),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":scheme", "https"),
|
|
||||||
pair(":path", "/index.html"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
pair("custom-key", "custom-value"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair("custom-key", "custom-value"),
|
|
||||||
pair("cache-control", "no-cache"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
},
|
|
||||||
164,
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// C.4 Request Examples with Huffman Coding
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.C.4
|
|
||||||
func TestDecodeC4_Huffman(t *testing.T) {
|
|
||||||
testDecodeSeries(t, 4096, []encAndWant{
|
|
||||||
{dehex("8286 8441 8cf1 e3c2 e5f2 3a6b a0ab 90f4 ff"),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":scheme", "http"),
|
|
||||||
pair(":path", "/"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
},
|
|
||||||
57,
|
|
||||||
},
|
|
||||||
{dehex("8286 84be 5886 a8eb 1064 9cbf"),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":scheme", "http"),
|
|
||||||
pair(":path", "/"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
pair("cache-control", "no-cache"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair("cache-control", "no-cache"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
},
|
|
||||||
110,
|
|
||||||
},
|
|
||||||
{dehex("8287 85bf 4088 25a8 49e9 5ba9 7d7f 8925 a849 e95b b8e8 b4bf"),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":scheme", "https"),
|
|
||||||
pair(":path", "/index.html"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
pair("custom-key", "custom-value"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair("custom-key", "custom-value"),
|
|
||||||
pair("cache-control", "no-cache"),
|
|
||||||
pair(":authority", "www.example.com"),
|
|
||||||
},
|
|
||||||
164,
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.C.5
|
|
||||||
// "This section shows several consecutive header lists, corresponding
|
|
||||||
// to HTTP responses, on the same connection. The HTTP/2 setting
|
|
||||||
// parameter SETTINGS_HEADER_TABLE_SIZE is set to the value of 256
|
|
||||||
// octets, causing some evictions to occur."
|
|
||||||
func TestDecodeC5_ResponsesNoHuff(t *testing.T) {
|
|
||||||
testDecodeSeries(t, 256, []encAndWant{
|
|
||||||
{dehex(`
|
|
||||||
4803 3330 3258 0770 7269 7661 7465 611d
|
|
||||||
4d6f 6e2c 2032 3120 4f63 7420 3230 3133
|
|
||||||
2032 303a 3133 3a32 3120 474d 546e 1768
|
|
||||||
7474 7073 3a2f 2f77 7777 2e65 7861 6d70
|
|
||||||
6c65 2e63 6f6d
|
|
||||||
`),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":status", "302"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:21 GMT"),
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:21 GMT"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
pair(":status", "302"),
|
|
||||||
},
|
|
||||||
222,
|
|
||||||
},
|
|
||||||
{dehex("4803 3330 37c1 c0bf"),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":status", "307"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:21 GMT"),
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":status", "307"),
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:21 GMT"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
},
|
|
||||||
222,
|
|
||||||
},
|
|
||||||
{dehex(`
|
|
||||||
88c1 611d 4d6f 6e2c 2032 3120 4f63 7420
|
|
||||||
3230 3133 2032 303a 3133 3a32 3220 474d
|
|
||||||
54c0 5a04 677a 6970 7738 666f 6f3d 4153
|
|
||||||
444a 4b48 514b 425a 584f 5157 454f 5049
|
|
||||||
5541 5851 5745 4f49 553b 206d 6178 2d61
|
|
||||||
6765 3d33 3630 303b 2076 6572 7369 6f6e
|
|
||||||
3d31
|
|
||||||
`),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":status", "200"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:22 GMT"),
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
pair("content-encoding", "gzip"),
|
|
||||||
pair("set-cookie", "foo=ASDJKHQKBZXOQWEOPIUAXQWEOIU; max-age=3600; version=1"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair("set-cookie", "foo=ASDJKHQKBZXOQWEOPIUAXQWEOIU; max-age=3600; version=1"),
|
|
||||||
pair("content-encoding", "gzip"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:22 GMT"),
|
|
||||||
},
|
|
||||||
215,
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// http://http2.github.io/http2-spec/compression.html#rfc.section.C.6
|
|
||||||
// "This section shows the same examples as the previous section, but
|
|
||||||
// using Huffman encoding for the literal values. The HTTP/2 setting
|
|
||||||
// parameter SETTINGS_HEADER_TABLE_SIZE is set to the value of 256
|
|
||||||
// octets, causing some evictions to occur. The eviction mechanism
|
|
||||||
// uses the length of the decoded literal values, so the same
|
|
||||||
// evictions occurs as in the previous section."
|
|
||||||
func TestDecodeC6_ResponsesHuffman(t *testing.T) {
|
|
||||||
testDecodeSeries(t, 256, []encAndWant{
|
|
||||||
{dehex(`
|
|
||||||
4882 6402 5885 aec3 771a 4b61 96d0 7abe
|
|
||||||
9410 54d4 44a8 2005 9504 0b81 66e0 82a6
|
|
||||||
2d1b ff6e 919d 29ad 1718 63c7 8f0b 97c8
|
|
||||||
e9ae 82ae 43d3
|
|
||||||
`),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":status", "302"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:21 GMT"),
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:21 GMT"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
pair(":status", "302"),
|
|
||||||
},
|
|
||||||
222,
|
|
||||||
},
|
|
||||||
{dehex("4883 640e ffc1 c0bf"),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":status", "307"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:21 GMT"),
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":status", "307"),
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:21 GMT"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
},
|
|
||||||
222,
|
|
||||||
},
|
|
||||||
{dehex(`
|
|
||||||
88c1 6196 d07a be94 1054 d444 a820 0595
|
|
||||||
040b 8166 e084 a62d 1bff c05a 839b d9ab
|
|
||||||
77ad 94e7 821d d7f2 e6c7 b335 dfdf cd5b
|
|
||||||
3960 d5af 2708 7f36 72c1 ab27 0fb5 291f
|
|
||||||
9587 3160 65c0 03ed 4ee5 b106 3d50 07
|
|
||||||
`),
|
|
||||||
[]HeaderField{
|
|
||||||
pair(":status", "200"),
|
|
||||||
pair("cache-control", "private"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:22 GMT"),
|
|
||||||
pair("location", "https://www.example.com"),
|
|
||||||
pair("content-encoding", "gzip"),
|
|
||||||
pair("set-cookie", "foo=ASDJKHQKBZXOQWEOPIUAXQWEOIU; max-age=3600; version=1"),
|
|
||||||
},
|
|
||||||
[]HeaderField{
|
|
||||||
pair("set-cookie", "foo=ASDJKHQKBZXOQWEOPIUAXQWEOIU; max-age=3600; version=1"),
|
|
||||||
pair("content-encoding", "gzip"),
|
|
||||||
pair("date", "Mon, 21 Oct 2013 20:13:22 GMT"),
|
|
||||||
},
|
|
||||||
215,
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func testDecodeSeries(t *testing.T, size uint32, steps []encAndWant) {
|
|
||||||
d := NewDecoder(size, nil)
|
|
||||||
for i, step := range steps {
|
|
||||||
hf, err := d.DecodeFull(step.enc)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("Error at step index %d: %v", i, err)
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(hf, step.want) {
|
|
||||||
t.Fatalf("At step index %d: Got headers %v; want %v", i, hf, step.want)
|
|
||||||
}
|
|
||||||
gotDynTab := d.dynTab.reverseCopy()
|
|
||||||
if !reflect.DeepEqual(gotDynTab, step.wantDynTab) {
|
|
||||||
t.Errorf("After step index %d, dynamic table = %v; want %v", i, gotDynTab, step.wantDynTab)
|
|
||||||
}
|
|
||||||
if d.dynTab.size != step.wantDynSize {
|
|
||||||
t.Errorf("After step index %d, dynamic table size = %v; want %v", i, d.dynTab.size, step.wantDynSize)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestHuffmanDecode(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
inHex, want string
|
|
||||||
}{
|
|
||||||
{"f1e3 c2e5 f23a 6ba0 ab90 f4ff", "www.example.com"},
|
|
||||||
{"a8eb 1064 9cbf", "no-cache"},
|
|
||||||
{"25a8 49e9 5ba9 7d7f", "custom-key"},
|
|
||||||
{"25a8 49e9 5bb8 e8b4 bf", "custom-value"},
|
|
||||||
{"6402", "302"},
|
|
||||||
{"aec3 771a 4b", "private"},
|
|
||||||
{"d07a be94 1054 d444 a820 0595 040b 8166 e082 a62d 1bff", "Mon, 21 Oct 2013 20:13:21 GMT"},
|
|
||||||
{"9d29 ad17 1863 c78f 0b97 c8e9 ae82 ae43 d3", "https://www.example.com"},
|
|
||||||
{"9bd9 ab", "gzip"},
|
|
||||||
{"94e7 821d d7f2 e6c7 b335 dfdf cd5b 3960 d5af 2708 7f36 72c1 ab27 0fb5 291f 9587 3160 65c0 03ed 4ee5 b106 3d50 07",
|
|
||||||
"foo=ASDJKHQKBZXOQWEOPIUAXQWEOIU; max-age=3600; version=1"},
|
|
||||||
}
|
|
||||||
for i, tt := range tests {
|
|
||||||
var buf bytes.Buffer
|
|
||||||
in, err := hex.DecodeString(strings.Replace(tt.inHex, " ", "", -1))
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("%d. hex input error: %v", i, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if _, err := HuffmanDecode(&buf, in); err != nil {
|
|
||||||
t.Errorf("%d. decode error: %v", i, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if got := buf.String(); tt.want != got {
|
|
||||||
t.Errorf("%d. decode = %q; want %q", i, got, tt.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestAppendHuffmanString(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
in, want string
|
|
||||||
}{
|
|
||||||
{"www.example.com", "f1e3 c2e5 f23a 6ba0 ab90 f4ff"},
|
|
||||||
{"no-cache", "a8eb 1064 9cbf"},
|
|
||||||
{"custom-key", "25a8 49e9 5ba9 7d7f"},
|
|
||||||
{"custom-value", "25a8 49e9 5bb8 e8b4 bf"},
|
|
||||||
{"302", "6402"},
|
|
||||||
{"private", "aec3 771a 4b"},
|
|
||||||
{"Mon, 21 Oct 2013 20:13:21 GMT", "d07a be94 1054 d444 a820 0595 040b 8166 e082 a62d 1bff"},
|
|
||||||
{"https://www.example.com", "9d29 ad17 1863 c78f 0b97 c8e9 ae82 ae43 d3"},
|
|
||||||
{"gzip", "9bd9 ab"},
|
|
||||||
{"foo=ASDJKHQKBZXOQWEOPIUAXQWEOIU; max-age=3600; version=1",
|
|
||||||
"94e7 821d d7f2 e6c7 b335 dfdf cd5b 3960 d5af 2708 7f36 72c1 ab27 0fb5 291f 9587 3160 65c0 03ed 4ee5 b106 3d50 07"},
|
|
||||||
}
|
|
||||||
for i, tt := range tests {
|
|
||||||
buf := []byte{}
|
|
||||||
want := strings.Replace(tt.want, " ", "", -1)
|
|
||||||
buf = AppendHuffmanString(buf, tt.in)
|
|
||||||
if got := hex.EncodeToString(buf); want != got {
|
|
||||||
t.Errorf("%d. encode = %q; want %q", i, got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestHuffmanMaxStrLen(t *testing.T) {
|
|
||||||
const msg = "Some string"
|
|
||||||
huff := AppendHuffmanString(nil, msg)
|
|
||||||
|
|
||||||
testGood := func(max int) {
|
|
||||||
var out bytes.Buffer
|
|
||||||
if err := huffmanDecode(&out, max, huff); err != nil {
|
|
||||||
t.Errorf("For maxLen=%d, unexpected error: %v", max, err)
|
|
||||||
}
|
|
||||||
if out.String() != msg {
|
|
||||||
t.Errorf("For maxLen=%d, out = %q; want %q", max, out.String(), msg)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
testGood(0)
|
|
||||||
testGood(len(msg))
|
|
||||||
testGood(len(msg) + 1)
|
|
||||||
|
|
||||||
var out bytes.Buffer
|
|
||||||
if err := huffmanDecode(&out, len(msg)-1, huff); err != ErrStringLength {
|
|
||||||
t.Errorf("err = %v; want ErrStringLength", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestHuffmanRoundtripStress(t *testing.T) {
|
|
||||||
const Len = 50 // of uncompressed string
|
|
||||||
input := make([]byte, Len)
|
|
||||||
var output bytes.Buffer
|
|
||||||
var huff []byte
|
|
||||||
|
|
||||||
n := 5000
|
|
||||||
if testing.Short() {
|
|
||||||
n = 100
|
|
||||||
}
|
|
||||||
seed := time.Now().UnixNano()
|
|
||||||
t.Logf("Seed = %v", seed)
|
|
||||||
src := rand.New(rand.NewSource(seed))
|
|
||||||
var encSize int64
|
|
||||||
for i := 0; i < n; i++ {
|
|
||||||
for l := range input {
|
|
||||||
input[l] = byte(src.Intn(256))
|
|
||||||
}
|
|
||||||
huff = AppendHuffmanString(huff[:0], string(input))
|
|
||||||
encSize += int64(len(huff))
|
|
||||||
output.Reset()
|
|
||||||
if err := huffmanDecode(&output, 0, huff); err != nil {
|
|
||||||
t.Errorf("Failed to decode %q -> %q -> error %v", input, huff, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !bytes.Equal(output.Bytes(), input) {
|
|
||||||
t.Errorf("Roundtrip failure on %q -> %q -> %q", input, huff, output.Bytes())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
t.Logf("Compressed size of original: %0.02f%% (%v -> %v)", 100*(float64(encSize)/(Len*float64(n))), Len*n, encSize)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestHuffmanDecodeFuzz(t *testing.T) {
|
|
||||||
const Len = 50 // of compressed
|
|
||||||
var buf, zbuf bytes.Buffer
|
|
||||||
|
|
||||||
n := 5000
|
|
||||||
if testing.Short() {
|
|
||||||
n = 100
|
|
||||||
}
|
|
||||||
seed := time.Now().UnixNano()
|
|
||||||
t.Logf("Seed = %v", seed)
|
|
||||||
src := rand.New(rand.NewSource(seed))
|
|
||||||
numFail := 0
|
|
||||||
for i := 0; i < n; i++ {
|
|
||||||
zbuf.Reset()
|
|
||||||
if i == 0 {
|
|
||||||
// Start with at least one invalid one.
|
|
||||||
zbuf.WriteString("00\x91\xff\xff\xff\xff\xc8")
|
|
||||||
} else {
|
|
||||||
for l := 0; l < Len; l++ {
|
|
||||||
zbuf.WriteByte(byte(src.Intn(256)))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
buf.Reset()
|
|
||||||
if err := huffmanDecode(&buf, 0, zbuf.Bytes()); err != nil {
|
|
||||||
if err == ErrInvalidHuffman {
|
|
||||||
numFail++
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
t.Errorf("Failed to decode %q: %v", zbuf.Bytes(), err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
t.Logf("%0.02f%% are invalid (%d / %d)", 100*float64(numFail)/float64(n), numFail, n)
|
|
||||||
if numFail < 1 {
|
|
||||||
t.Error("expected at least one invalid huffman encoding (test starts with one)")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadVarInt(t *testing.T) {
|
|
||||||
type res struct {
|
|
||||||
i uint64
|
|
||||||
consumed int
|
|
||||||
err error
|
|
||||||
}
|
|
||||||
tests := []struct {
|
|
||||||
n byte
|
|
||||||
p []byte
|
|
||||||
want res
|
|
||||||
}{
|
|
||||||
// Fits in a byte:
|
|
||||||
{1, []byte{0}, res{0, 1, nil}},
|
|
||||||
{2, []byte{2}, res{2, 1, nil}},
|
|
||||||
{3, []byte{6}, res{6, 1, nil}},
|
|
||||||
{4, []byte{14}, res{14, 1, nil}},
|
|
||||||
{5, []byte{30}, res{30, 1, nil}},
|
|
||||||
{6, []byte{62}, res{62, 1, nil}},
|
|
||||||
{7, []byte{126}, res{126, 1, nil}},
|
|
||||||
{8, []byte{254}, res{254, 1, nil}},
|
|
||||||
|
|
||||||
// Doesn't fit in a byte:
|
|
||||||
{1, []byte{1}, res{0, 0, errNeedMore}},
|
|
||||||
{2, []byte{3}, res{0, 0, errNeedMore}},
|
|
||||||
{3, []byte{7}, res{0, 0, errNeedMore}},
|
|
||||||
{4, []byte{15}, res{0, 0, errNeedMore}},
|
|
||||||
{5, []byte{31}, res{0, 0, errNeedMore}},
|
|
||||||
{6, []byte{63}, res{0, 0, errNeedMore}},
|
|
||||||
{7, []byte{127}, res{0, 0, errNeedMore}},
|
|
||||||
{8, []byte{255}, res{0, 0, errNeedMore}},
|
|
||||||
|
|
||||||
// Ignoring top bits:
|
|
||||||
{5, []byte{255, 154, 10}, res{1337, 3, nil}}, // high dummy three bits: 111
|
|
||||||
{5, []byte{159, 154, 10}, res{1337, 3, nil}}, // high dummy three bits: 100
|
|
||||||
{5, []byte{191, 154, 10}, res{1337, 3, nil}}, // high dummy three bits: 101
|
|
||||||
|
|
||||||
// Extra byte:
|
|
||||||
{5, []byte{191, 154, 10, 2}, res{1337, 3, nil}}, // extra byte
|
|
||||||
|
|
||||||
// Short a byte:
|
|
||||||
{5, []byte{191, 154}, res{0, 0, errNeedMore}},
|
|
||||||
|
|
||||||
// integer overflow:
|
|
||||||
{1, []byte{255, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128}, res{0, 0, errVarintOverflow}},
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
i, remain, err := readVarInt(tt.n, tt.p)
|
|
||||||
consumed := len(tt.p) - len(remain)
|
|
||||||
got := res{i, consumed, err}
|
|
||||||
if got != tt.want {
|
|
||||||
t.Errorf("readVarInt(%d, %v ~ %x) = %+v; want %+v", tt.n, tt.p, tt.p, got, tt.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Fuzz crash, originally reported at https://github.com/bradfitz/http2/issues/56
|
|
||||||
func TestHuffmanFuzzCrash(t *testing.T) {
|
|
||||||
got, err := HuffmanDecodeToString([]byte("00\x91\xff\xff\xff\xff\xc8"))
|
|
||||||
if got != "" {
|
|
||||||
t.Errorf("Got %q; want empty string", got)
|
|
||||||
}
|
|
||||||
if err != ErrInvalidHuffman {
|
|
||||||
t.Errorf("Err = %v; want ErrInvalidHuffman", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func dehex(s string) []byte {
|
|
||||||
s = strings.Replace(s, " ", "", -1)
|
|
||||||
s = strings.Replace(s, "\n", "", -1)
|
|
||||||
b, err := hex.DecodeString(s)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
return b
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestEmitEnabled(t *testing.T) {
|
|
||||||
var buf bytes.Buffer
|
|
||||||
enc := NewEncoder(&buf)
|
|
||||||
enc.WriteField(HeaderField{Name: "foo", Value: "bar"})
|
|
||||||
enc.WriteField(HeaderField{Name: "foo", Value: "bar"})
|
|
||||||
|
|
||||||
numCallback := 0
|
|
||||||
var dec *Decoder
|
|
||||||
dec = NewDecoder(8<<20, func(HeaderField) {
|
|
||||||
numCallback++
|
|
||||||
dec.SetEmitEnabled(false)
|
|
||||||
})
|
|
||||||
if !dec.EmitEnabled() {
|
|
||||||
t.Errorf("initial emit enabled = false; want true")
|
|
||||||
}
|
|
||||||
if _, err := dec.Write(buf.Bytes()); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
}
|
|
||||||
if numCallback != 1 {
|
|
||||||
t.Errorf("num callbacks = %d; want 1", numCallback)
|
|
||||||
}
|
|
||||||
if dec.EmitEnabled() {
|
|
||||||
t.Errorf("emit enabled = true; want false")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSaveBufLimit(t *testing.T) {
|
|
||||||
const maxStr = 1 << 10
|
|
||||||
var got []HeaderField
|
|
||||||
dec := NewDecoder(initialHeaderTableSize, func(hf HeaderField) {
|
|
||||||
got = append(got, hf)
|
|
||||||
})
|
|
||||||
dec.SetMaxStringLength(maxStr)
|
|
||||||
var frag []byte
|
|
||||||
frag = append(frag[:0], encodeTypeByte(false, false))
|
|
||||||
frag = appendVarInt(frag, 7, 3)
|
|
||||||
frag = append(frag, "foo"...)
|
|
||||||
frag = appendVarInt(frag, 7, 3)
|
|
||||||
frag = append(frag, "bar"...)
|
|
||||||
|
|
||||||
if _, err := dec.Write(frag); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
want := []HeaderField{{Name: "foo", Value: "bar"}}
|
|
||||||
if !reflect.DeepEqual(got, want) {
|
|
||||||
t.Errorf("After small writes, got %v; want %v", got, want)
|
|
||||||
}
|
|
||||||
|
|
||||||
frag = append(frag[:0], encodeTypeByte(false, false))
|
|
||||||
frag = appendVarInt(frag, 7, maxStr*3)
|
|
||||||
frag = append(frag, make([]byte, maxStr*3)...)
|
|
||||||
|
|
||||||
_, err := dec.Write(frag)
|
|
||||||
if err != ErrStringLength {
|
|
||||||
t.Fatalf("Write error = %v; want ErrStringLength", err)
|
|
||||||
}
|
|
||||||
}
|
|
190
vendor/golang.org/x/net/http2/hpack/huffman.go
generated
vendored
190
vendor/golang.org/x/net/http2/hpack/huffman.go
generated
vendored
@ -1,190 +0,0 @@
|
|||||||
// Copyright 2014 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 hpack
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"io"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
var bufPool = sync.Pool{
|
|
||||||
New: func() interface{} { return new(bytes.Buffer) },
|
|
||||||
}
|
|
||||||
|
|
||||||
// HuffmanDecode decodes the string in v and writes the expanded
|
|
||||||
// result to w, returning the number of bytes written to w and the
|
|
||||||
// Write call's return value. At most one Write call is made.
|
|
||||||
func HuffmanDecode(w io.Writer, v []byte) (int, error) {
|
|
||||||
buf := bufPool.Get().(*bytes.Buffer)
|
|
||||||
buf.Reset()
|
|
||||||
defer bufPool.Put(buf)
|
|
||||||
if err := huffmanDecode(buf, 0, v); err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
return w.Write(buf.Bytes())
|
|
||||||
}
|
|
||||||
|
|
||||||
// HuffmanDecodeToString decodes the string in v.
|
|
||||||
func HuffmanDecodeToString(v []byte) (string, error) {
|
|
||||||
buf := bufPool.Get().(*bytes.Buffer)
|
|
||||||
buf.Reset()
|
|
||||||
defer bufPool.Put(buf)
|
|
||||||
if err := huffmanDecode(buf, 0, v); err != nil {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
return buf.String(), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// ErrInvalidHuffman is returned for errors found decoding
|
|
||||||
// Huffman-encoded strings.
|
|
||||||
var ErrInvalidHuffman = errors.New("hpack: invalid Huffman-encoded data")
|
|
||||||
|
|
||||||
// huffmanDecode decodes v to buf.
|
|
||||||
// If maxLen is greater than 0, attempts to write more to buf than
|
|
||||||
// maxLen bytes will return ErrStringLength.
|
|
||||||
func huffmanDecode(buf *bytes.Buffer, maxLen int, v []byte) error {
|
|
||||||
n := rootHuffmanNode
|
|
||||||
cur, nbits := uint(0), uint8(0)
|
|
||||||
for _, b := range v {
|
|
||||||
cur = cur<<8 | uint(b)
|
|
||||||
nbits += 8
|
|
||||||
for nbits >= 8 {
|
|
||||||
idx := byte(cur >> (nbits - 8))
|
|
||||||
n = n.children[idx]
|
|
||||||
if n == nil {
|
|
||||||
return ErrInvalidHuffman
|
|
||||||
}
|
|
||||||
if n.children == nil {
|
|
||||||
if maxLen != 0 && buf.Len() == maxLen {
|
|
||||||
return ErrStringLength
|
|
||||||
}
|
|
||||||
buf.WriteByte(n.sym)
|
|
||||||
nbits -= n.codeLen
|
|
||||||
n = rootHuffmanNode
|
|
||||||
} else {
|
|
||||||
nbits -= 8
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for nbits > 0 {
|
|
||||||
n = n.children[byte(cur<<(8-nbits))]
|
|
||||||
if n.children != nil || n.codeLen > nbits {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
buf.WriteByte(n.sym)
|
|
||||||
nbits -= n.codeLen
|
|
||||||
n = rootHuffmanNode
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type node struct {
|
|
||||||
// children is non-nil for internal nodes
|
|
||||||
children []*node
|
|
||||||
|
|
||||||
// The following are only valid if children is nil:
|
|
||||||
codeLen uint8 // number of bits that led to the output of sym
|
|
||||||
sym byte // output symbol
|
|
||||||
}
|
|
||||||
|
|
||||||
func newInternalNode() *node {
|
|
||||||
return &node{children: make([]*node, 256)}
|
|
||||||
}
|
|
||||||
|
|
||||||
var rootHuffmanNode = newInternalNode()
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
if len(huffmanCodes) != 256 {
|
|
||||||
panic("unexpected size")
|
|
||||||
}
|
|
||||||
for i, code := range huffmanCodes {
|
|
||||||
addDecoderNode(byte(i), code, huffmanCodeLen[i])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func addDecoderNode(sym byte, code uint32, codeLen uint8) {
|
|
||||||
cur := rootHuffmanNode
|
|
||||||
for codeLen > 8 {
|
|
||||||
codeLen -= 8
|
|
||||||
i := uint8(code >> codeLen)
|
|
||||||
if cur.children[i] == nil {
|
|
||||||
cur.children[i] = newInternalNode()
|
|
||||||
}
|
|
||||||
cur = cur.children[i]
|
|
||||||
}
|
|
||||||
shift := 8 - codeLen
|
|
||||||
start, end := int(uint8(code<<shift)), int(1<<shift)
|
|
||||||
for i := start; i < start+end; i++ {
|
|
||||||
cur.children[i] = &node{sym: sym, codeLen: codeLen}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// AppendHuffmanString appends s, as encoded in Huffman codes, to dst
|
|
||||||
// and returns the extended buffer.
|
|
||||||
func AppendHuffmanString(dst []byte, s string) []byte {
|
|
||||||
rembits := uint8(8)
|
|
||||||
|
|
||||||
for i := 0; i < len(s); i++ {
|
|
||||||
if rembits == 8 {
|
|
||||||
dst = append(dst, 0)
|
|
||||||
}
|
|
||||||
dst, rembits = appendByteToHuffmanCode(dst, rembits, s[i])
|
|
||||||
}
|
|
||||||
|
|
||||||
if rembits < 8 {
|
|
||||||
// special EOS symbol
|
|
||||||
code := uint32(0x3fffffff)
|
|
||||||
nbits := uint8(30)
|
|
||||||
|
|
||||||
t := uint8(code >> (nbits - rembits))
|
|
||||||
dst[len(dst)-1] |= t
|
|
||||||
}
|
|
||||||
|
|
||||||
return dst
|
|
||||||
}
|
|
||||||
|
|
||||||
// HuffmanEncodeLength returns the number of bytes required to encode
|
|
||||||
// s in Huffman codes. The result is round up to byte boundary.
|
|
||||||
func HuffmanEncodeLength(s string) uint64 {
|
|
||||||
n := uint64(0)
|
|
||||||
for i := 0; i < len(s); i++ {
|
|
||||||
n += uint64(huffmanCodeLen[s[i]])
|
|
||||||
}
|
|
||||||
return (n + 7) / 8
|
|
||||||
}
|
|
||||||
|
|
||||||
// appendByteToHuffmanCode appends Huffman code for c to dst and
|
|
||||||
// returns the extended buffer and the remaining bits in the last
|
|
||||||
// element. The appending is not byte aligned and the remaining bits
|
|
||||||
// in the last element of dst is given in rembits.
|
|
||||||
func appendByteToHuffmanCode(dst []byte, rembits uint8, c byte) ([]byte, uint8) {
|
|
||||||
code := huffmanCodes[c]
|
|
||||||
nbits := huffmanCodeLen[c]
|
|
||||||
|
|
||||||
for {
|
|
||||||
if rembits > nbits {
|
|
||||||
t := uint8(code << (rembits - nbits))
|
|
||||||
dst[len(dst)-1] |= t
|
|
||||||
rembits -= nbits
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
t := uint8(code >> (nbits - rembits))
|
|
||||||
dst[len(dst)-1] |= t
|
|
||||||
|
|
||||||
nbits -= rembits
|
|
||||||
rembits = 8
|
|
||||||
|
|
||||||
if nbits == 0 {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
dst = append(dst, 0)
|
|
||||||
}
|
|
||||||
|
|
||||||
return dst, rembits
|
|
||||||
}
|
|
352
vendor/golang.org/x/net/http2/hpack/tables.go
generated
vendored
352
vendor/golang.org/x/net/http2/hpack/tables.go
generated
vendored
@ -1,352 +0,0 @@
|
|||||||
// Copyright 2014 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 hpack
|
|
||||||
|
|
||||||
func pair(name, value string) HeaderField {
|
|
||||||
return HeaderField{Name: name, Value: value}
|
|
||||||
}
|
|
||||||
|
|
||||||
// http://tools.ietf.org/html/draft-ietf-httpbis-header-compression-07#appendix-B
|
|
||||||
var staticTable = [...]HeaderField{
|
|
||||||
pair(":authority", ""), // index 1 (1-based)
|
|
||||||
pair(":method", "GET"),
|
|
||||||
pair(":method", "POST"),
|
|
||||||
pair(":path", "/"),
|
|
||||||
pair(":path", "/index.html"),
|
|
||||||
pair(":scheme", "http"),
|
|
||||||
pair(":scheme", "https"),
|
|
||||||
pair(":status", "200"),
|
|
||||||
pair(":status", "204"),
|
|
||||||
pair(":status", "206"),
|
|
||||||
pair(":status", "304"),
|
|
||||||
pair(":status", "400"),
|
|
||||||
pair(":status", "404"),
|
|
||||||
pair(":status", "500"),
|
|
||||||
pair("accept-charset", ""),
|
|
||||||
pair("accept-encoding", "gzip, deflate"),
|
|
||||||
pair("accept-language", ""),
|
|
||||||
pair("accept-ranges", ""),
|
|
||||||
pair("accept", ""),
|
|
||||||
pair("access-control-allow-origin", ""),
|
|
||||||
pair("age", ""),
|
|
||||||
pair("allow", ""),
|
|
||||||
pair("authorization", ""),
|
|
||||||
pair("cache-control", ""),
|
|
||||||
pair("content-disposition", ""),
|
|
||||||
pair("content-encoding", ""),
|
|
||||||
pair("content-language", ""),
|
|
||||||
pair("content-length", ""),
|
|
||||||
pair("content-location", ""),
|
|
||||||
pair("content-range", ""),
|
|
||||||
pair("content-type", ""),
|
|
||||||
pair("cookie", ""),
|
|
||||||
pair("date", ""),
|
|
||||||
pair("etag", ""),
|
|
||||||
pair("expect", ""),
|
|
||||||
pair("expires", ""),
|
|
||||||
pair("from", ""),
|
|
||||||
pair("host", ""),
|
|
||||||
pair("if-match", ""),
|
|
||||||
pair("if-modified-since", ""),
|
|
||||||
pair("if-none-match", ""),
|
|
||||||
pair("if-range", ""),
|
|
||||||
pair("if-unmodified-since", ""),
|
|
||||||
pair("last-modified", ""),
|
|
||||||
pair("link", ""),
|
|
||||||
pair("location", ""),
|
|
||||||
pair("max-forwards", ""),
|
|
||||||
pair("proxy-authenticate", ""),
|
|
||||||
pair("proxy-authorization", ""),
|
|
||||||
pair("range", ""),
|
|
||||||
pair("referer", ""),
|
|
||||||
pair("refresh", ""),
|
|
||||||
pair("retry-after", ""),
|
|
||||||
pair("server", ""),
|
|
||||||
pair("set-cookie", ""),
|
|
||||||
pair("strict-transport-security", ""),
|
|
||||||
pair("transfer-encoding", ""),
|
|
||||||
pair("user-agent", ""),
|
|
||||||
pair("vary", ""),
|
|
||||||
pair("via", ""),
|
|
||||||
pair("www-authenticate", ""),
|
|
||||||
}
|
|
||||||
|
|
||||||
var huffmanCodes = [256]uint32{
|
|
||||||
0x1ff8,
|
|
||||||
0x7fffd8,
|
|
||||||
0xfffffe2,
|
|
||||||
0xfffffe3,
|
|
||||||
0xfffffe4,
|
|
||||||
0xfffffe5,
|
|
||||||
0xfffffe6,
|
|
||||||
0xfffffe7,
|
|
||||||
0xfffffe8,
|
|
||||||
0xffffea,
|
|
||||||
0x3ffffffc,
|
|
||||||
0xfffffe9,
|
|
||||||
0xfffffea,
|
|
||||||
0x3ffffffd,
|
|
||||||
0xfffffeb,
|
|
||||||
0xfffffec,
|
|
||||||
0xfffffed,
|
|
||||||
0xfffffee,
|
|
||||||
0xfffffef,
|
|
||||||
0xffffff0,
|
|
||||||
0xffffff1,
|
|
||||||
0xffffff2,
|
|
||||||
0x3ffffffe,
|
|
||||||
0xffffff3,
|
|
||||||
0xffffff4,
|
|
||||||
0xffffff5,
|
|
||||||
0xffffff6,
|
|
||||||
0xffffff7,
|
|
||||||
0xffffff8,
|
|
||||||
0xffffff9,
|
|
||||||
0xffffffa,
|
|
||||||
0xffffffb,
|
|
||||||
0x14,
|
|
||||||
0x3f8,
|
|
||||||
0x3f9,
|
|
||||||
0xffa,
|
|
||||||
0x1ff9,
|
|
||||||
0x15,
|
|
||||||
0xf8,
|
|
||||||
0x7fa,
|
|
||||||
0x3fa,
|
|
||||||
0x3fb,
|
|
||||||
0xf9,
|
|
||||||
0x7fb,
|
|
||||||
0xfa,
|
|
||||||
0x16,
|
|
||||||
0x17,
|
|
||||||
0x18,
|
|
||||||
0x0,
|
|
||||||
0x1,
|
|
||||||
0x2,
|
|
||||||
0x19,
|
|
||||||
0x1a,
|
|
||||||
0x1b,
|
|
||||||
0x1c,
|
|
||||||
0x1d,
|
|
||||||
0x1e,
|
|
||||||
0x1f,
|
|
||||||
0x5c,
|
|
||||||
0xfb,
|
|
||||||
0x7ffc,
|
|
||||||
0x20,
|
|
||||||
0xffb,
|
|
||||||
0x3fc,
|
|
||||||
0x1ffa,
|
|
||||||
0x21,
|
|
||||||
0x5d,
|
|
||||||
0x5e,
|
|
||||||
0x5f,
|
|
||||||
0x60,
|
|
||||||
0x61,
|
|
||||||
0x62,
|
|
||||||
0x63,
|
|
||||||
0x64,
|
|
||||||
0x65,
|
|
||||||
0x66,
|
|
||||||
0x67,
|
|
||||||
0x68,
|
|
||||||
0x69,
|
|
||||||
0x6a,
|
|
||||||
0x6b,
|
|
||||||
0x6c,
|
|
||||||
0x6d,
|
|
||||||
0x6e,
|
|
||||||
0x6f,
|
|
||||||
0x70,
|
|
||||||
0x71,
|
|
||||||
0x72,
|
|
||||||
0xfc,
|
|
||||||
0x73,
|
|
||||||
0xfd,
|
|
||||||
0x1ffb,
|
|
||||||
0x7fff0,
|
|
||||||
0x1ffc,
|
|
||||||
0x3ffc,
|
|
||||||
0x22,
|
|
||||||
0x7ffd,
|
|
||||||
0x3,
|
|
||||||
0x23,
|
|
||||||
0x4,
|
|
||||||
0x24,
|
|
||||||
0x5,
|
|
||||||
0x25,
|
|
||||||
0x26,
|
|
||||||
0x27,
|
|
||||||
0x6,
|
|
||||||
0x74,
|
|
||||||
0x75,
|
|
||||||
0x28,
|
|
||||||
0x29,
|
|
||||||
0x2a,
|
|
||||||
0x7,
|
|
||||||
0x2b,
|
|
||||||
0x76,
|
|
||||||
0x2c,
|
|
||||||
0x8,
|
|
||||||
0x9,
|
|
||||||
0x2d,
|
|
||||||
0x77,
|
|
||||||
0x78,
|
|
||||||
0x79,
|
|
||||||
0x7a,
|
|
||||||
0x7b,
|
|
||||||
0x7ffe,
|
|
||||||
0x7fc,
|
|
||||||
0x3ffd,
|
|
||||||
0x1ffd,
|
|
||||||
0xffffffc,
|
|
||||||
0xfffe6,
|
|
||||||
0x3fffd2,
|
|
||||||
0xfffe7,
|
|
||||||
0xfffe8,
|
|
||||||
0x3fffd3,
|
|
||||||
0x3fffd4,
|
|
||||||
0x3fffd5,
|
|
||||||
0x7fffd9,
|
|
||||||
0x3fffd6,
|
|
||||||
0x7fffda,
|
|
||||||
0x7fffdb,
|
|
||||||
0x7fffdc,
|
|
||||||
0x7fffdd,
|
|
||||||
0x7fffde,
|
|
||||||
0xffffeb,
|
|
||||||
0x7fffdf,
|
|
||||||
0xffffec,
|
|
||||||
0xffffed,
|
|
||||||
0x3fffd7,
|
|
||||||
0x7fffe0,
|
|
||||||
0xffffee,
|
|
||||||
0x7fffe1,
|
|
||||||
0x7fffe2,
|
|
||||||
0x7fffe3,
|
|
||||||
0x7fffe4,
|
|
||||||
0x1fffdc,
|
|
||||||
0x3fffd8,
|
|
||||||
0x7fffe5,
|
|
||||||
0x3fffd9,
|
|
||||||
0x7fffe6,
|
|
||||||
0x7fffe7,
|
|
||||||
0xffffef,
|
|
||||||
0x3fffda,
|
|
||||||
0x1fffdd,
|
|
||||||
0xfffe9,
|
|
||||||
0x3fffdb,
|
|
||||||
0x3fffdc,
|
|
||||||
0x7fffe8,
|
|
||||||
0x7fffe9,
|
|
||||||
0x1fffde,
|
|
||||||
0x7fffea,
|
|
||||||
0x3fffdd,
|
|
||||||
0x3fffde,
|
|
||||||
0xfffff0,
|
|
||||||
0x1fffdf,
|
|
||||||
0x3fffdf,
|
|
||||||
0x7fffeb,
|
|
||||||
0x7fffec,
|
|
||||||
0x1fffe0,
|
|
||||||
0x1fffe1,
|
|
||||||
0x3fffe0,
|
|
||||||
0x1fffe2,
|
|
||||||
0x7fffed,
|
|
||||||
0x3fffe1,
|
|
||||||
0x7fffee,
|
|
||||||
0x7fffef,
|
|
||||||
0xfffea,
|
|
||||||
0x3fffe2,
|
|
||||||
0x3fffe3,
|
|
||||||
0x3fffe4,
|
|
||||||
0x7ffff0,
|
|
||||||
0x3fffe5,
|
|
||||||
0x3fffe6,
|
|
||||||
0x7ffff1,
|
|
||||||
0x3ffffe0,
|
|
||||||
0x3ffffe1,
|
|
||||||
0xfffeb,
|
|
||||||
0x7fff1,
|
|
||||||
0x3fffe7,
|
|
||||||
0x7ffff2,
|
|
||||||
0x3fffe8,
|
|
||||||
0x1ffffec,
|
|
||||||
0x3ffffe2,
|
|
||||||
0x3ffffe3,
|
|
||||||
0x3ffffe4,
|
|
||||||
0x7ffffde,
|
|
||||||
0x7ffffdf,
|
|
||||||
0x3ffffe5,
|
|
||||||
0xfffff1,
|
|
||||||
0x1ffffed,
|
|
||||||
0x7fff2,
|
|
||||||
0x1fffe3,
|
|
||||||
0x3ffffe6,
|
|
||||||
0x7ffffe0,
|
|
||||||
0x7ffffe1,
|
|
||||||
0x3ffffe7,
|
|
||||||
0x7ffffe2,
|
|
||||||
0xfffff2,
|
|
||||||
0x1fffe4,
|
|
||||||
0x1fffe5,
|
|
||||||
0x3ffffe8,
|
|
||||||
0x3ffffe9,
|
|
||||||
0xffffffd,
|
|
||||||
0x7ffffe3,
|
|
||||||
0x7ffffe4,
|
|
||||||
0x7ffffe5,
|
|
||||||
0xfffec,
|
|
||||||
0xfffff3,
|
|
||||||
0xfffed,
|
|
||||||
0x1fffe6,
|
|
||||||
0x3fffe9,
|
|
||||||
0x1fffe7,
|
|
||||||
0x1fffe8,
|
|
||||||
0x7ffff3,
|
|
||||||
0x3fffea,
|
|
||||||
0x3fffeb,
|
|
||||||
0x1ffffee,
|
|
||||||
0x1ffffef,
|
|
||||||
0xfffff4,
|
|
||||||
0xfffff5,
|
|
||||||
0x3ffffea,
|
|
||||||
0x7ffff4,
|
|
||||||
0x3ffffeb,
|
|
||||||
0x7ffffe6,
|
|
||||||
0x3ffffec,
|
|
||||||
0x3ffffed,
|
|
||||||
0x7ffffe7,
|
|
||||||
0x7ffffe8,
|
|
||||||
0x7ffffe9,
|
|
||||||
0x7ffffea,
|
|
||||||
0x7ffffeb,
|
|
||||||
0xffffffe,
|
|
||||||
0x7ffffec,
|
|
||||||
0x7ffffed,
|
|
||||||
0x7ffffee,
|
|
||||||
0x7ffffef,
|
|
||||||
0x7fffff0,
|
|
||||||
0x3ffffee,
|
|
||||||
}
|
|
||||||
|
|
||||||
var huffmanCodeLen = [256]uint8{
|
|
||||||
13, 23, 28, 28, 28, 28, 28, 28, 28, 24, 30, 28, 28, 30, 28, 28,
|
|
||||||
28, 28, 28, 28, 28, 28, 30, 28, 28, 28, 28, 28, 28, 28, 28, 28,
|
|
||||||
6, 10, 10, 12, 13, 6, 8, 11, 10, 10, 8, 11, 8, 6, 6, 6,
|
|
||||||
5, 5, 5, 6, 6, 6, 6, 6, 6, 6, 7, 8, 15, 6, 12, 10,
|
|
||||||
13, 6, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
|
|
||||||
7, 7, 7, 7, 7, 7, 7, 7, 8, 7, 8, 13, 19, 13, 14, 6,
|
|
||||||
15, 5, 6, 5, 6, 5, 6, 6, 6, 5, 7, 7, 6, 6, 6, 5,
|
|
||||||
6, 7, 6, 5, 5, 6, 7, 7, 7, 7, 7, 15, 11, 14, 13, 28,
|
|
||||||
20, 22, 20, 20, 22, 22, 22, 23, 22, 23, 23, 23, 23, 23, 24, 23,
|
|
||||||
24, 24, 22, 23, 24, 23, 23, 23, 23, 21, 22, 23, 22, 23, 23, 24,
|
|
||||||
22, 21, 20, 22, 22, 23, 23, 21, 23, 22, 22, 24, 21, 22, 23, 23,
|
|
||||||
21, 21, 22, 21, 23, 22, 23, 23, 20, 22, 22, 22, 23, 22, 22, 23,
|
|
||||||
26, 26, 20, 19, 22, 23, 22, 25, 26, 26, 26, 27, 27, 26, 24, 25,
|
|
||||||
19, 21, 26, 27, 27, 26, 27, 24, 21, 21, 26, 26, 28, 27, 27, 27,
|
|
||||||
20, 24, 20, 21, 22, 21, 21, 23, 22, 22, 25, 25, 24, 24, 26, 23,
|
|
||||||
26, 27, 26, 26, 27, 27, 27, 27, 27, 28, 27, 27, 27, 27, 27, 26,
|
|
||||||
}
|
|
429
vendor/golang.org/x/net/http2/http2.go
generated
vendored
429
vendor/golang.org/x/net/http2/http2.go
generated
vendored
@ -1,429 +0,0 @@
|
|||||||
// Copyright 2014 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 http2 implements the HTTP/2 protocol.
|
|
||||||
//
|
|
||||||
// This package is low-level and intended to be used directly by very
|
|
||||||
// few people. Most users will use it indirectly through the automatic
|
|
||||||
// use by the net/http package (from Go 1.6 and later).
|
|
||||||
// For use in earlier Go versions see ConfigureServer. (Transport support
|
|
||||||
// requires Go 1.6 or later)
|
|
||||||
//
|
|
||||||
// See https://http2.github.io/ for more information on HTTP/2.
|
|
||||||
//
|
|
||||||
// See https://http2.golang.org/ for a test server running this code.
|
|
||||||
package http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bufio"
|
|
||||||
"crypto/tls"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
VerboseLogs bool
|
|
||||||
logFrameWrites bool
|
|
||||||
logFrameReads bool
|
|
||||||
)
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
e := os.Getenv("GODEBUG")
|
|
||||||
if strings.Contains(e, "http2debug=1") {
|
|
||||||
VerboseLogs = true
|
|
||||||
}
|
|
||||||
if strings.Contains(e, "http2debug=2") {
|
|
||||||
VerboseLogs = true
|
|
||||||
logFrameWrites = true
|
|
||||||
logFrameReads = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const (
|
|
||||||
// ClientPreface is the string that must be sent by new
|
|
||||||
// connections from clients.
|
|
||||||
ClientPreface = "PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n"
|
|
||||||
|
|
||||||
// SETTINGS_MAX_FRAME_SIZE default
|
|
||||||
// http://http2.github.io/http2-spec/#rfc.section.6.5.2
|
|
||||||
initialMaxFrameSize = 16384
|
|
||||||
|
|
||||||
// NextProtoTLS is the NPN/ALPN protocol negotiated during
|
|
||||||
// HTTP/2's TLS setup.
|
|
||||||
NextProtoTLS = "h2"
|
|
||||||
|
|
||||||
// http://http2.github.io/http2-spec/#SettingValues
|
|
||||||
initialHeaderTableSize = 4096
|
|
||||||
|
|
||||||
initialWindowSize = 65535 // 6.9.2 Initial Flow Control Window Size
|
|
||||||
|
|
||||||
defaultMaxReadFrameSize = 1 << 20
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
clientPreface = []byte(ClientPreface)
|
|
||||||
)
|
|
||||||
|
|
||||||
type streamState int
|
|
||||||
|
|
||||||
const (
|
|
||||||
stateIdle streamState = iota
|
|
||||||
stateOpen
|
|
||||||
stateHalfClosedLocal
|
|
||||||
stateHalfClosedRemote
|
|
||||||
stateResvLocal
|
|
||||||
stateResvRemote
|
|
||||||
stateClosed
|
|
||||||
)
|
|
||||||
|
|
||||||
var stateName = [...]string{
|
|
||||||
stateIdle: "Idle",
|
|
||||||
stateOpen: "Open",
|
|
||||||
stateHalfClosedLocal: "HalfClosedLocal",
|
|
||||||
stateHalfClosedRemote: "HalfClosedRemote",
|
|
||||||
stateResvLocal: "ResvLocal",
|
|
||||||
stateResvRemote: "ResvRemote",
|
|
||||||
stateClosed: "Closed",
|
|
||||||
}
|
|
||||||
|
|
||||||
func (st streamState) String() string {
|
|
||||||
return stateName[st]
|
|
||||||
}
|
|
||||||
|
|
||||||
// Setting is a setting parameter: which setting it is, and its value.
|
|
||||||
type Setting struct {
|
|
||||||
// ID is which setting is being set.
|
|
||||||
// See http://http2.github.io/http2-spec/#SettingValues
|
|
||||||
ID SettingID
|
|
||||||
|
|
||||||
// Val is the value.
|
|
||||||
Val uint32
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s Setting) String() string {
|
|
||||||
return fmt.Sprintf("[%v = %d]", s.ID, s.Val)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Valid reports whether the setting is valid.
|
|
||||||
func (s Setting) Valid() error {
|
|
||||||
// Limits and error codes from 6.5.2 Defined SETTINGS Parameters
|
|
||||||
switch s.ID {
|
|
||||||
case SettingEnablePush:
|
|
||||||
if s.Val != 1 && s.Val != 0 {
|
|
||||||
return ConnectionError(ErrCodeProtocol)
|
|
||||||
}
|
|
||||||
case SettingInitialWindowSize:
|
|
||||||
if s.Val > 1<<31-1 {
|
|
||||||
return ConnectionError(ErrCodeFlowControl)
|
|
||||||
}
|
|
||||||
case SettingMaxFrameSize:
|
|
||||||
if s.Val < 16384 || s.Val > 1<<24-1 {
|
|
||||||
return ConnectionError(ErrCodeProtocol)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// A SettingID is an HTTP/2 setting as defined in
|
|
||||||
// http://http2.github.io/http2-spec/#iana-settings
|
|
||||||
type SettingID uint16
|
|
||||||
|
|
||||||
const (
|
|
||||||
SettingHeaderTableSize SettingID = 0x1
|
|
||||||
SettingEnablePush SettingID = 0x2
|
|
||||||
SettingMaxConcurrentStreams SettingID = 0x3
|
|
||||||
SettingInitialWindowSize SettingID = 0x4
|
|
||||||
SettingMaxFrameSize SettingID = 0x5
|
|
||||||
SettingMaxHeaderListSize SettingID = 0x6
|
|
||||||
)
|
|
||||||
|
|
||||||
var settingName = map[SettingID]string{
|
|
||||||
SettingHeaderTableSize: "HEADER_TABLE_SIZE",
|
|
||||||
SettingEnablePush: "ENABLE_PUSH",
|
|
||||||
SettingMaxConcurrentStreams: "MAX_CONCURRENT_STREAMS",
|
|
||||||
SettingInitialWindowSize: "INITIAL_WINDOW_SIZE",
|
|
||||||
SettingMaxFrameSize: "MAX_FRAME_SIZE",
|
|
||||||
SettingMaxHeaderListSize: "MAX_HEADER_LIST_SIZE",
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s SettingID) String() string {
|
|
||||||
if v, ok := settingName[s]; ok {
|
|
||||||
return v
|
|
||||||
}
|
|
||||||
return fmt.Sprintf("UNKNOWN_SETTING_%d", uint16(s))
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
errInvalidHeaderFieldName = errors.New("http2: invalid header field name")
|
|
||||||
errInvalidHeaderFieldValue = errors.New("http2: invalid header field value")
|
|
||||||
)
|
|
||||||
|
|
||||||
// validHeaderFieldName reports whether v is a valid header field name (key).
|
|
||||||
// RFC 7230 says:
|
|
||||||
// header-field = field-name ":" OWS field-value OWS
|
|
||||||
// field-name = token
|
|
||||||
// tchar = "!" / "#" / "$" / "%" / "&" / "'" / "*" / "+" / "-" / "." /
|
|
||||||
// "^" / "_" / "
|
|
||||||
// Further, http2 says:
|
|
||||||
// "Just as in HTTP/1.x, header field names are strings of ASCII
|
|
||||||
// characters that are compared in a case-insensitive
|
|
||||||
// fashion. However, header field names MUST be converted to
|
|
||||||
// lowercase prior to their encoding in HTTP/2. "
|
|
||||||
func validHeaderFieldName(v string) bool {
|
|
||||||
if len(v) == 0 {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
for _, r := range v {
|
|
||||||
if int(r) >= len(isTokenTable) || ('A' <= r && r <= 'Z') {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
if !isTokenTable[byte(r)] {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
// validHeaderFieldValue reports whether v is a valid header field value.
|
|
||||||
//
|
|
||||||
// RFC 7230 says:
|
|
||||||
// field-value = *( field-content / obs-fold )
|
|
||||||
// obj-fold = N/A to http2, and deprecated
|
|
||||||
// field-content = field-vchar [ 1*( SP / HTAB ) field-vchar ]
|
|
||||||
// field-vchar = VCHAR / obs-text
|
|
||||||
// obs-text = %x80-FF
|
|
||||||
// VCHAR = "any visible [USASCII] character"
|
|
||||||
//
|
|
||||||
// http2 further says: "Similarly, HTTP/2 allows header field values
|
|
||||||
// that are not valid. While most of the values that can be encoded
|
|
||||||
// will not alter header field parsing, carriage return (CR, ASCII
|
|
||||||
// 0xd), line feed (LF, ASCII 0xa), and the zero character (NUL, ASCII
|
|
||||||
// 0x0) might be exploited by an attacker if they are translated
|
|
||||||
// verbatim. Any request or response that contains a character not
|
|
||||||
// permitted in a header field value MUST be treated as malformed
|
|
||||||
// (Section 8.1.2.6). Valid characters are defined by the
|
|
||||||
// field-content ABNF rule in Section 3.2 of [RFC7230]."
|
|
||||||
//
|
|
||||||
// This function does not (yet?) properly handle the rejection of
|
|
||||||
// strings that begin or end with SP or HTAB.
|
|
||||||
func validHeaderFieldValue(v string) bool {
|
|
||||||
for i := 0; i < len(v); i++ {
|
|
||||||
if b := v[i]; b < ' ' && b != '\t' || b == 0x7f {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
var httpCodeStringCommon = map[int]string{} // n -> strconv.Itoa(n)
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
for i := 100; i <= 999; i++ {
|
|
||||||
if v := http.StatusText(i); v != "" {
|
|
||||||
httpCodeStringCommon[i] = strconv.Itoa(i)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func httpCodeString(code int) string {
|
|
||||||
if s, ok := httpCodeStringCommon[code]; ok {
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
return strconv.Itoa(code)
|
|
||||||
}
|
|
||||||
|
|
||||||
// from pkg io
|
|
||||||
type stringWriter interface {
|
|
||||||
WriteString(s string) (n int, err error)
|
|
||||||
}
|
|
||||||
|
|
||||||
// A gate lets two goroutines coordinate their activities.
|
|
||||||
type gate chan struct{}
|
|
||||||
|
|
||||||
func (g gate) Done() { g <- struct{}{} }
|
|
||||||
func (g gate) Wait() { <-g }
|
|
||||||
|
|
||||||
// A closeWaiter is like a sync.WaitGroup but only goes 1 to 0 (open to closed).
|
|
||||||
type closeWaiter chan struct{}
|
|
||||||
|
|
||||||
// Init makes a closeWaiter usable.
|
|
||||||
// It exists because so a closeWaiter value can be placed inside a
|
|
||||||
// larger struct and have the Mutex and Cond's memory in the same
|
|
||||||
// allocation.
|
|
||||||
func (cw *closeWaiter) Init() {
|
|
||||||
*cw = make(chan struct{})
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close marks the closeWaiter as closed and unblocks any waiters.
|
|
||||||
func (cw closeWaiter) Close() {
|
|
||||||
close(cw)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Wait waits for the closeWaiter to become closed.
|
|
||||||
func (cw closeWaiter) Wait() {
|
|
||||||
<-cw
|
|
||||||
}
|
|
||||||
|
|
||||||
// bufferedWriter is a buffered writer that writes to w.
|
|
||||||
// Its buffered writer is lazily allocated as needed, to minimize
|
|
||||||
// idle memory usage with many connections.
|
|
||||||
type bufferedWriter struct {
|
|
||||||
w io.Writer // immutable
|
|
||||||
bw *bufio.Writer // non-nil when data is buffered
|
|
||||||
}
|
|
||||||
|
|
||||||
func newBufferedWriter(w io.Writer) *bufferedWriter {
|
|
||||||
return &bufferedWriter{w: w}
|
|
||||||
}
|
|
||||||
|
|
||||||
var bufWriterPool = sync.Pool{
|
|
||||||
New: func() interface{} {
|
|
||||||
// TODO: pick something better? this is a bit under
|
|
||||||
// (3 x typical 1500 byte MTU) at least.
|
|
||||||
return bufio.NewWriterSize(nil, 4<<10)
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *bufferedWriter) Write(p []byte) (n int, err error) {
|
|
||||||
if w.bw == nil {
|
|
||||||
bw := bufWriterPool.Get().(*bufio.Writer)
|
|
||||||
bw.Reset(w.w)
|
|
||||||
w.bw = bw
|
|
||||||
}
|
|
||||||
return w.bw.Write(p)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *bufferedWriter) Flush() error {
|
|
||||||
bw := w.bw
|
|
||||||
if bw == nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
err := bw.Flush()
|
|
||||||
bw.Reset(nil)
|
|
||||||
bufWriterPool.Put(bw)
|
|
||||||
w.bw = nil
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func mustUint31(v int32) uint32 {
|
|
||||||
if v < 0 || v > 2147483647 {
|
|
||||||
panic("out of range")
|
|
||||||
}
|
|
||||||
return uint32(v)
|
|
||||||
}
|
|
||||||
|
|
||||||
// bodyAllowedForStatus reports whether a given response status code
|
|
||||||
// permits a body. See RFC2616, section 4.4.
|
|
||||||
func bodyAllowedForStatus(status int) bool {
|
|
||||||
switch {
|
|
||||||
case status >= 100 && status <= 199:
|
|
||||||
return false
|
|
||||||
case status == 204:
|
|
||||||
return false
|
|
||||||
case status == 304:
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
type httpError struct {
|
|
||||||
msg string
|
|
||||||
timeout bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e *httpError) Error() string { return e.msg }
|
|
||||||
func (e *httpError) Timeout() bool { return e.timeout }
|
|
||||||
func (e *httpError) Temporary() bool { return true }
|
|
||||||
|
|
||||||
var errTimeout error = &httpError{msg: "http2: timeout awaiting response headers", timeout: true}
|
|
||||||
|
|
||||||
var isTokenTable = [127]bool{
|
|
||||||
'!': true,
|
|
||||||
'#': true,
|
|
||||||
'$': true,
|
|
||||||
'%': true,
|
|
||||||
'&': true,
|
|
||||||
'\'': true,
|
|
||||||
'*': true,
|
|
||||||
'+': true,
|
|
||||||
'-': true,
|
|
||||||
'.': true,
|
|
||||||
'0': true,
|
|
||||||
'1': true,
|
|
||||||
'2': true,
|
|
||||||
'3': true,
|
|
||||||
'4': true,
|
|
||||||
'5': true,
|
|
||||||
'6': true,
|
|
||||||
'7': true,
|
|
||||||
'8': true,
|
|
||||||
'9': true,
|
|
||||||
'A': true,
|
|
||||||
'B': true,
|
|
||||||
'C': true,
|
|
||||||
'D': true,
|
|
||||||
'E': true,
|
|
||||||
'F': true,
|
|
||||||
'G': true,
|
|
||||||
'H': true,
|
|
||||||
'I': true,
|
|
||||||
'J': true,
|
|
||||||
'K': true,
|
|
||||||
'L': true,
|
|
||||||
'M': true,
|
|
||||||
'N': true,
|
|
||||||
'O': true,
|
|
||||||
'P': true,
|
|
||||||
'Q': true,
|
|
||||||
'R': true,
|
|
||||||
'S': true,
|
|
||||||
'T': true,
|
|
||||||
'U': true,
|
|
||||||
'W': true,
|
|
||||||
'V': true,
|
|
||||||
'X': true,
|
|
||||||
'Y': true,
|
|
||||||
'Z': true,
|
|
||||||
'^': true,
|
|
||||||
'_': true,
|
|
||||||
'`': true,
|
|
||||||
'a': true,
|
|
||||||
'b': true,
|
|
||||||
'c': true,
|
|
||||||
'd': true,
|
|
||||||
'e': true,
|
|
||||||
'f': true,
|
|
||||||
'g': true,
|
|
||||||
'h': true,
|
|
||||||
'i': true,
|
|
||||||
'j': true,
|
|
||||||
'k': true,
|
|
||||||
'l': true,
|
|
||||||
'm': true,
|
|
||||||
'n': true,
|
|
||||||
'o': true,
|
|
||||||
'p': true,
|
|
||||||
'q': true,
|
|
||||||
'r': true,
|
|
||||||
's': true,
|
|
||||||
't': true,
|
|
||||||
'u': true,
|
|
||||||
'v': true,
|
|
||||||
'w': true,
|
|
||||||
'x': true,
|
|
||||||
'y': true,
|
|
||||||
'z': true,
|
|
||||||
'|': true,
|
|
||||||
'~': true,
|
|
||||||
}
|
|
||||||
|
|
||||||
type connectionStater interface {
|
|
||||||
ConnectionState() tls.ConnectionState
|
|
||||||
}
|
|
174
vendor/golang.org/x/net/http2/http2_test.go
generated
vendored
174
vendor/golang.org/x/net/http2/http2_test.go
generated
vendored
@ -1,174 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"net/http"
|
|
||||||
"os/exec"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"golang.org/x/net/http2/hpack"
|
|
||||||
)
|
|
||||||
|
|
||||||
var knownFailing = flag.Bool("known_failing", false, "Run known-failing tests.")
|
|
||||||
|
|
||||||
func condSkipFailingTest(t *testing.T) {
|
|
||||||
if !*knownFailing {
|
|
||||||
t.Skip("Skipping known-failing test without --known_failing")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
DebugGoroutines = true
|
|
||||||
flag.BoolVar(&VerboseLogs, "verboseh2", false, "Verbose HTTP/2 debug logging")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSettingString(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
s Setting
|
|
||||||
want string
|
|
||||||
}{
|
|
||||||
{Setting{SettingMaxFrameSize, 123}, "[MAX_FRAME_SIZE = 123]"},
|
|
||||||
{Setting{1<<16 - 1, 123}, "[UNKNOWN_SETTING_65535 = 123]"},
|
|
||||||
}
|
|
||||||
for i, tt := range tests {
|
|
||||||
got := fmt.Sprint(tt.s)
|
|
||||||
if got != tt.want {
|
|
||||||
t.Errorf("%d. for %#v, string = %q; want %q", i, tt.s, got, tt.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type twriter struct {
|
|
||||||
t testing.TB
|
|
||||||
st *serverTester // optional
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w twriter) Write(p []byte) (n int, err error) {
|
|
||||||
if w.st != nil {
|
|
||||||
ps := string(p)
|
|
||||||
for _, phrase := range w.st.logFilter {
|
|
||||||
if strings.Contains(ps, phrase) {
|
|
||||||
return len(p), nil // no logging
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
w.t.Logf("%s", p)
|
|
||||||
return len(p), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// like encodeHeader, but don't add implicit psuedo headers.
|
|
||||||
func encodeHeaderNoImplicit(t *testing.T, headers ...string) []byte {
|
|
||||||
var buf bytes.Buffer
|
|
||||||
enc := hpack.NewEncoder(&buf)
|
|
||||||
for len(headers) > 0 {
|
|
||||||
k, v := headers[0], headers[1]
|
|
||||||
headers = headers[2:]
|
|
||||||
if err := enc.WriteField(hpack.HeaderField{Name: k, Value: v}); err != nil {
|
|
||||||
t.Fatalf("HPACK encoding error for %q/%q: %v", k, v, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return buf.Bytes()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Verify that curl has http2.
|
|
||||||
func requireCurl(t *testing.T) {
|
|
||||||
out, err := dockerLogs(curl(t, "--version"))
|
|
||||||
if err != nil {
|
|
||||||
t.Skipf("failed to determine curl features; skipping test")
|
|
||||||
}
|
|
||||||
if !strings.Contains(string(out), "HTTP2") {
|
|
||||||
t.Skip("curl doesn't support HTTP2; skipping test")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func curl(t *testing.T, args ...string) (container string) {
|
|
||||||
out, err := exec.Command("docker", append([]string{"run", "-d", "--net=host", "gohttp2/curl"}, args...)...).Output()
|
|
||||||
if err != nil {
|
|
||||||
t.Skipf("Failed to run curl in docker: %v, %s", err, out)
|
|
||||||
}
|
|
||||||
return strings.TrimSpace(string(out))
|
|
||||||
}
|
|
||||||
|
|
||||||
// Verify that h2load exists.
|
|
||||||
func requireH2load(t *testing.T) {
|
|
||||||
out, err := dockerLogs(h2load(t, "--version"))
|
|
||||||
if err != nil {
|
|
||||||
t.Skipf("failed to probe h2load; skipping test: %s", out)
|
|
||||||
}
|
|
||||||
if !strings.Contains(string(out), "h2load nghttp2/") {
|
|
||||||
t.Skipf("h2load not present; skipping test. (Output=%q)", out)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func h2load(t *testing.T, args ...string) (container string) {
|
|
||||||
out, err := exec.Command("docker", append([]string{"run", "-d", "--net=host", "--entrypoint=/usr/local/bin/h2load", "gohttp2/curl"}, args...)...).Output()
|
|
||||||
if err != nil {
|
|
||||||
t.Skipf("Failed to run h2load in docker: %v, %s", err, out)
|
|
||||||
}
|
|
||||||
return strings.TrimSpace(string(out))
|
|
||||||
}
|
|
||||||
|
|
||||||
type puppetCommand struct {
|
|
||||||
fn func(w http.ResponseWriter, r *http.Request)
|
|
||||||
done chan<- bool
|
|
||||||
}
|
|
||||||
|
|
||||||
type handlerPuppet struct {
|
|
||||||
ch chan puppetCommand
|
|
||||||
}
|
|
||||||
|
|
||||||
func newHandlerPuppet() *handlerPuppet {
|
|
||||||
return &handlerPuppet{
|
|
||||||
ch: make(chan puppetCommand),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *handlerPuppet) act(w http.ResponseWriter, r *http.Request) {
|
|
||||||
for cmd := range p.ch {
|
|
||||||
cmd.fn(w, r)
|
|
||||||
cmd.done <- true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *handlerPuppet) done() { close(p.ch) }
|
|
||||||
func (p *handlerPuppet) do(fn func(http.ResponseWriter, *http.Request)) {
|
|
||||||
done := make(chan bool)
|
|
||||||
p.ch <- puppetCommand{fn, done}
|
|
||||||
<-done
|
|
||||||
}
|
|
||||||
func dockerLogs(container string) ([]byte, error) {
|
|
||||||
out, err := exec.Command("docker", "wait", container).CombinedOutput()
|
|
||||||
if err != nil {
|
|
||||||
return out, err
|
|
||||||
}
|
|
||||||
exitStatus, err := strconv.Atoi(strings.TrimSpace(string(out)))
|
|
||||||
if err != nil {
|
|
||||||
return out, errors.New("unexpected exit status from docker wait")
|
|
||||||
}
|
|
||||||
out, err = exec.Command("docker", "logs", container).CombinedOutput()
|
|
||||||
exec.Command("docker", "rm", container).Run()
|
|
||||||
if err == nil && exitStatus != 0 {
|
|
||||||
err = fmt.Errorf("exit status %d: %s", exitStatus, out)
|
|
||||||
}
|
|
||||||
return out, err
|
|
||||||
}
|
|
||||||
|
|
||||||
func kill(container string) {
|
|
||||||
exec.Command("docker", "kill", container).Run()
|
|
||||||
exec.Command("docker", "rm", container).Run()
|
|
||||||
}
|
|
||||||
|
|
||||||
func cleanDate(res *http.Response) {
|
|
||||||
if d := res.Header["Date"]; len(d) == 1 {
|
|
||||||
d[0] = "XXX"
|
|
||||||
}
|
|
||||||
}
|
|
11
vendor/golang.org/x/net/http2/not_go15.go
generated
vendored
11
vendor/golang.org/x/net/http2/not_go15.go
generated
vendored
@ -1,11 +0,0 @@
|
|||||||
// Copyright 2015 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.
|
|
||||||
|
|
||||||
// +build !go1.5
|
|
||||||
|
|
||||||
package http2
|
|
||||||
|
|
||||||
import "net/http"
|
|
||||||
|
|
||||||
func requestCancel(req *http.Request) <-chan struct{} { return nil }
|
|
13
vendor/golang.org/x/net/http2/not_go16.go
generated
vendored
13
vendor/golang.org/x/net/http2/not_go16.go
generated
vendored
@ -1,13 +0,0 @@
|
|||||||
// Copyright 2015 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.
|
|
||||||
|
|
||||||
// +build !go1.6
|
|
||||||
|
|
||||||
package http2
|
|
||||||
|
|
||||||
import "net/http"
|
|
||||||
|
|
||||||
func configureTransport(t1 *http.Transport) (*Transport, error) {
|
|
||||||
return nil, errTransportVersion
|
|
||||||
}
|
|
147
vendor/golang.org/x/net/http2/pipe.go
generated
vendored
147
vendor/golang.org/x/net/http2/pipe.go
generated
vendored
@ -1,147 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"io"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
// pipe is a goroutine-safe io.Reader/io.Writer pair. It's like
|
|
||||||
// io.Pipe except there are no PipeReader/PipeWriter halves, and the
|
|
||||||
// underlying buffer is an interface. (io.Pipe is always unbuffered)
|
|
||||||
type pipe struct {
|
|
||||||
mu sync.Mutex
|
|
||||||
c sync.Cond // c.L lazily initialized to &p.mu
|
|
||||||
b pipeBuffer
|
|
||||||
err error // read error once empty. non-nil means closed.
|
|
||||||
breakErr error // immediate read error (caller doesn't see rest of b)
|
|
||||||
donec chan struct{} // closed on error
|
|
||||||
readFn func() // optional code to run in Read before error
|
|
||||||
}
|
|
||||||
|
|
||||||
type pipeBuffer interface {
|
|
||||||
Len() int
|
|
||||||
io.Writer
|
|
||||||
io.Reader
|
|
||||||
}
|
|
||||||
|
|
||||||
// Read waits until data is available and copies bytes
|
|
||||||
// from the buffer into p.
|
|
||||||
func (p *pipe) Read(d []byte) (n int, err error) {
|
|
||||||
p.mu.Lock()
|
|
||||||
defer p.mu.Unlock()
|
|
||||||
if p.c.L == nil {
|
|
||||||
p.c.L = &p.mu
|
|
||||||
}
|
|
||||||
for {
|
|
||||||
if p.breakErr != nil {
|
|
||||||
return 0, p.breakErr
|
|
||||||
}
|
|
||||||
if p.b.Len() > 0 {
|
|
||||||
return p.b.Read(d)
|
|
||||||
}
|
|
||||||
if p.err != nil {
|
|
||||||
if p.readFn != nil {
|
|
||||||
p.readFn() // e.g. copy trailers
|
|
||||||
p.readFn = nil // not sticky like p.err
|
|
||||||
}
|
|
||||||
return 0, p.err
|
|
||||||
}
|
|
||||||
p.c.Wait()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var errClosedPipeWrite = errors.New("write on closed buffer")
|
|
||||||
|
|
||||||
// Write copies bytes from p into the buffer and wakes a reader.
|
|
||||||
// It is an error to write more data than the buffer can hold.
|
|
||||||
func (p *pipe) Write(d []byte) (n int, err error) {
|
|
||||||
p.mu.Lock()
|
|
||||||
defer p.mu.Unlock()
|
|
||||||
if p.c.L == nil {
|
|
||||||
p.c.L = &p.mu
|
|
||||||
}
|
|
||||||
defer p.c.Signal()
|
|
||||||
if p.err != nil {
|
|
||||||
return 0, errClosedPipeWrite
|
|
||||||
}
|
|
||||||
return p.b.Write(d)
|
|
||||||
}
|
|
||||||
|
|
||||||
// CloseWithError causes the next Read (waking up a current blocked
|
|
||||||
// Read if needed) to return the provided err after all data has been
|
|
||||||
// read.
|
|
||||||
//
|
|
||||||
// The error must be non-nil.
|
|
||||||
func (p *pipe) CloseWithError(err error) { p.closeWithError(&p.err, err, nil) }
|
|
||||||
|
|
||||||
// BreakWithError causes the next Read (waking up a current blocked
|
|
||||||
// Read if needed) to return the provided err immediately, without
|
|
||||||
// waiting for unread data.
|
|
||||||
func (p *pipe) BreakWithError(err error) { p.closeWithError(&p.breakErr, err, nil) }
|
|
||||||
|
|
||||||
// closeWithErrorAndCode is like CloseWithError but also sets some code to run
|
|
||||||
// in the caller's goroutine before returning the error.
|
|
||||||
func (p *pipe) closeWithErrorAndCode(err error, fn func()) { p.closeWithError(&p.err, err, fn) }
|
|
||||||
|
|
||||||
func (p *pipe) closeWithError(dst *error, err error, fn func()) {
|
|
||||||
if err == nil {
|
|
||||||
panic("err must be non-nil")
|
|
||||||
}
|
|
||||||
p.mu.Lock()
|
|
||||||
defer p.mu.Unlock()
|
|
||||||
if p.c.L == nil {
|
|
||||||
p.c.L = &p.mu
|
|
||||||
}
|
|
||||||
defer p.c.Signal()
|
|
||||||
if *dst != nil {
|
|
||||||
// Already been done.
|
|
||||||
return
|
|
||||||
}
|
|
||||||
p.readFn = fn
|
|
||||||
*dst = err
|
|
||||||
p.closeDoneLocked()
|
|
||||||
}
|
|
||||||
|
|
||||||
// requires p.mu be held.
|
|
||||||
func (p *pipe) closeDoneLocked() {
|
|
||||||
if p.donec == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
// Close if unclosed. This isn't racy since we always
|
|
||||||
// hold p.mu while closing.
|
|
||||||
select {
|
|
||||||
case <-p.donec:
|
|
||||||
default:
|
|
||||||
close(p.donec)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Err returns the error (if any) first set by BreakWithError or CloseWithError.
|
|
||||||
func (p *pipe) Err() error {
|
|
||||||
p.mu.Lock()
|
|
||||||
defer p.mu.Unlock()
|
|
||||||
if p.breakErr != nil {
|
|
||||||
return p.breakErr
|
|
||||||
}
|
|
||||||
return p.err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Done returns a channel which is closed if and when this pipe is closed
|
|
||||||
// with CloseWithError.
|
|
||||||
func (p *pipe) Done() <-chan struct{} {
|
|
||||||
p.mu.Lock()
|
|
||||||
defer p.mu.Unlock()
|
|
||||||
if p.donec == nil {
|
|
||||||
p.donec = make(chan struct{})
|
|
||||||
if p.err != nil || p.breakErr != nil {
|
|
||||||
// Already hit an error.
|
|
||||||
p.closeDoneLocked()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return p.donec
|
|
||||||
}
|
|
109
vendor/golang.org/x/net/http2/pipe_test.go
generated
vendored
109
vendor/golang.org/x/net/http2/pipe_test.go
generated
vendored
@ -1,109 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"io"
|
|
||||||
"io/ioutil"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestPipeClose(t *testing.T) {
|
|
||||||
var p pipe
|
|
||||||
p.b = new(bytes.Buffer)
|
|
||||||
a := errors.New("a")
|
|
||||||
b := errors.New("b")
|
|
||||||
p.CloseWithError(a)
|
|
||||||
p.CloseWithError(b)
|
|
||||||
_, err := p.Read(make([]byte, 1))
|
|
||||||
if err != a {
|
|
||||||
t.Errorf("err = %v want %v", err, a)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPipeDoneChan(t *testing.T) {
|
|
||||||
var p pipe
|
|
||||||
done := p.Done()
|
|
||||||
select {
|
|
||||||
case <-done:
|
|
||||||
t.Fatal("done too soon")
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
p.CloseWithError(io.EOF)
|
|
||||||
select {
|
|
||||||
case <-done:
|
|
||||||
default:
|
|
||||||
t.Fatal("should be done")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPipeDoneChan_ErrFirst(t *testing.T) {
|
|
||||||
var p pipe
|
|
||||||
p.CloseWithError(io.EOF)
|
|
||||||
done := p.Done()
|
|
||||||
select {
|
|
||||||
case <-done:
|
|
||||||
default:
|
|
||||||
t.Fatal("should be done")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPipeDoneChan_Break(t *testing.T) {
|
|
||||||
var p pipe
|
|
||||||
done := p.Done()
|
|
||||||
select {
|
|
||||||
case <-done:
|
|
||||||
t.Fatal("done too soon")
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
p.BreakWithError(io.EOF)
|
|
||||||
select {
|
|
||||||
case <-done:
|
|
||||||
default:
|
|
||||||
t.Fatal("should be done")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPipeDoneChan_Break_ErrFirst(t *testing.T) {
|
|
||||||
var p pipe
|
|
||||||
p.BreakWithError(io.EOF)
|
|
||||||
done := p.Done()
|
|
||||||
select {
|
|
||||||
case <-done:
|
|
||||||
default:
|
|
||||||
t.Fatal("should be done")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPipeCloseWithError(t *testing.T) {
|
|
||||||
p := &pipe{b: new(bytes.Buffer)}
|
|
||||||
const body = "foo"
|
|
||||||
io.WriteString(p, body)
|
|
||||||
a := errors.New("test error")
|
|
||||||
p.CloseWithError(a)
|
|
||||||
all, err := ioutil.ReadAll(p)
|
|
||||||
if string(all) != body {
|
|
||||||
t.Errorf("read bytes = %q; want %q", all, body)
|
|
||||||
}
|
|
||||||
if err != a {
|
|
||||||
t.Logf("read error = %v, %v", err, a)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPipeBreakWithError(t *testing.T) {
|
|
||||||
p := &pipe{b: new(bytes.Buffer)}
|
|
||||||
io.WriteString(p, "foo")
|
|
||||||
a := errors.New("test err")
|
|
||||||
p.BreakWithError(a)
|
|
||||||
all, err := ioutil.ReadAll(p)
|
|
||||||
if string(all) != "" {
|
|
||||||
t.Errorf("read bytes = %q; want empty string", all)
|
|
||||||
}
|
|
||||||
if err != a {
|
|
||||||
t.Logf("read error = %v, %v", err, a)
|
|
||||||
}
|
|
||||||
}
|
|
118
vendor/golang.org/x/net/http2/priority_test.go
generated
vendored
118
vendor/golang.org/x/net/http2/priority_test.go
generated
vendored
@ -1,118 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestPriority(t *testing.T) {
|
|
||||||
// A -> B
|
|
||||||
// move A's parent to B
|
|
||||||
streams := make(map[uint32]*stream)
|
|
||||||
a := &stream{
|
|
||||||
parent: nil,
|
|
||||||
weight: 16,
|
|
||||||
}
|
|
||||||
streams[1] = a
|
|
||||||
b := &stream{
|
|
||||||
parent: a,
|
|
||||||
weight: 16,
|
|
||||||
}
|
|
||||||
streams[2] = b
|
|
||||||
adjustStreamPriority(streams, 1, PriorityParam{
|
|
||||||
Weight: 20,
|
|
||||||
StreamDep: 2,
|
|
||||||
})
|
|
||||||
if a.parent != b {
|
|
||||||
t.Errorf("Expected A's parent to be B")
|
|
||||||
}
|
|
||||||
if a.weight != 20 {
|
|
||||||
t.Errorf("Expected A's weight to be 20; got %d", a.weight)
|
|
||||||
}
|
|
||||||
if b.parent != nil {
|
|
||||||
t.Errorf("Expected B to have no parent")
|
|
||||||
}
|
|
||||||
if b.weight != 16 {
|
|
||||||
t.Errorf("Expected B's weight to be 16; got %d", b.weight)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPriorityExclusiveZero(t *testing.T) {
|
|
||||||
// A B and C are all children of the 0 stream.
|
|
||||||
// Exclusive reprioritization to any of the streams
|
|
||||||
// should bring the rest of the streams under the
|
|
||||||
// reprioritized stream
|
|
||||||
streams := make(map[uint32]*stream)
|
|
||||||
a := &stream{
|
|
||||||
parent: nil,
|
|
||||||
weight: 16,
|
|
||||||
}
|
|
||||||
streams[1] = a
|
|
||||||
b := &stream{
|
|
||||||
parent: nil,
|
|
||||||
weight: 16,
|
|
||||||
}
|
|
||||||
streams[2] = b
|
|
||||||
c := &stream{
|
|
||||||
parent: nil,
|
|
||||||
weight: 16,
|
|
||||||
}
|
|
||||||
streams[3] = c
|
|
||||||
adjustStreamPriority(streams, 3, PriorityParam{
|
|
||||||
Weight: 20,
|
|
||||||
StreamDep: 0,
|
|
||||||
Exclusive: true,
|
|
||||||
})
|
|
||||||
if a.parent != c {
|
|
||||||
t.Errorf("Expected A's parent to be C")
|
|
||||||
}
|
|
||||||
if a.weight != 16 {
|
|
||||||
t.Errorf("Expected A's weight to be 16; got %d", a.weight)
|
|
||||||
}
|
|
||||||
if b.parent != c {
|
|
||||||
t.Errorf("Expected B's parent to be C")
|
|
||||||
}
|
|
||||||
if b.weight != 16 {
|
|
||||||
t.Errorf("Expected B's weight to be 16; got %d", b.weight)
|
|
||||||
}
|
|
||||||
if c.parent != nil {
|
|
||||||
t.Errorf("Expected C to have no parent")
|
|
||||||
}
|
|
||||||
if c.weight != 20 {
|
|
||||||
t.Errorf("Expected C's weight to be 20; got %d", b.weight)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPriorityOwnParent(t *testing.T) {
|
|
||||||
streams := make(map[uint32]*stream)
|
|
||||||
a := &stream{
|
|
||||||
parent: nil,
|
|
||||||
weight: 16,
|
|
||||||
}
|
|
||||||
streams[1] = a
|
|
||||||
b := &stream{
|
|
||||||
parent: a,
|
|
||||||
weight: 16,
|
|
||||||
}
|
|
||||||
streams[2] = b
|
|
||||||
adjustStreamPriority(streams, 1, PriorityParam{
|
|
||||||
Weight: 20,
|
|
||||||
StreamDep: 1,
|
|
||||||
})
|
|
||||||
if a.parent != nil {
|
|
||||||
t.Errorf("Expected A's parent to be nil")
|
|
||||||
}
|
|
||||||
if a.weight != 20 {
|
|
||||||
t.Errorf("Expected A's weight to be 20; got %d", a.weight)
|
|
||||||
}
|
|
||||||
if b.parent != a {
|
|
||||||
t.Errorf("Expected B's parent to be A")
|
|
||||||
}
|
|
||||||
if b.weight != 16 {
|
|
||||||
t.Errorf("Expected B's weight to be 16; got %d", b.weight)
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
2149
vendor/golang.org/x/net/http2/server.go
generated
vendored
2149
vendor/golang.org/x/net/http2/server.go
generated
vendored
File diff suppressed because it is too large
Load Diff
3102
vendor/golang.org/x/net/http2/server_test.go
generated
vendored
3102
vendor/golang.org/x/net/http2/server_test.go
generated
vendored
File diff suppressed because it is too large
Load Diff
5021
vendor/golang.org/x/net/http2/testdata/draft-ietf-httpbis-http2.xml
generated
vendored
5021
vendor/golang.org/x/net/http2/testdata/draft-ietf-httpbis-http2.xml
generated
vendored
File diff suppressed because it is too large
Load Diff
1666
vendor/golang.org/x/net/http2/transport.go
generated
vendored
1666
vendor/golang.org/x/net/http2/transport.go
generated
vendored
File diff suppressed because it is too large
Load Diff
1740
vendor/golang.org/x/net/http2/transport_test.go
generated
vendored
1740
vendor/golang.org/x/net/http2/transport_test.go
generated
vendored
File diff suppressed because it is too large
Load Diff
263
vendor/golang.org/x/net/http2/write.go
generated
vendored
263
vendor/golang.org/x/net/http2/write.go
generated
vendored
@ -1,263 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"fmt"
|
|
||||||
"log"
|
|
||||||
"net/http"
|
|
||||||
"sort"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"golang.org/x/net/http2/hpack"
|
|
||||||
)
|
|
||||||
|
|
||||||
// writeFramer is implemented by any type that is used to write frames.
|
|
||||||
type writeFramer interface {
|
|
||||||
writeFrame(writeContext) error
|
|
||||||
}
|
|
||||||
|
|
||||||
// writeContext is the interface needed by the various frame writer
|
|
||||||
// types below. All the writeFrame methods below are scheduled via the
|
|
||||||
// frame writing scheduler (see writeScheduler in writesched.go).
|
|
||||||
//
|
|
||||||
// This interface is implemented by *serverConn.
|
|
||||||
//
|
|
||||||
// TODO: decide whether to a) use this in the client code (which didn't
|
|
||||||
// end up using this yet, because it has a simpler design, not
|
|
||||||
// currently implementing priorities), or b) delete this and
|
|
||||||
// make the server code a bit more concrete.
|
|
||||||
type writeContext interface {
|
|
||||||
Framer() *Framer
|
|
||||||
Flush() error
|
|
||||||
CloseConn() error
|
|
||||||
// HeaderEncoder returns an HPACK encoder that writes to the
|
|
||||||
// returned buffer.
|
|
||||||
HeaderEncoder() (*hpack.Encoder, *bytes.Buffer)
|
|
||||||
}
|
|
||||||
|
|
||||||
// endsStream reports whether the given frame writer w will locally
|
|
||||||
// close the stream.
|
|
||||||
func endsStream(w writeFramer) bool {
|
|
||||||
switch v := w.(type) {
|
|
||||||
case *writeData:
|
|
||||||
return v.endStream
|
|
||||||
case *writeResHeaders:
|
|
||||||
return v.endStream
|
|
||||||
case nil:
|
|
||||||
// This can only happen if the caller reuses w after it's
|
|
||||||
// been intentionally nil'ed out to prevent use. Keep this
|
|
||||||
// here to catch future refactoring breaking it.
|
|
||||||
panic("endsStream called on nil writeFramer")
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
type flushFrameWriter struct{}
|
|
||||||
|
|
||||||
func (flushFrameWriter) writeFrame(ctx writeContext) error {
|
|
||||||
return ctx.Flush()
|
|
||||||
}
|
|
||||||
|
|
||||||
type writeSettings []Setting
|
|
||||||
|
|
||||||
func (s writeSettings) writeFrame(ctx writeContext) error {
|
|
||||||
return ctx.Framer().WriteSettings([]Setting(s)...)
|
|
||||||
}
|
|
||||||
|
|
||||||
type writeGoAway struct {
|
|
||||||
maxStreamID uint32
|
|
||||||
code ErrCode
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *writeGoAway) writeFrame(ctx writeContext) error {
|
|
||||||
err := ctx.Framer().WriteGoAway(p.maxStreamID, p.code, nil)
|
|
||||||
if p.code != 0 {
|
|
||||||
ctx.Flush() // ignore error: we're hanging up on them anyway
|
|
||||||
time.Sleep(50 * time.Millisecond)
|
|
||||||
ctx.CloseConn()
|
|
||||||
}
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
type writeData struct {
|
|
||||||
streamID uint32
|
|
||||||
p []byte
|
|
||||||
endStream bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *writeData) String() string {
|
|
||||||
return fmt.Sprintf("writeData(stream=%d, p=%d, endStream=%v)", w.streamID, len(w.p), w.endStream)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *writeData) writeFrame(ctx writeContext) error {
|
|
||||||
return ctx.Framer().WriteData(w.streamID, w.endStream, w.p)
|
|
||||||
}
|
|
||||||
|
|
||||||
// handlerPanicRST is the message sent from handler goroutines when
|
|
||||||
// the handler panics.
|
|
||||||
type handlerPanicRST struct {
|
|
||||||
StreamID uint32
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hp handlerPanicRST) writeFrame(ctx writeContext) error {
|
|
||||||
return ctx.Framer().WriteRSTStream(hp.StreamID, ErrCodeInternal)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (se StreamError) writeFrame(ctx writeContext) error {
|
|
||||||
return ctx.Framer().WriteRSTStream(se.StreamID, se.Code)
|
|
||||||
}
|
|
||||||
|
|
||||||
type writePingAck struct{ pf *PingFrame }
|
|
||||||
|
|
||||||
func (w writePingAck) writeFrame(ctx writeContext) error {
|
|
||||||
return ctx.Framer().WritePing(true, w.pf.Data)
|
|
||||||
}
|
|
||||||
|
|
||||||
type writeSettingsAck struct{}
|
|
||||||
|
|
||||||
func (writeSettingsAck) writeFrame(ctx writeContext) error {
|
|
||||||
return ctx.Framer().WriteSettingsAck()
|
|
||||||
}
|
|
||||||
|
|
||||||
// writeResHeaders is a request to write a HEADERS and 0+ CONTINUATION frames
|
|
||||||
// for HTTP response headers or trailers from a server handler.
|
|
||||||
type writeResHeaders struct {
|
|
||||||
streamID uint32
|
|
||||||
httpResCode int // 0 means no ":status" line
|
|
||||||
h http.Header // may be nil
|
|
||||||
trailers []string // if non-nil, which keys of h to write. nil means all.
|
|
||||||
endStream bool
|
|
||||||
|
|
||||||
date string
|
|
||||||
contentType string
|
|
||||||
contentLength string
|
|
||||||
}
|
|
||||||
|
|
||||||
func encKV(enc *hpack.Encoder, k, v string) {
|
|
||||||
if VerboseLogs {
|
|
||||||
log.Printf("http2: server encoding header %q = %q", k, v)
|
|
||||||
}
|
|
||||||
enc.WriteField(hpack.HeaderField{Name: k, Value: v})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *writeResHeaders) writeFrame(ctx writeContext) error {
|
|
||||||
enc, buf := ctx.HeaderEncoder()
|
|
||||||
buf.Reset()
|
|
||||||
|
|
||||||
if w.httpResCode != 0 {
|
|
||||||
encKV(enc, ":status", httpCodeString(w.httpResCode))
|
|
||||||
}
|
|
||||||
|
|
||||||
encodeHeaders(enc, w.h, w.trailers)
|
|
||||||
|
|
||||||
if w.contentType != "" {
|
|
||||||
encKV(enc, "content-type", w.contentType)
|
|
||||||
}
|
|
||||||
if w.contentLength != "" {
|
|
||||||
encKV(enc, "content-length", w.contentLength)
|
|
||||||
}
|
|
||||||
if w.date != "" {
|
|
||||||
encKV(enc, "date", w.date)
|
|
||||||
}
|
|
||||||
|
|
||||||
headerBlock := buf.Bytes()
|
|
||||||
if len(headerBlock) == 0 && w.trailers == nil {
|
|
||||||
panic("unexpected empty hpack")
|
|
||||||
}
|
|
||||||
|
|
||||||
// For now we're lazy and just pick the minimum MAX_FRAME_SIZE
|
|
||||||
// that all peers must support (16KB). Later we could care
|
|
||||||
// more and send larger frames if the peer advertised it, but
|
|
||||||
// there's little point. Most headers are small anyway (so we
|
|
||||||
// generally won't have CONTINUATION frames), and extra frames
|
|
||||||
// only waste 9 bytes anyway.
|
|
||||||
const maxFrameSize = 16384
|
|
||||||
|
|
||||||
first := true
|
|
||||||
for len(headerBlock) > 0 {
|
|
||||||
frag := headerBlock
|
|
||||||
if len(frag) > maxFrameSize {
|
|
||||||
frag = frag[:maxFrameSize]
|
|
||||||
}
|
|
||||||
headerBlock = headerBlock[len(frag):]
|
|
||||||
endHeaders := len(headerBlock) == 0
|
|
||||||
var err error
|
|
||||||
if first {
|
|
||||||
first = false
|
|
||||||
err = ctx.Framer().WriteHeaders(HeadersFrameParam{
|
|
||||||
StreamID: w.streamID,
|
|
||||||
BlockFragment: frag,
|
|
||||||
EndStream: w.endStream,
|
|
||||||
EndHeaders: endHeaders,
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
err = ctx.Framer().WriteContinuation(w.streamID, endHeaders, frag)
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type write100ContinueHeadersFrame struct {
|
|
||||||
streamID uint32
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w write100ContinueHeadersFrame) writeFrame(ctx writeContext) error {
|
|
||||||
enc, buf := ctx.HeaderEncoder()
|
|
||||||
buf.Reset()
|
|
||||||
encKV(enc, ":status", "100")
|
|
||||||
return ctx.Framer().WriteHeaders(HeadersFrameParam{
|
|
||||||
StreamID: w.streamID,
|
|
||||||
BlockFragment: buf.Bytes(),
|
|
||||||
EndStream: false,
|
|
||||||
EndHeaders: true,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
type writeWindowUpdate struct {
|
|
||||||
streamID uint32 // or 0 for conn-level
|
|
||||||
n uint32
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wu writeWindowUpdate) writeFrame(ctx writeContext) error {
|
|
||||||
return ctx.Framer().WriteWindowUpdate(wu.streamID, wu.n)
|
|
||||||
}
|
|
||||||
|
|
||||||
func encodeHeaders(enc *hpack.Encoder, h http.Header, keys []string) {
|
|
||||||
// TODO: garbage. pool sorters like http1? hot path for 1 key?
|
|
||||||
if keys == nil {
|
|
||||||
keys = make([]string, 0, len(h))
|
|
||||||
for k := range h {
|
|
||||||
keys = append(keys, k)
|
|
||||||
}
|
|
||||||
sort.Strings(keys)
|
|
||||||
}
|
|
||||||
for _, k := range keys {
|
|
||||||
vv := h[k]
|
|
||||||
k = lowerHeader(k)
|
|
||||||
if !validHeaderFieldName(k) {
|
|
||||||
// TODO: return an error? golang.org/issue/14048
|
|
||||||
// For now just omit it.
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
isTE := k == "transfer-encoding"
|
|
||||||
for _, v := range vv {
|
|
||||||
if !validHeaderFieldValue(v) {
|
|
||||||
// TODO: return an error? golang.org/issue/14048
|
|
||||||
// For now just omit it.
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
// TODO: more of "8.1.2.2 Connection-Specific Header Fields"
|
|
||||||
if isTE && v != "trailers" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
encKV(enc, k, v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
283
vendor/golang.org/x/net/http2/writesched.go
generated
vendored
283
vendor/golang.org/x/net/http2/writesched.go
generated
vendored
@ -1,283 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import "fmt"
|
|
||||||
|
|
||||||
// frameWriteMsg is a request to write a frame.
|
|
||||||
type frameWriteMsg struct {
|
|
||||||
// write is the interface value that does the writing, once the
|
|
||||||
// writeScheduler (below) has decided to select this frame
|
|
||||||
// to write. The write functions are all defined in write.go.
|
|
||||||
write writeFramer
|
|
||||||
|
|
||||||
stream *stream // used for prioritization. nil for non-stream frames.
|
|
||||||
|
|
||||||
// done, if non-nil, must be a buffered channel with space for
|
|
||||||
// 1 message and is sent the return value from write (or an
|
|
||||||
// earlier error) when the frame has been written.
|
|
||||||
done chan error
|
|
||||||
}
|
|
||||||
|
|
||||||
// for debugging only:
|
|
||||||
func (wm frameWriteMsg) String() string {
|
|
||||||
var streamID uint32
|
|
||||||
if wm.stream != nil {
|
|
||||||
streamID = wm.stream.id
|
|
||||||
}
|
|
||||||
var des string
|
|
||||||
if s, ok := wm.write.(fmt.Stringer); ok {
|
|
||||||
des = s.String()
|
|
||||||
} else {
|
|
||||||
des = fmt.Sprintf("%T", wm.write)
|
|
||||||
}
|
|
||||||
return fmt.Sprintf("[frameWriteMsg stream=%d, ch=%v, type: %v]", streamID, wm.done != nil, des)
|
|
||||||
}
|
|
||||||
|
|
||||||
// writeScheduler tracks pending frames to write, priorities, and decides
|
|
||||||
// the next one to use. It is not thread-safe.
|
|
||||||
type writeScheduler struct {
|
|
||||||
// zero are frames not associated with a specific stream.
|
|
||||||
// They're sent before any stream-specific freams.
|
|
||||||
zero writeQueue
|
|
||||||
|
|
||||||
// maxFrameSize is the maximum size of a DATA frame
|
|
||||||
// we'll write. Must be non-zero and between 16K-16M.
|
|
||||||
maxFrameSize uint32
|
|
||||||
|
|
||||||
// sq contains the stream-specific queues, keyed by stream ID.
|
|
||||||
// when a stream is idle, it's deleted from the map.
|
|
||||||
sq map[uint32]*writeQueue
|
|
||||||
|
|
||||||
// canSend is a slice of memory that's reused between frame
|
|
||||||
// scheduling decisions to hold the list of writeQueues (from sq)
|
|
||||||
// which have enough flow control data to send. After canSend is
|
|
||||||
// built, the best is selected.
|
|
||||||
canSend []*writeQueue
|
|
||||||
|
|
||||||
// pool of empty queues for reuse.
|
|
||||||
queuePool []*writeQueue
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ws *writeScheduler) putEmptyQueue(q *writeQueue) {
|
|
||||||
if len(q.s) != 0 {
|
|
||||||
panic("queue must be empty")
|
|
||||||
}
|
|
||||||
ws.queuePool = append(ws.queuePool, q)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ws *writeScheduler) getEmptyQueue() *writeQueue {
|
|
||||||
ln := len(ws.queuePool)
|
|
||||||
if ln == 0 {
|
|
||||||
return new(writeQueue)
|
|
||||||
}
|
|
||||||
q := ws.queuePool[ln-1]
|
|
||||||
ws.queuePool = ws.queuePool[:ln-1]
|
|
||||||
return q
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ws *writeScheduler) empty() bool { return ws.zero.empty() && len(ws.sq) == 0 }
|
|
||||||
|
|
||||||
func (ws *writeScheduler) add(wm frameWriteMsg) {
|
|
||||||
st := wm.stream
|
|
||||||
if st == nil {
|
|
||||||
ws.zero.push(wm)
|
|
||||||
} else {
|
|
||||||
ws.streamQueue(st.id).push(wm)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ws *writeScheduler) streamQueue(streamID uint32) *writeQueue {
|
|
||||||
if q, ok := ws.sq[streamID]; ok {
|
|
||||||
return q
|
|
||||||
}
|
|
||||||
if ws.sq == nil {
|
|
||||||
ws.sq = make(map[uint32]*writeQueue)
|
|
||||||
}
|
|
||||||
q := ws.getEmptyQueue()
|
|
||||||
ws.sq[streamID] = q
|
|
||||||
return q
|
|
||||||
}
|
|
||||||
|
|
||||||
// take returns the most important frame to write and removes it from the scheduler.
|
|
||||||
// It is illegal to call this if the scheduler is empty or if there are no connection-level
|
|
||||||
// flow control bytes available.
|
|
||||||
func (ws *writeScheduler) take() (wm frameWriteMsg, ok bool) {
|
|
||||||
if ws.maxFrameSize == 0 {
|
|
||||||
panic("internal error: ws.maxFrameSize not initialized or invalid")
|
|
||||||
}
|
|
||||||
|
|
||||||
// If there any frames not associated with streams, prefer those first.
|
|
||||||
// These are usually SETTINGS, etc.
|
|
||||||
if !ws.zero.empty() {
|
|
||||||
return ws.zero.shift(), true
|
|
||||||
}
|
|
||||||
if len(ws.sq) == 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Next, prioritize frames on streams that aren't DATA frames (no cost).
|
|
||||||
for id, q := range ws.sq {
|
|
||||||
if q.firstIsNoCost() {
|
|
||||||
return ws.takeFrom(id, q)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Now, all that remains are DATA frames with non-zero bytes to
|
|
||||||
// send. So pick the best one.
|
|
||||||
if len(ws.canSend) != 0 {
|
|
||||||
panic("should be empty")
|
|
||||||
}
|
|
||||||
for _, q := range ws.sq {
|
|
||||||
if n := ws.streamWritableBytes(q); n > 0 {
|
|
||||||
ws.canSend = append(ws.canSend, q)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if len(ws.canSend) == 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer ws.zeroCanSend()
|
|
||||||
|
|
||||||
// TODO: find the best queue
|
|
||||||
q := ws.canSend[0]
|
|
||||||
|
|
||||||
return ws.takeFrom(q.streamID(), q)
|
|
||||||
}
|
|
||||||
|
|
||||||
// zeroCanSend is defered from take.
|
|
||||||
func (ws *writeScheduler) zeroCanSend() {
|
|
||||||
for i := range ws.canSend {
|
|
||||||
ws.canSend[i] = nil
|
|
||||||
}
|
|
||||||
ws.canSend = ws.canSend[:0]
|
|
||||||
}
|
|
||||||
|
|
||||||
// streamWritableBytes returns the number of DATA bytes we could write
|
|
||||||
// from the given queue's stream, if this stream/queue were
|
|
||||||
// selected. It is an error to call this if q's head isn't a
|
|
||||||
// *writeData.
|
|
||||||
func (ws *writeScheduler) streamWritableBytes(q *writeQueue) int32 {
|
|
||||||
wm := q.head()
|
|
||||||
ret := wm.stream.flow.available() // max we can write
|
|
||||||
if ret == 0 {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
if int32(ws.maxFrameSize) < ret {
|
|
||||||
ret = int32(ws.maxFrameSize)
|
|
||||||
}
|
|
||||||
if ret == 0 {
|
|
||||||
panic("internal error: ws.maxFrameSize not initialized or invalid")
|
|
||||||
}
|
|
||||||
wd := wm.write.(*writeData)
|
|
||||||
if len(wd.p) < int(ret) {
|
|
||||||
ret = int32(len(wd.p))
|
|
||||||
}
|
|
||||||
return ret
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ws *writeScheduler) takeFrom(id uint32, q *writeQueue) (wm frameWriteMsg, ok bool) {
|
|
||||||
wm = q.head()
|
|
||||||
// If the first item in this queue costs flow control tokens
|
|
||||||
// and we don't have enough, write as much as we can.
|
|
||||||
if wd, ok := wm.write.(*writeData); ok && len(wd.p) > 0 {
|
|
||||||
allowed := wm.stream.flow.available() // max we can write
|
|
||||||
if allowed == 0 {
|
|
||||||
// No quota available. Caller can try the next stream.
|
|
||||||
return frameWriteMsg{}, false
|
|
||||||
}
|
|
||||||
if int32(ws.maxFrameSize) < allowed {
|
|
||||||
allowed = int32(ws.maxFrameSize)
|
|
||||||
}
|
|
||||||
// TODO: further restrict the allowed size, because even if
|
|
||||||
// the peer says it's okay to write 16MB data frames, we might
|
|
||||||
// want to write smaller ones to properly weight competing
|
|
||||||
// streams' priorities.
|
|
||||||
|
|
||||||
if len(wd.p) > int(allowed) {
|
|
||||||
wm.stream.flow.take(allowed)
|
|
||||||
chunk := wd.p[:allowed]
|
|
||||||
wd.p = wd.p[allowed:]
|
|
||||||
// Make up a new write message of a valid size, rather
|
|
||||||
// than shifting one off the queue.
|
|
||||||
return frameWriteMsg{
|
|
||||||
stream: wm.stream,
|
|
||||||
write: &writeData{
|
|
||||||
streamID: wd.streamID,
|
|
||||||
p: chunk,
|
|
||||||
// even if the original had endStream set, there
|
|
||||||
// arebytes remaining because len(wd.p) > allowed,
|
|
||||||
// so we know endStream is false:
|
|
||||||
endStream: false,
|
|
||||||
},
|
|
||||||
// our caller is blocking on the final DATA frame, not
|
|
||||||
// these intermediates, so no need to wait:
|
|
||||||
done: nil,
|
|
||||||
}, true
|
|
||||||
}
|
|
||||||
wm.stream.flow.take(int32(len(wd.p)))
|
|
||||||
}
|
|
||||||
|
|
||||||
q.shift()
|
|
||||||
if q.empty() {
|
|
||||||
ws.putEmptyQueue(q)
|
|
||||||
delete(ws.sq, id)
|
|
||||||
}
|
|
||||||
return wm, true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ws *writeScheduler) forgetStream(id uint32) {
|
|
||||||
q, ok := ws.sq[id]
|
|
||||||
if !ok {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
delete(ws.sq, id)
|
|
||||||
|
|
||||||
// But keep it for others later.
|
|
||||||
for i := range q.s {
|
|
||||||
q.s[i] = frameWriteMsg{}
|
|
||||||
}
|
|
||||||
q.s = q.s[:0]
|
|
||||||
ws.putEmptyQueue(q)
|
|
||||||
}
|
|
||||||
|
|
||||||
type writeQueue struct {
|
|
||||||
s []frameWriteMsg
|
|
||||||
}
|
|
||||||
|
|
||||||
// streamID returns the stream ID for a non-empty stream-specific queue.
|
|
||||||
func (q *writeQueue) streamID() uint32 { return q.s[0].stream.id }
|
|
||||||
|
|
||||||
func (q *writeQueue) empty() bool { return len(q.s) == 0 }
|
|
||||||
|
|
||||||
func (q *writeQueue) push(wm frameWriteMsg) {
|
|
||||||
q.s = append(q.s, wm)
|
|
||||||
}
|
|
||||||
|
|
||||||
// head returns the next item that would be removed by shift.
|
|
||||||
func (q *writeQueue) head() frameWriteMsg {
|
|
||||||
if len(q.s) == 0 {
|
|
||||||
panic("invalid use of queue")
|
|
||||||
}
|
|
||||||
return q.s[0]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (q *writeQueue) shift() frameWriteMsg {
|
|
||||||
if len(q.s) == 0 {
|
|
||||||
panic("invalid use of queue")
|
|
||||||
}
|
|
||||||
wm := q.s[0]
|
|
||||||
// TODO: less copy-happy queue.
|
|
||||||
copy(q.s, q.s[1:])
|
|
||||||
q.s[len(q.s)-1] = frameWriteMsg{}
|
|
||||||
q.s = q.s[:len(q.s)-1]
|
|
||||||
return wm
|
|
||||||
}
|
|
||||||
|
|
||||||
func (q *writeQueue) firstIsNoCost() bool {
|
|
||||||
if df, ok := q.s[0].write.(*writeData); ok {
|
|
||||||
return len(df.p) == 0
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
356
vendor/golang.org/x/net/http2/z_spec_test.go
generated
vendored
356
vendor/golang.org/x/net/http2/z_spec_test.go
generated
vendored
@ -1,356 +0,0 @@
|
|||||||
// Copyright 2014 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 http2
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"encoding/xml"
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"os"
|
|
||||||
"reflect"
|
|
||||||
"regexp"
|
|
||||||
"sort"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"sync"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
var coverSpec = flag.Bool("coverspec", false, "Run spec coverage tests")
|
|
||||||
|
|
||||||
// The global map of sentence coverage for the http2 spec.
|
|
||||||
var defaultSpecCoverage specCoverage
|
|
||||||
|
|
||||||
var loadSpecOnce sync.Once
|
|
||||||
|
|
||||||
func loadSpec() {
|
|
||||||
if f, err := os.Open("testdata/draft-ietf-httpbis-http2.xml"); err != nil {
|
|
||||||
panic(err)
|
|
||||||
} else {
|
|
||||||
defaultSpecCoverage = readSpecCov(f)
|
|
||||||
f.Close()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// covers marks all sentences for section sec in defaultSpecCoverage. Sentences not
|
|
||||||
// "covered" will be included in report outputed by TestSpecCoverage.
|
|
||||||
func covers(sec, sentences string) {
|
|
||||||
loadSpecOnce.Do(loadSpec)
|
|
||||||
defaultSpecCoverage.cover(sec, sentences)
|
|
||||||
}
|
|
||||||
|
|
||||||
type specPart struct {
|
|
||||||
section string
|
|
||||||
sentence string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ss specPart) Less(oo specPart) bool {
|
|
||||||
atoi := func(s string) int {
|
|
||||||
n, err := strconv.Atoi(s)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
return n
|
|
||||||
}
|
|
||||||
a := strings.Split(ss.section, ".")
|
|
||||||
b := strings.Split(oo.section, ".")
|
|
||||||
for len(a) > 0 {
|
|
||||||
if len(b) == 0 {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
x, y := atoi(a[0]), atoi(b[0])
|
|
||||||
if x == y {
|
|
||||||
a, b = a[1:], b[1:]
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
return x < y
|
|
||||||
}
|
|
||||||
if len(b) > 0 {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
type bySpecSection []specPart
|
|
||||||
|
|
||||||
func (a bySpecSection) Len() int { return len(a) }
|
|
||||||
func (a bySpecSection) Less(i, j int) bool { return a[i].Less(a[j]) }
|
|
||||||
func (a bySpecSection) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
|
||||||
|
|
||||||
type specCoverage struct {
|
|
||||||
coverage map[specPart]bool
|
|
||||||
d *xml.Decoder
|
|
||||||
}
|
|
||||||
|
|
||||||
func joinSection(sec []int) string {
|
|
||||||
s := fmt.Sprintf("%d", sec[0])
|
|
||||||
for _, n := range sec[1:] {
|
|
||||||
s = fmt.Sprintf("%s.%d", s, n)
|
|
||||||
}
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
|
|
||||||
func (sc specCoverage) readSection(sec []int) {
|
|
||||||
var (
|
|
||||||
buf = new(bytes.Buffer)
|
|
||||||
sub = 0
|
|
||||||
)
|
|
||||||
for {
|
|
||||||
tk, err := sc.d.Token()
|
|
||||||
if err != nil {
|
|
||||||
if err == io.EOF {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
switch v := tk.(type) {
|
|
||||||
case xml.StartElement:
|
|
||||||
if skipElement(v) {
|
|
||||||
if err := sc.d.Skip(); err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
if v.Name.Local == "section" {
|
|
||||||
sub++
|
|
||||||
}
|
|
||||||
break
|
|
||||||
}
|
|
||||||
switch v.Name.Local {
|
|
||||||
case "section":
|
|
||||||
sub++
|
|
||||||
sc.readSection(append(sec, sub))
|
|
||||||
case "xref":
|
|
||||||
buf.Write(sc.readXRef(v))
|
|
||||||
}
|
|
||||||
case xml.CharData:
|
|
||||||
if len(sec) == 0 {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
buf.Write(v)
|
|
||||||
case xml.EndElement:
|
|
||||||
if v.Name.Local == "section" {
|
|
||||||
sc.addSentences(joinSection(sec), buf.String())
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (sc specCoverage) readXRef(se xml.StartElement) []byte {
|
|
||||||
var b []byte
|
|
||||||
for {
|
|
||||||
tk, err := sc.d.Token()
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
switch v := tk.(type) {
|
|
||||||
case xml.CharData:
|
|
||||||
if b != nil {
|
|
||||||
panic("unexpected CharData")
|
|
||||||
}
|
|
||||||
b = []byte(string(v))
|
|
||||||
case xml.EndElement:
|
|
||||||
if v.Name.Local != "xref" {
|
|
||||||
panic("expected </xref>")
|
|
||||||
}
|
|
||||||
if b != nil {
|
|
||||||
return b
|
|
||||||
}
|
|
||||||
sig := attrSig(se)
|
|
||||||
switch sig {
|
|
||||||
case "target":
|
|
||||||
return []byte(fmt.Sprintf("[%s]", attrValue(se, "target")))
|
|
||||||
case "fmt-of,rel,target", "fmt-,,rel,target":
|
|
||||||
return []byte(fmt.Sprintf("[%s, %s]", attrValue(se, "target"), attrValue(se, "rel")))
|
|
||||||
case "fmt-of,sec,target", "fmt-,,sec,target":
|
|
||||||
return []byte(fmt.Sprintf("[section %s of %s]", attrValue(se, "sec"), attrValue(se, "target")))
|
|
||||||
case "fmt-of,rel,sec,target":
|
|
||||||
return []byte(fmt.Sprintf("[section %s of %s, %s]", attrValue(se, "sec"), attrValue(se, "target"), attrValue(se, "rel")))
|
|
||||||
default:
|
|
||||||
panic(fmt.Sprintf("unknown attribute signature %q in %#v", sig, fmt.Sprintf("%#v", se)))
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
panic(fmt.Sprintf("unexpected tag %q", v))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var skipAnchor = map[string]bool{
|
|
||||||
"intro": true,
|
|
||||||
"Overview": true,
|
|
||||||
}
|
|
||||||
|
|
||||||
var skipTitle = map[string]bool{
|
|
||||||
"Acknowledgements": true,
|
|
||||||
"Change Log": true,
|
|
||||||
"Document Organization": true,
|
|
||||||
"Conventions and Terminology": true,
|
|
||||||
}
|
|
||||||
|
|
||||||
func skipElement(s xml.StartElement) bool {
|
|
||||||
switch s.Name.Local {
|
|
||||||
case "artwork":
|
|
||||||
return true
|
|
||||||
case "section":
|
|
||||||
for _, attr := range s.Attr {
|
|
||||||
switch attr.Name.Local {
|
|
||||||
case "anchor":
|
|
||||||
if skipAnchor[attr.Value] || strings.HasPrefix(attr.Value, "changes.since.") {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
case "title":
|
|
||||||
if skipTitle[attr.Value] {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func readSpecCov(r io.Reader) specCoverage {
|
|
||||||
sc := specCoverage{
|
|
||||||
coverage: map[specPart]bool{},
|
|
||||||
d: xml.NewDecoder(r)}
|
|
||||||
sc.readSection(nil)
|
|
||||||
return sc
|
|
||||||
}
|
|
||||||
|
|
||||||
func (sc specCoverage) addSentences(sec string, sentence string) {
|
|
||||||
for _, s := range parseSentences(sentence) {
|
|
||||||
sc.coverage[specPart{sec, s}] = false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (sc specCoverage) cover(sec string, sentence string) {
|
|
||||||
for _, s := range parseSentences(sentence) {
|
|
||||||
p := specPart{sec, s}
|
|
||||||
if _, ok := sc.coverage[p]; !ok {
|
|
||||||
panic(fmt.Sprintf("Not found in spec: %q, %q", sec, s))
|
|
||||||
}
|
|
||||||
sc.coverage[specPart{sec, s}] = true
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
var whitespaceRx = regexp.MustCompile(`\s+`)
|
|
||||||
|
|
||||||
func parseSentences(sens string) []string {
|
|
||||||
sens = strings.TrimSpace(sens)
|
|
||||||
if sens == "" {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
ss := strings.Split(whitespaceRx.ReplaceAllString(sens, " "), ". ")
|
|
||||||
for i, s := range ss {
|
|
||||||
s = strings.TrimSpace(s)
|
|
||||||
if !strings.HasSuffix(s, ".") {
|
|
||||||
s += "."
|
|
||||||
}
|
|
||||||
ss[i] = s
|
|
||||||
}
|
|
||||||
return ss
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSpecParseSentences(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
ss string
|
|
||||||
want []string
|
|
||||||
}{
|
|
||||||
{"Sentence 1. Sentence 2.",
|
|
||||||
[]string{
|
|
||||||
"Sentence 1.",
|
|
||||||
"Sentence 2.",
|
|
||||||
}},
|
|
||||||
{"Sentence 1. \nSentence 2.\tSentence 3.",
|
|
||||||
[]string{
|
|
||||||
"Sentence 1.",
|
|
||||||
"Sentence 2.",
|
|
||||||
"Sentence 3.",
|
|
||||||
}},
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, tt := range tests {
|
|
||||||
got := parseSentences(tt.ss)
|
|
||||||
if !reflect.DeepEqual(got, tt.want) {
|
|
||||||
t.Errorf("%d: got = %q, want %q", i, got, tt.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSpecCoverage(t *testing.T) {
|
|
||||||
if !*coverSpec {
|
|
||||||
t.Skip()
|
|
||||||
}
|
|
||||||
|
|
||||||
loadSpecOnce.Do(loadSpec)
|
|
||||||
|
|
||||||
var (
|
|
||||||
list []specPart
|
|
||||||
cv = defaultSpecCoverage.coverage
|
|
||||||
total = len(cv)
|
|
||||||
complete = 0
|
|
||||||
)
|
|
||||||
|
|
||||||
for sp, touched := range defaultSpecCoverage.coverage {
|
|
||||||
if touched {
|
|
||||||
complete++
|
|
||||||
} else {
|
|
||||||
list = append(list, sp)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
sort.Stable(bySpecSection(list))
|
|
||||||
|
|
||||||
if testing.Short() && len(list) > 5 {
|
|
||||||
list = list[:5]
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, p := range list {
|
|
||||||
t.Errorf("\tSECTION %s: %s", p.section, p.sentence)
|
|
||||||
}
|
|
||||||
|
|
||||||
t.Logf("%d/%d (%d%%) sentances covered", complete, total, (complete/total)*100)
|
|
||||||
}
|
|
||||||
|
|
||||||
func attrSig(se xml.StartElement) string {
|
|
||||||
var names []string
|
|
||||||
for _, attr := range se.Attr {
|
|
||||||
if attr.Name.Local == "fmt" {
|
|
||||||
names = append(names, "fmt-"+attr.Value)
|
|
||||||
} else {
|
|
||||||
names = append(names, attr.Name.Local)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
sort.Strings(names)
|
|
||||||
return strings.Join(names, ",")
|
|
||||||
}
|
|
||||||
|
|
||||||
func attrValue(se xml.StartElement, attr string) string {
|
|
||||||
for _, a := range se.Attr {
|
|
||||||
if a.Name.Local == attr {
|
|
||||||
return a.Value
|
|
||||||
}
|
|
||||||
}
|
|
||||||
panic("unknown attribute " + attr)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSpecPartLess(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
sec1, sec2 string
|
|
||||||
want bool
|
|
||||||
}{
|
|
||||||
{"6.2.1", "6.2", false},
|
|
||||||
{"6.2", "6.2.1", true},
|
|
||||||
{"6.10", "6.10.1", true},
|
|
||||||
{"6.10", "6.1.1", false}, // 10, not 1
|
|
||||||
{"6.1", "6.1", false}, // equal, so not less
|
|
||||||
}
|
|
||||||
for _, tt := range tests {
|
|
||||||
got := (specPart{tt.sec1, "foo"}).Less(specPart{tt.sec2, "foo"})
|
|
||||||
if got != tt.want {
|
|
||||||
t.Errorf("Less(%q, %q) = %v; want %v", tt.sec1, tt.sec2, got, tt.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
7
vendor/manifest
vendored
7
vendor/manifest
vendored
@ -247,13 +247,6 @@
|
|||||||
"branch": "master",
|
"branch": "master",
|
||||||
"path": "/ssh/terminal"
|
"path": "/ssh/terminal"
|
||||||
},
|
},
|
||||||
{
|
|
||||||
"importpath": "golang.org/x/net/http2",
|
|
||||||
"repository": "https://go.googlesource.com/net",
|
|
||||||
"revision": "6acef71eb69611914f7a30939ea9f6e194c78172",
|
|
||||||
"branch": "master",
|
|
||||||
"path": "/http2"
|
|
||||||
},
|
|
||||||
{
|
{
|
||||||
"importpath": "golang.org/x/text/transform",
|
"importpath": "golang.org/x/text/transform",
|
||||||
"repository": "https://go.googlesource.com/text",
|
"repository": "https://go.googlesource.com/text",
|
||||||
|
Loading…
Reference in New Issue
Block a user