[TinyProtocol] Format TinyProtocol

This commit is contained in:
Dan Mulloy 2015-02-17 16:53:09 -05:00
parent 66be4900a3
commit 115b5e8daa
3 changed files with 357 additions and 289 deletions

View File

@ -12,11 +12,13 @@ import org.bukkit.plugin.java.JavaPlugin;
import com.comphenix.tinyprotocol.Reflection.ConstructorInvoker; import com.comphenix.tinyprotocol.Reflection.ConstructorInvoker;
import com.comphenix.tinyprotocol.Reflection.FieldAccessor; import com.comphenix.tinyprotocol.Reflection.FieldAccessor;
/**
* Represents an example plugin utilizing TinyProtocol
*/
public class ExamplePlugin extends JavaPlugin { public class ExamplePlugin extends JavaPlugin {
// Chat packets // Chat packets
private FieldAccessor<String> CHAT_MESSAGE = Reflection.getField( private FieldAccessor<String> CHAT_MESSAGE = Reflection.getField("{nms}.PacketPlayInChat", String.class, 0);
"{nms}.PacketPlayInChat", String.class, 0);
// Explosion packet // Explosion packet
private Class<?> particleClass = Reflection.getClass("{nms}.PacketPlayOutWorldParticles"); private Class<?> particleClass = Reflection.getClass("{nms}.PacketPlayOutWorldParticles");
private FieldAccessor<String> particleName = Reflection.getField(particleClass, String.class, 0); private FieldAccessor<String> particleName = Reflection.getField(particleClass, String.class, 0);
@ -24,7 +26,7 @@ public class ExamplePlugin extends JavaPlugin {
private FieldAccessor<Float> particleY = Reflection.getField(particleClass, float.class, 1); private FieldAccessor<Float> particleY = Reflection.getField(particleClass, float.class, 1);
private FieldAccessor<Float> particleZ = Reflection.getField(particleClass, float.class, 2); private FieldAccessor<Float> particleZ = Reflection.getField(particleClass, float.class, 2);
private FieldAccessor<Integer> particleCount = Reflection.getField(particleClass, int.class, 0); private FieldAccessor<Integer> particleCount = Reflection.getField(particleClass, int.class, 0);
// Server info packet // Server info packet
private Class<?> serverInfoClass = Reflection.getClass("{nms}.PacketStatusOutServerInfo"); private Class<?> serverInfoClass = Reflection.getClass("{nms}.PacketStatusOutServerInfo");
private Class<Object> serverPingClass = Reflection.getUntypedClass("{nms}.ServerPing"); private Class<Object> serverPingClass = Reflection.getUntypedClass("{nms}.ServerPing");
@ -32,12 +34,13 @@ public class ExamplePlugin extends JavaPlugin {
private FieldAccessor<Object> serverPing = Reflection.getField(serverInfoClass, serverPingClass, 0); private FieldAccessor<Object> serverPing = Reflection.getField(serverInfoClass, serverPingClass, 0);
private FieldAccessor<Object> playerSample = Reflection.getField(serverPingClass, playerSampleClass, 0); private FieldAccessor<Object> playerSample = Reflection.getField(serverPingClass, playerSampleClass, 0);
private ConstructorInvoker playerSampleInvoker = Reflection.getConstructor(playerSampleClass, int.class, int.class); private ConstructorInvoker playerSampleInvoker = Reflection.getConstructor(playerSampleClass, int.class, int.class);
private TinyProtocol protocol; private TinyProtocol protocol;
@Override @Override
public void onEnable() { public void onEnable() {
protocol = new TinyProtocol(this) { protocol = new TinyProtocol(this) {
@Override @Override
public Object onPacketInAsync(Player sender, Channel channel, Object packet) { public Object onPacketInAsync(Player sender, Channel channel, Object packet) {
// Cancel chat packets // Cancel chat packets
@ -47,24 +50,28 @@ public class ExamplePlugin extends JavaPlugin {
return null; return null;
} }
} }
if (particleName.hasField(packet)) { if (particleName.hasField(packet)) {
System.out.println("Sending particle field:" + packet); System.out.println("Sending particle field:" + packet);
} }
return super.onPacketInAsync(sender, channel, packet); return super.onPacketInAsync(sender, channel, packet);
} }
@Override @Override
public Object onPacketOutAsync(Player reciever, Channel channel, Object packet) { public Object onPacketOutAsync(Player reciever, Channel channel, Object packet) {
if (serverInfoClass.isInstance(packet)) { if (serverInfoClass.isInstance(packet)) {
Object ping = serverPing.get(packet); Object ping = serverPing.get(packet);
playerSample.set(ping, playerSampleInvoker.invoke(1000, 0)); playerSample.set(ping, playerSampleInvoker.invoke(1000, 0));
// Which is equivalent to: // Which is equivalent to:
//serverPing.get(packet).setPlayerSample(new ServerPingPlayerSample(1000, 0)); // serverPing.get(packet).setPlayerSample(new ServerPingPlayerSample(1000, 0));
return packet; return packet;
} }
return super.onPacketOutAsync(reciever, channel, packet); return super.onPacketOutAsync(reciever, channel, packet);
} }
}; };
} }
@ -72,7 +79,7 @@ public class ExamplePlugin extends JavaPlugin {
public boolean onCommand(CommandSender sender, Command command, String label, String[] args) { public boolean onCommand(CommandSender sender, Command command, String label, String[] args) {
if (sender instanceof Player) { if (sender instanceof Player) {
Player player = (Player) sender; Player player = (Player) sender;
// Toggle injection // Toggle injection
if (protocol.hasInjected(player)) { if (protocol.hasInjected(player)) {
protocol.uninjectPlayer(player); protocol.uninjectPlayer(player);
@ -81,13 +88,15 @@ public class ExamplePlugin extends JavaPlugin {
protocol.injectPlayer(player); protocol.injectPlayer(player);
sender.sendMessage(ChatColor.DARK_GREEN + "Player " + player + " has been injected."); sender.sendMessage(ChatColor.DARK_GREEN + "Player " + player + " has been injected.");
} }
return true; return true;
} else { } else {
sender.sendMessage(ChatColor.RED + "Can only be invoked by a player."); sender.sendMessage(ChatColor.RED + "Can only be invoked by a player.");
} }
return false; return false;
} }
private void sendExplosion(Player player) { private void sendExplosion(Player player) {
try { try {
// Only visible for the client // Only visible for the client
@ -101,9 +110,8 @@ public class ExamplePlugin extends JavaPlugin {
// Send the packet to the player // Send the packet to the player
protocol.sendPacket(player, explosionPacket); protocol.sendPacket(player, explosionPacket);
} catch (Exception e) { } catch (Exception e) {
throw new RuntimeException("Cannot send packet.", e); throw new RuntimeException("Cannot send packet.", e);
} }
} }
} }

View File

@ -11,75 +11,83 @@ import org.bukkit.Bukkit;
/** /**
* An utility class that simplifies reflection in Bukkit plugins. * An utility class that simplifies reflection in Bukkit plugins.
*
* @author Kristian * @author Kristian
*/ */
public final class Reflection { public final class Reflection {
/** /**
* An interface for invoking a specific constructor. * An interface for invoking a specific constructor.
*/ */
public interface ConstructorInvoker { public interface ConstructorInvoker {
/** /**
* Invoke a constructor for a specific class. * Invoke a constructor for a specific class.
*
* @param arguments - the arguments to pass to the constructor. * @param arguments - the arguments to pass to the constructor.
* @return The constructed object. * @return The constructed object.
*/ */
public Object invoke(Object... arguments); public Object invoke(Object... arguments);
} }
/** /**
* An interface for invoking a specific method. * An interface for invoking a specific method.
*/ */
public interface MethodInvoker { public interface MethodInvoker {
/** /**
* Invoke a method on a specific target object. * Invoke a method on a specific target object.
*
* @param target - the target object, or NULL for a static method. * @param target - the target object, or NULL for a static method.
* @param arguments - the arguments to pass to the method. * @param arguments - the arguments to pass to the method.
* @return The return value, or NULL if is void. * @return The return value, or NULL if is void.
*/ */
public Object invoke(Object target, Object... arguments); public Object invoke(Object target, Object... arguments);
} }
/** /**
* An interface for retrieving the field content. * An interface for retrieving the field content.
*
* @param <T> - field type. * @param <T> - field type.
*/ */
public interface FieldAccessor<T> { public interface FieldAccessor<T> {
/** /**
* Retrieve the content of a field. * Retrieve the content of a field.
*
* @param target - the target object, or NULL for a static field. * @param target - the target object, or NULL for a static field.
* @return The value of the field. * @return The value of the field.
*/ */
public T get(Object target); public T get(Object target);
/** /**
* Set the content of a field. * Set the content of a field.
*
* @param target - the target object, or NULL for a static field. * @param target - the target object, or NULL for a static field.
* @param value - the new value of the field. * @param value - the new value of the field.
*/ */
public void set(Object target, Object value); public void set(Object target, Object value);
/** /**
* Determine if the given object has this field. * Determine if the given object has this field.
*
* @param target - the object to test. * @param target - the object to test.
* @return TRUE if it does, FALSE otherwise. * @return TRUE if it does, FALSE otherwise.
*/ */
public boolean hasField(Object target); public boolean hasField(Object target);
} }
// Deduce the net.minecraft.server.v* package // Deduce the net.minecraft.server.v* package
private static String OBC_PREFIX = Bukkit.getServer().getClass().getPackage().getName(); private static String OBC_PREFIX = Bukkit.getServer().getClass().getPackage().getName();
private static String NMS_PREFIX = OBC_PREFIX.replace("org.bukkit.craftbukkit", "net.minecraft.server"); private static String NMS_PREFIX = OBC_PREFIX.replace("org.bukkit.craftbukkit", "net.minecraft.server");
private static String VERSION = OBC_PREFIX.replace("org.bukkit.craftbukkit", "").replace(".", ""); private static String VERSION = OBC_PREFIX.replace("org.bukkit.craftbukkit", "").replace(".", "");
// Variable replacement // Variable replacement
private static Pattern MATCH_VARIABLE = Pattern.compile("\\{([^\\}]+)\\}"); private static Pattern MATCH_VARIABLE = Pattern.compile("\\{([^\\}]+)\\}");
private Reflection() { private Reflection() {
// Seal class // Seal class
} }
/** /**
* Retrieve a field accessor for a specific field type and name. * Retrieve a field accessor for a specific field type and name.
*
* @param target - the target type. * @param target - the target type.
* @param name - the name of the field, or NULL to ignore. * @param name - the name of the field, or NULL to ignore.
* @param fieldType - a compatible field type. * @param fieldType - a compatible field type.
@ -88,9 +96,10 @@ public final class Reflection {
public static <T> FieldAccessor<T> getField(Class<?> target, String name, Class<T> fieldType) { public static <T> FieldAccessor<T> getField(Class<?> target, String name, Class<T> fieldType) {
return getField(target, name, fieldType, 0); return getField(target, name, fieldType, 0);
} }
/** /**
* Retrieve a field accessor for a specific field type and name. * Retrieve a field accessor for a specific field type and name.
*
* @param className - lookup name of the class, see {@link #getClass(String)}. * @param className - lookup name of the class, see {@link #getClass(String)}.
* @param name - the name of the field, or NULL to ignore. * @param name - the name of the field, or NULL to ignore.
* @param fieldType - a compatible field type. * @param fieldType - a compatible field type.
@ -99,9 +108,10 @@ public final class Reflection {
public static <T> FieldAccessor<T> getField(String className, String name, Class<T> fieldType) { public static <T> FieldAccessor<T> getField(String className, String name, Class<T> fieldType) {
return getField(getClass(className), name, fieldType, 0); return getField(getClass(className), name, fieldType, 0);
} }
/** /**
* Retrieve a field accessor for a specific field type and name. * Retrieve a field accessor for a specific field type and name.
*
* @param target - the target type. * @param target - the target type.
* @param fieldType - a compatible field type. * @param fieldType - a compatible field type.
* @param index - the number of compatible fields to skip. * @param index - the number of compatible fields to skip.
@ -110,9 +120,10 @@ public final class Reflection {
public static <T> FieldAccessor<T> getField(Class<?> target, Class<T> fieldType, int index) { public static <T> FieldAccessor<T> getField(Class<?> target, Class<T> fieldType, int index) {
return getField(target, null, fieldType, index); return getField(target, null, fieldType, index);
} }
/** /**
* Retrieve a field accessor for a specific field type and name. * Retrieve a field accessor for a specific field type and name.
*
* @param className - lookup name of the class, see {@link #getClass(String)}. * @param className - lookup name of the class, see {@link #getClass(String)}.
* @param fieldType - a compatible field type. * @param fieldType - a compatible field type.
* @param index - the number of compatible fields to skip. * @param index - the number of compatible fields to skip.
@ -121,18 +132,18 @@ public final class Reflection {
public static <T> FieldAccessor<T> getField(String className, Class<T> fieldType, int index) { public static <T> FieldAccessor<T> getField(String className, Class<T> fieldType, int index) {
return getField(getClass(className), fieldType, index); return getField(getClass(className), fieldType, index);
} }
// Common method // Common method
private static <T> FieldAccessor<T> getField(Class<?> target, String name, Class<T> fieldType, int index) { private static <T> FieldAccessor<T> getField(Class<?> target, String name, Class<T> fieldType, int index) {
for (final Field field : target.getDeclaredFields()) { for (final Field field : target.getDeclaredFields()) {
if ((name == null || field.getName().equals(name)) && if ((name == null || field.getName().equals(name)) && fieldType.isAssignableFrom(field.getType()) && index-- <= 0) {
fieldType.isAssignableFrom(field.getType()) && index-- <= 0) {
field.setAccessible(true); field.setAccessible(true);
// A function for retrieving a specific field value // A function for retrieving a specific field value
return new FieldAccessor<T>() { return new FieldAccessor<T>() {
@SuppressWarnings("unchecked")
@Override @Override
@SuppressWarnings("unchecked")
public T get(Object target) { public T get(Object target) {
try { try {
return (T) field.get(target); return (T) field.get(target);
@ -140,7 +151,7 @@ public final class Reflection {
throw new RuntimeException("Cannot access reflection.", e); throw new RuntimeException("Cannot access reflection.", e);
} }
} }
@Override @Override
public void set(Object target, Object value) { public void set(Object target, Object value) {
try { try {
@ -149,7 +160,7 @@ public final class Reflection {
throw new RuntimeException("Cannot access reflection.", e); throw new RuntimeException("Cannot access reflection.", e);
} }
} }
@Override @Override
public boolean hasField(Object target) { public boolean hasField(Object target) {
// target instanceof DeclaringClass // target instanceof DeclaringClass
@ -158,218 +169,233 @@ public final class Reflection {
}; };
} }
} }
// Search in parent classes // Search in parent classes
if (target.getSuperclass() != null) if (target.getSuperclass() != null)
return getField(target.getSuperclass(), name, fieldType, index); return getField(target.getSuperclass(), name, fieldType, index);
throw new IllegalArgumentException("Cannot find field with type " + fieldType); throw new IllegalArgumentException("Cannot find field with type " + fieldType);
} }
/** /**
* Search for the first publically and privately defined method of the given name and parameter count. * Search for the first publically and privately defined method of the given name and parameter count.
* @param className - lookup name of the class, see {@link #getClass(String)}. *
* @param methodName - the method name, or NULL to skip. * @param className - lookup name of the class, see {@link #getClass(String)}.
* @param params - the expected parameters. * @param methodName - the method name, or NULL to skip.
* @return An object that invokes this specific method. * @param params - the expected parameters.
* @throws IllegalStateException If we cannot find this method. * @return An object that invokes this specific method.
*/ * @throws IllegalStateException If we cannot find this method.
*/
public static MethodInvoker getMethod(String className, String methodName, Class<?>... params) { public static MethodInvoker getMethod(String className, String methodName, Class<?>... params) {
return getTypedMethod(getClass(className), methodName, null, params); return getTypedMethod(getClass(className), methodName, null, params);
} }
/**
* Search for the first publically and privately defined method of the given name and parameter count.
* @param clazz - a class to start with.
* @param methodName - the method name, or NULL to skip.
* @param params - the expected parameters.
* @return An object that invokes this specific method.
* @throws IllegalStateException If we cannot find this method.
*/
public static MethodInvoker getMethod(Class<?> clazz, String methodName, Class<?>... params) {
return getTypedMethod(clazz, methodName, null, params);
}
/**
* Search for the first publically and privately defined method of the given name and parameter count.
* @param clazz - a class to start with.
* @param methodName - the method name, or NULL to skip.
* @param returnType - the expected return type, or NULL to ignore.
* @param params - the expected parameters.
* @return An object that invokes this specific method.
* @throws IllegalStateException If we cannot find this method.
*/
public static MethodInvoker getTypedMethod(Class<?> clazz, String methodName, Class<?> returnType, Class<?>... params) {
for (final Method method : clazz.getDeclaredMethods()) {
if ((methodName == null || method.getName().equals(methodName)) &&
(returnType == null) || method.getReturnType().equals(returnType) &&
Arrays.equals(method.getParameterTypes(), params)) {
method.setAccessible(true); /**
return new MethodInvoker() { * Search for the first publically and privately defined method of the given name and parameter count.
@Override *
public Object invoke(Object target, Object... arguments) { * @param clazz - a class to start with.
try { * @param methodName - the method name, or NULL to skip.
* @param params - the expected parameters.
* @return An object that invokes this specific method.
* @throws IllegalStateException If we cannot find this method.
*/
public static MethodInvoker getMethod(Class<?> clazz, String methodName, Class<?>... params) {
return getTypedMethod(clazz, methodName, null, params);
}
/**
* Search for the first publically and privately defined method of the given name and parameter count.
*
* @param clazz - a class to start with.
* @param methodName - the method name, or NULL to skip.
* @param returnType - the expected return type, or NULL to ignore.
* @param params - the expected parameters.
* @return An object that invokes this specific method.
* @throws IllegalStateException If we cannot find this method.
*/
public static MethodInvoker getTypedMethod(Class<?> clazz, String methodName, Class<?> returnType, Class<?>... params) {
for (final Method method : clazz.getDeclaredMethods()) {
if ((methodName == null || method.getName().equals(methodName)) && (returnType == null) || method.getReturnType().equals(returnType) && Arrays.equals(method.getParameterTypes(), params)) {
method.setAccessible(true);
return new MethodInvoker() {
@Override
public Object invoke(Object target, Object... arguments) {
try {
return method.invoke(target, arguments); return method.invoke(target, arguments);
} catch (Exception e) { } catch (Exception e) {
throw new RuntimeException("Cannot invoke method " + method, e); throw new RuntimeException("Cannot invoke method " + method, e);
} }
} }
};
}
}
// Search in every superclass
if (clazz.getSuperclass() != null)
return getMethod(clazz.getSuperclass(), methodName, params);
throw new IllegalStateException(String.format(
"Unable to find method %s (%s).", methodName, Arrays.asList(params)));
}
/**
* Search for the first publically and privately defined constructor of the given name and parameter count.
* @param className - lookup name of the class, see {@link #getClass(String)}.
* @param params - the expected parameters.
* @return An object that invokes this constructor.
* @throws IllegalStateException If we cannot find this method.
*/
public static ConstructorInvoker getConstructor(String className, Class<?>... params) {
return getConstructor(getClass(className), params);
}
/**
* Search for the first publically and privately defined constructor of the given name and parameter count.
* @param clazz - a class to start with.
* @param params - the expected parameters.
* @return An object that invokes this constructor.
* @throws IllegalStateException If we cannot find this method.
*/
public static ConstructorInvoker getConstructor(Class<?> clazz, Class<?>... params) {
for (final Constructor<?> constructor : clazz.getDeclaredConstructors()) {
if (Arrays.equals(constructor.getParameterTypes(), params)) {
constructor.setAccessible(true); };
return new ConstructorInvoker() { }
@Override }
public Object invoke(Object... arguments) {
try { // Search in every superclass
if (clazz.getSuperclass() != null)
return getMethod(clazz.getSuperclass(), methodName, params);
throw new IllegalStateException(String.format("Unable to find method %s (%s).", methodName, Arrays.asList(params)));
}
/**
* Search for the first publically and privately defined constructor of the given name and parameter count.
*
* @param className - lookup name of the class, see {@link #getClass(String)}.
* @param params - the expected parameters.
* @return An object that invokes this constructor.
* @throws IllegalStateException If we cannot find this method.
*/
public static ConstructorInvoker getConstructor(String className, Class<?>... params) {
return getConstructor(getClass(className), params);
}
/**
* Search for the first publically and privately defined constructor of the given name and parameter count.
*
* @param clazz - a class to start with.
* @param params - the expected parameters.
* @return An object that invokes this constructor.
* @throws IllegalStateException If we cannot find this method.
*/
public static ConstructorInvoker getConstructor(Class<?> clazz, Class<?>... params) {
for (final Constructor<?> constructor : clazz.getDeclaredConstructors()) {
if (Arrays.equals(constructor.getParameterTypes(), params)) {
constructor.setAccessible(true);
return new ConstructorInvoker() {
@Override
public Object invoke(Object... arguments) {
try {
return constructor.newInstance(arguments); return constructor.newInstance(arguments);
} catch (Exception e) { } catch (Exception e) {
throw new RuntimeException("Cannot invoke constructor " + constructor, e); throw new RuntimeException("Cannot invoke constructor " + constructor, e);
} }
} }
};
} };
} }
throw new IllegalStateException(String.format( }
"Unable to find constructor for %s (%s).", clazz, Arrays.asList(params)));
} throw new IllegalStateException(String.format("Unable to find constructor for %s (%s).", clazz, Arrays.asList(params)));
}
/**
* Retrieve a class from its full name, without knowing its type on compile time. /**
* <p> * Retrieve a class from its full name, without knowing its type on compile time.
* This is useful when looking up fields by a NMS or OBC type. * <p>
* <p> * This is useful when looking up fields by a NMS or OBC type.
* @see {@link #getClass()} for more information. * <p>
* @param lookupName - the class name with variables. *
* @return The class. * @see {@link #getClass()} for more information.
*/ * @param lookupName - the class name with variables.
public static Class<Object> getUntypedClass(String lookupName) { * @return The class.
@SuppressWarnings({"rawtypes", "unchecked"}) */
Class<Object> clazz = (Class<Object>)(Class)getClass(lookupName); public static Class<Object> getUntypedClass(String lookupName) {
return clazz; @SuppressWarnings({ "rawtypes", "unchecked" })
} Class<Object> clazz = (Class) getClass(lookupName);
return clazz;
/** }
* Retrieve a class from its full name.
* <p> /**
* Strings enclosed with curly brackets - such as {TEXT} - will be replaced according * Retrieve a class from its full name.
* to the following table: * <p>
* <p> * Strings enclosed with curly brackets - such as {TEXT} - will be replaced according to the following table:
* <p>
* <table border="1"> * <table border="1">
* <tr> * <tr>
* <th>Variable</th> * <th>Variable</th>
* <th>Content</th> * <th>Content</th>
* </tr> * </tr>
* <tr> * <tr>
* <td>{nms}</td> * <td>{nms}</td>
* <td>Actual package name of net.minecraft.server.VERSION</td> * <td>Actual package name of net.minecraft.server.VERSION</td>
* </tr> * </tr>
* <tr> * <tr>
* <td>{obc}</td> * <td>{obc}</td>
* <td>Actual pacakge name of org.bukkit.craftbukkit.VERSION</td> * <td>Actual pacakge name of org.bukkit.craftbukkit.VERSION</td>
* </tr> * </tr>
* <tr> * <tr>
* <td>{version}</td> * <td>{version}</td>
* <td>The current Minecraft package VERSION, if any.</td> * <td>The current Minecraft package VERSION, if any.</td>
* </tr> * </tr>
* </table> * </table>
* @param lookupName - the class name with variables. *
* @return The looked up class. * @param lookupName - the class name with variables.
* @throws IllegalArgumentException If a variable or class could not be found. * @return The looked up class.
*/ * @throws IllegalArgumentException If a variable or class could not be found.
public static Class<?> getClass(String lookupName) { */
return getCanonicalClass(expandVariables(lookupName)); public static Class<?> getClass(String lookupName) {
} return getCanonicalClass(expandVariables(lookupName));
}
/** /**
* Retrieve a class in the net.minecraft.server.VERSION.* package. * Retrieve a class in the net.minecraft.server.VERSION.* package.
*
* @param name - the name of the class, excluding the package. * @param name - the name of the class, excluding the package.
* @throws IllegalArgumentException If the class doesn't exist. * @throws IllegalArgumentException If the class doesn't exist.
*/ */
public static Class<?> getMinecraftClass(String name) { public static Class<?> getMinecraftClass(String name) {
return getCanonicalClass(NMS_PREFIX + "." + name); return getCanonicalClass(NMS_PREFIX + "." + name);
} }
/** /**
* Retrieve a class in the org.bukkit.craftbukkit.VERSION.* package. * Retrieve a class in the org.bukkit.craftbukkit.VERSION.* package.
*
* @param name - the name of the class, excluding the package. * @param name - the name of the class, excluding the package.
* @throws IllegalArgumentException If the class doesn't exist. * @throws IllegalArgumentException If the class doesn't exist.
*/ */
public static Class<?> getCraftBukkitClass(String name) { public static Class<?> getCraftBukkitClass(String name) {
return getCanonicalClass(OBC_PREFIX + "." + name); return getCanonicalClass(OBC_PREFIX + "." + name);
} }
/** /**
* Retrieve a class by its canonical name. * Retrieve a class by its canonical name.
*
* @param canonicalName - the canonical name. * @param canonicalName - the canonical name.
* @return The class. * @return The class.
*/ */
private static Class<?> getCanonicalClass(String canonicalName) { private static Class<?> getCanonicalClass(String canonicalName) {
try { try {
return Class.forName(canonicalName); return Class.forName(canonicalName);
} catch (ClassNotFoundException e) { } catch (ClassNotFoundException e) {
throw new IllegalArgumentException("Cannot find " + canonicalName, e); throw new IllegalArgumentException("Cannot find " + canonicalName, e);
} }
} }
/** /**
* Expand variables such as "{nms}" and "{obc}" to their corresponding packages. * Expand variables such as "{nms}" and "{obc}" to their corresponding packages.
*
* @param name - the full name of the class. * @param name - the full name of the class.
* @return The expanded string. * @return The expanded string.
*/ */
private static String expandVariables(String name) { private static String expandVariables(String name) {
StringBuffer output = new StringBuffer(); StringBuffer output = new StringBuffer();
Matcher matcher = MATCH_VARIABLE.matcher(name); Matcher matcher = MATCH_VARIABLE.matcher(name);
while (matcher.find()) { while (matcher.find()) {
String variable = matcher.group(1); String variable = matcher.group(1);
String replacement = ""; String replacement = "";
// Expand all detected variables // Expand all detected variables
if ("nms".equalsIgnoreCase(variable)) if ("nms".equalsIgnoreCase(variable))
replacement = NMS_PREFIX; replacement = NMS_PREFIX;
else if ("obc".equalsIgnoreCase(variable)) else if ("obc".equalsIgnoreCase(variable))
replacement = OBC_PREFIX; replacement = OBC_PREFIX;
else if ("version".equalsIgnoreCase(variable)) else if ("version".equalsIgnoreCase(variable))
replacement = VERSION; replacement = VERSION;
else else
throw new IllegalArgumentException("Unknown variable: " + variable); throw new IllegalArgumentException("Unknown variable: " + variable);
// Assume the expanded variables are all packages, and append a dot // Assume the expanded variables are all packages, and append a dot
if (replacement.length() > 0 && matcher.end() < name.length() && name.charAt(matcher.end()) != '.') if (replacement.length() > 0 && matcher.end() < name.length() && name.charAt(matcher.end()) != '.')
replacement += "."; replacement += ".";
matcher.appendReplacement(output, Matcher.quoteReplacement(replacement)); matcher.appendReplacement(output, Matcher.quoteReplacement(replacement));
} }
matcher.appendTail(output);
return output.toString(); matcher.appendTail(output);
return output.toString();
} }
} }

View File

@ -34,65 +34,67 @@ import com.google.common.collect.MapMaker;
import com.mojang.authlib.GameProfile; import com.mojang.authlib.GameProfile;
/** /**
* Represents a very tiny alternative to ProtocolLib in 1.7.2. * Represents a very tiny alternative to ProtocolLib.
* <p> * <p>
* It now supports intercepting packets during login and status ping (such as OUT_SERVER_PING)! * It now supports intercepting packets during login and status ping (such as OUT_SERVER_PING)!
*
* @author Kristian * @author Kristian
*/ */
public abstract class TinyProtocol { public abstract class TinyProtocol {
private static final AtomicInteger ID = new AtomicInteger(0); private static final AtomicInteger ID = new AtomicInteger(0);
// Used in order to lookup a channel // Used in order to lookup a channel
private static final MethodInvoker getPlayerHandle = Reflection.getMethod("{obc}.entity.CraftPlayer", "getHandle"); private static final MethodInvoker getPlayerHandle = Reflection.getMethod("{obc}.entity.CraftPlayer", "getHandle");
private static final FieldAccessor<Object> getConnection = Reflection.getField("{nms}.EntityPlayer", "playerConnection", Object.class); private static final FieldAccessor<Object> getConnection = Reflection.getField("{nms}.EntityPlayer", "playerConnection", Object.class);
private static final FieldAccessor<Object> getManager = Reflection.getField("{nms}.PlayerConnection", "networkManager", Object.class); private static final FieldAccessor<Object> getManager = Reflection.getField("{nms}.PlayerConnection", "networkManager", Object.class);
private static final FieldAccessor<Channel> getChannel = Reflection.getField("{nms}.NetworkManager", Channel.class, 0); private static final FieldAccessor<Channel> getChannel = Reflection.getField("{nms}.NetworkManager", Channel.class, 0);
// Looking up ServerConnection // Looking up ServerConnection
private static final Class<Object> minecraftServerClass = Reflection.getUntypedClass("{nms}.MinecraftServer"); private static final Class<Object> minecraftServerClass = Reflection.getUntypedClass("{nms}.MinecraftServer");
private static final Class<Object> serverConnectionClass = Reflection.getUntypedClass("{nms}.ServerConnection"); private static final Class<Object> serverConnectionClass = Reflection.getUntypedClass("{nms}.ServerConnection");
private static final FieldAccessor<Object> getMinecraftServer = Reflection.getField("{obc}.CraftServer", minecraftServerClass, 0); private static final FieldAccessor<Object> getMinecraftServer = Reflection.getField("{obc}.CraftServer", minecraftServerClass, 0);
private static final FieldAccessor<Object> getServerConnection = Reflection.getField(minecraftServerClass, serverConnectionClass, 0); private static final FieldAccessor<Object> getServerConnection = Reflection.getField(minecraftServerClass, serverConnectionClass, 0);
private static final MethodInvoker getNetworkMarkers = Reflection.getTypedMethod(serverConnectionClass, null, List.class, serverConnectionClass); private static final MethodInvoker getNetworkMarkers = Reflection.getTypedMethod(serverConnectionClass, null, List.class, serverConnectionClass);
// Packets we have to intercept // Packets we have to intercept
private static final Class<?> PACKET_LOGIN_IN_START = Reflection.getMinecraftClass("PacketLoginInStart"); private static final Class<?> PACKET_LOGIN_IN_START = Reflection.getMinecraftClass("PacketLoginInStart");
private static final FieldAccessor<GameProfile> getGameProfile = Reflection.getField(PACKET_LOGIN_IN_START, GameProfile.class, 0); private static final FieldAccessor<GameProfile> getGameProfile = Reflection.getField(PACKET_LOGIN_IN_START, GameProfile.class, 0);
// Speedup channel lookup // Speedup channel lookup
private Map<String, Channel> channelLookup = new MapMaker().weakValues().makeMap(); private Map<String, Channel> channelLookup = new MapMaker().weakValues().makeMap();
private Listener listener; private Listener listener;
// Channels that have already been removed // Channels that have already been removed
private Set<Channel> uninjectedChannels = Collections.newSetFromMap(new MapMaker().weakKeys().<Channel, Boolean>makeMap()); private Set<Channel> uninjectedChannels = Collections.newSetFromMap(new MapMaker().weakKeys().<Channel, Boolean>makeMap());
// List of network markers // List of network markers
private List<Object> networkManagers; private List<Object> networkManagers;
// Injected channel handlers // Injected channel handlers
private List<Channel> serverChannels = Lists.newArrayList(); private List<Channel> serverChannels = Lists.newArrayList();
private ChannelInboundHandlerAdapter serverChannelHandler; private ChannelInboundHandlerAdapter serverChannelHandler;
private ChannelInitializer<Channel> beginInitProtocol; private ChannelInitializer<Channel> beginInitProtocol;
private ChannelInitializer<Channel> endInitProtocol; private ChannelInitializer<Channel> endInitProtocol;
// Current handler name // Current handler name
private String handlerName; private String handlerName;
protected volatile boolean closed; protected volatile boolean closed;
protected Plugin plugin; protected Plugin plugin;
/** /**
* Construct a new instance of TinyProtocol, and start intercepting packets for all connected clients and future clients. * Construct a new instance of TinyProtocol, and start intercepting packets for all connected clients and future clients.
* <p> * <p>
* You can construct multiple instances per plugin. * You can construct multiple instances per plugin.
*
* @param plugin - the plugin. * @param plugin - the plugin.
*/ */
public TinyProtocol(Plugin plugin) { public TinyProtocol(Plugin plugin) {
this.plugin = plugin; this.plugin = plugin;
// Compute handler name // Compute handler name
this.handlerName = getHandlerName(); this.handlerName = getHandlerName();
// Prepare existing players // Prepare existing players
registerBukkitEvents(); registerBukkitEvents();
registerChannelHandler(); registerChannelHandler();
@ -100,109 +102,119 @@ public abstract class TinyProtocol {
} }
private void createServerChannelHandler() { private void createServerChannelHandler() {
// Handle connected channels // Handle connected channels
endInitProtocol = new ChannelInitializer<Channel>() { endInitProtocol = new ChannelInitializer<Channel>() {
@Override
protected void initChannel(Channel channel) throws Exception {
try {
// This can take a while, so we need to stop the main thread from interfering
synchronized (networkManagers) {
// Stop injecting channels
if (closed)
return;
injectChannelInternal(channel);
}
} catch (Exception e) {
plugin.getLogger().log(Level.SEVERE, "Cannot inject incomming channel " + channel, e);
}
}
};
// This is executed before Minecraft's channel handler
beginInitProtocol = new ChannelInitializer<Channel>() {
@Override
protected void initChannel(Channel channel) throws Exception {
channel.pipeline().addLast(endInitProtocol);
}
};
serverChannelHandler = new ChannelInboundHandlerAdapter() {
@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
Channel channel = (Channel) msg;
// Prepare to initialize ths channel @Override
channel.pipeline().addFirst(beginInitProtocol); protected void initChannel(Channel channel) throws Exception {
ctx.fireChannelRead(msg); try {
} // This can take a while, so we need to stop the main thread from interfering
}; synchronized (networkManagers) {
} // Stop injecting channels
if (!closed) {
injectChannelInternal(channel);
}
}
} catch (Exception e) {
plugin.getLogger().log(Level.SEVERE, "Cannot inject incomming channel " + channel, e);
}
}
};
// This is executed before Minecraft's channel handler
beginInitProtocol = new ChannelInitializer<Channel>() {
@Override
protected void initChannel(Channel channel) throws Exception {
channel.pipeline().addLast(endInitProtocol);
}
};
serverChannelHandler = new ChannelInboundHandlerAdapter() {
@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
Channel channel = (Channel) msg;
// Prepare to initialize ths channel
channel.pipeline().addFirst(beginInitProtocol);
ctx.fireChannelRead(msg);
}
};
}
/** /**
* Register bukkit events. * Register bukkit events.
*/ */
private void registerBukkitEvents() { private void registerBukkitEvents() {
listener = new Listener() { listener = new Listener() {
@EventHandler(priority = EventPriority.LOWEST) @EventHandler(priority = EventPriority.LOWEST)
public final void onPlayerLogin(PlayerLoginEvent e) { public final void onPlayerLogin(PlayerLoginEvent e) {
if (closed) if (closed)
return; return;
Channel channel = getChannel(e.getPlayer()); Channel channel = getChannel(e.getPlayer());
// Don't inject players that have been explicitly uninjected // Don't inject players that have been explicitly uninjected
if (!uninjectedChannels.contains(channel)) { if (!uninjectedChannels.contains(channel)) {
injectPlayer(e.getPlayer()); injectPlayer(e.getPlayer());
} }
} }
@EventHandler @EventHandler
public final void onPluginDisable(PluginDisableEvent e) { public final void onPluginDisable(PluginDisableEvent e) {
if (e.getPlugin().equals(plugin)) { if (e.getPlugin().equals(plugin)) {
close(); close();
} }
} }
}; };
this.plugin.getServer().getPluginManager().registerEvents(listener, plugin); plugin.getServer().getPluginManager().registerEvents(listener, plugin);
} }
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
private void registerChannelHandler() { private void registerChannelHandler() {
Object mcServer = getMinecraftServer.get(Bukkit.getServer()); Object mcServer = getMinecraftServer.get(Bukkit.getServer());
Object serverConnection = getServerConnection.get(mcServer); Object serverConnection = getServerConnection.get(mcServer);
boolean looking = true; boolean looking = true;
// We need to synchronize against this list // We need to synchronize against this list
networkManagers = (List<Object>) getNetworkMarkers.invoke(null, serverConnection); networkManagers = (List<Object>) getNetworkMarkers.invoke(null, serverConnection);
createServerChannelHandler(); createServerChannelHandler();
// Find the correct list, or implicitly throw an exception // Find the correct list, or implicitly throw an exception
for (int i = 0; looking; i++) { for (int i = 0; looking; i++) {
List<Object> list = Reflection.getField(serverConnection.getClass(), List.class, i).get(serverConnection); List<Object> list = Reflection.getField(serverConnection.getClass(), List.class, i).get(serverConnection);
for (Object item : list) { for (Object item : list) {
if (!ChannelFuture.class.isInstance(item)) if (!ChannelFuture.class.isInstance(item))
break; break;
// Channel future that contains the server connection // Channel future that contains the server connection
Channel serverChannel = ((ChannelFuture)item).channel(); Channel serverChannel = ((ChannelFuture) item).channel();
serverChannels.add(serverChannel); serverChannels.add(serverChannel);
serverChannel.pipeline().addFirst(serverChannelHandler); serverChannel.pipeline().addFirst(serverChannelHandler);
looking = false; looking = false;
} }
} }
} }
private void unregisterChannelHandler() { private void unregisterChannelHandler() {
if (serverChannelHandler == null) if (serverChannelHandler == null)
return; return;
for (Channel serverChannel : serverChannels) { for (Channel serverChannel : serverChannels) {
final ChannelPipeline pipeline = serverChannel.pipeline(); final ChannelPipeline pipeline = serverChannel.pipeline();
// Remove channel handler // Remove channel handler
serverChannel.eventLoop().execute(new Runnable() { serverChannel.eventLoop().execute(new Runnable() {
@Override @Override
public void run() { public void run() {
try { try {
@ -211,20 +223,22 @@ public abstract class TinyProtocol {
// That's fine // That's fine
} }
} }
}); });
} }
} }
private void registerPlayers(Plugin plugin) { private void registerPlayers(Plugin plugin) {
for (Player player : plugin.getServer().getOnlinePlayers()) { for (Player player : plugin.getServer().getOnlinePlayers()) {
injectPlayer(player); injectPlayer(player);
} }
} }
/** /**
* Invoked when the server is starting to send a packet to a player. * Invoked when the server is starting to send a packet to a player.
* <p> * <p>
* Note that this is not executed on the main thread. * Note that this is not executed on the main thread.
*
* @param reciever - the receiving player, NULL for early login/status packets. * @param reciever - the receiving player, NULL for early login/status packets.
* @param remoteAddress - remote address of the sending client. Never NULL. * @param remoteAddress - remote address of the sending client. Never NULL.
* @param packet - the packet being sent. * @param packet - the packet being sent.
@ -238,6 +252,7 @@ public abstract class TinyProtocol {
* Invoked when the server has received a packet from a given player. * Invoked when the server has received a packet from a given player.
* <p> * <p>
* Use {@link Channel#remoteAddress()} to get the remote address of the client. * Use {@link Channel#remoteAddress()} to get the remote address of the client.
*
* @param sender - the player that sent the packet, NULL for early login/status packets. * @param sender - the player that sent the packet, NULL for early login/status packets.
* @param channel - channel that received the packet. Never NULL. * @param channel - channel that received the packet. Never NULL.
* @param packet - the packet being received. * @param packet - the packet being received.
@ -246,133 +261,145 @@ public abstract class TinyProtocol {
public Object onPacketInAsync(Player sender, Channel channel, Object packet) { public Object onPacketInAsync(Player sender, Channel channel, Object packet) {
return packet; return packet;
} }
/** /**
* Send a packet to a particular player. * Send a packet to a particular player.
* <p> * <p>
* Note that {@link #onPacketOutAsync(Player, Object)} will be invoked with this packet. * Note that {@link #onPacketOutAsync(Player, Object)} will be invoked with this packet.
*
* @param player - the destination player. * @param player - the destination player.
* @param packet - the packet to send. * @param packet - the packet to send.
*/ */
public void sendPacket(Player player, Object packet) { public void sendPacket(Player player, Object packet) {
sendPacket(getChannel(player), packet); sendPacket(getChannel(player), packet);
} }
/** /**
* Send a packet to a particular client. * Send a packet to a particular client.
* <p> * <p>
* Note that {@link #onPacketOutAsync(Player, Object)} will be invoked with this packet. * Note that {@link #onPacketOutAsync(Player, Object)} will be invoked with this packet.
*
* @param channel - client identified by a channel. * @param channel - client identified by a channel.
* @param packet - the packet to send. * @param packet - the packet to send.
*/ */
public void sendPacket(Channel channel, Object packet) { public void sendPacket(Channel channel, Object packet) {
channel.pipeline().writeAndFlush(packet); channel.pipeline().writeAndFlush(packet);
} }
/** /**
* Pretend that a given packet has been received from a player. * Pretend that a given packet has been received from a player.
* <p> * <p>
* Note that {@link #onPacketInAsync(Player, Object)} will be invoked with this packet. * Note that {@link #onPacketInAsync(Player, Object)} will be invoked with this packet.
*
* @param player - the player that sent the packet. * @param player - the player that sent the packet.
* @param packet - the packet that will be received by the server. * @param packet - the packet that will be received by the server.
*/ */
public void receivePacket(Player player, Object packet) { public void receivePacket(Player player, Object packet) {
receivePacket(getChannel(player), packet); receivePacket(getChannel(player), packet);
} }
/** /**
* Pretend that a given packet has been received from a given client. * Pretend that a given packet has been received from a given client.
* <p> * <p>
* Note that {@link #onPacketInAsync(Player, Object)} will be invoked with this packet. * Note that {@link #onPacketInAsync(Player, Object)} will be invoked with this packet.
*
* @param channel - client identified by a channel. * @param channel - client identified by a channel.
* @param packet - the packet that will be received by the server. * @param packet - the packet that will be received by the server.
*/ */
public void receivePacket(Channel channel, Object packet) { public void receivePacket(Channel channel, Object packet) {
channel.pipeline().context("encoder").fireChannelRead(packet); channel.pipeline().context("encoder").fireChannelRead(packet);
} }
/** /**
* Retrieve the name of the channel injector, default implementation is "tiny-" + plugin name + "-" + a unique ID. * Retrieve the name of the channel injector, default implementation is "tiny-" + plugin name + "-" + a unique ID.
* <p> * <p>
* Note that this method will only be invoked once. It is no longer necessary to override this to support multiple instances. * Note that this method will only be invoked once. It is no longer necessary to override this to support multiple instances.
*
* @return A unique channel handler name. * @return A unique channel handler name.
*/ */
protected String getHandlerName() { protected String getHandlerName() {
return "tiny-" + plugin.getName() + "-" + ID.incrementAndGet(); return "tiny-" + plugin.getName() + "-" + ID.incrementAndGet();
} }
/** /**
* Add a custom channel handler to the given player's channel pipeline, * Add a custom channel handler to the given player's channel pipeline, allowing us to intercept sent and received packets.
* allowing us to intercept sent and received packets.
* <p> * <p>
* This will automatically be called when a player has logged in. * This will automatically be called when a player has logged in.
*
* @param player - the player to inject. * @param player - the player to inject.
*/ */
public void injectPlayer(Player player) { public void injectPlayer(Player player) {
injectChannelInternal(getChannel(player)).player = player; injectChannelInternal(getChannel(player)).player = player;
} }
/** /**
* Add a custom channel handler to the given channel. * Add a custom channel handler to the given channel.
*
* @param player - the channel to inject. * @param player - the channel to inject.
* @return The intercepted channel, or NULL if it has already been injected. * @return The intercepted channel, or NULL if it has already been injected.
*/ */
public void injectChannel(Channel channel) { public void injectChannel(Channel channel) {
injectChannelInternal(channel); injectChannelInternal(channel);
} }
/** /**
* Add a custom channel handler to the given channel. * Add a custom channel handler to the given channel.
*
* @param player - the channel to inject. * @param player - the channel to inject.
* @return The packet interceptor. * @return The packet interceptor.
*/ */
private PacketInterceptor injectChannelInternal(Channel channel) { private PacketInterceptor injectChannelInternal(Channel channel) {
try { try {
PacketInterceptor interceptor = (PacketInterceptor) channel.pipeline().get(handlerName); PacketInterceptor interceptor = (PacketInterceptor) channel.pipeline().get(handlerName);
// Inject our packet interceptor // Inject our packet interceptor
if (interceptor == null) { if (interceptor == null) {
interceptor = new PacketInterceptor(); interceptor = new PacketInterceptor();
channel.pipeline().addBefore("packet_handler", handlerName, interceptor); channel.pipeline().addBefore("packet_handler", handlerName, interceptor);
uninjectedChannels.remove(channel); uninjectedChannels.remove(channel);
} }
return interceptor; return interceptor;
} catch (IllegalArgumentException e) { } catch (IllegalArgumentException e) {
// Try again // Try again
return (PacketInterceptor) channel.pipeline().get(handlerName); return (PacketInterceptor) channel.pipeline().get(handlerName);
} }
} }
/** /**
* Retrieve the Netty channel associated with a player. This is cached. * Retrieve the Netty channel associated with a player. This is cached.
*
* @param player - the player. * @param player - the player.
* @return The Netty channel. * @return The Netty channel.
*/ */
public Channel getChannel(Player player) { public Channel getChannel(Player player) {
Channel channel = channelLookup.get(player.getName()); Channel channel = channelLookup.get(player.getName());
// Lookup channel again // Lookup channel again
if (channel == null) { if (channel == null) {
Object connection = getConnection.get(getPlayerHandle.invoke(player)); Object connection = getConnection.get(getPlayerHandle.invoke(player));
Object manager = getManager.get(connection); Object manager = getManager.get(connection);
channelLookup.put(player.getName(), channel = getChannel.get(manager)); channelLookup.put(player.getName(), channel = getChannel.get(manager));
} }
return channel; return channel;
} }
/** /**
* Uninject a specific player. * Uninject a specific player.
*
* @param player - the injected player. * @param player - the injected player.
*/ */
public void uninjectPlayer(Player player) { public void uninjectPlayer(Player player) {
uninjectChannel(getChannel(player)); uninjectChannel(getChannel(player));
} }
/** /**
* Uninject a specific channel. * Uninject a specific channel.
* <p> * <p>
* This will also disable the automatic channel injection that occurs when a player has properly logged in. * This will also disable the automatic channel injection that occurs when a player has properly logged in.
*
* @param channel - the injected channel. * @param channel - the injected channel.
*/ */
public void uninjectChannel(final Channel channel) { public void uninjectChannel(final Channel channel) {
@ -380,75 +407,82 @@ public abstract class TinyProtocol {
if (!closed) { if (!closed) {
uninjectedChannels.add(channel); uninjectedChannels.add(channel);
} }
// See ChannelInjector in ProtocolLib, line 590 // See ChannelInjector in ProtocolLib, line 590
channel.eventLoop().execute(new Runnable() { channel.eventLoop().execute(new Runnable() {
@Override @Override
public void run() { public void run() {
channel.pipeline().remove(handlerName); channel.pipeline().remove(handlerName);
} }
}); });
} }
/** /**
* Determine if the given player has been injected by TinyProtocol. * Determine if the given player has been injected by TinyProtocol.
*
* @param player - the player. * @param player - the player.
* @return TRUE if it is, FALSE otherwise. * @return TRUE if it is, FALSE otherwise.
*/ */
public boolean hasInjected(Player player) { public boolean hasInjected(Player player) {
return hasInjected(getChannel(player)); return hasInjected(getChannel(player));
} }
/** /**
* Determine if the given channel has been injected by TinyProtocol. * Determine if the given channel has been injected by TinyProtocol.
*
* @param channel - the channel. * @param channel - the channel.
* @return TRUE if it is, FALSE otherwise. * @return TRUE if it is, FALSE otherwise.
*/ */
public boolean hasInjected(Channel channel) { public boolean hasInjected(Channel channel) {
return channel.pipeline().get(handlerName) != null; return channel.pipeline().get(handlerName) != null;
} }
/** /**
* Cease listening for packets. This is called automatically when your plugin is disabled. * Cease listening for packets. This is called automatically when your plugin is disabled.
*/ */
public final void close() { public final void close() {
if (!closed) { if (!closed) {
closed = true; closed = true;
// Remove our handlers // Remove our handlers
for (Player player : plugin.getServer().getOnlinePlayers()) { for (Player player : plugin.getServer().getOnlinePlayers()) {
uninjectPlayer(player); uninjectPlayer(player);
} }
// Clean up Bukkit // Clean up Bukkit
HandlerList.unregisterAll(listener); HandlerList.unregisterAll(listener);
unregisterChannelHandler(); unregisterChannelHandler();
} }
} }
/** /**
* Channel handler that is inserted into the player's channel pipeline, allowing us to intercept sent and received packets. * Channel handler that is inserted into the player's channel pipeline, allowing us to intercept sent and received packets.
*
* @author Kristian * @author Kristian
*/ */
private final class PacketInterceptor extends ChannelDuplexHandler { private final class PacketInterceptor extends ChannelDuplexHandler {
// Updated by the login event // Updated by the login event
public volatile Player player; public volatile Player player;
@Override @Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception { public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
// Intercept channel // Intercept channel
final Channel channel = ctx.channel(); final Channel channel = ctx.channel();
handleLoginStart(channel, msg); handleLoginStart(channel, msg);
try { try {
msg = onPacketInAsync(player, channel, msg); msg = onPacketInAsync(player, channel, msg);
} catch (Exception e) { } catch (Exception e) {
plugin.getLogger().log(Level.SEVERE, "Error in onPacketInAsync().", e); plugin.getLogger().log(Level.SEVERE, "Error in onPacketInAsync().", e);
} }
if (msg != null) { if (msg != null) {
super.channelRead(ctx, msg); super.channelRead(ctx, msg);
} }
} }
@Override @Override
public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception { public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
try { try {
@ -456,12 +490,12 @@ public abstract class TinyProtocol {
} catch (Exception e) { } catch (Exception e) {
plugin.getLogger().log(Level.SEVERE, "Error in onPacketOutAsync().", e); plugin.getLogger().log(Level.SEVERE, "Error in onPacketOutAsync().", e);
} }
if (msg != null) { if (msg != null) {
super.write(ctx, msg, promise); super.write(ctx, msg, promise);
} }
} }
private void handleLoginStart(Channel channel, Object packet) { private void handleLoginStart(Channel channel, Object packet) {
if (PACKET_LOGIN_IN_START.isInstance(packet)) { if (PACKET_LOGIN_IN_START.isInstance(packet)) {
GameProfile profile = getGameProfile.get(packet); GameProfile profile = getGameProfile.get(packet);
@ -469,4 +503,4 @@ public abstract class TinyProtocol {
} }
} }
} }
} }