-
Notifications
You must be signed in to change notification settings - Fork 4.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Browse files
Browse the repository at this point in the history
* Implement proto parse * fix code formatting * fix tests with empty input * revert not needed code style changes * Implement parse of resources.pb for AAB Co-authored-by: bagipro <bugi@MacBook-Pro-2.local> Co-authored-by: Skylot <skylot@gmail.com>
- Loading branch information
3 people
committed
Mar 8, 2021
1 parent
4e5fac4
commit 9ef99a2
Showing
19 changed files
with
725 additions
and
138 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
141 changes: 141 additions & 0 deletions
141
jadx-core/src/main/java/jadx/core/xmlgen/ProtoXMLParser.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,141 @@ | ||
| package jadx.core.xmlgen; | ||
|
|
||
| import java.io.IOException; | ||
| import java.io.InputStream; | ||
| import java.util.HashMap; | ||
| import java.util.Map; | ||
| import java.util.Random; | ||
|
|
||
| import com.android.aapt.Resources.XmlAttribute; | ||
| import com.android.aapt.Resources.XmlElement; | ||
| import com.android.aapt.Resources.XmlNamespace; | ||
| import com.android.aapt.Resources.XmlNode; | ||
| import com.google.protobuf.InvalidProtocolBufferException; | ||
|
|
||
| import jadx.api.ICodeInfo; | ||
| import jadx.api.ICodeWriter; | ||
| import jadx.core.dex.nodes.RootNode; | ||
| import jadx.core.utils.StringUtils; | ||
|
|
||
| public class ProtoXMLParser { | ||
| private Map<String, String> nsMap; | ||
| private final Map<String, String> tagAttrDeobfNames = new HashMap<>(); | ||
|
|
||
| private ICodeWriter writer; | ||
|
|
||
| private final RootNode rootNode; | ||
| private String currentTag; | ||
| private String appPackageName; | ||
|
|
||
| public ProtoXMLParser(RootNode rootNode) { | ||
| this.rootNode = rootNode; | ||
| } | ||
|
|
||
| public synchronized ICodeInfo parse(InputStream inputStream) throws IOException { | ||
| nsMap = new HashMap<>(); | ||
| writer = rootNode.makeCodeWriter(); | ||
| writer.add("<?xml version=\"1.0\" encoding=\"utf-8\"?>"); | ||
| decode(decodeProto(inputStream)); | ||
| nsMap = null; | ||
| return writer.finish(); | ||
| } | ||
|
|
||
| private void decode(XmlNode n) throws IOException { | ||
| if (n.hasSource()) { | ||
| writer.attachSourceLine(n.getSource().getLineNumber()); | ||
| } | ||
| writer.add(StringUtils.escapeXML(n.getText().trim())); | ||
| if (n.hasElement()) { | ||
| decode(n.getElement()); | ||
| } | ||
| } | ||
|
|
||
| private void decode(XmlElement e) throws IOException { | ||
| String tag = deobfClassName(e.getName()); | ||
| tag = getValidTagAttributeName(tag); | ||
| currentTag = tag; | ||
| writer.startLine('<').add(tag); | ||
| for (int i = 0; i < e.getNamespaceDeclarationCount(); i++) { | ||
| decode(e.getNamespaceDeclaration(i)); | ||
| } | ||
| for (int i = 0; i < e.getAttributeCount(); i++) { | ||
| decode(e.getAttribute(i)); | ||
| } | ||
| if (e.getChildCount() > 0) { | ||
| writer.add('>'); | ||
| writer.incIndent(); | ||
| for (int i = 0; i < e.getChildCount(); i++) { | ||
| Map<String, String> oldNsMap = new HashMap<>(nsMap); | ||
| decode(e.getChild(i)); | ||
| nsMap = oldNsMap; | ||
| } | ||
| writer.decIndent(); | ||
| writer.startLine("</").add(tag).add('>'); | ||
| } else { | ||
| writer.add("/>"); | ||
| } | ||
| } | ||
|
|
||
| private void decode(XmlAttribute a) { | ||
| writer.add(' '); | ||
| String namespace = a.getNamespaceUri(); | ||
| if (!namespace.isEmpty()) { | ||
| writer.add(nsMap.get(namespace)).add(':'); | ||
| } | ||
| String name = a.getName(); | ||
| String value = deobfClassName(a.getValue()); | ||
| writer.add(name).add("=\"").add(value).add('\"'); | ||
| memorizePackageName(name, value); | ||
| } | ||
|
|
||
| private void decode(XmlNamespace n) { | ||
| String prefix = n.getPrefix(); | ||
| String uri = n.getUri(); | ||
| nsMap.put(uri, prefix); | ||
| writer.add(" xmlns:").add(prefix).add("=\"").add(uri).add('"'); | ||
| } | ||
|
|
||
| private void memorizePackageName(String attrName, String attrValue) { | ||
| if ("manifest".equals(currentTag) && "package".equals(attrName)) { | ||
| appPackageName = attrValue; | ||
| } | ||
| } | ||
|
|
||
| private String deobfClassName(String className) { | ||
| String newName = XmlDeobf.deobfClassName(rootNode, className, appPackageName); | ||
| if (newName != null) { | ||
| return newName; | ||
| } | ||
| return className; | ||
| } | ||
|
|
||
| private String getValidTagAttributeName(String originalName) { | ||
| if (XMLChar.isValidName(originalName)) { | ||
| return originalName; | ||
| } | ||
| if (tagAttrDeobfNames.containsKey(originalName)) { | ||
| return tagAttrDeobfNames.get(originalName); | ||
| } | ||
| String generated; | ||
| do { | ||
| generated = generateTagAttrName(); | ||
| } while (tagAttrDeobfNames.containsValue(generated)); | ||
| tagAttrDeobfNames.put(originalName, generated); | ||
| return generated; | ||
| } | ||
|
|
||
| private static String generateTagAttrName() { | ||
| final int length = 6; | ||
| Random r = new Random(); | ||
| StringBuilder sb = new StringBuilder(); | ||
| for (int i = 1; i <= length; i++) { | ||
| sb.append((char) (r.nextInt(26) + 'a')); | ||
| } | ||
| return sb.toString(); | ||
| } | ||
|
|
||
| private XmlNode decodeProto(InputStream inputStream) | ||
| throws InvalidProtocolBufferException, IOException { | ||
| return XmlNode.parseFrom(XmlGenUtils.readData(inputStream)); | ||
| } | ||
| } |
Oops, something went wrong.