read.go: Avoid handling ping after close frame has been sent

Closes nhooyr#298
bhallionOhbibi · Oct 19, 2023 · 69f55f3 · 69f55f3
1 parent c537145
commit 69f55f3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/read.go b/read.go
@@ -286,6 +286,13 @@ func (c *Conn) handleControl(ctx context.Context, h header) (err error) {
 
 	switch h.opcode {
 	case opPing:
+		c.closeMu.Lock()
+		wroteClose := c.wroteClose
+		c.closeMu.Unlock()
+		if wroteClose {
+			// Cannot respond to ping with a pong because we already sent a close frame.
+			return nil
+		}
 		return c.writeControl(ctx, opPong, b)
 	case opPong:
 		c.activePingsMu.Lock()