]> git.saurik.com Git - apt.git/blame - methods/server.cc
CMake: Do not use -lresolv if res_init exists in libc
[apt.git] / methods / server.cc
CommitLineData
7330f4df
DK
1// -*- mode: cpp; mode: fold -*-
2// Description /*{{{*/
3/* ######################################################################
4
5 HTTP and HTTPS share a lot of common code and these classes are
6 exactly the dumping ground for this common code
7
8 ##################################################################### */
9 /*}}}*/
10// Include Files /*{{{*/
11#include <config.h>
12
7330f4df
DK
13#include <apt-pkg/configuration.h>
14#include <apt-pkg/error.h>
453b82a3
DK
15#include <apt-pkg/fileutl.h>
16#include <apt-pkg/strutl.h>
7330f4df 17
453b82a3
DK
18#include <ctype.h>
19#include <signal.h>
20#include <stdio.h>
21#include <stdlib.h>
7330f4df
DK
22#include <sys/stat.h>
23#include <sys/time.h>
453b82a3 24#include <time.h>
7330f4df 25#include <unistd.h>
7330f4df 26#include <iostream>
453b82a3 27#include <limits>
7330f4df 28#include <map>
453b82a3
DK
29#include <string>
30#include <vector>
7330f4df 31
453b82a3 32#include "server.h"
7330f4df
DK
33
34#include <apti18n.h>
35 /*}}}*/
36using namespace std;
37
38string ServerMethod::FailFile;
39int ServerMethod::FailFd = -1;
40time_t ServerMethod::FailTime = 0;
41
42// ServerState::RunHeaders - Get the headers before the data /*{{{*/
43// ---------------------------------------------------------------------
44/* Returns 0 if things are OK, 1 if an IO error occurred and 2 if a header
45 parse error occurred */
9622b211
MV
46ServerState::RunHeadersResult ServerState::RunHeaders(FileFd * const File,
47 const std::string &Uri)
7330f4df 48{
ebdb6f18 49 Reset(false);
7330f4df
DK
50 Owner->Status(_("Waiting for headers"));
51
7330f4df
DK
52 do
53 {
54 string Data;
55 if (ReadHeaderLines(Data) == false)
56 continue;
57
58 if (Owner->Debug == true)
9622b211 59 clog << "Answer for: " << Uri << endl << Data;
7330f4df
DK
60
61 for (string::const_iterator I = Data.begin(); I < Data.end(); ++I)
62 {
63 string::const_iterator J = I;
64 for (; J != Data.end() && *J != '\n' && *J != '\r'; ++J);
65 if (HeaderLine(string(I,J)) == false)
66 return RUN_HEADERS_PARSE_ERROR;
67 I = J;
68 }
69
70 // 100 Continue is a Nop...
71 if (Result == 100)
72 continue;
73
1e3f4083 74 // Tidy up the connection persistence state.
7330f4df
DK
75 if (Encoding == Closes && HaveContent == true)
76 Persistent = false;
77
78 return RUN_HEADERS_OK;
79 }
80 while (LoadNextResponse(false, File) == true);
81
82 return RUN_HEADERS_IO_ERROR;
83}
84 /*}}}*/
85// ServerState::HeaderLine - Process a header line /*{{{*/
86// ---------------------------------------------------------------------
87/* */
88bool ServerState::HeaderLine(string Line)
89{
90 if (Line.empty() == true)
91 return true;
92
148c0491 93 if (Line.size() > 4 && stringcasecmp(Line.data(), Line.data()+4, "HTTP") == 0)
7330f4df
DK
94 {
95 // Evil servers return no version
96 if (Line[4] == '/')
97 {
98 int const elements = sscanf(Line.c_str(),"HTTP/%3u.%3u %3u%359[^\n]",&Major,&Minor,&Result,Code);
99 if (elements == 3)
100 {
101 Code[0] = '\0';
0c2dc43d 102 if (Owner != NULL && Owner->Debug == true)
b58e2c7c 103 clog << "HTTP server doesn't give Reason-Phrase for " << std::to_string(Result) << std::endl;
7330f4df
DK
104 }
105 else if (elements != 4)
106 return _error->Error(_("The HTTP server sent an invalid reply header"));
107 }
108 else
109 {
110 Major = 0;
111 Minor = 9;
112 if (sscanf(Line.c_str(),"HTTP %3u%359[^\n]",&Result,Code) != 2)
113 return _error->Error(_("The HTTP server sent an invalid reply header"));
114 }
115
1e3f4083 116 /* Check the HTTP response header to get the default persistence
7330f4df
DK
117 state. */
118 if (Major < 1)
119 Persistent = false;
120 else
121 {
122 if (Major == 1 && Minor == 0)
b6d88f39 123 {
7330f4df 124 Persistent = false;
b6d88f39 125 }
7330f4df 126 else
b6d88f39 127 {
7330f4df 128 Persistent = true;
b6d88f39
JAK
129 if (PipelineAllowed)
130 Pipeline = true;
131 }
7330f4df
DK
132 }
133
134 return true;
d3e8fbb3
DK
135 }
136
148c0491
DK
137 // Blah, some servers use "connection:closes", evil.
138 // and some even send empty header fields…
139 string::size_type Pos = Line.find(':');
140 if (Pos == string::npos)
141 return _error->Error(_("Bad header line"));
142 ++Pos;
143
144 // Parse off any trailing spaces between the : and the next word.
145 string::size_type Pos2 = Pos;
146 while (Pos2 < Line.length() && isspace_ascii(Line[Pos2]) != 0)
147 Pos2++;
148
149 string const Tag(Line,0,Pos);
150 string const Val(Line,Pos2);
151
7330f4df
DK
152 if (stringcasecmp(Tag,"Content-Length:") == 0)
153 {
154 if (Encoding == Closes)
155 Encoding = Stream;
156 HaveContent = true;
d3e8fbb3 157
ceafe8a6 158 unsigned long long * DownloadSizePtr = &DownloadSize;
4bba5a88 159 if (Result == 416 || (Result >= 300 && Result < 400))
ceafe8a6 160 DownloadSizePtr = &JunkSize;
7330f4df 161
ceafe8a6
MV
162 *DownloadSizePtr = strtoull(Val.c_str(), NULL, 10);
163 if (*DownloadSizePtr >= std::numeric_limits<unsigned long long>::max())
7330f4df 164 return _error->Errno("HeaderLine", _("The HTTP server sent an invalid Content-Length header"));
ceafe8a6 165 else if (*DownloadSizePtr == 0)
7330f4df 166 HaveContent = false;
ceafe8a6
MV
167
168 // On partial content (206) the Content-Length less than the real
169 // size, so do not set it here but leave that to the Content-Range
170 // header instead
6291f60e
MV
171 if(Result != 206 && TotalFileSize == 0)
172 TotalFileSize = DownloadSize;
ceafe8a6 173
7330f4df
DK
174 return true;
175 }
176
177 if (stringcasecmp(Tag,"Content-Type:") == 0)
178 {
179 HaveContent = true;
180 return true;
181 }
d3e8fbb3 182
7330f4df
DK
183 if (stringcasecmp(Tag,"Content-Range:") == 0)
184 {
185 HaveContent = true;
186
187 // §14.16 says 'byte-range-resp-spec' should be a '*' in case of 416
6291f60e 188 if (Result == 416 && sscanf(Val.c_str(), "bytes */%llu",&TotalFileSize) == 1)
ed793a19 189 ; // we got the expected filesize which is all we wanted
6291f60e 190 else if (sscanf(Val.c_str(),"bytes %llu-%*u/%llu",&StartPos,&TotalFileSize) != 2)
7330f4df 191 return _error->Error(_("The HTTP server sent an invalid Content-Range header"));
6291f60e 192 if ((unsigned long long)StartPos > TotalFileSize)
7330f4df 193 return _error->Error(_("This HTTP server has broken range support"));
ceafe8a6
MV
194
195 // figure out what we will download
6291f60e 196 DownloadSize = TotalFileSize - StartPos;
7330f4df
DK
197 return true;
198 }
d3e8fbb3 199
7330f4df
DK
200 if (stringcasecmp(Tag,"Transfer-Encoding:") == 0)
201 {
202 HaveContent = true;
203 if (stringcasecmp(Val,"chunked") == 0)
d3e8fbb3 204 Encoding = Chunked;
7330f4df
DK
205 return true;
206 }
207
208 if (stringcasecmp(Tag,"Connection:") == 0)
209 {
210 if (stringcasecmp(Val,"close") == 0)
9714d522 211 {
7330f4df 212 Persistent = false;
9714d522
DK
213 Pipeline = false;
214 /* Some servers send error pages (as they are dynamically generated)
215 for simplicity via a connection close instead of e.g. chunked,
216 so assuming an always closing server only if we get a file + close */
217 if (Result >= 200 && Result < 300)
218 PipelineAllowed = false;
219 }
220 else if (stringcasecmp(Val,"keep-alive") == 0)
7330f4df
DK
221 Persistent = true;
222 return true;
223 }
d3e8fbb3 224
7330f4df
DK
225 if (stringcasecmp(Tag,"Last-Modified:") == 0)
226 {
227 if (RFC1123StrToTime(Val.c_str(), Date) == false)
228 return _error->Error(_("Unknown date format"));
229 return true;
230 }
231
232 if (stringcasecmp(Tag,"Location:") == 0)
233 {
234 Location = Val;
235 return true;
236 }
237
d94b1d80
DK
238 if (stringcasecmp(Tag, "Accept-Ranges:") == 0)
239 {
240 std::string ranges = ',' + Val + ',';
241 ranges.erase(std::remove(ranges.begin(), ranges.end(), ' '), ranges.end());
242 if (ranges.find(",bytes,") == std::string::npos)
243 RangesAllowed = false;
244 return true;
245 }
246
7330f4df
DK
247 return true;
248}
249 /*}}}*/
250// ServerState::ServerState - Constructor /*{{{*/
2651f1c0
DK
251ServerState::ServerState(URI Srv, ServerMethod *Owner) :
252 DownloadSize(0), ServerName(Srv), TimeOut(120), Owner(Owner)
7330f4df
DK
253{
254 Reset();
255}
256 /*}}}*/
34faa8f7
DK
257bool ServerState::AddPartialFileToHashes(FileFd &File) /*{{{*/
258{
259 File.Truncate(StartPos);
260 return GetHashes()->AddFD(File, StartPos);
261}
262 /*}}}*/
ebdb6f18
DK
263void ServerState::Reset(bool const Everything) /*{{{*/
264{
265 Major = 0; Minor = 0; Result = 0; Code[0] = '\0';
266 TotalFileSize = 0; JunkSize = 0; StartPos = 0;
267 Encoding = Closes; time(&Date); HaveContent = false;
268 State = Header; MaximumSize = 0;
269 if (Everything)
270 {
271 Persistent = false; Pipeline = false; PipelineAllowed = true;
d94b1d80 272 RangesAllowed = true;
ebdb6f18
DK
273 }
274}
275 /*}}}*/
7330f4df 276
7330f4df
DK
277// ServerMethod::DealWithHeaders - Handle the retrieved header data /*{{{*/
278// ---------------------------------------------------------------------
279/* We look at the header data we got back from the server and decide what
280 to do. Returns DealWithHeadersResult (see http.h for details).
281 */
282ServerMethod::DealWithHeadersResult
283ServerMethod::DealWithHeaders(FetchResult &Res)
284{
285 // Not Modified
286 if (Server->Result == 304)
287 {
ce1f3a2c 288 RemoveFile("server", Queue->DestFile);
7330f4df
DK
289 Res.IMSHit = true;
290 Res.LastModified = Queue->LastModified;
4bba5a88 291 Res.Size = 0;
7330f4df
DK
292 return IMS_HIT;
293 }
dcbb364f 294
7330f4df
DK
295 /* Redirect
296 *
297 * Note that it is only OK for us to treat all redirection the same
298 * because we *always* use GET, not other HTTP methods. There are
299 * three redirection codes for which it is not appropriate that we
300 * redirect. Pass on those codes so the error handling kicks in.
301 */
302 if (AllowRedirect
303 && (Server->Result > 300 && Server->Result < 400)
304 && (Server->Result != 300 // Multiple Choices
305 && Server->Result != 304 // Not Modified
306 && Server->Result != 306)) // (Not part of HTTP/1.1, reserved)
307 {
4bba5a88
DK
308 if (Server->Location.empty() == true)
309 ;
7330f4df
DK
310 else if (Server->Location[0] == '/' && Queue->Uri.empty() == false)
311 {
312 URI Uri = Queue->Uri;
313 if (Uri.Host.empty() == false)
314 NextURI = URI::SiteOnly(Uri);
315 else
316 NextURI.clear();
317 NextURI.append(DeQuoteString(Server->Location));
57401c48
DK
318 if (Queue->Uri == NextURI)
319 {
320 SetFailReason("RedirectionLoop");
321 _error->Error("Redirection loop encountered");
322 if (Server->HaveContent == true)
323 return ERROR_WITH_CONTENT_PAGE;
324 return ERROR_UNRECOVERABLE;
325 }
7330f4df
DK
326 return TRY_AGAIN_OR_REDIRECT;
327 }
328 else
329 {
9082a1fc
DK
330 NextURI = DeQuoteString(Server->Location);
331 URI tmpURI = NextURI;
30060442 332 if (tmpURI.Access.find('+') != std::string::npos)
57401c48 333 {
30060442
DK
334 _error->Error("Server tried to trick us into using a specific implementation: %s", tmpURI.Access.c_str());
335 if (Server->HaveContent == true)
336 return ERROR_WITH_CONTENT_PAGE;
337 return ERROR_UNRECOVERABLE;
338 }
339 URI Uri = Queue->Uri;
340 if (Binary.find('+') != std::string::npos)
341 {
342 auto base = Binary.substr(0, Binary.find('+'));
343 if (base != tmpURI.Access)
344 {
345 tmpURI.Access = base + '+' + tmpURI.Access;
346 if (tmpURI.Access == Binary)
347 {
348 std::string tmpAccess = Uri.Access;
349 std::swap(tmpURI.Access, Uri.Access);
350 NextURI = tmpURI;
351 std::swap(tmpURI.Access, Uri.Access);
352 }
353 else
354 NextURI = tmpURI;
355 }
57401c48
DK
356 }
357 if (Queue->Uri == NextURI)
358 {
359 SetFailReason("RedirectionLoop");
360 _error->Error("Redirection loop encountered");
361 if (Server->HaveContent == true)
362 return ERROR_WITH_CONTENT_PAGE;
363 return ERROR_UNRECOVERABLE;
364 }
30060442 365 Uri.Access = Binary;
9082a1fc
DK
366 // same protocol redirects are okay
367 if (tmpURI.Access == Uri.Access)
368 return TRY_AGAIN_OR_REDIRECT;
369 // as well as http to https
4bba5a88 370 else if ((Uri.Access == "http" || Uri.Access == "https+http") && tmpURI.Access == "https")
9082a1fc 371 return TRY_AGAIN_OR_REDIRECT;
4bba5a88 372 else
30060442
DK
373 {
374 auto const tmpplus = tmpURI.Access.find('+');
375 if (tmpplus != std::string::npos && tmpURI.Access.substr(tmpplus + 1) == "https")
376 {
377 auto const uriplus = Uri.Access.find('+');
378 if (uriplus == std::string::npos)
379 {
380 if (Uri.Access == tmpURI.Access.substr(0, tmpplus)) // foo -> foo+https
381 return TRY_AGAIN_OR_REDIRECT;
382 }
383 else if (Uri.Access.substr(uriplus + 1) == "http" &&
384 Uri.Access.substr(0, uriplus) == tmpURI.Access.substr(0, tmpplus)) // foo+http -> foo+https
385 return TRY_AGAIN_OR_REDIRECT;
386 }
387 }
388 _error->Error("Redirection from %s to '%s' is forbidden", Uri.Access.c_str(), NextURI.c_str());
7330f4df
DK
389 }
390 /* else pass through for error message */
391 }
392 // retry after an invalid range response without partial data
393 else if (Server->Result == 416)
394 {
395 struct stat SBuf;
396 if (stat(Queue->DestFile.c_str(),&SBuf) >= 0 && SBuf.st_size > 0)
397 {
dcbb364f
DK
398 bool partialHit = false;
399 if (Queue->ExpectedHashes.usable() == true)
400 {
401 Hashes resultHashes(Queue->ExpectedHashes);
402 FileFd file(Queue->DestFile, FileFd::ReadOnly);
4fc6b757 403 Server->TotalFileSize = file.FileSize();
dcbb364f
DK
404 Server->Date = file.ModificationTime();
405 resultHashes.AddFD(file);
406 HashStringList const hashList = resultHashes.GetHashStringList();
407 partialHit = (Queue->ExpectedHashes == hashList);
408 }
4fc6b757 409 else if ((unsigned long long)SBuf.st_size == Server->TotalFileSize)
dcbb364f
DK
410 partialHit = true;
411 if (partialHit == true)
7330f4df
DK
412 {
413 // the file is completely downloaded, but was not moved
ed793a19
DK
414 if (Server->HaveContent == true)
415 {
4bba5a88
DK
416 // nuke the sent error page
417 Server->RunDataToDevNull();
418 Server->HaveContent = false;
ed793a19 419 }
6291f60e 420 Server->StartPos = Server->TotalFileSize;
7330f4df 421 Server->Result = 200;
7330f4df 422 }
ce1f3a2c 423 else if (RemoveFile("server", Queue->DestFile))
7330f4df
DK
424 {
425 NextURI = Queue->Uri;
426 return TRY_AGAIN_OR_REDIRECT;
427 }
428 }
429 }
430
3a8776a3 431 /* We have a reply we don't handle. This should indicate a perm server
7330f4df
DK
432 failure */
433 if (Server->Result < 200 || Server->Result >= 300)
434 {
4bba5a88
DK
435 if (_error->PendingError() == false)
436 {
437 std::string err;
438 strprintf(err, "HttpError%u", Server->Result);
439 SetFailReason(err);
440 _error->Error("%u %s", Server->Result, Server->Code);
441 }
7330f4df
DK
442 if (Server->HaveContent == true)
443 return ERROR_WITH_CONTENT_PAGE;
444 return ERROR_UNRECOVERABLE;
445 }
446
447 // This is some sort of 2xx 'data follows' reply
448 Res.LastModified = Server->Date;
6291f60e 449 Res.Size = Server->TotalFileSize;
7330f4df
DK
450 return FILE_IS_OPEN;
451}
452 /*}}}*/
453// ServerMethod::SigTerm - Handle a fatal signal /*{{{*/
454// ---------------------------------------------------------------------
1e3f4083 455/* This closes and timestamps the open file. This is necessary to get
7330f4df
DK
456 resume behavoir on user abort */
457void ServerMethod::SigTerm(int)
458{
459 if (FailFd == -1)
460 _exit(100);
9ce3cfc9 461
246bbb61 462 struct timeval times[2];
9ce3cfc9
DK
463 times[0].tv_sec = FailTime;
464 times[1].tv_sec = FailTime;
246bbb61
DK
465 times[0].tv_usec = times[1].tv_usec = 0;
466 utimes(FailFile.c_str(), times);
7330f4df 467 close(FailFd);
9ce3cfc9 468
7330f4df
DK
469 _exit(100);
470}
471 /*}}}*/
472// ServerMethod::Fetch - Fetch an item /*{{{*/
473// ---------------------------------------------------------------------
474/* This adds an item to the pipeline. We keep the pipeline at a fixed
475 depth. */
476bool ServerMethod::Fetch(FetchItem *)
477{
742f67ea 478 if (Server == nullptr || QueueBack == nullptr)
7330f4df
DK
479 return true;
480
742f67ea
DK
481 // If pipelining is disabled, we only queue 1 request
482 auto const AllowedDepth = Server->Pipeline ? PipelineDepth : 0;
483 // how deep is our pipeline currently?
484 decltype(PipelineDepth) CurrentDepth = 0;
485 for (FetchItem const *I = Queue; I != QueueBack; I = I->Next)
486 ++CurrentDepth;
353b7bab
DK
487 if (CurrentDepth > AllowedDepth)
488 return true;
742f67ea
DK
489
490 do {
7330f4df 491 // Make sure we stick with the same server
742f67ea
DK
492 if (Server->Comp(QueueBack->Uri) == false)
493 break;
494
495 bool const UsableHashes = QueueBack->ExpectedHashes.usable();
496 // if we have no hashes, do at most one such request
497 // as we can't fixup pipeling misbehaviors otherwise
498 if (CurrentDepth != 0 && UsableHashes == false)
7330f4df 499 break;
742f67ea
DK
500
501 if (UsableHashes && FileExists(QueueBack->DestFile))
7330f4df 502 {
742f67ea
DK
503 FileFd partial(QueueBack->DestFile, FileFd::ReadOnly);
504 Hashes wehave(QueueBack->ExpectedHashes);
505 if (QueueBack->ExpectedHashes.FileSize() == partial.FileSize())
506 {
507 if (wehave.AddFD(partial) &&
508 wehave.GetHashStringList() == QueueBack->ExpectedHashes)
509 {
510 FetchResult Res;
511 Res.Filename = QueueBack->DestFile;
512 Res.ResumePoint = QueueBack->ExpectedHashes.FileSize();
513 URIStart(Res);
514 // move item to the start of the queue as URIDone will
515 // always dequeued the first item in the queue
516 if (Queue != QueueBack)
517 {
518 FetchItem *Prev = Queue;
519 for (; Prev->Next != QueueBack; Prev = Prev->Next)
520 /* look for the previous queue item */;
521 Prev->Next = QueueBack->Next;
522 QueueBack->Next = Queue;
523 Queue = QueueBack;
524 QueueBack = Prev->Next;
525 }
526 Res.TakeHashes(wehave);
527 URIDone(Res);
528 continue;
529 }
530 else
531 RemoveFile("Fetch-Partial", QueueBack->DestFile);
532 }
7330f4df 533 }
742f67ea
DK
534 auto const Tmp = QueueBack;
535 QueueBack = QueueBack->Next;
536 SendReq(Tmp);
537 ++CurrentDepth;
538 } while (CurrentDepth <= AllowedDepth && QueueBack != nullptr);
539
7330f4df 540 return true;
d3e8fbb3 541}
7330f4df
DK
542 /*}}}*/
543// ServerMethod::Loop - Main loop /*{{{*/
544int ServerMethod::Loop()
545{
7330f4df
DK
546 signal(SIGTERM,SigTerm);
547 signal(SIGINT,SigTerm);
548
549 Server = 0;
550
551 int FailCounter = 0;
552 while (1)
553 {
554 // We have no commands, wait for some to arrive
555 if (Queue == 0)
556 {
557 if (WaitFd(STDIN_FILENO) == false)
558 return 0;
559 }
560
561 /* Run messages, we can accept 0 (no message) if we didn't
562 do a WaitFd above.. Otherwise the FD is closed. */
563 int Result = Run(true);
564 if (Result != -1 && (Result != 0 || Queue == 0))
565 {
566 if(FailReason.empty() == false ||
30060442 567 ConfigFindB("DependOnSTDIN", true) == true)
7330f4df
DK
568 return 100;
569 else
570 return 0;
571 }
572
573 if (Queue == 0)
574 continue;
575
576 // Connect to the server
577 if (Server == 0 || Server->Comp(Queue->Uri) == false)
30060442 578 {
7330f4df 579 Server = CreateServerState(Queue->Uri);
30060442
DK
580 setPostfixForMethodNames(::URI(Queue->Uri).Host.c_str());
581 AllowRedirect = ConfigFindB("AllowRedirect", true);
582 PipelineDepth = ConfigFindI("Pipeline-Depth", 10);
583 Debug = DebugEnabled();
584 }
830a1b8c 585
7330f4df
DK
586 /* If the server has explicitly said this is the last connection
587 then we pre-emptively shut down the pipeline and tear down
588 the connection. This will speed up HTTP/1.0 servers a tad
589 since we don't have to wait for the close sequence to
590 complete */
591 if (Server->Persistent == false)
592 Server->Close();
593
594 // Reset the pipeline
595 if (Server->IsOpen() == false)
596 QueueBack = Queue;
597
598 // Connnect to the host
599 if (Server->Open() == false)
600 {
601 Fail(true);
830a1b8c 602 Server = nullptr;
7330f4df
DK
603 continue;
604 }
605
606 // Fill the pipeline.
607 Fetch(0);
608
609 // Fetch the next URL header data from the server.
9622b211 610 switch (Server->RunHeaders(File, Queue->Uri))
7330f4df
DK
611 {
612 case ServerState::RUN_HEADERS_OK:
613 break;
614
615 // The header data is bad
616 case ServerState::RUN_HEADERS_PARSE_ERROR:
617 {
618 _error->Error(_("Bad header data"));
619 Fail(true);
cc0a4c82 620 Server->Close();
7330f4df
DK
621 RotateDNS();
622 continue;
623 }
624
625 // The server closed a connection during the header get..
626 default:
627 case ServerState::RUN_HEADERS_IO_ERROR:
628 {
629 FailCounter++;
630 _error->Discard();
631 Server->Close();
632 Server->Pipeline = false;
b6d88f39 633 Server->PipelineAllowed = false;
7330f4df
DK
634
635 if (FailCounter >= 2)
636 {
637 Fail(_("Connection failed"),true);
638 FailCounter = 0;
639 }
640
641 RotateDNS();
642 continue;
643 }
644 };
645
646 // Decide what to do.
647 FetchResult Res;
648 Res.Filename = Queue->DestFile;
649 switch (DealWithHeaders(Res))
650 {
651 // Ok, the file is Open
652 case FILE_IS_OPEN:
653 {
654 URIStart(Res);
655
656 // Run the data
657 bool Result = true;
dcd5856b
MV
658
659 // ensure we don't fetch too much
f2b47ba2
MV
660 // we could do "Server->MaximumSize = Queue->MaximumSize" here
661 // but that would break the clever pipeline messup detection
662 // so instead we use the size of the biggest item in the queue
663 Server->MaximumSize = FindMaximumObjectSizeInQueue();
dcd5856b 664
7330f4df
DK
665 if (Server->HaveContent)
666 Result = Server->RunData(File);
667
668 /* If the server is sending back sizeless responses then fill in
669 the size now */
670 if (Res.Size == 0)
671 Res.Size = File->Size();
672
673 // Close the file, destroy the FD object and timestamp it
674 FailFd = -1;
675 delete File;
676 File = 0;
677
678 // Timestamp
246bbb61 679 struct timeval times[2];
9ce3cfc9 680 times[0].tv_sec = times[1].tv_sec = Server->Date;
246bbb61
DK
681 times[0].tv_usec = times[1].tv_usec = 0;
682 utimes(Queue->DestFile.c_str(), times);
7330f4df
DK
683
684 // Send status to APT
685 if (Result == true)
686 {
895417ef
DK
687 Hashes * const resultHashes = Server->GetHashes();
688 HashStringList const hashList = resultHashes->GetHashStringList();
689 if (PipelineDepth != 0 && Queue->ExpectedHashes.usable() == true && Queue->ExpectedHashes != hashList)
690 {
691 // we did not get the expected hash… mhhh:
692 // could it be that server/proxy messed up pipelining?
693 FetchItem * BeforeI = Queue;
694 for (FetchItem *I = Queue->Next; I != 0 && I != QueueBack; I = I->Next)
695 {
696 if (I->ExpectedHashes.usable() == true && I->ExpectedHashes == hashList)
697 {
698 // yes, he did! Disable pipelining and rewrite queue
699 if (Server->Pipeline == true)
700 {
b9c20219 701 Warning(_("Automatically disabled %s due to incorrect response from server/proxy. (man 5 apt.conf)"), "Acquire::http::Pipeline-Depth");
895417ef 702 Server->Pipeline = false;
b6d88f39 703 Server->PipelineAllowed = false;
895417ef
DK
704 // we keep the PipelineDepth value so that the rest of the queue can be fixed up as well
705 }
706 Rename(Res.Filename, I->DestFile);
707 Res.Filename = I->DestFile;
708 BeforeI->Next = I->Next;
709 I->Next = Queue;
710 Queue = I;
711 break;
712 }
713 BeforeI = I;
714 }
715 }
716 Res.TakeHashes(*resultHashes);
7330f4df
DK
717 URIDone(Res);
718 }
719 else
720 {
721 if (Server->IsOpen() == false)
722 {
723 FailCounter++;
724 _error->Discard();
725 Server->Close();
726
727 if (FailCounter >= 2)
728 {
729 Fail(_("Connection failed"),true);
730 FailCounter = 0;
731 }
732
733 QueueBack = Queue;
734 }
735 else
a2d40703
MV
736 {
737 Server->Close();
7330f4df 738 Fail(true);
a2d40703 739 }
7330f4df
DK
740 }
741 break;
742 }
743
744 // IMS hit
745 case IMS_HIT:
746 {
747 URIDone(Res);
748 break;
749 }
750
751 // Hard server error, not found or something
752 case ERROR_UNRECOVERABLE:
753 {
754 Fail();
755 break;
756 }
757
758 // Hard internal error, kill the connection and fail
759 case ERROR_NOT_FROM_SERVER:
760 {
761 delete File;
762 File = 0;
763
764 Fail();
765 RotateDNS();
766 Server->Close();
767 break;
768 }
769
770 // We need to flush the data, the header is like a 404 w/ error text
771 case ERROR_WITH_CONTENT_PAGE:
772 {
773 Fail();
4bba5a88 774 Server->RunDataToDevNull();
7330f4df
DK
775 break;
776 }
57401c48
DK
777
778 // Try again with a new URL
779 case TRY_AGAIN_OR_REDIRECT:
780 {
781 // Clear rest of response if there is content
782 if (Server->HaveContent)
783 Server->RunDataToDevNull();
784 Redirect(NextURI);
785 break;
786 }
7330f4df
DK
787
788 default:
789 Fail(_("Internal error"));
790 break;
791 }
792
793 FailCounter = 0;
794 }
795
796 return 0;
f2b47ba2
MV
797}
798 /*}}}*/
830a1b8c 799unsigned long long ServerMethod::FindMaximumObjectSizeInQueue() const /*{{{*/
f2b47ba2
MV
800{
801 unsigned long long MaxSizeInQueue = 0;
8bb043fc 802 for (FetchItem *I = Queue; I != 0 && I != QueueBack; I = I->Next)
f2b47ba2
MV
803 MaxSizeInQueue = std::max(MaxSizeInQueue, I->MaximumSize);
804 return MaxSizeInQueue;
7330f4df
DK
805}
806 /*}}}*/
30060442
DK
807ServerMethod::ServerMethod(std::string &&Binary, char const * const Ver,unsigned long const Flags) :/*{{{*/
808 aptMethod(std::move(Binary), Ver, Flags), Server(nullptr), File(NULL), PipelineDepth(10),
830a1b8c
DK
809 AllowRedirect(false), Debug(false)
810{
811}
812 /*}}}*/
0568d325
DK
813bool ServerMethod::Configuration(std::string Message) /*{{{*/
814{
815 if (aptMethod::Configuration(Message) == false)
816 return false;
817
818 _config->CndSet("Acquire::tor::Proxy",
819 "socks5h://apt-transport-tor@localhost:9050");
820 return true;
821}
822 /*}}}*/
823bool ServerMethod::AddProxyAuth(URI &Proxy, URI const &Server) const /*{{{*/
824{
825 if (std::find(methodNames.begin(), methodNames.end(), "tor") != methodNames.end() &&
826 Proxy.User == "apt-transport-tor" && Proxy.Password.empty())
827 {
828 std::string pass = Server.Host;
829 pass.erase(std::remove_if(pass.begin(), pass.end(), [](char const c) { return std::isalnum(c) == 0; }), pass.end());
830 if (pass.length() > 255)
831 Proxy.Password = pass.substr(0, 255);
832 else
833 Proxy.Password = std::move(pass);
834 }
835 // FIXME: should we support auth.conf for proxies?
836 return true;
837}
838 /*}}}*/