// Test thread_local_quarantine_size_kb

// RUN: %clangxx_asan  %s -o %t
// RUN: %env_asan_opts=thread_local_quarantine_size_kb=256:verbosity=1 %run %t 2>&1 | \
// RUN:   FileCheck %s --check-prefix=CHECK-VALUE
// RUN: %env_asan_opts=thread_local_quarantine_size_kb=64:quarantine_size_mb=64 %run %t 2>&1 | \
// RUN:   FileCheck %s --allow-empty --check-prefix=CHECK-SMALL-LOCAL-CACHE-SMALL-OVERHEAD
// RUN: %env_asan_opts=thread_local_quarantine_size_kb=0:quarantine_size_mb=0 %run %t 2>&1 | \
// RUN:   FileCheck %s --check-prefix=CHECK-QUARANTINE-DISABLED
// RUN: %env_asan_opts=thread_local_quarantine_size_kb=0:quarantine_size_mb=64 not %run %t 2>&1 | \
// RUN:   FileCheck %s --check-prefix=CHECK-FOR-PARAMETER-ERROR

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sanitizer/allocator_interface.h>

// The idea is allocate a lot of small blocks, totaling 5Mb of user memory
// total, and verify that quarantine does not incur too much memory overhead.
// There's always an overhead for red zones, shadow memory and such, but
// quarantine accounting should not significantly contribute to that.
static const int kNumAllocs = 20000;
static const int kAllocSize = 256;
static const size_t kHeapSizeLimit = 12 << 20;

int main() {
  size_t old_heap_size = __sanitizer_get_heap_size();
  for (int i = 0; i < kNumAllocs; i++) {
    char *g = new char[kAllocSize];
    memset(g, -1, kAllocSize);
    delete [] (g);
  }
  size_t new_heap_size = __sanitizer_get_heap_size();
  fprintf(stderr, "heap size: new: %zd old: %zd\n", new_heap_size,
          old_heap_size);
  if (new_heap_size - old_heap_size > kHeapSizeLimit)
    fprintf(stderr, "Heap size limit exceeded");
}

// CHECK-VALUE: thread_local_quarantine_size_kb=256K
// CHECK-SMALL-LOCAL-CACHE-SMALL-OVERHEAD-NOT: Heap size limit exceeded
// CHECK-QUARANTINE-DISABLED-NOT: Heap size limit exceeded
// CHECK-FOR-PARAMETER-ERROR: thread_local_quarantine_size_kb can be set to 0 only when quarantine_size_mb is set to 0