]> sjero.net Git - wget/blob - src/main.c
Set stdout mode to binary under Windows.
[wget] / src / main.c
1 /* Command line parsing.
2    Copyright (C) 1996, 1997, 1998, 1999, 2000, 2001, 2002, 2003, 2004,
3    2005, 2006, 2007, 2008, 2009, 2010, 2011 Free Software Foundation,
4    Inc.
5
6 This file is part of GNU Wget.
7
8 GNU Wget is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
12
13 GNU Wget is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 GNU General Public License for more details.
17
18 You should have received a copy of the GNU General Public License
19 along with Wget.  If not, see <http://www.gnu.org/licenses/>.
20
21 Additional permission under GNU GPL version 3 section 7
22
23 If you modify this program, or any covered work, by linking or
24 combining it with the OpenSSL project's OpenSSL library (or a
25 modified version of that library), containing parts covered by the
26 terms of the OpenSSL or SSLeay licenses, the Free Software Foundation
27 grants you additional permission to convey the resulting work.
28 Corresponding Source for a non-source form of such a combination
29 shall include the source code for the parts of OpenSSL used as well
30 as that of the covered work.  */
31
32 #include "wget.h"
33
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <unistd.h>
37 #include <string.h>
38 #include <signal.h>
39 #ifdef ENABLE_NLS
40 # include <locale.h>
41 #endif
42 #include <assert.h>
43 #include <errno.h>
44 #include <time.h>
45
46 #include "exits.h"
47 #include "utils.h"
48 #include "init.h"
49 #include "retr.h"
50 #include "recur.h"
51 #include "host.h"
52 #include "url.h"
53 #include "progress.h"           /* for progress_handle_sigwinch */
54 #include "convert.h"
55 #include "spider.h"
56 #include "http.h"               /* for save_cookies */
57 #include "ptimer.h"
58
59 #include <getopt.h>
60 #include <getpass.h>
61 #include <quote.h>
62
63 #ifdef WINDOWS
64 # include <io.h>
65 # include <fcntl.h>
66 #endif
67
68 #ifdef __VMS
69 # include "vms.h"
70 #endif /* __VMS */
71
72 #ifndef PATH_SEPARATOR
73 # define PATH_SEPARATOR '/'
74 #endif
75
76 #ifndef ENABLE_IRI
77 struct iri dummy_iri;
78 #endif
79
80 struct options opt;
81
82 /* defined in version.c */
83 extern char *version_string;
84 extern char *compilation_string;
85 extern char *system_getrc;
86 extern char *link_string;
87 /* defined in build_info.c */
88 extern const char *compiled_features[];
89 /* Used for --version output in print_version */
90 #define MAX_CHARS_PER_LINE      72
91 #define TABULATION              4
92
93 #if defined(SIGHUP) || defined(SIGUSR1)
94 static void redirect_output_signal (int);
95 #endif
96
97 const char *exec_name;
98
99 /* Number of successfully downloaded URLs */
100 int numurls = 0;
101 \f
102 #ifndef TESTING
103 /* Initialize I18N/L10N.  That amounts to invoking setlocale, and
104    setting up gettext's message catalog using bindtextdomain and
105    textdomain.  Does nothing if NLS is disabled or missing.  */
106
107 static void
108 i18n_initialize (void)
109 {
110   /* ENABLE_NLS implies existence of functions invoked here.  */
111 #ifdef ENABLE_NLS
112   /* Set the current locale.  */
113   setlocale (LC_ALL, "");
114   /* Set the text message domain.  */
115   bindtextdomain ("wget", LOCALEDIR);
116   textdomain ("wget");
117 #endif /* ENABLE_NLS */
118 }
119 \f
120 /* Definition of command-line options. */
121
122 static void print_help (void);
123 static void print_version (void);
124
125 #ifdef HAVE_SSL
126 # define IF_SSL(x) x
127 #else
128 # define IF_SSL(x) NULL
129 #endif
130
131 #ifdef ENABLE_DEBUG
132 # define WHEN_DEBUG(x) x
133 #else
134 # define WHEN_DEBUG(x) NULL
135 #endif
136
137 struct cmdline_option {
138   const char *long_name;
139   char short_name;
140   enum {
141     OPT_VALUE,
142     OPT_BOOLEAN,
143     OPT_FUNCALL,
144     /* Non-standard options that have to be handled specially in
145        main().  */
146     OPT__APPEND_OUTPUT,
147     OPT__CLOBBER,
148     OPT__DONT_REMOVE_LISTING,
149     OPT__EXECUTE,
150     OPT__NO,
151     OPT__PARENT
152   } type;
153   const void *data;             /* for standard options */
154   int argtype;                  /* for non-standard options */
155 };
156
157 static struct cmdline_option option_data[] =
158   {
159     { "accept", 'A', OPT_VALUE, "accept", -1 },
160     { "adjust-extension", 'E', OPT_BOOLEAN, "adjustextension", -1 },
161     { "append-output", 'a', OPT__APPEND_OUTPUT, NULL, required_argument },
162     { "ask-password", 0, OPT_BOOLEAN, "askpassword", -1 },
163     { "auth-no-challenge", 0, OPT_BOOLEAN, "authnochallenge", -1 },
164     { "background", 'b', OPT_BOOLEAN, "background", -1 },
165     { "backup-converted", 'K', OPT_BOOLEAN, "backupconverted", -1 },
166     { "backups", 0, OPT_BOOLEAN, "backups", -1 },
167     { "base", 'B', OPT_VALUE, "base", -1 },
168     { "bind-address", 0, OPT_VALUE, "bindaddress", -1 },
169     { IF_SSL ("ca-certificate"), 0, OPT_VALUE, "cacertificate", -1 },
170     { IF_SSL ("ca-directory"), 0, OPT_VALUE, "cadirectory", -1 },
171     { "cache", 0, OPT_BOOLEAN, "cache", -1 },
172     { IF_SSL ("certificate"), 0, OPT_VALUE, "certificate", -1 },
173     { IF_SSL ("certificate-type"), 0, OPT_VALUE, "certificatetype", -1 },
174     { IF_SSL ("check-certificate"), 0, OPT_BOOLEAN, "checkcertificate", -1 },
175     { "clobber", 0, OPT__CLOBBER, NULL, optional_argument },
176     { "config", 0, OPT_VALUE, "chooseconfig", -1 },
177     { "connect-timeout", 0, OPT_VALUE, "connecttimeout", -1 },
178     { "continue", 'c', OPT_BOOLEAN, "continue", -1 },
179     { "convert-links", 'k', OPT_BOOLEAN, "convertlinks", -1 },
180     { "content-disposition", 0, OPT_BOOLEAN, "contentdisposition", -1 },
181     { "cookies", 0, OPT_BOOLEAN, "cookies", -1 },
182     { "cut-dirs", 0, OPT_VALUE, "cutdirs", -1 },
183     { WHEN_DEBUG ("debug"), 'd', OPT_BOOLEAN, "debug", -1 },
184     { "default-page", 0, OPT_VALUE, "defaultpage", -1 },
185     { "delete-after", 0, OPT_BOOLEAN, "deleteafter", -1 },
186     { "directories", 0, OPT_BOOLEAN, "dirstruct", -1 },
187     { "directory-prefix", 'P', OPT_VALUE, "dirprefix", -1 },
188     { "dns-cache", 0, OPT_BOOLEAN, "dnscache", -1 },
189     { "dns-timeout", 0, OPT_VALUE, "dnstimeout", -1 },
190     { "domains", 'D', OPT_VALUE, "domains", -1 },
191     { "dont-remove-listing", 0, OPT__DONT_REMOVE_LISTING, NULL, no_argument },
192     { "dot-style", 0, OPT_VALUE, "dotstyle", -1 }, /* deprecated */
193     { "egd-file", 0, OPT_VALUE, "egdfile", -1 },
194     { "exclude-directories", 'X', OPT_VALUE, "excludedirectories", -1 },
195     { "exclude-domains", 0, OPT_VALUE, "excludedomains", -1 },
196     { "execute", 'e', OPT__EXECUTE, NULL, required_argument },
197     { "follow-ftp", 0, OPT_BOOLEAN, "followftp", -1 },
198     { "follow-tags", 0, OPT_VALUE, "followtags", -1 },
199     { "force-directories", 'x', OPT_BOOLEAN, "dirstruct", -1 },
200     { "force-html", 'F', OPT_BOOLEAN, "forcehtml", -1 },
201     { "ftp-password", 0, OPT_VALUE, "ftppassword", -1 },
202 #ifdef __VMS
203     { "ftp-stmlf", 0, OPT_BOOLEAN, "ftpstmlf", -1 },
204 #endif /* def __VMS */
205     { "ftp-user", 0, OPT_VALUE, "ftpuser", -1 },
206     { "glob", 0, OPT_BOOLEAN, "glob", -1 },
207     { "header", 0, OPT_VALUE, "header", -1 },
208     { "help", 'h', OPT_FUNCALL, (void *)print_help, no_argument },
209     { "host-directories", 0, OPT_BOOLEAN, "addhostdir", -1 },
210     { "html-extension", 'E', OPT_BOOLEAN, "adjustextension", -1 }, /* deprecated */
211     { "htmlify", 0, OPT_BOOLEAN, "htmlify", -1 },
212     { "http-keep-alive", 0, OPT_BOOLEAN, "httpkeepalive", -1 },
213     { "http-passwd", 0, OPT_VALUE, "httppassword", -1 }, /* deprecated */
214     { "http-password", 0, OPT_VALUE, "httppassword", -1 },
215     { "http-user", 0, OPT_VALUE, "httpuser", -1 },
216     { "ignore-case", 0, OPT_BOOLEAN, "ignorecase", -1 },
217     { "ignore-length", 0, OPT_BOOLEAN, "ignorelength", -1 },
218     { "ignore-tags", 0, OPT_VALUE, "ignoretags", -1 },
219     { "include-directories", 'I', OPT_VALUE, "includedirectories", -1 },
220 #ifdef ENABLE_IPV6
221     { "inet4-only", '4', OPT_BOOLEAN, "inet4only", -1 },
222     { "inet6-only", '6', OPT_BOOLEAN, "inet6only", -1 },
223 #endif
224     { "input-file", 'i', OPT_VALUE, "input", -1 },
225     { "iri", 0, OPT_BOOLEAN, "iri", -1 },
226     { "keep-session-cookies", 0, OPT_BOOLEAN, "keepsessioncookies", -1 },
227     { "level", 'l', OPT_VALUE, "reclevel", -1 },
228     { "limit-rate", 0, OPT_VALUE, "limitrate", -1 },
229     { "load-cookies", 0, OPT_VALUE, "loadcookies", -1 },
230     { "local-encoding", 0, OPT_VALUE, "localencoding", -1 },
231     { "max-redirect", 0, OPT_VALUE, "maxredirect", -1 },
232     { "mirror", 'm', OPT_BOOLEAN, "mirror", -1 },
233     { "no", 'n', OPT__NO, NULL, required_argument },
234     { "no-clobber", 0, OPT_BOOLEAN, "noclobber", -1 },
235     { "no-parent", 0, OPT_BOOLEAN, "noparent", -1 },
236     { "output-document", 'O', OPT_VALUE, "outputdocument", -1 },
237     { "output-file", 'o', OPT_VALUE, "logfile", -1 },
238     { "page-requisites", 'p', OPT_BOOLEAN, "pagerequisites", -1 },
239     { "parent", 0, OPT__PARENT, NULL, optional_argument },
240     { "passive-ftp", 0, OPT_BOOLEAN, "passiveftp", -1 },
241     { "password", 0, OPT_VALUE, "password", -1 },
242     { "post-data", 0, OPT_VALUE, "postdata", -1 },
243     { "post-file", 0, OPT_VALUE, "postfile", -1 },
244     { "prefer-family", 0, OPT_VALUE, "preferfamily", -1 },
245     { "preserve-permissions", 0, OPT_BOOLEAN, "preservepermissions", -1 }, /* deprecated */
246     { IF_SSL ("private-key"), 0, OPT_VALUE, "privatekey", -1 },
247     { IF_SSL ("private-key-type"), 0, OPT_VALUE, "privatekeytype", -1 },
248     { "progress", 0, OPT_VALUE, "progress", -1 },
249     { "protocol-directories", 0, OPT_BOOLEAN, "protocoldirectories", -1 },
250     { "proxy", 0, OPT_BOOLEAN, "useproxy", -1 },
251     { "proxy__compat", 'Y', OPT_VALUE, "useproxy", -1 }, /* back-compatible */
252     { "proxy-passwd", 0, OPT_VALUE, "proxypassword", -1 }, /* deprecated */
253     { "proxy-password", 0, OPT_VALUE, "proxypassword", -1 },
254     { "proxy-user", 0, OPT_VALUE, "proxyuser", -1 },
255     { "quiet", 'q', OPT_BOOLEAN, "quiet", -1 },
256     { "quota", 'Q', OPT_VALUE, "quota", -1 },
257     { "random-file", 0, OPT_VALUE, "randomfile", -1 },
258     { "random-wait", 0, OPT_BOOLEAN, "randomwait", -1 },
259     { "read-timeout", 0, OPT_VALUE, "readtimeout", -1 },
260     { "recursive", 'r', OPT_BOOLEAN, "recursive", -1 },
261     { "referer", 0, OPT_VALUE, "referer", -1 },
262     { "reject", 'R', OPT_VALUE, "reject", -1 },
263     { "relative", 'L', OPT_BOOLEAN, "relativeonly", -1 },
264     { "remote-encoding", 0, OPT_VALUE, "remoteencoding", -1 },
265     { "remove-listing", 0, OPT_BOOLEAN, "removelisting", -1 },
266     { "restrict-file-names", 0, OPT_BOOLEAN, "restrictfilenames", -1 },
267     { "retr-symlinks", 0, OPT_BOOLEAN, "retrsymlinks", -1 },
268     { "retry-connrefused", 0, OPT_BOOLEAN, "retryconnrefused", -1 },
269     { "save-cookies", 0, OPT_VALUE, "savecookies", -1 },
270     { "save-headers", 0, OPT_BOOLEAN, "saveheaders", -1 },
271     { IF_SSL ("secure-protocol"), 0, OPT_VALUE, "secureprotocol", -1 },
272     { "server-response", 'S', OPT_BOOLEAN, "serverresponse", -1 },
273     { "span-hosts", 'H', OPT_BOOLEAN, "spanhosts", -1 },
274     { "spider", 0, OPT_BOOLEAN, "spider", -1 },
275     { "strict-comments", 0, OPT_BOOLEAN, "strictcomments", -1 },
276     { "timeout", 'T', OPT_VALUE, "timeout", -1 },
277     { "timestamping", 'N', OPT_BOOLEAN, "timestamping", -1 },
278     { "tries", 't', OPT_VALUE, "tries", -1 },
279     { "unlink", 0, OPT_BOOLEAN, "unlink", -1 },
280     { "trust-server-names", 0, OPT_BOOLEAN, "trustservernames", -1 },
281     { "use-server-timestamps", 0, OPT_BOOLEAN, "useservertimestamps", -1 },
282     { "user", 0, OPT_VALUE, "user", -1 },
283     { "user-agent", 'U', OPT_VALUE, "useragent", -1 },
284     { "verbose", 'v', OPT_BOOLEAN, "verbose", -1 },
285     { "verbose", 0, OPT_BOOLEAN, "verbose", -1 },
286     { "version", 'V', OPT_FUNCALL, (void *) print_version, no_argument },
287     { "wait", 'w', OPT_VALUE, "wait", -1 },
288     { "waitretry", 0, OPT_VALUE, "waitretry", -1 },
289 #ifdef USE_WATT32
290     { "wdebug", 0, OPT_BOOLEAN, "wdebug", -1 },
291 #endif
292   };
293
294 #undef WHEN_DEBUG
295 #undef IF_SSL
296
297 /* Return a string that contains S with "no-" prepended.  The string
298    is NUL-terminated and allocated off static storage at Wget
299    startup.  */
300
301 static char *
302 no_prefix (const char *s)
303 {
304   static char buffer[1024];
305   static char *p = buffer;
306
307   char *cp = p;
308   int size = 3 + strlen (s) + 1;  /* "no-STRING\0" */
309   if (p + size >= buffer + sizeof (buffer))
310     abort ();
311
312   cp[0] = 'n', cp[1] = 'o', cp[2] = '-';
313   strcpy (cp + 3, s);
314   p += size;
315   return cp;
316 }
317
318 /* The arguments that that main passes to getopt_long. */
319 static struct option long_options[2 * countof (option_data) + 1];
320 static char short_options[128];
321
322 /* Mapping between short option chars and option_data indices. */
323 static unsigned char optmap[96];
324
325 /* Marker for `--no-FOO' values in long_options.  */
326 #define BOOLEAN_NEG_MARKER 1024
327
328 /* Initialize the long_options array used by getopt_long from the data
329    in option_data.  */
330
331 static void
332 init_switches (void)
333 {
334   char *p = short_options;
335   size_t i, o = 0;
336   for (i = 0; i < countof (option_data); i++)
337     {
338       struct cmdline_option *opt = &option_data[i];
339       struct option *longopt;
340
341       if (!opt->long_name)
342         /* The option is disabled. */
343         continue;
344
345       longopt = &long_options[o++];
346       longopt->name = opt->long_name;
347       longopt->val = i;
348       if (opt->short_name)
349         {
350           *p++ = opt->short_name;
351           optmap[opt->short_name - 32] = longopt - long_options;
352         }
353       switch (opt->type)
354         {
355         case OPT_VALUE:
356           longopt->has_arg = required_argument;
357           if (opt->short_name)
358             *p++ = ':';
359           break;
360         case OPT_BOOLEAN:
361           /* Specify an optional argument for long options, so that
362              --option=off works the same as --no-option, for
363              compatibility with pre-1.10 Wget.  However, don't specify
364              optional arguments short-option booleans because they
365              prevent combining of short options.  */
366           longopt->has_arg = optional_argument;
367           /* For Boolean options, add the "--no-FOO" variant, which is
368              identical to "--foo", except it has opposite meaning and
369              it doesn't allow an argument.  */
370           longopt = &long_options[o++];
371           longopt->name = no_prefix (opt->long_name);
372           longopt->has_arg = no_argument;
373           /* Mask the value so we'll be able to recognize that we're
374              dealing with the false value.  */
375           longopt->val = i | BOOLEAN_NEG_MARKER;
376           break;
377         default:
378           assert (opt->argtype != -1);
379           longopt->has_arg = opt->argtype;
380           if (opt->short_name)
381             {
382               if (longopt->has_arg == required_argument)
383                 *p++ = ':';
384               /* Don't handle optional_argument */
385             }
386         }
387     }
388   /* Terminate short_options. */
389   *p = '\0';
390   /* No need for xzero(long_options[o]) because its storage is static
391      and it will be zeroed by default.  */
392   assert (o <= countof (long_options));
393 }
394
395 /* Print the usage message.  */
396 static void
397 print_usage (int error)
398 {
399   fprintf (error ? stderr : stdout, _("Usage: %s [OPTION]... [URL]...\n"),
400            exec_name);
401 }
402
403 /* Print the help message, describing all the available options.  If
404    you add an option, be sure to update this list.  */
405 static void
406 print_help (void)
407 {
408   /* We split the help text this way to ease translation of individual
409      entries.  */
410   static const char *help[] = {
411     "\n",
412     N_("\
413 Mandatory arguments to long options are mandatory for short options too.\n\n"),
414     N_("\
415 Startup:\n"),
416     N_("\
417   -V,  --version           display the version of Wget and exit.\n"),
418     N_("\
419   -h,  --help              print this help.\n"),
420     N_("\
421   -b,  --background        go to background after startup.\n"),
422     N_("\
423   -e,  --execute=COMMAND   execute a `.wgetrc'-style command.\n"),
424     "\n",
425
426     N_("\
427 Logging and input file:\n"),
428     N_("\
429   -o,  --output-file=FILE    log messages to FILE.\n"),
430     N_("\
431   -a,  --append-output=FILE  append messages to FILE.\n"),
432 #ifdef ENABLE_DEBUG
433     N_("\
434   -d,  --debug               print lots of debugging information.\n"),
435 #endif
436 #ifdef USE_WATT32
437     N_("\
438        --wdebug              print Watt-32 debug output.\n"),
439 #endif
440     N_("\
441   -q,  --quiet               quiet (no output).\n"),
442     N_("\
443   -v,  --verbose             be verbose (this is the default).\n"),
444     N_("\
445   -nv, --no-verbose          turn off verboseness, without being quiet.\n"),
446     N_("\
447   -i,  --input-file=FILE     download URLs found in local or external FILE.\n"),
448     N_("\
449   -F,  --force-html          treat input file as HTML.\n"),
450     N_("\
451   -B,  --base=URL            resolves HTML input-file links (-i -F)\n\
452                              relative to URL.\n"),
453     N_("\
454        --config=FILE         Specify config file to use.\n"), 
455     "\n",
456
457     N_("\
458 Download:\n"),
459     N_("\
460   -t,  --tries=NUMBER            set number of retries to NUMBER (0 unlimits).\n"),
461     N_("\
462        --retry-connrefused       retry even if connection is refused.\n"),
463     N_("\
464   -O,  --output-document=FILE    write documents to FILE.\n"),
465     N_("\
466   -nc, --no-clobber              skip downloads that would download to\n\
467                                  existing files (overwriting them).\n"),
468     N_("\
469   -c,  --continue                resume getting a partially-downloaded file.\n"),
470     N_("\
471        --progress=TYPE           select progress gauge type.\n"),
472     N_("\
473   -N,  --timestamping            don't re-retrieve files unless newer than\n\
474                                  local.\n"),
475     N_("\
476   --no-use-server-timestamps     don't set the local file's timestamp by\n\
477                                  the one on the server.\n"),
478     N_("\
479   -S,  --server-response         print server response.\n"),
480     N_("\
481        --spider                  don't download anything.\n"),
482     N_("\
483   -T,  --timeout=SECONDS         set all timeout values to SECONDS.\n"),
484     N_("\
485        --dns-timeout=SECS        set the DNS lookup timeout to SECS.\n"),
486     N_("\
487        --connect-timeout=SECS    set the connect timeout to SECS.\n"),
488     N_("\
489        --read-timeout=SECS       set the read timeout to SECS.\n"),
490     N_("\
491   -w,  --wait=SECONDS            wait SECONDS between retrievals.\n"),
492     N_("\
493        --waitretry=SECONDS       wait 1..SECONDS between retries of a retrieval.\n"),
494     N_("\
495        --random-wait             wait from 0.5*WAIT...1.5*WAIT secs between retrievals.\n"),
496     N_("\
497        --no-proxy                explicitly turn off proxy.\n"),
498     N_("\
499   -Q,  --quota=NUMBER            set retrieval quota to NUMBER.\n"),
500     N_("\
501        --bind-address=ADDRESS    bind to ADDRESS (hostname or IP) on local host.\n"),
502     N_("\
503        --limit-rate=RATE         limit download rate to RATE.\n"),
504     N_("\
505        --no-dns-cache            disable caching DNS lookups.\n"),
506     N_("\
507        --restrict-file-names=OS  restrict chars in file names to ones OS allows.\n"),
508     N_("\
509        --ignore-case             ignore case when matching files/directories.\n"),
510 #ifdef ENABLE_IPV6
511     N_("\
512   -4,  --inet4-only              connect only to IPv4 addresses.\n"),
513     N_("\
514   -6,  --inet6-only              connect only to IPv6 addresses.\n"),
515     N_("\
516        --prefer-family=FAMILY    connect first to addresses of specified family,\n\
517                                  one of IPv6, IPv4, or none.\n"),
518 #endif
519     N_("\
520        --user=USER               set both ftp and http user to USER.\n"),
521     N_("\
522        --password=PASS           set both ftp and http password to PASS.\n"),
523     N_("\
524        --ask-password            prompt for passwords.\n"),
525     N_("\
526        --no-iri                  turn off IRI support.\n"),
527     N_("\
528        --local-encoding=ENC      use ENC as the local encoding for IRIs.\n"),
529     N_("\
530        --remote-encoding=ENC     use ENC as the default remote encoding.\n"),
531     N_("\
532        --unlink                  remove file before clobber.\n"),
533     "\n",
534
535     N_("\
536 Directories:\n"),
537     N_("\
538   -nd, --no-directories           don't create directories.\n"),
539     N_("\
540   -x,  --force-directories        force creation of directories.\n"),
541     N_("\
542   -nH, --no-host-directories      don't create host directories.\n"),
543     N_("\
544        --protocol-directories     use protocol name in directories.\n"),
545     N_("\
546   -P,  --directory-prefix=PREFIX  save files to PREFIX/...\n"),
547     N_("\
548        --cut-dirs=NUMBER          ignore NUMBER remote directory components.\n"),
549     "\n",
550
551     N_("\
552 HTTP options:\n"),
553     N_("\
554        --http-user=USER        set http user to USER.\n"),
555     N_("\
556        --http-password=PASS    set http password to PASS.\n"),
557     N_("\
558        --no-cache              disallow server-cached data.\n"),
559     N_ ("\
560        --default-page=NAME     Change the default page name (normally\n\
561                                this is `index.html'.).\n"),
562     N_("\
563   -E,  --adjust-extension      save HTML/CSS documents with proper extensions.\n"),
564     N_("\
565        --ignore-length         ignore `Content-Length' header field.\n"),
566     N_("\
567        --header=STRING         insert STRING among the headers.\n"),
568     N_("\
569        --max-redirect          maximum redirections allowed per page.\n"),
570     N_("\
571        --proxy-user=USER       set USER as proxy username.\n"),
572     N_("\
573        --proxy-password=PASS   set PASS as proxy password.\n"),
574     N_("\
575        --referer=URL           include `Referer: URL' header in HTTP request.\n"),
576     N_("\
577        --save-headers          save the HTTP headers to file.\n"),
578     N_("\
579   -U,  --user-agent=AGENT      identify as AGENT instead of Wget/VERSION.\n"),
580     N_("\
581        --no-http-keep-alive    disable HTTP keep-alive (persistent connections).\n"),
582     N_("\
583        --no-cookies            don't use cookies.\n"),
584     N_("\
585        --load-cookies=FILE     load cookies from FILE before session.\n"),
586     N_("\
587        --save-cookies=FILE     save cookies to FILE after session.\n"),
588     N_("\
589        --keep-session-cookies  load and save session (non-permanent) cookies.\n"),
590     N_("\
591        --post-data=STRING      use the POST method; send STRING as the data.\n"),
592     N_("\
593        --post-file=FILE        use the POST method; send contents of FILE.\n"),
594     N_("\
595        --content-disposition   honor the Content-Disposition header when\n\
596                                choosing local file names (EXPERIMENTAL).\n"),
597     N_("\
598        --auth-no-challenge     send Basic HTTP authentication information\n\
599                                without first waiting for the server's\n\
600                                challenge.\n"),
601     "\n",
602
603 #ifdef HAVE_SSL
604     N_("\
605 HTTPS (SSL/TLS) options:\n"),
606     N_("\
607        --secure-protocol=PR     choose secure protocol, one of auto, SSLv2,\n\
608                                 SSLv3, and TLSv1.\n"),
609     N_("\
610        --no-check-certificate   don't validate the server's certificate.\n"),
611     N_("\
612        --certificate=FILE       client certificate file.\n"),
613     N_("\
614        --certificate-type=TYPE  client certificate type, PEM or DER.\n"),
615     N_("\
616        --private-key=FILE       private key file.\n"),
617     N_("\
618        --private-key-type=TYPE  private key type, PEM or DER.\n"),
619     N_("\
620        --ca-certificate=FILE    file with the bundle of CA's.\n"),
621     N_("\
622        --ca-directory=DIR       directory where hash list of CA's is stored.\n"),
623     N_("\
624        --random-file=FILE       file with random data for seeding the SSL PRNG.\n"),
625     N_("\
626        --egd-file=FILE          file naming the EGD socket with random data.\n"),
627     "\n",
628 #endif /* HAVE_SSL */
629
630     N_("\
631 FTP options:\n"),
632 #ifdef __VMS
633     N_("\
634        --ftp-stmlf             Use Stream_LF format for all binary FTP files.\n"),
635 #endif /* def __VMS */
636     N_("\
637        --ftp-user=USER         set ftp user to USER.\n"),
638     N_("\
639        --ftp-password=PASS     set ftp password to PASS.\n"),
640     N_("\
641        --no-remove-listing     don't remove `.listing' files.\n"),
642     N_("\
643        --no-glob               turn off FTP file name globbing.\n"),
644     N_("\
645        --no-passive-ftp        disable the \"passive\" transfer mode.\n"),
646     N_("\
647        --retr-symlinks         when recursing, get linked-to files (not dir).\n"),
648     "\n",
649
650     N_("\
651 Recursive download:\n"),
652     N_("\
653   -r,  --recursive          specify recursive download.\n"),
654     N_("\
655   -l,  --level=NUMBER       maximum recursion depth (inf or 0 for infinite).\n"),
656     N_("\
657        --delete-after       delete files locally after downloading them.\n"),
658     N_("\
659   -k,  --convert-links      make links in downloaded HTML or CSS point to\n\
660                             local files.\n"),
661 #ifdef __VMS
662     N_("\
663   -K,  --backup-converted   before converting file X, back up as X_orig.\n"),
664 #else /* def __VMS */
665     N_("\
666   -K,  --backup-converted   before converting file X, back up as X.orig.\n"),
667 #endif /* def __VMS [else] */
668     N_("\
669   -m,  --mirror             shortcut for -N -r -l inf --no-remove-listing.\n"),
670     N_("\
671   -p,  --page-requisites    get all images, etc. needed to display HTML page.\n"),
672     N_("\
673        --strict-comments    turn on strict (SGML) handling of HTML comments.\n"),
674     "\n",
675
676     N_("\
677 Recursive accept/reject:\n"),
678     N_("\
679   -A,  --accept=LIST               comma-separated list of accepted extensions.\n"),
680     N_("\
681   -R,  --reject=LIST               comma-separated list of rejected extensions.\n"),
682     N_("\
683   -D,  --domains=LIST              comma-separated list of accepted domains.\n"),
684     N_("\
685        --exclude-domains=LIST      comma-separated list of rejected domains.\n"),
686     N_("\
687        --follow-ftp                follow FTP links from HTML documents.\n"),
688     N_("\
689        --follow-tags=LIST          comma-separated list of followed HTML tags.\n"),
690     N_("\
691        --ignore-tags=LIST          comma-separated list of ignored HTML tags.\n"),
692     N_("\
693   -H,  --span-hosts                go to foreign hosts when recursive.\n"),
694     N_("\
695   -L,  --relative                  follow relative links only.\n"),
696     N_("\
697   -I,  --include-directories=LIST  list of allowed directories.\n"),
698     N_("\
699   --trust-server-names             use the name specified by the redirection\n\
700                                    url last component.\n"),
701     N_("\
702   -X,  --exclude-directories=LIST  list of excluded directories.\n"),
703     N_("\
704   -np, --no-parent                 don't ascend to the parent directory.\n"),
705     "\n",
706
707     N_("Mail bug reports and suggestions to <bug-wget@gnu.org>.\n")
708   };
709
710   size_t i;
711
712   printf (_("GNU Wget %s, a non-interactive network retriever.\n"),
713           version_string);
714   print_usage (0);
715
716   for (i = 0; i < countof (help); i++)
717     fputs (_(help[i]), stdout);
718
719   exit (0);
720 }
721
722 /* Return a human-readable printed representation of INTERVAL,
723    measured in seconds.  */
724
725 static char *
726 secs_to_human_time (double interval)
727 {
728   static char buf[32];
729   int secs = (int) (interval + 0.5);
730   int hours, mins, days;
731
732   days = secs / 86400, secs %= 86400;
733   hours = secs / 3600, secs %= 3600;
734   mins = secs / 60, secs %= 60;
735
736   if (days)
737     sprintf (buf, "%dd %dh %dm %ds", days, hours, mins, secs);
738   else if (hours)
739     sprintf (buf, "%dh %dm %ds", hours, mins, secs);
740   else if (mins)
741     sprintf (buf, "%dm %ds", mins, secs);
742   else
743     sprintf (buf, "%ss", print_decimal (interval));
744
745   return buf;
746 }
747
748 static char *
749 prompt_for_password (void)
750 {
751   if (opt.user)
752     fprintf (stderr, _("Password for user %s: "), quote (opt.user));
753   else
754     fprintf (stderr, _("Password: "));
755   return getpass("");
756 }
757
758 /* Function that prints the line argument while limiting it
759    to at most line_length. prefix is printed on the first line
760    and an appropriate number of spaces are added on subsequent
761    lines.*/
762 static void
763 format_and_print_line (const char *prefix, const char *line,
764                        int line_length)
765 {
766   int remaining_chars;
767   char *line_dup, *token;
768
769   assert (prefix != NULL);
770   assert (line != NULL);
771
772   line_dup = xstrdup (line);
773
774   if (line_length <= 0)
775     line_length = MAX_CHARS_PER_LINE - TABULATION;
776
777   printf ("%s", prefix);
778   remaining_chars = line_length;
779   /* We break on spaces. */
780   token = strtok (line_dup, " ");
781   while (token != NULL)
782     {
783       /* If however a token is much larger than the maximum
784          line length, all bets are off and we simply print the
785          token on the next line. */
786       if (remaining_chars <= strlen (token))
787         {
788           printf ("\n%*c", TABULATION, ' ');
789           remaining_chars = line_length - TABULATION;
790         }
791       printf ("%s ", token);
792       remaining_chars -= strlen (token) + 1;  /* account for " " */
793       token = strtok (NULL, " ");
794     }
795
796   printf ("\n");
797
798   xfree (line_dup);
799 }
800
801 static void
802 print_version (void)
803 {
804   const char *wgetrc_title  = _("Wgetrc: ");
805   const char *locale_title  = _("Locale: ");
806   const char *compile_title = _("Compile: ");
807   const char *link_title    = _("Link: ");
808   char *env_wgetrc, *user_wgetrc;
809   int i;
810
811   printf (_("GNU Wget %s built on %s.\n\n"), version_string, OS_TYPE);
812
813   for (i = 0; compiled_features[i] != NULL; )
814     {
815       int line_length = MAX_CHARS_PER_LINE;
816       while ((line_length > 0) && (compiled_features[i] != NULL))
817         {
818           printf ("%s ", compiled_features[i]);
819           line_length -= strlen (compiled_features[i]) + 2;
820           i++;
821         }
822       printf ("\n");
823     }
824   printf ("\n");
825
826   /* Handle the case when $WGETRC is unset and $HOME/.wgetrc is
827      absent. */
828   printf ("%s\n", wgetrc_title);
829   env_wgetrc = wgetrc_env_file_name ();
830   if (env_wgetrc && *env_wgetrc)
831     {
832       printf (_("    %s (env)\n"), env_wgetrc);
833       xfree (env_wgetrc);
834     }
835   user_wgetrc = wgetrc_user_file_name ();
836   if (user_wgetrc)
837     {
838       printf (_("    %s (user)\n"), user_wgetrc);
839       xfree (user_wgetrc);
840     }
841 #ifdef SYSTEM_WGETRC
842   printf (_("    %s (system)\n"), SYSTEM_WGETRC);
843 #endif
844
845 #ifdef ENABLE_NLS
846   format_and_print_line (locale_title,
847                         LOCALEDIR,
848                         MAX_CHARS_PER_LINE);
849 #endif /* def ENABLE_NLS */
850
851   if (compilation_string != NULL)
852     format_and_print_line (compile_title,
853                            compilation_string,
854                            MAX_CHARS_PER_LINE);
855
856   if (link_string != NULL)
857     format_and_print_line (link_title,
858                            link_string,
859                            MAX_CHARS_PER_LINE);
860
861   printf ("\n");
862   /* TRANSLATORS: When available, an actual copyright character
863      (cirle-c) should be used in preference to "(C)". */
864   fputs (_("\
865 Copyright (C) 2009 Free Software Foundation, Inc.\n"), stdout);
866   fputs (_("\
867 License GPLv3+: GNU GPL version 3 or later\n\
868 <http://www.gnu.org/licenses/gpl.html>.\n\
869 This is free software: you are free to change and redistribute it.\n\
870 There is NO WARRANTY, to the extent permitted by law.\n"), stdout);
871   /* TRANSLATORS: When available, please use the proper diacritics for
872      names such as this one. See en_US.po for reference. */
873   fputs (_("\nOriginally written by Hrvoje Niksic <hniksic@xemacs.org>.\n"),
874          stdout);
875   fputs (_("Please send bug reports and questions to <bug-wget@gnu.org>.\n"),
876          stdout);
877   exit (0);
878 }
879
880 char *program_name; /* Needed by lib/error.c. */
881
882 int
883 main (int argc, char **argv)
884 {
885   char **url, **t;
886   int i, ret, longindex;
887   int nurl;
888   bool append_to_log = false;
889
890   total_downloaded_bytes = 0;
891
892   program_name = argv[0];
893
894   struct ptimer *timer = ptimer_new ();
895   double start_time = ptimer_measure (timer);
896
897   i18n_initialize ();
898
899   /* Construct the name of the executable, without the directory part.  */
900 #ifdef __VMS
901   /* On VMS, lose the "dev:[dir]" prefix and the ".EXE;nnn" suffix. */
902   exec_name = vms_basename (argv[0]);
903 #else /* def __VMS */
904   exec_name = strrchr (argv[0], PATH_SEPARATOR);
905   if (!exec_name)
906     exec_name = argv[0];
907   else
908     ++exec_name;
909 #endif /* def __VMS [else] */
910
911 #ifdef WINDOWS
912   /* Drop extension (typically .EXE) from executable filename. */
913   windows_main ((char **) &exec_name);
914 #endif
915
916   /* Load the hard-coded defaults.  */
917   defaults ();
918
919   init_switches ();
920
921   /* This seperate getopt_long is needed to find the user config
922      and parse it before the other user options. */
923   longindex = -1;
924   int retconf;
925   bool use_userconfig = false;
926
927   while ((retconf = getopt_long (argc, argv,
928                                 short_options, long_options, &longindex)) != -1)
929     {
930       int confval;
931       bool userrc_ret = true;
932       struct cmdline_option *config_opt;
933       confval = long_options[longindex].val;
934       config_opt = &option_data[confval & ~BOOLEAN_NEG_MARKER];
935       if (strcmp (config_opt->long_name, "config") == 0)
936         {
937           userrc_ret &= run_wgetrc (optarg);
938           use_userconfig = true;
939         }
940       if (!userrc_ret)
941         {
942           printf ("Exiting due to error in %s\n", optarg);
943           exit (2);
944         }
945       else
946         break;
947     }
948
949   /* If the user did not specify a config, read the system wgetrc and ~/.wgetrc. */
950   if (use_userconfig == false)
951     initialize ();
952
953   opterr = 0;
954   optind = 0;
955
956   longindex = -1;
957   while ((ret = getopt_long (argc, argv,
958                              short_options, long_options, &longindex)) != -1)
959     {
960       int val;
961       struct cmdline_option *opt;
962
963       /* If LONGINDEX is unchanged, it means RET is referring a short
964          option.  */
965       if (longindex == -1)
966         {
967           if (ret == '?')
968             {
969               print_usage (0);
970               printf ("\n");
971               printf (_("Try `%s --help' for more options.\n"), exec_name);
972               exit (2);
973             }
974           /* Find the short option character in the mapping.  */
975           longindex = optmap[ret - 32];
976         }
977       val = long_options[longindex].val;
978
979       /* Use the retrieved value to locate the option in the
980          option_data array, and to see if we're dealing with the
981          negated "--no-FOO" variant of the boolean option "--foo".  */
982       opt = &option_data[val & ~BOOLEAN_NEG_MARKER];
983       switch (opt->type)
984         {
985         case OPT_VALUE:
986           setoptval (opt->data, optarg, opt->long_name);
987           break;
988         case OPT_BOOLEAN:
989           if (optarg)
990             /* The user has specified a value -- use it. */
991             setoptval (opt->data, optarg, opt->long_name);
992           else
993             {
994               /* NEG is true for `--no-FOO' style boolean options. */
995               bool neg = !!(val & BOOLEAN_NEG_MARKER);
996               setoptval (opt->data, neg ? "0" : "1", opt->long_name);
997             }
998           break;
999         case OPT_FUNCALL:
1000           {
1001             void (*func) (void) = (void (*) (void)) opt->data;
1002             func ();
1003           }
1004           break;
1005         case OPT__APPEND_OUTPUT:
1006           setoptval ("logfile", optarg, opt->long_name);
1007           append_to_log = true;
1008           break;
1009         case OPT__EXECUTE:
1010           run_command (optarg);
1011           break;
1012         case OPT__NO:
1013           {
1014             /* We support real --no-FOO flags now, but keep these
1015                short options for convenience and backward
1016                compatibility.  */
1017             char *p;
1018             for (p = optarg; p && *p; p++)
1019               switch (*p)
1020                 {
1021                 case 'v':
1022                   setoptval ("verbose", "0", opt->long_name);
1023                   break;
1024                 case 'H':
1025                   setoptval ("addhostdir", "0", opt->long_name);
1026                   break;
1027                 case 'd':
1028                   setoptval ("dirstruct", "0", opt->long_name);
1029                   break;
1030                 case 'c':
1031                   setoptval ("noclobber", "1", opt->long_name);
1032                   break;
1033                 case 'p':
1034                   setoptval ("noparent", "1", opt->long_name);
1035                   break;
1036                 default:
1037                   fprintf (stderr, _("%s: illegal option -- `-n%c'\n"),
1038                            exec_name, *p);
1039                   print_usage (1);
1040                   fprintf (stderr, "\n");
1041                   fprintf (stderr, _("Try `%s --help' for more options.\n"),
1042                            exec_name);
1043                   exit (1);
1044                 }
1045             break;
1046           }
1047         case OPT__PARENT:
1048         case OPT__CLOBBER:
1049           {
1050             /* The wgetrc commands are named noparent and noclobber,
1051                so we must revert the meaning of the cmdline options
1052                before passing the value to setoptval.  */
1053             bool flag = true;
1054             if (optarg)
1055               flag = (*optarg == '1' || c_tolower (*optarg) == 'y'
1056                       || (c_tolower (optarg[0]) == 'o'
1057                           && c_tolower (optarg[1]) == 'n'));
1058             setoptval (opt->type == OPT__PARENT ? "noparent" : "noclobber",
1059                        flag ? "0" : "1", opt->long_name);
1060             break;
1061           }
1062         case OPT__DONT_REMOVE_LISTING:
1063           setoptval ("removelisting", "0", opt->long_name);
1064           break;
1065         }
1066
1067       longindex = -1;
1068     }
1069
1070   nurl = argc - optind;
1071
1072   /* All user options have now been processed, so it's now safe to do
1073      interoption dependency checks. */
1074
1075   if (opt.noclobber && opt.convert_links)
1076     {
1077       fprintf (stderr,
1078                _("Both --no-clobber and --convert-links were specified,"
1079                  "only --convert-links will be used.\n"));
1080       opt.noclobber = false;
1081     }
1082
1083   if (opt.reclevel == 0)
1084       opt.reclevel = INFINITE_RECURSION; /* see recur.h for commentary */
1085
1086   if (opt.spider || opt.delete_after)
1087       opt.no_dirstruct = true;
1088
1089   if (opt.page_requisites && !opt.recursive)
1090     {
1091       /* Don't set opt.recursive here because it would confuse the FTP
1092          code.  Instead, call retrieve_tree below when either
1093          page_requisites or recursive is requested.  */
1094       opt.reclevel = 0;
1095       if (!opt.no_dirstruct)
1096         opt.dirstruct = 1;      /* normally handled by cmd_spec_recursive() */
1097     }
1098
1099   if (opt.verbose == -1)
1100     opt.verbose = !opt.quiet;
1101
1102   /* Sanity checks.  */
1103   if (opt.verbose && opt.quiet)
1104     {
1105       fprintf (stderr, _("Can't be verbose and quiet at the same time.\n"));
1106       print_usage (1);
1107       exit (1);
1108     }
1109   if (opt.timestamping && opt.noclobber)
1110     {
1111       fprintf (stderr, _("\
1112 Can't timestamp and not clobber old files at the same time.\n"));
1113       print_usage (1);
1114       exit (1);
1115     }
1116 #ifdef ENABLE_IPV6
1117   if (opt.ipv4_only && opt.ipv6_only)
1118     {
1119       fprintf (stderr,
1120                _("Cannot specify both --inet4-only and --inet6-only.\n"));
1121       print_usage (1);
1122       exit (1);
1123     }
1124 #endif
1125   if (opt.output_document)
1126     {
1127       if (opt.convert_links
1128           && (nurl > 1 || opt.page_requisites || opt.recursive))
1129         {
1130           fputs (_("\
1131 Cannot specify both -k and -O if multiple URLs are given, or in combination\n\
1132 with -p or -r. See the manual for details.\n\n"), stderr);
1133           print_usage (1);
1134           exit (1);
1135         }
1136       if (opt.page_requisites
1137           || opt.recursive)
1138         {
1139           logprintf (LOG_NOTQUIET, "%s", _("\
1140 WARNING: combining -O with -r or -p will mean that all downloaded content\n\
1141 will be placed in the single file you specified.\n\n"));
1142         }
1143       if (opt.timestamping)
1144         {
1145           logprintf (LOG_NOTQUIET, "%s", _("\
1146 WARNING: timestamping does nothing in combination with -O. See the manual\n\
1147 for details.\n\n"));
1148           opt.timestamping = false;
1149         }
1150       if (opt.noclobber && file_exists_p(opt.output_document))
1151            {
1152               /* Check if output file exists; if it does, exit. */
1153               logprintf (LOG_VERBOSE,
1154                          _("File `%s' already there; not retrieving.\n"),
1155                          opt.output_document);
1156               exit(1);
1157            }
1158     }
1159
1160   if (opt.ask_passwd && opt.passwd)
1161     {
1162       fprintf (stderr,
1163                _("Cannot specify both --ask-password and --password.\n"));
1164       print_usage (1);
1165       exit (1);
1166     }
1167
1168   if (!nurl && !opt.input_filename)
1169     {
1170       /* No URL specified.  */
1171       fprintf (stderr, _("%s: missing URL\n"), exec_name);
1172       print_usage (1);
1173       printf ("\n");
1174       /* #### Something nicer should be printed here -- similar to the
1175          pre-1.5 `--help' page.  */
1176       fprintf (stderr, _("Try `%s --help' for more options.\n"), exec_name);
1177       exit (1);
1178     }
1179
1180 #ifdef ENABLE_IRI
1181   if (opt.enable_iri)
1182     {
1183       if (opt.locale && !check_encoding_name (opt.locale))
1184         opt.locale = NULL;
1185
1186       if (!opt.locale)
1187         opt.locale = find_locale ();
1188
1189       if (opt.encoding_remote && !check_encoding_name (opt.encoding_remote))
1190         opt.encoding_remote = NULL;
1191     }
1192 #else
1193   memset (&dummy_iri, 0, sizeof (dummy_iri));
1194   if (opt.enable_iri || opt.locale || opt.encoding_remote)
1195     {
1196       /* sXXXav : be more specific... */
1197       fprintf (stderr, _("This version does not have support for IRIs\n"));
1198       exit(1);
1199     }
1200 #endif
1201
1202   if (opt.ask_passwd)
1203     {
1204       opt.passwd = prompt_for_password ();
1205
1206       if (opt.passwd == NULL || opt.passwd[0] == '\0')
1207         exit (1);
1208     }
1209
1210 #ifdef USE_WATT32
1211   if (opt.wdebug)
1212      dbug_init();
1213   sock_init();
1214 #else
1215   if (opt.background)
1216     fork_to_background ();
1217 #endif
1218
1219   /* Initialize progress.  Have to do this after the options are
1220      processed so we know where the log file is.  */
1221   if (opt.verbose)
1222     set_progress_implementation (opt.progress_type);
1223
1224   /* Fill in the arguments.  */
1225   url = alloca_array (char *, nurl + 1);
1226   for (i = 0; i < nurl; i++, optind++)
1227     {
1228       char *rewritten = rewrite_shorthand_url (argv[optind]);
1229       if (rewritten)
1230         url[i] = rewritten;
1231       else
1232         url[i] = xstrdup (argv[optind]);
1233     }
1234   url[i] = NULL;
1235
1236   /* Initialize logging.  */
1237   log_init (opt.lfilename, append_to_log);
1238
1239   DEBUGP (("DEBUG output created by Wget %s on %s.\n\n",
1240            version_string, OS_TYPE));
1241
1242   /* Open the output filename if necessary.  */
1243
1244 /* 2005-04-17 SMS.
1245    Note that having the output_stream ("-O") file opened here for an FTP
1246    URL rather than in getftp() (ftp.c) (and the http equivalent) rather
1247    limits the ability in VMS to open the file differently for ASCII
1248    versus binary FTP there.  (Of course, doing it here allows a open
1249    failure to be detected immediately, without first connecting to the
1250    server.)
1251 */
1252   if (opt.output_document)
1253     {
1254       if (HYPHENP (opt.output_document))
1255         {
1256 #ifdef WINDOWS
1257           _setmode (_fileno (stdout), _O_BINARY);
1258 #endif
1259           output_stream = stdout;
1260         }
1261       else
1262         {
1263           struct_fstat st;
1264
1265 #ifdef __VMS
1266 /* Common fopen() optional arguments:
1267    sequential access only, access callback function.
1268 */
1269 # define FOPEN_OPT_ARGS , "fop=sqo", "acc", acc_cb, &open_id
1270           int open_id = 7;
1271 #else /* def __VMS */
1272 # define FOPEN_OPT_ARGS
1273 #endif /* def __VMS [else] */
1274
1275           output_stream = fopen (opt.output_document,
1276                                  opt.always_rest ? "ab" : "wb"
1277                                  FOPEN_OPT_ARGS);
1278           if (output_stream == NULL)
1279             {
1280               perror (opt.output_document);
1281               exit (1);
1282             }
1283           if (fstat (fileno (output_stream), &st) == 0 && S_ISREG (st.st_mode))
1284             output_stream_regular = true;
1285         }
1286       if (!output_stream_regular && opt.convert_links)
1287         {
1288           fprintf (stderr, _("-k can be used together with -O only if \
1289 outputting to a regular file.\n"));
1290           print_usage (1);
1291           exit(1);
1292         }
1293     }
1294
1295 #ifdef __VMS
1296   /* Set global ODS5 flag according to the specified destination (if
1297      any), otherwise according to the current default device.
1298   */
1299   if (output_stream == NULL)
1300     set_ods5_dest( "SYS$DISK");
1301   else if (output_stream != stdout)
1302     set_ods5_dest( opt.output_document);
1303 #endif /* def __VMS */
1304
1305 #ifdef WINDOWS
1306   ws_startup ();
1307 #endif
1308
1309 #ifdef SIGHUP
1310   /* Setup the signal handler to redirect output when hangup is
1311      received.  */
1312   if (signal(SIGHUP, SIG_IGN) != SIG_IGN)
1313     signal(SIGHUP, redirect_output_signal);
1314 #endif
1315   /* ...and do the same for SIGUSR1.  */
1316 #ifdef SIGUSR1
1317   signal (SIGUSR1, redirect_output_signal);
1318 #endif
1319 #ifdef SIGPIPE
1320   /* Writing to a closed socket normally signals SIGPIPE, and the
1321      process exits.  What we want is to ignore SIGPIPE and just check
1322      for the return value of write().  */
1323   signal (SIGPIPE, SIG_IGN);
1324 #endif
1325 #ifdef SIGWINCH
1326   signal (SIGWINCH, progress_handle_sigwinch);
1327 #endif
1328
1329   /* Retrieve the URLs from argument list.  */
1330   for (t = url; *t; t++)
1331     {
1332       char *filename = NULL, *redirected_URL = NULL;
1333       int dt, url_err;
1334       /* Need to do a new struct iri every time, because
1335        * retrieve_url may modify it in some circumstances,
1336        * currently. */
1337       struct iri *iri = iri_new ();
1338       struct url *url_parsed;
1339
1340       set_uri_encoding (iri, opt.locale, true);
1341       url_parsed = url_parse (*t, &url_err, iri, true);
1342
1343       if (!url_parsed)
1344         {
1345           char *error = url_error (*t, url_err);
1346           logprintf (LOG_NOTQUIET, "%s: %s.\n",*t, error);
1347           xfree (error);
1348           inform_exit_status (URLERROR);
1349         }
1350       else
1351         {
1352           if ((opt.recursive || opt.page_requisites)
1353               && (url_scheme (*t) != SCHEME_FTP || url_uses_proxy (url_parsed)))
1354             {
1355               int old_follow_ftp = opt.follow_ftp;
1356
1357               /* Turn opt.follow_ftp on in case of recursive FTP retrieval */
1358               if (url_scheme (*t) == SCHEME_FTP)
1359                 opt.follow_ftp = 1;
1360
1361               retrieve_tree (url_parsed, NULL);
1362
1363               opt.follow_ftp = old_follow_ftp;
1364             }
1365           else
1366           {
1367             retrieve_url (url_parsed, *t, &filename, &redirected_URL, NULL,
1368                           &dt, opt.recursive, iri, true);
1369           }
1370
1371           if (opt.delete_after && file_exists_p(filename))
1372             {
1373               DEBUGP (("Removing file due to --delete-after in main():\n"));
1374               logprintf (LOG_VERBOSE, _("Removing %s.\n"), filename);
1375               if (unlink (filename))
1376                 logprintf (LOG_NOTQUIET, "unlink: %s\n", strerror (errno));
1377             }
1378           xfree_null (redirected_URL);
1379           xfree_null (filename);
1380           url_free (url_parsed);
1381         }
1382       iri_free (iri);
1383     }
1384
1385   /* And then from the input file, if any.  */
1386   if (opt.input_filename)
1387     {
1388       int count;
1389       int status;
1390       status = retrieve_from_file (opt.input_filename, opt.force_html, &count);
1391       inform_exit_status (status);
1392       if (!count)
1393         logprintf (LOG_NOTQUIET, _("No URLs found in %s.\n"),
1394                    opt.input_filename);
1395     }
1396
1397   /* Print broken links. */
1398   if (opt.recursive && opt.spider)
1399     print_broken_links ();
1400
1401   /* Print the downloaded sum.  */
1402   if ((opt.recursive || opt.page_requisites
1403        || nurl > 1
1404        || (opt.input_filename && total_downloaded_bytes != 0))
1405       &&
1406       total_downloaded_bytes != 0)
1407     {
1408       double end_time = ptimer_measure (timer);
1409       ptimer_destroy (timer);
1410
1411       char *wall_time = xstrdup (secs_to_human_time (end_time - start_time));
1412       char *download_time = xstrdup (secs_to_human_time (total_download_time));
1413       logprintf (LOG_NOTQUIET,
1414                  _("FINISHED --%s--\nTotal wall clock time: %s\n"
1415                    "Downloaded: %d files, %s in %s (%s)\n"),
1416                  datetime_str (time (NULL)),
1417                  wall_time,
1418                  numurls,
1419                  human_readable (total_downloaded_bytes),
1420                  download_time,
1421                  retr_rate (total_downloaded_bytes, total_download_time));
1422       xfree (wall_time);
1423       xfree (download_time);
1424
1425       /* Print quota warning, if exceeded.  */
1426       if (opt.quota && total_downloaded_bytes > opt.quota)
1427         logprintf (LOG_NOTQUIET,
1428                    _("Download quota of %s EXCEEDED!\n"),
1429                    human_readable (opt.quota));
1430     }
1431
1432   if (opt.cookies_output)
1433     save_cookies ();
1434
1435   if (opt.convert_links && !opt.delete_after)
1436     convert_all_links ();
1437
1438   log_close ();
1439   for (i = 0; i < nurl; i++)
1440     xfree (url[i]);
1441   cleanup ();
1442
1443   return get_exit_status ();
1444 }
1445 #endif /* TESTING */
1446 \f
1447 #if defined(SIGHUP) || defined(SIGUSR1)
1448
1449 /* So the signal_name check doesn't blow when only one is available. */
1450 #ifndef SIGHUP
1451 # define SIGHUP -1
1452 #endif
1453 #ifndef SIGUSR1
1454 # define SIGUSR1 -1
1455 #endif
1456
1457 /* Hangup signal handler.  When wget receives SIGHUP or SIGUSR1, it
1458    will proceed operation as usual, trying to write into a log file.
1459    If that is impossible, the output will be turned off.  */
1460
1461 static void
1462 redirect_output_signal (int sig)
1463 {
1464   const char *signal_name = (sig == SIGHUP ? "SIGHUP" :
1465                              (sig == SIGUSR1 ? "SIGUSR1" :
1466                               "WTF?!"));
1467   log_request_redirect_output (signal_name);
1468   progress_schedule_redirect ();
1469   signal (sig, redirect_output_signal);
1470 }
1471 #endif
1472
1473 /*
1474  * vim: et ts=2 sw=2
1475  */