1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/ext/objspace/object_tracing.c
nari fbece268c7 * ext/objspace/object_tracing.c: rename allocation_info to
lookup_allocation_info. At times I confused "struct
  allocation_info" with "function allocation_info".

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@41104 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2013-06-06 05:18:40 +00:00

247 lines
6.4 KiB
C

/**********************************************************************
object_tracing.c - Object Tracing mechanism/ObjectSpace extender for MRI.
$Author$
created at: Mon May 27 16:27:44 2013
NOTE: This extension library is not expected to exist except C Ruby.
NOTE: This feature is an example usage of internal event tracing APIs.
All the files in this distribution are covered under the Ruby's
license (see the file COPYING).
**********************************************************************/
#include "ruby/ruby.h"
#include "ruby/debug.h"
size_t rb_gc_count(void); /* from gc.c */
struct traceobj_arg {
VALUE newobj_trace;
VALUE freeobj_trace;
st_table *object_table;
st_table *str_table;
struct traceobj_arg *prev_traceobj_arg;
};
struct traceobj_arg *traceobj_arg; /* TODO: do not use GLOBAL VARIABLE!!! */
struct allocation_info {
const char *path;
unsigned long line;
const char *class_path;
VALUE mid;
size_t generation;
};
static const char *
make_unique_str(st_table *tbl, const char *str, long len)
{
if (!str) {
return NULL;
}
else {
st_data_t n;
char *result;
if (st_lookup(tbl, (st_data_t)str, &n)) {
st_insert(tbl, (st_data_t)str, n+1);
st_get_key(tbl, (st_data_t)str, (st_data_t *)&result);
}
else {
result = (char *)ruby_xmalloc(len+1);
strncpy(result, str, len);
result[len] = 0;
st_add_direct(tbl, (st_data_t)result, 1);
}
return result;
}
}
static void
delete_unique_str(st_table *tbl, const char *str)
{
if (str) {
st_data_t n;
st_lookup(tbl, (st_data_t)str, &n);
if (n == 1) {
st_delete(tbl, (st_data_t *)&str, 0);
ruby_xfree((char *)str);
}
else {
st_insert(tbl, (st_data_t)str, n-1);
}
}
}
static void
newobj_i(VALUE tpval, void *data)
{
struct traceobj_arg *arg = (struct traceobj_arg *)data;
rb_trace_arg_t *tparg = rb_tracearg_from_tracepoint(tpval);
VALUE obj = rb_tracearg_object(tparg);
VALUE path = rb_tracearg_path(tparg);
VALUE line = rb_tracearg_lineno(tparg);
VALUE mid = rb_tracearg_method_id(tparg);
VALUE klass = rb_tracearg_defined_class(tparg);
struct allocation_info *info = (struct allocation_info *)ruby_xmalloc(sizeof(struct allocation_info));
const char *path_cstr = RTEST(path) ? make_unique_str(arg->str_table, RSTRING_PTR(path), RSTRING_LEN(path)) : 0;
VALUE class_path = RTEST(klass) ? rb_class_path(klass) : Qnil;
const char *class_path_cstr = RTEST(class_path) ? make_unique_str(arg->str_table, RSTRING_PTR(class_path), RSTRING_LEN(class_path)) : 0;
info->path = path_cstr;
info->line = NUM2INT(line);
info->mid = mid;
info->class_path = class_path_cstr;
info->generation = rb_gc_count();
st_insert(arg->object_table, (st_data_t)obj, (st_data_t)info);
}
static void
freeobj_i(VALUE tpval, void *data)
{
struct traceobj_arg *arg = (struct traceobj_arg *)data;
rb_trace_arg_t *tparg = rb_tracearg_from_tracepoint(tpval);
VALUE obj = rb_tracearg_object(tparg);
struct allocation_info *info;
if (st_delete(arg->object_table, (st_data_t *)&obj, (st_data_t *)&info)) {
delete_unique_str(arg->str_table, info->path);
delete_unique_str(arg->str_table, info->class_path);
ruby_xfree(info);
}
}
static int
free_keys_i(st_data_t key, st_data_t value, void *data)
{
ruby_xfree((void *)key);
return ST_CONTINUE;
}
static int
free_values_i(st_data_t key, st_data_t value, void *data)
{
ruby_xfree((void *)value);
return ST_CONTINUE;
}
static VALUE
stop_trace_object_allocations(void *data)
{
struct traceobj_arg *arg = (struct traceobj_arg *)data;
rb_tracepoint_disable(arg->newobj_trace);
rb_tracepoint_disable(arg->freeobj_trace);
st_foreach(arg->object_table, free_values_i, 0);
st_foreach(arg->str_table, free_keys_i, 0);
st_free_table(arg->object_table);
st_free_table(arg->str_table);
traceobj_arg = arg->prev_traceobj_arg;
return Qnil;
}
static VALUE
trace_object_allocations(VALUE objspace)
{
struct traceobj_arg arg;
arg.newobj_trace = rb_tracepoint_new(0, RUBY_INTERNAL_EVENT_NEWOBJ, newobj_i, &arg);
arg.freeobj_trace = rb_tracepoint_new(0, RUBY_INTERNAL_EVENT_FREEOBJ, freeobj_i, &arg);
arg.object_table = st_init_numtable();
arg.str_table = st_init_strtable();
arg.prev_traceobj_arg = traceobj_arg;
traceobj_arg = &arg;
rb_tracepoint_enable(arg.newobj_trace);
rb_tracepoint_enable(arg.freeobj_trace);
return rb_ensure(rb_yield, Qnil, stop_trace_object_allocations, (VALUE)&arg);
}
static struct allocation_info *
lookup_allocation_info(VALUE obj)
{
if (traceobj_arg) {
struct allocation_info *info;
if (st_lookup(traceobj_arg->object_table, obj, (st_data_t *)&info)) {
return info;
}
}
return NULL;
}
static VALUE
allocation_sourcefile(VALUE objspace, VALUE obj)
{
struct allocation_info *info = lookup_allocation_info(obj);
if (info) {
return info->path ? rb_str_new2(info->path) : Qnil;
}
else {
return Qnil;
}
}
static VALUE
allocation_sourceline(VALUE objspace, VALUE obj)
{
struct allocation_info *info = lookup_allocation_info(obj);
if (info) {
return INT2FIX(info->line);
}
else {
return Qnil;
}
}
static VALUE
allocation_class_path(VALUE objspace, VALUE obj)
{
struct allocation_info *info = lookup_allocation_info(obj);
if (info) {
return info->class_path ? rb_str_new2(info->class_path) : Qnil;
}
else {
return Qnil;
}
}
static VALUE
allocation_method_id(VALUE objspace, VALUE obj)
{
struct allocation_info *info = lookup_allocation_info(obj);
if (info) {
return info->mid;
}
else {
return Qnil;
}
}
static VALUE
allocation_generation(VALUE objspace, VALUE obj)
{
struct allocation_info *info = lookup_allocation_info(obj);
if (info) {
return SIZET2NUM(info->generation);
}
else {
return Qnil;
}
}
void
Init_object_tracing(VALUE rb_mObjSpace)
{
rb_define_module_function(rb_mObjSpace, "trace_object_allocations", trace_object_allocations, 0);
rb_define_module_function(rb_mObjSpace, "allocation_sourcefile", allocation_sourcefile, 1);
rb_define_module_function(rb_mObjSpace, "allocation_sourceline", allocation_sourceline, 1);
rb_define_module_function(rb_mObjSpace, "allocation_class_path", allocation_class_path, 1);
rb_define_module_function(rb_mObjSpace, "allocation_method_id", allocation_method_id, 1);
rb_define_module_function(rb_mObjSpace, "allocation_generation", allocation_generation, 1);
}