WPS PixieDust attack support

Fixed encryption filtering.
More WPS-specific configurations.
Various fixes.

Wifite.py

@@ -1,37 +1,63 @@
 #!/usr/bin/python
 
+from py.Configuration import Configuration
 from py.Scanner import Scanner
 from py.Color import Color
 from py.AttackWEP import AttackWEP
 from py.AttackWPA import AttackWPA
+from py.AttackWPS import AttackWPS
 
 class Wifite(object):
     def __init__(self):
         pass
 
     def run(self):
+        '''
+            Main program.
+            1) Scans for targets, asks user to select targets
+            2) Attacks each target
+        '''
         s = Scanner()
         targets = s.select_targets()
         for t in targets:
             Color.pl('{+} starting attacks against {C}%s{W} ({C}%s{W})'
                 % (t.bssid, t.essid))
-            # TODO: Check if Configuration says to attack certain encryptions.
-            if 'WEP' in t.encryption:
+            if 'WEP' in t.encryption: # TODO: and configuration.use_wep
                 attack = AttackWEP(t)
+                attack.run()
             elif 'WPA' in t.encryption:
-                # TODO: Check if WPS, attack WPS
-                attack = AttackWPA(t)
-            attack.run()
+                if t.wps: # TODO: and Configuration.use_wps
+                    attack = AttackWPS(t)
+                    if attack.run():
+                        # We cracked it.
+                        break
+                    else: # TODO: and Configuration.use_wpa
+                        # WPS failed, try WPA handshake.
+                        attack = AttackWPA(t)
+                        attack.run()
+                else: # TODO: and Configuration.use_wpa
+                    # Not using WPS, try WPA handshake.
+                    attack = AttackWPA(t)
+                    attack.run()
+            else:
+                # TODO: Error: Can't attack - encryption not WEP or WPA
+                pass
+
+            # TODO: if attack.successful:
+            # TODO: Save attack.crack_result
         pass
 
+
 if __name__ == '__main__':
     w = Wifite()
     try:
         w.run()
     except Exception, e:
         Color.pl('\n{!} {R}Error:{O} %s{W}' % str(e))
-        #from traceback import format_exc
-        #format_exc().replace('\n', '\n    ')
+        from traceback import format_exc
+        print '\n    '
+        print format_exc().replace('\n', '\n    ')
     except KeyboardInterrupt:
         Color.pl('\n{!} {O}interrupted{W}')
+    Configuration.exit_gracefully(0)