Upstream/BungeeCord
1
0
Fork 0
mirror of https://github.com/SpigotMC/BungeeCord.git synced 2024-11-23 18:55:22 +01:00
Code Issues Projects Releases Wiki Activity

#3438: Fix possible race condition in duplicate player check

Browse Source
This commit is contained in:
BoomEaro 2023-11-01 21:32:31 +11:00 committed by md_5
parent df20effacc
commit f5157f12a4
No known key found for this signature in database
GPG Key ID: E8E901AC7C617C11
4 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
proxy/src/main/java/net/md_5/bungee/BungeeCord.java
View File

@ -753,7 +753,7 @@ public class BungeeCord extends ProxyServer
}
}
public void addConnection(UserConnection con)
public boolean addConnection(UserConnection con)
{
UUID offlineId = con.getPendingConnection().getOfflineId();
if ( offlineId != null && offlineId.version() != 3 )
@ -763,6 +763,10 @@ public class BungeeCord extends ProxyServer
connectionLock.writeLock().lock();
try
{
if ( connections.containsKey( con.getName() ) || connectionsByUUID.containsKey( con.getUniqueId() ) || connectionsByOfflineUUID.containsKey( offlineId ) )
{
return false;
}
connections.put( con.getName(), con );
connectionsByUUID.put( con.getUniqueId(), con );
connectionsByOfflineUUID.put( offlineId, con );
@ -770,6 +774,7 @@ public class BungeeCord extends ProxyServer
{
connectionLock.writeLock().unlock();
}
return true;
}
public void removeConnection(UserConnection con)

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

@ -153,7 +153,7 @@ public final class UserConnection implements ProxiedPlayer
}
};
public void init()
public boolean init()
{
this.entityRewrite = EntityMap.getEntityMap( getPendingConnection().getVersion() );
@ -172,6 +172,8 @@ public final class UserConnection implements ProxiedPlayer
// Set whether the connection has a 1.8 FML marker in the handshake.
forgeClientHandler.setFmlTokenInHandshake( this.getPendingConnection().getExtraDataInHandshake().contains( ForgeConstants.FML_HANDSHAKE_TOKEN ) );
return BungeeCord.getInstance().addConnection( this );
}
public void sendPacket(PacketWrapper packet)

6
proxy/src/main/java/net/md_5/bungee/connection/InitialHandler.java
View File

@ -613,7 +613,11 @@ public class InitialHandler extends PacketHandler implements PendingConnection
private void finish2()
{
userCon.init();
if ( !userCon.init() )
{
disconnect( bungee.getTranslation( "already_connected_proxy" ) );
return;
}
ch.getHandle().pipeline().get( HandlerBoss.class ).setHandler( new UpstreamBridge( bungee, userCon ) );
bungee.getPluginManager().callEvent( new PostLoginEvent( userCon ) );

1
proxy/src/main/java/net/md_5/bungee/connection/UpstreamBridge.java
View File

@ -54,7 +54,6 @@ public class UpstreamBridge extends PacketHandler
this.bungee = bungee;
this.con = con;
BungeeCord.getInstance().addConnection( con );
con.getTabListHandler().onConnect();
}