这个函数只是简单地从标准输入中读入,再送回到标准输出,书中为了演示存储器的越界引用和缓冲区溢出,使用了gets函数。通常,使用gets或其他能导致存储溢出的函数,都是不好的编程习惯。当编译一个含有调用gets的文件时,C编译器甚至会产生这样的出错信息:“the gets function is dangerous and should not be used(gets函数很危险,不应使用)。”
一个好的echo函数的要求如下:它从标准输入读入一行,再写回到标准输出,函数的实现必须对任一长度的输入行都能正常工作,即使在输入行需要比你为缓冲区分配的空间更大的空间时,仍能正确工作。代码还应该检查出错条件,当遇到错误是返回。echo函数实现如下:
#include
#include
#include
#define N 10 //定义初始的字符数组长度
void echo()
{
printf("Enter a string: ");
int count = 0, i, c;
char* buf = (char*)malloc(N + 1); //数组长度N+1,最后一个存放'\0'
if(buf == NULL)
{
printf("Could not allocate memory...\n");
return;
}char* dest = buf;
while(1)
{
++count;
for(i = 0; i < N; ++i)
{
if( (c = getchar()) == '\n' || c == EOF)
break;
*buf++ = c;
}
if(i < N)
{
*buf = '\0';
break;
}
dest = (char*)realloc(dest, (count + 1) * N + 1); //重新分配的内存大小为原来基础上加N
if(dest == NULL)
{
printf("Could not allocate memory...\n");
return;
}
buf = dest + count * N; //计算字符存储的正确位置
}
printf("The string you input is: %s\n", dest);
free(dest);
}