2015-12-16 00:07:31 -05:00
|
|
|
# frozen_string_literal: false
|
2003-09-04 12:18:59 -04:00
|
|
|
require 'test/unit'
|
2010-02-11 00:37:43 -05:00
|
|
|
require 'tempfile'
|
2008-02-18 01:59:47 -05:00
|
|
|
require_relative 'marshaltestlib'
|
2003-09-04 12:18:59 -04:00
|
|
|
|
2004-02-17 09:15:37 -05:00
|
|
|
class TestMarshal < Test::Unit::TestCase
|
|
|
|
include MarshalTestLib
|
|
|
|
|
2008-07-23 08:38:49 -04:00
|
|
|
def setup
|
|
|
|
@verbose = $VERBOSE
|
|
|
|
$VERBOSE = nil
|
|
|
|
end
|
|
|
|
|
|
|
|
def teardown
|
|
|
|
$VERBOSE = @verbose
|
|
|
|
end
|
|
|
|
|
2003-09-26 04:41:28 -04:00
|
|
|
def encode(o)
|
|
|
|
Marshal.dump(o)
|
|
|
|
end
|
|
|
|
|
|
|
|
def decode(s)
|
|
|
|
Marshal.load(s)
|
|
|
|
end
|
|
|
|
|
2003-09-04 12:18:59 -04:00
|
|
|
def fact(n)
|
|
|
|
return 1 if n == 0
|
|
|
|
f = 1
|
|
|
|
while n>0
|
|
|
|
f *= n
|
|
|
|
n -= 1
|
|
|
|
end
|
|
|
|
return f
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_marshal
|
2009-10-20 23:37:40 -04:00
|
|
|
a = [1, 2, 3, [4,5,"foo"], {1=>"bar"}, 2.5, fact(30)]
|
|
|
|
assert_equal a, Marshal.load(Marshal.dump(a))
|
2003-09-04 12:18:59 -04:00
|
|
|
|
|
|
|
[[1,2,3,4], [81, 2, 118, 3146]].each { |w,x,y,z|
|
2006-12-31 10:02:22 -05:00
|
|
|
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))
|
2003-09-04 12:18:59 -04:00
|
|
|
}
|
2010-08-05 05:25:39 -04:00
|
|
|
|
|
|
|
bug3659 = '[ruby-dev:41936]'
|
|
|
|
[1.0, 10.0, 100.0, 110.0].each {|x|
|
|
|
|
assert_equal(x, Marshal.load(Marshal.dump(x)), bug3659)
|
|
|
|
}
|
2003-09-04 12:18:59 -04:00
|
|
|
end
|
2006-12-31 10:02:22 -05:00
|
|
|
|
|
|
|
StrClone = String.clone
|
|
|
|
def test_marshal_cloned_class
|
|
|
|
assert_instance_of(StrClone, Marshal.load(Marshal.dump(StrClone.new("abc"))))
|
|
|
|
end
|
2007-09-02 02:39:58 -04:00
|
|
|
|
|
|
|
def test_inconsistent_struct
|
2007-09-07 19:06:33 -04:00
|
|
|
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
|
2007-09-02 02:39:58 -04:00
|
|
|
assert_raise(TypeError, "[ruby-dev:31709]") { Marshal.load(s) }
|
|
|
|
end
|
2007-09-07 19:06:33 -04:00
|
|
|
|
|
|
|
def test_struct_invalid_members
|
|
|
|
TestMarshal.const_set :StructInvalidMembers, Struct.new(:a)
|
|
|
|
assert_raise(TypeError, "[ruby-dev:31759]") {
|
2013-02-01 02:35:37 -05:00
|
|
|
Marshal.load("\004\bIc&TestMarshal::StructInvalidMembers\006:\020__members__\"\bfoo")
|
2007-09-07 19:06:33 -04:00
|
|
|
TestMarshal::StructInvalidMembers.members
|
|
|
|
}
|
|
|
|
end
|
2007-10-14 20:58:09 -04:00
|
|
|
|
|
|
|
class C
|
|
|
|
def initialize(str)
|
|
|
|
@str = str
|
|
|
|
end
|
2008-01-26 05:31:11 -05:00
|
|
|
attr_reader :str
|
2007-10-14 20:58:09 -04:00
|
|
|
def _dump(limit)
|
|
|
|
@str
|
|
|
|
end
|
|
|
|
def self._load(s)
|
|
|
|
new(s)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_too_long_string
|
2008-01-26 00:05:07 -05:00
|
|
|
data = Marshal.dump(C.new("a".force_encoding("ascii-8bit")))
|
|
|
|
data[-2, 1] = "\003\377\377\377"
|
2013-10-09 11:17:20 -04:00
|
|
|
assert_raise_with_message(ArgumentError, "marshal data too short", "[ruby-dev:32054]") {
|
2007-10-14 20:58:09 -04:00
|
|
|
Marshal.load(data)
|
|
|
|
}
|
|
|
|
end
|
2008-01-26 05:31:11 -05:00
|
|
|
|
|
|
|
|
|
|
|
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
|
2008-06-03 09:34:48 -04:00
|
|
|
|
|
|
|
def test_pipe
|
|
|
|
o1 = C.new("a" * 10000)
|
|
|
|
|
2014-05-31 19:54:12 -04:00
|
|
|
IO.pipe do |r, w|
|
2014-05-31 10:48:12 -04:00
|
|
|
th = Thread.new {Marshal.dump(o1, w)}
|
2014-05-31 19:54:12 -04:00
|
|
|
o2 = Marshal.load(r)
|
|
|
|
th.join
|
|
|
|
assert_equal(o1.str, o2.str)
|
2012-05-15 10:54:09 -04:00
|
|
|
end
|
2008-06-03 09:34:48 -04:00
|
|
|
|
2014-05-31 19:54:12 -04:00
|
|
|
IO.pipe do |r, w|
|
2014-05-31 10:48:12 -04:00
|
|
|
th = Thread.new {Marshal.dump(o1, w, 2)}
|
2014-05-31 19:54:12 -04:00
|
|
|
o2 = Marshal.load(r)
|
|
|
|
th.join
|
|
|
|
assert_equal(o1.str, o2.str)
|
2012-05-15 10:54:09 -04:00
|
|
|
end
|
2008-06-03 09:34:48 -04:00
|
|
|
|
|
|
|
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) }
|
2009-06-30 22:32:24 -04:00
|
|
|
assert_nothing_raised(ArgumentError, '[ruby-core:24100]') { Marshal.dump("\u3042", 1) }
|
2008-06-03 09:34:48 -04:00
|
|
|
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
|
2008-06-05 10:57:05 -04:00
|
|
|
|
|
|
|
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
|
* object.c (rb_obj_untrusted): new method Object#untrusted?.
(rb_obj_untrust): new method Object#untrust.
(rb_obj_trust): new method Object#trust.
* array.c, debug.c, time.c, include/ruby/ruby.h, re.c, variable.c,
string.c, io.c, dir.c, vm_method.c, struct.c, class.c, hash.c,
ruby.c, marshal.c: fixes for Object#untrusted?.
* test/ruby/test_module.rb, test/ruby/test_array.rb,
test/ruby/test_object.rb, test/ruby/test_string.rb,
test/ruby/test_marshal.rb, test/ruby/test_hash.rb: added tests for
Object#untrusted?.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@18568 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2008-08-13 03:25:05 -04:00
|
|
|
|
2013-06-13 01:41:44 -04:00
|
|
|
def test_taint
|
* object.c (rb_obj_untrusted): new method Object#untrusted?.
(rb_obj_untrust): new method Object#untrust.
(rb_obj_trust): new method Object#trust.
* array.c, debug.c, time.c, include/ruby/ruby.h, re.c, variable.c,
string.c, io.c, dir.c, vm_method.c, struct.c, class.c, hash.c,
ruby.c, marshal.c: fixes for Object#untrusted?.
* test/ruby/test_module.rb, test/ruby/test_array.rb,
test/ruby/test_object.rb, test/ruby/test_string.rb,
test/ruby/test_marshal.rb, test/ruby/test_hash.rb: added tests for
Object#untrusted?.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@18568 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2008-08-13 03:25:05 -04:00
|
|
|
x = Object.new
|
|
|
|
x.taint
|
|
|
|
s = Marshal.dump(x)
|
|
|
|
assert_equal(true, s.tainted?)
|
|
|
|
y = Marshal.load(s)
|
|
|
|
assert_equal(true, y.tainted?)
|
|
|
|
end
|
2009-08-07 17:18:33 -04:00
|
|
|
|
2013-06-13 01:41:44 -04:00
|
|
|
def test_taint_each_object
|
2009-12-28 08:01:28 -05:00
|
|
|
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?)
|
|
|
|
|
2013-06-13 01:41:44 -04:00
|
|
|
# tainted object causes tainted stream
|
2009-12-28 08:01:28 -05:00
|
|
|
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?)
|
|
|
|
|
2013-06-13 01:41:44 -04:00
|
|
|
# tainted stream causes tainted objects
|
2009-12-28 08:01:28 -05:00
|
|
|
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
|
|
|
|
|
2011-11-01 11:21:50 -04:00
|
|
|
def test_symbol2
|
2009-08-07 17:18:33 -04:00
|
|
|
[:ruby, :"\u{7d05}\u{7389}"].each do |sym|
|
|
|
|
assert_equal(sym, Marshal.load(Marshal.dump(sym)), '[ruby-core:24788]')
|
|
|
|
end
|
2010-01-07 23:12:29 -05:00
|
|
|
bug2548 = '[ruby-core:27375]'
|
|
|
|
ary = [:$1, nil]
|
|
|
|
assert_equal(ary, Marshal.load(Marshal.dump(ary)), bug2548)
|
2009-08-07 17:18:33 -04:00
|
|
|
end
|
2009-08-12 02:34:59 -04:00
|
|
|
|
2014-07-28 23:28:43 -04:00
|
|
|
def test_symlink
|
|
|
|
assert_include(Marshal.dump([:a, :a]), ';')
|
|
|
|
end
|
|
|
|
|
2015-03-22 13:03:31 -04:00
|
|
|
def test_symlink_in_ivar
|
|
|
|
bug10991 = '[ruby-core:68587] [Bug #10991]'
|
|
|
|
sym = Marshal.load("\x04\x08" +
|
|
|
|
"I" ":\x0bKernel" +
|
|
|
|
("\x06" +
|
|
|
|
("I" ":\x07@a" +
|
|
|
|
("\x06" ":\x07@b" "e;\x0""o:\x0bObject""\x0")) +
|
|
|
|
"0"))
|
|
|
|
assert_equal(:Kernel, sym, bug10991)
|
|
|
|
end
|
|
|
|
|
2009-08-12 02:34:59 -04:00
|
|
|
ClassUTF8 = eval("class R\u{e9}sum\u{e9}; self; end")
|
2009-10-17 09:11:05 -04:00
|
|
|
|
|
|
|
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
|
|
|
|
|
2009-08-12 02:34:59 -04:00
|
|
|
def test_class_nonascii
|
|
|
|
a = ClassUTF8.new
|
|
|
|
assert_instance_of(ClassUTF8, Marshal.load(Marshal.dump(a)), '[ruby-core:24790]')
|
2009-10-17 09:11:05 -04:00
|
|
|
|
|
|
|
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
|
2009-08-12 02:34:59 -04:00
|
|
|
end
|
2009-09-18 10:51:18 -04:00
|
|
|
|
2011-11-01 11:21:50 -04:00
|
|
|
def test_regexp2
|
2009-10-11 14:49:41 -04:00
|
|
|
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"))
|
|
|
|
|
2009-09-18 10:51:18 -04:00
|
|
|
bug2109 = '[ruby-core:25625]'
|
|
|
|
a = "\x82\xa0".force_encoding(Encoding::Windows_31J)
|
|
|
|
b = "\x82\xa2".force_encoding(Encoding::Windows_31J)
|
|
|
|
c = [/#{a}/, /#{b}/]
|
2009-10-11 14:49:41 -04:00
|
|
|
assert_equal(c, Marshal.load(Marshal.dump(c)), bug2109)
|
2010-02-11 00:37:43 -05:00
|
|
|
|
|
|
|
assert_nothing_raised(ArgumentError, '[ruby-dev:40386]') do
|
* test/csv/test_features.rb, test/logger/test_logger.rb
test/mkmf/test_have_macro.rb, test/net/http/test_http.rb,
test/openssl/test_config.rb, test/psych/test_encoding.rb,
test/psych/test_exception.rb, test/psych/test_psych.rb,
test/psych/test_tainted.rb, test/readline/test_readline.rb,
test/rexml/test_contrib.rb, test/ruby/test_autoload.rb,
test/ruby/test_beginendblock.rb, test/ruby/test_exception.rb,
test/ruby/test_file.rb, test/ruby/test_io.rb,
test/ruby/test_marshal.rb, test/ruby/test_process.rb,
test/ruby/test_require.rb, test/ruby/test_rubyoptions.rb,
test/syslog/test_syslog_logger.rb, test/webrick/test_httpauth.rb,
test/zlib/test_zlib.rb: Use Tempfile.create.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@40400 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2013-04-20 19:03:52 -04:00
|
|
|
re = Tempfile.create("marshal_regexp") do |f|
|
2010-02-11 00:37:43 -05:00
|
|
|
f.binmode.write("\x04\bI/\x00\x00\x06:\rencoding\"\rUS-ASCII")
|
* test/csv/test_features.rb, test/logger/test_logger.rb
test/mkmf/test_have_macro.rb, test/net/http/test_http.rb,
test/openssl/test_config.rb, test/psych/test_encoding.rb,
test/psych/test_exception.rb, test/psych/test_psych.rb,
test/psych/test_tainted.rb, test/readline/test_readline.rb,
test/rexml/test_contrib.rb, test/ruby/test_autoload.rb,
test/ruby/test_beginendblock.rb, test/ruby/test_exception.rb,
test/ruby/test_file.rb, test/ruby/test_io.rb,
test/ruby/test_marshal.rb, test/ruby/test_process.rb,
test/ruby/test_require.rb, test/ruby/test_rubyoptions.rb,
test/syslog/test_syslog_logger.rb, test/webrick/test_httpauth.rb,
test/zlib/test_zlib.rb: Use Tempfile.create.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@40400 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2013-04-20 19:03:52 -04:00
|
|
|
f.rewind
|
|
|
|
re2 = Marshal.load(f)
|
2013-01-28 22:06:40 -05:00
|
|
|
re2
|
2010-02-11 00:37:43 -05:00
|
|
|
end
|
|
|
|
assert_equal(//, re)
|
|
|
|
end
|
2009-09-18 10:51:18 -04:00
|
|
|
end
|
2009-10-04 06:30:56 -04:00
|
|
|
|
|
|
|
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
|
2009-11-21 02:38:51 -05:00
|
|
|
|
|
|
|
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
|
2009-12-05 01:35:07 -05:00
|
|
|
|
|
|
|
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)
|
2009-12-07 00:11:10 -05:00
|
|
|
assert_equal("foo", d.instance_variable_get(:@foo))
|
|
|
|
assert_equal(false, d.instance_variable_defined?(:@x))
|
2009-12-05 01:35:07 -05:00
|
|
|
end
|
2009-12-07 11:53:34 -05:00
|
|
|
|
|
|
|
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)
|
2011-02-17 08:37:55 -05:00
|
|
|
assert_equal(STDIN, o2.stdin)
|
2009-12-07 11:53:34 -05:00
|
|
|
end
|
2010-02-12 23:30:18 -05:00
|
|
|
|
2010-02-13 21:40:52 -05:00
|
|
|
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)
|
2010-02-12 23:30:18 -05:00
|
|
|
o2 = Marshal.load(m)
|
2010-02-13 21:40:52 -05:00
|
|
|
assert_equal(o1, o2)
|
2012-03-28 02:06:20 -04:00
|
|
|
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)
|
2010-02-12 23:30:18 -05:00
|
|
|
end
|
2011-01-28 12:57:27 -05:00
|
|
|
|
|
|
|
class PrivateClass
|
|
|
|
def initialize(foo)
|
|
|
|
@foo = foo
|
|
|
|
end
|
|
|
|
attr_reader :foo
|
|
|
|
end
|
|
|
|
private_constant :PrivateClass
|
2011-05-15 07:55:52 -04:00
|
|
|
|
2011-01-28 12:57:27 -05:00
|
|
|
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
|
2012-02-25 02:20:00 -05:00
|
|
|
|
|
|
|
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
|
2012-11-16 10:55:59 -05:00
|
|
|
|
|
|
|
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
|
2012-12-13 00:12:55 -05:00
|
|
|
|
|
|
|
class TestClass
|
|
|
|
end
|
|
|
|
|
|
|
|
module TestModule
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_marshal_load_should_not_taint_classes
|
|
|
|
bug7325 = '[ruby-core:49198]'
|
|
|
|
for c in [TestClass, TestModule]
|
2013-12-13 04:18:05 -05:00
|
|
|
assert_not_predicate(c, :tainted?)
|
2013-06-13 01:41:44 -04:00
|
|
|
c2 = Marshal.load(Marshal.dump(c).taint)
|
2012-12-13 00:12:55 -05:00
|
|
|
assert_same(c, c2)
|
2013-12-13 04:18:05 -05:00
|
|
|
assert_not_predicate(c, :tainted?, bug7325)
|
2012-12-13 00:12:55 -05:00
|
|
|
end
|
|
|
|
end
|
2013-01-26 08:39:15 -05:00
|
|
|
|
|
|
|
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
|
2013-02-01 02:35:37 -05:00
|
|
|
|
2013-04-22 04:09:44 -04:00
|
|
|
class LoadData
|
2013-04-19 00:03:29 -04:00
|
|
|
attr_reader :data
|
|
|
|
def initialize(data)
|
|
|
|
@data = data
|
|
|
|
end
|
|
|
|
alias marshal_dump data
|
|
|
|
alias marshal_load initialize
|
|
|
|
end
|
|
|
|
|
2013-04-22 04:09:44 -04:00
|
|
|
class Bug8276 < LoadData
|
|
|
|
def initialize(*)
|
|
|
|
super
|
|
|
|
freeze
|
|
|
|
end
|
|
|
|
alias marshal_load initialize
|
|
|
|
end
|
|
|
|
|
2013-05-02 04:32:48 -04:00
|
|
|
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
|
|
|
|
|
2013-04-19 00:03:29 -04:00
|
|
|
def test_marshal_dump_excess_encoding
|
|
|
|
bug8276 = '[ruby-core:54334] [Bug #8276]'
|
|
|
|
t = Bug8276.new(bug8276)
|
|
|
|
s = Marshal.dump(t)
|
2013-04-20 01:22:25 -04:00
|
|
|
assert_nothing_raised(RuntimeError, bug8276) {s = Marshal.load(s)}
|
2013-04-19 00:03:29 -04:00
|
|
|
assert_equal(t.data, s.data, bug8276)
|
|
|
|
end
|
|
|
|
|
2013-04-20 01:22:25 -04:00
|
|
|
def test_marshal_dump_ivar
|
|
|
|
s = "data with ivar"
|
|
|
|
s.instance_variable_set(:@t, 42)
|
|
|
|
t = Bug8276.new(s)
|
|
|
|
s = Marshal.dump(t)
|
2017-12-11 19:46:34 -05:00
|
|
|
assert_raise(FrozenError) {Marshal.load(s)}
|
2013-04-20 01:22:25 -04:00
|
|
|
end
|
|
|
|
|
2013-04-22 04:09:44 -04:00
|
|
|
def test_marshal_load_ivar
|
|
|
|
s = "data with ivar"
|
|
|
|
s.instance_variable_set(:@t, 42)
|
|
|
|
hook = ->(v) {
|
|
|
|
if LoadData === v
|
2013-05-02 04:32:48 -04:00
|
|
|
assert_send([v, :instance_variable_defined?, :@t], v.class.name)
|
|
|
|
assert_equal(42, v.instance_variable_get(:@t), v.class.name)
|
2013-04-22 04:09:44 -04:00
|
|
|
end
|
|
|
|
v
|
|
|
|
}
|
2013-05-02 04:32:48 -04:00
|
|
|
[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
|
2013-04-22 04:09:44 -04:00
|
|
|
end
|
|
|
|
|
2013-02-01 02:35:37 -05:00
|
|
|
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
|
2013-02-05 02:25:27 -05:00
|
|
|
|
|
|
|
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
|
2014-02-05 06:56:35 -05:00
|
|
|
|
|
|
|
def test_packed_string
|
|
|
|
packed = ["foo"].pack("p")
|
2014-02-05 19:50:22 -05:00
|
|
|
bare = "".force_encoding(Encoding::ASCII_8BIT) << packed
|
|
|
|
assert_equal(Marshal.dump(bare), Marshal.dump(packed))
|
2014-02-05 06:56:35 -05:00
|
|
|
end
|
2014-02-08 12:13:55 -05:00
|
|
|
|
|
|
|
def test_untainted_numeric
|
|
|
|
bug8945 = '[ruby-core:57346] [Bug #8945] Numerics never be tainted'
|
2017-04-05 22:10:40 -04:00
|
|
|
b = RbConfig::LIMITS['FIXNUM_MAX'] + 1
|
2014-02-08 12:13:55 -05:00
|
|
|
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
|
2014-02-17 03:41:57 -05:00
|
|
|
|
|
|
|
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)
|
2017-02-15 03:38:01 -05:00
|
|
|
GC.start
|
|
|
|
1000.times {"x"*1000}
|
|
|
|
GC.start
|
2014-02-17 03:41:57 -05:00
|
|
|
c.cc.call
|
|
|
|
end
|
|
|
|
end
|
2015-04-16 07:43:37 -04:00
|
|
|
|
|
|
|
def test_undumpable_message
|
|
|
|
c = Module.new {break module_eval("class IO\u{26a1} < IO;self;end")}
|
|
|
|
assert_raise_with_message(TypeError, /IO\u{26a1}/) {
|
|
|
|
Marshal.dump(c.new(0, autoclose: false))
|
|
|
|
}
|
|
|
|
end
|
2015-04-16 08:46:07 -04:00
|
|
|
|
|
|
|
def test_undumpable_data
|
|
|
|
c = Module.new {break module_eval("class T\u{23F0 23F3}<Time;undef _dump;self;end")}
|
|
|
|
assert_raise_with_message(TypeError, /T\u{23F0 23F3}/) {
|
|
|
|
Marshal.dump(c.new)
|
|
|
|
}
|
|
|
|
end
|
2015-04-16 10:25:19 -04:00
|
|
|
|
|
|
|
def test_unloadable_data
|
|
|
|
c = eval("class Unloadable\u{23F0 23F3}<Time;;self;end")
|
|
|
|
c.class_eval {
|
|
|
|
alias _dump_data _dump
|
|
|
|
undef _dump
|
|
|
|
}
|
|
|
|
d = Marshal.dump(c.new)
|
|
|
|
assert_raise_with_message(TypeError, /Unloadable\u{23F0 23F3}/) {
|
|
|
|
Marshal.load(d)
|
|
|
|
}
|
|
|
|
end
|
2015-04-16 10:25:33 -04:00
|
|
|
|
|
|
|
def test_unloadable_userdef
|
|
|
|
c = eval("class Userdef\u{23F0 23F3}<Time;self;end")
|
|
|
|
class << c
|
|
|
|
undef _load
|
|
|
|
end
|
|
|
|
d = Marshal.dump(c.new)
|
|
|
|
assert_raise_with_message(TypeError, /Userdef\u{23F0 23F3}/) {
|
|
|
|
Marshal.load(d)
|
|
|
|
}
|
|
|
|
end
|
2015-04-16 10:28:29 -04:00
|
|
|
|
|
|
|
def test_unloadable_usrmarshal
|
|
|
|
c = eval("class UsrMarshal\u{23F0 23F3}<Time;self;end")
|
|
|
|
c.class_eval {
|
|
|
|
alias marshal_dump _dump
|
|
|
|
}
|
|
|
|
d = Marshal.dump(c.new)
|
|
|
|
assert_raise_with_message(TypeError, /UsrMarshal\u{23F0 23F3}/) {
|
|
|
|
Marshal.load(d)
|
|
|
|
}
|
|
|
|
end
|
2015-10-27 10:36:35 -04:00
|
|
|
|
|
|
|
def test_no_internal_ids
|
|
|
|
opt = %w[--disable=gems]
|
|
|
|
args = [opt, 'Marshal.dump("",STDOUT)', true, true, encoding: Encoding::ASCII_8BIT]
|
|
|
|
out, err, status = EnvUtil.invoke_ruby(*args)
|
|
|
|
assert_empty(err)
|
|
|
|
assert_predicate(status, :success?)
|
|
|
|
expected = out
|
|
|
|
|
|
|
|
opt << "--enable=frozen-string-literal"
|
2015-11-21 04:57:31 -05:00
|
|
|
opt << "--debug=frozen-string-literal"
|
2015-10-27 10:36:35 -04:00
|
|
|
out, err, status = EnvUtil.invoke_ruby(*args)
|
|
|
|
assert_empty(err)
|
|
|
|
assert_predicate(status, :success?)
|
|
|
|
assert_equal(expected, out)
|
|
|
|
end
|
2016-01-21 03:36:01 -05:00
|
|
|
|
|
|
|
def test_marshal_honor_post_proc_value_for_link
|
|
|
|
str = 'x' # for link
|
|
|
|
obj = [str, str]
|
|
|
|
assert_equal(['X', 'X'], Marshal.load(Marshal.dump(obj), ->(v) { v == str ? v.upcase : v }))
|
|
|
|
end
|
2016-03-14 14:36:14 -04:00
|
|
|
|
|
|
|
def test_marshal_load_extended_class_crash
|
2016-06-18 22:19:45 -04:00
|
|
|
assert_separately([], "#{<<-"begin;"}\n#{<<-"end;"}")
|
|
|
|
begin;
|
|
|
|
assert_raise_with_message(ArgumentError, /undefined/) do
|
|
|
|
Marshal.load("\x04\be:\x0F\x00omparableo:\vObject\x00")
|
|
|
|
end
|
|
|
|
end;
|
2016-03-14 14:36:14 -04:00
|
|
|
end
|
2016-03-16 18:18:12 -04:00
|
|
|
|
|
|
|
def test_marshal_load_r_prepare_reference_crash
|
|
|
|
crash = "\x04\bI/\x05\x00\x06:\x06E{\x06@\x05T"
|
|
|
|
|
|
|
|
opt = %w[--disable=gems]
|
2016-03-16 21:23:03 -04:00
|
|
|
assert_separately(opt, <<-RUBY)
|
|
|
|
assert_raise_with_message(ArgumentError, /bad link/) do
|
|
|
|
Marshal.load(#{crash.dump})
|
|
|
|
end
|
2016-03-16 18:18:12 -04:00
|
|
|
RUBY
|
|
|
|
end
|
2016-06-24 13:33:35 -04:00
|
|
|
|
|
|
|
MethodMissingWithoutRespondTo = Struct.new(:wrapped_object) do
|
|
|
|
undef respond_to?
|
|
|
|
def method_missing(*args, &block)
|
|
|
|
wrapped_object.public_send(*args, &block)
|
|
|
|
end
|
|
|
|
def respond_to_missing?(name, private = false)
|
|
|
|
wrapped_object.respond_to?(name, false)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_method_missing_without_respond_to
|
|
|
|
bug12353 = "[ruby-core:75377] [Bug #12353]: try method_missing if" \
|
|
|
|
" respond_to? is undefined"
|
|
|
|
obj = MethodMissingWithoutRespondTo.new("foo")
|
|
|
|
dump = assert_nothing_raised(NoMethodError, bug12353) do
|
|
|
|
Marshal.dump(obj)
|
|
|
|
end
|
|
|
|
assert_equal(obj, Marshal.load(dump))
|
|
|
|
end
|
2016-11-24 06:07:12 -05:00
|
|
|
|
|
|
|
class Bug12974
|
|
|
|
def marshal_dump
|
|
|
|
dup
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_marshal_dump_recursion
|
|
|
|
assert_raise_with_message(RuntimeError, /same class instance/) do
|
|
|
|
Marshal.dump(Bug12974.new)
|
|
|
|
end
|
|
|
|
end
|
2018-01-05 06:44:31 -05:00
|
|
|
|
|
|
|
Bug14314 = Struct.new(:foo, keyword_init: true)
|
|
|
|
|
|
|
|
def test_marshal_keyword_init_struct
|
|
|
|
obj = Bug14314.new(foo: 42)
|
|
|
|
assert_equal obj, Marshal.load(Marshal.dump(obj))
|
|
|
|
end
|
2003-09-04 12:18:59 -04:00
|
|
|
end
|