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

Update main.py 

Potential fix for #187
This commit is contained in:
Tib3rius 2023-01-16 00:42:52 -05:00
parent 851ffbd316
commit 9cfbc4a7ce
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
autorecon/main.py
View File

@ -49,8 +49,8 @@ else:
warn('It looks like the plugins in ' + config['data_dir'] + ' are outdated. Please remove the ' + config['data_dir'] + ' directory and re-run AutoRecon to rebuild them.') warn('It looks like the plugins in ' + config['data_dir'] + ' are outdated. Please remove the ' + config['data_dir'] + ' directory and re-run AutoRecon to rebuild them.')
# Save current terminal settings so we can restore them. # Saves current terminal settings so we can restore them.
terminal_settings = termios.tcgetattr(sys.stdin.fileno()) terminal_settings = None
autorecon = AutoRecon() autorecon = AutoRecon()
@ -120,11 +120,11 @@ def cancel_all_tasks(sig, frame):
_, alive = psutil.wait_procs(processes, timeout=10) _, alive = psutil.wait_procs(processes, timeout=10)
if len(alive) > 0: if len(alive) > 0:
error('The following process IDs could not be killed: ' + ', '.join([str(x.pid) for x in sorted(alive, key=lambda x: x.pid)])) error('The following process IDs could not be killed: ' + ', '.join([str(x.pid) for x in sorted(alive, key=lambda x: x.pid)]))
if not config['disable_keyboard_control']: if not config['disable_keyboard_control']:
# Restore original terminal settings. # Restore original terminal settings.
termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, terminal_settings) if terminal_settings is not None:
termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, terminal_settings)
async def start_heartbeat(target, period=60): async def start_heartbeat(target, period=60):
while True: while True:
@ -1513,6 +1513,9 @@ async def run():
start_time = time.time() start_time = time.time()
if not config['disable_keyboard_control']:
terminal_settings = termios.tcgetattr(sys.stdin.fileno())
pending = [] pending = []
i = 0 i = 0
while autorecon.pending_targets: while autorecon.pending_targets:
@ -1616,7 +1619,8 @@ async def run():
if not config['disable_keyboard_control']: if not config['disable_keyboard_control']:
# Restore original terminal settings. # Restore original terminal settings.
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, terminal_settings) if terminal_settings is not None:
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, terminal_settings)
def main(): def main():
# Capture Ctrl+C and cancel everything. # Capture Ctrl+C and cancel everything.