mod.rs 27 KB

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