delete trailing white space from 'ipset save' output.
authorDietmar Maurer <dietmar@proxmox.com>
Tue, 1 Apr 2014 06:28:46 +0000 (08:28 +0200)
committerDietmar Maurer <dietmar@proxmox.com>
Tue, 1 Apr 2014 06:28:46 +0000 (08:28 +0200)
Also improve verbose output.

src/PVE/Firewall.pm

index 45f6a5d..54235fb 100644 (file)
@@ -892,8 +892,10 @@ sub ipset_get_chains {
 
        return if $line =~ m/^#/;
        return if $line =~ m/^\s*$/;
 
        return if $line =~ m/^#/;
        return if $line =~ m/^\s*$/;
-       if ($line =~ m/^(\S+)\s(\S+)\s(\S+)/) {
-           push @{$chains->{$2}}, $line;
+       if ($line =~ m/^(?:\S+)\s(\S+)\s(?:\S+).*/) {
+           my $chain = $1;
+           $line =~ s/\s+$//; # delete trailing white space
+           push @{$chains->{$chain}}, $line;
        } else {
            # simply ignore the rest
            return;
        } else {
            # simply ignore the rest
            return;
@@ -2404,15 +2406,23 @@ sub apply_ruleset {
 
     update_nf_conntrack_max($hostfw_conf);
 
 
     update_nf_conntrack_max($hostfw_conf);
 
-    my ($ipset_create_cmdlist, $ipset_delete_cmdlist) = get_ipset_cmdlist($ipset_ruleset, undef, $verbose);
-
-    my $cmdlist = get_ruleset_cmdlist($ruleset, $verbose);
+    my ($ipset_create_cmdlist, $ipset_delete_cmdlist, $ipset_changes) = 
+       get_ipset_cmdlist($ipset_ruleset, undef, $verbose);
 
 
-    print $ipset_create_cmdlist if $verbose;
+    my ($cmdlist, $changes) = get_ruleset_cmdlist($ruleset, $verbose);
 
 
-    print $ipset_delete_cmdlist if $verbose;
+    if ($verbose) {
+       if ($ipset_changes) {
+           print "ipset changes:\n";
+           print $ipset_create_cmdlist if $ipset_create_cmdlist;
+           print $ipset_delete_cmdlist if $ipset_delete_cmdlist;
+       }
 
 
-    print $cmdlist if $verbose;
+       if ($changes) {
+           print "iptables changes:\n";
+           print $cmdlist;
+       }
+    }
 
     ipset_restore_cmdlist($ipset_create_cmdlist);
 
 
     ipset_restore_cmdlist($ipset_create_cmdlist);
 
@@ -2422,7 +2432,7 @@ sub apply_ruleset {
 
     # test: re-read status and check if everything is up to date
     my $active_chains = iptables_get_chains();
 
     # test: re-read status and check if everything is up to date
     my $active_chains = iptables_get_chains();
-    my $statushash = get_ruleset_status($ruleset, $active_chains, \&iptables_chain_digest, $verbose);
+    my $statushash = get_ruleset_status($ruleset, $active_chains, \&iptables_chain_digest, 0);
 
     my $errors;
     foreach my $chain (sort keys %$ruleset) {
 
     my $errors;
     foreach my $chain (sort keys %$ruleset) {