fstream的问题读取正常,但不能写入用户输入字符串

时间:2017-06-02 10:56:17

标签: c++ fstream

我在代码中找不到问题。 readFile函数运行良好,但writeFile函数不会对文件进行任何更改:

#include <iostream>
#include <fstream>

using namespace std;
const int BUF_SIZE = 1024;

void readFile(fstream &file, char buffer[BUF_SIZE]);
void writeFile(fstream &file);

void readFile(fstream &file, char buffer[BUF_SIZE])
{
    int position;
    cout << "Please enter a position to read from the file some info" << endl;
    cin >> position;
    file.seekg(position, ios::beg);
    file.read((char*) buffer, BUF_SIZE); // <<<<<

    for(int i = 0; i < file.gcount(); i++){
        cout << buffer[i];
    }
}

void writeFile(fstream &file)
{
    char temp[100] = "HHHH";
    //cout << "Please enter some info to add to the file" << endl;
    file.write((char*) &temp, 100);
    for(int i = 0; i < file.gcount(); i++){
        cout << temp[i];
    }
}

int main(int argc, char const *argv[])
{
    char buffer[BUF_SIZE];

    if (argc != 2){
        cout << "Program usage: prog_name file_name";
        return 1;
    }

    fstream file(argv[1], ios::in | ios::out | ios::binary | ios::app);
    if (!file){
        cout << "File can not open or doesn't exist";
        return 1;
    }

    //Try to read & write some info from/to file in particular position

    readFile(file, buffer);
    writeFile(file);

    file.close();

    return 0;
}

当我创建新的ostream时效果很好,但我想了解为什么fstream进/出模式在我的代码中只能用于阅读。

1 个答案:

答案 0 :(得分:3)

我看到了几个问题:

  1. 写入问题背后的原因可能是因为你到达文件的末尾(文件是否小于BUF_SIZE字节?)。这将设置EOF位,这会使任何写操作失败。您必须先清除该位(使用std::fstream::clear方法):

    void readFile(fstream &file, char buffer[BUF_SIZE])
    {
        int position;
        cout << "Please enter a position to read from the file some info" << endl;
        cin >> position;
        file.seekg(position, ios::beg);
        file.read(buffer, BUF_SIZE);
    
        for(int i = 0; i < file.gcount(); i++){
            cout << buffer[i];
        }
    
        file.clear(); // clears EOF
    }
    
  2. file.write((char*) &temp, 100);是错误的,因为您实际上是将一个点传递给temp变量,该变量也是一个指针,但它被演员伪装。这些可以:file.write(temp, 100);file.write(&temp[0], 100);

  3. 打印书写字符时,您使用的是std::fstream::gcount,字面意思是获取计数(在上次获取操作中读取的字符数)。你正在写( put )没有阅读( get )。实际上,您实际上是在指示您愿意写多少字节,因此请使用它:

    file.write(temp, 100);
    for(int i = 0; i < 100; i++){
        cout << temp[i];
    }
    
  4. 最后,你总是写100个字符,可能包括来自缓冲区的一些垃圾。我看到你想让用户选择要写的内容(注释行),你可以改为:

    const size_t size = strlen(temp);
    file.write(temp, size);
    for(size_t i = 0; i < size; i++){
        cout << temp[i];
    }
    
  5. 此外,还有一些建议:

    1. 使用std::string读取用户输入,这样可以避免可能的缓冲区溢出(如果用户输入超过100个字符)。

      cin.ignore(numeric_limits<streamsize>::max(), '\n'); // read the link bel
      string temp;
      getline(cin, temp); // need #include <string>
      file.write(temp.c_str(), temp.size());
      

      您可能希望阅读this answer以了解有关第一行的更多信息(基本上它避免了在使用getline后跳过cin >> position)。

    2. 避免使用for循环打印用户输入。对于缓冲区和std::string选项,您只需cout << temp << endl;