3 #include "ractor_core.h"
7 static bool vm_barrier_finish_p(
rb_vm_t *vm);
12 return vm->ractor.sync.lock_owner == GET_RACTOR();
17 RUBY_ASSERT_vm_locking(
void)
19 if (rb_multi_ractor_p()) {
21 VM_ASSERT(vm_locked(vm));
26 RUBY_ASSERT_vm_unlocking(
void)
28 if (rb_multi_ractor_p()) {
30 VM_ASSERT(!vm_locked(vm));
38 return vm_locked(GET_VM());
42 vm_lock_enter(
rb_ractor_t *cr,
rb_vm_t *vm,
bool locked,
bool no_barrier,
unsigned int *lev APPEND_LOCATION_ARGS)
44 RUBY_DEBUG_LOG2(file, line,
"start locked:%d", locked);
52 VM_ASSERT(cr->sync.locked_by != rb_ractor_self(cr));
57 VM_ASSERT(vm->ractor.sync.lock_owner == NULL);
58 vm->ractor.sync.lock_owner = cr;
62 while (vm->ractor.sync.barrier_waiting) {
63 unsigned int barrier_cnt = vm->ractor.sync.barrier_cnt;
67 RB_GC_SAVE_MACHINE_CONTEXT(th);
69 if (rb_ractor_status_p(cr, ractor_running)) {
70 rb_vm_ractor_blocking_cnt_inc(vm, cr, __FILE__, __LINE__);
76 VM_ASSERT(rb_ractor_status_p(cr, ractor_blocking));
78 if (vm_barrier_finish_p(vm)) {
79 RUBY_DEBUG_LOG(
"wakeup barrier owner");
83 RUBY_DEBUG_LOG(
"wait for barrier finish");
87 while (barrier_cnt == vm->ractor.sync.barrier_cnt) {
88 vm->ractor.sync.lock_owner = NULL;
90 VM_ASSERT(vm->ractor.sync.lock_owner == NULL);
91 vm->ractor.sync.lock_owner = cr;
94 RUBY_DEBUG_LOG(
"barrier is released. Acquire vm_lock");
97 rb_vm_ractor_blocking_cnt_dec(vm, cr, __FILE__, __LINE__);
102 VM_ASSERT(vm->ractor.sync.lock_rec == 0);
103 VM_ASSERT(vm->ractor.sync.lock_owner == cr);
106 vm->ractor.sync.lock_rec++;
107 *lev = vm->ractor.sync.lock_rec;
109 RUBY_DEBUG_LOG2(file, line,
"rec:%u owner:%d", vm->ractor.sync.lock_rec, rb_ractor_id(vm->ractor.sync.lock_owner));
113 vm_lock_leave(
rb_vm_t *vm,
unsigned int *lev APPEND_LOCATION_ARGS)
115 RUBY_DEBUG_LOG2(file, line,
"rec:%u owner:%d", vm->ractor.sync.lock_rec, rb_ractor_id(vm->ractor.sync.lock_owner));
118 VM_ASSERT(vm->ractor.sync.lock_rec > 0);
119 VM_ASSERT(vm->ractor.sync.lock_rec == *lev);
121 vm->ractor.sync.lock_rec--;
122 *lev = vm->ractor.sync.lock_rec;
124 if (vm->ractor.sync.lock_rec == 0) {
125 vm->ractor.sync.lock_owner = NULL;
130 MJIT_FUNC_EXPORTED
void
131 rb_vm_lock_enter_body(
unsigned int *lev APPEND_LOCATION_ARGS)
135 vm_lock_enter(NULL, vm,
true,
false, lev APPEND_LOCATION_PARAMS);
138 vm_lock_enter(GET_RACTOR(), vm,
false,
false, lev APPEND_LOCATION_PARAMS);
142 MJIT_FUNC_EXPORTED
void
143 rb_vm_lock_enter_body_nb(
unsigned int *lev APPEND_LOCATION_ARGS)
147 vm_lock_enter(NULL, vm,
true,
true, lev APPEND_LOCATION_PARAMS);
150 vm_lock_enter(GET_RACTOR(), vm,
false,
true, lev APPEND_LOCATION_PARAMS);
154 MJIT_FUNC_EXPORTED
void
155 rb_vm_lock_enter_body_cr(
rb_ractor_t *cr,
unsigned int *lev APPEND_LOCATION_ARGS)
158 vm_lock_enter(cr, vm, vm_locked(vm),
false, lev APPEND_LOCATION_PARAMS);
161 MJIT_FUNC_EXPORTED
void
162 rb_vm_lock_leave_body(
unsigned int *lev APPEND_LOCATION_ARGS)
164 vm_lock_leave(GET_VM(), lev APPEND_LOCATION_PARAMS);
168 rb_vm_lock_body(LOCATION_ARGS)
171 ASSERT_vm_unlocking();
173 vm_lock_enter(GET_RACTOR(), vm,
false,
false, &vm->ractor.sync.lock_rec APPEND_LOCATION_PARAMS);
177 rb_vm_unlock_body(LOCATION_ARGS)
181 VM_ASSERT(vm->ractor.sync.lock_rec == 1);
182 vm_lock_leave(vm, &vm->ractor.sync.lock_rec APPEND_LOCATION_PARAMS);
186 vm_cond_wait(
rb_vm_t *vm, rb_nativethread_cond_t *cond,
unsigned long msec)
189 unsigned int lock_rec = vm->ractor.sync.lock_rec;
192 vm->ractor.sync.lock_rec = 0;
193 vm->ractor.sync.lock_owner = NULL;
200 vm->ractor.sync.lock_rec = lock_rec;
201 vm->ractor.sync.lock_owner = cr;
205 rb_vm_cond_wait(
rb_vm_t *vm, rb_nativethread_cond_t *cond)
207 vm_cond_wait(vm, cond, 0);
211 rb_vm_cond_timedwait(
rb_vm_t *vm, rb_nativethread_cond_t *cond,
unsigned long msec)
213 vm_cond_wait(vm, cond, msec);
217 vm_barrier_finish_p(
rb_vm_t *vm)
219 RUBY_DEBUG_LOG(
"cnt:%u living:%u blocking:%u",
220 vm->ractor.sync.barrier_cnt,
222 vm->ractor.blocking_cnt);
224 VM_ASSERT(vm->ractor.blocking_cnt <= vm->ractor.cnt);
225 return vm->ractor.blocking_cnt == vm->ractor.cnt;
231 RB_DEBUG_COUNTER_INC(vm_sync_barrier);
233 if (!rb_multi_ractor_p()) {
239 VM_ASSERT(vm->ractor.sync.barrier_waiting ==
false);
243 VM_ASSERT(cr == GET_RACTOR());
244 VM_ASSERT(rb_ractor_status_p(cr, ractor_running));
246 vm->ractor.sync.barrier_waiting =
true;
248 RUBY_DEBUG_LOG(
"barrier start. cnt:%u living:%u blocking:%u",
249 vm->ractor.sync.barrier_cnt,
251 vm->ractor.blocking_cnt);
253 rb_vm_ractor_blocking_cnt_inc(vm, cr, __FILE__, __LINE__);
257 list_for_each(&vm->ractor.set, r, vmlr_node) {
259 rb_ractor_vm_barrier_interrupt_running_thread(r);
264 while (!vm_barrier_finish_p(vm)) {
265 rb_vm_cond_wait(vm, &vm->ractor.sync.barrier_cond);
268 RUBY_DEBUG_LOG(
"cnt:%u barrier success", vm->ractor.sync.barrier_cnt);
270 rb_vm_ractor_blocking_cnt_dec(vm, cr, __FILE__, __LINE__);
272 vm->ractor.sync.barrier_waiting =
false;
273 vm->ractor.sync.barrier_cnt++;
275 list_for_each(&vm->ractor.set, r, vmlr_node) {
283 unsigned int recorded_lock_rec,
284 unsigned int current_lock_rec)
286 VM_ASSERT(recorded_lock_rec != current_lock_rec);
288 if (UNLIKELY(recorded_lock_rec > current_lock_rec)) {
289 rb_bug(
"unexpected situation - recordd:%u current:%u",
290 recorded_lock_rec, current_lock_rec);
293 while (recorded_lock_rec < current_lock_rec) {
294 RB_VM_LOCK_LEAVE_LEV(¤t_lock_rec);
298 VM_ASSERT(recorded_lock_rec == rb_ec_vm_lock_rec(ec));
void rb_bug(const char *fmt,...)
Interpreter panic switch.
void rb_native_mutex_lock(rb_nativethread_lock_t *lock)
Just another name of rb_nativethread_lock_lock.
void rb_native_mutex_unlock(rb_nativethread_lock_t *lock)
Just another name of rb_nativethread_lock_unlock.
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.