From 509b59b7aea7611d8f2203ebd7b6357ba772afb7 Mon Sep 17 00:00:00 2001
From: Mikhail Paulyshka <me@mixaill.tk>
Date: Tue, 25 Jul 2017 12:42:59 +0300
Subject: [PATCH] scipts/getWindowsSDK: use registry for SDK detection

---
 scripts/getWindowsSDK.ps1 | 34 +++++++++++++++++++---------------
 1 file changed, 19 insertions(+), 15 deletions(-)

diff --git a/scripts/getWindowsSDK.ps1 b/scripts/getWindowsSDK.ps1
index 7902707cb5..3aa23cc560 100644
--- a/scripts/getWindowsSDK.ps1
+++ b/scripts/getWindowsSDK.ps1
@@ -17,19 +17,14 @@ $scriptsDir = split-path -parent $MyInvocation.MyCommand.Definition
 
 $validInstances = New-Object System.Collections.ArrayList
 
-$CandidateProgramFiles = $(& $scriptsDir\getProgramFiles32bit.ps1), $(& $scriptsDir\getProgramFilesPlatformBitness.ps1)
-Write-Verbose "Program Files Candidate locations: $([system.String]::Join(',', $CandidateProgramFiles))"
-
 # Windows 10 SDK
-Write-Verbose "`n"
-Write-Verbose "Looking for Windows 10 SDK"
-foreach ($ProgramFiles in $CandidateProgramFiles)
+function CheckWindows10SDK($path)
 {
-    $folder = "$ProgramFiles\Windows Kits\10\Include"
+    $folder = "$path\Include"
     if (!(Test-Path $folder))
     {
         Write-Verbose "$folder - Not Found"
-        continue
+        return
     }
 
     Write-Verbose "$folder - Found"
@@ -42,7 +37,7 @@ foreach ($ProgramFiles in $CandidateProgramFiles)
         if (!(Test-Path $windowsheader))
         {
             Write-Verbose "$windowsheader - Not Found"
-            continue
+            return
         }
         Write-Verbose "$windowsheader - Found"
 
@@ -50,7 +45,7 @@ foreach ($ProgramFiles in $CandidateProgramFiles)
         if (!(Test-Path $ddkheader))
         {
             Write-Verbose "$ddkheader - Not Found"
-            continue
+            return
         }
 
         Write-Verbose "$ddkheader - Found"
@@ -60,16 +55,20 @@ foreach ($ProgramFiles in $CandidateProgramFiles)
     }
 }
 
-# Windows 8.1 SDK
 Write-Verbose "`n"
-Write-Verbose "Looking for Windows 8.1 SDK"
-foreach ($ProgramFiles in $CandidateProgramFiles)
+Write-Verbose "Looking for Windows 10 SDK"
+CheckWindows10SDK((Get-ItemProperty -Path 'HKLM:\SOFTWARE\Microsoft\Windows Kits\Installed Roots\' -Name 'KitsRoot10' -ErrorAction SilentlyContinue).KitsRoot10)
+CheckWindows10SDK((Get-ItemProperty -Path 'HKLM:\SOFTWARE\WOW6432Node\Microsoft\Windows Kits\Installed Roots\' -Name 'KitsRoot10' -ErrorAction SilentlyContinue).KitsRoot10)
+
+
+# Windows 8.1 SDK
+function CheckWindows81SDK($path)
 {
-    $folder = "$ProgramFiles\Windows Kits\8.1\Include"
+    $folder = "$path\Include"
     if (!(Test-Path $folder))
     {
         Write-Verbose "$folder - Not Found"
-        continue
+        return
     }
 
     Write-Verbose "$folder - Found"
@@ -78,6 +77,11 @@ foreach ($ProgramFiles in $CandidateProgramFiles)
     $validInstances.Add($win81sdkVersionString) > $null
 }
 
+Write-Verbose "`n"
+Write-Verbose "Looking for Windows 8.1 SDK"
+CheckWindows81SDK((Get-ItemProperty -Path 'HKLM:\SOFTWARE\Microsoft\Windows Kits\Installed Roots\' -Name 'KitsRoot81' -ErrorAction SilentlyContinue).KitsRoot81)
+CheckWindows81SDK((Get-ItemProperty -Path 'HKLM:\SOFTWARE\WOW6432Node\Microsoft\Windows Kits\Installed Roots\' -Name 'KitsRoot81' -ErrorAction SilentlyContinue).KitsRoot81)
+
 Write-Verbose "`n`n`n"
 Write-Verbose "The following Windows SDKs were found:"
 foreach ($instance in $validInstances)