1 /*
2 * Copyright (c) 2015, 2018, 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 8132734 8144062 8194070
27 * @summary Test the System properties for JarFile that support multi-release jar files
28 * @library /lib/testlibrary/java/util/jar
29 * @build Compiler JarBuilder CreateMultiReleaseTestJars
30 * @run testng MultiReleaseJarProperties
31 * @run testng/othervm -Djdk.util.jar.version=0 MultiReleaseJarProperties
32 * @run testng/othervm -Djdk.util.jar.version=8 MultiReleaseJarProperties
33 * @run testng/othervm -Djdk.util.jar.version=9 MultiReleaseJarProperties
34 * @run testng/othervm -Djdk.util.jar.version=10 MultiReleaseJarProperties
35 * @run testng/othervm -Djdk.util.jar.version=100 MultiReleaseJarProperties
36 * @run testng/othervm -Djdk.util.jar.version=8 -Djdk.util.jar.enableMultiRelease=false MultiReleaseJarProperties
37 * @run testng/othervm -Djdk.util.jar.version=9 -Djdk.util.jar.enableMultiRelease=false MultiReleaseJarProperties
38 * @run testng/othervm -Djdk.util.jar.version=10 -Djdk.util.jar.enableMultiRelease=false MultiReleaseJarProperties
39 * @run testng/othervm -Djdk.util.jar.version=8 -Djdk.util.jar.enableMultiRelease=force MultiReleaseJarProperties
40 * @run testng/othervm -Djdk.util.jar.version=9 -Djdk.util.jar.enableMultiRelease=force MultiReleaseJarProperties
41 * @run testng/othervm -Djdk.util.jar.version=10 -Djdk.util.jar.enableMultiRelease=force MultiReleaseJarProperties
42 * @run testng/othervm -Djdk.util.jar.enableMultiRelease=false MultiReleaseJarProperties
43 * @run testng/othervm -Djdk.util.jar.enableMultiRelease=force MultiReleaseJarProperties
44 */
45
46 import java.io.File;
47 import java.io.IOException;
48 import java.io.InputStream;
49 import java.lang.invoke.MethodHandle;
50 import java.lang.invoke.MethodHandles;
51 import java.lang.invoke.MethodType;
52 import java.net.URL;
53 import java.net.URLClassLoader;
54 import java.nio.file.Files;
55 import java.util.jar.JarEntry;
56 import java.util.jar.JarFile;
57
58 import org.testng.Assert;
59 import org.testng.annotations.AfterClass;
60 import org.testng.annotations.BeforeClass;
61 import org.testng.annotations.Test;
62
63
64 public class MultiReleaseJarProperties {
65 final static int BASE_VERSION = JarFile.baseVersion().major();
66
67 final static String userdir = System.getProperty("user.dir", ".");
68 final static File multirelease = new File(userdir, "multi-release.jar");
69 protected int rtVersion;
70 boolean force;
71 protected ClassLoader cldr;
72 protected Class<?> rootClass;
73
74 @BeforeClass
75 public void initialize() throws Exception {
76 CreateMultiReleaseTestJars creator = new CreateMultiReleaseTestJars();
77 creator.compileEntries();
78 creator.buildMultiReleaseJar();
79 int RUNTIME_VERSION = Runtime.version().major();
80 rtVersion = Integer.getInteger("jdk.util.jar.version", RUNTIME_VERSION);
81 String mrprop = System.getProperty("jdk.util.jar.enableMultiRelease", "");
82 if (mrprop.equals("false")) {
83 rtVersion = BASE_VERSION;
84 } else if (rtVersion < BASE_VERSION) {
85 rtVersion = BASE_VERSION;
86 } else if (rtVersion > RUNTIME_VERSION) {
87 rtVersion = RUNTIME_VERSION;
88 }
89 rtVersion = Math.min(rtVersion, 10); // Tests should be augmented for later releases
90 force = mrprop.equals("force");
91
92 initializeClassLoader();
93 }
94
95 protected void initializeClassLoader() throws Exception {
96 URL[] urls = new URL[]{multirelease.toURI().toURL()};
97 cldr = new URLClassLoader(urls);
98 // load any class, Main is convenient and in the root entries
99 rootClass = cldr.loadClass("version.Main");
100 }
101
102 @AfterClass
103 public void close() throws IOException {
104 ((URLClassLoader)cldr).close();
105 Files.delete(multirelease.toPath());
106 }
107
108 /*
109 * jdk.util.jar.enableMultiRelease=force is a no-op for URLClassLoader
110 */
111 @Test
112 public void testURLClassLoader() throws Throwable {
113 Class<?> vcls = cldr.loadClass("version.Version");
114 invokeMethod(vcls, rtVersion);
115 }
116
117 protected void invokeMethod(Class<?> vcls, int expected) throws Throwable {
118 MethodType mt = MethodType.methodType(int.class);
119 MethodHandle mh = MethodHandles.lookup().findVirtual(vcls, "getVersion", mt);
120 Assert.assertEquals(expected, (int) mh.invoke(vcls.newInstance()));
121 }
122
123 /*
124 * jdk.util.jar.enableMultiRelease=force should affect a custom class loader
125 */
126 @Test
127 public void testClassLoader() throws Throwable {
128 try (JarFile jf = new JarFile(multirelease)) { // do not set runtime versioning
129 ClassLoader cldr = new CustomClassLoader(jf);
130 Class<?> vcls = cldr.loadClass("version.Version");
131 if (rtVersion == 9) {
132 try {
133 cldr.loadClass("version.PackagePrivate");
134 } catch (ClassNotFoundException x) {
135 if (force) throw x;
136 }
137 }
138 invokeMethod(vcls, force ? rtVersion : BASE_VERSION);
139 }
140 }
141
142 private static class CustomClassLoader extends ClassLoader {
143 private final JarFile jf;
144
145 CustomClassLoader(JarFile jf) throws Exception {
146 super(null);
147 this.jf = jf;
148 }
149
150 protected Class<?> findClass(String name) throws ClassNotFoundException {
151 try {
152 byte[] b;
153 String entryName = name.replace(".", "/") + ".class";
154 JarEntry je = jf.getJarEntry(entryName);
155 if (je != null) {
156 try (InputStream is = jf.getInputStream(je)) {
157 b = new byte[(int) je.getSize()];
158 is.read(b);
159 }
160 return defineClass(name, b, 0, b.length);
161 }
162 throw new ClassNotFoundException(name);
163 } catch (IOException x) {
164 throw new ClassNotFoundException(x.getMessage());
165 }
166 }
167 }
168
169 @Test
170 public void testGetResourceAsStream() throws Exception {
171 String resource = rtVersion == 9 ? "/version/PackagePrivate.java" : "/version/Version.java";
172 // use fileRootClass as a base for getting resources
173 getResourceAsStream(rootClass, resource);
174 }
175
176 protected void getResourceAsStream(Class<?> rootClass, String resource) throws Exception {
177 try (InputStream is = rootClass.getResourceAsStream(resource)) {
178 byte[] bytes = is.readAllBytes();
179 resource = new String(bytes);
180 }
181 String match = "return " + rtVersion + ";";
182 Assert.assertTrue(resource.contains(match));
183 }
184
185 @Test
186 public void testGetResource() throws Exception {
187 String resource = rtVersion == 9 ? "/version/PackagePrivate.java" : "/version/Version.java";
188 // use rootClass as a base for getting resources
189 getResource(rootClass, resource);
190 }
191
192 protected void getResource(Class<?> rootClass, String resource) throws Exception {
193 URL url = rootClass.getResource(resource);
194 try (InputStream is = url.openStream()) {
195 byte[] bytes = is.readAllBytes();
196 resource = new String(bytes);
197 }
198 String match = "return " + rtVersion + ";";
199 Assert.assertTrue(resource.contains(match));
200 }
201 }