f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync/* $Id$ */
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync/** @file
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * IPRT - RTSystemShutdown, Windows.
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync */
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync/*
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * Copyright (C) 2012 Oracle Corporation
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync *
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * available from http://www.virtualbox.org. This file is free software;
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * you can redistribute it and/or modify it under the terms of the GNU
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * General Public License (GPL) as published by the Free Software
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync *
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * The contents of this file may alternatively be used under the terms
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * of the Common Development and Distribution License Version 1.0
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * VirtualBox OSE distribution, in which case the provisions of the
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * CDDL are applicable instead of those of the GPL.
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync *
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * You may elect to license modified versions of this file under the
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync * terms and conditions of either the GPL or the CDDL or both.
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync */
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync/*******************************************************************************
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync* Header Files *
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync*******************************************************************************/
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync#include <iprt/system.h>
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync#include "internal/iprt.h"
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync#include <iprt/assert.h>
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync#include <iprt/err.h>
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync#include <iprt/string.h>
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync#include <Windows.h>
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync
5984df8f9cfd9941dcd5a874848fbcd7e50d15ffvboxsync
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsyncRTDECL(int) RTSystemShutdown(RTMSINTERVAL cMsDelay, uint32_t fFlags, const char *pszLogMsg)
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync{
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync AssertPtrReturn(pszLogMsg, VERR_INVALID_POINTER);
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync AssertReturn(!(fFlags & ~RTSYSTEM_SHUTDOWN_VALID_MASK), VERR_INVALID_PARAMETER);
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync PRTUTF16 pwszLogMsg;
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync int rc = RTStrToUtf16(pszLogMsg, &pwszLogMsg);
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync if (RT_FAILURE(rc))
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync return rc;
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync DWORD cSecsTimeout = (cMsDelay + 499) / 1000;
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync BOOL fRebootAfterShutdown = (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_REBOOT
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync ? TRUE : FALSE;
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync BOOL fForceAppsClosed = fFlags & RTSYSTEM_SHUTDOWN_FORCE ? TRUE : FALSE;
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync /*
64eea8161bef2aa3c6516481383c830bca27abfevboxsync * Do the
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync */
64eea8161bef2aa3c6516481383c830bca27abfevboxsync if (InitiateSystemShutdownW(NULL /*pwszMachineName = NULL = localhost*/,
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync pwszLogMsg,
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync cSecsTimeout,
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync fForceAppsClosed,
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync fRebootAfterShutdown))
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync rc = (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_HALT ? VINF_SYS_MAY_POWER_OFF : VINF_SUCCESS;
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync else
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync {
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync /* If we failed because of missing privileges, try get the right to
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync shut down the system and call the api again. */
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync DWORD dwErr = GetLastError();
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync rc = RTErrConvertFromWin32(dwErr);
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync if (dwErr == ERROR_ACCESS_DENIED)
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync {
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync HANDLE hToken = NULL;
64eea8161bef2aa3c6516481383c830bca27abfevboxsync if (OpenThreadToken(GetCurrentThread(),
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync TOKEN_ADJUST_PRIVILEGES,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync TRUE /*OpenAsSelf*/,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync &hToken))
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync dwErr = NO_ERROR;
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync else
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync {
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync dwErr = GetLastError();
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync if (dwErr == ERROR_NO_TOKEN)
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync {
64eea8161bef2aa3c6516481383c830bca27abfevboxsync if (OpenProcessToken(GetCurrentProcess(),
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync TOKEN_ADJUST_PRIVILEGES,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync &hToken))
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync dwErr = NO_ERROR;
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync else
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync dwErr = GetLastError();
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync }
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync }
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync if (dwErr == NO_ERROR)
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync {
64eea8161bef2aa3c6516481383c830bca27abfevboxsync union
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync {
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync TOKEN_PRIVILEGES TokenPriv;
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync char ab[sizeof(TOKEN_PRIVILEGES) + sizeof(LUID_AND_ATTRIBUTES)];
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync } u;
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync u.TokenPriv.PrivilegeCount = 1;
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync u.TokenPriv.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync if (LookupPrivilegeValue(NULL /*localhost*/, SE_SHUTDOWN_NAME, &u.TokenPriv.Privileges[0].Luid))
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync {
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync if (AdjustTokenPrivileges(hToken,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync FALSE /*DisableAllPrivileges*/,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync &u.TokenPriv,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync RT_OFFSETOF(TOKEN_PRIVILEGES, Privileges[1]),
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync NULL,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync NULL) )
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync {
64eea8161bef2aa3c6516481383c830bca27abfevboxsync if (InitiateSystemShutdownW(NULL /*pwszMachineName = NULL = localhost*/,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync pwszLogMsg,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync cSecsTimeout,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync fForceAppsClosed,
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync fRebootAfterShutdown))
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync rc = (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_HALT ? VINF_SYS_MAY_POWER_OFF : VINF_SUCCESS;
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync else
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync {
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync dwErr = GetLastError();
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync rc = RTErrConvertFromWin32(dwErr);
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync }
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync }
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync CloseHandle(hToken);
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync }
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync }
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync }
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync }
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync RTUtf16Free(pwszLogMsg);
f84707e750cc4dcb56a0f2608224740e1db86e84vboxsync return rc;
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync}
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsyncRT_EXPORT_SYMBOL(RTSystemShutdown);
f8a829bc862da0c232b1291990e1e2c990d40cf4vboxsync