...
Run Format

Source file src/net/http/sniff_test.go

Documentation: net/http

  // Copyright 2011 The Go Authors. All rights reserved.
  // Use of this source code is governed by a BSD-style
  // license that can be found in the LICENSE file.
  
  package http_test
  
  import (
  	"bytes"
  	"fmt"
  	"io"
  	"io/ioutil"
  	"log"
  	. "net/http"
  	"reflect"
  	"strconv"
  	"strings"
  	"testing"
  )
  
  var sniffTests = []struct {
  	desc        string
  	data        []byte
  	contentType string
  }{
  	// Some nonsense.
  	{"Empty", []byte{}, "text/plain; charset=utf-8"},
  	{"Binary", []byte{1, 2, 3}, "application/octet-stream"},
  
  	{"HTML document #1", []byte(`<HtMl><bOdY>blah blah blah</body></html>`), "text/html; charset=utf-8"},
  	{"HTML document #2", []byte(`<HTML></HTML>`), "text/html; charset=utf-8"},
  	{"HTML document #3 (leading whitespace)", []byte(`   <!DOCTYPE HTML>...`), "text/html; charset=utf-8"},
  	{"HTML document #4 (leading CRLF)", []byte("\r\n<html>..."), "text/html; charset=utf-8"},
  
  	{"Plain text", []byte(`This is not HTML. It has ☃ though.`), "text/plain; charset=utf-8"},
  
  	{"XML", []byte("\n<?xml!"), "text/xml; charset=utf-8"},
  
  	// Image types.
  	{"GIF 87a", []byte(`GIF87a`), "image/gif"},
  	{"GIF 89a", []byte(`GIF89a...`), "image/gif"},
  
  	// Audio types.
  	{"MIDI audio", []byte("MThd\x00\x00\x00\x06\x00\x01"), "audio/midi"},
  	{"MP3 audio/MPEG audio", []byte("ID3\x03\x00\x00\x00\x00\x0f"), "audio/mpeg"},
  	{"WAV audio #1", []byte("RIFFb\xb8\x00\x00WAVEfmt \x12\x00\x00\x00\x06"), "audio/wave"},
  	{"WAV audio #2", []byte("RIFF,\x00\x00\x00WAVEfmt \x12\x00\x00\x00\x06"), "audio/wave"},
  	{"AIFF audio #1", []byte("FORM\x00\x00\x00\x00AIFFCOMM\x00\x00\x00\x12\x00\x01\x00\x00\x57\x55\x00\x10\x40\x0d\xf3\x34"), "audio/aiff"},
  
  	{"OGG audio", []byte("OggS\x00\x02\x00\x00\x00\x00\x00\x00\x00\x00\x7e\x46\x00\x00\x00\x00\x00\x00\x1f\xf6\xb4\xfc\x01\x1e\x01\x76\x6f\x72"), "application/ogg"},
  	{"Must not match OGG", []byte("owow\x00"), "application/octet-stream"},
  	{"Must not match OGG", []byte("oooS\x00"), "application/octet-stream"},
  	{"Must not match OGG", []byte("oggS\x00"), "application/octet-stream"},
  
  	// Video types.
  	{"MP4 video", []byte("\x00\x00\x00\x18ftypmp42\x00\x00\x00\x00mp42isom<\x06t\xbfmdat"), "video/mp4"},
  	{"AVI video #1", []byte("RIFF,O\n\x00AVI LISTÀ"), "video/avi"},
  	{"AVI video #2", []byte("RIFF,\n\x00\x00AVI LISTÀ"), "video/avi"},
  }
  
  func TestDetectContentType(t *testing.T) {
  	for _, tt := range sniffTests {
  		ct := DetectContentType(tt.data)
  		if ct != tt.contentType {
  			t.Errorf("%v: DetectContentType = %q, want %q", tt.desc, ct, tt.contentType)
  		}
  	}
  }
  
  func TestServerContentType_h1(t *testing.T) { testServerContentType(t, h1Mode) }
  func TestServerContentType_h2(t *testing.T) { testServerContentType(t, h2Mode) }
  
  func testServerContentType(t *testing.T, h2 bool) {
  	setParallel(t)
  	defer afterTest(t)
  	cst := newClientServerTest(t, h2, HandlerFunc(func(w ResponseWriter, r *Request) {
  		i, _ := strconv.Atoi(r.FormValue("i"))
  		tt := sniffTests[i]
  		n, err := w.Write(tt.data)
  		if n != len(tt.data) || err != nil {
  			log.Fatalf("%v: Write(%q) = %v, %v want %d, nil", tt.desc, tt.data, n, err, len(tt.data))
  		}
  	}))
  	defer cst.close()
  
  	for i, tt := range sniffTests {
  		resp, err := cst.c.Get(cst.ts.URL + "/?i=" + strconv.Itoa(i))
  		if err != nil {
  			t.Errorf("%v: %v", tt.desc, err)
  			continue
  		}
  		if ct := resp.Header.Get("Content-Type"); ct != tt.contentType {
  			t.Errorf("%v: Content-Type = %q, want %q", tt.desc, ct, tt.contentType)
  		}
  		data, err := ioutil.ReadAll(resp.Body)
  		if err != nil {
  			t.Errorf("%v: reading body: %v", tt.desc, err)
  		} else if !bytes.Equal(data, tt.data) {
  			t.Errorf("%v: data is %q, want %q", tt.desc, data, tt.data)
  		}
  		resp.Body.Close()
  	}
  }
  
  // Issue 5953: shouldn't sniff if the handler set a Content-Type header,
  // even if it's the empty string.
  func TestServerIssue5953_h1(t *testing.T) { testServerIssue5953(t, h1Mode) }
  func TestServerIssue5953_h2(t *testing.T) { testServerIssue5953(t, h2Mode) }
  func testServerIssue5953(t *testing.T, h2 bool) {
  	defer afterTest(t)
  	cst := newClientServerTest(t, h2, HandlerFunc(func(w ResponseWriter, r *Request) {
  		w.Header()["Content-Type"] = []string{""}
  		fmt.Fprintf(w, "<html><head></head><body>hi</body></html>")
  	}))
  	defer cst.close()
  
  	resp, err := cst.c.Get(cst.ts.URL)
  	if err != nil {
  		t.Fatal(err)
  	}
  
  	got := resp.Header["Content-Type"]
  	want := []string{""}
  	if !reflect.DeepEqual(got, want) {
  		t.Errorf("Content-Type = %q; want %q", got, want)
  	}
  	resp.Body.Close()
  }
  
  func TestContentTypeWithCopy_h1(t *testing.T) { testContentTypeWithCopy(t, h1Mode) }
  func TestContentTypeWithCopy_h2(t *testing.T) { testContentTypeWithCopy(t, h2Mode) }
  func testContentTypeWithCopy(t *testing.T, h2 bool) {
  	defer afterTest(t)
  
  	const (
  		input    = "\n<html>\n\t<head>\n"
  		expected = "text/html; charset=utf-8"
  	)
  
  	cst := newClientServerTest(t, h2, HandlerFunc(func(w ResponseWriter, r *Request) {
  		// Use io.Copy from a bytes.Buffer to trigger ReadFrom.
  		buf := bytes.NewBuffer([]byte(input))
  		n, err := io.Copy(w, buf)
  		if int(n) != len(input) || err != nil {
  			t.Errorf("io.Copy(w, %q) = %v, %v want %d, nil", input, n, err, len(input))
  		}
  	}))
  	defer cst.close()
  
  	resp, err := cst.c.Get(cst.ts.URL)
  	if err != nil {
  		t.Fatalf("Get: %v", err)
  	}
  	if ct := resp.Header.Get("Content-Type"); ct != expected {
  		t.Errorf("Content-Type = %q, want %q", ct, expected)
  	}
  	data, err := ioutil.ReadAll(resp.Body)
  	if err != nil {
  		t.Errorf("reading body: %v", err)
  	} else if !bytes.Equal(data, []byte(input)) {
  		t.Errorf("data is %q, want %q", data, input)
  	}
  	resp.Body.Close()
  }
  
  func TestSniffWriteSize_h1(t *testing.T) { testSniffWriteSize(t, h1Mode) }
  func TestSniffWriteSize_h2(t *testing.T) { testSniffWriteSize(t, h2Mode) }
  func testSniffWriteSize(t *testing.T, h2 bool) {
  	setParallel(t)
  	defer afterTest(t)
  	cst := newClientServerTest(t, h2, HandlerFunc(func(w ResponseWriter, r *Request) {
  		size, _ := strconv.Atoi(r.FormValue("size"))
  		written, err := io.WriteString(w, strings.Repeat("a", size))
  		if err != nil {
  			t.Errorf("write of %d bytes: %v", size, err)
  			return
  		}
  		if written != size {
  			t.Errorf("write of %d bytes wrote %d bytes", size, written)
  		}
  	}))
  	defer cst.close()
  	for _, size := range []int{0, 1, 200, 600, 999, 1000, 1023, 1024, 512 << 10, 1 << 20} {
  		res, err := cst.c.Get(fmt.Sprintf("%s/?size=%d", cst.ts.URL, size))
  		if err != nil {
  			t.Fatalf("size %d: %v", size, err)
  		}
  		if _, err := io.Copy(ioutil.Discard, res.Body); err != nil {
  			t.Fatalf("size %d: io.Copy of body = %v", size, err)
  		}
  		if err := res.Body.Close(); err != nil {
  			t.Fatalf("size %d: body Close = %v", size, err)
  		}
  	}
  }
  

View as plain text