From ed5640d8b587fbcfed7dd7967f3de04b37a76f26 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 7 Apr 2024 11:06:44 +0200 Subject: Adding upstream version 4:7.4.7. Signed-off-by: Daniel Baumann --- shell/source/win32/shlxthandler/util/registry.cxx | 173 ++++++++++++++++++++++ 1 file changed, 173 insertions(+) create mode 100644 shell/source/win32/shlxthandler/util/registry.cxx (limited to 'shell/source/win32/shlxthandler/util/registry.cxx') diff --git a/shell/source/win32/shlxthandler/util/registry.cxx b/shell/source/win32/shlxthandler/util/registry.cxx new file mode 100644 index 000000000..cc42c092e --- /dev/null +++ b/shell/source/win32/shlxthandler/util/registry.cxx @@ -0,0 +1,173 @@ +/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ +/* + * This file is part of the LibreOffice project. + * + * This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. + * + * This file incorporates work covered by the following license notice: + * + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed + * with this work for additional information regarding copyright + * ownership. The ASF licenses this file to you under the Apache + * License, Version 2.0 (the "License"); you may not use this file + * except in compliance with the License. You may obtain a copy of + * the License at http://www.apache.org/licenses/LICENSE-2.0 . + */ + + +#if !defined WIN32_LEAN_AND_MEAN +# define WIN32_LEAN_AND_MEAN +#endif +#include + +#include +#include + +#include + +bool SetRegistryKey(HKEY RootKey, const Filepath_char_t* KeyName, const Filepath_char_t* ValueName, const Filepath_char_t* Value) +{ + HKEY hSubKey; + + // open or create the desired key + wchar_t dummy[] = L""; + int rc = RegCreateKeyExW( + RootKey, KeyName, 0, dummy, REG_OPTION_NON_VOLATILE, KEY_WRITE, nullptr, &hSubKey, nullptr); + + if (ERROR_SUCCESS == rc) + { + rc = RegSetValueExW( + hSubKey, ValueName, 0, REG_SZ, reinterpret_cast(Value), + static_cast((wcslen(Value) + 1) * sizeof(*Value))); + + RegCloseKey(hSubKey); + } + + return (ERROR_SUCCESS == rc); +} + + +bool DeleteRegistryKey(HKEY RootKey, const Filepath_char_t* KeyName) +{ + HKEY hKey; + + int rc = RegOpenKeyExW( + RootKey, + KeyName, + 0, + KEY_READ | DELETE, + &hKey); + + if ( rc == ERROR_FILE_NOT_FOUND ) + return true; + + if (ERROR_SUCCESS == rc) + { + wchar_t* SubKey; + DWORD nMaxSubKeyLen; + + rc = RegQueryInfoKeyW( + hKey, nullptr, nullptr, nullptr, nullptr, + &nMaxSubKeyLen, + nullptr, nullptr, nullptr, nullptr, nullptr, nullptr); + + nMaxSubKeyLen++; // space for trailing '\0' + + SubKey = static_cast( + _alloca(nMaxSubKeyLen*sizeof(wchar_t))); + + while (ERROR_SUCCESS == rc) + { + DWORD nLen = nMaxSubKeyLen; + + rc = RegEnumKeyExW( + hKey, + 0, // always index zero + SubKey, + &nLen, + nullptr, nullptr, nullptr, nullptr); + + if (ERROR_NO_MORE_ITEMS == rc) + { + rc = RegDeleteKeyW(RootKey, KeyName); + break; + } + else if (rc == ERROR_SUCCESS) + { + DeleteRegistryKey(hKey, SubKey); + } + + } // while + + RegCloseKey(hKey); + + } // if + + return (ERROR_SUCCESS == rc); +} + +/** May be used to determine if the specified registry key has subkeys + The function returns true on success else if an error occurs false +*/ +bool HasSubkeysRegistryKey(HKEY RootKey, const Filepath_char_t* KeyName, /* out */ bool& bResult) +{ + HKEY hKey; + + LONG rc = RegOpenKeyExW(RootKey, KeyName, 0, KEY_READ, &hKey); + + if (ERROR_SUCCESS == rc) + { + DWORD nSubKeys = 0; + + rc = RegQueryInfoKeyW(hKey, nullptr, nullptr, nullptr, &nSubKeys, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr); + + RegCloseKey(hKey); + bResult = (nSubKeys > 0); + } + + return (ERROR_SUCCESS == rc); +} + +// Convert a CLSID to a char string. +Filepath_t ClsidToString(const CLSID& clsid) +{ + // Get CLSID + LPOLESTR wszCLSID = nullptr; + StringFromCLSID(clsid, &wszCLSID); + + std::wstring sResult = wszCLSID; + + // Free memory. + CoTaskMemFree(wszCLSID) ; + + return sResult; +} + + +bool QueryRegistryKey(HKEY RootKey, const Filepath_char_t* KeyName, const Filepath_char_t* ValueName, Filepath_char_t *pszData, DWORD dwBufLen) +{ + HKEY hKey; + + int rc = RegOpenKeyExW( + RootKey, + KeyName, + 0, + KEY_READ, + &hKey); + + if (ERROR_SUCCESS == rc) + { + DWORD dwBytes = dwBufLen * sizeof(*pszData); + rc = RegQueryValueExW( + hKey, ValueName, nullptr, nullptr, reinterpret_cast(pszData),&dwBytes); + + RegCloseKey(hKey); + } + + return (ERROR_SUCCESS == rc); +} + +/* vim:set shiftwidth=4 softtabstop=4 expandtab: */ -- cgit v1.2.3