linux系统编程之仿标准库函数获取一行字符个数getline()

/*              getline()函数                        
函数:my_getline(char** buf,size_t *n,FILE* stream)
参数：
(out)buf:一个字符缓冲区buf指针
(out)n:分配内存的大小
(in)stream:文件流
返回值:
成功：返回读取字符的个数
失败：返回-1
*/

#include<stdio.h>
#include<stdlib.h>
#include<errno.h>
#include<string.h>
#define STEP_SIZE 120
size_t my_getline(char** line,size_t* n,FILE* stream)
{
    size_t res = 0;//接收返回值
    char* buf = *line;
    size_t count = 0;//统计字符大小
    //首先判断传入的buf是否有效

    //判断stream是否有效
    if(stream == NULL)
    {
        fprintf(stderr,"传入文件流无效\n");
        return -1;
    }
    //如果第一次进来 则分配空间
    if(*n == 0 || buf == NULL)
    {
        *line = malloc(STEP_SIZE);
        //对分配好的内存进行初始化操作
        memset(*line, 0, STEP_SIZE);
        *n = STEP_SIZE;
        buf = *line;
    }
    while((res = fgetc(stream)) != '\n')
    {
        if(res == EOF)
            return -1;
        //如果有足够空间 不包含'\0''\n'
        if(count < *n - 2)
        {
           // printf("调试7%d\n",count);
            *(buf + count) = res;
        }
        else
        {
            //内存不够用 重新分配空间
            *n = *n + STEP_SIZE;
            buf = realloc(buf,*n);
            *(buf + count) = res;
        }
        count++;
    }
    *(buf + count) = '\0';
    *(buf + count++) = '\n';
//    printf("*n = %zu\n",*n);
    return count;
}
int main(int argc,char* argv[])
{
    
    FILE* fp; 
    char* buf = NULL;
    size_t n = 0;
    size_t l = 0;
   // printf("调试1\n");
    if(argc < 2)
    {
     //   printf("调试2\n");
        fprintf(stderr,"请使用命令行参数传入文件路径\n");
        exit(1);
    }
   // printf("调试3\n");
   // printf("调试4---%s\n",argv[1]);
    fp = fopen(argv[1],"r");
   // printf("调试5\n");
    if(fp == NULL)
    {
        perror("fopen:");
        exit(1);
    }
    while( -1 != (l = my_getline(&buf,&n,fp)))
    {
        printf("size:%zu\n",n);
        printf("line:%zu\n",l);
    }
   // printf("%s\n",buf);
    free(buf);
    buf = NULL;
    fclose(fp);
   // printf("n = %zu\n",n);
}