package net.minecraftforge.fml.common.network; import io.netty.channel.ChannelHandlerContext; import io.netty.channel.ChannelOutboundHandlerAdapter; import io.netty.channel.ChannelPromise; import io.netty.channel.embedded.EmbeddedChannel; import io.netty.util.AttributeKey; import java.util.List; import net.minecraft.entity.player.EntityPlayerMP; import net.minecraft.network.NetworkManager; import net.minecraftforge.fml.common.FMLCommonHandler; import net.minecraftforge.fml.common.network.NetworkRegistry.TargetPoint; import net.minecraftforge.fml.common.network.handshake.NetworkDispatcher; import net.minecraftforge.fml.common.network.internal.FMLProxyPacket; import net.minecraftforge.fml.relauncher.Side; import com.google.common.collect.ImmutableList; import com.google.common.collect.ImmutableSet; import com.google.common.collect.Sets; public class FMLOutboundHandler extends ChannelOutboundHandlerAdapter { public static final AttributeKey<OutboundTarget> FML_MESSAGETARGET = new AttributeKey<OutboundTarget>("fml:outboundTarget"); public static final AttributeKey<Object> FML_MESSAGETARGETARGS = new AttributeKey<Object>("fml:outboundTargetArgs"); public enum OutboundTarget { /** * The packet is sent nowhere. It will be on the {@link EmbeddedChannel#outboundMessages()} Queue. * * @author cpw * */ NOWHERE(Sets.immutableEnumSet(Side.CLIENT, Side.SERVER)) { @Override public void validateArgs(Object args) { // NOOP } @Override public List<NetworkDispatcher> selectNetworks(Object args, ChannelHandlerContext context, FMLProxyPacket packet) { return null; } }, /** * The packet is sent to the {@link NetworkDispatcher} supplied as an argument. * * @author cpw * */ DISPATCHER(Sets.immutableEnumSet(Side.SERVER)) { @Override public void validateArgs(Object args) { if (!(args instanceof NetworkDispatcher)) { throw new RuntimeException("DISPATCHER expects a NetworkDispatcher"); } } @Override public List<NetworkDispatcher> selectNetworks(Object args, ChannelHandlerContext context, FMLProxyPacket packet) { return ImmutableList.of((NetworkDispatcher)args); } }, /** * The packet is sent to the originator of the packet. This requires the inbound packet * to have it's originator information set. * * @author cpw * */ REPLY(Sets.immutableEnumSet(Side.SERVER)) { @Override public void validateArgs(Object args) { // NOOP } @Override public List<NetworkDispatcher> selectNetworks(Object args, ChannelHandlerContext context, FMLProxyPacket packet) { return ImmutableList.of(packet.getDispatcher()); } }, /** * The packet is sent to the {@link EntityPlayerMP} supplied as an argument. * * @author cpw * */ PLAYER(Sets.immutableEnumSet(Side.SERVER)) { @Override public void validateArgs(Object args) { if (!(args instanceof EntityPlayerMP)) { throw new RuntimeException("PLAYER target expects a Player arg"); } } @Override public List<NetworkDispatcher> selectNetworks(Object args, ChannelHandlerContext context, FMLProxyPacket packet) { EntityPlayerMP player = (EntityPlayerMP) args; NetworkDispatcher dispatcher = player == null ? null : player.playerNetServerHandler.netManager.channel().attr(NetworkDispatcher.FML_DISPATCHER).get(); return dispatcher == null ? ImmutableList.<NetworkDispatcher>of() : ImmutableList.of(dispatcher); } }, /** * The packet is dispatched to all players connected to the server. * @author cpw * */ ALL(Sets.immutableEnumSet(Side.SERVER)) { @Override public void validateArgs(Object args) { } @SuppressWarnings("unchecked") @Override public List<NetworkDispatcher> selectNetworks(Object args, ChannelHandlerContext context, FMLProxyPacket packet) { ImmutableList.Builder<NetworkDispatcher> builder = ImmutableList.<NetworkDispatcher>builder(); for (EntityPlayerMP player : (List<EntityPlayerMP>)FMLCommonHandler.instance().getMinecraftServerInstance().getConfigurationManager().playerEntityList) { NetworkDispatcher dispatcher = player.playerNetServerHandler.netManager.channel().attr(NetworkDispatcher.FML_DISPATCHER).get(); if (dispatcher != null) builder.add(dispatcher); } return builder.build(); } }, /** * The packet is sent to all players in the dimension identified by the integer argument. * @author cpw * */ DIMENSION(Sets.immutableEnumSet(Side.SERVER)) { @Override public void validateArgs(Object args) { if (!(args instanceof Integer)) { throw new RuntimeException("DIMENSION expects an integer argument"); } } @SuppressWarnings("unchecked") @Override public List<NetworkDispatcher> selectNetworks(Object args, ChannelHandlerContext context, FMLProxyPacket packet) { int dimension = (Integer)args; ImmutableList.Builder<NetworkDispatcher> builder = ImmutableList.<NetworkDispatcher>builder(); for (EntityPlayerMP player : (List<EntityPlayerMP>)FMLCommonHandler.instance().getMinecraftServerInstance().getConfigurationManager().playerEntityList) { if (dimension == player.dimension) { NetworkDispatcher dispatcher = player.playerNetServerHandler.netManager.channel().attr(NetworkDispatcher.FML_DISPATCHER).get(); // Null dispatchers may exist for fake players - skip them if (dispatcher != null) builder.add(dispatcher); } } return builder.build(); } }, /** * The packet is sent to all players within range of the {@link TargetPoint} argument supplied. * * @author cpw * */ ALLAROUNDPOINT(Sets.immutableEnumSet(Side.SERVER)) { @Override public void validateArgs(Object args) { if (!(args instanceof TargetPoint)) { throw new RuntimeException("ALLAROUNDPOINT expects a TargetPoint argument"); } } @SuppressWarnings("unchecked") @Override public List<NetworkDispatcher> selectNetworks(Object args, ChannelHandlerContext context, FMLProxyPacket packet) { TargetPoint tp = (TargetPoint)args; ImmutableList.Builder<NetworkDispatcher> builder = ImmutableList.<NetworkDispatcher>builder(); for (EntityPlayerMP player : (List<EntityPlayerMP>)FMLCommonHandler.instance().getMinecraftServerInstance().getConfigurationManager().playerEntityList) { if (player.dimension == tp.dimension) { double d4 = tp.x - player.posX; double d5 = tp.y - player.posY; double d6 = tp.z - player.posZ; if (d4 * d4 + d5 * d5 + d6 * d6 < tp.range * tp.range) { NetworkDispatcher dispatcher = player.playerNetServerHandler.netManager.channel().attr(NetworkDispatcher.FML_DISPATCHER).get(); if (dispatcher != null) builder.add(dispatcher); } } } return builder.build(); } }, /** * The packet is sent to the server this client is currently conversing with. * @author cpw * */ TOSERVER(Sets.immutableEnumSet(Side.CLIENT)) { @Override public void validateArgs(Object args) { } @Override public List<NetworkDispatcher> selectNetworks(Object args, ChannelHandlerContext context, FMLProxyPacket packet) { NetworkManager clientConnection = FMLCommonHandler.instance().getClientToServerNetworkManager(); return clientConnection == null || clientConnection.channel().attr(NetworkDispatcher.FML_DISPATCHER).get() == null ? ImmutableList.<NetworkDispatcher>of() : ImmutableList.of(clientConnection.channel().attr(NetworkDispatcher.FML_DISPATCHER).get()); } }; private OutboundTarget(ImmutableSet<Side> sides) { this.allowed = sides; } public final ImmutableSet<Side> allowed; public abstract void validateArgs(Object args); public abstract List<NetworkDispatcher> selectNetworks(Object args, ChannelHandlerContext context, FMLProxyPacket packet); } @Override public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception { if (!(msg instanceof FMLProxyPacket)) { return; } FMLProxyPacket pkt = (FMLProxyPacket) msg; OutboundTarget outboundTarget; Object args = null; NetworkDispatcher dispatcher = ctx.channel().attr(NetworkDispatcher.FML_DISPATCHER).get(); // INTERNAL message callback - let it pass out if (dispatcher != null) { ctx.write(msg, promise); return; } outboundTarget = ctx.channel().attr(FML_MESSAGETARGET).get(); Side channelSide = ctx.channel().attr(NetworkRegistry.CHANNEL_SOURCE).get(); if (outboundTarget != null && outboundTarget.allowed.contains(channelSide)) { args = ctx.channel().attr(FML_MESSAGETARGETARGS).get(); outboundTarget.validateArgs(args); } else if (channelSide == Side.CLIENT) { outboundTarget = OutboundTarget.TOSERVER; } else { throw new FMLNetworkException("Packet arrived at the outbound handler without a valid target!"); } List<NetworkDispatcher> dispatchers = outboundTarget.selectNetworks(args, ctx, pkt); // This will drop the messages into the output queue at the embedded channel if (dispatchers == null) { ctx.write(msg, promise); promise.setSuccess(); return; } for (NetworkDispatcher targetDispatcher : dispatchers) { targetDispatcher.sendProxy((FMLProxyPacket) msg); } promise.setSuccess(); } }