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

Look up ClassDefs in ClassPath using a hashmap

Browse Source
This commit is contained in:
Ben Gruver 2013-04-30 21:52:18 -07:00
parent 6893c660a3
commit 6fc32629c2
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
dexlib2/src/main/java/org/jf/dexlib2/analysis/ClassPath.java
View File

@ -55,6 +55,7 @@ public class ClassPath {
@Nonnull private final TypeProto unknownClass;
@Nonnull private DexFile[] dexFiles;
@Nonnull private HashMap<String, TypeProto> loadedClasses = Maps.newHashMap();
@Nonnull private HashMap<String, ClassDef> availableClasses = Maps.newHashMap();
/**
* Creates a new ClassPath instance that can load classes from the given dex files
@ -96,6 +97,15 @@ public class ClassPath {
loadPrimitiveType("F");
loadPrimitiveType("D");
loadPrimitiveType("L");
for (DexFile dexFile: dexFiles) {
for (ClassDef classDef: dexFile.getClasses()) {
ClassDef prev = availableClasses.get(classDef.getType());
if (prev == null) {
availableClasses.put(classDef.getType(), classDef);
}
}
}
}
private void loadPrimitiveType(String type) {
@ -134,15 +144,11 @@ public class ClassPath {
@Nonnull
public ClassDef getClassDef(String type) {
// TODO: need a <= O(log) way to look up classes
for (DexFile dexFile: dexFiles) {
for (ClassDef classDef: dexFile.getClasses()) {
if (classDef.getType().equals(type)) {
return classDef;
}
}
ClassDef ret = availableClasses.get(type);
if (ret == null) {
throw new UnresolvedClassException("Could not resolve class %s", type);
}
throw new UnresolvedClassException("Could not resolve class %s", type);
return ret;
}
@Nonnull