downloader.hpp

Go to the documentation of this file.

#ifndef H_KARAZEH_DOWNLOADER_H
#define H_KARAZEH_DOWNLOADER_H

#include <curl/curl.h>
#include <boost/filesystem.hpp>
#include "binreloc/binreloc.h"
#include "karazeh_export.h"
#include "karazeh/karazeh.hpp"
#include "karazeh/logger.hpp"
#include "karazeh/hasher.hpp"
#include "karazeh/file_manager.hpp"
#include "karazeh/config.hpp"

namespace kzh {
  struct download_t;

  class KARAZEH_EXPORT downloader : protected logger {
  public:
    downloader(config_t const&, file_manager const&);
    virtual ~downloader();

    int retry_count() const;
    void set_retry_count(int);

    bool fetch(string_t const& URI, string_t& out_buf) const;

    bool fetch(string_t const& URI, std::ostream& out_file) const;

    bool fetch(
      string_t const& URI,
      path_t const& path_to_file,
      string_t const& checksum,
      uint64_t expected_size_bytes = 0,
      int* const nr_retries = NULL
    ) const;

  private:
    const config_t &config_;
    const file_manager& file_manager_;

    bool fetch(string_t const& URI, download_t*, bool assume_ownership = true) const;

    int retry_count_;
  };

  struct KARAZEH_EXPORT download_t {
    inline explicit
    download_t(std::ostream& s)
    : buf(nullptr),
      to_file(false),
      status(false),
      size(0),
      retry_no(0),
      stream(s) {
    }

    string_t      *buf;
    string_t      uri;
    bool          status;
    bool          to_file;
    std::ostream  &stream;
    uint64_t      size;
    int           retry_no;
  };
} // end of namespace kzh

#endif