Clean up code in libpq that obtains user's home directory: make a single

subroutine that can hide platform dependencies.  The WIN32 path is still
a stub, but I await a fix from one of the win32 hackers.
Also clean up unnecessary #ifdef WIN32 ugliness in a couple of places.

diff --git a/src/interfaces/libpq/fe-auth.c b/src/interfaces/libpq/fe-auth.c
index 35f702058078e557a314be293ec0eaaf58e17ea2..f326335bf9114fa02309bfde866d8eba70602ff7 100644 (file)
--- a/src/interfaces/libpq/fe-auth.c
+++ b/src/interfaces/libpq/fe-auth.c
@@ -10,7 +10,7 @@
  * exceed INITIAL_EXPBUFFER_SIZE (currently 256 bytes).
  *
  * IDENTIFICATION
- *       $PostgreSQL: pgsql/src/interfaces/libpq/fe-auth.c,v 1.97 2004/12/31 22:03:50 pgsql Exp $
+ *       $PostgreSQL: pgsql/src/interfaces/libpq/fe-auth.c,v 1.98 2005/01/04 23:18:25 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -718,8 +718,16 @@ char *
 fe_getauthname(char *PQerrormsg)
 {
        const char *name = NULL;
-       char       *authn = NULL;
+       char       *authn;
        MsgType         authsvc;
+#ifdef WIN32
+       char            username[128];
+       DWORD           namesize = sizeof(username) - 1;
+#else
+       char            pwdbuf[BUFSIZ];
+       struct passwd pwdstr;
+       struct passwd *pw = NULL;
+#endif
 
        authsvc = fe_getauthsvc(PQerrormsg);
 
@@ -728,6 +736,7 @@ fe_getauthname(char *PQerrormsg)
                return NULL;                    /* leave original error message in place */
 
        pglock_thread();
+
 #ifdef KRB4
        if (authsvc == STARTUP_KRB4_MSG)
                name = pg_krb4_authname(PQerrormsg);
@@ -742,18 +751,10 @@ fe_getauthname(char *PQerrormsg)
                || (authsvc == STARTUP_KRB5_MSG && !name))
        {
 #ifdef WIN32
-               char            username[128];
-               DWORD           namesize = sizeof(username) - 1;
-
                if (GetUserName(username, &namesize))
                        name = username;
 #else
-               char            pwdbuf[BUFSIZ];
-               struct passwd pwdstr;
-               struct passwd *pw = NULL;
-
-               if (pqGetpwuid(geteuid(), &pwdstr,
-                                          pwdbuf, sizeof(pwdbuf), &pw) == 0)
+               if (pqGetpwuid(geteuid(), &pwdstr, pwdbuf, sizeof(pwdbuf), &pw) == 0)
                        name = pw->pw_name;
 #endif
        }
@@ -763,8 +764,9 @@ fe_getauthname(char *PQerrormsg)
                                 libpq_gettext("fe_getauthname: invalid authentication system: %d\n"),
                                 authsvc);
 
-       if (name && (authn = (char *) malloc(strlen(name) + 1)))
-               strcpy(authn, name);
+       authn = name ? strdup(name) : NULL;
+
        pgunlock_thread();
+
        return authn;
 }

diff --git a/src/interfaces/libpq/fe-connect.c b/src/interfaces/libpq/fe-connect.c
index 876ac997f0075a820b561d3ba1c9302b69b78fd2..95ed43f83bbbd0df3ff59a9d001e44716da1cc88 100644 (file)
--- a/src/interfaces/libpq/fe-connect.c
+++ b/src/interfaces/libpq/fe-connect.c
@@ -8,7 +8,7 @@
  *
  *
  * IDENTIFICATION
- *       $PostgreSQL: pgsql/src/interfaces/libpq/fe-connect.c,v 1.294 2004/12/31 22:03:50 pgsql Exp $
+ *       $PostgreSQL: pgsql/src/interfaces/libpq/fe-connect.c,v 1.295 2005/01/04 23:18:25 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -1944,7 +1944,7 @@ makeEmptyPGconn(void)
        PGconn     *conn;
 
 #ifdef WIN32
-       /* needed to use the static libpq under windows as well */
+       /* make sure socket support is up and running */
        WSADATA         wsaData;
 
        if (WSAStartup(MAKEWORD(1, 1), &wsaData))
@@ -2324,12 +2324,7 @@ retry5:
 
        /* All done */
        closesocket(tmpsock);
-#ifdef WIN32
-       WSASetLastError(save_errno);
-#else
-       errno = save_errno;
-#endif
-
+       SOCK_ERRNO_SET(save_errno);
        return TRUE;
 
 cancel_errReturn:
@@ -2346,12 +2341,7 @@ cancel_errReturn:
        }
        if (tmpsock >= 0)
                closesocket(tmpsock);
-#ifdef WIN32
-       WSASetLastError(save_errno);
-#else
-       errno = save_errno;
-#endif
-
+       SOCK_ERRNO_SET(save_errno);
        return FALSE;
 }
 

diff --git a/src/interfaces/libpq/fe-secure.c b/src/interfaces/libpq/fe-secure.c
index 74758e9bca1178c998de36bb6437fa340e6b3374..89064be506c03840f5980e056c080e23bbde297e 100644 (file)
--- a/src/interfaces/libpq/fe-secure.c
+++ b/src/interfaces/libpq/fe-secure.c
@@ -11,7 +11,7 @@
  *
  *
  * IDENTIFICATION
- *       $PostgreSQL: pgsql/src/interfaces/libpq/fe-secure.c,v 1.61 2004/12/31 22:03:50 pgsql Exp $
+ *       $PostgreSQL: pgsql/src/interfaces/libpq/fe-secure.c,v 1.62 2005/01/04 23:18:25 tgl Exp $
  *
  * NOTES
  *       [ Most of these notes are wrong/obsolete, but perhaps not all ]
@@ -492,6 +492,32 @@ pqsecure_write(PGconn *conn, const void *ptr, size_t len)
 /*                                               SSL specific code                                             */
 /* ------------------------------------------------------------ */
 #ifdef USE_SSL
+
+/*
+ * Obtain user's home directory, return in given buffer
+ *
+ * This code isn't really SSL-specific, but currently we only need it in
+ * SSL-related places.
+ */
+static bool
+pqGetHomeDirectory(char *buf, int bufsize)
+{
+#ifndef WIN32
+       char            pwdbuf[BUFSIZ];
+       struct passwd pwdstr;
+       struct passwd *pwd = NULL;
+
+       if (pqGetpwuid(geteuid(), &pwdstr, pwdbuf, sizeof(pwdbuf), &pwd) != 0)
+               return false;
+       StrNCpy(buf, pwd->pw_dir, bufsize);
+       return true;
+
+#else
+
+       return false;                           /* PLACEHOLDER */
+#endif
+}
+
 /*
  *     Certificate verification callback
  *
@@ -612,7 +638,7 @@ verify_peer(PGconn *conn)
 
        return -1;
 }
-#endif
+#endif /* NOT_USED */
 
 /*
  *     Load precomputed DH parameters.
@@ -624,23 +650,18 @@ verify_peer(PGconn *conn)
 static DH  *
 load_dh_file(int keylength)
 {
-#ifdef WIN32
-       return NULL;
-#else
-       char            pwdbuf[BUFSIZ];
-       struct passwd pwdstr;
-       struct passwd *pwd = NULL;
-       FILE       *fp;
+       char            homedir[MAXPGPATH];
        char            fnbuf[MAXPGPATH];
-       DH                 *dh = NULL;
+       FILE       *fp;
+       DH                 *dh;
        int                     codes;
 
-       if (pqGetpwuid(getuid(), &pwdstr, pwdbuf, sizeof(pwdbuf), &pwd) != 0)
+       if (!pqGetHomeDirectory(homedir, sizeof(homedir)))
                return NULL;
 
        /* attempt to open file.  It's not an error if it doesn't exist. */
        snprintf(fnbuf, sizeof(fnbuf), "%s/.postgresql/dh%d.pem",
-                        pwd->pw_dir, keylength);
+                        homedir, keylength);
 
        if ((fp = fopen(fnbuf, "r")) == NULL)
                return NULL;
@@ -667,7 +688,6 @@ load_dh_file(int keylength)
        }
 
        return dh;
-#endif
 }
 
 /*
@@ -771,12 +791,7 @@ tmp_dh_cb(SSL *s, int is_export, int keylength)
 static int
 client_cert_cb(SSL *ssl, X509 **x509, EVP_PKEY **pkey)
 {
-#ifdef WIN32
-       return 0;
-#else
-       char            pwdbuf[BUFSIZ];
-       struct passwd pwdstr;
-       struct passwd *pwd = NULL;
+       char            homedir[MAXPGPATH];
        struct stat buf,
                                buf2;
        char            fnbuf[MAXPGPATH];
@@ -785,7 +800,7 @@ client_cert_cb(SSL *ssl, X509 **x509, EVP_PKEY **pkey)
        int                     (*cb) () = NULL;        /* how to read user password */
        char            sebuf[256];
 
-       if (pqGetpwuid(getuid(), &pwdstr, pwdbuf, sizeof(pwdbuf), &pwd) != 0)
+       if (!pqGetHomeDirectory(homedir, sizeof(homedir)))
        {
                printfPQExpBuffer(&conn->errorMessage,
                                          libpq_gettext("could not get user information\n"));
@@ -794,7 +809,7 @@ client_cert_cb(SSL *ssl, X509 **x509, EVP_PKEY **pkey)
 
        /* read the user certificate */
        snprintf(fnbuf, sizeof(fnbuf), "%s/.postgresql/postgresql.crt",
-                        pwd->pw_dir);
+                        homedir);
        if ((fp = fopen(fnbuf, "r")) == NULL)
        {
                printfPQExpBuffer(&conn->errorMessage,
@@ -817,7 +832,7 @@ client_cert_cb(SSL *ssl, X509 **x509, EVP_PKEY **pkey)
 
        /* read the user key */
        snprintf(fnbuf, sizeof(fnbuf), "%s/.postgresql/postgresql.key",
-                        pwd->pw_dir);
+                        homedir);
        if (stat(fnbuf, &buf) == -1)
        {
                printfPQExpBuffer(&conn->errorMessage,
@@ -873,7 +888,6 @@ client_cert_cb(SSL *ssl, X509 **x509, EVP_PKEY **pkey)
        }
 
        return 1;
-#endif
 }
 
 #ifdef ENABLE_THREAD_SAFETY
@@ -885,6 +899,7 @@ pq_threadidcallback(void)
 }
 
 static pthread_mutex_t *pq_lockarray;
+
 static void
 pq_lockingcallback(int mode, int n, const char *file, int line)
 {
@@ -893,6 +908,7 @@ pq_lockingcallback(int mode, int n, const char *file, int line)
        else
                pthread_mutex_unlock(&pq_lockarray[n]);
 }
+
 #endif   /* ENABLE_THREAD_SAFETY */
 
 static int
@@ -969,23 +985,17 @@ init_ssl_system(PGconn *conn)
 static int
 initialize_SSL(PGconn *conn)
 {
-#ifndef WIN32
        struct stat buf;
-       char            pwdbuf[BUFSIZ];
-       struct passwd pwdstr;
-       struct passwd *pwd = NULL;
+       char            homedir[MAXPGPATH];
        char            fnbuf[MAXPGPATH];
-#endif
 
        if (init_ssl_system(conn))
                return -1;
 
-#ifndef WIN32
        /* Set up to verify server cert, if root.crt is present */
-       if (pqGetpwuid(getuid(), &pwdstr, pwdbuf, sizeof(pwdbuf), &pwd) == 0)
+       if (pqGetHomeDirectory(homedir, sizeof(homedir)))
        {
-               snprintf(fnbuf, sizeof(fnbuf), "%s/.postgresql/root.crt",
-                                pwd->pw_dir);
+               snprintf(fnbuf, sizeof(fnbuf), "%s/.postgresql/root.crt", homedir);
                if (stat(fnbuf, &buf) == 0)
                {
                        if (!SSL_CTX_load_verify_locations(SSL_context, fnbuf, NULL))
@@ -1009,7 +1019,6 @@ initialize_SSL(PGconn *conn)
 
        /* set up mechanism to provide client certificate, if available */
        SSL_CTX_set_client_cert_cb(SSL_context, client_cert_cb);
-#endif
 
        return 0;
 }
@@ -1232,15 +1241,19 @@ PQgetssl(PGconn *conn)
                return NULL;
        return conn->ssl;
 }
-#else
+
+#else   /* !USE_SSL */
+
 void *
 PQgetssl(PGconn *conn)
 {
        return NULL;
 }
+
 #endif   /* USE_SSL */
 
 #ifdef ENABLE_THREAD_SAFETY
+
 /*
  *     Block SIGPIPE for this thread.  This prevents send()/write() from exiting
  *     the application.
@@ -1322,4 +1335,5 @@ pq_reset_sigpipe(sigset_t *osigset, bool sigpipe_pending, bool got_epipe)
 
        SOCK_ERRNO_SET(save_errno);
 }
-#endif
+
+#endif /* ENABLE_THREAD_SAFETY */