teamy-mft 0.7.0

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
23
24
25
26
27
28
29
30
31
32
33

use eyre::bail;
use windows::Win32::Foundation::HANDLE;
use windows::Win32::Storage::FileSystem::FILE_BEGIN;
use windows::Win32::Storage::FileSystem::ReadFile;
use windows::Win32::Storage::FileSystem::SetFilePointerEx;

pub trait HandleReadExt {
    fn try_read_exact(&self, offset: i64, buf: &mut [u8]) -> eyre::Result<()>;
}
impl<T: AsRef<HANDLE>> HandleReadExt for T {
    fn try_read_exact(&self, offset: i64, buf: &mut [u8]) -> eyre::Result<()> {
        let handle = *self.as_ref();

        // Seek
        unsafe {
            SetFilePointerEx(handle, offset, None, FILE_BEGIN)?;
        }

        // Read
        let mut bytes_read = 0;
        unsafe {
            ReadFile(handle, Some(buf), Some(&mut bytes_read), None)?;
        }
        if bytes_read != buf.len() as u32 {
            bail!(
                "Failed to read from handle, expected to read {} bytes, got {}",
                buf.len(),
                bytes_read
            );
        }
        Ok(())
    }
}