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