1// Tests __hwasan_print_memory_usage.
2// REQUIRES: shell
3// RUN: %clang_hwasan %s -o %t
4// RUN: ulimit -s 1000
5// RUN: %run %t 2>&1 | FileCheck %s
6
7#include <pthread.h>
8#include <stdlib.h>
9#include <stdio.h>
10#include <string.h>
11
12#include <sanitizer/hwasan_interface.h>
13
14int state;
15__thread volatile char *sink;
16
17__attribute__((noinline))
18void *malloc_and_use(int size) {
19 char *x = (char*)malloc(size: size);
20 for (int i = 0; i < size; i++)
21 x[i] = 42; // make this memory used.
22 return x;
23}
24
25void *T1(void *arg) {
26 for (int i = 1; i <= (1 << 20); i *= 2) {
27 sink = malloc_and_use(size: i);
28 free(sink);
29 }
30
31 __sync_fetch_and_add(&state, 1);
32 while (__sync_fetch_and_add(&state, 0) != 4) {}
33 return NULL;
34}
35
36void *T4(void *arg) { return NULL; }
37
38int main() {
39 __hwasan_enable_allocator_tagging();
40 sink = malloc_and_use(size: 10);
41
42 __hwasan_print_memory_usage();
43 // CHECK: HWASAN pid: [[PID:[0-9]*]] rss: {{.*}} threads: 1 stacks: [[STACKS:[0-9]*]] thr_aux: {{.*}} stack_depot: {{.*}} uniq_stacks: [[UNIQ_STACKS:[0-9]*]] heap: [[HEAP:[0-9]*]]
44
45 void *one_meg = malloc_and_use(size: 1 << 20);
46
47 __hwasan_print_memory_usage();
48 // CHECK: HWASAN pid: [[PID]] rss: {{.*}} threads: 1 stacks: [[STACKS]] thr_aux: {{.*}} stack_depot: {{.*}}
49
50 free(ptr: one_meg);
51
52 __hwasan_print_memory_usage();
53 // CHECK: HWASAN pid: [[PID]] rss: {{.*}} threads: 1 stacks: [[STACKS]] thr_aux: {{.*}} stack_depot: {{.*}} uniq_stacks: {{.*}} heap: [[HEAP]]
54
55 pthread_t t1, t2, t3, t4;
56
57 pthread_create(newthread: &t1, NULL, start_routine: T1, NULL);
58 pthread_create(newthread: &t2, NULL, start_routine: T1, NULL);
59 pthread_create(newthread: &t3, NULL, start_routine: T1, NULL);
60 pthread_create(newthread: &t4, NULL, start_routine: T4, NULL);
61 while (__sync_fetch_and_add(&state, 0) != 3) {}
62 pthread_join(th: t4, NULL);
63
64 __hwasan_print_memory_usage();
65 // CHECK: HWASAN pid: [[PID]] rss: {{.*}} threads: 4 stacks:
66
67 __sync_fetch_and_add(&state, 1);
68 pthread_join(th: t1, NULL);
69 pthread_join(th: t2, NULL);
70 pthread_join(th: t3, NULL);
71 __hwasan_print_memory_usage();
72 // CHECK: HWASAN pid: [[PID]] rss: {{.*}} threads: 1 stacks: [[STACKS]]
73}
74

source code of compiler-rt/test/hwasan/TestCases/print-memory-usage.c