public static Object toRaw(String msg) {
   return Reflection.getTypedMethod(
           Reflection.getClass("{nms}.IChatBaseComponent$ChatSerializer"),
           "a",
           Reflection.getClass("{nms}.IChatBaseComponent"),
           String.class)
       .invoke(null, "{text:'" + msg + "'}");
 }
 public static String toPlain(Object obs) {
   String plain =
       (String)
           Reflection.getTypedMethod(
                   Reflection.getClass("{nms}.IChatBaseComponent$ChatSerializer"),
                   "a",
                   String.class,
                   Reflection.getClass("{nms}.IChatBaseComponent"))
               .invoke(null, obs);
   return plain.substring(1, plain.lastIndexOf('"'));
 }
Ejemplo n.º 3
0
  @SuppressWarnings("unchecked")
  private void registerChannelHandler() {
    Object mcServer = getMinecraftServer.get(Bukkit.getServer());
    Object serverConnection = getServerConnection.get(mcServer);
    boolean looking = true;

    networkManagers = (List<Object>) getNetworkMarkers.invoke(null, serverConnection);
    createServerChannelHandler();

    for (int i = 0; looking; i++) {
      List<Object> list =
          Reflection.getField(serverConnection.getClass(), List.class, i).get(serverConnection);

      for (Object item : list) {
        if (!ChannelFuture.class.isInstance(item)) break;

        Channel serverChannel = ((ChannelFuture) item).channel();

        serverChannels.add(serverChannel);
        serverChannel.pipeline().addFirst(serverChannelHandler);
        looking = false;
      }
    }
  }
Ejemplo n.º 4
0
/**
 * @author Kristian Modified by me. Patterned on PacketAPI (@BigTeddy98 ^^) Orginal TPAPI -
 *     https://github.com/aadnk/ProtocolLib/tree/master/modules/TinyProtocol/src/main/java/com/comphenix/tinyprotocol
 */
public class TinyProtocol extends Overlay {

  private static final AtomicInteger ID = new AtomicInteger(0);

  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> getManager =
      Reflection.getField("{nms}.PlayerConnection", "networkManager", Object.class);
  private static final FieldAccessor<Channel> getChannel =
      Reflection.getField("{nms}.NetworkManager", Channel.class, 0);

  private static final Class<Object> minecraftServerClass =
      Reflection.getUntypedClass("{nms}.MinecraftServer");
  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> getServerConnection =
      Reflection.getField(minecraftServerClass, serverConnectionClass, 0);
  private static final MethodInvoker getNetworkMarkers =
      Reflection.getTypedMethod(serverConnectionClass, null, List.class, serverConnectionClass);

  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 Map<String, Channel> channelLookup = new MapMaker().weakValues().makeMap();
  private Listener listener;

  private Set<Channel> uninjectedChannels =
      Collections.newSetFromMap(new MapMaker().weakKeys().<Channel, Boolean>makeMap());

  private List<Object> networkManagers;

  private List<Channel> serverChannels = Lists.newArrayList();
  private ChannelInboundHandlerAdapter serverChannelHandler;
  private ChannelInitializer<Channel> beginInitProtocol;
  private ChannelInitializer<Channel> endInitProtocol;

  private String handlerName;

  protected volatile boolean closed;
  protected Plugin plugin;

  public TinyProtocol(Plugin plugin) {
    this.plugin = plugin;
    this.handlerName = getHandlerName();
    registerBukkitEvents();
    registerChannelHandler();
    registerPlayers(plugin);
  }

  private void createServerChannelHandler() {
    endInitProtocol =
        new ChannelInitializer<Channel>() {

          @Override
          protected void initChannel(Channel channel) throws Exception {
            try {
              synchronized (networkManagers) {
                if (!closed) {
                  injectChannelInternal(channel);
                }
              }
            } catch (Exception e) {
              plugin.getLogger().log(Level.SEVERE, "Cannot inject incomming channel " + channel, e);
            }
          }
        };

    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;

            channel.pipeline().addFirst(beginInitProtocol);
            ctx.fireChannelRead(msg);
          }
        };
  }

  private void registerBukkitEvents() {
    listener =
        new Listener() {

          @EventHandler(priority = EventPriority.LOWEST)
          public final void onPlayerLogin(PlayerLoginEvent e) {
            if (closed) return;

            Channel channel = getChannel(e.getPlayer());

            if (!uninjectedChannels.contains(channel)) {
              injectPlayer(e.getPlayer());
            }
          }

          @EventHandler
          public final void onPluginDisable(PluginDisableEvent e) {
            if (e.getPlugin().equals(plugin)) {
              close();
            }
          }
        };

    plugin.getServer().getPluginManager().registerEvents(listener, plugin);
  }

  @SuppressWarnings("unchecked")
  private void registerChannelHandler() {
    Object mcServer = getMinecraftServer.get(Bukkit.getServer());
    Object serverConnection = getServerConnection.get(mcServer);
    boolean looking = true;

    networkManagers = (List<Object>) getNetworkMarkers.invoke(null, serverConnection);
    createServerChannelHandler();

    for (int i = 0; looking; i++) {
      List<Object> list =
          Reflection.getField(serverConnection.getClass(), List.class, i).get(serverConnection);

      for (Object item : list) {
        if (!ChannelFuture.class.isInstance(item)) break;

        Channel serverChannel = ((ChannelFuture) item).channel();

        serverChannels.add(serverChannel);
        serverChannel.pipeline().addFirst(serverChannelHandler);
        looking = false;
      }
    }
  }

  private void unregisterChannelHandler() {
    if (serverChannelHandler == null) return;

    for (Channel serverChannel : serverChannels) {
      final ChannelPipeline pipeline = serverChannel.pipeline();

      serverChannel
          .eventLoop()
          .execute(
              new Runnable() {

                @Override
                public void run() {
                  try {
                    pipeline.remove(serverChannelHandler);
                  } catch (NoSuchElementException e) {
                  }
                }
              });
    }
  }

  private void registerPlayers(Plugin plugin) {
    for (Player player : plugin.getServer().getOnlinePlayers()) {
      injectPlayer(player);
    }
  }

  public void sendPacket(Player player, Object packet) {
    sendPacket(getChannel(player), packet);
  }

  private void sendPacket(Channel channel, Object packet) {
    channel.pipeline().writeAndFlush(packet);
  }

  public void receivePacket(Player player, Object packet) {
    receivePacket(getChannel(player), packet);
  }

  private void receivePacket(Channel channel, Object packet) {
    channel.pipeline().context("encoder").fireChannelRead(packet);
  }

  protected String getHandlerName() {
    return "TinyProtocol(" + plugin.getName() + ")" + ID.incrementAndGet();
  }

  public void injectPlayer(Player player) {
    injectChannelInternal(getChannel(player)).player = player;
  }

  public void injectChannel(Channel channel) {
    injectChannelInternal(channel);
  }

  private PacketInterceptor injectChannelInternal(Channel channel) {
    try {
      PacketInterceptor interceptor = (PacketInterceptor) channel.pipeline().get(handlerName);

      if (interceptor == null) {
        interceptor = new PacketInterceptor();
        channel.pipeline().addBefore("packet_handler", handlerName, interceptor);
        uninjectedChannels.remove(channel);
      }

      return interceptor;
    } catch (IllegalArgumentException e) {
      return (PacketInterceptor) channel.pipeline().get(handlerName);
    }
  }

  public Channel getChannel(Player player) {
    Channel channel = channelLookup.get(player.getName());

    if (channel == null) {
      Object connection = getConnection.get(getPlayerHandle.invoke(player));
      Object manager = getManager.get(connection);

      channelLookup.put(player.getName(), channel = getChannel.get(manager));
    }

    return channel;
  }

  private void uninjectPlayer(Player player) {
    uninjectChannel(getChannel(player));
  }

  private void uninjectChannel(final Channel channel) {
    if (!closed) {
      uninjectedChannels.add(channel);
    }

    channel
        .eventLoop()
        .execute(
            new Runnable() {

              @Override
              public void run() {
                channel.pipeline().remove(handlerName);
              }
            });
  }

  public boolean hasInjected(Player player) {
    return hasInjected(getChannel(player));
  }

  private boolean hasInjected(Channel channel) {
    return channel.pipeline().get(handlerName) != null;
  }

  private final void close() {
    if (!closed) {
      closed = true;

      for (Player player : plugin.getServer().getOnlinePlayers()) {
        uninjectPlayer(player);
      }

      HandlerList.unregisterAll(listener);
      unregisterChannelHandler();
    }
  }

  private final class PacketInterceptor extends ChannelDuplexHandler {
    public volatile Player player;

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
      final Channel channel = ctx.channel();
      handleLoginStart(channel, msg);

      for (Entry<PacketListener, List<Method>> listener : IN.entrySet()) {
        for (Method method : listener.getValue()) {
          PacketHandler ann = method.getAnnotation(PacketHandler.class);
          if (!ann.packet().getName().equals(msg.getClass().getSimpleName())) continue;

          PacketEvent evt = new PacketEvent(player, msg);
          method.setAccessible(true);
          method.invoke(listener.getKey(), evt);
          method.setAccessible(false);
          msg = evt.getPacket();
        }
      }
      if (msg != null) {
        super.channelRead(ctx, msg);
      }
    }

    @Override
    public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise)
        throws Exception {
      for (Entry<PacketListener, List<Method>> listener : OUT.entrySet()) {
        for (Method method : listener.getValue()) {

          PacketHandler ann = method.getAnnotation(PacketHandler.class);
          if (!ann.packet().getName().equals(msg.getClass().getSimpleName())) continue;
          PacketEvent evt = new PacketEvent(player, msg);
          method.setAccessible(true);
          method.invoke(listener.getKey(), evt);
          msg = evt.getPacket();
        }
      }

      if (msg != null) {
        super.write(ctx, msg, promise);
      }
    }

    private void handleLoginStart(Channel channel, Object packet) {
      if (PACKET_LOGIN_IN_START.isInstance(packet)) {
        GameProfile profile = getGameProfile.get(packet);
        channelLookup.put(profile.getName(), channel);
      }
    }
  }
}