From 6af905c18fbce35e9acc61049a613b34154a9007 Mon Sep 17 00:00:00 2001
From: Dietmar Maurer <dietmar@proxmox.com>
Date: Thu, 30 May 2019 12:46:01 +0200
Subject: [PATCH] src/bin/proxmox-backup-client.rs: implement
 image/block-device upload

---
 src/bin/proxmox-backup-client.rs | 54 +++++++++++++++++++++++++-------
 1 file changed, 43 insertions(+), 11 deletions(-)

diff --git a/src/bin/proxmox-backup-client.rs b/src/bin/proxmox-backup-client.rs
index 7e12865d..545c208d 100644
--- a/src/bin/proxmox-backup-client.rs
+++ b/src/bin/proxmox-backup-client.rs
@@ -28,7 +28,7 @@ use futures::*;
 use tokio::sync::mpsc;
 
 lazy_static! {
-    static ref BACKUPSPEC_REGEX: Regex = Regex::new(r"^([a-zA-Z0-9_-]+\.(?:pxar|raw)):(.+)$").unwrap();
+    static ref BACKUPSPEC_REGEX: Regex = Regex::new(r"^([a-zA-Z0-9_-]+\.(?:pxar|img)):(.+)$").unwrap();
 }
 
 
@@ -139,6 +139,33 @@ fn backup_directory<P: AsRef<Path>>(
     Ok(())
 }
 
+fn backup_image<P: AsRef<Path>>(
+    client: &BackupClient,
+    image_path: P,
+    archive_name: &str,
+    image_size: u64,
+    chunk_size: Option<u64>,
+    verbose: bool,
+) -> Result<(), Error> {
+
+    if let Some(_size) = chunk_size {
+        unimplemented!();
+    }
+
+    let path = image_path.as_ref().to_owned();
+
+    let file = tokio::fs::File::open(path).wait()?;
+
+    let stream = tokio::codec::FramedRead::new(file, tokio::codec::BytesCodec::new())
+        .map_err(Error::from);
+
+    let stream = FixedChunkStream::new(stream, 4*1024*1024);
+
+    client.upload_stream(archive_name, stream, "fixed", Some(image_size)).wait()?;
+
+    Ok(())
+}
+
 /****
 fn backup_image(datastore: &DataStore, file: &std::fs::File, size: usize, target: &str, chunk_size: usize) -> Result<(), Error> {
 
@@ -408,6 +435,8 @@ fn create_backup(
 
     let mut upload_list = vec![];
 
+    enum BackupType { PXAR, IMAGE };
+
     for backupspec in backupspec_list {
         let (target, filename) = parse_backupspec(backupspec.as_str().unwrap())?;
 
@@ -421,7 +450,7 @@ fn create_backup(
 
         if file_type.is_dir() {
 
-            upload_list.push((filename.to_owned(), target.to_owned()));
+            upload_list.push((BackupType::PXAR, filename.to_owned(), target.to_owned(), 0));
 
         } else if file_type.is_file() || file_type.is_block_device() {
 
@@ -429,12 +458,7 @@ fn create_backup(
 
             if size == 0 { bail!("got zero-sized file '{}'", filename); }
 
-            panic!("implement me");
-
-            //backup_image(&datastore, &file, size, &target, chunk_size)?;
-
-            // let idx = datastore.open_image_reader(target)?;
-            // idx.print_info();
+            upload_list.push((BackupType::IMAGE, filename.to_owned(), target.to_owned(), size));
 
         } else {
             bail!("unsupported file type (expected a directory, file or block device)");
@@ -452,9 +476,17 @@ fn create_backup(
 
     let client = client.start_backup(repo.store(), "host", &backup_id, verbose).wait()?;
 
-    for (filename, target) in upload_list {
-        println!("Upload '{}' to '{:?}' as {}", filename, repo, target);
-        backup_directory(&client, &filename, &target, chunk_size_opt, all_file_systems, verbose)?;
+    for (backup_type, filename, target, size) in upload_list {
+        match backup_type {
+            BackupType::PXAR => {
+                println!("Upload directory '{}' to '{:?}' as {}", filename, repo, target);
+                backup_directory(&client, &filename, &target, chunk_size_opt, all_file_systems, verbose)?;
+            }
+            BackupType::IMAGE => {
+                println!("Upload image '{}' to '{:?}' as {}", filename, repo, target);
+                backup_image(&client, &filename, &target, size, chunk_size_opt, verbose)?;
+            }
+        }
     }
 
     client.finish().wait()?;