Fixing eviltwin. Lots of changes.

wifite/tools/iptables.py

@@ -4,14 +4,17 @@
 import re
 
 from .dependency import Dependency
+from ..util.process import Process
 
 class Iptables(Dependency):
+
     process_name = 'iptables'
 
     dependency_required = False
     dependency_name = process_name
     dependency_url = 'apt-get install iptables'
 
+
     @classmethod
     def exists(cls):
         return Process.exists(cls.process_name)
@@ -19,6 +22,8 @@ class Iptables(Dependency):
 
     @classmethod
     def __exec(cls, args, expect_return_code=0):
+        # Helper method for executing iptables commands.
+
         if type(args) is str:
             args = args.split(' ')
 
@@ -30,12 +35,13 @@ class Iptables(Dependency):
             raise Exception('Error executing %s:\n%s\n%s' % (' '.join(command), pid.stdout(), pid.stderr()))
 
 
+    # -N, --new-chain <chain>
     @classmethod
     def new_chain(cls, chain_name, table):
-        command = ['-N', name, '-t', table]
-        cls.__exec(command)
-
+        args = ['-N', chain_name, '-t', table]
+        cls.__exec(args)
 
+    # -A, --append <chain> <rule-specification>
     @classmethod
     def append(cls, chain, table=None, rules=[]):
         args = []
@@ -45,3 +51,23 @@ class Iptables(Dependency):
         args.extend(rules)
         cls.__exec(args)
 
+
+    # -F, --flush <chain>
+    @classmethod
+    def flush(cls, table=None):
+        args = []
+        if table is not None:
+            args.extend(['-t', table])
+        args.append('-F')
+        cls.__exec(args)
+
+
+    # -X, --delete-chain <chain>
+    @classmethod
+    def delete_chain(cls, table=None):
+        args = []
+        if table is not None:
+            args.extend(['-t', table])
+        args.append('-X')
+        cls.__exec(args)
+