undefect. CWE-407 — 63 sites patched across 27 ecosystems
Authors: russell@unturf.com · brackishbert@gmail.com · foxhop.net · TimeHexOn.com Patches, unit tests, benchmarks, whitepaper, and outreach briefs. Public domain — no copyright claimed. Use freely.
This commit is contained in:
commit
0a580b313d
70422 changed files with 17213626 additions and 0 deletions
131
test/langtools/jdk/jshell/ToolProviderTest.java
Normal file
131
test/langtools/jdk/jshell/ToolProviderTest.java
Normal file
|
|
@ -0,0 +1,131 @@
|
|||
/*
|
||||
* Copyright (c) 2016, 2025, Oracle and/or its affiliates. All rights reserved.
|
||||
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
||||
*
|
||||
* This code is free software; you can redistribute it and/or modify it
|
||||
* under the terms of the GNU General Public License version 2 only, as
|
||||
* published by the Free Software Foundation.
|
||||
*
|
||||
* This code is distributed in the hope that it will be useful, but WITHOUT
|
||||
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||||
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||||
* version 2 for more details (a copy is included in the LICENSE file that
|
||||
* accompanied this code).
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License version
|
||||
* 2 along with this work; if not, write to the Free Software Foundation,
|
||||
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
||||
*
|
||||
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
||||
* or visit www.oracle.com if you need additional information or have any
|
||||
* questions.
|
||||
*/
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.HashMap;
|
||||
import java.util.ServiceLoader;
|
||||
import java.util.function.Consumer;
|
||||
import java.util.function.Function;
|
||||
import javax.tools.Tool;
|
||||
import jdk.internal.jshell.tool.JShellToolProvider;
|
||||
import jdk.jshell.tool.JavaShellToolBuilder;
|
||||
import static org.junit.jupiter.api.Assertions.assertTrue;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
/*
|
||||
* @test
|
||||
* @bug 8170044 8171343 8179856 8185840 8190383 8353332
|
||||
* @summary Test ServiceLoader launching of jshell tool
|
||||
* @modules jdk.compiler/com.sun.tools.javac.api
|
||||
* jdk.compiler/com.sun.tools.javac.main
|
||||
* jdk.jdeps/com.sun.tools.javap
|
||||
* jdk.jshell/jdk.internal.jshell.tool:+open
|
||||
* jdk.jshell/jdk.internal.jshell.tool.resources:+open
|
||||
* @library /tools/lib
|
||||
* @build Compiler toolbox.ToolBox
|
||||
* @run junit/othervm --patch-module jdk.jshell=${test.src}/StartOptionTest-module-patch ToolProviderTest
|
||||
*/
|
||||
public class ToolProviderTest extends StartOptionTest {
|
||||
|
||||
// Through the provider, the console and console go to command out (we assume,
|
||||
// because it works with the current tests) that console and user output are
|
||||
// after command out.
|
||||
@Override
|
||||
protected void startExCoUoCeCn(int expectedExitCode,
|
||||
String expectedCmdOutput,
|
||||
String expectedUserOutput,
|
||||
String expectedError,
|
||||
String expectedConsole,
|
||||
String... args) {
|
||||
super.startExCoUoCeCn(expectedExitCode,
|
||||
(expectedCmdOutput == null? "" : expectedCmdOutput) +
|
||||
(expectedConsole == null? "" : expectedConsole) +
|
||||
(expectedUserOutput == null? "" : expectedUserOutput),
|
||||
null, expectedError, null, args);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void startCheckUserOutput(Consumer<String> checkUserOutput, String... args) {
|
||||
runShell(args);
|
||||
check(cmdout, checkUserOutput, "userout");
|
||||
check(usererr, null, "usererr");
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void startCheckCommandUserOutput(Consumer<String> checkCommandOutput,
|
||||
Consumer<String> checkUserOutput,
|
||||
Consumer<String> checkCombinedCommandUserOutput,
|
||||
String... args) {
|
||||
runShell(args);
|
||||
check(cmdout, checkCombinedCommandUserOutput, "userout");
|
||||
check(usererr, null, "usererr");
|
||||
}
|
||||
|
||||
@Override
|
||||
protected int runShell(String... args) {
|
||||
//make sure the JShell running during the test is not using persisted preferences from the machine:
|
||||
Function<JavaShellToolBuilder, JavaShellToolBuilder> prevAugmentedToolBuilder =
|
||||
getAndSetAugmentedToolBuilder(builder -> builder.persistence(getThisTestPersistence()));
|
||||
try {
|
||||
ServiceLoader<Tool> sl = ServiceLoader.load(Tool.class);
|
||||
for (Tool provider : sl) {
|
||||
if (provider.name().equals("jshell")) {
|
||||
return provider.run(cmdInStream, cmdout, cmderr, args);
|
||||
}
|
||||
}
|
||||
throw new AssertionError("Repl tool not found by ServiceLoader: " + sl);
|
||||
} finally {
|
||||
getAndSetAugmentedToolBuilder(prevAugmentedToolBuilder);
|
||||
}
|
||||
}
|
||||
|
||||
// Test --show-version
|
||||
@Override
|
||||
@Test
|
||||
public void testShowVersion() {
|
||||
startCo(s -> {
|
||||
assertTrue(s.startsWith("jshell "), "unexpected version: " + s);
|
||||
assertTrue(s.contains("Welcome"), "Expected start (but got no welcome): " + s);
|
||||
assertTrue(s.trim().contains("jshell>"), "Expected prompt, got: " + s);
|
||||
},
|
||||
"--show-version");
|
||||
}
|
||||
|
||||
private Function<JavaShellToolBuilder, JavaShellToolBuilder> getAndSetAugmentedToolBuilder
|
||||
(Function<JavaShellToolBuilder, JavaShellToolBuilder> augmentToolBuilder) {
|
||||
try {
|
||||
Field f = JShellToolProvider.class.getDeclaredField("augmentToolBuilder");
|
||||
|
||||
f.setAccessible(true);
|
||||
|
||||
Function<JavaShellToolBuilder, JavaShellToolBuilder> prev =
|
||||
(Function<JavaShellToolBuilder, JavaShellToolBuilder>) f.get(null);
|
||||
|
||||
f.set(null, augmentToolBuilder);
|
||||
|
||||
return prev;
|
||||
} catch (ReflectiveOperationException ex) {
|
||||
throw new IllegalStateException(ex);
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue