| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package javax.xml.crypto.test.dsig; |
| |
| import java.lang.reflect.Method; |
| import java.io.File; |
| import java.net.URL; |
| import java.net.URLClassLoader; |
| import java.security.AccessController; |
| import java.security.Provider; |
| import java.security.Security; |
| |
| import javax.xml.crypto.dsig.CanonicalizationMethod; |
| import org.apache.jcp.xml.dsig.internal.dom.XMLDSigRI; |
| |
| /** |
| * This test uses more than one classloader to load a class (Driver) that |
| * invokes the XMLSignature API. It tests that there are not provider class |
| * loading issues with more than one classloader (see 6380953). |
| */ |
| public class ClassLoaderTest { |
| |
| private static final org.slf4j.Logger LOG = |
| org.slf4j.LoggerFactory.getLogger(ClassLoaderTest.class); |
| |
| @SuppressWarnings("resource") |
| @org.junit.jupiter.api.Test |
| public void testMultipleLoaders() throws Exception { |
| |
| String baseDir = System.getProperty("basedir"); |
| String fs = System.getProperty("file.separator"); |
| File file0 = new File(baseDir + fs + "build" + fs + "classes" + fs); |
| File file1 = new File(baseDir + fs + "build" + fs + "test" + fs); |
| URL[] urls = new URL[2]; |
| urls[0] = file0.toURI().toURL(); |
| urls[1] = file1.toURI().toURL(); |
| URLClassLoader uc1 = new URLClassLoader |
| (urls, Thread.currentThread().getContextClassLoader()); |
| URLClassLoader uc2 = new URLClassLoader |
| (urls, Thread.currentThread().getContextClassLoader()); |
| |
| Class<?> c1 = uc1.loadClass("javax.xml.crypto.test.dsig.Driver"); |
| Class<?> c2 = uc2.loadClass("javax.xml.crypto.test.dsig.Driver"); |
| Object o1 = c1.newInstance(); |
| Object o2 = c2.newInstance(); |
| Method m1 = c1.getMethod("dsig", (Class[]) null); |
| Method m2 = c2.getMethod("dsig", (Class[]) null); |
| m1.invoke(o1, (Object[]) null); |
| m2.invoke(o2, (Object[]) null); |
| } |
| |
| @SuppressWarnings("resource") |
| @org.junit.jupiter.api.Test |
| public void testProviderMultipleLoaders() throws Exception { |
| String baseDir = System.getProperty("basedir"); |
| String fs = System.getProperty("file.separator"); |
| File file0 = new File(baseDir + fs + "build" + fs + "classes" + fs); |
| File file1 = new File(baseDir + fs + "build" + fs + "test" + fs); |
| URL[] urls = new URL[2]; |
| urls[0] = file0.toURI().toURL(); |
| urls[1] = file1.toURI().toURL(); |
| |
| URLClassLoader uc1 = new URLClassLoader |
| (urls, Thread.currentThread().getContextClassLoader()); |
| //load security provider using current class loader |
| final Provider provider = new XMLDSigRI(); |
| AccessController.doPrivileged((java.security.PrivilegedAction<Object>) |
| () -> Security.addProvider(provider)); |
| // get the provider from java.security.Security using URLClassLoader. |
| // Need to use introspection to invoke methods to avoid using the |
| // current class loader |
| String factoryName = "javax.xml.crypto.dsig.XMLSignatureFactory"; |
| Class<?> factoryClass = uc1.loadClass(factoryName); |
| Method factoryMethod = factoryClass.getDeclaredMethod |
| ("getInstance", new Class[]{String.class}); |
| Class<?> methodParameterClass = uc1.loadClass |
| ("javax.xml.crypto.dsig.spec.C14NMethodParameterSpec"); |
| Method canonicalizationMethod = factoryClass.getDeclaredMethod |
| ("newCanonicalizationMethod", |
| new Class[]{String.class, methodParameterClass}); |
| Object factory = factoryMethod.invoke(null, "DOM"); |
| long start = System.currentTimeMillis(); |
| for (int i = 0; i < 100; i++) { |
| canonicalizationMethod.invoke |
| (factory, new Object[]{CanonicalizationMethod.EXCLUSIVE, null}); |
| } |
| long end = System.currentTimeMillis(); |
| long elapsed = end-start; |
| LOG.debug("Elapsed: {}", elapsed); |
| } |
| |
| @SuppressWarnings("resource") |
| @org.junit.jupiter.api.Test |
| public void testProviderMultipleLoadersTwo() throws Exception { |
| |
| String baseDir = System.getProperty("basedir"); |
| String fs = System.getProperty("file.separator"); |
| File file0 = new File(baseDir + fs + "build" + fs + "classes" + fs); |
| File file1 = new File(baseDir + fs + "build" + fs + "test" + fs); |
| URL[] urls = new URL[2]; |
| urls[0] = file0.toURI().toURL(); |
| urls[1] = file1.toURI().toURL(); |
| URLClassLoader uc1 = new URLClassLoader |
| (urls, Thread.currentThread().getContextClassLoader()); |
| URLClassLoader uc2 = new URLClassLoader |
| (urls, Thread.currentThread().getContextClassLoader()); |
| |
| Class<?> c1 = uc1.loadClass("javax.xml.crypto.test.dsig.AppA"); |
| Class<?> c2 = uc2.loadClass("javax.xml.crypto.test.dsig.AppB"); |
| Object o1 = c1.newInstance(); |
| Object o2 = c2.newInstance(); |
| Method m1 = c1.getMethod("dsig", (Class[]) null); |
| Method m2 = c2.getMethod("dsig", (Class[]) null); |
| m1.invoke(o1, (Object[]) null); |
| m2.invoke(o2, (Object[]) null); |
| } |
| |
| } |