/*
|
* JStackAlloc (c) 2008 Martin Dvorak <jezek2@advel.cz>
|
*
|
* This software is provided 'as-is', without any express or implied warranty.
|
* In no event will the authors be held liable for any damages arising from
|
* the use of this software.
|
*
|
* Permission is granted to anyone to use this software for any purpose,
|
* including commercial applications, and to alter it and redistribute it
|
* freely, subject to the following restrictions:
|
*
|
* 1. The origin of this software must not be misrepresented; you must not
|
* claim that you wrote the original software. If you use this software
|
* in a product, an acknowledgment in the product documentation would be
|
* appreciated but is not required.
|
* 2. Altered source versions must be plainly marked as such, and must not be
|
* misrepresented as being the original software.
|
* 3. This notice may not be removed or altered from any source distribution.
|
*/
|
|
package cz.advel.stack.instrument;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Set;
|
import org.objectweb.asm.*;
|
import org.objectweb.asm.tree.FieldInsnNode;
|
import org.objectweb.asm.tree.InsnList;
|
import org.objectweb.asm.tree.InsnNode;
|
import org.objectweb.asm.tree.MethodInsnNode;
|
import org.objectweb.asm.tree.MethodNode;
|
import org.objectweb.asm.tree.TypeInsnNode;
|
|
/**
|
*
|
* @author jezek2
|
*/
|
class InstrumentClass extends ClassAdapter {
|
|
private Instrumenter instr;
|
private Set<String> methods;
|
private String className;
|
|
private boolean disableMethodInstrumentation = false;
|
private MethodNode clinitMethod;
|
|
private List<String> tempStaticFields = new ArrayList<String>();
|
|
public InstrumentClass(ClassVisitor cv, Instrumenter instr, Set<String> methods) {
|
super(cv);
|
this.instr = instr;
|
this.methods = methods;
|
}
|
|
@Override
|
public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
|
super.visit(version, access, name, signature, superName, interfaces);
|
className = name;
|
}
|
|
@Override
|
public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
|
if (disableMethodInstrumentation) {
|
return super.visitMethod(access, name, desc, signature, exceptions);
|
}
|
|
if (methods.contains(name)) {
|
InstrumentMethod im = new InstrumentMethod(access, name, desc, signature, exceptions, instr, this, className, cv);
|
if (isInitializerInstrumentationNeeded() && name.equals("<clinit>")) {
|
im.emitMethod = false;
|
clinitMethod = im;
|
}
|
return im;
|
}
|
|
if (isInitializerInstrumentationNeeded() && name.equals("<clinit>")) {
|
clinitMethod = new MethodNode(access, name, desc, signature, exceptions);
|
return clinitMethod;
|
}
|
|
return super.visitMethod(access, name, desc, signature, exceptions);
|
}
|
|
@Override
|
public void visitEnd() {
|
if (isInitializerInstrumentationNeeded()) {
|
if (tempStaticFields.size() > 0 && clinitMethod == null) {
|
clinitMethod = new MethodNode(Opcodes.ACC_STATIC, "<clinit>", "()V", null, null);
|
clinitMethod.instructions.add(new InsnNode(Opcodes.RETURN));
|
}
|
|
FieldVisitor fv;
|
for (int i=0; i<tempStaticFields.size(); i++) {
|
fv = visitField(Opcodes.ACC_PRIVATE | Opcodes.ACC_FINAL | Opcodes.ACC_STATIC, "$stackTemp"+i, "L"+tempStaticFields.get(i)+";", null, null);
|
fv.visitEnd();
|
}
|
|
if (clinitMethod != null) {
|
InsnList list = new InsnList();
|
for (int i=0; i<tempStaticFields.size(); i++) {
|
list.add(new TypeInsnNode(Opcodes.NEW, tempStaticFields.get(i)));
|
list.add(new InsnNode(Opcodes.DUP));
|
list.add(new MethodInsnNode(Opcodes.INVOKESPECIAL, tempStaticFields.get(i), "<init>", "()V"));
|
list.add(new FieldInsnNode(Opcodes.PUTSTATIC, className, "$stackTemp"+i, "L"+tempStaticFields.get(i)+";"));
|
}
|
clinitMethod.instructions.insertBefore(clinitMethod.instructions.getFirst(), list);
|
|
disableMethodInstrumentation = true;
|
clinitMethod.accept(this);
|
}
|
}
|
|
super.visitEnd();
|
}
|
|
private boolean isInitializerInstrumentationNeeded() {
|
return instr.isSingleThread();
|
}
|
|
public int registerStaticAlloc(String type) {
|
tempStaticFields.add(type);
|
return tempStaticFields.size() - 1;
|
}
|
|
}
|
