From 0bfc82c35674dec5bf288c45da60f34968e7020c Mon Sep 17 00:00:00 2001
From: derv82 <derv82@gmail.com>
Date: Sat, 10 Mar 2018 15:40:26 -0500
Subject: [PATCH] BUGFIX: Detect when Reaver/Bully is successful.

Reported in #60

Also removed PIN attack.
---
 py/AttackWPS.py | 20 +++-----------------
 1 file changed, 3 insertions(+), 17 deletions(-)

diff --git a/py/AttackWPS.py b/py/AttackWPS.py
index 8c96d53..effc289 100644
--- a/py/AttackWPS.py
+++ b/py/AttackWPS.py
@@ -25,11 +25,12 @@ class AttackWPS(Attack):
         # Pixie-Dust attack
         if Configuration.use_bully:
             # Bully: Pixie-dust
-            bully = Bully(self.target, pixie=True)
+            bully = Bully(self.target)
             bully.run()
             bully.stop()
             if bully.crack_result is not None:
                 self.crack_result = bully.crack_result
+                self.success = True
                 return True
         else:
             reaver = Reaver(self.target)
@@ -37,6 +38,7 @@ class AttackWPS(Attack):
                 # Reaver: Pixie-dust
                 reaver = Reaver(self.target)
                 if reaver.run_pixiedust_attack():
+                    self.success = True
                     return True
             else:
                 Color.pl("{!} {R}your version of 'reaver' does not support the {O}WPS pixie-dust attack{W}")
@@ -45,20 +47,4 @@ class AttackWPS(Attack):
             Color.pl('\r{!} {O}--pixie{R} set, ignoring WPS-PIN attack{W}')
             return False
 
-        ###################
-        # PIN attack
-        if Configuration.use_bully:
-            # Bully: PIN guessing
-            bully = Bully(self.target, pixie=False)
-            bully.run()
-            bully.stop()
-            if bully.crack_result is not None:
-                self.crack_result = bully.crack_result
-                return True
-        else:
-            # Reaver: PIN guessing
-            reaver = Reaver(self.target)
-            if reaver.run_wps_pin_attack():
-                return True
-
         return False