Update main.py

Fixed small logic bug.
This commit is contained in:
Tib3rius 2022-01-05 21:22:03 -05:00
parent d13b25c703
commit a4d08bae0d
1 changed files with 2 additions and 2 deletions

View File

@ -235,7 +235,7 @@ async def port_scan(plugin, target):
warn('A process was left running after port scan {bblue}' + plugin.name + ' {green}(' + plugin.slug + '){rst} against {byellow}' + target.address + '{rst} finished. Please ensure non-blocking processes are awaited before the run coroutine finishes. Awaiting now.', verbosity=2)
await process_dict['process'].wait()
if process_dict['process'].returncode != 0 or (process_dict['cmd'].contains('curl') and process_dict['process'].returncode != 22):
if process_dict['process'].returncode != 0:
errors = []
while True:
line = await process_dict['stderr'].readline()
@ -323,7 +323,7 @@ async def service_scan(plugin, service):
warn('A process was left running after service scan {bblue}' + plugin.name + ' {green}(' + tag + '){rst} against {byellow}' + service.target.address + '{rst} finished. Please ensure non-blocking processes are awaited before the run coroutine finishes. Awaiting now.', verbosity=2)
await process_dict['process'].wait()
if process_dict['process'].returncode != 0:
if process_dict['process'].returncode != 0 and not (process_dict['cmd'].startswith('curl') and process_dict['process'].returncode == 22):
errors = []
while True:
line = await process_dict['stderr'].readline()