chg: Taking into consideration if a user agent is specified in the module configuration

misp_modules/modules/expansion/macvendors.py

@@ -4,7 +4,7 @@ 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']}
-moduleconfig = ['user-agent']  # TODO take this into account in the code
+moduleconfig = ['user-agent']
 
 macvendors_api_url = 'https://api.macvendors.com/'
 default_user_agent = 'MISP-Module'
@@ -21,7 +21,8 @@ def handler(q=False):
     else:
         misperrors['error'] = "Unsupported attributes type"
         return misperrors
-    r = requests.get(macvendors_api_url + mac, headers={'user-agent': default_user_agent})  # Real request
+    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: