diff --git a/multiaddr/src/protocol.rs b/multiaddr/src/protocol.rs index 7c2e5962..fb1e98e3 100644 --- a/multiaddr/src/protocol.rs +++ b/multiaddr/src/protocol.rs @@ -20,7 +20,7 @@ const WS: u32 = 0x01dd; const WSS: u32 = 0x01de; const MEMORY: u32 = 0x0309; -const SHA256_CODE: u16 = 0x12; +const SHA256_CODE: u64 = 0x12; const SHA256_SIZE: u8 = 32; /// `Protocol` describes all possible multiaddress protocols. @@ -276,7 +276,7 @@ impl<'a> From for Protocol<'a> { } fn check_p2p(data: &[u8]) -> Result<(), Error> { - let (code, bytes) = unsigned_varint::decode::u16(data)?; + let (code, bytes) = unsigned_varint::decode::u64(data)?; if code != SHA256_CODE { return Err(Error::UnknownHash); diff --git a/secio/src/peer_id.rs b/secio/src/peer_id.rs index 1d26d394..8f450ff6 100644 --- a/secio/src/peer_id.rs +++ b/secio/src/peer_id.rs @@ -6,7 +6,7 @@ use unsigned_varint::{decode, encode}; use crate::handshake::handshake_struct::PublicKey; -const SHA256_CODE: u16 = 0x12; +const SHA256_CODE: u64 = 0x12; const SHA256_SIZE: u8 = 32; /// Identifier of a peer of the network @@ -31,7 +31,7 @@ impl PeerId { return Err(Error::Empty); } - let (code, bytes) = decode::u16(&data).map_err(|_| Error::InvalidData)?; + let (code, bytes) = decode::u64(&data).map_err(|_| Error::InvalidData)?; if code != SHA256_CODE { return Err(Error::NotSupportHashCode); @@ -57,8 +57,8 @@ impl PeerId { /// Return `PeerId` which used hashed seed as inner. fn from_seed(seed: &[u8]) -> Self { - let mut buf = encode::u16_buffer(); - let code = encode::u16(SHA256_CODE, &mut buf); + let mut buf = encode::u64_buffer(); + let code = encode::u64(SHA256_CODE, &mut buf); let header_len = code.len() + 1;