]> git.proxmox.com Git - pve-storage.git/blobdiff - PVE/Storage.pm
PVE::Storage::storage_can_replicate - hew helper
[pve-storage.git] / PVE / Storage.pm
index a904f4ec84fe18f0191f501dc9d2707815a261a7..cf1e5ec9a7565fccebdb5739484ae8a254aef39b 100755 (executable)
@@ -7,6 +7,7 @@ use Data::Dumper;
 use POSIX;
 use IO::Select;
 use IO::File;
+use IO::Socket::IP;
 use File::Basename;
 use File::Path;
 use Cwd 'abs_path';
@@ -144,6 +145,17 @@ sub storage_check_enabled {
     return storage_check_node($cfg, $storeid, $node, $noerr);
 }
 
+# storage_can_replicate:
+# return true if storage supports replication
+# (volumes alocated with vdisk_alloc() has replication feature)
+sub storage_can_replicate {
+    my ($cfg, $storeid, $format) = @_;
+
+    my $scfg = storage_config($cfg, $storeid);
+    my $plugin = PVE::Storage::Plugin->lookup($scfg->{type});
+    return $plugin->storage_can_replicate($scfg, $storeid, $format);
+}
+
 sub storage_ids {
     my ($cfg) = @_;
 
@@ -262,6 +274,22 @@ sub volume_has_feature {
     }
 }
 
+sub volume_snapshot_list {
+    my ($cfg, $volid) = @_;
+
+    my ($storeid, $volname) = parse_volume_id($volid, 1);
+    if ($storeid) {
+       my $scfg = storage_config($cfg, $storeid);
+       my $plugin = PVE::Storage::Plugin->lookup($scfg->{type});
+       return $plugin->volume_snapshot_list($scfg, $storeid, $volname);
+    } elsif ($volid =~ m|^(/.+)$| && -e $volid) {
+       die "send file/device '$volid' is not possible\n";
+    } else {
+       die "unable to parse volume ID '$volid'\n";
+    }
+    # return an empty array if dataset does not exist.
+}
+
 sub get_image_dir {
     my ($cfg, $storeid, $vmid) = @_;
 
@@ -344,6 +372,32 @@ sub parse_volume_id {
     return PVE::Storage::Plugin::parse_volume_id($volid, $noerr);
 }
 
+# test if we have read access to volid
+sub check_volume_access {
+    my ($rpcenv, $user, $cfg, $vmid, $volid) = @_;
+
+    my ($sid, $volname) = parse_volume_id($volid, 1);
+    if ($sid) {
+       my ($vtype, undef, $ownervm) = parse_volname($cfg, $volid);
+       if ($vtype eq 'iso' || $vtype eq 'vztmpl') {
+           # we simply allow access
+       } elsif (defined($ownervm) && defined($vmid) && ($ownervm == $vmid)) {
+           # we are owner - allow access
+       } elsif ($vtype eq 'backup' && $ownervm) {
+           $rpcenv->check($user, "/storage/$sid", ['Datastore.AllocateSpace']);
+           $rpcenv->check($user, "/vms/$ownervm", ['VM.Backup']);
+       } else {
+           # allow if we are Datastore administrator
+           $rpcenv->check($user, "/storage/$sid", ['Datastore.Allocate']);
+       }
+    } else {
+       die "Only root can pass arbitrary filesystem paths."
+           if $user ne 'root@pam';
+    }
+
+    return undef;
+}
+
 my $volume_is_base_and_used__no_lock = sub {
     my ($scfg, $storeid, $plugin, $volname) = @_;
 
@@ -482,7 +536,7 @@ sub abs_filesystem_path {
 }
 
 sub storage_migrate {
-    my ($cfg, $volid, $target_host, $target_storeid, $target_volname) = @_;
+    my ($cfg, $volid, $target_sshinfo, $target_storeid, $target_volname, $base_snapshot, $snapshot, $ratelimit_bps, $insecure, $with_snapshots) = @_;
 
     my ($storeid, $volname) = parse_volume_id($volid);
     $target_volname = $volname if !$target_volname;
@@ -496,141 +550,85 @@ sub storage_migrate {
 
     my $target_volid = "${target_storeid}:${target_volname}";
 
-    my $errstr = "unable to migrate '$volid' to '${target_volid}' on host '$target_host'";
-
-    my $sshoptions = "-o 'BatchMode=yes'";
-    my $ssh = "/usr/bin/ssh $sshoptions";
-
-    local $ENV{RSYNC_RSH} = $ssh;
-
-    # only implemented for file system based storage
-    if ($scfg->{path}) {
-       if ($tcfg->{path}) {
-
-           my $src_plugin = PVE::Storage::Plugin->lookup($scfg->{type});
-           my $dst_plugin = PVE::Storage::Plugin->lookup($tcfg->{type});
-           my $src = $src_plugin->path($scfg, $volname, $storeid);
-           my $dst = $dst_plugin->path($tcfg, $target_volname, $target_storeid);
-
-           my $dirname = dirname($dst);
-
-           if ($tcfg->{shared}) { # we can do a local copy
-
-               run_command(['/bin/mkdir', '-p', $dirname]);
+    my $target_ip = $target_sshinfo->{ip};
+    my $errstr = "unable to migrate '$volid' to '${target_volid}' on host '$target_sshinfo->{name}'";
 
-               run_command(['/bin/cp', $src, $dst]);
+    my $ssh = PVE::Cluster::ssh_info_to_command($target_sshinfo);
+    my $ssh_base = PVE::Cluster::ssh_info_to_command_base($target_sshinfo);
+    local $ENV{RSYNC_RSH} = PVE::Tools::cmd2string($ssh_base);
 
-           } else {
-               run_command(['/usr/bin/ssh', "root\@${target_host}",
-                            '/bin/mkdir', '-p', $dirname]);
+    my @cstream = ([ '/usr/bin/cstream', '-t', $ratelimit_bps ])
+       if defined($ratelimit_bps);
 
-               # we use rsync with --sparse, so we can't use --inplace,
-               # so we remove file on the target if it already exists to
-               # save space
-               my ($size, $format) = PVE::Storage::Plugin::file_size_info($src);
-               if ($format && ($format eq 'raw') && $size) {
-                   run_command(['/usr/bin/ssh', "root\@${target_host}",
-                                'rm', '-f', $dst],
-                               outfunc => sub {});
-               }
-
-               my $cmd;
-               if ($format eq 'subvol') {
-                   $cmd = ['/usr/bin/rsync', '--progress', '-X', '-A', '--numeric-ids',
-                           '-aH', '--delete', '--no-whole-file', '--inplace',
-                           '--one-file-system', "$src/", "[root\@${target_host}]:$dst"];
-               } else {
-                   $cmd = ['/usr/bin/rsync', '--progress', '--sparse', '--whole-file',
-                           $src, "[root\@${target_host}]:$dst"];
-               }
-
-               my $percent = -1;
-
-               run_command($cmd, outfunc => sub {
-                   my $line = shift;
-
-                   if ($line =~ m/^\s*(\d+\s+(\d+)%\s.*)$/) {
-                       if ($2 > $percent) {
-                           $percent = $2;
-                           print "rsync status: $1\n";
-                           *STDOUT->flush();
-                       }
-                   } else {
-                       print "$line\n";
-                       *STDOUT->flush();
-                   }
-               });
-           }
-       } else {
-           die "$errstr - target type '$tcfg->{type}' not implemented\n";
+    my $migration_snapshot;
+    if (!defined($snapshot)) {
+       if ($scfg->{type} eq 'zfspool') {
+           $migration_snapshot = 1;
+           $snapshot = '__migration__';
        }
+    }
 
-    } elsif ($scfg->{type} eq 'zfspool') {
-
-       if ($tcfg->{type} eq 'zfspool') {
-
-           die "$errstr - pool on target does not have the same name as on source!"
-               if $tcfg->{pool} ne $scfg->{pool};
-
-           my (undef, $volname) = parse_volname($cfg, $volid);
-
-           my $zfspath = "$scfg->{pool}\/$volname";
-
-           my $snap = ['zfs', 'snapshot', "$zfspath\@__migration__"];
-
-           my $send = [['zfs', 'send', '-Rpv', "$zfspath\@__migration__"], ['ssh', "root\@$target_host",
-                       'zfs', 'recv', $zfspath]];
-
-           my $destroy_target = ['ssh', "root\@$target_host", 'zfs', 'destroy', "$zfspath\@__migration__"];
-           run_command($snap);
-           eval{
-               run_command($send);
-           };
-           my $err;
-           if ($err = $@){
-               run_command(['zfs', 'destroy', "$zfspath\@__migration__"]);
-               die $err;
-           }
-           run_command($destroy_target);
-
-       } else {
-           die "$errstr - target type $tcfg->{type} is not valid\n";
-       }
-
-    } elsif ($scfg->{type} eq 'lvmthin' || $scfg->{type} eq 'lvm') {
+    my @formats = volume_transfer_formats($cfg, $volid, $volid, $snapshot, $base_snapshot, $with_snapshots);
+    die "cannot migrate from storage type '$scfg->{type}' to '$tcfg->{type}'\n" if !@formats;
+    my $format = $formats[0];
 
-       if (($scfg->{type} eq $tcfg->{type}) &&
-           ($tcfg->{type} eq 'lvmthin' || $tcfg->{type} eq 'lvm')) {
+    my @insecurecmd;
+    if ($insecure) {
+       @insecurecmd = ('pvecm', 'mtunnel', '-run-command', 1);
+       if (my $network = $target_sshinfo->{network}) {
+           push @insecurecmd, '-migration_network', $network;
+       }
+    }
 
-           my (undef, $volname, $vmid) = parse_volname($cfg, $volid);
-           my $size = volume_size_info($cfg, $volid, 5);
-           my $src = path($cfg, $volid);
-           my $dst = path($cfg, $target_volid);
+    $with_snapshots = $with_snapshots ? 1 : 0; # sanitize for passing as cli parameter
+    my $send = ['pvesm', 'export', $volid, $format, '-', '-with-snapshots', $with_snapshots];
+    my $recv = [@$ssh, @insecurecmd, '--', 'pvesm', 'import', $volid, $format, '-', '-with-snapshots', $with_snapshots];
+    if (defined($snapshot)) {
+       push @$send, '-snapshot', $snapshot
+    }
+    if ($migration_snapshot) {
+       push @$recv, '-delete-snapshot', $snapshot;
+    }
 
-           run_command(['/usr/bin/ssh', "root\@${target_host}",
-                        'pvesm', 'alloc', $target_storeid, $vmid,
-                         $target_volname, int($size/1024)]);
+    if (defined($base_snapshot)) {
+       # Check if the snapshot exists on the remote side:
+       push @$send, '-base', $base_snapshot;
+       push @$recv, '-base', $base_snapshot;
+    }
 
-           eval {
-               if ($tcfg->{type} eq 'lvmthin') {
-                   run_command([["dd", "if=$src"],["/usr/bin/ssh", "root\@${target_host}",
-                             "dd", 'conv=sparse', "of=$dst"]]);
-               } else {
-                   run_command([["dd", "if=$src"],["/usr/bin/ssh", "root\@${target_host}",
-                             "dd", "of=$dst"]]);
-               }
-           };
-           if (my $err = $@) {
-               run_command(['/usr/bin/ssh', "root\@${target_host}",
-                        'pvesm', 'free', $target_volid]);
-               die $err;
+    volume_snapshot($cfg, $volid, $snapshot) if $migration_snapshot;
+    eval {
+       if ($insecure) {
+           open(my $info, '-|', @$recv)
+               or die "receive command failed: $!\n";
+           my ($ip) = <$info> =~ /^($PVE::Tools::IPRE)$/ or die "no tunnel IP received\n";
+           my ($port) = <$info> =~ /^(\d+)$/ or die "no tunnel port received\n";
+           my $socket = IO::Socket::IP->new(PeerHost => $ip, PeerPort => $port, Type => SOCK_STREAM)
+               or die "failed to connect to tunnel at $ip:$port\n";
+           # we won't be reading from the socket
+           shutdown($socket, 0);
+           run_command([$send, @cstream], output => '>&'.fileno($socket));
+           # don't close the connection entirely otherwise the receiving end
+           # might not get all buffered data (and fails with 'connection reset by peer')
+           shutdown($socket, 1);
+           1 while <$info>; # wait for the remote process to finish
+           # now close the socket
+           close($socket);
+           if (!close($info)) { # does waitpid()
+               die "import failed: $!\n" if $!;
+               die "import failed: exit code ".($?>>8)."\n";
            }
        } else {
-           die "$errstr - migrate from source type '$scfg->{type}' to '$tcfg->{type}' not implemented\n";
+           run_command([$send, @cstream, $recv]);
        }
-    } else {
-       die "$errstr - source type '$scfg->{type}' not implemented\n";
+    };
+    my $err = $@;
+    warn "send/receive failed, cleaning up snapshot(s)..\n" if $err;
+    if ($migration_snapshot) {
+       eval { volume_snapshot_delete($cfg, $volid, $snapshot, 0) };
+       warn "could not remove source snapshot: $@\n" if $@;
     }
+    die $err if $err;
 }
 
 sub vdisk_clone {
@@ -1300,7 +1298,7 @@ sub extract_vzdump_config_tar {
 
     my $file;
     while (defined($file = <$fh>)) {
-       if ($file =~ m!$conf_re!) {
+       if ($file =~ $conf_re) {
            $file = $1; # untaint
            last;
        }
@@ -1368,7 +1366,7 @@ sub extract_vzdump_config_vma {
        my $rerr = $@;
 
        # use exit code if no stderr output and not just broken pipe
-       if (!$errstring && !$broken_pipe && $rc > 0 && $rc != 141) {
+       if (!$errstring && !$broken_pipe && $rc != 0 && $rc != 141) {
            die "$rerr\n" if $rerr;
            die "config extraction failed with exit code $rc\n";
        }
@@ -1386,9 +1384,9 @@ sub extract_vzdump_config {
 
     my $archive = abs_filesystem_path($cfg, $volid);
 
-    if ($volid =~ /\/vzdump-(lxc|openvz)-\d+-(\d{4})_(\d{2})_(\d{2})-(\d{2})_(\d{2})_(\d{2})\.(tgz|(tar(\.(gz|lzo))?))$/) {
-       return extract_vzdump_config_tar($archive,'^(\./etc/vzdump/(pct|vps)\.conf)$');
-    } elsif ($volid =~ /\/vzdump-qemu-\d+-(\d{4})_(\d{2})_(\d{2})-(\d{2})_(\d{2})_(\d{2})\.(tgz|((tar|vma)(\.(gz|lzo))?))$/) {
+    if ($volid =~ /vzdump-(lxc|openvz)-\d+-(\d{4})_(\d{2})_(\d{2})-(\d{2})_(\d{2})_(\d{2})\.(tgz|(tar(\.(gz|lzo))?))$/) {
+       return extract_vzdump_config_tar($archive, qr!^(\./etc/vzdump/(pct|vps)\.conf)$!);
+    } elsif ($volid =~ /vzdump-qemu-\d+-(\d{4})_(\d{2})_(\d{2})-(\d{2})_(\d{2})_(\d{2})\.(tgz|((tar|vma)(\.(gz|lzo))?))$/) {
        my $format;
        my $comp;
        if ($7 eq 'tgz') {
@@ -1409,6 +1407,60 @@ sub extract_vzdump_config {
     }
 }
 
+sub volume_export {
+    my ($cfg, $fh, $volid, $format, $snapshot, $base_snapshot, $with_snapshots) = @_;
+
+    my ($storeid, $volname) = parse_volume_id($volid, 1);
+    die "cannot export volume '$volid'\n" if !$storeid;
+    my $scfg = storage_config($cfg, $storeid);
+    my $plugin = PVE::Storage::Plugin->lookup($scfg->{type});
+    return $plugin->volume_export($scfg, $storeid, $fh, $volname, $format,
+                                  $snapshot, $base_snapshot, $with_snapshots);
+}
+
+sub volume_import {
+    my ($cfg, $fh, $volid, $format, $base_snapshot, $with_snapshots) = @_;
+
+    my ($storeid, $volname) = parse_volume_id($volid, 1);
+    die "cannot import into volume '$volid'\n" if !$storeid;
+    my $scfg = storage_config($cfg, $storeid);
+    my $plugin = PVE::Storage::Plugin->lookup($scfg->{type});
+    return $plugin->volume_import($scfg, $storeid, $fh, $volname, $format,
+                                  $base_snapshot, $with_snapshots);
+}
+
+sub volume_export_formats {
+    my ($cfg, $volid, $snapshot, $base_snapshot, $with_snapshots) = @_;
+
+    my ($storeid, $volname) = parse_volume_id($volid, 1);
+    return if !$storeid;
+    my $scfg = storage_config($cfg, $storeid);
+    my $plugin = PVE::Storage::Plugin->lookup($scfg->{type});
+    return $plugin->volume_export_formats($scfg, $storeid, $volname,
+                                          $snapshot, $base_snapshot,
+                                          $with_snapshots);
+}
+
+sub volume_import_formats {
+    my ($cfg, $volid, $base_snapshot, $with_snapshots) = @_;
+
+    my ($storeid, $volname) = parse_volume_id($volid, 1);
+    return if !$storeid;
+    my $scfg = storage_config($cfg, $storeid);
+    my $plugin = PVE::Storage::Plugin->lookup($scfg->{type});
+    return $plugin->volume_import_formats($scfg, $storeid, $volname,
+                                          $base_snapshot, $with_snapshots);
+}
+
+sub volume_transfer_formats {
+    my ($cfg, $src_volid, $dst_volid, $snapshot, $base_snapshot, $with_snapshots) = @_;
+    my @export_formats = volume_export_formats($cfg, $src_volid, $snapshot, $base_snapshot, $with_snapshots);
+    my @import_formats = volume_import_formats($cfg, $dst_volid, $base_snapshot, $with_snapshots);
+    my %import_hash = map { $_ => 1 } @import_formats;
+    my @common = grep { $import_hash{$_} } @export_formats;
+    return @common;
+}
+
 # bash completion helper
 
 sub complete_storage {