1/*
2 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24package compiler.aot.cli.jaotc;
25
26import java.io.File;
27import java.io.IOException;
28import jdk.test.lib.process.ExitCode;
29import jdk.test.lib.JDKToolLauncher;
30import jdk.test.lib.Utils;
31import jdk.test.lib.cli.CommandLineOptionTest;
32import jdk.test.lib.process.OutputAnalyzer;
33
34public class JaotcTestHelper {
35    public static final String DEFAULT_LIB_PATH = "./unnamed.so";
36    public static final String DEFAULT_LIBRARY_LOAD_MESSAGE = "loaded    " + DEFAULT_LIB_PATH
37            + "  aot library";
38    private static final String ENABLE_AOT = "-XX:+UseAOT";
39    private static final String AOT_LIBRARY = "-XX:AOTLibrary=" + DEFAULT_LIB_PATH;
40    private static final String PRINT_AOT = "-XX:+PrintAOT";
41
42    public static OutputAnalyzer compileLibrary(String... args) {
43        JDKToolLauncher launcher = JDKToolLauncher.createUsingTestJDK("jaotc");
44        for (String vmOpt : Utils.getTestJavaOpts()) {
45            launcher.addVMArg(vmOpt);
46        }
47        launcher.addToolArg("--compile-with-assertions");
48        for (String arg : args) {
49            launcher.addToolArg(arg);
50        }
51        String[] cmd = launcher.getCommand();
52        try {
53            return new OutputAnalyzer(new ProcessBuilder(cmd).start());
54        } catch (IOException e) {
55            throw new Error("Can't start test process: " + e, e);
56        }
57    }
58
59    public static void checkLibraryUsage(String classToRun) {
60        checkLibraryUsage(classToRun, new String[]{DEFAULT_LIBRARY_LOAD_MESSAGE}, null);
61    }
62
63    public static void checkLibraryUsage(String classToRun, String[] expectedOutput,
64            String[] unexpectedOutput) {
65        try {
66            CommandLineOptionTest.verifyJVMStartup(expectedOutput, unexpectedOutput,
67                    "Unexpected exit code", "Unexpected output", ExitCode.OK,
68                    /* addTestVMOpts */ true, ENABLE_AOT, AOT_LIBRARY, PRINT_AOT, classToRun);
69        } catch (Throwable t) {
70            throw new Error("Library usage verification failed: " + t, t);
71        }
72    }
73
74    public static String getClassAotCompilationFilename(Class<?> classToCompile) {
75        return classToCompile.getName().replaceAll("\\.", File.separator) + ".class";
76    }
77
78    public static String getClassAotCompilationName(Class<?> classToCompile) {
79        return classToCompile.getName();
80    }
81}
82