diff options
Diffstat (limited to 'src/main/java/makamys/neodymium/util/virtualjar')
4 files changed, 184 insertions, 0 deletions
diff --git a/src/main/java/makamys/neodymium/util/virtualjar/IVirtualJar.java b/src/main/java/makamys/neodymium/util/virtualjar/IVirtualJar.java new file mode 100644 index 0000000..11cef7e --- /dev/null +++ b/src/main/java/makamys/neodymium/util/virtualjar/IVirtualJar.java @@ -0,0 +1,11 @@ +package makamys.neodymium.util.virtualjar; + +import java.io.InputStream; + +public interface IVirtualJar { + + public String getName(); + + public InputStream getInputStream(String path); + +} diff --git a/src/main/java/makamys/neodymium/util/virtualjar/URLStreamHandlerHelper.java b/src/main/java/makamys/neodymium/util/virtualjar/URLStreamHandlerHelper.java new file mode 100644 index 0000000..7a8bef6 --- /dev/null +++ b/src/main/java/makamys/neodymium/util/virtualjar/URLStreamHandlerHelper.java @@ -0,0 +1,49 @@ +// From jimfs, adapted into a helper class +/* + * Copyright 2015 Google Inc. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package makamys.neodymium.util.virtualjar; + +import static com.google.common.base.Preconditions.checkArgument; + +import java.net.URLStreamHandler; + +public class URLStreamHandlerHelper { + + private static final String JAVA_PROTOCOL_HANDLER_PACKAGES = "java.protocol.handler.pkgs"; + + /** + * Generic method that would allow registration of any properly placed {@code Handler} class. + */ + public static void register(Class<? extends URLStreamHandler> handlerClass) { + checkArgument("Handler".equals(handlerClass.getSimpleName())); + + String pkg = handlerClass.getPackage().getName(); + int lastDot = pkg.lastIndexOf('.'); + checkArgument(lastDot > 0, "package for Handler (%s) must have a parent package", pkg); + + String parentPackage = pkg.substring(0, lastDot); + + String packages = System.getProperty(JAVA_PROTOCOL_HANDLER_PACKAGES); + if (packages == null) { + packages = parentPackage; + } else { + packages += "|" + parentPackage; + } + System.setProperty(JAVA_PROTOCOL_HANDLER_PACKAGES, packages); + } + +} diff --git a/src/main/java/makamys/neodymium/util/virtualjar/VirtualJar.java b/src/main/java/makamys/neodymium/util/virtualjar/VirtualJar.java new file mode 100644 index 0000000..8350d6f --- /dev/null +++ b/src/main/java/makamys/neodymium/util/virtualjar/VirtualJar.java @@ -0,0 +1,100 @@ +package makamys.neodymium.util.virtualjar; + +import static makamys.neodymium.Constants.MODID; +import static makamys.neodymium.Constants.PROTOCOL; + +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) { + return new URLConnection(url) { + public void connect() { + + } + + public Object getContent() throws IOException { + return super.getContent(); + } + + public String getHeaderField(String name) { + return super.getHeaderField(name); + } + + public InputStream getInputStream() { + String path = getURL().getPath(); + String nameSuffix = ".jar!"; + int nameEnd = path.indexOf(nameSuffix); + String name = path.substring(0, nameEnd); + + IVirtualJar jar = jars.get(name); + + return jar.getInputStream(path.substring(nameEnd + nameSuffix.length())); + } + + public java.io.OutputStream getOutputStream() throws IOException { + return super.getOutputStream(); + } + }; + } + } + +} diff --git a/src/main/java/makamys/neodymium/util/virtualjar/protocol/neodymiumvirtualjar/Handler.java b/src/main/java/makamys/neodymium/util/virtualjar/protocol/neodymiumvirtualjar/Handler.java new file mode 100644 index 0000000..9f0fe66 --- /dev/null +++ b/src/main/java/makamys/neodymium/util/virtualjar/protocol/neodymiumvirtualjar/Handler.java @@ -0,0 +1,24 @@ +package makamys.neodymium.util.virtualjar.protocol.neodymiumvirtualjar; + +import static makamys.neodymium.Constants.MODID; +import static makamys.neodymium.Constants.PROTOCOL; + +import java.net.URL; +import java.net.URLConnection; +import java.net.URLStreamHandler; + +import net.minecraft.launchwrapper.Launch; + +public class Handler extends URLStreamHandler { + + private static final IURLStreamHandlerImpl impl = (IURLStreamHandlerImpl)Launch.blackboard.get(MODID + "." + PROTOCOL + ".impl"); + + @Override + protected URLConnection openConnection(URL url) { + return impl.openConnection(url); + } + + public interface IURLStreamHandlerImpl { + URLConnection openConnection(URL url); + } +} |