@loucompanion/forge-bridge 0.1.3-dev.ba7d36726eda → 0.1.3-dev.bca7d0ccee53

package/runtime/windows-job-supervisor.ps1

@@ -4,7 +4,7 @@ param(
   [string]$Mode,
 
   [Parameter(Mandatory = $true)]
-  [string]$Payload,
+  [string]$PayloadPath,
 
   [string]$GateName
 )
@@ -159,8 +159,10 @@ public static class ForgeWindowsJob
 "@
 
 function Read-ForgeSupervisorPayload {
-  $bytes = [Convert]::FromBase64String($Payload)
-  $json = [Text.Encoding]::UTF8.GetString($bytes)
+  if (-not [IO.Path]::IsPathRooted($PayloadPath)) {
+    throw "The supervisor payload path must be absolute."
+  }
+  $json = [IO.File]::ReadAllText($PayloadPath, [Text.Encoding]::UTF8)
   return $json | ConvertFrom-Json
 }
 
@@ -170,22 +172,85 @@ function New-ForgeChildProcess(
   [string]$workingDirectory,
   [bool]$redirectStreams
 ) {
-  $startInfo = New-Object System.Diagnostics.ProcessStartInfo
-  $startInfo.FileName = $command
-  $startInfo.Arguments = [ForgeWindowsJob]::BuildArguments($arguments)
-  $startInfo.WorkingDirectory = $workingDirectory
-  $startInfo.UseShellExecute = $false
-  $startInfo.CreateNoWindow = $true
-  $startInfo.RedirectStandardInput = $redirectStreams
-  $startInfo.RedirectStandardOutput = $redirectStreams
-  $startInfo.RedirectStandardError = $redirectStreams
-
-  $child = New-Object System.Diagnostics.Process
-  $child.StartInfo = $startInfo
-  if (-not $child.Start()) {
-    throw "Failed to start supervised process '$command'."
+  $maximumAttempts = 3
+  for ($attempt = 1; $attempt -le $maximumAttempts; $attempt += 1) {
+    $startInfo = New-Object System.Diagnostics.ProcessStartInfo
+    $startInfo.FileName = $command
+    $startInfo.Arguments = [ForgeWindowsJob]::BuildArguments($arguments)
+    $startInfo.WorkingDirectory = $workingDirectory
+    $startInfo.UseShellExecute = $false
+    $startInfo.CreateNoWindow = $true
+    $startInfo.RedirectStandardInput = $redirectStreams
+    $startInfo.RedirectStandardOutput = $redirectStreams
+    $startInfo.RedirectStandardError = $redirectStreams
+
+    $child = New-Object System.Diagnostics.Process
+    $child.StartInfo = $startInfo
+    try {
+      if (-not $child.Start()) {
+        throw "Failed to start supervised process '$command'."
+      }
+      $child | Add-Member -NotePropertyName ForgeStartAttempt -NotePropertyValue $attempt
+      return $child
+    }
+    catch {
+      $nativeError = $_.Exception
+      while ($nativeError.InnerException) {
+        $nativeError = $nativeError.InnerException
+      }
+      $retryable = $nativeError -is [ComponentModel.Win32Exception] -and $nativeError.NativeErrorCode -in @(5, 32, 33)
+      $child.Dispose()
+      if (-not $retryable -or $attempt -eq $maximumAttempts) {
+        throw
+      }
+      Start-Sleep -Milliseconds (100 * $attempt)
+    }
+  }
+}
+
+function Write-ForgeProviderReady([object]$request, [Diagnostics.Process]$provider) {
+  $readyPath = [string]$request.readyPath
+  if ([string]::IsNullOrWhiteSpace($readyPath) -or -not [IO.Path]::IsPathRooted($readyPath)) {
+    throw "The provider-ready path must be absolute."
+  }
+  $payloadDirectory = [IO.Path]::GetDirectoryName([IO.Path]::GetFullPath($PayloadPath))
+  $readyDirectory = [IO.Path]::GetDirectoryName([IO.Path]::GetFullPath($readyPath))
+  if (-not [string]::Equals($payloadDirectory, $readyDirectory, [StringComparison]::OrdinalIgnoreCase)) {
+    throw "The provider-ready path must be beside the supervisor payload."
+  }
+
+  $ready = @{
+    schemaVersion = 1
+    supervisorId = [string]$request.supervisorId
+    rootPid = [int]$env:FORGE_SUPERVISOR_ROOT_PID
+    rootIdentity = [string]$env:FORGE_SUPERVISOR_ROOT_IDENTITY
+    providerPid = $provider.Id
+    providerCreatedAt = $provider.StartTime.ToUniversalTime().ToString("O")
+    providerIdentity = "win32:$($provider.StartTime.ToUniversalTime().Ticks)"
+    startAttempt = [int]$provider.ForgeStartAttempt
+  } | ConvertTo-Json -Compress
+  $temporaryPath = "$readyPath.$([Guid]::NewGuid().ToString('N')).tmp"
+  [IO.File]::WriteAllText($temporaryPath, "$ready`n", (New-Object Text.UTF8Encoding($false)))
+  [IO.File]::Move($temporaryPath, $readyPath)
+}
+
+function Wait-ForgeProviderStartGrant([object]$request) {
+  $grantPath = [string]$request.startGrantPath
+  if ([string]::IsNullOrWhiteSpace($grantPath) -or -not [IO.Path]::IsPathRooted($grantPath)) {
+    throw "The provider start-grant path must be absolute."
+  }
+  $payloadDirectory = [IO.Path]::GetDirectoryName([IO.Path]::GetFullPath($PayloadPath))
+  $grantDirectory = [IO.Path]::GetDirectoryName([IO.Path]::GetFullPath($grantPath))
+  if (-not [string]::Equals($payloadDirectory, $grantDirectory, [StringComparison]::OrdinalIgnoreCase)) {
+    throw "The provider start-grant path must be beside the supervisor payload."
+  }
+  for ($attempt = 0; $attempt -lt 1200; $attempt += 1) {
+    if ([IO.File]::Exists($grantPath)) {
+      return
+    }
+    Start-Sleep -Milliseconds 25
   }
-  return $child
+  throw "Timed out waiting for the provider start grant."
 }
 
 function Start-ForgeOutputCopies([Diagnostics.Process]$child) {
@@ -220,7 +285,9 @@ if ($Mode -eq "Inner") {
   }
 
   $provider = New-ForgeChildProcess ([string]$request.command) @($request.args) ([string]$request.cwd) $true
+  Write-ForgeProviderReady $request $provider
   $copies = Start-ForgeOutputCopies $provider
+  Wait-ForgeProviderStartGrant $request
   Write-ForgeStandardInput $provider $inputText
   $provider.WaitForExit()
   if (-not [Threading.Tasks.Task]::WaitAll([Threading.Tasks.Task[]]$copies, 1000)) {
@@ -232,6 +299,8 @@ if ($Mode -eq "Inner") {
 
 $inputText = [Console]::In.ReadToEnd()
 $request = Read-ForgeSupervisorPayload
+$env:FORGE_SUPERVISOR_ROOT_PID = [string]$PID
+$env:FORGE_SUPERVISOR_ROOT_IDENTITY = "win32:$([Diagnostics.Process]::GetCurrentProcess().StartTime.ToUniversalTime().Ticks)"
 $parent = [Diagnostics.Process]::GetProcessById([int]$request.parentPid)
 $jobName = "Local\Forge-$([string]$request.supervisorId)"
 $job = [ForgeWindowsJob]::CreateKillOnCloseJob($jobName)
@@ -252,8 +321,8 @@ try {
     $PSCommandPath,
     "-Mode",
     "Inner",
-    "-Payload",
-    $Payload,
+    "-PayloadPath",
+    $PayloadPath,
     "-GateName",
     $gateNameValue
   )