Added MongoDB output support (#407)

mongodb.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+
+import pymongo
+
+from twisted.python import log
+
+import cowrie.core.output
+
+
+class Output(cowrie.core.output.Output):
+    """
+    """
+
+    def __init__(self, cfg):
+        self.cfg = cfg
+        cowrie.core.output.Output.__init__(self, cfg)
+
+
+    def start(self):
+        """
+        """
+        db_addr = self.cfg.get('output_mongodb', 'connection_string')
+        db_name = self.cfg.get('output_mongodb', 'database')
+
+        try:
+            self.mongo_client = pymongo.MongoClient(db_addr)
+            self.mongo_db = self.mongo_client[db_name]
+            self.coll = self.mongo_db['events']
+        except Exception, e:
+            log.msg('output_mongodb: Error: %s' % str(e))
+
+
+    def stop(self):
+        """
+        """
+        self.mongo_client.close()
+
+
+    def write(self, entry):
+        """
+        """
+        for i in list(entry.keys()):
+            # Remove twisted 15 legacy keys
+            if i.startswith('log_'):
+                del entry[i]
+        try:
+            self.coll.insert_one(entry)
+        except Exception,e:
+            log.msg('output_mongodb: MongoDB Error: %s' % str(e))