sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2024-12-22 13:34:15 +03:00
Code

lib:util: Add path_expand_tilde()

Browse Source 
BUG: https://bugzilla.samba.org/show_bug.cgi?id=14370

Signed-off-by: Andreas Schneider <asn@samba.org>
Reviewed-by: Stefan Metzmacher <metze@samba.org>
This commit is contained in:
Andreas Schneider 2020-05-07 12:25:24 +02:00 committed by Andreas Schneider
parent 8b5e764413
commit 15457254be
2 changed files with 81 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
lib/util/util_paths.c
View File

@ -6,6 +6,7 @@
Copyright (C) Simo Sorce 2001
Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2003
Copyright (C) James Peach 2006
Copyright (c) 2020 Andreas Schneider <asn@samba.org>
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
@ -24,6 +25,7 @@
#include "includes.h"
#include "dynconfig/dynconfig.h"
#include "lib/util/util_paths.h"
#include "system/passwd.h"
/**
* @brief Returns an absolute path to a file in the Samba modules directory.
@ -62,3 +64,73 @@ const char *shlib_ext(void)
return get_dyn_SHLIBEXT();
}
static char *get_user_home_dir(TALLOC_CTX *mem_ctx)
{
struct passwd pwd = {0};
struct passwd *pwdbuf = NULL;
char buf[NSS_BUFLEN_PASSWD] = {0};
int rc;
rc = getpwuid_r(getuid(), &pwd, buf, NSS_BUFLEN_PASSWD, &pwdbuf);
if (rc != 0 || pwdbuf == NULL ) {
const char *szPath = getenv("HOME");
if (szPath == NULL) {
return NULL;
}
snprintf(buf, sizeof(buf), "%s", szPath);
return talloc_strdup(mem_ctx, buf);
}
return talloc_strdup(mem_ctx, pwd.pw_dir);
}
char *path_expand_tilde(TALLOC_CTX *mem_ctx, const char *d)
{
char *h = NULL, *r = NULL;
const char *p = NULL;
struct stat sb = {0};
int rc;
if (d[0] != '~') {
return talloc_strdup(mem_ctx, d);
}
d++;
/* handle ~user/path */
p = strchr(d, '/');
if (p != NULL && p > d) {
struct passwd *pw;
size_t s = p - d;
char u[128];
if (s >= sizeof(u)) {
return NULL;
}
memcpy(u, d, s);
u[s] = '\0';
pw = getpwnam(u);
if (pw == NULL) {
return NULL;
}
h = talloc_strdup(mem_ctx, pw->pw_dir);
} else {
p = d;
h = get_user_home_dir(mem_ctx);
}
if (h == NULL) {
return NULL;
}
rc = stat(h, &sb);
if (rc != 0) {
TALLOC_FREE(h);
return NULL;
}
r = talloc_asprintf(mem_ctx, "%s%s", h, p);
TALLOC_FREE(h);
return r;
}

9
lib/util/util_paths.h
View File

@ -51,4 +51,13 @@ char *data_path(TALLOC_CTX *mem_ctx, const char *name);
**/
const char *shlib_ext(void);
/**
* @brief Expand a directory starting with a tilde '~'
*
* @param[in] d The directory to expand.
*
* @return The expanded directory, NULL on error.
*/
char *path_expand_tilde(TALLOC_CTX *mem_ctx, const char *d);
#endif