写在前边
关于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;
}
}
}