186277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// Check hard_rss_limit_mb. Not all sanitizers implement it yet.
286277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// RUN: %clangxx -O2 %s -o %t
386277eb844c4983c81de62d7c050e92fe7155788Stephen Hines//
486277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// Run with limit should fail:
586277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// RUN: %tool_options=hard_rss_limit_mb=100                           not %run %t 2>&1 | FileCheck %s
686277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// This run uses getrusage:
786277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// RUN: %tool_options=hard_rss_limit_mb=100:can_use_proc_maps_statm=0 not %run %t 2>&1 | FileCheck %s
886277eb844c4983c81de62d7c050e92fe7155788Stephen Hines//
986277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// Run w/o limit or with a large enough limit should pass:
1086277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// RUN: %tool_options=hard_rss_limit_mb=1000 %run %t
1186277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// RUN: %run %t
1286277eb844c4983c81de62d7c050e92fe7155788Stephen Hines//
1386277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// FIXME: make it work for other sanitizers.
1486277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// XFAIL: lsan
1586277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// XFAIL: tsan
1686277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// XFAIL: msan
1786277eb844c4983c81de62d7c050e92fe7155788Stephen Hines
1886277eb844c4983c81de62d7c050e92fe7155788Stephen Hines#include <string.h>
1986277eb844c4983c81de62d7c050e92fe7155788Stephen Hines#include <stdio.h>
2086277eb844c4983c81de62d7c050e92fe7155788Stephen Hines#include <unistd.h>
2186277eb844c4983c81de62d7c050e92fe7155788Stephen Hines
2286277eb844c4983c81de62d7c050e92fe7155788Stephen Hinesconst int kNumAllocs = 200 * 1000;
2386277eb844c4983c81de62d7c050e92fe7155788Stephen Hinesconst int kAllocSize = 1000;
2486277eb844c4983c81de62d7c050e92fe7155788Stephen Hinesvolatile char *sink[kNumAllocs];
2586277eb844c4983c81de62d7c050e92fe7155788Stephen Hines
2686277eb844c4983c81de62d7c050e92fe7155788Stephen Hinesint main(int argc, char **argv) {
2786277eb844c4983c81de62d7c050e92fe7155788Stephen Hines  for (int i = 0; i < kNumAllocs; i++) {
2886277eb844c4983c81de62d7c050e92fe7155788Stephen Hines    if ((i % 1000) == 0) {
2986277eb844c4983c81de62d7c050e92fe7155788Stephen Hines      fprintf(stderr, "[%d]\n", i);
3086277eb844c4983c81de62d7c050e92fe7155788Stephen Hines    }
3186277eb844c4983c81de62d7c050e92fe7155788Stephen Hines    char *x = new char[kAllocSize];
3286277eb844c4983c81de62d7c050e92fe7155788Stephen Hines    memset(x, 0, kAllocSize);
3386277eb844c4983c81de62d7c050e92fe7155788Stephen Hines    sink[i] = x;
3486277eb844c4983c81de62d7c050e92fe7155788Stephen Hines  }
3586277eb844c4983c81de62d7c050e92fe7155788Stephen Hines  sleep(1);  // Make sure the background thread has time to kill the process.
3686277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// CHECK: hard rss limit exhausted
3786277eb844c4983c81de62d7c050e92fe7155788Stephen Hines}
38