Files
rdp-proxy/scripts/fabric/c19z3-remote-workspace-mailbox-stale-preflight-smoke.ps1
T
2026-05-14 23:30:34 +03:00

245 lines
13 KiB
PowerShell

param(
[string]$ApiBaseUrl = "http://192.168.200.61:18121/api/v1",
[string]$ClusterID = "cfc0743d-d960-49fb-9de8-96e063d5e4aa",
[string]$ActorUserID = "f67d943f-5397-4b3a-a229-695fe67ad700",
[string]$EntryNodeName = "test-1",
[string]$ExitNodeName = "test-2",
[string]$EntryBaseUrl = "http://192.168.200.61:19131",
[string]$ResultPath = "artifacts\c19z3-remote-workspace-mailbox-stale-preflight-smoke-result.json",
[switch]$SkipClose
)
Set-StrictMode -Version Latest
$ErrorActionPreference = "Stop"
$repoRoot = (Resolve-Path (Join-Path $PSScriptRoot "..\..")).ProviderPath
$runId = "c19z3-" + (Get-Date -Format "yyyyMMdd-HHmmss")
$consumerID = "rdp-worker-probe"
$routeID = ""
$adapterSessionID = ""
function Invoke-Api {
param([string]$Method, [string]$Path, [object]$Body = $null)
$uri = "$ApiBaseUrl$Path"
if ($null -eq $Body) { return Invoke-RestMethod -Method $Method -Uri $uri -TimeoutSec 30 }
return Invoke-RestMethod -Method $Method -Uri $uri -ContentType "application/json" -Body ($Body | ConvertTo-Json -Depth 80) -TimeoutSec 30
}
function Get-PropertyValue {
param([object]$Item, [string]$Name, [object]$Default = $null)
if ($null -eq $Item) { return $Default }
$property = $Item.PSObject.Properties[$Name]
if ($null -eq $property) { return $Default }
return $property.Value
}
function ConvertTo-Base64UrlJson {
param([object]$Value)
if ($Value -is [string]) { $json = $Value } else { $json = $Value | ConvertTo-Json -Depth 80 -Compress }
$bytes = [System.Text.Encoding]::UTF8.GetBytes($json)
return [Convert]::ToBase64String($bytes).TrimEnd("=").Replace("+", "-").Replace("/", "_")
}
function Get-Events {
param([object]$Item)
$events = Get-PropertyValue -Item $Item -Name "events" -Default $null
if ($null -eq $events) { return @() }
return @($events)
}
function Get-NodeByName {
param([string]$Name)
$nodes = (Invoke-Api -Method GET -Path "/clusters/$ClusterID/nodes?actor_user_id=$ActorUserID").nodes
$node = @($nodes | Where-Object { $_.name -eq $Name }) | Select-Object -First 1
if ($null -eq $node) { throw "Node '$Name' was not found in cluster $ClusterID" }
return $node
}
function Disable-ExistingRemoteWorkspaceRoutes {
param([string]$SourceNodeID, [string]$DestinationNodeID)
$items = (Invoke-Api -Method GET -Path "/clusters/$ClusterID/mesh/route-intents?actor_user_id=$ActorUserID").route_intents
foreach ($item in @($items)) {
if ([string](Get-PropertyValue -Item $item -Name "status" -Default "") -ne "active") { continue }
if ([string](Get-PropertyValue -Item $item -Name "service_class" -Default "") -ne "remote_workspace") { continue }
$sourceSelector = Get-PropertyValue -Item $item -Name "source_selector" -Default $null
$destinationSelector = Get-PropertyValue -Item $item -Name "destination_selector" -Default $null
if ([string](Get-PropertyValue -Item $sourceSelector -Name "node_id" -Default "") -ne $SourceNodeID) { continue }
if ([string](Get-PropertyValue -Item $destinationSelector -Name "node_id" -Default "") -ne $DestinationNodeID) { continue }
[void](Invoke-Api -Method POST -Path "/clusters/$ClusterID/mesh/route-intents/$($item.id)/disable" -Body @{
actor_user_id = $ActorUserID
reason = "c19z3 isolate stale mailbox preflight smoke"
})
}
}
function New-RemoteWorkspaceRouteIntent {
param([string]$SourceNodeID, [string]$DestinationNodeID)
$expiresAt = (Get-Date).ToUniversalTime().AddMinutes(5).ToString("o")
return Invoke-Api -Method POST -Path "/clusters/$ClusterID/mesh/route-intents" -Body @{
actor_user_id = $ActorUserID
source_selector = @{ node_id = $SourceNodeID }
destination_selector = @{ node_id = $DestinationNodeID }
service_class = "remote_workspace"
priority = 2100000000
policy = @{
synthetic_enabled = $true
route_version = "$runId-remote-workspace"
policy_version = "$runId-remote-workspace"
peer_directory_version = "$runId-remote-workspace"
hops = @($SourceNodeID, $DestinationNodeID)
allowed_channels = @("control", "interactive", "reliable", "bulk", "droppable")
max_ttl = 8
max_hops = 8
expires_at = $expiresAt
metadata = @{ smoke = "c19z3_remote_workspace_mailbox_stale_preflight"; run_id = $runId }
}
}
}
function New-FrameBatch {
param([int]$Index)
return [ordered]@{
schema_version = "rap.remote_workspace_frame_batch.v1"
probe_only = $true
service_class = "remote_workspace"
channel_class = "interactive"
adapter_contract_id = "rap.rdp_worker.remote_workspace_adapter_contract_probe.v1"
frames = @(@{ channel = "display"; direction = "adapter_to_client"; payload_encoding = "none"; payload_length = $Index; droppable = $true })
}
}
function Invoke-FrameBatch {
param([object]$FrameBatch, [hashtable]$Headers, [string]$Url)
$response = Invoke-WebRequest -Method POST -Uri $Url -Headers $Headers -Body ($FrameBatch | ConvertTo-Json -Depth 80 -Compress) -ContentType "application/vnd.rap.remote-workspace-frame-batch.v1+json" -TimeoutSec 30
return [ordered]@{ status_code = [int]$response.StatusCode; body = ($response.Content | ConvertFrom-Json) }
}
function Invoke-Mailbox {
param([string]$SessionID, [string]$Query = "")
$url = "$EntryBaseUrl/mesh/v1/remote-workspace/adapter-sessions/$SessionID/mailbox$Query"
$response = Invoke-WebRequest -Method GET -Uri $url -TimeoutSec 35
$json = $null
if ($response.Content) { $json = $response.Content | ConvertFrom-Json }
return [ordered]@{ status_code = [int]$response.StatusCode; body = $response.Content; json = $json }
}
function Invoke-Preflight {
param([string]$SessionID, [string]$Query = "")
$url = "$EntryBaseUrl/mesh/v1/remote-workspace/adapter-sessions/$SessionID/mailbox/preflight$Query"
$response = Invoke-WebRequest -Method GET -Uri $url -TimeoutSec 35
$json = $null
if ($response.Content) { $json = $response.Content | ConvertFrom-Json }
return [ordered]@{ status_code = [int]$response.StatusCode; body = $response.Content; json = $json }
}
function Invoke-Control {
param([string]$SessionID)
if ([string]::IsNullOrWhiteSpace($SessionID)) { return $null }
$url = "$EntryBaseUrl/mesh/v1/remote-workspace/adapter-sessions/$SessionID/control"
$body = @{ action = "close"; reason = "c19z3 stale mailbox preflight close" } | ConvertTo-Json -Compress
return Invoke-RestMethod -Method POST -Uri $url -ContentType "application/json" -Body $body -TimeoutSec 30
}
try {
$entryNode = Get-NodeByName -Name $EntryNodeName
$exitNode = Get-NodeByName -Name $ExitNodeName
Disable-ExistingRemoteWorkspaceRoutes -SourceNodeID $entryNode.id -DestinationNodeID $exitNode.id
$route = (New-RemoteWorkspaceRouteIntent -SourceNodeID $entryNode.id -DestinationNodeID $exitNode.id).route_intent
$routeID = [string]$route.id
$leaseResponse = Invoke-Api -Method POST -Path "/clusters/$ClusterID/fabric/service-channels/leases" -Body @{
actor_user_id = $ActorUserID
organization_id = "org-home"
user_id = "user-x"
resource_id = "$runId-remote-workspace"
service_class = "remote_workspace"
entry_node_ids = @([string]$entryNode.id)
exit_node_ids = @([string]$exitNode.id)
preferred_entry_node_id = [string]$entryNode.id
preferred_exit_node_id = [string]$exitNode.id
ttl_seconds = 120
metadata = @{ smoke = "c19z3_remote_workspace_mailbox_stale_preflight"; run_id = $runId }
}
$lease = $leaseResponse.fabric_service_channel_lease
$authorityPayload = Get-PropertyValue -Item $lease -Name "authority_payload" -Default $null
if ($authorityPayload -is [string] -and $authorityPayload.Length -gt 0) { $decodedAuthority = $authorityPayload | ConvertFrom-Json } else { $decodedAuthority = $authorityPayload }
$ingressUrl = "$EntryBaseUrl/api/v1/clusters/$ClusterID/fabric/service-channels/$($lease.channel_id)/remote-workspaces/$($lease.resource_id)/streams/interactive"
$headers = @{
"X-RAP-Service-Channel-Token" = [string]$lease.token.token
"X-RAP-Fabric-Channel-ID" = [string]$lease.channel_id
"X-RAP-Service-Channel-Authority-Payload" = ConvertTo-Base64UrlJson -Value $decodedAuthority
"X-RAP-Service-Channel-Authority-Signature" = ConvertTo-Base64UrlJson -Value (Get-PropertyValue -Item $lease -Name "authority_signature" -Default $null)
"X-RAP-Service-Class" = "remote_workspace"
"X-RAP-Channel-Class" = "interactive"
}
$firstDelivery = Invoke-FrameBatch -FrameBatch (New-FrameBatch -Index 1) -Headers $headers -Url $ingressUrl
$adapterSessionID = [string](Get-PropertyValue -Item $firstDelivery.body -Name "adapter_session_id" -Default "")
$seed = Invoke-Mailbox -SessionID $adapterSessionID -Query "?consumer_id=$consumerID&ack_sequence=1&limit=1"
$overflowDeliveries = @()
for ($i = 2; $i -le 19; $i++) {
$overflowDeliveries += ,(Invoke-FrameBatch -FrameBatch (New-FrameBatch -Index $i) -Headers $headers -Url $ingressUrl)
}
$preflight = Invoke-Preflight -SessionID $adapterSessionID -Query "?consumer_id=$consumerID&resume_from=ack&limit=3"
$control = $null
if (-not $SkipClose) {
$control = Invoke-Control -SessionID $adapterSessionID
}
$checks = [ordered]@{
lease_ready = ([string]$lease.status -eq "ready")
initial_delivery_accepted = ([int]$firstDelivery.status_code -eq 202 -and $adapterSessionID -match "^rap-rw-adapter-session-[0-9a-f]{24}$")
seed_ack_visible = ([int]$seed.status_code -eq 200 -and [int64]$seed.json.consumer_ack_sequence -eq 1)
overflow_deliveries_accepted = (@($overflowDeliveries | Where-Object { [int]$_.status_code -eq 202 }).Count -eq 18)
stale_preflight_visible = ([int]$preflight.status_code -eq 200 -and [bool]$preflight.json.read_only -and [bool]$preflight.json.stale_cursor -and [string]$preflight.json.diagnostic_state -eq "stale_cursor_gap" -and [string]$preflight.json.recommended_action -eq "reset_consumer_and_resync" -and @($preflight.json.action_hints).Contains("reset_consumer_cursor") -and @($preflight.json.action_hints).Contains("request_full_adapter_resync") -and @($preflight.json.action_hints).Contains("resume_from_checkpoint_after_resync") -and [int64]$preflight.json.resume_sequence -eq 1 -and [int]$preflight.json.mailbox_depth -eq 16 -and [int64]$preflight.json.mailbox_dropped_total -ge 3 -and [int64]$preflight.json.first_retained_sequence -gt [int64]$preflight.json.resume_sequence -and [int]$preflight.json.missing_dropped_count -eq ([int64]$preflight.json.first_retained_sequence - [int64]$preflight.json.resume_sequence - 1) -and [int64]$preflight.json.last_retained_sequence -ge [int64]$preflight.json.first_retained_sequence -and [int]$preflight.json.expected_returned_count -eq 3)
close_accepted = ($SkipClose -or ([bool]$control.accepted -and [string]$control.session_state -eq "closed"))
}
$failed = @($checks.GetEnumerator() | Where-Object { -not $_.Value } | ForEach-Object { $_.Key })
$result = [ordered]@{
schema_version = "c19z3.remote_workspace_mailbox_stale_preflight_smoke.v1"
run_id = $runId
cluster_id = $ClusterID
entry_node = @{ id = $entryNode.id; name = $EntryNodeName }
exit_node = @{ id = $exitNode.id; name = $ExitNodeName }
channel_id = $lease.channel_id
route_id = $routeID
adapter_session_id = $adapterSessionID
first_delivery = $firstDelivery
seed = $seed
overflow_delivery_count = $overflowDeliveries.Count
preflight = $preflight
control = $control
checks = $checks
failed_checks = $failed
passed = ($failed.Count -eq 0)
}
} finally {
if ($adapterSessionID -and -not $SkipClose) {
try { [void](Invoke-Control -SessionID $adapterSessionID) } catch {}
}
if ($routeID) {
try {
[void](Invoke-Api -Method POST -Path "/clusters/$ClusterID/mesh/route-intents/$routeID/disable" -Body @{
actor_user_id = $ActorUserID
reason = "c19z3 stale mailbox preflight cleanup"
})
} catch {
Write-Warning "cleanup failed after c19z3 smoke: $($_.Exception.Message)"
}
}
}
$fullResultPath = Join-Path $repoRoot $ResultPath
$resultDir = Split-Path -Parent $fullResultPath
if ($resultDir) { New-Item -ItemType Directory -Force -Path $resultDir | Out-Null }
$result | ConvertTo-Json -Depth 100 | Set-Content -Encoding UTF8 -Path $fullResultPath
if (-not $result.passed) {
throw "C19Z3 remote workspace mailbox stale preflight smoke failed. Result: $fullResultPath Failed: $($failed -join ', ')"
}
Write-Host "C19Z3 remote workspace mailbox stale preflight smoke passed. Result: $fullResultPath"
$result