(main, usage): Check for write error to stdout before exiting.
[platform/upstream/coreutils.git] / src / mkdir.c
1 /* mkdir -- make directories
2    Copyright (C) 90, 95, 96, 97, 1998 Free Software Foundation, Inc.
3
4    This program is free software; you can redistribute it and/or modify
5    it under the terms of the GNU General Public License as published by
6    the Free Software Foundation; either version 2, or (at your option)
7    any later version.
8
9    This program is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12    GNU General Public License for more details.
13
14    You should have received a copy of the GNU General Public License
15    along with this program; if not, write to the Free Software Foundation,
16    Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
17
18 /* David MacKenzie <djm@ai.mit.edu>  */
19
20 #include <config.h>
21 #include <stdio.h>
22 #include <getopt.h>
23 #include <sys/types.h>
24
25 #include "system.h"
26 #include "modechange.h"
27 #include "makepath.h"
28 #include "closeout.h"
29 #include "error.h"
30
31 /* The name this program was run with. */
32 char *program_name;
33
34 /* If nonzero, ensure that all parents of the specified directory exist.  */
35 static int path_mode;
36
37 /* If nonzero, display usage information and exit.  */
38 static int show_help;
39
40 /* If nonzero, print the version on standard output and exit.  */
41 static int show_version;
42
43 static struct option const longopts[] =
44 {
45   {"mode", required_argument, NULL, 'm'},
46   {"parents", no_argument, &path_mode, 1},
47   {"help", no_argument, &show_help, 1},
48   {"verbose", no_argument, NULL, 2},
49   {"version", no_argument, &show_version, 1},
50   {NULL, 0, NULL, 0}
51 };
52
53 static void
54 usage (int status)
55 {
56   if (status != 0)
57     fprintf (stderr, _("Try `%s --help' for more information.\n"),
58              program_name);
59   else
60     {
61       printf (_("Usage: %s [OPTION] DIRECTORY...\n"), program_name);
62       printf (_("\
63 Create the DIRECTORY(ies), if they do not already exist.\n\
64 \n\
65   -m, --mode=MODE   set permission mode (as in chmod), not rwxrwxrwx - umask\n\
66   -p, --parents     no error if existing, make parent directories as needed\n\
67       --verbose     print a message for each created directory\n\
68       --help        display this help and exit\n\
69       --version     output version information and exit\n\
70 "));
71       puts (_("\nReport bugs to <fileutils-bugs@gnu.org>."));
72       close_stdout ();
73     }
74   exit (status);
75 }
76
77 int
78 main (int argc, char **argv)
79 {
80   unsigned int newmode;
81   unsigned int parent_mode;
82   char *symbolic_mode = NULL;
83   const char *verbose_fmt_string = NULL;
84   int errors = 0;
85   int optc;
86
87   program_name = argv[0];
88   setlocale (LC_ALL, "");
89   bindtextdomain (PACKAGE, LOCALEDIR);
90   textdomain (PACKAGE);
91
92   path_mode = 0;
93
94   while ((optc = getopt_long (argc, argv, "pm:", longopts, NULL)) != -1)
95     {
96       switch (optc)
97         {
98         case 0:                 /* Long option. */
99           break;
100         case 'p':
101           path_mode = 1;
102           break;
103         case 'm':
104           symbolic_mode = optarg;
105           break;
106         case 2: /* --verbose  */
107           verbose_fmt_string = _("created directory `%s'");
108           break;
109         default:
110           usage (1);
111         }
112     }
113
114   if (show_version)
115     {
116       printf ("mkdir (%s) %s\n", GNU_PACKAGE, VERSION);
117       close_stdout ();
118       exit (0);
119     }
120
121   if (show_help)
122     usage (0);
123
124   if (optind == argc)
125     {
126       error (0, 0, _("too few arguments"));
127       usage (1);
128     }
129
130   newmode = 0777 & ~umask (0);
131   parent_mode = newmode | 0300; /* u+wx */
132   if (symbolic_mode)
133     {
134       struct mode_change *change = mode_compile (symbolic_mode, 0);
135       if (change == MODE_INVALID)
136         error (1, 0, _("invalid mode `%s'"), symbolic_mode);
137       else if (change == MODE_MEMORY_EXHAUSTED)
138         error (1, 0, _("virtual memory exhausted"));
139       newmode = mode_adjust (newmode, change);
140     }
141
142   for (; optind < argc; ++optind)
143     {
144       if (path_mode)
145         {
146           errors |= make_path (argv[optind], newmode, parent_mode,
147                                -1, -1, 1, verbose_fmt_string);
148         }
149       else if (mkdir (argv[optind], newmode))
150         {
151           error (0, errno, _("cannot make directory `%s'"), argv[optind]);
152           errors = 1;
153         }
154       else if (verbose_fmt_string)
155         {
156           error (0, 0, verbose_fmt_string, argv[optind]);
157         }
158     }
159
160   exit (errors);
161 }