46 lines
1.0 KiB
Rust
46 lines
1.0 KiB
Rust
use byteorder::{LittleEndian, ReadBytesExt};
|
|
use crc32c::crc32c;
|
|
|
|
use std::io::Cursor;
|
|
|
|
const BLOCK_SIZE: u64 = 4096;
|
|
const MAGIC: u64 = 0xa537a0aa6309ef77;
|
|
const SUPERBLOCK_CSUM_XOR: u32 = 160774;
|
|
const BITMAP_CSUM_XOR: u32 = 240779;
|
|
const INDEX_CSUM_XOR: u32 = 160478;
|
|
const BTREE_CSUM_XOR: u32 = 121107;
|
|
|
|
fn checksum(buf: &[u8]) -> u32 {
|
|
crc32c(&buf[4..]) ^ 0xffffffff
|
|
}
|
|
|
|
#[derive(Debug, PartialEq)]
|
|
pub enum BT {
|
|
SUPERBLOCK,
|
|
NODE,
|
|
INDEX,
|
|
BITMAP,
|
|
UNKNOWN,
|
|
}
|
|
|
|
pub fn metadata_block_type(buf: &[u8]) -> BT {
|
|
if buf.len() != BLOCK_SIZE as usize {
|
|
return BT::UNKNOWN;
|
|
}
|
|
|
|
// The checksum is always stored in the first u32 of the buffer.
|
|
let mut rdr = Cursor::new(buf);
|
|
let sum_on_disk = rdr.read_u32::<LittleEndian>().unwrap();
|
|
let csum = checksum(buf);
|
|
let btype = csum ^ sum_on_disk;
|
|
|
|
match btype {
|
|
SUPERBLOCK_CSUM_XOR => BT::SUPERBLOCK,
|
|
BTREE_CSUM_XOR => BT::NODE,
|
|
BITMAP_CSUM_XOR => BT::BITMAP,
|
|
INDEX_CSUM_XOR => BT::INDEX,
|
|
_ => BT::UNKNOWN,
|
|
}
|
|
}
|
|
|