c++ - 取消对 unique_ptr 的引用

Question

以下(不安全的)代码有效

#include <iostream>
#include <fstream>

std::ofstream* create_unsafe_stream(const char* filename) {
    return new std::ofstream {filename};
}

int main () {
    std::ofstream& u_out = *create_unsafe_stream("foo.txt");
    u_out << "foo\n" ;
    delete &u_out;
    return 0;
}

我试图制作一个更安全的版本

#include <iostream>
#include <fstream>
#include <memory>

using stream_ptr = std::unique_ptr<std::ostream>;

stream_ptr create_safe_stream(const char* filename) {
    return stream_ptr{ new std::ofstream {filename}};
}

int main() {
  std::ostream& s_out = *create_safe_stream("foo.txt");
  s_out << "foo\n" << std::endl; 
  return 0
}

编译但是，当我运行它时给我一个段错误。我在想问题是由 unique_ptr 超出范围引起的。所以我尝试将 main 修改为

int main() {
   stream_ptr ofile = create_safe_stream("foo.txt");
   std::ostream& s_out = *ofile;
   s_out << "foo\n"; 
}

再次起作用。

问题

有没有办法不使用像 ofile 这样的“中间”变量并在一行中完成所有操作？

编辑

函数create_safe_stream 是一个玩具模型what I want ，也就是说这个函数可能返回那个文件的 std::ofstream 或 std::cout，所以我想我真的需要返回一个指向 base 的指针类 std::ostream。我该怎么办？

Answer 1

你的假设是正确的。 create_safe_stream 返回一个 std::unique_ptr ，它立即超出范围，因此它持有的原始资源被 deleted 并尝试使用它是 UB .

不使用中间变量并在一行中完成所有操作的方法是只返回一个 std::ofstream 对象:

std::ofstream create_safe_stream(const char* filename) {
    return std::ofstream {filename};
}

int main() {
  std::ostream s_out = create_safe_stream("foo.txt");
  s_out << "foo\n" << std::endl; 
  return 0;
}