mirror of
https://github.com/Grasscutters/Grasscutter.git
synced 2025-12-19 10:24:47 +01:00
Merge branch 'development' into unstable
# Conflicts: # src/main/java/emu/grasscutter/server/http/dispatch/RegionHandler.java # src/main/java/emu/grasscutter/utils/Crypto.java
This commit is contained in:
@@ -3,12 +3,15 @@ package emu.grasscutter.server.http.dispatch;
|
|||||||
import static emu.grasscutter.config.Configuration.*;
|
import static emu.grasscutter.config.Configuration.*;
|
||||||
|
|
||||||
import com.google.protobuf.ByteString;
|
import com.google.protobuf.ByteString;
|
||||||
|
import emu.grasscutter.GameConstants;
|
||||||
import emu.grasscutter.Grasscutter;
|
import emu.grasscutter.Grasscutter;
|
||||||
import emu.grasscutter.Grasscutter.ServerRunMode;
|
import emu.grasscutter.Grasscutter.ServerRunMode;
|
||||||
import emu.grasscutter.net.proto.QueryCurrRegionHttpRspOuterClass.QueryCurrRegionHttpRsp;
|
import emu.grasscutter.net.proto.QueryCurrRegionHttpRspOuterClass.QueryCurrRegionHttpRsp;
|
||||||
import emu.grasscutter.net.proto.QueryRegionListHttpRspOuterClass.QueryRegionListHttpRsp;
|
import emu.grasscutter.net.proto.QueryRegionListHttpRspOuterClass.QueryRegionListHttpRsp;
|
||||||
import emu.grasscutter.net.proto.RegionInfoOuterClass.RegionInfo;
|
import emu.grasscutter.net.proto.RegionInfoOuterClass.RegionInfo;
|
||||||
import emu.grasscutter.net.proto.RegionSimpleInfoOuterClass.RegionSimpleInfo;
|
import emu.grasscutter.net.proto.RegionSimpleInfoOuterClass.RegionSimpleInfo;
|
||||||
|
import emu.grasscutter.net.proto.RetcodeOuterClass.Retcode;
|
||||||
|
import emu.grasscutter.net.proto.StopServerInfoOuterClass.StopServerInfo;
|
||||||
import emu.grasscutter.server.event.dispatch.QueryAllRegionsEvent;
|
import emu.grasscutter.server.event.dispatch.QueryAllRegionsEvent;
|
||||||
import emu.grasscutter.server.event.dispatch.QueryCurrentRegionEvent;
|
import emu.grasscutter.server.event.dispatch.QueryCurrentRegionEvent;
|
||||||
import emu.grasscutter.server.http.Router;
|
import emu.grasscutter.server.http.Router;
|
||||||
@@ -17,15 +20,14 @@ import emu.grasscutter.utils.Crypto;
|
|||||||
import emu.grasscutter.utils.Utils;
|
import emu.grasscutter.utils.Utils;
|
||||||
import io.javalin.Javalin;
|
import io.javalin.Javalin;
|
||||||
import io.javalin.http.Context;
|
import io.javalin.http.Context;
|
||||||
import java.io.ByteArrayOutputStream;
|
|
||||||
import java.security.Signature;
|
import javax.crypto.Cipher;
|
||||||
|
import java.time.Instant;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
import java.util.regex.Pattern;
|
import java.util.regex.Pattern;
|
||||||
import javax.crypto.Cipher;
|
|
||||||
import org.slf4j.Logger;
|
import org.slf4j.Logger;
|
||||||
|
|
||||||
/** Handles requests related to region queries. */
|
/** Handles requests related to region queries. */
|
||||||
@@ -124,6 +126,7 @@ public final class RegionHandler implements Router {
|
|||||||
|
|
||||||
String[] versionCode =
|
String[] versionCode =
|
||||||
versionName.replaceAll(Pattern.compile("[a-zA-Z]").pattern(), "").split("\\.");
|
versionName.replaceAll(Pattern.compile("[a-zA-Z]").pattern(), "").split("\\.");
|
||||||
|
String clientVersion = versionName.replaceAll(Pattern.compile("[a-zA-Z]").pattern(), "");
|
||||||
int versionMajor = Integer.parseInt(versionCode[0]);
|
int versionMajor = Integer.parseInt(versionCode[0]);
|
||||||
int versionMinor = Integer.parseInt(versionCode[1]);
|
int versionMinor = Integer.parseInt(versionCode[1]);
|
||||||
int versionFix = Integer.parseInt(versionCode[2]);
|
int versionFix = Integer.parseInt(versionCode[2]);
|
||||||
@@ -135,6 +138,30 @@ public final class RegionHandler implements Router {
|
|||||||
QueryCurrentRegionEvent event = new QueryCurrentRegionEvent(regionData);
|
QueryCurrentRegionEvent event = new QueryCurrentRegionEvent(regionData);
|
||||||
event.call();
|
event.call();
|
||||||
|
|
||||||
|
String key_id = ctx.queryParam("key_id");
|
||||||
|
|
||||||
|
if (!clientVersion.equals(GameConstants.VERSION)) { // Reject clients when there is a version mismatch
|
||||||
|
|
||||||
|
boolean updateClient = GameConstants.VERSION.compareTo(clientVersion) > 0;
|
||||||
|
|
||||||
|
QueryCurrRegionHttpRsp rsp = QueryCurrRegionHttpRsp.newBuilder()
|
||||||
|
.setRetcode(Retcode.RET_STOP_SERVER_VALUE)
|
||||||
|
.setMsg("Connection Failed!")
|
||||||
|
.setRegionInfo(RegionInfo.newBuilder())
|
||||||
|
.setStopServer(StopServerInfo.newBuilder()
|
||||||
|
.setUrl("https://discord.gg/grasscutters")
|
||||||
|
.setStopBeginTime((int) Instant.now().getEpochSecond())
|
||||||
|
.setStopEndTime((int) Instant.now().getEpochSecond()*2)
|
||||||
|
.setContentMsg(updateClient ? "\nVersion mismatch outdated client! \n\nServer version: %s\nClient version: %s".formatted(GameConstants.VERSION, clientVersion) : "\nVersion mismatch outdated server! \n\nServer version: %s\nClient version: %s".formatted(GameConstants.VERSION, clientVersion))
|
||||||
|
.build())
|
||||||
|
.buildPartial();
|
||||||
|
|
||||||
|
Grasscutter.getLogger().info(String.format("Connection denied for %s due to %s", ctx.ip(), updateClient ? "outdated client!" : "outdated server!"));
|
||||||
|
|
||||||
|
ctx.json(Crypto.encryptAndSignRegionData(rsp.toByteArray(), key_id));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
if (ctx.queryParam("dispatchSeed") == null) {
|
if (ctx.queryParam("dispatchSeed") == null) {
|
||||||
// More love for UA Patch players
|
// More love for UA Patch players
|
||||||
var rsp = new QueryCurRegionRspJson();
|
var rsp = new QueryCurRegionRspJson();
|
||||||
@@ -146,40 +173,13 @@ public final class RegionHandler implements Router {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
String key_id = ctx.queryParam("key_id");
|
|
||||||
|
|
||||||
if (key_id == null) throw new Exception("Key ID was not set");
|
if (key_id == null) throw new Exception("Key ID was not set");
|
||||||
|
|
||||||
Cipher cipher = Cipher.getInstance("RSA/ECB/PKCS1Padding");
|
Cipher cipher = Cipher.getInstance("RSA/ECB/PKCS1Padding");
|
||||||
cipher.init(Cipher.ENCRYPT_MODE, Crypto.EncryptionKeys.get(Integer.valueOf(key_id)));
|
cipher.init(Cipher.ENCRYPT_MODE, Crypto.EncryptionKeys.get(Integer.valueOf(key_id)));
|
||||||
var regionInfo = Utils.base64Decode(event.getRegionInfo());
|
var regionInfo = Utils.base64Decode(event.getRegionInfo());
|
||||||
|
|
||||||
// Encrypt regionInfo in chunks
|
ctx.json(Crypto.encryptAndSignRegionData(regionInfo, key_id));
|
||||||
ByteArrayOutputStream encryptedRegionInfoStream = new ByteArrayOutputStream();
|
|
||||||
|
|
||||||
// Thank you so much GH Copilot
|
|
||||||
int chunkSize = 256 - 11;
|
|
||||||
int regionInfoLength = regionInfo.length;
|
|
||||||
int numChunks = (int) Math.ceil(regionInfoLength / (double) chunkSize);
|
|
||||||
|
|
||||||
for (int i = 0; i < numChunks; i++) {
|
|
||||||
byte[] chunk =
|
|
||||||
Arrays.copyOfRange(
|
|
||||||
regionInfo, i * chunkSize, Math.min((i + 1) * chunkSize, regionInfoLength));
|
|
||||||
byte[] encryptedChunk = cipher.doFinal(chunk);
|
|
||||||
encryptedRegionInfoStream.write(encryptedChunk);
|
|
||||||
}
|
|
||||||
|
|
||||||
Signature privateSignature = Signature.getInstance("SHA256withRSA");
|
|
||||||
privateSignature.initSign(Crypto.CUR_SIGNING_KEY);
|
|
||||||
privateSignature.update(regionInfo);
|
|
||||||
|
|
||||||
var rsp = new QueryCurRegionRspJson();
|
|
||||||
|
|
||||||
rsp.content = Utils.base64Encode(encryptedRegionInfoStream.toByteArray());
|
|
||||||
rsp.sign = Utils.base64Encode(privateSignature.sign());
|
|
||||||
|
|
||||||
ctx.json(rsp);
|
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
Grasscutter.getLogger().error("An error occurred while handling query_cur_region.", e);
|
Grasscutter.getLogger().error("An error occurred while handling query_cur_region.", e);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,16 +1,22 @@
|
|||||||
package emu.grasscutter.utils;
|
package emu.grasscutter.utils;
|
||||||
|
|
||||||
import emu.grasscutter.Grasscutter;
|
import emu.grasscutter.Grasscutter;
|
||||||
|
import emu.grasscutter.server.http.objects.QueryCurRegionRspJson;
|
||||||
|
|
||||||
|
import java.io.ByteArrayOutputStream;
|
||||||
import java.nio.file.Path;
|
import java.nio.file.Path;
|
||||||
import java.security.KeyFactory;
|
import java.security.KeyFactory;
|
||||||
import java.security.PrivateKey;
|
import java.security.PrivateKey;
|
||||||
import java.security.PublicKey;
|
import java.security.PublicKey;
|
||||||
import java.security.SecureRandom;
|
import java.security.SecureRandom;
|
||||||
|
import java.security.Signature;
|
||||||
import java.security.spec.PKCS8EncodedKeySpec;
|
import java.security.spec.PKCS8EncodedKeySpec;
|
||||||
import java.security.spec.X509EncodedKeySpec;
|
import java.security.spec.X509EncodedKeySpec;
|
||||||
import java.util.HashMap;
|
import java.util.Arrays;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
import java.util.HashMap;
|
||||||
import java.util.regex.Pattern;
|
import java.util.regex.Pattern;
|
||||||
|
import javax.crypto.Cipher;
|
||||||
|
|
||||||
public final class Crypto {
|
public final class Crypto {
|
||||||
private static final SecureRandom secureRandom = new SecureRandom();
|
private static final SecureRandom secureRandom = new SecureRandom();
|
||||||
@@ -74,4 +80,38 @@ public final class Crypto {
|
|||||||
secureRandom.nextBytes(bytes);
|
secureRandom.nextBytes(bytes);
|
||||||
return bytes;
|
return bytes;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static QueryCurRegionRspJson encryptAndSignRegionData(byte[] regionInfo, String key_id) throws Exception {
|
||||||
|
if (key_id == null) {
|
||||||
|
throw new Exception("Key ID was not set");
|
||||||
|
}
|
||||||
|
|
||||||
|
Cipher cipher = Cipher.getInstance("RSA/ECB/PKCS1Padding");
|
||||||
|
cipher.init(Cipher.ENCRYPT_MODE, EncryptionKeys.get(Integer.valueOf(key_id)));
|
||||||
|
|
||||||
|
// Encrypt regionInfo in chunks
|
||||||
|
var encryptedRegionInfoStream = new ByteArrayOutputStream();
|
||||||
|
|
||||||
|
// Thank you so much GH Copilot
|
||||||
|
int chunkSize = 256 - 11;
|
||||||
|
int regionInfoLength = regionInfo.length;
|
||||||
|
int numChunks = (int) Math.ceil(regionInfoLength / (double) chunkSize);
|
||||||
|
|
||||||
|
for (int i = 0; i < numChunks; i++) {
|
||||||
|
byte[] chunk = Arrays.copyOfRange(regionInfo, i * chunkSize,
|
||||||
|
Math.min((i + 1) * chunkSize, regionInfoLength));
|
||||||
|
byte[] encryptedChunk = cipher.doFinal(chunk);
|
||||||
|
encryptedRegionInfoStream.write(encryptedChunk);
|
||||||
|
}
|
||||||
|
|
||||||
|
Signature privateSignature = Signature.getInstance("SHA256withRSA");
|
||||||
|
privateSignature.initSign(CUR_SIGNING_KEY);
|
||||||
|
privateSignature.update(regionInfo);
|
||||||
|
|
||||||
|
var rsp = new QueryCurRegionRspJson();
|
||||||
|
|
||||||
|
rsp.content = Utils.base64Encode(encryptedRegionInfoStream.toByteArray());
|
||||||
|
rsp.sign = Utils.base64Encode(privateSignature.sign());
|
||||||
|
return rsp;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user