Refactor SqlServerNetwork to not load assembly from GAC (issue dsccom…

…munity#1151)
johlju · Jul 4, 2018 · 3745098 · 3745098
1 parent 9b6f768
commit 3745098
Show file tree

Hide file tree

Showing 8 changed files with 481 additions and 413 deletions.
diff --git a/DSCResources/MSFT_SqlServerNetwork/MSFT_SqlServerNetwork.psm1 b/DSCResources/MSFT_SqlServerNetwork/MSFT_SqlServerNetwork.psm1
@@ -22,6 +22,11 @@ function Get-TargetResource
     [OutputType([System.Collections.Hashtable])]
     param
     (
+        [Parameter()]
+        [ValidateNotNullOrEmpty()]
+        [System.String]
+        $ServerName = $env:COMPUTERNAME,
+
         [Parameter(Mandatory = $true)]
         [System.String]
         $InstanceName,
@@ -33,31 +38,24 @@ function Get-TargetResource
         $ProtocolName
     )
 
-    try
-    {
-        $applicationDomainObject = Register-SqlWmiManagement -SQLInstanceName $InstanceName
-
-        $managedComputerObject = New-Object -TypeName Microsoft.SqlServer.Management.Smo.Wmi.ManagedComputer
+    Import-SQLPSModule
 
-        Write-Verbose -Message ($script:localizedData.GetNetworkProtocol -f $ProtocolName, $InstanceName)
-        $tcp = $managedComputerObject.ServerInstances[$InstanceName].ServerProtocols[$ProtocolName]
+    $managedComputerObject = New-Object -TypeName Microsoft.SqlServer.Management.Smo.Wmi.ManagedComputer
 
-        Write-Verbose -Message $script:localizedData.ReadingNetworkProperties
-        $returnValue = @{
-            InstanceName = $InstanceName
-            ProtocolName    = $ProtocolName
-            IsEnabled       = $tcp.IsEnabled
-            TcpDynamicPort  = ($tcp.IPAddresses['IPAll'].IPAddressProperties['TcpDynamicPorts'].Value -ge 0)
-            TcpPort         = $tcp.IPAddresses['IPAll'].IPAddressProperties['TcpPort'].Value
-        }
+    Write-Verbose -Message ($script:localizedData.GetNetworkProtocol -f $ProtocolName, $InstanceName)
+    $tcp = $managedComputerObject.ServerInstances[$InstanceName].ServerProtocols[$ProtocolName]
 
-        $returnValue.Keys | ForEach-Object {
-            Write-Verbose -Message "$_ = $($returnValue[$_])"
-        }
+    Write-Verbose -Message $script:localizedData.ReadingNetworkProperties
+    $returnValue = @{
+        InstanceName = $InstanceName
+        ProtocolName    = $ProtocolName
+        IsEnabled       = $tcp.IsEnabled
+        TcpDynamicPort  = ($tcp.IPAddresses['IPAll'].IPAddressProperties['TcpDynamicPorts'].Value -ge 0)
+        TcpPort         = $tcp.IPAddresses['IPAll'].IPAddressProperties['TcpPort'].Value
     }
-    finally
-    {
-        Unregister-SqlAssemblies -ApplicationDomain $applicationDomainObject
+
+    $returnValue.Keys | ForEach-Object {
+        Write-Verbose -Message "$_ = $($returnValue[$_])"
     }
 
     return $returnValue
@@ -145,90 +143,95 @@ function Set-TargetResource
 
     $getTargetResourceResult = Get-TargetResource -InstanceName $InstanceName -ProtocolName $ProtocolName
 
-    try
-    {
-        $applicationDomainObject = Register-SqlWmiManagement -SQLInstanceName $InstanceName
-
-        $desiredState = @{
-            InstanceName = $InstanceName
-            ProtocolName    = $ProtocolName
-            IsEnabled       = $IsEnabled
-            TcpDynamicPort  = $TcpDynamicPort
-            TcpPort         = $TcpPort
-        }
+    $desiredState = @{
+        InstanceName = $InstanceName
+        ProtocolName    = $ProtocolName
+        IsEnabled       = $IsEnabled
+        TcpDynamicPort  = $TcpDynamicPort
+        TcpPort         = $TcpPort
+    }
 
-        $isRestartNeeded = $false
+    $isRestartNeeded = $false
 
-        $managedComputerObject = New-Object -TypeName Microsoft.SqlServer.Management.Smo.Wmi.ManagedComputer
+    # Get-TargetResource makes the necessary calls so the type ManagedComputer is available.
+    $managedComputerObject = New-Object -TypeName Microsoft.SqlServer.Management.Smo.Wmi.ManagedComputer
 
-        Write-Verbose -Message ($script:localizedData.GetNetworkProtocol -f $ProtocolName, $InstanceName)
-        $tcp = $managedComputerObject.ServerInstances[$InstanceName].ServerProtocols[$ProtocolName]
+    Write-Verbose -Message ($script:localizedData.GetNetworkProtocol -f $ProtocolName, $InstanceName)
+    $tcp = $managedComputerObject.ServerInstances[$InstanceName].ServerProtocols[$ProtocolName]
 
-        Write-Verbose -Message ($script:localizedData.CheckingProperty -f 'IsEnabled')
-        if ($desiredState.IsEnabled -ine $getTargetResourceResult.IsEnabled)
-        {
-            Write-Verbose -Message ($script:localizedData.UpdatingProperty -f 'IsEnabled', $getTargetResourceResult.IsEnabled, $desiredState.IsEnabled)
-            $tcp.IsEnabled = $desiredState.IsEnabled
-            $tcp.Alter()
+    Write-Verbose -Message ($script:localizedData.CheckingProperty -f 'IsEnabled')
+    if ($desiredState.IsEnabled -ine $getTargetResourceResult.IsEnabled)
+    {
+        Write-Verbose -Message ($script:localizedData.UpdatingProperty -f 'IsEnabled', $getTargetResourceResult.IsEnabled, $desiredState.IsEnabled)
+        $tcp.IsEnabled = $desiredState.IsEnabled
+        $tcp.Alter()
+
+        $isRestartNeeded = $true
+    }
 
-            $isRestartNeeded = $true
+    Write-Verbose -Message ($script:localizedData.CheckingProperty -f 'TcpDynamicPort')
+    if ($desiredState.TcpDynamicPort -ne $getTargetResourceResult.TcpDynamicPort)
+    {
+        # Translates the current and desired state to a string for display
+        $dynamicPortDisplayValueTable = @{
+            $true = 'enabled'
+            $false = 'disabled'
         }
 
-        Write-Verbose -Message ($script:localizedData.CheckingProperty -f 'TcpDynamicPort')
-        if ($desiredState.TcpDynamicPort -ne $getTargetResourceResult.TcpDynamicPort)
-        {
-            # Translates the current and desired state to a string for display
-            $dynamicPortDisplayValueTable = @{
-                $true = 'enabled'
-                $false = 'disabled'
-            }
+        # Translates the desired state to a valid value
+        $desiredDynamicPortValue = @{
+            $true = '0'
+            $false = ''
+        }
 
-            # Translates the desired state to a valid value
-            $desiredDynamicPortValue = @{
-                $true = '0'
-                $false = ''
-            }
+        $fromTcpDynamicPortDisplayValue = $dynamicPortDisplayValueTable[$getTargetResourceResult.TcpDynamicPort]
+        $toTcpDynamicPortDisplayValue = $dynamicPortDisplayValueTable[$desiredState.TcpDynamicPort]
 
-            $fromTcpDynamicPortDisplayValue = $dynamicPortDisplayValueTable[$getTargetResourceResult.TcpDynamicPort]
-            $toTcpDynamicPortDisplayValue = $dynamicPortDisplayValueTable[$desiredState.TcpDynamicPort]
+        Write-Verbose -Message ($script:localizedData.UpdatingProperty -f 'TcpDynamicPorts', $fromTcpDynamicPortDisplayValue, $toTcpDynamicPortDisplayValue)
+        $tcp.IPAddresses['IPAll'].IPAddressProperties['TcpDynamicPorts'].Value = $desiredDynamicPortValue[$desiredState.TcpDynamicPort]
+        $tcp.Alter()
 
-            Write-Verbose -Message ($script:localizedData.UpdatingProperty -f 'TcpDynamicPorts', $fromTcpDynamicPortDisplayValue, $toTcpDynamicPortDisplayValue)
-            $tcp.IPAddresses['IPAll'].IPAddressProperties['TcpDynamicPorts'].Value = $desiredDynamicPortValue[$desiredState.TcpDynamicPort]
-            $tcp.Alter()
+        $isRestartNeeded = $true
+    }
 
-            $isRestartNeeded = $true
+    Write-Verbose -Message ($script:localizedData.CheckingProperty -f 'TcpPort')
+    if ($desiredState.TcpPort -ine $getTargetResourceResult.TcpPort)
+    {
+        $fromTcpPort = $getTargetResourceResult.TcpPort
+        if ($fromTcpPort -eq '')
+        {
+            $fromTcpPort = 'none'
         }
 
-        Write-Verbose -Message ($script:localizedData.CheckingProperty -f 'TcpPort')
-        if ($desiredState.TcpPort -ine $getTargetResourceResult.TcpPort)
+        $toTcpPort = $desiredState.TcpPort
+        if ($toTcpPort -eq '')
         {
-            $fromTcpPort = $getTargetResourceResult.TcpPort
-            if ($fromTcpPort -eq '')
-            {
-                $fromTcpPort = 'none'
-            }
+            $toTcpPort = 'none'
+        }
 
-            $toTcpPort = $desiredState.TcpPort
-            if ($toTcpPort -eq '')
-            {
-                $toTcpPort = 'none'
-            }
+        Write-Verbose -Message ($script:localizedData.UpdatingProperty -f 'TcpPort', $fromTcpPort, $toTcpPort)
+        $tcp.IPAddresses['IPAll'].IPAddressProperties['TcpPort'].Value = $desiredState.TcpPort
+        $tcp.Alter()
 
-            Write-Verbose -Message ($script:localizedData.UpdatingProperty -f 'TcpPort', $fromTcpPort, $toTcpPort)
-            $tcp.IPAddresses['IPAll'].IPAddressProperties['TcpPort'].Value = $desiredState.TcpPort
-            $tcp.Alter()
+        $isRestartNeeded = $true
+    }
 
-            $isRestartNeeded = $true
+    if ($RestartService -and $isRestartNeeded)
+    {
+        $restartSqlServiceParameters = @{
+            SQLServer = $ServerName
+            SQLInstanceName = $InstanceName
+            Timeout = $RestartTimeout
+            SkipClusterCheck = $true
         }
 
-        if ($RestartService -and $isRestartNeeded)
+        if ($PSBoundParameters.ContainsKey('IsEnabled') -and $IsEnabled -eq $false)
         {
-            Restart-SqlService -SQLServer $ServerName -SQLInstanceName $InstanceName -Timeout $RestartTimeout
+            # If the protocol is disabled it is not possible to connect to the instance.
+            $restartSqlServiceParameters['SkipWaitForOnline'] = $true
         }
-    }
-    finally
-    {
-        Unregister-SqlAssemblies -ApplicationDomain $applicationDomainObject
+
+        Restart-SqlService @restartSqlServiceParameters
     }
 }