我目前正在用C实现UnixHEAD
命令,并且只使用系统函数。到目前为止,它在文件上运行得很好,这些文件的行长度小于我为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
)。如何避免这种情况,并使其工作的线长度是什么?
最佳答案
不要一次读取一个字节。读取块(4096或8192字节是合理的大小,或者使用PIPE_BUF(from limits.h))进入缓冲区。在计算换行符时输出每个字符。如果打印足够的换行符,请终止。如果到达缓冲区的末尾,但没有打印足够的行,请将更多数据读入缓冲区。
关于c - C语言中的Unix HEAD命令实现在较大行上失败,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/47041443/