]>
Commit | Line | Data |
---|---|---|
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 |