forked from Hay1tsme/artemis
262 lines
9.4 KiB
Python
262 lines
9.4 KiB
Python
from starlette.requests import Request
|
|
from starlette.routing import Route
|
|
from starlette.responses import Response, PlainTextResponse, JSONResponse
|
|
import traceback
|
|
import sys
|
|
import yaml
|
|
import json
|
|
import re
|
|
import inflection
|
|
import logging, coloredlogs
|
|
from logging.handlers import TimedRotatingFileHandler
|
|
from typing import Dict, Tuple, List
|
|
from os import path
|
|
|
|
from core.config import CoreConfig
|
|
from core.title import BaseServlet
|
|
from core.utils import Utils
|
|
from .config import CxbConfig
|
|
from .const import CxbConstants
|
|
from .rev import CxbRev
|
|
from .rss1 import CxbRevSunriseS1
|
|
from .rss2 import CxbRevSunriseS2
|
|
|
|
|
|
class CxbServlet(BaseServlet):
|
|
def __init__(self, core_cfg: CoreConfig, cfg_dir: str) -> None:
|
|
self.isLeaf = True
|
|
self.cfg_dir = cfg_dir
|
|
self.core_cfg = core_cfg
|
|
self.game_cfg = CxbConfig()
|
|
if path.exists(f"{cfg_dir}/{CxbConstants.CONFIG_NAME}"):
|
|
self.game_cfg.update(
|
|
yaml.safe_load(open(f"{cfg_dir}/{CxbConstants.CONFIG_NAME}"))
|
|
)
|
|
|
|
self.logger = logging.getLogger("cxb")
|
|
if not hasattr(self.logger, "inited"):
|
|
log_fmt_str = "[%(asctime)s] CXB | %(levelname)s | %(message)s"
|
|
log_fmt = logging.Formatter(log_fmt_str)
|
|
fileHandler = TimedRotatingFileHandler(
|
|
"{0}/{1}.log".format(self.core_cfg.server.log_dir, "cxb"),
|
|
encoding="utf8",
|
|
when="d",
|
|
backupCount=10,
|
|
)
|
|
|
|
fileHandler.setFormatter(log_fmt)
|
|
|
|
consoleHandler = logging.StreamHandler()
|
|
consoleHandler.setFormatter(log_fmt)
|
|
|
|
self.logger.addHandler(fileHandler)
|
|
self.logger.addHandler(consoleHandler)
|
|
|
|
self.logger.setLevel(self.game_cfg.server.loglevel)
|
|
coloredlogs.install(
|
|
level=self.game_cfg.server.loglevel, logger=self.logger, fmt=log_fmt_str
|
|
)
|
|
self.logger.inited = True
|
|
|
|
self.versions = [
|
|
CxbRev(core_cfg, self.game_cfg),
|
|
CxbRevSunriseS1(core_cfg, self.game_cfg),
|
|
CxbRevSunriseS2(core_cfg, self.game_cfg),
|
|
]
|
|
|
|
def get_routes(self) -> List[Route]:
|
|
return [
|
|
Route("/data", self.handle_data, methods=['POST']),
|
|
Route("/action", self.handle_action, methods=['POST']),
|
|
Route("/v2/action", self.handle_action, methods=['POST']),
|
|
Route("/auth", self.handle_auth, methods=['POST']),
|
|
]
|
|
|
|
@classmethod
|
|
def is_game_enabled(cls, game_code: str, core_cfg: CoreConfig, cfg_dir: str) -> bool:
|
|
game_cfg = CxbConfig()
|
|
if path.exists(f"{cfg_dir}/{CxbConstants.CONFIG_NAME}"):
|
|
game_cfg.update(
|
|
yaml.safe_load(open(f"{cfg_dir}/{CxbConstants.CONFIG_NAME}"))
|
|
)
|
|
|
|
if not game_cfg.server.enable:
|
|
return False
|
|
|
|
return True
|
|
|
|
def get_allnet_info(self, game_code: str, game_ver: int, keychip: str) -> Tuple[str, str]:
|
|
title_port_int = Utils.get_title_port(self.core_cfg)
|
|
title_port_ssl_int = Utils.get_title_port_ssl(self.core_cfg)
|
|
|
|
proto = "https" if title_port_ssl_int != 443 else "http"
|
|
|
|
if proto == "https":
|
|
t_port = f":{title_port_ssl_int}" if title_port_ssl_int and not self.core_cfg.server.is_using_proxy else ""
|
|
|
|
else:
|
|
t_port = f":{title_port_int}" if title_port_int and not self.core_cfg.server.is_using_proxy else ""
|
|
|
|
return (
|
|
f"{proto}://{self.core_cfg.server.hostname}{t_port}",
|
|
"",
|
|
)
|
|
|
|
async def preprocess(self, req: Request) -> Dict:
|
|
req_bytes = await req.body()
|
|
|
|
try:
|
|
req_json: Dict = json.loads(req_bytes)
|
|
|
|
except Exception as e:
|
|
try:
|
|
req_json: Dict = json.loads(
|
|
req_bytes.decode().replace('"', '\\"').replace("'", '"')
|
|
)
|
|
|
|
except Exception as f:
|
|
self.logger.warning(
|
|
f"Error decoding json to /data endpoint: {e} / {f} - {req_bytes}"
|
|
)
|
|
return b""
|
|
|
|
return req_json
|
|
|
|
async def handle_data(self, request: Request) -> bytes:
|
|
req_json = await self.preprocess(request)
|
|
func_to_find = "handle_data_"
|
|
version_string = "Base"
|
|
internal_ver = 0
|
|
version = 0
|
|
|
|
if req_json == {}:
|
|
self.logger.warning(f"Empty json request to /data")
|
|
return Response()
|
|
|
|
subcmd = list(req_json.keys())[0]
|
|
if subcmd == "dldate":
|
|
|
|
if (
|
|
not type(req_json["dldate"]) is dict
|
|
or "filetype" not in req_json["dldate"]
|
|
):
|
|
self.logger.warning(f"Malformed dldate request: {req_json}")
|
|
return Response()
|
|
|
|
filetype = req_json["dldate"]["filetype"]
|
|
filetype_split = filetype.split("/")
|
|
|
|
if len(filetype_split) < 2 or not filetype_split[0].isnumeric():
|
|
self.logger.warning(f"Malformed dldate request: {req_json}")
|
|
return Response()
|
|
|
|
version = int(filetype_split[0])
|
|
filename = filetype_split[len(filetype_split) - 1]
|
|
|
|
match = re.match(
|
|
"^([A-Za-z]*)(\d\d\d\d)$", filetype_split[len(filetype_split) - 1]
|
|
)
|
|
if match:
|
|
func_to_find += f"{inflection.underscore(match.group(1))}xxxx"
|
|
else:
|
|
func_to_find += f"{inflection.underscore(filename)}"
|
|
else:
|
|
filetype = subcmd
|
|
func_to_find += filetype
|
|
|
|
func_to_find += "_request"
|
|
|
|
if version <= 10102:
|
|
version_string = "Rev"
|
|
internal_ver = CxbConstants.VER_CROSSBEATS_REV
|
|
|
|
elif version == 10113 or version == 10103:
|
|
version_string = "Rev SunriseS1"
|
|
internal_ver = CxbConstants.VER_CROSSBEATS_REV_SUNRISE_S1
|
|
|
|
elif version >= 10114 or version == 10104:
|
|
version_string = "Rev SunriseS2"
|
|
internal_ver = CxbConstants.VER_CROSSBEATS_REV_SUNRISE_S2
|
|
|
|
if not hasattr(self.versions[internal_ver], func_to_find):
|
|
self.logger.warn(f"{version_string} has no handler for filetype {filetype} / {func_to_find}")
|
|
return JSONResponse({"data":""})
|
|
|
|
self.logger.info(f"{version_string} request for filetype {filetype}")
|
|
self.logger.debug(req_json)
|
|
|
|
handler = getattr(self.versions[internal_ver], func_to_find)
|
|
|
|
try:
|
|
resp = await handler(req_json)
|
|
|
|
except Exception as e:
|
|
self.logger.error(f"Error handling request for file {filetype} - {e}")
|
|
if self.logger.level == logging.DEBUG:
|
|
tp, val, tb = sys.exc_info()
|
|
traceback.print_exception(tp, val, tb, limit=1)
|
|
with open("{0}/{1}.log".format(self.core_cfg.server.log_dir, "cxb"), "a") as f:
|
|
traceback.print_exception(tp, val, tb, limit=1, file=f)
|
|
return Response()
|
|
|
|
self.logger.debug(f"{version_string} Response {resp}")
|
|
return JSONResponse(resp, ensure_ascii=False)
|
|
|
|
async def handle_action(self, request: Request) -> bytes:
|
|
req_json = await self.preprocess(request)
|
|
subcmd = list(req_json.keys())[0]
|
|
func_to_find = f"handle_action_{subcmd}_request"
|
|
|
|
if not hasattr(self.versions[0], func_to_find):
|
|
self.logger.warn(f"No handler for action {subcmd} request")
|
|
return Response()
|
|
|
|
self.logger.info(f"Action {subcmd} Request")
|
|
self.logger.debug(req_json)
|
|
|
|
handler = getattr(self.versions[0], func_to_find)
|
|
|
|
try:
|
|
resp = await handler(req_json)
|
|
|
|
except Exception as e:
|
|
self.logger.error(f"Error handling action {subcmd} request - {e}")
|
|
if self.logger.level == logging.DEBUG:
|
|
tp, val, tb = sys.exc_info()
|
|
traceback.print_exception(tp, val, tb, limit=1)
|
|
with open("{0}/{1}.log".format(self.core_cfg.server.log_dir, "cxb"), "a") as f:
|
|
traceback.print_exception(tp, val, tb, limit=1, file=f)
|
|
return Response()
|
|
|
|
self.logger.debug(f"Response {resp}")
|
|
return JSONResponse(resp)
|
|
|
|
async def handle_auth(self, request: Request) -> bytes:
|
|
req_json = await self.preprocess(request)
|
|
subcmd = list(req_json.keys())[0]
|
|
func_to_find = f"handle_auth_{subcmd}_request"
|
|
|
|
if not hasattr(self.versions[0], func_to_find):
|
|
self.logger.warn(f"No handler for auth {subcmd} request")
|
|
return Response()
|
|
|
|
self.logger.info(f"Action {subcmd} Request")
|
|
self.logger.debug(req_json)
|
|
|
|
handler = getattr(self.versions[0], func_to_find)
|
|
|
|
try:
|
|
resp = await handler(req_json)
|
|
|
|
except Exception as e:
|
|
self.logger.error(f"Error handling auth {subcmd} request - {e}")
|
|
if self.logger.level == logging.DEBUG:
|
|
tp, val, tb = sys.exc_info()
|
|
traceback.print_exception(tp, val, tb, limit=1)
|
|
with open("{0}/{1}.log".format(self.core_cfg.server.log_dir, "cxb"), "a") as f:
|
|
traceback.print_exception(tp, val, tb, limit=1, file=f)
|
|
return Response()
|
|
|
|
self.logger.debug(f"Response {resp}")
|
|
return JSONResponse(resp)
|