separate check_kill

Author: jm33-m0

lib/cli/console.py

@@ -117,6 +117,18 @@ def completer(text, state):
 atexit.register(readline.write_history_file, HISTFILE)
 
 
+# kill process by name
+def check_kill_process(pstring):
+    '''
+    cross-platform way of killing process by name
+    '''
+    import psutil
+
+    for proc in psutil.process_iter():
+        if pstring in str(proc.cmdline):
+            proc.kill()
+
+
 def debug_except():
     '''
     display traceback info

lib/cli/main.py

@@ -14,7 +14,7 @@
 import lib.cli.console as console
 import lib.tools.baidu as baidu
 import lib.tools.exploits as ExecExp
-from lib.cli.console import debug_except, input_check
+from lib.cli.console import debug_except, input_check, check_kill_process
 from lib.tools import zoomeye
 
 
@@ -37,18 +37,6 @@ class SessionParameters:
         '\\data\\ss.json'
 
 
-# kill process by name
-def check_kill_process(pstring):
-    '''
-    cross-platform way of killing process by name
-    '''
-    import psutil
-
-    for proc in psutil.process_iter():
-        if pstring in str(proc.cmdline):
-            proc.kill()
-
-
 def list_exp():
     '''
     list all labeled exploits under the root of your exploit dir