-
Notifications
You must be signed in to change notification settings - Fork 331
/
string_spec.rb
125 lines (113 loc) · 4.01 KB
/
string_spec.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#
# This file is part of ruby-ffi.
# For licensing, see LICENSE.SPECS
#
require File.expand_path(File.join(File.dirname(__FILE__), "spec_helper"))
describe "String tests" do
include FFI
module StrLibTest
extend FFI::Library
ffi_lib TestLibrary::PATH
attach_function :ptr_ret_pointer, [ :pointer, :int], :string
attach_function :string_equals, [ :string, :string ], :int
attach_function :pointer_string_equals, :string_equals, [ :pointer, :string ], :int
attach_function :string_dummy, [ :string ], :void
attach_function :string_null, [ ], :string
end
it "MemoryPointer#get_string returns a tainted string" do
mp = FFI::MemoryPointer.new 1024
mp.put_string(0, "test\0")
str = mp.get_string(0)
expect(str.tainted?).to be true
end
it "String returned by a method is tainted" do
mp = FFI::MemoryPointer.new :pointer
sp = FFI::MemoryPointer.new 1024
sp.put_string(0, "test")
mp.put_pointer(0, sp)
str = StrLibTest.ptr_ret_pointer(mp, 0)
expect(str).to eq("test")
expect(str).to be_tainted
end
it "A String can be passed to a :pointer argument" do
str = "string buffer"
expect(StrLibTest.pointer_string_equals(str, str)).to eq(1)
expect(StrLibTest.pointer_string_equals(str + "a", str)).to eq(0)
end
it "Poison null byte raises error" do
s = "123\0abc"
expect { StrLibTest.string_equals(s, s) }.to raise_error(ArgumentError)
end
it "Tainted String parameter should throw a SecurityError" do
$SAFE = 1
str = "test"
str.taint
begin
expect(LibTest.string_equals(str, str)).to be false
rescue SecurityError
end
end if false
it "casts nil as NULL pointer" do
expect(StrLibTest.string_dummy(nil)).to be_nil
end
it "return nil for NULL char*" do
expect(StrLibTest.string_null).to be_nil
end
it "reads an array of strings until encountering a NULL pointer" do
strings = ["foo", "bar", "baz", "testing", "ffi"]
ptrary = FFI::MemoryPointer.new(:pointer, 6)
ary = strings.inject([]) do |a, str|
f = FFI::MemoryPointer.new(1024)
f.put_string(0, str)
a << f
end
ary.insert(3, nil)
ptrary.write_array_of_pointer(ary)
expect(ptrary.get_array_of_string(0)).to eq(["foo", "bar", "baz"])
end
it "reads an array of strings of the size specified, substituting nil when a pointer is NULL" do
strings = ["foo", "bar", "baz", "testing", "ffi"]
ptrary = FFI::MemoryPointer.new(:pointer, 6)
ary = strings.inject([]) do |a, str|
f = FFI::MemoryPointer.new(1024)
f.put_string(0, str)
a << f
end
ary.insert(2, nil)
ptrary.write_array_of_pointer(ary)
expect(ptrary.get_array_of_string(0, 4)).to eq(["foo", "bar", nil, "baz"])
end
it "reads an array of strings, taking a memory offset parameter" do
strings = ["foo", "bar", "baz", "testing", "ffi"]
ptrary = FFI::MemoryPointer.new(:pointer, 5)
ary = strings.inject([]) do |a, str|
f = FFI::MemoryPointer.new(1024)
f.put_string(0, str)
a << f
end
ptrary.write_array_of_pointer(ary)
expect(ptrary.get_array_of_string(2 * FFI.type_size(:pointer), 3)).to eq(["baz", "testing", "ffi"])
end
it "raises an IndexError when trying to read an array of strings out of bounds" do
strings = ["foo", "bar", "baz", "testing", "ffi"]
ptrary = FFI::MemoryPointer.new(:pointer, 5)
ary = strings.inject([]) do |a, str|
f = FFI::MemoryPointer.new(1024)
f.put_string(0, str)
a << f
end
ptrary.write_array_of_pointer(ary)
expect { ptrary.get_array_of_string(0, 6) }.to raise_error(IndexError)
end
it "raises an IndexError when trying to read an array of strings using a negative offset" do
strings = ["foo", "bar", "baz", "testing", "ffi"]
ptrary = FFI::MemoryPointer.new(:pointer, 5)
ary = strings.inject([]) do |a, str|
f = FFI::MemoryPointer.new(1024)
f.put_string(0, str)
a << f
end
ptrary.write_array_of_pointer(ary)
expect { ptrary.get_array_of_string(-1) }.to raise_error(IndexError)
end
end