]> git.saurik.com Git - apt.git/blob - methods/http.cc
Pass ExpectedSize to tthe backend method
[apt.git] / methods / http.cc
1 // -*- mode: cpp; mode: fold -*-
2 // Description /*{{{*/
3 // $Id: http.cc,v 1.59 2004/05/08 19:42:35 mdz Exp $
4 /* ######################################################################
5
6 HTTP Acquire Method - This is the HTTP acquire method for APT.
7
8 It uses HTTP/1.1 and many of the fancy options there-in, such as
9 pipelining, range, if-range and so on.
10
11 It is based on a doubly buffered select loop. A groupe of requests are
12 fed into a single output buffer that is constantly fed out the
13 socket. This provides ideal pipelining as in many cases all of the
14 requests will fit into a single packet. The input socket is buffered
15 the same way and fed into the fd for the file (may be a pipe in future).
16
17 This double buffering provides fairly substantial transfer rates,
18 compared to wget the http method is about 4% faster. Most importantly,
19 when HTTP is compared with FTP as a protocol the speed difference is
20 huge. In tests over the internet from two sites to llug (via ATM) this
21 program got 230k/s sustained http transfer rates. FTP on the other
22 hand topped out at 170k/s. That combined with the time to setup the
23 FTP connection makes HTTP a vastly superior protocol.
24
25 ##################################################################### */
26 /*}}}*/
27 // Include Files /*{{{*/
28 #include <config.h>
29
30 #include <apt-pkg/fileutl.h>
31 #include <apt-pkg/acquire-method.h>
32 #include <apt-pkg/configuration.h>
33 #include <apt-pkg/error.h>
34 #include <apt-pkg/hashes.h>
35 #include <apt-pkg/netrc.h>
36 #include <apt-pkg/strutl.h>
37
38 #include <stddef.h>
39 #include <stdlib.h>
40 #include <sys/select.h>
41 #include <cstring>
42 #include <sys/stat.h>
43 #include <sys/time.h>
44 #include <unistd.h>
45 #include <stdio.h>
46 #include <errno.h>
47 #include <iostream>
48 #include <sstream>
49
50 #include "config.h"
51 #include "connect.h"
52 #include "http.h"
53
54 #include <apti18n.h>
55 /*}}}*/
56 using namespace std;
57
58 unsigned long long CircleBuf::BwReadLimit=0;
59 unsigned long long CircleBuf::BwTickReadData=0;
60 struct timeval CircleBuf::BwReadTick={0,0};
61 const unsigned int CircleBuf::BW_HZ=10;
62
63 // CircleBuf::CircleBuf - Circular input buffer /*{{{*/
64 // ---------------------------------------------------------------------
65 /* */
66 CircleBuf::CircleBuf(unsigned long long Size)
67 : Size(Size), Hash(0), TotalWriten(0)
68 {
69 Buf = new unsigned char[Size];
70 Reset();
71
72 CircleBuf::BwReadLimit = _config->FindI("Acquire::http::Dl-Limit",0)*1024;
73 }
74 /*}}}*/
75 // CircleBuf::Reset - Reset to the default state /*{{{*/
76 // ---------------------------------------------------------------------
77 /* */
78 void CircleBuf::Reset()
79 {
80 InP = 0;
81 OutP = 0;
82 StrPos = 0;
83 TotalWriten = 0;
84 MaxGet = (unsigned long long)-1;
85 OutQueue = string();
86 if (Hash != 0)
87 {
88 delete Hash;
89 Hash = new Hashes;
90 }
91 }
92 /*}}}*/
93 // CircleBuf::Read - Read from a FD into the circular buffer /*{{{*/
94 // ---------------------------------------------------------------------
95 /* This fills up the buffer with as much data as is in the FD, assuming it
96 is non-blocking.. */
97 bool CircleBuf::Read(int Fd)
98 {
99 while (1)
100 {
101 // Woops, buffer is full
102 if (InP - OutP == Size)
103 return true;
104
105 // what's left to read in this tick
106 unsigned long long const BwReadMax = CircleBuf::BwReadLimit/BW_HZ;
107
108 if(CircleBuf::BwReadLimit) {
109 struct timeval now;
110 gettimeofday(&now,0);
111
112 unsigned long long d = (now.tv_sec-CircleBuf::BwReadTick.tv_sec)*1000000 +
113 now.tv_usec-CircleBuf::BwReadTick.tv_usec;
114 if(d > 1000000/BW_HZ) {
115 CircleBuf::BwReadTick = now;
116 CircleBuf::BwTickReadData = 0;
117 }
118
119 if(CircleBuf::BwTickReadData >= BwReadMax) {
120 usleep(1000000/BW_HZ);
121 return true;
122 }
123 }
124
125 // Write the buffer segment
126 ssize_t Res;
127 if(CircleBuf::BwReadLimit) {
128 Res = read(Fd,Buf + (InP%Size),
129 BwReadMax > LeftRead() ? LeftRead() : BwReadMax);
130 } else
131 Res = read(Fd,Buf + (InP%Size),LeftRead());
132
133 if(Res > 0 && BwReadLimit > 0)
134 CircleBuf::BwTickReadData += Res;
135
136 if (Res == 0)
137 return false;
138 if (Res < 0)
139 {
140 if (errno == EAGAIN)
141 return true;
142 return false;
143 }
144
145 if (InP == 0)
146 gettimeofday(&Start,0);
147 InP += Res;
148 }
149 }
150 /*}}}*/
151 // CircleBuf::Read - Put the string into the buffer /*{{{*/
152 // ---------------------------------------------------------------------
153 /* This will hold the string in and fill the buffer with it as it empties */
154 bool CircleBuf::Read(string Data)
155 {
156 OutQueue += Data;
157 FillOut();
158 return true;
159 }
160 /*}}}*/
161 // CircleBuf::FillOut - Fill the buffer from the output queue /*{{{*/
162 // ---------------------------------------------------------------------
163 /* */
164 void CircleBuf::FillOut()
165 {
166 if (OutQueue.empty() == true)
167 return;
168 while (1)
169 {
170 // Woops, buffer is full
171 if (InP - OutP == Size)
172 return;
173
174 // Write the buffer segment
175 unsigned long long Sz = LeftRead();
176 if (OutQueue.length() - StrPos < Sz)
177 Sz = OutQueue.length() - StrPos;
178 memcpy(Buf + (InP%Size),OutQueue.c_str() + StrPos,Sz);
179
180 // Advance
181 StrPos += Sz;
182 InP += Sz;
183 if (OutQueue.length() == StrPos)
184 {
185 StrPos = 0;
186 OutQueue = "";
187 return;
188 }
189 }
190 }
191 /*}}}*/
192 // CircleBuf::Write - Write from the buffer into a FD /*{{{*/
193 // ---------------------------------------------------------------------
194 /* This empties the buffer into the FD. */
195 bool CircleBuf::Write(int Fd)
196 {
197 while (1)
198 {
199 FillOut();
200
201 // Woops, buffer is empty
202 if (OutP == InP)
203 return true;
204
205 if (OutP == MaxGet)
206 return true;
207
208 // Write the buffer segment
209 ssize_t Res;
210 Res = write(Fd,Buf + (OutP%Size),LeftWrite());
211
212 if (Res == 0)
213 return false;
214 if (Res < 0)
215 {
216 if (errno == EAGAIN)
217 return true;
218
219 return false;
220 }
221
222 TotalWriten += Res;
223
224 if (Hash != 0)
225 Hash->Add(Buf + (OutP%Size),Res);
226
227 OutP += Res;
228 }
229 }
230 /*}}}*/
231 // CircleBuf::WriteTillEl - Write from the buffer to a string /*{{{*/
232 // ---------------------------------------------------------------------
233 /* This copies till the first empty line */
234 bool CircleBuf::WriteTillEl(string &Data,bool Single)
235 {
236 // We cheat and assume it is unneeded to have more than one buffer load
237 for (unsigned long long I = OutP; I < InP; I++)
238 {
239 if (Buf[I%Size] != '\n')
240 continue;
241 ++I;
242
243 if (Single == false)
244 {
245 if (I < InP && Buf[I%Size] == '\r')
246 ++I;
247 if (I >= InP || Buf[I%Size] != '\n')
248 continue;
249 ++I;
250 }
251
252 Data = "";
253 while (OutP < I)
254 {
255 unsigned long long Sz = LeftWrite();
256 if (Sz == 0)
257 return false;
258 if (I - OutP < Sz)
259 Sz = I - OutP;
260 Data += string((char *)(Buf + (OutP%Size)),Sz);
261 OutP += Sz;
262 }
263 return true;
264 }
265 return false;
266 }
267 /*}}}*/
268 // CircleBuf::Stats - Print out stats information /*{{{*/
269 // ---------------------------------------------------------------------
270 /* */
271 void CircleBuf::Stats()
272 {
273 if (InP == 0)
274 return;
275
276 struct timeval Stop;
277 gettimeofday(&Stop,0);
278 /* float Diff = Stop.tv_sec - Start.tv_sec +
279 (float)(Stop.tv_usec - Start.tv_usec)/1000000;
280 clog << "Got " << InP << " in " << Diff << " at " << InP/Diff << endl;*/
281 }
282 /*}}}*/
283 CircleBuf::~CircleBuf()
284 {
285 delete [] Buf;
286 delete Hash;
287 }
288
289 // HttpServerState::HttpServerState - Constructor /*{{{*/
290 HttpServerState::HttpServerState(URI Srv,HttpMethod *Owner) : ServerState(Srv, Owner), In(64*1024), Out(4*1024)
291 {
292 TimeOut = _config->FindI("Acquire::http::Timeout",TimeOut);
293 Reset();
294 }
295 /*}}}*/
296 // HttpServerState::Open - Open a connection to the server /*{{{*/
297 // ---------------------------------------------------------------------
298 /* This opens a connection to the server. */
299 bool HttpServerState::Open()
300 {
301 // Use the already open connection if possible.
302 if (ServerFd != -1)
303 return true;
304
305 Close();
306 In.Reset();
307 Out.Reset();
308 Persistent = true;
309
310 // Determine the proxy setting
311 string SpecificProxy = _config->Find("Acquire::http::Proxy::" + ServerName.Host);
312 if (!SpecificProxy.empty())
313 {
314 if (SpecificProxy == "DIRECT")
315 Proxy = "";
316 else
317 Proxy = SpecificProxy;
318 }
319 else
320 {
321 string DefProxy = _config->Find("Acquire::http::Proxy");
322 if (!DefProxy.empty())
323 {
324 Proxy = DefProxy;
325 }
326 else
327 {
328 char* result = getenv("http_proxy");
329 Proxy = result ? result : "";
330 }
331 }
332
333 // Parse no_proxy, a , separated list of domains
334 if (getenv("no_proxy") != 0)
335 {
336 if (CheckDomainList(ServerName.Host,getenv("no_proxy")) == true)
337 Proxy = "";
338 }
339
340 // Determine what host and port to use based on the proxy settings
341 int Port = 0;
342 string Host;
343 if (Proxy.empty() == true || Proxy.Host.empty() == true)
344 {
345 if (ServerName.Port != 0)
346 Port = ServerName.Port;
347 Host = ServerName.Host;
348 }
349 else
350 {
351 if (Proxy.Port != 0)
352 Port = Proxy.Port;
353 Host = Proxy.Host;
354 }
355
356 // Connect to the remote server
357 if (Connect(Host,Port,"http",80,ServerFd,TimeOut,Owner) == false)
358 return false;
359
360 return true;
361 }
362 /*}}}*/
363 // HttpServerState::Close - Close a connection to the server /*{{{*/
364 // ---------------------------------------------------------------------
365 /* */
366 bool HttpServerState::Close()
367 {
368 close(ServerFd);
369 ServerFd = -1;
370 return true;
371 }
372 /*}}}*/
373 // HttpServerState::RunData - Transfer the data from the socket /*{{{*/
374 bool HttpServerState::RunData(FileFd * const File)
375 {
376 State = Data;
377
378 // Chunked transfer encoding is fun..
379 if (Encoding == Chunked)
380 {
381 while (1)
382 {
383 // Grab the block size
384 bool Last = true;
385 string Data;
386 In.Limit(-1);
387 do
388 {
389 if (In.WriteTillEl(Data,true) == true)
390 break;
391 }
392 while ((Last = Go(false, File)) == true);
393
394 if (Last == false)
395 return false;
396
397 // See if we are done
398 unsigned long long Len = strtoull(Data.c_str(),0,16);
399 if (Len == 0)
400 {
401 In.Limit(-1);
402
403 // We have to remove the entity trailer
404 Last = true;
405 do
406 {
407 if (In.WriteTillEl(Data,true) == true && Data.length() <= 2)
408 break;
409 }
410 while ((Last = Go(false, File)) == true);
411 if (Last == false)
412 return false;
413 return !_error->PendingError();
414 }
415
416 // Transfer the block
417 In.Limit(Len);
418 while (Go(true, File) == true)
419 if (In.IsLimit() == true)
420 break;
421
422 // Error
423 if (In.IsLimit() == false)
424 return false;
425
426 // The server sends an extra new line before the next block specifier..
427 In.Limit(-1);
428 Last = true;
429 do
430 {
431 if (In.WriteTillEl(Data,true) == true)
432 break;
433 }
434 while ((Last = Go(false, File)) == true);
435 if (Last == false)
436 return false;
437 }
438 }
439 else
440 {
441 /* Closes encoding is used when the server did not specify a size, the
442 loss of the connection means we are done */
443 if (Encoding == Closes)
444 In.Limit(-1);
445 else
446 In.Limit(Size - StartPos);
447
448 // Just transfer the whole block.
449 do
450 {
451 if (In.IsLimit() == false)
452 continue;
453
454 In.Limit(-1);
455 return !_error->PendingError();
456 }
457 while (Go(true, File) == true);
458 }
459
460 return Owner->Flush() && !_error->PendingError();
461 }
462 /*}}}*/
463 bool HttpServerState::ReadHeaderLines(std::string &Data) /*{{{*/
464 {
465 return In.WriteTillEl(Data);
466 }
467 /*}}}*/
468 bool HttpServerState::LoadNextResponse(bool const ToFile, FileFd * const File)/*{{{*/
469 {
470 return Go(ToFile, File);
471 }
472 /*}}}*/
473 bool HttpServerState::WriteResponse(const std::string &Data) /*{{{*/
474 {
475 return Out.Read(Data);
476 }
477 /*}}}*/
478 APT_PURE bool HttpServerState::IsOpen() /*{{{*/
479 {
480 return (ServerFd != -1);
481 }
482 /*}}}*/
483 bool HttpServerState::InitHashes(FileFd &File) /*{{{*/
484 {
485 delete In.Hash;
486 In.Hash = new Hashes;
487
488 // Set the expected size and read file for the hashes
489 File.Truncate(StartPos);
490 return In.Hash->AddFD(File, StartPos);
491 }
492 /*}}}*/
493 APT_PURE Hashes * HttpServerState::GetHashes() /*{{{*/
494 {
495 return In.Hash;
496 }
497 /*}}}*/
498 // HttpServerState::Die - The server has closed the connection. /*{{{*/
499 bool HttpServerState::Die(FileFd &File)
500 {
501 unsigned int LErrno = errno;
502
503 // Dump the buffer to the file
504 if (State == ServerState::Data)
505 {
506 // on GNU/kFreeBSD, apt dies on /dev/null because non-blocking
507 // can't be set
508 if (File.Name() != "/dev/null")
509 SetNonBlock(File.Fd(),false);
510 while (In.WriteSpace() == true)
511 {
512 if (In.Write(File.Fd()) == false)
513 return _error->Errno("write",_("Error writing to the file"));
514
515 // Done
516 if (In.IsLimit() == true)
517 return true;
518 }
519 }
520
521 // See if this is because the server finished the data stream
522 if (In.IsLimit() == false && State != HttpServerState::Header &&
523 Encoding != HttpServerState::Closes)
524 {
525 Close();
526 if (LErrno == 0)
527 return _error->Error(_("Error reading from server. Remote end closed connection"));
528 errno = LErrno;
529 return _error->Errno("read",_("Error reading from server"));
530 }
531 else
532 {
533 In.Limit(-1);
534
535 // Nothing left in the buffer
536 if (In.WriteSpace() == false)
537 return false;
538
539 // We may have got multiple responses back in one packet..
540 Close();
541 return true;
542 }
543
544 return false;
545 }
546 /*}}}*/
547 // HttpServerState::Flush - Dump the buffer into the file /*{{{*/
548 // ---------------------------------------------------------------------
549 /* This takes the current input buffer from the Server FD and writes it
550 into the file */
551 bool HttpServerState::Flush(FileFd * const File)
552 {
553 if (File != NULL)
554 {
555 // on GNU/kFreeBSD, apt dies on /dev/null because non-blocking
556 // can't be set
557 if (File->Name() != "/dev/null")
558 SetNonBlock(File->Fd(),false);
559 if (In.WriteSpace() == false)
560 return true;
561
562 while (In.WriteSpace() == true)
563 {
564 if (In.Write(File->Fd()) == false)
565 return _error->Errno("write",_("Error writing to file"));
566 if (In.IsLimit() == true)
567 return true;
568 }
569
570 if (In.IsLimit() == true || Encoding == ServerState::Closes)
571 return true;
572 }
573 return false;
574 }
575 /*}}}*/
576 // HttpServerState::Go - Run a single loop /*{{{*/
577 // ---------------------------------------------------------------------
578 /* This runs the select loop over the server FDs, Output file FDs and
579 stdin. */
580 bool HttpServerState::Go(bool ToFile, FileFd * const File)
581 {
582 // Server has closed the connection
583 if (ServerFd == -1 && (In.WriteSpace() == false ||
584 ToFile == false))
585 return false;
586
587 fd_set rfds,wfds;
588 FD_ZERO(&rfds);
589 FD_ZERO(&wfds);
590
591 /* Add the server. We only send more requests if the connection will
592 be persisting */
593 if (Out.WriteSpace() == true && ServerFd != -1
594 && Persistent == true)
595 FD_SET(ServerFd,&wfds);
596 if (In.ReadSpace() == true && ServerFd != -1)
597 FD_SET(ServerFd,&rfds);
598
599 // Add the file
600 int FileFD = -1;
601 if (File != NULL)
602 FileFD = File->Fd();
603
604 if (In.WriteSpace() == true && ToFile == true && FileFD != -1)
605 FD_SET(FileFD,&wfds);
606
607 // Add stdin
608 if (_config->FindB("Acquire::http::DependOnSTDIN", true) == true)
609 FD_SET(STDIN_FILENO,&rfds);
610
611 // Figure out the max fd
612 int MaxFd = FileFD;
613 if (MaxFd < ServerFd)
614 MaxFd = ServerFd;
615
616 // Select
617 struct timeval tv;
618 tv.tv_sec = TimeOut;
619 tv.tv_usec = 0;
620 int Res = 0;
621 if ((Res = select(MaxFd+1,&rfds,&wfds,0,&tv)) < 0)
622 {
623 if (errno == EINTR)
624 return true;
625 return _error->Errno("select",_("Select failed"));
626 }
627
628 if (Res == 0)
629 {
630 _error->Error(_("Connection timed out"));
631 return Die(*File);
632 }
633
634 // Handle server IO
635 if (ServerFd != -1 && FD_ISSET(ServerFd,&rfds))
636 {
637 errno = 0;
638 if (In.Read(ServerFd) == false)
639 return Die(*File);
640 }
641
642 if (ServerFd != -1 && FD_ISSET(ServerFd,&wfds))
643 {
644 errno = 0;
645 if (Out.Write(ServerFd) == false)
646 return Die(*File);
647 }
648
649 // Send data to the file
650 if (FileFD != -1 && FD_ISSET(FileFD,&wfds))
651 {
652 if (In.Write(FileFD) == false)
653 return _error->Errno("write",_("Error writing to output file"));
654 }
655
656 if (ExpectedSize > 0 && In.TotalWriten > ExpectedSize)
657 return _error->Error("Writing more data than expected (%llu > %llu)",
658 In.TotalWriten, ExpectedSize);
659
660 // Handle commands from APT
661 if (FD_ISSET(STDIN_FILENO,&rfds))
662 {
663 if (Owner->Run(true) != -1)
664 exit(100);
665 }
666
667 return true;
668 }
669 /*}}}*/
670
671 // HttpMethod::SendReq - Send the HTTP request /*{{{*/
672 // ---------------------------------------------------------------------
673 /* This places the http request in the outbound buffer */
674 void HttpMethod::SendReq(FetchItem *Itm)
675 {
676 URI Uri = Itm->Uri;
677
678 // The HTTP server expects a hostname with a trailing :port
679 std::stringstream Req;
680 string ProperHost;
681
682 if (Uri.Host.find(':') != string::npos)
683 ProperHost = '[' + Uri.Host + ']';
684 else
685 ProperHost = Uri.Host;
686
687 /* RFC 2616 ยง5.1.2 requires absolute URIs for requests to proxies,
688 but while its a must for all servers to accept absolute URIs,
689 it is assumed clients will sent an absolute path for non-proxies */
690 std::string requesturi;
691 if (Server->Proxy.empty() == true || Server->Proxy.Host.empty())
692 requesturi = Uri.Path;
693 else
694 requesturi = Itm->Uri;
695
696 // The "+" is encoded as a workaround for a amazon S3 bug
697 // see LP bugs #1003633 and #1086997.
698 requesturi = QuoteString(requesturi, "+~ ");
699
700 /* Build the request. No keep-alive is included as it is the default
701 in 1.1, can cause problems with proxies, and we are an HTTP/1.1
702 client anyway.
703 C.f. https://tools.ietf.org/wg/httpbis/trac/ticket/158 */
704 Req << "GET " << requesturi << " HTTP/1.1\r\n";
705 if (Uri.Port != 0)
706 Req << "Host: " << ProperHost << ":" << Uri.Port << "\r\n";
707 else
708 Req << "Host: " << ProperHost << "\r\n";
709
710 // generate a cache control header (if needed)
711 if (_config->FindB("Acquire::http::No-Cache",false) == true)
712 Req << "Cache-Control: no-cache\r\n"
713 << "Pragma: no-cache\r\n";
714 else if (Itm->IndexFile == true)
715 Req << "Cache-Control: max-age=" << _config->FindI("Acquire::http::Max-Age",0) << "\r\n";
716 else if (_config->FindB("Acquire::http::No-Store",false) == true)
717 Req << "Cache-Control: no-store\r\n";
718
719 // If we ask for uncompressed files servers might respond with content-
720 // negotiation which lets us end up with compressed files we do not support,
721 // see 657029, 657560 and co, so if we have no extension on the request
722 // ask for text only. As a sidenote: If there is nothing to negotate servers
723 // seem to be nice and ignore it.
724 if (_config->FindB("Acquire::http::SendAccept", true) == true)
725 {
726 size_t const filepos = Itm->Uri.find_last_of('/');
727 string const file = Itm->Uri.substr(filepos + 1);
728 if (flExtension(file) == file)
729 Req << "Accept: text/*\r\n";
730 }
731
732 // Check for a partial file and send if-queries accordingly
733 struct stat SBuf;
734 if (stat(Itm->DestFile.c_str(),&SBuf) >= 0 && SBuf.st_size > 0)
735 Req << "Range: bytes=" << SBuf.st_size << "-\r\n"
736 << "If-Range: " << TimeRFC1123(SBuf.st_mtime) << "\r\n";
737 else if (Itm->LastModified != 0)
738 Req << "If-Modified-Since: " << TimeRFC1123(Itm->LastModified).c_str() << "\r\n";
739
740 if (Server->Proxy.User.empty() == false || Server->Proxy.Password.empty() == false)
741 Req << "Proxy-Authorization: Basic "
742 << Base64Encode(Server->Proxy.User + ":" + Server->Proxy.Password) << "\r\n";
743
744 maybe_add_auth (Uri, _config->FindFile("Dir::Etc::netrc"));
745 if (Uri.User.empty() == false || Uri.Password.empty() == false)
746 Req << "Authorization: Basic "
747 << Base64Encode(Uri.User + ":" + Uri.Password) << "\r\n";
748
749 Req << "User-Agent: " << _config->Find("Acquire::http::User-Agent",
750 "Debian APT-HTTP/1.3 (" PACKAGE_VERSION ")") << "\r\n";
751
752 Req << "\r\n";
753
754 if (Debug == true)
755 cerr << Req << endl;
756
757 Server->WriteResponse(Req.str());
758 }
759 /*}}}*/
760 // HttpMethod::Configuration - Handle a configuration message /*{{{*/
761 // ---------------------------------------------------------------------
762 /* We stash the desired pipeline depth */
763 bool HttpMethod::Configuration(string Message)
764 {
765 if (ServerMethod::Configuration(Message) == false)
766 return false;
767
768 AllowRedirect = _config->FindB("Acquire::http::AllowRedirect",true);
769 PipelineDepth = _config->FindI("Acquire::http::Pipeline-Depth",
770 PipelineDepth);
771 Debug = _config->FindB("Debug::Acquire::http",false);
772
773 // Get the proxy to use
774 AutoDetectProxy();
775
776 return true;
777 }
778 /*}}}*/
779 // HttpMethod::AutoDetectProxy - auto detect proxy /*{{{*/
780 // ---------------------------------------------------------------------
781 /* */
782 bool HttpMethod::AutoDetectProxy()
783 {
784 // option is "Acquire::http::Proxy-Auto-Detect" but we allow the old
785 // name without the dash ("-")
786 AutoDetectProxyCmd = _config->Find("Acquire::http::Proxy-Auto-Detect",
787 _config->Find("Acquire::http::ProxyAutoDetect"));
788
789 if (AutoDetectProxyCmd.empty())
790 return true;
791
792 if (Debug)
793 clog << "Using auto proxy detect command: " << AutoDetectProxyCmd << endl;
794
795 int Pipes[2] = {-1,-1};
796 if (pipe(Pipes) != 0)
797 return _error->Errno("pipe", "Failed to create Pipe");
798
799 pid_t Process = ExecFork();
800 if (Process == 0)
801 {
802 close(Pipes[0]);
803 dup2(Pipes[1],STDOUT_FILENO);
804 SetCloseExec(STDOUT_FILENO,false);
805
806 const char *Args[2];
807 Args[0] = AutoDetectProxyCmd.c_str();
808 Args[1] = 0;
809 execv(Args[0],(char **)Args);
810 cerr << "Failed to exec method " << Args[0] << endl;
811 _exit(100);
812 }
813 char buf[512];
814 int InFd = Pipes[0];
815 close(Pipes[1]);
816 int res = read(InFd, buf, sizeof(buf)-1);
817 ExecWait(Process, "ProxyAutoDetect", true);
818
819 if (res < 0)
820 return _error->Errno("read", "Failed to read");
821 if (res == 0)
822 return _error->Warning("ProxyAutoDetect returned no data");
823
824 // add trailing \0
825 buf[res] = 0;
826
827 if (Debug)
828 clog << "auto detect command returned: '" << buf << "'" << endl;
829
830 if (strstr(buf, "http://") == buf)
831 _config->Set("Acquire::http::proxy", _strstrip(buf));
832
833 return true;
834 }
835 /*}}}*/
836 ServerState * HttpMethod::CreateServerState(URI uri) /*{{{*/
837 {
838 return new HttpServerState(uri, this);
839 }
840 /*}}}*/
841 void HttpMethod::RotateDNS() /*{{{*/
842 {
843 ::RotateDNS();
844 }
845 /*}}}*/