181 lines
5.5 KiB
Plaintext
181 lines
5.5 KiB
Plaintext
' Copyright (c) 1997-2001 Microsoft Corporation
|
|
'***************************************************************************
|
|
' WMI script to change the OS boot default number
|
|
' based on WMI Sample Scripts (VBScript)
|
|
' Serguei Kouzmine [sergueik]
|
|
'
|
|
'***************************************************************************
|
|
|
|
|
|
'Initialize variables
|
|
ReDim objArgs(0)
|
|
'Get the command line arguments
|
|
If 0 = Wscript.arguments.count Then
|
|
' ShowUsage
|
|
Else
|
|
For i = 0 to Wscript.arguments.count-1
|
|
ReDim Preserve objArgs(i)
|
|
objArgs(i) = Wscript.arguments.Item(i)
|
|
Next
|
|
End If
|
|
|
|
ReDim Preserve KeyedArg(2)
|
|
Const lcDebug = 0
|
|
|
|
Dim pbDoBoot:pbDoBoot = False
|
|
Dim psHint:psHint = "SAFE"
|
|
If Not IsEmpty(objArgs) Then
|
|
For i = 0 to Ubound(objArgs)
|
|
ParseArg(objArgs(i))
|
|
If KeyedArg(0) = "l" Then
|
|
psHint = KeyedArg(1)
|
|
ElseIf KeyedArg(0) = "h" Then
|
|
ShowUsage
|
|
ElseIf KeyedArg(0) = "b" Then
|
|
pbDoBoot = True
|
|
End If
|
|
|
|
Next
|
|
End If
|
|
|
|
DoChange
|
|
if pbDoBoot = True Then
|
|
DoBoot
|
|
End If
|
|
|
|
sub DoChange()
|
|
|
|
Set posBootSet = GetObject("winmgmts:{(SystemEnvironment)}//./root/cimv2")._
|
|
InstancesOf ("Win32_ComputerSystem")
|
|
|
|
Set poRegEx = New RegExp
|
|
|
|
Dim pnChange:pnChange = 0
|
|
poRegEx.Global = True:poRegEx.IgnoreCase = True:poRegEx.Pattern = psHint
|
|
|
|
for each poBoot in posBootSet
|
|
|
|
if VBNull = VarType(poBoot.SystemStartupOptions) Then
|
|
WScript.echo "Fatal: invalid data in Win32_ComputerSystem.SystemStartupOptions"
|
|
WScript.echo "Giving up"
|
|
WScript.Quit(1)
|
|
Else
|
|
Dim pdSafePos: pdSafePos = 0
|
|
Dim pdCurPos : pdCurPos = 0
|
|
For Each lsLabel in poBoot.SystemStartupOptions
|
|
Set isMatches = poRegEx.Execute(lsLabel)
|
|
If isMatches.count <> 0 Then
|
|
pnSafePos = pdCurPos
|
|
pnChange = 1
|
|
End If
|
|
pdCurPos = pdCurPos + 1
|
|
Next
|
|
End If
|
|
Next
|
|
|
|
If pnChange = 0 Then
|
|
WScript.echo psHint & _
|
|
" not found " &_
|
|
VbCrLf &_
|
|
"Giving up"
|
|
WScript.Quit
|
|
End If
|
|
If pnSafePos = 0 Then
|
|
WScript.echo "Don't need swith boot OS" &_
|
|
VbCrLf
|
|
Exit Sub
|
|
End If
|
|
|
|
Set posBootSet = Nothing
|
|
Set posBootSet = GetObject("winmgmts:{impersonationLevel=impersonate,(SystemEnvironment)}")._
|
|
ExecQuery("select * from Win32_ComputerSystem")
|
|
for each poBoot in posBootSet
|
|
poBoot.SystemStartupSetting = pnSafePos
|
|
poBoot.Put_()
|
|
next
|
|
WScript.Echo "Boot up default OS changed to " & psHint
|
|
Set posBootSet = Nothing
|
|
|
|
End Sub
|
|
|
|
|
|
Private Sub ParseArg(lsLabel)
|
|
|
|
Set poSwitchRegEx = New RegExp
|
|
Dim psMaskSwitch:psMaskSwitch = "[-/][hdb]$" 'currently don't handle all switches.
|
|
poSwitchRegEx.Global = True:poSwitchRegEx.IgnoreCase = True:poSwitchRegEx.Pattern = psMaskSwitch
|
|
|
|
Set poFlagRegEx = New RegExp
|
|
Dim psMaskFlag:psMaskFlag = "[-/][l]:"
|
|
poFlagRegEx.Global = True:poFlagRegEx.IgnoreCase = True:poFlagRegEx.Pattern = psMaskFlag
|
|
|
|
Set poFullRegEx = New RegExp
|
|
Dim psMaskFull:psMaskFull = psMaskFlag & "\w+"
|
|
poFullRegEx.Global = True:poFullRegEx.IgnoreCase = True:poFullRegEx.Pattern = psMaskFull
|
|
|
|
Dim pnChange:pnChange = 0
|
|
|
|
set issSwitches = poSwitchRegEx.Execute(lsLabel)
|
|
If 0 <> issSwitches.Count Then
|
|
KeyedArg(0) = Mid(issSwitches(0),2,1)
|
|
KeyedArg(1) = "+"
|
|
' ShowUsage
|
|
Else
|
|
Set isMatches = poFullRegEx.Execute(lsLabel)
|
|
if 1 <> isMatches.count Then
|
|
WScript.echo "Bad Argument: " & lsLabel
|
|
WScript.quit(1)
|
|
End If
|
|
Set psFlag = poFlagRegEx.Execute(lsLabel)
|
|
psRes = poFlagRegEx.Replace(lsLabel,"")
|
|
KeyedArg(0) = Mid(psFlag(0),2,1)
|
|
KeyedArg(1) = psRes
|
|
End If
|
|
End Sub
|
|
|
|
Private Sub ShowUsage()
|
|
|
|
Dim strFullEngineName:strFullEngineName=WScript.FullName
|
|
Dim rPos:rPos = Len(strFullEngineName)
|
|
Dim lPos:lPos = InStrRev(strFullEngineName, "\")
|
|
strFullName = Mid(strFullEngineName, lPos + 1 , rPos - lPos)
|
|
|
|
Wscript.echo "" & _
|
|
"Usage:" & _
|
|
VbCrLf & _
|
|
Chr(9) & strFullName & Chr(32) & WScript.ScriptName & " [-l:<LABEL>] [-b] [-h]" & _
|
|
VbCrLf & _
|
|
Chr(9) & "Change the boot order of the BVT machine" & _
|
|
VbCrLf & _
|
|
"Where:" & _
|
|
VbCrLf & _
|
|
Chr(9) & "-l:<LABEL> boot into OS marked as <LABEL> (default is SAFE)" & _
|
|
VbCrLf & _
|
|
Chr(9) & "-b reboot the machine" & _
|
|
VbCrLf & _
|
|
Chr(9) & "-h view this message"
|
|
|
|
|
|
WScript.Quit(0)
|
|
|
|
End Sub
|
|
|
|
private Sub DoBoot()
|
|
|
|
Const lcWMIObject = "winmgmts:{impersonationLevel=impersonate, (Shutdown)}"
|
|
Dim lsWMIQueryString:lsWMIQueryString = "SELECT * FROM " & _
|
|
"Win32_OperatingSystem WHERE PRIMARY = true"
|
|
|
|
Set loWMISet = GetObject(lcWMIObject)._
|
|
ExecQuery(lsWMIQueryString, _
|
|
"WQL")
|
|
|
|
if lcDebug = 1 Then
|
|
WScript.echo TypeName(loWMISet)
|
|
WScript.echo loWMISet.Count
|
|
End If
|
|
|
|
For Each loWMI__ In loWMISet
|
|
loWMI__.Reboot()
|
|
Next
|
|
End Sub |