Git/mgeeky-Penetration-Testing-Tools
1
0
Fork 0
mirror of https://github.com/mgeeky/Penetration-Testing-Tools.git synced 2024-12-18 15:15:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixed some issues.

Browse Source
This commit is contained in:
Mariusz B. / mgeeky 2021-03-26 20:47:56 +01:00
parent 69b2ea5d3d
commit ca6a96ac07
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
red-teaming/C3-Client/c3-client.py
View File

@ -36,6 +36,8 @@ commands = {
] ]
} }
serverValidated = False
# BackendCommons.h: enum class Command : std::uint16_t # BackendCommons.h: enum class Command : std::uint16_t
commandsMap = { commandsMap = {
'AddDevice' : 0, 'AddDevice' : 0,
@ -92,6 +94,8 @@ def printJson(data):
print(json.dumps(data, sort_keys=True, indent=4)) print(json.dumps(data, sort_keys=True, indent=4))
def getRequest(url, rawResp = False, stream = False): def getRequest(url, rawResp = False, stream = False):
global serverValidated
auth = None auth = None
if config['httpauth']: if config['httpauth']:
user, _pass = config['httpauth'].split(':') user, _pass = config['httpauth'].split(':')
@ -103,11 +107,24 @@ def getRequest(url, rawResp = False, stream = False):
try: try:
resp = requests.get(fullurl, headers=headers, auth=auth, stream = stream, timeout = 5) resp = requests.get(fullurl, headers=headers, auth=auth, stream = stream, timeout = 5)
if not serverValidated:
try:
gateways = requests.get(config["host"] + '/api/gateway', headers=headers, auth=auth, stream = stream, timeout = 5)
if gateways.status_code < 200 or gateways.status_code > 300:
raise Exception()
serverValidated = True
except:
Logger.fatal('Server could not be validated. Are you sure your Host value points to a valid C3 webcontroller URL?')
except requests.exceptions.ConnectTimeout as e: except requests.exceptions.ConnectTimeout as e:
Logger.fatal(f'Connection with {config["host"]} timed-out.') Logger.fatal(f'Connection with {config["host"]} timed-out.')
except Exception as e: except Exception as e:
Logger.fatal(f'GET request failed ({url}): {e}') Logger.fatal(f'GET request failed ({url}): {e}')
Logger.dbg(f'First 512 bytes of response:\n{resp.text[:512]}')
if rawResp: if rawResp:
return resp return resp