From fb45e0e0c91e6d78f2bdb452c97eb66c9fe738c3 Mon Sep 17 00:00:00 2001 From: Hiroshi Shimoju Date: Mon, 4 Sep 2023 20:14:09 +0900 Subject: [PATCH] Use YAML.unsafe_load when it is available --- lib/prmd/multi_loader/yaml.rb | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/lib/prmd/multi_loader/yaml.rb b/lib/prmd/multi_loader/yaml.rb index 6f7445e6..99845358 100644 --- a/lib/prmd/multi_loader/yaml.rb +++ b/lib/prmd/multi_loader/yaml.rb @@ -9,7 +9,11 @@ module Yaml # @see (Prmd::MultiLoader::Loader#load_data) def self.load_data(data) - ::YAML.load(data) + if ::YAML.respond_to?(:unsafe_load) + ::YAML.unsafe_load(data) + else + ::YAML.load(data) + end end # register this loader for all .yaml and .yml files