From 141934a7b12756ad9dff78955996cc02c36b493a Mon Sep 17 00:00:00 2001
From: derv82 <derv82@gmail.com>
Date: Thu, 23 Aug 2018 19:05:20 -0700
Subject: [PATCH] Bully: Stop if AP becomes locked

---
 wifite/tools/bully.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/wifite/tools/bully.py b/wifite/tools/bully.py
index 8796f0e..fd143fe 100755
--- a/wifite/tools/bully.py
+++ b/wifite/tools/bully.py
@@ -105,7 +105,7 @@ class Bully(Attack, Dependency):
             # Update status
             self.pattack(self.get_status())
 
-                # Thresholds only apply to Pixie-Dust
+            # Thresholds only apply to Pixie-Dust
             if self.pixie_dust:
                 # Check if entire attack timed out.
                 if self.running_time() > Configuration.wps_pixie_timeout:
@@ -127,6 +127,12 @@ class Bully(Attack, Dependency):
                         Configuration.wps_fail_threshold), newline=True)
                     self.stop()
                     return
+            else:
+                if self.locked and not Configuration.wps_ignore_lock:
+                    self.pattack('{R}Failed: {O}AP became {R}Locked{O}', newline=True)
+                    self.stop()
+                    return
+
 
             time.sleep(0.5)