mirror of https://github.com/lukechilds/komodo.git
Browse Source
d282c1f
[Qt] catch Windows shutdown events while client is running (Philip Kaufmann)
try
Wladimir J. van der Laan
11 years ago
4 changed files with 111 additions and 3 deletions
@ -0,0 +1,57 @@ |
|||
// Copyright (c) 2014 The Bitcoin developers
|
|||
// Distributed under the MIT/X11 software license, see the accompanying
|
|||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|||
|
|||
#include "winshutdownmonitor.h" |
|||
|
|||
#if defined(Q_OS_WIN) && QT_VERSION >= 0x050000 |
|||
#include "init.h" |
|||
|
|||
#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.
|
|||
StartShutdown(); |
|||
*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)GetProcAddress(GetModuleHandleA("User32.dll"), "ShutdownBlockReasonCreate"); |
|||
if (shutdownBRCreate == NULL) { |
|||
qDebug() << "registerShutdownBlockReason : GetProcAddress for ShutdownBlockReasonCreate failed"; |
|||
return; |
|||
} |
|||
|
|||
if (shutdownBRCreate(mainWinId, strReason.toStdWString().c_str())) |
|||
qDebug() << "registerShutdownBlockReason : Successfully registered: " + strReason; |
|||
else |
|||
qDebug() << "registerShutdownBlockReason : Failed to register: " + strReason; |
|||
} |
|||
#endif |
@ -0,0 +1,29 @@ |
|||
// Copyright (c) 2014 The Bitcoin developers
|
|||
// Distributed under the MIT/X11 software license, see the accompanying
|
|||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|||
|
|||
#ifndef WINSHUTDOWNMONITOR_H |
|||
#define WINSHUTDOWNMONITOR_H |
|||
|
|||
#ifdef WIN32 |
|||
#include <QByteArray> |
|||
#include <QString> |
|||
|
|||
#if QT_VERSION >= 0x050000 |
|||
#include <windef.h> // for HWND |
|||
|
|||
#include <QAbstractNativeEventFilter> |
|||
|
|||
class WinShutdownMonitor : public QAbstractNativeEventFilter |
|||
{ |
|||
public: |
|||
/** Implements QAbstractNativeEventFilter interface for processing Windows messages */ |
|||
bool nativeEventFilter(const QByteArray &eventType, void *pMessage, long *pnResult); |
|||
|
|||
/** Register the reason for blocking shutdown on Windows to allow clean client exit */ |
|||
static void registerShutdownBlockReason(const QString& strReason, const HWND& mainWinId); |
|||
}; |
|||
#endif |
|||
#endif |
|||
|
|||
#endif // WINSHUTDOWNMONITOR_H
|
Loading…
Reference in new issue