【Rust调用Windows API】获取正在运行的全部进程信息

ops/2024/11/17 18:45:39/

前言

WIndows API官方文档 提供了C++的调用示例，最近想尝试用Rust去实现，本系列博客记录一下实现过程。

依赖

Rust调用Windows API需要引入依赖winapi，在Cargo.toml中添加依赖

winapi = "0.3.9"

调用不同的API集就需要使用相应的功能features，很好的一个判断方式是你在微软官方文档中看到的是在哪个头文件内，就添加哪个feature，例如本篇文章需要使用 tlhelp32.h 和 processthreadsapi.h 那么就将这俩feature添加进去

winapi = { version = "0.3.9", features = ["tlhelp32", "processthreadsapi"] }

实现

大致步骤：

  1. 创建进程快照，拿到快照句柄
  2. 遍历快照中的进程（以迭代器的方式实现），得到每个进程的数据
  3. 释放快照句柄

创建快照句柄

创建进程快照需要用到 CreateToolhelp32Snapshot 方法，它在 tlhelp32.h 头文件中定义。

use winapi::um::tlhelp32::{CreateToolhelp32Snapshot, TH32CS_SNAPPROCESS};
use winapi::um::handleapi::INVALID_HANDLE_VALUE;
use winapi::um::errhandlingapi::GetLastError;/// 保存进程快照并返回进程信息迭代器 `ProcessInformationIterator`
pub fn list() -> Result<ProcessInformationIterator, String> {let process_snapshot: HANDLE = unsafe { CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0) };if process_snapshot == INVALID_HANDLE_VALUE || process_snapshot.is_null() {unsafe {return Err(format!("Cannot list processes, code: {}", GetLastError()));}}Ok(ProcessInformationIterator::new(process_snapshot))
}

代码中的 ProcessInfomationIterator 是自定义的，为了结构更清晰，这里使用迭代器模式来读取。

如果保存进程快照失败，返回的句柄会是一个无效的值（这里用了两个条件或的关系去判断是否无效，其实任用其一都可以，他们都表示一个“空”内存或“空”指针），使用 GetLastError 方法可以获取错误代码，错误代码对应含义见系统错误代码说明，也可以通过API解析成可读文本，这个后面的文章再介绍，这里先用code简单表示一下。

实现迭代器

Rust中的迭代器模式实现方法这里就不多赘述，你只需要知道实现一个迭代器至少需要 一个迭代元素Item 和 一个实现了Iterator特征的迭代器 就可以了。

迭代元素Item

let vec = vec![1, 2]
for item in vec {...
}

上面代码的item就是迭代器中具体的元素，因为进程信息有很多，这里就使用一个结构体来存

use winapi::um::tlhelp32::PROCESSENTRY32;pub struct ProcessInformation {inner: PROCESSENTRY32,
}

这里并没有直接将进程的数据解析之后再存入结构体，而是直接将 PROCESSENTRY32 结构体做一个包装，这里是为了节省不必要的计算，从句柄中直接读取出来的 PROCESSENTRY32 并不是所有信息都是Rust直接可读的，在需要时才解析，并且通过getter方法读取数据更方便以后拓展，下面是ProcessInformation的具体方法实现。

use winapi::um::processthreadsapi::{GetPriorityClass, OpenProcess};
use winapi::um::errhandlingapi::GetLastError;
use winapi::um::tlhelp32::PROCESSENTRY32;
use winapi::um::winnt::{HANDLE, PROCESS_ALL_ACCESS};pub(crate) fn char_arr_to_string(chars: &[i8]) -> String {chars.into_iter().map(|&c| c as u8 as char).collect()
}impl ProcessInformation {pub(crate) fn new(entry: PROCESSENTRY32) -> ProcessInformation {ProcessInformation {inner: entry}}/// 获取进程IDpub fn get_pid(&self) -> u32 {self.inner.th32ProcessID as u32}/// 获取进程名pub fn get_name(&self) -> String {char_arr_to_string(&self.inner.szExeFile)}/// 获取父进程IDpub fn get_parent_pid(&self) -> u32 {self.inner.th32ParentProcessID as u32}/// 获取线程数量pub fn get_thread_count(&self) -> usize {self.inner.cntThreads as usize}/// 获取基础优先权值pub fn get_priority_base(&self) -> i64 {self.inner.pcPriClassBase as i64}/// 获取优先权类别，如果调用进程没有指定权限可能会获取失败，失败时返回 `None`pub fn get_priority_class(&self) -> Option<i32> {let mut priority_class = None;unsafe {let handle = OpenProcess(PROCESS_ALL_ACCESS, 0, self.inner.th32ProcessID);if !handle.is_null() {let class = GetPriorityClass(handle);CloseHandle(handle);priority_class = Some(class as i32);}}priority_class}
}

迭代器实现

迭代器中需要保存一些迭代遍历的状态，因此除了前面保存的快照句柄之外还要存储迭代的索引以及释放句柄的状态，迭代器是不可逆的。

use winapi::um::winnt::HANDLE;pub struct ProcessInformationIterator {process_snapshot: HANDLE,index: usize,finished: bool,
}impl ProcessInformationIterator {pub(crate) fn new(process_snapshot: HANDLE) -> ProcessInformationIterator {ProcessInformationIterator {process_snapshot,index: 0,finished: false,}}
}

然后就是迭代器的具体实现

use winapi::um::winnt::HANDLE;
use winapi::um::tlhelp32::{Process32First, Process32Next, PROCESSENTRY32};
use winapi::um::handleapi::CloseHandle;impl Iterator for ProcessInformationIterator {type Item = ProcessInformation;fn next(&mut self) -> Option<Self::Item> {if self.finished {return None;}self.index += 1;let mut entry: PROCESSENTRY32 = unsafe { std::mem::zeroed() };entry.dwSize = size_of::<PROCESSENTRY32>() as u32;//  读取快照中的第一个进程let res = unsafe {if self.index == 1 {Process32First(self.process_snapshot, &mut entry)} else {Process32Next(self.process_snapshot, &mut entry)}};if res == 0 {unsafe {CloseHandle(self.process_snapshot);}self.finished = true;return None;}Some(ProcessInformation::new(entry))}
}

上面的代码有几点需要说明一下：

  1. entry在初始化时需要先到一个全0值的内存空间，并不是分配为一个Rust引用空间，这里用 unsafe 方法 std::mem::zeroed()
  2. 在读取进程Entry之前需要先指定内存长度，这里用 size_of::<PROCESSENTRY32>() 来获取并赋值给 entry.dwSize
  3. 遍历时第一个元素需要调用 Process32First读取，后续的使用 Process32Next 读取
  4. 遍历完时记得关闭快照剧本 使用 CloseHandle 接口

特殊情况处理：如果用户并没有迭代完，上面的代码实现可能会出现快照句柄未释放的情况，所以还需要给迭代器实现一个Drop特征，在释放迭代器时释放快照句柄

impl Drop for ProcessInformationIterator {fn drop(&mut self) {if self.finished {return;}// 释放快照句柄。unsafe {CloseHandle(self.process_snapshot);}self.finished = true;}
}

代码汇总

我在写的时候放在了自定义的utils::process::win包下面，具体引用路径根据自己的情况调整

mod.rs文件

use crate::utils::process::win::process_information::ProcessInformationIterator;
use winapi::um::tlhelp32::{CreateToolhelp32Snapshot, TH32CS_SNAPPROCESS};
use winapi::um::errhandlingapi::GetLastError;
use winapi::um::handleapi::INVALID_HANDLE_VALUE;
use winapi::um::winnt::HANDLE;pub mod process_information;pub fn list() -> Result<ProcessInformationIterator, String> {let process_snapshot: HANDLE = unsafe { CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0) };if process_snapshot == INVALID_HANDLE_VALUE || process_snapshot.is_null() {unsafe {return Err(format!("Cannot list processes, code: {}", GetLastError()));}}Ok(ProcessInformationIterator::new(process_snapshot))
}pub(crate) fn char_arr_to_string(chars: &[i8]) -> String {chars.into_iter().map(|&c| c as u8 as char).collect()
}

process_information.rs文件

use crate::utils::process::win::char_arr_to_string;
use crate::utils::process::win::process_module::ProcessModuleIterator;
use winapi::um::errhandlingapi::GetLastError;
use winapi::um::handleapi::CloseHandle;
use winapi::um::processthreadsapi::{GetPriorityClass, OpenProcess};
use winapi::um::tlhelp32::{Process32First, Process32Next, PROCESSENTRY32};
use winapi::um::winnt::{HANDLE, PROCESS_ALL_ACCESS};/// [PROCESSENTRY32](https://learn.microsoft.com/zh-cn/windows/win32/api/tlhelp32/ns-tlhelp32-processentry32) 的Rust包装实现
pub struct ProcessInformation {inner: PROCESSENTRY32,
}impl ProcessInformation {pub(crate) fn new(entry: PROCESSENTRY32) -> ProcessInformation {ProcessInformation {inner: entry}}/// 获取进程IDpub fn get_pid(&self) -> u32 {self.inner.th32ProcessID as u32}/// 获取进程名pub fn get_name(&self) -> String {char_arr_to_string(&self.inner.szExeFile)}/// 获取父进程IDpub fn get_parent_pid(&self) -> u32 {self.inner.th32ParentProcessID as u32}/// 获取线程数量pub fn get_thread_count(&self) -> usize {self.inner.cntThreads as usize}/// 获取基础优先权值pub fn get_priority_base(&self) -> i64 {self.inner.pcPriClassBase as i64}/// 获取优先权类别，如果调用进程没有指定权限可能会获取失败，失败时返回 `None`pub fn get_priority_class(&self) -> Option<i32> {let mut priority_class = None;unsafe {let handle = OpenProcess(PROCESS_ALL_ACCESS, 0, self.inner.th32ProcessID);if !handle.is_null() {let class = GetPriorityClass(handle);CloseHandle(handle);priority_class = Some(class as i32);}}priority_class}
}pub struct ProcessInformationIterator {process_snapshot: HANDLE,index: usize,finished: bool,
}impl ProcessInformationIterator {pub(crate) fn new(process_snapshot: HANDLE) -> ProcessInformationIterator {ProcessInformationIterator {process_snapshot,index: 0,finished: false,}}
}impl Drop for ProcessInformationIterator {fn drop(&mut self) {if self.finished {return;}// 释放快照句柄。unsafe {CloseHandle(self.process_snapshot);}self.finished = true;}
}impl Iterator for ProcessInformationIterator {type Item = ProcessInformation;fn next(&mut self) -> Option<Self::Item> {if self.finished {return None;}self.index += 1;let mut entry: PROCESSENTRY32 = unsafe { std::mem::zeroed() };entry.dwSize = size_of::<PROCESSENTRY32>() as u32;//  读取快照中的第一个进程let res = unsafe {if self.index == 1 {Process32First(self.process_snapshot, &mut entry)} else {Process32Next(self.process_snapshot, &mut entry)}};if res == 0 {unsafe {CloseHandle(self.process_snapshot);}self.finished = true;return None;}Some(ProcessInformation::new(entry))}
}

测试

测试代码

#[test]
pub fn test_list() {println!("PID\tName\tParent PID\tThreads\tPriority Base\tPriority Class");let iter = list().unwrap();for process in iter {let pid = process.get_pid();let name = process.get_name();let parent_pid = process.get_parent_pid();let thread_count = process.get_thread_count();let priority_base = process.get_priority_base();let priority_class = process.get_priority_class();println!("{}\t{}\t{}\t{}\t{}\t{:?}", pid, name, parent_pid, thread_count, priority_base, priority_class)}
}

结果

PID     Name    Parent PID      Threads Priority Base   Priority Class
0       [System Process]        0       6       0       None
4       System  0       236     8       None
64      Secure System   4       0       8       None
132     Registry        4       4       8       None
504     smss.exe        4       2       11      None
728     csrss.exe       712     11      13      None
824     wininit.exe     712     1       13      None
832     csrss.exe       816     15      13      None...12624   chrome.exe      12148   19      8       Some(32)
16352   chrome.exe      12148   18      4       Some(64)
14904   chrome.exe      12148   17      4       Some(64)
14672   wslinstaller.exe        892     2       8       None
11160   chrome.exe      12148   20      4       Some(64)
18048   chrome.exe      12148   19      4       Some(64)
5452    chrome.exe      12148   14      4       Some(64)
14468   svchost.exe     892     3       8       None
18060   chrome.exe      12148   14      4       Some(64)
17748   dllhost.exe     688     8       8       Some(32)
16084   vctip.exe       16648   27      8       Some(32)
9008    OpenConsole.exe 10644   6       8       Some(32)
15516   cargo.exe       10644   4       8       Some(32)
11312   cargo.exe       15516   4       8       Some(32)

http://www.ppmy.cn/ops/134497.html

相关文章

Python - 初识Python；Python解释器下载安装；Python IDE（一）

Python - 初识Python；Python解释器下载安装；Python IDE（一）

一、初识Python Python 是一种高级编程语言&#xff0c;Python是一种面向对象的解释型计算机程序设计语言&#xff0c;Python由荷兰国家数学与计算机科学研究中心的吉多范罗苏姆&#xff08;&#xff09;Guido van Rossum吉多范罗苏姆&#xff08;&#xff09;于1989 年底发明…
阅读更多...
羊城杯2020Easyphp

羊城杯2020Easyphp

审题 看到url&#xff0c;可以想到伪协议读取 尝试过后可以发现&#xff0c;题目绕过了read后面的编码 我们可以尝试双重urlencode进行绕过 ?filephp://filter/read%25%36%33%25%36%66%25%36%65%25%37%36%25%36%35%25%37%32%25%37%34%25%32%65%25%36%32%25%36%31%25%37%33%…
阅读更多...
react 中 memo 模块作用

react 中 memo 模块作用

memo是一个用于优化组件性能的高阶组件。 一、避免不必要的重新渲染 1. 浅比较机制原理 1.1 组件渲染触发条件 在 React 中&#xff0c;当组件的props或state发生变化时&#xff0c;组件会重新渲染。然而&#xff0c;在某些情况下&#xff0c;即使父组件重新渲染并传递给子…
阅读更多...
STM32 ADC --- 任意单通道采样

STM32 ADC --- 任意单通道采样

STM32 ADC — 单通道采样 文章目录 STM32 ADC --- 单通道采样cubeMX配置代码修改&#xff1a;应用 使用cubeMX生成HAL工程 需求&#xff1a;有多个通道需要进行ADC采样&#xff0c;实现每次采样只采样一个通道&#xff0c;且可以随时采样不同通道的功能。 cubeMX配置 这里我们…
阅读更多...
HarmonyOS4+NEXT星河版入门与项目实战--------开发工具与环境准备

HarmonyOS4+NEXT星河版入门与项目实战--------开发工具与环境准备

文章目录 1、熟悉鸿蒙官网1、打开官网2、下载 DevEco Studio3、HarmonyOS 资源库4、开发指南与API 2、安装 DevEco Studio1、软件安装2、配置开发工具 1、熟悉鸿蒙官网 1、打开官网 百度搜索 鸿蒙开发者官网 点击进入开发者官网&#xff0c;点击开发&#xff0c;可以看到各种…
阅读更多...
leetcode 扫描线专题 06-leetcode.252 meeting room 力扣.252 会议室

leetcode 扫描线专题 06-leetcode.252 meeting room 力扣.252 会议室

扫描线专题 leetcode 扫描线专题 06-扫描线算法&#xff08;Sweep Line Algorithm&#xff09; leetcode 扫描线专题 06-leetcode.218 the-skyline-problem 力扣.218 天际线问题 leetcode 扫描线专题 06-leetcode.252 meeting room 力扣.252 会议室 leetcode 扫描线专题 06…
阅读更多...
【React】状态管理之Zustand

【React】状态管理之Zustand

&#x1f308;个人主页: 鑫宝Code &#x1f525;热门专栏: 闲话杂谈&#xff5c; 炫酷HTML | JavaScript基础 ​&#x1f4ab;个人格言: "如无必要&#xff0c;勿增实体" 文章目录 状态管理之Zustand引言1. Zustand 的核心特点1.1 简单直观的 API1.2 无需 Provi…
阅读更多...
【Linux篇】面试——用户和组、文件类型、权限、进程

【Linux篇】面试——用户和组、文件类型、权限、进程

目录 一、权限管理 1. 用户和组 &#xff08;1&#xff09;相关概念 &#xff08;2&#xff09;用户命令 ① useradd&#xff08;添加新的用户账号&#xff09; ② userdel&#xff08;删除帐号&#xff09; ③ usermod&#xff08;修改帐号&#xff09; ④ passwd&…
阅读更多...
最新文章

Copyright @ 2022~2023