From 8ea464a9c1b94bbe1f697bfd6c03e87411272701 Mon Sep 17 00:00:00 2001 From: hsbt Date: Sat, 20 Oct 2018 02:51:09 +0000 Subject: Support keyword arguments with Pysch.safe_load provided by after Psych 3.1.0.pre1. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@65192 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- lib/rubygems/safe_yaml.rb | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'lib/rubygems/safe_yaml.rb') diff --git a/lib/rubygems/safe_yaml.rb b/lib/rubygems/safe_yaml.rb index 789bb5e25a..949c7b4754 100644 --- a/lib/rubygems/safe_yaml.rb +++ b/lib/rubygems/safe_yaml.rb @@ -28,11 +28,19 @@ module Gem if ::YAML.respond_to? :safe_load def self.safe_load input - ::YAML.safe_load(input, WHITELISTED_CLASSES, WHITELISTED_SYMBOLS, true) + if Gem::Version.new(Psych::VERSION) >= Gem::Version.new('3.1.0.pre1') + ::YAML.safe_load(input, whitelist_classes: WHITELISTED_CLASSES, whitelist_symbols: WHITELISTED_SYMBOLS, aliases: true) + else + ::YAML.safe_load(input, WHITELISTED_CLASSES, WHITELISTED_SYMBOLS, true) + end end def self.load input - ::YAML.safe_load(input, [::Symbol]) + if Gem::Version.new(Psych::VERSION) >= Gem::Version.new('3.1.0.pre1') + ::YAML.safe_load(input, whitelist_classes: [::Symbol]) + else + ::YAML.safe_load(input, [::Symbol]) + end end else unless Gem::Deprecate.skip -- cgit v1.2.3