检查输入和stderr

时间:2015-11-01 22:01:54

标签: c input printf stderr

我正在尝试编写将从stdin读取整数并打印其gcd的程序。如果两个整数都是素数,我打印"prime"。在结束程序之前打印到stderr - "DONE"。当用户输入错误数据时,我要打印到stderr - "ERROR"。所以我写了这段代码:

#include "stdio.h"
#include "nd.h"
#include "nsd.h"

int main() {
    int in1, in2;
    int tmp;
    while (1) {
        tmp = scanf("%d %d", &in1, &in2);
        if (tmp == EOF) {
            fprintf(stderr, "DONE\n");
            break;
        }
        if (tmp != 2) {
            fprintf(stderr, "ERROR\n");
        } else if ((nd(in1) == 1) && (nd(in2) == 1)) printf("prime\n");
                    // nd(int a) return 1 when a is prime  
                else printf("%d\n", gcd(in1, in2));
    }
    return 0;
}

我希望在"ERROR"之后继续工作。但它不起作用,我尝试在continue;之后添加fprintf(stderr, "ERROR\n");,但它也不起作用。所以,我想:

- program run
5 10
5
1 3 
prime
1 0.9
error
// not break here!
10 2 
2
...
//EOF
DONE
- program stop

一切正常,除了"ERROR",我有这个:

- program run
5 0.9
ERROR
ERROR
ERROR
ERROR
...
//never stop

我理解在循环周期中它是正确的工作。我的问题是我需要改变什么来改变我所拥有的'到了我想要的东西'。

1 个答案:

答案 0 :(得分:2)

scanf()在尝试应对意外输入时遇到问题。而是使用fgets()

读取
char buf[100];
if (fgets(buf, sizeof buf, stdin) == NULL) {
  fprintf(stderr, "DONE\n");
  break;
}
if (sscanf(buf, "%d%d", &in1, &in2) != 2) {
  fprintf(stderr, "ERROR\n");
} else if ((nd(in1) == 1) && (nd(in2) == 1)) {
  printf("prime\n");
} else {
  printf("%d\n", gcd(in1, in2));
}

修改了在线上寻找额外文字的代码。

// if (sscanf(buf, "%d%d", &in1, &in2) != 2) {
//   fprintf(stderr, "ERROR\n");
int n = 0;
if (sscanf(buf, "%d%d %n", &in1, &in2, &n) != 2 || buf[n]) {
  fprintf(stderr, "ERROR\n");
} ...