lib/icinga/plugin/New-IcingaCheck.psm1
Import-IcingaLib icinga\enums; Import-IcingaLib core\tools; function New-IcingaCheck() { param( [string]$Name = '', $Value = $null, $Unit = $null, [string]$Minimum = '', [string]$Maximum = '', $ObjectExists = -1, $Translation = $null, [switch]$NoPerfData ); $Check = New-Object -TypeName PSObject; $Check | Add-Member -membertype NoteProperty -name 'name' -value $Name; $Check | Add-Member -membertype NoteProperty -name 'verbose' -value 0; $Check | Add-Member -membertype NoteProperty -name 'messages' -value @(); $Check | Add-Member -membertype NoteProperty -name 'oks' -value @(); $Check | Add-Member -membertype NoteProperty -name 'warnings' -value @(); $Check | Add-Member -membertype NoteProperty -name 'criticals' -value @(); $Check | Add-Member -membertype NoteProperty -name 'unknowns' -value @(); $Check | Add-Member -membertype NoteProperty -name 'okchecks' -value @(); $Check | Add-Member -membertype NoteProperty -name 'warningchecks' -value @(); $Check | Add-Member -membertype NoteProperty -name 'criticalchecks' -value @(); $Check | Add-Member -membertype NoteProperty -name 'unknownchecks' -value @(); $Check | Add-Member -membertype NoteProperty -name 'value' -value $Value; $Check | Add-Member -membertype NoteProperty -name 'exitcode' -value -1; $Check | Add-Member -membertype NoteProperty -name 'unit' -value $Unit; $Check | Add-Member -membertype NoteProperty -name 'spacing' -value 0; $Check | Add-Member -membertype NoteProperty -name 'compiled' -value $FALSE; $Check | Add-Member -membertype NoteProperty -name 'perfdata' -value (-Not $NoPerfData); $Check | Add-Member -membertype NoteProperty -name 'warning' -value ''; $Check | Add-Member -membertype NoteProperty -name 'critical' -value ''; $Check | Add-Member -membertype NoteProperty -name 'minimum' -value $Minimum; $Check | Add-Member -membertype NoteProperty -name 'maximum' -value $Maximum; $Check | Add-Member -membertype NoteProperty -name 'objectexists' -value $ObjectExists; $Check | Add-Member -membertype NoteProperty -name 'translation' -value $Translation; $Check | Add-Member -membertype NoteProperty -name 'checks' -value $null; $Check | Add-Member -membertype NoteProperty -name 'completed' -value $FALSE; $Check | Add-Member -membertype NoteProperty -name 'checkcommand' -value ''; $Check | Add-Member -membertype NoteProperty -name 'checkpackage' -value $FALSE; $Check | Add-Member -membertype ScriptMethod -name 'HandleDaemon' -value { # Only apply this once the checkcommand is set if ([string]::IsNullOrEmpty($this.checkcommand) -Or $global:IcingaDaemonData.FrameworkRunningAsDaemon -eq $FALSE) { return; } if ($global:IcingaDaemonData.ContainsKey('BackgroundDaemon') -eq $FALSE) { return; } if ($global:IcingaDaemonData.BackgroundDaemon.ContainsKey('ServiceCheckScheduler') -eq $FALSE) { return; } if ($global:IcingaDaemonData.BackgroundDaemon.ServiceCheckScheduler.ContainsKey($this.checkcommand)) { if ($global:IcingaDaemonData.BackgroundDaemon.ServiceCheckScheduler[$this.checkcommand]['results'].ContainsKey($this.name) -eq $FALSE) { $global:IcingaDaemonData.BackgroundDaemon.ServiceCheckScheduler[$this.checkcommand]['results'].Add( $this.name, [hashtable]::Synchronized(@{}) ); } $global:IcingaDaemonData.BackgroundDaemon.ServiceCheckScheduler[$this.checkcommand]['results'][$this.name].Add( (Get-IcingaUnixTime), $this.value ); } } $Check | Add-Member -membertype ScriptMethod -name 'AddSpacing' -value { $this.spacing += 1; } $Check | Add-Member -membertype ScriptMethod -name 'AssignCheckCommand' -value { param($CheckCommand); $this.checkcommand = $CheckCommand; $this.HandleDaemon(); } $Check | Add-Member -membertype ScriptMethod -name 'GetWarnings' -value { return $this.warningchecks; } $Check | Add-Member -membertype ScriptMethod -name 'GetCriticals' -value { return $this.criticalchecks; } $Check | Add-Member -membertype ScriptMethod -name 'GetUnknowns' -value { return $this.unknownchecks; } $Check | Add-Member -membertype ScriptMethod -name 'SetUnknown' -value { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Unknown, $null, $null ); return $this; } $Check | Add-Member -membertype ScriptMethod -name 'SetWarning' -value { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, $null, $null ); return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnOutOfRange' -value { param($warning); if ([string]::IsNullOrEmpty($warning)) { return $this; } if ((Test-Numeric $warning)) { $this.WarnIfGreaterThan($warning).WarnIfLowerThan(0) | Out-Null; } else { [array]$thresholds = $warning.Split(':'); [string]$rangeMin = $thresholds[0]; [string]$rangeMax = $thresholds[1]; $negate = $rangeMin.Contains('@'); $rangeMin = $rangeMin.Replace('@', ''); if (-Not $negate -And (Test-Numeric $rangeMin) -And (Test-Numeric $rangeMax)) { $this.WarnIfLowerThan($rangeMin).WarnIfGreaterThan($rangeMax) | Out-Null; } elseif ((Test-Numeric $rangeMin) -And [string]::IsNullOrEmpty($rangeMax) -eq $TRUE) { $this.WarnIfLowerThan($rangeMin) | Out-Null; } elseif ($rangeMin -eq '~' -And (Test-Numeric $rangeMax)) { $this.WarnIfGreaterThan($rangeMax) | Out-Null; } elseif ($negate -And (Test-Numeric $rangeMin) -And (Test-Numeric $rangeMax)) { $this.WarnIfBetweenAndEqual($rangeMin, $rangeMax) | Out-Null; } else { $this.AddMessage( [string]::Format( 'Invalid range specified for Warning argument: "{0}" of check {1}', $warning, $this.name ), $IcingaEnums.IcingaExitCode.Unknown ) $this.exitcode = $IcingaEnums.IcingaExitCode.Unknown; return $this; } } $this.warning = $warning; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfLike' -value { param($warning); if ($this.value -Like $warning) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, $warning, 'like' ); } $this.warning = $warning; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfNotLike' -value { param($warning); if (-Not ($this.value -Like $warning)) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, $warning, 'not like' ); } $this.warning = $warning; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfMatch' -value { param($warning); if ($this.value -eq $warning) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, $warning, 'matching' ); } $this.warning = $warning; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfNotMatch' -value { param($warning); if ($this.value -ne $warning) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, $warning, 'not matching' ); } $this.warning = $warning; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfBetweenAndEqual' -value { param($min, $max); if ($this.value -ge $min -And $this.value -le $max) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, [string]::Format('{0}:{1}', $min, $max), 'between' ); } $this.warning = [string]::Format('{0}:{1}', $min, $max); return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfBetween' -value { param($min, $max); if ($this.value -gt $min -And $this.value -lt $max) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, [string]::Format('{0}:{1}', $min, $max), 'between' ); } $this.warning = [string]::Format('{0}:{1}', $min, $max); return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfLowerThan' -value { param($warning); if ($this.value -lt $warning) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, $warning, 'lower than' ); } $this.warning = $warning; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfLowerEqualThan' -value { param($warning); if ($this.value -le $warning) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, $warning, 'lower or equal than' ); } $this.warning = $warning; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfGreaterThan' -value { param($warning); if ($this.value -gt $warning) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, $warning, 'greater than' ); } $this.warning = $warning; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'WarnIfGreaterEqualThan' -value { param($warning); if ($this.value -ge $warning) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Warning, $warning, 'greater or equal than' ); } $this.warning = $warning; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'SetCritical' -value { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, $null, $null ); return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritOutOfRange' -value { param($critical); if ([string]::IsNullOrEmpty($critical)) { return $this; } if ((Test-Numeric $critical)) { $this.CritIfGreaterThan($critical).CritIfLowerThan(0) | Out-Null; } else { [array]$thresholds = $critical.Split(':'); [string]$rangeMin = $thresholds[0]; [string]$rangeMax = $thresholds[1]; $negate = $rangeMin.Contains('@'); $rangeMin = $rangeMin.Replace('@', ''); if (-Not $negate -And (Test-Numeric $rangeMin) -And (Test-Numeric $rangeMax)) { $this.CritIfLowerThan($rangeMin).CritIfGreaterThan($rangeMax) | Out-Null; } elseif ((Test-Numeric $rangeMin) -And [string]::IsNullOrEmpty($rangeMax) -eq $TRUE) { $this.CritIfLowerThan($rangeMin) | Out-Null; } elseif ($rangeMin -eq '~' -And (Test-Numeric $rangeMax)) { $this.CritIfGreaterThan($rangeMax) | Out-Null; } elseif ($negate -And (Test-Numeric $rangeMin) -And (Test-Numeric $rangeMax)) { $this.CritIfBetweenAndEqual($rangeMin, $rangeMax) | Out-Null; } else { $this.AddMessage( [string]::Format( 'Invalid range specified for Critical argument: "{0}" of check {1}', $critical, $this.name ), $IcingaEnums.IcingaExitCode.Unknown ) $this.exitcode = $IcingaEnums.IcingaExitCode.Unknown; return $this; } } $this.critical = $critical; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfLike' -value { param($critical); if ($this.value -Like $critical) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, $critical, 'like' ); } $this.critical = $critical; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfNotLike' -value { param($critical); if (-Not ($this.value -Like $critical)) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, $critical, 'not like' ); } $this.critical = $critical; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfMatch' -value { param($critical); if ($this.value -eq $critical) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, $critical, 'matching' ); } $this.critical = $critical; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfNotMatch' -value { param($critical); if ($this.value -ne $critical) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, $critical, 'not matching' ); } $this.critical = $critical; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfBetweenAndEqual' -value { param($min, $max); if ($this.value -ge $min -And $this.value -le $max) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, [string]::Format('{0}:{1}', $min, $max), 'between' ); } $this.critical = [string]::Format('{0}:{1}', $min, $max); return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfBetween' -value { param($min, $max); if ($this.value -gt $min -And $this.value -lt $max) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, [string]::Format('{0}:{1}', $min, $max), 'between' ); } $this.critical = [string]::Format('{0}:{1}', $min, $max); return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfLowerThan' -value { param($critical); if ($this.value -lt $critical) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, $critical, 'lower than' ); } $this.critical = $critical; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfLowerEqualThan' -value { param($critical); if ($this.value -le $critical) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, $critical, 'lower or equal than' ); } $this.critical = $critical; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfGreaterThan' -value { param($critical); if ($this.value -gt $critical) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, $critical, 'greater than' ); } $this.critical = $critical; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'CritIfGreaterEqualThan' -value { param($critical); if ($this.value -ge $critical) { $this.AddInternalCheckMessage( $IcingaEnums.IcingaExitCode.Critical, $critical, 'greater or equal than' ); } $this.critical = $critical; return $this; } $Check | Add-Member -membertype ScriptMethod -name 'TranslateValue' -value { param($value); $value = Format-IcingaPerfDataValue $value; if ($null -eq $this.translation -Or $null -eq $value) { return $value; } $checkValue = $value; if ((Test-Numeric $checkValue)) { $checkValue = [int]$checkValue; } else { $checkValue = [string]$checkValue; } if ($this.translation.ContainsKey($checkValue)) { return $this.translation[$checkValue]; } return $value; } $Check | Add-Member -membertype ScriptMethod -name 'AddInternalCheckMessage' -value { param($state, $value, $type); if ($this.objectexists -ne -1 -And $null -eq $this.objectexists) { $this.SetExitCode($IcingaEnums.IcingaExitCode.Unknown); $this.AddMessage([string]::Format( '{0} does not exist', $this.name ), $IcingaEnums.IcingaExitCode.Unknown); return; } $this.SetExitCode($state); if ($null -eq $value -And $null -eq $type) { $this.AddMessage( $this.name, $state ); } else { $this.AddMessage( [string]::Format( '{0}: Value "{1}{4}" is {2} threshold "{3}{4}"', $this.name, $this.TranslateValue($this.value), $type, $this.TranslateValue($value), $this.unit ), $state ); } switch ($state) { $IcingaEnums.IcingaExitCode.Warning { $this.warning = $value; break; }; $IcingaEnums.IcingaExitCode.Critical { $this.critical = $value; break; }; } } $Check | Add-Member -membertype ScriptMethod -name 'AddMessage' -value { param($message, [int]$exitcode); [string]$outputMessage = [string]::Format( '{0} {1}', $IcingaEnums.IcingaExitCodeText[$exitcode], $message ); $this.messages += $outputMessage; switch ([int]$exitcode) { $IcingaEnums.IcingaExitCode.Ok { $this.oks += $outputMessage; break; }; $IcingaEnums.IcingaExitCode.Warning { $this.warnings += $outputMessage; break; }; $IcingaEnums.IcingaExitCode.Critical { $this.criticals += $outputMessage; break; }; $IcingaEnums.IcingaExitCode.Unknown { $this.unknowns += $outputMessage; break; }; } } $Check | Add-Member -membertype ScriptMethod -name 'AddCheckStateArrays' -value { switch ([int]$this.exitcode) { $IcingaEnums.IcingaExitCode.Ok { $this.okchecks += $this.name; break; }; $IcingaEnums.IcingaExitCode.Warning { $this.warningchecks += $this.name; break; }; $IcingaEnums.IcingaExitCode.Critical { $this.criticalchecks += $this.name; break; }; $IcingaEnums.IcingaExitCode.Unknown { $this.unknownchecks += $this.name; break; }; } } $Check | Add-Member -membertype ScriptMethod -name 'PrintOkMessages' -value { param([string]$spaces); $this.OutputMessageArray($this.oks, $spaces); } $Check | Add-Member -membertype ScriptMethod -name 'PrintWarningMessages' -value { param([string]$spaces); $this.OutputMessageArray($this.warnings, $spaces); } $Check | Add-Member -membertype ScriptMethod -name 'PrintCriticalMessages' -value { param([string]$spaces); $this.OutputMessageArray($this.criticals, $spaces); } $Check | Add-Member -membertype ScriptMethod -name 'PrintUnknownMessages' -value { param([string]$spaces); $this.OutputMessageArray($this.unknowns, $spaces); } $Check | Add-Member -membertype ScriptMethod -name 'PrintAllMessages' -value { [string]$spaces = New-StringTree $this.spacing; $this.OutputMessageArray($this.unknowns, $spaces); $this.OutputMessageArray($this.criticals, $spaces); $this.OutputMessageArray($this.warnings, $spaces); $this.OutputMessageArray($this.oks, $spaces); } $Check | Add-Member -membertype ScriptMethod -name 'OutputMessageArray' -value { param($msgArray, [string]$spaces); foreach ($msg in $msgArray) { Write-IcingaPluginOutput ([string]::Format('{0}{1}', $spaces, $msg)); } } $Check | Add-Member -membertype ScriptMethod -name 'PrintOutputMessages' -value { [string]$spaces = New-StringTree $this.spacing; if ($this.unknowns.Count -ne 0) { $this.PrintUnknownMessages($spaces); } elseif ($this.criticals.Count -ne 0) { $this.PrintCriticalMessages($spaces); } elseif ($this.warnings.Count -ne 0) { $this.PrintWarningMessages($spaces); } else { if ($this.oks.Count -ne 0) { $this.PrintOkMessages($spaces); } } } $Check | Add-Member -membertype ScriptMethod -name 'SetExitCode' -value { param([int]$code); # Only overwrite the exit code in case our new value is greater then # the current one Ok > Warning > Critical if ([int]$this.exitcode -gt $code) { return $this; } switch ($code) { 0 { break; }; 1 { $this.oks = @(); break; }; 2 { $this.oks = @(); $this.warnings = @(); break; }; 3 { $this.oks = @(); $this.warnings = @(); $this.criticals = @(); break; }; } $this.exitcode = $code; } $Check | Add-Member -membertype ScriptMethod -name 'ValidateUnit' -value { if ($null -ne $this.unit -And (-Not $IcingaEnums.IcingaMeasurementUnits.ContainsKey($this.unit))) { $this.AddMessage( [string]::Format( 'Error on check "{0}": Usage of invalid plugin unit "{1}". Allowed units are: {2}', $this.name, $this.unit, (($IcingaEnums.IcingaMeasurementUnits.Keys | Sort-Object name) -Join ', ') ), $IcingaEnums.IcingaExitCode.Unknown ) $this.unit = ''; $this.exitcode = $IcingaEnums.IcingaExitCode.Unknown; } } $Check | Add-Member -membertype ScriptMethod -name 'AddOkOutput' -value { if ([int]$this.exitcode -eq -1) { $this.exitcode = $IcingaEnums.IcingaExitCode.Ok; $this.AddMessage( [string]::Format( '{0}: {1}{2}', $this.name, $this.TranslateValue($this.value), $this.unit ), $IcingaEnums.IcingaExitCode.Ok ); } } $Check | Add-Member -membertype ScriptMethod -name 'SilentCompile' -value { if ($this.compiled) { return; } $this.AddOkOutput(); $this.compiled = $TRUE; $this.AddCheckStateArrays(); } $Check | Add-Member -membertype ScriptMethod -name 'Compile' -value { param([bool]$Verbose = $FALSE); if ($this.compiled) { return; } $this.AddOkOutput(); $this.compiled = $TRUE; if ($Verbose) { $this.PrintOutputMessages(); } $this.AddCheckStateArrays(); return $this.exitcode; } $Check | Add-Member -membertype ScriptMethod -name 'GetPerfData' -value { if ($this.completed -Or -Not $this.perfdata) { return $null; } $this.AutodiscoverMinMax(); $this.completed = $TRUE; [string]$LabelName = (Format-IcingaPerfDataLabel $this.name); $value = ConvertTo-Integer -Value $this.value -NullAsEmpty; $warning = ConvertTo-Integer -Value $this.warning -NullAsEmpty; $critical = ConvertTo-Integer -Value $this.critical -NullAsEmpty; $perfdata = @{ 'label' = $LabelName; 'perfdata' = ''; 'unit' = $this.unit; 'value' = (Format-IcingaPerfDataValue $value); 'warning' = (Format-IcingaPerfDataValue $warning); 'critical' = (Format-IcingaPerfDataValue $critical); 'minimum' = (Format-IcingaPerfDataValue $this.minimum); 'maximum' = (Format-IcingaPerfDataValue $this.maximum); 'package' = $FALSE; }; return $perfdata; } $Check | Add-Member -membertype ScriptMethod -name 'AutodiscoverMinMax' -value { if ([string]::IsNullOrEmpty($this.minimum) -eq $FALSE -Or [string]::IsNullOrEmpty($this.maximum) -eq $FALSE) { return; } switch ($this.unit) { '%' { $this.minimum = '0'; $this.maximum = '100'; if ($this.value -gt $this.maximum) { $this.maximum = $this.value } break; } } } $Check.ValidateUnit(); $Check.HandleDaemon(); return $Check; } |