Trying to add some error checking to matches and targets
[sfa.git] / sfatables / exec.py
index fcff3aa..ef0933d 100644 (file)
@@ -13,13 +13,13 @@ from sfatables.commands.List import *
 from sfatables.xmlrule import *
 
 class SFATablesRules:
-    sorted_rule_list = None
+    sorted_rule_list = []
 
     def __init__(self, chain_name):
         chain_dir_path = "%s/%s"%(sfatables_config,chain_name)
         rule_list = List().get_rule_list(chain_dir_path)
         for rule_number in rule_list:
-            self.sorted_rule_list.append(XMLRule(rule_number))
+            self.sorted_rule_list.append(XMLRule(chain_name, rule_number))
         return
 
     def apply(self, rspec):
@@ -28,3 +28,15 @@ class SFATablesRules:
             intermediate_rspec  = rule.apply(intermediate_rspec)
 
         return intermediate_rspec
+
+def main():
+    incoming = SFATablesRules('INCOMING')
+    outgoing = SFATablesRules('OUTGOING')
+
+    print "%d rules loaded for INCOMING chain\n"%len(incoming.sorted_rule_list)
+    print "%d rules loaded for OUTGOING chain\n"%len(outgoing.sorted_rule_list)
+
+    return
+
+if __name__=="__main__":
+    main()