197ccf6b8SFangrui Song // Test for __lsan_(un)register_root_region().
297ccf6b8SFangrui Song // RUN: %clangxx_lsan %s -o %t
3*d08e5d4cSClemens Wasser // RUN: %env_lsan_opts=use_stacks=0:use_registers=0 %run %t
4*d08e5d4cSClemens Wasser // RUN: %env_lsan_opts=use_stacks=0:use_registers=0 not %run %t foo 2>&1 | FileCheck %s
5*d08e5d4cSClemens Wasser // RUN: %env_lsan_opts=use_stacks=0:use_registers=0:use_root_regions=0 not %run %t 2>&1 | FileCheck %s
697ccf6b8SFangrui Song
797ccf6b8SFangrui Song #include <assert.h>
897ccf6b8SFangrui Song #include <stdio.h>
997ccf6b8SFangrui Song #include <stdlib.h>
1097ccf6b8SFangrui Song #include <sys/mman.h>
1197ccf6b8SFangrui Song #include <unistd.h>
1297ccf6b8SFangrui Song
1397ccf6b8SFangrui Song #include <sanitizer/lsan_interface.h>
1497ccf6b8SFangrui Song
main(int argc,char * argv[])1597ccf6b8SFangrui Song int main(int argc, char *argv[]) {
1697ccf6b8SFangrui Song size_t size = getpagesize() * 2;
1797ccf6b8SFangrui Song void *p =
1897ccf6b8SFangrui Song mmap(0, size, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANON, -1, 0);
1997ccf6b8SFangrui Song assert(p);
2097ccf6b8SFangrui Song // Make half of the memory inaccessible. LSan must not crash trying to read it.
2197ccf6b8SFangrui Song assert(0 == mprotect((char *)p + size / 2, size / 2, PROT_NONE));
2297ccf6b8SFangrui Song
2397ccf6b8SFangrui Song __lsan_register_root_region(p, size);
2497ccf6b8SFangrui Song *((void **)p) = malloc(1337);
2597ccf6b8SFangrui Song fprintf(stderr, "Test alloc: %p.\n", p);
2697ccf6b8SFangrui Song if (argc > 1)
2797ccf6b8SFangrui Song __lsan_unregister_root_region(p, size);
2897ccf6b8SFangrui Song return 0;
2997ccf6b8SFangrui Song }
3097ccf6b8SFangrui Song // CHECK: Test alloc: [[ADDR:.*]].
3197ccf6b8SFangrui Song // CHECK: SUMMARY: {{(Leak|Address)}}Sanitizer: 1337 byte(s) leaked in 1 allocation(s)
32