Don't crash initdb when we fail to get the current username.
authorMagnus Hagander <magnus@hagander.net>
Tue, 31 Mar 2009 18:58:34 +0000 (18:58 +0000)
committerMagnus Hagander <magnus@hagander.net>
Tue, 31 Mar 2009 18:58:34 +0000 (18:58 +0000)
Give an error message and exit instead, like we do elsewhere...

Per report from Wez Furlong and Robert Treat.

src/bin/initdb/initdb.c

index 55dbdf9284dad5bcc25f90126fb873c8571d302e..f965f0b2a5f8c2286218a30fcffc45afded61de3 100644 (file)
@@ -42,7 +42,7 @@
  * Portions Copyright (c) 1994, Regents of the University of California
  * Portions taken from FreeBSD.
  *
- * $PostgreSQL: pgsql/src/bin/initdb/initdb.c,v 1.125.2.2 2008/02/29 23:31:57 adunstan Exp $
+ * $PostgreSQL: pgsql/src/bin/initdb/initdb.c,v 1.125.2.3 2009/03/31 18:58:34 mha Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -647,6 +647,13 @@ get_id(void)
                progname);
        exit(1);
    }
+   if (!pw)
+   {
+       fprintf(stderr,
+               _("%s: could not obtain information about current user: %s\n"),
+               progname, strerror(errno));
+       exit(1);
+   }
 #else                          /* the windows code */
 
    struct passwd_win32
@@ -658,7 +665,12 @@ get_id(void)
    DWORD       pwname_size = sizeof(pass_win32.pw_name) - 1;
 
    pw->pw_uid = 1;
-   GetUserName(pw->pw_name, &pwname_size);
+   if (!GetUserName(pw->pw_name, &pwname_size))
+   {
+       fprintf(stderr, _("%s: could not get current user name: %s\n"),
+               progname, strerror(errno));
+       exit(1);
+   }
 #endif
 
    return xstrdup(pw->pw_name);