Dex.java revision df60589435658db76be61708bfa66515ffba40af
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; 39030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilsonimport java.util.zip.Adler32; 402bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.util.zip.ZipEntry; 412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonimport java.util.zip.ZipFile; 422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson/** 442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * The bytes of a dex file in memory for reading and writing. All int offsets 452bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * are unsigned. 462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilsonpublic final class Dex { 48030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private static final int CHECKSUM_OFFSET = 8; 49030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private static final int CHECKSUM_SIZE = 4; 50030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private static final int SIGNATURE_OFFSET = CHECKSUM_OFFSET + CHECKSUM_SIZE; 51030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private static final int SIGNATURE_SIZE = 20; 52030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 53030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private ByteBuffer data; 542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final TableOfContents tableOfContents = new TableOfContents(); 55030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private int nextSectionStart = 0; 562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final List<String> strings = new AbstractList<String>() { 582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public String get(int index) { 592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.stringIds.size); 602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.stringIds.off + (index * SizeOf.STRING_ID_ITEM)) 612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson .readString(); 622bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 632bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 642bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.stringIds.size; 652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final List<Integer> typeIds = new AbstractList<Integer>() { 692bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public Integer get(int index) { 702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.typeIds.size); 712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.typeIds.off + (index * SizeOf.TYPE_ID_ITEM)).readInt(); 722bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 732bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.typeIds.size; 752bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 762bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 772bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 782bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final List<String> typeNames = new AbstractList<String>() { 792bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public String get(int index) { 802bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.typeIds.size); 812bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return strings.get(typeIds.get(index)); 822bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 832bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 842bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.typeIds.size; 852bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 862bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 872bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 882bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final List<ProtoId> protoIds = new AbstractList<ProtoId>() { 892bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public ProtoId get(int index) { 902bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.protoIds.size); 912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.protoIds.off + (SizeOf.PROTO_ID_ITEM * index)) 922bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson .readProtoId(); 932bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.protoIds.size; 962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final List<FieldId> fieldIds = new AbstractList<FieldId>() { 1002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public FieldId get(int index) { 1012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.fieldIds.size); 1022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.fieldIds.off + (SizeOf.MEMBER_ID_ITEM * index)) 1032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson .readFieldId(); 1042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 1062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.fieldIds.size; 1072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 1092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final List<MethodId> methodIds = new AbstractList<MethodId>() { 1112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public MethodId get(int index) { 1122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson checkBounds(index, tableOfContents.methodIds.size); 1132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(tableOfContents.methodIds.off + (SizeOf.MEMBER_ID_ITEM * index)) 1142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson .readMethodId(); 1152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson @Override public int size() { 1172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents.methodIds.size; 1182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 1202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 122030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Creates a new dex that reads from {@code data}. It is an error to modify 123030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * {@code data} after using it to create a dex buffer. 1242bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 125030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public Dex(byte[] data) throws IOException { 126df60589435658db76be61708bfa66515ffba40afBrian Carlstrom this(ByteBuffer.wrap(data)); 127df60589435658db76be61708bfa66515ffba40afBrian Carlstrom } 128df60589435658db76be61708bfa66515ffba40afBrian Carlstrom 129df60589435658db76be61708bfa66515ffba40afBrian Carlstrom private Dex(ByteBuffer data) throws IOException { 130df60589435658db76be61708bfa66515ffba40afBrian Carlstrom this.data = data; 131030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data.order(ByteOrder.LITTLE_ENDIAN); 132030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.tableOfContents.readFrom(this); 1332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 136030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Creates a new empty dex of the specified size. 1372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 138030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public Dex(int byteCount) throws IOException { 139030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data = ByteBuffer.wrap(new byte[byteCount]); 140030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data.order(ByteOrder.LITTLE_ENDIAN); 1412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 1442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Creates a new dex buffer of the dex in {@code in}, and closes {@code in}. 1452bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 1462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Dex(InputStream in) throws IOException { 1472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson loadFrom(in); 1482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1502bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 1512bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Creates a new dex buffer from the dex file {@code file}. 1522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 1532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Dex(File file) throws IOException { 1542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (FileUtils.hasArchiveSuffix(file.getName())) { 1552bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ZipFile zipFile = new ZipFile(file); 1562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ZipEntry entry = zipFile.getEntry(DexFormat.DEX_IN_JAR_NAME); 1572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (entry != null) { 1582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson loadFrom(zipFile.getInputStream(entry)); 1592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson zipFile.close(); 1602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } else { 1612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new DexException("Expected " + DexFormat.DEX_IN_JAR_NAME + " in " + file); 1622bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1632bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } else if (file.getName().endsWith(".dex")) { 1642bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson loadFrom(new FileInputStream(file)); 1652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } else { 1662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new DexException("unknown output extension: " + file); 1672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 1692bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 170c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom /** 171c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom * Creates a new dex from the contents of {@code bytes}. This API supports 172030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * both {@code .dex} and {@code .odex} input. Calling this constructor 173030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * transfers ownership of {@code bytes} to the returned Dex: it is an error 174030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * to access the buffer after calling this method. 175c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom */ 176df60589435658db76be61708bfa66515ffba40afBrian Carlstrom public static Dex create(ByteBuffer data) throws IOException { 177df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data.order(ByteOrder.LITTLE_ENDIAN); 178030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 179c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom // if it's an .odex file, set position and limit to the .dex section 180df60589435658db76be61708bfa66515ffba40afBrian Carlstrom if (data.get(0) == 'd' 181df60589435658db76be61708bfa66515ffba40afBrian Carlstrom && data.get(1) == 'e' 182df60589435658db76be61708bfa66515ffba40afBrian Carlstrom && data.get(2) == 'y' 183df60589435658db76be61708bfa66515ffba40afBrian Carlstrom && data.get(3) == '\n') { 184df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data.position(8); 185df60589435658db76be61708bfa66515ffba40afBrian Carlstrom int offset = data.getInt(); 186df60589435658db76be61708bfa66515ffba40afBrian Carlstrom int length = data.getInt(); 187df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data.position(offset); 188df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data.limit(offset + length); 189df60589435658db76be61708bfa66515ffba40afBrian Carlstrom data = data.slice(); 190df60589435658db76be61708bfa66515ffba40afBrian Carlstrom } 191df60589435658db76be61708bfa66515ffba40afBrian Carlstrom 192c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom return new Dex(data); 193c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom } 194c69e85bfb7ddd6c4b0023ca29eeb0416087f9a8bBrian Carlstrom 1952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private void loadFrom(InputStream in) throws IOException { 1962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 1972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson byte[] buffer = new byte[8192]; 1982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 1992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int count; 2002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson while ((count = in.read(buffer)) != -1) { 2012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson bytesOut.write(buffer, 0, count); 2022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson in.close(); 2042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 205030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data = ByteBuffer.wrap(bytesOut.toByteArray()); 206030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data.order(ByteOrder.LITTLE_ENDIAN); 2072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson this.tableOfContents.readFrom(this); 2082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private static void checkBounds(int index, int length) { 2112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (index < 0 || index >= length) { 2122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IndexOutOfBoundsException("index:" + index + ", length=" + length); 2132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeTo(OutputStream out) throws IOException { 217030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] buffer = new byte[8192]; 218030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer data = this.data.duplicate(); // positioned ByteBuffers aren't thread safe 219030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.clear(); 220030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson while (data.hasRemaining()) { 221030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int count = Math.min(buffer.length, data.remaining()); 222030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(buffer, 0, count); 223030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson out.write(buffer, 0, count); 224030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 2252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeTo(File dexOut) throws IOException { 2282bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson OutputStream out = new FileOutputStream(dexOut); 2292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeTo(out); 2302bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson out.close(); 2312bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2322bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public TableOfContents getTableOfContents() { 2342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return tableOfContents; 2352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Section open(int position) { 238030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson if (position < 0 || position >= data.capacity()) { 239030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new IllegalArgumentException("position=" + position 240030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson + " length=" + data.capacity()); 241030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 242030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer sectionData = data.duplicate(); 243030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.order(ByteOrder.LITTLE_ENDIAN); // necessary? 244030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.position(position); 245030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.limit(data.capacity()); 246030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return new Section("section", sectionData); 2472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Section appendSection(int maxByteCount, String name) { 250030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson if ((maxByteCount & 3) != 0) { 251030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new IllegalStateException("Not four byte aligned!"); 252030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 253030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int limit = nextSectionStart + maxByteCount; 254030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer sectionData = data.duplicate(); 255030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.order(ByteOrder.LITTLE_ENDIAN); // necessary? 256030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.position(nextSectionStart); 257030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson sectionData.limit(limit); 258030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson Section result = new Section(name, sectionData); 259030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson nextSectionStart = limit; 2602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 2612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2622bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 263030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public int getLength() { 264030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.capacity(); 2652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 267030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public int getNextSectionStart() { 268030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return nextSectionStart; 2692bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private static int fourByteAlign(int position) { 2722bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return (position + 3) & ~3; 2732bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 275030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 276030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Returns a copy of the the bytes of this dex. 277030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 2782bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public byte[] getBytes() { 279030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer data = this.data.duplicate(); // positioned ByteBuffers aren't thread safe 280030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] result = new byte[data.capacity()]; 281030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(0); 282030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(result); 283030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return result; 2842bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2852bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2862bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<String> strings() { 2872bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return strings; 2882bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2892bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2902bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<Integer> typeIds() { 2912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return typeIds; 2922bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2932bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<String> typeNames() { 2952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return typeNames; 2962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 2972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 2982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<ProtoId> protoIds() { 2992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return protoIds; 3002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<FieldId> fieldIds() { 3032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return fieldIds; 3042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public List<MethodId> methodIds() { 3072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return methodIds; 3082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Iterable<ClassDef> classDefs() { 3112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new Iterable<ClassDef>() { 3122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Iterator<ClassDef> iterator() { 3132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (!tableOfContents.classDefs.exists()) { 3142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return Collections.<ClassDef>emptySet().iterator(); 3152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new Iterator<ClassDef>() { 3172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private Dex.Section in = open(tableOfContents.classDefs.off); 3182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private int count = 0; 3192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public boolean hasNext() { 3212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return count < tableOfContents.classDefs.size; 3222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public ClassDef next() { 3242bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (!hasNext()) { 3252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new NoSuchElementException(); 3262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson count++; 3282bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return in.readClassDef(); 3292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3302bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void remove() { 3312bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new UnsupportedOperationException(); 3322bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 3342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson }; 3362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public TypeList readTypeList(int offset) { 3392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (offset == 0) { 3402bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return TypeList.EMPTY; 3412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(offset).readTypeList(); 3432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3452bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public ClassData readClassData(ClassDef classDef) { 3462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int offset = classDef.getClassDataOffset(); 3472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (offset == 0) { 3482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IllegalArgumentException("offset == 0"); 3492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3502bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(offset).readClassData(); 3512bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 3532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Code readCode(ClassData.Method method) { 3542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int offset = method.getCodeOffset(); 3552bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (offset == 0) { 3562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IllegalArgumentException("offset == 0"); 3572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return open(offset).readCode(); 3592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 3602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 361030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 362030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Returns the signature of all but the first 32 bytes of this dex. The 363030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * first 32 bytes of dex files are not specified to be included in the 364030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * signature. 365030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 366030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public byte[] computeSignature() throws IOException { 367030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson MessageDigest digest; 368030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson try { 369030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson digest = MessageDigest.getInstance("SHA-1"); 370030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } catch (NoSuchAlgorithmException e) { 371030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new AssertionError(); 372030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 373030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] buffer = new byte[8192]; 374030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer data = this.data.duplicate(); // positioned ByteBuffers aren't thread safe 375030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.limit(data.capacity()); 376030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(SIGNATURE_OFFSET + SIGNATURE_SIZE); 377030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson while (data.hasRemaining()) { 378030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int count = Math.min(buffer.length, data.remaining()); 379030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(buffer, 0, count); 380030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson digest.update(buffer, 0, count); 381030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 382030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return digest.digest(); 383030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 384030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 385030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 386030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Returns the checksum of all but the first 12 bytes of {@code dex}. 387030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 388030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public int computeChecksum() throws IOException { 389030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson Adler32 adler32 = new Adler32(); 390030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] buffer = new byte[8192]; 391030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson ByteBuffer data = this.data.duplicate(); // positioned ByteBuffers aren't thread safe 392030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.limit(data.capacity()); 393030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(CHECKSUM_OFFSET + CHECKSUM_SIZE); 394030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson while (data.hasRemaining()) { 395030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int count = Math.min(buffer.length, data.remaining()); 396030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(buffer, 0, count); 397030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson adler32.update(buffer, 0, count); 398030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 399030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return (int) adler32.getValue(); 400030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 401030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 402030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 403030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Generates the signature and checksum of the dex file {@code out} and 404030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * writes them to the file. 405030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 406030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson public void writeHashes() throws IOException { 407030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson open(SIGNATURE_OFFSET).write(computeSignature()); 408030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson open(CHECKSUM_OFFSET).writeInt(computeChecksum()); 409030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 410030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 4112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public final class Section implements ByteInput, ByteOutput { 4122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private final String name; 413030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private final ByteBuffer data; 4142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 415030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private Section(String name, ByteBuffer data) { 4162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson this.name = name; 417030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data = data; 4182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int getPosition() { 421030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.position(); 4222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4242bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int readInt() { 425030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.getInt(); 4262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4282bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public short readShort() { 429030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.getShort(); 4302bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4312bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4322bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int readUnsignedShort() { 4332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return readShort() & 0xffff; 4342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public byte readByte() { 437030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.get(); 4382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4402bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public byte[] readByteArray(int length) { 441030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] result = new byte[length]; 442030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(result); 4432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 4442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4452bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public short[] readShortArray(int length) { 4472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short[] result = new short[length]; 4482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < length; i++) { 4492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson result[i] = readShort(); 4502bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4512bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 4522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int readUleb128() { 4552bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return Leb128.readUnsignedLeb128(this); 4562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int readSleb128() { 4592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return Leb128.readSignedLeb128(this); 4602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4622bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public TypeList readTypeList() { 4632bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int size = readInt(); 4642bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short[] types = new short[size]; 4652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < size; i++) { 4662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson types[i] = readShort(); 4672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson alignToFourBytes(); 4692bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new TypeList(Dex.this, types); 4702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4722bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public String readString() { 4732bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int offset = readInt(); 474030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int savedPosition = data.position(); 475030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int savedLimit = data.limit(); 476030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(offset); 477030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.limit(data.capacity()); 4782bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson try { 4792bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int expectedLength = readUleb128(); 4802bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson String result = Mutf8.decode(this, new char[expectedLength]); 4812bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (result.length() != expectedLength) { 4822bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new DexException("Declared length " + expectedLength 4832bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson + " doesn't match decoded length of " + result.length()); 4842bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4852bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 4862bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } catch (UTFDataFormatException e) { 4872bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new DexException(e); 4882bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } finally { 489030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(savedPosition); 490030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.limit(savedLimit); 4912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4922bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 4932bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 4942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public FieldId readFieldId() { 4952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int declaringClassIndex = readUnsignedShort(); 4962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int typeIndex = readUnsignedShort(); 4972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int nameIndex = readInt(); 4982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new FieldId(Dex.this, declaringClassIndex, typeIndex, nameIndex); 4992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public MethodId readMethodId() { 5022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int declaringClassIndex = readUnsignedShort(); 5032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int protoIndex = readUnsignedShort(); 5042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int nameIndex = readInt(); 5052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new MethodId(Dex.this, declaringClassIndex, protoIndex, nameIndex); 5062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public ProtoId readProtoId() { 5092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int shortyIndex = readInt(); 5102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int returnTypeIndex = readInt(); 5112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int parametersOffset = readInt(); 5122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new ProtoId(Dex.this, shortyIndex, returnTypeIndex, parametersOffset); 5132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public ClassDef readClassDef() { 5162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int offset = getPosition(); 5172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int type = readInt(); 5182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int accessFlags = readInt(); 5192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int supertype = readInt(); 5202bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int interfacesOffset = readInt(); 5212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int sourceFileIndex = readInt(); 5222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int annotationsOffset = readInt(); 5232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int classDataOffset = readInt(); 5242bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int staticValuesOffset = readInt(); 5252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new ClassDef(Dex.this, offset, type, accessFlags, supertype, 5262bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson interfacesOffset, sourceFileIndex, annotationsOffset, classDataOffset, 5272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson staticValuesOffset); 5282bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5302bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private Code readCode() { 5312bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int registersSize = readUnsignedShort(); 5322bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int insSize = readUnsignedShort(); 5332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int outsSize = readUnsignedShort(); 5342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int triesSize = readUnsignedShort(); 5352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int debugInfoOffset = readInt(); 5362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int instructionsSize = readInt(); 5372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short[] instructions = readShortArray(instructionsSize); 5382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson Code.Try[] tries = new Code.Try[triesSize]; 5392bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson Code.CatchHandler[] catchHandlers = new Code.CatchHandler[0]; 5402bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (triesSize > 0) { 5412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (instructions.length % 2 == 1) { 5422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson readShort(); // padding 5432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5452bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < triesSize; i++) { 5462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int startAddress = readInt(); 5472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int instructionCount = readUnsignedShort(); 5482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int handlerOffset = readUnsignedShort(); 5492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson tries[i] = new Code.Try(startAddress, instructionCount, handlerOffset); 5502bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5512bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int catchHandlersSize = readUleb128(); 5532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson catchHandlers = new Code.CatchHandler[catchHandlersSize]; 5542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < catchHandlersSize; i++) { 5552bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson catchHandlers[i] = readCatchHandler(); 5562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new Code(registersSize, insSize, outsSize, debugInfoOffset, instructions, 5592bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson tries, catchHandlers); 5602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5622bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private Code.CatchHandler readCatchHandler() { 5632bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int size = readSleb128(); 5642bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int handlersCount = Math.abs(size); 5652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int[] typeIndexes = new int[handlersCount]; 5662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int[] addresses = new int[handlersCount]; 5672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < handlersCount; i++) { 5682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson typeIndexes[i] = readUleb128(); 5692bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson addresses[i] = readUleb128(); 5702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int catchAllAddress = size <= 0 ? readUleb128() : -1; 5722bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new Code.CatchHandler(typeIndexes, addresses, catchAllAddress); 5732bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5752bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private ClassData readClassData() { 5762bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int staticFieldsSize = readUleb128(); 5772bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int instanceFieldsSize = readUleb128(); 5782bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int directMethodsSize = readUleb128(); 5792bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int virtualMethodsSize = readUleb128(); 5802bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Field[] staticFields = readFields(staticFieldsSize); 5812bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Field[] instanceFields = readFields(instanceFieldsSize); 5822bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Method[] directMethods = readMethods(directMethodsSize); 5832bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Method[] virtualMethods = readMethods(virtualMethodsSize); 5842bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return new ClassData(staticFields, instanceFields, directMethods, virtualMethods); 5852bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5862bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5872bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private ClassData.Field[] readFields(int count) { 5882bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Field[] result = new ClassData.Field[count]; 5892bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int fieldIndex = 0; 5902bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < count; i++) { 5912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson fieldIndex += readUleb128(); // field index diff 5922bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int accessFlags = readUleb128(); 5932bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson result[i] = new ClassData.Field(fieldIndex, accessFlags); 5942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 5962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 5972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 5982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson private ClassData.Method[] readMethods(int count) { 5992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson ClassData.Method[] result = new ClassData.Method[count]; 6002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int methodIndex = 0; 6012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (int i = 0; i < count; i++) { 6022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson methodIndex += readUleb128(); // method index diff 6032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int accessFlags = readUleb128(); 6042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int codeOff = readUleb128(); 6052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson result[i] = new ClassData.Method(methodIndex, accessFlags, codeOff); 6062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson return result; 6082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 610030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson /** 611030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * Returns a byte array containing the bytes from {@code start} to this 612030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson * section's current position. 613030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson */ 614030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson private byte[] getBytesFrom(int start) { 615030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int end = data.position(); 616030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson byte[] result = new byte[end - start]; 617030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.position(start); 618030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.get(result); 619030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return result; 620030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson } 621030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson 6222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public Annotation readAnnotation() { 6232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson byte visibility = readByte(); 624030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int start = data.position(); 6252bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson new EncodedValueReader(this, EncodedValueReader.ENCODED_ANNOTATION).skipValue(); 626030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return new Annotation(Dex.this, visibility, new EncodedValue(getBytesFrom(start))); 6272bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6282bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6292bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public EncodedValue readEncodedArray() { 630030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson int start = data.position(); 6312bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson new EncodedValueReader(this, EncodedValueReader.ENCODED_ARRAY).skipValue(); 632030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return new EncodedValue(getBytesFrom(start)); 6332bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6342bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6352bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 6362bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Writes 0x00 until the position is aligned to a multiple of 4. 6372bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 6382bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void alignToFourBytes() { 639030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson while ((data.position() & 3) != 0) { 640030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.put((byte) 0); 6412bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6422bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6432bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6442bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void assertFourByteAligned() { 645030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson if ((data.position() & 3) != 0) { 6462bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IllegalStateException("Not four byte aligned!"); 6472bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6482bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6492bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6502bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void write(byte[] bytes) { 651030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson this.data.put(bytes); 6522bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6532bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6542bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeByte(int b) { 655030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.put((byte) b); 6562bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6572bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6582bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeShort(short i) { 659030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.putShort(i); 6602bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6612bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6622bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeUnsignedShort(int i) { 6632bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short s = (short) i; 6642bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson if (i != (s & 0xffff)) { 6652bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new IllegalArgumentException("Expected an unsigned short: " + i); 6662bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6672bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeShort(s); 6682bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6692bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6702bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void write(short[] shorts) { 6712bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (short s : shorts) { 6722bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeShort(s); 6732bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6742bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6752bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6762bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeInt(int i) { 677030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson data.putInt(i); 6782bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6792bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6802bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeUleb128(int i) { 6812bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson try { 6822bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson Leb128.writeUnsignedLeb128(this, i); 6832bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } catch (ArrayIndexOutOfBoundsException e) { 684030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new DexException("Section limit " + data.limit() + " exceeded by " + name); 6852bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6862bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6872bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6882bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeSleb128(int i) { 6892bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson try { 6902bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson Leb128.writeSignedLeb128(this, i); 6912bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } catch (ArrayIndexOutOfBoundsException e) { 692030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson throw new DexException("Section limit " + data.limit() + " exceeded by " + name); 6932bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6942bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 6952bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 6962bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeStringData(String value) { 6972bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson try { 6982bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson int length = value.length(); 6992bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeUleb128(length); 7002bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson write(Mutf8.encode(value)); 7012bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeByte(0); 7022bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } catch (UTFDataFormatException e) { 7032bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson throw new AssertionError(); 7042bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7052bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7062bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 7072bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public void writeTypeList(TypeList typeList) { 7082bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson short[] types = typeList.getTypes(); 7092bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeInt(types.length); 7102bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson for (short type : types) { 7112bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson writeShort(type); 7122bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7132bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson alignToFourBytes(); 7142bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7152bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson 7162bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson /** 7172bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson * Returns the number of bytes remaining in this section. 7182bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson */ 7192bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson public int remaining() { 720030bd62e38ba41b8f464f009382955c51bd72f04Jesse Wilson return data.remaining(); 7212bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7222bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson } 7232bea5ee615b0f4add658d5660bd81c5145a0d05eJesse Wilson} 724