当前位置：首页 > article >正文

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

article 2024/11/14 18:24:56

前言

前面一篇文章写了使用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-openprocess
        let 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-terminateprocess
        let 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();
}