ReVanced/smali
1
0
Fork 0
mirror of https://github.com/revanced/smali.git synced 2025-05-08 10:24:31 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add multi-threading support for smali

Browse Source
This commit is contained in:
Ben Gruver 2013-04-30 01:05:16 -07:00
parent 03a4ffa1f4
commit 4c431a7ce7
1 changed files with 56 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
smali/src/main/java/org/jf/smali/main.java
View File

@ -28,6 +28,7 @@
package org.jf.smali; package org.jf.smali;
import com.google.common.collect.Lists;
import org.antlr.runtime.CommonTokenStream; import org.antlr.runtime.CommonTokenStream;
import org.antlr.runtime.Token; import org.antlr.runtime.Token;
import org.antlr.runtime.TokenSource; import org.antlr.runtime.TokenSource;
@ -40,10 +41,11 @@ import org.jf.util.SmaliHelpFormatter;
import javax.annotation.Nonnull; import javax.annotation.Nonnull;
import java.io.*; import java.io.*;
import java.util.LinkedHashSet; import java.util.*;
import java.util.Locale; import java.util.concurrent.Callable;
import java.util.Properties; import java.util.concurrent.ExecutorService;
import java.util.Set; import java.util.concurrent.Executors;
import java.util.concurrent.Future;
/** /**
* Main class for smali. It recognizes enough options to be able to dispatch * Main class for smali. It recognizes enough options to be able to dispatch
@ -99,6 +101,7 @@ public class main {
return; return;
} }
int jobs = -1;
boolean allowOdex = false; boolean allowOdex = false;
boolean verboseErrors = false; boolean verboseErrors = false;
boolean printTokens = false; boolean printTokens = false;
@ -137,6 +140,9 @@ public class main {
case 'a': case 'a':
apiLevel = Integer.parseInt(commandLine.getOptionValue("a")); apiLevel = Integer.parseInt(commandLine.getOptionValue("a"));
break; break;
case 'j':
jobs = Integer.parseInt(commandLine.getOptionValue("j"));
break;
case 'V': case 'V':
verboseErrors = true; verboseErrors = true;
break; break;
@ -170,15 +176,46 @@ public class main {
} }
} }
if (jobs <= 0) {
jobs = Runtime.getRuntime().availableProcessors();
if (jobs > 6) {
jobs = 6;
}
}
boolean errors = false; boolean errors = false;
DexBuilder dexBuilder = DexBuilder.makeDexBuilder(); final DexBuilder dexBuilder = DexBuilder.makeDexBuilder();
ExecutorService executor = Executors.newFixedThreadPool(jobs);
List<Future<Boolean>> tasks = Lists.newArrayList();
for (File file: filesToProcess) { final boolean finalVerboseErrors = verboseErrors;
if (!assembleSmaliFile(file, dexBuilder, verboseErrors, printTokens, allowOdex, apiLevel)) { final boolean finalPrintTokens = printTokens;
final boolean finalAllowOdex = allowOdex;
final int finalApiLevel = apiLevel;
for (final File file: filesToProcess) {
tasks.add(executor.submit(new Callable<Boolean>() {
@Override public Boolean call() throws Exception {
return assembleSmaliFile(file, dexBuilder, finalVerboseErrors, finalPrintTokens,
finalAllowOdex, finalApiLevel);
}
}));
}
for (Future<Boolean> task: tasks) {
while(true) {
try {
if (!task.get()) {
errors = true; errors = true;
} }
} catch (InterruptedException ex) {
continue;
} }
break;
}
}
executor.shutdown();
if (errors) { if (errors) {
System.exit(1); System.exit(1);
@ -320,6 +357,13 @@ public class main {
.withArgName("API_LEVEL") .withArgName("API_LEVEL")
.create("a"); .create("a");
Option jobsOption = OptionBuilder.withLongOpt("jobs")
.withDescription("The number of threads to use. Defaults to the number of cores available, up to a " +
"maximum of 6")
.hasArg()
.withArgName("NUM_THREADS")
.create("j");
Option verboseErrorsOption = OptionBuilder.withLongOpt("verbose-errors") Option verboseErrorsOption = OptionBuilder.withLongOpt("verbose-errors")
.withDescription("Generate verbose error messages") .withDescription("Generate verbose error messages")
.create("V"); .create("V");
@ -333,6 +377,7 @@ public class main {
basicOptions.addOption(outputOption); basicOptions.addOption(outputOption);
basicOptions.addOption(allowOdexOption); basicOptions.addOption(allowOdexOption);
basicOptions.addOption(apiLevelOption); basicOptions.addOption(apiLevelOption);
basicOptions.addOption(jobsOption);
debugOptions.addOption(verboseErrorsOption); debugOptions.addOption(verboseErrorsOption);
debugOptions.addOption(printTokensOption); debugOptions.addOption(printTokensOption);