Initial commit, basic helper classes created

py/Arguments.py

@@ -0,0 +1,98 @@
+#!/usr/bin/python
+
+import argparse
+
+class Arguments(object):
+    def __init__(self):
+        self.args = self.get_arguments()
+
+    def get_arguments(self):
+        description = 'Wrapper script around aircrack-ng and reaver'
+        description += ' https://github.com/derv82/wifite'
+        parser = argparse.ArgumentParser(
+            description=description)
+
+        # Global variables
+        glob = parser.add_argument_group('SETTINGS')
+        glob.add_argument('-i',
+            action='store',
+            dest='interface',
+            metavar='interface',
+            type=str,
+            help='Wireless interface to use (default: ask)')
+        glob.add_argument('-c',
+            action='store',
+            dest='channel',
+            metavar='channel',
+            type=int,
+            help='Wireless channel to scan (default: all channels)')
+
+        # WEP
+        wep = parser.add_argument_group('WEP-RELATED')
+        wep.add_argument('--wep',
+            action='store_true',
+            dest='wep_only',
+            help='Only target WEP-encrypted networks (ignores WPA)')
+
+        # WPA
+        wep = parser.add_argument_group('WPA-RELATED')
+        wep.add_argument('--wpa',
+            action='store_true',
+            dest='wpa_only',
+            help='Only target WPA-encrypted networks (ignores WEP)')
+
+        # WPS
+        wep = parser.add_argument_group('WPS-RELATED')
+        wep.add_argument('--wps',
+            action='store_true',
+            dest='wps_only',
+            help='Only target WPS-encrypted networks (ignores WEP/nonWPS)')
+        wep.add_argument('--pixie',
+            action='store_true',
+            dest='pixie_only',
+            help='Only use the WPS Pixie-Dust attack (do not crack PINs)')
+        
+        # Cracking
+        crack = parser.add_argument_group('CRACKING')
+        crack.add_argument('--cracked',
+            action='store_true',
+            dest='cracked',
+            help='Display previously-cracked access points')
+        crack.add_argument('--check',
+            action='store',
+            metavar='[file]',
+            dest='check',
+            help='Check a .cap file for WPA handshakes')
+        crack.add_argument('--crack-wpa',
+            action='store',
+            type=str,
+            dest='crackwpa',
+            metavar='[file]',
+            help='Crack a .cap file containing a WPA handshake')
+        crack.add_argument('--crack-wep',
+            action='store',
+            type=str,
+            dest='crackwep',
+            metavar='[file]',
+            help='Crack a .cap file containing WEP IVS')
+        crack.add_argument('--dict',
+            action='store',
+            type=str,
+            dest='wordlist',
+            metavar='[file]',
+            help='Dictionary/wordlist to use for cracking')
+        
+        # Misc
+        commands = parser.add_argument_group('FUNCTIONS')
+        commands.add_argument('--update',
+            action='store_true',
+            dest='update',
+            help='Update to latest version of Wifite (on github)')
+
+        return parser.parse_args()
+
+if __name__ == '__main__':
+    a = Arguments()
+    args = a.args
+    print args
+