Merge pull request #3535 from nspcc-dev/fix/possible-WS-deadlock-on-c…

…onnection-loss
nspcc-dev · Jul 29, 2024 · c2a1c4a · c2a1c4a
2 parents 8d4e8b6 + 9fadfef
commit c2a1c4a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pkg/rpcclient/wsclient.go b/pkg/rpcclient/wsclient.go
@@ -601,7 +601,13 @@ readloop:
 				connCloseErr = fmt.Errorf("unknown response channel for response %d", id)
 				break readloop // Unknown response (unexpected response ID).
 			}
-			ch <- &rr.Response
+			select {
+			case <-c.writerDone:
+				break readloop
+			case <-c.shutdown:
+				break readloop
+			case ch <- &rr.Response:
+			}
 		} else {
 			// Malformed response, neither valid request, nor valid response.
 			connCloseErr = fmt.Errorf("malformed response")