Basic callback done, taxii server will now push to a MISP server

misp_taxii_hooks/hooks.py

@@ -1,6 +1,12 @@
 #!/usr/bin/env python3
 
+######
+# TODO: DETECT DUPLICATE DATA
+#####
+
 import pymisp
+import tempfile
+import os
 
 from opentaxii.signals import (
     CONTENT_BLOCK_CREATED, INBOX_MESSAGE_CREATED
@@ -8,19 +14,37 @@ from opentaxii.signals import (
 
 ## CONFIG
 
+CONFIG = {
+            "MISP_URL" : "[URL]",
+            "MISP_API" : "[APIKEY]",
+        }
+
+MISP = pymisp.PyMISP( 
+                        CONFIG["MISP_URL"],
+                        CONFIG["MISP_API"],
+                )
+
 def post_stix(manager, content_block, collection_ids, service_id):
-    CONFIG = {
+    '''
-                "MISP_URL" : "localhost",
+        Callback function for when our taxii server gets new data
-                "MISP_API" : "DEADBEEF",
+        Will convert it to a MISPEvent and push to the server
-            }
+    '''
 
-    MISP = pymisp.PyMISP( 
+    # Create a temporary file to load STIX data from
-                            CONFIG["MISP_URL"],
+    f = tempfile.NamedTemporaryFile(delete=False, mode="w")
-                            CONFIG["MISP_API"],
+    f.write(content_block.content)
-                    )
+    f.close()
 
-    with open("/tmp/test.txt", "w") as f:
+    # Load the package
-        f.write("connect!")
+    package = pymisp.tools.stix.load_stix(f.name)
-    print("Content: {}".format(content_block.content))
     
+    # Delete that old temporary file
+    os.unlink(f.name)
+
+    # Push the event to MISP
+    # TODO: There's probably a proper method to do this rather than json_full
+    # But I don't wanna read docs
+    MISP.add_event(package._json_full())
+
+# Make TAXII call our push function whenever it gets new data
 CONTENT_BLOCK_CREATED.connect(post_stix)