From af1041731c8c0ce1846ff7e7b6052ed7327a5dbc Mon Sep 17 00:00:00 2001
From: jeff
Date: Mon, 9 Feb 2015 22:23:45 -0500
Subject: fix translation issues, particularly with fields

---
 .../enigma/analysis/RelatedMethodChecker.java      | 96 ++++++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100644 src/cuchaz/enigma/analysis/RelatedMethodChecker.java

(limited to 'src/cuchaz/enigma/analysis/RelatedMethodChecker.java')

diff --git a/src/cuchaz/enigma/analysis/RelatedMethodChecker.java b/src/cuchaz/enigma/analysis/RelatedMethodChecker.java
new file mode 100644
index 0000000..5bd67a0
--- /dev/null
+++ b/src/cuchaz/enigma/analysis/RelatedMethodChecker.java
@@ -0,0 +1,96 @@
+package cuchaz.enigma.analysis;
+
+import java.util.Map;
+import java.util.Set;
+
+import com.beust.jcommander.internal.Maps;
+import com.google.common.collect.Sets;
+
+import cuchaz.enigma.mapping.BehaviorEntry;
+import cuchaz.enigma.mapping.ClassEntry;
+import cuchaz.enigma.mapping.EntryFactory;
+import cuchaz.enigma.mapping.MethodEntry;
+import cuchaz.enigma.mapping.MethodMapping;
+
+public class RelatedMethodChecker {
+	
+	private JarIndex m_jarIndex;
+	private Map<Set<MethodEntry>,String> m_deobfNamesByGroup;
+	private Map<MethodEntry,String> m_deobfNamesByObfMethod;
+	private Map<MethodEntry,Set<MethodEntry>> m_groupsByObfMethod;
+	private Set<Set<MethodEntry>> m_inconsistentGroups;
+	
+	public RelatedMethodChecker(JarIndex jarIndex) {
+		m_jarIndex = jarIndex;
+		m_deobfNamesByGroup = Maps.newHashMap();
+		m_deobfNamesByObfMethod = Maps.newHashMap();
+		m_groupsByObfMethod = Maps.newHashMap();
+		m_inconsistentGroups = Sets.newHashSet();
+	}
+	
+	public void checkMethod(ClassEntry classEntry, MethodMapping methodMapping) {
+		
+		// TEMP: disable the expensive check for now, maybe we can optimize it later, or just use it for debugging
+		if (true) return;
+		
+		BehaviorEntry obfBehaviorEntry = EntryFactory.getObfBehaviorEntry(classEntry, methodMapping);
+		if (!(obfBehaviorEntry instanceof MethodEntry)) {
+			// only methods have related implementations
+			return;
+		}
+		MethodEntry obfMethodEntry = (MethodEntry)obfBehaviorEntry;
+		String deobfName = methodMapping.getDeobfName();
+		m_deobfNamesByObfMethod.put(obfMethodEntry, deobfName);
+		
+		// have we seen this method's group before?
+		Set<MethodEntry> group = m_groupsByObfMethod.get(obfMethodEntry);
+		if (group == null) {
+			
+			// no, compute the group and save the name
+			group = m_jarIndex.getRelatedMethodImplementations(obfMethodEntry);
+			m_deobfNamesByGroup.put(group, deobfName);
+			
+			assert(group.contains(obfMethodEntry));
+			for (MethodEntry relatedMethodEntry : group) {
+				m_groupsByObfMethod.put(relatedMethodEntry, group);
+			}
+		}
+		
+		// check the name
+		if (!sameName(m_deobfNamesByGroup.get(group), deobfName)) {
+			m_inconsistentGroups.add(group);
+		}
+	}
+	
+	private boolean sameName(String a, String b) {
+		if (a == null && b == null) {
+			return true;
+		} else if (a != null && b != null) {
+			return a.equals(b);
+		}
+		return false;
+	}
+
+	public boolean hasProblems() {
+		return m_inconsistentGroups.size() > 0;
+	}
+	
+	public String getReport() {
+		StringBuilder buf = new StringBuilder();
+		buf.append(m_inconsistentGroups.size());
+		buf.append(" groups of methods related by inheritance and/or interfaces have different deobf names!\n");
+		for (Set<MethodEntry> group : m_inconsistentGroups) {
+			buf.append("\tGroup with ");
+			buf.append(group.size());
+			buf.append(" methods:\n");
+			for (MethodEntry methodEntry : group) {
+				buf.append("\t\t");
+				buf.append(methodEntry.toString());
+				buf.append(" => ");
+				buf.append(m_deobfNamesByObfMethod.get(methodEntry));
+				buf.append("\n");
+			}
+		}
+		return buf.toString();
+	}
+}
-- 
cgit v1.2.3