5a14bc84e1
Linux for mips has a non-standard layout for the kernel sigaction struct. Adjust the layout by the minimally amount to get the test to pass, as we don't require the usage of the restorer function. llvm-svn: 314200
114 lines
3.4 KiB
C++
114 lines
3.4 KiB
C++
// clang-format off
|
|
// RUN: %clangxx -std=c++11 %s -o %t
|
|
// RUN: env LD_PRELOAD=%shared_libasan %env_asan_opts=handle_segv=1 not %run %t 2>&1 | FileCheck %s
|
|
// RUN: env LD_PRELOAD=%shared_libasan %env_asan_opts=handle_segv=2 not %run %t 2>&1 | FileCheck %s
|
|
|
|
// RUN: %clangxx -std=c++11 -DTEST_INSTALL_SIG_HANDLER %s -o %t
|
|
// RUN: env LD_PRELOAD=%shared_libasan %env_asan_opts=handle_segv=0 not %run %t 2>&1 | FileCheck %s --check-prefix=CHECK-HANDLER
|
|
// RUN: env LD_PRELOAD=%shared_libasan %env_asan_opts=handle_segv=1 not %run %t 2>&1 | FileCheck %s
|
|
// RUN: env LD_PRELOAD=%shared_libasan %env_asan_opts=handle_segv=2 not %run %t 2>&1 | FileCheck %s
|
|
|
|
// RUN: %clangxx -std=c++11 -DTEST_INSTALL_SIG_ACTION %s -o %t
|
|
// RUN: env LD_PRELOAD=%shared_libasan %env_asan_opts=handle_segv=0 not %run %t 2>&1 | FileCheck %s --check-prefix=CHECK-ACTION
|
|
// RUN: env LD_PRELOAD=%shared_libasan %env_asan_opts=handle_segv=1 not %run %t 2>&1 | FileCheck %s
|
|
// RUN: env LD_PRELOAD=%shared_libasan %env_asan_opts=handle_segv=2 not %run %t 2>&1 | FileCheck %s
|
|
|
|
// REQUIRES: asan-dynamic-runtime
|
|
|
|
// This way of setting LD_PRELOAD does not work with Android test runner.
|
|
// REQUIRES: !android
|
|
// clang-format on
|
|
|
|
#include <assert.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/syscall.h>
|
|
#include <unistd.h>
|
|
|
|
const char *handler = nullptr;
|
|
void SigHandler(int signum) { handler = "TestSigHandler"; }
|
|
void SigAction(int, siginfo_t *, void *) { handler = "TestSigAction"; }
|
|
|
|
struct KernelSigaction {
|
|
|
|
#if defined(__mips__)
|
|
unsigned long flags;
|
|
__sighandler_t handler;
|
|
#else
|
|
__sighandler_t handler;
|
|
unsigned long flags;
|
|
#endif
|
|
void (*restorer)();
|
|
char unused[1024];
|
|
};
|
|
|
|
#if defined(__x86_64__)
|
|
extern "C" void restorer();
|
|
asm("restorer:mov $15,%rax\nsyscall");
|
|
#endif
|
|
|
|
int InternalSigaction(int sig, KernelSigaction *act, KernelSigaction *oact) {
|
|
if (act) {
|
|
#if defined(__x86_64__)
|
|
act->flags |= 0x04000000;
|
|
act->restorer = &restorer;
|
|
#endif
|
|
}
|
|
return syscall(__NR_rt_sigaction, sig, act, oact, NSIG / 8);
|
|
}
|
|
|
|
struct KernelSigaction pre_asan = {};
|
|
|
|
static void Init() {
|
|
int res = InternalSigaction(SIGSEGV, nullptr, &pre_asan);
|
|
assert(res >= 0);
|
|
assert(pre_asan.handler == SIG_DFL || pre_asan.handler == SIG_IGN);
|
|
#if defined(TEST_INSTALL_SIG_HANDLER)
|
|
pre_asan = {};
|
|
pre_asan.handler = &SigHandler;
|
|
res = InternalSigaction(SIGSEGV, &pre_asan, nullptr);
|
|
assert(res >= 0);
|
|
#elif defined(TEST_INSTALL_SIG_ACTION)
|
|
pre_asan = {};
|
|
pre_asan.flags = SA_SIGINFO | SA_NODEFER;
|
|
pre_asan.handler = (__sighandler_t)&SigAction;
|
|
res = InternalSigaction(SIGSEGV, &pre_asan, nullptr);
|
|
assert(res >= 0);
|
|
#endif
|
|
}
|
|
|
|
__attribute__((section(".preinit_array"), used))
|
|
void (*__local_test_preinit)(void) = Init;
|
|
|
|
bool ExpectUserHandler() {
|
|
#if defined(TEST_INSTALL_SIG_HANDLER) || defined(TEST_INSTALL_SIG_ACTION)
|
|
return !strcmp(getenv("ASAN_OPTIONS"), "handle_segv=0");
|
|
#endif
|
|
return false;
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
KernelSigaction post_asan = {};
|
|
InternalSigaction(SIGSEGV, nullptr, &post_asan);
|
|
|
|
assert(post_asan.handler != SIG_DFL);
|
|
assert(post_asan.handler != SIG_IGN);
|
|
assert(ExpectUserHandler() ==
|
|
(post_asan.handler == pre_asan.handler));
|
|
|
|
raise(SIGSEGV);
|
|
printf("%s\n", handler);
|
|
return 1;
|
|
}
|
|
|
|
// CHECK-NOT: TestSig
|
|
// CHECK: AddressSanitizer:DEADLYSIGNAL
|
|
|
|
// CHECK-HANDLER-NOT: AddressSanitizer:DEADLYSIGNAL
|
|
// CHECK-HANDLER: TestSigHandler
|
|
|
|
// CHECK-ACTION-NOT: AddressSanitizer:DEADLYSIGNAL
|
|
// CHECK-ACTION: TestSigAction
|