122 lines
4.4 KiB
C++
122 lines
4.4 KiB
C++
/*
|
|
* Copyright (c) 2017, Intel Corporation
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
|
* copy of this software and associated documentation files (the "Software"),
|
|
* to deal in the Software without restriction, including without limitation
|
|
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
* and/or sell copies of the Software, and to permit persons to whom the
|
|
* Software is furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included
|
|
* in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
|
|
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
|
|
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
|
* OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
|
|
#include "runtime/os_interface/debug_settings_manager.h"
|
|
#include "runtime/utilities/debug_settings_reader.h"
|
|
#include <stdint.h>
|
|
#include "runtime/os_interface/windows/windows_wrapper.h"
|
|
#include "runtime/os_interface/windows/registry_reader.h"
|
|
|
|
namespace OCLRT {
|
|
SettingsReader *SettingsReader::createOsReader() {
|
|
return new RegistryReader;
|
|
}
|
|
|
|
bool RegistryReader::getSetting(const char *settingName, bool defaultValue) {
|
|
return getSetting(settingName, static_cast<int32_t>(defaultValue)) ? true : false;
|
|
}
|
|
|
|
int32_t RegistryReader::getSetting(const char *settingName, int32_t defaultValue) {
|
|
HKEY Key;
|
|
DWORD value = defaultValue;
|
|
DWORD success = ERROR_SUCCESS;
|
|
|
|
success = RegOpenKeyExA(HKEY_LOCAL_MACHINE,
|
|
igdrclRegKey.c_str(),
|
|
0,
|
|
KEY_READ,
|
|
&Key);
|
|
|
|
if (ERROR_SUCCESS == success) {
|
|
DWORD regType;
|
|
DWORD size = sizeof(ULONG);
|
|
|
|
success = RegQueryValueExA(Key,
|
|
settingName,
|
|
NULL,
|
|
®Type,
|
|
(LPBYTE)&value,
|
|
&size);
|
|
RegCloseKey(Key);
|
|
}
|
|
|
|
return value;
|
|
}
|
|
|
|
std::string RegistryReader::getSetting(const char *settingName, const std::string &value) {
|
|
HKEY Key;
|
|
DWORD success = ERROR_SUCCESS;
|
|
bool retFlag = false;
|
|
std::string keyValue = value;
|
|
|
|
success = RegOpenKeyExA(HKEY_LOCAL_MACHINE,
|
|
igdrclRegKey.c_str(),
|
|
0,
|
|
KEY_READ,
|
|
&Key);
|
|
|
|
if (ERROR_SUCCESS == success) {
|
|
DWORD regType = REG_NONE;
|
|
DWORD regSize = 0;
|
|
|
|
success = RegQueryValueExA(Key,
|
|
settingName,
|
|
NULL,
|
|
®Type,
|
|
NULL,
|
|
®Size);
|
|
|
|
if (success == ERROR_SUCCESS && regType == REG_SZ) {
|
|
char *regData = new char[regSize];
|
|
success = RegQueryValueExA(Key,
|
|
settingName,
|
|
NULL,
|
|
®Type,
|
|
(LPBYTE)regData,
|
|
®Size);
|
|
|
|
keyValue.assign(regData);
|
|
delete[] regData;
|
|
retFlag = true;
|
|
} else if (success == ERROR_SUCCESS && regType == REG_BINARY) {
|
|
std::unique_ptr<wchar_t[]> regData(new wchar_t[regSize]);
|
|
success = RegQueryValueExA(Key,
|
|
settingName,
|
|
NULL,
|
|
®Type,
|
|
(LPBYTE)regData.get(),
|
|
®Size);
|
|
|
|
size_t charsConverted = 0;
|
|
std::unique_ptr<char[]> convertedData(new char[regSize]);
|
|
|
|
wcstombs_s(&charsConverted, convertedData.get(), regSize, regData.get(), regSize);
|
|
|
|
keyValue.assign(convertedData.get());
|
|
retFlag = true;
|
|
}
|
|
|
|
RegCloseKey(Key);
|
|
}
|
|
return keyValue;
|
|
}
|
|
}; // namespace OCLRT
|