net.bytebuddy.implementation.MethodCall Maven / Gradle / Ivy
The newest version!
/*
* Copyright 2014 - Present Rafael Winterhalter
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package net.bytebuddy.implementation;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import net.bytebuddy.build.HashCodeAndEqualsPlugin;
import net.bytebuddy.description.enumeration.EnumerationDescription;
import net.bytebuddy.description.field.FieldDescription;
import net.bytebuddy.description.field.FieldList;
import net.bytebuddy.description.method.MethodDescription;
import net.bytebuddy.description.method.ParameterDescription;
import net.bytebuddy.description.method.ParameterList;
import net.bytebuddy.description.type.TypeDefinition;
import net.bytebuddy.description.type.TypeDescription;
import net.bytebuddy.dynamic.scaffold.FieldLocator;
import net.bytebuddy.dynamic.scaffold.InstrumentedType;
import net.bytebuddy.dynamic.scaffold.MethodGraph;
import net.bytebuddy.implementation.bytecode.*;
import net.bytebuddy.implementation.bytecode.assign.Assigner;
import net.bytebuddy.implementation.bytecode.collection.ArrayAccess;
import net.bytebuddy.implementation.bytecode.collection.ArrayFactory;
import net.bytebuddy.implementation.bytecode.constant.*;
import net.bytebuddy.implementation.bytecode.member.FieldAccess;
import net.bytebuddy.implementation.bytecode.member.MethodInvocation;
import net.bytebuddy.implementation.bytecode.member.MethodReturn;
import net.bytebuddy.implementation.bytecode.member.MethodVariableAccess;
import net.bytebuddy.matcher.ElementMatcher;
import net.bytebuddy.matcher.ElementMatchers;
import net.bytebuddy.utility.*;
import net.bytebuddy.utility.nullability.MaybeNull;
import net.bytebuddy.jar.asm.MethodVisitor;
import net.bytebuddy.jar.asm.Opcodes;
import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.lang.reflect.Type;
import java.util.*;
import java.util.concurrent.Callable;
import static net.bytebuddy.matcher.ElementMatchers.*;
/**
* This {@link Implementation} allows the invocation of a specified method while
* providing explicit arguments to this method.
*/
@HashCodeAndEqualsPlugin.Enhance
public class MethodCall implements Implementation.Composable {
/**
* The method locator to use.
*/
protected final MethodLocator.Factory methodLocator;
/**
* The target handler to use.
*/
protected final TargetHandler.Factory 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.Factory methodInvoker;
/**
* The termination handler to use.
*/
protected final TerminationHandler.Factory 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.Factory methodLocator,
TargetHandler.Factory targetHandler,
List argumentLoaders,
MethodInvoker.Factory methodInvoker,
TerminationHandler.Factory 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.
*
*
* Important: A constructor invocation can only be applied within another constructor to invoke the super constructor or an auxiliary
* constructor. To construct a new instance, use {@link MethodCall#construct(Constructor)}.
*
*
* @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.
*
*
* Important: A constructor invocation can only be applied within another constructor to invoke the super constructor or an auxiliary
* constructor. To construct a new instance, use {@link MethodCall#construct(MethodDescription)}.
*
*
* @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 unique virtual method or constructor of the instrumented type that is matched by the specified matcher.
*
* @param matcher The matcher to identify the method to invoke.
* @return A method call for the uniquely identified method.
*/
@SuppressWarnings("overloads")
public static WithoutSpecifiedTarget invoke(ElementMatcher matcher) {
return invoke(matcher, MethodGraph.Compiler.DEFAULT);
}
/**
* Invokes a unique virtual method or constructor of the instrumented type that is matched by the specified matcher.
*
* @param matcher The matcher to identify the method to invoke.
* @param methodGraphCompiler The method graph compiler to use.
* @return A method call for the uniquely identified method.
*/
public static WithoutSpecifiedTarget invoke(ElementMatcher matcher, MethodGraph.Compiler methodGraphCompiler) {
return invoke(new MethodLocator.ForElementMatcher.Factory(matcher, methodGraphCompiler));
}
/**
* 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.
*/
@SuppressWarnings("overloads")
public static WithoutSpecifiedTarget invoke(MethodLocator.Factory methodLocator) {
return new WithoutSpecifiedTarget(methodLocator);
}
/**
* Invokes the instrumented method recursively. Invoking this method on the same instance causes a {@link StackOverflowError} due to
* infinite recursion.
*
* @return A method call that invokes the method being instrumented.
*/
public static WithoutSpecifiedTarget invokeSelf() {
return new WithoutSpecifiedTarget(MethodLocator.ForInstrumentedMethod.INSTANCE);
}
/**
* Invokes the instrumented method as a super method call on the instance itself. This is a shortcut for {@code invokeSelf().onSuper()}.
*
* @return A method call that invokes the method being instrumented as a super method call.
*/
public static MethodCall invokeSuper() {
return invokeSelf().onSuper();
}
/**
* 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) {
try {
return invoke(Callable.class.getMethod("call")).on(callable, Callable.class).withAssigner(Assigner.DEFAULT, Assigner.Typing.DYNAMIC);
} catch (NoSuchMethodException exception) {
throw new IllegalStateException("Could not locate Callable::call method", exception);
}
}
/**
* 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) {
try {
return invoke(Runnable.class.getMethod("run")).on(runnable, Runnable.class).withAssigner(Assigner.DEFAULT, Assigner.Typing.DYNAMIC);
} catch (NoSuchMethodException exception) {
throw new IllegalStateException("Could not locate Runnable::run method", exception);
}
}
/**
* 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.Factory.INSTANCE,
Collections.emptyList(),
MethodInvoker.ForContextualInvocation.Factory.INSTANCE,
TerminationHandler.Simple.RETURNING,
Assigner.DEFAULT,
Assigner.Typing.STATIC);
}
/**
* 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}, method handles, types,
* method types as well as instances of {@link TypeDescription} or {@link JavaConstant} 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.ForStackManipulation.of(anArgument));
}
return with(argumentLoaders);
}
/**
* 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.ForStackManipulation(ClassConstant.of(aTypeDescription), Class.class));
}
return with(argumentLoaders);
}
/**
* 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.ForStackManipulation(FieldAccess.forEnumeration(anEnumerationDescription), anEnumerationDescription.getEnumerationType()));
}
return with(argumentLoaders);
}
/**
* 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 constant The constants to provide as arguments.
* @return A method call that hands the provided arguments to the invoked method.
*/
public MethodCall with(ConstantValue... constant) {
List argumentLoaders = new ArrayList(constant.length);
for (ConstantValue aConstant : constant) {
argumentLoaders.add(new ArgumentLoader.ForStackManipulation(aConstant.toStackManipulation(), aConstant.getTypeDescription()));
}
return with(argumentLoaders);
}
/**
* 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 constant The constants to provide as arguments.
* @return A method call that hands the provided arguments to the invoked method.
*/
public MethodCall with(JavaConstant... constant) {
return with((ConstantValue[]) constant);
}
/**
* 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.ForInstance.Factory(anArgument));
}
return with(argumentLoaders);
}
/**
* 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 with(argumentLoaders);
}
/**
* 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 of the instrumented method to the invoked method.
*/
public MethodCall withAllArguments() {
return with(ArgumentLoader.ForMethodParameter.OfInstrumentedMethod.INSTANCE);
}
/**
* Adds an array containing all arguments of the instrumented method to this method call.
*
* @return A method call that adds an array containing all arguments of the instrumented method to the invoked method.
*/
public MethodCall withArgumentArray() {
return with(ArgumentLoader.ForMethodParameterArray.ForInstrumentedMethod.INSTANCE);
}
/**
*
* Creates a method call where the parameter with {@code index} is expected to be an array and where each element of the array
* is expected to represent an argument for the method being invoked.
*
*
* Note: This is typically used in combination with dynamic type assignments which is activated via
* {@link MethodCall#withAssigner(Assigner, Assigner.Typing)} using a {@link Assigner.Typing#DYNAMIC}.
*
*
* @param index The index of the parameter.
* @return A method call that loads {@code size} elements from the array handed to the instrumented method as argument {@code index}.
*/
public MethodCall withArgumentArrayElements(int index) {
if (index < 0) {
throw new IllegalArgumentException("A parameter index cannot be negative: " + index);
}
return with(new ArgumentLoader.ForMethodParameterArrayElement.OfInvokedMethod(index));
}
/**
*
* Creates a method call where the parameter with {@code index} is expected to be an array and where {@code size} elements are loaded
* from the array as arguments for the invoked method.
*
*
* Note: This is typically used in combination with dynamic type assignments which is activated via
* {@link MethodCall#withAssigner(Assigner, Assigner.Typing)} using a {@link Assigner.Typing#DYNAMIC}.
*
*
* @param index The index of the parameter.
* @param size The amount of elements to load from the array.
* @return A method call that loads {@code size} elements from the array handed to the instrumented method as argument {@code index}.
*/
public MethodCall withArgumentArrayElements(int index, int size) {
return withArgumentArrayElements(index, 0, size);
}
/**
*
* Creates a method call where the parameter with {@code index} is expected to be an array and where {@code size} elements are loaded
* from the array as arguments for the invoked method. The first element is loaded from index {@code start}.
*
*
* Note: This is typically used in combination with dynamic type assignments which is activated via
* {@link MethodCall#withAssigner(Assigner, Assigner.Typing)} using a {@link Assigner.Typing#DYNAMIC}.
*
*
* @param index The index of the parameter.
* @param start The first array index to consider.
* @param size The amount of elements to load from the array with increasing index from {@code start}.
* @return A method call that loads {@code size} elements from the array handed to the instrumented method as argument {@code index}.
*/
public MethodCall withArgumentArrayElements(int index, int start, int size) {
if (index < 0) {
throw new IllegalArgumentException("A parameter index cannot be negative: " + index);
} else if (start < 0) {
throw new IllegalArgumentException("An array index cannot be negative: " + start);
} else if (size == 0) {
return this;
} else if (size < 0) {
throw new IllegalArgumentException("Size cannot be negative: " + size);
}
List argumentLoaders = new ArrayList(size);
for (int position = 0; position < size; position++) {
argumentLoaders.add(new ArgumentLoader.ForMethodParameterArrayElement.OfParameter(index, start + position));
}
return with(argumentLoaders);
}
/**
* 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 with(ArgumentLoader.ForThisReference.Factory.INSTANCE);
}
/**
* 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 with(ArgumentLoader.ForInstrumentedType.Factory.INSTANCE);
}
/**
* Defines a method call which fetches a value from a list of existing fields.
*
* @param name The names of the fields.
* @return A method call which assigns the next parameters to the values of the given fields.
*/
public MethodCall withField(String... name) {
return withField(FieldLocator.ForClassHierarchy.Factory.INSTANCE, name);
}
/**
* Defines a method call which fetches a value from a list of existing fields.
*
* @param fieldLocatorFactory The field locator factory to use.
* @param name The names of the fields.
* @return A method call which assigns the next parameters to the values of the given fields.
*/
public MethodCall withField(FieldLocator.Factory fieldLocatorFactory, String... name) {
List argumentLoaders = new ArrayList(name.length);
for (String aName : name) {
argumentLoaders.add(new ArgumentLoader.ForField.Factory(aName, fieldLocatorFactory));
}
return with(argumentLoaders);
}
/**
* Defines a method call which fetches a value from a method call.
*
* @param methodCall The method call to use.
* @return A method call which assigns the parameter to the result of the given method call.
*/
public MethodCall withMethodCall(MethodCall methodCall) {
return with(new ArgumentLoader.ForMethodCall.Factory(methodCall));
}
/**
* Adds a stack manipulation as an assignment to the next parameter.
*
* @param stackManipulation The stack manipulation loading the value.
* @param type The type of the argument being loaded.
* @return A method call that adds the stack manipulation as the next argument to the invoked method.
*/
public MethodCall with(StackManipulation stackManipulation, Type type) {
return with(stackManipulation, TypeDefinition.Sort.describe(type));
}
/**
* Adds a stack manipulation as an assignment to the next parameter.
*
* @param stackManipulation The stack manipulation loading the value.
* @param typeDefinition The type of the argument being loaded.
* @return A method call that adds the stack manipulation as the next argument to the invoked method.
*/
public MethodCall with(StackManipulation stackManipulation, TypeDefinition typeDefinition) {
return with(new ArgumentLoader.ForStackManipulation(stackManipulation, typeDefinition));
}
/**
* Defines a method call that resolves arguments by the supplied argument loader factories.
*
* @param argumentLoader The argument loaders to apply to the subsequent arguments of the
* @return A method call that adds the arguments of the supplied argument loaders to the invoked method.
*/
public MethodCall with(ArgumentLoader.Factory... argumentLoader) {
return with(Arrays.asList(argumentLoader));
}
/**
* Defines a method call that resolves arguments by the supplied argument loader factories.
*
* @param argumentLoaders The argument loaders to apply to the subsequent arguments of the
* @return A method call that adds the arguments of the supplied argument loaders to the invoked method.
*/
public MethodCall with(List argumentLoaders) {
return new MethodCall(methodLocator,
targetHandler,
CompoundList.of(this.argumentLoaders, argumentLoaders),
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* Sets the result of the method call as a value of the specified field. If the instrumented method does not
* return {@code void}, this instrumentation must be chained with another instrumentation.
*
* @param field The field to set.
* @return A new instance of this method call that sets the resulting value as the specified field's value.
*/
public FieldSetting setsField(Field field) {
return setsField(new FieldDescription.ForLoadedField(field));
}
/**
* Sets the result of the method call as a value of the specified field. If the instrumented method does not
* return {@code void}, this instrumentation must be chained with another instrumentation.
*
* @param fieldDescription The field to set.
* @return A new instance of this method call that sets the resulting value as the specified field's value.
*/
public FieldSetting setsField(FieldDescription fieldDescription) {
return new FieldSetting(new MethodCall(methodLocator,
targetHandler,
argumentLoaders,
methodInvoker,
new TerminationHandler.FieldSetting.Explicit(fieldDescription),
assigner,
typing));
}
/**
* Sets the result of the method call as a value of the specified field. If the instrumented method does not
* return {@code void}, this instrumentation must be chained with another instrumentation.
*
* @param matcher A matcher that locates a field in the instrumented type's hierarchy.
* @return A new instance of this method call that sets the resulting value as the specified field's value.
*/
public FieldSetting setsField(ElementMatcher matcher) {
return new FieldSetting(new MethodCall(methodLocator,
targetHandler,
argumentLoaders,
methodInvoker,
new TerminationHandler.FieldSetting.Implicit(matcher),
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 net.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 Composable withAssigner(Assigner assigner, Assigner.Typing typing) {
return new MethodCall(methodLocator,
targetHandler,
argumentLoaders,
methodInvoker,
terminationHandler,
assigner,
typing);
}
/**
* {@inheritDoc}
*/
public Implementation andThen(Implementation implementation) {
return new Implementation.Compound(new MethodCall(methodLocator,
targetHandler,
argumentLoaders,
methodInvoker,
TerminationHandler.Simple.DROPPING,
assigner,
typing), implementation);
}
/**
* {@inheritDoc}
*/
public Composable andThen(Composable implementation) {
return new Implementation.Compound.Composable(new MethodCall(methodLocator,
targetHandler,
argumentLoaders,
methodInvoker,
TerminationHandler.Simple.DROPPING,
assigner,
typing), implementation);
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
for (InstrumentedType.Prepareable prepareable : argumentLoaders) {
instrumentedType = prepareable.prepare(instrumentedType);
}
return targetHandler.prepare(instrumentedType);
}
/**
* {@inheritDoc}
*/
public ByteCodeAppender appender(Target implementationTarget) {
return new Appender(implementationTarget, terminationHandler.make(implementationTarget.getInstrumentedType()));
}
/**
* A method locator is responsible for identifying the method that is to be invoked
* by a {@link net.bytebuddy.implementation.MethodCall}.
*/
public interface MethodLocator {
/**
* Resolves the method to be invoked.
*
* @param targetType The type the method is called on.
* @param instrumentedMethod The method being instrumented.
* @return The method to invoke.
*/
MethodDescription resolve(TypeDescription targetType, MethodDescription instrumentedMethod);
/**
* A factory for creating a method locator.
*/
interface Factory {
/**
* Creates a method locator for a given instrumented type.
*
* @param instrumentedType The instrumented type.
* @return A suitable method locator.
*/
MethodLocator make(TypeDescription instrumentedType);
}
/**
* A method locator that simply returns the intercepted method.
*/
enum ForInstrumentedMethod implements MethodLocator, Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public MethodLocator make(TypeDescription instrumentedType) {
return this;
}
/**
* {@inheritDoc}
*/
public MethodDescription resolve(TypeDescription targetType, MethodDescription instrumentedMethod) {
return instrumentedMethod;
}
}
/**
* Invokes a given method.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForExplicitMethod implements MethodLocator, Factory {
/**
* 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.
*/
protected ForExplicitMethod(MethodDescription methodDescription) {
this.methodDescription = methodDescription;
}
/**
* {@inheritDoc}
*/
public MethodLocator make(TypeDescription instrumentedType) {
return this;
}
/**
* {@inheritDoc}
*/
public MethodDescription resolve(TypeDescription targetType, MethodDescription instrumentedMethod) {
return methodDescription;
}
}
/**
* A method locator that identifies a unique virtual method.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForElementMatcher implements MethodLocator {
/**
* The instrumented type.
*/
private final TypeDescription instrumentedType;
/**
* The matcher to use.
*/
private final ElementMatcher matcher;
/**
* The method graph compiler to use.
*/
private final MethodGraph.Compiler methodGraphCompiler;
/**
* Creates a new method locator for an element matcher.
*
* @param instrumentedType The instrumented type.
* @param matcher The matcher to use.
* @param methodGraphCompiler The method graph compiler to use.
*/
protected ForElementMatcher(TypeDescription instrumentedType, ElementMatcher matcher, MethodGraph.Compiler methodGraphCompiler) {
this.instrumentedType = instrumentedType;
this.matcher = matcher;
this.methodGraphCompiler = methodGraphCompiler;
}
/**
* {@inheritDoc}
*/
public MethodDescription resolve(TypeDescription targetType, MethodDescription instrumentedMethod) {
TypeDescription.Generic superClass = instrumentedType.getSuperClass();
List candidates = CompoundList.of(
superClass == null
? Collections.emptyList()
: superClass.getDeclaredMethods().filter(isConstructor().and(matcher)),
instrumentedType.getDeclaredMethods().filter(not(ElementMatchers.isVirtual()).and(matcher)),
methodGraphCompiler.compile((TypeDefinition) targetType, instrumentedType).listNodes().asMethodList().filter(matcher));
if (candidates.size() == 1) {
return candidates.get(0);
} else {
throw new IllegalStateException(instrumentedType + " does not define exactly one virtual method or constructor for " + matcher
+ " but contained " + candidates.size()
+ " candidates: " + candidates);
}
}
/**
* A factory for a method locator that uses a matcher on the instrumented type's available methods for identifing a target method.
*/
@HashCodeAndEqualsPlugin.Enhance
public static class Factory implements MethodLocator.Factory {
/**
* The matcher to use.
*/
private final ElementMatcher matcher;
/**
* The method graph compiler to use.
*/
private final MethodGraph.Compiler methodGraphCompiler;
/**
* Creates a factory for a method locator that identifies a method using a matcher.
*
* @param matcher The matcher to use.
* @param methodGraphCompiler The method graph compiler to use.
*/
public Factory(ElementMatcher matcher, MethodGraph.Compiler methodGraphCompiler) {
this.matcher = matcher;
this.methodGraphCompiler = methodGraphCompiler;
}
/**
* {@inheritDoc}
*/
public MethodLocator make(TypeDescription instrumentedType) {
return new ForElementMatcher(instrumentedType, matcher, methodGraphCompiler);
}
}
}
}
/**
* An argument loader is responsible for loading an argument for an invoked method
* onto the operand stack.
*/
public 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 toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing);
/**
* An argument provider is responsible for providing an argument loader for each argument to supply to a method.
*/
interface ArgumentProvider {
/**
* Resolves this provider to an argument loader for each provided argument.
*
* @param instrumentedMethod The instrumented method.
* @param invokedMethod The invoked method.
* @return A list of provided argument loaders.
*/
List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod);
}
/**
* A factory that produces {@link ArgumentLoader}s for a given instrumented method.
*/
interface Factory extends InstrumentedType.Prepareable {
/**
* Creates an argument provider for the supplied implementation target.
*
* @param implementationTarget The implementation target to use.
* @return An appropriate argument provider.
*/
ArgumentProvider make(Implementation.Target implementationTarget);
}
/**
* An argument loader that loads the {@code null} value onto the operand stack.
*/
enum ForNullConstant implements ArgumentLoader, ArgumentProvider, Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return this;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
return Collections.singletonList(this);
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
if (target.getType().isPrimitive()) {
throw new IllegalStateException("Cannot assign null to " + target);
}
return NullConstant.INSTANCE;
}
}
/**
* An argument loader that assigns the {@code this} reference to a parameter.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForThisReference implements ArgumentLoader, ArgumentProvider {
/**
* 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.
*/
public ForThisReference(TypeDescription instrumentedType) {
this.instrumentedType = instrumentedType;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
if (instrumentedMethod.isStatic()) {
throw new IllegalStateException(instrumentedMethod + " is static and cannot supply an invoker instance");
}
return Collections.singletonList(this);
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
MethodVariableAccess.loadThis(),
assigner.assign(instrumentedType.asGenericType(), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + instrumentedType + " to " + target);
}
return stackManipulation;
}
/**
* A factory for an argument loader that supplies the {@code this} value as an argument.
*/
public enum Factory implements ArgumentLoader.Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return new ForThisReference(implementationTarget.getInstrumentedType());
}
}
}
/**
* Loads the instrumented type onto the operand stack.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForInstrumentedType implements ArgumentLoader, ArgumentProvider {
/**
* 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.
*/
public ForInstrumentedType(TypeDescription instrumentedType) {
this.instrumentedType = instrumentedType;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
return Collections.singletonList(this);
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
ClassConstant.of(instrumentedType),
assigner.assign(TypeDescription.Generic.OfNonGenericType.ForLoadedType.of(Class.class), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign Class value to " + target);
}
return stackManipulation;
}
/**
* A factory for an argument loader that supplies the instrumented type as an argument.
*/
public enum Factory implements ArgumentLoader.Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return new ForInstrumentedType(implementationTarget.getInstrumentedType());
}
}
}
/**
* Loads a parameter of the instrumented method onto the operand stack.
*/
@HashCodeAndEqualsPlugin.Enhance
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.
*/
public ForMethodParameter(int index, MethodDescription instrumentedMethod) {
this.index = index;
this.instrumentedMethod = instrumentedMethod;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
ParameterDescription parameterDescription = instrumentedMethod.getParameters().get(index);
StackManipulation stackManipulation = new StackManipulation.Compound(
MethodVariableAccess.load(parameterDescription),
assigner.assign(parameterDescription.getType(), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + parameterDescription + " to " + target + " for " + instrumentedMethod);
}
return stackManipulation;
}
/**
* A factory for argument loaders that supplies all arguments of the instrumented method as arguments.
*/
protected enum OfInstrumentedMethod implements ArgumentLoader.Factory, ArgumentProvider {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return this;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
List argumentLoaders = new ArrayList(instrumentedMethod.getParameters().size());
for (ParameterDescription parameterDescription : instrumentedMethod.getParameters()) {
argumentLoaders.add(new ForMethodParameter(parameterDescription.getIndex(), instrumentedMethod));
}
return argumentLoaders;
}
}
/**
* A factory for an argument loader that supplies a method parameter as an argument.
*/
@HashCodeAndEqualsPlugin.Enhance
public static class Factory implements ArgumentLoader.Factory, ArgumentProvider {
/**
* 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.
*/
public Factory(int index) {
this.index = index;
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return this;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
if (index >= instrumentedMethod.getParameters().size()) {
throw new IllegalStateException(instrumentedMethod + " does not have a parameter with index " + index +
", " + instrumentedMethod.getParameters().size() + " defined");
}
return Collections.singletonList(new ForMethodParameter(index, instrumentedMethod));
}
}
}
/**
* Loads an array containing all arguments of a method.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForMethodParameterArray implements ArgumentLoader {
/**
* The parameters to load.
*/
private final ParameterList parameters;
/**
* Creates an argument loader that loads the supplied parameters onto the operand stack.
*
* @param parameters The parameters to load.
*/
public ForMethodParameterArray(ParameterList parameters) {
this.parameters = parameters;
}
/**
* {@inheritDoc}
*/
@SuppressFBWarnings(value = "NP_NULL_ON_SOME_PATH_FROM_RETURN_VALUE", justification = "Assuming component type for array type.")
public StackManipulation toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
TypeDescription.Generic componentType;
if (target.getType().represents(Object.class)) {
componentType = TypeDescription.Generic.OfNonGenericType.ForLoadedType.of(Object.class);
} else if (target.getType().isArray()) {
componentType = target.getType().getComponentType();
} else {
throw new IllegalStateException("Cannot set method parameter array for non-array type: " + target);
}
List stackManipulations = new ArrayList(parameters.size());
for (ParameterDescription parameter : parameters) {
StackManipulation stackManipulation = new StackManipulation.Compound(
MethodVariableAccess.load(parameter),
assigner.assign(parameter.getType(), componentType, typing)
);
if (stackManipulation.isValid()) {
stackManipulations.add(stackManipulation);
} else {
throw new IllegalStateException("Cannot assign " + parameter + " to " + componentType);
}
}
return new StackManipulation.Compound(ArrayFactory.forType(componentType).withValues(stackManipulations));
}
/**
* A factory that creates an arguments loader that loads all parameters of the instrumented method contained in an array.
*/
public enum ForInstrumentedMethod implements Factory, ArgumentProvider {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return this;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
return Collections.singletonList(new ForMethodParameterArray(instrumentedMethod.getParameters()));
}
}
}
/**
* An argument loader that loads an element of a parameter of an array type.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForMethodParameterArrayElement implements ArgumentLoader {
/**
* The parameter to load the array from.
*/
private final ParameterDescription parameterDescription;
/**
* The array index to load.
*/
private final int index;
/**
* Creates an argument loader for a parameter of the instrumented method where an array element is assigned to the invoked method.
*
* @param parameterDescription The parameter from which to load an array element.
* @param index The array index to load.
*/
public ForMethodParameterArrayElement(ParameterDescription parameterDescription, int index) {
this.parameterDescription = parameterDescription;
this.index = index;
}
/**
* {@inheritDoc}
*/
@SuppressFBWarnings(value = "NP_NULL_ON_SOME_PATH_FROM_RETURN_VALUE", justification = "Assuming component type for array type.")
public StackManipulation toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
MethodVariableAccess.load(parameterDescription),
IntegerConstant.forValue(index),
ArrayAccess.of(parameterDescription.getType().getComponentType()).load(),
assigner.assign(parameterDescription.getType().getComponentType(), target.getType(), typing)
);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + parameterDescription.getType().getComponentType() + " to " + target);
}
return stackManipulation;
}
/**
* Creates an argument loader for an array element that of a specific parameter.
*/
@HashCodeAndEqualsPlugin.Enhance
public static class OfParameter implements ArgumentLoader.Factory, ArgumentProvider {
/**
* The parameter index.
*/
private final int index;
/**
* The array index to load.
*/
private final int arrayIndex;
/**
* Creates a factory for an argument loader that loads a given parameter's array value.
*
* @param index The index of the parameter.
* @param arrayIndex The array index to load.
*/
public OfParameter(int index, int arrayIndex) {
this.index = index;
this.arrayIndex = arrayIndex;
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return this;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
if (instrumentedMethod.getParameters().size() <= index) {
throw new IllegalStateException(instrumentedMethod + " does not declare a parameter with index " + index +
", " + instrumentedMethod.getParameters().size() + " defined");
} else if (!instrumentedMethod.getParameters().get(index).getType().isArray()) {
throw new IllegalStateException("Cannot access an item from non-array parameter " + instrumentedMethod.getParameters().get(index) +
" at index " + index);
}
return Collections.singletonList(new ForMethodParameterArrayElement(instrumentedMethod.getParameters().get(index), arrayIndex));
}
}
/**
* An argument loader factory that loads an array element from a parameter for each argument of the invoked method.
*/
@HashCodeAndEqualsPlugin.Enhance
public static class OfInvokedMethod implements ArgumentLoader.Factory, ArgumentProvider {
/**
* The parameter index.
*/
private final int index;
/**
* Creates an argument loader factory for an invoked method.
*
* @param index The parameter index.
*/
public OfInvokedMethod(int index) {
this.index = index;
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return this;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
if (instrumentedMethod.getParameters().size() <= index) {
throw new IllegalStateException(instrumentedMethod + " does not declare a parameter with index " + index +
", " + instrumentedMethod.getParameters().size() + " defined");
} else if (!instrumentedMethod.getParameters().get(index).getType().isArray()) {
throw new IllegalStateException("Cannot access an item from non-array parameter " + instrumentedMethod.getParameters().get(index) +
" at index " + index);
}
List argumentLoaders = new ArrayList(invokedMethod.getParameters().size());
for (int index = 0; index < invokedMethod.getParameters().size(); index++) {
argumentLoaders.add(new ForMethodParameterArrayElement(instrumentedMethod.getParameters().get(this.index), index));
}
return argumentLoaders;
}
}
}
/**
* Loads a value onto the operand stack that is stored in a static field.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForInstance implements ArgumentLoader, ArgumentProvider {
/**
* 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.
*/
public ForInstance(FieldDescription fieldDescription) {
this.fieldDescription = fieldDescription;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
return Collections.singletonList(this);
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
FieldAccess.forField(fieldDescription).read(),
assigner.assign(fieldDescription.getType(), target.getType(), typing));
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + fieldDescription.getType() + " to " + target);
}
return stackManipulation;
}
/**
* A factory that supplies the value of a static field as an argument.
*/
@HashCodeAndEqualsPlugin.Enhance
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.
*/
@HashCodeAndEqualsPlugin.ValueHandling(HashCodeAndEqualsPlugin.ValueHandling.Sort.IGNORE)
private final String name;
/**
* Creates a factory that loads the value of a static field as an argument.
*
* @param value The value to supply as an argument.
*/
public Factory(Object value) {
this.value = value;
name = FIELD_PREFIX + "$" + RandomString.hashOf(value);
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType.withAuxiliaryField(new FieldDescription.Token(name,
Opcodes.ACC_SYNTHETIC | Opcodes.ACC_PUBLIC | Opcodes.ACC_STATIC,
TypeDescription.Generic.OfNonGenericType.ForLoadedType.of(value.getClass())), value);
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return new ForInstance(implementationTarget.getInstrumentedType().getDeclaredFields().filter(named(name)).getOnly());
}
}
}
/**
* Loads the value of an existing field onto the operand stack.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForField 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.
*/
public ForField(FieldDescription fieldDescription, MethodDescription instrumentedMethod) {
this.fieldDescription = fieldDescription;
this.instrumentedMethod = instrumentedMethod;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(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.loadThis(),
FieldAccess.forField(fieldDescription).read(),
assigner.assign(fieldDescription.getType(), target.getType(), typing)
);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign " + fieldDescription + " to " + target);
}
return stackManipulation;
}
/**
* An argument provider for a field access.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class ArgumentProvider implements ArgumentLoader.ArgumentProvider {
/**
* The field being accessed.
*/
private final FieldDescription fieldDescription;
/**
* Creates a new argument provider for a field access.
*
* @param fieldDescription The field being accessed.
*/
protected ArgumentProvider(FieldDescription fieldDescription) {
this.fieldDescription = fieldDescription;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
return Collections.singletonList(new ForField(fieldDescription, instrumentedMethod));
}
}
/**
* A factory for an argument loaded that loads the value of an existing field as an argument.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Factory implements ArgumentLoader.Factory {
/**
* The name of the field.
*/
private final String name;
/**
* The field locator to use.
*/
private final FieldLocator.Factory fieldLocatorFactory;
/**
* Creates a new argument loader for an existing field.
*
* @param name The name of the field.
* @param fieldLocatorFactory The field locator to use.
*/
public Factory(String name, FieldLocator.Factory fieldLocatorFactory) {
this.name = name;
this.fieldLocatorFactory = fieldLocatorFactory;
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentLoader.ArgumentProvider make(Implementation.Target implementationTarget) {
FieldLocator.Resolution resolution = fieldLocatorFactory.make(implementationTarget.getInstrumentedType()).locate(name);
if (!resolution.isResolved()) {
throw new IllegalStateException("Could not locate field '" + name + "' on " + implementationTarget.getInstrumentedType());
}
return new ArgumentProvider(resolution.getField());
}
}
}
/**
* Loads the return value of a method call onto the operand stack.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForMethodCall implements ArgumentLoader {
/**
* The method call's appender.
*/
private final Appender appender;
/**
* The invoked method.
*/
private final MethodDescription methodDescription;
/**
* The instrumented method.
*/
private final MethodDescription instrumentedMethod;
/**
* The resolved target handler to use.
*/
private final TargetHandler.Resolved targetHandler;
/**
* Creates a new argument loader for loading a method call's return value.
*
* @param appender The method call's appender.
* @param methodDescription The invoked method.
* @param instrumentedMethod The instrumented method.
* @param targetHandler The resolved target handler to use.
*/
public ForMethodCall(Appender appender, MethodDescription methodDescription, MethodDescription instrumentedMethod, TargetHandler.Resolved targetHandler) {
this.appender = appender;
this.methodDescription = methodDescription;
this.instrumentedMethod = instrumentedMethod;
this.targetHandler = targetHandler;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = new StackManipulation.Compound(
appender.toStackManipulation(instrumentedMethod, methodDescription, targetHandler),
assigner.assign(methodDescription.isConstructor()
? methodDescription.getDeclaringType().asGenericType()
: methodDescription.getReturnType(), target.getType(), typing)
);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign return type of " + methodDescription + " to " + target);
}
return stackManipulation;
}
/**
* An argument provider for a method call.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class ArgumentProvider implements ArgumentLoader.ArgumentProvider {
/**
* The method call's appender.
*/
private final Appender appender;
/**
* Creates a new argument provider for a method call.
*
* @param appender The method call's appender.
*/
protected ArgumentProvider(Appender appender) {
this.appender = appender;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
TargetHandler.Resolved targetHandler = appender.targetHandler.resolve(instrumentedMethod);
return Collections.singletonList(new ForMethodCall(appender,
appender.toInvokedMethod(instrumentedMethod, targetHandler),
instrumentedMethod,
targetHandler));
}
}
/**
* A factory for an argument loaded that loads the return value of a method call as an argument.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Factory implements ArgumentLoader.Factory {
/**
* The method call to use.
*/
private final MethodCall methodCall;
/**
* Creates a new argument loader for an existing method call.
*
* @param methodCall The method call to use.
*/
public Factory(MethodCall methodCall) {
this.methodCall = methodCall;
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return methodCall.prepare(instrumentedType);
}
/**
* {@inheritDoc}
*/
public ArgumentLoader.ArgumentProvider make(Implementation.Target implementationTarget) {
return new ArgumentProvider(methodCall.new Appender(implementationTarget, TerminationHandler.Simple.IGNORING));
}
}
}
/**
* Loads a stack manipulation resulting in a specific type as an argument.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForStackManipulation implements ArgumentLoader, ArgumentProvider, Factory {
/**
* The stack manipulation to load.
*/
private final StackManipulation stackManipulation;
/**
* The type of the resulting value.
*/
private final TypeDefinition typeDefinition;
/**
* Creates an argument loader that loads a stack manipulation as an argument.
*
* @param stackManipulation The stack manipulation to load.
* @param type The type of the resulting value.
*/
public ForStackManipulation(StackManipulation stackManipulation, Type type) {
this(stackManipulation, TypeDescription.Generic.Sort.describe(type));
}
/**
* Creates an argument loader that loads a stack manipulation as an argument.
*
* @param stackManipulation The stack manipulation to load.
* @param typeDefinition The type of the resulting value.
*/
public ForStackManipulation(StackManipulation stackManipulation, TypeDefinition typeDefinition) {
this.stackManipulation = stackManipulation;
this.typeDefinition = typeDefinition;
}
/**
* Creates an argument loader that loads the supplied value as a constant. If the value cannot be represented
* in the constant pool, a field is created to store the value.
*
* @param value The value to load as an argument or {@code null}.
* @return An appropriate argument loader.
*/
public static ArgumentLoader.Factory of(@MaybeNull Object value) {
if (value == null) {
return ForNullConstant.INSTANCE;
} else {
ConstantValue constant = ConstantValue.Simple.wrapOrNull(value);
return constant == null
? new ForInstance.Factory(value)
: new ForStackManipulation(constant.toStackManipulation(), constant.getTypeDescription());
}
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public ArgumentProvider make(Implementation.Target implementationTarget) {
return this;
}
/**
* {@inheritDoc}
*/
public List resolve(MethodDescription instrumentedMethod, MethodDescription invokedMethod) {
return Collections.singletonList(this);
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(ParameterDescription target, Assigner assigner, Assigner.Typing typing) {
StackManipulation assignment = assigner.assign(typeDefinition.asGenericType(), target.getType(), typing);
if (!assignment.isValid()) {
throw new IllegalStateException("Cannot assign " + target + " to " + typeDefinition);
}
return new StackManipulation.Compound(stackManipulation, assignment);
}
}
}
/**
* A target handler is responsible for invoking a method for a
* {@link net.bytebuddy.implementation.MethodCall}.
*/
protected interface TargetHandler {
/**
* Resolves this target handler.
*
* @param instrumentedMethod The instrumented method.
* @return The resolved target handler.
*/
Resolved resolve(MethodDescription instrumentedMethod);
/**
* A resolved target handler.
*/
interface Resolved {
/**
* Returns the target's type description.
*
* @return The target's type description.
*/
TypeDescription getTypeDescription();
/**
* Creates a stack manipulation to represent this resolved target handler.
*
* @param invokedMethod The invoked method.
* @param assigner The assigner to use.
* @param typing The typing to apply.
* @return A stack manipulation that implements this target handler.
*/
StackManipulation toStackManipulation(MethodDescription invokedMethod, Assigner assigner, Assigner.Typing typing);
}
/**
* A factory for creating a target handler.
*/
interface Factory extends InstrumentedType.Prepareable {
/**
* Creates a target handler for a given implementation target.
*
* @param implementationTarget The implementation target to use.
* @return The target handler to use.
*/
TargetHandler make(Implementation.Target implementationTarget);
}
/**
* A simple target handler that applies a given stack manipulation.
*/
@HashCodeAndEqualsPlugin.Enhance
class Simple implements TargetHandler, Factory, Resolved {
/**
* The type resolved by the stack manipulation.
*/
private final TypeDescription typeDescription;
/**
* The stack manipulation that loads a target for the method call.
*/
private final StackManipulation stackManipulation;
/**
* Creates a simple target handler.
*
* @param typeDescription The type resolved by the stack manipulation.
* @param stackManipulation The stack manipulation that loads a target for the method call.
*/
protected Simple(TypeDescription typeDescription, StackManipulation stackManipulation) {
this.typeDescription = typeDescription;
this.stackManipulation = stackManipulation;
}
/**
* {@inheritDoc}
*/
public TargetHandler make(Target implementationTarget) {
return this;
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public Resolved resolve(MethodDescription instrumentedMethod) {
return this;
}
/**
* {@inheritDoc}
*/
public TypeDescription getTypeDescription() {
return typeDescription;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Assigner assigner, Assigner.Typing typing) {
return stackManipulation;
}
}
/**
* A target handler that invokes a method either on the instance of the instrumented
* type or as a static method.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForSelfOrStaticInvocation implements TargetHandler {
/**
* The instrumented type.
*/
private final TypeDescription instrumentedType;
/**
* Creates a new target handler for a static or self-declared invocation.
*
* @param instrumentedType The instrumented type.
*/
protected ForSelfOrStaticInvocation(TypeDescription instrumentedType) {
this.instrumentedType = instrumentedType;
}
/**
* {@inheritDoc}
*/
public TargetHandler.Resolved resolve(MethodDescription instrumentedMethod) {
return new Resolved(instrumentedType, instrumentedMethod);
}
/**
* A resolved target handler for a static or self-declared invocation.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Resolved implements TargetHandler.Resolved {
/**
* The instrumented type.
*/
private final TypeDescription instrumentedType;
/**
* The instrumented method.
*/
private final MethodDescription instrumentedMethod;
/**
* Creates a resolved target handler for a static or self-declared invocation.
*
* @param instrumentedType The instrumented type.
* @param instrumentedMethod The instrumented method.
*/
protected Resolved(TypeDescription instrumentedType,
MethodDescription instrumentedMethod) {
this.instrumentedType = instrumentedType;
this.instrumentedMethod = instrumentedMethod;
}
/**
* {@inheritDoc}
*/
public TypeDescription getTypeDescription() {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
@SuppressFBWarnings(value = "NP_NULL_ON_SOME_PATH_FROM_RETURN_VALUE", justification = "Assuming declaring type for type member.")
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Assigner assigner, Assigner.Typing typing) {
if (instrumentedMethod.isStatic() && !invokedMethod.isStatic() && !invokedMethod.isConstructor()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " from " + instrumentedMethod);
} else if (invokedMethod.isConstructor() && (!instrumentedMethod.isConstructor()
|| !instrumentedType.equals(invokedMethod.getDeclaringType().asErasure())
&& (instrumentedType.getSuperClass() == null
|| !instrumentedType.getSuperClass().asErasure().equals(invokedMethod.getDeclaringType().asErasure())))) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " from " + instrumentedMethod + " in " + instrumentedType);
}
return new StackManipulation.Compound(
invokedMethod.isStatic()
? StackManipulation.Trivial.INSTANCE
: MethodVariableAccess.loadThis(),
invokedMethod.isConstructor()
? Duplication.SINGLE
: StackManipulation.Trivial.INSTANCE
);
}
}
/**
* A factory for invoking a static method or a self-declared method.
*/
protected enum Factory implements TargetHandler.Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public TargetHandler make(Implementation.Target implementationTarget) {
return new ForSelfOrStaticInvocation(implementationTarget.getInstrumentedType());
}
}
}
/**
* Invokes a method in order to construct a new instance.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForConstructingInvocation implements TargetHandler, Resolved {
/**
* The instrumented type.
*/
private final TypeDescription instrumentedType;
/**
* Creates a new target handle constructor invocation.
*
* @param instrumentedType The instrumented type.
*/
protected ForConstructingInvocation(TypeDescription instrumentedType) {
this.instrumentedType = instrumentedType;
}
/**
* {@inheritDoc}
*/
public Resolved resolve(MethodDescription instrumentedMethod) {
return this;
}
/**
* {@inheritDoc}
*/
public TypeDescription getTypeDescription() {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Assigner assigner, Assigner.Typing typing) {
return new StackManipulation.Compound(TypeCreation.of(invokedMethod.getDeclaringType().asErasure()), Duplication.SINGLE);
}
/**
* A target handler factory for a constructor invocation.
*/
enum Factory implements TargetHandler.Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public TargetHandler make(Implementation.Target implementationTarget) {
return new ForConstructingInvocation(implementationTarget.getInstrumentedType());
}
}
}
/**
* A target handler that invokes a method on an instance that is stored in a static field.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForValue implements TargetHandler, Resolved {
/**
* A description of the field that contains the value.
*/
private final FieldDescription.InDefinedShape fieldDescription;
/**
* Creates a new target handler for a field value.
*
* @param fieldDescription A description of the field that contains the value.
*/
protected ForValue(FieldDescription.InDefinedShape fieldDescription) {
this.fieldDescription = fieldDescription;
}
/**
* {@inheritDoc}
*/
public Resolved resolve(MethodDescription instrumentedMethod) {
return this;
}
/**
* {@inheritDoc}
*/
public TypeDescription getTypeDescription() {
return fieldDescription.getType().asErasure();
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = assigner.assign(fieldDescription.getType(), invokedMethod.getDeclaringType().asGenericType(), typing);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + fieldDescription);
}
return new StackManipulation.Compound(
FieldAccess.forField(fieldDescription).read(),
stackManipulation
);
}
/**
* A factory for a target handler that loads a specific value.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Factory implements TargetHandler.Factory {
/**
* 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 type of the field.
*/
private final TypeDescription.Generic fieldType;
/**
* The name of the field to store the target.
*/
@HashCodeAndEqualsPlugin.ValueHandling(HashCodeAndEqualsPlugin.ValueHandling.Sort.IGNORE)
private final String name;
/**
* Creates a new target handler for a static field.
*
* @param target The target on which the method is to be invoked.
* @param fieldType The type of the field.
*/
protected Factory(Object target, TypeDescription.Generic fieldType) {
this.target = target;
this.fieldType = fieldType;
name = FIELD_PREFIX + "$" + RandomString.hashOf(target);
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType.withAuxiliaryField(new FieldDescription.Token(name,
Opcodes.ACC_PUBLIC | Opcodes.ACC_STATIC | Opcodes.ACC_VOLATILE | Opcodes.ACC_SYNTHETIC,
fieldType), target);
}
/**
* {@inheritDoc}
*/
public TargetHandler make(Implementation.Target implementationTarget) {
return new ForValue(implementationTarget.getInstrumentedType().getDeclaredFields().filter(named(name)).getOnly());
}
}
}
/**
* Creates a target handler that stores the instance to invoke a method on in an instance field.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForField implements TargetHandler, Resolved {
/**
* A description of the field that is the target.
*/
private final FieldDescription fieldDescription;
/**
* Creates a new target handler for a field description.
*
* @param fieldDescription A description of the field that is the target.
*/
protected ForField(FieldDescription fieldDescription) {
this.fieldDescription = fieldDescription;
}
/**
* {@inheritDoc}
*/
public Resolved resolve(MethodDescription instrumentedMethod) {
return this;
}
/**
* {@inheritDoc}
*/
public TypeDescription getTypeDescription() {
return fieldDescription.getType().asErasure();
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Assigner assigner, Assigner.Typing typing) {
if (!invokedMethod.isMethod()
|| !invokedMethod.isVirtual()
|| !invokedMethod.isVisibleTo(fieldDescription.getType().asErasure())) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + fieldDescription);
}
StackManipulation stackManipulation = assigner.assign(fieldDescription.getType(), invokedMethod.getDeclaringType().asGenericType(), typing);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + fieldDescription);
}
return new StackManipulation.Compound(invokedMethod.isStatic() || fieldDescription.isStatic()
? StackManipulation.Trivial.INSTANCE
: MethodVariableAccess.loadThis(),
FieldAccess.forField(fieldDescription).read(), stackManipulation);
}
/**
* A location of a field.
*/
protected interface Location {
/**
* Resolves the field to invoke the method upon.
*
* @param instrumentedType The instrumented type.
* @return The field to invoke the method upon.
*/
FieldDescription resolve(TypeDescription instrumentedType);
/**
* An implicit field location.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForImplicitField implements Location {
/**
* The name of the field.
*/
private final String name;
/**
* The field locator factory to use.
*/
private final FieldLocator.Factory fieldLocatorFactory;
/**
* Creates an implicit field location.
*
* @param name The name of the field.
* @param fieldLocatorFactory The field locator factory to use.
*/
protected ForImplicitField(String name, FieldLocator.Factory fieldLocatorFactory) {
this.name = name;
this.fieldLocatorFactory = fieldLocatorFactory;
}
/**
* {@inheritDoc}
*/
public FieldDescription resolve(TypeDescription instrumentedType) {
FieldLocator.Resolution resolution = fieldLocatorFactory.make(instrumentedType).locate(name);
if (!resolution.isResolved()) {
throw new IllegalStateException("Could not locate field name " + name + " on " + instrumentedType);
}
return resolution.getField();
}
}
/**
* An explicit field location.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForExplicitField implements Location {
/**
* The field to resolve.
*/
private final FieldDescription fieldDescription;
/**
* Creates an explicit field location.
*
* @param fieldDescription The field to resolve.
*/
protected ForExplicitField(FieldDescription fieldDescription) {
this.fieldDescription = fieldDescription;
}
/**
* {@inheritDoc}
*/
public FieldDescription resolve(TypeDescription instrumentedType) {
return fieldDescription;
}
}
}
/**
* A factory for a field access.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Factory implements TargetHandler.Factory {
/**
* The field's location.
*/
private final Location location;
/**
* Creates a new target handler factory for a field location.
*
* @param location The field's location.
*/
protected Factory(Location location) {
this.location = location;
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
@SuppressFBWarnings(value = "NP_NULL_ON_SOME_PATH_FROM_RETURN_VALUE", justification = "Assuming declaring type for type member.")
public TargetHandler make(Implementation.Target implementationTarget) {
FieldDescription fieldDescription = location.resolve(implementationTarget.getInstrumentedType());
if (!fieldDescription.isStatic() && !implementationTarget.getInstrumentedType().isAssignableTo(fieldDescription.getDeclaringType().asErasure())) {
throw new IllegalStateException("Cannot access " + fieldDescription + " from " + implementationTarget.getInstrumentedType());
}
return new ForField(fieldDescription);
}
}
}
/**
* A target handler that loads the parameter of the given index as the target object.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForMethodParameter implements TargetHandler, Factory {
/**
* 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.
*/
protected ForMethodParameter(int index) {
this.index = index;
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return instrumentedType;
}
/**
* {@inheritDoc}
*/
public TargetHandler make(Implementation.Target implementationTarget) {
return this;
}
/**
* {@inheritDoc}
*/
public TargetHandler.Resolved resolve(MethodDescription instrumentedMethod) {
if (index >= instrumentedMethod.getParameters().size()) {
throw new IllegalArgumentException(instrumentedMethod + " does not have a parameter with index " + index);
}
return new Resolved(instrumentedMethod.getParameters().get(index));
}
/**
* A resolved target handler for a method parameter.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Resolved implements TargetHandler.Resolved {
/**
* The parameter that is the target of the method call.
*/
private final ParameterDescription parameterDescription;
/**
* Creates a new resolved target handler for a parameter.
*
* @param parameterDescription The parameter that is the target of the method call.
*/
protected Resolved(ParameterDescription parameterDescription) {
this.parameterDescription = parameterDescription;
}
/**
* {@inheritDoc}
*/
public TypeDescription getTypeDescription() {
return parameterDescription.getType().asErasure();
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Assigner assigner, Assigner.Typing typing) {
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.load(parameterDescription), stackManipulation);
}
}
}
/**
* A target handler that executes the method and uses it's return value as the target object.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForMethodCall implements TargetHandler {
/**
* The appender that is the target of the using method call.
*/
private final Appender appender;
/**
* Creates a new target handler for another method call.
*
* @param appender The appender that is the target of the using method call.
*/
protected ForMethodCall(Appender appender) {
this.appender = appender;
}
/**
* {@inheritDoc}
*/
public TargetHandler.Resolved resolve(MethodDescription instrumentedMethod) {
TargetHandler.Resolved targetHandler = appender.targetHandler.resolve(instrumentedMethod);
return new Resolved(appender, appender.toInvokedMethod(instrumentedMethod, targetHandler), instrumentedMethod,
targetHandler);
}
/**
* A resolved target handler for a method call.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Resolved implements TargetHandler.Resolved {
/**
* The appender to use.
*/
private final Appender appender;
/**
* The invoked method.
*/
private final MethodDescription methodDescription;
/**
* The instrumented method.
*/
private final MethodDescription instrumentedMethod;
/**
* The target handler to use.
*/
private final TargetHandler.Resolved targetHandler;
/**
* Creates a resolved target handler for a method call.
*
* @param appender The appender to use.
* @param methodDescription The invoked method.
* @param instrumentedMethod The instrumented method.
* @param targetHandler The target handler to use.
*/
protected Resolved(Appender appender, MethodDescription methodDescription, MethodDescription instrumentedMethod, TargetHandler.Resolved targetHandler) {
this.appender = appender;
this.methodDescription = methodDescription;
this.instrumentedMethod = instrumentedMethod;
this.targetHandler = targetHandler;
}
/**
* {@inheritDoc}
*/
public TypeDescription getTypeDescription() {
return methodDescription.isConstructor()
? methodDescription.getDeclaringType().asErasure()
: methodDescription.getReturnType().asErasure();
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = assigner.assign(methodDescription.isConstructor()
? methodDescription.getDeclaringType().asGenericType()
: methodDescription.getReturnType(), invokedMethod.getDeclaringType().asGenericType(), typing);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + (methodDescription.isConstructor()
? methodDescription.getDeclaringType()
: methodDescription.getReturnType()));
}
return new StackManipulation.Compound(appender.toStackManipulation(instrumentedMethod,
methodDescription,
targetHandler), stackManipulation);
}
}
/**
* Creates a factory for invoking a method on the return type of another method.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Factory implements TargetHandler.Factory {
/**
* The method call to invoke.
*/
private final MethodCall methodCall;
/**
* Creates a new factory for invoking another method call as a method target.
*
* @param methodCall The method call to invoke.
*/
public Factory(MethodCall methodCall) {
this.methodCall = methodCall;
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return methodCall.prepare(instrumentedType);
}
/**
* {@inheritDoc}
*/
public TargetHandler make(Implementation.Target implementationTarget) {
return new ForMethodCall(methodCall.new Appender(implementationTarget, TerminationHandler.Simple.IGNORING));
}
}
}
}
/**
* A method invoker is responsible for creating a method invocation that is to be applied by a
* {@link net.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 toStackManipulation(MethodDescription invokedMethod, Implementation.Target implementationTarget);
/**
* A factory for creating a method invoker.
*/
interface Factory {
/**
* Creates a method invoker.
*
* @param instrumentedType The instrumented type.
* @return The method invoker to use.
*/
MethodInvoker make(TypeDescription instrumentedType);
}
/**
* 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.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForContextualInvocation implements MethodInvoker {
/**
* The instrumented type.
*/
private final TypeDescription instrumentedType;
/**
* Creates a new method invoker for a contextual invocation.
*
* @param instrumentedType The instrumented type.
*/
protected ForContextualInvocation(TypeDescription instrumentedType) {
this.instrumentedType = instrumentedType;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Implementation.Target implementationTarget) {
if (invokedMethod.isVirtual() && !invokedMethod.isInvokableOn(instrumentedType)) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + instrumentedType);
}
return invokedMethod.isVirtual()
? MethodInvocation.invoke(invokedMethod).virtual(instrumentedType)
: MethodInvocation.invoke(invokedMethod);
}
/**
* A factory for creating a contextual method invoker.
*/
enum Factory implements MethodInvoker.Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public MethodInvoker make(TypeDescription instrumentedType) {
return new ForContextualInvocation(instrumentedType);
}
}
}
/**
* Applies a virtual invocation on a given type.
*/
@HashCodeAndEqualsPlugin.Enhance
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;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Implementation.Target implementationTarget) {
if (!invokedMethod.isInvokableOn(typeDescription)) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + typeDescription);
}
return MethodInvocation.invoke(invokedMethod).virtual(typeDescription);
}
/**
* A method invoker for a virtual method that uses an implicit target type.
*/
protected enum WithImplicitType implements MethodInvoker, MethodInvoker.Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public MethodInvoker make(TypeDescription instrumentedType) {
return this;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Implementation.Target implementationTarget) {
if (!invokedMethod.isAccessibleTo(implementationTarget.getInstrumentedType()) || !invokedMethod.isVirtual()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " virtually");
}
return MethodInvocation.invoke(invokedMethod);
}
}
/**
* A method invoker factory for a virtual method invocation.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Factory implements MethodInvoker.Factory {
/**
* The type on which the virtual method is invoked.
*/
private final TypeDescription typeDescription;
/**
* Creates a new method invoker factory for a virtual method call.
*
* @param typeDescription The type on which the virtual method is invoked.
*/
protected Factory(TypeDescription typeDescription) {
this.typeDescription = typeDescription;
}
/**
* {@inheritDoc}
*/
public MethodInvoker make(TypeDescription instrumentedType) {
if (!typeDescription.asErasure().isAccessibleTo(instrumentedType)) {
throw new IllegalStateException(typeDescription + " is not accessible to " + instrumentedType);
}
return new ForVirtualInvocation(typeDescription);
}
}
}
/**
* Applies a super method invocation of the provided method.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForSuperMethodInvocation implements MethodInvoker {
/**
* The instrumented type.
*/
private final TypeDescription instrumentedType;
/**
* Creates a method invoker for a super method invocation.
*
* @param instrumentedType The instrumented type.
*/
protected ForSuperMethodInvocation(TypeDescription instrumentedType) {
this.instrumentedType = instrumentedType;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Implementation.Target implementationTarget) {
if (!invokedMethod.isInvokableOn(implementationTarget.getOriginType().asErasure())) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " as super method of " + instrumentedType);
}
StackManipulation stackManipulation = implementationTarget
.invokeDominant(invokedMethod.asSignatureToken())
.withCheckedCompatibilityTo(invokedMethod.asTypeToken());
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " as a super method");
}
return stackManipulation;
}
/**
* A method invoker factory for a super method invocation.
*/
enum Factory implements MethodInvoker.Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public MethodInvoker make(TypeDescription instrumentedType) {
if (instrumentedType.getSuperClass() == null) {
throw new IllegalStateException("Cannot invoke super method for " + instrumentedType);
}
return new ForSuperMethodInvocation(instrumentedType);
}
}
}
/**
* Invokes a method as a Java 8 default method.
*/
@HashCodeAndEqualsPlugin.Enhance
class ForDefaultMethodInvocation implements MethodInvoker {
/**
* The instrumented type.
*/
private final TypeDescription instrumentedType;
/**
* Creates a new method invoker for a default method invocation.
*
* @param instrumentedType The instrumented type.
*/
protected ForDefaultMethodInvocation(TypeDescription instrumentedType) {
this.instrumentedType = instrumentedType;
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, Implementation.Target implementationTarget) {
if (!invokedMethod.isInvokableOn(instrumentedType)) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " as default method of " + instrumentedType);
}
StackManipulation stackManipulation = implementationTarget
.invokeDefault(invokedMethod.asSignatureToken(), invokedMethod.getDeclaringType().asErasure())
.withCheckedCompatibilityTo(invokedMethod.asTypeToken());
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot invoke " + invokedMethod + " on " + instrumentedType);
}
return stackManipulation;
}
/**
* A factory for creating a method invoker for a default method invocation.
*/
enum Factory implements MethodInvoker.Factory {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public MethodInvoker make(TypeDescription instrumentedType) {
return new ForDefaultMethodInvocation(instrumentedType);
}
}
}
}
/**
* A termination handler is responsible to handle the return value of a method that is invoked via a
* {@link net.bytebuddy.implementation.MethodCall}.
*/
protected interface TerminationHandler {
/**
* Returns a preparing stack manipulation to apply prior to the method call.
*
* @return The stack manipulation to apply prior to the method call.
*/
StackManipulation prepare();
/**
* 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 toStackManipulation(MethodDescription invokedMethod, MethodDescription instrumentedMethod, Assigner assigner, Assigner.Typing typing);
/**
* A factory for creating a termination handler.
*/
interface Factory {
/**
* Creates a termination handler for a given instrumented type.
*
* @param instrumentedType The instrumented type.
* @return A termination handler to apply for the instrumented type.
*/
TerminationHandler make(TypeDescription instrumentedType);
}
/**
* Simple termination handler implementations.
*/
enum Simple implements TerminationHandler, Factory {
/**
* A termination handler that returns the invoked method's return value.
*/
RETURNING {
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(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.of(instrumentedMethod.getReturnType()));
}
},
/**
* A termination handler that drops the invoked method's return value.
*/
DROPPING {
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, MethodDescription instrumentedMethod, Assigner assigner, Assigner.Typing typing) {
return Removal.of(invokedMethod.isConstructor()
? invokedMethod.getDeclaringType()
: invokedMethod.getReturnType());
}
},
/**
* A termination handler that does not apply any change.
*/
IGNORING {
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, MethodDescription instrumentedMethod, Assigner assigner, Assigner.Typing typing) {
return StackManipulation.Trivial.INSTANCE;
}
};
/**
* {@inheritDoc}
*/
public TerminationHandler make(TypeDescription instrumentedType) {
return this;
}
/**
* {@inheritDoc}
*/
public StackManipulation prepare() {
return StackManipulation.Trivial.INSTANCE;
}
}
/**
* A termination handler that sets a field.
*/
@HashCodeAndEqualsPlugin.Enhance
class FieldSetting implements TerminationHandler {
/**
* The field to set.
*/
private final FieldDescription fieldDescription;
/**
* Creates a new field-setting termination handler.
*
* @param fieldDescription The field to set.
*/
protected FieldSetting(FieldDescription fieldDescription) {
this.fieldDescription = fieldDescription;
}
/**
* {@inheritDoc}
*/
public StackManipulation prepare() {
return fieldDescription.isStatic()
? StackManipulation.Trivial.INSTANCE
: MethodVariableAccess.loadThis();
}
/**
* {@inheritDoc}
*/
public StackManipulation toStackManipulation(MethodDescription invokedMethod, MethodDescription instrumentedMethod, Assigner assigner, Assigner.Typing typing) {
StackManipulation stackManipulation = assigner.assign(invokedMethod.isConstructor()
? invokedMethod.getDeclaringType().asGenericType()
: invokedMethod.getReturnType(), fieldDescription.getType(), typing);
if (!stackManipulation.isValid()) {
throw new IllegalStateException("Cannot assign result of " + invokedMethod + " to " + fieldDescription);
}
return new StackManipulation.Compound(stackManipulation, FieldAccess.forField(fieldDescription).write());
}
/**
* A factory for a field-setting termination handler that locates a given field.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Explicit implements TerminationHandler.Factory {
/**
* The matcher being used for locating a field.
*/
private final FieldDescription fieldDescription;
/**
* Creates a factory for a field-setting termination handler.
*
* @param fieldDescription The field to set.
*/
protected Explicit(FieldDescription fieldDescription) {
this.fieldDescription = fieldDescription;
}
/**
* {@inheritDoc}
*/
@SuppressFBWarnings(value = "NP_NULL_ON_SOME_PATH_FROM_RETURN_VALUE", justification = "Assuming declaring type for type member.")
public TerminationHandler make(TypeDescription instrumentedType) {
if (!fieldDescription.isStatic() && !instrumentedType.isAssignableTo(fieldDescription.getDeclaringType().asErasure())) {
throw new IllegalStateException("Cannot set " + fieldDescription + " from " + instrumentedType);
} else if (!fieldDescription.isVisibleTo(instrumentedType)) {
throw new IllegalStateException("Cannot access " + fieldDescription + " from " + instrumentedType);
}
return new FieldSetting(fieldDescription);
}
}
/**
* A factory for a field-setting termination handler that uses a matcher to locate the target field on the insturmented type.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class Implicit implements TerminationHandler.Factory {
/**
* The matcher being used for locating a field.
*/
private final ElementMatcher matcher;
/**
* Creates a factory for a field-setting termination handler.
*
* @param matcher The matcher being used for locating a field.
*/
protected Implicit(ElementMatcher matcher) {
this.matcher = matcher;
}
/**
* {@inheritDoc}
*/
public TerminationHandler make(TypeDescription instrumentedType) {
TypeDefinition current = instrumentedType;
do {
FieldList candidates = current.getDeclaredFields().filter(isVisibleTo(instrumentedType).and(matcher));
if (candidates.size() == 1) {
return new FieldSetting(candidates.getOnly());
} else if (candidates.size() == 2) {
throw new IllegalStateException(matcher + " is ambiguous and resolved: " + candidates);
}
current = current.getSuperClass();
} while (current != null);
throw new IllegalStateException(matcher + " does not locate any accessible fields for " + instrumentedType);
}
}
}
}
/**
* Represents a {@link net.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.Factory methodLocator) {
super(methodLocator,
TargetHandler.ForSelfOrStaticInvocation.Factory.INSTANCE,
Collections.emptyList(),
MethodInvoker.ForContextualInvocation.Factory.INSTANCE,
TerminationHandler.Simple.RETURNING,
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.
*/
@SuppressWarnings("unchecked")
public MethodCall on(Object target) {
return on(target, (Class) target.getClass());
}
/**
* Invokes the specified method on the given instance.
*
* @param target The object on which the method is to be invoked upon.
* @param type The object's type.
* @param The type of the object.
* @return A method call that invokes the provided method on the given object.
*/
public MethodCall on(T target, Class type) {
return new MethodCall(methodLocator,
new TargetHandler.ForValue.Factory(target, TypeDescription.Generic.OfNonGenericType.ForLoadedType.of(type)),
argumentLoaders,
new MethodInvoker.ForVirtualInvocation.Factory(TypeDescription.ForLoadedType.of(type)),
terminationHandler,
assigner,
typing);
}
/**
* Invokes the specified method on an instance that is loaded by the provided stack manipulation.
*
* @param stackManipulation The stack manipulation that loads the instance that the method is invoked upon.
* @param type The type of the loaded instance.
* @return A method call that invokes the provided method on the value of the provided stack manipulation.
*/
public MethodCall on(StackManipulation stackManipulation, Class type) {
return on(stackManipulation, TypeDescription.ForLoadedType.of(type));
}
/**
* Invokes the specified method on an instance that is loaded by the provided stack manipulation.
*
* @param stackManipulation The stack manipulation that loads the instance that the method is invoked upon.
* @param typeDescription The type of the loaded instance.
* @return A method call that invokes the provided method on the value of the provided stack manipulation.
*/
public MethodCall on(StackManipulation stackManipulation, TypeDescription typeDescription) {
return new MethodCall(methodLocator,
new TargetHandler.Simple(typeDescription, stackManipulation),
argumentLoaders,
new MethodInvoker.ForVirtualInvocation.Factory(typeDescription),
terminationHandler,
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 A method 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,
assigner,
typing);
}
/**
* Invokes a method on the object stored in the specified field.
*
* @param name The name of the field.
* @return A method call that invokes the given method on an instance that is read from a field.
*/
public MethodCall onField(String name) {
return onField(name, FieldLocator.ForClassHierarchy.Factory.INSTANCE);
}
/**
* Invokes a method on the object stored in the specified field.
*
* @param name The name of the field.
* @param fieldLocatorFactory The field locator factory to use for locating the field.
* @return A method call that invokes the given method on an instance that is read from a field.
*/
public MethodCall onField(String name, FieldLocator.Factory fieldLocatorFactory) {
return new MethodCall(methodLocator,
new TargetHandler.ForField.Factory(new TargetHandler.ForField.Location.ForImplicitField(name, fieldLocatorFactory)),
argumentLoaders,
MethodInvoker.ForVirtualInvocation.WithImplicitType.INSTANCE,
terminationHandler,
assigner,
typing);
}
/**
* Invokes a method on the object stored in the specified field.
*
* @param field The field on which to invoke the method upon.
* @return A method call that invokes the given method on an instance that is read from a field.
*/
public MethodCall onField(Field field) {
return onField(new FieldDescription.ForLoadedField(field));
}
/**
* Invokes a method on the object stored in the specified field.
*
* @param fieldDescription The field on which to invoke the method upon.
* @return A method call that invokes the given method on an instance that is read from a field.
*/
public MethodCall onField(FieldDescription fieldDescription) {
return new MethodCall(methodLocator,
new TargetHandler.ForField.Factory(new TargetHandler.ForField.Location.ForExplicitField(fieldDescription)),
argumentLoaders,
MethodInvoker.ForVirtualInvocation.WithImplicitType.INSTANCE,
terminationHandler,
assigner,
typing);
}
/**
* Invokes a method on the method call's return value.
*
* @param methodCall The method call that return's value is to be used in this method call
* @return A method call that invokes the given method on an instance that is returned from a method call.
*/
public MethodCall onMethodCall(MethodCall methodCall) {
return new MethodCall(methodLocator,
new TargetHandler.ForMethodCall.Factory(methodCall),
argumentLoaders,
MethodInvoker.ForVirtualInvocation.WithImplicitType.INSTANCE,
terminationHandler,
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.Factory.INSTANCE,
argumentLoaders,
MethodInvoker.ForSuperMethodInvocation.Factory.INSTANCE,
terminationHandler,
assigner,
typing);
}
/**
* Invokes the given method by a Java 8 default 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.Factory.INSTANCE,
argumentLoaders,
MethodInvoker.ForDefaultMethodInvocation.Factory.INSTANCE,
terminationHandler,
assigner,
typing);
}
}
/**
* A {@link MethodCall} that sets the call's result as the value of a field.
*/
@HashCodeAndEqualsPlugin.Enhance
public static class FieldSetting implements Composable {
/**
* The represented method call.
*/
private final MethodCall methodCall;
/**
* Creates a new field setting method call.
*
* @param methodCall The represented method call.
*/
protected FieldSetting(MethodCall methodCall) {
this.methodCall = methodCall;
}
/**
* 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 field being set.
*
* @param assigner The assigner to use.
* @param typing Indicates if dynamic type castings should be attempted for incompatible assignments.
* @return This field-setting method call using the provided assigner.
*/
public Composable withAssigner(Assigner assigner, Assigner.Typing typing) {
return new FieldSetting((MethodCall) methodCall.withAssigner(assigner, typing));
}
/**
* {@inheritDoc}
*/
public InstrumentedType prepare(InstrumentedType instrumentedType) {
return methodCall.prepare(instrumentedType);
}
/**
* {@inheritDoc}
*/
public ByteCodeAppender appender(Target implementationTarget) {
return new ByteCodeAppender.Compound(methodCall.appender(implementationTarget), Appender.INSTANCE);
}
/**
* {@inheritDoc}
*/
public Implementation andThen(Implementation implementation) {
return new Compound(methodCall, implementation);
}
/**
* {@inheritDoc}
*/
public Composable andThen(Composable implementation) {
return new Compound.Composable(methodCall, implementation);
}
/**
* A byte code appender to implement a field-setting method call.
*/
protected enum Appender implements ByteCodeAppender {
/**
* The singleton instance.
*/
INSTANCE;
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Context implementationContext, MethodDescription instrumentedMethod) {
if (!instrumentedMethod.getReturnType().represents(void.class)) {
throw new IllegalStateException("Instrumented method " + instrumentedMethod + " does not return void for field setting method call");
}
return new Size(MethodReturn.VOID.apply(methodVisitor, implementationContext).getMaximalSize(), instrumentedMethod.getStackSize());
}
}
}
/**
* The appender being used to implement a {@link net.bytebuddy.implementation.MethodCall}.
*/
@HashCodeAndEqualsPlugin.Enhance(includeSyntheticFields = true)
protected class Appender implements ByteCodeAppender {
/**
* The implementation target.
*/
private final Implementation.Target implementationTarget;
/**
* The method locator to use.
*/
private final MethodLocator methodLocator;
/**
* The argument providers to use.
*/
private final List argumentProviders;
/**
* The method invoker to use.
*/
private final MethodInvoker methodInvoker;
/**
* The target handler to use.
*/
private final TargetHandler targetHandler;
/**
* The termination handler to use.
*/
private final TerminationHandler terminationHandler;
/**
* Creates a new appender.
*
* @param implementationTarget The implementation target.
* @param terminationHandler The termination handler to use.
*/
protected Appender(Target implementationTarget, TerminationHandler terminationHandler) {
this.implementationTarget = implementationTarget;
methodLocator = MethodCall.this.methodLocator.make(implementationTarget.getInstrumentedType());
argumentProviders = new ArrayList(argumentLoaders.size());
for (ArgumentLoader.Factory factory : argumentLoaders) {
argumentProviders.add(factory.make(implementationTarget));
}
methodInvoker = MethodCall.this.methodInvoker.make(implementationTarget.getInstrumentedType());
targetHandler = MethodCall.this.targetHandler.make(implementationTarget);
this.terminationHandler = terminationHandler;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Context implementationContext, MethodDescription instrumentedMethod) {
TargetHandler.Resolved targetHandler = this.targetHandler.resolve(instrumentedMethod);
return new Size(new StackManipulation.Compound(terminationHandler.prepare(), toStackManipulation(instrumentedMethod,
toInvokedMethod(instrumentedMethod, targetHandler),
targetHandler)).apply(methodVisitor, implementationContext).getMaximalSize(), instrumentedMethod.getStackSize());
}
/**
* Resolves this appender to the method being invoked.
*
* @param instrumentedMethod The instrumented method.
* @param targetHandler The resolved target handler to base the stack manipulation upon.
* @return The invoked method.
*/
protected MethodDescription toInvokedMethod(MethodDescription instrumentedMethod, TargetHandler.Resolved targetHandler) {
return methodLocator.resolve(targetHandler.getTypeDescription(), instrumentedMethod);
}
/**
* Resolves this appender to a stack manipulation.
*
* @param instrumentedMethod The instrumented method.
* @param invokedMethod The invoked method.
* @param targetHandler The resolved target handler to base the stack manipulation upon.
* @return A stack manipulation that represents this method call.
*/
protected StackManipulation toStackManipulation(MethodDescription instrumentedMethod, MethodDescription invokedMethod, TargetHandler.Resolved targetHandler) {
List argumentLoaders = new ArrayList();
for (ArgumentLoader.ArgumentProvider argumentProvider : argumentProviders) {
argumentLoaders.addAll(argumentProvider.resolve(instrumentedMethod, invokedMethod));
}
ParameterList parameters = invokedMethod.getParameters();
if (parameters.size() != argumentLoaders.size()) {
throw new IllegalStateException(invokedMethod + " does not accept " + argumentLoaders.size() + " arguments");
}
Iterator parameterIterator = parameters.iterator();
List argumentInstructions = new ArrayList(argumentLoaders.size());
for (ArgumentLoader argumentLoader : argumentLoaders) {
argumentInstructions.add(argumentLoader.toStackManipulation(parameterIterator.next(), assigner, typing));
}
return new StackManipulation.Compound(
targetHandler.toStackManipulation(invokedMethod, assigner, typing),
new StackManipulation.Compound(argumentInstructions),
methodInvoker.toStackManipulation(invokedMethod, implementationTarget),
terminationHandler.toStackManipulation(invokedMethod, instrumentedMethod, assigner, typing)
);
}
}
}