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

Merge pull request #26 from engn33r/master

Browse Source 
Simplify process_uri parsing logic with urlparse
This commit is contained in:
Evan 2021-04-28 18:24:28 -04:00 committed by GitHub
commit 2be871e615
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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)