非永続仮想デバイス インフラストラクチャ (NP-VDI)

.SYNOPSIS

非永続エージェントを更新し、更新の完了を待機してから、任意でエージェントを保護します。

.PARAMETER TimeoutSec

合計タイムアウト (秒)。

.PARAMETER UpgradeCode

更新中のエージェントのコードをアップグレードします。

.PARAMETER DoNotSeal

このパラメータは、更新後にエージェントを保護しないことを指定します。

#>

param

(

[int] $TimeoutSec = 300,

[guid] $UpgradeCode = "{0C5D7E9C-0424-462D-A570-2A7469EE3739}", # Neurons Agent

[switch] $DoNotSeal

)

function Check-UpdatersComplete

{

$updaters = Get-Process -Name "STAgentUpdater" -ErrorAction SilentlyContinue

return ($updaters.Count -eq 0)

}

function Wait-UpdatersComplete([DateTime] $timeoutDateTime, [int] $checkIntervalSec)

{

while ((Get-Date) -lt $timeoutDateTime)

{

if (Check-UpdatersComplete)

{

return

}

Start-Sleep -Seconds $checkIntervalSec

}

throw "すべての更新の完了待機がタイムアウトしました。"

}

function Get-AgentVersionFromManifest([string] $manifestPath, [string] $platformKey)

{

if (-not (Test-Path $manifestPath))

{

throw "'$manifestPath' でマニフェストが見つかりませんでした。"

}

[xml]$manifestXml = Get-Content $manifestPath

$agentFile = $manifestXml.manifest.files.file | Where-Object { $_.key -eq $platformKey }

if (-not ($agentFile))

{

throw "key='$platformKey' を持つエージェント エントリが見つかりません。"

}

if (-not ($agentFile.version))

{

throw "マニフェストでエージェント バージョンを検索できません。"

}

return [version]::Parse($agentFile.version)

}

function Get-CurrentAgentVersion([guid] $agentUpgradeCode)

{

$agentProductCode = Get-ProductCode -UpgradeCode $agentUpgradeCode

$agentObject = Get-WmiObject -Class Win32_Product | Where-Object { [guid]::Parse($_.IdentifyingNumber) -eq $agentProductCode }

if (-not ($agentObject))

{

throw "インストールされたエージェントが見つかりません。"

}

if (-not ($agentObject.Version))

{

throw "インストールされたエージェント バージョンが見つかりません。"

}

return [version]::Parse($agentObject.Version)

}

function Check-AgentUpToDate([string] $manifestPath, [guid] $agentUpgradeCode, [string] $platformKey)

{

$manifestVersion = Get-AgentVersionFromManifest -ManifestPath $manifestPath -PlatformKey $platformKey

$currentVersion = Get-CurrentAgentVersion -AgentUpgradeCode $agentUpgradeCode

return ($currentVersion -ge $manifestVersion)

}

function Wait-AgentUpdated([DateTime] $timeoutDateTime, [int] $checkIntervalSec, [string] $manifestPath, [guid] $agentUpgradeCode, [string] $platformKey)

{

while ((Get-Date) -lt $timeoutDateTime)

{

if (Check-AgentUpToDate -ManifestPath $manifestPath -AgentUpgradeCode $agentUpgradeCode -PlatformKey $platformKey)

{

return

}

Start-Sleep -Seconds $checkIntervalSec

}

throw "エージェント更新の待機がタイムアウトしました。"

}

function Update-Agent([DateTime] $timeoutDateTime, [string] $manifestPath, [guid] $agentUpgradeCode, [string] $agentCtlPath, [string] $platformKey)

{

$timeoutSec = [math]::Ceiling(($timeoutDateTime - (Get-Date)).TotalSeconds)

if ($timeoutSec -le 0)

{

throw "タイムアウトが短いため、チェックインを実行できません。"

}

& $agentCtlPath update --checkin --wait $timeoutSec

if (-not $?)

{

throw "チェックイン操作が失敗しました。"

}

& $agentCtlPath update --all

if (-not $?)

{

throw "更新操作が失敗しました。"

}

Wait-AgentUpdated -TimeoutDateTime $timeoutDateTime -CheckIntervalSec 1 -ManifestPath $manifestPath -AgentUpgradeCode $agentUpgradeCode -PlatformKey $platformKey

Wait-UpdatersComplete -TimeoutDateTime $timeoutDateTime -CheckIntervalSec 1

$timeoutSec = [math]::Ceiling(($timeoutDateTime - (Get-Date)).TotalSeconds)

if ($timeoutSec -le 0)

{

throw "タイムアウトが短いため、最終更新を実行できません。"

}

& $agentCtlPath update --all --wait $timeoutSec

if (-not $?)

{

throw "最終更新操作が失敗しました。"

}

}

function Get-ProductCode([guid] $upgradeCode)

{

$upgradeCodeStr = $upgradeCode.ToString("B") # Format as {xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx}

$wmiFilter = "Property='UpgradeCode' AND Value='$upgradeCodeStr'"

$productCode = @(Get-WmiObject -Class Win32_Property -Filter $wmiFilter).ProductCode

if ($productCode.Count -eq 0)

{

throw "アップグレード コード '$upgradeCode' が見つかりません。"

}

if ($productCode.Count -gt 1)

{

throw "同じアップグレード コード '$upgradeCode' を持つ製品が複数見つかりました。"

}

return [guid]::Parse($productCode)

}

function Get-AgentEnvironment([string] $installDirPath)

{

$configPath = Join-Path $installDirPath "AgentEnvironment.config"

if (-Not (Test-Path $configPath))

{

throw "$configPath に構成ファイルが見つかりません。"

}

[xml]$agentEnvironment = Get-Content $configPath

return $agentEnvironment

}

function Get-InstallLocation([guid] $productCode)

{

$productObject = Get-WmiObject -Class Win32_Product | Where-Object { [guid]::Parse($_.IdentifyingNumber) -eq $productCode }

if ($null -eq $productObject)

{

throw "製品 GUID が見つかりませんでした。"

}

return $productObject.InstallLocation

}

$timeoutDateTime = (Get-Date).AddSeconds($TimeoutSec)

Write-Host "情報を収集しています..."

try

{

$agentProductCode = Get-ProductCode -UpgradeCode $UpgradeCode

$installDirPath = Get-InstallLocation -ProductCode $agentProductCode

Write-Host "インストール ディレクトリ: $installDirPath"

$agentEnvironment = Get-AgentEnvironment -InstallDirPath $installDirPath

$dataDirPath = $agentEnvironment.agentEnvironment.agentDataDirectory

Write-Host "データ ディレクトリ: $dataDirPath"

$regPath = Join-Path "HKLM:" $agentEnvironment.agentEnvironment.registryPath

Write-Host "レジストリ キー: $regPath"

$platformKey = Get-ItemProperty -Path $regPath -Name "PlatformManifestKey" -ErrorAction Stop | Select-Object -ExpandProperty "PlatformManifestKey" -ErrorAction Stop

Write-Host "プラットフォーム キー: $platformKey"

$agentCtlPath = Join-Path $installDirPath "STAgentCtl.exe"

$manifestPath = Join-Path $dataDirPath "manifest.xml"

}

catch

{

Write-Error "エラー: $_"

exit -1

}

if ((Get-Date) -gt $timeoutDateTime)

{

Write-Error "エラー: エージェント更新を開始する前にタイムアウトが期限切れになりました。"

exit -1

}

Write-Host "エージェントの保護を解除しています..."

& $agentCtlPath unseal

if (-not $?)

{

Write-Error "エラー: 保護解除操作を実行できません。"

exit -1

}

Write-Host "エージェントを更新しています..."

$result = -1

try

{

Update-Agent -TimeoutDateTime $timeoutDateTime -ManifestPath $manifestPath -AgentUpgradeCode $UpgradeCode -AgentCtlPath $agentCtlPath -PlatformKey $platformKey

$result = 0

}

catch

{

Write-Error "エラー: $_"

}

if ($DoNotSeal)

{

if ($result -eq 0)

{

Write-Host "エージェントが正常に更新されました。"

}

else

{

Write-Error "エラー: エージェント更新に失敗しました。"

}

exit $result

}

Write-Host "エージェントを保護しています..."

& $agentCtlPath seal

if (-not $?)

{

Write-Error "エラー: エージェントを保護できません"

exit -1

}

if ($result -eq 0)

{

Write-Host "エージェントが正常に更新され、保護されました。"

}

exit $result

(@ECHO OFF

set pw=supersecret

set logfile=C:\ProgramData\Ivanti\Ivanti Cloud Agent\Logs\agentkeyrestore.log

set certFile=C:\Program Files\Ivanti\Ivanti Cloud Agent\agent.pfx

set scriptPath=%0%

ECHO %date% %time% エージェント キーの復元が開始されました >> "%logfile%"

IF EXIST "%certFile%" (

certutil -f -p "%pw%" -privatekey -importPFX "Ivanti Cloud Agent Agent" "%certFile%" NoCert,NoExport >> "%logfile%" 2>&1

IF %ERRORLEVEL% NEQ 0 (

ECHO エージェント キーのインポートに失敗しました >> "%logfile%"

)

ECHO エージェント キー ファイルを削除しています >> "%logfile%"

del "%certFile%" >> "%logfile%" 2>&1

IF %ERRORLEVEL% NEQ 0 (

ECHO エージェント キー ファイルの削除に失敗しました >> "%logfile%"

)

ECHO %date% %time% エージェント キーの復元が完了しました >> "%logfile%"

) ELSE (

ECHO %date% %time% エージェント キーの復元に失敗しました: エージェント キー ファイルが見つかりません >> "%logfile%"

)

ECHO スクリプトを削除しています >> "%logfile%"

del "%scriptPath%" >> "%logfile%" 2>&1

"REM スクリプト ファイルが削除されたため、この時点から、これ以上コマンドを実行することはできません"