mirrors/smuggler
1
0
Fork 0
mirror of https://github.com/defparam/smuggler synced 2024-11-14 16:05:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simplify process_uri parsing logic with urlparse

Browse Source
This commit is contained in:
engn33r 2021-04-28 11:01:48 -07:00
parent 7084d63497
commit bd21724ffc
1 changed files with 10 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
smuggler.py
View File

@ -35,6 +35,7 @@ from datetime import datetime
from lib.Payload import Payload, Chunked, EndChunk from lib.Payload import Payload, Chunked, EndChunk
from lib.EasySSL import EasySSL from lib.EasySSL import EasySSL
from lib.colorama import Fore, Style from lib.colorama import Fore, Style
from urllib.parse import urlparse
class Desyncr(): class Desyncr():
def __init__(self, configfile, smhost, smport=443, url="", method="POST", endpoint="/", SSLFlag=False, logh=None, smargs=None): def __init__(self, configfile, smhost, smport=443, url="", method="POST", endpoint="/", SSLFlag=False, logh=None, smargs=None):
@ -317,34 +318,22 @@ class Desyncr():
return False return False
def process_uri(uri): def process_uri(uri):
#remove shouldering white spaces and go lowercase u = urlparse(uri)
uri = uri.strip().lower()
#if it starts with https:// then strip it if u.scheme == "https":
if ((len(uri) > 8) and (uri[0:8] == "https://")):
uri = uri[8:]
ssl_flag = True ssl_flag = True
std_port = 443 std_port = 443
elif ((len(uri) > 7) and (uri[0:7] == "http://")): elif u.scheme == "http":
uri = uri[7:]
ssl_flag = False ssl_flag = False
std_port = 80 std_port = 80
else: else:
print_info("Error malformed URL not supported: %s" % (Fore.CYAN + uri)) print_info("Error malformed URL not supported: %s" % (Fore.CYAN + uri))
exit(1) exit(1)
#check for any forward slashes and filter only domain portion if u.port:
uri_tokenized = uri.split("/") return (u.hostname, u.port, u.path, ssl_flag)
uri = uri_tokenized[0] else:
smendpoint = '/' + '/'.join(uri_tokenized[1:]) return (u.hostname, std_port, u.path, ssl_flag)
#check for any port designators
uri = uri.split(":")
if len(uri) > 1:
return (uri[0], int(uri[1]), smendpoint, ssl_flag)
return (uri[0], std_port, smendpoint, ssl_flag)
def CF(text): def CF(text):
global NOCOLOR global NOCOLOR
@ -437,12 +426,9 @@ if __name__ == "__main__":
if server[0].lower().strip()[0:4] != "http": if server[0].lower().strip()[0:4] != "http":
server[0] = "https://" + server[0] server[0] = "https://" + server[0]
params = process_uri(server[0])
host, port, endpoint, SSLFlagval = process_uri(server[0])
method = server[1].upper() method = server[1].upper()
host = params[0]
port = params[1]
endpoint = params[2]
SSLFlagval = params[3]
configfile = Args.configfile configfile = Args.configfile
print_info("URL : %s"%(Fore.CYAN + server[0]), FileHandle) print_info("URL : %s"%(Fore.CYAN + server[0]), FileHandle)