zckhelper.h

Go to the documentation of this file.

/*---------------------------------------------------------------------\
|                          ____ _   __ __ ___                          |
|                         |__  / \ / / . \ . \                         |
|                           / / \ V /|  _/  _/                         |
|                          / /__ | | | | | |                           |
|                         /_____||_| |_| |_|                           |
|                                                                      |
----------------------------------------------------------------------/
*
* This file contains private API, this might break at any time between releases.
* You have been warned!
*
*/
#ifndef ZYPPNG_CURL_ZCKHELPER_H_INCLUDED
#define ZYPPNG_CURL_ZCKHELPER_H_INCLUDED

#include <zypp-curl/ng/network/curlmultiparthandler.h>
#include <zypp-curl/ng/network/rangedesc.h>

#include <zypp-core/AutoDispose.h>
#include <zypp-core/ng/base/Base>

extern "C" {
  typedef struct zckCtx zckCtx;
}

namespace zyppng {

  ZYPP_FWD_DECL_TYPE_WITH_REFS(ZckLoader);

  class ZckError : public zypp::Exception
  {
  public:
    ZckError( const std::string & msg_r );
    ZckError( std::string && msg_r );
  };

  class ZckLoader : public Base {
  public:

      enum State {
        Initial,
        DownloadLead,
        DownloadHeader,
        DownloadChunks,
        Finished
      };

    using Block = RangeDesc;

    struct PrepareResult {
      enum Code {
        Error,          // we got an error
        NothingToDo,    // Target file is already complete
        ExceedsMaxLen,  // Zchunk header reports a different filesize than what was expected
        Success         // Returns a list of blocks to fetch
      };

      Code _code;
      std::vector<Block> _blocks;
      zypp::ByteCount _bytesReused;
      std::string _message;
    };


    expected<void> buildZchunkFile( const zypp::Pathname &target, const zypp::Pathname &delta, const std::optional<zypp::ByteCount> &expectedFileSize, const std::optional<zypp::ByteCount> &zcKHeaderSize );

    expected<void> cont();

    void setFailed( const std::string &msg );

    SignalProxy<void(const std::vector<Block> &)> sigBlocksRequired();

    SignalProxy<void( PrepareResult )> sigFinished();


    static zypp::ByteCount minZchunkDownloadSize();

    static bool isZchunkFile(const zypp::Pathname &file);

    static PrepareResult prepareZck ( const zypp::Pathname &delta, const zypp::Pathname &target, const zypp::ByteCount &expectedFileSize );

    static bool validateZckFile( const zypp::Pathname &file, std::string &error );


  private:
    State _state = Initial;
    zypp::AutoDispose<zckCtx *> _zchunkContext;
    zypp::AutoFD _targetFd;
    zypp::ByteCount _bytesReused = 0;

    zypp::Pathname _target;
    zypp::Pathname _delta;
    std::optional<zypp::ByteCount> _expectedFileSize;
    std::optional<zypp::ByteCount> _zcKHeaderSize;

    Signal<void(const std::vector<Block> &)> _sigBlocksRequired;
    Signal<void( PrepareResult )> _sigFinished;
  };
}
#endif