improve project structure (i think)

2025-12-03 12:36:45 +00:00
parent 432ec8adea
commit a9a1c4870c
8 changed files with 21 additions and 28 deletions
@@ -0,0 +1,87 @@
+use crate::Error;
+use windows::Win32::System::Diagnostics::ToolHelp::{
+    CreateToolhelp32Snapshot, Module32FirstW, Module32NextW, Process32FirstW, Process32NextW,
+    MODULEENTRY32W, PROCESSENTRY32W, TH32CS_SNAPMODULE, TH32CS_SNAPMODULE32, TH32CS_SNAPPROCESS,
+};
+
+#[derive(Debug)]
+pub struct ProcessSnapshot {
+    pub id: u32,
+    pub name: String,
+}
+
+impl ProcessSnapshot {
+    pub fn get_processes() -> Result<Vec<Self>, Error> {
+        let snapshot =
+            unsafe { CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0) }.map_err(|why| {
+                Error::CreateSnapshotError(format!("failed to create process snapshot: {why}"))
+            })?;
+        let mut process_entry_32_w = PROCESSENTRY32W {
+            dwSize: u32::try_from(size_of::<PROCESSENTRY32W>())?,
+            ..Default::default()
+        };
+        unsafe { Process32FirstW(snapshot, &mut process_entry_32_w) }.map_err(|why| {
+            Error::CreateSnapshotError(format!("failed to get first process from snapshot: {why}"))
+        })?;
+        let mut processes = Vec::new();
+        loop {
+            let name = String::from_utf16_lossy(&process_entry_32_w.szExeFile)
+                .trim_end_matches('\0')
+                .to_string();
+            let process = Self {
+                id: process_entry_32_w.th32ProcessID,
+                name,
+            };
+            processes.push(process);
+            if unsafe { Process32NextW(snapshot, &mut process_entry_32_w) }.is_err() {
+                break;
+            };
+        }
+        Ok(processes)
+    }
+}
+
+#[derive(Debug)]
+pub struct ModuleSnapshot {
+    pub name: String,
+    pub path: String,
+    pub base_address: usize,
+    pub base_size: usize,
+}
+
+impl ModuleSnapshot {
+    pub fn get_modules(pid: u32) -> Result<Vec<Self>, Error> {
+        let snapshot =
+            unsafe { CreateToolhelp32Snapshot(TH32CS_SNAPMODULE | TH32CS_SNAPMODULE32, pid) }
+                .map_err(|why| {
+                    Error::CreateSnapshotError(format!("failed to create module snapshot: {why}"))
+                })?;
+        let mut module_entry_32_w = MODULEENTRY32W {
+            dwSize: u32::try_from(size_of::<MODULEENTRY32W>())?,
+            ..Default::default()
+        };
+        unsafe { Module32FirstW(snapshot, &mut module_entry_32_w) }.map_err(|why| {
+            Error::CreateSnapshotError(format!("failed to get first module from snapshot: {why}"))
+        })?;
+        let mut modules = Vec::new();
+        loop {
+            let name = String::from_utf16_lossy(&module_entry_32_w.szModule)
+                .trim_end_matches('\0')
+                .to_string();
+            let path = String::from_utf16_lossy(&module_entry_32_w.szExePath)
+                .trim_end_matches('\0')
+                .to_string();
+            let module = Self {
+                name,
+                path,
+                base_address: module_entry_32_w.modBaseAddr as usize,
+                base_size: module_entry_32_w.modBaseSize as usize,
+            };
+            modules.push(module);
+            if unsafe { Module32NextW(snapshot, &mut module_entry_32_w).is_err() } {
+                break;
+            }
+        }
+        Ok(modules)
+    }
+}