Arrange to call close_stdout only upon exit.
[platform/upstream/coreutils.git] / src / mkdir.c
1 /* mkdir -- make directories
2    Copyright (C) 90, 1995-2000 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 "error.h"
27 #include "makepath.h"
28 #include "modechange.h"
29
30 #ifndef S_IRWXUGO
31 # define S_IRWXUGO (S_IRWXU | S_IRWXG | S_IRWXO)
32 #endif
33
34 /* The official name of this program (e.g., no `g' prefix).  */
35 #define PROGRAM_NAME "mkdir"
36
37 #define AUTHORS "David MacKenzie"
38
39 /* The name this program was run with. */
40 char *program_name;
41
42 /* If nonzero, ensure that all parents of the specified directory exist.  */
43 static int create_parents;
44
45 static struct option const longopts[] =
46 {
47   {"mode", required_argument, NULL, 'm'},
48   {"parents", no_argument, NULL, 'p'},
49   {"verbose", no_argument, NULL, 2},
50   {GETOPT_HELP_OPTION_DECL},
51   {GETOPT_VERSION_OPTION_DECL},
52   {NULL, 0, NULL, 0}
53 };
54
55 void
56 usage (int status)
57 {
58   if (status != 0)
59     fprintf (stderr, _("Try `%s --help' for more information.\n"),
60              program_name);
61   else
62     {
63       printf (_("Usage: %s [OPTION] DIRECTORY...\n"), program_name);
64       printf (_("\
65 Create the DIRECTORY(ies), if they do not already exist.\n\
66 \n\
67   -m, --mode=MODE   set permission mode (as in chmod), not rwxrwxrwx - umask\n\
68   -p, --parents     no error if existing, make parent directories as needed\n\
69       --verbose     print a message for each created directory\n\
70       --help        display this help and exit\n\
71       --version     output version information and exit\n\
72 "));
73       puts (_("\nReport bugs to <bug-fileutils@gnu.org>."));
74     }
75   exit (status);
76 }
77
78 int
79 main (int argc, char **argv)
80 {
81   mode_t newmode;
82   mode_t parent_mode;
83   const char *symbolic_mode = NULL;
84   const char *verbose_fmt_string = NULL;
85   int errors = 0;
86   int optc;
87
88   program_name = argv[0];
89   setlocale (LC_ALL, "");
90   bindtextdomain (PACKAGE, LOCALEDIR);
91   textdomain (PACKAGE);
92
93   atexit (close_stdout);
94
95   create_parents = 0;
96
97   while ((optc = getopt_long (argc, argv, "pm:", longopts, NULL)) != -1)
98     {
99       switch (optc)
100         {
101         case 0:                 /* Long option. */
102           break;
103         case 'p':
104           create_parents = 1;
105           break;
106         case 'm':
107           symbolic_mode = optarg;
108           break;
109         case 2: /* --verbose  */
110           verbose_fmt_string = _("created directory `%s'");
111           break;
112         case_GETOPT_HELP_CHAR;
113         case_GETOPT_VERSION_CHAR (PROGRAM_NAME, AUTHORS);
114         default:
115           usage (1);
116         }
117     }
118
119   if (optind == argc)
120     {
121       error (0, 0, _("too few arguments"));
122       usage (1);
123     }
124
125   newmode = S_IRWXUGO & ~ umask (0);
126   parent_mode = S_IWUSR | S_IXUSR | newmode;
127   if (symbolic_mode)
128     {
129       struct mode_change *change = mode_compile (symbolic_mode, 0);
130       if (change == MODE_INVALID)
131         error (1, 0, _("invalid mode `%s'"), symbolic_mode);
132       else if (change == MODE_MEMORY_EXHAUSTED)
133         error (1, 0, _("virtual memory exhausted"));
134       newmode = mode_adjust (newmode, change);
135     }
136
137   for (; optind < argc; ++optind)
138     {
139       int fail = 0;
140       if (create_parents)
141         {
142           fail = make_path (argv[optind], newmode, parent_mode,
143                             -1, -1, 1, verbose_fmt_string);
144         }
145       else
146         {
147           fail = mkdir (argv[optind], newmode);
148           if (fail)
149             error (0, errno, _("cannot create directory `%s'"), argv[optind]);
150           else if (verbose_fmt_string)
151             error (0, 0, verbose_fmt_string, argv[optind]);
152
153           /* mkdir(2) is required to honor only the file permission bits.
154              In particular, it needn't do anything about `special' bits,
155              so if any were set in newmode, apply them with chmod.  */
156           if (fail == 0 && (newmode & ~S_IRWXUGO))
157             {
158               fail = chmod (argv[optind], newmode);
159               if (fail)
160                 error (0, errno, _("cannot set permissions of directory `%s'"),
161                        argv[optind]);
162             }
163         }
164
165       errors |= fail;
166     }
167
168   exit (errors);
169 }