【Soot学习记录1】新Class文件的生成及错误处理

在刚开始，我们总是发现生成的class文件无法正常使用，运行时会报错，这其中的一个原因就是未在class文件中添加构造函数。

生成流程

新建SootClass实例

完成SootClass的内容

将SootClass添加至Scene中

使用PackManager.v().writeClass(sootClass)来完成class文件的生成

问题解决

由于平时写Java代码时，编译器会自动为每个类提供一个默认构造函数，但是当接触到bytecode等底层的代码时，这个构造函数是必须手动添加的。

//创建Class文件
//以android开发为例，我们需要创建一个新的Activity类及其class文件
//activityName:新Activity类名
private SootClass createNewActivityClassFile(String activityName)
{
//如果已经存在该类，则不再创建
if(Scene.v().containsClass(activityName))
{
return Scene.v().getSootClass(activityName);
}

// Resolve dependencies
Scene.v().loadClassAndSupport("android.app.Activity");
Scene.v().loadClassAndSupport("android.os.Bundle");
Scene.v().loadClassAndSupport("android.content.Intent");

SootClass sClass;
SootMethod method;

sClass = new SootClass(activityName, Modifier.PUBLIC);
sClass.setSuperclass(Scene.v().getSootClass("android.app.Activity"));
Scene.v().addClass(sClass);
createConstrutor(sClass);//添加构造函数，如果没有这一步，那么运行调用这个类时将会报错
createOnCreateMethod(sClass);//添加onCreate函数，或者其它自定义的函数

PackManager.v().writeClass(sClass);

return sClass;
}

private void createConstrutor(SootClass sClass)
{
SootMethod method = new SootMethod("<init>", Arrays.asList(new Type[] {}), VoidType.v(), Modifier.PUBLIC);
sClass.addMethod(method);
//添加body
JimpleBody body = Jimple.v().newBody(method);
method.setActiveBody(body);
Chain<Unit> units = body.getUnits();

//添加this local
Local thisLocal = Jimple.v().newLocal("this", sClass.getType());
body.getLocals().add(thisLocal);
units.add(Jimple.v().newIdentityStmt(thisLocal,
Jimple.v().newThisRef(sClass.getType())));

//调用init
SootMethod oncreateMethod = Scene.v().getMethod("<android.app.Activity: void <init>()>");
SpecialInvokeExpr sie = Jimple.v().newSpecialInvokeExpr(thisLocal, oncreateMethod.makeRef());
units.add(Jimple.v().newInvokeStmt(sie));

// insert "return"
units.add(Jimple.v().newReturnVoidStmt());
}