2019-06-21 09:32:07 +00:00
|
|
|
use failure::*;
|
|
|
|
use std::convert::TryInto;
|
|
|
|
|
2019-07-01 08:36:59 +00:00
|
|
|
use proxmox::tools::io::{ReadExt, WriteExt};
|
2019-06-22 11:24:29 +00:00
|
|
|
|
2019-08-09 09:49:06 +00:00
|
|
|
const MAX_BLOB_SIZE: usize = 128*1024*1024;
|
|
|
|
|
2019-06-21 09:32:07 +00:00
|
|
|
use super::*;
|
|
|
|
|
|
|
|
/// Data blob binary storage format
|
|
|
|
///
|
2019-08-09 08:22:56 +00:00
|
|
|
/// Data blobs store arbitrary binary data (< 128MB), and can be
|
2019-06-21 09:32:07 +00:00
|
|
|
/// compressed and encrypted. A simply binary format is used to store
|
|
|
|
/// them on disk or transfer them over the network. Please use index
|
|
|
|
/// files to store large data files (".fidx" of ".didx").
|
|
|
|
///
|
|
|
|
pub struct DataBlob {
|
|
|
|
raw_data: Vec<u8>, // tagged, compressed, encryped data
|
|
|
|
}
|
|
|
|
|
|
|
|
impl DataBlob {
|
|
|
|
|
|
|
|
/// accessor to raw_data field
|
|
|
|
pub fn raw_data(&self) -> &[u8] {
|
|
|
|
&self.raw_data
|
|
|
|
}
|
|
|
|
|
2019-06-23 07:48:23 +00:00
|
|
|
/// Consume self and returns raw_data
|
|
|
|
pub fn into_inner(self) -> Vec<u8> {
|
|
|
|
self.raw_data
|
|
|
|
}
|
|
|
|
|
2019-06-21 09:32:07 +00:00
|
|
|
/// accessor to chunk type (magic number)
|
|
|
|
pub fn magic(&self) -> &[u8; 8] {
|
|
|
|
self.raw_data[0..8].try_into().unwrap()
|
|
|
|
}
|
|
|
|
|
2019-06-21 15:24:21 +00:00
|
|
|
/// accessor to crc32 checksum
|
|
|
|
pub fn crc(&self) -> u32 {
|
2019-06-22 07:12:25 +00:00
|
|
|
let crc_o = proxmox::tools::offsetof!(DataBlobHeader, crc);
|
|
|
|
u32::from_le_bytes(self.raw_data[crc_o..crc_o+4].try_into().unwrap())
|
2019-06-21 15:24:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// set the CRC checksum field
|
|
|
|
pub fn set_crc(&mut self, crc: u32) {
|
2019-06-22 07:12:25 +00:00
|
|
|
let crc_o = proxmox::tools::offsetof!(DataBlobHeader, crc);
|
|
|
|
self.raw_data[crc_o..crc_o+4].copy_from_slice(&crc.to_le_bytes());
|
2019-06-21 15:24:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// compute the CRC32 checksum
|
2019-06-23 07:48:23 +00:00
|
|
|
pub fn compute_crc(&self) -> u32 {
|
2019-06-21 15:24:21 +00:00
|
|
|
let mut hasher = crc32fast::Hasher::new();
|
2019-06-22 07:12:25 +00:00
|
|
|
let start = std::mem::size_of::<DataBlobHeader>(); // start after HEAD
|
|
|
|
hasher.update(&self.raw_data[start..]);
|
2019-06-21 15:24:21 +00:00
|
|
|
hasher.finalize()
|
|
|
|
}
|
|
|
|
|
2019-06-26 07:54:25 +00:00
|
|
|
/// verify the CRC32 checksum
|
|
|
|
pub fn verify_crc(&self) -> Result<(), Error> {
|
|
|
|
let expected_crc = self.compute_crc();
|
|
|
|
if expected_crc != self.crc() {
|
|
|
|
bail!("Data blob has wrong CRC checksum.");
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
2019-08-02 07:56:01 +00:00
|
|
|
/// Create a DataBlob, optionally compressed and/or encrypted
|
2019-06-21 09:32:07 +00:00
|
|
|
pub fn encode(
|
|
|
|
data: &[u8],
|
|
|
|
config: Option<&CryptConfig>,
|
|
|
|
compress: bool,
|
|
|
|
) -> Result<Self, Error> {
|
|
|
|
|
2019-08-09 09:49:06 +00:00
|
|
|
if data.len() > MAX_BLOB_SIZE {
|
2019-06-21 09:32:07 +00:00
|
|
|
bail!("data blob too large ({} bytes).", data.len());
|
|
|
|
}
|
|
|
|
|
2019-08-06 09:42:14 +00:00
|
|
|
let mut blob = if let Some(config) = config {
|
2019-06-21 09:32:07 +00:00
|
|
|
|
2019-06-22 11:02:53 +00:00
|
|
|
let compr_data;
|
|
|
|
let (_compress, data, magic) = if compress {
|
|
|
|
compr_data = zstd::block::compress(data, 1)?;
|
|
|
|
// Note: We only use compression if result is shorter
|
|
|
|
if compr_data.len() < data.len() {
|
|
|
|
(true, &compr_data[..], ENCR_COMPR_BLOB_MAGIC_1_0)
|
|
|
|
} else {
|
|
|
|
(false, data, ENCRYPTED_BLOB_MAGIC_1_0)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
(false, data, ENCRYPTED_BLOB_MAGIC_1_0)
|
|
|
|
};
|
|
|
|
|
|
|
|
let header_len = std::mem::size_of::<EncryptedDataBlobHeader>();
|
|
|
|
let mut raw_data = Vec::with_capacity(data.len() + header_len);
|
|
|
|
|
|
|
|
let dummy_head = EncryptedDataBlobHeader {
|
|
|
|
head: DataBlobHeader { magic: [0u8; 8], crc: [0; 4] },
|
|
|
|
iv: [0u8; 16],
|
|
|
|
tag: [0u8; 16],
|
|
|
|
};
|
2019-07-01 08:36:59 +00:00
|
|
|
unsafe {
|
|
|
|
raw_data.write_le_value(dummy_head)?;
|
|
|
|
}
|
2019-06-22 11:02:53 +00:00
|
|
|
|
|
|
|
let (iv, tag) = config.encrypt_to(data, &mut raw_data)?;
|
|
|
|
|
|
|
|
let head = EncryptedDataBlobHeader {
|
|
|
|
head: DataBlobHeader { magic, crc: [0; 4] }, iv, tag,
|
|
|
|
};
|
|
|
|
|
2019-07-01 08:36:59 +00:00
|
|
|
unsafe {
|
|
|
|
(&mut raw_data[0..header_len]).write_le_value(head)?;
|
|
|
|
}
|
2019-06-22 11:02:53 +00:00
|
|
|
|
2019-08-06 09:42:14 +00:00
|
|
|
DataBlob { raw_data }
|
2019-06-21 09:32:07 +00:00
|
|
|
} else {
|
|
|
|
|
2019-06-22 07:12:25 +00:00
|
|
|
let max_data_len = data.len() + std::mem::size_of::<DataBlobHeader>();
|
2019-06-21 09:32:07 +00:00
|
|
|
if compress {
|
2019-06-22 07:12:25 +00:00
|
|
|
let mut comp_data = Vec::with_capacity(max_data_len);
|
2019-06-21 09:32:07 +00:00
|
|
|
|
2019-06-22 07:12:25 +00:00
|
|
|
let head = DataBlobHeader {
|
|
|
|
magic: COMPRESSED_BLOB_MAGIC_1_0,
|
|
|
|
crc: [0; 4],
|
|
|
|
};
|
2019-07-01 08:36:59 +00:00
|
|
|
unsafe {
|
|
|
|
comp_data.write_le_value(head)?;
|
|
|
|
}
|
2019-06-21 15:24:21 +00:00
|
|
|
|
2019-06-21 09:32:07 +00:00
|
|
|
zstd::stream::copy_encode(data, &mut comp_data, 1)?;
|
|
|
|
|
2019-06-22 07:12:25 +00:00
|
|
|
if comp_data.len() < max_data_len {
|
2019-08-07 06:29:38 +00:00
|
|
|
let mut blob = DataBlob { raw_data: comp_data };
|
|
|
|
blob.set_crc(blob.compute_crc());
|
|
|
|
return Ok(blob);
|
2019-06-21 09:32:07 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-22 07:12:25 +00:00
|
|
|
let mut raw_data = Vec::with_capacity(max_data_len);
|
2019-06-21 09:32:07 +00:00
|
|
|
|
2019-06-22 07:12:25 +00:00
|
|
|
let head = DataBlobHeader {
|
|
|
|
magic: UNCOMPRESSED_BLOB_MAGIC_1_0,
|
|
|
|
crc: [0; 4],
|
|
|
|
};
|
2019-07-01 08:36:59 +00:00
|
|
|
unsafe {
|
|
|
|
raw_data.write_le_value(head)?;
|
|
|
|
}
|
2019-06-21 09:32:07 +00:00
|
|
|
raw_data.extend_from_slice(data);
|
|
|
|
|
2019-08-06 09:42:14 +00:00
|
|
|
DataBlob { raw_data }
|
|
|
|
};
|
|
|
|
|
|
|
|
blob.set_crc(blob.compute_crc());
|
|
|
|
|
|
|
|
Ok(blob)
|
2019-06-21 09:32:07 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Decode blob data
|
|
|
|
pub fn decode(self, config: Option<&CryptConfig>) -> Result<Vec<u8>, Error> {
|
|
|
|
|
|
|
|
let magic = self.magic();
|
|
|
|
|
|
|
|
if magic == &UNCOMPRESSED_BLOB_MAGIC_1_0 {
|
2019-06-22 07:12:25 +00:00
|
|
|
let data_start = std::mem::size_of::<DataBlobHeader>();
|
|
|
|
return Ok(self.raw_data[data_start..].to_vec());
|
2019-06-21 09:32:07 +00:00
|
|
|
} else if magic == &COMPRESSED_BLOB_MAGIC_1_0 {
|
2019-06-22 07:12:25 +00:00
|
|
|
let data_start = std::mem::size_of::<DataBlobHeader>();
|
2019-08-09 09:49:06 +00:00
|
|
|
let data = zstd::block::decompress(&self.raw_data[data_start..], MAX_BLOB_SIZE)?;
|
2019-06-21 09:32:07 +00:00
|
|
|
return Ok(data);
|
|
|
|
} else if magic == &ENCR_COMPR_BLOB_MAGIC_1_0 || magic == &ENCRYPTED_BLOB_MAGIC_1_0 {
|
2019-06-22 11:24:29 +00:00
|
|
|
let header_len = std::mem::size_of::<EncryptedDataBlobHeader>();
|
2019-06-22 14:29:10 +00:00
|
|
|
let head = unsafe {
|
|
|
|
(&self.raw_data[..header_len]).read_le_value::<EncryptedDataBlobHeader>()?
|
|
|
|
};
|
2019-06-22 11:24:29 +00:00
|
|
|
|
2019-06-21 09:32:07 +00:00
|
|
|
if let Some(config) = config {
|
|
|
|
let data = if magic == &ENCR_COMPR_BLOB_MAGIC_1_0 {
|
2019-06-22 11:24:29 +00:00
|
|
|
config.decode_compressed_chunk(&self.raw_data[header_len..], &head.iv, &head.tag)?
|
2019-06-21 09:32:07 +00:00
|
|
|
} else {
|
2019-06-22 11:24:29 +00:00
|
|
|
config.decode_uncompressed_chunk(&self.raw_data[header_len..], &head.iv, &head.tag)?
|
2019-06-21 09:32:07 +00:00
|
|
|
};
|
|
|
|
return Ok(data);
|
|
|
|
} else {
|
|
|
|
bail!("unable to decrypt blob - missing CryptConfig");
|
|
|
|
}
|
2019-08-02 07:56:01 +00:00
|
|
|
} else if magic == &AUTH_COMPR_BLOB_MAGIC_1_0 || magic == &AUTHENTICATED_BLOB_MAGIC_1_0 {
|
|
|
|
let header_len = std::mem::size_of::<AuthenticatedDataBlobHeader>();
|
|
|
|
let head = unsafe {
|
|
|
|
(&self.raw_data[..header_len]).read_le_value::<AuthenticatedDataBlobHeader>()?
|
|
|
|
};
|
|
|
|
|
|
|
|
let data_start = std::mem::size_of::<AuthenticatedDataBlobHeader>();
|
|
|
|
|
|
|
|
// Note: only verify if we have a crypt config
|
|
|
|
if let Some(config) = config {
|
|
|
|
let signature = config.compute_auth_tag(&self.raw_data[data_start..]);
|
|
|
|
if signature != head.tag {
|
|
|
|
bail!("verifying blob signature failed");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if magic == &AUTH_COMPR_BLOB_MAGIC_1_0 {
|
|
|
|
let data = zstd::block::decompress(&self.raw_data[data_start..], 16*1024*1024)?;
|
|
|
|
return Ok(data);
|
|
|
|
} else {
|
|
|
|
return Ok(self.raw_data[data_start..].to_vec());
|
|
|
|
}
|
2019-06-21 09:32:07 +00:00
|
|
|
} else {
|
|
|
|
bail!("Invalid blob magic number.");
|
|
|
|
}
|
|
|
|
}
|
2019-06-23 07:35:44 +00:00
|
|
|
|
2019-08-02 07:56:01 +00:00
|
|
|
/// Create a signed DataBlob, optionally compressed
|
|
|
|
pub fn create_signed(
|
|
|
|
data: &[u8],
|
|
|
|
config: &CryptConfig,
|
|
|
|
compress: bool,
|
|
|
|
) -> Result<Self, Error> {
|
|
|
|
|
2019-08-09 09:49:06 +00:00
|
|
|
if data.len() > MAX_BLOB_SIZE {
|
2019-08-02 07:56:01 +00:00
|
|
|
bail!("data blob too large ({} bytes).", data.len());
|
|
|
|
}
|
|
|
|
|
|
|
|
let compr_data;
|
|
|
|
let (_compress, data, magic) = if compress {
|
|
|
|
compr_data = zstd::block::compress(data, 1)?;
|
|
|
|
// Note: We only use compression if result is shorter
|
|
|
|
if compr_data.len() < data.len() {
|
|
|
|
(true, &compr_data[..], AUTH_COMPR_BLOB_MAGIC_1_0)
|
|
|
|
} else {
|
|
|
|
(false, data, AUTHENTICATED_BLOB_MAGIC_1_0)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
(false, data, AUTHENTICATED_BLOB_MAGIC_1_0)
|
|
|
|
};
|
|
|
|
|
|
|
|
let header_len = std::mem::size_of::<AuthenticatedDataBlobHeader>();
|
|
|
|
let mut raw_data = Vec::with_capacity(data.len() + header_len);
|
|
|
|
|
|
|
|
let head = AuthenticatedDataBlobHeader {
|
|
|
|
head: DataBlobHeader { magic, crc: [0; 4] },
|
|
|
|
tag: config.compute_auth_tag(data),
|
|
|
|
};
|
|
|
|
unsafe {
|
|
|
|
raw_data.write_le_value(head)?;
|
|
|
|
}
|
|
|
|
raw_data.extend_from_slice(data);
|
|
|
|
|
2019-08-06 09:42:14 +00:00
|
|
|
let mut blob = DataBlob { raw_data };
|
|
|
|
blob.set_crc(blob.compute_crc());
|
|
|
|
|
|
|
|
return Ok(blob);
|
2019-08-02 07:56:01 +00:00
|
|
|
}
|
|
|
|
|
2019-06-23 07:35:44 +00:00
|
|
|
/// Create Instance from raw data
|
|
|
|
pub fn from_raw(data: Vec<u8>) -> Result<Self, Error> {
|
|
|
|
|
|
|
|
if data.len() < std::mem::size_of::<DataBlobHeader>() {
|
|
|
|
bail!("blob too small ({} bytes).", data.len());
|
|
|
|
}
|
|
|
|
|
|
|
|
let magic = &data[0..8];
|
|
|
|
|
|
|
|
if magic == ENCR_COMPR_BLOB_MAGIC_1_0 || magic == ENCRYPTED_BLOB_MAGIC_1_0 {
|
|
|
|
|
|
|
|
if data.len() < std::mem::size_of::<EncryptedDataBlobHeader>() {
|
|
|
|
bail!("encrypted blob too small ({} bytes).", data.len());
|
|
|
|
}
|
|
|
|
|
|
|
|
let blob = DataBlob { raw_data: data };
|
|
|
|
|
|
|
|
Ok(blob)
|
|
|
|
} else if magic == COMPRESSED_BLOB_MAGIC_1_0 || magic == UNCOMPRESSED_BLOB_MAGIC_1_0 {
|
|
|
|
|
|
|
|
let blob = DataBlob { raw_data: data };
|
|
|
|
|
2019-08-02 07:56:01 +00:00
|
|
|
Ok(blob)
|
|
|
|
} else if magic == AUTH_COMPR_BLOB_MAGIC_1_0 || magic == AUTHENTICATED_BLOB_MAGIC_1_0 {
|
|
|
|
if data.len() < std::mem::size_of::<AuthenticatedDataBlobHeader>() {
|
|
|
|
bail!("authenticated blob too small ({} bytes).", data.len());
|
|
|
|
}
|
|
|
|
|
|
|
|
let blob = DataBlob { raw_data: data };
|
|
|
|
|
2019-06-23 07:35:44 +00:00
|
|
|
Ok(blob)
|
|
|
|
} else {
|
|
|
|
bail!("unable to parse raw blob - wrong magic");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-21 09:32:07 +00:00
|
|
|
}
|
2019-08-11 09:32:36 +00:00
|
|
|
|
|
|
|
// TODO: impl. other blob types
|
|
|
|
|
|
|
|
use std::io::{Read, BufRead, Write, Seek, SeekFrom};
|
|
|
|
|
2019-08-12 05:34:41 +00:00
|
|
|
enum BlobWriterState<'a, W: Write> {
|
2019-08-11 10:07:37 +00:00
|
|
|
Uncompressed { writer: W, hasher: crc32fast::Hasher },
|
|
|
|
Compressed { compr: zstd::stream::write::Encoder<W>, hasher: crc32fast::Hasher },
|
2019-08-12 05:34:41 +00:00
|
|
|
Signed {
|
|
|
|
writer: W,
|
|
|
|
hasher: crc32fast::Hasher,
|
|
|
|
signer: openssl::sign::Signer<'a>,
|
|
|
|
},
|
2019-08-11 10:07:37 +00:00
|
|
|
}
|
|
|
|
|
2019-08-11 09:32:36 +00:00
|
|
|
/// Write compressed data blobs
|
2019-08-12 05:34:41 +00:00
|
|
|
pub struct DataBlobWriter<'a, W: Write> {
|
|
|
|
state: BlobWriterState<'a, W>,
|
2019-08-11 09:32:36 +00:00
|
|
|
}
|
|
|
|
|
2019-08-12 05:34:41 +00:00
|
|
|
impl <'a, W: Write + Seek> DataBlobWriter<'a, W> {
|
2019-08-11 10:07:37 +00:00
|
|
|
|
|
|
|
pub fn new_uncompressed(mut writer: W) -> Result<Self, Error> {
|
|
|
|
let hasher = crc32fast::Hasher::new();
|
|
|
|
writer.seek(SeekFrom::Start(0))?;
|
|
|
|
let head = DataBlobHeader { magic: UNCOMPRESSED_BLOB_MAGIC_1_0, crc: [0; 4] };
|
|
|
|
unsafe {
|
|
|
|
writer.write_le_value(head)?;
|
|
|
|
}
|
|
|
|
let state = BlobWriterState::Uncompressed { writer, hasher };
|
|
|
|
Ok(Self { state })
|
|
|
|
}
|
2019-08-11 09:32:36 +00:00
|
|
|
|
2019-08-11 10:07:37 +00:00
|
|
|
pub fn new_compressed(mut writer: W) -> Result<Self, Error> {
|
|
|
|
let hasher = crc32fast::Hasher::new();
|
|
|
|
writer.seek(SeekFrom::Start(0))?;
|
2019-08-11 09:32:36 +00:00
|
|
|
let head = DataBlobHeader { magic: COMPRESSED_BLOB_MAGIC_1_0, crc: [0; 4] };
|
|
|
|
unsafe {
|
2019-08-11 10:07:37 +00:00
|
|
|
writer.write_le_value(head)?;
|
2019-08-11 09:32:36 +00:00
|
|
|
}
|
2019-08-11 10:07:37 +00:00
|
|
|
let compr = zstd::stream::write::Encoder::new(writer, 1)?;
|
|
|
|
let state = BlobWriterState::Compressed { compr, hasher };
|
|
|
|
Ok(Self { state })
|
2019-08-11 09:32:36 +00:00
|
|
|
}
|
|
|
|
|
2019-08-12 05:34:41 +00:00
|
|
|
pub fn new_signed(mut writer: W, config: &'a CryptConfig) -> Result<Self, Error> {
|
|
|
|
let hasher = crc32fast::Hasher::new();
|
|
|
|
writer.seek(SeekFrom::Start(0))?;
|
|
|
|
let head = AuthenticatedDataBlobHeader {
|
|
|
|
head: DataBlobHeader { magic: AUTHENTICATED_BLOB_MAGIC_1_0, crc: [0; 4] },
|
|
|
|
tag: [0u8; 32],
|
|
|
|
};
|
|
|
|
unsafe {
|
|
|
|
writer.write_le_value(head)?;
|
|
|
|
}
|
|
|
|
let signer = config.data_signer();
|
|
|
|
|
|
|
|
let state = BlobWriterState::Signed { writer, hasher, signer };
|
|
|
|
Ok(Self { state })
|
|
|
|
}
|
|
|
|
|
2019-08-11 10:07:37 +00:00
|
|
|
pub fn finish(self) -> Result<W, Error> {
|
|
|
|
match self.state {
|
|
|
|
BlobWriterState::Uncompressed { mut writer, hasher } => {
|
|
|
|
// write CRC
|
|
|
|
let crc = hasher.finalize();
|
|
|
|
let head = DataBlobHeader { magic: COMPRESSED_BLOB_MAGIC_1_0, crc: crc.to_le_bytes() };
|
2019-08-11 09:32:36 +00:00
|
|
|
|
2019-08-11 10:07:37 +00:00
|
|
|
writer.seek(SeekFrom::Start(0))?;
|
|
|
|
unsafe {
|
|
|
|
writer.write_le_value(head)?;
|
|
|
|
}
|
2019-08-11 09:32:36 +00:00
|
|
|
|
2019-08-11 10:07:37 +00:00
|
|
|
return Ok(writer)
|
|
|
|
}
|
|
|
|
BlobWriterState::Compressed { compr, hasher } => {
|
|
|
|
let mut writer = compr.finish()?;
|
|
|
|
|
|
|
|
// write CRC
|
|
|
|
let crc = hasher.finalize();
|
|
|
|
let head = DataBlobHeader { magic: COMPRESSED_BLOB_MAGIC_1_0, crc: crc.to_le_bytes() };
|
2019-08-11 09:32:36 +00:00
|
|
|
|
2019-08-11 10:07:37 +00:00
|
|
|
writer.seek(SeekFrom::Start(0))?;
|
|
|
|
unsafe {
|
|
|
|
writer.write_le_value(head)?;
|
|
|
|
}
|
|
|
|
|
2019-08-12 05:34:41 +00:00
|
|
|
return Ok(writer)
|
|
|
|
}
|
|
|
|
BlobWriterState::Signed { mut writer, hasher, signer, .. } => {
|
|
|
|
// write CRC and hmac
|
|
|
|
let crc = hasher.finalize();
|
|
|
|
|
|
|
|
let mut head = AuthenticatedDataBlobHeader {
|
|
|
|
head: DataBlobHeader { magic: AUTHENTICATED_BLOB_MAGIC_1_0, crc: crc.to_le_bytes() },
|
|
|
|
tag: [0u8; 32],
|
|
|
|
};
|
|
|
|
signer.sign(&mut head.tag)?;
|
|
|
|
|
|
|
|
writer.seek(SeekFrom::Start(0))?;
|
|
|
|
unsafe {
|
|
|
|
writer.write_le_value(head)?;
|
|
|
|
}
|
|
|
|
|
2019-08-11 10:07:37 +00:00
|
|
|
return Ok(writer)
|
|
|
|
}
|
|
|
|
}
|
2019-08-11 09:32:36 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-08-12 05:34:41 +00:00
|
|
|
impl <'a, W: Write + Seek> Write for DataBlobWriter<'a, W> {
|
2019-08-11 09:32:36 +00:00
|
|
|
|
|
|
|
fn write(&mut self, buf: &[u8]) -> Result<usize, std::io::Error> {
|
2019-08-11 10:07:37 +00:00
|
|
|
match self.state {
|
|
|
|
BlobWriterState::Uncompressed { ref mut writer, ref mut hasher } => {
|
|
|
|
hasher.update(buf);
|
|
|
|
writer.write(buf)
|
|
|
|
}
|
|
|
|
BlobWriterState::Compressed { ref mut compr, ref mut hasher } => {
|
|
|
|
hasher.update(buf);
|
|
|
|
compr.write(buf)
|
|
|
|
}
|
2019-08-12 05:34:41 +00:00
|
|
|
BlobWriterState::Signed { ref mut writer, ref mut hasher, ref mut signer, .. } => {
|
|
|
|
hasher.update(buf);
|
|
|
|
signer.update(buf).
|
|
|
|
map_err(|err| {
|
|
|
|
std::io::Error::new(
|
|
|
|
std::io::ErrorKind::Other,
|
|
|
|
format!("hmac update failed - {}", err))
|
|
|
|
})?;
|
|
|
|
writer.write(buf)
|
|
|
|
}
|
2019-08-11 10:07:37 +00:00
|
|
|
}
|
2019-08-11 09:32:36 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
fn flush(&mut self) -> Result<(), std::io::Error> {
|
2019-08-11 10:07:37 +00:00
|
|
|
match self.state {
|
|
|
|
BlobWriterState::Uncompressed { ref mut writer, .. } => {
|
|
|
|
writer.flush()
|
|
|
|
}
|
|
|
|
BlobWriterState::Compressed { ref mut compr, .. } => {
|
|
|
|
compr.flush()
|
|
|
|
}
|
2019-08-12 05:34:41 +00:00
|
|
|
BlobWriterState::Signed { ref mut writer, .. } => {
|
|
|
|
writer.flush()
|
|
|
|
}
|
2019-08-11 10:07:37 +00:00
|
|
|
}
|
2019-08-11 09:32:36 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Read compressed data blobs
|
|
|
|
pub struct CompressedDataBlobReader<R: BufRead> {
|
|
|
|
decompr: zstd::stream::read::Decoder<R>,
|
|
|
|
hasher: Option<crc32fast::Hasher>,
|
|
|
|
expected_crc: u32,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl <R: BufRead> CompressedDataBlobReader<R> {
|
|
|
|
|
|
|
|
pub fn new(mut reader: R) -> Result<Self, Error> {
|
|
|
|
|
|
|
|
let head: DataBlobHeader = unsafe { reader.read_le_value()? };
|
|
|
|
if head.magic != COMPRESSED_BLOB_MAGIC_1_0 {
|
|
|
|
bail!("got wrong magic number");
|
|
|
|
}
|
|
|
|
let expected_crc = u32::from_le_bytes(head.crc);
|
|
|
|
let decompr = zstd::stream::read::Decoder::with_buffer(reader)?;
|
|
|
|
Ok(Self { decompr: decompr, hasher: Some(crc32fast::Hasher::new()), expected_crc })
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl <R: BufRead> Read for CompressedDataBlobReader<R> {
|
|
|
|
|
|
|
|
fn read(&mut self, buf: &mut [u8]) -> Result<usize, std::io::Error> {
|
|
|
|
let count = self.decompr.read(buf)?;
|
|
|
|
if count == 0 { // EOF, verify crc
|
|
|
|
let hasher = self.hasher.take().expect("blob reader already finished");
|
|
|
|
let crc = hasher.finalize();
|
|
|
|
if crc != self.expected_crc {
|
|
|
|
return Err(std::io::Error::new(std::io::ErrorKind::Other, "blob reader crc error"));
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
let hasher = self.hasher.as_mut().expect("blob reader already finished");
|
|
|
|
hasher.update(buf);
|
|
|
|
}
|
|
|
|
Ok(count)
|
|
|
|
}
|
|
|
|
}
|