用 BCEL 设计字节码（二）直接在原方法上加指令

前一个是直接生成了一个一个方法对原方法进行调用，现在直接在原方法上进行修改
即将

public class StringBuilder

{

private String buildString(int length) {

String result = "";

for (int i = 0; i < length; i++) {

result += (char)(i%26 + 'a');

}

System.out.println(result);

return result;

}

public static void main(String[] argv) {

StringBuilder inst = new StringBuilder();

for (int i = 0; i < argv.length; i++) {

String result = inst.buildString(Integer.parseInt(argv[i]));

System.out.println("Constructed string of length " +

result.length());

}

改成：

public class StringBuilder

{

private String buildString(int length) {

System.out.println("start start start");
long starttime = System.currentTimeMillis();

String result = "";

for (int i = 0; i < length; i++) {

result += (char)(i%26 + 'a');

}

System.out.println(result);

        System.out.println("Call to buildString$impl took " +
                (System.currentTimeMillis()-starttime) + " ms.");
        System.out.println("end end end end");

return result;

}

public static void main(String[] argv) {

StringBuilder inst = new StringBuilder();

for (int i = 0; i < argv.length; i++) {

String result = inst.buildString(Integer.parseInt(argv[i]));

System.out.println("Constructed string of length " +

result.length());

}

修改代码如下：

import java.io.FileOutputStream;

import java.util.Iterator;

import org.apache.bcel.Constants;

import org.apache.bcel.classfile.ClassParser;

import org.apache.bcel.classfile.JavaClass;

import org.apache.bcel.classfile.LineNumberTable;

import org.apache.bcel.classfile.LocalVariableTable;

import org.apache.bcel.classfile.Method;

import org.apache.bcel.generic.ClassGen;

import org.apache.bcel.generic.ConstantPoolGen;

import org.apache.bcel.generic.InstructionConstants;

import org.apache.bcel.generic.InstructionFactory;

import org.apache.bcel.generic.InstructionHandle;

import org.apache.bcel.generic.InstructionList;

import org.apache.bcel.generic.InvokeInstruction;

import org.apache.bcel.generic.LocalVariableGen;

import org.apache.bcel.generic.MethodGen;

import org.apache.bcel.generic.ObjectType;

import org.apache.bcel.generic.PUSH;

import org.apache.bcel.generic.Type;

public class BCELTiming2 {

private static void modifyWrapper(ClassGen cgen,Method method){

InstructionFactory ifact = new InstructionFactory(cgen);

ConstantPoolGen pgen = cgen.getConstantPool();

String cname = cgen.getClassName();

MethodGen wrapgen = new MethodGen(method,cname,pgen);

//除掉这个原来的方法

cgen.removeMethod(method);

System.out.println("打印出这个方法看看---------------------------------start");

System.out.println(wrapgen.toString());

System.out.println("打印出这个方法看看---------------------------------end");

InstructionList ilist = wrapgen.getInstructionList();

System.out.println("看看这个方法的InstructionList-----------------------------start");

System.out.println(ilist.toString());

System.out.println("看看这个方法的InstructionList-----------------------------start");

System.out.println("看看这个方法InstructionList的各个InstructionHandle的信息--------------------start");

Iterator handleIt = ilist.iterator();

while(handleIt.hasNext()){

InstructionHandle iHandle = (InstructionHandle)handleIt.next();

System.out.println(iHandle.getAttributes());

System.out.println(iHandle.toString());

}

System.out.println("看看这个方法InstructionList的各个InstructionHandle的信息--------------------end");

System.out.println("先看看此方法的LocalVariableTable的信息--------------------------------------start");

LocalVariableTable lvt = wrapgen.getLocalVariableTable(pgen);

System.out.println(lvt.toString());

System.out.println("先看看此方法的LocalVariableTable的信息--------------------------------------end");

LineNumberTable lnt = wrapgen.getLineNumberTable(pgen);

System.out.println("LineNumberTable--------------------------------------start");

System.out.println(lnt.toString());

System.out.println("LineNumberTable--------------------------------------end");

Type returnType = wrapgen.getReturnType();

Type[] types = wrapgen.getArgumentTypes();

int slot = wrapgen.isStatic()?0:1;//非静态方法slot 0处应该存储的是this

//// 这种方式与Java如何处理方法调用有关。对于非静态的方法，每次调用的第一个（隐藏的）参数是目标对象的this引用（就是位置0储存的内容）。

for(int i = 0;i<types.length;i++){

slot += types[i].getSize();//long,double的size为2

}

* 判断原来的方法用到了哪些局部变量,获得局部变量中已经用到的最大索引

* 将新产生的局部变量加到最大索引后,本来准备自己写的

* BCEL中已经这样的方法了

LocalVariableGen lvg = wrapgen.addLocalVariable("starttime", Type.LONG, null, null);

//先插入一条打印语句不使用局部变量

InstructionList printlnList = ifact.createPrintln("start test start test start test start test");

InvokeInstruction invokestatic = ifact.createInvoke("java.lang.System",

"currentTimeMillis", Type.LONG, Type.NO_ARGS, Constants.INVOKESTATIC);

InstructionHandle firstHandle = printlnList.append(invokestatic);

//这个会不会将局部变量为2的地方的局部变量给覆盖,然后此局部变量也没有定义名字??slot这个index怎么来确定

//slot现在应该为2,之前代码块放的是result信息是aload,

//写到新产生的局部变量的索引处

printlnList.append(InstructionFactory.createStore(Type.LONG, lvg.getIndex()));

//加入到之前的inlist中

ilist.insert(printlnList);

//在语句最后在打印结束语句,不能简单插入在最后面,加入方法有返回值,则要插入在return 之前

//在return的时候，还要先将return的引用加入到操作数栈,怎么获取result在局部变量中index

InstructionHandle insertposition =null;

if(returnType.getType()!=Type.VOID.getType()){

insertposition = ilist.getEnd().getPrev()；

}else{

//返回值为Type.void的话,由于没有返回值，因此只需要在return指令前加就够了,往前取一个即可

insertposition = ilist.getEnd();

}

InstructionList tempList = new InstructionList();

tempList.append(ifact.createFieldAccess("java.lang.System"

, "out", new ObjectType("java.io.PrintStream"),

Constants.GETSTATIC));

tempList.append(InstructionFactory.DUP);

String text = "Call to method "+wrapgen.getName()+" took ";

tempList.append(new PUSH(pgen,text));

tempList.append(ifact.createInvoke("java.io.PrintStream",

"print", Type.VOID, new Type[]{Type.STRING},

Constants.INVOKEVIRTUAL));

tempList.append(ifact.createInvoke("java.lang.System",

"currentTimeMillis", Type.LONG, Type.NO_ARGS,

Constants.INVOKESTATIC));

//获取之前的的starttime局部变量

tempList.append(InstructionFactory.createLoad(Type.LONG, lvg.getIndex()));

tempList.append(InstructionConstants.LSUB);

tempList.append(ifact.createInvoke("java.io.PrintStream",

"print", Type.VOID, new Type[]{Type.LONG},

Constants.INVOKEVIRTUAL));

tempList.append(new PUSH(pgen," ms."));

tempList.append(ifact.createInvoke("java.io.PrintStream",

"println", Type.VOID, new Type[]{Type.STRING},

Constants.INVOKEVIRTUAL));

//在return之前的最后一句是打印end end end

.信息

tempList.append(ifact.createPrintln("end end end end end end end end end end"));

ilist.insert(insertposition,tempList);

// LocalVariableGen lvgen = new LocalVariableGen(slot,"start",Type.LONG,null,null);

//finalize the construted method

wrapgen.stripAttributes(false);

wrapgen.setMaxStack();

wrapgen.setMaxLocals();

cgen.addMethod(wrapgen.getMethod());

System.out.println();

System.out.println(wrapgen.getInstructionList());

ilist.dispose();

}

public static void main(String[] args) {

args[0]="D:\\java to eclipse\\javaeclipsestudy\\workspace\\BCELTest\\bin\\StringBuilder.class";

args[1]="buildString";

String targetClassfile = "StringBuilder.class";

if(args.length==2 && args[0].endsWith(".class")){

try{

JavaClass jclas = new ClassParser(args[0]).parse();

ClassGen cgen = new ClassGen(jclas);

Method[] methods = jclas.getMethods();

int index;

for(index =0;index<methods.length;index++){

if(methods[index].getName().equals(args[1])){

break;

}

if(index<methods.length){

modifyWrapper(cgen,methods[index]);

FileOutputStream fos = new FileOutputStream(targetClassfile);

cgen.getJavaClass().dump(fos);

fos.close();

}else{

System.err.println("Method " + args[1]+"not found in"+ args[0]);

}

}catch(Exception e){

e.printStackTrace();

}

}else{

System.out.println("usage: class-file method-name");

}

posted on 2009-08-13 22:53 Frank_Fang 阅读(1969) 评论(2) 编辑收藏所属分类: bcel javassist

常用链接

留言簿(1)

随笔分类(204)

随笔档案(100)

收藏夹(8)

牛人博客链接

搜索

最新评论

阅读排行榜

评论排行榜


只有注册用户登录后才能发表评论。




网站导航: 博客园 IT新闻知识库 C++博客博问管理
相关文章: API转换的问题的解决用 BCEL 设计字节码（三）直接在方法的调用处添加方法用 BCEL 设计字节码（二）直接在原方法上加指令【转】用 BCEL 设计字节码 API转换的问题的提出 Java的编程动态性