2068f089c8
-BEGIN VERIFY SCRIPT- mkdir -p src/util git mv src/util.h src/util/system.h git mv src/util.cpp src/util/system.cpp git mv src/utilmemory.h src/util/memory.h git mv src/utilmoneystr.h src/util/moneystr.h git mv src/utilmoneystr.cpp src/util/moneystr.cpp git mv src/utilstrencodings.h src/util/strencodings.h git mv src/utilstrencodings.cpp src/util/strencodings.cpp git mv src/utiltime.h src/util/time.h git mv src/utiltime.cpp src/util/time.cpp sed -i 's/<util\.h>/<util\/system\.h>/g' $(git ls-files 'src/*.h' 'src/*.cpp') sed -i 's/<utilmemory\.h>/<util\/memory\.h>/g' $(git ls-files 'src/*.h' 'src/*.cpp') sed -i 's/<utilmoneystr\.h>/<util\/moneystr\.h>/g' $(git ls-files 'src/*.h' 'src/*.cpp') sed -i 's/<utilstrencodings\.h>/<util\/strencodings\.h>/g' $(git ls-files 'src/*.h' 'src/*.cpp') sed -i 's/<utiltime\.h>/<util\/time\.h>/g' $(git ls-files 'src/*.h' 'src/*.cpp') sed -i 's/BITCOIN_UTIL_H/BITCOIN_UTIL_SYSTEM_H/g' src/util/system.h sed -i 's/BITCOIN_UTILMEMORY_H/BITCOIN_UTIL_MEMORY_H/g' src/util/memory.h sed -i 's/BITCOIN_UTILMONEYSTR_H/BITCOIN_UTIL_MONEYSTR_H/g' src/util/moneystr.h sed -i 's/BITCOIN_UTILSTRENCODINGS_H/BITCOIN_UTIL_STRENCODINGS_H/g' src/util/strencodings.h sed -i 's/BITCOIN_UTILTIME_H/BITCOIN_UTIL_TIME_H/g' src/util/time.h sed -i 's/ util\.\(h\|cpp\)/ util\/system\.\1/g' src/Makefile.am sed -i 's/utilmemory\.\(h\|cpp\)/util\/memory\.\1/g' src/Makefile.am sed -i 's/utilmoneystr\.\(h\|cpp\)/util\/moneystr\.\1/g' src/Makefile.am sed -i 's/utilstrencodings\.\(h\|cpp\)/util\/strencodings\.\1/g' src/Makefile.am sed -i 's/utiltime\.\(h\|cpp\)/util\/time\.\1/g' src/Makefile.am sed -i 's/-> util ->/-> util\/system ->/' test/lint/lint-circular-dependencies.sh sed -i 's/src\/util\.cpp/src\/util\/system\.cpp/g' test/lint/lint-format-strings.py test/lint/lint-locale-dependence.sh sed -i 's/src\/utilmoneystr\.cpp/src\/util\/moneystr\.cpp/g' test/lint/lint-locale-dependence.sh sed -i 's/src\/utilstrencodings\.\(h\|cpp\)/src\/util\/strencodings\.\1/g' test/lint/lint-locale-dependence.sh sed -i 's/src\\utilstrencodings\.cpp/src\\util\\strencodings\.cpp/' build_msvc/libbitcoinconsensus/libbitcoinconsensus.vcxproj -END VERIFY SCRIPT-
71 lines
2.4 KiB
C++
71 lines
2.4 KiB
C++
// Copyright (c) 2014-2018 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 <shutdown.h>
|
|
#include <util/system.h>
|
|
|
|
#include <windows.h>
|
|
|
|
#include <QDebug>
|
|
|
|
#include <openssl/rand.h>
|
|
|
|
// 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);
|
|
|
|
// Seed OpenSSL PRNG with Windows event data (e.g. mouse movements and other user interactions)
|
|
if (RAND_event(pMsg->message, pMsg->wParam, pMsg->lParam) == 0) {
|
|
// Warn only once as this is performance-critical
|
|
static bool warned = false;
|
|
if (!warned) {
|
|
LogPrintf("%s: OpenSSL RAND_event() failed to seed OpenSSL PRNG with enough data.\n", __func__);
|
|
warned = true;
|
|
}
|
|
}
|
|
|
|
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 == nullptr) {
|
|
qWarning() << "registerShutdownBlockReason: GetProcAddress for ShutdownBlockReasonCreate failed";
|
|
return;
|
|
}
|
|
|
|
if (shutdownBRCreate(mainWinId, strReason.toStdWString().c_str()))
|
|
qWarning() << "registerShutdownBlockReason: Successfully registered: " + strReason;
|
|
else
|
|
qWarning() << "registerShutdownBlockReason: Failed to register: " + strReason;
|
|
}
|
|
#endif
|