[WannaCrypt対策] SMB1 を無効化する

#####################################################
# SMB v1 を停止する
# Windows 7 / Windows Server 2008 R2 ～ 
#   Windows 10 RS2 / Windows Server 2016 対応
#
# 管理権限で PowerShell を開き、コピペする
# 再起動要求が来たら Y で再起動する
#####################################################
function DisableSMBv1(){
    $WindowsPrincipal = [Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()
    $AdminRole = $WindowsPrincipal.IsInRole("Administrators")
    if( $AdminRole -ne $true ){
        echo "Administrative authority required."
        return
    }

    $OSData = Get-WmiObject Win32_OperatingSystem
    $BuildNumber = $OSData.BuildNumber
    $strVersion = $OSData.Version
    $strVersion = $strVersion.Replace( ".$BuildNumber", "" )
    $WinVer = [decimal]$strVersion

    if( $WinVer -ge 6.2 ){
        # Windows 8 / WS12 以降
        # 役割の収集
        $Features = dism /Online /Get-Features

        # SMB v1 状態
        $SMBv1EnableFlag = $false

        $HitFlag = $false
        $SMBv1State = ""

        foreach( $Line in $Features ){
            if( $HitFlag -eq $true ){
                # SMBv1 の状態取得
                if( ($Line -match "状態 : (?<SMBv1State>.*?)$") -or ($Line -match "State : (?<SMBv1State>.*?)$") ){
                    $SMBv1State = ($Matches.SMBv1State).Trim()
                    if( ($SMBv1State -eq "有効") -or ($SMBv1State -eq "Enabled" ) ){
                        $SMBv1EnableFlag = $true
                    }
                    $HitFlag = $false
                }
            }

            # SMBv1 の役割発見
            if( $Line -match "SMB1Protocol" ){
                $HitFlag = $true
            }
        }

        # SMBv1 が有効になっていたら無効にする
        if( $SMBv1EnableFlag -eq $true ){
            dism /Online /Disable-Feature:SMB1Protocol
        }
        else {
            echo "SMB1 is disabled."
        }
    }
    else {
        # Windows 7 / WS08R2 以前

        $RegPath = "HKLM:\SYSTEM\CurrentControlSet\Services\LanmanServer\Parameters"
        $RegKey = "SMB1"
        $Result = Get-ItemProperty $RegPath -name $RegKey -ErrorAction SilentlyContinue
        if( $Result -ne $null ){
            # レジストリがあった
            if( $Result.$RegKey -eq 0 ){
                echo "SMB1 is disabled."
                return
            }
        }

        # SMBv1 が有効になっている
        Set-ItemProperty -Path $RegPath -name $RegKey -Type DWORD -Value 0 -Force
        echo "Restart now."
        sleep 3
        Restart-Computer
    }
}
DisableSMBv1