Реализация команды Unix HEAD в C терпит неудачу на больших строках

Question

Я в настоящее время внедряю Unix HEAD Команда с C и использование только системных функций. Пока что он отлично работает с файлами, длина строк которых меньше, чем та, которую я указал для моего buffer:

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#define LINES_TO_READ 10
#define BUFF_SIZE 4096

int main(int argc, char const *argv[]) {
    for (ssize_t i = 1; i < argc; i++) {
        const char *filename = argv[i];

        int fd = open(filename, O_RDONLY);

        if (fd < 0) {
            perror("open");
            return -1;
        }

        char ch, buffer[BUFF_SIZE];
        size_t index = 0, lines = 1;
        ssize_t rresult, wresult;

        // Read the file byte by byte
        while ((rresult = read(fd, &ch, 1)) != 0) {
            if (rresult < 0) {
                perror("read");
                return -1;
            }

            // Check if the current character is a new line (the line ends here)
            if (ch == '\n') {
                buffer[index] = ch;
                buffer[index + 1] = '\0';
                ch = 0;
                index = 0;

                // Print the line
                wresult = 0;
                ssize_t buffer_length = strlen(buffer);
                while (wresult != buffer_length) {
                    ssize_t res = write(STDOUT_FILENO, buffer + wresult, buffer_length - wresult);

                    if (wresult < 0) {
                        perror("write");
                        return -1;
                    }

                    wresult += res;
                }

                // Stop if we read 10 lines already
                if (lines == LINES_TO_READ) {
                    break;
                }

                lines++;
            } else {
                buffer[index++] = ch;
            }
        }

        if (close(fd) < 0) {
            perror("close");
            return -1;
        }
    }

    return 0;
}

И это работает с файлами, длина строки которых меньше BUFF_SIZE (как сейчас установлено, 4096).

Как избежать этого и заставить его работать при любой длине линии?

Answer 1

Не читайте по одному байту за раз. Считайте порцию (4096 или 8192 байта - разумные размеры, или используйте PIPE_BUF (из limit.h)) в буфер. Выведите каждый символ, считая новые строки. Если вы печатаете достаточно новых строк, прекратите. Если вы достигли конца буфера и не напечатали достаточно строк, считайте больше данных в буфер.