Dex.java revision dc5a528c10462726682b8f490772e0677b038c1f
12bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson/* 22bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Copyright (C) 2011 The Android Open Source Project 32bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * 42bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Licensed under the Apache License, Version 2.0 (the "License"); 52bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * you may not use this file except in compliance with the License. 62bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * You may obtain a copy of the License at 72bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * 82bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * http://www.apache.org/licenses/LICENSE-2.0 92bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * 102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Unless required by applicable law or agreed to in writing, software 112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * distributed under the License is distributed on an "AS IS" BASIS, 122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * See the License for the specific language governing permissions and 142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * limitations under the License. 152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonpackage com.android.dex; 182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport com.android.dex.util.ByteInput; 202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport com.android.dex.util.ByteOutput; 212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport com.android.dex.util.FileUtils; 222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.io.ByteArrayOutputStream; 232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.io.File; 242bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.io.FileInputStream; 252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.io.FileOutputStream; 262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.io.IOException; 272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.io.InputStream; 282bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.io.OutputStream; 292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.io.UTFDataFormatException; 30c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstromimport java.nio.ByteBuffer; 31c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstromimport java.nio.ByteOrder; 32030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilsonimport java.security.MessageDigest; 33030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilsonimport java.security.NoSuchAlgorithmException; 342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.util.AbstractList; 352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.util.Collections; 362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.util.Iterator; 372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.util.List; 382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.util.NoSuchElementException; 39a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilsonimport java.util.RandomAccess; 40030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilsonimport java.util.zip.Adler32; 412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.util.zip.ZipEntry; 422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.util.zip.ZipFile; 432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson/** 452bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * The bytes of a dex file in memory for reading and writing. All int offsets 462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * are unsigned. 472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonpublic final class Dex { 49030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private static final int CHECKSUM_OFFSET = 8; 50030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private static final int CHECKSUM_SIZE = 4; 51030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private static final int SIGNATURE_OFFSET = CHECKSUM_OFFSET + CHECKSUM_SIZE; 52030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private static final int SIGNATURE_SIZE = 20; 53030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 54030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private ByteBuffer data; 552bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final TableOfContents tableOfContents = new TableOfContents(); 56030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private int nextSectionStart = 0; 572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 58a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson private static abstract class AbstractRandomAccessList<T> 59a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson extends AbstractList<T> implements RandomAccess { 60a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson } 61a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson 62a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson private List<String> strings = new AbstractRandomAccessList<String>() { 632bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public String get(int index) { 642bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.stringIds.size); 652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.stringIds.off + (index * SizeOf.STRING_ID_ITEM)) 662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson .readString(); 672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 692bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.stringIds.size; 702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 722bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 73a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson private final List<Integer> typeIds = new AbstractRandomAccessList<Integer>() { 742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public Integer get(int index) { 752bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.typeIds.size); 762bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.typeIds.off + (index * SizeOf.TYPE_ID_ITEM)).readInt(); 772bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 782bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 792bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.typeIds.size; 802bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 812bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 822bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 83a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson private final List<String> typeNames = new AbstractRandomAccessList<String>() { 842bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public String get(int index) { 852bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.typeIds.size); 862bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return strings.get(typeIds.get(index)); 872bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 882bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 892bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.typeIds.size; 902bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 922bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 93a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson private final List<ProtoId> protoIds = new AbstractRandomAccessList<ProtoId>() { 942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public ProtoId get(int index) { 952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.protoIds.size); 962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.protoIds.off + (SizeOf.PROTO_ID_ITEM * index)) 972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson .readProtoId(); 982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 1002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.protoIds.size; 1012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 1032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 104a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson private final List<FieldId> fieldIds = new AbstractRandomAccessList<FieldId>() { 1052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public FieldId get(int index) { 1062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.fieldIds.size); 1072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.fieldIds.off + (SizeOf.MEMBER_ID_ITEM * index)) 1082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson .readFieldId(); 1092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 1112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.fieldIds.size; 1122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 1142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 115a34f79ba9d4983b96571bbde0a18d352d495f601Jesse Wilson private final List<MethodId> methodIds = new AbstractRandomAccessList<MethodId>() { 1162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public MethodId get(int index) { 1172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.methodIds.size); 1182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.methodIds.off + (SizeOf.MEMBER_ID_ITEM * index)) 1192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson .readMethodId(); 1202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 1222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.methodIds.size; 1232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1242bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 1252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 127030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Creates a new dex that reads from {@code data}. It is an error to modify 128030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * {@code data} after using it to create a dex buffer. 1292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 130030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public Dex(byte[] data) throws IOException { 131df60589435658db76be61708bfa66515ffba40afBrian Carlstrom this(ByteBuffer.wrap(data)); 132df60589435658db76be61708bfa66515ffba40afBrian Carlstrom } 133df60589435658db76be61708bfa66515ffba40afBrian Carlstrom 134df60589435658db76be61708bfa66515ffba40afBrian Carlstrom private Dex(ByteBuffer data) throws IOException { 135df60589435658db76be61708bfa66515ffba40afBrian Carlstrom this.data = data; 136030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data.order(ByteOrder.LITTLE_ENDIAN); 137030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.tableOfContents.readFrom(this); 1382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1402bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 141030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Creates a new empty dex of the specified size. 1422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 143030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public Dex(int byteCount) throws IOException { 144030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data = ByteBuffer.wrap(new byte[byteCount]); 145030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data.order(ByteOrder.LITTLE_ENDIAN); 1462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 1492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Creates a new dex buffer of the dex in {@code in}, and closes {@code in}. 1502bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 1512bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Dex(InputStream in) throws IOException { 1522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson loadFrom(in); 1532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1552bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 1562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Creates a new dex buffer from the dex file {@code file}. 1572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 1582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Dex(File file) throws IOException { 1592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (FileUtils.hasArchiveSuffix(file.getName())) { 1602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ZipFile zipFile = new ZipFile(file); 1612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ZipEntry entry = zipFile.getEntry(DexFormat.DEX_IN_JAR_NAME); 1622bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (entry != null) { 1632bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson loadFrom(zipFile.getInputStream(entry)); 1642bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson zipFile.close(); 1652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } else { 1662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new DexException("Expected " + DexFormat.DEX_IN_JAR_NAME + " in " + file); 1672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } else if (file.getName().endsWith(".dex")) { 1692bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson loadFrom(new FileInputStream(file)); 1702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } else { 1712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new DexException("unknown output extension: " + file); 1722bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1732bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 175c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom /** 176c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom * Creates a new dex from the contents of {@code bytes}. This API supports 177030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * both {@code .dex} and {@code .odex} input. Calling this constructor 178030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * transfers ownership of {@code bytes} to the returned Dex: it is an error 179030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * to access the buffer after calling this method. 180c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom */ 181df60589435658db76be61708bfa66515ffba40afBrian Carlstrom public static Dex create(ByteBuffer data) throws IOException { 182df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data.order(ByteOrder.LITTLE_ENDIAN); 183030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 184c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom // if it's an .odex file, set position and limit to the .dex section 185df60589435658db76be61708bfa66515ffba40afBrian Carlstrom if (data.get(0) == 'd' 186df60589435658db76be61708bfa66515ffba40afBrian Carlstrom && data.get(1) == 'e' 187df60589435658db76be61708bfa66515ffba40afBrian Carlstrom && data.get(2) == 'y' 188df60589435658db76be61708bfa66515ffba40afBrian Carlstrom && data.get(3) == '\n') { 189df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data.position(8); 190df60589435658db76be61708bfa66515ffba40afBrian Carlstrom int offset = data.getInt(); 191df60589435658db76be61708bfa66515ffba40afBrian Carlstrom int length = data.getInt(); 192df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data.position(offset); 193df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data.limit(offset + length); 194df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data = data.slice(); 195df60589435658db76be61708bfa66515ffba40afBrian Carlstrom } 196df60589435658db76be61708bfa66515ffba40afBrian Carlstrom 197c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom return new Dex(data); 198c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom } 199c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom 2002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private void loadFrom(InputStream in) throws IOException { 2012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 2022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson byte[] buffer = new byte[8192]; 2032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int count; 2052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson while ((count = in.read(buffer)) != -1) { 2062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson bytesOut.write(buffer, 0, count); 2072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson in.close(); 2092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 210030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data = ByteBuffer.wrap(bytesOut.toByteArray()); 211030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data.order(ByteOrder.LITTLE_ENDIAN); 2122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson this.tableOfContents.readFrom(this); 2132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private static void checkBounds(int index, int length) { 2162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (index < 0 || index >= length) { 2172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IndexOutOfBoundsException("index:" + index + ", length=" + length); 2182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeTo(OutputStream out) throws IOException { 222030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] buffer = new byte[8192]; 223030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer data = this.data.duplicate(); // positioned ByteBuffers aren't thread safe 224030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.clear(); 225030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson while (data.hasRemaining()) { 226030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int count = Math.min(buffer.length, data.remaining()); 227030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(buffer, 0, count); 228030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson out.write(buffer, 0, count); 229030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 2302bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2312bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2322bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeTo(File dexOut) throws IOException { 2332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson OutputStream out = new FileOutputStream(dexOut); 2342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeTo(out); 2352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson out.close(); 2362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public TableOfContents getTableOfContents() { 2392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents; 2402bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Section open(int position) { 243030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson if (position < 0 || position >= data.capacity()) { 244030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new IllegalArgumentException("position=" + position 245030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson + " length=" + data.capacity()); 246030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 247030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer sectionData = data.duplicate(); 248030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.order(ByteOrder.LITTLE_ENDIAN); // necessary? 249030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.position(position); 250030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.limit(data.capacity()); 251030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return new Section("section", sectionData); 2522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Section appendSection(int maxByteCount, String name) { 255030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson if ((maxByteCount & 3) != 0) { 256030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new IllegalStateException("Not four byte aligned!"); 257030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 258030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int limit = nextSectionStart + maxByteCount; 259030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer sectionData = data.duplicate(); 260030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.order(ByteOrder.LITTLE_ENDIAN); // necessary? 261030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.position(nextSectionStart); 262030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.limit(limit); 263030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson Section result = new Section(name, sectionData); 264030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson nextSectionStart = limit; 2652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 2662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 268030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public int getLength() { 269030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.capacity(); 2702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 272030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public int getNextSectionStart() { 273030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return nextSectionStart; 2742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2752bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 276030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 277030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Returns a copy of the the bytes of this dex. 278030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 2792bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public byte[] getBytes() { 280030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer data = this.data.duplicate(); // positioned ByteBuffers aren't thread safe 281030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] result = new byte[data.capacity()]; 282030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(0); 283030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(result); 284030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return result; 2852bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2862bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2872bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<String> strings() { 2882bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return strings; 2892bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2902bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<Integer> typeIds() { 2922bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return typeIds; 2932bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<String> typeNames() { 2962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return typeNames; 2972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<ProtoId> protoIds() { 3002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return protoIds; 3012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<FieldId> fieldIds() { 3042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return fieldIds; 3052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<MethodId> methodIds() { 3082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return methodIds; 3092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Iterable<ClassDef> classDefs() { 3122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new Iterable<ClassDef>() { 3132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Iterator<ClassDef> iterator() { 3142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (!tableOfContents.classDefs.exists()) { 3152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return Collections.<ClassDef>emptySet().iterator(); 3162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new Iterator<ClassDef>() { 3182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private Dex.Section in = open(tableOfContents.classDefs.off); 3192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private int count = 0; 3202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public boolean hasNext() { 3222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return count < tableOfContents.classDefs.size; 3232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3242bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public ClassDef next() { 3252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (!hasNext()) { 3262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new NoSuchElementException(); 3272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3282bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson count++; 3292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return in.readClassDef(); 3302bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3312bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void remove() { 3322bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new UnsupportedOperationException(); 3332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 3352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 3372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public TypeList readTypeList(int offset) { 3402bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (offset == 0) { 3412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return TypeList.EMPTY; 3422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(offset).readTypeList(); 3442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3452bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public ClassData readClassData(ClassDef classDef) { 3472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int offset = classDef.getClassDataOffset(); 3482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (offset == 0) { 3492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IllegalArgumentException("offset == 0"); 3502bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3512bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(offset).readClassData(); 3522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Code readCode(ClassData.Method method) { 3552bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int offset = method.getCodeOffset(); 3562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (offset == 0) { 3572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IllegalArgumentException("offset == 0"); 3582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(offset).readCode(); 3602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 362030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 363030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Returns the signature of all but the first 32 bytes of this dex. The 364030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * first 32 bytes of dex files are not specified to be included in the 365030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * signature. 366030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 367030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public byte[] computeSignature() throws IOException { 368030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson MessageDigest digest; 369030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson try { 370030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson digest = MessageDigest.getInstance("SHA-1"); 371030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } catch (NoSuchAlgorithmException e) { 372030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new AssertionError(); 373030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 374030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] buffer = new byte[8192]; 375030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer data = this.data.duplicate(); // positioned ByteBuffers aren't thread safe 376030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.limit(data.capacity()); 377030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(SIGNATURE_OFFSET + SIGNATURE_SIZE); 378030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson while (data.hasRemaining()) { 379030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int count = Math.min(buffer.length, data.remaining()); 380030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(buffer, 0, count); 381030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson digest.update(buffer, 0, count); 382030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 383030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return digest.digest(); 384030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 385030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 386030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 387030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Returns the checksum of all but the first 12 bytes of {@code dex}. 388030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 389030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public int computeChecksum() throws IOException { 390030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson Adler32 adler32 = new Adler32(); 391030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] buffer = new byte[8192]; 392030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer data = this.data.duplicate(); // positioned ByteBuffers aren't thread safe 393030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.limit(data.capacity()); 394030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(CHECKSUM_OFFSET + CHECKSUM_SIZE); 395030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson while (data.hasRemaining()) { 396030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int count = Math.min(buffer.length, data.remaining()); 397030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(buffer, 0, count); 398030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson adler32.update(buffer, 0, count); 399030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 400030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return (int) adler32.getValue(); 401030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 402030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 403030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 404030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Generates the signature and checksum of the dex file {@code out} and 405030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * writes them to the file. 406030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 407030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public void writeHashes() throws IOException { 408030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson open(SIGNATURE_OFFSET).write(computeSignature()); 409030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson open(CHECKSUM_OFFSET).writeInt(computeChecksum()); 410030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 411030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 4122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public final class Section implements ByteInput, ByteOutput { 4132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final String name; 414030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private final ByteBuffer data; 4150436f436f122e7e74285faf32b8db2259f56ded7Jesse Wilson private final int initialPosition; 4162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 417030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private Section(String name, ByteBuffer data) { 4182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson this.name = name; 419030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data = data; 4200436f436f122e7e74285faf32b8db2259f56ded7Jesse Wilson this.initialPosition = data.position(); 4212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int getPosition() { 424030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.position(); 4252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int readInt() { 428030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.getInt(); 4292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4302bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4312bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public short readShort() { 432030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.getShort(); 4332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int readUnsignedShort() { 4362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return readShort() & 0xffff; 4372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public byte readByte() { 440030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.get(); 4412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public byte[] readByteArray(int length) { 444030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] result = new byte[length]; 445030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(result); 4462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 4472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public short[] readShortArray(int length) { 4502bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short[] result = new short[length]; 4512bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < length; i++) { 4522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson result[i] = readShort(); 4532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 4552bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int readUleb128() { 4582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return Leb128.readUnsignedLeb128(this); 4592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 461dc5a528c10462726682b8f490772e0677b038c1fElliott Hughes public int readUleb128p1() { 462dc5a528c10462726682b8f490772e0677b038c1fElliott Hughes return Leb128.readUnsignedLeb128(this) - 1; 463dc5a528c10462726682b8f490772e0677b038c1fElliott Hughes } 464dc5a528c10462726682b8f490772e0677b038c1fElliott Hughes 4652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int readSleb128() { 4662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return Leb128.readSignedLeb128(this); 4672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 469dc5a528c10462726682b8f490772e0677b038c1fElliott Hughes public void writeUleb128p1(int i) { 470dc5a528c10462726682b8f490772e0677b038c1fElliott Hughes writeUleb128(i + 1); 471dc5a528c10462726682b8f490772e0677b038c1fElliott Hughes } 472dc5a528c10462726682b8f490772e0677b038c1fElliott Hughes 4732bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public TypeList readTypeList() { 4742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int size = readInt(); 4752bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short[] types = new short[size]; 4762bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < size; i++) { 4772bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson types[i] = readShort(); 4782bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4792bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson alignToFourBytes(); 4802bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new TypeList(Dex.this, types); 4812bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4822bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4832bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public String readString() { 4842bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int offset = readInt(); 485030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int savedPosition = data.position(); 486030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int savedLimit = data.limit(); 487030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(offset); 488030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.limit(data.capacity()); 4892bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson try { 4902bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int expectedLength = readUleb128(); 4912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson String result = Mutf8.decode(this, new char[expectedLength]); 4922bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (result.length() != expectedLength) { 4932bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new DexException("Declared length " + expectedLength 4942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson + " doesn't match decoded length of " + result.length()); 4952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 4972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } catch (UTFDataFormatException e) { 4982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new DexException(e); 4992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } finally { 500030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(savedPosition); 501030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.limit(savedLimit); 5022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public FieldId readFieldId() { 5062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int declaringClassIndex = readUnsignedShort(); 5072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int typeIndex = readUnsignedShort(); 5082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int nameIndex = readInt(); 5092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new FieldId(Dex.this, declaringClassIndex, typeIndex, nameIndex); 5102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public MethodId readMethodId() { 5132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int declaringClassIndex = readUnsignedShort(); 5142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int protoIndex = readUnsignedShort(); 5152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int nameIndex = readInt(); 5162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new MethodId(Dex.this, declaringClassIndex, protoIndex, nameIndex); 5172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public ProtoId readProtoId() { 5202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int shortyIndex = readInt(); 5212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int returnTypeIndex = readInt(); 5222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int parametersOffset = readInt(); 5232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new ProtoId(Dex.this, shortyIndex, returnTypeIndex, parametersOffset); 5242bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public ClassDef readClassDef() { 5272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int offset = getPosition(); 5282bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int type = readInt(); 5292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int accessFlags = readInt(); 5302bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int supertype = readInt(); 5312bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int interfacesOffset = readInt(); 5322bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int sourceFileIndex = readInt(); 5332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int annotationsOffset = readInt(); 5342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int classDataOffset = readInt(); 5352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int staticValuesOffset = readInt(); 5362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new ClassDef(Dex.this, offset, type, accessFlags, supertype, 5372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson interfacesOffset, sourceFileIndex, annotationsOffset, classDataOffset, 5382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson staticValuesOffset); 5392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5402bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private Code readCode() { 5422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int registersSize = readUnsignedShort(); 5432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int insSize = readUnsignedShort(); 5442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int outsSize = readUnsignedShort(); 5452bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int triesSize = readUnsignedShort(); 5462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int debugInfoOffset = readInt(); 5472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int instructionsSize = readInt(); 5482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short[] instructions = readShortArray(instructionsSize); 5492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson Code.Try[] tries = new Code.Try[triesSize]; 5502bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson Code.CatchHandler[] catchHandlers = new Code.CatchHandler[0]; 5512bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (triesSize > 0) { 5522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (instructions.length % 2 == 1) { 5532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson readShort(); // padding 5542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5552bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < triesSize; i++) { 5572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int startAddress = readInt(); 5582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int instructionCount = readUnsignedShort(); 5592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int handlerOffset = readUnsignedShort(); 5602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson tries[i] = new Code.Try(startAddress, instructionCount, handlerOffset); 5612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5622bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5632bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int catchHandlersSize = readUleb128(); 5642bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson catchHandlers = new Code.CatchHandler[catchHandlersSize]; 5652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < catchHandlersSize; i++) { 5662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson catchHandlers[i] = readCatchHandler(); 5672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5692bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new Code(registersSize, insSize, outsSize, debugInfoOffset, instructions, 5702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson tries, catchHandlers); 5712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5722bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5732bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private Code.CatchHandler readCatchHandler() { 5742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int size = readSleb128(); 5752bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int handlersCount = Math.abs(size); 5762bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int[] typeIndexes = new int[handlersCount]; 5772bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int[] addresses = new int[handlersCount]; 5782bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < handlersCount; i++) { 5792bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson typeIndexes[i] = readUleb128(); 5802bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson addresses[i] = readUleb128(); 5812bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5822bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int catchAllAddress = size <= 0 ? readUleb128() : -1; 5832bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new Code.CatchHandler(typeIndexes, addresses, catchAllAddress); 5842bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5852bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5862bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private ClassData readClassData() { 5872bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int staticFieldsSize = readUleb128(); 5882bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int instanceFieldsSize = readUleb128(); 5892bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int directMethodsSize = readUleb128(); 5902bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int virtualMethodsSize = readUleb128(); 5912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Field[] staticFields = readFields(staticFieldsSize); 5922bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Field[] instanceFields = readFields(instanceFieldsSize); 5932bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Method[] directMethods = readMethods(directMethodsSize); 5942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Method[] virtualMethods = readMethods(virtualMethodsSize); 5952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new ClassData(staticFields, instanceFields, directMethods, virtualMethods); 5962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private ClassData.Field[] readFields(int count) { 5992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Field[] result = new ClassData.Field[count]; 6002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int fieldIndex = 0; 6012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < count; i++) { 6022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson fieldIndex += readUleb128(); // field index diff 6032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int accessFlags = readUleb128(); 6042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson result[i] = new ClassData.Field(fieldIndex, accessFlags); 6052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 6072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private ClassData.Method[] readMethods(int count) { 6102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Method[] result = new ClassData.Method[count]; 6112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int methodIndex = 0; 6122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < count; i++) { 6132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson methodIndex += readUleb128(); // method index diff 6142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int accessFlags = readUleb128(); 6152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int codeOff = readUleb128(); 6162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson result[i] = new ClassData.Method(methodIndex, accessFlags, codeOff); 6172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 6192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 621030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 622030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Returns a byte array containing the bytes from {@code start} to this 623030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * section's current position. 624030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 625030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private byte[] getBytesFrom(int start) { 626030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int end = data.position(); 627030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] result = new byte[end - start]; 628030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(start); 629030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(result); 630030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return result; 631030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 632030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 6332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Annotation readAnnotation() { 6342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson byte visibility = readByte(); 635030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int start = data.position(); 6362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson new EncodedValueReader(this, EncodedValueReader.ENCODED_ANNOTATION).skipValue(); 637030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return new Annotation(Dex.this, visibility, new EncodedValue(getBytesFrom(start))); 6382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6402bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public EncodedValue readEncodedArray() { 641030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int start = data.position(); 6422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson new EncodedValueReader(this, EncodedValueReader.ENCODED_ARRAY).skipValue(); 643030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return new EncodedValue(getBytesFrom(start)); 6442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6452bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 647450a197f09d35a07be12a924d4db1ecd2bee65feBrian Carlstrom * Skips bytes until the position is aligned to a multiple of 4. 6482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 6492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void alignToFourBytes() { 650450a197f09d35a07be12a924d4db1ecd2bee65feBrian Carlstrom data.position((data.position() + 3) & ~3); 651450a197f09d35a07be12a924d4db1ecd2bee65feBrian Carlstrom } 652450a197f09d35a07be12a924d4db1ecd2bee65feBrian Carlstrom 653450a197f09d35a07be12a924d4db1ecd2bee65feBrian Carlstrom /** 654450a197f09d35a07be12a924d4db1ecd2bee65feBrian Carlstrom * Writes 0x00 until the position is aligned to a multiple of 4. 655450a197f09d35a07be12a924d4db1ecd2bee65feBrian Carlstrom */ 656450a197f09d35a07be12a924d4db1ecd2bee65feBrian Carlstrom public void alignToFourBytesWithZeroFill() { 657030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson while ((data.position() & 3) != 0) { 658030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.put((byte) 0); 6592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6622bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void assertFourByteAligned() { 663030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson if ((data.position() & 3) != 0) { 6642bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IllegalStateException("Not four byte aligned!"); 6652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void write(byte[] bytes) { 669030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data.put(bytes); 6702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6722bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeByte(int b) { 673030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.put((byte) b); 6742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6752bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6762bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeShort(short i) { 677030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.putShort(i); 6782bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6792bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6802bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeUnsignedShort(int i) { 6812bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short s = (short) i; 6822bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (i != (s & 0xffff)) { 6832bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IllegalArgumentException("Expected an unsigned short: " + i); 6842bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6852bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeShort(s); 6862bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6872bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6882bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void write(short[] shorts) { 6892bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (short s : shorts) { 6902bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeShort(s); 6912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6922bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6932bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeInt(int i) { 695030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.putInt(i); 6962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeUleb128(int i) { 6992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson try { 7002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson Leb128.writeUnsignedLeb128(this, i); 7012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } catch (ArrayIndexOutOfBoundsException e) { 702030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new DexException("Section limit " + data.limit() + " exceeded by " + name); 7032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 7062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeSleb128(int i) { 7072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson try { 7082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson Leb128.writeSignedLeb128(this, i); 7092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } catch (ArrayIndexOutOfBoundsException e) { 710030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new DexException("Section limit " + data.limit() + " exceeded by " + name); 7112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 7142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeStringData(String value) { 7152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson try { 7162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int length = value.length(); 7172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeUleb128(length); 7182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson write(Mutf8.encode(value)); 7192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeByte(0); 7202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } catch (UTFDataFormatException e) { 7212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new AssertionError(); 7222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7242bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 7252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeTypeList(TypeList typeList) { 7262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short[] types = typeList.getTypes(); 7272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeInt(types.length); 7282bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (short type : types) { 7292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeShort(type); 7302bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 731450a197f09d35a07be12a924d4db1ecd2bee65feBrian Carlstrom alignToFourBytesWithZeroFill(); 7322bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 7342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 7352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Returns the number of bytes remaining in this section. 7362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 7372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int remaining() { 738030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.remaining(); 7392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7400436f436f122e7e74285faf32b8db2259f56ded7Jesse Wilson 7410436f436f122e7e74285faf32b8db2259f56ded7Jesse Wilson /** 7420436f436f122e7e74285faf32b8db2259f56ded7Jesse Wilson * Returns the number of bytes used by this section. 7430436f436f122e7e74285faf32b8db2259f56ded7Jesse Wilson */ 7440436f436f122e7e74285faf32b8db2259f56ded7Jesse Wilson public int used () { 7450436f436f122e7e74285faf32b8db2259f56ded7Jesse Wilson return data.position() - initialPosition; 7460436f436f122e7e74285faf32b8db2259f56ded7Jesse Wilson } 7472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson} 749