]> sjero.net Git - wget/blob - src/retr.c
Get rid of the supplementary bool pointer in url_parse () arguments; UGLY :)
[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 "iri.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   reset_utf8_encode ();
629
630  second_try:
631   u = url_parse (url, &up_error_code);
632   if (!u)
633     {
634       logprintf (LOG_NOTQUIET, "%s: %s.\n", url, url_error (up_error_code));
635       xfree (url);
636       return URLERROR;
637     }
638
639   /*printf ("[Retrieving %s with %s (UTF-8=%d)\n", url, get_remote_charset (), utf8_encoded);*/
640
641   if (!refurl)
642     refurl = opt.referer;
643
644  redirected:
645
646   result = NOCONERROR;
647   mynewloc = NULL;
648   local_file = NULL;
649   proxy_url = NULL;
650
651   proxy = getproxy (u);
652   if (proxy)
653     {
654       /* sXXXav : support IRI for proxy */
655       /* Parse the proxy URL.  */
656       set_ugly_no_encode (true);
657       proxy_url = url_parse (proxy, &up_error_code);
658       set_ugly_no_encode (false);
659       if (!proxy_url)
660         {
661           logprintf (LOG_NOTQUIET, _("Error parsing proxy URL %s: %s.\n"),
662                      proxy, url_error (up_error_code));
663           xfree (url);
664           RESTORE_POST_DATA;
665           return PROXERR;
666         }
667       if (proxy_url->scheme != SCHEME_HTTP && proxy_url->scheme != u->scheme)
668         {
669           logprintf (LOG_NOTQUIET, _("Error in proxy URL %s: Must be HTTP.\n"), proxy);
670           url_free (proxy_url);
671           xfree (url);
672           RESTORE_POST_DATA;
673           return PROXERR;
674         }
675     }
676
677   if (u->scheme == SCHEME_HTTP
678 #ifdef HAVE_SSL
679       || u->scheme == SCHEME_HTTPS
680 #endif
681       || (proxy_url && proxy_url->scheme == SCHEME_HTTP))
682     {
683       result = http_loop (u, &mynewloc, &local_file, refurl, dt, proxy_url);
684     }
685   else if (u->scheme == SCHEME_FTP)
686     {
687       /* If this is a redirection, temporarily turn off opt.ftp_glob
688          and opt.recursive, both being undesirable when following
689          redirects.  */
690       bool oldrec = recursive, glob = opt.ftp_glob;
691       if (redirection_count)
692         oldrec = glob = false;
693
694       result = ftp_loop (u, dt, proxy_url, recursive, glob);
695       recursive = oldrec;
696
697       /* There is a possibility of having HTTP being redirected to
698          FTP.  In these cases we must decide whether the text is HTML
699          according to the suffix.  The HTML suffixes are `.html',
700          `.htm' and a few others, case-insensitive.  */
701       if (redirection_count && local_file && u->scheme == SCHEME_FTP)
702         {
703           if (has_html_suffix_p (local_file))
704             *dt |= TEXTHTML;
705         }
706     }
707
708   if (proxy_url)
709     {
710       url_free (proxy_url);
711       proxy_url = NULL;
712     }
713
714   location_changed = (result == NEWLOCATION);
715   if (location_changed)
716     {
717       char *construced_newloc;
718       struct url *newloc_parsed;
719
720       assert (mynewloc != NULL);
721
722       if (local_file)
723         xfree (local_file);
724
725       /* The HTTP specs only allow absolute URLs to appear in
726          redirects, but a ton of boneheaded webservers and CGIs out
727          there break the rules and use relative URLs, and popular
728          browsers are lenient about this, so wget should be too. */
729       construced_newloc = uri_merge (url, mynewloc);
730       xfree (mynewloc);
731       mynewloc = construced_newloc;
732
733       reset_utf8_encode ();
734
735       /* Now, see if this new location makes sense. */
736       newloc_parsed = url_parse (mynewloc, &up_error_code);
737       if (!newloc_parsed)
738         {
739           logprintf (LOG_NOTQUIET, "%s: %s.\n", escnonprint_uri (mynewloc),
740                      url_error (up_error_code));
741           url_free (u);
742           xfree (url);
743           xfree (mynewloc);
744           RESTORE_POST_DATA;
745           return result;
746         }
747
748       /* Now mynewloc will become newloc_parsed->url, because if the
749          Location contained relative paths like .././something, we
750          don't want that propagating as url.  */
751       xfree (mynewloc);
752       mynewloc = xstrdup (newloc_parsed->url);
753
754       /* Check for max. number of redirections.  */
755       if (++redirection_count > opt.max_redirect)
756         {
757           logprintf (LOG_NOTQUIET, _("%d redirections exceeded.\n"),
758                      opt.max_redirect);
759           url_free (newloc_parsed);
760           url_free (u);
761           xfree (url);
762           xfree (mynewloc);
763           RESTORE_POST_DATA;
764           return WRONGCODE;
765         }
766
767       xfree (url);
768       url = mynewloc;
769       url_free (u);
770       u = newloc_parsed;
771
772       /* If we're being redirected from POST, we don't want to POST
773          again.  Many requests answer POST with a redirection to an
774          index page; that redirection is clearly a GET.  We "suspend"
775          POST data for the duration of the redirections, and restore
776          it when we're done. */
777       if (!post_data_suspended)
778         SUSPEND_POST_DATA;
779
780       goto redirected;
781     }
782
783   /* Try to not encode in UTF-8 if fetching failed */
784   if (!(*dt & RETROKF) && get_utf8_encode ())
785     {
786       set_utf8_encode (false);
787       /*printf ("[Fallbacking to non-utf8 for `%s'\n", url);*/
788       goto second_try;
789     }
790
791   if (local_file && *dt & RETROKF)
792     {
793       register_download (u->url, local_file);
794       if (redirection_count && 0 != strcmp (origurl, u->url))
795         register_redirection (origurl, u->url);
796       if (*dt & TEXTHTML)
797         register_html (u->url, local_file);
798     }
799
800   if (file)
801     *file = local_file ? local_file : NULL;
802   else
803     xfree_null (local_file);
804
805   url_free (u);
806
807   if (redirection_count)
808     {
809       if (newloc)
810         *newloc = url;
811       else
812         xfree (url);
813     }
814   else
815     {
816       if (newloc)
817         *newloc = NULL;
818       xfree (url);
819     }
820
821   RESTORE_POST_DATA;
822
823   return result;
824 }
825
826 /* Find the URLs in the file and call retrieve_url() for each of them.
827    If HTML is true, treat the file as HTML, and construct the URLs
828    accordingly.
829
830    If opt.recursive is set, call retrieve_tree() for each file.  */
831
832 uerr_t
833 retrieve_from_file (const char *file, bool html, int *count)
834 {
835   uerr_t status;
836   struct urlpos *url_list, *cur_url;
837
838   url_list = (html ? get_urls_html (file, NULL, NULL)
839               : get_urls_file (file));
840   status = RETROK;             /* Suppose everything is OK.  */
841   *count = 0;                  /* Reset the URL count.  */
842
843   for (cur_url = url_list; cur_url; cur_url = cur_url->next, ++*count)
844     {
845       char *filename = NULL, *new_file = NULL;
846       int dt;
847
848       if (cur_url->ignore_when_downloading)
849         continue;
850
851       if (opt.quota && total_downloaded_bytes > opt.quota)
852         {
853           status = QUOTEXC;
854           break;
855         }
856       if ((opt.recursive || opt.page_requisites)
857           && (cur_url->url->scheme != SCHEME_FTP || getproxy (cur_url->url)))
858         {
859           int old_follow_ftp = opt.follow_ftp;
860
861           /* Turn opt.follow_ftp on in case of recursive FTP retrieval */
862           if (cur_url->url->scheme == SCHEME_FTP)
863             opt.follow_ftp = 1;
864
865           status = retrieve_tree (cur_url->url->url);
866
867           opt.follow_ftp = old_follow_ftp;
868         }
869       else
870         status = retrieve_url (cur_url->url->url, &filename, &new_file, NULL, &dt, opt.recursive);
871
872       if (filename && opt.delete_after && file_exists_p (filename))
873         {
874           DEBUGP (("\
875 Removing file due to --delete-after in retrieve_from_file():\n"));
876           logprintf (LOG_VERBOSE, _("Removing %s.\n"), filename);
877           if (unlink (filename))
878             logprintf (LOG_NOTQUIET, "unlink: %s\n", strerror (errno));
879           dt &= ~RETROKF;
880         }
881
882       xfree_null (new_file);
883       xfree_null (filename);
884     }
885
886   /* Free the linked list of URL-s.  */
887   free_urlpos (url_list);
888
889   return status;
890 }
891
892 /* Print `giving up', or `retrying', depending on the impending
893    action.  N1 and N2 are the attempt number and the attempt limit.  */
894 void
895 printwhat (int n1, int n2)
896 {
897   logputs (LOG_VERBOSE, (n1 == n2) ? _("Giving up.\n\n") : _("Retrying.\n\n"));
898 }
899
900 /* If opt.wait or opt.waitretry are specified, and if certain
901    conditions are met, sleep the appropriate number of seconds.  See
902    the documentation of --wait and --waitretry for more information.
903
904    COUNT is the count of current retrieval, beginning with 1. */
905
906 void
907 sleep_between_retrievals (int count)
908 {
909   static bool first_retrieval = true;
910
911   if (first_retrieval)
912     {
913       /* Don't sleep before the very first retrieval. */
914       first_retrieval = false;
915       return;
916     }
917
918   if (opt.waitretry && count > 1)
919     {
920       /* If opt.waitretry is specified and this is a retry, wait for
921          COUNT-1 number of seconds, or for opt.waitretry seconds.  */
922       if (count <= opt.waitretry)
923         xsleep (count - 1);
924       else
925         xsleep (opt.waitretry);
926     }
927   else if (opt.wait)
928     {
929       if (!opt.random_wait || count > 1)
930         /* If random-wait is not specified, or if we are sleeping
931            between retries of the same download, sleep the fixed
932            interval.  */
933         xsleep (opt.wait);
934       else
935         {
936           /* Sleep a random amount of time averaging in opt.wait
937              seconds.  The sleeping amount ranges from 0.5*opt.wait to
938              1.5*opt.wait.  */
939           double waitsecs = (0.5 + random_float ()) * opt.wait;
940           DEBUGP (("sleep_between_retrievals: avg=%f,sleep=%f\n",
941                    opt.wait, waitsecs));
942           xsleep (waitsecs);
943         }
944     }
945 }
946
947 /* Free the linked list of urlpos.  */
948 void
949 free_urlpos (struct urlpos *l)
950 {
951   while (l)
952     {
953       struct urlpos *next = l->next;
954       if (l->url)
955         url_free (l->url);
956       xfree_null (l->local_name);
957       xfree (l);
958       l = next;
959     }
960 }
961
962 /* Rotate FNAME opt.backups times */
963 void
964 rotate_backups(const char *fname)
965 {
966   int maxlen = strlen (fname) + 1 + numdigit (opt.backups) + 1;
967   char *from = (char *)alloca (maxlen);
968   char *to = (char *)alloca (maxlen);
969   struct_stat sb;
970   int i;
971
972   if (stat (fname, &sb) == 0)
973     if (S_ISREG (sb.st_mode) == 0)
974       return;
975
976   for (i = opt.backups; i > 1; i--)
977     {
978       sprintf (from, "%s.%d", fname, i - 1);
979       sprintf (to, "%s.%d", fname, i);
980       rename (from, to);
981     }
982
983   sprintf (to, "%s.%d", fname, 1);
984   rename(fname, to);
985 }
986
987 static bool no_proxy_match (const char *, const char **);
988
989 /* Return the URL of the proxy appropriate for url U.  */
990
991 static char *
992 getproxy (struct url *u)
993 {
994   char *proxy = NULL;
995   char *rewritten_url;
996   static char rewritten_storage[1024];
997
998   if (!opt.use_proxy)
999     return NULL;
1000   if (no_proxy_match (u->host, (const char **)opt.no_proxy))
1001     return NULL;
1002
1003   switch (u->scheme)
1004     {
1005     case SCHEME_HTTP:
1006       proxy = opt.http_proxy ? opt.http_proxy : getenv ("http_proxy");
1007       break;
1008 #ifdef HAVE_SSL
1009     case SCHEME_HTTPS:
1010       proxy = opt.https_proxy ? opt.https_proxy : getenv ("https_proxy");
1011       break;
1012 #endif
1013     case SCHEME_FTP:
1014       proxy = opt.ftp_proxy ? opt.ftp_proxy : getenv ("ftp_proxy");
1015       break;
1016     case SCHEME_INVALID:
1017       break;
1018     }
1019   if (!proxy || !*proxy)
1020     return NULL;
1021
1022   /* Handle shorthands.  `rewritten_storage' is a kludge to allow
1023      getproxy() to return static storage. */
1024   rewritten_url = rewrite_shorthand_url (proxy);
1025   if (rewritten_url)
1026     {
1027       strncpy (rewritten_storage, rewritten_url, sizeof (rewritten_storage));
1028       rewritten_storage[sizeof (rewritten_storage) - 1] = '\0';
1029       proxy = rewritten_storage;
1030     }
1031
1032   return proxy;
1033 }
1034
1035 /* Returns true if URL would be downloaded through a proxy. */
1036
1037 bool
1038 url_uses_proxy (const char *url)
1039 {
1040   bool ret;
1041   struct url *u;
1042   set_ugly_no_encode(true);
1043   u= url_parse (url, NULL);
1044   set_ugly_no_encode(false);
1045   if (!u)
1046     return false;
1047   ret = getproxy (u) != NULL;
1048   url_free (u);
1049   return ret;
1050 }
1051
1052 /* Should a host be accessed through proxy, concerning no_proxy?  */
1053 static bool
1054 no_proxy_match (const char *host, const char **no_proxy)
1055 {
1056   if (!no_proxy)
1057     return false;
1058   else
1059     return sufmatch (no_proxy, host);
1060 }