github.com/tcnksm/go@v0.0.0-20141208075154-439b32936367/src/net/http/serve_test.go (about)

     1  // Copyright 2010 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  // End-to-end serving tests
     6  
     7  package http_test
     8  
     9  import (
    10  	"bufio"
    11  	"bytes"
    12  	"crypto/tls"
    13  	"errors"
    14  	"fmt"
    15  	"io"
    16  	"io/ioutil"
    17  	"log"
    18  	"math/rand"
    19  	"net"
    20  	. "net/http"
    21  	"net/http/httptest"
    22  	"net/http/httputil"
    23  	"net/url"
    24  	"os"
    25  	"os/exec"
    26  	"reflect"
    27  	"runtime"
    28  	"strconv"
    29  	"strings"
    30  	"sync"
    31  	"sync/atomic"
    32  	"syscall"
    33  	"testing"
    34  	"time"
    35  )
    36  
    37  type dummyAddr string
    38  type oneConnListener struct {
    39  	conn net.Conn
    40  }
    41  
    42  func (l *oneConnListener) Accept() (c net.Conn, err error) {
    43  	c = l.conn
    44  	if c == nil {
    45  		err = io.EOF
    46  		return
    47  	}
    48  	err = nil
    49  	l.conn = nil
    50  	return
    51  }
    52  
    53  func (l *oneConnListener) Close() error {
    54  	return nil
    55  }
    56  
    57  func (l *oneConnListener) Addr() net.Addr {
    58  	return dummyAddr("test-address")
    59  }
    60  
    61  func (a dummyAddr) Network() string {
    62  	return string(a)
    63  }
    64  
    65  func (a dummyAddr) String() string {
    66  	return string(a)
    67  }
    68  
    69  type noopConn struct{}
    70  
    71  func (noopConn) LocalAddr() net.Addr                { return dummyAddr("local-addr") }
    72  func (noopConn) RemoteAddr() net.Addr               { return dummyAddr("remote-addr") }
    73  func (noopConn) SetDeadline(t time.Time) error      { return nil }
    74  func (noopConn) SetReadDeadline(t time.Time) error  { return nil }
    75  func (noopConn) SetWriteDeadline(t time.Time) error { return nil }
    76  
    77  type rwTestConn struct {
    78  	io.Reader
    79  	io.Writer
    80  	noopConn
    81  
    82  	closeFunc func() error // called if non-nil
    83  	closec    chan bool    // else, if non-nil, send value to it on close
    84  }
    85  
    86  func (c *rwTestConn) Close() error {
    87  	if c.closeFunc != nil {
    88  		return c.closeFunc()
    89  	}
    90  	select {
    91  	case c.closec <- true:
    92  	default:
    93  	}
    94  	return nil
    95  }
    96  
    97  type testConn struct {
    98  	readBuf  bytes.Buffer
    99  	writeBuf bytes.Buffer
   100  	closec   chan bool // if non-nil, send value to it on close
   101  	noopConn
   102  }
   103  
   104  func (c *testConn) Read(b []byte) (int, error) {
   105  	return c.readBuf.Read(b)
   106  }
   107  
   108  func (c *testConn) Write(b []byte) (int, error) {
   109  	return c.writeBuf.Write(b)
   110  }
   111  
   112  func (c *testConn) Close() error {
   113  	select {
   114  	case c.closec <- true:
   115  	default:
   116  	}
   117  	return nil
   118  }
   119  
   120  // reqBytes treats req as a request (with \n delimiters) and returns it with \r\n delimiters,
   121  // ending in \r\n\r\n
   122  func reqBytes(req string) []byte {
   123  	return []byte(strings.Replace(strings.TrimSpace(req), "\n", "\r\n", -1) + "\r\n\r\n")
   124  }
   125  
   126  type handlerTest struct {
   127  	handler Handler
   128  }
   129  
   130  func newHandlerTest(h Handler) handlerTest {
   131  	return handlerTest{h}
   132  }
   133  
   134  func (ht handlerTest) rawResponse(req string) string {
   135  	reqb := reqBytes(req)
   136  	var output bytes.Buffer
   137  	conn := &rwTestConn{
   138  		Reader: bytes.NewReader(reqb),
   139  		Writer: &output,
   140  		closec: make(chan bool, 1),
   141  	}
   142  	ln := &oneConnListener{conn: conn}
   143  	go Serve(ln, ht.handler)
   144  	<-conn.closec
   145  	return output.String()
   146  }
   147  
   148  func TestConsumingBodyOnNextConn(t *testing.T) {
   149  	conn := new(testConn)
   150  	for i := 0; i < 2; i++ {
   151  		conn.readBuf.Write([]byte(
   152  			"POST / HTTP/1.1\r\n" +
   153  				"Host: test\r\n" +
   154  				"Content-Length: 11\r\n" +
   155  				"\r\n" +
   156  				"foo=1&bar=1"))
   157  	}
   158  
   159  	reqNum := 0
   160  	ch := make(chan *Request)
   161  	servech := make(chan error)
   162  	listener := &oneConnListener{conn}
   163  	handler := func(res ResponseWriter, req *Request) {
   164  		reqNum++
   165  		ch <- req
   166  	}
   167  
   168  	go func() {
   169  		servech <- Serve(listener, HandlerFunc(handler))
   170  	}()
   171  
   172  	var req *Request
   173  	req = <-ch
   174  	if req == nil {
   175  		t.Fatal("Got nil first request.")
   176  	}
   177  	if req.Method != "POST" {
   178  		t.Errorf("For request #1's method, got %q; expected %q",
   179  			req.Method, "POST")
   180  	}
   181  
   182  	req = <-ch
   183  	if req == nil {
   184  		t.Fatal("Got nil first request.")
   185  	}
   186  	if req.Method != "POST" {
   187  		t.Errorf("For request #2's method, got %q; expected %q",
   188  			req.Method, "POST")
   189  	}
   190  
   191  	if serveerr := <-servech; serveerr != io.EOF {
   192  		t.Errorf("Serve returned %q; expected EOF", serveerr)
   193  	}
   194  }
   195  
   196  type stringHandler string
   197  
   198  func (s stringHandler) ServeHTTP(w ResponseWriter, r *Request) {
   199  	w.Header().Set("Result", string(s))
   200  }
   201  
   202  var handlers = []struct {
   203  	pattern string
   204  	msg     string
   205  }{
   206  	{"/", "Default"},
   207  	{"/someDir/", "someDir"},
   208  	{"someHost.com/someDir/", "someHost.com/someDir"},
   209  }
   210  
   211  var vtests = []struct {
   212  	url      string
   213  	expected string
   214  }{
   215  	{"http://localhost/someDir/apage", "someDir"},
   216  	{"http://localhost/otherDir/apage", "Default"},
   217  	{"http://someHost.com/someDir/apage", "someHost.com/someDir"},
   218  	{"http://otherHost.com/someDir/apage", "someDir"},
   219  	{"http://otherHost.com/aDir/apage", "Default"},
   220  	// redirections for trees
   221  	{"http://localhost/someDir", "/someDir/"},
   222  	{"http://someHost.com/someDir", "/someDir/"},
   223  }
   224  
   225  func TestHostHandlers(t *testing.T) {
   226  	defer afterTest(t)
   227  	mux := NewServeMux()
   228  	for _, h := range handlers {
   229  		mux.Handle(h.pattern, stringHandler(h.msg))
   230  	}
   231  	ts := httptest.NewServer(mux)
   232  	defer ts.Close()
   233  
   234  	conn, err := net.Dial("tcp", ts.Listener.Addr().String())
   235  	if err != nil {
   236  		t.Fatal(err)
   237  	}
   238  	defer conn.Close()
   239  	cc := httputil.NewClientConn(conn, nil)
   240  	for _, vt := range vtests {
   241  		var r *Response
   242  		var req Request
   243  		if req.URL, err = url.Parse(vt.url); err != nil {
   244  			t.Errorf("cannot parse url: %v", err)
   245  			continue
   246  		}
   247  		if err := cc.Write(&req); err != nil {
   248  			t.Errorf("writing request: %v", err)
   249  			continue
   250  		}
   251  		r, err := cc.Read(&req)
   252  		if err != nil {
   253  			t.Errorf("reading response: %v", err)
   254  			continue
   255  		}
   256  		switch r.StatusCode {
   257  		case StatusOK:
   258  			s := r.Header.Get("Result")
   259  			if s != vt.expected {
   260  				t.Errorf("Get(%q) = %q, want %q", vt.url, s, vt.expected)
   261  			}
   262  		case StatusMovedPermanently:
   263  			s := r.Header.Get("Location")
   264  			if s != vt.expected {
   265  				t.Errorf("Get(%q) = %q, want %q", vt.url, s, vt.expected)
   266  			}
   267  		default:
   268  			t.Errorf("Get(%q) unhandled status code %d", vt.url, r.StatusCode)
   269  		}
   270  	}
   271  }
   272  
   273  var serveMuxRegister = []struct {
   274  	pattern string
   275  	h       Handler
   276  }{
   277  	{"/dir/", serve(200)},
   278  	{"/search", serve(201)},
   279  	{"codesearch.google.com/search", serve(202)},
   280  	{"codesearch.google.com/", serve(203)},
   281  	{"example.com/", HandlerFunc(checkQueryStringHandler)},
   282  }
   283  
   284  // serve returns a handler that sends a response with the given code.
   285  func serve(code int) HandlerFunc {
   286  	return func(w ResponseWriter, r *Request) {
   287  		w.WriteHeader(code)
   288  	}
   289  }
   290  
   291  // checkQueryStringHandler checks if r.URL.RawQuery has the same value
   292  // as the URL excluding the scheme and the query string and sends 200
   293  // response code if it is, 500 otherwise.
   294  func checkQueryStringHandler(w ResponseWriter, r *Request) {
   295  	u := *r.URL
   296  	u.Scheme = "http"
   297  	u.Host = r.Host
   298  	u.RawQuery = ""
   299  	if "http://"+r.URL.RawQuery == u.String() {
   300  		w.WriteHeader(200)
   301  	} else {
   302  		w.WriteHeader(500)
   303  	}
   304  }
   305  
   306  var serveMuxTests = []struct {
   307  	method  string
   308  	host    string
   309  	path    string
   310  	code    int
   311  	pattern string
   312  }{
   313  	{"GET", "google.com", "/", 404, ""},
   314  	{"GET", "google.com", "/dir", 301, "/dir/"},
   315  	{"GET", "google.com", "/dir/", 200, "/dir/"},
   316  	{"GET", "google.com", "/dir/file", 200, "/dir/"},
   317  	{"GET", "google.com", "/search", 201, "/search"},
   318  	{"GET", "google.com", "/search/", 404, ""},
   319  	{"GET", "google.com", "/search/foo", 404, ""},
   320  	{"GET", "codesearch.google.com", "/search", 202, "codesearch.google.com/search"},
   321  	{"GET", "codesearch.google.com", "/search/", 203, "codesearch.google.com/"},
   322  	{"GET", "codesearch.google.com", "/search/foo", 203, "codesearch.google.com/"},
   323  	{"GET", "codesearch.google.com", "/", 203, "codesearch.google.com/"},
   324  	{"GET", "images.google.com", "/search", 201, "/search"},
   325  	{"GET", "images.google.com", "/search/", 404, ""},
   326  	{"GET", "images.google.com", "/search/foo", 404, ""},
   327  	{"GET", "google.com", "/../search", 301, "/search"},
   328  	{"GET", "google.com", "/dir/..", 301, ""},
   329  	{"GET", "google.com", "/dir/..", 301, ""},
   330  	{"GET", "google.com", "/dir/./file", 301, "/dir/"},
   331  
   332  	// The /foo -> /foo/ redirect applies to CONNECT requests
   333  	// but the path canonicalization does not.
   334  	{"CONNECT", "google.com", "/dir", 301, "/dir/"},
   335  	{"CONNECT", "google.com", "/../search", 404, ""},
   336  	{"CONNECT", "google.com", "/dir/..", 200, "/dir/"},
   337  	{"CONNECT", "google.com", "/dir/..", 200, "/dir/"},
   338  	{"CONNECT", "google.com", "/dir/./file", 200, "/dir/"},
   339  }
   340  
   341  func TestServeMuxHandler(t *testing.T) {
   342  	mux := NewServeMux()
   343  	for _, e := range serveMuxRegister {
   344  		mux.Handle(e.pattern, e.h)
   345  	}
   346  
   347  	for _, tt := range serveMuxTests {
   348  		r := &Request{
   349  			Method: tt.method,
   350  			Host:   tt.host,
   351  			URL: &url.URL{
   352  				Path: tt.path,
   353  			},
   354  		}
   355  		h, pattern := mux.Handler(r)
   356  		rr := httptest.NewRecorder()
   357  		h.ServeHTTP(rr, r)
   358  		if pattern != tt.pattern || rr.Code != tt.code {
   359  			t.Errorf("%s %s %s = %d, %q, want %d, %q", tt.method, tt.host, tt.path, rr.Code, pattern, tt.code, tt.pattern)
   360  		}
   361  	}
   362  }
   363  
   364  var serveMuxTests2 = []struct {
   365  	method  string
   366  	host    string
   367  	url     string
   368  	code    int
   369  	redirOk bool
   370  }{
   371  	{"GET", "google.com", "/", 404, false},
   372  	{"GET", "example.com", "/test/?example.com/test/", 200, false},
   373  	{"GET", "example.com", "test/?example.com/test/", 200, true},
   374  }
   375  
   376  // TestServeMuxHandlerRedirects tests that automatic redirects generated by
   377  // mux.Handler() shouldn't clear the request's query string.
   378  func TestServeMuxHandlerRedirects(t *testing.T) {
   379  	mux := NewServeMux()
   380  	for _, e := range serveMuxRegister {
   381  		mux.Handle(e.pattern, e.h)
   382  	}
   383  
   384  	for _, tt := range serveMuxTests2 {
   385  		tries := 1
   386  		turl := tt.url
   387  		for tries > 0 {
   388  			u, e := url.Parse(turl)
   389  			if e != nil {
   390  				t.Fatal(e)
   391  			}
   392  			r := &Request{
   393  				Method: tt.method,
   394  				Host:   tt.host,
   395  				URL:    u,
   396  			}
   397  			h, _ := mux.Handler(r)
   398  			rr := httptest.NewRecorder()
   399  			h.ServeHTTP(rr, r)
   400  			if rr.Code != 301 {
   401  				if rr.Code != tt.code {
   402  					t.Errorf("%s %s %s = %d, want %d", tt.method, tt.host, tt.url, rr.Code, tt.code)
   403  				}
   404  				break
   405  			}
   406  			if !tt.redirOk {
   407  				t.Errorf("%s %s %s, unexpected redirect", tt.method, tt.host, tt.url)
   408  				break
   409  			}
   410  			turl = rr.HeaderMap.Get("Location")
   411  			tries--
   412  		}
   413  		if tries < 0 {
   414  			t.Errorf("%s %s %s, too many redirects", tt.method, tt.host, tt.url)
   415  		}
   416  	}
   417  }
   418  
   419  // Tests for http://code.google.com/p/go/issues/detail?id=900
   420  func TestMuxRedirectLeadingSlashes(t *testing.T) {
   421  	paths := []string{"//foo.txt", "///foo.txt", "/../../foo.txt"}
   422  	for _, path := range paths {
   423  		req, err := ReadRequest(bufio.NewReader(strings.NewReader("GET " + path + " HTTP/1.1\r\nHost: test\r\n\r\n")))
   424  		if err != nil {
   425  			t.Errorf("%s", err)
   426  		}
   427  		mux := NewServeMux()
   428  		resp := httptest.NewRecorder()
   429  
   430  		mux.ServeHTTP(resp, req)
   431  
   432  		if loc, expected := resp.Header().Get("Location"), "/foo.txt"; loc != expected {
   433  			t.Errorf("Expected Location header set to %q; got %q", expected, loc)
   434  			return
   435  		}
   436  
   437  		if code, expected := resp.Code, StatusMovedPermanently; code != expected {
   438  			t.Errorf("Expected response code of StatusMovedPermanently; got %d", code)
   439  			return
   440  		}
   441  	}
   442  }
   443  
   444  func TestServerTimeouts(t *testing.T) {
   445  	if runtime.GOOS == "plan9" {
   446  		t.Skip("skipping test; see http://golang.org/issue/7237")
   447  	}
   448  	defer afterTest(t)
   449  	reqNum := 0
   450  	ts := httptest.NewUnstartedServer(HandlerFunc(func(res ResponseWriter, req *Request) {
   451  		reqNum++
   452  		fmt.Fprintf(res, "req=%d", reqNum)
   453  	}))
   454  	ts.Config.ReadTimeout = 250 * time.Millisecond
   455  	ts.Config.WriteTimeout = 250 * time.Millisecond
   456  	ts.Start()
   457  	defer ts.Close()
   458  
   459  	// Hit the HTTP server successfully.
   460  	tr := &Transport{DisableKeepAlives: true} // they interfere with this test
   461  	defer tr.CloseIdleConnections()
   462  	c := &Client{Transport: tr}
   463  	r, err := c.Get(ts.URL)
   464  	if err != nil {
   465  		t.Fatalf("http Get #1: %v", err)
   466  	}
   467  	got, _ := ioutil.ReadAll(r.Body)
   468  	expected := "req=1"
   469  	if string(got) != expected {
   470  		t.Errorf("Unexpected response for request #1; got %q; expected %q",
   471  			string(got), expected)
   472  	}
   473  
   474  	// Slow client that should timeout.
   475  	t1 := time.Now()
   476  	conn, err := net.Dial("tcp", ts.Listener.Addr().String())
   477  	if err != nil {
   478  		t.Fatalf("Dial: %v", err)
   479  	}
   480  	buf := make([]byte, 1)
   481  	n, err := conn.Read(buf)
   482  	latency := time.Since(t1)
   483  	if n != 0 || err != io.EOF {
   484  		t.Errorf("Read = %v, %v, wanted %v, %v", n, err, 0, io.EOF)
   485  	}
   486  	if latency < 200*time.Millisecond /* fudge from 250 ms above */ {
   487  		t.Errorf("got EOF after %s, want >= %s", latency, 200*time.Millisecond)
   488  	}
   489  
   490  	// Hit the HTTP server successfully again, verifying that the
   491  	// previous slow connection didn't run our handler.  (that we
   492  	// get "req=2", not "req=3")
   493  	r, err = Get(ts.URL)
   494  	if err != nil {
   495  		t.Fatalf("http Get #2: %v", err)
   496  	}
   497  	got, _ = ioutil.ReadAll(r.Body)
   498  	expected = "req=2"
   499  	if string(got) != expected {
   500  		t.Errorf("Get #2 got %q, want %q", string(got), expected)
   501  	}
   502  
   503  	if !testing.Short() {
   504  		conn, err := net.Dial("tcp", ts.Listener.Addr().String())
   505  		if err != nil {
   506  			t.Fatalf("Dial: %v", err)
   507  		}
   508  		defer conn.Close()
   509  		go io.Copy(ioutil.Discard, conn)
   510  		for i := 0; i < 5; i++ {
   511  			_, err := conn.Write([]byte("GET / HTTP/1.1\r\nHost: foo\r\n\r\n"))
   512  			if err != nil {
   513  				t.Fatalf("on write %d: %v", i, err)
   514  			}
   515  			time.Sleep(ts.Config.ReadTimeout / 2)
   516  		}
   517  	}
   518  }
   519  
   520  // golang.org/issue/4741 -- setting only a write timeout that triggers
   521  // shouldn't cause a handler to block forever on reads (next HTTP
   522  // request) that will never happen.
   523  func TestOnlyWriteTimeout(t *testing.T) {
   524  	if runtime.GOOS == "plan9" {
   525  		t.Skip("skipping test; see http://golang.org/issue/7237")
   526  	}
   527  	defer afterTest(t)
   528  	var conn net.Conn
   529  	var afterTimeoutErrc = make(chan error, 1)
   530  	ts := httptest.NewUnstartedServer(HandlerFunc(func(w ResponseWriter, req *Request) {
   531  		buf := make([]byte, 512<<10)
   532  		_, err := w.Write(buf)
   533  		if err != nil {
   534  			t.Errorf("handler Write error: %v", err)
   535  			return
   536  		}
   537  		conn.SetWriteDeadline(time.Now().Add(-30 * time.Second))
   538  		_, err = w.Write(buf)
   539  		afterTimeoutErrc <- err
   540  	}))
   541  	ts.Listener = trackLastConnListener{ts.Listener, &conn}
   542  	ts.Start()
   543  	defer ts.Close()
   544  
   545  	tr := &Transport{DisableKeepAlives: false}
   546  	defer tr.CloseIdleConnections()
   547  	c := &Client{Transport: tr}
   548  
   549  	errc := make(chan error)
   550  	go func() {
   551  		res, err := c.Get(ts.URL)
   552  		if err != nil {
   553  			errc <- err
   554  			return
   555  		}
   556  		_, err = io.Copy(ioutil.Discard, res.Body)
   557  		errc <- err
   558  	}()
   559  	select {
   560  	case err := <-errc:
   561  		if err == nil {
   562  			t.Errorf("expected an error from Get request")
   563  		}
   564  	case <-time.After(5 * time.Second):
   565  		t.Fatal("timeout waiting for Get error")
   566  	}
   567  	if err := <-afterTimeoutErrc; err == nil {
   568  		t.Error("expected write error after timeout")
   569  	}
   570  }
   571  
   572  // trackLastConnListener tracks the last net.Conn that was accepted.
   573  type trackLastConnListener struct {
   574  	net.Listener
   575  	last *net.Conn // destination
   576  }
   577  
   578  func (l trackLastConnListener) Accept() (c net.Conn, err error) {
   579  	c, err = l.Listener.Accept()
   580  	*l.last = c
   581  	return
   582  }
   583  
   584  // TestIdentityResponse verifies that a handler can unset
   585  func TestIdentityResponse(t *testing.T) {
   586  	defer afterTest(t)
   587  	handler := HandlerFunc(func(rw ResponseWriter, req *Request) {
   588  		rw.Header().Set("Content-Length", "3")
   589  		rw.Header().Set("Transfer-Encoding", req.FormValue("te"))
   590  		switch {
   591  		case req.FormValue("overwrite") == "1":
   592  			_, err := rw.Write([]byte("foo TOO LONG"))
   593  			if err != ErrContentLength {
   594  				t.Errorf("expected ErrContentLength; got %v", err)
   595  			}
   596  		case req.FormValue("underwrite") == "1":
   597  			rw.Header().Set("Content-Length", "500")
   598  			rw.Write([]byte("too short"))
   599  		default:
   600  			rw.Write([]byte("foo"))
   601  		}
   602  	})
   603  
   604  	ts := httptest.NewServer(handler)
   605  	defer ts.Close()
   606  
   607  	// Note: this relies on the assumption (which is true) that
   608  	// Get sends HTTP/1.1 or greater requests.  Otherwise the
   609  	// server wouldn't have the choice to send back chunked
   610  	// responses.
   611  	for _, te := range []string{"", "identity"} {
   612  		url := ts.URL + "/?te=" + te
   613  		res, err := Get(url)
   614  		if err != nil {
   615  			t.Fatalf("error with Get of %s: %v", url, err)
   616  		}
   617  		if cl, expected := res.ContentLength, int64(3); cl != expected {
   618  			t.Errorf("for %s expected res.ContentLength of %d; got %d", url, expected, cl)
   619  		}
   620  		if cl, expected := res.Header.Get("Content-Length"), "3"; cl != expected {
   621  			t.Errorf("for %s expected Content-Length header of %q; got %q", url, expected, cl)
   622  		}
   623  		if tl, expected := len(res.TransferEncoding), 0; tl != expected {
   624  			t.Errorf("for %s expected len(res.TransferEncoding) of %d; got %d (%v)",
   625  				url, expected, tl, res.TransferEncoding)
   626  		}
   627  		res.Body.Close()
   628  	}
   629  
   630  	// Verify that ErrContentLength is returned
   631  	url := ts.URL + "/?overwrite=1"
   632  	res, err := Get(url)
   633  	if err != nil {
   634  		t.Fatalf("error with Get of %s: %v", url, err)
   635  	}
   636  	res.Body.Close()
   637  
   638  	// Verify that the connection is closed when the declared Content-Length
   639  	// is larger than what the handler wrote.
   640  	conn, err := net.Dial("tcp", ts.Listener.Addr().String())
   641  	if err != nil {
   642  		t.Fatalf("error dialing: %v", err)
   643  	}
   644  	_, err = conn.Write([]byte("GET /?underwrite=1 HTTP/1.1\r\nHost: foo\r\n\r\n"))
   645  	if err != nil {
   646  		t.Fatalf("error writing: %v", err)
   647  	}
   648  
   649  	// The ReadAll will hang for a failing test, so use a Timer to
   650  	// fail explicitly.
   651  	goTimeout(t, 2*time.Second, func() {
   652  		got, _ := ioutil.ReadAll(conn)
   653  		expectedSuffix := "\r\n\r\ntoo short"
   654  		if !strings.HasSuffix(string(got), expectedSuffix) {
   655  			t.Errorf("Expected output to end with %q; got response body %q",
   656  				expectedSuffix, string(got))
   657  		}
   658  	})
   659  }
   660  
   661  func testTCPConnectionCloses(t *testing.T, req string, h Handler) {
   662  	defer afterTest(t)
   663  	s := httptest.NewServer(h)
   664  	defer s.Close()
   665  
   666  	conn, err := net.Dial("tcp", s.Listener.Addr().String())
   667  	if err != nil {
   668  		t.Fatal("dial error:", err)
   669  	}
   670  	defer conn.Close()
   671  
   672  	_, err = fmt.Fprint(conn, req)
   673  	if err != nil {
   674  		t.Fatal("print error:", err)
   675  	}
   676  
   677  	r := bufio.NewReader(conn)
   678  	res, err := ReadResponse(r, &Request{Method: "GET"})
   679  	if err != nil {
   680  		t.Fatal("ReadResponse error:", err)
   681  	}
   682  
   683  	didReadAll := make(chan bool, 1)
   684  	go func() {
   685  		select {
   686  		case <-time.After(5 * time.Second):
   687  			t.Error("body not closed after 5s")
   688  			return
   689  		case <-didReadAll:
   690  		}
   691  	}()
   692  
   693  	_, err = ioutil.ReadAll(r)
   694  	if err != nil {
   695  		t.Fatal("read error:", err)
   696  	}
   697  	didReadAll <- true
   698  
   699  	if !res.Close {
   700  		t.Errorf("Response.Close = false; want true")
   701  	}
   702  }
   703  
   704  // TestServeHTTP10Close verifies that HTTP/1.0 requests won't be kept alive.
   705  func TestServeHTTP10Close(t *testing.T) {
   706  	testTCPConnectionCloses(t, "GET / HTTP/1.0\r\n\r\n", HandlerFunc(func(w ResponseWriter, r *Request) {
   707  		ServeFile(w, r, "testdata/file")
   708  	}))
   709  }
   710  
   711  // TestClientCanClose verifies that clients can also force a connection to close.
   712  func TestClientCanClose(t *testing.T) {
   713  	testTCPConnectionCloses(t, "GET / HTTP/1.1\r\nConnection: close\r\n\r\n", HandlerFunc(func(w ResponseWriter, r *Request) {
   714  		// Nothing.
   715  	}))
   716  }
   717  
   718  // TestHandlersCanSetConnectionClose verifies that handlers can force a connection to close,
   719  // even for HTTP/1.1 requests.
   720  func TestHandlersCanSetConnectionClose11(t *testing.T) {
   721  	testTCPConnectionCloses(t, "GET / HTTP/1.1\r\n\r\n", HandlerFunc(func(w ResponseWriter, r *Request) {
   722  		w.Header().Set("Connection", "close")
   723  	}))
   724  }
   725  
   726  func TestHandlersCanSetConnectionClose10(t *testing.T) {
   727  	testTCPConnectionCloses(t, "GET / HTTP/1.0\r\nConnection: keep-alive\r\n\r\n", HandlerFunc(func(w ResponseWriter, r *Request) {
   728  		w.Header().Set("Connection", "close")
   729  	}))
   730  }
   731  
   732  func TestSetsRemoteAddr(t *testing.T) {
   733  	defer afterTest(t)
   734  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
   735  		fmt.Fprintf(w, "%s", r.RemoteAddr)
   736  	}))
   737  	defer ts.Close()
   738  
   739  	res, err := Get(ts.URL)
   740  	if err != nil {
   741  		t.Fatalf("Get error: %v", err)
   742  	}
   743  	body, err := ioutil.ReadAll(res.Body)
   744  	if err != nil {
   745  		t.Fatalf("ReadAll error: %v", err)
   746  	}
   747  	ip := string(body)
   748  	if !strings.HasPrefix(ip, "127.0.0.1:") && !strings.HasPrefix(ip, "[::1]:") {
   749  		t.Fatalf("Expected local addr; got %q", ip)
   750  	}
   751  }
   752  
   753  func TestChunkedResponseHeaders(t *testing.T) {
   754  	defer afterTest(t)
   755  	log.SetOutput(ioutil.Discard) // is noisy otherwise
   756  	defer log.SetOutput(os.Stderr)
   757  
   758  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
   759  		w.Header().Set("Content-Length", "intentional gibberish") // we check that this is deleted
   760  		w.(Flusher).Flush()
   761  		fmt.Fprintf(w, "I am a chunked response.")
   762  	}))
   763  	defer ts.Close()
   764  
   765  	res, err := Get(ts.URL)
   766  	if err != nil {
   767  		t.Fatalf("Get error: %v", err)
   768  	}
   769  	defer res.Body.Close()
   770  	if g, e := res.ContentLength, int64(-1); g != e {
   771  		t.Errorf("expected ContentLength of %d; got %d", e, g)
   772  	}
   773  	if g, e := res.TransferEncoding, []string{"chunked"}; !reflect.DeepEqual(g, e) {
   774  		t.Errorf("expected TransferEncoding of %v; got %v", e, g)
   775  	}
   776  	if _, haveCL := res.Header["Content-Length"]; haveCL {
   777  		t.Errorf("Unexpected Content-Length")
   778  	}
   779  }
   780  
   781  func TestIdentityResponseHeaders(t *testing.T) {
   782  	defer afterTest(t)
   783  	log.SetOutput(ioutil.Discard) // is noisy otherwise
   784  	defer log.SetOutput(os.Stderr)
   785  
   786  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
   787  		w.Header().Set("Transfer-Encoding", "identity")
   788  		w.(Flusher).Flush()
   789  		fmt.Fprintf(w, "I am an identity response.")
   790  	}))
   791  	defer ts.Close()
   792  
   793  	res, err := Get(ts.URL)
   794  	if err != nil {
   795  		t.Fatalf("Get error: %v", err)
   796  	}
   797  	defer res.Body.Close()
   798  
   799  	if g, e := res.TransferEncoding, []string(nil); !reflect.DeepEqual(g, e) {
   800  		t.Errorf("expected TransferEncoding of %v; got %v", e, g)
   801  	}
   802  	if _, haveCL := res.Header["Content-Length"]; haveCL {
   803  		t.Errorf("Unexpected Content-Length")
   804  	}
   805  	if !res.Close {
   806  		t.Errorf("expected Connection: close; got %v", res.Close)
   807  	}
   808  }
   809  
   810  // Test304Responses verifies that 304s don't declare that they're
   811  // chunking in their response headers and aren't allowed to produce
   812  // output.
   813  func Test304Responses(t *testing.T) {
   814  	defer afterTest(t)
   815  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
   816  		w.WriteHeader(StatusNotModified)
   817  		_, err := w.Write([]byte("illegal body"))
   818  		if err != ErrBodyNotAllowed {
   819  			t.Errorf("on Write, expected ErrBodyNotAllowed, got %v", err)
   820  		}
   821  	}))
   822  	defer ts.Close()
   823  	res, err := Get(ts.URL)
   824  	if err != nil {
   825  		t.Error(err)
   826  	}
   827  	if len(res.TransferEncoding) > 0 {
   828  		t.Errorf("expected no TransferEncoding; got %v", res.TransferEncoding)
   829  	}
   830  	body, err := ioutil.ReadAll(res.Body)
   831  	if err != nil {
   832  		t.Error(err)
   833  	}
   834  	if len(body) > 0 {
   835  		t.Errorf("got unexpected body %q", string(body))
   836  	}
   837  }
   838  
   839  // TestHeadResponses verifies that all MIME type sniffing and Content-Length
   840  // counting of GET requests also happens on HEAD requests.
   841  func TestHeadResponses(t *testing.T) {
   842  	defer afterTest(t)
   843  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
   844  		_, err := w.Write([]byte("<html>"))
   845  		if err != nil {
   846  			t.Errorf("ResponseWriter.Write: %v", err)
   847  		}
   848  
   849  		// Also exercise the ReaderFrom path
   850  		_, err = io.Copy(w, strings.NewReader("789a"))
   851  		if err != nil {
   852  			t.Errorf("Copy(ResponseWriter, ...): %v", err)
   853  		}
   854  	}))
   855  	defer ts.Close()
   856  	res, err := Head(ts.URL)
   857  	if err != nil {
   858  		t.Error(err)
   859  	}
   860  	if len(res.TransferEncoding) > 0 {
   861  		t.Errorf("expected no TransferEncoding; got %v", res.TransferEncoding)
   862  	}
   863  	if ct := res.Header.Get("Content-Type"); ct != "text/html; charset=utf-8" {
   864  		t.Errorf("Content-Type: %q; want text/html; charset=utf-8", ct)
   865  	}
   866  	if v := res.ContentLength; v != 10 {
   867  		t.Errorf("Content-Length: %d; want 10", v)
   868  	}
   869  	body, err := ioutil.ReadAll(res.Body)
   870  	if err != nil {
   871  		t.Error(err)
   872  	}
   873  	if len(body) > 0 {
   874  		t.Errorf("got unexpected body %q", string(body))
   875  	}
   876  }
   877  
   878  func TestTLSHandshakeTimeout(t *testing.T) {
   879  	if runtime.GOOS == "plan9" {
   880  		t.Skip("skipping test; see http://golang.org/issue/7237")
   881  	}
   882  	defer afterTest(t)
   883  	ts := httptest.NewUnstartedServer(HandlerFunc(func(w ResponseWriter, r *Request) {}))
   884  	errc := make(chanWriter, 10) // but only expecting 1
   885  	ts.Config.ReadTimeout = 250 * time.Millisecond
   886  	ts.Config.ErrorLog = log.New(errc, "", 0)
   887  	ts.StartTLS()
   888  	defer ts.Close()
   889  	conn, err := net.Dial("tcp", ts.Listener.Addr().String())
   890  	if err != nil {
   891  		t.Fatalf("Dial: %v", err)
   892  	}
   893  	defer conn.Close()
   894  	goTimeout(t, 10*time.Second, func() {
   895  		var buf [1]byte
   896  		n, err := conn.Read(buf[:])
   897  		if err == nil || n != 0 {
   898  			t.Errorf("Read = %d, %v; want an error and no bytes", n, err)
   899  		}
   900  	})
   901  	select {
   902  	case v := <-errc:
   903  		if !strings.Contains(v, "timeout") && !strings.Contains(v, "TLS handshake") {
   904  			t.Errorf("expected a TLS handshake timeout error; got %q", v)
   905  		}
   906  	case <-time.After(5 * time.Second):
   907  		t.Errorf("timeout waiting for logged error")
   908  	}
   909  }
   910  
   911  func TestTLSServer(t *testing.T) {
   912  	defer afterTest(t)
   913  	ts := httptest.NewTLSServer(HandlerFunc(func(w ResponseWriter, r *Request) {
   914  		if r.TLS != nil {
   915  			w.Header().Set("X-TLS-Set", "true")
   916  			if r.TLS.HandshakeComplete {
   917  				w.Header().Set("X-TLS-HandshakeComplete", "true")
   918  			}
   919  		}
   920  	}))
   921  	ts.Config.ErrorLog = log.New(ioutil.Discard, "", 0)
   922  	defer ts.Close()
   923  
   924  	// Connect an idle TCP connection to this server before we run
   925  	// our real tests.  This idle connection used to block forever
   926  	// in the TLS handshake, preventing future connections from
   927  	// being accepted. It may prevent future accidental blocking
   928  	// in newConn.
   929  	idleConn, err := net.Dial("tcp", ts.Listener.Addr().String())
   930  	if err != nil {
   931  		t.Fatalf("Dial: %v", err)
   932  	}
   933  	defer idleConn.Close()
   934  	goTimeout(t, 10*time.Second, func() {
   935  		if !strings.HasPrefix(ts.URL, "https://") {
   936  			t.Errorf("expected test TLS server to start with https://, got %q", ts.URL)
   937  			return
   938  		}
   939  		noVerifyTransport := &Transport{
   940  			TLSClientConfig: &tls.Config{
   941  				InsecureSkipVerify: true,
   942  			},
   943  		}
   944  		client := &Client{Transport: noVerifyTransport}
   945  		res, err := client.Get(ts.URL)
   946  		if err != nil {
   947  			t.Error(err)
   948  			return
   949  		}
   950  		if res == nil {
   951  			t.Errorf("got nil Response")
   952  			return
   953  		}
   954  		defer res.Body.Close()
   955  		if res.Header.Get("X-TLS-Set") != "true" {
   956  			t.Errorf("expected X-TLS-Set response header")
   957  			return
   958  		}
   959  		if res.Header.Get("X-TLS-HandshakeComplete") != "true" {
   960  			t.Errorf("expected X-TLS-HandshakeComplete header")
   961  		}
   962  	})
   963  }
   964  
   965  type serverExpectTest struct {
   966  	contentLength    int // of request body
   967  	chunked          bool
   968  	expectation      string // e.g. "100-continue"
   969  	readBody         bool   // whether handler should read the body (if false, sends StatusUnauthorized)
   970  	expectedResponse string // expected substring in first line of http response
   971  }
   972  
   973  func expectTest(contentLength int, expectation string, readBody bool, expectedResponse string) serverExpectTest {
   974  	return serverExpectTest{
   975  		contentLength:    contentLength,
   976  		expectation:      expectation,
   977  		readBody:         readBody,
   978  		expectedResponse: expectedResponse,
   979  	}
   980  }
   981  
   982  var serverExpectTests = []serverExpectTest{
   983  	// Normal 100-continues, case-insensitive.
   984  	expectTest(100, "100-continue", true, "100 Continue"),
   985  	expectTest(100, "100-cOntInUE", true, "100 Continue"),
   986  
   987  	// No 100-continue.
   988  	expectTest(100, "", true, "200 OK"),
   989  
   990  	// 100-continue but requesting client to deny us,
   991  	// so it never reads the body.
   992  	expectTest(100, "100-continue", false, "401 Unauthorized"),
   993  	// Likewise without 100-continue:
   994  	expectTest(100, "", false, "401 Unauthorized"),
   995  
   996  	// Non-standard expectations are failures
   997  	expectTest(0, "a-pony", false, "417 Expectation Failed"),
   998  
   999  	// Expect-100 requested but no body (is apparently okay: Issue 7625)
  1000  	expectTest(0, "100-continue", true, "200 OK"),
  1001  	// Expect-100 requested but handler doesn't read the body
  1002  	expectTest(0, "100-continue", false, "401 Unauthorized"),
  1003  	// Expect-100 continue with no body, but a chunked body.
  1004  	{
  1005  		expectation:      "100-continue",
  1006  		readBody:         true,
  1007  		chunked:          true,
  1008  		expectedResponse: "100 Continue",
  1009  	},
  1010  }
  1011  
  1012  // Tests that the server responds to the "Expect" request header
  1013  // correctly.
  1014  func TestServerExpect(t *testing.T) {
  1015  	defer afterTest(t)
  1016  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  1017  		// Note using r.FormValue("readbody") because for POST
  1018  		// requests that would read from r.Body, which we only
  1019  		// conditionally want to do.
  1020  		if strings.Contains(r.URL.RawQuery, "readbody=true") {
  1021  			ioutil.ReadAll(r.Body)
  1022  			w.Write([]byte("Hi"))
  1023  		} else {
  1024  			w.WriteHeader(StatusUnauthorized)
  1025  		}
  1026  	}))
  1027  	defer ts.Close()
  1028  
  1029  	runTest := func(test serverExpectTest) {
  1030  		conn, err := net.Dial("tcp", ts.Listener.Addr().String())
  1031  		if err != nil {
  1032  			t.Fatalf("Dial: %v", err)
  1033  		}
  1034  		defer conn.Close()
  1035  
  1036  		// Only send the body immediately if we're acting like an HTTP client
  1037  		// that doesn't send 100-continue expectations.
  1038  		writeBody := test.contentLength != 0 && strings.ToLower(test.expectation) != "100-continue"
  1039  
  1040  		go func() {
  1041  			contentLen := fmt.Sprintf("Content-Length: %d", test.contentLength)
  1042  			if test.chunked {
  1043  				contentLen = "Transfer-Encoding: chunked"
  1044  			}
  1045  			_, err := fmt.Fprintf(conn, "POST /?readbody=%v HTTP/1.1\r\n"+
  1046  				"Connection: close\r\n"+
  1047  				"%s\r\n"+
  1048  				"Expect: %s\r\nHost: foo\r\n\r\n",
  1049  				test.readBody, contentLen, test.expectation)
  1050  			if err != nil {
  1051  				t.Errorf("On test %#v, error writing request headers: %v", test, err)
  1052  				return
  1053  			}
  1054  			if writeBody {
  1055  				var targ io.WriteCloser = struct {
  1056  					io.Writer
  1057  					io.Closer
  1058  				}{
  1059  					conn,
  1060  					ioutil.NopCloser(nil),
  1061  				}
  1062  				if test.chunked {
  1063  					targ = httputil.NewChunkedWriter(conn)
  1064  				}
  1065  				body := strings.Repeat("A", test.contentLength)
  1066  				_, err = fmt.Fprint(targ, body)
  1067  				if err == nil {
  1068  					err = targ.Close()
  1069  				}
  1070  				if err != nil {
  1071  					if !test.readBody {
  1072  						// Server likely already hung up on us.
  1073  						// See larger comment below.
  1074  						t.Logf("On test %#v, acceptable error writing request body: %v", test, err)
  1075  						return
  1076  					}
  1077  					t.Errorf("On test %#v, error writing request body: %v", test, err)
  1078  				}
  1079  			}
  1080  		}()
  1081  		bufr := bufio.NewReader(conn)
  1082  		line, err := bufr.ReadString('\n')
  1083  		if err != nil {
  1084  			if writeBody && !test.readBody {
  1085  				// This is an acceptable failure due to a possible TCP race:
  1086  				// We were still writing data and the server hung up on us. A TCP
  1087  				// implementation may send a RST if our request body data was known
  1088  				// to be lost, which may trigger our reads to fail.
  1089  				// See RFC 1122 page 88.
  1090  				t.Logf("On test %#v, acceptable error from ReadString: %v", test, err)
  1091  				return
  1092  			}
  1093  			t.Fatalf("On test %#v, ReadString: %v", test, err)
  1094  		}
  1095  		if !strings.Contains(line, test.expectedResponse) {
  1096  			t.Errorf("On test %#v, got first line = %q; want %q", test, line, test.expectedResponse)
  1097  		}
  1098  	}
  1099  
  1100  	for _, test := range serverExpectTests {
  1101  		runTest(test)
  1102  	}
  1103  }
  1104  
  1105  // Under a ~256KB (maxPostHandlerReadBytes) threshold, the server
  1106  // should consume client request bodies that a handler didn't read.
  1107  func TestServerUnreadRequestBodyLittle(t *testing.T) {
  1108  	conn := new(testConn)
  1109  	body := strings.Repeat("x", 100<<10)
  1110  	conn.readBuf.Write([]byte(fmt.Sprintf(
  1111  		"POST / HTTP/1.1\r\n"+
  1112  			"Host: test\r\n"+
  1113  			"Content-Length: %d\r\n"+
  1114  			"\r\n", len(body))))
  1115  	conn.readBuf.Write([]byte(body))
  1116  
  1117  	done := make(chan bool)
  1118  
  1119  	ls := &oneConnListener{conn}
  1120  	go Serve(ls, HandlerFunc(func(rw ResponseWriter, req *Request) {
  1121  		defer close(done)
  1122  		if conn.readBuf.Len() < len(body)/2 {
  1123  			t.Errorf("on request, read buffer length is %d; expected about 100 KB", conn.readBuf.Len())
  1124  		}
  1125  		rw.WriteHeader(200)
  1126  		rw.(Flusher).Flush()
  1127  		if g, e := conn.readBuf.Len(), 0; g != e {
  1128  			t.Errorf("after WriteHeader, read buffer length is %d; want %d", g, e)
  1129  		}
  1130  		if c := rw.Header().Get("Connection"); c != "" {
  1131  			t.Errorf(`Connection header = %q; want ""`, c)
  1132  		}
  1133  	}))
  1134  	<-done
  1135  }
  1136  
  1137  // Over a ~256KB (maxPostHandlerReadBytes) threshold, the server
  1138  // should ignore client request bodies that a handler didn't read
  1139  // and close the connection.
  1140  func TestServerUnreadRequestBodyLarge(t *testing.T) {
  1141  	conn := new(testConn)
  1142  	body := strings.Repeat("x", 1<<20)
  1143  	conn.readBuf.Write([]byte(fmt.Sprintf(
  1144  		"POST / HTTP/1.1\r\n"+
  1145  			"Host: test\r\n"+
  1146  			"Content-Length: %d\r\n"+
  1147  			"\r\n", len(body))))
  1148  	conn.readBuf.Write([]byte(body))
  1149  	conn.closec = make(chan bool, 1)
  1150  
  1151  	ls := &oneConnListener{conn}
  1152  	go Serve(ls, HandlerFunc(func(rw ResponseWriter, req *Request) {
  1153  		if conn.readBuf.Len() < len(body)/2 {
  1154  			t.Errorf("on request, read buffer length is %d; expected about 1MB", conn.readBuf.Len())
  1155  		}
  1156  		rw.WriteHeader(200)
  1157  		rw.(Flusher).Flush()
  1158  		if conn.readBuf.Len() < len(body)/2 {
  1159  			t.Errorf("post-WriteHeader, read buffer length is %d; expected about 1MB", conn.readBuf.Len())
  1160  		}
  1161  	}))
  1162  	<-conn.closec
  1163  
  1164  	if res := conn.writeBuf.String(); !strings.Contains(res, "Connection: close") {
  1165  		t.Errorf("Expected a Connection: close header; got response: %s", res)
  1166  	}
  1167  }
  1168  
  1169  func TestTimeoutHandler(t *testing.T) {
  1170  	defer afterTest(t)
  1171  	sendHi := make(chan bool, 1)
  1172  	writeErrors := make(chan error, 1)
  1173  	sayHi := HandlerFunc(func(w ResponseWriter, r *Request) {
  1174  		<-sendHi
  1175  		_, werr := w.Write([]byte("hi"))
  1176  		writeErrors <- werr
  1177  	})
  1178  	timeout := make(chan time.Time, 1) // write to this to force timeouts
  1179  	ts := httptest.NewServer(NewTestTimeoutHandler(sayHi, timeout))
  1180  	defer ts.Close()
  1181  
  1182  	// Succeed without timing out:
  1183  	sendHi <- true
  1184  	res, err := Get(ts.URL)
  1185  	if err != nil {
  1186  		t.Error(err)
  1187  	}
  1188  	if g, e := res.StatusCode, StatusOK; g != e {
  1189  		t.Errorf("got res.StatusCode %d; expected %d", g, e)
  1190  	}
  1191  	body, _ := ioutil.ReadAll(res.Body)
  1192  	if g, e := string(body), "hi"; g != e {
  1193  		t.Errorf("got body %q; expected %q", g, e)
  1194  	}
  1195  	if g := <-writeErrors; g != nil {
  1196  		t.Errorf("got unexpected Write error on first request: %v", g)
  1197  	}
  1198  
  1199  	// Times out:
  1200  	timeout <- time.Time{}
  1201  	res, err = Get(ts.URL)
  1202  	if err != nil {
  1203  		t.Error(err)
  1204  	}
  1205  	if g, e := res.StatusCode, StatusServiceUnavailable; g != e {
  1206  		t.Errorf("got res.StatusCode %d; expected %d", g, e)
  1207  	}
  1208  	body, _ = ioutil.ReadAll(res.Body)
  1209  	if !strings.Contains(string(body), "<title>Timeout</title>") {
  1210  		t.Errorf("expected timeout body; got %q", string(body))
  1211  	}
  1212  
  1213  	// Now make the previously-timed out handler speak again,
  1214  	// which verifies the panic is handled:
  1215  	sendHi <- true
  1216  	if g, e := <-writeErrors, ErrHandlerTimeout; g != e {
  1217  		t.Errorf("expected Write error of %v; got %v", e, g)
  1218  	}
  1219  }
  1220  
  1221  // See issues 8209 and 8414.
  1222  func TestTimeoutHandlerRace(t *testing.T) {
  1223  	defer afterTest(t)
  1224  
  1225  	delayHi := HandlerFunc(func(w ResponseWriter, r *Request) {
  1226  		ms, _ := strconv.Atoi(r.URL.Path[1:])
  1227  		if ms == 0 {
  1228  			ms = 1
  1229  		}
  1230  		for i := 0; i < ms; i++ {
  1231  			w.Write([]byte("hi"))
  1232  			time.Sleep(time.Millisecond)
  1233  		}
  1234  	})
  1235  
  1236  	ts := httptest.NewServer(TimeoutHandler(delayHi, 20*time.Millisecond, ""))
  1237  	defer ts.Close()
  1238  
  1239  	var wg sync.WaitGroup
  1240  	gate := make(chan bool, 10)
  1241  	n := 50
  1242  	if testing.Short() {
  1243  		n = 10
  1244  		gate = make(chan bool, 3)
  1245  	}
  1246  	for i := 0; i < n; i++ {
  1247  		gate <- true
  1248  		wg.Add(1)
  1249  		go func() {
  1250  			defer wg.Done()
  1251  			defer func() { <-gate }()
  1252  			res, err := Get(fmt.Sprintf("%s/%d", ts.URL, rand.Intn(50)))
  1253  			if err == nil {
  1254  				io.Copy(ioutil.Discard, res.Body)
  1255  				res.Body.Close()
  1256  			}
  1257  		}()
  1258  	}
  1259  	wg.Wait()
  1260  }
  1261  
  1262  // See issues 8209 and 8414.
  1263  func TestTimeoutHandlerRaceHeader(t *testing.T) {
  1264  	defer afterTest(t)
  1265  
  1266  	delay204 := HandlerFunc(func(w ResponseWriter, r *Request) {
  1267  		w.WriteHeader(204)
  1268  	})
  1269  
  1270  	ts := httptest.NewServer(TimeoutHandler(delay204, time.Nanosecond, ""))
  1271  	defer ts.Close()
  1272  
  1273  	var wg sync.WaitGroup
  1274  	gate := make(chan bool, 50)
  1275  	n := 500
  1276  	if testing.Short() {
  1277  		n = 10
  1278  	}
  1279  	for i := 0; i < n; i++ {
  1280  		gate <- true
  1281  		wg.Add(1)
  1282  		go func() {
  1283  			defer wg.Done()
  1284  			defer func() { <-gate }()
  1285  			res, err := Get(ts.URL)
  1286  			if err != nil {
  1287  				t.Error(err)
  1288  				return
  1289  			}
  1290  			defer res.Body.Close()
  1291  			io.Copy(ioutil.Discard, res.Body)
  1292  		}()
  1293  	}
  1294  	wg.Wait()
  1295  }
  1296  
  1297  // Verifies we don't path.Clean() on the wrong parts in redirects.
  1298  func TestRedirectMunging(t *testing.T) {
  1299  	req, _ := NewRequest("GET", "http://example.com/", nil)
  1300  
  1301  	resp := httptest.NewRecorder()
  1302  	Redirect(resp, req, "/foo?next=http://bar.com/", 302)
  1303  	if g, e := resp.Header().Get("Location"), "/foo?next=http://bar.com/"; g != e {
  1304  		t.Errorf("Location header was %q; want %q", g, e)
  1305  	}
  1306  
  1307  	resp = httptest.NewRecorder()
  1308  	Redirect(resp, req, "http://localhost:8080/_ah/login?continue=http://localhost:8080/", 302)
  1309  	if g, e := resp.Header().Get("Location"), "http://localhost:8080/_ah/login?continue=http://localhost:8080/"; g != e {
  1310  		t.Errorf("Location header was %q; want %q", g, e)
  1311  	}
  1312  }
  1313  
  1314  func TestRedirectBadPath(t *testing.T) {
  1315  	// This used to crash. It's not valid input (bad path), but it
  1316  	// shouldn't crash.
  1317  	rr := httptest.NewRecorder()
  1318  	req := &Request{
  1319  		Method: "GET",
  1320  		URL: &url.URL{
  1321  			Scheme: "http",
  1322  			Path:   "not-empty-but-no-leading-slash", // bogus
  1323  		},
  1324  	}
  1325  	Redirect(rr, req, "", 304)
  1326  	if rr.Code != 304 {
  1327  		t.Errorf("Code = %d; want 304", rr.Code)
  1328  	}
  1329  }
  1330  
  1331  // TestZeroLengthPostAndResponse exercises an optimization done by the Transport:
  1332  // when there is no body (either because the method doesn't permit a body, or an
  1333  // explicit Content-Length of zero is present), then the transport can re-use the
  1334  // connection immediately. But when it re-uses the connection, it typically closes
  1335  // the previous request's body, which is not optimal for zero-lengthed bodies,
  1336  // as the client would then see http.ErrBodyReadAfterClose and not 0, io.EOF.
  1337  func TestZeroLengthPostAndResponse(t *testing.T) {
  1338  	defer afterTest(t)
  1339  	ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, r *Request) {
  1340  		all, err := ioutil.ReadAll(r.Body)
  1341  		if err != nil {
  1342  			t.Fatalf("handler ReadAll: %v", err)
  1343  		}
  1344  		if len(all) != 0 {
  1345  			t.Errorf("handler got %d bytes; expected 0", len(all))
  1346  		}
  1347  		rw.Header().Set("Content-Length", "0")
  1348  	}))
  1349  	defer ts.Close()
  1350  
  1351  	req, err := NewRequest("POST", ts.URL, strings.NewReader(""))
  1352  	if err != nil {
  1353  		t.Fatal(err)
  1354  	}
  1355  	req.ContentLength = 0
  1356  
  1357  	var resp [5]*Response
  1358  	for i := range resp {
  1359  		resp[i], err = DefaultClient.Do(req)
  1360  		if err != nil {
  1361  			t.Fatalf("client post #%d: %v", i, err)
  1362  		}
  1363  	}
  1364  
  1365  	for i := range resp {
  1366  		all, err := ioutil.ReadAll(resp[i].Body)
  1367  		if err != nil {
  1368  			t.Fatalf("req #%d: client ReadAll: %v", i, err)
  1369  		}
  1370  		if len(all) != 0 {
  1371  			t.Errorf("req #%d: client got %d bytes; expected 0", i, len(all))
  1372  		}
  1373  	}
  1374  }
  1375  
  1376  func TestHandlerPanicNil(t *testing.T) {
  1377  	testHandlerPanic(t, false, nil)
  1378  }
  1379  
  1380  func TestHandlerPanic(t *testing.T) {
  1381  	testHandlerPanic(t, false, "intentional death for testing")
  1382  }
  1383  
  1384  func TestHandlerPanicWithHijack(t *testing.T) {
  1385  	testHandlerPanic(t, true, "intentional death for testing")
  1386  }
  1387  
  1388  func testHandlerPanic(t *testing.T, withHijack bool, panicValue interface{}) {
  1389  	defer afterTest(t)
  1390  	// Unlike the other tests that set the log output to ioutil.Discard
  1391  	// to quiet the output, this test uses a pipe.  The pipe serves three
  1392  	// purposes:
  1393  	//
  1394  	//   1) The log.Print from the http server (generated by the caught
  1395  	//      panic) will go to the pipe instead of stderr, making the
  1396  	//      output quiet.
  1397  	//
  1398  	//   2) We read from the pipe to verify that the handler
  1399  	//      actually caught the panic and logged something.
  1400  	//
  1401  	//   3) The blocking Read call prevents this TestHandlerPanic
  1402  	//      function from exiting before the HTTP server handler
  1403  	//      finishes crashing. If this text function exited too
  1404  	//      early (and its defer log.SetOutput(os.Stderr) ran),
  1405  	//      then the crash output could spill into the next test.
  1406  	pr, pw := io.Pipe()
  1407  	log.SetOutput(pw)
  1408  	defer log.SetOutput(os.Stderr)
  1409  	defer pw.Close()
  1410  
  1411  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  1412  		if withHijack {
  1413  			rwc, _, err := w.(Hijacker).Hijack()
  1414  			if err != nil {
  1415  				t.Logf("unexpected error: %v", err)
  1416  			}
  1417  			defer rwc.Close()
  1418  		}
  1419  		panic(panicValue)
  1420  	}))
  1421  	defer ts.Close()
  1422  
  1423  	// Do a blocking read on the log output pipe so its logging
  1424  	// doesn't bleed into the next test.  But wait only 5 seconds
  1425  	// for it.
  1426  	done := make(chan bool, 1)
  1427  	go func() {
  1428  		buf := make([]byte, 4<<10)
  1429  		_, err := pr.Read(buf)
  1430  		pr.Close()
  1431  		if err != nil && err != io.EOF {
  1432  			t.Error(err)
  1433  		}
  1434  		done <- true
  1435  	}()
  1436  
  1437  	_, err := Get(ts.URL)
  1438  	if err == nil {
  1439  		t.Logf("expected an error")
  1440  	}
  1441  
  1442  	if panicValue == nil {
  1443  		return
  1444  	}
  1445  
  1446  	select {
  1447  	case <-done:
  1448  		return
  1449  	case <-time.After(5 * time.Second):
  1450  		t.Fatal("expected server handler to log an error")
  1451  	}
  1452  }
  1453  
  1454  func TestNoDate(t *testing.T) {
  1455  	defer afterTest(t)
  1456  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  1457  		w.Header()["Date"] = nil
  1458  	}))
  1459  	defer ts.Close()
  1460  	res, err := Get(ts.URL)
  1461  	if err != nil {
  1462  		t.Fatal(err)
  1463  	}
  1464  	_, present := res.Header["Date"]
  1465  	if present {
  1466  		t.Fatalf("Expected no Date header; got %v", res.Header["Date"])
  1467  	}
  1468  }
  1469  
  1470  func TestStripPrefix(t *testing.T) {
  1471  	defer afterTest(t)
  1472  	h := HandlerFunc(func(w ResponseWriter, r *Request) {
  1473  		w.Header().Set("X-Path", r.URL.Path)
  1474  	})
  1475  	ts := httptest.NewServer(StripPrefix("/foo", h))
  1476  	defer ts.Close()
  1477  
  1478  	res, err := Get(ts.URL + "/foo/bar")
  1479  	if err != nil {
  1480  		t.Fatal(err)
  1481  	}
  1482  	if g, e := res.Header.Get("X-Path"), "/bar"; g != e {
  1483  		t.Errorf("test 1: got %s, want %s", g, e)
  1484  	}
  1485  	res.Body.Close()
  1486  
  1487  	res, err = Get(ts.URL + "/bar")
  1488  	if err != nil {
  1489  		t.Fatal(err)
  1490  	}
  1491  	if g, e := res.StatusCode, 404; g != e {
  1492  		t.Errorf("test 2: got status %v, want %v", g, e)
  1493  	}
  1494  	res.Body.Close()
  1495  }
  1496  
  1497  func TestRequestLimit(t *testing.T) {
  1498  	defer afterTest(t)
  1499  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  1500  		t.Fatalf("didn't expect to get request in Handler")
  1501  	}))
  1502  	defer ts.Close()
  1503  	req, _ := NewRequest("GET", ts.URL, nil)
  1504  	var bytesPerHeader = len("header12345: val12345\r\n")
  1505  	for i := 0; i < ((DefaultMaxHeaderBytes+4096)/bytesPerHeader)+1; i++ {
  1506  		req.Header.Set(fmt.Sprintf("header%05d", i), fmt.Sprintf("val%05d", i))
  1507  	}
  1508  	res, err := DefaultClient.Do(req)
  1509  	if err != nil {
  1510  		// Some HTTP clients may fail on this undefined behavior (server replying and
  1511  		// closing the connection while the request is still being written), but
  1512  		// we do support it (at least currently), so we expect a response below.
  1513  		t.Fatalf("Do: %v", err)
  1514  	}
  1515  	defer res.Body.Close()
  1516  	if res.StatusCode != 413 {
  1517  		t.Fatalf("expected 413 response status; got: %d %s", res.StatusCode, res.Status)
  1518  	}
  1519  }
  1520  
  1521  type neverEnding byte
  1522  
  1523  func (b neverEnding) Read(p []byte) (n int, err error) {
  1524  	for i := range p {
  1525  		p[i] = byte(b)
  1526  	}
  1527  	return len(p), nil
  1528  }
  1529  
  1530  type countReader struct {
  1531  	r io.Reader
  1532  	n *int64
  1533  }
  1534  
  1535  func (cr countReader) Read(p []byte) (n int, err error) {
  1536  	n, err = cr.r.Read(p)
  1537  	atomic.AddInt64(cr.n, int64(n))
  1538  	return
  1539  }
  1540  
  1541  func TestRequestBodyLimit(t *testing.T) {
  1542  	defer afterTest(t)
  1543  	const limit = 1 << 20
  1544  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  1545  		r.Body = MaxBytesReader(w, r.Body, limit)
  1546  		n, err := io.Copy(ioutil.Discard, r.Body)
  1547  		if err == nil {
  1548  			t.Errorf("expected error from io.Copy")
  1549  		}
  1550  		if n != limit {
  1551  			t.Errorf("io.Copy = %d, want %d", n, limit)
  1552  		}
  1553  	}))
  1554  	defer ts.Close()
  1555  
  1556  	nWritten := new(int64)
  1557  	req, _ := NewRequest("POST", ts.URL, io.LimitReader(countReader{neverEnding('a'), nWritten}, limit*200))
  1558  
  1559  	// Send the POST, but don't care it succeeds or not.  The
  1560  	// remote side is going to reply and then close the TCP
  1561  	// connection, and HTTP doesn't really define if that's
  1562  	// allowed or not.  Some HTTP clients will get the response
  1563  	// and some (like ours, currently) will complain that the
  1564  	// request write failed, without reading the response.
  1565  	//
  1566  	// But that's okay, since what we're really testing is that
  1567  	// the remote side hung up on us before we wrote too much.
  1568  	_, _ = DefaultClient.Do(req)
  1569  
  1570  	if atomic.LoadInt64(nWritten) > limit*100 {
  1571  		t.Errorf("handler restricted the request body to %d bytes, but client managed to write %d",
  1572  			limit, nWritten)
  1573  	}
  1574  }
  1575  
  1576  // TestClientWriteShutdown tests that if the client shuts down the write
  1577  // side of their TCP connection, the server doesn't send a 400 Bad Request.
  1578  func TestClientWriteShutdown(t *testing.T) {
  1579  	if runtime.GOOS == "plan9" {
  1580  		t.Skip("skipping test; see http://golang.org/issue/7237")
  1581  	}
  1582  	defer afterTest(t)
  1583  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {}))
  1584  	defer ts.Close()
  1585  	conn, err := net.Dial("tcp", ts.Listener.Addr().String())
  1586  	if err != nil {
  1587  		t.Fatalf("Dial: %v", err)
  1588  	}
  1589  	err = conn.(*net.TCPConn).CloseWrite()
  1590  	if err != nil {
  1591  		t.Fatalf("Dial: %v", err)
  1592  	}
  1593  	donec := make(chan bool)
  1594  	go func() {
  1595  		defer close(donec)
  1596  		bs, err := ioutil.ReadAll(conn)
  1597  		if err != nil {
  1598  			t.Fatalf("ReadAll: %v", err)
  1599  		}
  1600  		got := string(bs)
  1601  		if got != "" {
  1602  			t.Errorf("read %q from server; want nothing", got)
  1603  		}
  1604  	}()
  1605  	select {
  1606  	case <-donec:
  1607  	case <-time.After(10 * time.Second):
  1608  		t.Fatalf("timeout")
  1609  	}
  1610  }
  1611  
  1612  // Tests that chunked server responses that write 1 byte at a time are
  1613  // buffered before chunk headers are added, not after chunk headers.
  1614  func TestServerBufferedChunking(t *testing.T) {
  1615  	conn := new(testConn)
  1616  	conn.readBuf.Write([]byte("GET / HTTP/1.1\r\n\r\n"))
  1617  	conn.closec = make(chan bool, 1)
  1618  	ls := &oneConnListener{conn}
  1619  	go Serve(ls, HandlerFunc(func(rw ResponseWriter, req *Request) {
  1620  		rw.(Flusher).Flush() // force the Header to be sent, in chunking mode, not counting the length
  1621  		rw.Write([]byte{'x'})
  1622  		rw.Write([]byte{'y'})
  1623  		rw.Write([]byte{'z'})
  1624  	}))
  1625  	<-conn.closec
  1626  	if !bytes.HasSuffix(conn.writeBuf.Bytes(), []byte("\r\n\r\n3\r\nxyz\r\n0\r\n\r\n")) {
  1627  		t.Errorf("response didn't end with a single 3 byte 'xyz' chunk; got:\n%q",
  1628  			conn.writeBuf.Bytes())
  1629  	}
  1630  }
  1631  
  1632  // Tests that the server flushes its response headers out when it's
  1633  // ignoring the response body and waits a bit before forcefully
  1634  // closing the TCP connection, causing the client to get a RST.
  1635  // See http://golang.org/issue/3595
  1636  func TestServerGracefulClose(t *testing.T) {
  1637  	defer afterTest(t)
  1638  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  1639  		Error(w, "bye", StatusUnauthorized)
  1640  	}))
  1641  	defer ts.Close()
  1642  
  1643  	conn, err := net.Dial("tcp", ts.Listener.Addr().String())
  1644  	if err != nil {
  1645  		t.Fatal(err)
  1646  	}
  1647  	defer conn.Close()
  1648  	const bodySize = 5 << 20
  1649  	req := []byte(fmt.Sprintf("POST / HTTP/1.1\r\nHost: foo.com\r\nContent-Length: %d\r\n\r\n", bodySize))
  1650  	for i := 0; i < bodySize; i++ {
  1651  		req = append(req, 'x')
  1652  	}
  1653  	writeErr := make(chan error)
  1654  	go func() {
  1655  		_, err := conn.Write(req)
  1656  		writeErr <- err
  1657  	}()
  1658  	br := bufio.NewReader(conn)
  1659  	lineNum := 0
  1660  	for {
  1661  		line, err := br.ReadString('\n')
  1662  		if err == io.EOF {
  1663  			break
  1664  		}
  1665  		if err != nil {
  1666  			t.Fatalf("ReadLine: %v", err)
  1667  		}
  1668  		lineNum++
  1669  		if lineNum == 1 && !strings.Contains(line, "401 Unauthorized") {
  1670  			t.Errorf("Response line = %q; want a 401", line)
  1671  		}
  1672  	}
  1673  	// Wait for write to finish. This is a broken pipe on both
  1674  	// Darwin and Linux, but checking this isn't the point of
  1675  	// the test.
  1676  	<-writeErr
  1677  }
  1678  
  1679  func TestCaseSensitiveMethod(t *testing.T) {
  1680  	defer afterTest(t)
  1681  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  1682  		if r.Method != "get" {
  1683  			t.Errorf(`Got method %q; want "get"`, r.Method)
  1684  		}
  1685  	}))
  1686  	defer ts.Close()
  1687  	req, _ := NewRequest("get", ts.URL, nil)
  1688  	res, err := DefaultClient.Do(req)
  1689  	if err != nil {
  1690  		t.Error(err)
  1691  		return
  1692  	}
  1693  	res.Body.Close()
  1694  }
  1695  
  1696  // TestContentLengthZero tests that for both an HTTP/1.0 and HTTP/1.1
  1697  // request (both keep-alive), when a Handler never writes any
  1698  // response, the net/http package adds a "Content-Length: 0" response
  1699  // header.
  1700  func TestContentLengthZero(t *testing.T) {
  1701  	ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, req *Request) {}))
  1702  	defer ts.Close()
  1703  
  1704  	for _, version := range []string{"HTTP/1.0", "HTTP/1.1"} {
  1705  		conn, err := net.Dial("tcp", ts.Listener.Addr().String())
  1706  		if err != nil {
  1707  			t.Fatalf("error dialing: %v", err)
  1708  		}
  1709  		_, err = fmt.Fprintf(conn, "GET / %v\r\nConnection: keep-alive\r\nHost: foo\r\n\r\n", version)
  1710  		if err != nil {
  1711  			t.Fatalf("error writing: %v", err)
  1712  		}
  1713  		req, _ := NewRequest("GET", "/", nil)
  1714  		res, err := ReadResponse(bufio.NewReader(conn), req)
  1715  		if err != nil {
  1716  			t.Fatalf("error reading response: %v", err)
  1717  		}
  1718  		if te := res.TransferEncoding; len(te) > 0 {
  1719  			t.Errorf("For version %q, Transfer-Encoding = %q; want none", version, te)
  1720  		}
  1721  		if cl := res.ContentLength; cl != 0 {
  1722  			t.Errorf("For version %q, Content-Length = %v; want 0", version, cl)
  1723  		}
  1724  		conn.Close()
  1725  	}
  1726  }
  1727  
  1728  func TestCloseNotifier(t *testing.T) {
  1729  	defer afterTest(t)
  1730  	gotReq := make(chan bool, 1)
  1731  	sawClose := make(chan bool, 1)
  1732  	ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, req *Request) {
  1733  		gotReq <- true
  1734  		cc := rw.(CloseNotifier).CloseNotify()
  1735  		<-cc
  1736  		sawClose <- true
  1737  	}))
  1738  	conn, err := net.Dial("tcp", ts.Listener.Addr().String())
  1739  	if err != nil {
  1740  		t.Fatalf("error dialing: %v", err)
  1741  	}
  1742  	diec := make(chan bool)
  1743  	go func() {
  1744  		_, err = fmt.Fprintf(conn, "GET / HTTP/1.1\r\nConnection: keep-alive\r\nHost: foo\r\n\r\n")
  1745  		if err != nil {
  1746  			t.Fatal(err)
  1747  		}
  1748  		<-diec
  1749  		conn.Close()
  1750  	}()
  1751  For:
  1752  	for {
  1753  		select {
  1754  		case <-gotReq:
  1755  			diec <- true
  1756  		case <-sawClose:
  1757  			break For
  1758  		case <-time.After(5 * time.Second):
  1759  			t.Fatal("timeout")
  1760  		}
  1761  	}
  1762  	ts.Close()
  1763  }
  1764  
  1765  func TestCloseNotifierChanLeak(t *testing.T) {
  1766  	defer afterTest(t)
  1767  	req := reqBytes("GET / HTTP/1.0\nHost: golang.org")
  1768  	for i := 0; i < 20; i++ {
  1769  		var output bytes.Buffer
  1770  		conn := &rwTestConn{
  1771  			Reader: bytes.NewReader(req),
  1772  			Writer: &output,
  1773  			closec: make(chan bool, 1),
  1774  		}
  1775  		ln := &oneConnListener{conn: conn}
  1776  		handler := HandlerFunc(func(rw ResponseWriter, r *Request) {
  1777  			// Ignore the return value and never read from
  1778  			// it, testing that we don't leak goroutines
  1779  			// on the sending side:
  1780  			_ = rw.(CloseNotifier).CloseNotify()
  1781  		})
  1782  		go Serve(ln, handler)
  1783  		<-conn.closec
  1784  	}
  1785  }
  1786  
  1787  func TestOptions(t *testing.T) {
  1788  	uric := make(chan string, 2) // only expect 1, but leave space for 2
  1789  	mux := NewServeMux()
  1790  	mux.HandleFunc("/", func(w ResponseWriter, r *Request) {
  1791  		uric <- r.RequestURI
  1792  	})
  1793  	ts := httptest.NewServer(mux)
  1794  	defer ts.Close()
  1795  
  1796  	conn, err := net.Dial("tcp", ts.Listener.Addr().String())
  1797  	if err != nil {
  1798  		t.Fatal(err)
  1799  	}
  1800  	defer conn.Close()
  1801  
  1802  	// An OPTIONS * request should succeed.
  1803  	_, err = conn.Write([]byte("OPTIONS * HTTP/1.1\r\nHost: foo.com\r\n\r\n"))
  1804  	if err != nil {
  1805  		t.Fatal(err)
  1806  	}
  1807  	br := bufio.NewReader(conn)
  1808  	res, err := ReadResponse(br, &Request{Method: "OPTIONS"})
  1809  	if err != nil {
  1810  		t.Fatal(err)
  1811  	}
  1812  	if res.StatusCode != 200 {
  1813  		t.Errorf("Got non-200 response to OPTIONS *: %#v", res)
  1814  	}
  1815  
  1816  	// A GET * request on a ServeMux should fail.
  1817  	_, err = conn.Write([]byte("GET * HTTP/1.1\r\nHost: foo.com\r\n\r\n"))
  1818  	if err != nil {
  1819  		t.Fatal(err)
  1820  	}
  1821  	res, err = ReadResponse(br, &Request{Method: "GET"})
  1822  	if err != nil {
  1823  		t.Fatal(err)
  1824  	}
  1825  	if res.StatusCode != 400 {
  1826  		t.Errorf("Got non-400 response to GET *: %#v", res)
  1827  	}
  1828  
  1829  	res, err = Get(ts.URL + "/second")
  1830  	if err != nil {
  1831  		t.Fatal(err)
  1832  	}
  1833  	res.Body.Close()
  1834  	if got := <-uric; got != "/second" {
  1835  		t.Errorf("Handler saw request for %q; want /second", got)
  1836  	}
  1837  }
  1838  
  1839  // Tests regarding the ordering of Write, WriteHeader, Header, and
  1840  // Flush calls.  In Go 1.0, rw.WriteHeader immediately flushed the
  1841  // (*response).header to the wire. In Go 1.1, the actual wire flush is
  1842  // delayed, so we could maybe tack on a Content-Length and better
  1843  // Content-Type after we see more (or all) of the output. To preserve
  1844  // compatibility with Go 1, we need to be careful to track which
  1845  // headers were live at the time of WriteHeader, so we write the same
  1846  // ones, even if the handler modifies them (~erroneously) after the
  1847  // first Write.
  1848  func TestHeaderToWire(t *testing.T) {
  1849  	tests := []struct {
  1850  		name    string
  1851  		handler func(ResponseWriter, *Request)
  1852  		check   func(output string) error
  1853  	}{
  1854  		{
  1855  			name: "write without Header",
  1856  			handler: func(rw ResponseWriter, r *Request) {
  1857  				rw.Write([]byte("hello world"))
  1858  			},
  1859  			check: func(got string) error {
  1860  				if !strings.Contains(got, "Content-Length:") {
  1861  					return errors.New("no content-length")
  1862  				}
  1863  				if !strings.Contains(got, "Content-Type: text/plain") {
  1864  					return errors.New("no content-length")
  1865  				}
  1866  				return nil
  1867  			},
  1868  		},
  1869  		{
  1870  			name: "Header mutation before write",
  1871  			handler: func(rw ResponseWriter, r *Request) {
  1872  				h := rw.Header()
  1873  				h.Set("Content-Type", "some/type")
  1874  				rw.Write([]byte("hello world"))
  1875  				h.Set("Too-Late", "bogus")
  1876  			},
  1877  			check: func(got string) error {
  1878  				if !strings.Contains(got, "Content-Length:") {
  1879  					return errors.New("no content-length")
  1880  				}
  1881  				if !strings.Contains(got, "Content-Type: some/type") {
  1882  					return errors.New("wrong content-type")
  1883  				}
  1884  				if strings.Contains(got, "Too-Late") {
  1885  					return errors.New("don't want too-late header")
  1886  				}
  1887  				return nil
  1888  			},
  1889  		},
  1890  		{
  1891  			name: "write then useless Header mutation",
  1892  			handler: func(rw ResponseWriter, r *Request) {
  1893  				rw.Write([]byte("hello world"))
  1894  				rw.Header().Set("Too-Late", "Write already wrote headers")
  1895  			},
  1896  			check: func(got string) error {
  1897  				if strings.Contains(got, "Too-Late") {
  1898  					return errors.New("header appeared from after WriteHeader")
  1899  				}
  1900  				return nil
  1901  			},
  1902  		},
  1903  		{
  1904  			name: "flush then write",
  1905  			handler: func(rw ResponseWriter, r *Request) {
  1906  				rw.(Flusher).Flush()
  1907  				rw.Write([]byte("post-flush"))
  1908  				rw.Header().Set("Too-Late", "Write already wrote headers")
  1909  			},
  1910  			check: func(got string) error {
  1911  				if !strings.Contains(got, "Transfer-Encoding: chunked") {
  1912  					return errors.New("not chunked")
  1913  				}
  1914  				if strings.Contains(got, "Too-Late") {
  1915  					return errors.New("header appeared from after WriteHeader")
  1916  				}
  1917  				return nil
  1918  			},
  1919  		},
  1920  		{
  1921  			name: "header then flush",
  1922  			handler: func(rw ResponseWriter, r *Request) {
  1923  				rw.Header().Set("Content-Type", "some/type")
  1924  				rw.(Flusher).Flush()
  1925  				rw.Write([]byte("post-flush"))
  1926  				rw.Header().Set("Too-Late", "Write already wrote headers")
  1927  			},
  1928  			check: func(got string) error {
  1929  				if !strings.Contains(got, "Transfer-Encoding: chunked") {
  1930  					return errors.New("not chunked")
  1931  				}
  1932  				if strings.Contains(got, "Too-Late") {
  1933  					return errors.New("header appeared from after WriteHeader")
  1934  				}
  1935  				if !strings.Contains(got, "Content-Type: some/type") {
  1936  					return errors.New("wrong content-length")
  1937  				}
  1938  				return nil
  1939  			},
  1940  		},
  1941  		{
  1942  			name: "sniff-on-first-write content-type",
  1943  			handler: func(rw ResponseWriter, r *Request) {
  1944  				rw.Write([]byte("<html><head></head><body>some html</body></html>"))
  1945  				rw.Header().Set("Content-Type", "x/wrong")
  1946  			},
  1947  			check: func(got string) error {
  1948  				if !strings.Contains(got, "Content-Type: text/html") {
  1949  					return errors.New("wrong content-length; want html")
  1950  				}
  1951  				return nil
  1952  			},
  1953  		},
  1954  		{
  1955  			name: "explicit content-type wins",
  1956  			handler: func(rw ResponseWriter, r *Request) {
  1957  				rw.Header().Set("Content-Type", "some/type")
  1958  				rw.Write([]byte("<html><head></head><body>some html</body></html>"))
  1959  			},
  1960  			check: func(got string) error {
  1961  				if !strings.Contains(got, "Content-Type: some/type") {
  1962  					return errors.New("wrong content-length; want html")
  1963  				}
  1964  				return nil
  1965  			},
  1966  		},
  1967  		{
  1968  			name: "empty handler",
  1969  			handler: func(rw ResponseWriter, r *Request) {
  1970  			},
  1971  			check: func(got string) error {
  1972  				if !strings.Contains(got, "Content-Type: text/plain") {
  1973  					return errors.New("wrong content-length; want text/plain")
  1974  				}
  1975  				if !strings.Contains(got, "Content-Length: 0") {
  1976  					return errors.New("want 0 content-length")
  1977  				}
  1978  				return nil
  1979  			},
  1980  		},
  1981  		{
  1982  			name: "only Header, no write",
  1983  			handler: func(rw ResponseWriter, r *Request) {
  1984  				rw.Header().Set("Some-Header", "some-value")
  1985  			},
  1986  			check: func(got string) error {
  1987  				if !strings.Contains(got, "Some-Header") {
  1988  					return errors.New("didn't get header")
  1989  				}
  1990  				return nil
  1991  			},
  1992  		},
  1993  		{
  1994  			name: "WriteHeader call",
  1995  			handler: func(rw ResponseWriter, r *Request) {
  1996  				rw.WriteHeader(404)
  1997  				rw.Header().Set("Too-Late", "some-value")
  1998  			},
  1999  			check: func(got string) error {
  2000  				if !strings.Contains(got, "404") {
  2001  					return errors.New("wrong status")
  2002  				}
  2003  				if strings.Contains(got, "Some-Header") {
  2004  					return errors.New("shouldn't have seen Too-Late")
  2005  				}
  2006  				return nil
  2007  			},
  2008  		},
  2009  	}
  2010  	for _, tc := range tests {
  2011  		ht := newHandlerTest(HandlerFunc(tc.handler))
  2012  		got := ht.rawResponse("GET / HTTP/1.1\nHost: golang.org")
  2013  		if err := tc.check(got); err != nil {
  2014  			t.Errorf("%s: %v\nGot response:\n%s", tc.name, err, got)
  2015  		}
  2016  	}
  2017  }
  2018  
  2019  // goTimeout runs f, failing t if f takes more than ns to complete.
  2020  func goTimeout(t *testing.T, d time.Duration, f func()) {
  2021  	ch := make(chan bool, 2)
  2022  	timer := time.AfterFunc(d, func() {
  2023  		t.Errorf("Timeout expired after %v", d)
  2024  		ch <- true
  2025  	})
  2026  	defer timer.Stop()
  2027  	go func() {
  2028  		defer func() { ch <- true }()
  2029  		f()
  2030  	}()
  2031  	<-ch
  2032  }
  2033  
  2034  type errorListener struct {
  2035  	errs []error
  2036  }
  2037  
  2038  func (l *errorListener) Accept() (c net.Conn, err error) {
  2039  	if len(l.errs) == 0 {
  2040  		return nil, io.EOF
  2041  	}
  2042  	err = l.errs[0]
  2043  	l.errs = l.errs[1:]
  2044  	return
  2045  }
  2046  
  2047  func (l *errorListener) Close() error {
  2048  	return nil
  2049  }
  2050  
  2051  func (l *errorListener) Addr() net.Addr {
  2052  	return dummyAddr("test-address")
  2053  }
  2054  
  2055  func TestAcceptMaxFds(t *testing.T) {
  2056  	log.SetOutput(ioutil.Discard) // is noisy otherwise
  2057  	defer log.SetOutput(os.Stderr)
  2058  
  2059  	ln := &errorListener{[]error{
  2060  		&net.OpError{
  2061  			Op:  "accept",
  2062  			Err: syscall.EMFILE,
  2063  		}}}
  2064  	err := Serve(ln, HandlerFunc(HandlerFunc(func(ResponseWriter, *Request) {})))
  2065  	if err != io.EOF {
  2066  		t.Errorf("got error %v, want EOF", err)
  2067  	}
  2068  }
  2069  
  2070  func TestWriteAfterHijack(t *testing.T) {
  2071  	req := reqBytes("GET / HTTP/1.1\nHost: golang.org")
  2072  	var buf bytes.Buffer
  2073  	wrotec := make(chan bool, 1)
  2074  	conn := &rwTestConn{
  2075  		Reader: bytes.NewReader(req),
  2076  		Writer: &buf,
  2077  		closec: make(chan bool, 1),
  2078  	}
  2079  	handler := HandlerFunc(func(rw ResponseWriter, r *Request) {
  2080  		conn, bufrw, err := rw.(Hijacker).Hijack()
  2081  		if err != nil {
  2082  			t.Error(err)
  2083  			return
  2084  		}
  2085  		go func() {
  2086  			bufrw.Write([]byte("[hijack-to-bufw]"))
  2087  			bufrw.Flush()
  2088  			conn.Write([]byte("[hijack-to-conn]"))
  2089  			conn.Close()
  2090  			wrotec <- true
  2091  		}()
  2092  	})
  2093  	ln := &oneConnListener{conn: conn}
  2094  	go Serve(ln, handler)
  2095  	<-conn.closec
  2096  	<-wrotec
  2097  	if g, w := buf.String(), "[hijack-to-bufw][hijack-to-conn]"; g != w {
  2098  		t.Errorf("wrote %q; want %q", g, w)
  2099  	}
  2100  }
  2101  
  2102  func TestDoubleHijack(t *testing.T) {
  2103  	req := reqBytes("GET / HTTP/1.1\nHost: golang.org")
  2104  	var buf bytes.Buffer
  2105  	conn := &rwTestConn{
  2106  		Reader: bytes.NewReader(req),
  2107  		Writer: &buf,
  2108  		closec: make(chan bool, 1),
  2109  	}
  2110  	handler := HandlerFunc(func(rw ResponseWriter, r *Request) {
  2111  		conn, _, err := rw.(Hijacker).Hijack()
  2112  		if err != nil {
  2113  			t.Error(err)
  2114  			return
  2115  		}
  2116  		_, _, err = rw.(Hijacker).Hijack()
  2117  		if err == nil {
  2118  			t.Errorf("got err = nil;  want err != nil")
  2119  		}
  2120  		conn.Close()
  2121  	})
  2122  	ln := &oneConnListener{conn: conn}
  2123  	go Serve(ln, handler)
  2124  	<-conn.closec
  2125  }
  2126  
  2127  // http://code.google.com/p/go/issues/detail?id=5955
  2128  // Note that this does not test the "request too large"
  2129  // exit path from the http server. This is intentional;
  2130  // not sending Connection: close is just a minor wire
  2131  // optimization and is pointless if dealing with a
  2132  // badly behaved client.
  2133  func TestHTTP10ConnectionHeader(t *testing.T) {
  2134  	defer afterTest(t)
  2135  
  2136  	mux := NewServeMux()
  2137  	mux.Handle("/", HandlerFunc(func(resp ResponseWriter, req *Request) {}))
  2138  	ts := httptest.NewServer(mux)
  2139  	defer ts.Close()
  2140  
  2141  	// net/http uses HTTP/1.1 for requests, so write requests manually
  2142  	tests := []struct {
  2143  		req    string   // raw http request
  2144  		expect []string // expected Connection header(s)
  2145  	}{
  2146  		{
  2147  			req:    "GET / HTTP/1.0\r\n\r\n",
  2148  			expect: nil,
  2149  		},
  2150  		{
  2151  			req:    "OPTIONS * HTTP/1.0\r\n\r\n",
  2152  			expect: nil,
  2153  		},
  2154  		{
  2155  			req:    "GET / HTTP/1.0\r\nConnection: keep-alive\r\n\r\n",
  2156  			expect: []string{"keep-alive"},
  2157  		},
  2158  	}
  2159  
  2160  	for _, tt := range tests {
  2161  		conn, err := net.Dial("tcp", ts.Listener.Addr().String())
  2162  		if err != nil {
  2163  			t.Fatal("dial err:", err)
  2164  		}
  2165  
  2166  		_, err = fmt.Fprint(conn, tt.req)
  2167  		if err != nil {
  2168  			t.Fatal("conn write err:", err)
  2169  		}
  2170  
  2171  		resp, err := ReadResponse(bufio.NewReader(conn), &Request{Method: "GET"})
  2172  		if err != nil {
  2173  			t.Fatal("ReadResponse err:", err)
  2174  		}
  2175  		conn.Close()
  2176  		resp.Body.Close()
  2177  
  2178  		got := resp.Header["Connection"]
  2179  		if !reflect.DeepEqual(got, tt.expect) {
  2180  			t.Errorf("wrong Connection headers for request %q. Got %q expect %q", tt.req, got, tt.expect)
  2181  		}
  2182  	}
  2183  }
  2184  
  2185  // See golang.org/issue/5660
  2186  func TestServerReaderFromOrder(t *testing.T) {
  2187  	defer afterTest(t)
  2188  	pr, pw := io.Pipe()
  2189  	const size = 3 << 20
  2190  	ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, req *Request) {
  2191  		rw.Header().Set("Content-Type", "text/plain") // prevent sniffing path
  2192  		done := make(chan bool)
  2193  		go func() {
  2194  			io.Copy(rw, pr)
  2195  			close(done)
  2196  		}()
  2197  		time.Sleep(25 * time.Millisecond) // give Copy a chance to break things
  2198  		n, err := io.Copy(ioutil.Discard, req.Body)
  2199  		if err != nil {
  2200  			t.Errorf("handler Copy: %v", err)
  2201  			return
  2202  		}
  2203  		if n != size {
  2204  			t.Errorf("handler Copy = %d; want %d", n, size)
  2205  		}
  2206  		pw.Write([]byte("hi"))
  2207  		pw.Close()
  2208  		<-done
  2209  	}))
  2210  	defer ts.Close()
  2211  
  2212  	req, err := NewRequest("POST", ts.URL, io.LimitReader(neverEnding('a'), size))
  2213  	if err != nil {
  2214  		t.Fatal(err)
  2215  	}
  2216  	res, err := DefaultClient.Do(req)
  2217  	if err != nil {
  2218  		t.Fatal(err)
  2219  	}
  2220  	all, err := ioutil.ReadAll(res.Body)
  2221  	if err != nil {
  2222  		t.Fatal(err)
  2223  	}
  2224  	res.Body.Close()
  2225  	if string(all) != "hi" {
  2226  		t.Errorf("Body = %q; want hi", all)
  2227  	}
  2228  }
  2229  
  2230  // Issue 6157, Issue 6685
  2231  func TestCodesPreventingContentTypeAndBody(t *testing.T) {
  2232  	for _, code := range []int{StatusNotModified, StatusNoContent, StatusContinue} {
  2233  		ht := newHandlerTest(HandlerFunc(func(w ResponseWriter, r *Request) {
  2234  			if r.URL.Path == "/header" {
  2235  				w.Header().Set("Content-Length", "123")
  2236  			}
  2237  			w.WriteHeader(code)
  2238  			if r.URL.Path == "/more" {
  2239  				w.Write([]byte("stuff"))
  2240  			}
  2241  		}))
  2242  		for _, req := range []string{
  2243  			"GET / HTTP/1.0",
  2244  			"GET /header HTTP/1.0",
  2245  			"GET /more HTTP/1.0",
  2246  			"GET / HTTP/1.1",
  2247  			"GET /header HTTP/1.1",
  2248  			"GET /more HTTP/1.1",
  2249  		} {
  2250  			got := ht.rawResponse(req)
  2251  			wantStatus := fmt.Sprintf("%d %s", code, StatusText(code))
  2252  			if !strings.Contains(got, wantStatus) {
  2253  				t.Errorf("Code %d: Wanted %q Modified for %q: %s", code, wantStatus, req, got)
  2254  			} else if strings.Contains(got, "Content-Length") {
  2255  				t.Errorf("Code %d: Got a Content-Length from %q: %s", code, req, got)
  2256  			} else if strings.Contains(got, "stuff") {
  2257  				t.Errorf("Code %d: Response contains a body from %q: %s", code, req, got)
  2258  			}
  2259  		}
  2260  	}
  2261  }
  2262  
  2263  func TestContentTypeOkayOn204(t *testing.T) {
  2264  	ht := newHandlerTest(HandlerFunc(func(w ResponseWriter, r *Request) {
  2265  		w.Header().Set("Content-Length", "123") // suppressed
  2266  		w.Header().Set("Content-Type", "foo/bar")
  2267  		w.WriteHeader(204)
  2268  	}))
  2269  	got := ht.rawResponse("GET / HTTP/1.1")
  2270  	if !strings.Contains(got, "Content-Type: foo/bar") {
  2271  		t.Errorf("Response = %q; want Content-Type: foo/bar", got)
  2272  	}
  2273  	if strings.Contains(got, "Content-Length: 123") {
  2274  		t.Errorf("Response = %q; don't want a Content-Length", got)
  2275  	}
  2276  }
  2277  
  2278  // Issue 6995
  2279  // A server Handler can receive a Request, and then turn around and
  2280  // give a copy of that Request.Body out to the Transport (e.g. any
  2281  // proxy).  So then two people own that Request.Body (both the server
  2282  // and the http client), and both think they can close it on failure.
  2283  // Therefore, all incoming server requests Bodies need to be thread-safe.
  2284  func TestTransportAndServerSharedBodyRace(t *testing.T) {
  2285  	defer afterTest(t)
  2286  
  2287  	const bodySize = 1 << 20
  2288  
  2289  	unblockBackend := make(chan bool)
  2290  	backend := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, req *Request) {
  2291  		io.CopyN(rw, req.Body, bodySize/2)
  2292  		<-unblockBackend
  2293  	}))
  2294  	defer backend.Close()
  2295  
  2296  	backendRespc := make(chan *Response, 1)
  2297  	proxy := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, req *Request) {
  2298  		if req.RequestURI == "/foo" {
  2299  			rw.Write([]byte("bar"))
  2300  			return
  2301  		}
  2302  		req2, _ := NewRequest("POST", backend.URL, req.Body)
  2303  		req2.ContentLength = bodySize
  2304  
  2305  		bresp, err := DefaultClient.Do(req2)
  2306  		if err != nil {
  2307  			t.Errorf("Proxy outbound request: %v", err)
  2308  			return
  2309  		}
  2310  		_, err = io.CopyN(ioutil.Discard, bresp.Body, bodySize/4)
  2311  		if err != nil {
  2312  			t.Errorf("Proxy copy error: %v", err)
  2313  			return
  2314  		}
  2315  		backendRespc <- bresp // to close later
  2316  
  2317  		// Try to cause a race: Both the DefaultTransport and the proxy handler's Server
  2318  		// will try to read/close req.Body (aka req2.Body)
  2319  		DefaultTransport.(*Transport).CancelRequest(req2)
  2320  		rw.Write([]byte("OK"))
  2321  	}))
  2322  	defer proxy.Close()
  2323  
  2324  	req, _ := NewRequest("POST", proxy.URL, io.LimitReader(neverEnding('a'), bodySize))
  2325  	res, err := DefaultClient.Do(req)
  2326  	if err != nil {
  2327  		t.Fatalf("Original request: %v", err)
  2328  	}
  2329  
  2330  	// Cleanup, so we don't leak goroutines.
  2331  	res.Body.Close()
  2332  	close(unblockBackend)
  2333  	(<-backendRespc).Body.Close()
  2334  }
  2335  
  2336  // Test that a hanging Request.Body.Read from another goroutine can't
  2337  // cause the Handler goroutine's Request.Body.Close to block.
  2338  func TestRequestBodyCloseDoesntBlock(t *testing.T) {
  2339  	t.Skipf("Skipping known issue; see golang.org/issue/7121")
  2340  	if testing.Short() {
  2341  		t.Skip("skipping in -short mode")
  2342  	}
  2343  	defer afterTest(t)
  2344  
  2345  	readErrCh := make(chan error, 1)
  2346  	errCh := make(chan error, 2)
  2347  
  2348  	server := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, req *Request) {
  2349  		go func(body io.Reader) {
  2350  			_, err := body.Read(make([]byte, 100))
  2351  			readErrCh <- err
  2352  		}(req.Body)
  2353  		time.Sleep(500 * time.Millisecond)
  2354  	}))
  2355  	defer server.Close()
  2356  
  2357  	closeConn := make(chan bool)
  2358  	defer close(closeConn)
  2359  	go func() {
  2360  		conn, err := net.Dial("tcp", server.Listener.Addr().String())
  2361  		if err != nil {
  2362  			errCh <- err
  2363  			return
  2364  		}
  2365  		defer conn.Close()
  2366  		_, err = conn.Write([]byte("POST / HTTP/1.1\r\nConnection: close\r\nHost: foo\r\nContent-Length: 100000\r\n\r\n"))
  2367  		if err != nil {
  2368  			errCh <- err
  2369  			return
  2370  		}
  2371  		// And now just block, making the server block on our
  2372  		// 100000 bytes of body that will never arrive.
  2373  		<-closeConn
  2374  	}()
  2375  	select {
  2376  	case err := <-readErrCh:
  2377  		if err == nil {
  2378  			t.Error("Read was nil. Expected error.")
  2379  		}
  2380  	case err := <-errCh:
  2381  		t.Error(err)
  2382  	case <-time.After(5 * time.Second):
  2383  		t.Error("timeout")
  2384  	}
  2385  }
  2386  
  2387  func TestResponseWriterWriteStringAllocs(t *testing.T) {
  2388  	ht := newHandlerTest(HandlerFunc(func(w ResponseWriter, r *Request) {
  2389  		if r.URL.Path == "/s" {
  2390  			io.WriteString(w, "Hello world")
  2391  		} else {
  2392  			w.Write([]byte("Hello world"))
  2393  		}
  2394  	}))
  2395  	before := testing.AllocsPerRun(50, func() { ht.rawResponse("GET / HTTP/1.0") })
  2396  	after := testing.AllocsPerRun(50, func() { ht.rawResponse("GET /s HTTP/1.0") })
  2397  	if int(after) >= int(before) {
  2398  		t.Errorf("WriteString allocs of %v >= Write allocs of %v", after, before)
  2399  	}
  2400  }
  2401  
  2402  func TestAppendTime(t *testing.T) {
  2403  	var b [len(TimeFormat)]byte
  2404  	t1 := time.Date(2013, 9, 21, 15, 41, 0, 0, time.FixedZone("CEST", 2*60*60))
  2405  	res := ExportAppendTime(b[:0], t1)
  2406  	t2, err := ParseTime(string(res))
  2407  	if err != nil {
  2408  		t.Fatalf("Error parsing time: %s", err)
  2409  	}
  2410  	if !t1.Equal(t2) {
  2411  		t.Fatalf("Times differ; expected: %v, got %v (%s)", t1, t2, string(res))
  2412  	}
  2413  }
  2414  
  2415  func TestServerConnState(t *testing.T) {
  2416  	defer afterTest(t)
  2417  	handler := map[string]func(w ResponseWriter, r *Request){
  2418  		"/": func(w ResponseWriter, r *Request) {
  2419  			fmt.Fprintf(w, "Hello.")
  2420  		},
  2421  		"/close": func(w ResponseWriter, r *Request) {
  2422  			w.Header().Set("Connection", "close")
  2423  			fmt.Fprintf(w, "Hello.")
  2424  		},
  2425  		"/hijack": func(w ResponseWriter, r *Request) {
  2426  			c, _, _ := w.(Hijacker).Hijack()
  2427  			c.Write([]byte("HTTP/1.0 200 OK\r\nConnection: close\r\n\r\nHello."))
  2428  			c.Close()
  2429  		},
  2430  		"/hijack-panic": func(w ResponseWriter, r *Request) {
  2431  			c, _, _ := w.(Hijacker).Hijack()
  2432  			c.Write([]byte("HTTP/1.0 200 OK\r\nConnection: close\r\n\r\nHello."))
  2433  			c.Close()
  2434  			panic("intentional panic")
  2435  		},
  2436  	}
  2437  	ts := httptest.NewUnstartedServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  2438  		handler[r.URL.Path](w, r)
  2439  	}))
  2440  	defer ts.Close()
  2441  
  2442  	var mu sync.Mutex // guard stateLog and connID
  2443  	var stateLog = map[int][]ConnState{}
  2444  	var connID = map[net.Conn]int{}
  2445  
  2446  	ts.Config.ErrorLog = log.New(ioutil.Discard, "", 0)
  2447  	ts.Config.ConnState = func(c net.Conn, state ConnState) {
  2448  		if c == nil {
  2449  			t.Errorf("nil conn seen in state %s", state)
  2450  			return
  2451  		}
  2452  		mu.Lock()
  2453  		defer mu.Unlock()
  2454  		id, ok := connID[c]
  2455  		if !ok {
  2456  			id = len(connID) + 1
  2457  			connID[c] = id
  2458  		}
  2459  		stateLog[id] = append(stateLog[id], state)
  2460  	}
  2461  	ts.Start()
  2462  
  2463  	mustGet(t, ts.URL+"/")
  2464  	mustGet(t, ts.URL+"/close")
  2465  
  2466  	mustGet(t, ts.URL+"/")
  2467  	mustGet(t, ts.URL+"/", "Connection", "close")
  2468  
  2469  	mustGet(t, ts.URL+"/hijack")
  2470  	mustGet(t, ts.URL+"/hijack-panic")
  2471  
  2472  	// New->Closed
  2473  	{
  2474  		c, err := net.Dial("tcp", ts.Listener.Addr().String())
  2475  		if err != nil {
  2476  			t.Fatal(err)
  2477  		}
  2478  		c.Close()
  2479  	}
  2480  
  2481  	// New->Active->Closed
  2482  	{
  2483  		c, err := net.Dial("tcp", ts.Listener.Addr().String())
  2484  		if err != nil {
  2485  			t.Fatal(err)
  2486  		}
  2487  		if _, err := io.WriteString(c, "BOGUS REQUEST\r\n\r\n"); err != nil {
  2488  			t.Fatal(err)
  2489  		}
  2490  		c.Close()
  2491  	}
  2492  
  2493  	// New->Idle->Closed
  2494  	{
  2495  		c, err := net.Dial("tcp", ts.Listener.Addr().String())
  2496  		if err != nil {
  2497  			t.Fatal(err)
  2498  		}
  2499  		if _, err := io.WriteString(c, "GET / HTTP/1.1\r\nHost: foo\r\n\r\n"); err != nil {
  2500  			t.Fatal(err)
  2501  		}
  2502  		res, err := ReadResponse(bufio.NewReader(c), nil)
  2503  		if err != nil {
  2504  			t.Fatal(err)
  2505  		}
  2506  		if _, err := io.Copy(ioutil.Discard, res.Body); err != nil {
  2507  			t.Fatal(err)
  2508  		}
  2509  		c.Close()
  2510  	}
  2511  
  2512  	want := map[int][]ConnState{
  2513  		1: {StateNew, StateActive, StateIdle, StateActive, StateClosed},
  2514  		2: {StateNew, StateActive, StateIdle, StateActive, StateClosed},
  2515  		3: {StateNew, StateActive, StateHijacked},
  2516  		4: {StateNew, StateActive, StateHijacked},
  2517  		5: {StateNew, StateClosed},
  2518  		6: {StateNew, StateActive, StateClosed},
  2519  		7: {StateNew, StateActive, StateIdle, StateClosed},
  2520  	}
  2521  	logString := func(m map[int][]ConnState) string {
  2522  		var b bytes.Buffer
  2523  		for id, l := range m {
  2524  			fmt.Fprintf(&b, "Conn %d: ", id)
  2525  			for _, s := range l {
  2526  				fmt.Fprintf(&b, "%s ", s)
  2527  			}
  2528  			b.WriteString("\n")
  2529  		}
  2530  		return b.String()
  2531  	}
  2532  
  2533  	for i := 0; i < 5; i++ {
  2534  		time.Sleep(time.Duration(i) * 50 * time.Millisecond)
  2535  		mu.Lock()
  2536  		match := reflect.DeepEqual(stateLog, want)
  2537  		mu.Unlock()
  2538  		if match {
  2539  			return
  2540  		}
  2541  	}
  2542  
  2543  	mu.Lock()
  2544  	t.Errorf("Unexpected events.\nGot log: %s\n   Want: %s\n", logString(stateLog), logString(want))
  2545  	mu.Unlock()
  2546  }
  2547  
  2548  func mustGet(t *testing.T, url string, headers ...string) {
  2549  	req, err := NewRequest("GET", url, nil)
  2550  	if err != nil {
  2551  		t.Fatal(err)
  2552  	}
  2553  	for len(headers) > 0 {
  2554  		req.Header.Add(headers[0], headers[1])
  2555  		headers = headers[2:]
  2556  	}
  2557  	res, err := DefaultClient.Do(req)
  2558  	if err != nil {
  2559  		t.Errorf("Error fetching %s: %v", url, err)
  2560  		return
  2561  	}
  2562  	_, err = ioutil.ReadAll(res.Body)
  2563  	defer res.Body.Close()
  2564  	if err != nil {
  2565  		t.Errorf("Error reading %s: %v", url, err)
  2566  	}
  2567  }
  2568  
  2569  func TestServerKeepAlivesEnabled(t *testing.T) {
  2570  	defer afterTest(t)
  2571  	ts := httptest.NewUnstartedServer(HandlerFunc(func(w ResponseWriter, r *Request) {}))
  2572  	ts.Config.SetKeepAlivesEnabled(false)
  2573  	ts.Start()
  2574  	defer ts.Close()
  2575  	res, err := Get(ts.URL)
  2576  	if err != nil {
  2577  		t.Fatal(err)
  2578  	}
  2579  	defer res.Body.Close()
  2580  	if !res.Close {
  2581  		t.Errorf("Body.Close == false; want true")
  2582  	}
  2583  }
  2584  
  2585  // golang.org/issue/7856
  2586  func TestServerEmptyBodyRace(t *testing.T) {
  2587  	defer afterTest(t)
  2588  	var n int32
  2589  	ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, req *Request) {
  2590  		atomic.AddInt32(&n, 1)
  2591  	}))
  2592  	defer ts.Close()
  2593  	var wg sync.WaitGroup
  2594  	const reqs = 20
  2595  	for i := 0; i < reqs; i++ {
  2596  		wg.Add(1)
  2597  		go func() {
  2598  			defer wg.Done()
  2599  			res, err := Get(ts.URL)
  2600  			if err != nil {
  2601  				t.Error(err)
  2602  				return
  2603  			}
  2604  			defer res.Body.Close()
  2605  			_, err = io.Copy(ioutil.Discard, res.Body)
  2606  			if err != nil {
  2607  				t.Error(err)
  2608  				return
  2609  			}
  2610  		}()
  2611  	}
  2612  	wg.Wait()
  2613  	if got := atomic.LoadInt32(&n); got != reqs {
  2614  		t.Errorf("handler ran %d times; want %d", got, reqs)
  2615  	}
  2616  }
  2617  
  2618  func TestServerConnStateNew(t *testing.T) {
  2619  	sawNew := false // if the test is buggy, we'll race on this variable.
  2620  	srv := &Server{
  2621  		ConnState: func(c net.Conn, state ConnState) {
  2622  			if state == StateNew {
  2623  				sawNew = true // testing that this write isn't racy
  2624  			}
  2625  		},
  2626  		Handler: HandlerFunc(func(w ResponseWriter, r *Request) {}), // irrelevant
  2627  	}
  2628  	srv.Serve(&oneConnListener{
  2629  		conn: &rwTestConn{
  2630  			Reader: strings.NewReader("GET / HTTP/1.1\r\nHost: foo\r\n\r\n"),
  2631  			Writer: ioutil.Discard,
  2632  		},
  2633  	})
  2634  	if !sawNew { // testing that this read isn't racy
  2635  		t.Error("StateNew not seen")
  2636  	}
  2637  }
  2638  
  2639  type closeWriteTestConn struct {
  2640  	rwTestConn
  2641  	didCloseWrite bool
  2642  }
  2643  
  2644  func (c *closeWriteTestConn) CloseWrite() error {
  2645  	c.didCloseWrite = true
  2646  	return nil
  2647  }
  2648  
  2649  func TestCloseWrite(t *testing.T) {
  2650  	var srv Server
  2651  	var testConn closeWriteTestConn
  2652  	c, err := ExportServerNewConn(&srv, &testConn)
  2653  	if err != nil {
  2654  		t.Fatal(err)
  2655  	}
  2656  	ExportCloseWriteAndWait(c)
  2657  	if !testConn.didCloseWrite {
  2658  		t.Error("didn't see CloseWrite call")
  2659  	}
  2660  }
  2661  
  2662  // This verifies that a handler can Flush and then Hijack.
  2663  //
  2664  // An similar test crashed once during development, but it was only
  2665  // testing this tangentially and temporarily until another TODO was
  2666  // fixed.
  2667  //
  2668  // So add an explicit test for this.
  2669  func TestServerFlushAndHijack(t *testing.T) {
  2670  	defer afterTest(t)
  2671  	ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  2672  		io.WriteString(w, "Hello, ")
  2673  		w.(Flusher).Flush()
  2674  		conn, buf, _ := w.(Hijacker).Hijack()
  2675  		buf.WriteString("6\r\nworld!\r\n0\r\n\r\n")
  2676  		if err := buf.Flush(); err != nil {
  2677  			t.Error(err)
  2678  		}
  2679  		if err := conn.Close(); err != nil {
  2680  			t.Error(err)
  2681  		}
  2682  	}))
  2683  	defer ts.Close()
  2684  	res, err := Get(ts.URL)
  2685  	if err != nil {
  2686  		t.Fatal(err)
  2687  	}
  2688  	defer res.Body.Close()
  2689  	all, err := ioutil.ReadAll(res.Body)
  2690  	if err != nil {
  2691  		t.Fatal(err)
  2692  	}
  2693  	if want := "Hello, world!"; string(all) != want {
  2694  		t.Errorf("Got %q; want %q", all, want)
  2695  	}
  2696  }
  2697  
  2698  // golang.org/issue/8534 -- the Server shouldn't reuse a connection
  2699  // for keep-alive after it's seen any Write error (e.g. a timeout) on
  2700  // that net.Conn.
  2701  //
  2702  // To test, verify we don't timeout or see fewer unique client
  2703  // addresses (== unique connections) than requests.
  2704  func TestServerKeepAliveAfterWriteError(t *testing.T) {
  2705  	if testing.Short() {
  2706  		t.Skip("skipping in -short mode")
  2707  	}
  2708  	defer afterTest(t)
  2709  	const numReq = 3
  2710  	addrc := make(chan string, numReq)
  2711  	ts := httptest.NewUnstartedServer(HandlerFunc(func(w ResponseWriter, r *Request) {
  2712  		addrc <- r.RemoteAddr
  2713  		time.Sleep(500 * time.Millisecond)
  2714  		w.(Flusher).Flush()
  2715  	}))
  2716  	ts.Config.WriteTimeout = 250 * time.Millisecond
  2717  	ts.Start()
  2718  	defer ts.Close()
  2719  
  2720  	errc := make(chan error, numReq)
  2721  	go func() {
  2722  		defer close(errc)
  2723  		for i := 0; i < numReq; i++ {
  2724  			res, err := Get(ts.URL)
  2725  			if res != nil {
  2726  				res.Body.Close()
  2727  			}
  2728  			errc <- err
  2729  		}
  2730  	}()
  2731  
  2732  	timeout := time.NewTimer(numReq * 2 * time.Second) // 4x overkill
  2733  	defer timeout.Stop()
  2734  	addrSeen := map[string]bool{}
  2735  	numOkay := 0
  2736  	for {
  2737  		select {
  2738  		case v := <-addrc:
  2739  			addrSeen[v] = true
  2740  		case err, ok := <-errc:
  2741  			if !ok {
  2742  				if len(addrSeen) != numReq {
  2743  					t.Errorf("saw %d unique client addresses; want %d", len(addrSeen), numReq)
  2744  				}
  2745  				if numOkay != 0 {
  2746  					t.Errorf("got %d successful client requests; want 0", numOkay)
  2747  				}
  2748  				return
  2749  			}
  2750  			if err == nil {
  2751  				numOkay++
  2752  			}
  2753  		case <-timeout.C:
  2754  			t.Fatal("timeout waiting for requests to complete")
  2755  		}
  2756  	}
  2757  }
  2758  
  2759  func BenchmarkClientServer(b *testing.B) {
  2760  	b.ReportAllocs()
  2761  	b.StopTimer()
  2762  	ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, r *Request) {
  2763  		fmt.Fprintf(rw, "Hello world.\n")
  2764  	}))
  2765  	defer ts.Close()
  2766  	b.StartTimer()
  2767  
  2768  	for i := 0; i < b.N; i++ {
  2769  		res, err := Get(ts.URL)
  2770  		if err != nil {
  2771  			b.Fatal("Get:", err)
  2772  		}
  2773  		all, err := ioutil.ReadAll(res.Body)
  2774  		res.Body.Close()
  2775  		if err != nil {
  2776  			b.Fatal("ReadAll:", err)
  2777  		}
  2778  		body := string(all)
  2779  		if body != "Hello world.\n" {
  2780  			b.Fatal("Got body:", body)
  2781  		}
  2782  	}
  2783  
  2784  	b.StopTimer()
  2785  }
  2786  
  2787  func BenchmarkClientServerParallel4(b *testing.B) {
  2788  	benchmarkClientServerParallel(b, 4, false)
  2789  }
  2790  
  2791  func BenchmarkClientServerParallel64(b *testing.B) {
  2792  	benchmarkClientServerParallel(b, 64, false)
  2793  }
  2794  
  2795  func BenchmarkClientServerParallelTLS4(b *testing.B) {
  2796  	benchmarkClientServerParallel(b, 4, true)
  2797  }
  2798  
  2799  func BenchmarkClientServerParallelTLS64(b *testing.B) {
  2800  	benchmarkClientServerParallel(b, 64, true)
  2801  }
  2802  
  2803  func benchmarkClientServerParallel(b *testing.B, parallelism int, useTLS bool) {
  2804  	b.ReportAllocs()
  2805  	ts := httptest.NewUnstartedServer(HandlerFunc(func(rw ResponseWriter, r *Request) {
  2806  		fmt.Fprintf(rw, "Hello world.\n")
  2807  	}))
  2808  	if useTLS {
  2809  		ts.StartTLS()
  2810  	} else {
  2811  		ts.Start()
  2812  	}
  2813  	defer ts.Close()
  2814  	b.ResetTimer()
  2815  	b.SetParallelism(parallelism)
  2816  	b.RunParallel(func(pb *testing.PB) {
  2817  		noVerifyTransport := &Transport{
  2818  			TLSClientConfig: &tls.Config{
  2819  				InsecureSkipVerify: true,
  2820  			},
  2821  		}
  2822  		defer noVerifyTransport.CloseIdleConnections()
  2823  		client := &Client{Transport: noVerifyTransport}
  2824  		for pb.Next() {
  2825  			res, err := client.Get(ts.URL)
  2826  			if err != nil {
  2827  				b.Logf("Get: %v", err)
  2828  				continue
  2829  			}
  2830  			all, err := ioutil.ReadAll(res.Body)
  2831  			res.Body.Close()
  2832  			if err != nil {
  2833  				b.Logf("ReadAll: %v", err)
  2834  				continue
  2835  			}
  2836  			body := string(all)
  2837  			if body != "Hello world.\n" {
  2838  				panic("Got body: " + body)
  2839  			}
  2840  		}
  2841  	})
  2842  }
  2843  
  2844  // A benchmark for profiling the server without the HTTP client code.
  2845  // The client code runs in a subprocess.
  2846  //
  2847  // For use like:
  2848  //   $ go test -c
  2849  //   $ ./http.test -test.run=XX -test.bench=BenchmarkServer -test.benchtime=15s -test.cpuprofile=http.prof
  2850  //   $ go tool pprof http.test http.prof
  2851  //   (pprof) web
  2852  func BenchmarkServer(b *testing.B) {
  2853  	b.ReportAllocs()
  2854  	// Child process mode;
  2855  	if url := os.Getenv("TEST_BENCH_SERVER_URL"); url != "" {
  2856  		n, err := strconv.Atoi(os.Getenv("TEST_BENCH_CLIENT_N"))
  2857  		if err != nil {
  2858  			panic(err)
  2859  		}
  2860  		for i := 0; i < n; i++ {
  2861  			res, err := Get(url)
  2862  			if err != nil {
  2863  				log.Panicf("Get: %v", err)
  2864  			}
  2865  			all, err := ioutil.ReadAll(res.Body)
  2866  			res.Body.Close()
  2867  			if err != nil {
  2868  				log.Panicf("ReadAll: %v", err)
  2869  			}
  2870  			body := string(all)
  2871  			if body != "Hello world.\n" {
  2872  				log.Panicf("Got body: %q", body)
  2873  			}
  2874  		}
  2875  		os.Exit(0)
  2876  		return
  2877  	}
  2878  
  2879  	var res = []byte("Hello world.\n")
  2880  	b.StopTimer()
  2881  	ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, r *Request) {
  2882  		rw.Header().Set("Content-Type", "text/html; charset=utf-8")
  2883  		rw.Write(res)
  2884  	}))
  2885  	defer ts.Close()
  2886  	b.StartTimer()
  2887  
  2888  	cmd := exec.Command(os.Args[0], "-test.run=XXXX", "-test.bench=BenchmarkServer")
  2889  	cmd.Env = append([]string{
  2890  		fmt.Sprintf("TEST_BENCH_CLIENT_N=%d", b.N),
  2891  		fmt.Sprintf("TEST_BENCH_SERVER_URL=%s", ts.URL),
  2892  	}, os.Environ()...)
  2893  	out, err := cmd.CombinedOutput()
  2894  	if err != nil {
  2895  		b.Errorf("Test failure: %v, with output: %s", err, out)
  2896  	}
  2897  }
  2898  
  2899  func BenchmarkServerFakeConnNoKeepAlive(b *testing.B) {
  2900  	b.ReportAllocs()
  2901  	req := reqBytes(`GET / HTTP/1.0
  2902  Host: golang.org
  2903  Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8
  2904  User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_8_2) AppleWebKit/537.17 (KHTML, like Gecko) Chrome/24.0.1312.52 Safari/537.17
  2905  Accept-Encoding: gzip,deflate,sdch
  2906  Accept-Language: en-US,en;q=0.8
  2907  Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.3
  2908  `)
  2909  	res := []byte("Hello world!\n")
  2910  
  2911  	conn := &testConn{
  2912  		// testConn.Close will not push into the channel
  2913  		// if it's full.
  2914  		closec: make(chan bool, 1),
  2915  	}
  2916  	handler := HandlerFunc(func(rw ResponseWriter, r *Request) {
  2917  		rw.Header().Set("Content-Type", "text/html; charset=utf-8")
  2918  		rw.Write(res)
  2919  	})
  2920  	ln := new(oneConnListener)
  2921  	for i := 0; i < b.N; i++ {
  2922  		conn.readBuf.Reset()
  2923  		conn.writeBuf.Reset()
  2924  		conn.readBuf.Write(req)
  2925  		ln.conn = conn
  2926  		Serve(ln, handler)
  2927  		<-conn.closec
  2928  	}
  2929  }
  2930  
  2931  // repeatReader reads content count times, then EOFs.
  2932  type repeatReader struct {
  2933  	content []byte
  2934  	count   int
  2935  	off     int
  2936  }
  2937  
  2938  func (r *repeatReader) Read(p []byte) (n int, err error) {
  2939  	if r.count <= 0 {
  2940  		return 0, io.EOF
  2941  	}
  2942  	n = copy(p, r.content[r.off:])
  2943  	r.off += n
  2944  	if r.off == len(r.content) {
  2945  		r.count--
  2946  		r.off = 0
  2947  	}
  2948  	return
  2949  }
  2950  
  2951  func BenchmarkServerFakeConnWithKeepAlive(b *testing.B) {
  2952  	b.ReportAllocs()
  2953  
  2954  	req := reqBytes(`GET / HTTP/1.1
  2955  Host: golang.org
  2956  Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8
  2957  User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_8_2) AppleWebKit/537.17 (KHTML, like Gecko) Chrome/24.0.1312.52 Safari/537.17
  2958  Accept-Encoding: gzip,deflate,sdch
  2959  Accept-Language: en-US,en;q=0.8
  2960  Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.3
  2961  `)
  2962  	res := []byte("Hello world!\n")
  2963  
  2964  	conn := &rwTestConn{
  2965  		Reader: &repeatReader{content: req, count: b.N},
  2966  		Writer: ioutil.Discard,
  2967  		closec: make(chan bool, 1),
  2968  	}
  2969  	handled := 0
  2970  	handler := HandlerFunc(func(rw ResponseWriter, r *Request) {
  2971  		handled++
  2972  		rw.Header().Set("Content-Type", "text/html; charset=utf-8")
  2973  		rw.Write(res)
  2974  	})
  2975  	ln := &oneConnListener{conn: conn}
  2976  	go Serve(ln, handler)
  2977  	<-conn.closec
  2978  	if b.N != handled {
  2979  		b.Errorf("b.N=%d but handled %d", b.N, handled)
  2980  	}
  2981  }
  2982  
  2983  // same as above, but representing the most simple possible request
  2984  // and handler. Notably: the handler does not call rw.Header().
  2985  func BenchmarkServerFakeConnWithKeepAliveLite(b *testing.B) {
  2986  	b.ReportAllocs()
  2987  
  2988  	req := reqBytes(`GET / HTTP/1.1
  2989  Host: golang.org
  2990  `)
  2991  	res := []byte("Hello world!\n")
  2992  
  2993  	conn := &rwTestConn{
  2994  		Reader: &repeatReader{content: req, count: b.N},
  2995  		Writer: ioutil.Discard,
  2996  		closec: make(chan bool, 1),
  2997  	}
  2998  	handled := 0
  2999  	handler := HandlerFunc(func(rw ResponseWriter, r *Request) {
  3000  		handled++
  3001  		rw.Write(res)
  3002  	})
  3003  	ln := &oneConnListener{conn: conn}
  3004  	go Serve(ln, handler)
  3005  	<-conn.closec
  3006  	if b.N != handled {
  3007  		b.Errorf("b.N=%d but handled %d", b.N, handled)
  3008  	}
  3009  }
  3010  
  3011  const someResponse = "<html>some response</html>"
  3012  
  3013  // A Response that's just no bigger than 2KB, the buffer-before-chunking threshold.
  3014  var response = bytes.Repeat([]byte(someResponse), 2<<10/len(someResponse))
  3015  
  3016  // Both Content-Type and Content-Length set. Should be no buffering.
  3017  func BenchmarkServerHandlerTypeLen(b *testing.B) {
  3018  	benchmarkHandler(b, HandlerFunc(func(w ResponseWriter, r *Request) {
  3019  		w.Header().Set("Content-Type", "text/html")
  3020  		w.Header().Set("Content-Length", strconv.Itoa(len(response)))
  3021  		w.Write(response)
  3022  	}))
  3023  }
  3024  
  3025  // A Content-Type is set, but no length. No sniffing, but will count the Content-Length.
  3026  func BenchmarkServerHandlerNoLen(b *testing.B) {
  3027  	benchmarkHandler(b, HandlerFunc(func(w ResponseWriter, r *Request) {
  3028  		w.Header().Set("Content-Type", "text/html")
  3029  		w.Write(response)
  3030  	}))
  3031  }
  3032  
  3033  // A Content-Length is set, but the Content-Type will be sniffed.
  3034  func BenchmarkServerHandlerNoType(b *testing.B) {
  3035  	benchmarkHandler(b, HandlerFunc(func(w ResponseWriter, r *Request) {
  3036  		w.Header().Set("Content-Length", strconv.Itoa(len(response)))
  3037  		w.Write(response)
  3038  	}))
  3039  }
  3040  
  3041  // Neither a Content-Type or Content-Length, so sniffed and counted.
  3042  func BenchmarkServerHandlerNoHeader(b *testing.B) {
  3043  	benchmarkHandler(b, HandlerFunc(func(w ResponseWriter, r *Request) {
  3044  		w.Write(response)
  3045  	}))
  3046  }
  3047  
  3048  func benchmarkHandler(b *testing.B, h Handler) {
  3049  	b.ReportAllocs()
  3050  	req := reqBytes(`GET / HTTP/1.1
  3051  Host: golang.org
  3052  `)
  3053  	conn := &rwTestConn{
  3054  		Reader: &repeatReader{content: req, count: b.N},
  3055  		Writer: ioutil.Discard,
  3056  		closec: make(chan bool, 1),
  3057  	}
  3058  	handled := 0
  3059  	handler := HandlerFunc(func(rw ResponseWriter, r *Request) {
  3060  		handled++
  3061  		h.ServeHTTP(rw, r)
  3062  	})
  3063  	ln := &oneConnListener{conn: conn}
  3064  	go Serve(ln, handler)
  3065  	<-conn.closec
  3066  	if b.N != handled {
  3067  		b.Errorf("b.N=%d but handled %d", b.N, handled)
  3068  	}
  3069  }
  3070  
  3071  func BenchmarkServerHijack(b *testing.B) {
  3072  	b.ReportAllocs()
  3073  	req := reqBytes(`GET / HTTP/1.1
  3074  Host: golang.org
  3075  `)
  3076  	h := HandlerFunc(func(w ResponseWriter, r *Request) {
  3077  		conn, _, err := w.(Hijacker).Hijack()
  3078  		if err != nil {
  3079  			panic(err)
  3080  		}
  3081  		conn.Close()
  3082  	})
  3083  	conn := &rwTestConn{
  3084  		Writer: ioutil.Discard,
  3085  		closec: make(chan bool, 1),
  3086  	}
  3087  	ln := &oneConnListener{conn: conn}
  3088  	for i := 0; i < b.N; i++ {
  3089  		conn.Reader = bytes.NewReader(req)
  3090  		ln.conn = conn
  3091  		Serve(ln, h)
  3092  		<-conn.closec
  3093  	}
  3094  }