From: Jason Ish <ish@unx.ca>
Date: Tue, 28 Nov 2017 22:50:17 +0000 (-0600)
Subject: et pro: mask the secret code
X-Git-Tag: 1.0.0a1~38
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=refs%2Fpull%2F10%2Fhead;p=thirdparty%2Fsuricata-update.git

et pro: mask the secret code
---

diff --git a/suricata/update/main.py b/suricata/update/main.py
index f427fef..219bdaf 100644
--- a/suricata/update/main.py
+++ b/suricata/update/main.py
@@ -45,8 +45,8 @@ if sys.argv[0] == __file__:
 import suricata.update.rule
 import suricata.update.engine
 import suricata.update.net
+import suricata.update.loghandler
 from suricata.update import configs
-from suricata.update.loghandler import SuriColourLogHandler
 from suricata.update import extract
 from suricata.update import util
 
@@ -54,7 +54,7 @@ from suricata.update import util
 if len(logging.root.handlers) == 0 and os.isatty(sys.stderr.fileno()):
     logger = logging.getLogger()
     logger.setLevel(level=logging.INFO)
-    logger.addHandler(SuriColourLogHandler())
+    logger.addHandler(suricata.update.loghandler.SuriColourLogHandler())
 else:
     logging.basicConfig(
         level=logging.INFO,
@@ -950,6 +950,7 @@ def load_sources(config, suricata_version):
                     code = source["code"]
                 else:
                     code = config.get("etpro")
+                suricata.update.loghandler.add_secret(code, "code")
                 if not code:
                     logger.error("ET-Pro source specified without code: %s",
                                  str(source))