summaryrefslogtreecommitdiffstats
path: root/third_party/heimdal/lib/roken/getuserinfo.c
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/heimdal/lib/roken/getuserinfo.c')
-rw-r--r--third_party/heimdal/lib/roken/getuserinfo.c334
1 files changed, 334 insertions, 0 deletions
diff --git a/third_party/heimdal/lib/roken/getuserinfo.c b/third_party/heimdal/lib/roken/getuserinfo.c
new file mode 100644
index 0000000..09f4c73
--- /dev/null
+++ b/third_party/heimdal/lib/roken/getuserinfo.c
@@ -0,0 +1,334 @@
+/*
+ * Copyright (c) 2017 Kungliga Tekniska Högskolan
+ * (Royal Institute of Technology, Stockholm, Sweden).
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * 3. Neither the name of the Institute nor the names of its contributors
+ * may be used to endorse or promote products derived from this software
+ * without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+
+#include <config.h>
+#include "roken.h"
+
+#ifdef WIN32
+#include <Shlobj.h> // need to include definitions of constants
+#define SECURITY_WIN32
+#include <security.h>
+#else
+#include <sys/types.h>
+#include <pwd.h>
+#include <unistd.h>
+#endif
+
+/**
+ * Returns the user's SHELL.
+ */
+ROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL
+roken_get_shell(char *shell, size_t shellsz)
+{
+ char *p;
+
+#ifndef WIN32
+#ifdef HAVE_GETPWNAM_R
+ size_t buflen = 2048;
+
+ if (sysconf(_SC_GETPW_R_SIZE_MAX) > 0)
+ buflen = sysconf(_SC_GETPW_R_SIZE_MAX);
+#endif
+
+ if (issuid())
+ return "/bin/sh";
+
+ p = secure_getenv("SHELL");
+ if (p != NULL && p[0] != '\0') {
+ if (strlcpy(shell, p, shellsz) < shellsz)
+ return shell;
+ errno = ERANGE;
+ return NULL;
+ }
+
+#ifdef HAVE_GETPWNAM_R
+ {
+ struct passwd pwd;
+ struct passwd *pwdp;
+ char buf[buflen];
+ char user[128];
+ const char *username = roken_get_username(user, sizeof(user));
+
+ if (username &&
+ getpwnam_r(username, &pwd, buf, buflen, &pwdp) == 0 &&
+ pwdp != NULL && pwdp->pw_shell != NULL) {
+ if (strlcpy(shell, pwdp->pw_shell, shellsz) < shellsz)
+ return shell;
+ errno = ERANGE;
+ return NULL;
+ }
+ }
+#endif
+ errno = 0;
+ return "/bin/sh";
+#else
+ /* Windows */
+ p = getenv("SHELL");
+ if (p != NULL && p[0] != '\0') {
+ if (strlcpy(shell, p, shellsz) < shellsz)
+ return shell;
+ errno = ERANGE;
+ return NULL;
+ }
+ errno = 0;
+ return NULL;
+#endif
+}
+
+/**
+ * Returns the home directory.
+ */
+ROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL
+roken_get_homedir(char *home, size_t homesz)
+{
+ char *p;
+
+#ifdef WIN32
+ if (homesz < MAX_PATH) {
+ errno = ERANGE;
+ return NULL;
+ }
+
+ if (SUCCEEDED(SHGetFolderPath(NULL, CSIDL_PROFILE, NULL,
+ SHGFP_TYPE_CURRENT, home)))
+ return home;
+
+ if ((p = getenv("HOMEDRIVE")) != NULL && p[0] != '\0') {
+ if (strlcpy(home, p, homesz) >= homesz) {
+ errno = ERANGE;
+ return NULL;
+ }
+ if ((p = getenv("HOMEPATH")) != NULL) {
+ if (strlcat(home, p, homesz) < homesz)
+ return home;
+ errno = ERANGE;
+ return NULL;
+ }
+ return home;
+ }
+ HEIM_FALLTHROUGH;
+#else
+#ifdef HAVE_GETPWNAM_R
+ size_t buflen = 2048;
+
+ if (sysconf(_SC_GETPW_R_SIZE_MAX) > 0)
+ buflen = sysconf(_SC_GETPW_R_SIZE_MAX);
+#endif
+
+ if (issuid()) {
+ errno = 0;
+ return NULL;
+ }
+
+ p = secure_getenv("HOME");
+ if (p != NULL && p[0] != '\0') {
+ if (strlcpy(home, p, homesz) < homesz)
+ return home;
+ errno = ERANGE;
+ return NULL;
+ }
+
+#ifdef HAVE_GETPWNAM_R
+ {
+ char user[128];
+ const char *username = roken_get_username(user, sizeof(user));
+ struct passwd pwd;
+ struct passwd *pwdp;
+ char buf[buflen];
+
+ if (username &&
+ getpwnam_r(username, &pwd, buf, buflen, &pwdp) == 0 &&
+ pwdp != NULL && pwdp->pw_dir != NULL) {
+ if (strlcpy(home, pwdp->pw_dir, homesz) < homesz)
+ return home;
+ errno = ERANGE;
+ return NULL;
+ }
+ }
+#endif
+#endif
+ errno = 0;
+ return NULL;
+}
+
+/**
+ * Returns the home directory on Unix, or the AppData directory on
+ * Windows.
+ */
+ROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL
+roken_get_appdatadir(char *appdata, size_t appdatasz)
+{
+#ifdef WIN32
+ char *p;
+#endif
+
+#ifndef WIN32
+ return roken_get_homedir(appdata, appdatasz);
+#else
+ if (appdatasz < MAX_PATH) {
+ errno = ERANGE;
+ return NULL;
+ }
+
+ if (SUCCEEDED(SHGetFolderPath(NULL, CSIDL_APPDATA, NULL,
+ SHGFP_TYPE_CURRENT, appdata)))
+ return appdata;
+
+ if ((p = getenv("APPDATA")) != NULL && p[0] != '\0') {
+ if (strlcpy(appdata, p, appdatasz) < appdatasz)
+ return appdata;
+ errno = ERANGE;
+ return NULL;
+ }
+
+ errno = 0;
+ return NULL;
+#endif
+}
+
+/**
+ * Return a bare username. This is used for, e.g., constructing default
+ * principal names.
+ *
+ * On POSIX systems, if the caller is not set-uid-like, then this will return
+ * the value of the USER or LOGNAME environment variables (in that order of
+ * preference), else the username found by looking up the effective UID.
+ */
+ROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL
+roken_get_username(char *user, size_t usersz)
+{
+ char *p;
+
+#ifdef WIN32
+ ULONG sz = usersz;
+
+ if (GetUserNameEx(NameSamCompatible, user, &sz)) {
+ /*
+ * There's no EXTENDED_NAME_FORMAT for "bare username". We we
+ * have to parse one.
+ */
+ p = strchr(user, '\\');
+ if (p != NULL) {
+ p++;
+ memmove(user, p, strlen(p) + 1);
+ }
+ return user;
+ } else {
+ DWORD err = GetLastError();
+ if (err == ERROR_MORE_DATA) {
+ errno = ERANGE;
+ return NULL;
+ }
+ /* %USERNAME% is generally bare */
+ p = getenv("USERNAME");
+ if (p != NULL && p[0] != '\0') {
+ if (strchr(p, '\\') != NULL)
+ p = strchr(p, '\\') + 1;
+ if (strlcpy(user, p, usersz) < usersz)
+ return user;
+ errno = ERANGE;
+ return NULL;
+ }
+ }
+#else
+#ifdef HAVE_GETPWUID_R
+ size_t buflen = 2048;
+
+ if (sysconf(_SC_GETPW_R_SIZE_MAX) > 0)
+ buflen = sysconf(_SC_GETPW_R_SIZE_MAX);
+#endif
+
+ p = secure_getenv("USER");
+ if (p == NULL || p[0] == '\0')
+ p = secure_getenv("LOGNAME");
+ if (p != NULL && p[0] != '\0') {
+ if (strlcpy(user, p, usersz) < usersz)
+ return user;
+ errno = ERANGE;
+ return NULL;
+ }
+
+#ifdef HAVE_GETPWUID_R
+ {
+ struct passwd pwd;
+ struct passwd *pwdp;
+ char buf[buflen];
+
+ if (getpwuid_r(getuid(), &pwd, buf, buflen, &pwdp) == 0 &&
+ pwdp != NULL && pwdp->pw_name != NULL) {
+ if (strlcpy(user, pwdp->pw_name, usersz) < usersz)
+ return user;
+ errno = ERANGE;
+ return NULL;
+ }
+ }
+#endif
+#endif
+ errno = 0;
+ return NULL;
+}
+
+/**
+ * Return a bare username. This is used for, e.g., constructing default
+ * principal names.
+ *
+ * On POSIX systems this returns the name recorded in the system as currently
+ * logged in on the current terminal.
+ */
+ROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL
+roken_get_loginname(char *user, size_t usersz)
+{
+#ifdef WIN32
+ return roken_get_username(user, usersz);
+#else
+#ifdef HAVE_GETLOGIN_R
+ if ((errno = getlogin_r(user, usersz)) == 0)
+ return user;
+ if (errno != ENOENT)
+ return NULL;
+#else
+#ifdef HAVE_GETLOGIN
+ if ((p = getlogin()) != NULL && p[0] != '\0') {
+ if (strlcpy(user, p, usersz) < usersz)
+ return user;
+ errno = ERANGE;
+ return NULL;
+ }
+ if (errno != ENOENT)
+ return NULL;
+#endif
+#endif
+ errno = 0;
+ return NULL;
+#endif
+}
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-28 00:07:36 +0000