从智能指针获取所有权，并通过相同的智能指针进一步访问原始指针

Question

我有一个这样的泄漏代码：

void f() {
  SomeClass *sc = new SomeClass;
  ...
  if (...) {
    g(sc); // g() takes ownership of memory pointed to by sc
  }
  ...
  sc->SomeMethod();
  ...
}

我想通过使用智能指针来消除泄漏，想到了以下解决方案：

void f() {
  smart_ptr<SomeClass> sc(new SomeClass);
  ...
  if (...) {
    g(sc.release()); // ownership is taken
  }
  ...
  sc->SomeMethod(); // sc points to the allocated memory regardless of whether ownership was taken or not
  ...
} // the memory is freed if ownership was not taken

尝试使用 std::unique_ptr 和 std::shared_ptr。 std::unique_ptr 指向释放后的空指针， std::shared_ptr 根本没有 release() 。手动增加 std::shared_ptr 的引用计数器会有所帮助，但据我所知，也没有这样的能力。

除了原始指针保持原样保留代码并且每次将原始指针传递给获得所有权的函数时调用 release() 之外，还考虑使用 std::unique_ptr，但这是一个非常混乱的解决方案。

我想知道标准库中是否有适合我目标的智能指针， 或者在我的情况下可能有一个理念或常见的技巧，或者我可能遗漏了什么？

Answer 1

您可以使用shared_ptr 获得相同的效果。 shared_ptr 没有 release() 函数，但它有 reset() 函数，我认为您正在寻找它。但是，您必须等到调用SomeClass::SomeMethod() 之后才能调用sc.reset()。

void g(shared_ptr<SomeClass> ptr);

void f()
{
   shared_ptr<SomeClass> sc(new SomeClass);

   if (/* check for condition */)
   {
      g(sc);
   }

   sc->SomeMethod();
   sc.reset();
}

Answer 2

我不明白您所说的 g 取得所有权是什么意思，因为在 g 退出后，您仍将使用指针。如果您总是想在g 之后使用指针，则将原始指针传递给g 并在块末尾释放指针会更有意义。

如果您确实希望g 获得所有权，并且希望仅在您的指针仍然存在时调用SomeMethod，请将原始指针传递给g 并让它返回一个原始指针。如果指针在g 中被释放，则返回nullptr 并在SomeMethod 调用之前检查它。

Answer 3

只需多一点代码，您就可以保留一个本地的、非拥有的视图：

void g(std::unique_ptr<Foo> p);     // takes ownership

int main()
{
    auto p = std::make_unique<Foo>();
    Foo & local_view = *p;

    if (/* condition */) { g(std::move(p)); }

    local_view.do_stuff();
}