diff options
Diffstat (limited to 'vm_insnhelper.c')
| -rw-r--r-- | vm_insnhelper.c | 31 |
1 files changed, 21 insertions, 10 deletions
diff --git a/vm_insnhelper.c b/vm_insnhelper.c index 25b0d5b3b..5b965d215 100644 --- a/vm_insnhelper.c +++ b/vm_insnhelper.c @@ -98,22 +98,23 @@ vm_pop_frame(rb_thread_t *th) /* method dispatch */ -#define VM_CALLEE_SETUP_ARG(ret, th, iseq, orig_argc, orig_argv, block) \ +#define VM_CALLEE_SETUP_ARG(ret, th, iseq, orig_argc, orig_argv, block, e) \ if (LIKELY(iseq->arg_simple & 0x01)) { \ /* simple check */ \ + e = -1; \ + ret = 0; \ if (orig_argc != iseq->argc) { \ - rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)", orig_argc, iseq->argc); \ + e = iseq->argc; \ } \ - ret = 0; \ } \ else { \ - ret = vm_callee_setup_arg_complex(th, iseq, orig_argc, orig_argv, block); \ + ret = vm_callee_setup_arg_complex(th, iseq, orig_argc, orig_argv, block, &e); \ } static inline int vm_callee_setup_arg_complex(rb_thread_t *th, const rb_iseq_t * iseq, int orig_argc, VALUE * orig_argv, - const rb_block_t **block) + const rb_block_t **block, int *argerr) { const int m = iseq->argc; int argc = orig_argc; @@ -124,8 +125,11 @@ vm_callee_setup_arg_complex(rb_thread_t *th, const rb_iseq_t * iseq, /* mandatory */ if (argc < (m + iseq->arg_post_len)) { /* check with post arg */ - rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)", - argc, m + iseq->arg_post_len); + *argerr = m + iseq->arg_post_len; + return 0; + } + else { + *argerr = -1; } argv += m; @@ -443,8 +447,9 @@ vm_setup_method(rb_thread_t *th, rb_control_frame_t *cfp, int opt_pc, i; VALUE *sp, *rsp = cfp->sp - argc; rb_iseq_t *iseq = me->def->body.iseq; + int eargc = 0; - VM_CALLEE_SETUP_ARG(opt_pc, th, iseq, argc, rsp, &blockptr); + VM_CALLEE_SETUP_ARG(opt_pc, th, iseq, argc, rsp, &blockptr, eargc); /* stack overflow check */ CHECK_STACK_OVERFLOW(cfp, iseq->stack_max); @@ -487,6 +492,9 @@ vm_setup_method(rb_thread_t *th, rb_control_frame_t *cfp, VM_FRAME_MAGIC_METHOD, recv, (VALUE) blockptr, iseq->iseq_encoded + opt_pc, sp, 0, 0); } + if (eargc >= 0) { + rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)", argc, eargc); + } } static inline VALUE @@ -886,8 +894,11 @@ vm_yield_setup_args(rb_thread_t * const th, const rb_iseq_t *iseq, if (lambda) { /* call as method */ - int opt_pc; - VM_CALLEE_SETUP_ARG(opt_pc, th, iseq, argc, argv, &blockptr); + int opt_pc, e; + VM_CALLEE_SETUP_ARG(opt_pc, th, iseq, argc, argv, &blockptr, e); + if (e >= 0) { + rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)", argc, e); + } return opt_pc; } else { |