MediaNetworkCommonHandler.h

Go to the documentation of this file.

/*---------------------------------------------------------------------\
|                          ____ _   __ __ ___                          |
|                         |__  / \ / / . \ . \                         |
|                           / / \ V /|  _/  _/                         |
|                          / /__ | | | | | |                           |
|                         /_____||_| |_| |_|                           |
|                                                                      |
\---------------------------------------------------------------------*/
#ifndef ZYPP_MEDIA_MEDIANETWORKCOMMONHANDLER_H
#define ZYPP_MEDIA_MEDIANETWORKCOMMONHANDLER_H

#include <zypp/media/MediaHandler.h>
#include <zypp-curl/TransferSettings>

namespace zypp
{
  namespace media
  {

    class CredentialManager;

    //  to avoid duplicated code.
    class MediaNetworkCommonHandler : public MediaHandler
    {
    public:

      static constexpr std::string_view MIRR_SETTINGS_KEY = "_zypp_transfersettings";

      MediaNetworkCommonHandler( const MirroredOrigin &origin_r,
                                 const Pathname & attach_point_r,
                                 const Pathname & urlpath_below_attachpoint_r,
                                 const bool       does_download_r );

    public:
      static zypp::Url findGeoIPRedirect ( const zypp::Url &url );

      void getFile( const OnMediaLocation & file ) const override;
      void getDir( const Pathname & dirname, bool recurse_r ) const override;
      void getDirInfo( std::list<std::string> & retlist,
                               const Pathname & dirname, bool dots = true ) const override;
      void getDirInfo( filesystem::DirContent & retlist,
                               const Pathname & dirname, bool dots = true ) const override;

    protected:

      void attachTo (bool next) override;

      bool checkAttachPoint(const Pathname &apoint) const override;

      Url clearQueryString(const Url &url) const;


      Url getFileUrl( int mirrorIdx, const Pathname & filename) const;


      void setupTransferSettings();
      void clearTransferSettings();

      bool authenticate( const Url &url, TransferSettings &settings, const std::string & availAuthTypes, bool firstTry );

      virtual void checkProtocol(const Url &url) const = 0;

      std::vector<unsigned> mirrorOrder( const OnMediaLocation &loc ) const;

      void deprioritizeMirror( unsigned mirr ) const;

    public:

      // standard auth procedure, shared with CommitPackagePreloader
      static bool authenticate( const Url &url, CredentialManager &cm, TransferSettings &settings, const std::string & availAuthTypes, bool firstTry);

      static const char *anonymousIdHeader();

      static const char *distributionFlavorHeader();

      static const char *agentString();

      template <typename Excpt>
      static bool canTryNextMirror( const Excpt &excpt_r ) {
        // check if we can retry on the next mirror
        bool fatal = (
              typeid(excpt_r) == typeid( MediaRequestCancelledException ) ||
              typeid(excpt_r) == typeid( MediaWriteException )            ||
              typeid(excpt_r) == typeid( MediaSystemException )           ||
              typeid(excpt_r) == typeid( MediaCurlInitException )         ||
              typeid(excpt_r) == typeid( MediaUnauthorizedException )
        );
        return !fatal;
      }

    protected:
      std::vector<Url> _redirTargets;
      mutable std::vector<unsigned> _mirrOrder;
    };

  } // namespace media
} // namespace zypp
#endif // ZYPP_MEDIA_MEDIANETWORKCOMMONHANDLER_H