mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-09 22:59:59 +00:00
houseme
112 lines
3.5 KiB
Rust
112 lines
3.5 KiB
Rust
// Copyright 2024 RustFS Team
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use crate::os::{DiskInfo, IOStats};
|
|
use rustix::fs::{StatVfs, statvfs};
|
|
use std::io::Error;
|
|
use std::path::Path;
|
|
|
|
/// Returns total and free bytes available in a directory, e.g. `/`.
|
|
pub fn get_info(p: impl AsRef<Path>) -> std::io::Result<DiskInfo> {
|
|
let path_display = p.as_ref().display();
|
|
let stat = statvfs(p.as_ref())?;
|
|
|
|
// According to POSIX statvfs definition:
|
|
// f_bsize: File system block size.
|
|
// f_frsize: Fundamental file system block size.
|
|
// f_blocks: Total number of blocks on file system in units of f_frsize.
|
|
//
|
|
// We should use f_frsize to calculate the size in bytes.
|
|
// If f_frsize is 0 (which shouldn't happen on compliant systems), fallback to f_bsize.
|
|
let bsize = if stat.f_frsize > 0 {
|
|
stat.f_frsize as u64
|
|
} else {
|
|
stat.f_bsize as u64
|
|
};
|
|
|
|
let bfree = stat.f_bfree as u64;
|
|
let bavail = stat.f_bavail as u64;
|
|
let blocks = stat.f_blocks as u64;
|
|
|
|
let reserved = match bfree.checked_sub(bavail) {
|
|
Some(reserved) => reserved,
|
|
None => {
|
|
return Err(Error::other(format!(
|
|
"detected f_bavail space ({bavail}) > f_bfree space ({bfree}), fs corruption at ({path_display}). please run 'fsck'",
|
|
)));
|
|
}
|
|
};
|
|
|
|
let total = match blocks.checked_sub(reserved) {
|
|
Some(total) => total * bsize,
|
|
None => {
|
|
return Err(Error::other(format!(
|
|
"detected reserved space ({reserved}) > blocks space ({blocks}), fs corruption at ({path_display}). please run 'fsck'",
|
|
)));
|
|
}
|
|
};
|
|
|
|
let free = bavail * bsize;
|
|
let used = match total.checked_sub(free) {
|
|
Some(used) => used,
|
|
None => {
|
|
return Err(Error::other(format!(
|
|
"detected free space ({free}) > total drive space ({total}), fs corruption at ({path_display}). please run 'fsck'"
|
|
)));
|
|
}
|
|
};
|
|
|
|
let st = rustix::fs::stat(p.as_ref())?;
|
|
|
|
Ok(DiskInfo {
|
|
total,
|
|
free,
|
|
used,
|
|
files: stat.f_files,
|
|
ffree: stat.f_ffree,
|
|
fstype: get_fs_type(&stat).to_string(),
|
|
major: rustix::fs::major(st.st_dev) as u64,
|
|
minor: rustix::fs::minor(st.st_dev) as u64,
|
|
..Default::default()
|
|
})
|
|
}
|
|
|
|
fn get_fs_type(_stat: &StatVfs) -> &'static str {
|
|
"UNKNOWN"
|
|
}
|
|
|
|
pub fn same_disk(disk1: &str, disk2: &str) -> std::io::Result<bool> {
|
|
let stat1 = rustix::fs::stat(disk1)?;
|
|
let stat2 = rustix::fs::stat(disk2)?;
|
|
|
|
Ok(stat1.st_dev == stat2.st_dev)
|
|
}
|
|
|
|
pub fn get_physical_device_ids(disk: &str) -> std::io::Result<Vec<String>> {
|
|
let stat = rustix::fs::stat(disk)?;
|
|
let major = rustix::fs::major(stat.st_dev);
|
|
let minor = rustix::fs::minor(stat.st_dev);
|
|
|
|
Ok(vec![format!("{major}:{minor}")])
|
|
}
|
|
|
|
pub fn check_cross_device_mounts(_paths: &[String]) -> std::io::Result<()> {
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(not(target_os = "linux"))]
|
|
pub fn get_drive_stats(_major: u32, _minor: u32) -> std::io::Result<IOStats> {
|
|
Ok(IOStats::default())
|
|
}
|