fugue-bytes 0.2.2

A binary analysis framework written in Rust
Documentation 
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

use paste::paste;

use crate::order::Order;

#[cfg(feature = "extra-integer-types")]
use crate::primitives::*;

pub trait ByteCast: Copy {
    const SIZEOF: usize;
    const SIGNED: bool;

    fn from_bytes<O: Order>(buf: &[u8]) -> Self;
    fn into_bytes<O: Order>(&self, buf: &mut [u8]);
}

macro_rules! impl_for {
    ($t:ident, $read:ident, $write:ident, $signed:ident) => {
        impl ByteCast for $t {
            const SIZEOF: usize = std::mem::size_of::<$t>();
            const SIGNED: bool = $signed;

            fn from_bytes<O: Order>(buf: &[u8]) -> Self {
                O::$read(buf)
            }

            fn into_bytes<O: Order>(&self, buf: &mut [u8]) {
                O::$write(buf, *self)
            }
        }
    };
}

macro_rules! impls_for {
    ([$($tname:ident),*], $signed:ident) => {
        $(
            paste! {
                impl_for!($tname, [<read_ $tname>], [<write_ $tname>], $signed);
            }
        )*
    };
}

impl ByteCast for bool {
    const SIZEOF: usize = 1;
    const SIGNED: bool = false;

    fn from_bytes<O: Order>(buf: &[u8]) -> Self {
        !buf.is_empty() && buf[0] != 0
    }

    fn into_bytes<O: Order>(&self, buf: &mut [u8]) {
        O::write_u8(buf, if *self { 1 } else { 0 })
    }
}

#[cfg(feature = "extra-integer-types")]
impl ByteCast for u24 {
    const SIZEOF: usize = 3;
    const SIGNED: bool = false;

    fn from_bytes<O: Order>(buf: &[u8]) -> Self {
        <O as Order>::read_u24(buf)
    }

    fn into_bytes<O: Order>(&self, buf: &mut [u8]) {
        <O as Order>::write_u24(buf, *self)
    }
}

impls_for! { [i8, i16, i32, i64, i128, isize], true }
impls_for! { [u8, u16, u32, u64, u128, usize], false }