-
Notifications
You must be signed in to change notification settings - Fork 353
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1275 from hcoles/feature/modulepath
Pass process parameters by file
- Loading branch information
Showing
14 changed files
with
190 additions
and
242 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
164 changes: 164 additions & 0 deletions
164
pitest-entry/src/main/java/org/pitest/process/Java9Process.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,164 @@ | ||
| package org.pitest.process; | ||
|
|
||
| import static java.util.Arrays.asList; | ||
| import static org.pitest.functional.prelude.Prelude.or; | ||
| import java.io.File; | ||
| import java.io.IOException; | ||
| import java.lang.management.ManagementFactory; | ||
| import java.lang.management.RuntimeMXBean; | ||
| import java.nio.file.Files; | ||
| import java.nio.file.Path; | ||
| import java.util.ArrayList; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import java.util.Optional; | ||
| import java.util.function.Predicate; | ||
|
|
||
| import org.pitest.functional.FCollection; | ||
|
|
||
| /** | ||
| * Process for java 9+, using file to pass all parameters | ||
| */ | ||
| public class Java9Process implements WrappingProcess { | ||
|
|
||
| private final int port; | ||
| private final ProcessArgs processArgs; | ||
| private final Class<?> minionClass; | ||
| private JavaProcess process; | ||
|
|
||
| private final long pid = ProcessHandle.current().pid(); | ||
|
|
||
| private static int counter = 0; | ||
|
|
||
| public Java9Process(int port, ProcessArgs args, Class<?> minionClass) { | ||
| this.port = port; | ||
| this.processArgs = args; | ||
| this.minionClass = minionClass; | ||
| } | ||
|
|
||
| public void start() throws IOException { | ||
| String[] args = { "" + this.port }; | ||
|
|
||
| ProcessBuilder processBuilder = createProcessBuilder( | ||
| this.processArgs.getJavaExecutable(), | ||
| this.processArgs.getJvmArgs(), | ||
| this.minionClass, asList(args), | ||
| this.processArgs.getJavaAgentFinder(), | ||
| this.processArgs.getLaunchClassPath()); | ||
|
|
||
|
|
||
| configureProcessBuilder(processBuilder, this.processArgs.getWorkingDir(), | ||
| this.processArgs.getEnvironmentVariables()); | ||
|
|
||
| Process process = processBuilder.start(); | ||
| this.process = new JavaProcess(process, this.processArgs.getStdout(), | ||
| this.processArgs.getStdErr()); | ||
| } | ||
|
|
||
| public boolean isAlive() { | ||
| return process.isAlive(); | ||
| } | ||
|
|
||
| private void configureProcessBuilder(ProcessBuilder processBuilder, | ||
| File workingDirectory, Map<String, String> environmentVariables) { | ||
| processBuilder.directory(workingDirectory); | ||
| Map<String, String> environment = processBuilder.environment(); | ||
|
|
||
| environment.putAll(environmentVariables); | ||
| } | ||
|
|
||
| public void destroy() { | ||
| this.process.destroy(); | ||
| } | ||
|
|
||
| public JavaProcess getProcess() { | ||
| return this.process; | ||
| } | ||
|
|
||
| private ProcessBuilder createProcessBuilder(String javaProc, | ||
| List<String> args, Class<?> mainClass, List<String> programArgs, | ||
| JavaAgent javaAgent, String classPath) { | ||
| List<String> cmd = createLaunchArgs(javaAgent, args, mainClass, | ||
| programArgs, classPath); | ||
|
|
||
| removeJacocoAgent(cmd); | ||
|
|
||
| try { | ||
| // all arguments are passed via a temporary file, thereby avoiding command line length limits | ||
| Path argsFile = createArgsFile(cmd); | ||
| return new ProcessBuilder(asList(javaProc, "@" + argsFile.toFile().getAbsolutePath())); | ||
| } catch (IOException e) { | ||
| throw new RuntimeException(e); | ||
| } | ||
|
|
||
| } | ||
| private void removeJacocoAgent(List<String> cmd) { | ||
| removeFromClassPath(cmd, line -> line.startsWith("-javaagent") && line.contains("jacoco")); | ||
| } | ||
|
|
||
| private static void removeFromClassPath(List<String> cmd, Predicate<String> match) { | ||
| for (int i = cmd.size() - 1; i >= 0; i--) { | ||
| if (match.test(cmd.get(i))) { | ||
| cmd.remove(i); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| private List<String> createLaunchArgs(JavaAgent agentJarLocator, List<String> args, Class<?> mainClass, | ||
| List<String> programArgs, String classPath) { | ||
|
|
||
| List<String> cmd = new ArrayList<>(); | ||
|
|
||
| cmd.add("-classpath"); | ||
| cmd.add(classPath); | ||
|
|
||
| addPITJavaAgent(agentJarLocator, cmd); | ||
|
|
||
| cmd.addAll(args); | ||
|
|
||
| addLaunchJavaAgents(cmd); | ||
|
|
||
| cmd.add(mainClass.getName()); | ||
| cmd.addAll(programArgs); | ||
| return cmd; | ||
| } | ||
|
|
||
| private Path createArgsFile(List<String> cmd) throws IOException { | ||
| // To avoid conflicts with running analysis, we use the PID as part of the file name | ||
| // To prevent conflicts between multiple threads counter is used | ||
| // All files should be deleted on process exit, although some garbage may be left | ||
| // if the process is killed. Files are however created in the system temp directory | ||
| // so should be cleaned up on reboot | ||
| String name = "pitest-args-" + pid + "-" + nextCounter(); | ||
| Path args = Files.createTempFile(name, ".args"); | ||
| args.toFile().deleteOnExit(); | ||
| Files.write(args, cmd); | ||
| return args; | ||
| } | ||
|
|
||
| private static void addPITJavaAgent(JavaAgent agentJarLocator, | ||
| List<String> cmd) { | ||
| final Optional<String> jarLocation = agentJarLocator.getJarLocation(); | ||
| jarLocation.ifPresent(l -> cmd.add("-javaagent:" + l)); | ||
| } | ||
|
|
||
| private static void addLaunchJavaAgents(List<String> cmd) { | ||
| RuntimeMXBean rt = ManagementFactory.getRuntimeMXBean(); | ||
| List<String> agents = FCollection.filter(rt.getInputArguments(), | ||
| or(isJavaAgentParam(), isEnvironmentSetting())); | ||
| cmd.addAll(agents); | ||
| } | ||
|
|
||
| private static Predicate<String> isEnvironmentSetting() { | ||
| return a -> a.startsWith("-D"); | ||
| } | ||
|
|
||
| private static Predicate<String> isJavaAgentParam() { | ||
| return a -> a.toLowerCase().startsWith("-javaagent"); | ||
| } | ||
|
|
||
| private static synchronized int nextCounter() { | ||
| return counter++; | ||
| } | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.