ReVanced-Mirror/Apktool
1
0
Fork 0
mirror of https://github.com/revanced/Apktool.git synced 2025-02-01 22:57:41 +01:00
Code Issues Projects Releases Wiki Activity

[SmaliMod] adjust handling for stream, files and dexlib2

Browse Source
This commit is contained in:
Connor Tumbleson 2013-08-13 15:43:35 -05:00
parent 3acff3ef34
commit 5aa18eee2d
1 changed files with 62 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

103
brut.apktool/apktool-lib/src/main/java/brut/androlib/mod/SmaliMod.java
View File

@ -20,6 +20,7 @@ import java.io.*;
import org.antlr.runtime.*; import org.antlr.runtime.*;
import org.antlr.runtime.tree.CommonTree; import org.antlr.runtime.tree.CommonTree;
import org.antlr.runtime.tree.CommonTreeNodeStream; import org.antlr.runtime.tree.CommonTreeNodeStream;
import org.apache.commons.io.IOUtils;
import org.jf.dexlib2.writer.builder.DexBuilder; import org.jf.dexlib2.writer.builder.DexBuilder;
import org.jf.smali.*; import org.jf.smali.*;
@ -28,53 +29,73 @@ import org.jf.smali.*;
*/ */
public class SmaliMod { public class SmaliMod {
public static boolean assembleSmaliFile(File smaliFile,DexBuilder dexBuilder, boolean verboseErrors, public static boolean assembleSmaliFile(String smali, DexBuilder dexBuilder, boolean verboseErrors,
boolean printTokens) throws IOException, boolean printTokens, File smaliFile) throws IOException, RuntimeException, RecognitionException {
RecognitionException {
CommonTokenStream tokens; InputStream is = new ByteArrayInputStream(smali.getBytes());
LexerErrorInterface lexer; return assembleSmaliFile(is, dexBuilder, verboseErrors, printTokens, smaliFile);
}
FileInputStream fis = new FileInputStream(smaliFile.getAbsolutePath()); public static boolean assembleSmaliFile(InputStream is,DexBuilder dexBuilder, boolean verboseErrors,
InputStreamReader reader = new InputStreamReader(fis, "UTF-8"); boolean printTokens, File smaliFile) throws IOException, RecognitionException {
lexer = new smaliFlexLexer(reader); // copy our filestream into a tmp file, so we don't overwrite
((smaliFlexLexer)lexer).setSourceFile(smaliFile); File tmp = File.createTempFile("BRUT",".bak");
tokens = new CommonTokenStream((TokenSource) lexer); tmp.deleteOnExit();
if (printTokens) { OutputStream os = new FileOutputStream(tmp);
tokens.getTokens(); IOUtils.copy(is, os);
os.close();
for (int i=0; i<tokens.size(); i++) { return assembleSmaliFile(tmp,dexBuilder, verboseErrors, printTokens);
Token token = tokens.get(i); }
if (token.getChannel() == smaliParser.HIDDEN) {
continue; public static boolean assembleSmaliFile(File smaliFile,DexBuilder dexBuilder, boolean verboseErrors,
boolean printTokens) throws IOException, RecognitionException {
CommonTokenStream tokens;
LexerErrorInterface lexer;
InputStream is = new FileInputStream(smaliFile);
InputStreamReader reader = new InputStreamReader(is, "UTF-8");
lexer = new smaliFlexLexer(reader);
((smaliFlexLexer)lexer).setSourceFile(smaliFile);
tokens = new CommonTokenStream((TokenSource) lexer);
if (printTokens) {
tokens.getTokens();
for (int i=0; i<tokens.size(); i++) {
Token token = tokens.get(i);
if (token.getChannel() == smaliParser.HIDDEN) {
continue;
}
System.out.println(smaliParser.tokenNames[token.getType()] + ": " + token.getText());
}
} }
System.out.println(smaliParser.tokenNames[token.getType()] + ": " + token.getText()); smaliParser parser = new smaliParser(tokens);
} parser.setVerboseErrors(verboseErrors);
smaliParser.smali_file_return result = parser.smali_file();
if (parser.getNumberOfSyntaxErrors() > 0 || lexer.getNumberOfSyntaxErrors() > 0) {
return false;
}
CommonTree t = (CommonTree) result.getTree();
CommonTreeNodeStream treeStream = new CommonTreeNodeStream(t);
treeStream.setTokenStream(tokens);
smaliTreeWalker dexGen = new smaliTreeWalker(treeStream);
dexGen.setVerboseErrors(verboseErrors);
dexGen.setDexBuilder(dexBuilder);
dexGen.smali_file();
return dexGen.getNumberOfSyntaxErrors() == 0;
} }
smaliParser parser = new smaliParser(tokens);
parser.setVerboseErrors(verboseErrors);
smaliParser.smali_file_return result = parser.smali_file();
if (parser.getNumberOfSyntaxErrors() > 0 || lexer.getNumberOfSyntaxErrors() > 0) {
return false;
}
CommonTree t = (CommonTree) result.getTree();
CommonTreeNodeStream treeStream = new CommonTreeNodeStream(t);
treeStream.setTokenStream(tokens);
smaliTreeWalker dexGen = new smaliTreeWalker(treeStream);
dexGen.setVerboseErrors(verboseErrors);
dexGen.setDexBuilder(dexBuilder);
dexGen.smali_file();
return dexGen.getNumberOfSyntaxErrors() == 0;
}
} }