able
/
ext2-rs
Archived
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

rewrite how block addresses are constructed

pull/3/head
Szymon Walter 2018-03-20 10:30:40 +01:00
parent 677c072b45
commit 2966b628fa
1 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/block.rs
View File

@ -42,9 +42,7 @@ pub struct Address<S: Size> {
}
impl<S: Size> Address<S> {
pub fn new(block: usize, offset: usize) -> Address<S> {
let block = block + (offset >> (S::LOG_SIZE + 10));
let offset = offset & S::OFFSET_MASK;
pub unsafe fn new_unchecked(block: usize, offset: usize) -> Address<S> {
let _phantom = PhantomData;
Address {
block,
@ -53,6 +51,12 @@ impl<S: Size> Address<S> {
}
}
pub fn new(block: usize, offset: isize) -> Address<S> {
let block = (block as isize + (offset >> (S::LOG_SIZE + 10))) as usize;
let offset = offset.abs() as usize & S::OFFSET_MASK;
unsafe { Address::new_unchecked(block, offset) }
}
pub fn into_index(&self) -> Option<usize> {
self.block
.checked_shl(S::LOG_SIZE + 10)
@ -102,29 +106,27 @@ impl<S: Size> From<usize> for Address<S> {
fn from(idx: usize) -> Address<S> {
let block = idx >> (S::LOG_SIZE + 10);
let offset = idx & S::OFFSET_MASK;
Address::new(block, offset)
Address::new(block, offset as isize)
}
}
impl<S: Size> Add for Address<S> {
type Output = Address<S>;
fn add(self, rhs: Address<S>) -> Address<S> {
let offset = self.offset + rhs.offset;
let block = offset >> (S::LOG_SIZE + 10);
let offset = offset & S::OFFSET_MASK;
Address::new(self.block + rhs.block + block, offset)
Address::new(
self.block + rhs.block,
(self.offset + rhs.offset) as isize,
)
}
}
impl<S: Size> Sub for Address<S> {
type Output = Address<S>;
fn sub(mut self, rhs: Address<S>) -> Address<S> {
if rhs.offset > self.offset {
self.offset += S::SIZE;
self.block -= 1;
}
let offset = self.offset - rhs.offset;
Address::new(self.block - rhs.block, offset)
fn sub(self, rhs: Address<S>) -> Address<S> {
Address::new(
self.block + rhs.block,
self.offset as isize - rhs.offset as isize,
)
}
}
@ -139,6 +141,11 @@ mod tests {
Address::<Size1024>::new(1, 0),
);
assert_eq!(
Address::<Size1024>::new(2, -512),
Address::<Size1024>::new(1, 512),
);
let a = Address::<Size2048>::new(0, 1024);
let b = Address::<Size2048>::new(0, 1024);
assert_eq!(a + b, Address::<Size2048>::new(1, 0));