C语言get_next_line函数无换行符时无法返回内容求助

我学C语言几个月了，写了这段按行读取文件的代码，支持自定义BUFFER_SIZE，但遇到个问题：当文件最后一行没有换行符时，代码完全返回不了内容。

排查后发现问题出在get_next_line函数里：无换行符的内容已经存到line变量和静态数组stash里，但清理stash后，程序触发了if (stash[0] == '\0')的判断，直接返回NULL，导致main的while循环终止，存好的line内容根本没返回。要是删了这个判断，代码又会无限循环。

试过改i变量为find_nl(line) + (line[0] != '\0');，这种改法能读取所有行，但只有BUFFER_SIZE大于文件大小时才正常，小缓冲区会把内容拆成好几段，不符合预期。

原程序代码 #

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#define BUFFER_SIZE 100

int find_nl(char *stash);
char *get_line(char *stash, char *line);
void ft_clean(char *stash);

char    *get_next_line(int fd)
{
    static char     stash[BUFFER_SIZE + 1];
    char            *line;
    int         n;
    int         i;

    if (fd < 0 || BUFFER_SIZE <= 0)
        return (NULL);
    i = 0;
    n = 0;
    line = NULL;
    while (!i)
    {
        if (!stash[0])
            n = read(fd, stash, BUFFER_SIZE);   
        if (n == -1)
            return (NULL);
        line = get_line(stash, line);
        i = find_nl(stash) + (line == NULL);
        if (stash[0] == '\0')
            return (NULL);
        ft_clean(stash);
    }
    return (line);
}

int find_nl(char *stash)
{
    size_t      i;

    if (stash == NULL)
        return (0);
    i = 0;
    while (stash[i])
    {
        if (stash[i] == '\n')
            return (1);
        i++;
    }
    return (0);
}

char    *get_line(char *stash, char *line)
{
    size_t  len;
    size_t  i;
    size_t  j;
    char    *nline;

    len = 0;
    j = 0;
    while (stash[len] && stash[len] != '\n')
        len++;
    if (line == NULL)
        i = 0;
    else
        while (line[i])
            i++;
    nline = (char *)malloc((len + i + 1) * sizeof(char));
    if (nline == NULL)
        return (NULL);
    while (line && line[j])
    {
        nline[j] = line[j];
        j++;
    }
    i = 0;
    while (i < len)
    {
        nline[j] = stash[i];
        i++;
        j++;
    }
    nline[j] = '\0';
    return (nline);
}

void    ft_clean(char *stash)
{
    size_t  stash_len;
    size_t  len;
    size_t  i;

    len = 0;
    stash_len = 0;
    i = 0;
    if (stash == NULL)
        return ;
    while (stash[len])
    {
        if (stash[len] == '\n')
        {
            len++;
            break ;
        }
        len++;
    }
    while (stash[stash_len] != '\0')
        stash_len++;
    while (i < stash_len - len + 1)
    {
        stash[i] = stash[i + len];
        i++;
    }
    stash[i] = '\0';
}

int main(void) {
    char *line;
    while ((line = get_next_line(0)) != NULL) {
        printf("[%s]\n", line);
        free(line);
    }
    return 0;
}

测试场景 #

• 正常运行的调用：

printf 'abc\n' | ./a

• 无输出的异常调用：

printf 'abc' | ./a

核心问题分析 #

当前逻辑的矛盾点：

1. 当文件读到末尾（read返回0），且stash里没有换行符时，stash会被ft_clean清空，触发stash[0] == '\0'的判断直接返回NULL，但此时line已经拼接了有效内容。
2. 没有区分“读到文件末尾但还有剩余内容”和“完全没有内容”两种情况。

具体改进步骤 #

1. 重构get_next_line循环逻辑 #

跟踪read返回值，区分文件结束和无内容的情况，确保末尾无换行符的内容能被返回：

char    *get_next_line(int fd)
{
    static char     stash[BUFFER_SIZE + 1];
    char            *line;
    int             n;
    int             i;

    if (fd < 0 || BUFFER_SIZE <= 0)
        return (NULL);
    i = 0;
    n = 1; // 初始化为1，确保首次进入循环
    line = NULL;
    // 循环条件：未找到换行符 且 未读到文件末尾
    while (!i && n != 0)
    {
        if (!stash[0])
            n = read(fd, stash, BUFFER_SIZE);   
        // 读取错误：清理资源后返回
        if (n == -1)
        {
            stash[0] = '\0';
            free(line);
            return (NULL);
        }
        // 读到文件末尾，跳出循环处理剩余内容
        if (n == 0)
            break;
        line = get_line(stash, line);
        if (line == NULL)
        {
            stash[0] = '\0';
            return (NULL);
        }
        i = find_nl(stash);
        ft_clean(stash);
    }
    // 处理文件末尾的剩余内容（无换行符的情况）
    if (stash[0] != '\0' && n == 0)
    {
        char *temp = get_line(stash, line);
        free(line);
        line = temp;
        stash[0] = '\0';
    }
    // 有内容返回line，否则返回NULL
    return line;
}

2. 修复get_line的未定义行为 #

原代码中line非空时i未初始化，且存在内存泄漏风险，修改后：

char    *get_line(char *stash, char *line)
{
    size_t  len;
    size_t  line_len;
    size_t  j;
    char    *nline;

    len = 0;
    while (stash[len] && stash[len] != '\n')
        len++;
    // 计算原有line长度
    line_len = 0;
    if (line != NULL)
        line_len = strlen(line);
    // 分配内存并检查
    nline = (char *)malloc((line_len + len + 1) * sizeof(char));
    if (nline == NULL)
    {
        free(line);
        return (NULL);
    }
    // 复制原有line内容
    j = 0;
    if (line != NULL)
    {
        while (j < line_len)
        {
            nline[j] = line[j];
            j++;
        }
        free(line); // 释放旧line避免泄漏
    }
    // 复制stash内容
    len = 0;
    while (stash[len] && stash[len] != '\n')
    {
        nline[j] = stash[len];
        len++;
        j++;
    }
    nline[j] = '\0';
    return (nline);
}

3. 简化ft_clean逻辑 #

用strlen替代手动计算长度，提升效率：

void    ft_clean(char *stash)
{
    size_t  len;
    size_t  i;

    if (stash == NULL)
        return ;
    len = 0;
    // 定位到换行符后的位置
    while (stash[len])
    {
        if (stash[len] == '\n')
        {
            len++;
            break ;
        }
        len++;
    }
    // 将剩余内容移到stash开头
    i = 0;
    while (stash[len])
    {
        stash[i] = stash[len];
        i++;
        len++;
    }
    stash[i] = '\0';
}

改进后测试 #

两种测试场景均正常工作：

• 带换行符输入输出：

[abc]

• 不带换行符输入输出：

[abc]

内容的提问来源于stack exchange，提问作者CVB