?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
* encoding.c (rb_filesystem_encoding): use default external encoding
[ruby-svn.git] / vm_insnhelper.c
bloba5967f736c3dae0cb2cedd9680500ce25b9e11c9
1 /**********************************************************************
2
3 insnhelper.c - instruction helper functions.
4
5 $Author$
6
7 Copyright (C) 2007 Koichi Sasada
8
9 **********************************************************************/
10
11 /* finish iseq array */
12 #include "insns.inc"
13
14 #include <math.h>
15
16 /* control stack frame */
17
18
19 #ifndef INLINE
20 #define INLINE inline
21 #endif
22
23 static inline rb_control_frame_t *
24 vm_push_frame(rb_thread_t * th, const rb_iseq_t * iseq,
25 VALUE type, VALUE self, VALUE specval,
26 const VALUE *pc, VALUE *sp, VALUE *lfp,
27 int local_size)
28 {
29 rb_control_frame_t * const cfp = th->cfp = th->cfp - 1;
30 int i;
31
32 /* setup vm value stack */
33
34 /* nil initialize */
35 for (i=0; i < local_size; i++) {
36 *sp = Qnil;
37 sp++;
38 }
39
40 /* set special val */
41 *sp = GC_GUARDED_PTR(specval);
42
43 if (lfp == 0) {
44 lfp = sp;
45 }
46
47 /* setup vm control frame stack */
48
49 cfp->pc = (VALUE *)pc;
50 cfp->sp = sp + 1;
51 cfp->bp = sp + 1;
52 cfp->iseq = (rb_iseq_t *) iseq;
53 cfp->flag = type;
54 cfp->self = self;
55 cfp->lfp = lfp;
56 cfp->dfp = sp;
57 cfp->proc = 0;
58
59 #define COLLECT_PROFILE 0
60 #if COLLECT_PROFILE
61 cfp->prof_time_self = clock();
62 cfp->prof_time_chld = 0;
63 #endif
64
65 if (VMDEBUG == 2) {
66 SDR();
67 }
68
69 return cfp;
70 }
71
72 static inline void
73 vm_pop_frame(rb_thread_t *th)
74 {
75 #if COLLECT_PROFILE
76 rb_control_frame_t *cfp = th->cfp;
77
78 if (RUBY_VM_NORMAL_ISEQ_P(cfp->iseq)) {
79 VALUE current_time = clock();
80 rb_control_frame_t *cfp = th->cfp;
81 cfp->prof_time_self = current_time - cfp->prof_time_self;
82 (cfp+1)->prof_time_chld += cfp->prof_time_self;
83
84 cfp->iseq->profile.count++;
85 cfp->iseq->profile.time_cumu = cfp->prof_time_self;
86 cfp->iseq->profile.time_self = cfp->prof_time_self - cfp->prof_time_chld;
87 }
88 else if (0 /* c method? */) {
89
90 }
91 #endif
92 th->cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(th->cfp);
93
94 if (VMDEBUG == 2) {
95 SDR();
96 }
97 }
98
99 /* method dispatch */
100
101 #define VM_CALLEE_SETUP_ARG(ret, th, iseq, orig_argc, orig_argv, block) \
102 if (LIKELY(iseq->arg_simple & 0x01)) { \
103 /* simple check */ \
104 if (orig_argc != iseq->argc) { \
105 rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)", orig_argc, iseq->argc); \
106 } \
107 ret = 0; \
108 } \
109 else { \
110 ret = vm_callee_setup_arg_complex(th, iseq, orig_argc, orig_argv, block); \
111 }
112
113 static inline int
114 vm_callee_setup_arg_complex(rb_thread_t *th, const rb_iseq_t * iseq,
115 int orig_argc, VALUE * orig_argv,
116 const rb_block_t **block)
117 {
118 const int m = iseq->argc;
119 int argc = orig_argc;
120 VALUE *argv = orig_argv;
121 int opt_pc = 0;
122
123 th->mark_stack_len = argc + iseq->arg_size;
124
125 /* mandatory */
126 if (argc < (m + iseq->arg_post_len)) { /* check with post arg */
127 rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)",
128 argc, m + iseq->arg_post_len);
129 }
130
131 argv += m;
132 argc -= m;
133
134 /* post arguments */
135 if (iseq->arg_post_len) {
136 if (!(orig_argc < iseq->arg_post_start)) {
137 VALUE *new_argv = ALLOCA_N(VALUE, argc);
138 MEMCPY(new_argv, argv, VALUE, argc);
139 argv = new_argv;
140 }
141
142 MEMCPY(&orig_argv[iseq->arg_post_start], &argv[argc -= iseq->arg_post_len],
143 VALUE, iseq->arg_post_len);
144 }
145
146 /* opt arguments */
147 if (iseq->arg_opts) {
148 const int opts = iseq->arg_opts - 1 /* no opt */;
149
150 if (iseq->arg_rest == -1 && argc > opts) {
151 rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)",
152 orig_argc, m + opts + iseq->arg_post_len);
153 }
154
155 if (argc > opts) {
156 argc -= opts;
157 argv += opts;
158 opt_pc = iseq->arg_opt_table[opts]; /* no opt */
159 }
160 else {
161 int i;
162 for (i = argc; i<opts; i++) {
163 orig_argv[i + m] = Qnil;
164 }
165 opt_pc = iseq->arg_opt_table[argc];
166 argc = 0;
167 }
168 }
169
170 /* rest arguments */
171 if (iseq->arg_rest != -1) {
172 orig_argv[iseq->arg_rest] = rb_ary_new4(argc, argv);
173 argc = 0;
174 }
175
176 /* block arguments */
177 if (block && iseq->arg_block != -1) {
178 VALUE blockval = Qnil;
179 const rb_block_t *blockptr = *block;
180
181 if (argc != 0) {
182 rb_raise(rb_eArgError, "wrong number of arguments (%d for %d)",
183 orig_argc, m + iseq->arg_post_len);
184 }
185
186 if (blockptr) {
187 /* make Proc object */
188 if (blockptr->proc == 0) {
189 rb_proc_t *proc;
190
191 blockval = vm_make_proc(th, th->cfp, blockptr);
192
193 GetProcPtr(blockval, proc);
194 *block = &proc->block;
195 }
196 else {
197 blockval = blockptr->proc;
198 }
199 }
200
201 orig_argv[iseq->arg_block] = blockval; /* Proc or nil */
202 }
203
204 th->mark_stack_len = 0;
205 return opt_pc;
206 }
207
208 static inline int
209 caller_setup_args(const rb_thread_t *th, rb_control_frame_t *cfp, VALUE flag,
210 int argc, rb_iseq_t *blockiseq, rb_block_t **block)
211 {
212 rb_block_t *blockptr = 0;
213
214 if (block) {
215 if (flag & VM_CALL_ARGS_BLOCKARG_BIT) {
216 rb_proc_t *po;
217 VALUE proc;
218
219 proc = *(--cfp->sp);
220
221 if (proc != Qnil) {
222 if (!rb_obj_is_proc(proc)) {
223 VALUE b = rb_check_convert_type(proc, T_DATA, "Proc", "to_proc");
224 if (NIL_P(b) || !rb_obj_is_proc(b)) {
225 rb_raise(rb_eTypeError,
226 "wrong argument type %s (expected Proc)",
227 rb_obj_classname(proc));
228 }
229 proc = b;
230 }
231 GetProcPtr(proc, po);
232 blockptr = &po->block;
233 RUBY_VM_GET_BLOCK_PTR_IN_CFP(cfp)->proc = proc;
234 *block = blockptr;
235 }
236 }
237 else if (blockiseq) {
238 blockptr = RUBY_VM_GET_BLOCK_PTR_IN_CFP(cfp);
239 blockptr->iseq = blockiseq;
240 blockptr->proc = 0;
241 *block = blockptr;
242 }
243 }
244
245 /* expand top of stack? */
246 if (flag & VM_CALL_ARGS_SPLAT_BIT) {
247 VALUE ary = *(cfp->sp - 1);
248 VALUE *ptr;
249 int i;
250 VALUE tmp = rb_check_convert_type(ary, T_ARRAY, "Array", "to_a");
251
252 if (NIL_P(tmp)) {
253 /* do nothing */
254 }
255 else {
256 int len = RARRAY_LEN(tmp);
257 ptr = RARRAY_PTR(tmp);
258 cfp->sp -= 1;
259
260 CHECK_STACK_OVERFLOW(cfp, len);
261
262 for (i = 0; i < len; i++) {
263 *cfp->sp++ = ptr[i];
264 }
265 argc += i-1;
266 }
267 }
268
269 return argc;
270 }
271
272 static inline VALUE
273 call_cfunc(VALUE (*func)(), VALUE recv,
274 int len, int argc, const VALUE *argv)
275 {
276 /* printf("len: %d, argc: %d\n", len, argc); */
277
278 if (len >= 0 && argc != len) {
279 rb_raise(rb_eArgError, "wrong number of arguments(%d for %d)",
280 argc, len);
281 }
282
283 switch (len) {
284 case -2:
285 return (*func) (recv, rb_ary_new4(argc, argv));
286 break;
287 case -1:
288 return (*func) (argc, argv, recv);
289 break;
290 case 0:
291 return (*func) (recv);
292 break;
293 case 1:
294 return (*func) (recv, argv[0]);
295 break;
296 case 2:
297 return (*func) (recv, argv[0], argv[1]);
298 break;
299 case 3:
300 return (*func) (recv, argv[0], argv[1], argv[2]);
301 break;
302 case 4:
303 return (*func) (recv, argv[0], argv[1], argv[2], argv[3]);
304 break;
305 case 5:
306 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4]);
307 break;
308 case 6:
309 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
310 argv[5]);
311 break;
312 case 7:
313 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
314 argv[5], argv[6]);
315 break;
316 case 8:
317 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
318 argv[5], argv[6], argv[7]);
319 break;
320 case 9:
321 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
322 argv[5], argv[6], argv[7], argv[8]);
323 break;
324 case 10:
325 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
326 argv[5], argv[6], argv[7], argv[8], argv[9]);
327 break;
328 case 11:
329 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
330 argv[5], argv[6], argv[7], argv[8], argv[9],
331 argv[10]);
332 break;
333 case 12:
334 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
335 argv[5], argv[6], argv[7], argv[8], argv[9],
336 argv[10], argv[11]);
337 break;
338 case 13:
339 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
340 argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
341 argv[11], argv[12]);
342 break;
343 case 14:
344 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
345 argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
346 argv[11], argv[12], argv[13]);
347 break;
348 case 15:
349 return (*func) (recv, argv[0], argv[1], argv[2], argv[3], argv[4],
350 argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
351 argv[11], argv[12], argv[13], argv[14]);
352 break;
353 default:
354 rb_raise(rb_eArgError, "too many arguments(%d)", len);
355 break;
356 }
357 return Qnil; /* not reached */
358 }
359
360 static inline VALUE
361 vm_call_cfunc(rb_thread_t *th, rb_control_frame_t *reg_cfp,
362 int num, ID id, VALUE recv, VALUE klass,
363 VALUE flag, const NODE *mn, const rb_block_t *blockptr)
364 {
365 VALUE val;
366
367 EXEC_EVENT_HOOK(th, RUBY_EVENT_C_CALL, recv, id, klass);
368 {
369 rb_control_frame_t *cfp =
370 vm_push_frame(th, 0, VM_FRAME_MAGIC_CFUNC,
371 recv, (VALUE) blockptr, 0, reg_cfp->sp, 0, 1);
372
373 cfp->method_id = id;
374 cfp->method_class = klass;
375
376 reg_cfp->sp -= num + 1;
377
378 val = call_cfunc(mn->nd_cfnc, recv, mn->nd_argc, num, reg_cfp->sp + 1);
379
380 if (reg_cfp != th->cfp + 1) {
381 rb_bug("cfp consistency error - send");
382 }
383
384 vm_pop_frame(th);
385 }
386 EXEC_EVENT_HOOK(th, RUBY_EVENT_C_RETURN, recv, id, klass);
387
388 return val;
389 }
390
391 static inline VALUE
392 vm_call_bmethod(rb_thread_t *th, ID id, VALUE procval, VALUE recv,
393 VALUE klass, int argc, VALUE *argv, rb_block_t *blockptr)
394 {
395 rb_control_frame_t *cfp = th->cfp;
396 rb_proc_t *proc;
397 VALUE val;
398
399 /* control block frame */
400 (cfp-2)->method_id = id;
401 (cfp-2)->method_class = klass;
402
403 GetProcPtr(procval, proc);
404 val = vm_invoke_proc(th, proc, recv, argc, argv, blockptr);
405 return val;
406 }
407
408 static inline VALUE
409 vm_method_missing(rb_thread_t *th, ID id, VALUE recv,
410 int num, rb_block_t *blockptr, int opt)
411 {
412 rb_control_frame_t * const reg_cfp = th->cfp;
413 VALUE *argv = STACK_ADDR_FROM_TOP(num + 1);
414 VALUE val;
415 argv[0] = ID2SYM(id);
416 th->method_missing_reason = opt;
417 th->passed_block = blockptr;
418 val = rb_funcall2(recv, idMethodMissing, num + 1, argv);
419 POPN(num + 1);
420 return val;
421 }
422
423 static inline void
424 vm_setup_method(rb_thread_t *th, rb_control_frame_t *cfp,
425 const int argc, const rb_block_t *blockptr, const VALUE flag,
426 const VALUE iseqval, const VALUE recv, const VALUE klass)
427 {
428 rb_iseq_t *iseq;
429 int opt_pc, i;
430 VALUE *sp, *rsp = cfp->sp - argc;
431
432 /* TODO: eliminate it */
433 GetISeqPtr(iseqval, iseq);
434 VM_CALLEE_SETUP_ARG(opt_pc, th, iseq, argc, rsp, &blockptr);
435
436 /* stack overflow check */
437 CHECK_STACK_OVERFLOW(cfp, iseq->stack_max);
438
439 sp = rsp + iseq->arg_size;
440
441 if (LIKELY(!(flag & VM_CALL_TAILCALL_BIT))) {
442 if (0) printf("local_size: %d, arg_size: %d\n",
443 iseq->local_size, iseq->arg_size);
444
445 /* clear local variables */
446 for (i = 0; i < iseq->local_size - iseq->arg_size; i++) {
447 *sp++ = Qnil;
448 }
449
450 vm_push_frame(th, iseq,
451 VM_FRAME_MAGIC_METHOD, recv, (VALUE) blockptr,
452 iseq->iseq_encoded + opt_pc, sp, 0, 0);
453
454 cfp->sp = rsp - 1 /* recv */;
455 }
456 else {
457 VALUE *p_rsp;
458 th->cfp++; /* pop cf */
459 p_rsp = th->cfp->sp;
460
461 /* copy arguments */
462 for (i=0; i < (sp - rsp); i++) {
463 p_rsp[i] = rsp[i];
464 }
465
466 sp -= rsp - p_rsp;
467
468 /* clear local variables */
469 for (i = 0; i < iseq->local_size - iseq->arg_size; i++) {
470 *sp++ = Qnil;
471 }
472
473 vm_push_frame(th, iseq,
474 VM_FRAME_MAGIC_METHOD, recv, (VALUE) blockptr,
475 iseq->iseq_encoded + opt_pc, sp, 0, 0);
476 }
477 }
478
479 static inline VALUE
480 vm_call_method(rb_thread_t * const th, rb_control_frame_t * const cfp,
481 const int num, rb_block_t * const blockptr, const VALUE flag,
482 const ID id, const NODE * mn, const VALUE recv, VALUE klass)
483 {
484 VALUE val;
485
486 start_method_dispatch:
487
488 if (mn != 0) {
489 if ((mn->nd_noex == 0)) {
490 /* dispatch method */
491 NODE *node;
492
493 normal_method_dispatch:
494
495 node = mn->nd_body;
496
497 switch (nd_type(node)) {
498 case RUBY_VM_METHOD_NODE:{
499 vm_setup_method(th, cfp, num, blockptr, flag, (VALUE)node->nd_body, recv, klass);
500 return Qundef;
501 }
502 case NODE_CFUNC:{
503 val = vm_call_cfunc(th, cfp, num, id, recv, mn->nd_clss, flag, node, blockptr);
504 break;
505 }
506 case NODE_ATTRSET:{
507 val = rb_ivar_set(recv, node->nd_vid, *(cfp->sp - 1));
508 cfp->sp -= 2;
509 break;
510 }
511 case NODE_IVAR:{
512 if (num != 0) {
513 rb_raise(rb_eArgError, "wrong number of arguments (%d for 0)",
514 num);
515 }
516 val = rb_attr_get(recv, node->nd_vid);
517 cfp->sp -= 1;
518 break;
519 }
520 case NODE_BMETHOD:{
521 VALUE *argv = cfp->sp - num;
522 val = vm_call_bmethod(th, id, node->nd_cval, recv, klass, num, argv, blockptr);
523 cfp->sp += - num - 1;
524 break;
525 }
526 case NODE_ZSUPER:{
527 klass = RCLASS_SUPER(mn->nd_clss);
528 mn = rb_method_node(klass, id);
529
530 if (mn != 0) {
531 goto normal_method_dispatch;
532 }
533 else {
534 goto start_method_dispatch;
535 }
536 }
537 default:{
538 printf("node: %s\n", ruby_node_name(nd_type(node)));
539 rb_bug("eval_invoke_method: unreachable");
540 /* unreachable */
541 break;
542 }
543 }
544 }
545 else {
546 int noex_safe;
547
548 if (!(flag & VM_CALL_FCALL_BIT) &&
549 (mn->nd_noex & NOEX_MASK) & NOEX_PRIVATE) {
550 int stat = NOEX_PRIVATE;
551
552 if (flag & VM_CALL_VCALL_BIT) {
553 stat |= NOEX_VCALL;
554 }
555 val = vm_method_missing(th, id, recv, num, blockptr, stat);
556 }
557 else if (((mn->nd_noex & NOEX_MASK) & NOEX_PROTECTED) &&
558 !(flag & VM_CALL_SEND_BIT)) {
559 VALUE defined_class = mn->nd_clss;
560
561 if (TYPE(defined_class) == T_ICLASS) {
562 defined_class = RBASIC(defined_class)->klass;
563 }
564
565 if (!rb_obj_is_kind_of(cfp->self, rb_class_real(defined_class))) {
566 val = vm_method_missing(th, id, recv, num, blockptr, NOEX_PROTECTED);
567 }
568 else {
569 goto normal_method_dispatch;
570 }
571 }
572 else if ((noex_safe = NOEX_SAFE(mn->nd_noex)) > th->safe_level &&
573 (noex_safe > 2)) {
574 rb_raise(rb_eSecurityError, "calling insecure method: %s", rb_id2name(id));
575 }
576 else {
577 goto normal_method_dispatch;
578 }
579 }
580 }
581 else {
582 /* method missing */
583 if (id == idMethodMissing) {
584 rb_bug("method missing");
585 }
586 else {
587 int stat = 0;
588 if (flag & VM_CALL_VCALL_BIT) {
589 stat |= NOEX_VCALL;
590 }
591 if (flag & VM_CALL_SUPER_BIT) {
592 stat |= NOEX_SUPER;
593 }
594 val = vm_method_missing(th, id, recv, num, blockptr, stat);
595 }
596 }
597
598 RUBY_VM_CHECK_INTS();
599 return val;
600 }
601
602 static inline void
603 vm_send_optimize(rb_control_frame_t * const reg_cfp, NODE ** const mn,
604 rb_num_t * const flag, rb_num_t * const num,
605 ID * const id, const VALUE klass)
606 {
607 if (*mn && nd_type((*mn)->nd_body) == NODE_CFUNC) {
608 NODE *node = (*mn)->nd_body;
609 extern VALUE rb_f_send(int argc, VALUE *argv, VALUE recv);
610
611 if (node->nd_cfnc == rb_f_send) {
612 int i = *num - 1;
613 VALUE sym = TOPN(i);
614 *id = SYMBOL_P(sym) ? SYM2ID(sym) : rb_to_id(sym);
615
616 /* shift arguments */
617 if (i > 0) {
618 MEMMOVE(&TOPN(i), &TOPN(i-1), VALUE, i);
619 }
620
621 *mn = rb_method_node(klass, *id);
622 *num -= 1;
623 DEC_SP(1);
624 *flag |= VM_CALL_FCALL_BIT;
625 }
626 }
627 }
628
629 /* yield */
630
631 static inline int
632 block_proc_is_lambda(const VALUE procval)
633 {
634 rb_proc_t *proc;
635
636 if (procval) {
637 GetProcPtr(procval, proc);
638 return proc->is_lambda;
639 }
640 else {
641 return 0;
642 }
643 }
644
645 static inline VALUE
646 vm_yield_with_cfunc(rb_thread_t *th, const rb_block_t *block,
647 VALUE self, int argc, const VALUE *argv,
648 const rb_block_t *blockptr)
649 {
650 NODE *ifunc = (NODE *) block->iseq;
651 VALUE val, arg, blockarg;
652 int lambda = block_proc_is_lambda(block->proc);
653
654 if (lambda) {
655 arg = rb_ary_new4(argc, argv);
656 }
657 else if (argc == 0) {
658 arg = Qnil;
659 }
660 else {
661 arg = argv[0];
662 }
663
664 if (blockptr) {
665 blockarg = vm_make_proc(th, th->cfp, blockptr);
666 }
667 else {
668 blockarg = Qnil;
669 }
670
671 vm_push_frame(th, 0, VM_FRAME_MAGIC_IFUNC,
672 self, (VALUE)block->dfp,
673 0, th->cfp->sp, block->lfp, 1);
674
675 val = (*ifunc->nd_cfnc) (arg, ifunc->nd_tval, argc, argv, blockarg);
676
677 th->cfp++;
678 return val;
679 }
680
681 static inline int
682 vm_yield_setup_args(rb_thread_t * const th, const rb_iseq_t *iseq,
683 int orig_argc, VALUE *argv,
684 const rb_block_t *blockptr, int lambda)
685 {
686 if (0) { /* for debug */
687 printf(" argc: %d\n", orig_argc);
688 printf("iseq argc: %d\n", iseq->argc);
689 printf("iseq opts: %d\n", iseq->arg_opts);
690 printf("iseq rest: %d\n", iseq->arg_rest);
691 printf("iseq post: %d\n", iseq->arg_post_len);
692 printf("iseq blck: %d\n", iseq->arg_block);
693 printf("iseq smpl: %d\n", iseq->arg_simple);
694 printf(" lambda: %s\n", lambda ? "true" : "false");
695 }
696
697 if (lambda) {
698 /* call as method */
699 int opt_pc;
700 VM_CALLEE_SETUP_ARG(opt_pc, th, iseq, orig_argc, argv, &blockptr);
701 return opt_pc;
702 }
703 else {
704 int i;
705 int argc = orig_argc;
706 const int m = iseq->argc;
707
708 th->mark_stack_len = argc;
709
710 /*
711 * yield [1, 2]
712 * => {|a|} => a = [1, 2]
713 * => {|a, b|} => a, b = [1, 2]
714 */
715 if (!(iseq->arg_simple & 0x02) &&
716 (m + iseq->arg_post_len) > 0 &&
717 argc == 1 && TYPE(argv[0]) == T_ARRAY) {
718 VALUE ary = argv[0];
719 th->mark_stack_len = argc = RARRAY_LEN(ary);
720
721 CHECK_STACK_OVERFLOW(th->cfp, argc);
722
723 MEMCPY(argv, RARRAY_PTR(ary), VALUE, argc);
724 }
725
726 for (i=argc; i<m; i++) {
727 argv[i] = Qnil;
728 }
729
730 if (iseq->arg_rest == -1) {
731 if (m < argc) {
732 /*
733 * yield 1, 2
734 * => {|a|} # truncate
735 */
736 th->mark_stack_len = argc = m;
737 }
738 }
739 else {
740 int r = iseq->arg_rest;
741
742 if (iseq->arg_post_len) {
743 int len = iseq->arg_post_len;
744 int start = iseq->arg_post_start;
745 int rsize = argc > m ? argc - m : 0;
746 int psize = rsize;
747 VALUE ary;
748
749 if (psize > len) psize = len;
750
751 ary = rb_ary_new4(rsize - psize, &argv[r]);
752
753 if (0) {
754 printf(" argc: %d\n", argc);
755 printf(" len: %d\n", len);
756 printf("start: %d\n", start);
757 printf("rsize: %d\n", rsize);
758 }
759
760 /* copy post argument */
761 MEMMOVE(&argv[start], &argv[r + rsize - psize], VALUE, psize);
762
763 for (i=psize; i<len; i++) {
764 argv[start + i] = Qnil;
765 }
766 argv[r] = ary;
767 }
768 else {
769 if (argc < r) {
770 /* yield 1
771 * => {|a, b, *r|}
772 */
773 for (i=argc; i<r; i++) {
774 argv[i] = Qnil;
775 }
776 argv[r] = rb_ary_new();
777 }
778 else {
779 argv[r] = rb_ary_new4(argc-r, &argv[r]);
780 }
781 }
782
783 th->mark_stack_len = iseq->arg_size;
784 }
785
786 /* {|&b|} */
787 if (iseq->arg_block != -1) {
788 VALUE procval = Qnil;
789
790 if (blockptr) {
791 procval = blockptr->proc;
792 }
793
794 argv[iseq->arg_block] = procval;
795 }
796
797 th->mark_stack_len = 0;
798 return 0;
799 }
800 }
801
802 static VALUE
803 vm_invoke_block(rb_thread_t *th, rb_control_frame_t *reg_cfp, rb_num_t num, rb_num_t flag)
804 {
805 rb_block_t * const block = GET_BLOCK_PTR();
806 rb_iseq_t *iseq;
807 int argc = num;
808
809 if (GET_ISEQ()->local_iseq->type != ISEQ_TYPE_METHOD || block == 0) {
810 vm_localjump_error("no block given (yield)", Qnil, 0);
811 }
812 iseq = block->iseq;
813
814 argc = caller_setup_args(th, GET_CFP(), flag, argc, 0, 0);
815
816 if (BUILTIN_TYPE(iseq) != T_NODE) {
817 int opt_pc;
818 const int arg_size = iseq->arg_size;
819 VALUE * const rsp = GET_SP() - argc;
820 SET_SP(rsp);
821
822 CHECK_STACK_OVERFLOW(GET_CFP(), iseq->stack_max);
823 opt_pc = vm_yield_setup_args(th, iseq, argc, rsp, 0,
824 block_proc_is_lambda(block->proc));
825
826 vm_push_frame(th, iseq,
827 VM_FRAME_MAGIC_BLOCK, block->self, (VALUE) block->dfp,
828 iseq->iseq_encoded + opt_pc, rsp + arg_size, block->lfp,
829 iseq->local_size - arg_size);
830
831 return Qundef;
832 }
833 else {
834 VALUE val = vm_yield_with_cfunc(th, block, block->self, argc, STACK_ADDR_FROM_TOP(argc), 0);
835 POPN(argc); /* TODO: should put before C/yield? */
836 return val;
837 }
838 }
839
840 /* svar */
841
842 static inline NODE *
843 lfp_svar_place(rb_thread_t *th, VALUE *lfp)
844 {
845 VALUE *svar;
846
847 if (lfp && th->local_lfp != lfp) {
848 svar = &lfp[-1];
849 }
850 else {
851 svar = &th->local_svar;
852 }
853 if (NIL_P(*svar)) {
854 *svar = (VALUE)NEW_IF(Qnil, Qnil, Qnil);
855 }
856 return (NODE *)*svar;
857 }
858
859 static VALUE
860 lfp_svar_get(rb_thread_t *th, VALUE *lfp, VALUE key)
861 {
862 NODE *svar = lfp_svar_place(th, lfp);
863
864 switch (key) {
865 case 0:
866 return svar->u1.value;
867 case 1:
868 return svar->u2.value;
869 default: {
870 const VALUE hash = svar->u3.value;
871
872 if (hash == Qnil) {
873 return Qnil;
874 }
875 else {
876 return rb_hash_lookup(hash, key);
877 }
878 }
879 }
880 }
881
882 static void
883 lfp_svar_set(rb_thread_t *th, VALUE *lfp, VALUE key, VALUE val)
884 {
885 NODE *svar = lfp_svar_place(th, lfp);
886
887 switch (key) {
888 case 0:
889 svar->u1.value = val;
890 return;
891 case 1:
892 svar->u2.value = val;
893 return;
894 default: {
895 VALUE hash = svar->u3.value;
896
897 if (hash == Qnil) {
898 svar->u3.value = hash = rb_hash_new();
899 }
900 rb_hash_aset(hash, key, val);
901 }
902 }
903 }
904
905 static inline VALUE
906 vm_getspecial(rb_thread_t *th, VALUE *lfp, VALUE key, rb_num_t type)
907 {
908 VALUE val;
909
910 if (type == 0) {
911 VALUE k = key;
912 if (FIXNUM_P(key)) {
913 k = FIX2INT(key);
914 }
915 val = lfp_svar_get(th, lfp, k);
916 }
917 else {
918 VALUE backref = lfp_svar_get(th, lfp, 1);
919
920 if (type & 0x01) {
921 switch (type >> 1) {
922 case '&':
923 val = rb_reg_last_match(backref);
924 break;
925 case '`':
926 val = rb_reg_match_pre(backref);
927 break;
928 case '\'':
929 val = rb_reg_match_post(backref);
930 break;
931 case '+':
932 val = rb_reg_match_last(backref);
933 break;
934 default:
935 rb_bug("unexpected back-ref");
936 }
937 }
938 else {
939 val = rb_reg_nth_match(type >> 1, backref);
940 }
941 }
942 return val;
943 }
944
945 static NODE *
946 vm_get_cref(const rb_iseq_t *iseq, const VALUE *lfp, const VALUE *dfp)
947 {
948 NODE *cref = 0;
949
950 while (1) {
951 if (lfp == dfp) {
952 cref = iseq->cref_stack;
953 break;
954 }
955 else if (dfp[-1] != Qnil) {
956 cref = (NODE *)dfp[-1];
957 break;
958 }
959 dfp = GET_PREV_DFP(dfp);
960 }
961
962 if (cref == 0) {
963 rb_bug("vm_get_cref: unreachable");
964 }
965 return cref;
966 }
967
968
969 static inline void
970 vm_check_if_namespace(VALUE klass)
971 {
972 switch (TYPE(klass)) {
973 case T_CLASS:
974 case T_MODULE:
975 break;
976 default:
977 rb_raise(rb_eTypeError, "%s is not a class/module",
978 RSTRING_PTR(rb_inspect(klass)));
979 }
980 }
981
982 static inline VALUE
983 vm_get_ev_const(rb_thread_t *th, const rb_iseq_t *iseq,
984 VALUE orig_klass, ID id, int is_defined)
985 {
986 VALUE val;
987
988 if (orig_klass == Qnil) {
989 /* in current lexical scope */
990 const NODE *root_cref = vm_get_cref(iseq, th->cfp->lfp, th->cfp->dfp);
991 const NODE *cref = root_cref;
992 VALUE klass = orig_klass;
993
994 while (cref && cref->nd_next) {
995 klass = cref->nd_clss;
996 cref = cref->nd_next;
997
998 if (!NIL_P(klass)) {
999 search_continue:
1000 if (RCLASS_IV_TBL(klass) &&
1001 st_lookup(RCLASS_IV_TBL(klass), id, &val)) {
1002 if (val == Qundef) {
1003 rb_autoload_load(klass, id);
1004 goto search_continue;
1005 }
1006 else {
1007 if (is_defined) {
1008 return 1;
1009 }
1010 else {
1011 return val;
1012 }
1013 }
1014 }
1015 }
1016 }
1017
1018 /* search self */
1019 klass = root_cref->nd_clss;
1020 if (NIL_P(klass)) {
1021 klass = CLASS_OF(th->cfp->self);
1022 }
1023
1024 if (is_defined) {
1025 return rb_const_defined(klass, id);
1026 }
1027 else {
1028 return rb_const_get(klass, id);
1029 }
1030 }
1031 else {
1032 vm_check_if_namespace(orig_klass);
1033 if (is_defined) {
1034 return rb_const_defined_from(orig_klass, id);
1035 }
1036 else {
1037 return rb_const_get_from(orig_klass, id);
1038 }
1039 }
1040 }
1041
1042 static inline VALUE
1043 vm_get_cvar_base(NODE *cref)
1044 {
1045 VALUE klass;
1046
1047 while (cref && cref->nd_next &&
1048 (NIL_P(cref->nd_clss) || FL_TEST(cref->nd_clss, FL_SINGLETON))) {
1049 cref = cref->nd_next;
1050
1051 if (!cref->nd_next) {
1052 rb_warn("class variable access from toplevel");
1053 }
1054 }
1055
1056 klass = cref->nd_clss;
1057
1058 if (NIL_P(klass)) {
1059 rb_raise(rb_eTypeError, "no class variables available");
1060 }
1061 return klass;
1062 }
1063
1064 static inline NODE *
1065 vm_method_search(VALUE id, VALUE klass, IC ic)
1066 {
1067 NODE *mn;
1068
1069 #if OPT_INLINE_METHOD_CACHE
1070 {
1071 if (LIKELY(klass == ic->ic_class) &&
1072 LIKELY(GET_VM_STATE_VERSION() == ic->ic_vmstat)) {
1073 mn = ic->ic_method;
1074 }
1075 else {
1076 mn = rb_method_node(klass, id);
1077 ic->ic_class = klass;
1078 ic->ic_method = mn;
1079 ic->ic_vmstat = GET_VM_STATE_VERSION();
1080 }
1081 }
1082 #else
1083 mn = rb_method_node(klass, id);
1084 #endif
1085 return mn;
1086 }
1087
1088 static inline VALUE
1089 vm_search_normal_superclass(VALUE klass, VALUE recv)
1090 {
1091 if (BUILTIN_TYPE(klass) == T_CLASS) {
1092 klass = RCLASS_SUPER(klass);
1093 }
1094 else if (BUILTIN_TYPE(klass) == T_MODULE) {
1095 VALUE k = CLASS_OF(recv);
1096 while (k) {
1097 if (BUILTIN_TYPE(k) == T_ICLASS && RBASIC(k)->klass == klass) {
1098 klass = RCLASS_SUPER(k);
1099 break;
1100 }
1101 k = RCLASS_SUPER(k);
1102 }
1103 }
1104 return klass;
1105 }
1106
1107 static void
1108 vm_search_superclass(rb_control_frame_t *reg_cfp, rb_iseq_t *ip,
1109 VALUE recv, VALUE sigval,
1110 ID *idp, VALUE *klassp)
1111 {
1112 ID id;
1113 VALUE klass;
1114
1115 while (ip && !ip->klass) {
1116 ip = ip->parent_iseq;
1117 }
1118
1119 if (ip == 0) {
1120 rb_raise(rb_eNoMethodError, "super called outside of method");
1121 }
1122
1123 id = ip->defined_method_id;
1124
1125 if (ip != ip->local_iseq) {
1126 /* defined by Module#define_method() */
1127 rb_control_frame_t *lcfp = GET_CFP();
1128
1129 while (lcfp->iseq != ip) {
1130 VALUE *tdfp = GET_PREV_DFP(lcfp->dfp);
1131 while (1) {
1132 lcfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(lcfp);
1133 if (lcfp->dfp == tdfp) {
1134 break;
1135 }
1136 }
1137 }
1138
1139 id = lcfp->method_id;
1140 klass = vm_search_normal_superclass(lcfp->method_class, recv);
1141
1142 if (sigval == Qfalse) {
1143 /* zsuper */
1144 rb_raise(rb_eRuntimeError, "implicit argument passing of super from method defined by define_method() is not supported. Specify all arguments explicitly.");
1145 }
1146 }
1147 else {
1148 klass = vm_search_normal_superclass(ip->klass, recv);
1149 }
1150
1151 *idp = id;
1152 *klassp = klass;
1153 }
1154
1155 static VALUE
1156 vm_throw(rb_thread_t *th, rb_control_frame_t *reg_cfp,
1157 rb_num_t throw_state, VALUE throwobj)
1158 {
1159 rb_num_t state = throw_state & 0xff;
1160 rb_num_t flag = throw_state & 0x8000;
1161 rb_num_t level = throw_state >> 16;
1162
1163 if (state != 0) {
1164 VALUE *pt = 0;
1165 int i;
1166 if (flag != 0) {
1167 pt = (void *) 1;
1168 }
1169 else {
1170 if (state == TAG_BREAK) {
1171 rb_control_frame_t *cfp = GET_CFP();
1172 VALUE *dfp = GET_DFP();
1173 int is_orphan = 1;
1174 rb_iseq_t *base_iseq = GET_ISEQ();
1175
1176 search_parent:
1177 if (cfp->iseq->type != ISEQ_TYPE_BLOCK) {
1178 dfp = GC_GUARDED_PTR_REF((VALUE *) *dfp);
1179 base_iseq = base_iseq->parent_iseq;
1180
1181 while ((VALUE *) cfp < th->stack + th->stack_size) {
1182 if (cfp->dfp == dfp) {
1183 goto search_parent;
1184 }
1185 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
1186 }
1187 rb_bug("VM (throw): can't find break base.");
1188 }
1189
1190 if (VM_FRAME_TYPE(cfp) == VM_FRAME_MAGIC_LAMBDA) {
1191 /* lambda{... break ...} */
1192 is_orphan = 0;
1193 pt = cfp->dfp;
1194 state = TAG_RETURN;
1195 }
1196 else {
1197 dfp = GC_GUARDED_PTR_REF((VALUE *) *dfp);
1198
1199 while ((VALUE *)cfp < th->stack + th->stack_size) {
1200 if (cfp->dfp == dfp) {
1201 VALUE epc = epc = cfp->pc - cfp->iseq->iseq_encoded;
1202 rb_iseq_t *iseq = cfp->iseq;
1203 int i;
1204
1205 for (i=0; i<iseq->catch_table_size; i++) {
1206 struct iseq_catch_table_entry *entry = &iseq->catch_table[i];
1207
1208 if (entry->type == CATCH_TYPE_BREAK &&
1209 entry->start < epc && entry->end >= epc) {
1210 if (entry->cont == epc) {
1211 goto found;
1212 }
1213 else {
1214 break;
1215 }
1216 }
1217 }
1218 break;
1219
1220 found:
1221 pt = dfp;
1222 is_orphan = 0;
1223 break;
1224 }
1225 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
1226 }
1227 }
1228
1229 if (is_orphan) {
1230 vm_localjump_error("break from proc-closure", throwobj, TAG_BREAK);
1231 }
1232 }
1233 else if (state == TAG_RETRY) {
1234 pt = GC_GUARDED_PTR_REF((VALUE *) * GET_DFP());
1235 for (i = 0; i < level; i++) {
1236 pt = GC_GUARDED_PTR_REF((VALUE *) * pt);
1237 }
1238 }
1239 else if (state == TAG_RETURN) {
1240 rb_control_frame_t *cfp = GET_CFP();
1241 VALUE *dfp = GET_DFP();
1242 VALUE * const lfp = GET_LFP();
1243
1244 /* check orphan and get dfp */
1245 while ((VALUE *) cfp < th->stack + th->stack_size) {
1246 if (cfp->lfp == lfp) {
1247 if (VM_FRAME_TYPE(cfp) == VM_FRAME_MAGIC_LAMBDA) {
1248 VALUE *tdfp = dfp;
1249
1250 while (lfp != tdfp) {
1251 if (cfp->dfp == tdfp) {
1252 /* in lambda */
1253 dfp = cfp->dfp;
1254 goto valid_return;
1255 }
1256 tdfp = GC_GUARDED_PTR_REF((VALUE *)*dfp);
1257 }
1258 }
1259 }
1260
1261 if (cfp->dfp == lfp && cfp->iseq->type == ISEQ_TYPE_METHOD) {
1262 dfp = lfp;
1263 goto valid_return;
1264 }
1265
1266 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
1267 }
1268
1269 vm_localjump_error("unexpected return", throwobj, TAG_RETURN);
1270
1271 valid_return:
1272 pt = dfp;
1273 }
1274 else {
1275 rb_bug("isns(throw): unsupport throw type");
1276 }
1277 }
1278 th->state = state;
1279 return (VALUE)NEW_THROW_OBJECT(throwobj, (VALUE) pt, state);
1280 }
1281 else {
1282 /* continue throw */
1283 VALUE err = throwobj;
1284
1285 if (FIXNUM_P(err)) {
1286 th->state = FIX2INT(err);
1287 }
1288 else if (SYMBOL_P(err)) {
1289 th->state = TAG_THROW;
1290 }
1291 else if (BUILTIN_TYPE(err) == T_NODE) {
1292 th->state = GET_THROWOBJ_STATE(err);
1293 }
1294 else {
1295 th->state = TAG_RAISE;
1296 /*th->state = FIX2INT(rb_ivar_get(err, idThrowState));*/
1297 }
1298 return err;
1299 }
1300 }
1301
1302 static inline void
1303 vm_expandarray(rb_control_frame_t *cfp, VALUE ary, int num, int flag)
1304 {
1305 int is_splat = flag & 0x01;
1306 int space_size = num + is_splat;
1307 VALUE *base = cfp->sp, *ptr;
1308 volatile VALUE tmp_ary;
1309 int len;
1310
1311 if (TYPE(ary) != T_ARRAY) {
1312 ary = rb_ary_to_ary(ary);
1313 }
1314
1315 cfp->sp += space_size;
1316
1317 tmp_ary = ary;
1318 ptr = RARRAY_PTR(ary);
1319 len = RARRAY_LEN(ary);
1320
1321 if (flag & 0x02) {
1322 /* post: ..., nil ,ary[-1], ..., ary[0..-num] # top */
1323 int i = 0, j;
1324
1325 if (len < num) {
1326 for (i=0; i<num-len; i++) {
1327 *base++ = Qnil;
1328 }
1329 }
1330 for (j=0; i<num; i++, j++) {
1331 VALUE v = ptr[len - j - 1];
1332 *base++ = v;
1333 }
1334 if (is_splat) {
1335 *base = rb_ary_new4(len - j, ptr);
1336 }
1337 }
1338 else {
1339 /* normal: ary[num..-1], ary[num-2], ary[num-3], ..., ary[0] # top */
1340 int i;
1341 VALUE *bptr = &base[space_size - 1];
1342
1343 for (i=0; i<num; i++) {
1344 if (len <= i) {
1345 for (; i<num; i++) {
1346 *bptr-- = Qnil;
1347 }
1348 break;
1349 }
1350 *bptr-- = ptr[i];
1351 }
1352 if (is_splat) {
1353 if (num > len) {
1354 *bptr = rb_ary_new();
1355 }
1356 else {
1357 *bptr = rb_ary_new4(len - num, ptr + num);
1358 }
1359 }
1360 }
1361 }
1362
1363 static inline int
1364 check_cfunc(const NODE *mn, const void *func)
1365 {
1366 if (mn && nd_type(mn->nd_body) == NODE_CFUNC &&
1367 mn->nd_body->nd_cfnc == func) {
1368 return 1;
1369 }
1370 else {
1371 return 0;
1372 }
1373 }
1374
1375 static
1376 #ifndef NO_BIG_INLINE
1377 inline
1378 #endif
1379 VALUE
1380 opt_eq_func(VALUE recv, VALUE obj, IC ic)
1381 {
1382 VALUE val = Qundef;
1383
1384 if (FIXNUM_2_P(recv, obj) &&
1385 BASIC_OP_UNREDEFINED_P(BOP_EQ)) {
1386 if (recv == obj) {
1387 val = Qtrue;
1388 }
1389 else {
1390 val = Qfalse;
1391 }
1392 }
1393 else if (!SPECIAL_CONST_P(recv) && !SPECIAL_CONST_P(obj)) {
1394 if (HEAP_CLASS_OF(recv) == rb_cFloat &&
1395 HEAP_CLASS_OF(obj) == rb_cFloat &&
1396 BASIC_OP_UNREDEFINED_P(BOP_EQ)) {
1397 double a = RFLOAT_VALUE(recv);
1398 double b = RFLOAT_VALUE(obj);
1399
1400 if (isnan(a) || isnan(b)) {
1401 val = Qfalse;
1402 }
1403 else if (a == b) {
1404 val = Qtrue;
1405 }
1406 else {
1407 val = Qfalse;
1408 }
1409 }
1410 else if (HEAP_CLASS_OF(recv) == rb_cString &&
1411 HEAP_CLASS_OF(obj) == rb_cString &&
1412 BASIC_OP_UNREDEFINED_P(BOP_EQ)) {
1413 val = rb_str_equal(recv, obj);
1414 }
1415 else {
1416 NODE *mn = vm_method_search(idEq, CLASS_OF(recv), ic);
1417 extern VALUE rb_obj_equal(VALUE obj1, VALUE obj2);
1418
1419 if (check_cfunc(mn, rb_obj_equal)) {
1420 return recv == obj ? Qtrue : Qfalse;
1421 }
1422 }
1423 }
1424
1425 return val;
1426 }
ruby svn mirror