我正在尝试编写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;
}
答案 0 :(得分:0)
此外,我认为第二次调用is.good()[如if(is.good())]会阻止复制文件的最后一个字符。