#21 create intermediate program to load misp-modules from Cortex

contrib/misp-modules-loader.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+
+import sys
+import os
+import getopt
+import json
+
+
+
+# modules_path = "./misp-modules/misp_modules/modules/expansion"
+# sys.path.append(modules_path)
+
+def run(argv):
+    try:
+        opts, args = getopt.getopt(argv, 'hp:i:r:', ["help","path=", "info=","run="])
+    except getopt.GetoptError as err:
+        print(__file__ + " --info <misp-module>")
+        print(__file__ + " --path <misp-modules path> --run <misp-module>")
+        print(str(err))
+        sys.exit(2)
+
+    module = None
+    path = None
+    for opt,arg in opts:
+
+        # TODO: check if module exist else exit
+        if opt in ('-h', '--help'):
+            print(__file__ + " --info <misp-module>")
+            print(__file__ + " --run <misp-module>")
+            sys.exit()
+
+        elif opt in ('-p', '--path'):
+            path = arg
+            sys.path.append(path)
+
+        elif opt in ('-r', '--run'):
+            module = arg
+            if path:
+                m = __import__(module)
+
+                data = json.load(sys.stdin)
+                print(json.dumps(m.handler(json.dumps(data))))
+                sys.exit(0)
+
+            else:
+                print("add module path")
+
+        elif opt in ('-i','--info'):
+            module = arg
+            print(module)
+            m = __import__(module)
+            print(({'name': module, 'mispattributes': m.mispattributes,
+                        'moduleinfo':m.moduleinfo}))
+
+
+
+
+if __name__ == '__main__':
+    if len(sys.argv[1:]) > 0:
+        run(sys.argv[1:])
+    else:
+        print(__file__ + " --info <misp-module>")
+        print(__file__ + " --run <misp-module>")
+        sys.exit(2)