org.pkl.thirdparty.truffle.host.HostMethodDesc Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of pkl-tools Show documentation
Show all versions of pkl-tools Show documentation
Fat Jar containing pkl-cli, pkl-codegen-java, pkl-codegen-kotlin, pkl-config-java, pkl-core, pkl-doc, and their shaded third-party dependencies.
/*
* Copyright (c) 2017, 2022, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* The Universal Permissive License (UPL), Version 1.0
*
* Subject to the condition set forth below, permission is hereby granted to any
* person obtaining a copy of this software, associated documentation and/or
* data (collectively the "Software"), free of charge and under any and all
* copyright rights in the Software, and any and all patent rights owned or
* freely licensable by each licensor hereunder covering either (i) the
* unmodified Software as contributed to or provided by such licensor, or (ii)
* the Larger Works (as defined below), to deal in both
*
* (a) the Software, and
*
* (b) any piece of software and/or hardware listed in the lrgrwrks.txt file if
* one is included with the Software each a "Larger Work" to which the Software
* is contributed by such licensors),
*
* without restriction, including without limitation the rights to copy, create
* derivative works of, display, perform, and distribute the Software and make,
* use, sell, offer for sale, import, export, have made, and have sold the
* Software and the Larger Work(s), and to sublicense the foregoing rights on
* either these or other terms.
*
* This license is subject to the following condition:
*
* The above copyright notice and either this complete permission notice or at a
* minimum a reference to the UPL must be included in all copies or substantial
* portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package org.pkl.thirdparty.truffle.host;
import java.lang.annotation.Annotation;
import java.lang.invoke.MethodHandle;
import java.lang.invoke.MethodHandles;
import java.lang.reflect.Array;
import java.lang.reflect.Constructor;
import java.lang.reflect.Executable;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.lang.reflect.Type;
import java.util.StringJoiner;
import org.pkl.thirdparty.truffle.api.CallTarget;
import org.pkl.thirdparty.truffle.api.CompilerAsserts;
import org.pkl.thirdparty.truffle.api.CompilerDirectives;
import org.pkl.thirdparty.truffle.api.CompilerDirectives.CompilationFinal;
import org.pkl.thirdparty.truffle.api.CompilerDirectives.TruffleBoundary;
import org.pkl.thirdparty.truffle.api.TruffleOptions;
import org.pkl.thirdparty.truffle.api.nodes.Node;
abstract class HostMethodDesc {
abstract String getName();
abstract String getDeclaringClassName();
abstract SingleMethod[] getOverloads();
boolean isInternal() {
return false;
}
abstract boolean isMethod();
abstract boolean isConstructor();
abstract static class SingleMethod extends HostMethodDesc {
static final int[] EMTPY_SCOPED_PARAMETERS = new int[0];
static final int NO_SCOPE = -1;
private final boolean varArgs;
@CompilationFinal(dimensions = 1) private final Class[] parameterTypes;
@CompilationFinal(dimensions = 1) private final Type[] genericParameterTypes;
@CompilationFinal(dimensions = 1) private final int[] scopedStaticParameters;
private final int scopedStaticParameterCount;
private final boolean onlyVisibleFromJniName;
private static final Class[] UNSCOPED_TYPES = {Boolean.class, Byte.class, Short.class, Character.class, Integer.class, Long.class, Float.class, Double.class, String.class};
protected SingleMethod(Executable executable, boolean parametersScoped, boolean onlyVisibleFromJniName) {
this.varArgs = executable.isVarArgs();
this.parameterTypes = executable.getParameterTypes();
this.genericParameterTypes = executable.getGenericParameterTypes();
int[] scopedParams = null;
int count = 0;
if (parametersScoped) {
scopedParams = new int[parameterTypes.length];
for (int i = 0; i < parameterTypes.length; i++) {
if (isScoped(parameterTypes[i])) {
scopedParams[i] = count++;
} else {
scopedParams[i] = NO_SCOPE;
}
}
}
this.scopedStaticParameterCount = count;
if (count > 0) {
assert scopedParams != null;
this.scopedStaticParameters = scopedParams;
} else {
this.scopedStaticParameters = EMTPY_SCOPED_PARAMETERS;
}
this.onlyVisibleFromJniName = onlyVisibleFromJniName;
}
private SingleMethod(boolean varArgs, Class[] parameterTypes, Type[] genericParameterTypes, int[] scopedStaticParameters, int scopedStaticParameterCount) {
this.varArgs = varArgs;
this.parameterTypes = parameterTypes;
this.genericParameterTypes = genericParameterTypes;
this.scopedStaticParameters = scopedStaticParameters;
this.scopedStaticParameterCount = scopedStaticParameterCount;
this.onlyVisibleFromJniName = false;
}
private static boolean isScoped(Class c) {
if (c.isPrimitive()) {
return false;
}
for (Class unscopedType : UNSCOPED_TYPES) {
if (unscopedType.isAssignableFrom(c)) {
return false;
}
}
return true;
}
public boolean isOnlyVisibleFromJniName() {
return onlyVisibleFromJniName;
}
public abstract Executable getReflectionMethod();
public final boolean isVarArgs() {
return varArgs;
}
public final Class[] getParameterTypes() {
return parameterTypes;
}
public final int getParameterCount() {
return parameterTypes.length;
}
public Type[] getGenericParameterTypes() {
return genericParameterTypes;
}
public final boolean hasScopedParameters() {
return scopedStaticParameterCount > 0;
}
public final int[] getScopedParameters() {
return this.scopedStaticParameters;
}
public final int getScopedParameterCount() {
return scopedStaticParameterCount;
}
@Override
public String getName() {
return getReflectionMethod().getName();
}
@Override
String getDeclaringClassName() {
return getReflectionMethod().getDeclaringClass().getName();
}
@Override
public SingleMethod[] getOverloads() {
return new SingleMethod[]{this};
}
public abstract Object invoke(Object receiver, Object[] arguments) throws Throwable;
public abstract Object invokeGuestToHost(Object receiver, Object[] arguments, GuestToHostCodeCache cache, HostContext context, Node node);
@Override
public boolean isMethod() {
return getReflectionMethod() instanceof Method;
}
@Override
public boolean isConstructor() {
return getReflectionMethod() instanceof Constructor;
}
static SingleMethod unreflect(Method reflectionMethod, boolean scoped, boolean onlyVisibleFromJniName) {
assert isAccessible(reflectionMethod);
if (TruffleOptions.AOT || isCallerSensitive(reflectionMethod)) {
return new MethodReflectImpl(reflectionMethod, scoped, onlyVisibleFromJniName);
} else {
return new MethodMHImpl(reflectionMethod, scoped, onlyVisibleFromJniName);
}
}
static SingleMethod unreflect(Constructor reflectionConstructor, boolean scoped) {
assert isAccessible(reflectionConstructor);
if (TruffleOptions.AOT || isCallerSensitive(reflectionConstructor)) {
return new ConstructorReflectImpl(reflectionConstructor, scoped);
} else {
return new ConstructorMHImpl(reflectionConstructor, scoped);
}
}
static boolean isAccessible(Executable method) {
return Modifier.isPublic(method.getModifiers()) && Modifier.isPublic(method.getDeclaringClass().getModifiers());
}
private static final Class callerSensitiveClass = getCallerSensitiveClass();
@SuppressWarnings("unchecked")
private static Class getCallerSensitiveClass() {
Class tmpCallerSensitiveClass = null;
try {
tmpCallerSensitiveClass = (Class) Class.forName("jdk.internal.reflect.CallerSensitive");
} catch (ClassNotFoundException e) {
try {
tmpCallerSensitiveClass = (Class) Class.forName("sun.reflect.CallerSensitive");
} catch (ClassNotFoundException ex) {
// ignore
}
}
return tmpCallerSensitiveClass;
}
static boolean isCallerSensitive(Executable method) {
return callerSensitiveClass != null && method.isAnnotationPresent(callerSensitiveClass);
}
@Override
public String toString() {
return "Method[" + getReflectionMethod().toString() + "]";
}
abstract static class ReflectBase extends SingleMethod {
ReflectBase(Executable executable, boolean scoped, boolean onlyVisibleFromJniName) {
super(executable, scoped, onlyVisibleFromJniName);
}
@Override
public Object invokeGuestToHost(Object receiver, Object[] arguments, GuestToHostCodeCache cache, HostContext hostContext, Node node) {
CallTarget target = cache.reflectionHostInvoke;
return GuestToHostRootNode.guestToHostCall(node, target, hostContext, receiver, this, arguments);
}
}
private static final class MethodReflectImpl extends ReflectBase {
private final Method reflectionMethod;
MethodReflectImpl(Method reflectionMethod, boolean scoped, boolean onlyVisibleFromJniName) {
super(reflectionMethod, scoped, onlyVisibleFromJniName);
this.reflectionMethod = reflectionMethod;
}
@Override
public Method getReflectionMethod() {
CompilerAsserts.neverPartOfCompilation();
return reflectionMethod;
}
@Override
public Object invoke(Object receiver, Object[] arguments) throws Throwable {
try {
return reflectInvoke(reflectionMethod, receiver, arguments);
} catch (InvocationTargetException e) {
throw e.getCause();
}
}
@TruffleBoundary
private static Object reflectInvoke(Method reflectionMethod, Object receiver, Object[] arguments)
throws IllegalAccessException, IllegalArgumentException, InvocationTargetException {
return reflectionMethod.invoke(receiver, arguments);
}
@Override
public boolean isInternal() {
return getReflectionMethod().getDeclaringClass() == Object.class;
}
}
private static final class ConstructorReflectImpl extends ReflectBase {
private final Constructor reflectionConstructor;
ConstructorReflectImpl(Constructor reflectionConstructor, boolean scoped) {
super(reflectionConstructor, scoped, false);
this.reflectionConstructor = reflectionConstructor;
}
@Override
public Constructor getReflectionMethod() {
CompilerAsserts.neverPartOfCompilation();
return reflectionConstructor;
}
@Override
public Object invoke(Object receiver, Object[] arguments) throws Throwable {
try {
return reflectNewInstance(reflectionConstructor, arguments);
} catch (InvocationTargetException e) {
throw e.getCause();
}
}
@TruffleBoundary
private static Object reflectNewInstance(Constructor reflectionConstructor, Object[] arguments)
throws InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
return reflectionConstructor.newInstance(arguments);
}
}
abstract static class MHBase extends SingleMethod {
@CompilationFinal private MethodHandle methodHandle;
MHBase(Executable executable, boolean scoped, boolean onlyVisibleFromJniName) {
super(executable, scoped, onlyVisibleFromJniName);
}
@Override
public final Object invoke(Object receiver, Object[] arguments) throws Throwable {
MethodHandle handle = methodHandle;
if (handle == null) {
CompilerDirectives.transferToInterpreterAndInvalidate();
handle = makeMethodHandle();
methodHandle = handle;
}
return invokeHandle(handle, receiver, arguments);
}
@TruffleBoundary(allowInlining = true)
static Object invokeHandle(MethodHandle invokeHandle, Object receiver, Object[] arguments) throws Throwable {
return invokeHandle.invokeExact(receiver, arguments);
}
protected abstract MethodHandle makeMethodHandle();
@TruffleBoundary
private MethodHandle makeMethodHandleBoundary() {
return makeMethodHandle();
}
protected static MethodHandle adaptSignature(MethodHandle originalHandle, boolean isStatic, int parameterCount) {
MethodHandle adaptedHandle = originalHandle;
adaptedHandle = adaptedHandle.asType(adaptedHandle.type().changeReturnType(Object.class));
if (isStatic) {
adaptedHandle = MethodHandles.dropArguments(adaptedHandle, 0, Object.class);
} else {
adaptedHandle = adaptedHandle.asType(adaptedHandle.type().changeParameterType(0, Object.class));
}
adaptedHandle = adaptedHandle.asSpreader(Object[].class, parameterCount);
return adaptedHandle;
}
@Override
public Object invokeGuestToHost(Object receiver, Object[] arguments, GuestToHostCodeCache cache, HostContext hostContext, Node node) {
MethodHandle handle = methodHandle;
if (handle == null) {
if (CompilerDirectives.isPartialEvaluationConstant(this)) {
// we must not repeatedly deoptimize if MHBase is uncached.
// it ok to modify the methodHandle here even though it is compilation final
// because it is always initialized to the same value.
CompilerDirectives.transferToInterpreterAndInvalidate();
}
methodHandle = handle = makeMethodHandleBoundary();
}
CallTarget target = cache.methodHandleHostInvoke;
CompilerAsserts.partialEvaluationConstant(target);
return GuestToHostRootNode.guestToHostCall(node, target, hostContext, receiver, handle, arguments);
}
}
private static final class MethodMHImpl extends MHBase {
private final Method reflectionMethod;
MethodMHImpl(Method reflectionMethod, boolean scoped, boolean onlyVisibleFromJniName) {
super(reflectionMethod, scoped, onlyVisibleFromJniName);
this.reflectionMethod = reflectionMethod;
}
@Override
public Method getReflectionMethod() {
CompilerAsserts.neverPartOfCompilation();
return reflectionMethod;
}
@Override
public boolean isInternal() {
return getReflectionMethod().getDeclaringClass() == Object.class;
}
@Override
@TruffleBoundary
protected MethodHandle makeMethodHandle() {
try {
Method m = reflectionMethod;
final MethodHandle methodHandle = MethodHandles.publicLookup().unreflect(m);
return adaptSignature(methodHandle, Modifier.isStatic(m.getModifiers()), m.getParameterCount());
} catch (IllegalAccessException e) {
throw new IllegalStateException(e);
}
}
}
private static final class ConstructorMHImpl extends MHBase {
private final Constructor reflectionConstructor;
ConstructorMHImpl(Constructor reflectionConstructor, boolean scoped) {
super(reflectionConstructor, scoped, false);
this.reflectionConstructor = reflectionConstructor;
}
@Override
public Constructor getReflectionMethod() {
CompilerAsserts.neverPartOfCompilation();
return reflectionConstructor;
}
@Override
@TruffleBoundary
protected MethodHandle makeMethodHandle() {
CompilerAsserts.neverPartOfCompilation();
try {
final MethodHandle methodHandle = MethodHandles.publicLookup().unreflectConstructor(reflectionConstructor);
return adaptSignature(methodHandle, true, getParameterCount());
} catch (IllegalAccessException e) {
throw new IllegalStateException(e);
}
}
}
static final class SyntheticArrayCloneMethod extends SingleMethod {
static final SyntheticArrayCloneMethod SINGLETON = new SyntheticArrayCloneMethod();
private SyntheticArrayCloneMethod() {
super(false, new Class[0], new Type[0], EMTPY_SCOPED_PARAMETERS, 0);
}
@Override
public String getName() {
return "clone";
}
@Override
String getDeclaringClassName() {
return null;
}
@Override
public String toString() {
return "Method[clone]";
}
@Override
public Executable getReflectionMethod() {
throw CompilerDirectives.shouldNotReachHere();
}
@Override
public Object invoke(Object receiver, Object[] arguments) {
assert receiver != null && receiver.getClass().isArray() && arguments.length == 0;
// Object#clone() is protected so clone the array via reflection.
int length = Array.getLength(receiver);
Object copy = Array.newInstance(receiver.getClass().getComponentType(), length);
System.arraycopy(receiver, 0, copy, 0, length);
return copy;
}
@Override
public Object invokeGuestToHost(Object receiver, Object[] arguments, GuestToHostCodeCache cache, HostContext context, Node node) {
return HostObject.forObject(invoke(receiver, arguments), context);
}
}
}
static final class OverloadedMethod extends HostMethodDesc {
private final SingleMethod[] overloads;
OverloadedMethod(SingleMethod[] overloads) {
this.overloads = overloads;
assert overloads.length >= 2;
}
@Override
public SingleMethod[] getOverloads() {
return overloads;
}
@Override
public String getName() {
return getOverloads()[0].getName();
}
@Override
String getDeclaringClassName() {
return getOverloads()[0].getDeclaringClassName();
}
@Override
public boolean isMethod() {
return getOverloads()[0].isMethod();
}
@Override
public boolean isConstructor() {
return getOverloads()[0].isConstructor();
}
@Override
public String toString() {
StringJoiner sj = new StringJoiner(", ", "Method[", "]");
for (SingleMethod overload : getOverloads()) {
sj.add(overload.getReflectionMethod().toString());
}
return sj.toString();
}
@Override
public boolean isInternal() {
for (SingleMethod overload : overloads) {
if (!overload.isInternal()) {
return false;
}
}
return true;
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy