mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
c7c2ad5749
In future versions of Psych, the `load` method will be mostly the same as the `safe_load` method. In other words, the `load` method won't allow arbitrary object deserialization (which can be used to escalate to an RCE). People that need to load *trusted* documents can use the `unsafe_load` method. This commit introduces the `unsafe_load` method so that people can incrementally upgrade. For example, if they try to upgrade to 4.0.0 and something breaks, they can downgrade, audit callsites, change to `safe_load` or `unsafe_load` as required, and then upgrade to 4.0.0 smoothly. https://github.com/ruby/psych/commit/cb50aa8d3f
97 lines
2 KiB
Ruby
97 lines
2 KiB
Ruby
# frozen_string_literal: true
|
|
require_relative 'helper'
|
|
|
|
class ObjectWithInstanceVariables
|
|
attr_accessor :var1, :var2
|
|
end
|
|
|
|
class SubStringWithInstanceVariables < String
|
|
attr_accessor :var1
|
|
end
|
|
|
|
module Psych
|
|
class TestAliasAndAnchor < TestCase
|
|
def test_mri_compatibility
|
|
yaml = <<EOYAML
|
|
---
|
|
- &id001 !ruby/object {}
|
|
|
|
- *id001
|
|
- *id001
|
|
EOYAML
|
|
result = Psych.unsafe_load yaml
|
|
result.each {|el| assert_same(result[0], el) }
|
|
end
|
|
|
|
def test_mri_compatibility_object_with_ivars
|
|
yaml = <<EOYAML
|
|
---
|
|
- &id001 !ruby/object:ObjectWithInstanceVariables
|
|
var1: test1
|
|
var2: test2
|
|
- *id001
|
|
- *id001
|
|
EOYAML
|
|
|
|
result = Psych.unsafe_load yaml
|
|
result.each do |el|
|
|
assert_same(result[0], el)
|
|
assert_equal('test1', el.var1)
|
|
assert_equal('test2', el.var2)
|
|
end
|
|
end
|
|
|
|
def test_mri_compatibility_substring_with_ivars
|
|
yaml = <<EOYAML
|
|
---
|
|
- &id001 !str:SubStringWithInstanceVariables
|
|
str: test
|
|
"@var1": test
|
|
- *id001
|
|
- *id001
|
|
EOYAML
|
|
result = Psych.unsafe_load yaml
|
|
result.each do |el|
|
|
assert_same(result[0], el)
|
|
assert_equal('test', el.var1)
|
|
end
|
|
end
|
|
|
|
def test_anchor_alias_round_trip
|
|
o = Object.new
|
|
original = [o,o,o]
|
|
|
|
yaml = Psych.dump original
|
|
result = Psych.unsafe_load yaml
|
|
result.each {|el| assert_same(result[0], el) }
|
|
end
|
|
|
|
def test_anchor_alias_round_trip_object_with_ivars
|
|
o = ObjectWithInstanceVariables.new
|
|
o.var1 = 'test1'
|
|
o.var2 = 'test2'
|
|
original = [o,o,o]
|
|
|
|
yaml = Psych.dump original
|
|
result = Psych.unsafe_load yaml
|
|
result.each do |el|
|
|
assert_same(result[0], el)
|
|
assert_equal('test1', el.var1)
|
|
assert_equal('test2', el.var2)
|
|
end
|
|
end
|
|
|
|
def test_anchor_alias_round_trip_substring_with_ivars
|
|
o = SubStringWithInstanceVariables.new
|
|
o.var1 = 'test'
|
|
original = [o,o,o]
|
|
|
|
yaml = Psych.dump original
|
|
result = Psych.unsafe_load yaml
|
|
result.each do |el|
|
|
assert_same(result[0], el)
|
|
assert_equal('test', el.var1)
|
|
end
|
|
end
|
|
end
|
|
end
|