From 15890dc952438821dd6d8cdd3eb69ec85e8ec381 Mon Sep 17 00:00:00 2001
From: Laurent Ghigonis <laurent@p1sec.com>
Date: Wed, 17 Apr 2013 02:36:38 +0200
Subject: pphidden_async: fix syntax error and use argparse

---
 toys/pphidden_async.py | 27 ++++++++++++++++-----------
 1 file changed, 16 insertions(+), 11 deletions(-)

(limited to 'toys')

diff --git a/toys/pphidden_async.py b/toys/pphidden_async.py
index 24dec8d..3f00b0a 100644
--- a/toys/pphidden_async.py
+++ b/toys/pphidden_async.py
@@ -3,6 +3,7 @@ import grequests
 from collections import deque
 import time
 import sys
+import argparse
 import warnings
 warnings.warn("XXX WORK IN PROGRESS")
 warnings.warn("XXX don't excpect this to work !")
@@ -38,11 +39,11 @@ class Brute:
         self.ips = ips # XXX TODO
         self.sessions = deque()
         self.ongoing = 0
-        for i in range(nb_sessions)
+        for i in range(nb_sessions):
             self.sessions.append(SessionQueue())
         self._send()
 
-    def _send(self)
+    def _send(self):
         while self.ongoing < self.nb_sessions * self.req_per_session:
             try:
                 url = self.url_iter.next()
@@ -53,9 +54,6 @@ class Brute:
                 s.add(url)
                 self.ongoing += 1
 
-    def _add_url(self, url)
-        self.
-
     def _cb_response(self, res):
         print "response: %s" % res
         self.ongoing -= 1
@@ -84,13 +82,20 @@ class Pp_url:
             raise StopIteration
         return res
 
-image = int(sys.argv[1])
-x = int(sys.argv[2])
-y = int(sys.argv[3])
-
-url_iter = Pp_url(image, x, y)
-b = Brute(ips=None, conn_per_ip=3, url_iter)
+parser = argparse.ArgumentParser(description='pphidden',
+                                 epilog="Example: %s 73 0 0" % sys.argv[0])
+parser.add_argument('image', action="store", type=int,
+                    help="image number")
+parser.add_argument('start_x', action="store", type=int,
+                    help="Start at coordinate X=")
+parser.add_argument('start_y', action="store", type=int,
+                    help="Start at coordinate Y=")
+parser.add_argument('-v', action="store_true", dest="verbose", default=False,
+                    help="verbose")
+args = parser.parse_args()                                               
 
+url_iter = Pp_url(args.image, args.start_x, args.start_y)
+b = Brute(url_iter)
 if b.found:
     print "[*] found: %s" % b.url
 else:
-- 
cgit v1.2.3-59-g8ed1b