289 lines
7.8 KiB
Python
289 lines
7.8 KiB
Python
import os
|
||
import re
|
||
import random
|
||
import shutil
|
||
import pathlib
|
||
import platform
|
||
from uuid import uuid4
|
||
from logger import info, warning, error
|
||
|
||
from config import CURSOR_PATH
|
||
|
||
|
||
# 颜色常量定义，保留用于日志输出
|
||
GREEN = "\033[92m"
|
||
RED = "\033[91m"
|
||
YELLOW = "\033[93m"
|
||
BLUE = "\033[96m"
|
||
PURPLE = "\033[95m"
|
||
RESET = "\033[0m"
|
||
|
||
SYSTEM = platform.system()
|
||
if SYSTEM not in ("Windows", "Linux", "Darwin"):
|
||
raise OSError(f"不支持的操作系统: {SYSTEM}")
|
||
|
||
|
||
def uuid():
|
||
"""生成随机UUID"""
|
||
return str(uuid4())
|
||
|
||
|
||
def path(path_str):
|
||
"""获取绝对路径"""
|
||
return pathlib.Path(path_str).resolve()
|
||
|
||
|
||
def randomuuid(randomuuid_str):
|
||
"""获取随机UUID，如果提供则使用提供的值"""
|
||
if not randomuuid_str:
|
||
randomuuid_str = uuid()
|
||
return randomuuid_str
|
||
|
||
|
||
def random_mac():
|
||
"""生成随机MAC地址"""
|
||
mac = [
|
||
0x00,
|
||
0x16,
|
||
0x3E,
|
||
random.randint(0x00, 0x7F),
|
||
random.randint(0x00, 0xFF),
|
||
random.randint(0x00, 0xFF),
|
||
]
|
||
return ":".join(map(lambda x: "%02x" % x, mac))
|
||
|
||
|
||
def load(file_path: pathlib.Path):
|
||
"""加载文件内容"""
|
||
with open(file_path, "rb") as f:
|
||
return f.read()
|
||
|
||
|
||
def save(file_path: pathlib.Path, data: bytes):
|
||
"""保存文件内容"""
|
||
with open(file_path, "wb") as f:
|
||
f.write(data)
|
||
|
||
|
||
def backup(file_path: pathlib.Path):
|
||
"""备份文件"""
|
||
backup_path = file_path.with_suffix(file_path.suffix + ".bak")
|
||
if not backup_path.exists():
|
||
shutil.copy2(file_path, backup_path)
|
||
print(f"已备份 {file_path} -> {backup_path}")
|
||
|
||
|
||
def replace(data: bytes, pattern: str, replace_str: str, probe: str = None) -> bytes:
|
||
"""替换文件内容"""
|
||
pattern_bytes = pattern.encode() if isinstance(pattern, str) else pattern
|
||
replace_bytes = (
|
||
replace_str.encode() if isinstance(replace_str, str) else replace_str
|
||
)
|
||
|
||
if probe:
|
||
probe_bytes = probe.encode() if isinstance(probe, str) else probe
|
||
if re.search(probe_bytes, data):
|
||
print("检测到已经被修补的代码，跳过...")
|
||
return data
|
||
|
||
return re.sub(pattern_bytes, replace_bytes, data)
|
||
|
||
|
||
def find_main_js():
|
||
"""查找Cursor的main.js文件"""
|
||
error(f"SYSTEM: {SYSTEM}")
|
||
if SYSTEM == "Windows":
|
||
localappdata = os.getenv("LOCALAPPDATA")
|
||
if not localappdata:
|
||
raise OSError("环境变量 %LOCALAPPDATA% 不存在")
|
||
|
||
# 使用本地变量保存路径
|
||
cursor_path = CURSOR_PATH
|
||
if not cursor_path:
|
||
error("当前windows系统, 环境变量 CURSOR_PATH 不存在，使用默认路径")
|
||
cursor_path = os.getenv("LOCALAPPDATA", "")
|
||
else:
|
||
info(f"当前windows系统, CURSOR_PATH: {cursor_path}")
|
||
|
||
# 常见的Cursor安装路径
|
||
paths = [
|
||
path(os.path.join(cursor_path, "resources", "app", "out", "main.js")),
|
||
path(
|
||
os.path.join(
|
||
localappdata,
|
||
"Programs",
|
||
"cursor",
|
||
"resources",
|
||
"app",
|
||
"out",
|
||
"main.js",
|
||
)
|
||
),
|
||
path(
|
||
os.path.join(
|
||
localappdata, "cursor", "resources", "app", "out", "main.js"
|
||
)
|
||
),
|
||
]
|
||
|
||
for p in paths:
|
||
info(f"检查路径: {p}")
|
||
if p.exists():
|
||
info(f"找到main.js: {p}")
|
||
return p
|
||
else:
|
||
warning(f"路径不存在: {p}")
|
||
|
||
elif SYSTEM == "Darwin": # macOS
|
||
paths = [
|
||
path("/Applications/Cursor.app/Contents/Resources/app/out/main.js"),
|
||
path(
|
||
os.path.expanduser(
|
||
"~/Applications/Cursor.app/Contents/Resources/app/out/main.js"
|
||
)
|
||
),
|
||
]
|
||
|
||
for p in paths:
|
||
if p.exists():
|
||
return p
|
||
|
||
elif SYSTEM == "Linux":
|
||
# Linux上常见的安装路径
|
||
paths = [
|
||
path("/usr/share/cursor/resources/app/out/main.js"),
|
||
path(os.path.expanduser("~/.local/share/cursor/resources/app/out/main.js")),
|
||
]
|
||
|
||
for p in paths:
|
||
if p.exists():
|
||
return p
|
||
|
||
raise FileNotFoundError("无法找到Cursor的main.js文件，请手动指定路径")
|
||
|
||
|
||
def patch_cursor(
|
||
js_path=None, machine_id=None, mac_addr=None, sqm_id=None, dev_id=None
|
||
):
|
||
"""
|
||
修补Cursor的main.js文件，替换机器ID等识别信息
|
||
|
||
参数:
|
||
js_path: main.js文件路径，如果为None则自动查找
|
||
machine_id: 机器ID，如果为None则随机生成
|
||
mac_addr: MAC地址，如果为None则随机生成
|
||
sqm_id: Windows SQM ID，如果为None则使用空字符串
|
||
dev_id: 设备ID，如果为None则随机生成
|
||
|
||
返回:
|
||
bool: 是否成功
|
||
"""
|
||
try:
|
||
# 查找main.js文件
|
||
if not js_path:
|
||
js_path = find_main_js()
|
||
else:
|
||
js_path = path(js_path)
|
||
|
||
# 如果找不到main.js文件
|
||
if not js_path.exists():
|
||
print(f"错误: 找不到文件 {js_path}")
|
||
return False
|
||
|
||
print(f"找到main.js文件: {js_path}")
|
||
|
||
# 随机生成ID
|
||
machine_id = randomuuid(machine_id)
|
||
mac_addr = mac_addr or random_mac()
|
||
sqm_id = sqm_id or ""
|
||
dev_id = randomuuid(dev_id)
|
||
|
||
# 加载文件内容
|
||
data = load(js_path)
|
||
|
||
# 备份文件
|
||
backup(js_path)
|
||
|
||
# 替换机器ID
|
||
data = replace(
|
||
data,
|
||
r"=.{0,50}timeout.{0,10}5e3.*?,",
|
||
f'=/*csp1*/"{machine_id}"/*1csp*/,',
|
||
r"=/\*csp1\*/.*?/\*1csp\*/,",
|
||
)
|
||
|
||
# 替换MAC地址
|
||
data = replace(
|
||
data,
|
||
r"(function .{0,50}\{).{0,300}Unable to retrieve mac address.*?(\})",
|
||
f'\\1return/*csp2*/"{mac_addr}"/*2csp*/;\\2',
|
||
r"()return/\*csp2\*/.*?/\*2csp\*/;()",
|
||
)
|
||
|
||
# 替换SQM ID
|
||
data = replace(
|
||
data,
|
||
r'return.{0,50}\.GetStringRegKey.*?HKEY_LOCAL_MACHINE.*?MachineId.*?\|\|.*?""',
|
||
f'return/*csp3*/"{sqm_id}"/*3csp*/',
|
||
r"return/\*csp3\*/.*?/\*3csp\*/",
|
||
)
|
||
|
||
# 替换设备ID
|
||
data = replace(
|
||
data,
|
||
r"return.{0,50}vscode\/deviceid.*?getDeviceId\(\)",
|
||
f'return/*csp4*/"{dev_id}"/*4csp*/',
|
||
r"return/\*csp4\*/.*?/\*4csp\*/",
|
||
)
|
||
|
||
# 保存修改后的文件
|
||
save(js_path, data)
|
||
|
||
print(f"成功修补 {js_path}")
|
||
print(f"机器ID: {machine_id}")
|
||
print(f"MAC地址: {mac_addr}")
|
||
print(f"SQM ID: {sqm_id}")
|
||
print(f"设备ID: {dev_id}")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"错误: {str(e)}")
|
||
import traceback
|
||
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
|
||
class CursorShadowPatcher:
|
||
"""Cursor机器标识修改器"""
|
||
|
||
@staticmethod
|
||
def reset_machine_ids():
|
||
"""重置所有机器标识"""
|
||
return patch_cursor()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 作为独立脚本运行时，执行交互式修补
|
||
print(f"\n{'=' * 50}")
|
||
print("Cursor 机器标识重置工具 (Shadow Patch 增强版)")
|
||
print(f"{'=' * 50}")
|
||
|
||
js_path = input("请输入main.js路径 (留空=自动检测): ")
|
||
machine_id = input("机器ID (留空=随机生成): ")
|
||
mac_addr = input("MAC地址 (留空=随机生成): ")
|
||
sqm_id = input("Windows SQM ID (留空=使用空值): ")
|
||
dev_id = input("设备ID (留空=随机生成): ")
|
||
|
||
success = patch_cursor(js_path, machine_id, mac_addr, sqm_id, dev_id)
|
||
|
||
if success:
|
||
print(f"\n{'=' * 50}")
|
||
print("修补成功!")
|
||
else:
|
||
print(f"\n{'=' * 50}")
|
||
print("修补失败!")
|
||
|
||
input("按回车键退出...")
|