15867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com/* 25867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * [The "BSD licence"] 300fc68adf2e39aeb9fed35293f2576bbe729ec4bJesusFreke@JesusFreke.com * Copyright (c) 2010 Ben Gruver (JesusFreke) 45867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * All rights reserved. 55867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * 65867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * Redistribution and use in source and binary forms, with or without 75867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * modification, are permitted provided that the following conditions 85867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * are met: 95867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * 1. Redistributions of source code must retain the above copyright 105867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * notice, this list of conditions and the following disclaimer. 115867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * 2. Redistributions in binary form must reproduce the above copyright 125867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * notice, this list of conditions and the following disclaimer in the 135867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * documentation and/or other materials provided with the distribution. 145867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * 3. The name of the author may not be used to endorse or promote products 155867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * derived from this software without specific prior written permission. 165867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * 175867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 185867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 195867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 205867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 215867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 225867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 235867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 245867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 255867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 265867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 275867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com */ 285867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 295867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.compackage org.jf.dexlib.Code.Format; 305867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 315867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.comimport org.jf.dexlib.Code.Instruction; 325867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.comimport org.jf.dexlib.Code.Opcode; 335867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.comimport org.jf.dexlib.DexFile; 34b3abca4c90929e31e6a8c52bc0178c44e3e53c6bJesusFreke@JesusFreke.comimport org.jf.dexlib.Util.AnnotatedOutput; 35b3abca4c90929e31e6a8c52bc0178c44e3e53c6bJesusFreke@JesusFreke.comimport org.jf.dexlib.Util.NumberUtils; 36b3abca4c90929e31e6a8c52bc0178c44e3e53c6bJesusFreke@JesusFreke.com 375867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.comimport java.util.Iterator; 385867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 395867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.compublic class ArrayDataPseudoInstruction extends Instruction { 405867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public static final Instruction.InstructionFactory Factory = new Factory(); 41fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com private int elementWidth; 42fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com private byte[] encodedValues; 435867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 445867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com @Override 4516a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com public int getSize(int codeAddress) { 4616a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com return ((encodedValues.length + 1)/2) + 4 + (codeAddress % 2); 475867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 485867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 49fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com public ArrayDataPseudoInstruction(int elementWidth, byte[] encodedValues) { 50fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com super(Opcode.NOP); 51fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com 525867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com if (encodedValues.length % elementWidth != 0) { 535867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com throw new RuntimeException("There are not a whole number of " + elementWidth + " byte elements"); 545867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 555867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 56fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com this.elementWidth = elementWidth; 57fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com this.encodedValues = encodedValues; 58fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com } 59fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com 60fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com public ArrayDataPseudoInstruction(byte[] buffer, int bufferIndex) { 61fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com super(Opcode.NOP); 62fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com 63fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com byte opcodeByte = buffer[bufferIndex]; 64fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com if (opcodeByte != 0x00) { 65fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com throw new RuntimeException("Invalid opcode byte for an ArrayData pseudo-instruction"); 66fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com } 67fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com 68fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com byte subopcodeByte = buffer[bufferIndex+1]; 69fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com if (subopcodeByte != 0x03) { 70fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com throw new RuntimeException("Invalid sub-opcode byte for an ArrayData pseudo-instruction"); 71fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com } 72fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com 73fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com this.elementWidth = NumberUtils.decodeUnsignedShort(buffer, bufferIndex+2); 74fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com int elementCount = NumberUtils.decodeInt(buffer, bufferIndex+4); 75fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com this.encodedValues = new byte[elementCount * elementWidth]; 76fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com System.arraycopy(buffer, bufferIndex+8, encodedValues, 0, elementCount * elementWidth); 77fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com } 78fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com 7916a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com protected void writeInstruction(AnnotatedOutput out, int currentCodeAddress) { 8016a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com out.alignTo(4); 815867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 825867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com int elementCount = encodedValues.length / elementWidth; 835867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 845867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com out.writeByte(0x00); 855867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com out.writeByte(0x03); 865867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com out.writeShort(elementWidth); 875867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com out.writeInt(elementCount); 885867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com out.write(encodedValues); 8916a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com 9016a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com //make sure we're written out an even number of bytes 9116a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com out.alignTo(2); 925867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 935867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 9416a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com protected void annotateInstruction(AnnotatedOutput out, int currentCodeAddress) { 9516a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com out.annotate(getSize(currentCodeAddress)*2, "[0x" + Integer.toHexString(currentCodeAddress) + "] " + 96fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com "fill-array-data instruction"); 975867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 985867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 995867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public Format getFormat() { 1005867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com return Format.ArrayData; 1015867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1025867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 1035867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public int getElementWidth() { 104fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com return elementWidth; 1055867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1065867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 1075867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public int getElementCount() { 108fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com return encodedValues.length / elementWidth; 1095867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1105867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 1115867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public static class ArrayElement { 1125867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public final byte[] buffer; 1135867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public int bufferIndex; 1145867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public final int elementWidth; 1155867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public ArrayElement(byte[] buffer, int elementWidth) { 1165867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com this.buffer = buffer; 1175867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com this.elementWidth = elementWidth; 1185867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1195867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1205867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 1215867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public Iterator<ArrayElement> getElements() { 1225867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com return new Iterator<ArrayElement>() { 1235867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com final int elementCount = getElementCount(); 1245867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com int i=0; 125fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com int position=0; 126fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com final ArrayElement arrayElement = new ArrayElement(encodedValues, getElementWidth()); 1275867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 1285867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public boolean hasNext() { 1295867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com return i<elementCount; 1305867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1315867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 1325867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public ArrayElement next() { 1335867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com arrayElement.bufferIndex = position; 1345867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com position += arrayElement.elementWidth; 1355867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com i++; 1365867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com return arrayElement; 1375867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1385867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 1395867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public void remove() { 1405867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1415867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com }; 1425867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1435867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com 1445867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com private static class Factory implements Instruction.InstructionFactory { 1455867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com public Instruction makeInstruction(DexFile dexFile, Opcode opcode, byte[] buffer, int bufferIndex) { 1465867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com if (opcode != Opcode.NOP) { 14716a911620ccf7b530b873fb71cf13e8dcb88ad28JesusFreke@JesusFreke.com throw new RuntimeException("The opcode for an ArrayDataPseudoInstruction must be NOP"); 1485867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1495867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com return new ArrayDataPseudoInstruction(buffer, bufferIndex); 1505867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1515867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com } 1525867263eb588f4671400895d1e6b01c01535061bJesusFreke@JesusFreke.com} 153