mte: make h_malloc_disable_memory_tagging() thread-safe

pull/226/head
Dmitry Muhomor 2023-10-29 09:43:57 +02:00 committed by Daniel Micay
parent 01a199e19e
commit 93aa9eefe4
1 changed files with 8 additions and 5 deletions

View File

@ -1181,13 +1181,14 @@ static inline void enforce_init(void) {
} }
} }
COLD static void init_slow_path(void) { static struct mutex init_lock = MUTEX_INITIALIZER;
static struct mutex lock = MUTEX_INITIALIZER;
mutex_lock(&lock); COLD static void init_slow_path(void) {
mutex_lock(&init_lock);
if (unlikely(is_init())) { if (unlikely(is_init())) {
mutex_unlock(&lock); mutex_unlock(&init_lock);
return; return;
} }
@ -1278,7 +1279,7 @@ COLD static void init_slow_path(void) {
} }
memory_set_name(&ro, sizeof(ro), "malloc read-only after init"); memory_set_name(&ro, sizeof(ro), "malloc read-only after init");
mutex_unlock(&lock); mutex_unlock(&init_lock);
// may allocate, so wait until the allocator is initialized to avoid deadlocking // may allocate, so wait until the allocator is initialized to avoid deadlocking
if (unlikely(pthread_atfork(full_lock, full_unlock, post_fork_child))) { if (unlikely(pthread_atfork(full_lock, full_unlock, post_fork_child))) {
@ -2157,6 +2158,7 @@ COLD EXPORT int h_malloc_set_state(UNUSED void *state) {
#ifdef __ANDROID__ #ifdef __ANDROID__
COLD EXPORT void h_malloc_disable_memory_tagging(void) { COLD EXPORT void h_malloc_disable_memory_tagging(void) {
#ifdef HAS_ARM_MTE #ifdef HAS_ARM_MTE
mutex_lock(&init_lock);
if (!ro.is_memtag_disabled) { if (!ro.is_memtag_disabled) {
if (is_init()) { if (is_init()) {
if (unlikely(memory_protect_rw(&ro, sizeof(ro)))) { if (unlikely(memory_protect_rw(&ro, sizeof(ro)))) {
@ -2171,6 +2173,7 @@ COLD EXPORT void h_malloc_disable_memory_tagging(void) {
ro.is_memtag_disabled = true; ro.is_memtag_disabled = true;
} }
} }
mutex_unlock(&init_lock);
#endif #endif
} }
#endif #endif