From: Petar Maymounkov <petarm@gmail.com>
Date: Wed, 10 Feb 2010 01:42:51 +0000 (-0800)
Subject: http: protect io.WriteString in Request/Response.Write with error checking,
X-Git-Tag: weekly.2010-02-17~51
X-Git-Url: http://www.git.cypherpunks.su/?a=commitdiff_plain;h=c5287ecb9c67eb731dd0bf7a5ffcaa1d3d23f52e;p=gostls13.git

http: protect io.WriteString in Request/Response.Write with error checking,
since they were causing a silent program exit (too many EPIPE's).

R=rsc
CC=golang-dev
https://golang.org/cl/204062
---

diff --git a/src/pkg/http/request.go b/src/pkg/http/request.go
index 17afc9cc1b..bd8f00d552 100644
--- a/src/pkg/http/request.go
+++ b/src/pkg/http/request.go
@@ -171,7 +171,10 @@ func (req *Request) Write(w io.Writer) os.Error {
 	// from Request, and introduce Request methods along the lines of
 	// Response.{GetHeader,AddHeader} and string constants for "Host",
 	// "User-Agent" and "Referer".
-	writeSortedKeyValue(w, req.Header, reqExcludeHeader)
+	err := writeSortedKeyValue(w, req.Header, reqExcludeHeader)
+	if err != nil {
+		return err
+	}
 
 	io.WriteString(w, "\r\n")
 
diff --git a/src/pkg/http/response.go b/src/pkg/http/response.go
index 30f8934917..b3743a88cb 100644
--- a/src/pkg/http/response.go
+++ b/src/pkg/http/response.go
@@ -459,7 +459,10 @@ func (resp *Response) Write(w io.Writer) os.Error {
 	}
 
 	// Rest of header
-	writeSortedKeyValue(w, resp.Header, respExcludeHeader)
+	err := writeSortedKeyValue(w, resp.Header, respExcludeHeader)
+	if err != nil {
+		return err
+	}
 
 	// End-of-header
 	io.WriteString(w, "\r\n")
@@ -494,7 +497,7 @@ func (resp *Response) Write(w io.Writer) os.Error {
 	return nil
 }
 
-func writeSortedKeyValue(w io.Writer, kvm map[string]string, exclude map[string]int) {
+func writeSortedKeyValue(w io.Writer, kvm map[string]string, exclude map[string]int) os.Error {
 	kva := make([]string, len(kvm))
 	i := 0
 	for k, v := range kvm {
@@ -506,6 +509,9 @@ func writeSortedKeyValue(w io.Writer, kvm map[string]string, exclude map[string]
 	kva = kva[0:i]
 	sort.SortStrings(kva)
 	for _, l := range kva {
-		io.WriteString(w, l)
+		if _, err := io.WriteString(w, l); err != nil {
+			return err
+		}
 	}
+	return nil
 }