C终端获取终端数据

写在前边

关于C语言从键盘获取数据，常用的有scanf，gets，getchar，fgets等等，但是scanf，gets，getchar等函数不会对输入的数据进行检查，会导致程序崩溃，所以一般都用fgets获取数据。

fgets问题

fgets(buf, n, stdin);
这个函数会从终端读取n-1个字符，会读取回车。若输入数据大小大于n-1，则再一次读取时，缓冲区数据会被接着读取。

看这个C语言菜单的例子：

char buf[2] = {0};
fgets(buf, 2, stdin);
switch (buf[0])
{
    case ‘a’: fun1(); break;
    case ‘b’: fun2(); break;
    case ‘c’: return 0;
}

分析一下，如果输入“abc”，那么，

第一次进入循环：buf[0] = ‘a’, buf[1] = ‘\0’;程序会执行fun1，
第二次进入循环：缓冲区数据被读取，buf[0] = ‘b’, buf[1] = ‘\0’;程序会执行fun2，
第三次进入循环：缓冲区数据被读取，buf[0] = ‘c’, buf[1] = ‘\0’;程序会执行return 0,程序直接退出。
最后，缓冲区数据’\n’被打印到终端

也就是说，我们输入“abc”，fun1，fun2，return 0，都被执行，但我们期望的是只执行fun1。
解决办法：我们可以先把第一位数据取出来，剩下的数据用临时变量读完，终止条件是’\n’，fgets读取的最后一位肯定是’\n’，所以用’\n’作为退出条件。

具体实现代码如下：

//是否存在'\n'，并将回车改为'\0'
int is_enter(char *p)
{
    while (*p != '\0')
    {
        if (*p == '\n')
        {
            *p = '\0';
            return 1;   
        }

        p++;
    }
    return 0;
}

//输入数据
void input(char *buf, int len)
{
    int flag = 0;
    while (1)
    {
        char temp[20] = {0};
        if (flag == 0)
        {
            flag = 1;
            fgets(buf, len, stdin);
            if (is_enter(buf) == 1)
                return;
        }
        else 
        {
            fgets(temp, sizeof(temp), stdin);
            if (is_enter(temp) == 1)
                return;
        }
    }
}