blob: d5049b4bcce0e3eacf6ed893eca51b9983a523af (
plain)
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
117
118
119
120
121
122
|
package makamys.neodymium.util.virtualjar;
import static makamys.neodymium.Constants.MODID;
import static makamys.neodymium.Constants.PROTOCOL;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.util.HashMap;
import java.util.Map;
import static makamys.neodymium.Constants.LOGGER;
import makamys.neodymium.util.virtualjar.protocol.neodymiumvirtualjar.Handler;
import net.minecraft.launchwrapper.Launch;
public class VirtualJar {
private static boolean registered;
private static Map<String, IVirtualJar> jars = new HashMap<>();
public static void registerHandler() {
if(registered) return;
LOGGER.debug("Registering URL protocol handler: " + PROTOCOL);
// We want the Handler to always be loaded by the same class loader.
Launch.classLoader.addClassLoaderExclusion("makamys." + MODID + ".util.virtualjar.protocol." + PROTOCOL);
// The Handler is loaded by the AppClassLoader, but it needs to access the state of VirtualJar, which is loaded
// by the LaunchClassLoader. The solution? Make the Handler just a proxy that delegates the real work to
// VirtualJar.StreamHandlerImpl. We use the blackboard as a class loader-agnostic way of sharing information.
Handler.IURLStreamHandlerImpl streamHandlerImpl = new StreamHandlerImpl();
Launch.blackboard.put(MODID + "." + PROTOCOL + ".impl", streamHandlerImpl);
URLStreamHandlerHelper.register(Handler.class);
registered = true;
}
public static void add(IVirtualJar jar) {
registerHandler();
LOGGER.trace("Adding virtual jar to class path: " + PROTOCOL + ":" + jar.getName() + ".jar");
String urlStr = PROTOCOL + ":" + jar.getName() + ".jar!/";
try {
URL url = new URL(urlStr);
Launch.classLoader.addURL(url);
// Forge expects all URLs in the sources list to be convertible to File objects, so we must remove it to
// avoid a crash.
Launch.classLoader.getSources().remove(url);
jars.put(jar.getName(), jar);
} catch(MalformedURLException e) {
LOGGER.fatal("Failed to add virtual jar to class path");
e.printStackTrace();
}
}
public static class StreamHandlerImpl implements Handler.IURLStreamHandlerImpl {
@Override
public URLConnection openConnection(URL url) throws IOException {
return new VirtualJarConnection(url);
}
public static class VirtualJarConnection extends URLConnection {
IVirtualJar jar;
private String filePath;
public VirtualJarConnection(URL url) throws IOException {
super(url);
String path = url.getPath();
String nameSuffix = ".jar!";
int nameEnd = path.indexOf(nameSuffix);
String name = path.substring(0, nameEnd);
jar = jars.get(name);
if(jar == null) {
throw new IOException();
}
filePath = path.substring(nameEnd + nameSuffix.length());
if(!jar.hasFile(filePath)) {
throw new IOException();
}
}
public void connect() throws IOException {
}
public Object getContent() throws IOException {
return super.getContent();
}
public String getHeaderField(String name) {
return super.getHeaderField(name);
}
public InputStream getInputStream() {
return jar.getInputStream(filePath);
}
public java.io.OutputStream getOutputStream() throws IOException {
return super.getOutputStream();
}
}
}
}
|
