pelite 0.10.0

Lightweight, memory-safe, zero-allocation library for reading and navigating PE binaries.
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

use crate::{Error, Result};
use crate::base_relocs::BaseRelocs;

use super::image::*;
use super::Pe;

pub(crate) fn try_from<'a, P: Pe<'a>>(pe: P) -> Result<BaseRelocs<'a>> {
	let datadir = pe.data_directory().get(IMAGE_DIRECTORY_ENTRY_BASERELOC).ok_or(Error::Bounds)?;
	let relocs = pe.slice(datadir.VirtualAddress, datadir.Size as usize, 4)?; // $1
	let relocs = unsafe { relocs.get_unchecked(..datadir.Size as usize) };
	Ok(unsafe { BaseRelocs::new(relocs) })
}

#[cfg(test)]
pub(crate) fn test<'a, P: Pe<'a>>(pe: P) -> Result<()> {
	let base_relocs = pe.base_relocs()?;
	let _ = format!("{:?}", base_relocs);

	let mut baseline = base_relocs
		.iter_blocks()
		.flat_map(move |block| {
			let _ = format!("{:?}", block);
			block.words()
				.iter()
				.filter(move |&word| block.type_of(word) != IMAGE_REL_BASED_ABSOLUTE)
				.map(move |word| block.rva_of(word))
		});

	base_relocs.for_each(|rva, _| {
		assert_eq!(baseline.next(), Some(rva));
	});
	assert_eq!(baseline.next(), None);

	Ok(())
}