From b50a60485fe185ddcf5c587a284a3fa228bae673 Mon Sep 17 00:00:00 2001
From: haraksin <george.haraksin@laverne.edu>
Date: Wed, 6 May 2020 17:28:04 -0700
Subject: [PATCH] adding Loader to push_published_to_taxii adding local-server
 default yaml and remote-servers default yaml

---
 config/local-server.default.yaml   | 12 ++++++++++++
 config/remote-servers.default.yaml | 18 ++++++++++++++++++
 scripts/push_published_to_taxii.py |  3 ++-
 3 files changed, 32 insertions(+), 1 deletion(-)
 create mode 100644 config/local-server.default.yaml
 create mode 100644 config/remote-servers.default.yaml

diff --git a/config/local-server.default.yaml b/config/local-server.default.yaml
new file mode 100644
index 0000000..2d90b5f
--- /dev/null
+++ b/config/local-server.default.yaml
@@ -0,0 +1,12 @@
+host: localhost
+port: 9000
+discovery_path: /services/discovery
+inbox_path: /services/inbox
+use_https: False
+taxii_version: '1.1'
+headers:
+auth:
+  username: test
+  password: test
+collections:
+  - collection
\ No newline at end of file
diff --git a/config/remote-servers.default.yaml b/config/remote-servers.default.yaml
new file mode 100644
index 0000000..be7a211
--- /dev/null
+++ b/config/remote-servers.default.yaml
@@ -0,0 +1,18 @@
+- name: 'default'
+  host: localhost
+  port: 9000
+  discovery_path:
+  use_https: False
+  taxii_version: '1.1'
+  headers:
+  auth:
+    username:
+    password:
+    cacert_path:
+    cert_file:
+    key_file:
+    key_password:
+    jwt_auth_url:
+    verify_ssl: True
+  collections:
+    - collection
\ No newline at end of file
diff --git a/scripts/push_published_to_taxii.py b/scripts/push_published_to_taxii.py
index f8ca786..f465fe9 100644
--- a/scripts/push_published_to_taxii.py
+++ b/scripts/push_published_to_taxii.py
@@ -4,6 +4,7 @@ import sys
 import json
 import pymisp
 from pyaml import yaml
+from yaml import Loader
 from cabby import create_client
 from misp_stix_converter.converters import lint_roller
 import logging
@@ -20,7 +21,7 @@ log.setLevel(logging.DEBUG)
 log.info("Starting...")
 # Try to load in config
 if "OPENTAXII_CONFIG" in os.environ:
-    config = yaml.load(open(os.environ["OPENTAXII_CONFIG"], "r"))
+    config = yaml.load(open(os.environ["OPENTAXII_CONFIG"], "r"), Loader=Loader)
 else:
     print("OPENTAXII CONFIG NOT EXPORTED")
     sys.exit()