mirror of https://github.com/bitcoin/bitcoin.git
57 lines
1.9 KiB
C++
57 lines
1.9 KiB
C++
// Copyright (c) 2014-2019 The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#include <qt/winshutdownmonitor.h>
|
|
|
|
#if defined(Q_OS_WIN)
|
|
|
|
#include <windows.h>
|
|
|
|
#include <QDebug>
|
|
|
|
// If we don't want a message to be processed by Qt, return true and set result to
|
|
// the value that the window procedure should return. Otherwise return false.
|
|
bool WinShutdownMonitor::nativeEventFilter(const QByteArray &eventType, void *pMessage, long *pnResult)
|
|
{
|
|
Q_UNUSED(eventType);
|
|
|
|
MSG *pMsg = static_cast<MSG *>(pMessage);
|
|
|
|
switch(pMsg->message)
|
|
{
|
|
case WM_QUERYENDSESSION:
|
|
{
|
|
// Initiate a client shutdown after receiving a WM_QUERYENDSESSION and block
|
|
// Windows session end until we have finished client shutdown.
|
|
m_shutdown_fn();
|
|
*pnResult = FALSE;
|
|
return true;
|
|
}
|
|
|
|
case WM_ENDSESSION:
|
|
{
|
|
*pnResult = FALSE;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void WinShutdownMonitor::registerShutdownBlockReason(const QString& strReason, const HWND& mainWinId)
|
|
{
|
|
typedef BOOL (WINAPI *PSHUTDOWNBRCREATE)(HWND, LPCWSTR);
|
|
PSHUTDOWNBRCREATE shutdownBRCreate = (PSHUTDOWNBRCREATE)(void*)GetProcAddress(GetModuleHandleA("User32.dll"), "ShutdownBlockReasonCreate");
|
|
if (shutdownBRCreate == nullptr) {
|
|
qWarning() << "registerShutdownBlockReason: GetProcAddress for ShutdownBlockReasonCreate failed";
|
|
return;
|
|
}
|
|
|
|
if (shutdownBRCreate(mainWinId, strReason.toStdWString().c_str()))
|
|
qInfo() << "registerShutdownBlockReason: Successfully registered: " + strReason;
|
|
else
|
|
qWarning() << "registerShutdownBlockReason: Failed to register: " + strReason;
|
|
}
|
|
#endif
|