如何從文件以及 std::cin 中讀取行？

Question

我正在編寫一個程序，它從文件中獲取要處理的文本行，用戶將其名稱作為參數傳遞，例如program <name of the file> 。 但如果未提供名稱，則從std::cin動態獲取輸入。 我試過的：

1. 重定向緩沖區（為什么會導致段錯誤）

if (argc == 2) {
    std::ifstream ifs(argv[1]);
    if (!ifs)
        std::cerr << "couldn't open " << argv[1] << " for reading" << '\n';
    std::cin.rdbuf(ifs.rdbuf());
}

for (;;) {
    std::string line;
    if (!std::getline(std::cin, line)) // Here the segfault happens 
        break;

2. 創建一個變量，其中存儲輸入源

std::ifstream ifs;
if (argc == 2) {
    ifs.open(argv[1]);
    if (!ifs)
        std::cerr << "couldn't open " << argv[1] << " for reading" << '\n';
} else
    ifs = std::cin;  // Doesn't work because of the different types

for (;;) {
    std::string line;
    if (!std::getline(ifs, line))
        break;

現在我正在考慮對文件結構/描述符做一些事情。 該怎么辦？

UPD：我希望有可能在程序的主循環中更新輸入源（見下文）。

Answer 1

第一個示例中的段錯誤是由於懸空指針引起的； 在您調用std::cin.rdbuf(ifs.rdbuf()) ， ifs被銷毀。 您應該按照@NathanOliver 的建議進行操作並編寫一個采用istream&的函數：

#include <iostream>
#include <fstream>
#include <string>

void foo(std::istream& stream) {
  std::string line;
  while (std::getline(stream, line)) {
    // do work
  }
}

int main(int argc, char* argv[]) {
  if (argc == 2) {
    std::ifstream file(argv[1]);
    foo(file);
  } else {
    foo(std::cin);
  }
}