grbrute: fix event processing interuption by user

now when the user callback return False it stops grbrute

1 files changed, 11 insertions, 5 deletions

diff --git a/toys/pphidden_async.py b/toys/pphidden_async.py
index 229a181..1b6a576 100644
--- a/toys/pphidden_async.py
+++ b/toys/pphidden_async.py
@@ -27,10 +27,13 @@ class Pp_url:
         return res
 
 def cb_response(res):
-    print "[-] %s : %d" % (res.url, res.status_code)
+    global found
+    if not found:
+        print "[-] %s : %d" % (res.url, res.status_code)
     if res.status_code != 404:
-        print "[*] found: %s" % res.url
-        sys.exit(0)
+        found = res.url
+        return False
+    return True
 
 parser = argparse.ArgumentParser(description='pphidden_async',
                                  epilog="Example: %s 73 0 0" % sys.argv[0])
@@ -44,7 +47,10 @@ parser.add_argument('-v', action="store_true", dest="verbose", default=False,
                     help="verbose")
 args = parser.parse_args()                                               
 
+found = None
 url_iter = Pp_url(args.image, args.start_x, args.start_y)
 grbrute.Grbrute(url_iter, cb_response, verbose=args.verbose)
-print "[*] not found"
-sys.exit(1)
+if found is False:
+    print "[*] not found"
+    sys.exit(1)
+print "[*] found: %s" % found