root/vm_insnhelper.c

/* [previous][next][first][last][top][bottom][index][help] */

DEFINITIONS

This source file includes following definitions.
  1. vm_push_frame
  2. vm_pop_frame
  3. vm_callee_setup_arg_complex
  4. caller_setup_args
  5. call_cfunc
  6. vm_call_cfunc
  7. vm_call_bmethod
  8. vm_method_missing
  9. vm_setup_method
  10. vm_call_method
  11. vm_send_optimize
  12. block_proc_is_lambda
  13. vm_yield_with_cfunc
  14. vm_yield_setup_block_args_complex
  15. vm_yield_setup_block_args
  16. vm_yield_setup_args
  17. vm_invoke_block
  18. lfp_svar_place
  19. lfp_svar_get
  20. lfp_svar_set
  21. vm_getspecial
  22. vm_get_cref
  23. vm_cref_push
  24. vm_get_cbase
  25. vm_check_if_namespace
  26. vm_get_ev_const
  27. vm_get_cvar_base
  28. vm_method_search
  29. vm_search_normal_superclass
  30. vm_search_superclass
  31. vm_throw
  32. vm_expandarray
  33. check_cfunc
  34. opt_eq_func

/**********************************************************************

  vm_insnhelper.c - instruction helper functions.

  $Author: yugui $

  Copyright (C) 2007 Koichi Sasada

**********************************************************************/

/* finish iseq array */
#include "insns.inc"
#include <math.h>

/* control stack frame */

#ifndef INLINE
#define INLINE inline
#endif

static rb_control_frame_t *vm_get_ruby_level_caller_cfp(rb_thread_t *th, rb_control_frame_t *cfp);

static inline rb_control_frame_t *
vm_push_frame(rb_thread_t * th, const rb_iseq_t * iseq,
              VALUE type, VALUE self, VALUE specval,
              const VALUE *pc, VALUE *sp, VALUE *lfp,
              int local_size)
{
    rb_control_frame_t * const cfp = th->cfp = th->cfp - 1;
    int i;

    /* setup vm value stack */
    
    /* nil initialize */
    for (i=0; i < local_size; i++) {
        *sp = Qnil;
        sp++;
    }

    /* set special val */
    *sp = GC_GUARDED_PTR(specval);

    if (lfp == 0) {
        lfp = sp;
    }

    /* setup vm control frame stack */

    cfp->pc = (VALUE *)pc;
    cfp->sp = sp + 1;
    cfp->bp = sp + 1;
    cfp->iseq = (rb_iseq_t *) iseq;
    cfp->flag = type;
    cfp->self = self;
    cfp->lfp = lfp;
    cfp->dfp = sp;
    cfp->proc = 0;

#define COLLECT_PROFILE 0
#if COLLECT_PROFILE
    cfp->prof_time_self = clock();
    cfp->prof_time_chld = 0;
#endif

    if (VMDEBUG == 2) {
        SDR();
    }

    return cfp;
}

static inline void
vm_pop_frame(rb_thread_t *th)
{
#if COLLECT_PROFILE
    rb_control_frame_t *cfp = th->cfp;

    if (RUBY_VM_NORMAL_ISEQ_P(cfp->iseq)) {
        VALUE current_time = clock();
        rb_control_frame_t *cfp = th->cfp;
        cfp->prof_time_self = current_time - cfp->prof_time_self;
        (cfp+1)->prof_time_chld += cfp->prof_time_self;

        cfp->iseq->profile.count++;
        cfp->iseq->profile.time_cumu = cfp->prof_time_self;
        cfp->iseq->profile.time_self = cfp->prof_time_self - cfp->prof_time_chld;
    }
    else if (0 /* c method? */) {

    }
#endif
    th->cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(th->cfp);

    if (VMDEBUG == 2) {
        SDR();
    }
}

/* method dispatch */

#define VM_CALLEE_SETUP_ARG(ret, th, iseq, orig_argc, orig_argv, block) \
    if (LIKELY(iseq->arg_simple & 0x01)) { \
        /* simple check */ \
        if (orig_argc != iseq->argc) { \
            rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)", orig_argc, iseq->argc); \
        } \
        ret = 0; \
    } \
    else { \
        ret = vm_callee_setup_arg_complex(th, iseq, orig_argc, orig_argv, block); \
    }

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 int m = iseq->argc;
    int argc = orig_argc;
    VALUE *argv = orig_argv;
    int opt_pc = 0;

    th->mark_stack_len = argc + iseq->arg_size;

    /* 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);
    }

    argv += m;
    argc -= m;

    /* post arguments */
    if (iseq->arg_post_len) {
        if (!(orig_argc < iseq->arg_post_start)) {
            VALUE *new_argv = ALLOCA_N(VALUE, argc);
            MEMCPY(new_argv, argv, VALUE, argc);
            argv = new_argv;
        }

        MEMCPY(&orig_argv[iseq->arg_post_start], &argv[argc -= iseq->arg_post_len],
               VALUE, iseq->arg_post_len);
    }

    /* opt arguments */
    if (iseq->arg_opts) {
        const int opts = iseq->arg_opts - 1 /* no opt */;

        if (iseq->arg_rest == -1 && argc > opts) {
            rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)",
                     orig_argc, m + opts + iseq->arg_post_len);
        }

        if (argc > opts) {
            argc -= opts;
            argv += opts;
            opt_pc = iseq->arg_opt_table[opts]; /* no opt */
        }
        else {
            int i;
            for (i = argc; i<opts; i++) {
                orig_argv[i + m] = Qnil;
            }
            opt_pc = iseq->arg_opt_table[argc];
            argc = 0;
        }
    }

    /* rest arguments */
    if (iseq->arg_rest != -1) {
        orig_argv[iseq->arg_rest] = rb_ary_new4(argc, argv);
        argc = 0;
    }

    /* block arguments */
    if (block && iseq->arg_block != -1) {
        VALUE blockval = Qnil;
        const rb_block_t *blockptr = *block;

        if (argc != 0) {
            rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)",
                     orig_argc, m + iseq->arg_post_len);
        }

        if (blockptr) {
            /* make Proc object */
            if (blockptr->proc == 0) {
                rb_proc_t *proc;
                blockval = rb_vm_make_proc(th, blockptr, rb_cProc);
                GetProcPtr(blockval, proc);
                *block = &proc->block;
            }
            else {
                blockval = blockptr->proc;
            }
        }

        orig_argv[iseq->arg_block] = blockval; /* Proc or nil */
    }

    th->mark_stack_len = 0;
    return opt_pc;
}

static inline int
caller_setup_args(const rb_thread_t *th, rb_control_frame_t *cfp, VALUE flag,
                  int argc, rb_iseq_t *blockiseq, rb_block_t **block)
{
    rb_block_t *blockptr = 0;

    if (block) {
        if (flag & VM_CALL_ARGS_BLOCKARG_BIT) {
            rb_proc_t *po;
            VALUE proc;

            proc = *(--cfp->sp);

            if (proc != Qnil) {
                if (!rb_obj_is_proc(proc)) {
                    VALUE b = rb_check_convert_type(proc, T_DATA, "Proc", "to_proc");
                    if (NIL_P(b) || !rb_obj_is_proc(b)) {
                        rb_raise(rb_eTypeError,
                                 "wrong argument type %s (expected Proc)",
                                 rb_obj_classname(proc));
                    }
                    proc = b;
                }
                GetProcPtr(proc, po);
                blockptr = &po->block;
                RUBY_VM_GET_BLOCK_PTR_IN_CFP(cfp)->proc = proc;
                *block = blockptr;
            }
        }
        else if (blockiseq) {
            blockptr = RUBY_VM_GET_BLOCK_PTR_IN_CFP(cfp);
            blockptr->iseq = blockiseq;
            blockptr->proc = 0;
            *block = blockptr;
        }
    }

    /* expand top of stack? */
    if (flag & VM_CALL_ARGS_SPLAT_BIT) {
        VALUE ary = *(cfp->sp - 1);
        VALUE *ptr;
        int i;
        VALUE tmp = rb_check_convert_type(ary, T_ARRAY, "Array", "to_a");

        if (NIL_P(tmp)) {
            /* do nothing */
        }
        else {
            int len = RARRAY_LEN(tmp);
            ptr = RARRAY_PTR(tmp);
            cfp->sp -= 1;

            CHECK_STACK_OVERFLOW(cfp, len);

            for (i = 0; i < len; i++) {
                *cfp->sp++ = ptr[i];
            }
            argc += i-1;
        }
    }

    return argc;
}

static inline VALUE
call_cfunc(VALUE (*func)(), VALUE recv,
           int len, int argc, const VALUE *argv)
{
    /* printf("len: %d, argc: %d\n", len, argc); */

    if (len >= 0 && argc != len) {
        rb_raise(rb_eArgError, "wrong number of arguments(%d for %d)",
                 argc, len);
    }

    switch (len) {
      case -2:
        return (*func) (recv, rb_ary_new4(argc, argv));
        break;
      case -1:
        return (*func) (argc, argv, recv);
        break;
      case 0:
        return (*func) (recv);
        break;
      case 1:
        return (*func) (recv, argv[0]);
        break;
      case 2:
        return (*func) (recv, argv[0], argv[1]);
        break;
      case 3:
        return (*func) (recv, argv[0], argv[1], argv[2]);
        break;
      case 4:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3]);
        break;
      case 5:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4]);
        break;
      case 6:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5]);
        break;
      case 7:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5], argv[6]);
        break;
      case 8:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5], argv[6], argv[7]);
        break;
      case 9:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5], argv[6], argv[7], argv[8]);
        break;
      case 10:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5], argv[6], argv[7], argv[8], argv[9]);
        break;
      case 11:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5], argv[6], argv[7], argv[8], argv[9],
                        argv[10]);
        break;
      case 12:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5], argv[6], argv[7], argv[8], argv[9],
                        argv[10], argv[11]);
        break;
      case 13:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
                        argv[11], argv[12]);
        break;
      case 14:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
                        argv[11], argv[12], argv[13]);
        break;
      case 15:
        return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
                        argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
                        argv[11], argv[12], argv[13], argv[14]);
        break;
      default:
        rb_raise(rb_eArgError, "too many arguments(%d)", len);
        break;
    }
    return Qnil;                /* not reached */
}

static inline VALUE
vm_call_cfunc(rb_thread_t *th, rb_control_frame_t *reg_cfp,
              int num, ID id, ID oid, VALUE recv, VALUE klass,
              VALUE flag, const NODE *mn, const rb_block_t *blockptr)
{
    VALUE val;

    EXEC_EVENT_HOOK(th, RUBY_EVENT_C_CALL, recv, id, klass);
    {
        rb_control_frame_t *cfp =
            vm_push_frame(th, 0, VM_FRAME_MAGIC_CFUNC,
                          recv, (VALUE) blockptr, 0, reg_cfp->sp, 0, 1);

        cfp->method_id = oid;
        cfp->method_class = klass;

        reg_cfp->sp -= num + 1;

        val = call_cfunc(mn->nd_cfnc, recv, mn->nd_argc, num, reg_cfp->sp + 1);

        if (reg_cfp != th->cfp + 1) {
            rb_bug("cfp consistency error - send");
        }

        vm_pop_frame(th);
    }
    EXEC_EVENT_HOOK(th, RUBY_EVENT_C_RETURN, recv, id, klass);

    return val;
}

static inline VALUE
vm_call_bmethod(rb_thread_t *th, ID id, VALUE procval, VALUE recv,
                VALUE klass, int argc, VALUE *argv, rb_block_t *blockptr)
{
    rb_control_frame_t *cfp = th->cfp;
    rb_proc_t *proc;
    VALUE val;

    /* control block frame */
    (cfp-2)->method_id = id;
    (cfp-2)->method_class = klass;

    GetProcPtr(procval, proc);
    val = rb_vm_invoke_proc(th, proc, recv, argc, argv, blockptr);
    return val;
}

static inline VALUE
vm_method_missing(rb_thread_t *th, ID id, VALUE recv,
                  int num, rb_block_t *blockptr, int opt)
{
    VALUE val;
    rb_control_frame_t * const reg_cfp = th->cfp;
    VALUE *argv = ALLOCA_N(VALUE, num + 1);
    MEMCPY(argv, STACK_ADDR_FROM_TOP(num + 1), VALUE, num + 1);
    argv[0] = ID2SYM(id);
    th->method_missing_reason = opt;
    th->passed_block = blockptr;
    POPN(num + 1);
    val = rb_funcall2(recv, idMethodMissing, num + 1, argv);
    return val;
}

static inline void
vm_setup_method(rb_thread_t *th, rb_control_frame_t *cfp,
                const int argc, const rb_block_t *blockptr, const VALUE flag,
                const VALUE iseqval, const VALUE recv)
{
    rb_iseq_t *iseq;
    int opt_pc, i;
    VALUE *sp, *rsp = cfp->sp - argc;

    /* TODO: eliminate it */
    GetISeqPtr(iseqval, iseq);
    VM_CALLEE_SETUP_ARG(opt_pc, th, iseq, argc, rsp, &blockptr);

    /* stack overflow check */
    CHECK_STACK_OVERFLOW(cfp, iseq->stack_max);

    sp = rsp + iseq->arg_size;

    if (LIKELY(!(flag & VM_CALL_TAILCALL_BIT))) {
        if (0) printf("local_size: %d, arg_size: %d\n",
                      iseq->local_size, iseq->arg_size);

        /* clear local variables */
        for (i = 0; i < iseq->local_size - iseq->arg_size; i++) {
            *sp++ = Qnil;
        }

        vm_push_frame(th, iseq,
                      VM_FRAME_MAGIC_METHOD, recv, (VALUE) blockptr,
                      iseq->iseq_encoded + opt_pc, sp, 0, 0);

        cfp->sp = rsp - 1 /* recv */;
    }
    else {
        VALUE *p_rsp;
        th->cfp++; /* pop cf */
        p_rsp = th->cfp->sp;

        /* copy arguments */
        for (i=0; i < (sp - rsp); i++) {
            p_rsp[i] = rsp[i];
        }

        sp -= rsp - p_rsp;

        /* clear local variables */
        for (i = 0; i < iseq->local_size - iseq->arg_size; i++) {
            *sp++ = Qnil;
        }

        vm_push_frame(th, iseq,
                      VM_FRAME_MAGIC_METHOD, recv, (VALUE) blockptr,
                      iseq->iseq_encoded + opt_pc, sp, 0, 0);
    }
}

static inline VALUE
vm_call_method(rb_thread_t * const th, rb_control_frame_t * const cfp,
               const int num, rb_block_t * const blockptr, const VALUE flag,
               const ID id, const NODE * mn, const VALUE recv)
{
    VALUE val;

  start_method_dispatch:

    if (mn != 0) {
        if ((mn->nd_noex == 0)) {
            /* dispatch method */
            NODE *node;

          normal_method_dispatch:

            node = mn->nd_body;

            switch (nd_type(node)) {
              case RUBY_VM_METHOD_NODE:{
                vm_setup_method(th, cfp, num, blockptr, flag, (VALUE)node->nd_body, recv);
                return Qundef;
              }
              case NODE_CFUNC:{
                val = vm_call_cfunc(th, cfp, num, id, (ID)mn->nd_file, recv, mn->nd_clss, flag, node, blockptr);
                break;
              }
              case NODE_ATTRSET:{
                val = rb_ivar_set(recv, node->nd_vid, *(cfp->sp - 1));
                cfp->sp -= 2;
                break;
              }
              case NODE_IVAR:{
                if (num != 0) {
                    rb_raise(rb_eArgError, "wrong number of arguments (%d for 0)",
                             num);
                }
                val = rb_attr_get(recv, node->nd_vid);
                cfp->sp -= 1;
                break;
              }
              case NODE_BMETHOD:{
                VALUE *argv = ALLOCA_N(VALUE, num);
                MEMCPY(argv, cfp->sp - num, VALUE, num);
                cfp->sp += - num - 1;
                val = vm_call_bmethod(th, (ID)mn->nd_file, node->nd_cval, recv, mn->nd_clss, num, argv, blockptr);
                break;
              }
              case NODE_ZSUPER:{
                VALUE klass;
                klass = RCLASS_SUPER(mn->nd_clss);
                mn = rb_method_node(klass, id);

                if (mn != 0) {
                    goto normal_method_dispatch;
                }
                else {
                    goto start_method_dispatch;
                }
              }
              default:{
                printf("node: %s\n", ruby_node_name(nd_type(node)));
                rb_bug("eval_invoke_method: unreachable");
                /* unreachable */
                break;
              }
            }
        }
        else {
            int noex_safe;

            if (!(flag & VM_CALL_FCALL_BIT) &&
                (mn->nd_noex & NOEX_MASK) & NOEX_PRIVATE) {
                int stat = NOEX_PRIVATE;

                if (flag & VM_CALL_VCALL_BIT) {
                    stat |= NOEX_VCALL;
                }
                val = vm_method_missing(th, id, recv, num, blockptr, stat);
            }
            else if (((mn->nd_noex & NOEX_MASK) & NOEX_PROTECTED) &&
                     !(flag & VM_CALL_SEND_BIT)) {
                VALUE defined_class = mn->nd_clss;

                if (TYPE(defined_class) == T_ICLASS) {
                    defined_class = RBASIC(defined_class)->klass;
                }

                if (!rb_obj_is_kind_of(cfp->self, rb_class_real(defined_class))) {
                    val = vm_method_missing(th, id, recv, num, blockptr, NOEX_PROTECTED);
                }
                else {
                    goto normal_method_dispatch;
                }
            }
            else if ((noex_safe = NOEX_SAFE(mn->nd_noex)) > th->safe_level &&
                     (noex_safe > 2)) {
                rb_raise(rb_eSecurityError, "calling insecure method: %s", rb_id2name(id));
            }
            else {
                goto normal_method_dispatch;
            }
        }
    }
    else {
        /* method missing */
        if (id == idMethodMissing) {
            rb_bug("method missing");
        }
        else {
            int stat = 0;
            if (flag & VM_CALL_VCALL_BIT) {
                stat |= NOEX_VCALL;
            }
            if (flag & VM_CALL_SUPER_BIT) {
                stat |= NOEX_SUPER;
            }
            val = vm_method_missing(th, id, recv, num, blockptr, stat);
        }
    }

    RUBY_VM_CHECK_INTS();
    return val;
}

static inline void
vm_send_optimize(rb_control_frame_t * const reg_cfp, NODE ** const mn,
                 rb_num_t * const flag, rb_num_t * const num,
                 ID * const id, const VALUE klass)
{
    if (*mn && nd_type((*mn)->nd_body) == NODE_CFUNC) {
        NODE *node = (*mn)->nd_body;
        extern VALUE rb_f_send(int argc, VALUE *argv, VALUE recv);

        if (node->nd_cfnc == rb_f_send) {
            int i = *num - 1;
            VALUE sym = TOPN(i);
            *id = SYMBOL_P(sym) ? SYM2ID(sym) : rb_to_id(sym);

            /* shift arguments */
            if (i > 0) {
                MEMMOVE(&TOPN(i), &TOPN(i-1), VALUE, i);
            }

            *mn = rb_method_node(klass, *id);
            *num -= 1;
            DEC_SP(1);
            *flag |= VM_CALL_FCALL_BIT;
        }
    }
}

/* yield */

static inline int
block_proc_is_lambda(const VALUE procval)
{
    rb_proc_t *proc;

    if (procval) {
        GetProcPtr(procval, proc);
        return proc->is_lambda;
    }
    else {
        return 0;
    }
}

static inline VALUE
vm_yield_with_cfunc(rb_thread_t *th, const rb_block_t *block,
                    VALUE self, int argc, const VALUE *argv,
                    const rb_block_t *blockargptr)
{
    NODE *ifunc = (NODE *) block->iseq;
    VALUE val, arg, blockarg;
    int lambda = block_proc_is_lambda(block->proc);

    if (lambda) {
        arg = rb_ary_new4(argc, argv);
    }
    else if (argc == 0) {
        arg = Qnil;
    }
    else {
        arg = argv[0];
    }

    if (blockargptr) {
        if (blockargptr->proc) {
            blockarg = blockargptr->proc;
        }
        else {
            blockarg = rb_vm_make_proc(th, blockargptr, rb_cProc);
        }
    }
    else {
        blockarg = Qnil;
    }

    vm_push_frame(th, 0, VM_FRAME_MAGIC_IFUNC,
                  self, (VALUE)block->dfp,
                  0, th->cfp->sp, block->lfp, 1);

    val = (*ifunc->nd_cfnc) (arg, ifunc->nd_tval, argc, argv, blockarg);

    th->cfp++;
    return val;
}


/*--
 * @brief on supplied all of optional, rest and post parameters.
 * @pre iseq is block style (not lambda style)
 */
static inline int
vm_yield_setup_block_args_complex(rb_thread_t *th, const rb_iseq_t * iseq,
        int argc, VALUE * argv)
{
    int opt_pc = 0;
    int i;
    const int m = iseq->argc;
    const int r = iseq->arg_rest;
    int len = iseq->arg_post_len;
    int start = iseq->arg_post_start;
    int rsize = argc > m ? argc - m : 0;    /* # of arguments which did not consumed yet */
    int psize = rsize > len ? len : rsize;  /* # of post arguments */
    int osize = 0;  /* # of opt arguments */
    VALUE ary;

    /* reserves arguments for post parameters */
    rsize -= psize;

    if (iseq->arg_opts) {
        const int opts = iseq->arg_opts - 1;
        if (rsize > opts) {
            osize = opts;
            opt_pc = iseq->arg_opt_table[opts];
        }
        else {
            osize = rsize;
            opt_pc = iseq->arg_opt_table[rsize];
        }
    }
    rsize -= osize;

    if (0) {
        printf(" argc: %d\n", argc);
        printf("  len: %d\n", len);
        printf("start: %d\n", start);
        printf("rsize: %d\n", rsize);
    }

    if (r == -1) {
        /* copy post argument */
        MEMMOVE(&argv[start], &argv[m+osize], VALUE, psize);
    }
    else {
        ary = rb_ary_new4(rsize, &argv[r]);

        /* copy post argument */
        MEMMOVE(&argv[start], &argv[m+rsize+osize], VALUE, psize);
        argv[r] = ary;
    }

    for (i=psize; i<len; i++) {
        argv[start + i] = Qnil;
    }

    return opt_pc;
}

static inline int
vm_yield_setup_block_args(rb_thread_t *th, const rb_iseq_t * iseq,
        int orig_argc, VALUE * argv,
        const rb_block_t *blockptr)
{
    int i;
    int argc = orig_argc;
    const int m = iseq->argc;
    VALUE ary;
    int opt_pc = 0;

    th->mark_stack_len = argc;

    /*
     * yield [1, 2]
     *  => {|a|} => a = [1, 2]
     *  => {|a, b|} => a, b = [1, 2]
     */
    if (!(iseq->arg_simple & 0x02) &&          /* exclude {|a|} */
            (m + iseq->arg_post_len) > 0 &&    /* this process is meaningful */
            argc == 1 && !NIL_P(ary = rb_check_array_type(argv[0]))) { /* rhs is only an array */
        th->mark_stack_len = argc = RARRAY_LEN(ary);

        CHECK_STACK_OVERFLOW(th->cfp, argc);

        MEMCPY(argv, RARRAY_PTR(ary), VALUE, argc);
    }

    for (i=argc; i<m; i++) {
        argv[i] = Qnil;
    }

    if (iseq->arg_rest == -1 && iseq->arg_opts == 0) {
        const int arg_size = iseq->arg_size;
        if (arg_size < argc) {
            /*
             * yield 1, 2
             * => {|a|} # truncate
             */
            th->mark_stack_len = argc = arg_size;
        }
    }
    else {
        int r = iseq->arg_rest;

        if (iseq->arg_post_len || 
                iseq->arg_opts) { /* TODO: implement simple version for (iseq->arg_post_len==0 && iseq->arg_opts > 0) */
            opt_pc = vm_yield_setup_block_args_complex(th, iseq, argc, argv);
        }
        else {
            if (argc < r) {
                /* yield 1
                 * => {|a, b, *r|}
                 */
                for (i=argc; i<r; i++) {
                    argv[i] = Qnil;
                }
                argv[r] = rb_ary_new();
            }
            else {
                argv[r] = rb_ary_new4(argc-r, &argv[r]);
            }
        }

        th->mark_stack_len = iseq->arg_size;
    }

    /* {|&b|} */
    if (iseq->arg_block != -1) {
        VALUE procval = Qnil;

        if (blockptr) {
            procval = blockptr->proc;
        }

        argv[iseq->arg_block] = procval;
    }

    th->mark_stack_len = 0;
    return opt_pc;
}

static inline int
vm_yield_setup_args(rb_thread_t * const th, const rb_iseq_t *iseq,
                    int argc, VALUE *argv,
                    const rb_block_t *blockptr, int lambda)
{
    if (0) { /* for debug */
        printf("     argc: %d\n", argc);
        printf("iseq argc: %d\n", iseq->argc);
        printf("iseq opts: %d\n", iseq->arg_opts);
        printf("iseq rest: %d\n", iseq->arg_rest);
        printf("iseq post: %d\n", iseq->arg_post_len);
        printf("iseq blck: %d\n", iseq->arg_block);
        printf("iseq smpl: %d\n", iseq->arg_simple);
        printf("   lambda: %s\n", lambda ? "true" : "false");
    }

    if (lambda) {
        /* call as method */
        int opt_pc;
        VM_CALLEE_SETUP_ARG(opt_pc, th, iseq, argc, argv, &blockptr);
        return opt_pc;
    }
    else {
        return vm_yield_setup_block_args(th, iseq, argc, argv, blockptr);
    }
}

static VALUE
vm_invoke_block(rb_thread_t *th, rb_control_frame_t *reg_cfp, rb_num_t num, rb_num_t flag)
{
    rb_block_t * const block = GET_BLOCK_PTR();
    rb_iseq_t *iseq;
    int argc = num;

    if (GET_ISEQ()->local_iseq->type != ISEQ_TYPE_METHOD || block == 0) {
        rb_vm_localjump_error("no block given (yield)", Qnil, 0);
    }
    iseq = block->iseq;

    argc = caller_setup_args(th, GET_CFP(), flag, argc, 0, 0);

    if (BUILTIN_TYPE(iseq) != T_NODE) {
        int opt_pc;
        const int arg_size = iseq->arg_size;
        VALUE * const rsp = GET_SP() - argc;
        SET_SP(rsp);

        CHECK_STACK_OVERFLOW(GET_CFP(), iseq->stack_max);
        opt_pc = vm_yield_setup_args(th, iseq, argc, rsp, 0,
                                     block_proc_is_lambda(block->proc));

        vm_push_frame(th, iseq,
                      VM_FRAME_MAGIC_BLOCK, block->self, (VALUE) block->dfp,
                      iseq->iseq_encoded + opt_pc, rsp + arg_size, block->lfp,
                      iseq->local_size - arg_size);

        return Qundef;
    }
    else {
        VALUE val = vm_yield_with_cfunc(th, block, block->self, argc, STACK_ADDR_FROM_TOP(argc), 0);
        POPN(argc); /* TODO: should put before C/yield? */
        return val;
    }
}

/* svar */

static inline NODE *
lfp_svar_place(rb_thread_t *th, VALUE *lfp)
{
    VALUE *svar;

    if (lfp && th->local_lfp != lfp) {
        svar = &lfp[-1];
    }
    else {
        svar = &th->local_svar;
    }
    if (NIL_P(*svar)) {
        *svar = (VALUE)NEW_IF(Qnil, Qnil, Qnil);
    }
    return (NODE *)*svar;
}

static VALUE
lfp_svar_get(rb_thread_t *th, VALUE *lfp, VALUE key)
{
    NODE *svar = lfp_svar_place(th, lfp);

    switch (key) {
      case 0:
        return svar->u1.value;
      case 1:
        return svar->u2.value;
      default: {
        const VALUE hash = svar->u3.value;

        if (hash == Qnil) {
            return Qnil;
        }
        else {
            return rb_hash_lookup(hash, key);
        }
      }
    }
}

static void
lfp_svar_set(rb_thread_t *th, VALUE *lfp, VALUE key, VALUE val)
{
    NODE *svar = lfp_svar_place(th, lfp);

    switch (key) {
      case 0:
        svar->u1.value = val;
        return;
      case 1:
        svar->u2.value = val;
        return;
      default: {
        VALUE hash = svar->u3.value;

        if (hash == Qnil) {
            svar->u3.value = hash = rb_hash_new();
        }
        rb_hash_aset(hash, key, val);
      }
    }
}

static inline VALUE
vm_getspecial(rb_thread_t *th, VALUE *lfp, VALUE key, rb_num_t type)
{
    VALUE val;

    if (type == 0) {
        VALUE k = key;
        if (FIXNUM_P(key)) {
            k = FIX2INT(key);
        }
        val = lfp_svar_get(th, lfp, k);
    }
    else {
        VALUE backref = lfp_svar_get(th, lfp, 1);

        if (type & 0x01) {
            switch (type >> 1) {
              case '&':
                val = rb_reg_last_match(backref);
                break;
              case '`':
                val = rb_reg_match_pre(backref);
                break;
              case '\'':
                val = rb_reg_match_post(backref);
                break;
              case '+':
                val = rb_reg_match_last(backref);
                break;
              default:
                rb_bug("unexpected back-ref");
            }
        }
        else {
            val = rb_reg_nth_match(type >> 1, backref);
        }
    }
    return val;
}

static NODE *
vm_get_cref(const rb_iseq_t *iseq, const VALUE *lfp, const VALUE *dfp)
{
    NODE *cref = 0;

    while (1) {
        if (lfp == dfp) {
            cref = iseq->cref_stack;
            break;
        }
        else if (dfp[-1] != Qnil) {
            cref = (NODE *)dfp[-1];
            break;
        }
        dfp = GET_PREV_DFP(dfp);
    }

    if (cref == 0) {
        rb_bug("vm_get_cref: unreachable");
    }
    return cref;
}

static NODE *
vm_cref_push(rb_thread_t *th, VALUE klass, int noex)
{
    rb_control_frame_t *cfp = vm_get_ruby_level_caller_cfp(th, th->cfp);
    NODE *cref = NEW_BLOCK(klass);
    cref->nd_file = 0;
    cref->nd_visi = noex;

    if (cfp) {
        cref->nd_next = vm_get_cref(cfp->iseq, cfp->lfp, cfp->dfp);
    }

    return cref;
}

static inline VALUE
vm_get_cbase(const rb_iseq_t *iseq, const VALUE *lfp, const VALUE *dfp)
{
    NODE *cref = vm_get_cref(iseq, lfp, dfp);
    VALUE klass = Qundef;

    while (cref) {
        if ((klass = cref->nd_clss) != 0) {
            break;
        }
        cref = cref->nd_next;
    }

    return klass;
}

static inline void
vm_check_if_namespace(VALUE klass)
{
    switch (TYPE(klass)) {
      case T_CLASS:
      case T_MODULE:
        break;
      default:
        rb_raise(rb_eTypeError, "%s is not a class/module",
                 RSTRING_PTR(rb_inspect(klass)));
    }
}

static inline VALUE
vm_get_ev_const(rb_thread_t *th, const rb_iseq_t *iseq,
                VALUE orig_klass, ID id, int is_defined)
{
    VALUE val;

    if (orig_klass == Qnil) {
        /* in current lexical scope */
        const NODE *root_cref = vm_get_cref(iseq, th->cfp->lfp, th->cfp->dfp);
        const NODE *cref = root_cref;
        VALUE klass = orig_klass;

        while (cref && cref->nd_next) {
            klass = cref->nd_clss;
            cref = cref->nd_next;

            if (!NIL_P(klass)) {
              search_continue:
                if (RCLASS_IV_TBL(klass) &&
                    st_lookup(RCLASS_IV_TBL(klass), id, &val)) {
                    if (val == Qundef) {
                        rb_autoload_load(klass, id);
                        goto search_continue;
                    }
                    else {
                        if (is_defined) {
                            return 1;
                        }
                        else {
                            return val;
                        }
                    }
                }
            }
        }

        /* search self */
        klass = root_cref->nd_clss;
        if (NIL_P(klass)) {
            klass = CLASS_OF(th->cfp->self);
        }

        if (is_defined) {
            return rb_const_defined(klass, id);
        }
        else {
            return rb_const_get(klass, id);
        }
    }
    else {
        vm_check_if_namespace(orig_klass);
        if (is_defined) {
            return rb_const_defined_from(orig_klass, id);
        }
        else {
            return rb_const_get_from(orig_klass, id);
        }
    }
}

static inline VALUE
vm_get_cvar_base(NODE *cref)
{
    VALUE klass;

    while (cref && cref->nd_next &&
           (NIL_P(cref->nd_clss) || FL_TEST(cref->nd_clss, FL_SINGLETON))) {
        cref = cref->nd_next;

        if (!cref->nd_next) {
            rb_warn("class variable access from toplevel");
        }
    }

    klass = cref->nd_clss;

    if (NIL_P(klass)) {
        rb_raise(rb_eTypeError, "no class variables available");
    }
    return klass;
}

static inline NODE *
vm_method_search(VALUE id, VALUE klass, IC ic)
{
    NODE *mn;

#if OPT_INLINE_METHOD_CACHE
    {
        if (LIKELY(klass == ic->ic_class) &&
            LIKELY(GET_VM_STATE_VERSION() == ic->ic_vmstat)) {
            mn = ic->ic_method;
        }
        else {
            mn = rb_method_node(klass, id);
            ic->ic_class = klass;
            ic->ic_method = mn;
            ic->ic_vmstat = GET_VM_STATE_VERSION();
        }
    }
#else
    mn = rb_method_node(klass, id);
#endif
    return mn;
}

static inline VALUE
vm_search_normal_superclass(VALUE klass, VALUE recv)
{
    if (BUILTIN_TYPE(klass) == T_CLASS) {
        return RCLASS_SUPER(klass);
    }
    else if (BUILTIN_TYPE(klass) == T_MODULE) {
        VALUE k = CLASS_OF(recv);
        while (k) {
            if (BUILTIN_TYPE(k) == T_ICLASS && RBASIC(k)->klass == klass) {
                return RCLASS_SUPER(k);
            }
            k = RCLASS_SUPER(k);
        }
        return rb_cObject;
    }
    else {
        rb_bug("vm_search_normal_superclass: should not be reach here");
    }
}

static void
vm_search_superclass(rb_control_frame_t *reg_cfp, rb_iseq_t *ip,
                     VALUE recv, VALUE sigval,
                     ID *idp, VALUE *klassp)
{
    ID id;
    VALUE klass;

    while (ip && !ip->klass) {
        ip = ip->parent_iseq;
    }

    if (ip == 0) {
        rb_raise(rb_eNoMethodError, "super called outside of method");
    }

    id = ip->defined_method_id;

    if (ip != ip->local_iseq) {
        /* defined by Module#define_method() */
        rb_control_frame_t *lcfp = GET_CFP();

        while (lcfp->iseq != ip) {
            VALUE *tdfp = GET_PREV_DFP(lcfp->dfp);
            while (1) {
                lcfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(lcfp);
                if (lcfp->dfp == tdfp) {
                    break;
                }
            }
        }

        id = lcfp->method_id;
        klass = vm_search_normal_superclass(lcfp->method_class, recv);

        if (sigval == Qfalse) {
            /* zsuper */
            rb_raise(rb_eRuntimeError, "implicit argument passing of super from method defined by define_method() is not supported. Specify all arguments explicitly.");
        }
    }
    else {
        klass = vm_search_normal_superclass(ip->klass, recv);
    }

    *idp = id;
    *klassp = klass;
}

static VALUE
vm_throw(rb_thread_t *th, rb_control_frame_t *reg_cfp,
         rb_num_t throw_state, VALUE throwobj)
{
    rb_num_t state = throw_state & 0xff;
    rb_num_t flag = throw_state & 0x8000;
    rb_num_t level = throw_state >> 16;

    if (state != 0) {
        VALUE *pt = 0;
        int i;
        if (flag != 0) {
            pt = (void *) 1;
        }
        else {
            if (state == TAG_BREAK) {
                rb_control_frame_t *cfp = GET_CFP();
                VALUE *dfp = GET_DFP();
                int is_orphan = 1;
                rb_iseq_t *base_iseq = GET_ISEQ();

              search_parent:
                if (cfp->iseq->type != ISEQ_TYPE_BLOCK) {
                    dfp = GC_GUARDED_PTR_REF((VALUE *) *dfp);
                    base_iseq = base_iseq->parent_iseq;

                    while ((VALUE *) cfp < th->stack + th->stack_size) {
                        if (cfp->dfp == dfp) {
                            goto search_parent;
                        }
                        cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
                    }
                    rb_bug("VM (throw): can't find break base.");
                }

                if (VM_FRAME_TYPE(cfp) == VM_FRAME_MAGIC_LAMBDA) {
                    /* lambda{... break ...} */
                    is_orphan = 0;
                    pt = cfp->dfp;
                    state = TAG_RETURN;
                }
                else {
                    dfp = GC_GUARDED_PTR_REF((VALUE *) *dfp);

                    while ((VALUE *)cfp < th->stack + th->stack_size) {
                        if (cfp->dfp == dfp) {
                            VALUE epc = epc = cfp->pc - cfp->iseq->iseq_encoded;
                            rb_iseq_t *iseq = cfp->iseq;
                            int i;

                            for (i=0; i<iseq->catch_table_size; i++) {
                                struct iseq_catch_table_entry *entry = &iseq->catch_table[i];

                                if (entry->type == CATCH_TYPE_BREAK &&
                                    entry->start < epc && entry->end >= epc) {
                                    if (entry->cont == epc) {
                                        goto found;
                                    }
                                    else {
                                        break;
                                    }
                                }
                            }
                            break;

                          found:
                            pt = dfp;
                            is_orphan = 0;
                            break;
                        }
                        cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
                    }
                }

                if (is_orphan) {
                    rb_vm_localjump_error("break from proc-closure", throwobj, TAG_BREAK);
                }
            }
            else if (state == TAG_RETRY) {
                pt = GC_GUARDED_PTR_REF((VALUE *) * GET_DFP());
                for (i = 0; i < level; i++) {
                    pt = GC_GUARDED_PTR_REF((VALUE *) * pt);
                }
            }
            else if (state == TAG_RETURN) {
                rb_control_frame_t *cfp = GET_CFP();
                VALUE *dfp = GET_DFP();
                VALUE * const lfp = GET_LFP();

                /* check orphan and get dfp */
                while ((VALUE *) cfp < th->stack + th->stack_size) {
                    if (cfp->lfp == lfp) {
                        if (VM_FRAME_TYPE(cfp) == VM_FRAME_MAGIC_LAMBDA) {
                            VALUE *tdfp = dfp;

                            while (lfp != tdfp) {
                                if (cfp->dfp == tdfp) {
                                    /* in lambda */
                                    dfp = cfp->dfp;
                                    goto valid_return;
                                }
                                tdfp = GC_GUARDED_PTR_REF((VALUE *)*dfp);
                            }
                        }
                    }

                    if (cfp->dfp == lfp && cfp->iseq->type == ISEQ_TYPE_METHOD) {
                        dfp = lfp;
                        goto valid_return;
                    }

                    cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
                }

                rb_vm_localjump_error("unexpected return", throwobj, TAG_RETURN);

              valid_return:
                pt = dfp;
            }
            else {
                rb_bug("isns(throw): unsupport throw type");
            }
        }
        th->state = state;
        return (VALUE)NEW_THROW_OBJECT(throwobj, (VALUE) pt, state);
    }
    else {
        /* continue throw */
        VALUE err = throwobj;

        if (FIXNUM_P(err)) {
            th->state = FIX2INT(err);
        }
        else if (SYMBOL_P(err)) {
            th->state = TAG_THROW;
        }
        else if (BUILTIN_TYPE(err) == T_NODE) {
            th->state = GET_THROWOBJ_STATE(err);
        }
        else {
            th->state = TAG_RAISE;
            /*th->state = FIX2INT(rb_ivar_get(err, idThrowState));*/
        }
        return err;
    }
}

static inline void
vm_expandarray(rb_control_frame_t *cfp, VALUE ary, int num, int flag)
{
    int is_splat = flag & 0x01;
    int space_size = num + is_splat;
    VALUE *base = cfp->sp, *ptr;
    volatile VALUE tmp_ary;
    int len;

    if (TYPE(ary) != T_ARRAY) {
        ary = rb_ary_to_ary(ary);
    }

    cfp->sp += space_size;

    tmp_ary = ary;
    ptr = RARRAY_PTR(ary);
    len = RARRAY_LEN(ary);

    if (flag & 0x02) {
        /* post: ..., nil ,ary[-1], ..., ary[0..-num] # top */
        int i = 0, j;

        if (len < num) {
            for (i=0; i<num-len; i++) {
                *base++ = Qnil;
            }
        }
        for (j=0; i<num; i++, j++) {
            VALUE v = ptr[len - j - 1];
            *base++ = v;
        }
        if (is_splat) {
            *base = rb_ary_new4(len - j, ptr);
        }
    }
    else {
        /* normal: ary[num..-1], ary[num-2], ary[num-3], ..., ary[0] # top */
        int i;
        VALUE *bptr = &base[space_size - 1];

        for (i=0; i<num; i++) {
            if (len <= i) {
                for (; i<num; i++) {
                    *bptr-- = Qnil;
                }
                break;
            }
            *bptr-- = ptr[i];
        }
        if (is_splat) {
            if (num > len) {
                *bptr = rb_ary_new();
            }
            else {
                *bptr = rb_ary_new4(len - num, ptr + num);
            }
        }
    }
}

static inline int
check_cfunc(const NODE *mn, VALUE (*func)())
{
    if (mn && nd_type(mn->nd_body) == NODE_CFUNC &&
        mn->nd_body->nd_cfnc == func) {
        return 1;
    }
    else {
        return 0;
    }
}

static
#ifndef NO_BIG_INLINE
inline
#endif
VALUE
opt_eq_func(VALUE recv, VALUE obj, IC ic)
{
    VALUE val = Qundef;

    if (FIXNUM_2_P(recv, obj) &&
        BASIC_OP_UNREDEFINED_P(BOP_EQ)) {
        if (recv == obj) {
            val = Qtrue;
        }
        else {
            val = Qfalse;
        }
    }
    else if (!SPECIAL_CONST_P(recv) && !SPECIAL_CONST_P(obj)) {
        if (HEAP_CLASS_OF(recv) == rb_cFloat &&
                 HEAP_CLASS_OF(obj) == rb_cFloat &&
                 BASIC_OP_UNREDEFINED_P(BOP_EQ)) {
            double a = RFLOAT_VALUE(recv);
            double b = RFLOAT_VALUE(obj);

            if (isnan(a) || isnan(b)) {
                val = Qfalse;
            }
            else if (a == b) {
                val = Qtrue;
            }
            else {
                val = Qfalse;
            }
        }
        else if (HEAP_CLASS_OF(recv) == rb_cString &&
                 HEAP_CLASS_OF(obj) == rb_cString &&
                 BASIC_OP_UNREDEFINED_P(BOP_EQ)) {
            val = rb_str_equal(recv, obj);
        }
        else {
            NODE *mn = vm_method_search(idEq, CLASS_OF(recv), ic);
            extern VALUE rb_obj_equal(VALUE obj1, VALUE obj2);

            if (check_cfunc(mn, rb_obj_equal)) {
                return recv == obj ? Qtrue : Qfalse;
            }
        }
    }

    return val;
}

/* [previous][next][first][last][top][bottom][index][help] */