Add DNSDB API parameters (#319)

TheHive-Project · Dec 10, 2018 · a7708cb · a7708cb
1 parent 23345ae
commit a7708cb
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/analyzers/DNSDB/dnsdb.py b/analyzers/DNSDB/dnsdb.py
@@ -19,13 +19,17 @@ def __init__(self):
 
     def execute_dnsdb_service(self, client):
         data = self.get_data()
+        rrtype = self.getParam('parameters.rrtype', None, None)
+        bailiwick = self.getParam('parameters.bailiwick', None, None)
+        before = self.getParam('parameters.before', None, None)
+        after = self.getParam('parameters.after', None, None)
 
         if self.service == 'domain_name' and self.data_type in ['domain', 'fqdn']:
-            return client.query_rrset(data)
+            return client.query_rrset(data, rrtype=rrtype, bailiwick=bailiwick, before=before, after=after)
         elif self.service == 'ip_history' and self.data_type == 'ip':
-            return client.query_rdata_ip(data)
+            return client.query_rdata_ip(data, before=before, after=after)
         elif self.service == 'name_history' and self.data_type in ['domain', 'fqdn']:
-            return client.query_rdata_name(data)
+            return client.query_rdata_name(data, rrtype=rrtype, before=before, after=after)
         else:
             self.error('Unknown DNSDB service or invalid data type')