为了避免缓冲区溢出,从终端读取输入时应当用fgets()代替gets()函数。
但是这也将带来一个问题,因为fgets()的调用格式是:
fgets (buf, MAX, fp)
fgets (buf, MAX, stdin)
buf是一个char数组的名称,MAX是字符串的最大长度,fp是FILE指针。
fgets()函数读取到它所遇到的第一个换行符的后面,或者读取比字符串的最大长度少一个的字符,或者读取到文件结尾。然后fgets()函数向末尾添加一个空字符以构成一个字符串。如果在达到字符最大数目之前读完一行,它将在字符串的空字符之前添加一个换行符以标识一行结束。
问题出在有时字符串的结尾处可能多出一个换行符,我们需要把它去掉。
#include <stdio.h>
2 #include <string.h>
3 #define LEN 5
4 int main() {
5 char str[LEN];
6 int i;
7 fgets(str, LEN, stdin);
8 printf("%s\n",str);
9 return 0;
10 }
输入一个:a
结果
a
fgets(str, LEN, stdin);
//fprintf(stderr, “%s %d\n”, str, strlen(str));