]> sjero.net Git - wget/blob - src/retr.c
[svn] Merging fix #20499: MAX_REDIRECTIONS should be configurable.
[wget] / src / retr.c
1 /* File retrieval.
2    Copyright (C) 1996-2006 Free Software Foundation, Inc.
3
4 This file is part of GNU Wget.
5
6 GNU Wget is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or (at
9 your option) any later version.
10
11 GNU Wget is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 GNU General Public License for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with Wget.  If not, see <http://www.gnu.org/licenses/>.
18
19 In addition, as a special exception, the Free Software Foundation
20 gives permission to link the code of its release of Wget with the
21 OpenSSL project's "OpenSSL" library (or with modified versions of it
22 that use the same license as the "OpenSSL" library), and distribute
23 the linked executables.  You must obey the GNU General Public License
24 in all respects for all of the code used other than "OpenSSL".  If you
25 modify this file, you may extend this exception to your version of the
26 file, but you are not obligated to do so.  If you do not wish to do
27 so, delete this exception statement from your version.  */
28
29 #include <config.h>
30
31 #include <stdio.h>
32 #include <stdlib.h>
33 #ifdef HAVE_UNISTD_H
34 # include <unistd.h>
35 #endif /* HAVE_UNISTD_H */
36 #include <errno.h>
37 #include <string.h>
38 #include <assert.h>
39
40 #include "wget.h"
41 #include "utils.h"
42 #include "retr.h"
43 #include "progress.h"
44 #include "url.h"
45 #include "recur.h"
46 #include "ftp.h"
47 #include "http.h"
48 #include "host.h"
49 #include "connect.h"
50 #include "hash.h"
51 #include "convert.h"
52 #include "ptimer.h"
53
54 /* Total size of downloaded files.  Used to enforce quota.  */
55 SUM_SIZE_INT total_downloaded_bytes;
56
57 /* Total download time in seconds. */
58 double total_download_time;
59
60 /* If non-NULL, the stream to which output should be written.  This
61    stream is initialized when `-O' is used.  */
62 FILE *output_stream;
63
64 /* Whether output_document is a regular file we can manipulate,
65    i.e. not `-' or a device file. */
66 bool output_stream_regular;
67 \f
68 static struct {
69   wgint chunk_bytes;
70   double chunk_start;
71   double sleep_adjust;
72 } limit_data;
73
74 static void
75 limit_bandwidth_reset (void)
76 {
77   xzero (limit_data);
78 }
79
80 /* Limit the bandwidth by pausing the download for an amount of time.
81    BYTES is the number of bytes received from the network, and TIMER
82    is the timer that started at the beginning of download.  */
83
84 static void
85 limit_bandwidth (wgint bytes, struct ptimer *timer)
86 {
87   double delta_t = ptimer_read (timer) - limit_data.chunk_start;
88   double expected;
89
90   limit_data.chunk_bytes += bytes;
91
92   /* Calculate the amount of time we expect downloading the chunk
93      should take.  If in reality it took less time, sleep to
94      compensate for the difference.  */
95   expected = (double) limit_data.chunk_bytes / opt.limit_rate;
96
97   if (expected > delta_t)
98     {
99       double slp = expected - delta_t + limit_data.sleep_adjust;
100       double t0, t1;
101       if (slp < 0.2)
102         {
103           DEBUGP (("deferring a %.2f ms sleep (%s/%.2f).\n",
104                    slp * 1000, number_to_static_string (limit_data.chunk_bytes),
105                    delta_t));
106           return;
107         }
108       DEBUGP (("\nsleeping %.2f ms for %s bytes, adjust %.2f ms\n",
109                slp * 1000, number_to_static_string (limit_data.chunk_bytes),
110                limit_data.sleep_adjust));
111
112       t0 = ptimer_read (timer);
113       xsleep (slp);
114       t1 = ptimer_measure (timer);
115
116       /* Due to scheduling, we probably slept slightly longer (or
117          shorter) than desired.  Calculate the difference between the
118          desired and the actual sleep, and adjust the next sleep by
119          that amount.  */
120       limit_data.sleep_adjust = slp - (t1 - t0);
121       /* If sleep_adjust is very large, it's likely due to suspension
122          and not clock inaccuracy.  Don't enforce those.  */
123       if (limit_data.sleep_adjust > 0.5)
124         limit_data.sleep_adjust = 0.5;
125       else if (limit_data.sleep_adjust < -0.5)
126         limit_data.sleep_adjust = -0.5;
127     }
128
129   limit_data.chunk_bytes = 0;
130   limit_data.chunk_start = ptimer_read (timer);
131 }
132
133 #ifndef MIN
134 # define MIN(i, j) ((i) <= (j) ? (i) : (j))
135 #endif
136
137 /* Write data in BUF to OUT.  However, if *SKIP is non-zero, skip that
138    amount of data and decrease SKIP.  Increment *TOTAL by the amount
139    of data written.  */
140
141 static int
142 write_data (FILE *out, const char *buf, int bufsize, wgint *skip,
143             wgint *written)
144 {
145   if (!out)
146     return 1;
147   if (*skip > bufsize)
148     {
149       *skip -= bufsize;
150       return 1;
151     }
152   if (*skip)
153     {
154       buf += *skip;
155       bufsize -= *skip;
156       *skip = 0;
157       if (bufsize == 0)
158         return 1;
159     }
160
161   fwrite (buf, 1, bufsize, out);
162   *written += bufsize;
163
164   /* Immediately flush the downloaded data.  This should not hinder
165      performance: fast downloads will arrive in large 16K chunks
166      (which stdio would write out immediately anyway), and slow
167      downloads wouldn't be limited by disk speed.  */
168   fflush (out);
169   return !ferror (out);
170 }
171
172 /* Read the contents of file descriptor FD until it the connection
173    terminates or a read error occurs.  The data is read in portions of
174    up to 16K and written to OUT as it arrives.  If opt.verbose is set,
175    the progress is shown.
176
177    TOREAD is the amount of data expected to arrive, normally only used
178    by the progress gauge.
179
180    STARTPOS is the position from which the download starts, used by
181    the progress gauge.  If QTYREAD is non-NULL, the value it points to
182    is incremented by the amount of data read from the network.  If
183    QTYWRITTEN is non-NULL, the value it points to is incremented by
184    the amount of data written to disk.  The time it took to download
185    the data is stored to ELAPSED.
186
187    The function exits and returns the amount of data read.  In case of
188    error while reading data, -1 is returned.  In case of error while
189    writing data, -2 is returned.  */
190
191 int
192 fd_read_body (int fd, FILE *out, wgint toread, wgint startpos,
193               wgint *qtyread, wgint *qtywritten, double *elapsed, int flags)
194 {
195   int ret = 0;
196
197   static char dlbuf[16384];
198   int dlbufsize = sizeof (dlbuf);
199
200   struct ptimer *timer = NULL;
201   double last_successful_read_tm = 0;
202
203   /* The progress gauge, set according to the user preferences. */
204   void *progress = NULL;
205
206   /* Non-zero if the progress gauge is interactive, i.e. if it can
207      continually update the display.  When true, smaller timeout
208      values are used so that the gauge can update the display when
209      data arrives slowly. */
210   bool progress_interactive = false;
211
212   bool exact = !!(flags & rb_read_exactly);
213   wgint skip = 0;
214
215   /* How much data we've read/written.  */
216   wgint sum_read = 0;
217   wgint sum_written = 0;
218
219   if (flags & rb_skip_startpos)
220     skip = startpos;
221
222   if (opt.verbose)
223     {
224       /* If we're skipping STARTPOS bytes, pass 0 as the INITIAL
225          argument to progress_create because the indicator doesn't
226          (yet) know about "skipping" data.  */
227       progress = progress_create (skip ? 0 : startpos, startpos + toread);
228       progress_interactive = progress_interactive_p (progress);
229     }
230
231   if (opt.limit_rate)
232     limit_bandwidth_reset ();
233
234   /* A timer is needed for tracking progress, for throttling, and for
235      tracking elapsed time.  If either of these are requested, start
236      the timer.  */
237   if (progress || opt.limit_rate || elapsed)
238     {
239       timer = ptimer_new ();
240       last_successful_read_tm = 0;
241     }
242
243   /* Use a smaller buffer for low requested bandwidths.  For example,
244      with --limit-rate=2k, it doesn't make sense to slurp in 16K of
245      data and then sleep for 8s.  With buffer size equal to the limit,
246      we never have to sleep for more than one second.  */
247   if (opt.limit_rate && opt.limit_rate < dlbufsize)
248     dlbufsize = opt.limit_rate;
249
250   /* Read from FD while there is data to read.  Normally toread==0
251      means that it is unknown how much data is to arrive.  However, if
252      EXACT is set, then toread==0 means what it says: that no data
253      should be read.  */
254   while (!exact || (sum_read < toread))
255     {
256       int rdsize = exact ? MIN (toread - sum_read, dlbufsize) : dlbufsize;
257       double tmout = opt.read_timeout;
258       if (progress_interactive)
259         {
260           /* For interactive progress gauges, always specify a ~1s
261              timeout, so that the gauge can be updated regularly even
262              when the data arrives very slowly or stalls.  */
263           tmout = 0.95;
264           if (opt.read_timeout)
265             {
266               double waittm;
267               waittm = ptimer_read (timer) - last_successful_read_tm;
268               if (waittm + tmout > opt.read_timeout)
269                 {
270                   /* Don't let total idle time exceed read timeout. */
271                   tmout = opt.read_timeout - waittm;
272                   if (tmout < 0)
273                     {
274                       /* We've already exceeded the timeout. */
275                       ret = -1, errno = ETIMEDOUT;
276                       break;
277                     }
278                 }
279             }
280         }
281       ret = fd_read (fd, dlbuf, rdsize, tmout);
282
283       if (progress_interactive && ret < 0 && errno == ETIMEDOUT)
284         ret = 0;                /* interactive timeout, handled above */
285       else if (ret <= 0)
286         break;                  /* EOF or read error */
287
288       if (progress || opt.limit_rate)
289         {
290           ptimer_measure (timer);
291           if (ret > 0)
292             last_successful_read_tm = ptimer_read (timer);
293         }
294
295       if (ret > 0)
296         {
297           sum_read += ret;
298           if (!write_data (out, dlbuf, ret, &skip, &sum_written))
299             {
300               ret = -2;
301               goto out;
302             }
303         }
304
305       if (opt.limit_rate)
306         limit_bandwidth (ret, timer);
307
308       if (progress)
309         progress_update (progress, ret, ptimer_read (timer));
310 #ifdef WINDOWS
311       if (toread > 0 && !opt.quiet)
312         ws_percenttitle (100.0 *
313                          (startpos + sum_read) / (startpos + toread));
314 #endif
315     }
316   if (ret < -1)
317     ret = -1;
318
319  out:
320   if (progress)
321     progress_finish (progress, ptimer_read (timer));
322
323   if (elapsed)
324     *elapsed = ptimer_read (timer);
325   if (timer)
326     ptimer_destroy (timer);
327
328   if (qtyread)
329     *qtyread += sum_read;
330   if (qtywritten)
331     *qtywritten += sum_written;
332
333   return ret;
334 }
335 \f
336 /* Read a hunk of data from FD, up until a terminator.  The hunk is
337    limited by whatever the TERMINATOR callback chooses as its
338    terminator.  For example, if terminator stops at newline, the hunk
339    will consist of a line of data; if terminator stops at two
340    newlines, it can be used to read the head of an HTTP response.
341    Upon determining the boundary, the function returns the data (up to
342    the terminator) in malloc-allocated storage.
343
344    In case of read error, NULL is returned.  In case of EOF and no
345    data read, NULL is returned and errno set to 0.  In case of having
346    read some data, but encountering EOF before seeing the terminator,
347    the data that has been read is returned, but it will (obviously)
348    not contain the terminator.
349
350    The TERMINATOR function is called with three arguments: the
351    beginning of the data read so far, the beginning of the current
352    block of peeked-at data, and the length of the current block.
353    Depending on its needs, the function is free to choose whether to
354    analyze all data or just the newly arrived data.  If TERMINATOR
355    returns NULL, it means that the terminator has not been seen.
356    Otherwise it should return a pointer to the charactre immediately
357    following the terminator.
358
359    The idea is to be able to read a line of input, or otherwise a hunk
360    of text, such as the head of an HTTP request, without crossing the
361    boundary, so that the next call to fd_read etc. reads the data
362    after the hunk.  To achieve that, this function does the following:
363
364    1. Peek at incoming data.
365
366    2. Determine whether the peeked data, along with the previously
367       read data, includes the terminator.
368
369       2a. If yes, read the data until the end of the terminator, and
370           exit.
371
372       2b. If no, read the peeked data and goto 1.
373
374    The function is careful to assume as little as possible about the
375    implementation of peeking.  For example, every peek is followed by
376    a read.  If the read returns a different amount of data, the
377    process is retried until all data arrives safely.
378
379    SIZEHINT is the buffer size sufficient to hold all the data in the
380    typical case (it is used as the initial buffer size).  MAXSIZE is
381    the maximum amount of memory this function is allowed to allocate,
382    or 0 if no upper limit is to be enforced.
383
384    This function should be used as a building block for other
385    functions -- see fd_read_line as a simple example.  */
386
387 char *
388 fd_read_hunk (int fd, hunk_terminator_t terminator, long sizehint, long maxsize)
389 {
390   long bufsize = sizehint;
391   char *hunk = xmalloc (bufsize);
392   int tail = 0;                 /* tail position in HUNK */
393
394   assert (maxsize >= bufsize);
395
396   while (1)
397     {
398       const char *end;
399       int pklen, rdlen, remain;
400
401       /* First, peek at the available data. */
402
403       pklen = fd_peek (fd, hunk + tail, bufsize - 1 - tail, -1);
404       if (pklen < 0)
405         {
406           xfree (hunk);
407           return NULL;
408         }
409       end = terminator (hunk, hunk + tail, pklen);
410       if (end)
411         {
412           /* The data contains the terminator: we'll drain the data up
413              to the end of the terminator.  */
414           remain = end - (hunk + tail);
415           assert (remain >= 0);
416           if (remain == 0)
417             {
418               /* No more data needs to be read. */
419               hunk[tail] = '\0';
420               return hunk;
421             }
422           if (bufsize - 1 < tail + remain)
423             {
424               bufsize = tail + remain + 1;
425               hunk = xrealloc (hunk, bufsize);
426             }
427         }
428       else
429         /* No terminator: simply read the data we know is (or should
430            be) available.  */
431         remain = pklen;
432
433       /* Now, read the data.  Note that we make no assumptions about
434          how much data we'll get.  (Some TCP stacks are notorious for
435          read returning less data than the previous MSG_PEEK.)  */
436
437       rdlen = fd_read (fd, hunk + tail, remain, 0);
438       if (rdlen < 0)
439         {
440           xfree_null (hunk);
441           return NULL;
442         }
443       tail += rdlen;
444       hunk[tail] = '\0';
445
446       if (rdlen == 0)
447         {
448           if (tail == 0)
449             {
450               /* EOF without anything having been read */
451               xfree (hunk);
452               errno = 0;
453               return NULL;
454             }
455           else
456             /* EOF seen: return the data we've read. */
457             return hunk;
458         }
459       if (end && rdlen == remain)
460         /* The terminator was seen and the remaining data drained --
461            we got what we came for.  */
462         return hunk;
463
464       /* Keep looping until all the data arrives. */
465
466       if (tail == bufsize - 1)
467         {
468           /* Double the buffer size, but refuse to allocate more than
469              MAXSIZE bytes.  */
470           if (maxsize && bufsize >= maxsize)
471             {
472               xfree (hunk);
473               errno = ENOMEM;
474               return NULL;
475             }
476           bufsize <<= 1;
477           if (maxsize && bufsize > maxsize)
478             bufsize = maxsize;
479           hunk = xrealloc (hunk, bufsize);
480         }
481     }
482 }
483
484 static const char *
485 line_terminator (const char *start, const char *peeked, int peeklen)
486 {
487   const char *p = memchr (peeked, '\n', peeklen);
488   if (p)
489     /* p+1 because the line must include '\n' */
490     return p + 1;
491   return NULL;
492 }
493
494 /* The maximum size of the single line we agree to accept.  This is
495    not meant to impose an arbitrary limit, but to protect the user
496    from Wget slurping up available memory upon encountering malicious
497    or buggy server output.  Define it to 0 to remove the limit.  */
498 #define FD_READ_LINE_MAX 4096
499
500 /* Read one line from FD and return it.  The line is allocated using
501    malloc, but is never larger than FD_READ_LINE_MAX.
502
503    If an error occurs, or if no data can be read, NULL is returned.
504    In the former case errno indicates the error condition, and in the
505    latter case, errno is NULL.  */
506
507 char *
508 fd_read_line (int fd)
509 {
510   return fd_read_hunk (fd, line_terminator, 128, FD_READ_LINE_MAX);
511 }
512 \f
513 /* Return a printed representation of the download rate, along with
514    the units appropriate for the download speed.  */
515
516 const char *
517 retr_rate (wgint bytes, double secs)
518 {
519   static char res[20];
520   static const char *rate_names[] = {"B/s", "KB/s", "MB/s", "GB/s" };
521   int units;
522
523   double dlrate = calc_rate (bytes, secs, &units);
524   /* Use more digits for smaller numbers (regardless of unit used),
525      e.g. "1022", "247", "12.5", "2.38".  */
526   sprintf (res, "%.*f %s",
527            dlrate >= 99.95 ? 0 : dlrate >= 9.995 ? 1 : 2,
528            dlrate, rate_names[units]);
529
530   return res;
531 }
532
533 /* Calculate the download rate and trim it as appropriate for the
534    speed.  Appropriate means that if rate is greater than 1K/s,
535    kilobytes are used, and if rate is greater than 1MB/s, megabytes
536    are used.
537
538    UNITS is zero for B/s, one for KB/s, two for MB/s, and three for
539    GB/s.  */
540
541 double
542 calc_rate (wgint bytes, double secs, int *units)
543 {
544   double dlrate;
545
546   assert (secs >= 0);
547   assert (bytes >= 0);
548
549   if (secs == 0)
550     /* If elapsed time is exactly zero, it means we're under the
551        resolution of the timer.  This can easily happen on systems
552        that use time() for the timer.  Since the interval lies between
553        0 and the timer's resolution, assume half the resolution.  */
554     secs = ptimer_resolution () / 2.0;
555
556   dlrate = bytes / secs;
557   if (dlrate < 1024.0)
558     *units = 0;
559   else if (dlrate < 1024.0 * 1024.0)
560     *units = 1, dlrate /= 1024.0;
561   else if (dlrate < 1024.0 * 1024.0 * 1024.0)
562     *units = 2, dlrate /= (1024.0 * 1024.0);
563   else
564     /* Maybe someone will need this, one day. */
565     *units = 3, dlrate /= (1024.0 * 1024.0 * 1024.0);
566
567   return dlrate;
568 }
569 \f
570
571 #define SUSPEND_POST_DATA do {                  \
572   post_data_suspended = true;                   \
573   saved_post_data = opt.post_data;              \
574   saved_post_file_name = opt.post_file_name;    \
575   opt.post_data = NULL;                         \
576   opt.post_file_name = NULL;                    \
577 } while (0)
578
579 #define RESTORE_POST_DATA do {                          \
580   if (post_data_suspended)                              \
581     {                                                   \
582       opt.post_data = saved_post_data;                  \
583       opt.post_file_name = saved_post_file_name;        \
584       post_data_suspended = false;                      \
585     }                                                   \
586 } while (0)
587
588 static char *getproxy (struct url *);
589
590 /* Retrieve the given URL.  Decides which loop to call -- HTTP, FTP,
591    FTP, proxy, etc.  */
592
593 /* #### This function should be rewritten so it doesn't return from
594    multiple points. */
595
596 uerr_t
597 retrieve_url (const char *origurl, char **file, char **newloc,
598               const char *refurl, int *dt, bool recursive)
599 {
600   uerr_t result;
601   char *url;
602   bool location_changed;
603   int dummy;
604   char *mynewloc, *proxy;
605   struct url *u, *proxy_url;
606   int up_error_code;            /* url parse error code */
607   char *local_file;
608   int redirection_count = 0;
609
610   bool post_data_suspended = false;
611   char *saved_post_data = NULL;
612   char *saved_post_file_name = NULL;
613
614   /* If dt is NULL, use local storage.  */
615   if (!dt)
616     {
617       dt = &dummy;
618       dummy = 0;
619     }
620   url = xstrdup (origurl);
621   if (newloc)
622     *newloc = NULL;
623   if (file)
624     *file = NULL;
625
626   u = url_parse (url, &up_error_code);
627   if (!u)
628     {
629       logprintf (LOG_NOTQUIET, "%s: %s.\n", url, url_error (up_error_code));
630       xfree (url);
631       return URLERROR;
632     }
633
634   if (!refurl)
635     refurl = opt.referer;
636
637  redirected:
638
639   result = NOCONERROR;
640   mynewloc = NULL;
641   local_file = NULL;
642   proxy_url = NULL;
643
644   proxy = getproxy (u);
645   if (proxy)
646     {
647       /* Parse the proxy URL.  */
648       proxy_url = url_parse (proxy, &up_error_code);
649       if (!proxy_url)
650         {
651           logprintf (LOG_NOTQUIET, _("Error parsing proxy URL %s: %s.\n"),
652                      proxy, url_error (up_error_code));
653           xfree (url);
654           RESTORE_POST_DATA;
655           return PROXERR;
656         }
657       if (proxy_url->scheme != SCHEME_HTTP && proxy_url->scheme != u->scheme)
658         {
659           logprintf (LOG_NOTQUIET, _("Error in proxy URL %s: Must be HTTP.\n"), proxy);
660           url_free (proxy_url);
661           xfree (url);
662           RESTORE_POST_DATA;
663           return PROXERR;
664         }
665     }
666
667   if (u->scheme == SCHEME_HTTP
668 #ifdef HAVE_SSL
669       || u->scheme == SCHEME_HTTPS
670 #endif
671       || (proxy_url && proxy_url->scheme == SCHEME_HTTP))
672     {
673       result = http_loop (u, &mynewloc, &local_file, refurl, dt, proxy_url);
674     }
675   else if (u->scheme == SCHEME_FTP)
676     {
677       /* If this is a redirection, temporarily turn off opt.ftp_glob
678          and opt.recursive, both being undesirable when following
679          redirects.  */
680       bool oldrec = recursive, glob = opt.ftp_glob;
681       if (redirection_count)
682         oldrec = glob = false;
683
684       result = ftp_loop (u, dt, proxy_url, recursive, glob);
685       recursive = oldrec;
686
687       /* There is a possibility of having HTTP being redirected to
688          FTP.  In these cases we must decide whether the text is HTML
689          according to the suffix.  The HTML suffixes are `.html',
690          `.htm' and a few others, case-insensitive.  */
691       if (redirection_count && local_file && u->scheme == SCHEME_FTP)
692         {
693           if (has_html_suffix_p (local_file))
694             *dt |= TEXTHTML;
695         }
696     }
697
698   if (proxy_url)
699     {
700       url_free (proxy_url);
701       proxy_url = NULL;
702     }
703
704   location_changed = (result == NEWLOCATION);
705   if (location_changed)
706     {
707       char *construced_newloc;
708       struct url *newloc_parsed;
709
710       assert (mynewloc != NULL);
711
712       if (local_file)
713         xfree (local_file);
714
715       /* The HTTP specs only allow absolute URLs to appear in
716          redirects, but a ton of boneheaded webservers and CGIs out
717          there break the rules and use relative URLs, and popular
718          browsers are lenient about this, so wget should be too. */
719       construced_newloc = uri_merge (url, mynewloc);
720       xfree (mynewloc);
721       mynewloc = construced_newloc;
722
723       /* Now, see if this new location makes sense. */
724       newloc_parsed = url_parse (mynewloc, &up_error_code);
725       if (!newloc_parsed)
726         {
727           logprintf (LOG_NOTQUIET, "%s: %s.\n", escnonprint_uri (mynewloc),
728                      url_error (up_error_code));
729           url_free (u);
730           xfree (url);
731           xfree (mynewloc);
732           RESTORE_POST_DATA;
733           return result;
734         }
735
736       /* Now mynewloc will become newloc_parsed->url, because if the
737          Location contained relative paths like .././something, we
738          don't want that propagating as url.  */
739       xfree (mynewloc);
740       mynewloc = xstrdup (newloc_parsed->url);
741
742       /* Check for max. number of redirections.  */
743       if (++redirection_count > opt.max_redirect)
744         {
745           logprintf (LOG_NOTQUIET, _("%d redirections exceeded.\n"),
746                      opt.max_redirect);
747           url_free (newloc_parsed);
748           url_free (u);
749           xfree (url);
750           xfree (mynewloc);
751           RESTORE_POST_DATA;
752           return WRONGCODE;
753         }
754
755       xfree (url);
756       url = mynewloc;
757       url_free (u);
758       u = newloc_parsed;
759
760       /* If we're being redirected from POST, we don't want to POST
761          again.  Many requests answer POST with a redirection to an
762          index page; that redirection is clearly a GET.  We "suspend"
763          POST data for the duration of the redirections, and restore
764          it when we're done. */
765       if (!post_data_suspended)
766         SUSPEND_POST_DATA;
767
768       goto redirected;
769     }
770
771   if (local_file)
772     {
773       if (*dt & RETROKF)
774         {
775           register_download (u->url, local_file);
776           if (redirection_count && 0 != strcmp (origurl, u->url))
777             register_redirection (origurl, u->url);
778           if (*dt & TEXTHTML)
779             register_html (u->url, local_file);
780         }
781     }
782
783   if (file)
784     *file = local_file ? local_file : NULL;
785   else
786     xfree_null (local_file);
787
788   url_free (u);
789
790   if (redirection_count)
791     {
792       if (newloc)
793         *newloc = url;
794       else
795         xfree (url);
796     }
797   else
798     {
799       if (newloc)
800         *newloc = NULL;
801       xfree (url);
802     }
803
804   RESTORE_POST_DATA;
805
806   return result;
807 }
808
809 /* Find the URLs in the file and call retrieve_url() for each of them.
810    If HTML is true, treat the file as HTML, and construct the URLs
811    accordingly.
812
813    If opt.recursive is set, call retrieve_tree() for each file.  */
814
815 uerr_t
816 retrieve_from_file (const char *file, bool html, int *count)
817 {
818   uerr_t status;
819   struct urlpos *url_list, *cur_url;
820
821   url_list = (html ? get_urls_html (file, NULL, NULL)
822               : get_urls_file (file));
823   status = RETROK;             /* Suppose everything is OK.  */
824   *count = 0;                  /* Reset the URL count.  */
825
826   for (cur_url = url_list; cur_url; cur_url = cur_url->next, ++*count)
827     {
828       char *filename = NULL, *new_file = NULL;
829       int dt;
830
831       if (cur_url->ignore_when_downloading)
832         continue;
833
834       if (opt.quota && total_downloaded_bytes > opt.quota)
835         {
836           status = QUOTEXC;
837           break;
838         }
839       if ((opt.recursive || opt.page_requisites)
840           && (cur_url->url->scheme != SCHEME_FTP || getproxy (cur_url->url)))
841         {
842           int old_follow_ftp = opt.follow_ftp;
843
844           /* Turn opt.follow_ftp on in case of recursive FTP retrieval */
845           if (cur_url->url->scheme == SCHEME_FTP) 
846             opt.follow_ftp = 1;
847           
848           status = retrieve_tree (cur_url->url->url);
849
850           opt.follow_ftp = old_follow_ftp;
851         }
852       else
853         status = retrieve_url (cur_url->url->url, &filename, &new_file, NULL, &dt, opt.recursive);
854
855       if (filename && opt.delete_after && file_exists_p (filename))
856         {
857           DEBUGP (("\
858 Removing file due to --delete-after in retrieve_from_file():\n"));
859           logprintf (LOG_VERBOSE, _("Removing %s.\n"), filename);
860           if (unlink (filename))
861             logprintf (LOG_NOTQUIET, "unlink: %s\n", strerror (errno));
862           dt &= ~RETROKF;
863         }
864
865       xfree_null (new_file);
866       xfree_null (filename);
867     }
868
869   /* Free the linked list of URL-s.  */
870   free_urlpos (url_list);
871
872   return status;
873 }
874
875 /* Print `giving up', or `retrying', depending on the impending
876    action.  N1 and N2 are the attempt number and the attempt limit.  */
877 void
878 printwhat (int n1, int n2)
879 {
880   logputs (LOG_VERBOSE, (n1 == n2) ? _("Giving up.\n\n") : _("Retrying.\n\n"));
881 }
882
883 /* If opt.wait or opt.waitretry are specified, and if certain
884    conditions are met, sleep the appropriate number of seconds.  See
885    the documentation of --wait and --waitretry for more information.
886
887    COUNT is the count of current retrieval, beginning with 1. */
888
889 void
890 sleep_between_retrievals (int count)
891 {
892   static bool first_retrieval = true;
893
894   if (first_retrieval)
895     {
896       /* Don't sleep before the very first retrieval. */
897       first_retrieval = false;
898       return;
899     }
900
901   if (opt.waitretry && count > 1)
902     {
903       /* If opt.waitretry is specified and this is a retry, wait for
904          COUNT-1 number of seconds, or for opt.waitretry seconds.  */
905       if (count <= opt.waitretry)
906         xsleep (count - 1);
907       else
908         xsleep (opt.waitretry);
909     }
910   else if (opt.wait)
911     {
912       if (!opt.random_wait || count > 1)
913         /* If random-wait is not specified, or if we are sleeping
914            between retries of the same download, sleep the fixed
915            interval.  */
916         xsleep (opt.wait);
917       else
918         {
919           /* Sleep a random amount of time averaging in opt.wait
920              seconds.  The sleeping amount ranges from 0.5*opt.wait to
921              1.5*opt.wait.  */
922           double waitsecs = (0.5 + random_float ()) * opt.wait;
923           DEBUGP (("sleep_between_retrievals: avg=%f,sleep=%f\n",
924                    opt.wait, waitsecs));
925           xsleep (waitsecs);
926         }
927     }
928 }
929
930 /* Free the linked list of urlpos.  */
931 void
932 free_urlpos (struct urlpos *l)
933 {
934   while (l)
935     {
936       struct urlpos *next = l->next;
937       if (l->url)
938         url_free (l->url);
939       xfree_null (l->local_name);
940       xfree (l);
941       l = next;
942     }
943 }
944
945 /* Rotate FNAME opt.backups times */
946 void
947 rotate_backups(const char *fname)
948 {
949   int maxlen = strlen (fname) + 1 + numdigit (opt.backups) + 1;
950   char *from = (char *)alloca (maxlen);
951   char *to = (char *)alloca (maxlen);
952   struct_stat sb;
953   int i;
954
955   if (stat (fname, &sb) == 0)
956     if (S_ISREG (sb.st_mode) == 0)
957       return;
958
959   for (i = opt.backups; i > 1; i--)
960     {
961       sprintf (from, "%s.%d", fname, i - 1);
962       sprintf (to, "%s.%d", fname, i);
963       rename (from, to);
964     }
965
966   sprintf (to, "%s.%d", fname, 1);
967   rename(fname, to);
968 }
969
970 static bool no_proxy_match (const char *, const char **);
971
972 /* Return the URL of the proxy appropriate for url U.  */
973
974 static char *
975 getproxy (struct url *u)
976 {
977   char *proxy = NULL;
978   char *rewritten_url;
979   static char rewritten_storage[1024];
980
981   if (!opt.use_proxy)
982     return NULL;
983   if (no_proxy_match (u->host, (const char **)opt.no_proxy))
984     return NULL;
985
986   switch (u->scheme)
987     {
988     case SCHEME_HTTP:
989       proxy = opt.http_proxy ? opt.http_proxy : getenv ("http_proxy");
990       break;
991 #ifdef HAVE_SSL
992     case SCHEME_HTTPS:
993       proxy = opt.https_proxy ? opt.https_proxy : getenv ("https_proxy");
994       break;
995 #endif
996     case SCHEME_FTP:
997       proxy = opt.ftp_proxy ? opt.ftp_proxy : getenv ("ftp_proxy");
998       break;
999     case SCHEME_INVALID:
1000       break;
1001     }
1002   if (!proxy || !*proxy)
1003     return NULL;
1004
1005   /* Handle shorthands.  `rewritten_storage' is a kludge to allow
1006      getproxy() to return static storage. */
1007   rewritten_url = rewrite_shorthand_url (proxy);
1008   if (rewritten_url)
1009     {
1010       strncpy (rewritten_storage, rewritten_url, sizeof (rewritten_storage));
1011       rewritten_storage[sizeof (rewritten_storage) - 1] = '\0';
1012       proxy = rewritten_storage;
1013     }
1014
1015   return proxy;
1016 }
1017
1018 /* Returns true if URL would be downloaded through a proxy. */
1019
1020 bool
1021 url_uses_proxy (const char *url)
1022 {
1023   bool ret;
1024   struct url *u = url_parse (url, NULL);
1025   if (!u)
1026     return false;
1027   ret = getproxy (u) != NULL;
1028   url_free (u);
1029   return ret;
1030 }
1031
1032 /* Should a host be accessed through proxy, concerning no_proxy?  */
1033 static bool
1034 no_proxy_match (const char *host, const char **no_proxy)
1035 {
1036   if (!no_proxy)
1037     return false;
1038   else
1039     return sufmatch (no_proxy, host);
1040 }