dylink 0.5.0

Run-time dynamic linking loader utilities
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

#[cfg(windows)]
#[test]
#[ignore = "this is just instrumentation"]
fn test_win32_alloc_instrumentation() {
	use dylink::*;
	use std::alloc::{GlobalAlloc, Layout, System};
	use std::sync::atomic::AtomicUsize;
	use std::sync::atomic::Ordering;

	struct MyAllocator(AtomicUsize);

	unsafe impl GlobalAlloc for MyAllocator {
		unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
			self.0.fetch_add(1, Ordering::SeqCst);
			System.alloc(layout)
		}

		unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
			System.dealloc(ptr, layout)
		}
	}

	#[global_allocator]
	static GLOBAL: MyAllocator = MyAllocator(AtomicUsize::new(0));

	// macro output: function
	#[dylink(name = "Kernel32.dll")]
	extern "C" {
		fn GetLastError() -> u32;
	}

	// factor in any allocs that aren't mine, take the difference and print it.
	unsafe {
		let test_allocs = GLOBAL.0.load(Ordering::Acquire);
		let _ = GetLastError();
		let dylink_allocs = GLOBAL.0.load(Ordering::Acquire) - test_allocs;
		println!("dylink allocations={dylink_allocs}");
	}
}