Mirror von
https://github.com/ViaVersion/ViaVersion.git
synchronisiert 2024-09-17 01:23:43 +02:00
Part 1/2 generate PS Packet Listener
Dieser Commit ist enthalten in:
Ursprung
527b5e6f99
Commit
76296d8d41
@ -3,8 +3,18 @@ package us.myles.ViaVersion.bukkit.classgenerator;
|
|||||||
import javassist.*;
|
import javassist.*;
|
||||||
import javassist.expr.ConstructorCall;
|
import javassist.expr.ConstructorCall;
|
||||||
import javassist.expr.ExprEditor;
|
import javassist.expr.ExprEditor;
|
||||||
|
|
||||||
|
import java.lang.reflect.Method;
|
||||||
|
import java.util.Arrays;
|
||||||
|
|
||||||
import org.bukkit.Bukkit;
|
import org.bukkit.Bukkit;
|
||||||
|
import org.bukkit.event.Event;
|
||||||
|
import org.bukkit.event.EventException;
|
||||||
|
import org.bukkit.event.EventPriority;
|
||||||
|
import org.bukkit.event.Listener;
|
||||||
|
import org.bukkit.plugin.EventExecutor;
|
||||||
import org.bukkit.plugin.Plugin;
|
import org.bukkit.plugin.Plugin;
|
||||||
|
import us.myles.ViaVersion.ViaVersionPlugin;
|
||||||
import us.myles.ViaVersion.api.ViaVersion;
|
import us.myles.ViaVersion.api.ViaVersion;
|
||||||
import us.myles.ViaVersion.bukkit.handlers.BukkitDecodeHandler;
|
import us.myles.ViaVersion.bukkit.handlers.BukkitDecodeHandler;
|
||||||
import us.myles.ViaVersion.bukkit.handlers.BukkitEncodeHandler;
|
import us.myles.ViaVersion.bukkit.handlers.BukkitEncodeHandler;
|
||||||
@ -13,6 +23,7 @@ import us.myles.ViaVersion.bukkit.util.NMSUtil;
|
|||||||
public class ClassGenerator {
|
public class ClassGenerator {
|
||||||
private static HandlerConstructor constructor = new BasicHandlerConstructor();
|
private static HandlerConstructor constructor = new BasicHandlerConstructor();
|
||||||
private static String psPackage = null;
|
private static String psPackage = null;
|
||||||
|
private static Class psConnectListener;
|
||||||
|
|
||||||
public static HandlerConstructor getConstructor() {
|
public static HandlerConstructor getConstructor() {
|
||||||
return constructor;
|
return constructor;
|
||||||
@ -27,16 +38,20 @@ public class ClassGenerator {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if (ViaVersion.getInstance().isCompatSpigotBuild()) {
|
if (ViaVersion.getInstance().isCompatSpigotBuild()) {
|
||||||
|
Bukkit.broadcastMessage("COMPAT SPIGOT");
|
||||||
Class decodeSuper = NMSUtil.nms("PacketDecoder");
|
Class decodeSuper = NMSUtil.nms("PacketDecoder");
|
||||||
Class encodeSuper = NMSUtil.nms("PacketEncoder");
|
Class encodeSuper = NMSUtil.nms("PacketEncoder");
|
||||||
// Generate the classes
|
// Generate the classes
|
||||||
addSpigotCompatibility(pool, BukkitDecodeHandler.class, decodeSuper);
|
addSpigotCompatibility(pool, BukkitDecodeHandler.class, decodeSuper);
|
||||||
addSpigotCompatibility(pool, BukkitEncodeHandler.class, encodeSuper);
|
addSpigotCompatibility(pool, BukkitEncodeHandler.class, encodeSuper);
|
||||||
} else {
|
} else {
|
||||||
|
Bukkit.broadcastMessage("COMPAT PROTOCOLSUPPORT");
|
||||||
// ProtocolSupport compatibility
|
// ProtocolSupport compatibility
|
||||||
Class encodeSuper;
|
Class encodeSuper;
|
||||||
Class decodeSuper;
|
Class decodeSuper;
|
||||||
if (isMultiplatformPS()) {
|
if (isMultiplatformPS()) {
|
||||||
|
Bukkit.broadcastMessage("NEW PROTOCOLSUPPORT!");
|
||||||
|
psConnectListener = makePSConnectListener(pool);
|
||||||
return;
|
return;
|
||||||
} else {
|
} else {
|
||||||
String psPackage = getOldPSPackage();
|
String psPackage = getOldPSPackage();
|
||||||
@ -166,6 +181,82 @@ public class ClassGenerator {
|
|||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private static Class makePSConnectListener(ClassPool pool) {
|
||||||
|
try {
|
||||||
|
// Reference classes
|
||||||
|
CtClass toExtend = pool.get("protocolsupport.api.Connection$PacketListener");
|
||||||
|
CtClass connectListenerClazz = pool.makeClass("us.myles.ViaVersion.classgenerator.generated.ProtocolSupportConnectListener");
|
||||||
|
connectListenerClazz.setSuperclass(toExtend);
|
||||||
|
// Import packages
|
||||||
|
pool.importPackage("java.util.Arrays");
|
||||||
|
pool.importPackage("org.bukkit.Bukkit");
|
||||||
|
pool.importPackage("us.myles.ViaVersion.api.protocol.ProtocolRegistry");
|
||||||
|
pool.importPackage("protocolsupport.api.ProtocolVersion");
|
||||||
|
pool.importPackage("protocolsupport.api.ProtocolType");
|
||||||
|
pool.importPackage("protocolsupport.api.Connection");
|
||||||
|
pool.importPackage("protocolsupport.api.Connection.PacketListener");
|
||||||
|
pool.importPackage("protocolsupport.api.Connection.PacketListener.PacketEvent");
|
||||||
|
pool.importPackage("protocolsupport.protocol.ConnectionImpl");
|
||||||
|
pool.importPackage(NMSUtil.nms("PacketHandshakingInSetProtocol").getName());
|
||||||
|
// Add connection reference field
|
||||||
|
connectListenerClazz.addField(CtField.make("private ConnectionImpl connection;", connectListenerClazz));
|
||||||
|
// Bake constructor
|
||||||
|
connectListenerClazz.addConstructor(CtNewConstructor.make(
|
||||||
|
"public ProtocolSupportConnectListener (ConnectionImpl connection) {\n"
|
||||||
|
+ " this.connection = connection;\n"
|
||||||
|
+ "}", connectListenerClazz));
|
||||||
|
// Add the listening method
|
||||||
|
connectListenerClazz.addMethod(CtNewMethod.make(
|
||||||
|
// On packet receive
|
||||||
|
"public void onPacketReceiving(protocolsupport.api.Connection.PacketListener.PacketEvent event) {\n"
|
||||||
|
// Check if we are getting handshake packet.
|
||||||
|
+ " if (event.getPacket() instanceof PacketHandshakingInSetProtocol) {\n"
|
||||||
|
// Get protocol version.
|
||||||
|
+ " int protoVersion = ((PacketHandshakingInSetProtocol) event.getPacket()).getProtocolVersion();\n"
|
||||||
|
// ViaVersion has at this point already spoofed the connectionversion. (Since it is higher up the pipeline)
|
||||||
|
// If via has put the protoVersion to the server we can spoof ProtocolSupport's version.
|
||||||
|
+ " if (protoVersion == us.myles.ViaVersion.api.protocol.ProtocolRegistry.SERVER_PROTOCOL) {\n"
|
||||||
|
+ " connection.setVersion(ProtocolVersion.getLatest(ProtocolType.PC));\n"
|
||||||
|
+ " }\n"
|
||||||
|
+ " }\n"
|
||||||
|
// Id version is not serverversion viaversion will not spoof. ProtocolSupport will handle the rest.
|
||||||
|
// In any case, remove the packet listener and wrap up.
|
||||||
|
+ " connection.removePacketListener(this);\n"
|
||||||
|
+ "}", connectListenerClazz));
|
||||||
|
return connectListenerClazz.toClass(HandlerConstructor.class.getClassLoader());
|
||||||
|
} catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void registerPSConnectListener(ViaVersionPlugin plugin) {
|
||||||
|
if (getPSConnectListener() != null) {
|
||||||
|
try {
|
||||||
|
Class<? extends Event> connectionOpenEvent = (Class<? extends Event>) Class.forName("protocolsupport.api.events.ConnectionOpenEvent");
|
||||||
|
Bukkit.getPluginManager().registerEvent(connectionOpenEvent, new Listener() { }, EventPriority.HIGH, new EventExecutor() {
|
||||||
|
@Override
|
||||||
|
public void execute(Listener listener, Event event) throws EventException {
|
||||||
|
try {
|
||||||
|
Object connection = event.getClass().getMethod("getConnection").invoke(event);
|
||||||
|
Object connectListener = getPSConnectListener().getConstructor(connection.getClass()).newInstance(connection);
|
||||||
|
Method addConnectListener = connection.getClass().getMethod("addPacketListener", Class.forName("protocolsupport.api.Connection$PacketListener"));
|
||||||
|
addConnectListener.invoke(connection, connectListener);
|
||||||
|
} catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}, plugin);
|
||||||
|
} catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static Class getPSConnectListener() {
|
||||||
|
return psConnectListener;
|
||||||
|
}
|
||||||
|
|
||||||
public static String getOldPSPackage() {
|
public static String getOldPSPackage() {
|
||||||
if (psPackage == null) {
|
if (psPackage == null) {
|
||||||
try {
|
try {
|
||||||
|
Laden…
In neuem Issue referenzieren
Einen Benutzer sperren