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.IOException;
25 import java.io.UncheckedIOException;
26 import java.nio.file.Files;
27 import java.nio.file.Path;
28 import java.util.Collections;
29 import java.util.List;
30 import java.util.Map;
31 import java.util.function.Function;
32
33 import jdk.tools.jlink.plugin.Plugin;
34 import jdk.tools.jlink.plugin.ResourcePool;
35 import jdk.tools.jlink.plugin.ResourcePoolBuilder;
36 import jdk.tools.jlink.internal.PluginRepository;
37 import jdk.tools.jlink.internal.PostProcessor;
38 import jdk.tools.jlink.internal.ExecutableImage;
39 import tests.Helper;
40
41 /*
42 * @test
43 * @summary Test post processing
44 * @author Jean-Francois Denise
45 * @library ../lib
46 * @modules java.base/jdk.internal.jimage
47 * jdk.jdeps/com.sun.tools.classfile
48 * jdk.jlink/jdk.tools.jlink.internal
49 * jdk.jlink/jdk.tools.jlink.plugin
50 * jdk.jlink/jdk.tools.jmod
51 * jdk.jlink/jdk.tools.jimage
52 * jdk.compiler
53 * @build tests.*
54 * @run main/othervm JLinkPostProcessingTest
55 */
56 public class JLinkPostProcessingTest {
57
58 private static class PPPlugin implements PostProcessor, Plugin {
59
60 private static ExecutableImage called;
61 private static final String NAME = "pp";
62
63 @Override
64 public List<String> process(ExecutableImage image) {
65 called = image;
66 Path gen = image.getHome().resolve("lib").resolve("toto.txt");
67 try {
68 Files.createFile(gen);
69 } catch (IOException ex) {
70 throw new UncheckedIOException(ex);
71 }
72 return null;
73 }
74
75 @Override
76 public ResourcePool transform(ResourcePool in, ResourcePoolBuilder out) {
77 in.transformAndCopy(Function.identity(), out);
78 return out.build();
79 }
80
81 @Override
82 public String getName() {
83 return NAME;
84 }
85
86 @Override
87 public Category getType() {
88 return Category.PROCESSOR;
89 }
90
91 @Override
92 public String getDescription() {
93 return NAME;
94 }
95 }
96
97 public static void main(String[] args) throws Exception {
98
99 Helper helper = Helper.newHelper();
100 if (helper == null) {
101 System.err.println("Test not run");
102 return;
103 }
104 helper.generateDefaultModules();
105
106 PluginRepository.registerPlugin(new PPPlugin());
107
108 // Generate an image and post-process in same jlink execution.
109 {
110 String[] userOptions = {"--pp"};
111 String moduleName = "postprocessing1";
112 helper.generateDefaultJModule(moduleName, "composite2");
113 String[] res = {};
114 String[] files = {};
115 Path imageDir = helper.generateDefaultImage(userOptions, moduleName).assertSuccess();
116 helper.checkImage(imageDir, moduleName, res, files);
117
118 test(imageDir);
119 }
120
121 // Generate an image, post-process in 2 jlink executions.
122 {
123 String[] userOptions = {};
124 String moduleName = "postprocessing2";
125 helper.generateDefaultJModule(moduleName, "composite2");
126 String[] res = {};
127 String[] files = {};
128 Path imageDir = helper.generateDefaultImage(userOptions, moduleName).assertSuccess();
129 helper.checkImage(imageDir, moduleName, res, files);
130
131 String[] ppOptions = {"--pp"};
132 helper.postProcessImage(imageDir, ppOptions);
133 test(imageDir);
134 }
135 }
136
137 private static void test(Path imageDir)
138 throws Exception {
139 if (PPPlugin.called == null) {
140 throw new Exception("Post processor not called.");
141 }
142 if (!PPPlugin.called.getHome().equals(imageDir)) {
143 throw new Exception("Not right imageDir " + PPPlugin.called.getHome());
144 }
145 if (PPPlugin.called.getExecutionArgs().isEmpty()) {
146 throw new Exception("No arguments to run java...");
147 }
148 Path gen = imageDir.resolve("lib").resolve("toto.txt");
149 if (!Files.exists(gen)) {
150 throw new Exception("Generated file doesn;t exist");
151 }
152 PPPlugin.called = null;
153 }
154 }