[pve-devel] [PATCH qemu-server v2 13/15] api: qmrestore: use volume content type assertion helpers

[Daniel Kral]

Make use of the newly introduced content type assertion helpers in the
preconditions check paths and the call to `vdisk_alloc` in the API and
CLI handler for restoring VMs to a specified target storage.

Since the preconditions existed before, adding a content type safeguard
at `vdisk_alloc` does not break the API.

Signed-off-by: Daniel Kral <d.kral at proxmox.com>
---
changes since v1:
- removed changes to fleecing images and vm state files

 PVE/QemuServer.pm | 12 +++++-------
 qmextract         |  4 +++-
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/PVE/QemuServer.pm b/PVE/QemuServer.pm
index f3178655..7025501f 100644
--- a/PVE/QemuServer.pm
+++ b/PVE/QemuServer.pm
@@ -6601,11 +6601,10 @@ my $parse_backup_hints = sub {
     my ($rpcenv, $user, $storecfg, $fh, $devinfo, $options) = @_;
 
     my $check_storage = sub { # assert if an image can be allocate
-	my ($storeid, $scfg) = @_;
-	die "Content type 'images' is not available on storage '$storeid'\n"
-	    if !$scfg->{content}->{images};
+	my ($storeid) = @_;
 	$rpcenv->check($user, "/storage/$storeid", ['Datastore.AllocateSpace'])
 	    if $user ne 'root at pam';
+	PVE::Storage::assert_content_type_supported($storecfg, $storeid, 'images');
     };
 
     my $virtdev_hash = {};
@@ -6626,8 +6625,7 @@ my $parse_backup_hints = sub {
 	    $devinfo->{$devname}->{format} = $format;
 	    $devinfo->{$devname}->{storeid} = $storeid;
 
-	    my $scfg = PVE::Storage::storage_config($storecfg, $storeid);
-	    $check_storage->($storeid, $scfg); # permission and content type check
+	    $check_storage->($storeid); # permission and content type check
 
 	    $virtdev_hash->{$virtdev} = $devinfo->{$devname};
 	} elsif ($line =~ m/^((?:ide|sata|scsi)\d+):\s*(.*)\s*$/) {
@@ -6637,10 +6635,9 @@ my $parse_backup_hints = sub {
 	    if (drive_is_cloudinit($drive)) {
 		my ($storeid, $volname) = PVE::Storage::parse_volume_id($drive->{file});
 		$storeid = $options->{storage} if defined ($options->{storage});
-		my $scfg = PVE::Storage::storage_config($storecfg, $storeid);
 		my $format = eval { checked_volume_format($storecfg, $drive->{file}) } // 'raw';
 
-		$check_storage->($storeid, $scfg); # permission and content type check
+		$check_storage->($storeid); # permission and content type check
 
 		$virtdev_hash->{$virtdev} = {
 		    format => $format,
@@ -6688,6 +6685,7 @@ my $restore_allocate_devices = sub {
 
 	my $volid = PVE::Storage::vdisk_alloc($storecfg, $storeid, $vmid, $d->{format}, $alloc_size, {
 	    name => $name,
+	    vtype => 'images',
 	});
 
 	print STDERR "new volume ID is '$volid'\n";
diff --git a/qmextract b/qmextract
index 0500c0b8..999fad9e 100755
--- a/qmextract
+++ b/qmextract
@@ -174,7 +174,9 @@ sub extract_archive {
 		if $format ne 'raw';
 	}
 
-	my $volid = PVE::Storage::vdisk_alloc($cfg, $storeid, $vmid, $format, $alloc_size);
+	my $volid = PVE::Storage::vdisk_alloc($cfg, $storeid, $vmid, $format, $alloc_size, {
+	    vtype => 'images',
+	});
 
 	print STDERR "new volume ID is '$volid'\n";
 
-- 
2.39.5