From f3933dc398f889414c003ec3c961793c05ac827e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=85=8E=E9=A5=BC=E6=9E=9C=E5=AD=90=E5=8D=B7=E9=B2=A8?= =?UTF-8?q?=E9=B1=BC=E8=BE=A3=E6=A4=92?= Date: Mon, 10 Feb 2025 17:01:50 +0800 Subject: [PATCH] refactor: Enhance Cursor app modification with robust file handling and error management - Significantly improved file modification process with comprehensive error checking - Added advanced file copying techniques using pv or rsync with progress tracking - Implemented detailed validation for file copy integrity - Enhanced signature removal process with timeout and fallback mechanisms - Added granular logging for each modification step - Improved error handling for Helper process signature removal - Implemented backup and restore strategy for modified files - Added more informative error messages and user guidance --- scripts/run/cursor_mac_id_modifier.sh | 152 ++++++++++++++++++-------- 1 file changed, 109 insertions(+), 43 deletions(-) diff --git a/scripts/run/cursor_mac_id_modifier.sh b/scripts/run/cursor_mac_id_modifier.sh index 56fd5e6..87f6ef4 100644 --- a/scripts/run/cursor_mac_id_modifier.sh +++ b/scripts/run/cursor_mac_id_modifier.sh @@ -297,104 +297,168 @@ generate_new_config() { modify_cursor_app_files() { log_info "正在安全修改 Cursor 主程序文件..." + # 验证应用是否存在 + if [ ! -d "$CURSOR_APP_PATH" ]; then + log_error "未找到 Cursor.app，请确认安装路径: $CURSOR_APP_PATH" + return 1 + } + # 创建临时工作目录 local timestamp=$(date +%Y%m%d_%H%M%S) local temp_dir="/tmp/cursor_mod_${timestamp}" local temp_app="${temp_dir}/Cursor.app" - # 复制应用到临时目录 - log_info "创建临时工作副本..." + # 清理可能存在的旧临时目录 + [ -d "$temp_dir" ] && rm -rf "$temp_dir" + + # 创建新的临时目录 mkdir -p "$temp_dir" || { - log_error "无法创建临时目录" + log_error "无法创建临时目录: $temp_dir" return 1 } - log_debug "正在复制应用文件: $CURSOR_APP_PATH -> $temp_dir" - rsync -a --progress "$CURSOR_APP_PATH/" "$temp_app/" || { + + # 复制应用到临时目录 + log_info "创建临时工作副本..." + { + # 使用 pv 显示进度（如果可用） + if command -v pv >/dev/null 2>&1; then + tar cf - -C "$(dirname "$CURSOR_APP_PATH")" "$(basename "$CURSOR_APP_PATH")" | pv -s $(du -sb "$CURSOR_APP_PATH" | awk '{print $1}') | tar xf - -C "$temp_dir" + else + # 使用 rsync 显示进度 + rsync -av --progress "$CURSOR_APP_PATH/" "$temp_app/" + fi + } || { log_error "复制应用失败" rm -rf "$temp_dir" return 1 } + + # 验证复制完整性 + local orig_size=$(du -s "$CURSOR_APP_PATH" | cut -f1) + local temp_size=$(du -s "$temp_app" | cut -f1) + if [ "$orig_size" != "$temp_size" ]; then + log_error "文件复制不完整，大小不匹配" + rm -rf "$temp_dir" + return 1 + } # 移除签名（增强兼容性） log_info "移除应用签名..." - find "$temp_app" -type f -exec chmod 644 {} \; - find "$temp_app" -type d -exec chmod 755 {} \; - timeout 120 codesign --remove-signature --all-architectures "$temp_app" > >(while read line; do echo -n .; done) 2>&1 || { - log_warn "签名移除超时，尝试快速模式..." - codesign --remove-signature --preserve-metadata=entitlements "$temp_app" >/dev/null 2>&1 || true + { + # 设置文件权限 + find "$temp_app" -type f -exec chmod 644 {} \; + find "$temp_app" -type d -exec chmod 755 {} \; + + # 移除主应用签名 + if ! timeout 60 codesign --remove-signature "$temp_app" 2>/dev/null; then + log_warn "标准签名移除超时，尝试快速模式..." + codesign --remove-signature --force "$temp_app" 2>/dev/null || true + fi + + # 处理所有Helper进程 + local helper_count=0 + while IFS= read -r helper; do + ((helper_count++)) + log_debug "处理Helper进程 ($helper_count): $helper" + timeout 30 codesign --remove-signature "$helper" 2>/dev/null || { + log_warn "移除Helper签名失败: $(basename "$helper")" + codesign --remove-signature --force "$helper" 2>/dev/null || true + } + done < <(find "${temp_app}/Contents/Frameworks" -name "*Helper*.app") + + log_info "已处理 $helper_count 个Helper进程" } - # 处理所有Helper进程 - find "${temp_app}/Contents/Frameworks" -name "*Helper*.app" | while read helper; do - log_debug "处理Helper进程: $helper" - timeout 30 codesign --remove-signature "$helper" >/dev/null 2>&1 || { - log_warn "无法移除 $helper 签名，尝试强制模式..." - codesign --remove-signature --force "$helper" >/dev/null 2>&1 || true - } - done - # 修改目标文件 - local files=("${temp_app}/Contents/Resources/app/out/main.js" - "${temp_app}/Contents/Resources/app/out/vs/code/node/cliProcessMain.js") + local modified_count=0 + local files=( + "${temp_app}/Contents/Resources/app/out/main.js" + "${temp_app}/Contents/Resources/app/out/vs/code/node/cliProcessMain.js" + ) for file in "${files[@]}"; do - [ ! -f "$file" ] && { + if [ ! -f "$file" ]; then log_warn "文件不存在: ${file/$temp_dir\//}" continue } - # 使用精确位置替换（避免sed正则问题） log_debug "处理文件: ${file/$temp_dir\//}" + + # 创建文件备份 + cp "$file" "${file}.bak" || { + log_error "无法创建文件备份: ${file/$temp_dir\//}" + continue + } + + # 使用精确位置替换 local content=$(<"$file") local uuid_pos=$(grep -b -o "IOPlatformUUID" <<< "$content" | cut -d: -f1) - [ -z "$uuid_pos" ] && { - log_warn "未找到IOPlatformUUID" + + if [ -z "$uuid_pos" ]; then + log_warn "未找到IOPlatformUUID标记: ${file/$temp_dir\//}" continue } # 定位最近的switch语句 local switch_pos=$(grep -b -o "switch" <<< "${content:0:$uuid_pos}" | tail -1 | cut -d: -f1) - [ -z "$switch_pos" ] && { - log_warn "未找到switch语句" + + if [ -z "$switch_pos" ]; then + log_warn "未找到switch语句: ${file/$temp_dir\//}" continue } # 精确修改内容 - new_content="${content:0:$switch_pos}return crypto.randomUUID();\n${content:$switch_pos}" - if ! printf "%b" "$new_content" > "$file"; then - log_error "文件写入失败" - return 1 + local new_content="${content:0:$switch_pos}return crypto.randomUUID();\n${content:$switch_pos}" + if printf "%b" "$new_content" > "$file"; then + ((modified_count++)) + log_info "成功修改文件: ${file/$temp_dir\//}" + else + log_error "文件写入失败: ${file/$temp_dir\//}" + mv "${file}.bak" "$file" fi - log_info "文件修改成功: ${file/$temp_dir\//}" + + # 清理备份 + rm -f "${file}.bak" done + if [ "$modified_count" -eq 0 ]; then + log_error "未能成功修改任何文件" + rm -rf "$temp_dir" + return 1 + } + # 重新签名应用 log_info "重新签名应用..." - xattr -cr "$temp_app" # 清除扩展属性 - codesign --force --deep --sign - --preserve-metadata=entitlements,identifier,flags "$temp_app" >/dev/null 2>&1 || { - log_warn "重新签名失败，尝试以下方法：" - log_warn "1. 前往 系统设置 -> 隐私与安全性" - log_warn "2. 在『安全性』中找到 Cursor 相关提示" - log_warn "3. 点击『仍要打开』" + { + # 清除扩展属性 + xattr -cr "$temp_app" + + # 尝试重新签名 + if ! codesign --force --deep --sign - --preserve-metadata=entitlements,identifier,flags "$temp_app" 2>/dev/null; then + log_warn "应用重新签名失败，您可能需要：" + log_warn "1. 前往 系统设置 -> 隐私与安全性" + log_warn "2. 在『安全性』中找到 Cursor 相关提示" + log_warn "3. 点击『仍要打开』" + fi } # 创建应用备份 local backup_app="/Applications/Cursor.backup.${timestamp}.app" log_info "创建应用备份: ${backup_app/$HOME/\~}" - mv "$CURSOR_APP_PATH" "$backup_app" || { + if ! mv "$CURSOR_APP_PATH" "$backup_app"; then log_error "创建备份失败" rm -rf "$temp_dir" return 1 - } + fi # 替换原应用 log_info "安装修改版应用..." - ditto "$temp_app" "$CURSOR_APP_PATH" || { + if ! ditto "$temp_app" "$CURSOR_APP_PATH"; then log_error "应用替换失败，正在恢复..." mv "$backup_app" "$CURSOR_APP_PATH" rm -rf "$temp_dir" return 1 - } + fi # 清理临时文件 rm -rf "$temp_dir" @@ -404,10 +468,12 @@ modify_cursor_app_files() { find "$CURSOR_APP_PATH" -type d -exec chmod 755 {} \; find "$CURSOR_APP_PATH" -type f -exec chmod 644 {} \; - log_info "正在重建LaunchServices数据库..." + # 重建 LaunchServices 数据库 + log_info "正在重建 LaunchServices 数据库..." /System/Library/Frameworks/CoreServices.framework/Frameworks/LaunchServices.framework/Support/lsregister -f "$CURSOR_APP_PATH" log_info "Cursor 主程序文件修改完成！原版备份在: ${backup_app/$HOME/\~}" + return 0 } # 显示文件树结构