Fix IPv6 rule creation in chain traversal.

ldx · Mar 5, 2013 · 5130dfe · 5130dfe
1 parent 031fd22
commit 5130dfe
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/iptc/ip4tc.py b/iptc/ip4tc.py
@@ -1206,7 +1206,7 @@ def _get_rules(self):
         while entry:
             entries.append(entry)
             entry = self.table.next_rule(entry)
-        return [Rule(e, self) for e in entries]
+        return [self.table.create_rule(e, self) for e in entries]
 
     rules = property(_get_rules)
     """This is the list of rules currently in the chain."""
@@ -1493,3 +1493,6 @@ def flush(self):
             if not self.builtin_chain(chain):
                 chain.flush()
                 chain.delete()
+
+    def create_rule(self, entry=None, chain=None):
+        return Rule(entry, chain)
diff --git a/iptc/ip6tc.py b/iptc/ip6tc.py
@@ -602,3 +602,6 @@ def __init__(self, name, autocommit=True):
         self.name = name
         self.autocommit = autocommit
         self.refresh()
+
+    def create_rule(self, entry=None, chain=None):
+        return Rule6(entry, chain)