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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
use super::system::get_all_data;
use utils;
use DiskExt;
use DiskType;
use libc::statvfs;
use std::ffi::{OsStr, OsString};
use std::mem;
use std::os::unix::ffi::OsStrExt;
use std::path::{Path, PathBuf};
fn find_type_for_name(name: &OsStr) -> DiskType {
let mut trimmed: &[u8] = name.as_bytes();
while trimmed.len() > 1
&& trimmed[trimmed.len() - 1] >= b'0'
&& trimmed[trimmed.len() - 1] <= b'9'
{
trimmed = &trimmed[..trimmed.len() - 1]
}
let trimmed: &OsStr = OsStrExt::from_bytes(trimmed);
let path = Path::new("/sys/block/")
.to_owned()
.join(trimmed)
.join("queue/rotational");
let rotational_int = get_all_data(path, 8).unwrap_or_default().trim().parse();
DiskType::from(rotational_int.unwrap_or(-1))
}
macro_rules! cast {
($x:expr) => {
u64::from($x)
};
}
pub fn new(name: &OsStr, mount_point: &Path, file_system: &[u8]) -> Disk {
let mount_point_cpath = utils::to_cpath(mount_point);
let type_ = find_type_for_name(name);
let mut total = 0;
let mut available = 0;
unsafe {
let mut stat: statvfs = mem::zeroed();
if statvfs(mount_point_cpath.as_ptr() as *const _, &mut stat) == 0 {
total = cast!(stat.f_bsize) * cast!(stat.f_blocks);
available = cast!(stat.f_bsize) * cast!(stat.f_bavail);
}
}
Disk {
type_,
name: name.to_owned(),
file_system: file_system.to_owned(),
mount_point: mount_point.to_owned(),
total_space: cast!(total),
available_space: cast!(available),
}
}
pub struct Disk {
type_: DiskType,
name: OsString,
file_system: Vec<u8>,
mount_point: PathBuf,
total_space: u64,
available_space: u64,
}
impl DiskExt for Disk {
fn get_type(&self) -> DiskType {
self.type_
}
fn get_name(&self) -> &OsStr {
&self.name
}
fn get_file_system(&self) -> &[u8] {
&self.file_system
}
fn get_mount_point(&self) -> &Path {
&self.mount_point
}
fn get_total_space(&self) -> u64 {
self.total_space
}
fn get_available_space(&self) -> u64 {
self.available_space
}
fn refresh(&mut self) -> bool {
unsafe {
let mut stat: statvfs = mem::zeroed();
let mount_point_cpath = utils::to_cpath(&self.mount_point);
if statvfs(mount_point_cpath.as_ptr() as *const _, &mut stat) == 0 {
let tmp = cast!(stat.f_bsize) * cast!(stat.f_bavail);
self.available_space = cast!(tmp);
true
} else {
false
}
}
}
}