iterate through subdirs to find all the isos/container
templates/snippets.

Signed-off-by: Noel Ullreich <[email protected]>
---
 PVE/Storage/Plugin.pm | 65 ++++++++++++++++++++++++++++---------------
 1 file changed, 43 insertions(+), 22 deletions(-)

diff --git a/PVE/Storage/Plugin.pm b/PVE/Storage/Plugin.pm
index ca7a0d4..bf1d564 100644
--- a/PVE/Storage/Plugin.pm
+++ b/PVE/Storage/Plugin.pm
@@ -9,6 +9,7 @@ use File::chdir;
 use File::Path;
 use File::Basename;
 use File::stat qw();
+use File::Find;
 
 use PVE::Tools qw(run_command);
 use PVE::JSONSchema qw(get_standard_option register_standard_option);
@@ -1275,46 +1276,66 @@ sub list_volumes {
     my $vmlist = PVE::Cluster::get_vmlist();
     foreach my $type (@$content_types) {
        my $data;
+       my @list_of_dirs;
 
        if ($type eq 'images' || $type eq 'rootdir') {
            $data = $class->list_images($storeid, $scfg, $vmid);
+           push (@list_of_dirs, $data) if $data; #fix this
        } elsif ($scfg->{path}) {
            my $path = $class->get_subdir($scfg, $type);
+           my @subdirs;
+           my $wanted = sub {
+               my ($dev,$ino,$mode,$nlink,$uid,$gid);
+
+               (($dev,$ino,$mode,$nlink,$uid,$gid) = lstat($_)) &&
+               -d _
+               && push(@subdirs, $File::Find::name);
+           };
+           File::Find::find({wanted => $wanted, untaint => 1}, $path);
 
            if ($type eq 'iso' && !defined($vmid)) {
-               $data = $get_subdir_files->($storeid, $path, 'iso');
+               for (@subdirs) {
+                   $data = $get_subdir_files->($storeid, $_, 'iso',);
+                   push @list_of_dirs, $data;
+               }
            } elsif ($type eq 'vztmpl'&& !defined($vmid)) {
-               $data = $get_subdir_files->($storeid, $path, 'vztmpl');
+               for (@subdirs) {
+                   $data = $get_subdir_files->($storeid, $_, 'vztmpl');
+                   push @list_of_dirs, $data;
+               }
            } elsif ($type eq 'backup') {
                $data = $get_subdir_files->($storeid, $path, 'backup', $vmid);
+               push(@list_of_dirs, $data) if $data;
            } elsif ($type eq 'snippets') {
-               $data = $get_subdir_files->($storeid, $path, 'snippets');
+               for (@subdirs){
+                   $data = $get_subdir_files->($storeid, $_, 'snippets');
+                   push(@list_of_dirs, $data) if $data;
+               }
            }
        }
-
-       next if !$data;
-
-       foreach my $item (@$data) {
-           if ($type eq 'images' || $type eq 'rootdir') {
-               my $vminfo = $vmlist->{ids}->{$item->{vmid}};
-               my $vmtype;
-               if (defined($vminfo)) {
-                   $vmtype = $vminfo->{type};
-               }
-               if (defined($vmtype) && $vmtype eq 'lxc') {
-                   $item->{content} = 'rootdir';
+       next if !@list_of_dirs; #if list of dirs is empty, there is no data 
either
+
+       for (@list_of_dirs) {
+           foreach my $item (@$_) {
+               if ($type eq 'images' || $type eq 'rootdir') {
+                   my $vminfo = $vmlist->{ids}->{$item->{vmid}};
+                   my $vmtype;
+                   if (defined($vminfo)) {
+                       $vmtype = $vminfo->{type};
+                   }
+                   if (defined($vmtype) && $vmtype eq 'lxc') {
+                       $item->{content} = 'rootdir';
+                   } else {
+                       $item->{content} = 'images';
+                   }
+                   next if $type ne $item->{content};
                } else {
-                   $item->{content} = 'images';
+                   $item->{content} = $type;
                }
-               next if $type ne $item->{content};
-           } else {
-               $item->{content} = $type;
+               push @$res, $item;
            }
-
-           push @$res, $item;
        }
     }
-
     return $res;
 }
 
-- 
2.30.2



_______________________________________________
pve-devel mailing list
[email protected]
https://lists.proxmox.com/cgi-bin/mailman/listinfo/pve-devel

Reply via email to