【Rust调用Windows API】杀掉指定进程（包括兄弟进程、子进程、父进程）

前言

前面一篇文章写了使用Rust调用Windows API 获取正在运行的全部进程信息，本篇实现杀掉指定进程。

通过标准库可以管理当前进程创建的子进程，要 kill 掉子进程也比较容易，这里不赘述了，主要实现通过调用Windows API来杀掉兄弟进程、子进程，甚至通过提权可以杀掉父进程。

依赖

进程管理主要用到 processthreadsapi.h 头文件，详细的接口文档请前往微软官方文档查看。

在Cargo.toml中添加 processthreadsapi 这个feature

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

实现

大致步骤：

获取指定进程的文件句柄
杀掉进程
释放进程句柄

提权：本篇只讲最简单的用户操作提权方法，调用Windows API提权的实现见本系列《Rust调用Windows API》后面的文章

获取进程句柄

进程句柄的获取需要调用 OpenProcess 这个函数

pub fn OpenProcess(dwDesiredAccess: DWORD,bInheritHandle: BOOL,dwProcessId: DWORD,
) -> HANDLE;

这个函数需要传入三个参数，主要关注第一个和第三个参数

dwDesiredAccess：对该进程的访问权限，多个权限用或位运算符|拼接，全部权限见官方文档。特例：如果调用方进程拥有并开启了 SeDebugPrivilege特权，此参数可以忽略，后续提权的方案也是通过获得该特权来实现对任意进程的完全访问，一般进程是没有这个特权的。
dwProcessId：指定进程的描述符，也就是进程PID

OpenProcess这个函数在日常开发中使用的频率是比较高的，每次使用需要的权限也不固定，因此这里我将第一个参数向外抛出，由调用方指定，下面是Rust调用实现

use winapi::shared::minwindef::DWORD;
use winapi::um::winnt::HANDLE;
use winapi::um::errhandlingapi::GetLastError;
use winapi::um::processthreadsapi::OpenProcess;fn open_process(pid: DWORD, access_privilege: DWORD) -> Result<HANDLE, String> {unsafe {// https://learn.microsoft.com/zh-cn/windows/win32/api/processthreadsapi/nf-processthreadsapi-openprocesslet handle = OpenProcess(access_privilege, 0, pid);if handle.is_null() {//  https://learn.microsoft.com/zh-cn/windows/win32/debug/system-error-codes--0-499-return Err(format!("Could not open process {}, code: {}", pid, GetLastError()));}Ok(handle)}
}

杀掉进程

上一步获得了进程的句柄，直接调用 TerminateProcess 函数就可以了，在Rust中定义如下：

pubfn TerminateProcess(hProcess: HANDLE,uExitCode: UINT,
) -> BOOL;

参数解释：

hProcess：需要杀掉进程的句柄，前面通过 OpenProcess 获得
uExitCode：被杀掉进程的退出代码，和平时调用 exit(0) 传入的 0 含义一样，在winapi中通过 GetExitCodeProcess 函数可以查出被杀掉进程的退出代码

Rust调用实现：

use winapi::um::winnt::HANDLE;
use winapi::um::errhandlingapi::GetLastError;
use winapi::um::processthreadsapi::TerminateProcess;fn kill_process(process_handle: HANDLE) -> Result<(), String> {unsafe {//  https://learn.microsoft.com/zh-cn/windows/win32/api/processthreadsapi/nf-processthreadsapi-terminateprocesslet success = TerminateProcess(process_handle, 0);if success == 0 {CloseHandle(process_handle);return Err(format!("Could not terminate process, code: {}", GetLastError()));}CloseHandle(process_handle);Ok(())}
}

测试

下面写个简单的测试程序验证一下

pub fn test_kill() {print!("Please enter the PID of the process to be queried: ");stdout().flush().unwrap();let mut input = String::new();stdin().read_line(&mut input).unwrap();let pid = input.trim().parse::<i64>().unwrap() as DWORD;let process_handle = open_process(pid, PROCESS_TERMINATE).unwrap();print!("Are you sure you want to kill this process? (y/n): ");stdout().flush().unwrap();input.clear();stdin().read_line(&mut input).unwrap();if input.trim().eq("y") {match kill_process(process_handle) {Ok(_) => println!("Killed"),Err(e) => {eprintln!("Failed to kill process, code: {}", e);input.clear();stdin().read_line(&mut input).unwrap();}}} else {println!("Ignored")}
}fn main() {test_kill();
}