MigrateRepos/ja-netfilter
1
0
Fork 0
mirror of https://gitee.com/ja-netfilter/ja-netfilter.git synced 2024-11-16 23:49:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix the f**k space escape

Browse Source 
Signed-off-by: pengzhile <pengzhile@gmail.com>
This commit is contained in:
pengzhile 2021-11-30 11:06:20 +08:00
parent af66da6219
commit 11bd215ce4
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/main/java/io/zhile/research/ja/netfilter/Launcher.java
View File

@ -8,7 +8,7 @@ import io.zhile.research.ja.netfilter.models.FilterConfig;
import java.io.File;
import java.lang.instrument.Instrumentation;
import java.lang.instrument.UnmodifiableClassException;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URL;
import java.util.jar.JarFile;
@ -20,18 +20,22 @@ public class Launcher {
public static void premain(String args, Instrumentation inst) {
printUsage();
URL jarURL = getJarURL();
if (null == jarURL) {
throw new RuntimeException("Can not locate ja-netfilter jar file.");
URI jarURI;
try {
jarURI = getJarURI();
} catch (Exception e) {
DebugInfo.output("ERROR: Can not locate ja-netfilter jar file.");
return;
}
try {
inst.appendToBootstrapClassLoaderSearch(new JarFile(jarURL.getPath()));
inst.appendToBootstrapClassLoaderSearch(new JarFile(jarURI.getPath()));
} catch (Throwable e) {
throw new RuntimeException("Can not access ja-netfilter jar file.", e);
DebugInfo.output("ERROR: Can not access ja-netfilter jar file.");
return;
}
File configFile = ConfigDetector.detect(new File(jarURL.getPath()).getParentFile().getPath(), args);
File configFile = ConfigDetector.detect(new File(jarURI.getPath()).getParentFile().getPath(), args);
if (null == configFile) {
DebugInfo.output("Could not find any configuration files.");
} else {
@ -68,29 +72,25 @@ public class Launcher {
System.out.flush();
}
private static URL getJarURL() {
private static URI getJarURI() throws Exception {
URL url = Launcher.class.getProtectionDomain().getCodeSource().getLocation();
if (null != url) {
return url;
return url.toURI();
}
String resourcePath = "/442fcf28466515a81d5434931496ffa64611cc8e.txt";
url = Launcher.class.getResource(resourcePath);
if (null == url) {
return null;
throw new Exception("Can not locate resource file.");
}
String path = url.getPath();
if (!path.endsWith("!" + resourcePath)) {
return null;
throw new Exception("Invalid resource path.");
}
path = path.substring(0, path.length() - resourcePath.length() - 1);
try {
return new URL(path);
} catch (MalformedURLException e) {
return null;
}
return new URI(path);
}
}