teamy-mft 0.7.1

TeamDman's Master File Table CLI and library for NTFS.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use eyre::eyre;
use windows::Win32::Foundation::CloseHandle;
use windows::Win32::Foundation::HANDLE;
use windows::Win32::System::Threading::GetExitCodeProcess;
use windows::Win32::System::Threading::INFINITE;
use windows::Win32::System::Threading::WaitForSingleObject;

#[derive(Debug)]
pub struct ElevatedChildProcess {
    pub h_process: HANDLE,
}

impl ElevatedChildProcess {
    pub fn wait(self) -> eyre::Result<u32> {
        unsafe { WaitForSingleObject(self.h_process, INFINITE) };
        let mut code = 0u32;
        unsafe { GetExitCodeProcess(self.h_process, &mut code) }
            .map_err(|e| eyre!("Failed to get exit code: {}", e))?;
        unsafe { CloseHandle(self.h_process) }?;
        Ok(code)
    }
}