when 'import-from' contains a disk image that needs extraction
(currently only from an 'ova' archive), do that in 'create_disks'
and overwrite the '$source' volid.

Collect the names into a 'delete_sources' list, that we use later
to clean it up again (either when we're finished with importing or in an
error case).

Signed-off-by: Dominik Csapak <d.csa...@proxmox.com>
---
changes from v5:
* changed the pve-managed check to be correct

 PVE/API2/Qemu.pm          | 51 +++++++++++++++++++++++++++++++++------
 PVE/QemuServer.pm         | 12 +++++++++
 PVE/QemuServer/Helpers.pm |  5 ++++
 3 files changed, 60 insertions(+), 8 deletions(-)

diff --git a/PVE/API2/Qemu.pm b/PVE/API2/Qemu.pm
index b9c63af8..1aa42585 100644
--- a/PVE/API2/Qemu.pm
+++ b/PVE/API2/Qemu.pm
@@ -24,6 +24,7 @@ use PVE::JSONSchema qw(get_standard_option);
 use PVE::RESTHandler;
 use PVE::ReplicationConfig;
 use PVE::GuestHelpers qw(assert_tag_permissions);
+use PVE::GuestImport;
 use PVE::QemuConfig;
 use PVE::QemuServer;
 use PVE::QemuServer::Cloudinit;
@@ -163,10 +164,20 @@ my $check_storage_access = sub {
 
        if (my $src_image = $drive->{'import-from'}) {
            my $src_vmid;
-           if (PVE::Storage::parse_volume_id($src_image, 1)) { # PVE-managed 
volume
-               (my $vtype, undef, $src_vmid) = 
PVE::Storage::parse_volname($storecfg, $src_image);
-               raise_param_exc({ $ds => "$src_image has wrong type '$vtype' - 
not an image" })
-                   if $vtype ne 'images';
+           my ($storeid, $volname) = PVE::Storage::parse_volume_id($src_image, 
1);
+           if ($storeid) { # PVE-managed volume
+               my $scfg = PVE::Storage::storage_config($storecfg, $storeid);
+               my $plugin = PVE::Storage::Plugin->lookup($scfg->{type});
+               (my $vtype, undef, $src_vmid, undef, undef, undef, my $fmt) = 
$plugin->parse_volname($volname);
+
+               raise_param_exc({ $ds => "$src_image has wrong type '$vtype' - 
needs to be 'images' or 'import'" })
+                   if $vtype ne 'images' && $vtype ne 'import';
+
+               if (PVE::QemuServer::Helpers::needs_extraction($vtype, $fmt)) {
+                   raise_param_exc({ $ds => "$src_image is not on an storage 
with 'images' content type."})
+                       if !$scfg->{content}->{images};
+                   $rpcenv->check($authuser, "/storage/$storeid", 
['Datastore.AllocateSpace']);
+               }
            }
 
            if ($src_vmid) { # might be actively used by VM and will be copied 
via clone_disk()
@@ -416,6 +427,23 @@ my sub create_disks : prototype($$$$$$$$$$) {
                my ($source_storage, $source_volid) = 
PVE::Storage::parse_volume_id($source, 1);
 
                if ($source_storage) { # PVE-managed volume
+                   my ($vtype, undef, undef, undef, undef, undef, $fmt)
+                       = PVE::Storage::parse_volname($storecfg, $source);
+                   my $needs_extraction = 
PVE::QemuServer::Helpers::needs_extraction($vtype, $fmt);
+                   if ($needs_extraction) {
+                       print "extracting $source\n";
+                       my $extracted_volid
+                            = 
PVE::GuestImport::extract_disk_from_import_file($source, $vmid);
+                       print "finished extracting to $extracted_volid\n";
+                       push @$vollist, $extracted_volid;
+                       $source = $extracted_volid;
+
+                       my (undef, undef, undef, $parent)
+                           = PVE::Storage::volume_size_info($storecfg, 
$source);
+                       die "importing from extracted images with backing file 
($parent) not allowed\n"
+                           if $parent;
+                   }
+
                    if ($live_import && $ds ne 'efidisk0') {
                        my $path = PVE::Storage::path($storecfg, $source)
                            or die "failed to get a path for '$source'\n";
@@ -424,9 +452,11 @@ my sub create_disks : prototype($$$$$$$$$$) {
 
                        die "could not get file size of $source\n" if !$size;
                        $live_import_mapping->{$ds} = {
-                           path => $source,
+                           path => $path,
                            format => $source_format,
                        };
+                       $live_import_mapping->{$ds}->{'delete-after-finish'} = 
$source
+                           if $needs_extraction;
                    } else {
                        my $dest_info = {
                            vmid => $vmid,
@@ -438,8 +468,14 @@ my sub create_disks : prototype($$$$$$$$$$) {
                        $dest_info->{efisize} = 
PVE::QemuServer::get_efivars_size($conf, $disk)
                            if $ds eq 'efidisk0';
 
-                       ($dst_volid, $size) = eval {
-                           $import_from_volid->($storecfg, $source, 
$dest_info, $vollist);
+                       eval {
+                           ($dst_volid, $size)
+                               = $import_from_volid->($storecfg, $source, 
$dest_info, $vollist);
+
+                           # remove extracted volumes after importing
+                           PVE::Storage::vdisk_free($storecfg, $source) if 
$needs_extraction;
+                           print "cleaned up extracted image $source\n";
+                           @$vollist = grep { $_ ne $source } @$vollist;
                        };
                        die "cannot import from '$source' - $@" if $@;
                    }
@@ -1964,7 +2000,6 @@ my $update_vm_api  = sub {
 
                    assert_scsi_feature_compatibility($opt, $conf, $storecfg, 
$param->{$opt})
                        if $opt =~ m/^scsi\d+$/;
-
                    my (undef, $created_opts) = create_disks(
                        $rpcenv,
                        $authuser,
diff --git a/PVE/QemuServer.pm b/PVE/QemuServer.pm
index cb1e0b82..706343a4 100644
--- a/PVE/QemuServer.pm
+++ b/PVE/QemuServer.pm
@@ -7400,6 +7400,7 @@ sub live_import_from_files {
     my ($mapping, $vmid, $conf, $restore_options) = @_;
 
     my $live_restore_backing = {};
+    my $sources_to_remove = [];
     for my $dev (keys %$mapping) {
        die "disk not support for live-restoring: '$dev'\n"
            if !is_valid_drivename($dev) || $dev =~ /^(?:efidisk|tpmstate)/;
@@ -7420,6 +7421,9 @@ sub live_import_from_files {
            . ",read-only=on"
            . ",file.driver=file,file.filename=$path"
        };
+
+       my $source_volid = $info->{'delete-after-finish'};
+       push $sources_to_remove->@*, $source_volid if defined($source_volid);
     };
 
     my $storecfg = PVE::Storage::config();
@@ -7464,6 +7468,14 @@ sub live_import_from_files {
 
     my $err = $@;
 
+    for my $volid ($sources_to_remove->@*) {
+       eval {
+           PVE::Storage::vdisk_free($storecfg, $volid);
+           print "cleaned up extracted image $volid\n";
+       };
+       warn "An error occurred while cleaning up source images: $@\n" if $@;
+    }
+
     if ($err) {
        warn "An error occurred during live-restore: $err\n";
        _do_vm_stop($storecfg, $vmid, 1, 1, 10, 0, 1);
diff --git a/PVE/QemuServer/Helpers.pm b/PVE/QemuServer/Helpers.pm
index 0afb6317..15e2496c 100644
--- a/PVE/QemuServer/Helpers.pm
+++ b/PVE/QemuServer/Helpers.pm
@@ -225,4 +225,9 @@ sub windows_version {
     return $winversion;
 }
 
+sub needs_extraction {
+    my ($vtype, $fmt) = @_;
+    return $vtype eq 'import' && $fmt =~ m/^ova\+(.*)$/;
+}
+
 1;
-- 
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