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

use std::ops::Range;
use crate::*;
use super::Wrap;

/// Describes the PE headers.
impl<'a, Pe32: pe32::Pe<'a>, Pe64: pe64::Pe<'a>> Wrap<pe32::headers::Headers<Pe32>, pe64::headers::Headers<Pe64>> {
	/// Gets the PE instance.
	#[inline]
	pub fn pe(&self) -> Wrap<Pe32, Pe64> {
		match self {
			Wrap::T32(headers) => Wrap::T32(headers.pe()),
			Wrap::T64(headers) => Wrap::T64(headers.pe()),
		}
	}
	/// Gets the PE headers as a byte slice.
	#[inline]
	pub fn image(&self) -> &'a [u8] {
		match self {
			Wrap::T32(headers) => headers.image(),
			Wrap::T64(headers) => headers.image(),
		}
	}
	/// Calculates the optional header's CheckSum.
	#[inline]
	pub fn check_sum(&self) -> u32 {
		match self {
			Wrap::T32(headers) => headers.check_sum(),
			Wrap::T64(headers) => headers.check_sum(),
		}
	}
	/// Gets the code range from the optional header.
	#[inline]
	pub fn code_range(&self) -> Range<u32> {
		match self {
			Wrap::T32(headers) => headers.code_range(),
			Wrap::T64(headers) => headers.code_range(),
		}
	}
	/// Gets the full image range excluding the PE headers.
	#[inline]
	pub fn image_range(&self) -> Range<u32> {
		match self {
			Wrap::T32(headers) => headers.image_range(),
			Wrap::T64(headers) => headers.image_range(),
		}
	}
}