ruby/mini_builtin.c

56 строки
1.4 KiB
C

#include "internal.h"
#include "vm_core.h"
#include "iseq.h"
#include "builtin.h"
// included from miniinit.c
static struct st_table *loaded_builtin_table;
rb_ast_t *rb_builtin_ast(const char *feature_name, VALUE *name_str);
void
rb_load_with_builtin_functions(const char *feature_name, const struct rb_builtin_function *table)
{
VALUE name_str = 0;
rb_ast_t *ast = rb_builtin_ast(feature_name, &name_str);
GET_VM()->builtin_function_table = table;
const rb_iseq_t *iseq = rb_iseq_new(&ast->body, name_str, name_str, Qnil, NULL, ISEQ_TYPE_TOP);
GET_VM()->builtin_function_table = NULL;
rb_ast_dispose(ast);
// register (loaded iseq will not be freed)
st_insert(loaded_builtin_table, (st_data_t)feature_name, (st_data_t)iseq);
rb_gc_register_mark_object((VALUE)iseq);
// eval
rb_iseq_eval(iseq);
}
static int
each_builtin_i(st_data_t key, st_data_t val, st_data_t dmy)
{
const char *feature = (const char *)key;
const rb_iseq_t *iseq = (const rb_iseq_t *)val;
rb_yield_values(2, rb_str_new2(feature), rb_iseqw_new(iseq));
return ST_CONTINUE;
}
static VALUE
each_builtin(VALUE self)
{
st_foreach(loaded_builtin_table, each_builtin_i, 0);
return Qnil;
}
void
Init_builtin(void)
{
rb_define_singleton_method(rb_cRubyVM, "each_builtin", each_builtin, 0);
loaded_builtin_table = st_init_strtable();
}