Hugh.C
5 years ago
42 changed files with 571 additions and 500 deletions
@ -0,0 +1,78 @@
|
||||
/* |
||||
* Copyright 2006-2018 the original author or authors. |
||||
* |
||||
* Licensed 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 com.fr.third.org.objenesis.instantiator.util; |
||||
|
||||
import com.fr.third.org.objenesis.ObjenesisException; |
||||
|
||||
/** |
||||
* Helper class for to play with classes. It contains everything needed to play with a class
|
||||
* except the dodgy (Java 8) code you will find in {@link ClassDefinitionUtils}. |
||||
* |
||||
* @author Henri Tremblay |
||||
*/ |
||||
public final class ClassUtils { |
||||
|
||||
private ClassUtils() { } |
||||
|
||||
/** |
||||
* Will convert a class name to its name in the class definition format (e.g {@code com.fr.third.org.objenesis.EmptyClass} |
||||
* becomes {@code org/objenesis/EmptyClass}) |
||||
* |
||||
* @param className full class name including the package
|
||||
* @return the internal name |
||||
*/ |
||||
public static String classNameToInternalClassName(String className) { |
||||
return className.replace('.', '/'); |
||||
} |
||||
|
||||
/** |
||||
* Will convert a class name to its class loader resource name (e.g {@code com.fr.third.org.objenesis.EmptyClass} |
||||
* becomes {@code org/objenesis/EmptyClass.class}) |
||||
* |
||||
* @param className full class name including the package
|
||||
* @return the resource name |
||||
*/ |
||||
public static String classNameToResource(String className) { |
||||
return classNameToInternalClassName(className) + ".class"; |
||||
} |
||||
|
||||
/** |
||||
* Check if this class already exists in the class loader and return it if it does |
||||
* |
||||
* @param <T> type of the class returned |
||||
* @param classLoader Class loader where to search the class
|
||||
* @param className Class name with full path |
||||
* @return the class if it already exists or null |
||||
*/ |
||||
@SuppressWarnings("unchecked") |
||||
public static <T> Class<T> getExistingClass(ClassLoader classLoader, String className) { |
||||
try { |
||||
return (Class<T>) Class.forName(className, true, classLoader); |
||||
} |
||||
catch (ClassNotFoundException e) { |
||||
return null; |
||||
} |
||||
} |
||||
|
||||
@SuppressWarnings("deprecation") |
||||
public static <T> T newInstance(Class<T> clazz) { |
||||
try { |
||||
return clazz.newInstance(); |
||||
} catch (Exception e) { |
||||
throw new ObjenesisException(e); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,147 @@
|
||||
/* |
||||
* Copyright 2006-2018 the original author or authors. |
||||
* |
||||
* Licensed 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 com.fr.third.org.objenesis.instantiator.util; |
||||
|
||||
import sun.misc.Unsafe; |
||||
import com.fr.third.org.objenesis.ObjenesisException; |
||||
import com.fr.third.org.objenesis.strategy.PlatformDescription; |
||||
|
||||
import java.lang.invoke.MethodHandle; |
||||
import java.lang.invoke.MethodHandles; |
||||
import java.lang.invoke.MethodType; |
||||
import java.security.ProtectionDomain; |
||||
|
||||
/** |
||||
* Java 11+ removed sun.misc.Unsafe.defineClass. This class bridges the gap to work from Java 1.8 up to 11. |
||||
* <p> |
||||
* It was inspired from <a href="https://github.com/jboss-javassist/javassist/blob/master/src/main/javassist/util/proxy/DefineClassHelper.java">javassist</a>. |
||||
* |
||||
* @author Henri Tremblay |
||||
*/ |
||||
public final class DefineClassHelper { |
||||
|
||||
private static abstract class Helper { |
||||
abstract Class<?> defineClass(String name, byte[] b, int off, int len, Class<?> neighbor, |
||||
ClassLoader loader, ProtectionDomain protectionDomain); |
||||
} |
||||
|
||||
private static class Java8 extends Helper { |
||||
|
||||
private final MethodHandle defineClass = defineClass(); |
||||
|
||||
private MethodHandle defineClass() { |
||||
MethodType mt = MethodType.methodType(Class.class, String.class, byte[].class, int.class, int.class, ClassLoader.class, ProtectionDomain.class); |
||||
MethodHandle m; |
||||
try { |
||||
m = MethodHandles.publicLookup().findVirtual(Unsafe.class, "defineClass", mt); |
||||
} catch(Exception e) { |
||||
throw new ObjenesisException(e); |
||||
} |
||||
Unsafe unsafe = UnsafeUtils.getUnsafe(); |
||||
return m.bindTo(unsafe); |
||||
} |
||||
|
||||
@Override |
||||
Class<?> defineClass(String className, byte[] b, int off, int len, Class<?> neighbor, ClassLoader loader, ProtectionDomain protectionDomain) { |
||||
try { |
||||
return (Class<?>) defineClass.invokeExact(className, b, off, len, loader, protectionDomain); |
||||
} catch (Throwable e) { |
||||
if(e instanceof Error) { |
||||
throw (Error) e; |
||||
} |
||||
if(e instanceof RuntimeException) { |
||||
throw (RuntimeException) e; |
||||
} |
||||
throw new ObjenesisException(e); |
||||
} |
||||
} |
||||
} |
||||
|
||||
private static class Java11 extends Helper { |
||||
|
||||
private final Class<?> module = module(); |
||||
private final MethodHandles.Lookup lookup = MethodHandles.lookup(); |
||||
private final MethodHandle getModule = getModule(); |
||||
private final MethodHandle addReads = addReads(); |
||||
private final MethodHandle privateLookupIn = privateLookupIn(); |
||||
private final MethodHandle defineClass = defineClass(); |
||||
|
||||
private Class<?> module() { |
||||
try { |
||||
return Class.forName("java.lang.Module"); |
||||
} catch (ClassNotFoundException e) { |
||||
throw new ObjenesisException(e); |
||||
} |
||||
} |
||||
|
||||
private MethodHandle getModule() { |
||||
try { |
||||
return lookup.findVirtual(Class.class, "getModule", MethodType.methodType(module)); |
||||
} catch (Exception e) { |
||||
throw new ObjenesisException(e); |
||||
} |
||||
} |
||||
|
||||
private MethodHandle addReads() { |
||||
try { |
||||
return lookup.findVirtual(module, "addReads", MethodType.methodType(module, module)); |
||||
} catch (Exception e) { |
||||
throw new ObjenesisException(e); |
||||
} |
||||
} |
||||
|
||||
private MethodHandle privateLookupIn() { |
||||
try { |
||||
return lookup.findStatic(MethodHandles.class, "privateLookupIn", MethodType.methodType(MethodHandles.Lookup.class, Class.class, MethodHandles.Lookup.class)); |
||||
} catch (Exception e) { |
||||
throw new ObjenesisException(e); |
||||
} |
||||
} |
||||
|
||||
private MethodHandle defineClass() { |
||||
try { |
||||
return lookup.findVirtual(MethodHandles.Lookup.class, "defineClass", MethodType.methodType(Class.class, byte[].class)); |
||||
} catch (Exception e) { |
||||
throw new ObjenesisException(e); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
Class<?> defineClass(String className, byte[] b, int off, int len, Class<?> neighbor, ClassLoader loader, ProtectionDomain protectionDomain) { |
||||
try { |
||||
Object module = getModule.invokeWithArguments(DefineClassHelper.class); |
||||
Object neighborModule = getModule.invokeWithArguments(neighbor); |
||||
addReads.invokeWithArguments(module, neighborModule); |
||||
MethodHandles.Lookup prvlookup = (MethodHandles.Lookup) privateLookupIn.invokeExact(neighbor, lookup); |
||||
return (Class<?>) defineClass.invokeExact(prvlookup, b); |
||||
} catch (Throwable e) { |
||||
throw new ObjenesisException(neighbor.getName() + " has no permission to define the class", e); |
||||
} |
||||
} |
||||
} |
||||
|
||||
// Java 11+ removed sun.misc.Unsafe.defineClass, so we fallback to invoking defineClass on
|
||||
// ClassLoader until we have an implementation that uses MethodHandles.Lookup.defineClass
|
||||
private static final Helper privileged = PlatformDescription.isAfterJava11() ? |
||||
new Java11() : new Java8(); |
||||
|
||||
public static Class<?> defineClass(String name, byte[] b, int off, int len, Class<?> neighbor, |
||||
ClassLoader loader, ProtectionDomain protectionDomain) { |
||||
return privileged.defineClass(name, b, off, len, neighbor, loader, protectionDomain); |
||||
} |
||||
|
||||
private DefineClassHelper() {} |
||||
} |
||||
Loading…
Reference in new issue