sjero.net Git - wget/blob - src/init.c

   1 /* Reading/parsing the initialization file.
   2    Copyright (C) 1995, 1996, 1997, 1998, 2000, 2001
   3    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 2 of the License, or
  10 (at 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, write to the Free Software
  19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  20
  21 #include <config.h>
  22
  23 #include <stdio.h>
  24 #include <sys/types.h>
  25 #include <stdlib.h>
  26 #ifdef HAVE_UNISTD_H
  27 # include <unistd.h>
  28 #endif
  29 #ifdef HAVE_STRING_H
  30 # include <string.h>
  31 #else
  32 # include <strings.h>
  33 #endif
  34 #include <errno.h>
  35
  36 #ifdef WINDOWS
  37 # include <winsock.h>
  38 #else
  39 # include <sys/socket.h>
  40 # include <netinet/in.h>
  41 #ifndef __BEOS__
  42 # include <arpa/inet.h>
  43 #endif
  44 #endif
  45
  46 #ifdef HAVE_PWD_H
  47 #include <pwd.h>
  48 #endif
  49
  50 #include "wget.h"
  51 #include "utils.h"
  52 #include "init.h"
  53 #include "host.h"
  54 #include "recur.h"
  55 #include "netrc.h"
  56 #include "cookies.h"            /* for cookies_cleanup */
  57 #include "progress.h"
  58
  59 #ifndef errno
  60 extern int errno;
  61 #endif
  62
  63 /* We want tilde expansion enabled only when reading `.wgetrc' lines;
  64    otherwise, it will be performed by the shell.  This variable will
  65    be set by the wgetrc-reading function.  */
  66
  67 static int enable_tilde_expansion;
  68
  69
  70 #define CMD_DECLARE(func) static int func \
  71   PARAMS ((const char *, const char *, void *))
  72
  73 CMD_DECLARE (cmd_boolean);
  74 CMD_DECLARE (cmd_bytes);
  75 CMD_DECLARE (cmd_directory_vector);
  76 CMD_DECLARE (cmd_lockable_boolean);
  77 CMD_DECLARE (cmd_number);
  78 CMD_DECLARE (cmd_number_inf);
  79 CMD_DECLARE (cmd_string);
  80 CMD_DECLARE (cmd_file);
  81 CMD_DECLARE (cmd_directory);
  82 CMD_DECLARE (cmd_time);
  83 CMD_DECLARE (cmd_vector);
  84
  85 CMD_DECLARE (cmd_spec_dirstruct);
  86 CMD_DECLARE (cmd_spec_header);
  87 CMD_DECLARE (cmd_spec_htmlify);
  88 CMD_DECLARE (cmd_spec_mirror);
  89 CMD_DECLARE (cmd_spec_progress);
  90 CMD_DECLARE (cmd_spec_recursive);
  91 CMD_DECLARE (cmd_spec_useragent);
  92
  93 /* List of recognized commands, each consisting of name, closure and function.
  94    When adding a new command, simply add it to the list, but be sure to keep the
  95    list sorted alphabetically, as comind() depends on it.  Also, be sure to add
  96    any entries that allocate memory (e.g. cmd_string and cmd_vector guys) to the
  97    cleanup() function below. */
  98 static struct {
  99   char *name;
 100   void *closure;
 101   int (*action) PARAMS ((const char *, const char *, void *));
 102 } commands[] = {
 103   { "accept",           &opt.accepts,           cmd_vector },
 104   { "addhostdir",       &opt.add_hostdir,       cmd_boolean },
 105   { "alwaysrest",       &opt.always_rest,       cmd_boolean }, /* deprecated */
 106   { "background",       &opt.background,        cmd_boolean },
 107   { "backupconverted",  &opt.backup_converted,  cmd_boolean },
 108   { "backups",          &opt.backups,           cmd_number },
 109   { "base",             &opt.base_href,         cmd_string },
 110   { "bindaddress",      &opt.bind_address,      cmd_string },
 111   { "cache",            &opt.allow_cache,       cmd_boolean },
 112   { "continue",         &opt.always_rest,       cmd_boolean },
 113   { "convertlinks",     &opt.convert_links,     cmd_boolean },
 114   { "cookies",          &opt.cookies,           cmd_boolean },
 115   { "cutdirs",          &opt.cut_dirs,          cmd_number },
 116 #ifdef DEBUG
 117   { "debug",            &opt.debug,             cmd_boolean },
 118 #endif
 119   { "deleteafter",      &opt.delete_after,      cmd_boolean },
 120   { "dirprefix",        &opt.dir_prefix,        cmd_directory },
 121   { "dirstruct",        NULL,                   cmd_spec_dirstruct },
 122   { "domains",          &opt.domains,           cmd_vector },
 123   { "dotbytes",         &opt.dot_bytes,         cmd_bytes },
 124   { "dotsinline",       &opt.dots_in_line,      cmd_number },
 125   { "dotspacing",       &opt.dot_spacing,       cmd_number },
 126   { "dotstyle",         &opt.dot_style,         cmd_string },
 127   { "excludedirectories", &opt.excludes,        cmd_directory_vector },
 128   { "excludedomains",   &opt.exclude_domains,   cmd_vector },
 129   { "followftp",        &opt.follow_ftp,        cmd_boolean },
 130   { "followtags",       &opt.follow_tags,       cmd_vector },
 131   { "forcehtml",        &opt.force_html,        cmd_boolean },
 132   { "ftpproxy",         &opt.ftp_proxy,         cmd_string },
 133   { "glob",             &opt.ftp_glob,          cmd_boolean },
 134   { "header",           NULL,                   cmd_spec_header },
 135   { "htmlextension",    &opt.html_extension,    cmd_boolean },
 136   { "htmlify",          NULL,                   cmd_spec_htmlify },
 137   { "httpkeepalive",    &opt.http_keep_alive,   cmd_boolean },
 138   { "httppasswd",       &opt.http_passwd,       cmd_string },
 139   { "httpproxy",        &opt.http_proxy,        cmd_string },
 140   { "httpsproxy",       &opt.https_proxy,       cmd_string },
 141   { "httpuser",         &opt.http_user,         cmd_string },
 142   { "ignorelength",     &opt.ignore_length,     cmd_boolean },
 143   { "ignoretags",       &opt.ignore_tags,       cmd_vector },
 144   { "includedirectories", &opt.includes,        cmd_directory_vector },
 145   { "input",            &opt.input_filename,    cmd_file },
 146   { "killlonger",       &opt.kill_longer,       cmd_boolean },
 147   { "limitrate",        &opt.limit_rate,        cmd_bytes },
 148   { "loadcookies",      &opt.cookies_input,     cmd_file },
 149   { "logfile",          &opt.lfilename,         cmd_file },
 150   { "login",            &opt.ftp_acc,           cmd_string },
 151   { "mirror",           NULL,                   cmd_spec_mirror },
 152   { "netrc",            &opt.netrc,             cmd_boolean },
 153   { "noclobber",        &opt.noclobber,         cmd_boolean },
 154   { "noparent",         &opt.no_parent,         cmd_boolean },
 155   { "noproxy",          &opt.no_proxy,          cmd_vector },
 156   { "numtries",         &opt.ntry,              cmd_number_inf },/* deprecated*/
 157   { "outputdocument",   &opt.output_document,   cmd_file },
 158   { "pagerequisites",   &opt.page_requisites,   cmd_boolean },
 159   { "passiveftp",       &opt.ftp_pasv,          cmd_lockable_boolean },
 160   { "passwd",           &opt.ftp_pass,          cmd_string },
 161   { "progress",         &opt.progress_type,     cmd_spec_progress },
 162   { "proxypasswd",      &opt.proxy_passwd,      cmd_string },
 163   { "proxyuser",        &opt.proxy_user,        cmd_string },
 164   { "quiet",            &opt.quiet,             cmd_boolean },
 165   { "quota",            &opt.quota,             cmd_bytes },
 166   { "randomwait",       &opt.random_wait,       cmd_boolean },
 167   { "reclevel",         &opt.reclevel,          cmd_number_inf },
 168   { "recursive",        NULL,                   cmd_spec_recursive },
 169   { "referer",          &opt.referer,           cmd_string },
 170   { "reject",           &opt.rejects,           cmd_vector },
 171   { "relativeonly",     &opt.relative_only,     cmd_boolean },
 172   { "removelisting",    &opt.remove_listing,    cmd_boolean },
 173   { "retrsymlinks",     &opt.retr_symlinks,     cmd_boolean },
 174   { "robots",           &opt.use_robots,        cmd_boolean },
 175   { "savecookies",      &opt.cookies_output,    cmd_file },
 176   { "saveheaders",      &opt.save_headers,      cmd_boolean },
 177   { "serverresponse",   &opt.server_response,   cmd_boolean },
 178   { "spanhosts",        &opt.spanhost,          cmd_boolean },
 179   { "spider",           &opt.spider,            cmd_boolean },
 180 #ifdef HAVE_SSL
 181   { "sslcertfile",      &opt.sslcertfile,       cmd_file },
 182   { "sslcertkey",       &opt.sslcertkey,        cmd_file },
 183   { "egdfile",          &opt.sslegdsock,        cmd_file },
 184 #endif /* HAVE_SSL */
 185   { "timeout",          &opt.timeout,           cmd_time },
 186   { "timestamping",     &opt.timestamping,      cmd_boolean },
 187   { "tries",            &opt.ntry,              cmd_number_inf },
 188   { "useproxy",         &opt.use_proxy,         cmd_boolean },
 189   { "useragent",        NULL,                   cmd_spec_useragent },
 190   { "verbose",          &opt.verbose,           cmd_boolean },
 191   { "wait",             &opt.wait,              cmd_time },
 192   { "waitretry",        &opt.waitretry,         cmd_time }
 193 };
 194
 195 /* Return index of COM if it is a valid command, or -1 otherwise.  COM
 196    is looked up in `commands' using binary search algorithm.  */
 197 static int
 198 comind (const char *com)
 199 {
 200   int min = 0, max = ARRAY_SIZE (commands) - 1;
 201
 202   do
 203     {
 204       int i = (min + max) / 2;
 205       int cmp = strcasecmp (com, commands[i].name);
 206       if (cmp == 0)
 207         return i;
 208       else if (cmp < 0)
 209         max = i - 1;
 210       else
 211         min = i + 1;
 212     }
 213   while (min <= max);
 214   return -1;
 215 }
 216 \f
 217 /* Reset the variables to default values.  */
 218 static void
 219 defaults (void)
 220 {
 221   char *tmp;
 222
 223   /* Most of the default values are 0.  Just reset everything, and
 224      fill in the non-zero values.  Note that initializing pointers to
 225      NULL this way is technically illegal, but porting Wget to a
 226      machine where NULL is not all-zero bit pattern will be the least
 227      of the implementors' worries.  */
 228   memset (&opt, 0, sizeof (opt));
 229
 230   opt.cookies = 1;
 231
 232   opt.verbose = -1;
 233   opt.dir_prefix = xstrdup (".");
 234   opt.ntry = 20;
 235   opt.reclevel = 5;
 236   opt.add_hostdir = 1;
 237   opt.ftp_acc  = xstrdup ("anonymous");
 238   opt.ftp_pass = xstrdup ("-wget@");
 239   opt.netrc = 1;
 240   opt.ftp_glob = 1;
 241   opt.htmlify = 1;
 242   opt.http_keep_alive = 1;
 243   opt.use_proxy = 1;
 244   tmp = getenv ("no_proxy");
 245   if (tmp)
 246     opt.no_proxy = sepstring (tmp);
 247   opt.allow_cache = 1;
 248
 249 #ifdef HAVE_SELECT
 250   opt.timeout = 900;
 251 #endif
 252   opt.use_robots = 1;
 253
 254   opt.remove_listing = 1;
 255
 256   opt.dot_bytes = 1024;
 257   opt.dot_spacing = 10;
 258   opt.dots_in_line = 50;
 259 }
 260 \f
 261 /* Return the user's home directory (strdup-ed), or NULL if none is
 262    found.  */
 263 char *
 264 home_dir (void)
 265 {
 266   char *home = getenv ("HOME");
 267
 268   if (!home)
 269     {
 270 #ifndef WINDOWS
 271       /* If HOME is not defined, try getting it from the password
 272          file.  */
 273       struct passwd *pwd = getpwuid (getuid ());
 274       if (!pwd || !pwd->pw_dir)
 275         return NULL;
 276       home = pwd->pw_dir;
 277 #else  /* WINDOWS */
 278       home = "C:\\";
 279       /* #### Maybe I should grab home_dir from registry, but the best
 280          that I could get from there is user's Start menu.  It sucks!  */
 281 #endif /* WINDOWS */
 282     }
 283
 284   return home ? xstrdup (home) : NULL;
 285 }
 286
 287 /* Return the path to the user's .wgetrc.  This is either the value of
 288    `WGETRC' environment variable, or `$HOME/.wgetrc'.
 289
 290    If the `WGETRC' variable exists but the file does not exist, the
 291    function will exit().  */
 292 static char *
 293 wgetrc_file_name (void)
 294 {
 295   char *env, *home;
 296   char *file = NULL;
 297
 298   /* Try the environment.  */
 299   env = getenv ("WGETRC");
 300   if (env && *env)
 301     {
 302       if (!file_exists_p (env))
 303         {
 304           fprintf (stderr, "%s: %s: %s.\n", exec_name, env, strerror (errno));
 305           exit (1);
 306         }
 307       return xstrdup (env);
 308     }
 309
 310 #ifndef WINDOWS
 311   /* If that failed, try $HOME/.wgetrc.  */
 312   home = home_dir ();
 313   if (home)
 314     {
 315       file = (char *)xmalloc (strlen (home) + 1 + strlen (".wgetrc") + 1);
 316       sprintf (file, "%s/.wgetrc", home);
 317     }
 318   FREE_MAYBE (home);
 319 #else  /* WINDOWS */
 320   /* Under Windows, "home" is (for the purposes of this function) the
 321      directory where `wget.exe' resides, and `wget.ini' will be used
 322      as file name.  SYSTEM_WGETRC should not be defined under WINDOWS.
 323
 324      It is not as trivial as I assumed, because on 95 argv[0] is full
 325      path, but on NT you get what you typed in command line.  --dbudor */
 326   home = ws_mypath ();
 327   if (home)
 328     {
 329       file = (char *)xmalloc (strlen (home) + strlen ("wget.ini") + 1);
 330       sprintf (file, "%swget.ini", home);
 331     }
 332 #endif /* WINDOWS */
 333
 334   if (!file)
 335     return NULL;
 336   if (!file_exists_p (file))
 337     {
 338       xfree (file);
 339       return NULL;
 340     }
 341   return file;
 342 }
 343
 344 /* Initialize variables from a wgetrc file */
 345 static void
 346 run_wgetrc (const char *file)
 347 {
 348   FILE *fp;
 349   char *line;
 350   int ln;
 351
 352   fp = fopen (file, "rb");
 353   if (!fp)
 354     {
 355       fprintf (stderr, _("%s: Cannot read %s (%s).\n"), exec_name,
 356                file, strerror (errno));
 357       return;
 358     }
 359   enable_tilde_expansion = 1;
 360   ln = 1;
 361   while ((line = read_whole_line (fp)))
 362     {
 363       char *com, *val;
 364       int status;
 365
 366       /* Parse the line.  */
 367       status = parse_line (line, &com, &val);
 368       xfree (line);
 369       /* If everything is OK, set the value.  */
 370       if (status == 1)
 371         {
 372           if (!setval (com, val))
 373             fprintf (stderr, _("%s: Error in %s at line %d.\n"), exec_name,
 374                      file, ln);
 375           xfree (com);
 376           xfree (val);
 377         }
 378       else if (status == 0)
 379         fprintf (stderr, _("%s: Error in %s at line %d.\n"), exec_name,
 380                  file, ln);
 381       ++ln;
 382     }
 383   enable_tilde_expansion = 0;
 384   fclose (fp);
 385 }
 386
 387 /* Initialize the defaults and run the system wgetrc and user's own
 388    wgetrc.  */
 389 void
 390 initialize (void)
 391 {
 392   char *file;
 393
 394   /* Load the hard-coded defaults.  */
 395   defaults ();
 396
 397   /* If SYSTEM_WGETRC is defined, use it.  */
 398 #ifdef SYSTEM_WGETRC
 399   if (file_exists_p (SYSTEM_WGETRC))
 400     run_wgetrc (SYSTEM_WGETRC);
 401 #endif
 402   /* Override it with your own, if one exists.  */
 403   file = wgetrc_file_name ();
 404   if (!file)
 405     return;
 406   /* #### We should somehow canonicalize `file' and SYSTEM_WGETRC,
 407      really.  */
 408 #ifdef SYSTEM_WGETRC
 409   if (!strcmp (file, SYSTEM_WGETRC))
 410     {
 411       fprintf (stderr, _("\
 412 %s: Warning: Both system and user wgetrc point to `%s'.\n"),
 413                exec_name, file);
 414     }
 415   else
 416 #endif
 417     run_wgetrc (file);
 418   xfree (file);
 419   return;
 420 }
 421 \f
 422 /* Parse the line pointed by line, with the syntax:
 423    <sp>* command <sp>* = <sp>* value <newline>
 424    Uses malloc to allocate space for command and value.
 425    If the line is invalid, data is freed and 0 is returned.
 426
 427    Return values:
 428     1 - success
 429     0 - failure
 430    -1 - empty */
 431 int
 432 parse_line (const char *line, char **com, char **val)
 433 {
 434   const char *p = line;
 435   const char *orig_comptr, *end;
 436   char *new_comptr;
 437
 438   /* Skip whitespace.  */
 439   while (*p && ISSPACE (*p))
 440     ++p;
 441
 442   /* Don't process empty lines.  */
 443   if (!*p || *p == '#')
 444     return -1;
 445
 446   for (orig_comptr = p; ISALPHA (*p) || *p == '_' || *p == '-'; p++)
 447     ;
 448   /* The next char should be space or '='.  */
 449   if (!ISSPACE (*p) && (*p != '='))
 450     return 0;
 451   /* Here we cannot use strdupdelim() as we normally would because we
 452      want to skip the `-' and `_' characters in the input string.  */
 453   *com = (char *)xmalloc (p - orig_comptr + 1);
 454   for (new_comptr = *com; orig_comptr < p; orig_comptr++)
 455     {
 456       if (*orig_comptr == '_' || *orig_comptr == '-')
 457         continue;
 458       *new_comptr++ = *orig_comptr;
 459     }
 460   *new_comptr = '\0';
 461   /* If the command is invalid, exit now.  */
 462   if (comind (*com) == -1)
 463     {
 464       xfree (*com);
 465       return 0;
 466     }
 467
 468   /* Skip spaces before '='.  */
 469   for (; ISSPACE (*p); p++);
 470   /* If '=' not found, bail out.  */
 471   if (*p != '=')
 472     {
 473       xfree (*com);
 474       return 0;
 475     }
 476   /* Skip spaces after '='.  */
 477   for (++p; ISSPACE (*p); p++);
 478   /* Get the ending position for VAL by starting with the end of the
 479      line and skipping whitespace.  */
 480   end = line + strlen (line) - 1;
 481   while (end > p && ISSPACE (*end))
 482     --end;
 483   *val = strdupdelim (p, end + 1);
 484   return 1;
 485 }
 486
 487 /* Set COM to VAL.  This is the meat behind processing `.wgetrc'.  No
 488    fatals -- error signal prints a warning and resets to default
 489    value.  All error messages are printed to stderr, *not* to
 490    opt.lfile, since opt.lfile wasn't even generated yet.  */
 491 int
 492 setval (const char *com, const char *val)
 493 {
 494   int ind;
 495
 496   if (!com || !val)
 497     return 0;
 498   ind = comind (com);
 499   if (ind == -1)
 500     {
 501       /* #### Should I just abort()?  */
 502 #ifdef DEBUG
 503       fprintf (stderr, _("%s: BUG: unknown command `%s', value `%s'.\n"),
 504                exec_name, com, val);
 505 #endif
 506       return 0;
 507     }
 508   return ((*commands[ind].action) (com, val, commands[ind].closure));
 509 }
 510 \f
 511 /* Generic helper functions, for use with `commands'. */
 512
 513 static int myatoi PARAMS ((const char *s));
 514
 515 /* Store the boolean value from VAL to CLOSURE.  COM is ignored,
 516    except for error messages.  */
 517 static int
 518 cmd_boolean (const char *com, const char *val, void *closure)
 519 {
 520   int bool_value;
 521
 522   if (!strcasecmp (val, "on")
 523       || (*val == '1' && !*(val + 1)))
 524     bool_value = 1;
 525   else if (!strcasecmp (val, "off")
 526            || (*val == '0' && !*(val + 1)))
 527     bool_value = 0;
 528   else
 529     {
 530       fprintf (stderr, _("%s: %s: Please specify on or off.\n"),
 531                exec_name, com);
 532       return 0;
 533     }
 534
 535   *(int *)closure = bool_value;
 536   return 1;
 537 }
 538
 539 /* Store the lockable_boolean {2, 1, 0, -1} value from VAL to CLOSURE.  COM is
 540    ignored, except for error messages.  Values 2 and -1 indicate that once
 541    defined, the value may not be changed by successive wgetrc files or
 542    command-line arguments.
 543
 544    Values: 2 - Enable a particular option for good ("always")
 545            1 - Enable an option ("on")
 546            0 - Disable an option ("off")
 547           -1 - Disable an option for good ("never") */
 548 static int
 549 cmd_lockable_boolean (const char *com, const char *val, void *closure)
 550 {
 551   int lockable_boolean_value;
 552
 553   /*
 554    * If a config file said "always" or "never", don't allow command line
 555    * arguments to override the config file.
 556    */
 557   if (*(int *)closure == -1 || *(int *)closure == 2)
 558     return 1;
 559
 560   if (!strcasecmp (val, "always")
 561       || (*val == '2' && !*(val + 1)))
 562     lockable_boolean_value = 2;
 563   else if (!strcasecmp (val, "on")
 564       || (*val == '1' && !*(val + 1)))
 565     lockable_boolean_value = 1;
 566   else if (!strcasecmp (val, "off")
 567           || (*val == '0' && !*(val + 1)))
 568     lockable_boolean_value = 0;
 569   else if (!strcasecmp (val, "never")
 570       || (*val == '-' && *(val + 1) == '1' && !*(val + 2)))
 571     lockable_boolean_value = -1;
 572   else
 573     {
 574       fprintf (stderr, _("%s: %s: Please specify always, on, off, "
 575                          "or never.\n"),
 576                exec_name, com);
 577       return 0;
 578     }
 579
 580   *(int *)closure = lockable_boolean_value;
 581   return 1;
 582 }
 583
 584 /* Set the non-negative integer value from VAL to CLOSURE.  With
 585    incorrect specification, the number remains unchanged.  */
 586 static int
 587 cmd_number (const char *com, const char *val, void *closure)
 588 {
 589   int num = myatoi (val);
 590
 591   if (num == -1)
 592     {
 593       fprintf (stderr, _("%s: %s: Invalid specification `%s'.\n"),
 594                exec_name, com, val);
 595       return 0;
 596     }
 597   *(int *)closure = num;
 598   return 1;
 599 }
 600
 601 /* Similar to cmd_number(), only accepts `inf' as a synonym for 0.  */
 602 static int
 603 cmd_number_inf (const char *com, const char *val, void *closure)
 604 {
 605   if (!strcasecmp (val, "inf"))
 606     {
 607       *(int *)closure = 0;
 608       return 1;
 609     }
 610   return cmd_number (com, val, closure);
 611 }
 612
 613 /* Copy (strdup) the string at COM to a new location and place a
 614    pointer to *CLOSURE.  */
 615 static int
 616 cmd_string (const char *com, const char *val, void *closure)
 617 {
 618   char **pstring = (char **)closure;
 619
 620   FREE_MAYBE (*pstring);
 621   *pstring = xstrdup (val);
 622   return 1;
 623 }
 624
 625 /* Like the above, but handles tilde-expansion when reading a user's
 626    `.wgetrc'.  In that case, and if VAL begins with `~', the tilde
 627    gets expanded to the user's home directory.  */
 628 static int
 629 cmd_file (const char *com, const char *val, void *closure)
 630 {
 631   char **pstring = (char **)closure;
 632
 633   FREE_MAYBE (*pstring);
 634
 635   /* #### If VAL is empty, perhaps should set *CLOSURE to NULL.  */
 636
 637   if (!enable_tilde_expansion || !(*val == '~' && (*(val + 1) == '/'
 638 #ifdef WINDOWS
 639           || *(val + 1) == '\\'
 640 #endif
 641           )))
 642     {
 643     noexpand:
 644       *pstring = xstrdup (val);
 645     }
 646   else
 647     {
 648       char *result;
 649       int homelen;
 650       char *home = home_dir ();
 651       if (!home)
 652         goto noexpand;
 653
 654       homelen = strlen (home);
 655       while (homelen && (home[homelen - 1] == '/'
 656 #ifdef WINDOWS
 657             || home[homelen - 1] == '\\'
 658 #endif
 659             ))
 660         home[--homelen] = '\0';
 661
 662       /* Skip the leading "~/". */
 663 #ifdef WINDOWS
 664       for (++val; *val == '/' || *val == '\\'; val++)
 665         ;
 666 #else
 667       for (++val; *val == '/'; val++)
 668         ;
 669 #endif
 670
 671       result = xmalloc (homelen + 1 + strlen (val));
 672       memcpy (result, home, homelen);
 673       result[homelen] = '/';
 674       strcpy (result + homelen + 1, val);
 675
 676       *pstring = result;
 677     }
 678 #ifdef WINDOWS
 679   /* Convert "\" to "/". */
 680   {
 681     char *s;
 682     for (s = *pstring; *s; s++)
 683       if (*s == '\\')
 684         *s = '/';
 685   }
 686 #endif
 687   return 1;
 688 }
 689
 690 /* Like cmd_file, but strips trailing '/' characters.  */
 691 static int
 692 cmd_directory (const char *com, const char *val, void *closure)
 693 {
 694   char *s, *t;
 695
 696   /* Call cmd_file() for tilde expansion and separator
 697      canonicalization (backslash -> slash under Windows).  These
 698      things should perhaps be in a separate function.  */
 699   if (!cmd_file (com, val, closure))
 700     return 0;
 701
 702   s = *(char **)closure;
 703   t = s + strlen (s);
 704   while (t > s && *--t == '/')
 705     *t = '\0';
 706
 707   return 1;
 708 }
 709
 710 /* Merge the vector (array of strings separated with `,') in COM with
 711    the vector (NULL-terminated array of strings) pointed to by
 712    CLOSURE.  */
 713 static int
 714 cmd_vector (const char *com, const char *val, void *closure)
 715 {
 716   char ***pvec = (char ***)closure;
 717
 718   if (*val)
 719     *pvec = merge_vecs (*pvec, sepstring (val));
 720   else
 721     {
 722       free_vec (*pvec);
 723       *pvec = NULL;
 724     }
 725   return 1;
 726 }
 727
 728 static int
 729 cmd_directory_vector (const char *com, const char *val, void *closure)
 730 {
 731   char ***pvec = (char ***)closure;
 732
 733   if (*val)
 734     {
 735       /* Strip the trailing slashes from directories.  */
 736       char **t, **seps;
 737
 738       seps = sepstring (val);
 739       for (t = seps; t && *t; t++)
 740         {
 741           int len = strlen (*t);
 742           /* Skip degenerate case of root directory.  */
 743           if (len > 1)
 744             {
 745               if ((*t)[len - 1] == '/')
 746                 (*t)[len - 1] = '\0';
 747             }
 748         }
 749       *pvec = merge_vecs (*pvec, seps);
 750     }
 751   else
 752     {
 753       free_vec (*pvec);
 754       *pvec = NULL;
 755     }
 756   return 1;
 757 }
 758
 759 /* Set the value stored in VAL to CLOSURE (which should point to a
 760    long int), allowing several postfixes, with the following syntax
 761    (regexp):
 762
 763    [0-9]+       -> bytes
 764    [0-9]+[kK]   -> bytes * 1024
 765    [0-9]+[mM]   -> bytes * 1024 * 1024
 766    inf          -> 0
 767
 768    Anything else is flagged as incorrect, and CLOSURE is unchanged.  */
 769 static int
 770 cmd_bytes (const char *com, const char *val, void *closure)
 771 {
 772   long result;
 773   long *out = (long *)closure;
 774   const char *p;
 775
 776   result = 0;
 777   p = val;
 778   /* Check for "inf".  */
 779   if (p[0] == 'i' && p[1] == 'n' && p[2] == 'f' && p[3] == '\0')
 780     {
 781       *out = 0;
 782       return 1;
 783     }
 784   /* Search for digits and construct result.  */
 785   for (; *p && ISDIGIT (*p); p++)
 786     result = (10 * result) + (*p - '0');
 787   /* If no digits were found, or more than one character is following
 788      them, bail out.  */
 789   if (p == val || (*p != '\0' && *(p + 1) != '\0'))
 790     {
 791       printf (_("%s: Invalid specification `%s'\n"), com, val);
 792       return 0;
 793     }
 794   /* Search for a designator.  */
 795   switch (TOLOWER (*p))
 796     {
 797     case '\0':
 798       /* None */
 799       break;
 800     case 'k':
 801       /* Kilobytes */
 802       result *= 1024;
 803       break;
 804     case 'm':
 805       /* Megabytes */
 806       result *= (long)1024 * 1024;
 807       break;
 808     case 'g':
 809       /* Gigabytes */
 810       result *= (long)1024 * 1024 * 1024;
 811       break;
 812     default:
 813       printf (_("%s: Invalid specification `%s'\n"), com, val);
 814       return 0;
 815     }
 816   *out = result;
 817   return 1;
 818 }
 819
 820 /* Store the value of VAL to *OUT, allowing suffixes for minutes and
 821    hours.  */
 822 static int
 823 cmd_time (const char *com, const char *val, void *closure)
 824 {
 825   long result = 0;
 826   const char *p = val;
 827
 828   /* Search for digits and construct result.  */
 829   for (; *p && ISDIGIT (*p); p++)
 830     result = (10 * result) + (*p - '0');
 831   /* If no digits were found, or more than one character is following
 832      them, bail out.  */
 833   if (p == val || (*p != '\0' && *(p + 1) != '\0'))
 834     {
 835       printf (_("%s: Invalid specification `%s'\n"), com, val);
 836       return 0;
 837     }
 838   /* Search for a suffix.  */
 839   switch (TOLOWER (*p))
 840     {
 841     case '\0':
 842       /* None */
 843       break;
 844     case 'm':
 845       /* Minutes */
 846       result *= 60;
 847       break;
 848     case 'h':
 849       /* Seconds */
 850       result *= 3600;
 851       break;
 852     case 'd':
 853       /* Days (overflow on 16bit machines) */
 854       result *= 86400L;
 855       break;
 856     case 'w':
 857       /* Weeks :-) */
 858       result *= 604800L;
 859       break;
 860     default:
 861       printf (_("%s: Invalid specification `%s'\n"), com, val);
 862       return 0;
 863     }
 864   *(long *)closure = result;
 865   return 1;
 866 }
 867 \f
 868 /* Specialized helper functions, used by `commands' to handle some
 869    options specially.  */
 870
 871 static int check_user_specified_header PARAMS ((const char *));
 872
 873 static int
 874 cmd_spec_dirstruct (const char *com, const char *val, void *closure)
 875 {
 876   if (!cmd_boolean (com, val, &opt.dirstruct))
 877     return 0;
 878   /* Since dirstruct behaviour is explicitly changed, no_dirstruct
 879      must be affected inversely.  */
 880   if (opt.dirstruct)
 881     opt.no_dirstruct = 0;
 882   else
 883     opt.no_dirstruct = 1;
 884   return 1;
 885 }
 886
 887 static int
 888 cmd_spec_header (const char *com, const char *val, void *closure)
 889 {
 890   if (!*val)
 891     {
 892       /* Empty header means reset headers.  */
 893       FREE_MAYBE (opt.user_header);
 894       opt.user_header = NULL;
 895     }
 896   else
 897     {
 898       int i;
 899
 900       if (!check_user_specified_header (val))
 901         {
 902           fprintf (stderr, _("%s: %s: Invalid specification `%s'.\n"),
 903                    exec_name, com, val);
 904           return 0;
 905         }
 906       i = opt.user_header ? strlen (opt.user_header) : 0;
 907       opt.user_header = (char *)xrealloc (opt.user_header, i + strlen (val)
 908                                           + 2 + 1);
 909       strcpy (opt.user_header + i, val);
 910       i += strlen (val);
 911       opt.user_header[i++] = '\r';
 912       opt.user_header[i++] = '\n';
 913       opt.user_header[i] = '\0';
 914     }
 915   return 1;
 916 }
 917
 918 static int
 919 cmd_spec_htmlify (const char *com, const char *val, void *closure)
 920 {
 921   int flag = cmd_boolean (com, val, &opt.htmlify);
 922   if (flag && !opt.htmlify)
 923     opt.remove_listing = 0;
 924   return flag;
 925 }
 926
 927 static int
 928 cmd_spec_mirror (const char *com, const char *val, void *closure)
 929 {
 930   int mirror;
 931
 932   if (!cmd_boolean (com, val, &mirror))
 933     return 0;
 934   if (mirror)
 935     {
 936       opt.recursive = 1;
 937       if (!opt.no_dirstruct)
 938         opt.dirstruct = 1;
 939       opt.timestamping = 1;
 940       opt.reclevel = INFINITE_RECURSION;
 941       opt.remove_listing = 0;
 942     }
 943   return 1;
 944 }
 945
 946 static int
 947 cmd_spec_progress (const char *com, const char *val, void *closure)
 948 {
 949   if (!valid_progress_implementation_p (val))
 950     {
 951       fprintf (stderr, _("%s: %s: Invalid progress type `%s'.\n"),
 952                exec_name, com, val);
 953       return 0;
 954     }
 955   FREE_MAYBE (opt.progress_type);
 956
 957   /* Don't call set_progress_implementation here.  It will be called
 958      in main() when it becomes clear what the log output is.  */
 959   opt.progress_type = xstrdup (val);
 960   return 1;
 961 }
 962
 963 static int
 964 cmd_spec_recursive (const char *com, const char *val, void *closure)
 965 {
 966   if (!cmd_boolean (com, val, &opt.recursive))
 967     return 0;
 968   else
 969     {
 970       if (opt.recursive && !opt.no_dirstruct)
 971         opt.dirstruct = 1;
 972     }
 973   return 1;
 974 }
 975
 976 static int
 977 cmd_spec_useragent (const char *com, const char *val, void *closure)
 978 {
 979   /* Just check for empty string and newline, so we don't throw total
 980      junk to the server.  */
 981   if (!*val || strchr (val, '\n'))
 982     {
 983       fprintf (stderr, _("%s: %s: Invalid specification `%s'.\n"),
 984                exec_name, com, val);
 985       return 0;
 986     }
 987   opt.useragent = xstrdup (val);
 988   return 1;
 989 }
 990 \f
 991 /* Miscellaneous useful routines.  */
 992
 993 /* Return the integer value of a positive integer written in S, or -1
 994    if an error was encountered.  */
 995 static int
 996 myatoi (const char *s)
 997 {
 998   int res;
 999   const char *orig = s;
1000
1001   for (res = 0; *s && ISDIGIT (*s); s++)
1002     res = 10 * res + (*s - '0');
1003   if (*s || orig == s)
1004     return -1;
1005   else
1006     return res;
1007 }
1008
1009 #define ISODIGIT(x) ((x) >= '0' && (x) <= '7')
1010
1011 static int
1012 check_user_specified_header (const char *s)
1013 {
1014   const char *p;
1015
1016   for (p = s; *p && *p != ':' && !ISSPACE (*p); p++);
1017   /* The header MUST contain `:' preceded by at least one
1018      non-whitespace character.  */
1019   if (*p != ':' || p == s)
1020     return 0;
1021   /* The header MUST NOT contain newlines.  */
1022   if (strchr (s, '\n'))
1023     return 0;
1024   return 1;
1025 }
1026 \f
1027 void cleanup_html_url PARAMS ((void));
1028 void res_cleanup PARAMS ((void));
1029 void downloaded_files_free PARAMS ((void));
1030 void http_cleanup PARAMS ((void));
1031
1032
1033 /* Free the memory allocated by global variables.  */
1034 void
1035 cleanup (void)
1036 {
1037   /* Free external resources, close files, etc. */
1038
1039   if (opt.dfp)
1040     fclose (opt.dfp);
1041
1042   /* We're exiting anyway so there's no real need to call free()
1043      hundreds of times.  Skipping the frees will make Wget exit
1044      faster.
1045
1046      However, when detecting leaks, it's crucial to free() everything
1047      because then you can find the real leaks, i.e. the allocated
1048      memory which grows with the size of the program.  */
1049
1050 #ifdef DEBUG_MALLOC
1051   recursive_cleanup ();
1052   res_cleanup ();
1053   http_cleanup ();
1054   cleanup_html_url ();
1055   downloaded_files_free ();
1056   cookies_cleanup ();
1057   host_cleanup ();
1058
1059   {
1060     extern acc_t *netrc_list;
1061     free_netrc (netrc_list);
1062   }
1063   FREE_MAYBE (opt.lfilename);
1064   xfree (opt.dir_prefix);
1065   FREE_MAYBE (opt.input_filename);
1066   FREE_MAYBE (opt.output_document);
1067   free_vec (opt.accepts);
1068   free_vec (opt.rejects);
1069   free_vec (opt.excludes);
1070   free_vec (opt.includes);
1071   free_vec (opt.domains);
1072   free_vec (opt.follow_tags);
1073   free_vec (opt.ignore_tags);
1074   FREE_MAYBE (opt.progress_type);
1075   xfree (opt.ftp_acc);
1076   FREE_MAYBE (opt.ftp_pass);
1077   FREE_MAYBE (opt.ftp_proxy);
1078   FREE_MAYBE (opt.https_proxy);
1079   FREE_MAYBE (opt.http_proxy);
1080   free_vec (opt.no_proxy);
1081   FREE_MAYBE (opt.useragent);
1082   FREE_MAYBE (opt.referer);
1083   FREE_MAYBE (opt.http_user);
1084   FREE_MAYBE (opt.http_passwd);
1085   FREE_MAYBE (opt.user_header);
1086 #ifdef HAVE_SSL
1087   FREE_MAYBE (opt.sslcertkey);
1088   FREE_MAYBE (opt.sslcertfile);
1089 #endif /* HAVE_SSL */
1090   FREE_MAYBE (opt.bind_address);
1091   FREE_MAYBE (opt.cookies_input);
1092   FREE_MAYBE (opt.cookies_output);
1093 #endif
1094 }