* move count_iptables_rule to Iptables::Mgr and update it's usage

1 files changed, 11 insertions, 1 deletions

diff --git a/lib/Vyatta/IpTables/Mgr.pm b/lib/Vyatta/IpTables/Mgr.pm
index 9247a44..ff468cb 100755
--- a/lib/Vyatta/IpTables/Mgr.pm
+++ b/lib/Vyatta/IpTables/Mgr.pm
@@ -29,7 +29,7 @@ use warnings;
 
 use base 'Exporter';
 our @EXPORT = qw(ipt_find_chain_rule ipt_enable_conntrack
-                 ipt_disable_conntrack);
+                 ipt_disable_conntrack count_iptables_rules);
 
 
 sub ipt_find_chain_rule {
@@ -104,4 +104,14 @@ sub ipt_disable_conntrack {
     return 0;
 }
 
+sub count_iptables_rules {
+    my ($command, $table, $chain) = @_;
+    my @lines = `sudo $command -t $table -L $chain -n --line`;
+    my $cnt = 0;
+    foreach my $line (@lines) {
+      $cnt++ if $line =~ /^\d/;
+    }
+    return $cnt;
+}
+
 1;