Fix flaky behaviour during S/R.

PiperOrigin-RevId: 280280156
google · Nov 13, 2019 · 6dd4c9e · 6dd4c9e
1 parent 683e879
commit 6dd4c9e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/tcpip/transport/tcp/connect.go b/pkg/tcpip/transport/tcp/connect.go
@@ -1229,7 +1229,9 @@ func (e *endpoint) protocolMainLoop(handshake bool) *tcpip.Error {
 							return err
 						}
 					}
-					if e.state != StateError {
+					if e.state != StateClose && e.state != StateError {
+						// Only block the worker if the endpoint
+						// is not in closed state or error state.
 						close(e.drainDone)
 						<-e.undrain
 					}