First, the provider is asked about what restore mechanism to use.
Currently, only 'qemu-img' is possible. Then the configuration files
are restored, the provider gives information about volumes contained
in the backup and finally the volumes are restored via
'qemu-img convert'.

The code for the restore_external_archive() function was copied and
adapted from the restore_proxmox_backup_archive() function. Together
with restore_vma_archive() it seems sensible to extract the common
parts and use a dedicated module for restore code.

The parse_restore_archive() helper was renamed, because it's not just
parsing.

Signed-off-by: Fiona Ebner <f.eb...@proxmox.com>
---
 PVE/API2/Qemu.pm  |  30 +++++++++--
 PVE/QemuServer.pm | 132 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 159 insertions(+), 3 deletions(-)

diff --git a/PVE/API2/Qemu.pm b/PVE/API2/Qemu.pm
index 9c644ff6..bcbfaa92 100644
--- a/PVE/API2/Qemu.pm
+++ b/PVE/API2/Qemu.pm
@@ -845,7 +845,7 @@ __PACKAGE__->register_method({
        return $res;
     }});
 
-my $parse_restore_archive = sub {
+my $classify_restore_archive = sub {
     my ($storecfg, $archive) = @_;
 
     my ($archive_storeid, $archive_volname) = 
PVE::Storage::parse_volume_id($archive, 1);
@@ -859,6 +859,21 @@ my $parse_restore_archive = sub {
            $res->{type} = 'pbs';
            return $res;
        }
+       my $log_function = sub {
+           my ($log_level, $message) = @_;
+           my $prefix = $log_level eq 'err' ? 'ERROR' : uc($log_level);
+           print "$prefix: $message\n";
+       };
+       my $backup_provider = PVE::Storage::new_backup_provider(
+           $storecfg,
+           $archive_storeid,
+           $log_function,
+       );
+       if ($backup_provider) {
+           $res->{type} = 'external';
+           $res->{'backup-provider'} = $backup_provider;
+           return $res;
+       }
     }
     my $path = PVE::Storage::abs_filesystem_path($storecfg, $archive);
     $res->{type} = 'file';
@@ -1011,7 +1026,7 @@ __PACKAGE__->register_method({
                    'backup',
                );
 
-               $archive = $parse_restore_archive->($storecfg, $archive);
+               $archive = $classify_restore_archive->($storecfg, $archive);
            }
        }
 
@@ -1056,6 +1071,7 @@ __PACKAGE__->register_method({
                    live => $live_restore,
                    override_conf => $param,
                };
+
                if (my $volid = $archive->{volid}) {
                    # best effort, real check is after restoring!
                    my $merged = eval {
@@ -1069,7 +1085,15 @@ __PACKAGE__->register_method({
                        PVE::QemuServer::check_restore_permissions($rpcenv, 
$authuser, $merged);
                    }
                }
-               if ($archive->{type} eq 'file' || $archive->{type} eq 'pipe') {
+               if (my $backup_provider = $archive->{'backup-provider'}) {
+                   PVE::QemuServer::restore_external_archive(
+                       $backup_provider,
+                       $archive->{volid},
+                       $vmid,
+                       $authuser,
+                       $restore_options,
+                   );
+               } elsif ($archive->{type} eq 'file' || $archive->{type} eq 
'pipe') {
                    die "live-restore is only compatible with backup images 
from a Proxmox Backup Server\n"
                        if $live_restore;
                    PVE::QemuServer::restore_file_archive($archive->{path} // 
'-', $vmid, $authuser, $restore_options);
diff --git a/PVE/QemuServer.pm b/PVE/QemuServer.pm
index 0db9f667..0c52e78d 100644
--- a/PVE/QemuServer.pm
+++ b/PVE/QemuServer.pm
@@ -7245,6 +7245,138 @@ sub restore_proxmox_backup_archive {
     }
 }
 
+sub restore_external_archive {
+    my ($backup_provider, $archive, $vmid, $user, $options) = @_;
+
+    die "live restore from backup provider is not implemented\n" if 
$options->{live};
+
+    my $storecfg = PVE::Storage::config();
+
+    my ($storeid, $volname) = PVE::Storage::parse_volume_id($archive);
+    my $scfg = PVE::Storage::storage_config($storecfg, $storeid);
+
+    my $tmpdir = "/var/tmp/vzdumptmp$$";
+    rmtree $tmpdir;
+    mkpath $tmpdir;
+
+    my $conffile = PVE::QemuConfig->config_file($vmid);
+    # disable interrupts (always do cleanups)
+    local $SIG{INT} =
+       local $SIG{TERM} =
+       local $SIG{QUIT} =
+       local $SIG{HUP} = sub { print STDERR "got interrupt - ignored\n"; };
+
+    # Note: $oldconf is undef if VM does not exists
+    my $cfs_path = PVE::QemuConfig->cfs_config_path($vmid);
+    my $oldconf = PVE::Cluster::cfs_read_file($cfs_path);
+    my $new_conf_raw = '';
+
+    my $rpcenv = PVE::RPCEnvironment::get();
+    my $devinfo = {}; # info about drives included in backup
+    my $virtdev_hash = {}; # info about allocated drives
+
+    eval {
+       # enable interrupts
+       local $SIG{INT} =
+           local $SIG{TERM} =
+           local $SIG{QUIT} =
+           local $SIG{HUP} =
+           local $SIG{PIPE} = sub { die "interrupted by signal\n"; };
+
+       my $cfgfn = "$tmpdir/qemu-server.conf";
+       my $firewall_config_fn = "$tmpdir/fw.conf";
+
+       my $cmd = "restore";
+
+       my ($mechanism, $vmtype) =
+           $backup_provider->restore_get_mechanism($volname, $storeid);
+       die "mechanism '$mechanism' requested by backup provider is not 
supported for VMs\n"
+           if $mechanism ne 'qemu-img';
+       die "cannot restore non-VM guest of type '$vmtype'\n" if $vmtype ne 
'qemu';
+
+       $devinfo = $backup_provider->restore_qemu_get_device_info($volname, 
$storeid);
+
+       my $data = $backup_provider->extract_guest_config($volname, $storeid)
+           or die "backup provider failed to extract guest configuration\n";
+       PVE::Tools::file_set_contents($cfgfn, $data);
+
+       if ($data = $backup_provider->extract_firewall_config($volname, 
$storeid)) {
+           PVE::Tools::file_set_contents($firewall_config_fn, $data);
+           my $pve_firewall_dir = '/etc/pve/firewall';
+           mkdir $pve_firewall_dir; # make sure the dir exists
+           PVE::Tools::file_copy($firewall_config_fn, 
"${pve_firewall_dir}/$vmid.fw");
+       }
+
+       my $fh = IO::File->new($cfgfn, "r") or die "unable to read 
qemu-server.conf - $!\n";
+
+       $virtdev_hash = $parse_backup_hints->($rpcenv, $user, $storecfg, $fh, 
$devinfo, $options);
+
+       # create empty/temp config
+       PVE::Tools::file_set_contents($conffile, "memory: 128\nlock: create");
+
+       $restore_cleanup_oldconf->($storecfg, $vmid, $oldconf, $virtdev_hash) 
if $oldconf;
+
+       # allocate volumes
+       my $map = $restore_allocate_devices->($storecfg, $virtdev_hash, $vmid);
+
+       for my $virtdev (sort keys $virtdev_hash->%*) {
+           my $d = $virtdev_hash->{$virtdev};
+           next if $d->{is_cloudinit}; # no need to restore cloudinit
+
+           my $source_path =
+               $backup_provider->restore_qemu_img_init($volname, $storeid, 
$d->{devname});
+           eval {
+               qemu_img_convert(
+                   $source_path, $d->{volid}, $d->{size}, undef, 0, 
$options->{bwlimit});
+           };
+           my $err = $@;
+           eval { $backup_provider->restore_qemu_img_cleanup($volname, 
$storeid, $d->{devname}); };
+           if (my $cleanup_err = $@) {
+               die $cleanup_err if !$err;
+               warn $cleanup_err;
+           }
+           die $err if $err
+       }
+
+       $fh->seek(0, 0) || die "seek failed - $!\n";
+
+       my $cookie = { netcount => 0 };
+       while (defined(my $line = <$fh>)) {
+           $new_conf_raw .= restore_update_config_line(
+               $cookie,
+               $map,
+               $line,
+               $options->{unique},
+           );
+       }
+
+       $fh->close();
+    };
+    my $err = $@;
+
+    if ($err) {
+       $restore_deactivate_volumes->($storecfg, $virtdev_hash);
+    }
+
+    rmtree $tmpdir;
+
+    if ($err) {
+       $restore_destroy_volumes->($storecfg, $virtdev_hash);
+       die $err;
+    }
+
+    my $new_conf = restore_merge_config($conffile, $new_conf_raw, 
$options->{override_conf});
+    check_restore_permissions($rpcenv, $user, $new_conf);
+    PVE::QemuConfig->write_config($vmid, $new_conf);
+
+    eval { rescan($vmid, 1); };
+    warn $@ if $@;
+
+    PVE::AccessControl::add_vm_to_pool($vmid, $options->{pool}) if 
$options->{pool};
+
+    return;
+}
+
 sub pbs_live_restore {
     my ($vmid, $conf, $storecfg, $restored_disks, $opts) = @_;
 
-- 
2.39.2



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

Reply via email to