1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

* ext/dl/cfunc.c (rb_dlcfunc_initialize): cleaning up C macros

* ext/dl/cfunc.c (**): adding documentation
* test/dl/test_cfunc.rb (test_set_calltype) testing calltype

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@25679 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
tenderlove 2009-11-06 22:50:40 +00:00
parent bc52ea0fde
commit 35da6fdcf1
2 changed files with 53 additions and 9 deletions

View file

@ -133,7 +133,14 @@ rb_dlcfunc_kind_p(VALUE func)
return rb_typeddata_is_kind_of(func, &dlcfunc_data_type); return rb_typeddata_is_kind_of(func, &dlcfunc_data_type);
} }
VALUE /*
* call-seq:
* DL::CFunc.new(address, type=DL::TYPE_VOID, name=nil, calltype=:cdecl)
*
* Create a new function that points to +address+ with an optional return type
* of +type+, a name of +name+ and a calltype of +calltype+.
*/
static VALUE
rb_dlcfunc_initialize(int argc, VALUE argv[], VALUE self) rb_dlcfunc_initialize(int argc, VALUE argv[], VALUE self)
{ {
VALUE addr, name, type, calltype; VALUE addr, name, type, calltype;
@ -150,13 +157,19 @@ rb_dlcfunc_initialize(int argc, VALUE argv[], VALUE self)
if( data->name ) xfree(data->name); if( data->name ) xfree(data->name);
data->ptr = saddr; data->ptr = saddr;
data->name = sname ? strdup(sname) : 0; data->name = sname ? strdup(sname) : 0;
data->type = (type == Qnil) ? DLTYPE_VOID : NUM2INT(type); data->type = NIL_P(type) ? DLTYPE_VOID : NUM2INT(type);
data->calltype = (calltype == Qnil) ? CFUNC_CDECL : SYM2ID(calltype); data->calltype = NIL_P(calltype) ? CFUNC_CDECL : SYM2ID(calltype);
return Qnil; return Qnil;
} }
VALUE /*
* call-seq:
* name => str
*
* Get the name of this function
*/
static VALUE
rb_dlcfunc_name(VALUE self) rb_dlcfunc_name(VALUE self)
{ {
struct cfunc_data *cfunc; struct cfunc_data *cfunc;
@ -165,7 +178,14 @@ rb_dlcfunc_name(VALUE self)
return cfunc->name ? rb_tainted_str_new2(cfunc->name) : Qnil; return cfunc->name ? rb_tainted_str_new2(cfunc->name) : Qnil;
} }
VALUE /*
* call-seq:
* cfunc.ctype => num
*
* Get the C function return value type. See DL for a list of constants
* corresponding to this method's return value.
*/
static VALUE
rb_dlcfunc_ctype(VALUE self) rb_dlcfunc_ctype(VALUE self)
{ {
struct cfunc_data *cfunc; struct cfunc_data *cfunc;
@ -174,7 +194,13 @@ rb_dlcfunc_ctype(VALUE self)
return INT2NUM(cfunc->type); return INT2NUM(cfunc->type);
} }
VALUE /*
* call-seq:
* cfunc.ctype = type
*
* Set the C function return value type to +type+.
*/
static VALUE
rb_dlcfunc_set_ctype(VALUE self, VALUE ctype) rb_dlcfunc_set_ctype(VALUE self, VALUE ctype)
{ {
struct cfunc_data *cfunc; struct cfunc_data *cfunc;
@ -184,7 +210,13 @@ rb_dlcfunc_set_ctype(VALUE self, VALUE ctype)
return ctype; return ctype;
} }
VALUE /*
* call-seq:
* cfunc.calltype => symbol
*
* Get the call type of this function.
*/
static VALUE
rb_dlcfunc_calltype(VALUE self) rb_dlcfunc_calltype(VALUE self)
{ {
struct cfunc_data *cfunc; struct cfunc_data *cfunc;
@ -193,7 +225,13 @@ rb_dlcfunc_calltype(VALUE self)
return ID2SYM(cfunc->calltype); return ID2SYM(cfunc->calltype);
} }
VALUE /*
* call-seq:
* cfunc.calltype = symbol
*
* Set the call type for this function.
*/
static VALUE
rb_dlcfunc_set_calltype(VALUE self, VALUE sym) rb_dlcfunc_set_calltype(VALUE self, VALUE sym)
{ {
struct cfunc_data *cfunc; struct cfunc_data *cfunc;
@ -264,7 +302,7 @@ rb_dlcfunc_inspect(VALUE self)
} }
VALUE static VALUE
rb_dlcfunc_call(VALUE self, VALUE ary) rb_dlcfunc_call(VALUE self, VALUE ary)
{ {
struct cfunc_data *cfunc; struct cfunc_data *cfunc;

View file

@ -9,6 +9,11 @@ module DL
@cf = CFunc.new(@libc[@name], TYPE_VOIDP, @name) @cf = CFunc.new(@libc[@name], TYPE_VOIDP, @name)
end end
def test_set_calltype
@cf.calltype = :foo
assert_equal :foo, @cf.calltype
end
def test_new_ptr_type_name def test_new_ptr_type_name
assert_equal @name, @cf.name assert_equal @name, @cf.name
assert @cf.name.tainted?, 'name should be tainted' assert @cf.name.tainted?, 'name should be tainted'
@ -47,6 +52,7 @@ module DL
def test_to_i def test_to_i
assert_equal @cf.to_i, @cf.ptr assert_equal @cf.to_i, @cf.ptr
assert_equal @libc[@name], @cf.to_i
end end
def test_last_error def test_last_error