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