ayafs/src/block_device/disk.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use std::cell::RefCell;
use std::fs::File;
use std::io::{Read, Seek, SeekFrom, Write};
use std::path::{Path, PathBuf};
use crate::block_device::{BLOCK_SIZE, BlockDevice};

pub struct Disk {
    disk_path: PathBuf,
    device: RefCell<File>,
}

impl Disk {
    pub fn new(disk_path: PathBuf) -> Self {
        let device = File::options()
            .read(true)
            .write(true)
            .open(disk_path.as_path())
            .unwrap();
        // let device = File::open(disk_path.as_path()).unwrap();
        Self {
            disk_path,
            device: RefCell::new(device),
        }
    }
}

impl BlockDevice for Disk {
    fn read(&self, block_id: usize, buffer: &mut [u8]) {
        let mut device = self.device.borrow_mut();
        device
            .seek(SeekFrom::Start((block_id * BLOCK_SIZE) as u64))
            .expect("Unable to seek!");
        device
            .read_exact(buffer)
            .expect("Failed to read 4096 bytes!");
    }

    fn write(&self, block_id: usize, buffer: &[u8]) {
        let mut device = self.device.borrow_mut();
        device
            .seek(SeekFrom::Start((block_id * BLOCK_SIZE) as u64))
            .expect("Unable to seek!");
        device
            .write_all(buffer)
            .expect("Unable to write 4096 bytes!");
    }
}