ExchangeServerMaintenanceMode.psm1

# Special thanks to Michael 'Van Hybrid' for the idea and implementation

#region Classes
add-type @'
namespace CPolydorou.Exchange
{
    public class ExchangeServerMaintenanceMode
    {
        public string Name;
        public bool MaintenanceModeEnabled;
        public bool DatabaseCopyActivationDisabledAndMoveNow;
        public string DatabaseCopyAutoActivationPolicy;
        public string HubTransportState;
        public int ActiveMailboxDatabaseCopies;
    }
}
'@

#endregion

#region Functions

#region StopExchangeServerMaintenanceMode
Function Stop-ExchangeServerMaintenanceMode
{

    <#  .Synopsis    Take an Exchange 2013 server out of maintenance mode. .DESCRIPTION      This function will take all the necessary steps to take an Exchange Server 2013 out of maintenance mode.    .EXAMPLE       Stop-ExchangeServerMaintenanceMode -Server Server1  #> 
 
    [CmdletBinding(
        SupportsShouldProcess = $true,
        ConfirmImpact = "High"    
    )] 

    [OutputType([int])] 
    Param 
    ( 
     # determine what server to put in maintenance mode 
     [Parameter(Mandatory=$true, 
                ValueFromPipelineByPropertyName=$true, 
                Position=0)] 
        [Alias('Name')]
     [string[]]$Server = $env:COMPUTERNAME 
    ) 
 
    Begin{}
 
    Process
    {
        foreach($s in $Server)
        {
            if($PSCmdlet.ShouldProcess($s))
            {
                $errorCount = 0
            
                # Get the exchange server object
                Write-Verbose "Searching for exchange server $s"
                New-Variable discoveredServer -Force
                try
                {
                    $discoveredServer = Get-ExchangeServer -Identity $s -ErrorAction Stop | Select IsHubTransportServer,IsFrontendTransportServer,AdminDisplayVersion
                    Write-Verbose "Exchange server $s found."
  }
                catch
                {
                    Write-Error "Could not find exchange server $s"
                    return
                }

                #Check for Administrative credentials 
                If (-NOT ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator")){ 
     Write-Error "You do not have Administrator rights to run this script!`nPlease re-run this script as an Administrator!" 
     continue
                }

                # Check the exchange server version 
                if($discoveredServer.AdminDisplayVersion.Major -ne "15")
                { 
         Write-Error "The specified Exchange Server is not an Exchange 2013 server!" 
         Write-Error "Aborting script..." 
         continue
                } 
 
                # Enable all server components 
                Write-Verbose "Reactivating all server components..."
                try
                {
     Set-ServerComponentState $s -Component ServerWideOffline -State Active -Requester Maintenance -ErrorAction Stop
                    Write-Verbose "Server component states changed back into active state using requester 'Maintenance'"
                }
                catch
                {
                    Write-Error "Could not set the components."
                    $errorCount++
                }

                # if the server is a mailbox server
                if($discoveredServer.IsHubTransportServer -eq $true)
                {      
     $mailboxserver = Get-MailboxServer -Identity $s | Select DatabaseAvailabilityGroup 
     
     if($mailboxserver.DatabaseAvailabilityGroup -ne $null)
                    { 
         Write-Verbose "Server $s is a member of a Database Availability Group."
                        Write-Verbose "Resuming cluster node."
         try
                        {
                            $output = Invoke-Command -ComputerName $s -ArgumentList $s {Resume-ClusterNode $args[0]} 
                            Write-Verbose ("`tName: " + $output.Name)
                            Write-Verbose ("`tCluster: " + $output.Cluster)
                            Write-Verbose ("`tState: " + $output.State)
                        }
                        catch
                        {
                            Write-Warning "Failed to resume the cluster node."
                            $errorCount++
                        }

                        try
                        {
         Set-MailboxServer $s -DatabaseCopyActivationDisabledAndMoveNow $false -ErrorAction Stop
                            Write-Verbose "The DatabaseCopyActivationDisabledAndMoveNow setting is set to False"
                        }
                        catch
                        {
                            Write-Error "Could not set the DatabaseCopyActivationDisabledAndMoveNow setting"
                            $errorCount++
                        }

                        try
                        {
         Set-MailboxServer $s -DatabaseCopyAutoActivationPolicy Unrestricted -ErrorAction Stop
                            Write-Verbose "The DatabaseCopyAutoActivationPolicy has been set to Unrestricted."
                        }
                        catch
                        {
                            Write-Error "Could not set the DatabaseCopyAutoActivationPolicy."
                            $errorCount++
                        }
     } 
     
     Write-Verbose "Resuming Transport Service..."
                    try
                    {
     Set-ServerComponentState â€“Identity $s -Component HubTransport -State Active -Requester Maintenance -ErrorAction Stop
                    }
                    catch
                    {
                        Write-Warning "Could not activate the HubTransport component."
                        $errorCount++
                    }

     Write-Verbose "Restarting the MSExchangeTransport Service on server $s..."
                    try
                    {
     Invoke-Command -ComputerName $s {Restart-Service MSExchangeTransport} | Out-Null
                        Write-Verbose "The MSExchangeTransport service has been restarted on server $s."
                    }
                    catch
                    {
                        Write-Warning "Could not restart the MSExchangeTransport service on server $s."
                        Write-Warning "Please restart the service manually."
                        $errorCount++
                    }
                } 
 
                #restart FE Transport Services if server is also CAS 
                if($discoveredServer.IsFrontendTransportServer -eq $true)
                { 
     Write-Verbose "Restarting the MSExchangeFrontEndTransport Service on server $s..."
     try
                    {
                        Invoke-Command -ComputerName $s {Restart-Service MSExchangeFrontEndTransport} | Out-Null
     Write-Verbose "The MSExchangeFrontEndTransport Service has been restarted on server $s"
                    }
                    catch
                    {
                        Write-Error "Could not restart the MSExchangeFrontEndTransport service on server $s."
                        Write-Error "Please restart the service manually."
                        $errorCount++
                    }
                } 

                # check for components that are still inactive
                $components = Get-ServerComponentstate -Identity $s | Where-Object {$_.State -ne "Active"}

                foreach($component in $components)
                {
                    $component.LocalStates |
                        Select-Object -Property @{Name = "Identity"; Expression={$component.Identity}}, `
                                                @{Name = "Component"; Expression={$component.Component}}, `
                                                @{Name = "State"; Expression={$_.State}}, `
                                                @{Name="Requester"; Expression={$_.Requester}}
                }

                if( $errorCount -gt 0)
                {
                    Write-Warning "Process completed. Server $s is taken out of Maintenance Mode with warnings."
                }
                else
                {
                    Write-Verbose "Process completed. Server $s is taken out of Maintenance Mode."
  }
            }
        }
    }

    End{}

}
#endregion

#region StartExchangeServerMaintenanceMode
function Start-ExchangeServerMaintenanceMode
{
    <#  .Synopsis      Put an Exchange 2013 Server into Maintenance Mode.  .DESCRIPTION      This function will take all the necessary steps to put an Exchange Server 2013 into maintenance mode.    .EXAMPLE      Start-ExchangeServerMaintenanceMode -Server Server1 -TargetServerFQDN Server2.domain.com  #> 


    [CmdletBinding(
        SupportsShouldProcess = $true,
        ConfirmImpact = "high"
    )] 

    [OutputType([int])] 
    Param 
    ( 
     # determine what server to put in maintenance mode 
     [Parameter(Mandatory=$true, 
                ValueFromPipelineByPropertyName=$true, 
                Position=0)] 
        [Alias('Name')]
     [string[]]$Server=$env:COMPUTERNAME, 
 
     [Parameter(Mandatory=$false, 
                ValueFromPipelineByPropertyName=$true, 
                Position=1)] 
     [string]$TargetServer
    ) 
 
    Begin{}
    
    Process
    {
        function evaluatequeues()
        {
            Param
            (
                [string]$Server
            )

            do
            {  
     $MessageCount = Get-Queue -Server $Server | ?{$_.Identity -notlike "*\Poison" -and $_.Identity -notlike"*\Shadow\*"} | Measure-Object -Property MessageCount -Sum
     if($MessageCount.Sum -ne 0)
                { 
         Write-Warning "The transport queues are not empty. Sleeping for 60 seconds before checking again..."
         Start-Sleep -s 30 
     } 
     else
                { 
         Write-Verbose "The transport queues are empty."
         Write-Verbose "Putting the entire server into maintenance mode..."
         if(Set-ServerComponentState $Server -Component ServerWideOffline -State Inactive -Requester Maintenance)
                    { 
             Write-Verbose "The components of $Server have successfully been placed into an inactive state!" 
         } 

         Write-Verbose "Restarting MSExchangeTransport service on server $Server..."
             #Restarting transport services based on info from http://blogs.technet.com/b/exchange/archive/2013/09/26/server-component-states-in-exchange-2013.aspx 
             #Restarting the services will cause the transport services to immediately pick up the changed state rather than having to wait for a MA responder to take action 
                    try
                    {
             Invoke-Command -ComputerName $Server {Restart-Service MSExchangeTransport | Out-Null} 
     }
                    catch
                    {
                        Write-Warning "Cound not restart the MSExchangeTransport service on server $server."
                    }

         #restart FE Transport Services if server is also CAS 
         if($discoveredServer.IsFrontendTransportServer -eq $true)
                    { 
             Write-Verbose "Restarting the MSExchangeFrontEndTransport Service on server $Server..."
                        try
                        {
                            Invoke-Command -ComputerName $Server {Restart-Service MSExchangeFrontEndTransport} | Out-Null 
                        }
                        catch
                        {
                            Write-Warning "Could not restart the MSExchangeFrontEndTransport service on server $server."
                        }
         } 
         Write-Verbose "Done! Server $Server is put succesfully into maintenance mode!"
                    break
     }
            }
            While($MessageCount.Sum -gt 0)
 
        } 

        foreach($s in $Server)
        {
            if($PSCmdlet.ShouldProcess($s))
            {
                $errorCount = 0

                New-Variable discoveredServer -Force
                Write-Verbose "Searching exchange server $s..."
                try
                {
                    $discoveredServer = Get-ExchangeServer -Identity $s | Select IsHubTransportServer,IsFrontendTransportServer,AdminDisplayVersion
                    Write-Verbose "Exchange server found."
  }
                catch
                {
                    Write-Error "Could not find exchange server $s."
                    continue
                }

                #Check for Administrative credentials 
                If (-NOT ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator")){ 
     Write-Error "You do not have Administrator rights to run this script!`nPlease re-run this script as an Administrator!" 
     Continue
                } 
 
 
                #check if the server is an Exchange 2013 server 
                if($discoveredServer.AdminDisplayVersion.Major -ne "15"){ 
     Write-Error "The specified Exchange Server is not an Exchange 2013 server!" 
     Write-Error "Aborting script..." 
     continue
                } 
                else{ 
 
     if($discoveredServer.IsHubTransportServer -eq $True){ 
         if(-NOT ($TargetServer)){ 
             Write-Warning "TargetServer is required." 
             $TargetServer = Read-Host -Prompt "Please enter the TargetServer: " 
         } 
         
         #Get the FQDN of the Target Server through DNS, even if the input is just a host name 
         try
                        { 
             $TargetServerName = ([System.Net.Dns]::GetHostByName($TargetServer)).Hostname 
         } 
         catch
                        { 
             Write-Error "Could not resolve ServerFQDN: $TargetServer"
                            continue
         } 
 
         if((Get-ExchangeServer -Identity $TargetServerName | Select IsHubTransportServer).IsHubTransportServer -ne $True)
                        { 
             Write-Error "The target server is not a valid Mailbox server." 
             Write-Error "Aborting script..." 
             Continue
         } 
 
         #Redirecting messages to target system 
         Write-Verbose "Suspending Transport Service. Draining remaining messages..."
         try
                        {
                            Write-Verbose "Disabling server components using requestor 'Maintenance'..."
                            Set-ServerComponentState $s -Component HubTransport -State Draining -Requester Maintenance
                            Write-Verbose "Server components disabled."
                        }
                        catch
                        {
                            Write-Warning "Could not disable the server component states."
                            $errorCount++
                        }

                        try
                        {
                            Write-Verbose "Redirecting messages..."
         Redirect-Message -Server $s -Target $TargetServerName -Confirm:$false 
                            Write-Verbose "Messages redirected."
  }
                        catch
                        {
                            Write-Warning "Could not redirect messages."
                            $errorCount++
                        }

         #suspending cluster node (if the server is part of a DAG) 
         $mailboxserver = Get-MailboxServer -Identity $s | Select DatabaseAvailabilityGroup 
         if($mailboxserver.DatabaseAvailabilityGroup -ne $null){ 
             Write-Verbose "Server $s is a member of a Database Availability Group."
                            Write-Verbose "Suspending the node now."
             try
                            {
                                Write-Verbose "Suspending cluster node..."
                                $output = Invoke-Command -ComputerName $s -ArgumentList $s {Suspend-ClusterNode $args[0]}
                                Write-Verbose ("`tName: " + $output.Name)
                                Write-Verbose ("`tCluster: " + $output.Cluster)
                                Write-Verbose ("`tState: " + $output.State)
                                Write-Verbose "Node suspended."
                            }
                            catch
                            {
                                Write-Verbose "Could not suspend the node."
                                $errorCount++
                            }

                            try
                            {
                                Write-Verbose "Setting DatabaseCopyActivationDisabledAndMoveNow..."
             Set-MailboxServer $s -DatabaseCopyActivationDisabledAndMoveNow $true
                                Write-Verbose "DatabaseCopyActivationDisabledAndMoveNow is now set to True."
                            }
                            catch
                            {
                                Write-Warning "Could not set the DatabaseCopyActivationDisabledAndMoveNow."
                                $errorCount++
                            }

                            try
                            {
                                Write-Verbose "Setting DatabaseCopyAutoActivationPolicy..."
             Set-MailboxServer $s -DatabaseCopyAutoActivationPolicy Blocked
                                Write-Verbose "DatabaseCopyAutoActivationPolicy is now set to Blocked."
                            }
                            catch
                            {
                                Write-Warning "Could not set the DatabaseCopyAutoActivationPolicy."
                                $errorCount++
                            }

         } 
 
         #Evaluate the Transport Queues and put into maintenance mode once all queues are empty 
         evaluatequeues -Server $s
 
     } 
     else{ 
         Write-Vebose "Server $s is a Client Access Server-only server."
         Write-Verbose "Putting the server components into inactive state"
                        try
                        {
         Set-ServerComponentState $s -Component ServerWideOffline -State Inactive -Requester Maintenance
                        }
                        catch
                        {
                            Write-Warning "Could not set the server components."
                            $errorCount++
                        }

         Write-Verbose "Restarting transport services..."
         try
                        {
                            Invoke-Command -ComputerName $s {Restart-Service MSExchangeFrontEndTransport | Out-Null} 
             Write-Verbose "Successfully restarted MSExchangeFrontEndTransport service"
         }
                        catch
                        {
                            Write-Warning "Failed to restart MSExchangeFrontEndTransport service"
                        }

                        if($errorCount -gt 0)
                        {
         Write-Verbose "Process completed with errors. Server $s is now maintenance mode!"
                        }
                        else
                        {
                            Write-Warning "Process completed. Server $s in now in maintenance mode."
                        }
     } 
                }
            }
        }
    }

    End{}
}
#endregion

#region Get-ExchangeServerMaintenanceMode
Function Get-ExchangeServerMaintenanceMode
{
    <#
    .SYNOPSIS
 
    Check if an exchange server is in maintenance mode.
    .DESCRIPTION
 
    The Get-ExchangeServerMaintenanceMode cmdlet checks if an Exchange 2013 server is in maintenance mode.
    .PARAMETER Server
 
    The name of the server to check
    .EXAMPLE
 
    Get-ExchangeServerMaintenanceMode -Server Exchange1
 
    This command will check if the exchange server Exchange1 is in maintenance mode
    .EXAMPLE
 
    Get-MailboxServer | Get-ExchangeServerMaintenanceMode
    #>


    [CmdletBinding()]
    Param
    (
        [parameter(Mandatory=$false, Position=0, ValueFromPipeline=$true)]
        [Alias("Name","ComputerName")]
        [string[]]$Server = $env:COMPUTERNAME
    )

    Begin{}

    Process
    {
        foreach($s in $Server)
        {
            # TODO: should check the version of the exchange server here

            $mailboxserver = Get-MailboxServer $s
            $componentstate = Get-ServerComponentState $s -Component HubTransport
            $mailboxdatabases = Get-MailboxDatabaseCopyStatus -Server $s |
                                    Where-Object {$_.status -eq "Mounted"}

            $obj = New-Object -TypeName CPolydorou.Exchange.ExchangeServerMaintenanceMode
            $obj.Name = $s

            if( $mailboxserver.DatabaseCopyActivationDisabledAndMoveNow -eq $true -and
                $mailboxserver.DatabaseCopyAutoActivationPolicy -eq "Blocked" -and
                $mailboxdatabases.Count -eq 0 -and
                $componentstate.State -eq "Inactive" )
            {
                $obj.MaintenanceModeEnabled = $true
            }
            else
            {
                $obj.MaintenanceModeEnabled = $false
            }

            $obj.DatabaseCopyActivationDisabledAndMoveNow = $mailboxserver.DatabaseCopyActivationDisabledAndMoveNow
            $obj.DatabaseCopyAutoActivationPolicy = $mailboxserver.DatabaseCopyAutoActivationPolicy
            $obj.HubTransportState = $componentstate.State
            $obj.ActiveMailboxDatabaseCopies = $mailboxdatabases.Count

            $obj
        }
    }

    End{}
}
#endregion

#endregion

#region Exports
Export-ModuleMember -Function Start-ExchangeServerMaintenanceMode
Export-ModuleMember -Function Stop-ExchangeServerMaintenanceMode
Export-ModuleMember -Function Get-ExchangeServerMaintenanceMode
#endregion