use crate::block::Block;
use crate::blockhandle::BlockHandle;
use crate::error::{err, Result, StatusCode};
use crate::filter;
use crate::filter_block::FilterBlockReader;
use crate::options::{self, CompressionType, Options};
use crate::table_builder;
use crate::types::{unmask_crc, RandomAccess};
use crc::{Crc, CRC_32_ISCSI};
use integer_encoding::FixedInt;
use snap::raw::Decoder;
fn read_bytes(f: &dyn RandomAccess, location: &BlockHandle) -> Result<Vec<u8>> {
let mut buf = vec![0; location.size()];
f.read_at(location.offset(), &mut buf).map(|_| buf)
}
pub fn read_filter_block(
src: &dyn RandomAccess,
location: &BlockHandle,
policy: filter::BoxedFilterPolicy,
) -> Result<FilterBlockReader> {
if location.size() == 0 {
return err(
StatusCode::InvalidArgument,
"no filter block in empty location",
);
}
let buf = read_bytes(src, location)?;
Ok(FilterBlockReader::new_owned(policy, buf))
}
pub fn read_table_block(
opt: Options,
f: &dyn RandomAccess,
location: &BlockHandle,
) -> Result<Block> {
let block_data_size = location.size();
let location = BlockHandle::new(
location.offset(),
block_data_size
+ table_builder::TABLE_BLOCK_CKSUM_LEN
+ table_builder::TABLE_BLOCK_COMPRESS_LEN,
);
let mut buf = read_bytes(f, &location)?;
let mut compress = buf.split_off(block_data_size);
let cksum = compress.split_off(table_builder::TABLE_BLOCK_COMPRESS_LEN);
if !verify_table_block(&buf, compress[0], unmask_crc(u32::decode_fixed(&cksum))) {
return err(
StatusCode::Corruption,
&format!(
"checksum verification failed for block at {}",
location.offset()
),
);
}
if let Some(ctype) = options::int_to_compressiontype(compress[0] as u32) {
match ctype {
CompressionType::CompressionNone => Ok(Block::new(opt, buf)),
CompressionType::CompressionSnappy => {
let decoded = Decoder::new().decompress_vec(&buf)?;
Ok(Block::new(opt, decoded))
}
}
} else {
err(StatusCode::InvalidData, "invalid compression type")
}
}
fn verify_table_block(data: &[u8], compression: u8, want: u32) -> bool {
let crc = Crc::<u32>::new(&CRC_32_ISCSI);
let mut digest = crc.digest();
digest.update(data);
digest.update(&[compression; 1]);
digest.finalize() == want
}