add console

LoRexxar · Sep 9, 2020 · cb7bfde · cb7bfde
1 parent 86095f4
commit cb7bfde
Show file tree

Hide file tree

Showing 8 changed files with 1,593 additions and 10 deletions.
diff --git a/Kunlun_M/settings.py.bak b/Kunlun_M/settings.py.bak
@@ -161,4 +161,11 @@ EXAMPLES_PATH = os.path.join(TESTS_PATH, 'examples')
 RULES_PATH = os.path.join(PROJECT_DIRECTORY, 'rules')
 CONFIG_PATH = os.path.join(PROJECT_DIRECTORY, 'config')
 
-
+# check platform
+PLATFORM = "Linux"
+if "Windows" in platform.system():
+    PLATFORM = "windows"
+elif "Linux" in platform.system():
+    PLATFORM = "linux"
+elif "Darwin" in platform.system():
+    PLATFORM = "mac"
diff --git a/core/__init__.py b/core/__init__.py
@@ -30,6 +30,7 @@
 from .__version__ import __copyright__, __epilog__, __scan_epilog__
 
 from core.rule import RuleCheck, TamperCheck
+from core.console import KunlunInterpreter
 from web.index.models import ScanTask
 
 try:
@@ -71,7 +72,14 @@ def main():
                                        help='show all rules & tanmpers')
 
         parser_group_show.add_argument('-k', '--key', dest='listkey', action='store', default="all",
-                                       help='key for show rule & tamper')
+                                       help='key for show rule & tamper. eg: 1001/wordpress')
+
+        parser_group_console = subparsers.add_parser('console', help='enter console mode',
+                                                     description=__introduction__.format(detail='enter console mode'),
+                                                     formatter_class=argparse.RawDescriptionHelpFormatter,
+                                                     usage=argparse.SUPPRESS, add_help=True)
+        parser_group_console.add_argument('console', action='store_true', default=True,
+                                          help='enter console mode')
 
         args = parser.parse_args()
 
@@ -114,10 +122,23 @@ def main():
                 logger.info("[INIT] TamperRecover finished.")
                 exit()
 
+            else:
+                parser_group_core.print_help()
+                exit()
+
         if hasattr(args, "list"):
             if args.list:
                 logger.info("Show {}:\n{}".format(args.list, show_info(args.list, args.listkey.strip(""))))
                 exit()
+            else:
+                parser_group_show.print_help()
+                exit()
+
+        if hasattr(args, "console"):
+            logger.info("[INIT] Enter KunLun-M console mode.")
+            shell = KunlunInterpreter()
+            shell.start()
+            exit()
 
         if not hasattr(args, "target") or args.target == '':
             parser.print_help()

diff --git a/core/cli.py b/core/cli.py
@@ -14,6 +14,7 @@
 
 import os
 import codecs
+import pprint
 from prettytable import PrettyTable
 
 from .detection import Detection
@@ -332,11 +333,11 @@ def list_parse(rules_path, istamp=False):
     {}
 
 Filter Func:
-    {}
+{}
     
 Input Control:
-    {}
-""".format(tampname, filter_func, input_control)
+{}
+""".format(tampname, pprint.pformat(filter_func, indent=4), pprint.pformat(input_control, indent=4))
         else:
             logger.error("[Info] no tamper name {]".format(key))