heim-disk 0.0.11

Cross-platform disk information
Documentation
use std::mem;
use std::ptr;

use heim_common::prelude::*;

#[allow(unused)]
pub const MNT_WAIT: libc::c_int = 1;
pub const MNT_NOWAIT: libc::c_int = 2;

extern "C" {
    fn getfsstat64(buf: *mut libc::statfs, bufsize: libc::c_int, flags: libc::c_int)
        -> libc::c_int;
}

// TODO: Since `MNT_NOWAIT` might return inconsistent data (see `getfsstat(2)`)
// it might be a good idea (maybe?) to wrap it into a `blocking` call
// and switch to the `MNT_WAIT` mode?
// Should be considered later.
pub fn mounts() -> Result<Vec<libc::statfs>> {
    let expected_len = unsafe { getfsstat64(ptr::null_mut(), 0, MNT_NOWAIT) };
    let mut mounts: Vec<libc::statfs> = Vec::with_capacity(expected_len as usize);
    let result = unsafe {
        getfsstat64(
            mounts.as_mut_ptr(),
            mem::size_of::<libc::statfs>() as libc::c_int * expected_len,
            MNT_NOWAIT,
        )
    };
    if result == -1 {
        return Err(Error::last_os_error());
    } else {
        debug_assert!(
            expected_len == result,
            "Expected {} statfs entries, but got {}",
            expected_len,
            result
        );
        unsafe {
            mounts.set_len(result as usize);
        }
    }

    Ok(mounts)
}