github.com/c0deoo1/golang1.5@v0.0.0-20220525150107-c87c805d4593/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 err error 501 } 502 503 func (f *fakeFileInfo) Name() string { return f.basename } 504 func (f *fakeFileInfo) Sys() interface{} { return nil } 505 func (f *fakeFileInfo) ModTime() time.Time { return f.modtime } 506 func (f *fakeFileInfo) IsDir() bool { return f.dir } 507 func (f *fakeFileInfo) Size() int64 { return int64(len(f.contents)) } 508 func (f *fakeFileInfo) Mode() os.FileMode { 509 if f.dir { 510 return 0755 | os.ModeDir 511 } 512 return 0644 513 } 514 515 type fakeFile struct { 516 io.ReadSeeker 517 fi *fakeFileInfo 518 path string // as opened 519 entpos int 520 } 521 522 func (f *fakeFile) Close() error { return nil } 523 func (f *fakeFile) Stat() (os.FileInfo, error) { return f.fi, nil } 524 func (f *fakeFile) Readdir(count int) ([]os.FileInfo, error) { 525 if !f.fi.dir { 526 return nil, os.ErrInvalid 527 } 528 var fis []os.FileInfo 529 530 limit := f.entpos + count 531 if count <= 0 || limit > len(f.fi.ents) { 532 limit = len(f.fi.ents) 533 } 534 for ; f.entpos < limit; f.entpos++ { 535 fis = append(fis, f.fi.ents[f.entpos]) 536 } 537 538 if len(fis) == 0 && count > 0 { 539 return fis, io.EOF 540 } else { 541 return fis, nil 542 } 543 } 544 545 type fakeFS map[string]*fakeFileInfo 546 547 func (fs fakeFS) Open(name string) (File, error) { 548 name = path.Clean(name) 549 f, ok := fs[name] 550 if !ok { 551 return nil, os.ErrNotExist 552 } 553 if f.err != nil { 554 return nil, f.err 555 } 556 return &fakeFile{ReadSeeker: strings.NewReader(f.contents), fi: f, path: name}, nil 557 } 558 559 func TestDirectoryIfNotModified(t *testing.T) { 560 defer afterTest(t) 561 const indexContents = "I am a fake index.html file" 562 fileMod := time.Unix(1000000000, 0).UTC() 563 fileModStr := fileMod.Format(TimeFormat) 564 dirMod := time.Unix(123, 0).UTC() 565 indexFile := &fakeFileInfo{ 566 basename: "index.html", 567 modtime: fileMod, 568 contents: indexContents, 569 } 570 fs := fakeFS{ 571 "/": &fakeFileInfo{ 572 dir: true, 573 modtime: dirMod, 574 ents: []*fakeFileInfo{indexFile}, 575 }, 576 "/index.html": indexFile, 577 } 578 579 ts := httptest.NewServer(FileServer(fs)) 580 defer ts.Close() 581 582 res, err := Get(ts.URL) 583 if err != nil { 584 t.Fatal(err) 585 } 586 b, err := ioutil.ReadAll(res.Body) 587 if err != nil { 588 t.Fatal(err) 589 } 590 if string(b) != indexContents { 591 t.Fatalf("Got body %q; want %q", b, indexContents) 592 } 593 res.Body.Close() 594 595 lastMod := res.Header.Get("Last-Modified") 596 if lastMod != fileModStr { 597 t.Fatalf("initial Last-Modified = %q; want %q", lastMod, fileModStr) 598 } 599 600 req, _ := NewRequest("GET", ts.URL, nil) 601 req.Header.Set("If-Modified-Since", lastMod) 602 603 res, err = DefaultClient.Do(req) 604 if err != nil { 605 t.Fatal(err) 606 } 607 if res.StatusCode != 304 { 608 t.Fatalf("Code after If-Modified-Since request = %v; want 304", res.StatusCode) 609 } 610 res.Body.Close() 611 612 // Advance the index.html file's modtime, but not the directory's. 613 indexFile.modtime = indexFile.modtime.Add(1 * time.Hour) 614 615 res, err = DefaultClient.Do(req) 616 if err != nil { 617 t.Fatal(err) 618 } 619 if res.StatusCode != 200 { 620 t.Fatalf("Code after second If-Modified-Since request = %v; want 200; res is %#v", res.StatusCode, res) 621 } 622 res.Body.Close() 623 } 624 625 func mustStat(t *testing.T, fileName string) os.FileInfo { 626 fi, err := os.Stat(fileName) 627 if err != nil { 628 t.Fatal(err) 629 } 630 return fi 631 } 632 633 func TestServeContent(t *testing.T) { 634 defer afterTest(t) 635 type serveParam struct { 636 name string 637 modtime time.Time 638 content io.ReadSeeker 639 contentType string 640 etag string 641 } 642 servec := make(chan serveParam, 1) 643 ts := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) { 644 p := <-servec 645 if p.etag != "" { 646 w.Header().Set("ETag", p.etag) 647 } 648 if p.contentType != "" { 649 w.Header().Set("Content-Type", p.contentType) 650 } 651 ServeContent(w, r, p.name, p.modtime, p.content) 652 })) 653 defer ts.Close() 654 655 type testCase struct { 656 // One of file or content must be set: 657 file string 658 content io.ReadSeeker 659 660 modtime time.Time 661 serveETag string // optional 662 serveContentType string // optional 663 reqHeader map[string]string 664 wantLastMod string 665 wantContentType string 666 wantStatus int 667 } 668 htmlModTime := mustStat(t, "testdata/index.html").ModTime() 669 tests := map[string]testCase{ 670 "no_last_modified": { 671 file: "testdata/style.css", 672 wantContentType: "text/css; charset=utf-8", 673 wantStatus: 200, 674 }, 675 "with_last_modified": { 676 file: "testdata/index.html", 677 wantContentType: "text/html; charset=utf-8", 678 modtime: htmlModTime, 679 wantLastMod: htmlModTime.UTC().Format(TimeFormat), 680 wantStatus: 200, 681 }, 682 "not_modified_modtime": { 683 file: "testdata/style.css", 684 modtime: htmlModTime, 685 reqHeader: map[string]string{ 686 "If-Modified-Since": htmlModTime.UTC().Format(TimeFormat), 687 }, 688 wantStatus: 304, 689 }, 690 "not_modified_modtime_with_contenttype": { 691 file: "testdata/style.css", 692 serveContentType: "text/css", // explicit content type 693 modtime: htmlModTime, 694 reqHeader: map[string]string{ 695 "If-Modified-Since": htmlModTime.UTC().Format(TimeFormat), 696 }, 697 wantStatus: 304, 698 }, 699 "not_modified_etag": { 700 file: "testdata/style.css", 701 serveETag: `"foo"`, 702 reqHeader: map[string]string{ 703 "If-None-Match": `"foo"`, 704 }, 705 wantStatus: 304, 706 }, 707 "not_modified_etag_no_seek": { 708 content: panicOnSeek{nil}, // should never be called 709 serveETag: `"foo"`, 710 reqHeader: map[string]string{ 711 "If-None-Match": `"foo"`, 712 }, 713 wantStatus: 304, 714 }, 715 "range_good": { 716 file: "testdata/style.css", 717 serveETag: `"A"`, 718 reqHeader: map[string]string{ 719 "Range": "bytes=0-4", 720 }, 721 wantStatus: StatusPartialContent, 722 wantContentType: "text/css; charset=utf-8", 723 }, 724 // An If-Range resource for entity "A", but entity "B" is now current. 725 // The Range request should be ignored. 726 "range_no_match": { 727 file: "testdata/style.css", 728 serveETag: `"A"`, 729 reqHeader: map[string]string{ 730 "Range": "bytes=0-4", 731 "If-Range": `"B"`, 732 }, 733 wantStatus: 200, 734 wantContentType: "text/css; charset=utf-8", 735 }, 736 "range_with_modtime": { 737 file: "testdata/style.css", 738 modtime: time.Date(2014, 6, 25, 17, 12, 18, 0 /* nanos */, time.UTC), 739 reqHeader: map[string]string{ 740 "Range": "bytes=0-4", 741 "If-Range": "Wed, 25 Jun 2014 17:12:18 GMT", 742 }, 743 wantStatus: StatusPartialContent, 744 wantContentType: "text/css; charset=utf-8", 745 wantLastMod: "Wed, 25 Jun 2014 17:12:18 GMT", 746 }, 747 "range_with_modtime_nanos": { 748 file: "testdata/style.css", 749 modtime: time.Date(2014, 6, 25, 17, 12, 18, 123 /* nanos */, time.UTC), 750 reqHeader: map[string]string{ 751 "Range": "bytes=0-4", 752 "If-Range": "Wed, 25 Jun 2014 17:12:18 GMT", 753 }, 754 wantStatus: StatusPartialContent, 755 wantContentType: "text/css; charset=utf-8", 756 wantLastMod: "Wed, 25 Jun 2014 17:12:18 GMT", 757 }, 758 "unix_zero_modtime": { 759 content: strings.NewReader("<html>foo"), 760 modtime: time.Unix(0, 0), 761 wantStatus: StatusOK, 762 wantContentType: "text/html; charset=utf-8", 763 }, 764 } 765 for testName, tt := range tests { 766 var content io.ReadSeeker 767 if tt.file != "" { 768 f, err := os.Open(tt.file) 769 if err != nil { 770 t.Fatalf("test %q: %v", testName, err) 771 } 772 defer f.Close() 773 content = f 774 } else { 775 content = tt.content 776 } 777 778 servec <- serveParam{ 779 name: filepath.Base(tt.file), 780 content: content, 781 modtime: tt.modtime, 782 etag: tt.serveETag, 783 contentType: tt.serveContentType, 784 } 785 req, err := NewRequest("GET", ts.URL, nil) 786 if err != nil { 787 t.Fatal(err) 788 } 789 for k, v := range tt.reqHeader { 790 req.Header.Set(k, v) 791 } 792 res, err := DefaultClient.Do(req) 793 if err != nil { 794 t.Fatal(err) 795 } 796 io.Copy(ioutil.Discard, res.Body) 797 res.Body.Close() 798 if res.StatusCode != tt.wantStatus { 799 t.Errorf("test %q: status = %d; want %d", testName, res.StatusCode, tt.wantStatus) 800 } 801 if g, e := res.Header.Get("Content-Type"), tt.wantContentType; g != e { 802 t.Errorf("test %q: content-type = %q, want %q", testName, g, e) 803 } 804 if g, e := res.Header.Get("Last-Modified"), tt.wantLastMod; g != e { 805 t.Errorf("test %q: last-modified = %q, want %q", testName, g, e) 806 } 807 } 808 } 809 810 func TestServeContentErrorMessages(t *testing.T) { 811 defer afterTest(t) 812 fs := fakeFS{ 813 "/500": &fakeFileInfo{ 814 err: errors.New("random error"), 815 }, 816 "/403": &fakeFileInfo{ 817 err: &os.PathError{Err: os.ErrPermission}, 818 }, 819 } 820 ts := httptest.NewServer(FileServer(fs)) 821 defer ts.Close() 822 for _, code := range []int{403, 404, 500} { 823 res, err := DefaultClient.Get(fmt.Sprintf("%s/%d", ts.URL, code)) 824 if err != nil { 825 t.Errorf("Error fetching /%d: %v", code, err) 826 continue 827 } 828 if res.StatusCode != code { 829 t.Errorf("For /%d, status code = %d; want %d", code, res.StatusCode, code) 830 } 831 res.Body.Close() 832 } 833 } 834 835 // verifies that sendfile is being used on Linux 836 func TestLinuxSendfile(t *testing.T) { 837 defer afterTest(t) 838 if runtime.GOOS != "linux" { 839 t.Skip("skipping; linux-only test") 840 } 841 if _, err := exec.LookPath("strace"); err != nil { 842 t.Skip("skipping; strace not found in path") 843 } 844 845 ln, err := net.Listen("tcp", "127.0.0.1:0") 846 if err != nil { 847 t.Fatal(err) 848 } 849 lnf, err := ln.(*net.TCPListener).File() 850 if err != nil { 851 t.Fatal(err) 852 } 853 defer ln.Close() 854 855 var buf bytes.Buffer 856 child := exec.Command("strace", "-f", "-q", "-e", "trace=sendfile,sendfile64", os.Args[0], "-test.run=TestLinuxSendfileChild") 857 child.ExtraFiles = append(child.ExtraFiles, lnf) 858 child.Env = append([]string{"GO_WANT_HELPER_PROCESS=1"}, os.Environ()...) 859 child.Stdout = &buf 860 child.Stderr = &buf 861 if err := child.Start(); err != nil { 862 t.Skipf("skipping; failed to start straced child: %v", err) 863 } 864 865 res, err := Get(fmt.Sprintf("http://%s/", ln.Addr())) 866 if err != nil { 867 t.Fatalf("http client error: %v", err) 868 } 869 _, err = io.Copy(ioutil.Discard, res.Body) 870 if err != nil { 871 t.Fatalf("client body read error: %v", err) 872 } 873 res.Body.Close() 874 875 // Force child to exit cleanly. 876 Get(fmt.Sprintf("http://%s/quit", ln.Addr())) 877 child.Wait() 878 879 rx := regexp.MustCompile(`sendfile(64)?\(\d+,\s*\d+,\s*NULL,\s*\d+\)\s*=\s*\d+\s*\n`) 880 rxResume := regexp.MustCompile(`<\.\.\. sendfile(64)? resumed> \)\s*=\s*\d+\s*\n`) 881 out := buf.String() 882 if !rx.MatchString(out) && !rxResume.MatchString(out) { 883 t.Errorf("no sendfile system call found in:\n%s", out) 884 } 885 } 886 887 func getBody(t *testing.T, testName string, req Request) (*Response, []byte) { 888 r, err := DefaultClient.Do(&req) 889 if err != nil { 890 t.Fatalf("%s: for URL %q, send error: %v", testName, req.URL.String(), err) 891 } 892 b, err := ioutil.ReadAll(r.Body) 893 if err != nil { 894 t.Fatalf("%s: for URL %q, reading body: %v", testName, req.URL.String(), err) 895 } 896 return r, b 897 } 898 899 // TestLinuxSendfileChild isn't a real test. It's used as a helper process 900 // for TestLinuxSendfile. 901 func TestLinuxSendfileChild(*testing.T) { 902 if os.Getenv("GO_WANT_HELPER_PROCESS") != "1" { 903 return 904 } 905 defer os.Exit(0) 906 fd3 := os.NewFile(3, "ephemeral-port-listener") 907 ln, err := net.FileListener(fd3) 908 if err != nil { 909 panic(err) 910 } 911 mux := NewServeMux() 912 mux.Handle("/", FileServer(Dir("testdata"))) 913 mux.HandleFunc("/quit", func(ResponseWriter, *Request) { 914 os.Exit(0) 915 }) 916 s := &Server{Handler: mux} 917 err = s.Serve(ln) 918 if err != nil { 919 panic(err) 920 } 921 } 922 923 func TestFileServerCleanPath(t *testing.T) { 924 tests := []struct { 925 path string 926 wantCode int 927 wantOpen []string 928 }{ 929 {"/", 200, []string{"/", "/index.html"}}, 930 {"/dir", 301, []string{"/dir"}}, 931 {"/dir/", 200, []string{"/dir", "/dir/index.html"}}, 932 } 933 for _, tt := range tests { 934 var log []string 935 rr := httptest.NewRecorder() 936 req, _ := NewRequest("GET", "http://foo.localhost"+tt.path, nil) 937 FileServer(fileServerCleanPathDir{&log}).ServeHTTP(rr, req) 938 if !reflect.DeepEqual(log, tt.wantOpen) { 939 t.Logf("For %s: Opens = %q; want %q", tt.path, log, tt.wantOpen) 940 } 941 if rr.Code != tt.wantCode { 942 t.Logf("For %s: Response code = %d; want %d", tt.path, rr.Code, tt.wantCode) 943 } 944 } 945 } 946 947 type fileServerCleanPathDir struct { 948 log *[]string 949 } 950 951 func (d fileServerCleanPathDir) Open(path string) (File, error) { 952 *(d.log) = append(*(d.log), path) 953 if path == "/" || path == "/dir" || path == "/dir/" { 954 // Just return back something that's a directory. 955 return Dir(".").Open(".") 956 } 957 return nil, os.ErrNotExist 958 } 959 960 type panicOnSeek struct{ io.ReadSeeker }