Git-Mirrors/hardened_malloc
1
0
Fork 0
mirror of https://github.com/GrapheneOS/hardened_malloc.git synced 2024-07-01 09:01:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

mte: make h_malloc_disable_memory_tagging() thread-safe

Browse Source
This commit is contained in:
Dmitry Muhomor 2023-10-29 09:43:57 +02:00
parent 225f0b03a3
commit 2fe0e3fb73
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
h_malloc.c
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