--- Begin Message ---
From: Arnaud Pascal <r...@arnaud.sh>

Currently allowing only `.yml` and `.yaml` files.

Signed-off-by: Arnaud Pascal <r...@arnaud.sh>
---
 src/PVE/API2/Storage/Status.pm | 14 ++++++++++++--
 src/PVE/Storage.pm             | 11 +++++++++++
 2 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/src/PVE/API2/Storage/Status.pm b/src/PVE/API2/Storage/Status.pm
index dc6cc69..d3310e8 100644
--- a/src/PVE/API2/Storage/Status.pm
+++ b/src/PVE/API2/Storage/Status.pm
@@ -382,7 +382,7 @@ __PACKAGE__->register_method ({
            content => {
                description => "Content type.",
                type => 'string', format => 'pve-storage-content',
-               enum => ['iso', 'vztmpl'],
+               enum => ['iso', 'vztmpl', 'snippets'],
            },
            filename => {
                description => "The name of the file to create. Caution: This 
will be normalized!",
@@ -448,6 +448,11 @@ __PACKAGE__->register_method ({
                raise_param_exc({ filename => "wrong file extension" });
            }
            $path = PVE::Storage::get_vztmpl_dir($cfg, $param->{storage});
+       } elsif ($content eq 'snippets') {
+           if ($filename !~ m![^/]+$PVE::Storage::SNIPPETS_EXT_RE_0$!) {
+               raise_param_exc({ filename => "wrong file extension" });
+           }
+           $path = PVE::Storage::get_snippets_dir($cfg, $param->{storage});
        } else {
            raise_param_exc({ content => "upload content type '$content' not 
allowed" });
        }
@@ -572,7 +577,7 @@ __PACKAGE__->register_method({
            content => {
                description => "Content type.", # TODO: could be optional & 
detected in most cases
                type => 'string', format => 'pve-storage-content',
-               enum => ['iso', 'vztmpl'],
+               enum => ['iso', 'vztmpl', 'snippets'],
            },
            filename => {
                description => "The name of the file to create. Caution: This 
will be normalized!",
@@ -642,6 +647,11 @@ __PACKAGE__->register_method({
                raise_param_exc({ filename => "wrong file extension" });
            }
            $path = PVE::Storage::get_vztmpl_dir($cfg, $storage);
+       } elsif ($content eq 'snippets') {
+           if ($filename !~ m![^/]+$PVE::Storage::SNIPPETS_EXT_RE_0$!) {
+               raise_param_exc({ filename => "wrong file extension" });
+           }
+           $path = PVE::Storage::get_snippets_dir($cfg, $storage);
        } else {
            raise_param_exc({ content => "upload content-type '$content' is not 
allowed" });
        }
diff --git a/src/PVE/Storage.pm b/src/PVE/Storage.pm
index 57b2038..73cf991 100755
--- a/src/PVE/Storage.pm
+++ b/src/PVE/Storage.pm
@@ -112,6 +112,8 @@ our $ISO_EXT_RE_0 = qr/\.(?:iso|img)/i;
 
 our $VZTMPL_EXT_RE_1 = qr/\.tar\.(gz|xz|zst)/i;
 
+our $SNIPPETS_EXT_RE_0 = qr/\.(?:yml|yaml)/i;
+
 our $BACKUP_EXT_RE_2 = 
qr/\.(tgz|(?:tar|vma)(?:\.(${\PVE::Storage::Plugin::COMPRESSOR_RE}))?)/;
 
 # FIXME remove with PVE 9.0, add versioned breaks for pve-manager
@@ -469,6 +471,15 @@ sub get_vztmpl_dir {
     return $plugin->get_subdir($scfg, 'vztmpl');
 }
 
+sub get_snippets_dir {
+    my ($cfg, $storeid) = @_;
+
+    my $scfg = storage_config($cfg, $storeid);
+    my $plugin = PVE::Storage::Plugin->lookup($scfg->{type});
+
+    return $plugin->get_subdir($scfg, 'snippets');
+}
+
 sub get_backup_dir {
     my ($cfg, $storeid) = @_;
 
-- 
2.39.2




--- End Message ---
_______________________________________________
pve-devel mailing list
pve-devel@lists.proxmox.com
https://lists.proxmox.com/cgi-bin/mailman/listinfo/pve-devel

Reply via email to