1 /*
2 * Copyright (c) 1983, 1992, 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the University nor the names of its contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 *
29 * @(#) Copyright (c) 1983, 1992, 1993 The Regents of the University of California. All rights reserved.
30 * @(#)mkdir.c 8.2 (Berkeley) 1/25/94
31 * $FreeBSD: src/bin/mkdir/mkdir.c,v 1.19.2.2 2001/08/01 04:42:37 obrien Exp $
32 */
33
34 #include <sys/types.h>
35 #include <sys/stat.h>
36
37 #include <err.h>
38 #include <errno.h>
39 #include <libgen.h>
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <string.h>
43 #include <sysexits.h>
44 #include <unistd.h>
45
46 static int build(char *, mode_t);
47 static int mkdir_race(const char *path, int nmode);
48 static void usage(void);
49
50 static int vflag;
51
52 int
main(int argc,char ** argv)53 main(int argc, char **argv)
54 {
55 int ch, exitval, success, pflag;
56 void *set;
57 mode_t omode;
58 char *mode;
59
60 pflag = 0;
61 mode = NULL;
62 while ((ch = getopt(argc, argv, "m:pv")) != -1)
63 switch(ch) {
64 case 'm':
65 mode = optarg;
66 break;
67 case 'p':
68 pflag = 1;
69 break;
70 case 'v':
71 vflag = 1;
72 break;
73 default:
74 usage();
75 }
76
77 argc -= optind;
78 argv += optind;
79 if (argv[0] == NULL)
80 usage();
81
82 if (mode == NULL) {
83 omode = S_IRWXU | S_IRWXG | S_IRWXO;
84 } else {
85 errno = 0;
86 if ((set = setmode(mode)) == NULL) {
87 if (!errno)
88 errx(1, "invalid file mode: %s", mode);
89 else
90 /* A malloc(3) error, not a invaild file mode. */
91 err(1, "setmode failed");
92 }
93 omode = getmode(set, S_IRWXU | S_IRWXG | S_IRWXO);
94 free(set);
95 }
96
97 for (exitval = 0; *argv != NULL; ++argv) {
98 success = 1;
99 if (pflag) {
100 if (build(*argv, omode))
101 success = 0;
102 } else if (mkdir(*argv, omode) < 0) {
103 if (errno == ENOTDIR || errno == ENOENT)
104 warn("%s", dirname(*argv));
105 else
106 warn("%s", *argv);
107 success = 0;
108 } else if (vflag)
109 printf("%s\n", *argv);
110
111 if (!success)
112 exitval = 1;
113 /*
114 * The mkdir() and umask() calls both honor only the low
115 * nine bits, so if you try to set a mode including the
116 * sticky, setuid, setgid bits you lose them. Don't do
117 * this unless the user has specifically requested a mode,
118 * as chmod will (obviously) ignore the umask.
119 */
120 if (success && mode != NULL && chmod(*argv, omode) == -1) {
121 warn("%s", *argv);
122 exitval = 1;
123 }
124 }
125 exit(exitval);
126 }
127
128 static int
build(char * path,mode_t omode)129 build(char *path, mode_t omode)
130 {
131 struct stat sb;
132 mode_t numask, oumask;
133 int first, last, retval;
134 char *p;
135
136 p = path;
137 oumask = 0;
138 retval = 0;
139 if (p[0] == '/') /* Skip leading '/'. */
140 ++p;
141 for (first = 1, last = 0; !last ; ++p) {
142 if (p[0] == '\0')
143 last = 1;
144 else if (p[0] != '/')
145 continue;
146 *p = '\0';
147 if (!last && p[1] == '\0')
148 last = 1;
149 if (first) {
150 /*
151 * POSIX 1003.2:
152 * For each dir operand that does not name an existing
153 * directory, effects equivalent to those cased by the
154 * following command shall occcur:
155 *
156 * mkdir -p -m $(umask -S),u+wx $(dirname dir) &&
157 * mkdir [-m mode] dir
158 *
159 * We change the user's umask and then restore it,
160 * instead of doing chmod's.
161 */
162 oumask = umask(0);
163 numask = oumask & ~(S_IWUSR | S_IXUSR);
164 umask(numask);
165 first = 0;
166 }
167 if (last)
168 umask(oumask);
169 if (stat(path, &sb)) {
170 if (errno != ENOENT ||
171 mkdir_race(path, last ? omode :
172 S_IRWXU | S_IRWXG | S_IRWXO) < 0) {
173 warn("%s", path);
174 retval = 1;
175 break;
176 } else if (vflag)
177 printf("%s\n", path);
178 }
179 else if ((sb.st_mode & S_IFMT) != S_IFDIR) {
180 if (last)
181 errno = EEXIST;
182 else
183 errno = ENOTDIR;
184 warn("%s", path);
185 retval = 1;
186 break;
187 }
188 if (!last)
189 *p = '/';
190 }
191 if (!first && !last)
192 umask(oumask);
193 return (retval);
194 }
195
196 int
mkdir_race(const char * path,int nmode)197 mkdir_race(const char *path, int nmode)
198 {
199 int res;
200 struct stat sb;
201
202 res = mkdir(path, nmode);
203 if (res < 0 && errno == EEXIST) {
204 if (stat(path, &sb) == 0 && S_ISDIR(sb.st_mode))
205 return(0);
206 res = mkdir(path, nmode);
207 }
208 return (res);
209 }
210
211 static void
usage(void)212 usage(void)
213 {
214
215 fprintf(stderr, "usage: mkdir [-pv] [-m mode] directory ...\n");
216 exit (EX_USAGE);
217 }
218