reference, declaration → definition
definition → references, declarations, derived classes, virtual overrides
reference to multiple definitions → definitions
unreferenced

    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
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72

// Tests __hwasan_print_memory_usage.
// RUN: %clang_hwasan %s -o %t
// RUN: ulimit -s 1000
// RUN: %run %t 2>&1 | FileCheck %s
// REQUIRES: stable-runtime

#include <pthread.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#include <sanitizer/hwasan_interface.h>

int state;
__thread volatile char *sink;

__attribute__((noinline))
void *malloc_and_use(int size) {
  char *x = (char*)malloc(size);
  for (int i = 0; i < size; i++)
    x[i] = 42;  // make this memory used.
  return x;
}

void *T1(void *arg) {

  for (int i = 1; i <= (1 << 20); i *= 2)
    sink = malloc_and_use(i);

  __sync_fetch_and_add(&state, 1);
  while (__sync_fetch_and_add(&state, 0) != 4) {}
  return NULL;
}

void *T4(void *arg) { return NULL; }

int main() {
  __hwasan_enable_allocator_tagging();
  sink = malloc_and_use(10);

  __hwasan_print_memory_usage();
  // 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]*]]

  void *one_meg = malloc_and_use(1 << 20);

  __hwasan_print_memory_usage();
  // CHECK: HWASAN pid: [[PID]] rss: {{.*}} threads: 1 stacks: [[STACKS]] thr_aux: {{.*}} stack_depot: {{.*}}

  free(one_meg);

  __hwasan_print_memory_usage();
  // CHECK: HWASAN pid: [[PID]] rss: {{.*}} threads: 1 stacks: [[STACKS]] thr_aux: {{.*}} stack_depot: {{.*}} uniq_stacks: {{.*}} heap: [[HEAP]]

  pthread_t t1, t2, t3, t4;

  pthread_create(&t1, NULL, T1, NULL);
  pthread_create(&t2, NULL, T1, NULL);
  pthread_create(&t3, NULL, T1, NULL);
  pthread_create(&t4, NULL, T4, NULL);
  while (__sync_fetch_and_add(&state, 0) != 3) {}
  pthread_join(t4, NULL);

  __hwasan_print_memory_usage();
  // CHECK: HWASAN pid: [[PID]] rss: {{.*}} threads: 4 stacks:

  __sync_fetch_and_add(&state, 1);
  pthread_join(t1, NULL);
  pthread_join(t2, NULL);
  pthread_join(t3, NULL);
  __hwasan_print_memory_usage();
  // CHECK: HWASAN pid: [[PID]] rss: {{.*}} threads: 1 stacks: [[STACKS]]
}