DSCResources/DSC_SqlProtocolTcpIp/DSC_SqlProtocolTcpIp.psm1

$script:sqlServerDscHelperModulePath = Join-Path -Path $PSScriptRoot -ChildPath '..\..\Modules\SqlServerDsc.Common'
$script:resourceHelperModulePath = Join-Path -Path $PSScriptRoot -ChildPath '..\..\Modules\DscResource.Common'

Import-Module -Name $script:sqlServerDscHelperModulePath
Import-Module -Name $script:resourceHelperModulePath

$script:localizedData = Get-LocalizedData -DefaultUICulture 'en-US'

<#
    .SYNOPSIS
        Returns the current state of the SQL Server TCP/IP protocol IP address
        group for the specified SQL Server instance.
 
    .PARAMETER InstanceName
        Specifies the name of the SQL Server instance to enable the protocol for.
 
    .PARAMETER IpAddressGroup
        Specifies the name of the IP address group in the TCP/IP protocol, e.g.
        'IP1', 'IP2' etc., or 'IPAll'.
 
    .PARAMETER ServerName
        Specifies the host name of the SQL Server to be configured. If the
        SQL Server belongs to a cluster or availability group specify the host
        name for the listener or cluster group. Default value is the
        current computer name.
 
    .PARAMETER SuppressRestart
        If set to $true then the any attempt by the resource to restart the service
        is suppressed. The default value is $false.
 
    .PARAMETER RestartTimeout
        Timeout value for restarting the SQL Server services. The default value
        is 120 seconds.
 
    .NOTES
        The parameters SuppressRestart and RestartTimeout are part of the function
        Get-TargetResource to be able to return the value that the configuration
        have set, or the default values if not. If they weren't passed to the
        function Get-TargetResource we would have to always return $null which
        would indicate that they are not set at all.
 
        Thought this function should throw an exception if the address group is
        missing, but voted against it since run Test-DscConfiguration before
        running a configuration (that would configure NICs) would then fail.
        Instead choose to output a warning message indicating that the current
        state cannot be evaluated.
#>

function Get-TargetResource
{
    [CmdletBinding()]
    [OutputType([System.Collections.Hashtable])]
    param
    (
        [Parameter(Mandatory = $true)]
        [System.String]
        $InstanceName,

        [Parameter(Mandatory = $true)]
        [System.String]
        $IpAddressGroup,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String]
        $ServerName = (Get-ComputerName),

        [Parameter()]
        [System.Boolean]
        $SuppressRestart = $false,

        [Parameter()]
        [System.UInt16]
        $RestartTimeout = 120
    )

    $IpAddressGroup = Convert-IpAdressGroupCasing -IpAddressGroup $IpAddressGroup

    $returnValue = @{
        InstanceName      = $InstanceName
        IpAddressGroup    = $IpAddressGroup
        ServerName        = $ServerName
        SuppressRestart   = $SuppressRestart
        RestartTimeout    = $RestartTimeout
        Enabled           = $false
        IpAddress         = $null
        UseTcpDynamicPort = $false
        TcpPort           = $null
        IsActive          = $false
        AddressFamily     = $null
        TcpDynamicPort    = $null
    }

    # Getting the server protocol properties by using the computer name.
    $computerName = Get-ComputerName

    Write-Verbose -Message (
        $script:localizedData.GetCurrentState -f $IpAddressGroup, $InstanceName, $computerName
    )

    Import-SQLPSModule

    <#
        Must connect to the local machine name because $ServerName can point
        to a cluster instance or availability group listener.
    #>

    $getServerProtocolObjectParameters = @{
        ServerName   = $computerName
        Instance     = $InstanceName
        ProtocolName = 'TcpIp'
    }

    $serverProtocolProperties = Get-ServerProtocolObject @getServerProtocolObjectParameters

    if ($serverProtocolProperties)
    {
        if ($IpAddressGroup -in $serverProtocolProperties.IPAddresses.Name)
        {
            $ipAddressGroupObject = $serverProtocolProperties.IPAddresses[$IpAddressGroup]

            # Values for all IP adress groups.
            $currentTcpPort = $ipAddressGroupObject.IPAddressProperties['TcpPort'].Value
            $currentTcpDynamicPort = $ipAddressGroupObject.IPAddressProperties['TcpDynamicPorts'].Value

            # Get the current state of TcpDynamicPort.
            if (-not (
                    [System.String]::IsNullOrEmpty($currentTcpDynamicPort) `
                        -or [System.String]::IsNullOrWhiteSpace($currentTcpDynamicPort)
                )
            )
            {
                $returnValue.UseTcpDynamicPort = $true
                $returnValue.TcpDynamicPort = $currentTcpDynamicPort
            }

            # Get the current state of TcpPort.
            if (-not (
                    [System.String]::IsNullOrEmpty($currentTcpPort) `
                        -or [System.String]::IsNullOrWhiteSpace($currentTcpPort)
                )
            )
            {
                $returnValue.TcpPort = $currentTcpPort
            }

            # Values for all individual IP adress groups.
            switch ($IpAddressGroup)
            {
                'IPAll'
                {
                    <#
                        Left blank intentionally. There are no individual IP address
                        properties for the IP address group 'IPAll'.
                    #>

                }

                default
                {
                    $returnValue.AddressFamily = $ipAddressGroupObject.IPAddress.AddressFamily
                    $returnValue.IpAddress = $ipAddressGroupObject.IPAddress.IPAddressToString
                    $returnValue.Enabled = $ipAddressGroupObject.IPAddressProperties['Enabled'].Value
                    $returnValue.IsActive = $ipAddressGroupObject.IPAddressProperties['Active'].Value
                }
            }
        }
        else
        {
            <#
                The IP address groups are created automatically so if this happens
                there is something wrong with the network interfaces on the node
                that this resource can not solve.
            #>

            Write-Warning -Message (
                $script:localizedData.GetMissingIpAddressGroup -f $IpAddressGroup
            )
        }
    }

    return $returnValue
}

<#
    .SYNOPSIS
        Sets the desired state of the SQL Server TCP/IP protocol IP address
        group for the specified SQL Server instance.
 
    .PARAMETER InstanceName
        Specifies the name of the SQL Server instance to enable the protocol for.
 
    .PARAMETER IpAddressGroup
        Specifies the name of the IP address group in the TCP/IP protocol, e.g.
        'IP1', 'IP2' etc., or 'IPAll'.
 
    .PARAMETER ServerName
        Specifies the host name of the SQL Server to be configured. If the SQL
        Server belongs to a cluster or availability group specify the host name
        for the listener or cluster group. Default value is the current computer
        name.
 
    .PARAMETER Enabled
        Specified if the IP address group should be enabled or disabled. Only used if
        the IP address group is not set to 'IPAll'. If not specified, the existing
        value will not be changed.
 
    .PARAMETER IpAddress
        Specifies the IP address for the IP adress group. Only used if the IP address
        group is not set to 'IPAll'. If not specified, the existing value will not be
        changed.
 
    .PARAMETER UseTcpDynamicPort
        Specifies whether the SQL Server instance should use a dynamic port. If
        not specified the existing value will not be changed. This parameter is
        not allowed to be used at the same time as the parameter TcpPort.
 
    .PARAMETER TcpPort
        Specifies the TCP port(s) that SQL Server should be listening on. If the
        IP address should listen on more than one port, list all ports as a string
        value with the port numbers separated with a comma, e.g. '1433,1500,1501'.
        This parameter is limited to 2047 characters. If not specified, the existing
        value will not be changed.This parameter is not allowed to be used at the
        same time as the parameter UseTcpDynamicPort.
 
    .PARAMETER SuppressRestart
        If set to $true then the any attempt by the resource to restart the service
        is suppressed. The default value is $false.
 
    .PARAMETER RestartTimeout
        Timeout value for restarting the SQL Server services. The default value
        is 120 seconds.
#>

function Set-TargetResource
{
    [System.Diagnostics.CodeAnalysis.SuppressMessageAttribute('SqlServerDsc.AnalyzerRules\Measure-CommandsNeededToLoadSMO', '', Justification='The command Import-SQLPSModule is implicitly called when calling Compare-TargetResourceState')]
    [CmdletBinding()]
    param
    (
        [Parameter(Mandatory = $true)]
        [System.String]
        $InstanceName,

        [Parameter(Mandatory = $true)]
        [System.String]
        $IpAddressGroup,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String]
        $ServerName = (Get-ComputerName),

        [Parameter()]
        [System.Boolean]
        $Enabled,

        [Parameter()]
        [System.String]
        $IpAddress,

        [Parameter()]
        [System.Boolean]
        $UseTcpDynamicPort,

        [Parameter()]
        [System.String]
        $TcpPort,

        [Parameter()]
        [System.Boolean]
        $SuppressRestart = $false,

        [Parameter()]
        [System.UInt16]
        $RestartTimeout = 120
    )

    $IpAddressGroup = Convert-IpAdressGroupCasing -IpAddressGroup $IpAddressGroup

    <#
        Compare the current state against the desired state. Calling this will
        also import the necessary module to later call Get-ServerProtocolObject
        which uses the SMO class ManagedComputer.
    #>

    $propertyState = Compare-TargetResourceState @PSBoundParameters

    # Get all properties that are not in desired state.
    $propertiesNotInDesiredState = $propertyState.Where( { -not $_.InDesiredState })

    if ($propertiesNotInDesiredState.Count -gt 0)
    {
        # Getting the server protocol properties by using the computer name.
        $computerName = Get-ComputerName

        Write-Verbose -Message (
            $script:localizedData.SetDesiredState -f $IpAddressGroup, $InstanceName, $computerName
        )

        <#
            Must connect to the local machine name because $ServerName can point
            to a cluster instance or availability group listener.
        #>

        $getServerProtocolObjectParameters = @{
            ServerName   = $computerName
            Instance     = $InstanceName
            ProtocolName = 'TcpIp'
        }

        $serverProtocolProperties = Get-ServerProtocolObject @getServerProtocolObjectParameters

        if ($serverProtocolProperties)
        {
            if ($IpAddressGroup -in $serverProtocolProperties.IPAddresses.Name)
            {
                $ipAddressGroupObject = $serverProtocolProperties.IPAddresses[$IpAddressGroup]

                $isRestartNeeded = $false

                # Check if TcpPort property need updating.
                if ($propertiesNotInDesiredState.Where( { $_.ParameterName -eq 'TcpPort' }))
                {
                    $ipAddressGroupObject.IPAddressProperties['TcpPort'].Value = $TcpPort

                    # Should be using TcpPort, make sure dynamic ports are disabled.
                    $ipAddressGroupObject.IPAddressProperties['TcpDynamicPorts'].Value = ''

                    Write-Verbose -Message (
                        $script:localizedData.TcpPortHasBeenSet -f $TcpPort, $IpAddressGroup
                    )

                    $isRestartNeeded = $true
                }

                # Check if TcpDynamicPort property need updating.
                if ($propertiesNotInDesiredState.Where( { $_.ParameterName -eq 'UseTcpDynamicPort' }))
                {
                    <#
                        Enable TCP dynamic ports using a '0'. When the SQL Server
                        Database Engine is restarted it will get a dynamic port.
                    #>

                    $ipAddressGroupObject.IPAddressProperties['TcpDynamicPorts'].Value = '0'

                    <#
                        Should be using dynamic TCP port, make sure static TCP port
                        are disabled.
                    #>

                    $ipAddressGroupObject.IPAddressProperties['TcpPort'].Value = ''

                    Write-Verbose -Message (
                        $script:localizedData.TcpDynamicPortHasBeenSet -f $IpAddressGroup
                    )

                    $isRestartNeeded = $true
                }

                # Set individual protocol properties.
                switch ($IpAddressGroup)
                {
                    'IPAll'
                    {
                        <#
                            Left blank intentionally. There are no individual protocol
                            properties for the IP address group IPAll.
                        #>

                    }

                    default
                    {
                        # Check if Enable property need updating.
                        if ($propertiesNotInDesiredState.Where( { $_.ParameterName -eq 'Enabled' }))
                        {
                            $ipAddressGroupObject.IPAddressProperties['Enabled'].Value = $Enabled

                            if ($Enabled)
                            {
                                Write-Verbose -Message (
                                    $script:localizedData.GroupHasBeenEnabled -f $IpAddressGroup, $InstanceName
                                )
                            }
                            else
                            {
                                Write-Verbose -Message (
                                    $script:localizedData.GroupHasBeenDisabled -f $IpAddressGroup, $InstanceName
                                )
                            }

                            $isRestartNeeded = $true
                        }

                        # Check if Enabled property need updating.
                        if ($propertiesNotInDesiredState.Where( { $_.ParameterName -eq 'IpAddress' }))
                        {
                            # Casing of the property IpAddress is important!
                            $ipAddressGroupObject.IPAddressProperties['IpAddress'].Value = $IpAddress

                            Write-Verbose -Message (
                                $script:localizedData.IpAddressHasBeenSet -f $IpAddressGroup, $IpAddress
                            )

                            $isRestartNeeded = $true
                        }
                    }
                }

                $serverProtocolProperties.Alter()
            }
            else
            {
                $errorMessage = $script:localizedData.SetMissingIpAddressGroup -f $IpAddressGroup

                New-ObjectNotFoundException -Message $errorMessage
            }
        }
        else
        {
            $errorMessage = $script:localizedData.FailedToGetSqlServerProtocol

            New-InvalidOperationException -Message $errorMessage
        }

        if (-not $SuppressRestart -and $isRestartNeeded)
        {
            <#
                This is using the $ServerName to be able to restart a cluster
                instance or availability group listener.
            #>

            $restartSqlServiceParameters = @{
                ServerName   = $ServerName
                InstanceName = $InstanceName
                Timeout      = $RestartTimeout
                OwnerNode    = Get-ComputerName
            }

            Restart-SqlService @restartSqlServiceParameters
        }
        elseif ($isRestartNeeded)
        {
            Write-Warning -Message $script:localizedData.RestartSuppressed
        }
    }
    else
    {
        Write-Verbose -Message (
            $script:localizedData.GroupIsInDesiredState -f $IpAddressGroup, $InstanceName
        )
    }
}

<#
    .SYNOPSIS
        Determines the current state of the SQL Server TCP/IP protocol IP address
        group for the specified SQL Server instance.
 
    .PARAMETER InstanceName
        Specifies the name of the SQL Server instance to enable the protocol for.
 
    .PARAMETER IpAddressGroup
        Specifies the name of the IP address group in the TCP/IP protocol, e.g.
        'IP1', 'IP2' etc., or 'IPAll'.
 
    .PARAMETER ServerName
        Specifies the host name of the SQL Server to be configured. If the
        SQL Server belongs to a cluster or availability group specify the host
        name for the listener or cluster group. Default value is the current
        computer name.
 
    .PARAMETER Enabled
        Specified if the IP address group should be enabled or disabled. Only used if
        the IP address group is not set to 'IPAll'. If not specified, the existing
        value will not be changed.
 
    .PARAMETER IpAddress
        Specifies the IP address for the IP adress group. Only used if the IP address
        group is not set to 'IPAll'. If not specified, the existing value will not be
        changed.
 
    .PARAMETER UseTcpDynamicPort
        Specifies whether the SQL Server instance should use a dynamic port. If
        not specified the existing value will not be changed. This parameter is
        not allowed to be used at the same time as the parameter TcpPort.
 
    .PARAMETER TcpPort
        Specifies the TCP port(s) that SQL Server should be listening on. If the
        IP address should listen on more than one port, list all ports as a string
        value with the port numbers separated with a comma, e.g. '1433,1500,1501'.
        This parameter is limited to 2047 characters. If not specified, the existing
        value will not be changed.This parameter is not allowed to be used at the
        same time as the parameter UseTcpDynamicPort.
 
    .PARAMETER SuppressRestart
        If set to $true then the any attempt by the resource to restart the service
        is suppressed. The default value is $false.
 
    .PARAMETER RestartTimeout
        Timeout value for restarting the SQL Server services. The default value
        is 120 seconds.
#>

function Test-TargetResource
{
    [System.Diagnostics.CodeAnalysis.SuppressMessageAttribute('SqlServerDsc.AnalyzerRules\Measure-CommandsNeededToLoadSMO', '', Justification='The command Import-SQLPSModule is implicitly called when calling Compare-TargetResourceState')]
    [CmdletBinding()]
    [OutputType([System.Boolean])]
    param
    (
        [Parameter(Mandatory = $true)]
        [System.String]
        $InstanceName,

        [Parameter(Mandatory = $true)]
        [System.String]
        $IpAddressGroup,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String]
        $ServerName = (Get-ComputerName),

        [Parameter()]
        [System.Boolean]
        $Enabled,

        [Parameter()]
        [System.String]
        $IpAddress,

        [Parameter()]
        [System.Boolean]
        $UseTcpDynamicPort,

        [Parameter()]
        [System.String]
        $TcpPort,

        [Parameter()]
        [System.Boolean]
        $SuppressRestart = $false,

        [Parameter()]
        [System.UInt16]
        $RestartTimeout = 120
    )

    Write-Verbose -Message (
        $script:localizedData.TestDesiredState -f $IpAddressGroup, $InstanceName, $ServerName
    )

    $propertyState = Compare-TargetResourceState @PSBoundParameters


    if ($false -in $propertyState.InDesiredState)
    {
        $testTargetResourceReturnValue = $false

        Write-Verbose -Message (
            $script:localizedData.NotInDesiredState -f $IpAddressGroup, $InstanceName
        )
    }
    else
    {
        $testTargetResourceReturnValue = $true

        Write-Verbose -Message (
            $script:localizedData.InDesiredState -f $IpAddressGroup, $InstanceName
        )
    }

    return $testTargetResourceReturnValue
}

<#
    .SYNOPSIS
        Compares the properties in the current state with the properties of the
        desired state and returns a hashtable with the comparison result.
 
    .PARAMETER InstanceName
        Specifies the name of the SQL Server instance to enable the protocol for.
 
    .PARAMETER IpAddressGroup
        Specifies the name of the IP address group in the TCP/IP protocol, e.g.
        'IP1', 'IP2' etc., or 'IPAll'.
 
    .PARAMETER ServerName
        Specifies the host name of the SQL Server to be configured. If the
        SQL Server belongs to a cluster or availability group specify the host
        name for the listener or cluster group. Default value is the current
        computer name.
 
    .PARAMETER Enabled
        Specified if the IP address group should be enabled or disabled. Only used if
        the IP address group is not set to 'IPAll'. If not specified, the existing
        value will not be changed.
 
    .PARAMETER IpAddress
        Specifies the IP address for the IP adress group. Only used if the IP address
        group is not set to 'IPAll'. If not specified, the existing value will not be
        changed.
 
    .PARAMETER UseTcpDynamicPort
        Specifies whether the SQL Server instance should use a dynamic port. If
        not specified the existing value will not be changed. This parameter is
        not allowed to be used at the same time as the parameter TcpPort.
 
    .PARAMETER TcpPort
        Specifies the TCP port(s) that SQL Server should be listening on. If the
        IP address should listen on more than one port, list all ports as a string
        value with the port numbers separated with a comma, e.g. '1433,1500,1501'.
        This parameter is limited to 2047 characters. If not specified, the existing
        value will not be changed.This parameter is not allowed to be used at the
        same time as the parameter UseTcpDynamicPort.
 
    .PARAMETER SuppressRestart
        If set to $true then the any attempt by the resource to restart the service
        is suppressed. The default value is $false.
 
    .PARAMETER RestartTimeout
        Timeout value for restarting the SQL Server services. The default value
        is 120 seconds.
#>

function Compare-TargetResourceState
{
    [CmdletBinding()]
    param
    (
        [Parameter(Mandatory = $true)]
        [System.String]
        $InstanceName,

        [Parameter(Mandatory = $true)]
        [System.String]
        $IpAddressGroup,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String]
        $ServerName = (Get-ComputerName),

        [Parameter()]
        [System.Boolean]
        $Enabled,

        [Parameter()]
        [System.String]
        $IpAddress,

        [Parameter()]
        [System.Boolean]
        $UseTcpDynamicPort,

        [Parameter()]
        [System.String]
        $TcpPort,

        [Parameter()]
        [System.Boolean]
        $SuppressRestart = $false,

        [Parameter()]
        [System.UInt16]
        $RestartTimeout = 120
    )

    $assertBoundParameterParameters = @{
        BoundParameterList     = $PSBoundParameters
        MutuallyExclusiveList1 = @(
            'UseTcpDynamicPort'
        )
        MutuallyExclusiveList2 = @(
            'TcpPort'
        )
    }

    Assert-BoundParameter @assertBoundParameterParameters

    if ($PSBoundParameters.ContainsKey('IpAddress'))
    {
        Assert-IpAddress -Address $IpAddress
    }

    $getTargetResourceParameters = @{
        InstanceName    = $InstanceName
        IpAddressGroup  = $IpAddressGroup
        ServerName      = $ServerName
        SuppressRestart = $SuppressRestart
        RestartTimeout  = $RestartTimeout
    }

    <#
        We remove any parameters not passed by $PSBoundParameters so that
        Get-TargetResource can also evaluate $PSBoundParameters correctly.
 
        Need the @() around the Keys property to get a new array to enumerate.
    #>

    @($getTargetResourceParameters.Keys) | ForEach-Object -Process {
        if (-not $PSBoundParameters.ContainsKey($_))
        {
            $getTargetResourceParameters.Remove($_)
        }
    }

    $getTargetResourceResult = Get-TargetResource @getTargetResourceParameters

    # Get individual IP address group properties to evaluate.
    switch ($IpAddressGroup)
    {
        'IPAll'
        {
            $propertiesToEvaluate = @(
                'UseTcpDynamicPort'
                'TcpPort'
            )
        }

        default
        {
            $propertiesToEvaluate = @(
                'Enabled'
                'IpAddress'
                'UseTcpDynamicPort'
                'TcpPort'
            )
        }
    }

    $compareTargetResourceStateParameters = @{
        CurrentValues = $getTargetResourceResult
        DesiredValues = $PSBoundParameters
        Properties    = $propertiesToEvaluate
    }

    return Compare-ResourcePropertyState @compareTargetResourceStateParameters
}

<#
    .SYNOPSIS
        Converts a IP address group name to the correct casing.
 
    .PARAMETER IpAddressGroup
        Specifies the name of the IP address group in the TCP/IP protocol, e.g.
        'IP1', 'IP2' etc., or 'IPAll'.
 
    .NOTES
        SMO is case-sensitive with the address group names.
#>

function Convert-IpAdressGroupCasing
{
    [CmdletBinding()]
    [OutputType([System.String])]
    param
    (
        [Parameter(Mandatory = $true)]
        [System.String]
        $IpAddressGroup
    )

    return ($IpAddressGroup.ToUpper() -replace 'IPALL', 'IPAll')
}