tyler
/
proxmox-backup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

src/bin/proxmox-backup-client.rs - reatore: avoid loading (large) blobs into memory

This commit is contained in:
Dietmar Maurer 2019-09-02 14:13:31 +02:00
parent 26c945d3a3
commit 0d9862803b
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/bin/proxmox-backup-client.rs
View File

@ -888,20 +888,22 @@ fn restore(
.open("/tmp")?; .open("/tmp")?;
if server_archive_name.ends_with(".blob") { if server_archive_name.ends_with(".blob") {
let mut tmpfile = client.download(&server_archive_name, tmpfile).wait()?;
let writer = Vec::with_capacity(1024*1024); tmpfile.seek(SeekFrom::Start(0))?;
let blob_data = client.download(&server_archive_name, writer).wait()?; let mut reader = DataBlobReader::new(tmpfile, crypt_config)?;
let blob = DataBlob::from_raw(blob_data)?;
blob.verify_crc()?;
let raw_data = blob.decode(crypt_config)?;
if let Some(target) = target { if let Some(target) = target {
file_set_contents(target, &raw_data, None)?; let mut writer = std::fs::OpenOptions::new()
.write(true)
.create(true)
.create_new(true)
.open(target)
.map_err(|err| format_err!("unable to create target file {:?} - {}", target, err))?;
std::io::copy(&mut reader, &mut writer)?;
} else { } else {
let stdout = std::io::stdout(); let stdout = std::io::stdout();
let mut writer = stdout.lock(); let mut writer = stdout.lock();
writer.write_all(&raw_data) std::io::copy(&mut reader, &mut writer)
.map_err(|err| format_err!("unable to pipe data - {}", err))?; .map_err(|err| format_err!("unable to pipe data - {}", err))?;
} }