首页 > 解决方案 > 如何将 GetProfileBinary 的结果保存到智能指针中?

问题描述

目前我在一个类中有以下成员变量:

BYTE *m_pbyImportColumnMappings;

在其中一个类中,我们尝试从注册表中读取现有数据,如果不存在,我们分配它。到目前为止,我已将其更改为:

void CImportOCLMAssignmentHistoryDlg::ReadSettings()
{
    UINT uSize;

    m_dwImportFlags = theApp.GetNumberSetting(theApp.GetActiveScheduleSection(_T("Options")),
        _T("ImportFlags"), ImportAssignment::None);

    theApp.GetProfileBinary(theApp.GetActiveScheduleSection(_T("Options")), 
        _T("ImportColumnMappings"), (LPBYTE*)&m_pbyImportColumnMappings, &uSize);
    // Reset memory buffer (if required)
    if (uSize != (sizeof(BYTE) * 15))
    {
        if (uSize > 0)
        {
            delete[] m_pbyImportColumnMappings;
            m_pbyImportColumnMappings = nullptr;
        }

        m_pbyImportColumnMappings = new BYTE[15];
        // Default values
        const gsl::span column_mappings(m_pbyImportColumnMappings, 15);
        std::fill(begin(column_mappings), end(column_mappings), -1);
        /*
        m_pbyImportColumnMappings[0] = -1;
        m_pbyImportColumnMappings[1] = -1;
        m_pbyImportColumnMappings[2] = -1;
        m_pbyImportColumnMappings[3] = -1;
        m_pbyImportColumnMappings[4] = -1;
        m_pbyImportColumnMappings[5] = -1;
        m_pbyImportColumnMappings[6] = -1;
        m_pbyImportColumnMappings[7] = -1;
        m_pbyImportColumnMappings[8] = -1;
        m_pbyImportColumnMappings[9] = -1;
        m_pbyImportColumnMappings[10] = -1;
        m_pbyImportColumnMappings[11] = -1;
        m_pbyImportColumnMappings[12] = -1;
        m_pbyImportColumnMappings[13] = -1;
        m_pbyImportColumnMappings[14] = -1;
        */

    }
}

我最初的改变是使用 agsl::span来抑制关于使用指针算术的几个警告。但我不知道如何m_pbyImportColumnMappings变成智能指针,因为我们最初试图从GetProfileBinary.

如果我可以把它变成一个智能指针,那么当类超出范围时我不需要释放内存。

在相关答案中,建议使用此代码:

theApp.GetProfileBinary(strSection, strEntry,
        reinterpret_cast<LPBYTE*>(&pALI), &uBytesRead);
std::unique_ptr<BYTE[]> cleanup(reinterpret_cast<BYTE*>(pALI)); 

cleanup但是,鉴于我们正在处理类的成员变量而不是函数中的孤立变量,我不确定如何应用该方法。

标签: visual-c++mfcsmart-pointerscode-analysis

解决方案


要获得更简洁的代码,请考虑使用std::vector和 临时缓冲区

std::vector<BYTE> m_mapping;
m_mapping.resize(15, -1);
...

UINT len = 0;
BYTE* temp = nullptr;
AfxGetApp()->GetProfileBinary(_T("setting"), _T("key"), &temp, &len);
std::unique_ptr<BYTE[]> cleanup(temp);
if (len == m_mapping.size() * sizeof(m_mapping[0]))
    memcpy(m_mapping.data(), temp, len);
else
    std::fill(m_mapping.begin(), m_mapping.end(), -1);

std::vector还具有自动清理和附加方法。


否则,使用此成员数据std::unique_ptr替换new/delete,可能会有点噩梦。例子:

m_mapping = nullptr;
GetProfileBinary("setting", "key", &m_mapping, &uSize);
if (uSize != (sizeof(BYTE) * 15))
{
    { std::unique_ptr<BYTE[]> cleanup(m_mapping); }
    //delete memory immediately after exiting scope
    //note the extra brackets

    //allocate new memory and don't manage it anymore
    m_mapping = std::make_unique<BYTE[]>(15).release();
    if(m_mapping)
        for (int i = 0; i < 15; i++) m_mapping[i] = -1;
}

在这里,我们无法利用std::unique_ptr内存管理,它仅用于关闭警告。

您在这里不需要任何强制转换,因为它只是发生m_pbyImportColumnMappings并且BYTE期望GetProfileBinaryBYTE使用分配内存new BYTE


推荐阅读