forked from OSchip/llvm-project
89 lines
2.3 KiB
C++
89 lines
2.3 KiB
C++
// RUN: %clangxx_cfi_dso -std=c++11 -g -DSHARED_LIB %s -fPIC -shared -o %t-cfi-so.so
|
|
// RUN: %clangxx -std=c++11 -g -DSHARED_LIB %s -fPIC -shared -o %t-nocfi-so.so
|
|
// RUN: %clangxx_cfi_dso -std=c++11 -g %s -o %t
|
|
|
|
// RUN: %expect_crash %t start 2>&1 | FileCheck %s
|
|
// RUN: %expect_crash %t mmap 2>&1 | FileCheck %s
|
|
// RUN: %expect_crash %t dlopen %t-cfi-so.so 2>&1 | FileCheck %s
|
|
// RUN: %expect_crash %t dlclose %t-cfi-so.so 2>&1 | FileCheck %s
|
|
// RUN: %expect_crash %t dlopen %t-nocfi-so.so 2>&1 | FileCheck %s
|
|
// RUN: %expect_crash %t dlclose %t-nocfi-so.so 2>&1 | FileCheck %s
|
|
|
|
// Tests that shadow is read-only most of the time.
|
|
// REQUIRES: cxxabi
|
|
|
|
// Uses private API that is not available on Android.
|
|
// UNSUPPORTED: android
|
|
|
|
#include <assert.h>
|
|
#include <dlfcn.h>
|
|
#include <stdio.h>
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/mman.h>
|
|
|
|
struct A {
|
|
virtual void f();
|
|
};
|
|
|
|
#ifdef SHARED_LIB
|
|
|
|
void A::f() {}
|
|
|
|
extern "C" A *create_A() { return new A(); }
|
|
|
|
#else
|
|
|
|
constexpr unsigned kShadowGranularity = 12;
|
|
|
|
namespace __cfi {
|
|
uintptr_t GetShadow();
|
|
}
|
|
|
|
void write_shadow(void *ptr) {
|
|
uintptr_t base = __cfi::GetShadow();
|
|
uint16_t *s =
|
|
(uint16_t *)(base + (((uintptr_t)ptr >> kShadowGranularity) << 1));
|
|
fprintf(stderr, "going to crash\n");
|
|
// CHECK: going to crash
|
|
*s = 42;
|
|
fprintf(stderr, "did not crash\n");
|
|
// CHECK-NOT: did not crash
|
|
exit(1);
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
assert(argc > 1);
|
|
const bool test_mmap = strcmp(argv[1], "mmap") == 0;
|
|
const bool test_start = strcmp(argv[1], "start") == 0;
|
|
const bool test_dlopen = strcmp(argv[1], "dlopen") == 0;
|
|
const bool test_dlclose = strcmp(argv[1], "dlclose") == 0;
|
|
const char *lib = argc > 2 ? argv[2] : nullptr;
|
|
|
|
if (test_start)
|
|
write_shadow((void *)&main);
|
|
|
|
if (test_mmap) {
|
|
void *p = mmap(nullptr, 1 << 20, PROT_READ | PROT_WRITE | PROT_EXEC,
|
|
MAP_PRIVATE | MAP_ANONYMOUS, 0, 0);
|
|
assert(p != MAP_FAILED);
|
|
write_shadow((char *)p + 100);
|
|
} else {
|
|
void *handle = dlopen(lib, RTLD_NOW);
|
|
assert(handle);
|
|
void *create_A = dlsym(handle, "create_A");
|
|
assert(create_A);
|
|
|
|
if (test_dlopen)
|
|
write_shadow(create_A);
|
|
|
|
int res = dlclose(handle);
|
|
assert(res == 0);
|
|
|
|
if (test_dlclose)
|
|
write_shadow(create_A);
|
|
}
|
|
}
|
|
#endif
|