MISP
/
misp-modules
mirror of https://github.com/MISP/misp-modules
2
0
Fork 0
Code Issues Releases Wiki Activity

fix: pep8 foobar.

pull/290/head
Raphaël Vinot 2019-04-02 16:01:33 +02:00
parent 9cb21f98e1
commit f82933779f
2 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
misp_modules/modules/expansion/greynoise.py
View File

@ -4,11 +4,12 @@ import json
misperrors = {'error': 'Error'} misperrors = {'error': 'Error'}
mispattributes = {'input': ['ip-dst', 'ip-src'], 'output': ['text']} mispattributes = {'input': ['ip-dst', 'ip-src'], 'output': ['text']}
moduleinfo = {'version': '0.1', 'author': 'Aurélien Schwab <aurelien.schwab+dev@gmail.com>', 'description': 'Module to access GreyNoise.io API.', 'module-type': ['hover']} moduleinfo = {'version': '0.1', 'author': 'Aurélien Schwab <aurelien.schwab+dev@gmail.com>', 'description': 'Module to access GreyNoise.io API.', 'module-type': ['hover']}
moduleconfig = ['user-agent']#TODO take this into account in the code moduleconfig = ['user-agent'] # TODO take this into account in the code
greynoise_api_url = 'http://api.greynoise.io:8888/v1/query/ip' greynoise_api_url = 'http://api.greynoise.io:8888/v1/query/ip'
default_user_agent = 'MISP-Module' default_user_agent = 'MISP-Module'
def handler(q=False): def handler(q=False):
if q is False: if q is False:
return False return False
@ -20,21 +21,23 @@ def handler(q=False):
else: else:
misperrors['error'] = "Unsupported attributes type" misperrors['error'] = "Unsupported attributes type"
return misperrors return misperrors
data = {'ip':ip} data = {'ip': ip}
r = requests.post(greynoise_api_url, data=data, headers={'user-agent': default_user_agent})#Real request r = requests.post(greynoise_api_url, data=data, headers={'user-agent': default_user_agent}) # Real request
if r.status_code == 200:#OK (record found) if r.status_code == 200: # OK (record found)
response = json.loads(r.text) response = json.loads(r.text)
if response: if response:
return {'results': [{'types': mispattributes['output'], 'values': response}]} return {'results': [{'types': mispattributes['output'], 'values': response}]}
elif r.status_code == 404:#Not found (not an error) elif r.status_code == 404: # Not found (not an error)
return {'results': [{'types': mispattributes['output'], 'values': 'No data'}]} return {'results': [{'types': mispattributes['output'], 'values': 'No data'}]}
else:#Real error else: # Real error
misperrors['error'] = 'GreyNoise API not accessible (HTTP ' + str(r.status_code) + ')' misperrors['error'] = 'GreyNoise API not accessible (HTTP ' + str(r.status_code) + ')'
return misperrors['error'] return misperrors['error']
def introspection(): def introspection():
return mispattributes return mispattributes
def version(): def version():
moduleinfo['config'] = moduleconfig moduleinfo['config'] = moduleconfig
return moduleinfo return moduleinfo

17
misp_modules/modules/expansion/hibp.py
View File

@ -2,12 +2,13 @@ import requests
import json import json
misperrors = {'error': 'Error'} misperrors = {'error': 'Error'}
mispattributes = {'input': ['email-dst', 'email-src'], 'output': ['text']}#All mails as input mispattributes = {'input': ['email-dst', 'email-src'], 'output': ['text']} # All mails as input
moduleinfo = {'version': '0.1', 'author': 'Aurélien Schwab', 'description': 'Module to access haveibeenpwned.com API.', 'module-type': ['hover']} moduleinfo = {'version': '0.1', 'author': 'Aurélien Schwab', 'description': 'Module to access haveibeenpwned.com API.', 'module-type': ['hover']}
moduleconfig = ['user-agent']#TODO take this into account in the code moduleconfig = ['user-agent'] # TODO take this into account in the code
haveibeenpwned_api_url = 'https://api.haveibeenpwned.com/api/v2/breachedaccount/' haveibeenpwned_api_url = 'https://api.haveibeenpwned.com/api/v2/breachedaccount/'
default_user_agent = 'MISP-Module'#User agent (must be set, requiered by API)) default_user_agent = 'MISP-Module' # User agent (must be set, requiered by API))
def handler(q=False): def handler(q=False):
if q is False: if q is False:
@ -21,20 +22,22 @@ def handler(q=False):
misperrors['error'] = "Unsupported attributes type" misperrors['error'] = "Unsupported attributes type"
return misperrors return misperrors
r = requests.get(haveibeenpwned_api_url + email, headers={'user-agent': default_user_agent})#Real request r = requests.get(haveibeenpwned_api_url + email, headers={'user-agent': default_user_agent}) # Real request
if r.status_code == 200:##OK (record found) if r.status_code == 200: # OK (record found)
breaches = json.loads(r.text) breaches = json.loads(r.text)
if breaches: if breaches:
return {'results': [{'types': mispattributes['output'], 'values': breaches}]} return {'results': [{'types': mispattributes['output'], 'values': breaches}]}
elif r.status_code == 404:#Not found (not an error) elif r.status_code == 404: # Not found (not an error)
return {'results': [{'types': mispattributes['output'], 'values': 'OK (Not Found)'}]} return {'results': [{'types': mispattributes['output'], 'values': 'OK (Not Found)'}]}
else:#Real error else: # Real error
misperrors['error'] = 'haveibeenpwned.com API not accessible (HTTP ' + str(r.status_code) + ')' misperrors['error'] = 'haveibeenpwned.com API not accessible (HTTP ' + str(r.status_code) + ')'
return misperrors['error'] return misperrors['error']
def introspection(): def introspection():
return mispattributes return mispattributes
def version(): def version():
moduleinfo['config'] = moduleconfig moduleinfo['config'] = moduleconfig
return moduleinfo return moduleinfo