C++自定义Open File Dialog如何解决硬编码文件名缓冲区不足问题

核心原理 #

当GetOpenFileName（或GetSaveFileName）执行时如果传入的缓冲区不足以存储选中的所有文件名，会返回FALSE，此时调用CommDlgExtendedError()会返回错误码FNERR_BUFFERTOOSMALL，且OPENFILENAME结构的lpstrFile指向的缓冲区前2个字节会写入实际需要的缓冲区总字符数，我们可以基于该机制动态扩容后重新唤起对话框，无需硬编码固定缓冲区大小。

具体代码修改 #

1. 类定义修改 #

替换固定大小的栈数组为动态堆指针，删除硬编码的固定缓冲区大小：

class DLLEXPORT CustomOpenFileDialog
{
    ......
private:
    OPENFILENAME m_OpenFileName;
    long m_fileNameBufSize; // 改为非静态成员，动态记录当前缓冲区大小
    TCHAR* m_fileNameBuf;   // 改为指针，动态分配内存
    ......
public:
    CustomOpenFileDialog();
    ~CustomOpenFileDialog(); // 新增析构函数释放内存
    BOOL ShowDialog(); // 假设这是你弹出文件对话框的成员函数
    void SetFileName(TCHAR* name);
    ......
};

2. 构造、析构函数实现 #

初始化动态缓冲区，析构时释放内存：

CustomOpenFileDialog::CustomOpenFileDialog()
{
    .....
    // 初始分配合理大小的缓冲区，可按需设置初始值
    m_fileNameBufSize = 5000;
    m_fileNameBuf = new TCHAR[m_fileNameBufSize];
    ZeroMemory(m_fileNameBuf, m_fileNameBufSize * sizeof(TCHAR));

    m_OpenFileName.lpstrFile = m_fileNameBuf;
    m_OpenFileName.nMaxFile = m_fileNameBufSize;
}

CustomOpenFileDialog::~CustomOpenFileDialog()
{
    delete[] m_fileNameBuf;
    m_fileNameBuf = nullptr;
}

3. SetFileName函数适配 #

void CustomOpenFileDialog::SetFileName(TCHAR* name)
{
    _tcsncpy_s(m_fileNameBuf, m_fileNameBufSize, name, _TRUNCATE);
    .....
}

4. 对话框弹出函数增加动态扩容逻辑 #

BOOL CustomOpenFileDialog::ShowDialog()
{
    BOOL result;
    while (TRUE)
    {
        result = GetOpenFileName(&m_OpenFileName);
        if (result)
        {
            // 选择成功，退出循环
            break;
        }
        
        DWORD err = CommDlgExtendedError();
        if (err == FNERR_BUFFERTOOSMALL)
        {
            // 缓冲区不足，读取需要的大小
            long requiredSize = *(reinterpret_cast<WORD*>(m_fileNameBuf));
            // 扩容缓冲区
            delete[] m_fileNameBuf;
            m_fileNameBufSize = requiredSize;
            m_fileNameBuf = new TCHAR[m_fileNameBufSize];
            ZeroMemory(m_fileNameBuf, m_fileNameBufSize * sizeof(TCHAR));
            // 更新OPENFILENAME结构的缓冲区参数
            m_OpenFileName.lpstrFile = m_fileNameBuf;
            m_OpenFileName.nMaxFile = m_fileNameBufSize;
            // 继续循环重新唤起对话框
        }
        else
        {
            // 其他错误，直接返回
            break;
        }
    }
    return result;
}

注意事项 #

• 若你的项目使用C++11及以上版本，可改用std::vector<TCHAR>管理缓冲区，无需手动管理内存，避免内存泄漏风险
• 多次扩容重新唤起对话框对用户完全无感知，不会出现对话框多次弹出的问题
• 若需要支持Windows XP及更早系统，需确保缓冲区大小不超过系统限制的最大可分配内存阈值

内容的提问来源于stack exchange，提问作者StackUser