lib/util/util_paths.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Andrew Tridgell 1992-1998
   5    Copyright (C) Jeremy Allison 2001-2007
   6    Copyright (C) Simo Sorce 2001
   7    Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2003
   8    Copyright (C) James Peach 2006
   9    Copyright (c) 2020      Andreas Schneider <asn@samba.org>
  10
  11    This program is free software; you can redistribute it and/or modify
  12    it under the terms of the GNU General Public License as published by
  13    the Free Software Foundation; either version 3 of the License, or
  14    (at your option) any later version.
  15
  16    This program is distributed in the hope that it will be useful,
  17    but WITHOUT ANY WARRANTY; without even the implied warranty of
  18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19    GNU General Public License for more details.
  20
  21    You should have received a copy of the GNU General Public License
  22    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  23 */
  24
  25 #include "replace.h"
  26 #include "dynconfig/dynconfig.h"
  27 #include "lib/util/util_paths.h"
  28 #include "system/passwd.h"
  29 #include "system/filesys.h"
  30
  31 /**
  32  * @brief Returns an absolute path to a file in the Samba modules directory.
  33  *
  34  * @param name File to find, relative to MODULESDIR.
  35  *
  36  * @retval Pointer to a string containing the full path.
  37  **/
  38
  39 char *modules_path(TALLOC_CTX *mem_ctx, const char *name)
  40 {
  41         return talloc_asprintf(mem_ctx, "%s/%s", get_dyn_MODULESDIR(), name);
  42 }
  43
  44 /**
  45  * @brief Returns an absolute path to a file in the Samba data directory.
  46  *
  47  * @param name File to find, relative to CODEPAGEDIR.
  48  *
  49  * @retval Pointer to a talloc'ed string containing the full path.
  50  **/
  51
  52 char *data_path(TALLOC_CTX *mem_ctx, const char *name)
  53 {
  54         return talloc_asprintf(mem_ctx, "%s/%s", get_dyn_CODEPAGEDIR(), name);
  55 }
  56
  57 /**
  58  * @brief Returns the platform specific shared library extension.
  59  *
  60  * @retval Pointer to a const char * containing the extension.
  61  **/
  62
  63 const char *shlib_ext(void)
  64 {
  65         return get_dyn_SHLIBEXT();
  66 }
  67
  68 static char *get_user_home_dir(TALLOC_CTX *mem_ctx)
  69 {
  70         struct passwd pwd = {0};
  71         struct passwd *pwdbuf = NULL;
  72         char *buf = NULL;
  73         char *out = NULL;
  74         long int initlen;
  75         size_t len;
  76         int rc;
  77
  78         initlen = sysconf(_SC_GETPW_R_SIZE_MAX);
  79         if (initlen == -1) {
  80                 len = 1024;
  81         } else {
  82                 len = (size_t)initlen;
  83         }
  84         buf = talloc_size(mem_ctx, len);
  85         if (buf == NULL) {
  86                 return NULL;
  87         }
  88
  89         rc = getpwuid_r(getuid(), &pwd, buf, len, &pwdbuf);
  90         while (rc == ERANGE) {
  91                 size_t newlen = 2 * len;
  92                 if (newlen < len) {
  93                         /* Overflow */
  94                         goto done;
  95                 }
  96                 len = newlen;
  97                 buf = talloc_realloc_size(mem_ctx, buf, len);
  98                 if (buf == NULL) {
  99                         goto done;
 100                 }
 101                 rc = getpwuid_r(getuid(), &pwd, buf, len, &pwdbuf);
 102         }
 103         if (rc != 0 || pwdbuf == NULL ) {
 104                 const char *szPath = getenv("HOME");
 105                 if (szPath == NULL) {
 106                         goto done;
 107                 }
 108                 len = strnlen(szPath, PATH_MAX);
 109                 if (len >= PATH_MAX) {
 110                         goto done;
 111                 }
 112                 out = talloc_strdup(mem_ctx, szPath);
 113                 goto done;
 114         }
 115
 116         out = talloc_strdup(mem_ctx, pwd.pw_dir);
 117 done:
 118         TALLOC_FREE(buf);
 119         return out;
 120 }
 121
 122 char *path_expand_tilde(TALLOC_CTX *mem_ctx, const char *d)
 123 {
 124         char *h = NULL, *r = NULL;
 125         const char *p = NULL;
 126         struct stat sb = {0};
 127         int rc;
 128
 129         if (d[0] != '~') {
 130                 return talloc_strdup(mem_ctx, d);
 131         }
 132         d++;
 133
 134         /* handle ~user/path */
 135         p = strchr(d, '/');
 136         if (p != NULL && p > d) {
 137                 struct passwd *pw;
 138                 size_t s = p - d;
 139                 char u[128];
 140
 141                 if (s >= sizeof(u)) {
 142                         return NULL;
 143                 }
 144                 memcpy(u, d, s);
 145                 u[s] = '\0';
 146
 147                 pw = getpwnam(u);
 148                 if (pw == NULL) {
 149                         return NULL;
 150                 }
 151                 h = talloc_strdup(mem_ctx, pw->pw_dir);
 152         } else {
 153                 p = d;
 154                 h = get_user_home_dir(mem_ctx);
 155         }
 156         if (h == NULL) {
 157                 return NULL;
 158         }
 159
 160         rc = stat(h, &sb);
 161         if (rc != 0) {
 162                 TALLOC_FREE(h);
 163                 return NULL;
 164         }
 165
 166         r = talloc_asprintf(mem_ctx, "%s%s", h, p);
 167         TALLOC_FREE(h);
 168
 169         return r;
 170 }