xref: /dragonfly/bin/mkdir/mkdir.c (revision 71126e33)
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 void	usage(void);
53 
54 int vflag;
55 
56 int
57 main(int argc, char **argv)
58 {
59 	int ch, exitval, success, pflag;
60 	void *set;
61 	mode_t omode;
62 	char *mode;
63 
64 	omode = pflag = 0;
65 	mode = NULL;
66 	while ((ch = getopt(argc, argv, "m:pv")) != -1)
67 		switch(ch) {
68 		case 'm':
69 			mode = optarg;
70 			break;
71 		case 'p':
72 			pflag = 1;
73 			break;
74 		case 'v':
75 			vflag = 1;
76 			break;
77 		default:
78 			usage();
79 		}
80 
81 	argc -= optind;
82 	argv += optind;
83 	if (argv[0] == NULL)
84 		usage();
85 
86 	if (mode == NULL) {
87 		omode = S_IRWXU | S_IRWXG | S_IRWXO;
88 	} else {
89 		errno = 0;
90 		if ((set = setmode(mode)) == NULL) {
91 			if (!errno)
92 				errx(1, "invalid file mode: %s", mode);
93 			else
94 				/* A malloc(3) error, not a invaild file mode. */
95 				err(1, "setmode failed");
96 		}
97 		omode = getmode(set, S_IRWXU | S_IRWXG | S_IRWXO);
98 		free(set);
99 	}
100 
101 	for (exitval = 0; *argv != NULL; ++argv) {
102 		success = 1;
103 		if (pflag) {
104 			if (build(*argv, omode))
105 				success = 0;
106 		} else if (mkdir(*argv, omode) < 0) {
107 			if (errno == ENOTDIR || errno == ENOENT)
108 				warn("%s", dirname(*argv));
109 			else
110 				warn("%s", *argv);
111 			success = 0;
112 		} else if (vflag)
113 			printf("%s\n", *argv);
114 
115 		if (!success)
116 			exitval = 1;
117 		/*
118 		 * The mkdir() and umask() calls both honor only the low
119 		 * nine bits, so if you try to set a mode including the
120 		 * sticky, setuid, setgid bits you lose them.  Don't do
121 		 * this unless the user has specifically requested a mode,
122 		 * as chmod will (obviously) ignore the umask.
123 		 */
124 		if (success && mode != NULL && chmod(*argv, omode) == -1) {
125 			warn("%s", *argv);
126 			exitval = 1;
127 		}
128 	}
129 	exit(exitval);
130 }
131 
132 static int
133 build(char *path, mode_t omode)
134 {
135 	struct stat sb;
136 	mode_t numask, oumask;
137 	int first, last, retval;
138 	char *p;
139 
140 	p = path;
141 	oumask = 0;
142 	retval = 0;
143 	if (p[0] == '/')		/* Skip leading '/'. */
144 		++p;
145 	for (first = 1, last = 0; !last ; ++p) {
146 		if (p[0] == '\0')
147 			last = 1;
148 		else if (p[0] != '/')
149 			continue;
150 		*p = '\0';
151 		if (p[1] == '\0')
152 			last = 1;
153 		if (first) {
154 			/*
155 			 * POSIX 1003.2:
156 			 * For each dir operand that does not name an existing
157 			 * directory, effects equivalent to those cased by the
158 			 * following command shall occcur:
159 			 *
160 			 * mkdir -p -m $(umask -S),u+wx $(dirname dir) &&
161 			 *    mkdir [-m mode] dir
162 			 *
163 			 * We change the user's umask and then restore it,
164 			 * instead of doing chmod's.
165 			 */
166 			oumask = umask(0);
167 			numask = oumask & ~(S_IWUSR | S_IXUSR);
168 			umask(numask);
169 			first = 0;
170 		}
171 		if (last)
172 			umask(oumask);
173 		if (stat(path, &sb)) {
174 			if (errno != ENOENT ||
175 			    mkdir(path, last ? omode :
176 				  S_IRWXU | S_IRWXG | S_IRWXO) < 0) {
177 				warn("%s", path);
178 				retval = 1;
179 				break;
180 			} else if (vflag)
181 				printf("%s\n", path);
182 		}
183 		else if ((sb.st_mode & S_IFMT) != S_IFDIR) {
184 			if (last)
185 				errno = EEXIST;
186 			else
187 				errno = ENOTDIR;
188 			warn("%s", path);
189 			retval = 1;
190 			break;
191 		}
192 		if (!last)
193 		    *p = '/';
194 	}
195 	if (!first && !last)
196 		umask(oumask);
197 	return (retval);
198 }
199 
200 static void
201 usage(void)
202 {
203 
204 	fprintf(stderr, "usage: mkdir [-pv] [-m mode] directory ...\n");
205 	exit (EX_USAGE);
206 }
207