mirror of
				https://github.com/ruby/ruby.git
				synced 2022-11-09 12:17:21 -05:00 
			
		
		
		
	 360a865246
			
		
	
	
		360a865246
		
	
	
	
	
		
			
			* marshal.c (w_symbol): fix symlink index. register the given symbols, not its name. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@46993 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
		
			
				
	
	
		
			639 lines
		
	
	
	
		
			16 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			639 lines
		
	
	
	
		
			16 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| require 'test/unit'
 | |
| require 'tempfile'
 | |
| require_relative 'marshaltestlib'
 | |
| 
 | |
| class TestMarshal < Test::Unit::TestCase
 | |
|   include MarshalTestLib
 | |
| 
 | |
|   def setup
 | |
|     @verbose = $VERBOSE
 | |
|     $VERBOSE = nil
 | |
|   end
 | |
| 
 | |
|   def teardown
 | |
|     $VERBOSE = @verbose
 | |
|   end
 | |
| 
 | |
|   def encode(o)
 | |
|     Marshal.dump(o)
 | |
|   end
 | |
| 
 | |
|   def decode(s)
 | |
|     Marshal.load(s)
 | |
|   end
 | |
| 
 | |
|   def fact(n)
 | |
|     return 1 if n == 0
 | |
|     f = 1
 | |
|     while n>0
 | |
|       f *= n
 | |
|       n -= 1
 | |
|     end
 | |
|     return f
 | |
|   end
 | |
| 
 | |
|   def test_marshal
 | |
|     a = [1, 2, 3, [4,5,"foo"], {1=>"bar"}, 2.5, fact(30)]
 | |
|     assert_equal a, Marshal.load(Marshal.dump(a))
 | |
| 
 | |
|     [[1,2,3,4], [81, 2, 118, 3146]].each { |w,x,y,z|
 | |
|       obj = (x.to_f + y.to_f / z.to_f) * Math.exp(w.to_f / (x.to_f + y.to_f / z.to_f))
 | |
|       assert_equal obj, Marshal.load(Marshal.dump(obj))
 | |
|     }
 | |
| 
 | |
|     bug3659 = '[ruby-dev:41936]'
 | |
|     [1.0, 10.0, 100.0, 110.0].each {|x|
 | |
|       assert_equal(x, Marshal.load(Marshal.dump(x)), bug3659)
 | |
|     }
 | |
|   end
 | |
| 
 | |
|   StrClone = String.clone
 | |
|   def test_marshal_cloned_class
 | |
|     assert_instance_of(StrClone, Marshal.load(Marshal.dump(StrClone.new("abc"))))
 | |
|   end
 | |
| 
 | |
|   def test_inconsistent_struct
 | |
|     TestMarshal.const_set :StructOrNot, Struct.new(:a)
 | |
|     s = Marshal.dump(StructOrNot.new(1))
 | |
|     TestMarshal.instance_eval { remove_const :StructOrNot }
 | |
|     TestMarshal.const_set :StructOrNot, Class.new
 | |
|     assert_raise(TypeError, "[ruby-dev:31709]") { Marshal.load(s) }
 | |
|   end
 | |
| 
 | |
|   def test_struct_invalid_members
 | |
|     TestMarshal.const_set :StructInvalidMembers, Struct.new(:a)
 | |
|     assert_raise(TypeError, "[ruby-dev:31759]") {
 | |
|       Marshal.load("\004\bIc&TestMarshal::StructInvalidMembers\006:\020__members__\"\bfoo")
 | |
|       TestMarshal::StructInvalidMembers.members
 | |
|     }
 | |
|   end
 | |
| 
 | |
|   class C
 | |
|     def initialize(str)
 | |
|       @str = str
 | |
|     end
 | |
|     attr_reader :str
 | |
|     def _dump(limit)
 | |
|       @str
 | |
|     end
 | |
|     def self._load(s)
 | |
|       new(s)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_too_long_string
 | |
|     data = Marshal.dump(C.new("a".force_encoding("ascii-8bit")))
 | |
|     data[-2, 1] = "\003\377\377\377"
 | |
|     assert_raise_with_message(ArgumentError, "marshal data too short", "[ruby-dev:32054]") {
 | |
|       Marshal.load(data)
 | |
|     }
 | |
|   end
 | |
| 
 | |
| 
 | |
|   def test_userdef_encoding
 | |
|     s1 = "\xa4\xa4".force_encoding("euc-jp")
 | |
|     o1 = C.new(s1)
 | |
|     m = Marshal.dump(o1)
 | |
|     o2 = Marshal.load(m)
 | |
|     s2 = o2.str
 | |
|     assert_equal(s1, s2)
 | |
|   end
 | |
| 
 | |
|   def test_pipe
 | |
|     o1 = C.new("a" * 10000)
 | |
| 
 | |
|     IO.pipe do |r, w|
 | |
|       th = Thread.new {Marshal.dump(o1, w)}
 | |
|       o2 = Marshal.load(r)
 | |
|       th.join
 | |
|       assert_equal(o1.str, o2.str)
 | |
|     end
 | |
| 
 | |
|     IO.pipe do |r, w|
 | |
|       th = Thread.new {Marshal.dump(o1, w, 2)}
 | |
|       o2 = Marshal.load(r)
 | |
|       th.join
 | |
|       assert_equal(o1.str, o2.str)
 | |
|     end
 | |
| 
 | |
|     assert_raise(TypeError) { Marshal.dump("foo", Object.new) }
 | |
|     assert_raise(TypeError) { Marshal.load(Object.new) }
 | |
|   end
 | |
| 
 | |
|   def test_limit
 | |
|     assert_equal([[[]]], Marshal.load(Marshal.dump([[[]]], 3)))
 | |
|     assert_raise(ArgumentError) { Marshal.dump([[[]]], 2) }
 | |
|     assert_nothing_raised(ArgumentError, '[ruby-core:24100]') { Marshal.dump("\u3042", 1) }
 | |
|   end
 | |
| 
 | |
|   def test_userdef_invalid
 | |
|     o = C.new(nil)
 | |
|     assert_raise(TypeError) { Marshal.dump(o) }
 | |
|   end
 | |
| 
 | |
|   def test_class
 | |
|     o = class << Object.new; self; end
 | |
|     assert_raise(TypeError) { Marshal.dump(o) }
 | |
|     assert_equal(Object, Marshal.load(Marshal.dump(Object)))
 | |
|     assert_equal(Enumerable, Marshal.load(Marshal.dump(Enumerable)))
 | |
|   end
 | |
| 
 | |
|   class C2
 | |
|     def initialize(ary)
 | |
|       @ary = ary
 | |
|     end
 | |
|     def _dump(s)
 | |
|       @ary.clear
 | |
|       "foo"
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_modify_array_during_dump
 | |
|     a = []
 | |
|     o = C2.new(a)
 | |
|     a << o << nil
 | |
|     assert_raise(RuntimeError) { Marshal.dump(a) }
 | |
|   end
 | |
| 
 | |
|   def test_change_class_name
 | |
|     eval("class C3; def _dump(s); 'foo'; end; end")
 | |
|     m = Marshal.dump(C3.new)
 | |
|     assert_raise(TypeError) { Marshal.load(m) }
 | |
|     eval("C3 = nil")
 | |
|     assert_raise(TypeError) { Marshal.load(m) }
 | |
|   end
 | |
| 
 | |
|   def test_change_struct
 | |
|     eval("C3 = Struct.new(:foo, :bar)")
 | |
|     m = Marshal.dump(C3.new("FOO", "BAR"))
 | |
|     eval("C3 = Struct.new(:foo)")
 | |
|     assert_raise(TypeError) { Marshal.load(m) }
 | |
|     eval("C3 = Struct.new(:foo, :baz)")
 | |
|     assert_raise(TypeError) { Marshal.load(m) }
 | |
|   end
 | |
| 
 | |
|   class C4
 | |
|     def initialize(gc)
 | |
|       @gc = gc
 | |
|     end
 | |
|     def _dump(s)
 | |
|       GC.start if @gc
 | |
|       "foo"
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_gc
 | |
|     assert_nothing_raised do
 | |
|       Marshal.dump((0..1000).map {|x| C4.new(x % 50 == 25) })
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_taint
 | |
|     x = Object.new
 | |
|     x.taint
 | |
|     s = Marshal.dump(x)
 | |
|     assert_equal(true, s.tainted?)
 | |
|     y = Marshal.load(s)
 | |
|     assert_equal(true, y.tainted?)
 | |
|   end
 | |
| 
 | |
|   def test_taint_each_object
 | |
|     x = Object.new
 | |
|     obj = [[x]]
 | |
| 
 | |
|     # clean object causes crean stream
 | |
|     assert_equal(false, obj.tainted?)
 | |
|     assert_equal(false, obj.first.tainted?)
 | |
|     assert_equal(false, obj.first.first.tainted?)
 | |
|     s = Marshal.dump(obj)
 | |
|     assert_equal(false, s.tainted?)
 | |
| 
 | |
|     # tainted object causes tainted stream
 | |
|     x.taint
 | |
|     assert_equal(false, obj.tainted?)
 | |
|     assert_equal(false, obj.first.tainted?)
 | |
|     assert_equal(true, obj.first.first.tainted?)
 | |
|     t = Marshal.dump(obj)
 | |
|     assert_equal(true, t.tainted?)
 | |
| 
 | |
|     # clean stream causes clean objects
 | |
|     assert_equal(false, s.tainted?)
 | |
|     y = Marshal.load(s)
 | |
|     assert_equal(false, y.tainted?)
 | |
|     assert_equal(false, y.first.tainted?)
 | |
|     assert_equal(false, y.first.first.tainted?)
 | |
| 
 | |
|     # tainted stream causes tainted objects
 | |
|     assert_equal(true, t.tainted?)
 | |
|     y = Marshal.load(t)
 | |
|     assert_equal(true, y.tainted?)
 | |
|     assert_equal(true, y.first.tainted?)
 | |
|     assert_equal(true, y.first.first.tainted?)
 | |
| 
 | |
|     # same tests by different senario
 | |
|     s.taint
 | |
|     assert_equal(true, s.tainted?)
 | |
|     y = Marshal.load(s)
 | |
|     assert_equal(true, y.tainted?)
 | |
|     assert_equal(true, y.first.tainted?)
 | |
|     assert_equal(true, y.first.first.tainted?)
 | |
|   end
 | |
| 
 | |
|   def test_symbol2
 | |
|     [:ruby, :"\u{7d05}\u{7389}"].each do |sym|
 | |
|       assert_equal(sym, Marshal.load(Marshal.dump(sym)), '[ruby-core:24788]')
 | |
|     end
 | |
|     bug2548 = '[ruby-core:27375]'
 | |
|     ary = [:$1, nil]
 | |
|     assert_equal(ary, Marshal.load(Marshal.dump(ary)), bug2548)
 | |
|   end
 | |
| 
 | |
|   def test_symlink
 | |
|     assert_include(Marshal.dump([:a, :a]), ';')
 | |
|   end
 | |
| 
 | |
|   ClassUTF8 = eval("class R\u{e9}sum\u{e9}; self; end")
 | |
| 
 | |
|   iso_8859_1 = Encoding::ISO_8859_1
 | |
| 
 | |
|   structISO8859_1 = Struct.new("r\xe9sum\xe9".force_encoding(iso_8859_1).intern)
 | |
|   const_set("R\xe9sum\xe9".force_encoding(iso_8859_1), structISO8859_1)
 | |
|   structISO8859_1.name
 | |
|   StructISO8859_1 = structISO8859_1
 | |
|   classISO8859_1 = Class.new do
 | |
|     attr_accessor "r\xe9sum\xe9".force_encoding(iso_8859_1)
 | |
|     eval("def initialize(x) @r\xe9sum\xe9 = x; end".force_encoding(iso_8859_1))
 | |
|   end
 | |
|   const_set("R\xe9sum\xe92".force_encoding(iso_8859_1), classISO8859_1)
 | |
|   classISO8859_1.name
 | |
|   ClassISO8859_1 = classISO8859_1
 | |
| 
 | |
|   def test_class_nonascii
 | |
|     a = ClassUTF8.new
 | |
|     assert_instance_of(ClassUTF8, Marshal.load(Marshal.dump(a)), '[ruby-core:24790]')
 | |
| 
 | |
|     bug1932 = '[ruby-core:24882]'
 | |
| 
 | |
|     a = StructISO8859_1.new(10)
 | |
|     assert_nothing_raised(bug1932) do
 | |
|       assert_equal(a, Marshal.load(Marshal.dump(a)), bug1932)
 | |
|     end
 | |
|     a.__send__("#{StructISO8859_1.members[0]}=", a)
 | |
|     assert_nothing_raised(bug1932) do
 | |
|       assert_equal(a, Marshal.load(Marshal.dump(a)), bug1932)
 | |
|     end
 | |
| 
 | |
|     a = ClassISO8859_1.new(10)
 | |
|     assert_nothing_raised(bug1932) do
 | |
|       b = Marshal.load(Marshal.dump(a))
 | |
|       assert_equal(ClassISO8859_1, b.class, bug1932)
 | |
|       assert_equal(a.instance_variables, b.instance_variables, bug1932)
 | |
|       a.instance_variables.each do |i|
 | |
|         assert_equal(a.instance_variable_get(i), b.instance_variable_get(i), bug1932)
 | |
|       end
 | |
|     end
 | |
|     a.__send__(a.methods(true).grep(/=\z/)[0], a)
 | |
|     assert_nothing_raised(bug1932) do
 | |
|       b = Marshal.load(Marshal.dump(a))
 | |
|       assert_equal(ClassISO8859_1, b.class, bug1932)
 | |
|       assert_equal(a.instance_variables, b.instance_variables, bug1932)
 | |
|       assert_equal(b, b.instance_variable_get(a.instance_variables[0]), bug1932)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_regexp2
 | |
|     assert_equal(/\\u/, Marshal.load("\004\b/\b\\\\u\000"))
 | |
|     assert_equal(/u/, Marshal.load("\004\b/\a\\u\000"))
 | |
|     assert_equal(/u/, Marshal.load("\004\bI/\a\\u\000\006:\016@encoding\"\vEUC-JP"))
 | |
| 
 | |
|     bug2109 = '[ruby-core:25625]'
 | |
|     a = "\x82\xa0".force_encoding(Encoding::Windows_31J)
 | |
|     b = "\x82\xa2".force_encoding(Encoding::Windows_31J)
 | |
|     c = [/#{a}/, /#{b}/]
 | |
|     assert_equal(c, Marshal.load(Marshal.dump(c)), bug2109)
 | |
| 
 | |
|     assert_nothing_raised(ArgumentError, '[ruby-dev:40386]') do
 | |
|       re = Tempfile.create("marshal_regexp") do |f|
 | |
|         f.binmode.write("\x04\bI/\x00\x00\x06:\rencoding\"\rUS-ASCII")
 | |
|         f.rewind
 | |
|         re2 = Marshal.load(f)
 | |
|         re2
 | |
|       end
 | |
|       assert_equal(//, re)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   class DumpTest
 | |
|     def marshal_dump
 | |
|       @@block.call(:marshal_dump)
 | |
|     end
 | |
| 
 | |
|     def dump_each(&block)
 | |
|       @@block = block
 | |
|       Marshal.dump(self)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   class LoadTest
 | |
|     def marshal_dump
 | |
|       nil
 | |
|     end
 | |
|     def marshal_load(obj)
 | |
|       @@block.call(:marshal_load)
 | |
|     end
 | |
|     def self.load_each(m, &block)
 | |
|       @@block = block
 | |
|       Marshal.load(m)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_context_switch
 | |
|     o = DumpTest.new
 | |
|     e = o.enum_for(:dump_each)
 | |
|     assert_equal(:marshal_dump, e.next)
 | |
|     GC.start
 | |
|     assert(true, '[ruby-dev:39425]')
 | |
|     assert_raise(StopIteration) {e.next}
 | |
| 
 | |
|     o = LoadTest.new
 | |
|     m = Marshal.dump(o)
 | |
|     e = LoadTest.enum_for(:load_each, m)
 | |
|     assert_equal(:marshal_load, e.next)
 | |
|     GC.start
 | |
|     assert(true, '[ruby-dev:39425]')
 | |
|     assert_raise(StopIteration) {e.next}
 | |
|   end
 | |
| 
 | |
|   def test_dump_buffer
 | |
|     bug2390 = '[ruby-dev:39744]'
 | |
|     w = ""
 | |
|     def w.write(str)
 | |
|       self << str.to_s
 | |
|     end
 | |
|     Marshal.dump(Object.new, w)
 | |
|     assert_not_empty(w, bug2390)
 | |
|   end
 | |
| 
 | |
|   class C5
 | |
|     def marshal_dump
 | |
|       "foo"
 | |
|     end
 | |
|     def marshal_load(foo)
 | |
|       @foo = foo
 | |
|     end
 | |
|     def initialize(x)
 | |
|       @x = x
 | |
|     end
 | |
|   end
 | |
|   def test_marshal_dump
 | |
|     c = C5.new("bar")
 | |
|     s = Marshal.dump(c)
 | |
|     d = Marshal.load(s)
 | |
|     assert_equal("foo", d.instance_variable_get(:@foo))
 | |
|     assert_equal(false, d.instance_variable_defined?(:@x))
 | |
|   end
 | |
| 
 | |
|   class C6
 | |
|     def initialize
 | |
|       @stdin = STDIN
 | |
|     end
 | |
|     attr_reader :stdin
 | |
|     def marshal_dump
 | |
|       1
 | |
|     end
 | |
|     def marshal_load(x)
 | |
|       @stdin = STDIN
 | |
|     end
 | |
|   end
 | |
|   def test_marshal_dump_extra_iv
 | |
|     o = C6.new
 | |
|     m = nil
 | |
|     assert_nothing_raised("[ruby-dev:21475] [ruby-dev:39845]") {
 | |
|       m = Marshal.dump(o)
 | |
|     }
 | |
|     o2 = Marshal.load(m)
 | |
|     assert_equal(STDIN, o2.stdin)
 | |
|   end
 | |
| 
 | |
|   def test_marshal_string_encoding
 | |
|     o1 = ["foo".force_encoding("EUC-JP")] + [ "bar" ] * 2
 | |
|     m = Marshal.dump(o1)
 | |
|     o2 = Marshal.load(m)
 | |
|     assert_equal(o1, o2, "[ruby-dev:40388]")
 | |
|   end
 | |
| 
 | |
|   def test_marshal_regexp_encoding
 | |
|     o1 = [Regexp.new("r1".force_encoding("EUC-JP"))] + ["r2"] * 2
 | |
|     m = Marshal.dump(o1)
 | |
|     o2 = Marshal.load(m)
 | |
|     assert_equal(o1, o2, "[ruby-dev:40416]")
 | |
|   end
 | |
| 
 | |
|   def test_marshal_encoding_encoding
 | |
|     o1 = [Encoding.find("EUC-JP")] + ["r2"] * 2
 | |
|     m = Marshal.dump(o1)
 | |
|     o2 = Marshal.load(m)
 | |
|     assert_equal(o1, o2)
 | |
|   end
 | |
| 
 | |
|   def test_marshal_symbol_ascii8bit
 | |
|     bug6209 = '[ruby-core:43762]'
 | |
|     o1 = "\xff".force_encoding("ASCII-8BIT").intern
 | |
|     m = Marshal.dump(o1)
 | |
|     o2 = nil
 | |
|     assert_nothing_raised(EncodingError, bug6209) {o2 = Marshal.load(m)}
 | |
|     assert_equal(o1, o2, bug6209)
 | |
|   end
 | |
| 
 | |
|   class PrivateClass
 | |
|     def initialize(foo)
 | |
|       @foo = foo
 | |
|     end
 | |
|     attr_reader :foo
 | |
|   end
 | |
|   private_constant :PrivateClass
 | |
| 
 | |
|   def test_marshal_private_class
 | |
|     o1 = PrivateClass.new("test")
 | |
|     o2 = Marshal.load(Marshal.dump(o1))
 | |
|     assert_equal(o1.class, o2.class)
 | |
|     assert_equal(o1.foo, o2.foo)
 | |
|   end
 | |
| 
 | |
|   def test_marshal_complex
 | |
|     assert_raise(ArgumentError){Marshal.load("\x04\bU:\fComplex[\x05")}
 | |
|     assert_raise(ArgumentError){Marshal.load("\x04\bU:\fComplex[\x06i\x00")}
 | |
|     assert_equal(Complex(1, 2), Marshal.load("\x04\bU:\fComplex[\ai\x06i\a"))
 | |
|     assert_raise(ArgumentError){Marshal.load("\x04\bU:\fComplex[\bi\x00i\x00i\x00")}
 | |
|   end
 | |
| 
 | |
|   def test_marshal_rational
 | |
|     assert_raise(ArgumentError){Marshal.load("\x04\bU:\rRational[\x05")}
 | |
|     assert_raise(ArgumentError){Marshal.load("\x04\bU:\rRational[\x06i\x00")}
 | |
|     assert_equal(Rational(1, 2), Marshal.load("\x04\bU:\rRational[\ai\x06i\a"))
 | |
|     assert_raise(ArgumentError){Marshal.load("\x04\bU:\rRational[\bi\x00i\x00i\x00")}
 | |
|   end
 | |
| 
 | |
|   def test_marshal_flonum_reference
 | |
|     bug7348 = '[ruby-core:49323]'
 | |
|     e = []
 | |
|     ary = [ [2.0, e], [e] ]
 | |
|     assert_equal(ary, Marshal.load(Marshal.dump(ary)), bug7348)
 | |
|   end
 | |
| 
 | |
|   class TestClass
 | |
|   end
 | |
| 
 | |
|   module TestModule
 | |
|   end
 | |
| 
 | |
|   def test_marshal_load_should_not_taint_classes
 | |
|     bug7325 = '[ruby-core:49198]'
 | |
|     for c in [TestClass, TestModule]
 | |
|       assert_not_predicate(c, :tainted?)
 | |
|       c2 = Marshal.load(Marshal.dump(c).taint)
 | |
|       assert_same(c, c2)
 | |
|       assert_not_predicate(c, :tainted?, bug7325)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   class Bug7627 < Struct.new(:bar)
 | |
|     attr_accessor :foo
 | |
| 
 | |
|     def marshal_dump; 'dump'; end  # fake dump data
 | |
|     def marshal_load(*); end       # do nothing
 | |
|   end
 | |
| 
 | |
|   def test_marshal_dump_struct_ivar
 | |
|     bug7627 = '[ruby-core:51163]'
 | |
|     obj = Bug7627.new
 | |
|     obj.foo = '[Bug #7627]'
 | |
| 
 | |
|     dump   = Marshal.dump(obj)
 | |
|     loaded = Marshal.load(dump)
 | |
| 
 | |
|     assert_equal(obj, loaded, bug7627)
 | |
|     assert_nil(loaded.foo, bug7627)
 | |
|   end
 | |
| 
 | |
|   class LoadData
 | |
|     attr_reader :data
 | |
|     def initialize(data)
 | |
|       @data = data
 | |
|     end
 | |
|     alias marshal_dump data
 | |
|     alias marshal_load initialize
 | |
|   end
 | |
| 
 | |
|   class Bug8276 < LoadData
 | |
|     def initialize(*)
 | |
|       super
 | |
|       freeze
 | |
|     end
 | |
|     alias marshal_load initialize
 | |
|   end
 | |
| 
 | |
|   class FrozenData < LoadData
 | |
|     def marshal_load(data)
 | |
|       super
 | |
|       data.instance_variables.each do |iv|
 | |
|         instance_variable_set(iv, data.instance_variable_get(iv))
 | |
|       end
 | |
|       freeze
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_marshal_dump_excess_encoding
 | |
|     bug8276 = '[ruby-core:54334] [Bug #8276]'
 | |
|     t = Bug8276.new(bug8276)
 | |
|     s = Marshal.dump(t)
 | |
|     assert_nothing_raised(RuntimeError, bug8276) {s = Marshal.load(s)}
 | |
|     assert_equal(t.data, s.data, bug8276)
 | |
|   end
 | |
| 
 | |
|   def test_marshal_dump_ivar
 | |
|     s = "data with ivar"
 | |
|     s.instance_variable_set(:@t, 42)
 | |
|     t = Bug8276.new(s)
 | |
|     s = Marshal.dump(t)
 | |
|     assert_raise(RuntimeError) {Marshal.load(s)}
 | |
|   end
 | |
| 
 | |
|   def test_marshal_load_ivar
 | |
|     s = "data with ivar"
 | |
|     s.instance_variable_set(:@t, 42)
 | |
|     hook = ->(v) {
 | |
|       if LoadData === v
 | |
|         assert_send([v, :instance_variable_defined?, :@t], v.class.name)
 | |
|         assert_equal(42, v.instance_variable_get(:@t), v.class.name)
 | |
|       end
 | |
|       v
 | |
|     }
 | |
|     [LoadData, FrozenData].each do |klass|
 | |
|       t = klass.new(s)
 | |
|       d = Marshal.dump(t)
 | |
|       v = assert_nothing_raised(RuntimeError) {break Marshal.load(d, hook)}
 | |
|       assert_send([v, :instance_variable_defined?, :@t], klass.name)
 | |
|       assert_equal(42, v.instance_variable_get(:@t), klass.name)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_class_ivar
 | |
|     assert_raise(TypeError) {Marshal.load("\x04\x08Ic\x1bTestMarshal::TestClass\x06:\x0e@ivar_bug\"\x08bug")}
 | |
|     assert_raise(TypeError) {Marshal.load("\x04\x08IM\x1bTestMarshal::TestClass\x06:\x0e@ivar_bug\"\x08bug")}
 | |
|     assert_not_operator(TestClass, :instance_variable_defined?, :@bug)
 | |
|   end
 | |
| 
 | |
|   def test_module_ivar
 | |
|     assert_raise(TypeError) {Marshal.load("\x04\x08Im\x1cTestMarshal::TestModule\x06:\x0e@ivar_bug\"\x08bug")}
 | |
|     assert_raise(TypeError) {Marshal.load("\x04\x08IM\x1cTestMarshal::TestModule\x06:\x0e@ivar_bug\"\x08bug")}
 | |
|     assert_not_operator(TestModule, :instance_variable_defined?, :@bug)
 | |
|   end
 | |
| 
 | |
|   class TestForRespondToFalse
 | |
|     def respond_to?(a)
 | |
|       false
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_marshal_respond_to_arity
 | |
|     assert_nothing_raised(ArgumentError, '[Bug #7722]') do
 | |
|       Marshal.dump(TestForRespondToFalse.new)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_packed_string
 | |
|     packed = ["foo"].pack("p")
 | |
|     bare = "".force_encoding(Encoding::ASCII_8BIT) << packed
 | |
|     assert_equal(Marshal.dump(bare), Marshal.dump(packed))
 | |
|   end
 | |
| 
 | |
|   def test_untainted_numeric
 | |
|     bug8945 = '[ruby-core:57346] [Bug #8945] Numerics never be tainted'
 | |
|     b = 1 << 32
 | |
|     b *= b until Bignum === b
 | |
|     tainted = [0, 1.0, 1.72723e-77, b].select do |x|
 | |
|       Marshal.load(Marshal.dump(x).taint).tainted?
 | |
|     end
 | |
|     assert_empty(tainted.map {|x| [x, x.class]}, bug8945)
 | |
|   end
 | |
| 
 | |
|   class Bug9523
 | |
|     attr_reader :cc
 | |
|     def marshal_dump
 | |
|       callcc {|c| @cc = c }
 | |
|       nil
 | |
|     end
 | |
|     def marshal_load(v)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def test_continuation
 | |
|     require "continuation"
 | |
|     c = Bug9523.new
 | |
|     assert_raise_with_message(RuntimeError, /Marshal\.dump reentered at marshal_dump/) do
 | |
|       Marshal.dump(c)
 | |
|       c.cc.call
 | |
|     end
 | |
|   end
 | |
| end
 |