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
108
109
110
111
112
113
114
115
116
|
/*
* WarmRoast
* Copyright (C) 2013 Albert Pham <http://www.sk89q.com>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package com.sk89q.warmroast;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.apache.commons.io.FileUtils;
import au.com.bytecode.opencsv.CSVReader;
public class McpMapping {
private static final Pattern clPattern =
Pattern.compile("CL: (?<obfuscated>[^ ]+) (?<actual>[^ ]+)");
private static final Pattern mdPattern =
Pattern.compile("MD: (?<obfuscatedClass>[^ /]+)/(?<obfuscatedMethod>[^ ]+) " +
"[^ ]+ (?<method>[^ ]+) [^ ]+");
private final Map<String, ClassMapping> classes = new HashMap<>();
private final Map<String, String> methods = new HashMap<>();
public ClassMapping mapClass(String obfuscated) {
return classes.get(obfuscated);
}
public void read(File joinedFile, File methodsFile) throws IOException {
try (FileReader r = new FileReader(methodsFile)) {
try (CSVReader reader = new CSVReader(r)) {
List<String[]> entries = reader.readAll();
processMethodNames(entries);
}
}
List<String> lines = FileUtils.readLines(joinedFile, "UTF-8");
processClasses(lines);
processMethods(lines);
}
public String mapMethodId(String id) {
return methods.get(id);
}
public String fromMethodId(String id) {
String method = methods.get(id);
if (method == null) {
return id;
}
return method;
}
private void processMethodNames(List<String[]> entries) {
boolean first = true;
for (String[] entry : entries) {
if (entry.length < 2) {
continue;
}
if (first) { // Header
first = false;
continue;
}
methods.put(entry[0], entry[1]);
}
}
private void processClasses(List<String> lines) {
for (String line : lines) {
Matcher m = clPattern.matcher(line);
if (m.matches()) {
String obfuscated = m.group("obfuscated");
String actual = m.group("actual").replace("/", ".");
classes.put(obfuscated, new ClassMapping(obfuscated, actual));
}
}
}
private void processMethods(List<String> lines) {
for (String line : lines) {
Matcher m = mdPattern.matcher(line);
if (m.matches()) {
String obfuscatedClass = m.group("obfuscatedClass");
String obfuscatedMethod = m.group("obfuscatedMethod");
String method = m.group("method");
String methodId = method.substring(method.lastIndexOf('/') + 1);
ClassMapping mapping = mapClass(obfuscatedClass);
if (mapping != null) {
mapping.addMethod(obfuscatedMethod,
fromMethodId(methodId));
}
}
}
}
}
|
