autorecon.py: added termios to restore the flags

2019-10-14 22:07:39 +08:00 · 2019-10-14 22:07:39 +08:00 · a5f9509c01
parent a965613262
commit a5f9509c01
1 changed files with 5 additions and 1 deletions
--- a/autorecon.py
+++ b/autorecon.py
@ -20,13 +20,17 @@ import socket
 import string
 import sys
 import toml
+import termios


 def _quit():
-    colorama.deinit()
+    termios.tcsetattr(sys.stdin.fileno(), TERM_FLAGS)
+

 atexit.register(_quit)

+TERM_FLAGS = termios.tcgetattr(sys.stdin.fileno())
+
 verbose = 0
 nmap = '-vv --reason -Pn'
 srvname = ''