Warn on unhandled v2 protocol messages

anacrolix · Feb 28, 2024 · fd97a99 · fd97a99
1 parent e9355d9
commit fd97a99
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/peer_protocol/protocol.go b/peer_protocol/protocol.go
@@ -41,6 +41,11 @@ const (
 
 	// BEP 10
 	Extended MessageType = 20
+
+	// BEP 52
+	HashRequest = 21
+	Hashes      = 22
+	HashReject  = 23
 )
 
 const (

diff --git a/peerconn.go b/peerconn.go
@@ -862,6 +862,8 @@ func (c *PeerConn) mainReadLoop() (err error) {
 			c.updateRequests("PeerConn.mainReadLoop allowed fast")
 		case pp.Extended:
 			err = c.onReadExtendedMsg(msg.ExtendedID, msg.ExtendedPayload)
+		case pp.HashRequest, pp.Hashes, pp.HashReject:
+			err = log.WithLevel(log.Warning, fmt.Errorf("received unimplemented BitTorrent v2 message: %v", msg.Type))
 		default:
 			err = fmt.Errorf("received unknown message type: %#v", msg.Type)
 		}