diff options
Diffstat (limited to 'src/pkg/mime/multipart')
-rw-r--r-- | src/pkg/mime/multipart/example_test.go | 53 | ||||
-rw-r--r-- | src/pkg/mime/multipart/formdata_test.go | 3 | ||||
-rw-r--r-- | src/pkg/mime/multipart/multipart.go | 10 | ||||
-rw-r--r-- | src/pkg/mime/multipart/quotedprintable_test.go | 2 |
4 files changed, 63 insertions, 5 deletions
diff --git a/src/pkg/mime/multipart/example_test.go b/src/pkg/mime/multipart/example_test.go new file mode 100644 index 000000000..26135b785 --- /dev/null +++ b/src/pkg/mime/multipart/example_test.go @@ -0,0 +1,53 @@ +// Copyright 2014 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 multipart_test + +import ( + "fmt" + "io" + "io/ioutil" + "log" + "mime" + "mime/multipart" + "net/mail" + "strings" +) + +func ExampleNewReader() { + msg := &mail.Message{ + Header: map[string][]string{ + "Content-Type": []string{"multipart/mixed; boundary=foo"}, + }, + Body: strings.NewReader( + "--foo\r\nFoo: one\r\n\r\nA section\r\n" + + "--foo\r\nFoo: two\r\n\r\nAnd another\r\n" + + "--foo--\r\n"), + } + mediaType, params, err := mime.ParseMediaType(msg.Header.Get("Content-Type")) + if err != nil { + log.Fatal(err) + } + if strings.HasPrefix(mediaType, "multipart/") { + mr := multipart.NewReader(msg.Body, params["boundary"]) + for { + p, err := mr.NextPart() + if err == io.EOF { + return + } + if err != nil { + log.Fatal(err) + } + slurp, err := ioutil.ReadAll(p) + if err != nil { + log.Fatal(err) + } + fmt.Printf("Part %q: %q\n", p.Header.Get("Foo"), slurp) + } + } + + // Output: + // Part "one": "A section" + // Part "two": "And another" +} diff --git a/src/pkg/mime/multipart/formdata_test.go b/src/pkg/mime/multipart/formdata_test.go index 4bc464931..6e2388baf 100644 --- a/src/pkg/mime/multipart/formdata_test.go +++ b/src/pkg/mime/multipart/formdata_test.go @@ -9,12 +9,13 @@ import ( "io" "os" "regexp" + "strings" "testing" ) func TestReadForm(t *testing.T) { testBody := regexp.MustCompile("\n").ReplaceAllString(message, "\r\n") - b := bytes.NewBufferString(testBody) + b := strings.NewReader(testBody) r := NewReader(b, boundary) f, err := r.ReadForm(25) if err != nil { diff --git a/src/pkg/mime/multipart/multipart.go b/src/pkg/mime/multipart/multipart.go index 2b4f5b433..7382efab9 100644 --- a/src/pkg/mime/multipart/multipart.go +++ b/src/pkg/mime/multipart/multipart.go @@ -81,12 +81,16 @@ func (p *Part) parseContentDisposition() { } } -// NewReader creates a new multipart Reader reading from reader using the +// NewReader creates a new multipart Reader reading from r using the // given MIME boundary. -func NewReader(reader io.Reader, boundary string) *Reader { +// +// The boundary is usually obtained from the "boundary" parameter of +// the message's "Content-Type" header. Use mime.ParseMediaType to +// parse such headers. +func NewReader(r io.Reader, boundary string) *Reader { b := []byte("\r\n--" + boundary + "--") return &Reader{ - bufReader: bufio.NewReader(reader), + bufReader: bufio.NewReader(r), nl: b[:2], nlDashBoundary: b[:len(b)-2], diff --git a/src/pkg/mime/multipart/quotedprintable_test.go b/src/pkg/mime/multipart/quotedprintable_test.go index 8a95f7f03..c4de3eb75 100644 --- a/src/pkg/mime/multipart/quotedprintable_test.go +++ b/src/pkg/mime/multipart/quotedprintable_test.go @@ -131,7 +131,7 @@ func TestQPExhaustive(t *testing.T) { return } if strings.HasSuffix(errStr, "0x0a") || strings.HasSuffix(errStr, "0x0d") { - // bunch of cases; since whitespace at the end of of a line before \n is removed. + // bunch of cases; since whitespace at the end of a line before \n is removed. return } } |