/
TestLimitCPU.java
133 lines (126 loc) · 4.54 KB
/
TestLimitCPU.java
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
129
130
131
132
133
package delight.nashornsandbox.tests;
import delight.nashornsandbox.NashornSandbox;
import delight.nashornsandbox.NashornSandboxes;
import delight.nashornsandbox.exceptions.ScriptCPUAbuseException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import org.eclipse.xtend2.lib.StringConcatenation;
import org.junit.Test;
@SuppressWarnings("all")
public class TestLimitCPU {
@Test(expected = ScriptCPUAbuseException.class)
public void test() {
final NashornSandbox sandbox = NashornSandboxes.create();
try {
sandbox.setMaxCPUTime(50);
ExecutorService _newSingleThreadExecutor = Executors.newSingleThreadExecutor();
sandbox.setExecutor(_newSingleThreadExecutor);
StringConcatenation _builder = new StringConcatenation();
_builder.append("var x = 1;");
_builder.newLine();
_builder.append("while (true) {");
_builder.newLine();
_builder.append("\t");
_builder.append("x=x+1;");
_builder.newLine();
_builder.append("}");
_builder.newLine();
sandbox.eval(_builder.toString());
} finally {
ExecutorService _executor = sandbox.getExecutor();
_executor.shutdown();
}
}
@Test(expected = ScriptCPUAbuseException.class)
public void test_evil_script() {
final NashornSandbox sandbox = NashornSandboxes.create();
try {
sandbox.setMaxCPUTime(50);
ExecutorService _newSingleThreadExecutor = Executors.newSingleThreadExecutor();
sandbox.setExecutor(_newSingleThreadExecutor);
StringConcatenation _builder = new StringConcatenation();
_builder.append("var x = 1;");
_builder.newLine();
_builder.append("while (true) { }");
_builder.newLine();
sandbox.eval(_builder.toString());
} finally {
ExecutorService _executor = sandbox.getExecutor();
_executor.shutdown();
}
}
@Test
public void test_nice_script() {
final NashornSandbox sandbox = NashornSandboxes.create();
sandbox.setMaxCPUTime(50);
ExecutorService _newSingleThreadExecutor = Executors.newSingleThreadExecutor();
sandbox.setExecutor(_newSingleThreadExecutor);
StringConcatenation _builder = new StringConcatenation();
_builder.append("var x = 1;");
_builder.newLine();
_builder.append("for (var i=0;i<=1000;i++) {");
_builder.newLine();
_builder.append("\t");
_builder.append("x = x + i");
_builder.newLine();
_builder.append("}");
_builder.newLine();
sandbox.eval(_builder.toString());
ExecutorService _executor = sandbox.getExecutor();
_executor.shutdown();
}
@Test(expected = ScriptCPUAbuseException.class)
public void test_only_while() {
final NashornSandbox sandbox = NashornSandboxes.create();
try {
sandbox.setMaxCPUTime(50);
ExecutorService _newSingleThreadExecutor = Executors.newSingleThreadExecutor();
sandbox.setExecutor(_newSingleThreadExecutor);
StringConcatenation _builder = new StringConcatenation();
_builder.append("while (true);");
_builder.newLine();
sandbox.eval(_builder.toString());
} finally {
ExecutorService _executor = sandbox.getExecutor();
_executor.shutdown();
}
}
@Test(expected = ScriptCPUAbuseException.class)
public void test_while_plus_iteration() {
final NashornSandbox sandbox = NashornSandboxes.create();
try {
sandbox.setMaxCPUTime(50);
ExecutorService _newSingleThreadExecutor = Executors.newSingleThreadExecutor();
sandbox.setExecutor(_newSingleThreadExecutor);
StringConcatenation _builder = new StringConcatenation();
_builder.append("var x=0;");
_builder.newLine();
_builder.append("while (true) x++;");
_builder.newLine();
sandbox.eval(_builder.toString());
} finally {
ExecutorService _executor = sandbox.getExecutor();
_executor.shutdown();
}
}
@Test(expected = ScriptCPUAbuseException.class)
public void test_do_while() {
final NashornSandbox sandbox = NashornSandboxes.create();
try {
sandbox.setMaxCPUTime(50);
ExecutorService _newSingleThreadExecutor = Executors.newSingleThreadExecutor();
sandbox.setExecutor(_newSingleThreadExecutor);
StringConcatenation _builder = new StringConcatenation();
_builder.append("do {");
_builder.newLine();
_builder.append("\t");
_builder.newLine();
_builder.append("} while (true);");
_builder.newLine();
sandbox.eval(_builder.toString());
} finally {
ExecutorService _executor = sandbox.getExecutor();
_executor.shutdown();
}
}
}