Chapter 3: Denial of Service vulnerability

[mtlynch]

Chapter 3 accepts arbitrary uploads from remote users, but it doesn't limit the size of the upload.

goblueprints/chapter3/chat/upload.go

Line 12 in aae50b4

userID := req.FormValue("userid")

A malicious user could upload a very large file and exhaust server RAM or fill the disk.

A possible fix is to use http.MaxBytesReader to limit the size of the upload to some reasonable maximum for image files (e.g. 10 MiB).

Calling ParseMultipartForm with an explicit RAM limit also protects the server from exhausting RAM trying to process large uploads:

const maxImageBytes = 10 << 20 // 10 MiB

func uploaderHandler(w http.ResponseWriter, req *http.Request) {
	// Read a maximum of maxImageBytes, plus a little extra room for multipart fields.
	r.Body = http.MaxBytesReader(w, r.Body, maxImageBytes +1024)
	if err := r.ParseMultipartForm(maxImageBytes); err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}
	userID := req.FormValue("userid")
	...