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