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

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

 

一 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;

}