dnfile 0.3.0

.NET binary extractor
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

use crate::{Result, error::Error};

#[derive(Debug, Clone, serde::Serialize)]
pub struct UserStringHeap {
    #[serde(skip_serializing)]
    data: Vec<u8>,
}

impl UserStringHeap {
    pub fn get(&self, index: usize) -> Result<Vec<u8>> {
        if index >= self.data.len() {
            return Err(Error::UserStringHeapReadOutOfBound(index, self.data.len()));
        }

        let (data_length, length_size) = crate::utils::read_compressed_usize(
            self.data
                .get(index..)
                .ok_or(Error::UserStringHeapReadOutOfBound(index, self.data.len()))?,
        )?;

        let end_index = index
            .checked_add(length_size)
            .and_then(|i| i.checked_add(data_length));
        match end_index {
            Some(end) if end <= self.data.len() => Ok(self.data[index + length_size..end].to_vec()),
            _ => Err(Error::UserStringHeapReadOutOfBound(index, self.data.len())),
        }
    }

    pub fn get_us(&self, index: usize) -> Result<String> {
        let data = self.get(index)?;
        let utf16: Vec<u16> = data
            .chunks_exact(2)
            .map(|chunk| u16::from_le_bytes([chunk[0], chunk[1]]))
            .collect();
        Ok(String::from_utf16(&utf16)?)
    }
}

impl crate::DnPe {
    pub fn new_user_string_heap(
        &self,
        _metadata_rva: &u32,
        _stream_offset: &u32,
        _stream_size: &usize,
        _stream_name: &str,
        stream_data: Vec<u8>,
    ) -> Result<super::Stream> {
        Ok(super::Stream::UserStringHeap(UserStringHeap {
            data: stream_data,
        }))
    }
}