c++ - 有没有办法检查此内存泄漏？

Question

我有一个非常基本的函数 bool read_binary( string filename, double*& buf, int &N) 从二进制文件中读取数据。我称之为:

int N;
double* A = NULL;
read_binfile(inputfile, A, N);
delete[] A;

read_binfile 在哪里

bool read_binfile( string fname, double*& buf, int &N ) { 
    ifstream finp(fname.c_str(), ifstream::binary);

    finp.seekg(0, finp.end);
    N = finp.tellg()/8;
    buf = new double[N];
    finp.seekg(0, finp.beg);
    printf("N = %i\n", N); 
    finp.read( reinterpret_cast<char*>(buf), sizeof(buf[0])*N);

    if( finp ) {
        return true;
    } else {
        return false;
    }
}

我发现如果我(天真地)循环 only read_binfile 那么这会导致内存泄漏，而如果我包含 double* A = NULL 和 delete[] A 在循环中，则不会发生此泄漏。有没有什么方法(在 read_binfile 内部)检查这种类型的错误，或者这是否需要正确使用 read_binfile 其中 delete[] 必须跟随而不被再次调用？

Answer 1

您可以使用 std::vector<double>对于 buf .这将使内存管理自动化。然后，您不需要传递 N作为一个论点。您可以从 vector 中获取缓冲区的大小.

bool read_binfile(std::string const& fname,
                  std::vector<double>& buf)
{
   std::ifstream finp(fname.c_str(), std::ifstream::binary);

   finp.seekg(0, finp.end);
   size_t N = finp.tellg()/sizeof(buf[0]);
   buf.resize(N);

   finp.seekg(0, finp.beg);
   finp.read(reinterpret_cast<char*>(buf.data()), sizeof(buf[0])*N);

   if( finp ) {
      return true;
   } else {
      return false;
   }
}

并将其用作:

std::vector<double> A;
read_binfile(inputfile, A);
size_t N = A.size(); // Only if needed.