diff options
author | k0kubun <k0kubun@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2019-03-18 16:32:48 +0000 |
---|---|---|
committer | k0kubun <k0kubun@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2019-03-18 16:32:48 +0000 |
commit | 4d426e28d2cbbc09208987c6cf86180ea3ba3522 (patch) | |
tree | 18485082db1ad71c26e01f8850a6a8358be532e8 /mjit_worker.c | |
parent | 08e9c58d519599def6b786ea211ac3a3d8b64d52 (diff) |
Resurrect r67287 and r67288
I noticed that r67287 was illegal because memory allocated by `alloca`
was used after the stack is expired.
So I just replaced that with `malloc` and `free` for now.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@67299 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'mjit_worker.c')
-rw-r--r-- | mjit_worker.c | 114 |
1 files changed, 68 insertions, 46 deletions
diff --git a/mjit_worker.c b/mjit_worker.c index 94eda53176..a7e6bcea65 100644 --- a/mjit_worker.c +++ b/mjit_worker.c @@ -986,7 +986,7 @@ compile_prelude(FILE *f) /* Compile ISeq in UNIT and return function pointer of JIT-ed code. It may return NOT_COMPILED_JIT_ISEQ_FUNC if something went wrong. */ static mjit_func_t -convert_unit_to_func(struct rb_mjit_unit *unit, struct rb_call_cache *cc_entries, union iseq_inline_storage_entry *is_entries) +convert_unit_to_func(struct rb_mjit_unit *unit) { char c_file_buff[MAXPATHLEN], *c_file = c_file_buff, *so_file, funcname[35]; /* TODO: reconsider `35` */ int fd; @@ -1067,7 +1067,7 @@ convert_unit_to_func(struct rb_mjit_unit *unit, struct rb_call_cache *cc_entries verbose(2, "start compilation: %s@%s:%d -> %s", label, path, lineno, c_file); fprintf(f, "/* %s@%s:%d */\n\n", label, path, lineno); } - bool success = mjit_compile(f, unit->iseq->body, funcname, cc_entries, is_entries); + bool success = mjit_compile(f, unit->iseq, funcname); /* release blocking mjit_gc_start_hook */ CRITICAL_SECTION_START(3, "after mjit_compile to wakeup client for GC"); @@ -1141,33 +1141,73 @@ static void mjit_copy_job_handler(void *data); /* vm_trace.c */ int rb_workqueue_register(unsigned flags, rb_postponed_job_func_t , void *); -/* We're lazily copying cache values from main thread because these cache values - could be different between ones on enqueue timing and ones on dequeue timing. - Return true if copy succeeds. */ -static bool -copy_cache_from_main_thread(mjit_copy_job_t *job) +// Copy inline cache values of `iseq` to `*cc_entries` and `*is_entries`. +// Return true if copy succeeds or is not needed. +// +// We're lazily copying cache values from main thread because these cache values +// could be different between ones on enqueue timing and ones on dequeue timing. +bool +mjit_copy_cache_from_main_thread(const rb_iseq_t *iseq, struct rb_call_cache **cc_entries, union iseq_inline_storage_entry **is_entries) { - CRITICAL_SECTION_START(3, "in copy_cache_from_main_thread"); + mjit_copy_job_t *job = &mjit_copy_job; // just a short hand + + CRITICAL_SECTION_START(3, "in mjit_copy_cache_from_main_thread"); + job->finish_p = true; // disable dispatching this job in mjit_copy_job_handler while it's being modified + CRITICAL_SECTION_FINISH(3, "in mjit_copy_cache_from_main_thread"); + + const struct rb_iseq_constant_body *body = iseq->body; + job->cc_entries = NULL; + if (body->ci_size > 0 || body->ci_kw_size > 0) { + job->cc_entries = malloc(sizeof(struct rb_call_cache) * (body->ci_size + body->ci_kw_size)); + if (!job->cc_entries) return false; + } + job->is_entries = NULL; + if (body->is_size > 0) { + job->is_entries = malloc(sizeof(union iseq_inline_storage_entry) * body->is_size); + if (!job->is_entries) { + free(job->cc_entries); + return false; + } + } + + // If ISeq has no inline cache, there's no need to run a copy job. + if (job->cc_entries == NULL && job->is_entries == NULL) { + *cc_entries = job->cc_entries; + *is_entries = job->is_entries; + return true; + } + + CRITICAL_SECTION_START(3, "in mjit_copy_cache_from_main_thread"); + job->iseq = iseq; // Prevernt GC of this ISeq from here job->finish_p = false; // allow dispatching this job in mjit_copy_job_handler - CRITICAL_SECTION_FINISH(3, "in copy_cache_from_main_thread"); + CRITICAL_SECTION_FINISH(3, "in mjit_copy_cache_from_main_thread"); if (UNLIKELY(mjit_opts.wait)) { mjit_copy_job_handler((void *)job); - return job->finish_p; + } else if (rb_workqueue_register(0, mjit_copy_job_handler, (void *)job)) { + CRITICAL_SECTION_START(3, "in MJIT copy job wait"); + // checking `stop_worker_p` too because `RUBY_VM_CHECK_INTS(ec)` may not + // lush mjit_copy_job_handler when EC_EXEC_TAG() is not TAG_NONE, and then + // `stop_worker()` could dead lock with this function. + while (!job->finish_p && !stop_worker_p) { + rb_native_cond_wait(&mjit_worker_wakeup, &mjit_engine_mutex); + verbose(3, "Getting wakeup from client"); + } + CRITICAL_SECTION_FINISH(3, "in MJIT copy job wait"); } - if (!rb_workqueue_register(0, mjit_copy_job_handler, (void *)job)) - return false; - CRITICAL_SECTION_START(3, "in MJIT copy job wait"); - /* checking `stop_worker_p` too because `RUBY_VM_CHECK_INTS(ec)` may not - lush mjit_copy_job_handler when EC_EXEC_TAG() is not TAG_NONE, and then - `stop_worker()` could dead lock with this function. */ - while (!job->finish_p && !stop_worker_p) { - rb_native_cond_wait(&mjit_worker_wakeup, &mjit_engine_mutex); - verbose(3, "Getting wakeup from client"); - } - CRITICAL_SECTION_FINISH(3, "in MJIT copy job wait"); - return job->finish_p; + // Set result values. + *cc_entries = job->cc_entries; + *is_entries = job->is_entries; + + bool result = job->finish_p; + CRITICAL_SECTION_START(3, "in mjit_copy_cache_from_main_thread"); + job->iseq = NULL; // Allow GC of this ISeq from here + // Disable dispatching this job in mjit_copy_job_handler while memory allocated by alloca + // could be expired after finishing this function. + job->finish_p = true; + CRITICAL_SECTION_FINISH(3, "in mjit_copy_cache_from_main_thread"); + return result; } /* The function implementing a worker. It is executed in a separate @@ -1176,8 +1216,6 @@ copy_cache_from_main_thread(mjit_copy_job_t *job) void mjit_worker(void) { - mjit_copy_job_t *job = &mjit_copy_job; /* just a shorthand */ - #ifndef _MSC_VER if (pch_status == PCH_NOT_READY) { make_pch(); @@ -1204,28 +1242,16 @@ mjit_worker(void) verbose(3, "Getting wakeup from client"); } unit = get_from_list(&unit_queue); - if (unit) job->iseq = unit->iseq; - job->finish_p = true; // disable dispatching this job in mjit_copy_job_handler while it's being modified CRITICAL_SECTION_FINISH(3, "in worker dequeue"); if (unit) { - const struct rb_iseq_constant_body *body = unit->iseq->body; - job->cc_entries = NULL; - if (body->ci_size > 0 || body->ci_kw_size > 0) - job->cc_entries = alloca(sizeof(struct rb_call_cache) * (body->ci_size + body->ci_kw_size)); - job->is_entries = NULL; - if (body->is_size > 0) - job->is_entries = alloca(sizeof(union iseq_inline_storage_entry) * body->is_size); - - /* Copy ISeq's inline caches values to avoid race condition. */ - if (job->cc_entries != NULL || job->is_entries != NULL) { - if (copy_cache_from_main_thread(job) == false) { - continue; /* retry postponed_job failure, or stop worker */ - } - } - // JIT compile - mjit_func_t func = convert_unit_to_func(unit, job->cc_entries, job->is_entries); + mjit_func_t func = convert_unit_to_func(unit); + + // `mjit_copy_cache_from_main_thread` in `mjit_compile` may wait for a long time + // and worker may be stopped during the compilation. + if (stop_worker_p) + break; CRITICAL_SECTION_START(3, "in jit func replace"); while (in_gc) { /* Make sure we're not GC-ing when touching ISeq */ @@ -1248,10 +1274,6 @@ mjit_worker(void) } } - // Disable dispatching this job in mjit_copy_job_handler while memory allocated by alloca - // could be expired after finishing this function. - job->finish_p = true; - // To keep mutex unlocked when it is destroyed by mjit_finish, don't wrap CRITICAL_SECTION here. worker_stopped = true; } |