'USB_Stock_Block.vbs
'===========================================================================================
CheckOS     ' 检查操作系统版本
CheckMeState  ' 检查程序运行状态
main      ' 执行主程序

'===========================================================================================
'主函数
Sub main()
    Dim wso, windir, EnableUSB
    Set wso = CreateObject("WScript.Shell")
    Set objNetwork = CreateObject("wscript.network")
        strComputer = objNetwork.ComputerName
        
    If wso.Popup(VbCrLf & "禁用 USB 存储设备，请按“确定”"& VbCrLf & _
                VbCrLf & "启用 USB 存储设备，请按“取消”   (6秒后自动取消)" _
                , 6, "USB 存储设备控制 - 主菜单", 48+4096+1) = 1 Then
        EnableUSB = 0
    Else
        EnableUSB = 1
    End If
    
    If Exist( "C:\windows\system32\cmd.exe" ) Then windir = "windows"
    If Exist( "C:\winnt\system32\cmd.exe" ) Then windir = "winnt"

    If EnableUSB = 1 Then
        wso.RegWrite "HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Control\StorageDevicePolicies\WriteProtect","1","REG_DWORD" '禁止写入
        wso.RegWrite "HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\USBSTOR\Start","3","REG_DWORD"           '启用USBStor
        
        Move "C:\" & windir & "\inf\usbstor.pnf_" , "C:\" & windir & "\inf\usbstor.pnf"
        Move "C:\" & windir & "\inf\usbstor.inf_" , "C:\" & windir & "\inf\usbstor.inf"
        Move "C:\" & windir & "\system32\drivers\usbstor.sys_" , "C:\" & windir & "\system32\drivers\usbstor.sys"
        
        If (Not Exist( "C:\" & windir & "\inf\usbstor.pnf_" )) And (regKeyRead( "HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\USBSTOR\Start" ) = 3 ) Then
            wso.Popup VbCrLf & "启用 USB 存储设备成功。　　　　", 5, "USB 存储设备控制 - 操作完成", 64+4096
        Else
            wso.Popup VbCrLf & "启用 USB 存储设备失败。　　　　", 5, "USB 存储设备控制 - 操作完成", 16+4096
        End If
    Else
        wso.RegWrite "HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Control\StorageDevicePolicies\WriteProtect","1","REG_DWORD" '禁止写入
        wso.RegWrite "HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\USBSTOR\Start","4","REG_DWORD"           '禁用用USBStor
        
        Move "C:\" & windir & "\inf\usbstor.pnf" , "C:\" & windir & "\inf\usbstor.pnf_"
        Move "C:\" & windir & "\inf\usbstor.inf" , "C:\" & windir & "\inf\usbstor.inf_"
        Move "C:\" & windir & "\system32\drivers\usbstor.sys" , "C:\" & windir & "\system32\drivers\usbstor.sys_"
        
        If (Not Exist( "C:\" & windir & "\inf\usbstor.pnf" )) And (regKeyRead( "HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\USBSTOR\Start" ) = 4 ) Then
            wso.Popup VbCrLf & "禁用 USB 存储设备成功。　　　　", 5, "USB 存储设备控制 - 操作完成", 64+4096
        Else
            wso.Popup VbCrLf & "禁用 USB 存储设备失败。　　　　", 5, "USB 存储设备控制 - 操作完成", 16+4096
        End If
    End if
        
    Set wso = Nothing
End Sub

'===========================================================================================
'小函数
Function Exist( strPath )
    'On Error Resume Next
    Set fso = CreateObject("Scripting.FileSystemObject")
    If ((fso.FolderExists( strPath )) Or (fso.FileExists( strPath ))) then
        Exist = True
    Else
        Exist = False
    End if
    Set fso = Nothing
End Function
Sub Move( strSource, strDestination )
    On Error Resume Next
    If Exist( strSource ) Then
        Set fso = CreateObject("Scripting.FileSystemObject")
        If (fso.FileExists(strSource)) Then fso.MoveFile strSource, strDestination
        If (fso.FolderExists(strSource)) Then fso.MoveFolder strSource, strDestination
        Set fso = Nothing
    Else
        WarningInfo "警告", "找不到 " & strSource & " 文件！", 2
    End If
    If Not Exist( strDestination ) Then WarningInfo "警告", "移动失败，无法移动 " & VbCrLf & strSource & " 至" & VbCrLf & strDestination, 2
End Sub
Function regKeyRead( strKey )
    Set wso = CreateObject("WScript.Shell")
    regKeyRead = wso.RegRead( strKey )  'strKey = "HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\Run\DocTip"
    Set wso = Nothing
End Function

'===========================================================================================
'是否重复运行
Sub CheckMeState()
    If IsRun( WScript.ScriptFullName ) Then
        Set wso = CreateObject("WScript.Shell")
        If wso.Popup("程序已运行，请不要重复运行本程序！" & VbCrLf & VbCrLf & _
                            "退出已运行程序，请按“确定”，否则请按“取消”。(3秒后自动取消)" _
                            , 3, "警告", 1) = 1 Then
            KillMeAllRun
        End If
        Set wso = Nothing
        'WarningInfo "警告：", "程序已运行，请不要重复运行本程序！！", 1
        WScript.Quit
    End If
End Sub
' 检测是否重复运行
Function IsRun(appPath)
    IsRun=False
    For Each ps in GetObject("winmgmts:\\.\root\cimv2:win32_process").instances_
        'IF Lcase(ps.name)="mshta.exe" Then
        IF Lcase(ps.name)="wscript.exe" Then
            IF instr(Lcase(ps.CommandLine),Lcase(appPath)) Then i=i+1
        End IF
    next
    if i>1 then
        IsRun=True
    end if
End Function
'终止自身
Function KillMeAllRun()
    Dim MeAllPid
    Set pid = Getobject("winmgmts:\\.").InstancesOf("Win32_Process")
    For Each ps In pid
        'if LCase(ps.name) = LCase("mshta.exe") then
        IF Lcase(ps.name)="wscript.exe" Or Lcase(ps.name)="cscript.exe"Then
            IF instr(Lcase(ps.CommandLine),Lcase(WScript.ScriptFullName)) Then MeAllPid = MeAllPid & "/PID " & ps.ProcessID & " "
        end if
    next
    Set wso = CreateObject("WScript.Shell")
    wso.Run "TASKKILL " & MeAllPid & " /F /T", 0, False
    Set wso = Nothing
    Set pid = Nothing
End Function

'===========================================================================================
'检查操作系统版本
Sub CheckOS()
    Dim os_ver
    os_ver = GetSystemVersion
    If os_ver >= 60 Or os_ver <= 50 Then
        Msgbox "不支持该操作系统！　　　　", 48+4096, "警告"
        WScript.Quit  ' 退出程序
    End If
End Sub
'取得操作系统版本
Function GetSystemVersion()
    Dim os_obj, os_version, os_version_arr
    Set os_obj = GetObject("winmgmts:").InstancesOf("Win32_OperatingSystem")
    For Each os_info In os_obj
        os_version = os_info.Version
        If os_version <> "" Then Exit For
    Next
    Set os_obj = Nothing
    os_version_arr = Split( os_info.Version, ".")
    GetSystemVersion = Cint( os_version_arr( 0 ) & os_version_arr( 1 ) )
End Function