apt-pkg/acquire-method.h

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: acquire-method.h,v 1.15.2.1 2003/12/24 23:09:17 mdz Exp $
   4 /* ######################################################################
   5
   6    Acquire Method - Method helper class + functions
   7
   8    These functions are designed to be used within the method task to
   9    ease communication with APT.
  10
  11    ##################################################################### */
  12                                                                         /*}}}*/
  13
  14 /** \addtogroup acquire
  15  *  @{
  16  *
  17  *  \file acquire-method.h
  18  */
  19
  20 #ifndef PKGLIB_ACQUIRE_METHOD_H
  21 #define PKGLIB_ACQUIRE_METHOD_H
  22
  23 #include <apt-pkg/hashes.h>
  24 #include <apt-pkg/macros.h>
  25
  26 #include <stdarg.h>
  27 #include <time.h>
  28
  29 #include <string>
  30 #include <vector>
  31
  32 #ifndef APT_8_CLEANER_HEADERS
  33 #include <apt-pkg/configuration.h>
  34 #include <apt-pkg/strutl.h>
  35 #endif
  36
  37 class pkgAcqMethod
  38 {
  39    protected:
  40
  41    struct FetchItem
  42    {
  43       FetchItem *Next;
  44
  45       std::string Uri;
  46       std::string DestFile;
  47       int DestFileFd;
  48       time_t LastModified;
  49       bool IndexFile;
  50       bool FailIgnore;
  51       HashStringList ExpectedHashes;
  52       // a maximum size we will download, this can be the exact filesize
  53       // for when we know it or a arbitrary limit when we don't know the
  54       // filesize (like a InRelease file)
  55       unsigned long long MaximumSize;
  56    };
  57
  58    struct FetchResult
  59    {
  60       HashStringList Hashes;
  61       std::vector<std::string> GPGVOutput;
  62       time_t LastModified;
  63       bool IMSHit;
  64       std::string Filename;
  65       unsigned long long Size;
  66       unsigned long long ResumePoint;
  67
  68       void TakeHashes(class Hashes &Hash);
  69       FetchResult();
  70    };
  71
  72    // State
  73    std::vector<std::string> Messages;
  74    FetchItem *Queue;
  75    FetchItem *QueueBack;
  76    std::string FailReason;
  77    std::string UsedMirror;
  78    std::string IP;
  79
  80    // Handlers for messages
  81    virtual bool Configuration(std::string Message);
  82    virtual bool Fetch(FetchItem * /*Item*/) {return true;};
  83    virtual bool URIAcquire(std::string const &/*Message*/, FetchItem *Itm) { return Fetch(Itm); };
  84
  85    // Outgoing messages
  86    void Fail(bool Transient = false);
  87    inline void Fail(const char *Why, bool Transient = false) {Fail(std::string(Why),Transient);};
  88    virtual void Fail(std::string Why, bool Transient = false);
  89    virtual void URIStart(FetchResult &Res);
  90    virtual void URIDone(FetchResult &Res,FetchResult *Alt = 0);
  91
  92    bool MediaFail(std::string Required,std::string Drive);
  93    virtual void Exit() {};
  94
  95    void PrintStatus(char const * const header, const char* Format, va_list &args) const;
  96
  97    public:
  98    enum CnfFlags {SingleInstance = (1<<0),
  99                   Pipeline = (1<<1), SendConfig = (1<<2),
 100                   LocalOnly = (1<<3), NeedsCleanup = (1<<4),
 101                   Removable = (1<<5)};
 102
 103    void Log(const char *Format,...);
 104    void Status(const char *Format,...);
 105
 106    void Redirect(const std::string &NewURI);
 107
 108    int Run(bool Single = false);
 109    inline void SetFailReason(std::string Msg) {FailReason = Msg;};
 110    inline void SetIP(std::string aIP) {IP = aIP;};
 111
 112    pkgAcqMethod(const char *Ver,unsigned long Flags = 0);
 113    virtual ~pkgAcqMethod();
 114    void DropPrivsOrDie();
 115    private:
 116    APT_HIDDEN void Dequeue();
 117 };
 118
 119 /** @} */
 120
 121 #endif