92 lines
2.7 KiB
C
Vendored
92 lines
2.7 KiB
C
Vendored
#include <mruby.h>
|
|
#include <mruby/class.h>
|
|
#include <mruby/string.h>
|
|
#include <mruby/istruct.h>
|
|
|
|
static mrb_value
|
|
istruct_test_initialize(mrb_state *mrb, mrb_value self)
|
|
{
|
|
char *string = (char*)mrb_istruct_ptr(self);
|
|
mrb_int size = mrb_istruct_size();
|
|
mrb_value object = mrb_get_arg1(mrb);
|
|
|
|
if (mrb_integer_p(object)) {
|
|
strncpy(string, "fixnum", size-1);
|
|
}
|
|
#ifndef MRB_NO_FLOAT
|
|
else if (mrb_float_p(object)) {
|
|
strncpy(string, "float", size-1);
|
|
}
|
|
#endif
|
|
else if (mrb_string_p(object)) {
|
|
strncpy(string, "string", size-1);
|
|
}
|
|
else {
|
|
strncpy(string, "anything", size-1);
|
|
}
|
|
|
|
string[size - 1] = 0; // force NULL at the end
|
|
return self;
|
|
}
|
|
|
|
static mrb_value
|
|
istruct_test_to_s(mrb_state *mrb, mrb_value self)
|
|
{
|
|
return mrb_str_new_cstr(mrb, (const char*)mrb_istruct_ptr(self));
|
|
}
|
|
|
|
static mrb_value
|
|
istruct_test_length(mrb_state *mrb, mrb_value self)
|
|
{
|
|
return mrb_fixnum_value(mrb_istruct_size());
|
|
}
|
|
|
|
static mrb_value
|
|
istruct_test_test_receive(mrb_state *mrb, mrb_value self)
|
|
{
|
|
mrb_value object = mrb_get_arg1(mrb);
|
|
|
|
if (mrb_obj_class(mrb, object) != mrb_class_get(mrb, "InlineStructTest"))
|
|
{
|
|
mrb_raise(mrb, E_TYPE_ERROR, "Expected InlineStructTest");
|
|
}
|
|
return mrb_bool_value(((char*)mrb_istruct_ptr(object))[0] == 's');
|
|
}
|
|
|
|
static mrb_value
|
|
istruct_test_test_receive_direct(mrb_state *mrb, mrb_value self)
|
|
{
|
|
mrb_value is = mrb_get_arg1(mrb);
|
|
struct RClass *klass = mrb_class_get(mrb, "InlineStructTest");
|
|
/* if you need to protect istruct retrieval from untrusted code,
|
|
you need to care about class replacing.
|
|
See mrbgem/mruby-random/src/random.c for detail */
|
|
if (mrb_istruct_p(is) && mrb_obj_is_kind_of(mrb, is, klass)) {
|
|
char *ptr = (char*)mrb_istruct_ptr(is);
|
|
return mrb_bool_value(ptr[0] == 's');
|
|
}
|
|
mrb_raise(mrb, E_TYPE_ERROR, "InlineStructTest");
|
|
return mrb_false_value();
|
|
}
|
|
|
|
static mrb_value
|
|
istruct_test_mutate(mrb_state *mrb, mrb_value self)
|
|
{
|
|
char *ptr = (char*)mrb_istruct_ptr(self);
|
|
memcpy(ptr, "mutate", 6);
|
|
return mrb_nil_value();
|
|
}
|
|
|
|
void mrb_mruby_test_inline_struct_gem_test(mrb_state *mrb)
|
|
{
|
|
struct RClass *cls;
|
|
|
|
cls = mrb_define_class(mrb, "InlineStructTest", mrb->object_class);
|
|
MRB_SET_INSTANCE_TT(cls, MRB_TT_ISTRUCT);
|
|
mrb_define_method(mrb, cls, "initialize", istruct_test_initialize, MRB_ARGS_REQ(1));
|
|
mrb_define_method(mrb, cls, "to_s", istruct_test_to_s, MRB_ARGS_NONE());
|
|
mrb_define_method(mrb, cls, "mutate", istruct_test_mutate, MRB_ARGS_NONE());
|
|
mrb_define_class_method(mrb, cls, "length", istruct_test_length, MRB_ARGS_NONE());
|
|
mrb_define_class_method(mrb, cls, "test_receive", istruct_test_test_receive, MRB_ARGS_REQ(1));
|
|
mrb_define_class_method(mrb, cls, "test_receive_direct", istruct_test_test_receive_direct, MRB_ARGS_REQ(1));
|
|
}
|