image.rs

use std::fs::File;
use std::io::{Read, Write, Seek, SeekFrom};

use redoxfs::Disk;

use system::error::{Error, Result, EIO};

macro_rules! try_disk {
    ($expr:expr) => (match $expr {
        Ok(val) => val,
        Err(err) => {
            println!("Disk I/O Error: {}", err);
            return Err(Error::new(EIO));
        }
    })
}

pub struct Image {
    file: File
}

impl Image {
    pub fn open(path: &str) -> Result<Image> {
        let file = try_disk!(File::open(path));
        Ok(Image {
            file: file
        })
    }
}

impl Disk for Image {
    fn read_at(&mut self, block: u64, buffer: &mut [u8]) -> Result<usize> {
        //println!("Image read at {}", block);
        try_disk!(self.file.seek(SeekFrom::Start(block * 512)));
        let count = try_disk!(self.file.read(buffer));
        Ok(count)
    }

    fn write_at(&mut self, block: u64, buffer: &[u8]) -> Result<usize> {
        //println!("Image write at {}", block);
        try_disk!(self.file.seek(SeekFrom::Start(block * 512)));
        let count = try_disk!(self.file.write(buffer));
        Ok(count)
    }

    fn size(&mut self) -> Result<u64> {
        let size = try_disk!(self.file.seek(SeekFrom::End(0)));
        Ok(size)
    }
}