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