Roman/HackingScripts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

template update

Browse Source
This commit is contained in:
Roman Hergenreder 2023-12-18 00:35:23 +01:00
parent 6007c50279
commit e4132c3468
2 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
template.py
View File

@ -1,23 +1,36 @@
#!/usr/bin/env python #!/usr/bin/env python
import sys import sys
import json
import urllib.parse
def generate_template(base_url, features): def generate_template(base_url, features):
variables = {
"IP_ADDRESS": "util.get_address()",
"BASE_URL": f'"{base_url}" if "LOCAL" not in sys.argv else "http://127.0.0.1:1337"',
"PROXIES": json.dumps({"http":"http://127.0.0.1:8080", "https":"http://127.0.0.1:8080"})
}
if "proxies" in features or "burp" in features: if "proxies" in features or "burp" in features:
proxy = """ proxy = """
if \"proxies\" not in kwargs: if \"proxies\" not in kwargs:
kwargs[\"proxies\"] = {\"http\":\"http://127.0.0.1:8080\", \"https\":\"http://127.0.0.1:8080\"} kwargs[\"proxies\"] = PROXIES
""" """
else: else:
proxy = "" proxy = ""
variables = { if "vhost" in features or "subdomain" in features:
"BASE_URL": f'"{base_url}" if "LOCAL" not in sys.argv else "http://127.0.0.1:1337"', url_parts = urllib.parse.urlparse(base_url)
"IP_ADDRESS": "util.get_address()", host_name = url_parts.netloc
} variables["HOST_NAME"] = f"'{host_name}' if \"LOCAL\" not in sys.argv else \"127.0.0.1:1337\""
vhost_param = ", vhost=None"
full_url = f"f'{url_parts.scheme}://{{vhost}}.{{HOST_NAME}}{{uri}}' if vhost else BASE_URL + uri"
else:
vhost_param = ""
full_url = "BASE_URL + uri"
request_method = f"""def request(method, uri, **kwargs): request_method = f"""def request(method, uri{vhost_param}, **kwargs):
if not uri.startswith("/") and uri != "": if not uri.startswith("/") and uri != "":
uri = "/" + uri uri = "/" + uri
@ -32,7 +45,8 @@ def generate_template(base_url, features):
if "verify" not in kwargs: if "verify" not in kwargs:
kwargs["verify"] = False kwargs["verify"] = False
{proxy} {proxy}
return client.request(method, BASE_URL + uri, **kwargs) url = {full_url}
return client.request(method, url, **kwargs)
""" """
methods = [request_method] methods = [request_method]
@ -76,6 +90,7 @@ if __name__ == "__main__":
# https://git.romanh.de/Roman/HackingScripts # https://git.romanh.de/Roman/HackingScripts
import os import os
import io
import re import re
import sys import sys
import json import json
@ -87,7 +102,6 @@ import urllib.parse
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from hackingscripts import util, rev_shell from hackingscripts import util, rev_shell
from hackingscripts.fileserver import HttpFileServer from hackingscripts.fileserver import HttpFileServer
from urllib3.exceptions import InsecureRequestWarning from urllib3.exceptions import InsecureRequestWarning
requests.packages.urllib3.disable_warnings(category=InsecureRequestWarning) requests.packages.urllib3.disable_warnings(category=InsecureRequestWarning)

2
util.py
View File

@ -153,7 +153,7 @@ def assert_regex_match(pattern, data, err=None):
return match return match
err = f"[-] Data does not match pattern '{pattern}': '{data}'" if err is None else err err = f"[-] Data does not match pattern '{pattern}': '{data}'" if err is None else err
exit_with_error(res, err) exit_with_error(None, err)
def open_server(address, ports=None, retry=True): def open_server(address, ports=None, retry=True):
listen_port = None listen_port = None