1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
#!/usr/bin/env python
import json
from bson.objectid import ObjectId
from pymongo import MongoClient
from libnmap.parser import NmapParser
from libnmap.plugins.backendplugin import NmapBackendPlugin
from libnmap.reportjson import ReportEncoder
class NmapMongodbPlugin(NmapBackendPlugin):
"""
This class handle the persistence of NmapRepport object in mongodb
Implementation is made using pymongo
Object of this class must be create via the
BackendPluginFactory.create(**url) where url is a named dict like
{'plugin_name': "mongodb"} this dict may receive all the param
MongoClient() support
"""
def __init__(self, dbname=None, store=None, **kwargs):
NmapBackendPlugin.__init__(self)
if dbname is not None:
self.dbname = dbname
if store is not None:
self.store = store
self.dbclient = MongoClient(**kwargs)
self.collection = self.dbclient[self.dbname][self.store]
def insert(self, report):
"""
create a json object from an NmapReport instance
:param NmapReport: obj to insert
:return: str id
"""
j = json.dumps(report, cls=ReportEncoder)
try:
oid = self.collection.insert(json.loads(j))
except Exception as e:
em = "Failed to insert nmap object in MongoDB: {0}".format(e)
raise Exception(em)
return str(oid)
def get(self, str_report_id=None):
"""select a NmapReport by Id
:param str: id
:return: NmapReport object
"""
rid = str_report_id
nmapreport = None
if str_report_id is not None and isinstance(str_report_id, str):
rid = ObjectId(str_report_id)
if isinstance(rid, ObjectId):
# get a specific report by mongo's id
resultset = self.collection.find({"_id": rid})
if resultset.count() == 1:
# search by id means only one in the iterator
record = resultset[0]
# remove mongo's id to recreate the NmapReport Obj
del record["_id"]
nmapreport = NmapParser.parse_fromdict(record)
return nmapreport
def getall(self, dict_filter=None):
"""return a list of tuple (id,NmapReport) saved in the backend
TODO : add a filter capability
"""
nmapreportlist = []
resultset = self.collection.find()
for report in resultset:
oid = report["_id"]
del report["_id"]
nmapreport = NmapParser.parse_fromdict(report)
nmapreportlist.append((oid, nmapreport))
return nmapreportlist
def delete(self, report_id=None):
"""
delete an obj from the backend
:param str: id
:return: dict document with result or None
"""
if report_id is not None and isinstance(report_id, str):
return self.collection.remove({"_id": ObjectId(report_id)})
else:
return self.collection.remove({"_id": report_id})
|
