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/ZYppCallbacks.h>
#include <zypp/media/MediaNetworkCommonHandler.h>
 
#include <curl/curl.h>
 
namespace zypp {
  namespace media {
 
//
//      CLASS NAME : MediaCurl
class MediaCurl : public MediaNetworkCommonHandler
{
  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 OnMediaLocation & file ) 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 OnMediaLocation& srcFile, const Pathname & targetFilename ) const override;
 
    virtual void doGetFileCopy( const OnMediaLocation &srcFile, const Pathname & targetFilename, callback::SendReport<DownloadProgressReport> & _report,  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(...) {} }
 
    static void setCookieFile( const Pathname & );
 
    class Callbacks
    {
      public:
        virtual ~Callbacks();
        virtual bool progress( int percent ) = 0;
    };
 
  protected:
    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();
 
    void evaluateCurlCode(const zypp::Pathname &filename, CURLcode code, bool timeout) const;
 
    void doGetFileCopyFile( const OnMediaLocation & srcFile, const Pathname & dest, FILE *file, callback::SendReport<DownloadProgressReport> & report, 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:
    std::string _currentCookieFile;
    static Pathname _cookieFile;
 
    mutable std::string _lastRedirect;  
 
  protected:
    CURL *_curl;
    char _curlError[ CURL_ERROR_SIZE ];
    curl_slist *_customHeaders;
};
ZYPP_DECLARE_OPERATORS_FOR_FLAGS(MediaCurl::RequestOptions);
 
 
  } // namespace media
} // namespace zypp
 
#endif // ZYPP_MEDIA_MEDIACURL_H