Compare commits

...

7 Commits

Author SHA1 Message Date
Asif Bacchus cccd7ca3d6 feature(wakeup): only display connection err on last iteration 2021-09-06 16:08:33 -06:00
Asif Bacchus 140edc1cb1 fix(wakeup): fix trailing countdown error 2021-09-06 16:04:36 -06:00
Asif Bacchus 56a910d714 refactor(wakeup): use verbose output stream
- move connection check error details to verbose stream
2021-09-06 16:04:14 -06:00
Asif Bacchus fc3a39b0b9 refactor(wakeup): use cmdletbinding, change var db name 2021-09-06 16:03:34 -06:00
Asif Bacchus 47c313a26a refactor(wakeup): rewrite connectivity check timer loop
- use do loop to run at least one check
- check total time vs timeout in trailing while condition
2021-09-06 15:45:06 -06:00
Asif Bacchus 5332818990 feature(wakeup): run multiple connectivity checks up to timeout 2021-09-06 15:38:36 -06:00
Asif Bacchus 0119923ed7 refactor(wakeup): rework params, conn-check multiple err msg
- rename connectivity check parameters for consistency
- allow multiple errors to be reported for connectivity check for
troubleshooting
2021-09-06 15:16:21 -06:00
1 changed files with 82 additions and 37 deletions

View File

@ -1,30 +1,37 @@
<#
#>
[CmdletBinding()]
param
(
# Array of strings listing computer name(s) to wake up. Names will be matched using RegEx
[Alias("Computer")]
[String[]]
$TargetComputer = "",
# XML database file containing computer details
[Alias("List", "database", "db")]
# XML database file containing computer details (Default: WOLDatabase.xml)
[Alias("List", "Database")]
[String]
$WolDatabase = "WOLDatabase.xml",
# Skip connectivity readiness check
# Skip connectivity check
[Alias("NoCheck", "NoReadiness")]
[Switch]
$NoConnectivityCheck = $false,
# Initial wait period in seconds before first connectivity check
$NoConnectivityCheck,
# Delay before running first connectivity check (Default: 10)
[Alias("Delay")]
[Int]
$WaitBeforeConnectivityCheck = 10,
# connectivity readiness check total timeout
$ConnectivityCheckDelay = 10,
# Seconds before aborting connectivity check (Default: 300)
[Alias("Timeout")]
[Int]
$Timeout = 300,
# connectivity readiness port to check
$ConnectivityCheckTimeout = 300,
# Seconds between connectivity checks (Default: 30)
[Alias("Interval")]
[Int]
$ConnectivityCheckInterval = 30,
# Port to use for connectivity check (Default: 3389-RDP)
[Alias("CheckPort")]
[Int]
$ConnectivityPort = 3389
$ConnectivityCheckPort = 3389
)
function exitError($errMessage, $PSItem, $exitCode = 1)
@ -84,14 +91,14 @@ if ( [String]::IsNullOrWhiteSpace($TargetComputer))
Write-Host
# read database and assemble list of target computers
[xml]$db = Get-Content -Path $WolDatabase
$broadcastIP = $db.WOLDatabase.Configuration.BroadcastAddress
$port = $db.WOLDatabase.Configuration.Port
$dnsSuffix = $db.WOLDatabase.Configuration.DnsSuffix
[xml]$wolDb = Get-Content -Path $WolDatabase
$broadcastIP = $wolDb.WOLDatabase.Configuration.BroadcastAddress
$port = $wolDb.WOLDatabase.Configuration.Port
$dnsSuffix = $wolDb.WOLDatabase.Configuration.DnsSuffix
$targetComputers = [System.Collections.Generic.List[PSObject]]::new()
$TargetComputer | ForEach-Object {
$tgt = $_
$db.WOLDatabase.Computers.Computer | Where-Object { $_.name -match $tgt } | ForEach-Object { $targetComputers.Add($_) }
$wolDb.WOLDatabase.Computers.Computer | Where-Object { $_.name -match $tgt } | ForEach-Object { $targetComputers.Add($_) }
}
$removeFromTargetComputers = [System.Collections.Generic.List[String]]::new()
@ -125,6 +132,7 @@ $targetComputers | ForEach-Object {
else
{
Write-Host -ForegroundColor Red "[ERROR]"
# queue for removal from computers on which to run a connectivity check
$removeFromTargetComputers.Add($name)
}
}
@ -143,41 +151,78 @@ if ($NoConnectivityCheck)
}
elseif ($targetComputers.Count -eq 0)
{
exitError "No successful wake-up packets sent" -exitCode 5
exitError "No wake-up packets sent successfully" -exitCode 5
}
# wait for initialWaitTime seconds to let computer(s) wake up
for ($i = $WaitBeforeConnectivityCheck; $i -gt 0; $i--) {
# wait for initial delay seconds to let computer(s) wake up
for ($i = $ConnectivityCheckDelay; $i -gt 0; $i--) {
Write-Progress -Activity "Waiting for computer(s) to wake-up..." -SecondsRemaining $i
Start-Sleep 1
}
Write-Progress -Activity "Waiting for computer(s) to wake-up..." -Completed
# iterate computers and test connectivity
$targetComputers | ForEach-Object {
$name = $_.name
$friendlyName = $_.friendlyName
$fqdn = -join ($name, $dnsSuffix)
$connectionError = ""
# TODO: run connectivity test as background job
$connCheckTotalTime = 0
$removeFromTargetComputers.Clear()
do
{
$targetComputers | ForEach-Object {
$name = $_.name
$friendlyName = $_.friendlyName
$fqdn = -join ($name, $dnsSuffix)
$connectionError = @()
if ( [String]::IsNullOrWhiteSpace($friendlyName))
{
Write-Host "Testing connection readiness of '$name'... " -NoNewline
if ( [String]::IsNullOrWhiteSpace($friendlyName))
{
Write-Host "Testing connection readiness of '$name'... " -NoNewline
}
else
{
Write-Host "Testing connection readiness of '$friendlyName'... " -NoNewline
}
if (!(Test-NetConnection -ComputerName $fqdn -Port $ConnectivityCheckPort -InformationLevel Quiet -WarningAction SilentlyContinue -ErrorAction SilentlyContinue -ErrorVariable +connectionError))
{
if (($connCheckTotalTime + $ConnectivityCheckInterval) -gt $ConnectivityCheckTimeout)
{
# last iteration, display 'error'
Write-Host -ForegroundColor Red "[ERROR]"
$connectionError | ForEach-Object {
$errMsg = $_.ToString()
Write-Verbose "`t(Additional information: $errMsg)"
}
}
else
{
# more iterations pending, display 'retrying'
Write-Host -ForegroundColor Yellow "[RETRYING]"
}
}
else
{
Write-Host -ForegroundColor Green "[OK]"
# queue for removal from computers to check next round
$removeFromTargetComputers.Add($name)
}
}
else
{
Write-Host "Testing connection readiness of '$friendlyName'... " -NoNewline
# remove computers from targetComputers if connectivity check already successful
$removeFromTargetComputers | ForEach-Object {
$removeName = $_
$targetComputers.Remove(($targetComputers | Where-Object { $_.name -eq $removeName })) | Out-Null
}
if (!(Test-NetConnection -ComputerName $fqdn -Port $ConnectivityPort -InformationLevel Quiet -WarningAction SilentlyContinue -ErrorAction SilentlyContinue -ErrorVariable connectionError))
# sleep until next test interval and increase time counter
$connCheckTotalTime += $ConnectivityCheckInterval
if ($connCheckTotalTime -le $ConnectivityCheckTimeout -and $targetComputers.Count -gt 0)
{
Write-Host -ForegroundColor Red "[ERROR]"
Write-Host -ForegroundColor Red "`tAdditional information:", $connectionError[-1].ToString()
for ($i = $ConnectivityCheckInterval; $i -gt 0; $i--) {
Write-Progress -Activity "Waiting for next connectivity check..." -SecondsRemaining $i
Start-Sleep 1
}
Write-Progress -Activity "Waiting for next connectivity check..." -Completed
}
else
{
Write-Host -ForegroundColor Green "[OK]"
}
}
} while ($connCheckTotalTime -le $ConnectivityCheckTimeout -and $targetComputers.Count -gt 0)
# exit gracefully
exitGracefully