From 060234624981b74a782466a1457527f4a0da2cb0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E1=B4=8B=E1=B4=87=CA=8F?= <24655118+gh0stkey@users.noreply.github.com> Date: Thu, 19 Oct 2023 22:43:29 +0800 Subject: [PATCH] Version: 2.5.2 Update --- src/main/java/burp/rule/utils/YamlTool.java | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/main/java/burp/rule/utils/YamlTool.java b/src/main/java/burp/rule/utils/YamlTool.java index 08de7be..54b4c25 100644 --- a/src/main/java/burp/rule/utils/YamlTool.java +++ b/src/main/java/burp/rule/utils/YamlTool.java @@ -1,8 +1,9 @@ package burp.rule.utils; +import java.nio.file.Files; +import java.nio.file.Paths; import java.util.Map; -import burp.BurpExtender; import org.yaml.snakeyaml.DumperOptions; import org.yaml.snakeyaml.Yaml; import java.io.FileInputStream; @@ -25,13 +26,11 @@ public class YamlTool { public static Map loadYaml(String filePath) { try { - InputStream inputStream = new FileInputStream(filePath); - Yaml yaml = newStandardYaml(); - return yaml.load(inputStream); - } catch (FileNotFoundException e) { + InputStream inputStream = Files.newInputStream(Paths.get(filePath)); + return newStandardYaml().load(inputStream); + } catch (Exception e) { e.printStackTrace(); return null; } } } -