C获取用户输入的安全且正确的方法[关闭]

Question

安全输入比我想象的要难。我环顾四周，找到了this，但它是从 2011 年开始的，从那时起情况可能已经发生了变化。

我正在寻找可以处理字符串输入和单字符输入的东西。例如：

Hello, what's your name?
My name is: _

Are you sure?
[Y] Yes, hello
[N] No, goodbye
_

这是我现在的做法：

char input[16];
fgets(input, 16, stdin);

char input = getchar();

我的问题一直是用户可能输入任意长的输入或者无效的输入。如何安全地读取输入并确保以后的输入不会被打扰？

寻找适用于 C 和 Linux/Windows 的解决方案。

感谢您的宝贵时间。

Answer 1

您可以安全地读取无限标准输入。您可以只使用不断重新分配的缓冲区。

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

char* get_line(size_t* length) {
    size_t len = 0;
    size_t cap = 4;
    char* buffer = malloc(cap);
    if (buffer == NULL) {
        fprintf(stderr, "Virtual memory exhausted\n");
        return NULL;
    }

    char current;
    while ((current = getchar()) != '\n') {
        if (len + 1 > cap) {
            cap *= 2;
            buffer = realloc(buffer, cap);
        }
        buffer[len++] = current;
    }
    // One last time for NUL terminator
    if (len + 1 > cap) {
        cap *= 2;
        buffer = realloc(buffer, cap);
    }        
    buffer[len] = 0;

    if (length) *length = len;
    return buffer;
}

此函数将逐个字符地读取stdin 并将其添加到返回的字符串中。您可以选择询问它的长度。别忘了free。

int main() {
    char* line = get_line(NULL);
    printf("You entered: %s\n", line);
    free(line);
}

注意：这段代码的意思是很少冗长，所以省略了一些检查。在生产代码中检查realloc 并使用int for getchar() 来测试EOF 是必要的。