use of org.jf.smali.LexerErrorInterface in project atlas by alibaba.
the class SmaliMod method assembleSmaliFile.
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());
}
}
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;
}
use of org.jf.smali.LexerErrorInterface in project otertool by wuntee.
the class SmaliWorkshop method buildDex.
private static void buildDex(File smaliSourceDirectory, File destination) throws RecognitionException, SmaliSyntaxException, SmaliDexException, IOException {
DexFile dexFile = new DexFile();
// Load files into set
logger.debug("Loading smali files");
LinkedHashSet<File> filesToProcess = new LinkedHashSet<File>();
getSmaliFilesInDir(smaliSourceDirectory, filesToProcess);
// Process each file
logger.debug("Processing files");
for (File file : filesToProcess) {
logger.debug("Processing: " + file.getAbsolutePath());
FileInputStream fis = new FileInputStream(file.getAbsolutePath());
InputStreamReader reader = new InputStreamReader(fis, "UTF-8");
LexerErrorInterface lexer = new smaliFlexLexer(reader);
((smaliFlexLexer) lexer).setSourceFile(file);
CommonTokenStream tokens = new CommonTokenStream((TokenSource) lexer);
smaliParser parser = new smaliParser(tokens);
smaliParser.smali_file_return result = parser.smali_file();
// Errors
if (parser.getNumberOfSyntaxErrors() > 0) {
throw new SmaliSyntaxException(file, parser.getNumberOfSyntaxErrors());
}
if (lexer.getNumberOfSyntaxErrors() > 0) {
throw new SmaliSyntaxException(file, lexer.getNumberOfSyntaxErrors());
}
CommonTree t = (CommonTree) result.getTree();
CommonTreeNodeStream treeStream = new CommonTreeNodeStream(t);
treeStream.setTokenStream(tokens);
smaliTreeWalker dexGen = new smaliTreeWalker(treeStream);
dexGen.dexFile = dexFile;
dexGen.smali_file();
if (dexGen.getNumberOfSyntaxErrors() > 0) {
throw new SmaliDexException(file, dexGen.getNumberOfSyntaxErrors());
}
}
// Calculate signatures and write file
logger.debug("Writing dex file.");
dexFile.place();
ByteArrayAnnotatedOutput out = new ByteArrayAnnotatedOutput();
dexFile.writeTo(out);
byte[] bytes = out.toByteArray();
DexFile.calcSignature(bytes);
DexFile.calcChecksum(bytes);
FileOutputStream fileOutputStream = new FileOutputStream(destination);
fileOutputStream.write(bytes);
fileOutputStream.close();
}
Aggregations