malloc_zone.cpp
1.58 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
// Check that malloc_default_zone and malloc_zone_from_ptr return the
// sanitizer-installed malloc zone even when MallocStackLogging (MSL) is
// requested. This prevents crashes in certain situations. Note that the
// sanitizers and MSL cannot be used together. If both are enabled, MSL
// functionality is essentially deactivated since it only hooks the default
// allocator which is replaced by a custom sanitizer allocator.
//
// MSL=lite creates its own special malloc zone, copies the passed zone name,
// and leaks it.
// RUN: echo "leak:create_and_insert_msl_lite_zone" >> lsan.supp
//
// RUN: %clangxx -g %s -o %t
// RUN: %run %t | FileCheck %s
// RUN: %env MallocStackLogging=lite LSAN_OPTIONS=suppressions=lsan.supp %run %t | FileCheck %s
// RUN: %env MallocStackLogging=full %run %t | FileCheck %s
//
// UBSan does not install a malloc zone.
// XFAIL: ubsan
//
#include <malloc/malloc.h>
#include <stdlib.h>
#include <stdio.h>
int main(void) {
malloc_zone_t *default_zone = malloc_default_zone();
printf("default zone name: %s\n", malloc_get_zone_name(default_zone));
// CHECK: default zone name: {{a|l|t}}san
void *ptr1 = malloc(10);
void *ptr2 = malloc_zone_malloc(default_zone, 10);
malloc_zone_t* zone1 = malloc_zone_from_ptr(ptr1);
malloc_zone_t* zone2 = malloc_zone_from_ptr(ptr2);
printf("zone1: %d\n", zone1 == default_zone);
printf("zone2: %d\n", zone2 == default_zone);
// CHECK: zone1: 1
// CHECK: zone2: 1
free(ptr1);
malloc_zone_free(zone2, ptr2);
return 0;
}