小编Eug*_*g G的帖子

我有一个程序可以从文件或键盘读取输入.我找到了一种方法来决定是从文件还是从键盘插入输入.

我需要以一种简单的方式实现该决定,因为我不想为每种类型的输入复制我的代码2次.

我尝试了类似下面的代码,但仅用于输入文件(arv == 1),而它从stdin获取分段错误.

调试arg == 0后出错

(_IO_fgets (buf=0xbffffaac "\030\031\023", n=100, fp=0x0) at iofgets.c:52)

码:

FILE *fp;

if (arg = 1)
    fp = fopen (operationfile, "r");
else
    fp = stdin;

while (fgets(buffer, sizeof(buffer), fp) != NULL)

有没有快速的方法来做到这一点？