SourceProvideFile.cc

Go to the documentation of this file.

/*---------------------------------------------------------------------\
|                          ____ _   __ __ ___                          |
|                         |__  / \ / / . \ . \                         |
|                           / / \ V /|  _/  _/                         |
|                          / /__ | | | | | |                           |
|                         /_____||_| |_| |_|                           |
|                                                                      |
\---------------------------------------------------------------------*/
#include <iostream>
#include <fstream>
#include <sstream>
#include "zypp/base/Logger.h"

#include "zypp/source/SourceProvideFile.h"
#include "zypp/ZYppCallbacks.h"

using std::endl;

namespace zypp
{ 

  namespace source
  { 

    //
    //  CLASS NAME : ProvideFilePolicy
    //

    namespace
    { 

      bool yes() { return true; }
      bool no()  { return false; }

    } // namespace

    ProvideFilePolicy & ProvideFilePolicy::failOnChecksumErrorCB( bool yesno_r )
    {
      _failOnChecksumErrorCB = (yesno_r ? &yes : &no);
      return *this;
    }

    bool ProvideFilePolicy::progress( int value ) const
    {
      if ( _progressCB )
        return _progressCB( value );
      return true;
    }

    bool ProvideFilePolicy::failOnChecksumError() const
    {
      if ( _failOnChecksumErrorCB )
        return _failOnChecksumErrorCB();
      return true;
    }

    //
    //  provideFile
    //

    namespace
    { 

      struct DownloadFileReportHack : public callback::ReceiveReport<source::SourceReport>
      {
        virtual bool progress( int value )
        {
          if ( _redirect )
            return _redirect( value );
          return true;
        }
        function<bool ( int )> _redirect;
      };

    } // namespace

    ManagedFile provideFile( Source_Ref source_r,
                             const source::OnMediaLocation & loc_r,
                             const ProvideFilePolicy & policy_r )
    {
      MIL << "sourceProvideFile " << loc_r << endl;
      // Arrange DownloadFileReportHack to recieve the source::DownloadFileReport
      // and redirect download progress triggers to call the ProvideFilePolicy
      // callback.
      DownloadFileReportHack dumb;
      dumb._redirect = bind( mem_fun_ref( &ProvideFilePolicy::progress ),
                             ref( policy_r ), _1 );
      callback::TempConnect<source::SourceReport> temp( dumb );


      ManagedFile ret( source_r.provideFile( loc_r.filename(), loc_r.medianr() ),
                       bind( &Source_Ref::releaseFile, source_r, _1, loc_r.medianr() ) );

      if ( loc_r.checksum().empty() )
        {
          // no checksum in metadata
          WAR << "No checksum in metadata " << loc_r << endl;
        }
      else
        {
          std::ifstream input( ret->asString().c_str() );
          CheckSum retChecksum( loc_r.checksum().type(), input );
          input.close();

          if ( loc_r.checksum() != retChecksum )
            {
              // failed integity check
              std::ostringstream err;
              err << "File " << ret << " fails integrity check. Expected: [" << loc_r.checksum() << "] Got: [";
              if ( retChecksum.empty() )
                err << "Failed to compute checksum";
              else
                err << retChecksum;
              err << "]";

              if ( policy_r.failOnChecksumError() )
                ZYPP_THROW( Exception( err.str() ) );
              else
                WAR << "NO failOnChecksumError: " << err.str() << endl;
            }
        }

      MIL << "sourceProvideFile at " << ret << endl;
      return ret;
    }

  } // namespace source
} // namespace zypp

---