1 /*
   2  * Copyright (c) 2015, 2016, 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.lang.module.Configuration;
  25 import java.lang.module.ModuleFinder;
  26 import java.lang.reflect.Layer;
  27 import java.lang.reflect.Proxy;
  28 import java.nio.file.Path;
  29 import java.nio.file.Paths;
  30 import java.util.Arrays;
  31 import java.util.List;
  32 
  33 import jdk.testlibrary.tasks.JavaTask;
  34 
  35 import org.testng.annotations.BeforeTest;
  36 import org.testng.annotations.Test;
  37 import static org.testng.Assert.*;
  38 
  39 /**
  40  * @test
  41  * @library /lib/testlibrary
  42  * @modules jdk.compiler
  43  * @build ProxyClassAccessTest q.NP CompilerUtils jdk.testlibrary.tasks.JavaTask
  44  * @run testng ProxyClassAccessTest
  45  * @summary Driver for testing proxy class doesn't have access to
  46  *          types referenced by proxy interfaces
  47  */
  48 
  49 public class ProxyClassAccessTest {
  50 
  51     private static final String TEST_SRC = System.getProperty("test.src");
  52     private static final String TEST_CLASSES = System.getProperty("test.classes");
  53 
  54     private static final Path SRC_DIR = Paths.get(TEST_SRC, "src");
  55     private static final Path MODS_DIR = Paths.get("mods");
  56 
  57     // the names of the modules in this test
  58     private static List<String> modules = Arrays.asList("m1", "m2", "m3", "test");
  59 
  60     /**
  61      * Compiles all modules used by the test
  62      */
  63     @BeforeTest
  64     public void compileAll() throws Exception {
  65         for (String mn : modules) {
  66             Path msrc = SRC_DIR.resolve(mn);
  67             assertTrue(CompilerUtils.compile(msrc, MODS_DIR, "--module-source-path", SRC_DIR.toString()));
  68         }
  69     }
  70 
  71     /**
  72      * Run the modular test
  73      */
  74     @Test
  75     public void runTest() {
  76         new JavaTask().ignoreStandardModuleOptions()
  77             .modulePath(MODS_DIR)
  78             .module("test", "jdk.test.ProxyClassAccess")
  79             .run();
  80     }
  81 
  82     /**
  83      * Test unnamed module has no access to other proxy interface
  84      */
  85     @Test
  86     public void testNoReadAccess() throws ClassNotFoundException {
  87         ModuleFinder finder = ModuleFinder.of(MODS_DIR);
  88         Layer bootLayer = Layer.boot();
  89         Configuration cf = bootLayer
  90                 .configuration()
  91                 .resolveRequiresAndUses(ModuleFinder.of(), finder, modules);
  92         ClassLoader parentLoader = this.getClass().getClassLoader();
  93         Layer layer = bootLayer.defineModulesWithOneLoader(cf, parentLoader);
  94 
  95         ClassLoader loader = layer.findLoader("m1");
  96         Class<?>[] interfaces = new Class<?>[] {
  97                 Class.forName("p.one.I", false, loader),
  98                 Class.forName("q.NP", false, loader)     // non-public interface in unnamed module
  99         };
 100         checkIAE(loader, interfaces);
 101     }
 102 
 103     private void checkIAE(ClassLoader loader, Class<?>[] interfaces) {
 104         try {
 105             Proxy.getProxyClass(loader, interfaces);
 106             throw new RuntimeException("Expected IllegalArgumentException thrown");
 107         } catch (IllegalArgumentException e) {}
 108 
 109         try {
 110             Proxy.newProxyInstance(loader, interfaces,
 111                 (proxy, m, params) -> { throw new RuntimeException(m.toString()); });
 112             throw new RuntimeException("Expected IllegalArgumentException thrown");
 113         } catch (IllegalArgumentException e) {}
 114     }
 115 
 116 }