mod.rs 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827
  1. use core::intrinsics::size_of;
  2. use alloc::{
  3. borrow::ToOwned,
  4. collections::BTreeMap,
  5. format,
  6. string::{String, ToString},
  7. sync::{Arc, Weak},
  8. vec::Vec,
  9. };
  10. use system_error::SystemError;
  11. use crate::{
  12. arch::mm::LockedFrameAllocator,
  13. driver::base::device::device_number::DeviceNumber,
  14. filesystem::vfs::{
  15. core::{generate_inode_id, ROOT_INODE},
  16. FileType,
  17. },
  18. kerror, kinfo,
  19. libs::{
  20. once::Once,
  21. rwlock::RwLock,
  22. spinlock::{SpinLock, SpinLockGuard},
  23. },
  24. mm::allocator::page_frame::FrameAllocator,
  25. process::{Pid, ProcessManager},
  26. time::TimeSpec,
  27. };
  28. use super::vfs::{
  29. file::{FileMode, FilePrivateData},
  30. syscall::ModeType,
  31. FileSystem, FsInfo, IndexNode, InodeId, Magic, Metadata, SuperBlock,
  32. };
  33. pub mod kmsg;
  34. pub mod log;
  35. mod syscall;
  36. /// @brief 进程文件类型
  37. /// @usage 用于定义进程文件夹下的各类文件类型
  38. #[derive(Debug)]
  39. #[repr(u8)]
  40. pub enum ProcFileType {
  41. ///展示进程状态信息
  42. ProcStatus = 0,
  43. /// meminfo
  44. ProcMeminfo = 1,
  45. /// kmsg
  46. ProcKmsg = 2,
  47. //todo: 其他文件类型
  48. ///默认文件类型
  49. Default,
  50. }
  51. impl From<u8> for ProcFileType {
  52. fn from(value: u8) -> Self {
  53. match value {
  54. 0 => ProcFileType::ProcStatus,
  55. 1 => ProcFileType::ProcMeminfo,
  56. 2 => ProcFileType::ProcKmsg,
  57. _ => ProcFileType::Default,
  58. }
  59. }
  60. }
  61. /// @brief 节点私有信息结构体
  62. /// @usage 用于传入各类文件所需的信息
  63. #[derive(Debug)]
  64. pub struct InodeInfo {
  65. ///进程的pid
  66. pid: Pid,
  67. ///文件类型
  68. ftype: ProcFileType,
  69. //其他需要传入的信息在此定义
  70. }
  71. /// @brief procfs的inode名称的最大长度
  72. const PROCFS_MAX_NAMELEN: usize = 64;
  73. const PROCFS_BLOCK_SIZE: u64 = 512;
  74. /// @brief procfs文件系统的Inode结构体
  75. #[derive(Debug)]
  76. pub struct LockedProcFSInode(SpinLock<ProcFSInode>);
  77. /// @brief procfs文件系统结构体
  78. #[derive(Debug)]
  79. pub struct ProcFS {
  80. /// procfs的root inode
  81. root_inode: Arc<LockedProcFSInode>,
  82. super_block: RwLock<SuperBlock>,
  83. }
  84. #[derive(Debug, Clone)]
  85. pub struct ProcfsFilePrivateData {
  86. data: Vec<u8>,
  87. }
  88. impl ProcfsFilePrivateData {
  89. pub fn new() -> Self {
  90. return ProcfsFilePrivateData { data: Vec::new() };
  91. }
  92. }
  93. impl Default for ProcfsFilePrivateData {
  94. fn default() -> Self {
  95. Self::new()
  96. }
  97. }
  98. /// @brief procfs文件系统的Inode结构体(不包含锁)
  99. #[derive(Debug)]
  100. pub struct ProcFSInode {
  101. /// 指向父Inode的弱引用
  102. parent: Weak<LockedProcFSInode>,
  103. /// 指向自身的弱引用
  104. self_ref: Weak<LockedProcFSInode>,
  105. /// 子Inode的B树
  106. children: BTreeMap<String, Arc<LockedProcFSInode>>,
  107. /// 当前inode的数据部分
  108. data: Vec<u8>,
  109. /// 当前inode的元数据
  110. metadata: Metadata,
  111. /// 指向inode所在的文件系统对象的指针
  112. fs: Weak<ProcFS>,
  113. /// 储存私有信息
  114. fdata: InodeInfo,
  115. }
  116. /// 对ProcFSInode实现获取各类文件信息的函数
  117. impl ProcFSInode {
  118. /// @brief 去除Vec中所有的\0,并在结尾添加\0
  119. #[inline]
  120. fn trim_string(&self, data: &mut Vec<u8>) {
  121. data.retain(|x| *x != 0);
  122. data.push(0);
  123. }
  124. // todo:其他数据获取函数实现
  125. /// @brief 打开status文件
  126. ///
  127. fn open_status(&self, pdata: &mut ProcfsFilePrivateData) -> Result<i64, SystemError> {
  128. // 获取该pid对应的pcb结构体
  129. let pid = self.fdata.pid;
  130. let pcb = ProcessManager::find(pid);
  131. let pcb = if let Some(pcb) = pcb {
  132. pcb
  133. } else {
  134. kerror!(
  135. "ProcFS: Cannot find pcb for pid {:?} when opening its 'status' file.",
  136. pid
  137. );
  138. return Err(SystemError::ESRCH);
  139. };
  140. // 传入数据
  141. let pdata: &mut Vec<u8> = &mut pdata.data;
  142. pdata.append(
  143. &mut format!("Name:\t{}", pcb.basic().name())
  144. .as_bytes()
  145. .to_owned(),
  146. );
  147. let sched_info_guard = pcb.sched_info();
  148. let state = sched_info_guard.inner_lock_read_irqsave().state();
  149. let cpu_id = sched_info_guard
  150. .on_cpu()
  151. .map(|cpu| cpu.data() as i32)
  152. .unwrap_or(-1);
  153. let priority = sched_info_guard.priority();
  154. let vrtime = sched_info_guard.virtual_runtime();
  155. pdata.append(&mut format!("\nState:\t{:?}", state).as_bytes().to_owned());
  156. pdata.append(
  157. &mut format!("\nPid:\t{}", pcb.pid().into())
  158. .as_bytes()
  159. .to_owned(),
  160. );
  161. pdata.append(
  162. &mut format!("\nPpid:\t{}", pcb.basic().ppid().into())
  163. .as_bytes()
  164. .to_owned(),
  165. );
  166. pdata.append(&mut format!("\ncpu_id:\t{}", cpu_id).as_bytes().to_owned());
  167. pdata.append(
  168. &mut format!("\npriority:\t{}", priority.data())
  169. .as_bytes()
  170. .to_owned(),
  171. );
  172. pdata.append(
  173. &mut format!("\npreempt:\t{}", pcb.preempt_count())
  174. .as_bytes()
  175. .to_owned(),
  176. );
  177. pdata.append(&mut format!("\nvrtime:\t{}", vrtime).as_bytes().to_owned());
  178. if let Some(user_vm) = pcb.basic().user_vm() {
  179. let address_space_guard = user_vm.read();
  180. // todo: 当前进程运行过程中占用内存的峰值
  181. let hiwater_vm: u64 = 0;
  182. // 进程代码段的大小
  183. let text = (address_space_guard.end_code - address_space_guard.start_code) / 1024;
  184. // 进程数据段的大小
  185. let data = (address_space_guard.end_data - address_space_guard.start_data) / 1024;
  186. drop(address_space_guard);
  187. pdata.append(
  188. &mut format!("\nVmPeak:\t{} kB", hiwater_vm)
  189. .as_bytes()
  190. .to_owned(),
  191. );
  192. pdata.append(&mut format!("\nVmData:\t{} kB", data).as_bytes().to_owned());
  193. pdata.append(&mut format!("\nVmExe:\t{} kB", text).as_bytes().to_owned());
  194. }
  195. pdata.append(
  196. &mut format!("\nflags: {:?}\n", pcb.flags().clone())
  197. .as_bytes()
  198. .to_owned(),
  199. );
  200. // 去除多余的\0
  201. self.trim_string(pdata);
  202. return Ok((pdata.len() * size_of::<u8>()) as i64);
  203. }
  204. /// 打开 meminfo 文件
  205. fn open_meminfo(&self, pdata: &mut ProcfsFilePrivateData) -> Result<i64, SystemError> {
  206. // 获取内存信息
  207. let usage = unsafe { LockedFrameAllocator.usage() };
  208. // 传入数据
  209. let data: &mut Vec<u8> = &mut pdata.data;
  210. data.append(
  211. &mut format!("MemTotal:\t{} kB\n", usage.total().bytes() >> 10)
  212. .as_bytes()
  213. .to_owned(),
  214. );
  215. data.append(
  216. &mut format!("MemFree:\t{} kB\n", usage.free().bytes() >> 10)
  217. .as_bytes()
  218. .to_owned(),
  219. );
  220. // 去除多余的\0
  221. self.trim_string(data);
  222. return Ok((data.len() * size_of::<u8>()) as i64);
  223. }
  224. /// proc文件系统读取函数
  225. fn proc_read(
  226. &self,
  227. offset: usize,
  228. len: usize,
  229. buf: &mut [u8],
  230. pdata: &mut ProcfsFilePrivateData,
  231. ) -> Result<usize, SystemError> {
  232. let start = pdata.data.len().min(offset);
  233. let end = pdata.data.len().min(offset + len);
  234. // buffer空间不足
  235. if buf.len() < (end - start) {
  236. return Err(SystemError::ENOBUFS);
  237. }
  238. // 拷贝数据
  239. let src = &pdata.data[start..end];
  240. buf[0..src.len()].copy_from_slice(src);
  241. return Ok(src.len());
  242. }
  243. }
  244. impl FileSystem for ProcFS {
  245. fn root_inode(&self) -> Arc<dyn super::vfs::IndexNode> {
  246. return self.root_inode.clone();
  247. }
  248. fn info(&self) -> FsInfo {
  249. return FsInfo {
  250. blk_dev_id: 0,
  251. max_name_len: PROCFS_MAX_NAMELEN,
  252. };
  253. }
  254. fn as_any_ref(&self) -> &dyn core::any::Any {
  255. self
  256. }
  257. fn name(&self) -> &str {
  258. "procfs"
  259. }
  260. fn super_block(&self) -> SuperBlock {
  261. self.super_block.read().clone()
  262. }
  263. }
  264. impl ProcFS {
  265. pub fn new() -> Arc<Self> {
  266. let super_block = SuperBlock::new(
  267. Magic::PROC_MAGIC,
  268. PROCFS_BLOCK_SIZE,
  269. PROCFS_MAX_NAMELEN as u64,
  270. );
  271. // 初始化root inode
  272. let root: Arc<LockedProcFSInode> =
  273. Arc::new(LockedProcFSInode(SpinLock::new(ProcFSInode {
  274. parent: Weak::default(),
  275. self_ref: Weak::default(),
  276. children: BTreeMap::new(),
  277. data: Vec::new(),
  278. metadata: Metadata {
  279. dev_id: 0,
  280. inode_id: generate_inode_id(),
  281. size: 0,
  282. blk_size: 0,
  283. blocks: 0,
  284. atime: TimeSpec::default(),
  285. mtime: TimeSpec::default(),
  286. ctime: TimeSpec::default(),
  287. file_type: FileType::Dir,
  288. mode: ModeType::from_bits_truncate(0o555),
  289. nlinks: 1,
  290. uid: 0,
  291. gid: 0,
  292. raw_dev: DeviceNumber::default(),
  293. },
  294. fs: Weak::default(),
  295. fdata: InodeInfo {
  296. pid: Pid::new(0),
  297. ftype: ProcFileType::Default,
  298. },
  299. })));
  300. let result: Arc<ProcFS> = Arc::new(ProcFS {
  301. root_inode: root,
  302. super_block: RwLock::new(super_block),
  303. });
  304. // 对root inode加锁,并继续完成初始化工作
  305. let mut root_guard: SpinLockGuard<ProcFSInode> = result.root_inode.0.lock();
  306. root_guard.parent = Arc::downgrade(&result.root_inode);
  307. root_guard.self_ref = Arc::downgrade(&result.root_inode);
  308. root_guard.fs = Arc::downgrade(&result);
  309. // 释放锁
  310. drop(root_guard);
  311. // 创建meminfo文件
  312. let inode = result.root_inode();
  313. let binding = inode.create(
  314. "meminfo",
  315. FileType::File,
  316. ModeType::from_bits_truncate(0o444),
  317. );
  318. if let Ok(meminfo) = binding {
  319. let meminfo_file = meminfo
  320. .as_any_ref()
  321. .downcast_ref::<LockedProcFSInode>()
  322. .unwrap();
  323. meminfo_file.0.lock().fdata.pid = Pid::new(0);
  324. meminfo_file.0.lock().fdata.ftype = ProcFileType::ProcMeminfo;
  325. } else {
  326. panic!("create meminfo error");
  327. }
  328. // 创建kmsg文件
  329. let binding = inode.create("kmsg", FileType::File, ModeType::from_bits_truncate(0o444));
  330. if let Ok(kmsg) = binding {
  331. let kmsg_file = kmsg
  332. .as_any_ref()
  333. .downcast_ref::<LockedProcFSInode>()
  334. .unwrap();
  335. kmsg_file.0.lock().fdata.pid = Pid::new(1);
  336. kmsg_file.0.lock().fdata.ftype = ProcFileType::ProcKmsg;
  337. } else {
  338. panic!("create ksmg error");
  339. }
  340. return result;
  341. }
  342. /// @brief 进程注册函数
  343. /// @usage 在进程中调用并创建进程对应文件
  344. pub fn register_pid(&self, pid: Pid) -> Result<(), SystemError> {
  345. // 获取当前inode
  346. let inode: Arc<dyn IndexNode> = self.root_inode();
  347. // 创建对应进程文件夹
  348. let pid_dir: Arc<dyn IndexNode> = inode.create(
  349. &pid.to_string(),
  350. FileType::Dir,
  351. ModeType::from_bits_truncate(0o555),
  352. )?;
  353. // 创建相关文件
  354. // status文件
  355. let binding: Arc<dyn IndexNode> = pid_dir.create(
  356. "status",
  357. FileType::File,
  358. ModeType::from_bits_truncate(0o444),
  359. )?;
  360. let status_file: &LockedProcFSInode = binding
  361. .as_any_ref()
  362. .downcast_ref::<LockedProcFSInode>()
  363. .unwrap();
  364. status_file.0.lock().fdata.pid = pid;
  365. status_file.0.lock().fdata.ftype = ProcFileType::ProcStatus;
  366. //todo: 创建其他文件
  367. return Ok(());
  368. }
  369. /// @brief 解除进程注册
  370. ///
  371. pub fn unregister_pid(&self, pid: Pid) -> Result<(), SystemError> {
  372. // 获取当前inode
  373. let proc: Arc<dyn IndexNode> = self.root_inode();
  374. // 获取进程文件夹
  375. let pid_dir: Arc<dyn IndexNode> = proc.find(&pid.to_string())?;
  376. // 删除进程文件夹下文件
  377. pid_dir.unlink("status")?;
  378. // 查看进程文件是否还存在
  379. // let pf= pid_dir.find("status").expect("Cannot find status");
  380. // 删除进程文件夹
  381. proc.unlink(&pid.to_string())?;
  382. return Ok(());
  383. }
  384. }
  385. impl IndexNode for LockedProcFSInode {
  386. fn open(
  387. &self,
  388. mut data: SpinLockGuard<FilePrivateData>,
  389. _mode: &FileMode,
  390. ) -> Result<(), SystemError> {
  391. // 加锁
  392. let mut inode: SpinLockGuard<ProcFSInode> = self.0.lock();
  393. // 如果inode类型为文件夹,则直接返回成功
  394. if let FileType::Dir = inode.metadata.file_type {
  395. return Ok(());
  396. }
  397. let mut private_data = ProcfsFilePrivateData::new();
  398. // 根据文件类型获取相应数据
  399. let file_size = match inode.fdata.ftype {
  400. ProcFileType::ProcStatus => inode.open_status(&mut private_data)?,
  401. ProcFileType::ProcMeminfo => inode.open_meminfo(&mut private_data)?,
  402. _ => {
  403. todo!()
  404. }
  405. };
  406. *data = FilePrivateData::Procfs(private_data);
  407. // 更新metadata里面的文件大小数值
  408. inode.metadata.size = file_size;
  409. return Ok(());
  410. }
  411. fn close(&self, mut data: SpinLockGuard<FilePrivateData>) -> Result<(), SystemError> {
  412. let guard: SpinLockGuard<ProcFSInode> = self.0.lock();
  413. // 如果inode类型为文件夹,则直接返回成功
  414. if let FileType::Dir = guard.metadata.file_type {
  415. return Ok(());
  416. }
  417. // 释放data
  418. *data = FilePrivateData::Procfs(ProcfsFilePrivateData::new());
  419. return Ok(());
  420. }
  421. fn read_at(
  422. &self,
  423. offset: usize,
  424. len: usize,
  425. buf: &mut [u8],
  426. data: SpinLockGuard<FilePrivateData>,
  427. ) -> Result<usize, SystemError> {
  428. if buf.len() < len {
  429. return Err(SystemError::EINVAL);
  430. }
  431. // 加锁
  432. let inode: SpinLockGuard<ProcFSInode> = self.0.lock();
  433. // 检查当前inode是否为一个文件夹,如果是的话,就返回错误
  434. if inode.metadata.file_type == FileType::Dir {
  435. return Err(SystemError::EISDIR);
  436. }
  437. // 获取数据信息
  438. let mut private_data = match &*data {
  439. FilePrivateData::Procfs(p) => p.clone(),
  440. _ => {
  441. panic!("ProcFS: FilePrivateData mismatch!");
  442. }
  443. };
  444. // 根据文件类型读取相应数据
  445. match inode.fdata.ftype {
  446. ProcFileType::ProcStatus => {
  447. return inode.proc_read(offset, len, buf, &mut private_data)
  448. }
  449. ProcFileType::ProcMeminfo => {
  450. return inode.proc_read(offset, len, buf, &mut private_data)
  451. }
  452. ProcFileType::ProcKmsg => (),
  453. ProcFileType::Default => (),
  454. };
  455. // 默认读取
  456. let start = inode.data.len().min(offset);
  457. let end = inode.data.len().min(offset + len);
  458. // buffer空间不足
  459. if buf.len() < (end - start) {
  460. return Err(SystemError::ENOBUFS);
  461. }
  462. // 拷贝数据
  463. let src = &inode.data[start..end];
  464. buf[0..src.len()].copy_from_slice(src);
  465. return Ok(src.len());
  466. }
  467. fn write_at(
  468. &self,
  469. _offset: usize,
  470. _len: usize,
  471. _buf: &[u8],
  472. _data: SpinLockGuard<FilePrivateData>,
  473. ) -> Result<usize, SystemError> {
  474. return Err(SystemError::EOPNOTSUPP_OR_ENOTSUP);
  475. }
  476. fn fs(&self) -> Arc<dyn FileSystem> {
  477. return self.0.lock().fs.upgrade().unwrap();
  478. }
  479. fn as_any_ref(&self) -> &dyn core::any::Any {
  480. self
  481. }
  482. fn metadata(&self) -> Result<Metadata, SystemError> {
  483. let inode = self.0.lock();
  484. let metadata = inode.metadata.clone();
  485. return Ok(metadata);
  486. }
  487. fn set_metadata(&self, metadata: &Metadata) -> Result<(), SystemError> {
  488. let mut inode = self.0.lock();
  489. inode.metadata.atime = metadata.atime;
  490. inode.metadata.mtime = metadata.mtime;
  491. inode.metadata.ctime = metadata.ctime;
  492. inode.metadata.mode = metadata.mode;
  493. inode.metadata.uid = metadata.uid;
  494. inode.metadata.gid = metadata.gid;
  495. return Ok(());
  496. }
  497. fn resize(&self, len: usize) -> Result<(), SystemError> {
  498. let mut inode = self.0.lock();
  499. if inode.metadata.file_type == FileType::File {
  500. inode.data.resize(len, 0);
  501. return Ok(());
  502. } else {
  503. return Err(SystemError::EINVAL);
  504. }
  505. }
  506. fn create_with_data(
  507. &self,
  508. name: &str,
  509. file_type: FileType,
  510. mode: ModeType,
  511. data: usize,
  512. ) -> Result<Arc<dyn IndexNode>, SystemError> {
  513. // 获取当前inode
  514. let mut inode = self.0.lock();
  515. // 如果当前inode不是文件夹,则返回
  516. if inode.metadata.file_type != FileType::Dir {
  517. return Err(SystemError::ENOTDIR);
  518. }
  519. // 如果有重名的,则返回
  520. if inode.children.contains_key(name) {
  521. return Err(SystemError::EEXIST);
  522. }
  523. // 创建inode
  524. let result: Arc<LockedProcFSInode> =
  525. Arc::new(LockedProcFSInode(SpinLock::new(ProcFSInode {
  526. parent: inode.self_ref.clone(),
  527. self_ref: Weak::default(),
  528. children: BTreeMap::new(),
  529. data: Vec::new(),
  530. metadata: Metadata {
  531. dev_id: 0,
  532. inode_id: generate_inode_id(),
  533. size: 0,
  534. blk_size: 0,
  535. blocks: 0,
  536. atime: TimeSpec::default(),
  537. mtime: TimeSpec::default(),
  538. ctime: TimeSpec::default(),
  539. file_type,
  540. mode,
  541. nlinks: 1,
  542. uid: 0,
  543. gid: 0,
  544. raw_dev: DeviceNumber::from(data as u32),
  545. },
  546. fs: inode.fs.clone(),
  547. fdata: InodeInfo {
  548. pid: Pid::new(0),
  549. ftype: ProcFileType::Default,
  550. },
  551. })));
  552. // 初始化inode的自引用的weak指针
  553. result.0.lock().self_ref = Arc::downgrade(&result);
  554. // 将子inode插入父inode的B树中
  555. inode.children.insert(String::from(name), result.clone());
  556. return Ok(result);
  557. }
  558. fn link(&self, name: &str, other: &Arc<dyn IndexNode>) -> Result<(), SystemError> {
  559. let other: &LockedProcFSInode = other
  560. .downcast_ref::<LockedProcFSInode>()
  561. .ok_or(SystemError::EPERM)?;
  562. let mut inode: SpinLockGuard<ProcFSInode> = self.0.lock();
  563. let mut other_locked: SpinLockGuard<ProcFSInode> = other.0.lock();
  564. // 如果当前inode不是文件夹,那么报错
  565. if inode.metadata.file_type != FileType::Dir {
  566. return Err(SystemError::ENOTDIR);
  567. }
  568. // 如果另一个inode是文件夹,那么也报错
  569. if other_locked.metadata.file_type == FileType::Dir {
  570. return Err(SystemError::EISDIR);
  571. }
  572. // 如果当前文件夹下已经有同名文件,也报错。
  573. if inode.children.contains_key(name) {
  574. return Err(SystemError::EEXIST);
  575. }
  576. inode
  577. .children
  578. .insert(String::from(name), other_locked.self_ref.upgrade().unwrap());
  579. // 增加硬链接计数
  580. other_locked.metadata.nlinks += 1;
  581. return Ok(());
  582. }
  583. fn unlink(&self, name: &str) -> Result<(), SystemError> {
  584. let mut inode: SpinLockGuard<ProcFSInode> = self.0.lock();
  585. // 如果当前inode不是目录,那么也没有子目录/文件的概念了,因此要求当前inode的类型是目录
  586. if inode.metadata.file_type != FileType::Dir {
  587. return Err(SystemError::ENOTDIR);
  588. }
  589. // 不允许删除当前文件夹,也不允许删除上一个目录
  590. if name == "." || name == ".." {
  591. return Err(SystemError::ENOTEMPTY);
  592. }
  593. // 获得要删除的文件的inode
  594. let to_delete = inode.children.get(name).ok_or(SystemError::ENOENT)?;
  595. // 减少硬链接计数
  596. to_delete.0.lock().metadata.nlinks -= 1;
  597. // 在当前目录中删除这个子目录项
  598. inode.children.remove(name);
  599. return Ok(());
  600. }
  601. fn move_to(
  602. &self,
  603. _old_name: &str,
  604. _target: &Arc<dyn IndexNode>,
  605. _new_name: &str,
  606. ) -> Result<(), SystemError> {
  607. return Err(SystemError::EOPNOTSUPP_OR_ENOTSUP);
  608. }
  609. fn find(&self, name: &str) -> Result<Arc<dyn IndexNode>, SystemError> {
  610. let inode = self.0.lock();
  611. if inode.metadata.file_type != FileType::Dir {
  612. return Err(SystemError::ENOTDIR);
  613. }
  614. match name {
  615. "" | "." => {
  616. return Ok(inode.self_ref.upgrade().ok_or(SystemError::ENOENT)?);
  617. }
  618. ".." => {
  619. return Ok(inode.parent.upgrade().ok_or(SystemError::ENOENT)?);
  620. }
  621. name => {
  622. // 在子目录项中查找
  623. return Ok(inode.children.get(name).ok_or(SystemError::ENOENT)?.clone());
  624. }
  625. }
  626. }
  627. fn get_entry_name(&self, ino: InodeId) -> Result<String, SystemError> {
  628. let inode: SpinLockGuard<ProcFSInode> = self.0.lock();
  629. if inode.metadata.file_type != FileType::Dir {
  630. return Err(SystemError::ENOTDIR);
  631. }
  632. match ino.into() {
  633. 0 => {
  634. return Ok(String::from("."));
  635. }
  636. 1 => {
  637. return Ok(String::from(".."));
  638. }
  639. ino => {
  640. // 暴力遍历所有的children,判断inode id是否相同
  641. // TODO: 优化这里,这个地方性能很差!
  642. let mut key: Vec<String> = inode
  643. .children
  644. .keys()
  645. .filter(|k| {
  646. inode
  647. .children
  648. .get(*k)
  649. .unwrap()
  650. .0
  651. .lock()
  652. .metadata
  653. .inode_id
  654. .into()
  655. == ino
  656. })
  657. .cloned()
  658. .collect();
  659. match key.len() {
  660. 0=>{return Err(SystemError::ENOENT);}
  661. 1=>{return Ok(key.remove(0));}
  662. _ => panic!("Procfs get_entry_name: key.len()={key_len}>1, current inode_id={inode_id:?}, to find={to_find:?}", key_len=key.len(), inode_id = inode.metadata.inode_id, to_find=ino)
  663. }
  664. }
  665. }
  666. }
  667. fn list(&self) -> Result<Vec<String>, SystemError> {
  668. let info = self.metadata()?;
  669. if info.file_type != FileType::Dir {
  670. return Err(SystemError::ENOTDIR);
  671. }
  672. let mut keys: Vec<String> = Vec::new();
  673. keys.push(String::from("."));
  674. keys.push(String::from(".."));
  675. keys.append(&mut self.0.lock().children.keys().cloned().collect());
  676. return Ok(keys);
  677. }
  678. }
  679. /// @brief 向procfs注册进程
  680. pub fn procfs_register_pid(pid: Pid) -> Result<(), SystemError> {
  681. let procfs_inode = ROOT_INODE().find("proc")?;
  682. let procfs_inode = procfs_inode
  683. .downcast_ref::<LockedProcFSInode>()
  684. .expect("Failed to find procfs' root inode");
  685. let fs = procfs_inode.fs();
  686. let procfs: &ProcFS = fs.as_any_ref().downcast_ref::<ProcFS>().unwrap();
  687. // 调用注册函数
  688. procfs.register_pid(pid)?;
  689. return Ok(());
  690. }
  691. /// @brief 在ProcFS中,解除进程的注册
  692. pub fn procfs_unregister_pid(pid: Pid) -> Result<(), SystemError> {
  693. // 获取procfs实例
  694. let procfs_inode: Arc<dyn IndexNode> = ROOT_INODE().find("proc")?;
  695. let procfs_inode: &LockedProcFSInode = procfs_inode
  696. .downcast_ref::<LockedProcFSInode>()
  697. .expect("Failed to find procfs' root inode");
  698. let fs: Arc<dyn FileSystem> = procfs_inode.fs();
  699. let procfs: &ProcFS = fs.as_any_ref().downcast_ref::<ProcFS>().unwrap();
  700. // 调用解除注册函数
  701. return procfs.unregister_pid(pid);
  702. }
  703. pub fn procfs_init() -> Result<(), SystemError> {
  704. static INIT: Once = Once::new();
  705. let mut result = None;
  706. INIT.call_once(|| {
  707. kinfo!("Initializing ProcFS...");
  708. // 创建 procfs 实例
  709. let procfs: Arc<ProcFS> = ProcFS::new();
  710. // procfs 挂载
  711. let _t = ROOT_INODE()
  712. .find("proc")
  713. .expect("Cannot find /proc")
  714. .mount(procfs)
  715. .expect("Failed to mount proc");
  716. kinfo!("ProcFS mounted.");
  717. result = Some(Ok(()));
  718. });
  719. return result.unwrap();
  720. }