Unblock handshake ch on close

association.go

@@ -1376,7 +1376,11 @@ func (a *Association) handleCookieEcho(c *chunkCookieEcho) []*packet {
 
 		a.setState(established)
 		// Note: This is a future place where the user could be notified (COMMUNICATION UP)
-		a.handshakeCompletedCh <- nil
+		select {
+		case a.handshakeCompletedCh <- nil:
+		case <-a.closeWriteLoopCh: // check the write side since we are in a read path
+			return nil
+		}
 	}
 
 	p := &packet{
@@ -1405,7 +1409,10 @@ func (a *Association) handleCookieAck() {
 
 	a.setState(established)
 	// Note: This is a future place where the user could be notified (COMMUNICATION UP)
-	a.handshakeCompletedCh <- nil
+	select {
+	case a.handshakeCompletedCh <- nil:
+	case <-a.closeWriteLoopCh: // check the write side since we are in a read path
+	}
 }
 
 // The caller should hold the lock.
@@ -2698,13 +2705,21 @@ func (a *Association) onRetransmissionFailure(id int) {
 
 	if id == timerT1Init {
 		a.log.Errorf("[%s] retransmission failure: T1-init", a.name)
-		a.handshakeCompletedCh <- ErrHandshakeInitAck
+		select {
+		case a.handshakeCompletedCh <- ErrHandshakeInitAck:
+		case <-a.closeWriteLoopCh: // check the read/write sides
+		case <-a.readLoopCloseCh:
+		}
 		return
 	}
 
 	if id == timerT1Cookie {
 		a.log.Errorf("[%s] retransmission failure: T1-cookie", a.name)
-		a.handshakeCompletedCh <- ErrHandshakeCookieEcho
+		select {
+		case a.handshakeCompletedCh <- ErrHandshakeCookieEcho:
+		case <-a.closeWriteLoopCh: // check the read/write sides
+		case <-a.readLoopCloseCh:
+		}
 		return
 	}
 

association_test.go

@@ -3222,6 +3222,10 @@ func TestAssociation_Abort(t *testing.T) {
 
 // TestAssociation_createClientWithContext tests that the client is closed when the context is canceled.
 func TestAssociation_createClientWithContext(t *testing.T) {
+	// Limit runtime in case of deadlocks
+	lim := test.TimeOut(time.Second * 5)
+	defer lim.Stop()
+
 	checkGoroutineLeaks(t)
 
 	udp1, udp2 := createUDPConnPair()