pewter 0.0.3

A library for reading and writing PE files
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

use core::ops::Deref;
use core::ops::DerefMut;

use crate::vec::Vec;

use crate::error::Result;
use crate::io::ReadData;
use crate::io::Reader;

#[derive(Debug, Clone, Default, PartialEq, Eq)]
pub struct Table<T>(pub Vec<T>);

impl<T> Table<T> {
    pub fn new() -> Self {
        Self(Vec::new())
    }

    pub fn with_capacity(capacity: usize) -> Self {
        Self(Vec::with_capacity(capacity))
    }
}

impl<T: ReadData> Table<T> {
    pub fn new_linear(data_ptr: &mut &[u8], items_count: usize) -> Result<Self> {
        let mut items = Self::with_capacity(items_count);
        for _ in 0..items.capacity() {
            items.push(data_ptr.read()?);
        }
        Ok(items)
    }
}

impl<T> Table<T> {
    pub fn new_with_reader(
        data_ptr: &mut &[u8],
        items_count: usize,
        mut read_item: impl FnMut(&mut &[u8]) -> Result<T>,
    ) -> Result<Self> {
        let mut items = Self::with_capacity(items_count);
        for _ in 0..items.capacity() {
            items.push(read_item(data_ptr)?);
        }
        Ok(items)
    }
}

impl<T> Deref for Table<T> {
    type Target = Vec<T>;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl<T> DerefMut for Table<T> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.0
    }
}