1*d21b3d34SFangrui Song // RUN: %clangxx_msan -DPRE1 -O0 %s -o %t && not %run %t 2>&1
2*d21b3d34SFangrui Song // RUN: %clangxx_msan -DPRE2 -O0 %s -o %t && not %run %t 2>&1
3*d21b3d34SFangrui Song // RUN: %clangxx_msan -DPRE3 -O0 %s -o %t && not %run %t 2>&1
4*d21b3d34SFangrui Song // RUN: %clangxx_msan -O0 %s -o %t && %run %t 2>&1
5*d21b3d34SFangrui Song
6*d21b3d34SFangrui Song #include <assert.h>
7*d21b3d34SFangrui Song #include <signal.h>
8*d21b3d34SFangrui Song #include <string.h>
9*d21b3d34SFangrui Song
10*d21b3d34SFangrui Song #include <sanitizer/linux_syscall_hooks.h>
11*d21b3d34SFangrui Song #include <sanitizer/msan_interface.h>
12*d21b3d34SFangrui Song
13*d21b3d34SFangrui Song struct my_kernel_sigaction {
14*d21b3d34SFangrui Song #if defined(__mips__)
15*d21b3d34SFangrui Song long flags, handler;
16*d21b3d34SFangrui Song #else
17*d21b3d34SFangrui Song long handler, flags, restorer;
18*d21b3d34SFangrui Song #endif
19*d21b3d34SFangrui Song uint64_t mask[20]; // larger than any known platform
20*d21b3d34SFangrui Song };
21*d21b3d34SFangrui Song
main()22*d21b3d34SFangrui Song int main() {
23*d21b3d34SFangrui Song my_kernel_sigaction act = {}, oldact = {};
24*d21b3d34SFangrui Song
25*d21b3d34SFangrui Song #if defined(PRE1)
26*d21b3d34SFangrui Song __msan_poison(&act.handler, sizeof(act.handler));
27*d21b3d34SFangrui Song __sanitizer_syscall_pre_rt_sigaction(SIGUSR1, &act, &oldact, 20 * 8);
28*d21b3d34SFangrui Song #elif defined(PRE2)
29*d21b3d34SFangrui Song __msan_poison(&act.flags, sizeof(act.flags));
30*d21b3d34SFangrui Song __sanitizer_syscall_pre_rt_sigaction(SIGUSR1, &act, &oldact, 20 * 8);
31*d21b3d34SFangrui Song #elif defined(PRE3)
32*d21b3d34SFangrui Song __msan_poison(&act.mask, 1);
33*d21b3d34SFangrui Song __sanitizer_syscall_pre_rt_sigaction(SIGUSR1, &act, &oldact, 20 * 8);
34*d21b3d34SFangrui Song #else
35*d21b3d34SFangrui Song // Uninit past the end of the mask is ignored.
36*d21b3d34SFangrui Song __msan_poison(((char *)&act.mask) + 5, 1);
37*d21b3d34SFangrui Song __sanitizer_syscall_pre_rt_sigaction(SIGUSR1, &act, &oldact, 5);
38*d21b3d34SFangrui Song
39*d21b3d34SFangrui Song memset(&act, 0, sizeof(act));
40*d21b3d34SFangrui Song __msan_poison(&oldact, sizeof(oldact));
41*d21b3d34SFangrui Song __sanitizer_syscall_post_rt_sigaction(0, SIGUSR1, &act, &oldact, 5);
42*d21b3d34SFangrui Song #if defined(__mips__)
43*d21b3d34SFangrui Song assert(__msan_test_shadow(&oldact, sizeof(oldact)) == sizeof(long)*2 + 5);
44*d21b3d34SFangrui Song #else
45*d21b3d34SFangrui Song assert(__msan_test_shadow(&oldact, sizeof(oldact)) == sizeof(long)*3 + 5);
46*d21b3d34SFangrui Song #endif
47*d21b3d34SFangrui Song #endif
48*d21b3d34SFangrui Song }
49