1*673dc3d4SNico Weber #include "sanitizer\allocator_interface.h"
2*673dc3d4SNico Weber #include <cassert>
3*673dc3d4SNico Weber #include <stdio.h>
4*673dc3d4SNico Weber #include <windows.h>
5*673dc3d4SNico Weber // RUN: %clang_cl_asan %s -o%t
6*673dc3d4SNico Weber // RUN: %env_asan_opts=windows_hook_rtl_allocators=true %run %t 2>&1 | FileCheck %s
7*673dc3d4SNico Weber // UNSUPPORTED: asan-64-bits
8*673dc3d4SNico Weber
main()9*673dc3d4SNico Weber int main() {
10*673dc3d4SNico Weber //owned by rtl
11*673dc3d4SNico Weber void *alloc = HeapAlloc(GetProcessHeap(), HEAP_GENERATE_EXCEPTIONS, 100);
12*673dc3d4SNico Weber assert(alloc);
13*673dc3d4SNico Weber // still owned by rtl
14*673dc3d4SNico Weber alloc = HeapReAlloc(GetProcessHeap(), HEAP_GENERATE_EXCEPTIONS, alloc, 100);
15*673dc3d4SNico Weber assert(alloc && !__sanitizer_get_ownership(alloc) && HeapValidate(GetProcessHeap(), 0, alloc));
16*673dc3d4SNico Weber //convert to asan owned
17*673dc3d4SNico Weber void *realloc = HeapReAlloc(GetProcessHeap(), 0, alloc, 500);
18*673dc3d4SNico Weber alloc = nullptr;
19*673dc3d4SNico Weber assert(realloc && __sanitizer_get_ownership(realloc));
20*673dc3d4SNico Weber //convert back to rtl owned;
21*673dc3d4SNico Weber alloc = HeapReAlloc(GetProcessHeap(), HEAP_GENERATE_EXCEPTIONS, realloc, 100);
22*673dc3d4SNico Weber assert(alloc && !__sanitizer_get_ownership(alloc) && HeapValidate(GetProcessHeap(), 0, alloc));
23*673dc3d4SNico Weber printf("Success\n");
24*673dc3d4SNico Weber }
25*673dc3d4SNico Weber
26*673dc3d4SNico Weber // CHECK-NOT: assert
27*673dc3d4SNico Weber // CHECK-NOT: AddressSanitizer
28*673dc3d4SNico Weber // CHECK: Success