org.jf.baksmali.BaksmaliOptions Java Examples

/**
 * 生成解析成smali代码的选项
 *
 * @return
 */
private static BaksmaliOptions createBaksmaliOptions(ClassDef classDef) {
    BaksmaliOptions options = new BaksmaliOptions();
    options.deodex = false;
    options.parameterRegisters = false;
    options.localsDirective = true;
    options.sequentialLabels = true;
    options.debugInfo = false;
    options.codeOffsets = false;
    options.accessorComments = false;
    options.registerInfo = 0;// 128
    options.inlineResolver = null;
    options.apiLevel = DEFAULT_API_LEVEL;
    List<ClassDef> classDefs = Lists.newArrayList();
    classDefs.add(classDef);
    options.syntheticAccessorResolver = new SyntheticAccessorResolver(Opcodes.getDefault(),classDefs);
    return options;
}
 

public static BaksmaliOptions getBuildOption(Iterable<? extends ClassDef> collection, int apiLevel) {
    BaksmaliOptions options = new BaksmaliOptions();

    options.deodex = false;
    options.parameterRegisters = false;
    options.localsDirective = true;
    options.sequentialLabels = true;
    options.debugInfo = true;
    options.codeOffsets = false;
    options.accessorComments = false;
    options.registerInfo = 0;// 128
    options.inlineResolver = null;
    options.apiLevel = apiLevel;
    if (!options.accessorComments) {
        options.syntheticAccessorResolver = new SyntheticAccessorResolver(Opcodes.getDefault(),collection);
    }

    return options;
}
 

private BaksmaliOptions getSmaliOptions(final DexBackedDexFile dexFile) {
    final BaksmaliOptions options = new BaksmaliOptions();

    options.deodex = false;
    options.implicitReferences = false;
    options.parameterRegisters = true;
    options.localsDirective = true;
    options.sequentialLabels = true;
    options.debugInfo = false;
    options.codeOffsets = false;
    options.accessorComments = false;
    options.registerInfo = 0;

    if (dexFile instanceof DexBackedOdexFile) {
        options.inlineResolver =
                InlineMethodResolver.createInlineMethodResolver(
                    ((DexBackedOdexFile)dexFile).getOdexVersion());
    } else {
        options.inlineResolver = null;
    }

    return options;
}
 

@Override
public boolean writeTo(IndentingWriter writer) throws IOException {
    int registerInfo = registerFormatter.options.registerInfo;
    int registerCount = analyzedInstruction.getRegisterCount();
    BitSet registers = new BitSet(registerCount);

    if ((registerInfo & BaksmaliOptions.ALL) != 0) {
        registers.set(0, registerCount);
    } else {
        if ((registerInfo & BaksmaliOptions.ALLPOST) != 0) {
            registers.set(0, registerCount);
        } else if ((registerInfo & BaksmaliOptions.DEST) != 0) {
            addDestRegs(registers, registerCount);
        }
    }

    return writeRegisterInfo(writer, registers);
}
 

public CatchMethodItem(@Nonnull BaksmaliOptions options, @Nonnull MethodDefinition.LabelCache labelCache,
                       int codeAddress, @Nullable String exceptionType, int startAddress, int endAddress,
                       int handlerAddress) {
    super(codeAddress);
    this.exceptionType = exceptionType;

    tryStartLabel = labelCache.internLabel(new LabelMethodItem(options, startAddress, "try_start_"));

    //use the address from the last covered instruction, but make the label
    //name refer to the address of the next instruction
    tryEndLabel = labelCache.internLabel(new EndTryLabelMethodItem(options, codeAddress, endAddress));

    if (exceptionType == null) {
        handlerLabel = labelCache.internLabel(new LabelMethodItem(options, handlerAddress, "catchall_"));
    } else {
        handlerLabel = labelCache.internLabel(new LabelMethodItem(options, handlerAddress, "catch_"));
    }
}
 

private void decode() throws AndrolibException {
    try {
        final BaksmaliOptions options = new BaksmaliOptions();

        // options
        options.deodex = false;
        options.implicitReferences = false;
        options.parameterRegisters = true;
        options.localsDirective = true;
        options.sequentialLabels = true;
        options.debugInfo = mBakDeb;
        options.codeOffsets = false;
        options.accessorComments = false;
        options.registerInfo = 0;
        options.inlineResolver = null;

        // set jobs automatically
        int jobs = Runtime.getRuntime().availableProcessors();
        if (jobs > 6) {
            jobs = 6;
        }

        // create the dex
        DexBackedDexFile dexFile = DexFileFactory.loadDexEntry(mApkFile, mDexFile, true, Opcodes.forApi(mApi));

        if (dexFile.isOdexFile()) {
            throw new AndrolibException("Warning: You are disassembling an odex file without deodexing it.");
        }

        if (dexFile instanceof DexBackedOdexFile) {
            options.inlineResolver =
                    InlineMethodResolver.createInlineMethodResolver(((DexBackedOdexFile)dexFile).getOdexVersion());
        }

        Baksmali.disassembleDexFile(dexFile, mOutDir, jobs, options);
    } catch (IOException ex) {
        throw new AndrolibException(ex);
    }
}
 

public static void writeEmptyMethodTo(IndentingWriter writer, Method method,
                                      BaksmaliOptions options) throws IOException {
    writer.write(".method ");
    writeAccessFlags(writer, method.getAccessFlags());
    writer.write(method.getName());
    writer.write("(");
    ImmutableList<MethodParameter> methodParameters = ImmutableList.copyOf(method.getParameters());
    for (MethodParameter parameter : methodParameters) {
        writer.write(parameter.getType());
    }
    writer.write(")");
    writer.write(method.getReturnType());
    writer.write('\n');

    writer.indent(4);
    writeParameters(writer, method, methodParameters, options);

    String containingClass = null;
    if (options.implicitReferences) {
        containingClass = method.getDefiningClass();
    }
    //如果是修改的方法，需要添加ReplaceAnnotation
    if (DexDiffInfo.modifiedMethods.contains(method)) {
        MethodReplaceAnnotation replaceAnnotation = new MethodReplaceAnnotation(method.getDefiningClass(), method.getName());
        AnnotationFormatter.writeTo(writer, method.getAnnotations(), containingClass, replaceAnnotation);
    } else {
        AnnotationFormatter.writeTo(writer, method.getAnnotations(), containingClass);
    }
    writer.deindent(4);
    writer.write(".end method\n");
}
 

private static void writeParameters(IndentingWriter writer, Method method,
                                    List<? extends MethodParameter> parameters,
                                    BaksmaliOptions options) throws IOException {
    boolean isStatic = AccessFlags.STATIC.isSet(method.getAccessFlags());
    int registerNumber = isStatic ? 0 : 1;
    for (MethodParameter parameter : parameters) {
        String parameterType = parameter.getType();
        String parameterName = parameter.getName();
        Collection<? extends Annotation> annotations = parameter.getAnnotations();
        if (parameterName != null || annotations.size() != 0) {
            writer.write(".param p");
            writer.printSignedIntAsDec(registerNumber);

            if (parameterName != null && options.debugInfo) {
                writer.write(", ");
                ReferenceFormatter.writeStringReference(writer, parameterName);
            }
            writer.write("    # ");
            writer.write(parameterType);
            writer.write("\n");
            if (annotations.size() > 0) {
                writer.indent(4);

                String containingClass = null;
                if (options.implicitReferences) {
                    containingClass = method.getDefiningClass();
                }
                AnnotationFormatter.writeTo(writer, annotations, containingClass);
                writer.deindent(4);
                writer.write(".end param\n");
            }
        }

        registerNumber++;
        if (TypeUtils.isWideType(parameterType)) {
            registerNumber++;
        }
    }
}
 

public OffsetInstructionFormatMethodItem(@Nonnull BaksmaliOptions options, @Nonnull MethodDefinition methodDef,
                                         int codeAddress, OffsetInstruction instruction) {
    super(methodDef, codeAddress, instruction);

    label = new LabelMethodItem(options, codeAddress + instruction.getCodeOffset(), getLabelPrefix());
    label = methodDef.getLabelCache().internLabel(label);
}
 

private boolean isAllowedOdex(@Nonnull Opcode opcode) {
    BaksmaliOptions options = methodDef.classDef.options;
    if (options.allowOdex) {
        return true;
    }

    if (methodDef.classDef.options.apiLevel >= 14) {
        return false;
    }

    return opcode.isVolatileFieldAccessor() || opcode == Opcode.THROW_VERIFICATION_ERROR;
}
 

/**
 * 生成解析成smali代码的选项
 * @return
 */
private static BaksmaliOptions createBaksmaliOptions() {
    BaksmaliOptions options = new BaksmaliOptions();
    options.deodex = false;
    options.parameterRegisters = false;
    options.localsDirective = true;
    options.sequentialLabels = true;
    options.debugInfo = true;
    options.codeOffsets = false;
    options.accessorComments = false;
    options.registerInfo = 0;// 128
    options.inlineResolver = null;
    options.apiLevel = DEFAULT_API_LEVEL;
    return options;
}
 

public ClassDefinition(@Nonnull BaksmaliOptions options, @Nonnull ClassDef classDef,
                       @Nonnull boolean isScan, @Nonnull boolean fullMethod) {
    this.options = options;
    this.classDef = classDef;
    this.isScan = isScan;
    this.fullMethod = fullMethod;
    fieldsSetInStaticConstructor = findFieldsSetInStaticConstructor();
}
 

public ClassDefinition(BaksmaliOptions options, ClassDef classDef) {
    this.options = options;
    this.classDef = classDef;
    this.isScan = false;
    this.fieldsSetInStaticConstructor = findFieldsSetInStaticConstructor();
    this.fullMethod = true;
}
 

private static void getSmaliCode(DexBackedClassDef classDef, StringWriter stringWriter) throws IOException {
	ClassDefinition classDefinition = new ClassDefinition(new BaksmaliOptions(), classDef);
	classDefinition.writeTo(new IndentingWriter(stringWriter));
}
 

private static String getSmaliCode(DexBackedClassDef classDef) throws IOException {
	ClassDefinition classDefinition = new ClassDefinition(new BaksmaliOptions(), classDef);
	StringWriter sw = new StringWriter();
	classDefinition.writeTo(new IndentingWriter(sw));
	return sw.toString();
}
 

/**
 * .dex转换为smali
 *
 * @param dexFile       dex/apk文件
 * @param outDir        smali文件输出目录
 * @return  是否转换成功
 */
public static boolean dex2smali(File dexFile, File outDir){

    if (dexFile == null || !dexFile.exists()){
        LogUtils.w( "dex2smali dexFile is null or not exists : " + dexFile.getPath());
        return false;
    }

    // dex文件的处理
    BaksmaliOptions options = new BaksmaliOptions();

    // options
    options.deodex = false;
    options.implicitReferences = false;
    options.parameterRegisters = true;
    options.localsDirective = true;
    options.sequentialLabels = true;
    options.debugInfo = true;
    options.codeOffsets = false;
    options.accessorComments = false;
    options.registerInfo = 0;
    options.inlineResolver = null;

    // set jobs automatically
    int jobs = Runtime.getRuntime().availableProcessors();
    if (jobs > 6) {
        jobs = 6;
    }

    try {
        //brut/androlib/ApkDecoder.mApi default value is 15
        // create the dex
        DexBackedDexFile dexBackedDexFile = DexFileFactory.loadDexFile(dexFile, Opcodes.forApi(15));

        if (dexBackedDexFile.isOdexFile()) {
            LogUtils.w("Warning: You are disassembling an odex file without deodexing it.");
        }

        if (dexBackedDexFile instanceof DexBackedOdexFile) {
            options.inlineResolver =
                    InlineMethodResolver.createInlineMethodResolver(((DexBackedOdexFile)dexBackedDexFile).getOdexVersion());
        }

        Baksmali.disassembleDexFile(dexBackedDexFile, outDir, jobs, options);
    } catch (IOException e) {
        e.printStackTrace();
    }
    return false;
}
 

public LabelMethodItem(@Nonnull BaksmaliOptions options, int codeAddress, @Nonnull String labelPrefix) {
    super(codeAddress);
    this.options = options;
    this.labelPrefix = labelPrefix;
}
 

public EndTryLabelMethodItem(@Nonnull BaksmaliOptions options, int codeAddress, int endTryAddress) {
    super(options, codeAddress, "try_end_");
    this.endTryAddress = endTryAddress;
}
 

public RegisterFormatter(@Nonnull BaksmaliOptions options, int registerCount, int parameterRegisterCount) {
    this.options = options;
    this.registerCount = registerCount;
    this.parameterRegisterCount = parameterRegisterCount;
}
 

private static String getSmaliCode(DexBackedClassDef classDef) throws IOException {
	ClassDefinition classDefinition = new ClassDefinition(new BaksmaliOptions(), classDef);
	StringWriter sw = new StringWriter();
	classDefinition.writeTo(new IndentingWriter(sw));
	return sw.toString();
}