From 1f3142a4477a10bfe566c17837c2a0b9f6cba00f Mon Sep 17 00:00:00 2001 From: ko1 Date: Fri, 25 May 2012 04:50:10 +0000 Subject: * vm.c: refactoring backtrace related funcitons. (1) unify similar functions (rb_backtrace_each() and backtrace_object()). backtrace_each() is a unified function. variation: a) backtrace_object(): create backtrace object. b) vm_backtrace_str_ary(): create bt as an array of string. c) vm_backtrace_print(): print backtrace to specified file. d) rb_backtrace_print_as_bugreport(): print backtrace on bugreport style. (2) remove rb_backtrace_each(). Use backtrace_each() instead. (3) chang the type of lev parameter to size_t. a) lev == 0 means current frame (exception, etc use it). b) lev == 1 means upper frame (caller(0) use it). * vm_core.h, vm_dump.c, vm_eval.c: ditto. * vm.c (backtrace_object(), vm_backtrace_str_ary()): fix to return a correct size of caller(lev) array. Let n be a "caller(0).size" then ln as caller(lev).size should be (n - lev). However, the previous implementation returns a wrong size array (ln > n - lev). [ruby-dev:45673] * test/ruby/test_backtrace.rb: add tests for backtrace. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@35787 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- vm.c | 359 ++++++++++++++++++++++++++++++++++++++++++++++--------------------- 1 file changed, 249 insertions(+), 110 deletions(-) (limited to 'vm.c') diff --git a/vm.c b/vm.c index 9ce0a8e713..a6f1ce5a47 100644 --- a/vm.c +++ b/vm.c @@ -736,7 +736,7 @@ rb_lastline_set(VALUE val) /* backtrace */ inline static int -calc_line_no(const rb_iseq_t *iseq, VALUE *pc) +calc_line_no(const rb_iseq_t *iseq, const VALUE *pc) { return rb_iseq_line_no(iseq, pc - iseq->iseq_encoded); } @@ -762,8 +762,8 @@ typedef struct rb_frame_info_struct { union { struct { - rb_iseq_t *iseq; - VALUE *pc; + const rb_iseq_t *iseq; + const VALUE *pc; } iseq; struct { ID mid; @@ -824,8 +824,9 @@ frame_info_to_str_override(rb_frame_info_t *fi, VALUE *args) typedef struct rb_backtrace_struct { rb_frame_info_t *backtrace; + rb_frame_info_t *backtrace_base; size_t backtrace_size; - VALUE str; + VALUE strary; } rb_backtrace_t; static void @@ -837,7 +838,7 @@ backtrace_mark(void *ptr) for (i=0; ibacktrace[i]); - rb_gc_mark(bt->str); + rb_gc_mark(bt->strary); } } } @@ -847,7 +848,7 @@ backtrace_free(void *ptr) { if (ptr) { rb_backtrace_t *bt = (rb_backtrace_t *)ptr; - if (bt->backtrace) ruby_xfree(bt->backtrace); + if (bt->backtrace) ruby_xfree(bt->backtrace_base); ruby_xfree(bt); } } @@ -875,71 +876,133 @@ backtrace_alloc(VALUE klass) { rb_backtrace_t *bt; VALUE obj = TypedData_Make_Struct(klass, rb_backtrace_t, &backtrace_data_type, bt); - bt->backtrace = 0; - bt->backtrace_size = 0; - bt->str = 0; return obj; } -static VALUE -backtrace_object(rb_thread_t *th, int lev, ptrdiff_t n) +static void +backtrace_each(rb_thread_t *th, + void (*init)(void *arg, int size), + void (*iter_iseq)(void *arg, const rb_iseq_t *iseq, const VALUE *pc), + void (*iter_cfunc)(void *arg, ID mid), + void *arg) { - VALUE btobj = backtrace_alloc(rb_cBacktrace); - rb_backtrace_t *bt; rb_control_frame_t *last_cfp = th->cfp; rb_control_frame_t *start_cfp = RUBY_VM_END_CONTROL_FRAME(th); rb_control_frame_t *cfp; - ptrdiff_t size, i, j; + ptrdiff_t size, i; + + /* <- start_cfp (end control frame) + * top frame (dummy) + * top frame (dummy) + * top frame <- start_cfp + * top frame + * ... + * 2nd frame <- lev:0 + * current frame <- th->cfp + */ - start_cfp = RUBY_VM_NEXT_CONTROL_FRAME( - RUBY_VM_NEXT_CONTROL_FRAME( - RUBY_VM_NEXT_CONTROL_FRAME(start_cfp))); /* skip top frames */ - size = (start_cfp - last_cfp) + 1; + start_cfp = + RUBY_VM_NEXT_CONTROL_FRAME( + RUBY_VM_NEXT_CONTROL_FRAME( + RUBY_VM_NEXT_CONTROL_FRAME(start_cfp))); /* skip top frames */ - if (n <= 0) { - n = size + n; - if (n < 0) { - n = 0; - } + if (start_cfp < last_cfp) { + size = 0; } - if (lev < 0) { - lev = 0; + else { + size = start_cfp - last_cfp + 1; } - GetCoreDataFromValue(btobj, rb_backtrace_t, bt); - bt->backtrace = ruby_xmalloc(sizeof(rb_frame_info_t) * n); - bt->backtrace_size = n; - - for (i=lev, j=0, cfp = start_cfp; ibacktrace[j]; + init(arg, size); + /* SDR(); */ + for (i=0, cfp = start_cfp; istack + th->stack_size) - cfp); */ if (cfp->iseq) { if (cfp->pc) { - fi->type = FRAME_INFO_TYPE_ISEQ; - fi->body.iseq.iseq = cfp->iseq; - fi->body.iseq.pc = cfp->pc; - j++; + iter_iseq(arg, cfp->iseq, cfp->pc); } } else if (RUBYVM_CFUNC_FRAME_P(cfp)) { ID mid = cfp->me->def ? cfp->me->def->original_id : cfp->me->called_id; if (mid != ID_ALLOCATOR) { - fi->type = FRAME_INFO_TYPE_CFUNC; - fi->body.cfunc.mid = mid; - j++; + iter_cfunc(arg, mid); } } } +} + +struct bt_iter_arg { + rb_backtrace_t *bt; + VALUE btobj; +}; + +static void +bt_init(void *ptr, int size) +{ + struct bt_iter_arg *arg = (struct bt_iter_arg *)ptr; + arg->btobj = backtrace_alloc(rb_cBacktrace); + GetCoreDataFromValue(arg->btobj, rb_backtrace_t, arg->bt); + arg->bt->backtrace_base = arg->bt->backtrace = ruby_xmalloc(sizeof(rb_frame_info_t) * size); + arg->bt->backtrace_size = 0; +} + +static void +bt_iter_iseq(void *ptr, const rb_iseq_t *iseq, const VALUE *pc) +{ + struct bt_iter_arg *arg = (struct bt_iter_arg *)ptr; + rb_frame_info_t *fi = &arg->bt->backtrace[arg->bt->backtrace_size++]; + fi->type = FRAME_INFO_TYPE_ISEQ; + fi->body.iseq.iseq = iseq; + fi->body.iseq.pc = pc; +} + +static void +bt_iter_cfunc(void *ptr, ID mid) +{ + struct bt_iter_arg *arg = (struct bt_iter_arg *)ptr; + rb_frame_info_t *fi = &arg->bt->backtrace[arg->bt->backtrace_size++]; + fi->type = FRAME_INFO_TYPE_CFUNC; + fi->body.cfunc.mid = mid; +} + +static VALUE +backtrace_object(rb_thread_t *th, size_t lev, size_t n) +{ + struct bt_iter_arg arg; + + backtrace_each(th, + bt_init, + bt_iter_iseq, + bt_iter_cfunc, + &arg); - if (j > 0) { - bt->backtrace_size = j; /* TODO: realloc? */ - return btobj; + if (lev > 0) { + if (lev > arg.bt->backtrace_size) { + arg.bt->backtrace = 0; + arg.bt->backtrace_size = 0; + arg.btobj = Qnil; + } + else { + arg.bt->backtrace += lev; + arg.bt->backtrace_size -= lev; + } } - else { - /* gc will free object */ - return Qnil; + + if (n > 0) { + if (n < arg.bt->backtrace_size) { + arg.bt->backtrace_size = n; /* trim */ + } } + + return arg.btobj; +} + +VALUE +rb_vm_backtrace_object(void) +{ + return backtrace_object(GET_THREAD(), 0, 0); } static VALUE @@ -971,12 +1034,12 @@ rb_backtrace_to_str_ary(VALUE self) rb_backtrace_t *bt; GetCoreDataFromValue(self, rb_backtrace_t, bt); - if (bt->str) { - return bt->str; + if (bt->strary) { + return bt->strary; } else { - bt->str = backtreace_collect(bt, frame_info_to_str_override); - return bt->str; + bt->strary = backtreace_collect(bt, frame_info_to_str_override); + return bt->strary; } } @@ -988,9 +1051,9 @@ rb_backtrace_to_frame_ary(VALUE self) } static VALUE -vm_backtrace_frame_ary(rb_thread_t *th, int lev, int n) +backtrace_frame_ary(rb_thread_t *th, size_t lev, size_t n) { - return rb_backtrace_to_frame_ary(vm_backtrace_create(th, lev, n)); + return rb_backtrace_to_frame_ary(backtrace_create(th, lev, n)); } #endif @@ -1006,72 +1069,56 @@ backtrace_load_data(VALUE self, VALUE str) { rb_backtrace_t *bt; GetCoreDataFromValue(self, rb_backtrace_t, bt); - bt->str = str; + bt->strary = str; return self; } -/* old style backtrace for compatibility */ +/* old style backtrace directly */ -static int -vm_backtrace_each(rb_thread_t *th, int lev, ptrdiff_t n, void (*init)(void *), rb_backtrace_iter_func *iter, void *arg) +struct oldbt_arg { + VALUE filename; + int line_no; + void (*func)(void *data, VALUE file, int line_no, VALUE name); + void *data; /* result */ +}; + +static void +oldbt_init(void *ptr, int dmy) { - const rb_control_frame_t *limit_cfp = th->cfp; - const rb_control_frame_t *cfp = (void *)(th->stack + th->stack_size); - VALUE file = Qnil; - int line_no = 0; + struct oldbt_arg *arg = (struct oldbt_arg *)ptr; + rb_thread_t *th = GET_THREAD(); - if (n <= 0) { - n = cfp - limit_cfp; - } + arg->filename = th->vm->progname ? th->vm->progname : ruby_engine_name;; + arg->line_no = 0; + arg->data = (void *)rb_ary_new(); +} - cfp -= 2; - while (lev-- >= 0 && n > 0) { - if (++limit_cfp > cfp) { - return FALSE; - } - } - if (init) (*init)(arg); - limit_cfp = RUBY_VM_NEXT_CONTROL_FRAME(limit_cfp); - if (th->vm->progname) file = th->vm->progname; - while (cfp > limit_cfp && n>0) { - if (cfp->iseq != 0) { - if (cfp->pc != 0) { - rb_iseq_t *iseq = cfp->iseq; +static void +oldbt_iter_iseq(void *ptr, const rb_iseq_t *iseq, const VALUE *pc) +{ + struct oldbt_arg *arg = (struct oldbt_arg *)ptr; + VALUE file = arg->filename = iseq->location.filename; + VALUE name = iseq->location.name; + int line_no = arg->line_no = calc_line_no(iseq, pc); - line_no = rb_vm_get_sourceline(cfp); - file = iseq->location.filename; - n--; - if ((*iter)(arg, file, line_no, iseq->location.name)) break; - } - } - else if (RUBYVM_CFUNC_FRAME_P(cfp)) { - ID id; - - n--; - if (NIL_P(file)) file = ruby_engine_name; - if (cfp->me->def) - id = cfp->me->def->original_id; - else - id = cfp->me->called_id; - if (id != ID_ALLOCATOR && (*iter)(arg, file, line_no, rb_id2str(id))) - break; - } - cfp = RUBY_VM_NEXT_CONTROL_FRAME(cfp); - } - return TRUE; + (arg->func)(arg->data, file, line_no, name); } static void -vm_backtrace_alloc(void *arg) +oldbt_iter_cfunc(void *ptr, ID mid) { - VALUE *aryp = arg; - *aryp = rb_ary_new(); + struct oldbt_arg *arg = (struct oldbt_arg *)ptr; + VALUE file = arg->filename; + VALUE name = rb_id2str(mid); + int line_no = arg->line_no; + + (arg->func)(arg->data, file, line_no, name); } -static int -vm_backtrace_push(void *arg, VALUE file, int line_no, VALUE name) +static void +oldbt_push(void *data, VALUE file, int line_no, VALUE name) { - VALUE *aryp = arg; + VALUE ary = (VALUE)data; VALUE bt; if (line_no) { @@ -1082,23 +1129,115 @@ vm_backtrace_push(void *arg, VALUE file, int line_no, VALUE name) bt = rb_enc_sprintf(rb_enc_compatible(file, name), "%s:in `%s'", RSTRING_PTR(file), RSTRING_PTR(name)); } - rb_ary_push(*aryp, bt); - return 0; + rb_ary_push(ary, bt); } static VALUE -vm_backtrace_str_ary(rb_thread_t *th, int lev, int n) +vm_backtrace_str_ary(rb_thread_t *th, size_t lev, size_t n) +{ + struct oldbt_arg arg; + VALUE ary, result; + int i; + size_t size; + VALUE *ptr; + + arg.func = oldbt_push; + + backtrace_each(th, + oldbt_init, + oldbt_iter_iseq, + oldbt_iter_cfunc, + &arg); + + ary = (VALUE)arg.data; + size = RARRAY_LEN(ary); + + /* ["top", "2nd", ..........., "size-th"] */ + /* <-- n --> <-- lev --> */ + /* return: [.......] */ + + if (n == 0) { + n = size; + } + + if (size < lev) { + return Qnil; + } + + result = rb_ary_new(); + ptr = RARRAY_PTR(ary); + + for (i=0; i<(int)(size - lev) && i<(int)n; i++) { + rb_ary_push(result, ptr[(size - 1) - lev - i]); + } + + return result; +} + +static void +oldbt_print(void *data, VALUE file, int line_no, VALUE name) { - VALUE ary = 0; + FILE *fp = (FILE *)data; + + if (NIL_P(name)) { + fprintf(fp, "\tfrom %s:%d:in unknown method\n", + RSTRING_PTR(file), line_no); + } + else { + fprintf(fp, "\tfrom %s:%d:in `%s'\n", + RSTRING_PTR(file), line_no, RSTRING_PTR(name)); + } +} - if (lev < 0) { - ary = rb_ary_new(); +static void +vm_backtrace_print(FILE *fp) +{ + struct oldbt_arg arg; + + arg.func = oldbt_print; + arg.data = (void *)fp; + backtrace_each(GET_THREAD(), + oldbt_init, + oldbt_iter_iseq, + oldbt_iter_cfunc, + &arg); +} + +static void +oldbt_bugreport(void *arg, VALUE file, int line, VALUE method) +{ + const char *filename = NIL_P(file) ? "ruby" : RSTRING_PTR(file); + if (!*(int *)arg) { + fprintf(stderr, "-- Ruby level backtrace information " + "----------------------------------------\n"); + *(int *)arg = 1; + } + if (NIL_P(method)) { + fprintf(stderr, "%s:%d:in unknown method\n", filename, line); + } + else { + fprintf(stderr, "%s:%d:in `%s'\n", filename, line, RSTRING_PTR(method)); } - vm_backtrace_each(th, lev, 0, vm_backtrace_alloc, vm_backtrace_push, &ary); - if (!ary) return Qnil; - return rb_ary_reverse(ary); } +void +rb_backtrace_print_as_bugreport(void) +{ + struct oldbt_arg arg; + int i; + + arg.func = oldbt_bugreport; + arg.data = (int *)&i; + + backtrace_each(GET_THREAD(), + oldbt_init, + oldbt_iter_iseq, + oldbt_iter_cfunc, + &arg); +} + +/* misc */ + VALUE rb_sourcefilename(void) { -- cgit v1.2.3