From 096cc865a434a016db15d6e9fdfa717af49280cd Mon Sep 17 00:00:00 2001
From: Szymon Walter <walter.szymon.98@gmail.com>
Date: Sun, 18 Mar 2018 18:38:54 +0100
Subject: [PATCH] add locating `Superblock` and `[BlockGroupDescriptor]`

---
 src/block_group.rs |  1 +
 src/inode.rs       |  1 +
 src/superblock.rs  | 58 ++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 60 insertions(+)

diff --git a/src/block_group.rs b/src/block_group.rs
index 739fa32..5f0d49f 100644
--- a/src/block_group.rs
+++ b/src/block_group.rs
@@ -11,6 +11,7 @@
 /// Remember that blocks are numbered starting at 0, and that block numbers
 /// don't usually correspond to physical block addresses.
 #[repr(C, packed)]
+#[derive(Clone, Debug, PartialEq, Eq, Hash)]
 pub struct BlockGroupDescriptor {
     /// Block address of block usage bitmap
     block_usage_addr: u32,
diff --git a/src/inode.rs b/src/inode.rs
index 072ad45..c1f4792 100644
--- a/src/inode.rs
+++ b/src/inode.rs
@@ -1,4 +1,5 @@
 #[repr(C, packed)]
+#[derive(Clone, Debug, PartialEq, Eq, Hash)]
 pub struct Inode {
     /// Type and Permissions (see below)
     type_perm: u16,
diff --git a/src/superblock.rs b/src/superblock.rs
index 258e8b2..6fc818a 100644
--- a/src/superblock.rs
+++ b/src/superblock.rs
@@ -1,3 +1,16 @@
+#[cfg(test)]
+use std::mem;
+#[cfg(test)]
+use std::slice;
+
+#[cfg(not(test))]
+use core::mem;
+#[cfg(not(test))]
+use core::slice;
+
+use error::Error;
+use block_group::BlockGroupDescriptor;
+
 /// Ext2 signature (0xef53), used to help confirm the presence of Ext2 on a
 /// volume
 pub const EXT2_MAGIC: u16 = 0xef53;
@@ -34,6 +47,7 @@ pub const OS_LITE: u32 = 4;
 /// 512 byte sectors, the Superblock will begin at LBA 2 and will occupy all of
 /// sector 2 and 3.
 #[repr(C, packed)]
+#[derive(Clone, Debug, PartialEq, Eq, Hash)]
 pub struct Superblock {
     // taken from https://wiki.osdev.org/Ext2
     /// Total number of inodes in file system
@@ -139,6 +153,50 @@ pub struct Superblock {
 }
 
 impl Superblock {
+    pub fn find<'a>(
+        haystack: &'a mut [u8],
+    ) -> Result<&'a mut Superblock, Error> {
+        let offset = 1024;
+        let end = offset + mem::size_of::<Superblock>();
+        if haystack.len() < end {
+            return Err(Error::OutOfBounds(end));
+        }
+
+        let superblock: &mut Superblock = unsafe {
+            let ptr =
+                haystack.as_ptr().offset(offset as isize) as *mut Superblock;
+            ptr.as_mut().unwrap()
+        };
+
+        if superblock.magic != EXT2_MAGIC {
+            Err(Error::BadMagic(superblock.magic))
+        } else {
+            Ok(superblock)
+        }
+    }
+
+    pub fn find_block_table<'a>(
+        &self,
+        haystack: &'a mut [u8],
+    ) -> Result<&'a mut [BlockGroupDescriptor], Error> {
+        let count = self.block_group_count()
+            .map_err(|(a, b)| Error::BadBlockGroupCount(a, b))?
+            as usize;
+
+        let offset = 2048;
+        let end = offset + count * mem::size_of::<BlockGroupDescriptor>();
+        if haystack.len() < end {
+            return Err(Error::OutOfBounds(end));
+        }
+
+        let ptr = unsafe {
+            haystack.as_ptr().offset(offset as isize)
+                as *mut BlockGroupDescriptor
+        };
+        let slice = unsafe { slice::from_raw_parts_mut(ptr, count) };
+        Ok(slice)
+    }
+
     #[inline]
     pub fn block_size(&self) -> usize {
         1024 << self.log_block_size