| /* |
| * Copyright (c) 2015, 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. |
| */ |
| |
| package compiler.compilercontrol.share.scenario; |
| |
| import compiler.compilercontrol.share.actions.BaseAction; |
| import compiler.compilercontrol.share.method.MethodDescriptor; |
| import compiler.compilercontrol.share.processors.CommandProcessor; |
| import compiler.compilercontrol.share.processors.LogProcessor; |
| import compiler.compilercontrol.share.processors.PrintProcessor; |
| import compiler.compilercontrol.share.processors.QuietProcessor; |
| import jdk.test.lib.Asserts; |
| import jdk.test.lib.OutputAnalyzer; |
| import jdk.test.lib.Pair; |
| import jdk.test.lib.ProcessTools; |
| import jdk.test.lib.dcmd.CommandExecutorException; |
| import jdk.test.lib.dcmd.JcmdExecutor; |
| import pool.PoolHelper; |
| |
| import java.io.BufferedReader; |
| import java.io.IOException; |
| import java.io.InputStreamReader; |
| import java.io.PrintWriter; |
| import java.lang.reflect.Executable; |
| import java.net.ServerSocket; |
| import java.net.Socket; |
| import java.util.ArrayList; |
| import java.util.Arrays; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.LinkedHashSet; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Set; |
| import java.util.concurrent.Callable; |
| import java.util.function.Consumer; |
| |
| /** |
| * Test scenario |
| */ |
| public final class Scenario { |
| private final boolean isValid; |
| private final List<String> vmopts; |
| private final Map<Executable, State> states; |
| private final List<Consumer<OutputAnalyzer>> processors; |
| private final List<String> jcmdExecCommands; |
| |
| private Scenario(boolean isValid, |
| List<String> vmopts, |
| Map<Executable, State> states, |
| List<CompileCommand> compileCommands, |
| List<JcmdCommand> jcmdCommands) { |
| this.isValid = isValid; |
| this.vmopts = vmopts; |
| this.states = states; |
| processors = new ArrayList<>(); |
| processors.add(new LogProcessor(states)); |
| processors.add(new PrintProcessor(states)); |
| List<CompileCommand> nonQuieted = new ArrayList<>(); |
| List<CompileCommand> quieted = new ArrayList<>(); |
| boolean metQuiet = false; |
| for (CompileCommand cc : compileCommands) { |
| metQuiet |= cc.command == Command.QUIET; |
| if (metQuiet) { |
| quieted.add(cc); |
| } else { |
| nonQuieted.add(cc); |
| } |
| } |
| processors.add(new CommandProcessor(nonQuieted)); |
| processors.add(new QuietProcessor(quieted)); |
| jcmdExecCommands = new ArrayList<>(); |
| boolean addCommandMet = false; |
| for (JcmdCommand cmd : jcmdCommands) { |
| switch (cmd.jcmdType) { |
| case ADD: |
| if (!addCommandMet) { |
| jcmdExecCommands.add(JcmdType.ADD.command); |
| } |
| addCommandMet = true; |
| break; |
| default: |
| jcmdExecCommands.add(cmd.jcmdType.command); |
| break; |
| } |
| } |
| } |
| |
| /** |
| * Executes scenario |
| */ |
| public void execute() { |
| // Construct execution command with CompileCommand and class |
| List<String> argsList = new ArrayList<>(); |
| // Add VM options |
| argsList.addAll(vmopts); |
| // Add class name that would be executed in a separate VM |
| String classCmd = BaseAction.class.getName(); |
| argsList.add(classCmd); |
| OutputAnalyzer output; |
| try (ServerSocket serverSocket = new ServerSocket(0)) { |
| if (isValid) { |
| // Get port test VM will connect to |
| int port = serverSocket.getLocalPort(); |
| if (port == -1) { |
| throw new Error("Socket is not bound: " + port); |
| } |
| argsList.add(String.valueOf(port)); |
| // Start separate thread to connect with test VM |
| new Thread(() -> connectTestVM(serverSocket)).start(); |
| } |
| // Start test VM |
| output = ProcessTools.executeTestJvmAllArgs( |
| argsList.toArray(new String[argsList.size()])); |
| } catch (Throwable thr) { |
| throw new Error("Execution failed", thr); |
| } |
| if (isValid) { |
| output.shouldHaveExitValue(0); |
| for (Consumer<OutputAnalyzer> processor : processors) { |
| processor.accept(output); |
| } |
| } else { |
| Asserts.assertNE(output.getExitValue(), 0, "VM should exit with " |
| + "error for incorrect directives"); |
| output.shouldContain("Parsing of compiler directives failed"); |
| } |
| } |
| |
| /* |
| * Performs connection with a test VM, sends method states and performs |
| * JCMD operations on a test VM. |
| */ |
| private void connectTestVM(ServerSocket serverSocket) { |
| /* |
| * There are no way to prove that accept was invoked before we started |
| * test VM that connects to this serverSocket. Connection timeout is |
| * enough |
| */ |
| try ( |
| Socket socket = serverSocket.accept(); |
| PrintWriter pw = new PrintWriter(socket.getOutputStream(), |
| true); |
| BufferedReader in = new BufferedReader(new InputStreamReader( |
| socket.getInputStream()))) { |
| // Get pid of the executed process |
| int pid = Integer.parseInt(in.readLine()); |
| Asserts.assertNE(pid, 0, "Got incorrect pid"); |
| executeJCMD(pid); |
| // serialize and send state map |
| for (Executable x : states.keySet()) { |
| pw.println("{"); |
| pw.println(x.toGenericString()); |
| pw.println(states.get(x).toString()); |
| pw.println("}"); |
| } |
| } catch (IOException e) { |
| throw new Error("Failed to write data", e); |
| } |
| } |
| |
| // Executes all diagnostic commands |
| private void executeJCMD(int pid) { |
| for (String command : jcmdExecCommands) { |
| new JcmdExecutor() { |
| @Override |
| protected List<String> createCommandLine(String cmd) |
| throws CommandExecutorException { |
| return Arrays.asList(jcmdBinary, Integer.toString(pid), |
| cmd); |
| } |
| }.execute(command); |
| } |
| } |
| |
| /** |
| * Gets states of methods for this scenario |
| * |
| * @return pairs of executable and its state |
| */ |
| public Map<Executable, State> getStates() { |
| return states; |
| } |
| |
| public static enum Compiler { |
| C1("c1"), |
| C2("c2"); |
| |
| public final String name; |
| |
| Compiler(String name) { |
| this.name = name; |
| } |
| } |
| |
| /** |
| * Type of diagnostic (jcmd) command |
| */ |
| public static enum JcmdType { |
| ADD("Compiler.directives_add " + Type.JCMD.fileName), |
| PRINT("Compiler.directives_print"), |
| CLEAR("Compiler.directives_clear"), |
| REMOVE("Compiler.directives_remove"); |
| |
| public final String command; |
| private JcmdType(String command) { |
| this.command = command; |
| } |
| } |
| |
| /** |
| * Type of the compile command |
| */ |
| public static enum Type { |
| OPTION(""), |
| FILE("command_file"), |
| DIRECTIVE("directives.json"), |
| JCMD("jcmd_directives.json") { |
| @Override |
| public CompileCommand createCompileCommand(Command command, |
| MethodDescriptor md, Compiler compiler) { |
| return new JcmdCommand(command, md, compiler, this, |
| JcmdType.ADD); |
| } |
| }; |
| |
| public final String fileName; |
| |
| public CompileCommand createCompileCommand(Command command, |
| MethodDescriptor md, Compiler compiler) { |
| return new CompileCommand(command, md, compiler, this); |
| } |
| |
| private Type(String fileName) { |
| this.fileName = fileName; |
| } |
| } |
| |
| public static Builder getBuilder() { |
| return new Builder(); |
| } |
| |
| public static class Builder { |
| private final Set<String> vmopts = new LinkedHashSet<>(); |
| private final Map<Type, StateBuilder<CompileCommand>> builders |
| = new HashMap<>(); |
| private final JcmdStateBuilder jcmdStateBuilder; |
| |
| public Builder() { |
| builders.put(Type.FILE, new CommandFileBuilder(Type.FILE.fileName)); |
| builders.put(Type.OPTION, new CommandOptionsBuilder()); |
| builders.put(Type.DIRECTIVE, new DirectiveBuilder( |
| Type.DIRECTIVE.fileName)); |
| jcmdStateBuilder = new JcmdStateBuilder(Type.JCMD.fileName); |
| } |
| |
| public void add(CompileCommand compileCommand) { |
| String[] vmOptions = compileCommand.command.vmOpts; |
| Collections.addAll(vmopts, vmOptions); |
| if (compileCommand.type == Type.JCMD) { |
| jcmdStateBuilder.add((JcmdCommand) compileCommand); |
| } else { |
| StateBuilder<CompileCommand> builder = builders.get( |
| compileCommand.type); |
| if (builder == null) { |
| throw new Error("TESTBUG: Missing builder for the type: " |
| + compileCommand.type); |
| } |
| builder.add(compileCommand); |
| } |
| } |
| |
| public Scenario build() { |
| boolean isValid = true; |
| |
| // Get states from each of the state builders |
| Map<Executable, State> commandFileStates |
| = builders.get(Type.FILE).getStates(); |
| Map<Executable, State> commandOptionStates |
| = builders.get(Type.OPTION).getStates(); |
| Map<Executable, State> directiveFileStates |
| = builders.get(Type.DIRECTIVE).getStates(); |
| |
| // get all jcmd commands |
| List<JcmdCommand> jcmdCommands = jcmdStateBuilder |
| .getCompileCommands(); |
| boolean isClearedState = false; |
| if (jcmdClearedState(jcmdCommands)) { |
| isClearedState = true; |
| } |
| |
| // Merge states |
| List<Pair<Executable, Callable<?>>> methods = new PoolHelper() |
| .getAllMethods(); |
| Map<Executable, State> finalStates = new HashMap<>(); |
| Map<Executable, State> jcmdStates = jcmdStateBuilder.getStates(); |
| for (Pair<Executable, Callable<?>> pair : methods) { |
| Executable x = pair.first; |
| State commandOptionState = commandOptionStates.get(x); |
| State commandFileState = commandFileStates.get(x); |
| State st = State.merge(commandOptionState, commandFileState); |
| if (!isClearedState) { |
| State directiveState = directiveFileStates.get(x); |
| if (directiveState != null) { |
| st = directiveState; |
| } |
| } |
| State jcmdState = jcmdStates.get(x); |
| st = State.merge(st, jcmdState); |
| |
| finalStates.put(x, st); |
| } |
| |
| /* |
| * Create a list of commands from options and file |
| * to handle quiet command |
| */ |
| List<CompileCommand> ccList = new ArrayList<>(); |
| ccList.addAll(builders.get(Type.OPTION).getCompileCommands()); |
| ccList.addAll(builders.get(Type.FILE).getCompileCommands()); |
| |
| // Get all VM options after we build all states and files |
| List<String> options = new ArrayList<>(); |
| options.addAll(vmopts); |
| for (StateBuilder<?> builder : builders.values()) { |
| options.addAll(builder.getOptions()); |
| isValid &= builder.isValid(); |
| } |
| options.addAll(jcmdStateBuilder.getOptions()); |
| return new Scenario(isValid, options, finalStates, ccList, |
| jcmdCommands); |
| } |
| |
| // shows if jcmd have passed a clear command |
| private boolean jcmdClearedState(List<JcmdCommand> jcmdCommands) { |
| for (JcmdCommand jcmdCommand : jcmdCommands) { |
| if (jcmdCommand.jcmdType == JcmdType.CLEAR) { |
| return true; |
| } |
| } |
| return false; |
| } |
| } |
| } |