Fix file type detection

This commit is contained in:
Sean Whalen
2018-03-14 20:55:10 -04:00
parent 1c1280a0a2
commit 1f73dcfe8c

View File

@@ -947,12 +947,12 @@ def parse_report_email(input_, nameservers=None, timeout=6.0):
result = OrderedDict([("report_type", "forensic"),
("report", forensic_report)])
payload = b64decode(part.get_payload())
if payload.startswith(MAGIC_ZIP) or \
payload.startswith(MAGIC_GZIP) or \
payload.startswith(MAGIC_XML):
ns = nameservers
xml = b64decode(part.get_payload())
aggregate_report = parse_aggregate_report_file(xml,
aggregate_report = parse_aggregate_report_file(payload,
nameservers=ns,
timeout=timeout)
result = OrderedDict([("report_type", "aggregate"),