Return error when can't get conn to send ACK

Signed-off-by: bill fort <fxbao@hotmail.com>
nknorg · Feb 20, 2023 · ed5f036 · ed5f036
1 parent c86a46b
commit ed5f036
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/errors.go b/errors.go
@@ -12,6 +12,7 @@ var (
 	ErrRecvWindowFull        = NewGenericError("receive window full", false, true)
 	ErrNotHandshake          = NewGenericError("first packet is not handshake packet", false, true)
 	ErrConnClosed            = NewGenericError("connection closed", false, false)
+	ErrConnNotFound          = NewGenericError("connection not found", false, false)
 )
 
 type GenericError struct {

diff --git a/session.go b/session.go
@@ -296,6 +296,8 @@ func (session *Session) ReceiveWith(localClientID, remoteClientID string, buf []
 
 		if conn, ok := session.connections[connKey(localClientID, remoteClientID)]; ok {
 			conn.SendAck(packet.SequenceId)
+		} else {
+			return ErrConnNotFound
 		}
 	}