C ++检测输入是否不满足条件

问题描述

更新：请通过明确的答案和用法的小示例指导我

我正在从 binary 文件中读取数据，其中第一个数字表示要读取的输入或值的数量，第二个数字表示第一个输入的长度，然后是输入本身，然后第二个输入的长度，然后是输入本身的长度，依此类推。

因此编写了以下代码：

    std::ifstream infile(filename,std::ios_base::binary);
    unsigned int NumberOfInputs;
    infile.read((char *) &NumberOfInputs,sizeof(unsigned int));

    for (unsigned int i = 0; i < NumberOfInputs; i++) {
        unsigned int Input_Lengh;
        infile.read((char *) &Input_Lengh,sizeof(unsigned int));
        string data;
        while (Input_Lengh) {
        char letter;
        infile.read((char *) &letter,sizeof(char));
        data += letter;
        Input_Lengh--;
        }
    }

但是，如果文件损坏并且用户告诉我输入的数量是10，而我只读了2个（例如，因为我进入了EOF）怎么办？

解决方法

来自文档。之std::istream::read()：

提取并存储字符，直到出现以下任何一种情况为止：

• 提取并存储了计数字符
• 文件条件结束出现在输入序列上（在这种情况下，称为setstate(failbit|eofbit)）。可以使用gcount()查询成功提取的字符数。

我做了一个小型的MCVE，以演示如何检查输入流的状态以及std::istream::gcount()的用法：

#include <iostream>
#include <fstream>
#include <vector>

int main(int argc,char **argv)
{
  std::ifstream infile(argv[1],std::ios_base::binary);
  unsigned n;
  if (!infile.read((char*)&n,sizeof n)) {
    std::cerr << "ERROR! Failed to read number of inputs.\n";
    return 1;
  }
  for (unsigned i = 0; i < n; ++i) {
    unsigned len;
    if (!infile.read((char*)&len,sizeof len)) {
      std::cerr << "ERROR! Failed to read length of input " << (i + 1) << ".\n";
      return 1;
    }
    std::vector<char> data(len);
    if (!infile.read(&data[0],len)) {
      std::cerr << "ERROR! Failed to read data of input " << (i + 1) << ".\n"
        << "Expected " << len << " bytes but got " << infile.gcount() << ".\n";
      return 1;
    }
  }
  std::cout << "File '" << argv[1] << "' successfully read.\n";
}

测试：

$ g++ -std=c++17 -O2 -Wall -pedantic -pthread main.cpp

$ echo "Good:" ; \
  echo -ne '\x01\x00\x00\x00\x04\x00\x00\x00''abcd' > test.dat ; \
  ./a.out test.dat
Good:
File 'test.dat' successfully read.

$ echo "Bad n:" ; \
  echo -ne '\x02\x00\x00\x00\x04\x00\x00\x00''abcd' > test.dat ; \
  ./a.out test.dat
Bad n:
ERROR! Failed to read length of input 2.

$ echo "Bad len:"; \
  echo -ne '\x01\x00\x00\x00\x04\x00\x00\x00''abc' > test.dat ; \
  ./a.out test.dat
Bad len:
ERROR! Failed to read data of input 1.
Expected 4 bytes but got 3.

Live Demo on coliru

注释：

• unsigned的大小取决于编译器/目标体系结构。 可能少于4个字节。
• unsigned的流行程度取决于平台。
  在此示例中，假定使用低端字节序，但仍然存在具有大端字节序的平台。
  （至少，在过去，平台具有更大的异国情调。）

要解决这两个问题，可以使用位算术运算符读取预期的字节数并将其组合为整数。

相关问题/答案：

• SO: C++ istream: Is gcount() always set after a read() even if it fails?
• SO: Ifstream reading error

,

使用std :: istream :: read（）时，您可以测试eofbit和failbit标志。当到达文件末尾时，都将同时设置它们。对于您而言，在每次读取操作之后，我建议您测试是否设置了这些标志。如果是这样，只需中断您的文件读取顺序即可。

例如：

if ( (infile.rdstate() & std::ifstream::failbit ) != 0 )
{
/* error handling here*/
}

或者，您可以使用gcount（）std :: istream方法获取成功读取的字符数。

例如：

if (infile.gcount() < N_EXPECTED_CHARS)
{
    /* error handling here*/
}