写作&使用Windows API读取Windows注册表

时间:2018-01-24 17:50:22

标签: c++ winapi visual-studio-2017

我正在尝试从Windows寄存器中读取和读取:

书面方式:

std::string path = "c:\\"
LPCTSTR str_data = TEXT(path.c_str());
auto size = static_cast<DWORD>(strlen(str_data));
LONG setRes = RegSetValueEx(*key, TEXT("DumpFolder"), 0, REG_EXPAND_SZ, (LPBYTE)str_data, size);

读:

char str_data[1028];
DWORD keyType;
DWORD size;
auto sk = TEXT("SOFTWARE\\Microsoft\\Windows\\Windows Error reporting\\LocalDumps");
auto status = RegGetValue(HKEY_LOCAL_MACHINE, sk, TEXT("DumpFolder"), RF_RT_REG_EXPAND_SZ, &keyType, str_data, &size);

写入似乎工作正常,至少在regedit.exe中看起来很好。

使用ERROR_INVALID_PARAMETER = 87读取失败。如果我将RF_RT_REG_EXPAND_SZ更改为RRF_RT_ANY,它在调试模式下工作,但在发布时仍然失败,错误代码为ERROR_MORE_DATA = 234.我试过了:

std::string path = "c:\\";
path = path + "\0"  (it should be null terminated anyway

但它无济于事

更新

首先,感谢您的回答,我现在理解的事情好一点。不幸的是,我仍然无法成功读取字符串。

以下是以下答案结合的测试示例:

HKEY registry_key;
LPCTSTR sk = "SOFTWARE\\Microsoft\\Windows\\Windows Error Reporting";

// open registry key
auto openRes = RegOpenKey(HKEY_CURRENT_USER, sk, &registry_key);

// set default dump options
HKEY default_key;
auto createRes = RegCreateKey(registry_key, "LocalDumps", &default_key);
if (createRes != ERROR_SUCCESS) {
    auto b = createRes;
}    

std::string path = "c:\\";
LONG setRes = RegSetValueExA(default_key, "DumpFolder", 0, REG_EXPAND_SZ, (LPCBYTE)path.c_str(), path.size() + 1);

std::string str_data;
DWORD size = 0;
const char *sak = "SOFTWARE\\Microsoft\\Windows\\Windows Error reporting\\LocalDumps";
auto status = RegGetValueA(HKEY_CURRENT_USER, sak, "DumpFolder", RRF_RT_REG_EXPAND_SZ, NULL, NULL, &size);
if ((status == ERROR_SUCCESS) && (size > 1)) {
    str_data.resize(size - 1);
    status = RegGetValueA(HKEY_CURRENT_USER, sk, "DumpFolder", RRF_RT_REG_EXPAND_SZ, NULL, &str_data[0], &size);
}

再次写入工作正常(在regedit中检查,并返回错误代码)。另一方面,读取字符串寄存器的大小会将大小设置为0并返回错误代码87 = ERROR_INVALID_PARAMETER。

显然,我仍然缺少一些东西。 (项目设置为多字节字符集)

在解决了以下答案提出的问题后,以下代码对我有用:

#include <Windows.h>
#include <string>
#include <iostream>

#define reg_type HKEY_LOCAL_MACHINE

void main() {

    const std::string reg_path = "Software\\Microsoft\\Windows\\Windows Error Reporting\\LocalDumps";
    const std::string dump_folder = "DumpFolder";
    const std::string path = "c:\\";

    // WRITING
    HKEY default_key;
    auto status = RegCreateKeyExA(reg_type, reg_path.c_str(), 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE | KEY_QUERY_VALUE, NULL, &default_key, NULL);
    if (status != ERROR_SUCCESS) {
        std::cout << "Creating key failed.";
        return;
    }
    status = RegSetValueExA(default_key, dump_folder.c_str(), 0, REG_EXPAND_SZ, (LPCBYTE)path.c_str(), path.size() + 1);
    if (status != ERROR_SUCCESS) {
        std::cout << "Setting key value failed.";
        return;
    }

    // READING
    std::string str_data;
    DWORD size = 0;
    status = RegGetValueA(default_key, "", dump_folder.c_str(), RRF_NOEXPAND | RRF_RT_REG_EXPAND_SZ, NULL, NULL, &size);
    if ((status == ERROR_SUCCESS) && (size > 1)){
        str_data.resize(size - 1);
        status = RegGetValueA(default_key, "", dump_folder.c_str(), RRF_NOEXPAND | RRF_RT_REG_EXPAND_SZ, NULL, &str_data[0], &size);
        std::cout << "Successfully read key value: " << str_data;
    } else {
        std::cout << "Unable to retrive value. Error: " << status;
    }

    RegCloseKey(default_key);
}

我发现,RegGetValueA应该用

调用
RRF_NOEXPAND | RRF_RT_REG_EXPAND_SZ 

标志,看起来很奇怪,但是在标题中描述了它,所以我猜它是正确的。如果仅使用

RRF_RT_REG_EXPAND_SZ

错误87出现ERROR_INVALID_PARAMETER。

1 个答案:

答案 0 :(得分:4)

在写作方面:

std::string使用char个元素,但TCHAR映射到charwchar_t,具体取决于您的代码是否使用UNICODE定义编译或不。

TEXT()宏仅适用于编译时文字,您无法将其与运行时数据一起使用。 TEXT(path.c_str())是无效的类型广播,如果启用了UNICODE,则无法编译。

您显然正在处理char数据,因此您应该使用基于char的API函数而不是基于TCHAR的函数。

您也没有遵循RegSetValueEx()最重要的规则之一:

  

对于基于字符串的类型,例如REG_SZ,字符串必须以空值终止。对于REG_MULTI_SZ数据类型,字符串必须以两个空字符终止... lpData参数指向的信息大小(以字节为单位)。 如果数据类型为REG_SZ,REG_EXPAND_SZ或REG_MULTI_SZ,则cbData必须包含终止空字符或的大小。

std::string::c_str()返回指向以空值终止的数据的指针,但在报告要写入注册表的数据大小时,不包括空终止符。 RegGetValue()知道如何处理这个错误,但RegGetValueEx()却没有。您可能不是唯一一个读取该值的人,因此请确保正确包含空终止符。

请改为尝试:

std::string path = "c:\\";
LONG setRes = RegSetValueExA(*key, "DumpFolder", 0, REG_EXPAND_SZ, (LPCBYTE)path.c_str(), path.size()+1);

在阅读方面:

您收到错误是因为您没有告诉RegGetValue() str_data缓冲区有多大。在传递之前,您必须将size变量设置为str_data的大小(以字节为单位)。

请改为尝试:

char str_data[1028];
DWORD size = sizeof(str_data);
DWORD dwFlags = RRF_RT_REG_EXPAND_SZ;
// NOTE: when using RRF_RT_REG_EXPAND_SZ, RRF_NOEXPAND is *required* prior to Windows 8.1!
auto status = RegGetValueA(HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows\\Windows Error reporting\\LocalDumps", "DumpFolder", RRF_RT_REG_EXPAND_SZ | RRF_NOEXPAND, NULL, str_data, &size);

可替换地:

std:string str_data;
DWORD size = 0;
const char *sk = "SOFTWARE\\Microsoft\\Windows\\Windows Error reporting\\LocalDumps";
// NOTE: when using RRF_RT_REG_EXPAND_SZ, RRF_NOEXPAND is *required* prior to Windows 8.1!
const DWORD dwFlags = RRF_RT_REG_EXPAND_SZ | RRF_NOEXPAND;
auto status = RegGetValueA(HKEY_LOCAL_MACHINE, sk, "DumpFolder", dwFlags, NULL, NULL, &size);
if ((status == ERROR_SUCCESS) && (size > 1))
{
    str_data.resize(size-1);
    status = RegGetValueA(HKEY_LOCAL_MACHINE, sk, "DumpFolder", dwFlags, NULL, &str_data[0], &size);
}

更新:您的新代码失败,因为您引入了新错误。

您正在使用适用于16位应用的旧版注册表功能。您需要使用RegOpenKeyEx / RegCreateKeyEx而不是RegOpenKey / RegCreateKey,然后您只能指定实际需要的特定访问权限(创建子项,设置值,读取值等)。更好的是,RegCreateKeyEx()会为您创建缺少的密钥,因此您不需要单独手动打开父密钥来创建新的子密钥。

此外,您已将HKEY_LOCAL_MACHINE更改为HKEY_CURRENT_USER,但不一致。您的一些步骤使用一个根,其他步骤使用另一个根。您无法回读正在撰写的值,因为您没有从您写入的相同密钥中读取。

请改为尝试:

LPCSTR sk = "SOFTWARE\\Microsoft\\Windows\\Windows Error Reporting\\LocalDumps";
HKEY default_key;
auto status = RegCreateKeyExA(HKEY_LOCAL_MACHINE, sk, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_SET_VALUE, NULL, &default_key, NULL); 
if (status == ERROR_SUCCESS)
{
    std::string path = "c:\\";
    status = RegSetValueExA(default_key, "DumpFolder", 0, REG_EXPAND_SZ, (LPCBYTE)path.c_str(), path.size() + 1);
    RegCloseKey(default_key);
}

LPCSTR sk = "SOFTWARE\\Microsoft\\Windows\\Windows Error reporting\\LocalDumps";
std::string str_data;
DWORD size = 0;
// NOTE: when using RRF_RT_REG_EXPAND_SZ, RRF_NOEXPAND is *required* prior to Windows 8.1!
const DWORD dwFlags = RRF_RT_REG_EXPAND_SZ | RRF_NOEXPAND;
auto status = RegGetValueA(HKEY_LOCAL_MACHINE, sk, "DumpFolder", dwFlags, NULL, NULL, &size);
if ((status == ERROR_SUCCESS) && (size > 1))
{
    str_data.resize(size - 1);
    status = RegGetValueA(HKEY_LOCAL_MACHINE, sk, "DumpFolder", dwFlags, NULL, &str_data[0], &size);
}

另一方面,当你必须进行多次API调用来读取一个值(即查询大小,然后查询数据)时,你应该先显式打开父键:

const char *sk = "SOFTWARE\\Microsoft\\Windows\\Windows Error reporting\\LocalDumps";
std:string str_data;
HKEY default_key;
auto status = RegOpenKeyExA(HKEY_LOCAL_MACHINE, sk, 0, KEY_QUERY_VALUE, &dumps_key); 
if (status == ERROR_SUCCESS)
{
    DWORD size = 0;
    // NOTE: when using RRF_RT_REG_EXPAND_SZ, RRF_NOEXPAND is *required* prior to Windows 8.1!
    const DWORD dwFlags = RRF_RT_REG_EXPAND_SZ | RRF_NOEXPAND;

    status = RegGetValueA(default_key, "", "DumpFolder", dwFlags, NULL, NULL, &size);
    if ((status == ERROR_SUCCESS) && (size > 1))
    {
        str_data.resize(size-1);
        status = RegGetValueA(default_key, "", "DumpFolder", dwFlags, NULL, &str_data[0], &size);
    }

    RegCloseKey(default_key);
}