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 2086063f91102a5fc21997167de268e3ea76bd07..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
@@ -42,6 +47,6 @@ const (
        HandshakeExtendedID = 0
 
        RequestMetadataExtensionMsgType ExtendedMetadataRequestMsgType = 0
-       DataMetadataExtensionMsgType                                   = 1
-       RejectMetadataExtensionMsgType                                 = 2
+       DataMetadataExtensionMsgType    ExtendedMetadataRequestMsgType = 1
+       RejectMetadataExtensionMsgType  ExtendedMetadataRequestMsgType = 2
 )