fix: restore automatic unload on normal backend shutdown

Move the unload sequence back into a detached autouninstall child process so the backend can exit normally while still unloading OpenSysKit and DriverLoader automatically.

Made-with: Cursor

Author: yxc0915

cmd/server/autouninstall_mode_windows.go

@@ -4,8 +4,149 @@ package main
 
 import (
 	"fmt"
+	"log"
+	"os"
+	"sort"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/OpenSysKit/backend/internal/driver"
 )
 
 func runAutoUninstallMode() error {
-	return fmt.Errorf("autouninstall 已禁用：请使用显式 uninstall 命令执行完整卸载，避免主进程退出阶段蓝屏")
+	delay := parseAutoUninstallDelay()
+	handles, err := parseAutoUninstallHandles()
+	if err != nil {
+		return err
+	}
+	if len(handles) == 0 {
+		return fmt.Errorf("autouninstall 未提供可卸载的 handle")
+	}
+
+	log.Printf("autouninstall 子进程启动: delay=%s handles=%s", delay, formatHandleList(handles))
+	if delay > 0 {
+		time.Sleep(delay)
+	}
+
+	log.Printf("autouninstall 开始执行完整卸载")
+	if err := runAutoUninstall(handles); err != nil {
+		return fmt.Errorf("autouninstall 卸载失败: %w", err)
+	}
+	log.Printf("autouninstall 卸载完成")
+	return nil
+}
+
+func runAutoUninstall(handles []uint64) error {
+	loader, err := driver.OpenExistingLoader()
+	if err != nil {
+		return fmt.Errorf("autouninstall 无法连接 WinDrive(\\\\.\\DriverLoader): %w", err)
+	}
+	defer loader.Close()
+
+	drivers, err := loader.ListMappedDrivers()
+	if err != nil {
+		return err
+	}
+	if len(drivers) == 0 {
+		log.Printf("autouninstall 未检测到映射驱动，跳过 OpenSysKit 卸载")
+	} else {
+		existing := make(map[uint64]struct{}, len(drivers))
+		for _, row := range drivers {
+			existing[row.Handle] = struct{}{}
+		}
+
+		filtered := make([]uint64, 0, len(handles))
+		for _, h := range dedupHandles(handles) {
+			if _, ok := existing[h]; ok {
+				filtered = append(filtered, h)
+			} else {
+				log.Printf("autouninstall 跳过不存在的 handle=%d (当前映射: %s)", h, formatHandles(drivers))
+			}
+		}
+		if len(filtered) == 0 {
+			return fmt.Errorf("autouninstall 指定 handle 均不存在，当前映射: %s", formatHandles(drivers))
+		}
+
+		sort.Slice(filtered, func(i, j int) bool { return filtered[i] > filtered[j] })
+		if waitForDeviceRelease(`\\.\OpenSysKit`, 15*time.Second) {
+			log.Println("autouninstall 检测到设备引用已释放，开始卸载")
+		} else {
+			log.Println("autouninstall 警告: 设备可能仍被占用，仍尝试卸载")
+		}
+
+		log.Printf("autouninstall 计划卸载 handle: %s", formatHandleList(filtered))
+		for _, handle := range filtered {
+			if err = unloadHandleWithRetry(loader, handle, 5); err != nil {
+				return err
+			}
+		}
+	}
+
+	after, err := loader.ListMappedDrivers()
+	if err != nil {
+		return err
+	}
+	if len(after) != 0 {
+		return fmt.Errorf("autouninstall 卸载后仍存在映射驱动: %s", formatHandles(after))
+	}
+
+	log.Println("autouninstall 下发 WinDrive allow-unload")
+	if err = loader.AllowUnload(); err != nil {
+		return err
+	}
+
+	loader.Close()
+	log.Println("autouninstall 卸载 DriverLoader 服务")
+	if err = driver.UninstallLoaderService(); err != nil {
+		return err
+	}
+	return nil
+}
+
+func parseAutoUninstallDelay() time.Duration {
+	for _, arg := range os.Args[2:] {
+		if strings.HasPrefix(arg, "--delay=") {
+			raw := strings.TrimSpace(strings.TrimPrefix(arg, "--delay="))
+			if raw == "" {
+				continue
+			}
+			seconds, err := strconv.Atoi(raw)
+			if err != nil || seconds < 0 {
+				log.Printf("警告: 非法 autouninstall delay 参数 %q，改用 1 秒", raw)
+				return time.Second
+			}
+			if seconds == 0 {
+				return 0
+			}
+			return time.Duration(seconds) * time.Second
+		}
+	}
+	return time.Second
+}
+
+func parseAutoUninstallHandles() ([]uint64, error) {
+	for _, arg := range os.Args[2:] {
+		if strings.HasPrefix(arg, "--handles=") {
+			raw := strings.TrimSpace(strings.TrimPrefix(arg, "--handles="))
+			if raw == "" {
+				return nil, nil
+			}
+			parts := strings.Split(raw, ",")
+			handles := make([]uint64, 0, len(parts))
+			for _, part := range parts {
+				part = strings.TrimSpace(part)
+				if part == "" {
+					continue
+				}
+				h, err := strconv.ParseUint(part, 10, 64)
+				if err != nil {
+					return nil, fmt.Errorf("解析 autouninstall handle 失败 %q: %w", part, err)
+				}
+				handles = append(handles, h)
+			}
+			return handles, nil
+		}
+	}
+	return nil, nil
 }

cmd/server/main.go

@@ -192,15 +192,21 @@ func main() {
 		loader = nil
 	}
 
-	// 卸载不能再放在主进程退出阶段自动执行：
-	// 在当前系统环境下会触发 KiSystemServiceExitPico / APC_INDEX_MISMATCH 蓝屏。
-	// 因此这里只释放本进程句柄，并明确提示使用显式 uninstall 命令完成完整卸载链路。
+	// 自动卸载仍由独立子进程执行，但主进程只负责调度并正常退出，
+	// 避免在当前进程退出临界区直接执行卸载链路触发蓝屏。
+	if !autoUninstallEnabled() {
+		log.Printf("自动卸载已禁用: mapped_by_this_process=%t, handles=%s，请手动执行 OpenSysKit.exe uninstall", mappedByThisProcess, formatHandleList(mappedHandles))
+		log.Println("主进程已完成资源释放，正常退出")
+		return
+	}
+
 	if mappedByThisProcess && len(mappedHandles) > 0 {
-		log.Printf("已跳过退出阶段自动卸载以避免蓝屏: mapped_by_this_process=%t, handles=%s，请稍后手动执行 OpenSysKit.exe uninstall --handles=%s",
-			mappedByThisProcess,
-			formatHandleList(mappedHandles),
-			formatHandleList(mappedHandles),
-		)
+		log.Printf("调度自动卸载子进程: handles=%s", formatHandleList(mappedHandles))
+		if err := scheduleSelfUninstall(1*time.Second, mappedHandles); err != nil {
+			log.Printf("警告: 调度自动卸载失败: %v，请手动执行 OpenSysKit.exe uninstall --handles=%s", err, formatHandleList(mappedHandles))
+		} else {
+			log.Printf("自动卸载子进程已启动，将执行 WinDrive 安全卸载链路: handles=%s", formatHandleList(mappedHandles))
+		}
 	} else {
 		log.Printf("退出时无需卸载映射驱动: mapped_by_this_process=%t, handles=%s", mappedByThisProcess, formatHandleList(mappedHandles))
 	}