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  * @bug 8132734 8144062 8159785
  27  * @summary Test that URL connections to multi-release jars can be runtime versioned
  28  * @library /lib/testlibrary/java/util/jar
  29  * @modules java.compiler
  30  *          jdk.httpserver
  31  *          jdk.jartool
  32  * @build Compiler JarBuilder CreateMultiReleaseTestJars SimpleHttpServer
  33  * @run testng MultiReleaseJarURLConnection
  34  */
  35 
  36 import java.io.IOException;
  37 import java.io.InputStream;
  38 import java.lang.invoke.MethodHandle;
  39 import java.lang.invoke.MethodHandles;
  40 import java.lang.invoke.MethodType;
  41 import java.net.JarURLConnection;
  42 import java.net.URL;
  43 import java.net.URLClassLoader;
  44 import java.net.URLConnection;
  45 import java.nio.file.Files;
  46 import java.nio.file.Paths;
  47 import java.util.Enumeration;
  48 import java.util.jar.JarFile;
  49 
  50 import org.testng.Assert;
  51 import org.testng.annotations.AfterClass;
  52 import org.testng.annotations.BeforeClass;
  53 import org.testng.annotations.DataProvider;
  54 import org.testng.annotations.Test;
  55 
  56 public class MultiReleaseJarURLConnection {
  57     String userdir = System.getProperty("user.dir",".");
  58     String unversioned = userdir + "/unversioned.jar";
  59     String unsigned = userdir + "/multi-release.jar";
  60     String signed = userdir + "/signed-multi-release.jar";
  61     SimpleHttpServer server;
  62 
  63     @BeforeClass
  64     public void initialize() throws Exception {
  65         CreateMultiReleaseTestJars creator = new CreateMultiReleaseTestJars();
  66         creator.compileEntries();
  67         creator.buildUnversionedJar();
  68         creator.buildMultiReleaseJar();
  69         creator.buildSignedMultiReleaseJar();
  70 
  71         server = new SimpleHttpServer();
  72         server.start();
  73 
  74     }
  75 
  76     @AfterClass
  77     public void close() throws IOException {
  78         // Windows requires server to stop before file is deleted
  79         if (server != null)
  80             server.stop();
  81         Files.delete(Paths.get(unversioned));
  82         Files.delete(Paths.get(unsigned));
  83         Files.delete(Paths.get(signed));
  84     }
  85 
  86     @DataProvider(name = "data")
  87     public Object[][] createData() {
  88         return new Object[][]{
  89                 {"unversioned", unversioned},
  90                 {"unsigned", unsigned},
  91                 {"signed", signed}
  92         };
  93     }
  94 
  95     @Test(dataProvider = "data")
  96     public void testRuntimeVersioning(String style, String file) throws Exception {
  97         String urlFile = "jar:file:" + file + "!/";
  98         String baseUrlEntry = urlFile + "version/Version.java";
  99         String rtreturn = "return " + Runtime.version().major();
 100 
 101         Assert.assertTrue(readAndCompare(new URL(baseUrlEntry), "return 8"));
 102         // #runtime is "magic" for a multi-release jar, but not for unversioned jar
 103         Assert.assertTrue(readAndCompare(new URL(baseUrlEntry + "#runtime"),
 104                 style.equals("unversioned") ? "return 8" : rtreturn));
 105         // #fragment or any other fragment is not magic
 106         Assert.assertTrue(readAndCompare(new URL(baseUrlEntry + "#fragment"), "return 8"));
 107         // cached entities not affected
 108         Assert.assertTrue(readAndCompare(new URL(baseUrlEntry), "return 8"));
 109 
 110         // the following tests will not work with unversioned jars
 111         if (style.equals("unversioned")) return;
 112 
 113         // direct access to versioned entry
 114         String versUrlEntry = urlFile + "META-INF/versions/" + Runtime.version().major()
 115                 + "/version/Version.java";
 116         Assert.assertTrue(readAndCompare(new URL(versUrlEntry), rtreturn));
 117         // adding any fragment does not change things
 118         Assert.assertTrue(readAndCompare(new URL(versUrlEntry + "#runtime"), rtreturn));
 119         Assert.assertTrue(readAndCompare(new URL(versUrlEntry + "#fragment"), rtreturn));
 120 
 121         // it really doesn't change things
 122         versUrlEntry = urlFile + "META-INF/versions/10/version/Version.java";
 123         Assert.assertTrue(readAndCompare(new URL(versUrlEntry), "return 10"));
 124         Assert.assertTrue(readAndCompare(new URL(versUrlEntry + "#runtime"), "return 10"));
 125         Assert.assertTrue(readAndCompare(new URL(versUrlEntry + "#fragment"), "return 10"));
 126     }
 127 
 128     @Test(dataProvider = "data")
 129     public void testCachedJars(String style, String file) throws Exception {
 130         String urlFile = "jar:file:" + file + "!/";
 131 
 132         URL rootUrl = new URL(urlFile);
 133         JarURLConnection juc = (JarURLConnection)rootUrl.openConnection();
 134         JarFile rootJar = juc.getJarFile();
 135         Runtime.Version root = rootJar.getVersion();
 136 
 137         URL runtimeUrl = new URL(urlFile + "#runtime");
 138         juc = (JarURLConnection)runtimeUrl.openConnection();
 139         JarFile runtimeJar = juc.getJarFile();
 140         Runtime.Version runtime = runtimeJar.getVersion();
 141         if (style.equals("unversioned")) {
 142             Assert.assertEquals(root, runtime);
 143         } else {
 144             Assert.assertNotEquals(root, runtime);
 145         }
 146 
 147         juc = (JarURLConnection)rootUrl.openConnection();
 148         JarFile jar = juc.getJarFile();
 149         Assert.assertEquals(jar.getVersion(), root);
 150         Assert.assertEquals(jar, rootJar);
 151 
 152         juc = (JarURLConnection)runtimeUrl.openConnection();
 153         jar = juc.getJarFile();
 154         Assert.assertEquals(jar.getVersion(), runtime);
 155         Assert.assertEquals(jar, runtimeJar);
 156 
 157         rootJar.close();
 158         runtimeJar.close();
 159         jar.close(); // probably not needed
 160     }
 161 
 162     @DataProvider(name = "resourcedata")
 163     public Object[][] createResourceData() throws Exception {
 164         return new Object[][]{
 165                 {"unversioned", Paths.get(unversioned).toUri().toURL()},
 166                 {"unsigned", Paths.get(unsigned).toUri().toURL()},
 167                 {"signed", Paths.get(signed).toUri().toURL()},
 168                 {"unversioned", new URL("file:" + unversioned)},
 169                 {"unsigned", new URL("file:" + unsigned)},
 170                 {"signed", new URL("file:" + signed)},
 171                 {"unversioned", new URL("jar:file:" + unversioned + "!/")},
 172                 {"unsigned", new URL("jar:file:" + unsigned + "!/")},
 173                 {"signed", new URL("jar:file:" + signed + "!/")},
 174                 // external jar received via http protocol
 175                 {"http", new URL("jar:http://localhost:" + server.getPort() + "/multi-release.jar!/")},
 176                 {"http", new URL("http://localhost:" + server.getPort() + "/multi-release.jar")},
 177 
 178         };
 179     }
 180 
 181     @Test(dataProvider = "resourcedata")
 182     public void testResources(String style, URL url) throws Throwable {
 183         //System.out.println("  testing " + style + " url: " + url);
 184         URL[] urls = {url};
 185         URLClassLoader cldr = new URLClassLoader(urls);
 186         Class<?> vcls = cldr.loadClass("version.Version");
 187 
 188         // verify we are loading a runtime versioned class
 189         MethodType mt = MethodType.methodType(int.class);
 190         MethodHandle mh = MethodHandles.lookup().findVirtual(vcls, "getVersion", mt);
 191         Assert.assertEquals((int)mh.invoke(vcls.newInstance()),
 192                 style.equals("unversioned") ? 8 : Runtime.version().major());
 193 
 194         // now get a resource and verify that we don't have a fragment attached
 195         Enumeration<URL> vclsUrlEnum = cldr.getResources("version/Version.class");
 196         Assert.assertTrue(vclsUrlEnum.hasMoreElements());
 197         URL vclsUrls[] = new URL[] {
 198             vcls.getResource("/version/Version.class"),
 199             vcls.getResource("Version.class"),
 200             cldr.getResource("version/Version.class"),
 201             vclsUrlEnum.nextElement()
 202         };
 203         Assert.assertFalse(vclsUrlEnum.hasMoreElements());
 204         for (URL vclsUrl : vclsUrls) {
 205             String fragment = vclsUrl.getRef();
 206             Assert.assertNull(fragment);
 207 
 208             // and verify that the the url is a reified pointer to the runtime entry
 209             String rep = vclsUrl.toString();
 210             //System.out.println("    getResource(\"/version/Version.class\") returned: " + rep);
 211             if (style.equals("http")) {
 212                 Assert.assertTrue(rep.startsWith("jar:http:"));
 213             } else {
 214                 Assert.assertTrue(rep.startsWith("jar:file:"));
 215             }
 216             String suffix;
 217             if (style.equals("unversioned")) {
 218                 suffix = ".jar!/version/Version.class";
 219             } else {
 220                 suffix = ".jar!/META-INF/versions/" + Runtime.version().major()
 221                         + "/version/Version.class";
 222             }
 223             Assert.assertTrue(rep.endsWith(suffix));
 224         }
 225         cldr.close();
 226     }
 227 
 228 
 229     private boolean readAndCompare(URL url, String match) throws Exception {
 230         boolean result;
 231         // necessary to do it this way, instead of openStream(), so we can
 232         // close underlying JarFile, otherwise windows can't delete the file
 233         URLConnection conn = url.openConnection();
 234         try (InputStream is = conn.getInputStream()) {
 235             byte[] bytes = is.readAllBytes();
 236             result = (new String(bytes)).contains(match);
 237         }
 238         if (conn instanceof JarURLConnection) {
 239             ((JarURLConnection)conn).getJarFile().close();
 240         }
 241         return result;
 242     }
 243 }