bugfixes, error handling and new function (da734541) · Commits · Zhang Yushan / cve-search

lib/PluginManager.py

+19 −5

Original line number	Diff line number	Diff line
		@@ -77,7 +77,7 @@ class PluginManager():

		def getWebPluginsWithPage(self):
		plugins = []
		for plug in self.getWebPlugins:
		for plug in self.getWebPlugins():
		page = plug.getPage()
		if page and page[0]: # Make sure there is a page
		plugins.append(plug)
		@@ -87,10 +87,13 @@ class PluginManager():
		actions = []
		for plugin in self.getWebPlugins():
		auth = plugin.requiresAuth
		try:
		for action in plugin.getCVEActions(**args):
		action['auth'] = auth
		action['plugin'] = plugin.getUID()
		actions.append(action)
		except:
		print("[!] Plugin %s failed on fetching CVE actions!"%plugin.getName())
		return actions

		def requiresAuth(self, plugin):
		@@ -109,3 +112,14 @@ class PluginManager():
		else:
		return ("error.html", {'status': {'except': 'plugin-page-missing'}})
		return ("error.html", {'status': {'except': 'plugin-not-loaded'}})

		def cvePluginInfo(self, cve, **args):
		cveInfo = []
		for plugin in self.getWebPlugins():
		try:
		data = plugin.cvePluginInfo(cve, **args)
		if type(data) == dict and 'title' in data and 'data' in data:
		cveInfo.append(data)
		except:
		print("[!] Plugin %s failed on fetching CVE plugin info!"%plugin.getName())
		return cveInfo