syscall.rs 34 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066
  1. use core::ffi::CStr;
  2. use alloc::{
  3. string::{String, ToString},
  4. sync::Arc,
  5. vec::Vec,
  6. };
  7. use crate::{
  8. driver::base::{block::SeekFrom, device::DeviceNumber},
  9. filesystem::vfs::file::FileDescriptorVec,
  10. include::bindings::bindings::verify_area,
  11. kerror,
  12. libs::rwlock::RwLockWriteGuard,
  13. mm::VirtAddr,
  14. process::ProcessManager,
  15. syscall::{
  16. user_access::{check_and_clone_cstr, UserBufferReader, UserBufferWriter},
  17. Syscall, SystemError,
  18. },
  19. time::TimeSpec,
  20. };
  21. use super::{
  22. core::{do_mkdir, do_remove_dir, do_unlink_at},
  23. fcntl::{AtFlags, FcntlCommand, FD_CLOEXEC},
  24. file::{File, FileMode},
  25. open::{do_faccessat, do_fchmodat, do_sys_open},
  26. utils::{rsplit_path, user_path_at},
  27. Dirent, FileType, IndexNode, MAX_PATHLEN, ROOT_INODE, VFS_MAX_FOLLOW_SYMLINK_TIMES,
  28. };
  29. // use crate::kdebug;
  30. pub const SEEK_SET: u32 = 0;
  31. pub const SEEK_CUR: u32 = 1;
  32. pub const SEEK_END: u32 = 2;
  33. pub const SEEK_MAX: u32 = 3;
  34. bitflags! {
  35. /// 文件类型和权限
  36. #[repr(C)]
  37. pub struct ModeType: u32 {
  38. /// 掩码
  39. const S_IFMT = 0o0_170_000;
  40. /// 文件类型
  41. const S_IFSOCK = 0o140000;
  42. const S_IFLNK = 0o120000;
  43. const S_IFREG = 0o100000;
  44. const S_IFBLK = 0o060000;
  45. const S_IFDIR = 0o040000;
  46. const S_IFCHR = 0o020000;
  47. const S_IFIFO = 0o010000;
  48. const S_ISUID = 0o004000;
  49. const S_ISGID = 0o002000;
  50. const S_ISVTX = 0o001000;
  51. /// 文件用户权限
  52. const S_IRWXU = 0o0700;
  53. const S_IRUSR = 0o0400;
  54. const S_IWUSR = 0o0200;
  55. const S_IXUSR = 0o0100;
  56. /// 文件组权限
  57. const S_IRWXG = 0o0070;
  58. const S_IRGRP = 0o0040;
  59. const S_IWGRP = 0o0020;
  60. const S_IXGRP = 0o0010;
  61. /// 文件其他用户权限
  62. const S_IRWXO = 0o0007;
  63. const S_IROTH = 0o0004;
  64. const S_IWOTH = 0o0002;
  65. const S_IXOTH = 0o0001;
  66. /// 0o777
  67. const S_IRWXUGO = Self::S_IRWXU.bits | Self::S_IRWXG.bits | Self::S_IRWXO.bits;
  68. /// 0o7777
  69. const S_IALLUGO = Self::S_ISUID.bits | Self::S_ISGID.bits | Self::S_ISVTX.bits| Self::S_IRWXUGO.bits;
  70. /// 0o444
  71. const S_IRUGO = Self::S_IRUSR.bits | Self::S_IRGRP.bits | Self::S_IROTH.bits;
  72. /// 0o222
  73. const S_IWUGO = Self::S_IWUSR.bits | Self::S_IWGRP.bits | Self::S_IWOTH.bits;
  74. /// 0o111
  75. const S_IXUGO = Self::S_IXUSR.bits | Self::S_IXGRP.bits | Self::S_IXOTH.bits;
  76. }
  77. }
  78. #[repr(C)]
  79. /// # 文件信息结构体
  80. pub struct PosixKstat {
  81. /// 硬件设备ID
  82. dev_id: u64,
  83. /// inode号
  84. inode: u64,
  85. /// 硬链接数
  86. nlink: u64,
  87. /// 文件权限
  88. mode: ModeType,
  89. /// 所有者用户ID
  90. uid: i32,
  91. /// 所有者组ID
  92. gid: i32,
  93. /// 设备ID
  94. rdev: i64,
  95. /// 文件大小
  96. size: i64,
  97. /// 文件系统块大小
  98. blcok_size: i64,
  99. /// 分配的512B块数
  100. blocks: u64,
  101. /// 最后访问时间
  102. atime: TimeSpec,
  103. /// 最后修改时间
  104. mtime: TimeSpec,
  105. /// 最后状态变化时间
  106. ctime: TimeSpec,
  107. /// 用于填充结构体大小的空白数据
  108. pub _pad: [i8; 24],
  109. }
  110. impl PosixKstat {
  111. fn new() -> Self {
  112. Self {
  113. inode: 0,
  114. dev_id: 0,
  115. mode: ModeType { bits: 0 },
  116. nlink: 0,
  117. uid: 0,
  118. gid: 0,
  119. rdev: 0,
  120. size: 0,
  121. atime: TimeSpec {
  122. tv_sec: 0,
  123. tv_nsec: 0,
  124. },
  125. mtime: TimeSpec {
  126. tv_sec: 0,
  127. tv_nsec: 0,
  128. },
  129. ctime: TimeSpec {
  130. tv_sec: 0,
  131. tv_nsec: 0,
  132. },
  133. blcok_size: 0,
  134. blocks: 0,
  135. _pad: Default::default(),
  136. }
  137. }
  138. }
  139. ///
  140. /// Arguments for how openat2(2) should open the target path. If only @flags and
  141. /// @mode are non-zero, then openat2(2) operates very similarly to openat(2).
  142. ///
  143. /// However, unlike openat(2), unknown or invalid bits in @flags result in
  144. /// -EINVAL rather than being silently ignored. @mode must be zero unless one of
  145. /// {O_CREAT, O_TMPFILE} are set.
  146. ///
  147. /// ## 成员变量
  148. ///
  149. /// - flags: O_* flags.
  150. /// - mode: O_CREAT/O_TMPFILE file mode.
  151. /// - resolve: RESOLVE_* flags.
  152. #[derive(Debug, Clone, Copy)]
  153. #[repr(C)]
  154. pub struct PosixOpenHow {
  155. pub flags: u64,
  156. pub mode: u64,
  157. pub resolve: u64,
  158. }
  159. impl PosixOpenHow {
  160. #[allow(dead_code)]
  161. pub fn new(flags: u64, mode: u64, resolve: u64) -> Self {
  162. Self {
  163. flags,
  164. mode,
  165. resolve,
  166. }
  167. }
  168. }
  169. #[derive(Debug, Clone, Copy)]
  170. pub struct OpenHow {
  171. pub o_flags: FileMode,
  172. pub mode: ModeType,
  173. pub resolve: OpenHowResolve,
  174. }
  175. impl OpenHow {
  176. pub fn new(mut o_flags: FileMode, mut mode: ModeType, resolve: OpenHowResolve) -> Self {
  177. if !o_flags.contains(FileMode::O_CREAT) {
  178. mode = ModeType::empty();
  179. }
  180. if o_flags.contains(FileMode::O_PATH) {
  181. o_flags = o_flags.intersection(FileMode::O_PATH_FLAGS);
  182. }
  183. Self {
  184. o_flags,
  185. mode,
  186. resolve,
  187. }
  188. }
  189. }
  190. impl From<PosixOpenHow> for OpenHow {
  191. fn from(posix_open_how: PosixOpenHow) -> Self {
  192. let o_flags = FileMode::from_bits_truncate(posix_open_how.flags as u32);
  193. let mode = ModeType::from_bits_truncate(posix_open_how.mode as u32);
  194. let resolve = OpenHowResolve::from_bits_truncate(posix_open_how.resolve as u64);
  195. return Self::new(o_flags, mode, resolve);
  196. }
  197. }
  198. bitflags! {
  199. pub struct OpenHowResolve: u64{
  200. /// Block mount-point crossings
  201. /// (including bind-mounts).
  202. const RESOLVE_NO_XDEV = 0x01;
  203. /// Block traversal through procfs-style
  204. /// "magic-links"
  205. const RESOLVE_NO_MAGICLINKS = 0x02;
  206. /// Block traversal through all symlinks
  207. /// (implies OEXT_NO_MAGICLINKS)
  208. const RESOLVE_NO_SYMLINKS = 0x04;
  209. /// Block "lexical" trickery like
  210. /// "..", symlinks, and absolute
  211. const RESOLVE_BENEATH = 0x08;
  212. /// Make all jumps to "/" and ".."
  213. /// be scoped inside the dirfd
  214. /// (similar to chroot(2)).
  215. const RESOLVE_IN_ROOT = 0x10;
  216. // Only complete if resolution can be
  217. // completed through cached lookup. May
  218. // return -EAGAIN if that's not
  219. // possible.
  220. const RESOLVE_CACHED = 0x20;
  221. }
  222. }
  223. impl Syscall {
  224. /// @brief 为当前进程打开一个文件
  225. ///
  226. /// @param path 文件路径
  227. /// @param o_flags 打开文件的标志位
  228. ///
  229. /// @return 文件描述符编号,或者是错误码
  230. pub fn open(
  231. path: &str,
  232. flags: FileMode,
  233. mode: ModeType,
  234. follow_symlink: bool,
  235. ) -> Result<usize, SystemError> {
  236. return do_sys_open(AtFlags::AT_FDCWD.bits(), path, flags, mode, follow_symlink);
  237. }
  238. pub fn openat(
  239. dirfd: i32,
  240. path: &str,
  241. o_flags: FileMode,
  242. mode: ModeType,
  243. follow_symlink: bool,
  244. ) -> Result<usize, SystemError> {
  245. return do_sys_open(dirfd, path, o_flags, mode, follow_symlink);
  246. }
  247. /// @brief 关闭文件
  248. ///
  249. /// @param fd 文件描述符编号
  250. ///
  251. /// @return 成功返回0,失败返回错误码
  252. pub fn close(fd: usize) -> Result<usize, SystemError> {
  253. let binding = ProcessManager::current_pcb().fd_table();
  254. let mut fd_table_guard = binding.write();
  255. let res = fd_table_guard.drop_fd(fd as i32).map(|_| 0);
  256. return res;
  257. }
  258. /// @brief 发送命令到文件描述符对应的设备,
  259. ///
  260. /// @param fd 文件描述符编号
  261. /// @param cmd 设备相关的请求类型
  262. ///
  263. /// @return Ok(usize) 成功返回0
  264. /// @return Err(SystemError) 读取失败,返回posix错误码
  265. pub fn ioctl(fd: usize, cmd: u32, data: usize) -> Result<usize, SystemError> {
  266. let binding = ProcessManager::current_pcb().fd_table();
  267. let fd_table_guard = binding.read();
  268. let file = fd_table_guard
  269. .get_file_by_fd(fd as i32)
  270. .ok_or(SystemError::EBADF)?;
  271. // drop guard 以避免无法调度的问题
  272. drop(fd_table_guard);
  273. let r = file.lock_no_preempt().inode().ioctl(cmd, data);
  274. return r;
  275. }
  276. /// @brief 根据文件描述符,读取文件数据。尝试读取的数据长度与buf的长度相同。
  277. ///
  278. /// @param fd 文件描述符编号
  279. /// @param buf 输出缓冲区。
  280. ///
  281. /// @return Ok(usize) 成功读取的数据的字节数
  282. /// @return Err(SystemError) 读取失败,返回posix错误码
  283. pub fn read(fd: i32, buf: &mut [u8]) -> Result<usize, SystemError> {
  284. let binding = ProcessManager::current_pcb().fd_table();
  285. let fd_table_guard = binding.read();
  286. let file = fd_table_guard.get_file_by_fd(fd);
  287. if file.is_none() {
  288. return Err(SystemError::EBADF);
  289. }
  290. // drop guard 以避免无法调度的问题
  291. drop(fd_table_guard);
  292. let file = file.unwrap();
  293. return file.lock_no_preempt().read(buf.len(), buf);
  294. }
  295. /// @brief 根据文件描述符,向文件写入数据。尝试写入的数据长度与buf的长度相同。
  296. ///
  297. /// @param fd 文件描述符编号
  298. /// @param buf 输入缓冲区。
  299. ///
  300. /// @return Ok(usize) 成功写入的数据的字节数
  301. /// @return Err(SystemError) 写入失败,返回posix错误码
  302. pub fn write(fd: i32, buf: &[u8]) -> Result<usize, SystemError> {
  303. let binding = ProcessManager::current_pcb().fd_table();
  304. let fd_table_guard = binding.read();
  305. let file = fd_table_guard
  306. .get_file_by_fd(fd)
  307. .ok_or(SystemError::EBADF)?;
  308. // drop guard 以避免无法调度的问题
  309. drop(fd_table_guard);
  310. return file.lock_no_preempt().write(buf.len(), buf);
  311. }
  312. /// @brief 调整文件操作指针的位置
  313. ///
  314. /// @param fd 文件描述符编号
  315. /// @param seek 调整的方式
  316. ///
  317. /// @return Ok(usize) 调整后,文件访问指针相对于文件头部的偏移量
  318. /// @return Err(SystemError) 调整失败,返回posix错误码
  319. pub fn lseek(fd: i32, seek: SeekFrom) -> Result<usize, SystemError> {
  320. let binding = ProcessManager::current_pcb().fd_table();
  321. let fd_table_guard = binding.read();
  322. let file = fd_table_guard
  323. .get_file_by_fd(fd)
  324. .ok_or(SystemError::EBADF)?;
  325. // drop guard 以避免无法调度的问题
  326. drop(fd_table_guard);
  327. return file.lock_no_preempt().lseek(seek);
  328. }
  329. /// @brief 切换工作目录
  330. ///
  331. /// @param dest_path 目标路径
  332. ///
  333. /// @return 返回码 描述
  334. /// 0 | 成功
  335. ///
  336. /// EACCESS | 权限不足
  337. ///
  338. /// ELOOP | 解析path时遇到路径循环
  339. ///
  340. /// ENAMETOOLONG | 路径名过长
  341. ///
  342. /// ENOENT | 目标文件或目录不存在
  343. ///
  344. /// ENODIR | 检索期间发现非目录项
  345. ///
  346. /// ENOMEM | 系统内存不足
  347. ///
  348. /// EFAULT | 错误的地址
  349. ///
  350. /// ENAMETOOLONG | 路径过长
  351. pub fn chdir(dest_path: &str) -> Result<usize, SystemError> {
  352. let proc = ProcessManager::current_pcb();
  353. // Copy path to kernel space to avoid some security issues
  354. let path = dest_path.to_string();
  355. let mut new_path = String::from("");
  356. if path.len() > 0 {
  357. let cwd = match path.as_bytes()[0] {
  358. b'/' => String::from("/"),
  359. _ => proc.basic().cwd(),
  360. };
  361. let mut cwd_vec: Vec<_> = cwd.split("/").filter(|&x| x != "").collect();
  362. let path_split = path.split("/").filter(|&x| x != "");
  363. for seg in path_split {
  364. if seg == ".." {
  365. cwd_vec.pop();
  366. } else if seg == "." {
  367. // 当前目录
  368. } else {
  369. cwd_vec.push(seg);
  370. }
  371. }
  372. //proc.basic().set_path(String::from(""));
  373. for seg in cwd_vec {
  374. new_path.push_str("/");
  375. new_path.push_str(seg);
  376. }
  377. if new_path == "" {
  378. new_path = String::from("/");
  379. }
  380. }
  381. let inode =
  382. match ROOT_INODE().lookup_follow_symlink(&new_path, VFS_MAX_FOLLOW_SYMLINK_TIMES) {
  383. Err(e) => {
  384. kerror!("Change Directory Failed, Error = {:?}", e);
  385. return Err(SystemError::ENOENT);
  386. }
  387. Ok(i) => i,
  388. };
  389. let metadata = inode.metadata()?;
  390. if metadata.file_type == FileType::Dir {
  391. proc.basic_mut().set_cwd(String::from(new_path));
  392. return Ok(0);
  393. } else {
  394. return Err(SystemError::ENOTDIR);
  395. }
  396. }
  397. /// @brief 获取当前进程的工作目录路径
  398. ///
  399. /// @param buf 指向缓冲区的指针
  400. /// @param size 缓冲区的大小
  401. ///
  402. /// @return 成功,返回的指针指向包含工作目录路径的字符串
  403. /// @return 错误,没有足够的空间
  404. pub fn getcwd(buf: &mut [u8]) -> Result<VirtAddr, SystemError> {
  405. let proc = ProcessManager::current_pcb();
  406. let cwd = proc.basic().cwd();
  407. let cwd_bytes = cwd.as_bytes();
  408. let cwd_len = cwd_bytes.len();
  409. if cwd_len + 1 > buf.len() {
  410. return Err(SystemError::ENOMEM);
  411. }
  412. buf[..cwd_len].copy_from_slice(cwd_bytes);
  413. buf[cwd_len] = 0;
  414. return Ok(VirtAddr::new(buf.as_ptr() as usize));
  415. }
  416. /// @brief 获取目录中的数据
  417. ///
  418. /// TODO: 这个函数的语义与Linux不一致,需要修改!!!
  419. ///
  420. /// @param fd 文件描述符号
  421. /// @param buf 输出缓冲区
  422. ///
  423. /// @return 成功返回读取的字节数,失败返回错误码
  424. pub fn getdents(fd: i32, buf: &mut [u8]) -> Result<usize, SystemError> {
  425. let dirent =
  426. unsafe { (buf.as_mut_ptr() as *mut Dirent).as_mut() }.ok_or(SystemError::EFAULT)?;
  427. if fd < 0 || fd as usize > FileDescriptorVec::PROCESS_MAX_FD {
  428. return Err(SystemError::EBADF);
  429. }
  430. // 获取fd
  431. let binding = ProcessManager::current_pcb().fd_table();
  432. let fd_table_guard = binding.read();
  433. let file = fd_table_guard
  434. .get_file_by_fd(fd)
  435. .ok_or(SystemError::EBADF)?;
  436. // drop guard 以避免无法调度的问题
  437. drop(fd_table_guard);
  438. let res = file.lock_no_preempt().readdir(dirent).map(|x| x as usize);
  439. return res;
  440. }
  441. /// @brief 创建文件夹
  442. ///
  443. /// @param path(r8) 路径 / mode(r9) 模式
  444. ///
  445. /// @return uint64_t 负数错误码 / 0表示成功
  446. pub fn mkdir(path: &str, mode: usize) -> Result<usize, SystemError> {
  447. return do_mkdir(path, FileMode::from_bits_truncate(mode as u32)).map(|x| x as usize);
  448. }
  449. /// **删除文件夹、取消文件的链接、删除文件的系统调用**
  450. ///
  451. /// ## 参数
  452. ///
  453. /// - `dirfd`:文件夹的文件描述符.目前暂未实现
  454. /// - `pathname`:文件夹的路径
  455. /// - `flags`:标志位
  456. ///
  457. ///
  458. pub fn unlinkat(dirfd: i32, pathname: &str, flags: u32) -> Result<usize, SystemError> {
  459. let flags = AtFlags::from_bits(flags as i32).ok_or(SystemError::EINVAL)?;
  460. if flags.contains(AtFlags::AT_REMOVEDIR) {
  461. // kdebug!("rmdir");
  462. match do_remove_dir(dirfd, &pathname) {
  463. Err(err) => {
  464. kerror!("Failed to Remove Directory, Error Code = {:?}", err);
  465. return Err(err);
  466. }
  467. Ok(_) => {
  468. return Ok(0);
  469. }
  470. }
  471. }
  472. match do_unlink_at(dirfd, &pathname) {
  473. Err(err) => {
  474. kerror!("Failed to Remove Directory, Error Code = {:?}", err);
  475. return Err(err);
  476. }
  477. Ok(_) => {
  478. return Ok(0);
  479. }
  480. }
  481. }
  482. pub fn unlink(pathname: *const u8) -> Result<usize, SystemError> {
  483. if pathname.is_null() {
  484. return Err(SystemError::EFAULT);
  485. }
  486. let ureader = UserBufferReader::new(pathname, MAX_PATHLEN, true)?;
  487. let buf: &[u8] = ureader.buffer(0).unwrap();
  488. let pathname: &CStr = CStr::from_bytes_until_nul(buf).map_err(|_| SystemError::EINVAL)?;
  489. let pathname: &str = pathname.to_str().map_err(|_| SystemError::EINVAL)?;
  490. if pathname.len() >= MAX_PATHLEN {
  491. return Err(SystemError::ENAMETOOLONG);
  492. }
  493. let pathname = pathname.trim();
  494. return do_unlink_at(AtFlags::AT_FDCWD.bits(), pathname).map(|v| v as usize);
  495. }
  496. /// @brief 根据提供的文件描述符的fd,复制对应的文件结构体,并返回新复制的文件结构体对应的fd
  497. pub fn dup(oldfd: i32) -> Result<usize, SystemError> {
  498. let binding = ProcessManager::current_pcb().fd_table();
  499. let mut fd_table_guard = binding.write();
  500. let old_file = fd_table_guard
  501. .get_file_by_fd(oldfd)
  502. .ok_or(SystemError::EBADF)?;
  503. let new_file = old_file
  504. .lock_no_preempt()
  505. .try_clone()
  506. .ok_or(SystemError::EBADF)?;
  507. // 申请文件描述符,并把文件对象存入其中
  508. let res = fd_table_guard.alloc_fd(new_file, None).map(|x| x as usize);
  509. return res;
  510. }
  511. /// 根据提供的文件描述符的fd,和指定新fd,复制对应的文件结构体,
  512. /// 并返回新复制的文件结构体对应的fd.
  513. /// 如果新fd已经打开,则会先关闭新fd.
  514. ///
  515. /// ## 参数
  516. ///
  517. /// - `oldfd`:旧文件描述符
  518. /// - `newfd`:新文件描述符
  519. ///
  520. /// ## 返回值
  521. ///
  522. /// - 成功:新文件描述符
  523. /// - 失败:错误码
  524. pub fn dup2(oldfd: i32, newfd: i32) -> Result<usize, SystemError> {
  525. let binding = ProcessManager::current_pcb().fd_table();
  526. let mut fd_table_guard = binding.write();
  527. return Self::do_dup2(oldfd, newfd, &mut fd_table_guard);
  528. }
  529. fn do_dup2(
  530. oldfd: i32,
  531. newfd: i32,
  532. fd_table_guard: &mut RwLockWriteGuard<'_, FileDescriptorVec>,
  533. ) -> Result<usize, SystemError> {
  534. // 确认oldfd, newid是否有效
  535. if !(FileDescriptorVec::validate_fd(oldfd) && FileDescriptorVec::validate_fd(newfd)) {
  536. return Err(SystemError::EBADF);
  537. }
  538. if oldfd == newfd {
  539. // 若oldfd与newfd相等
  540. return Ok(newfd as usize);
  541. }
  542. let new_exists = fd_table_guard.get_file_by_fd(newfd).is_some();
  543. if new_exists {
  544. // close newfd
  545. if let Err(_) = fd_table_guard.drop_fd(newfd) {
  546. // An I/O error occurred while attempting to close fildes2.
  547. return Err(SystemError::EIO);
  548. }
  549. }
  550. let old_file = fd_table_guard
  551. .get_file_by_fd(oldfd)
  552. .ok_or(SystemError::EBADF)?;
  553. let new_file = old_file
  554. .lock_no_preempt()
  555. .try_clone()
  556. .ok_or(SystemError::EBADF)?;
  557. // 申请文件描述符,并把文件对象存入其中
  558. let res = fd_table_guard
  559. .alloc_fd(new_file, Some(newfd))
  560. .map(|x| x as usize);
  561. return res;
  562. }
  563. /// # fcntl
  564. ///
  565. /// ## 参数
  566. ///
  567. /// - `fd`:文件描述符
  568. /// - `cmd`:命令
  569. /// - `arg`:参数
  570. pub fn fcntl(fd: i32, cmd: FcntlCommand, arg: i32) -> Result<usize, SystemError> {
  571. match cmd {
  572. FcntlCommand::DupFd => {
  573. if arg < 0 || arg as usize >= FileDescriptorVec::PROCESS_MAX_FD {
  574. return Err(SystemError::EBADF);
  575. }
  576. let arg = arg as usize;
  577. for i in arg..FileDescriptorVec::PROCESS_MAX_FD {
  578. let binding = ProcessManager::current_pcb().fd_table();
  579. let mut fd_table_guard = binding.write();
  580. if fd_table_guard.get_file_by_fd(fd).is_none() {
  581. return Self::do_dup2(fd, i as i32, &mut fd_table_guard);
  582. }
  583. }
  584. return Err(SystemError::EMFILE);
  585. }
  586. FcntlCommand::GetFd => {
  587. // Get file descriptor flags.
  588. let binding = ProcessManager::current_pcb().fd_table();
  589. let fd_table_guard = binding.read();
  590. if let Some(file) = fd_table_guard.get_file_by_fd(fd) {
  591. // drop guard 以避免无法调度的问题
  592. drop(fd_table_guard);
  593. if file.lock().close_on_exec() {
  594. return Ok(FD_CLOEXEC as usize);
  595. }
  596. }
  597. return Err(SystemError::EBADF);
  598. }
  599. FcntlCommand::SetFd => {
  600. // Set file descriptor flags.
  601. let binding = ProcessManager::current_pcb().fd_table();
  602. let fd_table_guard = binding.write();
  603. if let Some(file) = fd_table_guard.get_file_by_fd(fd) {
  604. // drop guard 以避免无法调度的问题
  605. drop(fd_table_guard);
  606. let arg = arg as u32;
  607. if arg & FD_CLOEXEC != 0 {
  608. file.lock().set_close_on_exec(true);
  609. } else {
  610. file.lock().set_close_on_exec(false);
  611. }
  612. return Ok(0);
  613. }
  614. return Err(SystemError::EBADF);
  615. }
  616. FcntlCommand::GetFlags => {
  617. // Get file status flags.
  618. let binding = ProcessManager::current_pcb().fd_table();
  619. let fd_table_guard = binding.read();
  620. if let Some(file) = fd_table_guard.get_file_by_fd(fd) {
  621. // drop guard 以避免无法调度的问题
  622. drop(fd_table_guard);
  623. return Ok(file.lock_no_preempt().mode().bits() as usize);
  624. }
  625. return Err(SystemError::EBADF);
  626. }
  627. FcntlCommand::SetFlags => {
  628. // Set file status flags.
  629. let binding = ProcessManager::current_pcb().fd_table();
  630. let fd_table_guard = binding.write();
  631. if let Some(file) = fd_table_guard.get_file_by_fd(fd) {
  632. let arg = arg as u32;
  633. let mode = FileMode::from_bits(arg).ok_or(SystemError::EINVAL)?;
  634. // drop guard 以避免无法调度的问题
  635. drop(fd_table_guard);
  636. file.lock_no_preempt().set_mode(mode)?;
  637. return Ok(0);
  638. }
  639. return Err(SystemError::EBADF);
  640. }
  641. _ => {
  642. // TODO: unimplemented
  643. // 未实现的命令,返回0,不报错。
  644. // kwarn!("fcntl: unimplemented command: {:?}, defaults to 0.", cmd);
  645. return Ok(0);
  646. }
  647. }
  648. }
  649. /// # ftruncate
  650. ///
  651. /// ## 描述
  652. ///
  653. /// 改变文件大小.
  654. /// 如果文件大小大于原来的大小,那么文件的内容将会被扩展到指定的大小,新的空间将会用0填充.
  655. /// 如果文件大小小于原来的大小,那么文件的内容将会被截断到指定的大小.
  656. ///
  657. /// ## 参数
  658. ///
  659. /// - `fd`:文件描述符
  660. /// - `len`:文件大小
  661. ///
  662. /// ## 返回值
  663. ///
  664. /// 如果成功,返回0,否则返回错误码.
  665. pub fn ftruncate(fd: i32, len: usize) -> Result<usize, SystemError> {
  666. let binding = ProcessManager::current_pcb().fd_table();
  667. let fd_table_guard = binding.read();
  668. if let Some(file) = fd_table_guard.get_file_by_fd(fd) {
  669. // drop guard 以避免无法调度的问题
  670. drop(fd_table_guard);
  671. let r = file.lock_no_preempt().ftruncate(len).map(|_| 0);
  672. return r;
  673. }
  674. return Err(SystemError::EBADF);
  675. }
  676. fn do_fstat(fd: i32) -> Result<PosixKstat, SystemError> {
  677. let binding = ProcessManager::current_pcb().fd_table();
  678. let fd_table_guard = binding.read();
  679. let file = fd_table_guard
  680. .get_file_by_fd(fd)
  681. .ok_or(SystemError::EBADF)?;
  682. // drop guard 以避免无法调度的问题
  683. drop(fd_table_guard);
  684. let mut kstat = PosixKstat::new();
  685. // 获取文件信息
  686. let metadata = file.lock().metadata()?;
  687. kstat.size = metadata.size as i64;
  688. kstat.dev_id = metadata.dev_id as u64;
  689. kstat.inode = metadata.inode_id.into() as u64;
  690. kstat.blcok_size = metadata.blk_size as i64;
  691. kstat.blocks = metadata.blocks as u64;
  692. kstat.atime.tv_sec = metadata.atime.tv_sec;
  693. kstat.atime.tv_nsec = metadata.atime.tv_nsec;
  694. kstat.mtime.tv_sec = metadata.mtime.tv_sec;
  695. kstat.mtime.tv_nsec = metadata.mtime.tv_nsec;
  696. kstat.ctime.tv_sec = metadata.ctime.tv_sec;
  697. kstat.ctime.tv_nsec = metadata.ctime.tv_nsec;
  698. kstat.nlink = metadata.nlinks as u64;
  699. kstat.uid = metadata.uid as i32;
  700. kstat.gid = metadata.gid as i32;
  701. kstat.rdev = metadata.raw_dev as i64;
  702. kstat.mode = metadata.mode;
  703. match file.lock().file_type() {
  704. FileType::File => kstat.mode.insert(ModeType::S_IFREG),
  705. FileType::Dir => kstat.mode.insert(ModeType::S_IFDIR),
  706. FileType::BlockDevice => kstat.mode.insert(ModeType::S_IFBLK),
  707. FileType::CharDevice => kstat.mode.insert(ModeType::S_IFCHR),
  708. FileType::SymLink => kstat.mode.insert(ModeType::S_IFLNK),
  709. FileType::Socket => kstat.mode.insert(ModeType::S_IFSOCK),
  710. FileType::Pipe => kstat.mode.insert(ModeType::S_IFIFO),
  711. FileType::KvmDevice => kstat.mode.insert(ModeType::S_IFCHR),
  712. }
  713. return Ok(kstat);
  714. }
  715. pub fn fstat(fd: i32, usr_kstat: *mut PosixKstat) -> Result<usize, SystemError> {
  716. let kstat = Self::do_fstat(fd)?;
  717. if usr_kstat.is_null() {
  718. return Err(SystemError::EFAULT);
  719. }
  720. unsafe {
  721. *usr_kstat = kstat;
  722. }
  723. return Ok(0);
  724. }
  725. pub fn stat(path: &str, user_kstat: *mut PosixKstat) -> Result<usize, SystemError> {
  726. let fd = Self::open(path, FileMode::O_RDONLY, ModeType::empty(), true)?;
  727. let r = Self::fstat(fd as i32, user_kstat);
  728. Self::close(fd).ok();
  729. return r;
  730. }
  731. pub fn lstat(path: &str, user_kstat: *mut PosixKstat) -> Result<usize, SystemError> {
  732. let fd = Self::open(path, FileMode::O_RDONLY, ModeType::empty(), false)?;
  733. let r = Self::fstat(fd as i32, user_kstat);
  734. Self::close(fd).ok();
  735. return r;
  736. }
  737. pub fn mknod(
  738. path_ptr: *const i8,
  739. mode: ModeType,
  740. dev_t: DeviceNumber,
  741. ) -> Result<usize, SystemError> {
  742. // 安全检验
  743. let len = unsafe { CStr::from_ptr(path_ptr).to_bytes().len() };
  744. let user_buffer = UserBufferReader::new(path_ptr, len, true)?;
  745. let buf = user_buffer.read_from_user::<u8>(0)?;
  746. let path = core::str::from_utf8(buf).map_err(|_| SystemError::EINVAL)?;
  747. // 文件名过长
  748. if path.len() > MAX_PATHLEN as usize {
  749. return Err(SystemError::ENAMETOOLONG);
  750. }
  751. let inode: Result<Arc<dyn IndexNode>, SystemError> =
  752. ROOT_INODE().lookup_follow_symlink(path, VFS_MAX_FOLLOW_SYMLINK_TIMES);
  753. if inode.is_ok() {
  754. return Err(SystemError::EEXIST);
  755. }
  756. let (filename, parent_path) = rsplit_path(path);
  757. // 查找父目录
  758. let parent_inode: Arc<dyn IndexNode> = ROOT_INODE()
  759. .lookup_follow_symlink(parent_path.unwrap_or("/"), VFS_MAX_FOLLOW_SYMLINK_TIMES)?;
  760. // 创建nod
  761. parent_inode.mknod(filename, mode, dev_t)?;
  762. return Ok(0);
  763. }
  764. pub fn writev(fd: i32, iov: usize, count: usize) -> Result<usize, SystemError> {
  765. // IoVecs会进行用户态检验
  766. let iovecs = unsafe { IoVecs::from_user(iov as *const IoVec, count, false) }?;
  767. let data = iovecs.gather();
  768. Self::write(fd, &data)
  769. }
  770. pub fn readv(fd: i32, iov: usize, count: usize) -> Result<usize, SystemError> {
  771. // IoVecs会进行用户态检验
  772. let mut iovecs = unsafe { IoVecs::from_user(iov as *const IoVec, count, true) }?;
  773. let mut data = Vec::new();
  774. data.resize(iovecs.0.iter().map(|x| x.len()).sum(), 0);
  775. let len = Self::read(fd, &mut data)?;
  776. iovecs.scatter(&data[..len]);
  777. return Ok(len);
  778. }
  779. pub fn readlink_at(
  780. dirfd: i32,
  781. path: *const u8,
  782. user_buf: *mut u8,
  783. buf_size: usize,
  784. ) -> Result<usize, SystemError> {
  785. let path = check_and_clone_cstr(path, Some(MAX_PATHLEN))?;
  786. let mut user_buf = UserBufferWriter::new(user_buf, buf_size, true)?;
  787. if path.len() == 0 {
  788. return Err(SystemError::EINVAL);
  789. }
  790. let (inode, path) = user_path_at(&ProcessManager::current_pcb(), dirfd, &path)?;
  791. let inode = inode.lookup(path.as_str())?;
  792. if inode.metadata()?.file_type != FileType::SymLink {
  793. return Err(SystemError::EINVAL);
  794. }
  795. let ubuf = user_buf.buffer::<u8>(0).unwrap();
  796. let mut file = File::new(inode, FileMode::O_RDONLY)?;
  797. let len = file.read(buf_size, ubuf)?;
  798. return Ok(len);
  799. }
  800. pub fn readlink(
  801. path: *const u8,
  802. user_buf: *mut u8,
  803. buf_size: usize,
  804. ) -> Result<usize, SystemError> {
  805. return Self::readlink_at(AtFlags::AT_FDCWD.bits(), path, user_buf, buf_size);
  806. }
  807. pub fn access(pathname: *const u8, mode: u32) -> Result<usize, SystemError> {
  808. return do_faccessat(
  809. AtFlags::AT_FDCWD.bits(),
  810. pathname,
  811. ModeType::from_bits(mode).ok_or(SystemError::EINVAL)?,
  812. 0,
  813. );
  814. }
  815. pub fn faccessat2(
  816. dirfd: i32,
  817. pathname: *const u8,
  818. mode: u32,
  819. flags: u32,
  820. ) -> Result<usize, SystemError> {
  821. return do_faccessat(
  822. dirfd,
  823. pathname,
  824. ModeType::from_bits(mode).ok_or(SystemError::EINVAL)?,
  825. flags,
  826. );
  827. }
  828. pub fn chmod(pathname: *const u8, mode: u32) -> Result<usize, SystemError> {
  829. return do_fchmodat(
  830. AtFlags::AT_FDCWD.bits(),
  831. pathname,
  832. ModeType::from_bits(mode).ok_or(SystemError::EINVAL)?,
  833. );
  834. }
  835. pub fn fchmodat(dirfd: i32, pathname: *const u8, mode: u32) -> Result<usize, SystemError> {
  836. return do_fchmodat(
  837. dirfd,
  838. pathname,
  839. ModeType::from_bits(mode).ok_or(SystemError::EINVAL)?,
  840. );
  841. }
  842. pub fn fchmod(fd: i32, mode: u32) -> Result<usize, SystemError> {
  843. let _mode = ModeType::from_bits(mode).ok_or(SystemError::EINVAL)?;
  844. let binding = ProcessManager::current_pcb().fd_table();
  845. let fd_table_guard = binding.read();
  846. let _file = fd_table_guard
  847. .get_file_by_fd(fd)
  848. .ok_or(SystemError::EBADF)?;
  849. // fchmod没完全实现,因此不修改文件的权限
  850. // todo: 实现fchmod
  851. kwarn!("fchmod not fully implemented");
  852. return Ok(0);
  853. }
  854. }
  855. #[repr(C)]
  856. #[derive(Debug, Clone, Copy)]
  857. pub struct IoVec {
  858. /// 缓冲区的起始地址
  859. pub iov_base: *mut u8,
  860. /// 缓冲区的长度
  861. pub iov_len: usize,
  862. }
  863. /// 用于存储多个来自用户空间的IoVec
  864. ///
  865. /// 由于目前内核中的文件系统还不支持分散读写,所以暂时只支持将用户空间的IoVec聚合成一个缓冲区,然后进行操作。
  866. /// TODO:支持分散读写
  867. #[derive(Debug)]
  868. pub struct IoVecs(Vec<&'static mut [u8]>);
  869. impl IoVecs {
  870. /// 从用户空间的IoVec中构造IoVecs
  871. ///
  872. /// @param iov 用户空间的IoVec
  873. /// @param iovcnt 用户空间的IoVec的数量
  874. /// @param readv 是否为readv系统调用
  875. ///
  876. /// @return 构造成功返回IoVecs,否则返回错误码
  877. pub unsafe fn from_user(
  878. iov: *const IoVec,
  879. iovcnt: usize,
  880. _readv: bool,
  881. ) -> Result<Self, SystemError> {
  882. // 检查iov指针所在空间是否合法
  883. if !verify_area(
  884. iov as usize as u64,
  885. (iovcnt * core::mem::size_of::<IoVec>()) as u64,
  886. ) {
  887. return Err(SystemError::EFAULT);
  888. }
  889. // 将用户空间的IoVec转换为引用(注意:这里的引用是静态的,因为用户空间的IoVec不会被释放)
  890. let iovs: &[IoVec] = core::slice::from_raw_parts(iov, iovcnt);
  891. let mut slices: Vec<&mut [u8]> = vec![];
  892. slices.reserve(iovs.len());
  893. for iov in iovs.iter() {
  894. if iov.iov_len == 0 {
  895. continue;
  896. }
  897. if !verify_area(iov.iov_base as usize as u64, iov.iov_len as u64) {
  898. return Err(SystemError::EFAULT);
  899. }
  900. slices.push(core::slice::from_raw_parts_mut(iov.iov_base, iov.iov_len));
  901. }
  902. return Ok(Self(slices));
  903. }
  904. /// @brief 将IoVecs中的数据聚合到一个缓冲区中
  905. ///
  906. /// @return 返回聚合后的缓冲区
  907. pub fn gather(&self) -> Vec<u8> {
  908. let mut buf = Vec::new();
  909. for slice in self.0.iter() {
  910. buf.extend_from_slice(slice);
  911. }
  912. return buf;
  913. }
  914. /// @brief 将给定的数据分散写入到IoVecs中
  915. pub fn scatter(&mut self, data: &[u8]) {
  916. let mut data: &[u8] = data;
  917. for slice in self.0.iter_mut() {
  918. let len = core::cmp::min(slice.len(), data.len());
  919. if len == 0 {
  920. continue;
  921. }
  922. slice[..len].copy_from_slice(&data[..len]);
  923. data = &data[len..];
  924. }
  925. }
  926. /// @brief 创建与IoVecs等长的缓冲区
  927. ///
  928. /// @param set_len 是否设置返回的Vec的len。
  929. /// 如果为true,则返回的Vec的len为所有IoVec的长度之和;
  930. /// 否则返回的Vec的len为0,capacity为所有IoVec的长度之和.
  931. ///
  932. /// @return 返回创建的缓冲区
  933. pub fn new_buf(&self, set_len: bool) -> Vec<u8> {
  934. let total_len: usize = self.0.iter().map(|slice| slice.len()).sum();
  935. let mut buf: Vec<u8> = Vec::with_capacity(total_len);
  936. if set_len {
  937. buf.resize(total_len, 0);
  938. }
  939. return buf;
  940. }
  941. }