1 /*
2 * Copyright (c) 2015, 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 import java.io.File;
25 import java.io.FileOutputStream;
26 import java.io.IOException;
27 import java.nio.file.Files;
28 import java.nio.file.Path;
29 import java.nio.file.Paths;
30 import java.security.KeyStore;
31 import java.security.PrivateKey;
32 import java.security.cert.CertPath;
33 import java.security.cert.CertificateFactory;
34 import java.util.Arrays;
35 import java.util.HashMap;
36 import java.util.Map;
37 import java.util.zip.ZipFile;
38 import jdk.security.jarsigner.JarSigner;
39
40 public class CreateMultiReleaseTestJars {
41 final private String main =
42 "package version;\n\n"
43 + "public class Main {\n"
44 + " public static void main(String[] args) {\n"
45 + " Version v = new Version();\n"
46 + " System.out.println(\"I am running on version \" + v.getVersion());\n"
47 + " }\n"
48 + "}\n";
49 final private String java8 =
50 "package version;\n\n"
51 + "public class Version {\n"
52 + " public int getVersion() {\n"
53 + " return 8;\n"
54 + " }\n"
55 + "}\n";
56 final private String java9 =
57 "package version;\n\n"
58 + "public class Version {\n"
59 + " public int getVersion() {\n"
60 + " int version = (new PackagePrivate()).getVersion();\n"
61 + " if (version == 9) return 9;\n" // strange I know, but easy to test
62 + " return version;\n"
63 + " }\n"
64 + "}\n";
65 final private String ppjava9 =
66 "package version;\n\n"
67 + "class PackagePrivate {\n"
68 + " int getVersion() {\n"
69 + " return 9;\n"
70 + " }\n"
71 + "}\n";
72 final private String java10 = java8.replace("8", "10");
73 final String readme8 = "This is the root readme file";
74 final String readme9 = "This is the version nine readme file";
75 final String readme10 = "This is the version ten readme file";
76 private Map<String,byte[]> rootClasses;
77 private Map<String,byte[]> version9Classes;
78 private Map<String,byte[]> version10Classes;
79
80 public void buildUnversionedJar() throws IOException {
81 JarBuilder jb = new JarBuilder("unversioned.jar");
82 jb.addEntry("README", readme8.getBytes());
83 jb.addEntry("version/Main.java", main.getBytes());
84 jb.addEntry("version/Main.class", rootClasses.get("version.Main"));
85 jb.addEntry("version/Version.java", java8.getBytes());
86 jb.addEntry("version/Version.class", rootClasses.get("version.Version"));
87 jb.build();
88 }
89
90 public void buildMultiReleaseJar() throws IOException {
91 JarBuilder jb = customMultiReleaseJar("multi-release.jar", "true");
92 addEntries(jb);
93 jb.build();
94 }
95
96 private JarBuilder customMultiReleaseJar(String filename, String multiReleaseValue)
97 throws IOException {
98 JarBuilder jb = new JarBuilder(filename);
99 jb.addAttribute("Multi-Release", multiReleaseValue);
100 return jb;
101 }
102
103 public void buildCustomMultiReleaseJar(String filename, String multiReleaseValue,
104 Map<String, String> extraAttributes) throws IOException {
105 JarBuilder jb = new JarBuilder(filename);
106 extraAttributes.entrySet()
107 .forEach(entry -> jb.addAttribute(entry.getKey(), entry.getValue()));
108 jb.addAttribute("Multi-Release", multiReleaseValue);
109 jb.build();
110 }
111
112 private void addEntries(JarBuilder jb) {
113 jb.addEntry("README", readme8.getBytes());
114 jb.addEntry("version/Main.java", main.getBytes());
115 jb.addEntry("version/Main.class", rootClasses.get("version.Main"));
116 jb.addEntry("version/Version.java", java8.getBytes());
117 jb.addEntry("version/Version.class", rootClasses.get("version.Version"));
118 jb.addEntry("META-INF/versions/9/README", readme9.getBytes());
119 jb.addEntry("META-INF/versions/9/version/Version.java", java9.getBytes());
120 jb.addEntry("META-INF/versions/9/version/PackagePrivate.java", ppjava9.getBytes());
121 jb.addEntry("META-INF/versions/9/version/Version.class", version9Classes.get("version.Version"));
122 jb.addEntry("META-INF/versions/9/version/PackagePrivate.class", version9Classes.get("version.PackagePrivate"));
123 jb.addEntry("META-INF/versions/10/README", readme10.getBytes());
124 jb.addEntry("META-INF/versions/10/version/Version.java", java10.getBytes());
125 jb.addEntry("META-INF/versions/10/version/Version.class", version10Classes.get("version.Version"));
126 }
127
128 public void buildSignedMultiReleaseJar() throws Exception {
129 String testsrc = System.getProperty("test.src",".");
130 String testdir = findTestDir(testsrc);
131 String keystore = testdir + "/sun/security/tools/jarsigner/JarSigning.keystore";
132
133 // jarsigner -keystore keystore -storepass "bbbbbb"
134 // -signedJar signed-multi-release.jar multi-release.jar b
135
136 char[] password = "bbbbbb".toCharArray();
137 KeyStore ks = KeyStore.getInstance(new File(keystore), password);
138 PrivateKey pkb = (PrivateKey)ks.getKey("b", password);
139 CertPath cp = CertificateFactory.getInstance("X.509")
140 .generateCertPath(Arrays.asList(ks.getCertificateChain("b")));
141 JarSigner js = new JarSigner.Builder(pkb, cp).build();
142 try (ZipFile in = new ZipFile("multi-release.jar");
143 FileOutputStream os = new FileOutputStream("signed-multi-release.jar"))
144 {
145 js.sign(in, os);
146 }
147 }
148
149 String findTestDir(String dir) throws IOException {
150 Path path = Paths.get(dir).toAbsolutePath();
151 while (path != null && !path.endsWith("test")) {
152 path = path.getParent();
153 }
154 if (path == null) {
155 throw new IllegalArgumentException(dir + " is not in a test directory");
156 }
157 if (!Files.isDirectory(path)) {
158 throw new IOException(path.toString() + " is not a directory");
159 }
160 return path.toString();
161 }
162
163 void compileEntries() {
164 Map<String,String> input = new HashMap<>();
165 input.put("version.Main", main);
166 input.put("version.Version", java8);
167 rootClasses = (new Compiler(input)).setRelease(8).compile();
168 input.clear();
169 input.put("version.Version", java9);
170 input.put("version.PackagePrivate", ppjava9);
171 version9Classes = (new Compiler(input)).setRelease(9).compile();
172 input.clear();
173 input.put("version.Version", java10);
174 version10Classes = (new Compiler(input)).setRelease(9).compile(); // fixme in JDK 10
175 }
176 }