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