import json import vulners misperrors = {'error': 'Error'} mispattributes = {'input': ['vulnerability'], 'output': ['text']} moduleinfo = {'version': '0.1', 'author': 'Igor Ivanov', 'description': 'An expansion hover module to expand information about CVE id using Vulners API.', 'module-type': ['hover']} # Get API key from https://vulners.com/userinfo moduleconfig = ["apikey"] def handler(q=False): if q is False: return False request = json.loads(q) if not request.get('vulnerability'): misperrors['error'] = 'Vulnerability id missing' return misperrors ai_summary = '' exploit_summary = '' vuln_summary = '' if not request.get('config') or not request['config'].get('apikey'): return {'error': "A Vulners api key is required for this module."} key = request['config']['apikey'] vulners_api = vulners.Vulners(api_key=key) vulnerability = request.get('vulnerability') vulners_document = vulners_api.document(vulnerability) # Get AI scoring from the document if it's already calculated # There is no need to call AI Scoring method if 'score' in vulners_document.get('enchantments', {}): vulners_ai_score = vulners_document['enchantments']['score']['value'] else: vulners_ai_score = vulners_api.get_ai_score(vulnerability) if len(vulners_ai_score) == 2: vulners_ai_score = vulners_ai_score[0] vulners_exploits = vulners_api.searchExploit(vulnerability) if vulners_document: vuln_summary += vulners_document.get('description') else: vuln_summary += 'Non existing CVE' if vulners_ai_score: ai_summary += 'Vulners AI Score is ' + str(vulners_ai_score) + " " if vulners_exploits: exploit_summary += " || " + str(len(vulners_exploits)) + " Public exploits available:\n " for exploit in vulners_exploits: exploit_summary += exploit['title'] + " " + exploit['href'] + "\n " exploit_summary += "|| Vulnerability Description: " + vuln_summary summary = ai_summary + exploit_summary + vuln_summary r = {'results': [{'types': mispattributes['output'], 'values': summary}]} return r def introspection(): return mispattributes def version(): moduleinfo['config'] = moduleconfig return moduleinfo