graphene/LibOS/shim/test/fs/copy_mmap_seq.c

#include "common.h"

void copy_data(int fi, int fo, const char* input_path, const char* output_path, size_t size) {
    if (size > 0) {
        size_t max_step = 16;
        if (size > 65536)
            max_step = 256;

        // map whole input/output file
        void* in = mmap_fd(input_path, fi, PROT_READ, 0, size);
        printf("mmap_fd(%zu) input OK\n", size);
        void* out = mmap_fd(output_path, fo, PROT_WRITE, 0, size);
        printf("mmap_fd(%zu) output OK\n", size);

        // copy data
        ssize_t offset = 0;
        ssize_t step;
        while (offset < size) {
            if (offset + max_step <= size)
                step = rand() % max_step + 1;
            else
                step = size - offset;

            if (ftruncate(fo, offset + step) != 0)
                fatal_error("ftruncate(%s, %zu) failed: %s\n", output_path, offset + step, strerror(errno));

            memcpy(out + offset, in + offset, step);
            offset += step;
        }

        // unmap
        munmap_fd(input_path, in, size);
        printf("munmap_fd(%zu) input OK\n", size);
        munmap_fd(output_path, out, size);
        printf("munmap_fd(%zu) output OK\n", size);
    }
}