com.fitbur.mockito.bytebuddy.implementation.MethodCall Maven / Gradle / Ivy
package com.fitbur.mockito.bytebuddy.implementation;
import com.fitbur.mockito.bytebuddy.description.enumeration.EnumerationDescription;
import com.fitbur.mockito.bytebuddy.description.field.FieldDescription;
import com.fitbur.mockito.bytebuddy.description.field.FieldList;
import com.fitbur.mockito.bytebuddy.description.method.MethodDescription;
import com.fitbur.mockito.bytebuddy.description.method.ParameterDescription;
import com.fitbur.mockito.bytebuddy.description.method.ParameterList;
import com.fitbur.mockito.bytebuddy.description.type.TypeDefinition;
import com.fitbur.mockito.bytebuddy.description.type.TypeDescription;
import com.fitbur.mockito.bytebuddy.dynamic.scaffold.InstrumentedType;
import com.fitbur.mockito.bytebuddy.implementation.bytecode.*;
import com.fitbur.mockito.bytebuddy.implementation.bytecode.assign.Assigner;
import com.fitbur.mockito.bytebuddy.implementation.bytecode.constant.*;
import com.fitbur.mockito.bytebuddy.implementation.bytecode.member.FieldAccess;
import com.fitbur.mockito.bytebuddy.implementation.bytecode.member.MethodInvocation;
import com.fitbur.mockito.bytebuddy.implementation.bytecode.member.MethodReturn;
import com.fitbur.mockito.bytebuddy.implementation.bytecode.member.MethodVariableAccess;
import com.fitbur.mockito.bytebuddy.utility.CompoundList;
import com.fitbur.mockito.bytebuddy.utility.JavaConstant;
import com.fitbur.mockito.bytebuddy.utility.JavaType;
import com.fitbur.mockito.bytebuddy.utility.RandomString;
import com.fitbur.mockito.bytebuddy.jar.asm.MethodVisitor;
import com.fitbur.mockito.bytebuddy.jar.asm.Opcodes;
import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.concurrent.Callable;
import static com.fitbur.mockito.bytebuddy.matcher.ElementMatchers.isVisibleTo;
import static com.fitbur.mockito.bytebuddy.matcher.ElementMatchers.named;
/**
* This {@link Implementation} allows the invocation of a specified method while
* providing explicit arguments to this method.
*/
public class MethodCall implements Implementation.Composable {
/**
* A reference to {@link Callable#call()}.
*/
private static final MethodDescription CALL;
/**
* A reference to {@link Runnable#run()}.
*/
private static final MethodDescription RUN;
/*
* Looks up references to known methods.
*/
static {
try {
CALL = new MethodDescription.ForLoadedMethod(Callable.class.getDeclaredMethod("call"));
RUN = new MethodDescription.ForLoadedMethod(Runnable.class.getDeclaredMethod("run"));
} catch (NoSuchMethodException exception) {
throw new RuntimeException("Cannot find standard method", exception);
}
}
/**
* The method locator to use.
*/
protected final MethodLocator methodLocator;
/**
* The target handler to use.
*/
protected final TargetHandler targetHandler;
/**
* The argument loader to load arguments onto the operand stack in their application order.
*/
protected final List argumentLoaders;
/**
* The method invoker to use.
*/
protected final MethodInvoker methodInvoker;
/**
* The termination handler to use.
*/
protected final TerminationHandler terminationHandler;
/**
* The assigner to use.
*/
protected final Assigner assigner;
/**
* Indicates if dynamic type castings should be attempted for incompatible assignments.
*/
protected final Assigner.Typing typing;
/**
* Creates a new method call implementation.
*
* @param methodLocator The method locator to use.
* @param targetHandler The target handler to use.
* @param argumentLoaders The argument loader to load arguments onto the operand stack in
* their application order.
* @param methodInvoker The method invoker to use.
* @param terminationHandler The termination handler to use.
* @param assigner The assigner to use.
* @param typing Indicates if dynamic type castings should be attempted for incompatible assignments.
*/
protected MethodCall(MethodLocator methodLocator,
TargetHandler targetHandler,
List argumentLoaders,
MethodInvoker methodInvoker,
TerminationHandler terminationHandler,
Assigner assigner,
Assigner.Typing typing) {
this.methodLocator = methodLocator;
this.targetHandler = targetHandler;
this.argumentLoaders = argumentLoaders;
this.methodInvoker = methodInvoker;
this.terminationHandler = terminationHandler;
this.assigner = assigner;
this.typing = typing;
}
/**
* Invokes the given method. Without further specification, the method is invoked without any arguments on
* the instance of the instrumented class or statically, if the given method is {@code static}.
*
* @param method The method to invoke.
* @return A method call implementation that invokes the given method without providing any arguments.
*/
public static WithoutSpecifiedTarget invoke(Method method) {
return invoke(new MethodDescription.ForLoadedMethod(method));
}
/**
* Invokes the given constructor on the instance of the instrumented type.
*
* @param constructor The constructor to invoke.
* @return A method call implementation that invokes the given constructor without providing any arguments.
*/
public static WithoutSpecifiedTarget invoke(Constructor constructor) {
return invoke(new MethodDescription.ForLoadedConstructor(constructor));
}
/**
* Invokes the given method. If the method description describes a constructor, it is automatically invoked as
* a special method invocation on the instance of the instrumented type. The same is true for {@code private}
* methods. Finally, {@code static} methods are invoked statically.
*
* @param methodDescription The method to invoke.
* @return A method call implementation that invokes the given method without providing any arguments.
*/
public static WithoutSpecifiedTarget invoke(MethodDescription methodDescription) {
return invoke(new MethodLocator.ForExplicitMethod(methodDescription));
}
/**
* Invokes a method using the provided method locator.
*
* @param methodLocator The method locator to apply for locating the method to invoke given the instrumented
* method.
* @return A method call implementation that uses the provided method locator for resolving the method
* to be invoked.
*/
public static WithoutSpecifiedTarget invoke(MethodLocator methodLocator) {
return new WithoutSpecifiedTarget(methodLocator);
}
/**
* Implements a method by invoking the provided {@link Callable}. The return value of the provided object is casted to the implemented method's
* return type, if necessary.
*
* @param callable The callable to invoke when a method is intercepted.
* @return A composable method implementation that invokes the given callable.
*/
public static Composable call(Callable callable) {
return invoke(CALL).on(callable).withAssigner(Assigner.DEFAULT, Assigner.Typing.DYNAMIC);
}
/**
* Implements a method by invoking the provided {@link Runnable}. If the instrumented method returns a value, {@code null} is returned.
*
* @param runnable The runnable to invoke when a method is intercepted.
* @return A composable method implementation that invokes the given runnable.
*/
public static Composable run(Runnable runnable) {
return invoke(RUN).on(runnable).withAssigner(Assigner.DEFAULT, Assigner.Typing.DYNAMIC);
}
/**
* Invokes the given constructor in order to create an instance.
*
* @param constructor The constructor to invoke.
* @return A method call that invokes the given constructor without providing any arguments.
*/
public static MethodCall construct(Constructor constructor) {
return construct(new MethodDescription.ForLoadedConstructor(constructor));
}
/**
* Invokes the given constructor in order to create an instance.
*
* @param methodDescription A description of the constructor to invoke.
* @return A method call that invokes the given constructor without providing any arguments.
*/
public static MethodCall construct(MethodDescription methodDescription) {
if (!methodDescription.isConstructor()) {
throw new IllegalArgumentException("Not a constructor: " + methodDescription);
}
return new MethodCall(new MethodLocator.ForExplicitMethod(methodDescription),
TargetHandler.ForConstructingInvocation.INSTANCE,
Collections.emptyList(),
MethodInvoker.ForContextualInvocation.INSTANCE,
TerminationHandler.ForMethodReturn.INSTANCE,
Assigner.DEFAULT,
Assigner.Typing.STATIC);
}
/**
* Invokes the method that is instrumented by the returned instance by a super method invocation.
*
* @return A method call that invokes the method being instrumented.
*/
public static MethodCall invokeSuper() {
return new WithoutSpecifiedTarget(MethodLocator.ForInstrumentedMethod.INSTANCE).onSuper();
}
/**
* Defines a number of arguments to be handed to the method that is being invoked by this implementation. Any
* wrapper type instances for primitive values, instances of {@link java.lang.String} or {@code null} are loaded
* directly onto the operand stack. This might corrupt referential identity for these values. Any other values
* are stored within a {@code static} field that is added to the instrumented type.
*
* @param argument The arguments to provide to the method that is being called in their order.
* @return A method call that hands the provided arguments to the invoked method.
*/
public MethodCall with(Object... argument) {
List argumentLoaders = new ArrayList(argument.length);
for (Object anArgument : argument) {
argumentLoaders.add(ArgumentLoader.ForStaticField.Factory.of(anArgument));
}
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(this.argumentLoaders, argumentLoaders),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Defines the given types to be provided as arguments to the invoked method where the represented types
* are stored in the generated class's constant pool.
*
* @param typeDescription The type descriptions to provide as arguments.
* @return A method call that hands the provided arguments to the invoked method.
*/
public MethodCall with(TypeDescription... typeDescription) {
List argumentLoaders = new ArrayList(typeDescription.length);
for (TypeDescription aTypeDescription : typeDescription) {
argumentLoaders.add(new ArgumentLoader.ForClassConstant(aTypeDescription));
}
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(this.argumentLoaders, argumentLoaders),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Defines the given enumeration values to be provided as arguments to the invoked method where the values
* are read from the enumeration class on demand.
*
* @param enumerationDescription The enumeration descriptions to provide as arguments.
* @return A method call that hands the provided arguments to the invoked method.
*/
public MethodCall with(EnumerationDescription... enumerationDescription) {
List argumentLoaders = new ArrayList(enumerationDescription.length);
for (EnumerationDescription anEnumerationDescription : enumerationDescription) {
argumentLoaders.add(new ArgumentLoader.ForEnumerationValue(anEnumerationDescription));
}
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(this.argumentLoaders, argumentLoaders),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Defines the given Java instances to be provided as arguments to the invoked method where the given
* instances are stored in the generated class's constant pool.
*
* @param javaConstant The Java instances to provide as arguments.
* @return A method call that hands the provided arguments to the invoked method.
*/
public MethodCall with(JavaConstant... javaConstant) {
List argumentLoaders = new ArrayList(javaConstant.length);
for (JavaConstant aJavaConstant : javaConstant) {
argumentLoaders.add(new ArgumentLoader.ForJavaConstant(aJavaConstant));
}
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(this.argumentLoaders, argumentLoaders),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Defines a number of arguments to be handed to the method that is being invoked by this implementation. Any
* value is stored within a field in order to preserve referential identity. As an exception, the {@code null}
* value is not stored within a field.
*
* @param argument The arguments to provide to the method that is being called in their order.
* @return A method call that hands the provided arguments to the invoked method.
*/
public MethodCall withReference(Object... argument) {
List argumentLoaders = new ArrayList(argument.length);
for (Object anArgument : argument) {
argumentLoaders.add(anArgument == null
? ArgumentLoader.ForNullConstant.INSTANCE
: new ArgumentLoader.ForStaticField.Factory(anArgument));
}
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(this.argumentLoaders, argumentLoaders),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Defines a number of arguments of the instrumented method by their parameter indices to be handed
* to the invoked method as an argument.
*
* @param index The parameter indices of the instrumented method to be handed to the invoked method as an
* argument in their order. The indices are zero-based.
* @return A method call that hands the provided arguments to the invoked method.
*/
public MethodCall withArgument(int... index) {
List argumentLoaders = new ArrayList(index.length);
for (int anIndex : index) {
if (anIndex < 0) {
throw new IllegalArgumentException("Negative index: " + anIndex);
}
argumentLoaders.add(new ArgumentLoader.ForMethodParameter.Factory(anIndex));
}
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(this.argumentLoaders, argumentLoaders),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Adds all arguments of the instrumented method as arguments to the invoked method to this method call.
*
* @return A method call that hands all arguments arguments of the instrumented method to the invoked method.
*/
public MethodCall withAllArguments() {
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(this.argumentLoaders, ArgumentLoader.ForMethodParameter.OfInstrumentedMethod.INSTANCE),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Assigns the {@code this} reference to the next parameter.
*
* @return This method call where the next parameter is a assigned a reference to the {@code this} reference
* of the instance of the intercepted method.
*/
public MethodCall withThis() {
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(argumentLoaders, ArgumentLoader.ForThisReference.Factory.INSTANCE),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Assigns the {@link java.lang.Class} value of the instrumented type.
*
* @return This method call where the next parameter is a assigned a reference to the {@link java.lang.Class}
* value of the instrumented type.
*/
public MethodCall withOwnType() {
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(argumentLoaders, ArgumentLoader.ForInstrumentedType.Factory.INSTANCE),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Defines a method call which fetches a value from an instance field. The value of the field needs to be
* defined manually and is initialized with {@code null}. The field itself is defined by this implementation.
*
* @param type The type of the field.
* @param name The name of the field.
* @return A method call which assigns the next parameter to the value of the instance field.
*/
public MethodCall withInstanceField(Type type, String name) {
return withInstanceField(TypeDefinition.Sort.describe(type), name);
}
/**
* Defines a method call which fetches a value from an instance field. The value of the field needs to be
* defined manually and is initialized with {@code null}. The field itself is defined by this implementation.
*
* @param typeDefinition The type of the field.
* @param name The name of the field.
* @return A method call which assigns the next parameter to the value of the instance field.
*/
public MethodCall withInstanceField(TypeDefinition typeDefinition, String name) {
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(argumentLoaders, new ArgumentLoader.ForInstanceField.Factory(typeDefinition.asGenericType(), name)),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Defines a method call which fetches a value from an existing field. The field is not defines by this
* implementation.
*
* @param fieldName The name of the field.
* @return A method call which assigns the next parameter to the value of the given field.
*/
public MethodCall withField(String... fieldName) {
List argumentLoaders = new ArrayList(fieldName.length);
for (String aFieldName : fieldName) {
argumentLoaders.add(new ArgumentLoader.ForExistingField.Factory(aFieldName));
}
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(this.argumentLoaders, argumentLoaders),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Defines an assigner to be used for assigning values to the parameters of the invoked method. This assigner
* is also used for assigning the invoked method's return value to the return type of the instrumented method,
* if this method is not chained with
* {@link com.fitbur.mockito.bytebuddy.implementation.MethodCall#andThen(Implementation)} such
* that a return value of this method call is discarded.
*
* @param assigner The assigner to use.
* @param typing Indicates if dynamic type castings should be attempted for incompatible assignments.
* @return This method call using the provided assigner.
*/
public MethodCall withAssigner(Assigner assigner, Assigner.Typing typing) {
return new MethodCall(methodLocator,
targetHandler,
argumentLoaders,
methodInvoker,
terminationHandler,
assigner,
typing);
}
@Override
public Implementation andThen(Implementation implementation) {
return new Implementation.Compound(new MethodCall(methodLocator,
targetHandler,
argumentLoaders,
methodInvoker,
TerminationHandler.ForChainedInvocation.INSTANCE,
assigner,
typing), implementation);
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
for (ArgumentLoader.Factory argumentLoader : argumentLoaders) {
instrumentedType = argumentLoader.prepare(instrumentedType);
}
return targetHandler.prepare(instrumentedType);
}
@Override
public ByteCodeAppender appender(Target implementationTarget) {
return new Appender(implementationTarget);
}
@Override
public boolean equals(Object other) {
if (this == other) return true;
if (!(other instanceof MethodCall)) return false;
MethodCall that = (MethodCall) other;
return typing == that.typing
&& argumentLoaders.equals(that.argumentLoaders)
&& assigner.equals(that.assigner)
&& methodInvoker.equals(that.methodInvoker)
&& methodLocator.equals(that.methodLocator)
&& targetHandler.equals(that.targetHandler)
&& terminationHandler.equals(that.terminationHandler);
}
@Override
public int hashCode() {
int result = methodLocator.hashCode();
result = 31 * result + targetHandler.hashCode();
result = 31 * result + argumentLoaders.hashCode();
result = 31 * result + methodInvoker.hashCode();
result = 31 * result + terminationHandler.hashCode();
result = 31 * result + assigner.hashCode();
result = 31 * result + typing.hashCode();
return result;
}
@Override
public String toString() {
return "MethodCall{" +
"methodLocator=" + methodLocator +
", targetHandler=" + targetHandler +
", argumentLoaders=" + argumentLoaders +
", methodInvoker=" + methodInvoker +
", terminationHandler=" + terminationHandler +
", assigner=" + assigner +
", typing=" + typing +
'}';
}
/**
* A method locator is responsible for identifying the method that is to be invoked
* by a {@link com.fitbur.mockito.bytebuddy.implementation.MethodCall}.
*/
public interface MethodLocator {
/**
* Resolves the method to be invoked.
*
* @param instrumentedMethod The method being instrumented.
* @return The method to invoke.
*/
MethodDescription resolve(MethodDescription instrumentedMethod);
/**
* A method locator that simply returns the intercepted method.
*/
enum ForInstrumentedMethod implements MethodLocator {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public MethodDescription resolve(MethodDescription instrumentedMethod) {
return instrumentedMethod;
}
@Override
public String toString() {
return "MethodCall.MethodLocator.ForInstrumentedMethod." + name();
}
}
/**
* Invokes a given method.
*/
class ForExplicitMethod implements MethodLocator {
/**
* The method to be invoked.
*/
private final MethodDescription methodDescription;
/**
* Creates a new method locator for a given method.
*
* @param methodDescription The method to be invoked.
*/
public ForExplicitMethod(MethodDescription methodDescription) {
this.methodDescription = methodDescription;
}
@Override
public MethodDescription resolve(MethodDescription instrumentedMethod) {
return methodDescription;
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& methodDescription.equals(((ForExplicitMethod) other).methodDescription);
}
@Override
public int hashCode() {
return methodDescription.hashCode();
}
@Override
public String toString() {
return "MethodCall.MethodLocator.ForExplicitMethod{" +
"methodDescription=" + methodDescription +
'}';
}
}
}
/**
* A target handler is responsible for invoking a method for a
* {@link com.fitbur.mockito.bytebuddy.implementation.MethodCall}.
*/
protected interface TargetHandler {
/**
* Creates a stack manipulation that represents the method's invocation.
*
* @param invokedMethod The method to be invoked.
* @param instrumentedMethod The instrumented method.
* @param instrumentedType The instrumented type. @return A stack manipulation that invokes the method.
* @param assigner The assigner to use.
* @param typing The typing to apply.
* @return A stack manipulation that loads the method target onto the operand stack.
*/
StackManipulation resolve(MethodDescription invokedMethod,
MethodDescription instrumentedMethod,
TypeDescription instrumentedType,
Assigner assigner,
Assigner.Typing typing);
/**
* Prepares the instrumented type in order to allow for the represented invocation.
*
* @param instrumentedType The instrumented type.
* @return The prepared instrumented type.
*/
InstrumentedType prepare(InstrumentedType instrumentedType);
/**
* A target handler that invokes a method either on the instance of the instrumented
* type or as a static method.
*/
enum ForSelfOrStaticInvocation implements TargetHandler {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public StackManipulation resolve(MethodDescription invokedMethod, MethodDescription instrumentedMethod, TypeDescription instrumentedType, Assigner assigner, Assigner.Typing typing) {
return new StackManipulation.Compound(
invokedMethod.isStatic()
? StackManipulation.Trivial.INSTANCE
: MethodVariableAccess.REFERENCE.loadOffset(0),
invokedMethod.isConstructor()
? Duplication.SINGLE
: StackManipulation.Trivial.INSTANCE
);
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public String toString() {
return "MethodCall.TargetHandler.ForSelfOrStaticInvocation." + name();
}
}
/**
* Invokes a method in order to construct a new instance.
*/
enum ForConstructingInvocation implements TargetHandler {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public StackManipulation resolve(MethodDescription invokedMethod, MethodDescription instrumentedMethod, TypeDescription instrumentedType, Assigner assigner, Assigner.Typing typing) {
return new StackManipulation.Compound(TypeCreation.of(invokedMethod.getDeclaringType().asErasure()), Duplication.SINGLE);
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public String toString() {
return "MethodCall.TargetHandler.ForConstructingInvocation." + name();
}
}
/**
* A target handler that invokes a method on an instance that is stored in a static field.
*/
class ForStaticField implements TargetHandler {
/**
* The name prefix of the field to store the instance.
*/
private static final String FIELD_PREFIX = "invocationTarget";
/**
* The target on which the method is to be invoked.
*/
private final Object target;
/**
* The name of the field to store the target.
*/
private final String fieldName;
/**
* Creates a new target handler for a static field.
*
* @param target The target on which the method is to be invoked.
*/
public ForStaticField(Object target) {
this.target = target;
fieldName = String.format("%s$%s", FIELD_PREFIX, RandomString.make());
}
@Override
public StackManipulation resolve(MethodDescription invokedMethod, MethodDescription instrumentedMethod, TypeDescription instrumentedType, Assigner assigner, Assigner.Typing typing) {
return FieldAccess.forField(instrumentedType.getDeclaredFields().filter(named(fieldName)).getOnly()).getter();
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType
.withField(new FieldDescription.Token(fieldName,
Opcodes.ACC_SYNTHETIC | Opcodes.ACC_PUBLIC | Opcodes.ACC_STATIC,
new TypeDescription.Generic.OfNonGenericType.ForLoadedType(target.getClass())))
.withInitializer(new LoadedTypeInitializer.ForStaticField(fieldName, target));
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& target.equals(((ForStaticField) other).target);
}
@Override
public int hashCode() {
return target.hashCode();
}
@Override
public String toString() {
return "MethodCall.TargetHandler.ForStaticField{" +
"target=" + target +
", fieldName='" + fieldName + '\'' +
'}';
}
}
/**
* Creates a target handler that stores the instance to invoke a method on in an instance field.
*/
class ForInstanceField implements TargetHandler {
/**
* The name of the field.
*/
private final String fieldName;
/**
* The type of the field.
*/
private final TypeDescription.Generic fieldType;
/**
* Creates a new target handler for storing a method invocation target in an
* instance field.
*
* @param fieldName The name of the field.
* @param fieldType The type of the field.
*/
public ForInstanceField(String fieldName, TypeDescription.Generic fieldType) {
this.fieldName = fieldName;
this.fieldType = fieldType;
}
@Override
public StackManipulation resolve(MethodDescription invokedMethod, MethodDescription instrumentedMethod, TypeDescription instrumentedType, Assigner assigner, Assigner.Typing typing) {
return new StackManipulation.Compound(
invokedMethod.isStatic()
? StackManipulation.Trivial.INSTANCE
: MethodVariableAccess.REFERENCE.loadOffset(0),
FieldAccess.forField(instrumentedType.getDeclaredFields().filter(named(fieldName)).getOnly()).getter());
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
if (instrumentedType.isInterface()) {
throw new IllegalStateException("Cannot define non-static field '" + fieldName + "' on " + instrumentedType);
}
return instrumentedType.withField(new FieldDescription.Token(fieldName, Opcodes.ACC_SYNTHETIC | Opcodes.ACC_PUBLIC, fieldType));
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass()) && fieldName
.equals(((ForInstanceField) other).fieldName) && fieldType
.equals(((ForInstanceField) other).fieldType);
}
@Override
public int hashCode() {
int result = fieldName.hashCode();
result = 31 * result + fieldType.hashCode();
return result;
}
@Override
public String toString() {
return "MethodCall.TargetHandler.ForInstanceField{" +
"fieldName='" + fieldName + '\'' +
", fieldType=" + fieldType +
'}';
}
}
/**
* A target handler that loads the parameter of the given index as the target object.
*/
class ForMethodParameter implements TargetHandler {
/**
* The index of the instrumented method's parameter that is the target of the method invocation.
*/
private final int index;
/**
* Creates a new target handler for the instrumented method's argument.
*
* @param index The index of the instrumented method's parameter that is the target of the method invocation.
*/
public ForMethodParameter(int index) {
this.index = index;
}
@Override
public StackManipulation resolve(MethodDescription invokedMethod,
MethodDescription instrumentedMethod,
TypeDescription instrumentedType,
Assigner assigner,
Assigner.Typing typing) {
if (instrumentedMethod.getParameters().size() < index) {
throw new IllegalArgumentException(instrumentedMethod + " does not have a parameter with index " + index);
}
ParameterDescription parameterDescription = instrumentedMethod.getParameters().get(index);
StackManipulation stackManipulation = assigner.assign(parameterDescription.getType(), invokedMethod.getDeclaringType().asGenericType(), typing);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + parameterDescription.getType());
}
return new StackManipulation.Compound(MethodVariableAccess.of(parameterDescription.getType()).loadOffset(parameterDescription.getOffset()), stackManipulation);
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass()) && index == ((ForMethodParameter) other).index;
}
@Override
public int hashCode() {
return index;
}
@Override
public String toString() {
return "MethodCall.TargetHandler.ForMethodParameter{" +
"index=" + index +
'}';
}
}
}
/**
* An argument loader is responsible for loading an argument for an invoked method
* onto the operand stack.
*/
protected interface ArgumentLoader {
/**
* Loads the argument that is represented by this instance onto the operand stack.
*
* @param target The target parameter.
* @param assigner The assigner to be used.
* @param typing Indicates if dynamic type castings should be attempted for incompatible assignments.
* @return The stack manipulation that loads the represented argument onto the stack.
*/
StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing);
/**
* A factory that produces {@link ArgumentLoader}s for a given instrumented method.
*/
interface Factory {
/**
* Prepares the instrumented type in order to allow the loading of the represented argument.
*
* @param instrumentedType The instrumented type.
* @return The prepared instrumented type.
*/
InstrumentedType prepare(InstrumentedType instrumentedType);
/**
* Creates any number of argument loaders for an instrumentation.
*
* @param instrumentedType The instrumented type.
* @param instrumentedMethod The instrumented method.
* @return Any number of argument loaders to supply for the method call.
*/
List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod);
}
/**
* An argument loader that loads the {@code null} value onto the operand stack.
*/
enum ForNullConstant implements ArgumentLoader, Factory {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
if (target.getType().isPrimitive()) {
throw new IllegalStateException("Cannot assign null to " + target);
}
return NullConstant.INSTANCE;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForNullConstant." + name();
}
}
/**
* An argument loader that assigns the {@code this} reference to a parameter.
*/
class ForThisReference implements ArgumentLoader {
/**
* The instrumented type.
*/
private final TypeDescription instrumentedType;
/**
* Creates an argument loader that supplies the {@code this} instance as an argument.
*
* @param instrumentedType The instrumented type.
*/
protected ForThisReference(TypeDescription instrumentedType) {
this.instrumentedType = instrumentedType;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
MethodVariableAccess.REFERENCE.loadOffset(0),
assigner.assign(instrumentedType.asGenericType(), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + instrumentedType + " to " + target);
}
return stackManipulation;
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
ForThisReference that = (ForThisReference) object;
return instrumentedType.equals(that.instrumentedType);
}
@Override
public int hashCode() {
return instrumentedType.hashCode();
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForThisReference{" +
"instrumentedType=" + instrumentedType +
'}';
}
/**
* A factory for an argument loader that supplies the {@code this} value as an argument.
*/
protected enum Factory implements ArgumentLoader.Factory {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
if (instrumentedMethod.isStatic()) {
throw new IllegalStateException(instrumentedMethod + " is static and cannot supply an invoker instance");
}
return Collections.singletonList(new ForThisReference(instrumentedType));
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForThisReference.Factory." + name();
}
}
}
/**
* Loads the instrumented type onto the operand stack.
*/
class ForInstrumentedType implements ArgumentLoader {
/**
* The instrumented type.
*/
private final TypeDescription instrumentedType;
/**
* Creates an argument loader for supporting the instrumented type as a type constant as an argument.
*
* @param instrumentedType The instrumented type.
*/
protected ForInstrumentedType(TypeDescription instrumentedType) {
this.instrumentedType = instrumentedType;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
ClassConstant.of(instrumentedType),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(Class.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign Class value to " + target);
}
return stackManipulation;
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
ForInstrumentedType that = (ForInstrumentedType) object;
return instrumentedType.equals(that.instrumentedType);
}
@Override
public int hashCode() {
return instrumentedType.hashCode();
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForInstrumentedType{" +
"instrumentedType=" + instrumentedType +
'}';
}
/**
* A factory for an argument loader that supplies the instrumented type as an argument.
*/
protected enum Factory implements ArgumentLoader.Factory {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(new ForInstrumentedType(instrumentedType));
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForInstrumentedType.Factory." + name();
}
}
}
/**
* Loads a parameter of the instrumented method onto the operand stack.
*/
class ForMethodParameter implements ArgumentLoader {
/**
* The index of the parameter to be loaded onto the operand stack.
*/
private final int index;
/**
* The instrumented method.
*/
private final MethodDescription instrumentedMethod;
/**
* Creates an argument loader for a parameter of the instrumented method.
*
* @param index The index of the parameter to be loaded onto the operand stack.
* @param instrumentedMethod The instrumented method.
*/
protected ForMethodParameter(int index, MethodDescription instrumentedMethod) {
this.index = index;
this.instrumentedMethod = instrumentedMethod;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
ParameterDescription parameterDescription = instrumentedMethod.getParameters().get(index);
StackManipulation stackManipulation = new StackManipulation.Compound(
MethodVariableAccess.of(parameterDescription.getType().asErasure()).loadOffset(parameterDescription.getOffset()),
assigner.assign(parameterDescription.getType(), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + parameterDescription + " to " + target + " for " + instrumentedMethod);
}
return stackManipulation;
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
ForMethodParameter that = (ForMethodParameter) object;
if (index != that.index) return false;
return instrumentedMethod.equals(that.instrumentedMethod);
}
@Override
public int hashCode() {
int result = index;
result = 31 * result + instrumentedMethod.hashCode();
return result;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForMethodParameter{" +
"index=" + index +
", instrumentedMethod=" + instrumentedMethod +
'}';
}
/**
* A factory for argument loaders that supplies all arguments of the instrumented method as arguments.
*/
protected enum OfInstrumentedMethod implements ArgumentLoader.Factory {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
List argumentLoaders = new ArrayList(instrumentedMethod.getParameters().size());
for (ParameterDescription parameterDescription : instrumentedMethod.getParameters()) {
argumentLoaders.add(new ForMethodParameter(parameterDescription.getIndex(), instrumentedMethod));
}
return argumentLoaders;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForMethodParameter.OfInstrumentedMethod." + name();
}
}
/**
* A factory for an argument loader that supplies a method parameter as an argument.
*/
protected static class Factory implements ArgumentLoader.Factory {
/**
* The index of the parameter to be loaded onto the operand stack.
*/
private final int index;
/**
* Creates a factory for an argument loader that supplies a method parameter as an argument.
*
* @param index The index of the parameter to supply.
*/
protected Factory(int index) {
this.index = index;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
if (index >= instrumentedMethod.getParameters().size()) {
throw new IllegalStateException(instrumentedMethod + " does not have a parameter with index " + index);
}
return Collections.singletonList(new ForMethodParameter(index, instrumentedMethod));
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
Factory factory = (Factory) object;
return index == factory.index;
}
@Override
public int hashCode() {
return index;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForMethodParameter.Factory{" +
"index=" + index +
'}';
}
}
}
/**
* Loads a value onto the operand stack that is stored in a static field.
*/
class ForStaticField implements ArgumentLoader {
/**
* The description of the field.
*/
private final FieldDescription fieldDescription;
/**
* Creates an argument loader that supplies the value of a static field as an argument.
*
* @param fieldDescription The description of the field.
*/
protected ForStaticField(FieldDescription fieldDescription) {
this.fieldDescription = fieldDescription;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
FieldAccess.forField(fieldDescription).getter(),
assigner.assign(fieldDescription.getType(), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + fieldDescription.getType() + " to " + target);
}
return stackManipulation;
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
ForStaticField that = (ForStaticField) object;
return fieldDescription.equals(that.fieldDescription);
}
@Override
public int hashCode() {
return fieldDescription.hashCode();
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForStaticField{" +
"fieldDescription=" + fieldDescription +
'}';
}
/**
* A factory that supplies the value of a static field as an argument.
*/
protected static class Factory implements ArgumentLoader.Factory {
/**
* The name prefix of the field to store the argument.
*/
private static final String FIELD_PREFIX = "methodCall";
/**
* The value to be stored in the field.
*/
private final Object value;
/**
* The name of the field.
*/
private final String fieldName;
/**
* Creates a factory that loads the value of a static field as an argument.
*
* @param value The value to supply as an argument.
*/
protected Factory(Object value) {
this.value = value;
fieldName = String.format("%s$%s", FIELD_PREFIX, RandomString.make());
}
/**
* Creates a factory for loading the supplied value as an argument, either stored as a static field or as a constant pool value.
*
* @param value The value to supply as an argument.
* @return An appropriate factory for an argument loader.
*/
protected static ArgumentLoader.Factory of(Object value) {
if (value == null) {
return ForNullConstant.INSTANCE;
} else if (value instanceof String) {
return new ForTextConstant((String) value);
} else if (value instanceof Boolean) {
return new ForBooleanConstant((Boolean) value);
} else if (value instanceof Byte) {
return new ForByteConstant((Byte) value);
} else if (value instanceof Short) {
return new ForShortConstant((Short) value);
} else if (value instanceof Character) {
return new ForCharacterConstant((Character) value);
} else if (value instanceof Integer) {
return new ForIntegerConstant((Integer) value);
} else if (value instanceof Long) {
return new ForLongConstant((Long) value);
} else if (value instanceof Float) {
return new ForFloatConstant((Float) value);
} else if (value instanceof Double) {
return new ForDoubleConstant((Double) value);
} else if (value instanceof Class) {
return new ForClassConstant(new TypeDescription.ForLoadedType((Class) value));
} else if (JavaType.METHOD_HANDLE.getTypeStub().isInstance(value)) {
return new ForJavaConstant(JavaConstant.MethodHandle.ofLoaded(value));
} else if (JavaType.METHOD_TYPE.getTypeStub().isInstance(value)) {
return new ForJavaConstant(JavaConstant.MethodType.ofLoaded(value));
} else if (value instanceof Enum) {
return new ForEnumerationValue(new EnumerationDescription.ForLoadedEnumeration((Enum) value));
} else {
return new ForStaticField.Factory(value);
}
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType
.withField(new FieldDescription.Token(fieldName,
Opcodes.ACC_SYNTHETIC | Opcodes.ACC_PUBLIC | Opcodes.ACC_STATIC,
new TypeDescription.Generic.OfNonGenericType.ForLoadedType(value.getClass())))
.withInitializer(new LoadedTypeInitializer.ForStaticField(fieldName, value));
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(new ForStaticField(instrumentedType.getDeclaredFields().filter(named(fieldName)).getOnly()));
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
Factory factory = (Factory) object;
return value.equals(factory.value);
}
@Override
public int hashCode() {
int result = value.hashCode();
return result;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForStaticField.Factory{" +
"value=" + value +
", fieldName='" + fieldName + '\'' +
'}';
}
}
}
/**
* Loads a value onto the operand stack that is stored in an instance field.
*/
class ForInstanceField implements ArgumentLoader {
/**
* The field for which the value is loaded.
*/
private final FieldDescription fieldDescription;
/**
* Creates an argument loader for an instance field.
*
* @param fieldDescription The field for which the value is loaded.
*/
protected ForInstanceField(FieldDescription fieldDescription) {
this.fieldDescription = fieldDescription;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
MethodVariableAccess.REFERENCE.loadOffset(0),
FieldAccess.forField(fieldDescription).getter(),
assigner.assign(fieldDescription.getType(), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + fieldDescription + " to " + target);
}
return stackManipulation;
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
ForInstanceField that = (ForInstanceField) object;
return fieldDescription.equals(that.fieldDescription);
}
@Override
public int hashCode() {
return fieldDescription.hashCode();
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForInstanceField{" +
"fieldDescription=" + fieldDescription +
'}';
}
/**
* A factory for an argument loader that loads the value of an instance field as an argument.
*/
protected static class Factory implements ArgumentLoader.Factory {
/**
* The type of the field.
*/
private final TypeDescription.Generic fieldType;
/**
* The name of the field.
*/
private final String fieldName;
/**
* Creates a factory for defining a field that is used as an argument.
*
* @param fieldType The type of the field.
* @param fieldName The name of the field.
*/
protected Factory(TypeDescription.Generic fieldType, String fieldName) {
this.fieldType = fieldType;
this.fieldName = fieldName;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
if (instrumentedType.isInterface()) {
throw new IllegalStateException("Cannot define non-static field '" + fieldName + "' for " + instrumentedType);
}
return instrumentedType.withField(new FieldDescription.Token(fieldName, Opcodes.ACC_SYNTHETIC | Opcodes.ACC_PUBLIC, fieldType));
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
if (instrumentedMethod.isStatic()) {
throw new IllegalStateException("Cannot access instance field from static " + instrumentedMethod);
}
return Collections.singletonList(new ForInstanceField(instrumentedType.getDeclaredFields().filter(named(fieldName)).getOnly()));
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
Factory factory = (Factory) object;
if (!fieldType.equals(factory.fieldType)) return false;
return fieldName.equals(factory.fieldName);
}
@Override
public int hashCode() {
int result = fieldType.hashCode();
result = 31 * result + fieldName.hashCode();
return result;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForInstanceField.Factory{" +
"fieldType=" + fieldType +
", fieldName='" + fieldName + '\'' +
'}';
}
}
}
/**
* Loads the value of an existing field onto the operand stack.
*/
class ForExistingField implements ArgumentLoader {
/**
* The field containing the loaded value.
*/
private final FieldDescription fieldDescription;
/**
* The instrumented method.
*/
private final MethodDescription instrumentedMethod;
/**
* Creates a new argument loader for loading an existing field.
*
* @param fieldDescription The field containing the loaded value.
* @param instrumentedMethod The instrumented method.
*/
protected ForExistingField(FieldDescription fieldDescription, MethodDescription instrumentedMethod) {
this.fieldDescription = fieldDescription;
this.instrumentedMethod = instrumentedMethod;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
if (!fieldDescription.isStatic() && instrumentedMethod.isStatic()) {
throw new IllegalStateException("Cannot access non-static " + fieldDescription + " from " + instrumentedMethod);
}
StackManipulation stackManipulation = new StackManipulation.Compound(
fieldDescription.isStatic()
? StackManipulation.Trivial.INSTANCE
: MethodVariableAccess.REFERENCE.loadOffset(0),
FieldAccess.forField(fieldDescription).getter(),
assigner.assign(fieldDescription.getType(), target.getType(), typing)
);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + fieldDescription + " to " + target);
}
return stackManipulation;
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
ForExistingField that = (ForExistingField) object;
if (!fieldDescription.equals(that.fieldDescription)) return false;
return instrumentedMethod.equals(that.instrumentedMethod);
}
@Override
public int hashCode() {
int result = fieldDescription.hashCode();
result = 31 * result + instrumentedMethod.hashCode();
return result;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForExistingField{" +
"fieldDescription=" + fieldDescription +
", instrumentedMethod=" + instrumentedMethod +
'}';
}
/**
* A factory for an argument loaded that loads the value of an existing field as an argument.
*/
protected static class Factory implements ArgumentLoader.Factory {
/**
* The name of the field.
*/
private final String fieldName;
/**
* Creates a new argument loader for an existing field.
*
* @param fieldName The name of the field.
*/
protected Factory(String fieldName) {
this.fieldName = fieldName;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
for (TypeDefinition currentType : instrumentedType) {
FieldList fieldList = currentType.getDeclaredFields().filter(named(fieldName).and(isVisibleTo(instrumentedType)));
if (fieldList.size() != 0) {
return Collections.singletonList(new ForExistingField(fieldList.getOnly(), instrumentedMethod));
}
}
throw new IllegalStateException(instrumentedType + " does not define a visible field " + fieldName);
}
@Override
public boolean equals(Object object) {
if (this == object) return true;
if (object == null || getClass() != object.getClass()) return false;
Factory factory = (Factory) object;
return fieldName.equals(factory.fieldName);
}
@Override
public int hashCode() {
return fieldName.hashCode();
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForExistingField.Factory{" +
"fieldName='" + fieldName + '\'' +
'}';
}
}
}
/**
* Loads a {@code boolean} value onto the operand stack.
*/
class ForBooleanConstant implements ArgumentLoader, Factory {
/**
* The {@code boolean} value to load onto the operand stack.
*/
private final boolean value;
/**
* Creates a new argument loader for a {@code boolean} value.
*
* @param value The {@code boolean} value to load onto the operand stack.
*/
protected ForBooleanConstant(boolean value) {
this.value = value;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
IntegerConstant.forValue(value),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(boolean.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign boolean value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& value == ((ForBooleanConstant) other).value;
}
@Override
public int hashCode() {
return (value ? 1 : 0);
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForBooleanConstant{value=" + value + '}';
}
}
/**
* Loads a {@code byte} value onto the operand stack.
*/
class ForByteConstant implements ArgumentLoader, Factory {
/**
* The {@code boolean} value to load onto the operand stack.
*/
private final byte value;
/**
* Creates a new argument loader for a {@code boolean} value.
*
* @param value The {@code boolean} value to load onto the operand stack.
*/
protected ForByteConstant(byte value) {
this.value = value;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
IntegerConstant.forValue(value),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(byte.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign byte value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& value == ((ForByteConstant) other).value;
}
@Override
public int hashCode() {
return value;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForByteConstant{value=" + value + '}';
}
}
/**
* Loads a {@code short} value onto the operand stack.
*/
class ForShortConstant implements ArgumentLoader, Factory {
/**
* The {@code short} value to load onto the operand stack.
*/
private final short value;
/**
* Creates a new argument loader for a {@code short} value.
*
* @param value The {@code short} value to load onto the operand stack.
*/
protected ForShortConstant(short value) {
this.value = value;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
IntegerConstant.forValue(value),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(short.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign short value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& value == ((ForShortConstant) other).value;
}
@Override
public int hashCode() {
return value;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForShortConstant{value=" + value + '}';
}
}
/**
* Loads a {@code char} value onto the operand stack.
*/
class ForCharacterConstant implements ArgumentLoader, Factory {
/**
* The {@code char} value to load onto the operand stack.
*/
private final char value;
/**
* Creates a new argument loader for a {@code char} value.
*
* @param value The {@code char} value to load onto the operand stack.
*/
protected ForCharacterConstant(char value) {
this.value = value;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
IntegerConstant.forValue(value),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(char.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign char value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& value == ((ForCharacterConstant) other).value;
}
@Override
public int hashCode() {
return value;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForCharacterConstant{value=" + value + '}';
}
}
/**
* Loads an {@code int} value onto the operand stack.
*/
class ForIntegerConstant implements ArgumentLoader, Factory {
/**
* The {@code int} value to load onto the operand stack.
*/
private final int value;
/**
* Creates a new argument loader for a {@code int} value.
*
* @param value The {@code int} value to load onto the operand stack.
*/
protected ForIntegerConstant(int value) {
this.value = value;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
IntegerConstant.forValue(value),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(int.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign integer value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& value == ((ForIntegerConstant) other).value;
}
@Override
public int hashCode() {
return value;
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForIntegerConstant{value=" + value + '}';
}
}
/**
* Loads a {@code long} value onto the operand stack.
*/
class ForLongConstant implements ArgumentLoader, Factory {
/**
* The {@code long} value to load onto the operand stack.
*/
private final long value;
/**
* Creates a new argument loader for a {@code long} value.
*
* @param value The {@code long} value to load onto the operand stack.
*/
protected ForLongConstant(long value) {
this.value = value;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
LongConstant.forValue(value),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(long.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign long value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& value == ((ForLongConstant) other).value;
}
@Override
public int hashCode() {
return (int) (value ^ (value >>> 32));
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForLongConstant{" +
"value=" + value +
'}';
}
}
/**
* Loads a {@code float} value onto the operand stack.
*/
class ForFloatConstant implements ArgumentLoader, Factory {
/**
* The {@code float} value to load onto the operand stack.
*/
private final float value;
/**
* Creates a new argument loader for a {@code float} value.
*
* @param value The {@code float} value to load onto the operand stack.
*/
protected ForFloatConstant(float value) {
this.value = value;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
FloatConstant.forValue(value),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(float.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign float value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& Float.compare(((ForFloatConstant) other).value, value) == 0;
}
@Override
public int hashCode() {
return (value != +0.0f ? Float.floatToIntBits(value) : 0);
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForFloatConstant{value=" + value + '}';
}
}
/**
* Loads a {@code double} value onto the operand stack.
*/
class ForDoubleConstant implements ArgumentLoader, Factory {
/**
* The {@code double} value to load onto the operand stack.
*/
private final double value;
/**
* Creates a new argument loader for a {@code double} value.
*
* @param value The {@code double} value to load onto the operand stack.
*/
protected ForDoubleConstant(double value) {
this.value = value;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
DoubleConstant.forValue(value),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(double.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign double value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& Double.compare(((ForDoubleConstant) other).value, value) == 0;
}
@Override
public int hashCode() {
long temp = Double.doubleToLongBits(value);
return (int) (temp ^ (temp >>> 32));
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForDoubleConstant{value=" + value + '}';
}
}
/**
* Loads a {@link java.lang.String} value onto the operand stack.
*/
class ForTextConstant implements ArgumentLoader, Factory {
/**
* The {@link java.lang.String} value to load onto the operand stack.
*/
private final String value;
/**
* Creates a new argument loader for a {@link java.lang.String} value.
*
* @param value The {@link java.lang.String} value to load onto the operand stack.
*/
protected ForTextConstant(String value) {
this.value = value;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
new TextConstant(value),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(String.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign String value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& value.equals(((ForTextConstant) other).value);
}
@Override
public int hashCode() {
return value.hashCode();
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForTextConstant{" +
"value='" + value + '\'' +
'}';
}
}
/**
* Loads a {@link java.lang.Class} value onto the operand stack.
*/
class ForClassConstant implements ArgumentLoader, Factory {
/**
* The type to load onto the operand stack.
*/
private final TypeDescription typeDescription;
/**
* Creates a new class constant representation.
*
* @param typeDescription The type to represent.
*/
protected ForClassConstant(TypeDescription typeDescription) {
this.typeDescription = typeDescription;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
ClassConstant.of(typeDescription),
assigner.assign(new TypeDescription.Generic.OfNonGenericType.ForLoadedType(Class.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign class value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& typeDescription.equals(((ForClassConstant) other).typeDescription);
}
@Override
public int hashCode() {
return typeDescription.hashCode();
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForClassConstant{" +
"typeDescription=" + typeDescription +
'}';
}
}
/**
* An argument loader that loads an enumeration constant.
*/
class ForEnumerationValue implements ArgumentLoader, Factory {
/**
* The enumeration to describe.
*/
private final EnumerationDescription enumerationDescription;
/**
* Creates a new argument loader for an enumeration constant.
*
* @param enumerationDescription The enumeration to describe.
*/
protected ForEnumerationValue(EnumerationDescription enumerationDescription) {
this.enumerationDescription = enumerationDescription;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
FieldAccess.forEnumeration(enumerationDescription),
assigner.assign(enumerationDescription.getEnumerationType().asGenericType(), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + enumerationDescription.getEnumerationType() + " value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& enumerationDescription.equals(((ForEnumerationValue) other).enumerationDescription);
}
@Override
public int hashCode() {
return enumerationDescription.hashCode();
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForEnumerationValue{" +
"enumerationDescription=" + enumerationDescription +
'}';
}
}
/**
* Loads a Java instance onto the operand stack.
*/
class ForJavaConstant implements ArgumentLoader, Factory {
/**
* The Java instance to load onto the operand stack.
*/
private final JavaConstant javaConstant;
/**
* Creates a new argument loader for a Java instance.
*
* @param javaConstant The Java instance to load as an argument.
*/
public ForJavaConstant(JavaConstant javaConstant) {
this.javaConstant = javaConstant;
}
@Override
public StackManipulation resolve(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
javaConstant.asStackManipulation(),
assigner.assign(javaConstant.getType().asGenericType(), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign Class value to " + target);
}
return stackManipulation;
}
@Override
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
@Override
public List make(TypeDescription instrumentedType, MethodDescription instrumentedMethod) {
return Collections.singletonList(this);
}
@Override
public boolean equals(Object other) {
return this == other || !(other == null || getClass() != other.getClass())
&& javaConstant.equals(((ForJavaConstant) other).javaConstant);
}
@Override
public int hashCode() {
return javaConstant.hashCode();
}
@Override
public String toString() {
return "MethodCall.ArgumentLoader.ForJavaConstant{" +
"javaConstant=" + javaConstant +
'}';
}
}
}
/**
* A method invoker is responsible for creating a method invocation that is to be applied by a
* {@link com.fitbur.mockito.bytebuddy.implementation.MethodCall}.
*/
protected interface MethodInvoker {
/**
* Invokes the method.
*
* @param invokedMethod The method to be invoked.
* @param implementationTarget The implementation target of the instrumented instance.
* @return A stack manipulation that represents the method invocation.
*/
StackManipulation invoke(MethodDescription invokedMethod, Target implementationTarget);
/**
* Applies a contextual invocation of the provided method, i.e. a static invocation for static methods,
* a special invocation for constructors and private methods and a virtual invocation for any other method.
*/
enum ForContextualInvocation implements MethodInvoker {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public StackManipulation invoke(MethodDescription invokedMethod, Target implementationTarget) {
if (invokedMethod.isVirtual() && !invokedMethod.isInvokableOn(implementationTarget.getInstrumentedType())) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + implementationTarget.getInstrumentedType());
} else if (!invokedMethod.isVisibleTo(implementationTarget.getInstrumentedType())) {
throw new IllegalStateException(implementationTarget.getInstrumentedType() + " cannot see " + invokedMethod);
}
return invokedMethod.isVirtual()
? MethodInvocation.invoke(invokedMethod).virtual(implementationTarget.getInstrumentedType())
: MethodInvocation.invoke(invokedMethod);
}
@Override
public String toString() {
return "MethodCall.MethodInvoker.ForContextualInvocation." + name();
}
}
/**
* Applies a virtual invocation on a given type.
*/
class ForVirtualInvocation implements MethodInvoker {
/**
* The type description to virtually invoke the method upon.
*/
private final TypeDescription typeDescription;
/**
* Creates a new method invoking for a virtual method invocation.
*
* @param typeDescription The type description to virtually invoke the method upon.
*/
protected ForVirtualInvocation(TypeDescription typeDescription) {
this.typeDescription = typeDescription;
}
/**
* Creates a new method invoking for a virtual method invocation.
*
* @param type The type to virtually invoke the method upon.
*/
protected ForVirtualInvocation(Class type) {
this(new TypeDescription.ForLoadedType(type));
}
@Override
public StackManipulation invoke(MethodDescription invokedMethod, Target implementationTarget) {
if (!invokedMethod.isVirtual()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " virtually");
} else if (!invokedMethod.isInvokableOn(typeDescription.asErasure())) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + typeDescription);
} else if (!typeDescription.asErasure().isVisibleTo(implementationTarget.getInstrumentedType())) {
throw new IllegalStateException(typeDescription + " is not visible to " + implementationTarget.getInstrumentedType());
}
return MethodInvocation.invoke(invokedMethod).virtual(typeDescription.asErasure());
}
@Override
public boolean equals(Object other) {
if (this == other) return true;
if (other == null || getClass() != other.getClass()) return false;
ForVirtualInvocation that = (ForVirtualInvocation) other;
return typeDescription.equals(that.typeDescription);
}
@Override
public int hashCode() {
return typeDescription.hashCode();
}
@Override
public String toString() {
return "MethodCall.MethodInvoker.ForVirtualInvocation{" +
"typeDescription=" + typeDescription +
'}';
}
/**
* A method invoker for a virtual method that uses an implicit target type.
*/
public enum WithImplicitType implements MethodInvoker {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public StackManipulation invoke(MethodDescription invokedMethod, Target implementationTarget) {
if (!invokedMethod.isVirtual()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " virtually");
}
return MethodInvocation.invoke(invokedMethod);
}
@Override
public String toString() {
return "MethodCall.MethodInvoker.ForVirtualInvocation.WithImplicitType." + name();
}
}
}
/**
* Applies a super method invocation of the provided method.
*/
enum ForSuperMethodInvocation implements MethodInvoker {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public StackManipulation invoke(MethodDescription invokedMethod, Target implementationTarget) {
if (implementationTarget.getInstrumentedType().getSuperClass() == null) {
throw new IllegalStateException("Cannot invoke super method for " + implementationTarget.getInstrumentedType());
} else if (!invokedMethod.isInvokableOn(implementationTarget.getOriginType().asErasure())) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " as super method of " + implementationTarget.getInstrumentedType());
}
StackManipulation stackManipulation = implementationTarget.invokeDominant(invokedMethod.asSignatureToken());
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " as a super method");
}
return stackManipulation;
}
@Override
public String toString() {
return "MethodCall.MethodInvoker.ForSuperMethodInvocation." + name();
}
}
/**
* Invokes a method as a Java 8 default method.
*/
enum ForDefaultMethodInvocation implements MethodInvoker {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public StackManipulation invoke(MethodDescription invokedMethod, Target implementationTarget) {
if (!invokedMethod.isInvokableOn(implementationTarget.getInstrumentedType())) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " as default method of " + implementationTarget.getInstrumentedType());
}
StackManipulation stackManipulation = implementationTarget.invokeDefault(invokedMethod.getDeclaringType().asErasure(), invokedMethod.asSignatureToken());
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + implementationTarget.getInstrumentedType());
}
return stackManipulation;
}
@Override
public String toString() {
return "MethodCall.MethodInvoker.ForDefaultMethodInvocation." + name();
}
}
}
/**
* A termination handler is responsible to handle the return value of a method that is invoked via a
* {@link com.fitbur.mockito.bytebuddy.implementation.MethodCall}.
*/
protected interface TerminationHandler {
/**
* Returns a stack manipulation that handles the method return.
*
* @param invokedMethod The method that was invoked by the method call.
* @param instrumentedMethod The method being intercepted.
* @param assigner The assigner to be used.
* @param typing Indicates if dynamic type castings should be attempted for incompatible assignments.
* @return A stack manipulation that handles the method return.
*/
StackManipulation resolve(MethodDescription invokedMethod,
MethodDescription instrumentedMethod,
Assigner assigner,
Assigner.Typing typing);
/**
* Returns the return value of the method call from the intercepted method.
*/
enum ForMethodReturn implements TerminationHandler {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public StackManipulation resolve(MethodDescription invokedMethod, MethodDescription instrumentedMethod, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = assigner.assign(invokedMethod.isConstructor()
? invokedMethod.getDeclaringType().asGenericType()
: invokedMethod.getReturnType(), instrumentedMethod.getReturnType(), typing);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot return " + invokedMethod.getReturnType() + " from " + instrumentedMethod);
}
return new StackManipulation.Compound(stackManipulation,
MethodReturn.returning(instrumentedMethod.getReturnType().asErasure()));
}
@Override
public String toString() {
return "MethodCall.TerminationHandler.ForMethodReturn." + name();
}
}
/**
* Drops the return value of the called method from the operand stack without returning from the intercepted
* method.
*/
enum ForChainedInvocation implements TerminationHandler {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public StackManipulation resolve(MethodDescription invokedMethod, MethodDescription instrumentedMethod, Assigner assigner, Assigner.Typing typing) {
return Removal.pop(invokedMethod.isConstructor()
? invokedMethod.getDeclaringType().asErasure()
: invokedMethod.getReturnType().asErasure());
}
@Override
public String toString() {
return "MethodCall.TerminationHandler.ForChainedInvocation." + name();
}
}
}
/**
* Represents a {@link com.fitbur.mockito.bytebuddy.implementation.MethodCall} that invokes a method without specifying
* an invocation method. Some methods can for example be invoked both virtually or as a super method invocation.
* Similarly, interface methods can be invoked virtually or as an explicit invocation of a default method. If
* no explicit invocation type is set, a method is always invoked virtually unless the method
* represents a static methods or a constructor.
*/
public static class WithoutSpecifiedTarget extends MethodCall {
/**
* Creates a new method call without a specified target.
*
* @param methodLocator The method locator to use.
*/
protected WithoutSpecifiedTarget(MethodLocator methodLocator) {
super(methodLocator,
TargetHandler.ForSelfOrStaticInvocation.INSTANCE,
Collections.emptyList(),
MethodInvoker.ForContextualInvocation.INSTANCE,
TerminationHandler.ForMethodReturn.INSTANCE,
Assigner.DEFAULT,
Assigner.Typing.STATIC);
}
/**
* Invokes the specified method on the given instance.
*
* @param target The object on which the method is to be invoked upon.
* @return A method call that invokes the provided method on the given object.
*/
public MethodCall on(Object target) {
return new MethodCall(methodLocator,
new TargetHandler.ForStaticField(target),
argumentLoaders,
new MethodInvoker.ForVirtualInvocation(target.getClass()),
TerminationHandler.ForMethodReturn.INSTANCE,
assigner,
typing);
}
/**
* Invokes the specified method on the instrumented method's argument of the given index.
*
* @param index The index of the method's argument on which the specified method should be invoked.
* @return Amethod call that invokes the provided method on the given method argument.
*/
public MethodCall onArgument(int index) {
if (index < 0) {
throw new IllegalArgumentException("An argument index cannot be negative: " + index);
}
return new MethodCall(methodLocator,
new TargetHandler.ForMethodParameter(index),
argumentLoaders,
MethodInvoker.ForVirtualInvocation.WithImplicitType.INSTANCE,
TerminationHandler.ForMethodReturn.INSTANCE,
assigner,
typing);
}
/**
* Invokes the given method on an instance that is stored in an instance field. This field's value needs
* to be set by the user such that the method call does not throw a {@link java.lang.NullPointerException}.
*
* @param type The type of the field.
* @param fieldName The name of the field.
* @return A method call that invokes the given method on an instance that is read from an instance field.
*/
public MethodCall onInstanceField(Type type, String fieldName) {
return onInstanceField(TypeDefinition.Sort.describe(type), fieldName);
}
/**
* Invokes the given method on an instance that is stored in an instance field. This field's value needs
* to be set by the user such that the method call does not throw a {@link java.lang.NullPointerException}.
*
* @param typeDescription The type of the field.
* @param fieldName The name of the field.
* @return A method call that invokes the given method on an instance that is read from an instance field.
*/
public MethodCall onInstanceField(TypeDescription.Generic typeDescription, String fieldName) {
return new MethodCall(methodLocator,
new TargetHandler.ForInstanceField(fieldName, typeDescription),
argumentLoaders,
new MethodInvoker.ForVirtualInvocation(typeDescription.asErasure()),
TerminationHandler.ForMethodReturn.INSTANCE,
assigner,
typing);
}
/**
* Invokes the given method by a super method invocation on the instance of the instrumented type.
* Note that the super method is resolved depending on the type of implementation when this method is called.
* In case that a subclass is created, the super type is invoked. If a type is rebased, the rebased method
* is invoked if such a method exists.
*
* @return A method call where the given method is invoked as a super method invocation.
*/
public MethodCall onSuper() {
return new MethodCall(methodLocator,
TargetHandler.ForSelfOrStaticInvocation.INSTANCE,
argumentLoaders,
MethodInvoker.ForSuperMethodInvocation.INSTANCE,
TerminationHandler.ForMethodReturn.INSTANCE,
assigner,
typing);
}
/**
* Invokes the given method by a Java 8default method invocation on the instance of the instrumented type.
*
* @return A method call where the given method is invoked as a super method invocation.
*/
public MethodCall onDefault() {
return new MethodCall(methodLocator,
TargetHandler.ForSelfOrStaticInvocation.INSTANCE,
argumentLoaders,
MethodInvoker.ForDefaultMethodInvocation.INSTANCE,
TerminationHandler.ForMethodReturn.INSTANCE,
assigner,
typing);
}
@Override
public String toString() {
return "MethodCall.WithoutSpecifiedTarget{" +
"methodLocator=" + methodLocator +
", targetHandler=" + targetHandler +
", argumentLoaders=" + argumentLoaders +
", methodInvoker=" + methodInvoker +
", terminationHandler=" + terminationHandler +
", assigner=" + assigner +
", typing=" + typing +
'}';
}
}
/**
* The appender being used to implement a {@link com.fitbur.mockito.bytebuddy.implementation.MethodCall}.
*/
protected class Appender implements ByteCodeAppender {
/**
* The implementation target of the current implementation.
*/
private final Target implementationTarget;
/**
* Creates a new appender.
*
* @param implementationTarget The implementation target of the current implementation.
*/
protected Appender(Target implementationTarget) {
this.implementationTarget = implementationTarget;
}
@Override
public Size apply(MethodVisitor methodVisitor, Context implementationContext, MethodDescription instrumentedMethod) {
MethodDescription invokedMethod = methodLocator.resolve(instrumentedMethod);
List argumentLoaders = new ArrayList(MethodCall.this.argumentLoaders.size());
for (ArgumentLoader.Factory argumentLoader : MethodCall.this.argumentLoaders) {
argumentLoaders.addAll(argumentLoader.make(implementationTarget.getInstrumentedType(), instrumentedMethod));
}
ParameterList parameters = invokedMethod.getParameters();
Iterator parameterIterator = parameters.iterator();
if (parameters.size() != argumentLoaders.size()) {
throw new IllegalStateException(invokedMethod + " does not take " + argumentLoaders.size() + " arguments");
}
List argumentInstructions = new ArrayList(argumentLoaders.size());
for (ArgumentLoader argumentLoader : argumentLoaders) {
argumentInstructions.add(argumentLoader.resolve(parameterIterator.next(), assigner, typing));
}
StackManipulation.Size size = new StackManipulation.Compound(
targetHandler.resolve(invokedMethod, instrumentedMethod, implementationTarget.getInstrumentedType(), assigner, typing),
new StackManipulation.Compound(argumentInstructions),
methodInvoker.invoke(invokedMethod, implementationTarget),
terminationHandler.resolve(invokedMethod, instrumentedMethod, assigner, typing)
).apply(methodVisitor, implementationContext);
return new Size(size.getMaximalSize(), instrumentedMethod.getStackSize());
}
/**
* Returns the outer instance.
*
* @return The outer instance.
*/
private MethodCall getOuter() {
return MethodCall.this;
}
@Override
public boolean equals(Object other) {
if (this == other) return true;
if (other == null || getClass() != other.getClass()) return false;
Appender appender = (Appender) other;
return implementationTarget.equals(appender.implementationTarget)
&& MethodCall.this.equals(appender.getOuter());
}
@Override
public int hashCode() {
return implementationTarget.hashCode() + 31 * MethodCall.this.hashCode();
}
@Override
public String toString() {
return "MethodCall.Appender{" +
"methodCall=" + MethodCall.this +
", implementationTarget=" + implementationTarget +
'}';
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy