package http_test import ( "bufio" "net/http" "strings" "testing" . "v2ray.com/core/proxy/http" . "v2ray.com/ext/assert" _ "v2ray.com/core/transport/internet/tcp" ) func TestHopByHopHeadersStrip(t *testing.T) { assert := With(t) rawRequest := `GET /pkg/net/http/ HTTP/1.1 Host: golang.org Connection: keep-alive,Foo, Bar Foo: foo Bar: bar Proxy-Connection: keep-alive Proxy-Authenticate: abc User-Agent: Mozilla/5.0 (Macintosh; U; Intel Mac OS X; de-de) AppleWebKit/523.10.3 (KHTML, like Gecko) Version/3.0.4 Safari/523.10 Accept-Encoding: gzip Accept-Charset: ISO-8859-1,UTF-8;q=0.7,*;q=0.7 Cache-Control: no-cache Accept-Language: de,en;q=0.7,en-us;q=0.3 ` b := bufio.NewReader(strings.NewReader(rawRequest)) req, err := http.ReadRequest(b) assert(err, IsNil) assert(req.Header.Get("Foo"), Equals, "foo") assert(req.Header.Get("Bar"), Equals, "bar") assert(req.Header.Get("Connection"), Equals, "keep-alive,Foo, Bar") assert(req.Header.Get("Proxy-Connection"), Equals, "keep-alive") assert(req.Header.Get("Proxy-Authenticate"), Equals, "abc") StripHopByHopHeaders(req.Header) assert(req.Header.Get("Connection"), IsEmpty) assert(req.Header.Get("Foo"), IsEmpty) assert(req.Header.Get("Bar"), IsEmpty) assert(req.Header.Get("Proxy-Connection"), IsEmpty) assert(req.Header.Get("Proxy-Authenticate"), IsEmpty) }