2008年8月17日星期日

Reboots a Windows 2000 PC for VB

 Reboots a Windows 2000 PC. Many examples shell to the kernel and just kill the PC. This does it properly and takes into account a user privilages.

'API Calls used for RebootPC

Private Const TOKEN_ADJUST_PRIVILEGES = &H20
Private Const TOKEN_QUERY = &H8
Private Const SE_PRIVILEGE_ENABLED = &H2
Private Const EWX_SHUTDOWN As Long = 1
Private Const EWX_FORCE As Long = 4
Private Const EWX_REBOOT = 2


Private Type LUID
 UsedPart As Long
 IgnoredForNowHigh32BitPart As Long
End Type

Private Type TOKEN_PRIVILEGES
 PrivilegeCount As Long
 TheLuid As LUID
 Attributes As Long
End Type

Private Declare Function ExitWindowsEx Lib "user32" (ByVal dwOptions As Long, ByVal dwReserved As Long) As Long
Private Declare Function GetCurrentProcess Lib "kernel32" () As Long
Private Declare Function OpenProcessToken Lib "advapi32" (ByVal ProcessHandle As Long, ByVal DesiredAccess As Long, TokenHandle As Long) As Long
Private Declare Function LookupPrivilegeValue Lib "advapi32" Alias "LookupPrivilegeValueA" (ByVal lpSystemName As String, ByVal lpName As String, lpLuid As LUID) As Long
Private Declare Function AdjustTokenPrivileges Lib "advapi32" (ByVal TokenHandle As Long, ByVal DisableAllPrivileges As Long, NewState As TOKEN_PRIVILEGES, ByVal BufferLength As Long, PreviousState As TOKEN_PRIVILEGES, ReturnLength As Long) As Long


Sub RebootPC()
 On Local Error GoTo RebootPC_ErrorHandler
 Const csProcName = "RebootPC"
 
 Dim hProcessHandle As Long
 Dim hTokenHandle As Long
 Dim tmpLuid As LUID
 Dim tkpNew As TOKEN_PRIVILEGES
 Dim tkpPrevious As TOKEN_PRIVILEGES
 Dim lBufferNeeded As Long

 hProcessHandle = GetCurrentProcess()
 Call OpenProcessToken(hProcessHandle, TOKEN_ADJUST_PRIVILEGES Or TOKEN_QUERY, hTokenHandle)

' Get the LUID for the shutdown privilege
 Call LookupPrivilegeValue("", "SeShutdownPrivilege", tmpLuid)

 tkpNew.PrivilegeCount = 1 ' One privilege to set
 tkpNew.TheLuid = tmpLuid
 tkpNew.Attributes = SE_PRIVILEGE_ENABLED

' Enable the shutdown privilege in the access token of this process.
 lBufferNeeded = 0
 Call AdjustTokenPrivileges(hTokenHandle, False, tkpNew, Len(tkpPrevious), tkpPrevious, lBufferNeeded)

' Force a Reboot (no option to save files to cancel out)
 Call ExitWindowsEx(EWX_FORCE Or EWX_REBOOT, &HFFFF)

 Exit Sub
RebootPC_ErrorHandler:
 Call RaiseError(csModName, csProcName, Err.Number, Err.Description)
End Sub

没有评论: