1*1496ba7bSLoGin use alloc::{ 2*1496ba7bSLoGin string::{String, ToString}, 3*1496ba7bSLoGin sync::Arc, 4*1496ba7bSLoGin vec::Vec, 5*1496ba7bSLoGin }; 6004e86ffSlogin 7004e86ffSlogin use crate::{ 8b087521eSChiichen driver::base::block::SeekFrom, 92b771e32SGou Ngai filesystem::vfs::file::FileDescriptorVec, 10aa0367d6SLoGin include::bindings::bindings::{verify_area, AT_REMOVEDIR, PAGE_4K_SIZE, PROC_MAX_FD_NUM}, 112b771e32SGou Ngai kerror, 12*1496ba7bSLoGin libs::rwlock::RwLockWriteGuard, 13*1496ba7bSLoGin mm::VirtAddr, 14*1496ba7bSLoGin process::ProcessManager, 15ab5c8ca4Slogin syscall::{Syscall, SystemError}, 1667b48188Shoumkh time::TimeSpec, 17004e86ffSlogin }; 18004e86ffSlogin 19004e86ffSlogin use super::{ 20ab5c8ca4Slogin core::{do_mkdir, do_remove_dir, do_unlink_at}, 216d81180bSLoGin fcntl::{FcntlCommand, FD_CLOEXEC}, 22004e86ffSlogin file::{File, FileMode}, 23ab5c8ca4Slogin utils::rsplit_path, 24ab5c8ca4Slogin Dirent, FileType, IndexNode, ROOT_INODE, 25004e86ffSlogin }; 26004e86ffSlogin 27ab5c8ca4Slogin pub const SEEK_SET: u32 = 0; 28ab5c8ca4Slogin pub const SEEK_CUR: u32 = 1; 29ab5c8ca4Slogin pub const SEEK_END: u32 = 2; 30ab5c8ca4Slogin pub const SEEK_MAX: u32 = 3; 31004e86ffSlogin 3267b48188Shoumkh bitflags! { 3367b48188Shoumkh /// 文件类型和权限 3467b48188Shoumkh pub struct ModeType: u32 { 3567b48188Shoumkh /// 掩码 3667b48188Shoumkh const S_IFMT = 0o0_170_000; 3767b48188Shoumkh /// 文件类型 3867b48188Shoumkh const S_IFSOCK = 0o140000; 3967b48188Shoumkh const S_IFLNK = 0o120000; 4067b48188Shoumkh const S_IFREG = 0o100000; 4167b48188Shoumkh const S_IFBLK = 0o060000; 4267b48188Shoumkh const S_IFDIR = 0o040000; 4367b48188Shoumkh const S_IFCHR = 0o020000; 4467b48188Shoumkh const S_IFIFO = 0o010000; 4567b48188Shoumkh 4667b48188Shoumkh const S_ISUID = 0o004000; 4767b48188Shoumkh const S_ISGID = 0o002000; 4867b48188Shoumkh const S_ISVTX = 0o001000; 4967b48188Shoumkh /// 文件用户权限 5067b48188Shoumkh const S_IRWXU = 0o0700; 5167b48188Shoumkh const S_IRUSR = 0o0400; 5267b48188Shoumkh const S_IWUSR = 0o0200; 5367b48188Shoumkh const S_IXUSR = 0o0100; 5467b48188Shoumkh /// 文件组权限 5567b48188Shoumkh const S_IRWXG = 0o0070; 5667b48188Shoumkh const S_IRGRP = 0o0040; 5767b48188Shoumkh const S_IWGRP = 0o0020; 5867b48188Shoumkh const S_IXGRP = 0o0010; 5967b48188Shoumkh /// 文件其他用户权限 6067b48188Shoumkh const S_IRWXO = 0o0007; 6167b48188Shoumkh const S_IROTH = 0o0004; 6267b48188Shoumkh const S_IWOTH = 0o0002; 6367b48188Shoumkh const S_IXOTH = 0o0001; 6467b48188Shoumkh } 6567b48188Shoumkh } 6667b48188Shoumkh 6767b48188Shoumkh #[repr(C)] 6867b48188Shoumkh /// # 文件信息结构体 6967b48188Shoumkh pub struct PosixKstat { 7067b48188Shoumkh /// 硬件设备ID 7167b48188Shoumkh dev_id: u64, 7267b48188Shoumkh /// inode号 7367b48188Shoumkh inode: u64, 7467b48188Shoumkh /// 硬链接数 7567b48188Shoumkh nlink: u64, 7667b48188Shoumkh /// 文件权限 7767b48188Shoumkh mode: ModeType, 7867b48188Shoumkh /// 所有者用户ID 7967b48188Shoumkh uid: i32, 8067b48188Shoumkh /// 所有者组ID 8167b48188Shoumkh gid: i32, 8267b48188Shoumkh /// 设备ID 8367b48188Shoumkh rdev: i64, 8467b48188Shoumkh /// 文件大小 8567b48188Shoumkh size: i64, 8667b48188Shoumkh /// 文件系统块大小 8767b48188Shoumkh blcok_size: i64, 8867b48188Shoumkh /// 分配的512B块数 8967b48188Shoumkh blocks: u64, 9067b48188Shoumkh /// 最后访问时间 9167b48188Shoumkh atime: TimeSpec, 9267b48188Shoumkh /// 最后修改时间 9367b48188Shoumkh mtime: TimeSpec, 9467b48188Shoumkh /// 最后状态变化时间 9567b48188Shoumkh ctime: TimeSpec, 9667b48188Shoumkh /// 用于填充结构体大小的空白数据 9767b48188Shoumkh pub _pad: [i8; 24], 9867b48188Shoumkh } 9967b48188Shoumkh impl PosixKstat { 10067b48188Shoumkh fn new() -> Self { 10167b48188Shoumkh Self { 10267b48188Shoumkh inode: 0, 10367b48188Shoumkh dev_id: 0, 10467b48188Shoumkh mode: ModeType { bits: 0 }, 10567b48188Shoumkh nlink: 0, 10667b48188Shoumkh uid: 0, 10767b48188Shoumkh gid: 0, 10867b48188Shoumkh rdev: 0, 10967b48188Shoumkh size: 0, 11067b48188Shoumkh atime: TimeSpec { 11167b48188Shoumkh tv_sec: 0, 11267b48188Shoumkh tv_nsec: 0, 11367b48188Shoumkh }, 11467b48188Shoumkh mtime: TimeSpec { 11567b48188Shoumkh tv_sec: 0, 11667b48188Shoumkh tv_nsec: 0, 11767b48188Shoumkh }, 11867b48188Shoumkh ctime: TimeSpec { 11967b48188Shoumkh tv_sec: 0, 12067b48188Shoumkh tv_nsec: 0, 12167b48188Shoumkh }, 12267b48188Shoumkh blcok_size: 0, 12367b48188Shoumkh blocks: 0, 12467b48188Shoumkh _pad: Default::default(), 12567b48188Shoumkh } 12667b48188Shoumkh } 12767b48188Shoumkh } 128ab5c8ca4Slogin impl Syscall { 129ab5c8ca4Slogin /// @brief 为当前进程打开一个文件 130ab5c8ca4Slogin /// 131ab5c8ca4Slogin /// @param path 文件路径 132ab5c8ca4Slogin /// @param o_flags 打开文件的标志位 133ab5c8ca4Slogin /// 134ab5c8ca4Slogin /// @return 文件描述符编号,或者是错误码 135ab5c8ca4Slogin pub fn open(path: &str, mode: FileMode) -> Result<usize, SystemError> { 1366d81180bSLoGin // kdebug!("open: path: {}, mode: {:?}", path, mode); 137*1496ba7bSLoGin 138ab5c8ca4Slogin // 文件名过长 139ab5c8ca4Slogin if path.len() > PAGE_4K_SIZE as usize { 140ab5c8ca4Slogin return Err(SystemError::ENAMETOOLONG); 141ab5c8ca4Slogin } 142ab5c8ca4Slogin 143ab5c8ca4Slogin let inode: Result<Arc<dyn IndexNode>, SystemError> = ROOT_INODE().lookup(path); 144ab5c8ca4Slogin 145ab5c8ca4Slogin let inode: Arc<dyn IndexNode> = if inode.is_err() { 146ab5c8ca4Slogin let errno = inode.unwrap_err(); 147ab5c8ca4Slogin // 文件不存在,且需要创建 148ab5c8ca4Slogin if mode.contains(FileMode::O_CREAT) 149ab5c8ca4Slogin && !mode.contains(FileMode::O_DIRECTORY) 150ab5c8ca4Slogin && errno == SystemError::ENOENT 151ab5c8ca4Slogin { 152ab5c8ca4Slogin let (filename, parent_path) = rsplit_path(path); 153ab5c8ca4Slogin // 查找父目录 154ab5c8ca4Slogin let parent_inode: Arc<dyn IndexNode> = 155ab5c8ca4Slogin ROOT_INODE().lookup(parent_path.unwrap_or("/"))?; 156ab5c8ca4Slogin // 创建文件 157ab5c8ca4Slogin let inode: Arc<dyn IndexNode> = 158ab5c8ca4Slogin parent_inode.create(filename, FileType::File, 0o777)?; 159ab5c8ca4Slogin inode 160004e86ffSlogin } else { 161ab5c8ca4Slogin // 不需要创建文件,因此返回错误码 162ab5c8ca4Slogin return Err(errno); 163004e86ffSlogin } 164004e86ffSlogin } else { 165ab5c8ca4Slogin inode.unwrap() 166004e86ffSlogin }; 167004e86ffSlogin 168ab5c8ca4Slogin let file_type: FileType = inode.metadata()?.file_type; 169ab5c8ca4Slogin // 如果要打开的是文件夹,而目标不是文件夹 170ab5c8ca4Slogin if mode.contains(FileMode::O_DIRECTORY) && file_type != FileType::Dir { 171ab5c8ca4Slogin return Err(SystemError::ENOTDIR); 172004e86ffSlogin } 173ab5c8ca4Slogin 174ab5c8ca4Slogin // 如果O_TRUNC,并且,打开模式包含O_RDWR或O_WRONLY,清空文件 175ab5c8ca4Slogin if mode.contains(FileMode::O_TRUNC) 176ab5c8ca4Slogin && (mode.contains(FileMode::O_RDWR) || mode.contains(FileMode::O_WRONLY)) 177ab5c8ca4Slogin && file_type == FileType::File 178ab5c8ca4Slogin { 179ab5c8ca4Slogin inode.truncate(0)?; 180ab5c8ca4Slogin } 181ab5c8ca4Slogin 182ab5c8ca4Slogin // 创建文件对象 183ab5c8ca4Slogin let mut file: File = File::new(inode, mode)?; 184ab5c8ca4Slogin 185ab5c8ca4Slogin // 打开模式为“追加” 186ab5c8ca4Slogin if mode.contains(FileMode::O_APPEND) { 187ab5c8ca4Slogin file.lseek(SeekFrom::SeekEnd(0))?; 188ab5c8ca4Slogin } 189ab5c8ca4Slogin 190ab5c8ca4Slogin // 把文件对象存入pcb 191*1496ba7bSLoGin let r = ProcessManager::current_pcb() 192*1496ba7bSLoGin .fd_table() 193*1496ba7bSLoGin .write() 194*1496ba7bSLoGin .alloc_fd(file, None) 195*1496ba7bSLoGin .map(|fd| fd as usize); 196*1496ba7bSLoGin 1976d81180bSLoGin return r; 198ab5c8ca4Slogin } 199ab5c8ca4Slogin 200ab5c8ca4Slogin /// @brief 关闭文件 201ab5c8ca4Slogin /// 202ab5c8ca4Slogin /// @param fd 文件描述符编号 203ab5c8ca4Slogin /// 204ab5c8ca4Slogin /// @return 成功返回0,失败返回错误码 205ab5c8ca4Slogin pub fn close(fd: usize) -> Result<usize, SystemError> { 206*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 207*1496ba7bSLoGin let mut fd_table_guard = binding.write(); 208*1496ba7bSLoGin 209*1496ba7bSLoGin return fd_table_guard.drop_fd(fd as i32).map(|_| 0); 210ab5c8ca4Slogin } 211ab5c8ca4Slogin 212ab5c8ca4Slogin /// @brief 根据文件描述符,读取文件数据。尝试读取的数据长度与buf的长度相同。 213ab5c8ca4Slogin /// 214ab5c8ca4Slogin /// @param fd 文件描述符编号 215ab5c8ca4Slogin /// @param buf 输出缓冲区。 216ab5c8ca4Slogin /// 217ab5c8ca4Slogin /// @return Ok(usize) 成功读取的数据的字节数 218ab5c8ca4Slogin /// @return Err(SystemError) 读取失败,返回posix错误码 219ab5c8ca4Slogin pub fn read(fd: i32, buf: &mut [u8]) -> Result<usize, SystemError> { 220*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 221*1496ba7bSLoGin let fd_table_guard = binding.read(); 222*1496ba7bSLoGin 223*1496ba7bSLoGin let file = fd_table_guard.get_file_by_fd(fd); 224ab5c8ca4Slogin if file.is_none() { 225ab5c8ca4Slogin return Err(SystemError::EBADF); 226ab5c8ca4Slogin } 227*1496ba7bSLoGin // drop guard 以避免无法调度的问题 228*1496ba7bSLoGin drop(fd_table_guard); 229*1496ba7bSLoGin let file = file.unwrap(); 230ab5c8ca4Slogin 231*1496ba7bSLoGin return file.lock_no_preempt().read(buf.len(), buf); 232ab5c8ca4Slogin } 233ab5c8ca4Slogin 234ab5c8ca4Slogin /// @brief 根据文件描述符,向文件写入数据。尝试写入的数据长度与buf的长度相同。 235ab5c8ca4Slogin /// 236ab5c8ca4Slogin /// @param fd 文件描述符编号 237ab5c8ca4Slogin /// @param buf 输入缓冲区。 238ab5c8ca4Slogin /// 239ab5c8ca4Slogin /// @return Ok(usize) 成功写入的数据的字节数 240ab5c8ca4Slogin /// @return Err(SystemError) 写入失败,返回posix错误码 241ab5c8ca4Slogin pub fn write(fd: i32, buf: &[u8]) -> Result<usize, SystemError> { 242*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 243*1496ba7bSLoGin let fd_table_guard = binding.read(); 244ab5c8ca4Slogin 245*1496ba7bSLoGin let file = fd_table_guard 246*1496ba7bSLoGin .get_file_by_fd(fd) 247*1496ba7bSLoGin .ok_or(SystemError::EBADF)?; 248*1496ba7bSLoGin 249*1496ba7bSLoGin // drop guard 以避免无法调度的问题 250*1496ba7bSLoGin drop(fd_table_guard); 251*1496ba7bSLoGin return file.lock_no_preempt().write(buf.len(), buf); 252ab5c8ca4Slogin } 253ab5c8ca4Slogin 254ab5c8ca4Slogin /// @brief 调整文件操作指针的位置 255ab5c8ca4Slogin /// 256ab5c8ca4Slogin /// @param fd 文件描述符编号 257ab5c8ca4Slogin /// @param seek 调整的方式 258ab5c8ca4Slogin /// 259ab5c8ca4Slogin /// @return Ok(usize) 调整后,文件访问指针相对于文件头部的偏移量 260ab5c8ca4Slogin /// @return Err(SystemError) 调整失败,返回posix错误码 261ab5c8ca4Slogin pub fn lseek(fd: i32, seek: SeekFrom) -> Result<usize, SystemError> { 262*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 263*1496ba7bSLoGin let fd_table_guard = binding.read(); 264*1496ba7bSLoGin let file = fd_table_guard 265*1496ba7bSLoGin .get_file_by_fd(fd) 266*1496ba7bSLoGin .ok_or(SystemError::EBADF)?; 267*1496ba7bSLoGin 268*1496ba7bSLoGin // drop guard 以避免无法调度的问题 269*1496ba7bSLoGin drop(fd_table_guard); 270*1496ba7bSLoGin return file.lock_no_preempt().lseek(seek); 271004e86ffSlogin } 272004e86ffSlogin 273004e86ffSlogin /// @brief 切换工作目录 274004e86ffSlogin /// 275004e86ffSlogin /// @param dest_path 目标路径 276004e86ffSlogin /// 277004e86ffSlogin /// @return 返回码 描述 278004e86ffSlogin /// 0 | 成功 279004e86ffSlogin /// 280004e86ffSlogin /// EACCESS | 权限不足 281004e86ffSlogin /// 282004e86ffSlogin /// ELOOP | 解析path时遇到路径循环 283004e86ffSlogin /// 284004e86ffSlogin /// ENAMETOOLONG | 路径名过长 285004e86ffSlogin /// 286004e86ffSlogin /// ENOENT | 目标文件或目录不存在 287004e86ffSlogin /// 288004e86ffSlogin /// ENODIR | 检索期间发现非目录项 289004e86ffSlogin /// 290004e86ffSlogin /// ENOMEM | 系统内存不足 291004e86ffSlogin /// 292004e86ffSlogin /// EFAULT | 错误的地址 293004e86ffSlogin /// 294004e86ffSlogin /// ENAMETOOLONG | 路径过长 295ab5c8ca4Slogin pub fn chdir(dest_path: &str) -> Result<usize, SystemError> { 296*1496ba7bSLoGin let proc = ProcessManager::current_pcb(); 297ab5c8ca4Slogin // Copy path to kernel space to avoid some security issues 298*1496ba7bSLoGin let path = dest_path.to_string(); 299*1496ba7bSLoGin let mut new_path = String::from(""); 300*1496ba7bSLoGin if path.len() > 0 { 301*1496ba7bSLoGin let cwd = match path.as_bytes()[0] { 302*1496ba7bSLoGin b'/' => String::from("/"), 303*1496ba7bSLoGin _ => proc.basic().cwd(), 304*1496ba7bSLoGin }; 305*1496ba7bSLoGin let mut cwd_vec: Vec<_> = cwd.split("/").filter(|&x| x != "").collect(); 306*1496ba7bSLoGin let path_split = path.split("/").filter(|&x| x != ""); 307*1496ba7bSLoGin for seg in path_split { 308*1496ba7bSLoGin if seg == ".." { 309*1496ba7bSLoGin cwd_vec.pop(); 310*1496ba7bSLoGin } else if seg == "." { 311*1496ba7bSLoGin // 当前目录 312*1496ba7bSLoGin } else { 313*1496ba7bSLoGin cwd_vec.push(seg); 314*1496ba7bSLoGin } 315*1496ba7bSLoGin } 316*1496ba7bSLoGin //proc.basic().set_path(String::from("")); 317*1496ba7bSLoGin for seg in cwd_vec { 318*1496ba7bSLoGin new_path.push_str("/"); 319*1496ba7bSLoGin new_path.push_str(seg); 320*1496ba7bSLoGin } 321*1496ba7bSLoGin if new_path == "" { 322*1496ba7bSLoGin new_path = String::from("/"); 323*1496ba7bSLoGin } 324*1496ba7bSLoGin } 325*1496ba7bSLoGin let inode = match ROOT_INODE().lookup(&new_path) { 326004e86ffSlogin Err(e) => { 327676b8ef6SMork kerror!("Change Directory Failed, Error = {:?}", e); 328ab5c8ca4Slogin return Err(SystemError::ENOENT); 329004e86ffSlogin } 330004e86ffSlogin Ok(i) => i, 331004e86ffSlogin }; 332*1496ba7bSLoGin let metadata = inode.metadata()?; 333*1496ba7bSLoGin if metadata.file_type == FileType::Dir { 334*1496ba7bSLoGin proc.basic_mut().set_cwd(String::from(new_path)); 335ab5c8ca4Slogin return Ok(0); 336004e86ffSlogin } else { 337ab5c8ca4Slogin return Err(SystemError::ENOTDIR); 338004e86ffSlogin } 339004e86ffSlogin } 340*1496ba7bSLoGin 341*1496ba7bSLoGin /// @brief 获取当前进程的工作目录路径 342*1496ba7bSLoGin /// 343*1496ba7bSLoGin /// @param buf 指向缓冲区的指针 344*1496ba7bSLoGin /// @param size 缓冲区的大小 345*1496ba7bSLoGin /// 346*1496ba7bSLoGin /// @return 成功,返回的指针指向包含工作目录路径的字符串 347*1496ba7bSLoGin /// @return 错误,没有足够的空间 348*1496ba7bSLoGin pub fn getcwd(buf: &mut [u8]) -> Result<VirtAddr, SystemError> { 349*1496ba7bSLoGin let proc = ProcessManager::current_pcb(); 350*1496ba7bSLoGin let cwd = proc.basic().cwd(); 351*1496ba7bSLoGin 352*1496ba7bSLoGin let cwd_bytes = cwd.as_bytes(); 353*1496ba7bSLoGin let cwd_len = cwd_bytes.len(); 354*1496ba7bSLoGin if cwd_len + 1 > buf.len() { 355*1496ba7bSLoGin return Err(SystemError::ENOMEM); 356004e86ffSlogin } 357*1496ba7bSLoGin buf[..cwd_len].copy_from_slice(cwd_bytes); 358*1496ba7bSLoGin buf[cwd_len] = 0; 359*1496ba7bSLoGin 360*1496ba7bSLoGin return Ok(VirtAddr::new(buf.as_ptr() as usize)); 361004e86ffSlogin } 362004e86ffSlogin 363004e86ffSlogin /// @brief 获取目录中的数据 364004e86ffSlogin /// 365ab5c8ca4Slogin /// TODO: 这个函数的语义与Linux不一致,需要修改!!! 366ab5c8ca4Slogin /// 367004e86ffSlogin /// @param fd 文件描述符号 368ab5c8ca4Slogin /// @param buf 输出缓冲区 369ab5c8ca4Slogin /// 370ab5c8ca4Slogin /// @return 成功返回读取的字节数,失败返回错误码 371ab5c8ca4Slogin pub fn getdents(fd: i32, buf: &mut [u8]) -> Result<usize, SystemError> { 372ab5c8ca4Slogin let dirent = 373ab5c8ca4Slogin unsafe { (buf.as_mut_ptr() as *mut Dirent).as_mut() }.ok_or(SystemError::EFAULT)?; 374004e86ffSlogin 375004e86ffSlogin if fd < 0 || fd as u32 > PROC_MAX_FD_NUM { 376ab5c8ca4Slogin return Err(SystemError::EBADF); 377004e86ffSlogin } 378004e86ffSlogin 379004e86ffSlogin // 获取fd 380*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 381*1496ba7bSLoGin let fd_table_guard = binding.read(); 382*1496ba7bSLoGin let file = fd_table_guard 383*1496ba7bSLoGin .get_file_by_fd(fd) 384*1496ba7bSLoGin .ok_or(SystemError::EBADF)?; 385004e86ffSlogin 386*1496ba7bSLoGin // drop guard 以避免无法调度的问题 387*1496ba7bSLoGin drop(fd_table_guard); 388*1496ba7bSLoGin return file.lock_no_preempt().readdir(dirent).map(|x| x as usize); 389004e86ffSlogin } 390004e86ffSlogin 391004e86ffSlogin /// @brief 创建文件夹 392004e86ffSlogin /// 393004e86ffSlogin /// @param path(r8) 路径 / mode(r9) 模式 394004e86ffSlogin /// 395004e86ffSlogin /// @return uint64_t 负数错误码 / 0表示成功 396ab5c8ca4Slogin pub fn mkdir(path: &str, mode: usize) -> Result<usize, SystemError> { 397ab5c8ca4Slogin return do_mkdir(path, FileMode::from_bits_truncate(mode as u32)).map(|x| x as usize); 398004e86ffSlogin } 399004e86ffSlogin 400ab5c8ca4Slogin /// **删除文件夹、取消文件的链接、删除文件的系统调用** 401004e86ffSlogin /// 402ab5c8ca4Slogin /// ## 参数 403004e86ffSlogin /// 404ab5c8ca4Slogin /// - `dirfd`:文件夹的文件描述符.目前暂未实现 405ab5c8ca4Slogin /// - `pathname`:文件夹的路径 406ab5c8ca4Slogin /// - `flags`:标志位 407004e86ffSlogin /// 408004e86ffSlogin /// 409ab5c8ca4Slogin pub fn unlinkat(_dirfd: i32, pathname: &str, flags: u32) -> Result<usize, SystemError> { 410004e86ffSlogin // kdebug!("sys_unlink_at={path:?}"); 411ab5c8ca4Slogin if (flags & (!AT_REMOVEDIR)) != 0 { 412ab5c8ca4Slogin return Err(SystemError::EINVAL); 413004e86ffSlogin } 414004e86ffSlogin 415ab5c8ca4Slogin if (flags & AT_REMOVEDIR) > 0 { 416004e86ffSlogin // kdebug!("rmdir"); 417ab5c8ca4Slogin match do_remove_dir(&pathname) { 418004e86ffSlogin Err(err) => { 419676b8ef6SMork kerror!("Failed to Remove Directory, Error Code = {:?}", err); 420ab5c8ca4Slogin return Err(err); 421004e86ffSlogin } 422004e86ffSlogin Ok(_) => { 423ab5c8ca4Slogin return Ok(0); 424004e86ffSlogin } 425004e86ffSlogin } 426004e86ffSlogin } 427004e86ffSlogin 428ab5c8ca4Slogin match do_unlink_at(&pathname, FileMode::from_bits_truncate(flags as u32)) { 429004e86ffSlogin Err(err) => { 430676b8ef6SMork kerror!("Failed to Remove Directory, Error Code = {:?}", err); 431ab5c8ca4Slogin return Err(err); 432004e86ffSlogin } 433004e86ffSlogin Ok(_) => { 434ab5c8ca4Slogin return Ok(0); 435004e86ffSlogin } 436004e86ffSlogin } 437004e86ffSlogin } 4382b771e32SGou Ngai 439ab5c8ca4Slogin /// @brief 根据提供的文件描述符的fd,复制对应的文件结构体,并返回新复制的文件结构体对应的fd 440ab5c8ca4Slogin pub fn dup(oldfd: i32) -> Result<usize, SystemError> { 441*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 442*1496ba7bSLoGin let mut fd_table_guard = binding.write(); 443ab5c8ca4Slogin 444*1496ba7bSLoGin let old_file = fd_table_guard 445*1496ba7bSLoGin .get_file_by_fd(oldfd) 446*1496ba7bSLoGin .ok_or(SystemError::EBADF)?; 447*1496ba7bSLoGin 448*1496ba7bSLoGin let new_file = old_file 449*1496ba7bSLoGin .lock_no_preempt() 450*1496ba7bSLoGin .try_clone() 451*1496ba7bSLoGin .ok_or(SystemError::EBADF)?; 4522b771e32SGou Ngai // 申请文件描述符,并把文件对象存入其中 453*1496ba7bSLoGin let res = fd_table_guard.alloc_fd(new_file, None).map(|x| x as usize); 4542b771e32SGou Ngai return res; 4552b771e32SGou Ngai } 4562b771e32SGou Ngai 457ab5c8ca4Slogin /// 根据提供的文件描述符的fd,和指定新fd,复制对应的文件结构体, 458ab5c8ca4Slogin /// 并返回新复制的文件结构体对应的fd. 459ab5c8ca4Slogin /// 如果新fd已经打开,则会先关闭新fd. 460ab5c8ca4Slogin /// 461ab5c8ca4Slogin /// ## 参数 462ab5c8ca4Slogin /// 463ab5c8ca4Slogin /// - `oldfd`:旧文件描述符 464ab5c8ca4Slogin /// - `newfd`:新文件描述符 465ab5c8ca4Slogin /// 466ab5c8ca4Slogin /// ## 返回值 467ab5c8ca4Slogin /// 468ab5c8ca4Slogin /// - 成功:新文件描述符 469ab5c8ca4Slogin /// - 失败:错误码 470ab5c8ca4Slogin pub fn dup2(oldfd: i32, newfd: i32) -> Result<usize, SystemError> { 471*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 472*1496ba7bSLoGin let mut fd_table_guard = binding.write(); 473*1496ba7bSLoGin return Self::do_dup2(oldfd, newfd, &mut fd_table_guard); 474*1496ba7bSLoGin } 475*1496ba7bSLoGin 476*1496ba7bSLoGin fn do_dup2( 477*1496ba7bSLoGin oldfd: i32, 478*1496ba7bSLoGin newfd: i32, 479*1496ba7bSLoGin fd_table_guard: &mut RwLockWriteGuard<'_, FileDescriptorVec>, 480*1496ba7bSLoGin ) -> Result<usize, SystemError> { 4812b771e32SGou Ngai // 确认oldfd, newid是否有效 482*1496ba7bSLoGin if !(FileDescriptorVec::validate_fd(oldfd) && FileDescriptorVec::validate_fd(newfd)) { 483*1496ba7bSLoGin return Err(SystemError::EBADF); 484*1496ba7bSLoGin } 485*1496ba7bSLoGin 4862b771e32SGou Ngai if oldfd == newfd { 4872b771e32SGou Ngai // 若oldfd与newfd相等 488ab5c8ca4Slogin return Ok(newfd as usize); 4892b771e32SGou Ngai } 490*1496ba7bSLoGin let new_exists = fd_table_guard.get_file_by_fd(newfd).is_some(); 491*1496ba7bSLoGin if new_exists { 4922b771e32SGou Ngai // close newfd 493*1496ba7bSLoGin if let Err(_) = fd_table_guard.drop_fd(newfd) { 4942b771e32SGou Ngai // An I/O error occurred while attempting to close fildes2. 4952b771e32SGou Ngai return Err(SystemError::EIO); 4962b771e32SGou Ngai } 4972b771e32SGou Ngai } 4982b771e32SGou Ngai 499*1496ba7bSLoGin let old_file = fd_table_guard 500*1496ba7bSLoGin .get_file_by_fd(oldfd) 501*1496ba7bSLoGin .ok_or(SystemError::EBADF)?; 502*1496ba7bSLoGin let new_file = old_file 503*1496ba7bSLoGin .lock_no_preempt() 504*1496ba7bSLoGin .try_clone() 505*1496ba7bSLoGin .ok_or(SystemError::EBADF)?; 5062b771e32SGou Ngai // 申请文件描述符,并把文件对象存入其中 507*1496ba7bSLoGin let res = fd_table_guard 508*1496ba7bSLoGin .alloc_fd(new_file, Some(newfd)) 509ab5c8ca4Slogin .map(|x| x as usize); 5102b771e32SGou Ngai return res; 5112b771e32SGou Ngai } 5126d81180bSLoGin 5136d81180bSLoGin /// # fcntl 5146d81180bSLoGin /// 5156d81180bSLoGin /// ## 参数 5166d81180bSLoGin /// 5176d81180bSLoGin /// - `fd`:文件描述符 5186d81180bSLoGin /// - `cmd`:命令 5196d81180bSLoGin /// - `arg`:参数 5206d81180bSLoGin pub fn fcntl(fd: i32, cmd: FcntlCommand, arg: i32) -> Result<usize, SystemError> { 5216d81180bSLoGin match cmd { 5226d81180bSLoGin FcntlCommand::DupFd => { 5236d81180bSLoGin if arg < 0 || arg as usize >= FileDescriptorVec::PROCESS_MAX_FD { 5246d81180bSLoGin return Err(SystemError::EBADF); 5256d81180bSLoGin } 5266d81180bSLoGin let arg = arg as usize; 5276d81180bSLoGin for i in arg..FileDescriptorVec::PROCESS_MAX_FD { 528*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 529*1496ba7bSLoGin let mut fd_table_guard = binding.write(); 530*1496ba7bSLoGin if fd_table_guard.get_file_by_fd(fd).is_none() { 531*1496ba7bSLoGin return Self::do_dup2(fd, i as i32, &mut fd_table_guard); 5326d81180bSLoGin } 5336d81180bSLoGin } 5346d81180bSLoGin return Err(SystemError::EMFILE); 5356d81180bSLoGin } 5366d81180bSLoGin FcntlCommand::GetFd => { 5376d81180bSLoGin // Get file descriptor flags. 538*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 539*1496ba7bSLoGin let fd_table_guard = binding.read(); 540*1496ba7bSLoGin if let Some(file) = fd_table_guard.get_file_by_fd(fd) { 541*1496ba7bSLoGin // drop guard 以避免无法调度的问题 542*1496ba7bSLoGin drop(fd_table_guard); 5436d81180bSLoGin 544*1496ba7bSLoGin if file.lock().close_on_exec() { 5456d81180bSLoGin return Ok(FD_CLOEXEC as usize); 5466d81180bSLoGin } 5476d81180bSLoGin } 5486d81180bSLoGin return Err(SystemError::EBADF); 5496d81180bSLoGin } 5506d81180bSLoGin FcntlCommand::SetFd => { 5516d81180bSLoGin // Set file descriptor flags. 552*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 553*1496ba7bSLoGin let fd_table_guard = binding.write(); 554*1496ba7bSLoGin 555*1496ba7bSLoGin if let Some(file) = fd_table_guard.get_file_by_fd(fd) { 556*1496ba7bSLoGin // drop guard 以避免无法调度的问题 557*1496ba7bSLoGin drop(fd_table_guard); 5586d81180bSLoGin let arg = arg as u32; 5596d81180bSLoGin if arg & FD_CLOEXEC != 0 { 560*1496ba7bSLoGin file.lock().set_close_on_exec(true); 5616d81180bSLoGin } else { 562*1496ba7bSLoGin file.lock().set_close_on_exec(false); 5636d81180bSLoGin } 5646d81180bSLoGin return Ok(0); 5656d81180bSLoGin } 5666d81180bSLoGin return Err(SystemError::EBADF); 5676d81180bSLoGin } 5686d81180bSLoGin 5696d81180bSLoGin FcntlCommand::GetFlags => { 5706d81180bSLoGin // Get file status flags. 571*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 572*1496ba7bSLoGin let fd_table_guard = binding.read(); 573*1496ba7bSLoGin 574*1496ba7bSLoGin if let Some(file) = fd_table_guard.get_file_by_fd(fd) { 575*1496ba7bSLoGin // drop guard 以避免无法调度的问题 576*1496ba7bSLoGin drop(fd_table_guard); 577*1496ba7bSLoGin return Ok(file.lock_no_preempt().mode().bits() as usize); 5786d81180bSLoGin } 579*1496ba7bSLoGin 5806d81180bSLoGin return Err(SystemError::EBADF); 5816d81180bSLoGin } 5826d81180bSLoGin FcntlCommand::SetFlags => { 5836d81180bSLoGin // Set file status flags. 584*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 585*1496ba7bSLoGin let fd_table_guard = binding.write(); 586*1496ba7bSLoGin 587*1496ba7bSLoGin if let Some(file) = fd_table_guard.get_file_by_fd(fd) { 5886d81180bSLoGin let arg = arg as u32; 5896d81180bSLoGin let mode = FileMode::from_bits(arg).ok_or(SystemError::EINVAL)?; 590*1496ba7bSLoGin // drop guard 以避免无法调度的问题 591*1496ba7bSLoGin drop(fd_table_guard); 592*1496ba7bSLoGin file.lock_no_preempt().set_mode(mode)?; 5936d81180bSLoGin return Ok(0); 5946d81180bSLoGin } 595*1496ba7bSLoGin 5966d81180bSLoGin return Err(SystemError::EBADF); 5976d81180bSLoGin } 5986d81180bSLoGin _ => { 5996d81180bSLoGin // TODO: unimplemented 6006d81180bSLoGin // 未实现的命令,返回0,不报错。 6016d81180bSLoGin 6026d81180bSLoGin // kwarn!("fcntl: unimplemented command: {:?}, defaults to 0.", cmd); 6036d81180bSLoGin return Ok(0); 6046d81180bSLoGin } 6056d81180bSLoGin } 6066d81180bSLoGin } 6076d81180bSLoGin 6086d81180bSLoGin /// # ftruncate 6096d81180bSLoGin /// 6106d81180bSLoGin /// ## 描述 6116d81180bSLoGin /// 6126d81180bSLoGin /// 改变文件大小. 6136d81180bSLoGin /// 如果文件大小大于原来的大小,那么文件的内容将会被扩展到指定的大小,新的空间将会用0填充. 6146d81180bSLoGin /// 如果文件大小小于原来的大小,那么文件的内容将会被截断到指定的大小. 6156d81180bSLoGin /// 6166d81180bSLoGin /// ## 参数 6176d81180bSLoGin /// 6186d81180bSLoGin /// - `fd`:文件描述符 6196d81180bSLoGin /// - `len`:文件大小 6206d81180bSLoGin /// 6216d81180bSLoGin /// ## 返回值 6226d81180bSLoGin /// 6236d81180bSLoGin /// 如果成功,返回0,否则返回错误码. 6246d81180bSLoGin pub fn ftruncate(fd: i32, len: usize) -> Result<usize, SystemError> { 625*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 626*1496ba7bSLoGin let fd_table_guard = binding.read(); 627*1496ba7bSLoGin 628*1496ba7bSLoGin if let Some(file) = fd_table_guard.get_file_by_fd(fd) { 629*1496ba7bSLoGin // drop guard 以避免无法调度的问题 630*1496ba7bSLoGin drop(fd_table_guard); 631*1496ba7bSLoGin let r = file.lock_no_preempt().ftruncate(len).map(|_| 0); 6326d81180bSLoGin return r; 6336d81180bSLoGin } 634*1496ba7bSLoGin 6356d81180bSLoGin return Err(SystemError::EBADF); 6366d81180bSLoGin } 637*1496ba7bSLoGin 63867b48188Shoumkh fn do_fstat(fd: i32) -> Result<PosixKstat, SystemError> { 639*1496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 640*1496ba7bSLoGin let fd_table_guard = binding.read(); 641*1496ba7bSLoGin let file = fd_table_guard 642*1496ba7bSLoGin .get_file_by_fd(fd) 643*1496ba7bSLoGin .ok_or(SystemError::EBADF)?; 644*1496ba7bSLoGin // drop guard 以避免无法调度的问题 645*1496ba7bSLoGin drop(fd_table_guard); 646*1496ba7bSLoGin 64767b48188Shoumkh let mut kstat = PosixKstat::new(); 64867b48188Shoumkh // 获取文件信息 649*1496ba7bSLoGin let metadata = file.lock().metadata()?; 65067b48188Shoumkh kstat.size = metadata.size as i64; 65167b48188Shoumkh kstat.dev_id = metadata.dev_id as u64; 65267b48188Shoumkh kstat.inode = metadata.inode_id as u64; 65367b48188Shoumkh kstat.blcok_size = metadata.blk_size as i64; 65467b48188Shoumkh kstat.blocks = metadata.blocks as u64; 65567b48188Shoumkh 65667b48188Shoumkh kstat.atime.tv_sec = metadata.atime.tv_sec; 65767b48188Shoumkh kstat.atime.tv_nsec = metadata.atime.tv_nsec; 65867b48188Shoumkh kstat.mtime.tv_sec = metadata.mtime.tv_sec; 65967b48188Shoumkh kstat.mtime.tv_nsec = metadata.mtime.tv_nsec; 66067b48188Shoumkh kstat.ctime.tv_sec = metadata.ctime.tv_sec; 66167b48188Shoumkh kstat.ctime.tv_nsec = metadata.ctime.tv_nsec; 66267b48188Shoumkh 66367b48188Shoumkh kstat.nlink = metadata.nlinks as u64; 66467b48188Shoumkh kstat.uid = metadata.uid as i32; 66567b48188Shoumkh kstat.gid = metadata.gid as i32; 66667b48188Shoumkh kstat.rdev = metadata.raw_dev as i64; 66767b48188Shoumkh kstat.mode.bits = metadata.mode; 668*1496ba7bSLoGin match file.lock().file_type() { 66967b48188Shoumkh FileType::File => kstat.mode.insert(ModeType::S_IFMT), 67067b48188Shoumkh FileType::Dir => kstat.mode.insert(ModeType::S_IFDIR), 67167b48188Shoumkh FileType::BlockDevice => kstat.mode.insert(ModeType::S_IFBLK), 67267b48188Shoumkh FileType::CharDevice => kstat.mode.insert(ModeType::S_IFCHR), 67367b48188Shoumkh FileType::SymLink => kstat.mode.insert(ModeType::S_IFLNK), 67467b48188Shoumkh FileType::Socket => kstat.mode.insert(ModeType::S_IFSOCK), 67567b48188Shoumkh FileType::Pipe => kstat.mode.insert(ModeType::S_IFIFO), 67667b48188Shoumkh } 67767b48188Shoumkh 67867b48188Shoumkh return Ok(kstat); 67967b48188Shoumkh } 680*1496ba7bSLoGin 68167b48188Shoumkh pub fn fstat(fd: i32, usr_kstat: *mut PosixKstat) -> Result<usize, SystemError> { 682*1496ba7bSLoGin let kstat = Self::do_fstat(fd)?; 68367b48188Shoumkh if usr_kstat.is_null() { 68467b48188Shoumkh return Err(SystemError::EFAULT); 68567b48188Shoumkh } 68667b48188Shoumkh unsafe { 68767b48188Shoumkh *usr_kstat = kstat; 68867b48188Shoumkh } 68967b48188Shoumkh return Ok(0); 69067b48188Shoumkh } 69167b48188Shoumkh } 692cde5492fSlogin #[repr(C)] 693cde5492fSlogin #[derive(Debug, Clone, Copy)] 694cde5492fSlogin pub struct IoVec { 695cde5492fSlogin /// 缓冲区的起始地址 696cde5492fSlogin pub iov_base: *mut u8, 697cde5492fSlogin /// 缓冲区的长度 698cde5492fSlogin pub iov_len: usize, 699cde5492fSlogin } 700cde5492fSlogin 701cde5492fSlogin /// 用于存储多个来自用户空间的IoVec 702cde5492fSlogin /// 703cde5492fSlogin /// 由于目前内核中的文件系统还不支持分散读写,所以暂时只支持将用户空间的IoVec聚合成一个缓冲区,然后进行操作。 704cde5492fSlogin /// TODO:支持分散读写 705cde5492fSlogin #[derive(Debug)] 706cde5492fSlogin pub struct IoVecs(Vec<&'static mut [u8]>); 707cde5492fSlogin 708cde5492fSlogin impl IoVecs { 709cde5492fSlogin /// 从用户空间的IoVec中构造IoVecs 710cde5492fSlogin /// 711cde5492fSlogin /// @param iov 用户空间的IoVec 712cde5492fSlogin /// @param iovcnt 用户空间的IoVec的数量 713cde5492fSlogin /// @param readv 是否为readv系统调用 714cde5492fSlogin /// 715cde5492fSlogin /// @return 构造成功返回IoVecs,否则返回错误码 716cde5492fSlogin pub unsafe fn from_user( 717cde5492fSlogin iov: *const IoVec, 718cde5492fSlogin iovcnt: usize, 719cde5492fSlogin _readv: bool, 720cde5492fSlogin ) -> Result<Self, SystemError> { 721cde5492fSlogin // 检查iov指针所在空间是否合法 722cde5492fSlogin if !verify_area( 723cde5492fSlogin iov as usize as u64, 724cde5492fSlogin (iovcnt * core::mem::size_of::<IoVec>()) as u64, 725cde5492fSlogin ) { 726cde5492fSlogin return Err(SystemError::EFAULT); 727cde5492fSlogin } 728cde5492fSlogin 729cde5492fSlogin // 将用户空间的IoVec转换为引用(注意:这里的引用是静态的,因为用户空间的IoVec不会被释放) 730cde5492fSlogin let iovs: &[IoVec] = core::slice::from_raw_parts(iov, iovcnt); 731cde5492fSlogin 732cde5492fSlogin let mut slices: Vec<&mut [u8]> = vec![]; 733cde5492fSlogin slices.reserve(iovs.len()); 734cde5492fSlogin 735cde5492fSlogin for iov in iovs.iter() { 736cde5492fSlogin if iov.iov_len == 0 { 737cde5492fSlogin continue; 738cde5492fSlogin } 739cde5492fSlogin 740cde5492fSlogin if !verify_area(iov.iov_base as usize as u64, iov.iov_len as u64) { 741cde5492fSlogin return Err(SystemError::EFAULT); 742cde5492fSlogin } 743cde5492fSlogin 744cde5492fSlogin slices.push(core::slice::from_raw_parts_mut(iov.iov_base, iov.iov_len)); 745cde5492fSlogin } 746cde5492fSlogin 747cde5492fSlogin return Ok(Self(slices)); 748cde5492fSlogin } 749cde5492fSlogin 750cde5492fSlogin /// @brief 将IoVecs中的数据聚合到一个缓冲区中 751cde5492fSlogin /// 752cde5492fSlogin /// @return 返回聚合后的缓冲区 753cde5492fSlogin pub fn gather(&self) -> Vec<u8> { 754cde5492fSlogin let mut buf = Vec::new(); 755cde5492fSlogin for slice in self.0.iter() { 756cde5492fSlogin buf.extend_from_slice(slice); 757cde5492fSlogin } 758cde5492fSlogin return buf; 759cde5492fSlogin } 760cde5492fSlogin 761cde5492fSlogin /// @brief 将给定的数据分散写入到IoVecs中 762cde5492fSlogin pub fn scatter(&mut self, data: &[u8]) { 763cde5492fSlogin let mut data: &[u8] = data; 764cde5492fSlogin for slice in self.0.iter_mut() { 765cde5492fSlogin let len = core::cmp::min(slice.len(), data.len()); 766cde5492fSlogin if len == 0 { 767cde5492fSlogin continue; 768cde5492fSlogin } 769cde5492fSlogin 770cde5492fSlogin slice[..len].copy_from_slice(&data[..len]); 771cde5492fSlogin data = &data[len..]; 772cde5492fSlogin } 773cde5492fSlogin } 774cde5492fSlogin 775cde5492fSlogin /// @brief 创建与IoVecs等长的缓冲区 776cde5492fSlogin /// 777cde5492fSlogin /// @param set_len 是否设置返回的Vec的len。 778cde5492fSlogin /// 如果为true,则返回的Vec的len为所有IoVec的长度之和; 779cde5492fSlogin /// 否则返回的Vec的len为0,capacity为所有IoVec的长度之和. 780cde5492fSlogin /// 781cde5492fSlogin /// @return 返回创建的缓冲区 782cde5492fSlogin pub fn new_buf(&self, set_len: bool) -> Vec<u8> { 783cde5492fSlogin let total_len: usize = self.0.iter().map(|slice| slice.len()).sum(); 784cde5492fSlogin let mut buf: Vec<u8> = Vec::with_capacity(total_len); 785cde5492fSlogin 786cde5492fSlogin if set_len { 787cde5492fSlogin unsafe { 788cde5492fSlogin buf.set_len(total_len); 789cde5492fSlogin } 790cde5492fSlogin } 791cde5492fSlogin return buf; 792cde5492fSlogin } 793cde5492fSlogin } 794