From 4c34414b26bf71e747ea3ecb2586645bab4aba52 Mon Sep 17 00:00:00 2001
From: Chuyan Zhang <me@zcy.moe>
Date: Fri, 1 Dec 2023 19:42:13 -0800
Subject: Multiple bugfix, it works!

---
 ayafs-core/src/bin/main.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'ayafs-core/src/bin/main.rs')

diff --git a/ayafs-core/src/bin/main.rs b/ayafs-core/src/bin/main.rs
index 25504b9..a2d838c 100644
--- a/ayafs-core/src/bin/main.rs
+++ b/ayafs-core/src/bin/main.rs
@@ -9,6 +9,7 @@ use aya::block_device::disk::Disk;
 #[derive(Parser, Debug)]
 #[command(author, version, about)]
 struct Args {
+    device_path: Option<String>,
     mount_point: Option<String>,
     #[arg(short, action = clap::ArgAction::Count)]
     verbosity: u8,
@@ -20,6 +21,7 @@ struct Args {
 
 fn main() {
     let args = Args::parse();
+    let device_path = args.device_path.unwrap();
     let mount_point = args.mount_point.unwrap();
     let verbosity = args.verbosity;
     let log_level = match verbosity {
@@ -36,9 +38,8 @@ fn main() {
         MountOption::AutoUnmount,
         MountOption::AllowRoot,
     ];
-    let disk = Arc::new(Disk::new(PathBuf::from("/dev/nvme0n1p4")));
-    // let disk = Arc::new(MemoryDisk::new(16384));
-    let filesystem = AyaFS::new(disk, 7864320);
-
+    let disk = Arc::new(Disk::new(PathBuf::from(device_path)));
+    let filesystem = AyaFS::load(disk);
+    
     fuser::mount2(filesystem, mount_point, &options).unwrap();
 }
\ No newline at end of file
-- 
cgit v1.2.3-70-g09d2