OSchip
/
llvm-project
mirror of https://github.com/THU-DSP-LAB/llvm-project.git
13
0
Fork 6
Code Issues Packages Projects Releases Wiki Activity

[asan] use TLS on Linux to get the FakeStack. Saves 15% performance 

llvm-svn: 190661
This commit is contained in:
Kostya Serebryany 2013-09-13 06:04:18 +00:00
parent 77caab4d7b
commit 096413ba3b
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
compiler-rt/lib/asan/asan_fake_stack.cc
View File

@ -122,10 +122,25 @@ NOINLINE void FakeStack::GC(uptr real_stack) {
needs_gc_ = false;
}
ALWAYS_INLINE uptr OnMalloc(uptr class_id, uptr size, uptr real_stack) {
static FakeStack *GetFakeStack() {
AsanThread *t = GetCurrentThread();
if (!t) return real_stack;
FakeStack *fs = t->fake_stack();
if (!t) return 0;
return t->fake_stack();
}
static FakeStack *GetFakeStackFast() {
#if SANITIZER_LINUX
static THREADLOCAL FakeStack *fake_stack;
if (!fake_stack)
fake_stack = GetFakeStack();
return fake_stack;
#else
return GetFakeStack();
#endif
}
ALWAYS_INLINE uptr OnMalloc(uptr class_id, uptr size, uptr real_stack) {
FakeStack *fs = GetFakeStackFast();
if (!fs) return real_stack;
FakeFrame *ff = fs->Allocate(fs->stack_size_log(), class_id, real_stack);
uptr ptr = reinterpret_cast<uptr>(ff);
@ -136,9 +151,7 @@ ALWAYS_INLINE uptr OnMalloc(uptr class_id, uptr size, uptr real_stack) {
ALWAYS_INLINE void OnFree(uptr ptr, uptr class_id, uptr size, uptr real_stack) {
if (ptr == real_stack)
return;
AsanThread *t = GetCurrentThread();
if (!t) return;
FakeStack *fs = t->fake_stack();
FakeStack *fs = GetFakeStackFast(); // Must not be 0.
FakeFrame *ff = reinterpret_cast<FakeFrame *>(ptr);
fs->Deallocate(ff, fs->stack_size_log(), class_id, real_stack);
SetShadow(ptr, size, class_id, kMagic8);