github.com/bgentry/go@v0.0.0-20150121062915-6cf5a733d54d/src/net/http/fs_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 package http_test 6 7 import ( 8 "bytes" 9 "errors" 10 "fmt" 11 "io" 12 "io/ioutil" 13 "mime" 14 "mime/multipart" 15 "net" 16 . "net/http" 17 "net/http/httptest" 18 "net/url" 19 "os" 20 "os/exec" 21 "path" 22 "path/filepath" 23 "reflect" 24 "regexp" 25 "runtime" 26 "strconv" 27 "strings" 28 "testing" 29 "time" 30 ) 31 32 const ( 33 testFile = "testdata/file" 34 testFileLen = 11 35 ) 36 37 type wantRange struct { 38 start, end int64 // range [start,end) 39 } 40 41 var itoa = strconv.Itoa 42 43 var ServeFileRangeTests = []struct { 44 r string 45 code int 46 ranges []wantRange 47 }{ 48 {r: "", code: StatusOK}, 49 {r: "bytes=0-4", code: StatusPartialContent, ranges: []wantRange{{0, 5}}}, 50 {r: "bytes=2-", code: StatusPartialContent, ranges: []wantRange{{2, testFileLen}}}, 51 {r: "bytes=-5", code: StatusPartialContent, ranges: []wantRange{{testFileLen - 5, testFileLen}}}, 52 {r: "bytes=3-7", code: StatusPartialContent, ranges: []wantRange{{3, 8}}}, 53 {r: "bytes=0-0,-2", code: StatusPartialContent, ranges: []wantRange{{0, 1}, {testFileLen - 2, testFileLen}}}, 54 {r: "bytes=0-1,5-8", code: StatusPartialContent, ranges: []wantRange{{0, 2}, {5, 9}}}, 55 {r: "bytes=0-1,5-", code: StatusPartialContent, ranges: []wantRange{{0, 2}, {5, testFileLen}}}, 56 {r: "bytes=5-1000", code: StatusPartialContent, ranges: []wantRange{{5, testFileLen}}}, 57 {r: "bytes=0-,1-,2-,3-,4-", code: StatusOK}, // ignore wasteful range request 58 {r: "bytes=0-9", code: StatusPartialContent, ranges: []wantRange{{0, testFileLen - 1}}}, 59 {r: "bytes=0-10", code: StatusPartialContent, ranges: []wantRange{{0, testFileLen}}}, 60 {r: "bytes=0-11", code: StatusPartialContent, ranges: []wantRange{{0, testFileLen}}}, 61 {r: "bytes=10-11", code: StatusPartialContent, ranges: []wantRange{{testFileLen - 1, testFileLen}}}, 62 {r: "bytes=10-", code: StatusPartialContent, ranges: []wantRange{{testFileLen - 1, testFileLen}}}, 63 {r: "bytes=11-", code: StatusRequestedRangeNotSatisfiable}, 64 {r: "bytes=11-12", code: StatusRequestedRangeNotSatisfiable}, 65 {r: "bytes=12-12", code: StatusRequestedRangeNotSatisfiable}, 66 {r: "bytes=11-100", code: StatusRequestedRangeNotSatisfiable}, 67 {r: "bytes=12-100", code: StatusRequestedRangeNotSatisfiable}, 68 {r: "bytes=100-", code: StatusRequestedRangeNotSatisfiable}, 69 {r: "bytes=100-1000", code: StatusRequestedRangeNotSatisfiable}, 70 } 71 72 func TestServeFile(t *testing.T) { 73 defer afterTest(t) 74 ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) { 75 ServeFile(w, r, "testdata/file") 76 })) 77 defer ts.Close() 78 79 var err error 80 81 file, err := ioutil.ReadFile(testFile) 82 if err != nil { 83 t.Fatal("reading file:", err) 84 } 85 86 // set up the Request (re-used for all tests) 87 var req Request 88 req.Header = make(Header) 89 if req.URL, err = url.Parse(ts.URL); err != nil { 90 t.Fatal("ParseURL:", err) 91 } 92 req.Method = "GET" 93 94 // straight GET 95 _, body := getBody(t, "straight get", req) 96 if !bytes.Equal(body, file) { 97 t.Fatalf("body mismatch: got %q, want %q", body, file) 98 } 99 100 // Range tests 101 Cases: 102 for _, rt := range ServeFileRangeTests { 103 if rt.r != "" { 104 req.Header.Set("Range", rt.r) 105 } 106 resp, body := getBody(t, fmt.Sprintf("range test %q", rt.r), req) 107 if resp.StatusCode != rt.code { 108 t.Errorf("range=%q: StatusCode=%d, want %d", rt.r, resp.StatusCode, rt.code) 109 } 110 if rt.code == StatusRequestedRangeNotSatisfiable { 111 continue 112 } 113 wantContentRange := "" 114 if len(rt.ranges) == 1 { 115 rng := rt.ranges[0] 116 wantContentRange = fmt.Sprintf("bytes %d-%d/%d", rng.start, rng.end-1, testFileLen) 117 } 118 cr := resp.Header.Get("Content-Range") 119 if cr != wantContentRange { 120 t.Errorf("range=%q: Content-Range = %q, want %q", rt.r, cr, wantContentRange) 121 } 122 ct := resp.Header.Get("Content-Type") 123 if len(rt.ranges) == 1 { 124 rng := rt.ranges[0] 125 wantBody := file[rng.start:rng.end] 126 if !bytes.Equal(body, wantBody) { 127 t.Errorf("range=%q: body = %q, want %q", rt.r, body, wantBody) 128 } 129 if strings.HasPrefix(ct, "multipart/byteranges") { 130 t.Errorf("range=%q content-type = %q; unexpected multipart/byteranges", rt.r, ct) 131 } 132 } 133 if len(rt.ranges) > 1 { 134 typ, params, err := mime.ParseMediaType(ct) 135 if err != nil { 136 t.Errorf("range=%q content-type = %q; %v", rt.r, ct, err) 137 continue 138 } 139 if typ != "multipart/byteranges" { 140 t.Errorf("range=%q content-type = %q; want multipart/byteranges", rt.r, typ) 141 continue 142 } 143 if params["boundary"] == "" { 144 t.Errorf("range=%q content-type = %q; lacks boundary", rt.r, ct) 145 continue 146 } 147 if g, w := resp.ContentLength, int64(len(body)); g != w { 148 t.Errorf("range=%q Content-Length = %d; want %d", rt.r, g, w) 149 continue 150 } 151 mr := multipart.NewReader(bytes.NewReader(body), params["boundary"]) 152 for ri, rng := range rt.ranges { 153 part, err := mr.NextPart() 154 if err != nil { 155 t.Errorf("range=%q, reading part index %d: %v", rt.r, ri, err) 156 continue Cases 157 } 158 wantContentRange = fmt.Sprintf("bytes %d-%d/%d", rng.start, rng.end-1, testFileLen) 159 if g, w := part.Header.Get("Content-Range"), wantContentRange; g != w { 160 t.Errorf("range=%q: part Content-Range = %q; want %q", rt.r, g, w) 161 } 162 body, err := ioutil.ReadAll(part) 163 if err != nil { 164 t.Errorf("range=%q, reading part index %d body: %v", rt.r, ri, err) 165 continue Cases 166 } 167 wantBody := file[rng.start:rng.end] 168 if !bytes.Equal(body, wantBody) { 169 t.Errorf("range=%q: body = %q, want %q", rt.r, body, wantBody) 170 } 171 } 172 _, err = mr.NextPart() 173 if err != io.EOF { 174 t.Errorf("range=%q; expected final error io.EOF; got %v", rt.r, err) 175 } 176 } 177 } 178 } 179 180 var fsRedirectTestData = []struct { 181 original, redirect string 182 }{ 183 {"/test/index.html", "/test/"}, 184 {"/test/testdata", "/test/testdata/"}, 185 {"/test/testdata/file/", "/test/testdata/file"}, 186 } 187 188 func TestFSRedirect(t *testing.T) { 189 defer afterTest(t) 190 ts := httptest.NewServer(StripPrefix("/test", FileServer(Dir(".")))) 191 defer ts.Close() 192 193 for _, data := range fsRedirectTestData { 194 res, err := Get(ts.URL + data.original) 195 if err != nil { 196 t.Fatal(err) 197 } 198 res.Body.Close() 199 if g, e := res.Request.URL.Path, data.redirect; g != e { 200 t.Errorf("redirect from %s: got %s, want %s", data.original, g, e) 201 } 202 } 203 } 204 205 type testFileSystem struct { 206 open func(name string) (File, error) 207 } 208 209 func (fs *testFileSystem) Open(name string) (File, error) { 210 return fs.open(name) 211 } 212 213 func TestFileServerCleans(t *testing.T) { 214 defer afterTest(t) 215 ch := make(chan string, 1) 216 fs := FileServer(&testFileSystem{func(name string) (File, error) { 217 ch <- name 218 return nil, errors.New("file does not exist") 219 }}) 220 tests := []struct { 221 reqPath, openArg string 222 }{ 223 {"/foo.txt", "/foo.txt"}, 224 {"//foo.txt", "/foo.txt"}, 225 {"/../foo.txt", "/foo.txt"}, 226 } 227 req, _ := NewRequest("GET", "http://example.com", nil) 228 for n, test := range tests { 229 rec := httptest.NewRecorder() 230 req.URL.Path = test.reqPath 231 fs.ServeHTTP(rec, req) 232 if got := <-ch; got != test.openArg { 233 t.Errorf("test %d: got %q, want %q", n, got, test.openArg) 234 } 235 } 236 } 237 238 func TestFileServerEscapesNames(t *testing.T) { 239 defer afterTest(t) 240 const dirListPrefix = "<pre>\n" 241 const dirListSuffix = "\n</pre>\n" 242 tests := []struct { 243 name, escaped string 244 }{ 245 {`simple_name`, `<a href="simple_name">simple_name</a>`}, 246 {`"'<>&`, `<a href="%22%27%3C%3E&">"'<>&</a>`}, 247 {`?foo=bar#baz`, `<a href="%3Ffoo=bar%23baz">?foo=bar#baz</a>`}, 248 {`<combo>?foo`, `<a href="%3Ccombo%3E%3Ffoo"><combo>?foo</a>`}, 249 } 250 251 // We put each test file in its own directory in the fakeFS so we can look at it in isolation. 252 fs := make(fakeFS) 253 for i, test := range tests { 254 testFile := &fakeFileInfo{basename: test.name} 255 fs[fmt.Sprintf("/%d", i)] = &fakeFileInfo{ 256 dir: true, 257 modtime: time.Unix(1000000000, 0).UTC(), 258 ents: []*fakeFileInfo{testFile}, 259 } 260 fs[fmt.Sprintf("/%d/%s", i, test.name)] = testFile 261 } 262 263 ts := httptest.NewServer(FileServer(&fs)) 264 defer ts.Close() 265 for i, test := range tests { 266 url := fmt.Sprintf("%s/%d", ts.URL, i) 267 res, err := Get(url) 268 if err != nil { 269 t.Fatalf("test %q: Get: %v", test.name, err) 270 } 271 b, err := ioutil.ReadAll(res.Body) 272 if err != nil { 273 t.Fatalf("test %q: read Body: %v", test.name, err) 274 } 275 s := string(b) 276 if !strings.HasPrefix(s, dirListPrefix) || !strings.HasSuffix(s, dirListSuffix) { 277 t.Errorf("test %q: listing dir, full output is %q, want prefix %q and suffix %q", test.name, s, dirListPrefix, dirListSuffix) 278 } 279 if trimmed := strings.TrimSuffix(strings.TrimPrefix(s, dirListPrefix), dirListSuffix); trimmed != test.escaped { 280 t.Errorf("test %q: listing dir, filename escaped to %q, want %q", test.name, trimmed, test.escaped) 281 } 282 res.Body.Close() 283 } 284 } 285 286 func mustRemoveAll(dir string) { 287 err := os.RemoveAll(dir) 288 if err != nil { 289 panic(err) 290 } 291 } 292 293 func TestFileServerImplicitLeadingSlash(t *testing.T) { 294 defer afterTest(t) 295 tempDir, err := ioutil.TempDir("", "") 296 if err != nil { 297 t.Fatalf("TempDir: %v", err) 298 } 299 defer mustRemoveAll(tempDir) 300 if err := ioutil.WriteFile(filepath.Join(tempDir, "foo.txt"), []byte("Hello world"), 0644); err != nil { 301 t.Fatalf("WriteFile: %v", err) 302 } 303 ts := httptest.NewServer(StripPrefix("/bar/", FileServer(Dir(tempDir)))) 304 defer ts.Close() 305 get := func(suffix string) string { 306 res, err := Get(ts.URL + suffix) 307 if err != nil { 308 t.Fatalf("Get %s: %v", suffix, err) 309 } 310 b, err := ioutil.ReadAll(res.Body) 311 if err != nil { 312 t.Fatalf("ReadAll %s: %v", suffix, err) 313 } 314 res.Body.Close() 315 return string(b) 316 } 317 if s := get("/bar/"); !strings.Contains(s, ">foo.txt<") { 318 t.Logf("expected a directory listing with foo.txt, got %q", s) 319 } 320 if s := get("/bar/foo.txt"); s != "Hello world" { 321 t.Logf("expected %q, got %q", "Hello world", s) 322 } 323 } 324 325 func TestDirJoin(t *testing.T) { 326 if runtime.GOOS == "windows" { 327 t.Skip("skipping test on windows") 328 } 329 wfi, err := os.Stat("/etc/hosts") 330 if err != nil { 331 t.Skip("skipping test; no /etc/hosts file") 332 } 333 test := func(d Dir, name string) { 334 f, err := d.Open(name) 335 if err != nil { 336 t.Fatalf("open of %s: %v", name, err) 337 } 338 defer f.Close() 339 gfi, err := f.Stat() 340 if err != nil { 341 t.Fatalf("stat of %s: %v", name, err) 342 } 343 if !os.SameFile(gfi, wfi) { 344 t.Errorf("%s got different file", name) 345 } 346 } 347 test(Dir("/etc/"), "/hosts") 348 test(Dir("/etc/"), "hosts") 349 test(Dir("/etc/"), "../../../../hosts") 350 test(Dir("/etc"), "/hosts") 351 test(Dir("/etc"), "hosts") 352 test(Dir("/etc"), "../../../../hosts") 353 354 // Not really directories, but since we use this trick in 355 // ServeFile, test it: 356 test(Dir("/etc/hosts"), "") 357 test(Dir("/etc/hosts"), "/") 358 test(Dir("/etc/hosts"), "../") 359 } 360 361 func TestEmptyDirOpenCWD(t *testing.T) { 362 test := func(d Dir) { 363 name := "fs_test.go" 364 f, err := d.Open(name) 365 if err != nil { 366 t.Fatalf("open of %s: %v", name, err) 367 } 368 defer f.Close() 369 } 370 test(Dir("")) 371 test(Dir(".")) 372 test(Dir("./")) 373 } 374 375 func TestServeFileContentType(t *testing.T) { 376 defer afterTest(t) 377 const ctype = "icecream/chocolate" 378 ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) { 379 switch r.FormValue("override") { 380 case "1": 381 w.Header().Set("Content-Type", ctype) 382 case "2": 383 // Explicitly inhibit sniffing. 384 w.Header()["Content-Type"] = []string{} 385 } 386 ServeFile(w, r, "testdata/file") 387 })) 388 defer ts.Close() 389 get := func(override string, want []string) { 390 resp, err := Get(ts.URL + "?override=" + override) 391 if err != nil { 392 t.Fatal(err) 393 } 394 if h := resp.Header["Content-Type"]; !reflect.DeepEqual(h, want) { 395 t.Errorf("Content-Type mismatch: got %v, want %v", h, want) 396 } 397 resp.Body.Close() 398 } 399 get("0", []string{"text/plain; charset=utf-8"}) 400 get("1", []string{ctype}) 401 get("2", nil) 402 } 403 404 func TestServeFileMimeType(t *testing.T) { 405 defer afterTest(t) 406 ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) { 407 ServeFile(w, r, "testdata/style.css") 408 })) 409 defer ts.Close() 410 resp, err := Get(ts.URL) 411 if err != nil { 412 t.Fatal(err) 413 } 414 resp.Body.Close() 415 want := "text/css; charset=utf-8" 416 if h := resp.Header.Get("Content-Type"); h != want { 417 t.Errorf("Content-Type mismatch: got %q, want %q", h, want) 418 } 419 } 420 421 func TestServeFileFromCWD(t *testing.T) { 422 defer afterTest(t) 423 ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) { 424 ServeFile(w, r, "fs_test.go") 425 })) 426 defer ts.Close() 427 r, err := Get(ts.URL) 428 if err != nil { 429 t.Fatal(err) 430 } 431 r.Body.Close() 432 if r.StatusCode != 200 { 433 t.Fatalf("expected 200 OK, got %s", r.Status) 434 } 435 } 436 437 func TestServeFileWithContentEncoding(t *testing.T) { 438 defer afterTest(t) 439 ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) { 440 w.Header().Set("Content-Encoding", "foo") 441 ServeFile(w, r, "testdata/file") 442 })) 443 defer ts.Close() 444 resp, err := Get(ts.URL) 445 if err != nil { 446 t.Fatal(err) 447 } 448 resp.Body.Close() 449 if g, e := resp.ContentLength, int64(-1); g != e { 450 t.Errorf("Content-Length mismatch: got %d, want %d", g, e) 451 } 452 } 453 454 func TestServeIndexHtml(t *testing.T) { 455 defer afterTest(t) 456 const want = "index.html says hello\n" 457 ts := httptest.NewServer(FileServer(Dir("."))) 458 defer ts.Close() 459 460 for _, path := range []string{"/testdata/", "/testdata/index.html"} { 461 res, err := Get(ts.URL + path) 462 if err != nil { 463 t.Fatal(err) 464 } 465 b, err := ioutil.ReadAll(res.Body) 466 if err != nil { 467 t.Fatal("reading Body:", err) 468 } 469 if s := string(b); s != want { 470 t.Errorf("for path %q got %q, want %q", path, s, want) 471 } 472 res.Body.Close() 473 } 474 } 475 476 func TestFileServerZeroByte(t *testing.T) { 477 defer afterTest(t) 478 ts := httptest.NewServer(FileServer(Dir("."))) 479 defer ts.Close() 480 481 res, err := Get(ts.URL + "/..\x00") 482 if err != nil { 483 t.Fatal(err) 484 } 485 b, err := ioutil.ReadAll(res.Body) 486 if err != nil { 487 t.Fatal("reading Body:", err) 488 } 489 if res.StatusCode == 200 { 490 t.Errorf("got status 200; want an error. Body is:\n%s", string(b)) 491 } 492 } 493 494 type fakeFileInfo struct { 495 dir bool 496 basename string 497 modtime time.Time 498 ents []*fakeFileInfo 499 contents string 500 } 501 502 func (f *fakeFileInfo) Name() string { return f.basename } 503 func (f *fakeFileInfo) Sys() interface{} { return nil } 504 func (f *fakeFileInfo) ModTime() time.Time { return f.modtime } 505 func (f *fakeFileInfo) IsDir() bool { return f.dir } 506 func (f *fakeFileInfo) Size() int64 { return int64(len(f.contents)) } 507 func (f *fakeFileInfo) Mode() os.FileMode { 508 if f.dir { 509 return 0755 | os.ModeDir 510 } 511 return 0644 512 } 513 514 type fakeFile struct { 515 io.ReadSeeker 516 fi *fakeFileInfo 517 path string // as opened 518 entpos int 519 } 520 521 func (f *fakeFile) Close() error { return nil } 522 func (f *fakeFile) Stat() (os.FileInfo, error) { return f.fi, nil } 523 func (f *fakeFile) Readdir(count int) ([]os.FileInfo, error) { 524 if !f.fi.dir { 525 return nil, os.ErrInvalid 526 } 527 var fis []os.FileInfo 528 529 limit := f.entpos + count 530 if count <= 0 || limit > len(f.fi.ents) { 531 limit = len(f.fi.ents) 532 } 533 for ; f.entpos < limit; f.entpos++ { 534 fis = append(fis, f.fi.ents[f.entpos]) 535 } 536 537 if len(fis) == 0 && count > 0 { 538 return fis, io.EOF 539 } else { 540 return fis, nil 541 } 542 } 543 544 type fakeFS map[string]*fakeFileInfo 545 546 func (fs fakeFS) Open(name string) (File, error) { 547 name = path.Clean(name) 548 f, ok := fs[name] 549 if !ok { 550 return nil, os.ErrNotExist 551 } 552 return &fakeFile{ReadSeeker: strings.NewReader(f.contents), fi: f, path: name}, nil 553 } 554 555 func TestDirectoryIfNotModified(t *testing.T) { 556 defer afterTest(t) 557 const indexContents = "I am a fake index.html file" 558 fileMod := time.Unix(1000000000, 0).UTC() 559 fileModStr := fileMod.Format(TimeFormat) 560 dirMod := time.Unix(123, 0).UTC() 561 indexFile := &fakeFileInfo{ 562 basename: "index.html", 563 modtime: fileMod, 564 contents: indexContents, 565 } 566 fs := fakeFS{ 567 "/": &fakeFileInfo{ 568 dir: true, 569 modtime: dirMod, 570 ents: []*fakeFileInfo{indexFile}, 571 }, 572 "/index.html": indexFile, 573 } 574 575 ts := httptest.NewServer(FileServer(fs)) 576 defer ts.Close() 577 578 res, err := Get(ts.URL) 579 if err != nil { 580 t.Fatal(err) 581 } 582 b, err := ioutil.ReadAll(res.Body) 583 if err != nil { 584 t.Fatal(err) 585 } 586 if string(b) != indexContents { 587 t.Fatalf("Got body %q; want %q", b, indexContents) 588 } 589 res.Body.Close() 590 591 lastMod := res.Header.Get("Last-Modified") 592 if lastMod != fileModStr { 593 t.Fatalf("initial Last-Modified = %q; want %q", lastMod, fileModStr) 594 } 595 596 req, _ := NewRequest("GET", ts.URL, nil) 597 req.Header.Set("If-Modified-Since", lastMod) 598 599 res, err = DefaultClient.Do(req) 600 if err != nil { 601 t.Fatal(err) 602 } 603 if res.StatusCode != 304 { 604 t.Fatalf("Code after If-Modified-Since request = %v; want 304", res.StatusCode) 605 } 606 res.Body.Close() 607 608 // Advance the index.html file's modtime, but not the directory's. 609 indexFile.modtime = indexFile.modtime.Add(1 * time.Hour) 610 611 res, err = DefaultClient.Do(req) 612 if err != nil { 613 t.Fatal(err) 614 } 615 if res.StatusCode != 200 { 616 t.Fatalf("Code after second If-Modified-Since request = %v; want 200; res is %#v", res.StatusCode, res) 617 } 618 res.Body.Close() 619 } 620 621 func mustStat(t *testing.T, fileName string) os.FileInfo { 622 fi, err := os.Stat(fileName) 623 if err != nil { 624 t.Fatal(err) 625 } 626 return fi 627 } 628 629 func TestServeContent(t *testing.T) { 630 defer afterTest(t) 631 type serveParam struct { 632 name string 633 modtime time.Time 634 content io.ReadSeeker 635 contentType string 636 etag string 637 } 638 servec := make(chan serveParam, 1) 639 ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) { 640 p := <-servec 641 if p.etag != "" { 642 w.Header().Set("ETag", p.etag) 643 } 644 if p.contentType != "" { 645 w.Header().Set("Content-Type", p.contentType) 646 } 647 ServeContent(w, r, p.name, p.modtime, p.content) 648 })) 649 defer ts.Close() 650 651 type testCase struct { 652 // One of file or content must be set: 653 file string 654 content io.ReadSeeker 655 656 modtime time.Time 657 serveETag string // optional 658 serveContentType string // optional 659 reqHeader map[string]string 660 wantLastMod string 661 wantContentType string 662 wantStatus int 663 } 664 htmlModTime := mustStat(t, "testdata/index.html").ModTime() 665 tests := map[string]testCase{ 666 "no_last_modified": { 667 file: "testdata/style.css", 668 wantContentType: "text/css; charset=utf-8", 669 wantStatus: 200, 670 }, 671 "with_last_modified": { 672 file: "testdata/index.html", 673 wantContentType: "text/html; charset=utf-8", 674 modtime: htmlModTime, 675 wantLastMod: htmlModTime.UTC().Format(TimeFormat), 676 wantStatus: 200, 677 }, 678 "not_modified_modtime": { 679 file: "testdata/style.css", 680 modtime: htmlModTime, 681 reqHeader: map[string]string{ 682 "If-Modified-Since": htmlModTime.UTC().Format(TimeFormat), 683 }, 684 wantStatus: 304, 685 }, 686 "not_modified_modtime_with_contenttype": { 687 file: "testdata/style.css", 688 serveContentType: "text/css", // explicit content type 689 modtime: htmlModTime, 690 reqHeader: map[string]string{ 691 "If-Modified-Since": htmlModTime.UTC().Format(TimeFormat), 692 }, 693 wantStatus: 304, 694 }, 695 "not_modified_etag": { 696 file: "testdata/style.css", 697 serveETag: `"foo"`, 698 reqHeader: map[string]string{ 699 "If-None-Match": `"foo"`, 700 }, 701 wantStatus: 304, 702 }, 703 "not_modified_etag_no_seek": { 704 content: panicOnSeek{nil}, // should never be called 705 serveETag: `"foo"`, 706 reqHeader: map[string]string{ 707 "If-None-Match": `"foo"`, 708 }, 709 wantStatus: 304, 710 }, 711 "range_good": { 712 file: "testdata/style.css", 713 serveETag: `"A"`, 714 reqHeader: map[string]string{ 715 "Range": "bytes=0-4", 716 }, 717 wantStatus: StatusPartialContent, 718 wantContentType: "text/css; charset=utf-8", 719 }, 720 // An If-Range resource for entity "A", but entity "B" is now current. 721 // The Range request should be ignored. 722 "range_no_match": { 723 file: "testdata/style.css", 724 serveETag: `"A"`, 725 reqHeader: map[string]string{ 726 "Range": "bytes=0-4", 727 "If-Range": `"B"`, 728 }, 729 wantStatus: 200, 730 wantContentType: "text/css; charset=utf-8", 731 }, 732 "range_with_modtime": { 733 file: "testdata/style.css", 734 modtime: time.Date(2014, 6, 25, 17, 12, 18, 0 /* nanos */, time.UTC), 735 reqHeader: map[string]string{ 736 "Range": "bytes=0-4", 737 "If-Range": "Wed, 25 Jun 2014 17:12:18 GMT", 738 }, 739 wantStatus: StatusPartialContent, 740 wantContentType: "text/css; charset=utf-8", 741 wantLastMod: "Wed, 25 Jun 2014 17:12:18 GMT", 742 }, 743 "range_with_modtime_nanos": { 744 file: "testdata/style.css", 745 modtime: time.Date(2014, 6, 25, 17, 12, 18, 123 /* nanos */, time.UTC), 746 reqHeader: map[string]string{ 747 "Range": "bytes=0-4", 748 "If-Range": "Wed, 25 Jun 2014 17:12:18 GMT", 749 }, 750 wantStatus: StatusPartialContent, 751 wantContentType: "text/css; charset=utf-8", 752 wantLastMod: "Wed, 25 Jun 2014 17:12:18 GMT", 753 }, 754 } 755 for testName, tt := range tests { 756 var content io.ReadSeeker 757 if tt.file != "" { 758 f, err := os.Open(tt.file) 759 if err != nil { 760 t.Fatalf("test %q: %v", testName, err) 761 } 762 defer f.Close() 763 content = f 764 } else { 765 content = tt.content 766 } 767 768 servec <- serveParam{ 769 name: filepath.Base(tt.file), 770 content: content, 771 modtime: tt.modtime, 772 etag: tt.serveETag, 773 contentType: tt.serveContentType, 774 } 775 req, err := NewRequest("GET", ts.URL, nil) 776 if err != nil { 777 t.Fatal(err) 778 } 779 for k, v := range tt.reqHeader { 780 req.Header.Set(k, v) 781 } 782 res, err := DefaultClient.Do(req) 783 if err != nil { 784 t.Fatal(err) 785 } 786 io.Copy(ioutil.Discard, res.Body) 787 res.Body.Close() 788 if res.StatusCode != tt.wantStatus { 789 t.Errorf("test %q: status = %d; want %d", testName, res.StatusCode, tt.wantStatus) 790 } 791 if g, e := res.Header.Get("Content-Type"), tt.wantContentType; g != e { 792 t.Errorf("test %q: content-type = %q, want %q", testName, g, e) 793 } 794 if g, e := res.Header.Get("Last-Modified"), tt.wantLastMod; g != e { 795 t.Errorf("test %q: last-modified = %q, want %q", testName, g, e) 796 } 797 } 798 } 799 800 // verifies that sendfile is being used on Linux 801 func TestLinuxSendfile(t *testing.T) { 802 defer afterTest(t) 803 if runtime.GOOS != "linux" { 804 t.Skip("skipping; linux-only test") 805 } 806 if _, err := exec.LookPath("strace"); err != nil { 807 t.Skip("skipping; strace not found in path") 808 } 809 810 ln, err := net.Listen("tcp", "127.0.0.1:0") 811 if err != nil { 812 t.Fatal(err) 813 } 814 lnf, err := ln.(*net.TCPListener).File() 815 if err != nil { 816 t.Fatal(err) 817 } 818 defer ln.Close() 819 820 var buf bytes.Buffer 821 child := exec.Command("strace", "-f", "-q", "-e", "trace=sendfile,sendfile64", os.Args[0], "-test.run=TestLinuxSendfileChild") 822 child.ExtraFiles = append(child.ExtraFiles, lnf) 823 child.Env = append([]string{"GO_WANT_HELPER_PROCESS=1"}, os.Environ()...) 824 child.Stdout = &buf 825 child.Stderr = &buf 826 if err := child.Start(); err != nil { 827 t.Skipf("skipping; failed to start straced child: %v", err) 828 } 829 830 res, err := Get(fmt.Sprintf("http://%s/", ln.Addr())) 831 if err != nil { 832 t.Fatalf("http client error: %v", err) 833 } 834 _, err = io.Copy(ioutil.Discard, res.Body) 835 if err != nil { 836 t.Fatalf("client body read error: %v", err) 837 } 838 res.Body.Close() 839 840 // Force child to exit cleanly. 841 Get(fmt.Sprintf("http://%s/quit", ln.Addr())) 842 child.Wait() 843 844 rx := regexp.MustCompile(`sendfile(64)?\(\d+,\s*\d+,\s*NULL,\s*\d+\)\s*=\s*\d+\s*\n`) 845 rxResume := regexp.MustCompile(`<\.\.\. sendfile(64)? resumed> \)\s*=\s*\d+\s*\n`) 846 out := buf.String() 847 if !rx.MatchString(out) && !rxResume.MatchString(out) { 848 t.Errorf("no sendfile system call found in:\n%s", out) 849 } 850 } 851 852 func getBody(t *testing.T, testName string, req Request) (*Response, []byte) { 853 r, err := DefaultClient.Do(&req) 854 if err != nil { 855 t.Fatalf("%s: for URL %q, send error: %v", testName, req.URL.String(), err) 856 } 857 b, err := ioutil.ReadAll(r.Body) 858 if err != nil { 859 t.Fatalf("%s: for URL %q, reading body: %v", testName, req.URL.String(), err) 860 } 861 return r, b 862 } 863 864 // TestLinuxSendfileChild isn't a real test. It's used as a helper process 865 // for TestLinuxSendfile. 866 func TestLinuxSendfileChild(*testing.T) { 867 if os.Getenv("GO_WANT_HELPER_PROCESS") != "1" { 868 return 869 } 870 defer os.Exit(0) 871 fd3 := os.NewFile(3, "ephemeral-port-listener") 872 ln, err := net.FileListener(fd3) 873 if err != nil { 874 panic(err) 875 } 876 mux := NewServeMux() 877 mux.Handle("/", FileServer(Dir("testdata"))) 878 mux.HandleFunc("/quit", func(ResponseWriter, *Request) { 879 os.Exit(0) 880 }) 881 s := &Server{Handler: mux} 882 err = s.Serve(ln) 883 if err != nil { 884 panic(err) 885 } 886 } 887 888 func TestFileServerCleanPath(t *testing.T) { 889 tests := []struct { 890 path string 891 wantCode int 892 wantOpen []string 893 }{ 894 {"/", 200, []string{"/", "/index.html"}}, 895 {"/dir", 301, []string{"/dir"}}, 896 {"/dir/", 200, []string{"/dir", "/dir/index.html"}}, 897 } 898 for _, tt := range tests { 899 var log []string 900 rr := httptest.NewRecorder() 901 req, _ := NewRequest("GET", "http://foo.localhost"+tt.path, nil) 902 FileServer(fileServerCleanPathDir{&log}).ServeHTTP(rr, req) 903 if !reflect.DeepEqual(log, tt.wantOpen) { 904 t.Logf("For %s: Opens = %q; want %q", tt.path, log, tt.wantOpen) 905 } 906 if rr.Code != tt.wantCode { 907 t.Logf("For %s: Response code = %d; want %d", tt.path, rr.Code, tt.wantCode) 908 } 909 } 910 } 911 912 type fileServerCleanPathDir struct { 913 log *[]string 914 } 915 916 func (d fileServerCleanPathDir) Open(path string) (File, error) { 917 *(d.log) = append(*(d.log), path) 918 if path == "/" || path == "/dir" || path == "/dir/" { 919 // Just return back something that's a directory. 920 return Dir(".").Open(".") 921 } 922 return nil, os.ErrNotExist 923 } 924 925 type panicOnSeek struct{ io.ReadSeeker }