Chinaunix

标题: 帮忙看看这个getLine函数 [打印本页]
作者: iwinux    时间: 2006-08-02 10:06
标题: 帮忙看看这个getLine函数
因为刚刚学 C 语言，经常做练习的时候都需要用到getchar()之类的函数，为了方便，我参考《C语言的科学和艺术》这本书的代码，实现了一个getLine函数，这里我把代码贴出来，大家帮忙看看有什么bug，谢谢！



  2. #define BUFSIZE 255
  3. /*reference:The Art and Science of C*/
  4. int getLine(char ** line) {
  5.         char * tmp;
  6.         int c, n = 0, size = BUFSIZE;

  8.         if (*line) {
  9.                 free(*line);
  10.         }

  12.         *line  = (char *) malloc(size + 1);

  14.         if (!*line) {
  15.                 return 0;
  16.         }

  18.         while ((c = getchar()) != '\n' && c != EOF) {
  19.                 if (n == size) {
  20.                         size *= 2;
  21.                         tmp = (char *) malloc(size + 1);
  22.                         strncpy(tmp, *line, n);
  23.                         free(*line);
  24.                         *line = tmp;
  25.                 }
  26.                 *(*line + n) = c;
  27.                 n++;
  28.         }

  30.         if (n == 0 && c == EOF) {
  31.                 free(*line);
  32.                 return 0;
  33.         }
  34.        
  35.         *(*line + n)= '\0';
  36.         tmp = 0;

  38.         return n;
  39. }
复制代码
作者: flw10000    时间: 2006-08-02 11:11
原帖由 iwinux 于 2006-8-2 10:06 发表
因为刚刚学 C 语言，经常做练习的时候都需要用到getchar()之类的函数，为了方便，我参考《C语言的科学和艺术》这本书的代码，实现了一个getLine函数，这里我把代码贴出来，大家帮忙看看有什么bug，谢谢！


        if (*line) {
                free(*line);
        }
[c ...



这里还没有malloc就free，会有问题吧！
作者: flw10000    时间: 2006-08-02 11:18
明白你的意思了

如果调用时已经malloc了，要先free，重新malloc，因为没法知道调用者malloc了多大空间！
作者: MackedNice    时间: 2006-08-02 12:58
if (*line) {
                free(*line);
        }
有问题. line 只是一个指针的地址。这个指针所指的内存是分配在函数栈上还是分配在堆上或者可能这个指针什么也没有指向，
你都是不知道的。所以你这样做是错误的。
当然你可以判断一下指针的地址是不是有效
if(line == NULL)
     return -1;


其他的好象没有问题。写得不错。
-------------------------------------
我学c的时候还写不出这样代码。
作者: Bayweb    时间: 2006-08-02 13:00
原帖由 flw10000 于 2006-8-2 11:18 发表
明白你的意思了

如果调用时已经malloc了，要先free，重新malloc，因为没法知道调用者malloc了多大空间！

>>
>>
>>一般malloc之前都会检查一下是否为空，或者size是否为零。如果不是十分必要，
>>在函数的结尾务必free一下。
>>
>>
作者: flw10000    时间: 2006-08-02 18:03
原帖由 Bayweb 于 2006-8-2 13:00 发表

>>
>>
>>一般malloc之前都会检查一下是否为空，或者size是否为零。如果不是十分必要，
>>在函数的结尾务必free一下。
>>
>>


咋俩的说法并不矛盾

我说的是楼主在程序开始处就判断*line如果为真就free，可能考虑的是并不知道该函数的调用者为*line申请了多大空间，所以free了，重新malloc了（size + 1）个内存空间！
作者: 默难    时间: 2006-08-02 18:26
LZ看以下几个函数：
fgets realloc
作者: yuxh    时间: 2006-08-02 18:29
没有大的问题.
但在最后

  2.         if (n == 0 && c == EOF) {
  3.                 free(*line);
  4.                 return 0;
  5.         }
复制代码

如果不加上一句*line = NULL;的话,再次调入函数会重复free,程序会core哦.
作者: flw    时间: 2006-08-02 18:36
man 3 getline
man 3 fgets
man 3 realloc
作者: iwinux    时间: 2006-08-02 21:45
Google 了一下
这篇文章说realloc有潜在的隐患
http://www.csdn.net/develop/article/27/27950.shtm

不过C-FAQ又说：
Since you always want each block of dynamically-allocated memory to be contiguous (so that you can treat it as if it were an array), you and realloc have to worry about the case where realloc can't make the old block of memory bigger ``in place,'' but rather has to relocate it elsewhere in order to find enough contiguous space for the new requested size. realloc does this by returning a new pointer. If realloc was able to make the old block of memory bigger, it returns the same pointer. If realloc has to go elsewhere to get enough contiguous memory, it returns a pointer to the new memory, after copying your old data there. (In this case, after it makes the copy, it frees the old block.) Finally, if realloc can't find enough memory to satisfy the new request at all, it returns a null pointer. Therefore, you usually don't want to overwrite your old pointer with realloc's return value until you've tested it to make sure it's not a null pointer. You might use code like this:


看起来很安全 -____-

[ 本帖最后由 iwinux 于 2006-8-2 21:57 编辑 ]
作者: iwinux    时间: 2006-08-02 22:46
修改后的代码


  2. /*reference:The Art and Science of C*/
  3. int readLine(FILE * infile, char ** line) {
  4.         char * tmp, *test;
  5.         int c, n = 0, size = BUFSIZE;

  7.         if (*line) {
  8.                 free(*line);
  9.         }

  11.         *line  = (char *) malloc(size + 1);

  13.         if (!*line) {
  14.                 return 0;
  15.         }

  17.         while ((c = fgetc(infile)) != '\n' && c != EOF) {
  18.                 if (n == size) {
  19.                         size *= 2;
  20.                         //tmp = (char *) malloc(size + 1);
  21.                         tmp = (char *) realloc(*line, size + 1);
  22.                        
  23.                         if (!tmp) {
  24.                                 //line = 0;
  25.                                 //return 0;
  26.                                 return n;
  27.                         }
  28.                        
  29.                         //strncpy(tmp, *line, n);
  30.                         //free(*line);
  31.                         test = *line;
  32.                         *line = tmp;
  33.                 }
  34.                 *(*line + n) = c;
  35.                 n++;
  36.         }

  38.         if (n == 0 && c == EOF) {
  39.                 *line = NULL;
  40.                 free(*line);
  41.                 return 0;
  42.         }
  43.        
  44.         *(*line + n)= '\0';
  45.         tmp = 0;

  47.         return n;
  48. }
复制代码




欢迎光临 Chinaunix (http://bbs.chinaunix.net/) Powered by Discuz! X3.2