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 <zypp-curl/private/curlhelper_p.h>
 
#include <curl/curl.h>
 
namespace zypp {
  namespace media {

//
//      CLASS NAME : MediaCurl
class MediaCurl : public MediaNetworkCommonHandler, public internal::CurlPollHelper::CurlPoll
{
  struct RequestData {
    int mirror = -1;
    CURL *curl = nullptr;
  };
 
  protected:
 
    void releaseFrom( const std::string & ejectDev ) override;

    bool getDoesFileExist( const Pathname & filename ) const override;

    bool doGetDoesFileExist( const int mirror, const Pathname & filename );

    void disconnectFrom() override;
    void getFileCopy( const OnMediaLocation& srcFile, const Pathname & target ) const override;
 
    void getFileCopyFromMirror(const int mirror, const OnMediaLocation& srcFile, const Pathname & target );
 
  public:
 
    MediaCurl( const MirroredOrigin &origin_r,
               const Pathname &attach_point_hint_r );
 
    ~MediaCurl() override;
 
    static void setCookieFile( const Pathname & );
 
  protected:
    static int aliveCallback( void *clientp, curl_off_t dltotal, curl_off_t dlnow, curl_off_t ultotal, curl_off_t ulnow );
    static int progressCallback( void *clientp, curl_off_t dltotal, curl_off_t dlnow, curl_off_t ultotal, curl_off_t ulnow );
    static size_t writeCallback( char *ptr, size_t size, size_t nmemb, void *userdata );
 
    void checkProtocol(const Url &url) const override;

    void setupEasy(RequestData &rData, TransferSettings &settings );

    void evaluateCurlCode( RequestData &rData, const zypp::Pathname &fileName, CURLcode code, bool timeout ) const;
 
    static void resetExpectedFileSize ( void *clientp, const ByteCount &expectedFileSize );
 
  private:
 
    CURLcode executeCurl( RequestData &rData );

    std::string getAuthHint(CURL *curl) const;
 
    //bool authenticate(const std::string & availAuthTypes, bool firstTry) const;
 
    bool detectDirIndex() const;
 
  private:
    std::string _currentCookieFile;
    static Pathname _cookieFile;
    char _curlError[ CURL_ERROR_SIZE ];
 
    mutable std::string _lastRedirect;     
    curl_slist *_customHeaders;
 
    const char* curlError() const { return _curlError; };
    void setCurlError(const char* error);
};

 
  } // namespace media
} // namespace zypp
 
#endif // ZYPP_MEDIA_MEDIACURL_H