mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
59 lines
1.6 KiB
Ruby
59 lines
1.6 KiB
Ruby
module Gem
|
|
|
|
###
|
|
# This module is used for safely loading YAML specs from a gem. The
|
|
# `safe_load` method defined on this module is specifically designed for
|
|
# loading Gem specifications. For loading other YAML safely, please see
|
|
# Psych.safe_load
|
|
|
|
module SafeYAML
|
|
PERMITTED_CLASSES = %w[
|
|
Symbol
|
|
Time
|
|
Date
|
|
Gem::Dependency
|
|
Gem::Platform
|
|
Gem::Requirement
|
|
Gem::Specification
|
|
Gem::Version
|
|
Gem::Version::Requirement
|
|
YAML::Syck::DefaultKey
|
|
Syck::DefaultKey
|
|
].freeze
|
|
|
|
PERMITTED_SYMBOLS = %w[
|
|
development
|
|
runtime
|
|
].freeze
|
|
|
|
if ::YAML.respond_to? :safe_load
|
|
def self.safe_load(input)
|
|
if Gem::Version.new(Psych::VERSION) >= Gem::Version.new('3.1.0.pre1')
|
|
::YAML.safe_load(input, permitted_classes: PERMITTED_CLASSES, permitted_symbols: PERMITTED_SYMBOLS, aliases: true)
|
|
else
|
|
::YAML.safe_load(input, PERMITTED_CLASSES, PERMITTED_SYMBOLS, true)
|
|
end
|
|
end
|
|
|
|
def self.load(input)
|
|
if Gem::Version.new(Psych::VERSION) >= Gem::Version.new('3.1.0.pre1')
|
|
::YAML.safe_load(input, permitted_classes: [::Symbol])
|
|
else
|
|
::YAML.safe_load(input, [::Symbol])
|
|
end
|
|
end
|
|
else
|
|
unless Gem::Deprecate.skip
|
|
warn "YAML safe loading is not available. Please upgrade psych to a version that supports safe loading (>= 2.0)."
|
|
end
|
|
|
|
def self.safe_load(input, *args)
|
|
::YAML.load input
|
|
end
|
|
|
|
def self.load(input)
|
|
::YAML.load input
|
|
end
|
|
end
|
|
end
|
|
end
|