extern "C" { pub static stdout: *mut libc::FILE; } use getset::{CopyGetters, Getters}; use std::path::PathBuf; #[derive(Getters, CopyGetters)] pub struct FileSystem { /// External UUID of the bcachefs #[getset(get = "pub")] uuid: uuid::Uuid, /// Whether filesystem is encrypted #[getset(get_copy = "pub")] encrypted: bool, /// Super block #[getset(get = "pub")] sb: bcachefs::bch_sb_handle, /// Member devices for this filesystem #[getset(get = "pub")] devices: Vec, } impl std::fmt::Debug for FileSystem { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { f.debug_struct("FileSystem") .field("uuid", &self.uuid) .field("encrypted", &self.encrypted) .field("devices", &self.device_string()) .finish() } } use std::fmt; impl std::fmt::Display for FileSystem { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { let devs = self.device_string(); write!( f, "{:?}: locked?={lock} ({}) ", self.uuid, devs, lock = self.encrypted ) } } impl FileSystem { pub(crate) fn new(sb: bcachefs::bch_sb_handle) -> Self { Self { uuid: sb.sb().uuid(), encrypted: sb.sb().crypt().is_some(), sb: sb, devices: Vec::new(), } } pub fn device_string(&self) -> String { use itertools::Itertools; self.devices.iter().map(|d| d.display()).join(":") } pub fn mount( &self, target: impl AsRef, options: impl AsRef, ) -> anyhow::Result<()> { tracing::info_span!("mount").in_scope(|| { let src = self.device_string(); let (data, mountflags) = parse_mount_options(options); // let fstype = c_str!("bcachefs"); tracing::info!(msg="mounting bcachefs filesystem", target=%target.as_ref().display()); mount_inner(src, target, "bcachefs", mountflags, data) }) } } fn mount_inner( src: String, target: impl AsRef, fstype: &str, mountflags: u64, data: Option, ) -> anyhow::Result<()> { use std::{ ffi::{c_void, CString}, os::{raw::c_char, unix::ffi::OsStrExt}, }; // bind the CStrings to keep them alive let src = CString::new(src)?; let target = CString::new(target.as_ref().as_os_str().as_bytes())?; let data = data.map(CString::new).transpose()?; let fstype = CString::new(fstype)?; // convert to pointers for ffi let src = src.as_c_str().to_bytes_with_nul().as_ptr() as *const c_char; let target = target.as_c_str().to_bytes_with_nul().as_ptr() as *const c_char; let data = data.as_ref().map_or(std::ptr::null(), |data| { data.as_c_str().to_bytes_with_nul().as_ptr() as *const c_void }); let fstype = fstype.as_c_str().to_bytes_with_nul().as_ptr() as *const c_char; let ret = {let _entered = tracing::info_span!("libc::mount").entered(); tracing::info!("mounting filesystem"); // REQUIRES: CAP_SYS_ADMIN unsafe { libc::mount(src, target, fstype, mountflags, data) } }; match ret { 0 => Ok(()), _ => Err(crate::ErrnoError(errno::errno()).into()), } } /// Parse a comma-separated mount options and split out mountflags and filesystem /// specific options. #[tracing_attributes::instrument(skip(options))] fn parse_mount_options(options: impl AsRef) -> (Option, u64) { use either::Either::*; tracing::debug!(msg="parsing mount options", options=?options.as_ref()); let (opts, flags) = options .as_ref() .split(",") .map(|o| match o { "dirsync" => Left(libc::MS_DIRSYNC), "lazytime" => Left(1 << 25), // MS_LAZYTIME "mand" => Left(libc::MS_MANDLOCK), "noatime" => Left(libc::MS_NOATIME), "nodev" => Left(libc::MS_NODEV), "nodiratime" => Left(libc::MS_NODIRATIME), "noexec" => Left(libc::MS_NOEXEC), "nosuid" => Left(libc::MS_NOSUID), "ro" => Left(libc::MS_RDONLY), "rw" => Left(0), "relatime" => Left(libc::MS_RELATIME), "strictatime" => Left(libc::MS_STRICTATIME), "sync" => Left(libc::MS_SYNCHRONOUS), "" => Left(0), o @ _ => Right(o), }) .fold((Vec::new(), 0), |(mut opts, flags), next| match next { Left(f) => (opts, flags | f), Right(o) => { opts.push(o); (opts, flags) } }); use itertools::Itertools; ( if opts.len() == 0 { None } else { Some(opts.iter().join(",")) }, flags, ) } use bch_bindgen::bcachefs; use std::collections::HashMap; use uuid::Uuid; #[tracing_attributes::instrument] pub fn probe_filesystems() -> anyhow::Result> { tracing::trace!("enumerating udev devices"); let mut udev = udev::Enumerator::new()?; udev.match_subsystem("block")?; // find kernel block devices let mut fs_map = HashMap::new(); let devresults = udev.scan_devices()? .into_iter() .filter_map(|dev| dev.devnode().map(ToOwned::to_owned)); for pathbuf in devresults { match get_super_block_uuid(&pathbuf)? { Ok((uuid_key, superblock)) => { let fs = fs_map.entry(uuid_key).or_insert_with(|| { tracing::info!(msg="found bcachefs pool", uuid=?uuid_key); FileSystem::new(superblock) }); fs.devices.push(pathbuf); }, Err(e) => { tracing::debug!(inner2_error=?e);} } } tracing::info!(msg = "found filesystems", count = fs_map.len()); Ok(fs_map) } // #[tracing_attributes::instrument(skip(dev, fs_map))] fn get_super_block_uuid(path: &std::path::Path) -> std::io::Result> { let sb = bch_bindgen::rs::read_super(&path)?; let super_block = match sb { Err(e) => { return Ok(Err(e)); } Ok(sb) => sb, }; let uuid = (&super_block).sb().uuid(); tracing::debug!(found="bcachefs superblock", devnode=?path, ?uuid); Ok(Ok((uuid, super_block))) }