add controls

TheHive-Project · Nov 3, 2017 · 45529ba · 45529ba
1 parent 8d37026
commit 45529ba
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/analyzers/Shodan/shodan_api.py b/analyzers/Shodan/shodan_api.py
@@ -22,9 +22,9 @@ def info_domains(self, domain):
             ips = list(set([r['ip_str'] for r in results['matches']]))
             ports = list(set([r['port'] for r in results['matches']]))
             transports = list(set([r['transport'] for r in results['matches']]))
-            isp = list(set([r['isp'] for r in results['matches']]))
-            asn = list(set([r['asn'] for r in results['matches']]))
-            orgs = list(set([r['org'] for r in results['matches']]))
+            isp = list(set([r['isp'] for r in results['matches'] if 'isp' in r]))
+            asn = list(set([r['asn'] for r in results['matches'] if 'asn' in r]))
+            orgs = list(set([r['org'] for r in results['matches'] if 'org' in r]))
 
             return {'all_domains': all_domains, 'ips': ips, 'ports': ports, 'transports': transports,
                     'isp': isp, 'asn': asn, 'orgs': orgs