Skip to content

Commit

Permalink
Merge pull request #1 from LaZyDK/patch-2
Browse files Browse the repository at this point in the history
Update Umbrella.py
  • Loading branch information
LaZyDK authored Sep 30, 2020
2 parents 0f60fcd + 8aa7212 commit c1fa5a1
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion analyzers/Umbrella/Umbrella.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,17 @@ def summary(self, raw):


def run(self):
# Map The Hive observable types to Umbrella observable types
observable_mapping = {
"domain": "domain",
"fqdn": "domain",
}

if self.service == 'get':
if self.data_type == 'domain':
dataType = self.get_param("dataType")

# Validate the supplied observable type is supported
if dataType in observable_mapping.keys():
data = self.get_param('data', None, 'Data is missing')
r = self.umbrella_runreport(data)
self.report(r)
Expand Down

0 comments on commit c1fa5a1

Please sign in to comment.