From fd125947c9db0b33761414e65e919f73d9bf1815 Mon Sep 17 00:00:00 2001
From: Chuyan Zhang <me@zcy.moe>
Date: Thu, 30 Nov 2023 12:01:11 -0800
Subject: Refactor workspace

---
 mkfs.aya/src/block_device.rs |  5 -----
 mkfs.aya/src/ioctl.rs        |  6 ++++++
 mkfs.aya/src/main.rs         | 43 ++++++++++++++++++++++++++++++++++---------
 3 files changed, 40 insertions(+), 14 deletions(-)
 delete mode 100644 mkfs.aya/src/block_device.rs
 create mode 100644 mkfs.aya/src/ioctl.rs

(limited to 'mkfs.aya/src')

diff --git a/mkfs.aya/src/block_device.rs b/mkfs.aya/src/block_device.rs
deleted file mode 100644
index 295b357..0000000
--- a/mkfs.aya/src/block_device.rs
+++ /dev/null
@@ -1,5 +0,0 @@
-pub(crate) const BLOCK_SIZE: usize = 4096;
-pub(crate) trait BlockDevice {
-    fn read(&self, block_id: usize, buffer: &mut [u8]);
-    fn write(&self, block_id: usize, buffer: &[u8]);
-}
\ No newline at end of file
diff --git a/mkfs.aya/src/ioctl.rs b/mkfs.aya/src/ioctl.rs
new file mode 100644
index 0000000..7327f4b
--- /dev/null
+++ b/mkfs.aya/src/ioctl.rs
@@ -0,0 +1,6 @@
+use nix::ioctl_read;
+
+const BLKGETSIZE64_CODE: u8 = 0x12;
+const BLKGETSIZE64_SEQ: u8 = 114;
+
+ioctl_read!(ioctl_blkgetsize64, BLKGETSIZE64_CODE, BLKGETSIZE64_SEQ, u64);
\ No newline at end of file
diff --git a/mkfs.aya/src/main.rs b/mkfs.aya/src/main.rs
index 57ca3fc..3a96d0c 100644
--- a/mkfs.aya/src/main.rs
+++ b/mkfs.aya/src/main.rs
@@ -1,21 +1,46 @@
-mod block_device;
+mod ioctl;
 
-use std::fs::File;
-use std::path::PathBuf;
 use clap::Parser;
-use users::{get_current_gid, get_current_uid};
+use std::fs::File;
+use std::os::fd::AsRawFd;
+use std::path::{Path, PathBuf};
+use std::sync::Arc;
+use aya::AyaFS;
+use aya::block_device::{BLOCK_SIZE, BlockDevice};
+use aya::block_device::disk::Disk;
+use crate::ioctl::ioctl_blkgetsize64;
 
 #[derive(Parser, Debug)]
 #[command(author, version, about)]
 struct Args {
     block_device: Option<PathBuf>,
-    start_point: Option<usize>,
-    length: Option<usize>,
 }
 
+fn get_device_size<T: AsRef<Path>>(path: T) -> u64 {
+    let device = File::options()
+        .write(true)
+        .open(path.as_ref())
+        .unwrap();
+    let device_raw_fd = device.as_raw_fd();
+    let mut device_size = 0u64;
+    let device_size_ptr = &mut device_size as *mut u64;
+    unsafe {
+        ioctl_blkgetsize64(device_raw_fd, device_size_ptr).unwrap();
+    }
+    device_size
+}
+
+
+
 fn main() {
     let args = Args::parse();
     let device_path = args.block_device.unwrap();
-    println!("{:?}", device_path.as_path());
-    let mut device = File::open(device_path.as_path()).unwrap();
-}
\ No newline at end of file
+    let device_size = get_device_size(device_path.as_path());
+    let device = File::options()
+        .read(true)
+        .write(true)
+        .open(device_path.as_path())
+        .unwrap();
+    let disk =Arc::new( Disk::new(device_path));
+    let fs = AyaFS::new(disk, device_size as usize / BLOCK_SIZE);
+}
-- 
cgit v1.2.3-70-g09d2