Skip to content

Commit

Permalink
Implement better support for Floodgate
Browse files Browse the repository at this point in the history
Closes #9

Squashed commit of the following:

commit 04bb63c454ee82bf1a715840182dc3ccf4132122
Author: bundabrg <bundabrg@grieve.com.au>
Date:   Tue Nov 24 14:26:52 2020 +0800

    Provide support for Floodgate with MCPE and MCEE

    * If a blank XUID is provided (in the case of MCEE) we will just generate one based upon a hash of the identity

commit 3169d90
Author: bundabrg <bundabrg@grieve.com.au>
Date:   Mon Nov 23 16:48:45 2020 +0800

    Cleanup

commit 0b692c7
Author: bundabrg <bundabrg@grieve.com.au>
Date:   Mon Nov 23 16:24:43 2020 +0800

    Move LoginPacket handler from Reversion side to Geyser side
  • Loading branch information
bundabrg committed Nov 24, 2020
1 parent ad18fde commit a5050c9
Show file tree
Hide file tree
Showing 3 changed files with 150 additions and 76 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@

import au.com.grieve.geyser.reversion.GeyserReversionExtension;
import au.com.grieve.geyser.reversion.server.GeyserServerSession;
import au.com.grieve.geyser.reversion.server.GeyserUpstreamPacketHandler;
import au.com.grieve.reversion.editions.bedrock.BedrockReversionSession;
import au.com.grieve.reversion.shaded.nukkitx.protocol.bedrock.BedrockPong;
import au.com.grieve.reversion.shaded.nukkitx.protocol.bedrock.BedrockServerEventHandler;
Expand All @@ -34,7 +35,6 @@
import io.netty.channel.socket.DatagramPacket;
import lombok.Getter;
import org.geysermc.connector.GeyserConnector;
import org.geysermc.connector.network.UpstreamPacketHandler;
import org.geysermc.connector.network.session.GeyserSession;

import javax.annotation.ParametersAreNonnullByDefault;
Expand Down Expand Up @@ -83,7 +83,7 @@ public void onSessionCreation(BedrockServerSession bedrockServerSession) {
GeyserServerSession facadeSession = new GeyserServerSession((BedrockReversionSession) bedrockServerSession);
GeyserSession geyserSession = new GeyserSession(GeyserConnector.getInstance(), facadeSession);

facadeSession.setPacketHandler(new UpstreamPacketHandler(GeyserConnector.getInstance(), geyserSession));
facadeSession.setPacketHandler(new GeyserUpstreamPacketHandler((BedrockReversionSession) bedrockServerSession, GeyserConnector.getInstance(), geyserSession));
bedrockServerSession.setPacketHandler(new BedrockEditionUpstreamPacketHandler((BedrockReversionSession) bedrockServerSession,
geyserSession, facadeSession));
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,30 +24,16 @@

package au.com.grieve.geyser.reversion.editions.bedrock.handlers;

import au.com.grieve.geyser.reversion.GeyserReversionExtension;
import au.com.grieve.geyser.reversion.server.GeyserServerSession;
import au.com.grieve.reversion.api.LoginData;
import au.com.grieve.reversion.api.Translator;
import au.com.grieve.reversion.editions.bedrock.BedrockReversionSession;
import au.com.grieve.reversion.exceptions.LoginException;
import au.com.grieve.reversion.shaded.nukkitx.protocol.bedrock.BedrockPacket;
import au.com.grieve.reversion.shaded.nukkitx.protocol.bedrock.handler.BedrockPacketHandler;
import au.com.grieve.reversion.shaded.nukkitx.protocol.bedrock.packet.*;
import com.fasterxml.jackson.databind.JsonNode;
import com.nukkitx.protocol.bedrock.BedrockPacketCodec;
import com.nukkitx.protocol.bedrock.packet.PlayStatusPacket;
import com.nukkitx.protocol.bedrock.packet.ResourcePacksInfoPacket;
import com.nukkitx.protocol.bedrock.packet.ServerToClientHandshakePacket;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import lombok.Getter;
import org.geysermc.connector.network.BedrockProtocol;
import org.geysermc.connector.network.session.GeyserSession;
import org.geysermc.connector.network.session.auth.AuthData;
import org.geysermc.connector.network.session.auth.BedrockClientData;
import org.geysermc.connector.utils.LanguageUtils;

import java.util.UUID;


@Getter
Expand All @@ -62,66 +48,6 @@ public BedrockEditionUpstreamPacketHandler(BedrockReversionSession serverSession
this.facadeSession = facadeSession;
}

@Override
public boolean handle(LoginPacket loginPacket) {

// Check that we support the codec
BedrockPacketCodec packetCodec = BedrockProtocol.getBedrockCodec(loginPacket.getProtocolVersion());
if (packetCodec == null) {
if (loginPacket.getProtocolVersion() > BedrockProtocol.DEFAULT_BEDROCK_CODEC.getProtocolVersion()) {
geyserSession.disconnect(LanguageUtils.getLocaleStringLog("geyser.network.outdated.server", BedrockProtocol.DEFAULT_BEDROCK_CODEC.getMinecraftVersion()));
return true;
}

if (loginPacket.getProtocolVersion() < BedrockProtocol.DEFAULT_BEDROCK_CODEC.getProtocolVersion()) {
geyserSession.disconnect(LanguageUtils.getLocaleStringLog("geyser.network.outdated.client", BedrockProtocol.DEFAULT_BEDROCK_CODEC.getMinecraftVersion()));
return true;
}
}

// Provide some debug about our translation chain
Translator translator = serverSession.getTranslator();
if (translator != null) {
GeyserReversionExtension.getInstance().getLogger().debug("Translator Chain:");
while (translator != null) {
GeyserReversionExtension.getInstance().getLogger().debug(" " + translator);
translator = translator.getDownstreamTranslator();
}
}

// Encrypt Connection
serverSession.enableEncryption(serverSession.getLoginData().getEncryptionKey());

try {
ServerToClientHandshakePacket packet = new ServerToClientHandshakePacket();
packet.setJwt(serverSession.getLoginData().getHandshakeJwt().serialize());
geyserSession.sendUpstreamPacketImmediately(packet);
} catch (LoginException e) {
geyserSession.disconnect("You are not able to connect. Please make sure your account is authorized to connect or contact the server administrator.");
geyserSession.getConnector().getLogger().error("Failed to encrypt connection: " + e.getMessage());
return true;
}

// Setup Session
JsonNode extraData = serverSession.getLoginData().getPayload().get("extraData");
geyserSession.setAuthenticationData(new AuthData(
extraData.get("displayName").asText(),
UUID.fromString(extraData.get("identity").asText()),
extraData.get("XUID").asText()
));

serverSession.getLoginData();
geyserSession.setClientData(LoginData.JSON_MAPPER.convertValue(serverSession.getLoginData().getClientData(), BedrockClientData.class));

PlayStatusPacket playStatus = new PlayStatusPacket();
playStatus.setStatus(PlayStatusPacket.Status.LOGIN_SUCCESS);
geyserSession.sendUpstreamPacket(playStatus);

ResourcePacksInfoPacket resourcePacksInfo = new ResourcePacksInfoPacket();
geyserSession.sendUpstreamPacket(resourcePacksInfo);
return true;
}

public boolean handlePacket(BedrockPacket original) {
// Isolate Reversion protocol from Geyser Protocol in case there are overlapping differences
ByteBuf buffer = ByteBufAllocator.DEFAULT.ioBuffer();
Expand All @@ -134,6 +60,21 @@ public boolean handlePacket(BedrockPacket original) {
return translated.handle(facadeSession.getPacketHandler());
}

@Override
public boolean handle(LoginPacket packet) {
return handlePacket(packet);
}

@Override
public boolean handle(MotionPredictionHintsPacket packet) {
return handlePacket(packet);
}

@Override
public boolean handle(ItemComponentPacket packet) {
return handlePacket(packet);
}

@Override
public boolean handle(AdventureSettingsPacket packet) {
return handlePacket(packet);
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,133 @@
/*
* MIT License
*
* Copyright (c) 2020 GeyserReversion Developers
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/

package au.com.grieve.geyser.reversion.server;

import au.com.grieve.geyser.reversion.GeyserReversionExtension;
import au.com.grieve.reversion.api.LoginData;
import au.com.grieve.reversion.api.Translator;
import au.com.grieve.reversion.editions.bedrock.BedrockReversionSession;
import au.com.grieve.reversion.exceptions.LoginException;
import com.fasterxml.jackson.databind.JsonNode;
import com.nukkitx.protocol.bedrock.BedrockPacketCodec;
import com.nukkitx.protocol.bedrock.packet.LoginPacket;
import com.nukkitx.protocol.bedrock.packet.PlayStatusPacket;
import com.nukkitx.protocol.bedrock.packet.ResourcePacksInfoPacket;
import com.nukkitx.protocol.bedrock.packet.ServerToClientHandshakePacket;
import org.geysermc.connector.GeyserConnector;
import org.geysermc.connector.event.EventResult;
import org.geysermc.connector.event.events.packet.UpstreamPacketReceiveEvent;
import org.geysermc.connector.event.events.packet.upstream.LoginPacketReceive;
import org.geysermc.connector.network.BedrockProtocol;
import org.geysermc.connector.network.UpstreamPacketHandler;
import org.geysermc.connector.network.session.GeyserSession;
import org.geysermc.connector.network.session.auth.AuthData;
import org.geysermc.connector.network.session.auth.BedrockClientData;
import org.geysermc.connector.utils.LanguageUtils;
import org.geysermc.connector.utils.ResourcePack;
import org.geysermc.connector.utils.ResourcePackManifest;

import java.util.UUID;

public class GeyserUpstreamPacketHandler extends UpstreamPacketHandler {
private final BedrockReversionSession serverSession;

public GeyserUpstreamPacketHandler(BedrockReversionSession serverSession, GeyserConnector connector, GeyserSession session) {
super(connector, session);

this.serverSession = serverSession;
}

@Override
public boolean handle(LoginPacket loginPacket) {
EventResult<LoginPacketReceive> result = connector.getEventManager().triggerEvent(UpstreamPacketReceiveEvent.of(session, loginPacket));
if (result.isCancelled()) {
return true;
}

loginPacket = result.getEvent().getPacket();

// Check that we support the codec
BedrockPacketCodec packetCodec = BedrockProtocol.getBedrockCodec(loginPacket.getProtocolVersion());
if (packetCodec == null) {
if (loginPacket.getProtocolVersion() > BedrockProtocol.DEFAULT_BEDROCK_CODEC.getProtocolVersion()) {
// Too early to determine session locale
session.getConnector().getLogger().info(LanguageUtils.getLocaleStringLog("geyser.network.outdated.server", BedrockProtocol.DEFAULT_BEDROCK_CODEC.getMinecraftVersion()));
session.disconnect(LanguageUtils.getLocaleStringLog("geyser.network.outdated.server", BedrockProtocol.DEFAULT_BEDROCK_CODEC.getMinecraftVersion()));
return true;
} else if (loginPacket.getProtocolVersion() < BedrockProtocol.DEFAULT_BEDROCK_CODEC.getProtocolVersion()) {
session.getConnector().getLogger().info(LanguageUtils.getLocaleStringLog("geyser.network.outdated.client", BedrockProtocol.DEFAULT_BEDROCK_CODEC.getMinecraftVersion()));
session.disconnect(LanguageUtils.getLocaleStringLog("geyser.network.outdated.client", BedrockProtocol.DEFAULT_BEDROCK_CODEC.getMinecraftVersion()));
return true;
}
}

// Provide some debug about our translation chain
Translator translator = serverSession.getTranslator();
if (translator != null) {
GeyserReversionExtension.getInstance().getLogger().debug("Translator Chain:");
while (translator != null) {
GeyserReversionExtension.getInstance().getLogger().debug(" " + translator);
translator = translator.getDownstreamTranslator();
}
}

// Encrypt Connection
serverSession.enableEncryption(serverSession.getLoginData().getEncryptionKey());

try {
ServerToClientHandshakePacket packet = new ServerToClientHandshakePacket();
packet.setJwt(serverSession.getLoginData().getHandshakeJwt().serialize());
session.sendUpstreamPacketImmediately(packet);
} catch (LoginException e) {
session.disconnect("You are not able to connect. Please make sure your account is authorized to connect or contact the server administrator.");
session.getConnector().getLogger().error("Failed to encrypt connection: " + e.getMessage());
return true;
}

// Setup Session
JsonNode extraData = serverSession.getLoginData().getPayload().get("extraData");
session.setAuthenticationData(new AuthData(
extraData.get("displayName").asText(),
UUID.fromString(extraData.get("identity").asText()),
extraData.get("XUID").asText().isEmpty() ? String.valueOf(UUID.fromString(extraData.get("identity").asText()).hashCode()) : extraData.get("XUID").asText()
));

serverSession.getLoginData();
session.setClientData(LoginData.JSON_MAPPER.convertValue(serverSession.getLoginData().getClientData(), BedrockClientData.class));

PlayStatusPacket playStatus = new PlayStatusPacket();
playStatus.setStatus(PlayStatusPacket.Status.LOGIN_SUCCESS);
session.sendUpstreamPacket(playStatus);

ResourcePacksInfoPacket resourcePacksInfo = new ResourcePacksInfoPacket();
for (ResourcePack resourcePack : ResourcePack.PACKS.values()) {
ResourcePackManifest.Header header = resourcePack.getManifest().getHeader();
resourcePacksInfo.getResourcePackInfos().add(new ResourcePacksInfoPacket.Entry(header.getUuid().toString(), header.getVersionString(), resourcePack.getFile().length(), "", "", "", false));
}
resourcePacksInfo.setForcedToAccept(GeyserConnector.getInstance().getConfig().isForceResourcePacks());
session.sendUpstreamPacket(resourcePacksInfo);
return true;
}
}

0 comments on commit a5050c9

Please sign in to comment.