12#ifdef THREAD_SYSTEM_DEPENDENT_IMPLEMENTATION
14#include "internal/gc.h"
15#include "internal/sanitizers.h"
17#ifdef HAVE_SYS_RESOURCE_H
18#include <sys/resource.h>
20#ifdef HAVE_THR_STKSEGMENT
23#if defined(HAVE_FCNTL_H)
25#elif defined(HAVE_SYS_FCNTL_H)
28#ifdef HAVE_SYS_PRCTL_H
31#if defined(HAVE_SYS_TIME_H)
38#include <sys/syscall.h>
44# include <AvailabilityMacros.h>
47#if defined(HAVE_SYS_EVENTFD_H) && defined(HAVE_EVENTFD)
48# define USE_EVENTFD (1)
49# include <sys/eventfd.h>
51# define USE_EVENTFD (0)
54#if defined(HAVE_PTHREAD_CONDATTR_SETCLOCK) && \
55 defined(CLOCK_REALTIME) && defined(CLOCK_MONOTONIC) && \
56 defined(HAVE_CLOCK_GETTIME)
57static pthread_condattr_t condattr_mono;
58static pthread_condattr_t *condattr_monotonic = &condattr_mono;
60static const void *
const condattr_monotonic = NULL;
65#ifndef HAVE_SYS_EVENT_H
66#define HAVE_SYS_EVENT_H 0
69#ifndef HAVE_SYS_EPOLL_H
70#define HAVE_SYS_EPOLL_H 0
78 #if defined(__EMSCRIPTEN__) || defined(COROUTINE_PTHREAD_CONTEXT)
81 #define USE_MN_THREADS 0
82 #elif HAVE_SYS_EPOLL_H
83 #include <sys/epoll.h>
84 #define USE_MN_THREADS 1
85 #elif HAVE_SYS_EVENT_H
86 #include <sys/event.h>
87 #define USE_MN_THREADS 1
89 #define USE_MN_THREADS 0
93#ifdef HAVE_SCHED_YIELD
94#define native_thread_yield() (void)sched_yield()
96#define native_thread_yield() ((void)0)
101#define NATIVE_MUTEX_LOCK_DEBUG 0
102#define NATIVE_MUTEX_LOCK_DEBUG_YIELD 0
105mutex_debug(
const char *msg,
void *lock)
107 if (NATIVE_MUTEX_LOCK_DEBUG) {
109 static pthread_mutex_t dbglock = PTHREAD_MUTEX_INITIALIZER;
111 if ((r = pthread_mutex_lock(&dbglock)) != 0) {exit(EXIT_FAILURE);}
112 fprintf(stdout,
"%s: %p\n", msg, lock);
113 if ((r = pthread_mutex_unlock(&dbglock)) != 0) {exit(EXIT_FAILURE);}
121#if NATIVE_MUTEX_LOCK_DEBUG_YIELD
122 native_thread_yield();
124 mutex_debug(
"lock", lock);
125 if ((r = pthread_mutex_lock(lock)) != 0) {
134 mutex_debug(
"unlock", lock);
135 if ((r = pthread_mutex_unlock(lock)) != 0) {
144 mutex_debug(
"trylock", lock);
145 if ((r = pthread_mutex_trylock(lock)) != 0) {
159 int r = pthread_mutex_init(lock, 0);
160 mutex_debug(
"init", lock);
169 int r = pthread_mutex_destroy(lock);
170 mutex_debug(
"destroy", lock);
179 int r = pthread_cond_init(cond, condattr_monotonic);
188 int r = pthread_cond_destroy(cond);
209 r = pthread_cond_signal(cond);
210 }
while (r == EAGAIN);
221 r = pthread_cond_broadcast(cond);
222 }
while (r == EAGAIN);
231 int r = pthread_cond_wait(cond, mutex);
238native_cond_timedwait(rb_nativethread_cond_t *cond, pthread_mutex_t *mutex,
const rb_hrtime_t *abs)
250 rb_hrtime2timespec(&ts, abs);
251 r = pthread_cond_timedwait(cond, mutex, &ts);
252 }
while (r == EINTR);
254 if (r != 0 && r != ETIMEDOUT) {
262native_cond_timeout(rb_nativethread_cond_t *cond,
const rb_hrtime_t rel)
264 if (condattr_monotonic) {
265 return rb_hrtime_add(rb_hrtime_now(), rel);
271 return rb_hrtime_add(rb_timespec2hrtime(&ts), rel);
278 rb_hrtime_t hrmsec = native_cond_timeout(cond, RB_HRTIME_PER_MSEC * msec);
279 native_cond_timedwait(cond, mutex, &hrmsec);
284static rb_internal_thread_event_hook_t *rb_internal_thread_event_hooks = NULL;
285static void rb_thread_execute_hooks(
rb_event_flag_t event, rb_thread_t *th);
306#define RB_INTERNAL_THREAD_HOOK(event, th) \
307 if (UNLIKELY(rb_internal_thread_event_hooks)) { \
308 fprintf(stderr, "[thread=%"PRIxVALUE"] %s in %s (%s:%d)\n", th->self, event_name(event), __func__, __FILE__, __LINE__); \
309 rb_thread_execute_hooks(event, th); \
312#define RB_INTERNAL_THREAD_HOOK(event, th) if (UNLIKELY(rb_internal_thread_event_hooks)) { rb_thread_execute_hooks(event, th); }
315static rb_serial_t current_fork_gen = 1;
317#if defined(SIGVTALRM) && !defined(__EMSCRIPTEN__)
318# define USE_UBF_LIST 1
321static void threadptr_trap_interrupt(rb_thread_t *);
323static void native_thread_dedicated_inc(rb_vm_t *vm, rb_ractor_t *cr,
struct rb_native_thread *nt);
324static void native_thread_dedicated_dec(rb_vm_t *vm, rb_ractor_t *cr,
struct rb_native_thread *nt);
325static void native_thread_assign(
struct rb_native_thread *nt, rb_thread_t *th);
327static void ractor_sched_enq(rb_vm_t *vm, rb_ractor_t *r);
328static void timer_thread_wakeup(
void);
329static void timer_thread_wakeup_locked(rb_vm_t *vm);
330static void timer_thread_wakeup_force(
void);
331static void thread_sched_switch(rb_thread_t *cth, rb_thread_t *next_th);
335#define thread_sched_dump(s) thread_sched_dump_(__FILE__, __LINE__, s)
338th_has_dedicated_nt(
const rb_thread_t *th)
341 return th->nt->dedicated > 0;
346thread_sched_dump_(const
char *file,
int line, struct
rb_thread_sched *sched)
348 fprintf(stderr,
"@%s:%d running:%d\n", file, line, sched->running ? (
int)sched->running->serial : -1);
351 ccan_list_for_each(&sched->readyq, th, sched.node.readyq) {
352 i++;
if (i>10) rb_bug(
"too many");
353 fprintf(stderr,
" ready:%d (%sNT:%d)\n", th->serial,
354 th->nt ? (th->nt->dedicated ?
"D" :
"S") :
"x",
355 th->nt ? (
int)th->nt->serial : -1);
359#define ractor_sched_dump(s) ractor_sched_dump_(__FILE__, __LINE__, s)
363ractor_sched_dump_(const
char *file,
int line, rb_vm_t *vm)
367 fprintf(stderr,
"ractor_sched_dump %s:%d\n", file, line);
370 ccan_list_for_each(&vm->ractor.sched.grq, r, threads.sched.grq_node) {
372 if (i>10) rb_bug(
"!!");
373 fprintf(stderr,
" %d ready:%d\n", i, rb_ractor_id(r));
377#define thread_sched_lock(a, b) thread_sched_lock_(a, b, __FILE__, __LINE__)
378#define thread_sched_unlock(a, b) thread_sched_unlock_(a, b, __FILE__, __LINE__)
384 VM_ASSERT(sched->lock_owner == NULL);
386 sched->lock_owner = th;
391thread_sched_set_unlocked(
struct rb_thread_sched *sched, rb_thread_t *th)
394 VM_ASSERT(sched->lock_owner == th);
396 sched->lock_owner = NULL;
401thread_sched_lock_(
struct rb_thread_sched *sched, rb_thread_t *th,
const char *file,
int line)
406 RUBY_DEBUG_LOG2(file, line,
"r:%d th:%u", th ? (
int)rb_ractor_id(th->ractor) : -1, rb_th_serial(th));
408 RUBY_DEBUG_LOG2(file, line,
"th:%u", rb_th_serial(th));
411 thread_sched_set_locked(sched, th);
415thread_sched_unlock_(
struct rb_thread_sched *sched, rb_thread_t *th,
const char *file,
int line)
417 RUBY_DEBUG_LOG2(file, line,
"th:%u", rb_th_serial(th));
419 thread_sched_set_unlocked(sched, th);
425ASSERT_thread_sched_locked(
struct rb_thread_sched *sched, rb_thread_t *th)
431 VM_ASSERT(sched->lock_owner == th);
434 VM_ASSERT(sched->lock_owner != NULL);
439#define ractor_sched_lock(a, b) ractor_sched_lock_(a, b, __FILE__, __LINE__)
440#define ractor_sched_unlock(a, b) ractor_sched_unlock_(a, b, __FILE__, __LINE__)
444rb_ractor_serial(const rb_ractor_t *r)
447 return rb_ractor_id(r);
455ractor_sched_set_locked(rb_vm_t *vm, rb_ractor_t *cr)
458 VM_ASSERT(vm->ractor.sched.lock_owner == NULL);
459 VM_ASSERT(vm->ractor.sched.locked ==
false);
461 vm->ractor.sched.lock_owner = cr;
462 vm->ractor.sched.locked =
true;
467ractor_sched_set_unlocked(rb_vm_t *vm, rb_ractor_t *cr)
470 VM_ASSERT(vm->ractor.sched.locked);
471 VM_ASSERT(vm->ractor.sched.lock_owner == cr);
473 vm->ractor.sched.locked =
false;
474 vm->ractor.sched.lock_owner = NULL;
479ractor_sched_lock_(rb_vm_t *vm, rb_ractor_t *cr,
const char *file,
int line)
484 RUBY_DEBUG_LOG2(file, line,
"cr:%u prev_owner:%u", rb_ractor_serial(cr), rb_ractor_serial(vm->ractor.sched.lock_owner));
486 RUBY_DEBUG_LOG2(file, line,
"cr:%u", rb_ractor_serial(cr));
489 ractor_sched_set_locked(vm, cr);
493ractor_sched_unlock_(rb_vm_t *vm, rb_ractor_t *cr,
const char *file,
int line)
495 RUBY_DEBUG_LOG2(file, line,
"cr:%u", rb_ractor_serial(cr));
497 ractor_sched_set_unlocked(vm, cr);
502ASSERT_ractor_sched_locked(rb_vm_t *vm, rb_ractor_t *cr)
505 VM_ASSERT(vm->ractor.sched.locked);
506 VM_ASSERT(cr == NULL || vm->ractor.sched.lock_owner == cr);
511ractor_sched_running_threads_contain_p(rb_vm_t *vm, rb_thread_t *th)
514 ccan_list_for_each(&vm->ractor.sched.running_threads, rth, sched.node.running_threads) {
515 if (rth == th)
return true;
522ractor_sched_running_threads_size(rb_vm_t *vm)
526 ccan_list_for_each(&vm->ractor.sched.running_threads, th, sched.node.running_threads) {
534ractor_sched_timeslice_threads_size(rb_vm_t *vm)
538 ccan_list_for_each(&vm->ractor.sched.timeslice_threads, th, sched.node.timeslice_threads) {
546ractor_sched_timeslice_threads_contain_p(rb_vm_t *vm, rb_thread_t *th)
549 ccan_list_for_each(&vm->ractor.sched.timeslice_threads, rth, sched.node.timeslice_threads) {
550 if (rth == th)
return true;
555static void ractor_sched_barrier_join_signal_locked(rb_vm_t *vm);
559thread_sched_setup_running_threads(
struct rb_thread_sched *sched, rb_ractor_t *cr, rb_vm_t *vm,
560 rb_thread_t *add_th, rb_thread_t *del_th, rb_thread_t *add_timeslice_th)
562#if USE_RUBY_DEBUG_LOG
563 unsigned int prev_running_cnt = vm->ractor.sched.running_cnt;
566 rb_thread_t *del_timeslice_th;
568 if (del_th && sched->is_running_timeslice) {
569 del_timeslice_th = del_th;
570 sched->is_running_timeslice =
false;
573 del_timeslice_th = NULL;
576 RUBY_DEBUG_LOG(
"+:%u -:%u +ts:%u -ts:%u",
577 rb_th_serial(add_th), rb_th_serial(del_th),
578 rb_th_serial(add_timeslice_th), rb_th_serial(del_timeslice_th));
580 ractor_sched_lock(vm, cr);
584 VM_ASSERT(ractor_sched_running_threads_contain_p(vm, del_th));
585 VM_ASSERT(del_timeslice_th != NULL ||
586 !ractor_sched_timeslice_threads_contain_p(vm, del_th));
588 ccan_list_del_init(&del_th->sched.node.running_threads);
589 vm->ractor.sched.running_cnt--;
591 if (UNLIKELY(vm->ractor.sched.barrier_waiting)) {
592 ractor_sched_barrier_join_signal_locked(vm);
594 sched->is_running =
false;
598 if (vm->ractor.sched.barrier_waiting) {
600 RUBY_DEBUG_LOG(
"barrier_waiting");
601 RUBY_VM_SET_VM_BARRIER_INTERRUPT(add_th->ec);
604 VM_ASSERT(!ractor_sched_running_threads_contain_p(vm, add_th));
605 VM_ASSERT(!ractor_sched_timeslice_threads_contain_p(vm, add_th));
607 ccan_list_add(&vm->ractor.sched.running_threads, &add_th->sched.node.running_threads);
608 vm->ractor.sched.running_cnt++;
609 sched->is_running =
true;
612 if (add_timeslice_th) {
614 int was_empty = ccan_list_empty(&vm->ractor.sched.timeslice_threads);
615 VM_ASSERT(!ractor_sched_timeslice_threads_contain_p(vm, add_timeslice_th));
616 ccan_list_add(&vm->ractor.sched.timeslice_threads, &add_timeslice_th->sched.node.timeslice_threads);
617 sched->is_running_timeslice =
true;
619 timer_thread_wakeup_locked(vm);
623 if (del_timeslice_th) {
624 VM_ASSERT(ractor_sched_timeslice_threads_contain_p(vm, del_timeslice_th));
625 ccan_list_del_init(&del_timeslice_th->sched.node.timeslice_threads);
628 VM_ASSERT(ractor_sched_running_threads_size(vm) == vm->ractor.sched.running_cnt);
629 VM_ASSERT(ractor_sched_timeslice_threads_size(vm) <= vm->ractor.sched.running_cnt);
631 ractor_sched_unlock(vm, cr);
636 RUBY_DEBUG_LOG(
"run:%u->%u", prev_running_cnt, vm->ractor.sched.running_cnt);
640thread_sched_add_running_thread(
struct rb_thread_sched *sched, rb_thread_t *th)
642 ASSERT_thread_sched_locked(sched, th);
643 VM_ASSERT(sched->running == th);
645 rb_vm_t *vm = th->vm;
646 thread_sched_setup_running_threads(sched, th->ractor, vm, th, NULL, ccan_list_empty(&sched->readyq) ? NULL : th);
650thread_sched_del_running_thread(
struct rb_thread_sched *sched, rb_thread_t *th)
652 ASSERT_thread_sched_locked(sched, th);
654 rb_vm_t *vm = th->vm;
655 thread_sched_setup_running_threads(sched, th->ractor, vm, NULL, th, NULL);
659rb_add_running_thread(rb_thread_t *th)
663 thread_sched_lock(sched, th);
665 thread_sched_add_running_thread(sched, th);
667 thread_sched_unlock(sched, th);
671rb_del_running_thread(rb_thread_t *th)
675 thread_sched_lock(sched, th);
677 thread_sched_del_running_thread(sched, th);
679 thread_sched_unlock(sched, th);
687thread_sched_set_running(
struct rb_thread_sched *sched, rb_thread_t *th)
689 RUBY_DEBUG_LOG(
"th:%u->th:%u", rb_th_serial(sched->running), rb_th_serial(th));
690 VM_ASSERT(sched->running != th);
697thread_sched_readyq_contain_p(struct
rb_thread_sched *sched, rb_thread_t *th)
700 ccan_list_for_each(&sched->readyq, rth, sched.node.readyq) {
701 if (rth == th)
return true;
713 ASSERT_thread_sched_locked(sched, NULL);
714 rb_thread_t *next_th;
716 VM_ASSERT(sched->running != NULL);
718 if (ccan_list_empty(&sched->readyq)) {
722 next_th = ccan_list_pop(&sched->readyq, rb_thread_t, sched.node.readyq);
724 VM_ASSERT(sched->readyq_cnt > 0);
726 ccan_list_node_init(&next_th->sched.node.readyq);
729 RUBY_DEBUG_LOG(
"next_th:%u readyq_cnt:%d", rb_th_serial(next_th), sched->readyq_cnt);
738 ASSERT_thread_sched_locked(sched, NULL);
739 RUBY_DEBUG_LOG(
"ready_th:%u readyq_cnt:%d", rb_th_serial(ready_th), sched->readyq_cnt);
741 VM_ASSERT(sched->running != NULL);
742 VM_ASSERT(!thread_sched_readyq_contain_p(sched, ready_th));
744 if (sched->is_running) {
745 if (ccan_list_empty(&sched->readyq)) {
747 thread_sched_setup_running_threads(sched, ready_th->ractor, ready_th->vm, NULL, NULL, sched->running);
755 ccan_list_add_tail(&sched->readyq, &ready_th->sched.node.readyq);
762thread_sched_wakeup_running_thread(
struct rb_thread_sched *sched, rb_thread_t *next_th,
bool will_switch)
764 ASSERT_thread_sched_locked(sched, NULL);
765 VM_ASSERT(sched->running == next_th);
769 if (th_has_dedicated_nt(next_th)) {
770 RUBY_DEBUG_LOG(
"pinning th:%u", next_th->serial);
775 RUBY_DEBUG_LOG(
"th:%u is already running.", next_th->serial);
780 RUBY_DEBUG_LOG(
"th:%u (do nothing)", rb_th_serial(next_th));
783 RUBY_DEBUG_LOG(
"th:%u (enq)", rb_th_serial(next_th));
784 ractor_sched_enq(next_th->vm, next_th->ractor);
789 RUBY_DEBUG_LOG(
"no waiting threads%s",
"");
795thread_sched_to_ready_common(
struct rb_thread_sched *sched, rb_thread_t *th,
bool wakeup,
bool will_switch)
797 RUBY_DEBUG_LOG(
"th:%u running:%u redyq_cnt:%d", rb_th_serial(th), rb_th_serial(sched->running), sched->readyq_cnt);
799 VM_ASSERT(sched->running != th);
800 VM_ASSERT(!thread_sched_readyq_contain_p(sched, th));
803 if (sched->running == NULL) {
804 thread_sched_set_running(sched, th);
805 if (wakeup) thread_sched_wakeup_running_thread(sched, th, will_switch);
808 thread_sched_enq(sched, th);
820 RUBY_DEBUG_LOG(
"th:%u", rb_th_serial(th));
822 thread_sched_lock(sched, th);
824 thread_sched_to_ready_common(sched, th,
true,
false);
826 thread_sched_unlock(sched, th);
831thread_sched_wait_running_turn(
struct rb_thread_sched *sched, rb_thread_t *th,
bool can_direct_transfer)
833 RUBY_DEBUG_LOG(
"th:%u", rb_th_serial(th));
835 ASSERT_thread_sched_locked(sched, th);
836 VM_ASSERT(th == rb_ec_thread_ptr(rb_current_ec_noinline()));
838 if (th != sched->running) {
843 rb_thread_t *next_th;
844 while((next_th = sched->running) != th) {
845 if (th_has_dedicated_nt(th)) {
846 RUBY_DEBUG_LOG(
"(nt) sleep th:%u running:%u", rb_th_serial(th), rb_th_serial(sched->running));
848 thread_sched_set_unlocked(sched, th);
850 RUBY_DEBUG_LOG(
"nt:%d cond:%p", th->nt->serial, &th->nt->cond.readyq);
853 thread_sched_set_locked(sched, th);
855 RUBY_DEBUG_LOG(
"(nt) wakeup %s", sched->running == th ?
"success" :
"failed");
856 if (th == sched->running) {
857 rb_ractor_thread_switch(th->ractor, th,
false);
862 if (can_direct_transfer &&
863 (next_th = sched->running) != NULL &&
867 RUBY_DEBUG_LOG(
"th:%u->%u (direct)", rb_th_serial(th), rb_th_serial(next_th));
869 thread_sched_set_unlocked(sched, th);
871 rb_ractor_set_current_ec(th->ractor, NULL);
872 thread_sched_switch(th, next_th);
874 thread_sched_set_locked(sched, th);
879 native_thread_assign(NULL, th);
881 RUBY_DEBUG_LOG(
"th:%u->%u (ractor scheduling)", rb_th_serial(th), rb_th_serial(next_th));
883 thread_sched_set_unlocked(sched, th);
885 rb_ractor_set_current_ec(th->ractor, NULL);
886 coroutine_transfer0(th->sched.context, nt->nt_context,
false);
888 thread_sched_set_locked(sched, th);
891 VM_ASSERT(rb_current_ec_noinline() == th->ec);
895 VM_ASSERT(th->nt != NULL);
896 VM_ASSERT(rb_current_ec_noinline() == th->ec);
897 VM_ASSERT(th->sched.waiting_reason.flags == thread_sched_waiting_none);
900 thread_sched_add_running_thread(sched, th);
909thread_sched_to_running_common(
struct rb_thread_sched *sched, rb_thread_t *th)
911 RUBY_DEBUG_LOG(
"th:%u dedicated:%d", rb_th_serial(th), th_has_dedicated_nt(th));
913 VM_ASSERT(sched->running != th);
914 VM_ASSERT(th_has_dedicated_nt(th));
915 VM_ASSERT(GET_THREAD() == th);
917 native_thread_dedicated_dec(th->vm, th->ractor, th->nt);
920 thread_sched_to_ready_common(sched, th,
false,
false);
922 if (sched->running == th) {
923 thread_sched_add_running_thread(sched, th);
927 thread_sched_wait_running_turn(sched, th,
false);
939 thread_sched_lock(sched, th);
941 thread_sched_to_running_common(sched, th);
943 thread_sched_unlock(sched, th);
954thread_sched_wakeup_next_thread(
struct rb_thread_sched *sched, rb_thread_t *th,
bool will_switch)
956 ASSERT_thread_sched_locked(sched, th);
958 VM_ASSERT(sched->running == th);
959 VM_ASSERT(sched->running->nt != NULL);
961 rb_thread_t *next_th = thread_sched_deq(sched);
963 RUBY_DEBUG_LOG(
"next_th:%u", rb_th_serial(next_th));
964 VM_ASSERT(th != next_th);
966 thread_sched_set_running(sched, next_th);
967 VM_ASSERT(next_th == sched->running);
968 thread_sched_wakeup_running_thread(sched, next_th, will_switch);
971 thread_sched_del_running_thread(sched, th);
977thread_sched_to_dead_common(
struct rb_thread_sched *sched, rb_thread_t *th)
979 RUBY_DEBUG_LOG(
"th:%u DNT:%d", rb_th_serial(th), th->nt->dedicated);
983 thread_sched_wakeup_next_thread(sched, th, !th_has_dedicated_nt(th));
992 thread_sched_lock(sched, th);
994 thread_sched_to_dead_common(sched, th);
996 thread_sched_unlock(sched, th);
1003thread_sched_to_waiting_common(
struct rb_thread_sched *sched, rb_thread_t *th)
1005 RUBY_DEBUG_LOG(
"th:%u DNT:%d", rb_th_serial(th), th->nt->dedicated);
1009 native_thread_dedicated_inc(th->vm, th->ractor, th->nt);
1010 thread_sched_wakeup_next_thread(sched, th,
false);
1017thread_sched_to_waiting(
struct rb_thread_sched *sched, rb_thread_t *th)
1019 thread_sched_lock(sched, th);
1021 thread_sched_to_waiting_common(sched, th);
1023 thread_sched_unlock(sched, th);
1031 VM_ASSERT(func != NULL);
1034 if (RUBY_VM_INTERRUPTED(th->ec)) {
1035 RUBY_DEBUG_LOG(
"interrupted:0x%x", th->ec->interrupt_flag);
1041 if (!th->ec->raised_flag && RUBY_VM_INTERRUPTED(th->ec)) {
1046 VM_ASSERT(th->unblock.func == NULL);
1047 th->unblock.func = func;
1048 th->unblock.arg = arg;
1056ubf_clear(rb_thread_t *th)
1058 if (th->unblock.func) {
1061 th->unblock.func = NULL;
1062 th->unblock.arg = NULL;
1069ubf_waiting(
void *ptr)
1071 rb_thread_t *th = (rb_thread_t *)ptr;
1075 th->unblock.func = NULL;
1076 th->unblock.arg = NULL;
1078 RUBY_DEBUG_LOG(
"th:%u", rb_th_serial(th));
1080 thread_sched_lock(sched, th);
1082 if (sched->running == th) {
1086 thread_sched_to_ready_common(sched, th,
true,
false);
1089 thread_sched_unlock(sched, th);
1096thread_sched_to_waiting_until_wakeup(
struct rb_thread_sched *sched, rb_thread_t *th)
1098 RUBY_DEBUG_LOG(
"th:%u", rb_th_serial(th));
1100 RB_VM_SAVE_MACHINE_CONTEXT(th);
1105 thread_sched_lock(sched, th);
1108 if (ubf_set(th, ubf_waiting, (
void *)th)) {
1109 RUBY_DEBUG_LOG(
"th:%u interrupted", rb_th_serial(th));
1112 bool can_direct_transfer = !th_has_dedicated_nt(th);
1114 thread_sched_wakeup_next_thread(sched, th, can_direct_transfer);
1115 thread_sched_wait_running_turn(sched, th, can_direct_transfer);
1118 thread_sched_unlock(sched, th);
1128 RUBY_DEBUG_LOG(
"th:%d sched->readyq_cnt:%d", (
int)th->serial, sched->readyq_cnt);
1130 thread_sched_lock(sched, th);
1132 if (!ccan_list_empty(&sched->readyq)) {
1134 thread_sched_wakeup_next_thread(sched, th, !th_has_dedicated_nt(th));
1135 bool can_direct_transfer = !th_has_dedicated_nt(th);
1136 thread_sched_to_ready_common(sched, th,
false, can_direct_transfer);
1137 thread_sched_wait_running_turn(sched, th, can_direct_transfer);
1138 th->status = THREAD_RUNNABLE;
1141 VM_ASSERT(sched->readyq_cnt == 0);
1144 thread_sched_unlock(sched, th);
1153 sched->lock_owner = NULL;
1156 ccan_list_head_init(&sched->readyq);
1157 sched->readyq_cnt = 0;
1160 if (!atfork) sched->enable_mn_threads =
true;
1167#ifdef RUBY_ASAN_ENABLED
1168 void **fake_stack = to_dead ? NULL : &transfer_from->fake_stack;
1169 __sanitizer_start_switch_fiber(fake_stack, transfer_to->stack_base, transfer_to->stack_size);
1173 struct
coroutine_context *returning_from = coroutine_transfer(transfer_from, transfer_to);
1177 VM_ASSERT(!to_dead);
1178#ifdef RUBY_ASAN_ENABLED
1179 __sanitizer_finish_switch_fiber(transfer_from->fake_stack,
1180 (
const void**)&returning_from->stack_base, &returning_from->stack_size);
1188 VM_ASSERT(!nt->dedicated);
1189 VM_ASSERT(next_th->nt == NULL);
1191 RUBY_DEBUG_LOG(
"next_th:%u", rb_th_serial(next_th));
1193 ruby_thread_set_native(next_th);
1194 native_thread_assign(nt, next_th);
1196 coroutine_transfer0(current_cont, next_th->sched.context, to_dead);
1200thread_sched_switch(rb_thread_t *cth, rb_thread_t *next_th)
1203 native_thread_assign(NULL, cth);
1204 RUBY_DEBUG_LOG(
"th:%u->%u on nt:%d", rb_th_serial(cth), rb_th_serial(next_th), nt->serial);
1205 thread_sched_switch0(cth->sched.context, next_th, nt, cth->status == THREAD_KILLED);
1208#if VM_CHECK_MODE > 0
1211grq_size(rb_vm_t *vm, rb_ractor_t *cr)
1213 ASSERT_ractor_sched_locked(vm, cr);
1215 rb_ractor_t *r, *prev_r = NULL;
1218 ccan_list_for_each(&vm->ractor.sched.grq, r, threads.sched.grq_node) {
1221 VM_ASSERT(r != prev_r);
1229ractor_sched_enq(rb_vm_t *vm, rb_ractor_t *r)
1232 rb_ractor_t *cr = NULL;
1234 VM_ASSERT(sched->running != NULL);
1235 VM_ASSERT(sched->running->nt == NULL);
1237 ractor_sched_lock(vm, cr);
1239#if VM_CHECK_MODE > 0
1242 ccan_list_for_each(&vm->ractor.sched.grq,
tr, threads.sched.grq_node) {
1247 ccan_list_add_tail(&vm->ractor.sched.grq, &sched->grq_node);
1248 vm->ractor.sched.grq_cnt++;
1249 VM_ASSERT(grq_size(vm, cr) == vm->ractor.sched.grq_cnt);
1251 RUBY_DEBUG_LOG(
"r:%u th:%u grq_cnt:%u", rb_ractor_id(r), rb_th_serial(sched->running), vm->ractor.sched.grq_cnt);
1257 ractor_sched_unlock(vm, cr);
1261#ifndef SNT_KEEP_SECONDS
1262#define SNT_KEEP_SECONDS 0
1267#define MINIMUM_SNT 0
1271ractor_sched_deq(rb_vm_t *vm, rb_ractor_t *cr)
1275 ractor_sched_lock(vm, cr);
1277 RUBY_DEBUG_LOG(
"empty? %d", ccan_list_empty(&vm->ractor.sched.grq));
1280 VM_ASSERT(rb_current_execution_context(
false) == NULL);
1281 VM_ASSERT(grq_size(vm, cr) == vm->ractor.sched.grq_cnt);
1283 while ((r = ccan_list_pop(&vm->ractor.sched.grq, rb_ractor_t, threads.sched.grq_node)) == NULL) {
1284 RUBY_DEBUG_LOG(
"wait grq_cnt:%d", (
int)vm->ractor.sched.grq_cnt);
1286#if SNT_KEEP_SECONDS > 0
1287 rb_hrtime_t abs = rb_hrtime_add(rb_hrtime_now(), RB_HRTIME_PER_SEC * SNT_KEEP_SECONDS);
1288 if (native_cond_timedwait(&vm->ractor.sched.cond, &vm->ractor.sched.lock, &abs) == ETIMEDOUT) {
1289 RUBY_DEBUG_LOG(
"timeout, grq_cnt:%d", (
int)vm->ractor.sched.grq_cnt);
1290 VM_ASSERT(r == NULL);
1291 vm->ractor.sched.snt_cnt--;
1292 vm->ractor.sched.running_cnt--;
1296 RUBY_DEBUG_LOG(
"wakeup grq_cnt:%d", (
int)vm->ractor.sched.grq_cnt);
1299 ractor_sched_set_unlocked(vm, cr);
1301 ractor_sched_set_locked(vm, cr);
1303 RUBY_DEBUG_LOG(
"wakeup grq_cnt:%d", (
int)vm->ractor.sched.grq_cnt);
1307 VM_ASSERT(rb_current_execution_context(
false) == NULL);
1310 VM_ASSERT(vm->ractor.sched.grq_cnt > 0);
1311 vm->ractor.sched.grq_cnt--;
1312 RUBY_DEBUG_LOG(
"r:%d grq_cnt:%u", (
int)rb_ractor_id(r), vm->ractor.sched.grq_cnt);
1315 VM_ASSERT(SNT_KEEP_SECONDS > 0);
1319 ractor_sched_unlock(vm, cr);
1324void rb_ractor_lock_self(rb_ractor_t *r);
1325void rb_ractor_unlock_self(rb_ractor_t *r);
1330rb_ractor_sched_wait(rb_execution_context_t *ec, rb_ractor_t *cr,
rb_unblock_function_t *ubf,
void *ubf_arg)
1334 RUBY_DEBUG_LOG(
"start%s",
"");
1336 rb_thread_t *
volatile th = rb_ec_thread_ptr(ec);
1339 if (ubf_set(th, ubf, ubf_arg)) {
1344 thread_sched_lock(sched, th);
1345 rb_ractor_unlock_self(cr);
1348 bool can_direct_transfer = !th_has_dedicated_nt(th);
1349 RB_VM_SAVE_MACHINE_CONTEXT(th);
1350 th->status = THREAD_STOPPED_FOREVER;
1352 thread_sched_wakeup_next_thread(sched, th, can_direct_transfer);
1354 thread_sched_wait_running_turn(sched, th, can_direct_transfer);
1355 th->status = THREAD_RUNNABLE;
1357 thread_sched_unlock(sched, th);
1358 rb_ractor_lock_self(cr);
1362 RUBY_DEBUG_LOG(
"end%s",
"");
1366rb_ractor_sched_wakeup(rb_ractor_t *r, rb_thread_t *r_th)
1371 RUBY_DEBUG_LOG(
"r:%u th:%d", (
unsigned int)rb_ractor_id(r), r_th->serial);
1373 thread_sched_lock(sched, r_th);
1375 if (r_th->status == THREAD_STOPPED_FOREVER) {
1376 thread_sched_to_ready_common(sched, r_th,
true,
false);
1379 thread_sched_unlock(sched, r_th);
1383ractor_sched_barrier_completed_p(rb_vm_t *vm)
1385 RUBY_DEBUG_LOG(
"run:%u wait:%u", vm->ractor.sched.running_cnt, vm->ractor.sched.barrier_waiting_cnt);
1386 VM_ASSERT(vm->ractor.sched.running_cnt - 1 >= vm->ractor.sched.barrier_waiting_cnt);
1388 return (vm->ractor.sched.running_cnt - vm->ractor.sched.barrier_waiting_cnt) == 1;
1392rb_ractor_sched_barrier_start(rb_vm_t *vm, rb_ractor_t *cr)
1394 VM_ASSERT(cr == GET_RACTOR());
1395 VM_ASSERT(vm->ractor.sync.lock_owner == cr);
1396 VM_ASSERT(!vm->ractor.sched.barrier_waiting);
1397 VM_ASSERT(vm->ractor.sched.barrier_waiting_cnt == 0);
1398 VM_ASSERT(vm->ractor.sched.barrier_ractor == NULL);
1399 VM_ASSERT(vm->ractor.sched.barrier_lock_rec == 0);
1401 RUBY_DEBUG_LOG(
"start serial:%u", vm->ractor.sched.barrier_serial);
1403 unsigned int lock_rec;
1405 ractor_sched_lock(vm, cr);
1407 vm->ractor.sched.barrier_waiting =
true;
1408 vm->ractor.sched.barrier_ractor = cr;
1409 vm->ractor.sched.barrier_lock_rec = vm->ractor.sync.lock_rec;
1412 lock_rec = vm->ractor.sync.lock_rec;
1413 vm->ractor.sync.lock_rec = 0;
1414 vm->ractor.sync.lock_owner = NULL;
1419 ccan_list_for_each(&vm->ractor.sched.running_threads, ith, sched.node.running_threads) {
1420 if (ith->ractor != cr) {
1421 RUBY_DEBUG_LOG(
"barrier request to th:%u", rb_th_serial(ith));
1422 RUBY_VM_SET_VM_BARRIER_INTERRUPT(ith->ec);
1427 while (!ractor_sched_barrier_completed_p(vm)) {
1428 ractor_sched_set_unlocked(vm, cr);
1430 ractor_sched_set_locked(vm, cr);
1433 RUBY_DEBUG_LOG(
"completed seirial:%u", vm->ractor.sched.barrier_serial);
1436 vm->ractor.sched.barrier_serial++;
1437 vm->ractor.sched.barrier_waiting_cnt = 0;
1442 vm->ractor.sync.lock_rec = lock_rec;
1443 vm->ractor.sync.lock_owner = cr;
1452rb_ractor_sched_barrier_end(rb_vm_t *vm, rb_ractor_t *cr)
1454 RUBY_DEBUG_LOG(
"serial:%u", (
unsigned int)vm->ractor.sched.barrier_serial - 1);
1455 VM_ASSERT(vm->ractor.sched.barrier_waiting);
1456 VM_ASSERT(vm->ractor.sched.barrier_ractor);
1457 VM_ASSERT(vm->ractor.sched.barrier_lock_rec > 0);
1459 vm->ractor.sched.barrier_waiting =
false;
1460 vm->ractor.sched.barrier_ractor = NULL;
1461 vm->ractor.sched.barrier_lock_rec = 0;
1462 ractor_sched_unlock(vm, cr);
1466ractor_sched_barrier_join_signal_locked(rb_vm_t *vm)
1468 if (ractor_sched_barrier_completed_p(vm)) {
1474ractor_sched_barrier_join_wait_locked(rb_vm_t *vm, rb_thread_t *th)
1476 VM_ASSERT(vm->ractor.sched.barrier_waiting);
1478 unsigned int barrier_serial = vm->ractor.sched.barrier_serial;
1480 while (vm->ractor.sched.barrier_serial == barrier_serial) {
1481 RUBY_DEBUG_LOG(
"sleep serial:%u", barrier_serial);
1482 RB_VM_SAVE_MACHINE_CONTEXT(th);
1484 rb_ractor_t *cr = th->ractor;
1485 ractor_sched_set_unlocked(vm, cr);
1487 ractor_sched_set_locked(vm, cr);
1489 RUBY_DEBUG_LOG(
"wakeup serial:%u", barrier_serial);
1494rb_ractor_sched_barrier_join(rb_vm_t *vm, rb_ractor_t *cr)
1496 VM_ASSERT(cr->threads.sched.running != NULL);
1497 VM_ASSERT(cr == GET_RACTOR());
1498 VM_ASSERT(vm->ractor.sync.lock_owner == NULL);
1499 VM_ASSERT(vm->ractor.sched.barrier_waiting);
1501#if USE_RUBY_DEBUG_LOG || VM_CHECK_MODE > 0
1502 unsigned int barrier_serial = vm->ractor.sched.barrier_serial;
1505 RUBY_DEBUG_LOG(
"join");
1509 VM_ASSERT(vm->ractor.sched.barrier_waiting);
1510 VM_ASSERT(vm->ractor.sched.barrier_serial == barrier_serial);
1512 ractor_sched_lock(vm, cr);
1515 vm->ractor.sched.barrier_waiting_cnt++;
1516 RUBY_DEBUG_LOG(
"waiting_cnt:%u serial:%u", vm->ractor.sched.barrier_waiting_cnt, barrier_serial);
1518 ractor_sched_barrier_join_signal_locked(vm);
1519 ractor_sched_barrier_join_wait_locked(vm, cr->threads.sched.running);
1521 ractor_sched_unlock(vm, cr);
1531static void clear_thread_cache_altstack(
void);
1544 clear_thread_cache_altstack();
1548#ifdef RB_THREAD_T_HAS_NATIVE_ID
1550get_native_thread_id(
void)
1553 return (
int)syscall(SYS_gettid);
1554#elif defined(__FreeBSD__)
1555 return pthread_getthreadid_np();
1560#if defined(HAVE_WORKING_FORK)
1561void rb_internal_thread_event_hooks_rw_lock_atfork(
void);
1567 rb_thread_sched_init(sched,
true);
1568 rb_thread_t *th = GET_THREAD();
1569 rb_vm_t *vm = GET_VM();
1571 if (th_has_dedicated_nt(th)) {
1572 vm->ractor.sched.snt_cnt = 0;
1575 vm->ractor.sched.snt_cnt = 1;
1577 vm->ractor.sched.running_cnt = 0;
1580#if VM_CHECK_MODE > 0
1581 vm->ractor.sched.lock_owner = NULL;
1582 vm->ractor.sched.locked =
false;
1590 ccan_list_head_init(&vm->ractor.sched.grq);
1591 ccan_list_head_init(&vm->ractor.sched.timeslice_threads);
1592 ccan_list_head_init(&vm->ractor.sched.running_threads);
1594 rb_internal_thread_event_hooks_rw_lock_atfork();
1596 VM_ASSERT(sched->is_running);
1597 sched->is_running_timeslice =
false;
1599 if (sched->running != th) {
1600 thread_sched_to_running(sched, th);
1603 thread_sched_setup_running_threads(sched, th->ractor, vm, th, NULL, NULL);
1606#ifdef RB_THREAD_T_HAS_NATIVE_ID
1608 th->nt->tid = get_native_thread_id();
1615#ifdef RB_THREAD_LOCAL_SPECIFIER
1616static RB_THREAD_LOCAL_SPECIFIER rb_thread_t *ruby_native_thread;
1618static pthread_key_t ruby_native_thread_key;
1630ruby_thread_from_native(
void)
1632#ifdef RB_THREAD_LOCAL_SPECIFIER
1633 return ruby_native_thread;
1635 return pthread_getspecific(ruby_native_thread_key);
1640ruby_thread_set_native(rb_thread_t *th)
1644 ccan_list_node_init(&th->sched.node.ubf);
1651 rb_ractor_set_current_ec(th->ractor, th->ec);
1653#ifdef RB_THREAD_LOCAL_SPECIFIER
1654 ruby_native_thread = th;
1657 return pthread_setspecific(ruby_native_thread_key, th) == 0;
1665Init_native_thread(rb_thread_t *main_th)
1667#if defined(HAVE_PTHREAD_CONDATTR_SETCLOCK)
1668 if (condattr_monotonic) {
1669 int r = pthread_condattr_init(condattr_monotonic);
1671 r = pthread_condattr_setclock(condattr_monotonic, CLOCK_MONOTONIC);
1673 if (r) condattr_monotonic = NULL;
1677#ifndef RB_THREAD_LOCAL_SPECIFIER
1678 if (pthread_key_create(&ruby_native_thread_key, 0) == EAGAIN) {
1679 rb_bug(
"pthread_key_create failed (ruby_native_thread_key)");
1681 if (pthread_key_create(&ruby_current_ec_key, 0) == EAGAIN) {
1682 rb_bug(
"pthread_key_create failed (ruby_current_ec_key)");
1685 ruby_posix_signal(SIGVTALRM, null_func);
1688 rb_vm_t *vm = main_th->vm;
1694 ccan_list_head_init(&vm->ractor.sched.grq);
1695 ccan_list_head_init(&vm->ractor.sched.timeslice_threads);
1696 ccan_list_head_init(&vm->ractor.sched.running_threads);
1699 main_th->nt->thread_id = pthread_self();
1700 main_th->nt->serial = 1;
1701#ifdef RUBY_NT_SERIAL
1704 ruby_thread_set_native(main_th);
1705 native_thread_setup(main_th->nt);
1706 native_thread_setup_on_thread(main_th->nt);
1708 TH_SCHED(main_th)->running = main_th;
1709 main_th->has_dedicated_nt = 1;
1711 thread_sched_setup_running_threads(TH_SCHED(main_th), main_th->ractor, vm, main_th, NULL, NULL);
1714 main_th->nt->dedicated = 1;
1715 main_th->nt->vm = vm;
1718 vm->ractor.sched.dnt_cnt = 1;
1721extern int ruby_mn_threads_enabled;
1724ruby_mn_threads_params(
void)
1726 rb_vm_t *vm = GET_VM();
1727 rb_ractor_t *main_ractor = GET_RACTOR();
1729 const char *mn_threads_cstr = getenv(
"RUBY_MN_THREADS");
1730 bool enable_mn_threads =
false;
1732 if (USE_MN_THREADS && mn_threads_cstr && (enable_mn_threads = atoi(mn_threads_cstr) > 0)) {
1734 ruby_mn_threads_enabled = 1;
1736 main_ractor->threads.sched.enable_mn_threads = enable_mn_threads;
1738 const char *max_cpu_cstr = getenv(
"RUBY_MAX_CPU");
1739 const int default_max_cpu = 8;
1740 int max_cpu = default_max_cpu;
1742 if (USE_MN_THREADS && max_cpu_cstr) {
1743 int given_max_cpu = atoi(max_cpu_cstr);
1744 if (given_max_cpu > 0) {
1745 max_cpu = given_max_cpu;
1749 vm->ractor.sched.max_cpu = max_cpu;
1753native_thread_dedicated_inc(rb_vm_t *vm, rb_ractor_t *cr,
struct rb_native_thread *nt)
1755 RUBY_DEBUG_LOG(
"nt:%d %d->%d", nt->serial, nt->dedicated, nt->dedicated + 1);
1757 if (nt->dedicated == 0) {
1758 ractor_sched_lock(vm, cr);
1760 vm->ractor.sched.snt_cnt--;
1761 vm->ractor.sched.dnt_cnt++;
1763 ractor_sched_unlock(vm, cr);
1770native_thread_dedicated_dec(rb_vm_t *vm, rb_ractor_t *cr,
struct rb_native_thread *nt)
1772 RUBY_DEBUG_LOG(
"nt:%d %d->%d", nt->serial, nt->dedicated, nt->dedicated - 1);
1773 VM_ASSERT(nt->dedicated > 0);
1776 if (nt->dedicated == 0) {
1777 ractor_sched_lock(vm, cr);
1779 nt->vm->ractor.sched.snt_cnt++;
1780 nt->vm->ractor.sched.dnt_cnt--;
1782 ractor_sched_unlock(vm, cr);
1789#if USE_RUBY_DEBUG_LOG
1792 RUBY_DEBUG_LOG(
"th:%d nt:%d->%d", (
int)th->serial, (
int)th->nt->serial, (
int)nt->serial);
1795 RUBY_DEBUG_LOG(
"th:%d nt:NULL->%d", (
int)th->serial, (
int)nt->serial);
1800 RUBY_DEBUG_LOG(
"th:%d nt:%d->NULL", (
int)th->serial, (
int)th->nt->serial);
1803 RUBY_DEBUG_LOG(
"th:%d nt:NULL->NULL", (
int)th->serial);
1826 RB_ALTSTACK_FREE(nt->altstack);
1838 if (&nt->cond.readyq != &nt->cond.intr) {
1842 native_thread_destroy_atfork(nt);
1846#if defined HAVE_PTHREAD_GETATTR_NP || defined HAVE_PTHREAD_ATTR_GET_NP
1847#define STACKADDR_AVAILABLE 1
1848#elif defined HAVE_PTHREAD_GET_STACKADDR_NP && defined HAVE_PTHREAD_GET_STACKSIZE_NP
1849#define STACKADDR_AVAILABLE 1
1850#undef MAINSTACKADDR_AVAILABLE
1851#define MAINSTACKADDR_AVAILABLE 1
1852void *pthread_get_stackaddr_np(pthread_t);
1853size_t pthread_get_stacksize_np(pthread_t);
1854#elif defined HAVE_THR_STKSEGMENT || defined HAVE_PTHREAD_STACKSEG_NP
1855#define STACKADDR_AVAILABLE 1
1856#elif defined HAVE_PTHREAD_GETTHRDS_NP
1857#define STACKADDR_AVAILABLE 1
1858#elif defined __HAIKU__
1859#define STACKADDR_AVAILABLE 1
1862#ifndef MAINSTACKADDR_AVAILABLE
1863# ifdef STACKADDR_AVAILABLE
1864# define MAINSTACKADDR_AVAILABLE 1
1866# define MAINSTACKADDR_AVAILABLE 0
1869#if MAINSTACKADDR_AVAILABLE && !defined(get_main_stack)
1870# define get_main_stack(addr, size) get_stack(addr, size)
1873#ifdef STACKADDR_AVAILABLE
1878get_stack(
void **addr,
size_t *size)
1880#define CHECK_ERR(expr) \
1881 {int err = (expr); if (err) return err;}
1882#ifdef HAVE_PTHREAD_GETATTR_NP
1883 pthread_attr_t attr;
1885 STACK_GROW_DIR_DETECTION;
1886 CHECK_ERR(pthread_getattr_np(pthread_self(), &attr));
1887# ifdef HAVE_PTHREAD_ATTR_GETSTACK
1888 CHECK_ERR(pthread_attr_getstack(&attr, addr, size));
1889 STACK_DIR_UPPER((
void)0, (
void)(*addr = (
char *)*addr + *size));
1891 CHECK_ERR(pthread_attr_getstackaddr(&attr, addr));
1892 CHECK_ERR(pthread_attr_getstacksize(&attr, size));
1894# ifdef HAVE_PTHREAD_ATTR_GETGUARDSIZE
1895 CHECK_ERR(pthread_attr_getguardsize(&attr, &guard));
1897 guard = getpagesize();
1900 pthread_attr_destroy(&attr);
1901#elif defined HAVE_PTHREAD_ATTR_GET_NP
1902 pthread_attr_t attr;
1903 CHECK_ERR(pthread_attr_init(&attr));
1904 CHECK_ERR(pthread_attr_get_np(pthread_self(), &attr));
1905# ifdef HAVE_PTHREAD_ATTR_GETSTACK
1906 CHECK_ERR(pthread_attr_getstack(&attr, addr, size));
1908 CHECK_ERR(pthread_attr_getstackaddr(&attr, addr));
1909 CHECK_ERR(pthread_attr_getstacksize(&attr, size));
1911 STACK_DIR_UPPER((
void)0, (
void)(*addr = (
char *)*addr + *size));
1912 pthread_attr_destroy(&attr);
1913#elif (defined HAVE_PTHREAD_GET_STACKADDR_NP && defined HAVE_PTHREAD_GET_STACKSIZE_NP)
1914 pthread_t th = pthread_self();
1915 *addr = pthread_get_stackaddr_np(th);
1916 *size = pthread_get_stacksize_np(th);
1917#elif defined HAVE_THR_STKSEGMENT || defined HAVE_PTHREAD_STACKSEG_NP
1919# if defined HAVE_THR_STKSEGMENT
1920 CHECK_ERR(thr_stksegment(&stk));
1922 CHECK_ERR(pthread_stackseg_np(pthread_self(), &stk));
1925 *size = stk.ss_size;
1926#elif defined HAVE_PTHREAD_GETTHRDS_NP
1927 pthread_t th = pthread_self();
1928 struct __pthrdsinfo thinfo;
1930 int regsiz=
sizeof(reg);
1931 CHECK_ERR(pthread_getthrds_np(&th, PTHRDSINFO_QUERY_ALL,
1932 &thinfo,
sizeof(thinfo),
1934 *addr = thinfo.__pi_stackaddr;
1938 *size = thinfo.__pi_stackend - thinfo.__pi_stackaddr;
1939 STACK_DIR_UPPER((
void)0, (
void)(*addr = (
char *)*addr + *size));
1940#elif defined __HAIKU__
1942 STACK_GROW_DIR_DETECTION;
1943 CHECK_ERR(get_thread_info(find_thread(NULL), &info));
1944 *addr = info.stack_base;
1945 *size = (uintptr_t)info.stack_end - (uintptr_t)info.stack_base;
1946 STACK_DIR_UPPER((
void)0, (
void)(*addr = (
char *)*addr + *size));
1948#error STACKADDR_AVAILABLE is defined but not implemented.
1956 rb_nativethread_id_t id;
1957 size_t stack_maxsize;
1959} native_main_thread;
1961#ifdef STACK_END_ADDRESS
1962extern void *STACK_END_ADDRESS;
1966 RUBY_STACK_SPACE_LIMIT = 1024 * 1024,
1967 RUBY_STACK_SPACE_RATIO = 5
1971space_size(
size_t stack_size)
1973 size_t space_size = stack_size / RUBY_STACK_SPACE_RATIO;
1974 if (space_size > RUBY_STACK_SPACE_LIMIT) {
1975 return RUBY_STACK_SPACE_LIMIT;
1983native_thread_init_main_thread_stack(
void *addr)
1985 native_main_thread.id = pthread_self();
1986#ifdef RUBY_ASAN_ENABLED
1987 addr = asan_get_real_stack_addr((
void *)addr);
1990#if MAINSTACKADDR_AVAILABLE
1991 if (native_main_thread.stack_maxsize)
return;
1995 if (get_main_stack(&stackaddr, &size) == 0) {
1996 native_main_thread.stack_maxsize = size;
1997 native_main_thread.stack_start = stackaddr;
2002#ifdef STACK_END_ADDRESS
2003 native_main_thread.stack_start = STACK_END_ADDRESS;
2005 if (!native_main_thread.stack_start ||
2006 STACK_UPPER((
VALUE *)(
void *)&addr,
2007 native_main_thread.stack_start > (
VALUE *)addr,
2008 native_main_thread.stack_start < (
VALUE *)addr)) {
2009 native_main_thread.stack_start = (
VALUE *)addr;
2013#if defined(HAVE_GETRLIMIT)
2014#if defined(PTHREAD_STACK_DEFAULT)
2015# if PTHREAD_STACK_DEFAULT < RUBY_STACK_SPACE*5
2016# error "PTHREAD_STACK_DEFAULT is too small"
2018 size_t size = PTHREAD_STACK_DEFAULT;
2020 size_t size = RUBY_VM_THREAD_VM_STACK_SIZE;
2023 int pagesize = getpagesize();
2025 STACK_GROW_DIR_DETECTION;
2026 if (getrlimit(RLIMIT_STACK, &rlim) == 0) {
2027 size = (size_t)rlim.rlim_cur;
2029 addr = native_main_thread.stack_start;
2030 if (IS_STACK_DIR_UPPER()) {
2031 space = ((size_t)((
char *)addr + size) / pagesize) * pagesize - (size_t)addr;
2034 space = (size_t)addr - ((
size_t)((
char *)addr - size) / pagesize + 1) * pagesize;
2036 native_main_thread.stack_maxsize = space;
2040#if MAINSTACKADDR_AVAILABLE
2047 STACK_GROW_DIR_DETECTION;
2049 if (IS_STACK_DIR_UPPER()) {
2050 start = native_main_thread.stack_start;
2051 end = (
char *)native_main_thread.stack_start + native_main_thread.stack_maxsize;
2054 start = (
char *)native_main_thread.stack_start - native_main_thread.stack_maxsize;
2055 end = native_main_thread.stack_start;
2058 if ((
void *)addr < start || (
void *)addr > end) {
2060 native_main_thread.stack_start = (
VALUE *)addr;
2061 native_main_thread.stack_maxsize = 0;
2066#define CHECK_ERR(expr) \
2067 {int err = (expr); if (err) {rb_bug_errno(#expr, err);}}
2070native_thread_init_stack(rb_thread_t *th,
void *local_in_parent_frame)
2072 rb_nativethread_id_t curr = pthread_self();
2073#ifdef RUBY_ASAN_ENABLED
2074 local_in_parent_frame = asan_get_real_stack_addr(local_in_parent_frame);
2075 th->ec->machine.asan_fake_stack_handle = asan_get_thread_fake_stack_handle();
2078 if (!native_main_thread.id) {
2081 native_thread_init_main_thread_stack(local_in_parent_frame);
2084 if (pthread_equal(curr, native_main_thread.id)) {
2085 th->ec->machine.stack_start = native_main_thread.stack_start;
2086 th->ec->machine.stack_maxsize = native_main_thread.stack_maxsize;
2089#ifdef STACKADDR_AVAILABLE
2090 if (th_has_dedicated_nt(th)) {
2094 if (get_stack(&start, &size) == 0) {
2095 uintptr_t diff = (uintptr_t)start - (uintptr_t)local_in_parent_frame;
2096 th->ec->machine.stack_start = local_in_parent_frame;
2097 th->ec->machine.stack_maxsize = size - diff;
2101 rb_raise(
rb_eNotImpError,
"ruby engine can initialize only in the main thread");
2110 struct rb_native_thread *nt;
2120 pthread_attr_t attr;
2122 const size_t stack_size = nt->vm->default_params.thread_machine_stack_size;
2123 const size_t space = space_size(stack_size);
2125 nt->machine_stack_maxsize = stack_size - space;
2127#ifdef USE_SIGALTSTACK
2128 nt->altstack = rb_allocate_sigaltstack();
2131 CHECK_ERR(pthread_attr_init(&attr));
2133# ifdef PTHREAD_STACK_MIN
2134 RUBY_DEBUG_LOG(
"stack size: %lu", (
unsigned long)stack_size);
2135 CHECK_ERR(pthread_attr_setstacksize(&attr, stack_size));
2138# ifdef HAVE_PTHREAD_ATTR_SETINHERITSCHED
2139 CHECK_ERR(pthread_attr_setinheritsched(&attr, PTHREAD_INHERIT_SCHED));
2141 CHECK_ERR(pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED));
2143 err = pthread_create(&nt->thread_id, &attr, nt_start, nt);
2145 RUBY_DEBUG_LOG(
"nt:%d err:%d", (
int)nt->serial, err);
2147 CHECK_ERR(pthread_attr_destroy(&attr));
2158 if (&nt->cond.readyq != &nt->cond.intr) {
2167#ifdef RB_THREAD_T_HAS_NATIVE_ID
2168 nt->tid = get_native_thread_id();
2172 RB_ALTSTACK_INIT(nt->altstack, nt->altstack);
2176native_thread_alloc(
void)
2179 native_thread_setup(nt);
2185#if USE_RUBY_DEBUG_LOG
2193native_thread_create_dedicated(rb_thread_t *th)
2195 th->nt = native_thread_alloc();
2196 th->nt->vm = th->vm;
2197 th->nt->running_thread = th;
2198 th->nt->dedicated = 1;
2201 size_t vm_stack_word_size = th->vm->default_params.thread_vm_stack_size /
sizeof(
VALUE);
2203 th->sched.malloc_stack =
true;
2204 rb_ec_initialize_vm_stack(th->ec, vm_stack, vm_stack_word_size);
2205 th->sched.context_stack = vm_stack;
2208 int err = native_thread_create0(th->nt);
2211 thread_sched_to_ready(TH_SCHED(th), th);
2217call_thread_start_func_2(rb_thread_t *th)
2225 VALUE stack_start = 0;
2226 VALUE *stack_start_addr = asan_get_real_stack_addr(&stack_start);
2228 native_thread_init_stack(th, stack_start_addr);
2229 thread_start_func_2(th, th->ec->machine.stack_start);
2236 rb_vm_t *vm = nt->vm;
2238 native_thread_setup_on_thread(nt);
2241#ifdef RB_THREAD_T_HAS_NATIVE_ID
2242 nt->tid = get_native_thread_id();
2245#if USE_RUBY_DEBUG_LOG && defined(RUBY_NT_SERIAL)
2246 ruby_nt_serial = nt->serial;
2249 RUBY_DEBUG_LOG(
"nt:%u", nt->serial);
2251 if (!nt->dedicated) {
2252 coroutine_initialize_main(nt->nt_context);
2256 if (nt->dedicated) {
2258 rb_thread_t *th = nt->running_thread;
2261 RUBY_DEBUG_LOG(
"on dedicated th:%u", rb_th_serial(th));
2262 ruby_thread_set_native(th);
2264 thread_sched_lock(sched, th);
2266 if (sched->running == th) {
2267 thread_sched_add_running_thread(sched, th);
2269 thread_sched_wait_running_turn(sched, th,
false);
2271 thread_sched_unlock(sched, th);
2274 call_thread_start_func_2(th);
2278 RUBY_DEBUG_LOG(
"check next");
2279 rb_ractor_t *r = ractor_sched_deq(vm, NULL);
2284 thread_sched_lock(sched, NULL);
2286 rb_thread_t *next_th = sched->running;
2288 if (next_th && next_th->nt == NULL) {
2289 RUBY_DEBUG_LOG(
"nt:%d next_th:%d", (
int)nt->serial, (
int)next_th->serial);
2290 thread_sched_switch0(nt->nt_context, next_th, nt,
false);
2293 RUBY_DEBUG_LOG(
"no schedulable threads -- next_th:%p", next_th);
2296 thread_sched_unlock(sched, NULL);
2303 if (nt->dedicated) {
2313static int native_thread_create_shared(rb_thread_t *th);
2316static void nt_free_stack(
void *mstack);
2320rb_threadptr_remove(rb_thread_t *th)
2323 if (th->sched.malloc_stack) {
2328 rb_vm_t *vm = th->vm;
2329 th->sched.finished =
false;
2332 ccan_list_add(&vm->ractor.sched.zombie_threads, &th->sched.node.zombie_threads);
2339rb_threadptr_sched_free(rb_thread_t *th)
2342 if (th->sched.malloc_stack) {
2345 native_thread_destroy(th->nt);
2348 nt_free_stack(th->sched.context_stack);
2353 th->sched.context = NULL;
2357 native_thread_destroy(th->nt);
2364rb_thread_sched_mark_zombies(rb_vm_t *vm)
2366 if (!ccan_list_empty(&vm->ractor.sched.zombie_threads)) {
2367 rb_thread_t *zombie_th, *next_zombie_th;
2368 ccan_list_for_each_safe(&vm->ractor.sched.zombie_threads, zombie_th, next_zombie_th, sched.node.zombie_threads) {
2369 if (zombie_th->sched.finished) {
2370 ccan_list_del_init(&zombie_th->sched.node.zombie_threads);
2373 rb_gc_mark(zombie_th->self);
2380native_thread_create(rb_thread_t *th)
2382 VM_ASSERT(th->nt == 0);
2383 RUBY_DEBUG_LOG(
"th:%d has_dnt:%d", th->serial, th->has_dedicated_nt);
2386 if (!th->ractor->threads.sched.enable_mn_threads) {
2387 th->has_dedicated_nt = 1;
2390 if (th->has_dedicated_nt) {
2391 return native_thread_create_dedicated(th);
2394 return native_thread_create_shared(th);
2398#if USE_NATIVE_THREAD_PRIORITY
2401native_thread_apply_priority(rb_thread_t *th)
2403#if defined(_POSIX_PRIORITY_SCHEDULING) && (_POSIX_PRIORITY_SCHEDULING > 0)
2404 struct sched_param sp;
2406 int priority = 0 - th->priority;
2408 pthread_getschedparam(th->nt->thread_id, &policy, &sp);
2409 max = sched_get_priority_max(policy);
2410 min = sched_get_priority_min(policy);
2412 if (min > priority) {
2415 else if (max < priority) {
2419 sp.sched_priority = priority;
2420 pthread_setschedparam(th->nt->thread_id, policy, &sp);
2431 return rb_fd_select(n, readfds, writefds, exceptfds, timeout);
2435ubf_pthread_cond_signal(
void *ptr)
2437 rb_thread_t *th = (rb_thread_t *)ptr;
2438 RUBY_DEBUG_LOG(
"th:%u on nt:%d", rb_th_serial(th), (
int)th->nt->serial);
2443native_cond_sleep(rb_thread_t *th, rb_hrtime_t *rel)
2445 rb_nativethread_lock_t *lock = &th->interrupt_lock;
2446 rb_nativethread_cond_t *cond = &th->nt->cond.intr;
2456 const rb_hrtime_t max = (rb_hrtime_t)100000000 * RB_HRTIME_PER_SEC;
2458 THREAD_BLOCKING_BEGIN(th);
2461 th->unblock.func = ubf_pthread_cond_signal;
2462 th->unblock.arg = th;
2464 if (RUBY_VM_INTERRUPTED(th->ec)) {
2466 RUBY_DEBUG_LOG(
"interrupted before sleep th:%u", rb_th_serial(th));
2479 end = native_cond_timeout(cond, *rel);
2480 native_cond_timedwait(cond, lock, &end);
2483 th->unblock.func = 0;
2487 THREAD_BLOCKING_END(th);
2489 RUBY_DEBUG_LOG(
"done th:%u", rb_th_serial(th));
2493static CCAN_LIST_HEAD(ubf_list_head);
2494static rb_nativethread_lock_t ubf_list_lock = RB_NATIVETHREAD_LOCK_INIT;
2497ubf_list_atfork(
void)
2499 ccan_list_head_init(&ubf_list_head);
2505ubf_list_contain_p(rb_thread_t *th)
2507 rb_thread_t *list_th;
2508 ccan_list_for_each(&ubf_list_head, list_th, sched.node.ubf) {
2509 if (list_th == th)
return true;
2516register_ubf_list(rb_thread_t *th)
2518 RUBY_DEBUG_LOG(
"th:%u", rb_th_serial(th));
2519 struct ccan_list_node *node = &th->sched.node.ubf;
2521 VM_ASSERT(th->unblock.func != NULL);
2526 if (ccan_list_empty((
struct ccan_list_head*)node)) {
2527 VM_ASSERT(!ubf_list_contain_p(th));
2528 ccan_list_add(&ubf_list_head, node);
2533 timer_thread_wakeup();
2538unregister_ubf_list(rb_thread_t *th)
2540 RUBY_DEBUG_LOG(
"th:%u", rb_th_serial(th));
2541 struct ccan_list_node *node = &th->sched.node.ubf;
2544 VM_ASSERT(th->unblock.func == NULL);
2546 if (!ccan_list_empty((
struct ccan_list_head*)node)) {
2549 VM_ASSERT(ubf_list_contain_p(th));
2550 ccan_list_del_init(node);
2561ubf_wakeup_thread(rb_thread_t *th)
2563 RUBY_DEBUG_LOG(
"th:%u thread_id:%p", rb_th_serial(th), (
void *)th->nt->thread_id);
2565 pthread_kill(th->nt->thread_id, SIGVTALRM);
2569ubf_select(
void *ptr)
2571 rb_thread_t *th = (rb_thread_t *)ptr;
2572 RUBY_DEBUG_LOG(
"wakeup th:%u", rb_th_serial(th));
2573 ubf_wakeup_thread(th);
2574 register_ubf_list(th);
2578ubf_threads_empty(
void)
2580 return ccan_list_empty(&ubf_list_head) != 0;
2584ubf_wakeup_all_threads(
void)
2589 ccan_list_for_each(&ubf_list_head, th, sched.node.ubf) {
2590 ubf_wakeup_thread(th);
2597#define register_ubf_list(th) (void)(th)
2598#define unregister_ubf_list(th) (void)(th)
2600static void ubf_wakeup_all_threads(
void) {
return; }
2601static bool ubf_threads_empty(
void) {
return true; }
2602#define ubf_list_atfork() do {} while (0)
2606#define WRITE_CONST(fd, str) (void)(write((fd),(str),sizeof(str)-1)<0)
2609rb_thread_wakeup_timer_thread(
int sig)
2615 timer_thread_wakeup_force();
2619 rb_vm_t *vm = GET_VM();
2620 rb_thread_t *main_th = vm->ractor.main_thread;
2623 volatile rb_execution_context_t *main_th_ec = ACCESS_ONCE(rb_execution_context_t *, main_th->ec);
2626 RUBY_VM_SET_TRAP_INTERRUPT(main_th_ec);
2628 if (vm->ubf_async_safe && main_th->unblock.func) {
2629 (main_th->unblock.func)(main_th->unblock.arg);
2636#define CLOSE_INVALIDATE_PAIR(expr) \
2637 close_invalidate_pair(expr,"close_invalidate: "#expr)
2639close_invalidate(
int *fdp,
const char *msg)
2644 if (close(fd) < 0) {
2645 async_bug_fd(msg,
errno, fd);
2650close_invalidate_pair(
int fds[2],
const char *msg)
2652 if (USE_EVENTFD && fds[0] == fds[1]) {
2654 close_invalidate(&fds[0], msg);
2657 close_invalidate(&fds[1], msg);
2658 close_invalidate(&fds[0], msg);
2668 oflags = fcntl(fd, F_GETFL);
2671 oflags |= O_NONBLOCK;
2672 err = fcntl(fd, F_SETFL, oflags);
2679setup_communication_pipe_internal(
int pipes[2])
2683 if (pipes[0] > 0 || pipes[1] > 0) {
2684 VM_ASSERT(pipes[0] > 0);
2685 VM_ASSERT(pipes[1] > 0);
2693#if USE_EVENTFD && defined(EFD_NONBLOCK) && defined(EFD_CLOEXEC)
2694 pipes[0] = pipes[1] = eventfd(0, EFD_NONBLOCK|EFD_CLOEXEC);
2696 if (pipes[0] >= 0) {
2704 rb_bug(
"can not create communication pipe");
2708 set_nonblock(pipes[0]);
2709 set_nonblock(pipes[1]);
2712#if !defined(SET_CURRENT_THREAD_NAME) && defined(__linux__) && defined(PR_SET_NAME)
2713# define SET_CURRENT_THREAD_NAME(name) prctl(PR_SET_NAME, name)
2718#if defined(__linux__)
2720#elif defined(__APPLE__)
2728static VALUE threadptr_invoke_proc_location(rb_thread_t *th);
2731native_set_thread_name(rb_thread_t *th)
2733#ifdef SET_CURRENT_THREAD_NAME
2735 if (!
NIL_P(loc = th->name)) {
2736 SET_CURRENT_THREAD_NAME(RSTRING_PTR(loc));
2738 else if ((loc = threadptr_invoke_proc_location(th)) !=
Qnil) {
2740 char buf[THREAD_NAME_MAX];
2745 p = strrchr(name,
'/');
2753 if (
len >=
sizeof(buf)) {
2754 buf[
sizeof(buf)-2] =
'*';
2755 buf[
sizeof(buf)-1] =
'\0';
2757 SET_CURRENT_THREAD_NAME(buf);
2763native_set_another_thread_name(rb_nativethread_id_t thread_id,
VALUE name)
2765#if defined SET_ANOTHER_THREAD_NAME || defined SET_CURRENT_THREAD_NAME
2766 char buf[THREAD_NAME_MAX];
2768# if !defined SET_ANOTHER_THREAD_NAME
2769 if (!pthread_equal(pthread_self(), thread_id))
return;
2774 if (n >= (
int)
sizeof(buf)) {
2775 memcpy(buf, s,
sizeof(buf)-1);
2776 buf[
sizeof(buf)-1] =
'\0';
2780# if defined SET_ANOTHER_THREAD_NAME
2781 SET_ANOTHER_THREAD_NAME(thread_id, s);
2782# elif defined SET_CURRENT_THREAD_NAME
2783 SET_CURRENT_THREAD_NAME(s);
2788#if defined(RB_THREAD_T_HAS_NATIVE_ID) || defined(__APPLE__)
2790native_thread_native_thread_id(rb_thread_t *target_th)
2792 if (!target_th->nt)
return Qnil;
2794#ifdef RB_THREAD_T_HAS_NATIVE_ID
2795 int tid = target_th->nt->tid;
2796 if (tid == 0)
return Qnil;
2798#elif defined(__APPLE__)
2804# if (!defined(MAC_OS_X_VERSION_10_6) || \
2805 (MAC_OS_X_VERSION_MAX_ALLOWED < MAC_OS_X_VERSION_10_6) || \
2806 defined(__POWERPC__) )
2807 const bool no_pthread_threadid_np =
true;
2808# define NO_PTHREAD_MACH_THREAD_NP 1
2809# elif MAC_OS_X_VERSION_MIN_REQUIRED >= MAC_OS_X_VERSION_10_6
2810 const bool no_pthread_threadid_np =
false;
2812# if !(defined(__has_attribute) && __has_attribute(availability))
2814 __attribute__((weak))
int pthread_threadid_np(pthread_t, uint64_t*);
2817 const bool no_pthread_threadid_np = !&pthread_threadid_np;
2819 if (no_pthread_threadid_np) {
2820 return ULL2NUM(pthread_mach_thread_np(pthread_self()));
2822# ifndef NO_PTHREAD_MACH_THREAD_NP
2823 int e = pthread_threadid_np(target_th->nt->thread_id, &tid);
2825 return ULL2NUM((
unsigned long long)tid);
2829# define USE_NATIVE_THREAD_NATIVE_THREAD_ID 1
2831# define USE_NATIVE_THREAD_NATIVE_THREAD_ID 0
2835 rb_serial_t created_fork_gen;
2836 pthread_t pthread_id;
2840#if (HAVE_SYS_EPOLL_H || HAVE_SYS_EVENT_H) && USE_MN_THREADS
2843#if HAVE_SYS_EPOLL_H && USE_MN_THREADS
2844#define EPOLL_EVENTS_MAX 0x10
2845 struct epoll_event finished_events[EPOLL_EVENTS_MAX];
2846#elif HAVE_SYS_EVENT_H && USE_MN_THREADS
2847#define KQUEUE_EVENTS_MAX 0x10
2848 struct kevent finished_events[KQUEUE_EVENTS_MAX];
2852 struct ccan_list_head waiting;
2853 pthread_mutex_t waiting_lock;
2855 .created_fork_gen = 0,
2858#define TIMER_THREAD_CREATED_P() (timer_th.created_fork_gen == current_fork_gen)
2860static void timer_thread_check_timeslice(rb_vm_t *vm);
2861static int timer_thread_set_timeout(rb_vm_t *vm);
2862static void timer_thread_wakeup_thread(rb_thread_t *th, uint32_t event_serial);
2864#include "thread_pthread_mn.c"
2870 return (rb_thread_t *)((size_t)w - offsetof(rb_thread_t, sched.waiting_reason));
2878timer_thread_set_timeout(rb_vm_t *vm)
2885 ractor_sched_lock(vm, NULL);
2887 if ( !ccan_list_empty(&vm->ractor.sched.timeslice_threads)
2888 || !ubf_threads_empty()
2889 || vm->ractor.sched.grq_cnt > 0
2892 RUBY_DEBUG_LOG(
"timeslice:%d ubf:%d grq:%d",
2893 !ccan_list_empty(&vm->ractor.sched.timeslice_threads),
2894 !ubf_threads_empty(),
2895 (vm->ractor.sched.grq_cnt > 0));
2898 vm->ractor.sched.timeslice_wait_inf =
false;
2901 vm->ractor.sched.timeslice_wait_inf =
true;
2904 ractor_sched_unlock(vm, NULL);
2911 rb_thread_t *th = thread_sched_waiting_thread(w);
2913 if (th && (th->sched.waiting_reason.flags & thread_sched_waiting_timeout)) {
2914 rb_hrtime_t now = rb_hrtime_now();
2915 rb_hrtime_t hrrel = rb_hrtime_sub(th->sched.waiting_reason.data.timeout, now);
2917 RUBY_DEBUG_LOG(
"th:%u now:%lu rel:%lu", rb_th_serial(th), (
unsigned long)now, (
unsigned long)hrrel);
2920 int thread_timeout = (int)((hrrel + RB_HRTIME_PER_MSEC - 1) / RB_HRTIME_PER_MSEC);
2923 if (timeout < 0 || thread_timeout < timeout) {
2924 timeout = thread_timeout;
2930 RUBY_DEBUG_LOG(
"timeout:%d inf:%d", timeout, (
int)vm->ractor.sched.timeslice_wait_inf);
2938timer_thread_check_signal(rb_vm_t *vm)
2942 int signum = rb_signal_buff_size();
2943 if (UNLIKELY(signum > 0) && vm->ractor.main_thread) {
2944 RUBY_DEBUG_LOG(
"signum:%d", signum);
2945 threadptr_trap_interrupt(vm->ractor.main_thread);
2950timer_thread_check_exceed(rb_hrtime_t abs, rb_hrtime_t now)
2955 else if (abs - now < RB_HRTIME_PER_MSEC) {
2964timer_thread_deq_wakeup(rb_vm_t *vm, rb_hrtime_t now, uint32_t *event_serial)
2969 (w->flags & thread_sched_waiting_timeout) &&
2970 timer_thread_check_exceed(w->data.timeout, now)) {
2972 RUBY_DEBUG_LOG(
"wakeup th:%u", rb_th_serial(thread_sched_waiting_thread(w)));
2975 ccan_list_del_init(&w->node);
2978 w->flags = thread_sched_waiting_none;
2981 rb_thread_t *th = thread_sched_waiting_thread(w);
2982 *event_serial = w->data.event_serial;
2990timer_thread_wakeup_thread_locked(
struct rb_thread_sched *sched, rb_thread_t *th, uint32_t event_serial)
2992 if (sched->running != th && th->sched.event_serial == event_serial) {
2993 thread_sched_to_ready_common(sched, th,
true,
false);
2998timer_thread_wakeup_thread(rb_thread_t *th, uint32_t event_serial)
3000 RUBY_DEBUG_LOG(
"th:%u", rb_th_serial(th));
3003 thread_sched_lock(sched, th);
3005 timer_thread_wakeup_thread_locked(sched, th, event_serial);
3007 thread_sched_unlock(sched, th);
3011timer_thread_check_timeout(rb_vm_t *vm)
3013 rb_hrtime_t now = rb_hrtime_now();
3015 uint32_t event_serial;
3019 while ((th = timer_thread_deq_wakeup(vm, now, &event_serial)) != NULL) {
3021 timer_thread_wakeup_thread(th, event_serial);
3029timer_thread_check_timeslice(rb_vm_t *vm)
3033 ccan_list_for_each(&vm->ractor.sched.timeslice_threads, th, sched.node.timeslice_threads) {
3034 RUBY_DEBUG_LOG(
"timeslice th:%u", rb_th_serial(th));
3035 RUBY_VM_SET_TIMER_INTERRUPT(th->ec);
3043 pthread_sigmask(0, NULL, &oldmask);
3044 if (sigismember(&oldmask, SIGVTALRM)) {
3048 RUBY_DEBUG_LOG(
"ok");
3053timer_thread_func(
void *ptr)
3055 rb_vm_t *vm = (rb_vm_t *)ptr;
3056#if defined(RUBY_NT_SERIAL)
3060 RUBY_DEBUG_LOG(
"started%s",
"");
3063 timer_thread_check_signal(vm);
3064 timer_thread_check_timeout(vm);
3065 ubf_wakeup_all_threads();
3068 timer_thread_polling(vm);
3071 RUBY_DEBUG_LOG(
"terminated");
3077signal_communication_pipe(
int fd)
3080 const uint64_t buff = 1;
3082 const char buff =
'!';
3089 if ((result = write(fd, &buff,
sizeof(buff))) <= 0) {
3092 case EINTR:
goto retry;
3094#if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
3099 async_bug_fd(
"rb_thread_wakeup_timer_thread: write", e, fd);
3102 if (TT_DEBUG) WRITE_CONST(2,
"rb_thread_wakeup_timer_thread: write\n");
3110timer_thread_wakeup_force(
void)
3113 signal_communication_pipe(timer_th.comm_fds[1]);
3117timer_thread_wakeup_locked(rb_vm_t *vm)
3120 ASSERT_ractor_sched_locked(vm, NULL);
3122 if (timer_th.created_fork_gen == current_fork_gen) {
3123 if (vm->ractor.sched.timeslice_wait_inf) {
3124 RUBY_DEBUG_LOG(
"wakeup with fd:%d", timer_th.comm_fds[1]);
3125 timer_thread_wakeup_force();
3128 RUBY_DEBUG_LOG(
"will be wakeup...");
3134timer_thread_wakeup(
void)
3136 rb_vm_t *vm = GET_VM();
3138 ractor_sched_lock(vm, NULL);
3140 timer_thread_wakeup_locked(vm);
3142 ractor_sched_unlock(vm, NULL);
3146rb_thread_create_timer_thread(
void)
3148 rb_serial_t created_fork_gen = timer_th.created_fork_gen;
3150 RUBY_DEBUG_LOG(
"fork_gen create:%d current:%d", (
int)created_fork_gen, (
int)current_fork_gen);
3152 timer_th.created_fork_gen = current_fork_gen;
3154 if (created_fork_gen != current_fork_gen) {
3155 if (created_fork_gen != 0) {
3156 RUBY_DEBUG_LOG(
"forked child process");
3158 CLOSE_INVALIDATE_PAIR(timer_th.comm_fds);
3159#if HAVE_SYS_EPOLL_H && USE_MN_THREADS
3160 close_invalidate(&timer_th.event_fd,
"close event_fd");
3165 ccan_list_head_init(&timer_th.waiting);
3169 setup_communication_pipe_internal(timer_th.comm_fds);
3172 timer_thread_setup_mn();
3175 pthread_create(&timer_th.pthread_id, NULL, timer_thread_func, GET_VM());
3179native_stop_timer_thread(
void)
3183 RUBY_DEBUG_LOG(
"wakeup send %d", timer_th.comm_fds[1]);
3184 timer_thread_wakeup_force();
3185 RUBY_DEBUG_LOG(
"wakeup sent");
3186 pthread_join(timer_th.pthread_id, NULL);
3188 if (TT_DEBUG) fprintf(stderr,
"stop timer thread\n");
3194native_reset_timer_thread(
void)
3199#ifdef HAVE_SIGALTSTACK
3201ruby_stack_overflowed_p(
const rb_thread_t *th,
const void *addr)
3205 const size_t water_mark = 1024 * 1024;
3206 STACK_GROW_DIR_DETECTION;
3209 size = th->ec->machine.stack_maxsize;
3210 base = (
char *)th->ec->machine.stack_start - STACK_DIR_UPPER(0, size);
3212#ifdef STACKADDR_AVAILABLE
3213 else if (get_stack(&base, &size) == 0) {
3215 if (pthread_equal(th->nt->thread_id, native_main_thread.id)) {
3217 if (getrlimit(RLIMIT_STACK, &rlim) == 0 && rlim.rlim_cur > size) {
3218 size = (size_t)rlim.rlim_cur;
3222 base = (
char *)base + STACK_DIR_UPPER(+size, -size);
3229 size /= RUBY_STACK_SPACE_RATIO;
3230 if (size > water_mark) size = water_mark;
3231 if (IS_STACK_DIR_UPPER()) {
3232 if (size > ~(
size_t)base+1) size = ~(size_t)base+1;
3233 if (addr > base && addr <= (
void *)((
char *)base + size))
return 1;
3236 if (size > (
size_t)base) size = (size_t)base;
3237 if (addr > (
void *)((
char *)base - size) && addr <= base)
return 1;
3247 if (fd < 0)
return 0;
3249 if (fd == timer_th.comm_fds[0] ||
3250 fd == timer_th.comm_fds[1]
3251#if (HAVE_SYS_EPOLL_H || HAVE_SYS_EVENT_H) && USE_MN_THREADS
3252 || fd == timer_th.event_fd
3255 goto check_fork_gen;
3260 if (timer_th.created_fork_gen == current_fork_gen) {
3272 return pthread_self();
3275#if defined(USE_POLL) && !defined(HAVE_PPOLL)
3278ruby_ppoll(
struct pollfd *fds, nfds_t nfds,
3279 const struct timespec *ts,
const sigset_t *sigmask)
3286 if (ts->tv_sec > INT_MAX/1000)
3287 timeout_ms = INT_MAX;
3289 tmp = (int)(ts->tv_sec * 1000);
3291 tmp2 = (int)((ts->tv_nsec + 999999L) / (1000L * 1000L));
3292 if (INT_MAX - tmp < tmp2)
3293 timeout_ms = INT_MAX;
3295 timeout_ms = (int)(tmp + tmp2);
3301 return poll(fds, nfds, timeout_ms);
3303# define ppoll(fds,nfds,ts,sigmask) ruby_ppoll((fds),(nfds),(ts),(sigmask))
3316#define THREAD_BLOCKING_YIELD(th) do { \
3317 const rb_thread_t *next_th; \
3318 struct rb_thread_sched *sched = TH_SCHED(th); \
3319 RB_VM_SAVE_MACHINE_CONTEXT(th); \
3320 thread_sched_to_waiting(sched, (th)); \
3321 next_th = sched->running; \
3322 rb_native_mutex_unlock(&sched->lock_); \
3323 native_thread_yield(); \
3324 if (!next_th && rb_ractor_living_thread_num(th->ractor) > 1) { \
3325 native_thread_yield(); \
3329native_sleep(rb_thread_t *th, rb_hrtime_t *rel)
3333 RUBY_DEBUG_LOG(
"rel:%d", rel ? (
int)*rel : 0);
3335 if (th_has_dedicated_nt(th)) {
3336 native_cond_sleep(th, rel);
3339 thread_sched_wait_events(sched, th, -1, thread_sched_waiting_timeout, rel);
3343 thread_sched_to_waiting_until_wakeup(sched, th);
3346 RUBY_DEBUG_LOG(
"wakeup");
3350static pthread_rwlock_t rb_thread_fork_rw_lock = PTHREAD_RWLOCK_INITIALIZER;
3353rb_thread_release_fork_lock(
void)
3356 if ((r = pthread_rwlock_unlock(&rb_thread_fork_rw_lock))) {
3362rb_thread_reset_fork_lock(
void)
3365 if ((r = pthread_rwlock_destroy(&rb_thread_fork_rw_lock))) {
3369 if ((r = pthread_rwlock_init(&rb_thread_fork_rw_lock, NULL))) {
3375rb_thread_prevent_fork(
void *(*func)(
void *),
void *data)
3378 if ((r = pthread_rwlock_rdlock(&rb_thread_fork_rw_lock))) {
3381 void *result = func(data);
3382 rb_thread_release_fork_lock();
3387rb_thread_acquire_fork_lock(
void)
3390 if ((r = pthread_rwlock_wrlock(&rb_thread_fork_rw_lock))) {
3397struct rb_internal_thread_event_hook {
3398 rb_internal_thread_event_callback callback;
3402 struct rb_internal_thread_event_hook *next;
3405static pthread_rwlock_t rb_internal_thread_event_hooks_rw_lock = PTHREAD_RWLOCK_INITIALIZER;
3407#if defined(HAVE_WORKING_FORK)
3409rb_internal_thread_event_hooks_rw_lock_atfork(
void)
3418 rb_internal_thread_event_hooks_rw_lock =
3419 (pthread_rwlock_t)PTHREAD_RWLOCK_INITIALIZER;
3423rb_internal_thread_event_hook_t *
3426 rb_internal_thread_event_hook_t *hook =
ALLOC_N(rb_internal_thread_event_hook_t, 1);
3427 hook->callback = callback;
3428 hook->user_data = user_data;
3429 hook->event = internal_event;
3432 if ((r = pthread_rwlock_wrlock(&rb_internal_thread_event_hooks_rw_lock))) {
3436 hook->next = rb_internal_thread_event_hooks;
3437 ATOMIC_PTR_EXCHANGE(rb_internal_thread_event_hooks, hook);
3439 if ((r = pthread_rwlock_unlock(&rb_internal_thread_event_hooks_rw_lock))) {
3449 if ((r = pthread_rwlock_wrlock(&rb_internal_thread_event_hooks_rw_lock))) {
3453 bool success = FALSE;
3455 if (rb_internal_thread_event_hooks == hook) {
3456 ATOMIC_PTR_EXCHANGE(rb_internal_thread_event_hooks, hook->next);
3460 rb_internal_thread_event_hook_t *h = rb_internal_thread_event_hooks;
3463 if (h->next == hook) {
3464 h->next = hook->next;
3468 }
while ((h = h->next));
3471 if ((r = pthread_rwlock_unlock(&rb_internal_thread_event_hooks_rw_lock))) {
3485 if ((r = pthread_rwlock_rdlock(&rb_internal_thread_event_hooks_rw_lock))) {
3489 if (rb_internal_thread_event_hooks) {
3490 rb_internal_thread_event_hook_t *h = rb_internal_thread_event_hooks;
3492 if (h->event & event) {
3493 rb_internal_thread_event_data_t event_data = {
3496 (*h->callback)(event, &event_data, h->user_data);
3498 }
while((h = h->next));
3500 if ((r = pthread_rwlock_unlock(&rb_internal_thread_event_hooks_rw_lock))) {
3510 rb_thread_t *th = GET_THREAD();
3511 bool is_snt = th->nt->dedicated == 0;
3512 native_thread_dedicated_inc(th->vm, th->ractor, th->nt);
3518rb_thread_malloc_stack_set(rb_thread_t *th,
void *stack)
3520 th->sched.malloc_stack =
true;
3521 th->sched.context_stack = stack;
std::atomic< unsigned > rb_atomic_t
Type that is eligible for atomic operations.
#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...
#define RUBY_ATOMIC_LOAD(var)
Atomic load.
#define RUBY_ATOMIC_SET(var, val)
Identical to RUBY_ATOMIC_EXCHANGE, except for the return type.
uint32_t rb_event_flag_t
Represents event(s).
#define INT2FIX
Old name of RB_INT2FIX.
#define ZALLOC
Old name of RB_ZALLOC.
#define ALLOC_N
Old name of RB_ALLOC_N.
#define ULL2NUM
Old name of RB_ULL2NUM.
#define NUM2INT
Old name of RB_NUM2INT.
#define Qnil
Old name of RUBY_Qnil.
#define NIL_P
Old name of RB_NIL_P.
VALUE rb_eNotImpError
NotImplementedError exception.
void rb_syserr_fail(int e, const char *mesg)
Raises appropriate exception that represents a C errno.
void rb_bug_errno(const char *mesg, int errno_arg)
This is a wrapper of rb_bug() which automatically constructs appropriate message from the passed errn...
int rb_cloexec_pipe(int fildes[2])
Opens a pipe with closing on exec.
void rb_update_max_fd(int fd)
Informs the interpreter that the passed fd can be the max.
int rb_reserved_fd_p(int fd)
Queries if the given FD is reserved or not.
void rb_unblock_function_t(void *)
This is the type of UBFs.
void rb_timespec_now(struct timespec *ts)
Fills the current time into the given struct.
int len
Length of the buffer.
#define RUBY_INTERNAL_THREAD_EVENT_RESUMED
Triggered when a thread successfully acquired the GVL.
rb_internal_thread_event_hook_t * rb_internal_thread_add_event_hook(rb_internal_thread_event_callback func, rb_event_flag_t events, void *data)
Registers a thread event hook function.
#define RUBY_INTERNAL_THREAD_EVENT_EXITED
Triggered when a thread exits.
#define RUBY_INTERNAL_THREAD_EVENT_SUSPENDED
Triggered when a thread released the GVL.
bool rb_thread_lock_native_thread(void)
Declare the current Ruby thread should acquire a dedicated native thread on M:N thread scheduler.
#define RUBY_INTERNAL_THREAD_EVENT_STARTED
Triggered when a new thread is started.
bool rb_internal_thread_remove_event_hook(rb_internal_thread_event_hook_t *hook)
Unregister the passed hook.
#define RUBY_INTERNAL_THREAD_EVENT_READY
Triggered when a thread attempt to acquire the GVL.
#define RBIMPL_ATTR_MAYBE_UNUSED()
Wraps (or simulates) [[maybe_unused]].
#define RB_GC_GUARD(v)
Prevents premature destruction of local objects.
#define rb_fd_select
Waits for multiple file descriptors at once.
#define RARRAY_AREF(a, i)
#define RSTRING_GETMEM(str, ptrvar, lenvar)
Convenient macro to obtain the contents and length at once.
#define errno
Ractor-aware version of errno.
The data structure which wraps the fd_set bitmap used by select(2).
rb_nativethread_id_t rb_nativethread_self(void)
Queries the ID of the native thread that is calling this function.
void rb_native_mutex_lock(rb_nativethread_lock_t *lock)
Just another name of rb_nativethread_lock_lock.
void rb_native_cond_initialize(rb_nativethread_cond_t *cond)
Fills the passed condition variable with an initial value.
int rb_native_mutex_trylock(rb_nativethread_lock_t *lock)
Identical to rb_native_mutex_lock(), except it doesn't block in case rb_native_mutex_lock() would.
void rb_native_cond_broadcast(rb_nativethread_cond_t *cond)
Signals a condition variable.
void rb_native_mutex_initialize(rb_nativethread_lock_t *lock)
Just another name of rb_nativethread_lock_initialize.
void rb_native_mutex_unlock(rb_nativethread_lock_t *lock)
Just another name of rb_nativethread_lock_unlock.
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.
void rb_native_cond_signal(rb_nativethread_cond_t *cond)
Signals a condition variable.
void rb_native_cond_wait(rb_nativethread_cond_t *cond, rb_nativethread_lock_t *mutex)
Waits for the passed condition variable to be signalled.
void rb_native_cond_timedwait(rb_nativethread_cond_t *cond, rb_nativethread_lock_t *mutex, unsigned long msec)
Identical to rb_native_cond_wait(), except it additionally takes timeout in msec resolution.
uintptr_t VALUE
Type that represents a Ruby object.
void ruby_xfree(void *ptr)
Deallocates a storage instance.
void * ruby_xmalloc(size_t size)
Allocates a storage instance.