2025-04-24 20:07:06 +08:00

264 lines
9.5 KiB
Rust

// SPDX-License-Identifier: MPL-2.0
use super::SyscallReturn;
use crate::{
fs::{
exfat::{ExfatFS, ExfatMountOptions},
ext2::Ext2,
fs_resolver::{FsPath, AT_FDCWD},
overlayfs::OverlayFS,
path::Dentry,
utils::{FileSystem, InodeType},
},
prelude::*,
syscall::constants::MAX_FILENAME_LEN,
};
/// The `data` argument is interpreted by the different filesystems.
/// Typically it is a string of comma-separated options understood by
/// this filesystem. The current implementation only considers the case
/// where it is `NULL`. Because it should be interpreted by the specific filesystems.
pub fn sys_mount(
devname_addr: Vaddr,
dirname_addr: Vaddr,
fstype_addr: Vaddr,
flags: u64,
data: Vaddr,
ctx: &Context,
) -> Result<SyscallReturn> {
let user_space = ctx.user_space();
let devname = user_space.read_cstring(devname_addr, MAX_FILENAME_LEN)?;
let dirname = user_space.read_cstring(dirname_addr, MAX_FILENAME_LEN)?;
let mount_flags = MountFlags::from_bits_truncate(flags as u32);
debug!(
"devname = {:?}, dirname = {:?}, fstype = 0x{:x}, flags = {:?}, data = 0x{:x}",
devname, dirname, fstype_addr, mount_flags, data,
);
let dst_dentry = {
let dirname = dirname.to_string_lossy();
if dirname.is_empty() {
return_errno_with_message!(Errno::ENOENT, "dirname is empty");
}
let fs_path = FsPath::new(AT_FDCWD, dirname.as_ref())?;
ctx.posix_thread.fs().resolver().read().lookup(&fs_path)?
};
if mount_flags.contains(MountFlags::MS_REMOUNT) && mount_flags.contains(MountFlags::MS_BIND) {
do_reconfigure_mnt()?;
} else if mount_flags.contains(MountFlags::MS_REMOUNT) {
do_remount()?;
} else if mount_flags.contains(MountFlags::MS_BIND) {
do_bind_mount(
devname,
dst_dentry,
mount_flags.contains(MountFlags::MS_REC),
ctx,
)?;
} else if mount_flags.contains(MountFlags::MS_SHARED)
| mount_flags.contains(MountFlags::MS_PRIVATE)
| mount_flags.contains(MountFlags::MS_SLAVE)
| mount_flags.contains(MountFlags::MS_UNBINDABLE)
{
do_change_type()?;
} else if mount_flags.contains(MountFlags::MS_MOVE) {
do_move_mount_old(devname, dst_dentry, ctx)?;
} else {
do_new_mount(devname, fstype_addr, dst_dentry, data, ctx)?;
}
Ok(SyscallReturn::Return(0))
}
fn do_reconfigure_mnt() -> Result<()> {
return_errno_with_message!(Errno::EINVAL, "do_reconfigure_mnt is not supported");
}
fn do_remount() -> Result<()> {
return_errno_with_message!(Errno::EINVAL, "do_remount is not supported");
}
/// Bind a mount to a dst location.
///
/// If recursive is true, then bind the mount recursively.
/// Such as use user command `mount --rbind src dst`.
fn do_bind_mount(
src_name: CString,
dst_dentry: Dentry,
recursive: bool,
ctx: &Context,
) -> Result<()> {
let src_dentry = {
let src_name = src_name.to_string_lossy();
if src_name.is_empty() {
return_errno_with_message!(Errno::ENOENT, "src_name is empty");
}
let fs_path = FsPath::new(AT_FDCWD, src_name.as_ref())?;
ctx.posix_thread.fs().resolver().read().lookup(&fs_path)?
};
if src_dentry.type_() != InodeType::Dir {
return_errno_with_message!(Errno::ENOTDIR, "src_name must be directory");
};
src_dentry.bind_mount_to(&dst_dentry, recursive)?;
Ok(())
}
fn do_change_type() -> Result<()> {
return_errno_with_message!(Errno::EINVAL, "do_change_type is not supported");
}
/// Move a mount from src location to dst location.
fn do_move_mount_old(src_name: CString, dst_dentry: Dentry, ctx: &Context) -> Result<()> {
let src_dentry = {
let src_name = src_name.to_string_lossy();
if src_name.is_empty() {
return_errno_with_message!(Errno::ENOENT, "src_name is empty");
}
let fs_path = FsPath::new(AT_FDCWD, src_name.as_ref())?;
ctx.posix_thread.fs().resolver().read().lookup(&fs_path)?
};
if !src_dentry.is_root_of_mount() {
return_errno_with_message!(Errno::EINVAL, "src_name can not be moved");
};
if src_dentry.mount_node().parent().is_none() {
return_errno_with_message!(Errno::EINVAL, "src_name can not be moved");
}
src_dentry.mount_node().graft_mount_node_tree(&dst_dentry)?;
Ok(())
}
/// Mount a new filesystem.
fn do_new_mount(
devname: CString,
fs_type: Vaddr,
target_dentry: Dentry,
data: Vaddr,
ctx: &Context,
) -> Result<()> {
if target_dentry.type_() != InodeType::Dir {
return_errno_with_message!(Errno::ENOTDIR, "mountpoint must be directory");
};
let fs_type = ctx.user_space().read_cstring(fs_type, MAX_FILENAME_LEN)?;
if fs_type.is_empty() {
return_errno_with_message!(Errno::EINVAL, "fs_type is empty");
}
let fs = get_fs(fs_type, devname, data, ctx)?;
target_dentry.mount(fs)?;
Ok(())
}
/// Get the filesystem by fs_type and devname.
fn get_fs(
fs_type: CString,
devname: CString,
data: Vaddr,
ctx: &Context,
) -> Result<Arc<dyn FileSystem>> {
let user_space = ctx.user_space();
let data = user_space.read_cstring(data, MAX_FILENAME_LEN)?;
let data = data.to_string_lossy();
let fs_type = fs_type.to_str().unwrap();
match fs_type {
"ext2" => {
let device = aster_block::get_device(devname.to_str().unwrap()).ok_or(
Error::with_message(Errno::ENOENT, "device for ext2 does not exist"),
)?;
let ext2_fs = Ext2::open(device)?;
Ok(ext2_fs)
}
"exfat" => {
let device = aster_block::get_device(devname.to_str().unwrap()).ok_or(
Error::with_message(Errno::ENOENT, "device for exfat does not exist"),
)?;
let exfat_fs = ExfatFS::open(device, ExfatMountOptions::default())?;
Ok(exfat_fs)
}
"overlay" => {
let overlay_fs = create_overlayfs(data.as_ref(), ctx)?;
Ok(overlay_fs)
}
_ => return_errno_with_message!(Errno::EINVAL, "Invalid fs type"),
}
}
// TODO: Support read-only mount (no upper) and customized features
fn create_overlayfs(data: &str, ctx: &Context) -> Result<Arc<OverlayFS>> {
let mut lower = Vec::new();
let mut upper = "";
let mut work = "";
for entry in data.split(',') {
let mut parts = entry.split('=');
match (parts.next(), parts.next()) {
// Handle lowerdir, split by ':'
(Some("upperdir"), Some(path)) => {
if path.is_empty() {
return_errno_with_message!(Errno::ENOENT, "upperdir is empty");
}
upper = path;
}
(Some("lowerdir"), Some(paths)) => {
for path in paths.split(':') {
if path.is_empty() {
return_errno_with_message!(Errno::ENOENT, "lowerdir is empty");
}
lower.push(path);
}
}
(Some("workdir"), Some(path)) => {
if path.is_empty() {
return_errno_with_message!(Errno::ENOENT, "workdir is empty");
}
work = path;
}
_ => (),
}
}
let fs = ctx.posix_thread.fs().resolver().read();
let upper = fs.lookup(&FsPath::new(AT_FDCWD, upper)?)?;
let lower = lower
.iter()
.map(|lower| fs.lookup(&FsPath::new(AT_FDCWD, lower).unwrap()).unwrap())
.collect();
let work = fs.lookup(&FsPath::new(AT_FDCWD, work)?)?;
let overlayfs = OverlayFS::new(upper, lower, work)?;
Ok(overlayfs)
}
bitflags! {
struct MountFlags: u32 {
const MS_RDONLY = 1 << 0; // Mount read-only.
const MS_NOSUID = 1 << 1; // Ignore suid and sgid bits.
const MS_NODEV = 1 << 2; // Disallow access to device special files.
const MS_NOEXEC = 1 << 3; // Disallow program execution.
const MS_SYNCHRONOUS = 1 << 4; // Writes are synced at once.
const MS_REMOUNT = 1 << 5; // Alter flags of a mounted FS.
const MS_MANDLOCK = 1 << 6; // Allow mandatory locks on an FS.
const MS_DIRSYNC = 1 << 7; // Directory modifications are synchronous.
const MS_NOSYMFOLLOW = 1 << 8; // Do not follow symlinks.
const MS_NOATIME = 1 << 10; // Do not update access times.
const MS_NODIRATIME = 1 << 11; // Do not update directory access times.
const MS_BIND = 1 << 12; // Bind directory at different place.
const MS_MOVE = 1 << 13; // Move mount from old to new.
const MS_REC = 1 << 14; // Create recursive mount.
const MS_SILENT = 1 << 15; // Suppress certain messages in kernel log.
const MS_POSIXACL = 1 << 16; // VFS does not apply the umask.
const MS_UNBINDABLE = 1 << 17; // Change to unbindable.
const MS_PRIVATE = 1 << 18; // Change to private.
const MS_SLAVE = 1 << 19; // Change to slave.
const MS_SHARED = 1 << 20; // Change to shared.
const MS_RELATIME = 1 << 21; // Update atime relative to mtime/ctime.
const MS_KERNMOUNT = 1 << 22; // This is a kern_mount call.
}
}