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.k...@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@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



_______________________________________________
pve-devel mailing list
pve-devel@lists.proxmox.com
https://lists.proxmox.com/cgi-bin/mailman/listinfo/pve-devel

Reply via email to