mirror of https://github.com/MISP/misp-modules
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
import requests
|
|
import json
|
|
|
|
misperrors = {'error': 'Error'}
|
|
mispattributes = {'input': ['mac-address'], 'output': ['text']}
|
|
moduleinfo = {
|
|
'version': '0.1',
|
|
'author': 'Aurélien Schwab',
|
|
'description': 'Module to access Macvendors API.',
|
|
'module-type': ['hover'],
|
|
'name': 'Macvendors Lookup',
|
|
'logo': 'macvendors.png',
|
|
'requirements': [],
|
|
'features': 'The module takes a MAC address as input and queries macvendors.com for some information about it. The API returns the name of the vendor related to the address.',
|
|
'references': ['https://macvendors.com/', 'https://macvendors.com/api'],
|
|
'input': 'A MAC address.',
|
|
'output': 'Additional information about the MAC address.',
|
|
}
|
|
moduleconfig = ['user-agent']
|
|
|
|
macvendors_api_url = 'https://api.macvendors.com/'
|
|
default_user_agent = 'MISP-Module'
|
|
|
|
|
|
def handler(q=False):
|
|
if q is False:
|
|
return False
|
|
request = json.loads(q)
|
|
for input_type in mispattributes['input']:
|
|
if input_type in request:
|
|
mac = request[input_type]
|
|
break
|
|
else:
|
|
misperrors['error'] = "Unsupported attributes type"
|
|
return misperrors
|
|
user_agent = request['config']['user-agent'] if request.get('config') and request['config'].get('user-agent') else default_user_agent
|
|
r = requests.get(macvendors_api_url + mac, headers={'user-agent': user_agent}) # Real request
|
|
if r.status_code == 200: # OK (record found)
|
|
response = r.text
|
|
if response:
|
|
return {'results': [{'types': mispattributes['output'], 'values': response}]}
|
|
elif r.status_code == 404: # Not found (not an error)
|
|
return {'results': [{'types': mispattributes['output'], 'values': 'Not found'}]}
|
|
else: # Real error
|
|
misperrors['error'] = 'MacVendors API not accessible (HTTP ' + str(r.status_code) + ')'
|
|
return misperrors['error']
|
|
|
|
|
|
def introspection():
|
|
return mispattributes
|
|
|
|
|
|
def version():
|
|
moduleinfo['config'] = moduleconfig
|
|
return moduleinfo
|