]> git.saurik.com Git - apt.git/blame_incremental - apt-pkg/acquire-method.h
allow ratelimiting progress reporting for testcases
[apt.git] / apt-pkg / acquire-method.h
... / ...
CommitLineData
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
37class 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