88 lines
2.4 KiB
C++
88 lines
2.4 KiB
C++
|
//+----------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// File: cmexitwin.cpp
|
||
|
|
//
|
||
|
|
// Module: Common Code
|
||
|
|
//
|
||
|
|
// Synopsis: Implements the function MyExitWindowsEx.
|
||
|
|
//
|
||
|
|
// Copyright (c) 1998-1999 Microsoft Corporation
|
||
|
|
//
|
||
|
|
// Author: quintinb Created Heaser 08/19/99
|
||
|
|
//
|
||
|
|
//+----------------------------------------------------------------------------
|
||
|
|
#include <windows.h>
|
||
|
|
|
||
|
|
BOOL MyExitWindowsEx(UINT uFlags,
|
||
|
|
DWORD dwRsvd)
|
||
|
|
{
|
||
|
|
BOOL bRes;
|
||
|
|
|
||
|
|
//
|
||
|
|
// If platform is NT, we will have to adjust privileges before rebooting
|
||
|
|
//
|
||
|
|
if (OS_NT)
|
||
|
|
{
|
||
|
|
HANDLE hToken; // handle to process token
|
||
|
|
TOKEN_PRIVILEGES tkp; // ptr. to token structure
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get the current process token handle
|
||
|
|
// so we can get shutdown privilege.
|
||
|
|
//
|
||
|
|
if (!OpenProcessToken(GetCurrentProcess(),
|
||
|
|
TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
|
||
|
|
&hToken))
|
||
|
|
{
|
||
|
|
CMTRACE1(TEXT("MyExitWindowsEx() OpenThreadToken() failed, GLE=%u."), GetLastError());
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get the LUID for shutdown privilege
|
||
|
|
//
|
||
|
|
bRes = LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME,
|
||
|
|
&tkp.Privileges[0].Luid);
|
||
|
|
#ifdef DEBUG
|
||
|
|
if (!bRes)
|
||
|
|
{
|
||
|
|
CMTRACE1(TEXT("MyExitWindowsEx() LookupPrivilegeValue() failed, GLE=%u."), GetLastError());
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
tkp.PrivilegeCount = 1;
|
||
|
|
tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get shutdown privilege for this process
|
||
|
|
//
|
||
|
|
AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,
|
||
|
|
(PTOKEN_PRIVILEGES) NULL, 0);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Cannot reliably test the return value of AdjustTokenPrivileges
|
||
|
|
//
|
||
|
|
if (GetLastError() != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
CMTRACE1(TEXT("MyExitWindowsEx() AdjustTokenPrivileges() failed, GLE=%u."), GetLastError());
|
||
|
|
|
||
|
|
CloseHandle(hToken);
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseHandle(hToken);
|
||
|
|
}
|
||
|
|
|
||
|
|
bRes = ExitWindowsEx(uFlags,dwRsvd);
|
||
|
|
#ifdef DEBUG
|
||
|
|
if (!bRes)
|
||
|
|
{
|
||
|
|
CMTRACE1(TEXT("MyExitWindowsEx() ExitWindowsEx() failed, GLE=%u."), GetLastError());
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
return (bRes);
|
||
|
|
}
|