X-Git-Url: https://git.proxmox.com/?a=blobdiff_plain;f=PVE%2FStorage.pm;h=4772c6c06c1f2b64b8b4903d457ba77ffbb25481;hb=e8a7e764afbee31609583ecce4300ba24cfff199;hp=c27e9cf30ecbae65523ef4e81a9a2192d272a25c;hpb=02e797b8e10d5fe326c3c054962371e253a11e57;p=pve-storage.git diff --git a/PVE/Storage.pm b/PVE/Storage.pm index c27e9cf..4772c6c 100755 --- a/PVE/Storage.pm +++ b/PVE/Storage.pm @@ -7,13 +7,14 @@ 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'; use Socket; -use PVE::Tools qw(run_command file_read_firstline $IPV6RE); -use PVE::Cluster qw(cfs_read_file cfs_lock_file); +use PVE::Tools qw(run_command file_read_firstline dir_glob_foreach $IPV6RE); +use PVE::Cluster qw(cfs_read_file cfs_write_file cfs_lock_file); use PVE::Exception qw(raise_param_exc); use PVE::JSONSchema; use PVE::INotify; @@ -22,6 +23,7 @@ use PVE::RPCEnvironment; use PVE::Storage::Plugin; use PVE::Storage::DirPlugin; use PVE::Storage::LVMPlugin; +use PVE::Storage::LvmThinPlugin; use PVE::Storage::NFSPlugin; use PVE::Storage::ISCSIPlugin; use PVE::Storage::RBDPlugin; @@ -32,9 +34,13 @@ use PVE::Storage::ZFSPoolPlugin; use PVE::Storage::ZFSPlugin; use PVE::Storage::DRBDPlugin; -# load and initialize all plugins +# Storage API version. Icrement it on changes in storage API interface. +use constant APIVER => 1; + +# load standard plugins PVE::Storage::DirPlugin->register(); PVE::Storage::LVMPlugin->register(); +PVE::Storage::LvmThinPlugin->register(); PVE::Storage::NFSPlugin->register(); PVE::Storage::ISCSIPlugin->register(); PVE::Storage::RBDPlugin->register(); @@ -44,6 +50,34 @@ PVE::Storage::GlusterfsPlugin->register(); PVE::Storage::ZFSPoolPlugin->register(); PVE::Storage::ZFSPlugin->register(); PVE::Storage::DRBDPlugin->register(); + +# load third-party plugins +if ( -d '/usr/share/perl5/PVE/Storage/Custom' ) { + dir_glob_foreach('/usr/share/perl5/PVE/Storage/Custom', '.*\.pm$', sub { + my ($file) = @_; + my $modname = 'PVE::Storage::Custom::' . $file; + $modname =~ s!\.pm$!!; + $file = 'PVE/Storage/Custom/' . $file; + + eval { + require $file; + }; + if ($@) { + warn $@; + # Check storage API version and that file is really storage plugin. + } elsif ($modname->isa('PVE::Storage::Plugin') && $modname->can('api') && $modname->api() == APIVER) { + eval { + import $file; + $modname->register(); + }; + warn $@ if $@; + } else { + warn "Error loading storage plugin \"$modname\" because of API version mismatch. Please, update it.\n" + } + }); +} + +# initialize all plugins PVE::Storage::Plugin->init(); my $UDEVADM = '/sbin/udevadm'; @@ -54,6 +88,12 @@ sub config { return cfs_read_file("storage.cfg"); } +sub write_config { + my ($cfg) = @_; + + cfs_write_file('storage.cfg', $cfg); +} + sub lock_storage_config { my ($code, $errmsg) = @_; @@ -67,7 +107,7 @@ sub lock_storage_config { sub storage_config { my ($cfg, $storeid, $noerr) = @_; - die "no storage id specified\n" if !$storeid; + die "no storage ID specified\n" if !$storeid; my $scfg = $cfg->{ids}->{$storeid}; @@ -223,6 +263,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) = @_; @@ -282,6 +338,9 @@ sub parse_vmid { return int($vmid); } +# NOTE: basename and basevmid are always undef for LVM-thin, where the +# clone -> base reference is not encoded in the volume ID. +# see note in PVE::Storage::LvmThinPlugin for details. sub parse_volname { my ($cfg, $volid) = @_; @@ -302,25 +361,71 @@ sub parse_volume_id { return PVE::Storage::Plugin::parse_volume_id($volid, $noerr); } -sub volume_is_base { - my ($cfg, $volid) = @_; +# 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); - return 0 if !$sid; - - if (my $scfg = $cfg->{ids}->{$sid}) { - my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); - my ($vtype, $name, $vmid, $basename, $basevmid, $isBase) = - $plugin->parse_volname($volname); - return $isBase ? 1 : 0; - } else { - # stale volid with undefined storage - so we can just guess - if ($volid =~ m/base-/) { - return 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) = @_; + + my ($vtype, $name, $vmid, undef, undef, $isBase, undef) = + $plugin->parse_volname($volname); + + if ($isBase) { + my $vollist = $plugin->list_images($storeid, $scfg); + foreach my $info (@$vollist) { + my (undef, $tmpvolname) = parse_volume_id($info->{volid}); + my $basename = undef; + my $basevmid = undef; + + eval{ + (undef, undef, undef, $basename, $basevmid) = + $plugin->parse_volname($tmpvolname); + }; + + if ($basename && defined($basevmid) && $basevmid == $vmid && $basename eq $name) { + return 1; + } + } + } return 0; +}; + +# NOTE: this check does not work for LVM-thin, where the clone -> base +# reference is not encoded in the volume ID. +# see note in PVE::Storage::LvmThinPlugin for details. +sub volume_is_base_and_used { + my ($cfg, $volid) = @_; + + my ($storeid, $volname) = parse_volume_id($volid); + my $scfg = storage_config($cfg, $storeid); + my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); + + $plugin->cluster_lock_storage($storeid, $scfg->{shared}, undef, sub { + return &$volume_is_base_and_used__no_lock($scfg, $storeid, $plugin, $volname); + }); } # try to map a filesystem path to a volume identifier @@ -420,7 +525,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; @@ -434,102 +539,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"; + my $target_ip = $target_sshinfo->{ip}; + my $errstr = "unable to migrate '$volid' to '${target_volid}' on host '$target_sshinfo->{name}'"; - local $ENV{RSYNC_RSH} = $ssh; + 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); - # only implemented for file system based storage - if ($scfg->{path}) { - if ($tcfg->{path}) { + my @cstream = ([ '/usr/bin/cstream', '-t', $ratelimit_bps ]) + if defined($ratelimit_bps); - 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 $migration_snapshot; + if (!defined($snapshot)) { + if ($scfg->{type} eq 'zfspool') { + $migration_snapshot = 1; + $snapshot = '__migration__'; + } + } - run_command(['/bin/cp', $src, $dst]); + 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]; - } else { + my @insecurecmd; + if ($insecure) { + @insecurecmd = ('pvecm', 'mtunnel', '-run-command', 1); + if (my $network = $target_sshinfo->{network}) { + push @insecurecmd, '-migration_network', $network; + } + } - run_command(['/usr/bin/ssh', "root\@${target_host}", - '/bin/mkdir', '-p', $dirname]); + $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; + } - # 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 {}); - } + if (defined($base_snapshot)) { + # Check if the snapshot exists on the remote side: + push @$send, '-base', $base_snapshot; + push @$recv, '-base', $base_snapshot; + } - my $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(); - } - }); + 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 - target type '$tcfg->{type}' not implemented\n"; + run_command([$send, @cstream, $recv]); } - - } elsif ($scfg->{type} eq 'zfspool') { - - if ($tcfg->{type} eq 'zfspool') { - - die "$errstr - pool on target has not same name as 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 -v $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"; - } - } 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 { @@ -571,7 +659,7 @@ sub vdisk_create_base { sub vdisk_alloc { my ($cfg, $storeid, $vmid, $fmt, $name, $size) = @_; - die "no storage id specified\n" if !$storeid; + die "no storage ID specified\n" if !$storeid; PVE::JSONSchema::parse_storage_id($storeid); @@ -604,9 +692,7 @@ sub vdisk_free { my ($cfg, $volid) = @_; my ($storeid, $volname) = parse_volume_id($volid); - my $scfg = storage_config($cfg, $storeid); - my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); activate_storage($cfg, $storeid); @@ -615,27 +701,12 @@ sub vdisk_free { # lock shared storage $plugin->cluster_lock_storage($storeid, $scfg->{shared}, undef, sub { + # LVM-thin allows deletion of still referenced base volumes! + die "base volume '$volname' is still in use by linked clones\n" + if &$volume_is_base_and_used__no_lock($scfg, $storeid, $plugin, $volname); - my ($vtype, $name, $vmid, undef, undef, $isBase, $format) = + my (undef, undef, undef, undef, undef, $isBase, $format) = $plugin->parse_volname($volname); - if ($isBase) { - my $vollist = $plugin->list_images($storeid, $scfg); - foreach my $info (@$vollist) { - my (undef, $tmpvolname) = parse_volume_id($info->{volid}); - my $basename = undef; - my $basevmid = undef; - - eval{ - (undef, undef, undef, $basename, $basevmid) = - $plugin->parse_volname($tmpvolname); - }; - - if ($basename && defined($basevmid) && $basevmid == $vmid && $basename eq $name) { - die "base volume '$volname' is still in use " . - "(use by '$tmpvolname')\n"; - } - } - } $cleanup_worker = $plugin->free_image($storeid, $scfg, $volname, $isBase, $format); }); @@ -759,6 +830,44 @@ sub vdisk_list { return $res; } +sub volume_list { + my ($cfg, $storeid, $vmid, $content) = @_; + + my @ctypes = qw(images vztmpl iso backup); + + my $cts = $content ? [ $content ] : [ @ctypes ]; + + my $scfg = PVE::Storage::storage_config($cfg, $storeid); + + my $res = []; + foreach my $ct (@$cts) { + my $data; + if ($ct eq 'images') { + $data = vdisk_list($cfg, $storeid, $vmid); + } elsif ($ct eq 'iso' && !defined($vmid)) { + $data = template_list($cfg, $storeid, 'iso'); + } elsif ($ct eq 'vztmpl'&& !defined($vmid)) { + $data = template_list ($cfg, $storeid, 'vztmpl'); + } elsif ($ct eq 'backup') { + $data = template_list ($cfg, $storeid, 'backup'); + foreach my $item (@{$data->{$storeid}}) { + if (defined($vmid)) { + @{$data->{$storeid}} = grep { $_->{volid} =~ m/\S+-$vmid-\S+/ } @{$data->{$storeid}}; + } + } + } + + next if !$data || !$data->{$storeid}; + + foreach my $item (@{$data->{$storeid}}) { + $item->{content} = $ct; + push @$res, $item; + } + } + + return $res; +} + sub uevent_seqnum { my $filename = "/sys/kernel/uevent_seqnum"; @@ -876,7 +985,7 @@ sub deactivate_volumes { } } - die "volume deativation failed: " . join(' ', @errlist) + die "volume deactivation failed: " . join(' ', @errlist) if scalar(@errlist); } @@ -886,9 +995,9 @@ sub storage_info { my $ids = $cfg->{ids}; my $info = {}; - + my @ctypes = PVE::Tools::split_list($content); - + my $slist = []; foreach my $storeid (keys %$ids) { @@ -904,7 +1013,7 @@ sub storage_info { } next if !$want_ctype; } - + my $type = $ids->{$storeid}->{type}; $info->{$storeid} = { @@ -937,9 +1046,9 @@ sub storage_info { eval { ($total, $avail, $used, $active) = $plugin->status($storeid, $scfg, $cache); }; warn $@ if $@; next if !$active; - $info->{$storeid}->{total} = $total; - $info->{$storeid}->{avail} = $avail; - $info->{$storeid}->{used} = $used; + $info->{$storeid}->{total} = int($total); + $info->{$storeid}->{avail} = int($avail); + $info->{$storeid}->{used} = int($used); $info->{$storeid}->{active} = $active; } @@ -997,7 +1106,8 @@ sub scan_zfs { my $size = PVE::Storage::ZFSPoolPlugin::zfs_parse_size($size_str); my $used = PVE::Storage::ZFSPoolPlugin::zfs_parse_size($used_str); # ignore subvolumes generated by our ZFSPoolPlugin - return if $pool =~ m!/subvol-\d+-[^/]+$!; + return if $pool =~ m!/subvol-\d+-[^/]+$!; + return if $pool =~ m!/basevol-\d+-[^/]+$!; push @$res, { pool => $pool, size => $size, free => $size-$used }; } }); @@ -1167,25 +1277,240 @@ sub foreach_volid { } } +sub extract_vzdump_config_tar { + my ($archive, $conf_re) = @_; + + die "ERROR: file '$archive' does not exist\n" if ! -f $archive; + + my $pid = open(my $fh, '-|', 'tar', 'tf', $archive) || + die "unable to open file '$archive'\n"; + + my $file; + while (defined($file = <$fh>)) { + if ($file =~ $conf_re) { + $file = $1; # untaint + last; + } + } + + kill 15, $pid; + waitpid $pid, 0; + close $fh; + + die "ERROR: archive contains no configuration file\n" if !$file; + chomp $file; + + my $raw = ''; + my $out = sub { + my $output = shift; + $raw .= "$output\n"; + }; + + PVE::Tools::run_command(['tar', '-xpOf', $archive, $file, '--occurrence'], outfunc => $out); + + return wantarray ? ($raw, $file) : $raw; +} + +sub extract_vzdump_config_vma { + my ($archive, $comp) = @_; + + my $cmd; + my $raw = ''; + my $out = sub { + my $output = shift; + $raw .= "$output\n"; + }; + + + if ($comp) { + my $uncomp; + if ($comp eq 'gz') { + $uncomp = ["zcat", $archive]; + } elsif ($comp eq 'lzo') { + $uncomp = ["lzop", "-d", "-c", $archive]; + } else { + die "unknown compression method '$comp'\n"; + } + $cmd = [$uncomp, ["vma", "config", "-"]]; + + # in some cases, lzop/zcat exits with 1 when its stdout pipe is + # closed early by vma, detect this and ignore the exit code later + my $broken_pipe; + my $errstring; + my $err = sub { + my $output = shift; + if ($output =~ m/lzop: Broken pipe: / || $output =~ m/gzip: stdout: Broken pipe/) { + $broken_pipe = 1; + } elsif (!defined ($errstring) && $output !~ m/^\s*$/) { + $errstring = "Failed to extract config from VMA archive: $output\n"; + } + }; + + # in other cases, the pipeline will exit with exit code 141 + # because of the broken pipe, handle / ignore this as well + my $rc; + eval { + $rc = PVE::Tools::run_command($cmd, outfunc => $out, errfunc => $err, noerr => 1); + }; + my $rerr = $@; + + # use exit code if no stderr output and not just broken pipe + if (!$errstring && !$broken_pipe && $rc != 0 && $rc != 141) { + die "$rerr\n" if $rerr; + die "config extraction failed with exit code $rc\n"; + } + die "$errstring\n" if $errstring; + } else { + # simple case without compression and weird piping behaviour + PVE::Tools::run_command(["vma", "config", $archive], outfunc => $out); + } + + return wantarray ? ($raw, undef) : $raw; +} + +sub extract_vzdump_config { + my ($cfg, $volid) = @_; + + 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, 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') { + $format = 'tar'; + $comp = 'gz'; + } else { + $format = $9; + $comp = $11 if defined($11); + } + + if ($format eq 'tar') { + return extract_vzdump_config_tar($archive, qr!\(\./qemu-server\.conf\)!); + } else { + return extract_vzdump_config_vma($archive, $comp); + } + } else { + die "cannot determine backup guest type for backup archive '$volid'\n"; + } +} + +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 { - my ($cmdname, $pname, $cvalue) = @_; + my ($cmdname, $pname, $cvalue) = @_; - return $cmdname eq 'add' ? [] : [ PVE::Storage::storage_ids() ]; + my $cfg = PVE::Storage::config(); + + return $cmdname eq 'add' ? [] : [ PVE::Storage::storage_ids($cfg) ]; } sub complete_storage_enabled { - my ($cmdname, $pname, $cvalue) = @_; + my ($cmdname, $pname, $cvalue) = @_; + + my $res = []; + + my $cfg = PVE::Storage::config(); + foreach my $sid (keys %{$cfg->{ids}}) { + next if !storage_check_enabled($cfg, $sid, undef, 1); + push @$res, $sid; + } + return $res; +} + +sub complete_content_type { + my ($cmdname, $pname, $cvalue) = @_; + + return [qw(rootdir images vztmpl iso backup)]; +} + +sub complete_volume { + my ($cmdname, $pname, $cvalue) = @_; - my $res = []; + my $cfg = config(); - my $cfg = PVE::Storage::config(); - foreach my $sid (keys %{$cfg->{ids}}) { - next if !storage_check_enabled($cfg, $sid, undef, 1); - push @$res, $sid; - } - return $res; + my $storage_list = complete_storage_enabled(); + + if ($cvalue =~ m/^([^:]+):/) { + $storage_list = [ $1 ]; + } else { + if (scalar(@$storage_list) > 1) { + # only list storage IDs to avoid large listings + my $res = []; + foreach my $storeid (@$storage_list) { + # Hack: simply return 2 artificial values, so that + # completions does not finish + push @$res, "$storeid:volname", "$storeid:..."; + } + return $res; + } + } + + my $res = []; + foreach my $storeid (@$storage_list) { + my $vollist = PVE::Storage::volume_list($cfg, $storeid); + + foreach my $item (@$vollist) { + push @$res, $item->{volid}; + } + } + + return $res; } 1;