1 files changed, 76 insertions, 29 deletions

diff --git a/common/userpref.c b/common/userpref.c
index d363b0e..ab3a3cf 100644
--- a/common/userpref.c
+++ b/common/userpref.c
@@ -26,6 +26,9 @@
 #include <stdint.h>
 #include <stdlib.h>
 #include <string.h>
+#include <sys/types.h>
+#include <pwd.h>
+#include <unistd.h>
 #ifdef HAVE_OPENSSL
 #include <openssl/pem.h>
 #include <openssl/rsa.h>
@@ -48,6 +51,7 @@
 
 #include "userpref.h"
 #include "debug.h"
+#include "utils.h"
 
 #define LIBIMOBILEDEVICE_CONF_DIR  "libimobiledevice"
 #define LIBIMOBILEDEVICE_CONF_FILE "libimobiledevicerc"
@@ -65,7 +69,7 @@
 #define DIR_SEP_S "/"
 #endif
 
-static char __config_dir[512] = {0, };
+static char *__config_dir = NULL;
 
 #ifdef WIN32
 static char *userpref_utf16_to_utf8(wchar_t *unistr, long len, long *items_read, long *items_written)
@@ -103,24 +107,60 @@ static char *userpref_utf16_to_utf8(wchar_t *unistr, long len, long *items_read,
 #endif
 
 #ifndef WIN32
-static const char *userpref_get_tmp_dir()
+static char *get_home_dir_from_system(void)
 {
-	const char *cdir = getenv("TMPDIR");
-	if (cdir && cdir[0])
-		return cdir;
-	cdir = getenv("TMP");
-	if (cdir && cdir[0])
-		return cdir;
-	cdir = getenv("TEMP");
-	if (cdir && cdir[0])
-		return cdir;
-	return "/tmp";
+	long bufsize;
+	char *buf;
+	int error;
+	struct passwd pwd;
+	struct passwd *ppwd;
+	char *result = NULL;
+
+	bufsize = sysconf (_SC_GETPW_R_SIZE_MAX);
+
+	if (bufsize < 0)
+		bufsize = 1024;
+
+	buf = NULL;
+
+	do {
+		free (buf);
+		buf = malloc (bufsize);
+		if (!buf)
+			return NULL;
+
+		errno = 0;
+		error = getpwuid_r (getuid (), &pwd, buf, bufsize, &ppwd);
+
+		error = (error < 0) ? errno : error;
+
+		if (!ppwd) {
+			if (error == 0 || error == ENOENT)
+				break;
+			else if (bufsize > 32 * 1024)
+				break; /* Unreasonable size; let's bail out */
+
+			bufsize *= 2;
+		}
+	} while (!ppwd);
+
+	if (ppwd && ppwd->pw_dir)
+		result = strdup (ppwd->pw_dir);
+
+	free (buf);
+
+	return result;
 }
 #endif
 
 static const char *userpref_get_config_dir()
 {
-	if (__config_dir[0]) return __config_dir;
+	char *base_config_dir;
+	int use_dot_config;
+
+	if (__config_dir)
+		return __config_dir;
+
 #ifdef WIN32
 	wchar_t path[MAX_PATH+1];
 	HRESULT hr;
@@ -131,38 +171,45 @@ static const char *userpref_get_config_dir()
 	if (hr == S_OK) {
 		b = SHGetPathFromIDListW (pidl, path);
 		if (b) {
-			char *cdir = userpref_utf16_to_utf8 (path, wcslen(path), NULL, NULL);
-			strcpy(__config_dir, cdir);
-			free(cdir);
+			base_config_dir = userpref_utf16_to_utf8 (path, wcslen(path), NULL, NULL);
 			CoTaskMemFree (pidl);
 		}
 	}
+
+	use_dot_config = 0;
 #else
 	const char *cdir = getenv("XDG_CONFIG_HOME");
 	if (!cdir) {
 		cdir = getenv("HOME");
 		if (!cdir || !cdir[0]) {
-			const char *tdir = userpref_get_tmp_dir();
-			strcpy(__config_dir, tdir);
-			strcat(__config_dir, DIR_SEP_S);
-			strcat(__config_dir, "root");
+			base_config_dir = get_home_dir_from_system();
+			if (!base_config_dir)
+				return NULL;
 		} else {
-			strcpy(__config_dir, cdir);
+			base_config_dir = strdup(cdir);
 		}
-		strcat(__config_dir, DIR_SEP_S);
-		strcat(__config_dir, ".config");
+
+		use_dot_config = 1;
 	} else {
-		strcpy(__config_dir, cdir);
+		base_config_dir = strdup(cdir);
+		use_dot_config = 0;
 	}
 #endif
-	strcat(__config_dir, DIR_SEP_S);
-	strcat(__config_dir, LIBIMOBILEDEVICE_CONF_DIR);
 
-	int i = strlen(__config_dir)-1;	
-	while ((i > 0) && (__config_dir[i] == DIR_SEP)) {
-		__config_dir[i--] = '\0';
+	if (use_dot_config)
+		__config_dir = string_concat(base_config_dir, DIR_SEP_S, ".config", DIR_SEP_S, LIBIMOBILEDEVICE_CONF_DIR);
+	else
+		__config_dir = string_concat(base_config_dir, DIR_SEP_S, LIBIMOBILEDEVICE_CONF_DIR);
+
+	if (__config_dir) {
+		int i = strlen(__config_dir)-1;	
+		while ((i > 0) && (__config_dir[i] == DIR_SEP)) {
+			__config_dir[i--] = '\0';
+		}
 	}
 
+	free(base_config_dir);
+
 	return __config_dir;
 }