c++ - 如何在 lambda 中捕获智能指针？

Question

在 lambda 中捕获智能指针的最佳方法是什么？我的一次尝试导致了释放后使用错误。

示例代码:

#include <cstring>
#include <functional>
#include <memory>
#include <iostream>

std::function<const char *(const char *)> test(const char *input);

int main()
{
  std::cout.sync_with_stdio(false);
  std::function<const char *(const char *)> a = test("I love you");
  const char *c;
  while ((c = a(" "))){
    std::cout << c << std::endl;
  }
  return 0;
}
std::function<const char *(const char *)> test(const char *input)
{
  char* stored = strdup(input);
  char *tmpstorage = nullptr;
  std::shared_ptr<char> pointer = std::shared_ptr<char>(stored, free);
  return [=](const char * delim) mutable -> const char *
  {
    const char *b = strtok_r(stored, delim, &tmpstorage);
    stored = nullptr;
    return b;
  };
}

失败，如 AddressSanitizer 所示。

Answer 1

lambda(即使是像 [=] 这样具有通用捕获的 lambda)实际上只捕获其定义中使用的变量。由于在您的示例中，pointer 从未在 lambda 内部使用，因此不会被捕获，因此当它超出范围时，它是引用 stored 和 free() 被调用。

如果你想捕获指针，你可以强制使用它:

return [=](const char * delim) mutable -> const char *
{
  pointer;
  const char *b = strtok_r(stored, delim, &tmpstorage);
  stored = nullptr;
  return b;
};

然而，这是相当骇人听闻的。你希望你的仿函数有状态并且具有非平凡的状态管理。对我来说，这是一个强有力的指标，一个实际的命名类(而不是 lambda)将是有序的。所以我会这样改变它:

std::function<const char *(const char *)> test(const char *input)
{
  struct Tokenizer
  {
    std::shared_ptr<char> pointer;
    char* stored;
    char* tmpstorage;
    explicit Tokenizer(char* stored) : pointer(stored, free), stored(stored), tmpstorage(nullptr) {}
    const char* operator() (const char * delim)
    {
      const char *b = strtok_r(stored, delim, &tmpstorage);
      stored = nullptr;
      return b;
    }
  };
  return Tokenizer(strdup(input));
}