diff options
author | Florian Mayer <fmayer@google.com> | 2021-06-04 15:40:18 +0000 |
---|---|---|
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2021-06-04 15:40:18 +0000 |
commit | 6018a9a481691033aef3955f13b1e2cca60db016 (patch) | |
tree | 790d939aa65ad043194b9604e8fe13b68d862955 | |
parent | d37ab49cc8db7eed6969a1f46b542b52585929cc (diff) | |
parent | b887dc0b5c5ac8ffa993a1ac6899ae92b6dbe8cb (diff) | |
download | bionic-6018a9a481691033aef3955f13b1e2cca60db016.tar.gz |
Fix dangling pointer in heapprofd API. am: b887dc0b5c
Original change: https://googleplex-android-review.googlesource.com/c/platform/bionic/+/14830782
Change-Id: Ic5925108a9e8d30c8cee1ecdbf9563fade5c9e71
-rw-r--r-- | libc/bionic/malloc_heapprofd.cpp | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/libc/bionic/malloc_heapprofd.cpp b/libc/bionic/malloc_heapprofd.cpp index 198bcbab7..741b45e98 100644 --- a/libc/bionic/malloc_heapprofd.cpp +++ b/libc/bionic/malloc_heapprofd.cpp @@ -325,12 +325,12 @@ void HeapprofdRememberHookConflict() { static void CommonInstallHooks(libc_globals* globals) { void* impl_handle = atomic_load(&gHeapprofdHandle); - bool reusing_handle = impl_handle != nullptr; - if (!reusing_handle) { + if (impl_handle == nullptr) { impl_handle = LoadSharedLibrary(kHeapprofdSharedLib, kHeapprofdPrefix, &globals->malloc_dispatch_table); if (impl_handle == nullptr) { return; } + atomic_store(&gHeapprofdHandle, impl_handle); } else if (!InitSharedLibrary(impl_handle, kHeapprofdSharedLib, kHeapprofdPrefix, &globals->malloc_dispatch_table)) { return; } @@ -341,11 +341,7 @@ static void CommonInstallHooks(libc_globals* globals) { // MaybeModifyGlobals locks at this point. atomic_store(&gPreviousDefaultDispatchTable, GetDefaultDispatchTable()); - if (FinishInstallHooks(globals, nullptr, kHeapprofdPrefix)) { - atomic_store(&gHeapprofdHandle, impl_handle); - } else if (!reusing_handle) { - dlclose(impl_handle); - } + FinishInstallHooks(globals, nullptr, kHeapprofdPrefix); } void HeapprofdInstallHooksAtInit(libc_globals* globals) { |