Mirror/Grasscutter
1
0
Fork 0
mirror of https://github.com/Grasscutters/Grasscutter.git synced 2025-12-19 10:24:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'development' into unstable

Browse Source 
# Conflicts:
#	src/main/java/emu/grasscutter/server/http/dispatch/RegionHandler.java
#	src/main/java/emu/grasscutter/utils/Crypto.java
This commit is contained in:
KingRainbow44
2023-04-10 22:11:51 -04:00
parent a96506ccf9 3c60f792ca
commit 559261963d
2 changed files with 447 additions and 407 deletions
Download Patch File Download Diff File Expand all files Collapse all files

660
src/main/java/emu/grasscutter/server/http/dispatch/RegionHandler.java
View File

@@ -1,330 +1,330 @@
package emu.grasscutter.server.http.dispatch; 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.Grasscutter; import emu.grasscutter.GameConstants;
import emu.grasscutter.Grasscutter.ServerRunMode; import emu.grasscutter.Grasscutter;
import emu.grasscutter.net.proto.QueryCurrRegionHttpRspOuterClass.QueryCurrRegionHttpRsp; import emu.grasscutter.Grasscutter.ServerRunMode;
import emu.grasscutter.net.proto.QueryRegionListHttpRspOuterClass.QueryRegionListHttpRsp; import emu.grasscutter.net.proto.QueryCurrRegionHttpRspOuterClass.QueryCurrRegionHttpRsp;
import emu.grasscutter.net.proto.RegionInfoOuterClass.RegionInfo; import emu.grasscutter.net.proto.QueryRegionListHttpRspOuterClass.QueryRegionListHttpRsp;
import emu.grasscutter.net.proto.RegionSimpleInfoOuterClass.RegionSimpleInfo; import emu.grasscutter.net.proto.RegionInfoOuterClass.RegionInfo;
import emu.grasscutter.server.event.dispatch.QueryAllRegionsEvent; import emu.grasscutter.net.proto.RegionSimpleInfoOuterClass.RegionSimpleInfo;
import emu.grasscutter.server.event.dispatch.QueryCurrentRegionEvent; import emu.grasscutter.net.proto.RetcodeOuterClass.Retcode;
import emu.grasscutter.server.http.Router; import emu.grasscutter.net.proto.StopServerInfoOuterClass.StopServerInfo;
import emu.grasscutter.server.http.objects.QueryCurRegionRspJson; import emu.grasscutter.server.event.dispatch.QueryAllRegionsEvent;
import emu.grasscutter.utils.Crypto; import emu.grasscutter.server.event.dispatch.QueryCurrentRegionEvent;
import emu.grasscutter.utils.Utils; import emu.grasscutter.server.http.Router;
import io.javalin.Javalin; import emu.grasscutter.server.http.objects.QueryCurRegionRspJson;
import io.javalin.http.Context; import emu.grasscutter.utils.Crypto;
import java.io.ByteArrayOutputStream; import emu.grasscutter.utils.Utils;
import java.security.Signature; import io.javalin.Javalin;
import java.util.ArrayList; import io.javalin.http.Context;
import java.util.Arrays;
import java.util.List; import javax.crypto.Cipher;
import java.util.Map; import java.time.Instant;
import java.util.concurrent.ConcurrentHashMap; import java.util.ArrayList;
import java.util.regex.Pattern; import java.util.List;
import javax.crypto.Cipher; import java.util.Map;
import org.slf4j.Logger; import java.util.concurrent.ConcurrentHashMap;
import java.util.regex.Pattern;
/** Handles requests related to region queries. */ import org.slf4j.Logger;
public final class RegionHandler implements Router {
private static final Map<String, RegionData> regions = new ConcurrentHashMap<>(); /** Handles requests related to region queries. */
private static String regionListResponse; public final class RegionHandler implements Router {
private static String regionListResponsecn; private static final Map<String, RegionData> regions = new ConcurrentHashMap<>();
private static String regionListResponse;
public RegionHandler() { private static String regionListResponsecn;
try { // Read & initialize region data.
this.initialize(); public RegionHandler() {
} catch (Exception exception) { try { // Read & initialize region data.
Grasscutter.getLogger().error("Failed to initialize region data.", exception); this.initialize();
} } catch (Exception exception) {
} Grasscutter.getLogger().error("Failed to initialize region data.", exception);
}
/** }
* Handle query region list request.
* /**
* @param ctx The context object for handling the request. * Handle query region list request.
* @route /query_region_list *
*/ * @param ctx The context object for handling the request.
private static void queryRegionList(Context ctx) { * @route /query_region_list
// Get logger and query parameters. */
Logger logger = Grasscutter.getLogger(); private static void queryRegionList(Context ctx) {
if (ctx.queryParamMap().containsKey("version") && ctx.queryParamMap().containsKey("platform")) { // Get logger and query parameters.
String versionName = ctx.queryParam("version"); Logger logger = Grasscutter.getLogger();
String versionCode = versionName.replaceAll("[/.0-9]*", ""); if (ctx.queryParamMap().containsKey("version") && ctx.queryParamMap().containsKey("platform")) {
String platformName = ctx.queryParam("platform"); String versionName = ctx.queryParam("version");
String versionCode = versionName.replaceAll("[/.0-9]*", "");
// Determine the region list to use based on the version and platform. String platformName = ctx.queryParam("platform");
if ("CNRELiOS".equals(versionCode)
|| "CNRELWin".equals(versionCode) // Determine the region list to use based on the version and platform.
|| "CNRELAndroid".equals(versionCode)) { if ("CNRELiOS".equals(versionCode)
// Use the CN region list. || "CNRELWin".equals(versionCode)
QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponsecn); || "CNRELAndroid".equals(versionCode)) {
event.call(); // Use the CN region list.
logger.debug("Connect to Chinese version"); QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponsecn);
event.call();
// Respond with the event result. logger.debug("Connect to Chinese version");
ctx.result(event.getRegionList());
} else if ("OSRELiOS".equals(versionCode) // Respond with the event result.
|| "OSRELWin".equals(versionCode) ctx.result(event.getRegionList());
|| "OSRELAndroid".equals(versionCode)) { } else if ("OSRELiOS".equals(versionCode)
// Use the OS region list. || "OSRELWin".equals(versionCode)
QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse); || "OSRELAndroid".equals(versionCode)) {
event.call(); // Use the OS region list.
logger.debug("Connect to global version"); QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse);
event.call();
// Respond with the event result. logger.debug("Connect to global version");
ctx.result(event.getRegionList());
} else { // Respond with the event result.
/* ctx.result(event.getRegionList());
* String regionListResponse = "CP///////////wE="; } else {
* QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse); /*
* event.call(); * String regionListResponse = "CP///////////wE=";
* ctx.result(event.getRegionList()); * QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse);
* return; * event.call();
*/ * ctx.result(event.getRegionList());
// Use the default region list. * return;
QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse); */
event.call(); // Use the default region list.
logger.debug("Connect to global version"); QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse);
event.call();
// Respond with the event result. logger.debug("Connect to global version");
ctx.result(event.getRegionList());
} // Respond with the event result.
} else { ctx.result(event.getRegionList());
// Use the default region list. }
QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse); } else {
event.call(); // Use the default region list.
logger.debug("Connect to global version"); QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse);
event.call();
// Respond with the event result. logger.debug("Connect to global version");
ctx.result(event.getRegionList());
} // Respond with the event result.
// Log the request to the console. ctx.result(event.getRegionList());
Grasscutter.getLogger() }
.info(String.format("[Dispatch] Client %s request: query_region_list", ctx.ip())); // Log the request to the console.
} Grasscutter.getLogger()
.info(String.format("[Dispatch] Client %s request: query_region_list", ctx.ip()));
/** }
* @route /query_cur_region/{region}
*/ /**
private static void queryCurrentRegion(Context ctx) { * @route /query_cur_region/{region}
// Get region to query. */
String regionName = ctx.pathParam("region"); private static void queryCurrentRegion(Context ctx) {
String versionName = ctx.queryParam("version"); // Get region to query.
var region = regions.get(regionName); String regionName = ctx.pathParam("region");
String versionName = ctx.queryParam("version");
// Get region data. var region = regions.get(regionName);
String regionData = "CAESGE5vdCBGb3VuZCB2ZXJzaW9uIGNvbmZpZw==";
if (ctx.queryParamMap().values().size() > 0) { // Get region data.
if (region != null) regionData = region.getBase64(); String regionData = "CAESGE5vdCBGb3VuZCB2ZXJzaW9uIGNvbmZpZw==";
} if (ctx.queryParamMap().values().size() > 0) {
if (region != null) regionData = region.getBase64();
String[] versionCode = }
versionName.replaceAll(Pattern.compile("[a-zA-Z]").pattern(), "").split("\\.");
int versionMajor = Integer.parseInt(versionCode[0]); String[] versionCode =
int versionMinor = Integer.parseInt(versionCode[1]); versionName.replaceAll(Pattern.compile("[a-zA-Z]").pattern(), "").split("\\.");
int versionFix = Integer.parseInt(versionCode[2]); String clientVersion = versionName.replaceAll(Pattern.compile("[a-zA-Z]").pattern(), "");
int versionMajor = Integer.parseInt(versionCode[0]);
if (versionMajor >= 3 int versionMinor = Integer.parseInt(versionCode[1]);
|| (versionMajor == 2 && versionMinor == 7 && versionFix >= 50) int versionFix = Integer.parseInt(versionCode[2]);
|| (versionMajor == 2 && versionMinor == 8)) {
try { if (versionMajor >= 3
QueryCurrentRegionEvent event = new QueryCurrentRegionEvent(regionData); || (versionMajor == 2 && versionMinor == 7 && versionFix >= 50)
event.call(); || (versionMajor == 2 && versionMinor == 8)) {
try {
if (ctx.queryParam("dispatchSeed") == null) { QueryCurrentRegionEvent event = new QueryCurrentRegionEvent(regionData);
// More love for UA Patch players event.call();
var rsp = new QueryCurRegionRspJson();
String key_id = ctx.queryParam("key_id");
rsp.content = event.getRegionInfo();
rsp.sign = "TW9yZSBsb3ZlIGZvciBVQSBQYXRjaCBwbGF5ZXJz"; if (!clientVersion.equals(GameConstants.VERSION)) { // Reject clients when there is a version mismatch
ctx.json(rsp); boolean updateClient = GameConstants.VERSION.compareTo(clientVersion) > 0;
return;
} QueryCurrRegionHttpRsp rsp = QueryCurrRegionHttpRsp.newBuilder()
.setRetcode(Retcode.RET_STOP_SERVER_VALUE)
String key_id = ctx.queryParam("key_id"); .setMsg("Connection Failed!")
.setRegionInfo(RegionInfo.newBuilder())
if (key_id == null) throw new Exception("Key ID was not set"); .setStopServer(StopServerInfo.newBuilder()
.setUrl("https://discord.gg/grasscutters")
Cipher cipher = Cipher.getInstance("RSA/ECB/PKCS1Padding"); .setStopBeginTime((int) Instant.now().getEpochSecond())
cipher.init(Cipher.ENCRYPT_MODE, Crypto.EncryptionKeys.get(Integer.valueOf(key_id))); .setStopEndTime((int) Instant.now().getEpochSecond()*2)
var regionInfo = Utils.base64Decode(event.getRegionInfo()); .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())
// Encrypt regionInfo in chunks .buildPartial();
ByteArrayOutputStream encryptedRegionInfoStream = new ByteArrayOutputStream();
Grasscutter.getLogger().info(String.format("Connection denied for %s due to %s", ctx.ip(), updateClient ? "outdated client!" : "outdated server!"));
// Thank you so much GH Copilot
int chunkSize = 256 - 11; ctx.json(Crypto.encryptAndSignRegionData(rsp.toByteArray(), key_id));
int regionInfoLength = regionInfo.length; return;
int numChunks = (int) Math.ceil(regionInfoLength / (double) chunkSize); }
for (int i = 0; i < numChunks; i++) { if (ctx.queryParam("dispatchSeed") == null) {
byte[] chunk = // More love for UA Patch players
Arrays.copyOfRange( var rsp = new QueryCurRegionRspJson();
regionInfo, i * chunkSize, Math.min((i + 1) * chunkSize, regionInfoLength));
byte[] encryptedChunk = cipher.doFinal(chunk); rsp.content = event.getRegionInfo();
encryptedRegionInfoStream.write(encryptedChunk); rsp.sign = "TW9yZSBsb3ZlIGZvciBVQSBQYXRjaCBwbGF5ZXJz";
}
ctx.json(rsp);
Signature privateSignature = Signature.getInstance("SHA256withRSA"); return;
privateSignature.initSign(Crypto.CUR_SIGNING_KEY); }
privateSignature.update(regionInfo);
if (key_id == null) throw new Exception("Key ID was not set");
var rsp = new QueryCurRegionRspJson();
Cipher cipher = Cipher.getInstance("RSA/ECB/PKCS1Padding");
rsp.content = Utils.base64Encode(encryptedRegionInfoStream.toByteArray()); cipher.init(Cipher.ENCRYPT_MODE, Crypto.EncryptionKeys.get(Integer.valueOf(key_id)));
rsp.sign = Utils.base64Encode(privateSignature.sign()); var regionInfo = Utils.base64Decode(event.getRegionInfo());
ctx.json(rsp); ctx.json(Crypto.encryptAndSignRegionData(regionInfo, key_id));
} 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);
} }
} else { } else {
// Invoke event. // Invoke event.
QueryCurrentRegionEvent event = new QueryCurrentRegionEvent(regionData); QueryCurrentRegionEvent event = new QueryCurrentRegionEvent(regionData);
event.call(); event.call();
// Respond with event result. // Respond with event result.
ctx.result(event.getRegionInfo()); ctx.result(event.getRegionInfo());
} }
// Log to console. // Log to console.
Grasscutter.getLogger() Grasscutter.getLogger()
.info(String.format("Client %s request: query_cur_region/%s", ctx.ip(), regionName)); .info(String.format("Client %s request: query_cur_region/%s", ctx.ip(), regionName));
} }
/** /**
* Gets the current region query. * Gets the current region query.
* *
* @return A {@link QueryCurrRegionHttpRsp} object. * @return A {@link QueryCurrRegionHttpRsp} object.
*/ */
public static QueryCurrRegionHttpRsp getCurrentRegion() { public static QueryCurrRegionHttpRsp getCurrentRegion() {
return SERVER.runMode == ServerRunMode.HYBRID ? regions.get("os_usa").getRegionQuery() : null; return SERVER.runMode == ServerRunMode.HYBRID ? regions.get("os_usa").getRegionQuery() : null;
} }
/** Configures region data according to configuration. */ /** Configures region data according to configuration. */
private void initialize() { private void initialize() {
String dispatchDomain = String dispatchDomain =
"http" "http"
+ (HTTP_ENCRYPTION.useInRouting ? "s" : "") + (HTTP_ENCRYPTION.useInRouting ? "s" : "")
+ "://" + "://"
+ lr(HTTP_INFO.accessAddress, HTTP_INFO.bindAddress) + lr(HTTP_INFO.accessAddress, HTTP_INFO.bindAddress)
+ ":" + ":"
+ lr(HTTP_INFO.accessPort, HTTP_INFO.bindPort); + lr(HTTP_INFO.accessPort, HTTP_INFO.bindPort);
// Create regions. // Create regions.
List<RegionSimpleInfo> servers = new ArrayList<>(); List<RegionSimpleInfo> servers = new ArrayList<>();
List<String> usedNames = new ArrayList<>(); // List to check for potential naming conflicts. List<String> usedNames = new ArrayList<>(); // List to check for potential naming conflicts.
var configuredRegions = new ArrayList<>(List.of(DISPATCH_INFO.regions)); var configuredRegions = new ArrayList<>(List.of(DISPATCH_INFO.regions));
if (SERVER.runMode != ServerRunMode.HYBRID && configuredRegions.size() == 0) { if (SERVER.runMode != ServerRunMode.HYBRID && configuredRegions.size() == 0) {
Grasscutter.getLogger() Grasscutter.getLogger()
.error( .error(
"[Dispatch] There are no game servers available. Exiting due to unplayable state."); "[Dispatch] There are no game servers available. Exiting due to unplayable state.");
System.exit(1); System.exit(1);
} else if (configuredRegions.size() == 0) } else if (configuredRegions.size() == 0)
configuredRegions.add( configuredRegions.add(
new Region( new Region(
"os_usa", "os_usa",
DISPATCH_INFO.defaultName, DISPATCH_INFO.defaultName,
lr(GAME_INFO.accessAddress, GAME_INFO.bindAddress), lr(GAME_INFO.accessAddress, GAME_INFO.bindAddress),
lr(GAME_INFO.accessPort, GAME_INFO.bindPort))); lr(GAME_INFO.accessPort, GAME_INFO.bindPort)));
configuredRegions.forEach( configuredRegions.forEach(
region -> { region -> {
if (usedNames.contains(region.Name)) { if (usedNames.contains(region.Name)) {
Grasscutter.getLogger().error("Region name already in use."); Grasscutter.getLogger().error("Region name already in use.");
return; return;
} }
// Create a region identifier. // Create a region identifier.
var identifier = var identifier =
RegionSimpleInfo.newBuilder() RegionSimpleInfo.newBuilder()
.setName(region.Name) .setName(region.Name)
.setTitle(region.Title) .setTitle(region.Title)
.setType("DEV_PUBLIC") .setType("DEV_PUBLIC")
.setDispatchUrl(dispatchDomain + "/query_cur_region/" + region.Name) .setDispatchUrl(dispatchDomain + "/query_cur_region/" + region.Name)
.build(); .build();
usedNames.add(region.Name); usedNames.add(region.Name);
servers.add(identifier); servers.add(identifier);
// Create a region info object. // Create a region info object.
var regionInfo = var regionInfo =
RegionInfo.newBuilder() RegionInfo.newBuilder()
.setGateserverIp(region.Ip) .setGateserverIp(region.Ip)
.setGateserverPort(region.Port) .setGateserverPort(region.Port)
.setSecretKey(ByteString.copyFrom(Crypto.DISPATCH_SEED)) .setSecretKey(ByteString.copyFrom(Crypto.DISPATCH_SEED))
.build(); .build();
// Create an updated region query. // Create an updated region query.
var updatedQuery = QueryCurrRegionHttpRsp.newBuilder().setRegionInfo(regionInfo).build(); var updatedQuery = QueryCurrRegionHttpRsp.newBuilder().setRegionInfo(regionInfo).build();
regions.put( regions.put(
region.Name, region.Name,
new RegionData( new RegionData(
updatedQuery, Utils.base64Encode(updatedQuery.toByteString().toByteArray()))); updatedQuery, Utils.base64Encode(updatedQuery.toByteString().toByteArray())));
}); });
// Create a config object. // Create a config object.
byte[] customConfig = byte[] customConfig =
"{\"sdkenv\":\"2\",\"checkdevice\":\"false\",\"loadPatch\":\"false\",\"showexception\":\"false\",\"regionConfig\":\"pm|fk|add\",\"downloadMode\":\"0\"}" "{\"sdkenv\":\"2\",\"checkdevice\":\"false\",\"loadPatch\":\"false\",\"showexception\":\"false\",\"regionConfig\":\"pm|fk|add\",\"downloadMode\":\"0\"}"
.getBytes(); .getBytes();
Crypto.xor(customConfig, Crypto.DISPATCH_KEY); // XOR the config with the key. Crypto.xor(customConfig, Crypto.DISPATCH_KEY); // XOR the config with the key.
// Create an updated region list. // Create an updated region list.
QueryRegionListHttpRsp updatedRegionList = QueryRegionListHttpRsp updatedRegionList =
QueryRegionListHttpRsp.newBuilder() QueryRegionListHttpRsp.newBuilder()
.addAllRegionList(servers) .addAllRegionList(servers)
.setClientSecretKey(ByteString.copyFrom(Crypto.DISPATCH_SEED)) .setClientSecretKey(ByteString.copyFrom(Crypto.DISPATCH_SEED))
.setClientCustomConfigEncrypted(ByteString.copyFrom(customConfig)) .setClientCustomConfigEncrypted(ByteString.copyFrom(customConfig))
.setEnableLoginPc(true) .setEnableLoginPc(true)
.build(); .build();
// Set the region list response. // Set the region list response.
regionListResponse = Utils.base64Encode(updatedRegionList.toByteString().toByteArray()); regionListResponse = Utils.base64Encode(updatedRegionList.toByteString().toByteArray());
// CN // CN
// Create a config object. // Create a config object.
byte[] customConfigcn = byte[] customConfigcn =
"{\"sdkenv\":\"0\",\"checkdevice\":\"true\",\"loadPatch\":\"false\",\"showexception\":\"false\",\"regionConfig\":\"pm|fk|add\",\"downloadMode\":\"0\"}" "{\"sdkenv\":\"0\",\"checkdevice\":\"true\",\"loadPatch\":\"false\",\"showexception\":\"false\",\"regionConfig\":\"pm|fk|add\",\"downloadMode\":\"0\"}"
.getBytes(); .getBytes();
Crypto.xor(customConfigcn, Crypto.DISPATCH_KEY); // XOR the config with the key. Crypto.xor(customConfigcn, Crypto.DISPATCH_KEY); // XOR the config with the key.
// Create an updated region list. // Create an updated region list.
QueryRegionListHttpRsp updatedRegionListcn = QueryRegionListHttpRsp updatedRegionListcn =
QueryRegionListHttpRsp.newBuilder() QueryRegionListHttpRsp.newBuilder()
.addAllRegionList(servers) .addAllRegionList(servers)
.setClientSecretKey(ByteString.copyFrom(Crypto.DISPATCH_SEED)) .setClientSecretKey(ByteString.copyFrom(Crypto.DISPATCH_SEED))
.setClientCustomConfigEncrypted(ByteString.copyFrom(customConfigcn)) .setClientCustomConfigEncrypted(ByteString.copyFrom(customConfigcn))
.setEnableLoginPc(true) .setEnableLoginPc(true)
.build(); .build();
// Set the region list response. // Set the region list response.
regionListResponsecn = Utils.base64Encode(updatedRegionListcn.toByteString().toByteArray()); regionListResponsecn = Utils.base64Encode(updatedRegionListcn.toByteString().toByteArray());
} }
@Override @Override
public void applyRoutes(Javalin javalin) { public void applyRoutes(Javalin javalin) {
javalin.get("/query_region_list", RegionHandler::queryRegionList); javalin.get("/query_region_list", RegionHandler::queryRegionList);
javalin.get("/query_cur_region/{region}", RegionHandler::queryCurrentRegion); javalin.get("/query_cur_region/{region}", RegionHandler::queryCurrentRegion);
} }
/** Region data container. */ /** Region data container. */
public static class RegionData { public static class RegionData {
private final QueryCurrRegionHttpRsp regionQuery; private final QueryCurrRegionHttpRsp regionQuery;
private final String base64; private final String base64;
public RegionData(QueryCurrRegionHttpRsp prq, String b64) { public RegionData(QueryCurrRegionHttpRsp prq, String b64) {
this.regionQuery = prq; this.regionQuery = prq;
this.base64 = b64; this.base64 = b64;
} }
public QueryCurrRegionHttpRsp getRegionQuery() { public QueryCurrRegionHttpRsp getRegionQuery() {
return this.regionQuery; return this.regionQuery;
} }
public String getBase64() { public String getBase64() {
return this.base64; return this.base64;
} }
} }
} }

194
src/main/java/emu/grasscutter/utils/Crypto.java
View File

@@ -1,77 +1,117 @@
package emu.grasscutter.utils; package emu.grasscutter.utils;
import emu.grasscutter.Grasscutter; import emu.grasscutter.Grasscutter;
import java.nio.file.Path; import emu.grasscutter.server.http.objects.QueryCurRegionRspJson;
import java.security.KeyFactory;
import java.security.PrivateKey; import java.io.ByteArrayOutputStream;
import java.security.PublicKey; import java.nio.file.Path;
import java.security.SecureRandom; import java.security.KeyFactory;
import java.security.spec.PKCS8EncodedKeySpec; import java.security.PrivateKey;
import java.security.spec.X509EncodedKeySpec; import java.security.PublicKey;
import java.util.HashMap; import java.security.SecureRandom;
import java.util.Map; import java.security.Signature;
import java.util.regex.Pattern; import java.security.spec.PKCS8EncodedKeySpec;
import java.security.spec.X509EncodedKeySpec;
public final class Crypto { import java.util.Arrays;
private static final SecureRandom secureRandom = new SecureRandom(); import java.util.Map;
import java.util.HashMap;
public static byte[] DISPATCH_KEY; import java.util.regex.Pattern;
public static byte[] DISPATCH_SEED; import javax.crypto.Cipher;
public static byte[] ENCRYPT_KEY; public final class Crypto {
public static long ENCRYPT_SEED = Long.parseUnsignedLong("11468049314633205968"); private static final SecureRandom secureRandom = new SecureRandom();
public static byte[] ENCRYPT_SEED_BUFFER = new byte[0];
public static byte[] DISPATCH_KEY;
public static PrivateKey CUR_SIGNING_KEY; public static byte[] DISPATCH_SEED;
public static Map<Integer, PublicKey> EncryptionKeys = new HashMap<>(); public static byte[] ENCRYPT_KEY;
public static long ENCRYPT_SEED = Long.parseUnsignedLong("11468049314633205968");
public static void loadKeys() { public static byte[] ENCRYPT_SEED_BUFFER = new byte[0];
DISPATCH_KEY = FileUtils.readResource("/keys/dispatchKey.bin");
DISPATCH_SEED = FileUtils.readResource("/keys/dispatchSeed.bin"); public static PrivateKey CUR_SIGNING_KEY;
ENCRYPT_KEY = FileUtils.readResource("/keys/secretKey.bin"); public static Map<Integer, PublicKey> EncryptionKeys = new HashMap<>();
ENCRYPT_SEED_BUFFER = FileUtils.readResource("/keys/secretKeyBuffer.bin");
public static void loadKeys() {
try { DISPATCH_KEY = FileUtils.readResource("/keys/dispatchKey.bin");
CUR_SIGNING_KEY = DISPATCH_SEED = FileUtils.readResource("/keys/dispatchSeed.bin");
KeyFactory.getInstance("RSA")
.generatePrivate( ENCRYPT_KEY = FileUtils.readResource("/keys/secretKey.bin");
new PKCS8EncodedKeySpec(FileUtils.readResource("/keys/SigningKey.der"))); ENCRYPT_SEED_BUFFER = FileUtils.readResource("/keys/secretKeyBuffer.bin");
Pattern pattern = Pattern.compile("([0-9]*)_Pub\\.der"); try {
for (Path path : FileUtils.getPathsFromResource("/keys/game_keys")) { CUR_SIGNING_KEY =
if (path.toString().endsWith("_Pub.der")) { KeyFactory.getInstance("RSA")
.generatePrivate(
var m = pattern.matcher(path.getFileName().toString()); new PKCS8EncodedKeySpec(FileUtils.readResource("/keys/SigningKey.der")));
if (m.matches()) { Pattern pattern = Pattern.compile("([0-9]*)_Pub\\.der");
var key = for (Path path : FileUtils.getPathsFromResource("/keys/game_keys")) {
KeyFactory.getInstance("RSA") if (path.toString().endsWith("_Pub.der")) {
.generatePublic(new X509EncodedKeySpec(FileUtils.read(path)));
var m = pattern.matcher(path.getFileName().toString());
EncryptionKeys.put(Integer.valueOf(m.group(1)), key);
} if (m.matches()) {
} var key =
} KeyFactory.getInstance("RSA")
} catch (Exception e) { .generatePublic(new X509EncodedKeySpec(FileUtils.read(path)));
Grasscutter.getLogger().error("An error occurred while loading keys.", e);
} EncryptionKeys.put(Integer.valueOf(m.group(1)), key);
} }
}
public static void xor(byte[] packet, byte[] key) { }
try { } catch (Exception e) {
for (int i = 0; i < packet.length; i++) { Grasscutter.getLogger().error("An error occurred while loading keys.", e);
packet[i] ^= key[i % key.length]; }
} }
} catch (Exception e) {
Grasscutter.getLogger().error("Crypto error.", e); public static void xor(byte[] packet, byte[] key) {
} try {
} for (int i = 0; i < packet.length; i++) {
packet[i] ^= key[i % key.length];
public static byte[] createSessionKey(int length) { }
byte[] bytes = new byte[length]; } catch (Exception e) {
secureRandom.nextBytes(bytes); Grasscutter.getLogger().error("Crypto error.", e);
return bytes; }
} }
}
public static byte[] createSessionKey(int length) {
byte[] bytes = new byte[length];
secureRandom.nextBytes(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;
}
}