emoose/UpdateDLSS.ps1

## UpdateDLSS.ps1
# Global DLSS update script by emoose - https://gist.github.com/emoose/11271bbb3b42fb3b1b0e1c83eef47c05
# Allows setting up driver to use a single global DLSS DLL for majority of DLSS2/3 games
# If the global version is newer than the one included with game, it should get loaded automatically
# (how this works: https://forums.guru3d.com/threads/.439761/page-143#post-6221767)
#
# Almost all DLSS3 games should work, DLSS2 is hit-and-miss
# (DLSS2 games with customized appid probably won't work, thanks to some pointless nvngx checks)
# DLSSD/DLSSG should work fine with nearly all games (re-run the script for each DLL you want to update)
#
# Usage:
# - press the Raw button on the right side, select all contents with CTRL+A, copy with CTRL+C (don't download as file as PS won't allow execution)
# - paste contents into notepad, save as UpdateDLSS.ps1
# - find the saved script, right click -> "Run with PowerShell"
# - when asked for DLL path, drag+drop an nvngx_dlss file into the window, and press enter
# - if Run with PowerShell option doesn't appear, you can also run through command prompt or powershell: "powershell C:\Users\emoose\Downloads\UpdateDLSS.ps1"
#
# To undo the changes from this script just clear out the "C:\ProgramData\NVIDIA\NGX\models" folder
# This folder should regen the next time DLSS is used, and games will go back to using the DLL from their game folder
#
# Advanced setup (installing as PowerShell module):
# - save script as UpdateDLSS.psm1 into powershell modules folder, so you can use it from any location
#    Depending on PS version, save to either:
#      - Documents\WindowsPowerShell\Modules\UpdateDLSS\UpdateDLSS.psm1
#      - Documents\PowerShell\Modules\UpdateDLSS\UpdateDLSS.psm1
#    Alternatively System32 should work for all PS versions:
#      - C:\Windows\system32\WindowsPowerShell\v1.0\Modules\UpdateDLSS\UpdateDLSS.psm1
#    With it saved into modules folder, you should be able to open a new powershell window and then run "UpdateDLSS" from any location
#    (or run "UpdateDLSS -DllPath C:\DLLs\nvngx_dlss.dll")
#
# If ran without -DllPath param you'll be prompted for the path to the DLSS DLL
# When prompted you can usually just drag+drop the DLL into the powershell window
# Admin access shouldn't be required for this script
#
# Changelog:
#  2025-01-27: further snippet type improvements, renamed to UpdateDLSS.ps1 for easier usage
#  2025-01-26: improved snippet type detection, updated comments
#  2025-01-24: updated to get correct version of v310 files

function Update-IniFile {
    param (
        [string]$FilePath,
        [string]$Section,
        [string]$Key,
        [string]$Value
    )

    $content = Get-Content $FilePath

    $sectionExists = $false
    $keyExists = $false

    # Loop through each line in the content
    for ($i = 0; $i -lt $content.Count; $i++) {
        $line = $content[$i]

        # Check if the line contains the section header
        if ($line -match "^\[$Section\]") {
            $sectionExists = $true

            # Search for the key in subsequent lines within the same section
            for ($j = $i + 1; $j -lt $content.Count; $j++) {
                $line = $content[$j]

                # Break out of loop if new section has started
                if ($line.StartsWith("[") -And $line.EndsWith("]")) {
                    break
                }

                # Check if the line contains the key
                if ($line -match "^$Key\s*=") {
                    # Update the value
                    $content[$j] = "$Key = $Value"
                    $keyExists = $true
                    break
                }
            }

            # If the key doesn't exist, add it to the section
            if (-not $keyExists) {
                # Insert the key-value pair at the start of this section
                $content[$i] = $content[$i] + "`r`n$Key = $Value"
            }

            break
        }
    }

    # If the section doesn't exist, add it along with the key
    if (-not $sectionExists) {
        $content += "[$Section]`r`n"
        $content += "$Key = $Value`r`n"
    }

    # Write the updated content back to the INI file
    $content | Set-Content $FilePath
}

function UpdateDLSS {
    param (
        [Parameter(Mandatory=$true)]
        [string]$DllPath
    )

    $nvngxDllPath = $DllPath.Trim(" ").Trim('"').Trim(" ")
    $nvngxDllType = $null
    $fileName = (Split-Path $nvngxDllPath -Leaf)
    if ($fileName -notlike '*_*') {
        Write-Host "Error: Invalid input DLL filename (no underscores?), aborting script."
        return
    }
    $fileName = $fileName.Split('_')[1] -replace '\.dll$'

    # User might have passed us a filename with extraneous chars, eg nvngx_dlss (1).dll
    # Try extracting relevant part from it and check against known snippet names

    # Extract the first 5 characters if the string is long enough, otherwise extract 4
    $extractedValue = if ($fileName.Length -ge 5) { $fileName.Substring(0, 5) } else { $fileName.Substring(0, 4) }

    # Check if the extracted value matches one of the desired values
    if ($extractedValue -in 'dlss', 'dlssg', 'dlssd') {
        $nvngxDllType = $extractedValue
    } elseif ($fileName -in 'dlss', 'dlssg', 'dlssd') {
        # Handle cases where the filename is exactly one of the desired values
        $nvngxDllType = $fileName
    }

    # Check if $nvngxDllType is still null
    if ($null -eq $nvngxDllType) {
        # Last chance, check if it might be dlss
        if ($fileName.Substring(0, 4) -eq "dlss") {
            $nvngxDllType = "dlss"
        } else {
            Write-Host "Error: Unable to determine the DLL type from the filename. Please check the input: $nvngxDllPath"
            Write-Host "Aborting script."
            return
        }
    }

    # DllImport needs full DLL path, try figuring it out:
    $DllPath = [System.IO.Path]::GetFullPath($nvngxDllPath)
    if (-not (Test-Path $DllPath)) {
        $DllPath = Join-Path -Path $PWD -ChildPath $nvngxDllPath
        if (-not (Test-Path $DllPath)) {
            $DllPath = Join-Path -Path $PSScriptRoot -ChildPath $nvngxDllPath
            if (-not (Test-Path $DllPath)) {
                Write-Host "Error: Input DLL path not found: $nvngx_filename"
                Write-Host "Aborting script."
                return
            }
        }
    }

    Write-Host "Input DLL path: $DllPath"

    $signature = Get-AuthenticodeSignature -FilePath $DllPath
    if ($signature.Status -ne "Valid") {
        Write-Host "Error: Input DLL doesn't have a valid authenticode signature."
        Write-Host "UpdateDLSS requires loading/executing the DLL to work, which may be a risk."
        Write-Host "Aborting script."
        return
    }

    # DllImport code
    $dllimport = @"
        [DllImport(@"$DllPath", CallingConvention = CallingConvention.Cdecl)]
        public static extern int NVSDK_NGX_GetSnippetVersion();
"@

    # Generate random string to use for class name above, so script can be ran multiple times in same session
    $length = 10
    $randomString = -join ((65..90) + (97..122) | Get-Random -Count $length | ForEach-Object { [char]$_ })

    $type = Add-Type -MemberDefinition $dllimport -Name "NGX$randomString" -Namespace "NV" -PassThru

    $nvngxVersion = 0
    try {
        $nvngxVersion = $type::NVSDK_NGX_GetSnippetVersion()
    } catch {
        Write-Host "Error: Failed to execute NVSDK_NGX_GetSnippetVersion, error: $_"
        return
    }

    $ver = [BitConverter]::GetBytes($nvngxVersion)
    $verMajor = [BitConverter]::ToUInt16($ver, 2)
    # Seems we don't need the result of NVSDK_NGX_GetGPUArchitecture, 160 always works fine?
    #$nvngxDllArch = $type::NVSDK_NGX_GetGPUArchitecture().ToString('X')
    $nvngxDllArch = "160"
    $nvngxBasePath = "C:\ProgramData\NVIDIA\NGX\models"
    $dllDestPath = "$nvngxBasePath\$nvngxDllType\versions\$nvngxVersion\files\$($nvngxDllArch)_E658703.bin"
    $configFilePath = "$nvngxBasePath\nvngx_config.txt"
    $configKey = "app_E658703"
    $configValue = "$($verMajor).$($ver[1]).$($ver[0])"

    Write-Host " "
    Write-Host "Detected NVNGX DLL info:"
    Write-Host "  Type: $nvngxDllType"
    Write-Host "  Version: $configValue"
    Write-Host "  Signature: $($signature.Status)"
    Write-Host " "
    Write-Host "The following operations will be performed:"
    Write-Host "1. Copy DLL to $dllDestPath"
    Write-Host "2. Update config file $configFilePath with the following:"
    Write-Host "   [$nvngxDllType]"
    Write-Host "   $configKey = $configValue"
    Write-Host " "

    if (Test-Path $dllDestPath) {
        Write-Host "Error: DLL destination already exists, script aborted."
        return
    }

    $confirmation = Read-Host "Do you want to proceed? (y/n)"

    if ($confirmation.ToLower() -ne 'y') {
        Write-Host "Operation canceled by user."
        return
    }

    # Create folders if needed
    if (-not (Test-Path (Split-Path $dllDestPath))) {
        New-Item -ItemType Directory -Force -Path (Split-Path $dllDestPath) | Out-Null
    }

    # Copy DLL
    Copy-Item -Path $DllPath -Destination $dllDestPath -Force

    if (-not (Test-Path $configFilePath)) {
        Write-Host "Config file doesn't exist, creating $configFilePath"
        New-Item -Path $configFilePath -ItemType File > $null
    }

    # Update config file
    Update-IniFile -FilePath $configFilePath -Section $nvngxDllType -Key $configKey -Value $configValue

    Write-Host "Operations completed successfully."
}

try
{
    Export-ModuleMember -Function "UpdateDLSS"
} catch {
    UpdateDLSS
}
	# Global DLSS update script by emoose - https://gist.github.com/emoose/11271bbb3b42fb3b1b0e1c83eef47c05
	# Allows setting up driver to use a single global DLSS DLL for majority of DLSS2/3 games
	# If the global version is newer than the one included with game, it should get loaded automatically
	# (how this works: https://forums.guru3d.com/threads/.439761/page-143#post-6221767)
	#
	# Almost all DLSS3 games should work, DLSS2 is hit-and-miss
	# (DLSS2 games with customized appid probably won't work, thanks to some pointless nvngx checks)
	# DLSSD/DLSSG should work fine with nearly all games (re-run the script for each DLL you want to update)
	#
	# Usage:
	# - press the Raw button on the right side, select all contents with CTRL+A, copy with CTRL+C (don't download as file as PS won't allow execution)
	# - paste contents into notepad, save as UpdateDLSS.ps1
	# - find the saved script, right click -> "Run with PowerShell"
	# - when asked for DLL path, drag+drop an nvngx_dlss file into the window, and press enter
	# - if Run with PowerShell option doesn't appear, you can also run through command prompt or powershell: "powershell C:\Users\emoose\Downloads\UpdateDLSS.ps1"
	#
	# To undo the changes from this script just clear out the "C:\ProgramData\NVIDIA\NGX\models" folder
	# This folder should regen the next time DLSS is used, and games will go back to using the DLL from their game folder
	#
	# Advanced setup (installing as PowerShell module):
	# - save script as UpdateDLSS.psm1 into powershell modules folder, so you can use it from any location
	# Depending on PS version, save to either:
	# - Documents\WindowsPowerShell\Modules\UpdateDLSS\UpdateDLSS.psm1
	# - Documents\PowerShell\Modules\UpdateDLSS\UpdateDLSS.psm1
	# Alternatively System32 should work for all PS versions:
	# - C:\Windows\system32\WindowsPowerShell\v1.0\Modules\UpdateDLSS\UpdateDLSS.psm1
	# With it saved into modules folder, you should be able to open a new powershell window and then run "UpdateDLSS" from any location
	# (or run "UpdateDLSS -DllPath C:\DLLs\nvngx_dlss.dll")
	#
	# If ran without -DllPath param you'll be prompted for the path to the DLSS DLL
	# When prompted you can usually just drag+drop the DLL into the powershell window
	# Admin access shouldn't be required for this script
	#
	# Changelog:
	# 2025-01-27: further snippet type improvements, renamed to UpdateDLSS.ps1 for easier usage
	# 2025-01-26: improved snippet type detection, updated comments
	# 2025-01-24: updated to get correct version of v310 files

	function Update-IniFile {
	param (
	[string]$FilePath,
	[string]$Section,
	[string]$Key,
	[string]$Value
	)

	$content = Get-Content $FilePath

	$sectionExists = $false
	$keyExists = $false

	# Loop through each line in the content
	for ($i = 0; $i -lt $content.Count; $i++) {
	$line = $content[$i]

	# Check if the line contains the section header
	if ($line -match "^\[$Section\]") {
	$sectionExists = $true

	# Search for the key in subsequent lines within the same section
	for ($j = $i + 1; $j -lt $content.Count; $j++) {
	$line = $content[$j]

	# Break out of loop if new section has started
	if ($line.StartsWith("[") -And $line.EndsWith("]")) {
	break
	}

	# Check if the line contains the key
	if ($line -match "^$Key\s*=") {
	# Update the value
	$content[$j] = "$Key = $Value"
	$keyExists = $true
	break
	}
	}

	# If the key doesn't exist, add it to the section
	if (-not $keyExists) {
	# Insert the key-value pair at the start of this section
	$content[$i] = $content[$i] + "`r`n$Key = $Value"
	}

	break
	}
	}

	# If the section doesn't exist, add it along with the key
	if (-not $sectionExists) {
	$content += "[$Section]`r`n"
	$content += "$Key = $Value`r`n"
	}

	# Write the updated content back to the INI file
	$content \| Set-Content $FilePath
	}

	function UpdateDLSS {
	param (
	[Parameter(Mandatory=$true)]
	[string]$DllPath
	)

	$nvngxDllPath = $DllPath.Trim(" ").Trim('"').Trim(" ")
	$nvngxDllType = $null
	$fileName = (Split-Path $nvngxDllPath -Leaf)
	if ($fileName -notlike '_') {
	Write-Host "Error: Invalid input DLL filename (no underscores?), aborting script."
	return
	}
	$fileName = $fileName.Split('_')[1] -replace '\.dll$'

	# User might have passed us a filename with extraneous chars, eg nvngx_dlss (1).dll
	# Try extracting relevant part from it and check against known snippet names

	# Extract the first 5 characters if the string is long enough, otherwise extract 4
	$extractedValue = if ($fileName.Length -ge 5) { $fileName.Substring(0, 5) } else { $fileName.Substring(0, 4) }

	# Check if the extracted value matches one of the desired values
	if ($extractedValue -in 'dlss', 'dlssg', 'dlssd') {
	$nvngxDllType = $extractedValue
	} elseif ($fileName -in 'dlss', 'dlssg', 'dlssd') {
	# Handle cases where the filename is exactly one of the desired values
	$nvngxDllType = $fileName
	}

	# Check if $nvngxDllType is still null
	if ($null -eq $nvngxDllType) {
	# Last chance, check if it might be dlss
	if ($fileName.Substring(0, 4) -eq "dlss") {
	$nvngxDllType = "dlss"
	} else {
	Write-Host "Error: Unable to determine the DLL type from the filename. Please check the input: $nvngxDllPath"
	Write-Host "Aborting script."
	return
	}
	}

	# DllImport needs full DLL path, try figuring it out:
	$DllPath = [System.IO.Path]::GetFullPath($nvngxDllPath)
	if (-not (Test-Path $DllPath)) {
	$DllPath = Join-Path -Path $PWD -ChildPath $nvngxDllPath
	if (-not (Test-Path $DllPath)) {
	$DllPath = Join-Path -Path $PSScriptRoot -ChildPath $nvngxDllPath
	if (-not (Test-Path $DllPath)) {
	Write-Host "Error: Input DLL path not found: $nvngx_filename"
	Write-Host "Aborting script."
	return
	}
	}
	}

	Write-Host "Input DLL path: $DllPath"

	$signature = Get-AuthenticodeSignature -FilePath $DllPath
	if ($signature.Status -ne "Valid") {
	Write-Host "Error: Input DLL doesn't have a valid authenticode signature."
	Write-Host "UpdateDLSS requires loading/executing the DLL to work, which may be a risk."
	Write-Host "Aborting script."
	return
	}

	# DllImport code
	$dllimport = @"
	[DllImport(@"$DllPath", CallingConvention = CallingConvention.Cdecl)]
	public static extern int NVSDK_NGX_GetSnippetVersion();
	"@

	# Generate random string to use for class name above, so script can be ran multiple times in same session
	$length = 10
	$randomString = -join ((65..90) + (97..122) \| Get-Random -Count $length \| ForEach-Object { [char]$_ })

	$type = Add-Type -MemberDefinition $dllimport -Name "NGX$randomString" -Namespace "NV" -PassThru

	$nvngxVersion = 0
	try {
	$nvngxVersion = $type::NVSDK_NGX_GetSnippetVersion()
	} catch {
	Write-Host "Error: Failed to execute NVSDK_NGX_GetSnippetVersion, error: $_"
	return
	}

	$ver = [BitConverter]::GetBytes($nvngxVersion)
	$verMajor = [BitConverter]::ToUInt16($ver, 2)
	# Seems we don't need the result of NVSDK_NGX_GetGPUArchitecture, 160 always works fine?
	#$nvngxDllArch = $type::NVSDK_NGX_GetGPUArchitecture().ToString('X')
	$nvngxDllArch = "160"
	$nvngxBasePath = "C:\ProgramData\NVIDIA\NGX\models"
	$dllDestPath = "$nvngxBasePath\$nvngxDllType\versions\$nvngxVersion\files\$($nvngxDllArch)_E658703.bin"
	$configFilePath = "$nvngxBasePath\nvngx_config.txt"
	$configKey = "app_E658703"
	$configValue = "$($verMajor).$($ver[1]).$($ver[0])"

	Write-Host " "
	Write-Host "Detected NVNGX DLL info:"
	Write-Host " Type: $nvngxDllType"
	Write-Host " Version: $configValue"
	Write-Host " Signature: $($signature.Status)"
	Write-Host " "
	Write-Host "The following operations will be performed:"
	Write-Host "1. Copy DLL to $dllDestPath"
	Write-Host "2. Update config file $configFilePath with the following:"
	Write-Host " [$nvngxDllType]"
	Write-Host " $configKey = $configValue"
	Write-Host " "

	if (Test-Path $dllDestPath) {
	Write-Host "Error: DLL destination already exists, script aborted."
	return
	}

	$confirmation = Read-Host "Do you want to proceed? (y/n)"

	if ($confirmation.ToLower() -ne 'y') {
	Write-Host "Operation canceled by user."
	return
	}

	# Create folders if needed
	if (-not (Test-Path (Split-Path $dllDestPath))) {
	New-Item -ItemType Directory -Force -Path (Split-Path $dllDestPath) \| Out-Null
	}

	# Copy DLL
	Copy-Item -Path $DllPath -Destination $dllDestPath -Force

	if (-not (Test-Path $configFilePath)) {
	Write-Host "Config file doesn't exist, creating $configFilePath"
	New-Item -Path $configFilePath -ItemType File > $null
	}

	# Update config file
	Update-IniFile -FilePath $configFilePath -Section $nvngxDllType -Key $configKey -Value $configValue

	Write-Host "Operations completed successfully."
	}

	try
	{
	Export-ModuleMember -Function "UpdateDLSS"
	} catch {
	UpdateDLSS
	}