net.bytebuddy.asm.ClassVisitorFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of redisson-all Show documentation
Show all versions of redisson-all Show documentation
Easy Redis Java client and Real-Time Data Platform. Valkey compatible. Sync/Async/RxJava3/Reactive API. Client side caching. Over 50 Redis based Java objects and services: JCache API, Apache Tomcat, Hibernate, Spring, Set, Multimap, SortedSet, Map, List, Queue, Deque, Semaphore, Lock, AtomicLong, Map Reduce, Bloom filter, Scheduler, RPC
/*
* 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.asm;
import net.bytebuddy.ByteBuddy;
import net.bytebuddy.build.AccessControllerPlugin;
import net.bytebuddy.build.HashCodeAndEqualsPlugin;
import net.bytebuddy.description.field.FieldDescription;
import net.bytebuddy.description.method.MethodDescription;
import net.bytebuddy.description.modifier.FieldManifestation;
import net.bytebuddy.description.modifier.Ownership;
import net.bytebuddy.description.modifier.Visibility;
import net.bytebuddy.description.type.TypeDefinition;
import net.bytebuddy.description.type.TypeDescription;
import net.bytebuddy.dynamic.DynamicType;
import net.bytebuddy.dynamic.loading.MultipleParentClassLoader;
import net.bytebuddy.dynamic.scaffold.TypeValidation;
import net.bytebuddy.dynamic.scaffold.subclass.ConstructorStrategy;
import net.bytebuddy.implementation.ExceptionMethod;
import net.bytebuddy.implementation.FieldAccessor;
import net.bytebuddy.implementation.Implementation;
import net.bytebuddy.implementation.MethodCall;
import net.bytebuddy.implementation.bytecode.ByteCodeAppender;
import net.bytebuddy.implementation.bytecode.Duplication;
import net.bytebuddy.implementation.bytecode.StackManipulation;
import net.bytebuddy.implementation.bytecode.TypeCreation;
import net.bytebuddy.implementation.bytecode.assign.Assigner;
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.ElementMatchers;
import net.bytebuddy.utility.CompoundList;
import net.bytebuddy.utility.OpenedClassReader;
import net.bytebuddy.utility.nullability.MaybeNull;
import net.bytebuddy.jar.asm.*;
import java.lang.reflect.Method;
import java.security.PrivilegedAction;
import java.util.*;
import static net.bytebuddy.matcher.ElementMatchers.*;
/**
* A factory for wrapping a {@link ClassVisitor} in Byte Buddy's package namespace to a
* {@link ClassVisitor} in any other namespace. Note that this API does not allow for the
* passing of {@link Attribute}s. which must be filtered or propagated past this translation.
* If the supplied visitors do not declare a method that Byte Buddy's version of ASM is aware
* of, an {@link UnsupportedOperationException} is thrown.
*
* @param The type of the mapped class visitor.
*/
@HashCodeAndEqualsPlugin.Enhance
public abstract class ClassVisitorFactory {
/**
* The name of the delegate field containing an equivalent visitor.
*/
private static final String DELEGATE = "delegate";
/**
* The name of a map with labels that have been translated previously.
*/
private static final String LABELS = "labels";
/**
* The name of the method that wraps a translated visitor, including a {@code null} check.
*/
private static final String WRAP = "wrap";
/**
* The type of the represented class visitor wrapper.
*/
private final Class type;
/**
* Creates a new factory.
*
* @param type The type of the represented class visitor wrapper.
*/
protected ClassVisitorFactory(Class type) {
this.type = type;
}
/**
* Returns the {@link ClassVisitor} type that this factory represents.
*
* @return The {@link ClassVisitor} type that this factory represents.
*/
public Class getType() {
return type;
}
/**
* Returns a class visitor factory for the supplied {@link ClassVisitor} type.
*
* @param classVisitor The type of the translated class visitor.
* @param The type of the class visitor to map to.
* @return A factory for wrapping {@link ClassVisitor}s in Byte Buddy's and the supplied package namespace.
*/
public static ClassVisitorFactory of(Class classVisitor) {
return of(classVisitor, new ByteBuddy().with(TypeValidation.DISABLED));
}
/**
* Returns a class visitor factory for the supplied {@link ClassVisitor} type.
*
* @param classVisitor The type of the translated {@link ClassVisitor}.
* @param byteBuddy The Byte Buddy instance to use.
* @param The type of the class visitor to map to.
* @return A factory for wrapping {@link ClassVisitor}s in Byte Buddy's and the supplied package namespace.
*/
public static ClassVisitorFactory of(Class classVisitor, ByteBuddy byteBuddy) {
return doPrivileged(new CreateClassVisitorFactory(classVisitor, byteBuddy));
}
/**
* A proxy for {@code java.security.AccessController#doPrivileged} that is activated if available.
*
* @param action The action to execute from a privileged context.
* @param The type of the action's resolved value.
* @return The action's resolved value.
*/
@AccessControllerPlugin.Enhance
private static T doPrivileged(PrivilegedAction action) {
return action.run();
}
/**
* Creates a builder for a visitor type.
*
* @param byteBuddy The Byte Buddy instance to use.
* @param sourceVisitor The visitor type to map from.
* @param targetVisitor The visitor type to map to.
* @param sourceTypePath The {@link TypePath} source type.
* @param targetTypePath The {@link TypePath} target type.
* @param appendix The implementation to append to the constructor.
* @return The created builder.
* @throws Exception If an exception occurs.
*/
private static DynamicType.Builder toVisitorBuilder(ByteBuddy byteBuddy,
Class sourceVisitor,
Class targetVisitor,
@MaybeNull Class sourceTypePath,
@MaybeNull Class targetTypePath,
Implementation appendix) throws Exception {
DynamicType.Builder builder = byteBuddy.subclass(sourceVisitor, ConstructorStrategy.Default.NO_CONSTRUCTORS)
.defineField(DELEGATE, targetVisitor, Visibility.PRIVATE, FieldManifestation.FINAL)
.defineConstructor(Visibility.PUBLIC)
.withParameters(targetVisitor)
.intercept(MethodCall.invoke(sourceVisitor.getDeclaredConstructor(int.class))
.with(OpenedClassReader.ASM_API)
.andThen(FieldAccessor.ofField(DELEGATE).setsArgumentAt(0))
.andThen(appendix))
.defineMethod(WRAP, sourceVisitor, Visibility.PUBLIC, Ownership.STATIC)
.withParameters(targetVisitor)
.intercept(new Implementation.Simple(new NullCheckedConstruction(targetVisitor)));
if (sourceTypePath == null || targetTypePath == null) {
return builder;
} else {
return builder
.defineMethod(TypePathTranslator.NAME, targetTypePath, Visibility.PRIVATE, Ownership.STATIC)
.withParameters(sourceTypePath)
.intercept(new Implementation.Simple(new TypePathTranslator(sourceTypePath, targetTypePath)));
}
}
/**
* Creates a builder for a method visitor type.
*
* @param byteBuddy The Byte Buddy instance to use.
* @param sourceVisitor The visitor type to map from.
* @param targetVisitor The visitor type to map to.
* @param sourceTypePath The {@link TypePath} source type.
* @param targetTypePath The {@link TypePath} target type.
* @param sourceLabel The {@link Label} source type.
* @param targetLabel The {@link Label} target type.
* @param sourceType The {@link Type} source type.
* @param targetType The {@link Type} target type.
* @param sourceHandle The {@link Handle} source type.
* @param targetHandle The {@link Handle} target type.
* @param sourceConstantDynamic The {@link ConstantDynamic} source type.
* @param targetConstantDynamic The {@link ConstantDynamic} target type.
* @return The created builder.
* @throws Exception If an exception occurs.
*/
private static DynamicType.Builder toMethodVisitorBuilder(ByteBuddy byteBuddy,
Class sourceVisitor,
Class targetVisitor,
@MaybeNull Class sourceTypePath,
@MaybeNull Class targetTypePath,
@MaybeNull Class sourceLabel,
@MaybeNull Class targetLabel,
@MaybeNull Class sourceType,
@MaybeNull Class targetType,
@MaybeNull Class sourceHandle,
@MaybeNull Class targetHandle,
@MaybeNull Class sourceConstantDynamic,
@MaybeNull Class targetConstantDynamic) throws Exception {
DynamicType.Builder builder = toVisitorBuilder(byteBuddy,
sourceVisitor,
targetVisitor,
sourceTypePath,
targetTypePath,
FieldAccessor.ofField(LABELS).setsValue(new StackManipulation.Compound(TypeCreation.of(TypeDescription.ForLoadedType.of(HashMap.class)),
Duplication.SINGLE,
MethodInvocation.invoke(TypeDescription.ForLoadedType.of(HashMap.class)
.getDeclaredMethods()
.filter(ElementMatchers.isConstructor().and(ElementMatchers.takesArguments(0)))
.getOnly())), Map.class));
if (sourceLabel != null && targetLabel != null) {
builder = builder
.defineField(LABELS, Map.class, Visibility.PRIVATE, FieldManifestation.FINAL)
.defineMethod(LabelTranslator.NAME, targetLabel, Visibility.PRIVATE)
.withParameters(sourceLabel)
.intercept(new Implementation.Simple(new LabelTranslator(targetLabel)))
.defineMethod(LabelArrayTranslator.NAME, TypeDescription.ArrayProjection.of(TypeDescription.ForLoadedType.ForLoadedType.of(targetLabel)), Visibility.PRIVATE)
.withParameters(TypeDescription.ArrayProjection.of(TypeDescription.ForLoadedType.ForLoadedType.of(sourceLabel)))
.intercept(new Implementation.Simple(new LabelArrayTranslator(sourceLabel, targetLabel)))
.defineMethod(FrameTranslator.NAME, Object[].class, Visibility.PRIVATE)
.withParameters(Object[].class)
.intercept(new Implementation.Simple(new FrameTranslator(sourceLabel, targetLabel)));
}
if (sourceHandle != null && targetHandle != null) {
builder = builder
.defineMethod(HandleTranslator.NAME, targetHandle, Visibility.PRIVATE, Ownership.STATIC)
.withParameters(sourceHandle)
.intercept(new Implementation.Simple(new HandleTranslator(sourceHandle, targetHandle)));
}
if (sourceConstantDynamic != null && targetConstantDynamic != null && sourceHandle != null && targetHandle != null) {
builder = builder
.defineMethod(ConstantDynamicTranslator.NAME, targetConstantDynamic, Visibility.PRIVATE, Ownership.STATIC)
.withParameters(sourceConstantDynamic)
.intercept(new Implementation.Simple(new ConstantDynamicTranslator(sourceConstantDynamic, targetConstantDynamic, sourceHandle, targetHandle)));
}
return builder
.defineMethod(ConstantTranslator.NAME, Object.class, Visibility.PRIVATE, Ownership.STATIC)
.withParameters(Object.class)
.intercept(new Implementation.Simple(new ConstantTranslator(sourceHandle, targetHandle, sourceType, targetType, sourceConstantDynamic, targetConstantDynamic)))
.defineMethod(ConstantArrayTranslator.NAME, Object[].class, Visibility.PRIVATE, Ownership.STATIC)
.withParameters(Object[].class)
.intercept(new Implementation.Simple(new ConstantArrayTranslator()));
}
/**
* Creates an argument loader for a method parameter that requires conversion.
*
* @param source The source type.
* @param target The target type.
* @param method The name of the method.
* @param offset The parameter offset
* @param virtual {@code true} if the invoked method is virtual.
* @return An appropriate argument loader factory.
*/
private static MethodCall.ArgumentLoader.Factory toConvertedParameter(TypeDescription source, Class target, String method, int offset, boolean virtual) {
return new MethodCall.ArgumentLoader.ForStackManipulation(new StackManipulation.Compound(virtual ? MethodVariableAccess.loadThis() : StackManipulation.Trivial.INSTANCE,
MethodVariableAccess.REFERENCE.loadFrom(offset),
MethodInvocation.invoke(source.getDeclaredMethods().filter(named(method)).getOnly())), target);
}
/**
* Creates a wrapper type for an {@link Attribute} to pass attribues along a visitor chain.
*
* @param builder The builder to use for the wrapper type.
* @param source The {@link Attribute} type in the original namespace.
* @param target The {@link Attribute} type in the targeted namespace.
* @param sourceWrapper The wrapper type for the {@link Attribute} type in the original namespace.
* @param targetWrapper The wrapper type for the {@link Attribute} type in the targeted namespace.
* @return The created dynamic type.
* @throws Exception If the dynamic type cannot be built.
*/
private static DynamicType toAttributeWrapper(DynamicType.Builder builder, Class source, Class target, TypeDescription sourceWrapper, TypeDescription targetWrapper) throws Exception {
return builder
.defineField(DELEGATE, target, Visibility.PUBLIC, FieldManifestation.FINAL)
.defineConstructor(Visibility.PUBLIC)
.withParameters(target)
.intercept(MethodCall.invoke(source.getDeclaredConstructor(String.class))
.onSuper()
.with(new StackManipulation.Compound(
MethodVariableAccess.REFERENCE.loadFrom(1),
FieldAccess.forField(new FieldDescription.ForLoadedField(target.getField("type"))).read()), String.class)
.andThen(FieldAccessor.ofField(DELEGATE).setsArgumentAt(0)))
.defineMethod(AttributeTranslator.NAME, source, Visibility.PUBLIC, Ownership.STATIC)
.withParameters(target)
.intercept(new Implementation.Simple(new AttributeTranslator(source, target, sourceWrapper, targetWrapper)))
.method(isProtected())
.intercept(ExceptionMethod.throwing(UnsupportedOperationException.class))
.method(named("isUnknown"))
.intercept(MethodCall.invoke(target.getMethod("isUnknown")).onField(DELEGATE))
.method(named("isCodeAttribute"))
.intercept(MethodCall.invoke(target.getMethod("isCodeAttribute")).onField(DELEGATE))
.make();
}
/**
* Wraps a {@link ClassVisitor} within an instance of the supplied class visitor type.
*
* @param classVisitor The class visitor to wrap.
* @return A class visitor that wraps the supplied class visitor.
*/
public abstract T wrap(ClassVisitor classVisitor);
/**
* Unwraps an instance of the supplied class visitor as a {@link ClassVisitor}.
*
* @param classVisitor The class visitor to unwrap.
* @return A class visitor that unwraps the supplied class visitor.
*/
public abstract ClassVisitor unwrap(T classVisitor);
/**
* An appender that performs a {@code null}-checked construction.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class NullCheckedConstruction implements ByteCodeAppender {
/**
* The constructed type.
*/
private final Class type;
/**
* Creates a byte code appender for creating a {@code null}-checked construction.
*
* @param type The constructed type.
*/
protected NullCheckedConstruction(Class type) {
this.type = type;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor,
Implementation.Context implementationContext,
MethodDescription instrumentedMethod) {
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
Label label = new Label();
methodVisitor.visitJumpInsn(Opcodes.IFNULL, label);
methodVisitor.visitTypeInsn(Opcodes.NEW, implementationContext.getInstrumentedType().getInternalName());
methodVisitor.visitInsn(Opcodes.DUP);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKESPECIAL,
implementationContext.getInstrumentedType().getInternalName(),
MethodDescription.CONSTRUCTOR_INTERNAL_NAME,
Type.getMethodDescriptor(Type.VOID_TYPE, Type.getType(type)),
false);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(label);
implementationContext.getFrameGeneration().same(methodVisitor, instrumentedMethod.getParameters().asTypeList());
methodVisitor.visitInsn(Opcodes.ACONST_NULL);
methodVisitor.visitInsn(Opcodes.ARETURN);
return new Size(3, 1);
}
}
/**
* A method to translate a {@link Label} from one namespace to another.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class LabelTranslator implements ByteCodeAppender {
/**
* The name of the method.
*/
protected static final String NAME = "label";
/**
* The label type that is targeted.
*/
private final Class target;
/**
* Creates a new label translator.
*
* @param target The label type that is targeted.
*/
protected LabelTranslator(Class target) {
this.target = target;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Implementation.Context implementationContext, MethodDescription instrumentedMethod) {
Label nullCheck = new Label(), end = new Label();
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitJumpInsn(Opcodes.IFNONNULL, nullCheck);
methodVisitor.visitInsn(Opcodes.ACONST_NULL);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(nullCheck);
implementationContext.getFrameGeneration().same(methodVisitor, CompoundList.of(
implementationContext.getInstrumentedType(),
instrumentedMethod.getParameters().asTypeList()));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitFieldInsn(Opcodes.GETFIELD,
implementationContext.getInstrumentedType().getInternalName(),
LABELS,
Type.getDescriptor(Map.class));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitMethodInsn(Opcodes.INVOKEINTERFACE, Type.getInternalName(Map.class),
"get",
Type.getMethodDescriptor(Type.getType(Object.class), Type.getType(Object.class)),
true);
methodVisitor.visitTypeInsn(Opcodes.CHECKCAST, Type.getInternalName(target));
methodVisitor.visitVarInsn(Opcodes.ASTORE, 2);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 2);
methodVisitor.visitJumpInsn(Opcodes.IFNONNULL, end);
methodVisitor.visitTypeInsn(Opcodes.NEW, Type.getInternalName(target));
methodVisitor.visitInsn(Opcodes.DUP);
methodVisitor.visitMethodInsn(Opcodes.INVOKESPECIAL,
Type.getInternalName(target),
MethodDescription.CONSTRUCTOR_INTERNAL_NAME,
"()V",
false);
methodVisitor.visitVarInsn(Opcodes.ASTORE, 2);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitFieldInsn(Opcodes.GETFIELD,
implementationContext.getInstrumentedType().getInternalName(),
LABELS,
Type.getDescriptor(Map.class));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 2);
methodVisitor.visitMethodInsn(Opcodes.INVOKEINTERFACE,
Type.getInternalName(Map.class),
"put",
Type.getMethodDescriptor(Type.getType(Object.class), Type.getType(Object.class), Type.getType(Object.class)),
true);
methodVisitor.visitInsn(Opcodes.POP);
methodVisitor.visitLabel(end);
implementationContext.getFrameGeneration().append(methodVisitor,
Collections.singletonList(TypeDescription.ForLoadedType.of(target)),
CompoundList.of(implementationContext.getInstrumentedType(), instrumentedMethod.getParameters().asTypeList()));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 2);
methodVisitor.visitInsn(Opcodes.ARETURN);
return new Size(3, 3);
}
}
/**
* A method to translate an array of {@link Label}s from one namespace to another.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class LabelArrayTranslator implements ByteCodeAppender {
/**
* The name of the method.
*/
protected static final String NAME = "labels";
/**
* The {@link Label} type in the original namespace.
*/
private final Class sourceLabel;
/**
* The {@link Label} type in the targeted namespace.
*/
private final Class targetLabel;
/**
* Creates a new label array translator.
*
* @param sourceLabel The {@link Label} type in the original namespace.
* @param targetLabel The {@link Label} type in the targeted namespace.
*/
protected LabelArrayTranslator(Class sourceLabel, Class targetLabel) {
this.sourceLabel = sourceLabel;
this.targetLabel = targetLabel;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Implementation.Context implementationContext, MethodDescription instrumentedMethod) {
Label nullCheck = new Label(), loop = new Label(), end = new Label();
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitJumpInsn(Opcodes.IFNONNULL, nullCheck);
methodVisitor.visitInsn(Opcodes.ACONST_NULL);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(nullCheck);
implementationContext.getFrameGeneration().same(methodVisitor, CompoundList.of(
implementationContext.getInstrumentedType(),
instrumentedMethod.getParameters().asTypeList()));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitInsn(Opcodes.ARRAYLENGTH);
methodVisitor.visitTypeInsn(Opcodes.ANEWARRAY, Type.getInternalName(targetLabel));
methodVisitor.visitVarInsn(Opcodes.ASTORE, 2);
methodVisitor.visitInsn(Opcodes.ICONST_0);
methodVisitor.visitVarInsn(Opcodes.ISTORE, 3);
methodVisitor.visitLabel(loop);
implementationContext.getFrameGeneration().append(methodVisitor,
Arrays.asList(TypeDescription.ArrayProjection.of(TypeDescription.ForLoadedType.of(targetLabel)), TypeDescription.ForLoadedType.of(int.class)),
CompoundList.of(implementationContext.getInstrumentedType(), instrumentedMethod.getParameters().asTypeList()));
methodVisitor.visitVarInsn(Opcodes.ILOAD, 3);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitInsn(Opcodes.ARRAYLENGTH);
methodVisitor.visitJumpInsn(Opcodes.IF_ICMPGE, end);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 2);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 3);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 3);
methodVisitor.visitInsn(Opcodes.AALOAD);
methodVisitor.visitMethodInsn(Opcodes.INVOKESPECIAL,
implementationContext.getInstrumentedType().getInternalName(),
LabelTranslator.NAME,
Type.getMethodDescriptor(Type.getType(targetLabel), Type.getType(sourceLabel)),
false);
methodVisitor.visitInsn(Opcodes.AASTORE);
methodVisitor.visitIincInsn(3, 1);
methodVisitor.visitJumpInsn(Opcodes.GOTO, loop);
methodVisitor.visitLabel(end);
implementationContext.getFrameGeneration().chop(methodVisitor, 1, CompoundList.of(
Collections.singletonList(implementationContext.getInstrumentedType()),
instrumentedMethod.getParameters().asTypeList(),
Collections.singletonList(TypeDescription.ArrayProjection.of(TypeDescription.ForLoadedType.of(targetLabel)))));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 2);
methodVisitor.visitInsn(Opcodes.ARETURN);
return new Size(5, 4);
}
}
/**
* A method to translate a {@link Handle} from one namespace to another.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class HandleTranslator implements ByteCodeAppender {
/**
* The name of the method.
*/
protected static final String NAME = "handle";
/**
* The {@link Handle} type in the original namespace.
*/
private final Class sourceHandle;
/**
* The {@link Handle} type in the targeted namespace.
*/
private final Class targetHandle;
/**
* Creates a new handle translator.
*
* @param sourceHandle The {@link Handle} type in the original namespace.
* @param targetHandle The {@link Handle} type in the targeted namespace.
*/
protected HandleTranslator(Class sourceHandle, Class targetHandle) {
this.sourceHandle = sourceHandle;
this.targetHandle = targetHandle;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Implementation.Context implementationContext, MethodDescription instrumentedMethod) {
Label nullCheck = new Label();
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitJumpInsn(Opcodes.IFNONNULL, nullCheck);
methodVisitor.visitInsn(Opcodes.ACONST_NULL);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(nullCheck);
implementationContext.getFrameGeneration().same(methodVisitor, instrumentedMethod.getParameters().asTypeList());
methodVisitor.visitTypeInsn(Opcodes.NEW, Type.getInternalName(targetHandle));
methodVisitor.visitInsn(Opcodes.DUP);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceHandle),
"getTag",
Type.getMethodDescriptor(Type.INT_TYPE),
false);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceHandle),
"getOwner",
Type.getMethodDescriptor(Type.getType(String.class)),
false);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceHandle),
"getName",
Type.getMethodDescriptor(Type.getType(String.class)),
false);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL, Type.getInternalName(sourceHandle),
"getDesc",
Type.getMethodDescriptor(Type.getType(String.class)),
false);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceHandle),
"isInterface",
Type.getMethodDescriptor(Type.BOOLEAN_TYPE),
false);
methodVisitor.visitMethodInsn(Opcodes.INVOKESPECIAL,
Type.getInternalName(targetHandle),
MethodDescription.CONSTRUCTOR_INTERNAL_NAME,
Type.getMethodDescriptor(Type.VOID_TYPE, Type.INT_TYPE, Type.getType(String.class), Type.getType(String.class), Type.getType(String.class), Type.BOOLEAN_TYPE),
false);
methodVisitor.visitInsn(Opcodes.ARETURN);
return new Size(7, 1);
}
}
/**
* A method to translate a {@link ConstantDynamic} from one namespace to another.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class ConstantDynamicTranslator implements ByteCodeAppender {
/**
* The name of the method.
*/
protected static final String NAME = "constantDyanmic";
/**
* The {@link ConstantDynamic} type in the original namespace.
*/
private final Class sourceConstantDynamic;
/**
* The {@link ConstantDynamic} type in the targeted namespace.
*/
private final Class targetConstantDynamic;
/**
* The {@link Handle} type in the original namespace.
*/
private final Class sourceHandle;
/**
* The {@link Handle} type in the targeted namespace.
*/
private final Class targetHandle;
/**
* Creates a new constant dynamic translator.
*
* @param sourceConstantDynamic The {@link ConstantDynamic} type in the original namespace.
* @param targetConstantDynamic The {@link ConstantDynamic} type in the targeted namespace.
* @param sourceHandle The {@link Handle} type in the original namespace.
* @param targetHandle The {@link Handle} type in the targeted namespace.
*/
protected ConstantDynamicTranslator(Class sourceConstantDynamic, Class targetConstantDynamic, Class sourceHandle, Class targetHandle) {
this.sourceConstantDynamic = sourceConstantDynamic;
this.targetConstantDynamic = targetConstantDynamic;
this.sourceHandle = sourceHandle;
this.targetHandle = targetHandle;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Implementation.Context implementationContext, MethodDescription instrumentedMethod) {
Label loop = new Label(), end = new Label();
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceConstantDynamic),
"getBootstrapMethodArgumentCount",
Type.getMethodDescriptor(Type.INT_TYPE),
false);
methodVisitor.visitTypeInsn(Opcodes.ANEWARRAY, Type.getInternalName(Object.class));
methodVisitor.visitVarInsn(Opcodes.ASTORE, 1);
methodVisitor.visitInsn(Opcodes.ICONST_0);
methodVisitor.visitVarInsn(Opcodes.ISTORE, 2);
methodVisitor.visitLabel(loop);
implementationContext.getFrameGeneration().append(methodVisitor,
Arrays.asList(TypeDescription.ForLoadedType.of(Object[].class), TypeDescription.ForLoadedType.of(int.class)),
instrumentedMethod.getParameters().asTypeList());
methodVisitor.visitVarInsn(Opcodes.ILOAD, 2);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitInsn(Opcodes.ARRAYLENGTH);
methodVisitor.visitJumpInsn(Opcodes.IF_ICMPGE, end);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 2);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 2);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceConstantDynamic),
"getBootstrapMethodArgument",
Type.getMethodDescriptor(Type.getType(Object.class), Type.INT_TYPE),
false);
methodVisitor.visitMethodInsn(Opcodes.INVOKESTATIC,
implementationContext.getInstrumentedType().getInternalName(),
"ldc",
Type.getMethodDescriptor(Type.getType(Object.class), Type.getType(Object.class)),
false);
methodVisitor.visitInsn(Opcodes.AASTORE);
methodVisitor.visitIincInsn(2, 1);
methodVisitor.visitJumpInsn(Opcodes.GOTO, loop);
methodVisitor.visitLabel(end);
implementationContext.getFrameGeneration().chop(methodVisitor, 1, CompoundList.of(
instrumentedMethod.getParameters().asTypeList(),
TypeDescription.ForLoadedType.of(Object[].class)));
methodVisitor.visitTypeInsn(Opcodes.NEW, Type.getInternalName(targetConstantDynamic));
methodVisitor.visitInsn(Opcodes.DUP);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceConstantDynamic),
"getName",
Type.getMethodDescriptor(Type.getType(String.class)),
false);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceConstantDynamic),
"getDescriptor",
Type.getMethodDescriptor(Type.getType(String.class)),
false);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceConstantDynamic),
"getBootstrapMethod",
Type.getMethodDescriptor(Type.getType(sourceHandle)),
false);
methodVisitor.visitMethodInsn(Opcodes.INVOKESTATIC,
implementationContext.getInstrumentedType().getInternalName(),
HandleTranslator.NAME,
Type.getMethodDescriptor(Type.getType(targetHandle), Type.getType(sourceHandle)),
false);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitMethodInsn(Opcodes.INVOKESPECIAL,
Type.getInternalName(targetConstantDynamic),
MethodDescription.CONSTRUCTOR_INTERNAL_NAME,
Type.getMethodDescriptor(Type.VOID_TYPE, Type.getType(String.class), Type.getType(String.class), Type.getType(targetHandle), Type.getType(Object[].class)),
false);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitMaxs(6, 3);
return new Size(6, 3);
}
}
/**
* A method to translate a constant value from one namespace to another.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class ConstantTranslator implements ByteCodeAppender {
/**
* The name of the method.
*/
protected static final String NAME = "constant";
/**
* The {@link Handle} type in the original namespace.
*/
@MaybeNull
@HashCodeAndEqualsPlugin.ValueHandling(HashCodeAndEqualsPlugin.ValueHandling.Sort.REVERSE_NULLABILITY)
private final Class sourceHandle;
/**
* The {@link Handle} type in the targeted namespace.
*/
@MaybeNull
@HashCodeAndEqualsPlugin.ValueHandling(HashCodeAndEqualsPlugin.ValueHandling.Sort.REVERSE_NULLABILITY)
private final Class targetHandle;
/**
* The {@link Type} type in the original namespace.
*/
@MaybeNull
@HashCodeAndEqualsPlugin.ValueHandling(HashCodeAndEqualsPlugin.ValueHandling.Sort.REVERSE_NULLABILITY)
private final Class sourceType;
/**
* The {@link Type} type in the targeted namespace.
*/
@MaybeNull
@HashCodeAndEqualsPlugin.ValueHandling(HashCodeAndEqualsPlugin.ValueHandling.Sort.REVERSE_NULLABILITY)
private final Class targetType;
/**
* The {@link ConstantDynamic} type in the original namespace.
*/
@MaybeNull
@HashCodeAndEqualsPlugin.ValueHandling(HashCodeAndEqualsPlugin.ValueHandling.Sort.REVERSE_NULLABILITY)
private final Class sourceConstantDynamic;
/**
* The {@link ConstantDynamic} type in the targeted namespace.
*/
@MaybeNull
@HashCodeAndEqualsPlugin.ValueHandling(HashCodeAndEqualsPlugin.ValueHandling.Sort.REVERSE_NULLABILITY)
private final Class targetConstantDynamic;
/**
* Creates a new constant translator.
*
* @param sourceHandle The {@link Handle} type in the original namespace.
* @param targetHandle The {@link Handle} type in the targeted namespace.
* @param sourceType The {@link Type} type in the original namespace.
* @param targetType The {@link Type} type in the targeted namespace.
* @param sourceConstantDynamic The {@link ConstantDynamic} type in the original namespace.
* @param targetConstantDynamic The {@link ConstantDynamic} type in the targeted namespace.
*/
protected ConstantTranslator(@MaybeNull Class sourceHandle,
@MaybeNull Class targetHandle,
@MaybeNull Class sourceType,
@MaybeNull Class targetType,
@MaybeNull Class sourceConstantDynamic,
@MaybeNull Class targetConstantDynamic) {
this.sourceHandle = sourceHandle;
this.targetHandle = targetHandle;
this.sourceType = sourceType;
this.targetType = targetType;
this.sourceConstantDynamic = sourceConstantDynamic;
this.targetConstantDynamic = targetConstantDynamic;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Implementation.Context implementationContext, MethodDescription instrumentedMethod) {
Label noType = new Label(), noHandle = new Label(), noConstantDynamic = new Label();
if (sourceType != null && targetType != null) {
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitTypeInsn(Opcodes.INSTANCEOF, Type.getInternalName(sourceType));
methodVisitor.visitJumpInsn(Opcodes.IFEQ, noType);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitTypeInsn(Opcodes.CHECKCAST, Type.getInternalName(sourceType));
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceType),
"getDescriptor",
Type.getMethodDescriptor(Type.getType(String.class)),
false);
methodVisitor.visitMethodInsn(Opcodes.INVOKESTATIC,
Type.getInternalName(targetType),
"getType",
Type.getMethodDescriptor(Type.getType(targetType), Type.getType(String.class)),
false);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(noType);
implementationContext.getFrameGeneration().same(methodVisitor, instrumentedMethod.getParameters().asTypeList());
}
if (sourceHandle != null && targetHandle != null) {
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitTypeInsn(Opcodes.INSTANCEOF, Type.getInternalName(sourceHandle));
methodVisitor.visitJumpInsn(Opcodes.IFEQ, noHandle);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitTypeInsn(Opcodes.CHECKCAST, Type.getInternalName(sourceHandle));
methodVisitor.visitMethodInsn(Opcodes.INVOKESTATIC,
implementationContext.getInstrumentedType().getInternalName(),
HandleTranslator.NAME,
Type.getMethodDescriptor(Type.getType(targetHandle), Type.getType(sourceHandle)),
false);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(noHandle);
implementationContext.getFrameGeneration().same(methodVisitor, instrumentedMethod.getParameters().asTypeList());
}
if (sourceConstantDynamic != null && targetConstantDynamic != null) {
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitTypeInsn(Opcodes.INSTANCEOF, Type.getInternalName(sourceConstantDynamic));
methodVisitor.visitJumpInsn(Opcodes.IFEQ, noConstantDynamic);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitTypeInsn(Opcodes.CHECKCAST, Type.getInternalName(sourceConstantDynamic));
methodVisitor.visitMethodInsn(Opcodes.INVOKESTATIC,
implementationContext.getInstrumentedType().getInternalName(),
ConstantDynamicTranslator.NAME,
Type.getMethodDescriptor(Type.getType(targetConstantDynamic), Type.getType(sourceConstantDynamic)),
false);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(noConstantDynamic);
implementationContext.getFrameGeneration().same(methodVisitor, instrumentedMethod.getParameters().asTypeList());
}
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitInsn(Opcodes.ARETURN);
return new Size(1, 1);
}
}
/**
* A method to translate an array of constants from one namespace to another.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class ConstantArrayTranslator implements ByteCodeAppender {
/**
* The name of the method.
*/
protected static final String NAME = "constants";
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Implementation.Context implementationContext, MethodDescription instrumentedMethod) {
Label nullCheck = new Label(), loop = new Label(), end = new Label();
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitJumpInsn(Opcodes.IFNONNULL, nullCheck);
methodVisitor.visitInsn(Opcodes.ACONST_NULL);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(nullCheck);
implementationContext.getFrameGeneration().same(methodVisitor, instrumentedMethod.getParameters().asTypeList());
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitInsn(Opcodes.ARRAYLENGTH);
methodVisitor.visitTypeInsn(Opcodes.ANEWARRAY, Type.getInternalName(Object.class));
methodVisitor.visitVarInsn(Opcodes.ASTORE, 1);
methodVisitor.visitInsn(Opcodes.ICONST_0);
methodVisitor.visitVarInsn(Opcodes.ISTORE, 2);
methodVisitor.visitLabel(loop);
implementationContext.getFrameGeneration().append(methodVisitor,
Arrays.asList(TypeDescription.ForLoadedType.of(Object[].class), TypeDescription.ForLoadedType.of(int.class)),
instrumentedMethod.getParameters().asTypeList());
methodVisitor.visitVarInsn(Opcodes.ILOAD, 2);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitInsn(Opcodes.ARRAYLENGTH);
methodVisitor.visitJumpInsn(Opcodes.IF_ICMPGE, end);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 2);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 2);
methodVisitor.visitInsn(Opcodes.AALOAD);
methodVisitor.visitMethodInsn(Opcodes.INVOKESTATIC,
implementationContext.getInstrumentedType().getInternalName(),
ConstantTranslator.NAME,
Type.getMethodDescriptor(Type.getType(Object.class), Type.getType(Object.class)),
false);
methodVisitor.visitInsn(Opcodes.AASTORE);
methodVisitor.visitIincInsn(2, 1);
methodVisitor.visitJumpInsn(Opcodes.GOTO, loop);
methodVisitor.visitLabel(end);
implementationContext.getFrameGeneration().chop(methodVisitor, 1, CompoundList.of(
instrumentedMethod.getParameters().asTypeList(),
TypeDescription.ForLoadedType.of(Object[].class)));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitInsn(Opcodes.ARETURN);
return new Size(4, 3);
}
}
/**
* A method to translate a stack map frame array from one namespace to another.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class FrameTranslator implements ByteCodeAppender {
/**
* The name of the method.
*/
protected static final String NAME = "frames";
/**
* The {@link Label} type in the original namespace.
*/
private final Class sourceLabel;
/**
* The {@link Label} type in the targeted namespace.
*/
private final Class targetLabel;
/**
* Creates a new frame translator.
*
* @param sourceLabel The {@link Label} type in the original namespace.
* @param targetLabel The {@link Label} type in the targeted namespace.
*/
protected FrameTranslator(Class sourceLabel, Class targetLabel) {
this.sourceLabel = sourceLabel;
this.targetLabel = targetLabel;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Implementation.Context implementationContext, MethodDescription instrumentedMethod) {
Label nullCheck = new Label(), loop = new Label(), store = new Label(), end = new Label(), label = new Label();
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitJumpInsn(Opcodes.IFNONNULL, nullCheck);
methodVisitor.visitInsn(Opcodes.ACONST_NULL);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(nullCheck);
implementationContext.getFrameGeneration().same(methodVisitor, CompoundList.of(
implementationContext.getInstrumentedType(),
instrumentedMethod.getParameters().asTypeList()));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitInsn(Opcodes.ARRAYLENGTH);
methodVisitor.visitTypeInsn(Opcodes.ANEWARRAY, Type.getInternalName(Object.class));
methodVisitor.visitVarInsn(Opcodes.ASTORE, 2);
methodVisitor.visitInsn(Opcodes.ICONST_0);
methodVisitor.visitVarInsn(Opcodes.ISTORE, 3);
methodVisitor.visitLabel(loop);
implementationContext.getFrameGeneration().append(methodVisitor,
Arrays.asList(TypeDescription.ForLoadedType.of(Object[].class), TypeDescription.ForLoadedType.of(int.class)),
CompoundList.of(implementationContext.getInstrumentedType(), instrumentedMethod.getParameters().asTypeList()));
methodVisitor.visitVarInsn(Opcodes.ILOAD, 3);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitInsn(Opcodes.ARRAYLENGTH);
methodVisitor.visitJumpInsn(Opcodes.IF_ICMPGE, end);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 2);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 3);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 3);
methodVisitor.visitInsn(Opcodes.AALOAD);
methodVisitor.visitTypeInsn(Opcodes.INSTANCEOF, Type.getInternalName(sourceLabel));
methodVisitor.visitJumpInsn(Opcodes.IFEQ, label);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 3);
methodVisitor.visitInsn(Opcodes.AALOAD);
methodVisitor.visitTypeInsn(Opcodes.CHECKCAST, Type.getInternalName(sourceLabel));
methodVisitor.visitMethodInsn(Opcodes.INVOKESPECIAL,
implementationContext.getInstrumentedType().getInternalName(),
LabelTranslator.NAME,
Type.getMethodDescriptor(Type.getType(targetLabel), Type.getType(sourceLabel)),
false);
methodVisitor.visitJumpInsn(Opcodes.GOTO, store);
methodVisitor.visitLabel(label);
implementationContext.getFrameGeneration().full(methodVisitor,
Arrays.asList(TypeDescription.ForLoadedType.of(Object[].class), TypeDescription.ForLoadedType.of(int.class)),
CompoundList.of(
Collections.singletonList(implementationContext.getInstrumentedType()),
instrumentedMethod.getParameters().asTypeList(),
Arrays.asList(TypeDescription.ForLoadedType.of(Object[].class), TypeDescription.ForLoadedType.of(int.class))));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 1);
methodVisitor.visitVarInsn(Opcodes.ILOAD, 3);
methodVisitor.visitInsn(Opcodes.AALOAD);
methodVisitor.visitLabel(store);
implementationContext.getFrameGeneration().full(methodVisitor,
Arrays.asList(TypeDescription.ForLoadedType.of(Object[].class), TypeDescription.ForLoadedType.of(int.class), TypeDescription.ForLoadedType.of(Object.class)),
CompoundList.of(
Collections.singletonList(implementationContext.getInstrumentedType()),
instrumentedMethod.getParameters().asTypeList(),
Arrays.asList(TypeDescription.ForLoadedType.of(Object[].class), TypeDescription.ForLoadedType.of(int.class))));
methodVisitor.visitInsn(Opcodes.AASTORE);
methodVisitor.visitIincInsn(3, 1);
methodVisitor.visitJumpInsn(Opcodes.GOTO, loop);
methodVisitor.visitLabel(end);
implementationContext.getFrameGeneration().chop(methodVisitor, 1, CompoundList.of(
Collections.singletonList(implementationContext.getInstrumentedType()),
instrumentedMethod.getParameters().asTypeList(),
Collections.singletonList(TypeDescription.ForLoadedType.of(Object[].class))));
methodVisitor.visitVarInsn(Opcodes.ALOAD, 2);
methodVisitor.visitInsn(Opcodes.ARETURN);
return new Size(5, 4);
}
}
/**
* A method to translate a {@link TypePath} type from one namespace to another.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class TypePathTranslator implements ByteCodeAppender {
/**
* The name of the method.
*/
protected static final String NAME = "typePath";
/**
* The {@link TypePath} type in the original namespace.
*/
private final Class sourceTypePath;
/**
* The {@link TypePath} type in the targeted namespace.
*/
private final Class targetTypePath;
/**
* Creates a new type path translator.
*
* @param sourceTypePath The {@link TypePath} type in the original namespace.
* @param targetTypePath The {@link TypePath} type in the targeted namespace.
*/
protected TypePathTranslator(Class sourceTypePath, Class targetTypePath) {
this.sourceTypePath = sourceTypePath;
this.targetTypePath = targetTypePath;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Implementation.Context implementationContext, MethodDescription instrumentedMethod) {
Label nullCheck = new Label(), end = new Label();
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitJumpInsn(Opcodes.IFNONNULL, nullCheck);
methodVisitor.visitInsn(Opcodes.ACONST_NULL);
methodVisitor.visitJumpInsn(Opcodes.GOTO, end);
implementationContext.getFrameGeneration().same(methodVisitor, instrumentedMethod.getParameters().asTypeList());
methodVisitor.visitLabel(nullCheck);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
Type.getInternalName(sourceTypePath),
"toString",
Type.getMethodDescriptor(Type.getType(String.class)),
false);
methodVisitor.visitMethodInsn(Opcodes.INVOKESTATIC,
Type.getInternalName(targetTypePath),
"fromString",
Type.getMethodDescriptor(Type.getType(targetTypePath), Type.getType(String.class)),
false);
methodVisitor.visitLabel(end);
implementationContext.getFrameGeneration().same1(methodVisitor,
TypeDescription.ForLoadedType.of(targetTypePath),
instrumentedMethod.getParameters().asTypeList());
methodVisitor.visitInsn(Opcodes.ARETURN);
return new Size(1, 2);
}
}
/**
* A method to wrap an {@link Attribute}.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class AttributeTranslator implements ByteCodeAppender {
/**
* The name of the method.
*/
protected static final String NAME = "attribute";
/**
* The {@link Attribute} type in the original namespace.
*/
private final Class sourceAttribute;
/**
* The {@link Attribute} type in the targeted namespace.
*/
private final Class targetAttribute;
/**
* The wrapper type for the {@link Attribute} type in the original namespace.
*/
private final TypeDescription sourceWrapper;
/**
* The wrapper type for the {@link Attribute} type in the targeted namespace.
*/
private final TypeDescription targetWrapper;
/**
* Creates a new attribute translator.
*
* @param sourceAttribute The {@link Attribute} type in the original namespace.
* @param targetAttribute The {@link Attribute} type in the targeted namespace.
* @param sourceWrapper The wrapper type for the {@link Attribute} type in the original namespace.
* @param targetWrapper The wrapper type for the {@link Attribute} type in the targeted namespace.
*/
protected AttributeTranslator(Class sourceAttribute, Class targetAttribute, TypeDescription sourceWrapper, TypeDescription targetWrapper) {
this.sourceAttribute = sourceAttribute;
this.targetAttribute = targetAttribute;
this.sourceWrapper = sourceWrapper;
this.targetWrapper = targetWrapper;
}
/**
* {@inheritDoc}
*/
public Size apply(MethodVisitor methodVisitor, Implementation.Context implementationContext, MethodDescription instrumentedMethod) {
Label nullCheck = new Label(), wrapperCheck = new Label();
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitJumpInsn(Opcodes.IFNONNULL, nullCheck);
methodVisitor.visitInsn(Opcodes.ACONST_NULL);
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(nullCheck);
implementationContext.getFrameGeneration().same(methodVisitor, instrumentedMethod.getParameters().asTypeList());
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitTypeInsn(Opcodes.INSTANCEOF, targetWrapper.getInternalName());
methodVisitor.visitJumpInsn(Opcodes.IFEQ, wrapperCheck);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitTypeInsn(Opcodes.CHECKCAST, targetWrapper.getInternalName());
methodVisitor.visitFieldInsn(Opcodes.GETFIELD, targetWrapper.getInternalName(), DELEGATE, Type.getDescriptor(sourceAttribute));
methodVisitor.visitInsn(Opcodes.ARETURN);
methodVisitor.visitLabel(wrapperCheck);
implementationContext.getFrameGeneration().same(methodVisitor, instrumentedMethod.getParameters().asTypeList());
methodVisitor.visitTypeInsn(Opcodes.NEW, sourceWrapper.getInternalName());
methodVisitor.visitInsn(Opcodes.DUP);
methodVisitor.visitVarInsn(Opcodes.ALOAD, 0);
methodVisitor.visitMethodInsn(Opcodes.INVOKESPECIAL,
sourceWrapper.getInternalName(),
MethodDescription.CONSTRUCTOR_INTERNAL_NAME,
Type.getMethodDescriptor(Type.VOID_TYPE, Type.getType(targetAttribute)),
false);
methodVisitor.visitInsn(Opcodes.ARETURN);
return new Size(3, 1);
}
}
/**
* A factory for creating a wrapper for a {@link ClassVisitor}.
*
* @param The type of the class visitor to map to.
*/
@HashCodeAndEqualsPlugin.Enhance
protected static class CreateClassVisitorFactory implements PrivilegedAction> {
/**
* The type of the translated {@link ClassVisitor}.
*/
private final Class classVisitor;
/**
* The Byte Buddy instance to use.
*/
private final ByteBuddy byteBuddy;
/**
* Creates a new factory for a class visitor wrapper.
*
* @param classVisitor The type of the translated {@link ClassVisitor}.
* @param byteBuddy The Byte Buddy instance to use.
*/
protected CreateClassVisitorFactory(Class classVisitor, ByteBuddy byteBuddy) {
this.classVisitor = classVisitor;
this.byteBuddy = byteBuddy;
}
/**
* {@inheritDoc}
*/
public ClassVisitorFactory run() {
if (!ClassVisitor.class.getSimpleName().equals(classVisitor.getSimpleName())) {
throw new IllegalArgumentException("Expected a class named " + ClassVisitor.class.getSimpleName() + ": " + classVisitor);
}
try {
String prefix = classVisitor.getPackage().getName();
Map, Class> utilities = new HashMap, Class>();
for (Class type : Arrays.>asList(
Attribute.class,
Label.class,
Type.class,
TypePath.class,
Handle.class,
ConstantDynamic.class
)) {
Class utility;
try {
utility = Class.forName(prefix + "." + type.getSimpleName(), false, classVisitor.getClassLoader());
} catch (ClassNotFoundException ignored) {
continue;
}
utilities.put(type, utility);
}
if (utilities.containsKey(Label.class)) {
utilities.put(Label[].class, Class.forName("[L" + utilities.get(Label.class).getName() + ";", false, classVisitor.getClassLoader()));
}
Map, Class> equivalents = new HashMap, Class>();
Map, DynamicType.Builder> builders = new HashMap, DynamicType.Builder>();
for (Class type : Arrays.>asList(
ClassVisitor.class,
AnnotationVisitor.class,
ModuleVisitor.class,
RecordComponentVisitor.class,
FieldVisitor.class,
MethodVisitor.class
)) {
Class equivalent;
try {
equivalent = Class.forName(prefix + "." + type.getSimpleName(), false, classVisitor.getClassLoader());
} catch (ClassNotFoundException ignored) {
continue;
}
DynamicType.Builder wrapper, unwrapper;
if (type == MethodVisitor.class) {
wrapper = toMethodVisitorBuilder(byteBuddy,
type, equivalent,
TypePath.class, utilities.get(TypePath.class),
Label.class, utilities.get(Label.class),
Type.class, utilities.get(Type.class),
Handle.class, utilities.get(Handle.class),
ConstantDynamic.class, utilities.get(ConstantDynamic.class));
unwrapper = toMethodVisitorBuilder(byteBuddy,
equivalent, type,
utilities.get(TypePath.class), TypePath.class,
utilities.get(Label.class), Label.class,
utilities.get(Type.class), Type.class,
utilities.get(Handle.class), Handle.class,
utilities.get(ConstantDynamic.class), ConstantDynamic.class);
} else {
wrapper = toVisitorBuilder(byteBuddy,
type, equivalent,
TypePath.class, utilities.get(TypePath.class),
new Implementation.Simple(MethodReturn.VOID));
unwrapper = toVisitorBuilder(byteBuddy,
equivalent, type,
utilities.get(TypePath.class), TypePath.class,
new Implementation.Simple(MethodReturn.VOID));
}
equivalents.put(type, equivalent);
builders.put(type, wrapper);
builders.put(equivalent, unwrapper);
}
List dynamicTypes = new ArrayList();
Map, TypeDescription> generated = new HashMap, TypeDescription>();
DynamicType sourceAttribute, targetAttribute;
if (utilities.containsKey(Attribute.class)) {
DynamicType.Builder source = byteBuddy.subclass(Attribute.class, ConstructorStrategy.Default.NO_CONSTRUCTORS);
DynamicType.Builder target = byteBuddy.subclass(utilities.get(Attribute.class), ConstructorStrategy.Default.NO_CONSTRUCTORS);
sourceAttribute = toAttributeWrapper(source, Attribute.class, utilities.get(Attribute.class), source.toTypeDescription(), target.toTypeDescription());
dynamicTypes.add(sourceAttribute);
targetAttribute = toAttributeWrapper(target, utilities.get(Attribute.class), Attribute.class, target.toTypeDescription(), source.toTypeDescription());
dynamicTypes.add(targetAttribute);
} else {
sourceAttribute = null;
targetAttribute = null;
}
for (Map.Entry, Class> entry : equivalents.entrySet()) {
DynamicType.Builder wrapper = builders.get(entry.getKey()), unwrapper = builders.get(entry.getValue());
for (Method method : entry.getKey().getMethods()) {
if (method.getDeclaringClass() == Object.class) {
continue;
}
Class[] parameter = method.getParameterTypes(), match = new Class[parameter.length];
List left = new ArrayList(parameter.length);
List right = new ArrayList(match.length);
boolean unsupported = false, unresolved = false;
int offset = 1;
for (int index = 0; index < parameter.length; index++) {
if (entry.getKey() == MethodVisitor.class && parameter[index] == Label.class) {
match[index] = utilities.get(Label.class);
left.add(toConvertedParameter(builders.get(entry.getKey()).toTypeDescription(), match[index], LabelTranslator.NAME, offset, true));
right.add(toConvertedParameter(builders.get(entry.getValue()).toTypeDescription(), parameter[index], LabelTranslator.NAME, offset, true));
} else if (entry.getKey() == MethodVisitor.class && parameter[index] == Label[].class) {
match[index] = utilities.get(Label[].class);
left.add(toConvertedParameter(builders.get(entry.getKey()).toTypeDescription(), match[index], LabelArrayTranslator.NAME, offset, true));
right.add(toConvertedParameter(builders.get(entry.getValue()).toTypeDescription(), parameter[index], LabelArrayTranslator.NAME, offset, true));
} else if (parameter[index] == TypePath.class) {
match[index] = utilities.get(TypePath.class);
left.add(toConvertedParameter(builders.get(entry.getKey()).toTypeDescription(), match[index], TypePathTranslator.NAME, offset, false));
right.add(toConvertedParameter(builders.get(entry.getValue()).toTypeDescription(), parameter[index], TypePathTranslator.NAME, offset, false));
} else if (entry.getKey() == MethodVisitor.class && parameter[index] == Handle.class) {
match[index] = utilities.get(Handle.class);
left.add(toConvertedParameter(builders.get(entry.getKey()).toTypeDescription(), match[index], HandleTranslator.NAME, offset, false));
right.add(toConvertedParameter(builders.get(entry.getValue()).toTypeDescription(), parameter[index], HandleTranslator.NAME, offset, false));
} else if (entry.getKey() == MethodVisitor.class && parameter[index] == Object.class) {
match[index] = Object.class;
left.add(toConvertedParameter(builders.get(entry.getKey()).toTypeDescription(), Object.class, ConstantTranslator.NAME, offset, false));
right.add(toConvertedParameter(builders.get(entry.getValue()).toTypeDescription(), Object.class, ConstantTranslator.NAME, offset, false));
} else if (entry.getKey() == MethodVisitor.class && parameter[index] == Object[].class) {
match[index] = Object[].class;
if (method.getName().equals("visitFrame")) {
left.add(toConvertedParameter(builders.get(entry.getKey()).toTypeDescription(), Object[].class, FrameTranslator.NAME, offset, true));
right.add(toConvertedParameter(builders.get(entry.getValue()).toTypeDescription(), Object[].class, FrameTranslator.NAME, offset, true));
} else {
left.add(toConvertedParameter(builders.get(entry.getKey()).toTypeDescription(), Object[].class, ConstantArrayTranslator.NAME, offset, false));
right.add(toConvertedParameter(builders.get(entry.getValue()).toTypeDescription(), Object[].class, ConstantArrayTranslator.NAME, offset, false));
}
} else if (parameter[index] == Attribute.class) {
match[index] = utilities.get(Attribute.class);
if (sourceAttribute != null && targetAttribute != null) {
left.add(toConvertedParameter(targetAttribute.getTypeDescription(), utilities.get(Attribute.class), AttributeTranslator.NAME, offset, false));
right.add(toConvertedParameter(sourceAttribute.getTypeDescription(), Attribute.class, AttributeTranslator.NAME, offset, false));
} else {
unsupported = true;
}
} else {
match[index] = parameter[index];
left.add(new MethodCall.ArgumentLoader.ForMethodParameter.Factory(index));
right.add(new MethodCall.ArgumentLoader.ForMethodParameter.Factory(index));
}
if (match[index] == null) {
unresolved = true;
break;
}
offset += parameter[index] == long.class || parameter[index] == double.class ? 2 : 1;
}
Method target;
if (unresolved) {
target = null;
unsupported = true;
} else {
try {
target = entry.getValue().getMethod(method.getName(), match);
} catch (NoSuchMethodException ignored) {
target = null;
unsupported = true;
}
}
if (unsupported) {
wrapper = wrapper.method(is(method)).intercept(ExceptionMethod.throwing(UnsupportedOperationException.class));
if (target != null) {
unwrapper = unwrapper.method(is(target)).intercept(ExceptionMethod.throwing(UnsupportedOperationException.class));
}
} else {
MethodCall wrapping = MethodCall.invoke(target).onField(DELEGATE).with(left);
MethodCall unwrapping = MethodCall.invoke(method).onField(DELEGATE).with(right);
Class returned = equivalents.get(method.getReturnType());
if (returned != null) {
wrapping = MethodCall.invoke(builders.get(method.getReturnType())
.toTypeDescription()
.getDeclaredMethods()
.filter(named(WRAP))
.getOnly()).withMethodCall(wrapping);
unwrapping = MethodCall.invoke(builders.get(returned).toTypeDescription()
.getDeclaredMethods()
.filter(named(WRAP))
.getOnly()).withMethodCall(unwrapping);
}
wrapper = wrapper.method(is(method)).intercept(wrapping);
unwrapper = unwrapper.method(is(target)).intercept(unwrapping);
}
}
DynamicType left = wrapper.make(), right = unwrapper.make();
generated.put(entry.getKey(), left.getTypeDescription());
generated.put(entry.getValue(), right.getTypeDescription());
dynamicTypes.add(left);
dynamicTypes.add(right);
}
ClassLoader classLoader = new MultipleParentClassLoader.Builder(false)
.appendMostSpecific(ClassVisitor.class, classVisitor)
.build();
@SuppressWarnings("unchecked")
ClassVisitorFactory factory = byteBuddy.subclass(ClassVisitorFactory.class, ConstructorStrategy.Default.IMITATE_SUPER_CLASS_OPENING)
.method(named("wrap")).intercept(MethodCall.construct(generated.get(classVisitor)
.getDeclaredMethods()
.filter(ElementMatchers.isConstructor())
.getOnly()).withArgument(0))
.method(named("unwrap")).intercept(MethodCall.construct(generated.get(ClassVisitor.class)
.getDeclaredMethods()
.filter(ElementMatchers.isConstructor())
.getOnly()).withArgument(0).withAssigner(Assigner.DEFAULT, Assigner.Typing.DYNAMIC))
.make()
.include(dynamicTypes)
.load(classLoader)
.getLoaded()
.getConstructor(Class.class)
.newInstance(classVisitor);
if (classLoader instanceof MultipleParentClassLoader
&& classLoader != ClassVisitor.class.getClassLoader()
&& classLoader != classVisitor.getClassLoader()
&& !((MultipleParentClassLoader) classLoader).seal()) {
throw new IllegalStateException("Failed to seal multiple parent class loader: " + classLoader);
}
return factory;
} catch (Exception exception) {
throw new IllegalArgumentException("Failed to generate factory for " + classVisitor.getName(), exception);
}
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy