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