diff --git a/misp_taxii_hooks/hooks.py b/misp_taxii_hooks/hooks.py index 70b5a36..4d863ec 100644 --- a/misp_taxii_hooks/hooks.py +++ b/misp_taxii_hooks/hooks.py @@ -77,15 +77,18 @@ def post_stix(manager, content_block, collection_ids, service_id): # make sure collections, if specified are supposed to be sent to if CONFIG["misp"]["collections"] != "UNKNOWN" or CONFIG["misp"]["collections"] == False: should_send_to_misp = False + collection_names = [collection.name for collection in manager.get_collections(service_id)] for collection in CONFIG["misp"]["collections"]: - if collection in collection_ids: + if collection in collection_names or collection in collection_ids: should_send_to_misp = True if should_send_to_misp == False: log.info('''No collections match misp.collections; aborting MISP extraction. Collection ids whitelisted: {} - Collection ids sent to: {}'''.format( + Collection ids sent to: {} + Collection names sent to: {}'''.format( CONFIG["misp"]["collections"], - collection_ids + collection_ids, + collection_names )) return None