fix: Segmentation Error

Harry-kp · Aug 19, 2024 · 90f29ea · 90f29ea
1 parent 92a0bda
commit 90f29ea
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/message/message.go b/message/message.go
@@ -92,7 +92,7 @@ func Read(r io.Reader) (*Message, error) {
 	}
 	length := binary.BigEndian.Uint32(lengthBuf)
 
-	// keet the connection alive
+	// keep the connection alive
 	if length == 0 {
 		return nil, nil
 	}

diff --git a/p2p/p2p.go b/p2p/p2p.go
@@ -54,7 +54,10 @@ func (state *pieceProgress) readMessage() error {
 	if err != nil {
 		return err
 	}
-
+	// keep alive connection
+	if msg == nil {
+		return nil
+	}
 	switch msg.ID {
 	case message.MsgUnchoke:
 		state.client.Choked = false