1 // RUN: %clangxx %s -o %t
2 // RUN: %env_tool_opts=detect_write_exec=1 %run %t 2>&1 | FileCheck %s
3 // RUN: %env_tool_opts=detect_write_exec=0 %run %t 2>&1 | FileCheck %s \
4 // RUN:   --check-prefix=CHECK-DISABLED
5 // ubsan and lsan do not install mmap interceptors UNSUPPORTED: ubsan, lsan
6 
7 // TODO: Fix option on Android, it hangs there for unknown reasons.
8 // XFAIL: android
9 
10 #include <pthread.h>
11 #include <stdio.h>
12 #include <sys/mman.h>
13 
14 int main(int argc, char **argv) {
15   char *p = (char *)mmap(0, 1024, PROT_READ | PROT_WRITE | PROT_EXEC,
16                          MAP_ANONYMOUS | MAP_PRIVATE, -1, 0);
17   // CHECK: WARNING: {{.*}}Sanitizer: writable-executable page usage
18   // CHECK: #{{[0-9]+.*}}main{{.*}}mmap_write_exec.cpp:[[@LINE-3]]
19   // CHECK: SUMMARY: {{.*}}Sanitizer: w-and-x-usage
20 
21   char *q = (char *)mmap(p, 64, PROT_READ | PROT_WRITE,
22                          MAP_ANONYMOUS | MAP_PRIVATE | MAP_FIXED, -1, 0);
23   (void)mprotect(q, 64, PROT_WRITE | PROT_EXEC);
24   // CHECK: WARNING: {{.*}}Sanitizer: writable-executable page usage
25   // CHECK: #{{[0-9]+.*}}main{{.*}}mmap_write_exec.cpp:[[@LINE-2]]
26   // CHECK: SUMMARY: {{.*}}Sanitizer: w-and-x-usage
27 
28   char *a = (char *)mmap(0, 1024, PROT_READ | PROT_WRITE,
29                          MAP_ANONYMOUS | MAP_PRIVATE, -1, 0);
30   char *b = (char *)mmap(a, 64, PROT_READ | PROT_WRITE,
31                          MAP_ANONYMOUS | MAP_PRIVATE | MAP_FIXED, -1, 0);
32   (void)mprotect(q, 64, PROT_READ | PROT_EXEC);
33   // CHECK-NOT: Sanitizer
34 #if defined(__APPLE__)
35   pthread_jit_write_protect_np(false);
36   char *c = (char *)mmap(0, 128, PROT_WRITE | PROT_EXEC,
37 		         MAP_ANONYMOUS | MAP_PRIVATE | MAP_JIT, -1, 0);
38   // CHECK-NOT: Sanitizer
39 #endif
40 
41   printf("done\n");
42   // CHECK-DISABLED-NOT: Sanitizer
43   // CHECK-DISABLED: done
44 }
45