From 5f3d0f523328be8197fd3d410a1b3bd3febfcbc2 Mon Sep 17 00:00:00 2001
From: tenderlove <tenderlove@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sun, 1 Nov 2009 01:46:27 +0000
Subject: * test/dl/test_dl2.rb (**) testing malloc and realloc * ext/dl/dl.c
 (**) adding documentation

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@25605 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 test/dl/test_dl2.rb | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

(limited to 'test/dl')

diff --git a/test/dl/test_dl2.rb b/test/dl/test_dl2.rb
index 5a9daba42a..d391e976e5 100644
--- a/test/dl/test_dl2.rb
+++ b/test/dl/test_dl2.rb
@@ -3,6 +3,53 @@ require 'dl/callback'
 
 module DL
 class TestDL < TestBase
+  # TODO: refactor test repetition
+
+  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!*"))
-- 
cgit v1.2.3