forked from GuChen/maimaiDX-API-Web-Server
Initial commit: Add maimaiDX API web application with AimeDB scanning and logging features
This commit is contained in:
kejiz
352
backend/API_TitleServer.py
Normal file
352
backend/API_TitleServer.py
Normal file
@@ -0,0 +1,352 @@
|
||||
# 舞萌DX
|
||||
# 标题服务器通讯实现
|
||||
import zlib
|
||||
import hashlib
|
||||
import random
|
||||
import time
|
||||
|
||||
import httpx
|
||||
from loguru import logger
|
||||
from ctypes import c_int32
|
||||
from Crypto.Cipher import AES
|
||||
from Crypto.Util.Padding import pad, unpad
|
||||
|
||||
from Config import (
|
||||
useProxy,
|
||||
proxyUrl,
|
||||
)
|
||||
|
||||
|
||||
AesKey = "a>32bVP7v<63BVLkY[xM>daZ1s9MBP<R"
|
||||
AesIV = "d6xHIKq]1J]Dt^ue"
|
||||
ObfuscateParam = "B44df8yT"
|
||||
|
||||
|
||||
def use2024Api():
|
||||
global AesKey, AesIV, ObfuscateParam
|
||||
|
||||
AesKey = "n7bx6:@Fg_:2;5E89Phy7AyIcpxEQ:R@"
|
||||
AesIV = ";;KjR1C3hgB1ovXa"
|
||||
ObfuscateParam = "BEs2D5vW"
|
||||
|
||||
|
||||
class SDGBApiError(Exception):
|
||||
pass
|
||||
|
||||
|
||||
class SDGBRequestError(SDGBApiError):
|
||||
pass
|
||||
|
||||
|
||||
class SDGBResponseError(SDGBApiError):
|
||||
pass
|
||||
|
||||
|
||||
class aes_pkcs7(object):
|
||||
def __init__(self, key: str, iv: str):
|
||||
self.key = key.encode("utf-8")
|
||||
self.iv = iv.encode("utf-8")
|
||||
self.mode = AES.MODE_CBC
|
||||
|
||||
def encrypt(self, content: bytes) -> bytes:
|
||||
cipher = AES.new(self.key, self.mode, self.iv)
|
||||
content_padded = pad(content, AES.block_size)
|
||||
encrypted_bytes = cipher.encrypt(content_padded)
|
||||
return encrypted_bytes
|
||||
|
||||
def decrypt(self, content):
|
||||
cipher = AES.new(self.key, self.mode, self.iv)
|
||||
decrypted_padded = cipher.decrypt(content)
|
||||
decrypted = unpad(decrypted_padded, AES.block_size)
|
||||
return decrypted
|
||||
|
||||
def pkcs7unpadding(self, text):
|
||||
length = len(text)
|
||||
unpadding = ord(text[length - 1])
|
||||
return text[0 : length - unpadding]
|
||||
|
||||
def pkcs7padding(self, text):
|
||||
bs = 16
|
||||
length = len(text)
|
||||
bytes_length = len(text.encode("utf-8"))
|
||||
padding_size = length if (bytes_length == length) else bytes_length
|
||||
padding = bs - padding_size % bs
|
||||
padding_text = chr(padding) * padding
|
||||
return text + padding_text
|
||||
|
||||
|
||||
def getSDGBApiHash(api):
|
||||
# API 的 Hash 的生成
|
||||
# 有空做一下 Hash 的彩虹表？
|
||||
return hashlib.md5((api + "MaimaiChn" + ObfuscateParam).encode()).hexdigest()
|
||||
|
||||
|
||||
def apiSDGB(
|
||||
data: str,
|
||||
targetApi: str,
|
||||
userAgentExtraData: str,
|
||||
noLog: bool = False,
|
||||
timeout: int = 5,
|
||||
maxRetries: int = 3,
|
||||
) -> str:
|
||||
"""
|
||||
舞萌DX 2025 API 通讯用函数
|
||||
|
||||
:param data: 请求数据
|
||||
:param targetApi: 使用的 API
|
||||
:param userAgentExtraData: UA 附加信息，机台相关则为狗号（如A63E01E9564），用户相关则为 UID
|
||||
:param noLog: 是否不记录日志
|
||||
:param timeout: 请求超时时间（秒）
|
||||
:return: 解码后的响应数据
|
||||
"""
|
||||
# 处理参数
|
||||
agentExtra = str(userAgentExtraData)
|
||||
aes = aes_pkcs7(AesKey, AesIV)
|
||||
endpoint = "https://maimai-gm.wahlap.com:42081/Maimai2Servlet/"
|
||||
|
||||
# 准备好请求数据
|
||||
requestDataFinal = aes.encrypt(zlib.compress(data.encode("utf-8")))
|
||||
|
||||
if not noLog:
|
||||
logger.debug(f"[Stage 1] 准备开始请求 {targetApi}，以 {data}")
|
||||
|
||||
retries = 0
|
||||
while retries < maxRetries:
|
||||
try:
|
||||
# 配置 HTTP 客户端
|
||||
if useProxy and proxyUrl:
|
||||
logger.debug("使用代理")
|
||||
httpClient = httpx.Client(proxy=proxyUrl, verify=False)
|
||||
else:
|
||||
httpClient = httpx.Client(verify=False)
|
||||
|
||||
api_hash = getSDGBApiHash(targetApi)
|
||||
|
||||
logger.info(f"hash: {api_hash}")
|
||||
|
||||
# 发送请求
|
||||
response = httpClient.post(
|
||||
url=endpoint + api_hash,
|
||||
headers={
|
||||
"User-Agent": f"{api_hash}#{agentExtra}",
|
||||
"Content-Type": "application/json",
|
||||
"Mai-Encoding": "1.50",
|
||||
"Accept-Encoding": "",
|
||||
"Charset": "UTF-8",
|
||||
"Content-Encoding": "deflate",
|
||||
"Expect": "100-continue",
|
||||
},
|
||||
content=requestDataFinal, # 数据
|
||||
timeout=timeout,
|
||||
)
|
||||
|
||||
if not noLog:
|
||||
logger.info(f"[Stage 2] {targetApi} 请求结果: {response.status_code}")
|
||||
|
||||
if response.status_code != 200:
|
||||
errorMessage = f"[Stage 2] 请求失败: {response.status_code}"
|
||||
logger.error(errorMessage)
|
||||
raise SDGBRequestError(errorMessage)
|
||||
|
||||
# 处理响应内容
|
||||
responseContentRaw = response.content
|
||||
|
||||
# 先尝试解密
|
||||
try:
|
||||
responseContentDecrypted = aes.decrypt(responseContentRaw)
|
||||
if not noLog:
|
||||
logger.debug("[Stage 3] Decryption SUCCESS.")
|
||||
except Exception as e:
|
||||
logger.warning(
|
||||
f"[Stage 3] Decryption FAILED. Raw Content: {responseContentRaw}, Error: {e}"
|
||||
)
|
||||
raise SDGBResponseError("Decryption failed")
|
||||
# 然后尝试解压
|
||||
try:
|
||||
# 看看文件头是否是压缩过的
|
||||
if responseContentDecrypted.startswith(b"\x78\x9c"):
|
||||
logger.debug("[Stage 4] Zlib detected, decompressing...")
|
||||
responseContentFinal = zlib.decompress(
|
||||
responseContentDecrypted
|
||||
).decode("utf-8")
|
||||
else:
|
||||
logger.warning(
|
||||
f"[Stage 4] Not Zlib Format!! using raw content: {responseContentDecrypted}"
|
||||
)
|
||||
responseContentFinal = responseContentDecrypted.decode("utf-8")
|
||||
# 完成解压
|
||||
if not noLog:
|
||||
logger.debug(
|
||||
f"[Stage 4] Process OK, Content: {responseContentFinal}"
|
||||
)
|
||||
# 最终处理，检查是否是 JSON 格式
|
||||
if responseContentFinal.startswith(
|
||||
"{"
|
||||
) and responseContentFinal.endswith("}"):
|
||||
# 如果是 JSON 格式，直接返回
|
||||
logger.debug("[Stage 5] Response is JSON, returning.")
|
||||
return responseContentFinal
|
||||
else:
|
||||
# 如果不是 JSON 格式，直接返回但是警告
|
||||
logger.warning(
|
||||
"[Stage 5] Response is not JSON, returning as is, take care!"
|
||||
)
|
||||
return responseContentFinal
|
||||
except Exception:
|
||||
logger.warning(f"解压失败，原始响应: {responseContentDecrypted}")
|
||||
raise SDGBResponseError("解压失败")
|
||||
except SDGBRequestError as e:
|
||||
logger.error(f"请求格式错误: {e}")
|
||||
raise
|
||||
except SDGBResponseError as e:
|
||||
logger.warning(f"响应错误，将重试: {e}")
|
||||
retries += 1
|
||||
time.sleep(2)
|
||||
except Exception as e:
|
||||
logger.warning(f"请求失败，将重试: {e}")
|
||||
retries += 1
|
||||
time.sleep(2)
|
||||
|
||||
finally:
|
||||
if "httpClient" in locals():
|
||||
httpClient.close()
|
||||
|
||||
raise SDGBApiError("重试多次仍然无法成功请求服务器")
|
||||
|
||||
|
||||
def calcPlaySpecial():
|
||||
"""使用 c_int32 实现的 SpecialNumber 算法"""
|
||||
rng = random.SystemRandom()
|
||||
num2 = rng.randint(1, 1037933) * 2069
|
||||
num2 += 1024 # GameManager.CalcSpecialNum()
|
||||
num2 = c_int32(num2).value
|
||||
result = c_int32(0)
|
||||
for _ in range(32):
|
||||
result.value <<= 1
|
||||
result.value += num2 & 1
|
||||
num2 >>= 1
|
||||
return c_int32(result.value).value
|
||||
|
||||
|
||||
class AESPKCS7_2024:
|
||||
# 实现了 maimai 通讯所用的 AES 加密的类
|
||||
def __init__(self, key: str, iv: str):
|
||||
self.key = key.encode("utf-8")
|
||||
self.iv = iv.encode("utf-8")
|
||||
self.mode = AES.MODE_CBC
|
||||
|
||||
# 加密
|
||||
def encrypt(self, content) -> bytes:
|
||||
# if content is str, convert to bytes
|
||||
if isinstance(content, str):
|
||||
encodedData = content.encode("utf-8")
|
||||
cipher = AES.new(self.key, self.mode, self.iv)
|
||||
content_padded = pad(encodedData, AES.block_size)
|
||||
encrypted_bytes = cipher.encrypt(content_padded)
|
||||
return encrypted_bytes
|
||||
|
||||
# 解密
|
||||
def decrypt(self, encrypted_content: bytes) -> str:
|
||||
cipher = AES.new(self.key, self.mode, self.iv)
|
||||
decrypted_padded = cipher.decrypt(encrypted_content)
|
||||
decrypted = unpad(decrypted_padded, AES.block_size)
|
||||
return decrypted
|
||||
|
||||
|
||||
def apiSDGB_2024(
|
||||
data: str,
|
||||
targetApi: str,
|
||||
userAgentExtraData: str,
|
||||
noLog: bool = False,
|
||||
timeout: int = 5,
|
||||
):
|
||||
"""
|
||||
舞萌DX 2024 API 通讯用函数
|
||||
:param data: 请求数据
|
||||
:param targetApi: 使用的 API
|
||||
:param userAgentExtraData: UA 附加信息，机台相关则为狗号（如A63E01E9564），用户相关则为 UID
|
||||
:param noLog: 是否不记录日志
|
||||
"""
|
||||
maxRetries = 3
|
||||
agentExtra = str(userAgentExtraData)
|
||||
aes = AESPKCS7_2024(AesKey, AesIV)
|
||||
reqData_encrypted = aes.encrypt(data)
|
||||
reqData_deflated = zlib.compress(reqData_encrypted)
|
||||
endpoint = "https://maimai-gm.wahlap.com:42081/Maimai2Servlet/"
|
||||
if not noLog:
|
||||
logger.debug(f"开始请求 {targetApi}，以 {data}")
|
||||
|
||||
retries = 0
|
||||
while retries < maxRetries:
|
||||
try:
|
||||
if useProxy:
|
||||
# 使用代理
|
||||
logger.debug("使用代理")
|
||||
httpClient = httpx.Client(proxy=proxyUrl, verify=False)
|
||||
else:
|
||||
# 不使用代理
|
||||
logger.debug("不使用代理")
|
||||
httpClient = httpx.Client(verify=False)
|
||||
responseOriginal = httpClient.post(
|
||||
url=endpoint + getSDGBApiHash(targetApi),
|
||||
headers={
|
||||
"User-Agent": f"{getSDGBApiHash(targetApi)}#{agentExtra}",
|
||||
"Content-Type": "application/json",
|
||||
"Mai-Encoding": "1.40",
|
||||
"Accept-Encoding": "",
|
||||
"Charset": "UTF-8",
|
||||
"Content-Encoding": "deflate",
|
||||
"Expect": "100-continue",
|
||||
},
|
||||
content=reqData_deflated,
|
||||
# 经测试，加 Verify 之后速度慢好多，因此建议选择性开
|
||||
# verify=certifi.where(),
|
||||
# verify=False,
|
||||
timeout=timeout,
|
||||
)
|
||||
|
||||
if not noLog:
|
||||
logger.info(f"{targetApi} 请求结果: {responseOriginal.status_code}")
|
||||
|
||||
if responseOriginal.status_code == 200:
|
||||
logger.debug("200 OK!")
|
||||
else:
|
||||
errorMessage = f"请求失败: {responseOriginal.status_code}"
|
||||
logger.error(errorMessage)
|
||||
raise SDGBRequestError(errorMessage)
|
||||
|
||||
responseRAWContent = responseOriginal.content
|
||||
|
||||
try:
|
||||
responseDecompressed = zlib.decompress(responseRAWContent)
|
||||
logger.debug("成功解压响应！")
|
||||
except Exception:
|
||||
logger.warning(f"无法解压，得到的原始响应: {responseRAWContent}")
|
||||
raise SDGBResponseError("解压失败")
|
||||
try:
|
||||
resultResponse = aes.decrypt(responseDecompressed)
|
||||
logger.debug("成功解密响应！")
|
||||
except Exception:
|
||||
logger.warning(f"解密失败，得到的原始响应: {responseDecompressed}")
|
||||
raise SDGBResponseError("解密失败")
|
||||
|
||||
if not noLog:
|
||||
logger.debug(f"响应: {resultResponse}")
|
||||
return resultResponse
|
||||
|
||||
# 异常处理
|
||||
except SDGBRequestError:
|
||||
# 请求格式错误，不需要重试
|
||||
raise SDGBRequestError("请求格式错误")
|
||||
except SDGBResponseError as e:
|
||||
# 响应解析错误，这种有一定可能是我们的问题，所以只重试一次
|
||||
logger.warning(f"将重试一次 Resp Err: {e}")
|
||||
retries += 2
|
||||
time.sleep(2)
|
||||
except Exception as e:
|
||||
# 其他错误，重试多次
|
||||
logger.warning(f"将开始重试请求. {e}")
|
||||
retries += 1
|
||||
time.sleep(2)
|
||||
|
||||
raise SDGBApiError("重试多次仍然无法成功请求服务器")
|
||||
Reference in New Issue
Block a user