Skip to content

Commit

Permalink
windows impl
Browse files Browse the repository at this point in the history
  • Loading branch information
@dlowe
dlowe committed Nov 26, 2024
1 parent 85194b4 commit 1d07bd3
Showing 1 changed file with 74 additions and 52 deletions.
126 changes: 74 additions & 52 deletions src/windows/disk.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,25 +167,32 @@ impl DiskInner {
self.is_read_only
}

pub(crate) fn refresh_specifics(&mut self, _refreshes: DiskRefreshKind) -> bool {
let Some((read_bytes, written_bytes)) = get_disk_io(&self.device_path, None) else {
sysinfo_debug!("Failed to update disk i/o stats");
return false;
};
pub(crate) fn refresh_specifics(&mut self, refreshes: DiskRefreshKind) -> bool {
if refreshes.io_usage() {
let Some((read_bytes, written_bytes)) = get_disk_io(&self.device_path, None) else {
sysinfo_debug!("Failed to update disk i/o stats");
return false;
};

self.old_read_bytes = self.read_bytes;
self.old_written_bytes = self.written_bytes;
self.read_bytes = read_bytes;
self.written_bytes = written_bytes;

if self.total_space != 0 {
unsafe {
let mut tmp = 0;
let lpdirectoryname = PCWSTR::from_raw(self.mount_point.as_ptr());
if GetDiskFreeSpaceExW(lpdirectoryname, None, None, Some(&mut tmp)).is_ok() {
self.available_space = tmp;
return true;
}
self.old_read_bytes = self.read_bytes;
self.old_written_bytes = self.written_bytes;
self.read_bytes = read_bytes;
self.written_bytes = written_bytes;
}

if refreshes.details() {
if self.total_space != 0 {
let available_space = unsafe {
let mut tmp = 0;
let lpdirectoryname = PCWSTR::from_raw(self.mount_point.as_ptr());
if GetDiskFreeSpaceExW(lpdirectoryname, None, None, Some(&mut tmp)).is_ok() {
tmp
} else {
0
}
};

self.available_space = available_space;
}
}
false
Expand Down Expand Up @@ -220,9 +227,9 @@ impl DisksInner {
self.disks
}

pub(crate) fn refresh_list_specifics(&mut self, _refreshes: DiskRefreshKind) {
pub(crate) fn refresh_list_specifics(&mut self, refreshes: DiskRefreshKind) {
unsafe {
self.disks = get_list();
self.disks = get_list(refreshes);
}
}

Expand Down Expand Up @@ -259,7 +266,7 @@ unsafe fn get_drive_size(mount_point: &[u16]) -> Option<(u64, u64)> {
}
}

pub(crate) unsafe fn get_list() -> Vec<Disk> {
pub(crate) unsafe fn get_list(refreshes: DiskRefreshKind) -> Vec<Disk> {
#[cfg(feature = "multithread")]
use rayon::iter::ParallelIterator;

Expand All @@ -268,7 +275,11 @@ pub(crate) unsafe fn get_list() -> Vec<Disk> {
let raw_volume_name = PCWSTR::from_raw(volume_name.as_ptr());
let drive_type = GetDriveTypeW(raw_volume_name);

let is_removable = drive_type == DRIVE_REMOVABLE;
let is_removable = if refreshes.details() {
drive_type == DRIVE_REMOVABLE
} else {
false
};

if drive_type != DRIVE_FIXED && drive_type != DRIVE_REMOVABLE {
return Vec::new();
Expand All @@ -292,7 +303,11 @@ pub(crate) unsafe fn get_list() -> Vec<Disk> {
);
return Vec::new();
}
let is_read_only = (flags & FILE_READ_ONLY_VOLUME) != 0;
let is_read_only = if refreshes.details() {
(flags & FILE_READ_ONLY_VOLUME) != 0
} else {
false
};

let mount_paths = get_volume_path_names_for_volume_name(&volume_name[..]);
if mount_paths.is_empty() {
Expand All @@ -309,47 +324,54 @@ pub(crate) unsafe fn get_list() -> Vec<Disk> {
else {
return Vec::new();
};
let Some((total_space, available_space)) = get_drive_size(&mount_paths[0][..]) else {
return Vec::new();

let (total_space, available_space) = if refreshes.details() {
get_drive_size(&mount_paths[0][..]).unwrap_or_default()
} else {
(0, 0)
};
if total_space == 0 {
sysinfo_debug!("total_space == 0");
return Vec::new();
}
let spq_trim = STORAGE_PROPERTY_QUERY {
PropertyId: StorageDeviceSeekPenaltyProperty,
QueryType: PropertyStandardQuery,
AdditionalParameters: [0],
};
let mut result: DEVICE_SEEK_PENALTY_DESCRIPTOR = std::mem::zeroed();

let mut dw_size = 0;
let device_io_control = DeviceIoControl(
handle.0,
IOCTL_STORAGE_QUERY_PROPERTY,
Some(&spq_trim as *const STORAGE_PROPERTY_QUERY as *const c_void),
size_of::<STORAGE_PROPERTY_QUERY>() as u32,
Some(&mut result as *mut DEVICE_SEEK_PENALTY_DESCRIPTOR as *mut c_void),
size_of::<DEVICE_SEEK_PENALTY_DESCRIPTOR>() as u32,
Some(&mut dw_size),
None,
)
.is_ok();
let type_ = if !device_io_control
|| dw_size != size_of::<DEVICE_SEEK_PENALTY_DESCRIPTOR>() as u32
{
DiskKind::Unknown(-1)
} else {
let is_hdd = result.IncursSeekPenalty.as_bool();
if is_hdd {
DiskKind::HDD
let type_ = if refreshes.kind() {
let mut dw_size = 0;
let device_io_control = DeviceIoControl(
handle.0,
IOCTL_STORAGE_QUERY_PROPERTY,
Some(&spq_trim as *const STORAGE_PROPERTY_QUERY as *const c_void),
size_of::<STORAGE_PROPERTY_QUERY>() as u32,
Some(&mut result as *mut DEVICE_SEEK_PENALTY_DESCRIPTOR as *mut c_void),
size_of::<DEVICE_SEEK_PENALTY_DESCRIPTOR>() as u32,
Some(&mut dw_size),
None,
)
.is_ok();

if !device_io_control
|| dw_size != size_of::<DEVICE_SEEK_PENALTY_DESCRIPTOR>() as u32
{
DiskKind::Unknown(-1)
} else {
DiskKind::SSD
let is_hdd = result.IncursSeekPenalty.as_bool();
if is_hdd {
DiskKind::HDD
} else {
DiskKind::SSD
}
}
} else {
DiskKind::Unknown(-1)
};

let (read_bytes, written_bytes) =
get_disk_io(&device_path, Some(handle)).unwrap_or_default();
let (read_bytes, written_bytes) = if refreshes.io_usage() {
get_disk_io(&device_path, Some(handle)).unwrap_or_default()
} else {
(0, 0)
};

let name = os_string_from_zero_terminated(&name);
let file_system = os_string_from_zero_terminated(&file_system);
Expand Down

0 comments on commit 1d07bd3

Please sign in to comment.