From 823822bbf14d3e7a0b1436d332ce7864abbc7191 Mon Sep 17 00:00:00 2001 From: haraksin Date: Wed, 13 May 2020 13:49:34 -0700 Subject: [PATCH] Making dedup optional --- misp_taxii_hooks/hooks.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/misp_taxii_hooks/hooks.py b/misp_taxii_hooks/hooks.py index e7be3c9..25a3964 100644 --- a/misp_taxii_hooks/hooks.py +++ b/misp_taxii_hooks/hooks.py @@ -22,6 +22,8 @@ from opentaxii.signals import ( if "OPENTAXII_CONFIG" in os.environ: print("Using config from {}".format(os.environ["OPENTAXII_CONFIG"])) CONFIG = yaml.load(open(os.environ["OPENTAXII_CONFIG"], "r"), Loader=Loader) + if "dedup" not in CONFIG["misp"]: + CONFIG["misp"]["dedup"] = "UNKNOWN" else: print("Trying to use env variables...") if "MISP_URL" in os.environ: @@ -70,7 +72,11 @@ def post_stix(manager, content_block, collection_ids, service_id): log.info("STIX loaded succesfully.") values = [x.value for x in package.attributes] log.info("Extracted %s", values) - if CONFIG['MISP_DEDUP'] == "true" or CONFIG['MISP_DEDUP'] == "True" or CONFIG['MISP_DEDUP'] == "TRUE" or CONFIG['MISP_DEDUP'] == "UNKNOWN": + if ( + CONFIG["misp"]["dedup"] == True or + CONFIG["misp"]["dedup"] == "True" or + CONFIG["misp"]["dedup"] == "UNKNOWN" + ): for attrib in values: log.info("Checking for existence of %s", attrib) search = MISP.search("attributes", values=str(attrib))