ReVanced/multidexlib2
1
0
Fork 0
mirror of https://github.com/revanced/multidexlib2.git synced 2025-05-02 15:44:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Refactor DexIO.writeCommon

Browse Source
This commit is contained in:
Lanchon 2017-09-23 13:49:39 -03:00
parent 7b27d0ca5a
commit 9ee3cce268
1 changed files with 27 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
src/main/java/lanchon/multidexlib2/DexIO.java
View File

@ -128,26 +128,11 @@ public class DexIO {
int fileClassCount = 0;
while (currentClass != null) {
if (minimalMainDex && fileClassCount >= minMainDexClassCount) break;
dexPool.mark();
dexPool.internClass(currentClass);
fileClassCount++;
boolean overflow =
dexPool.typeSection.getItemCount() > maxDexPoolSize ||
//dexPool.protoSection.getItemCount() > maxDexPoolSize ||
dexPool.fieldSection.getItemCount() > maxDexPoolSize ||
dexPool.methodSection.getItemCount() > maxDexPoolSize ||
//dexPool.classSection.getItemCount() > maxDexPoolSize ||
false;
if (overflow) {
if (fileClassCount <= minMainDexClassCount) throw new DexPoolOverflowException(
"Dex pool overflowed while writing type " + fileClassCount +
" of " + minMainDexClassCount);
if (fileClassCount == 1) throw new DexPoolOverflowException(
"Type too big for dex pool: " + currentClass.getType());
dexPool.reset();
fileClassCount--;
if (!internClass(dexPool, currentClass, maxDexPoolSize)) {
checkDexPoolOverflow(currentClass, fileClassCount, minMainDexClassCount);
break;
}
fileClassCount++;
currentClass = getQueueItem(queue, classIterator, lock);
}
synchronized (lock) {
@ -165,6 +150,30 @@ public class DexIO {
} while (currentClass != null);
}
private static boolean internClass(DexPool dexPool, ClassDef classDef, int maxDexPoolSize) {
dexPool.mark();
dexPool.internClass(classDef);
if (
dexPool.typeSection.getItemCount() > maxDexPoolSize ||
//dexPool.protoSection.getItemCount() > maxDexPoolSize ||
dexPool.fieldSection.getItemCount() > maxDexPoolSize ||
dexPool.methodSection.getItemCount() > maxDexPoolSize ||
//dexPool.classSection.getItemCount() > maxDexPoolSize ||
false
) {
dexPool.reset(); // roll back interning on pool overflow
return false;
}
return true;
}
private static void checkDexPoolOverflow(ClassDef classDef, int classCount, int minClassCount) {
if (classCount < minClassCount) throw new DexPoolOverflowException(
"Dex pool overflowed while writing type " + (classCount + 1) + " of " + minClassCount);
if (classCount == 0) throw new DexPoolOverflowException(
"Type too big for dex pool: " + classDef.getType());
}
private static <T> T getQueueItem(Queue<T> queue, Iterator<? extends T> iterator, Object lock) {
T item = queue.poll();
if (item == null) {