147 lines
3.7 KiB
Ruby
147 lines
3.7 KiB
Ruby
require_relative 'test_base.rb'
|
|
require 'dl/callback'
|
|
|
|
module DL
|
|
class TestDL < TestBase
|
|
# TODO: refactor test repetition
|
|
|
|
def test_free_secure
|
|
assert_raises(SecurityError) do
|
|
Thread.new do
|
|
$SAFE = 4
|
|
DL.free(0)
|
|
end.join
|
|
end
|
|
end
|
|
|
|
def test_realloc
|
|
str = "abc"
|
|
ptr_id = DL.realloc(0, 4)
|
|
ptr = CPtr.new(ptr_id, 4)
|
|
|
|
assert_equal ptr_id, ptr.to_i
|
|
|
|
cfunc = CFunc.new(@libc['strcpy'], TYPE_VOIDP, 'strcpy')
|
|
x = cfunc.call([ptr_id,str].pack("l!p").unpack("l!*"))
|
|
assert_equal("abc\0", ptr[0,4])
|
|
DL.free ptr_id
|
|
end
|
|
|
|
def test_realloc_secure
|
|
assert_raises(SecurityError) do
|
|
Thread.new do
|
|
$SAFE = 4
|
|
DL.realloc(0, 4)
|
|
end.join
|
|
end
|
|
end
|
|
|
|
def test_malloc
|
|
str = "abc"
|
|
|
|
ptr_id = DL.malloc(4)
|
|
ptr = CPtr.new(ptr_id, 4)
|
|
|
|
assert_equal ptr_id, ptr.to_i
|
|
|
|
cfunc = CFunc.new(@libc['strcpy'], TYPE_VOIDP, 'strcpy')
|
|
x = cfunc.call([ptr_id,str].pack("l!p").unpack("l!*"))
|
|
assert_equal("abc\0", ptr[0,4])
|
|
DL.free ptr_id
|
|
end
|
|
|
|
def test_malloc_security
|
|
assert_raises(SecurityError) do
|
|
Thread.new do
|
|
$SAFE = 4
|
|
DL.malloc(4)
|
|
end.join
|
|
end
|
|
end
|
|
|
|
def test_call_int()
|
|
cfunc = CFunc.new(@libc['atoi'], TYPE_INT, 'atoi')
|
|
x = cfunc.call(["100"].pack("p").unpack("l!*"))
|
|
assert_equal(100, x)
|
|
|
|
cfunc = CFunc.new(@libc['atoi'], TYPE_INT, 'atoi')
|
|
x = cfunc.call(["-100"].pack("p").unpack("l!*"))
|
|
assert_equal(-100, x)
|
|
end
|
|
|
|
def test_call_long()
|
|
cfunc = CFunc.new(@libc['atol'], TYPE_LONG, 'atol')
|
|
x = cfunc.call(["100"].pack("p").unpack("l!*"))
|
|
assert_equal(100, x)
|
|
cfunc = CFunc.new(@libc['atol'], TYPE_LONG, 'atol')
|
|
x = cfunc.call(["-100"].pack("p").unpack("l!*"))
|
|
assert_equal(-100, x)
|
|
end
|
|
|
|
def test_call_double()
|
|
cfunc = CFunc.new(@libc['atof'], TYPE_DOUBLE, 'atof')
|
|
x = cfunc.call(["0.1"].pack("p").unpack("l!*"))
|
|
assert_in_delta(0.1, x)
|
|
|
|
cfunc = CFunc.new(@libc['atof'], TYPE_DOUBLE, 'atof')
|
|
x = cfunc.call(["-0.1"].pack("p").unpack("l!*"))
|
|
assert_in_delta(-0.1, x)
|
|
end
|
|
|
|
def test_sin()
|
|
pi_2 = Math::PI/2
|
|
cfunc = CFunc.new(@libm['sin'], TYPE_DOUBLE, 'sin')
|
|
x = cfunc.call([pi_2].pack("d").unpack("l!*"))
|
|
assert_equal(Math.sin(pi_2), x)
|
|
|
|
cfunc = CFunc.new(@libm['sin'], TYPE_DOUBLE, 'sin')
|
|
x = cfunc.call([-pi_2].pack("d").unpack("l!*"))
|
|
assert_equal(Math.sin(-pi_2), x)
|
|
end
|
|
|
|
def test_strlen()
|
|
cfunc = CFunc.new(@libc['strlen'], TYPE_INT, 'strlen')
|
|
x = cfunc.call(["abc"].pack("p").unpack("l!*"))
|
|
assert_equal("abc".size, x)
|
|
end
|
|
|
|
def test_strcpy()
|
|
buff = "xxxx"
|
|
str = "abc"
|
|
cfunc = CFunc.new(@libc['strcpy'], TYPE_VOIDP, 'strcpy')
|
|
x = cfunc.call([buff,str].pack("pp").unpack("l!*"))
|
|
assert_equal("abc\0", buff)
|
|
assert_equal("abc\0", CPtr.new(x).to_s(4))
|
|
|
|
buff = "xxxx"
|
|
str = "abc"
|
|
cfunc = CFunc.new(@libc['strncpy'], TYPE_VOIDP, 'strncpy')
|
|
x = cfunc.call([buff,str,3].pack("ppL!").unpack("l!*"))
|
|
assert_equal("abcx", buff)
|
|
assert_equal("abcx", CPtr.new(x).to_s(4))
|
|
|
|
ptr = CPtr.malloc(4)
|
|
str = "abc"
|
|
cfunc = CFunc.new(@libc['strcpy'], TYPE_VOIDP, 'strcpy')
|
|
x = cfunc.call([ptr.to_i,str].pack("l!p").unpack("l!*"))
|
|
assert_equal("abc\0", ptr[0,4])
|
|
assert_equal("abc\0", CPtr.new(x).to_s(4))
|
|
end
|
|
|
|
def test_callback()
|
|
buff = "foobarbaz"
|
|
cb = set_callback(TYPE_INT,2){|x,y| CPtr.new(x)[0] <=> CPtr.new(y)[0]}
|
|
cfunc = CFunc.new(@libc['qsort'], TYPE_VOID, 'qsort')
|
|
cfunc.call([buff, buff.size, 1, cb].pack("pL!L!L!").unpack("l!*"))
|
|
assert_equal('aabbfoorz', buff)
|
|
end
|
|
|
|
def test_dlwrap()
|
|
ary = [0,1,2,4,5]
|
|
addr = dlwrap(ary)
|
|
ary2 = dlunwrap(addr)
|
|
assert_equal(ary, ary2)
|
|
end
|
|
end
|
|
end # module DL
|