Converted Tab to Spaces

core/requester.py

@@ -1,12 +1,19 @@
 import urllib3
 import requests
+from core.colors import bad
 
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
+# Added better error handling.
+# Added verbose options.
 
 def requester(url, scheme, headers, origin):
     headers['Origin'] = scheme + origin
+    try:
         response = requests.get(url, headers=headers, verify=False).headers
         for key, value in response.items():
             if key.lower() == 'access-control-allow-origin':
                 return response
+    except requests.exceptions.RequestException as e:
+        if 'Failed to establish a new connection' in str(e):
+            print ( ' ' + bad + ' ' + url + ' seems to be down')

core/utils.py

@@ -2,6 +2,7 @@ import os
 import tld
 import json
 import tempfile
+import re
 
 
 def host(string):
@@ -34,6 +35,15 @@ def create_url_list(target_url, inp_file):
         urls.append(target_url)
     return urls
 
+def create_stdin_list(target_url, inp_file):
+    urls = []
+    if inp_file:
+        for line in inp_file.readlines():
+            if line.startswith(('http://', 'https://')):
+                urls.append(line.rstrip('\n'))
+    if target_url and target_url.startswith(('http://', 'https://')):
+        urls.append(target_url)
+    return urls
 
 def prompt(default=None):
     editor = 'nano'

corsy.py

@@ -6,7 +6,7 @@ import json
 import argparse
 
 from core.tests import active_tests
-from core.utils import host, prompt, format_result, create_url_list, create_stdin_list
+from core.utils import host, prompt, format_result, create_url_list, create_stdin_list, extractHeaders
 from core.colors import bad, end, red, run, good, grey, green, white, yellow