如何使用重试逻辑安全输入 int？

Question

我正在尝试写入一个程序，该程序从终端接收一个整数并安全地检查它是否真的是一个 int 以及它是否等于或介于 0 和 5 之间。代码还应该具有重试逻辑。

这是我写的代码：

#include <stdio.h>
#include <stdlib.h>


int main(){
    int input;
    char i[3];
    char *p;

    while(fgets(i, (sizeof(i)),stdin)){
        input=strtol(i,&p,10);
        if(input<0 || input>5 || p==i || (*p)!='\n'){
            printf("please enter a integer larger or equal to 0 or smaller or equal to 5\n");   
        }
        else{
            printf("%d",input);
            break;
        }
    }
}

我遇到的问题是，如果我输入的字符多于fgets() 读取的字符，例如“aaaaa”，我会得到输出：

please enter a integer larger or equal to 0 or smaller or equal to 5
please enter a integer larger or equal to 0 or smaller or equal to 5
please enter a integer larger or equal to 0 or smaller or equal to 5 

我假设问题是fgets 未读取的字符留在缓冲区中，并在触发 if 语句并因此重复输出的后续迭代中被读取。我假设解决方案是以某种方式清除缓冲区，但从我读过的fflush() 来看是不允许的。有没有一种简单的方法来清除缓冲区？如果这确实是问题的问题，你会怎么做？

Answer 1

简单的方法从更大的缓冲区开始。

#define BUF_N 100
char buf[BUF_N];

然后是一个不那么简单的方法来测试各种麻烦。我怀疑存在简化，但至少这个处理了许多角落并且是说明性的。

为简单起见，如果输入行过长，我们就说输入行不好，不管读取的文本如何。

  #define BUF_N 100
  char buf[BUF_N];

  while (fgets(buf, sizeof buf, stdin)) {
    // If entire buffer filled, the line may be longer, read it.
    if (strlen(buf) == BUF_N - 1 && buf[BUF_N - 2] != '\n') {
      printf("Excessive input\n");
      // Consume rest of the line 
      int ch;
      while ((ch = fgetc(stdin)) != '\n') {
        if (ch == EOF) {
          return EOF;
        }
      }
      continue;
    }

    char *p;
    long input = strtol(buf, &p, 10); // use long

    if (buf == p) {
      printf("Non-numeric input\n");
      continue;
    }

    if (input < 0 || input > 5) {
      printf("Out of range %ld\n", input);
      continue;
    }

    // Look for trailing junk
    while (isspace(*(unsigned char* )p)) {
      p++;
    }
    if (*p) {
      printf("Trailing junk\n");
      continue;
    }

    return (int) input;
  }
  return EOF;
}

上面对读取空字符不是那么宽容。

Answer 2

使用3 的缓冲区大小调用函数fgets 将最多读取一个字符，除了换行符和终止空字符。如果输入应该是介于0 和5 之间的数字，那么这种行为是可以接受的，但是您应该知道您的程序将无法读取诸如05 之类的输入，即使它满足范围要求。

如果你想允许输入如05，那么你应该增加缓冲区大小。

您可以轻松检测用户在行中输入的字符是否多于缓冲区。由于函数fgets 也读取换行符并将其存储到缓冲区中，因此您可以检查此换行符是否存在，例如使用函数strchr。如果您确定fgets 没有读取换行符，则这意味着输入太长而无法放入缓冲区。在这种情况下，您可能希望拒绝输入并丢弃输入的其余部分，以便下次调用 fgets 时不会读取它。这就是您的程序中发生的情况，导致同一行被多次处理。

丢弃剩余输入的一种方法是调用getchar，它一次读取一个字符。您应该重复此操作，直到遇到换行符，这意味着已到达行尾。

这就是代码的样子：

while( fgets( i, sizeof i, stdin ) != NULL )
{
    //verify that entire line of input was read
    if ( strchr( i, '\n' ) == NULL )
    {
        int c;

        //verify that stream state is as expected
        if ( feof(stdin) || ferror(stdin) )
        {
            fprintf( stderr, "unexpected input error!\n" );
            exit( EXIT_FAILURE );
        }

        //print error message
        printf( "Line input was too long to fit into buffer!\n" );

        //discard remainder of line
        do
        {
            c = getchar();

            if ( c == EOF )
            {
                fprintf( stderr, "unexpected input error!\n" );
                exit( EXIT_FAILURE );
            }

        } while ( c != '\n' );

        continue;
    }

    //input was successful, so we can now process it with strtol
}

请注意，您必须另外 #include <string.h> 才能使用函数 strchr。