Блог

Как я могу заставить этот код использовать библиотеку строк boost C или стандартную библиотеку, чтобы избежать wchar_t определения размера и получить более динамичную строку, которую можно обрабатывать намного проще? Этот код использует MFC CString , но я бы предпочел вместо этого использовать стандартную библиотеку или boost.

 TCHAR       drive[_MAX_DRIVE];
TCHAR       dir[_MAX_DIR];
TCHAR       fname[_MAX_FNAME];
TCHAR       ext[_MAX_EXT];
CString     cstr;

GetModuleFileName(NULL,cstr.GetBuffer(MAX_PATH),MAX_PATH);
cstr.ReleaseBuffer();

_wsplitpath_s(cstr,drive,dir,fname,ext);
cstr=drive;
cstr =dir;
cstr =_T("\myfile.dat");
 

Вам следует взглянуть на стандартную <filesystem> библиотеку C , в частности на ее path класс, в котором есть replace_filename() метод, например:

 #include <filesystem>
#include <windows.h>

WCHAR szFileName[MAX_PATH] = {};
GetModuleFileNameW(NULL, szFileName, MAX_PATH);
std:wstring str = std::filesystem::path(szFileName).replace_filename(L"myfile.dat").wstring();
 

Или, поскольку вы все равно используете Win32 API, вы можете просто использовать PathRemoveFileSpec() и PathAppend() вместо этого (или их более безопасные Cch аналоги), например:

 #include <windows.h>

WCHAR szFileName[MAX_PATH] = {};
GetModuleFileNameW(NULL, szFileName, MAX_PATH);
PathRemoveFileSpecW(szFileName);
PathAppendW(szFileName, L"myfile.dat");
std:wstring str = szFileName;
 

Если вы действительно хотите избежать MAX_PATH ограничения, вам придется вызывать GetModuleFileName() цикл, увеличивая размер буфера на каждой итерации, пока он, наконец, не завершится успешно:

 std::wstring wFileName(MAX_PATH, L'');
do {
    DWORD dwSize = GetModuleFileNameW(NULL, wFileName.data(), wFileName.size()); // or amp;wFileName[0] before C  17
    if (dwSize < wFileName.size()) {
        wFileName.resize(dwSize);
        break;
    }
    wFileName.resize(wFileName.size() * 2);
}
while (true);
// use wFileName as needed...
 

Этот маленький помощник работает всякий раз, когда вам приходится иметь дело с API, которые содержат параметры возврата по ссылке в стиле C, но вы хотите использовать исключительно типы, следующие за RAI.

 #include <cstddef>
#include <string>
#include <iostream>

/**
 * Wrap an instance of class S for C-style return-by-reference
 * with base-type T and maximum length N.
 *
 * Use is required when S only supports modification by methods,
 * and direct modification of internal buffer is forbidden.
 */
template<class S, class T, size_t N = 1>
class ref_param
{
public:
    ref_param() = delete;
    ref_param(const ref_paramamp;) = delete;
    ref_param(ref_paramamp;amp;) = delete;
    
    ref_param(Samp; ref) : _ref(ref), _storage{} {}
    ~ref_param()
    {
        _ref = _storage;
    }
    
    operator T*()
    {
        return _storage;
    }
    
private:
    Samp; _ref;
    T _storage[N];
};

// Specialization for single value
template<class S, class T>
class ref_param<S, T, 1>
{
    public:
    ref_param() = delete;
    ref_param(const ref_paramamp;) = delete;
    ref_param(ref_paramamp;amp;) = delete;
    
    ref_param(Samp; ref) : _ref(ref), _storage{} {}
    ~ref_param()
    {
        _ref = _storage;
    }
    
    operator T*()
    {
        return amp;_storage;
    }
    
private:
    Samp; _ref;
    T _storage;
};


extern "C"
{
    // example function with C style return-by-reference
    void foo(char* ret, size_t size)
    {
        for(size_t i = 0; i < size - 1;   i)
        {
            ret[i] = i   'a';
        }
        ret[size - 1] = 0;
    }
}

int main()
{
    std::string bar;
    foo(ref_param<std::string, char, 32>(bar), 32);
    std::cout << bar;
    return 0;
}
 

Этот шаблон работает и с другими типами, такими как ComPtr s, по сути, все, что имеет оператор присваивания на стороне C , просто работает.