Prevent all writes after close

Closes #213
coder · Apr 14, 2020 · d34c89a · d34c89a
1 parent c4d4650
commit d34c89a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/write.go b/write.go
@@ -246,7 +246,13 @@ func (c *Conn) writeFrame(ctx context.Context, fin bool, flate bool, opcode opco
 	if err != nil {
 		return 0, err
 	}
-	defer c.writeFrameMu.unlock()
+	defer func() {
+		// We leave it locked when writing the close frame to avoid
+		// any other goroutine writing any other frame.
+		if opcode != opClose {
+			c.writeFrameMu.unlock()
+		}
+	}()
 
 	select {
 	case <-c.closed: