From: Matt Joiner <anacrolix@gmail.com>
Date: Tue, 27 Feb 2024 12:31:43 +0000 (+1100)
Subject: Warn on unhandled v2 protocol messages
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=1fd47d6748161c5c915c24fd7528e9d4add24319;p=btrtrc.git

Warn on unhandled v2 protocol messages
---

diff --git a/peer_protocol/protocol.go b/peer_protocol/protocol.go
index bfeb6a04..2f92ab32 100644
--- 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
index a1e98b12..ded9b483 100644
--- 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)
 		}