与 fstream::read 不一致的结果

Question

我的程序使用小型 SQLite3 数据库。为了确保在程序启动时它确实存在，我在一个文件中有一个数据库创建脚本，它会被执行。

脚本运行没有问题。

但是，当使用 C++ I/O 函数从该文件中读取数据时，我经常在文件末尾收到无效字符，这导致脚本包含错误并且 SQLite 库无法正确执行。这是显示缓冲区内容时的示例： // Proper content from the file, then a random character is there 1 Error executing request: near "1": syntax error

其他字符也会出现，空格、数字、字母... 这是我加载脚本的代码：

std::cerr << "Creating database if needed...\n";
char sql_script[] = "/path/to/script.sql";
int script_length;
bool result = false;
std::ifstream script_fs(sql_script, std::fstream::binary | std::fstream::in);

if (script_fs) {
    char* buffer;
    char** err_msg = NULL;
    script_fs.seekg(0, script_fs.end);
    script_length = script_fs.tellg();
    script_fs.seekg(0, script_fs.beg);

    buffer = new char[script_length];
    script_fs.read(buffer, script_length);

    std::cout << "sql:\n" << buffer << "\n";

    if (sqlite3_exec(m_db, buffer, NULL, NULL, err_msg) == SQLITE_OK){
        result = true;
    } else {
        std::cerr << "Error executing: " << sqlite3_errmsg(m_db) << "\n" << err_msg << "\n";
    }
    delete buffer;
    script_fs.close();
} else {
    std::cerr << "Error opening script: " << strerror(errno) << "\n";
}

return result;
}

为什么会发生这种情况，我该如何解决？

Answer 1

您需要确保您有一个以 null 结尾的字符串。

1. 为多一个字符分配内存。
2. 将空字符分配给buffer 的最后一个元素。

---

buffer = new char[script_length+1];
script_fs.read(buffer, script_length);
buffer[script_length] = '\0';

---

另外，使用delete的数组形式。

delete [] buffer;

Answer 2

不要混合使用 C 和 C++，如果您想使用 ifstream 在 C++ 中读取 sql 查询文件，那么下面的 C++ 代码可能是您不需要管理内存的一种方法，请注意分配一个'\0' 等的额外字符：

#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
int main() {
  ifstream fin("test.sql", std::fstream::binary | std::fstream::in);
  std::string sqlquery = std::string(std::istreambuf_iterator<char>(fin), std::istreambuf_iterator<char>());
  std::cout<<sqlquery<<std::endl;
  return 0;
}