org.datanucleus.enhancer.asm.JdoPropertyGetterAdapter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of datanucleus-enhancer Show documentation
Show all versions of datanucleus-enhancer Show documentation
DataNucleus Enhancer is a Java byte-code enhancer for use with DataNucleus.
The newest version!
/**********************************************************************
Copyright (c) 2007 Andy Jefferson and others. All rights reserved.
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.
Contributors:
...
**********************************************************************/
package org.datanucleus.enhancer.asm;
import javax.jdo.spi.PersistenceCapable;
import org.datanucleus.enhancer.ClassEnhancer;
import org.datanucleus.enhancer.ClassMethod;
import org.datanucleus.enhancer.DataNucleusEnhancer;
import org.datanucleus.metadata.AbstractClassMetaData;
import org.datanucleus.metadata.AbstractMemberMetaData;
import org.datanucleus.util.Localiser;
import org.objectweb.asm.AnnotationVisitor;
import org.objectweb.asm.Attribute;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.Label;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;
/**
* Adapter for property getter methods in JDO-enabled classes.
* This adapter processes the getXXX method and
*
* - Creates aaaGetXXX with the same code as was present in getXXX
* - Changes getXXX to have the code below
*
* When detachable this will be (CHECK_READ variant)
*
* YYY aaaGetZZZ()
* {
* if (aaaFlags > 0 && aaaStateManager != null && !aaaStateManager.isLoaded(this, 0))
* return (Integer) aaaStateManager.getObjectField(this, 0, aaaGetXXX());
* if (aaaIsDetached() && !((BitSet) aaaDetachedState[2]).get(0))
* throw new DetachedFieldAccessException
* ("You have just attempted to access property \"id\" yet this field was not detached ...");
* return aaaGetXXX();
* }
*
* and when not detachable
*
* YYY aaaGetZZZ()
* {
* if (aaaFlags > 0 && aaaStateManager != null && !aaaStateManager.isLoaded(this, 0))
* return (Integer) aaaStateManager.getObjectField(this, 0, aaaGetXXX());
* return aaaGetXXX();
* }
*
* There are other variants for MEDIATE_READ and NORMAL_READ
*/
public class JdoPropertyGetterAdapter extends MethodVisitor
{
/** Localisation of messages. */
protected static final Localiser LOCALISER = Localiser.getInstance("org.datanucleus.enhancer.Localisation", ClassEnhancer.class.getClassLoader());
/** The enhancer for this class. */
protected ASMClassEnhancer enhancer;
/** Name for the method being adapted. */
protected String methodName;
/** Descriptor for the method being adapted. */
protected String methodDescriptor;
/** MetaData for the property. */
protected AbstractMemberMetaData mmd;
/** Visitor for the aaaGetXXX method. */
protected MethodVisitor visitor = null;
/**
* Constructor for the method adapter.
* @param mv MethodVisitor
* @param enhancer ClassEnhancer for the class with the method
* @param methodName Name of the method
* @param methodDesc Method descriptor
* @param mmd MetaData for the property
* @param cv ClassVisitor
*/
public JdoPropertyGetterAdapter(MethodVisitor mv, ASMClassEnhancer enhancer, String methodName, String methodDesc,
AbstractMemberMetaData mmd, ClassVisitor cv)
{
super(ASMClassEnhancer.ASM_API_VERSION, mv);
this.enhancer = enhancer;
this.methodName = methodName;
this.methodDescriptor = methodDesc;
this.mmd = mmd;
// Generate aaaGetXXX method to include code that this getXXX currently has
int access = (mmd.isPublic() ? Opcodes.ACC_PUBLIC : 0) |
(mmd.isProtected() ? Opcodes.ACC_PROTECTED : 0) |
(mmd.isPrivate() ? Opcodes.ACC_PRIVATE : 0) |
(mmd.isAbstract() ? Opcodes.ACC_ABSTRACT : 0);
this.visitor = cv.visitMethod(access, enhancer.getGetMethodPrefixMethodName() + mmd.getName(), methodDesc,
null, null);
}
/**
* Method called at the end of visiting the getXXX method.
* This is used to add the aaaGetXXX method with the same code as is present originally in the getXXX method.
*/
public void visitEnd()
{
visitor.visitEnd();
if (DataNucleusEnhancer.LOGGER.isDebugEnabled())
{
String msg = ClassMethod.getMethodAdditionMessage(enhancer.getGetMethodPrefixMethodName() + mmd.getName(),
mmd.getType(), null, null);
DataNucleusEnhancer.LOGGER.debug(LOCALISER.msg("Enhancer.AddMethod", msg));
}
if (!mmd.isAbstract())
{
// Property is not abstract so generate the getXXX method to use the aaaGetXXX we just added
generateGetXXXMethod(mv, mmd, enhancer.getASMClassName(), enhancer.getClassDescriptor(), false, ASMClassEnhancer.useFrames());
}
}
/**
* Convenience method to use the MethodVisitor to generate the code for the method getXXX() for the
* property with the specified MetaData.
* @param mv MethodVisitor
* @param mmd MetaData for the property
* @param asmClassName ASM class name for the owning class
* @param asmClassDesc ASM descriptor for the owning class
* @param detachListener true if the generate code must support DetachListener
*/
public static void generateGetXXXMethod(MethodVisitor mv, AbstractMemberMetaData mmd,
String asmClassName, String asmClassDesc, boolean detachListener, boolean includeFrames)
{
String[] argNames = new String[] {"objPC"};
String fieldTypeDesc = Type.getDescriptor(mmd.getType());
mv.visitCode();
AbstractClassMetaData cmd = mmd.getAbstractClassMetaData();
if ((mmd.getPersistenceFlags() & PersistenceCapable.MEDIATE_READ) == PersistenceCapable.MEDIATE_READ)
{
// MEDIATE_READ
Label startLabel = new Label();
mv.visitLabel(startLabel);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName,
ASMClassEnhancer.FN_StateManager, "L" + ASMClassEnhancer.ACN_StateManager + ";");
Label l1 = new Label();
mv.visitJumpInsn(Opcodes.IFNULL, l1);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName,
ASMClassEnhancer.FN_StateManager, "L" + ASMClassEnhancer.ACN_StateManager + ";");
mv.visitVarInsn(Opcodes.ALOAD, 0);
ASMUtils.addBIPUSHToMethod(mv, mmd.getFieldId());
if (cmd.getPersistenceCapableSuperclass() != null)
{
mv.visitFieldInsn(Opcodes.GETSTATIC, asmClassName, ASMClassEnhancer.FN_InheritedFieldCount, "I");
mv.visitInsn(Opcodes.IADD);
}
mv.visitMethodInsn(Opcodes.INVOKEINTERFACE, ASMClassEnhancer.ACN_StateManager,
"isLoaded", "(L" + ASMClassEnhancer.ACN_Persistable + ";I)Z");
mv.visitJumpInsn(Opcodes.IFNE, l1);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName,
ASMClassEnhancer.FN_StateManager, "L" + ASMClassEnhancer.ACN_StateManager + ";");
mv.visitVarInsn(Opcodes.ALOAD, 0);
ASMUtils.addBIPUSHToMethod(mv, mmd.getFieldId());
if (cmd.getPersistenceCapableSuperclass() != null)
{
mv.visitFieldInsn(Opcodes.GETSTATIC, asmClassName, ASMClassEnhancer.FN_InheritedFieldCount, "I");
mv.visitInsn(Opcodes.IADD);
}
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, asmClassName,
ASMClassEnhancer.MN_GetterPrefix + mmd.getName(), "()" + fieldTypeDesc);
String methodName = "get" + ASMUtils.getTypeNameForJDOMethod(mmd.getType()) + "Field";
String argTypeDesc = fieldTypeDesc;
if (methodName.equals("getObjectField"))
{
argTypeDesc = ASMUtils.CD_Object;
}
mv.visitMethodInsn(Opcodes.INVOKEINTERFACE, ASMClassEnhancer.ACN_StateManager,
methodName, "(L" + ASMClassEnhancer.ACN_Persistable + ";I" + argTypeDesc + ")" + argTypeDesc);
if (methodName.equals("getObjectField"))
{
// Cast any object fields to the correct type
mv.visitTypeInsn(Opcodes.CHECKCAST, mmd.getTypeName().replace('.', '/'));
}
ASMUtils.addReturnForType(mv, mmd.getType());
mv.visitLabel(l1);
Label l4 = new Label();
if (cmd.isDetachable())
{
// "if (objPC.aaaIsDetached() != false && ((BitSet) objPC.aaaDetachedState[2]).get(5) != true)"
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, asmClassName, ASMClassEnhancer.MN_IsDetached, "()Z");
mv.visitJumpInsn(Opcodes.IFEQ, l4);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName,
ASMClassEnhancer.FN_DetachedState, "[Ljava/lang/Object;");
mv.visitInsn(Opcodes.ICONST_2);
mv.visitInsn(Opcodes.AALOAD);
mv.visitTypeInsn(Opcodes.CHECKCAST, "java/util/BitSet");
ASMUtils.addBIPUSHToMethod(mv, mmd.getFieldId());
if (cmd.getPersistenceCapableSuperclass() != null)
{
mv.visitFieldInsn(Opcodes.GETSTATIC, asmClassName, ASMClassEnhancer.FN_InheritedFieldCount, "I");
mv.visitInsn(Opcodes.IADD);
}
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/util/BitSet", "get", "(I)Z");
mv.visitJumpInsn(Opcodes.IFNE, l4);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName,
ASMClassEnhancer.FN_DetachedState, "[Ljava/lang/Object;");
mv.visitInsn(Opcodes.ICONST_3);
mv.visitInsn(Opcodes.AALOAD);
mv.visitTypeInsn(Opcodes.CHECKCAST, "java/util/BitSet");
ASMUtils.addBIPUSHToMethod(mv, mmd.getFieldId());
if (cmd.getPersistenceCapableSuperclass() != null)
{
mv.visitFieldInsn(Opcodes.GETSTATIC, asmClassName, ASMClassEnhancer.FN_InheritedFieldCount, "I");
mv.visitInsn(Opcodes.IADD);
}
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/util/BitSet", "get", "(I)Z");
mv.visitJumpInsn(Opcodes.IFNE, l4);
if (detachListener)
{
mv.visitMethodInsn(Opcodes.INVOKESTATIC, "org/datanucleus/util/DetachListener", "getInstance",
"()Lorg/datanucleus/util/DetachListener;");
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitLdcInsn(mmd.getName());
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "org/datanucleus/util/DetachListener",
"undetachedFieldAccess", "(Ljava/lang/Object;Ljava/lang/String;)V");
}
else
{
// "throw new JDODetachedFieldAccessException(...)"
mv.visitTypeInsn(Opcodes.NEW, ASMClassEnhancer.ACN_DetachedFieldAccessException);
mv.visitInsn(Opcodes.DUP);
mv.visitLdcInsn(LOCALISER.msg("Enhancer.DetachedPropertyAccess", mmd.getName()));
mv.visitMethodInsn(Opcodes.INVOKESPECIAL, ASMClassEnhancer.ACN_DetachedFieldAccessException,
"", "(Ljava/lang/String;)V");
mv.visitInsn(Opcodes.ATHROW);
}
}
mv.visitLabel(l4);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, asmClassName,
ASMClassEnhancer.MN_GetterPrefix + mmd.getName(), "()" + fieldTypeDesc);
ASMUtils.addReturnForType(mv, mmd.getType());
Label endLabel = new Label();
mv.visitLabel(endLabel);
mv.visitLocalVariable(argNames[0], asmClassDesc, null, startLabel, endLabel, 0);
mv.visitMaxs(4, 1);
}
else if ((mmd.getPersistenceFlags() & PersistenceCapable.CHECK_READ) == PersistenceCapable.CHECK_READ)
{
// CHECK_READ
Label startLabel = new Label();
mv.visitLabel(startLabel);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName, ASMClassEnhancer.FN_Flag, "B");
Label l1 = new Label();
mv.visitJumpInsn(Opcodes.IFLE, l1);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName,
ASMClassEnhancer.FN_StateManager, "L" + ASMClassEnhancer.ACN_StateManager + ";");
mv.visitJumpInsn(Opcodes.IFNULL, l1);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName,
ASMClassEnhancer.FN_StateManager, "L" + ASMClassEnhancer.ACN_StateManager + ";");
mv.visitVarInsn(Opcodes.ALOAD, 0);
ASMUtils.addBIPUSHToMethod(mv, mmd.getFieldId());
if (cmd.getPersistenceCapableSuperclass() != null)
{
mv.visitFieldInsn(Opcodes.GETSTATIC, asmClassName,
ASMClassEnhancer.FN_InheritedFieldCount, "I");
mv.visitInsn(Opcodes.IADD);
}
mv.visitMethodInsn(Opcodes.INVOKEINTERFACE, ASMClassEnhancer.ACN_StateManager,
"isLoaded", "(L" + ASMClassEnhancer.ACN_Persistable + ";I)Z");
mv.visitJumpInsn(Opcodes.IFNE, l1);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName,
ASMClassEnhancer.FN_StateManager, "L" + ASMClassEnhancer.ACN_StateManager + ";");
mv.visitVarInsn(Opcodes.ALOAD, 0);
ASMUtils.addBIPUSHToMethod(mv, mmd.getFieldId());
if (cmd.getPersistenceCapableSuperclass() != null)
{
mv.visitFieldInsn(Opcodes.GETSTATIC, asmClassName, ASMClassEnhancer.FN_InheritedFieldCount, "I");
mv.visitInsn(Opcodes.IADD);
}
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, asmClassName, ASMClassEnhancer.MN_GetterPrefix + mmd.getName(), "()" + fieldTypeDesc);
String methodName = "get" + ASMUtils.getTypeNameForJDOMethod(mmd.getType()) + "Field";
String argTypeDesc = fieldTypeDesc;
if (methodName.equals("getObjectField"))
{
argTypeDesc = ASMUtils.CD_Object;
}
mv.visitMethodInsn(Opcodes.INVOKEINTERFACE, ASMClassEnhancer.ACN_StateManager,
methodName, "(L" + ASMClassEnhancer.ACN_Persistable + ";I" + argTypeDesc + ")" + argTypeDesc);
if (methodName.equals("getObjectField"))
{
// Cast any object fields to the correct type
mv.visitTypeInsn(Opcodes.CHECKCAST, mmd.getTypeName().replace('.', '/'));
}
ASMUtils.addReturnForType(mv, mmd.getType());
mv.visitLabel(l1);
if (cmd.isDetachable())
{
// "if (objPC.aaaIsDetached() != false && ((BitSet) objPC.aaaDetachedState[2]).get(5) != true)"
if (includeFrames)
{
mv.visitFrame(Opcodes.F_SAME, 0, null, 0, null);
}
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, asmClassName, ASMClassEnhancer.MN_IsDetached, "()Z");
Label l4 = new Label();
mv.visitJumpInsn(Opcodes.IFEQ, l4);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitFieldInsn(Opcodes.GETFIELD, asmClassName,
ASMClassEnhancer.FN_DetachedState, "[Ljava/lang/Object;");
mv.visitInsn(Opcodes.ICONST_2);
mv.visitInsn(Opcodes.AALOAD);
mv.visitTypeInsn(Opcodes.CHECKCAST, "java/util/BitSet");
ASMUtils.addBIPUSHToMethod(mv, mmd.getFieldId());
if (cmd.getPersistenceCapableSuperclass() != null)
{
mv.visitFieldInsn(Opcodes.GETSTATIC, asmClassName,
ASMClassEnhancer.FN_InheritedFieldCount, "I");
mv.visitInsn(Opcodes.IADD);
}
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/util/BitSet", "get", "(I)Z");
mv.visitJumpInsn(Opcodes.IFNE, l4);
if (detachListener)
{
mv.visitMethodInsn(Opcodes.INVOKESTATIC, "org/datanucleus/util/DetachListener", "getInstance",
"()Lorg/datanucleus/util/DetachListener;");
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitLdcInsn(mmd.getName());
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "org/datanucleus/util/DetachListener",
"undetachedFieldAccess", "(Ljava/lang/Object;Ljava/lang/String;)V");
}
else
{
// "throw new JDODetachedFieldAccessException(...)"
mv.visitTypeInsn(Opcodes.NEW, ASMClassEnhancer.ACN_DetachedFieldAccessException);
mv.visitInsn(Opcodes.DUP);
mv.visitLdcInsn(LOCALISER.msg("Enhancer.DetachedPropertyAccess", mmd.getName()));
mv.visitMethodInsn(Opcodes.INVOKESPECIAL, ASMClassEnhancer.ACN_DetachedFieldAccessException,
"", "(Ljava/lang/String;)V");
mv.visitInsn(Opcodes.ATHROW);
mv.visitLabel(l4);
}
}
if (includeFrames)
{
mv.visitFrame(Opcodes.F_SAME, 0, null, 0, null);
}
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, asmClassName,
ASMClassEnhancer.MN_GetterPrefix + mmd.getName(), "()" + fieldTypeDesc);
ASMUtils.addReturnForType(mv, mmd.getType());
Label endLabel = new Label();
mv.visitLabel(endLabel);
mv.visitLocalVariable(argNames[0], asmClassDesc, null, startLabel, endLabel, 0);
mv.visitMaxs(4, 1);
}
else
{
// NORMAL
Label startLabel = new Label();
mv.visitLabel(startLabel);
mv.visitVarInsn(Opcodes.ALOAD, 0);
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, asmClassName,
ASMClassEnhancer.MN_GetterPrefix + mmd.getName(), "()" + fieldTypeDesc);
ASMUtils.addReturnForType(mv, mmd.getType());
Label endLabel = new Label();
mv.visitLabel(endLabel);
mv.visitLocalVariable(argNames[0], asmClassDesc, null, startLabel, endLabel, 0);
mv.visitMaxs(1, 1);
}
mv.visitEnd();
}
public AnnotationVisitor visitAnnotation(String arg0, boolean arg1)
{
// Keep any annotations on the getXXX method
return mv.visitAnnotation(arg0, arg1);
}
public AnnotationVisitor visitAnnotationDefault()
{
return visitor.visitAnnotationDefault();
}
public void visitAttribute(Attribute arg0)
{
visitor.visitAttribute(arg0);
}
public void visitCode()
{
visitor.visitCode();
}
public void visitFieldInsn(int arg0, String arg1, String arg2, String arg3)
{
visitor.visitFieldInsn(arg0, arg1, arg2, arg3);
}
public void visitFrame(int arg0, int arg1, Object[] arg2, int arg3, Object[] arg4)
{
visitor.visitFrame(arg0, arg1, arg2, arg3, arg4);
}
public void visitIincInsn(int arg0, int arg1)
{
visitor.visitIincInsn(arg0, arg1);
}
public void visitInsn(int arg0)
{
visitor.visitInsn(arg0);
}
public void visitIntInsn(int arg0, int arg1)
{
visitor.visitIntInsn(arg0, arg1);
}
public void visitJumpInsn(int arg0, Label arg1)
{
visitor.visitJumpInsn(arg0, arg1);
}
public void visitLabel(Label arg0)
{
visitor.visitLabel(arg0);
}
public void visitLdcInsn(Object arg0)
{
visitor.visitLdcInsn(arg0);
}
public void visitLineNumber(int arg0, Label arg1)
{
visitor.visitLineNumber(arg0, arg1);
}
public void visitLocalVariable(String arg0, String arg1, String arg2, Label arg3, Label arg4, int arg5)
{
visitor.visitLocalVariable(arg0, arg1, arg2, arg3, arg4, arg5);
}
public void visitLookupSwitchInsn(Label arg0, int[] arg1, Label[] arg2)
{
visitor.visitLookupSwitchInsn(arg0, arg1, arg2);
}
public void visitMaxs(int arg0, int arg1)
{
visitor.visitMaxs(arg0, arg1);
}
public void visitMethodInsn(int arg0, String arg1, String arg2, String arg3)
{
visitor.visitMethodInsn(arg0, arg1, arg2, arg3);
}
public void visitMultiANewArrayInsn(String arg0, int arg1)
{
visitor.visitMultiANewArrayInsn(arg0, arg1);
}
public AnnotationVisitor visitParameterAnnotation(int arg0, String arg1, boolean arg2)
{
return visitor.visitParameterAnnotation(arg0, arg1, arg2);
}
public void visitTableSwitchInsn(int arg0, int arg1, Label arg2, Label... arg3)
{
visitor.visitTableSwitchInsn(arg0, arg1, arg2, arg3);
}
public void visitTryCatchBlock(Label arg0, Label arg1, Label arg2, String arg3)
{
visitor.visitTryCatchBlock(arg0, arg1, arg2, arg3);
}
public void visitTypeInsn(int arg0, String arg1)
{
visitor.visitTypeInsn(arg0, arg1);
}
public void visitVarInsn(int arg0, int arg1)
{
visitor.visitVarInsn(arg0, arg1);
}
}