C++ 合并 2 wchar_t*

Question

我有一个有趣的问题，我不知道为什么。

inline void mergeWChar(wchar_t*& dest, const wchar_t* source) {
    if (dest == nullptr) {
        dest = const_cast<wchar_t*>(source);
        return;
    }
    int size = wcslen(dest);
    wcscat_s(dest, size, source);
}
inline void test() {
    wchar_t* msg = nullptr;
    wchar_t* t = L"BLA.BLA.BLA";
    mergeWChar(msg, L"Ts is jut a phrase in order to test my function");
    mergeWChar(msg, t);
}

我正在尝试使用 wcscat_s 函数创建一个将合并更严格的 2 wchar_t* 的函数，但由于某种原因无法正常工作，并且我收到了此错误消息。

在 ConsoleApplication1.exe 中的 0x5D8C3A18 (ucrtbased.dll) 处引发异常：0xC0000005：访问冲突写入位置 0x0082DD60。

另外，我已经检查了这两个问题，并没有真正帮助我。

wcscat_s problem

wcscat_s function - buffer error

我不能做什么

我正在使用一些遗留代码，我可以不使用以下代码：

1. 字符串
2. to_wstring
3. wchar_t var[size]

Answer 1

这里：

mergeWChar(msg, t);

msg 指向字符串文字L"Ts is jut a phrase in order to test my function"。您不能修改字符串文字。

你想要这个：

inline void mergeWChar(wchar_t*& dest, const wchar_t* source) {
  if (dest == nullptr) {
    dest = const_cast<wchar_t*>(source);
    return;
  }

  wchar_t *newdest = (wchar_t*)malloc((wcslen(dest) + wcslen(source) + 1) * sizeof *newdest);
  wcscpy(newdest, dest);      // the *_s functions are useless here
  wcscat(newdest, source);    // as you know that newdest point to
                              // a long enough buffer
  dest = newdest;
}

免责声明：为简洁起见，此处未进行错误检查。

问题的根源在于 mergeWChar 函数的糟糕设计和 const_cast<wchar_t*> 的可疑使用，它基本上允许您将指向常量字符串的指针分配给普通指针，然后允许您修改它最终失败的持续记忆。

但是这种设计存在一个问题：您无法释放分配的内存，因为dest 指针不一定是通过malloc 获得的指针。因此，以下设计是更好的 IMO：

inline void mergeWChar(wchar_t*& dest, const wchar_t* source) {
  int size = (dest ? wcslen(dest) : 0) + wcslen(source) + 1;
  wchar_t *newdest = (wchar_t*)malloc(size * sizeof *newdest);
  if (dest)
    wcscpy(newdest, dest);
  else
    newdest[0] = 0;
  wcscat(newdest, source);
  dest = newdest;
}

这样就没有丑陋的强制转换，也没有指向常量内存的非 const 指针的风险，并且您可以确保无论如何都可以释放 mergeWChar 修改的指针。代价是您需要更多内存。

使用示例：

...
wchar_t* msg = nullptr;
wchar_t* t = L"BLA.BLA.BLA";
mergeWChar(msg, L"Ts is jut a phrase in order to test my function");
mergeWChar(msg, t);
...
free(msg);
...