[转] 克服C语言gets函数缺陷的方法初探

转自：http://blog.csdn.net/wangzhicheng1983/article/details/7780508

克服C语言gets函数缺陷的方法初探  

一 gets函数简介  

gets函数是C语言用于从stdin流中读取字符串的函数。gets函数接收用户从键盘上输入的字符直到遇到回车时终止。原型是：  

char* gets(char *buff);  

举例：  

#include <stdio.h>  
int main()  
{  
	char str[30];  

	while (!str != gets(str));  

	printf(“%s\n”,str);  

	return 0;  
}  

如果读取成功，gets函数返回值是和str的值相同的指针，否则，返回空指针。  

二 gets函数的漏洞  

gets函数是个危险的函数，原因是用户在键盘上输入的字符个数可能大于buf的最大界限，而gets函数并不对其检查。如在上例中，如果用户在键盘上输入1111111111111111…………….输入1的个数大于30，这时会造成程序崩溃。  

三 gets函数漏洞的解决方案  

我们可以重写一个新的函数Gets  

原型是char* Gets(int maxlen)  

这个函数让程序员指定一个最多输入字符的个数，在函数中为字符分配存储空间，函数返回char*  

这个函数是针对动态数组而做的，如  

int main()  
{  

  	char* p;  

	p = Gets(18);  

}  

Gets函数的参数之所以舍弃了传入的指针，是因为传入函数的指针可能不可靠，会造成程序崩溃，比如传入一个野指针。  

另一个Gets函数原型是  

char* const Gets(char* const array, int maxlen);  

这个函数针对固定长度字符数组的输入，如  

int main()  
{  

  	char str[20];  

	Gets(str, 20);  

  	return 0;  

}  

此时Gets函数参数之一之所以是char* const类型，是因为我们容许程序员修改这个类型指针所指向的内容，而不能修改指针本身，这和char str[30]中的str[0]=’1’，但不能str++效果一样。  

具体实现代码如下：  

#include <string.h>  
#include <stdio.h>  
#include <conio.h> 　  
#include <stdlib.h>　  
#include <io.h>  

char* Gets(int maxlen)  //最多从键盘上读入maxlen个字符,返回字符串指针类型  
{  
  	int i;  
  	static char* str;  
  	char c;  
  	str = (char*)malloc(sizeof(char)*maxlen);  

  	if(!str)  
	{  
		perror("memeoryallocation error!\n");  
       		return0;  
	}  
	else  
  	{  
		for(i=0; i<maxlen; i++) 
       		{  
             		c = getchar();  
             		if (c != '\n') str[i]=c;  
             		else
             			break;  
       		}  
       		str[i] = '\0';  
       		
       		return str;  
  	}  
}  

char* const Gets(char* const array,int maxlen)  
{  
  	int i;  
  	char c;  

  	for(i=0;i<maxlen;i++)  
  	{  
       		c=getchar();  
       		if(c!='\n')
       			array[i]=c;  
       		else
       			break;  
	} 
  	array[i]='\0';  
  	return array;  
}  

int main()  
{  
  	char s[8];  
  	Gets(s,8);  
  	puts(s);  
	fflush(stdin); //刷新输入缓冲区,很重要,否则会影响下一个Gets函数  
	char*p=Gets(8);  
  	puts(p);  
  
  	return 0;  
}