diff options
Diffstat (limited to 'test/ruby')
-rw-r--r-- | test/ruby/test_pack.rb | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/test/ruby/test_pack.rb b/test/ruby/test_pack.rb index b59faec9f3..76843b2eae 100644 --- a/test/ruby/test_pack.rb +++ b/test/ruby/test_pack.rb @@ -829,4 +829,20 @@ EXPECTED ret = []; "A".unpack("B*") {|v| ret << v } assert_equal ["01000001"], ret end + + def test_pack_infection + tainted_array_string = ["123456"] + tainted_array_string.first.taint + ['a', 'A', 'Z', 'B', 'b', 'H', 'h', 'u', 'M', 'm', 'P', 'p'].each do |f| + assert_predicate(tainted_array_string.pack(f), :tainted?) + end + end + + def test_unpack_infection + tainted_string = "123456" + tainted_string.taint + ['a', 'A', 'Z', 'B', 'b', 'H', 'h', 'u', 'M', 'm'].each do |f| + assert_predicate(tainted_string.unpack(f).first, :tainted?) + end + end end |