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 }