From 1f73dcfe8c0cc5637896246dc0a76afe83f25da3 Mon Sep 17 00:00:00 2001 From: Sean Whalen Date: Wed, 14 Mar 2018 20:55:10 -0400 Subject: [PATCH] Fix file type detection --- parsedmarc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/parsedmarc.py b/parsedmarc.py index 98f1a40..12cc412 100644 --- a/parsedmarc.py +++ b/parsedmarc.py @@ -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"),