1 /**
2 * Copyright (c) 2015, 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
24 /*
25 * @test
26 * @summary Basic test of jlink to create jmods and images
27 * @author Andrei Eremeev
28 * @library /lib/testlibrary
29 * @modules java.base/jdk.internal.module
30 * jdk.jlink
31 * jdk.compiler
32 * @build jdk.testlibrary.ProcessTools
33 * jdk.testlibrary.OutputAnalyzer
34 * JarUtils CompilerUtils
35 * @run main BasicTest
36 */
37
38 import java.io.File;
39 import java.io.PrintWriter;
40 import java.nio.file.Files;
41 import java.nio.file.Path;
42 import java.nio.file.Paths;
43 import java.util.ArrayList;
44 import java.util.Collections;
45 import java.util.List;
46 import java.util.spi.ToolProvider;
47
48 import jdk.testlibrary.OutputAnalyzer;
49 import jdk.testlibrary.ProcessTools;
50
51 public class BasicTest {
52 static final ToolProvider JMOD_TOOL = ToolProvider.findFirst("jmod")
53 .orElseThrow(() ->
54 new RuntimeException("jmod tool not found")
55 );
56
57 static final ToolProvider JLINK_TOOL = ToolProvider.findFirst("jlink")
58 .orElseThrow(() ->
59 new RuntimeException("jlink tool not found")
60 );
61
62 private final Path jdkHome = Paths.get(System.getProperty("test.jdk"));
63 private final Path jdkMods = jdkHome.resolve("jmods");
64 private final Path testSrc = Paths.get(System.getProperty("test.src"));
65 private final Path src = testSrc.resolve("src");
66 private final Path classes = Paths.get("classes");
67 private final Path jmods = Paths.get("jmods");
68 private final Path jars = Paths.get("jars");
69
70 public static void main(String[] args) throws Throwable {
71 new BasicTest().run();
72 }
73
74 public void run() throws Throwable {
75 if (Files.notExists(jdkMods)) {
76 return;
77 }
78
79 if (!CompilerUtils.compile(src, classes)) {
80 throw new AssertionError("Compilation failure. See log.");
81 }
82
83 String modName = "test";
84 Files.createDirectories(jmods);
85 Files.createDirectories(jars);
86 Path jarfile = jars.resolve("test.jar");
87 JarUtils.createJarFile(jarfile, classes);
88
89 Path image = Paths.get("mysmallimage");
90 runJmod(jarfile.toString(), modName);
91 runJlink(image, modName, "--compress", "2");
92 execute(image, modName);
93
94 Files.delete(jmods.resolve(modName + ".jmod"));
95
96 image = Paths.get("myimage");
97 runJmod(classes.toString(), modName);
98 runJlink(image, modName);
99 execute(image, modName);
100 }
101
102 private void execute(Path image, String moduleName) throws Throwable {
103 String cmd = image.resolve("bin").resolve(moduleName).toString();
104 OutputAnalyzer analyzer;
105 if (System.getProperty("os.name").startsWith("Windows")) {
106 analyzer = ProcessTools.executeProcess("sh.exe", cmd, "1", "2", "3");
107 } else {
108 analyzer = ProcessTools.executeProcess(cmd, "1", "2", "3");
109 }
110 if (analyzer.getExitValue() != 0) {
111 throw new AssertionError("Image invocation failed: rc=" + analyzer.getExitValue());
112 }
113 }
114
115 private void runJlink(Path image, String modName, String... options) {
116 List<String> args = new ArrayList<>();
117 Collections.addAll(args,
118 "--module-path", jdkMods + File.pathSeparator + jmods,
119 "--add-modules", modName,
120 "--output", image.toString());
121 Collections.addAll(args, options);
122
123 PrintWriter pw = new PrintWriter(System.out);
124 int rc = JLINK_TOOL.run(pw, pw, args.toArray(new String[args.size()]));
125 if (rc != 0) {
126 throw new AssertionError("Jlink failed: rc = " + rc);
127 }
128 }
129
130 private void runJmod(String cp, String modName) {
131 int rc = JMOD_TOOL.run(System.out, System.out, new String[] {
132 "create",
133 "--class-path", cp,
134 "--module-version", "1.0",
135 "--main-class", "jdk.test.Test",
136 jmods.resolve(modName + ".jmod").toString(),
137 });
138 if (rc != 0) {
139 throw new AssertionError("Jmod failed: rc = " + rc);
140 }
141 }
142 }