From 5cc5f99c45f626409ad9993632ec970111dbbd9e Mon Sep 17 00:00:00 2001 From: haraksin Date: Thu, 14 May 2020 16:38:28 -0700 Subject: [PATCH] Making sure collections returned from manager instance are only the ones we want --- misp_taxii_hooks/hooks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/misp_taxii_hooks/hooks.py b/misp_taxii_hooks/hooks.py index 89bf7da..9eccc7e 100644 --- a/misp_taxii_hooks/hooks.py +++ b/misp_taxii_hooks/hooks.py @@ -78,7 +78,7 @@ def post_stix(manager, content_block, collection_ids, service_id): if CONFIG["misp"]["collections"] != "UNKNOWN" or CONFIG["misp"]["collections"] == False: log.info("Using collections") should_send_to_misp = False - collection_names = [collection.name for collection in manager.get_collections(service_id)] + collection_names = [collection.name for collection in manager.get_collections(service_id) if collection.id in collection_ids] for collection in CONFIG["misp"]["collections"]: if collection in collection_names or collection in collection_ids: log.info("Collection specified matches push collection: {}".format(collection))