Ignore dial rate limits for holepunch connects

[btrtrc.git] / peer_protocol / protocol.go

diff --git a/peer_protocol/protocol.go b/peer_protocol/protocol.go
index 506e020eeba8b1cc0e738a269b1fdac47e6302d0..bfeb6a0448873da8b3fdc96dd133520da0c8d360 100644 (file)
--- a/peer_protocol/protocol.go
+++ b/peer_protocol/protocol.go
@@ -12,6 +12,11 @@ func (mt MessageType) FastExtension() bool {
        return mt >= Suggest && mt <= AllowedFast
 }
 
+func (mt *MessageType) UnmarshalBinary(b []byte) error {
+       *mt = MessageType(b[0])
+       return nil
+}
+
 const (
        // BEP 3
        Choke         MessageType = 0
@@ -41,7 +46,7 @@ const (
 const (
        HandshakeExtendedID = 0
 
-       RequestMetadataExtensionMsgType = 0
-       DataMetadataExtensionMsgType    = 1
-       RejectMetadataExtensionMsgType  = 2
+       RequestMetadataExtensionMsgType ExtendedMetadataRequestMsgType = 0
+       DataMetadataExtensionMsgType    ExtendedMetadataRequestMsgType = 1
+       RejectMetadataExtensionMsgType  ExtendedMetadataRequestMsgType = 2
 )