|
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316 |
- // Copyright 2015 Google Inc. All rights reserved.
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
-
- package martian
-
- import (
- "bufio"
- "bytes"
- "crypto/tls"
- "crypto/x509"
- "errors"
- "fmt"
- "io"
- "io/ioutil"
- "net"
- "net/http"
- "net/url"
- "os"
- "strings"
- "testing"
- "time"
-
- "github.com/google/martian/log"
- "github.com/google/martian/martiantest"
- "github.com/google/martian/mitm"
- "github.com/google/martian/proxyutil"
- )
-
- type tempError struct{}
-
- func (e *tempError) Error() string { return "temporary" }
- func (e *tempError) Timeout() bool { return true }
- func (e *tempError) Temporary() bool { return true }
-
- type timeoutListener struct {
- net.Listener
- errCount int
- err error
- }
-
- func newTimeoutListener(l net.Listener, errCount int) net.Listener {
- return &timeoutListener{
- Listener: l,
- errCount: errCount,
- err: &tempError{},
- }
- }
-
- func (l *timeoutListener) Accept() (net.Conn, error) {
- if l.errCount > 0 {
- l.errCount--
- return nil, l.err
- }
-
- return l.Listener.Accept()
- }
-
- func TestIntegrationTemporaryTimeout(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- tr := martiantest.NewTransport()
- p.SetRoundTripper(tr)
- p.SetTimeout(200 * time.Millisecond)
-
- // Start the proxy with a listener that will return a temporary error on
- // Accept() three times.
- go p.Serve(newTimeoutListener(l, 3))
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("GET", "http://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
- req.Header.Set("Connection", "close")
-
- // GET http://example.com/ HTTP/1.1
- // Host: example.com
- if err := req.WriteProxy(conn); err != nil {
- t.Fatalf("req.WriteProxy(): got %v, want no error", err)
- }
-
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- defer res.Body.Close()
-
- if got, want := res.StatusCode, 200; got != want {
- t.Errorf("res.StatusCode: got %d, want %d", got, want)
- }
- }
-
- func TestIntegrationHTTP(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- p.SetRequestModifier(nil)
- p.SetResponseModifier(nil)
-
- tr := martiantest.NewTransport()
- p.SetRoundTripper(tr)
- p.SetTimeout(200 * time.Millisecond)
-
- tm := martiantest.NewModifier()
-
- tm.RequestFunc(func(req *http.Request) {
- ctx := NewContext(req)
- ctx.Set("martian.test", "true")
- })
-
- tm.ResponseFunc(func(res *http.Response) {
- ctx := NewContext(res.Request)
- v, _ := ctx.Get("martian.test")
-
- res.Header.Set("Martian-Test", v.(string))
- })
-
- p.SetRequestModifier(tm)
- p.SetResponseModifier(tm)
-
- go p.Serve(l)
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("GET", "http://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // GET http://example.com/ HTTP/1.1
- // Host: example.com
- if err := req.WriteProxy(conn); err != nil {
- t.Fatalf("req.WriteProxy(): got %v, want no error", err)
- }
-
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
-
- if got, want := res.StatusCode, 200; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
-
- if got, want := res.Header.Get("Martian-Test"), "true"; got != want {
- t.Errorf("res.Header.Get(%q): got %q, want %q", "Martian-Test", got, want)
- }
- }
-
- func TestIntegrationHTTP100Continue(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- p.SetTimeout(2 * time.Second)
-
- sl, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- go func() {
- conn, err := sl.Accept()
- if err != nil {
- log.Errorf("proxy_test: failed to accept connection: %v", err)
- return
- }
- defer conn.Close()
-
- log.Infof("proxy_test: accepted connection: %s", conn.RemoteAddr())
-
- req, err := http.ReadRequest(bufio.NewReader(conn))
- if err != nil {
- log.Errorf("proxy_test: failed to read request: %v", err)
- return
- }
-
- if req.Header.Get("Expect") == "100-continue" {
- log.Infof("proxy_test: received 100-continue request")
-
- conn.Write([]byte("HTTP/1.1 100 Continue\r\n\r\n"))
-
- log.Infof("proxy_test: sent 100-continue response")
- } else {
- log.Infof("proxy_test: received non 100-continue request")
-
- res := proxyutil.NewResponse(417, nil, req)
- res.Header.Set("Connection", "close")
- res.Write(conn)
- return
- }
-
- res := proxyutil.NewResponse(200, req.Body, req)
- res.Header.Set("Connection", "close")
- res.Write(conn)
-
- log.Infof("proxy_test: sent 200 response")
- }()
-
- tm := martiantest.NewModifier()
- p.SetRequestModifier(tm)
- p.SetResponseModifier(tm)
-
- go p.Serve(l)
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- host := sl.Addr().String()
- raw := fmt.Sprintf("POST http://%s/ HTTP/1.1\r\n"+
- "Host: %s\r\n"+
- "Content-Length: 12\r\n"+
- "Expect: 100-continue\r\n\r\n", host, host)
-
- if _, err := conn.Write([]byte(raw)); err != nil {
- t.Fatalf("conn.Write(headers): got %v, want no error", err)
- }
-
- go func() {
- select {
- case <-time.After(time.Second):
- conn.Write([]byte("body content"))
- }
- }()
-
- res, err := http.ReadResponse(bufio.NewReader(conn), nil)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- defer res.Body.Close()
-
- if got, want := res.StatusCode, 200; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
-
- got, err := ioutil.ReadAll(res.Body)
- if err != nil {
- t.Fatalf("ioutil.ReadAll(): got %v, want no error", err)
- }
-
- if want := []byte("body content"); !bytes.Equal(got, want) {
- t.Errorf("res.Body: got %q, want %q", got, want)
- }
-
- if !tm.RequestModified() {
- t.Error("tm.RequestModified(): got false, want true")
- }
- if !tm.ResponseModified() {
- t.Error("tm.ResponseModified(): got false, want true")
- }
- }
-
- func TestIntegrationHTTPDownstreamProxy(t *testing.T) {
- t.Parallel()
-
- // Start first proxy to use as downstream.
- dl, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- downstream := NewProxy()
- defer downstream.Close()
-
- dtr := martiantest.NewTransport()
- dtr.Respond(299)
- downstream.SetRoundTripper(dtr)
- downstream.SetTimeout(600 * time.Millisecond)
-
- go downstream.Serve(dl)
-
- // Start second proxy as upstream proxy, will write to downstream proxy.
- ul, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- upstream := NewProxy()
- defer upstream.Close()
-
- // Set upstream proxy's downstream proxy to the host:port of the first proxy.
- upstream.SetDownstreamProxy(&url.URL{
- Host: dl.Addr().String(),
- })
- upstream.SetTimeout(600 * time.Millisecond)
-
- go upstream.Serve(ul)
-
- // Open connection to upstream proxy.
- conn, err := net.Dial("tcp", ul.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("GET", "http://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // GET http://example.com/ HTTP/1.1
- // Host: example.com
- if err := req.WriteProxy(conn); err != nil {
- t.Fatalf("req.WriteProxy(): got %v, want no error", err)
- }
-
- // Response from downstream proxy.
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
-
- if got, want := res.StatusCode, 299; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
- }
-
- func TestIntegrationHTTPDownstreamProxyError(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- // Set proxy's downstream proxy to invalid host:port to force failure.
- p.SetDownstreamProxy(&url.URL{
- Host: "[::]:0",
- })
- p.SetTimeout(600 * time.Millisecond)
-
- tm := martiantest.NewModifier()
- reserr := errors.New("response error")
- tm.ResponseError(reserr)
-
- p.SetResponseModifier(tm)
-
- go p.Serve(l)
-
- // Open connection to upstream proxy.
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("CONNECT", "//example.com:443", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // CONNECT example.com:443 HTTP/1.1
- // Host: example.com
- if err := req.Write(conn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- // Response from upstream proxy, assuming downstream proxy failed to CONNECT.
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
-
- if got, want := res.StatusCode, 502; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
- if got, want := res.Header["Warning"][1], reserr.Error(); !strings.Contains(got, want) {
- t.Errorf("res.Header.get(%q): got %q, want to contain %q", "Warning", got, want)
- }
- }
-
- func TestIntegrationTLSHandshakeErrorCallback(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- // Test TLS server.
- ca, priv, err := mitm.NewAuthority("martian.proxy", "Martian Authority", time.Hour)
- if err != nil {
- t.Fatalf("mitm.NewAuthority(): got %v, want no error", err)
- }
- mc, err := mitm.NewConfig(ca, priv)
- if err != nil {
- t.Fatalf("mitm.NewConfig(): got %v, want no error", err)
- }
-
- var herr error
- mc.SetHandshakeErrorCallback(func(_ *http.Request, err error) { herr = fmt.Errorf("handshake error") })
- p.SetMITM(mc)
-
- tl, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("tls.Listen(): got %v, want no error", err)
- }
- tl = tls.NewListener(tl, mc.TLS())
-
- go http.Serve(tl, http.HandlerFunc(
- func(rw http.ResponseWriter, req *http.Request) {
- rw.WriteHeader(200)
- }))
-
- tm := martiantest.NewModifier()
-
- // Force the CONNECT request to dial the local TLS server.
- tm.RequestFunc(func(req *http.Request) {
- req.URL.Host = tl.Addr().String()
- })
-
- go p.Serve(l)
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("CONNECT", "//example.com:443", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // CONNECT example.com:443 HTTP/1.1
- // Host: example.com
- //
- // Rewritten to CONNECT to host:port in CONNECT request modifier.
- if err := req.Write(conn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- // CONNECT response after establishing tunnel.
- if _, err := http.ReadResponse(bufio.NewReader(conn), req); err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
-
- tlsconn := tls.Client(conn, &tls.Config{
- ServerName: "example.com",
- // Client has no cert so it will get "x509: certificate signed by unknown authority" from the
- // handshake and send "remote error: bad certificate" to the server.
- RootCAs: x509.NewCertPool(),
- })
- defer tlsconn.Close()
-
- req, err = http.NewRequest("GET", "https://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
- req.Header.Set("Connection", "close")
-
- if got, want := req.Write(tlsconn), "x509: certificate signed by unknown authority"; !strings.Contains(got.Error(), want) {
- t.Fatalf("Got incorrect error from Client Handshake(), got: %v, want: %v", got, want)
- }
-
- // TODO: herr is not being asserted against. It should be pushed on to a channel
- // of err, and the assertion should pull off of it and assert. That design resulted in the test
- // hanging for unknown reasons.
- t.Skip("skipping assertion of handshake error callback error due to mysterious deadlock")
- if got, want := herr, "remote error: bad certificate"; !strings.Contains(got.Error(), want) {
- t.Fatalf("Got incorrect error from Server Handshake(), got: %v, want: %v", got, want)
- }
- }
-
- func TestIntegrationConnect(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- // Test TLS server.
- ca, priv, err := mitm.NewAuthority("martian.proxy", "Martian Authority", time.Hour)
- if err != nil {
- t.Fatalf("mitm.NewAuthority(): got %v, want no error", err)
- }
- mc, err := mitm.NewConfig(ca, priv)
- if err != nil {
- t.Fatalf("mitm.NewConfig(): got %v, want no error", err)
- }
-
- tl, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("tls.Listen(): got %v, want no error", err)
- }
- tl = tls.NewListener(tl, mc.TLS())
-
- go http.Serve(tl, http.HandlerFunc(
- func(rw http.ResponseWriter, req *http.Request) {
- rw.WriteHeader(299)
- }))
-
- tm := martiantest.NewModifier()
- reqerr := errors.New("request error")
- reserr := errors.New("response error")
-
- // Force the CONNECT request to dial the local TLS server.
- tm.RequestFunc(func(req *http.Request) {
- req.URL.Host = tl.Addr().String()
- })
-
- tm.RequestError(reqerr)
- tm.ResponseError(reserr)
-
- p.SetRequestModifier(tm)
- p.SetResponseModifier(tm)
-
- go p.Serve(l)
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("CONNECT", "//example.com:443", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // CONNECT example.com:443 HTTP/1.1
- // Host: example.com
- //
- // Rewritten to CONNECT to host:port in CONNECT request modifier.
- if err := req.Write(conn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- // CONNECT response after establishing tunnel.
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
-
- if got, want := res.StatusCode, 200; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
-
- if !tm.RequestModified() {
- t.Error("tm.RequestModified(): got false, want true")
- }
- if !tm.ResponseModified() {
- t.Error("tm.ResponseModified(): got false, want true")
- }
- if got, want := res.Header.Get("Warning"), reserr.Error(); !strings.Contains(got, want) {
- t.Errorf("res.Header.Get(%q): got %q, want to contain %q", "Warning", got, want)
- }
-
- roots := x509.NewCertPool()
- roots.AddCert(ca)
-
- tlsconn := tls.Client(conn, &tls.Config{
- ServerName: "example.com",
- RootCAs: roots,
- })
- defer tlsconn.Close()
-
- req, err = http.NewRequest("GET", "https://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
- req.Header.Set("Connection", "close")
-
- // GET / HTTP/1.1
- // Host: example.com
- // Connection: close
- if err := req.Write(tlsconn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- res, err = http.ReadResponse(bufio.NewReader(tlsconn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- defer res.Body.Close()
-
- if got, want := res.StatusCode, 299; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
- if got, want := res.Header.Get("Warning"), reserr.Error(); strings.Contains(got, want) {
- t.Errorf("res.Header.Get(%q): got %s, want to not contain %s", "Warning", got, want)
- }
- }
-
- func TestIntegrationConnectDownstreamProxy(t *testing.T) {
- t.Parallel()
-
- // Start first proxy to use as downstream.
- dl, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- downstream := NewProxy()
- defer downstream.Close()
-
- dtr := martiantest.NewTransport()
- dtr.Respond(299)
- downstream.SetRoundTripper(dtr)
-
- ca, priv, err := mitm.NewAuthority("martian.proxy", "Martian Authority", 2*time.Hour)
- if err != nil {
- t.Fatalf("mitm.NewAuthority(): got %v, want no error", err)
- }
-
- mc, err := mitm.NewConfig(ca, priv)
- if err != nil {
- t.Fatalf("mitm.NewConfig(): got %v, want no error", err)
- }
- downstream.SetMITM(mc)
-
- go downstream.Serve(dl)
-
- // Start second proxy as upstream proxy, will CONNECT to downstream proxy.
- ul, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- upstream := NewProxy()
- defer upstream.Close()
-
- // Set upstream proxy's downstream proxy to the host:port of the first proxy.
- upstream.SetDownstreamProxy(&url.URL{
- Host: dl.Addr().String(),
- })
-
- go upstream.Serve(ul)
-
- // Open connection to upstream proxy.
- conn, err := net.Dial("tcp", ul.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("CONNECT", "//example.com:443", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // CONNECT example.com:443 HTTP/1.1
- // Host: example.com
- if err := req.Write(conn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- // Response from downstream proxy starting MITM.
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
-
- if got, want := res.StatusCode, 200; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
-
- roots := x509.NewCertPool()
- roots.AddCert(ca)
-
- tlsconn := tls.Client(conn, &tls.Config{
- // Validate the hostname.
- ServerName: "example.com",
- // The certificate will have been MITM'd, verify using the MITM CA
- // certificate.
- RootCAs: roots,
- })
- defer tlsconn.Close()
-
- req, err = http.NewRequest("GET", "https://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // GET / HTTP/1.1
- // Host: example.com
- if err := req.Write(tlsconn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- // Response from MITM in downstream proxy.
- res, err = http.ReadResponse(bufio.NewReader(tlsconn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- defer res.Body.Close()
-
- if got, want := res.StatusCode, 299; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
- }
-
- func TestIntegrationMITM(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- tr := martiantest.NewTransport()
- tr.Func(func(req *http.Request) (*http.Response, error) {
- res := proxyutil.NewResponse(200, nil, req)
- res.Header.Set("Request-Scheme", req.URL.Scheme)
-
- return res, nil
- })
-
- p.SetRoundTripper(tr)
- p.SetTimeout(600 * time.Millisecond)
-
- ca, priv, err := mitm.NewAuthority("martian.proxy", "Martian Authority", 2*time.Hour)
- if err != nil {
- t.Fatalf("mitm.NewAuthority(): got %v, want no error", err)
- }
-
- mc, err := mitm.NewConfig(ca, priv)
- if err != nil {
- t.Fatalf("mitm.NewConfig(): got %v, want no error", err)
- }
- p.SetMITM(mc)
-
- tm := martiantest.NewModifier()
- reqerr := errors.New("request error")
- reserr := errors.New("response error")
- tm.RequestError(reqerr)
- tm.ResponseError(reserr)
-
- p.SetRequestModifier(tm)
- p.SetResponseModifier(tm)
-
- go p.Serve(l)
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("CONNECT", "//example.com:443", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // CONNECT example.com:443 HTTP/1.1
- // Host: example.com
- if err := req.Write(conn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- // Response MITM'd from proxy.
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- if got, want := res.StatusCode, 200; got != want {
-
- t.Errorf("res.StatusCode: got %d, want %d", got, want)
- }
- if got, want := res.Header.Get("Warning"), reserr.Error(); !strings.Contains(got, want) {
- t.Errorf("res.Header.Get(%q): got %q, want to contain %q", "Warning", got, want)
- }
-
- roots := x509.NewCertPool()
- roots.AddCert(ca)
-
- tlsconn := tls.Client(conn, &tls.Config{
- ServerName: "example.com",
- RootCAs: roots,
- })
- defer tlsconn.Close()
-
- req, err = http.NewRequest("GET", "https://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // GET / HTTP/1.1
- // Host: example.com
- if err := req.Write(tlsconn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- // Response from MITM proxy.
- res, err = http.ReadResponse(bufio.NewReader(tlsconn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- defer res.Body.Close()
-
- if got, want := res.StatusCode, 200; got != want {
- t.Errorf("res.StatusCode: got %d, want %d", got, want)
- }
- if got, want := res.Header.Get("Request-Scheme"), "https"; got != want {
- t.Errorf("res.Header.Get(%q): got %q, want %q", "Request-Scheme", got, want)
- }
- if got, want := res.Header.Get("Warning"), reserr.Error(); !strings.Contains(got, want) {
- t.Errorf("res.Header.Get(%q): got %q, want to contain %q", "Warning", got, want)
- }
- }
-
- func TestIntegrationTransparentHTTP(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- tr := martiantest.NewTransport()
- p.SetRoundTripper(tr)
- p.SetTimeout(200 * time.Millisecond)
-
- tm := martiantest.NewModifier()
- p.SetRequestModifier(tm)
- p.SetResponseModifier(tm)
-
- go p.Serve(l)
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("GET", "http://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // GET / HTTP/1.1
- // Host: www.example.com
- if err := req.Write(conn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
-
- if got, want := res.StatusCode, 200; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
-
- if !tm.RequestModified() {
- t.Error("tm.RequestModified(): got false, want true")
- }
- if !tm.ResponseModified() {
- t.Error("tm.ResponseModified(): got false, want true")
- }
- }
-
- func TestIntegrationTransparentMITM(t *testing.T) {
- t.Parallel()
-
- ca, priv, err := mitm.NewAuthority("martian.proxy", "Martian Authority", 2*time.Hour)
- if err != nil {
- t.Fatalf("mitm.NewAuthority(): got %v, want no error", err)
- }
-
- mc, err := mitm.NewConfig(ca, priv)
- if err != nil {
- t.Fatalf("mitm.NewConfig(): got %v, want no error", err)
- }
-
- // Start TLS listener with config that will generate certificates based on
- // SNI from connection.
- //
- // BUG: tls.Listen will not accept a tls.Config where Certificates is empty,
- // even though it is supported by tls.Server when GetCertificate is not nil.
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
- l = tls.NewListener(l, mc.TLS())
-
- p := NewProxy()
- defer p.Close()
-
- tr := martiantest.NewTransport()
- tr.Func(func(req *http.Request) (*http.Response, error) {
- res := proxyutil.NewResponse(200, nil, req)
- res.Header.Set("Request-Scheme", req.URL.Scheme)
-
- return res, nil
- })
-
- p.SetRoundTripper(tr)
-
- tm := martiantest.NewModifier()
- p.SetRequestModifier(tm)
- p.SetResponseModifier(tm)
-
- go p.Serve(l)
-
- roots := x509.NewCertPool()
- roots.AddCert(ca)
-
- tlsconn, err := tls.Dial("tcp", l.Addr().String(), &tls.Config{
- // Verify the hostname is example.com.
- ServerName: "example.com",
- // The certificate will have been generated during MITM, so we need to
- // verify it with the generated CA certificate.
- RootCAs: roots,
- })
- if err != nil {
- t.Fatalf("tls.Dial(): got %v, want no error", err)
- }
- defer tlsconn.Close()
-
- req, err := http.NewRequest("GET", "https://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // Write Encrypted request directly, no CONNECT.
- // GET / HTTP/1.1
- // Host: example.com
- if err := req.Write(tlsconn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- res, err := http.ReadResponse(bufio.NewReader(tlsconn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- defer res.Body.Close()
-
- if got, want := res.StatusCode, 200; got != want {
- t.Fatalf("res.StatusCode: got %d, want %d", got, want)
- }
- if got, want := res.Header.Get("Request-Scheme"), "https"; got != want {
- t.Errorf("res.Header.Get(%q): got %q, want %q", "Request-Scheme", got, want)
- }
-
- if !tm.RequestModified() {
- t.Errorf("tm.RequestModified(): got false, want true")
- }
- if !tm.ResponseModified() {
- t.Errorf("tm.ResponseModified(): got false, want true")
- }
- }
-
- func TestIntegrationFailedRoundTrip(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- tr := martiantest.NewTransport()
- trerr := errors.New("round trip error")
- tr.RespondError(trerr)
- p.SetRoundTripper(tr)
- p.SetTimeout(200 * time.Millisecond)
-
- go p.Serve(l)
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("GET", "http://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // GET http://example.com/ HTTP/1.1
- // Host: example.com
- if err := req.WriteProxy(conn); err != nil {
- t.Fatalf("req.WriteProxy(): got %v, want no error", err)
- }
-
- // Response from failed round trip.
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- defer res.Body.Close()
-
- if got, want := res.StatusCode, 502; got != want {
- t.Errorf("res.StatusCode: got %d, want %d", got, want)
- }
-
- if got, want := res.Header.Get("Warning"), trerr.Error(); !strings.Contains(got, want) {
- t.Errorf("res.Header.Get(%q): got %q, want to contain %q", "Warning", got, want)
- }
- }
-
- func TestIntegrationSkipRoundTrip(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- // Transport will be skipped, no 500.
- tr := martiantest.NewTransport()
- tr.Respond(500)
- p.SetRoundTripper(tr)
- p.SetTimeout(200 * time.Millisecond)
-
- tm := martiantest.NewModifier()
- tm.RequestFunc(func(req *http.Request) {
- ctx := NewContext(req)
- ctx.SkipRoundTrip()
- })
- p.SetRequestModifier(tm)
-
- go p.Serve(l)
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("GET", "http://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // GET http://example.com/ HTTP/1.1
- // Host: example.com
- if err := req.WriteProxy(conn); err != nil {
- t.Fatalf("req.WriteProxy(): got %v, want no error", err)
- }
-
- // Response from skipped round trip.
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- defer res.Body.Close()
-
- if got, want := res.StatusCode, 200; got != want {
- t.Errorf("res.StatusCode: got %d, want %d", got, want)
- }
- }
-
- func TestHTTPThroughConnectWithMITM(t *testing.T) {
- t.Parallel()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- p := NewProxy()
- defer p.Close()
-
- tm := martiantest.NewModifier()
- tm.RequestFunc(func(req *http.Request) {
- ctx := NewContext(req)
- ctx.SkipRoundTrip()
-
- if req.Method != "GET" && req.Method != "CONNECT" {
- t.Errorf("unexpected method on request handler: %v", req.Method)
- }
- })
- p.SetRequestModifier(tm)
-
- ca, priv, err := mitm.NewAuthority("martian.proxy", "Martian Authority", 2*time.Hour)
- if err != nil {
- t.Fatalf("mitm.NewAuthority(): got %v, want no error", err)
- }
-
- mc, err := mitm.NewConfig(ca, priv)
- if err != nil {
- t.Fatalf("mitm.NewConfig(): got %v, want no error", err)
- }
- p.SetMITM(mc)
-
- go p.Serve(l)
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("CONNECT", "//example.com:80", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // CONNECT example.com:80 HTTP/1.1
- // Host: example.com
- if err := req.Write(conn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
-
- // Response skipped round trip.
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- res.Body.Close()
-
- if got, want := res.StatusCode, 200; got != want {
- t.Errorf("res.StatusCode: got %d, want %d", got, want)
- }
-
- req, err = http.NewRequest("GET", "http://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // GET http://example.com/ HTTP/1.1
- // Host: example.com
- if err := req.WriteProxy(conn); err != nil {
- t.Fatalf("req.WriteProxy(): got %v, want no error", err)
- }
-
- // Response from skipped round trip.
- res, err = http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- res.Body.Close()
-
- if got, want := res.StatusCode, 200; got != want {
- t.Errorf("res.StatusCode: got %d, want %d", got, want)
- }
-
- req, err = http.NewRequest("GET", "http://example.com", nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // GET http://example.com/ HTTP/1.1
- // Host: example.com
- if err := req.WriteProxy(conn); err != nil {
- t.Fatalf("req.WriteProxy(): got %v, want no error", err)
- }
-
- // Response from skipped round trip.
- res, err = http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- res.Body.Close()
-
- if got, want := res.StatusCode, 200; got != want {
- t.Errorf("res.StatusCode: got %d, want %d", got, want)
- }
- }
-
- func TestServerClosesConnection(t *testing.T) {
- t.Parallel()
-
- dstl, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("Failed to create http listener: %v", err)
- }
- defer dstl.Close()
-
- go func() {
- t.Logf("Waiting for server side connection")
- conn, err := dstl.Accept()
- if err != nil {
- t.Fatalf("Got error while accepting connection on destination listener: %v", err)
- }
- t.Logf("Accepted server side connection")
-
- buf := make([]byte, 16384)
- if _, err := conn.Read(buf); err != nil {
- t.Fatalf("Error reading: %v", err)
- }
-
- _, err = conn.Write([]byte("HTTP/1.1 301 MOVED PERMANENTLY\r\n" +
- "Server: \r\n" +
- "Date: \r\n" +
- "Referer: \r\n" +
- "Location: http://www.foo.com/\r\n" +
- "Content-type: text/html\r\n" +
- "Connection: close\r\n\r\n"))
- if err != nil {
- t.Fatalf("Got error while writting to connection on destination listener: %v", err)
- }
- conn.Close()
- }()
-
- l, err := net.Listen("tcp", "[::]:0")
- if err != nil {
- t.Fatalf("net.Listen(): got %v, want no error", err)
- }
-
- ca, priv, err := mitm.NewAuthority("martian.proxy", "Martian Authority", 2*time.Hour)
- if err != nil {
- t.Fatalf("mitm.NewAuthority(): got %v, want no error", err)
- }
-
- mc, err := mitm.NewConfig(ca, priv)
- if err != nil {
- t.Fatalf("mitm.NewConfig(): got %v, want no error", err)
- }
- p := NewProxy()
- p.SetMITM(mc)
- defer p.Close()
-
- // Start the proxy with a listener that will return a temporary error on
- // Accept() three times.
- go p.Serve(newTimeoutListener(l, 3))
-
- conn, err := net.Dial("tcp", l.Addr().String())
- if err != nil {
- t.Fatalf("net.Dial(): got %v, want no error", err)
- }
- defer conn.Close()
-
- req, err := http.NewRequest("CONNECT", fmt.Sprintf("//%s", dstl.Addr().String()), nil)
- if err != nil {
- t.Fatalf("http.NewRequest(): got %v, want no error", err)
- }
-
- // CONNECT example.com:443 HTTP/1.1
- // Host: example.com
- if err := req.Write(conn); err != nil {
- t.Fatalf("req.Write(): got %v, want no error", err)
- }
- res, err := http.ReadResponse(bufio.NewReader(conn), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- res.Body.Close()
-
- _, err = conn.Write([]byte("GET / HTTP/1.1\r\n" +
- "User-Agent: curl/7.35.0\r\n" +
- fmt.Sprintf("Host: %s\r\n", dstl.Addr()) +
- "Accept: */*\r\n\r\n"))
- if err != nil {
- t.Fatalf("Error while writing GET request: %v", err)
- }
-
- res, err = http.ReadResponse(bufio.NewReader(io.TeeReader(conn, os.Stderr)), req)
- if err != nil {
- t.Fatalf("http.ReadResponse(): got %v, want no error", err)
- }
- _, err = ioutil.ReadAll(res.Body)
- if err != nil {
- t.Fatalf("error while ReadAll: %v", err)
- }
- defer res.Body.Close()
- }
|