ton_lib 0.0.38

A collection of types and utilities for interacting with the TON network
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

use num_bigint::BigUint;
use num_traits::{One, Zero};

/// All functions except `add_leading_bit` expect 1 extra leading bit in `val` set to 1
pub fn all_bits_same(val: &BigUint) -> bool {
    if val.is_zero() {
        return true;
    }
    let origin_bits = val.bits();
    let all_zero = (val - 1u32).bits() != origin_bits;
    let all_ones = (val + 1u32).bits() != origin_bits;
    all_zero || all_ones
}

pub fn common_prefix_len(a: &BigUint, b: &BigUint) -> usize {
    let xor = a ^ b;
    (a.bits() - xor.bits() - 1) as usize // don't forget leading zero
}

pub fn remove_leading_bit(val: &BigUint) -> BigUint {
    let bits = val.bits();
    let mask = BigUint::one() << (bits - 1);
    val ^ mask
}

pub fn add_leading_bit(val: &BigUint, val_bit_len: usize) -> BigUint {
    let leading_bit = BigUint::one() << val_bit_len;
    leading_bit | val
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_all_bits_same() {
        for val in [BigUint::from(0u32), BigUint::from(0b1111u32), BigUint::from(0b1000u32)] {
            assert!(all_bits_same(&val));
        }

        let val = BigUint::from(0b1011u32);
        assert!(!all_bits_same(&val));
    }

    #[test]
    fn test_common_prefix_len() {
        let a = BigUint::from(0b1011u32);
        let b = BigUint::from(0b1010u32);
        assert_eq!(common_prefix_len(&a, &b), 2);

        let a = BigUint::from(0b1011u32);
        let b = BigUint::from(0b1011u32);
        assert_eq!(common_prefix_len(&a, &b), 3);
    }

    #[test]
    fn test_remove_leading_bit() {
        let val = BigUint::from(0b1011u32);
        assert_eq!(remove_leading_bit(&val), BigUint::from(0b011u32));

        let val = BigUint::from(0b1111u32);
        assert_eq!(remove_leading_bit(&val), BigUint::from(0b111u32));

        let val = BigUint::from(0b1u32);
        assert_eq!(remove_leading_bit(&val), BigUint::from(0u32));
    }

    #[test]
    fn test_add_leading_bit() {
        let val = BigUint::from(0b1011u32);
        assert_eq!(add_leading_bit(&val, 4), BigUint::from(0b11011u32));

        let val = BigUint::from(0b1111u32);
        assert_eq!(add_leading_bit(&val, 4), BigUint::from(0b11111u32));

        let val = BigUint::from(0u32);
        assert_eq!(add_leading_bit(&val, 1), BigUint::from(0b10u32));
    }
}