pve-manager/PVE/VZDump.pm
Aaron Lauterer df5875b41e vzdump: move remaining guest include logic to single method
The `guest include` logic handling `all` and `exclude` parameters was in
the `PVE::VZDump->exec_backup()` method. Moving this logic into the
`get_included_guests` method allows us to simplify and generalize it.

This helps to make the overall logic easier to test and develop other
features around vzdump backup jobs.

The method now returns a hash with node names as keys mapped to arrays
of VMIDs on these nodes that are included in the vzdump job.

The VZDump API call to create a new backup is adapted to use the new
method to create the list of local VMIDs and the skiplist.

Permission checks are kept where they are to be able to handle missing
permissions according to the current context. The old behavior to die
on a backup job when the user is missing the permission to a guest and
the job is not an 'all' or 'exclude' job is kept.

Signed-off-by: Aaron Lauterer <a.lauterer@proxmox.com>
2020-06-17 15:16:06 +02:00

1206 lines
31 KiB
Perl

package PVE::VZDump;
use strict;
use warnings;
use Fcntl ':flock';
use File::Path;
use IO::File;
use IO::Select;
use IPC::Open3;
use POSIX qw(strftime);
use Time::Local;
use PVE::Cluster qw(cfs_read_file);
use PVE::DataCenterConfig;
use PVE::Exception qw(raise_param_exc);
use PVE::HA::Config;
use PVE::HA::Env::PVE2;
use PVE::JSONSchema qw(get_standard_option);
use PVE::RPCEnvironment;
use PVE::Storage;
use PVE::VZDump::Common;
use PVE::VZDump::Plugin;
use PVE::Tools qw(extract_param);
use PVE::API2Tools;
my @posix_filesystems = qw(ext3 ext4 nfs nfs4 reiserfs xfs);
my $lockfile = '/var/run/vzdump.lock';
my $pidfile = '/var/run/vzdump.pid';
my $logdir = '/var/log/vzdump';
my @plugins = qw();
my $confdesc = PVE::VZDump::Common::get_confdesc();
# Load available plugins
my @pve_vzdump_classes = qw(PVE::VZDump::QemuServer PVE::VZDump::LXC);
foreach my $plug (@pve_vzdump_classes) {
my $filename = "/usr/share/perl5/$plug.pm";
$filename =~ s!::!/!g;
if (-f $filename) {
eval { require $filename; };
if (!$@) {
$plug->import ();
push @plugins, $plug;
} else {
die $@;
}
}
}
# helper functions
sub debugmsg {
my ($mtype, $msg, $logfd, $syslog) = @_;
PVE::VZDump::Plugin::debugmsg(@_);
}
sub run_command {
my ($logfd, $cmdstr, %param) = @_;
my $logfunc = sub {
my $line = shift;
debugmsg ('info', $line, $logfd);
};
PVE::Tools::run_command($cmdstr, %param, logfunc => $logfunc);
}
sub storage_info {
my $storage = shift;
my $cfg = PVE::Storage::config();
my $scfg = PVE::Storage::storage_config($cfg, $storage);
my $type = $scfg->{type};
die "can't use storage type '$type' for backup\n"
if (!($type eq 'dir' || $type eq 'nfs' || $type eq 'glusterfs'
|| $type eq 'cifs' || $type eq 'cephfs' || $type eq 'pbs'));
die "can't use storage '$storage' for backups - wrong content type\n"
if (!$scfg->{content}->{backup});
PVE::Storage::activate_storage($cfg, $storage);
if ($type eq 'pbs') {
return {
scfg => $scfg,
maxfiles => $scfg->{maxfiles},
pbs => 1,
};
} else {
return {
scfg => $scfg,
dumpdir => PVE::Storage::get_backup_dir($cfg, $storage),
maxfiles => $scfg->{maxfiles},
};
}
}
sub format_size {
my $size = shift;
my $kb = $size / 1024;
if ($kb < 1024) {
return int ($kb) . "KB";
}
my $mb = $size / (1024*1024);
if ($mb < 1024) {
return int ($mb) . "MB";
}
my $gb = $mb / 1024;
if ($gb < 1024) {
return sprintf ("%.2fGB", $gb);
}
my $tb = $gb / 1024;
return sprintf ("%.2fTB", $tb);
}
sub format_time {
my $seconds = shift;
my $hours = int ($seconds/3600);
$seconds = $seconds - $hours*3600;
my $min = int ($seconds/60);
$seconds = $seconds - $min*60;
return sprintf ("%02d:%02d:%02d", $hours, $min, $seconds);
}
sub encode8bit {
my ($str) = @_;
$str =~ s/^(.{990})/$1\n/mg; # reduce line length
return $str;
}
sub escape_html {
my ($str) = @_;
$str =~ s/&/&amp;/g;
$str =~ s/</&lt;/g;
$str =~ s/>/&gt;/g;
return $str;
}
sub check_bin {
my ($bin) = @_;
foreach my $p (split (/:/, $ENV{PATH})) {
my $fn = "$p/$bin";
if (-x $fn) {
return $fn;
}
}
die "unable to find command '$bin'\n";
}
sub check_vmids {
my (@vmids) = @_;
my $res = [];
foreach my $vmid (@vmids) {
die "ERROR: strange VM ID '${vmid}'\n" if $vmid !~ m/^\d+$/;
$vmid = int ($vmid); # remove leading zeros
next if !$vmid;
push @$res, $vmid;
}
return $res;
}
sub read_vzdump_defaults {
my $fn = "/etc/vzdump.conf";
my $defaults = {
map {
my $default = $confdesc->{$_}->{default};
defined($default) ? ($_ => $default) : ()
} keys %$confdesc
};
my $raw;
eval { $raw = PVE::Tools::file_get_contents($fn); };
return $defaults if $@;
my $conf_schema = { type => 'object', properties => $confdesc, };
my $res = PVE::JSONSchema::parse_config($conf_schema, $fn, $raw);
if (my $excludes = $res->{'exclude-path'}) {
$res->{'exclude-path'} = PVE::Tools::split_args($excludes);
}
if (defined($res->{mailto})) {
my @mailto = PVE::Tools::split_list($res->{mailto});
$res->{mailto} = [ @mailto ];
}
foreach my $key (keys %$defaults) {
$res->{$key} = $defaults->{$key} if !defined($res->{$key});
}
return $res;
}
use constant MAX_MAIL_SIZE => 1024*1024;
sub sendmail {
my ($self, $tasklist, $totaltime, $err, $detail_pre, $detail_post) = @_;
my $opts = $self->{opts};
my $mailto = $opts->{mailto};
return if !($mailto && scalar(@$mailto));
my $cmdline = $self->{cmdline};
my $ecount = 0;
foreach my $task (@$tasklist) {
$ecount++ if $task->{state} ne 'ok';
chomp $task->{msg} if $task->{msg};
$task->{backuptime} = 0 if !$task->{backuptime};
$task->{size} = 0 if !$task->{size};
$task->{tarfile} = 'unknown' if !$task->{tarfile};
$task->{hostname} = "VM $task->{vmid}" if !$task->{hostname};
if ($task->{state} eq 'todo') {
$task->{msg} = 'aborted';
}
}
my $notify = $opts->{mailnotification} || 'always';
return if (!$ecount && !$err && ($notify eq 'failure'));
my $stat = ($ecount || $err) ? 'backup failed' : 'backup successful';
if ($err) {
if ($err =~ /\n/) {
$stat .= ": multiple problems";
} else {
$stat .= ": $err";
$err = undef;
}
}
my $hostname = `hostname -f` || PVE::INotify::nodename();
chomp $hostname;
# text part
my $text = $err ? "$err\n\n" : '';
$text .= sprintf ("%-10s %-6s %10s %10s %s\n", qw(VMID STATUS TIME SIZE FILENAME));
foreach my $task (@$tasklist) {
my $vmid = $task->{vmid};
if ($task->{state} eq 'ok') {
$text .= sprintf ("%-10s %-6s %10s %10s %s\n", $vmid,
$task->{state},
format_time($task->{backuptime}),
format_size ($task->{size}),
$task->{tarfile});
} else {
$text .= sprintf ("%-10s %-6s %10s %8.2fMB %s\n", $vmid,
$task->{state},
format_time($task->{backuptime}),
0, '-');
}
}
my $text_log_part;
$text_log_part .= "\nDetailed backup logs:\n\n";
$text_log_part .= "$cmdline\n\n";
$text_log_part .= $detail_pre . "\n" if defined($detail_pre);
foreach my $task (@$tasklist) {
my $vmid = $task->{vmid};
my $log = $task->{tmplog};
if (!$log) {
$text_log_part .= "$vmid: no log available\n\n";
next;
}
if (open (TMP, "$log")) {
while (my $line = <TMP>) {
next if $line =~ /^status: \d+/; # not useful in mails
$text_log_part .= encode8bit ("$vmid: $line");
}
} else {
$text_log_part .= "$vmid: Could not open log file\n\n";
}
close (TMP);
$text_log_part .= "\n";
}
$text_log_part .= $detail_post if defined($detail_post);
# html part
my $html = "<html><body>\n";
$html .= "<p>" . (escape_html($err) =~ s/\n/<br>/gr) . "</p>\n" if $err;
$html .= "<table border=1 cellpadding=3>\n";
$html .= "<tr><td>VMID<td>NAME<td>STATUS<td>TIME<td>SIZE<td>FILENAME</tr>\n";
my $ssize = 0;
foreach my $task (@$tasklist) {
my $vmid = $task->{vmid};
my $name = $task->{hostname};
if ($task->{state} eq 'ok') {
$ssize += $task->{size};
$html .= sprintf ("<tr><td>%s<td>%s<td>OK<td>%s<td align=right>%s<td>%s</tr>\n",
$vmid, $name,
format_time($task->{backuptime}),
format_size ($task->{size}),
escape_html ($task->{tarfile}));
} else {
$html .= sprintf ("<tr><td>%s<td>%s<td><font color=red>FAILED<td>%s<td colspan=2>%s</tr>\n",
$vmid, $name, format_time($task->{backuptime}),
escape_html ($task->{msg}));
}
}
$html .= sprintf ("<tr><td align=left colspan=3>TOTAL<td>%s<td>%s<td></tr>",
format_time ($totaltime), format_size ($ssize));
$html .= "\n</table><br><br>\n";
my $html_log_part;
$html_log_part .= "Detailed backup logs:<br /><br />\n";
$html_log_part .= "<pre>\n";
$html_log_part .= escape_html($cmdline) . "\n\n";
$html_log_part .= escape_html($detail_pre) . "\n" if defined($detail_pre);
foreach my $task (@$tasklist) {
my $vmid = $task->{vmid};
my $log = $task->{tmplog};
if (!$log) {
$html_log_part .= "$vmid: no log available\n\n";
next;
}
if (open (TMP, "$log")) {
while (my $line = <TMP>) {
next if $line =~ /^status: \d+/; # not useful in mails
if ($line =~ m/^\S+\s\d+\s+\d+:\d+:\d+\s+(ERROR|WARN):/) {
$html_log_part .= encode8bit ("$vmid: <font color=red>".
escape_html ($line) . "</font>");
} else {
$html_log_part .= encode8bit ("$vmid: " . escape_html ($line));
}
}
} else {
$html_log_part .= "$vmid: Could not open log file\n\n";
}
close (TMP);
$html_log_part .= "\n";
}
$html_log_part .= escape_html($detail_post) if defined($detail_post);
$html_log_part .= "</pre>";
my $html_end .= "\n</body></html>\n";
# end html part
if (length($text) + length($text_log_part) +
length($html) + length($html_log_part) < MAX_MAIL_SIZE)
{
$html .= $html_log_part;
$text .= $text_log_part;
} else {
my $msg = "Log output was too long to be sent by mail. ".
"See Task History for details!\n";
$text .= $msg;
$html .= "<p>$msg</p>";
$html .= $html_end;
}
my $subject = "vzdump backup status ($hostname) : $stat";
my $dcconf = PVE::Cluster::cfs_read_file('datacenter.cfg');
my $mailfrom = $dcconf->{email_from} || "root";
PVE::Tools::sendmail($mailto, $subject, $text, $html, $mailfrom, "vzdump backup tool");
};
sub new {
my ($class, $cmdline, $opts, $skiplist) = @_;
mkpath $logdir;
check_bin ('cp');
check_bin ('df');
check_bin ('sendmail');
check_bin ('rsync');
check_bin ('tar');
check_bin ('mount');
check_bin ('umount');
check_bin ('cstream');
check_bin ('ionice');
if ($opts->{mode} && $opts->{mode} eq 'snapshot') {
check_bin ('lvcreate');
check_bin ('lvs');
check_bin ('lvremove');
}
my $defaults = read_vzdump_defaults();
my $maxfiles = $opts->{maxfiles}; # save here, because we overwrite with default
$opts->{remove} = 1 if !defined($opts->{remove});
foreach my $k (keys %$defaults) {
next if $k eq 'exclude-path'; # dealt with separately
if ($k eq 'dumpdir' || $k eq 'storage') {
$opts->{$k} = $defaults->{$k} if !defined ($opts->{dumpdir}) &&
!defined ($opts->{storage});
} else {
$opts->{$k} = $defaults->{$k} if !defined ($opts->{$k});
}
}
$opts->{dumpdir} =~ s|/+$|| if ($opts->{dumpdir});
$opts->{tmpdir} =~ s|/+$|| if ($opts->{tmpdir});
$skiplist = [] if !$skiplist;
my $self = bless { cmdline => $cmdline, opts => $opts, skiplist => $skiplist };
my $findexcl = $self->{findexcl} = [];
if ($defaults->{'exclude-path'}) {
push @$findexcl, @{$defaults->{'exclude-path'}};
}
if ($opts->{'exclude-path'}) {
push @$findexcl, @{$opts->{'exclude-path'}};
}
if ($opts->{stdexcludes}) {
push @$findexcl, '/tmp/?*',
'/var/tmp/?*',
'/var/run/?*.pid';
}
foreach my $p (@plugins) {
my $pd = $p->new ($self);
push @{$self->{plugins}}, $pd;
}
if (defined($opts->{storage}) && $opts->{stdout}) {
die "unable to use option 'storage' with option 'stdout'\n";
}
if (!$opts->{dumpdir} && !$opts->{storage}) {
$opts->{storage} = 'local';
}
my $errors = '';
if ($opts->{storage}) {
my $info = eval { storage_info ($opts->{storage}) };
$errors .= "could not get storage information for '$opts->{storage}': $@"
if ($@);
$opts->{dumpdir} = $info->{dumpdir};
$opts->{scfg} = $info->{scfg};
$opts->{pbs} = $info->{pbs};
$maxfiles //= $info->{maxfiles};
} elsif ($opts->{dumpdir}) {
$errors .= "dumpdir '$opts->{dumpdir}' does not exist"
if ! -d $opts->{dumpdir};
} else {
die "internal error";
}
if ($opts->{tmpdir} && ! -d $opts->{tmpdir}) {
$errors .= "\n" if $errors;
$errors .= "tmpdir '$opts->{tmpdir}' does not exist";
}
if ($errors) {
eval { $self->sendmail([], 0, $errors); };
debugmsg ('err', $@) if $@;
die "$errors\n";
}
$opts->{maxfiles} = $maxfiles if defined($maxfiles);
return $self;
}
sub get_mount_info {
my ($dir) = @_;
# Note: df 'available' can be negative, and percentage set to '-'
my $cmd = [ 'df', '-P', '-T', '-B', '1', $dir];
my $res;
my $parser = sub {
my $line = shift;
if (my ($fsid, $fstype, undef, $mp) = $line =~
m!(\S+.*)\s+(\S+)\s+\d+\s+\-?\d+\s+\d+\s+(\d+%|-)\s+(/.*)$!) {
$res = {
device => $fsid,
fstype => $fstype,
mountpoint => $mp,
};
}
};
eval { PVE::Tools::run_command($cmd, errfunc => sub {}, outfunc => $parser); };
warn $@ if $@;
return $res;
}
sub getlock {
my ($self, $upid) = @_;
my $fh;
my $maxwait = $self->{opts}->{lockwait} || $self->{lockwait};
die "missimg UPID" if !$upid; # should not happen
if (!open (SERVER_FLCK, ">>$lockfile")) {
debugmsg ('err', "can't open lock on file '$lockfile' - $!", undef, 1);
die "can't open lock on file '$lockfile' - $!";
}
if (!flock (SERVER_FLCK, LOCK_EX|LOCK_NB)) {
if (!$maxwait) {
debugmsg ('err', "can't acquire lock '$lockfile' (wait = 0)", undef, 1);
die "can't acquire lock '$lockfile' (wait = 0)";
}
debugmsg('info', "trying to get global lock - waiting...", undef, 1);
eval {
alarm ($maxwait * 60);
local $SIG{ALRM} = sub { alarm (0); die "got timeout\n"; };
if (!flock (SERVER_FLCK, LOCK_EX)) {
my $err = $!;
close (SERVER_FLCK);
alarm (0);
die "$err\n";
}
alarm (0);
};
alarm (0);
my $err = $@;
if ($err) {
debugmsg ('err', "can't acquire lock '$lockfile' - $err", undef, 1);
die "can't acquire lock '$lockfile' - $err";
}
debugmsg('info', "got global lock", undef, 1);
}
PVE::Tools::file_set_contents($pidfile, $upid);
}
sub run_hook_script {
my ($self, $phase, $task, $logfd) = @_;
my $opts = $self->{opts};
my $script = $opts->{script};
return if !$script;
if (!-x $script) {
die "The hook script '$script' is not executable.\n";
}
my $cmd = "$script $phase";
$cmd .= " $task->{mode} $task->{vmid}" if ($task);
local %ENV;
# set immutable opts directly (so they are available in all phases)
$ENV{STOREID} = $opts->{storage} if $opts->{storage};
$ENV{DUMPDIR} = $opts->{dumpdir} if $opts->{dumpdir};
foreach my $ek (qw(vmtype hostname tarfile logfile)) {
$ENV{uc($ek)} = $task->{$ek} if $task->{$ek};
}
run_command ($logfd, $cmd);
}
sub compressor_info {
my ($opts) = @_;
my $opt_compress = $opts->{compress};
if (!$opt_compress || $opt_compress eq '0') {
return undef;
} elsif ($opt_compress eq '1' || $opt_compress eq 'lzo') {
return ('lzop', 'lzo');
} elsif ($opt_compress eq 'gzip') {
if ($opts->{pigz} > 0) {
my $pigz_threads = $opts->{pigz};
if ($pigz_threads == 1) {
my $cpuinfo = PVE::ProcFSTools::read_cpuinfo();
$pigz_threads = int(($cpuinfo->{cpus} + 1)/2);
}
return ("pigz -p ${pigz_threads} --rsyncable", 'gz');
} else {
return ('gzip --rsyncable', 'gz');
}
} elsif ($opt_compress eq 'zstd') {
my $zstd_threads = $opts->{zstd} // 1;
if ($zstd_threads == 0) {
my $cpuinfo = PVE::ProcFSTools::read_cpuinfo();
$zstd_threads = int(($cpuinfo->{cpus} + 1)/2);
}
return ("zstd --rsyncable --threads=${zstd_threads}", 'zst');
} else {
die "internal error - unknown compression option '$opt_compress'";
}
}
sub get_backup_file_list {
my ($dir, $bkname, $exclude_fn) = @_;
my $bklist = [];
foreach my $fn (<$dir/${bkname}-*>) {
next if $exclude_fn && $fn eq $exclude_fn;
if ($fn =~ m!/(${bkname}-(\d{4})_(\d{2})_(\d{2})-(\d{2})_(\d{2})_(\d{2})\.(tgz|((tar|vma)(\.(${\PVE::Storage::Plugin::COMPRESSOR_RE}))?)))$!) {
$fn = "$dir/$1"; # untaint
my $t = timelocal ($7, $6, $5, $4, $3 - 1, $2);
push @$bklist, [$fn, $t];
}
}
return $bklist;
}
sub exec_backup_task {
my ($self, $task) = @_;
my $opts = $self->{opts};
my $vmid = $task->{vmid};
my $plugin = $task->{plugin};
my $vmtype = $plugin->type();
$task->{backup_time} = time();
my $pbs_group_name;
my $pbs_snapshot_name;
if ($self->{opts}->{pbs}) {
if ($vmtype eq 'lxc') {
$pbs_group_name = "ct/$vmid";
} elsif ($vmtype eq 'qemu') {
$pbs_group_name = "vm/$vmid";
} else {
die "pbs backup not implemented for plugin type '$vmtype'\n";
}
my $btime = strftime("%FT%TZ", gmtime($task->{backup_time}));
$pbs_snapshot_name = "$pbs_group_name/$btime";
}
my $vmstarttime = time ();
my $logfd;
my $cleanup = {};
my $log_vm_online_again = sub {
return if !defined($task->{vmstoptime});
$task->{vmconttime} //= time();
my $delay = $task->{vmconttime} - $task->{vmstoptime};
debugmsg ('info', "guest is online again after $delay seconds", $logfd);
};
eval {
die "unable to find VM '$vmid'\n" if !$plugin;
# for now we deny backups of a running ha managed service in *stop* mode
# as it interferes with the HA stack (started services should not stop).
if ($opts->{mode} eq 'stop' &&
PVE::HA::Config::vm_is_ha_managed($vmid, 'started'))
{
die "Cannot execute a backup with stop mode on a HA managed and".
" enabled Service. Use snapshot mode or disable the Service.\n";
}
my $tmplog = "$logdir/$vmtype-$vmid.log";
my $bkname = "vzdump-$vmtype-$vmid";
my $basename = $bkname . strftime("-%Y_%m_%d-%H_%M_%S", localtime($task->{backup_time}));
my $maxfiles = $opts->{maxfiles};
if ($maxfiles && !$opts->{remove}) {
my $count;
if ($self->{opts}->{pbs}) {
my $res = PVE::Storage::PBSPlugin::run_client_cmd($opts->{scfg}, $opts->{storage}, 'snapshots', $pbs_group_name);
$count = scalar(@$res);
} else {
my $bklist = get_backup_file_list($opts->{dumpdir}, $bkname);
$count = scalar(@$bklist);
}
die "There is a max backup limit of ($maxfiles) enforced by the".
" target storage or the vzdump parameters.".
" Either increase the limit or delete old backup(s).\n"
if $count >= $maxfiles;
}
if (!$self->{opts}->{pbs}) {
$task->{logfile} = "$opts->{dumpdir}/$basename.log";
}
my $ext = $vmtype eq 'qemu' ? '.vma' : '.tar';
my ($comp, $comp_ext) = compressor_info($opts);
if ($comp && $comp_ext) {
$ext .= ".${comp_ext}";
}
if ($self->{opts}->{pbs}) {
die "unable to pipe backup to stdout\n" if $opts->{stdout};
} else {
if ($opts->{stdout}) {
$task->{tarfile} = '-';
} else {
my $tarfile = $task->{tarfile} = "$opts->{dumpdir}/$basename$ext";
$task->{tmptar} = $task->{tarfile};
$task->{tmptar} =~ s/\.[^\.]+$/\.dat/;
unlink $task->{tmptar};
}
}
$task->{vmtype} = $vmtype;
if ($self->{opts}->{pbs}) {
$task->{tmpdir} = "/var/tmp/vzdumptmp$$"; #fixme
} elsif ($opts->{tmpdir}) {
$task->{tmpdir} = "$opts->{tmpdir}/vzdumptmp$$";
} else {
# dumpdir is posix? then use it as temporary dir
my $info = get_mount_info($opts->{dumpdir});
if ($vmtype eq 'qemu' ||
grep ($_ eq $info->{fstype}, @posix_filesystems)) {
$task->{tmpdir} = "$opts->{dumpdir}/$basename.tmp";
} else {
$task->{tmpdir} = "/var/tmp/vzdumptmp$$";
debugmsg ('info', "filesystem type on dumpdir is '$info->{fstype}' -" .
"using $task->{tmpdir} for temporary files", $logfd);
}
}
rmtree $task->{tmpdir};
mkdir $task->{tmpdir};
-d $task->{tmpdir} ||
die "unable to create temporary directory '$task->{tmpdir}'";
$logfd = IO::File->new (">$tmplog") ||
die "unable to create log file '$tmplog'";
$task->{dumpdir} = $opts->{dumpdir};
$task->{storeid} = $opts->{storage};
$task->{scfg} = $opts->{scfg};
$task->{tmplog} = $tmplog;
unlink $task->{logfile} if defined($task->{logfile});
debugmsg ('info', "Starting Backup of VM $vmid ($vmtype)", $logfd, 1);
debugmsg ('info', "Backup started at " . strftime("%F %H:%M:%S", localtime()));
$plugin->set_logfd ($logfd);
# test is VM is running
my ($running, $status_text) = $plugin->vm_status ($vmid);
debugmsg ('info', "status = ${status_text}", $logfd);
# lock VM (prevent config changes)
$plugin->lock_vm ($vmid);
$cleanup->{unlock} = 1;
# prepare
my $mode = $running ? $task->{mode} : 'stop';
if ($mode eq 'snapshot') {
my %saved_task = %$task;
eval { $plugin->prepare ($task, $vmid, $mode); };
if (my $err = $@) {
die $err if $err !~ m/^mode failure/;
debugmsg ('info', $err, $logfd);
debugmsg ('info', "trying 'suspend' mode instead", $logfd);
$mode = 'suspend'; # so prepare is called again below
%$task = %saved_task;
}
}
$cleanup->{prepared} = 1;
$task->{mode} = $mode;
debugmsg ('info', "backup mode: $mode", $logfd);
debugmsg ('info', "bandwidth limit: $opts->{bwlimit} KB/s", $logfd)
if $opts->{bwlimit};
debugmsg ('info', "ionice priority: $opts->{ionice}", $logfd);
if ($mode eq 'stop') {
$plugin->prepare ($task, $vmid, $mode);
$self->run_hook_script ('backup-start', $task, $logfd);
if ($running) {
debugmsg ('info', "stopping vm", $logfd);
$task->{vmstoptime} = time();
$self->run_hook_script ('pre-stop', $task, $logfd);
$plugin->stop_vm ($task, $vmid);
$cleanup->{restart} = 1;
}
} elsif ($mode eq 'suspend') {
$plugin->prepare ($task, $vmid, $mode);
$self->run_hook_script ('backup-start', $task, $logfd);
if ($vmtype eq 'lxc') {
# pre-suspend rsync
$plugin->copy_data_phase1($task, $vmid);
}
debugmsg ('info', "suspend vm", $logfd);
$task->{vmstoptime} = time ();
$self->run_hook_script ('pre-stop', $task, $logfd);
$plugin->suspend_vm ($task, $vmid);
$cleanup->{resume} = 1;
if ($vmtype eq 'lxc') {
# post-suspend rsync
$plugin->copy_data_phase2($task, $vmid);
debugmsg ('info', "resume vm", $logfd);
$cleanup->{resume} = 0;
$self->run_hook_script('pre-restart', $task, $logfd);
$plugin->resume_vm($task, $vmid);
$self->run_hook_script('post-restart', $task, $logfd);
$log_vm_online_again->();
}
} elsif ($mode eq 'snapshot') {
$self->run_hook_script ('backup-start', $task, $logfd);
my $snapshot_count = $task->{snapshot_count} || 0;
$self->run_hook_script ('pre-stop', $task, $logfd);
if ($snapshot_count > 1) {
debugmsg ('info', "suspend vm to make snapshot", $logfd);
$task->{vmstoptime} = time ();
$plugin->suspend_vm ($task, $vmid);
$cleanup->{resume} = 1;
}
$plugin->snapshot ($task, $vmid);
$self->run_hook_script ('pre-restart', $task, $logfd);
if ($snapshot_count > 1) {
debugmsg ('info', "resume vm", $logfd);
$cleanup->{resume} = 0;
$plugin->resume_vm ($task, $vmid);
$log_vm_online_again->();
}
$self->run_hook_script ('post-restart', $task, $logfd);
} else {
die "internal error - unknown mode '$mode'\n";
}
# assemble archive image
$plugin->assemble ($task, $vmid);
# produce archive
if ($opts->{stdout}) {
debugmsg ('info', "sending archive to stdout", $logfd);
$plugin->archive($task, $vmid, $task->{tmptar}, $comp);
$self->run_hook_script ('backup-end', $task, $logfd);
return;
}
# fixme: ??
if ($self->{opts}->{pbs}) {
debugmsg ('info', "creating pbs archive on storage '$opts->{storage}'", $logfd);
} else {
debugmsg ('info', "creating archive '$task->{tarfile}'", $logfd);
}
$plugin->archive($task, $vmid, $task->{tmptar}, $comp);
if ($self->{opts}->{pbs}) {
# fixme: log size ?
debugmsg ('info', "pbs upload finished", $logfd);
} else {
rename ($task->{tmptar}, $task->{tarfile}) ||
die "unable to rename '$task->{tmptar}' to '$task->{tarfile}'\n";
# determine size
$task->{size} = (-s $task->{tarfile}) || 0;
my $cs = format_size ($task->{size});
debugmsg ('info', "archive file size: $cs", $logfd);
}
# purge older backup
if ($maxfiles && $opts->{remove}) {
if ($self->{opts}->{pbs}) {
my $args = [$pbs_group_name, '--quiet', '1', '--keep-last', $maxfiles];
my $logfunc = sub { my $line = shift; debugmsg ('info', $line, $logfd); };
PVE::Storage::PBSPlugin::run_raw_client_cmd(
$opts->{scfg}, $opts->{storage}, 'prune', $args, logfunc => $logfunc);
} else {
my $bklist = get_backup_file_list($opts->{dumpdir}, $bkname, $task->{tarfile});
$bklist = [ sort { $b->[1] <=> $a->[1] } @$bklist ];
while (scalar (@$bklist) >= $maxfiles) {
my $d = pop @$bklist;
debugmsg ('info', "delete old backup '$d->[0]'", $logfd);
unlink $d->[0];
my $logfn = $d->[0];
$logfn =~ s/\.(tgz|((tar|vma)(\.(${\PVE::Storage::Plugin::COMPRESSOR_RE}))?))$/\.log/;
unlink $logfn;
}
}
}
$self->run_hook_script ('backup-end', $task, $logfd);
};
my $err = $@;
if ($plugin) {
# clean-up
if ($cleanup->{unlock}) {
eval { $plugin->unlock_vm ($vmid); };
warn $@ if $@;
}
if ($cleanup->{prepared}) {
# only call cleanup when necessary (when prepare was executed)
eval { $plugin->cleanup ($task, $vmid) };
warn $@ if $@;
}
eval { $plugin->set_logfd (undef); };
warn $@ if $@;
if ($cleanup->{resume} || $cleanup->{restart}) {
eval {
$self->run_hook_script ('pre-restart', $task, $logfd);
if ($cleanup->{resume}) {
debugmsg ('info', "resume vm", $logfd);
$plugin->resume_vm ($task, $vmid);
} else {
my $running = $plugin->vm_status($vmid);
if (!$running) {
debugmsg ('info', "restarting vm", $logfd);
$plugin->start_vm ($task, $vmid);
}
}
$self->run_hook_script ('post-restart', $task, $logfd);
};
my $err = $@;
if ($err) {
warn $err;
} else {
$log_vm_online_again->();
}
}
}
eval { unlink $task->{tmptar} if $task->{tmptar} && -f $task->{tmptar}; };
warn $@ if $@;
eval { rmtree $task->{tmpdir} if $task->{tmpdir} && -d $task->{tmpdir}; };
warn $@ if $@;
my $delay = $task->{backuptime} = time () - $vmstarttime;
if ($err) {
$task->{state} = 'err';
$task->{msg} = $err;
debugmsg ('err', "Backup of VM $vmid failed - $err", $logfd, 1);
debugmsg ('info', "Failed at " . strftime("%F %H:%M:%S", localtime()));
eval { $self->run_hook_script ('backup-abort', $task, $logfd); };
} else {
$task->{state} = 'ok';
my $tstr = format_time ($delay);
debugmsg ('info', "Finished Backup of VM $vmid ($tstr)", $logfd, 1);
debugmsg ('info', "Backup finished at " . strftime("%F %H:%M:%S", localtime()));
}
close ($logfd) if $logfd;
if ($task->{tmplog}) {
if ($self->{opts}->{pbs}) {
if ($task->{state} eq 'ok') {
my $param = [$pbs_snapshot_name, $task->{tmplog}];
PVE::Storage::PBSPlugin::run_raw_client_cmd(
$opts->{scfg}, $opts->{storage}, 'upload-log', $param, errmsg => "upload log failed");
}
} elsif ($task->{logfile}) {
system {'cp'} 'cp', $task->{tmplog}, $task->{logfile};
}
}
eval { $self->run_hook_script ('log-end', $task); };
die $err if $err && $err =~ m/^interrupted by signal$/;
}
sub exec_backup {
my ($self, $rpcenv, $authuser) = @_;
my $opts = $self->{opts};
debugmsg ('info', "starting new backup job: $self->{cmdline}", undef, 1);
debugmsg ('info', "skip external VMs: " . join(', ', @{$self->{skiplist}}))
if scalar(@{$self->{skiplist}});
my $tasklist = [];
my $vzdump_plugins = {};
foreach my $plugin (@{$self->{plugins}}) {
my $type = $plugin->type();
next if exists $vzdump_plugins->{$type};
$vzdump_plugins->{$type} = $plugin;
}
my $vmlist = PVE::Cluster::get_vmlist();
foreach my $vmid (sort @{$opts->{vmids}}) {
my $guest_type = $vmlist->{ids}->{$vmid}->{type};
my $plugin = $vzdump_plugins->{$guest_type};
next if !$rpcenv->check($authuser, "/vms/$vmid", [ 'VM.Backup' ], $opts->{all});
push @$tasklist, {
vmid => $vmid,
state => 'todo',
plugin => $plugin,
mode => $opts->{mode} };
}
# Use in-memory files for the outer hook logs to pass them to sendmail.
my $job_start_log = '';
my $job_end_log = '';
open my $job_start_fd, '>', \$job_start_log;
open my $job_end_fd, '>', \$job_end_log;
my $starttime = time();
my $errcount = 0;
eval {
$self->run_hook_script ('job-start', undef, $job_start_fd);
foreach my $task (@$tasklist) {
$self->exec_backup_task ($task);
$errcount += 1 if $task->{state} ne 'ok';
}
$self->run_hook_script ('job-end', undef, $job_end_fd);
};
my $err = $@;
$self->run_hook_script ('job-abort', undef, $job_end_fd) if $err;
if ($err) {
debugmsg ('err', "Backup job failed - $err", undef, 1);
} else {
if ($errcount) {
debugmsg ('info', "Backup job finished with errors", undef, 1);
} else {
debugmsg ('info', "Backup job finished successfully", undef, 1);
}
}
close $job_start_fd;
close $job_end_fd;
my $totaltime = time() - $starttime;
eval { $self->sendmail ($tasklist, $totaltime, undef, $job_start_log, $job_end_log); };
debugmsg ('err', $@) if $@;
die $err if $err;
die "job errors\n" if $errcount;
unlink $pidfile;
}
sub option_exists {
my $key = shift;
return defined($confdesc->{$key});
}
sub verify_vzdump_parameters {
my ($param, $check_missing) = @_;
raise_param_exc({ all => "option conflicts with option 'vmid'"})
if $param->{all} && $param->{vmid};
raise_param_exc({ exclude => "option conflicts with option 'vmid'"})
if $param->{exclude} && $param->{vmid};
raise_param_exc({ pool => "option conflicts with option 'vmid'"})
if $param->{pool} && $param->{vmid};
$param->{all} = 1 if (defined($param->{exclude}) && !$param->{pool});
warn "option 'size' is deprecated and will be removed in a future " .
"release, please update your script/configuration!\n"
if defined($param->{size});
return if !$check_missing;
raise_param_exc({ vmid => "property is missing"})
if !($param->{all} || $param->{stop} || $param->{pool}) && !$param->{vmid};
}
sub stop_running_backups {
my($self) = @_;
my $upid = PVE::Tools::file_read_firstline($pidfile);
return if !$upid;
my $task = PVE::Tools::upid_decode($upid);
if (PVE::ProcFSTools::check_process_running($task->{pid}, $task->{pstart}) &&
PVE::ProcFSTools::read_proc_starttime($task->{pid}) == $task->{pstart}) {
kill(15, $task->{pid});
# wait max 15 seconds to shut down (else, do nothing for now)
my $i;
for ($i = 15; $i > 0; $i--) {
last if !PVE::ProcFSTools::check_process_running(($task->{pid}, $task->{pstart}));
sleep (1);
}
die "stopping backup process $task->{pid} failed\n" if $i == 0;
}
}
sub get_included_guests {
my ($job) = @_;
my $nodename = PVE::INotify::nodename();
my $vmids = [];
my $vmids_per_node = {};
my $vmlist = PVE::Cluster::get_vmlist();
if ($job->{pool}) {
$vmids = PVE::API2Tools::get_resource_pool_guest_members($job->{pool});
} elsif ($job->{vmid}) {
$vmids = [ PVE::Tools::split_list($job->{vmid}) ];
} else {
# all or exclude
my @exclude = PVE::Tools::split_list($job->{exclude});
@exclude = @{PVE::VZDump::check_vmids(@exclude)};
my $excludehash = { map { $_ => 1 } @exclude };
for my $id (keys %{ $vmlist->{ids} }) {
next if $excludehash->{$id};
push @$vmids, $id;
}
}
$vmids = [ sort {$a <=> $b} @$vmids];
$vmids = PVE::VZDump::check_vmids(@$vmids);
foreach my $vmid (@$vmids) {
my $vmid_data = $vmlist->{ids}->{$vmid};
my $node = $vmid_data->{node};
next if (defined $job->{node} && $job->{node} ne $node);
push @{$vmids_per_node->{$node}}, $vmid;
}
return $vmids_per_node;
}
1;