/*
* Copyright (C) 2011 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.taobao.dx.instruction;
import com.taobao.dex.DexException;
import com.taobao.dx.util.Hex;
import com.taobao.utils.SparseIntArray;
/**
* *** This file is NOT a part of AOSP. ***
*
* Created by tangyinsheng on 2016/8/11.
*/
public final class InstructionPromoter extends InstructionVisitor {
private final SparseIntArray addressMap = new SparseIntArray();
// Notice that the unit of currentPromotedAddress is not 'byte'
// but 'short'
private int currentPromotedAddress = 0;
public InstructionPromoter() {
super(null);
}
private void mapAddressIfNeeded(int currentAddress) {
if (currentAddress != this.currentPromotedAddress) {
addressMap.append(currentAddress, this.currentPromotedAddress);
}
}
public int getPromotedAddress(int currentAddress) {
int index = addressMap.indexOfKey(currentAddress);
if (index < 0) {
return currentAddress;
} else {
return addressMap.valueAt(index);
}
}
public int getPromotedAddressCount() {
return addressMap.size();
}
@Override
public void visitZeroRegisterInsn(int currentAddress, int opcode, int index, int indexType, int target, long literal) {
mapAddressIfNeeded(currentAddress);
switch (opcode) {
case Opcodes.SPECIAL_FORMAT:
case Opcodes.NOP:
case Opcodes.RETURN_VOID: {
this.currentPromotedAddress += 1;
break;
}
case Opcodes.GOTO: {
int relativeTarget = InstructionCodec.getTarget(target, this.currentPromotedAddress);
if (relativeTarget != (byte) relativeTarget) {
if (relativeTarget != (short) relativeTarget) {
this.currentPromotedAddress += 3;
} else {
this.currentPromotedAddress += 2;
}
} else {
this.currentPromotedAddress += 1;
}
break;
}
case Opcodes.GOTO_16: {
int relativeTarget = InstructionCodec.getTarget(target, this.currentPromotedAddress);
if (relativeTarget != (short) relativeTarget) {
this.currentPromotedAddress += 3;
} else {
this.currentPromotedAddress += 2;
}
break;
}
case Opcodes.GOTO_32: {
this.currentPromotedAddress += 3;
break;
}
case Opcodes.FILLED_NEW_ARRAY:
case Opcodes.INVOKE_VIRTUAL:
case Opcodes.INVOKE_SUPER:
case Opcodes.INVOKE_DIRECT:
case Opcodes.INVOKE_STATIC:
case Opcodes.INVOKE_INTERFACE: {
this.currentPromotedAddress += 3;
break;
}
default: {
throw new IllegalStateException("unexpected opcode: " + Hex.u2or4(opcode));
}
}
}
@Override
public void visitOneRegisterInsn(int currentAddress, int opcode, int index, int indexType, int target, long literal, int a) {
mapAddressIfNeeded(currentAddress);
switch (opcode) {
case Opcodes.CONST_STRING: {
if (index > 0xFFFF) {
this.currentPromotedAddress += 3;
} else {
this.currentPromotedAddress += 2;
}
break;
}
case Opcodes.CONST_STRING_JUMBO: {
this.currentPromotedAddress += 3;
break;
}
case Opcodes.CONST_4:
case Opcodes.MOVE_RESULT:
case Opcodes.MOVE_RESULT_WIDE:
case Opcodes.MOVE_RESULT_OBJECT:
case Opcodes.MOVE_EXCEPTION:
case Opcodes.RETURN:
case Opcodes.RETURN_WIDE:
case Opcodes.RETURN_OBJECT:
case Opcodes.MONITOR_ENTER:
case Opcodes.MONITOR_EXIT:
case Opcodes.THROW: {
this.currentPromotedAddress += 1;
break;
}
case Opcodes.IF_EQZ:
case Opcodes.IF_NEZ:
case Opcodes.IF_LTZ:
case Opcodes.IF_GEZ:
case Opcodes.IF_GTZ:
case Opcodes.IF_LEZ:
case Opcodes.CONST_16:
case Opcodes.CONST_WIDE_16:
case Opcodes.CONST_HIGH16:
case Opcodes.CONST_WIDE_HIGH16:
case Opcodes.CONST_CLASS:
case Opcodes.CHECK_CAST:
case Opcodes.NEW_INSTANCE:
case Opcodes.SGET:
case Opcodes.SGET_WIDE:
case Opcodes.SGET_OBJECT:
case Opcodes.SGET_BOOLEAN:
case Opcodes.SGET_BYTE:
case Opcodes.SGET_CHAR:
case Opcodes.SGET_SHORT:
case Opcodes.SPUT:
case Opcodes.SPUT_WIDE:
case Opcodes.SPUT_OBJECT:
case Opcodes.SPUT_BOOLEAN:
case Opcodes.SPUT_BYTE:
case Opcodes.SPUT_CHAR:
case Opcodes.SPUT_SHORT: {
this.currentPromotedAddress += 2;
break;
}
case Opcodes.CONST:
case Opcodes.CONST_WIDE_32:
case Opcodes.FILL_ARRAY_DATA:
case Opcodes.PACKED_SWITCH:
case Opcodes.SPARSE_SWITCH:
case Opcodes.FILLED_NEW_ARRAY:
case Opcodes.INVOKE_VIRTUAL:
case Opcodes.INVOKE_SUPER:
case Opcodes.INVOKE_DIRECT:
case Opcodes.INVOKE_STATIC:
case Opcodes.INVOKE_INTERFACE: {
this.currentPromotedAddress += 3;
break;
}
case Opcodes.CONST_WIDE: {
this.currentPromotedAddress += 5;
break;
}
default: {
throw new IllegalStateException("unexpected opcode: " + Hex.u2or4(opcode));
}
}
}
@Override
public void visitTwoRegisterInsn(int currentAddress, int opcode, int index, int indexType, int target, long literal, int a, int b) {
mapAddressIfNeeded(currentAddress);
switch (opcode) {
case Opcodes.MOVE:
case Opcodes.MOVE_WIDE:
case Opcodes.MOVE_OBJECT:
case Opcodes.ARRAY_LENGTH:
case Opcodes.NEG_INT:
case Opcodes.NOT_INT:
case Opcodes.NEG_LONG:
case Opcodes.NOT_LONG:
case Opcodes.NEG_FLOAT:
case Opcodes.NEG_DOUBLE:
case Opcodes.INT_TO_LONG:
case Opcodes.INT_TO_FLOAT:
case Opcodes.INT_TO_DOUBLE:
case Opcodes.LONG_TO_INT:
case Opcodes.LONG_TO_FLOAT:
case Opcodes.LONG_TO_DOUBLE:
case Opcodes.FLOAT_TO_INT:
case Opcodes.FLOAT_TO_LONG:
case Opcodes.FLOAT_TO_DOUBLE:
case Opcodes.DOUBLE_TO_INT:
case Opcodes.DOUBLE_TO_LONG:
case Opcodes.DOUBLE_TO_FLOAT:
case Opcodes.INT_TO_BYTE:
case Opcodes.INT_TO_CHAR:
case Opcodes.INT_TO_SHORT:
case Opcodes.ADD_INT_2ADDR:
case Opcodes.SUB_INT_2ADDR:
case Opcodes.MUL_INT_2ADDR:
case Opcodes.DIV_INT_2ADDR:
case Opcodes.REM_INT_2ADDR:
case Opcodes.AND_INT_2ADDR:
case Opcodes.OR_INT_2ADDR:
case Opcodes.XOR_INT_2ADDR:
case Opcodes.SHL_INT_2ADDR:
case Opcodes.SHR_INT_2ADDR:
case Opcodes.USHR_INT_2ADDR:
case Opcodes.ADD_LONG_2ADDR:
case Opcodes.SUB_LONG_2ADDR:
case Opcodes.MUL_LONG_2ADDR:
case Opcodes.DIV_LONG_2ADDR:
case Opcodes.REM_LONG_2ADDR:
case Opcodes.AND_LONG_2ADDR:
case Opcodes.OR_LONG_2ADDR:
case Opcodes.XOR_LONG_2ADDR:
case Opcodes.SHL_LONG_2ADDR:
case Opcodes.SHR_LONG_2ADDR:
case Opcodes.USHR_LONG_2ADDR:
case Opcodes.ADD_FLOAT_2ADDR:
case Opcodes.SUB_FLOAT_2ADDR:
case Opcodes.MUL_FLOAT_2ADDR:
case Opcodes.DIV_FLOAT_2ADDR:
case Opcodes.REM_FLOAT_2ADDR:
case Opcodes.ADD_DOUBLE_2ADDR:
case Opcodes.SUB_DOUBLE_2ADDR:
case Opcodes.MUL_DOUBLE_2ADDR:
case Opcodes.DIV_DOUBLE_2ADDR:
case Opcodes.REM_DOUBLE_2ADDR: {
this.currentPromotedAddress += 1;
break;
}
case Opcodes.MOVE_FROM16:
case Opcodes.MOVE_WIDE_FROM16:
case Opcodes.MOVE_OBJECT_FROM16: {
this.currentPromotedAddress += 2;
break;
}
case Opcodes.ADD_INT_LIT8:
case Opcodes.RSUB_INT_LIT8:
case Opcodes.MUL_INT_LIT8:
case Opcodes.DIV_INT_LIT8:
case Opcodes.REM_INT_LIT8:
case Opcodes.AND_INT_LIT8:
case Opcodes.OR_INT_LIT8:
case Opcodes.XOR_INT_LIT8:
case Opcodes.SHL_INT_LIT8:
case Opcodes.SHR_INT_LIT8:
case Opcodes.USHR_INT_LIT8:
case Opcodes.IF_EQ:
case Opcodes.IF_NE:
case Opcodes.IF_LT:
case Opcodes.IF_GE:
case Opcodes.IF_GT:
case Opcodes.IF_LE:
case Opcodes.ADD_INT_LIT16:
case Opcodes.RSUB_INT:
case Opcodes.MUL_INT_LIT16:
case Opcodes.DIV_INT_LIT16:
case Opcodes.REM_INT_LIT16:
case Opcodes.AND_INT_LIT16:
case Opcodes.OR_INT_LIT16:
case Opcodes.XOR_INT_LIT16:
case Opcodes.INSTANCE_OF:
case Opcodes.NEW_ARRAY:
case Opcodes.IGET:
case Opcodes.IGET_WIDE:
case Opcodes.IGET_OBJECT:
case Opcodes.IGET_BOOLEAN:
case Opcodes.IGET_BYTE:
case Opcodes.IGET_CHAR:
case Opcodes.IGET_SHORT:
case Opcodes.IPUT:
case Opcodes.IPUT_WIDE:
case Opcodes.IPUT_OBJECT:
case Opcodes.IPUT_BOOLEAN:
case Opcodes.IPUT_BYTE:
case Opcodes.IPUT_CHAR:
case Opcodes.IPUT_SHORT: {
this.currentPromotedAddress += 2;
break;
}
case Opcodes.MOVE_16:
case Opcodes.MOVE_WIDE_16:
case Opcodes.MOVE_OBJECT_16:
case Opcodes.FILLED_NEW_ARRAY:
case Opcodes.INVOKE_VIRTUAL:
case Opcodes.INVOKE_SUPER:
case Opcodes.INVOKE_DIRECT:
case Opcodes.INVOKE_STATIC:
case Opcodes.INVOKE_INTERFACE: {
this.currentPromotedAddress += 3;
break;
}
default: {
throw new IllegalStateException("unexpected opcode: " + Hex.u2or4(opcode));
}
}
}
@Override
public void visitThreeRegisterInsn(int currentAddress, int opcode, int index, int indexType, int target, long literal, int a, int b, int c) {
mapAddressIfNeeded(currentAddress);
switch (opcode) {
case Opcodes.CMPL_FLOAT:
case Opcodes.CMPG_FLOAT:
case Opcodes.CMPL_DOUBLE:
case Opcodes.CMPG_DOUBLE:
case Opcodes.CMP_LONG:
case Opcodes.AGET:
case Opcodes.AGET_WIDE:
case Opcodes.AGET_OBJECT:
case Opcodes.AGET_BOOLEAN:
case Opcodes.AGET_BYTE:
case Opcodes.AGET_CHAR:
case Opcodes.AGET_SHORT:
case Opcodes.APUT:
case Opcodes.APUT_WIDE:
case Opcodes.APUT_OBJECT:
case Opcodes.APUT_BOOLEAN:
case Opcodes.APUT_BYTE:
case Opcodes.APUT_CHAR:
case Opcodes.APUT_SHORT:
case Opcodes.ADD_INT:
case Opcodes.SUB_INT:
case Opcodes.MUL_INT:
case Opcodes.DIV_INT:
case Opcodes.REM_INT:
case Opcodes.AND_INT:
case Opcodes.OR_INT:
case Opcodes.XOR_INT:
case Opcodes.SHL_INT:
case Opcodes.SHR_INT:
case Opcodes.USHR_INT:
case Opcodes.ADD_LONG:
case Opcodes.SUB_LONG:
case Opcodes.MUL_LONG:
case Opcodes.DIV_LONG:
case Opcodes.REM_LONG:
case Opcodes.AND_LONG:
case Opcodes.OR_LONG:
case Opcodes.XOR_LONG:
case Opcodes.SHL_LONG:
case Opcodes.SHR_LONG:
case Opcodes.USHR_LONG:
case Opcodes.ADD_FLOAT:
case Opcodes.SUB_FLOAT:
case Opcodes.MUL_FLOAT:
case Opcodes.DIV_FLOAT:
case Opcodes.REM_FLOAT:
case Opcodes.ADD_DOUBLE:
case Opcodes.SUB_DOUBLE:
case Opcodes.MUL_DOUBLE:
case Opcodes.DIV_DOUBLE:
case Opcodes.REM_DOUBLE: {
this.currentPromotedAddress += 2;
break;
}
case Opcodes.FILLED_NEW_ARRAY:
case Opcodes.INVOKE_VIRTUAL:
case Opcodes.INVOKE_SUPER:
case Opcodes.INVOKE_DIRECT:
case Opcodes.INVOKE_STATIC:
case Opcodes.INVOKE_INTERFACE: {
this.currentPromotedAddress += 3;
break;
}
default: {
throw new IllegalStateException("unexpected opcode: " + Hex.u2or4(opcode));
}
}
}
@Override
public void visitFourRegisterInsn(int currentAddress, int opcode, int index, int indexType, int target, long literal, int a, int b, int c, int d) {
mapAddressIfNeeded(currentAddress);
switch (opcode) {
case Opcodes.FILLED_NEW_ARRAY:
case Opcodes.INVOKE_VIRTUAL:
case Opcodes.INVOKE_SUPER:
case Opcodes.INVOKE_DIRECT:
case Opcodes.INVOKE_STATIC:
case Opcodes.INVOKE_INTERFACE: {
this.currentPromotedAddress += 3;
break;
}
default: {
throw new IllegalStateException("unexpected opcode: " + Hex.u2or4(opcode));
}
}
}
@Override
public void visitFiveRegisterInsn(int currentAddress, int opcode, int index, int indexType, int target, long literal, int a, int b, int c, int d, int e) {
mapAddressIfNeeded(currentAddress);
switch (opcode) {
case Opcodes.FILLED_NEW_ARRAY:
case Opcodes.INVOKE_VIRTUAL:
case Opcodes.INVOKE_SUPER:
case Opcodes.INVOKE_DIRECT:
case Opcodes.INVOKE_STATIC:
case Opcodes.INVOKE_INTERFACE: {
this.currentPromotedAddress += 3;
break;
}
default: {
throw new IllegalStateException("unexpected opcode: " + Hex.u2or4(opcode));
}
}
}
@Override
public void visitRegisterRangeInsn(int currentAddress, int opcode, int index, int indexType, int target, long literal, int a, int registerCount) {
mapAddressIfNeeded(currentAddress);
switch (opcode) {
case Opcodes.FILLED_NEW_ARRAY_RANGE:
case Opcodes.INVOKE_VIRTUAL_RANGE:
case Opcodes.INVOKE_SUPER_RANGE:
case Opcodes.INVOKE_DIRECT_RANGE:
case Opcodes.INVOKE_STATIC_RANGE:
case Opcodes.INVOKE_INTERFACE_RANGE: {
this.currentPromotedAddress += 3;
break;
}
default: {
throw new IllegalStateException("unexpected opcode: " + Hex.u2or4(opcode));
}
}
}
@Override
public void visitSparseSwitchPayloadInsn(int currentAddress, int opcode, int[] keys, int[] targets) {
mapAddressIfNeeded(currentAddress);
this.currentPromotedAddress += 2;
this.currentPromotedAddress += keys.length * 2;
this.currentPromotedAddress += targets.length * 2;
}
@Override
public void visitPackedSwitchPayloadInsn(int currentAddress, int opcode, int firstKey, int[] targets) {
mapAddressIfNeeded(currentAddress);
this.currentPromotedAddress += 2 + 2;
this.currentPromotedAddress += targets.length * 2;
}
@Override
public void visitFillArrayDataPayloadInsn(int currentAddress, int opcode, Object data, int size, int elementWidth) {
mapAddressIfNeeded(currentAddress);
this.currentPromotedAddress += 2 + 2;
switch (elementWidth) {
case 1: {
int length = ((byte[]) data).length;
this.currentPromotedAddress += (length >> 1) + (length & 1);
break;
}
case 2: {
this.currentPromotedAddress += ((short[]) data).length * 1;
break;
}
case 4: {
this.currentPromotedAddress += ((int[]) data).length * 2;
break;
}
case 8: {
this.currentPromotedAddress += ((long[]) data).length * 4;
break;
}
default: {
throw new DexException("bogus element_width: " + Hex.u2(elementWidth));
}
}
}
}