1 // RUN: %clang_asan -O0 %s -o %t
2 // RUN: %run %t 0x00 2>&1 | FileCheck %s -check-prefix=X00
3 // RUN: not %run %t 0xf1 2>&1 | FileCheck %s -check-prefix=XF1
4 // RUN: not %run %t 0xf2 2>&1 | FileCheck %s -check-prefix=XF2
5 // RUN: not %run %t 0xf3 2>&1 | FileCheck %s -check-prefix=XF3
6 // RUN: not %run %t 0xf5 2>&1 | FileCheck %s -check-prefix=XF5
7 // RUN: not %run %t 0xf8 2>&1 | FileCheck %s -check-prefix=XF8
8 
9 // XFAIL: windows-msvc
10 
11 #include <assert.h>
12 #include <sanitizer/asan_interface.h>
13 #include <stddef.h>
14 #include <stdio.h>
15 #include <stdlib.h>
16 
17 void __asan_set_shadow_00(size_t addr, size_t size);
18 void __asan_set_shadow_f1(size_t addr, size_t size);
19 void __asan_set_shadow_f2(size_t addr, size_t size);
20 void __asan_set_shadow_f3(size_t addr, size_t size);
21 void __asan_set_shadow_f5(size_t addr, size_t size);
22 void __asan_set_shadow_f8(size_t addr, size_t size);
23 
24 char a __attribute__((aligned(8)));
25 
f(long arg)26 void f(long arg) {
27   size_t shadow_offset;
28   size_t shadow_scale;
29   __asan_get_shadow_mapping(&shadow_scale, &shadow_offset);
30   size_t addr = (((size_t)&a) >> shadow_scale) + shadow_offset;
31 
32   switch (arg) {
33   // X00-NOT: AddressSanitizer
34   // X00: PASS
35   case 0x00:
36     return __asan_set_shadow_00(addr, 1);
37   // XF1: AddressSanitizer: stack-buffer-underflow
38   // XF1: [f1]
39   case 0xf1:
40     return __asan_set_shadow_f1(addr, 1);
41   // XF2: AddressSanitizer: stack-buffer-overflow
42   // XF2: [f2]
43   case 0xf2:
44     return __asan_set_shadow_f2(addr, 1);
45   // XF3: AddressSanitizer: stack-buffer-overflow
46   // XF3: [f3]
47   case 0xf3:
48     return __asan_set_shadow_f3(addr, 1);
49   // XF5: AddressSanitizer: stack-use-after-return
50   // XF5: [f5]
51   case 0xf5:
52     return __asan_set_shadow_f5(addr, 1);
53   // XF8: AddressSanitizer: stack-use-after-scope
54   // XF8: [f8]
55   case 0xf8:
56     return __asan_set_shadow_f8(addr, 1);
57   }
58   assert(0);
59 }
60 
main(int argc,char ** argv)61 int main(int argc, char **argv) {
62   assert(argc > 1);
63 
64   long arg = strtol(argv[1], 0, 16);
65   f(arg);
66   a = 1;
67   printf("PASS\n");
68   return 0;
69 }
70