From f64ae11ed52a735d9f590878d2279ea8d78c3871 Mon Sep 17 00:00:00 2001 From: Andy Date: Wed, 23 Jul 2025 15:57:46 +0800 Subject: [PATCH] refactor(app): use in-memory stream for license generation --- app.py | 161 ++++++++++++++++++++++++++------------------------------- 1 file changed, 72 insertions(+), 89 deletions(-) diff --git a/app.py b/app.py index ddb5a7f..ab9bc8a 100644 --- a/app.py +++ b/app.py @@ -1,19 +1,21 @@ -#/usr/bin/env python3 +#!/usr/bin/env python3 -import os, sys, zipfile -from flask import Flask, request, send_file -import os.path +import os +import sys +import zipfile +import io # <--- 引入 io 模块，用于内存操作 +from flask import Flask, request, send_file, make_response app = Flask(__name__) +# --- 核心加解密和编码逻辑 (这部分无需改动) --- VariantBase64Table = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=' -VariantBase64Dict = { i : VariantBase64Table[i] for i in range(len(VariantBase64Table)) } -VariantBase64ReverseDict = { VariantBase64Table[i] : i for i in range(len(VariantBase64Table)) } +VariantBase64Dict = {i: VariantBase64Table[i] for i in range(len(VariantBase64Table))} +VariantBase64ReverseDict = {VariantBase64Table[i]: i for i in range(len(VariantBase64Table))} -def VariantBase64Encode(bs : bytes): +def VariantBase64Encode(bs: bytes): result = b'' blocks_count, left_bytes = divmod(len(bs), 3) - for i in range(blocks_count): coding_int = int.from_bytes(bs[3 * i:3 * i + 3], 'little') block = VariantBase64Dict[coding_int & 0x3f] @@ -21,7 +23,6 @@ def VariantBase64Encode(bs : bytes): block += VariantBase64Dict[(coding_int >> 12) & 0x3f] block += VariantBase64Dict[(coding_int >> 18) & 0x3f] result += block.encode() - if left_bytes == 0: return result elif left_bytes == 1: @@ -38,106 +39,88 @@ def VariantBase64Encode(bs : bytes): result += block.encode() return result -def VariantBase64Decode(s : str): - result = b'' - blocks_count, left_bytes = divmod(len(s), 4) - - for i in range(blocks_count): - block = VariantBase64ReverseDict[s[4 * i]] - block += VariantBase64ReverseDict[s[4 * i + 1]] << 6 - block += VariantBase64ReverseDict[s[4 * i + 2]] << 12 - block += VariantBase64ReverseDict[s[4 * i + 3]] << 18 - result += block.to_bytes(3, 'little') - - if left_bytes == 0: - return result - elif left_bytes == 2: - block = VariantBase64ReverseDict[s[4 * blocks_count]] - block += VariantBase64ReverseDict[s[4 * blocks_count + 1]] << 6 - result += block.to_bytes(1, 'little') - return result - elif left_bytes == 3: - block = VariantBase64ReverseDict[s[4 * blocks_count]] - block += VariantBase64ReverseDict[s[4 * blocks_count + 1]] << 6 - block += VariantBase64ReverseDict[s[4 * blocks_count + 2]] << 12 - result += block.to_bytes(2, 'little') - return result - else: - raise ValueError('Invalid encoding.') - -def EncryptBytes(key : int, bs : bytes): +def EncryptBytes(key: int, bs: bytes): result = bytearray() for i in range(len(bs)): result.append(bs[i] ^ ((key >> 8) & 0xff)) key = result[-1] & key | 0x482D return bytes(result) -def DecryptBytes(key : int, bs : bytes): - result = bytearray() - for i in range(len(bs)): - result.append(bs[i] ^ ((key >> 8) & 0xff)) - key = bs[i] & key | 0x482D - return bytes(result) - class LicenseType: Professional = 1 Educational = 3 Persional = 4 -def GenerateLicense(Type : LicenseType, Count : int, UserName : str, MajorVersion : int, MinorVersion): - assert(Count >= 0) - LicenseString = '%d#%s|%d%d#%d#%d3%d6%d#%d#%d#%d#' % (Type, - UserName, MajorVersion, MinorVersion, - Count, - MajorVersion, MinorVersion, MinorVersion, - 0, # Unknown - 0, # No Games flag. 0 means "NoGames = false". But it does not work. - 0) # No Plugins flag. 0 means "NoPlugins = false". But it does not work. +# --- 重构后的核心功能 --- +def GenerateLicenseInMemory(Type: LicenseType, Count: int, UserName: str, MajorVersion: int, MinorVersion: int): + """ + 重构后的函数：在内存中生成许可证 ZIP 文件，并返回一个 BytesIO 对象。 + 不再向磁盘写入任何文件。 + """ + assert Count >= 0 + LicenseString = '%d#%s|%d%d#%d#%d3%d6%d#%d#%d#%d#' % ( + Type, UserName, MajorVersion, MinorVersion, + Count, + MajorVersion, MinorVersion, MinorVersion, + 0, 0, 0 + ) EncodedLicenseString = VariantBase64Encode(EncryptBytes(0x787, LicenseString.encode())).decode() - FileName = EncodedLicenseString.replace('/','').replace('\\','') - with zipfile.ZipFile(FileName, 'w') as f: - f.writestr('Pro.key', data = EncodedLicenseString) - return FileName - - -#@app.route('/gen') -def get_lc(): - name = request.args.get('name', '') - version = request.args.get('ver', '') - count = int(request.args.get('count', '1')) - try: - MajorVersion, MinorVersion = version.split('.')[0:2] - except: - return - MajorVersion = int(MajorVersion) - MinorVersion = int(MinorVersion) - lc = GenerateLicense(LicenseType.Professional, count, name, MajorVersion, MinorVersion) - return lc - - -#@app.route('/download/') -def download_lc(lc): - if lc and len(lc) > 5 and os.path.exists('./' + lc): - return send_file(lc, - as_attachment=True, - attachment_filename='Custom.mxtpro') - else: - return "请检查用户名版本号是否正确！" - - -@app.route('/gen') -def get(): - lc = get_lc() - return download_lc(lc) + + # 1. 创建一个内存中的二进制流对象 + memory_file = io.BytesIO() + + # 2. 像操作普通文件一样，在内存中创建 ZIP 归档 + with zipfile.ZipFile(memory_file, 'w', zipfile.ZIP_DEFLATED) as zf: + zf.writestr('Pro.key', data=EncodedLicenseString) + + # 3. 将内存流的指针移到开头，以便 send_file 从头读取 + memory_file.seek(0) + + return memory_file +# --- 重构后的 Flask 路由 --- @app.route('/') def index(): + """提供一个简单的使用说明页面""" return send_file('index.html') +@app.route('/gen') +def generate_and_download_license(): + """ + 一个统一的路由，处理参数、生成许可证并直接提供下载。 + """ + # 1. 获取和验证参数 + name = request.args.get('name') + version = request.args.get('ver') + + if not name or not version: + return make_response("错误：必须提供 'name' 和 'ver' 参数 (例如: /gen?name=MyName&ver=25.2)", 400) + + try: + count = int(request.args.get('count', '1')) + MajorVersion, MinorVersion = version.split('.')[0:2] + MajorVersion = int(MajorVersion) + MinorVersion = int(MinorVersion) + except (ValueError, IndexError): + return make_response("错误：版本号 'ver' 格式不正确，应为 '主版本号.次版本号' (例如: 25.2)", 400) + + # 2. 在内存中生成许可证文件 + license_file_stream = GenerateLicenseInMemory( + LicenseType.Professional, count, name, MajorVersion, MinorVersion + ) + + # 3. 使用 send_file 直接发送内存中的文件流 + return send_file( + license_file_stream, + mimetype='application/zip', + as_attachment=True, + download_name='Custom.mxtpro' # <--- 使用了正确的参数名 + ) if __name__ == '__main__': - app.run(host='0.0.0.0', port=5000, debug=False) + # 建议开启 debug=True 进行开发调试，部署时设为 False + app.run(host='0.0.0.0', port=5000, debug=True)