Shutdown/sleep/restart/logout class

Source: Internet
Author: User

Imports System
Imports System. Text
Imports System. Diagnostics
Imports System. Runtime. InteropServices
Public Class ExitWindwos
Public Enum RestartOptions
LogOff = 0
PowerOff = 8
Reboot = 2
ShutDown = 1
Suspend =-1
Hibernate =-2
End Enum

Public Structure LUID
Dim LowPart As Integer
Dim HighPart As Integer
End Structure


Dim pLuid As LUID

Dim Attributes As Integer
End Structure


Dim PrivilegeCount As Integer

End Structure

Private Const TOKEN_QUERY = & H8
Private Const EWX_FORCE = 4
Private Declare Function LoadLibrary Lib "kernel32" Alias "LoadLibraryA" (ByVal lpLibFileName As String) As IntPtr
Private Declare Function FreeLibrary Lib "kernel32" (ByVal hLibModule As IntPtr) As Integer
Private Declare Function GetProcAddress Lib "kernel32" (ByVal hModule As IntPtr, ByVal lpProcName As String) As IntPtr
Private Declare Function SetSuspendState Lib "Powrprof" (ByVal Hibernate As Integer, ByVal ForceCritical As Integer, ByVal DisableWakeEvent As Integer) As Integer
Private Declare Function OpenProcessToken Lib "advapi32.dll" (ByVal ProcessHandle As IntPtr, ByVal DesiredAccess As Integer, ByRef TokenHandle As IntPtr) As Integer
Private Declare Function LookupPrivilegeValue Lib "advapi32.dll" Alias "LookupPrivilegeValueA" (ByVal lpSystemName As String, ByVal lpName As String, ByRef lpLuid As LUID) As Integer
Private Declare Function using Lib "values" (ByVal TokenHandle As IntPtr, ByVal DisableAllPrivileges As Integer, ByRef NewState As values, ByVal BufferLength As Integer, ByRef previusstate As values, ByRef ReturnLength As Integer) as Integer
Private Declare Function ExitWindowsEx Lib "user32" (ByVal uFlags As Integer, ByVal dwReserved As Integer) As Integer
Private Declare Function FormatMessage Lib "kernel32" Alias "FormatMessageA" (ByVal dwFlags As Integer, ByVal lpSource As IntPtr, ByVal dwMessageId As Integer, ByVal dww.ageid As Integer, ByVal lpBuffer As StringBuilder, byVal nSize As Integer, ByVal Arguments As Integer) As Integer

Private Sub ExitWindows (ByVal how As RestartOptions, ByVal force As Boolean)
Select Case how
Case RestartOptions. Suspend
SuspendSystem (False, force)
Case RestartOptions. Hibernate
SuspendSystem (True, force)
Case Else
ExitWindows (Convert. ToInt32 (how), force)
End Select
End Sub

Private Sub ExitWindows (ByVal how As Integer, ByVal force As Boolean)
EnableToken ("SeShutdownPrivilege ")
If force Then how = how Or EWX_FORCE
If (ExitWindowsEx (how, 0) = 0) Then Throw New PrivilegeException (FormatError (Marshal. GetLastWin32Error ()))
End Sub

Private Sub EnableToken (ByVal privilege As String)
If Not CheckEntryPoint ("advapi32.dll", "AdjustTokenPrivileges") Then Return
Dim tokenHandle As IntPtr = IntPtr. Zero
Dim privilegeLUID = New LUID
Dim newPrivileges = New TOKEN_PRIVILEGES
Dim tokenPrivileges As TOKEN_PRIVILEGES
If (OpenProcessToken (Process. GetCurrentProcess (). Handle, TOKEN_ADJUST_PRIVILEGES Or TOKEN_QUERY, tokenHandle) = 0 Then Throw New PrivilegeException (FormatError (Marshal. GetLastWin32Error ()))
If (LookupPrivilegeValue ("", privilege, privilegeLUID) = 0 Then Throw New PrivilegeException (FormatError (Marshal. GetLastWin32Error ()))
TokenPrivileges. PrivilegeCount = 1
TokenPrivileges. Privileges. Attributes = SE_PRIVILEGE_ENABLED
TokenPrivileges. Privileges. pLuid = privilegeLUID
Dim Size As Integer = 4
If (AdjustTokenPrivileges (tokenHandle, 0, tokenPrivileges, 4 + (12 * tokenPrivileges. privilegeCount), newPrivileges, Size) = 0 Then Throw New PrivilegeException (FormatError (Marshal. getLastWin32Error ()))
End Sub

Private Sub SuspendSystem (ByVal hibernate As Boolean, ByVal force As Boolean)
If Not CheckEntryPoint ("powrprof. dll", "SetSuspendState") Then Throw New PlatformNotSupportedException ("The SetSuspendState method is not supported on this system! ")
SetSuspendState (Convert. ToInt32 (IIf (hibernate, 1, 0), Convert. ToInt32 (IIf (force, 1, 0), 0)
End Sub

Private Function CheckEntryPoint (ByVal library As String, ByVal method As String) As Boolean
Dim libPtr As IntPtr = LoadLibrary (library)
If Not libPtr. Equals (IntPtr. Zero) Then
If Not GetProcAddress (libPtr, method). Equals (IntPtr. Zero) Then
FreeLibrary (libPtr)
Return True
End If
FreeLibrary (libPtr)
End If
Return False
End Function

Private Function FormatError (ByVal number As Integer) As String
Dim Buffer = New S

Related Article

Contact Us

The content source of this page is from Internet, which doesn't represent Alibaba Cloud's opinion; products and services mentioned on that page don't have any relationship with Alibaba Cloud. If the content of the page makes you feel confusing, please write us an email, we will handle the problem within 5 days after receiving your email.

If you find any instances of plagiarism from the community, please send an email to: and provide relevant evidence. A staff member will contact you within 5 working days.

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.