Bug fixes, new major version

[apt.git] / apt-pkg / acquire.h

diff --git a/apt-pkg/acquire.h b/apt-pkg/acquire.h
index acfbb3ead8b0b2093719fbcaafa9189e23ec147d..de1474f56f6c83eab230a262d45f74c5eb31bc0f 100644 (file)
--- a/apt-pkg/acquire.h
+++ b/apt-pkg/acquire.h
@@ -1,6 +1,6 @@
 // -*- mode: cpp; mode: fold -*-
 // Description                                                         /*{{{*/
 // -*- mode: cpp; mode: fold -*-
 // Description                                                         /*{{{*/

-// $Id: acquire.h,v 1.13 1998/11/14 01:39:46 jgg Exp $
+// $Id: acquire.h,v 1.27 2000/01/27 04:15:09 jgg Exp $

 /* ######################################################################
 
    Acquire - File Acquiration
 /* ######################################################################
 
    Acquire - File Acquiration


@@ -82,8 +82,8 @@ class pkgAcquire
    string QueueName(string URI,MethodConfig const *&Config);
 
    // FDSET managers for derived classes
    string QueueName(string URI,MethodConfig const *&Config);
 
    // FDSET managers for derived classes

-   void SetFds(int &Fd,fd_set *RSet,fd_set *WSet);
-   void RunFds(fd_set *RSet,fd_set *WSet);   
+   virtual void SetFds(int &Fd,fd_set *RSet,fd_set *WSet);
+   virtual void RunFds(fd_set *RSet,fd_set *WSet);   

 
    // A queue calls this when it dequeues an item
    void Bump();
 
    // A queue calls this when it dequeues an item
    void Bump();


@@ -91,23 +91,33 @@ class pkgAcquire
    public:
 
    MethodConfig *GetConfig(string Access);
    public:
 
    MethodConfig *GetConfig(string Access);

-   bool Run();

 
 

+   enum RunResult {Continue,Failed,Cancelled};
+
+   RunResult Run();
+   void Shutdown();
+   

    // Simple iteration mechanism
    inline Worker *WorkersBegin() {return Workers;};
    Worker *WorkerStep(Worker *I);
    inline Item **ItemsBegin() {return Items.begin();};
    inline Item **ItemsEnd() {return Items.end();};
    // Simple iteration mechanism
    inline Worker *WorkersBegin() {return Workers;};
    Worker *WorkerStep(Worker *I);
    inline Item **ItemsBegin() {return Items.begin();};
    inline Item **ItemsEnd() {return Items.end();};

-
+   
+   // Iterate over queued Item URIs
+   class UriIterator;
+   UriIterator UriBegin();
+   UriIterator UriEnd();
+   

    // Cleans out the download dir
    bool Clean(string Dir);
 
    // Returns the size of the total download set
    unsigned long TotalNeeded();
    unsigned long FetchNeeded();
    // Cleans out the download dir
    bool Clean(string Dir);
 
    // Returns the size of the total download set
    unsigned long TotalNeeded();
    unsigned long FetchNeeded();

+   unsigned long PartialPresent();

    
    pkgAcquire(pkgAcquireStatus *Log = 0);
    
    pkgAcquire(pkgAcquireStatus *Log = 0);

-   ~pkgAcquire();
+   virtual ~pkgAcquire();

 };
 
 // Description of an Item+URI
 };
 
 // Description of an Item+URI


@@ -123,6 +133,7 @@ struct pkgAcquire::ItemDesc
 class pkgAcquire::Queue
 {
    friend pkgAcquire;
 class pkgAcquire::Queue
 {
    friend pkgAcquire;

+   friend pkgAcquire::UriIterator;

    Queue *Next;
    
    protected:
    Queue *Next;
    
    protected:


@@ -149,6 +160,8 @@ class pkgAcquire::Queue
    QItem *Items;
    pkgAcquire::Worker *Workers;
    pkgAcquire *Owner;
    QItem *Items;
    pkgAcquire::Worker *Workers;
    pkgAcquire *Owner;

+   signed long PipeDepth;
+   unsigned long MaxPipeDepth;

    
    public:
    
    
    public:
    


@@ -162,7 +175,7 @@ class pkgAcquire::Queue
    bool ItemDone(QItem *Itm);
    
    bool Startup();
    bool ItemDone(QItem *Itm);
    
    bool Startup();

-   bool Shutdown();
+   bool Shutdown(bool Final);

    bool Cycle();
    void Bump();
    
    bool Cycle();
    void Bump();
    


@@ -170,6 +183,40 @@ class pkgAcquire::Queue
    ~Queue();
 };
 
    ~Queue();
 };
 

+class pkgAcquire::UriIterator
+{
+   pkgAcquire::Queue *CurQ;
+   pkgAcquire::Queue::QItem *CurItem;
+   
+   public:
+   
+   // Advance to the next item
+   inline void operator ++() {operator ++();};
+   void operator ++(int)
+   {
+      CurItem = CurItem->Next;
+      while (CurItem == 0 && CurQ != 0)
+      {
+        CurItem = CurQ->Items;
+        CurQ = CurQ->Next;
+      }
+   };
+   
+   // Accessors
+   inline pkgAcquire::ItemDesc const *operator ->() const {return CurItem;};
+   inline bool operator !=(UriIterator const &rhs) const {return rhs.CurQ != CurQ || rhs.CurItem != CurItem;};
+   inline bool operator ==(UriIterator const &rhs) const {return rhs.CurQ == CurQ && rhs.CurItem == CurItem;};
+   
+   UriIterator(pkgAcquire::Queue *Q) : CurQ(Q), CurItem(0)
+   {
+      while (CurItem == 0 && CurQ != 0)
+      {
+        CurItem = CurQ->Items;
+        CurQ = CurQ->Next;
+      }
+   }   
+};
+

 // Configuration information from each method
 struct pkgAcquire::MethodConfig
 {
 // Configuration information from each method
 struct pkgAcquire::MethodConfig
 {


@@ -179,11 +226,12 @@ struct pkgAcquire::MethodConfig
 
    string Version;
    bool SingleInstance;
 
    string Version;
    bool SingleInstance;

-   bool PreScan;

    bool Pipeline;
    bool SendConfig;
    bool LocalOnly;
    bool Pipeline;
    bool SendConfig;
    bool LocalOnly;

-      
+   bool NeedsCleanup;
+   bool Removable;
+   

    MethodConfig();
 };
 
    MethodConfig();
 };
 


@@ -199,20 +247,26 @@ class pkgAcquireStatus
    unsigned long TotalBytes;
    unsigned long FetchedBytes;
    unsigned long ElapsedTime;
    unsigned long TotalBytes;
    unsigned long FetchedBytes;
    unsigned long ElapsedTime;

+   unsigned long TotalItems;
+   unsigned long CurrentItems;

    
    public:
 
    bool Update;
    
    public:
 
    bool Update;

-   
+   bool MorePulses;
+      

    // Called by items when they have finished a real download
    virtual void Fetched(unsigned long Size,unsigned long ResumePoint);
    
    // Called by items when they have finished a real download
    virtual void Fetched(unsigned long Size,unsigned long ResumePoint);
    

+   // Called to change media
+   virtual bool MediaChange(string Media,string Drive) = 0;
+   

    // Each of these is called by the workers when an event occures
    // Each of these is called by the workers when an event occures

-   virtual void IMSHit(pkgAcquire::ItemDesc &Itm) {};
-   virtual void Fetch(pkgAcquire::ItemDesc &Itm) {};
-   virtual void Done(pkgAcquire::ItemDesc &Itm) {};
-   virtual void Fail(pkgAcquire::ItemDesc &Itm) {};   
-   virtual void Pulse(pkgAcquire *Owner);
+   virtual void IMSHit(pkgAcquire::ItemDesc &/*Itm*/) {};
+   virtual void Fetch(pkgAcquire::ItemDesc &/*Itm*/) {};
+   virtual void Done(pkgAcquire::ItemDesc &/*Itm*/) {};
+   virtual void Fail(pkgAcquire::ItemDesc &/*Itm*/) {};
+   virtual bool Pulse(pkgAcquire *Owner); // returns false on user cancel

    virtual void Start();
    virtual void Stop();
    
    virtual void Start();
    virtual void Stop();