progress

lfi.py

@@ -129,10 +129,11 @@ def testPayload( url, payload ):
 
 def testURL( url ):
     time.sleep( 0.01 )
-    # print(url)
+    t_multiproc['n_current'] = t_multiproc['n_current'] + 1
+
     if _verbose <= 1:
         sys.stdout.write( 'progress: %d/%d\r' %  (t_multiproc['n_current'],t_multiproc['n_total']) )
-        t_multiproc['n_current'] = t_multiproc['n_current'] + 1
+        # t_multiproc['n_current'] = t_multiproc['n_current'] + 1
 
     pool = Pool( 10 )
     pool.map( partial(testPayload,url), t_payloads )
@@ -159,7 +160,7 @@ def realDoTest( t_params ):
 
     if _verbose <= 1:
         sys.stdout.write( 'progress: %d/%d\r' %  (t_multiproc['n_current'],t_multiproc['n_total']) )
-        t_multiproc['n_current'] = t_multiproc['n_current'] + 1
+        # t_multiproc['n_current'] = t_multiproc['n_current'] + 1
 
     t_urlparse = urllib.parse.urlparse(url)
     u = t_urlparse.scheme + '_' + t_urlparse.netloc
@@ -272,6 +273,15 @@ def realDoTest( t_params ):
         fp.close()
     else:
         t_urls.append( args.urls )
+else:
+    while True:
+        try:
+            url = input()
+        except EOFError:
+            break
+        else:
+            t_urls.append( url )
+
 n_urls = len(t_urls)
 if _verbose < 4:
     sys.stdout.write( '%s[+] %d urls found: %s%s\n' % (fg('green'),n_urls,args.urls,attr(0)) )

rce.py

@@ -127,10 +127,11 @@ def testPayload( url, payload ):
 
 def testURL( url ):
     time.sleep( 0.01 )
+    t_multiproc['n_current'] = t_multiproc['n_current'] + 1
 
     if _verbose <= 1:
         sys.stdout.write( 'progress: %d/%d\r' %  (t_multiproc['n_current'],t_multiproc['n_total']) )
-        t_multiproc['n_current'] = t_multiproc['n_current'] + 1
+        # t_multiproc['n_current'] = t_multiproc['n_current'] + 1
 
     pool = Pool( 10 )
     pool.map( partial(testPayload,url), t_payloads )
@@ -153,7 +154,7 @@ def realDoTest( t_params ):
 
     if _verbose <= 1:
         sys.stdout.write( 'progress: %d/%d\r' %  (t_multiproc['n_current'],t_multiproc['n_total']) )
-        t_multiproc['n_current'] = t_multiproc['n_current'] + 1
+        # t_multiproc['n_current'] = t_multiproc['n_current'] + 1
 
     t_urlparse = urllib.parse.urlparse(url)
     u = t_urlparse.scheme + '_' + t_urlparse.netloc
@@ -266,6 +267,15 @@ def realDoTest( t_params ):
         fp.close()
     else:
         t_urls.append( args.urls )
+else:
+    while True:
+        try:
+            url = input()
+        except EOFError:
+            break
+        else:
+            t_urls.append( url )
+
 n_urls = len(t_urls)
 if _verbose < 4:
     sys.stdout.write( '%s[+] %d urls found: %s%s\n' % (fg('green'),n_urls,args.urls,attr(0)) )

xss.py

@@ -73,8 +73,8 @@ def testParams( t_urlparse, payload ):
             url = urllib.parse.urlunparse(t_urlparse)
             doTest( url )
             # convert get params to post
-            t_urlparse = t_urlparse._replace(query='')
-            url = urllib.parse.urlunparse(t_urlparse)
+            # t_urlparse = t_urlparse._replace(query='')
+            # url = urllib.parse.urlunparse(t_urlparse)
             # doTest( url, 'POST', new_query )
 
 
@@ -115,10 +115,11 @@ def testPayload( url, payload ):
 
 def testURL( url ):
     time.sleep( 0.01 )
+    t_multiproc['n_current'] = t_multiproc['n_current'] + 1
 
     if _verbose <= 1:
         sys.stdout.write( 'progress: %d/%d\r' %  (t_multiproc['n_current'],t_multiproc['n_total']) )
-        t_multiproc['n_current'] = t_multiproc['n_current'] + 1
+        # t_multiproc['n_current'] = t_multiproc['n_current'] + 1
 
     pool = Pool( 10 )
     pool.map( partial(testPayload,url), t_payloads )
@@ -129,6 +130,7 @@ def testURL( url ):
 def doTest( url, method='GET', post_params='' ):
 
     realDoTest( [url,method,post_params] )
+
     # t_realdotest.append( [url,method,post_params] )
     return
 
@@ -142,7 +144,7 @@ def realDoTest( t_params ):
 
     if _verbose <= 1:
         sys.stdout.write( 'progress: %d/%d\r' %  (t_multiproc['n_current'],t_multiproc['n_total']) )
-        t_multiproc['n_current'] = t_multiproc['n_current'] + 1
+        # t_multiproc['n_current'] = t_multiproc['n_current'] + 1
 
     t_urlparse = urllib.parse.urlparse( url )
     t_params = [ method, url, post_params, _cookies, t_urlparse.netloc ]
@@ -244,6 +246,15 @@ def realDoTest( t_params ):
         fp.close()
     else:
         t_urls.append( args.urls )
+else:
+    while True:
+        try:
+            url = input()
+        except EOFError:
+            break
+        else:
+            t_urls.append( url )
+
 n_urls = len(t_urls)
 if _verbose < 4:
     sys.stdout.write( '%s[+] %d urls found: %s%s\n' % (fg('green'),n_urls,args.urls,attr(0)) )