-
-
Notifications
You must be signed in to change notification settings - Fork 44
/
PatchMain.java
93 lines (85 loc) · 3.29 KB
/
PatchMain.java
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
package me.nallar.tickthreading.patcher;
import javax.xml.transform.TransformerException;
import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.logging.Level;
import javassist.NotFoundException;
import me.nallar.tickthreading.Log;
import me.nallar.tickthreading.mappings.MCPMappings;
import me.nallar.tickthreading.mappings.Mappings;
import me.nallar.tickthreading.util.CollectionsUtil;
import me.nallar.tickthreading.util.VersionUtil;
import org.xml.sax.SAXException;
public class PatchMain {
public static void main(String[] argv) {
Log.info("Running " + VersionUtil.versionString());
if (argv.length < 1) {
Log.severe("Type must be passed");
return;
}
String type = argv[0];
String[] args = Arrays.copyOfRange(argv, 1, argv.length);
if ("obfuscator".equalsIgnoreCase(type)) {
obfuscator(args);
} else if ("patcher".equalsIgnoreCase(type)) {
patcher(args);
} else {
Log.severe(type + " is not a valid action.");
}
}
private static void obfuscator(String[] args) {
String mcpConfigPath = args.length > 0 ? args[0] : "build/forge/mcp/conf";
String inputPatchPath = args.length > 1 ? args[1] : "resources/patches-deobfuscated.xml";
String outputPatchPath = args.length > 2 ? args[2] : "build/classes/patches.xml";
Log.info("Obfuscating " + inputPatchPath + " to " + outputPatchPath + " via " + mcpConfigPath);
try {
Mappings mcpMappings = new MCPMappings(new File(mcpConfigPath));
PatchManager patchManager = new PatchManager(new File(inputPatchPath).toURI().toURL().openStream(), Patches.class);
try {
patchManager.obfuscate(mcpMappings);
patchManager.save(new File(outputPatchPath));
} catch (TransformerException e) {
Log.severe("Failed to save obfuscated patch");
}
} catch (IOException e) {
Log.severe("Failed to read input file", e);
} catch (SAXException e) {
Log.severe("Failed to parse input file", e);
}
}
private static void patcher(String[] args) {
Log.setFileName("patcher", Level.FINEST, Log.LOGGER);
// TODO: Auto force-patch if Patches.class changes
boolean forcePatching = true;// args.length >= 2 && "force".equalsIgnoreCase(args[1]);
PatchManager patchManager;
try {
patchManager = new PatchManager(PatchMain.class.getResourceAsStream("/patches.xml"), Patches.class);
} catch (Exception e) {
Log.severe("Failed to initialize Patch Manager", e);
return;
}
try {
Iterable<File> filesToLoad = (Iterable<File>) CollectionsUtil.toObjects(CollectionsUtil.split(args[0]), File.class);
patchManager.classRegistry.forcePatching = forcePatching;
patchManager.loadBackups(filesToLoad);
patchManager.classRegistry.loadFiles(filesToLoad);
try {
patchManager.classRegistry.getClass("org.bukkit.craftbukkit.util.WatchdogThread");
patchManager.patchEnvironment = "mcpc";
} catch (NotFoundException ignored) {
}
Log.info("Patching in environment: " + patchManager.patchEnvironment);
patchManager.runPatches();
} catch (IOException e) {
Log.severe("Failed to load jars", e);
} catch (Exception e) {
Log.severe("Unhandled patching failure", e);
}
try {
Log.info("Done. Press enter to exit.");
System.console().readLine();
} catch (Exception ignored) {
}
}
}