1 // Test the behavior of malloc/calloc/realloc when the allocation causes OOM
2 // in the secondary allocator.
3 // By default (allocator_may_return_null=0) the process should crash.
4 // With allocator_may_return_null=1 the allocator should return 0.
5 // Set the limit to 20.5T on 64 bits to account for ASan shadow memory,
6 // allocator buffers etc. so that the test allocation of ~1T will trigger OOM.
7 // Limit this test to Linux since we're relying on allocator internal
8 // limits (shadow memory size, allocation limits etc.)
9
10 // RUN: %clangxx_asan -O0 %s -o %t
11 // RUN: ulimit -v 22024290304
12 // RUN: not %run %t malloc 2>&1 \
13 // RUN: | FileCheck %s --check-prefixes=CHECK-MALLOC,CHECK-CRASH
14 // RUN: %env_asan_opts=allocator_may_return_null=0 not %run %t malloc 2>&1 \
15 // RUN: | FileCheck %s --check-prefixes=CHECK-MALLOC,CHECK-CRASH
16 // RUN: %env_asan_opts=allocator_may_return_null=1 %run %t malloc 2>&1 \
17 // RUN: | FileCheck %s --check-prefixes=CHECK-MALLOC,CHECK-NULL
18 // RUN: %env_asan_opts=allocator_may_return_null=0 not %run %t calloc 2>&1 \
19 // RUN: | FileCheck %s --check-prefixes=CHECK-CALLOC,CHECK-CRASH
20 // RUN: %env_asan_opts=allocator_may_return_null=1 %run %t calloc 2>&1 \
21 // RUN: | FileCheck %s --check-prefixes=CHECK-CALLOC,CHECK-NULL
22 // RUN: %env_asan_opts=allocator_may_return_null=0 not %run %t realloc 2>&1 \
23 // RUN: | FileCheck %s --check-prefixes=CHECK-REALLOC,CHECK-CRASH
24 // RUN: %env_asan_opts=allocator_may_return_null=1 %run %t realloc 2>&1 \
25 // RUN: | FileCheck %s --check-prefixes=CHECK-REALLOC,CHECK-NULL
26 // RUN: %env_asan_opts=allocator_may_return_null=0 not %run %t realloc-after-malloc 2>&1 \
27 // RUN: | FileCheck %s --check-prefixes=CHECK-MALLOC-REALLOC,CHECK-CRASH
28 // RUN: %env_asan_opts=allocator_may_return_null=1 %run %t realloc-after-malloc 2>&1 \
29 // RUN: | FileCheck %s --check-prefixes=CHECK-MALLOC-REALLOC,CHECK-NULL
30
31 // ASan shadow memory on s390 is too large for this test.
32 // AArch64 bots fail on this test.
33 // TODO(alekseys): Android lit do not run ulimit on device.
34 // REQUIRES: shadow-scale-3
35 // UNSUPPORTED: s390,android,aarch64,powerpc64le
36
37 #include <stdlib.h>
38 #include <string.h>
39 #include <stdio.h>
40 #include <assert.h>
41
main(int argc,char ** argv)42 int main(int argc, char **argv) {
43 assert(argc == 2);
44 const char *action = argv[1];
45 fprintf(stderr, "%s:\n", action);
46
47 // Allocate just a bit less than max allocation size enforced by ASan's
48 // allocator (currently 1T and 3G).
49 const size_t size =
50 #if __LP64__
51 (1ULL << 40) - (1ULL << 30);
52 #else
53 (3ULL << 30) - (1ULL << 20);
54 #endif
55
56 void *x = 0;
57
58 if (!strcmp(action, "malloc")) {
59 x = malloc(size);
60 } else if (!strcmp(action, "calloc")) {
61 x = calloc(size / 4, 4);
62 } else if (!strcmp(action, "realloc")) {
63 x = realloc(0, size);
64 } else if (!strcmp(action, "realloc-after-malloc")) {
65 char *t = (char*)malloc(100);
66 *t = 42;
67 x = realloc(t, size);
68 assert(*t == 42);
69 free(t);
70 } else {
71 assert(0);
72 }
73
74 // The NULL pointer is printed differently on different systems, while (long)0
75 // is always the same.
76 fprintf(stderr, "x: %lx\n", (long)x);
77 free(x);
78
79 return x != 0;
80 }
81
82 // CHECK-MALLOC: malloc:
83 // CHECK-CALLOC: calloc:
84 // CHECK-REALLOC: realloc:
85 // CHECK-MALLOC-REALLOC: realloc-after-malloc:
86
87 // CHECK-CRASH: SUMMARY: AddressSanitizer: out-of-memory
88 // CHECK-NULL: x: 0
89