database/postgres
database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-05-01 01:04:50 +03:00
Code

libpq: pass a memory allocation failure error up to PQconndefaults()

Browse Source 
Previously user name memory allocation failures were ignored and the
default user name set to NULL.
This commit is contained in:
Bruce Momjian 2014-03-20 11:48:31 -04:00
parent d1bdab2fa3
commit a4c8f14364
2 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/interfaces/libpq/fe-auth.c
View File

@ -741,8 +741,11 @@ pg_fe_getauthname(void)
*/ */
pglock_thread(); pglock_thread();
if (!name) /*
{ * We document PQconndefaults() to return NULL for a memory allocation
* failure. We don't have an API to return a user name lookup failure,
* so we just assume it always succeeds.
*/
#ifdef WIN32 #ifdef WIN32
if (GetUserName(username, &namesize)) if (GetUserName(username, &namesize))
name = username; name = username;
@ -750,7 +753,6 @@ pg_fe_getauthname(void)
if (pqGetpwuid(geteuid(), &pwdstr, pwdbuf, sizeof(pwdbuf), &pw) == 0) if (pqGetpwuid(geteuid(), &pwdstr, pwdbuf, sizeof(pwdbuf), &pw) == 0)
name = pw->pw_name; name = pw->pw_name;
#endif #endif
}
authn = name ? strdup(name) : NULL; authn = name ? strdup(name) : NULL;

7
src/interfaces/libpq/fe-connect.c
View File

@ -4482,6 +4482,13 @@ conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage)
if (strcmp(option->keyword, "user") == 0) if (strcmp(option->keyword, "user") == 0)
{ {
option->val = pg_fe_getauthname(); option->val = pg_fe_getauthname();
if (!option->val)
{
if (errorMessage)
printfPQExpBuffer(errorMessage,
libpq_gettext("out of memory\n"));
return false;
}
continue; continue;
} }
} }