1 /**
   2  * Copyright (c) 2015, 2017, 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  * @summary Basic test of jlink to create jmods and images
  27  * @author Andrei Eremeev
  28  * @library /lib/testlibrary /test/lib
  29  * @modules java.base/jdk.internal.module
  30  *          jdk.jlink
  31  *          jdk.compiler
  32  * @build jdk.testlibrary.ProcessTools
  33  *        jdk.testlibrary.OutputAnalyzer
  34  *        JarUtils jdk.test.lib.compiler.CompilerUtils
  35  * @run main BasicTest
  36  */
  37 
  38 import java.io.File;
  39 import java.io.PrintWriter;
  40 import java.nio.file.Files;
  41 import java.nio.file.Path;
  42 import java.nio.file.Paths;
  43 import java.util.ArrayList;
  44 import java.util.Collections;
  45 import java.util.List;
  46 import java.util.spi.ToolProvider;
  47 
  48 import jdk.test.lib.compiler.CompilerUtils;
  49 import jdk.testlibrary.OutputAnalyzer;
  50 import jdk.testlibrary.ProcessTools;
  51 
  52 public class BasicTest {
  53     static final ToolProvider JMOD_TOOL = ToolProvider.findFirst("jmod")
  54         .orElseThrow(() ->
  55             new RuntimeException("jmod tool not found")
  56         );
  57 
  58     static final ToolProvider JLINK_TOOL = ToolProvider.findFirst("jlink")
  59         .orElseThrow(() ->
  60             new RuntimeException("jlink tool not found")
  61         );
  62 
  63     private final String TEST_MODULE = "test";
  64     private final Path jdkHome = Paths.get(System.getProperty("test.jdk"));
  65     private final Path jdkMods = jdkHome.resolve("jmods");
  66     private final Path testSrc = Paths.get(System.getProperty("test.src"));
  67     private final Path src = testSrc.resolve("src").resolve(TEST_MODULE);
  68     private final Path classes = Paths.get("classes");
  69     private final Path jmods = Paths.get("jmods");
  70     private final Path jars = Paths.get("jars");
  71 
  72     public static void main(String[] args) throws Throwable {
  73         new BasicTest().run();
  74     }
  75 
  76     public void run() throws Throwable {
  77         if (Files.notExists(jdkMods)) {
  78             return;
  79         }
  80 
  81         if (!CompilerUtils.compile(src, classes)) {
  82             throw new AssertionError("Compilation failure. See log.");
  83         }
  84 
  85         Files.createDirectories(jmods);
  86         Files.createDirectories(jars);
  87         Path jarfile = jars.resolve("test.jar");
  88         JarUtils.createJarFile(jarfile, classes);
  89 
  90         Path image = Paths.get("mysmallimage");
  91         runJmod(jarfile.toString(), TEST_MODULE, true);
  92         runJlink(image, TEST_MODULE, "--compress", "2", "--launcher", "foo=" + TEST_MODULE);
  93         execute(image, "foo");
  94 
  95         Files.delete(jmods.resolve(TEST_MODULE + ".jmod"));
  96 
  97         image = Paths.get("myimage");
  98         runJmod(classes.toString(), TEST_MODULE, true);
  99         runJlink(image, TEST_MODULE, "--launcher", "bar=" + TEST_MODULE);
 100         execute(image, "bar");
 101 
 102         Files.delete(jmods.resolve(TEST_MODULE + ".jmod"));
 103 
 104         image = Paths.get("myimage2");
 105         runJmod(classes.toString(), TEST_MODULE, false /* no ModuleMainClass! */);
 106         // specify main class in --launcher command line
 107         runJlink(image, TEST_MODULE, "--launcher", "bar2=" + TEST_MODULE + "/jdk.test.Test");
 108         execute(image, "bar2");
 109 
 110     }
 111 
 112     private void execute(Path image, String scriptName) throws Throwable {
 113         String cmd = image.resolve("bin").resolve(scriptName).toString();
 114         OutputAnalyzer analyzer;
 115         if (System.getProperty("os.name").startsWith("Windows")) {
 116             analyzer = ProcessTools.executeProcess("sh.exe", cmd, "1", "2", "3");
 117         } else {
 118             analyzer = ProcessTools.executeProcess(cmd, "1", "2", "3");
 119         }
 120         if (analyzer.getExitValue() != 0) {
 121             throw new AssertionError("Image invocation failed: rc=" + analyzer.getExitValue());
 122         }
 123     }
 124 
 125     private void runJlink(Path image, String modName, String... options) {
 126         List<String> args = new ArrayList<>();
 127         Collections.addAll(args,
 128                 "--module-path", jdkMods + File.pathSeparator + jmods,
 129                 "--add-modules", modName,
 130                 "--output", image.toString());
 131         Collections.addAll(args, options);
 132 
 133         PrintWriter pw = new PrintWriter(System.out);
 134         int rc = JLINK_TOOL.run(pw, pw, args.toArray(new String[args.size()]));
 135         if (rc != 0) {
 136             throw new AssertionError("Jlink failed: rc = " + rc);
 137         }
 138     }
 139 
 140     private void runJmod(String cp, String modName, boolean main) {
 141         int rc;
 142         if (main) {
 143             rc = JMOD_TOOL.run(System.out, System.out, new String[] {
 144                 "create",
 145                 "--class-path", cp,
 146                 "--module-version", "1.0",
 147                 "--main-class", "jdk.test.Test",
 148                 jmods.resolve(modName + ".jmod").toString()
 149             });
 150         } else {
 151             rc = JMOD_TOOL.run(System.out, System.out, new String[] {
 152                 "create",
 153                 "--class-path", cp,
 154                 "--module-version", "1.0",
 155                 jmods.resolve(modName + ".jmod").toString(),
 156             });
 157         }
 158 
 159         if (rc != 0) {
 160             throw new AssertionError("Jmod failed: rc = " + rc);
 161         }
 162     }
 163 }