IT学习联盟

 找回密码
 立即注册
搜索
热搜: 活动 交友 discuz
IT学习联盟 门户 文章 C/C++/VC++/MFC/QT 查看内容

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

2012-8-18 02:48| 发布者: admin| 查看: 572| 评论: 0

一 gets函数简介
gets函数是C语言用于从stdin流中读取字符串的函数。gets函数接收用户从键盘上输入的字符直到遇到回车时终止。原型是:
char* gets(char *buff);
举例:
#include <stdio.h>
int main()
{
      charstr[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()
{
      charstr[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;
      staticchar* 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;
                 elsebreak;
           }
           str[i]='\0';
           returnstr;
      }
}
 
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;
           elsebreak;
      }
      array[i]='\0';
      returnarray;
}
 
int main()
{
      char s[8];
      Gets(s,8);
      puts(s);
fflush(stdin); //刷新输入缓冲区,很重要,否则会影响下一个Gets函数
   char*p=Gets(8);
      puts(p);
      return 0;
}
Archiver|IT学习联盟| 网站地图

Powered by Discuz! X2.5© 2001-2010 世界学习室 大伟制作

GMT+8, 2018-2-25 09:54 , Processed in 0.092244 second(s), 12 queries .