RomanNum3ral
/
AutoRecon
mirror of https://github.com/Tib3rius/AutoRecon.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update autorecon.py 

Added forced service check. If no services are defined, AutoRecon will error out.
This commit is contained in:
Tib3rius 2021-08-27 09:31:33 -04:00
parent 98037302f7
commit 288af6a961
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
autorecon.py
View File

@ -439,6 +439,7 @@ class AutoRecon(object):
'accessible': False, 'accessible': False,
'verbose': 0 'verbose': 0
} }
self.errors = False
self.lock = asyncio.Lock() self.lock = asyncio.Lock()
self.load_slug = None self.load_slug = None
self.load_module = None self.load_module = None
@ -937,6 +938,11 @@ async def scan_target(target):
if services: if services:
pending.append(asyncio.create_task(asyncio.sleep(0))) pending.append(asyncio.create_task(asyncio.sleep(0)))
else:
error('No services were defined. Please check your service syntax: [tcp|udp]/<port>/<service-name>/[secure|insecure]')
heartbeat.cancel()
autorecon.errors = True
return
else: else:
for plugin in target.autorecon.plugin_types['port']: for plugin in target.autorecon.plugin_types['port']:
plugin_tag_set = set(plugin.tags) plugin_tag_set = set(plugin.tags)
@ -1526,6 +1532,12 @@ async def main():
timed_out = False timed_out = False
while pending: while pending:
done, pending = await asyncio.wait(pending, return_when=asyncio.FIRST_COMPLETED, timeout=1) done, pending = await asyncio.wait(pending, return_when=asyncio.FIRST_COMPLETED, timeout=1)
# If something failed in scan_target, autorecon.errors will be true.
if autorecon.errors:
cancel_all_tasks(None, None)
sys.exit(1)
# Check if global timeout has occurred. # Check if global timeout has occurred.
if autorecon.config['timeout'] is not None: if autorecon.config['timeout'] is not None:
elapsed_seconds = round(time.time() - start_time) elapsed_seconds = round(time.time() - start_time)