backup/chunk_store.rs: implement read_chunk()

This commit is contained in:
Dietmar Maurer 2019-01-04 12:50:54 +01:00
parent 691c89a0fb
commit 96df2fb44b
2 changed files with 34 additions and 0 deletions

View File

@ -129,6 +129,21 @@ impl <'a> ArchiveIndexReader<'a> {
}
Ok(())
}
pub fn dump_catar(&self, mut writer: Box<Write>) -> Result<(), Error> {
for pos in 0..self.index_entries {
let offset = unsafe { *(self.index.add(pos*40) as *const u64) };
let digest = unsafe { std::slice::from_raw_parts(self.index.add(pos*40+8), 32) };
let chunk = self.store.read_chunk(digest)?;
println!("Dump {:08x} {}", offset, chunk.len());
writer.write_all(&chunk)?;
}
Ok(())
}
}
@ -240,6 +255,7 @@ impl <'a> ArchiveIndexWriter<'a> {
match self.store.insert_chunk(&self.chunk_buffer) {
Ok((is_duplicate, digest)) => {
println!("ADD CHUNK {:016x} {} {} {}", self.chunk_offset, chunk_size, is_duplicate, digest_to_hex(&digest));
let chunk_end =
self.writer.write(unsafe { &std::mem::transmute::<u64, [u8;8]>(self.chunk_offset as u64) })?;
self.writer.write(&digest)?;
self.chunk_buffer.truncate(0);

View File

@ -165,6 +165,24 @@ impl ChunkStore {
Ok(())
}
pub fn read_chunk(&self, digest:&[u8]) -> Result<Vec<u8>, Error> {
let mut chunk_path = self.chunk_dir.clone();
let prefix = digest_to_prefix(&digest);
chunk_path.push(&prefix);
let digest_str = digest_to_hex(&digest);
chunk_path.push(&digest_str);
let mut f = std::fs::File::open(&chunk_path)?;
let mut chunk = Vec::with_capacity(64*1024);
use std::io::Read;
f.read_to_end(&mut chunk)?;
Ok(chunk)
}
fn sweep_old_files(&self, handle: &mut nix::dir::Dir, status: &mut GarbageCollectionStatus) -> Result<(), Error> {
let rawfd = handle.as_raw_fd();