]>
Commit | Line | Data |
---|---|---|
be4401bf AL |
1 | // -*- mode: cpp; mode: fold -*- |
2 | // Description /*{{{*/ | |
b98f2859 | 3 | // $Id: http.h,v 1.4 1998/11/11 06:54:22 jgg Exp $ |
be4401bf AL |
4 | /* ###################################################################### |
5 | ||
6 | HTTP Aquire Method - This is the HTTP aquire method for APT. | |
7 | ||
8 | ##################################################################### */ | |
9 | /*}}}*/ | |
10 | ||
11 | #ifndef APT_HTTP_H | |
12 | #define APT_HTTP_H | |
13 | ||
14 | #define MAXLEN 360 | |
15 | ||
16 | class HttpMethod; | |
17 | ||
18 | class CircleBuf | |
19 | { | |
20 | unsigned char *Buf; | |
21 | unsigned long Size; | |
22 | unsigned long InP; | |
23 | unsigned long OutP; | |
24 | string OutQueue; | |
25 | unsigned long StrPos; | |
26 | unsigned long MaxGet; | |
27 | struct timeval Start; | |
28 | ||
29 | unsigned long LeftRead() | |
30 | { | |
31 | unsigned long Sz = Size - (InP - OutP); | |
32 | if (Sz > Size - (InP%Size)) | |
33 | Sz = Size - (InP%Size); | |
34 | return Sz; | |
35 | } | |
36 | unsigned long LeftWrite() | |
37 | { | |
38 | unsigned long Sz = InP - OutP; | |
39 | if (InP > MaxGet) | |
40 | Sz = MaxGet - OutP; | |
41 | if (Sz > Size - (OutP%Size)) | |
42 | Sz = Size - (OutP%Size); | |
43 | return Sz; | |
44 | } | |
45 | void FillOut(); | |
46 | ||
47 | public: | |
48 | ||
49 | MD5Summation *MD5; | |
50 | ||
51 | // Read data in | |
52 | bool Read(int Fd); | |
53 | bool Read(string Data); | |
54 | ||
55 | // Write data out | |
56 | bool Write(int Fd); | |
57 | bool WriteTillEl(string &Data,bool Single = false); | |
58 | ||
59 | // Control the write limit | |
60 | void Limit(long Max) {if (Max == -1) MaxGet = 0-1; else MaxGet = OutP + Max;} | |
61 | bool IsLimit() {return MaxGet == OutP;}; | |
62 | void Print() {cout << MaxGet << ',' << OutP << endl;}; | |
63 | ||
64 | // Test for free space in the buffer | |
65 | bool ReadSpace() {return Size - (InP - OutP) > 0;}; | |
66 | bool WriteSpace() {return InP - OutP > 0;}; | |
67 | ||
68 | // Dump everything | |
69 | void Reset(); | |
70 | void Stats(); | |
71 | ||
72 | CircleBuf(unsigned long Size); | |
73 | ~CircleBuf() {delete [] Buf;}; | |
74 | }; | |
75 | ||
76 | struct ServerState | |
77 | { | |
78 | // This is the last parsed Header Line | |
79 | unsigned int Major; | |
80 | unsigned int Minor; | |
81 | unsigned int Result; | |
82 | char Code[MAXLEN]; | |
83 | ||
92e889c8 | 84 | // These are some statistics from the last parsed header lines |
be4401bf AL |
85 | unsigned long Size; |
86 | signed long StartPos; | |
87 | time_t Date; | |
92e889c8 | 88 | bool HaveContent; |
be4401bf AL |
89 | enum {Chunked,Stream,Closes} Encoding; |
90 | enum {Header, Data} State; | |
91 | ||
92 | HttpMethod *Owner; | |
93 | ||
94 | // This is the connection itself. Output is data FROM the server | |
95 | CircleBuf In; | |
96 | CircleBuf Out; | |
97 | int ServerFd; | |
98 | URI ServerName; | |
99 | ||
100 | bool HeaderLine(string Line); | |
101 | bool Comp(URI Other) {return Other.Host == ServerName.Host && Other.Port == ServerName.Port;}; | |
102 | void Reset() {Major = 0; Minor = 0; Result = 0; Size = 0; StartPos = 0; | |
103 | Encoding = Closes; time(&Date); ServerFd = -1;}; | |
92e889c8 | 104 | int RunHeaders(); |
be4401bf AL |
105 | bool RunData(); |
106 | ||
107 | bool Open(); | |
108 | bool Close(); | |
109 | ||
110 | ServerState(URI Srv,HttpMethod *Owner); | |
111 | ~ServerState() {Close();}; | |
112 | }; | |
113 | ||
114 | class HttpMethod : public pkgAcqMethod | |
115 | { | |
116 | void SendReq(FetchItem *Itm,CircleBuf &Out); | |
117 | bool Go(bool ToFile,ServerState *Srv); | |
118 | bool Flush(ServerState *Srv); | |
119 | bool ServerDie(ServerState *Srv); | |
120 | int DealWithHeaders(FetchResult &Res,ServerState *Srv); | |
121 | ||
492f957a AL |
122 | // In the event of a fatal signal this file will be closed and timestamped. |
123 | static string FailFile; | |
124 | static int FailFd; | |
125 | static time_t FailTime; | |
126 | static void SigTerm(int); | |
127 | ||
be4401bf AL |
128 | public: |
129 | friend ServerState; | |
130 | ||
be4401bf AL |
131 | int Depth; |
132 | FileFd *File; | |
133 | ||
134 | int Loop(); | |
135 | ||
b98f2859 | 136 | HttpMethod() : pkgAcqMethod("1.2",Pipeline | SendConfig) |
be4401bf AL |
137 | { |
138 | Depth = 0; | |
be4401bf AL |
139 | File = 0; |
140 | Depth = 0; | |
141 | }; | |
142 | }; | |
143 | ||
144 | URI Proxy; | |
145 | ||
146 | #endif |