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
50 lines
1 KiB
Ruby
50 lines
1 KiB
Ruby
# frozen_string_literal: true
|
|
require_relative 'helper'
|
|
|
|
class PsychStructWithIvar < Struct.new(:foo)
|
|
attr_reader :bar
|
|
def initialize *args
|
|
super
|
|
@bar = 'hello'
|
|
end
|
|
end
|
|
|
|
module Psych
|
|
class TestStruct < TestCase
|
|
class StructSubclass < Struct.new(:foo)
|
|
def initialize foo, bar
|
|
super(foo)
|
|
@bar = bar
|
|
end
|
|
end
|
|
|
|
def test_self_referential_struct
|
|
ss = StructSubclass.new(nil, 'foo')
|
|
ss.foo = ss
|
|
|
|
loaded = Psych.unsafe_load(Psych.dump(ss))
|
|
assert_instance_of(StructSubclass, loaded.foo)
|
|
|
|
assert_equal(ss, loaded)
|
|
end
|
|
|
|
def test_roundtrip
|
|
thing = PsychStructWithIvar.new('bar')
|
|
struct = Psych.unsafe_load(Psych.dump(thing))
|
|
|
|
assert_equal 'hello', struct.bar
|
|
assert_equal 'bar', struct.foo
|
|
end
|
|
|
|
def test_load
|
|
obj = Psych.unsafe_load(<<-eoyml)
|
|
--- !ruby/struct:PsychStructWithIvar
|
|
:foo: bar
|
|
:@bar: hello
|
|
eoyml
|
|
|
|
assert_equal 'hello', obj.bar
|
|
assert_equal 'bar', obj.foo
|
|
end
|
|
end
|
|
end
|