forked from OSchip/llvm-project
115 lines
2.9 KiB
C++
115 lines
2.9 KiB
C++
|
|
#include "hwasan.h"
|
|
#include "hwasan_mapping.h"
|
|
#include "hwasan_thread.h"
|
|
#include "hwasan_poisoning.h"
|
|
#include "hwasan_interface_internal.h"
|
|
|
|
#include "sanitizer_common/sanitizer_tls_get_addr.h"
|
|
|
|
namespace __hwasan {
|
|
|
|
static u32 RandomSeed() {
|
|
u32 seed;
|
|
do {
|
|
if (UNLIKELY(!GetRandom(reinterpret_cast<void *>(&seed), sizeof(seed),
|
|
/*blocking=*/false))) {
|
|
seed = static_cast<u32>(
|
|
(NanoTime() >> 12) ^
|
|
(reinterpret_cast<uptr>(__builtin_frame_address(0)) >> 4));
|
|
}
|
|
} while (!seed);
|
|
return seed;
|
|
}
|
|
|
|
HwasanThread *HwasanThread::Create(thread_callback_t start_routine,
|
|
void *arg) {
|
|
uptr PageSize = GetPageSizeCached();
|
|
uptr size = RoundUpTo(sizeof(HwasanThread), PageSize);
|
|
HwasanThread *thread = (HwasanThread*)MmapOrDie(size, __func__);
|
|
thread->start_routine_ = start_routine;
|
|
thread->arg_ = arg;
|
|
thread->destructor_iterations_ = GetPthreadDestructorIterations();
|
|
thread->random_state_ = flags()->random_tags ? RandomSeed() : 0;
|
|
|
|
return thread;
|
|
}
|
|
|
|
void HwasanThread::SetThreadStackAndTls() {
|
|
uptr tls_size = 0;
|
|
uptr stack_size = 0;
|
|
GetThreadStackAndTls(IsMainThread(), &stack_bottom_, &stack_size,
|
|
&tls_begin_, &tls_size);
|
|
stack_top_ = stack_bottom_ + stack_size;
|
|
tls_end_ = tls_begin_ + tls_size;
|
|
|
|
int local;
|
|
CHECK(AddrIsInStack((uptr)&local));
|
|
}
|
|
|
|
void HwasanThread::Init() {
|
|
SetThreadStackAndTls();
|
|
CHECK(MEM_IS_APP(stack_bottom_));
|
|
CHECK(MEM_IS_APP(stack_top_ - 1));
|
|
}
|
|
|
|
void HwasanThread::TSDDtor(void *tsd) {
|
|
HwasanThread *t = (HwasanThread*)tsd;
|
|
t->Destroy();
|
|
}
|
|
|
|
void HwasanThread::ClearShadowForThreadStackAndTLS() {
|
|
TagMemory(stack_bottom_, stack_top_ - stack_bottom_, 0);
|
|
if (tls_begin_ != tls_end_)
|
|
TagMemory(tls_begin_, tls_end_ - tls_begin_, 0);
|
|
}
|
|
|
|
void HwasanThread::Destroy() {
|
|
malloc_storage().CommitBack();
|
|
ClearShadowForThreadStackAndTLS();
|
|
uptr size = RoundUpTo(sizeof(HwasanThread), GetPageSizeCached());
|
|
UnmapOrDie(this, size);
|
|
DTLS_Destroy();
|
|
}
|
|
|
|
thread_return_t HwasanThread::ThreadStart() {
|
|
Init();
|
|
|
|
if (!start_routine_) {
|
|
// start_routine_ == 0 if we're on the main thread or on one of the
|
|
// OS X libdispatch worker threads. But nobody is supposed to call
|
|
// ThreadStart() for the worker threads.
|
|
return 0;
|
|
}
|
|
|
|
thread_return_t res = start_routine_(arg_);
|
|
|
|
return res;
|
|
}
|
|
|
|
static u32 xorshift(u32 state) {
|
|
state ^= state << 13;
|
|
state ^= state >> 17;
|
|
state ^= state << 5;
|
|
return state;
|
|
}
|
|
|
|
// Generate a (pseudo-)random non-zero tag.
|
|
tag_t HwasanThread::GenerateRandomTag() {
|
|
tag_t tag;
|
|
do {
|
|
if (flags()->random_tags) {
|
|
if (!random_buffer_)
|
|
random_buffer_ = random_state_ = xorshift(random_state_);
|
|
CHECK(random_buffer_);
|
|
tag = random_buffer_ & 0xFF;
|
|
random_buffer_ >>= 8;
|
|
} else {
|
|
tag = random_state_ = (random_state_ + 1) & 0xFF;
|
|
}
|
|
} while (!tag);
|
|
return tag;
|
|
}
|
|
|
|
} // namespace __hwasan
|