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&">&#34;&#39;&lt;&gt;&amp;</a>`},
   247  		{`?foo=bar#baz`, `<a href="%3Ffoo=bar%23baz">?foo=bar#baz</a>`},
   248  		{`<combo>?foo`, `<a href="%3Ccombo%3E%3Ffoo">&lt;combo&gt;?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 }