/* * Copyright (c) 2015, 2018, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ import java.io.File; import java.io.FileOutputStream; import java.io.IOException; import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; import java.security.KeyStore; import java.security.PrivateKey; import java.security.cert.CertPath; import java.security.cert.CertificateFactory; import java.util.Arrays; import java.util.HashMap; import java.util.Map; import java.util.zip.ZipFile; import jdk.security.jarsigner.JarSigner; public class CreateMultiReleaseTestJars { final private String main = "package version;\n\n" + "public class Main {\n" + " public static void main(String[] args) {\n" + " Version v = new Version();\n" + " System.out.println(\"I am running on version \" + v.getVersion());\n" + " }\n" + "}\n"; final private String java8 = "package version;\n\n" + "public class Version {\n" + " public int getVersion() {\n" + " return 8;\n" + " }\n" + "}\n"; final private String java9 = "package version;\n\n" + "public class Version {\n" + " public int getVersion() {\n" + " int version = (new PackagePrivate()).getVersion();\n" + " if (version == 9) return 9;\n" // strange I know, but easy to test + " return version;\n" + " }\n" + "}\n"; final private String ppjava9 = "package version;\n\n" + "class PackagePrivate {\n" + " int getVersion() {\n" + " return 9;\n" + " }\n" + "}\n"; final int currentVersion = Runtime.version().major(); final String currentVersionStr = Integer.toString(currentVersion); final private String javaCurrent = java8.replace("8", currentVersionStr); final String readme8 = "This is the root readme file"; final String readme9 = "This is the version nine readme file"; final String readmeCurrent = "This is the current version readme file"; private Map rootClasses; private Map version9Classes; private Map versionCurrentClasses; public void buildUnversionedJar() throws IOException { JarBuilder jb = new JarBuilder("unversioned.jar"); jb.addEntry("README", readme8.getBytes()); jb.addEntry("version/Main.java", main.getBytes()); jb.addEntry("version/Main.class", rootClasses.get("version.Main")); jb.addEntry("version/Version.java", java8.getBytes()); jb.addEntry("version/Version.class", rootClasses.get("version.Version")); jb.build(); } public void buildMultiReleaseJar() throws IOException { JarBuilder jb = customMultiReleaseJar("multi-release.jar", "true"); addEntries(jb); jb.addEntry("META-INF/versions/9/version/Version.class", version9Classes.get("version.Version")); jb.build(); } public void buildShortMultiReleaseJar() throws IOException { JarBuilder jb = customMultiReleaseJar("short-multi-release.jar", "true"); addEntries(jb); jb.build(); } private JarBuilder customMultiReleaseJar(String filename, String multiReleaseValue) throws IOException { JarBuilder jb = new JarBuilder(filename); jb.addAttribute("Multi-Release", multiReleaseValue); return jb; } public void buildCustomMultiReleaseJar(String filename, String multiReleaseValue, Map extraAttributes) throws IOException { buildCustomMultiReleaseJar(filename, multiReleaseValue, extraAttributes, false); } public void buildCustomMultiReleaseJar(String filename, String multiReleaseValue, Map extraAttributes, boolean addEntries) throws IOException { JarBuilder jb = new JarBuilder(filename); extraAttributes.entrySet() .forEach(entry -> jb.addAttribute(entry.getKey(), entry.getValue())); if (addEntries) { addEntries(jb); } jb.addAttribute("Multi-Release", multiReleaseValue); jb.build(); } private void addEntries(JarBuilder jb) { jb.addEntry("README", readme8.getBytes()); jb.addEntry("version/Main.java", main.getBytes()); jb.addEntry("version/Main.class", rootClasses.get("version.Main")); jb.addEntry("version/Version.java", java8.getBytes()); jb.addEntry("version/Version.class", rootClasses.get("version.Version")); jb.addEntry("META-INF/versions/9/README", readme9.getBytes()); jb.addEntry("META-INF/versions/9/version/Version.java", java9.getBytes()); jb.addEntry("META-INF/versions/9/version/PackagePrivate.java", ppjava9.getBytes()); jb.addEntry("META-INF/versions/9/version/PackagePrivate.class", version9Classes.get("version.PackagePrivate")); jb.addEntry("META-INF/versions/" + currentVersionStr + "/README", readmeCurrent.getBytes()); jb.addEntry("META-INF/versions/" + currentVersionStr + "/version/Version.java", javaCurrent.getBytes()); jb.addEntry("META-INF/versions/" + currentVersionStr + "/version/Version.class", versionCurrentClasses.get("version.Version")); } public void buildSignedMultiReleaseJar() throws Exception { String testsrc = System.getProperty("test.src","."); String testdir = findTestDir(testsrc); String keystore = testdir + "/sun/security/tools/jarsigner/JarSigning.keystore"; // jarsigner -keystore keystore -storepass "bbbbbb" // -signedJar signed-multi-release.jar multi-release.jar b char[] password = "bbbbbb".toCharArray(); KeyStore ks = KeyStore.getInstance(new File(keystore), password); PrivateKey pkb = (PrivateKey)ks.getKey("b", password); CertPath cp = CertificateFactory.getInstance("X.509") .generateCertPath(Arrays.asList(ks.getCertificateChain("b"))); JarSigner js = new JarSigner.Builder(pkb, cp).build(); try (ZipFile in = new ZipFile("multi-release.jar"); FileOutputStream os = new FileOutputStream("signed-multi-release.jar")) { js.sign(in, os); } } String findTestDir(String dir) throws IOException { Path path = Paths.get(dir).toAbsolutePath(); Path child = null; while (path != null && !path.endsWith("test")) { child = path; path = path.getParent(); } if (child == null) { throw new IllegalArgumentException(dir + " is not in a test directory"); } if (!Files.isDirectory(child)) { throw new IOException(child.toString() + " is not a directory"); } return child.toString(); } void compileEntries() { Map input = new HashMap<>(); input.put("version.Main", main); input.put("version.Version", java8); rootClasses = (new Compiler(input)).setRelease(8).compile(); input.clear(); input.put("version.Version", java9); input.put("version.PackagePrivate", ppjava9); version9Classes = (new Compiler(input)).setRelease(9).compile(); input.clear(); input.put("version.Version", javaCurrent); versionCurrentClasses = (new Compiler(input)).compile(); // Use default release } }