如何在lambda中捕获智能指针?

时间:2013-12-19 07:07:50

标签: c++ smart-pointers lambda

在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所示。

2 个答案:

答案 0 :(得分:5)

lambda(即使是具有[=]等通用捕获的lambda)实际上只捕获其定义中使用的变量。因为在你的例子中,pointer从未在lambda中使用过,所以它没有被捕获,因此当它超出范围时,它是指向stored的最后一个共享指针, free()被称为。{/ p>

如果要捕获pointer,可以强制使用它:

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

然而,这是相当hackish。您希望您的仿函数具有状态并且具有非常重要的状态管理。对我来说,这是一个强有力的指标,实际的命名类(而不是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));
}

答案 1 :(得分:3)

只需按值捕获变量,让复制构造函数和析构函数担心所有权语义 - 这就是智能指针的用途。