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
24 /*
25 * @test
26 * @bug 8157068
27 * @summary Patch java.base and user module with Hashes attribute tied with
28 * other module.
29 * @library /lib/testlibrary
30 * @modules jdk.compiler
31 * @build CompilerUtils
32 * @run testng PatchSystemModules
33 */
34
35 import java.io.File;
36 import java.nio.file.Files;
37 import java.nio.file.Path;
38 import java.nio.file.Paths;
39 import java.util.ArrayList;
40 import java.util.List;
41 import java.util.stream.Stream;
42
43 import jdk.testlibrary.FileUtils;
44 import jdk.testlibrary.JDKToolFinder;
45 import org.testng.annotations.BeforeTest;
46 import org.testng.annotations.Test;
47
48 import static jdk.testlibrary.ProcessTools.executeCommand;
49 import static org.testng.Assert.*;
50
51 public class PatchSystemModules {
52 private static final String JAVA_HOME = System.getProperty("java.home");
53
54 private static final Path TEST_SRC = Paths.get(System.getProperty("test.src"));
55 private static final Path PATCH_SRC_DIR = TEST_SRC.resolve("src1");
56
57 private static final Path JMODS = Paths.get(JAVA_HOME, "jmods");
58 private static final Path MODS_DIR = Paths.get("mods");
59 private static final Path JARS_DIR = Paths.get("jars");
60 private static final Path PATCH_DIR = Paths.get("patches");
61 private static final Path IMAGE = Paths.get("image");
62
63 private static final String JAVA_BASE = "java.base";
64 private final String[] modules = new String[] { "m1", "m2" };
65
66 @BeforeTest
67 private void setup() throws Throwable {
68 Path src = TEST_SRC.resolve("src");
69 for (String name : modules) {
70 assertTrue(CompilerUtils.compile(src.resolve(name),
71 MODS_DIR,
72 "--module-source-path", src.toString()));
73 }
74
75 // compile patched source
76 String patchDir = PATCH_SRC_DIR.resolve(JAVA_BASE).toString();
77 assertTrue(CompilerUtils.compile(PATCH_SRC_DIR.resolve(JAVA_BASE),
78 PATCH_DIR.resolve(JAVA_BASE),
79 "--patch-module", "java.base=" + patchDir));
80 assertTrue(CompilerUtils.compile(PATCH_SRC_DIR.resolve("m2"),
81 PATCH_DIR.resolve("m2")));
82
83 // create an image with only m1 and m2
84 if (Files.exists(JMODS)) {
85 // create an image with m1,m2
86 createImage();
87 }
88 }
89
90 @Test
91 public void test() throws Throwable {
92 Path patchedJavaBase = PATCH_DIR.resolve(JAVA_BASE);
93 Path patchedM2 = PATCH_DIR.resolve("m2");
94
95 Path home = Paths.get(JAVA_HOME);
96 runTest(home,
97 "--module-path", MODS_DIR.toString(),
98 "-m", "m1/p1.Main", "1");
99 runTest(home,
100 "--patch-module", "java.base=" + patchedJavaBase,
101 "--module-path", MODS_DIR.toString(),
102 "-m", "m1/p1.Main", "1");
103
104 runTest(home,
105 "--patch-module", "m2=" + patchedM2.toString(),
106 "--module-path", MODS_DIR.toString(),
107 "-m", "m1/p1.Main", "2");
108 }
109
110 @Test
111 public void testImage() throws Throwable {
112 if (Files.notExists(JMODS))
113 return;
114
115 Path patchedJavaBase = PATCH_DIR.resolve(JAVA_BASE);
116 Path patchedM2 = PATCH_DIR.resolve("m2");
117
118 runTest(IMAGE,
119 "-m", "m1/p1.Main", "1");
120 runTest(IMAGE,
121 "--patch-module", "java.base=" + patchedJavaBase,
122 "-m", "m1/p1.Main", "1");
123 runTest(IMAGE,
124 "--patch-module", "m2=" + patchedM2.toString(),
125 "-m", "m1/p1.Main", "2");
126 }
127
128 @Test
129 public void upgradeTiedModule() throws Throwable {
130 if (Files.notExists(JMODS))
131 return;
132
133 Path m1 = MODS_DIR.resolve("m1.jar");
134
135 // create another m1.jar
136 jar("--create",
137 "--file=" + m1.toString(),
138 "-C", MODS_DIR.resolve("m1").toString(), ".");
139
140 // Fail to upgrade m1.jar with mismatched hash
141 runTestWithExitCode(getJava(IMAGE),
142 "--upgrade-module-path", m1.toString(),
143 "-m", "m1/p1.Main");
144
145 runTestWithExitCode(getJava(IMAGE),
146 "--patch-module", "java.base=" + PATCH_DIR.resolve(JAVA_BASE),
147 "--upgrade-module-path", m1.toString(),
148 "-m", "m1/p1.Main", "1");
149 }
150
151 private void runTestWithExitCode(String... options) throws Throwable {
152 assertTrue(executeCommand(options)
153 .outputTo(System.out)
154 .errorTo(System.out)
155 .shouldContain("differs to expected hash")
156 .getExitValue() != 0);
157 }
158
159 private void runTest(Path image, String... opts) throws Throwable {
160 String[] options =
161 Stream.concat(Stream.of(getJava(image)),
162 Stream.of(opts))
163 .toArray(String[]::new);
164
165 ProcessBuilder pb = new ProcessBuilder(options);
166 int exitValue = executeCommand(pb)
167 .outputTo(System.out)
168 .errorTo(System.out)
169 .getExitValue();
170
171 assertTrue(exitValue == 0);
172 }
173
174 static void createImage() throws Throwable {
175 FileUtils.deleteFileTreeUnchecked(JARS_DIR);
176 FileUtils.deleteFileTreeUnchecked(IMAGE);
177
178 Files.createDirectories(JARS_DIR);
179 Path m1 = JARS_DIR.resolve("m1.jar");
180 Path m2 = JARS_DIR.resolve("m2.jar");
181
182 // hash m1 in m2's Hashes attribute
183 jar("--create",
184 "--file=" + m1.toString(),
185 "-C", MODS_DIR.resolve("m1").toString(), ".");
186
187 jar("--create",
188 "--file=" + m2.toString(),
189 "--module-path", JARS_DIR.toString(),
190 "--hash-modules", "m1",
191 "-C", MODS_DIR.resolve("m2").toString(), ".");
192
193
194 String mpath = JARS_DIR.toString() + File.pathSeparator + JMODS.toString();
195 execTool("jlink", "--module-path", mpath,
196 "--add-modules", "m1",
197 "--output", IMAGE.toString());
198 }
199
200 static void jar(String... args) throws Throwable {
201 execTool("jar", args);
202 }
203
204 static void execTool(String tool, String... args) throws Throwable {
205 String path = JDKToolFinder.getJDKTool(tool);
206 List<String> commands = new ArrayList<>();
207 commands.add(path);
208 Stream.of(args).forEach(commands::add);
209 ProcessBuilder pb = new ProcessBuilder(commands);
210 int exitValue = executeCommand(pb)
211 .outputTo(System.out)
212 .errorTo(System.out)
213 .shouldNotContain("no module is recorded in hash")
214 .getExitValue();
215
216 assertTrue(exitValue == 0);
217 }
218
219 static String getJava(Path image) {
220 boolean isWindows = System.getProperty("os.name").startsWith("Windows");
221 Path java = image.resolve("bin").resolve(isWindows ? "java.exe" : "java");
222 if (Files.notExists(java))
223 throw new RuntimeException(java + " not found");
224 return java.toAbsolutePath().toString();
225 }
226 }