Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions ext/fiddle/fiddle.h
Original file line number Diff line number Diff line change
Expand Up @@ -199,5 +199,8 @@ extern VALUE rb_eFiddleDLError;

VALUE rb_fiddle_new_function(VALUE address, VALUE arg_types, VALUE ret_type);

typedef void (*rb_fiddle_freefunc_t)(void*);
VALUE rb_fiddle_ptr_new_wrap(void *ptr, long size, rb_fiddle_freefunc_t func, VALUE wrap0, VALUE wrap1);

#endif
/* vim: set noet sws=4 sw=4: */
17 changes: 16 additions & 1 deletion ext/fiddle/handle.c
Original file line number Diff line number Diff line change
Expand Up @@ -259,7 +259,21 @@ rb_fiddle_handle_to_i(VALUE self)
struct dl_handle *fiddle_handle;

TypedData_Get_Struct(self, struct dl_handle, &fiddle_handle_data_type, fiddle_handle);
return PTR2NUM(fiddle_handle);
return PTR2NUM(fiddle_handle->ptr);
}

/*
* call-seq: to_ptr
*
* Returns the Fiddle::Pointer of this handle.
*/
static VALUE
rb_fiddle_handle_to_ptr(VALUE self)
{
struct dl_handle *fiddle_handle;

TypedData_Get_Struct(self, struct dl_handle, &fiddle_handle_data_type, fiddle_handle);
return rb_fiddle_ptr_new_wrap(fiddle_handle->ptr, 0, 0, self, 0);
}

static VALUE fiddle_handle_sym(void *handle, VALUE symbol);
Expand Down Expand Up @@ -466,6 +480,7 @@ Init_fiddle_handle(void)

rb_define_method(rb_cHandle, "initialize", rb_fiddle_handle_initialize, -1);
rb_define_method(rb_cHandle, "to_i", rb_fiddle_handle_to_i, 0);
rb_define_method(rb_cHandle, "to_ptr", rb_fiddle_handle_to_ptr, 0);
rb_define_method(rb_cHandle, "close", rb_fiddle_handle_close, 0);
rb_define_method(rb_cHandle, "sym", rb_fiddle_handle_sym, 1);
rb_define_method(rb_cHandle, "[]", rb_fiddle_handle_sym, 1);
Expand Down
16 changes: 12 additions & 4 deletions ext/fiddle/pointer.c
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@

VALUE rb_cPointer;

typedef void (*freefunc_t)(void*);
typedef rb_fiddle_freefunc_t freefunc_t;

struct ptr_data {
void *ptr;
Expand Down Expand Up @@ -125,7 +125,7 @@ static const rb_memory_view_entry_t fiddle_ptr_memory_view_entry = {
#endif

static VALUE
rb_fiddle_ptr_new2(VALUE klass, void *ptr, long size, freefunc_t func)
rb_fiddle_ptr_new2(VALUE klass, void *ptr, long size, freefunc_t func, VALUE wrap0, VALUE wrap1)
{
struct ptr_data *data;
VALUE val;
Expand All @@ -135,14 +135,22 @@ rb_fiddle_ptr_new2(VALUE klass, void *ptr, long size, freefunc_t func)
data->free = func;
data->freed = false;
data->size = size;
data->wrap[0] = wrap0;
data->wrap[1] = wrap1;

return val;
}

VALUE
rb_fiddle_ptr_new_wrap(void *ptr, long size, freefunc_t func, VALUE wrap0, VALUE wrap1)
{
return rb_fiddle_ptr_new2(rb_cPointer, ptr, size, func, wrap0, wrap1);
}

static VALUE
rb_fiddle_ptr_new(void *ptr, long size, freefunc_t func)
{
return rb_fiddle_ptr_new2(rb_cPointer, ptr, size, func);
return rb_fiddle_ptr_new2(rb_cPointer, ptr, size, func, 0, 0);
}

static VALUE
Expand All @@ -152,7 +160,7 @@ rb_fiddle_ptr_malloc(VALUE klass, long size, freefunc_t func)

ptr = ruby_xmalloc((size_t)size);
memset(ptr,0,(size_t)size);
return rb_fiddle_ptr_new2(klass, ptr, size, func);
return rb_fiddle_ptr_new2(klass, ptr, size, func, 0, 0);
}

static void *
Expand Down
6 changes: 6 additions & 0 deletions test/fiddle/test_handle.rb
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,12 @@ def test_to_i
assert_kind_of Integer, handle.to_i
end

def test_to_ptr
handle = Fiddle::Handle.new(LIBC_SO)
ptr = handle.to_ptr
assert_equal ptr.to_i, handle.to_i
end

def test_static_sym_unknown
assert_raise(DLError) { Fiddle::Handle.sym('fooo') }
assert_raise(DLError) { Fiddle::Handle['fooo'] }
Expand Down