@echo off
chcp 65001 >nul
title 极域U盘禁用一键恢复_Win10专业版_含组策略全重置
color 0A
echo ==============================================
echo          极域U盘禁用一键恢复脚本
echo          适配：Win10专业版 丨 需管理员权限
echo          含：组策略全项未配置+注册表+驱动修复
echo          执行中请勿关闭窗口，等待自动完成！
echo ==============================================
echo.

:: 1. 强制结束极域学生端进程（核心，防止实时还原设置）
echo 【1/7】正在结束极域学生端进程...
taskkill /f /im StudentMain.exe >nul 2>&1
echo 极域进程结束完成！
echo.

:: 2. 还原USBSTOR核心注册表（多ControlSet，避免重启失效）
echo 【2/7】正在还原USB存储驱动注册表...
reg add "HKLM\SYSTEM\CurrentControlSet\Services\USBSTOR" /v Start /t REG_DWORD /d 3 /f >nul
reg add "HKLM\SYSTEM\ControlSet001\Services\USBSTOR" /v Start /t REG_DWORD /d 3 /f >nul
reg add "HKLM\SYSTEM\ControlSet002\Services\USBSTOR" /v Start /t REG_DWORD /d 3 /f >nul
echo USB驱动注册表还原完成！
echo.

:: 3. 重置极域自身USB配置注册表
echo 【3/7】正在恢复极域自身USB配置...
reg add "HKLM\SOFTWARE\TopDomain\e-learningClassStandard" /v USBEnable /t REG_DWORD /d 1 /f >nul 2>&1
reg add "HKLM\SOFTWARE\TopDomain\e-learningClassStandard" /v USBDiskEnable /t REG_DWORD /d 1 /f >nul 2>&1
reg add "HKCU\SOFTWARE\TopDomain\e-learningClassStandard" /v USBEnable /t REG_DWORD /d 1 /f >nul 2>&1
reg add "HKCU\SOFTWARE\TopDomain\e-learningClassStandard" /v USBDiskEnable /t REG_DWORD /d 1 /f >nul 2>&1
echo 极域配置恢复完成！
echo.

:: 4. 核心：组策略可移动存储访问全项设为未配置（删除对应策略注册表项）
echo 【4/7】正在重置组策略（所有项设为未配置）...
:: 删除计算机配置下可移动存储访问所有策略
reg delete "HKLM\SOFTWARE\Policies\Microsoft\Windows\RemovableStorageDevices" /f >nul 2>&1
:: 额外清理组策略模板缓存项，避免残留
reg delete "HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\Group Policy\State\Machine\Scripts" /f >nul 2>&1
reg delete "HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\Group Policy\State\User\Scripts" /f >nul 2>&1
echo 组策略全项已设为【未配置】！
echo.

:: 5. 修复USB驱动文件（usbstor.inf/pnf）权限
echo 【5/7】正在修复USB驱动文件权限...
icacls "C:\Windows\inf\usbstor.inf" /remove:d Everyone >nul 2>&1
icacls "C:\Windows\inf\usbstor.inf" /grant SYSTEM:(R,X) Administrators:(R,X) Users:(R) /f >nul 2>&1
icacls "C:\Windows\inf\usbstor.pnf" /remove:d Everyone >nul 2>&1
icacls "C:\Windows\inf\usbstor.pnf" /grant SYSTEM:(R,X) Administrators:(R,X) Users:(R) /f >nul 2>&1
echo 驱动文件权限修复完成！
echo.

:: 6. 启动USBSTOR服务+强制刷新组策略
echo 【6/7】正在启动USB服务并刷新组策略...
sc config USBSTOR start= demand >nul
sc start USBSTOR >nul 2>&1
:: 强制刷新组策略，立即生效
gpupdate /force >nul 2>&1
echo USB服务启动完成！组策略已强制刷新！
echo.

:: 7. 扫描硬件改动，重新识别USB设备
echo 【7/7】正在扫描硬件改动，重新加载USB设备...
devcon rescan >nul 2>&1
echo 硬件扫描完成！
echo.
echo ==============================================
echo          所有恢复操作执行完毕！✅
echo          关键说明：
echo          1. 组策略已全部设为【未配置】
echo          2. 建议重启电脑后插入U盘验证
echo          3. 若仍异常，彻底卸载极域学生端后重试
echo ==============================================
pause >nul