dnfile 0.4.2

.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
54

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

#[derive(Debug, Clone, serde::Serialize)]
pub struct BlobHeap<'a> {
    #[serde(skip_serializing)]
    data: &'a [u8],
}

impl<'a> BlobHeap<'a> {
    /// Construct from a borrowed slice into the parent file buffer.
    #[must_use]
    pub fn new(data: &'a [u8]) -> Self {
        Self { data }
    }

    /// Returns a copy of the blob at `index` (length-prefixed in #Blob format).
    pub fn get(&self, index: usize) -> Result<Vec<u8>> {
        Ok(self.get_ref(index)?.to_vec())
    }

    /// Returns a borrowed slice into the blob at `index` — zero-copy.
    pub fn get_ref(&self, index: usize) -> Result<&'a [u8]> {
        if index >= self.data.len() {
            return Err(Error::BlobHeapReadOutOfBound(index, self.data.len()));
        }
        // Pass at most 4 bytes — `read_compressed_usize` now handles short input.
        let header_end = index.saturating_add(4).min(self.data.len());
        let (data_length, length_size) =
            crate::utils::read_compressed_usize(&self.data[index..header_end])?;
        let payload_start = index
            .checked_add(length_size)
            .ok_or(Error::BlobHeapReadOutOfBound(index, self.data.len()))?;
        let payload_end = payload_start
            .checked_add(data_length)
            .ok_or(Error::BlobHeapReadOutOfBound(index, self.data.len()))?;
        if payload_end > self.data.len() {
            return Err(Error::BlobHeapReadOutOfBound(payload_end, self.data.len()));
        }
        Ok(&self.data[payload_start..payload_end])
    }
}

impl<'a> crate::DnPe<'a> {
    pub fn new_blob_heap(
        &self,
        _metadata_rva: &u32,
        _stream_offset: &u32,
        _stream_size: &usize,
        _stream_name: &str,
        stream_data: &'a [u8],
    ) -> Result<super::Stream<'a>> {
        Ok(super::Stream::BlobHeap(BlobHeap::new(stream_data)))
    }
}