1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
/*******************************************************************************
* Copyright (c) 2015 Jeff Martin.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the GNU Lesser General Public
* License v3.0 which accompanies this distribution, and is available at
* http://www.gnu.org/licenses/lgpl.html
* <p>
* Contributors:
* Jeff Martin - initial API and implementation
******************************************************************************/
package cuchaz.enigma.mapping;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import cuchaz.enigma.analysis.JarIndex;
import java.util.Map;
public class MappingsChecker {
private JarIndex index;
private Map<ClassEntry, ClassMapping> droppedClassMappings;
private Map<ClassEntry, ClassMapping> droppedInnerClassMappings;
private Map<FieldEntry, FieldMapping> droppedFieldMappings;
private Map<MethodEntry, MethodMapping> droppedMethodMappings;
public MappingsChecker(JarIndex index) {
this.index = index;
this.droppedClassMappings = Maps.newHashMap();
this.droppedInnerClassMappings = Maps.newHashMap();
this.droppedFieldMappings = Maps.newHashMap();
this.droppedMethodMappings = Maps.newHashMap();
}
public Map<ClassEntry, ClassMapping> getDroppedClassMappings() {
return this.droppedClassMappings;
}
public Map<ClassEntry, ClassMapping> getDroppedInnerClassMappings() {
return this.droppedInnerClassMappings;
}
public Map<FieldEntry, FieldMapping> getDroppedFieldMappings() {
return this.droppedFieldMappings;
}
public Map<MethodEntry, MethodMapping> getDroppedMethodMappings() {
return this.droppedMethodMappings;
}
public void dropBrokenMappings(Mappings mappings) {
for (ClassMapping classMapping : Lists.newArrayList(mappings.classes())) {
if (!checkClassMapping(classMapping)) {
mappings.removeClassMapping(classMapping);
this.droppedClassMappings.put(EntryFactory.getObfClassEntry(this.index, classMapping), classMapping);
}
}
}
private boolean checkClassMapping(ClassMapping classMapping) {
// check the class
ClassEntry classEntry = EntryFactory.getObfClassEntry(this.index, classMapping);
if (!this.index.getObfClassEntries().contains(classEntry)) {
return false;
}
// check the fields
for (FieldMapping fieldMapping : Lists.newArrayList(classMapping.fields())) {
FieldEntry obfFieldEntry = EntryFactory.getObfFieldEntry(classMapping, fieldMapping);
if (!this.index.containsObfField(obfFieldEntry)) {
classMapping.removeFieldMapping(fieldMapping);
this.droppedFieldMappings.put(obfFieldEntry, fieldMapping);
}
}
// check methods
for (MethodMapping methodMapping : Lists.newArrayList(classMapping.methods())) {
MethodEntry obfMethodEntry = EntryFactory.getObfMethodEntry(classEntry, methodMapping);
if (!this.index.containsObfMethod(obfMethodEntry)) {
classMapping.removeMethodMapping(methodMapping);
this.droppedMethodMappings.put(obfMethodEntry, methodMapping);
}
}
// check inner classes
for (ClassMapping innerClassMapping : Lists.newArrayList(classMapping.innerClasses())) {
if (!checkClassMapping(innerClassMapping)) {
classMapping.removeInnerClassMapping(innerClassMapping);
this.droppedInnerClassMappings.put(EntryFactory.getObfClassEntry(this.index, innerClassMapping), innerClassMapping);
}
}
return true;
}
}
|
