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

   1 /* Command line parsing.
   2    Copyright (C) 1995, 1996, 1997, 1998, 2000 Free Software Foundation, Inc.
   3
   4 This file is part of Wget.
   5
   6 This program 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 This program 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 this program; if not, write to the Free Software
  18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  19
  20 #include <config.h>
  21
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #ifdef HAVE_UNISTD_H
  25 # include <unistd.h>
  26 #endif /* HAVE_UNISTD_H */
  27 #include <sys/types.h>
  28 #ifdef HAVE_STRING_H
  29 # include <string.h>
  30 #else
  31 # include <strings.h>
  32 #endif /* HAVE_STRING_H */
  33 #ifdef HAVE_SIGNAL_H
  34 # include <signal.h>
  35 #endif
  36 #ifdef HAVE_NLS
  37 #ifdef HAVE_LOCALE_H
  38 # include <locale.h>
  39 #endif /* HAVE_LOCALE_H */
  40 #endif /* HAVE_NLS */
  41 #include <errno.h>
  42
  43 #define OPTIONS_DEFINED_HERE    /* for options.h */
  44
  45 #include "wget.h"
  46 #include "utils.h"
  47 #include "getopt.h"
  48 #include "init.h"
  49 #include "retr.h"
  50 #include "recur.h"
  51 #include "host.h"
  52 #include "cookies.h"
  53
  54 #ifndef PATH_SEPARATOR
  55 # define PATH_SEPARATOR '/'
  56 #endif
  57
  58 extern char *version_string;
  59
  60 #ifndef errno
  61 extern int errno;
  62 #endif
  63
  64 struct options opt;
  65
  66 /* From log.c.  */
  67 void log_init PARAMS ((const char *, int));
  68 void log_close PARAMS ((void));
  69 void redirect_output PARAMS ((const char *));
  70
  71 static RETSIGTYPE redirect_output_signal PARAMS ((int));
  72
  73 const char *exec_name;
  74 \f
  75 /* Initialize I18N.  The initialization amounts to invoking
  76    setlocale(), bindtextdomain() and textdomain().
  77    Does nothing if NLS is disabled or missing.  */
  78 static void
  79 i18n_initialize (void)
  80 {
  81   /* If HAVE_NLS is defined, assume the existence of the three
  82      functions invoked here.  */
  83 #ifdef HAVE_NLS
  84   /* Set the current locale.  */
  85   /* Here we use LC_MESSAGES instead of LC_ALL, for two reasons.
  86      First, message catalogs are all of I18N Wget uses anyway.
  87      Second, setting LC_ALL has a dangerous potential of messing
  88      things up.  For example, when in a foreign locale, Solaris
  89      strptime() fails to handle international dates correctly, which
  90      makes http_atotm() malfunction.  */
  91 #ifdef LC_MESSAGES
  92   setlocale (LC_MESSAGES, "");
  93   setlocale (LC_CTYPE, "");
  94 #else
  95   setlocale (LC_ALL, "");
  96 #endif
  97   /* Set the text message domain.  */
  98   bindtextdomain ("wget", LOCALEDIR);
  99   textdomain ("wget");
 100 #endif /* HAVE_NLS */
 101 }
 102
 103 /* It's kosher to declare these here because their interface _has_ to
 104    be void foo(void).  */
 105 void url_init PARAMS ((void));
 106 void host_init PARAMS ((void));
 107
 108 /* This just calls the various initialization functions from the
 109    modules that need one-time initialization. */
 110 static void
 111 private_initialize (void)
 112 {
 113   url_init ();
 114   host_init ();
 115 }
 116 \f
 117 /* Print the usage message.  */
 118 static void
 119 print_usage (void)
 120 {
 121   printf (_("Usage: %s [OPTION]... [URL]...\n"), exec_name);
 122 }
 123
 124 /* Print the help message, describing all the available options.  If
 125    you add an option, be sure to update this list.  */
 126 static void
 127 print_help (void)
 128 {
 129   printf (_("GNU Wget %s, a non-interactive network retriever.\n"),
 130           version_string);
 131   print_usage ();
 132   /* Had to split this in parts, so the #@@#%# Ultrix compiler and cpp
 133      don't bitch.  Also, it makes translation much easier.  */
 134   fputs (_("\
 135 \n\
 136 Mandatory arguments to long options are mandatory for short options too.\n\
 137 \n"), stdout);
 138   fputs (_("\
 139 Startup:\n\
 140   -V,  --version           display the version of Wget and exit.\n\
 141   -h,  --help              print this help.\n\
 142   -b,  --background        go to background after startup.\n\
 143   -e,  --execute=COMMAND   execute a `.wgetrc\'-style command.\n\
 144 \n"), stdout);
 145   fputs (_("\
 146 Logging and input file:\n\
 147   -o,  --output-file=FILE     log messages to FILE.\n\
 148   -a,  --append-output=FILE   append messages to FILE.\n\
 149   -d,  --debug                print debug output.\n\
 150   -q,  --quiet                quiet (no output).\n\
 151   -v,  --verbose              be verbose (this is the default).\n\
 152   -nv, --non-verbose          turn off verboseness, without being quiet.\n\
 153   -i,  --input-file=FILE      download URLs found in FILE.\n\
 154   -F,  --force-html           treat input file as HTML.\n\
 155   -B,  --base=URL             prepends URL to relative links in -F -i file.\n\
 156        --sslcertfile=FILE     optional client certificate.\n\
 157        --sslcertkey=KEYFILE   optional keyfile for this certificate.\n\
 158 \n"), stdout);
 159   fputs (_("\
 160 Download:\n\
 161        --bind-address=ADDRESS   bind to ADDRESS (hostname or IP) on local host.\n\
 162   -t,  --tries=NUMBER           set number of retries to NUMBER (0 unlimits).\n\
 163   -O   --output-document=FILE   write documents to FILE.\n\
 164   -nc, --no-clobber             don\'t clobber existing files or use .# suffixes.\n\
 165   -c,  --continue               resume getting a partially-downloaded file.\n\
 166        --dot-style=STYLE        set retrieval display style.\n\
 167   -N,  --timestamping           don\'t re-retrieve files unless newer than local.\n\
 168   -S,  --server-response        print server response.\n\
 169        --spider                 don\'t download anything.\n\
 170   -T,  --timeout=SECONDS        set the read timeout to SECONDS.\n\
 171   -w,  --wait=SECONDS           wait SECONDS between retrievals.\n\
 172        --waitretry=SECONDS      wait 1...SECONDS between retries of a retrieval.\n\
 173   -Y,  --proxy=on/off           turn proxy on or off.\n\
 174   -Q,  --quota=NUMBER           set retrieval quota to NUMBER.\n\
 175 \n"), stdout);
 176   fputs (_("\
 177 Directories:\n\
 178   -nd  --no-directories            don\'t create directories.\n\
 179   -x,  --force-directories         force creation of directories.\n\
 180   -nH, --no-host-directories       don\'t create host directories.\n\
 181   -P,  --directory-prefix=PREFIX   save files to PREFIX/...\n\
 182        --cut-dirs=NUMBER           ignore NUMBER remote directory components.\n\
 183 \n"), stdout);
 184   fputs (_("\
 185 HTTP options:\n\
 186        --http-user=USER      set http user to USER.\n\
 187        --http-passwd=PASS    set http password to PASS.\n\
 188   -C,  --cache=on/off        (dis)allow server-cached data (normally allowed).\n\
 189   -E,  --html-extension      save all text/html documents with .html extension.\n\
 190        --ignore-length       ignore `Content-Length\' header field.\n\
 191        --header=STRING       insert STRING among the headers.\n\
 192        --proxy-user=USER     set USER as proxy username.\n\
 193        --proxy-passwd=PASS   set PASS as proxy password.\n\
 194        --referer=URL         include `Referer: URL\' header in HTTP request.\n\
 195   -s,  --save-headers        save the HTTP headers to file.\n\
 196   -U,  --user-agent=AGENT    identify as AGENT instead of Wget/VERSION.\n\
 197        --cookies=no          don't use cookies.\n\
 198        --load-cookies=FILE   load cookies from FILE before session.\n\
 199        --save-cookies=FILE   save cookies to FILE after session.\n\
 200 \n"), stdout);
 201   fputs (_("\
 202 FTP options:\n\
 203   -nr, --dont-remove-listing   don\'t remove `.listing\' files.\n\
 204   -g,  --glob=on/off           turn file name globbing on or off.\n\
 205        --passive-ftp           use the \"passive\" transfer mode.\n\
 206        --retr-symlinks         when recursing, get linked-to files (not dirs).\n\
 207 \n"), stdout);
 208   fputs (_("\
 209 Recursive retrieval:\n\
 210   -r,  --recursive          recursive web-suck -- use with care!\n\
 211   -l,  --level=NUMBER       maximum recursion depth (inf or 0 for infinite).\n\
 212        --delete-after       delete files locally after downloading them.\n\
 213   -k,  --convert-links      convert non-relative links to relative.\n\
 214   -K,  --backup-converted   before converting file X, back up as X.orig.\n\
 215   -m,  --mirror             shortcut option equivalent to -r -N -l inf -nr.\n\
 216   -p,  --page-requisites    get all images, etc. needed to display HTML page.\n\
 217 \n"), stdout);
 218   fputs (_("\
 219 Recursive accept/reject:\n\
 220   -A,  --accept=LIST                comma-separated list of accepted extensions.\n\
 221   -R,  --reject=LIST                comma-separated list of rejected extensions.\n\
 222   -D,  --domains=LIST               comma-separated list of accepted domains.\n\
 223        --exclude-domains=LIST       comma-separated list of rejected domains.\n\
 224        --follow-ftp                 follow FTP links from HTML documents.\n\
 225        --follow-tags=LIST           comma-separated list of followed HTML tags.\n\
 226   -G,  --ignore-tags=LIST           comma-separated list of ignored HTML tags.\n\
 227   -H,  --span-hosts                 go to foreign hosts when recursive.\n\
 228   -L,  --relative                   follow relative links only.\n\
 229   -I,  --include-directories=LIST   list of allowed directories.\n\
 230   -X,  --exclude-directories=LIST   list of excluded directories.\n\
 231   -nh, --no-host-lookup             don\'t DNS-lookup hosts.\n\
 232   -np, --no-parent                  don\'t ascend to the parent directory.\n\
 233 \n"), stdout);
 234   fputs (_("Mail bug reports and suggestions to <bug-wget@gnu.org>.\n"),
 235          stdout);
 236 }
 237 \f
 238 int
 239 main (int argc, char *const *argv)
 240 {
 241   char **url, **t;
 242   int i, c, nurl, status, append_to_log;
 243   int wr = 0;
 244
 245   static struct option long_options[] =
 246   {
 247     /* Options without arguments: */
 248     { "background", no_argument, NULL, 'b' },
 249     { "backup-converted", no_argument, NULL, 'K' },
 250     { "continue", no_argument, NULL, 'c' },
 251     { "convert-links", no_argument, NULL, 'k' },
 252     { "cookies", no_argument, NULL, 160 },
 253     { "debug", no_argument, NULL, 'd' },
 254     { "delete-after", no_argument, NULL, 136 },
 255     { "dont-remove-listing", no_argument, NULL, 149 },
 256     { "email-address", no_argument, NULL, 154 }, /* undocumented (debug) */
 257     { "follow-ftp", no_argument, NULL, 142 },
 258     { "force-directories", no_argument, NULL, 'x' },
 259     { "force-hier", no_argument, NULL, 'x' }, /* obsolete */
 260     { "force-html", no_argument, NULL, 'F'},
 261     { "help", no_argument, NULL, 'h' },
 262     { "html-extension", no_argument, NULL, 'E' },
 263     { "ignore-length", no_argument, NULL, 138 },
 264     { "mirror", no_argument, NULL, 'm' },
 265     { "no-clobber", no_argument, NULL, 141 },
 266     { "no-directories", no_argument, NULL, 147 },
 267     { "no-host-directories", no_argument, NULL, 148 },
 268     { "no-host-lookup", no_argument, NULL, 150 },
 269     { "no-http-keep-alive", no_argument, NULL, 156 },
 270     { "no-parent", no_argument, NULL, 133 },
 271     { "non-verbose", no_argument, NULL, 146 },
 272     { "passive-ftp", no_argument, NULL, 139 },
 273     { "page-requisites", no_argument, NULL, 'p' },
 274     { "quiet", no_argument, NULL, 'q' },
 275     { "recursive", no_argument, NULL, 'r' },
 276     { "relative", no_argument, NULL, 'L' },
 277     { "retr-symlinks", no_argument, NULL, 137 },
 278     { "save-headers", no_argument, NULL, 's' },
 279     { "server-response", no_argument, NULL, 'S' },
 280     { "span-hosts", no_argument, NULL, 'H' },
 281     { "spider", no_argument, NULL, 132 },
 282     { "timestamping", no_argument, NULL, 'N' },
 283     { "verbose", no_argument, NULL, 'v' },
 284     { "version", no_argument, NULL, 'V' },
 285
 286     /* Options accepting an argument: */
 287     { "accept", required_argument, NULL, 'A' },
 288     { "append-output", required_argument, NULL, 'a' },
 289     { "backups", required_argument, NULL, 151 }, /* undocumented */
 290     { "base", required_argument, NULL, 'B' },
 291     { "bind-address", required_argument, NULL, 155 },
 292     { "cache", required_argument, NULL, 'C' },
 293     { "cookie-file", required_argument, NULL, 161 },
 294     { "cut-dirs", required_argument, NULL, 145 },
 295     { "directory-prefix", required_argument, NULL, 'P' },
 296     { "domains", required_argument, NULL, 'D' },
 297     { "dot-style", required_argument, NULL, 134 },
 298     { "execute", required_argument, NULL, 'e' },
 299     { "exclude-directories", required_argument, NULL, 'X' },
 300     { "exclude-domains", required_argument, NULL, 140 },
 301     { "follow-tags", required_argument, NULL, 153 },
 302     { "glob", required_argument, NULL, 'g' },
 303     { "header", required_argument, NULL, 131 },
 304     { "htmlify", required_argument, NULL, 135 },
 305     { "http-passwd", required_argument, NULL, 130 },
 306     { "http-user", required_argument, NULL, 129 },
 307     { "ignore-tags", required_argument, NULL, 'G' },
 308     { "include-directories", required_argument, NULL, 'I' },
 309     { "input-file", required_argument, NULL, 'i' },
 310     { "level", required_argument, NULL, 'l' },
 311     { "load-cookies", required_argument, NULL, 162 },
 312     { "no", required_argument, NULL, 'n' },
 313     { "output-document", required_argument, NULL, 'O' },
 314     { "output-file", required_argument, NULL, 'o' },
 315     { "proxy", required_argument, NULL, 'Y' },
 316     { "proxy-passwd", required_argument, NULL, 144 },
 317     { "proxy-user", required_argument, NULL, 143 },
 318     { "quota", required_argument, NULL, 'Q' },
 319     { "reject", required_argument, NULL, 'R' },
 320     { "save-cookies", required_argument, NULL, 163 },
 321     { "timeout", required_argument, NULL, 'T' },
 322     { "tries", required_argument, NULL, 't' },
 323     { "user-agent", required_argument, NULL, 'U' },
 324     { "referer", required_argument, NULL, 157 },
 325     { "use-proxy", required_argument, NULL, 'Y' },
 326 #ifdef HAVE_SSL
 327     { "sslcertfile", required_argument, NULL, 158 },
 328     { "sslcertkey", required_argument, NULL, 159 },
 329 #endif /* HAVE_SSL */
 330     { "wait", required_argument, NULL, 'w' },
 331     { "waitretry", required_argument, NULL, 152 },
 332     { 0, 0, 0, 0 }
 333   };
 334
 335   i18n_initialize ();
 336   private_initialize ();
 337
 338   append_to_log = 0;
 339
 340   /* Construct the name of the executable, without the directory part.  */
 341   exec_name = strrchr (argv[0], PATH_SEPARATOR);
 342   if (!exec_name)
 343     exec_name = argv[0];
 344   else
 345     ++exec_name;
 346
 347 #ifdef WINDOWS
 348   windows_main_junk (&argc, (char **) argv, (char **) &exec_name);
 349 #endif
 350
 351   initialize (); /* sets option defaults; reads the system wgetrc and .wgetrc */
 352
 353   /* [Is the order of the option letters significant?  If not, they should be
 354       alphabetized, like the long_options.  The only thing I know for sure is
 355       that the options with required arguments must be followed by a ':'.
 356       -- Dan Harkless <wget@harkless.org>] */
 357   while ((c = getopt_long (argc, argv, "\
 358 hpVqvdkKsxmNWrHSLcFbEY:G:g:T:U:O:l:n:i:o:a:t:D:A:R:P:B:e:Q:X:I:w:C:",
 359                            long_options, (int *)0)) != EOF)
 360     {
 361       switch (c)
 362         {
 363           /* Options without arguments: */
 364         case 132:
 365           setval ("spider", "on");
 366           break;
 367         case 133:
 368           setval ("noparent", "on");
 369           break;
 370         case 136:
 371           setval ("deleteafter", "on");
 372           break;
 373         case 137:
 374           setval ("retrsymlinks", "on");
 375           break;
 376         case 138:
 377           setval ("ignorelength", "on");
 378           break;
 379         case 139:
 380           setval ("passiveftp", "on");
 381           break;
 382         case 141:
 383           setval ("noclobber", "on");
 384           break;
 385         case 142:
 386           setval ("followftp", "on");
 387           break;
 388         case 145:
 389           setval ("cutdirs", optarg);
 390           break;
 391         case 146:
 392           setval ("verbose", "off");
 393           break;
 394         case 147:
 395           setval ("dirstruct", "off");
 396           break;
 397         case 148:
 398           setval ("addhostdir", "off");
 399           break;
 400         case 149:
 401           setval ("removelisting", "off");
 402           break;
 403         case 150:
 404           setval ("simplehostcheck", "on");
 405           break;
 406         case 154:
 407           /* For debugging purposes.  */
 408           printf ("%s\n", ftp_getaddress ());
 409           exit (0);
 410           break;
 411         case 155:
 412           setval ("bindaddress", optarg);
 413           break;
 414         case 156:
 415           setval ("httpkeepalive", "off");
 416           break;
 417         case 'b':
 418           setval ("background", "on");
 419           break;
 420         case 'c':
 421           setval ("continue", "on");
 422           break;
 423         case 'd':
 424 #ifdef DEBUG
 425           setval ("debug", "on");
 426 #else  /* not DEBUG */
 427           fprintf (stderr, _("%s: debug support not compiled in.\n"),
 428                    exec_name);
 429 #endif /* not DEBUG */
 430           break;
 431         case 'E':
 432           setval ("htmlextension", "on");
 433           break;
 434         case 'F':
 435           setval ("forcehtml", "on");
 436           break;
 437         case 'H':
 438           setval ("spanhosts", "on");
 439           break;
 440         case 'h':
 441           print_help ();
 442 #ifdef WINDOWS
 443           ws_help (exec_name);
 444 #endif
 445           exit (0);
 446           break;
 447         case 'K':
 448           setval ("backupconverted", "on");
 449           break;
 450         case 'k':
 451           setval ("convertlinks", "on");
 452           break;
 453         case 'L':
 454           setval ("relativeonly", "on");
 455           break;
 456         case 'm':
 457           setval ("mirror", "on");
 458           break;
 459         case 'N':
 460           setval ("timestamping", "on");
 461           break;
 462         case 'p':
 463           setval ("pagerequisites", "on");
 464           break;
 465         case 'S':
 466           setval ("serverresponse", "on");
 467           break;
 468         case 's':
 469           setval ("saveheaders", "on");
 470           break;
 471         case 'q':
 472           setval ("quiet", "on");
 473           break;
 474         case 'r':
 475           setval ("recursive", "on");
 476           break;
 477         case 'V':
 478           printf ("GNU Wget %s\n\n", version_string);
 479           printf ("%s", _("\
 480 Copyright (C) 1995, 1996, 1997, 1998, 2000 Free Software Foundation, Inc.\n\
 481 This program is distributed in the hope that it will be useful,\n\
 482 but WITHOUT ANY WARRANTY; without even the implied warranty of\n\
 483 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n\
 484 GNU General Public License for more details.\n"));
 485           printf (_("\nOriginally written by Hrvoje Niksic <hniksic@arsdigita.com>.\n"));
 486           exit (0);
 487           break;
 488         case 'v':
 489           setval ("verbose", "on");
 490           break;
 491         case 'x':
 492           setval ("dirstruct", "on");
 493           break;
 494
 495           /* Options accepting an argument: */
 496         case 129:
 497           setval ("httpuser", optarg);
 498           break;
 499         case 130:
 500           setval ("httppasswd", optarg);
 501           break;
 502         case 131:
 503           setval ("header", optarg);
 504           break;
 505         case 134:
 506           setval ("dotstyle", optarg);
 507           break;
 508         case 135:
 509           setval ("htmlify", optarg);
 510           break;
 511         case 140:
 512           setval ("excludedomains", optarg);
 513           break;
 514         case 143:
 515           setval ("proxyuser", optarg);
 516           break;
 517         case 144:
 518           setval ("proxypasswd", optarg);
 519           break;
 520         case 151:
 521           setval ("backups", optarg);
 522           break;
 523         case 152:
 524           setval ("waitretry", optarg);
 525           wr = 1;
 526           break;
 527         case 153:
 528           setval ("followtags", optarg);
 529           break;
 530         case 160:
 531           setval ("cookies", "on");
 532           break;
 533         case 161:
 534           setval ("cookies", "on");
 535           setval ("cookiein", optarg);
 536           setval ("cookieout", optarg);
 537           break;
 538         case 162:
 539           setval ("cookies", "on");
 540           setval ("cookiein", optarg);
 541           break;
 542         case 163:
 543           setval ("cookies", "on");
 544           setval ("cookieout", optarg);
 545           break;
 546         case 157:
 547           setval ("referer", optarg);
 548           break;
 549 #ifdef HAVE_SSL
 550         case 158:
 551           setval ("sslcertfile", optarg);
 552           break;
 553         case 159:
 554           setval ("sslcertkey", optarg);
 555           break;
 556 #endif /* HAVE_SSL */
 557         case 'A':
 558           setval ("accept", optarg);
 559           break;
 560         case 'a':
 561           setval ("logfile", optarg);
 562           append_to_log = 1;
 563           break;
 564         case 'B':
 565           setval ("base", optarg);
 566           break;
 567         case 'C':
 568           setval ("cache", optarg);
 569           break;
 570         case 'D':
 571           setval ("domains", optarg);
 572           break;
 573         case 'e':
 574           {
 575             char *com, *val;
 576             if (parse_line (optarg, &com, &val))
 577               {
 578                 if (!setval (com, val))
 579                   exit (1);
 580               }
 581             else
 582               {
 583                 fprintf (stderr, _("%s: %s: invalid command\n"), exec_name,
 584                          optarg);
 585                 exit (1);
 586               }
 587             xfree (com);
 588             xfree (val);
 589           }
 590           break;
 591         case 'G':
 592           setval ("ignoretags", optarg);
 593           break;
 594         case 'g':
 595           setval ("glob", optarg);
 596           break;
 597         case 'I':
 598           setval ("includedirectories", optarg);
 599           break;
 600         case 'i':
 601           setval ("input", optarg);
 602           break;
 603         case 'l':
 604           setval ("reclevel", optarg);
 605           break;
 606         case 'n':
 607           {
 608             /* #### The n? options are utter crock!  */
 609             char *p;
 610
 611             for (p = optarg; *p; p++)
 612               switch (*p)
 613                 {
 614                 case 'v':
 615                   setval ("verbose", "off");
 616                   break;
 617                 case 'h':
 618                   setval ("simplehostcheck", "on");
 619                   break;
 620                 case 'H':
 621                   setval ("addhostdir", "off");
 622                   break;
 623                 case 'd':
 624                   setval ("dirstruct", "off");
 625                   break;
 626                 case 'c':
 627                   setval ("noclobber", "on");
 628                   break;
 629                 case 'r':
 630                   setval ("removelisting", "off");
 631                   break;
 632                 case 'p':
 633                   setval ("noparent", "on");
 634                   break;
 635                 case 'k':
 636                   setval ("httpkeepalive", "off");
 637                   break;
 638                 default:
 639                   printf (_("%s: illegal option -- `-n%c'\n"), exec_name, *p);
 640                   print_usage ();
 641                   printf ("\n");
 642                   printf (_("Try `%s --help\' for more options.\n"), exec_name);
 643                   exit (1);
 644                 }
 645             break;
 646           }
 647         case 'O':
 648           setval ("outputdocument", optarg);
 649           break;
 650         case 'o':
 651           setval ("logfile", optarg);
 652           break;
 653         case 'P':
 654           setval ("dirprefix", optarg);
 655           break;
 656         case 'Q':
 657           setval ("quota", optarg);
 658           break;
 659         case 'R':
 660           setval ("reject", optarg);
 661           break;
 662         case 'T':
 663           setval ("timeout", optarg);
 664           break;
 665         case 't':
 666           setval ("tries", optarg);
 667           break;
 668         case 'U':
 669           setval ("useragent", optarg);
 670           break;
 671         case 'w':
 672           setval ("wait", optarg);
 673           break;
 674         case 'X':
 675           setval ("excludedirectories", optarg);
 676           break;
 677         case 'Y':
 678           setval ("useproxy", optarg);
 679           break;
 680
 681         case '?':
 682           print_usage ();
 683           printf ("\n");
 684           printf (_("Try `%s --help' for more options.\n"), exec_name);
 685           exit (0);
 686           break;
 687         }
 688     }
 689
 690   /* All user options have now been processed, so it's now safe to do
 691      interoption dependency checks. */
 692
 693   if (opt.reclevel == 0)
 694     opt.reclevel = INFINITE_RECURSION;  /* see wget.h for commentary on this */
 695
 696   if (opt.page_requisites && !opt.recursive)
 697     {
 698       opt.recursive = TRUE;
 699       opt.reclevel = 0;
 700       if (!opt.no_dirstruct)
 701         opt.dirstruct = TRUE;  /* usually handled by cmd_spec_recursive() */
 702     }
 703
 704   if (opt.verbose == -1)
 705     opt.verbose = !opt.quiet;
 706
 707   /* Retain compatibility with previous scripts.
 708      if wait has been set, but waitretry has not, give it the wait value.
 709      A simple check on the values is not enough, I could have set
 710      wait to n>0 and waitretry to 0 - HEH */
 711   if (opt.wait && !wr)
 712     {
 713       char  opt_wait_str[256];  /* bigger than needed buf to prevent overflow */
 714
 715       sprintf(opt_wait_str, "%ld", opt.wait);
 716       setval ("waitretry", opt_wait_str);
 717     }
 718
 719   /* Sanity checks.  */
 720   if (opt.verbose && opt.quiet)
 721     {
 722       printf (_("Can't be verbose and quiet at the same time.\n"));
 723       print_usage ();
 724       exit (1);
 725     }
 726   if (opt.timestamping && opt.noclobber)
 727     {
 728       printf (_("\
 729 Can't timestamp and not clobber old files at the same time.\n"));
 730       print_usage ();
 731       exit (1);
 732     }
 733   nurl = argc - optind;
 734   if (!nurl && !opt.input_filename)
 735     {
 736       /* No URL specified.  */
 737       printf (_("%s: missing URL\n"), exec_name);
 738       print_usage ();
 739       printf ("\n");
 740       /* #### Something nicer should be printed here -- similar to the
 741          pre-1.5 `--help' page.  */
 742       printf (_("Try `%s --help' for more options.\n"), exec_name);
 743       exit (1);
 744     }
 745
 746   if (opt.background)
 747     fork_to_background ();
 748
 749   /* Allocate basic pointer.  */
 750   url = ALLOCA_ARRAY (char *, nurl + 1);
 751   /* Fill in the arguments.  */
 752   for (i = 0; i < nurl; i++, optind++)
 753     {
 754       char *irix4_cc_needs_this;
 755       STRDUP_ALLOCA (irix4_cc_needs_this, argv[optind]);
 756       url[i] = irix4_cc_needs_this;
 757     }
 758   url[i] = NULL;
 759
 760   /* Change the title of console window on Windows.  #### I think this
 761      statement should belong to retrieve_url().  --hniksic.  */
 762 #ifdef WINDOWS
 763   ws_changetitle (*url, nurl);
 764 #endif
 765
 766   /* Initialize logging.  */
 767   log_init (opt.lfilename, append_to_log);
 768
 769   DEBUGP (("DEBUG output created by Wget %s on %s.\n\n", version_string,
 770            OS_TYPE));
 771
 772   /* Open the output filename if necessary.  */
 773   if (opt.output_document)
 774     {
 775       if (HYPHENP (opt.output_document))
 776         opt.dfp = stdout;
 777       else
 778         {
 779           struct stat st;
 780           opt.dfp = fopen (opt.output_document, opt.always_rest ? "ab" : "wb");
 781           if (opt.dfp == NULL)
 782             {
 783               perror (opt.output_document);
 784               exit (1);
 785             }
 786           if (fstat (fileno (opt.dfp), &st) == 0 && S_ISREG (st.st_mode))
 787             opt.od_known_regular = 1;
 788         }
 789     }
 790
 791 #ifdef WINDOWS
 792   ws_startup ();
 793 #endif
 794
 795   if (opt.cookies_input)
 796       load_cookies (opt.cookies_input);
 797
 798   /* Setup the signal handler to redirect output when hangup is
 799      received.  */
 800 #ifdef HAVE_SIGNAL
 801   if (signal(SIGHUP, SIG_IGN) != SIG_IGN)
 802     signal(SIGHUP, redirect_output_signal);
 803   /* ...and do the same for SIGUSR1.  */
 804   signal (SIGUSR1, redirect_output_signal);
 805   /* Writing to a closed socket normally signals SIGPIPE, and the
 806      process exits.  What we want is to ignore SIGPIPE and just check
 807      for the return value of write().  */
 808   signal (SIGPIPE, SIG_IGN);
 809 #endif /* HAVE_SIGNAL */
 810
 811   status = RETROK;              /* initialize it, just-in-case */
 812   recursive_reset ();
 813   /* Retrieve the URLs from argument list.  */
 814   for (t = url; *t; t++)
 815     {
 816       char *filename, *redirected_URL;
 817       int dt;
 818
 819       status = retrieve_url (*t, &filename, &redirected_URL, NULL, &dt);
 820       if (opt.recursive && status == RETROK && (dt & TEXTHTML))
 821         status = recursive_retrieve (filename,
 822                                      redirected_URL ? redirected_URL : *t);
 823
 824       if (opt.delete_after && file_exists_p(filename))
 825         {
 826           DEBUGP (("Removing file due to --delete-after in main():\n"));
 827           logprintf (LOG_VERBOSE, _("Removing %s.\n"), filename);
 828           if (unlink (filename))
 829             logprintf (LOG_NOTQUIET, "unlink: %s\n", strerror (errno));
 830         }
 831
 832       FREE_MAYBE (redirected_URL);
 833       FREE_MAYBE (filename);
 834     }
 835
 836   /* And then from the input file, if any.  */
 837   if (opt.input_filename)
 838     {
 839       int count;
 840       status = retrieve_from_file (opt.input_filename, opt.force_html, &count);
 841       if (!count)
 842         logprintf (LOG_NOTQUIET, _("No URLs found in %s.\n"),
 843                    opt.input_filename);
 844     }
 845   /* Print the downloaded sum.  */
 846   if (opt.recursive
 847       || nurl > 1
 848       || (opt.input_filename && opt.downloaded != 0))
 849     {
 850       logprintf (LOG_NOTQUIET,
 851                  _("\nFINISHED --%s--\nDownloaded: %s bytes in %d files\n"),
 852                  time_str (NULL),
 853                  (opt.downloaded_overflow ?
 854                   "<overflow>" : legible_very_long (opt.downloaded)),
 855                  opt.numurls);
 856       /* Print quota warning, if exceeded.  */
 857       if (downloaded_exceeds_quota ())
 858         logprintf (LOG_NOTQUIET,
 859                    _("Download quota (%s bytes) EXCEEDED!\n"),
 860                    legible (opt.quota));
 861     }
 862
 863   if (opt.cookies_output)
 864     save_cookies (opt.cookies_output);
 865
 866   if (opt.convert_links && !opt.delete_after)
 867     {
 868       convert_all_links ();
 869     }
 870   log_close ();
 871   cleanup ();
 872 #ifdef DEBUG_MALLOC
 873   print_malloc_debug_stats ();
 874 #endif
 875   if (status == RETROK)
 876     return 0;
 877   else
 878     return 1;
 879 }
 880 \f
 881 /* Hangup signal handler.  When wget receives SIGHUP or SIGUSR1, it
 882    will proceed operation as usual, trying to write into a log file.
 883    If that is impossible, the output will be turned off.  */
 884
 885 #ifdef HAVE_SIGNAL
 886 static RETSIGTYPE
 887 redirect_output_signal (int sig)
 888 {
 889   char tmp[100];
 890   signal (sig, redirect_output_signal);
 891   /* Please note that the double `%' in `%%s' is intentional, because
 892      redirect_output passes tmp through printf.  */
 893   sprintf (tmp, _("%s received, redirecting output to `%%s'.\n"),
 894            (sig == SIGHUP ? "SIGHUP" :
 895             (sig == SIGUSR1 ? "SIGUSR1" :
 896              "WTF?!")));
 897   redirect_output (tmp);
 898 }
 899 #endif /* HAVE_SIGNAL */