busybox/libbb/make_directory.c
<<
>>
Prefs
   1/* vi: set sw=4 ts=4: */
   2/*
   3 * parse_mode implementation for busybox
   4 *
   5 * Copyright (C) 2003  Manuel Novoa III  <mjn3@codepoet.org>
   6 *
   7 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
   8 */
   9
  10/* Mar 5, 2003    Manuel Novoa III
  11 *
  12 * This is the main work function for the 'mkdir' applet.  As such, it
  13 * strives to be SUSv3 compliant in it's behaviour when recursively
  14 * making missing parent dirs, and in it's mode setting of the final
  15 * directory 'path'.
  16 *
  17 * To recursively build all missing intermediate directories, make
  18 * sure that (flags & FILEUTILS_RECUR) is non-zero.  Newly created
  19 * intermediate directories will have at least u+wx perms.
  20 *
  21 * To set specific permissions on 'path', pass the appropriate 'mode'
  22 * val.  Otherwise, pass -1 to get default permissions.
  23 */
  24
  25#include "libbb.h"
  26
  27/* This function is used from NOFORK applets. It must not allocate anything */
  28
  29int FAST_FUNC bb_make_directory(char *path, long mode, int flags)
  30{
  31        mode_t cur_mask;
  32        mode_t org_mask;
  33        const char *fail_msg;
  34        char *s;
  35        char c;
  36        struct stat st;
  37
  38        /* Happens on bb_make_directory(dirname("no_slashes"),...) */
  39        if (LONE_CHAR(path, '.'))
  40                return 0;
  41
  42        org_mask = cur_mask = (mode_t)-1L;
  43        s = path;
  44        while (1) {
  45                c = '\0';
  46
  47                if (flags & FILEUTILS_RECUR) {  /* Get the parent */
  48                        /* Bypass leading non-'/'s and then subsequent '/'s */
  49                        while (*s) {
  50                                if (*s == '/') {
  51                                        do {
  52                                                ++s;
  53                                        } while (*s == '/');
  54                                        c = *s; /* Save the current char */
  55                                        *s = '\0'; /* and replace it with nul */
  56                                        break;
  57                                }
  58                                ++s;
  59                        }
  60                }
  61
  62                if (c != '\0') {
  63                        /* Intermediate dirs: must have wx for user */
  64                        if (cur_mask == (mode_t)-1L) { /* wasn't done yet? */
  65                                mode_t new_mask;
  66                                org_mask = umask(0);
  67                                cur_mask = 0;
  68                                /* Clear u=wx in umask - this ensures
  69                                 * they won't be cleared on mkdir */
  70                                new_mask = (org_mask & ~(mode_t)0300);
  71                                //bb_error_msg("org_mask:%o cur_mask:%o", org_mask, new_mask);
  72                                if (new_mask != cur_mask) {
  73                                        cur_mask = new_mask;
  74                                        umask(new_mask);
  75                                }
  76                        }
  77                } else {
  78                        /* Last component: uses original umask */
  79                        //bb_error_msg("1 org_mask:%o", org_mask);
  80                        if (org_mask != cur_mask) {
  81                                cur_mask = org_mask;
  82                                umask(org_mask);
  83                        }
  84                }
  85
  86                if (mkdir(path, 0777) < 0) {
  87                        /* If we failed for any other reason than the directory
  88                         * already exists, output a diagnostic and return -1 */
  89                        if (errno != EEXIST
  90                         || !(flags & FILEUTILS_RECUR)
  91                         || ((stat(path, &st) < 0) || !S_ISDIR(st.st_mode))
  92                        ) {
  93                                fail_msg = "create";
  94                                break;
  95                        }
  96                        /* Since the directory exists, don't attempt to change
  97                         * permissions if it was the full target.  Note that
  98                         * this is not an error condition. */
  99                        if (!c) {
 100                                goto ret0;
 101                        }
 102                }
 103
 104                if (!c) {
 105                        /* Done.  If necessary, update perms on the newly
 106                         * created directory.  Failure to update here _is_
 107                         * an error. */
 108                        if ((mode != -1) && (chmod(path, mode) < 0)) {
 109                                fail_msg = "set permissions of";
 110                                break;
 111                        }
 112                        goto ret0;
 113                }
 114
 115                /* Remove any inserted nul from the path (recursive mode) */
 116                *s = c;
 117        } /* while (1) */
 118
 119        bb_perror_msg("can't %s directory '%s'", fail_msg, path);
 120        flags = -1;
 121        goto ret;
 122 ret0:
 123        flags = 0;
 124 ret:
 125        //bb_error_msg("2 org_mask:%o", org_mask);
 126        if (org_mask != cur_mask)
 127                umask(org_mask);
 128        return flags;
 129}
 130