lief 1.0.0

Official Rust bindings for LIEF
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

use lief_ffi as ffi;

use bitflags::bitflags;
use std::{fmt, marker::PhantomData};

use crate::common::{FromFFI, into_optional};

use super::{Symbol, commands::Dylib};

/// This structure represents an export (info) in a Mach-O binary
pub struct ExportInfo<'a> {
    ptr: cxx::UniquePtr<ffi::MachO_ExportInfo>,
    _owner: PhantomData<&'a ()>,
}

#[allow(non_camel_case_types)]
#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
pub enum Kind {
    REGULAR,
    THREAD_LOCAL,
    ABSOLUTE,
    UNKNOWN(u64),
}

impl From<u64> for Kind {
    fn from(value: u64) -> Self {
        match value {
            0x00000000 => Kind::REGULAR,
            0x00000001 => Kind::THREAD_LOCAL,
            0x00000002 => Kind::ABSOLUTE,
            _ => Kind::UNKNOWN(value),
        }
    }
}

bitflags! {
    #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
    pub struct Flags: u64 {
        const WEAK_DEFINITION = 0x4;
        const REEXPORT = 0x8;
        const STUB_AND_RESOLVER = 0x10;
        const STATIC_RESOLVER = 0x20;
    }
}

impl From<u64> for Flags {
    fn from(value: u64) -> Self {
        Flags::from_bits_truncate(value)
    }
}
impl From<Flags> for u64 {
    fn from(value: Flags) -> Self {
        value.bits()
    }
}
impl std::fmt::Display for Flags {
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        bitflags::parser::to_writer(self, f)
    }
}

impl fmt::Debug for ExportInfo<'_> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("ExportInfo")
            .field("node_offset", &self.node_offset())
            .field("flags", &self.flags())
            .field("address", &self.address())
            .field("other", &self.other())
            .field("kind", &self.kind())
            .finish()
    }
}

impl ExportInfo<'_> {
    /// Original offset in the export Trie
    pub fn node_offset(&self) -> u64 {
        self.ptr.node_offset()
    }

    pub fn flags(&self) -> Flags {
        Flags::from(self.ptr.flags())
    }

    /// The address of the export
    pub fn address(&self) -> u64 {
        self.ptr.address()
    }

    pub fn other(&self) -> u64 {
        self.ptr.other()
    }

    /// The export's kind (regular, thread local, absolute, ...)
    pub fn kind(&self) -> Kind {
        Kind::from(self.ptr.kind())
    }

    /// Symbol associated with this export
    pub fn symbol(&self) -> Option<Symbol<'_>> {
        into_optional(self.ptr.symbol())
    }

    /// If the export is a re-export ([`Flags::REEXPORT`]) this function returns
    /// the symbol being re-exported
    pub fn alias(&self) -> Option<Symbol<'_>> {
        into_optional(self.ptr.alias())
    }

    /// If the export is a re-export ([`Flags::REEXPORT`]) this function returns
    /// the library from which the symbol is re-exported
    pub fn alias_library(&self) -> Option<Dylib<'_>> {
        into_optional(self.ptr.alias_library())
    }
}

impl<'a> FromFFI<ffi::MachO_ExportInfo> for ExportInfo<'a> {
    fn from_ffi(ptr: cxx::UniquePtr<ffi::MachO_ExportInfo>) -> Self {
        Self {
            ptr,
            _owner: PhantomData,
        }
    }
}