from twisted.web.http import Request import logging, coloredlogs from logging.handlers import TimedRotatingFileHandler import zlib import yaml import json import inflection import string from Crypto.Cipher import AES from Crypto.Util.Padding import pad from core import CoreConfig from titles.chuni.config import ChuniConfig from titles.chuni.const import ChuniConstants from titles.chuni.base import ChuniBase from titles.chuni.plus import ChuniPlus from titles.chuni.air import ChuniAir from titles.chuni.airplus import ChuniAirPlus from titles.chuni.star import ChuniStar from titles.chuni.starplus import ChuniStarPlus from titles.chuni.amazon import ChuniAmazon from titles.chuni.amazonplus import ChuniAmazonPlus from titles.chuni.crystal import ChuniCrystal from titles.chuni.crystalplus import ChuniCrystalPlus from titles.chuni.paradise import ChuniParadise from titles.chuni.new import ChuniNew from titles.chuni.newplus import ChuniNewPlus class ChuniServlet(): def __init__(self, core_cfg: CoreConfig, cfg_dir: str) -> None: self.core_cfg = core_cfg self.game_cfg = ChuniConfig() self.game_cfg.update(yaml.safe_load(open(f"{cfg_dir}/chuni.yaml"))) self.versions = [ ChuniBase(core_cfg, self.game_cfg), ChuniPlus(core_cfg, self.game_cfg), ChuniAir(core_cfg, self.game_cfg), ChuniAirPlus(core_cfg, self.game_cfg), ChuniStar(core_cfg, self.game_cfg), ChuniStarPlus(core_cfg, self.game_cfg), ChuniAmazon(core_cfg, self.game_cfg), ChuniAmazonPlus(core_cfg, self.game_cfg), ChuniCrystal(core_cfg, self.game_cfg), ChuniCrystalPlus(core_cfg, self.game_cfg), ChuniParadise(core_cfg, self.game_cfg), ChuniNew(core_cfg, self.game_cfg), ChuniNewPlus(core_cfg, self.game_cfg), ] self.logger = logging.getLogger("chuni") if not hasattr(self.logger, "inited"): log_fmt_str = "[%(asctime)s] Chunithm | %(levelname)s | %(message)s" log_fmt = logging.Formatter(log_fmt_str) fileHandler = TimedRotatingFileHandler("{0}/{1}.log".format(self.core_cfg.server.log_dir, "chuni"), 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 def render_POST(self, request: Request, version: int, url_path: str) -> bytes: req_raw = request.content.getvalue() url_split = url_path.split("/") encrtped = False internal_ver = 0 endpoint = url_split[len(url_split) - 1] if version < 105: # 1.0 internal_ver = ChuniConstants.VER_CHUNITHM elif version >= 105 and version < 110: # Plus internal_ver = ChuniConstants.VER_CHUNITHM_PLUS elif version >= 110 and version < 115: # Air internal_ver = ChuniConstants.VER_CHUNITHM_AIR elif version >= 115 and version < 120: # Air Plus internal_ver = ChuniConstants.VER_CHUNITHM_AIR_PLUS elif version >= 120 and version < 125: # Star internal_ver = ChuniConstants.VER_CHUNITHM_STAR elif version >= 125 and version < 130: # Star Plus internal_ver = ChuniConstants.VER_CHUNITHM_STAR_PLUS elif version >= 130 and version < 135: # Amazon internal_ver = ChuniConstants.VER_CHUNITHM_AMAZON elif version >= 135 and version < 140: # Amazon Plus internal_ver = ChuniConstants.VER_CHUNITHM_AMAZON_PLUS elif version >= 140 and version < 145: # Crystal internal_ver = ChuniConstants.VER_CHUNITHM_CRYSTAL elif version >= 145 and version < 150: # Crystal Plus internal_ver = ChuniConstants.VER_CHUNITHM_CRYSTAL_PLUS elif version >= 150 and version < 200: # Paradise internal_ver = ChuniConstants.VER_CHUNITHM_PARADISE elif version >= 200 and version < 205: # New internal_ver = ChuniConstants.VER_CHUNITHM_NEW elif version >= 205 and version < 210: # New Plus internal_ver = ChuniConstants.VER_CHUNITHM_NEW_PLUS if all(c in string.hexdigits for c in endpoint) and len(endpoint) == 32: # If we get a 32 character long hex string, it's a hash and we're # doing encrypted. The likelyhood of false positives is low but # technically not 0 endpoint = request.getHeader("User-Agent").split("#")[0] try: crypt = AES.new( bytes.fromhex(self.game_cfg.crypto.keys[str(internal_ver)][0]), AES.MODE_CBC, bytes.fromhex(self.game_cfg.crypto.keys[str(internal_ver)][1]) ) req_raw = crypt.decrypt(req_raw) except: self.logger.error(f"Failed to decrypt v{version} request to {endpoint} -> {req_raw}") return zlib.compress("{\"stat\": \"0\"}".encode("utf-8")) encrtped = True if not encrtped and self.game_cfg.crypto.encrypted_only: self.logger.error(f"Unencrypted v{version} {endpoint} request, but config is set to encrypted only: {req_raw}") return zlib.compress("{\"stat\": \"0\"}".encode("utf-8")) try: unzip = zlib.decompress(req_raw) except zlib.error as e: self.logger.error(f"Failed to decompress v{version} {endpoint} request -> {e}") return b"" req_data = json.loads(unzip) self.logger.info(f"v{version} {endpoint} request - {req_data}") func_to_find = "handle_" + inflection.underscore(endpoint) + "_request" try: handler = getattr(self.versions[internal_ver], func_to_find) resp = handler(req_data) except AttributeError as e: self.logger.warning(f"Unhandled v{version} request {endpoint} - {e}") return zlib.compress("{\"stat\": \"0\"}".encode("utf-8")) except Exception as e: self.logger.error(f"Error handling v{version} method {endpoint} - {e}") return zlib.compress("{\"stat\": \"0\"}".encode("utf-8")) if resp == None: resp = {'returnCode': 1} self.logger.info(f"Response {resp}") zipped = zlib.compress(json.dumps(resp, ensure_ascii=False).encode("utf-8")) if not encrtped: return zipped padded = pad(zipped, 16) crypt = AES.new( bytes.fromhex(self.game_cfg.crypto.keys[str(internal_ver)][0]), AES.MODE_CBC, bytes.fromhex(self.game_cfg.crypto.keys[str(internal_ver)][1]) ) return crypt.encrypt(padded)