fromsoftware-shared 0.13.0

Helpers for dealing with pointers and other common stuff across games
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

use std::sync::LazyLock;

use pelite::pe64::{Pe, PeFile, PeObject, PeView};
use windows::Win32::System::LibraryLoader::GetModuleHandleA;
use windows::core::PCSTR;

#[derive(Copy, Clone)]
pub enum Program<'a> {
    File(PeFile<'a>),
    Mapping(PeView<'a>),
}

static CURRENT_BASE: LazyLock<Program> = LazyLock::new(|| {
    let module = unsafe { GetModuleHandleA(PCSTR(std::ptr::null())).unwrap().0 } as *const u8;
    Program::Mapping(unsafe { PeView::module(module) })
});

impl Program<'_> {
    /// Returns the currently running programing.
    pub fn current() -> Self {
        *CURRENT_BASE
    }
}

unsafe impl<'a> Pe<'a> for Program<'a> {}
unsafe impl<'a> PeObject<'a> for Program<'a> {
    fn image(&self) -> &'a [u8] {
        match self {
            Self::File(file) => file.image(),
            Self::Mapping(mapping) => mapping.image(),
        }
    }

    fn align(&self) -> pelite::Align {
        match self {
            Self::File(file) => file.align(),
            Self::Mapping(mapping) => mapping.align(),
        }
    }
}