Java Code Examples for org.jf.dexlib2.iface.instruction.SwitchElement#getOffset()

The following examples show how to use org.jf.dexlib2.iface.instruction.SwitchElement#getOffset() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source File: PackedSwitchInstruction.java    From JAADAS with GNU General Public License v3.0 6 votes vote down vote up
protected Stmt switchStatement(DexBody body, Instruction targetData, Local key) {
     PackedSwitchPayload i = (PackedSwitchPayload) targetData;
     List<? extends SwitchElement> seList = i.getSwitchElements();

     // the default target always follows the switch statement
     int defaultTargetAddress = codeAddress + instruction.getCodeUnits();
     Unit defaultTarget = body.instructionAtAddress(defaultTargetAddress).getUnit();

     List<IntConstant> lookupValues = new ArrayList<IntConstant>();
     List<Unit> targets = new ArrayList<Unit>();
     for(SwitchElement se: seList) {
       lookupValues.add(IntConstant.v(se.getKey()));
       int offset = se.getOffset();
       targets.add(body.instructionAtAddress(codeAddress + offset).getUnit());
     }
     switchStmt = Jimple.v().newLookupSwitchStmt(key, lookupValues, targets, defaultTarget);
     setUnit(switchStmt);
     
     if (IDalvikTyper.ENABLE_DVKTYPER) {
Debug.printDbg(IDalvikTyper.DEBUG, "constraint: "+ switchStmt);
         DalvikTyper.v().setType(switchStmt.getKeyBox(), IntType.v(), true);
     }
     
     return switchStmt;
 }
 
Example 2
Source File: SparseSwitchInstruction.java    From JAADAS with GNU General Public License v3.0 6 votes vote down vote up
protected Stmt switchStatement(DexBody body, Instruction targetData, Local key) {
     SparseSwitchPayload i = (SparseSwitchPayload) targetData;
     List<? extends SwitchElement> seList = i.getSwitchElements();

     // the default target always follows the switch statement
     int defaultTargetAddress = codeAddress + instruction.getCodeUnits();
     Unit defaultTarget = body.instructionAtAddress(defaultTargetAddress).getUnit();

     List<IntConstant> lookupValues = new ArrayList<IntConstant>();
     List<Unit> targets = new ArrayList<Unit>();
     for(SwitchElement se: seList) {
       lookupValues.add(IntConstant.v(se.getKey()));
       int offset = se.getOffset();
       targets.add(body.instructionAtAddress(codeAddress + offset).getUnit());
     }
     switchStmt = Jimple.v().newLookupSwitchStmt(key, lookupValues, targets, defaultTarget);
     setUnit(switchStmt);
     
     if (IDalvikTyper.ENABLE_DVKTYPER) {
Debug.printDbg(IDalvikTyper.DEBUG, "constraint: "+ switchStmt);
         DalvikTyper.v().setType(switchStmt.getKeyBox(), IntType.v(), true);
     }
     
     return switchStmt;
 }
 
Example 3
Source File: ImmutableSwitchElement.java    From ZjDroid with Apache License 2.0 5 votes vote down vote up
@Nonnull
public static ImmutableSwitchElement of(SwitchElement switchElement) {
    if (switchElement instanceof  ImmutableSwitchElement) {
        return (ImmutableSwitchElement)switchElement;
    }
    return new ImmutableSwitchElement(
            switchElement.getKey(),
            switchElement.getOffset());
}
 
Example 4
Source File: ImmutableSwitchElement.java    From zjdroid with Apache License 2.0 5 votes vote down vote up
@Nonnull
public static ImmutableSwitchElement of(SwitchElement switchElement) {
    if (switchElement instanceof  ImmutableSwitchElement) {
        return (ImmutableSwitchElement)switchElement;
    }
    return new ImmutableSwitchElement(
            switchElement.getKey(),
            switchElement.getOffset());
}
 
Example 5
Source File: ImmutableSwitchElement.java    From HeyGirl with Apache License 2.0 5 votes vote down vote up
@Nonnull
public static ImmutableSwitchElement of(SwitchElement switchElement) {
    if (switchElement instanceof  ImmutableSwitchElement) {
        return (ImmutableSwitchElement)switchElement;
    }
    return new ImmutableSwitchElement(
            switchElement.getKey(),
            switchElement.getOffset());
}
 
Example 6
Source File: ImmutableSwitchElement.java    From ZjDroid with Apache License 2.0 5 votes vote down vote up
@Nonnull
public static ImmutableSwitchElement of(SwitchElement switchElement) {
    if (switchElement instanceof  ImmutableSwitchElement) {
        return (ImmutableSwitchElement)switchElement;
    }
    return new ImmutableSwitchElement(
            switchElement.getKey(),
            switchElement.getOffset());
}
 
Example 7
Source File: ControlFlowGraph.java    From CFGScanDroid with GNU General Public License v2.0 4 votes vote down vote up
private boolean parseDestinations(BasicBlockInstruction bbinsn, List<Integer> leaders, Map<Integer, Integer> switchMap) {
	int offset = -1;
	switch(bbinsn.instruction.getOpcode()) {
    	case PACKED_SWITCH_PAYLOAD:		// switch payloads
    	case SPARSE_SWITCH_PAYLOAD:	
    		Integer sourceAddress = switchMap.get(bbinsn.address);
    		if(sourceAddress != null) {
				for(SwitchElement switchElement : ((SwitchPayload) bbinsn.instruction).getSwitchElements()){
					offset = switchElement.getOffset() + (int)sourceAddress;
					if(bbinsn.destinations == null)
						bbinsn.destinations = new ArrayList<Integer>();
					bbinsn.destinations.add(offset);
					if(!leaders.contains(offset))
						leaders.add(offset);
				}
			}
			break;
		case RETURN_VOID:				// returns
		case RETURN:
		case RETURN_WIDE:
		case RETURN_OBJECT:
			break;
		case GOTO:						// gotos
		case GOTO_16:
		case GOTO_32:
			if(bbinsn.destinations == null)
				bbinsn.destinations = new ArrayList<Integer>();
			offset = ((OffsetInstruction) bbinsn.instruction).getCodeOffset() + bbinsn.address;
			bbinsn.destinations.add(offset);
			if(!leaders.contains(offset))
				leaders.add(offset);

			break;
		case PACKED_SWITCH:				// switches (to payload)
		case SPARSE_SWITCH:
			offset = ((OffsetInstruction) bbinsn.instruction).getCodeOffset();
			switchMap.put(bbinsn.address+offset, bbinsn.address);
		case IF_EQ:						// ifs reg cmp reg
		case IF_NE:
		case IF_LT:
		case IF_GE:
		case IF_GT:
		case IF_LE:
		case IF_EQZ: 					// ifs reg cmp zero
		case IF_NEZ:
		case IF_LTZ:
		case IF_GEZ:
		case IF_GTZ:
		case IF_LEZ:
			if(bbinsn.destinations == null)
				bbinsn.destinations = new ArrayList<Integer>();

			offset = ((OffsetInstruction) bbinsn.instruction).getCodeOffset() + bbinsn.address;
			if(!leaders.contains(offset))
				leaders.add(offset);
			bbinsn.destinations.add(offset);
			
			offset = bbinsn.address + bbinsn.instruction.getCodeUnits();
			if(!leaders.contains(offset))
				leaders.add(offset);
			bbinsn.destinations.add(offset);
			
			break;
	}

	return true;
}