Merge pull request #17 from ailuckly/dev
refactor(app): use in-memory stream for license generationmain
malaohu
GitHub
commit
f071851d57
157
app.py
157
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,
|
||||
# --- 重构后的核心功能 ---
|
||||
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, # 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.
|
||||
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/<lc>')
|
||||
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)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue