Line data    Source code

       1             : /* SPDX-License-Identifier: LGPL-2.1+ */
       2             : #pragma once
       3             : 
       4             : #include <gcrypt.h>
       5             : 
       6             : #include "curl-util.h"
       7             : #include "import-compress.h"
       8             : #include "macro.h"
       9             : 
      10             : typedef struct PullJob PullJob;
      11             : 
      12             : typedef void (*PullJobFinished)(PullJob *job);
      13             : typedef int (*PullJobOpenDisk)(PullJob *job);
      14             : typedef int (*PullJobHeader)(PullJob *job, const char *header, size_t sz);
      15             : typedef void (*PullJobProgress)(PullJob *job);
      16             : 
      17             : typedef enum PullJobState {
      18             :         PULL_JOB_INIT,
      19             :         PULL_JOB_ANALYZING, /* Still reading into ->payload, to figure out what we have */
      20             :         PULL_JOB_RUNNING,  /* Writing to destination */
      21             :         PULL_JOB_DONE,
      22             :         PULL_JOB_FAILED,
      23             :         _PULL_JOB_STATE_MAX,
      24             :         _PULL_JOB_STATE_INVALID = -1,
      25             : } PullJobState;
      26             : 
      27             : typedef enum VerificationStyle {
      28             :         VERIFICATION_STYLE_UNSET,
      29             :         VERIFICATION_PER_FILE,        /* SuSE-style ".sha256" files with inline signature */
      30             :         VERIFICATION_PER_DIRECTORY,   /* Ubuntu-style SHA256SUM files with detach SHA256SUM.gpg signatures */
      31             : } VerificationStyle;
      32             : 
      33             : #define PULL_JOB_IS_COMPLETE(j) (IN_SET((j)->state, PULL_JOB_DONE, PULL_JOB_FAILED))
      34             : 
      35             : struct PullJob {
      36             :         PullJobState state;
      37             :         int error;
      38             : 
      39             :         char *url;
      40             : 
      41             :         void *userdata;
      42             :         PullJobFinished on_finished;
      43             :         PullJobOpenDisk on_open_disk;
      44             :         PullJobHeader on_header;
      45             :         PullJobProgress on_progress;
      46             : 
      47             :         CurlGlue *glue;
      48             :         CURL *curl;
      49             :         struct curl_slist *request_header;
      50             : 
      51             :         char *etag;
      52             :         char **old_etags;
      53             :         bool etag_exists;
      54             : 
      55             :         uint64_t content_length;
      56             :         uint64_t written_compressed;
      57             :         uint64_t written_uncompressed;
      58             : 
      59             :         uint64_t uncompressed_max;
      60             :         uint64_t compressed_max;
      61             : 
      62             :         uint8_t *payload;
      63             :         size_t payload_size;
      64             :         size_t payload_allocated;
      65             : 
      66             :         int disk_fd;
      67             : 
      68             :         usec_t mtime;
      69             : 
      70             :         ImportCompress compress;
      71             : 
      72             :         unsigned progress_percent;
      73             :         usec_t start_usec;
      74             :         usec_t last_status_usec;
      75             : 
      76             :         bool allow_sparse;
      77             : 
      78             :         bool calc_checksum;
      79             :         gcry_md_hd_t checksum_context;
      80             : 
      81             :         char *checksum;
      82             : 
      83             :         VerificationStyle style;
      84             : };
      85             : 
      86             : int pull_job_new(PullJob **job, const char *url, CurlGlue *glue, void *userdata);
      87             : PullJob* pull_job_unref(PullJob *job);
      88             : 
      89             : int pull_job_begin(PullJob *j);
      90             : 
      91             : void pull_job_curl_on_finished(CurlGlue *g, CURL *curl, CURLcode result);
      92             : 
      93           0 : DEFINE_TRIVIAL_CLEANUP_FUNC(PullJob*, pull_job_unref);