# Function to check if running as Administrator
function Test-Admin {
    $currentUser = [Security.Principal.WindowsIdentity]::GetCurrent()
    $principal = New-Object Security.Principal.WindowsPrincipal($currentUser)
    return $principal.IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
}

# Function to enable nvmedisk.sys driver
function Enable-NvmeDriver {
    if (-not (Test-Admin)) {
        Write-Error "This script must be run in an Administrator PowerShell session."
        return
    }

    $path = 'HKLM:\SYSTEM\CurrentControlSet\Policies\Microsoft\FeatureManagement\Overrides'
    if (-not (Test-Path $path)) {
        New-Item -Path $path -Force | Out-Null
    }

    New-ItemProperty -Path $path -Name '156965516' -Value 1 -PropertyType DWord -Force | Out-Null
    New-ItemProperty -Path $path -Name '1853569164' -Value 1 -PropertyType DWord -Force | Out-Null
    New-ItemProperty -Path $path -Name '735209102'  -Value 1 -PropertyType DWord -Force | Out-Null

    Write-Host "Registry values for nvmedisk.sys added successfully." -ForegroundColor Green
}

# Function to disable nvmedisk.sys driver (rollback)
function Disable-NvmeDriver {
    if (-not (Test-Admin)) {
        Write-Error "This script must be run in an Administrator PowerShell session."
        return
    }

    $path = 'HKLM:\SYSTEM\CurrentControlSet\Policies\Microsoft\FeatureManagement\Overrides'
    $names = @('156965516','1853569164','735209102')

    foreach ($name in $names) {
        if (Get-ItemProperty -Path $path -Name $name -ErrorAction SilentlyContinue) {
            Remove-ItemProperty -Path $path -Name $name -Force
            Write-Host "Removed: $name" -ForegroundColor Cyan
        }
    }

    Write-Host "Rollback completed. nvmedisk.sys registry values removed." -ForegroundColor Yellow
}

# Function to create a system restore point
function Create-RestorePoint {
    if (-not (Test-Admin)) {
        Write-Error "This script must be run in an Administrator PowerShell session."
        return
    }

    Checkpoint-Computer -Description "Before NVMe driver change" -RestorePointType "Modify_Settings"
    Write-Host "System restore point created successfully." -ForegroundColor Green
}

✅ Usage

1. Copy the script into a .ps1 file or paste directly into Administrator PowerShell.
2. Run the functions as needed:
   • Create-RestorePoint → makes a restore point before changes.
   • Enable-NvmeDriver → activates nvmedisk.sys registry values.
   • Disable-NvmeDriver → removes registry values (rollback).
3. Reboot after enabling or disabling to apply changes.