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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* Copyright (C) 2009 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
public class Main {
static class ArrayMemEater {
static boolean sawOome;
static void blowup(char[][] holder) {
try {
for (int i = 0; i < holder.length; ++i) {
holder[i] = new char[1024 * 1024];
}
} catch (OutOfMemoryError oome) {
ArrayMemEater.sawOome = true;
}
}
}
static class InstanceMemEater {
static boolean sawOome;
static InstanceMemEater hook;
InstanceMemEater next;
double d1, d2, d3, d4, d5, d6, d7, d8; // Bloat this object so we fill the heap faster.
static InstanceMemEater allocate() {
try {
return new InstanceMemEater();
} catch (OutOfMemoryError e) {
InstanceMemEater.sawOome = true;
return null;
}
}
static void confuseCompilerOptimization(InstanceMemEater instance) {
hook = instance;
}
}
static boolean triggerArrayOOM() {
ArrayMemEater.blowup(new char[128 * 1024][]);
return ArrayMemEater.sawOome;
}
static boolean triggerInstanceOOM() {
InstanceMemEater memEater = InstanceMemEater.allocate();
InstanceMemEater lastMemEater = memEater;
do {
lastMemEater.next = InstanceMemEater.allocate();
lastMemEater = lastMemEater.next;
} while (lastMemEater != null);
memEater.confuseCompilerOptimization(memEater);
InstanceMemEater.hook = null;
return InstanceMemEater.sawOome;
}
public static void main(String[] args) {
if (triggerReflectionOOM()) {
System.out.println("Test reflection correctly threw");
}
if (triggerArrayOOM()) {
System.out.println("NEW_ARRAY correctly threw OOME");
}
if (triggerInstanceOOM()) {
System.out.println("NEW_INSTANCE correctly threw OOME");
}
}
static Object[] holder;
public static void blowup() throws Exception {
int size = 32 * 1024 * 1024;
for (int i = 0; i < holder.length; ) {
try {
holder[i] = new char[size];
i++;
} catch (OutOfMemoryError oome) {
size = size / 2;
if (size == 0) {
break;
}
}
}
holder[0] = new char[100000];
}
static boolean triggerReflectionOOM() {
try {
Class<?> c = Main.class;
Method m = c.getMethod("blowup", (Class[]) null);
holder = new Object[1000000];
m.invoke(null);
holder = null;
System.out.println("Didn't throw from blowup");
} catch (OutOfMemoryError e) {
holder = null;
} catch (InvocationTargetException e) {
holder = null;
if (!(e.getCause() instanceof OutOfMemoryError)) {
System.out.println("InvocationTargetException cause not OOME " + e.getCause());
return false;
}
} catch (Exception e) {
holder = null;
System.out.println("Unexpected exception " + e);
return false;
}
return true;
}
}
|
