From 67f9c52deff425e9db415f83bf794fbe4bdc9202 Mon Sep 17 00:00:00 2001 From: haraksin Date: Mon, 15 Jun 2020 19:00:47 -0700 Subject: [PATCH] adding try/except for pymisp error --- misp_taxii_hooks/hooks.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/misp_taxii_hooks/hooks.py b/misp_taxii_hooks/hooks.py index c64218d..f0fe453 100644 --- a/misp_taxii_hooks/hooks.py +++ b/misp_taxii_hooks/hooks.py @@ -12,6 +12,7 @@ from pyaml import yaml from yaml import Loader from io import StringIO from requests.exceptions import ConnectionError +from pymisp.exceptions import PyMISPError logging_level = logging.INFO log = logging.getLogger("__main__") @@ -70,12 +71,14 @@ else: "collections": misp_collections } } - -MISP = pymisp.PyMISP( +try: + MISP = pymisp.PyMISP( CONFIG["misp"]["url"], CONFIG["misp"]["api"], ssl = CONFIG["misp"].get("verifySSL", True) ) +except PyMISPError: + log.error("Cannot connect to MISP; please ensure that MISP is up and running at {}. Skipping MISP upload.".format(CONFIG['misp']['url'])) def post_stix(manager, content_block, collection_ids, service_id): ''' @@ -151,7 +154,7 @@ def post_stix(manager, content_block, collection_ids, service_id): try: event = MISP.add_event(package) except ConnectionError: - log.error("MISP-Taxii-Server - Cannot connect to MISP; please ensure that MISP is up and running at {}. Skipping MISP upload.".format(CONFIG['misp']['url'])) + log.error("Cannot push to MISP; please ensure that MISP is up and running at {}. Skipping MISP upload.".format(CONFIG['misp']['url'])) if ( CONFIG["misp"]["publish"] == True or CONFIG["misp"]["publish"] == "True"