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

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