From 2262d82cc70c0476177ef7fd15c46614fe49d73d Mon Sep 17 00:00:00 2001 From: Hannah Ward Date: Fri, 3 Nov 2017 11:47:27 +0000 Subject: [PATCH] fix: Just in case taxii DOESN'T give us bytes Fixes #26 --- misp_taxii_hooks/hooks.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/misp_taxii_hooks/hooks.py b/misp_taxii_hooks/hooks.py index 7d3786c..4a036c6 100644 --- a/misp_taxii_hooks/hooks.py +++ b/misp_taxii_hooks/hooks.py @@ -55,7 +55,11 @@ def post_stix(manager, content_block, collection_ids, service_id): # Load the package log.info("Posting STIX...") - package = pymisp.tools.stix.load_stix(StringIO(content_block.content.decode())) + block = content_block.content + if isinstance(block, bytes): + block = block.decode() + + package = pymisp.tools.stix.load_stix(StringIO(block)) log.info("STIX loaded succesfully.") values = [x.value for x in package.attributes] log.info("Extracted %s", values)