mirror of https://github.com/MISP/misp-modules
Merge pull request #335 from FafnerKeyZee/patch-2
Travis should not be complaining with the tests after the latest update on "test_cve"pull/338/head
commit
205342996a
|
@ -3,10 +3,12 @@ import requests
|
||||||
|
|
||||||
misperrors = {'error': 'Error'}
|
misperrors = {'error': 'Error'}
|
||||||
mispattributes = {'input': ['vulnerability'], 'output': ['text']}
|
mispattributes = {'input': ['vulnerability'], 'output': ['text']}
|
||||||
moduleinfo = {'version': '0.2', 'author': 'Alexandre Dulaunoy', 'description': 'An expansion hover module to expand information about CVE id.', 'module-type': ['hover']}
|
moduleinfo = {'version': '0.3', 'author': 'Alexandre Dulaunoy', 'description': 'An expansion hover module to expand information about CVE id.', 'module-type': ['hover']}
|
||||||
moduleconfig = []
|
moduleconfig = ["custom_API"]
|
||||||
cveapi_url = 'https://cve.circl.lu/api/cve/'
|
cveapi_url = 'https://cve.circl.lu/api/cve/'
|
||||||
|
|
||||||
|
def check_url(url):
|
||||||
|
return "{}/".format(url) if not url.endswith('/') else url
|
||||||
|
|
||||||
def handler(q=False):
|
def handler(q=False):
|
||||||
if q is False:
|
if q is False:
|
||||||
|
@ -16,7 +18,8 @@ def handler(q=False):
|
||||||
misperrors['error'] = 'Vulnerability id missing'
|
misperrors['error'] = 'Vulnerability id missing'
|
||||||
return misperrors
|
return misperrors
|
||||||
|
|
||||||
r = requests.get(cveapi_url + request.get('vulnerability'))
|
api_url = check_url(request['config']['custom_API']) if request['config'].get('custom_API') else cveapi_url
|
||||||
|
r = requests.get("{}{}".format(api_url, request.get('vulnerability')))
|
||||||
if r.status_code == 200:
|
if r.status_code == 200:
|
||||||
vulnerability = json.loads(r.text)
|
vulnerability = json.loads(r.text)
|
||||||
if vulnerability:
|
if vulnerability:
|
||||||
|
@ -25,7 +28,7 @@ def handler(q=False):
|
||||||
else:
|
else:
|
||||||
summary = 'Non existing CVE'
|
summary = 'Non existing CVE'
|
||||||
else:
|
else:
|
||||||
misperrors['error'] = 'cve.circl.lu API not accessible'
|
misperrors['error'] = 'API not accessible'
|
||||||
return misperrors['error']
|
return misperrors['error']
|
||||||
|
|
||||||
r = {'results': [{'types': mispattributes['output'], 'values': summary}]}
|
r = {'results': [{'types': mispattributes['output'], 'values': summary}]}
|
||||||
|
|
Loading…
Reference in New Issue