mirror of https://github.com/MISP/misp-modules
145 lines
4.8 KiB
Python
Executable File
145 lines
4.8 KiB
Python
Executable File
import json
|
|
import logging
|
|
import sys
|
|
|
|
from domaintools import API
|
|
|
|
|
|
log = logging.getLogger('domaintools')
|
|
log.setLevel(logging.DEBUG)
|
|
ch = logging.StreamHandler(sys.stdout)
|
|
ch.setLevel(logging.DEBUG)
|
|
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
ch.setFormatter(formatter)
|
|
log.addHandler(ch)
|
|
|
|
misperrors = {'error': 'Error'}
|
|
mispattributes = {
|
|
'input': ['domain'],
|
|
'output': ['whois-registrant-email', 'whois-registrant-phone', 'whois-registrant-name',
|
|
'whois-registrar', 'whois-creation-date', 'freetext']
|
|
}
|
|
|
|
moduleinfo = {
|
|
'version': '0.1',
|
|
'author': 'Raphaël Vinot',
|
|
'description': 'DomainTools MISP expansion module.',
|
|
'module-type': ['expansion', 'hover']
|
|
}
|
|
|
|
moduleconfig = ['username', 'api_key']
|
|
|
|
|
|
class DomainTools(object):
|
|
|
|
def __init__(self):
|
|
self.reg_mail = {}
|
|
self.reg_phone = {}
|
|
self.reg_name = {}
|
|
self.registrar = set()
|
|
self.creation_date = set()
|
|
self.freetext = ''
|
|
|
|
def _add_value(self, value_type, value, comment):
|
|
if value_type.get(value):
|
|
if comment:
|
|
value_type[value] += ' - {}'.format(comment)
|
|
else:
|
|
value_type[value] = comment
|
|
return value_type
|
|
|
|
def add_mail(self, mail, comment=None):
|
|
self.reg_mail = self._add_value(self.reg_mail, mail, comment)
|
|
|
|
def add_phone(self, phone, comment=None):
|
|
self.reg_phone = self._add_value(self.reg_phone, phone, comment)
|
|
|
|
def add_name(self, name, comment=None):
|
|
self.reg_name = self._add_value(self.reg_name, name, comment)
|
|
|
|
def dump(self):
|
|
to_return = []
|
|
if self.reg_mail:
|
|
for mail, comment in self.reg_mail.items():
|
|
to_return.append({'type': ['whois-registrant-email'], 'values': [mail], 'comment': comment})
|
|
if self.reg_phone:
|
|
for phone, comment in self.reg_phone.items():
|
|
to_return.append({'type': ['whois-registrant-phone'], 'values': [phone], 'comment': comment})
|
|
if self.reg_name:
|
|
for name, comment in self.reg_name.items():
|
|
to_return.append({'type': ['whois-registrant-name'], 'values': [name], 'comment': comment})
|
|
if self.registrar:
|
|
to_return.append({'type': ['whois-registrar'], 'values': list(self.registrar)})
|
|
if self.creation_date:
|
|
to_return.append({'type': ['whois-creation-date'], 'values': list(self.creation_date)})
|
|
if self.freetext:
|
|
to_return.append({'type': ['freetext'], 'values': [self.freetext], 'comment': 'Freetext import'})
|
|
return to_return
|
|
|
|
|
|
def handler(q=False):
|
|
if not q:
|
|
return q
|
|
|
|
request = json.loads(q)
|
|
to_query = None
|
|
for t in mispattributes['input']:
|
|
to_query = request.get(t)
|
|
if to_query:
|
|
break
|
|
if not to_query:
|
|
misperrors['error'] = "Unsupported attributes type"
|
|
return misperrors
|
|
|
|
if request.get('config'):
|
|
if (request['config'].get('username') is None) or (request['config'].get('api_key') is None):
|
|
misperrors['error'] = 'DomainTools authentication is incomplete'
|
|
return misperrors
|
|
else:
|
|
domtools = API(request['config'].get('username'), request['config'].get('api_key'))
|
|
else:
|
|
misperrors['error'] = 'DomainTools authentication is missing'
|
|
return misperrors
|
|
|
|
whois_entry = domtools.parsed_whois(to_query)
|
|
print(whois_entry)
|
|
values = DomainTools()
|
|
|
|
if whois_entry.get('error'):
|
|
misperrors['error'] = whois_entry['error']['message']
|
|
return misperrors
|
|
|
|
if whois_entry.get('registrant'):
|
|
values.add_name(whois_entry['registrant'], 'Parsed registrant')
|
|
|
|
if whois_entry.get('registration'):
|
|
values.creation_date.add(whois_entry['registration']['created'])
|
|
|
|
if whois_entry.get('whois'):
|
|
values.freetext = whois_entry['whois']['record']
|
|
if whois_entry.get('parsed_whois'):
|
|
if whois_entry['parsed_whois']['created_date']:
|
|
values.creation_date.add(whois_entry['parsed_whois']['created_date'])
|
|
if whois_entry['parsed_whois']['registrar']['name']:
|
|
values.registrar.add(whois_entry['parsed_whois']['registrar']['name'])
|
|
for key, entry in whois_entry['parsed_whois']['contacts'].items():
|
|
if entry['email']:
|
|
values.add_mail(entry['email'], key)
|
|
if entry['phone']:
|
|
values.add_phone(entry['phone'], key)
|
|
if entry['name']:
|
|
values.add_name(entry['name'], key)
|
|
if whois_entry.emails():
|
|
for mail in whois_entry.emails():
|
|
values.add_mail(mail)
|
|
return json.dumps({'results': values.dump()})
|
|
|
|
|
|
def introspection():
|
|
return mispattributes
|
|
|
|
|
|
def version():
|
|
moduleinfo['config'] = moduleconfig
|
|
return moduleinfo
|