diff options
author | Robert Loehning <robert.loehning@nokia.com> | 2010-03-23 16:13:00 +0100 |
---|---|---|
committer | Robert Loehning <robert.loehning@nokia.com> | 2010-03-23 16:16:32 +0100 |
commit | be3d859f4f81ec539aeee0385758042b4891b024 (patch) | |
tree | 8799d8b1760c0f83504c8acadbc920f7ecdb354b /src/shared/registryaccess | |
parent | 2f0e30c15030098bd13342081249509bc45405ea (diff) |
Moved code for accessing Windows Registry to src/shared.
Diffstat (limited to 'src/shared/registryaccess')
-rw-r--r-- | src/shared/registryaccess/registryaccess.cpp | 143 | ||||
-rw-r--r-- | src/shared/registryaccess/registryaccess.h | 83 | ||||
-rw-r--r-- | src/shared/registryaccess/registryaccess.pri | 10 |
3 files changed, 236 insertions, 0 deletions
diff --git a/src/shared/registryaccess/registryaccess.cpp b/src/shared/registryaccess/registryaccess.cpp new file mode 100644 index 0000000000..6ee650f252 --- /dev/null +++ b/src/shared/registryaccess/registryaccess.cpp @@ -0,0 +1,143 @@ +/************************************************************************** +** +** This file is part of Qt Creator +** +** Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies). +** +** Contact: Nokia Corporation (qt-info@nokia.com) +** +** Commercial Usage +** +** Licensees holding valid Qt Commercial licenses may use this file in +** accordance with the Qt Commercial License Agreement provided with the +** Software or, alternatively, in accordance with the terms contained in +** a written agreement between you and Nokia. +** +** GNU Lesser General Public License Usage +** +** Alternatively, this file may be used under the terms of the GNU Lesser +** General Public License version 2.1 as published by the Free Software +** Foundation and appearing in the file LICENSE.LGPL included in the +** packaging of this file. Please review the following information to +** ensure the GNU Lesser General Public License version 2.1 requirements +** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html. +** +** If you are unsure which license is appropriate for your use, please +** contact the sales department at http://qt.nokia.com/contact. +** +**************************************************************************/ + +#include "registryaccess.h" + +#include <QtGui/QApplication> + +#include <QtCore/QDir> +#include <QtCore/QTextStream> + +namespace RegistryAccess { + +static QString winErrorMessage(unsigned long error) +{ + QString rc = QString::fromLatin1("#%1: ").arg(error); + ushort *lpMsgBuf; + + const int len = FormatMessage( + FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, + NULL, error, 0, (LPTSTR)&lpMsgBuf, 0, NULL); + if (len) { + rc = QString::fromUtf16(lpMsgBuf, len); + LocalFree(lpMsgBuf); + } else { + rc += QString::fromLatin1("<unknown error>"); + } + return rc; +} + +QString msgFunctionFailed(const char *f, unsigned long error) +{ + return QString::fromLatin1("'%1' failed: %2").arg(QLatin1String(f), winErrorMessage(error)); +} + +static bool registryReadBinaryKey(HKEY handle, // HKEY_LOCAL_MACHINE, etc. + const WCHAR *valueName, + QByteArray *data, + QString *errorMessage) +{ + data->clear(); + DWORD type; + DWORD size; + // get size and retrieve + LONG rc = RegQueryValueEx(handle, valueName, 0, &type, 0, &size); + if (rc != ERROR_SUCCESS) { + *errorMessage = msgRegistryOperationFailed("read", valueName, msgFunctionFailed("RegQueryValueEx1", rc)); + return false; + } + BYTE *dataC = new BYTE[size + 1]; + // Will be Utf16 in case of a string + rc = RegQueryValueEx(handle, valueName, 0, &type, dataC, &size); + if (rc != ERROR_SUCCESS) { + *errorMessage = msgRegistryOperationFailed("read", valueName, msgFunctionFailed("RegQueryValueEx2", rc)); + return false; + } + *data = QByteArray(reinterpret_cast<const char*>(dataC), size); + delete [] dataC; + return true; +} + +bool registryReadStringKey(HKEY handle, // HKEY_LOCAL_MACHINE, etc. + const WCHAR *valueName, + QString *s, + QString *errorMessage) +{ + QByteArray data; + if (!registryReadBinaryKey(handle, valueName, &data, errorMessage)) + return false; + data += '\0'; + data += '\0'; + *s = QString::fromUtf16(reinterpret_cast<const unsigned short*>(data.data())); + return true; +} + +bool openRegistryKey(HKEY category, // HKEY_LOCAL_MACHINE, etc. + const WCHAR *key, + bool readWrite, + HKEY *keyHandle, + QString *errorMessage) +{ + + REGSAM accessRights = KEY_READ; + if (readWrite) + accessRights |= KEY_SET_VALUE; + const LONG rc = RegOpenKeyEx(category, key, 0, accessRights, keyHandle); + if (rc != ERROR_SUCCESS) { + *errorMessage = msgFunctionFailed("RegOpenKeyEx", rc); + return false; + } + return true; +} + +// Installation helpers: Format the debugger call with placeholders for PID and event +// '"[path]\qtcdebugger" [-wow] %ld %ld'. + +QString debuggerCall(const QString &additionalOption) +{ + QString rc; + QTextStream str(&rc); + str << '"' << QDir::toNativeSeparators(QApplication::applicationDirPath() + QLatin1Char('/') + + debuggerApplicationFileC + QLatin1String(".exe")) << '"'; + if (!additionalOption.isEmpty()) + str << ' ' << additionalOption; + str << " %ld %ld"; + return rc; +} + +bool isRegistered(HKEY handle, const QString &call, QString *errorMessage, QString *oldDebugger) +{ + QString registeredDebugger; + registryReadStringKey(handle, debuggerRegistryValueNameC, ®isteredDebugger, errorMessage); + if (oldDebugger) + *oldDebugger = registeredDebugger; + return !registeredDebugger.compare(call, Qt::CaseInsensitive); +} + +} // namespace RegistryAccess diff --git a/src/shared/registryaccess/registryaccess.h b/src/shared/registryaccess/registryaccess.h new file mode 100644 index 0000000000..dc1341f5e1 --- /dev/null +++ b/src/shared/registryaccess/registryaccess.h @@ -0,0 +1,83 @@ +/************************************************************************** +** +** This file is part of Qt Creator +** +** Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies). +** +** Contact: Nokia Corporation (qt-info@nokia.com) +** +** Commercial Usage +** +** Licensees holding valid Qt Commercial licenses may use this file in +** accordance with the Qt Commercial License Agreement provided with the +** Software or, alternatively, in accordance with the terms contained in +** a written agreement between you and Nokia. +** +** GNU Lesser General Public License Usage +** +** Alternatively, this file may be used under the terms of the GNU Lesser +** General Public License version 2.1 as published by the Free Software +** Foundation and appearing in the file LICENSE.LGPL included in the +** packaging of this file. Please review the following information to +** ensure the GNU Lesser General Public License version 2.1 requirements +** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html. +** +** If you are unsure which license is appropriate for your use, please +** contact the sales department at http://qt.nokia.com/contact. +** +**************************************************************************/ + +// Some functions used by qtcreator.exe and qtcdebugger.exe to check if +// qtcdebugger is currently registered for post-mortem debugging. +// This is only needed on Windows. + +#ifndef REGISTRYACCESS_H +#define REGISTRYACCESS_H + +#include <QtCore/QString> +#include <QtCore/QLatin1String> + +#include <Windows.h> + +namespace RegistryAccess { + +static const char *debuggerApplicationFileC = "qtcdebugger"; +static const WCHAR *debuggerRegistryKeyC = L"Software\\Microsoft\\Windows NT\\CurrentVersion\\AeDebug"; +static const WCHAR *debuggerRegistryValueNameC = L"Debugger"; + +static inline QString wCharToQString(const WCHAR *w) +{ + return QString::fromUtf16(reinterpret_cast<const ushort *>(w)); +} + +QString msgFunctionFailed(const char *f, unsigned long error); + +static inline QString msgRegistryOperationFailed(const char *op, const WCHAR *valueName, const QString &why) +{ + QString rc = QLatin1String("Registry "); + rc += QLatin1String(op); + rc += QLatin1String(" of "); + rc += wCharToQString(valueName); + rc += QLatin1String(" failed: "); + rc += why; + return rc; +} + +bool registryReadStringKey(HKEY handle, // HKEY_LOCAL_MACHINE, etc. + const WCHAR *valueName, + QString *s, + QString *errorMessage); + +bool openRegistryKey(HKEY category, // HKEY_LOCAL_MACHINE, etc. + const WCHAR *key, + bool readWrite, + HKEY *keyHandle, + QString *errorMessage); + +QString debuggerCall(const QString &additionalOption = QString()); + +bool isRegistered(HKEY handle, const QString &call, QString *errorMessage, QString *oldDebugger = 0); + +} // namespace RegistryAccess + +#endif // REGISTRYACCESS_H diff --git a/src/shared/registryaccess/registryaccess.pri b/src/shared/registryaccess/registryaccess.pri new file mode 100644 index 0000000000..0594521073 --- /dev/null +++ b/src/shared/registryaccess/registryaccess.pri @@ -0,0 +1,10 @@ +INCLUDEPATH *= $$PWD + +SOURCES += $$PWD/registryaccess.cpp +HEADERS += $$PWD/registryaccess.h + +LIBS *= -lpsapi +# PS API and registry functions +contains(QMAKE_CXX, cl) { + LIBS *= -ladvapi32 +} |