为了避免缓冲区溢出,从终端读取输入时应当用fgets()代替gets()函数。
但是这也将带来一个问题,因为fgets()的调用格式是:
-
fgets (buf, MAX, fp)
-
fgets (buf, MAX, stdin)
buf是一个char数组的名称,MAX是字符串的最大长度,fp是FILE指针。
fgets()函数读取到它所遇到的第一个换行符的后面,或者读取比字符串的最大长度少一个的字符,或者读取到文件结尾。然后fgets()函数向末尾添加一个空字符以构成一个字符串。如果在达到字符最大数目之前读完一行,它将在字符串的空字符之前添加一个换行符以标识一行结束。
问题出在有时字符串的结尾处可能多出一个换行符,我们需要把它去掉。
-
#include <stdio.h>
-
#include <string.h>
-
#define LEN 5
-
int main()
-
{
-
char str[LEN];
-
fgets(str, LEN, stdin);
-
//fprintf(stderr, "%s %d\n", str, strlen(str));
-
for(int i = 0; i < LEN; i++)
-
printf("%d\t", str[i]);
-
printf("\n");
-
if(str[strlen(str)-1] == '\n')
-
str[strlen(str)-1] = '\0';
-
for(int i = 0; i < LEN; i++)
-
printf("%d\t", str[i]);
-
printf("\n");
-
return 0;
-
}
输入:
abc
输出:
-
97 98 99 10 0
-
97 98 99 0 0
说明,当输入的字符少于指定数目时,会将最后一个换行符保存在s[len-1]的位置,s[len]处恒为'\0'。
输入:
abcdefg
输出:
-
97 98 99 100 0
-
97 98 99 100 0
说明,当输入的字符大于指定数目时,保存指定字符串长度-1个字符,不保存换行符,s[len]处恒为'\0'。
用如下的函数来处理:
-
void clearEnter(char *p)
-
{
-
while(*p)
-
{
-
if(*p == '\n')
-
*p = '\0';
-
p++;
-
}
-
}
但该函数有个问题,在此处处理fgets()函数获得的字符串尾的换行是有效的。假如字符串中间不是从fgets()函数获得,而字符串中间有换行符,则会将字符串中第一个换行符替换为'\0',也就丢弃了字符串的其他部分。