basic FileSystem trait

Signed-off-by: Julien CLEMENT <julien.clement@epita.fr>

src/fd/mod.rs

@@ -33,7 +33,7 @@ impl FDTable {
         FDTable { table: BTreeMap::new() }
     }
 
-    pub async fn register_fd(&mut self, fd: FDt) {
+    pub fn register_fd(&mut self, fd: FDt) {
         // TODO
         self.table.insert(fd.borrow().get_fd(), fd.clone());
         println!("Registered fd: {:?}", self.table.get(&FDId(1)).unwrap().borrow().get_fd());

src/fs/iso/fd.rs

@@ -1,5 +1,6 @@
 use crate::println;
 use crate::fd::{FDId, FileDescriptor, FDt};
+use crate::utils::mutex::AsyncMutex;
 
 use alloc::{sync::Arc, boxed::Box};
 use async_trait::async_trait;

src/fs/iso/mod.rs

@@ -6,21 +6,19 @@ use crate::drivers::atapi::{DRIVE};
 use crate::fd::{FD_TABLE, FDt};
 use crate::utils::unserialize;
 
+use super::FileSystem;
 use iso9660::{IsoPrimVolDesc, IsoDir};
 use fd::IsoFD;
 
-pub async fn get_prim_vol_desc() -> IsoPrimVolDesc {
-    let desc_block = DRIVE
-        .lock()
-        .await
-        .as_mut()
-        .unwrap()
-        .read_block(iso9660::ISO_PRIM_VOLDESC_BLOCK)
-        .await;
-    *unserialize::<IsoPrimVolDesc>(desc_block.as_ptr())
+use alloc::{sync::Arc, boxed::Box};
+use async_trait::async_trait;
+
+pub struct IsoFS {
 }
 
-pub async fn open(path: &str, flags: u32) -> Option<FDt> {
+#[async_trait(?Send)]
+impl FileSystem for IsoFS {
+    async fn open(path: &str, flags: u32) -> Option<FDt> {
         if flags != crate::syscalls::io::O_RDONLY {
             return None;
         }
@@ -32,6 +30,18 @@ pub async fn open(path: &str, flags: u32) -> Option<FDt> {
         }
 
         let fd = IsoFD::new();
-    FD_TABLE.lock().await.register_fd(fd.clone()).await;
-    Some(fd)
+        FD_TABLE.lock().await.register_fd(fd.clone());
+        Some(fd.clone())
+    }
+}
+
+pub async fn get_prim_vol_desc() -> IsoPrimVolDesc {
+    let desc_block = DRIVE
+        .lock()
+        .await
+        .as_mut()
+        .unwrap()
+        .read_block(iso9660::ISO_PRIM_VOLDESC_BLOCK)
+        .await;
+    *unserialize::<IsoPrimVolDesc>(desc_block.as_ptr())
 }

src/fs/mod.rs

@@ -1 +1,11 @@
 pub mod iso;
+
+use crate::fd::FDt;
+
+use alloc::{sync::Arc, boxed::Box};
+use async_trait::async_trait;
+
+#[async_trait(?Send)]
+pub trait FileSystem {
+    async fn open(path: &str, flags: u32) -> Option<FDt>;
+}

src/lib.rs

@@ -20,6 +20,7 @@ use core::panic::PanicInfo;
 use drivers::vga::{self, Color, ColorCode};
 use multiboot2::BootInformation;
 use task::{executor::Executor, keyboard, Task};
+use crate::fs::FileSystem;
 
 #[alloc_error_handler]
 fn alloc_error_handler(layout: alloc::alloc::Layout) -> ! {
@@ -65,6 +66,6 @@ pub extern "C" fn julios_main(multiboot_info_addr: usize) -> ! {
 
 
 async fn get_file() {
-    let fd = fs::iso::open("test", syscalls::io::O_RDONLY).await.unwrap();
+    let fd = fs::iso::IsoFS::open("test", syscalls::io::O_RDONLY).await.unwrap();
     fd.borrow_mut().read(&[], 0).await;
 }

src/utils/mutex.rs

@@ -14,7 +14,7 @@ struct Lock {
     waker: Arc<AtomicWaker>,
 }
 
-pub struct AsyncMutex<T> {
+pub struct AsyncMutex<T: ?Sized> {
     lock: Lock,
     inner: UnsafeCell<T>,
 }