Add support for constructing a DexBackedDexFile from a DexBuffer

JesusFreke · Jan 23, 2020 · b7a1052 · b7a1052
1 parent bf65e57
commit b7a1052
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/DexBackedDexFile.java b/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/DexBackedDexFile.java
@@ -98,6 +98,35 @@ protected DexBackedDexFile(@Nullable Opcodes opcodes, @Nonnull byte[] buf, int o
         mapOffset = dexBuffer.readSmallUint(HeaderItem.MAP_OFFSET);
     }
 
+    protected DexBackedDexFile(@Nullable Opcodes opcodes, @Nonnull DexBuffer dexBuffer, @Nonnull DexBuffer dataBuffer, int offset, boolean verifyMagic) {
+        this.dexBuffer = dexBuffer;
+        this.dataBuffer = dataBuffer;
+
+        byte[] headerBuf = dexBuffer.readByteRange(offset, /* lengthOfHeader= */ 112);
+
+        int dexVersion = getVersion(headerBuf, offset, verifyMagic);
+
+        if (opcodes == null) {
+            this.opcodes = getDefaultOpcodes(dexVersion);
+        } else {
+            this.opcodes = opcodes;
+        }
+
+        stringCount = dexBuffer.readSmallUint(HeaderItem.STRING_COUNT_OFFSET);
+        stringStartOffset = dexBuffer.readSmallUint(HeaderItem.STRING_START_OFFSET);
+        typeCount = dexBuffer.readSmallUint(HeaderItem.TYPE_COUNT_OFFSET);
+        typeStartOffset = dexBuffer.readSmallUint(HeaderItem.TYPE_START_OFFSET);
+        protoCount = dexBuffer.readSmallUint(HeaderItem.PROTO_COUNT_OFFSET);
+        protoStartOffset = dexBuffer.readSmallUint(HeaderItem.PROTO_START_OFFSET);
+        fieldCount = dexBuffer.readSmallUint(HeaderItem.FIELD_COUNT_OFFSET);
+        fieldStartOffset = dexBuffer.readSmallUint(HeaderItem.FIELD_START_OFFSET);
+        methodCount = dexBuffer.readSmallUint(HeaderItem.METHOD_COUNT_OFFSET);
+        methodStartOffset = dexBuffer.readSmallUint(HeaderItem.METHOD_START_OFFSET);
+        classCount = dexBuffer.readSmallUint(HeaderItem.CLASS_COUNT_OFFSET);
+        classStartOffset = dexBuffer.readSmallUint(HeaderItem.CLASS_START_OFFSET);
+        mapOffset = dexBuffer.readSmallUint(HeaderItem.MAP_OFFSET);
+    }
+
     /**
      * @return The offset that various data offsets are relative to. This is always 0 for a dex file, but may be
      * different for other related formats (e.g. cdex).