summaryrefslogtreecommitdiffstats
path: root/func
diff options
context:
space:
mode:
authorSerenaFeng <feng.xiaowei@zte.com.cn>2016-09-20 14:59:28 +0800
committerSerenaFeng <feng.xiaowei@zte.com.cn>2016-09-20 14:59:28 +0800
commitfd3040c8eaa96ed14a75fbd34e009da1e59dd947 (patch)
tree5fc20bcaf3b3eb25e0250a937b8d6c70aedac04f /func
parentfac2a3bff97efed6d4e50bfe249c8107100b8f52 (diff)
Fix issues found in security review
JIRA: QTIP-105 Change-Id: I08108a982f9059fd19b92df0d219fa8f13f288ad Signed-off-by: SerenaFeng <feng.xiaowei@zte.com.cn>
Diffstat (limited to 'func')
-rw-r--r--func/env_setup.py2
-rw-r--r--func/spawn_vm.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/func/env_setup.py b/func/env_setup.py
index 96f984cb..3697f235 100644
--- a/func/env_setup.py
+++ b/func/env_setup.py
@@ -174,7 +174,7 @@ class Env_setup:
def parse(self, config_file_path):
try:
f_name = open(config_file_path, 'r+')
- doc = yaml.load(f_name)
+ doc = yaml.safe_load(f_name)
f_name.close()
if doc['Scenario']['benchmark']:
self.benchmark = doc['Scenario']['benchmark']
diff --git a/func/spawn_vm.py b/func/spawn_vm.py
index 15c26861..5710308b 100644
--- a/func/spawn_vm.py
+++ b/func/spawn_vm.py
@@ -65,7 +65,7 @@ class SpawnVM(Env_setup):
Heat_Dic = {}
try:
with open('./heat/SampleHeat.yaml', 'r+') as H_temp:
- Heat_Dic = yaml.load(H_temp)
+ Heat_Dic = yaml.safe_load(H_temp)
except yaml.YAMLError as exc:
if hasattr(exc, 'problem_mark'):
mark = exc.problem_mark