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
76 lines
1.7 KiB
Ruby
76 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
require_relative 'helper'
|
|
|
|
module Psych
|
|
class TestObjectReferences < TestCase
|
|
def test_range_has_references
|
|
assert_reference_trip 1..2
|
|
end
|
|
|
|
def test_module_has_references
|
|
assert_reference_trip Psych
|
|
end
|
|
|
|
def test_class_has_references
|
|
assert_reference_trip TestObjectReferences
|
|
end
|
|
|
|
def test_rational_has_references
|
|
assert_reference_trip Rational('1.2')
|
|
end
|
|
|
|
def test_complex_has_references
|
|
assert_reference_trip Complex(1, 2)
|
|
end
|
|
|
|
def test_datetime_has_references
|
|
assert_reference_trip DateTime.now
|
|
end
|
|
|
|
def test_struct_has_references
|
|
assert_reference_trip Struct.new(:foo).new(1)
|
|
end
|
|
|
|
def assert_reference_trip obj
|
|
yml = Psych.dump([obj, obj])
|
|
assert_match(/\*-?\d+/, yml)
|
|
begin
|
|
data = Psych.load yml
|
|
rescue Psych::DisallowedClass
|
|
data = Psych.unsafe_load yml
|
|
end
|
|
assert_equal data.first.object_id, data.last.object_id
|
|
end
|
|
|
|
def test_float_references
|
|
data = Psych.unsafe_load <<-eoyml
|
|
---\s
|
|
- &name 1.2
|
|
- *name
|
|
eoyml
|
|
assert_equal data.first, data.last
|
|
assert_equal data.first.object_id, data.last.object_id
|
|
end
|
|
|
|
def test_binary_references
|
|
data = Psych.unsafe_load <<-eoyml
|
|
---
|
|
- &name !binary |-
|
|
aGVsbG8gd29ybGQh
|
|
- *name
|
|
eoyml
|
|
assert_equal data.first, data.last
|
|
assert_equal data.first.object_id, data.last.object_id
|
|
end
|
|
|
|
def test_regexp_references
|
|
data = Psych.unsafe_load <<-eoyml
|
|
---\s
|
|
- &name !ruby/regexp /pattern/i
|
|
- *name
|
|
eoyml
|
|
assert_equal data.first, data.last
|
|
assert_equal data.first.object_id, data.last.object_id
|
|
end
|
|
end
|
|
end
|