org.jf.dexlib2.builder.BuilderInstruction Java Examples

public List<BuilderInstruction> getRealInsns(LabelAssigner labelAssigner) {
	List<BuilderInstruction> finalInsns = new ArrayList<BuilderInstruction>();
	for (Insn i : insns) {
		if (i instanceof AddressInsn) {
			continue; // skip non-insns
		}
		BuilderInstruction realInsn = i.getRealInsn(labelAssigner);
		finalInsns.add(realInsn);
           if (insnStmtMap.containsKey(i)) { // get tags
               instructionInsnMap.put(realInsn, i);
           }
           if (insnRegisterMap.containsKey(i)) {
           	instructionRegisterMap.put(realInsn, insnRegisterMap.get(i));
           }
           if (i instanceof SwitchPayload)
           	instructionPayloadMap.put(realInsn, (SwitchPayload) i);
	}
	return finalInsns;
}
 

private static MethodImplementation modifyMethodTpatch(@Nonnull MethodImplementation implementation, Method method) {
        MutableMethodImplementation mutableImplementation = new MutableMethodImplementation(implementation);
        System.out.println(mutableImplementation.getRegisterCount());
        List<BuilderInstruction> instructions = mutableImplementation.getInstructions();
        boolean isModified = false;
        for (int i = 0; i < instructions.size(); i++) {
            isModified = false;
            if (instructions.get(i).getOpcode() == Opcode.INVOKE_DIRECT) {
                if (!isModified) {
                    mutableImplementation.addInstruction(i++,
                            new BuilderInstruction21c(Opcode.CONST_STRING, 0,
                                    new ImmutableStringReference("tpatch:" + method.getDefiningClass().replace("/", "."))));
                    mutableImplementation.addInstruction(i++,
                            new BuilderInstruction35c(Opcode.INVOKE_STATIC, 1,
                                    0, 0, 0, 0, 0,
                                    new ImmutableMethodReference("Landroid/util/Log;", "e",
                                            Lists.newArrayList("Ljava/lang/String;", "Ljava/lang/String;"), "I")));
                    isModified = true;
                    break;

                }

            }
//            mutableImplementation.addInstruction(instructions.get(i));
        }

        return mutableImplementation;
    }
 

private static MethodImplementation modifyMethodAndFix(@Nonnull MethodImplementation implementation, Method method) {
    MutableMethodImplementation mutableImplementation = new MutableMethodImplementation(implementation);
    System.out.println(mutableImplementation.getRegisterCount());
    List<BuilderInstruction> instructions = mutableImplementation.getInstructions();
    mutableImplementation.addInstruction(0,
            new BuilderInstruction21c(Opcode.CONST_STRING, 0,
                    new ImmutableStringReference("AndFix:" + method.getDefiningClass().replace("/", "."))));
    mutableImplementation.addInstruction(1,
            new BuilderInstruction35c(Opcode.INVOKE_STATIC, 1,
                    0, 0, 0, 0, 0,
                    new ImmutableMethodReference("Landroid/util/Log;", "e",
                            Lists.newArrayList("Ljava/lang/String;", "Ljava/lang/String;"), "I")));

    return mutableImplementation;
}
 

private int getDistanceBetween(List<BuilderInstruction> instructions,
		int i, int j) {
	if (i == j)
		return 0;
	
	int dist = 0;
	for (int idx = Math.min(i, j); idx < Math.max(i, j); idx++) {
		BuilderInstruction bi = instructions.get(idx);
		dist += (bi.getFormat().size / 2);
	}
	return dist;
}
 

@Override
public BuilderInstruction getRealInsn(LabelAssigner assigner) {
	if (hasIncompatibleRegs()) {
		throw new RuntimeException("the instruction still has incompatible registers: " + getIncompatibleRegs());
	}
	return getRealInsn0(assigner);
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	List<SwitchLabelElement> elements = new ArrayList<SwitchLabelElement>();
	for (int i = 0; i < keys.length; i++)
		elements.add(new SwitchLabelElement(keys[i],
				assigner.getOrCreateLabel((Stmt) targets.get(i))));
	return new BuilderSparseSwitchPayload(elements);
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	List<Label> elements = new ArrayList<Label>();
	for (int i = 0; i < targets.size(); i++)
		elements.add(assigner.getOrCreateLabel((Stmt) targets.get(i)));
	return new BuilderPackedSwitchPayload(firstKey, elements);
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	int[] realRegNumbers = getRealRegNumbers();
	byte regDNumber = (byte) realRegNumbers[REG_D_IDX];
	byte regENumber = (byte) realRegNumbers[REG_E_IDX];
	byte regFNumber = (byte) realRegNumbers[REG_F_IDX];
	byte regGNumber = (byte) realRegNumbers[REG_G_IDX];
	byte regANumber = (byte) realRegNumbers[REG_A_IDX];
	return new BuilderInstruction35c(opc, regCount, regDNumber, regENumber,
			regFNumber, regGNumber, regANumber, referencedItem);
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction23x(opc, (short) getRegA().getNumber(),
			(short) getRegB().getNumber(), (short) getRegC().getNumber());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction30t(opc, assigner.getOrCreateLabel(target));
}
 

protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction22x(opc, (short) getRegA().getNumber(),
			getRegB().getNumber());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction22c(opc, getRegA().getNumber(),
			getRegB().getNumber(), referencedItem);
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction32x(opc, getRegA().getNumber(),
			getRegB().getNumber());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction31t(opc, (short) getRegA().getNumber(),
			assigner.getOrCreateLabel(payload));
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	if (target == null)
		throw new RuntimeException("Cannot jump to a NULL target");
	return new BuilderInstruction10t(opc, assigner.getOrCreateLabel(target));
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction10x(opc);
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction21s(opc, (short) getRegA().getNumber(), getLitB());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction31i(opc, (short) getRegA().getNumber(), getLitB());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction22s(opc, (byte) getRegA().getNumber(),
			(byte) getRegB().getNumber(), getLitC());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction22b(opc, (short) getRegA().getNumber(),
			(short) getRegB().getNumber(), getLitC());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction11n(opc, (byte) getRegA().getNumber(), getLitB());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction21c(opc, (short) getRegA().getNumber(), referencedItem);
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction20t(opc, assigner.getOrCreateLabel(target));
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction22t(opc, (byte) getRegA().getNumber(),
			(byte) getRegB().getNumber(), assigner.getOrCreateLabel(target));
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction51l(opc, (short) getRegA().getNumber(), getLitB());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	Register startReg = regs.get(0);
	return new BuilderInstruction3rc(opc, startReg.getNumber(), regCount, referencedItem);
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction12x(opc, (byte) getRegA().getNumber(), (byte) getRegB().getNumber());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction21t(opc, (short) getRegA().getNumber(),
			assigner.getOrCreateLabel(target));
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return new BuilderInstruction11x(opc, (short) getRegA().getNumber());
}
 

@Override
protected BuilderInstruction getRealInsn0(LabelAssigner assigner) {
	return null;
}