Merge pull request #44 from microhack44/patch-12

SAC Script debugged and updated less prone to failures

src/windows/sac-enabler.ps1

@@ -5,36 +5,60 @@ cmd /c color 0A
 $host.UI.RawUI.WindowTitle = "                                                                                  --== AUTO SAC ENABLER by Daniel Muñoz L ==--"
 
 # Rescue OS variable
-$diska='c'
+$diska='C'
 
 # FINDER FOR FAULTY OS DISK
-$diskarray = "q","w","e","r","t","y","u","i","o","p","s","d","f","g","h","j","k","l","z","x","v","n","m"
+$diskarray = "Q","W","E","R","T","Y","U","I","O","P","S","D","F","G","H","J","K","L","Z","X","V","N","M"
 $diskb="000"
 foreach ($diskt in $diskarray)
 {
    if (Test-Path -Path "$($diskt):\Windows") {$diskb=$diskt} 
 }
 
-# ADDING BCD DRIVE
-$diskc="000"
-foreach ($diskt in $diskarray)
-{
-   if ((Test-Path -Path "$($diskt):\efi") -or (Test-Path -Path "$($diskt):\boot")){$diskc=$diskt} 
-}
-
-# ADDING BCD PATH
-$diskd="000"
-foreach ($diskt in $diskarray)
-{
-   if (Test-Path -Path "$($diskt):\efi") 
-      {write-output "VM is GEN2";$diskd="$($diskc):\efi\Microsoft\boot\bcd"}
-   elseif (Test-Path -Path "$($diskt):\boot")
-      {write-output "VM is GEN1";$diskd="$($diskc):\boot\bcd"}
-}
-
-# IN CASE OF FINDER FAILURE WITH MITIGATION REASURE OS DISK IS MOUNTED AS DATA BEFORE CHECKING BOOTMGR
+# IN CASE OF FINDER FAILURE WITH MITIGATION REASURE IF OS DISK EXIST AND IS MOUNTED AS DATADISK BEFORE PROCEDING
 if ($diskb -eq "000") {write-output "SCRIPT COULD NOT FIND A RESCUE OS DISK ATTACHED, EXITING";start-sleep 10;Exit}
-if ($diskc -eq "000") {write-output "SCRIPT COULD NOT FIND A BOOT FOLDER, EXITING";start-sleep 10;Exit}
+
+# DETECT IF GEN2
+$partboot='777'
+$diskd='000'
+$disknumber=$_
+
+Get-WmiObject Win32_DiskDrive | ForEach-Object {
+  $disk = $_
+  $partitions = "ASSOCIATORS OF " +
+                "{Win32_DiskDrive.DeviceID='$($disk.DeviceID)'} " +
+                "WHERE AssocClass = Win32_DiskDriveToDiskPartition"
+  Get-WmiObject -Query $partitions | ForEach-Object {
+    $partition = $_
+    $drives = "ASSOCIATORS OF " +
+              "{Win32_DiskPartition.DeviceID='$($partition.DeviceID)'} " +
+              "WHERE AssocClass = Win32_LogicalDiskToPartition"
+    Get-WmiObject -Query $drives | ForEach-Object {
+      New-Object -Type PSCustomObject -Property @{
+        X = $_.DeviceID
+        Y  = $partition.diskindex
+      }
+    }
+  }
+} > "$($diskb):\txtemp"
+
+Select-String -Pattern "$($diskb)" -Path "$($diskb):\txtemp" -List -CaseSensitive | select-object -First 1 | %{$disknumber=$_.Line.Split('')[0]}
+
+get-partition -disknumber $disknumber > "$($diskb):\txtempvar"
+Select-String -Pattern "System" -Path "$($diskb):\txtempvar" -list -SimpleMatch | select-object -First 1 | %{$partboot=$_.Line.Split('')[0]}
+Get-Partition -DiskNumber $disknumber -PartitionNumber $partboot | Set-Partition -NewDriveLetter z
+
+	if (Test-Path -Path "Z:\efi\Microsoft") {write-output "VM is GEN2";$diskd="Z:\efi\Microsoft\boot\bcd"}
+
+# DETECT IF GEN1
+if ($diskd -eq '000')
+{
+	foreach ($diskt in $diskarray)
+	{
+	   	if (Test-Path -Path "$($diskt):\boot\bcd")
+      	{write-output "VM is GEN1";$diskd="$($diskt):\boot\bcd"}
+	}
+}
 
 # SAC ENABLE
 bcdedit /store "$($diskd)" /set "{bootmgr}" displaybootmenu yes
@@ -43,6 +67,7 @@ bcdedit /store "$($diskd)" /set "{bootmgr}" bootems yes
 bcdedit /store "$($diskd)" /ems "{default}" ON
 bcdedit /store "$($diskd)" /emssettings EMSPORT:1 EMSBAUDRATE:115200
 
+Remove-Item -force "$($diskb):\txtempvar"
+
 write-output "          ---------------          SCRIPT FINISHED PROPERLY, CHANGES APPLIED          ---------------          "
 start-sleep 10
-