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

+DexFileBuilder

Browse Source
This commit is contained in:
Ryszard Wiśniewski 2010-04-26 18:28:12 +02:00
parent 3955e7082c
commit 5afc491ebc
2 changed files with 121 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

88
src/brut/androlib/AndrolibSmali.java
View File

@ -17,24 +17,13 @@
package brut.androlib;
import brut.androlib.util.DexFileBuilder;
import brut.directory.DirectoryException;
import brut.directory.FileDirectory;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import org.antlr.runtime.ANTLRInputStream;
import org.antlr.runtime.CommonTokenStream;
import org.antlr.runtime.RecognitionException;
import org.antlr.runtime.tree.CommonTree;
import org.antlr.runtime.tree.CommonTreeNodeStream;
import org.jf.baksmali.baksmali;
import org.jf.dexlib.CodeItem;
import org.jf.dexlib.DexFile;
import org.jf.dexlib.Util.ByteArrayAnnotatedOutput;
import org.jf.smali.smaliLexer;
import org.jf.smali.smaliParser;
import org.jf.smali.smaliTreeWalker;
/**
* @author Ryszard Wiśniewski <brut.alll@gmail.com>
@ -53,77 +42,20 @@ public class AndrolibSmali {
}
}
public void smali(String dir, String dexFile) throws AndrolibException {
smali(new File(dir), new File(dexFile));
}
public void smali(File dir, File dexFile) throws AndrolibException {
smali(dir.getAbsolutePath(), dexFile.getAbsolutePath());
}
public void smali(String dir, String dexFileName) throws AndrolibException {
try {
DexFile dexFile = new DexFile();
DexFileBuilder builder = new DexFileBuilder();
for (String fileName : new FileDirectory(dir).getFiles(true)) {
if (! assembleSmaliFile(
new File(dir + "/" + fileName), dexFile)) {
throw new AndrolibException(
"Could not smali file: " + fileName);
}
builder.addSmaliFile(new File(dir + "/" + fileName));
}
dexFile.place();
for (CodeItem codeItem: dexFile.CodeItemsSection.getItems()) {
codeItem.fixInstructions(true, true);
}
dexFile.place();
ByteArrayAnnotatedOutput out = new ByteArrayAnnotatedOutput();
dexFile.writeTo(out);
byte[] bytes = out.toByteArray();
DexFile.calcSignature(bytes);
DexFile.calcChecksum(bytes);
FileOutputStream fileOutputStream = new FileOutputStream(dexFileName);
fileOutputStream.write(bytes);
fileOutputStream.close();
} catch (IOException ex) {
throw new AndrolibException("Could not smali files", ex);
builder.writeTo(dexFile);
} catch (DirectoryException ex) {
throw new AndrolibException("Could not smali files", ex);
} catch (RecognitionException ex) {
throw new AndrolibException("Could not smali files", ex);
throw new AndrolibException(
"Could not smali " + dir + " to " + dexFile, ex);
}
}
private static boolean assembleSmaliFile(File smaliFile, DexFile dexFile) throws IOException, RecognitionException {
ANTLRInputStream input = new ANTLRInputStream(new FileInputStream(smaliFile));
input.name = smaliFile.getAbsolutePath();
smaliLexer lexer = new smaliLexer(input);
CommonTokenStream tokens = new CommonTokenStream(lexer);
smaliParser parser = new smaliParser(tokens);
smaliParser.smali_file_return result = parser.smali_file();
if (parser.getNumberOfSyntaxErrors() > 0 || lexer.getNumberOfLexerErrors() > 0) {
return false;
}
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) {
return false;
}
return true;
}
}

111
src/brut/androlib/util/DexFileBuilder.java Normal file
View File

@ -0,0 +1,111 @@
/*
* Copyright 2010 Ryszard Wiśniewski <brut.alll@gmail.com>.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* under the License.
*/
package brut.androlib.util;
import brut.androlib.AndrolibException;
import java.io.*;
import org.antlr.runtime.*;
import org.antlr.runtime.tree.CommonTree;
import org.antlr.runtime.tree.CommonTreeNodeStream;
import org.jf.dexlib.CodeItem;
import org.jf.dexlib.DexFile;
import org.jf.dexlib.Util.ByteArrayAnnotatedOutput;
import org.jf.smali.*;
/**
* @author Ryszard Wiśniewski <brut.alll@gmail.com>
*/
public class DexFileBuilder {
public void addSmaliFile(File smaliFile) throws AndrolibException {
try {
if (!assembleSmaliFile(smaliFile)) {
throw new AndrolibException(
"Could not smali file: " + smaliFile);
}
} catch (IOException ex) {
throw new AndrolibException(
"Could not smali file: " + smaliFile, ex);
} catch (RecognitionException ex) {
throw new AndrolibException(
"Could not smali file: " + smaliFile, ex);
}
}
public void writeTo(File dexFile) throws AndrolibException {
try {
OutputStream out = new FileOutputStream(dexFile);
out.write(getAsByteArray());
out.close();
} catch (IOException ex) {
throw new AndrolibException(
"Could not write dex to file: " + dexFile, ex);
}
}
public byte[] getAsByteArray() {
mDexFile.place();
for (CodeItem codeItem: mDexFile.CodeItemsSection.getItems()) {
codeItem.fixInstructions(true, true);
}
mDexFile.place();
ByteArrayAnnotatedOutput out = new ByteArrayAnnotatedOutput();
mDexFile.writeTo(out);
byte[] bytes = out.toByteArray();
DexFile.calcSignature(bytes);
DexFile.calcChecksum(bytes);
return bytes;
}
private boolean assembleSmaliFile(File smaliFile)
throws IOException, RecognitionException {
ANTLRInputStream input = new ANTLRInputStream(new FileInputStream(smaliFile), "UTF8");
input.name = smaliFile.getAbsolutePath();
smaliLexer lexer = new smaliLexer(input);
CommonTokenStream tokens = new CommonTokenStream(lexer);
smaliParser parser = new smaliParser(tokens);
smaliParser.smali_file_return result = parser.smali_file();
if (parser.getNumberOfSyntaxErrors() > 0 || lexer.getNumberOfLexerErrors() > 0) {
return false;
}
CommonTree t = (CommonTree) result.getTree();
CommonTreeNodeStream treeStream = new CommonTreeNodeStream(t);
treeStream.setTokenStream(tokens);
smaliTreeWalker dexGen = new smaliTreeWalker(treeStream);
dexGen.dexFile = mDexFile;
dexGen.smali_file();
if (dexGen.getNumberOfSyntaxErrors() > 0) {
return false;
}
return true;
}
private final DexFile mDexFile = new DexFile();
}