Ensure context isn't nil in NoiseSocket.consumeFrames

Fixes tulir#236
Kryvchun · Nov 26, 2022 · dc4f3fd · dc4f3fd
1 parent 4810a44
commit dc4f3fd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/socket/noisesocket.go b/socket/noisesocket.go
@@ -47,6 +47,10 @@ func newNoiseSocket(fs *FrameSocket, writeKey, readKey cipher.AEAD, frameHandler
 }
 
 func (ns *NoiseSocket) consumeFrames(ctx context.Context, frames <-chan []byte) {
+	if ctx == nil {
+		// ctx being nil implies the connection already closed somehow
+		return
+	}
 	ctxDone := ctx.Done()
 	for {
 		select {