溢出缓冲区时如何防止fgets多次运行?

问题描述 投票:2回答:1

我有一个问题,如果用户输入的输入大于最大字符数,例如“abcdefg”,当max为5时,循环内的fgets()运行两次而不是“abcd”,而是打印出“abcd” “那么”efg“。

我认为这是因为fgets()只处理4个字符和一个空终止符,但stdin缓冲区中仍然存在“efg”。

我想知道是否有办法只抓取“abcd”并丢弃剩余的任何超过分配缓冲区最大大小的剩余输入。

#define INPUT_MAX 5
int main(int argc, char* argv[]){
    char input[INPUT_MAX];
    while(1){
        printf("prompt> ");
        fgets(input, INPUT_MAX, stdin);
        printf("\n%s\n", input);
    }
}

示例运行:

prompt> abcdefg   <-- I press enter once here for /n

abcd
prompt> 
efg

prompt> .         <-- I end up here after enter command

我发现fflush(stdin)不是冲洗stdin的正确方法。

c
1个回答
3
投票

没有什么是自动的。检查输入是否以换行结束。如果没有,请在循环中调用getchar(),直到获得换行符或EOF。

while(1){
    printf("prompt> ");
    if (fgets(input, INPUT_MAX, stdin) == NULL) {
        break;
    }
    printf("\n%s\n", input);
    if (input[strlen(input)-1] != '\n') {
        for (int c; (c = getchar()) != EOF && c != '\n';)
            ;
}
© www.soinside.com 2019 - 2024. All rights reserved.