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