| | |
|---|
| | | import os |
|---|
| | | import subprocess |
|---|
| | | import time |
|---|
| | | import logging |
|---|
| | | import sys |
|---|
| | | import getpass |
|---|
| | | |
|---|
| | | APP_NAME = "sening_vdp" |
|---|
| | | LOG_FILE = "sening_vdp.log" |
|---|
| | | |
|---|
| | | BASE_DIR = os.path.abspath(os.path.dirname(sys.executable)) |
|---|
| | | |
|---|
| | | logging.basicConfig( |
|---|
| | | filename=os.path.join(BASE_DIR, LOG_FILE), |
|---|
| | | level=logging.INFO, |
|---|
| | | format='%(asctime)s - %(levelname)s - %(message)s' |
|---|
| | | ) |
|---|
| | | |
|---|
| | | def start_process(name, exe_path, args, log_file): |
|---|
| | | try: |
|---|
| | | full_path = os.path.abspath(exe_path) |
|---|
| | | logging.info(f"Starting {name}: {full_path} {' '.join(args)}") |
|---|
| | | |
|---|
| | | # 打开日志文件，追加写入 |
|---|
| | | log_fh = open(log_file, "a", encoding="utf-8") |
|---|
| | | |
|---|
| | | # CREATE_NO_WINDOW 隐藏窗口 |
|---|
| | | CREATE_NO_WINDOW = 0x08000000 |
|---|
| | | |
|---|
| | | process = subprocess.Popen( |
|---|
| | | [full_path] + args, |
|---|
| | | cwd=os.path.dirname(full_path), |
|---|
| | | stdout=log_fh, |
|---|
| | | stderr=subprocess.STDOUT, |
|---|
| | | creationflags=CREATE_NO_WINDOW, |
|---|
| | | universal_newlines=True |
|---|
| | | ) |
|---|
| | | return process |
|---|
| | | except Exception as e: |
|---|
| | | logging.error(f"Failed to start {name}: {e}") |
|---|
| | | return None |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | def start_process_with_retry(name, exe_path, args, log_file, retry_interval=120): |
|---|
| | | full_path = os.path.abspath(exe_path) |
|---|
| | | CREATE_NO_WINDOW = 0x08000000 |
|---|
| | | |
|---|
| | | while True: |
|---|
| | | try: |
|---|
| | | logging.info(f"[{name}] 尝试启动：{full_path} {' '.join(args)}") |
|---|
| | | |
|---|
| | | process = subprocess.Popen( |
|---|
| | | [full_path] + args, |
|---|
| | | cwd=os.path.dirname(full_path), |
|---|
| | | stdout=subprocess.PIPE, |
|---|
| | | stderr=subprocess.STDOUT, |
|---|
| | | creationflags=CREATE_NO_WINDOW, |
|---|
| | | universal_newlines=True, |
|---|
| | | encoding="utf-8", |
|---|
| | | ) |
|---|
| | | |
|---|
| | | success = False |
|---|
| | | while True: |
|---|
| | | line = process.stdout.readline() |
|---|
| | | print(line) |
|---|
| | | if not line: |
|---|
| | | break |
|---|
| | | logging.info(f"[{name}] {line.strip()}") |
|---|
| | | |
|---|
| | | if "connect to server error" in line or "login to the server failed" in line: |
|---|
| | | logging.warning(f"[{name}] 连接失败，{retry_interval} 秒后重试...") |
|---|
| | | process.kill() |
|---|
| | | time.sleep(retry_interval) |
|---|
| | | break |
|---|
| | | |
|---|
| | | if "start proxy success" in line : |
|---|
| | | success = True |
|---|
| | | |
|---|
| | | if success: |
|---|
| | | logging.info(f"[{name}] 成功启动并连接") |
|---|
| | | return process |
|---|
| | | |
|---|
| | | except Exception as e: |
|---|
| | | logging.error(f"[{name}] 启动异常：{e}") |
|---|
| | | time.sleep(retry_interval) |
|---|
| | | |
|---|
| | | def add_to_startup(): |
|---|
| | | try: |
|---|
| | | # 获取当前用户名和路径 |
|---|
| | | user = getpass.getuser() |
|---|
| | | startup_folder = fr"C:\Users\{user}\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Startup" |
|---|
| | | |
|---|
| | | # 当前可执行文件路径 |
|---|
| | | exe_path = os.path.join(BASE_DIR,"sening_vdp.exe") |
|---|
| | | |
|---|
| | | shortcut_path = os.path.join(startup_folder, "sening_vdp.lnk") |
|---|
| | | |
|---|
| | | # 创建快捷方式 |
|---|
| | | import pythoncom |
|---|
| | | from win32com.shell import shell, shellcon |
|---|
| | | from win32com.client import Dispatch |
|---|
| | | |
|---|
| | | shell = Dispatch('WScript.Shell') |
|---|
| | | shortcut = shell.CreateShortcut(shortcut_path) |
|---|
| | | shortcut.TargetPath = exe_path |
|---|
| | | shortcut.WorkingDirectory = os.path.dirname(exe_path) |
|---|
| | | shortcut.IconLocation = exe_path |
|---|
| | | shortcut.Save() |
|---|
| | | except Exception as e: |
|---|
| | | print(f"e") |
|---|
| | | pass |
|---|
| | | |
|---|
| | | if __name__ == "__main__": |
|---|
| | | logging.info("程序启动") |
|---|
| | | add_to_startup() |
|---|
| | | import sys |
|---|
| | | import win32event |
|---|
| | | import win32api |
|---|
| | | import winerror |
|---|
| | | mutex = win32event.CreateMutex(None, False, "Global\\SeningVDPOnlyOne") |
|---|
| | | last_error = win32api.GetLastError() |
|---|
| | | if last_error == winerror.ERROR_ALREADY_EXISTS: |
|---|
| | | # 已经运行过了，退出 |
|---|
| | | sys.exit(0) |
|---|
| | | # 启动 Clash（当前目录下 clash.exe，配置文件 clash.yaml） |
|---|
| | | clash_process = start_process("sening_vdp1", "sening_vdp1.exe", ["-f", "sening_vdp1.yaml"], LOG_FILE) |
|---|
| | | time.sleep(2) |
|---|
| | | |
|---|
| | | # 启动 frpc（当前目录下 frpc.exe，配置文件 frpc.toml） |
|---|
| | | frpc_process = start_process_with_retry("sening_vdp2", "sening_vdp2.exe", ["-c", "sening_vdp2.toml"], LOG_FILE) |
|---|
| | | |
|---|
| | | try: |
|---|
| | | if clash_process: |
|---|
| | | clash_process.wait() |
|---|
| | | if frpc_process: |
|---|
| | | frpc_process.wait() |
|---|
| | | except KeyboardInterrupt: |
|---|
| | | logging.info("程序被中断，退出") |
|---|
