]> sjero.net Git - wget/blob - src/retr.c
-i accepts URLs (stsc). For #21265.
[wget] / src / retr.c
1 /* File retrieval.
2    Copyright (C) 1996, 1997, 1998, 1999, 2000, 2001, 2002, 2003,
3    2004, 2005, 2006, 2007, 2008 Free Software Foundation, Inc.
4
5 This file is part of GNU Wget.
6
7 GNU Wget is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or (at
10 your option) any later version.
11
12 GNU Wget is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with Wget.  If not, see <http://www.gnu.org/licenses/>.
19
20 Additional permission under GNU GPL version 3 section 7
21
22 If you modify this program, or any covered work, by linking or
23 combining it with the OpenSSL project's OpenSSL library (or a
24 modified version of that library), containing parts covered by the
25 terms of the OpenSSL or SSLeay licenses, the Free Software Foundation
26 grants you additional permission to convey the resulting work.
27 Corresponding Source for a non-source form of such a combination
28 shall include the source code for the parts of OpenSSL used as well
29 as that of the covered work.  */
30
31 #include "wget.h"
32
33 #include <stdio.h>
34 #include <stdlib.h>
35 #ifdef HAVE_UNISTD_H
36 # include <unistd.h>
37 #endif /* HAVE_UNISTD_H */
38 #include <errno.h>
39 #include <string.h>
40 #include <assert.h>
41
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
573 #define SUSPEND_POST_DATA do {                  \
574   post_data_suspended = true;                   \
575   saved_post_data = opt.post_data;              \
576   saved_post_file_name = opt.post_file_name;    \
577   opt.post_data = NULL;                         \
578   opt.post_file_name = NULL;                    \
579 } while (0)
580
581 #define RESTORE_POST_DATA do {                          \
582   if (post_data_suspended)                              \
583     {                                                   \
584       opt.post_data = saved_post_data;                  \
585       opt.post_file_name = saved_post_file_name;        \
586       post_data_suspended = false;                      \
587     }                                                   \
588 } while (0)
589
590 static char *getproxy (struct url *);
591
592 /* Retrieve the given URL.  Decides which loop to call -- HTTP, FTP,
593    FTP, proxy, etc.  */
594
595 /* #### This function should be rewritten so it doesn't return from
596    multiple points. */
597
598 uerr_t
599 retrieve_url (const char *origurl, char **file, char **newloc,
600               const char *refurl, int *dt, bool recursive)
601 {
602   uerr_t result;
603   char *url;
604   bool location_changed;
605   int dummy;
606   char *mynewloc, *proxy;
607   struct url *u, *proxy_url;
608   int up_error_code;            /* url parse error code */
609   char *local_file;
610   int redirection_count = 0;
611
612   bool post_data_suspended = false;
613   char *saved_post_data = NULL;
614   char *saved_post_file_name = NULL;
615
616   /* If dt is NULL, use local storage.  */
617   if (!dt)
618     {
619       dt = &dummy;
620       dummy = 0;
621     }
622   url = xstrdup (origurl);
623   if (newloc)
624     *newloc = NULL;
625   if (file)
626     *file = NULL;
627
628   u = url_parse (url, &up_error_code);
629   if (!u)
630     {
631       logprintf (LOG_NOTQUIET, "%s: %s.\n", url, url_error (up_error_code));
632       xfree (url);
633       return URLERROR;
634     }
635
636   if (!refurl)
637     refurl = opt.referer;
638
639  redirected:
640
641   result = NOCONERROR;
642   mynewloc = NULL;
643   local_file = NULL;
644   proxy_url = NULL;
645
646   proxy = getproxy (u);
647   if (proxy)
648     {
649       /* Parse the proxy URL.  */
650       proxy_url = url_parse (proxy, &up_error_code);
651       if (!proxy_url)
652         {
653           logprintf (LOG_NOTQUIET, _("Error parsing proxy URL %s: %s.\n"),
654                      proxy, url_error (up_error_code));
655           xfree (url);
656           RESTORE_POST_DATA;
657           return PROXERR;
658         }
659       if (proxy_url->scheme != SCHEME_HTTP && proxy_url->scheme != u->scheme)
660         {
661           logprintf (LOG_NOTQUIET, _("Error in proxy URL %s: Must be HTTP.\n"), proxy);
662           url_free (proxy_url);
663           xfree (url);
664           RESTORE_POST_DATA;
665           return PROXERR;
666         }
667     }
668
669   if (u->scheme == SCHEME_HTTP
670 #ifdef HAVE_SSL
671       || u->scheme == SCHEME_HTTPS
672 #endif
673       || (proxy_url && proxy_url->scheme == SCHEME_HTTP))
674     {
675       result = http_loop (u, &mynewloc, &local_file, refurl, dt, proxy_url);
676     }
677   else if (u->scheme == SCHEME_FTP)
678     {
679       /* If this is a redirection, temporarily turn off opt.ftp_glob
680          and opt.recursive, both being undesirable when following
681          redirects.  */
682       bool oldrec = recursive, glob = opt.ftp_glob;
683       if (redirection_count)
684         oldrec = glob = false;
685
686       result = ftp_loop (u, dt, proxy_url, recursive, glob);
687       recursive = oldrec;
688
689       /* There is a possibility of having HTTP being redirected to
690          FTP.  In these cases we must decide whether the text is HTML
691          according to the suffix.  The HTML suffixes are `.html',
692          `.htm' and a few others, case-insensitive.  */
693       if (redirection_count && local_file && u->scheme == SCHEME_FTP)
694         {
695           if (has_html_suffix_p (local_file))
696             *dt |= TEXTHTML;
697         }
698     }
699
700   if (proxy_url)
701     {
702       url_free (proxy_url);
703       proxy_url = NULL;
704     }
705
706   location_changed = (result == NEWLOCATION);
707   if (location_changed)
708     {
709       char *construced_newloc;
710       struct url *newloc_parsed;
711
712       assert (mynewloc != NULL);
713
714       if (local_file)
715         xfree (local_file);
716
717       /* The HTTP specs only allow absolute URLs to appear in
718          redirects, but a ton of boneheaded webservers and CGIs out
719          there break the rules and use relative URLs, and popular
720          browsers are lenient about this, so wget should be too. */
721       construced_newloc = uri_merge (url, mynewloc);
722       xfree (mynewloc);
723       mynewloc = construced_newloc;
724
725       /* Now, see if this new location makes sense. */
726       newloc_parsed = url_parse (mynewloc, &up_error_code);
727       if (!newloc_parsed)
728         {
729           logprintf (LOG_NOTQUIET, "%s: %s.\n", escnonprint_uri (mynewloc),
730                      url_error (up_error_code));
731           url_free (u);
732           xfree (url);
733           xfree (mynewloc);
734           RESTORE_POST_DATA;
735           return result;
736         }
737
738       /* Now mynewloc will become newloc_parsed->url, because if the
739          Location contained relative paths like .././something, we
740          don't want that propagating as url.  */
741       xfree (mynewloc);
742       mynewloc = xstrdup (newloc_parsed->url);
743
744       /* Check for max. number of redirections.  */
745       if (++redirection_count > opt.max_redirect)
746         {
747           logprintf (LOG_NOTQUIET, _("%d redirections exceeded.\n"),
748                      opt.max_redirect);
749           url_free (newloc_parsed);
750           url_free (u);
751           xfree (url);
752           xfree (mynewloc);
753           RESTORE_POST_DATA;
754           return WRONGCODE;
755         }
756
757       xfree (url);
758       url = mynewloc;
759       url_free (u);
760       u = newloc_parsed;
761
762       /* If we're being redirected from POST, we don't want to POST
763          again.  Many requests answer POST with a redirection to an
764          index page; that redirection is clearly a GET.  We "suspend"
765          POST data for the duration of the redirections, and restore
766          it when we're done. */
767       if (!post_data_suspended)
768         SUSPEND_POST_DATA;
769
770       goto redirected;
771     }
772
773   if (local_file)
774     {
775       if (*dt & RETROKF)
776         {
777           register_download (u->url, local_file);
778           if (redirection_count && 0 != strcmp (origurl, u->url))
779             register_redirection (origurl, u->url);
780           if (*dt & TEXTHTML)
781             register_html (u->url, local_file);
782           if (*dt & TEXTCSS)
783             register_css (u->url, local_file);
784         }
785     }
786
787   if (file)
788     *file = local_file ? local_file : NULL;
789   else
790     xfree_null (local_file);
791
792   url_free (u);
793
794   if (redirection_count)
795     {
796       if (newloc)
797         *newloc = url;
798       else
799         xfree (url);
800     }
801   else
802     {
803       if (newloc)
804         *newloc = NULL;
805       xfree (url);
806     }
807
808   RESTORE_POST_DATA;
809
810   return result;
811 }
812
813 /* Find the URLs in the file and call retrieve_url() for each of them.
814    If HTML is true, treat the file as HTML, and construct the URLs
815    accordingly.
816
817    If opt.recursive is set, call retrieve_tree() for each file.  */
818
819 uerr_t
820 retrieve_from_file (const char *file, bool html, int *count)
821 {
822   uerr_t status;
823   struct urlpos *url_list, *cur_url;
824
825   char *input_file = NULL;
826   const char *url = file;
827
828   status = RETROK;             /* Suppose everything is OK.  */
829   *count = 0;                  /* Reset the URL count.  */
830   
831   if (url_has_scheme (url))
832     {
833       uerr_t status;
834       status = retrieve_url (url, &input_file, NULL, NULL, NULL, false);
835       if (status != RETROK)
836         return status;
837     }
838   else
839     input_file = (char *) file;
840
841   url_list = (html ? get_urls_html (input_file, NULL, NULL)
842               : get_urls_file (input_file));
843
844   for (cur_url = url_list; cur_url; cur_url = cur_url->next, ++*count)
845     {
846       char *filename = NULL, *new_file = NULL;
847       int dt;
848
849       if (cur_url->ignore_when_downloading)
850         continue;
851
852       if (opt.quota && total_downloaded_bytes > opt.quota)
853         {
854           status = QUOTEXC;
855           break;
856         }
857       if ((opt.recursive || opt.page_requisites)
858           && (cur_url->url->scheme != SCHEME_FTP || getproxy (cur_url->url)))
859         {
860           int old_follow_ftp = opt.follow_ftp;
861
862           /* Turn opt.follow_ftp on in case of recursive FTP retrieval */
863           if (cur_url->url->scheme == SCHEME_FTP) 
864             opt.follow_ftp = 1;
865           
866           status = retrieve_tree (cur_url->url->url);
867
868           opt.follow_ftp = old_follow_ftp;
869         }
870       else
871         status = retrieve_url (cur_url->url->url, &filename, &new_file, NULL, &dt, opt.recursive);
872
873       if (filename && opt.delete_after && file_exists_p (filename))
874         {
875           DEBUGP (("\
876 Removing file due to --delete-after in retrieve_from_file():\n"));
877           logprintf (LOG_VERBOSE, _("Removing %s.\n"), filename);
878           if (unlink (filename))
879             logprintf (LOG_NOTQUIET, "unlink: %s\n", strerror (errno));
880           dt &= ~RETROKF;
881         }
882
883       xfree_null (new_file);
884       xfree_null (filename);
885     }
886
887   /* Free the linked list of URL-s.  */
888   free_urlpos (url_list);
889
890   return status;
891 }
892
893 /* Print `giving up', or `retrying', depending on the impending
894    action.  N1 and N2 are the attempt number and the attempt limit.  */
895 void
896 printwhat (int n1, int n2)
897 {
898   logputs (LOG_VERBOSE, (n1 == n2) ? _("Giving up.\n\n") : _("Retrying.\n\n"));
899 }
900
901 /* If opt.wait or opt.waitretry are specified, and if certain
902    conditions are met, sleep the appropriate number of seconds.  See
903    the documentation of --wait and --waitretry for more information.
904
905    COUNT is the count of current retrieval, beginning with 1. */
906
907 void
908 sleep_between_retrievals (int count)
909 {
910   static bool first_retrieval = true;
911
912   if (first_retrieval)
913     {
914       /* Don't sleep before the very first retrieval. */
915       first_retrieval = false;
916       return;
917     }
918
919   if (opt.waitretry && count > 1)
920     {
921       /* If opt.waitretry is specified and this is a retry, wait for
922          COUNT-1 number of seconds, or for opt.waitretry seconds.  */
923       if (count <= opt.waitretry)
924         xsleep (count - 1);
925       else
926         xsleep (opt.waitretry);
927     }
928   else if (opt.wait)
929     {
930       if (!opt.random_wait || count > 1)
931         /* If random-wait is not specified, or if we are sleeping
932            between retries of the same download, sleep the fixed
933            interval.  */
934         xsleep (opt.wait);
935       else
936         {
937           /* Sleep a random amount of time averaging in opt.wait
938              seconds.  The sleeping amount ranges from 0.5*opt.wait to
939              1.5*opt.wait.  */
940           double waitsecs = (0.5 + random_float ()) * opt.wait;
941           DEBUGP (("sleep_between_retrievals: avg=%f,sleep=%f\n",
942                    opt.wait, waitsecs));
943           xsleep (waitsecs);
944         }
945     }
946 }
947
948 /* Free the linked list of urlpos.  */
949 void
950 free_urlpos (struct urlpos *l)
951 {
952   while (l)
953     {
954       struct urlpos *next = l->next;
955       if (l->url)
956         url_free (l->url);
957       xfree_null (l->local_name);
958       xfree (l);
959       l = next;
960     }
961 }
962
963 /* Rotate FNAME opt.backups times */
964 void
965 rotate_backups(const char *fname)
966 {
967   int maxlen = strlen (fname) + 1 + numdigit (opt.backups) + 1;
968   char *from = (char *)alloca (maxlen);
969   char *to = (char *)alloca (maxlen);
970   struct_stat sb;
971   int i;
972
973   if (stat (fname, &sb) == 0)
974     if (S_ISREG (sb.st_mode) == 0)
975       return;
976
977   for (i = opt.backups; i > 1; i--)
978     {
979       sprintf (from, "%s.%d", fname, i - 1);
980       sprintf (to, "%s.%d", fname, i);
981       rename (from, to);
982     }
983
984   sprintf (to, "%s.%d", fname, 1);
985   rename(fname, to);
986 }
987
988 static bool no_proxy_match (const char *, const char **);
989
990 /* Return the URL of the proxy appropriate for url U.  */
991
992 static char *
993 getproxy (struct url *u)
994 {
995   char *proxy = NULL;
996   char *rewritten_url;
997   static char rewritten_storage[1024];
998
999   if (!opt.use_proxy)
1000     return NULL;
1001   if (no_proxy_match (u->host, (const char **)opt.no_proxy))
1002     return NULL;
1003
1004   switch (u->scheme)
1005     {
1006     case SCHEME_HTTP:
1007       proxy = opt.http_proxy ? opt.http_proxy : getenv ("http_proxy");
1008       break;
1009 #ifdef HAVE_SSL
1010     case SCHEME_HTTPS:
1011       proxy = opt.https_proxy ? opt.https_proxy : getenv ("https_proxy");
1012       break;
1013 #endif
1014     case SCHEME_FTP:
1015       proxy = opt.ftp_proxy ? opt.ftp_proxy : getenv ("ftp_proxy");
1016       break;
1017     case SCHEME_INVALID:
1018       break;
1019     }
1020   if (!proxy || !*proxy)
1021     return NULL;
1022
1023   /* Handle shorthands.  `rewritten_storage' is a kludge to allow
1024      getproxy() to return static storage. */
1025   rewritten_url = rewrite_shorthand_url (proxy);
1026   if (rewritten_url)
1027     {
1028       strncpy (rewritten_storage, rewritten_url, sizeof (rewritten_storage));
1029       rewritten_storage[sizeof (rewritten_storage) - 1] = '\0';
1030       proxy = rewritten_storage;
1031     }
1032
1033   return proxy;
1034 }
1035
1036 /* Returns true if URL would be downloaded through a proxy. */
1037
1038 bool
1039 url_uses_proxy (const char *url)
1040 {
1041   bool ret;
1042   struct url *u = url_parse (url, NULL);
1043   if (!u)
1044     return false;
1045   ret = getproxy (u) != NULL;
1046   url_free (u);
1047   return ret;
1048 }
1049
1050 /* Should a host be accessed through proxy, concerning no_proxy?  */
1051 static bool
1052 no_proxy_match (const char *host, const char **no_proxy)
1053 {
1054   if (!no_proxy)
1055     return false;
1056   else
1057     return sufmatch (no_proxy, host);
1058 }