add threat list expansion

pull/197/head
Sebdraven 2018-06-22 11:59:09 +02:00
parent 1d1fd36569
commit e230c88c15
1 changed files with 36 additions and 4 deletions

View File

@ -82,7 +82,7 @@ def handle_ip(api, ip, misperrors):
if status_ok:
result_filtered['results'].extend(r)
else:
misperrors['error'] = 'Error datascan result %s' % status_ok
misperrors['error'] = 'Error datascan result '
return misperrors
r, status_ok = expand_forward(api, ip, misperrors)
@ -90,7 +90,7 @@ def handle_ip(api, ip, misperrors):
if status_ok:
result_filtered['results'].extend(r)
else:
misperrors['error'] = 'Error forward result %s' % status_ok
misperrors['error'] = 'Error forward result'
return misperrors
r, status_ok = expand_reverse(api, ip, misperrors)
@ -101,7 +101,14 @@ def handle_ip(api, ip, misperrors):
misperrors['error'] = 'Error reverse result'
return misperrors
print(result_filtered)
r, status_ok = expand_threatlist(api, misperrors, ip=ip)
if status_ok:
result_filtered['results'].extend(r)
else:
misperrors['error'] = 'Error threat list'
return misperrors
return result_filtered
@ -168,7 +175,6 @@ def expand_datascan(api, misperror,**kwargs):
results = api.datascan(query)
if results['status'] == 'ok':
status_ok = True
for elem in results['results']:
@ -308,6 +314,32 @@ def expand_pastries(api, misperror, **kwargs):
return r, status_ok
def expand_threatlist(api, misperror,**kwargs):
status_ok = False
r = []
query = None
threat_list = []
if 'ip' in kwargs:
query = kwargs.get('ip')
else:
query = kwargs.get('domain')
results = api.threatlist(query)
if results['status'] == 'ok':
status_ok = True
threat_list = ['seen %s on %s ' % (item['seen_date'], item['threatlist'])
for item in results['results']]
r.append({'types': ['comment'],
'categories': ['Other'],
'values': [threat_list]
})
return r,status_ok
def introspection():
return mispattributes