DSCResources/cSNMPTrapDestination/cSNMPTrapDestination.psm1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
function Get-TargetResource 
{
    [CmdletBinding()]
    [OutputType([System.Collections.Hashtable])]
    param 
    (
        [parameter(Mandatory = $true)]
        [System.String]
        $Destination,

        [System.String]
        $Community        
    )

    Write-Verbose "Gathering all permitted Managers"
    $Destinations = [PSCustomObject](Get-ItemProperty "HKLM:\SYSTEM\CurrentControlSet\Services\SNMP\Parameters\TrapConfiguration\$Community").psbase.properties | ? { 
            $_.Name -notin @('PSDrive','PSProvider','PSCHildName','PSPath','PSParentPath') 
         } | Select Name,Value

    if (Get-ItemProperty "HKLM:\SYSTEM\CurrentControlSet\Services\SNMP\Parameters\TrapConfiguration\$Community") {
        $Script:DestinationList = ""
        $ofs = "="
        $Destinations | % { $Script:DestinationList += ","+"$($_.Name,$_.Value)" }
    
    
        $ReturnValue = @{
            CommunityList=$Script:DestinationList.substring(1)
        }
    }
    else {
        $ReturnValue = @{
            CommunityList="No destinations specified"
        }
    }
    $ReturnValue
}


function Set-TargetResource 
{
    [CmdletBinding()]
    param 
    (
        [parameter(Mandatory = $true)]
        [System.String]
        $Destination,
        
        [System.String]
        $Community,

        [ValidateSet("Present","Absent")]
        [System.String]
        $Ensure

    )
    
    # Gather all registered permitted managers
    $Destinations = [PSCustomObject](Get-ItemProperty "HKLM:\SYSTEM\CurrentControlSet\Services\SNMP\Parameters\TrapConfiguration\$Community").psbase.properties | ? { 
            $_.Name -notin @('PSDrive','PSProvider','PSCHildName','PSPath','PSParentPath') 
         } | Select Name,Value
    switch ($Ensure) {
        "Present" {
            $Destinations | % { $_.Name = [int]$_.Name }
            [Int]$LastNum = ($Destinations |  Sort-Object Name | Select Name -Last 1).Name
            $LastNum++
            Write-Verbose "Adding new Manager to permitted list"
            New-ItemProperty -Path "HKLM:\SYSTEM\CurrentControlSet\Services\SNMP\Parameters\TrapConfiguration\$Community" -Name $LastNum -PropertyType String -Value $Destination
        }
        "Absent" {
            Write-Verbose "Removing Manager of permitted list"
            $Destinations | ? { $_.value -eq $Destination } | % {
                $Name = $_.Name.Trim()
                Remove-ItemProperty "HKLM:\SYSTEM\CurrentControlSet\Services\SNMP\Parameters\TrapConfiguration\$Community" -Name $Name
            }
        }
    }
}


function Test-TargetResource 
{
    [CmdletBinding()]
    param 
    (
        [parameter(Mandatory = $true)]
        [System.String]
        $Destination,

        [System.String]
        $Community,

        [ValidateSet("Present","Absent")]
        [System.String]
        $Ensure
    )

    Write-Verbose "Gathering all permitted Managers"
    $Destinations = [PSCustomObject](Get-ItemProperty "HKLM:\SYSTEM\CurrentControlSet\Services\SNMP\Parameters\TrapConfiguration\$Community").psbase.properties | ? { 
            $_.Name -notin @('PSDrive','PSProvider','PSCHildName','PSPath','PSParentPath') 
         } | Select Name,Value

    #Building the Hashtable
    $Destinations | ? { $_.Value -eq $Destination } | % {
        if ($Ensure -eq "Present") {
            if ($_.Value -eq $Destination) { $Return = $true }
            else { $Return = $false }
        }
        elseif ($Ensure -eq "Absent") {
            if ($_.Value -eq $Destination) { $Return = $false }
            else { $Return = $true }
        }
    }
    $Return
}


Export-ModuleMember -Function *-TargetResource