// -*- mode: cpp; mode: fold -*- // Description /*{{{*/ // $Id: acquire-item.h,v 1.26.2.3 2004/01/02 18:51:00 mdz Exp $ /* ###################################################################### Acquire Item - Item to acquire When an item is instantiated it will add it self to the local list in the Owner Acquire class. Derived classes will then call QueueURI to register all the URI's they wish to fetch at the initial moment. Two item classes are provided to provide functionality for downloading of Index files and downloading of Packages. A Archive class is provided for downloading .deb files. It does Md5 checking and source location as well as a retry algorithm. ##################################################################### */ /*}}}*/ #ifndef PKGLIB_ACQUIRE_ITEM_H #define PKGLIB_ACQUIRE_ITEM_H #include #include #include #include #include #include #ifdef __GNUG__ #pragma interface "apt-pkg/acquire-item.h" #endif // Item to acquire class pkgAcquire::Item { protected: // Some private helper methods for registering URIs pkgAcquire *Owner; inline void QueueURI(ItemDesc &Item) {Owner->Enqueue(Item);}; inline void Dequeue() {Owner->Dequeue(this);}; // Safe rename function with timestamp preservation void Rename(string From,string To); public: // State of the item enum {StatIdle, StatFetching, StatDone, StatError, StatAuthError} Status; string ErrorText; unsigned long FileSize; unsigned long PartialSize; const char *Mode; unsigned long ID; bool Complete; bool Local; // Number of queues we are inserted into unsigned int QueueCounter; // File to write the fetch into string DestFile; // Action members invoked by the worker virtual void Failed(string Message,pkgAcquire::MethodConfig *Cnf); virtual void Done(string Message,unsigned long Size,string Md5Hash, pkgAcquire::MethodConfig *Cnf); virtual void Start(string Message,unsigned long Size); virtual string Custom600Headers() {return string();}; virtual string DescURI() = 0; virtual string ShortDesc() {return DescURI();} virtual void Finished() {}; // Inquire functions virtual string MD5Sum() {return string();}; pkgAcquire *GetOwner() {return Owner;}; virtual bool IsTrusted() {return false;}; Item(pkgAcquire *Owner); virtual ~Item(); }; // item for index diffs class pkgAcqDiffIndex : public pkgAcquire::Item { protected: bool Debug; string RealURI; public: // Specialized action members virtual void Failed(string Message,pkgAcquire::MethodConfig *Cnf); virtual void Done(string Message,unsigned long Size,string Md5Hash, pkgAcquire::MethodConfig *Cnf); virtual string DescURI() {return RealURI + "Index";}; virtual string Custom600Headers(); // helpers bool ParseDiffIndex(string IndexDiffFile); pkgAcqDiffIndex(pkgAcquire *Owner,string URI,string URIDesc, string ShortDesct, string ExpectedMD5); } class pkgAcqIndexDiffs : public pkgAcquire::Item { protected: bool Debug; pkgAcquire::ItemDesc Desc; string RealURI; string ExpectedMD5; // this is the SHA-1 sum we expect after the patching string ServerSha1; string CurrentPackagesFile; string Description; struct DiffInfo { string file; string sha1; unsigned long size; }; vector available_patches; enum {StateFetchIndex,StateFetchDiff,StateUnzipDiff,StateApplyDiff} State; public: // Specialized action members virtual void Failed(string Message,pkgAcquire::MethodConfig *Cnf); virtual void Done(string Message,unsigned long Size,string Md5Hash, pkgAcquire::MethodConfig *Cnf); virtual string DescURI() {return RealURI + "Index";}; virtual string Custom600Headers(); // various helpers bool QueueNextDiff(); bool ApplyDiff(string PatchFile); void Finish(bool allDone=false); pkgAcqIndexDiffs(pkgAcquire *Owner,string URI,string URIDesc, string ShortDesct, string ExpectedMD5, vector diffs=vector()); }; // Item class for index files class pkgAcqIndex : public pkgAcquire::Item { protected: bool Decompression; bool Erase; pkgAcquire::ItemDesc Desc; string RealURI; string ExpectedMD5; public: // Specialized action members virtual void Failed(string Message,pkgAcquire::MethodConfig *Cnf); virtual void Done(string Message,unsigned long Size,string Md5Hash, pkgAcquire::MethodConfig *Cnf); virtual string Custom600Headers(); virtual string DescURI() {return RealURI + ".gz";}; pkgAcqIndex(pkgAcquire *Owner,string URI,string URIDesc, string ShortDesct, string ExpectedMD5, string compressExt=""); }; struct IndexTarget { string URI; string Description; string ShortDesc; string MetaKey; }; // Item class for index signatures class pkgAcqMetaSig : public pkgAcquire::Item { protected: pkgAcquire::ItemDesc Desc; string RealURI,MetaIndexURI,MetaIndexURIDesc,MetaIndexShortDesc; indexRecords* MetaIndexParser; const vector* IndexTargets; public: // Specialized action members virtual void Failed(string Message,pkgAcquire::MethodConfig *Cnf); virtual void Done(string Message,unsigned long Size,string Md5Hash, pkgAcquire::MethodConfig *Cnf); virtual string Custom600Headers(); virtual string DescURI() {return RealURI; }; pkgAcqMetaSig(pkgAcquire *Owner,string URI,string URIDesc, string ShortDesc, string MetaIndexURI, string MetaIndexURIDesc, string MetaIndexShortDesc, const vector* IndexTargets, indexRecords* MetaIndexParser); }; // Item class for index signatures class pkgAcqMetaIndex : public pkgAcquire::Item { protected: pkgAcquire::ItemDesc Desc; string RealURI; // FIXME: is this redundant w/ Desc.URI? string SigFile; const vector* IndexTargets; indexRecords* MetaIndexParser; bool AuthPass; bool VerifyVendor(); void RetrievalDone(string Message); void AuthDone(string Message); void QueueIndexes(bool verify); public: // Specialized action members virtual void Failed(string Message,pkgAcquire::MethodConfig *Cnf); virtual void Done(string Message,unsigned long Size,string Md5Hash, pkgAcquire::MethodConfig *Cnf); virtual string Custom600Headers(); virtual string DescURI() {return RealURI; }; pkgAcqMetaIndex(pkgAcquire *Owner, string URI,string URIDesc, string ShortDesc, string SigFile, const vector* IndexTargets, indexRecords* MetaIndexParser); }; // Item class for archive files class pkgAcqArchive : public pkgAcquire::Item { protected: // State information for the retry mechanism pkgCache::VerIterator Version; pkgAcquire::ItemDesc Desc; pkgSourceList *Sources; pkgRecords *Recs; string MD5; string &StoreFilename; pkgCache::VerFileIterator Vf; unsigned int Retries; bool Trusted; // Queue the next available file for download. bool QueueNext(); public: // Specialized action members virtual void Failed(string Message,pkgAcquire::MethodConfig *Cnf); virtual void Done(string Message,unsigned long Size,string Md5Hash, pkgAcquire::MethodConfig *Cnf); virtual string MD5Sum() {return MD5;}; virtual string DescURI() {return Desc.URI;}; virtual string ShortDesc() {return Desc.ShortDesc;}; virtual void Finished(); virtual bool IsTrusted(); pkgAcqArchive(pkgAcquire *Owner,pkgSourceList *Sources, pkgRecords *Recs,pkgCache::VerIterator const &Version, string &StoreFilename); }; // Fetch a generic file to the current directory class pkgAcqFile : public pkgAcquire::Item { pkgAcquire::ItemDesc Desc; string Md5Hash; unsigned int Retries; public: // Specialized action members virtual void Failed(string Message,pkgAcquire::MethodConfig *Cnf); virtual void Done(string Message,unsigned long Size,string Md5Hash, pkgAcquire::MethodConfig *Cnf); virtual string MD5Sum() {return Md5Hash;}; virtual string DescURI() {return Desc.URI;}; pkgAcqFile(pkgAcquire *Owner,string URI,string MD5,unsigned long Size, string Desc,string ShortDesc); }; #endif