#!/usr/bin/perl
-my $PROGNAME = "pve-zsync";
-my $CONFIG_PATH = '/var/lib/'.$PROGNAME.'/';
-my $CONFIG = "$PROGNAME.cfg";
-my $CRONJOBS = '/etc/cron.d/'.$PROGNAME;
-my $VMCONFIG = '/var/lib/'.$PROGNAME.'/';
-my $PATH = "/usr/sbin/";
-my $QEMU_CONF = '/etc/pve/local/qemu-server/';
-my $DEBUG = 0;
-
use strict;
use warnings;
use Data::Dumper qw(Dumper);
use Fcntl qw(:flock SEEK_END);
-use Getopt::Long;
-use Switch;
+use Getopt::Long qw(GetOptionsFromArray);
+use File::Copy qw(move);
+use File::Path qw(make_path);
+use JSON;
+use IO::File;
+use String::ShellQuote 'shell_quote';
+
+my $PROGNAME = "pve-zsync";
+my $CONFIG_PATH = "/var/lib/${PROGNAME}";
+my $STATE = "${CONFIG_PATH}/sync_state";
+my $CRONJOBS = "/etc/cron.d/$PROGNAME";
+my $PATH = "/usr/sbin";
+my $PVE_DIR = "/etc/pve/local";
+my $QEMU_CONF = "${PVE_DIR}/qemu-server";
+my $LXC_CONF = "${PVE_DIR}/lxc";
+my $LOCKFILE = "$CONFIG_PATH/${PROGNAME}.lock";
+my $PROG_PATH = "$PATH/${PROGNAME}";
+my $INTERVAL = 15;
+my $DEBUG = 0;
+
+my $IPV4OCTET = "(?:25[0-5]|(?:[1-9]|1[0-9]|2[0-4])?[0-9])";
+my $IPV4RE = "(?:(?:$IPV4OCTET\\.){3}$IPV4OCTET)";
+my $IPV6H16 = "(?:[0-9a-fA-F]{1,4})";
+my $IPV6LS32 = "(?:(?:$IPV4RE|$IPV6H16:$IPV6H16))";
+
+my $IPV6RE = "(?:" .
+ "(?:(?:" . "(?:$IPV6H16:){6})$IPV6LS32)|" .
+ "(?:(?:" . "::(?:$IPV6H16:){5})$IPV6LS32)|" .
+ "(?:(?:(?:" . "$IPV6H16)?::(?:$IPV6H16:){4})$IPV6LS32)|" .
+ "(?:(?:(?:(?:$IPV6H16:){0,1}$IPV6H16)?::(?:$IPV6H16:){3})$IPV6LS32)|" .
+ "(?:(?:(?:(?:$IPV6H16:){0,2}$IPV6H16)?::(?:$IPV6H16:){2})$IPV6LS32)|" .
+ "(?:(?:(?:(?:$IPV6H16:){0,3}$IPV6H16)?::(?:$IPV6H16:){1})$IPV6LS32)|" .
+ "(?:(?:(?:(?:$IPV6H16:){0,4}$IPV6H16)?::" . ")$IPV6LS32)|" .
+ "(?:(?:(?:(?:$IPV6H16:){0,5}$IPV6H16)?::" . ")$IPV6H16)|" .
+ "(?:(?:(?:(?:$IPV6H16:){0,6}$IPV6H16)?::" . ")))";
+
+my $HOSTv4RE0 = "(?:[\\w\\.\\-_]+|$IPV4RE)"; # hostname or ipv4 address
+my $HOSTv4RE1 = "(?:$HOSTv4RE0|\\[$HOSTv4RE0\\])"; # these may be in brackets, too
+my $HOSTRE = "(?:$HOSTv4RE1|\\[$IPV6RE\\])"; # ipv6 must always be in brackets
+# targets are either a VMID, or a 'host:zpool/path' with 'host:' being optional
+my $TARGETRE = qr!^(?:($HOSTRE):)?(\d+|(?:[\w\-_]+)(/.+)?)$!;
+
+my $command = $ARGV[0];
+
+if (defined($command) && $command ne 'help' && $command ne 'printpod') {
+ check_bin ('cstream');
+ check_bin ('zfs');
+ check_bin ('ssh');
+ check_bin ('scp');
+}
-check_bin ('cstream');
-check_bin ('zfs');
-check_bin ('ssh');
-check_bin ('scp');
+$SIG{TERM} = $SIG{QUIT} = $SIG{PIPE} = $SIG{HUP} = $SIG{KILL} = $SIG{INT} =
+ sub {
+ die "Signal aborting sync\n";
+ };
sub check_bin {
my ($bin) = @_;
foreach my $p (split (/:/, $ENV{PATH})) {
- my $fn = "$p/$bin";
- if (-x $fn) {
- return $fn;
+ my $fn = "$p/$bin";
+ if (-x $fn) {
+ return $fn;
}
}
- warn "unable to find command '$bin'\n";
+ die "unable to find command '$bin'\n";
}
-sub cut_to_width {
- my ($text, $max) = @_;
+sub cut_target_width {
+ my ($path, $maxlen) = @_;
+ $path =~ s@/+@/@g;
+
+ return $path if length($path) <= $maxlen;
+
+ return '..'.substr($path, -$maxlen+2) if $path !~ m@/@;
- return $text if (length($text) <= $max);
- my @spl = split('/', $text);
+ $path =~ s@/([^/]+/?)$@@;
+ my $tail = $1;
- my $count = length($spl[@spl-1]);
- return "..\/".substr($spl[@spl-1],($count-$max)+3 ,$count) if $count > $max;
+ if (length($tail)+3 == $maxlen) {
+ return "../$tail";
+ } elsif (length($tail)+2 >= $maxlen) {
+ return '..'.substr($tail, -$maxlen+2)
+ }
- $count += length($spl[0]) if @spl > 1;
- return substr($spl[0], 0, $max-4-length($spl[@spl-1]))."\/..\/".$spl[@spl-1] if $count > $max;
+ $path =~ s@(/[^/]+)(?:/|$)@@;
+ my $head = $1;
+ my $both = length($head) + length($tail);
+ my $remaining = $maxlen-$both-4; # -4 for "/../"
- my $rest = 1 ;
- $rest = $max-$count if ($max-$count > 0);
+ if ($remaining < 0) {
+ return substr($head, 0, $maxlen - length($tail) - 3) . "../$tail"; # -3 for "../"
+ }
- return "$spl[0]".substr($text, length($spl[0]), $rest)."..\/".$spl[@spl-1];
+ substr($path, ($remaining/2), (length($path)-$remaining), '..');
+ return "$head/" . $path . "/$tail";
}
sub lock {
my ($fh) = @_;
- flock($fh, LOCK_EX) or die "Cannot lock config - $!\n";
-
- seek($fh, 0, SEEK_END) or die "Cannot seek - $!\n";
+ flock($fh, LOCK_EX) || die "Can't lock config - $!\n";
}
sub unlock {
my ($fh) = @_;
- flock($fh, LOCK_UN) or die "Cannot unlock config- $!\n";
+ flock($fh, LOCK_UN) || die "Can't unlock config- $!\n";
}
-sub check_config {
- my ($source, $name, $cfg) = @_;
+sub get_status {
+ my ($source, $name, $status) = @_;
- if ($source->{vmid} && $cfg->{$source->{vmid}}->{$name}->{locked}){
- return "active" if $cfg->{$source->{vmid}}->{$name}->{locked} eq 'yes';
- return "exist" if $cfg->{$source->{vmid}}->{$name}->{locked} eq 'no';
- } elsif ($cfg->{$source->{abs_path}}->{$name}->{locked}) {
- return "active" if $cfg->{$source->{abs_path}}->{$name}->{locked} eq 'yes';
- return "exist" if $cfg->{$source->{abs_path}}->{$name}->{locked} eq 'no';
+ if ($status->{$source->{all}}->{$name}->{status}) {
+ return $status;
}
return undef;
}
+sub check_pool_exists {
+ my ($target, $user) = @_;
-sub check_pool_exsits {
- my ($pool, $ip) = @_;
+ my $cmd = [];
- my $cmd = '';
- $cmd = "ssh root\@$ip " if $ip;
- $cmd .= "zfs list $pool -H";
+ if ($target->{ip}) {
+ push @$cmd, 'ssh', "$user\@$target->{ip}", '--';
+ }
+ push @$cmd, 'zfs', 'list', '-H', '--', $target->{all};
eval {
run_cmd($cmd);
};
- if($@){
- return 1;
+ if ($@) {
+ return 0;
}
- return undef;
+ return 1;
}
-sub write_to_config {
- my ($cfg) = @_;
+sub parse_target {
+ my ($text) = @_;
- open(my $fh, ">", "$CONFIG_PATH$CONFIG")
- or die "cannot open >$CONFIG_PATH$CONFIG: $!\n";
+ my $errstr = "$text : is not a valid input! Use [IP:]<VMID> or [IP:]<ZFSPool>[/Path]";
+ my $target = {};
- my $text = decode_config($cfg);
+ if ($text !~ $TARGETRE) {
+ die "$errstr\n";
+ }
+ $target->{all} = $2;
+ $target->{ip} = $1 if $1;
+ my @parts = split('/', $2);
- print($fh $text);
+ $target->{ip} =~ s/^\[(.*)\]$/$1/ if $target->{ip};
- close($fh);
-}
+ my $pool = $target->{pool} = shift(@parts);
+ die "$errstr\n" if !$pool;
-sub read_from_config {
+ if ($pool =~ m/^\d+$/) {
+ $target->{vmid} = $pool;
+ delete $target->{pool};
+ }
- unless(-e "$CONFIG_PATH$CONFIG") {
- return undef;
+ return $target if (@parts == 0);
+ $target->{last_part} = pop(@parts);
+
+ if ($target->{ip}) {
+ pop(@parts);
+ }
+ if (@parts > 0) {
+ $target->{path} = join('/', @parts);
}
- open(my $fh, "<", "$CONFIG_PATH$CONFIG")
- or die "cannot open > $CONFIG_PATH$CONFIG: $!\n";
+ return $target;
+}
- $/ = undef;
+sub read_cron {
- my $text = <$fh>;
+ #This is for the first use to init file;
+ if (!-e $CRONJOBS) {
+ my $new_fh = IO::File->new("> $CRONJOBS");
+ die "Could not create $CRONJOBS: $!\n" if !$new_fh;
+ close($new_fh);
+ return undef;
+ }
- unlock($fh);
+ my $fh = IO::File->new("< $CRONJOBS");
+ die "Could not open file $CRONJOBS: $!\n" if !$fh;
+
+ my @text = <$fh>;
close($fh);
- my $cfg = encode_config($text);
+ return encode_cron(@text);
+}
- return $cfg;
+sub parse_argv {
+ my (@arg) = @_;
+
+ my $param = {
+ dest => undef,
+ source => undef,
+ verbose => undef,
+ limit => undef,
+ maxsnap => undef,
+ name => undef,
+ skip => undef,
+ method => undef,
+ source_user => undef,
+ dest_user => undef,
+ };
+
+ my ($ret) = GetOptionsFromArray(
+ \@arg,
+ 'dest=s' => \$param->{dest},
+ 'source=s' => \$param->{source},
+ 'verbose' => \$param->{verbose},
+ 'limit=i' => \$param->{limit},
+ 'maxsnap=i' => \$param->{maxsnap},
+ 'name=s' => \$param->{name},
+ 'skip' => \$param->{skip},
+ 'method=s' => \$param->{method},
+ 'source-user=s' => \$param->{source_user},
+ 'dest-user=s' => \$param->{dest_user}
+ );
+
+ die "can't parse options\n" if $ret == 0;
+
+ $param->{name} //= "default";
+ $param->{maxsnap} //= 1;
+ $param->{method} //= "ssh";
+ $param->{source_user} //= "root";
+ $param->{dest_user} //= "root";
+
+ return $param;
}
-sub decode_config {
- my ($cfg) = @_;
- my $raw = '';
- foreach my $source (sort keys%{$cfg}){
- foreach my $sync_name (sort keys%{$cfg->{$source}}){
- $raw .= "$source: $sync_name\n";
- foreach my $parameter (sort keys%{$cfg->{$source}->{$sync_name}}){
- $raw .= "\t$parameter: $cfg->{$source}->{$sync_name}->{$parameter}\n";
- }
- }
+sub add_state_to_job {
+ my ($job) = @_;
+
+ my $states = read_state();
+ my $state = $states->{$job->{source}}->{$job->{name}};
+
+ $job->{state} = $state->{state};
+ $job->{lsync} = $state->{lsync};
+ $job->{vm_type} = $state->{vm_type};
+
+ for (my $i = 0; $state->{"snap$i"}; $i++) {
+ $job->{"snap$i"} = $state->{"snap$i"};
}
- return $raw;
+
+ return $job;
}
-sub encode_config {
- my ($raw) = @_;
+sub encode_cron {
+ my (@text) = @_;
+
my $cfg = {};
- my $source;
- my $check = 0;
- my $sync_name;
- while ($raw && $raw =~ s/^(.*?)(\n|$)//) {
- my $line = $1;
+ while (my $line = shift(@text)) {
- next if $line =~ m/^\#/;
- next if $line =~ m/^\s*$/;
-
- if ($line =~ m/^(\t| )(\w+): (.+)/){
- my $par = $2;
- my $value = $3;
-
- if ($par eq 'source_pool') {
- $cfg->{$source}->{$sync_name}->{$par} = $value;
- die "error in Config: SourcePool value doubled\n" if ($check & 1);
- $check += 1;
- } elsif ($par eq 'source_ip') {
- $cfg->{$source}->{$sync_name}->{$par} = $value;
- die "error in Config: SourceIP value doubled\n" if ($check & 2);
- $check += 2;
- } elsif ($par eq 'locked') {
- $cfg->{$source}->{$sync_name}->{$par} = $value;
- die "error in Config: Locked value doubled\n" if ($check & 4);
- $check += 4;
- } elsif ($par eq 'method') {
- $cfg -> {$source}->{$sync_name}->{$par} = $value;
- die "error in Config: Method value doubled\n" if ($check & 8);
- $check += 8;
- } elsif ($par eq 'interval') {
- $cfg -> {$source}->{$sync_name}->{$par} = $value;
- die "error in Config: Iterval value doubled\n" if ($check & 16);
- $check += 16;
- } elsif ($par eq 'limit') {
- $cfg -> {$source}->{$sync_name}->{$par} = $value;
- die "error in Config: Limit value doubled\n" if ($check & 32);
- $check += 32;
- } elsif ($par eq 'dest_pool') {
- $cfg -> {$source}->{$sync_name}->{$par} = $value;
- die "error in Config: DestPool value doubled\n" if ($check & 64);
- $check += 64;
- } elsif ($par eq 'dest_ip') {
- $cfg -> {$source}->{$sync_name}->{$par} = $value;
- die "error in Config: DestIp value doubled\n" if ($check & 128);
- $check += 128;
- } elsif ($par eq 'dest_path') {
- $cfg -> {$source}->{$sync_name}->{$par} = $value;
- die "error in Config: DestPath value doubled\n" if ($check & 256);
- $check += 256;
- } elsif ($par eq 'source_path') {
- $cfg -> {$source}->{$sync_name}->{$par} = $value;
- die "error in Config: SourcePath value doubled\n" if ($check & 512);
- $check += 512;
- } elsif ($par eq 'vmid') {
- $cfg -> {$source}->{$sync_name}->{$par} = $value;
- die "error in Config: Vmid value doubled\n" if ($check & 1024);
- $check += 1024;
- } elsif ($par =~ 'lsync') {
- $cfg->{$source}->{$sync_name}->{$par} = $value;
- die "error in Config: lsync value doubled\n" if ($check & 2048);
- $check += 2048;
- } elsif ($par =~ 'maxsnap') {
- $cfg->{$source}->{$sync_name}->{$par} = $value;
- die "error in Config: maxsnap value doubled\n" if ($check & 4096);
- $check += 4096;
- } else {
- die "error in Config\n";
- }
- } elsif ($line =~ m/^((\d+.\d+.\d+.\d+):)?([\w\-\_\/]+): (.+){0,1}/){
- $source = $3;
- $sync_name = $4 ? $4 : 'default' ;
- $cfg->{$source}->{$sync_name} = undef;
- $cfg->{$source}->{$sync_name}->{source_ip} = $2 if $2;
- $check = 0;
+ my @arg = split('\s', $line);
+ my $param = parse_argv(@arg);
+
+ if ($param->{source} && $param->{dest}) {
+ my $source = delete $param->{source};
+ my $name = delete $param->{name};
+
+ $cfg->{$source}->{$name} = $param;
}
}
+
return $cfg;
}
-sub parse_target {
- my ($text) = @_;
+sub param_to_job {
+ my ($param) = @_;
+
+ my $job = {};
- if ($text =~ m/^((\d+.\d+.\d+.\d+):)?((\w+)\/?)([\w\/\-\_]*)?$/) {
+ my $source = parse_target($param->{source});
+ my $dest = parse_target($param->{dest}) if $param->{dest};
+
+ $job->{name} = !$param->{name} ? "default" : $param->{name};
+ $job->{dest} = $param->{dest} if $param->{dest};
+ $job->{method} = "local" if !$dest->{ip} && !$source->{ip};
+ $job->{method} = "ssh" if !$job->{method};
+ $job->{limit} = $param->{limit};
+ $job->{maxsnap} = $param->{maxsnap} if $param->{maxsnap};
+ $job->{source} = $param->{source};
+ $job->{source_user} = $param->{source_user};
+ $job->{dest_user} = $param->{dest_user};
+
+ return $job;
+}
- die "Input not valid\n" if !$3;
- my $tmp = $3;
- my $target = {};
+sub read_state {
- if ($2) {
- $target->{ip} = $2 ;
+ if (!-e $STATE) {
+ make_path $CONFIG_PATH;
+ my $new_fh = IO::File->new("> $STATE");
+ die "Could not create $STATE: $!\n" if !$new_fh;
+ print $new_fh "{}";
+ close($new_fh);
+ return undef;
+ }
+
+ my $fh = IO::File->new("< $STATE");
+ die "Could not open file $STATE: $!\n" if !$fh;
+
+ my $text = <$fh>;
+ my $states = decode_json($text);
+
+ close($fh);
+
+ return $states;
+}
+
+sub update_state {
+ my ($job) = @_;
+ my $text;
+ my $in_fh;
+
+ eval {
+
+ $in_fh = IO::File->new("< $STATE");
+ die "Could not open file $STATE: $!\n" if !$in_fh;
+ lock($in_fh);
+ $text = <$in_fh>;
+ };
+
+ my $out_fh = IO::File->new("> $STATE.new");
+ die "Could not open file ${STATE}.new: $!\n" if !$out_fh;
+
+ my $states = {};
+ my $state = {};
+ if ($text){
+ $states = decode_json($text);
+ $state = $states->{$job->{source}}->{$job->{name}};
+ }
+
+ if ($job->{state} ne "del") {
+ $state->{state} = $job->{state};
+ $state->{lsync} = $job->{lsync};
+ $state->{vm_type} = $job->{vm_type};
+
+ for (my $i = 0; $job->{"snap$i"} ; $i++) {
+ $state->{"snap$i"} = $job->{"snap$i"};
}
+ $states->{$job->{source}}->{$job->{name}} = $state;
+ } else {
+
+ delete $states->{$job->{source}}->{$job->{name}};
+ delete $states->{$job->{source}} if !keys %{$states->{$job->{source}}};
+ }
+
+ $text = encode_json($states);
+ print $out_fh $text;
+
+ close($out_fh);
+ move("$STATE.new", $STATE);
+ eval {
+ close($in_fh);
+ };
+
+ return $states;
+}
- if ($tmp =~ m/^(\d\d\d+)$/){
- $target->{vmid} = $tmp;
+sub update_cron {
+ my ($job) = @_;
+
+ my $updated;
+ my $has_header;
+ my $line_no = 0;
+ my $text = "";
+ my $header = "SHELL=/bin/sh\n";
+ $header .= "PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin\n\n";
+
+ my $fh = IO::File->new("< $CRONJOBS");
+ die "Could not open file $CRONJOBS: $!\n" if !$fh;
+ lock($fh);
+
+ my @test = <$fh>;
+
+ while (my $line = shift(@test)) {
+ chomp($line);
+ if ($line =~ m/source $job->{source} .*name $job->{name} /) {
+ $updated = 1;
+ next if $job->{state} eq "del";
+ $text .= format_job($job, $line);
} else {
- $target->{pool} = $4;
- my $abs_path = $4;
- if ($5) {
- $target->{path} = "\/$5";
- $abs_path .= "\/$5";
+ if (($line_no < 3) && ($line =~ /^(PATH|SHELL)/ )) {
+ $has_header = 1;
}
- $target->{abs_path} = $abs_path;
+ $text .= "$line\n";
}
+ $line_no++;
+ }
+
+ if (!$has_header) {
+ $text = "$header$text";
+ }
+
+ if (!$updated) {
+ $text .= format_job($job);
+ }
+ my $new_fh = IO::File->new("> ${CRONJOBS}.new");
+ die "Could not open file ${CRONJOBS}.new: $!\n" if !$new_fh;
+
+ die "can't write to $CRONJOBS.new\n" if !print($new_fh $text);
+ close ($new_fh);
- return $target;
+ die "can't move $CRONJOBS.new: $!\n" if !move("${CRONJOBS}.new", "$CRONJOBS");
+ close ($fh);
+}
+
+sub format_job {
+ my ($job, $line) = @_;
+ my $text = "";
+
+ if ($job->{state} eq "stopped") {
+ $text = "#";
+ }
+ if ($line) {
+ $line =~ /^#*\s*((?:\S+\s+){4}\S+)\s+root/;
+ $text .= $1;
+ } else {
+ $text .= "*/$INTERVAL * * * *";
}
- die "Input not valid\n";
+ $text .= " root";
+ $text .= " $PROGNAME sync --source $job->{source} --dest $job->{dest}";
+ $text .= " --name $job->{name} --maxsnap $job->{maxsnap}";
+ $text .= " --limit $job->{limit}" if $job->{limit};
+ $text .= " --method $job->{method}";
+ $text .= " --verbose" if $job->{verbose};
+ $text .= " --source-user $job->{source_user}";
+ $text .= " --dest-user $job->{dest_user}";
+ $text .= "\n";
+
+ return $text;
}
sub list {
- my $cfg = read_from_config("$CONFIG_PATH$CONFIG");
-
- my $list = sprintf("%-25s%-15s%-7s%-20s%-10s%-5s\n" , "SOURCE", "NAME", "ACTIVE", "LAST SYNC", "INTERVAL", "TYPE");
-
- foreach my $source (sort keys%{$cfg}){
- foreach my $sync_name (sort keys%{$cfg->{$source}}){
- my $source_name = $source;
- $source_name = $cfg->{$source}->{$sync_name}->{source_ip}.":".$source if $cfg->{$source}->{$sync_name}->{source_ip};
- $list .= sprintf("%-25s%-15s", cut_to_width($source_name,25), cut_to_width($sync_name,15));
- $list .= sprintf("%-7s",$cfg->{$source}->{$sync_name}->{locked});
- $list .= sprintf("%-20s",$cfg->{$source}->{$sync_name}->{lsync});
- $list .= sprintf("%-10s",$cfg->{$source}->{$sync_name}->{interval});
- $list .= sprintf("%-5s\n",$cfg->{$source}->{$sync_name}->{method});
+ my $cfg = read_cron();
+
+ my $list = sprintf("%-25s%-25s%-10s%-20s%-6s%-5s\n" , "SOURCE", "NAME", "STATE", "LAST SYNC", "TYPE", "CON");
+
+ my $states = read_state();
+ foreach my $source (sort keys%{$cfg}) {
+ foreach my $name (sort keys%{$cfg->{$source}}) {
+ $list .= sprintf("%-25s", cut_target_width($source, 25));
+ $list .= sprintf("%-25s", cut_target_width($name, 25));
+ $list .= sprintf("%-10s", $states->{$source}->{$name}->{state});
+ $list .= sprintf("%-20s", $states->{$source}->{$name}->{lsync});
+ $list .= sprintf("%-6s", defined($states->{$source}->{$name}->{vm_type}) ? $states->{$source}->{$name}->{vm_type} : "undef");
+ $list .= sprintf("%-5s\n", $cfg->{$source}->{$name}->{method});
}
}
}
sub vm_exists {
- my ($target) = @_;
+ my ($target, $user) = @_;
- my $cmd = "";
- $cmd = "ssh root\@$target->{ip} " if ($target->{ip});
- $cmd .= "qm status $target->{vmid}";
+ my @cmd = ('ssh', "$user\@$target->{ip}", '--') if $target->{ip};
- my $res = run_cmd($cmd);
+ my $res = undef;
+
+ return undef if !defined($target->{vmid});
+
+ eval { $res = run_cmd([@cmd, 'ls', "$QEMU_CONF/$target->{vmid}.conf"]) };
+
+ return "qemu" if $res;
+
+ eval { $res = run_cmd([@cmd, 'ls', "$LXC_CONF/$target->{vmid}.conf"]) };
+
+ return "lxc" if $res;
- return 1 if ($res =~ m/^status.*$/);
return undef;
}
sub init {
my ($param) = @_;
- my $cfg = read_from_config;
+ my $cfg = read_cron();
- my $vm = {};
+ my $job = param_to_job($param);
- my $name = $param->{name} ? $param->{name} : "default";
- my $interval = $param->{interval} ? $param->{interval} : 15;
+ $job->{state} = "ok";
+ $job->{lsync} = 0;
my $source = parse_target($param->{source});
my $dest = parse_target($param->{dest});
- $vm->{$name}->{dest_pool} = $dest->{pool};
- $vm->{$name}->{dest_ip} = $dest->{ip} if $dest->{ip};
- $vm->{$name}->{dest_path} = $dest->{path} if $dest->{path};
-
- $param->{method} = "local" if !$dest->{ip} && !$source->{ip};
- $vm->{$name}->{locked} = "no";
- $vm->{$name}->{interval} = $interval;
- $vm->{$name}->{method} = $param->{method} ? $param->{method} : "ssh";
- $vm->{$name}->{limit} = $param->{limit} if $param->{limit};
- $vm->{$name}->{maxsnap} = $param->{maxsnap} if $param->{maxsnap};
-
- if ( my $ip = $vm->{$name}->{dest_ip} ) {
- run_cmd("ssh-copy-id -i /root/.ssh/id_rsa.pub root\@$ip");
+ if (my $ip = $dest->{ip}) {
+ run_cmd(['ssh-copy-id', '-i', '/root/.ssh/id_rsa.pub', "$param->{dest_user}\@$ip"]);
}
- if ( my $ip = $source->{ip} ) {
- run_cmd("ssh-copy-id -i /root/.ssh/id_rsa.pub root\@$ip");
+ if (my $ip = $source->{ip}) {
+ run_cmd(['ssh-copy-id', '-i', '/root/.ssh/id_rsa.pub', "$param->{source_user}\@$ip"]);
}
- die "Pool $dest->{abs_path} does not exists\n" if check_pool_exsits($dest->{abs_path}, $dest->{ip});
+ die "Pool $dest->{all} does not exists\n" if !check_pool_exists($dest, $param->{dest_user});
- my $check = check_pool_exsits($source->{abs_path}, $source->{ip}) if !$source->{vmid} && $source->{abs_path};
+ if (!defined($source->{vmid})) {
+ die "Pool $source->{all} does not exists\n" if !check_pool_exists($source, $param->{source_user});
+ }
- die "Pool $source->{abs_path} does not exists\n" if undef($check);
+ my $vm_type = vm_exists($source, $param->{source_user});
+ $job->{vm_type} = $vm_type;
+ $source->{vm_type} = $vm_type;
- my $add_job = sub {
- my ($vm, $name) = @_;
- my $source = "";
+ die "VM $source->{vmid} doesn't exist\n" if $source->{vmid} && !$vm_type;
- if ($vm->{$name}->{vmid}) {
- $source = "$vm->{$name}->{source_ip}:" if $vm->{$name}->{source_ip};
- $source .= $vm->{$name}->{vmid};
- } else {
- $source = $vm->{$name}->{source_pool};
- $source .= $vm->{$name}->{source_path} if $vm->{$name}->{source_path};
- }
- die "Config already exists\n" if $cfg->{$source}->{$name};
+ die "Config already exists\n" if $cfg->{$job->{source}}->{$job->{name}};
- cron_add($vm);
+ #check if vm has zfs disks if not die;
+ get_disks($source, $param->{source_user}) if $source->{vmid};
- $cfg->{$source}->{$name} = $vm->{$name};
+ update_cron($job);
+ update_state($job);
- write_to_config($cfg);
+ eval {
+ sync($param) if !$param->{skip};
};
-
- if ($source->{vmid}) {
- die "VM $source->{vmid} doesn't exist\n" if !vm_exists($source);
- my $disks = get_disks($source);
- $vm->{$name}->{vmid} = $source->{vmid};
- $vm->{$name}->{lsync} = 0;
- $vm->{$name}->{source_ip} = $source->{ip} if $source->{ip};
-
- &$add_job($vm, $name);
-
- } else {
- $vm->{$name}->{source_pool} = $source->{pool};
- $vm->{$name}->{source_ip} = $source->{ip} if $source->{ip};
- $vm->{$name}->{source_path} = $source->{path} if $source->{path};
- $vm->{$name}->{lsync} = 0;
-
- &$add_job($vm, $name);
- }
-
- eval {sync($param) if !$param->{skip};};
if(my $err = $@) {
- destroy($param);
+ destroy_job($param);
print $err;
}
}
-sub destroy {
+sub get_job {
my ($param) = @_;
- my $cfg = read_from_config("$CONFIG_PATH$CONFIG");
- my $name = $param->{name} ? $param->{name} : "default";
-
- my $source = parse_target($param->{source});
-
- my $delete_cron = sub {
- my ($path, $name, $cfg) = @_;
-
- die "Source does not exist!\n" unless $cfg->{$path} ;
-
- die "Sync Name does not exist!\n" unless $cfg->{$path}->{$name};
+ my $cfg = read_cron();
- delete $cfg->{$path}->{$name};
-
- delete $cfg->{$path} if keys%{$cfg->{$path}} == 0;
-
- write_to_config($cfg);
-
- cron_del($path, $name);
- };
-
-
- if ($source->{vmid}) {
- my $path = $source->{vmid};
+ if (!$cfg->{$param->{source}}->{$param->{name}}) {
+ die "Job with source $param->{source} and name $param->{name} does not exist\n" ;
+ }
+ my $job = $cfg->{$param->{source}}->{$param->{name}};
+ $job->{name} = $param->{name};
+ $job->{source} = $param->{source};
+ $job = add_state_to_job($job);
- &$delete_cron($path, $name, $cfg)
+ return $job;
+}
- } else {
+sub destroy_job {
+ my ($param) = @_;
- my $path = $source->{pool};
- $path .= $source->{path} if $source->{path};
+ my $job = get_job($param);
+ $job->{state} = "del";
- &$delete_cron($path, $name, $cfg);
- }
+ update_cron($job);
+ update_state($job);
}
sub sync {
my ($param) = @_;
- my $cfg = read_from_config("$CONFIG_PATH$CONFIG");
+ my $lock_fh = IO::File->new("> $LOCKFILE");
+ die "Can't open Lock File: $LOCKFILE $!\n" if !$lock_fh;
+ lock($lock_fh);
- my $name = $param->{name} ? $param->{name} : "default";
- my $max_snap = $param->{maxsnap} ? $param->{maxsnap} : 1;
- my $method = $param->{method} ? $param->{method} : "ssh";
+ my $date = get_date();
+ my $job;
+ eval {
+ $job = get_job($param);
+ };
+
+ if ($job && $job->{state} eq "syncing") {
+ die "Job --source $param->{source} --name $param->{name} is syncing at the moment";
+ }
my $dest = parse_target($param->{dest});
my $source = parse_target($param->{source});
my $sync_path = sub {
- my ($source, $name, $cfg, $max_snap, $dest, $method) = @_;
+ my ($source, $dest, $job, $param, $date) = @_;
- ($source->{old_snap},$source->{last_snap}) = snapshot_get($source, $dest, $max_snap, $name);
+ ($source->{old_snap}, $source->{last_snap}) = snapshot_get($source, $dest, $param->{maxsnap}, $param->{name}, $param->{source_user});
- my $job_status = check_config($source, $name, $cfg) if $cfg;
- die "VM syncing at the moment!\n" if ($job_status && $job_status eq "active");
+ snapshot_add($source, $dest, $param->{name}, $date, $param->{source_user}, $param->{dest_user});
- if ($job_status && $job_status eq "exist") {
- my $conf_name = $source->{abs_path};
- $conf_name = $source->{vmid} if $source->{vmid};
- $cfg->{$conf_name}->{$name}->{locked} = "yes";
- write_to_config($cfg);
- }
+ send_image($source, $dest, $param);
- my $date = snapshot_add($source, $dest, $name);
+ snapshot_destroy($source, $dest, $param->{method}, $source->{old_snap}, $param->{source_user}, $param->{dest_user}) if ($source->{destroy} && $source->{old_snap});
- send_image($source, $dest, $method, $param->{verbose}, $param->{limit});
+ };
- snapshot_destroy($source, $dest, $method, $source->{old_snap}) if ($source->{destroy} && $source->{old_snap});
+ my $vm_type = vm_exists($source, $param->{source_user});
+ $source->{vm_type} = $vm_type;
+ if ($job) {
+ $job->{state} = "syncing";
+ $job->{vm_type} = $vm_type if !$job->{vm_type};
+ update_state($job);
+ }
- if ($job_status && $job_status eq "exist") {
- my $conf_name = $source->{abs_path};
- $conf_name = $source->{vmid} if $source->{vmid};
- $cfg->{$conf_name}->{$name}->{locked} = "no";
- $cfg->{$conf_name}->{$name}->{lsync} = $date;
- write_to_config($cfg);
+ eval{
+ if ($source->{vmid}) {
+ die "VM $source->{vmid} doesn't exist\n" if !$vm_type;
+ die "source-user has to be root for syncing VMs\n" if ($param->{source_user} ne "root");
+ my $disks = get_disks($source, $param->{source_user});
+
+ foreach my $disk (sort keys %{$disks}) {
+ $source->{all} = $disks->{$disk}->{all};
+ $source->{pool} = $disks->{$disk}->{pool};
+ $source->{path} = $disks->{$disk}->{path} if $disks->{$disk}->{path};
+ $source->{last_part} = $disks->{$disk}->{last_part};
+ &$sync_path($source, $dest, $job, $param, $date);
+ }
+ if ($param->{method} eq "ssh" && ($source->{ip} || $dest->{ip})) {
+ send_config($source, $dest,'ssh', $param->{source_user}, $param->{dest_user});
+ } else {
+ send_config($source, $dest,'local', $param->{source_user}, $param->{dest_user});
+ }
+ } else {
+ &$sync_path($source, $dest, $job, $param, $date);
}
};
-
- $param->{method} = "ssh" if !$param->{method};
-
- if ($source->{vmid}) {
- die "VM $source->{vmid} doesn't exist\n" if !vm_exists($source);
- my $disks = get_disks($source);
-
- foreach my $disk (keys %{$disks}) {
- $source->{abs_path} = $disks->{$disk}->{pool};
- $source->{abs_path} .= "\/$disks->{$disk}->{path}" if $disks->{$disk}->{path};
-
- $source->{pool} = $disks->{$disk}->{pool};
- $source->{path} = "\/$disks->{$disk}->{path}";
-
- &$sync_path($source, $name, $cfg, $max_snap, $dest, $method);
+ if(my $err = $@) {
+ if ($job) {
+ $job->{state} = "error";
+ update_state($job);
+ unlock($lock_fh);
+ close($lock_fh);
+ print "Job --source $param->{source} --name $param->{name} got an ERROR!!!\nERROR Message:\n";
}
- } else {
- &$sync_path($source, $name, $cfg, $max_snap, $dest, $method);
+ die "$err\n";
}
+
+ if ($job) {
+ $job->{state} = "ok";
+ $job->{lsync} = $date;
+ update_state($job);
+ }
+
+ unlock($lock_fh);
+ close($lock_fh);
}
sub snapshot_get{
- my ($source, $dest, $max_snap, $name) = @_;
-
- my $cmd = "zfs list -r -t snapshot -Ho name, -S creation ";
+ my ($source, $dest, $max_snap, $name, $source_user) = @_;
- $cmd .= $source->{abs_path};
- $cmd = "ssh root\@$source->{ip} ".$cmd if $source->{ip};
+ my $cmd = [];
+ push @$cmd, 'ssh', "$source_user\@$source->{ip}", '--', if $source->{ip};
+ push @$cmd, 'zfs', 'list', '-r', '-t', 'snapshot', '-Ho', 'name', '-S', 'creation';
+ push @$cmd, $source->{all};
my $raw = run_cmd($cmd);
- my $index = 1;
+ my $index = 0;
my $line = "";
my $last_snap = undef;
+ my $old_snap;
while ($raw && $raw =~ s/^(.*?)(\n|$)//) {
$line = $1;
- $last_snap = $line if $index == 1;
- if ($index == $max_snap) {
- $source->{destroy} = 1;
- last;
- };
- $index++;
+ if ($line =~ m/(rep_\Q${name}\E_\d{4}-\d{2}-\d{2}_\d{2}:\d{2}:\d{2})$/) {
+
+ $last_snap = $1 if (!$last_snap);
+ $old_snap = $1;
+ $index++;
+ if ($index == $max_snap) {
+ $source->{destroy} = 1;
+ last;
+ };
+ }
}
- $line =~ m/^(.+)\@(rep_$name\_.+)(\n|$)/;
- return ($2, $last_snap) if $2;
+ return ($old_snap, $last_snap) if $last_snap;
return undef;
}
sub snapshot_add {
- my ($source, $dest, $name) = @_;
-
- my $date = get_date();
+ my ($source, $dest, $name, $date, $source_user, $dest_user) = @_;
my $snap_name = "rep_$name\_".$date;
$source->{new_snap} = $snap_name;
- my $path = $source->{abs_path}."\@".$snap_name;
-
- my $cmd = "zfs snapshot $path";
- $cmd = "ssh root\@$source->{ip} ".$cmd if $source->{ip};
+ my $path = "$source->{all}\@$snap_name";
+ my $cmd = [];
+ push @$cmd, 'ssh', "$source_user\@$source->{ip}", '--', if $source->{ip};
+ push @$cmd, 'zfs', 'snapshot', $path;
eval{
run_cmd($cmd);
};
- if (my $err = $@){
- snapshot_destroy($source, $dest, 'ssh', $snap_name);
+ if (my $err = $@) {
+ snapshot_destroy($source, $dest, 'ssh', $snap_name, $source_user, $dest_user);
die "$err\n";
}
- return $date;
-}
-
-sub cron_add {
- my ($vm) = @_;
-
- open(my $fh, '>>', "$CRONJOBS")
- or die "Could not open file: $!\n";
-
- foreach my $name (keys%{$vm}){
- my $text = "*/$vm->{$name}->{interval} * * * * root ";
- $text .= "$PATH$PROGNAME sync";
- $text .= " -source ";
- if ($vm->{$name}->{vmid}) {
- $text .= "$vm->{$name}->{source_ip}:" if $vm->{$name}->{source_ip};
- $text .= "$vm->{$name}->{vmid} ";
- } else {
- $text .= "$vm->{$name}->{source_ip}:" if $vm->{$name}->{source_ip};
- $text .= "$vm->{$name}->{source_pool}";
- $text .= "$vm->{$name}->{source_path}" if $vm->{$name}->{source_path};
- }
- $text .= " -dest ";
- $text .= "$vm->{$name}->{dest_ip}:" if $vm->{$name}->{dest_ip};
- $text .= "$vm->{$name}->{dest_pool}";
- $text .= "$vm->{$name}->{dest_path}" if $vm->{$name}->{dest_path};
- $text .= " -name $name ";
- $text .= " -limit $vm->{$name}->{limit}" if $vm->{$name}->{limit};
- $text .= " -maxsnap $vm->{$name}->{maxsnap}" if $vm->{$name}->{maxsnap};
- $text .= "\n";
- print($fh $text);
- }
- close($fh);
}
-sub cron_del {
- my ($source, $name) = @_;
+sub write_cron {
+ my ($cfg) = @_;
- open(my $fh, '<', "$CRONJOBS")
- or die "Could not open file: $!\n";
+ my $text = "SHELL=/bin/sh\n";
+ $text .= "PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin\n";
- $/ = undef;
+ my $fh = IO::File->new("> $CRONJOBS");
+ die "Could not open file: $!\n" if !$fh;
- my $text = <$fh>;
- my $buffer = "";
- close($fh);
- while ($text && $text =~ s/^(.*?)(\n|$)//) {
- my $line = $1.$2;
- if ($line !~ m/.*$PROGNAME.*$source.*$name.*/){
- $buffer .= $line;
+ foreach my $source (sort keys%{$cfg}) {
+ foreach my $sync_name (sort keys%{$cfg->{$source}}) {
+ next if $cfg->{$source}->{$sync_name}->{status} ne 'ok';
+ $text .= "$PROG_PATH sync";
+ $text .= " -source ";
+ if ($cfg->{$source}->{$sync_name}->{vmid}) {
+ $text .= "$cfg->{$source}->{$sync_name}->{source_ip}:" if $cfg->{$source}->{$sync_name}->{source_ip};
+ $text .= "$cfg->{$source}->{$sync_name}->{vmid} ";
+ } else {
+ $text .= "$cfg->{$source}->{$sync_name}->{source_ip}:" if $cfg->{$source}->{$sync_name}->{source_ip};
+ $text .= "$cfg->{$source}->{$sync_name}->{source_pool}";
+ $text .= "$cfg->{$source}->{$sync_name}->{source_path}" if $cfg->{$source}->{$sync_name}->{source_path};
+ }
+ $text .= " -dest ";
+ $text .= "$cfg->{$source}->{$sync_name}->{dest_ip}:" if $cfg->{$source}->{$sync_name}->{dest_ip};
+ $text .= "$cfg->{$source}->{$sync_name}->{dest_pool}";
+ $text .= "$cfg->{$source}->{$sync_name}->{dest_path}" if $cfg->{$source}->{$sync_name}->{dest_path};
+ $text .= " -name $sync_name ";
+ $text .= " -limit $cfg->{$source}->{$sync_name}->{limit}" if $cfg->{$source}->{$sync_name}->{limit};
+ $text .= " -maxsnap $cfg->{$source}->{$sync_name}->{maxsnap}" if $cfg->{$source}->{$sync_name}->{maxsnap};
+ $text .= "\n";
}
}
- open($fh, '>', "$CRONJOBS")
- or die "Could not open file: $!\n";
- print($fh $buffer);
+ die "Can't write to cron\n" if (!print($fh $text));
close($fh);
}
sub get_disks {
- my ($target) = @_;
+ my ($target, $user) = @_;
+
+ my $cmd = [];
+ push @$cmd, 'ssh', "$user\@$target->{ip}", '--', if $target->{ip};
- my $cmd = "";
- $cmd = "ssh root\@$target->{ip} " if $target->{ip};
- $cmd .= "qm config $target->{vmid}";
+ if ($target->{vm_type} eq 'qemu') {
+ push @$cmd, 'qm', 'config', $target->{vmid};
+ } elsif ($target->{vm_type} eq 'lxc') {
+ push @$cmd, 'pct', 'config', $target->{vmid};
+ } else {
+ die "VM Type unknown\n";
+ }
my $res = run_cmd($cmd);
- my $disks = parse_disks($res, $target->{ip});
+ my $disks = parse_disks($res, $target->{ip}, $target->{vm_type}, $user);
return $disks;
}
my ($cmd) = @_;
print "Start CMD\n" if $DEBUG;
print Dumper $cmd if $DEBUG;
+ if (ref($cmd) eq 'ARRAY') {
+ $cmd = join(' ', map { ref($_) ? $$_ : shell_quote($_) } @$cmd);
+ }
my $output = `$cmd 2>&1`;
- die $output if 0 != $?;
+ die "COMMAND:\n\t$cmd\nGET ERROR:\n\t$output" if 0 != $?;
chomp($output);
print Dumper $output if $DEBUG;
}
sub parse_disks {
- my ($text, $ip) = @_;
+ my ($text, $ip, $vm_type, $user) = @_;
my $disks;
my $num = 0;
- my $cmd = "";
- $cmd .= "ssh root\@$ip " if $ip;
- $cmd .= "pvesm zfsscan";
- my $zfs_pools = run_cmd($cmd);
while ($text && $text =~ s/^(.*?)(\n|$)//) {
my $line = $1;
+
+ next if $line =~ /media=cdrom/;
+ next if $line !~ m/^(?:((?:virtio|ide|scsi|sata|mp)\d+)|rootfs): /;
+
+ #QEMU if backup is not set include in sync
+ next if $vm_type eq 'qemu' && ($line =~ m/backup=(?i:0|no|off|false)/);
+
+ #LXC if backup is not set do no in sync
+ next if $vm_type eq 'lxc' && ($line =~ m/^mp\d:/) && ($line !~ m/backup=(?i:1|yes|on|true)/);
+
my $disk = undef;
my $stor = undef;
- if($line =~ m/^(virtio\d+: )(.+:)([A-Za-z0-9\-]+),(.*)$/) {
- $disk = $3;
- $stor = $2;
- } elsif($line =~ m/^(ide\d+: )(.+:)([A-Za-z0-9\-]+),(.*)$/) {
- $disk = $3;
- $stor = $2;
- } elsif($line =~ m/^(scsi\d+: )(.+:)([A-Za-z0-9\-]+),(.*)$/) {
- $disk = $3;
- $stor = $2;
- } elsif($line =~ m/^(sata\d+: )(.+:)([A-Za-z0-9\-]+),(.*)$/) {
- $disk = $3;
- $stor = $2;
+ if($line =~ m/^(?:(?:(?:virtio|ide|scsi|sata|mp)\d+)|rootfs): (.*)$/) {
+ my @parameter = split(/,/,$1);
+
+ foreach my $opt (@parameter) {
+ if ($opt =~ m/^(?:file=|volume=)?([^:]+:)([A-Za-z0-9\-]+)$/){
+ $disk = $2;
+ $stor = $1;
+ last;
+ }
+ }
+ }
+ if (!defined($disk) || !defined($stor)) {
+ print "Disk: \"$line\" has no valid zfs dataset format and will be skipped\n";
+ next;
}
- if($disk && $disk ne "none" && $disk !~ m/cdrom/ ) {
- my $cmd = "";
- $cmd .= "ssh root\@$ip " if $ip;
- $cmd .= "pvesm path $stor$disk";
- my $path = run_cmd($cmd);
+ my $cmd = [];
+ push @$cmd, 'ssh', "$user\@$ip", '--' if $ip;
+ push @$cmd, 'pvesm', 'path', "$stor$disk";
+ my $path = run_cmd($cmd);
+
+ die "Get no path from pvesm path $stor$disk\n" if !$path;
+
+ if ($vm_type eq 'qemu' && $path =~ m/^\/dev\/zvol\/(\w+.*)(\/$disk)$/) {
+
+ my @array = split('/', $1);
+ $disks->{$num}->{pool} = shift(@array);
+ $disks->{$num}->{all} = $disks->{$num}->{pool};
+ if (0 < @array) {
+ $disks->{$num}->{path} = join('/', @array);
+ $disks->{$num}->{all} .= "\/$disks->{$num}->{path}";
+ }
+ $disks->{$num}->{last_part} = $disk;
+ $disks->{$num}->{all} .= "\/$disk";
- if ($path =~ m/^\/dev\/zvol\/(\w+).*(\/$disk)$/){
+ $num++;
+ } elsif ($vm_type eq 'lxc' && $path =~ m/^\/(\w+.+)(\/(\w+.*))*(\/$disk)$/) {
- $disks->{$num}->{pool} = $1;
- $disks->{$num}->{path} = $disk;
- $num++;
+ $disks->{$num}->{pool} = $1;
+ $disks->{$num}->{all} = $disks->{$num}->{pool};
- } else {
- die "ERROR: in path\n";
+ if ($2) {
+ $disks->{$num}->{path} = $3;
+ $disks->{$num}->{all} .= "\/$disks->{$num}->{path}";
}
+
+ $disks->{$num}->{last_part} = $disk;
+ $disks->{$num}->{all} .= "\/$disk";
+
+ $num++;
+
+ } else {
+ die "ERROR: in path\n";
}
}
- die "disk is not on ZFS Storage\n" if $num == 0;
+
+ die "Vm include no disk on zfs.\n" if !$disks->{0};
return $disks;
}
sub snapshot_destroy {
- my ($source, $dest, $method, $snap) = @_;
+ my ($source, $dest, $method, $snap, $source_user, $dest_user) = @_;
- my $zfscmd = "zfs destroy ";
- my $name = "$source->{path}\@$snap";
+ my @zfscmd = ('zfs', 'destroy');
+ my $snapshot = "$source->{all}\@$snap";
eval {
if($source->{ip} && $method eq 'ssh'){
- run_cmd("ssh root\@$source->{ip} $zfscmd $source->{pool}$name");
+ run_cmd(['ssh', "$source_user\@$source->{ip}", '--', @zfscmd, $snapshot]);
} else {
- run_cmd("$zfscmd $source->{pool}$name");
+ run_cmd([@zfscmd, $snapshot]);
}
};
if (my $erro = $@) {
warn "WARN: $erro";
}
- if ($dest){
- my $ssh = $dest->{ip} ? "ssh root\@$dest->{ip}" : "";
+ if ($dest) {
+ my @ssh = $dest->{ip} ? ('ssh', "$dest_user\@$dest->{ip}", '--') : ();
- my $path = "";
- $path ="$dest->{path}" if $dest->{path};
+ my $path = "$dest->{all}";
+ $path .= "/$source->{last_part}" if $source->{last_part};
- my @dir = split(/\//, $source->{path});
eval {
- run_cmd("$ssh $zfscmd $dest->{pool}$path\/$dir[@dir-1]\@$snap ");
+ run_cmd([@ssh, @zfscmd, "$path\@$snap"]);
};
if (my $erro = $@) {
warn "WARN: $erro";
}
sub snapshot_exist {
- my ($source ,$dest, $method) = @_;
+ my ($source , $dest, $method, $source_user) = @_;
+
+ my $cmd = [];
+ push @$cmd, 'ssh', "$source_user\@$dest->{ip}", '--' if $dest->{ip};
+ push @$cmd, 'zfs', 'list', '-rt', 'snapshot', '-Ho', 'name';
+
+ my $path = $dest->{all};
+ $path .= "/$source->{last_part}" if $source->{last_part};
+ $path .= "\@$source->{old_snap}";
+
+ push @$cmd, $path;
- my $cmd = "";
- $cmd = "ssh root\@$dest->{ip} " if $dest->{ip};
- $cmd .= "zfs list -rt snapshot -Ho name $dest->{pool}";
- $cmd .= "$dest->{path}" if $dest->{path};
- my @dir = split(/\//, $source->{path});
- $cmd .= "\/$dir[@dir-1]\@$source->{old_snap}";
my $text = "";
eval {$text =run_cmd($cmd);};
- if (my $erro = $@) {
+ if (my $erro =$@) {
warn "WARN: $erro";
return undef;
}
while ($text && $text =~ s/^(.*?)(\n|$)//) {
- my $line = $1;
+ my $line =$1;
return 1 if $line =~ m/^.*$source->{old_snap}$/;
}
}
sub send_image {
- my ($source, $dest, $method, $verbose, $limit) = @_;
+ my ($source, $dest, $param) = @_;
- my $cmd = "";
+ my $cmd = [];
- $cmd .= "ssh root\@$source->{ip} " if $source->{ip};
- $cmd .= "zfs send ";
- $cmd .= "-v " if $verbose;
+ push @$cmd, 'ssh', '-o', 'BatchMode=yes', "$param->{source_user}\@$source->{ip}", '--' if $source->{ip};
+ push @$cmd, 'zfs', 'send';
+ push @$cmd, '-v' if $param->{verbose};
- if($source->{last_snap} && snapshot_exist($source ,$dest, $method)) {
- $cmd .= "-i $source->{abs_path}\@$source->{old_snap} $source->{abs_path}\@$source->{new_snap} ";
- } else {
- $cmd .= "$source->{abs_path}\@$source->{new_snap} ";
+ if($source->{last_snap} && snapshot_exist($source , $dest, $param->{method}, $param->{source_user})) {
+ push @$cmd, '-i', "$source->{all}\@$source->{last_snap}";
}
+ push @$cmd, '--', "$source->{all}\@$source->{new_snap}";
- if ($limit){
- my $bwl = $limit*1024;
- $cmd .= "| cstream -t $bwl";
+ if ($param->{limit}){
+ my $bwl = $param->{limit}*1024;
+ push @$cmd, \'|', 'cstream', '-t', $bwl;
}
- $cmd .= "| ";
- $cmd .= "ssh root\@$dest->{ip} " if $dest->{ip};
- $cmd .= "zfs recv $dest->{pool}";
- $cmd .= "$dest->{path}" if $dest->{path};
+ my $target = "$dest->{all}";
+ $target .= "/$source->{last_part}" if $source->{last_part};
+ $target =~ s!/+!/!g;
- my @dir = split(/\//,$source->{path});
- $cmd .= "\/$dir[@dir-1]\@$source->{new_snap}";
+ push @$cmd, \'|';
+ push @$cmd, 'ssh', '-o', 'BatchMode=yes', "$param->{dest_user}\@$dest->{ip}", '--' if $dest->{ip};
+ push @$cmd, 'zfs', 'recv', '-F', '--';
+ push @$cmd, "$target";
eval {
run_cmd($cmd)
};
if (my $erro = $@) {
- snapshot_destroy($source, undef, $method, $source->{new_snap});
+ snapshot_destroy($source, undef, $param->{method}, $source->{new_snap}, $param->{source_user}, $param->{dest_user});
die $erro;
};
-
- if ($source->{vmid}) {
- if ($method eq "ssh") {
- send_config($source, $dest,'ssh');
- }
- }
}
sub send_config{
- my ($source, $dest, $method) = @_;
+ my ($source, $dest, $method, $source_user, $dest_user) = @_;
+
+ my $source_target = $source->{vm_type} eq 'qemu' ? "$QEMU_CONF/$source->{vmid}.conf": "$LXC_CONF/$source->{vmid}.conf";
+ my $dest_target_new ="$source->{vmid}.conf.$source->{vm_type}.$source->{new_snap}";
+
+ my $config_dir = $dest->{last_part} ? "${CONFIG_PATH}/$dest->{last_part}" : $CONFIG_PATH;
+
+ $dest_target_new = $config_dir.'/'.$dest_target_new;
if ($method eq 'ssh'){
if ($dest->{ip} && $source->{ip}) {
- run_cmd("ssh root\@$dest->{ip} mkdir $VMCONFIG -p");
- run_cmd("scp root\@$source->{ip}:$QEMU_CONF$source->{vmid}.conf root\@$dest->{ip}:$VMCONFIG$source->{vmid}.conf.$source->{new_snap}");
+ run_cmd(['ssh', "$dest_user\@$dest->{ip}", '--', 'mkdir', '-p', '--', $config_dir]);
+ run_cmd(['scp', '--', "$source_user\@[$source->{ip}]:$source_target", "$dest_user\@[$dest->{ip}]:$dest_target_new"]);
} elsif ($dest->{ip}) {
- run_cmd("ssh root\@$dest->{ip} mkdir $VMCONFIG -p");
- run_cmd("scp $QEMU_CONF$source->{vmid}.conf root\@$dest->{ip}:$VMCONFIG$source->{vmid}.conf.$source->{new_snap}");
+ run_cmd(['ssh', "$source_user\@$dest->{ip}", '--', 'mkdir', '-p', '--', $config_dir]);
+ run_cmd(['scp', '--', $source_target, "$dest_user\@[$dest->{ip}]:$dest_target_new"]);
} elsif ($source->{ip}) {
- run_cmd("mkdir $VMCONFIG -p");
- run_cmd("scp root\@$source->{ip}:$QEMU_CONF$source->{vmid}.conf $VMCONFIG$source->{vmid}.conf.$source->{new_snap}");
+ run_cmd(['mkdir', '-p', '--', $config_dir]);
+ run_cmd(['scp', '--', "$source_user\@[$source->{ip}]:$source_target", $dest_target_new]);
}
if ($source->{destroy}){
+ my $dest_target_old ="${config_dir}/$source->{vmid}.conf.$source->{vm_type}.$source->{old_snap}";
if($dest->{ip}){
- run_cmd("ssh root\@$dest->{ip} rm -f $VMCONFIG$source->{vmid}.conf.$source->{old_snap}");
+ run_cmd(['ssh', "$dest_user\@$dest->{ip}", '--', 'rm', '-f', '--', $dest_target_old]);
} else {
- run_cmd("rm -f $VMCONFIG$source->{vmid}.conf.$source->{old_snap}");
+ run_cmd(['rm', '-f', '--', $dest_target_old]);
}
}
+ } elsif ($method eq 'local') {
+ run_cmd(['mkdir', '-p', '--', $config_dir]);
+ run_cmd(['cp', $source_target, $dest_target_new]);
}
}
sub get_date {
- my ($sec,$min,$hour,$mday,$mon,$year,$wday,$yday,$isdst) = localtime(time);
- my $datestamp = sprintf ( "%04d-%02d-%02d_%02d:%02d:%02d",$year+1900,$mon+1,$mday,$hour,$min,$sec);
+ my ($sec, $min, $hour, $mday, $mon, $year, $wday, $yday, $isdst) = localtime(time);
+ my $datestamp = sprintf ("%04d-%02d-%02d_%02d:%02d:%02d", $year+1900, $mon+1, $mday, $hour, $min, $sec);
return $datestamp;
}
sub status {
- my $cfg = read_from_config("$CONFIG_PATH$CONFIG");
-
- my $status_list = sprintf("%-25s%-15s%-10s\n","SOURCE","NAME","STATUS");
+ my $cfg = read_cron();
- foreach my $source (sort keys%{$cfg}){
- foreach my $sync_name (sort keys%{$cfg->{$source}}){
- my $status;
+ my $status_list = sprintf("%-25s%-25s%-10s\n", "SOURCE", "NAME", "STATUS");
- my $source_name = $source;
+ my $states = read_state();
- $source_name = $cfg->{$source}->{$sync_name}->{source_ip}.":".$source if $cfg->{$source}->{$sync_name}->{source_ip};
-
- if ($cfg->{$source}->{$sync_name}->{locked} eq 'no'){
- $status = sprintf("%-10s","OK");
- } elsif ($cfg->{$source}->{$sync_name}->{locked} eq 'yes' &&
- $cfg->{$source}->{$sync_name}->{failure}) {
- $status = sprintf("%-10s","sync error");
- } else {
- $status = sprintf("%-10s","syncing");
- }
-
- $status_list .= sprintf("%-25s%-15s", cut_to_width($source_name,25), cut_to_width($sync_name,15));
- $status_list .= "$status\n";
+ foreach my $source (sort keys%{$cfg}) {
+ foreach my $sync_name (sort keys%{$cfg->{$source}}) {
+ $status_list .= sprintf("%-25s", cut_target_width($source, 25));
+ $status_list .= sprintf("%-25s", cut_target_width($sync_name, 25));
+ $status_list .= "$states->{$source}->{$sync_name}->{state}\n";
}
}
return $status_list;
}
+sub enable_job {
+ my ($param) = @_;
-my $command = $ARGV[0];
+ my $job = get_job($param);
+ $job->{state} = "ok";
+ update_state($job);
+ update_cron($job);
+}
-my $commands = {'destroy' => 1,
- 'create' => 1,
- 'sync' => 1,
- 'list' => 1,
- 'status' => 1,
- 'help' => 1};
+sub disable_job {
+ my ($param) = @_;
-if (!$command || !$commands->{$command}) {
- usage();
- die "\n";
+ my $job = get_job($param);
+ $job->{state} = "stopped";
+ update_state($job);
+ update_cron($job);
}
-my $dest = '';
-my $source = '';
-my $verbose = '';
-my $interval = '';
-my $limit = '';
-my $maxsnap = '';
-my $name = '';
-my $skip = '';
-
-my $help_sync = "zfs-zsync sync -dest <string> -source <string> [OPTIONS]\n
-\twill sync one time\n
-\t-dest\tstring\n
-\t\tthe destination target is like [IP:]<Pool>[/Path]\n
-\t-limit\tinteger\n
-\t\tmax sync speed in kBytes/s, default unlimited\n
-\t-maxsnap\tinteger\n
-\t\thow much snapshots will be kept before get erased, default 1/n
-\t-name\tstring\n
-\t\tname of the sync job, if not set it is default.
-\tIt is only necessary if scheduler allready contains this source.\n
-\t-source\tstring\n
-\t\tthe source can be an <VMID> or [IP:]<ZFSPool>[/Path]\n";
-
-my $help_create = "zfs-zsync create -dest <string> -source <string> [OPTIONS]/n
-\tCreate a sync Job\n
-\t-dest\tstringn\n
-\t\tthe destination target is like [IP]:<Pool>[/Path]\n
-\t-interval\tinteger\n
-\t\tthe interval in min in witch the zfs will sync,
-\t\tdefault is 15 min\n
-\t-limit\tinteger\n
-\t\tmax sync speed, default unlimited\n
-\t-maxsnap\tstring\n
-\t\thow much snapshots will be kept before get erased, default 1\n
-\t-name\tstring\n
-\t\tname of the sync job, if not set it is default\n
-\t-skip\tboolean\n
-\t\tif this flag is set it will skip the first sync\n
-\t-source\tstring\n
-\t\tthe source can be an <VMID> or [IP:]<ZFSPool>[/Path]\n";
-
-my $help_destroy = "zfs-zsync destroy -source <string> [OPTIONS]\n
-\tremove a sync Job from the scheduler\n
-\t-name\tstring\n
-\t\tname of the sync job, if not set it is default\n
-\t-source\tstring\n
-\t\tthe source can be an <VMID> or [IP:]<ZFSPool>[/Path]\n";
-
-my $help_help = "zfs-zsync help <cmd> [OPTIONS]\n
-\tGet help about specified command.\n
-\t<cmd>\tstring\n
-\t\tCommand name\n
-\t-verbose\tboolean\n
-\t\tVerbose output format.\n";
-
-my $help_list = "zfs-zsync list\n
-\tGet a List of all scheduled Sync Jobs\n";
-
-my $help_status = "zfs-zsync status\n
-\tGet the status of all scheduled Sync Jobs\n";
-
-sub help{
- my ($command) = @_;
-
- switch($command){
- case 'help'
- {
- die "$help_help\n";
- }
- case 'sync'
- {
- die "$help_sync\n";
- }
- case 'destroy'
- {
- die "$help_destroy\n";
- }
- case 'create'
- {
- die "$help_create\n";
- }
- case 'list'
- {
- die "$help_list\n";
- }
- case 'status'
- {
- die "$help_status\n";
- }
- }
+my $cmd_help = {
+ destroy => qq{
+$PROGNAME destroy -source <string> [OPTIONS]
-}
+ remove a sync Job from the scheduler
-my $err = GetOptions ('dest=s' => \$dest,
- 'source=s' => \$source,
- 'verbose' => \$verbose,
- 'interval=i' => \$interval,
- 'limit=i' => \$limit,
- 'maxsnap=i' => \$maxsnap,
- 'name=s' => \$name,
- 'skip' => \$skip);
-
-if ($err == 0) {
- die "can't parse options\n";
-}
+ -name string
-my $param;
-$param->{dest} = $dest;
-$param->{source} = $source;
-$param->{verbose} = $verbose;
-$param->{interval} = $interval;
-$param->{limit} = $limit;
-$param->{maxsnap} = $maxsnap;
-$param->{name} = $name;
-$param->{skip} = $skip;
-
-switch($command){
- case "destroy"
- {
- die "$help_destroy\n" if !$source;
- check_target($source);
- destroy($param);
- }
- case "sync"
- {
- die "$help_sync\n" if !$source || !$dest;
- check_target($source);
- check_target($dest);
- sync($param);
- }
- case "create"
- {
- die "$help_create\n" if !$source || !$dest;
- check_target($source);
- check_target($dest);
- init($param);
- }
- case "status"
- {
- print status();
- }
- case "list"
- {
- print list();
- }
- case "help"
- {
- my $help_command = $ARGV[1];
- if ($help_command && $commands->{$help_command}) {
- print help($help_command);
- }
- if ($verbose){
- exec("man $PROGNAME");
- } else {
- usage(1);
- }
- }
-}
+ name of the sync job, if not set it is default
-sub usage{
- my ($help) = @_;
+ -source string
- print("ERROR:\tno command specified\n") if !$help;
- print("USAGE:\t$PROGNAME <COMMAND> [ARGS] [OPTIONS]\n");
- print("\tpve-zsync help [<cmd>] [OPTIONS]\n\n");
- print("\tpve-zsync create -dest <string> -source <string> [OPTIONS]\n");
- print("\tpve-zsync destroy -source <string> [OPTIONS]\n");
- print("\tpve-zsync list\n");
- print("\tpve-zsync status\n");
- print("\tpve-zsync sync -dest <string> -source <string> [OPTIONS]\n");
-}
+ the source can be an <VMID> or [IP:]<ZFSPool>[/Path]
+ },
+ create => qq{
+$PROGNAME create -dest <string> -source <string> [OPTIONS]
-sub check_target{
- my ($target) = @_;
+ Create a sync Job
- chomp($target);
+ -dest string
- if($target !~ m/(\d+.\d+.\d+.\d+:)?([\w\-\_\/]+)(\/.+)?/){
- print("ERROR:\t$target is not valid.\n\tUse [IP:]<ZFSPool>[/Path]!\n");
- return 1;
- }
- return undef;
-}
+ the destination target is like [IP]:<Pool>[/Path]
-__END__
+ -dest-user string
-=head1 NAME
+ name of the user on the destination target, root by default
-pve-zsync - PVE ZFS Replication Manager
+ -limit integer
-=head1 SYNOPSIS
+ max sync speed in kBytes/s, default unlimited
+
+ -maxsnap string
-zfs-zsync <COMMAND> [ARGS] [OPTIONS]
+ how much snapshots will be kept before get erased, default 1
+
+ -name string
+
+ name of the sync job, if not set it is default
+
+ -skip boolean
+
+ if this flag is set it will skip the first sync
+
+ -source string
+
+ the source can be an <VMID> or [IP:]<ZFSPool>[/Path]
+
+ -source-user string
+
+ name of the user on the source target, root by default
+ },
+ sync => qq{
+$PROGNAME sync -dest <string> -source <string> [OPTIONS]\n
+
+ will sync one time
+
+ -dest string
+
+ the destination target is like [IP:]<Pool>[/Path]
+
+ -dest-user string
+
+ name of the user on the destination target, root by default
+
+ -limit integer
-zfs-zsync help <cmd> [OPTIONS]
+ max sync speed in kBytes/s, default unlimited
+
+ -maxsnap integer
+
+ how much snapshots will be kept before get erased, default 1
+
+ -name string
+
+ name of the sync job, if not set it is default.
+ It is only necessary if scheduler allready contains this source.
+
+ -source string
+
+ the source can be an <VMID> or [IP:]<ZFSPool>[/Path]
+
+ -source-user string
+
+ name of the user on the source target, root by default
+
+ -verbose boolean
+
+ print out the sync progress.
+ },
+ list => qq{
+$PROGNAME list
+
+ Get a List of all scheduled Sync Jobs
+ },
+ status => qq{
+$PROGNAME status
+
+ Get the status of all scheduled Sync Jobs
+ },
+ help => qq{
+$PROGNAME help <cmd> [OPTIONS]
Get help about specified command.
-verbose boolean
Verbose output format.
+ },
+ enable => qq{
+$PROGNAME enable -source <string> [OPTIONS]
-zfs-zsync create -dest <string> -source <string> [OPTIONS]
+ enable a syncjob and reset error
- Create a sync Job
+ -name string
- -dest string
+ name of the sync job, if not set it is default
- the destination target is like [IP]:<Pool>[/Path]
+ -source string
- -interval integer
+ the source can be an <VMID> or [IP:]<ZFSPool>[/Path]
+ },
+ disable => qq{
+$PROGNAME disable -source <string> [OPTIONS]
- the interval in min in witch the zfs will sync, default is 15 min
+ pause a sync job
- -limit integer
+ -name string
- max sync speed, default unlimited
+ name of the sync job, if not set it is default
- -maxsnap string
+ -source string
- how much snapshots will be kept before get erased, default 1
+ the source can be an <VMID> or [IP:]<ZFSPool>[/Path]
+ },
+ printpod => 'internal command',
- -name string
+};
- name of the sync job, if not set it is default
+if (!$command) {
+ usage(); die "\n";
+} elsif (!$cmd_help->{$command}) {
+ print "ERROR: unknown command '$command'";
+ usage(1); die "\n";
+}
- -skip boolean
+my @arg = @ARGV;
+my $param = parse_argv(@arg);
- if this flag is set it will skip the first sync
+sub check_params {
+ for (@_) {
+ die "$cmd_help->{$command}\n" if !$param->{$_};
+ }
+}
- -source string
+if ($command eq 'destroy') {
+ check_params(qw(source));
- the source can be an <VMID> or [IP:]<ZFSPool>[/Path]
+ check_target($param->{source});
+ destroy_job($param);
-zfs-zsync destroy -source <string> [OPTIONS]
+} elsif ($command eq 'sync') {
+ check_params(qw(source dest));
- remove a sync Job from the scheduler
+ check_target($param->{source});
+ check_target($param->{dest});
+ sync($param);
- -name string
+} elsif ($command eq 'create') {
+ check_params(qw(source dest));
- name of the sync job, if not set it is default
+ check_target($param->{source});
+ check_target($param->{dest});
+ init($param);
- -source string
+} elsif ($command eq 'status') {
+ print status();
- the source can be an <VMID> or [IP:]<ZFSPool>[/Path]
+} elsif ($command eq 'list') {
+ print list();
-zfs-zsync list
+} elsif ($command eq 'help') {
+ my $help_command = $ARGV[1];
- Get a List of all scheduled Sync Jobs
+ if ($help_command && $cmd_help->{$help_command}) {
+ die "$cmd_help->{$help_command}\n";
-zfs-zsync status
+ }
+ if ($param->{verbose}) {
+ exec("man $PROGNAME");
- Get the status of all scheduled Sync Jobs
+ } else {
+ usage(1);
-zfs-zsync sync -dest <string> -source <string> [OPTIONS]
+ }
- will sync one time
+} elsif ($command eq 'enable') {
+ check_params(qw(source));
- -dest string
+ check_target($param->{source});
+ enable_job($param);
- the destination target is like [IP:]<Pool>[/Path]
+} elsif ($command eq 'disable') {
+ check_params(qw(source));
- -limit integer
+ check_target($param->{source});
+ disable_job($param);
- max sync speed in kBytes/s, default unlimited
+} elsif ($command eq 'printpod') {
+ print_pod();
+}
- -maxsnap integer
+sub usage {
+ my ($help) = @_;
- how much snapshots will be kept before get erased, default 1
+ print("ERROR:\tno command specified\n") if !$help;
+ print("USAGE:\t$PROGNAME <COMMAND> [ARGS] [OPTIONS]\n");
+ print("\t$PROGNAME help [<cmd>] [OPTIONS]\n\n");
+ print("\t$PROGNAME create -dest <string> -source <string> [OPTIONS]\n");
+ print("\t$PROGNAME destroy -source <string> [OPTIONS]\n");
+ print("\t$PROGNAME disable -source <string> [OPTIONS]\n");
+ print("\t$PROGNAME enable -source <string> [OPTIONS]\n");
+ print("\t$PROGNAME list\n");
+ print("\t$PROGNAME status\n");
+ print("\t$PROGNAME sync -dest <string> -source <string> [OPTIONS]\n");
+}
- -name string
+sub check_target {
+ my ($target) = @_;
+ parse_target($target);
+}
- name of the sync job, if not set it is default.
- It is only necessary if scheduler allready contains this source.
+sub print_pod {
- -source string
+ my $synopsis = join("\n", sort values %$cmd_help);
- the source can be an <VMID> or [IP:]<ZFSPool>[/Path]
+ print <<EOF;
+=head1 NAME
+
+pve-zsync - PVE ZFS Replication Manager
+
+=head1 SYNOPSIS
+
+pve-zsync <COMMAND> [ARGS] [OPTIONS]
+
+$synopsis
=head1 DESCRIPTION
This Tool helps you to sync your VM or directory which stored on ZFS between 2 servers.
This tool also has the capability to add jobs to cron so the sync will be automatically done.
+The default syncing interval is set to 15 min, if you want to change this value you can do this in /etc/cron.d/pve-zsync.
+To config cron see man crontab.
=head2 PVE ZFS Storage sync Tool
=head1 EXAMPLES
add sync job from local VM to remote ZFS Server
-zfs-zsync -source=100 -dest=192.168.1.2:zfspool
+pve-zsync create -source=100 -dest=192.168.1.2:zfspool
=head1 IMPORTANT FILES
-Where the cron jobs are stored /etc/cron.d/pve-zsync
-Where the VM config get copied on the destination machine /var/pve-zsync
-Where the config is stored /var/pve-zsync
+Cron jobs and config are stored at /etc/cron.d/pve-zsync
+
+The VM config get copied on the destination machine to /var/lib/pve-zsync/
+
+=head1 COPYRIGHT AND DISCLAIMER
Copyright (C) 2007-2015 Proxmox Server Solutions GmbH
You should have received a copy of the GNU Affero General Public
License along with this program. If not, see
<http://www.gnu.org/licenses/>.
+
+EOF
+}
projects / pve-zsync.git / blobdiff