MediaCurl.h

Go to the documentation of this file.

/*---------------------------------------------------------------------\
|                          ____ _   __ __ ___                          |
|                         |__  / \ / / . \ . \                         |
|                           / / \ V /|  _/  _/                         |
|                          / /__ | | | | | |                           |
|                         /_____||_| |_| |_|                           |
|                                                                      |
\---------------------------------------------------------------------*/
#ifndef ZYPP_MEDIA_MEDIACURL_H
#define ZYPP_MEDIA_MEDIACURL_H
 
#include <zypp/base/Flags.h>
#include <zypp/media/TransferSettings.h>
#include <zypp/media/MediaHandler.h>
#include <zypp/ZYppCallbacks.h>
 
#include <curl/curl.h>
 
namespace zypp {
  namespace media {
 
//
//      CLASS NAME : MediaCurl
class MediaCurl : public MediaHandler
{
  public:
    enum RequestOption
    {
        OPTION_NONE = 0x0,
        OPTION_RANGE = 0x1,
        OPTION_HEAD = 0x02,
        OPTION_NO_IFMODSINCE = 0x04,
        OPTION_NO_REPORT_START = 0x08,
    };
    ZYPP_DECLARE_FLAGS(RequestOptions,RequestOption);
 
  protected:
 
    Url clearQueryString(const Url &url) const;
 
    virtual void attachTo (bool next = false) override;
    virtual void releaseFrom( const std::string & ejectDev ) override;
    virtual void getFile( const Pathname & filename, const ByteCount &expectedFileSize_r ) const override;
    virtual void getDir( const Pathname & dirname, bool recurse_r ) const override;
    virtual void getDirInfo( std::list<std::string> & retlist,
                             const Pathname & dirname, bool dots = true ) const override;
    virtual void getDirInfo( filesystem::DirContent & retlist,
                             const Pathname & dirname, bool dots = true ) const override;
    virtual bool getDoesFileExist( const Pathname & filename ) const override;
 
    virtual bool doGetDoesFileExist( const Pathname & filename ) const;
 
    virtual void disconnectFrom() override;
    virtual void getFileCopy( const Pathname & srcFilename, const Pathname & targetFilename, const ByteCount &expectedFileSize_r) const override;
 
    virtual void doGetFileCopy( const Pathname & srcFilename, const Pathname & targetFilename, callback::SendReport<DownloadProgressReport> & _report, const ByteCount &expectedFileSize_r,  RequestOptions options = OPTION_NONE ) const;
 
 
    virtual bool checkAttachPoint(const Pathname &apoint) const override;
 
  public:
 
    MediaCurl( const Url &      url_r,
               const Pathname & attach_point_hint_r );
 
    virtual ~MediaCurl() override { try { release(); } catch(...) {} }
 
    TransferSettings & settings();
 
    static void setCookieFile( const Pathname & );
 
    class Callbacks
    {
      public:
        virtual ~Callbacks();
        virtual bool progress( int percent ) = 0;
    };
 
  protected:
//     /** Callback sending just an alive trigger to the UI, without stats (e.g. during metalink download). */
    static int aliveCallback( void *clientp, double dltotal, double dlnow, double ultotal, double ulnow );
    static int progressCallback( void *clientp, double dltotal, double dlnow, double ultotal, double ulnow );
    static CURL *progressCallback_getcurl( void *clientp );
    void checkProtocol(const Url &url) const;
 
    virtual void setupEasy();
    Url getFileUrl(const Pathname & filename) const;
 
    void evaluateCurlCode(const zypp::Pathname &filename, CURLcode code, bool timeout) const;
 
    void doGetFileCopyFile( const Pathname & srcFilename, const Pathname & dest, FILE *file, callback::SendReport<DownloadProgressReport> & _report, const ByteCount &expectedFileSize_r, RequestOptions options = OPTION_NONE ) const;
 
    static void resetExpectedFileSize ( void *clientp, const ByteCount &expectedFileSize );
 
  private:
    std::string getAuthHint() const;
 
    bool authenticate(const std::string & availAuthTypes, bool firstTry) const;
 
    bool detectDirIndex() const;
 
  private:
    long _curlDebug;
 
    std::string _currentCookieFile;
    static Pathname _cookieFile;
 
    mutable std::string _lastRedirect;  
 
  protected:
    CURL *_curl;
    char _curlError[ CURL_ERROR_SIZE ];
    curl_slist *_customHeaders;
    TransferSettings _settings;
};
ZYPP_DECLARE_OPERATORS_FOR_FLAGS(MediaCurl::RequestOptions);
 
 
  } // namespace media
} // namespace zypp
 
#endif // ZYPP_MEDIA_MEDIACURL_H