@echo off
:: USB Write Protection Toggle Tool
net session >nul 2>&1
if %errorlevel% neq 0 (
    echo Please run this script as Administrator.
    pause
    exit
)

echo ==========================================
echo USB Write Protection - Enable/Disable Tool
echo ==========================================
echo.
echo 1. Enable Write Protection
echo 2. Disable Write Protection
echo 3. Exit
echo.
set /p choice="Enter your choice (1/2/3): "
set choice=%choice:~0,1%

if "%choice%"=="1" goto enable
if "%choice%"=="2" goto disable
if "%choice%"=="3" exit
goto invalid

:enable
reg add "HKLM\SYSTEM\CurrentControlSet\Control\StorageDevicePolicies" /f
reg add "HKLM\SYSTEM\CurrentControlSet\Control\StorageDevicePolicies" /v WriteProtect /t REG_DWORD /d 1 /f
echo Write Protection has been ENABLED.
goto end

:disable
reg add "HKLM\SYSTEM\CurrentControlSet\Control\StorageDevicePolicies" /f
reg add "HKLM\SYSTEM\CurrentControlSet\Control\StorageDevicePolicies" /v WriteProtect /t REG_DWORD /d 0 /f
echo Write Protection has been DISABLED.
goto end

:invalid
echo Invalid choice. Please run the script again.
goto end

:end
pause
exit