xref: /netbsd/bin/mkdir/mkdir.c (revision c4a72b64)
1 /* $NetBSD: mkdir.c,v 1.27 2002/11/24 23:40:07 chs Exp $ */
2 
3 /*
4  * Copyright (c) 1983, 1992, 1993
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. All advertising materials mentioning features or use of this software
16  *    must display the following acknowledgement:
17  *	This product includes software developed by the University of
18  *	California, Berkeley and its contributors.
19  * 4. Neither the name of the University nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  */
35 
36 #include <sys/cdefs.h>
37 #ifndef lint
38 __COPYRIGHT("@(#) Copyright (c) 1983, 1992, 1993\n\
39 	The Regents of the University of California.  All rights reserved.\n");
40 #endif /* not lint */
41 
42 #ifndef lint
43 #if 0
44 static char sccsid[] = "@(#)mkdir.c	8.2 (Berkeley) 1/25/94";
45 #else
46 __RCSID("$NetBSD: mkdir.c,v 1.27 2002/11/24 23:40:07 chs Exp $");
47 #endif
48 #endif /* not lint */
49 
50 #include <sys/types.h>
51 #include <sys/stat.h>
52 
53 #include <err.h>
54 #include <errno.h>
55 #include <locale.h>
56 #include <stdio.h>
57 #include <stdlib.h>
58 #include <string.h>
59 #include <unistd.h>
60 
61 int	mkpath(char *, mode_t, mode_t);
62 void	usage(void);
63 int	main(int, char *[]);
64 
65 int
66 main(int argc, char *argv[])
67 {
68 	int ch, exitval, pflag;
69 	void *set;
70 	mode_t mode, dir_mode;
71 
72 	setprogname(argv[0]);
73 	(void)setlocale(LC_ALL, "");
74 
75 	/*
76 	 * The default file mode is a=rwx (0777) with selected permissions
77 	 * removed in accordance with the file mode creation mask.  For
78 	 * intermediate path name components, the mode is the default modified
79 	 * by u+wx so that the subdirectories can always be created.
80 	 */
81 	mode = (S_IRWXU | S_IRWXG | S_IRWXO) & ~umask(0);
82 	dir_mode = mode | S_IWUSR | S_IXUSR;
83 
84 	pflag = 0;
85 	while ((ch = getopt(argc, argv, "m:p")) != -1)
86 		switch (ch) {
87 		case 'p':
88 			pflag = 1;
89 			break;
90 		case 'm':
91 			if ((set = setmode(optarg)) == NULL)
92 				errx(1, "invalid file mode: %s", optarg);
93 			mode = getmode(set, S_IRWXU | S_IRWXG | S_IRWXO);
94 			free(set);
95 			break;
96 		case '?':
97 		default:
98 			usage();
99 		}
100 	argc -= optind;
101 	argv += optind;
102 
103 	if (*argv == NULL)
104 		usage();
105 
106 	for (exitval = EXIT_SUCCESS; *argv != NULL; ++argv) {
107 		char *slash;
108 
109 		/* Remove trailing slashes, per POSIX. */
110 		slash = strrchr(*argv, '\0');
111 		while (--slash > *argv && *slash == '/')
112 			*slash = '\0';
113 
114 		if (pflag) {
115 			if (mkpath(*argv, mode, dir_mode) < 0)
116 				exitval = EXIT_FAILURE;
117 		} else {
118 			if (mkdir(*argv, mode) < 0) {
119 				warn("%s", *argv);
120 				exitval = EXIT_FAILURE;
121 			} else {
122 				/*
123 				 * The mkdir() and umask() calls both honor
124 				 * only the file permission bits, so if you try
125 				 * to set a mode including the sticky, setuid,
126 				 * setgid bits you lose them. So chmod().
127 				 */
128 				if ((mode & ~(S_IRWXU|S_IRWXG|S_IRWXO)) != 0 &&
129 				    chmod(*argv, mode) == -1) {
130 					warn("%s", *argv);
131 					exitval = EXIT_FAILURE;
132 				}
133 			}
134 		}
135 	}
136 	exit(exitval);
137 	/* NOTREACHED */
138 }
139 
140 /*
141  * mkpath -- create directories.
142  *	path     - path
143  *	mode     - file mode of terminal directory
144  *	dir_mode - file mode of intermediate directories
145  */
146 int
147 mkpath(char *path, mode_t mode, mode_t dir_mode)
148 {
149 	struct stat sb;
150 	char *slash;
151 	int done, rv;
152 
153 	done = 0;
154 	slash = path;
155 
156 	for (;;) {
157 		slash += strspn(slash, "/");
158 		slash += strcspn(slash, "/");
159 
160 		done = (*slash == '\0');
161 		*slash = '\0';
162 
163 		rv = mkdir(path, done ? mode : dir_mode);
164 		if (rv < 0 && errno != EEXIST) {
165 			warn("%s", path);
166 			return (-1);
167 		}
168 		if (done) {
169 			break;
170 		}
171 		*slash = '/';
172 	}
173 
174 	/*
175 	 * Check for the final component being something other than
176 	 * a directory.
177 	 */
178 
179 	if (rv < 0) {
180 		if (stat(path, &sb) < 0) {
181 			warn("stat %s failed", path);
182 			return (-1);
183 		}
184 		if (!S_ISDIR(sb.st_mode)) {
185 			errno = ENOTDIR;
186 			warn("%s", path);
187 			return (-1);
188 		}
189 	}
190 
191 	/*
192 	 * The mkdir() and umask() calls both honor only the
193 	 * file permission bits, so if you try to set a mode
194 	 * including the sticky, setuid, setgid bits you lose
195 	 * them. So chmod().
196 	 */
197 
198 	if ((mode & ~(S_IRWXU|S_IRWXG|S_IRWXU)) != 0 &&
199 	    chmod(path, mode) == -1) {
200 		warn("%s", path);
201 		return (-1);
202 	}
203 	return (0);
204 }
205 
206 void
207 usage(void)
208 {
209 
210 	(void)fprintf(stderr, "usage: %s [-p] [-m mode] dirname ...\n",
211 	    getprogname());
212 	exit(EXIT_FAILURE);
213 	/* NOTREACHED */
214 }
215