src/basic/mkdir.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2010 Lennart Poettering
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   Lesser General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 #include <errno.h>
  21 #include <stdbool.h>
  22 #include <string.h>
  23 #include <sys/stat.h>
  24
  25 #include "fs-util.h"
  26 #include "macro.h"
  27 #include "missing.h"
  28 #include "mkdir.h"
  29 #include "path-util.h"
  30 #include "stat-util.h"
  31 #include "user-util.h"
  32
  33 int mkdir_safe_internal(const char *path, mode_t mode, uid_t uid, gid_t gid, mkdir_func_t _mkdir) {
  34         struct stat st;
  35
  36         if (_mkdir(path, mode) >= 0)
  37                 if (chmod_and_chown(path, mode, uid, gid) < 0)
  38                         return -errno;
  39
  40         if (lstat(path, &st) < 0)
  41                 return -errno;
  42
  43         if ((st.st_mode & 0007) > (mode & 0007) ||
  44             (st.st_mode & 0070) > (mode & 0070) ||
  45             (st.st_mode & 0700) > (mode & 0700) ||
  46             (uid != UID_INVALID && st.st_uid != uid) ||
  47             (gid != GID_INVALID && st.st_gid != gid) ||
  48             !S_ISDIR(st.st_mode))
  49                 return -EEXIST;
  50
  51         return 0;
  52 }
  53
  54 int mkdir_safe(const char *path, mode_t mode, uid_t uid, gid_t gid) {
  55         return mkdir_safe_internal(path, mode, uid, gid, mkdir);
  56 }
  57
  58 int mkdir_parents_internal(const char *prefix, const char *path, mode_t mode, mkdir_func_t _mkdir) {
  59         const char *p, *e;
  60         int r;
  61
  62         assert(path);
  63
  64         if (prefix && !path_startswith(path, prefix))
  65                 return -ENOTDIR;
  66
  67         /* return immediately if directory exists */
  68         e = strrchr(path, '/');
  69         if (!e)
  70                 return -EINVAL;
  71
  72         if (e == path)
  73                 return 0;
  74
  75         p = strndupa(path, e - path);
  76         r = is_dir(p, true);
  77         if (r > 0)
  78                 return 0;
  79         if (r == 0)
  80                 return -ENOTDIR;
  81
  82         /* create every parent directory in the path, except the last component */
  83         p = path + strspn(path, "/");
  84         for (;;) {
  85                 char t[strlen(path) + 1];
  86
  87                 e = p + strcspn(p, "/");
  88                 p = e + strspn(e, "/");
  89
  90                 /* Is this the last component? If so, then we're
  91                  * done */
  92                 if (*p == 0)
  93                         return 0;
  94
  95                 memcpy(t, path, e - path);
  96                 t[e-path] = 0;
  97
  98                 if (prefix && path_startswith(prefix, t))
  99                         continue;
 100
 101                 r = _mkdir(t, mode);
 102                 if (r < 0 && errno != EEXIST)
 103                         return -errno;
 104         }
 105 }
 106
 107 int mkdir_parents(const char *path, mode_t mode) {
 108         return mkdir_parents_internal(NULL, path, mode, mkdir);
 109 }
 110
 111 int mkdir_p_internal(const char *prefix, const char *path, mode_t mode, mkdir_func_t _mkdir) {
 112         int r;
 113
 114         /* Like mkdir -p */
 115
 116         r = mkdir_parents_internal(prefix, path, mode, _mkdir);
 117         if (r < 0)
 118                 return r;
 119
 120         r = _mkdir(path, mode);
 121         if (r < 0 && (errno != EEXIST || is_dir(path, true) <= 0))
 122                 return -errno;
 123
 124         return 0;
 125 }
 126
 127 int mkdir_p(const char *path, mode_t mode) {
 128         return mkdir_p_internal(NULL, path, mode, mkdir);
 129 }