Skip to content

Commit 6377a44

Browse files
valapheevalaphee
committed
Remove most-times unnecessary client reconfiguration
1 parent 8f3dea5 commit 6377a44

File tree

8 files changed

+227
-27
lines changed

8 files changed

+227
-27
lines changed

‎proxy/src/main/java/com/velocitypowered/proxy/connection/MinecraftSessionHandler.java‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
import com.velocitypowered.proxy.protocol.packet.LoginAcknowledgedPacket;
3838
import com.velocitypowered.proxy.protocol.packet.LoginPluginMessagePacket;
3939
import com.velocitypowered.proxy.protocol.packet.LoginPluginResponsePacket;
40+
import com.velocitypowered.proxy.protocol.packet.ObjectivePacket;
4041
import com.velocitypowered.proxy.protocol.packet.PingIdentifyPacket;
4142
import com.velocitypowered.proxy.protocol.packet.PluginMessagePacket;
4243
import com.velocitypowered.proxy.protocol.packet.RemovePlayerInfoPacket;
@@ -364,4 +365,8 @@ default boolean handle(ClientboundCustomReportDetailsPacket packet) {
364365
default boolean handle(ClientboundServerLinksPacket packet) {
365366
return false;
366367
}
368+
369+
default boolean handle(ObjectivePacket packet) {
370+
return false;
371+
}
367372
}

‎proxy/src/main/java/com/velocitypowered/proxy/connection/backend/BackendPlaySessionHandler.java‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@
5656
import com.velocitypowered.proxy.protocol.packet.DisconnectPacket;
5757
import com.velocitypowered.proxy.protocol.packet.KeepAlivePacket;
5858
import com.velocitypowered.proxy.protocol.packet.LegacyPlayerListItemPacket;
59+
import com.velocitypowered.proxy.protocol.packet.ObjectivePacket;
5960
import com.velocitypowered.proxy.protocol.packet.PluginMessagePacket;
6061
import com.velocitypowered.proxy.protocol.packet.RemovePlayerInfoPacket;
6162
import com.velocitypowered.proxy.protocol.packet.RemoveResourcePackPacket;
@@ -183,6 +184,16 @@ public boolean handle(BossBarPacket packet) {
183184
return false; // forward
184185
}
185186

187+
@Override
188+
public boolean handle(ObjectivePacket packet) {
189+
if (packet.getAction() == ObjectivePacket.ADD) {
190+
playerSessionHandler.getServerObjectives().add(packet.getName());
191+
} else if (packet.getAction() == ObjectivePacket.REMOVE) {
192+
playerSessionHandler.getServerObjectives().remove(packet.getName());
193+
}
194+
return false; // forward
195+
}
196+
186197
@Override
187198
public boolean handle(final ResourcePackRequestPacket packet) {
188199
final ResourcePackInfo.Builder builder = new VelocityResourcePackInfo.BuilderImpl(

‎proxy/src/main/java/com/velocitypowered/proxy/connection/backend/ConfigSessionHandler.java‎

Lines changed: 45 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@
3838
import com.velocitypowered.proxy.connection.util.ConnectionRequestResults;
3939
import com.velocitypowered.proxy.connection.util.ConnectionRequestResults.Impl;
4040
import com.velocitypowered.proxy.protocol.MinecraftPacket;
41+
import com.velocitypowered.proxy.protocol.ProtocolUtils;
4142
import com.velocitypowered.proxy.protocol.StateRegistry;
4243
import com.velocitypowered.proxy.protocol.netty.MinecraftDecoder;
4344
import com.velocitypowered.proxy.protocol.packet.ClientboundCookieRequestPacket;
@@ -52,6 +53,7 @@
5253
import com.velocitypowered.proxy.protocol.packet.config.ClientboundCustomReportDetailsPacket;
5354
import com.velocitypowered.proxy.protocol.packet.config.ClientboundServerLinksPacket;
5455
import com.velocitypowered.proxy.protocol.packet.config.FinishedUpdatePacket;
56+
import com.velocitypowered.proxy.protocol.packet.config.KnownPacksPacket;
5557
import com.velocitypowered.proxy.protocol.packet.config.RegistrySyncPacket;
5658
import com.velocitypowered.proxy.protocol.packet.config.StartUpdatePacket;
5759
import com.velocitypowered.proxy.protocol.packet.config.TagsUpdatePacket;
@@ -61,6 +63,8 @@
6163
import java.io.IOException;
6264
import java.net.InetSocketAddress;
6365
import java.util.concurrent.CompletableFuture;
66+
import io.netty.buffer.ByteBuf;
67+
import io.netty.buffer.Unpooled;
6468
import net.kyori.adventure.key.Key;
6569
import org.apache.logging.log4j.LogManager;
6670
import org.apache.logging.log4j.Logger;
@@ -230,24 +234,21 @@ public boolean handle(RemoveResourcePackPacket packet) {
230234
public boolean handle(FinishedUpdatePacket packet) {
231235
final MinecraftConnection smc = serverConn.ensureConnected();
232236
final ConnectedPlayer player = serverConn.getPlayer();
233-
final ClientConfigSessionHandler configHandler = (ClientConfigSessionHandler) player.getConnection().getActiveSessionHandler();
234237

235238
smc.getChannel().pipeline().get(MinecraftDecoder.class).setState(StateRegistry.PLAY);
236-
//noinspection DataFlowIssue
237-
configHandler.handleBackendFinishUpdate(serverConn).thenRunAsync(() -> {
238-
smc.write(FinishedUpdatePacket.INSTANCE);
239-
if (serverConn == player.getConnectedServer()) {
240-
smc.setActiveSessionHandler(StateRegistry.PLAY);
241-
player.sendPlayerListHeaderAndFooter(player.getPlayerListHeader(), player.getPlayerListFooter());
242-
// The client cleared the tab list. TODO: Restore changes done via TabList API
243-
player.getTabList().clearAllSilent();
244-
} else {
245-
smc.setActiveSessionHandler(StateRegistry.PLAY, new TransitionSessionHandler(server, serverConn, resultFuture));
246-
}
247-
if (player.resourcePackHandler().getFirstAppliedPack() == null && resourcePackToApply != null) {
248-
player.resourcePackHandler().queueResourcePack(resourcePackToApply);
239+
if (player.getConnection().getActiveSessionHandler() instanceof ClientConfigSessionHandler configHandler) {
240+
configHandler.handleBackendFinishUpdate(serverConn).thenRunAsync(this::finish, smc.eventLoop());
241+
} else {
242+
final String brand = serverConn.getPlayer().getClientBrand();
243+
if (brand != null) {
244+
final ByteBuf buf = Unpooled.buffer();
245+
ProtocolUtils.writeString(buf, brand);
246+
final PluginMessagePacket brandPacket = new PluginMessagePacket("minecraft:brand", buf);
247+
smc.write(brandPacket);
249248
}
250-
}, smc.eventLoop());
249+
250+
finish();
251+
}
251252
return true;
252253
}
253254

@@ -322,6 +323,17 @@ public boolean handle(TransferPacket packet) {
322323
return true;
323324
}
324325

326+
@Override
327+
public boolean handle(KnownPacksPacket packet) {
328+
// Server expects us to reply to this packet
329+
if (serverConn.getPlayer().getConnection().getState() != StateRegistry.CONFIG) {
330+
// TODO: just replay the first packet the user sent
331+
serverConn.ensureConnected().write(packet);
332+
return true;
333+
}
334+
return false; // forward
335+
}
336+
325337
@Override
326338
public boolean handle(ClientboundStoreCookiePacket packet) {
327339
server.getEventManager()
@@ -374,6 +386,24 @@ private void switchFailure(Throwable cause) {
374386
resultFuture.completeExceptionally(cause);
375387
}
376388

389+
private void finish() {
390+
final MinecraftConnection smc = serverConn.ensureConnected();
391+
final ConnectedPlayer player = serverConn.getPlayer();
392+
393+
smc.write(FinishedUpdatePacket.INSTANCE);
394+
if (serverConn == player.getConnectedServer()) {
395+
smc.setActiveSessionHandler(StateRegistry.PLAY);
396+
player.sendPlayerListHeaderAndFooter(player.getPlayerListHeader(), player.getPlayerListFooter());
397+
// The client cleared the tab list. TODO: Restore changes done via TabList API
398+
player.getTabList().clearAllSilent();
399+
} else {
400+
smc.setActiveSessionHandler(StateRegistry.PLAY, new TransitionSessionHandler(server, serverConn, resultFuture));
401+
}
402+
if (player.resourcePackHandler().getFirstAppliedPack() == null && resourcePackToApply != null) {
403+
player.resourcePackHandler().queueResourcePack(resourcePackToApply);
404+
}
405+
}
406+
377407
/**
378408
* Represents the state of the configuration stage.
379409
*/

‎proxy/src/main/java/com/velocitypowered/proxy/connection/backend/LoginSessionHandler.java‎

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -163,10 +163,7 @@ public boolean handle(ServerLoginSuccessPacket packet) {
163163
if (player.getClientSettingsPacket() != null) {
164164
smc.write(player.getClientSettingsPacket());
165165
}
166-
if (player.getConnection().getActiveSessionHandler() instanceof ClientPlaySessionHandler clientPlaySessionHandler) {
167-
smc.setAutoReading(false);
168-
clientPlaySessionHandler.doSwitch().thenAcceptAsync((unused) -> smc.setAutoReading(true), smc.eventLoop());
169-
} else {
166+
if (!(player.getConnection().getActiveSessionHandler() instanceof ClientPlaySessionHandler)) {
170167
// Initial login - the player is already in configuration state.
171168
server.getEventManager().fireAndForget(new PlayerEnteredConfigurationEvent(player, serverConn));
172169
}

‎proxy/src/main/java/com/velocitypowered/proxy/connection/client/ClientConfigSessionHandler.java‎

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -153,8 +153,9 @@ public boolean handle(final PluginMessagePacket packet) {
153153
logger.error("Exception while handling plugin message packet for {}", player, ex);
154154
return null;
155155
});
156+
return true;
156157
}
157-
return true;
158+
return false;
158159
}
159160

160161
@Override
@@ -169,14 +170,12 @@ public boolean handle(PingIdentifyPacket packet) {
169170

170171
@Override
171172
public boolean handle(KnownPacksPacket packet) {
172-
callConfigurationEvent().thenRun(() -> {
173-
player.getConnectionInFlightOrConnectedServer().ensureConnected().write(packet);
174-
}).exceptionally(ex -> {
175-
logger.error("Error forwarding known packs response to backend:", ex);
176-
return null;
177-
});
173+
if (player.getConnectionInFlight() != null) {
174+
player.getConnectionInFlight().ensureConnected().write(packet);
175+
return true;
176+
}
178177

179-
return true;
178+
return false;
180179
}
181180

182181
@Override

‎proxy/src/main/java/com/velocitypowered/proxy/connection/client/ClientPlaySessionHandler.java‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@
4747
import com.velocitypowered.proxy.protocol.packet.ClientSettingsPacket;
4848
import com.velocitypowered.proxy.protocol.packet.JoinGamePacket;
4949
import com.velocitypowered.proxy.protocol.packet.KeepAlivePacket;
50+
import com.velocitypowered.proxy.protocol.packet.ObjectivePacket;
5051
import com.velocitypowered.proxy.protocol.packet.PluginMessagePacket;
5152
import com.velocitypowered.proxy.protocol.packet.ResourcePackResponsePacket;
5253
import com.velocitypowered.proxy.protocol.packet.RespawnPacket;
@@ -81,8 +82,10 @@
8182
import java.time.Instant;
8283
import java.util.ArrayList;
8384
import java.util.Collection;
85+
import java.util.HashSet;
8486
import java.util.List;
8587
import java.util.Queue;
88+
import java.util.Set;
8689
import java.util.UUID;
8790
import java.util.concurrent.CompletableFuture;
8891
import java.util.concurrent.ConcurrentLinkedQueue;
@@ -104,6 +107,7 @@ public class ClientPlaySessionHandler implements MinecraftSessionHandler {
104107
private final ConnectedPlayer player;
105108
private boolean spawned = false;
106109
private final List<UUID> serverBossBars = new ArrayList<>();
110+
private final Set<String> serverObjectives = new HashSet<>();
107111
private final Queue<PluginMessagePacket> loginPluginMessages = new ConcurrentLinkedQueue<>();
108112
private final VelocityServer server;
109113
private @Nullable TabCompleteRequestPacket outstandingTabComplete;
@@ -526,6 +530,7 @@ public CompletableFuture<Void> doSwitch() {
526530
// Config state clears everything in the client. No need to clear later.
527531
spawned = false;
528532
serverBossBars.clear();
533+
serverObjectives.clear();
529534
player.clearPlayerListHeaderAndFooterSilent();
530535
player.getTabList().clearAllSilent();
531536
}
@@ -574,6 +579,13 @@ public void handleBackendJoinGame(JoinGamePacket joinGame, VelocityServerConnect
574579
player.getConnection().delayedWrite(deletePacket);
575580
}
576581
serverBossBars.clear();
582+
for (String serverObjective : serverObjectives) {
583+
ObjectivePacket deletePacket = new ObjectivePacket();
584+
deletePacket.setName(serverObjective);
585+
deletePacket.setAction(ObjectivePacket.REMOVE);
586+
player.getConnection().delayedWrite(deletePacket);
587+
}
588+
serverObjectives.clear();
577589

578590
// Tell the server about the proxy's plugin message channels.
579591
ProtocolVersion serverVersion = serverMc.getProtocolVersion();
@@ -645,6 +657,10 @@ public List<UUID> getServerBossBars() {
645657
return serverBossBars;
646658
}
647659

660+
public Set<String> getServerObjectives() {
661+
return serverObjectives;
662+
}
663+
648664
private boolean handleCommandTabComplete(TabCompleteRequestPacket packet) {
649665
// In 1.13+, we need to do additional work for the richer suggestions available.
650666
String command = packet.getCommand().substring(1);

‎proxy/src/main/java/com/velocitypowered/proxy/protocol/StateRegistry.java‎

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,7 @@
6868
import com.velocitypowered.proxy.protocol.packet.LoginAcknowledgedPacket;
6969
import com.velocitypowered.proxy.protocol.packet.LoginPluginMessagePacket;
7070
import com.velocitypowered.proxy.protocol.packet.LoginPluginResponsePacket;
71+
import com.velocitypowered.proxy.protocol.packet.ObjectivePacket;
7172
import com.velocitypowered.proxy.protocol.packet.PingIdentifyPacket;
7273
import com.velocitypowered.proxy.protocol.packet.PluginMessagePacket;
7374
import com.velocitypowered.proxy.protocol.packet.RemovePlayerInfoPacket;
@@ -706,6 +707,24 @@ public enum StateRegistry {
706707
ClientboundServerLinksPacket::new,
707708
map(0x7B, MINECRAFT_1_21, false),
708709
map(0x82, MINECRAFT_1_21_2, false));
710+
clientbound.register(
711+
ObjectivePacket.class,
712+
ObjectivePacket::new,
713+
map(0x3B, MINECRAFT_1_8, false),
714+
map(0x3F, MINECRAFT_1_9, false),
715+
map(0x41, MINECRAFT_1_12, false),
716+
map(0x42, MINECRAFT_1_12_1, false),
717+
map(0x45, MINECRAFT_1_13, false),
718+
map(0x49, MINECRAFT_1_14, false),
719+
map(0x4A, MINECRAFT_1_15, false),
720+
map(0x53, MINECRAFT_1_17, false),
721+
map(0x56, MINECRAFT_1_19_1, false),
722+
map(0x54, MINECRAFT_1_19_3, false),
723+
map(0x58, MINECRAFT_1_19_4, false),
724+
map(0x5A, MINECRAFT_1_20_2, false),
725+
map(0x5C, MINECRAFT_1_20_3, false),
726+
map(0x5E, MINECRAFT_1_20_5, false),
727+
map(0x64, MINECRAFT_1_21_2, false));
709728
}
710729
},
711730
LOGIN {

0 commit comments

Comments
 (0)