]> sjero.net Git - wget/blob - src/utils.c
[svn] Doc update.
[wget] / src / utils.c
1 /* Various utility functions.
2    Copyright (C) 1996-2005 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
9 (at 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 #include <string.h>
35 #include <time.h>
36 #ifdef HAVE_SYS_TIME_H
37 # include <sys/time.h>
38 #endif
39 #ifdef HAVE_UNISTD_H
40 # include <unistd.h>
41 #endif
42 #ifdef HAVE_MMAP
43 # include <sys/mman.h>
44 #endif
45 #ifdef HAVE_UTIME_H
46 # include <utime.h>
47 #endif
48 #ifdef HAVE_SYS_UTIME_H
49 # include <sys/utime.h>
50 #endif
51 #include <errno.h>
52 #include <fcntl.h>
53 #include <assert.h>
54 #include <stdarg.h>
55 #include <locale.h>
56
57 /* For TIOCGWINSZ and friends: */
58 #ifdef HAVE_SYS_IOCTL_H
59 # include <sys/ioctl.h>
60 #endif
61 #ifdef HAVE_TERMIOS_H
62 # include <termios.h>
63 #endif
64
65 /* Needed for Unix version of run_with_timeout. */
66 #include <signal.h>
67 #include <setjmp.h>
68
69 #ifndef HAVE_SIGSETJMP
70 /* If sigsetjmp is a macro, configure won't pick it up. */
71 # ifdef sigsetjmp
72 #  define HAVE_SIGSETJMP
73 # endif
74 #endif
75
76 #if defined HAVE_SIGSETJMP || defined HAVE_SIGBLOCK
77 # define USE_SIGNAL_TIMEOUT
78 #endif
79
80 #include "wget.h"
81 #include "utils.h"
82 #include "hash.h"
83
84 /* Utility function: like xstrdup(), but also lowercases S.  */
85
86 char *
87 xstrdup_lower (const char *s)
88 {
89   char *copy = xstrdup (s);
90   char *p = copy;
91   for (; *p; p++)
92     *p = TOLOWER (*p);
93   return copy;
94 }
95
96 /* Copy the string formed by two pointers (one on the beginning, other
97    on the char after the last char) to a new, malloc-ed location.
98    0-terminate it.  */
99 char *
100 strdupdelim (const char *beg, const char *end)
101 {
102   char *res = xmalloc (end - beg + 1);
103   memcpy (res, beg, end - beg);
104   res[end - beg] = '\0';
105   return res;
106 }
107
108 /* Parse a string containing comma-separated elements, and return a
109    vector of char pointers with the elements.  Spaces following the
110    commas are ignored.  */
111 char **
112 sepstring (const char *s)
113 {
114   char **res;
115   const char *p;
116   int i = 0;
117
118   if (!s || !*s)
119     return NULL;
120   res = NULL;
121   p = s;
122   while (*s)
123     {
124       if (*s == ',')
125         {
126           res = xrealloc (res, (i + 2) * sizeof (char *));
127           res[i] = strdupdelim (p, s);
128           res[++i] = NULL;
129           ++s;
130           /* Skip the blanks following the ','.  */
131           while (ISSPACE (*s))
132             ++s;
133           p = s;
134         }
135       else
136         ++s;
137     }
138   res = xrealloc (res, (i + 2) * sizeof (char *));
139   res[i] = strdupdelim (p, s);
140   res[i + 1] = NULL;
141   return res;
142 }
143 \f
144 /* Like sprintf, but prints into a string of sufficient size freshly
145    allocated with malloc, which is returned.  If unable to print due
146    to invalid format, returns NULL.  Inability to allocate needed
147    memory results in abort, as with xmalloc.  This is in spirit
148    similar to the GNU/BSD extension asprintf, but somewhat easier to
149    use.
150
151    Internally the function either calls vasprintf or loops around
152    vsnprintf until the correct size is found.  Since Wget also ships a
153    fallback implementation of vsnprintf, this should be portable.  */
154
155 char *
156 aprintf (const char *fmt, ...)
157 {
158 #ifdef HAVE_VASPRINTF
159   /* Use vasprintf. */
160   int ret;
161   va_list args;
162   char *str;
163   va_start (args, fmt);
164   ret = vasprintf (&str, fmt, args);
165   va_end (args);
166   if (ret < 0 && errno == ENOMEM)
167     abort ();                   /* for consistency with xmalloc/xrealloc */
168   else if (ret < 0)
169     return NULL;
170   return str;
171 #else  /* not HAVE_VASPRINTF */
172
173   /* vasprintf is unavailable.  snprintf into a small buffer and
174      resize it as necessary. */
175   int size = 32;
176   char *str = xmalloc (size);
177
178   /* #### This code will infloop and eventually abort in xrealloc if
179      passed a FMT that causes snprintf to consistently return -1.  */
180
181   while (1)
182     {
183       int n;
184       va_list args;
185
186       va_start (args, fmt);
187       n = vsnprintf (str, size, fmt, args);
188       va_end (args);
189
190       /* If the printing worked, return the string. */
191       if (n > -1 && n < size)
192         return str;
193
194       /* Else try again with a larger buffer. */
195       if (n > -1)               /* C99 */
196         size = n + 1;           /* precisely what is needed */
197       else
198         size <<= 1;             /* twice the old size */
199       str = xrealloc (str, size);
200     }
201 #endif /* not HAVE_VASPRINTF */
202 }
203
204 /* Concatenate the NULL-terminated list of string arguments into
205    freshly allocated space.  */
206
207 char *
208 concat_strings (const char *str0, ...)
209 {
210   va_list args;
211   int saved_lengths[5];         /* inspired by Apache's apr_pstrcat */
212   char *ret, *p;
213
214   const char *next_str;
215   int total_length = 0;
216   int argcount;
217
218   /* Calculate the length of and allocate the resulting string. */
219
220   argcount = 0;
221   va_start (args, str0);
222   for (next_str = str0; next_str != NULL; next_str = va_arg (args, char *))
223     {
224       int len = strlen (next_str);
225       if (argcount < countof (saved_lengths))
226         saved_lengths[argcount++] = len;
227       total_length += len;
228     }
229   va_end (args);
230   p = ret = xmalloc (total_length + 1);
231
232   /* Copy the strings into the allocated space. */
233
234   argcount = 0;
235   va_start (args, str0);
236   for (next_str = str0; next_str != NULL; next_str = va_arg (args, char *))
237     {
238       int len;
239       if (argcount < countof (saved_lengths))
240         len = saved_lengths[argcount++];
241       else
242         len = strlen (next_str);
243       memcpy (p, next_str, len);
244       p += len;
245     }
246   va_end (args);
247   *p = '\0';
248
249   return ret;
250 }
251 \f
252 /* Return pointer to a static char[] buffer in which zero-terminated
253    string-representation of TM (in form hh:mm:ss) is printed.
254
255    If TM is NULL, the current time will be used.  */
256
257 char *
258 time_str (time_t *tm)
259 {
260   static char output[15];
261   struct tm *ptm;
262   time_t secs = tm ? *tm : time (NULL);
263
264   if (secs == -1)
265     {
266       /* In case of error, return the empty string.  Maybe we should
267          just abort if this happens?  */
268       *output = '\0';
269       return output;
270     }
271   ptm = localtime (&secs);
272   sprintf (output, "%02d:%02d:%02d", ptm->tm_hour, ptm->tm_min, ptm->tm_sec);
273   return output;
274 }
275
276 /* Like the above, but include the date: YYYY-MM-DD hh:mm:ss.  */
277
278 char *
279 datetime_str (time_t *tm)
280 {
281   static char output[20];       /* "YYYY-MM-DD hh:mm:ss" + \0 */
282   struct tm *ptm;
283   time_t secs = tm ? *tm : time (NULL);
284
285   if (secs == -1)
286     {
287       /* In case of error, return the empty string.  Maybe we should
288          just abort if this happens?  */
289       *output = '\0';
290       return output;
291     }
292   ptm = localtime (&secs);
293   sprintf (output, "%04d-%02d-%02d %02d:%02d:%02d",
294            ptm->tm_year + 1900, ptm->tm_mon + 1, ptm->tm_mday,
295            ptm->tm_hour, ptm->tm_min, ptm->tm_sec);
296   return output;
297 }
298 \f
299 /* The Windows versions of the following two functions are defined in
300    mswindows.c.  */
301
302 #ifndef WINDOWS
303 void
304 fork_to_background (void)
305 {
306   pid_t pid;
307   /* Whether we arrange our own version of opt.lfilename here.  */
308   bool logfile_changed = false;
309
310   if (!opt.lfilename)
311     {
312       /* We must create the file immediately to avoid either a race
313          condition (which arises from using unique_name and failing to
314          use fopen_excl) or lying to the user about the log file name
315          (which arises from using unique_name, printing the name, and
316          using fopen_excl later on.)  */
317       FILE *new_log_fp = unique_create (DEFAULT_LOGFILE, false, &opt.lfilename);
318       if (new_log_fp)
319         {
320           logfile_changed = true;
321           fclose (new_log_fp);
322         }
323     }
324   pid = fork ();
325   if (pid < 0)
326     {
327       /* parent, error */
328       perror ("fork");
329       exit (1);
330     }
331   else if (pid != 0)
332     {
333       /* parent, no error */
334       printf (_("Continuing in background, pid %d.\n"), (int) pid);
335       if (logfile_changed)
336         printf (_("Output will be written to `%s'.\n"), opt.lfilename);
337       exit (0);                 /* #### should we use _exit()? */
338     }
339
340   /* child: give up the privileges and keep running. */
341   setsid ();
342   freopen ("/dev/null", "r", stdin);
343   freopen ("/dev/null", "w", stdout);
344   freopen ("/dev/null", "w", stderr);
345 }
346 #endif /* not WINDOWS */
347 \f
348 /* "Touch" FILE, i.e. make its mtime ("modified time") equal the time
349    specified with TM.  The atime ("access time") is set to the current
350    time.  */
351
352 void
353 touch (const char *file, time_t tm)
354 {
355 #ifdef HAVE_STRUCT_UTIMBUF
356   struct utimbuf times;
357 #else
358   struct {
359     time_t actime;
360     time_t modtime;
361   } times;
362 #endif
363   times.modtime = tm;
364   times.actime = time (NULL);
365   if (utime (file, &times) == -1)
366     logprintf (LOG_NOTQUIET, "utime(%s): %s\n", file, strerror (errno));
367 }
368
369 /* Checks if FILE is a symbolic link, and removes it if it is.  Does
370    nothing under MS-Windows.  */
371 int
372 remove_link (const char *file)
373 {
374   int err = 0;
375   struct_stat st;
376
377   if (lstat (file, &st) == 0 && S_ISLNK (st.st_mode))
378     {
379       DEBUGP (("Unlinking %s (symlink).\n", file));
380       err = unlink (file);
381       if (err != 0)
382         logprintf (LOG_VERBOSE, _("Failed to unlink symlink `%s': %s\n"),
383                    file, strerror (errno));
384     }
385   return err;
386 }
387
388 /* Does FILENAME exist?  This is quite a lousy implementation, since
389    it supplies no error codes -- only a yes-or-no answer.  Thus it
390    will return that a file does not exist if, e.g., the directory is
391    unreadable.  I don't mind it too much currently, though.  The
392    proper way should, of course, be to have a third, error state,
393    other than true/false, but that would introduce uncalled-for
394    additional complexity to the callers.  */
395 bool
396 file_exists_p (const char *filename)
397 {
398 #ifdef HAVE_ACCESS
399   return access (filename, F_OK) >= 0;
400 #else
401   struct_stat buf;
402   return stat (filename, &buf) >= 0;
403 #endif
404 }
405
406 /* Returns 0 if PATH is a directory, 1 otherwise (any kind of file).
407    Returns 0 on error.  */
408 bool
409 file_non_directory_p (const char *path)
410 {
411   struct_stat buf;
412   /* Use lstat() rather than stat() so that symbolic links pointing to
413      directories can be identified correctly.  */
414   if (lstat (path, &buf) != 0)
415     return false;
416   return S_ISDIR (buf.st_mode) ? false : true;
417 }
418
419 /* Return the size of file named by FILENAME, or -1 if it cannot be
420    opened or seeked into. */
421 wgint
422 file_size (const char *filename)
423 {
424 #if defined(HAVE_FSEEKO) && defined(HAVE_FTELLO)
425   wgint size;
426   /* We use fseek rather than stat to determine the file size because
427      that way we can also verify that the file is readable without
428      explicitly checking for permissions.  Inspired by the POST patch
429      by Arnaud Wylie.  */
430   FILE *fp = fopen (filename, "rb");
431   if (!fp)
432     return -1;
433   fseeko (fp, 0, SEEK_END);
434   size = ftello (fp);
435   fclose (fp);
436   return size;
437 #else
438   struct_stat st;
439   if (stat (filename, &st) < 0)
440     return -1;
441   return st.st_size;
442 #endif
443 }
444
445 /* stat file names named PREFIX.1, PREFIX.2, etc., until one that
446    doesn't exist is found.  Return a freshly allocated copy of the
447    unused file name.  */
448
449 static char *
450 unique_name_1 (const char *prefix)
451 {
452   int count = 1;
453   int plen = strlen (prefix);
454   char *template = (char *)alloca (plen + 1 + 24);
455   char *template_tail = template + plen;
456
457   memcpy (template, prefix, plen);
458   *template_tail++ = '.';
459
460   do
461     number_to_string (template_tail, count++);
462   while (file_exists_p (template));
463
464   return xstrdup (template);
465 }
466
467 /* Return a unique file name, based on FILE.
468
469    More precisely, if FILE doesn't exist, it is returned unmodified.
470    If not, FILE.1 is tried, then FILE.2, etc.  The first FILE.<number>
471    file name that doesn't exist is returned.
472
473    The resulting file is not created, only verified that it didn't
474    exist at the point in time when the function was called.
475    Therefore, where security matters, don't rely that the file created
476    by this function exists until you open it with O_EXCL or
477    equivalent.
478
479    If ALLOW_PASSTHROUGH is 0, it always returns a freshly allocated
480    string.  Otherwise, it may return FILE if the file doesn't exist
481    (and therefore doesn't need changing).  */
482
483 char *
484 unique_name (const char *file, bool allow_passthrough)
485 {
486   /* If the FILE itself doesn't exist, return it without
487      modification. */
488   if (!file_exists_p (file))
489     return allow_passthrough ? (char *)file : xstrdup (file);
490
491   /* Otherwise, find a numeric suffix that results in unused file name
492      and return it.  */
493   return unique_name_1 (file);
494 }
495
496 /* Create a file based on NAME, except without overwriting an existing
497    file with that name.  Providing O_EXCL is correctly implemented,
498    this function does not have the race condition associated with
499    opening the file returned by unique_name.  */
500
501 FILE *
502 unique_create (const char *name, bool binary, char **opened_name)
503 {
504   /* unique file name, based on NAME */
505   char *uname = unique_name (name, false);
506   FILE *fp;
507   while ((fp = fopen_excl (uname, binary)) == NULL && errno == EEXIST)
508     {
509       xfree (uname);
510       uname = unique_name (name, false);
511     }
512   if (opened_name && fp != NULL)
513     {
514       if (fp)
515         *opened_name = uname;
516       else
517         {
518           *opened_name = NULL;
519           xfree (uname);
520         }
521     }
522   else
523     xfree (uname);
524   return fp;
525 }
526
527 /* Open the file for writing, with the addition that the file is
528    opened "exclusively".  This means that, if the file already exists,
529    this function will *fail* and errno will be set to EEXIST.  If
530    BINARY is set, the file will be opened in binary mode, equivalent
531    to fopen's "wb".
532
533    If opening the file fails for any reason, including the file having
534    previously existed, this function returns NULL and sets errno
535    appropriately.  */
536    
537 FILE *
538 fopen_excl (const char *fname, bool binary)
539 {
540   int fd;
541 #ifdef O_EXCL
542   int flags = O_WRONLY | O_CREAT | O_EXCL;
543 # ifdef O_BINARY
544   if (binary)
545     flags |= O_BINARY;
546 # endif
547   fd = open (fname, flags, 0666);
548   if (fd < 0)
549     return NULL;
550   return fdopen (fd, binary ? "wb" : "w");
551 #else  /* not O_EXCL */
552   /* Manually check whether the file exists.  This is prone to race
553      conditions, but systems without O_EXCL haven't deserved
554      better.  */
555   if (file_exists_p (fname))
556     {
557       errno = EEXIST;
558       return NULL;
559     }
560   return fopen (fname, binary ? "wb" : "w");
561 #endif /* not O_EXCL */
562 }
563 \f
564 /* Create DIRECTORY.  If some of the pathname components of DIRECTORY
565    are missing, create them first.  In case any mkdir() call fails,
566    return its error status.  Returns 0 on successful completion.
567
568    The behaviour of this function should be identical to the behaviour
569    of `mkdir -p' on systems where mkdir supports the `-p' option.  */
570 int
571 make_directory (const char *directory)
572 {
573   int i, ret, quit = 0;
574   char *dir;
575
576   /* Make a copy of dir, to be able to write to it.  Otherwise, the
577      function is unsafe if called with a read-only char *argument.  */
578   STRDUP_ALLOCA (dir, directory);
579
580   /* If the first character of dir is '/', skip it (and thus enable
581      creation of absolute-pathname directories.  */
582   for (i = (*dir == '/'); 1; ++i)
583     {
584       for (; dir[i] && dir[i] != '/'; i++)
585         ;
586       if (!dir[i])
587         quit = 1;
588       dir[i] = '\0';
589       /* Check whether the directory already exists.  Allow creation of
590          of intermediate directories to fail, as the initial path components
591          are not necessarily directories!  */
592       if (!file_exists_p (dir))
593         ret = mkdir (dir, 0777);
594       else
595         ret = 0;
596       if (quit)
597         break;
598       else
599         dir[i] = '/';
600     }
601   return ret;
602 }
603
604 /* Merge BASE with FILE.  BASE can be a directory or a file name, FILE
605    should be a file name.
606
607    file_merge("/foo/bar", "baz")  => "/foo/baz"
608    file_merge("/foo/bar/", "baz") => "/foo/bar/baz"
609    file_merge("foo", "bar")       => "bar"
610
611    In other words, it's a simpler and gentler version of uri_merge.  */
612
613 char *
614 file_merge (const char *base, const char *file)
615 {
616   char *result;
617   const char *cut = (const char *)strrchr (base, '/');
618
619   if (!cut)
620     return xstrdup (file);
621
622   result = xmalloc (cut - base + 1 + strlen (file) + 1);
623   memcpy (result, base, cut - base);
624   result[cut - base] = '/';
625   strcpy (result + (cut - base) + 1, file);
626
627   return result;
628 }
629 \f
630 /* Like fnmatch, but performs a case-insensitive match.  */
631
632 int
633 fnmatch_nocase (const char *pattern, const char *string, int flags)
634 {
635 #ifdef FNM_CASEFOLD
636   /* The FNM_CASEFOLD flag started as a GNU extension, but it is now
637      also present on *BSD platforms, and possibly elsewhere.  */
638   return fnmatch (pattern, string, flags | FNM_CASEFOLD);
639 #else
640   /* Turn PATTERN and STRING to lower case and call fnmatch on them. */
641   char *patcopy = (char *) alloca (strlen (pattern) + 1);
642   char *strcopy = (char *) alloca (strlen (string) + 1);
643   char *p;
644   for (p = patcopy; *pattern; pattern++, p++)
645     *p = TOLOWER (*pattern);
646   *p = '\0';
647   for (p = strcopy; *string; string++, p++)
648     *p = TOLOWER (*string);
649   *p = '\0';
650   return fnmatch (patcopy, strcopy, flags);
651 #endif
652 }
653
654 static bool in_acclist (const char *const *, const char *, bool);
655
656 /* Determine whether a file is acceptable to be followed, according to
657    lists of patterns to accept/reject.  */
658 bool
659 acceptable (const char *s)
660 {
661   int l = strlen (s);
662
663   while (l && s[l] != '/')
664     --l;
665   if (s[l] == '/')
666     s += (l + 1);
667   if (opt.accepts)
668     {
669       if (opt.rejects)
670         return (in_acclist ((const char *const *)opt.accepts, s, true)
671                 && !in_acclist ((const char *const *)opt.rejects, s, true));
672       else
673         return in_acclist ((const char *const *)opt.accepts, s, true);
674     }
675   else if (opt.rejects)
676     return !in_acclist ((const char *const *)opt.rejects, s, true);
677   return true;
678 }
679
680 /* Compare S1 and S2 frontally; S2 must begin with S1.  E.g. if S1 is
681    `/something', frontcmp() will return true only if S2 begins with
682    `/something'.  */
683 bool
684 frontcmp (const char *s1, const char *s2)
685 {
686   if (!opt.ignore_case)
687     for (; *s1 && *s2 && (*s1 == *s2); ++s1, ++s2)
688       ;
689   else
690     for (; *s1 && *s2 && (TOLOWER (*s1) == TOLOWER (*s2)); ++s1, ++s2)
691       ;
692   return *s1 == '\0';
693 }
694
695 /* Iterate through STRLIST, and return the first element that matches
696    S, through wildcards or front comparison (as appropriate).  */
697 static char *
698 proclist (char **strlist, const char *s)
699 {
700   char **x;
701   int (*matcher) (const char *, const char *, int)
702     = opt.ignore_case ? fnmatch_nocase : fnmatch;
703
704   for (x = strlist; *x; x++)
705     {
706       /* Remove leading '/' */
707       char *p = *x + (**x == '/');
708       if (has_wildcards_p (p))
709         {
710           if (matcher (p, s, FNM_PATHNAME) == 0)
711             break;
712         }
713       else
714         {
715           if (frontcmp (p, s))
716             break;
717         }
718     }
719   return *x;
720 }
721
722 /* Returns whether DIRECTORY is acceptable for download, wrt the
723    include/exclude lists.
724
725    The leading `/' is ignored in paths; relative and absolute paths
726    may be freely intermixed.  */
727
728 bool
729 accdir (const char *directory)
730 {
731   /* Remove starting '/'.  */
732   if (*directory == '/')
733     ++directory;
734   if (opt.includes)
735     {
736       if (!proclist (opt.includes, directory))
737         return false;
738     }
739   if (opt.excludes)
740     {
741       if (proclist (opt.excludes, directory))
742         return false;
743     }
744   return true;
745 }
746
747 /* Return true if STRING ends with TAIL.  For instance:
748
749    match_tail ("abc", "bc", false)  -> 1
750    match_tail ("abc", "ab", false)  -> 0
751    match_tail ("abc", "abc", false) -> 1
752
753    If FOLD_CASE is true, the comparison will be case-insensitive.  */
754
755 bool
756 match_tail (const char *string, const char *tail, bool fold_case)
757 {
758   int i, j;
759
760   /* We want this to be fast, so we code two loops, one with
761      case-folding, one without. */
762
763   if (!fold_case)
764     {
765       for (i = strlen (string), j = strlen (tail); i >= 0 && j >= 0; i--, j--)
766         if (string[i] != tail[j])
767           break;
768     }
769   else
770     {
771       for (i = strlen (string), j = strlen (tail); i >= 0 && j >= 0; i--, j--)
772         if (TOLOWER (string[i]) != TOLOWER (tail[j]))
773           break;
774     }
775
776   /* If the tail was exhausted, the match was succesful.  */
777   if (j == -1)
778     return true;
779   else
780     return false;
781 }
782
783 /* Checks whether string S matches each element of ACCEPTS.  A list
784    element are matched either with fnmatch() or match_tail(),
785    according to whether the element contains wildcards or not.
786
787    If the BACKWARD is false, don't do backward comparison -- just compare
788    them normally.  */
789 static bool
790 in_acclist (const char *const *accepts, const char *s, bool backward)
791 {
792   for (; *accepts; accepts++)
793     {
794       if (has_wildcards_p (*accepts))
795         {
796           int res = opt.ignore_case
797             ? fnmatch_nocase (*accepts, s, 0) : fnmatch (*accepts, s, 0);
798           /* fnmatch returns 0 if the pattern *does* match the string.  */
799           if (res == 0)
800             return true;
801         }
802       else
803         {
804           if (backward)
805             {
806               if (match_tail (s, *accepts, opt.ignore_case))
807                 return true;
808             }
809           else
810             {
811               int cmp = opt.ignore_case
812                 ? strcasecmp (s, *accepts) : strcmp (s, *accepts);
813               if (cmp == 0)
814                 return true;
815             }
816         }
817     }
818   return false;
819 }
820
821 /* Return the location of STR's suffix (file extension).  Examples:
822    suffix ("foo.bar")       -> "bar"
823    suffix ("foo.bar.baz")   -> "baz"
824    suffix ("/foo/bar")      -> NULL
825    suffix ("/foo.bar/baz")  -> NULL  */
826 char *
827 suffix (const char *str)
828 {
829   int i;
830
831   for (i = strlen (str); i && str[i] != '/' && str[i] != '.'; i--)
832     ;
833
834   if (str[i++] == '.')
835     return (char *)str + i;
836   else
837     return NULL;
838 }
839
840 /* Return true if S contains globbing wildcards (`*', `?', `[' or
841    `]').  */
842
843 bool
844 has_wildcards_p (const char *s)
845 {
846   for (; *s; s++)
847     if (*s == '*' || *s == '?' || *s == '[' || *s == ']')
848       return true;
849   return false;
850 }
851
852 /* Return true if FNAME ends with a typical HTML suffix.  The
853    following (case-insensitive) suffixes are presumed to be HTML
854    files:
855    
856      html
857      htm
858      ?html (`?' matches one character)
859
860    #### CAVEAT.  This is not necessarily a good indication that FNAME
861    refers to a file that contains HTML!  */
862 bool
863 has_html_suffix_p (const char *fname)
864 {
865   char *suf;
866
867   if ((suf = suffix (fname)) == NULL)
868     return false;
869   if (!strcasecmp (suf, "html"))
870     return true;
871   if (!strcasecmp (suf, "htm"))
872     return true;
873   if (suf[0] && !strcasecmp (suf + 1, "html"))
874     return true;
875   return false;
876 }
877
878 /* Read a line from FP and return the pointer to freshly allocated
879    storage.  The storage space is obtained through malloc() and should
880    be freed with free() when it is no longer needed.
881
882    The length of the line is not limited, except by available memory.
883    The newline character at the end of line is retained.  The line is
884    terminated with a zero character.
885
886    After end-of-file is encountered without anything being read, NULL
887    is returned.  NULL is also returned on error.  To distinguish
888    between these two cases, use the stdio function ferror().  */
889
890 char *
891 read_whole_line (FILE *fp)
892 {
893   int length = 0;
894   int bufsize = 82;
895   char *line = xmalloc (bufsize);
896
897   while (fgets (line + length, bufsize - length, fp))
898     {
899       length += strlen (line + length);
900       if (length == 0)
901         /* Possible for example when reading from a binary file where
902            a line begins with \0.  */
903         continue;
904
905       if (line[length - 1] == '\n')
906         break;
907
908       /* fgets() guarantees to read the whole line, or to use up the
909          space we've given it.  We can double the buffer
910          unconditionally.  */
911       bufsize <<= 1;
912       line = xrealloc (line, bufsize);
913     }
914   if (length == 0 || ferror (fp))
915     {
916       xfree (line);
917       return NULL;
918     }
919   if (length + 1 < bufsize)
920     /* Relieve the memory from our exponential greediness.  We say
921        `length + 1' because the terminating \0 is not included in
922        LENGTH.  We don't need to zero-terminate the string ourselves,
923        though, because fgets() does that.  */
924     line = xrealloc (line, length + 1);
925   return line;
926 }
927 \f
928 /* Read FILE into memory.  A pointer to `struct file_memory' are
929    returned; use struct element `content' to access file contents, and
930    the element `length' to know the file length.  `content' is *not*
931    zero-terminated, and you should *not* read or write beyond the [0,
932    length) range of characters.
933
934    After you are done with the file contents, call read_file_free to
935    release the memory.
936
937    Depending on the operating system and the type of file that is
938    being read, read_file() either mmap's the file into memory, or
939    reads the file into the core using read().
940
941    If file is named "-", fileno(stdin) is used for reading instead.
942    If you want to read from a real file named "-", use "./-" instead.  */
943
944 struct file_memory *
945 read_file (const char *file)
946 {
947   int fd;
948   struct file_memory *fm;
949   long size;
950   bool inhibit_close = false;
951
952   /* Some magic in the finest tradition of Perl and its kin: if FILE
953      is "-", just use stdin.  */
954   if (HYPHENP (file))
955     {
956       fd = fileno (stdin);
957       inhibit_close = true;
958       /* Note that we don't inhibit mmap() in this case.  If stdin is
959          redirected from a regular file, mmap() will still work.  */
960     }
961   else
962     fd = open (file, O_RDONLY);
963   if (fd < 0)
964     return NULL;
965   fm = xnew (struct file_memory);
966
967 #ifdef HAVE_MMAP
968   {
969     struct_fstat buf;
970     if (fstat (fd, &buf) < 0)
971       goto mmap_lose;
972     fm->length = buf.st_size;
973     /* NOTE: As far as I know, the callers of this function never
974        modify the file text.  Relying on this would enable us to
975        specify PROT_READ and MAP_SHARED for a marginal gain in
976        efficiency, but at some cost to generality.  */
977     fm->content = mmap (NULL, fm->length, PROT_READ | PROT_WRITE,
978                         MAP_PRIVATE, fd, 0);
979     if (fm->content == (char *)MAP_FAILED)
980       goto mmap_lose;
981     if (!inhibit_close)
982       close (fd);
983
984     fm->mmap_p = 1;
985     return fm;
986   }
987
988  mmap_lose:
989   /* The most common reason why mmap() fails is that FD does not point
990      to a plain file.  However, it's also possible that mmap() doesn't
991      work for a particular type of file.  Therefore, whenever mmap()
992      fails, we just fall back to the regular method.  */
993 #endif /* HAVE_MMAP */
994
995   fm->length = 0;
996   size = 512;                   /* number of bytes fm->contents can
997                                    hold at any given time. */
998   fm->content = xmalloc (size);
999   while (1)
1000     {
1001       wgint nread;
1002       if (fm->length > size / 2)
1003         {
1004           /* #### I'm not sure whether the whole exponential-growth
1005              thing makes sense with kernel read.  On Linux at least,
1006              read() refuses to read more than 4K from a file at a
1007              single chunk anyway.  But other Unixes might optimize it
1008              better, and it doesn't *hurt* anything, so I'm leaving
1009              it.  */
1010
1011           /* Normally, we grow SIZE exponentially to make the number
1012              of calls to read() and realloc() logarithmic in relation
1013              to file size.  However, read() can read an amount of data
1014              smaller than requested, and it would be unreasonable to
1015              double SIZE every time *something* was read.  Therefore,
1016              we double SIZE only when the length exceeds half of the
1017              entire allocated size.  */
1018           size <<= 1;
1019           fm->content = xrealloc (fm->content, size);
1020         }
1021       nread = read (fd, fm->content + fm->length, size - fm->length);
1022       if (nread > 0)
1023         /* Successful read. */
1024         fm->length += nread;
1025       else if (nread < 0)
1026         /* Error. */
1027         goto lose;
1028       else
1029         /* EOF */
1030         break;
1031     }
1032   if (!inhibit_close)
1033     close (fd);
1034   if (size > fm->length && fm->length != 0)
1035     /* Due to exponential growth of fm->content, the allocated region
1036        might be much larger than what is actually needed.  */
1037     fm->content = xrealloc (fm->content, fm->length);
1038   fm->mmap_p = 0;
1039   return fm;
1040
1041  lose:
1042   if (!inhibit_close)
1043     close (fd);
1044   xfree (fm->content);
1045   xfree (fm);
1046   return NULL;
1047 }
1048
1049 /* Release the resources held by FM.  Specifically, this calls
1050    munmap() or xfree() on fm->content, depending whether mmap or
1051    malloc/read were used to read in the file.  It also frees the
1052    memory needed to hold the FM structure itself.  */
1053
1054 void
1055 read_file_free (struct file_memory *fm)
1056 {
1057 #ifdef HAVE_MMAP
1058   if (fm->mmap_p)
1059     {
1060       munmap (fm->content, fm->length);
1061     }
1062   else
1063 #endif
1064     {
1065       xfree (fm->content);
1066     }
1067   xfree (fm);
1068 }
1069 \f
1070 /* Free the pointers in a NULL-terminated vector of pointers, then
1071    free the pointer itself.  */
1072 void
1073 free_vec (char **vec)
1074 {
1075   if (vec)
1076     {
1077       char **p = vec;
1078       while (*p)
1079         xfree (*p++);
1080       xfree (vec);
1081     }
1082 }
1083
1084 /* Append vector V2 to vector V1.  The function frees V2 and
1085    reallocates V1 (thus you may not use the contents of neither
1086    pointer after the call).  If V1 is NULL, V2 is returned.  */
1087 char **
1088 merge_vecs (char **v1, char **v2)
1089 {
1090   int i, j;
1091
1092   if (!v1)
1093     return v2;
1094   if (!v2)
1095     return v1;
1096   if (!*v2)
1097     {
1098       /* To avoid j == 0 */
1099       xfree (v2);
1100       return v1;
1101     }
1102   /* Count v1.  */
1103   for (i = 0; v1[i]; i++)
1104     ;
1105   /* Count v2.  */
1106   for (j = 0; v2[j]; j++)
1107     ;
1108   /* Reallocate v1.  */
1109   v1 = xrealloc (v1, (i + j + 1) * sizeof (char **));
1110   memcpy (v1 + i, v2, (j + 1) * sizeof (char *));
1111   xfree (v2);
1112   return v1;
1113 }
1114
1115 /* Append a freshly allocated copy of STR to VEC.  If VEC is NULL, it
1116    is allocated as needed.  Return the new value of the vector. */
1117
1118 char **
1119 vec_append (char **vec, const char *str)
1120 {
1121   int cnt;                      /* count of vector elements, including
1122                                    the one we're about to append */
1123   if (vec != NULL)
1124     {
1125       for (cnt = 0; vec[cnt]; cnt++)
1126         ;
1127       ++cnt;
1128     }
1129   else
1130     cnt = 1;
1131   /* Reallocate the array to fit the new element and the NULL. */
1132   vec = xrealloc (vec, (cnt + 1) * sizeof (char *));
1133   /* Append a copy of STR to the vector. */
1134   vec[cnt - 1] = xstrdup (str);
1135   vec[cnt] = NULL;
1136   return vec;
1137 }
1138 \f
1139 /* Sometimes it's useful to create "sets" of strings, i.e. special
1140    hash tables where you want to store strings as keys and merely
1141    query for their existence.  Here is a set of utility routines that
1142    makes that transparent.  */
1143
1144 void
1145 string_set_add (struct hash_table *ht, const char *s)
1146 {
1147   /* First check whether the set element already exists.  If it does,
1148      do nothing so that we don't have to free() the old element and
1149      then strdup() a new one.  */
1150   if (hash_table_contains (ht, s))
1151     return;
1152
1153   /* We use "1" as value.  It provides us a useful and clear arbitrary
1154      value, and it consumes no memory -- the pointers to the same
1155      string "1" will be shared by all the key-value pairs in all `set'
1156      hash tables.  */
1157   hash_table_put (ht, xstrdup (s), "1");
1158 }
1159
1160 /* Synonym for hash_table_contains... */
1161
1162 int
1163 string_set_contains (struct hash_table *ht, const char *s)
1164 {
1165   return hash_table_contains (ht, s);
1166 }
1167
1168 static int
1169 string_set_to_array_mapper (void *key, void *value_ignored, void *arg)
1170 {
1171   char ***arrayptr = (char ***) arg;
1172   *(*arrayptr)++ = (char *) key;
1173   return 0;
1174 }
1175
1176 /* Convert the specified string set to array.  ARRAY should be large
1177    enough to hold hash_table_count(ht) char pointers.  */
1178
1179 void string_set_to_array (struct hash_table *ht, char **array)
1180 {
1181   hash_table_map (ht, string_set_to_array_mapper, &array);
1182 }
1183
1184 static int
1185 string_set_free_mapper (void *key, void *value_ignored, void *arg_ignored)
1186 {
1187   xfree (key);
1188   return 0;
1189 }
1190
1191 void
1192 string_set_free (struct hash_table *ht)
1193 {
1194   hash_table_map (ht, string_set_free_mapper, NULL);
1195   hash_table_destroy (ht);
1196 }
1197
1198 static int
1199 free_keys_and_values_mapper (void *key, void *value, void *arg_ignored)
1200 {
1201   xfree (key);
1202   xfree (value);
1203   return 0;
1204 }
1205
1206 /* Another utility function: call free() on all keys and values of HT.  */
1207
1208 void
1209 free_keys_and_values (struct hash_table *ht)
1210 {
1211   hash_table_map (ht, free_keys_and_values_mapper, NULL);
1212 }
1213
1214 \f
1215 /* Get grouping data, the separator and grouping info, by calling
1216    localeconv().  The information is cached after the first call to
1217    the function.
1218
1219    In locales that don't set a thousand separator (such as the "C"
1220    locale), this forces it to be ",".  We are now only showing
1221    thousand separators in one place, so this shouldn't be a problem in
1222    practice.  */
1223
1224 static void
1225 get_grouping_data (const char **sep, const char **grouping)
1226 {
1227   static const char *cached_sep;
1228   static const char *cached_grouping;
1229   static bool initialized;
1230   if (!initialized)
1231     {
1232       /* Get the grouping info from the locale. */
1233       struct lconv *lconv = localeconv ();
1234       cached_sep = lconv->thousands_sep;
1235       cached_grouping = lconv->grouping;
1236       if (!*cached_sep)
1237         {
1238           /* Many locales (such as "C" or "hr_HR") don't specify
1239              grouping, which we still want to use it for legibility.
1240              In those locales set the sep char to ',', unless that
1241              character is used for decimal point, in which case set it
1242              to ".".  */
1243           if (*lconv->decimal_point != ',')
1244             cached_sep = ",";
1245           else
1246             cached_sep = ".";
1247           cached_grouping = "\x03";
1248         }
1249       initialized = true;
1250     }
1251   *sep = cached_sep;
1252   *grouping = cached_grouping;
1253 }
1254
1255 /* Return a printed representation of N with thousand separators.
1256    This should respect locale settings, with the exception of the "C"
1257    locale which mandates no separator, but we use one anyway.
1258
1259    Unfortunately, we cannot use %'d (in fact it would be %'j) to get
1260    the separators because it's too non-portable, and it's hard to test
1261    for this feature at configure time.  Besides, it wouldn't work in
1262    the "C" locale, which many Unix users still work in.  */
1263
1264 const char *
1265 with_thousand_seps (wgint n)
1266 {
1267   static char outbuf[48];
1268   char *p = outbuf + sizeof outbuf;
1269
1270   /* Info received from locale */
1271   const char *grouping, *sep;
1272   int seplen;
1273
1274   /* State information */
1275   int i = 0, groupsize;
1276   const char *atgroup;
1277
1278   bool negative = n < 0;
1279
1280   /* Initialize grouping data. */
1281   get_grouping_data (&sep, &grouping);
1282   seplen = strlen (sep);
1283   atgroup = grouping;
1284   groupsize = *atgroup++;
1285
1286   /* This will overflow on WGINT_MIN, but we're not using this to
1287      print negative numbers anyway.  */
1288   if (negative)
1289     n = -n;
1290
1291   /* Write the number into the buffer, backwards, inserting the
1292      separators as necessary.  */
1293   *--p = '\0';
1294   while (1)
1295     {
1296       *--p = n % 10 + '0';
1297       n /= 10;
1298       if (n == 0)
1299         break;
1300       /* Prepend SEP to every groupsize'd digit and get new groupsize.  */
1301       if (++i == groupsize)
1302         {
1303           if (seplen == 1)
1304             *--p = *sep;
1305           else
1306             memcpy (p -= seplen, sep, seplen);
1307           i = 0;
1308           if (*atgroup)
1309             groupsize = *atgroup++;
1310         }
1311     }
1312   if (negative)
1313     *--p = '-';
1314
1315   return p;
1316 }
1317
1318 /* N, a byte quantity, is converted to a human-readable abberviated
1319    form a la sizes printed by `ls -lh'.  The result is written to a
1320    static buffer, a pointer to which is returned.
1321
1322    Unlike `with_thousand_seps', this approximates to the nearest unit.
1323    Quoting GNU libit: "Most people visually process strings of 3-4
1324    digits effectively, but longer strings of digits are more prone to
1325    misinterpretation.  Hence, converting to an abbreviated form
1326    usually improves readability."
1327
1328    This intentionally uses kilobyte (KB), megabyte (MB), etc. in their
1329    original computer-related meaning of "powers of 1024".  Powers of
1330    1000 would be useless since Wget already displays sizes with
1331    thousand separators.  We don't use the "*bibyte" names invented in
1332    1998, and seldom used in practice.  Wikipedia's entry on kilobyte
1333    discusses this in some detail.  */
1334
1335 char *
1336 human_readable (HR_NUMTYPE n)
1337 {
1338   /* These suffixes are compatible with those of GNU `ls -lh'. */
1339   static char powers[] =
1340     {
1341       'K',                      /* kilobyte, 2^10 bytes */
1342       'M',                      /* megabyte, 2^20 bytes */
1343       'G',                      /* gigabyte, 2^30 bytes */
1344       'T',                      /* terabyte, 2^40 bytes */
1345       'P',                      /* petabyte, 2^50 bytes */
1346       'E',                      /* exabyte,  2^60 bytes */
1347     };
1348   static char buf[8];
1349   int i;
1350
1351   /* If the quantity is smaller than 1K, just print it. */
1352   if (n < 1024)
1353     {
1354       snprintf (buf, sizeof (buf), "%d", (int) n);
1355       return buf;
1356     }
1357
1358   /* Loop over powers, dividing N with 1024 in each iteration.  This
1359      works unchanged for all sizes of wgint, while still avoiding
1360      non-portable `long double' arithmetic.  */
1361   for (i = 0; i < countof (powers); i++)
1362     {
1363       /* At each iteration N is greater than the *subsequent* power.
1364          That way N/1024.0 produces a decimal number in the units of
1365          *this* power.  */
1366       if ((n / 1024) < 1024 || i == countof (powers) - 1)
1367         {
1368           double val = n / 1024.0;
1369           /* Print values smaller than 10 with one decimal digits, and
1370              others without any decimals.  */
1371           snprintf (buf, sizeof (buf), "%.*f%c",
1372                     val < 10 ? 1 : 0, val, powers[i]);
1373           return buf;
1374         }
1375       n /= 1024;
1376     }
1377   return NULL;                  /* unreached */
1378 }
1379
1380 /* Count the digits in the provided number.  Used to allocate space
1381    when printing numbers.  */
1382
1383 int
1384 numdigit (wgint number)
1385 {
1386   int cnt = 1;
1387   if (number < 0)
1388     ++cnt;                      /* accomodate '-' */
1389   while ((number /= 10) != 0)
1390     ++cnt;
1391   return cnt;
1392 }
1393
1394 #define PR(mask) *p++ = n / (mask) + '0'
1395
1396 /* DIGITS_<D> is used to print a D-digit number and should be called
1397    with mask==10^(D-1).  It prints n/mask (the first digit), reducing
1398    n to n%mask (the remaining digits), and calling DIGITS_<D-1>.
1399    Recursively this continues until DIGITS_1 is invoked.  */
1400
1401 #define DIGITS_1(mask) PR (mask)
1402 #define DIGITS_2(mask) PR (mask), n %= (mask), DIGITS_1 ((mask) / 10)
1403 #define DIGITS_3(mask) PR (mask), n %= (mask), DIGITS_2 ((mask) / 10)
1404 #define DIGITS_4(mask) PR (mask), n %= (mask), DIGITS_3 ((mask) / 10)
1405 #define DIGITS_5(mask) PR (mask), n %= (mask), DIGITS_4 ((mask) / 10)
1406 #define DIGITS_6(mask) PR (mask), n %= (mask), DIGITS_5 ((mask) / 10)
1407 #define DIGITS_7(mask) PR (mask), n %= (mask), DIGITS_6 ((mask) / 10)
1408 #define DIGITS_8(mask) PR (mask), n %= (mask), DIGITS_7 ((mask) / 10)
1409 #define DIGITS_9(mask) PR (mask), n %= (mask), DIGITS_8 ((mask) / 10)
1410 #define DIGITS_10(mask) PR (mask), n %= (mask), DIGITS_9 ((mask) / 10)
1411
1412 /* DIGITS_<11-20> are only used on machines with 64-bit wgints. */
1413
1414 #define DIGITS_11(mask) PR (mask), n %= (mask), DIGITS_10 ((mask) / 10)
1415 #define DIGITS_12(mask) PR (mask), n %= (mask), DIGITS_11 ((mask) / 10)
1416 #define DIGITS_13(mask) PR (mask), n %= (mask), DIGITS_12 ((mask) / 10)
1417 #define DIGITS_14(mask) PR (mask), n %= (mask), DIGITS_13 ((mask) / 10)
1418 #define DIGITS_15(mask) PR (mask), n %= (mask), DIGITS_14 ((mask) / 10)
1419 #define DIGITS_16(mask) PR (mask), n %= (mask), DIGITS_15 ((mask) / 10)
1420 #define DIGITS_17(mask) PR (mask), n %= (mask), DIGITS_16 ((mask) / 10)
1421 #define DIGITS_18(mask) PR (mask), n %= (mask), DIGITS_17 ((mask) / 10)
1422 #define DIGITS_19(mask) PR (mask), n %= (mask), DIGITS_18 ((mask) / 10)
1423
1424 /* SPRINTF_WGINT is used by number_to_string to handle pathological
1425    cases and to portably support strange sizes of wgint.  Ideally this
1426    would just use "%j" and intmax_t, but many systems don't support
1427    it, so it's used only if nothing else works.  */
1428 #if SIZEOF_LONG >= SIZEOF_WGINT
1429 # define SPRINTF_WGINT(buf, n) sprintf (buf, "%ld", (long) (n))
1430 #elif SIZEOF_LONG_LONG >= SIZEOF_WGINT
1431 # define SPRINTF_WGINT(buf, n) sprintf (buf, "%lld", (long long) (n))
1432 #elif defined(WINDOWS)
1433 # define SPRINTF_WGINT(buf, n) sprintf (buf, "%I64d", (__int64) (n))
1434 #else
1435 # define SPRINTF_WGINT(buf, n) sprintf (buf, "%j", (intmax_t) (n))
1436 #endif
1437
1438 /* Shorthand for casting to wgint. */
1439 #define W wgint
1440
1441 /* Print NUMBER to BUFFER in base 10.  This is equivalent to
1442    `sprintf(buffer, "%lld", (long long) number)', only typically much
1443    faster and portable to machines without long long.
1444
1445    The speedup may make a difference in programs that frequently
1446    convert numbers to strings.  Some implementations of sprintf,
1447    particularly the one in GNU libc, have been known to be extremely
1448    slow when converting integers to strings.
1449
1450    Return the pointer to the location where the terminating zero was
1451    printed.  (Equivalent to calling buffer+strlen(buffer) after the
1452    function is done.)
1453
1454    BUFFER should be big enough to accept as many bytes as you expect
1455    the number to take up.  On machines with 64-bit longs the maximum
1456    needed size is 24 bytes.  That includes the digits needed for the
1457    largest 64-bit number, the `-' sign in case it's negative, and the
1458    terminating '\0'.  */
1459
1460 char *
1461 number_to_string (char *buffer, wgint number)
1462 {
1463   char *p = buffer;
1464   wgint n = number;
1465
1466 #if (SIZEOF_WGINT != 4) && (SIZEOF_WGINT != 8)
1467   /* We are running in a strange or misconfigured environment.  Let
1468      sprintf cope with it.  */
1469   SPRINTF_WGINT (buffer, n);
1470   p += strlen (buffer);
1471 #else  /* (SIZEOF_WGINT == 4) || (SIZEOF_WGINT == 8) */
1472
1473   if (n < 0)
1474     {
1475       if (n < -WGINT_MAX)
1476         {
1477           /* -n would overflow.  Have sprintf deal with this.  */
1478           SPRINTF_WGINT (buffer, n);
1479           p += strlen (buffer);
1480           return p;
1481         }
1482
1483       *p++ = '-';
1484       n = -n;
1485     }
1486
1487   /* Use the DIGITS_ macro appropriate for N's number of digits.  That
1488      way printing any N is fully open-coded without a loop or jump.
1489      (Also see description of DIGITS_*.)  */
1490
1491   if      (n < 10)                       DIGITS_1 (1);
1492   else if (n < 100)                      DIGITS_2 (10);
1493   else if (n < 1000)                     DIGITS_3 (100);
1494   else if (n < 10000)                    DIGITS_4 (1000);
1495   else if (n < 100000)                   DIGITS_5 (10000);
1496   else if (n < 1000000)                  DIGITS_6 (100000);
1497   else if (n < 10000000)                 DIGITS_7 (1000000);
1498   else if (n < 100000000)                DIGITS_8 (10000000);
1499   else if (n < 1000000000)               DIGITS_9 (100000000);
1500 #if SIZEOF_WGINT == 4
1501   /* wgint is 32 bits wide: no number has more than 10 digits. */
1502   else                                   DIGITS_10 (1000000000);
1503 #else
1504   /* wgint is 64 bits wide: handle numbers with 9-19 decimal digits.
1505      Constants are constructed by compile-time multiplication to avoid
1506      dealing with different notations for 64-bit constants
1507      (nL/nLL/nI64, depending on the compiler and architecture).  */
1508   else if (n < 10*(W)1000000000)         DIGITS_10 (1000000000);
1509   else if (n < 100*(W)1000000000)        DIGITS_11 (10*(W)1000000000);
1510   else if (n < 1000*(W)1000000000)       DIGITS_12 (100*(W)1000000000);
1511   else if (n < 10000*(W)1000000000)      DIGITS_13 (1000*(W)1000000000);
1512   else if (n < 100000*(W)1000000000)     DIGITS_14 (10000*(W)1000000000);
1513   else if (n < 1000000*(W)1000000000)    DIGITS_15 (100000*(W)1000000000);
1514   else if (n < 10000000*(W)1000000000)   DIGITS_16 (1000000*(W)1000000000);
1515   else if (n < 100000000*(W)1000000000)  DIGITS_17 (10000000*(W)1000000000);
1516   else if (n < 1000000000*(W)1000000000) DIGITS_18 (100000000*(W)1000000000);
1517   else                                   DIGITS_19 (1000000000*(W)1000000000);
1518 #endif
1519
1520   *p = '\0';
1521 #endif /* (SIZEOF_WGINT == 4) || (SIZEOF_WGINT == 8) */
1522
1523   return p;
1524 }
1525
1526 #undef PR
1527 #undef W
1528 #undef SPRINTF_WGINT
1529 #undef DIGITS_1
1530 #undef DIGITS_2
1531 #undef DIGITS_3
1532 #undef DIGITS_4
1533 #undef DIGITS_5
1534 #undef DIGITS_6
1535 #undef DIGITS_7
1536 #undef DIGITS_8
1537 #undef DIGITS_9
1538 #undef DIGITS_10
1539 #undef DIGITS_11
1540 #undef DIGITS_12
1541 #undef DIGITS_13
1542 #undef DIGITS_14
1543 #undef DIGITS_15
1544 #undef DIGITS_16
1545 #undef DIGITS_17
1546 #undef DIGITS_18
1547 #undef DIGITS_19
1548
1549 #define RING_SIZE 3
1550
1551 /* Print NUMBER to a statically allocated string and return a pointer
1552    to the printed representation.
1553
1554    This function is intended to be used in conjunction with printf.
1555    It is hard to portably print wgint values:
1556     a) you cannot use printf("%ld", number) because wgint can be long
1557        long on 32-bit machines with LFS.
1558     b) you cannot use printf("%lld", number) because NUMBER could be
1559        long on 32-bit machines without LFS, or on 64-bit machines,
1560        which do not require LFS.  Also, Windows doesn't support %lld.
1561     c) you cannot use printf("%j", (int_max_t) number) because not all
1562        versions of printf support "%j", the most notable being the one
1563        on Windows.
1564     d) you cannot #define WGINT_FMT to the appropriate format and use
1565        printf(WGINT_FMT, number) because that would break translations
1566        for user-visible messages, such as printf("Downloaded: %d
1567        bytes\n", number).
1568
1569    What you should use instead is printf("%s", number_to_static_string
1570    (number)).
1571
1572    CAVEAT: since the function returns pointers to static data, you
1573    must be careful to copy its result before calling it again.
1574    However, to make it more useful with printf, the function maintains
1575    an internal ring of static buffers to return.  That way things like
1576    printf("%s %s", number_to_static_string (num1),
1577    number_to_static_string (num2)) work as expected.  Three buffers
1578    are currently used, which means that "%s %s %s" will work, but "%s
1579    %s %s %s" won't.  If you need to print more than three wgints,
1580    bump the RING_SIZE (or rethink your message.)  */
1581
1582 char *
1583 number_to_static_string (wgint number)
1584 {
1585   static char ring[RING_SIZE][24];
1586   static int ringpos;
1587   char *buf = ring[ringpos];
1588   number_to_string (buf, number);
1589   ringpos = (ringpos + 1) % RING_SIZE;
1590   return buf;
1591 }
1592 \f
1593 /* Determine the width of the terminal we're running on.  If that's
1594    not possible, return 0.  */
1595
1596 int
1597 determine_screen_width (void)
1598 {
1599   /* If there's a way to get the terminal size using POSIX
1600      tcgetattr(), somebody please tell me.  */
1601 #ifdef TIOCGWINSZ
1602   int fd;
1603   struct winsize wsz;
1604
1605   if (opt.lfilename != NULL)
1606     return 0;
1607
1608   fd = fileno (stderr);
1609   if (ioctl (fd, TIOCGWINSZ, &wsz) < 0)
1610     return 0;                   /* most likely ENOTTY */
1611
1612   return wsz.ws_col;
1613 #elif defined(WINDOWS)
1614   CONSOLE_SCREEN_BUFFER_INFO csbi;
1615   if (!GetConsoleScreenBufferInfo (GetStdHandle (STD_ERROR_HANDLE), &csbi))
1616     return 0;
1617   return csbi.dwSize.X;
1618 #else  /* neither TIOCGWINSZ nor WINDOWS */
1619   return 0;
1620 #endif /* neither TIOCGWINSZ nor WINDOWS */
1621 }
1622 \f
1623 /* Whether the rnd system (either rand or [dl]rand48) has been
1624    seeded.  */
1625 static int rnd_seeded;
1626
1627 /* Return a random number between 0 and MAX-1, inclusive.
1628
1629    If the system does not support lrand48 and MAX is greater than the
1630    value of RAND_MAX+1 on the system, the returned value will be in
1631    the range [0, RAND_MAX].  This may be fixed in a future release.
1632    The random number generator is seeded automatically the first time
1633    it is called.
1634
1635    This uses lrand48 where available, rand elsewhere.  DO NOT use it
1636    for cryptography.  It is only meant to be used in situations where
1637    quality of the random numbers returned doesn't really matter.  */
1638
1639 int
1640 random_number (int max)
1641 {
1642 #ifdef HAVE_DRAND48
1643   if (!rnd_seeded)
1644     {
1645       srand48 ((long) time (NULL) ^ (long) getpid ());
1646       rnd_seeded = 1;
1647     }
1648   return lrand48 () % max;
1649 #else  /* not HAVE_DRAND48 */
1650
1651   double bounded;
1652   int rnd;
1653   if (!rnd_seeded)
1654     {
1655       srand ((unsigned) time (NULL) ^ (unsigned) getpid ());
1656       rnd_seeded = 1;
1657     }
1658   rnd = rand ();
1659
1660   /* Like rand() % max, but uses the high-order bits for better
1661      randomness on architectures where rand() is implemented using a
1662      simple congruential generator.  */
1663
1664   bounded = (double) max * rnd / (RAND_MAX + 1.0);
1665   return (int) bounded;
1666
1667 #endif /* not HAVE_DRAND48 */
1668 }
1669
1670 /* Return a random uniformly distributed floating point number in the
1671    [0, 1) range.  Uses drand48 where available, and a really lame
1672    kludge elsewhere.  */
1673
1674 double
1675 random_float (void)
1676 {
1677 #ifdef HAVE_DRAND48
1678   if (!rnd_seeded)
1679     {
1680       srand48 ((long) time (NULL) ^ (long) getpid ());
1681       rnd_seeded = 1;
1682     }
1683   return drand48 ();
1684 #else  /* not HAVE_DRAND48 */
1685   return (  random_number (10000) / 10000.0
1686           + random_number (10000) / (10000.0 * 10000.0)
1687           + random_number (10000) / (10000.0 * 10000.0 * 10000.0)
1688           + random_number (10000) / (10000.0 * 10000.0 * 10000.0 * 10000.0));
1689 #endif /* not HAVE_DRAND48 */
1690 }
1691 \f
1692 /* Implementation of run_with_timeout, a generic timeout-forcing
1693    routine for systems with Unix-like signal handling.  */
1694
1695 #ifdef USE_SIGNAL_TIMEOUT
1696 # ifdef HAVE_SIGSETJMP
1697 #  define SETJMP(env) sigsetjmp (env, 1)
1698
1699 static sigjmp_buf run_with_timeout_env;
1700
1701 static void
1702 abort_run_with_timeout (int sig)
1703 {
1704   assert (sig == SIGALRM);
1705   siglongjmp (run_with_timeout_env, -1);
1706 }
1707 # else /* not HAVE_SIGSETJMP */
1708 #  define SETJMP(env) setjmp (env)
1709
1710 static jmp_buf run_with_timeout_env;
1711
1712 static void
1713 abort_run_with_timeout (int sig)
1714 {
1715   assert (sig == SIGALRM);
1716   /* We don't have siglongjmp to preserve the set of blocked signals;
1717      if we longjumped out of the handler at this point, SIGALRM would
1718      remain blocked.  We must unblock it manually. */
1719   int mask = siggetmask ();
1720   mask &= ~sigmask (SIGALRM);
1721   sigsetmask (mask);
1722
1723   /* Now it's safe to longjump. */
1724   longjmp (run_with_timeout_env, -1);
1725 }
1726 # endif /* not HAVE_SIGSETJMP */
1727
1728 /* Arrange for SIGALRM to be delivered in TIMEOUT seconds.  This uses
1729    setitimer where available, alarm otherwise.
1730
1731    TIMEOUT should be non-zero.  If the timeout value is so small that
1732    it would be rounded to zero, it is rounded to the least legal value
1733    instead (1us for setitimer, 1s for alarm).  That ensures that
1734    SIGALRM will be delivered in all cases.  */
1735
1736 static void
1737 alarm_set (double timeout)
1738 {
1739 #ifdef ITIMER_REAL
1740   /* Use the modern itimer interface. */
1741   struct itimerval itv;
1742   xzero (itv);
1743   itv.it_value.tv_sec = (long) timeout;
1744   itv.it_value.tv_usec = 1000000 * (timeout - (long)timeout);
1745   if (itv.it_value.tv_sec == 0 && itv.it_value.tv_usec == 0)
1746     /* Ensure that we wait for at least the minimum interval.
1747        Specifying zero would mean "wait forever".  */
1748     itv.it_value.tv_usec = 1;
1749   setitimer (ITIMER_REAL, &itv, NULL);
1750 #else  /* not ITIMER_REAL */
1751   /* Use the old alarm() interface. */
1752   int secs = (int) timeout;
1753   if (secs == 0)
1754     /* Round TIMEOUTs smaller than 1 to 1, not to zero.  This is
1755        because alarm(0) means "never deliver the alarm", i.e. "wait
1756        forever", which is not what someone who specifies a 0.5s
1757        timeout would expect.  */
1758     secs = 1;
1759   alarm (secs);
1760 #endif /* not ITIMER_REAL */
1761 }
1762
1763 /* Cancel the alarm set with alarm_set. */
1764
1765 static void
1766 alarm_cancel (void)
1767 {
1768 #ifdef ITIMER_REAL
1769   struct itimerval disable;
1770   xzero (disable);
1771   setitimer (ITIMER_REAL, &disable, NULL);
1772 #else  /* not ITIMER_REAL */
1773   alarm (0);
1774 #endif /* not ITIMER_REAL */
1775 }
1776
1777 /* Call FUN(ARG), but don't allow it to run for more than TIMEOUT
1778    seconds.  Returns true if the function was interrupted with a
1779    timeout, false otherwise.
1780
1781    This works by setting up SIGALRM to be delivered in TIMEOUT seconds
1782    using setitimer() or alarm().  The timeout is enforced by
1783    longjumping out of the SIGALRM handler.  This has several
1784    advantages compared to the traditional approach of relying on
1785    signals causing system calls to exit with EINTR:
1786
1787      * The callback function is *forcibly* interrupted after the
1788        timeout expires, (almost) regardless of what it was doing and
1789        whether it was in a syscall.  For example, a calculation that
1790        takes a long time is interrupted as reliably as an IO
1791        operation.
1792
1793      * It works with both SYSV and BSD signals because it doesn't
1794        depend on the default setting of SA_RESTART.
1795
1796      * It doesn't require special handler setup beyond a simple call
1797        to signal().  (It does use sigsetjmp/siglongjmp, but they're
1798        optional.)
1799
1800    The only downside is that, if FUN allocates internal resources that
1801    are normally freed prior to exit from the functions, they will be
1802    lost in case of timeout.  */
1803
1804 bool
1805 run_with_timeout (double timeout, void (*fun) (void *), void *arg)
1806 {
1807   int saved_errno;
1808
1809   if (timeout == 0)
1810     {
1811       fun (arg);
1812       return false;
1813     }
1814
1815   signal (SIGALRM, abort_run_with_timeout);
1816   if (SETJMP (run_with_timeout_env) != 0)
1817     {
1818       /* Longjumped out of FUN with a timeout. */
1819       signal (SIGALRM, SIG_DFL);
1820       return true;
1821     }
1822   alarm_set (timeout);
1823   fun (arg);
1824
1825   /* Preserve errno in case alarm() or signal() modifies it. */
1826   saved_errno = errno;
1827   alarm_cancel ();
1828   signal (SIGALRM, SIG_DFL);
1829   errno = saved_errno;
1830
1831   return false;
1832 }
1833
1834 #else  /* not USE_SIGNAL_TIMEOUT */
1835
1836 #ifndef WINDOWS
1837 /* A stub version of run_with_timeout that just calls FUN(ARG).  Don't
1838    define it under Windows, because Windows has its own version of
1839    run_with_timeout that uses threads.  */
1840
1841 int
1842 run_with_timeout (double timeout, void (*fun) (void *), void *arg)
1843 {
1844   fun (arg);
1845   return false;
1846 }
1847 #endif /* not WINDOWS */
1848 #endif /* not USE_SIGNAL_TIMEOUT */
1849 \f
1850 #ifndef WINDOWS
1851
1852 /* Sleep the specified amount of seconds.  On machines without
1853    nanosleep(), this may sleep shorter if interrupted by signals.  */
1854
1855 void
1856 xsleep (double seconds)
1857 {
1858 #ifdef HAVE_NANOSLEEP
1859   /* nanosleep is the preferred interface because it offers high
1860      accuracy and, more importantly, because it allows us to reliably
1861      restart receiving a signal such as SIGWINCH.  (There was an
1862      actual Debian bug report about --limit-rate malfunctioning while
1863      the terminal was being resized.)  */
1864   struct timespec sleep, remaining;
1865   sleep.tv_sec = (long) seconds;
1866   sleep.tv_nsec = 1000000000 * (seconds - (long) seconds);
1867   while (nanosleep (&sleep, &remaining) < 0 && errno == EINTR)
1868     /* If nanosleep has been interrupted by a signal, adjust the
1869        sleeping period and return to sleep.  */
1870     sleep = remaining;
1871 #elif defined(HAVE_USLEEP)
1872   /* If usleep is available, use it in preference to select.  */
1873   if (seconds >= 1)
1874     {
1875       /* On some systems, usleep cannot handle values larger than
1876          1,000,000.  If the period is larger than that, use sleep
1877          first, then add usleep for subsecond accuracy.  */
1878       sleep (seconds);
1879       seconds -= (long) seconds;
1880     }
1881   usleep (seconds * 1000000);
1882 #else /* fall back select */
1883   /* Note that, although Windows supports select, it can't be used to
1884      implement sleeping because Winsock's select doesn't implement
1885      timeout when it is passed NULL pointers for all fd sets.  (But it
1886      does under Cygwin, which implements Unix-compatible select.)  */
1887   struct timeval sleep;
1888   sleep.tv_sec = (long) seconds;
1889   sleep.tv_usec = 1000000 * (seconds - (long) seconds);
1890   select (0, NULL, NULL, NULL, &sleep);
1891   /* If select returns -1 and errno is EINTR, it means we were
1892      interrupted by a signal.  But without knowing how long we've
1893      actually slept, we can't return to sleep.  Using gettimeofday to
1894      track sleeps is slow and unreliable due to clock skew.  */
1895 #endif
1896 }
1897
1898 #endif /* not WINDOWS */
1899
1900 /* Encode the string STR of length LENGTH to base64 format and place it
1901    to B64STORE.  The output will be \0-terminated, and must point to a
1902    writable buffer of at least 1+BASE64_LENGTH(length) bytes.  It
1903    returns the length of the resulting base64 data, not counting the
1904    terminating zero.
1905
1906    This implementation will not emit newlines after 76 characters of
1907    base64 data.  */
1908
1909 int
1910 base64_encode (const char *str, int length, char *b64store)
1911 {
1912   /* Conversion table.  */
1913   static char tbl[64] = {
1914     'A','B','C','D','E','F','G','H',
1915     'I','J','K','L','M','N','O','P',
1916     'Q','R','S','T','U','V','W','X',
1917     'Y','Z','a','b','c','d','e','f',
1918     'g','h','i','j','k','l','m','n',
1919     'o','p','q','r','s','t','u','v',
1920     'w','x','y','z','0','1','2','3',
1921     '4','5','6','7','8','9','+','/'
1922   };
1923   int i;
1924   const unsigned char *s = (const unsigned char *) str;
1925   char *p = b64store;
1926
1927   /* Transform the 3x8 bits to 4x6 bits, as required by base64.  */
1928   for (i = 0; i < length; i += 3)
1929     {
1930       *p++ = tbl[s[0] >> 2];
1931       *p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
1932       *p++ = tbl[((s[1] & 0xf) << 2) + (s[2] >> 6)];
1933       *p++ = tbl[s[2] & 0x3f];
1934       s += 3;
1935     }
1936
1937   /* Pad the result if necessary...  */
1938   if (i == length + 1)
1939     *(p - 1) = '=';
1940   else if (i == length + 2)
1941     *(p - 1) = *(p - 2) = '=';
1942
1943   /* ...and zero-terminate it.  */
1944   *p = '\0';
1945
1946   return p - b64store;
1947 }
1948
1949 /* Store in C the next non-whitespace character from the string, or \0
1950    when end of string is reached.  */
1951 #define NEXT_CHAR(c, p) do {                    \
1952   c = (unsigned char) *p++;                     \
1953 } while (ISSPACE (c))
1954
1955 #define IS_ASCII(c) (((c) & 0x80) == 0)
1956
1957 /* Decode data from BASE64 (pointer to \0-terminated text) into memory
1958    pointed to by TO.  TO should be large enough to accomodate the
1959    decoded data, which is guaranteed to be less than strlen(base64).
1960
1961    Since TO is assumed to contain binary data, it is not
1962    NUL-terminated.  The function returns the length of the data
1963    written to TO.  -1 is returned in case of error caused by malformed
1964    base64 input.  */
1965
1966 int
1967 base64_decode (const char *base64, char *to)
1968 {
1969   /* Table of base64 values for first 128 characters.  Note that this
1970      assumes ASCII (but so does Wget in other places).  */
1971   static signed char base64_char_to_value[128] =
1972     {
1973       -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  /*   0-  9 */
1974       -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  /*  10- 19 */
1975       -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  /*  20- 29 */
1976       -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  -1,  /*  30- 39 */
1977       -1,  -1,  -1,  62,  -1,  -1,  -1,  63,  52,  53,  /*  40- 49 */
1978       54,  55,  56,  57,  58,  59,  60,  61,  -1,  -1,  /*  50- 59 */
1979       -1,  -1,  -1,  -1,  -1,  0,   1,   2,   3,   4,   /*  60- 69 */
1980       5,   6,   7,   8,   9,   10,  11,  12,  13,  14,  /*  70- 79 */
1981       15,  16,  17,  18,  19,  20,  21,  22,  23,  24,  /*  80- 89 */
1982       25,  -1,  -1,  -1,  -1,  -1,  -1,  26,  27,  28,  /*  90- 99 */
1983       29,  30,  31,  32,  33,  34,  35,  36,  37,  38,  /* 100-109 */
1984       39,  40,  41,  42,  43,  44,  45,  46,  47,  48,  /* 110-119 */
1985       49,  50,  51,  -1,  -1,  -1,  -1,  -1             /* 120-127 */
1986     };
1987 #define BASE64_CHAR_TO_VALUE(c) ((int) base64_char_to_value[c])
1988 #define IS_BASE64(c) ((IS_ASCII (c) && BASE64_CHAR_TO_VALUE (c) >= 0) || c == '=')
1989
1990   const char *p = base64;
1991   char *q = to;
1992
1993   while (1)
1994     {
1995       unsigned char c;
1996       unsigned long value;
1997
1998       /* Process first byte of a quadruplet.  */
1999       NEXT_CHAR (c, p);
2000       if (!c)
2001         break;
2002       if (c == '=' || !IS_BASE64 (c))
2003         return -1;              /* illegal char while decoding base64 */
2004       value = BASE64_CHAR_TO_VALUE (c) << 18;
2005
2006       /* Process second byte of a quadruplet.  */
2007       NEXT_CHAR (c, p);
2008       if (!c)
2009         return -1;              /* premature EOF while decoding base64 */
2010       if (c == '=' || !IS_BASE64 (c))
2011         return -1;              /* illegal char while decoding base64 */
2012       value |= BASE64_CHAR_TO_VALUE (c) << 12;
2013       *q++ = value >> 16;
2014
2015       /* Process third byte of a quadruplet.  */
2016       NEXT_CHAR (c, p);
2017       if (!c)
2018         return -1;              /* premature EOF while decoding base64 */
2019       if (!IS_BASE64 (c))
2020         return -1;              /* illegal char while decoding base64 */
2021
2022       if (c == '=')
2023         {
2024           NEXT_CHAR (c, p);
2025           if (!c)
2026             return -1;          /* premature EOF while decoding base64 */
2027           if (c != '=')
2028             return -1;          /* padding `=' expected but not found */
2029           continue;
2030         }
2031
2032       value |= BASE64_CHAR_TO_VALUE (c) << 6;
2033       *q++ = 0xff & value >> 8;
2034
2035       /* Process fourth byte of a quadruplet.  */
2036       NEXT_CHAR (c, p);
2037       if (!c)
2038         return -1;              /* premature EOF while decoding base64 */
2039       if (c == '=')
2040         continue;
2041       if (!IS_BASE64 (c))
2042         return -1;              /* illegal char while decoding base64 */
2043
2044       value |= BASE64_CHAR_TO_VALUE (c);
2045       *q++ = 0xff & value;
2046     }
2047 #undef IS_BASE64
2048 #undef BASE64_CHAR_TO_VALUE
2049
2050   return q - to;
2051 }
2052
2053 #undef IS_ASCII
2054 #undef NEXT_CHAR
2055 \f
2056 /* Simple merge sort for use by stable_sort.  Implementation courtesy
2057    Zeljko Vrba with additional debugging by Nenad Barbutov.  */
2058
2059 static void
2060 mergesort_internal (void *base, void *temp, size_t size, size_t from, size_t to,
2061                     int (*cmpfun) (const void *, const void *))
2062 {
2063 #define ELT(array, pos) ((char *)(array) + (pos) * size)
2064   if (from < to)
2065     {
2066       size_t i, j, k;
2067       size_t mid = (to + from) / 2;
2068       mergesort_internal (base, temp, size, from, mid, cmpfun);
2069       mergesort_internal (base, temp, size, mid + 1, to, cmpfun);
2070       i = from;
2071       j = mid + 1;
2072       for (k = from; (i <= mid) && (j <= to); k++)
2073         if (cmpfun (ELT (base, i), ELT (base, j)) <= 0)
2074           memcpy (ELT (temp, k), ELT (base, i++), size);
2075         else
2076           memcpy (ELT (temp, k), ELT (base, j++), size);
2077       while (i <= mid)
2078         memcpy (ELT (temp, k++), ELT (base, i++), size);
2079       while (j <= to)
2080         memcpy (ELT (temp, k++), ELT (base, j++), size);
2081       for (k = from; k <= to; k++)
2082         memcpy (ELT (base, k), ELT (temp, k), size);
2083     }
2084 #undef ELT
2085 }
2086
2087 /* Stable sort with interface exactly like standard library's qsort.
2088    Uses mergesort internally, allocating temporary storage with
2089    alloca.  */
2090
2091 void
2092 stable_sort (void *base, size_t nmemb, size_t size,
2093              int (*cmpfun) (const void *, const void *))
2094 {
2095   if (size > 1)
2096     {
2097       void *temp = alloca (nmemb * size * sizeof (void *));
2098       mergesort_internal (base, temp, size, 0, nmemb - 1, cmpfun);
2099     }
2100 }
2101 \f
2102 /* Print a decimal number.  If it is equal to or larger than ten, the
2103    number is rounded.  Otherwise it is printed with one significant
2104    digit without trailing zeros and with no more than three fractional
2105    digits total.  For example, 0.1 is printed as "0.1", 0.035 is
2106    printed as "0.04", 0.0091 as "0.009", and 0.0003 as simply "0".
2107
2108    This is useful for displaying durations because it provides
2109    order-of-magnitude information without unnecessary clutter --
2110    long-running downloads are shown without the fractional part, and
2111    short ones still retain one significant digit.  */
2112
2113 const char *
2114 print_decimal (double number)
2115 {
2116   static char buf[32];
2117   double n = number >= 0 ? number : -number;
2118
2119   if (n >= 9.95)
2120     /* Cut off at 9.95 because the below %.1f would round 9.96 to
2121        "10.0" instead of "10".  OTOH 9.94 will print as "9.9".  */
2122     snprintf (buf, sizeof buf, "%.0f", number);
2123   else if (n >= 0.95)
2124     snprintf (buf, sizeof buf, "%.1f", number);
2125   else if (n >= 0.001)
2126     snprintf (buf, sizeof buf, "%.1g", number);
2127   else if (n >= 0.0005)
2128     /* round [0.0005, 0.001) to 0.001 */
2129     snprintf (buf, sizeof buf, "%.3f", number);
2130   else
2131     /* print numbers close to 0 as 0, not 0.000 */
2132     strcpy (buf, "0");
2133
2134   return buf;
2135 }