newport_os 0.2.0

Os abstractions for engine use
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

#[cfg(target_os = "windows")]
use crate::win32::*;

use std::ffi::CString;
use std::ptr::copy_nonoverlapping;

#[macro_export]
macro_rules! proc_address {
	($lib:expr, $proc:expr) => {{
		let func_ptr = $lib.raw_address($proc);
		if func_ptr.is_null() {
			None
		} else {
			Some(unsafe { std::mem::transmute(func_ptr) })
		}
	}};
}

pub struct Library {
	#[cfg(target_os = "windows")]
	handle: HMODULE,
}

impl Library {
	pub fn new(path: &str) -> Result<Library, ()> {
		let path = CString::new(path.as_bytes()).unwrap(); // TODO: Temp allocator
		unsafe {
			let handle = LoadLibraryA(path.as_ptr() as LPCSTR);
			if handle == INVALID_HANDLE_VALUE {
				return Err(());
			}
			Ok(Library { handle: handle })
		}
	}

	pub fn raw_address(&self, proc: &str) -> LPVOID {
		assert!(proc.len() < 512);
		let mut buffer = [0 as u8; 512];
		unsafe { copy_nonoverlapping(proc.as_ptr(), buffer.as_mut_ptr(), proc.len()) };
		buffer[proc.len()] = 0;

		unsafe { GetProcAddress(self.handle, buffer.as_ptr() as LPCSTR) as _ }
	}
}

impl Drop for Library {
	fn drop(&mut self) {
		unsafe {
			FreeLibrary(self.handle);
		}
	}
}

unsafe impl Send for Library {}
unsafe impl Sync for Library {}