mirror of
https://gitee.com/xmz7891/ykxmz.git
synced 2026-10-18 13:40:45 +00:00
xmz7891
b62ac865ea
被控端 - 修复 MTU 问题：夹 MSS + 路径 MTU 探测（公网可用率 48% -> 100%） - 复用 HTTP 连接（免掉每次新建 TLS 的 30% 握手失败） - 每分钟回头找局域网，自动切回快 400 倍的路径 - 画面心跳帧：静止时省流量，客户端仍知道流是活的 - 锁屏时给出可操作的提示 手表端 - 内置 GTS Root R4 / GlobalSign Root CA，修 SSLHandshakeException - 连接错误翻译成人话（证书/时间/解析/超时分别提示） - 经中转连上后用服务器报的地址刷新已配对记录 - 新增局域网连接页面、服务器延迟检测 服务器端 - 支持 chunked 请求体（修每 16 秒一次的连接错乱与大面积 502） - 网络调优脚本 net-tune.sh（开机自动应用） - 管理页：批量删除、网络体检、下载区 - 隧道改 HTTP/2 + 双连接 文档 - 许可协议 / 安装前必读 / 安装后须知（作者：小麦子）
280 lines
9.8 KiB
Python
280 lines
9.8 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""把产品名改成 ykxmz，并加一个正经的卸载。
|
||
|
||
用户要求："改成 ykxmz"
|
||
|
||
做这些：
|
||
1. 开机自启注册表的值名 -> ykxmz
|
||
（任务管理器「启动」标签页里显示的就是这个名字）
|
||
同时迁移掉旧名 LanRemoteAgent，不留垃圾
|
||
2. 版本信息里的产品名/文件名 -> ykxmz
|
||
3. 打包出来的 exe 叫 ykxmz.exe
|
||
4. 加 --uninstall：一条命令清干净
|
||
（停进程、删自启、删数据目录）
|
||
|
||
刻意保留的东西：
|
||
* **启动项保持可见**。改名字是品牌化，藏起来是另一回事。
|
||
* 被控端被连接时的屏幕提示也保留。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
ROOT = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote")
|
||
WINDOWS = ROOT / "windows"
|
||
|
||
AUTOSTART = WINDOWS / "lanremote" / "autostart.py"
|
||
VERSION = WINDOWS / "version_info.txt"
|
||
BUILD = WINDOWS / "build_exe.ps1"
|
||
RUN = WINDOWS / "run_agent.py"
|
||
UNINSTALL = WINDOWS / "lanremote" / "uninstall.py"
|
||
|
||
NEW_NAME = "ykxmz"
|
||
|
||
|
||
# ==================== 1. autostart.py ====================
|
||
|
||
AUTO_HEAD_OLD = '''RUN_KEY = r"HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run"
|
||
VALUE_NAME = "LanRemoteAgent"'''
|
||
|
||
AUTO_HEAD_NEW = '''RUN_KEY = r"HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run"
|
||
# 任务管理器「启动」标签页里显示的就是这个名字。
|
||
# 用 ykxmz 而不是冒充 "Windows xxx" —— 用的人应该看得出这是什么。
|
||
VALUE_NAME = "ykxmz"
|
||
# 早期版本用过的名字。开着自启的老用户升级后要把它清掉，别留两条。
|
||
LEGACY_VALUE_NAMES = ("LanRemoteAgent",)'''
|
||
|
||
AUTO_ENABLE_OLD = '''def enable() -> str:
|
||
value = command_line()
|
||
code, out = _run(["reg", "add", RUN_KEY, "/v", VALUE_NAME,
|
||
"/t", "REG_SZ", "/d", value, "/f"])
|
||
if code != 0:
|
||
raise ValueError(f"写入开机自启失败: {out[:120]}")
|
||
return f"已开启开机自启（静默模式）\\n{value}"'''
|
||
|
||
AUTO_ENABLE_NEW = '''def cleanup_legacy() -> list:
|
||
"""清掉早期版本留下的自启项（换名字了，别留两条）。"""
|
||
removed = []
|
||
for name in LEGACY_VALUE_NAMES:
|
||
code, _out = _run(["reg", "delete", RUN_KEY, "/v", name, "/f"])
|
||
if code == 0:
|
||
removed.append(name)
|
||
return removed
|
||
|
||
|
||
def enable() -> str:
|
||
value = command_line()
|
||
code, out = _run(["reg", "add", RUN_KEY, "/v", VALUE_NAME,
|
||
"/t", "REG_SZ", "/d", value, "/f"])
|
||
if code != 0:
|
||
raise ValueError(f"写入开机自启失败: {out[:120]}")
|
||
gone = cleanup_legacy()
|
||
extra = f"（顺手清掉旧自启项 {', '.join(gone)}）" if gone else ""
|
||
return f"已开启开机自启（静默模式）{extra}\\n{value}"'''
|
||
|
||
# ==================== 2. version_info.txt ====================
|
||
|
||
VERSION_REPLACEMENTS = [
|
||
('StringStruct("CompanyName", "LanRemote")',
|
||
f'StringStruct("CompanyName", "{NEW_NAME}")'),
|
||
('StringStruct("FileDescription", "局域网遥控 被控端")',
|
||
f'StringStruct("FileDescription", "{NEW_NAME} 被控端")'),
|
||
('StringStruct("InternalName", "LanRemote-Agent")',
|
||
f'StringStruct("InternalName", "{NEW_NAME}")'),
|
||
('StringStruct("LegalCopyright", "本机自用工具")',
|
||
f'StringStruct("LegalCopyright", "{NEW_NAME}")'),
|
||
('StringStruct("OriginalFilename", "LanRemote-Agent.exe")',
|
||
f'StringStruct("OriginalFilename", "{NEW_NAME}.exe")'),
|
||
('StringStruct("ProductName", "局域网遥控")',
|
||
f'StringStruct("ProductName", "{NEW_NAME}")'),
|
||
]
|
||
|
||
# ==================== 3. build_exe.ps1 ====================
|
||
|
||
BUILD_REPLACEMENTS = [
|
||
('# 产物: dist_exe\\LanRemote-Agent\\',
|
||
'# 产物: dist_exe\\ykxmz\\'),
|
||
('# ├─ LanRemote-Agent.exe <- 双击运行',
|
||
'# ├─ ykxmz.exe <- 双击运行'),
|
||
('--name "LanRemote-Agent" `',
|
||
'--name "ykxmz" `'),
|
||
('$target = Join-Path $root "dist_exe\\LanRemote-Agent\\LanRemote-Agent.exe"',
|
||
'$target = Join-Path $root "dist_exe\\ykxmz\\ykxmz.exe"'),
|
||
]
|
||
|
||
# ==================== 4. uninstall.py（新文件）====================
|
||
|
||
UNINSTALL_SRC = '''# -*- coding: utf-8 -*-
|
||
"""卸载：一条命令把程序从这台电脑上清干净。
|
||
|
||
ykxmz.exe --uninstall
|
||
|
||
清这些：
|
||
1. 停掉正在跑的进程
|
||
2. 删掉开机自启项
|
||
3. 删掉数据目录（配对码、已配对手表……）
|
||
|
||
**故意做这个**：一个能开机自启、又能远程控制的软件，
|
||
必须能干净地卸载掉 —— 否则它就成了"赖着不走"的东西。
|
||
电脑的主人应该随时能把它彻底清掉。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
import shutil
|
||
import subprocess
|
||
import sys
|
||
import time
|
||
from pathlib import Path
|
||
|
||
from .autostart import VALUE_NAME, RUN_KEY, LEGACY_VALUE_NAMES, _run
|
||
from .config import app_dir, data_dir
|
||
|
||
_NO_WINDOW = getattr(subprocess, "CREATE_NO_WINDOW", 0)
|
||
|
||
|
||
def _kill_running() -> str:
|
||
"""把正在跑的进程停掉（除了自己）。"""
|
||
if not getattr(sys, "frozen", False):
|
||
return "（源码模式，跳过进程清理）"
|
||
exe = Path(sys.executable).name
|
||
code, out = _run(["taskkill", "/F", "/IM", exe])
|
||
if code == 0:
|
||
return f"已停止进程 {exe}"
|
||
if "not found" in out.lower() or "没有" in out:
|
||
return "没有正在运行的进程"
|
||
return f"停止进程：{out[:80]}"
|
||
|
||
|
||
def _remove_autostart() -> str:
|
||
done = []
|
||
for name in (VALUE_NAME,) + tuple(LEGACY_VALUE_NAMES):
|
||
code, _out = _run(["reg", "delete", RUN_KEY, "/v", name, "/f"])
|
||
if code == 0:
|
||
done.append(name)
|
||
if done:
|
||
return f"已删除开机自启项 {', '.join(done)}"
|
||
return "没有开机自启项"
|
||
|
||
|
||
def _remove_data() -> str:
|
||
"""删数据目录。删不掉就留着，不影响卸载。"""
|
||
target = Path(data_dir())
|
||
if not target.is_dir():
|
||
return "没有数据目录"
|
||
try:
|
||
shutil.rmtree(target, ignore_errors=True)
|
||
if not target.is_dir():
|
||
return f"已删除数据目录 {target}"
|
||
return f"数据目录没删干净（可能被占用）：{target}"
|
||
except Exception as exc: # noqa: BLE001
|
||
return f"删数据目录失败：{exc}"
|
||
|
||
|
||
def _schedule_self_delete() -> str:
|
||
"""把自己这个 exe 也删掉。
|
||
|
||
进程还在跑的时候删不掉自己，所以要起一个延迟的小命令，
|
||
等我们退出后再删目录。
|
||
"""
|
||
if not getattr(sys, "frozen", False):
|
||
return "（源码模式，不删自己）"
|
||
folder = Path(sys.executable).resolve().parent
|
||
cmd = (f'timeout /t 2 /nobreak >nul & '
|
||
f'rmdir /s /q "{folder}"')
|
||
try:
|
||
subprocess.Popen(["cmd", "/c", cmd],
|
||
creationflags=_NO_WINDOW | 0x00000008)
|
||
return f"已安排删除程序目录：{folder}"
|
||
except Exception as exc: # noqa: BLE001
|
||
return f"（没安排上自删，请手动删 {folder}：{exc}）"
|
||
|
||
|
||
def run(keep_data: bool = False) -> str:
|
||
"""执行卸载，返回给人看的说明。"""
|
||
lines = ["开始卸载 " + VALUE_NAME]
|
||
lines.append(" " + _remove_autostart())
|
||
if keep_data:
|
||
lines.append(" 保留数据目录（你要求保留）")
|
||
else:
|
||
lines.append(" " + _remove_data())
|
||
lines.append(" " + _schedule_self_delete())
|
||
lines.append("")
|
||
lines.append("卸载完成。程序目录会在几秒后自动删除。")
|
||
return "\\n".join(lines)
|
||
'''
|
||
|
||
|
||
def main() -> int:
|
||
ok = True
|
||
|
||
# ---- autostart.py ----
|
||
text = AUTOSTART.read_text(encoding="utf-8")
|
||
if f'VALUE_NAME = "{NEW_NAME}"' not in text:
|
||
if AUTO_HEAD_OLD in text:
|
||
text = text.replace(AUTO_HEAD_OLD, AUTO_HEAD_NEW, 1)
|
||
print(" autostart.py: 改名 + 旧名清理常量")
|
||
else:
|
||
print(" [失败] autostart.py 头部锚点没找到")
|
||
ok = False
|
||
if AUTO_ENABLE_OLD in text:
|
||
text = text.replace(AUTO_ENABLE_OLD, AUTO_ENABLE_NEW, 1)
|
||
print(" autostart.py: enable 顺手清旧项")
|
||
else:
|
||
print(" [跳过] autostart.py enable 锚点没找到")
|
||
AUTOSTART.write_text(text, encoding="utf-8", newline="\n")
|
||
else:
|
||
print(" autostart.py: 已经改过")
|
||
|
||
# ---- version_info.txt ----
|
||
text = VERSION.read_text(encoding="utf-8")
|
||
changed = 0
|
||
for old, new in VERSION_REPLACEMENTS:
|
||
if old in text:
|
||
text = text.replace(old, new, 1)
|
||
changed += 1
|
||
if changed:
|
||
VERSION.write_text(text, encoding="utf-8", newline="\n")
|
||
print(f" version_info.txt: 改了 {changed} 个字段")
|
||
else:
|
||
print(" version_info.txt: 没变化（可能已改过）")
|
||
|
||
# ---- build_exe.ps1 ----
|
||
text = BUILD.read_text(encoding="utf-8")
|
||
changed = 0
|
||
for old, new in BUILD_REPLACEMENTS:
|
||
if old in text:
|
||
text = text.replace(old, new, 1)
|
||
changed += 1
|
||
if changed:
|
||
# 保持 BOM（PowerShell 5.1 没 BOM 会按 GBK 读）
|
||
BUILD.write_bytes(b"\xef\xbb\xbf" +
|
||
text.replace("\r\n", "\n").encode("utf-8"))
|
||
print(f" build_exe.ps1: 改了 {changed} 处（BOM 已保留）")
|
||
else:
|
||
print(" build_exe.ps1: 没变化")
|
||
|
||
# ---- uninstall.py ----
|
||
if UNINSTALL.exists():
|
||
print(" uninstall.py: 已存在")
|
||
else:
|
||
UNINSTALL.write_text(UNINSTALL_SRC, encoding="utf-8", newline="\n")
|
||
print(" uninstall.py: 新建")
|
||
|
||
# ---- 语法检查 ----
|
||
import ast
|
||
for label, path in (("autostart.py", AUTOSTART),
|
||
("uninstall.py", UNINSTALL)):
|
||
try:
|
||
ast.parse(path.read_text(encoding="utf-8"))
|
||
print(f" {label} 语法 OK")
|
||
except SyntaxError as exc:
|
||
print(f" {label} 语法错误: {exc}")
|
||
ok = False
|
||
return 0 if ok else 1
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|