package org.jf.dexlib2.dexbacked;
import com.google.common.collect.ImmutableList;
import org.jf.dexlib2.dexbacked.instruction.DexBackedInstruction;
import org.jf.dexlib2.dexbacked.raw.CodeItem;
import org.jf.dexlib2.dexbacked.util.DebugInfo;
import org.jf.dexlib2.dexbacked.util.FixedSizeList;
import org.jf.dexlib2.dexbacked.util.VariableSizeListIterator;
import org.jf.dexlib2.dexbacked.util.VariableSizeLookaheadIterator;
import org.jf.dexlib2.iface.MethodImplementation;
import org.jf.dexlib2.iface.debug.DebugItem;
import org.jf.dexlib2.iface.instruction.Instruction;
import org.jf.dexlib2.util.AlignmentUtils;
import org.jf.util.ExceptionWithContext;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.Iterator;
import java.util.List;
public class DexBackedMethodImplementation implements MethodImplementation {
@Nonnull public final DexBackedDexFile dexFile;
@Nonnull public final DexBackedMethod method;
protected final int codeOffset;
protected DexBackedMethodImplementation(@Nonnull DexBackedDexFile dexFile,
@Nonnull DexBackedMethod method,
int codeOffset) {
this.dexFile = dexFile;
this.method = method;
this.codeOffset = codeOffset;
}
@Override public int getRegisterCount() {
return dexFile.getDataBuffer().readUshort(codeOffset);
}
protected int getInstructionsSize() {
return dexFile.getDataBuffer().readSmallUint(codeOffset + CodeItem.INSTRUCTION_COUNT_OFFSET);
}
protected int getInstructionsStartOffset() {
return codeOffset + CodeItem.INSTRUCTION_START_OFFSET;
}
@Nonnull @Override public Iterable<? extends Instruction> getInstructions() {
int instructionsSize = getInstructionsSize();
final int instructionsStartOffset = getInstructionsStartOffset();
final int endOffset = instructionsStartOffset + (instructionsSize*2);
return new Iterable<Instruction>() {
@Override
public Iterator<Instruction> iterator() {
return new VariableSizeLookaheadIterator<Instruction>(
dexFile.getDataBuffer(), instructionsStartOffset) {
@Override
protected Instruction readNextItem(@Nonnull DexReader reader) {
if (reader.getOffset() >= endOffset) {
return endOfData();
}
Instruction instruction = DexBackedInstruction.readFrom(dexFile, reader);
int offset = reader.getOffset();
if (offset > endOffset || offset < 0) {
throw new ExceptionWithContext("The last instruction in method %s is truncated", method);
}
return instruction;
}
};
}
};
}
protected int getTriesSize() {
return dexFile.getDataBuffer().readUshort(codeOffset + CodeItem.TRIES_SIZE_OFFSET);
}
@Nonnull
@Override
public List<? extends DexBackedTryBlock> getTryBlocks() {
final int triesSize = getTriesSize();
if (triesSize > 0) {
int instructionsSize = getInstructionsSize();
final int triesStartOffset = AlignmentUtils.alignOffset(
getInstructionsStartOffset() + (instructionsSize*2), 4);
final int handlersStartOffset = triesStartOffset + triesSize*CodeItem.TryItem.ITEM_SIZE;
return new FixedSizeList<DexBackedTryBlock>() {
@Nonnull
@Override
public DexBackedTryBlock readItem(int index) {
return new DexBackedTryBlock(dexFile,
triesStartOffset + index*CodeItem.TryItem.ITEM_SIZE,
handlersStartOffset);
}
@Override
public int size() {
return triesSize;
}
};
}
return ImmutableList.of();
}
protected int getDebugOffset() {
return dexFile.getDataBuffer().readInt(codeOffset + CodeItem.DEBUG_INFO_OFFSET);
}
@Nonnull
private DebugInfo getDebugInfo() {
int debugOffset = getDebugOffset();
if (debugOffset == -1 || debugOffset == 0) {
return DebugInfo.newOrEmpty(dexFile, 0, this);
}
if (debugOffset < 0) {
System.err.println(String.format("%s: Invalid debug offset", method));
return DebugInfo.newOrEmpty(dexFile, 0, this);
}
if ((debugOffset + dexFile.getBaseDataOffset()) >= dexFile.getBuffer().buf.length) {
System.err.println(String.format("%s: Invalid debug offset", method));
return DebugInfo.newOrEmpty(dexFile, 0, this);
}
return DebugInfo.newOrEmpty(dexFile, debugOffset, this);
}
@Nonnull @Override
public Iterable<? extends DebugItem> getDebugItems() {
return getDebugInfo();
}
@Nonnull
public Iterator<String> getParameterNames(@Nullable DexReader dexReader) {
return getDebugInfo().getParameterNames(dexReader);
}
public int getSize() {
int debugSize = getDebugInfo().getSize();
int lastOffset = getInstructionsStartOffset();
lastOffset += getInstructionsSize() * 2;
for (DexBackedTryBlock tryBlock: getTryBlocks()) {
Iterator<? extends DexBackedExceptionHandler> tryHandlerIter =
tryBlock.getExceptionHandlers().iterator();
while (tryHandlerIter.hasNext()) {
tryHandlerIter.next();
}
lastOffset = ((VariableSizeListIterator)tryHandlerIter).getReaderOffset();
}
return debugSize + (lastOffset - codeOffset);
}
}