Question

我想复制一个文件，跳过前n行，最后m行使用open，read，write和lseek （例如，n = 1，m = 2，源文件：
AAAAAAA
BBBBBBB
CCCCCCC
DDDDDDD

目标文件：
BBBBBBB）

我知道如何复制文件但不知道如何跳过这些行。这是我的复制代码：

    char buf[128];
    size_t size;

    int source = open(argv[1], O_RDONLY);
    int dest = open(argv[2], O_CREAT | O_APPEND | O_WRONLY);

    if(source == -1) {
            printf("error");
            return;
    }
    if(dest == -1) {
            printf("error");
            return;
    }

    while((size = read(source, buf, sizeof(buf))) > 0) {
            write(dest, buf, size);
    }

    close(source);
    close(dest);

我该如何解决这个问题？

Answer 1

您应该使用fgets来阅读您的文件，因为它会逐行读取。

因为我总是建议你有多少行是不容易的

使用fgets逐行读取文件
跳过输出第一个n行
将其余部分写入输出文件并计算行数并记住每行的长度。
使用ftruncate截断最后m行。

这应该可以解决问题：

void copy_nm(char * source, char * dest, int n, int m) {
    FILE * in = fopen(source, "r");
    FILE * out = fopen(dest, "w");

    size_t file_length = 0;
    size_t line_lengths[m + 1];
    memset(line_lengths, 0 , sizeof(line_lengths));

    int lengths_iterator = 0;


    char buffer[0x400];

    while (fgets(buffer, sizeof(buffer), in)) {
        size_t length = strlen(buffer);

        if (n) { // skip this line
            if (buffer[length - 1] == '\n') // only if it is a real line
                n--;
            continue;
        }

        fwrite(buffer, length, 1, out);
        line_lengths[lengths_iterator] += length;
        file_length += length;

        if (buffer[length - 1] != '\n') { // line was longer then the buffer
            continue;
        }

        lengths_iterator++;
        lengths_iterator %= m+1;

        line_lengths[lengths_iterator] = 0;
    }

    for (lengths_iterator = 0; lengths_iterator < m+1; lengths_iterator++) {
        file_length -= line_lengths[lengths_iterator];
    }

    fseek(out, 0, SEEK_SET); // rewind before truncating
    ftruncate(fileno(out), file_length);

}

复制文件跳过前n行和后m行

1 个答案: