mercurial-scm/hg: rust/hg-core/src/revlog/node.rs annotate

annotate rust/hg-core/src/revlog/node.rs @ 46427:6380efb82191

rust: replace Node::encode_hex with std::fmt::LowerHex This avoids allocating intermediate strings. Differential Revision: https://phab.mercurial-scm.org/D9860

author	Simon Sapin <simon.sapin@octobus.net>
date	Mon, 25 Jan 2021 12:00:23 +0100
parents	cfb6c10c08c2
children	5893706af3de

rev	line source
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	1 // Copyright 2019-2020 Georges Racinet <georges.racinet@octobus.net>
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	2 //
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	3 // This software may be used and distributed according to the terms of the
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	4 // GNU General Public License version 2 or any later version.
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	5
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	6 //! Definitions and utilities for Revision nodes
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	7 //!
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	8 //! In Mercurial code base, it is customary to call "a node" the binary SHA
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	9 //! of a revision.
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	10
46391 cfb6c10c08c2 rust: replace an unsafe use of transmute with a safe use of bytes-cast Simon Sapin <simon.sapin@octobus.net> parents: 46033 diff changeset	11 use bytes_cast::BytesCast;
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	12 use hex::{self, FromHex, FromHexError};
46391 cfb6c10c08c2 rust: replace an unsafe use of transmute with a safe use of bytes-cast Simon Sapin <simon.sapin@octobus.net> parents: 46033 diff changeset	13 use std::convert::TryFrom;
46427 6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	14 use std::fmt;
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	15
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	16 /// The length in bytes of a `Node`
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	17 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	18 /// This constant is meant to ease refactors of this module, and
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	19 /// are private so that calling code does not expect all nodes have
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	20 /// the same size, should we support several formats concurrently in
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	21 /// the future.
45531 b0d6309ff50c hg-core: check data integrity in `Revlog` Antoine Cezar <antoine.cezar@octobus.net> parents: 44973 diff changeset	22 pub const NODE_BYTES_LENGTH: usize = 20;
b0d6309ff50c hg-core: check data integrity in `Revlog` Antoine Cezar <antoine.cezar@octobus.net> parents: 44973 diff changeset	23
b0d6309ff50c hg-core: check data integrity in `Revlog` Antoine Cezar <antoine.cezar@octobus.net> parents: 44973 diff changeset	24 /// Id of the null node.
b0d6309ff50c hg-core: check data integrity in `Revlog` Antoine Cezar <antoine.cezar@octobus.net> parents: 44973 diff changeset	25 ///
b0d6309ff50c hg-core: check data integrity in `Revlog` Antoine Cezar <antoine.cezar@octobus.net> parents: 44973 diff changeset	26 /// Used to indicate the absence of node.
b0d6309ff50c hg-core: check data integrity in `Revlog` Antoine Cezar <antoine.cezar@octobus.net> parents: 44973 diff changeset	27 pub const NULL_NODE_ID: [u8; NODE_BYTES_LENGTH] = [0u8; NODE_BYTES_LENGTH];
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	28
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	29 /// The length in bytes of a `Node`
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	30 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	31 /// see also `NODES_BYTES_LENGTH` about it being private.
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	32 const NODE_NYBBLES_LENGTH: usize = 2 * NODE_BYTES_LENGTH;
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	33
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	34 /// Private alias for readability and to ease future change
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	35 type NodeData = [u8; NODE_BYTES_LENGTH];
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	36
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	37 /// Binary revision SHA
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	38 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	39 /// ## Future changes of hash size
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	40 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	41 /// To accomodate future changes of hash size, Rust callers
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	42 /// should use the conversion methods at the boundaries (FFI, actual
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	43 /// computation of hashes and I/O) only, and only if required.
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	44 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	45 /// All other callers outside of unit tests should just handle `Node` values
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	46 /// and never make any assumption on the actual length, using [`nybbles_len`]
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	47 /// if they need a loop boundary.
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	48 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	49 /// All methods that create a `Node` either take a type that enforces
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	50 /// the size or fail immediately at runtime with [`ExactLengthRequired`].
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	51 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	52 /// [`nybbles_len`]: #method.nybbles_len
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	53 /// [`ExactLengthRequired`]: struct.NodeError#variant.ExactLengthRequired
46391 cfb6c10c08c2 rust: replace an unsafe use of transmute with a safe use of bytes-cast Simon Sapin <simon.sapin@octobus.net> parents: 46033 diff changeset	54 #[derive(Clone, Debug, PartialEq, BytesCast)]
44502 166349510398 revlog: using two new functions in C capsule from Rust code Georges Racinet <georges.racinet@octobus.net> parents: 44388 diff changeset	55 #[repr(transparent)]
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	56 pub struct Node {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	57 data: NodeData,
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	58 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	59
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	60 /// The node value for NULL_REVISION
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	61 pub const NULL_NODE: Node = Node {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	62 data: [0; NODE_BYTES_LENGTH],
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	63 };
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	64
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	65 impl From<NodeData> for Node {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	66 fn from(data: NodeData) -> Node {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	67 Node { data }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	68 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	69 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	70
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	71 /// Return an error if the slice has an unexpected length
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	72 impl<'a> TryFrom<&'a [u8]> for &'a Node {
46391 cfb6c10c08c2 rust: replace an unsafe use of transmute with a safe use of bytes-cast Simon Sapin <simon.sapin@octobus.net> parents: 46033 diff changeset	73 type Error = ();
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	74
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	75 #[inline]
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	76 fn try_from(bytes: &'a [u8]) -> Result<&'a Node, Self::Error> {
46391 cfb6c10c08c2 rust: replace an unsafe use of transmute with a safe use of bytes-cast Simon Sapin <simon.sapin@octobus.net> parents: 46033 diff changeset	77 match Node::from_bytes(bytes) {
cfb6c10c08c2 rust: replace an unsafe use of transmute with a safe use of bytes-cast Simon Sapin <simon.sapin@octobus.net> parents: 46033 diff changeset	78 Ok((node, rest)) if rest.is_empty() => Ok(node),
cfb6c10c08c2 rust: replace an unsafe use of transmute with a safe use of bytes-cast Simon Sapin <simon.sapin@octobus.net> parents: 46033 diff changeset	79 _ => Err(()),
cfb6c10c08c2 rust: replace an unsafe use of transmute with a safe use of bytes-cast Simon Sapin <simon.sapin@octobus.net> parents: 46033 diff changeset	80 }
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	81 }
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	82 }
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	83
46427 6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	84 impl fmt::LowerHex for Node {
6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	85 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	86 for &byte in &self.data {
6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	87 write!(f, "{:02x}", byte)?
6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	88 }
6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	89 Ok(())
6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	90 }
6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	91 }
6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	92
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	93 #[derive(Debug, PartialEq)]
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	94 pub enum NodeError {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	95 ExactLengthRequired(usize, String),
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	96 PrefixTooLong(String),
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	97 HexError(FromHexError, String),
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	98 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	99
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	100 /// Low level utility function, also for prefixes
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	101 fn get_nybble(s: &[u8], i: usize) -> u8 {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	102 if i % 2 == 0 {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	103 s[i / 2] >> 4
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	104 } else {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	105 s[i / 2] & 0x0f
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	106 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	107 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	108
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	109 impl Node {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	110 /// Retrieve the `i`th half-byte of the binary data.
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	111 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	112 /// This is also the `i`th hexadecimal digit in numeric form,
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	113 /// also called a [nybble](https://en.wikipedia.org/wiki/Nibble).
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	114 pub fn get_nybble(&self, i: usize) -> u8 {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	115 get_nybble(&self.data, i)
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	116 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	117
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	118 /// Length of the data, in nybbles
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	119 pub fn nybbles_len(&self) -> usize {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	120 // public exposure as an instance method only, so that we can
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	121 // easily support several sizes of hashes if needed in the future.
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	122 NODE_NYBBLES_LENGTH
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	123 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	124
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	125 /// Convert from hexadecimal string representation
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	126 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	127 /// Exact length is required.
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	128 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	129 /// To be used in FFI and I/O only, in order to facilitate future
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	130 /// changes of hash format.
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	131 pub fn from_hex(hex: impl AsRef<[u8]>) -> Result<Node, NodeError> {
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	132 Ok(NodeData::from_hex(hex.as_ref())
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	133 .map_err(\|e\| NodeError::from((e, hex)))?
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	134 .into())
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	135 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	136
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	137 /// Provide access to binary data
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	138 ///
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	139 /// This is needed by FFI layers, for instance to return expected
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	140 /// binary values to Python.
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	141 pub fn as_bytes(&self) -> &[u8] {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	142 &self.data
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	143 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	144 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	145
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	146 impl<T: AsRef<[u8]>> From<(FromHexError, T)> for NodeError {
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	147 fn from(err_offender: (FromHexError, T)) -> Self {
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	148 let (err, offender) = err_offender;
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	149 let offender = String::from_utf8_lossy(offender.as_ref()).into_owned();
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	150 match err {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	151 FromHexError::InvalidStringLength => {
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	152 NodeError::ExactLengthRequired(NODE_NYBBLES_LENGTH, offender)
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	153 }
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	154 _ => NodeError::HexError(err, offender),
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	155 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	156 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	157 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	158
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	159 /// The beginning of a binary revision SHA.
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	160 ///
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	161 /// Since it can potentially come from an hexadecimal representation with
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	162 /// odd length, it needs to carry around whether the last 4 bits are relevant
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	163 /// or not.
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	164 #[derive(Debug, PartialEq)]
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	165 pub struct NodePrefix {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	166 buf: Vec<u8>,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	167 is_odd: bool,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	168 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	169
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	170 impl NodePrefix {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	171 /// Convert from hexadecimal string representation
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	172 ///
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	173 /// Similarly to `hex::decode`, can be used with Unicode string types
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	174 /// (`String`, `&str`) as well as bytes.
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	175 ///
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	176 /// To be used in FFI and I/O only, in order to facilitate future
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	177 /// changes of hash format.
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	178 pub fn from_hex(hex: impl AsRef<[u8]>) -> Result<Self, NodeError> {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	179 let hex = hex.as_ref();
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	180 let len = hex.len();
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	181 if len > NODE_NYBBLES_LENGTH {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	182 return Err(NodeError::PrefixTooLong(
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	183 String::from_utf8_lossy(hex).to_owned().to_string(),
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	184 ));
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	185 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	186
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	187 let is_odd = len % 2 == 1;
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	188 let even_part = if is_odd { &hex[..len - 1] } else { hex };
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	189 let mut buf: Vec<u8> =
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	190 Vec::from_hex(&even_part).map_err(\|e\| (e, hex))?;
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	191
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	192 if is_odd {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	193 let latest_char = char::from(hex[len - 1]);
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	194 let latest_nybble = latest_char.to_digit(16).ok_or_else(\|\| {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	195 (
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	196 FromHexError::InvalidHexCharacter {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	197 c: latest_char,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	198 index: len - 1,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	199 },
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	200 hex,
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	201 )
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	202 })? as u8;
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	203 buf.push(latest_nybble << 4);
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	204 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	205 Ok(NodePrefix { buf, is_odd })
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	206 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	207
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	208 pub fn borrow(&self) -> NodePrefixRef {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	209 NodePrefixRef {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	210 buf: &self.buf,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	211 is_odd: self.is_odd,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	212 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	213 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	214 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	215
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	216 #[derive(Clone, Debug, PartialEq)]
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	217 pub struct NodePrefixRef<'a> {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	218 buf: &'a [u8],
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	219 is_odd: bool,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	220 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	221
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	222 impl<'a> NodePrefixRef<'a> {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	223 pub fn len(&self) -> usize {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	224 if self.is_odd {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	225 self.buf.len() * 2 - 1
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	226 } else {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	227 self.buf.len() * 2
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	228 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	229 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	230
44973 26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	231 pub fn is_empty(&self) -> bool {
26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	232 self.len() == 0
26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	233 }
26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	234
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	235 pub fn is_prefix_of(&self, node: &Node) -> bool {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	236 if self.is_odd {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	237 let buf = self.buf;
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	238 let last_pos = buf.len() - 1;
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	239 node.data.starts_with(buf.split_at(last_pos).0)
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	240 && node.data[last_pos] >> 4 == buf[last_pos] >> 4
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	241 } else {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	242 node.data.starts_with(self.buf)
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	243 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	244 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	245
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	246 /// Retrieve the `i`th half-byte from the prefix.
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	247 ///
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	248 /// This is also the `i`th hexadecimal digit in numeric form,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	249 /// also called a [nybble](https://en.wikipedia.org/wiki/Nibble).
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	250 pub fn get_nybble(&self, i: usize) -> u8 {
44187 be52b7372ec2 rust-node: avoid meaningless read at the end of odd prefix Georges Racinet <georges.racinet@octobus.net> parents: 44183 diff changeset	251 assert!(i < self.len());
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	252 get_nybble(self.buf, i)
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	253 }
44388 5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	254
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	255 /// Return the index first nybble that's different from `node`
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	256 ///
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	257 /// If the return value is `None` that means that `self` is
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	258 /// a prefix of `node`, but the current method is a bit slower
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	259 /// than `is_prefix_of`.
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	260 ///
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	261 /// Returned index is as in `get_nybble`, i.e., starting at 0.
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	262 pub fn first_different_nybble(&self, node: &Node) -> Option<usize> {
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	263 let buf = self.buf;
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	264 let until = if self.is_odd {
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	265 buf.len() - 1
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	266 } else {
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	267 buf.len()
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	268 };
44973 26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	269 for (i, item) in buf.iter().enumerate().take(until) {
26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	270 if *item != node.data[i] {
26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	271 return if *item & 0xf0 == node.data[i] & 0xf0 {
26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	272 Some(2 * i + 1)
44388 5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	273 } else {
44973 26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	274 Some(2 * i)
26114bd6ec60 rust: do a clippy pass Rapha?l Gom?s <rgomes@octobus.net> parents: 44502 diff changeset	275 };
44388 5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	276 }
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	277 }
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	278 if self.is_odd && buf[until] & 0xf0 != node.data[until] & 0xf0 {
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	279 Some(until * 2)
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	280 } else {
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	281 None
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	282 }
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	283 }
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	284 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	285
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	286 /// A shortcut for full `Node` references
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	287 impl<'a> From<&'a Node> for NodePrefixRef<'a> {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	288 fn from(node: &'a Node) -> Self {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	289 NodePrefixRef {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	290 buf: &node.data,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	291 is_odd: false,
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	292 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	293 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	294 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	295
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	296 impl PartialEq<Node> for NodePrefixRef<'_> {
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	297 fn eq(&self, other: &Node) -> bool {
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	298 !self.is_odd && self.buf == other.data
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	299 }
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	300 }
88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	301
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	302 #[cfg(test)]
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	303 mod tests {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	304 use super::*;
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	305
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	306 fn sample_node() -> Node {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	307 let mut data = [0; NODE_BYTES_LENGTH];
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	308 data.copy_from_slice(&[
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	309 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0xfe, 0xdc, 0xba,
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	310 0x98, 0x76, 0x54, 0x32, 0x10, 0xde, 0xad, 0xbe, 0xef,
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	311 ]);
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	312 data.into()
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	313 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	314
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	315 /// Pad an hexadecimal string to reach `NODE_NYBBLES_LENGTH`
46033 88e741bf2d93 rust: use NodePrefix::from_hex instead of hex::decode directly Simon Sapin <simon-commits@exyr.org> parents: 45531 diff changeset	316 ///check_hash
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	317 /// The padding is made with zeros
44183 e52401a95b94 rust-nodemap: NodeMap trait with simplest implementation Georges Racinet <georges.racinet@octobus.net> parents: 44182 diff changeset	318 pub fn hex_pad_right(hex: &str) -> String {
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	319 let mut res = hex.to_string();
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	320 while res.len() < NODE_NYBBLES_LENGTH {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	321 res.push('0');
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	322 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	323 res
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	324 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	325
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	326 fn sample_node_hex() -> String {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	327 hex_pad_right("0123456789abcdeffedcba9876543210deadbeef")
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	328 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	329
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	330 #[test]
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	331 fn test_node_from_hex() {
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	332 assert_eq!(Node::from_hex(&sample_node_hex()), Ok(sample_node()));
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	333
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	334 let mut short = hex_pad_right("0123");
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	335 short.pop();
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	336 short.pop();
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	337 assert_eq!(
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	338 Node::from_hex(&short),
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	339 Err(NodeError::ExactLengthRequired(NODE_NYBBLES_LENGTH, short)),
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	340 );
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	341
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	342 let not_hex = hex_pad_right("012... oops");
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	343 assert_eq!(
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	344 Node::from_hex(&not_hex),
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	345 Err(NodeError::HexError(
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	346 FromHexError::InvalidHexCharacter { c: '.', index: 3 },
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	347 not_hex,
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	348 )),
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	349 );
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	350 }
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	351
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	352 #[test]
7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	353 fn test_node_encode_hex() {
46427 6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	354 assert_eq!(format!("{:x}", sample_node()), sample_node_hex());
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	355 }
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	356
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	357 #[test]
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	358 fn test_prefix_from_hex() -> Result<(), NodeError> {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	359 assert_eq!(
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	360 NodePrefix::from_hex("0e1")?,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	361 NodePrefix {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	362 buf: vec![14, 16],
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	363 is_odd: true
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	364 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	365 );
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	366 assert_eq!(
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	367 NodePrefix::from_hex("0e1a")?,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	368 NodePrefix {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	369 buf: vec![14, 26],
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	370 is_odd: false
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	371 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	372 );
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	373
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	374 // checking limit case
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	375 let node_as_vec = sample_node().data.iter().cloned().collect();
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	376 assert_eq!(
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	377 NodePrefix::from_hex(sample_node_hex())?,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	378 NodePrefix {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	379 buf: node_as_vec,
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	380 is_odd: false
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	381 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	382 );
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	383
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	384 Ok(())
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	385 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	386
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	387 #[test]
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	388 fn test_prefix_from_hex_errors() {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	389 assert_eq!(
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	390 NodePrefix::from_hex("testgr"),
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	391 Err(NodeError::HexError(
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	392 FromHexError::InvalidHexCharacter { c: 't', index: 0 },
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	393 "testgr".to_string()
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	394 ))
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	395 );
46427 6380efb82191 rust: replace Node::encode_hex with std::fmt::LowerHex Simon Sapin <simon.sapin@octobus.net> parents: 46391 diff changeset	396 let mut long = format!("{:x}", NULL_NODE);
44182 9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	397 long.push('c');
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	398 match NodePrefix::from_hex(&long)
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	399 .expect_err("should be refused as too long")
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	400 {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	401 NodeError::PrefixTooLong(s) => assert_eq!(s, long),
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	402 err => panic!(format!("Should have been TooLong, got {:?}", err)),
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	403 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	404 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	405
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	406 #[test]
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	407 fn test_is_prefix_of() -> Result<(), NodeError> {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	408 let mut node_data = [0; NODE_BYTES_LENGTH];
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	409 node_data[0] = 0x12;
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	410 node_data[1] = 0xca;
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	411 let node = Node::from(node_data);
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	412 assert!(NodePrefix::from_hex("12")?.borrow().is_prefix_of(&node));
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	413 assert!(!NodePrefix::from_hex("1a")?.borrow().is_prefix_of(&node));
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	414 assert!(NodePrefix::from_hex("12c")?.borrow().is_prefix_of(&node));
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	415 assert!(!NodePrefix::from_hex("12d")?.borrow().is_prefix_of(&node));
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	416 Ok(())
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	417 }
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	418
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	419 #[test]
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	420 fn test_get_nybble() -> Result<(), NodeError> {
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	421 let prefix = NodePrefix::from_hex("dead6789cafe")?;
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	422 assert_eq!(prefix.borrow().get_nybble(0), 13);
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	423 assert_eq!(prefix.borrow().get_nybble(7), 9);
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	424 Ok(())
9896a8d0d3d2 rust-node: handling binary Node prefix Georges Racinet <georges.racinet@octobus.net> parents: 44143 diff changeset	425 }
44388 5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	426
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	427 #[test]
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	428 fn test_first_different_nybble_even_prefix() {
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	429 let prefix = NodePrefix::from_hex("12ca").unwrap();
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	430 let prefref = prefix.borrow();
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	431 let mut node = Node::from([0; NODE_BYTES_LENGTH]);
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	432 assert_eq!(prefref.first_different_nybble(&node), Some(0));
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	433 node.data[0] = 0x13;
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	434 assert_eq!(prefref.first_different_nybble(&node), Some(1));
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	435 node.data[0] = 0x12;
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	436 assert_eq!(prefref.first_different_nybble(&node), Some(2));
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	437 node.data[1] = 0xca;
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	438 // now it is a prefix
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	439 assert_eq!(prefref.first_different_nybble(&node), None);
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	440 }
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	441
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	442 #[test]
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	443 fn test_first_different_nybble_odd_prefix() {
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	444 let prefix = NodePrefix::from_hex("12c").unwrap();
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	445 let prefref = prefix.borrow();
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	446 let mut node = Node::from([0; NODE_BYTES_LENGTH]);
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	447 assert_eq!(prefref.first_different_nybble(&node), Some(0));
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	448 node.data[0] = 0x13;
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	449 assert_eq!(prefref.first_different_nybble(&node), Some(1));
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	450 node.data[0] = 0x12;
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	451 assert_eq!(prefref.first_different_nybble(&node), Some(2));
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	452 node.data[1] = 0xca;
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	453 // now it is a prefix
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	454 assert_eq!(prefref.first_different_nybble(&node), None);
5ac1eecc9c64 rust-nodemap: core implementation for shortest Georges Racinet <georges.racinet@octobus.net> parents: 44187 diff changeset	455 }
44143 7f86426fdd2c rust-node: binary Node ID and conversion utilities Georges Racinet <georges.racinet@octobus.net> parents: diff changeset	456 }
44183 e52401a95b94 rust-nodemap: NodeMap trait with simplest implementation Georges Racinet <georges.racinet@octobus.net> parents: 44182 diff changeset	457
e52401a95b94 rust-nodemap: NodeMap trait with simplest implementation Georges Racinet <georges.racinet@octobus.net> parents: 44182 diff changeset	458 #[cfg(test)]
e52401a95b94 rust-nodemap: NodeMap trait with simplest implementation Georges Racinet <georges.racinet@octobus.net> parents: 44182 diff changeset	459 pub use tests::hex_pad_right;

Mercurial > public > mercurial-scm > hg

annotate rust/hg-core/src/revlog/node.rs @ 46427:6380efb82191