diff --git a/src/api2/types/mod.rs b/src/api2/types/mod.rs index cfb4ec74..66b4d258 100644 --- a/src/api2/types/mod.rs +++ b/src/api2/types/mod.rs @@ -1354,6 +1354,18 @@ pub struct ArchiveEntry { impl ArchiveEntry { pub fn new(filepath: &[u8], entry_type: Option<&DirEntryAttribute>) -> Self { + let size = match entry_type { + Some(DirEntryAttribute::File { size, .. }) => Some(*size), + _ => None, + }; + Self::new_with_size(filepath, entry_type, size) + } + + pub fn new_with_size( + filepath: &[u8], + entry_type: Option<&DirEntryAttribute>, + size: Option, + ) -> Self { Self { filepath: base64::encode(filepath), text: String::from_utf8_lossy(filepath.split(|x| *x == b'/').last().unwrap()) @@ -1363,13 +1375,10 @@ impl ArchiveEntry { None => "v".to_owned(), }, leaf: !matches!(entry_type, None | Some(DirEntryAttribute::Directory { .. })), - size: match entry_type { - Some(DirEntryAttribute::File { size, .. }) => Some(*size), - _ => None - }, + size, mtime: match entry_type { Some(DirEntryAttribute::File { mtime, .. }) => Some(*mtime), - _ => None + _ => None, }, } } diff --git a/src/bin/proxmox-file-restore.rs b/src/bin/proxmox-file-restore.rs index ffc8b0a5..5766b279 100644 --- a/src/bin/proxmox-file-restore.rs +++ b/src/bin/proxmox-file-restore.rs @@ -194,7 +194,7 @@ async fn list( } else { None }; - entries.push(ArchiveEntry::new(path.as_bytes(), attr)); + entries.push(ArchiveEntry::new_with_size(path.as_bytes(), attr, Some(file.size))); } Ok(entries) diff --git a/src/bin/proxmox_restore_daemon/api.rs b/src/bin/proxmox_restore_daemon/api.rs index 82ead647..f1d601ce 100644 --- a/src/bin/proxmox_restore_daemon/api.rs +++ b/src/bin/proxmox_restore_daemon/api.rs @@ -200,11 +200,12 @@ fn list( for c in comps { let mut c_path = path.clone(); c_path.push(b'/'); - c_path.extend(c.as_bytes()); - res.push(ArchiveEntry::new( + c_path.extend(c.0.as_bytes()); + res.push(ArchiveEntry::new_with_size( &c_path[..], // this marks the beginning of a filesystem, i.e. '/', so this is a Directory Some(&DirEntryAttribute::Directory { start: 0 }), + Some(c.1), )); } } diff --git a/src/bin/proxmox_restore_daemon/disk.rs b/src/bin/proxmox_restore_daemon/disk.rs index f9d7c8aa..08fe7490 100644 --- a/src/bin/proxmox_restore_daemon/disk.rs +++ b/src/bin/proxmox_restore_daemon/disk.rs @@ -36,13 +36,14 @@ lazy_static! { pub enum ResolveResult { Path(PathBuf), BucketTypes(Vec<&'static str>), - BucketComponents(Vec), + BucketComponents(Vec<(String, u64)>), } struct PartitionBucketData { dev_node: String, number: i32, mountpoint: Option, + size: u64, } /// A "Bucket" represents a mapping found on a disk, e.g. a partition, a zfs dataset or an LV. A @@ -83,6 +84,12 @@ impl Bucket { Bucket::Partition(data) => data.number.to_string(), } } + + fn size(&self) -> u64 { + match self { + Bucket::Partition(data) => data.size, + } + } } /// Functions related to the local filesystem. This mostly exists so we can use 'supported_fs' in @@ -209,15 +216,23 @@ impl DiskState { .trim() .parse::()?; + // this *always* contains the number of 512-byte sectors, regardless of the true + // blocksize of this disk - which should always be 512 here anyway + let size = fs::file_read_firstline(&format!("{}/size", part_path))? + .trim() + .parse::()? + * 512; + info!( - "drive '{}' ('{}'): found partition '{}' ({})", - name, fidx, devnode, number + "drive '{}' ('{}'): found partition '{}' ({}, {}B)", + name, fidx, devnode, number, size ); let bucket = Bucket::Partition(PartitionBucketData { dev_node: devnode, mountpoint: None, number, + size, }); parts.push(bucket); @@ -281,7 +296,7 @@ impl DiskState { let comps = buckets .iter() .filter(|b| b.type_string() == bucket_type) - .map(Bucket::component_string) + .map(|b| (b.component_string(), b.size())) .collect(); return Ok(ResolveResult::BucketComponents(comps)); }