108 lines
4.0 KiB
Java
108 lines
4.0 KiB
Java
package com.janetfilter.core.plugin;
|
|
|
|
import com.janetfilter.core.Dispatcher;
|
|
import com.janetfilter.core.Environment;
|
|
import com.janetfilter.core.commons.ConfigParser;
|
|
import com.janetfilter.core.commons.DebugInfo;
|
|
import com.janetfilter.core.utils.StringUtils;
|
|
|
|
import java.io.File;
|
|
import java.lang.instrument.Instrumentation;
|
|
import java.util.Arrays;
|
|
import java.util.concurrent.ExecutorService;
|
|
import java.util.concurrent.Executors;
|
|
import java.util.concurrent.TimeUnit;
|
|
import java.util.jar.JarFile;
|
|
import java.util.jar.Manifest;
|
|
|
|
public final class PluginManager {
|
|
private static final String ENTRY_NAME = "JANF-Plugin-Entry";
|
|
|
|
private final Instrumentation inst;
|
|
private final Dispatcher dispatcher;
|
|
private final Environment environment;
|
|
|
|
public PluginManager(Dispatcher dispatcher, Environment environment) {
|
|
this.inst = environment.getInstrumentation();
|
|
this.dispatcher = dispatcher;
|
|
this.environment = environment;
|
|
}
|
|
|
|
public void loadPlugins() {
|
|
long startTime = System.currentTimeMillis();
|
|
|
|
File pluginsDirectory = environment.getPluginsDir();
|
|
if (!pluginsDirectory.exists() || !pluginsDirectory.isDirectory()) {
|
|
return;
|
|
}
|
|
|
|
File[] pluginFiles = pluginsDirectory.listFiles((d, n) -> n.endsWith(".jar"));
|
|
if (null == pluginFiles) {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
ExecutorService executorService = Executors.newCachedThreadPool();
|
|
for (File pluginFile : pluginFiles) {
|
|
executorService.submit(new PluginLoadTask(pluginFile));
|
|
}
|
|
|
|
executorService.shutdown();
|
|
if (!executorService.awaitTermination(30L, TimeUnit.SECONDS)) {
|
|
throw new RuntimeException("Load plugin timeout");
|
|
}
|
|
|
|
DebugInfo.debug(String.format("============ All plugins loaded, %.2fs elapsed ============", (System.currentTimeMillis() - startTime) / 1000D));
|
|
} catch (Throwable e) {
|
|
DebugInfo.error("Load plugin failed", e);
|
|
}
|
|
}
|
|
|
|
private class PluginLoadTask implements Runnable {
|
|
private final File pluginFile;
|
|
|
|
public PluginLoadTask(File pluginFile) {
|
|
this.pluginFile = pluginFile;
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
try {
|
|
if (pluginFile.getName().endsWith(environment.getDisabledPluginSuffix())) {
|
|
DebugInfo.debug("Disabled plugin: " + pluginFile + ", ignored.");
|
|
return;
|
|
}
|
|
|
|
JarFile jarFile = new JarFile(pluginFile);
|
|
Manifest manifest = jarFile.getManifest();
|
|
String entryClass = manifest.getMainAttributes().getValue(ENTRY_NAME);
|
|
if (StringUtils.isEmpty(entryClass)) {
|
|
return;
|
|
}
|
|
|
|
PluginClassLoader classLoader = new PluginClassLoader(jarFile);
|
|
Class<?> klass = Class.forName(entryClass, false, classLoader);
|
|
if (!Arrays.asList(klass.getInterfaces()).contains(PluginEntry.class)) {
|
|
return;
|
|
}
|
|
|
|
synchronized (inst) {
|
|
inst.appendToBootstrapClassLoaderSearch(jarFile);
|
|
}
|
|
|
|
PluginEntry pluginEntry = (PluginEntry) Class.forName(entryClass).newInstance();
|
|
|
|
File configFile = new File(environment.getConfigDir(), pluginEntry.getName().toLowerCase() + ".conf");
|
|
PluginConfig pluginConfig = new PluginConfig(configFile, ConfigParser.parse(configFile));
|
|
pluginEntry.init(environment, pluginConfig);
|
|
|
|
dispatcher.addTransformers(pluginEntry.getTransformers());
|
|
|
|
DebugInfo.debug("Plugin loaded: {name=" + pluginEntry.getName() + ", version=" + pluginEntry.getVersion() + ", author=" + pluginEntry.getAuthor() + "}");
|
|
} catch (Throwable e) {
|
|
DebugInfo.error("Parse plugin info failed", e);
|
|
}
|
|
}
|
|
}
|
|
}
|