http2: fix CloseNotify data race

Fixes golang/go#17868 Change-Id: I7f9d85f404198d68757d7ca5bb8b1e7aafaafa8b Reviewed-on: https://go-review.googlesource.com/33013 Run-TryBot: Brad Fitzpatrick <[email protected]> TryBot-Result: Gobot Gobot <[email protected]> Reviewed-by: Tom Bergan <[email protected]> Reviewed-by: Brad Fitzpatrick <[email protected]>
golang · Nov 9, 2016 · 9ef2211 · 9ef2211
1 parent 40d3034
commit 9ef2211
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/http2/server.go b/http2/server.go
@@ -2276,8 +2276,9 @@ func (w *responseWriter) CloseNotify() <-chan bool {
 	if ch == nil {
 		ch = make(chan bool, 1)
 		rws.closeNotifierCh = ch
+		cw := rws.stream.cw
 		go func() {
-			rws.stream.cw.Wait() // wait for close
+			cw.Wait() // wait for close
 			ch <- true
 		}()
 	}