diff options
author | Russ Cox <rsc@golang.org> | 2011-05-03 23:53:31 -0400 |
---|---|---|
committer | Russ Cox <rsc@golang.org> | 2011-05-03 23:53:31 -0400 |
commit | 0bd41f1e48205160e5c084287b9592bba2408730 (patch) | |
tree | 72b5e62dde854e62b700c39fcc7e8e9990481956 | |
parent | ca2cb382ba89d1a1534f71ec2218a23659e3d491 (diff) | |
download | go-0bd41f1e48205160e5c084287b9592bba2408730.tar.gz go-0bd41f1e48205160e5c084287b9592bba2408730.zip |
[release-branch.r57] http: fix FormFile nil pointer dereference on missing multipart form
««« CL 4463042 / 6294ca33c042
http: fix FormFile nil pointer dereference on missing multipart form
R=rsc
CC=golang-dev
https://golang.org/cl/4463042
»»»
R=adg
CC=golang-dev
https://golang.org/cl/4467042
-rw-r--r-- | src/pkg/http/request.go | 8 | ||||
-rw-r--r-- | src/pkg/http/request_test.go | 28 |
2 files changed, 33 insertions, 3 deletions
diff --git a/src/pkg/http/request.go b/src/pkg/http/request.go index b8e9a21423..4852ca3e1f 100644 --- a/src/pkg/http/request.go +++ b/src/pkg/http/request.go @@ -715,9 +715,11 @@ func (r *Request) FormFile(key string) (multipart.File, *multipart.FileHeader, o return nil, nil, err } } - if fhs := r.MultipartForm.File[key]; len(fhs) > 0 { - f, err := fhs[0].Open() - return f, fhs[0], err + if r.MultipartForm != nil && r.MultipartForm.File != nil { + if fhs := r.MultipartForm.File[key]; len(fhs) > 0 { + f, err := fhs[0].Open() + return f, fhs[0], err + } } return nil, nil, ErrMissingFile } diff --git a/src/pkg/http/request_test.go b/src/pkg/http/request_test.go index f982471d8d..f79d3a2424 100644 --- a/src/pkg/http/request_test.go +++ b/src/pkg/http/request_test.go @@ -200,6 +200,29 @@ func TestMultipartRequestAuto(t *testing.T) { validateTestMultipartContents(t, req, true) } +func TestEmptyMultipartRequest(t *testing.T) { + // Test that FormValue and FormFile automatically invoke + // ParseMultipartForm and return the right values. + req, err := NewRequest("GET", "/", nil) + if err != nil { + t.Errorf("NewRequest err = %q", err) + } + testMissingFile(t, req) +} + +func testMissingFile(t *testing.T, req *Request) { + f, fh, err := req.FormFile("missing") + if f != nil { + t.Errorf("FormFile file = %q, want nil", f, nil) + } + if fh != nil { + t.Errorf("FormFile file header = %q, want nil", fh, nil) + } + if err != ErrMissingFile { + t.Errorf("FormFile err = %q, want nil", err, ErrMissingFile) + } +} + func newTestMultipartRequest(t *testing.T) *Request { b := bytes.NewBufferString(strings.Replace(message, "\n", "\r\n", -1)) req, err := NewRequest("POST", "/", b) @@ -218,6 +241,9 @@ func validateTestMultipartContents(t *testing.T, req *Request, allMem bool) { if g, e := req.FormValue("texta"), textaValue; g != e { t.Errorf("texta value = %q, want %q", g, e) } + if g := req.FormValue("missing"); g != "" { + t.Errorf("missing value = %q, want empty string", g) + } assertMem := func(n string, fd multipart.File) { if _, ok := fd.(*os.File); ok { @@ -234,6 +260,8 @@ func validateTestMultipartContents(t *testing.T, req *Request, allMem bool) { t.Errorf("fileb has unexpected underlying type %T", fd) } } + + testMissingFile(t, req) } func testMultipartFile(t *testing.T, req *Request, key, expectFilename, expectContent string) multipart.File { |