xref: /llvm-project/compiler-rt/test/msan/msan_copy_shadow.cpp (revision d21b3d346af2f6189638d853182e389555e7ccb9)
1*d21b3d34SFangrui Song // Test that __msan_copy_shadow copies shadow, updates origin and does not touch
2*d21b3d34SFangrui Song // the application memory.
3*d21b3d34SFangrui Song // RUN: %clangxx_msan -fsanitize-memory-track-origins=0 -O0 %s -o %t && not %run %t 2>&1
4*d21b3d34SFangrui Song // RUN: %clangxx_msan -fsanitize-memory-track-origins=2 -O0 %s -o %t && not %run %t 2>&1 | FileCheck --check-prefix=CHECK --check-prefix=CHECK-%short-stack %s
5*d21b3d34SFangrui Song 
6*d21b3d34SFangrui Song #include <assert.h>
7*d21b3d34SFangrui Song #include <string.h>
8*d21b3d34SFangrui Song #include <sanitizer/msan_interface.h>
9*d21b3d34SFangrui Song 
main()10*d21b3d34SFangrui Song int main() {
11*d21b3d34SFangrui Song   char *a = new char[4];
12*d21b3d34SFangrui Song   char *b = new char[4];
13*d21b3d34SFangrui Song   a[1] = 1;
14*d21b3d34SFangrui Song   a[3] = 2;
15*d21b3d34SFangrui Song   memset(b, 42, 4);
16*d21b3d34SFangrui Song 
17*d21b3d34SFangrui Song   // Test that __msan_copy_shadow does not touch the contents of b[].
18*d21b3d34SFangrui Song   __msan_copy_shadow(b, a, 4);
19*d21b3d34SFangrui Song   __msan_unpoison(b, 4);
20*d21b3d34SFangrui Song   assert(b[0] == 42 && b[1] == 42 && b[2] == 42 && b[3] == 42);
21*d21b3d34SFangrui Song 
22*d21b3d34SFangrui Song   // Test that __msan_copy_shadow correctly updates shadow and origin of b[].
23*d21b3d34SFangrui Song   __msan_copy_shadow(b, a, 4);
24*d21b3d34SFangrui Song   assert(__msan_test_shadow(b, 4) == 0);
25*d21b3d34SFangrui Song   assert(__msan_test_shadow(b + 1, 3) == 1);
26*d21b3d34SFangrui Song   assert(__msan_test_shadow(b + 3, 1) == -1);
27*d21b3d34SFangrui Song   __msan_check_mem_is_initialized(b, 4);
28*d21b3d34SFangrui Song   // CHECK: use-of-uninitialized-value
29*d21b3d34SFangrui Song   // CHECK:   {{in main.*msan_copy_shadow.cpp:}}[[@LINE-2]]
30*d21b3d34SFangrui Song   // CHECK: Uninitialized value was stored to memory at
31*d21b3d34SFangrui Song   // CHECK-FULL-STACK:   {{in main.*msan_copy_shadow.cpp:}}[[@LINE-8]]
32*d21b3d34SFangrui Song   // CHECK-SHORT-STACK:   {{in __msan_copy_shadow .*msan_interceptors.cpp:}}
33*d21b3d34SFangrui Song   // CHECK: Uninitialized value was created by a heap allocation
34*d21b3d34SFangrui Song   // CHECK:   {{in main.*msan_copy_shadow.cpp:}}[[@LINE-23]]
35*d21b3d34SFangrui Song }
36