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&">&#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  }
   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 }