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