mirror of
https://github.com/Grasscutters/Grasscutter.git
synced 2025-05-09 05:25:55 +08:00
353 lines
15 KiB
Java
353 lines
15 KiB
Java
package emu.grasscutter.server.http.dispatch;
|
|
|
|
import static emu.grasscutter.config.Configuration.*;
|
|
|
|
import com.google.gson.*;
|
|
import com.google.protobuf.ByteString;
|
|
import emu.grasscutter.*;
|
|
import emu.grasscutter.Grasscutter.ServerRunMode;
|
|
import emu.grasscutter.net.proto.QueryCurrRegionHttpRspOuterClass.QueryCurrRegionHttpRsp;
|
|
import emu.grasscutter.net.proto.QueryRegionListHttpRspOuterClass.QueryRegionListHttpRsp;
|
|
import emu.grasscutter.net.proto.RegionInfoOuterClass.RegionInfo;
|
|
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.*;
|
|
import emu.grasscutter.server.http.Router;
|
|
import emu.grasscutter.server.http.objects.QueryCurRegionRspJson;
|
|
import emu.grasscutter.utils.*;
|
|
import io.javalin.Javalin;
|
|
import io.javalin.http.Context;
|
|
import java.time.Instant;
|
|
import java.util.*;
|
|
import java.util.concurrent.ConcurrentHashMap;
|
|
import java.util.regex.Pattern;
|
|
import org.slf4j.Logger;
|
|
|
|
/** Handles requests related to region queries. */
|
|
public final class RegionHandler implements Router {
|
|
private static final Map<String, RegionData> regions = new ConcurrentHashMap<>();
|
|
private static String regionListResponse;
|
|
private static String regionListResponseCN;
|
|
|
|
public RegionHandler() {
|
|
try { // Read & initialize region data.
|
|
this.initialize();
|
|
} catch (Exception exception) {
|
|
Grasscutter.getLogger().error("Failed to initialize region data.", exception);
|
|
}
|
|
}
|
|
|
|
/** Configures region data according to configuration. */
|
|
private void initialize() {
|
|
var dispatchDomain =
|
|
"http"
|
|
+ (HTTP_ENCRYPTION.useInRouting ? "s" : "")
|
|
+ "://"
|
|
+ lr(HTTP_INFO.accessAddress, HTTP_INFO.bindAddress)
|
|
+ ":"
|
|
+ lr(HTTP_INFO.accessPort, HTTP_INFO.bindPort);
|
|
|
|
// Create regions.
|
|
var servers = new ArrayList<RegionSimpleInfo>();
|
|
var usedNames = new ArrayList<String>(); // List to check for potential naming conflicts.
|
|
|
|
var configuredRegions = new ArrayList<>(DISPATCH_INFO.regions);
|
|
if (Grasscutter.getRunMode() != ServerRunMode.HYBRID && configuredRegions.size() == 0) {
|
|
Grasscutter.getLogger()
|
|
.error(
|
|
"[Dispatch] There are no game servers available. Exiting due to unplayable state.");
|
|
System.exit(1);
|
|
} else if (configuredRegions.size() == 0)
|
|
configuredRegions.add(
|
|
new Region(
|
|
"os_usa",
|
|
DISPATCH_INFO.defaultName,
|
|
lr(GAME_INFO.accessAddress, GAME_INFO.bindAddress),
|
|
lr(GAME_INFO.accessPort, GAME_INFO.bindPort)));
|
|
|
|
configuredRegions.forEach(
|
|
region -> {
|
|
if (usedNames.contains(region.Name)) {
|
|
Grasscutter.getLogger().error("Region name already in use.");
|
|
return;
|
|
}
|
|
|
|
// Create a region identifier.
|
|
var identifier =
|
|
RegionSimpleInfo.newBuilder()
|
|
.setName(region.Name)
|
|
.setTitle(region.Title)
|
|
.setType("DEV_PUBLIC")
|
|
.setDispatchUrl(dispatchDomain + "/query_cur_region/" + region.Name)
|
|
.build();
|
|
usedNames.add(region.Name);
|
|
servers.add(identifier);
|
|
|
|
// Create a region info object.
|
|
var regionInfo =
|
|
RegionInfo.newBuilder()
|
|
.setGateserverIp(region.Ip)
|
|
.setGateserverPort(region.Port)
|
|
.build();
|
|
// Create an updated region query.
|
|
var updatedQuery =
|
|
QueryCurrRegionHttpRsp.newBuilder()
|
|
.setRegionInfo(regionInfo)
|
|
.setClientSecretKey(ByteString.copyFrom(Crypto.DISPATCH_SEED))
|
|
.build();
|
|
regions.put(
|
|
region.Name,
|
|
new RegionData(
|
|
updatedQuery, Utils.base64Encode(updatedQuery.toByteString().toByteArray())));
|
|
});
|
|
|
|
// Determine config settings.
|
|
var hiddenIcons = new JsonArray();
|
|
hiddenIcons.add(40);
|
|
var codeSwitch = new JsonArray();
|
|
codeSwitch.add(3628);
|
|
|
|
// Create a config object.
|
|
var customConfig = new JsonObject();
|
|
customConfig.addProperty("sdkenv", "2");
|
|
customConfig.addProperty("checkdevice", "false");
|
|
customConfig.addProperty("loadPatch", "false");
|
|
customConfig.addProperty("showexception", String.valueOf(GameConstants.DEBUG));
|
|
customConfig.addProperty("regionConfig", "pm|fk|add");
|
|
customConfig.addProperty("downloadMode", "0");
|
|
customConfig.add("codeSwitch", codeSwitch);
|
|
customConfig.add("coverSwitch", hiddenIcons);
|
|
|
|
// XOR the config with the key.
|
|
var encodedConfig = JsonUtils.encode(customConfig).getBytes();
|
|
Crypto.xor(encodedConfig, Crypto.DISPATCH_KEY);
|
|
|
|
// Create an updated region list.
|
|
var updatedRegionList =
|
|
QueryRegionListHttpRsp.newBuilder()
|
|
.addAllRegionList(servers)
|
|
.setClientSecretKey(ByteString.copyFrom(Crypto.DISPATCH_SEED))
|
|
.setClientCustomConfigEncrypted(ByteString.copyFrom(encodedConfig))
|
|
.setEnableLoginPc(true)
|
|
.build();
|
|
|
|
// Set the region list response.
|
|
regionListResponse = Utils.base64Encode(updatedRegionList.toByteString().toByteArray());
|
|
|
|
// CN
|
|
// Modify the existing config option.
|
|
customConfig.addProperty("sdkenv", "0");
|
|
// XOR the config with the key.
|
|
encodedConfig = JsonUtils.encode(customConfig).getBytes();
|
|
Crypto.xor(encodedConfig, Crypto.DISPATCH_KEY);
|
|
|
|
// Create an updated region list.
|
|
var updatedRegionListCN =
|
|
QueryRegionListHttpRsp.newBuilder()
|
|
.addAllRegionList(servers)
|
|
.setClientSecretKey(ByteString.copyFrom(Crypto.DISPATCH_SEED))
|
|
.setClientCustomConfigEncrypted(ByteString.copyFrom(encodedConfig))
|
|
.setEnableLoginPc(true)
|
|
.build();
|
|
|
|
// Set the region list response.
|
|
regionListResponseCN = Utils.base64Encode(updatedRegionListCN.toByteString().toByteArray());
|
|
}
|
|
|
|
@Override
|
|
public void applyRoutes(Javalin javalin) {
|
|
javalin.get("/query_region_list", RegionHandler::queryRegionList);
|
|
javalin.get("/query_cur_region/{region}", RegionHandler::queryCurrentRegion);
|
|
}
|
|
|
|
/**
|
|
* Handle query region list request.
|
|
*
|
|
* @param ctx The context object for handling the request.
|
|
* @route /query_region_list
|
|
*/
|
|
private static void queryRegionList(Context ctx) {
|
|
// Get logger and query parameters.
|
|
Logger logger = Grasscutter.getLogger();
|
|
if (ctx.queryParamMap().containsKey("version") && ctx.queryParamMap().containsKey("platform")) {
|
|
String versionName = ctx.queryParam("version");
|
|
String versionCode = versionName.replaceAll("[/.0-9]*", "");
|
|
String platformName = ctx.queryParam("platform");
|
|
|
|
// Determine the region list to use based on the version and platform.
|
|
if ("CNRELiOS".equals(versionCode)
|
|
|| "CNRELWin".equals(versionCode)
|
|
|| "CNRELAndroid".equals(versionCode)) {
|
|
// Use the CN region list.
|
|
QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponseCN);
|
|
event.call();
|
|
|
|
// Respond with the event result.
|
|
ctx.result(event.getRegionList());
|
|
} else if ("OSRELiOS".equals(versionCode)
|
|
|| "OSRELWin".equals(versionCode)
|
|
|| "OSRELAndroid".equals(versionCode)) {
|
|
// Use the OS region list.
|
|
QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse);
|
|
event.call();
|
|
|
|
// Respond with the event result.
|
|
ctx.result(event.getRegionList());
|
|
} else {
|
|
/*
|
|
* String regionListResponse = "CP///////////wE=";
|
|
* QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse);
|
|
* event.call();
|
|
* ctx.result(event.getRegionList());
|
|
* return;
|
|
*/
|
|
// Use the default region list.
|
|
QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse);
|
|
event.call();
|
|
|
|
// Respond with the event result.
|
|
ctx.result(event.getRegionList());
|
|
}
|
|
} else {
|
|
// Use the default region list.
|
|
QueryAllRegionsEvent event = new QueryAllRegionsEvent(regionListResponse);
|
|
event.call();
|
|
|
|
// Respond with the event result.
|
|
ctx.result(event.getRegionList());
|
|
}
|
|
// Log the request to the console.
|
|
Grasscutter.getLogger()
|
|
.info(String.format("[Dispatch] Client %s request: query_region_list", Utils.address(ctx)));
|
|
}
|
|
|
|
/**
|
|
* @route /query_cur_region/{region}
|
|
*/
|
|
private static void queryCurrentRegion(Context ctx) {
|
|
// Get region to query.
|
|
String regionName = ctx.pathParam("region");
|
|
String versionName = ctx.queryParam("version");
|
|
var region = regions.get(regionName);
|
|
|
|
// Get region data.
|
|
String regionData = "CAESGE5vdCBGb3VuZCB2ZXJzaW9uIGNvbmZpZw==";
|
|
if (!ctx.queryParamMap().values().isEmpty()) {
|
|
if (region != null) regionData = region.getBase64();
|
|
}
|
|
|
|
var clientVersion = versionName.replaceAll(Pattern.compile("[a-zA-Z]").pattern(), "");
|
|
var versionCode = clientVersion.split("\\.");
|
|
var versionMajor = Integer.parseInt(versionCode[0]);
|
|
var versionMinor = Integer.parseInt(versionCode[1]);
|
|
var versionFix = Integer.parseInt(versionCode[2]);
|
|
|
|
if (versionMajor >= 3
|
|
|| (versionMajor == 2 && versionMinor == 7 && versionFix >= 50)
|
|
|| (versionMajor == 2 && versionMinor == 8)) {
|
|
try {
|
|
QueryCurrentRegionEvent event = new QueryCurrentRegionEvent(regionData);
|
|
event.call();
|
|
|
|
String key_id = ctx.queryParam("key_id");
|
|
|
|
if (versionMajor != GameConstants.VERSION_PARTS[0]
|
|
|| versionMinor != GameConstants.VERSION_PARTS[1]
|
|
// The 'fix' or 'patch' version is not checked because it is only used
|
|
// when miHoYo is desperate and fucks up big time.
|
|
) { // 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() + 1)
|
|
.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()
|
|
.debug(
|
|
String.format(
|
|
"Connection denied for %s due to %s.",
|
|
Utils.address(ctx), updateClient ? "outdated client!" : "outdated server!"));
|
|
|
|
ctx.json(Crypto.encryptAndSignRegionData(rsp.toByteArray(), key_id));
|
|
return;
|
|
}
|
|
|
|
if (ctx.queryParam("dispatchSeed") == null) {
|
|
// More love for UA Patch players
|
|
var rsp = new QueryCurRegionRspJson();
|
|
|
|
rsp.content = event.getRegionInfo();
|
|
rsp.sign = "TW9yZSBsb3ZlIGZvciBVQSBQYXRjaCBwbGF5ZXJz";
|
|
|
|
ctx.json(rsp);
|
|
return;
|
|
}
|
|
|
|
var regionInfo = Utils.base64Decode(event.getRegionInfo());
|
|
|
|
ctx.json(Crypto.encryptAndSignRegionData(regionInfo, key_id));
|
|
} catch (Exception e) {
|
|
Grasscutter.getLogger().error("An error occurred while handling query_cur_region.", e);
|
|
}
|
|
} else {
|
|
// Invoke event.
|
|
QueryCurrentRegionEvent event = new QueryCurrentRegionEvent(regionData);
|
|
event.call();
|
|
// Respond with event result.
|
|
ctx.result(event.getRegionInfo());
|
|
}
|
|
// Log to console.
|
|
Grasscutter.getLogger()
|
|
.info(
|
|
String.format(
|
|
"Client %s request: query_cur_region/%s", Utils.address(ctx), regionName));
|
|
}
|
|
|
|
/** Region data container. */
|
|
public static class RegionData {
|
|
private final QueryCurrRegionHttpRsp regionQuery;
|
|
private final String base64;
|
|
|
|
public RegionData(QueryCurrRegionHttpRsp prq, String b64) {
|
|
this.regionQuery = prq;
|
|
this.base64 = b64;
|
|
}
|
|
|
|
public QueryCurrRegionHttpRsp getRegionQuery() {
|
|
return this.regionQuery;
|
|
}
|
|
|
|
public String getBase64() {
|
|
return this.base64;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Gets the current region query.
|
|
*
|
|
* @return A {@link QueryCurrRegionHttpRsp} object.
|
|
*/
|
|
public static QueryCurrRegionHttpRsp getCurrentRegion() {
|
|
return Grasscutter.getRunMode() == ServerRunMode.HYBRID
|
|
? regions.get("os_usa").getRegionQuery()
|
|
: null;
|
|
}
|
|
}
|