]> git.proxmox.com Git - pve-firewall.git/blobdiff - src/PVE/API2/Firewall/IPSet.pm
rename save_rules to save_ipset
[pve-firewall.git] / src / PVE / API2 / Firewall / IPSet.pm
index ec6e783d1ac9c6f127c8a09e811413c7c38f4924..45851d442dc7fed406b067938f0650854a0a2532 100644 (file)
@@ -14,10 +14,7 @@ my $api_properties = {
        description => "Network/IP specification in CIDR format.",
        type => 'string', format => 'IPv4orCIDR',
     },
-    name => {
-       description => "IP set name.",
-       type => 'string',
-    },
+    name => get_standard_option('ipset-name'),
     comment => {
        type => 'string',
        optional => 1,
@@ -32,11 +29,9 @@ sub load_config {
     my ($class, $param) = @_;
 
     die "implement this in subclass";
-
-    #return ($fw_conf, $rules);
 }
 
-sub save_rules {
+sub save_ipset {
     my ($class, $param, $fw_conf, $rules) = @_;
 
     die "implement this in subclass";
@@ -89,7 +84,8 @@ sub register_get_ipset {
                    nomatch => {
                        type => 'boolean',
                        optional => 1,
-                   },                  
+                   },
+                   digest => get_standard_option('pve-config-digest', { optional => 0} ),      
                },
            },
            links => [ { rel => 'child', href => "{cidr}" } ],
@@ -99,11 +95,11 @@ sub register_get_ipset {
 
            my ($fw_conf, $ipset) = $class->load_config($param);
 
-           return $ipset;
+           return PVE::Firewall::copy_list_with_digest($ipset);
        }});
 }
 
-sub register_add_ip {
+sub register_create_ip {
     my ($class) = @_;
 
     my $properties = $class->additional_parameters();
@@ -112,9 +108,9 @@ sub register_add_ip {
     $properties->{cidr} = $api_properties->{cidr};
     $properties->{nomatch} = $api_properties->{nomatch};
     $properties->{comment} = $api_properties->{comment};
-    
+
     $class->register_method({
-       name => 'add_ip',
+       name => 'create_ip',
        path => '',
        method => 'POST',
        description => "Add IP or Network to IPSet.",
@@ -148,7 +144,7 @@ sub register_add_ip {
        }});
 }
 
-sub register_remove_ip {
+sub register_read_ip {
     my ($class) = @_;
 
     my $properties = $class->additional_parameters();
@@ -156,6 +152,87 @@ sub register_remove_ip {
     $properties->{name} = $api_properties->{name};
     $properties->{cidr} = $api_properties->{cidr};
     
+    $class->register_method({
+       name => 'read_ip',
+       path => '{cidr}',
+       method => 'GET',
+       description => "Read IP or Network settings from IPSet.",
+       protected => 1,
+       parameters => {
+           additionalProperties => 0,
+           properties => $properties,
+       },
+       returns => { type => "object" },
+       code => sub {
+           my ($param) = @_;
+
+           my ($fw_conf, $ipset) = $class->load_config($param);
+
+           my $list = PVE::Firewall::copy_list_with_digest($ipset);
+
+           foreach my $entry (@$list) {
+               if ($entry->{cidr} eq $param->{cidr}) {
+                   return $entry;
+               }
+           }
+
+           raise_param_exc({ cidr => "no such IP/Network" });
+       }});
+}
+
+sub register_update_ip {
+    my ($class) = @_;
+
+    my $properties = $class->additional_parameters();
+
+    $properties->{name} = $api_properties->{name};
+    $properties->{cidr} = $api_properties->{cidr};
+    $properties->{nomatch} = $api_properties->{nomatch};
+    $properties->{comment} = $api_properties->{comment};
+    $properties->{digest} = get_standard_option('pve-config-digest');
+
+    $class->register_method({
+       name => 'update_ip',
+       path => '{cidr}',
+       method => 'PUT',
+       description => "Update IP or Network settings",
+       protected => 1,
+       parameters => {
+           additionalProperties => 0,
+           properties => $properties,
+       },
+       returns => { type => "null" },
+       code => sub {
+           my ($param) = @_;
+
+           my ($fw_conf, $ipset) = $class->load_config($param);
+
+           my (undef, $digest) = PVE::Firewall::copy_list_with_digest($ipset);
+           PVE::Tools::assert_if_modified($digest, $param->{digest});
+           warn "TEST:$digest:$param->{digest}:\n";
+
+           foreach my $entry (@$ipset) {
+               if($entry->{cidr} eq $param->{cidr}) {
+                   $entry->{nomatch} = $param->{nomatch};
+                   $entry->{comment} = $param->{comment};
+                   $class->save_ipset($param, $fw_conf, $ipset);
+                   return;
+               }
+           }
+
+           raise_param_exc({ cidr => "no such IP/Network" });
+       }});
+}
+
+sub register_delete_ip {
+    my ($class) = @_;
+
+    my $properties = $class->additional_parameters();
+
+    $properties->{name} = $api_properties->{name};
+    $properties->{cidr} = $api_properties->{cidr};
+    $properties->{digest} = get_standard_option('pve-config-digest');
+
     $class->register_method({
        name => 'remove_ip',
        path => '{cidr}',
@@ -172,6 +249,9 @@ sub register_remove_ip {
 
            my ($fw_conf, $ipset) = $class->load_config($param);
 
+           my (undef, $digest) = PVE::Firewall::copy_list_with_digest($ipset);
+           PVE::Tools::assert_if_modified($digest, $param->{digest});
+
            my $new = [];
    
            foreach my $entry (@$ipset) {
@@ -188,8 +268,10 @@ sub register_handlers {
     my ($class) = @_;
 
     $class->register_get_ipset();
-    $class->register_add_ip();
-    $class->register_remove_ip();
+    $class->register_create_ip();
+    $class->register_read_ip();
+    $class->register_update_ip();
+    $class->register_delete_ip();
 }
 
 package PVE::API2::Firewall::ClusterIPset;
@@ -218,4 +300,202 @@ sub save_ipset {
 
 __PACKAGE__->register_handlers();
 
+package PVE::API2::Firewall::BaseIPSetList;
+
+use strict;
+use warnings;
+use PVE::JSONSchema qw(get_standard_option);
+use PVE::Exception qw(raise_param_exc);
+use PVE::Firewall;
+
+use base qw(PVE::RESTHandler);
+
+my $get_ipset_list = sub {
+    my ($fw_conf) = @_;
+
+    my $res = [];
+    foreach my $name (keys %{$fw_conf->{ipset}}) {
+       my $data = { 
+           name => $name,
+       };
+       if (my $comment = $fw_conf->{ipset_comments}->{$name}) {
+           $data->{comment} = $comment;
+       }
+       push @$res, $data;
+    }
+
+    my ($list, $digest) = PVE::Firewall::copy_list_with_digest($res);
+
+    return wantarray ? ($list, $digest) : $list;
+};
+
+sub register_index {
+    my ($class) = @_;
+
+    $class->register_method({
+       name => 'ipset_index',
+       path => '',
+       method => 'GET',
+       description => "List IPSets",
+       parameters => {
+           additionalProperties => 0,
+       },
+       returns => {
+           type => 'array',
+           items => {
+               type => "object",
+               properties => { 
+                   name => get_standard_option('ipset-name'),
+                   digest => get_standard_option('pve-config-digest', { optional => 0} ),
+                   comment => { 
+                       type => 'string',
+                       optional => 1,
+                   }
+               },
+           },
+           links => [ { rel => 'child', href => "{name}" } ],
+       },
+       code => sub {
+           my ($param) = @_;
+           
+           my $fw_conf = $class->load_config();
+
+           return &$get_ipset_list($fw_conf); 
+       }});
+}
+
+sub register_create {
+    my ($class) = @_;
+
+    $class->register_method({
+       name => 'create_ipset',
+       path => '',
+       method => 'POST',
+       description => "Create new IPSet",
+       protected => 1,
+       parameters => {
+           additionalProperties => 0,
+           properties => { 
+               name => get_standard_option('ipset-name'),
+               comment => {
+                   type => 'string',
+                   optional => 1,
+               },
+               rename => get_standard_option('ipset-name', {
+                   description => "Rename an existing IPSet. You can set 'rename' to the same value as 'name' to update the 'comment' of an existing IPSet.",
+                   optional => 1,
+               }),
+               digest => get_standard_option('pve-config-digest'),
+           }
+       },
+       returns => { type => 'null' },
+       code => sub {
+           my ($param) = @_;
+           
+           my $fw_conf = $class->load_config();
+
+           if ($param->{rename}) {
+               my (undef, $digest) = &$get_ipset_list($fw_conf);
+               PVE::Tools::assert_if_modified($digest, $param->{digest});
+
+               raise_param_exc({ name => "IPSet '$param->{rename}' does not exists" }) 
+                   if !$fw_conf->{ipset}->{$param->{rename}};
+
+               my $data = delete $fw_conf->{ipset}->{$param->{rename}};
+               $fw_conf->{ipset}->{$param->{name}} = $data;
+               if (my $comment = delete $fw_conf->{ipset_comments}->{$param->{rename}}) {
+                   $fw_conf->{ipset_comments}->{$param->{name}} = $comment;
+               }
+               $fw_conf->{ipset_comments}->{$param->{name}} = $param->{comment} if defined($param->{comment});
+           } else { 
+               foreach my $name (keys %{$fw_conf->{ipset}}) {
+                   raise_param_exc({ name => "IPSet '$name' already exists" }) 
+                       if $name eq $param->{name};
+               }
+
+               $fw_conf->{ipset}->{$param->{name}} = [];
+               $fw_conf->{ipset_comments}->{$param->{name}} = $param->{comment} if defined($param->{comment});
+           }
+
+           $class->save_config($fw_conf);
+
+           return undef;
+       }});
+}
+
+sub register_delete {
+    my ($class) = @_;
+
+    $class->register_method({
+       name => 'delete_ipset',
+       path => '{name}',
+       method => 'DELETE',
+       description => "Delete IPSet",
+       protected => 1,
+       parameters => {
+           additionalProperties => 0,
+           properties => { 
+               name => get_standard_option('ipset-name'),
+               digest => get_standard_option('pve-config-digest'),
+           },
+       },
+       returns => { type => 'null' },
+       code => sub {
+           my ($param) = @_;
+           
+           my $fw_conf = $class->load_config();
+
+           return undef if !$fw_conf->{ipset}->{$param->{name}};
+
+           my (undef, $digest) = &$get_ipset_list($fw_conf);
+           PVE::Tools::assert_if_modified($digest, $param->{digest});
+
+           die "IPSet '$param->{name}' is not empty\n" 
+               if scalar(@{$fw_conf->{ipset}->{$param->{name}}});
+
+           delete $fw_conf->{ipset}->{$param->{name}};
+
+           $class->save_config($fw_conf);
+
+           return undef;
+       }});
+}
+
+sub register_handlers {
+    my ($class) = @_;
+
+    $class->register_index();
+    $class->register_create();
+    $class->register_delete();
+}
+
+package PVE::API2::Firewall::ClusterIPSetList;
+
+use strict;
+use warnings;
+use PVE::Firewall;
+
+use base qw(PVE::API2::Firewall::BaseIPSetList);
+
+sub load_config {
+    my ($class) = @_;
+    return PVE::Firewall::load_clusterfw_conf();
+}
+
+sub save_config {
+    my ($class, $fw_conf) = @_;
+
+    PVE::Firewall::save_clusterfw_conf($fw_conf);
+}
+
+__PACKAGE__->register_handlers();
+
+__PACKAGE__->register_method ({
+    subclass => "PVE::API2::Firewall::ClusterIPset",  
+    path => '{name}',
+    # set fragment delimiter (no subdirs) - we need that, because CIDR address contain a slash '/' 
+    fragmentDelimiter => '', 
+});
+
 1;