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
98
99
100
101
102
103
104
105
106
107
|
/*******************************************************************************
* 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
*
* Contributors:
* Jeff Martin - initial API and implementation
******************************************************************************/
package cuchaz.enigma.mapping;
import java.util.Map;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import cuchaz.enigma.analysis.JarIndex;
import cuchaz.enigma.analysis.RelatedMethodChecker;
public class MappingsChecker {
private JarIndex m_index;
private RelatedMethodChecker m_relatedMethodChecker;
private Map<ClassEntry,ClassMapping> m_droppedClassMappings;
private Map<ClassEntry,ClassMapping> m_droppedInnerClassMappings;
private Map<FieldEntry,FieldMapping> m_droppedFieldMappings;
private Map<BehaviorEntry,MethodMapping> m_droppedMethodMappings;
public MappingsChecker(JarIndex index) {
m_index = index;
m_relatedMethodChecker = new RelatedMethodChecker(m_index);
m_droppedClassMappings = Maps.newHashMap();
m_droppedInnerClassMappings = Maps.newHashMap();
m_droppedFieldMappings = Maps.newHashMap();
m_droppedMethodMappings = Maps.newHashMap();
}
public RelatedMethodChecker getRelatedMethodChecker() {
return m_relatedMethodChecker;
}
public Map<ClassEntry,ClassMapping> getDroppedClassMappings() {
return m_droppedClassMappings;
}
public Map<ClassEntry,ClassMapping> getDroppedInnerClassMappings() {
return m_droppedInnerClassMappings;
}
public Map<FieldEntry,FieldMapping> getDroppedFieldMappings() {
return m_droppedFieldMappings;
}
public Map<BehaviorEntry,MethodMapping> getDroppedMethodMappings() {
return m_droppedMethodMappings;
}
public void dropBrokenMappings(Mappings mappings) {
for (ClassMapping classMapping : Lists.newArrayList(mappings.classes())) {
if (!checkClassMapping(classMapping)) {
mappings.removeClassMapping(classMapping);
m_droppedClassMappings.put(EntryFactory.getObfClassEntry(m_index, classMapping), classMapping);
}
}
}
private boolean checkClassMapping(ClassMapping classMapping) {
// check the class
ClassEntry classEntry = EntryFactory.getObfClassEntry(m_index, classMapping);
if (!m_index.getObfClassEntries().contains(classEntry)) {
return false;
}
// check the fields
for (FieldMapping fieldMapping : Lists.newArrayList(classMapping.fields())) {
FieldEntry obfFieldEntry = EntryFactory.getObfFieldEntry(classMapping, fieldMapping);
if (!m_index.containsObfField(obfFieldEntry)) {
classMapping.removeFieldMapping(fieldMapping);
m_droppedFieldMappings.put(obfFieldEntry, fieldMapping);
}
}
// check methods
for (MethodMapping methodMapping : Lists.newArrayList(classMapping.methods())) {
BehaviorEntry obfBehaviorEntry = EntryFactory.getObfBehaviorEntry(classEntry, methodMapping);
if (!m_index.containsObfBehavior(obfBehaviorEntry)) {
classMapping.removeMethodMapping(methodMapping);
m_droppedMethodMappings.put(obfBehaviorEntry, methodMapping);
}
m_relatedMethodChecker.checkMethod(classEntry, methodMapping);
}
// check inner classes
for (ClassMapping innerClassMapping : Lists.newArrayList(classMapping.innerClasses())) {
if (!checkClassMapping(innerClassMapping)) {
classMapping.removeInnerClassMapping(innerClassMapping);
m_droppedInnerClassMappings.put(EntryFactory.getObfClassEntry(m_index, innerClassMapping), innerClassMapping);
}
}
return true;
}
}
|
