TheArchive/Magisk
1
0
Fork 0
mirror of https://github.com/topjohnwu/Magisk.git synced 2025-01-11 16:33:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

pthread_cond_signal might wake multiple threads

Browse Source 
Close #4759
This commit is contained in:
topjohnwu 2021-10-19 21:32:37 -07:00
parent 8276a0775d
commit fe41df87bb
1 changed files with 17 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
native/jni/core/thread.cpp
View File

@ -18,8 +18,8 @@ static pthread_cond_t send_task = PTHREAD_COND_INITIALIZER_MONOTONIC_NP;
static pthread_cond_t recv_task = PTHREAD_COND_INITIALIZER_MONOTONIC_NP;
// The following variables should be guarded by lock
static int available_threads = 0;
static int active_threads = 0;
static int idle_threads = 0;
static int total_threads = 0;
static function<void()> pending_task;
static void operator+=(timespec &a, const timespec &b) {
@ -40,8 +40,8 @@ static void reset_pool() {
send_task = PTHREAD_COND_INITIALIZER_MONOTONIC_NP;
pthread_cond_destroy(&recv_task);
recv_task = PTHREAD_COND_INITIALIZER_MONOTONIC_NP;
available_threads = 0;
active_threads = 0;
idle_threads = 0;
total_threads = 0;
pending_task = nullptr;
}
@ -58,7 +58,7 @@ static void *thread_pool_loop(void * const is_core_pool) {
function<void()> local_task;
{
mutex_guard g(lock);
++available_threads;
++idle_threads;
if (!pending_task) {
if (is_core_pool) {
pthread_cond_wait(&send_task, &lock);
@ -68,17 +68,20 @@ static void *thread_pool_loop(void * const is_core_pool) {
ts += { THREAD_IDLE_MAX_SEC, 0 };
if (pthread_cond_timedwait(&send_task, &lock, &ts) == ETIMEDOUT) {
// Terminate thread after max idle time
--available_threads;
--active_threads;
--idle_threads;
--total_threads;
return nullptr;
}
}
}
local_task.swap(pending_task);
pthread_cond_signal(&recv_task);
--available_threads;
if (pending_task) {
local_task.swap(pending_task);
pthread_cond_signal(&recv_task);
}
--idle_threads;
}
local_task();
if (local_task)
local_task();
if (getpid() == gettid())
exit(0);
}
@ -87,9 +90,9 @@ static void *thread_pool_loop(void * const is_core_pool) {
void exec_task(function<void()> &&task) {
mutex_guard g(lock);
pending_task.swap(task);
if (available_threads == 0) {
++active_threads;
long is_core_pool = active_threads <= CORE_POOL_SIZE;
if (idle_threads == 0) {
++total_threads;
long is_core_pool = total_threads <= CORE_POOL_SIZE;
new_daemon_thread(thread_pool_loop, (void *) is_core_pool);
} else {
pthread_cond_signal(&send_task);