1use crate::raw::{
11 BTRFS_SCRUB_READONLY, btrfs_ioc_scrub, btrfs_ioc_scrub_cancel, btrfs_ioc_scrub_progress,
12 btrfs_ioctl_scrub_args,
13};
14use std::{
15 mem,
16 os::{fd::AsRawFd, unix::io::BorrowedFd},
17};
18
19#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
22pub struct ScrubProgress {
23 pub data_extents_scrubbed: u64,
25 pub tree_extents_scrubbed: u64,
27 pub data_bytes_scrubbed: u64,
29 pub tree_bytes_scrubbed: u64,
31 pub read_errors: u64,
33 pub csum_errors: u64,
35 pub verify_errors: u64,
37 pub no_csum: u64,
39 pub csum_discards: u64,
41 pub super_errors: u64,
43 pub malloc_errors: u64,
45 pub uncorrectable_errors: u64,
47 pub corrected_errors: u64,
49 pub last_physical: u64,
51 pub unverified_errors: u64,
53}
54
55impl ScrubProgress {
56 pub fn error_count(&self) -> u64 {
58 self.read_errors + self.super_errors + self.verify_errors + self.csum_errors
59 }
60
61 pub fn bytes_scrubbed(&self) -> u64 {
63 self.data_bytes_scrubbed + self.tree_bytes_scrubbed
64 }
65
66 pub fn is_clean(&self) -> bool {
68 self.error_count() == 0 && self.corrected_errors == 0 && self.uncorrectable_errors == 0
69 }
70}
71
72fn from_raw(raw: &btrfs_ioctl_scrub_args) -> ScrubProgress {
73 let p = &raw.progress;
74 ScrubProgress {
75 data_extents_scrubbed: p.data_extents_scrubbed,
76 tree_extents_scrubbed: p.tree_extents_scrubbed,
77 data_bytes_scrubbed: p.data_bytes_scrubbed,
78 tree_bytes_scrubbed: p.tree_bytes_scrubbed,
79 read_errors: p.read_errors,
80 csum_errors: p.csum_errors,
81 verify_errors: p.verify_errors,
82 no_csum: p.no_csum,
83 csum_discards: p.csum_discards,
84 super_errors: p.super_errors,
85 malloc_errors: p.malloc_errors,
86 uncorrectable_errors: p.uncorrectable_errors,
87 corrected_errors: p.corrected_errors,
88 last_physical: p.last_physical,
89 unverified_errors: p.unverified_errors,
90 }
91}
92
93pub fn scrub_start(fd: BorrowedFd, devid: u64, readonly: bool) -> nix::Result<ScrubProgress> {
101 let mut args: btrfs_ioctl_scrub_args = unsafe { mem::zeroed() };
102 args.devid = devid;
103 args.start = 0;
104 args.end = u64::MAX;
105 if readonly {
106 args.flags = BTRFS_SCRUB_READONLY as u64;
107 }
108 unsafe { btrfs_ioc_scrub(fd.as_raw_fd(), &mut args) }?;
109 Ok(from_raw(&args))
110}
111
112pub fn scrub_cancel(fd: BorrowedFd) -> nix::Result<()> {
114 unsafe { btrfs_ioc_scrub_cancel(fd.as_raw_fd()) }?;
115 Ok(())
116}
117
118pub fn scrub_progress(fd: BorrowedFd, devid: u64) -> nix::Result<Option<ScrubProgress>> {
123 let mut args: btrfs_ioctl_scrub_args = unsafe { mem::zeroed() };
124 args.devid = devid;
125 args.start = 0;
126 args.end = u64::MAX;
127
128 match unsafe { btrfs_ioc_scrub_progress(fd.as_raw_fd(), &mut args) } {
129 Err(nix::errno::Errno::ENOTCONN) => Ok(None),
130 Err(e) => Err(e),
131 Ok(_) => Ok(Some(from_raw(&args))),
132 }
133}