获取文件数据后刷新cout的问题

时间:2012-04-10 00:35:29

标签: c++ file segmentation-fault buffer ifstream

我正在尝试编写rc4的实现。我正在使用ifstream从文件中读取纯文本。我注意到它没有在文件末尾输出,所以我尝试了各种方式明确清除缓冲区。无论哪种方式(使用endl,附加\ n,调用cout.flush())我尝试刷新缓冲区,我得到一个段错误。作为一个完整性检查,我用网上的一个例子替换了我的代码,我也分别进行了测试。它可以工作,如果我把它放在自己的文件中并编译它(例如,它打印出文件的内容,没有段错误,并且不需要任何调用flush()或endl这样做),但不在我的代码中。

以下是令人讨厌的代码(在我的代码之外工作正常;它几乎直接从cplusplus.com复制)

 ifstream is;
 is.open("plain");
 char c;
 while (is.good())     // loop while extraction from file is possible
 {
     c = is.get();       // get character from file
     if (is.good())
         cout << c;
//       cout.flush();
 }
 is.close();           // close file*/

以下是完整代码:(警告,许多注释掉的代码)

#include <iostream>
#include <fstream>
#include <string.h>
#include <vector>

using namespace std;
static char s[256], k[256];
//static char *i, *j;
void swap(int m, int n, char t[256]){
        char tmp = t[m];
        t[m] = t[n];
        t[n] = tmp;
}

char getByte(){
        static char i(0), j(0);
        i = (i+1)%256;
        j = (j + s[i])%256;
        swap(i, j, s);
        return s[(s[i]+s[j]) % 256];
}

int main(int argc, char ** argv){
        /*string key = argv[1];*/
        if(argc < 4){
                cout << "Usage: \n rc4 keyfile plaintextfile outputfile" << endl;
                return -1;
        }
        string key;
        ifstream keyfile (argv[1]);
        keyfile >> k;
        cout << "Key = " << k << endl;
        keyfile.close();
        /*ifstream plaintextf;
        plaintextf.open(argv[2]);*/

        ofstream ciphertextf (argv[3]);

        for(int q = 0; q < 256; q++){
                s[q] = q;
        }
        int i, j;
        for(int m = 0; m < 256; m++){
                j = (j + s[m] + k[m % sizeof(k)])%256;
                swap(m, j, s);
        }
//      vector<char> bytes(plaintext.begin(), plaintext.end());
//      bytes.push_back('\0');
//      vector<char>::iterator it = bytes.begin();
/*      char pt;
        while(plaintextf.good()){
                pt = plaintextf.get();
                if(plaintextf.good()){
                        cout << pt;

                      ciphertextf <<(char) (pt ^ getByte());
                }

        } */
        ifstream is;
        is.open("plain");
        char c;
         while (is.good())     // loop while extraction from file is possible
         {
            c = is.get();       // get character from file
            if (is.good())
              cout << c;
//              cout.flush();
         }
  is.close();           // close file*/

/*//    plaintextf.close();
        ciphertextf.close();
        keyfile.close();
        */
        return 0;
}

1 个答案:

答案 0 :(得分:0)

此外,我认为第二次调用is.good()[如if(is.good())]会阻止复制文件的最后一个字符。