xref: /llvm-project/compiler-rt/test/sanitizer_common/TestCases/get_allocated_begin.cpp (revision 8362ddb6579657a1cb81735837d40e2c68f8f56e)
1 // RUN: %clangxx -O0 -g %s -o %t && %run %t
2 
3 // Must not be implemented, no other reason to install interceptors.
4 // XFAIL: ubsan
5 
6 #include <assert.h>
7 #include <sanitizer/allocator_interface.h>
8 #include <stdio.h>
9 #include <stdlib.h>
10 #include <string.h>
11 
12 // Based on lib/msan/tests/msan_test.cpp::get_allocated_size_and_ownership
main(void)13 int main(void) {
14   int sizes[] = {10, 100, 1000, 10000, 100000, 1000000};
15 
16   for (int i = 0; i < sizeof(sizes) / sizeof(int); i++) {
17     printf("Testing size %d\n", sizes[i]);
18 
19     char *array = reinterpret_cast<char *>(malloc(sizes[i]));
20     int *int_ptr = new int;
21     printf("array: %p\n", array);
22     printf("int_ptr: %p\n", int_ptr);
23 
24     // Bogus value to unpoison start. Calling __sanitizer_get_allocated_begin
25     // does not unpoison it.
26     const void *start = NULL;
27     for (int j = 0; j < sizes[i]; j++) {
28 
29       start = __sanitizer_get_allocated_begin(array + j);
30       if (array != start) {
31         printf("j: %d\n", j);
32         printf("Start: %p (expected: %p)\n", start, array);
33         fflush(stdout);
34       }
35       assert(array == start);
36     }
37 
38     start = __sanitizer_get_allocated_begin(int_ptr);
39     assert(int_ptr == start);
40 
41     void *wild_addr = reinterpret_cast<void *>(4096 * 160);
42     assert(__sanitizer_get_allocated_begin(wild_addr) == NULL);
43 
44     wild_addr = reinterpret_cast<void *>(0x1);
45     assert(__sanitizer_get_allocated_begin(wild_addr) == NULL);
46 
47     // NULL is a valid argument for GetAllocatedSize but is not owned.
48     assert(__sanitizer_get_allocated_begin(NULL) == NULL);
49 
50     free(array);
51     for (int j = 0; j < sizes[i]; j++) {
52       assert(__sanitizer_get_allocated_begin(array + j) == NULL);
53     }
54 
55     delete int_ptr;
56     assert(__sanitizer_get_allocated_begin(int_ptr) == NULL);
57   }
58 
59   return 0;
60 }
61