PandacubeFr/BungeeCord
1
0
Fork 0
Code Releases Activity

Fix entity id remapping

Browse Source 
The old system only worked in ints and 1.7 changed some to varints. Since the length of the varint is variable more work is needed to remap ids
This commit is contained in:
Thinkofdeath 2013-12-08 13:47:49 +00:00
parent 85c27f30ee
commit cfda905d98
2 changed files with 73 additions and 64 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

123
proxy/src/main/java/net/md_5/bungee/EntityMap.java
View File

@ -10,83 +10,94 @@ import net.md_5.bungee.protocol.PacketWrapper;
public class EntityMap
{
public final static int[][] entityIds = new int[ 256 ][];
private final static boolean[] clientboundInts = new boolean[256];
private final static boolean[] clientboundVarInts = new boolean[256];
static
{
entityIds[0x0A] = new int[]
{
0
};
entityIds[0x0D] = new int[]
{
4
};
entityIds[0x12] = new int[]
{
0
};
entityIds[0x1A] = new int[]
{
0
};
entityIds[0x1B] = new int[]
{
0, 4
};
entityIds[0x1C] = new int[]
{
0 // TODO: Meta
};
entityIds[0x1D] = new int[]
{
0
};
entityIds[0x1E] = new int[]
{
0
};
entityIds[0x20] = new int[]
{
0
};
clientboundInts[0x04] = true;
clientboundInts[0x0A] = true;
clientboundVarInts[0x0B] = true;
clientboundVarInts[0x0C] = true;
clientboundInts[0x0D] = true;
clientboundVarInts[0x0E] = true;
clientboundVarInts[0x0F] = true;
clientboundVarInts[0x10] = true;
clientboundVarInts[0x11] = true;
clientboundInts[0x12] = true;
clientboundInts[0x14] = true;
clientboundInts[0x15] = true;
clientboundInts[0x16] = true;
clientboundInts[0x17] = true;
clientboundInts[0x18] = true;
clientboundInts[0x19] = true;
clientboundInts[0x1A] = true;
clientboundInts[0x1B] = true;
clientboundInts[0x1C] = true;
clientboundInts[0x1D] = true;
clientboundInts[0x1E] = true;
clientboundInts[0x20] = true;
clientboundVarInts[0x25] = true;
clientboundVarInts[0x2C] = true;
}
public static void rewrite(ByteBuf packet, int oldId, int newId)
public static void rewriteClientbound(ByteBuf packet, int serverEntityId, int clientEntityId)
{
int readerIndex = packet.readerIndex();
int packetId = DefinedPacket.readVarInt( packet );
int packetIdLength = packet.readerIndex() - readerIndex;
int[] idArray = entityIds[packetId];
if ( idArray != null )
if ( clientboundInts[packetId] )
{
for ( int pos : idArray )
int readId = packet.getInt( packetIdLength );
if ( readId == serverEntityId )
{
int readId = packet.getInt( packetIdLength + pos );
if ( readId == oldId )
packet.setInt( packetIdLength, clientEntityId );
} else if ( readId == clientEntityId )
{
packet.setInt( packetIdLength + pos, newId );
}
}
packet.setInt( packetIdLength, serverEntityId );
}
if ( packetId == 0x0E )
if ( packetId == 0x0D || packetId == 0x1B )
{
DefinedPacket.readVarInt( packet );
byte type = packet.readByte();
if ( type == 60 || type == 90 )
readId = packet.getInt( packetIdLength + 4 );
if ( readId == serverEntityId )
{
packet.skipBytes( 14 );
int pos = packet.readerIndex();
int shooterId = packet.getInt( pos );
if ( shooterId == oldId )
packet.setInt( packetIdLength + 4, clientEntityId );
} else if ( readId == clientEntityId )
{
packet.setInt( pos, newId );
packet.setInt( packetIdLength + 4, serverEntityId );
}
}
} else if ( clientboundVarInts[packetId] )
{
// Need to rewrite the packet because VarInts are variable length
int readId = DefinedPacket.readVarInt( packet );
int readIdLength = packet.readerIndex() - readerIndex - packetIdLength;
if ( readId == serverEntityId || readId == clientEntityId )
{
ByteBuf data = packet.slice();
packet.readerIndex( readerIndex );
packet.writerIndex( packetIdLength );
DefinedPacket.writeVarInt( readId == serverEntityId ? clientEntityId : serverEntityId, packet );
packet.writeBytes( data );
data.release();
}
} else if ( packetId == 0x13 )
{
int count = packet.getByte( packetIdLength );
for ( int i = 0; i < count; i++ )
{
int readId = packet.getInt( packetIdLength + 1 + i * 4);
if ( readId == serverEntityId )
{
packet.setInt( packetIdLength + 1 + i * 4, clientEntityId );
} else if ( readId == clientEntityId )
{
packet.setInt( packetIdLength + 1 + i * 4, serverEntityId );
}
}
}
packet.readerIndex( readerIndex );
}
}

4
proxy/src/main/java/net/md_5/bungee/connection/DownstreamBridge.java
View File

@ -3,10 +3,8 @@ package net.md_5.bungee.connection;
import com.google.common.io.ByteArrayDataOutput;
import com.google.common.io.ByteStreams;
import java.io.DataInput;
import java.util.List;
import java.util.Objects;
import lombok.RequiredArgsConstructor;
import net.md_5.bungee.BungeeCord;
import net.md_5.bungee.EntityMap;
import net.md_5.bungee.ServerConnection;
import net.md_5.bungee.api.event.ServerDisconnectEvent;
@ -80,7 +78,7 @@ public class DownstreamBridge extends PacketHandler
{
if ( !server.isObsolete() )
{
EntityMap.rewrite( packet.buf, con.getServerEntityId(), con.getClientEntityId() );
EntityMap.rewriteClientbound( packet.buf, con.getServerEntityId(), con.getClientEntityId() );
con.sendPacket( packet );
}
}