src/import/pull-job.h

432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering#pragma once
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering/***
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  This file is part of systemd.
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  Copyright 2015 Lennart Poettering
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  systemd is free software; you can redistribute it and/or modify it
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  under the terms of the GNU Lesser General Public License as published by
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  the Free Software Foundation; either version 2.1 of the License, or
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  (at your option) any later version.
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  systemd is distributed in the hope that it will be useful, but
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  WITHOUT ANY WARRANTY; without even the implied warranty of
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  Lesser General Public License for more details.
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  You should have received a copy of the GNU Lesser General Public License
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering  along with systemd; If not, see <http://www.gnu.org/licenses/>.
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering***/
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering
a8fbdf5424be099ba1b2b1ec261c02b8759d6b0cThomas Hindoe Paaboel Andersen#include <gcrypt.h>
a8fbdf5424be099ba1b2b1ec261c02b8759d6b0cThomas Hindoe Paaboel Andersen
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering#include "curl-util.h"
432cea008720480f6eaae7e75e6ad255e91fe2b4Lennart Poettering#include "import-compress.h"
b6e676ce41508e2aeea22202fc8f234126177f52Lennart Poettering#include "macro.h"
b6e676ce41508e2aeea22202fc8f234126177f52Lennart Poettering
b6e676ce41508e2aeea22202fc8f234126177f52Lennart Poetteringtypedef struct PullJob PullJob;
4cee5eede280b7fd48c18a1942616c4ac896a554Lennart Poettering
26166c88e0b47b83972f32b5057ecbffe06bf904Lennart Poetteringtypedef void (*PullJobFinished)(PullJob *job);
typedef int (*PullJobOpenDisk)(PullJob *job);
typedef int (*PullJobHeader)(PullJob *job, const char *header, size_t sz);
typedef void (*PullJobProgress)(PullJob *job);

typedef enum PullJobState {
        PULL_JOB_INIT,
        PULL_JOB_ANALYZING, /* Still reading into ->payload, to figure out what we have */
        PULL_JOB_RUNNING,  /* Writing to destination */
        PULL_JOB_DONE,
        PULL_JOB_FAILED,
        _PULL_JOB_STATE_MAX,
        _PULL_JOB_STATE_INVALID = -1,
} PullJobState;

#define PULL_JOB_IS_COMPLETE(j) (IN_SET((j)->state, PULL_JOB_DONE, PULL_JOB_FAILED))

typedef enum PullJobCompression {
        PULL_JOB_UNCOMPRESSED,
        PULL_JOB_XZ,
        PULL_JOB_GZIP,
        PULL_JOB_BZIP2,
        _PULL_JOB_COMPRESSION_MAX,
        _PULL_JOB_COMPRESSION_INVALID = -1,
} PullJobCompression;

struct PullJob {
        PullJobState state;
        int error;

        char *url;

        void *userdata;
        PullJobFinished on_finished;
        PullJobOpenDisk on_open_disk;
        PullJobHeader on_header;
        PullJobProgress on_progress;

        CurlGlue *glue;
        CURL *curl;
        struct curl_slist *request_header;

        char *etag;
        char **old_etags;
        bool etag_exists;

        uint64_t content_length;
        uint64_t written_compressed;
        uint64_t written_uncompressed;

        uint64_t uncompressed_max;
        uint64_t compressed_max;

        uint8_t *payload;
        size_t payload_size;
        size_t payload_allocated;

        int disk_fd;

        usec_t mtime;

        ImportCompress compress;

        unsigned progress_percent;
        usec_t start_usec;
        usec_t last_status_usec;

        bool allow_sparse;

        bool calc_checksum;
        gcry_md_hd_t checksum_context;

        char *checksum;

        bool grow_machine_directory;
        uint64_t written_since_last_grow;
};

int pull_job_new(PullJob **job, const char *url, CurlGlue *glue, void *userdata);
PullJob* pull_job_unref(PullJob *job);

int pull_job_begin(PullJob *j);

void pull_job_curl_on_finished(CurlGlue *g, CURL *curl, CURLcode result);

DEFINE_TRIVIAL_CLEANUP_FUNC(PullJob*, pull_job_unref);