Ruby 4.1.0dev (2026-08-15 revision d1c079751b80352d347452c8d134ffb177838adb)
vm.c (d1c079751b80352d347452c8d134ffb177838adb)
1/**********************************************************************
2
3 Vm.c -
4
5 $Author$
6
7 Copyright (C) 2004-2007 Koichi Sasada
8
9**********************************************************************/
10
11#define vm_exec rb_vm_exec
12
13#include "eval_intern.h"
14#include "internal.h"
15#include "internal/box.h"
16#include "internal/class.h"
17#include "internal/compile.h"
18#include "internal/cont.h"
19#include "internal/error.h"
20#include "internal/encoding.h"
21#include "internal/eval.h"
22#include "internal/gc.h"
23#include "internal/inits.h"
24#include "internal/missing.h"
25#include "internal/object.h"
26#include "internal/proc.h"
27#include "internal/ractor.h"
28#include "internal/re.h"
29#include "internal/ruby_parser.h"
30#include "internal/st.h"
31#include "internal/symbol.h"
32#include "internal/thread.h"
33#include "internal/transcode.h"
34#include "internal/vm.h"
35#include "internal/sanitizers.h"
36#include "internal/variable.h"
37#include "iseq.h"
38#include "symbol.h" // This includes a macro for a more performant rb_id2sym.
39#include "yjit.h"
40#include "insns.inc"
41#include "zjit.h"
42#include "ruby/st.h"
43#include "ruby/vm.h"
44#include "vm_core.h"
45#include "vm_callinfo.h"
46#include "vm_debug.h"
47#include "ruby/debug.h"
48#include "vm_exec.h"
49#include "vm_insnhelper.h"
50#include "ractor_core.h"
51#include "vm_sync.h"
52#include "shape.h"
53
54#include "builtin.h"
55
56#include "probes.h"
57#include "probes_helper.h"
58
59#ifdef RUBY_ASSERT_CRITICAL_SECTION
60int ruby_assert_critical_section_entered = 0;
61#endif
62
63static void *native_main_thread_stack_top;
64
65bool ruby_vm_during_cleanup = false;
66
67VALUE rb_str_concat_literals(size_t, const VALUE*);
68
70
71extern const char *const rb_debug_counter_names[];
72
73PUREFUNC(static inline const VALUE *VM_EP_LEP(const VALUE *));
74static inline const VALUE *
75VM_EP_LEP(const VALUE *ep)
76{
77 while (!VM_ENV_LOCAL_P(ep)) {
78 ep = VM_ENV_PREV_EP(ep);
79 }
80 return ep;
81}
82
83static inline const rb_control_frame_t *
84rb_vm_search_cf_from_ep(const rb_execution_context_t *ec, const rb_control_frame_t *cfp, const VALUE * const ep)
85{
86 if (!ep) {
87 return NULL;
88 }
89 else {
90 const rb_control_frame_t * const eocfp = RUBY_VM_END_CONTROL_FRAME(ec); /* end of control frame pointer */
91
92 while (cfp < eocfp) {
93 if (cfp->ep == ep) {
94 return cfp;
95 }
96 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
97 }
98
99 return NULL;
100 }
101}
102
103#if VM_CHECK_MODE > 0
104// ruby_box_crashed defined in internal/box.h
105#define VM_BOX_CRASHED() {ruby_box_crashed = true;}
106#define VM_BOX_ASSERT(expr, msg) \
107 if (!(expr)) { ruby_box_crashed = true; rb_bug(msg); }
108#else
109#define VM_BOX_CRASHED() {}
110#define VM_BOX_ASSERT(expr, msg) ((void)0)
111#endif
112
113static const VALUE *
114VM_EP_RUBY_LEP(const rb_execution_context_t *ec, const rb_control_frame_t *current_cfp)
115{
116 // rb_vmdebug_box_env_dump_raw() simulates this function
117 const VALUE *ep = current_cfp->ep;
118 const rb_control_frame_t * const eocfp = RUBY_VM_END_CONTROL_FRAME(ec); /* end of control frame pointer */
119 const rb_control_frame_t *cfp = current_cfp;
120
121 if (VM_ENV_FRAME_TYPE_P(ep, VM_FRAME_MAGIC_IFUNC)) {
122 ep = VM_EP_LEP(current_cfp->ep);
151 VM_ASSERT(VM_ENV_FRAME_TYPE_P(ep, VM_FRAME_MAGIC_CFUNC));
152 return ep;
153 }
154
155 while (VM_ENV_FRAME_TYPE_P(ep, VM_FRAME_MAGIC_CFUNC)) {
156 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
157
158 VM_BOX_ASSERT(cfp, "CFUNC should have a valid previous control frame");
159 VM_BOX_ASSERT(cfp < eocfp, "CFUNC should have a valid caller frame");
160 if (!cfp || cfp >= eocfp) {
161 return NULL;
162 }
163
164 VM_BOX_ASSERT(cfp->ep, "CFUNC should have a valid caller frame with env");
165 ep = cfp->ep;
166 if (!ep) {
167 return NULL;
168 }
169 }
170
171 while (!VM_ENV_LOCAL_P(ep)) {
172 ep = VM_ENV_PREV_EP(ep);
173 }
174
175 return ep;
176}
177
178const VALUE *
179rb_vm_ep_local_ep(const VALUE *ep)
180{
181 return VM_EP_LEP(ep);
182}
183
184PUREFUNC(static inline const VALUE *VM_CF_LEP(const rb_control_frame_t * const cfp));
185static inline const VALUE *
186VM_CF_LEP(const rb_control_frame_t * const cfp)
187{
188 return VM_EP_LEP(cfp->ep);
189}
190
191static inline const VALUE *
192VM_CF_PREV_EP(const rb_control_frame_t * const cfp)
193{
194 return VM_ENV_PREV_EP(cfp->ep);
195}
196
197PUREFUNC(static inline VALUE VM_CF_BLOCK_HANDLER(const rb_control_frame_t * const cfp));
198static inline VALUE
199VM_CF_BLOCK_HANDLER(const rb_control_frame_t * const cfp)
200{
201 const VALUE *ep;
202 if (VM_ENV_BOXED_P(cfp->ep)) {
203 VM_ASSERT(VM_ENV_LOCAL_P(cfp->ep));
204 /* Never set black_handler for VM_FRAME_MAGIC_TOP or VM_FRAME_MAGIC_CLASS
205 * and the specval is used for boxes (rb_box_t) in these case
206 */
207 return VM_BLOCK_HANDLER_NONE;
208 }
209 ep = VM_CF_LEP(cfp);
210 return VM_ENV_BLOCK_HANDLER(ep);
211}
212
213int
214rb_vm_cframe_keyword_p(const rb_control_frame_t *cfp)
215{
216 return VM_FRAME_CFRAME_KW_P(cfp);
217}
218
219VALUE
220rb_vm_frame_block_handler(const rb_control_frame_t *cfp)
221{
222 return VM_CF_BLOCK_HANDLER(cfp);
223}
224
225#if VM_CHECK_MODE > 0
226static int
227VM_CFP_IN_HEAP_P(const rb_execution_context_t *ec, const rb_control_frame_t *cfp)
228{
229 const VALUE *start = ec->vm_stack;
230 const VALUE *end = (VALUE *)ec->vm_stack + ec->vm_stack_size;
231 VM_ASSERT(start != NULL);
232
233 if (start <= (VALUE *)cfp && (VALUE *)cfp < end) {
234 return FALSE;
235 }
236 else {
237 return TRUE;
238 }
239}
240
241static int
242VM_EP_IN_HEAP_P(const rb_execution_context_t *ec, const VALUE *ep)
243{
244 const VALUE *start = ec->vm_stack;
245 const VALUE *end = (VALUE *)ec->cfp;
246 VM_ASSERT(start != NULL);
247
248 if (start <= ep && ep < end) {
249 return FALSE;
250 }
251 else {
252 return TRUE;
253 }
254}
255
256static int
257vm_ep_in_heap_p_(const rb_execution_context_t *ec, const VALUE *ep)
258{
259 if (VM_EP_IN_HEAP_P(ec, ep)) {
260 VALUE envval = ep[VM_ENV_DATA_INDEX_ENV]; /* VM_ENV_ENVVAL(ep); */
261
262 if (!UNDEF_P(envval)) {
263 const rb_env_t *env = (const rb_env_t *)envval;
264
265 VM_ASSERT(imemo_type_p(envval, imemo_env));
266 VM_ASSERT(VM_ENV_FLAGS(ep, VM_ENV_FLAG_ESCAPED));
267 VM_ASSERT(env->ep == ep);
268 }
269 return TRUE;
270 }
271 else {
272 return FALSE;
273 }
274}
275
276int
277rb_vm_ep_in_heap_p(const VALUE *ep)
278{
279 const rb_execution_context_t *ec = GET_EC();
280 if (ec->vm_stack == NULL) return TRUE;
281 return vm_ep_in_heap_p_(ec, ep);
282}
283#endif
284
285static struct rb_captured_block *
286VM_CFP_TO_CAPTURED_BLOCK(const rb_control_frame_t *cfp)
287{
288 VM_ASSERT(!VM_CFP_IN_HEAP_P(GET_EC(), cfp));
289 return (struct rb_captured_block *)&cfp->self;
290}
291
292static rb_control_frame_t *
293VM_CAPTURED_BLOCK_TO_CFP(const struct rb_captured_block *captured)
294{
295 rb_control_frame_t *cfp = ((rb_control_frame_t *)((VALUE *)(captured) - 3));
296 VM_ASSERT(!VM_CFP_IN_HEAP_P(GET_EC(), cfp));
297 VM_ASSERT(sizeof(rb_control_frame_t)/sizeof(VALUE) == 7 + VM_DEBUG_BP_CHECK ? 1 : 0);
298 return cfp;
299}
300
301static int
302VM_BH_FROM_CFP_P(VALUE block_handler, const rb_control_frame_t *cfp)
303{
304 const struct rb_captured_block *captured = VM_CFP_TO_CAPTURED_BLOCK(cfp);
305 return VM_TAGGED_PTR_REF(block_handler, 0x03) == captured;
306}
307
308static VALUE
309vm_passed_block_handler(rb_execution_context_t *ec)
310{
311 VALUE block_handler = ec->passed_block_handler;
312 ec->passed_block_handler = VM_BLOCK_HANDLER_NONE;
313 vm_block_handler_verify(block_handler);
314 return block_handler;
315}
316
317static rb_cref_t *
318vm_cref_new0(VALUE klass, rb_method_visibility_t visi, int module_func, rb_cref_t *prev_cref, int pushed_by_eval, int use_prev_prev, int singleton)
319{
320 VALUE refinements = Qnil;
321 int omod_shared = FALSE;
322
323 /* scope */
324 rb_scope_visibility_t scope_visi;
325 scope_visi.method_visi = visi;
326 scope_visi.module_func = module_func;
327
328 /* refinements */
329 if (prev_cref != NULL && prev_cref != (void *)1 /* TODO: why CREF_NEXT(cref) is 1? */) {
330 refinements = CREF_REFINEMENTS(prev_cref);
331
332 if (!NIL_P(refinements)) {
333 omod_shared = TRUE;
334 CREF_OMOD_SHARED_SET(prev_cref);
335 }
336 }
337
338 VM_ASSERT(singleton || klass);
339
340 rb_cref_t *cref = SHAREABLE_IMEMO_NEW(rb_cref_t, imemo_cref, refinements);
341 /* A cref is born shareable, so possibly-unshareable children (a singleton cref's
342 * self, `using`'s refinements hash) go through the write barrier to record a shref;
343 * a plain store would let the owner's local GC collect the child under the pinned
344 * cref. next is always a cref (shareable): plain store. */
345 if (!SPECIAL_CONST_P(refinements)) RB_OBJ_WRITTEN(cref, Qundef, refinements);
346 if (klass) {
347 RB_OBJ_WRITE(cref, &cref->klass_or_self, klass);
348 }
349 else {
350 cref->klass_or_self = 0;
351 }
352 cref->next = use_prev_prev ? CREF_NEXT(prev_cref) : prev_cref;
353 *((rb_scope_visibility_t *)&cref->scope_visi) = scope_visi;
354
355 if (pushed_by_eval) CREF_PUSHED_BY_EVAL_SET(cref);
356 if (omod_shared) CREF_OMOD_SHARED_SET(cref);
357 if (singleton) CREF_SINGLETON_SET(cref);
358
359 return cref;
360}
361
362static rb_cref_t *
363vm_cref_new(VALUE klass, rb_method_visibility_t visi, int module_func, rb_cref_t *prev_cref, int pushed_by_eval, int singleton)
364{
365 return vm_cref_new0(klass, visi, module_func, prev_cref, pushed_by_eval, FALSE, singleton);
366}
367
368static int
369ref_delete_symkey(VALUE key, VALUE value, VALUE unused)
370{
371 return SYMBOL_P(key) ? ST_DELETE : ST_CONTINUE;
372}
373
374rb_cref_t *
375rb_vm_cref_dup(const rb_cref_t *cref)
376{
377 const rb_scope_visibility_t *visi = CREF_SCOPE_VISI(cref);
378 rb_cref_t *next_cref = CREF_NEXT(cref), *new_cref;
379 int pushed_by_eval = CREF_PUSHED_BY_EVAL(cref);
380 int singleton = CREF_SINGLETON(cref);
381
382 new_cref = vm_cref_new(cref->klass_or_self, visi->method_visi, visi->module_func, next_cref, pushed_by_eval, singleton);
383
384 if (!NIL_P(CREF_REFINEMENTS(cref))) {
385 VALUE ref = rb_hash_dup(CREF_REFINEMENTS(cref));
386 rb_hash_foreach(ref, ref_delete_symkey, Qnil);
387 CREF_REFINEMENTS_SET(new_cref, ref);
388 CREF_OMOD_SHARED_UNSET(new_cref);
389 }
390
391 return new_cref;
392}
393
394rb_cref_t *
395rb_vm_cref_dup_without_refinements(const rb_cref_t *cref)
396{
397 const rb_scope_visibility_t *visi = CREF_SCOPE_VISI(cref);
398 rb_cref_t *next_cref = CREF_NEXT(cref), *new_cref;
399 int pushed_by_eval = CREF_PUSHED_BY_EVAL(cref);
400 int singleton = CREF_SINGLETON(cref);
401
402 new_cref = vm_cref_new(cref->klass_or_self, visi->method_visi, visi->module_func, next_cref, pushed_by_eval, singleton);
403
404 if (!NIL_P(CREF_REFINEMENTS(cref))) {
405 CREF_REFINEMENTS_SET(new_cref, Qnil);
406 CREF_OMOD_SHARED_UNSET(new_cref);
407 }
408
409 return new_cref;
410}
411
412static rb_cref_t *
413vm_cref_new_toplevel(rb_execution_context_t *ec)
414{
415 rb_cref_t *cref = vm_cref_new(rb_cObject, METHOD_VISI_PRIVATE /* toplevel visibility is private */, FALSE, NULL, FALSE, FALSE);
416 VALUE top_wrapper = rb_ec_thread_ptr(ec)->top_wrapper;
417
418 if (top_wrapper) {
419 cref = vm_cref_new(top_wrapper, METHOD_VISI_PRIVATE, FALSE, cref, FALSE, FALSE);
420 }
421
422 return cref;
423}
424
425rb_cref_t *
426rb_vm_cref_new_toplevel(void)
427{
428 return vm_cref_new_toplevel(GET_EC());
429}
430
431static void
432vm_cref_dump(const char *mesg, const rb_cref_t *cref)
433{
434 ruby_debug_printf("vm_cref_dump: %s (%p)\n", mesg, (void *)cref);
435
436 while (cref) {
437 VALUE path = rb_class_path(CREF_CLASS(cref));
438 ruby_debug_printf("= cref| klass: %.*s\n",
439 RSTRING_LENINT(path), RSTRING_PTR(path));
440 cref = CREF_NEXT(cref);
441 }
442}
443
444void
445rb_vm_block_ep_update(VALUE obj, const struct rb_block *dst, const VALUE *ep)
446{
447 *((const VALUE **)&dst->as.captured.ep) = ep;
448 RB_OBJ_WRITTEN(obj, Qundef, VM_ENV_ENVVAL(ep));
449}
450
451static void
452vm_bind_update_env(VALUE bindval, rb_binding_t *bind, VALUE envval)
453{
454 const rb_env_t *env = (rb_env_t *)envval;
455 RB_OBJ_WRITE(bindval, &bind->block.as.captured.code.iseq, env->iseq);
456 rb_vm_block_ep_update(bindval, &bind->block, env->ep);
457}
458
459#if VM_COLLECT_USAGE_DETAILS
460static void vm_collect_usage_operand(int insn, int n, VALUE op);
461static void vm_collect_usage_insn(int insn);
462static void vm_collect_usage_register(int reg, int isset);
463#endif
464
465static VALUE vm_make_env_object(const rb_execution_context_t *ec, rb_control_frame_t *cfp);
466static VALUE vm_invoke_bmethod(rb_execution_context_t *ec, rb_proc_t *proc, VALUE self,
467 int argc, const VALUE *argv, int kw_splat, VALUE block_handler,
469static VALUE vm_invoke_proc(rb_execution_context_t *ec, rb_proc_t *proc, VALUE self, int argc, const VALUE *argv, int kw_splat, VALUE block_handler, const rb_cref_t *cref);
470
471#if USE_YJIT
472// Counter to serve as a proxy for execution time, total number of calls
473static uint64_t yjit_total_entry_hits = 0;
474
475// Number of calls used to estimate how hot an ISEQ is
476#define YJIT_CALL_COUNT_INTERV 20u
477
479static inline bool
480rb_yjit_threshold_hit(const rb_iseq_t *iseq, uint64_t entry_calls)
481{
482 yjit_total_entry_hits += 1;
483
484 // Record the number of calls at the beginning of the interval
485 if (entry_calls + YJIT_CALL_COUNT_INTERV == rb_yjit_call_threshold) {
486 iseq->body->yjit_calls_at_interv = yjit_total_entry_hits;
487 }
488
489 // Try to estimate the total time taken (total number of calls) to reach 20 calls to this ISEQ
490 // This give us a ratio of how hot/cold this ISEQ is
491 if (entry_calls == rb_yjit_call_threshold) {
492 // We expect threshold 1 to compile everything immediately
493 if (rb_yjit_call_threshold < YJIT_CALL_COUNT_INTERV) {
494 return true;
495 }
496
497 uint64_t num_calls = yjit_total_entry_hits - iseq->body->yjit_calls_at_interv;
498
499 // Reject ISEQs that don't get called often enough
500 if (num_calls > rb_yjit_cold_threshold) {
501 rb_yjit_incr_counter("cold_iseq_entry");
502 return false;
503 }
504
505 return true;
506 }
507
508 return false;
509}
510#else
511#define rb_yjit_threshold_hit(iseq, entry_calls) false
512#endif
513
514#if USE_YJIT
515// Generate JIT code that supports the following kinds of ISEQ entries:
516// * The first ISEQ on vm_exec (e.g. <main>, or Ruby methods/blocks
517// called by a C method). The current frame has VM_FRAME_FLAG_FINISH.
518// The current vm_exec stops if JIT code returns a non-Qundef value.
519// * ISEQs called by the interpreter on vm_sendish (e.g. Ruby methods or
520// blocks called by a Ruby frame that isn't compiled or side-exited).
521// The current frame doesn't have VM_FRAME_FLAG_FINISH. The current
522// vm_exec does NOT stop whether JIT code returns Qundef or not.
523static inline rb_jit_func_t
524yjit_compile(rb_execution_context_t *ec)
525{
526 const rb_iseq_t *iseq = CFP_ISEQ(ec->cfp);
527 struct rb_iseq_constant_body *body = ISEQ_BODY(iseq);
528
529 // Increment the ISEQ's call counter and trigger JIT compilation if not compiled
530 if (body->jit_entry == NULL) {
531 body->jit_entry_calls++;
532 if (rb_yjit_threshold_hit(iseq, body->jit_entry_calls)) {
533 rb_yjit_compile_iseq(iseq, ec, false);
534 }
535 }
536 return body->jit_entry;
537}
538#else
539# define yjit_compile(ec) ((rb_jit_func_t)0)
540#endif
541
542#if USE_ZJIT
543static inline rb_jit_func_t
544zjit_compile(rb_execution_context_t *ec)
545{
546 const rb_iseq_t *iseq = CFP_ISEQ(ec->cfp);
547 struct rb_iseq_constant_body *body = ISEQ_BODY(iseq);
548
549 if (body->jit_entry == NULL) {
550 body->jit_entry_calls++;
551
552 // At profile-threshold, rewrite some of the YARV instructions
553 // to zjit_* instructions to profile these instructions.
554 if (body->jit_entry_calls == rb_zjit_profile_threshold) {
555 rb_zjit_profile_enable(iseq);
556 }
557
558 // At call-threshold, compile the ISEQ with ZJIT.
559 if (body->jit_entry_calls == rb_zjit_call_threshold) {
560 rb_zjit_compile_iseq(iseq, ec, false);
561 }
562 }
563 return body->jit_entry;
564}
565#else
566# define zjit_compile(ec) ((rb_jit_func_t)0)
567#endif
568
569#if USE_YJIT || USE_ZJIT
570// Execute JIT code compiled by yjit_compile() or zjit_compile()
571static inline VALUE
572jit_exec(rb_execution_context_t *ec)
573{
574#if USE_YJIT
575 if (rb_yjit_enabled_p) {
576 rb_jit_func_t func = yjit_compile(ec);
577 if (func) {
578 return func(ec, ec->cfp);
579 }
580 return Qundef;
581 }
582#endif
583
584#if USE_ZJIT
585 void *zjit_entry = rb_zjit_entry;
586 if (zjit_entry) {
587 rb_jit_func_t func = zjit_compile(ec);
588 if (func) {
589 VALUE result = ((rb_zjit_func_t)zjit_entry)(ec, ec->cfp, func);
590 return result;
591 }
592 }
593#endif
594 return Qundef;
595}
596
597// Generate JIT code that supports the following kind of ISEQ entry:
598// * The first ISEQ pushed by vm_exec_handle_exception. The frame would
599// point to a location specified by a catch table, and it doesn't have
600// VM_FRAME_FLAG_FINISH. The current vm_exec stops if JIT code returns
601// a non-Qundef value. So you should not return a non-Qundef value
602// until ec->cfp is changed to a frame with VM_FRAME_FLAG_FINISH.
603static inline rb_jit_func_t
604jit_compile_exception(rb_execution_context_t *ec)
605{
606 const rb_iseq_t *iseq = CFP_ISEQ(ec->cfp);
607 struct rb_iseq_constant_body *body = ISEQ_BODY(iseq);
608
609#if USE_ZJIT
610 if (body->jit_exception == NULL && rb_zjit_enabled_p) {
611 body->jit_exception_calls++;
612
613 // At profile-threshold, rewrite some of the YARV instructions
614 // to zjit_* instructions to profile these instructions.
615 if (body->jit_exception_calls == rb_zjit_profile_threshold) {
616 rb_zjit_profile_enable(iseq);
617 }
618
619 // At call-threshold, compile the ISEQ with ZJIT.
620 if (body->jit_exception_calls == rb_zjit_call_threshold) {
621 rb_zjit_compile_iseq(iseq, ec, true);
622 }
623 }
624#endif
625
626#if USE_YJIT
627 // Increment the ISEQ's call counter and trigger JIT compilation if not compiled
628 if (body->jit_exception == NULL && rb_yjit_enabled_p) {
629 body->jit_exception_calls++;
630 if (body->jit_exception_calls == rb_yjit_call_threshold) {
631 rb_yjit_compile_iseq(iseq, ec, true);
632 }
633 }
634#endif
635 return body->jit_exception;
636}
637
638// Execute JIT code compiled by jit_compile_exception()
639static inline VALUE
640jit_exec_exception(rb_execution_context_t *ec)
641{
642 rb_jit_func_t func = jit_compile_exception(ec);
643 if (func) {
644 // Call the JIT code
645 return func(ec, ec->cfp);
646 }
647 else {
648 return Qundef;
649 }
650}
651#else
652# define jit_compile_exception(ec) ((rb_jit_func_t)0)
653# define jit_exec(ec) Qundef
654# define jit_exec_exception(ec) Qundef
655#endif
656
657static void add_opt_method_entry(const rb_method_entry_t *me);
658
659#define RB_TYPE_2_P(obj, type1, type2) \
660 (RB_TYPE_P(obj, type1) || RB_TYPE_P(obj, type2))
661#define RB_TYPE_3_P(obj, type1, type2, type3) \
662 (RB_TYPE_P(obj, type1) || RB_TYPE_P(obj, type2) || RB_TYPE_P(obj, type3))
663
664#define VM_ASSERT_TYPE(obj, type) \
665 VM_ASSERT(RB_TYPE_P(obj, type), #obj ": %s", rb_obj_info(obj))
666#define VM_ASSERT_TYPE2(obj, type1, type2) \
667 VM_ASSERT(RB_TYPE_2_P(obj, type1, type2), #obj ": %s", rb_obj_info(obj))
668#define VM_ASSERT_TYPE3(obj, type1, type2, type3) \
669 VM_ASSERT(RB_TYPE_3_P(obj, type1, type2, type3), #obj ": %s", rb_obj_info(obj))
670
671#include "vm_insnhelper.c"
672
673#include "vm_exec.c"
674
675#include "vm_method.c"
676#include "vm_eval.c"
677
678#define PROCDEBUG 0
679
680VALUE rb_cRubyVM;
682VALUE rb_mRubyVMFrozenCore;
683VALUE rb_block_param_proxy;
684
685VALUE ruby_vm_const_missing_count = 0;
686rb_vm_t *ruby_current_vm_ptr = NULL;
687rb_ractor_t *ruby_single_main_ractor;
688bool ruby_vm_keep_script_lines;
689
690#ifdef RB_THREAD_LOCAL_SPECIFIER
691RB_THREAD_LOCAL_SPECIFIER rb_execution_context_t *ruby_current_ec;
692
693#ifdef RUBY_NT_SERIAL
694RB_THREAD_LOCAL_SPECIFIER rb_atomic_t ruby_nt_serial;
695#endif
696
697// no-inline decl on vm_core.h
699rb_current_ec_noinline(void)
700{
701 return ruby_current_ec;
702}
703
704void
705rb_current_ec_set(rb_execution_context_t *ec)
706{
707 ruby_current_ec = ec;
708}
709
710
711#ifdef RB_THREAD_CURRENT_EC_NOINLINE
713rb_current_ec(void)
714{
715 return ruby_current_ec;
716}
717
718#endif
719#else
720native_tls_key_t ruby_current_ec_key;
721
722// no-inline decl on vm_core.h
724rb_current_ec_noinline(void)
725{
726 return native_tls_get(ruby_current_ec_key);
727}
728
729#endif
730
731rb_event_flag_t ruby_vm_event_flags = 0;
732rb_event_flag_t ruby_vm_event_enabled_global_flags = 0;
733unsigned int ruby_vm_c_events_enabled = 0;
734unsigned int ruby_vm_iseq_events_enabled = 0;
735
736rb_serial_t ruby_vm_constant_cache_invalidations = 0;
737rb_serial_t ruby_vm_constant_cache_misses = 0;
738rb_serial_t ruby_vm_global_cvar_state = 1;
739
740static const struct rb_callcache vm_empty_cc = {
741 .flags = T_IMEMO | (imemo_callcache << FL_USHIFT) | VM_CALLCACHE_UNMARKABLE,
742 .klass = Qundef,
743 .cme_ = NULL,
744 .call_ = vm_call_general,
745 .aux_ = {
746 .v = Qfalse,
747 }
748};
749
750static const struct rb_callcache vm_empty_cc_for_super = {
751 .flags = T_IMEMO | (imemo_callcache << FL_USHIFT) | VM_CALLCACHE_UNMARKABLE,
752 .klass = Qundef,
753 .cme_ = NULL,
754 .call_ = vm_call_super_method,
755 .aux_ = {
756 .v = Qfalse,
757 }
758};
759
760static void thread_free(void *ptr);
761
762void
763rb_vm_inc_const_missing_count(void)
764{
765 ruby_vm_const_missing_count +=1;
766}
767
768int
769rb_dtrace_setup(rb_execution_context_t *ec, VALUE klass, ID id,
770 struct ruby_dtrace_method_hook_args *args)
771{
773 if (!klass) {
774 if (!ec) ec = GET_EC();
775 if (!rb_ec_frame_method_id_and_class(ec, &id, 0, &klass) || !klass)
776 return FALSE;
777 }
778 if (RB_TYPE_P(klass, T_ICLASS)) {
779 klass = RBASIC(klass)->klass;
780 }
781 else if (RCLASS_SINGLETON_P(klass)) {
782 klass = RCLASS_ATTACHED_OBJECT(klass);
783 if (NIL_P(klass)) return FALSE;
784 }
785 type = BUILTIN_TYPE(klass);
786 if (type == T_CLASS || type == T_ICLASS || type == T_MODULE) {
787 VALUE name = rb_class_path(klass);
788 const char *classname, *filename;
789 const char *methodname = rb_id2name(id);
790 if (methodname && (filename = rb_source_location_cstr(&args->line_no)) != 0) {
791 if (NIL_P(name) || !(classname = StringValuePtr(name)))
792 classname = "<unknown>";
793 args->classname = classname;
794 args->methodname = methodname;
795 args->filename = filename;
796 args->klass = klass;
797 args->name = name;
798 return TRUE;
799 }
800 }
801 return FALSE;
802}
803
804extern unsigned int redblack_buffer_size;
805
806/*
807 * call-seq:
808 * RubyVM.stat -> Hash
809 * RubyVM.stat(hsh) -> hsh
810 * RubyVM.stat(Symbol) -> Numeric
811 *
812 * Returns a Hash containing implementation-dependent counters inside the VM.
813 *
814 * This hash includes information about method/constant caches:
815 *
816 * {
817 * :constant_cache_invalidations=>2,
818 * :constant_cache_misses=>14,
819 * :global_cvar_state=>27
820 * }
821 *
822 * If <tt>USE_DEBUG_COUNTER</tt> is enabled, debug counters will be included.
823 *
824 * The contents of the hash are implementation specific and may be changed in
825 * the future.
826 *
827 * This method is only expected to work on C Ruby.
828 */
829static VALUE
830vm_stat(int argc, VALUE *argv, VALUE self)
831{
832 static VALUE sym_constant_cache_invalidations, sym_constant_cache_misses, sym_global_cvar_state, sym_next_shape_id;
833 static VALUE sym_shape_cache_size;
834 VALUE arg = Qnil;
835 VALUE hash = Qnil, key = Qnil;
836
837 if (rb_check_arity(argc, 0, 1) == 1) {
838 arg = argv[0];
839 if (SYMBOL_P(arg))
840 key = arg;
841 else if (RB_TYPE_P(arg, T_HASH))
842 hash = arg;
843 else
844 rb_raise(rb_eTypeError, "non-hash or symbol given");
845 }
846 else {
847 hash = rb_hash_new();
848 }
849
850#define S(s) sym_##s = ID2SYM(rb_intern_const(#s))
851 S(constant_cache_invalidations);
852 S(constant_cache_misses);
853 S(global_cvar_state);
854 S(next_shape_id);
855 S(shape_cache_size);
856#undef S
857
858#define SET(name, attr) \
859 if (key == sym_##name) \
860 return SERIALT2NUM(attr); \
861 else if (hash != Qnil) \
862 rb_hash_aset(hash, sym_##name, SERIALT2NUM(attr));
863
864 SET(constant_cache_invalidations, ruby_vm_constant_cache_invalidations);
865 SET(constant_cache_misses, ruby_vm_constant_cache_misses);
866 SET(global_cvar_state, ruby_vm_global_cvar_state);
867 SET(next_shape_id, (rb_serial_t)rb_shapes_count());
868 SET(shape_cache_size, (rb_serial_t)rb_shapes_cache_size());
869#undef SET
870
871#if USE_DEBUG_COUNTER
872 ruby_debug_counter_show_at_exit(FALSE);
873 for (size_t i = 0; i < RB_DEBUG_COUNTER_MAX; i++) {
874 const VALUE name = rb_sym_intern_ascii_cstr(rb_debug_counter_names[i]);
875 const VALUE boxed_value = SIZET2NUM(rb_debug_counter[i]);
876
877 if (key == name) {
878 return boxed_value;
879 }
880 else if (hash != Qnil) {
881 rb_hash_aset(hash, name, boxed_value);
882 }
883 }
884#endif
885
886 if (!NIL_P(key)) { /* matched key should return above */
887 rb_raise(rb_eArgError, "unknown key: %"PRIsVALUE, rb_sym2str(key));
888 }
889
890 return hash;
891}
892
893/* control stack frame */
894
895static void
896vm_set_top_stack(rb_execution_context_t *ec, const rb_iseq_t *iseq, const rb_box_t *box)
897{
898 if (ISEQ_BODY(iseq)->type != ISEQ_TYPE_TOP) {
899 rb_raise(rb_eTypeError, "Not a toplevel InstructionSequence");
900 }
901
902 /* for return */
903 vm_push_frame(ec, iseq, VM_FRAME_MAGIC_TOP | VM_ENV_FLAG_LOCAL | VM_FRAME_FLAG_FINISH,
904 rb_ec_thread_ptr(ec)->top_self,
905 GC_GUARDED_PTR(box),
906 (VALUE)vm_cref_new_toplevel(ec), /* cref or me */
907 ISEQ_BODY(iseq)->iseq_encoded, ec->cfp->sp,
908 ISEQ_BODY(iseq)->local_table_size, ISEQ_BODY(iseq)->stack_max);
909}
910
911static void
912vm_set_eval_stack(rb_execution_context_t *ec, const rb_iseq_t *iseq, const rb_cref_t *cref, const struct rb_block *base_block)
913{
914 vm_push_frame(ec, iseq, VM_FRAME_MAGIC_EVAL | VM_FRAME_FLAG_FINISH,
915 vm_block_self(base_block), VM_GUARDED_PREV_EP(vm_block_ep(base_block)),
916 (VALUE)cref, /* cref or me */
917 ISEQ_BODY(iseq)->iseq_encoded,
918 ec->cfp->sp, ISEQ_BODY(iseq)->local_table_size,
919 ISEQ_BODY(iseq)->stack_max);
920}
921
922static void
923vm_set_main_stack(rb_execution_context_t *ec, const rb_iseq_t *iseq)
924{
925 VALUE toplevel_binding = rb_const_get(rb_cObject, rb_intern("TOPLEVEL_BINDING"));
926 rb_binding_t *bind;
927
928 GetBindingPtr(toplevel_binding, bind);
929 RUBY_ASSERT_MESG(bind, "TOPLEVEL_BINDING is not built");
930
931 vm_set_eval_stack(ec, iseq, 0, &bind->block);
932
933 /* save binding */
934 if (ISEQ_BODY(iseq)->local_table_size > 0) {
935 vm_bind_update_env(toplevel_binding, bind, vm_make_env_object(ec, ec->cfp));
936 }
937}
938
940rb_vm_get_binding_creatable_next_cfp(const rb_execution_context_t *ec, const rb_control_frame_t *cfp)
941{
942 while (!RUBY_VM_CONTROL_FRAME_STACK_OVERFLOW_P(ec, cfp)) {
943 if (CFP_ISEQ(cfp)) {
944 return (rb_control_frame_t *)cfp;
945 }
946 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
947 }
948 return 0;
949}
950
952rb_vm_get_ruby_level_next_cfp(const rb_execution_context_t *ec, const rb_control_frame_t *cfp)
953{
954 while (!RUBY_VM_CONTROL_FRAME_STACK_OVERFLOW_P(ec, cfp)) {
955 if (VM_FRAME_RUBYFRAME_P(cfp)) {
956 return (rb_control_frame_t *)cfp;
957 }
958 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
959 }
960 return 0;
961}
962
963static rb_control_frame_t *
964vm_get_ruby_level_caller_cfp(const rb_execution_context_t *ec, const rb_control_frame_t *cfp)
965{
966 if (VM_FRAME_RUBYFRAME_P(cfp)) {
967 return (rb_control_frame_t *)cfp;
968 }
969
970 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
971
972 while (!RUBY_VM_CONTROL_FRAME_STACK_OVERFLOW_P(ec, cfp)) {
973 if (VM_FRAME_RUBYFRAME_P(cfp)) {
974 return (rb_control_frame_t *)cfp;
975 }
976
977 if (VM_ENV_FLAGS(cfp->ep, VM_FRAME_FLAG_PASSED) == FALSE) {
978 break;
979 }
980 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
981 }
982 return 0;
983}
984
985void
986rb_vm_pop_cfunc_frame(void)
987{
988 rb_execution_context_t *ec = GET_EC();
989 rb_control_frame_t *cfp = ec->cfp;
990 const rb_callable_method_entry_t *me = rb_vm_frame_method_entry(cfp);
991
992 EXEC_EVENT_HOOK(ec, RUBY_EVENT_C_RETURN, cfp->self, me->def->original_id, me->called_id, me->owner, Qnil);
993 RUBY_DTRACE_CMETHOD_RETURN_HOOK(ec, me->owner, me->def->original_id);
994 vm_pop_frame(ec, cfp, cfp->ep);
995}
996
997void
998rb_vm_rewind_cfp(rb_execution_context_t *ec, rb_control_frame_t *cfp)
999{
1000 /* check skipped frame */
1001 while (ec->cfp != cfp) {
1002#if VMDEBUG
1003 printf("skipped frame: %s\n", vm_frametype_name(ec->cfp));
1004#endif
1005 if (VM_FRAME_TYPE(ec->cfp) != VM_FRAME_MAGIC_CFUNC) {
1006 rb_vm_pop_frame(ec);
1007 }
1008 else { /* unlikely path */
1009 rb_vm_pop_cfunc_frame();
1010 }
1011 }
1012}
1013
1014/* at exit */
1015
1016void
1017ruby_vm_at_exit(void (*func)(rb_vm_t *))
1018{
1019 rb_vm_t *vm = GET_VM();
1021 nl->func = func;
1022 nl->next = vm->at_exit;
1023 vm->at_exit = nl;
1024}
1025
1026static void
1027ruby_vm_run_at_exit_hooks(rb_vm_t *vm)
1028{
1029 rb_at_exit_list *l = vm->at_exit;
1030
1031 while (l) {
1032 rb_at_exit_list* t = l->next;
1033 rb_vm_at_exit_func *func = l->func;
1034 ruby_xfree(l);
1035 l = t;
1036 (*func)(vm);
1037 }
1038}
1039
1040/* Env */
1041
1042static VALUE check_env_value(const rb_env_t *env);
1043
1044static int
1045check_env(const rb_env_t *env)
1046{
1047 fputs("---\n", stderr);
1048 ruby_debug_printf("envptr: %p\n", (void *)&env->ep[0]);
1049 ruby_debug_printf("envval: %10p ", (void *)env->ep[1]);
1050 dp(env->ep[1]);
1051 ruby_debug_printf("ep: %10p\n", (void *)env->ep);
1052 if (rb_vm_env_prev_env(env)) {
1053 fputs(">>\n", stderr);
1054 check_env_value(rb_vm_env_prev_env(env));
1055 fputs("<<\n", stderr);
1056 }
1057 return 1;
1058}
1059
1060static VALUE
1061check_env_value(const rb_env_t *env)
1062{
1063 if (check_env(env)) {
1064 return (VALUE)env;
1065 }
1066 rb_bug("invalid env");
1067 return Qnil; /* unreachable */
1068}
1069
1070static VALUE
1071vm_block_handler_escape(const rb_execution_context_t *ec, VALUE block_handler)
1072{
1073 switch (vm_block_handler_type(block_handler)) {
1074 case block_handler_type_ifunc:
1075 case block_handler_type_iseq:
1076 return rb_vm_make_proc(ec, VM_BH_TO_CAPT_BLOCK(block_handler), rb_cProc);
1077
1078 case block_handler_type_symbol:
1079 case block_handler_type_proc:
1080 return block_handler;
1081 }
1082 VM_UNREACHABLE(vm_block_handler_escape);
1083 return Qnil;
1084}
1085
1086static VALUE
1087vm_make_env_each(const rb_execution_context_t * const ec, rb_control_frame_t *const cfp)
1088{
1089 const VALUE * const ep = cfp->ep;
1090 VALUE *env_body, *env_ep;
1091 int local_size, env_size;
1092
1093 if (VM_ENV_ESCAPED_P(ep)) {
1094 return VM_ENV_ENVVAL(ep);
1095 }
1096
1097 if (!VM_ENV_LOCAL_P(ep)) {
1098 const VALUE *prev_ep = VM_ENV_PREV_EP(ep);
1099 if (!VM_ENV_ESCAPED_P(prev_ep)) {
1100 rb_control_frame_t *prev_cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
1101
1102 while (prev_cfp->ep != prev_ep) {
1103 prev_cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(prev_cfp);
1104 VM_ASSERT(prev_cfp->ep != NULL);
1105 }
1106
1107 vm_make_env_each(ec, prev_cfp);
1108 VM_FORCE_WRITE_SPECIAL_CONST(&ep[VM_ENV_DATA_INDEX_SPECVAL], VM_GUARDED_PREV_EP(prev_cfp->ep));
1109 }
1110 }
1111 else {
1112 VM_ASSERT(VM_ENV_LOCAL_P(ep));
1113 VALUE block_handler = VM_ENV_BLOCK_HANDLER(ep);
1114
1115 if (block_handler != VM_BLOCK_HANDLER_NONE) {
1116 VALUE blockprocval = vm_block_handler_escape(ec, block_handler);
1117 VM_STACK_ENV_WRITE(ep, VM_ENV_DATA_INDEX_SPECVAL, blockprocval);
1118 }
1119 }
1120
1121 const rb_iseq_t *iseq = CFP_ISEQ(cfp);
1122 if (!VM_FRAME_RUBYFRAME_P(cfp)) {
1123 local_size = VM_ENV_DATA_SIZE;
1124 }
1125 else {
1126 local_size = ISEQ_BODY(iseq)->local_table_size;
1127 if (ISEQ_BODY(iseq)->param.flags.forwardable && VM_ENV_LOCAL_P(cfp->ep)) {
1128 int ci_offset = local_size - ISEQ_BODY(iseq)->param.size + VM_ENV_DATA_SIZE;
1129
1130 CALL_INFO ci = (CALL_INFO)VM_CF_LEP(cfp)[-ci_offset];
1131 local_size += vm_ci_argc(ci);
1132 }
1133 local_size += VM_ENV_DATA_SIZE;
1134 }
1135
1136 // Invalidate JIT code that assumes cfp->ep == vm_base_ptr(cfp).
1137 // This is done before creating the imemo_env because VM_STACK_ENV_WRITE
1138 // below leaves the on-stack ep in a state that is unsafe to GC.
1139 // Once the enabled JIT has recorded this iseq's escape, the invalidations
1140 // are no longer useful and can slow down Ractors.
1141 if (VM_FRAME_RUBYFRAME_P(cfp) &&
1142 !rbimpl_atomic_load(&ISEQ_BODY(iseq)->jit_ep_escape_recorded, RBIMPL_ATOMIC_RELAXED)) {
1143 rb_yjit_invalidate_ep_is_bp(iseq);
1144 rb_zjit_invalidate_no_ep_escape(iseq);
1145 }
1146
1147 /*
1148 * # local variables on a stack frame (N == local_size)
1149 * [lvar1, lvar2, ..., lvarN, SPECVAL]
1150 * ^
1151 * ep[0]
1152 *
1153 * # moved local variables
1154 * [lvar1, lvar2, ..., lvarN, SPECVAL, Envval, BlockProcval (if needed)]
1155 * ^ ^
1156 * env->env[0] ep[0]
1157 */
1158
1159 env_size = local_size +
1160 1 /* envval */;
1161
1162 // Careful with order in the following sequence. Each allocation can move objects.
1163 env_body = ALLOC_N(VALUE, env_size);
1164 rb_env_t *env = IMEMO_NEW(rb_env_t, imemo_env, 0);
1165
1166 // Set up env without WB since it's brand new (similar to newobj_init(), newobj_fill())
1167 MEMCPY(env_body, ep - (local_size - 1 /* specval */), VALUE, local_size);
1168
1169 env_ep = &env_body[local_size - 1 /* specval */];
1170 env_ep[VM_ENV_DATA_INDEX_ENV] = (VALUE)env;
1171
1172 env->iseq = (rb_iseq_t *)(VM_FRAME_RUBYFRAME_P(cfp) ? iseq : NULL);
1173 env->ep = env_ep;
1174 env->env = env_body;
1175 env->env_size = env_size;
1176
1177 cfp->ep = env_ep;
1178 VM_ENV_FLAGS_SET(env_ep, VM_ENV_FLAG_ESCAPED | VM_ENV_FLAG_WB_REQUIRED);
1179 VM_STACK_ENV_WRITE(ep, 0, (VALUE)env); /* GC mark */
1180
1181#if 0
1182 for (i = 0; i < local_size; i++) {
1183 if (VM_FRAME_RUBYFRAME_P(cfp)) {
1184 /* clear value stack for GC */
1185 ep[-local_size + i] = 0;
1186 }
1187 }
1188#endif
1189
1190 return (VALUE)env;
1191}
1192
1193static VALUE
1194vm_make_env_object(const rb_execution_context_t *ec, rb_control_frame_t *cfp)
1195{
1196 VALUE envval = vm_make_env_each(ec, cfp);
1197
1198 if (PROCDEBUG) {
1199 check_env_value((const rb_env_t *)envval);
1200 }
1201
1202 return envval;
1203}
1204
1205void
1206rb_vm_stack_to_heap(rb_execution_context_t *ec)
1207{
1208 rb_control_frame_t *cfp = ec->cfp;
1209 while ((cfp = rb_vm_get_binding_creatable_next_cfp(ec, cfp)) != 0) {
1210 vm_make_env_object(ec, cfp);
1211 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
1212 }
1213}
1214
1215const rb_env_t *
1216rb_vm_env_prev_env(const rb_env_t *env)
1217{
1218 const VALUE *ep = env->ep;
1219
1220 if (VM_ENV_LOCAL_P(ep)) {
1221 return NULL;
1222 }
1223 else {
1224 const VALUE *prev_ep = VM_ENV_PREV_EP(ep);
1225 return VM_ENV_ENVVAL_PTR(prev_ep);
1226 }
1227}
1228
1229static int
1230collect_local_variables_in_iseq(const rb_iseq_t *iseq, const struct local_var_list *vars)
1231{
1232 unsigned int i;
1233 if (!iseq) return 0;
1234 for (i = 0; i < ISEQ_BODY(iseq)->local_table_size; i++) {
1235 local_var_list_add(vars, ISEQ_BODY(iseq)->local_table[i]);
1236 }
1237 return 1;
1238}
1239
1240static void
1241collect_local_variables_in_env(const rb_env_t *env, const struct local_var_list *vars)
1242{
1243 do {
1244 if (VM_ENV_FLAGS(env->ep, VM_ENV_FLAG_ISOLATED)) break;
1245 collect_local_variables_in_iseq(env->iseq, vars);
1246 } while ((env = rb_vm_env_prev_env(env)) != NULL);
1247}
1248
1249static int
1250vm_collect_local_variables_in_heap(const VALUE *ep, const struct local_var_list *vars)
1251{
1252 if (VM_ENV_ESCAPED_P(ep)) {
1253 collect_local_variables_in_env(VM_ENV_ENVVAL_PTR(ep), vars);
1254 return 1;
1255 }
1256 else {
1257 return 0;
1258 }
1259}
1260
1261VALUE
1262rb_vm_env_local_variables(const rb_env_t *env)
1263{
1264 struct local_var_list vars;
1265 local_var_list_init(&vars);
1266 collect_local_variables_in_env(env, &vars);
1267 return local_var_list_finish(&vars);
1268}
1269
1270VALUE
1271rb_vm_env_numbered_parameters(const rb_env_t *env)
1272{
1273 struct local_var_list vars;
1274 local_var_list_init(&vars);
1275 // if (VM_ENV_FLAGS(env->ep, VM_ENV_FLAG_ISOLATED)) break; // TODO: is this needed?
1276 const rb_iseq_t *iseq = env->iseq;
1277 unsigned int i;
1278 if (!iseq) return 0;
1279 for (i = 0; i < ISEQ_BODY(iseq)->local_table_size; i++) {
1280 numparam_list_add(&vars, ISEQ_BODY(iseq)->local_table[i]);
1281 }
1282 return local_var_list_finish(&vars);
1283}
1284
1285VALUE
1286rb_iseq_local_variables(const rb_iseq_t *iseq)
1287{
1288 struct local_var_list vars;
1289 local_var_list_init(&vars);
1290 while (collect_local_variables_in_iseq(iseq, &vars)) {
1291 iseq = ISEQ_BODY(iseq)->parent_iseq;
1292 }
1293 return local_var_list_finish(&vars);
1294}
1295
1296/* Proc */
1297
1298static VALUE
1299vm_proc_create_from_captured(VALUE klass,
1300 const struct rb_captured_block *captured,
1301 enum rb_block_type block_type,
1302 int8_t is_from_method, int8_t is_lambda)
1303{
1304 VALUE procval = rb_proc_alloc(klass);
1305 rb_proc_t *proc = RTYPEDDATA_DATA(procval);
1306
1307 VM_ASSERT(VM_EP_IN_HEAP_P(GET_EC(), captured->ep));
1308
1309 /* copy block */
1310 RB_OBJ_WRITE(procval, &proc->block.as.captured.code.val, captured->code.val);
1311 RB_OBJ_WRITE(procval, &proc->block.as.captured.self, captured->self);
1312 rb_vm_block_ep_update(procval, &proc->block, captured->ep);
1313
1314 vm_block_type_set(&proc->block, block_type);
1315 proc->is_from_method = is_from_method;
1316 proc->is_lambda = is_lambda;
1317
1318 return procval;
1319}
1320
1321void
1322rb_vm_block_copy(VALUE obj, const struct rb_block *dst, const struct rb_block *src)
1323{
1324 /* copy block */
1325 switch (vm_block_type(src)) {
1326 case block_type_iseq:
1327 case block_type_ifunc:
1328 RB_OBJ_WRITE(obj, &dst->as.captured.self, src->as.captured.self);
1329 RB_OBJ_WRITE(obj, &dst->as.captured.code.val, src->as.captured.code.val);
1330 rb_vm_block_ep_update(obj, dst, src->as.captured.ep);
1331 break;
1332 case block_type_symbol:
1333 RB_OBJ_WRITE(obj, &dst->as.symbol, src->as.symbol);
1334 break;
1335 case block_type_proc:
1336 RB_OBJ_WRITE(obj, &dst->as.proc, src->as.proc);
1337 break;
1338 }
1339}
1340
1341static VALUE
1342proc_create(VALUE klass, const struct rb_block *block, int8_t is_from_method, int8_t is_lambda)
1343{
1344 VALUE procval = rb_proc_alloc(klass);
1345 rb_proc_t *proc = RTYPEDDATA_DATA(procval);
1346
1347 VM_ASSERT(VM_EP_IN_HEAP_P(GET_EC(), vm_block_ep(block)));
1348 rb_vm_block_copy(procval, &proc->block, block);
1349 vm_block_type_set(&proc->block, block->type);
1350 proc->is_from_method = is_from_method;
1351 proc->is_lambda = is_lambda;
1352
1353 return procval;
1354}
1355
1356VALUE
1357rb_proc_dup_0(VALUE self)
1358{
1359 VALUE procval;
1360 rb_proc_t *src;
1361
1362 GetProcPtr(self, src);
1363
1364 switch (vm_block_type(&src->block)) {
1365 case block_type_ifunc:
1366 procval = rb_func_proc_dup(self);
1367 break;
1368 default:
1369 procval = proc_create(rb_obj_class(self), &src->block, src->is_from_method, src->is_lambda);
1370 break;
1371 }
1372
1373 if (src->is_refined) {
1374 rb_proc_t *dst;
1375 GetProcPtr(procval, dst);
1376 dst->is_refined = 1;
1377 }
1378
1379 if (RB_OBJ_SHAREABLE_P(self)) RB_OBJ_SET_SHAREABLE(procval);
1380 RB_GC_GUARD(self); /* for: body = rb_proc_dup(body) */
1381 return procval;
1382}
1383
1384VALUE
1385rb_proc_dup(VALUE self)
1386{
1387 VALUE procval = rb_proc_dup_0(self);
1388 VALUE recipe = rb_proc_refinements_recipe(self);
1389 if (!NIL_P(recipe)) rb_proc_set_refinements_recipe(procval, recipe);
1390 return procval;
1391}
1392
1393/* Proc#refined: build a Proc that runs `iseq` with the refinements of
1394 * `recipe`, sharing self's environment. `iseq` is normally self's own block
1395 * iseq, which the copy replaces on the first call. */
1396VALUE
1397rb_proc_dup_with_iseq_and_recipe(VALUE self, const rb_iseq_t *iseq, VALUE recipe)
1398{
1399 rb_proc_t *src;
1400 GetProcPtr(self, src);
1401 VM_ASSERT(vm_block_type(&src->block) == block_type_iseq);
1402
1403 struct rb_block block = src->block;
1404 block.as.captured.code.iseq = iseq;
1405
1406 VALUE procval = proc_create(rb_obj_class(self), &block, src->is_from_method, src->is_lambda);
1407 rb_proc_set_refinements_recipe(procval, recipe);
1408
1409 RB_GC_GUARD(self);
1410 return procval;
1411}
1412
1414 VALUE ary;
1415 VALUE read_only;
1416 bool yield;
1417 bool isolate;
1418};
1419
1420static VALUE
1421ID2NUM(ID id)
1422{
1423 if (SIZEOF_VOIDP > SIZEOF_LONG)
1424 return ULL2NUM(id);
1425 else
1426 return ULONG2NUM(id);
1427}
1428
1429static ID
1430NUM2ID(VALUE num)
1431{
1432 if (SIZEOF_VOIDP > SIZEOF_LONG)
1433 return (ID)NUM2ULL(num);
1434 else
1435 return (ID)NUM2ULONG(num);
1436}
1437
1438static enum rb_id_table_iterator_result
1439collect_outer_variable_names(ID id, VALUE val, void *ptr)
1440{
1442
1443 if (id == rb_intern("yield")) {
1444 data->yield = true;
1445 }
1446 else {
1447 VALUE *store;
1448 if (data->isolate ||
1449 val == Qtrue /* write */) {
1450 store = &data->ary;
1451 }
1452 else {
1453 store = &data->read_only;
1454 }
1455 if (*store == Qfalse) *store = rb_ary_new();
1456 rb_ary_push(*store, ID2NUM(id));
1457 }
1458 return ID_TABLE_CONTINUE;
1459}
1460
1461static const rb_env_t *
1462env_copy(const VALUE *src_ep, VALUE read_only_variables)
1463{
1464 const rb_env_t *src_env = (rb_env_t *)VM_ENV_ENVVAL(src_ep);
1465 VM_ASSERT(src_env->ep == src_ep);
1466
1467 VALUE *env_body = ZALLOC_N(VALUE, src_env->env_size); // fill with Qfalse
1468 VALUE *ep = &env_body[src_env->env_size - 2];
1469 const rb_env_t *copied_env = vm_env_new(ep, env_body, src_env->env_size, src_env->iseq);
1470
1471 // Copy after allocations above, since they can move objects in src_ep.
1472 VALUE svar_val = src_ep[VM_ENV_DATA_INDEX_ME_CREF];
1473 if (imemo_type_p(svar_val, imemo_svar)) {
1474 const struct vm_svar *svar = (struct vm_svar *)svar_val;
1475
1476 if (svar->cref_or_me) {
1477 svar_val = svar->cref_or_me;
1478 }
1479 else {
1480 svar_val = Qfalse;
1481 }
1482 }
1483 RB_OBJ_WRITE(copied_env, &ep[VM_ENV_DATA_INDEX_ME_CREF], svar_val);
1484
1485 ep[VM_ENV_DATA_INDEX_FLAGS] = src_ep[VM_ENV_DATA_INDEX_FLAGS] | VM_ENV_FLAG_ISOLATED;
1486 if (!VM_ENV_LOCAL_P(src_ep)) {
1487 VM_ENV_FLAGS_SET(ep, VM_ENV_FLAG_LOCAL);
1488 }
1489
1490 if (read_only_variables) {
1491 for (int i=RARRAY_LENINT(read_only_variables)-1; i>=0; i--) {
1492 ID id = NUM2ID(RARRAY_AREF(read_only_variables, i));
1493
1494 const struct rb_iseq_constant_body *body = ISEQ_BODY(src_env->iseq);
1495 for (unsigned int j=0; j<body->local_table_size; j++) {
1496 if (id == body->local_table[j]) {
1497 // check reassignment
1498 if (body->lvar_states[j] == lvar_reassigned) {
1499 VALUE name = rb_id2str(id);
1500 VALUE msg = rb_sprintf("cannot make a shareable Proc because "
1501 "the outer variable '%" PRIsVALUE "' may be reassigned.", name);
1502 rb_exc_raise(rb_exc_new_str(rb_eRactorIsolationError, msg));
1503 }
1504
1505 // check shareable
1506 VALUE v = src_env->env[j];
1507 if (!rb_ractor_shareable_p(v)) {
1508 VALUE name = rb_id2str(id);
1509 VALUE msg = rb_sprintf("cannot make a shareable Proc because it can refer"
1510 " unshareable object %+" PRIsVALUE " from ", v);
1511 if (name)
1512 rb_str_catf(msg, "variable '%" PRIsVALUE "'", name);
1513 else
1514 rb_str_cat_cstr(msg, "a hidden variable");
1515 rb_exc_raise(rb_exc_new_str(rb_eRactorIsolationError, msg));
1516 }
1517 RB_OBJ_WRITE((VALUE)copied_env, &env_body[j], v);
1518 rb_ary_delete_at(read_only_variables, i);
1519 break;
1520 }
1521 }
1522 }
1523 }
1524
1525 if (!VM_ENV_LOCAL_P(src_ep)) {
1526 const VALUE *prev_ep = VM_ENV_PREV_EP(src_env->ep);
1527 const rb_env_t *new_prev_env = env_copy(prev_ep, read_only_variables);
1528 ep[VM_ENV_DATA_INDEX_SPECVAL] = VM_GUARDED_PREV_EP(new_prev_env->ep);
1529 RB_OBJ_WRITTEN(copied_env, Qundef, new_prev_env);
1530 VM_ENV_FLAGS_UNSET(ep, VM_ENV_FLAG_LOCAL);
1531 }
1532 else {
1533 ep[VM_ENV_DATA_INDEX_SPECVAL] = VM_BLOCK_HANDLER_NONE;
1534 }
1535
1536 RB_OBJ_SET_SHAREABLE((VALUE)copied_env);
1537 return copied_env;
1538}
1539
1540static void
1541proc_isolate_env(VALUE self, rb_proc_t *proc, VALUE read_only_variables)
1542{
1543 const struct rb_captured_block *captured = &proc->block.as.captured;
1544 const rb_env_t *env = env_copy(captured->ep, read_only_variables);
1545 *((const VALUE **)&proc->block.as.captured.ep) = env->ep;
1546 RB_OBJ_WRITTEN(self, Qundef, env);
1547}
1548
1549static VALUE
1550proc_shared_outer_variables(struct rb_id_table *outer_variables, bool isolate, const char *message)
1551{
1552 struct collect_outer_variable_name_data data = {
1553 .isolate = isolate,
1554 .ary = Qfalse,
1555 .read_only = Qfalse,
1556 .yield = false,
1557 };
1558 rb_id_table_foreach(outer_variables, collect_outer_variable_names, (void *)&data);
1559
1560 if (data.ary != Qfalse) {
1561 VALUE str = rb_sprintf("can not %s because it accesses outer variables", message);
1562 VALUE ary = data.ary;
1563 const char *sep = " (";
1564 for (long i = 0; i < RARRAY_LEN(ary); i++) {
1565 VALUE name = rb_id2str(NUM2ID(RARRAY_AREF(ary, i)));
1566 if (!name) continue;
1567 rb_str_cat_cstr(str, sep);
1568 sep = ", ";
1569 rb_str_append(str, name);
1570 }
1571 if (*sep == ',') rb_str_cat_cstr(str, ")");
1572 rb_str_cat_cstr(str, data.yield ? " and uses 'yield'." : ".");
1573 rb_exc_raise(rb_exc_new_str(rb_eRactorIsolationError, str));
1574 }
1575 else if (data.yield) {
1576 rb_raise(rb_eRactorIsolationError, "can not %s because it uses 'yield'.", message);
1577 }
1578
1579 return data.read_only;
1580}
1581
1582VALUE
1583rb_proc_isolate_bang(VALUE self, VALUE replace_self)
1584{
1585 const rb_iseq_t *iseq = vm_proc_iseq(self);
1586
1587 if (iseq) {
1588 rb_proc_t *proc = (rb_proc_t *)RTYPEDDATA_DATA(self);
1589
1590 if (!UNDEF_P(replace_self)) {
1591 VM_ASSERT(rb_ractor_shareable_p(replace_self));
1592 RB_OBJ_WRITE(self, &proc->block.as.captured.self, replace_self);
1593 }
1594
1595 if (proc->block.type != block_type_iseq) rb_raise(rb_eRuntimeError, "not supported yet");
1596
1597 if (ISEQ_BODY(iseq)->outer_variables) {
1598 proc_shared_outer_variables(ISEQ_BODY(iseq)->outer_variables, true, "isolate a Proc");
1599 }
1600
1601 proc_isolate_env(self, proc, Qfalse);
1602 proc->is_isolated = TRUE;
1603 RB_OBJ_WRITE(self, &proc->block.as.captured.self, Qnil);
1604 }
1605
1606 RB_OBJ_SET_SHAREABLE(self);
1607 return self;
1608}
1609
1610VALUE
1611rb_proc_isolate(VALUE self)
1612{
1613 VALUE dst = rb_proc_dup(self);
1614 rb_proc_isolate_bang(dst, Qundef);
1615 return dst;
1616}
1617
1618VALUE
1619rb_proc_ractor_make_shareable(VALUE self, VALUE replace_self)
1620{
1621 const rb_iseq_t *iseq = vm_proc_iseq(self);
1622
1623 if (iseq) {
1624 rb_proc_t *proc = (rb_proc_t *)RTYPEDDATA_DATA(self);
1625
1626 if (!UNDEF_P(replace_self)) {
1627 RB_OBJ_WRITE(self, &proc->block.as.captured.self, replace_self);
1628 }
1629
1630 if (proc->block.type != block_type_iseq) rb_raise(rb_eRuntimeError, "not supported yet");
1631
1632 if (!rb_ractor_shareable_p(vm_block_self(&proc->block))) {
1633 rb_raise(rb_eRactorIsolationError,
1634 "Proc's self is not shareable: %" PRIsVALUE,
1635 self);
1636 }
1637
1638 VALUE read_only_variables = Qfalse;
1639
1640 if (ISEQ_BODY(iseq)->outer_variables) {
1641 read_only_variables =
1642 proc_shared_outer_variables(ISEQ_BODY(iseq)->outer_variables, false, "make a Proc shareable");
1643 }
1644
1645 proc_isolate_env(self, proc, read_only_variables);
1646 proc->is_isolated = TRUE;
1647 }
1648 else {
1649 const struct rb_block *block = vm_proc_block(self);
1650 if (block->type != block_type_symbol) rb_raise(rb_eRuntimeError, "not supported yet");
1651
1652 VALUE proc_self = vm_block_self(block);
1653 if (!rb_ractor_shareable_p(proc_self)) {
1654 rb_raise(rb_eRactorIsolationError,
1655 "Proc's self is not shareable: %" PRIsVALUE,
1656 self);
1657 }
1658 }
1659
1660 RB_OBJ_SET_FROZEN_SHAREABLE(self);
1661 return self;
1662}
1663
1664VALUE
1665rb_vm_make_proc_lambda(const rb_execution_context_t *ec, const struct rb_captured_block *captured, VALUE klass, int8_t is_lambda)
1666{
1667 VALUE procval;
1668 enum imemo_type code_type = imemo_type(captured->code.val);
1669
1670 if (!VM_ENV_ESCAPED_P(captured->ep)) {
1671 rb_control_frame_t *cfp = VM_CAPTURED_BLOCK_TO_CFP(captured);
1672 vm_make_env_object(ec, cfp);
1673 }
1674
1675 VM_ASSERT(VM_EP_IN_HEAP_P(ec, captured->ep));
1676 VM_ASSERT(code_type == imemo_iseq || code_type == imemo_ifunc);
1677
1678 procval = vm_proc_create_from_captured(klass, captured,
1679 code_type == imemo_iseq ? block_type_iseq : block_type_ifunc,
1680 FALSE, is_lambda);
1681
1682 if (code_type == imemo_ifunc) {
1683 struct vm_ifunc *ifunc = (struct vm_ifunc *)captured->code.val;
1684 if (ifunc->svar_lep) {
1685 VALUE ep0 = ifunc->svar_lep[0];
1686 if (RB_TYPE_P(ep0, T_IMEMO) && imemo_type_p(ep0, imemo_env)) {
1687 // `ep0 == imemo_env` means this ep is escaped to heap (in env object).
1688 const rb_env_t *env = (const rb_env_t *)ep0;
1689 ifunc->svar_lep = (VALUE *)env->ep;
1690 }
1691 else {
1692 VM_ASSERT(FIXNUM_P(ep0));
1693 if (ep0 & VM_ENV_FLAG_ESCAPED) {
1694 // ok. do nothing
1695 }
1696 else {
1697 ifunc->svar_lep = NULL;
1698 }
1699 }
1700 }
1701 }
1702
1703 return procval;
1704}
1705
1706/* Binding */
1707
1708VALUE
1709rb_vm_make_binding(const rb_execution_context_t *ec, const rb_control_frame_t *src_cfp)
1710{
1711 rb_control_frame_t *cfp = rb_vm_get_binding_creatable_next_cfp(ec, src_cfp);
1712 rb_control_frame_t *ruby_level_cfp = rb_vm_get_ruby_level_next_cfp(ec, src_cfp);
1713 VALUE bindval, envval;
1714 rb_binding_t *bind;
1715
1716 if (cfp == 0 || ruby_level_cfp == 0) {
1717 rb_raise(rb_eRuntimeError, "Can't create Binding Object on top of Fiber.");
1718 }
1719 if (!VM_FRAME_RUBYFRAME_P(src_cfp) &&
1720 !VM_FRAME_RUBYFRAME_P(RUBY_VM_PREVIOUS_CONTROL_FRAME(src_cfp))) {
1721 rb_raise(rb_eRuntimeError, "Cannot create Binding object for non-Ruby caller");
1722 }
1723
1724 envval = vm_make_env_object(ec, cfp);
1725 bindval = rb_binding_alloc(rb_cBinding);
1726 GetBindingPtr(bindval, bind);
1727 vm_bind_update_env(bindval, bind, envval);
1728 RB_OBJ_WRITE(bindval, &bind->block.as.captured.self, cfp->self);
1729 RB_OBJ_WRITE(bindval, &bind->block.as.captured.code.iseq, CFP_ISEQ(cfp));
1730 RB_OBJ_WRITE(bindval, &bind->pathobj, ISEQ_BODY(CFP_ISEQ(ruby_level_cfp))->location.pathobj);
1731 bind->first_lineno = rb_vm_get_sourceline(ruby_level_cfp);
1732
1733 return bindval;
1734}
1735
1736const VALUE *
1737rb_binding_add_dynavars(VALUE bindval, rb_binding_t *bind, int dyncount, const ID *dynvars)
1738{
1739 VALUE envval, pathobj = bind->pathobj;
1740 VALUE path = pathobj_path(pathobj);
1741 VALUE realpath = pathobj_realpath(pathobj);
1742 const struct rb_block *base_block;
1743 const rb_env_t *env;
1744 rb_execution_context_t *ec = GET_EC();
1745 const rb_iseq_t *base_iseq, *iseq;
1746 rb_node_scope_t tmp_node;
1747
1748 if (dyncount < 0) return 0;
1749
1750 base_block = &bind->block;
1751 base_iseq = vm_block_iseq(base_block);
1752
1753 VALUE idtmp = 0;
1754 rb_ast_id_table_t *dyns = ALLOCV(idtmp, sizeof(rb_ast_id_table_t) + dyncount * sizeof(ID));
1755 dyns->size = dyncount;
1756 MEMCPY(dyns->ids, dynvars, ID, dyncount);
1757
1758 rb_node_init(RNODE(&tmp_node), NODE_SCOPE);
1759 tmp_node.nd_tbl = dyns;
1760 tmp_node.nd_body = 0;
1761 tmp_node.nd_parent = NULL;
1762 tmp_node.nd_args = 0;
1763
1764 VALUE ast_value = rb_ruby_ast_new(RNODE(&tmp_node));
1765
1766 if (base_iseq) {
1767 iseq = rb_iseq_new(ast_value, ISEQ_BODY(base_iseq)->location.label, path, realpath, base_iseq, ISEQ_TYPE_EVAL);
1768 }
1769 else {
1770 VALUE tempstr = rb_fstring_lit("<temp>");
1771 iseq = rb_iseq_new_top(ast_value, tempstr, tempstr, tempstr, NULL);
1772 }
1773 tmp_node.nd_tbl = 0; /* reset table */
1774 ALLOCV_END(idtmp);
1775
1776 vm_set_eval_stack(ec, iseq, 0, base_block);
1777 vm_bind_update_env(bindval, bind, envval = vm_make_env_object(ec, ec->cfp));
1778 rb_vm_pop_frame(ec);
1779
1780 env = (const rb_env_t *)envval;
1781 return env->env;
1782}
1783
1784/* C -> Ruby: block */
1785
1786static inline void
1787invoke_block(rb_execution_context_t *ec, const rb_iseq_t *iseq, VALUE self, const struct rb_captured_block *captured, const rb_cref_t *cref, VALUE type, int opt_pc)
1788{
1789 int arg_size = ISEQ_BODY(iseq)->param.size;
1790
1791 vm_push_frame(ec, iseq, type | VM_FRAME_FLAG_FINISH, self,
1792 VM_GUARDED_PREV_EP(captured->ep),
1793 (VALUE)cref, /* cref or method */
1794 ISEQ_BODY(iseq)->iseq_encoded + opt_pc,
1795 ec->cfp->sp + arg_size,
1796 ISEQ_BODY(iseq)->local_table_size - arg_size,
1797 ISEQ_BODY(iseq)->stack_max);
1798}
1799
1800static inline void
1801invoke_bmethod(rb_execution_context_t *ec, const rb_iseq_t *iseq, VALUE self, const struct rb_captured_block *captured, const rb_callable_method_entry_t *me, VALUE type, int opt_pc)
1802{
1803 /* bmethod call from outside the VM */
1804 int arg_size = ISEQ_BODY(iseq)->param.size;
1805
1806 VM_ASSERT(me->def->type == VM_METHOD_TYPE_BMETHOD);
1807
1808 vm_push_frame(ec, iseq, type | VM_FRAME_FLAG_BMETHOD, self,
1809 VM_GUARDED_PREV_EP(captured->ep),
1810 (VALUE)me,
1811 ISEQ_BODY(iseq)->iseq_encoded + opt_pc,
1812 ec->cfp->sp + 1 /* self */ + arg_size,
1813 ISEQ_BODY(iseq)->local_table_size - arg_size,
1814 ISEQ_BODY(iseq)->stack_max);
1815
1816 VM_ENV_FLAGS_SET(ec->cfp->ep, VM_FRAME_FLAG_FINISH);
1817}
1818
1819ALWAYS_INLINE(static VALUE
1820 invoke_iseq_block_from_c(rb_execution_context_t *ec, const struct rb_captured_block *captured,
1821 VALUE self, int argc, const VALUE *argv, int kw_splat, VALUE passed_block_handler,
1822 const rb_cref_t *cref, int is_lambda, const rb_callable_method_entry_t *me));
1823
1824static inline VALUE
1825invoke_iseq_block_from_c(rb_execution_context_t *ec, const struct rb_captured_block *captured,
1826 VALUE self, int argc, const VALUE *argv, int kw_splat, VALUE passed_block_handler,
1827 const rb_cref_t *cref, int is_lambda, const rb_callable_method_entry_t *me)
1828{
1829 const rb_iseq_t *iseq = rb_iseq_check(captured->code.iseq);
1830 int opt_pc;
1831 VALUE type = VM_FRAME_MAGIC_BLOCK | (is_lambda ? VM_FRAME_FLAG_LAMBDA : 0);
1832 rb_control_frame_t *cfp = ec->cfp;
1833 VALUE *sp = cfp->sp;
1834 int flags = (kw_splat ? VM_CALL_KW_SPLAT : 0);
1835 VALUE *use_argv = (VALUE *)argv;
1836 VALUE av[2];
1837
1838 stack_check(ec);
1839
1840 if (UNLIKELY(argc > VM_ARGC_STACK_MAX) &&
1841 (VM_ARGC_STACK_MAX >= 1 ||
1842 /* Skip ruby array for potential autosplat case */
1843 (argc != 1 || is_lambda))) {
1844 use_argv = vm_argv_ruby_array(av, argv, &flags, &argc, kw_splat);
1845 }
1846
1847 CHECK_VM_STACK_OVERFLOW(cfp, argc + 1);
1848 vm_check_canary(ec, sp);
1849
1850 VALUE *stack_argv = sp;
1851 if (me) {
1852 *sp = self; // bemthods need `self` on the VM stack
1853 stack_argv++;
1854 }
1855 cfp->sp = stack_argv + argc;
1856 MEMCPY(stack_argv, use_argv, VALUE, argc); // restrict: new stack space
1857
1858 opt_pc = vm_yield_setup_args(ec, iseq, argc, stack_argv, flags, passed_block_handler,
1859 (is_lambda ? arg_setup_method : arg_setup_block));
1860 cfp->sp = sp;
1861
1862 if (me == NULL) {
1863 invoke_block(ec, iseq, self, captured, cref, type, opt_pc);
1864 }
1865 else {
1866 invoke_bmethod(ec, iseq, self, captured, me, type, opt_pc);
1867 }
1868
1869 return vm_exec(ec);
1870}
1871
1872static VALUE
1873invoke_block_from_c_bh(rb_execution_context_t *ec, VALUE block_handler,
1874 int argc, const VALUE *argv,
1875 int kw_splat, VALUE passed_block_handler, const rb_cref_t *cref,
1876 int is_lambda, int force_blockarg)
1877{
1878 again:
1879 switch (vm_block_handler_type(block_handler)) {
1880 case block_handler_type_iseq:
1881 {
1882 const struct rb_captured_block *captured = VM_BH_TO_ISEQ_BLOCK(block_handler);
1883 return invoke_iseq_block_from_c(ec, captured, captured->self,
1884 argc, argv, kw_splat, passed_block_handler,
1885 cref, is_lambda, NULL);
1886 }
1887 case block_handler_type_ifunc:
1888 return vm_yield_with_cfunc(ec, VM_BH_TO_IFUNC_BLOCK(block_handler),
1889 VM_BH_TO_IFUNC_BLOCK(block_handler)->self,
1890 argc, argv, kw_splat, passed_block_handler, NULL);
1891 case block_handler_type_symbol:
1892 return vm_yield_with_symbol(ec, VM_BH_TO_SYMBOL(block_handler),
1893 argc, argv, kw_splat, passed_block_handler);
1894 case block_handler_type_proc:
1895 {
1896 VALUE procval = VM_BH_TO_PROC(block_handler);
1897 rb_proc_t *po;
1898 GetProcPtr(procval, po);
1899 if (po->is_refined) cref = rb_proc_refinements_cref_for_call(procval);
1900 if (force_blockarg == FALSE) {
1901 is_lambda = po->is_lambda;
1902 }
1903 block_handler = vm_block_to_block_handler(&po->block);
1904 goto again;
1905 }
1906 }
1907 VM_UNREACHABLE(invoke_block_from_c_splattable);
1908 return Qundef;
1909}
1910
1911static inline VALUE
1912check_block_handler(rb_execution_context_t *ec)
1913{
1914 VALUE block_handler = VM_CF_BLOCK_HANDLER(ec->cfp);
1915 vm_block_handler_verify(block_handler);
1916 if (UNLIKELY(block_handler == VM_BLOCK_HANDLER_NONE)) {
1917 rb_vm_localjump_error("no block given", Qnil, 0);
1918 }
1919
1920 return block_handler;
1921}
1922
1923static VALUE
1924vm_yield_with_cref(rb_execution_context_t *ec, int argc, const VALUE *argv, int kw_splat, const rb_cref_t *cref, int is_lambda)
1925{
1926 return invoke_block_from_c_bh(ec, check_block_handler(ec),
1927 argc, argv, kw_splat, VM_BLOCK_HANDLER_NONE,
1928 cref, is_lambda, FALSE);
1929}
1930
1931static VALUE
1932vm_yield(rb_execution_context_t *ec, int argc, const VALUE *argv, int kw_splat)
1933{
1934 return vm_yield_with_cref(ec, argc, argv, kw_splat, NULL, FALSE);
1935}
1936
1937static VALUE
1938vm_yield_with_block(rb_execution_context_t *ec, int argc, const VALUE *argv, VALUE block_handler, int kw_splat)
1939{
1940 return invoke_block_from_c_bh(ec, check_block_handler(ec),
1941 argc, argv, kw_splat, block_handler,
1942 NULL, FALSE, FALSE);
1943}
1944
1945static VALUE
1946vm_yield_force_blockarg(rb_execution_context_t *ec, VALUE args)
1947{
1948 return invoke_block_from_c_bh(ec, check_block_handler(ec), 1, &args,
1949 RB_NO_KEYWORDS, VM_BLOCK_HANDLER_NONE, NULL, FALSE, TRUE);
1950}
1951
1952ALWAYS_INLINE(static VALUE
1953 invoke_block_from_c_proc(rb_execution_context_t *ec, const rb_proc_t *proc,
1954 VALUE self, int argc, const VALUE *argv,
1955 int kw_splat, VALUE passed_block_handler, int is_lambda,
1956 const rb_cref_t *cref,
1957 const rb_callable_method_entry_t *me));
1958
1959static inline VALUE
1960invoke_block_from_c_proc(rb_execution_context_t *ec, const rb_proc_t *proc,
1961 VALUE self, int argc, const VALUE *argv,
1962 int kw_splat, VALUE passed_block_handler, int is_lambda,
1963 const rb_cref_t *cref,
1965{
1966 const struct rb_block *block = &proc->block;
1967
1968 again:
1969 switch (vm_block_type(block)) {
1970 case block_type_iseq:
1971 return invoke_iseq_block_from_c(ec, &block->as.captured, self, argc, argv, kw_splat, passed_block_handler, cref, is_lambda, me);
1972 case block_type_ifunc:
1973 if (kw_splat == 1) {
1974 VALUE keyword_hash = argv[argc-1];
1975 if (!RB_TYPE_P(keyword_hash, T_HASH)) {
1976 keyword_hash = rb_to_hash_type(keyword_hash);
1977 }
1978 if (RHASH_EMPTY_P(keyword_hash)) {
1979 argc--;
1980 }
1981 else {
1982 ((VALUE *)argv)[argc-1] = rb_hash_dup(keyword_hash);
1983 }
1984 }
1985 return vm_yield_with_cfunc(ec, &block->as.captured, self, argc, argv, kw_splat, passed_block_handler, me);
1986 case block_type_symbol:
1987 return vm_yield_with_symbol(ec, block->as.symbol, argc, argv, kw_splat, passed_block_handler);
1988 case block_type_proc:
1989 is_lambda = block_proc_is_lambda(block->as.proc);
1990 block = vm_proc_block(block->as.proc);
1991 goto again;
1992 }
1993 VM_UNREACHABLE(invoke_block_from_c_proc);
1994 return Qundef;
1995}
1996
1997static VALUE
1998vm_invoke_proc(rb_execution_context_t *ec, rb_proc_t *proc, VALUE self,
1999 int argc, const VALUE *argv, int kw_splat, VALUE passed_block_handler,
2000 const rb_cref_t *cref)
2001{
2002 return invoke_block_from_c_proc(ec, proc, self, argc, argv, kw_splat, passed_block_handler, proc->is_lambda, cref, NULL);
2003}
2004
2005static VALUE
2006vm_invoke_bmethod(rb_execution_context_t *ec, rb_proc_t *proc, VALUE self,
2007 int argc, const VALUE *argv, int kw_splat, VALUE block_handler, const rb_callable_method_entry_t *me)
2008{
2009 /* bmethod procs never carry a refinement cref (Proc#refined rejects them) */
2010 return invoke_block_from_c_proc(ec, proc, self, argc, argv, kw_splat, block_handler, TRUE, NULL, me);
2011}
2012
2013VALUE
2014rb_vm_invoke_proc(rb_execution_context_t *ec, rb_proc_t *proc,
2015 int argc, const VALUE *argv, int kw_splat, VALUE passed_block_handler,
2016 const rb_cref_t *cref)
2017{
2018 VALUE self = vm_block_self(&proc->block);
2019 vm_block_handler_verify(passed_block_handler);
2020
2021 if (proc->is_from_method) {
2022 return vm_invoke_bmethod(ec, proc, self, argc, argv, kw_splat, passed_block_handler, NULL);
2023 }
2024 else {
2025 return vm_invoke_proc(ec, proc, self, argc, argv, kw_splat, passed_block_handler, cref);
2026 }
2027}
2028
2029VALUE
2030rb_vm_invoke_proc_with_self(rb_execution_context_t *ec, rb_proc_t *proc, VALUE self,
2031 int argc, const VALUE *argv, int kw_splat, VALUE passed_block_handler,
2032 const rb_cref_t *cref)
2033{
2034 vm_block_handler_verify(passed_block_handler);
2035
2036 if (proc->is_from_method) {
2037 return vm_invoke_bmethod(ec, proc, self, argc, argv, kw_splat, passed_block_handler, NULL);
2038 }
2039 else {
2040 return vm_invoke_proc(ec, proc, self, argc, argv, kw_splat, passed_block_handler, cref);
2041 }
2042}
2043
2044/* special variable */
2045
2046VALUE *
2047rb_vm_svar_lep(const rb_execution_context_t *ec, const rb_control_frame_t *cfp)
2048{
2049 while (!CFP_PC(cfp) || !CFP_ISEQ(cfp)) {
2050 if (VM_FRAME_TYPE(cfp) == VM_FRAME_MAGIC_IFUNC) {
2051 struct vm_ifunc *ifunc = (struct vm_ifunc *)CFP_ISEQ(cfp);
2052 return ifunc->svar_lep;
2053 }
2054 else {
2055 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
2056 }
2057
2058 if (RUBY_VM_CONTROL_FRAME_STACK_OVERFLOW_P(ec, cfp)) {
2059 return NULL;
2060 }
2061 }
2062
2063 return (VALUE *)VM_CF_LEP(cfp);
2064}
2065
2066static VALUE
2067vm_cfp_svar_get(const rb_execution_context_t *ec, rb_control_frame_t *cfp, VALUE key)
2068{
2069 return lep_svar_get(ec, rb_vm_svar_lep(ec, cfp), key);
2070}
2071
2072static void
2073vm_cfp_svar_set(const rb_execution_context_t *ec, rb_control_frame_t *cfp, VALUE key, const VALUE val)
2074{
2075 lep_svar_set(ec, rb_vm_svar_lep(ec, cfp), key, val);
2076}
2077
2078static VALUE
2079vm_svar_get(const rb_execution_context_t *ec, VALUE key)
2080{
2081 return vm_cfp_svar_get(ec, ec->cfp, key);
2082}
2083
2084static void
2085vm_svar_set(const rb_execution_context_t *ec, VALUE key, VALUE val)
2086{
2087 vm_cfp_svar_set(ec, ec->cfp, key, val);
2088}
2089
2090VALUE
2092{
2093 return vm_svar_get(GET_EC(), VM_SVAR_BACKREF);
2094}
2095
2096void
2098{
2099 vm_svar_set(GET_EC(), VM_SVAR_BACKREF, val);
2100}
2101
2102VALUE
2104{
2105 return vm_svar_get(GET_EC(), VM_SVAR_LASTLINE);
2106}
2107
2108void
2110{
2111 vm_svar_set(GET_EC(), VM_SVAR_LASTLINE, val);
2112}
2113
2114void
2115rb_lastline_set_up(VALUE val, unsigned int up)
2116{
2117 rb_control_frame_t * cfp = GET_EC()->cfp;
2118
2119 for(unsigned int i = 0; i < up; i++) {
2120 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
2121 }
2122 vm_cfp_svar_set(GET_EC(), cfp, VM_SVAR_LASTLINE, val);
2123}
2124
2125/* misc */
2126
2127const char *
2129{
2130 const rb_execution_context_t *ec = GET_EC();
2131 const rb_control_frame_t *cfp = rb_vm_get_ruby_level_next_cfp(ec, ec->cfp);
2132
2133 if (cfp) {
2134 return RSTRING_PTR(rb_iseq_path(CFP_ISEQ(cfp)));
2135 }
2136 else {
2137 return 0;
2138 }
2139}
2140
2141int
2143{
2144 const rb_execution_context_t *ec = GET_EC();
2145 const rb_control_frame_t *cfp = rb_vm_get_ruby_level_next_cfp(ec, ec->cfp);
2146
2147 if (cfp) {
2148 return rb_vm_get_sourceline(cfp);
2149 }
2150 else {
2151 return 0;
2152 }
2153}
2154
2155VALUE
2156rb_source_location(int *pline)
2157{
2158 const rb_execution_context_t *ec = GET_EC();
2159 const rb_control_frame_t *cfp = rb_vm_get_ruby_level_next_cfp(ec, ec->cfp);
2160
2161 if (cfp && VM_FRAME_RUBYFRAME_P(cfp)) {
2162 if (pline) *pline = rb_vm_get_sourceline(cfp);
2163 return rb_iseq_path(CFP_ISEQ(cfp));
2164 }
2165 else {
2166 if (pline) *pline = 0;
2167 return Qnil;
2168 }
2169}
2170
2171const char *
2172rb_source_location_cstr(int *pline)
2173{
2174 VALUE path = rb_source_location(pline);
2175 if (NIL_P(path)) return NULL;
2176 return RSTRING_PTR(path);
2177}
2178
2179rb_cref_t *
2180rb_vm_cref(void)
2181{
2182 const rb_execution_context_t *ec = GET_EC();
2183 return vm_ec_cref(ec);
2184}
2185
2186rb_cref_t *
2187rb_vm_cref_replace_with_duplicated_cref(void)
2188{
2189 const rb_execution_context_t *ec = GET_EC();
2190 const rb_control_frame_t *cfp = rb_vm_get_ruby_level_next_cfp(ec, ec->cfp);
2191 rb_cref_t *cref = vm_cref_replace_with_duplicated_cref(cfp->ep);
2192 ASSUME(cref);
2193 return cref;
2194}
2195
2196const rb_cref_t *
2197rb_vm_cref_in_context(VALUE self, VALUE cbase)
2198{
2199 const rb_execution_context_t *ec = GET_EC();
2200 const rb_control_frame_t *cfp = rb_vm_get_ruby_level_next_cfp(ec, ec->cfp);
2201 const rb_cref_t *cref;
2202 if (!cfp || cfp->self != self) return NULL;
2203 if (!vm_env_cref_by_cref(cfp->ep)) return NULL;
2204 cref = vm_get_cref(cfp->ep);
2205 if (CREF_CLASS(cref) != cbase) return NULL;
2206 return cref;
2207}
2208
2209#if 0
2210void
2211debug_cref(rb_cref_t *cref)
2212{
2213 while (cref) {
2214 dp(CREF_CLASS(cref));
2215 printf("%ld\n", CREF_VISI(cref));
2216 cref = CREF_NEXT(cref);
2217 }
2218}
2219#endif
2220
2221VALUE
2222rb_vm_cbase(void)
2223{
2224 const rb_execution_context_t *ec = GET_EC();
2225 const rb_control_frame_t *cfp = rb_vm_get_ruby_level_next_cfp(ec, ec->cfp);
2226
2227 if (cfp == 0) {
2228 rb_raise(rb_eRuntimeError, "Can't call on top of Fiber or Thread");
2229 }
2230 return vm_get_cbase(cfp->ep);
2231}
2232
2233/* jump */
2234
2235static VALUE
2236make_localjump_error(const char *mesg, VALUE value, int reason)
2237{
2240 ID id;
2241
2242 switch (reason) {
2243 case TAG_BREAK:
2244 CONST_ID(id, "break");
2245 break;
2246 case TAG_REDO:
2247 CONST_ID(id, "redo");
2248 break;
2249 case TAG_RETRY:
2250 CONST_ID(id, "retry");
2251 break;
2252 case TAG_NEXT:
2253 CONST_ID(id, "next");
2254 break;
2255 case TAG_RETURN:
2256 CONST_ID(id, "return");
2257 break;
2258 default:
2259 CONST_ID(id, "noreason");
2260 break;
2261 }
2262 rb_iv_set(exc, "@exit_value", value);
2263 rb_iv_set(exc, "@reason", ID2SYM(id));
2264 return exc;
2265}
2266
2267void
2268rb_vm_localjump_error(const char *mesg, VALUE value, int reason)
2269{
2270 VALUE exc = make_localjump_error(mesg, value, reason);
2271 rb_exc_raise(exc);
2272}
2273
2274VALUE
2275rb_vm_make_jump_tag_but_local_jump(enum ruby_tag_type state, VALUE val)
2276{
2277 const char *mesg;
2278
2279 switch (state) {
2280 case TAG_RETURN:
2281 mesg = "unexpected return";
2282 break;
2283 case TAG_BREAK:
2284 mesg = "unexpected break";
2285 break;
2286 case TAG_NEXT:
2287 mesg = "unexpected next";
2288 break;
2289 case TAG_REDO:
2290 mesg = "unexpected redo";
2291 val = Qnil;
2292 break;
2293 case TAG_RETRY:
2294 mesg = "retry outside of rescue clause";
2295 val = Qnil;
2296 break;
2297 default:
2298 return Qnil;
2299 }
2300 if (UNDEF_P(val)) {
2301 val = GET_EC()->tag->retval;
2302 }
2303 return make_localjump_error(mesg, val, state);
2304}
2305
2306void
2307rb_vm_jump_tag_but_local_jump(enum ruby_tag_type state)
2308{
2309 VALUE exc = rb_vm_make_jump_tag_but_local_jump(state, Qundef);
2310 if (!NIL_P(exc)) rb_exc_raise(exc);
2311 EC_JUMP_TAG(GET_EC(), state);
2312}
2313
2314static rb_control_frame_t *
2315next_not_local_frame(rb_control_frame_t *cfp)
2316{
2317 while (VM_ENV_LOCAL_P(cfp->ep)) {
2318 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
2319 }
2320 return cfp;
2321}
2322
2323NORETURN(static void vm_iter_break(rb_execution_context_t *ec, VALUE val));
2324
2325static void
2326vm_iter_break(rb_execution_context_t *ec, VALUE val)
2327{
2328 rb_control_frame_t *cfp = next_not_local_frame(ec->cfp);
2329 const VALUE *ep = VM_CF_PREV_EP(cfp);
2330 const rb_control_frame_t *target_cfp = rb_vm_search_cf_from_ep(ec, cfp, ep);
2331
2332 if (!target_cfp) {
2333 rb_vm_localjump_error("unexpected break", val, TAG_BREAK);
2334 }
2335
2336 ec->errinfo = (VALUE)THROW_DATA_NEW(val, target_cfp, TAG_BREAK);
2337 EC_JUMP_TAG(ec, TAG_BREAK);
2338}
2339
2340void
2342{
2343 vm_iter_break(GET_EC(), Qnil);
2344}
2345
2346void
2348{
2349 vm_iter_break(GET_EC(), val);
2350}
2351
2352/* optimization: redefine management */
2353
2354short ruby_vm_redefined_flag[BOP_LAST_];
2355static st_table *vm_opt_method_def_table = 0;
2356static st_table *vm_opt_mid_table = 0;
2357
2358static void
2359rb_free_vm_opt_tables(void)
2360{
2361 st_free_table(vm_opt_method_def_table);
2362 st_free_table(vm_opt_mid_table);
2363}
2364
2365static int
2366vm_redefinition_check_flag(VALUE klass)
2367{
2368 if (klass == rb_cInteger) return INTEGER_REDEFINED_OP_FLAG;
2369 if (klass == rb_cFloat) return FLOAT_REDEFINED_OP_FLAG;
2370 if (klass == rb_cString) return STRING_REDEFINED_OP_FLAG;
2371 if (klass == rb_cArray) return ARRAY_REDEFINED_OP_FLAG;
2372 if (klass == rb_cHash) return HASH_REDEFINED_OP_FLAG;
2373 if (klass == rb_cSymbol) return SYMBOL_REDEFINED_OP_FLAG;
2374#if 0
2375 if (klass == rb_cTime) return TIME_REDEFINED_OP_FLAG;
2376#endif
2377 if (klass == rb_cRegexp) return REGEXP_REDEFINED_OP_FLAG;
2378 if (klass == rb_cNilClass) return NIL_REDEFINED_OP_FLAG;
2379 if (klass == rb_cTrueClass) return TRUE_REDEFINED_OP_FLAG;
2380 if (klass == rb_cFalseClass) return FALSE_REDEFINED_OP_FLAG;
2381 if (klass == rb_cProc) return PROC_REDEFINED_OP_FLAG;
2382 return 0;
2383}
2384
2385int
2386rb_vm_check_optimizable_mid(VALUE mid)
2387{
2388 if (!vm_opt_mid_table) {
2389 return FALSE;
2390 }
2391
2392 return st_lookup(vm_opt_mid_table, mid, NULL);
2393}
2394
2395static int
2396vm_redefinition_check_method_type(const rb_method_entry_t *me)
2397{
2398 if (me->called_id != me->def->original_id) {
2399 return FALSE;
2400 }
2401
2402 if (METHOD_ENTRY_BASIC(me)) return TRUE;
2403
2404 const rb_method_definition_t *def = me->def;
2405 switch (def->type) {
2406 case VM_METHOD_TYPE_CFUNC:
2407 case VM_METHOD_TYPE_OPTIMIZED:
2408 return TRUE;
2409 default:
2410 return FALSE;
2411 }
2412}
2413
2414static void
2415rb_vm_check_redefinition_opt_method(const rb_method_entry_t *me, VALUE klass)
2416{
2417 st_data_t bop;
2418 if (RB_TYPE_P(klass, T_ICLASS) && RICLASS_IS_ORIGIN_P(klass) &&
2419 RB_TYPE_P(RBASIC_CLASS(klass), T_CLASS)) {
2420 klass = RBASIC_CLASS(klass);
2421 }
2422 if (vm_redefinition_check_method_type(me)) {
2423 if (st_lookup(vm_opt_method_def_table, (st_data_t)me->def, &bop)) {
2424 int flag = vm_redefinition_check_flag(klass);
2425 if (flag != 0) {
2428 "Redefining '%s#%s' disables interpreter and JIT optimizations",
2429 rb_class2name(me->owner),
2430 rb_id2name(me->called_id)
2431 );
2432 rb_yjit_bop_redefined(flag, (enum ruby_basic_operators)bop);
2433 rb_zjit_bop_redefined(flag, (enum ruby_basic_operators)bop);
2434 ruby_vm_redefined_flag[bop] |= flag;
2435 }
2436 }
2437 }
2438}
2439
2440static enum rb_id_table_iterator_result
2441check_redefined_method(ID mid, VALUE value, void *data)
2442{
2443 VALUE klass = (VALUE)data;
2444 const rb_method_entry_t *me = (rb_method_entry_t *)value;
2445 const rb_method_entry_t *newme = rb_method_entry(klass, mid);
2446
2447 if (newme != me) rb_vm_check_redefinition_opt_method(me, me->owner);
2448
2449 return ID_TABLE_CONTINUE;
2450}
2451
2452void
2453rb_vm_check_redefinition_by_prepend(VALUE klass)
2454{
2455 if (!vm_redefinition_check_flag(klass)) return;
2456 rb_id_table_foreach(RCLASS_M_TBL(RCLASS_ORIGIN(klass)), check_redefined_method, (void *)klass);
2457}
2458
2459static void
2460add_opt_method_entry_bop(const rb_method_entry_t *me, ID mid, enum ruby_basic_operators bop)
2461{
2462 st_insert(vm_opt_method_def_table, (st_data_t)me->def, (st_data_t)bop);
2463 st_insert(vm_opt_mid_table, (st_data_t)mid, (st_data_t)Qtrue);
2464}
2465
2466static void
2467add_opt_method(VALUE klass, ID mid, enum ruby_basic_operators bop)
2468{
2469 const rb_method_entry_t *me = rb_method_entry_at(klass, mid);
2470
2471 if (me && vm_redefinition_check_method_type(me)) {
2472 add_opt_method_entry_bop(me, mid, bop);
2473 }
2474 else {
2475 rb_bug("undefined optimized method: %s", rb_id2name(mid));
2476 }
2477}
2478
2479static enum ruby_basic_operators vm_redefinition_bop_for_id(ID mid);
2480
2481static void
2482add_opt_method_entry(const rb_method_entry_t *me)
2483{
2484 if (me && vm_redefinition_check_method_type(me)) {
2485 ID mid = me->called_id;
2486 enum ruby_basic_operators bop = vm_redefinition_bop_for_id(mid);
2487 if ((int)bop >= 0) {
2488 add_opt_method_entry_bop(me, mid, bop);
2489 }
2490 }
2491}
2492
2493static void
2494vm_init_redefined_flag(void)
2495{
2496 ID mid;
2497 enum ruby_basic_operators bop;
2498
2499#define OP(mid_, bop_) (mid = id##mid_, bop = BOP_##bop_, ruby_vm_redefined_flag[bop] = 0)
2500#define C(k) add_opt_method(rb_c##k, mid, bop)
2501 OP(PLUS, PLUS), (C(Integer), C(Float), C(String), C(Array));
2502 OP(MINUS, MINUS), (C(Integer), C(Float));
2503 OP(MULT, MULT), (C(Integer), C(Float));
2504 OP(DIV, DIV), (C(Integer), C(Float));
2505 OP(MOD, MOD), (C(Integer), C(Float));
2506 OP(Eq, EQ), (C(Integer), C(Float), C(String), C(Symbol));
2507 OP(Eqq, EQQ), (C(Integer), C(Float), C(Symbol), C(String),
2508 C(NilClass), C(TrueClass), C(FalseClass));
2509 OP(LT, LT), (C(Integer), C(Float));
2510 OP(LE, LE), (C(Integer), C(Float));
2511 OP(GT, GT), (C(Integer), C(Float));
2512 OP(GE, GE), (C(Integer), C(Float));
2513 OP(LTLT, LTLT), (C(String), C(Array));
2514 OP(GTGT, GTGT), (C(Integer));
2515 OP(AREF, AREF), (C(Array), C(Hash), C(Integer));
2516 OP(ASET, ASET), (C(Array), C(Hash));
2517 OP(Length, LENGTH), (C(Array), C(String), C(Hash));
2518 OP(Size, SIZE), (C(Array), C(String), C(Hash));
2519 OP(EmptyP, EMPTY_P), (C(Array), C(String), C(Hash));
2520 OP(Succ, SUCC), (C(Integer), C(String));
2521 OP(EqTilde, MATCH), (C(Regexp), C(String));
2522 OP(Freeze, FREEZE), (C(String), C(Array), C(Hash));
2523 OP(UMinus, UMINUS), (C(String));
2524 OP(Max, MAX), (C(Array));
2525 OP(Min, MIN), (C(Array));
2526 OP(Hash, HASH), (C(Array));
2527 OP(Call, CALL), (C(Proc));
2528 OP(Yield, YIELD), (C(Proc));
2529 OP(And, AND), (C(Integer));
2530 OP(Or, OR), (C(Integer));
2531 OP(NilP, NIL_P), (C(NilClass));
2532 OP(Cmp, CMP), (C(Integer), C(Float), C(String));
2533 OP(Default, DEFAULT), (C(Hash));
2534 OP(IncludeP, INCLUDE_P), (C(Array));
2535#undef C
2536#undef OP
2537}
2538
2539static enum ruby_basic_operators
2540vm_redefinition_bop_for_id(ID mid)
2541{
2542 switch (mid) {
2543#define OP(mid_, bop_) case id##mid_: return BOP_##bop_
2544 OP(PLUS, PLUS);
2545 OP(MINUS, MINUS);
2546 OP(MULT, MULT);
2547 OP(DIV, DIV);
2548 OP(MOD, MOD);
2549 OP(Eq, EQ);
2550 OP(Eqq, EQQ);
2551 OP(LT, LT);
2552 OP(LE, LE);
2553 OP(GT, GT);
2554 OP(GE, GE);
2555 OP(LTLT, LTLT);
2556 OP(AREF, AREF);
2557 OP(ASET, ASET);
2558 OP(Length, LENGTH);
2559 OP(Size, SIZE);
2560 OP(EmptyP, EMPTY_P);
2561 OP(Succ, SUCC);
2562 OP(EqTilde, MATCH);
2563 OP(Freeze, FREEZE);
2564 OP(UMinus, UMINUS);
2565 OP(Max, MAX);
2566 OP(Min, MIN);
2567 OP(Hash, HASH);
2568 OP(Call, CALL);
2569 OP(Yield, YIELD);
2570 OP(And, AND);
2571 OP(Or, OR);
2572 OP(NilP, NIL_P);
2573 OP(Cmp, CMP);
2574 OP(Default, DEFAULT);
2575 OP(Pack, PACK);
2576#undef OP
2577 }
2578 return -1;
2579}
2580
2581/* for vm development */
2582
2583#if VMDEBUG
2584static const char *
2585vm_frametype_name(const rb_control_frame_t *cfp)
2586{
2587 switch (VM_FRAME_TYPE(cfp)) {
2588 case VM_FRAME_MAGIC_METHOD: return "method";
2589 case VM_FRAME_MAGIC_BLOCK: return "block";
2590 case VM_FRAME_MAGIC_CLASS: return "class";
2591 case VM_FRAME_MAGIC_TOP: return "top";
2592 case VM_FRAME_MAGIC_CFUNC: return "cfunc";
2593 case VM_FRAME_MAGIC_IFUNC: return "ifunc";
2594 case VM_FRAME_MAGIC_EVAL: return "eval";
2595 case VM_FRAME_MAGIC_RESCUE: return "rescue";
2596 default:
2597 rb_bug("unknown frame");
2598 }
2599}
2600#endif
2601
2602static VALUE
2603frame_return_value(const struct vm_throw_data *err)
2604{
2605 if (THROW_DATA_P(err) &&
2606 THROW_DATA_STATE(err) == TAG_BREAK &&
2607 THROW_DATA_CONSUMED_P(err) == FALSE) {
2608 return THROW_DATA_VAL(err);
2609 }
2610 else {
2611 return Qnil;
2612 }
2613}
2614
2615#if 0
2616/* for debug */
2617static const char *
2618frame_name(const rb_control_frame_t *cfp)
2619{
2620 unsigned long type = VM_FRAME_TYPE(cfp);
2621#define C(t) if (type == VM_FRAME_MAGIC_##t) return #t
2622 C(METHOD);
2623 C(BLOCK);
2624 C(CLASS);
2625 C(TOP);
2626 C(CFUNC);
2627 C(PROC);
2628 C(IFUNC);
2629 C(EVAL);
2630 C(LAMBDA);
2631 C(RESCUE);
2632 C(DUMMY);
2633#undef C
2634 return "unknown";
2635}
2636#endif
2637
2638// cfp_returning_with_value:
2639// Whether cfp is the last frame in the unwinding process for a non-local return.
2640static void
2641hook_before_rewind(rb_execution_context_t *ec, bool cfp_returning_with_value, int state, struct vm_throw_data *err)
2642{
2643 if (state == TAG_RAISE && RBASIC(err)->klass == rb_eSysStackError) {
2644 return;
2645 }
2646 else {
2647 const rb_iseq_t *iseq = CFP_ISEQ(ec->cfp);
2648 rb_hook_list_t *local_hooks = NULL;
2649 unsigned int local_hooks_cnt = iseq->aux.exec.local_hooks_cnt;
2650 if (RB_UNLIKELY(local_hooks_cnt > 0)) {
2651 local_hooks = rb_iseq_local_hooks(iseq, rb_ec_ractor_ptr(ec), false);
2652 }
2653
2654 switch (VM_FRAME_TYPE(ec->cfp)) {
2655 case VM_FRAME_MAGIC_METHOD:
2656 RUBY_DTRACE_METHOD_RETURN_HOOK(ec, 0, 0);
2657 EXEC_EVENT_HOOK_AND_POP_FRAME(ec, RUBY_EVENT_RETURN, ec->cfp->self, 0, 0, 0, frame_return_value(err));
2658
2659 if (UNLIKELY(local_hooks && local_hooks->events & RUBY_EVENT_RETURN)) {
2660 rb_exec_event_hook_orig(ec, local_hooks, RUBY_EVENT_RETURN,
2661 ec->cfp->self, 0, 0, 0, frame_return_value(err), TRUE);
2662 }
2663
2664 THROW_DATA_CONSUMED_SET(err);
2665 break;
2666 case VM_FRAME_MAGIC_BLOCK:
2667 if (VM_FRAME_BMETHOD_P(ec->cfp)) {
2668 VALUE bmethod_return_value = frame_return_value(err);
2669 if (cfp_returning_with_value) {
2670 // Non-local return terminating at a BMETHOD control frame.
2671 bmethod_return_value = THROW_DATA_VAL(err);
2672 }
2673
2674
2675 EXEC_EVENT_HOOK_AND_POP_FRAME(ec, RUBY_EVENT_B_RETURN, ec->cfp->self, 0, 0, 0, bmethod_return_value);
2676 if (UNLIKELY(local_hooks && local_hooks->events & RUBY_EVENT_B_RETURN)) {
2677 rb_exec_event_hook_orig(ec, local_hooks, RUBY_EVENT_B_RETURN,
2678 ec->cfp->self, 0, 0, 0, bmethod_return_value, TRUE);
2679 }
2680
2681 const rb_callable_method_entry_t *me = rb_vm_frame_method_entry(ec->cfp);
2682
2683 EXEC_EVENT_HOOK_AND_POP_FRAME(ec, RUBY_EVENT_RETURN, ec->cfp->self,
2684 rb_vm_frame_method_entry(ec->cfp)->def->original_id,
2685 rb_vm_frame_method_entry(ec->cfp)->called_id,
2686 rb_vm_frame_method_entry(ec->cfp)->owner,
2687 bmethod_return_value);
2688
2689 VM_ASSERT(me->def->type == VM_METHOD_TYPE_BMETHOD);
2690 unsigned int local_hooks_cnt = me->def->body.bmethod.local_hooks_cnt;
2691 if (UNLIKELY(local_hooks_cnt > 0)) {
2692 local_hooks = rb_method_def_local_hooks(me->def, rb_ec_ractor_ptr(ec), false);
2693 if (local_hooks && local_hooks->events & RUBY_EVENT_RETURN) {
2694 rb_exec_event_hook_orig(ec, local_hooks, RUBY_EVENT_RETURN, ec->cfp->self,
2695 rb_vm_frame_method_entry(ec->cfp)->def->original_id,
2696 rb_vm_frame_method_entry(ec->cfp)->called_id,
2697 rb_vm_frame_method_entry(ec->cfp)->owner,
2698 bmethod_return_value, TRUE);
2699 }
2700 }
2701
2702 THROW_DATA_CONSUMED_SET(err);
2703 }
2704 else {
2705 EXEC_EVENT_HOOK_AND_POP_FRAME(ec, RUBY_EVENT_B_RETURN, ec->cfp->self, 0, 0, 0, frame_return_value(err));
2706 if (UNLIKELY(local_hooks && local_hooks->events & RUBY_EVENT_B_RETURN)) {
2707 rb_exec_event_hook_orig(ec, local_hooks, RUBY_EVENT_B_RETURN,
2708 ec->cfp->self, 0, 0, 0, frame_return_value(err), TRUE);
2709 }
2710 THROW_DATA_CONSUMED_SET(err);
2711 }
2712 break;
2713 case VM_FRAME_MAGIC_CLASS:
2714 EXEC_EVENT_HOOK_AND_POP_FRAME(ec, RUBY_EVENT_END, ec->cfp->self, 0, 0, 0, Qnil);
2715 break;
2716 }
2717 }
2718}
2719
2720/* evaluator body */
2721
2722/* finish
2723 VMe (h1) finish
2724 VM finish F1 F2
2725 cfunc finish F1 F2 C1
2726 rb_funcall finish F1 F2 C1
2727 VMe finish F1 F2 C1
2728 VM finish F1 F2 C1 F3
2729
2730 F1 - F3 : pushed by VM
2731 C1 : pushed by send insn (CFUNC)
2732
2733 struct CONTROL_FRAME {
2734 VALUE *pc; // cfp[0], program counter
2735 VALUE *sp; // cfp[1], stack pointer
2736 rb_iseq_t *iseq; // cfp[2], iseq
2737 VALUE self; // cfp[3], self
2738 const VALUE *ep; // cfp[4], env pointer
2739 const void *block_code; // cfp[5], block code
2740 };
2741
2742 struct rb_captured_block {
2743 VALUE self;
2744 VALUE *ep;
2745 union code;
2746 };
2747
2748 struct METHOD_ENV {
2749 VALUE param0;
2750 ...
2751 VALUE paramN;
2752 VALUE lvar1;
2753 ...
2754 VALUE lvarM;
2755 VALUE cref; // ep[-2]
2756 VALUE special; // ep[-1]
2757 VALUE flags; // ep[ 0] == lep[0]
2758 };
2759
2760 struct BLOCK_ENV {
2761 VALUE block_param0;
2762 ...
2763 VALUE block_paramN;
2764 VALUE block_lvar1;
2765 ...
2766 VALUE block_lvarM;
2767 VALUE cref; // ep[-2]
2768 VALUE special; // ep[-1]
2769 VALUE flags; // ep[ 0]
2770 };
2771
2772 struct CLASS_ENV {
2773 VALUE class_lvar0;
2774 ...
2775 VALUE class_lvarN;
2776 VALUE cref;
2777 VALUE prev_ep; // for frame jump
2778 VALUE flags;
2779 };
2780
2781 struct C_METHOD_CONTROL_FRAME {
2782 VALUE *pc; // 0
2783 VALUE *sp; // stack pointer
2784 rb_iseq_t *iseq; // cmi
2785 VALUE self; // ?
2786 VALUE *ep; // ep == lep
2787 void *code; //
2788 };
2789
2790 struct C_BLOCK_CONTROL_FRAME {
2791 VALUE *pc; // point only "finish" insn
2792 VALUE *sp; // sp
2793 rb_iseq_t *iseq; // ?
2794 VALUE self; //
2795 VALUE *ep; // ep
2796 void *code; //
2797 };
2798 */
2799
2800static inline VALUE
2801vm_exec_handle_exception(rb_execution_context_t *ec, enum ruby_tag_type state, VALUE errinfo);
2802static inline VALUE
2803vm_exec_loop(rb_execution_context_t *ec, enum ruby_tag_type state, struct rb_vm_tag *tag, VALUE result);
2804
2805// for non-Emscripten Wasm build, use vm_exec with optimized setjmp for runtime performance
2806#if defined(__wasm__) && !defined(__EMSCRIPTEN__)
2807
2808struct rb_vm_exec_context {
2809 rb_execution_context_t *const ec;
2810 struct rb_vm_tag *const tag;
2811
2812 VALUE result;
2813};
2814
2815static void
2816vm_exec_bottom_main(void *context)
2817{
2818 struct rb_vm_exec_context *ctx = context;
2819 rb_execution_context_t *ec = ctx->ec;
2820
2821 ctx->result = vm_exec_loop(ec, TAG_NONE, ctx->tag, vm_exec_core(ec));
2822}
2823
2824static void
2825vm_exec_bottom_rescue(void *context)
2826{
2827 struct rb_vm_exec_context *ctx = context;
2828 rb_execution_context_t *ec = ctx->ec;
2829
2830 ctx->result = vm_exec_loop(ec, rb_ec_tag_state(ec), ctx->tag, ec->errinfo);
2831}
2832#endif
2833
2834VALUE
2835vm_exec(rb_execution_context_t *ec)
2836{
2837 VALUE result = Qundef;
2838
2839 EC_PUSH_TAG(ec);
2840
2841 _tag.retval = Qnil;
2842
2843#if defined(__wasm__) && !defined(__EMSCRIPTEN__)
2844 struct rb_vm_exec_context ctx = {
2845 .ec = ec,
2846 .tag = &_tag,
2847 };
2848 struct rb_wasm_try_catch try_catch;
2849
2850 EC_REPUSH_TAG();
2851
2852 rb_wasm_try_catch_init(&try_catch, vm_exec_bottom_main, vm_exec_bottom_rescue, &ctx);
2853
2854 rb_wasm_try_catch_loop_run(&try_catch, &RB_VM_TAG_JMPBUF_GET(_tag.buf));
2855
2856 result = ctx.result;
2857#else
2858 enum ruby_tag_type state;
2859 if ((state = EC_EXEC_TAG()) == TAG_NONE) {
2860 if (UNDEF_P(result = jit_exec(ec))) {
2861 result = vm_exec_core(ec);
2862 }
2863 /* fallback to the VM */
2864 result = vm_exec_loop(ec, TAG_NONE, &_tag, result);
2865 }
2866 else {
2867 result = vm_exec_loop(ec, state, &_tag, ec->errinfo);
2868 }
2869#endif
2870
2871 EC_POP_TAG();
2872 return result;
2873}
2874
2875static inline VALUE
2876vm_exec_loop(rb_execution_context_t *ec, enum ruby_tag_type state,
2877 struct rb_vm_tag *tag, VALUE result)
2878{
2879 if (state == TAG_NONE) { /* no jumps, result is discarded */
2880 goto vm_loop_start;
2881 }
2882
2883 rb_ec_raised_reset(ec, RAISED_STACKOVERFLOW | RAISED_NOMEMORY);
2884 while (UNDEF_P(result = vm_exec_handle_exception(ec, state, result))) {
2885 // caught a jump, exec the handler. JIT code in jit_exec_exception()
2886 // may return Qundef to run remaining frames with vm_exec_core().
2887 if (UNDEF_P(result = jit_exec_exception(ec))) {
2888 result = vm_exec_core(ec);
2889 }
2890 vm_loop_start:
2891 VM_ASSERT(ec->tag == tag);
2892 /* when caught `throw`, `tag.state` is set. */
2893 if ((state = tag->state) == TAG_NONE) break;
2894 tag->state = TAG_NONE;
2895 }
2896
2897 return result;
2898}
2899
2900#if USE_ZJIT
2901// Materialize JITFrame-enabled CFP into interpreter-compatible CFP
2902static void
2903zjit_materialize_frames(const rb_execution_context_t *ec, rb_control_frame_t *cfp, bool materialize_target)
2904{
2905 if (!rb_zjit_enabled_p) return;
2906 const rb_control_frame_t *end_cfp = ec->tag->cfp;
2907 VM_ASSERT(cfp <= end_cfp);
2908
2909 while (true) {
2910 // If materialize_target is false, we skip materializing ec->tag->cfp.
2911 //
2912 // When JIT code calls a C function that does the same number of setjmps and
2913 // longjmps, e.g. rb_hash_aref, it calls zjit_materialize_frames but goes
2914 // back to the JIT code. In that case, we don't want to materialize the frame
2915 // and clear cfp->jit_return, which will still be used by the JIT code.
2916 //
2917 // When JIT code calls a C function that does more longjmps than setjmps,
2918 // it would not go back to the JIT code. So ec->tag->cfp should be materialized
2919 // in that case.
2920 if (cfp == end_cfp && !materialize_target) break;
2921
2922 if (CFP_ZJIT_FRAME_P(cfp)) {
2923 const zjit_jit_frame_t *jit_frame = CFP_ZJIT_FRAME(cfp);
2924 cfp->pc = jit_frame->pc;
2925 cfp->_iseq = (rb_iseq_t *)jit_frame->iseq;
2926 if (jit_frame->materialize_block_code) {
2927 cfp->block_code = NULL;
2928 }
2929
2930 // Materialize Ruby stack slots kept off the VM stack. On side exit,
2931 // the exiting frame is already written by compile_exit_save_state()
2932 // and skipped here after materialize_exit_trampoline clears its
2933 // jit_return, so this restores older ZJIT frames from stack maps.
2934 int32_t stack_size = (int32_t)jit_frame->stack_size;
2935 if (stack_size > 0) {
2936 VALUE *stack = cfp->sp;
2937 for (int32_t i = 0; i < stack_size; i++) {
2938 VALUE entry = jit_frame->stack[i];
2939 if (ZJIT_STACK_MAP_VREG_P(entry)) {
2940 // Decode a native stack slot offset generated by ZJIT's backend.
2941 // It's an offset from NATIVE_BASE_PTR, which is copied into
2942 // cfp->jit_return, to the encoded stack slot.
2943 stack--;
2944 *stack = ((VALUE *)cfp->jit_return)[-(ssize_t)ZJIT_STACK_MAP_VREG_INDEX(entry)];
2945 }
2946 else if (ZJIT_STACK_MAP_SKIP_P(entry)) {
2947 stack -= ZJIT_STACK_MAP_SKIP_SIZE(entry);
2948 }
2949 else if (ZJIT_STACK_MAP_BASE_PTR_P(entry)) {
2950 // This has to be the first code to align the write cursor for other entries
2951 RUBY_ASSERT_ALWAYS(0 == i, "base_ptr stack map code only makes sense at 0");
2952 VALUE *base_ptr = (VALUE *)((VALUE *)cfp->jit_return)[-(ssize_t)ZJIT_STACK_MAP_BASE_PTR_SLOT_INDEX(entry)];
2953 stack = base_ptr + ZJIT_STACK_MAP_BASE_PTR_STACK_SIZE(entry);
2954 }
2955 else {
2956 stack--;
2957 *stack = entry;
2958 }
2959 }
2960 }
2961 cfp->jit_return = 0;
2962 }
2963 if (end_cfp == cfp) break;
2964 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
2965 }
2966}
2967
2968void
2969rb_zjit_materialize_frames(const rb_execution_context_t *ec, rb_control_frame_t *cfp)
2970{
2971 zjit_materialize_frames(ec, cfp, true);
2972}
2973
2974void
2975rb_zjit_materialize_frames_for_longjmp(const rb_execution_context_t *ec, rb_control_frame_t *cfp)
2976{
2977 // A ZJIT frame active before the tag's setjmp is below it on the native
2978 // stack and survives longjmp. Materialize only the frames unwound above it.
2979 zjit_materialize_frames(ec, cfp, !ec->tag->zjit_frame_active);
2980}
2981#endif
2982
2983static inline VALUE
2984vm_exec_handle_exception(rb_execution_context_t *ec, enum ruby_tag_type state, VALUE errinfo)
2985{
2986 struct vm_throw_data *err = (struct vm_throw_data *)errinfo;
2987
2988 for (;;) {
2989 unsigned int i;
2990 const struct iseq_catch_table_entry *entry;
2991 const struct iseq_catch_table *ct;
2992 unsigned long epc, cont_pc, cont_sp;
2993 const rb_iseq_t *catch_iseq;
2994 VALUE type;
2995 const rb_control_frame_t *escape_cfp;
2996
2997 cont_pc = cont_sp = 0;
2998 catch_iseq = NULL;
2999
3000 while (CFP_PC(ec->cfp) == 0 || CFP_ISEQ(ec->cfp) == 0) {
3001 if (UNLIKELY(VM_FRAME_TYPE(ec->cfp) == VM_FRAME_MAGIC_CFUNC)) {
3002 EXEC_EVENT_HOOK_AND_POP_FRAME(ec, RUBY_EVENT_C_RETURN, ec->cfp->self,
3003 rb_vm_frame_method_entry(ec->cfp)->def->original_id,
3004 rb_vm_frame_method_entry(ec->cfp)->called_id,
3005 rb_vm_frame_method_entry(ec->cfp)->owner, Qnil);
3006 RUBY_DTRACE_CMETHOD_RETURN_HOOK(ec,
3007 rb_vm_frame_method_entry(ec->cfp)->owner,
3008 rb_vm_frame_method_entry(ec->cfp)->def->original_id);
3009 }
3010 rb_vm_pop_frame(ec);
3011 }
3012
3013 rb_control_frame_t *const cfp = ec->cfp;
3014 epc = CFP_PC(cfp) - ISEQ_BODY(CFP_ISEQ(cfp))->iseq_encoded;
3015
3016 escape_cfp = NULL;
3017 if (state == TAG_BREAK || state == TAG_RETURN) {
3018 escape_cfp = THROW_DATA_CATCH_FRAME(err);
3019
3020 if (cfp == escape_cfp) {
3021 if (state == TAG_RETURN) {
3022 if (!VM_FRAME_FINISHED_P(cfp)) {
3023 THROW_DATA_CATCH_FRAME_SET(err, cfp + 1);
3024 THROW_DATA_STATE_SET(err, state = TAG_BREAK);
3025 }
3026 else {
3027 ct = ISEQ_BODY(CFP_ISEQ(cfp))->catch_table;
3028 if (ct) for (i = 0; i < ct->size; i++) {
3029 entry = UNALIGNED_MEMBER_PTR(ct, entries[i]);
3030 if (entry->start < epc && entry->end >= epc) {
3031 if (entry->type == CATCH_TYPE_ENSURE) {
3032 catch_iseq = entry->iseq;
3033 cont_pc = entry->cont;
3034 cont_sp = entry->sp;
3035 break;
3036 }
3037 }
3038 }
3039 if (catch_iseq == NULL) {
3040 ec->errinfo = Qnil;
3041 THROW_DATA_CATCH_FRAME_SET(err, cfp + 1);
3042 // cfp == escape_cfp here so calling with cfp_returning_with_value = true
3043 hook_before_rewind(ec, true, state, err);
3044 rb_vm_pop_frame(ec);
3045 return THROW_DATA_VAL(err);
3046 }
3047 }
3048 /* through */
3049 }
3050 else {
3051 /* TAG_BREAK */
3052 *cfp->sp++ = THROW_DATA_VAL(err);
3053 ec->errinfo = Qnil;
3054 rb_zjit_materialize_frames(ec, cfp);
3055 return Qundef;
3056 }
3057 }
3058 }
3059
3060 if (state == TAG_RAISE) {
3061 ct = ISEQ_BODY(CFP_ISEQ(cfp))->catch_table;
3062 if (ct) for (i = 0; i < ct->size; i++) {
3063 entry = UNALIGNED_MEMBER_PTR(ct, entries[i]);
3064 if (entry->start < epc && entry->end >= epc) {
3065
3066 if (entry->type == CATCH_TYPE_RESCUE ||
3067 entry->type == CATCH_TYPE_ENSURE) {
3068 catch_iseq = entry->iseq;
3069 cont_pc = entry->cont;
3070 cont_sp = entry->sp;
3071 break;
3072 }
3073 }
3074 }
3075 }
3076 else if (state == TAG_RETRY) {
3077 ct = ISEQ_BODY(CFP_ISEQ(cfp))->catch_table;
3078 if (ct) for (i = 0; i < ct->size; i++) {
3079 entry = UNALIGNED_MEMBER_PTR(ct, entries[i]);
3080 if (entry->start < epc && entry->end >= epc) {
3081
3082 if (entry->type == CATCH_TYPE_ENSURE) {
3083 catch_iseq = entry->iseq;
3084 cont_pc = entry->cont;
3085 cont_sp = entry->sp;
3086 break;
3087 }
3088 else if (entry->type == CATCH_TYPE_RETRY) {
3089 const rb_control_frame_t *escape_cfp;
3090 escape_cfp = THROW_DATA_CATCH_FRAME(err);
3091 if (cfp == escape_cfp) {
3092 rb_zjit_materialize_frames(ec, cfp);
3093 cfp->pc = ISEQ_BODY(CFP_ISEQ(cfp))->iseq_encoded + entry->cont;
3094 ec->errinfo = Qnil;
3095 return Qundef;
3096 }
3097 }
3098 }
3099 }
3100 }
3101 else if ((state == TAG_BREAK && !escape_cfp) ||
3102 (state == TAG_REDO) ||
3103 (state == TAG_NEXT)) {
3104 type = (const enum rb_catch_type[TAG_MASK]) {
3105 [TAG_BREAK] = CATCH_TYPE_BREAK,
3106 [TAG_NEXT] = CATCH_TYPE_NEXT,
3107 [TAG_REDO] = CATCH_TYPE_REDO,
3108 /* otherwise = dontcare */
3109 }[state];
3110
3111 ct = ISEQ_BODY(CFP_ISEQ(cfp))->catch_table;
3112 if (ct) for (i = 0; i < ct->size; i++) {
3113 entry = UNALIGNED_MEMBER_PTR(ct, entries[i]);
3114
3115 if (entry->start < epc && entry->end >= epc) {
3116 if (entry->type == CATCH_TYPE_ENSURE) {
3117 catch_iseq = entry->iseq;
3118 cont_pc = entry->cont;
3119 cont_sp = entry->sp;
3120 break;
3121 }
3122 else if (entry->type == type) {
3123 rb_zjit_materialize_frames(ec, cfp);
3124 cfp->pc = ISEQ_BODY(CFP_ISEQ(cfp))->iseq_encoded + entry->cont;
3125 cfp->sp = vm_base_ptr(cfp) + entry->sp;
3126
3127 if (state != TAG_REDO) {
3128 *cfp->sp++ = THROW_DATA_VAL(err);
3129 }
3130 ec->errinfo = Qnil;
3131 VM_ASSERT(ec->tag->state == TAG_NONE);
3132 return Qundef;
3133 }
3134 }
3135 }
3136 }
3137 else {
3138 ct = ISEQ_BODY(CFP_ISEQ(cfp))->catch_table;
3139 if (ct) for (i = 0; i < ct->size; i++) {
3140 entry = UNALIGNED_MEMBER_PTR(ct, entries[i]);
3141 if (entry->start < epc && entry->end >= epc) {
3142
3143 if (entry->type == CATCH_TYPE_ENSURE) {
3144 catch_iseq = entry->iseq;
3145 cont_pc = entry->cont;
3146 cont_sp = entry->sp;
3147 break;
3148 }
3149 }
3150 }
3151 }
3152
3153 if (catch_iseq != NULL) { /* found catch table */
3154 /* enter catch scope */
3155 const int arg_size = 1;
3156
3157 rb_iseq_check(catch_iseq);
3158 rb_zjit_materialize_frames(ec, cfp); // vm_base_ptr looks at cfp->_iseq
3159 cfp->sp = vm_base_ptr(cfp) + cont_sp;
3160 cfp->pc = ISEQ_BODY(CFP_ISEQ(cfp))->iseq_encoded + cont_pc;
3161
3162 /* push block frame */
3163 cfp->sp[0] = (VALUE)err;
3164 vm_push_frame(ec, catch_iseq, VM_FRAME_MAGIC_RESCUE,
3165 cfp->self,
3166 VM_GUARDED_PREV_EP(cfp->ep),
3167 0, /* cref or me */
3168 ISEQ_BODY(catch_iseq)->iseq_encoded,
3169 cfp->sp + arg_size /* push value */,
3170 ISEQ_BODY(catch_iseq)->local_table_size - arg_size,
3171 ISEQ_BODY(catch_iseq)->stack_max);
3172
3173 state = 0;
3174 ec->tag->state = TAG_NONE;
3175 ec->errinfo = Qnil;
3176
3177 return Qundef;
3178 }
3179 else {
3180 hook_before_rewind(ec, (cfp == escape_cfp), state, err);
3181
3182 if (VM_FRAME_FINISHED_P(ec->cfp)) {
3183 rb_vm_pop_frame(ec);
3184 ec->errinfo = (VALUE)err;
3185 rb_vm_tag_jmpbuf_deinit(&ec->tag->buf);
3186 ec->tag = ec->tag->prev;
3187 EC_JUMP_TAG(ec, state);
3188 }
3189 else {
3190 rb_vm_pop_frame(ec);
3191 }
3192 }
3193 }
3194}
3195
3196/* misc */
3197
3198VALUE
3199rb_iseq_eval(const rb_iseq_t *iseq, const rb_box_t *box)
3200{
3201 rb_execution_context_t *ec = GET_EC();
3202 VALUE val;
3203 vm_set_top_stack(ec, iseq, box);
3204 val = vm_exec(ec);
3205 return val;
3206}
3207
3208VALUE
3209rb_iseq_eval_main(const rb_iseq_t *iseq)
3210{
3211 rb_execution_context_t *ec = GET_EC();
3212 VALUE val;
3213 vm_set_main_stack(ec, iseq);
3214 val = vm_exec(ec);
3215 return val;
3216}
3217
3218int
3219rb_vm_control_frame_id_and_class(const rb_control_frame_t *cfp, ID *idp, ID *called_idp, VALUE *klassp)
3220{
3221 const rb_callable_method_entry_t *me = rb_vm_frame_method_entry(cfp);
3222
3223 if (me) {
3224 if (idp) *idp = me->def->original_id;
3225 if (called_idp) *called_idp = me->called_id;
3226 if (klassp) *klassp = me->owner;
3227 return TRUE;
3228 }
3229 else {
3230 return FALSE;
3231 }
3232}
3233
3234int
3235rb_ec_frame_method_id_and_class(const rb_execution_context_t *ec, ID *idp, ID *called_idp, VALUE *klassp)
3236{
3237 return rb_vm_control_frame_id_and_class(ec->cfp, idp, called_idp, klassp);
3238}
3239
3240int
3242{
3243 return rb_ec_frame_method_id_and_class(GET_EC(), idp, 0, klassp);
3244}
3245
3246VALUE
3247rb_vm_call_cfunc(VALUE recv, VALUE (*func)(VALUE), VALUE arg,
3248 VALUE block_handler, VALUE filename)
3249{
3250 rb_execution_context_t *ec = GET_EC();
3251 const rb_control_frame_t *reg_cfp = ec->cfp;
3252 const rb_iseq_t *iseq = rb_iseq_new(Qnil, filename, filename, Qnil, 0, ISEQ_TYPE_TOP);
3253 const rb_box_t *box = rb_current_box();
3254 VALUE val;
3255
3256 vm_push_frame(ec, iseq, VM_FRAME_MAGIC_TOP | VM_ENV_FLAG_LOCAL | VM_FRAME_FLAG_FINISH,
3257 recv, GC_GUARDED_PTR(box),
3258 (VALUE)vm_cref_new_toplevel(ec), /* cref or me */
3259 0, reg_cfp->sp, 0, 0);
3260
3261 val = (*func)(arg);
3262
3263 rb_vm_pop_frame(ec);
3264 return val;
3265}
3266
3267/* Ruby::Box */
3268
3269VALUE
3270rb_vm_call_cfunc_in_box(VALUE recv, VALUE (*func)(VALUE, VALUE), VALUE arg1, VALUE arg2,
3271 VALUE filename, const rb_box_t *box)
3272{
3273 rb_execution_context_t *ec = GET_EC();
3274 const rb_control_frame_t *reg_cfp = ec->cfp;
3275 const rb_iseq_t *iseq = rb_iseq_new(Qnil, filename, filename, Qnil, 0, ISEQ_TYPE_TOP);
3276 VALUE val;
3277
3278 vm_push_frame(ec, iseq, VM_FRAME_MAGIC_TOP | VM_ENV_FLAG_LOCAL | VM_FRAME_FLAG_FINISH,
3279 recv, GC_GUARDED_PTR(box),
3280 (VALUE)vm_cref_new_toplevel(ec), /* cref or me */
3281 0, reg_cfp->sp, 0, 0);
3282
3283 val = (*func)(arg1, arg2);
3284
3285 rb_vm_pop_frame(ec);
3286 return val;
3287}
3288
3289void
3290rb_vm_frame_flag_set_box_require(const rb_execution_context_t *ec)
3291{
3292 VM_ASSERT(rb_box_available());
3293 VM_ENV_FLAGS_SET(ec->cfp->ep, VM_FRAME_FLAG_BOX_REQUIRE);
3294}
3295
3296static const rb_box_t *
3297current_box_on_cfp(const rb_execution_context_t *ec, const rb_control_frame_t *cfp)
3298{
3300 const rb_box_t *box;
3301 const VALUE *lep = VM_EP_RUBY_LEP(ec, cfp);
3302 VM_BOX_ASSERT(lep, "lep should be valid");
3303 VM_BOX_ASSERT(rb_box_available(), "box should be available here");
3304
3305 if (VM_ENV_FRAME_TYPE_P(lep, VM_FRAME_MAGIC_METHOD) || VM_ENV_FRAME_TYPE_P(lep, VM_FRAME_MAGIC_CFUNC)) {
3306 cme = check_method_entry(lep[VM_ENV_DATA_INDEX_ME_CREF], TRUE);
3307 VM_BOX_ASSERT(cme, "cme should be valid");
3308 VM_BOX_ASSERT(cme->def, "cme->def shold be valid");
3309 return cme->def->box;
3310 }
3311 else if (VM_ENV_FRAME_TYPE_P(lep, VM_FRAME_MAGIC_TOP) || VM_ENV_FRAME_TYPE_P(lep, VM_FRAME_MAGIC_CLASS)) {
3312 VM_BOX_ASSERT(VM_ENV_LOCAL_P(lep), "lep should be local on MAGIC_TOP or MAGIC_CLASS frames");
3313 return VM_ENV_BOX(lep);
3314 }
3315 else if (VM_ENV_FRAME_TYPE_P(lep, VM_FRAME_MAGIC_DUMMY)) {
3316 // No valid local ep found (just after process boot?)
3317 // return the root box (the only valid box) until the main is initialized
3318 box = rb_main_box();
3319 if (box)
3320 return box;
3321 return rb_root_box();
3322 }
3323 else {
3324 VM_BOX_CRASHED();
3325 rb_bug("BUG: Local ep without cme/box, flags: %08lX", (unsigned long)lep[VM_ENV_DATA_INDEX_FLAGS]);
3326 }
3328}
3329
3330const rb_box_t *
3331rb_vm_current_box(const rb_execution_context_t *ec)
3332{
3333 return current_box_on_cfp(ec, ec->cfp);
3334}
3335
3336static const rb_control_frame_t *
3337find_loader_control_frame(const rb_execution_context_t *ec, const rb_control_frame_t *cfp, const rb_control_frame_t *end_cfp)
3338{
3339 while (RUBY_VM_VALID_CONTROL_FRAME_P(cfp, end_cfp)) {
3340 if (!VM_ENV_FRAME_TYPE_P(cfp->ep, VM_FRAME_MAGIC_CFUNC))
3341 break;
3342 if (!BOX_MASTER_P(current_box_on_cfp(ec, cfp)))
3343 break;
3344 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
3345 }
3346 VM_ASSERT(RUBY_VM_VALID_CONTROL_FRAME_P(cfp, end_cfp));
3347 return cfp;
3348}
3349
3350const rb_box_t *
3351rb_vm_loading_box(const rb_execution_context_t *ec)
3352{
3353 const rb_control_frame_t *cfp, *current_cfp, *end_cfp;
3354
3355 if (!rb_box_available() || !ec)
3356 return rb_root_box();
3357
3358 cfp = ec->cfp;
3359 current_cfp = cfp;
3360 end_cfp = RUBY_VM_END_CONTROL_FRAME(ec);
3361
3362 while (RUBY_VM_VALID_CONTROL_FRAME_P(cfp, end_cfp)) {
3363 if (VM_ENV_FLAGS(cfp->ep, VM_FRAME_FLAG_BOX_REQUIRE)) {
3364 if (RTEST(cfp->self) && BOX_OBJ_P(cfp->self)) {
3365 // Box#require, #require_relative, #load
3366 return rb_get_box_t(cfp->self);
3367 }
3368 // Kernel#require, #require_relative, #load
3369 cfp = find_loader_control_frame(ec, cfp, end_cfp);
3370 return current_box_on_cfp(ec, cfp);
3371 }
3372 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
3373 }
3374 // no require/load with explicit boxes.
3375 return current_box_on_cfp(ec, current_cfp);
3376}
3377
3378/* vm */
3379
3380void
3381rb_vm_update_references(void *ptr)
3382{
3383 if (ptr) {
3384 rb_vm_t *vm = ptr;
3385
3386 vm->self = rb_gc_location(vm->self);
3387 vm->orig_progname = rb_gc_location(vm->orig_progname);
3388 vm->cc_refinement_set = rb_gc_location(vm->cc_refinement_set);
3389
3390 if (vm->root_box)
3391 rb_box_gc_update_references(vm->root_box);
3392 if (vm->main_box)
3393 rb_box_gc_update_references(vm->main_box);
3394
3395 rb_gc_update_values(RUBY_NSIG, vm->trap_list.cmd);
3396
3397 if (vm->coverages) {
3398 vm->coverages = rb_gc_location(vm->coverages);
3399 vm->cme2counter = rb_gc_location(vm->cme2counter);
3400 vm->me_set = rb_gc_location(vm->me_set);
3401 }
3402 }
3403}
3404
3405void
3406rb_vm_each_stack_value(void *ptr, void (*cb)(VALUE, void*), void *ctx)
3407{
3408 if (ptr) {
3409 rb_vm_t *vm = ptr;
3410 rb_ractor_t *r = 0;
3411 ccan_list_for_each(&vm->ractor.set, r, vmlr_node) {
3412 VM_ASSERT(rb_ractor_status_p(r, ractor_blocking) ||
3413 rb_ractor_status_p(r, ractor_running));
3414 if (r->threads.cnt > 0) {
3415 rb_thread_t *th = 0;
3416 ccan_list_for_each(&r->threads.set, th, lt_node) {
3417 VM_ASSERT(th != NULL);
3418 rb_execution_context_t * ec = th->ec;
3419 if (ec->vm_stack) {
3420 VALUE *p = ec->vm_stack;
3421 VALUE *sp = ec->cfp->sp;
3422 while (p < sp) {
3423 if (!RB_SPECIAL_CONST_P(*p)) {
3424 cb(*p, ctx);
3425 }
3426 p++;
3427 }
3428 }
3429 }
3430 }
3431 }
3432 }
3433}
3434
3435static enum rb_id_table_iterator_result
3436vm_mark_negative_cme(VALUE val, void *dmy)
3437{
3438 rb_gc_mark(val);
3439 return ID_TABLE_CONTINUE;
3440}
3441
3442
3443void
3444rb_vm_mark(void *ptr)
3445{
3446 RUBY_MARK_ENTER("vm");
3447 RUBY_GC_INFO("-------------------------------------------------\n");
3448 if (ptr) {
3449 rb_vm_t *vm = ptr;
3450 rb_ractor_t *r = 0;
3451 long i;
3452
3453 ccan_list_for_each(&vm->ractor.set, r, vmlr_node) {
3454 // ractor.set only contains blocking or running ractors
3455 VM_ASSERT(rb_ractor_status_p(r, ractor_blocking) ||
3456 rb_ractor_status_p(r, ractor_running));
3457 rb_gc_mark(rb_ractor_self(r));
3458 }
3459
3460 rb_gc_mark_movable(vm->self);
3461
3462 if (vm->root_box) {
3463 rb_box_entry_mark(vm->root_box);
3464 }
3465 if (vm->main_box) {
3466 rb_box_entry_mark(vm->main_box);
3467 }
3468
3469 rb_gc_mark_movable(vm->orig_progname);
3470 rb_gc_mark_movable(vm->coverages);
3471 rb_gc_mark_movable(vm->cme2counter);
3472 rb_gc_mark_movable(vm->me_set);
3473 rb_gc_mark_movable(vm->cc_refinement_set);
3474
3475 rb_hook_list_mark(&vm->global_hooks);
3476
3477 rb_id_table_foreach_values(&vm->negative_cme_table, vm_mark_negative_cme, NULL);
3478 rb_mark_tbl_no_pin(&vm->overloaded_cme_table);
3479 for (i=0; i<VM_GLOBAL_CC_CACHE_TABLE_SIZE; i++) {
3480 const struct rb_callcache *cc = vm->global_cc_cache_table[i];
3481
3482 if (cc != NULL) {
3483 if (!vm_cc_invalidated_p(cc)) {
3484 rb_gc_mark((VALUE)cc);
3485 }
3486 else {
3487 vm->global_cc_cache_table[i] = NULL;
3488 }
3489 }
3490 }
3491 }
3492
3493 RUBY_MARK_LEAVE("vm");
3494}
3495
3496#undef rb_vm_register_special_exception
3497void
3498rb_vm_register_special_exception_str(enum ruby_special_exceptions sp, VALUE cls, VALUE mesg)
3499{
3500 rb_vm_t *vm = GET_VM();
3501 VALUE exc = rb_exc_new3(cls, rb_obj_freeze(mesg));
3502 OBJ_FREEZE(exc);
3503 ((VALUE *)vm->special_exceptions)[sp] = exc;
3504 rb_vm_register_global_object(exc);
3505}
3506
3507void rb_objspace_free_objects(void *objspace);
3508
3509int
3511{
3512 {
3513 // wait for native threads still winding down a dead coroutine: their
3514 // reclaim frees through the objspace this function is about to
3515 // destroy (see coroutine_thread_terminated)
3516 void rb_thread_sched_wait_winding(rb_vm_t *vm);
3517 rb_thread_sched_wait_winding(vm);
3518 }
3519
3520 RUBY_FREE_ENTER("vm");
3521 ruby_vm_during_cleanup = true;
3522 rb_gc_stash_cleanup_objspace();
3523
3524 if (vm) {
3525 rb_thread_t *th = vm->ractor.main_thread;
3526
3527 if (rb_free_at_exit) {
3528 rb_free_encoded_insn_data();
3529 rb_free_global_enc_table();
3530 rb_free_loaded_builtin_table();
3531 rb_free_global_symbol_table();
3532
3533 rb_free_shared_fiber_pool();
3534 rb_free_transcoder_table();
3535 rb_free_vm_opt_tables();
3536 rb_free_warning();
3537 rb_free_rb_global_tbl();
3538
3539 rb_id_table_free_items(&vm->negative_cme_table);
3540 st_free_embedded_table(&vm->overloaded_cme_table);
3541
3542 // TODO: Is this ignorable for classext->m_tbl ?
3543 // rb_id_table_free(RCLASS(rb_mRubyVMFrozenCore)->m_tbl);
3544
3545 st_free_embedded_table(&vm->static_ext_inits);
3546
3547 rb_id_table_free_items(&vm->constant_cache);
3548 set_free_embedded_table(&vm->unused_block_warning_table);
3549
3550 rb_thread_free_native_thread(th);
3551
3552#ifndef HAVE_SETPROCTITLE
3553 ruby_free_proctitle();
3554#endif
3555 }
3556 else {
3557 rb_fiber_reset_root_local_storage(th);
3558 thread_free(th);
3559 }
3560
3561 void *objspace = vm->ractor.main_ractor ? vm->ractor.main_ractor->objspace : NULL;
3562
3563 rb_vm_living_threads_init(vm);
3564 ruby_vm_run_at_exit_hooks(vm);
3565 st_free_embedded_table(&vm->ci_table);
3566 RB_ALTSTACK_FREE(vm->main_altstack);
3567
3568 if (objspace) {
3569 if (rb_free_at_exit) {
3570 rb_objspace_free_objects(objspace);
3571 rb_free_generic_fields_tbl_();
3572 rb_free_default_rand_key();
3573 }
3574 rb_objspace_free(objspace);
3575 }
3576 rb_native_mutex_destroy(&vm->workqueue_lock);
3577 rb_native_mutex_destroy(&vm->once_lock);
3578 rb_native_cond_destroy(&vm->once_cond);
3579 /* after freeing objspace, you *can't* use ruby_xfree() */
3580 ruby_current_vm_ptr = NULL;
3581
3582 if (rb_free_at_exit) {
3583#if USE_YJIT
3584 rb_yjit_free_at_exit();
3585#endif
3586 }
3587 }
3588 RUBY_FREE_LEAVE("vm");
3589 return 0;
3590}
3591
3592size_t rb_vm_memsize_workqueue(struct ccan_list_head *workqueue); // vm_trace.c
3593
3594// Used for VM memsize reporting. Returns the size of the at_exit list by
3595// looping through the linked list and adding up the size of the structs.
3596static enum rb_id_table_iterator_result
3597vm_memsize_constant_cache_i(ID id, VALUE ics, void *size)
3598{
3599 *((size_t *) size) += rb_set_memsize((set_table *) ics);
3600 return ID_TABLE_CONTINUE;
3601}
3602
3603// Returns a size_t representing the memory footprint of the VM's constant
3604// cache, which is the memsize of the table as well as the memsize of all of the
3605// nested tables.
3606static size_t
3607vm_memsize_constant_cache(void)
3608{
3609 rb_vm_t *vm = GET_VM();
3610 size_t size = rb_id_table_memsize(&vm->constant_cache) - sizeof(struct rb_id_table);
3611
3612 rb_id_table_foreach(&vm->constant_cache, vm_memsize_constant_cache_i, &size);
3613 return size;
3614}
3615
3616static size_t
3617vm_memsize_at_exit_list(rb_at_exit_list *at_exit)
3618{
3619 size_t size = 0;
3620
3621 while (at_exit) {
3622 size += sizeof(rb_at_exit_list);
3623 at_exit = at_exit->next;
3624 }
3625
3626 return size;
3627}
3628
3629// Used for VM memsize reporting. Returns the size of the builtin function
3630// table if it has been defined.
3631static size_t
3632vm_memsize_builtin_function_table(const struct rb_builtin_function *builtin_function_table)
3633{
3634 return builtin_function_table == NULL ? 0 : sizeof(struct rb_builtin_function);
3635}
3636
3637// Reports the memsize of the VM struct object and the structs that are
3638// associated with it.
3639static size_t
3640vm_memsize(const void *ptr)
3641{
3642 rb_vm_t *vm = GET_VM();
3643
3644 return (
3645 sizeof(rb_vm_t) +
3646 rb_vm_memsize_postponed_job_queue() +
3647 rb_vm_memsize_workqueue(&vm->workqueue) +
3648 vm_memsize_at_exit_list(vm->at_exit) +
3649 (rb_st_memsize(&vm->ci_table) - sizeof(struct st_table)) +
3650 vm_memsize_builtin_function_table(vm->builtin_function_table) +
3651 (rb_id_table_memsize(&vm->negative_cme_table) - sizeof(struct rb_id_table)) +
3652 (rb_st_memsize(&vm->overloaded_cme_table) - sizeof(struct st_table)) +
3653 vm_memsize_constant_cache()
3654 );
3655
3656 // TODO
3657 // struct { struct ccan_list_head set; } ractor;
3658 // void *main_altstack; #ifdef USE_SIGALTSTACK
3659 // struct rb_objspace *objspace;
3660}
3661
3662const rb_data_type_t ruby_vm_data_type = {
3663 "VM",
3664 {0, 0, vm_memsize,},
3665 0, 0, RUBY_TYPED_THREAD_SAFE_FREE
3666};
3667
3668#define vm_data_type ruby_vm_data_type
3669
3670static VALUE
3671vm_default_params(void)
3672{
3673 rb_vm_t *vm = GET_VM();
3674 VALUE result = rb_hash_new_with_size(4);
3675#define SET(name) rb_hash_aset(result, ID2SYM(rb_intern(#name)), SIZET2NUM(vm->default_params.name));
3676 SET(thread_vm_stack_size);
3677 SET(thread_machine_stack_size);
3678 SET(fiber_vm_stack_size);
3679 SET(fiber_machine_stack_size);
3680#undef SET
3681 rb_obj_freeze(result);
3682 return result;
3683}
3684
3685static size_t
3686get_param(const char *name, size_t default_value, size_t min_value)
3687{
3688 const char *envval;
3689 size_t result = default_value;
3690 if ((envval = getenv(name)) != 0) {
3691 long val = atol(envval);
3692 if (val < (long)min_value) {
3693 val = (long)min_value;
3694 }
3695 result = (size_t)(((val -1 + RUBY_VM_SIZE_ALIGN) / RUBY_VM_SIZE_ALIGN) * RUBY_VM_SIZE_ALIGN);
3696 }
3697 if (0) ruby_debug_printf("%s: %"PRIuSIZE"\n", name, result); /* debug print */
3698
3699 return result;
3700}
3701
3702static void
3703check_machine_stack_size(size_t *sizep)
3704{
3705#ifdef PTHREAD_STACK_MIN
3706 size_t size = *sizep;
3707#endif
3708
3709#ifdef PTHREAD_STACK_MIN
3710 if (size < (size_t)PTHREAD_STACK_MIN) {
3711 *sizep = (size_t)PTHREAD_STACK_MIN * 2;
3712 }
3713#endif
3714}
3715
3716static void
3717vm_default_params_setup(rb_vm_t *vm)
3718{
3719 vm->default_params.thread_vm_stack_size =
3720 get_param("RUBY_THREAD_VM_STACK_SIZE",
3721 RUBY_VM_THREAD_VM_STACK_SIZE,
3722 RUBY_VM_THREAD_VM_STACK_SIZE_MIN);
3723
3724 vm->default_params.thread_machine_stack_size =
3725 get_param("RUBY_THREAD_MACHINE_STACK_SIZE",
3726 RUBY_VM_THREAD_MACHINE_STACK_SIZE,
3727 RUBY_VM_THREAD_MACHINE_STACK_SIZE_MIN);
3728
3729 vm->default_params.fiber_vm_stack_size =
3730 get_param("RUBY_FIBER_VM_STACK_SIZE",
3731 RUBY_VM_FIBER_VM_STACK_SIZE,
3732 RUBY_VM_FIBER_VM_STACK_SIZE_MIN);
3733
3734 vm->default_params.fiber_machine_stack_size =
3735 get_param("RUBY_FIBER_MACHINE_STACK_SIZE",
3736 RUBY_VM_FIBER_MACHINE_STACK_SIZE,
3737 RUBY_VM_FIBER_MACHINE_STACK_SIZE_MIN);
3738
3739 /* environment dependent check */
3740 check_machine_stack_size(&vm->default_params.thread_machine_stack_size);
3741 check_machine_stack_size(&vm->default_params.fiber_machine_stack_size);
3742}
3743
3744static void
3745vm_init2(rb_vm_t *vm)
3746{
3747 rb_vm_living_threads_init(vm);
3748 vm->thread_report_on_exception = 1;
3749 vm->src_encoding_index = -1;
3750
3751 vm_default_params_setup(vm);
3752}
3753
3754void
3755rb_execution_context_update(rb_execution_context_t *ec)
3756{
3757 /* update VM stack */
3758 if (ec->vm_stack) {
3759 long i;
3760 VM_ASSERT(ec->cfp);
3761 VALUE *p = ec->vm_stack;
3762 VALUE *sp = ec->cfp->sp;
3763 rb_control_frame_t *cfp = ec->cfp;
3764 rb_control_frame_t *limit_cfp = (void *)(ec->vm_stack + ec->vm_stack_size);
3765
3766 // ZJIT leaves uninitialized slots on the VM stack, so we cannot
3767 // safely use rb_gc_location on such slots.
3768 if (!rb_zjit_enabled_p) {
3769 for (i = 0; i < (long)(sp - p); i++) {
3770 VALUE ref = p[i];
3771 VALUE update = rb_gc_location(ref);
3772 if (ref != update) {
3773 p[i] = update;
3774 }
3775 }
3776 }
3777
3778 while (cfp != limit_cfp) {
3779 const VALUE *ep = cfp->ep;
3780 cfp->self = rb_gc_location(cfp->self);
3781 if (CFP_ZJIT_FRAME_P(cfp)) {
3782 const zjit_jit_frame_t *jit_frame = CFP_ZJIT_FRAME(cfp);
3783 rb_zjit_jit_frame_update_references((zjit_jit_frame_t *)jit_frame);
3784 // materialize_block_code means cfp->block_code is lazy and may
3785 // still contain stale data from a previous frame. Otherwise it
3786 // was initialized by ZJIT and may have been written later by
3787 // vm_caller_setup_arg_block (ISEQ frames) or rb_iterate0 (C frames).
3788 if (!jit_frame->materialize_block_code) {
3789 cfp->block_code = (void *)rb_gc_location((VALUE)cfp->block_code);
3790 }
3791 }
3792 else {
3793 cfp->_iseq = (rb_iseq_t *)rb_gc_location((VALUE)cfp->_iseq);
3794 cfp->block_code = (void *)rb_gc_location((VALUE)cfp->block_code);
3795 }
3796
3797 if (!VM_ENV_LOCAL_P(ep)) {
3798 const VALUE *prev_ep = VM_ENV_PREV_EP(ep);
3799 if (VM_ENV_FLAGS(prev_ep, VM_ENV_FLAG_ESCAPED)) {
3800 VM_FORCE_WRITE(&prev_ep[VM_ENV_DATA_INDEX_ENV], rb_gc_location(prev_ep[VM_ENV_DATA_INDEX_ENV]));
3801 }
3802
3803 if (VM_ENV_FLAGS(ep, VM_ENV_FLAG_ESCAPED)) {
3804 VM_FORCE_WRITE(&ep[VM_ENV_DATA_INDEX_ENV], rb_gc_location(ep[VM_ENV_DATA_INDEX_ENV]));
3805 VM_FORCE_WRITE(&ep[VM_ENV_DATA_INDEX_ME_CREF], rb_gc_location(ep[VM_ENV_DATA_INDEX_ME_CREF]));
3806 }
3807 }
3808
3809 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
3810 }
3811 }
3812
3813 ec->storage = rb_gc_location(ec->storage);
3814
3815 ec->gen_fields_cache.obj = rb_gc_location(ec->gen_fields_cache.obj);
3816 ec->gen_fields_cache.fields_obj = rb_gc_location(ec->gen_fields_cache.fields_obj);
3817}
3818
3819static enum rb_id_table_iterator_result
3820mark_local_storage_i(VALUE local, void *data)
3821{
3822 rb_gc_mark(local);
3823 return ID_TABLE_CONTINUE;
3824}
3825
3826void
3827rb_execution_context_mark(const rb_execution_context_t *ec)
3828{
3829 /* mark VM stack */
3830 if (ec->vm_stack) {
3831 VM_ASSERT(ec->cfp);
3832 VALUE *p = ec->vm_stack;
3833 VALUE *sp = ec->cfp->sp;
3834 rb_control_frame_t *cfp = ec->cfp;
3835 rb_control_frame_t *limit_cfp = (void *)(ec->vm_stack + ec->vm_stack_size);
3836
3837 for (long i = 0; i < (long)(sp - p); i++) {
3838 if (rb_zjit_enabled_p) {
3839 // ZJIT leaves uninitialized slots on the VM stack, so we need
3840 // to mark such slots conservatively.
3841 rb_gc_mark_maybe(p[i]);
3842 }
3843 else {
3844 rb_gc_mark_movable(p[i]);
3845 }
3846 }
3847
3848 while (cfp != limit_cfp) {
3849 const VALUE *ep = cfp->ep;
3850 VM_ASSERT(!!VM_ENV_FLAGS(ep, VM_ENV_FLAG_ESCAPED) == vm_ep_in_heap_p_(ec, ep));
3851
3852 rb_gc_mark_movable(cfp->self);
3853 if (CFP_ZJIT_FRAME_P(cfp)) {
3854 const zjit_jit_frame_t *jit_frame = CFP_ZJIT_FRAME(cfp);
3855 rb_gc_mark_movable((VALUE)jit_frame->iseq);
3856 // materialize_block_code means cfp->block_code is lazy and may
3857 // still contain stale data from a previous frame. Otherwise it
3858 // was initialized by ZJIT and may have been written later by
3859 // vm_caller_setup_arg_block (ISEQ frames) or rb_iterate0 (C frames).
3860 if (!jit_frame->materialize_block_code) {
3861 rb_gc_mark_movable((VALUE)cfp->block_code);
3862 }
3863 }
3864 else {
3865 rb_gc_mark_movable((VALUE)cfp->_iseq);
3866 rb_gc_mark_movable((VALUE)cfp->block_code);
3867 }
3868
3869 if (VM_ENV_LOCAL_P(ep) && VM_ENV_BOXED_P(ep)) {
3870 const rb_box_t *box = VM_ENV_BOX(ep);
3871 if (BOX_USER_P(box)) {
3872 rb_gc_mark_movable(box->box_object);
3873 }
3874 }
3875
3876 if (!VM_ENV_LOCAL_P(ep)) {
3877 const VALUE *prev_ep = VM_ENV_PREV_EP(ep);
3878 if (VM_ENV_FLAGS(prev_ep, VM_ENV_FLAG_ESCAPED)) {
3879 rb_gc_mark_movable(prev_ep[VM_ENV_DATA_INDEX_ENV]);
3880 }
3881
3882 if (VM_ENV_FLAGS(ep, VM_ENV_FLAG_ESCAPED)) {
3883 rb_gc_mark_movable(ep[VM_ENV_DATA_INDEX_ENV]);
3884 rb_gc_mark(ep[VM_ENV_DATA_INDEX_ME_CREF]);
3885 }
3886 }
3887
3888 cfp = RUBY_VM_PREVIOUS_CONTROL_FRAME(cfp);
3889 }
3890 }
3891
3892 /* mark machine stack */
3893 if (ec->machine.stack_start && ec->machine.stack_end &&
3894 /* marked for current ec at the first stage of marking */
3895 ec != rb_gc_get_ec()) {
3896 rb_gc_mark_machine_context(ec);
3897 }
3898
3899 rb_gc_mark(ec->errinfo);
3900 rb_gc_mark(ec->root_svar);
3901 if (ec->local_storage) {
3902 rb_id_table_foreach_values(ec->local_storage, mark_local_storage_i, NULL);
3903 }
3904 rb_gc_mark(ec->local_storage_recursive_hash);
3905 rb_gc_mark(ec->local_storage_recursive_hash_for_trace);
3906 rb_gc_mark(ec->private_const_reference);
3907
3908 /* Snapshots of copy receives being materialized; off the queue, this is their only
3909 * root. A snapshot is sender-resident, skipped as foreign by our local GC; the
3910 * global GC marks it and re-pins its shrefs (its clear pass dropped all). Move
3911 * couriers are covered by the in-flight registry instead (ractor.c). */
3912 for (const struct ractor_materialize_frame *f = ec->materialize_frames; f != NULL; f = f->prev) {
3913 rb_gc_mark(f->snapshot);
3914 if (f->snapshot && !RB_SPECIAL_CONST_P(f->snapshot) && rb_gc_during_global_gc_p()) {
3915 /* Every node, not just the root: if compaction moved a snapshot node,
3916 * the address-keyed generic_fields entries and the dedup table would
3917 * break. */
3918 rb_gc_pin_in_flight_message(f->snapshot);
3919 for (size_t i = 0; i < f->pinned_cnt; i++) {
3920 rb_gc_pin_in_flight_message(f->pinned[i]);
3921 }
3922 }
3923 }
3924
3925 rb_gc_mark_movable(ec->storage);
3926}
3927
3928void rb_fiber_mark_self(rb_fiber_t *fib);
3929void rb_fiber_update_self(rb_fiber_t *fib);
3930void rb_threadptr_root_fiber_setup(rb_thread_t *th);
3931void rb_root_fiber_obj_setup(rb_thread_t *th);
3932void rb_threadptr_root_fiber_release(rb_thread_t *th);
3933
3934static void
3935thread_compact(void *ptr)
3936{
3937 rb_thread_t *th = ptr;
3938
3939 th->self = rb_gc_location(th->self);
3940}
3941
3942/* Mark the heap objects a thread owns (the caller handles ec and fiber). Split
3943 * out of thread_mark so that a local GC can root them straight from the Ractor's
3944 * local roots (rb_ractor_mark_local_roots). */
3945void
3946rb_thread_mark_owned_roots(rb_thread_t *th)
3947{
3948 /* mark ruby objects */
3949 switch (th->invoke_type) {
3950 case thread_invoke_type_proc:
3951 case thread_invoke_type_ractor_proc:
3952 rb_gc_mark(th->invoke_arg.proc.proc);
3953 rb_gc_mark(th->invoke_arg.proc.args);
3954 break;
3955 case thread_invoke_type_func:
3956 rb_gc_mark_maybe((VALUE)th->invoke_arg.func.arg);
3957 break;
3958 default:
3959 break;
3960 }
3961
3962 rb_gc_mark(th->thgroup);
3963 rb_gc_mark(th->value);
3964 rb_gc_mark(th->pending_interrupt_queue);
3965 rb_gc_mark(th->pending_interrupt_mask_stack);
3966 rb_gc_mark(th->top_self);
3967 rb_gc_mark(th->top_wrapper);
3968 rb_gc_mark(th->last_status);
3969 rb_gc_mark(th->locking_mutex);
3970 rb_gc_mark(th->name);
3971 rb_gc_mark(th->scheduler);
3972
3973 rb_threadptr_interrupt_exec_task_mark(th);
3974}
3975
3976static void
3977thread_mark(void *ptr)
3978{
3979 rb_thread_t *th = ptr;
3980 RUBY_MARK_ENTER("thread");
3981
3982 // ec is null when setting up the thread in rb_threadptr_root_fiber_setup
3983 if (th->ec) {
3984 rb_fiber_mark_self(th->ec->fiber_ptr);
3985 }
3986
3987 /* A live thread wrapper keeps its Ractor object alive (and through its dfree the
3988 * rb_ractor_t), so an inherited Thread keeps a dead Ractor alive just as it does
3989 * upstream. */
3990 if (th->ractor) rb_gc_mark(rb_ractor_self(th->ractor));
3991 if (th->root_fiber) rb_fiber_mark_self(th->root_fiber);
3992
3993 RUBY_ASSERT(th->ec == NULL || th->ec == rb_fiberptr_get_ec(th->ec->fiber_ptr));
3994
3995 rb_thread_mark_owned_roots(th);
3996
3997 RUBY_MARK_LEAVE("thread");
3998}
3999
4000void rb_threadptr_sched_free(rb_thread_t *th); // thread_*.c
4001
4002static void
4003thread_free(void *ptr)
4004{
4005 rb_thread_t *th = ptr;
4006
4007 /* The final self collection sweeps the wrapper of the very thread running it; the
4008 * struct is still under that thread's feet (GET_EC resolves through it), so the
4009 * thread frees the struct itself at its last step (rb_ractor_postmortem_free). */
4010 if (th->ec != NULL && th->ec == rb_current_execution_context(false)) {
4011 th->self = 0;
4012 return;
4013 }
4014 rb_thread_free_body(ptr);
4015}
4016
4017void
4018rb_thread_free_body(void *ptr)
4019{
4020 rb_thread_t *th = ptr;
4021 RUBY_FREE_ENTER("thread");
4022
4023 rb_threadptr_sched_free(th);
4024 // destroyed here rather than during teardown: nothing can interrupt a
4025 // thread that is unreachable and off its Ractor's living set
4026 rb_native_mutex_destroy(&th->interrupt_lock);
4027
4028 if (th->locking_mutex != Qfalse) {
4029 rb_bug("thread_free: locking_mutex must be NULL (%p:%p)", (void *)th, (void *)th->locking_mutex);
4030 }
4031 if (th->keeping_mutexes != NULL) {
4032 rb_bug("thread_free: keeping_mutexes must be NULL (%p:%p)", (void *)th, (void *)th->keeping_mutexes);
4033 }
4034
4035 ruby_xfree(th->specific_storage);
4036
4037 if (th->vm && th->vm->ractor.main_thread == th) {
4038 RUBY_GC_INFO("MRI main thread\n");
4039 }
4040 else {
4041 // ruby_xfree(th->nt);
4042 // TODO: MN system collect nt, but without MN system it should be freed here.
4043 if (!th->main_thread) {
4044 ruby_xfree(th);
4045 }
4046 }
4047
4048 RUBY_FREE_LEAVE("thread");
4049}
4050
4051static size_t
4052thread_memsize(const void *ptr)
4053{
4054 const rb_thread_t *th = ptr;
4055 size_t size = sizeof(rb_thread_t);
4056
4057 if (!th->root_fiber) {
4058 size += th->ec->vm_stack_size * sizeof(VALUE);
4059 }
4060 if (th->ec->local_storage) {
4061 size += rb_id_table_memsize(th->ec->local_storage);
4062 }
4063 return size;
4064}
4065
4066#define thread_data_type ruby_threadptr_data_type
4067const rb_data_type_t ruby_threadptr_data_type = {
4068 "VM/thread",
4069 {
4070 thread_mark,
4071 thread_free,
4072 thread_memsize,
4073 thread_compact,
4074 },
4075 0, 0, RUBY_TYPED_THREAD_SAFE_FREE
4076};
4077
4078VALUE
4079rb_obj_is_thread(VALUE obj)
4080{
4081 return RBOOL(rb_typeddata_is_kind_of(obj, &thread_data_type));
4082}
4083
4084static VALUE
4085thread_alloc(VALUE klass)
4086{
4087 rb_thread_t *th;
4088 return TypedData_Make_Struct(klass, rb_thread_t, &thread_data_type, th);
4089}
4090
4091void
4092rb_ec_set_vm_stack(rb_execution_context_t *ec, VALUE *stack, size_t size)
4093{
4094 ec->vm_stack = stack;
4095 ec->vm_stack_size = size;
4096}
4097
4098void
4099rb_ec_initialize_vm_stack(rb_execution_context_t *ec, VALUE *stack, size_t size)
4100{
4101 rb_ec_set_vm_stack(ec, stack, size);
4102
4103#if VM_CHECK_MODE > 0
4104 MEMZERO(stack, VALUE, size); // malloc memory could have the VM canary in it
4105#endif
4106
4107 ec->cfp = (void *)(ec->vm_stack + ec->vm_stack_size);
4108
4109 vm_push_frame(ec,
4110 NULL /* dummy iseq */,
4111 VM_FRAME_MAGIC_DUMMY | VM_ENV_FLAG_LOCAL | VM_FRAME_FLAG_FINISH | VM_FRAME_FLAG_CFRAME /* dummy frame */,
4112 Qnil /* dummy self */, VM_BLOCK_HANDLER_NONE /* dummy block ptr */,
4113 0 /* dummy cref/me */,
4114 0 /* dummy pc */, ec->vm_stack, 0, 0
4115 );
4116}
4117
4118void
4119rb_ec_clear_vm_stack(rb_execution_context_t *ec)
4120{
4121 // set cfp to NULL before clearing the stack in case `thread_profile_frames`
4122 // gets called in this middle of `rb_ec_set_vm_stack` via signal handler.
4123 ec->cfp = NULL;
4124 rb_ec_set_vm_stack(ec, NULL, 0);
4125}
4126
4127void
4128rb_ec_close(rb_execution_context_t *ec)
4129{
4130 // Fiber storage is not accessible from outside the running fiber, so it is safe to clear it here.
4131 ec->storage = Qnil;
4132}
4133
4134static void
4135th_init(rb_thread_t *th, VALUE self, rb_vm_t *vm)
4136{
4137 const rb_box_t *box = rb_current_box();
4138
4139 th->self = self;
4140
4141 ccan_list_head_init(&th->interrupt_exec_tasks);
4142 // initialized here (not at thread creation) so that every Thread object
4143 // -- including allocated-but-never-started ones -- owns a valid mutex:
4144 // thread_free destroys it unconditionally
4145 rb_native_mutex_initialize(&th->interrupt_lock);
4146
4147 rb_threadptr_root_fiber_setup(th);
4148
4149 /* All threads are blocking until a non-blocking fiber is scheduled */
4150 th->blocking = 1;
4151 th->scheduler = Qnil;
4152
4153 if (self == 0) {
4154 size_t size = vm->default_params.thread_vm_stack_size / sizeof(VALUE);
4155 VALUE *stack = ALLOC_N(VALUE, size);
4156 rb_ec_initialize_vm_stack(th->ec, stack, size);
4157 rb_thread_malloc_stack_set(th, stack, size);
4158 }
4159 else {
4160 VM_ASSERT(th->ec->cfp == NULL);
4161 VM_ASSERT(th->ec->vm_stack == NULL);
4162 VM_ASSERT(th->ec->vm_stack_size == 0);
4163 }
4164
4165 th->status = THREAD_RUNNABLE;
4166 th->last_status = Qnil;
4167 th->top_wrapper = 0;
4168 if (box->top_self) {
4169 th->top_self = box->top_self;
4170 }
4171 else {
4172 th->top_self = 0;
4173 }
4174 th->value = Qundef;
4175
4176 th->ec->errinfo = Qnil;
4177 th->ec->root_svar = Qfalse;
4178 th->ec->local_storage_recursive_hash = Qnil;
4179 th->ec->local_storage_recursive_hash_for_trace = Qnil;
4180
4181 th->ec->storage = Qnil;
4182 th->ec->ractor_id = rb_ractor_id(th->ractor);
4183
4184#if OPT_CALL_THREADED_CODE
4185 th->retval = Qundef;
4186#endif
4187 th->name = Qnil;
4188 th->report_on_exception = vm->thread_report_on_exception;
4189 th->ext_config.ractor_safe = true;
4190
4191#if USE_RUBY_DEBUG_LOG
4192 static rb_atomic_t thread_serial = 1;
4193 th->serial = RUBY_ATOMIC_FETCH_ADD(thread_serial, 1);
4194
4195 RUBY_DEBUG_LOG("th:%u", th->serial);
4196#endif
4197}
4198
4199VALUE
4200rb_thread_alloc(VALUE klass)
4201{
4202 VALUE self = thread_alloc(klass);
4203 rb_thread_t *target_th = rb_thread_ptr(self);
4204 target_th->ractor = GET_RACTOR();
4205 th_init(target_th, self, target_th->vm = GET_VM());
4206 rb_root_fiber_obj_setup(target_th);
4207 return self;
4208}
4209
4210#define REWIND_CFP(expr) do { \
4211 rb_execution_context_t *ec__ = GET_EC(); \
4212 VALUE *const curr_sp = (ec__->cfp++)->sp; \
4213 VALUE *const saved_sp = ec__->cfp->sp; \
4214 ec__->cfp->sp = curr_sp; \
4215 expr; \
4216 (ec__->cfp--)->sp = saved_sp; \
4217} while (0)
4218
4219static VALUE
4220m_core_set_method_alias(VALUE self, VALUE cbase, VALUE sym1, VALUE sym2)
4221{
4222 REWIND_CFP({
4223 rb_alias(cbase, SYM2ID(sym1), SYM2ID(sym2));
4224 });
4225 return Qnil;
4226}
4227
4228static VALUE
4229m_core_set_variable_alias(VALUE self, VALUE sym1, VALUE sym2)
4230{
4231 REWIND_CFP({
4232 rb_alias_variable(SYM2ID(sym1), SYM2ID(sym2));
4233 });
4234 return Qnil;
4235}
4236
4237static VALUE
4238m_core_undef_method(VALUE self, VALUE cbase, VALUE sym)
4239{
4240 REWIND_CFP({
4241 ID mid = SYM2ID(sym);
4242 rb_undef(cbase, mid);
4243 rb_clear_method_cache(self, mid);
4244 });
4245 return Qnil;
4246}
4247
4248static VALUE
4249m_core_set_postexe(VALUE self)
4250{
4251 rb_ractor_ensure_main_ractor("can not use END{} in non-main Ractors");
4252 rb_set_end_proc(rb_call_end_proc, rb_block_proc());
4253 return Qnil;
4254}
4255
4256static VALUE core_hash_merge_kwd(VALUE hash, VALUE kw);
4257
4258static VALUE
4259core_hash_merge(VALUE hash, long argc, const VALUE *argv)
4260{
4261 Check_Type(hash, T_HASH);
4262 VM_ASSERT(argc % 2 == 0);
4263 rb_hash_bulk_insert(argc, argv, hash);
4264 return hash;
4265}
4266
4267static VALUE
4268m_core_hash_merge_ptr(int argc, VALUE *argv, VALUE recv)
4269{
4270 VALUE hash = argv[0];
4271
4272 REWIND_CFP(hash = core_hash_merge(hash, argc-1, argv+1));
4273
4274 return hash;
4275}
4276
4277static int
4278kwmerge_i(VALUE key, VALUE value, VALUE hash)
4279{
4280 rb_hash_aset(hash, key, value);
4281 return ST_CONTINUE;
4282}
4283
4284static VALUE
4285m_core_hash_merge_kwd(VALUE recv, VALUE hash, VALUE kw)
4286{
4287 if (!NIL_P(kw)) {
4288 REWIND_CFP(hash = core_hash_merge_kwd(hash, kw));
4289 }
4290 return hash;
4291}
4292
4293static VALUE
4294m_core_make_shareable(VALUE recv, VALUE obj)
4295{
4296 return rb_ractor_make_shareable(obj);
4297}
4298
4299static VALUE
4300m_core_make_shareable_copy(VALUE recv, VALUE obj)
4301{
4303}
4304
4305static VALUE
4306m_core_ensure_shareable(VALUE recv, VALUE obj, VALUE name)
4307{
4308 return rb_ractor_ensure_shareable(obj, name);
4309}
4310
4311static VALUE
4312core_hash_merge_kwd(VALUE hash, VALUE kw)
4313{
4314 rb_hash_foreach(rb_to_hash_type(kw), kwmerge_i, hash);
4315 return hash;
4316}
4317
4318extern VALUE *rb_gc_stack_start;
4319extern size_t rb_gc_stack_maxsize;
4320
4321/* debug functions */
4322
4323/* :nodoc: */
4324static VALUE
4325sdr(VALUE self)
4326{
4327 rb_vm_bugreport(NULL, stderr);
4328 return Qnil;
4329}
4330
4331/* :nodoc: */
4332static VALUE
4333nsdr(VALUE self)
4334{
4335 VALUE ary = rb_ary_new();
4336#ifdef HAVE_BACKTRACE
4337#include <execinfo.h>
4338#define MAX_NATIVE_TRACE 1024
4339 static void *trace[MAX_NATIVE_TRACE];
4340 int n = (int)backtrace(trace, MAX_NATIVE_TRACE);
4341 char **syms = backtrace_symbols(trace, n);
4342 int i;
4343
4344 if (syms == 0) {
4345 rb_memerror();
4346 }
4347
4348 for (i=0; i<n; i++) {
4349 rb_ary_push(ary, rb_str_new2(syms[i]));
4350 }
4351 free(syms); /* OK */
4352#endif
4353 return ary;
4354}
4355
4356#if VM_COLLECT_USAGE_DETAILS
4357static VALUE usage_analysis_insn_start(VALUE self);
4358static VALUE usage_analysis_operand_start(VALUE self);
4359static VALUE usage_analysis_register_start(VALUE self);
4360static VALUE usage_analysis_insn_stop(VALUE self);
4361static VALUE usage_analysis_operand_stop(VALUE self);
4362static VALUE usage_analysis_register_stop(VALUE self);
4363static VALUE usage_analysis_insn_running(VALUE self);
4364static VALUE usage_analysis_operand_running(VALUE self);
4365static VALUE usage_analysis_register_running(VALUE self);
4366static VALUE usage_analysis_insn_clear(VALUE self);
4367static VALUE usage_analysis_operand_clear(VALUE self);
4368static VALUE usage_analysis_register_clear(VALUE self);
4369#endif
4370
4371static VALUE
4372f_raise(int c, VALUE *v, VALUE _)
4373{
4374 return rb_f_raise(c, v);
4375}
4376
4377static VALUE
4378f_proc(VALUE _)
4379{
4380 return rb_block_proc();
4381}
4382
4383static VALUE
4384f_lambda(VALUE _)
4385{
4386 return rb_block_lambda();
4387}
4388
4389static VALUE
4390f_sprintf(int c, const VALUE *v, VALUE _)
4391{
4392 return rb_f_sprintf(c, v);
4393}
4394
4395/* :nodoc: */
4396static VALUE
4397vm_mtbl(VALUE self, VALUE obj, VALUE sym)
4398{
4399 vm_mtbl_dump(CLASS_OF(obj), RTEST(sym) ? SYM2ID(sym) : 0);
4400 return Qnil;
4401}
4402
4403/* :nodoc: */
4404static VALUE
4405vm_mtbl2(VALUE self, VALUE obj, VALUE sym)
4406{
4407 vm_mtbl_dump(obj, RTEST(sym) ? SYM2ID(sym) : 0);
4408 return Qnil;
4409}
4410
4411/*
4412 * call-seq:
4413 * RubyVM.keep_script_lines -> true or false
4414 *
4415 * Return current +keep_script_lines+ status. Now it only returns
4416 * +true+ of +false+, but it can return other objects in future.
4417 *
4418 * Note that this is an API for ruby internal use, debugging,
4419 * and research. Do not use this for any other purpose.
4420 * The compatibility is not guaranteed.
4421 */
4422static VALUE
4423vm_keep_script_lines(VALUE self)
4424{
4425 return RBOOL(ruby_vm_keep_script_lines);
4426}
4427
4428/*
4429 * call-seq:
4430 * RubyVM.keep_script_lines = true / false
4431 *
4432 * It set +keep_script_lines+ flag. If the flag is set, all
4433 * loaded scripts are recorded in a interpreter process.
4434 *
4435 * Note that this is an API for ruby internal use, debugging,
4436 * and research. Do not use this for any other purpose.
4437 * The compatibility is not guaranteed.
4438 */
4439static VALUE
4440vm_keep_script_lines_set(VALUE self, VALUE flags)
4441{
4442 ruby_vm_keep_script_lines = RTEST(flags);
4443 return flags;
4444}
4445
4446void
4447Init_VM(void)
4448{
4449 VALUE opts;
4450 VALUE klass;
4451 VALUE fcore;
4452
4453 /*
4454 * Document-class: RubyVM
4455 *
4456 * The RubyVM module only exists on MRI. +RubyVM+ is not defined in
4457 * other Ruby implementations such as JRuby and TruffleRuby.
4458 *
4459 * The RubyVM module provides some access to MRI internals.
4460 * This module is for very limited purposes, such as debugging,
4461 * prototyping, and research. Normal users must not use it.
4462 * This module is not portable between Ruby implementations.
4463 */
4464 rb_cRubyVM = rb_define_class("RubyVM", rb_cObject);
4465 rb_undef_alloc_func(rb_cRubyVM);
4466 rb_undef_method(CLASS_OF(rb_cRubyVM), "new");
4467 rb_define_singleton_method(rb_cRubyVM, "stat", vm_stat, -1);
4468 rb_define_singleton_method(rb_cRubyVM, "keep_script_lines", vm_keep_script_lines, 0);
4469 rb_define_singleton_method(rb_cRubyVM, "keep_script_lines=", vm_keep_script_lines_set, 1);
4470
4471#if USE_DEBUG_COUNTER
4472 rb_define_singleton_method(rb_cRubyVM, "reset_debug_counters", rb_debug_counter_reset, 0);
4473 rb_define_singleton_method(rb_cRubyVM, "show_debug_counters", rb_debug_counter_show, 0);
4474#endif
4475
4476 /* FrozenCore (hidden) */
4478 rb_set_class_path(fcore, rb_cRubyVM, "FrozenCore");
4479 rb_vm_register_global_object(rb_mod_name(fcore));
4480 klass = rb_singleton_class(fcore);
4481 rb_define_method_id(klass, id_core_set_method_alias, m_core_set_method_alias, 3);
4482 rb_define_method_id(klass, id_core_set_variable_alias, m_core_set_variable_alias, 2);
4483 rb_define_method_id(klass, id_core_undef_method, m_core_undef_method, 2);
4484 rb_define_method_id(klass, id_core_set_postexe, m_core_set_postexe, 0);
4485 rb_define_method_id(klass, id_core_hash_merge_ptr, m_core_hash_merge_ptr, -1);
4486 rb_define_method_id(klass, id_core_hash_merge_kwd, m_core_hash_merge_kwd, 2);
4487 rb_define_method_id(klass, id_core_raise, f_raise, -1);
4488 rb_define_method_id(klass, id_core_sprintf, f_sprintf, -1);
4489 rb_define_method_id(klass, idProc, f_proc, 0);
4490 rb_define_method_id(klass, idLambda, f_lambda, 0);
4491 rb_define_method(klass, "make_shareable", m_core_make_shareable, 1);
4492 rb_define_method(klass, "make_shareable_copy", m_core_make_shareable_copy, 1);
4493 rb_define_method(klass, "ensure_shareable", m_core_ensure_shareable, 2);
4494 rb_obj_freeze(fcore);
4495 RBASIC_CLEAR_CLASS(klass);
4496 rb_obj_freeze(klass);
4497 rb_vm_register_global_object(fcore);
4498 rb_mRubyVMFrozenCore = fcore;
4499
4500 /*
4501 * Document-class: Thread
4502 *
4503 * Threads are the Ruby implementation for a concurrent programming model.
4504 *
4505 * Programs that require multiple threads of execution are a perfect
4506 * candidate for Ruby's Thread class.
4507 *
4508 * For example, we can create a new thread separate from the main thread's
4509 * execution using ::new.
4510 *
4511 * thr = Thread.new { puts "What's the big deal" }
4512 *
4513 * Then we are able to pause the execution of the main thread and allow
4514 * our new thread to finish, using #join:
4515 *
4516 * thr.join #=> "What's the big deal"
4517 *
4518 * If we don't call +thr.join+ before the main thread terminates, then all
4519 * other threads including +thr+ will be killed.
4520 *
4521 * Alternatively, you can use an array for handling multiple threads at
4522 * once, like in the following example:
4523 *
4524 * threads = []
4525 * threads << Thread.new { puts "What's the big deal" }
4526 * threads << Thread.new { 3.times { puts "Threads are fun!" } }
4527 *
4528 * After creating a few threads we wait for them all to finish
4529 * consecutively.
4530 *
4531 * threads.each { |thr| thr.join }
4532 *
4533 * To retrieve the last value of a thread, use #value
4534 *
4535 * thr = Thread.new { sleep 1; "Useful value" }
4536 * thr.value #=> "Useful value"
4537 *
4538 * === Thread initialization
4539 *
4540 * In order to create new threads, Ruby provides ::new, ::start, and
4541 * ::fork. A block must be provided with each of these methods, otherwise
4542 * a ThreadError will be raised.
4543 *
4544 * When subclassing the Thread class, the +initialize+ method of your
4545 * subclass will be ignored by ::start and ::fork. Otherwise, be sure to
4546 * call super in your +initialize+ method.
4547 *
4548 * === Thread termination
4549 *
4550 * For terminating threads, Ruby provides a variety of ways to do this.
4551 *
4552 * The class method ::kill, is meant to exit a given thread:
4553 *
4554 * thr = Thread.new { sleep }
4555 * Thread.kill(thr) # sends exit() to thr
4556 *
4557 * Alternatively, you can use the instance method #exit, or any of its
4558 * aliases #kill or #terminate.
4559 *
4560 * thr.exit
4561 *
4562 * === Thread status
4563 *
4564 * Ruby provides a few instance methods for querying the state of a given
4565 * thread. To get a string with the current thread's state use #status
4566 *
4567 * thr = Thread.new { sleep }
4568 * thr.status # => "sleep"
4569 * thr.exit
4570 * thr.status # => false
4571 *
4572 * You can also use #alive? to tell if the thread is running or sleeping,
4573 * and #stop? if the thread is dead or sleeping.
4574 *
4575 * === Thread variables and scope
4576 *
4577 * Since threads are created with blocks, the same rules apply to other
4578 * Ruby blocks for variable scope. Any local variables created within this
4579 * block are accessible to only this thread.
4580 *
4581 * ==== Fiber-local vs. Thread-local
4582 *
4583 * Each fiber has its own bucket for Thread#[] storage. When you set a
4584 * new fiber-local it is only accessible within this Fiber. To illustrate:
4585 *
4586 * Thread.new {
4587 * Thread.current[:foo] = "bar"
4588 * Fiber.new {
4589 * p Thread.current[:foo] # => nil
4590 * }.resume
4591 * }.join
4592 *
4593 * This example uses #[] for getting and #[]= for setting fiber-locals,
4594 * you can also use #keys to list the fiber-locals for a given
4595 * thread and #key? to check if a fiber-local exists.
4596 *
4597 * When it comes to thread-locals, they are accessible within the entire
4598 * scope of the thread. Given the following example:
4599 *
4600 * Thread.new{
4601 * Thread.current.thread_variable_set(:foo, 1)
4602 * p Thread.current.thread_variable_get(:foo) # => 1
4603 * Fiber.new{
4604 * Thread.current.thread_variable_set(:foo, 2)
4605 * p Thread.current.thread_variable_get(:foo) # => 2
4606 * }.resume
4607 * p Thread.current.thread_variable_get(:foo) # => 2
4608 * }.join
4609 *
4610 * You can see that the thread-local +:foo+ carried over into the fiber
4611 * and was changed to +2+ by the end of the thread.
4612 *
4613 * This example makes use of #thread_variable_set to create new
4614 * thread-locals, and #thread_variable_get to reference them.
4615 *
4616 * There is also #thread_variables to list all thread-locals, and
4617 * #thread_variable? to check if a given thread-local exists.
4618 *
4619 * === Exception handling
4620 *
4621 * When an unhandled exception is raised inside a thread, it will
4622 * terminate. By default, this exception will not propagate to other
4623 * threads. The exception is stored and when another thread calls #value
4624 * or #join, the exception will be re-raised in that thread.
4625 *
4626 * t = Thread.new{ raise 'something went wrong' }
4627 * t.value #=> RuntimeError: something went wrong
4628 *
4629 * An exception can be raised from outside the thread using the
4630 * Thread#raise instance method, which takes the same parameters as
4631 * Kernel#raise.
4632 *
4633 * Setting Thread.abort_on_exception = true, Thread#abort_on_exception =
4634 * true, or $DEBUG = true will cause a subsequent unhandled exception
4635 * raised in a thread to be automatically re-raised in the main thread.
4636 *
4637 * With the addition of the class method ::handle_interrupt, you can now
4638 * handle exceptions asynchronously with threads.
4639 *
4640 * === Scheduling
4641 *
4642 * Ruby provides a few ways to support scheduling threads in your program.
4643 *
4644 * The first way is by using the class method ::stop, to put the current
4645 * running thread to sleep and schedule the execution of another thread.
4646 *
4647 * Once a thread is asleep, you can use the instance method #wakeup to
4648 * mark your thread as eligible for scheduling.
4649 *
4650 * You can also try ::pass, which attempts to pass execution to another
4651 * thread but is dependent on the OS whether a running thread will switch
4652 * or not. The same goes for #priority, which lets you hint to the thread
4653 * scheduler which threads you want to take precedence when passing
4654 * execution. This method is also dependent on the OS and may be ignored
4655 * on some platforms.
4656 *
4657 */
4658 rb_cThread = rb_define_class("Thread", rb_cObject);
4660
4661#if VM_COLLECT_USAGE_DETAILS
4662 /* ::RubyVM::USAGE_ANALYSIS_* */
4663#define define_usage_analysis_hash(name) /* shut up rdoc -C */ \
4664 rb_define_const(rb_cRubyVM, "USAGE_ANALYSIS_" #name, rb_hash_new())
4665 define_usage_analysis_hash(INSN);
4666 define_usage_analysis_hash(REGS);
4667 define_usage_analysis_hash(INSN_BIGRAM);
4668
4669 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_INSN_START", usage_analysis_insn_start, 0);
4670 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_OPERAND_START", usage_analysis_operand_start, 0);
4671 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_REGISTER_START", usage_analysis_register_start, 0);
4672 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_INSN_STOP", usage_analysis_insn_stop, 0);
4673 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_OPERAND_STOP", usage_analysis_operand_stop, 0);
4674 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_REGISTER_STOP", usage_analysis_register_stop, 0);
4675 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_INSN_RUNNING", usage_analysis_insn_running, 0);
4676 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_OPERAND_RUNNING", usage_analysis_operand_running, 0);
4677 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_REGISTER_RUNNING", usage_analysis_register_running, 0);
4678 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_INSN_CLEAR", usage_analysis_insn_clear, 0);
4679 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_OPERAND_CLEAR", usage_analysis_operand_clear, 0);
4680 rb_define_singleton_method(rb_cRubyVM, "USAGE_ANALYSIS_REGISTER_CLEAR", usage_analysis_register_clear, 0);
4681#endif
4682
4683 /* ::RubyVM::OPTS
4684 * An Array of VM build options.
4685 * This constant is MRI specific.
4686 */
4687 rb_define_const(rb_cRubyVM, "OPTS", opts = rb_ary_new());
4688
4689#if OPT_DIRECT_THREADED_CODE
4690 rb_ary_push(opts, rb_str_new2("direct threaded code"));
4691#elif OPT_TOKEN_THREADED_CODE
4692 rb_ary_push(opts, rb_str_new2("token threaded code"));
4693#elif OPT_CALL_THREADED_CODE
4694 rb_ary_push(opts, rb_str_new2("call threaded code"));
4695#endif
4696
4697#if OPT_OPERANDS_UNIFICATION
4698 rb_ary_push(opts, rb_str_new2("operands unification"));
4699#endif
4700#if OPT_INSTRUCTIONS_UNIFICATION
4701 rb_ary_push(opts, rb_str_new2("instructions unification"));
4702#endif
4703#if OPT_INLINE_METHOD_CACHE
4704 rb_ary_push(opts, rb_str_new2("inline method cache"));
4705#endif
4706
4707 /* ::RubyVM::INSTRUCTION_NAMES
4708 * A list of bytecode instruction names in MRI.
4709 * This constant is MRI specific.
4710 */
4711 rb_define_const(rb_cRubyVM, "INSTRUCTION_NAMES", rb_insns_name_array());
4712
4713 /* ::RubyVM::DEFAULT_PARAMS
4714 * This constant exposes the VM's default parameters.
4715 * Note that changing these values does not affect VM execution.
4716 * Specification is not stable and you should not depend on this value.
4717 * Of course, this constant is MRI specific.
4718 */
4719 rb_define_const(rb_cRubyVM, "DEFAULT_PARAMS", vm_default_params());
4720
4721 /* debug functions ::RubyVM::SDR(), ::RubyVM::NSDR() */
4722#if VMDEBUG
4723 rb_define_singleton_method(rb_cRubyVM, "SDR", sdr, 0);
4724 rb_define_singleton_method(rb_cRubyVM, "NSDR", nsdr, 0);
4725 rb_define_singleton_method(rb_cRubyVM, "mtbl", vm_mtbl, 2);
4726 rb_define_singleton_method(rb_cRubyVM, "mtbl2", vm_mtbl2, 2);
4727#else
4728 (void)sdr;
4729 (void)nsdr;
4730 (void)vm_mtbl;
4731 (void)vm_mtbl2;
4732#endif
4733
4734 /* VM bootstrap: phase 2 */
4735 {
4736 rb_vm_t *vm = ruby_current_vm_ptr;
4737 rb_thread_t *th = GET_THREAD();
4738 VALUE filename = rb_fstring_lit("<main>");
4739 const rb_iseq_t *iseq = rb_iseq_new(Qnil, filename, filename, Qnil, 0, ISEQ_TYPE_TOP);
4740
4741 // Ractor setup
4742 rb_ractor_main_setup(vm, th->ractor, th);
4743
4744 /* create vm object */
4745 vm->self = TypedData_Wrap_Struct(rb_cRubyVM, &vm_data_type, vm);
4746
4747 /* create main thread */
4748 th->self = TypedData_Wrap_Struct(rb_cThread, &thread_data_type, th);
4749 vm->ractor.main_thread = th;
4750 vm->ractor.main_ractor = th->ractor;
4751 th->vm = vm;
4752 th->top_wrapper = 0;
4753 th->top_self = rb_vm_top_self();
4754
4755 rb_root_fiber_obj_setup(th);
4756
4757 rb_vm_register_global_object((VALUE)iseq);
4758 th->ec->cfp->_iseq = iseq;
4759 th->ec->cfp->pc = ISEQ_BODY(iseq)->iseq_encoded;
4760 th->ec->cfp->self = th->top_self;
4761
4762 VM_ENV_FLAGS_UNSET(th->ec->cfp->ep, VM_FRAME_FLAG_CFRAME);
4763 VM_STACK_ENV_WRITE(th->ec->cfp->ep, VM_ENV_DATA_INDEX_ME_CREF, (VALUE)vm_cref_new(rb_cObject, METHOD_VISI_PRIVATE, FALSE, NULL, FALSE, FALSE));
4764
4765 /*
4766 * The Binding of the top level scope
4767 */
4768 rb_define_global_const("TOPLEVEL_BINDING", rb_binding_new());
4769
4770#ifdef _WIN32
4771 rb_objspace_gc_enable(vm->ractor.main_ractor->objspace);
4772#endif
4773 }
4774 vm_init_redefined_flag();
4775
4776 rb_block_param_proxy = rb_obj_alloc(rb_cObject);
4777 VALUE proxy_singleton = rb_singleton_class(rb_block_param_proxy);
4778 rb_add_method_optimized(proxy_singleton, idCall, OPTIMIZED_METHOD_TYPE_BLOCK_CALL, 0, METHOD_VISI_PUBLIC);
4779 rb_add_method_optimized(proxy_singleton, idAREF, OPTIMIZED_METHOD_TYPE_BLOCK_CALL, 0, METHOD_VISI_PUBLIC);
4780 rb_add_method_optimized(proxy_singleton, idYield, OPTIMIZED_METHOD_TYPE_BLOCK_CALL, 0, METHOD_VISI_PUBLIC);
4781 rb_add_method_optimized(proxy_singleton, idEqq, OPTIMIZED_METHOD_TYPE_BLOCK_CALL, 0, METHOD_VISI_PUBLIC);
4782 rb_obj_freeze(rb_block_param_proxy);
4783 rb_vm_register_global_object(rb_block_param_proxy);
4784
4785 /* vm_backtrace.c */
4786 Init_vm_backtrace();
4787}
4788
4789void
4790rb_vm_set_progname(VALUE filename)
4791{
4792 rb_thread_t *th = GET_VM()->ractor.main_thread;
4793 rb_control_frame_t *cfp = (void *)(th->ec->vm_stack + th->ec->vm_stack_size);
4794 --cfp;
4795
4796 filename = rb_str_new_frozen(filename);
4797 rb_iseq_pathobj_set(CFP_ISEQ(cfp), filename, rb_iseq_realpath(CFP_ISEQ(cfp)));
4798}
4799
4800extern const struct st_hash_type rb_fstring_hash_type;
4801
4802static rb_vm_t _vm;
4803static rb_thread_t _main_thread = {
4804 .vm = &_vm,
4805 .main_thread = 1,
4806};
4807
4808void
4809Init_BareVM(void)
4810{
4811 /* VM bootstrap: phase 1 */
4812 rb_vm_t *vm = &_vm;
4813 rb_thread_t *th = &_main_thread;
4814
4815 // setup the VM
4816 vm_init2(vm);
4817
4818 ruby_current_vm_ptr = vm;
4819 /* The boot objspace belongs to the main Ractor, so the main Ractor has to exist
4820 * before rb_gc_init_objspaces allocates it. */
4821 vm->ractor.main_ractor = rb_ractor_main_alloc();
4822 rb_gc_init_objspaces();
4823 vm->ractor.main_ractor->newobj_cache = rb_gc_ractor_cache_alloc(vm->ractor.main_ractor);
4824 rb_id_table_init(&vm->negative_cme_table, 16);
4825 st_init_existing_numtable_with_size(&vm->overloaded_cme_table, 0);
4826 st_init_existing_strtable_with_size(&vm->static_ext_inits, 0);
4827 set_init_embedded_numtable_with_size(&vm->unused_block_warning_table, 0);
4828 vm->global_hooks.type = hook_list_type_global;
4829
4830 // setup main thread
4831 th->nt = ZALLOC(struct rb_native_thread);
4832 th->ractor = vm->ractor.main_ractor;
4833 Init_native_thread(th);
4834 rb_jit_cont_init();
4835 th_init(th, 0, vm);
4836
4837 rb_ractor_set_current_ec(th->ractor, th->ec);
4838
4839 /* n.b. native_main_thread_stack_top is set by the INIT_STACK macro */
4840 ruby_thread_init_stack(th, native_main_thread_stack_top);
4841
4842 // setup ractor system
4843 rb_native_mutex_initialize(&vm->ractor.sync.lock);
4844 rb_native_cond_initialize(&vm->ractor.sync.terminate_cond);
4845 rb_native_mutex_initialize(&vm->ractor.generic_fields_lock);
4846 rb_native_mutex_initialize(&vm->ractor.move_courier_registry_lock);
4847 ccan_list_head_init(&vm->ractor.move_courier_registry);
4848 rb_native_mutex_initialize(&vm->gc.registered_globals.lock);
4849 vm->gc.orphan_merge_pjob = POSTPONED_JOB_HANDLE_INVALID;
4850
4851 vm_opt_method_def_table = st_init_numtable();
4852 vm_opt_mid_table = st_init_numtable();
4853
4854#ifdef RUBY_THREAD_WIN32_H
4855 rb_native_cond_initialize(&vm->ractor.sync.barrier_complete_cond);
4856 rb_native_cond_initialize(&vm->ractor.sync.barrier_release_cond);
4857#endif
4858}
4859
4860void
4862{
4863 native_main_thread_stack_top = addr;
4864}
4865
4866#ifndef _WIN32
4867#include <unistd.h>
4868#include <sys/mman.h>
4869#endif
4870
4871
4872void
4873rb_vm_register_global_object(VALUE obj)
4874{
4876 if (RB_SPECIAL_CONST_P(obj)) {
4877 return;
4878 }
4879
4880 switch (RB_BUILTIN_TYPE(obj)) {
4881 case T_CLASS:
4882 case T_MODULE:
4883 if (FL_TEST(obj, RCLASS_IS_ROOT)) {
4884 return;
4885 }
4886 FL_SET(obj, RCLASS_IS_ROOT);
4887 break;
4888 default:
4889 break;
4890 }
4891 /* Register in the current Ractor's own pin list (a raw array). No lock: only the
4892 * owner appends and only the owner's GC marks it; the merge that inherits a list
4893 * runs stop-the-world. */
4894 rb_ractor_t *cr = GET_RACTOR();
4895 if (cr->registered_marks_cnt == cr->registered_marks_capa) {
4896 size_t nc = cr->registered_marks_capa ? cr->registered_marks_capa * 2 : 64;
4897 VALUE *p = realloc(cr->registered_marks, nc * sizeof(VALUE));
4898 if (!p) rb_bug("rb_vm_register_global_object: out of memory");
4899 cr->registered_marks = p;
4900 cr->registered_marks_capa = nc;
4901 }
4902 cr->registered_marks[cr->registered_marks_cnt++] = obj;
4903 RB_GC_GUARD(obj);
4904}
4905
4906VALUE rb_cc_refinement_set_create(void);
4907
4908void
4909Init_vm_objects(void)
4910{
4911 rb_vm_t *vm = GET_VM();
4912 st_init_existing_table_with_size(&vm->ci_table, &vm_ci_hashtype, 0);
4913 vm->cc_refinement_set = rb_cc_refinement_set_create();
4914}
4915
4916// Whether JIT is enabled or not, we need to load/undef `#with_jit` for other builtins.
4917#include "jit_hook.rbinc"
4918#include "jit_undef.rbinc"
4919
4920// Stub for builtin function when not building YJIT units
4921#if !USE_YJIT
4922void Init_builtin_yjit(void) {}
4923#endif
4924
4925// Stub for builtin function when not building ZJIT units
4926#if !USE_ZJIT
4927void Init_builtin_zjit(void) {}
4928#endif
4929
4930/* top self */
4931
4932static VALUE
4933main_to_s(VALUE obj)
4934{
4935 return rb_str_new2("main");
4936}
4937
4938VALUE
4939rb_vm_top_self(void)
4940{
4941 const rb_box_t *box = rb_current_box();
4942 VM_ASSERT(box);
4943 VM_ASSERT(box->top_self);
4944 return box->top_self;
4945}
4946
4947void
4948Init_top_self(void)
4949{
4950 rb_vm_t *vm = GET_VM();
4951 vm->root_box = (rb_box_t *)rb_root_box();
4952 vm->root_box->top_self = rb_obj_alloc(rb_cObject);
4953 rb_define_singleton_method(vm->root_box->top_self, "to_s", main_to_s, 0);
4954 rb_define_alias(rb_singleton_class(vm->root_box->top_self), "inspect", "to_s");
4955}
4956
4957VALUE *
4959{
4960 rb_ractor_t *cr = GET_RACTOR();
4961 return &cr->verbose;
4962}
4963
4964VALUE *
4966{
4967 rb_ractor_t *cr = GET_RACTOR();
4968 return &cr->debug;
4969}
4970
4971bool rb_free_at_exit = false;
4972
4973bool
4974ruby_free_at_exit_p(void)
4975{
4976 return rb_free_at_exit;
4977}
4978
4979/* iseq.c */
4980VALUE rb_insn_operand_intern(const rb_iseq_t *iseq,
4981 VALUE insn, int op_no, VALUE op,
4982 int len, size_t pos, VALUE *pnop, VALUE child);
4983
4984#if VM_COLLECT_USAGE_DETAILS
4985
4986#define HASH_ASET(h, k, v) rb_hash_aset((h), (st_data_t)(k), (st_data_t)(v))
4987
4988/* uh = {
4989 * insn(Fixnum) => ihash(Hash)
4990 * }
4991 * ihash = {
4992 * -1(Fixnum) => count, # insn usage
4993 * 0(Fixnum) => ophash, # operand usage
4994 * }
4995 * ophash = {
4996 * val(interned string) => count(Fixnum)
4997 * }
4998 */
4999static void
5000vm_analysis_insn(int insn)
5001{
5002 ID usage_hash;
5003 ID bigram_hash;
5004 static int prev_insn = -1;
5005
5006 VALUE uh;
5007 VALUE ihash;
5008 VALUE cv;
5009
5010 CONST_ID(usage_hash, "USAGE_ANALYSIS_INSN");
5011 CONST_ID(bigram_hash, "USAGE_ANALYSIS_INSN_BIGRAM");
5012 uh = rb_const_get(rb_cRubyVM, usage_hash);
5013 if (NIL_P(ihash = rb_hash_aref(uh, INT2FIX(insn)))) {
5014 ihash = rb_hash_new();
5015 HASH_ASET(uh, INT2FIX(insn), ihash);
5016 }
5017 if (NIL_P(cv = rb_hash_aref(ihash, INT2FIX(-1)))) {
5018 cv = INT2FIX(0);
5019 }
5020 HASH_ASET(ihash, INT2FIX(-1), INT2FIX(FIX2INT(cv) + 1));
5021
5022 /* calc bigram */
5023 if (prev_insn != -1) {
5024 VALUE bi;
5025 VALUE ary[2];
5026 VALUE cv;
5027
5028 ary[0] = INT2FIX(prev_insn);
5029 ary[1] = INT2FIX(insn);
5030 bi = rb_ary_new4(2, &ary[0]);
5031
5032 uh = rb_const_get(rb_cRubyVM, bigram_hash);
5033 if (NIL_P(cv = rb_hash_aref(uh, bi))) {
5034 cv = INT2FIX(0);
5035 }
5036 HASH_ASET(uh, bi, INT2FIX(FIX2INT(cv) + 1));
5037 }
5038 prev_insn = insn;
5039}
5040
5041static void
5042vm_analysis_operand(int insn, int n, VALUE op)
5043{
5044 ID usage_hash;
5045
5046 VALUE uh;
5047 VALUE ihash;
5048 VALUE ophash;
5049 VALUE valstr;
5050 VALUE cv;
5051
5052 CONST_ID(usage_hash, "USAGE_ANALYSIS_INSN");
5053
5054 uh = rb_const_get(rb_cRubyVM, usage_hash);
5055 if (NIL_P(ihash = rb_hash_aref(uh, INT2FIX(insn)))) {
5056 ihash = rb_hash_new();
5057 HASH_ASET(uh, INT2FIX(insn), ihash);
5058 }
5059 if (NIL_P(ophash = rb_hash_aref(ihash, INT2FIX(n)))) {
5060 ophash = rb_hash_new();
5061 HASH_ASET(ihash, INT2FIX(n), ophash);
5062 }
5063 /* intern */
5064 valstr = rb_insn_operand_intern(CFP_ISEQ(GET_EC()->cfp), insn, n, op, 0, 0, 0, 0);
5065
5066 /* set count */
5067 if (NIL_P(cv = rb_hash_aref(ophash, valstr))) {
5068 cv = INT2FIX(0);
5069 }
5070 HASH_ASET(ophash, valstr, INT2FIX(FIX2INT(cv) + 1));
5071}
5072
5073static void
5074vm_analysis_register(int reg, int isset)
5075{
5076 ID usage_hash;
5077 VALUE uh;
5078 VALUE valstr;
5079 static const char regstrs[][5] = {
5080 "pc", /* 0 */
5081 "sp", /* 1 */
5082 "ep", /* 2 */
5083 "cfp", /* 3 */
5084 "self", /* 4 */
5085 "iseq", /* 5 */
5086 };
5087 static const char getsetstr[][4] = {
5088 "get",
5089 "set",
5090 };
5091 static VALUE syms[sizeof(regstrs) / sizeof(regstrs[0])][2];
5092
5093 VALUE cv;
5094
5095 CONST_ID(usage_hash, "USAGE_ANALYSIS_REGS");
5096 if (syms[0] == 0) {
5097 char buff[0x10];
5098 int i;
5099
5100 for (i = 0; i < (int)(sizeof(regstrs) / sizeof(regstrs[0])); i++) {
5101 int j;
5102 for (j = 0; j < 2; j++) {
5103 snprintf(buff, 0x10, "%d %s %-4s", i, getsetstr[j], regstrs[i]);
5104 syms[i][j] = ID2SYM(rb_intern(buff));
5105 }
5106 }
5107 }
5108 valstr = syms[reg][isset];
5109
5110 uh = rb_const_get(rb_cRubyVM, usage_hash);
5111 if (NIL_P(cv = rb_hash_aref(uh, valstr))) {
5112 cv = INT2FIX(0);
5113 }
5114 HASH_ASET(uh, valstr, INT2FIX(FIX2INT(cv) + 1));
5115}
5116
5117#undef HASH_ASET
5118
5119static void (*ruby_vm_collect_usage_func_insn)(int insn) = NULL;
5120static void (*ruby_vm_collect_usage_func_operand)(int insn, int n, VALUE op) = NULL;
5121static void (*ruby_vm_collect_usage_func_register)(int reg, int isset) = NULL;
5122
5123/* :nodoc: */
5124static VALUE
5125usage_analysis_insn_start(VALUE self)
5126{
5127 ruby_vm_collect_usage_func_insn = vm_analysis_insn;
5128 return Qnil;
5129}
5130
5131/* :nodoc: */
5132static VALUE
5133usage_analysis_operand_start(VALUE self)
5134{
5135 ruby_vm_collect_usage_func_operand = vm_analysis_operand;
5136 return Qnil;
5137}
5138
5139/* :nodoc: */
5140static VALUE
5141usage_analysis_register_start(VALUE self)
5142{
5143 ruby_vm_collect_usage_func_register = vm_analysis_register;
5144 return Qnil;
5145}
5146
5147/* :nodoc: */
5148static VALUE
5149usage_analysis_insn_stop(VALUE self)
5150{
5151 ruby_vm_collect_usage_func_insn = 0;
5152 return Qnil;
5153}
5154
5155/* :nodoc: */
5156static VALUE
5157usage_analysis_operand_stop(VALUE self)
5158{
5159 ruby_vm_collect_usage_func_operand = 0;
5160 return Qnil;
5161}
5162
5163/* :nodoc: */
5164static VALUE
5165usage_analysis_register_stop(VALUE self)
5166{
5167 ruby_vm_collect_usage_func_register = 0;
5168 return Qnil;
5169}
5170
5171/* :nodoc: */
5172static VALUE
5173usage_analysis_insn_running(VALUE self)
5174{
5175 return RBOOL(ruby_vm_collect_usage_func_insn != 0);
5176}
5177
5178/* :nodoc: */
5179static VALUE
5180usage_analysis_operand_running(VALUE self)
5181{
5182 return RBOOL(ruby_vm_collect_usage_func_operand != 0);
5183}
5184
5185/* :nodoc: */
5186static VALUE
5187usage_analysis_register_running(VALUE self)
5188{
5189 return RBOOL(ruby_vm_collect_usage_func_register != 0);
5190}
5191
5192static VALUE
5193usage_analysis_clear(VALUE self, ID usage_hash)
5194{
5195 VALUE uh;
5196 uh = rb_const_get(self, usage_hash);
5197 rb_hash_clear(uh);
5198
5199 return Qtrue;
5200}
5201
5202
5203/* :nodoc: */
5204static VALUE
5205usage_analysis_insn_clear(VALUE self)
5206{
5207 ID usage_hash;
5208 ID bigram_hash;
5209
5210 CONST_ID(usage_hash, "USAGE_ANALYSIS_INSN");
5211 CONST_ID(bigram_hash, "USAGE_ANALYSIS_INSN_BIGRAM");
5212 usage_analysis_clear(rb_cRubyVM, usage_hash);
5213 return usage_analysis_clear(rb_cRubyVM, bigram_hash);
5214}
5215
5216/* :nodoc: */
5217static VALUE
5218usage_analysis_operand_clear(VALUE self)
5219{
5220 ID usage_hash;
5221
5222 CONST_ID(usage_hash, "USAGE_ANALYSIS_INSN");
5223 return usage_analysis_clear(self, usage_hash);
5224}
5225
5226/* :nodoc: */
5227static VALUE
5228usage_analysis_register_clear(VALUE self)
5229{
5230 ID usage_hash;
5231
5232 CONST_ID(usage_hash, "USAGE_ANALYSIS_REGS");
5233 return usage_analysis_clear(self, usage_hash);
5234}
5235
5236/* @param insn instruction number */
5237static void
5238vm_collect_usage_insn(int insn)
5239{
5240 if (RUBY_DTRACE_INSN_ENABLED()) {
5241 RUBY_DTRACE_INSN(rb_insns_name(insn));
5242 }
5243 if (ruby_vm_collect_usage_func_insn)
5244 (*ruby_vm_collect_usage_func_insn)(insn);
5245}
5246
5247/* @param insn instruction number
5248 * @param n n-th operand
5249 * @param op operand value
5250 */
5251static void
5252vm_collect_usage_operand(int insn, int n, VALUE op)
5253{
5254 if (RUBY_DTRACE_INSN_OPERAND_ENABLED()) {
5255 VALUE valstr;
5256
5257 valstr = rb_insn_operand_intern(CFP_ISEQ(GET_EC()->cfp), insn, n, op, 0, 0, 0, 0);
5258
5259 RUBY_DTRACE_INSN_OPERAND(RSTRING_PTR(valstr), rb_insns_name(insn));
5260 RB_GC_GUARD(valstr);
5261 }
5262 if (ruby_vm_collect_usage_func_operand)
5263 (*ruby_vm_collect_usage_func_operand)(insn, n, op);
5264}
5265
5266/* @param reg register id. see code of vm_analysis_register() */
5267/* @param isset 0: read, 1: write */
5268static void
5269vm_collect_usage_register(int reg, int isset)
5270{
5271 if (ruby_vm_collect_usage_func_register)
5272 (*ruby_vm_collect_usage_func_register)(reg, isset);
5273}
5274#endif
5275
5276const struct rb_callcache *
5277rb_vm_empty_cc(void)
5278{
5279 return &vm_empty_cc;
5280}
5281
5282const struct rb_callcache *
5283rb_vm_empty_cc_for_super(void)
5284{
5285 return &vm_empty_cc_for_super;
5286}
5287
5288#include "vm_call_iseq_optimized.inc" /* required from vm_insnhelper.c */
#define RUBY_ASSERT_ALWAYS(expr,...)
A variant of RUBY_ASSERT that does not interface with RUBY_DEBUG.
Definition assert.h:199
#define RUBY_ASSERT_MESG(expr,...)
Asserts that the expression is truthy.
Definition assert.h:186
#define RUBY_ASSERT(...)
Asserts that the given expression is truthy if and only if RUBY_DEBUG is truthy.
Definition assert.h:219
std::atomic< unsigned > rb_atomic_t
Type that is eligible for atomic operations.
Definition atomic.h:69
#define RUBY_ATOMIC_FETCH_ADD(var, val)
Atomically replaces the value pointed by var with the result of addition of val to the old value of v...
Definition atomic.h:118
#define rb_define_method(klass, mid, func, arity)
Defines klass#mid.
#define rb_define_method_id(klass, mid, func, arity)
Defines klass#mid.
#define rb_define_singleton_method(klass, mid, func, arity)
Defines klass.mid.
#define RUBY_EVENT_END
Encountered an end of a class clause.
Definition event.h:40
#define RUBY_EVENT_B_RETURN
Encountered a next statement.
Definition event.h:56
#define RUBY_EVENT_RETURN
Encountered a return statement.
Definition event.h:42
#define RUBY_EVENT_C_RETURN
Return from a method, written in C.
Definition event.h:44
uint32_t rb_event_flag_t
Represents event(s).
Definition event.h:108
VALUE rb_class_new(VALUE super)
Creates a new, anonymous class.
Definition class.c:789
VALUE rb_singleton_class(VALUE obj)
Finds or creates the singleton class of the passed object.
Definition class.c:2854
void rb_define_alias(VALUE klass, const char *name1, const char *name2)
Defines an alias of a method.
Definition class.c:2897
void rb_undef_method(VALUE klass, const char *name)
Defines an undef of a method.
Definition class.c:2707
#define rb_str_new2
Old name of rb_str_new_cstr.
Definition string.h:1676
#define NUM2ULONG
Old name of RB_NUM2ULONG.
Definition long.h:52
#define ALLOCV
Old name of RB_ALLOCV.
Definition memory.h:404
#define ALLOC
Old name of RB_ALLOC.
Definition memory.h:400
#define Qundef
Old name of RUBY_Qundef.
#define INT2FIX
Old name of RB_INT2FIX.
Definition long.h:48
#define T_IMEMO
Old name of RUBY_T_IMEMO.
Definition value_type.h:67
#define ID2SYM
Old name of RB_ID2SYM.
Definition symbol.h:44
#define SPECIAL_CONST_P
Old name of RB_SPECIAL_CONST_P.
#define OBJ_FREEZE
Old name of RB_OBJ_FREEZE.
Definition fl_type.h:131
#define ULONG2NUM
Old name of RB_ULONG2NUM.
Definition long.h:60
#define UNREACHABLE_RETURN
Old name of RBIMPL_UNREACHABLE_RETURN.
Definition assume.h:29
#define SYM2ID
Old name of RB_SYM2ID.
Definition symbol.h:45
#define ZALLOC
Old name of RB_ZALLOC.
Definition memory.h:402
#define CLASS_OF
Old name of rb_class_of.
Definition globals.h:205
#define rb_ary_new4
Old name of rb_ary_new_from_values.
Definition array.h:659
#define SIZET2NUM
Old name of RB_SIZE2NUM.
Definition size_t.h:62
#define rb_exc_new2
Old name of rb_exc_new_cstr.
Definition error.h:37
#define FIX2INT
Old name of RB_FIX2INT.
Definition int.h:41
#define T_MODULE
Old name of RUBY_T_MODULE.
Definition value_type.h:70
#define ZALLOC_N
Old name of RB_ZALLOC_N.
Definition memory.h:401
#define ASSUME
Old name of RBIMPL_ASSUME.
Definition assume.h:27
#define T_ICLASS
Old name of RUBY_T_ICLASS.
Definition value_type.h:66
#define T_HASH
Old name of RUBY_T_HASH.
Definition value_type.h:65
#define ALLOC_N
Old name of RB_ALLOC_N.
Definition memory.h:399
#define FL_SET
Old name of RB_FL_SET.
Definition fl_type.h:125
#define rb_exc_new3
Old name of rb_exc_new_str.
Definition error.h:38
#define ULL2NUM
Old name of RB_ULL2NUM.
Definition long_long.h:31
#define Qtrue
Old name of RUBY_Qtrue.
#define Qnil
Old name of RUBY_Qnil.
#define Qfalse
Old name of RUBY_Qfalse.
#define NIL_P
Old name of RB_NIL_P.
#define NUM2ULL
Old name of RB_NUM2ULL.
Definition long_long.h:35
#define T_CLASS
Old name of RUBY_T_CLASS.
Definition value_type.h:58
#define BUILTIN_TYPE
Old name of RB_BUILTIN_TYPE.
Definition value_type.h:85
#define FL_TEST
Old name of RB_FL_TEST.
Definition fl_type.h:127
#define FIXNUM_P
Old name of RB_FIXNUM_P.
#define FL_USHIFT
Old name of RUBY_FL_USHIFT.
Definition fl_type.h:67
#define CONST_ID
Old name of RUBY_CONST_ID.
Definition symbol.h:47
#define ALLOCV_END
Old name of RB_ALLOCV_END.
Definition memory.h:406
#define SYMBOL_P
Old name of RB_SYMBOL_P.
Definition value_type.h:88
void ruby_init_stack(void *addr)
Set stack bottom of Ruby implementation.
Definition vm.c:4861
VALUE rb_eLocalJumpError
LocalJumpError exception.
Definition eval.c:49
void rb_category_warn(rb_warning_category_t category, const char *fmt,...)
Identical to rb_category_warning(), except it reports unless $VERBOSE is nil.
Definition error.c:478
void rb_exc_raise(VALUE mesg)
Raises an exception in the current thread.
Definition eval.c:672
void rb_iter_break(void)
Breaks from a block.
Definition vm.c:2341
VALUE rb_eTypeError
TypeError exception.
Definition error.c:1431
void rb_iter_break_value(VALUE val)
Identical to rb_iter_break(), except it additionally takes the "value" of this breakage.
Definition vm.c:2347
VALUE rb_eRuntimeError
RuntimeError exception.
Definition error.c:1429
VALUE * rb_ruby_verbose_ptr(void)
This is an implementation detail of ruby_verbose.
Definition vm.c:4958
VALUE rb_exc_new_str(VALUE etype, VALUE str)
Identical to rb_exc_new_cstr(), except it takes a Ruby's string instead of C's.
Definition error.c:1482
VALUE * rb_ruby_debug_ptr(void)
This is an implementation detail of ruby_debug.
Definition vm.c:4965
VALUE rb_eSysStackError
SystemStackError exception.
Definition eval.c:50
@ RB_WARN_CATEGORY_PERFORMANCE
Warning is for performance issues (not enabled by -w).
Definition error.h:54
VALUE rb_cTime
Time class.
Definition time.c:702
VALUE rb_cArray
Array class.
VALUE rb_cObject
Object class.
Definition object.c:58
VALUE rb_obj_alloc(VALUE klass)
Allocates an instance of the given class.
Definition object.c:2239
VALUE rb_cInteger
Module class.
Definition numeric.c:202
VALUE rb_cNilClass
NilClass class.
Definition object.c:63
VALUE rb_cBinding
Binding class.
Definition proc.c:45
VALUE rb_cRegexp
Regexp class.
Definition re.c:2825
VALUE rb_cHash
Hash class.
Definition hash.c:117
VALUE rb_cFalseClass
FalseClass class.
Definition object.c:65
VALUE rb_obj_class(VALUE obj)
Queries the class of an object.
Definition object.c:232
VALUE rb_cSymbol
Symbol class.
Definition string.c:85
VALUE rb_cBasicObject
BasicObject class.
Definition object.c:56
VALUE rb_cThread
Thread class.
Definition vm.c:681
VALUE rb_obj_freeze(VALUE obj)
Just calls rb_obj_freeze_inline() inside.
Definition object.c:1297
VALUE rb_cFloat
Float class.
Definition numeric.c:201
VALUE rb_cProc
Proc class.
Definition proc.c:46
VALUE rb_cTrueClass
TrueClass class.
Definition object.c:64
VALUE rb_cString
String class.
Definition string.c:84
#define RB_OBJ_WRITTEN(old, oldv, young)
Identical to RB_OBJ_WRITE(), except it doesn't write any values, but only a WB declaration.
Definition gc.h:468
#define RB_OBJ_WRITE(old, slot, young)
Declaration of a "back" pointer.
Definition gc.h:456
Defines RBIMPL_HAS_BUILTIN.
VALUE rb_ary_delete_at(VALUE ary, long pos)
Destructively removes an element which resides at the specific index of the passed array.
VALUE rb_ary_new(void)
Allocates a new, empty array.
VALUE rb_ary_push(VALUE ary, VALUE elem)
Special case of rb_ary_cat() that it adds only one element.
void rb_undef(VALUE mod, ID mid)
Inserts a method entry that hides previous method definition of the given name.
Definition vm_method.c:2466
static int rb_check_arity(int argc, int min, int max)
Ensures that the passed integer is in the passed range.
Definition error.h:284
VALUE rb_backref_get(void)
Queries the last match, or Regexp.last_match, or the $~.
Definition vm.c:2091
void rb_lastline_set(VALUE str)
Updates $_.
Definition vm.c:2109
VALUE rb_lastline_get(void)
Queries the last line, or the $_.
Definition vm.c:2103
void rb_backref_set(VALUE md)
Updates $~.
Definition vm.c:2097
VALUE rb_block_proc(void)
Constructs a Proc object from implicitly passed components.
Definition proc.c:1549
VALUE rb_block_lambda(void)
Identical to rb_proc_new(), except it returns a lambda.
Definition proc.c:1568
VALUE rb_binding_new(void)
Snapshots the current execution context and turn it into an instance of rb_cBinding.
Definition proc.c:880
VALUE rb_str_append(VALUE dst, VALUE src)
Identical to rb_str_buf_append(), except it converts the right hand side before concatenating.
Definition string.c:3880
VALUE rb_str_new_frozen(VALUE str)
Creates a frozen copy of the string, if necessary.
Definition string.c:1537
#define rb_str_cat_cstr(buf, str)
Identical to rb_str_cat(), except it assumes the passed pointer is a pointer to a C string.
Definition string.h:1657
VALUE rb_const_get(VALUE space, ID name)
Identical to rb_const_defined(), except it returns the actual defined value.
Definition variable.c:3408
void rb_set_class_path(VALUE klass, VALUE space, const char *name)
Names a class.
Definition variable.c:455
VALUE rb_mod_name(VALUE mod)
Queries the name of a module.
Definition variable.c:150
void rb_alias_variable(ID dst, ID src)
Aliases a global variable.
Definition variable.c:1180
VALUE rb_class_path(VALUE mod)
Identical to rb_mod_name(), except it returns #<Class: ...> style inspection for anonymous modules.
Definition variable.c:394
void rb_undef_alloc_func(VALUE klass)
Deletes the allocator function of a class.
Definition vm_method.c:1799
const char * rb_sourcefile(void)
Resembles __FILE__.
Definition vm.c:2128
void rb_alias(VALUE klass, ID dst, ID src)
Resembles alias.
Definition vm_method.c:2849
int rb_frame_method_id_and_class(ID *idp, VALUE *klassp)
Resembles __method__.
Definition vm.c:3241
int rb_sourceline(void)
Resembles __LINE__.
Definition vm.c:2142
VALUE rb_sym2str(VALUE symbol)
Obtain a frozen string representation of a symbol (not including the leading colon).
Definition symbol.c:1147
void rb_define_global_const(const char *name, VALUE val)
Identical to rb_define_const(), except it defines that of "global", i.e.
Definition variable.c:3997
VALUE rb_iv_set(VALUE obj, const char *name, VALUE val)
Assigns to an instance variable.
Definition variable.c:4471
int len
Length of the buffer.
Definition io.h:8
VALUE rb_ractor_make_shareable_copy(VALUE obj)
Identical to rb_ractor_make_shareable(), except it returns a (deep) copy of the passed one instead of...
Definition ractor.c:1940
static bool rb_ractor_shareable_p(VALUE obj)
Queries if multiple Ractors can share the passed object or not.
Definition ractor.h:249
#define RB_OBJ_SHAREABLE_P(obj)
Queries if the passed object has previously classified as shareable or not.
Definition ractor.h:235
VALUE rb_ractor_make_shareable(VALUE obj)
Destructively transforms the passed object so that multiple Ractors can share it.
Definition ractor.c:1929
void ruby_vm_at_exit(void(*func)(ruby_vm_t *))
ruby_vm_at_exit registers a function func to be invoked when a VM passed away.
Definition vm.c:1017
int ruby_vm_destruct(ruby_vm_t *vm)
Destructs the passed VM.
Definition vm.c:3510
VALUE rb_f_sprintf(int argc, const VALUE *argv)
Identical to rb_str_format(), except how the arguments are arranged.
Definition sprintf.c:221
#define MEMCPY(p1, p2, type, n)
Handy macro to call memcpy.
Definition memory.h:372
#define MEMZERO(p, type, n)
Handy macro to erase a region of memory.
Definition memory.h:360
#define RB_GC_GUARD(v)
Prevents premature destruction of local objects.
Definition memory.h:167
VALUE type(ANYARGS)
ANYARGS-ed function type.
void rb_hash_foreach(VALUE q, int_type *w, VALUE e)
Iteration over the given hash.
#define RARRAY_LEN
Just another name of rb_array_len.
Definition rarray.h:50
static int RARRAY_LENINT(VALUE ary)
Identical to rb_array_len(), except it differs for the return type.
Definition rarray.h:280
#define RARRAY_AREF(a, i)
Definition rarray.h:402
static VALUE RBASIC_CLASS(VALUE obj)
Queries the class of an object.
Definition rbasic.h:166
#define RBASIC(obj)
Convenient casting macro.
Definition rbasic.h:40
#define RHASH_EMPTY_P(h)
Checks if the hash is empty.
Definition rhash.h:79
#define StringValuePtr(v)
Identical to StringValue, except it returns a char*.
Definition rstring.h:76
static int RSTRING_LENINT(VALUE str)
Identical to RSTRING_LEN(), except it differs for the return type.
Definition rstring.h:438
#define RTYPEDDATA_DATA(v)
Convenient getter macro.
Definition rtypeddata.h:106
#define TypedData_Wrap_Struct(klass, data_type, sval)
Converts sval, a pointer to your struct, into a Ruby object.
Definition rtypeddata.h:557
#define TypedData_Make_Struct(klass, type, data_type, sval)
Identical to TypedData_Wrap_Struct, except it allocates a new data region internally instead of takin...
Definition rtypeddata.h:604
const char * rb_class2name(VALUE klass)
Queries the name of the passed class.
Definition variable.c:520
#define RB_NO_KEYWORDS
Do not pass keywords.
Definition scan_args.h:69
static bool RB_SPECIAL_CONST_P(VALUE obj)
Checks if the given object is of enum ruby_special_consts.
#define RTEST
This is an old name of RB_TEST.
#define _(args)
This was a transition path from K&R to ANSI.
Definition stdarg.h:35
Definition proc.c:31
Definition iseq.h:286
Internal header for Ruby Box.
Definition box.h:14
Definition method.h:63
CREF (Class REFerence)
Definition method.h:45
This is the struct that holds necessary info for a struct.
Definition rtypeddata.h:238
Definition method.h:55
Definition st.h:79
IFUNC (Internal FUNCtion)
Definition imemo.h:87
SVAR (Special VARiable)
Definition imemo.h:52
const VALUE cref_or_me
class reference or rb_method_entry_t
Definition imemo.h:54
THROW_DATA.
Definition imemo.h:61
void rb_native_cond_initialize(rb_nativethread_cond_t *cond)
Fills the passed condition variable with an initial value.
void rb_native_mutex_initialize(rb_nativethread_lock_t *lock)
Just another name of rb_nativethread_lock_initialize.
void rb_native_mutex_destroy(rb_nativethread_lock_t *lock)
Just another name of rb_nativethread_lock_destroy.
void rb_native_cond_destroy(rb_nativethread_cond_t *cond)
Destroys the passed condition variable.
uintptr_t ID
Type that represents a Ruby identifier such as a variable name.
Definition value.h:52
uintptr_t VALUE
Type that represents a Ruby object.
Definition value.h:40
static enum ruby_value_type RB_BUILTIN_TYPE(VALUE obj)
Queries the type of the object.
Definition value_type.h:182
static void Check_Type(VALUE v, enum ruby_value_type t)
Identical to RB_TYPE_P(), except it raises exceptions on predication failure.
Definition value_type.h:425
static bool RB_TYPE_P(VALUE obj, enum ruby_value_type t)
Queries if the given object is of given type.
Definition value_type.h:376
ruby_value_type
C-level type of an object.
Definition value_type.h:113