读取REG_BINARY会破坏方法的参数

时间:2014-07-22 20:55:46

标签: c++ registry

我正在尝试使用以下代码读取REG_BINARY。 value以正确的字符串结束,但在return语句后出现Acces违规错误。经过一些调试后,我发现RegQueryValueExW执行后,方法的参数被crap覆盖(例如,name以“test”开头,变为“Î\0ÎÎÎvv|& ...”在RegQueryValueExW之后。)

string GetREG_BINARYValue(HKEY hkey, DWORD cbData, wstring name)
{
    DWORD* data = new DWORD[cbData]; 
    DWORD size = cbData;

    if( RegQueryValueExW(hkey,name.c_str(), NULL, NULL, reinterpret_cast<LPBYTE>(&data), &size)  != ERROR_SUCCESS)
    {
        RegCloseKey(hkey);
        return "Could not read REG_BINARY registry value";
    }    

    string value =  ByteArrayToString((BYTE*)&data, size);
    return value;
}

从该方法调用该函数:

string RegistryReader::ReadRegValue(HKEY root, string key, string name)
{
    HKEY hkey;
    DWORD type;
    DWORD cbData;
    string result;
    wstring wkey, wname;
    std::wstring_convert<std::codecvt_utf8_utf16<wchar_t>> converter;
    wkey=converter.from_bytes(key);
    wname=converter.from_bytes(name);

    if (RegOpenKeyExW(root, wkey.c_str(), 0, KEY_ACCESS_DEF64 , &hkey) != ERROR_SUCCESS)
    {
        if (RegOpenKeyExW(root, wkey.c_str(), 0,  KEY_ACCESS_DEF32, &hkey) != ERROR_SUCCESS)
        {
            return "Could not find registry key";      
        }  
    }

    if (RegQueryValueExW(hkey, wname.c_str(), NULL, &type, NULL, &cbData) != ERROR_SUCCESS)
    {
        RegCloseKey(hkey);
        return "Could not open registry value";
    }

    switch (type)
    {
    case REG_SZ:
        result=GetREG_SZValue(hkey, cbData, wname);
        break;
    case REG_DWORD:
        result=GetREG_DWORDValue(hkey, cbData, wname);
        break;
    case REG_BINARY:
        result=GetREG_BINARYValue(hkey, cbData, wname); //<--Here
        break;
    default:
        return "Invalid data type";
        break;
    }
    return result;
}

起初我虽然这是一个数组大小的问题,但我最终得到了正确的结果,cbData总是包含我期望的大小。我可能在RegQueryValueExW做错了,但我很难找到明确的文档。这段代码有明显的错误吗?

1 个答案:

答案 0 :(得分:3)

您的代码有很多问题。我在评论中列出了很多这些问题。根本问题是您必须向RegQueryValueExW传递指向长度为size的数组的指针。但你没有。

现在,data是一个指向DWORD数组的指针,长度为size。所以它包含的字节数是你需要的四倍。这本身并不是那么糟糕。但是,不是传递data,而是传递&data。哪个是指针的地址而不是数组的地址。然后,当API函数写入指针而不是写入它指向的缓冲区时,您将调用未定义的行为。

你必须施放的事实应该是第一个警示标志。尽量不要施放。我再说一遍,尽量不要施展。当你施放时,你停止编译器发现你的错误。当编译器告诉你你犯了一个错误时,不要用强制转换忽略它。听听它。

如果我们必须使用这些全大写Windows类型,则需要分配BYTE数组:

BYTE* data = new BYTE[cbData]; 

现在,您可以将data传递给API函数而不是&data。显然也将data传递给ByteArrayToString。并且记得在完成后删除数组。

正如我在对这个问题的评论中所说,还有很多其他错误,我真的不想进入调试和纠正所有错误。我希望在评论中给你一个很好的引导,其余的由你决定。