casualx / external Goto Github PK

fn suspend_process(proc_id: ProcessId) -> Result<()> {
    let threads = EnumThreads::create().context("Failed to collect running threads")?;
    let threads = threads.filter(|thread| thread.process_id() == proc_id);

    threads.for_each(|thread_entry| {
        if let Ok(thread) =
            Thread::attach(thread_entry.thread_id(), ThreadRights::new().suspend_resume())
        {
            unsafe { SuspendThread(thread.into_inner()) };
        }
    });

    Ok(())
}

fn resume_process(proc_id: ProcessId) -> Result<()> {
    let threads = EnumThreads::create().context("Failed to collect running threads")?;
    let threads = threads.filter(|thread| thread.process_id() == proc_id);

    threads.for_each(|thread_entry| {
        if let Ok(thread) =
            Thread::attach(thread_entry.thread_id(), ThreadRights::new().suspend_resume())
        {
            let handle = thread.into_inner();
            loop {
                if matches!(unsafe { ResumeThread(handle) }, 0 | 0xFFFFFFFF) {
                    break;
                }
            }
        }
    });

    Ok(())
}