227 lines
13 KiB
Java
227 lines
13 KiB
Java
/*
|
|
* This file is part of ViaVersion - https://github.com/ViaVersion/ViaVersion
|
|
* Copyright (C) 2023 ViaVersion and contributors
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package com.viaversion.viaversion.protocols.protocol1_20_2to1_20;
|
|
|
|
import com.viaversion.viaversion.api.connection.UserConnection;
|
|
import com.viaversion.viaversion.api.data.MappingData;
|
|
import com.viaversion.viaversion.api.data.MappingDataBase;
|
|
import com.viaversion.viaversion.api.minecraft.entities.Entity1_19_4Types;
|
|
import com.viaversion.viaversion.api.protocol.AbstractProtocol;
|
|
import com.viaversion.viaversion.api.protocol.packet.Direction;
|
|
import com.viaversion.viaversion.api.protocol.packet.PacketWrapper;
|
|
import com.viaversion.viaversion.api.protocol.packet.State;
|
|
import com.viaversion.viaversion.api.protocol.remapper.PacketHandler;
|
|
import com.viaversion.viaversion.api.rewriter.EntityRewriter;
|
|
import com.viaversion.viaversion.api.rewriter.ItemRewriter;
|
|
import com.viaversion.viaversion.api.type.Type;
|
|
import com.viaversion.viaversion.data.entity.EntityTrackerBase;
|
|
import com.viaversion.viaversion.exception.CancelException;
|
|
import com.viaversion.viaversion.protocols.base.ClientboundLoginPackets;
|
|
import com.viaversion.viaversion.protocols.base.ServerboundLoginPackets;
|
|
import com.viaversion.viaversion.protocols.protocol1_19_4to1_19_3.ClientboundPackets1_19_4;
|
|
import com.viaversion.viaversion.protocols.protocol1_19_4to1_19_3.ServerboundPackets1_19_4;
|
|
import com.viaversion.viaversion.protocols.protocol1_20_2to1_20.packet.ClientboundConfigurationPackets1_20_2;
|
|
import com.viaversion.viaversion.protocols.protocol1_20_2to1_20.packet.ClientboundPackets1_20_2;
|
|
import com.viaversion.viaversion.protocols.protocol1_20_2to1_20.packet.ServerboundConfigurationPackets1_20_2;
|
|
import com.viaversion.viaversion.protocols.protocol1_20_2to1_20.packet.ServerboundPackets1_20_2;
|
|
import com.viaversion.viaversion.protocols.protocol1_20_2to1_20.rewriter.BlockItemPacketRewriter1_20_2;
|
|
import com.viaversion.viaversion.protocols.protocol1_20_2to1_20.rewriter.EntityPacketRewriter1_20_2;
|
|
import com.viaversion.viaversion.protocols.protocol1_20_2to1_20.storage.ConfigurationState;
|
|
import com.viaversion.viaversion.protocols.protocol1_20_2to1_20.storage.ConfigurationState.BridgePhase;
|
|
import com.viaversion.viaversion.rewriter.SoundRewriter;
|
|
import java.util.UUID;
|
|
import org.checkerframework.checker.nullness.qual.Nullable;
|
|
|
|
public final class Protocol1_20_2To1_20 extends AbstractProtocol<ClientboundPackets1_19_4, ClientboundPackets1_20_2, ServerboundPackets1_19_4, ServerboundPackets1_20_2> {
|
|
|
|
public static final MappingData MAPPINGS = new MappingDataBase("1.20", "1.20.2");
|
|
private final EntityPacketRewriter1_20_2 entityPacketRewriter = new EntityPacketRewriter1_20_2(this);
|
|
private final BlockItemPacketRewriter1_20_2 itemPacketRewriter = new BlockItemPacketRewriter1_20_2(this);
|
|
|
|
public Protocol1_20_2To1_20() {
|
|
super(ClientboundPackets1_19_4.class, ClientboundPackets1_20_2.class, ServerboundPackets1_19_4.class, ServerboundPackets1_20_2.class);
|
|
}
|
|
|
|
@Override
|
|
protected void registerPackets() {
|
|
// Close your eyes and turn around while you still can
|
|
super.registerPackets();
|
|
|
|
final SoundRewriter<ClientboundPackets1_19_4> soundRewriter = new SoundRewriter<>(this);
|
|
soundRewriter.register1_19_3Sound(ClientboundPackets1_19_4.SOUND);
|
|
soundRewriter.registerEntitySound(ClientboundPackets1_19_4.ENTITY_SOUND);
|
|
|
|
registerClientbound(ClientboundPackets1_19_4.DISPLAY_SCOREBOARD, wrapper -> {
|
|
final byte slot = wrapper.read(Type.BYTE);
|
|
wrapper.write(Type.VAR_INT, (int) slot);
|
|
});
|
|
|
|
registerServerbound(State.LOGIN, ServerboundLoginPackets.HELLO.getId(), ServerboundLoginPackets.HELLO.getId(), wrapper -> {
|
|
wrapper.passthrough(Type.STRING); // Name
|
|
|
|
final UUID uuid = wrapper.read(Type.UUID);
|
|
wrapper.write(Type.OPTIONAL_UUID, uuid);
|
|
});
|
|
|
|
// Deal with the new CONFIGURATION protocol state the client expects
|
|
// After the game profile is received by the client, it will send its login ack,
|
|
// switch to the configuration protocol state and send its brand.
|
|
// We need to wait for it send the login ack before actually sending the play login,
|
|
// hence packets are added to a queue. With the data from the login packet, we sent what is needed
|
|
// during the configuration phase before finally transitioning to the play state with the client as well.
|
|
registerClientbound(State.LOGIN, ClientboundLoginPackets.GAME_PROFILE.getId(), ClientboundLoginPackets.GAME_PROFILE.getId(), wrapper -> {
|
|
wrapper.user().get(ConfigurationState.class).setBridgePhase(BridgePhase.PROFILE_SENT);
|
|
|
|
// Set the state according to what the server expects. All packets between now and when the client
|
|
// switches to PLAY as well will be discarded after being dealt with.
|
|
wrapper.user().getProtocolInfo().setState(State.PLAY);
|
|
});
|
|
|
|
registerServerbound(State.LOGIN, ServerboundLoginPackets.LOGIN_ACKNOWLEDGED.getId(), -1, wrapper -> {
|
|
wrapper.cancel();
|
|
|
|
final ConfigurationState configurationState = wrapper.user().get(ConfigurationState.class);
|
|
configurationState.setBridgePhase(BridgePhase.CONFIGURATION);
|
|
configurationState.sendQueuedPackets(wrapper.user());
|
|
});
|
|
cancelServerbound(State.LOGIN, ServerboundLoginPackets.CUSTOM_QUERY_ANSWER.getId());
|
|
|
|
registerServerbound(State.CONFIGURATION, ServerboundConfigurationPackets1_20_2.FINISH_CONFIGURATION.getId(), -1, wrapper -> {
|
|
wrapper.cancel();
|
|
|
|
final ConfigurationState configurationState = wrapper.user().get(ConfigurationState.class);
|
|
configurationState.setBridgePhase(BridgePhase.NONE);
|
|
configurationState.sendQueuedPackets(wrapper.user());
|
|
configurationState.clear();
|
|
});
|
|
registerServerbound(State.CONFIGURATION, ServerboundConfigurationPackets1_20_2.CLIENT_INFORMATION.getId(), -1, queueServerboundPacket(ServerboundPackets1_20_2.CLIENT_SETTINGS));
|
|
registerServerbound(State.CONFIGURATION, ServerboundConfigurationPackets1_20_2.CUSTOM_PAYLOAD.getId(), -1, queueServerboundPacket(ServerboundPackets1_20_2.PLUGIN_MESSAGE));
|
|
registerServerbound(State.CONFIGURATION, ServerboundConfigurationPackets1_20_2.KEEP_ALIVE.getId(), -1, queueServerboundPacket(ServerboundPackets1_20_2.KEEP_ALIVE));
|
|
registerServerbound(State.CONFIGURATION, ServerboundConfigurationPackets1_20_2.PONG.getId(), -1, queueServerboundPacket(ServerboundPackets1_20_2.PONG));
|
|
registerServerbound(State.CONFIGURATION, ServerboundConfigurationPackets1_20_2.RESOURCE_PACK.getId(), -1, queueServerboundPacket(ServerboundPackets1_20_2.RESOURCE_PACK_STATUS));
|
|
|
|
cancelClientbound(ClientboundPackets1_19_4.UPDATE_ENABLED_FEATURES); // TODO Sad emoji
|
|
cancelServerbound(ServerboundPackets1_20_2.CONFIGURATION_ACKNOWLEDGED);
|
|
cancelServerbound(ServerboundPackets1_20_2.CHUNK_BATCH_RECEIVED);
|
|
|
|
registerServerbound(ServerboundPackets1_20_2.PING_REQUEST, null, wrapper -> {
|
|
wrapper.cancel();
|
|
final long time = wrapper.read(Type.LONG);
|
|
|
|
final PacketWrapper responsePacket = wrapper.create(ClientboundPackets1_20_2.PONG_RESPONSE);
|
|
responsePacket.write(Type.LONG, time);
|
|
responsePacket.sendFuture(Protocol1_20_2To1_20.class);
|
|
});
|
|
}
|
|
|
|
private PacketHandler queueServerboundPacket(final ServerboundPackets1_20_2 packetType) {
|
|
return wrapper -> {
|
|
wrapper.setPacketType(packetType);
|
|
wrapper.user().get(ConfigurationState.class).addPacketToQueue(wrapper, false);
|
|
wrapper.cancel();
|
|
};
|
|
}
|
|
|
|
@Override
|
|
public void transform(final Direction direction, final State state, final PacketWrapper packetWrapper) throws Exception {
|
|
final ConfigurationState configurationBridge = packetWrapper.user().get(ConfigurationState.class);
|
|
final BridgePhase phase = configurationBridge.bridgePhase();
|
|
if (phase == BridgePhase.NONE) {
|
|
super.transform(direction, state, packetWrapper);
|
|
return;
|
|
}
|
|
|
|
if (direction == Direction.SERVERBOUND) {
|
|
// Client and server will be on different protocol states, pick the right client protocol state
|
|
super.transform(direction, phase == BridgePhase.PROFILE_SENT ? State.LOGIN : State.CONFIGURATION, packetWrapper);
|
|
return;
|
|
}
|
|
|
|
if (phase == BridgePhase.PROFILE_SENT) {
|
|
// Queue packets sent by the server while we wait for the client to transition to the configuration state
|
|
configurationBridge.addPacketToQueue(packetWrapper, true);
|
|
throw CancelException.generate();
|
|
}
|
|
|
|
if (packetWrapper.getPacketType() == null || packetWrapper.getPacketType().state() != State.CONFIGURATION) {
|
|
// Map some of them to their configuration state counterparts, but make sure to let join game through
|
|
final int unmappedId = packetWrapper.getId();
|
|
if (unmappedId == ClientboundPackets1_19_4.JOIN_GAME.getId()) {
|
|
super.transform(direction, state, packetWrapper);
|
|
return;
|
|
}
|
|
|
|
if (configurationBridge.queuedOrSentJoinGame()) {
|
|
// Don't try to send configuration packets after the join game packet has been queued or sent
|
|
configurationBridge.addPacketToQueue(packetWrapper, true);
|
|
throw CancelException.generate();
|
|
}
|
|
|
|
if (unmappedId == ClientboundPackets1_19_4.PLUGIN_MESSAGE.getId()) {
|
|
packetWrapper.setPacketType(ClientboundConfigurationPackets1_20_2.CUSTOM_PAYLOAD);
|
|
} else if (unmappedId == ClientboundPackets1_19_4.DISCONNECT.getId()) {
|
|
packetWrapper.setPacketType(ClientboundConfigurationPackets1_20_2.DISCONNECT);
|
|
} else if (unmappedId == ClientboundPackets1_19_4.KEEP_ALIVE.getId()) {
|
|
packetWrapper.setPacketType(ClientboundConfigurationPackets1_20_2.KEEP_ALIVE);
|
|
} else if (unmappedId == ClientboundPackets1_19_4.PING.getId()) {
|
|
packetWrapper.setPacketType(ClientboundConfigurationPackets1_20_2.PING);
|
|
} else if (unmappedId == ClientboundPackets1_19_4.RESOURCE_PACK.getId()) {
|
|
packetWrapper.setPacketType(ClientboundConfigurationPackets1_20_2.RESOURCE_PACK);
|
|
} else if (unmappedId == ClientboundPackets1_19_4.UPDATE_ENABLED_FEATURES.getId()) {
|
|
packetWrapper.setPacketType(ClientboundConfigurationPackets1_20_2.UPDATE_ENABLED_FEATURES);
|
|
} else if (unmappedId == ClientboundPackets1_19_4.TAGS.getId()) {
|
|
packetWrapper.setPacketType(ClientboundConfigurationPackets1_20_2.UPDATE_TAGS);
|
|
} else {
|
|
// Not a packet that can be mapped to the configuration protocol
|
|
configurationBridge.addPacketToQueue(packetWrapper, true);
|
|
throw CancelException.generate();
|
|
}
|
|
return;
|
|
}
|
|
|
|
// Redirect packets during the fake configuration phase
|
|
// This might mess up people using Via API/other protocols down the line, but such is life. We can't have different states for server and client
|
|
super.transform(direction, State.CONFIGURATION, packetWrapper);
|
|
}
|
|
|
|
@Override
|
|
public MappingData getMappingData() {
|
|
return MAPPINGS;
|
|
}
|
|
|
|
@Override
|
|
protected @Nullable ServerboundPackets1_20_2 configurationAcknowledgedPacket() {
|
|
return null; // Don't handle it in the transitioning protocol
|
|
}
|
|
|
|
@Override
|
|
public void init(final UserConnection user) {
|
|
user.put(new ConfigurationState());
|
|
addEntityTracker(user, new EntityTrackerBase(user, Entity1_19_4Types.PLAYER));
|
|
}
|
|
|
|
@Override
|
|
public EntityRewriter<Protocol1_20_2To1_20> getEntityRewriter() {
|
|
return entityPacketRewriter;
|
|
}
|
|
|
|
@Override
|
|
public ItemRewriter<Protocol1_20_2To1_20> getItemRewriter() {
|
|
return itemPacketRewriter;
|
|
}
|
|
} |