Home | History | Annotate | Download | only in TestCases 
      1 // Checks the ASan memory address type debugging API, makes sure it returns
      2 // the correct memory type for heap, stack, global and shadow addresses and
      3 // that it correctly finds out which region (and name and size) the address
      4 // belongs to.
      5 // RUN: %clangxx_asan -O0 %s -o %t && %run %t 2>&1
      6 
      7 #include <assert.h>
      8 #include <sanitizer/asan_interface.h>
      9 #include <stdio.h>
     10 #include <stdlib.h>
     11 #include <string.h>
     12 
     13 int global_var;
     14 
     15 int main() {
     16   int local_var;
     17   char *heap_ptr = (char *)malloc(10);
     18 
     19   char name[100];
     20   void *region_address;
     21   size_t region_size;
     22   const char *type;
     23 
     24   type = __asan_locate_address(&global_var, name, 100,
     25                                &region_address, &region_size);
     26   assert(0 == strcmp(name, "global_var"));
     27   assert(0 == strcmp(type, "global"));
     28   assert(region_address == &global_var);
     29   assert(region_size == sizeof(global_var));
     30 
     31   type = __asan_locate_address((char *)(&global_var)+1, name, 100,
     32                                &region_address, &region_size);
     33   assert(0 == strcmp(name, "global_var"));
     34   assert(0 == strcmp(type, "global"));
     35   assert(region_address == &global_var);
     36   assert(region_size == sizeof(global_var));
     37 
     38   type = __asan_locate_address(&local_var, name, 100,
     39                                &region_address, &region_size);
     40   assert(0 == strcmp(name, "local_var"));
     41   assert(0 == strcmp(type, "stack"));
     42   assert(region_address == &local_var);
     43   assert(region_size == sizeof(local_var));
     44 
     45   type = __asan_locate_address((char *)(&local_var)+1, name, 100,
     46                                &region_address, &region_size);
     47   assert(0 == strcmp(name, "local_var"));
     48   assert(0 == strcmp(type, "stack"));
     49   assert(region_address == &local_var);
     50   assert(region_size == sizeof(local_var));
     51 
     52   type = __asan_locate_address(heap_ptr, name, 100,
     53                                &region_address, &region_size);
     54   assert(0 == strcmp(type, "heap"));
     55   assert(region_address == heap_ptr);
     56   assert(10 == region_size);
     57 
     58   type = __asan_locate_address(heap_ptr+1, name, 100,
     59                                &region_address, &region_size);
     60   assert(0 == strcmp(type, "heap"));
     61   assert(region_address == heap_ptr);
     62   assert(10 == region_size);
     63 
     64   size_t shadow_scale;
     65   size_t shadow_offset;
     66   __asan_get_shadow_mapping(&shadow_scale, &shadow_offset);
     67 
     68   uintptr_t shadow_ptr = (((uintptr_t)heap_ptr) >> shadow_scale)
     69                          + shadow_offset;
     70   type = __asan_locate_address((void *)shadow_ptr, NULL, 0, NULL, NULL);
     71   assert((0 == strcmp(type, "high shadow")) || 0 == strcmp(type, "low shadow"));
     72 
     73   uintptr_t shadow_gap = (shadow_ptr >> shadow_scale) + shadow_offset;
     74   type = __asan_locate_address((void *)shadow_gap, NULL, 0, NULL, NULL);
     75   assert(0 == strcmp(type, "shadow gap"));
     76 
     77   free(heap_ptr);
     78 
     79   return 0;
     80 }
     81