lneto/http/httpraw/multipart_test.go

package httpraw

import (
	"io"
	"strconv"
	"strings"
	"testing"
)

// A two part body: a text field and a PNG upload whose bytes contain CRLFs and
// even the boundary text, which must not desync the parser.
const (
	multiBoundary = "----abc123"
	multiBody     = "------abc123\r\n" +
		"Content-Disposition: form-data; name=\"caption\"\r\n" +
		"\r\n" +
		"hi there\r\n" +
		"------abc123\r\n" +
		"Content-Disposition: form-data; name=\"photo\"; filename=\"beach.png\"\r\n" +
		"Content-Type: image/png\r\n" +
		"\r\n" +
		"\x89PNG\r\n--not-the-boundary\r\n\x00\xff\r\n" +
		"------abc123--\r\n"
)

func TestMultipartBoundary(t *testing.T) {
	var mp Multipart
	for _, test := range []struct {
		contentType string
		want        string
	}{
		{contentType: "multipart/form-data; boundary=abc123", want: "abc123"},
		{contentType: "multipart/form-data; boundary=\"a b\"", want: "a b"},
		{contentType: "multipart/form-data; charset=utf-8; boundary=xyz", want: "xyz"},
		{contentType: "multipart/form-data; BOUNDARY=xyz", want: "xyz"}, // Keys are case insensitive.
		{contentType: "multipart/form-data", want: ""},                  // Absent.
		{contentType: "application/x-www-form-urlencoded", want: ""},
	} {
		err := mp.SetContentType([]byte(test.contentType))
		if err != nil {
			t.Skip("asdasd")
		}
		got := string(mp.Boundary)
		if got != test.want {
			t.Errorf("%q: want %q, got %q", test.contentType, test.want, got)
		}
	}
}

func TestContentParam(t *testing.T) {
	for _, test := range []struct {
		value string
		key   string
		want  string
	}{
		{value: "text/plain; charset=utf-8", key: "charset", want: "utf-8"},
		{value: "text/plain;charset=utf-8", key: "charset", want: "utf-8"}, // No space.
		{value: "text/plain; charset=\"utf-8\"", key: "charset", want: "utf-8"},
		{value: "form-data; name=\"photo\"; filename=\"a;b.png\"", key: "filename", want: "a;b.png"},
		{value: "form-data; name=\"photo\"", key: "nope", want: ""},
		{value: "form-data; names=x; name=y", key: "name", want: "y"}, // Prefix must not match.
		{value: "text/plain", key: "charset", want: ""},
	} {
		got := ContentParam([]byte(test.value), test.key)
		if string(got) != test.want {
			t.Errorf("%q key %q: want %q, got %q", test.value, test.key, test.want, got)
		}
	}
}

func TestNextPartHeader(t *testing.T) {
	m := Multipart{Boundary: []byte(multiBoundary)}
	var hdr MultipartHeader
	rest, err := m.NextHeader(&hdr, []byte(multiBody))
	if err != nil {
		t.Fatal(err)
	}
	const wantHdr = "Content-Disposition: form-data; name=\"caption\"\r\n"
	if string(hdr.Part) != wantHdr {
		t.Errorf("want header %q, got %q", wantHdr, hdr.Part)
	}
	if string(hdr.Name) != "caption" {
		t.Errorf("want name %q, got %q", "caption", hdr.Name)
	}
	if hdr.Filename != nil {
		t.Errorf("want nil filename for a non file part, got %q", hdr.Filename)
	}
	if !strings.HasPrefix(string(rest), "hi there\r\n") {
		t.Errorf("want rest at part body, got %q", rest)
	}
}

// Incomplete data must ask for more, never guess.
func TestNextPartHeaderNeedMore(t *testing.T) {
	m := Multipart{Boundary: []byte(multiBoundary)}
	for _, data := range []string{
		"",
		"------abc",        // Delimiter cut short.
		"------abc123\r\n", // No header block yet.
		"------abc123\r\nContent-Disposition: form-", // Header block unterminated.
	} {
		var hdr MultipartHeader
		if _, err := m.NextHeader(&hdr, []byte(data)); err != ErrNeedMoreData {
			t.Errorf("%q: want ErrNeedMoreData, got %v", data, err)
		}
	}
}

// The closing delimiter ends iteration.
func TestNextPartHeaderEnd(t *testing.T) {
	m := Multipart{Boundary: []byte(multiBoundary)}
	var hdr MultipartHeader
	if _, err := m.NextHeader(&hdr, []byte("------abc123--\r\n")); err != io.EOF {
		t.Errorf("want io.EOF, got %v", err)
	}
}

func TestNextPartBody(t *testing.T) {
	m := Multipart{Boundary: []byte(multiBoundary)}
	var hdr MultipartHeader
	rest, err := m.NextHeader(&hdr, []byte(multiBody))
	if err != nil {
		t.Fatal(err)
	}
	body, rest, done := m.NextBody(rest)
	if !done {
		t.Fatal("want the part to end within the buffer")
	}
	if string(body) != "hi there" {
		t.Errorf("want body %q, got %q", "hi there", body)
	}
	if !strings.HasPrefix(string(rest), "------abc123\r\n") {
		t.Errorf("want rest at next delimiter, got %q", rest)
	}
}

// A part whose bytes contain CRLFs and boundary-like text must survive intact.
func TestNextPartBodyBinary(t *testing.T) {
	m := Multipart{Boundary: []byte(multiBoundary)}
	data := []byte(multiBody)
	var hdr MultipartHeader
	rest, err := m.NextHeader(&hdr, data) // caption part.
	if err != nil {
		t.Fatal(err)
	}
	_, rest, _ = m.NextBody(rest)
	rest, err = m.NextHeader(&hdr, rest) // photo part.
	if err != nil {
		t.Fatal(err)
	}
	body, rest, done := m.NextBody(rest)
	if !done {
		t.Fatal("want the part to end within the buffer")
	}
	const want = "\x89PNG\r\n--not-the-boundary\r\n\x00\xff"
	if string(body) != want {
		t.Errorf("want body %q, got %q", want, body)
	}
	if _, err = m.NextHeader(&hdr, rest); err != io.EOF {
		t.Errorf("want io.EOF after last part, got %v", err)
	}
}

// A delimiter split across two reads must not be mistaken for part data: the
// tail is held back until proven not to be a delimiter.
func TestNextPartBodySplitDelimiter(t *testing.T) {
	m := Multipart{Boundary: []byte(multiBoundary)}
	const part = "hi there"
	full := part + "\r\n------abc123\r\n"
	for split := 1; split < len(full); split++ {
		body, rest, done := m.NextBody([]byte(full[:split]))
		if done {
			continue // Whole delimiter already present, nothing to prove.
		}
		if len(body) > len(part) {
			t.Fatalf("split %d: emitted %q, past the end of the part", split, body)
		}
		if string(body)+string(rest) != full[:split] {
			t.Fatalf("split %d: body+rest %q%q does not reconstruct input", split, body, rest)
		}
	}
}

// A file part carries both parameters, and the raw block stays available.
func TestNextHeaderFilePart(t *testing.T) {
	m := Multipart{Boundary: []byte(multiBoundary)}
	var hdr MultipartHeader
	rest, err := m.NextHeader(&hdr, []byte(multiBody)) // caption part.
	if err != nil {
		t.Fatal(err)
	}
	_, rest, _ = m.NextBody(rest)
	if _, err = m.NextHeader(&hdr, rest); err != nil { // photo part.
		t.Fatal(err)
	}
	if got := string(hdr.Name); got != "photo" {
		t.Errorf("want name %q, got %q", "photo", got)
	}
	if got := string(hdr.Filename); got != "beach.png" {
		t.Errorf("want filename %q, got %q", "beach.png", got)
	}
	if !strings.Contains(string(hdr.Part), "Content-Type: image/png") {
		t.Errorf("want the raw block to hold every field, got %q", hdr.Part)
	}
}

// A failed call must not leave the previous part's fields behind.
func TestNextHeaderZeroesOnError(t *testing.T) {
	m := Multipart{Boundary: []byte(multiBoundary)}
	var hdr MultipartHeader
	if _, err := m.NextHeader(&hdr, []byte(multiBody)); err != nil {
		t.Fatal(err)
	}
	if _, err := m.NextHeader(&hdr, []byte("------abc123--\r\n")); err != io.EOF {
		t.Fatalf("want io.EOF, got %v", err)
	}
	if hdr.Part != nil || hdr.Name != nil || hdr.Filename != nil {
		t.Errorf("want zeroed header on error, got %+v", hdr)
	}
}

// The whole loop, as a caller writes it.
func TestMultipartLoop(t *testing.T) {
	m := Multipart{Boundary: []byte(multiBoundary)}
	rest := []byte(multiBody)
	var got []string
	var hdr MultipartHeader
	for {
		next, err := m.NextHeader(&hdr, rest)
		if err == io.EOF {
			break
		} else if err != nil {
			t.Fatal(err)
		}
		name := string(hdr.Name)
		total := 0
		rest = next
		for {
			body, next, done := m.NextBody(rest)
			total += len(body)
			rest = next
			if done {
				break
			}
			t.Fatal("part must complete within the buffer")
		}
		got = append(got, name+":"+strconv.Itoa(total))
	}
	want := "caption:8|photo:28"
	if strings.Join(got, "|") != want {
		t.Errorf("want %q, got %q", want, strings.Join(got, "|"))
	}
}