@Override
public List<MarkedRegion> markWithDefaultWeavingReg(MethodNode method) {
List<MarkedRegion> regions = new LinkedList<MarkedRegion>();
CtrlFlowGraph cfg = new CtrlFlowGraph(method);
cfg.visit(method.instructions.getFirst());
for (TryCatchBlockNode tcb : method.tryCatchBlocks) {
List<AbstractInsnNode> exits = cfg.visit(tcb.handler);
regions.add(new MarkedRegion(AsmHelper.skipVirtualInsns(tcb.handler, true), exits));
}
return regions;
}
@Override
protected void onMethodEnter() {
final String THREAD_CLASS_NAME = Type.getType(Thread.class).getInternalName();
final String CURRENTTHREAD_METHOD_NAME = "currentThread";
final String CURRENTTHREAD_METHOD_SIG = "()L" + THREAD_CLASS_NAME + ";";
// for each thread local var insert initialization
for (ThreadLocalVar tlv : threadLocalVars) {
Label getDefaultValue = new Label();
Label putValue = new Label();
// put this on the stack - for putfield
visitVarInsn(ALOAD, 0);
// -- inherited value --
if (tlv.isInheritable()) {
// put current thread instance on the stack
visitMethodInsn(
INVOKESTATIC, THREAD_CLASS_NAME, CURRENTTHREAD_METHOD_NAME, CURRENTTHREAD_METHOD_SIG);
// if null, go to "get default value"
visitJumpInsn(IFNULL, getDefaultValue);
// put current thread instance on the stack
visitMethodInsn(
INVOKESTATIC, THREAD_CLASS_NAME, CURRENTTHREAD_METHOD_NAME, CURRENTTHREAD_METHOD_SIG);
// get value from parent thread ant put it on the stack
visitFieldInsn(GETFIELD, THREAD_CLASS_NAME, tlv.getName(), tlv.getTypeAsDesc());
// go to "put value"
visitJumpInsn(GOTO, putValue);
}
// -- default value --
visitLabel(getDefaultValue);
// put the default value on the stack
Object defaultVal = tlv.getDefaultValue();
if (defaultVal != null) {
// default value
visitLdcInsn(defaultVal);
} else {
// if object or array
if (AsmHelper.isReferenceType(tlv.getType())) {
// insert null
visitInsn(ACONST_NULL);
}
// if basic type
else {
// insert 0 as default
visitLdcInsn(0);
}
}
// -- put value to the field --
visitLabel(putValue);
visitFieldInsn(PUTFIELD, THREAD_CLASS_NAME, tlv.getName(), tlv.getTypeAsDesc());
}
}