github.com/bgentry/go@v0.0.0-20150121062915-6cf5a733d54d/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 // test that ResponseWriter implements io.stringWriter. 2388 func TestResponseWriterWriteString(t *testing.T) { 2389 okc := make(chan bool, 1) 2390 ht := newHandlerTest(HandlerFunc(func(w ResponseWriter, r *Request) { 2391 type stringWriter interface { 2392 WriteString(s string) (n int, err error) 2393 } 2394 _, ok := w.(stringWriter) 2395 okc <- ok 2396 })) 2397 ht.rawResponse("GET / HTTP/1.0") 2398 select { 2399 case ok := <-okc: 2400 if !ok { 2401 t.Error("ResponseWriter did not implement io.stringWriter") 2402 } 2403 default: 2404 t.Error("handler was never called") 2405 } 2406 } 2407 2408 func TestAppendTime(t *testing.T) { 2409 var b [len(TimeFormat)]byte 2410 t1 := time.Date(2013, 9, 21, 15, 41, 0, 0, time.FixedZone("CEST", 2*60*60)) 2411 res := ExportAppendTime(b[:0], t1) 2412 t2, err := ParseTime(string(res)) 2413 if err != nil { 2414 t.Fatalf("Error parsing time: %s", err) 2415 } 2416 if !t1.Equal(t2) { 2417 t.Fatalf("Times differ; expected: %v, got %v (%s)", t1, t2, string(res)) 2418 } 2419 } 2420 2421 func TestServerConnState(t *testing.T) { 2422 defer afterTest(t) 2423 handler := map[string]func(w ResponseWriter, r *Request){ 2424 "/": func(w ResponseWriter, r *Request) { 2425 fmt.Fprintf(w, "Hello.") 2426 }, 2427 "/close": func(w ResponseWriter, r *Request) { 2428 w.Header().Set("Connection", "close") 2429 fmt.Fprintf(w, "Hello.") 2430 }, 2431 "/hijack": func(w ResponseWriter, r *Request) { 2432 c, _, _ := w.(Hijacker).Hijack() 2433 c.Write([]byte("HTTP/1.0 200 OK\r\nConnection: close\r\n\r\nHello.")) 2434 c.Close() 2435 }, 2436 "/hijack-panic": func(w ResponseWriter, r *Request) { 2437 c, _, _ := w.(Hijacker).Hijack() 2438 c.Write([]byte("HTTP/1.0 200 OK\r\nConnection: close\r\n\r\nHello.")) 2439 c.Close() 2440 panic("intentional panic") 2441 }, 2442 } 2443 ts := httptest.NewUnstartedServer(HandlerFunc(func(w ResponseWriter, r *Request) { 2444 handler[r.URL.Path](w, r) 2445 })) 2446 defer ts.Close() 2447 2448 var mu sync.Mutex // guard stateLog and connID 2449 var stateLog = map[int][]ConnState{} 2450 var connID = map[net.Conn]int{} 2451 2452 ts.Config.ErrorLog = log.New(ioutil.Discard, "", 0) 2453 ts.Config.ConnState = func(c net.Conn, state ConnState) { 2454 if c == nil { 2455 t.Errorf("nil conn seen in state %s", state) 2456 return 2457 } 2458 mu.Lock() 2459 defer mu.Unlock() 2460 id, ok := connID[c] 2461 if !ok { 2462 id = len(connID) + 1 2463 connID[c] = id 2464 } 2465 stateLog[id] = append(stateLog[id], state) 2466 } 2467 ts.Start() 2468 2469 mustGet(t, ts.URL+"/") 2470 mustGet(t, ts.URL+"/close") 2471 2472 mustGet(t, ts.URL+"/") 2473 mustGet(t, ts.URL+"/", "Connection", "close") 2474 2475 mustGet(t, ts.URL+"/hijack") 2476 mustGet(t, ts.URL+"/hijack-panic") 2477 2478 // New->Closed 2479 { 2480 c, err := net.Dial("tcp", ts.Listener.Addr().String()) 2481 if err != nil { 2482 t.Fatal(err) 2483 } 2484 c.Close() 2485 } 2486 2487 // New->Active->Closed 2488 { 2489 c, err := net.Dial("tcp", ts.Listener.Addr().String()) 2490 if err != nil { 2491 t.Fatal(err) 2492 } 2493 if _, err := io.WriteString(c, "BOGUS REQUEST\r\n\r\n"); err != nil { 2494 t.Fatal(err) 2495 } 2496 c.Close() 2497 } 2498 2499 // New->Idle->Closed 2500 { 2501 c, err := net.Dial("tcp", ts.Listener.Addr().String()) 2502 if err != nil { 2503 t.Fatal(err) 2504 } 2505 if _, err := io.WriteString(c, "GET / HTTP/1.1\r\nHost: foo\r\n\r\n"); err != nil { 2506 t.Fatal(err) 2507 } 2508 res, err := ReadResponse(bufio.NewReader(c), nil) 2509 if err != nil { 2510 t.Fatal(err) 2511 } 2512 if _, err := io.Copy(ioutil.Discard, res.Body); err != nil { 2513 t.Fatal(err) 2514 } 2515 c.Close() 2516 } 2517 2518 want := map[int][]ConnState{ 2519 1: {StateNew, StateActive, StateIdle, StateActive, StateClosed}, 2520 2: {StateNew, StateActive, StateIdle, StateActive, StateClosed}, 2521 3: {StateNew, StateActive, StateHijacked}, 2522 4: {StateNew, StateActive, StateHijacked}, 2523 5: {StateNew, StateClosed}, 2524 6: {StateNew, StateActive, StateClosed}, 2525 7: {StateNew, StateActive, StateIdle, StateClosed}, 2526 } 2527 logString := func(m map[int][]ConnState) string { 2528 var b bytes.Buffer 2529 for id, l := range m { 2530 fmt.Fprintf(&b, "Conn %d: ", id) 2531 for _, s := range l { 2532 fmt.Fprintf(&b, "%s ", s) 2533 } 2534 b.WriteString("\n") 2535 } 2536 return b.String() 2537 } 2538 2539 for i := 0; i < 5; i++ { 2540 time.Sleep(time.Duration(i) * 50 * time.Millisecond) 2541 mu.Lock() 2542 match := reflect.DeepEqual(stateLog, want) 2543 mu.Unlock() 2544 if match { 2545 return 2546 } 2547 } 2548 2549 mu.Lock() 2550 t.Errorf("Unexpected events.\nGot log: %s\n Want: %s\n", logString(stateLog), logString(want)) 2551 mu.Unlock() 2552 } 2553 2554 func mustGet(t *testing.T, url string, headers ...string) { 2555 req, err := NewRequest("GET", url, nil) 2556 if err != nil { 2557 t.Fatal(err) 2558 } 2559 for len(headers) > 0 { 2560 req.Header.Add(headers[0], headers[1]) 2561 headers = headers[2:] 2562 } 2563 res, err := DefaultClient.Do(req) 2564 if err != nil { 2565 t.Errorf("Error fetching %s: %v", url, err) 2566 return 2567 } 2568 _, err = ioutil.ReadAll(res.Body) 2569 defer res.Body.Close() 2570 if err != nil { 2571 t.Errorf("Error reading %s: %v", url, err) 2572 } 2573 } 2574 2575 func TestServerKeepAlivesEnabled(t *testing.T) { 2576 defer afterTest(t) 2577 ts := httptest.NewUnstartedServer(HandlerFunc(func(w ResponseWriter, r *Request) {})) 2578 ts.Config.SetKeepAlivesEnabled(false) 2579 ts.Start() 2580 defer ts.Close() 2581 res, err := Get(ts.URL) 2582 if err != nil { 2583 t.Fatal(err) 2584 } 2585 defer res.Body.Close() 2586 if !res.Close { 2587 t.Errorf("Body.Close == false; want true") 2588 } 2589 } 2590 2591 // golang.org/issue/7856 2592 func TestServerEmptyBodyRace(t *testing.T) { 2593 defer afterTest(t) 2594 var n int32 2595 ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, req *Request) { 2596 atomic.AddInt32(&n, 1) 2597 })) 2598 defer ts.Close() 2599 var wg sync.WaitGroup 2600 const reqs = 20 2601 for i := 0; i < reqs; i++ { 2602 wg.Add(1) 2603 go func() { 2604 defer wg.Done() 2605 res, err := Get(ts.URL) 2606 if err != nil { 2607 t.Error(err) 2608 return 2609 } 2610 defer res.Body.Close() 2611 _, err = io.Copy(ioutil.Discard, res.Body) 2612 if err != nil { 2613 t.Error(err) 2614 return 2615 } 2616 }() 2617 } 2618 wg.Wait() 2619 if got := atomic.LoadInt32(&n); got != reqs { 2620 t.Errorf("handler ran %d times; want %d", got, reqs) 2621 } 2622 } 2623 2624 func TestServerConnStateNew(t *testing.T) { 2625 sawNew := false // if the test is buggy, we'll race on this variable. 2626 srv := &Server{ 2627 ConnState: func(c net.Conn, state ConnState) { 2628 if state == StateNew { 2629 sawNew = true // testing that this write isn't racy 2630 } 2631 }, 2632 Handler: HandlerFunc(func(w ResponseWriter, r *Request) {}), // irrelevant 2633 } 2634 srv.Serve(&oneConnListener{ 2635 conn: &rwTestConn{ 2636 Reader: strings.NewReader("GET / HTTP/1.1\r\nHost: foo\r\n\r\n"), 2637 Writer: ioutil.Discard, 2638 }, 2639 }) 2640 if !sawNew { // testing that this read isn't racy 2641 t.Error("StateNew not seen") 2642 } 2643 } 2644 2645 type closeWriteTestConn struct { 2646 rwTestConn 2647 didCloseWrite bool 2648 } 2649 2650 func (c *closeWriteTestConn) CloseWrite() error { 2651 c.didCloseWrite = true 2652 return nil 2653 } 2654 2655 func TestCloseWrite(t *testing.T) { 2656 var srv Server 2657 var testConn closeWriteTestConn 2658 c, err := ExportServerNewConn(&srv, &testConn) 2659 if err != nil { 2660 t.Fatal(err) 2661 } 2662 ExportCloseWriteAndWait(c) 2663 if !testConn.didCloseWrite { 2664 t.Error("didn't see CloseWrite call") 2665 } 2666 } 2667 2668 // This verifies that a handler can Flush and then Hijack. 2669 // 2670 // An similar test crashed once during development, but it was only 2671 // testing this tangentially and temporarily until another TODO was 2672 // fixed. 2673 // 2674 // So add an explicit test for this. 2675 func TestServerFlushAndHijack(t *testing.T) { 2676 defer afterTest(t) 2677 ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) { 2678 io.WriteString(w, "Hello, ") 2679 w.(Flusher).Flush() 2680 conn, buf, _ := w.(Hijacker).Hijack() 2681 buf.WriteString("6\r\nworld!\r\n0\r\n\r\n") 2682 if err := buf.Flush(); err != nil { 2683 t.Error(err) 2684 } 2685 if err := conn.Close(); err != nil { 2686 t.Error(err) 2687 } 2688 })) 2689 defer ts.Close() 2690 res, err := Get(ts.URL) 2691 if err != nil { 2692 t.Fatal(err) 2693 } 2694 defer res.Body.Close() 2695 all, err := ioutil.ReadAll(res.Body) 2696 if err != nil { 2697 t.Fatal(err) 2698 } 2699 if want := "Hello, world!"; string(all) != want { 2700 t.Errorf("Got %q; want %q", all, want) 2701 } 2702 } 2703 2704 // golang.org/issue/8534 -- the Server shouldn't reuse a connection 2705 // for keep-alive after it's seen any Write error (e.g. a timeout) on 2706 // that net.Conn. 2707 // 2708 // To test, verify we don't timeout or see fewer unique client 2709 // addresses (== unique connections) than requests. 2710 func TestServerKeepAliveAfterWriteError(t *testing.T) { 2711 if testing.Short() { 2712 t.Skip("skipping in -short mode") 2713 } 2714 defer afterTest(t) 2715 const numReq = 3 2716 addrc := make(chan string, numReq) 2717 ts := httptest.NewUnstartedServer(HandlerFunc(func(w ResponseWriter, r *Request) { 2718 addrc <- r.RemoteAddr 2719 time.Sleep(500 * time.Millisecond) 2720 w.(Flusher).Flush() 2721 })) 2722 ts.Config.WriteTimeout = 250 * time.Millisecond 2723 ts.Start() 2724 defer ts.Close() 2725 2726 errc := make(chan error, numReq) 2727 go func() { 2728 defer close(errc) 2729 for i := 0; i < numReq; i++ { 2730 res, err := Get(ts.URL) 2731 if res != nil { 2732 res.Body.Close() 2733 } 2734 errc <- err 2735 } 2736 }() 2737 2738 timeout := time.NewTimer(numReq * 2 * time.Second) // 4x overkill 2739 defer timeout.Stop() 2740 addrSeen := map[string]bool{} 2741 numOkay := 0 2742 for { 2743 select { 2744 case v := <-addrc: 2745 addrSeen[v] = true 2746 case err, ok := <-errc: 2747 if !ok { 2748 if len(addrSeen) != numReq { 2749 t.Errorf("saw %d unique client addresses; want %d", len(addrSeen), numReq) 2750 } 2751 if numOkay != 0 { 2752 t.Errorf("got %d successful client requests; want 0", numOkay) 2753 } 2754 return 2755 } 2756 if err == nil { 2757 numOkay++ 2758 } 2759 case <-timeout.C: 2760 t.Fatal("timeout waiting for requests to complete") 2761 } 2762 } 2763 } 2764 2765 func BenchmarkClientServer(b *testing.B) { 2766 b.ReportAllocs() 2767 b.StopTimer() 2768 ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, r *Request) { 2769 fmt.Fprintf(rw, "Hello world.\n") 2770 })) 2771 defer ts.Close() 2772 b.StartTimer() 2773 2774 for i := 0; i < b.N; i++ { 2775 res, err := Get(ts.URL) 2776 if err != nil { 2777 b.Fatal("Get:", err) 2778 } 2779 all, err := ioutil.ReadAll(res.Body) 2780 res.Body.Close() 2781 if err != nil { 2782 b.Fatal("ReadAll:", err) 2783 } 2784 body := string(all) 2785 if body != "Hello world.\n" { 2786 b.Fatal("Got body:", body) 2787 } 2788 } 2789 2790 b.StopTimer() 2791 } 2792 2793 func BenchmarkClientServerParallel4(b *testing.B) { 2794 benchmarkClientServerParallel(b, 4, false) 2795 } 2796 2797 func BenchmarkClientServerParallel64(b *testing.B) { 2798 benchmarkClientServerParallel(b, 64, false) 2799 } 2800 2801 func BenchmarkClientServerParallelTLS4(b *testing.B) { 2802 benchmarkClientServerParallel(b, 4, true) 2803 } 2804 2805 func BenchmarkClientServerParallelTLS64(b *testing.B) { 2806 benchmarkClientServerParallel(b, 64, true) 2807 } 2808 2809 func benchmarkClientServerParallel(b *testing.B, parallelism int, useTLS bool) { 2810 b.ReportAllocs() 2811 ts := httptest.NewUnstartedServer(HandlerFunc(func(rw ResponseWriter, r *Request) { 2812 fmt.Fprintf(rw, "Hello world.\n") 2813 })) 2814 if useTLS { 2815 ts.StartTLS() 2816 } else { 2817 ts.Start() 2818 } 2819 defer ts.Close() 2820 b.ResetTimer() 2821 b.SetParallelism(parallelism) 2822 b.RunParallel(func(pb *testing.PB) { 2823 noVerifyTransport := &Transport{ 2824 TLSClientConfig: &tls.Config{ 2825 InsecureSkipVerify: true, 2826 }, 2827 } 2828 defer noVerifyTransport.CloseIdleConnections() 2829 client := &Client{Transport: noVerifyTransport} 2830 for pb.Next() { 2831 res, err := client.Get(ts.URL) 2832 if err != nil { 2833 b.Logf("Get: %v", err) 2834 continue 2835 } 2836 all, err := ioutil.ReadAll(res.Body) 2837 res.Body.Close() 2838 if err != nil { 2839 b.Logf("ReadAll: %v", err) 2840 continue 2841 } 2842 body := string(all) 2843 if body != "Hello world.\n" { 2844 panic("Got body: " + body) 2845 } 2846 } 2847 }) 2848 } 2849 2850 // A benchmark for profiling the server without the HTTP client code. 2851 // The client code runs in a subprocess. 2852 // 2853 // For use like: 2854 // $ go test -c 2855 // $ ./http.test -test.run=XX -test.bench=BenchmarkServer -test.benchtime=15s -test.cpuprofile=http.prof 2856 // $ go tool pprof http.test http.prof 2857 // (pprof) web 2858 func BenchmarkServer(b *testing.B) { 2859 b.ReportAllocs() 2860 // Child process mode; 2861 if url := os.Getenv("TEST_BENCH_SERVER_URL"); url != "" { 2862 n, err := strconv.Atoi(os.Getenv("TEST_BENCH_CLIENT_N")) 2863 if err != nil { 2864 panic(err) 2865 } 2866 for i := 0; i < n; i++ { 2867 res, err := Get(url) 2868 if err != nil { 2869 log.Panicf("Get: %v", err) 2870 } 2871 all, err := ioutil.ReadAll(res.Body) 2872 res.Body.Close() 2873 if err != nil { 2874 log.Panicf("ReadAll: %v", err) 2875 } 2876 body := string(all) 2877 if body != "Hello world.\n" { 2878 log.Panicf("Got body: %q", body) 2879 } 2880 } 2881 os.Exit(0) 2882 return 2883 } 2884 2885 var res = []byte("Hello world.\n") 2886 b.StopTimer() 2887 ts := httptest.NewServer(HandlerFunc(func(rw ResponseWriter, r *Request) { 2888 rw.Header().Set("Content-Type", "text/html; charset=utf-8") 2889 rw.Write(res) 2890 })) 2891 defer ts.Close() 2892 b.StartTimer() 2893 2894 cmd := exec.Command(os.Args[0], "-test.run=XXXX", "-test.bench=BenchmarkServer") 2895 cmd.Env = append([]string{ 2896 fmt.Sprintf("TEST_BENCH_CLIENT_N=%d", b.N), 2897 fmt.Sprintf("TEST_BENCH_SERVER_URL=%s", ts.URL), 2898 }, os.Environ()...) 2899 out, err := cmd.CombinedOutput() 2900 if err != nil { 2901 b.Errorf("Test failure: %v, with output: %s", err, out) 2902 } 2903 } 2904 2905 func BenchmarkServerFakeConnNoKeepAlive(b *testing.B) { 2906 b.ReportAllocs() 2907 req := reqBytes(`GET / HTTP/1.0 2908 Host: golang.org 2909 Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8 2910 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 2911 Accept-Encoding: gzip,deflate,sdch 2912 Accept-Language: en-US,en;q=0.8 2913 Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.3 2914 `) 2915 res := []byte("Hello world!\n") 2916 2917 conn := &testConn{ 2918 // testConn.Close will not push into the channel 2919 // if it's full. 2920 closec: make(chan bool, 1), 2921 } 2922 handler := HandlerFunc(func(rw ResponseWriter, r *Request) { 2923 rw.Header().Set("Content-Type", "text/html; charset=utf-8") 2924 rw.Write(res) 2925 }) 2926 ln := new(oneConnListener) 2927 for i := 0; i < b.N; i++ { 2928 conn.readBuf.Reset() 2929 conn.writeBuf.Reset() 2930 conn.readBuf.Write(req) 2931 ln.conn = conn 2932 Serve(ln, handler) 2933 <-conn.closec 2934 } 2935 } 2936 2937 // repeatReader reads content count times, then EOFs. 2938 type repeatReader struct { 2939 content []byte 2940 count int 2941 off int 2942 } 2943 2944 func (r *repeatReader) Read(p []byte) (n int, err error) { 2945 if r.count <= 0 { 2946 return 0, io.EOF 2947 } 2948 n = copy(p, r.content[r.off:]) 2949 r.off += n 2950 if r.off == len(r.content) { 2951 r.count-- 2952 r.off = 0 2953 } 2954 return 2955 } 2956 2957 func BenchmarkServerFakeConnWithKeepAlive(b *testing.B) { 2958 b.ReportAllocs() 2959 2960 req := reqBytes(`GET / HTTP/1.1 2961 Host: golang.org 2962 Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8 2963 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 2964 Accept-Encoding: gzip,deflate,sdch 2965 Accept-Language: en-US,en;q=0.8 2966 Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.3 2967 `) 2968 res := []byte("Hello world!\n") 2969 2970 conn := &rwTestConn{ 2971 Reader: &repeatReader{content: req, count: b.N}, 2972 Writer: ioutil.Discard, 2973 closec: make(chan bool, 1), 2974 } 2975 handled := 0 2976 handler := HandlerFunc(func(rw ResponseWriter, r *Request) { 2977 handled++ 2978 rw.Header().Set("Content-Type", "text/html; charset=utf-8") 2979 rw.Write(res) 2980 }) 2981 ln := &oneConnListener{conn: conn} 2982 go Serve(ln, handler) 2983 <-conn.closec 2984 if b.N != handled { 2985 b.Errorf("b.N=%d but handled %d", b.N, handled) 2986 } 2987 } 2988 2989 // same as above, but representing the most simple possible request 2990 // and handler. Notably: the handler does not call rw.Header(). 2991 func BenchmarkServerFakeConnWithKeepAliveLite(b *testing.B) { 2992 b.ReportAllocs() 2993 2994 req := reqBytes(`GET / HTTP/1.1 2995 Host: golang.org 2996 `) 2997 res := []byte("Hello world!\n") 2998 2999 conn := &rwTestConn{ 3000 Reader: &repeatReader{content: req, count: b.N}, 3001 Writer: ioutil.Discard, 3002 closec: make(chan bool, 1), 3003 } 3004 handled := 0 3005 handler := HandlerFunc(func(rw ResponseWriter, r *Request) { 3006 handled++ 3007 rw.Write(res) 3008 }) 3009 ln := &oneConnListener{conn: conn} 3010 go Serve(ln, handler) 3011 <-conn.closec 3012 if b.N != handled { 3013 b.Errorf("b.N=%d but handled %d", b.N, handled) 3014 } 3015 } 3016 3017 const someResponse = "<html>some response</html>" 3018 3019 // A Response that's just no bigger than 2KB, the buffer-before-chunking threshold. 3020 var response = bytes.Repeat([]byte(someResponse), 2<<10/len(someResponse)) 3021 3022 // Both Content-Type and Content-Length set. Should be no buffering. 3023 func BenchmarkServerHandlerTypeLen(b *testing.B) { 3024 benchmarkHandler(b, HandlerFunc(func(w ResponseWriter, r *Request) { 3025 w.Header().Set("Content-Type", "text/html") 3026 w.Header().Set("Content-Length", strconv.Itoa(len(response))) 3027 w.Write(response) 3028 })) 3029 } 3030 3031 // A Content-Type is set, but no length. No sniffing, but will count the Content-Length. 3032 func BenchmarkServerHandlerNoLen(b *testing.B) { 3033 benchmarkHandler(b, HandlerFunc(func(w ResponseWriter, r *Request) { 3034 w.Header().Set("Content-Type", "text/html") 3035 w.Write(response) 3036 })) 3037 } 3038 3039 // A Content-Length is set, but the Content-Type will be sniffed. 3040 func BenchmarkServerHandlerNoType(b *testing.B) { 3041 benchmarkHandler(b, HandlerFunc(func(w ResponseWriter, r *Request) { 3042 w.Header().Set("Content-Length", strconv.Itoa(len(response))) 3043 w.Write(response) 3044 })) 3045 } 3046 3047 // Neither a Content-Type or Content-Length, so sniffed and counted. 3048 func BenchmarkServerHandlerNoHeader(b *testing.B) { 3049 benchmarkHandler(b, HandlerFunc(func(w ResponseWriter, r *Request) { 3050 w.Write(response) 3051 })) 3052 } 3053 3054 func benchmarkHandler(b *testing.B, h Handler) { 3055 b.ReportAllocs() 3056 req := reqBytes(`GET / HTTP/1.1 3057 Host: golang.org 3058 `) 3059 conn := &rwTestConn{ 3060 Reader: &repeatReader{content: req, count: b.N}, 3061 Writer: ioutil.Discard, 3062 closec: make(chan bool, 1), 3063 } 3064 handled := 0 3065 handler := HandlerFunc(func(rw ResponseWriter, r *Request) { 3066 handled++ 3067 h.ServeHTTP(rw, r) 3068 }) 3069 ln := &oneConnListener{conn: conn} 3070 go Serve(ln, handler) 3071 <-conn.closec 3072 if b.N != handled { 3073 b.Errorf("b.N=%d but handled %d", b.N, handled) 3074 } 3075 } 3076 3077 func BenchmarkServerHijack(b *testing.B) { 3078 b.ReportAllocs() 3079 req := reqBytes(`GET / HTTP/1.1 3080 Host: golang.org 3081 `) 3082 h := HandlerFunc(func(w ResponseWriter, r *Request) { 3083 conn, _, err := w.(Hijacker).Hijack() 3084 if err != nil { 3085 panic(err) 3086 } 3087 conn.Close() 3088 }) 3089 conn := &rwTestConn{ 3090 Writer: ioutil.Discard, 3091 closec: make(chan bool, 1), 3092 } 3093 ln := &oneConnListener{conn: conn} 3094 for i := 0; i < b.N; i++ { 3095 conn.Reader = bytes.NewReader(req) 3096 ln.conn = conn 3097 Serve(ln, h) 3098 <-conn.closec 3099 } 3100 }