umd-livepatch/umdiff/compare.c

/**
 * @file       compare.c
 * @author     Karim Vergnes <me@thesola.io>
 * @copyright  GPLv2
 * @brief      rsync-based diff calculator
 *
 * This file leverages librsync to compute the difference between two files,
 * and convert it in-memory into the UMDiff format.
 */

#include "compare.h"

#include <librsync.h>

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

#define BUFFERS_SIZE 1048576

#define _impl_umdiff_alignBufferSize$(x) \
    ( x + BUFFERS_SIZE - (x % BUFFERS_SIZE) )

typedef union {
    int fd;
    void *opaque;
} _impl_umdiff_OpaqueFd;

rs_signature_t *workSignatures;

char buf_next_in[BUFFERS_SIZE];
char buf_next_out[BUFFERS_SIZE];

size_t workSigs_counter = 0;
size_t workSigs_size = 0;

rs_result
_impl_umdiff_sigJobSink(rs_job_t *job, rs_buffers_t *buf, void *dest)
{
    int res;

    if (!buf->next_out) {
        buf->next_out = (char *) workSignatures;
        buf->avail_out = workSigs_size;
        return RS_DONE;
    }

    if (BUFFERS_SIZE + workSigs_counter > workSigs_size) {
        workSignatures = realloc(workSignatures, workSigs_size + _impl_umdiff_alignBufferSize$(buf->avail_out));
        workSigs_size += _impl_umdiff_alignBufferSize$(buf->avail_out);
    }
    workSigs_counter += BUFFERS_SIZE;

    buf->avail_out = workSigs_size - workSigs_counter;
    //buf->next_out = dest + workSigs_counter;

    return RS_DONE;
}

rs_result
_impl_umdiff_deltaJobSink(rs_job_t *job, rs_buffers_t *buf, void *file_)
{
    umdiff_File *file = file_;
    size_t output_size = buf->next_out - buf_next_out;

    if (buf->next_out)
        umdiff_File_feedCommands(file, buf->next_out, output_size);

    buf->next_out = buf_next_out;
    buf->avail_out = BUFFERS_SIZE;

    return RS_DONE;
}

rs_result
_impl_umdiff_fileJobSource(rs_job_t *job, rs_buffers_t *buf, void *fd_)
{
    int res;
    _impl_umdiff_OpaqueFd fd = { .opaque = fd_ };

    buf->next_in = buf_next_in;
    buf->avail_in = read(fd.fd, buf->next_in, BUFFERS_SIZE);
    if (!buf->avail_in) {
        buf->eof_in = 1;
        return RS_INPUT_ENDED;
    }

    return RS_DONE;
}


int
umdiff_File_fromCompare(umdiff_File *file, int source_fd, int target_fd)
{
    _impl_umdiff_OpaqueFd source_fd_ = { .fd = source_fd };
    _impl_umdiff_OpaqueFd target_fd_ = { .fd = target_fd };

    rs_job_t *sigJob, *deltaJob;
    rs_buffers_t buffers = {
        .eof_in = 0,
        .avail_in = 0,
        .avail_out = 0,
        .next_in = NULL,
        .next_out = NULL
    };

    workSignatures = malloc(BUFFERS_SIZE);
    workSigs_size = BUFFERS_SIZE;

    sigJob = rs_sig_begin(ISO_SECTOR_SIZE,
                          RS_DEFAULT_MIN_STRONG_LEN,
                          RS_RK_BLAKE2_SIG_MAGIC);
    deltaJob = rs_delta_begin(workSignatures);

    rs_job_drive(sigJob, &buffers,
                _impl_umdiff_fileJobSource, source_fd_.opaque,
                _impl_umdiff_sigJobSink,    workSignatures);
    buffers = (rs_buffers_t) {
        .eof_in = 0,
        .avail_in = 0,
        .avail_out = 0,
        .next_in = NULL,
        .next_out = NULL
    };
    rs_job_drive(deltaJob, &buffers,
                _impl_umdiff_fileJobSource, target_fd_.opaque,
                _impl_umdiff_deltaJobSink,  file);

    free(workSignatures);

    return 0;
}

// vim: ft=c.doxygen