Move sdb_get_homedir() from tools/sysdb to utils/os.

[sysdb.git] / src / utils / os.c

diff --git a/src/utils/os.c b/src/utils/os.c
index 747a3700014bd248f19230bfd0363ccba55437ee..e975f96e743b6510102426d6450dbe9270690e1c 100644 (file)
--- a/src/utils/os.c
+++ b/src/utils/os.c
@@ -35,6 +35,7 @@
 #include <errno.h>
 
 #include <sys/types.h>
+#include <sys/socket.h>
 #include <sys/stat.h>
 
 #include <dirent.h>
@@ -45,12 +46,44 @@
 #include <unistd.h>
 
 #include <libgen.h>
+#include <netdb.h>
 #include <pwd.h>
 
 /*
  * public API
  */
 
+char *
+sdb_get_homedir(void)
+{
+       char *username = sdb_get_current_user();
+
+       struct passwd pw_entry;
+       struct passwd *result = NULL;
+
+       char buf[4096];
+
+       int status;
+
+       if (username) {
+               memset(&pw_entry, 0, sizeof(pw_entry));
+               status = getpwnam_r(username, &pw_entry, buf, sizeof(buf), &result);
+       }
+       else
+               status = -1;
+
+       if (status || (! result)) {
+               char errbuf[1024];
+               sdb_log(SDB_LOG_WARNING, "os: Failed to determine home directory "
+                               "for user %s: %s", username,
+                               sdb_strerror(errno, errbuf, sizeof(errbuf)));
+               free(username);
+               return NULL;
+       }
+       free(username);
+       return strdup(result->pw_dir);
+} /* sdb_get_homedir */
+
 int
 sdb_mkdir_all(const char *pathname, mode_t mode)
 {
@@ -150,7 +183,7 @@ sdb_get_current_user(void)
 
        uid_t uid;
 
-       char buf[1024];
+       char buf[4096];
        int status;
 
        uid = geteuid();
@@ -248,5 +281,63 @@ sdb_write(int fd, size_t msg_len, const void *msg)
        return (ssize_t)msg_len;
 } /* sdb_write */
 
+int
+sdb_resolve(int network, const char *address, struct addrinfo **res)
+{
+       struct addrinfo ai_hints;
+       const char *host;
+       char *port;
+       int status;
+
+       if (! res) {
+               errno = EINVAL;
+               return EAI_SYSTEM;
+       }
+
+       if (address) {
+               host = address;
+               port = strchr(host, ':');
+               if (port) {
+                       *port = '\0';
+                       ++port;
+               }
+               if (! *host)
+                       host = NULL;
+       }
+       else {
+               host = NULL;
+               port = NULL;
+       }
+
+       memset(&ai_hints, 0, sizeof(ai_hints));
+       ai_hints.ai_flags = AI_PASSIVE | AI_ADDRCONFIG;
+       if (network & SDB_NET_V4)
+               ai_hints.ai_family = AF_INET;
+       else if (network & SDB_NET_V6)
+               ai_hints.ai_family = AF_INET6;
+       else
+               ai_hints.ai_family = AF_UNSPEC;
+
+       if ((network & SDB_NET_IP) == SDB_NET_IP) {
+               ai_hints.ai_socktype = 0;
+               ai_hints.ai_protocol = 0;
+       }
+       else if (network & SDB_NET_TCP) {
+               ai_hints.ai_socktype = SOCK_STREAM;
+               ai_hints.ai_protocol = IPPROTO_TCP;
+       }
+       else if (network & SDB_NET_UDP) {
+               ai_hints.ai_socktype = SOCK_DGRAM;
+               ai_hints.ai_protocol = IPPROTO_UDP;
+       }
+
+       status = getaddrinfo(host, port, &ai_hints, res);
+       if (port) {
+               --port;
+               *port = ':';
+       }
+       return status;
+} /* sdb_resolve */
+
 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */