bugfix (d80bd88e) · Commits · Zhang Yushan / cve-search

lib/DatabaseLayer.py

+3 −2

Original line number	Original line	Diff line number	Diff line
	@@ -12,6 +12,7 @@
	# imports		# imports
	from lib.Config import Configuration as conf		from lib.Config import Configuration as conf
	import pymongo		import pymongo
			import re

	# Variables		# Variables
	db=conf.getMongoConnection()		db=conf.getMongoConnection()
	@@ -142,8 +143,8 @@ def getRules(list):
	else:		else:
	return []		return []
	rlist=col.find({'type':'cpe'}).distinct('id')		rlist=col.find({'type':'cpe'}).distinct('id')
	hardware=["cpe:2.3:([^:]*:){9}"+re.escape(x) for x in col.find({'type':'cpe'}).distinct('id')]		hardware=["cpe:2.3:([^:]*:){9}"+re.escape(x) for x in col.find({'type':'targethardware'}).distinct('id')]
	software=["cpe:2.3:([^:]*:){8}"+re.escape(x) for x in col.find({'type':'cpe'}).distinct('id')]		software=["cpe:2.3:([^:]*:){8}"+re.escape(x) for x in col.find({'type':'targetsoftware'}).distinct('id')]
	rlist.extend(hardware)		rlist.extend(hardware)
	rlist.extend(software)		rlist.extend(software)
	return rlist		return rlist