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 import java.io.IOException;
  25 import java.io.UncheckedIOException;
  26 import java.nio.file.Files;
  27 import java.nio.file.Path;
  28 import java.nio.file.Paths;
  29 import java.util.Arrays;
  30 import java.util.stream.Stream;
  31 
  32 import jdk.test.lib.util.FileUtils;
  33 import static jdk.testlibrary.ProcessTools.*;
  34 
  35 import org.testng.annotations.BeforeClass;
  36 import org.testng.annotations.Test;
  37 import static org.testng.Assert.assertTrue;
  38 
  39 /**
  40  * @test
  41  * @bug 8087335
  42  * @summary Tests for Class.forName(Module,String)
  43  * @library /lib/testlibrary /test/lib
  44  * @modules jdk.compiler
  45  * @build TestDriver CompilerUtils jdk.testlibrary.ProcessTools TestMain TestLayer
  46  * @run testng TestDriver
  47  */
  48 
  49 public class TestDriver {
  50 
  51     private static final String TEST_SRC =
  52             Paths.get(System.getProperty("test.src")).toString();
  53     private static final String TEST_CLASSES =
  54             Paths.get(System.getProperty("test.classes")).toString();
  55 
  56     private static final Path MOD_SRC_DIR = Paths.get(TEST_SRC, "src");
  57     private static final Path MOD_DEST_DIR = Paths.get("mods");
  58 
  59     private static final String[] modules = new String[] {"m1", "m2", "m3"};
  60 
  61     /**
  62      * Compiles all modules used by the test.
  63      */
  64     @BeforeClass
  65     public void setup() throws Exception {
  66         assertTrue(CompilerUtils.compile(
  67                         MOD_SRC_DIR, MOD_DEST_DIR,
  68                         "--module-source-path",
  69                         MOD_SRC_DIR.toString()));
  70 
  71         copyDirectories(MOD_DEST_DIR.resolve("m1"), Paths.get("mods1"));
  72         copyDirectories(MOD_DEST_DIR.resolve("m2"), Paths.get("mods2"));
  73     }
  74 
  75     @Test
  76     public void test() throws Exception {
  77         String[] options = new String[] {
  78                 "-cp", TEST_CLASSES,
  79                 "--module-path", MOD_DEST_DIR.toString(),
  80                 "--add-modules", String.join(",", modules),
  81                 "-m", "m2/p2.test.Main"
  82         };
  83         runTest(options);
  84     }
  85 
  86     @Test
  87     public void testUnnamedModule() throws Exception {
  88         String[] options = new String[] {
  89                 "-cp", TEST_CLASSES,
  90                 "--module-path", MOD_DEST_DIR.toString(),
  91                 "--add-modules", String.join(",", modules),
  92                 "TestMain"
  93         };
  94         runTest(options);
  95     }
  96 
  97     @Test
  98     public void testLayer() throws Exception {
  99         String[] options = new String[] {
 100                 "-cp", TEST_CLASSES,
 101                 "TestLayer"
 102         };
 103 
 104         runTest(options);
 105     }
 106 
 107     @Test
 108     public void testDeniedClassLoaderAccess() throws Exception {
 109         String[] options = new String[] {
 110                 "--module-path", MOD_DEST_DIR.toString(),
 111                 "--add-modules", String.join(",", modules),
 112                 "-m", "m3/p3.NoGetClassLoaderAccess"
 113         };
 114         assertTrue(executeTestJava(options)
 115                         .outputTo(System.out)
 116                         .errorTo(System.err)
 117                         .getExitValue() == 0);
 118     }
 119 
 120     @Test
 121     public void testDeniedAccess() throws Exception {
 122         Path policyFile = Paths.get(TEST_SRC, "policy.denied");
 123 
 124         String[] options = new String[] {
 125                 "-Djava.security.manager",
 126                 "-Djava.security.policy=" + policyFile.toString(),
 127                 "--module-path", MOD_DEST_DIR.toString(),
 128                 "--add-modules", String.join(",", modules),
 129                 "-m", "m3/p3.NoAccess"
 130         };
 131         assertTrue(executeTestJava(options)
 132                         .outputTo(System.out)
 133                         .errorTo(System.err)
 134                         .getExitValue() == 0);
 135     }
 136 
 137     private String[] runWithSecurityManager(String[] options) {
 138         Path policyFile = Paths.get(TEST_SRC, "policy");
 139         Stream<String> opts = Stream.concat(Stream.of("-Djava.security.manager",
 140                                                       "-Djava.security.policy=" + policyFile.toString()),
 141                                             Arrays.stream(options));
 142         return opts.toArray(String[]::new);
 143     }
 144 
 145     private void runTest(String[] options) throws Exception {
 146         assertTrue(executeTestJava(options)
 147                         .outputTo(System.out)
 148                         .errorTo(System.err)
 149                         .getExitValue() == 0);
 150 
 151         assertTrue(executeTestJava(runWithSecurityManager(options))
 152                         .outputTo(System.out)
 153                         .errorTo(System.err)
 154                         .getExitValue() == 0);
 155     }
 156 
 157     private void copyDirectories(Path source, Path dest) throws IOException {
 158         if (Files.exists(dest))
 159             FileUtils.deleteFileTreeWithRetry(dest);
 160         Files.walk(source, Integer.MAX_VALUE)
 161                 .filter(Files::isRegularFile)
 162                 .forEach(p -> {
 163                     try {
 164                         Path to = dest.resolve(source.relativize(p));
 165                         Files.createDirectories(to.getParent());
 166                         Files.copy(p, to);
 167                     } catch (IOException e) {
 168                         throw new UncheckedIOException(e);
 169                     }
 170                 });
 171     }
 172 }