blob: 9d238e3ada1093d7c40c4049f40be86da9e6d9b2 (
plain) (
blame)
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
|
package cuchaz.enigma.command;
import cuchaz.enigma.Enigma;
import cuchaz.enigma.EnigmaProject;
import cuchaz.enigma.ProgressListener;
import cuchaz.enigma.analysis.index.JarIndex;
import cuchaz.enigma.translation.mapping.EntryMapping;
import cuchaz.enigma.translation.mapping.MappingSaveParameters;
import cuchaz.enigma.translation.mapping.serde.MappingFormat;
import cuchaz.enigma.translation.mapping.tree.EntryTree;
import cuchaz.enigma.translation.representation.entry.ClassEntry;
import java.nio.file.Path;
import java.util.Set;
import java.util.stream.Collectors;
public class CheckMappingsCommand extends Command {
public CheckMappingsCommand() {
super("checkmappings");
}
@Override
public String getUsage() {
return "<in jar> <mappings file>";
}
@Override
public boolean isValidArgument(int length) {
return length == 2;
}
@Override
public void run(String... args) throws Exception {
Path fileJarIn = getReadableFile(getArg(args, 0, "in jar", true)).toPath();
Path fileMappings = getReadablePath(getArg(args, 1, "mappings file", true));
Enigma enigma = Enigma.create();
System.out.println("Reading JAR...");
EnigmaProject project = enigma.openJar(fileJarIn, ProgressListener.none());
System.out.println("Reading mappings...");
MappingFormat format = chooseEnigmaFormat(fileMappings);
MappingSaveParameters saveParameters = enigma.getProfile().getMappingSaveParameters();
EntryTree<EntryMapping> mappings = format.read(fileMappings, ProgressListener.none(), saveParameters);
project.setMappings(mappings);
JarIndex idx = project.getJarIndex();
boolean error = false;
for (Set<ClassEntry> partition : idx.getPackageVisibilityIndex().getPartitions()) {
long packages = partition.stream()
.map(project.getMapper()::deobfuscate)
.map(ClassEntry::getPackageName)
.distinct()
.count();
if (packages > 1) {
error = true;
System.err.println("ERROR: Must be in one package:\n" + partition.stream()
.map(project.getMapper()::deobfuscate)
.map(ClassEntry::toString)
.sorted()
.collect(Collectors.joining("\n"))
);
}
}
if (error) {
throw new IllegalStateException("Errors in package visibility detected, see SysErr above");
}
}
}
|
