'USB_Stock_Block.vbs
'===========================================================================================
CheckOS ' 檢查操作系統(tǒng)版本
CheckMeState ' 檢查程序運行狀態(tài)
main ' 執(zhí)行主程序
'===========================================================================================
'主函數(shù)
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
'===========================================================================================
'小函數(shù)
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
'===========================================================================================
'檢查操作系統(tǒng)版本
Sub CheckOS()
Dim os_ver
os_ver = GetSystemVersion
If os_ver >= 60 Or os_ver = 50 Then
Msgbox "不支持該操作系統(tǒng)！ ", 48+4096, "警告"
WScript.Quit ' 退出程序
End If
End Sub
'取得操作系統(tǒng)版本
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
