mirror/postgresql
2
0
Fork 0
mirror of https://git.postgresql.org/git/postgresql.git synced 2025-01-24 18:55:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

libpq: change PQconndefaults() to ignore invalid service files

Browse Source 
Previously missing or invalid service files returned NULL.  Also fix
pg_upgrade to report "out of memory" for a null return from
PQconndefaults().

Patch by Steve Singer, rewritten by me
This commit is contained in:
Bruce Momjian 2013-12-03 11:11:56 -05:00
parent 95e3d50539
commit 9e0a97f1c8
5 changed files with 23 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
contrib/pg_upgrade/server.c
View File

@ -325,6 +325,9 @@ check_pghost_envvar(void)
start = PQconndefaults(); start = PQconndefaults();
if (!start)
pg_fatal("out of memory\n");
for (option = start; option->keyword != NULL; option++) for (option = start; option->keyword != NULL; option++)
{ {
if (option->envvar && (strcmp(option->envvar, "PGHOST") == 0 || if (option->envvar && (strcmp(option->envvar, "PGHOST") == 0 ||

3
doc/src/sgml/libpq.sgml
View File

@ -483,7 +483,8 @@ typedef struct
with an entry having a null <structfield>keyword</> pointer. The with an entry having a null <structfield>keyword</> pointer. The
null pointer is returned if memory could not be allocated. Note that null pointer is returned if memory could not be allocated. Note that
the current default values (<structfield>val</structfield> fields) the current default values (<structfield>val</structfield> fields)
will depend on environment variables and other context. Callers will depend on environment variables and other context. A
missing or invalid service file will be silently ignored. Callers
must treat the connection options data as read-only. must treat the connection options data as read-only.
</para> </para>

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

@ -982,7 +982,7 @@ pg_fe_sendauth(AuthRequest areq, PGconn *conn)
* if there is an error, return NULL with an error message in errorMessage * if there is an error, return NULL with an error message in errorMessage
*/ */
char * char *
pg_fe_getauthname(PQExpBuffer errorMessage) pg_fe_getauthname(void)
{ {
const char *name = NULL; const char *name = NULL;
char *authn; char *authn;

2
src/interfaces/libpq/fe-auth.h
View File

@ -19,6 +19,6 @@
extern int pg_fe_sendauth(AuthRequest areq, PGconn *conn); extern int pg_fe_sendauth(AuthRequest areq, PGconn *conn);
extern char *pg_fe_getauthname(PQExpBuffer errorMessage); extern char *pg_fe_getauthname(void);
#endif /* FE_AUTH_H */ #endif /* FE_AUTH_H */

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

@ -875,7 +875,8 @@ PQconndefaults(void)
connOptions = conninfo_init(&errorBuf); connOptions = conninfo_init(&errorBuf);
if (connOptions != NULL) if (connOptions != NULL)
{ {
if (!conninfo_add_defaults(connOptions, &errorBuf)) /* pass NULL errorBuf to ignore errors */
if (!conninfo_add_defaults(connOptions, NULL))
{ {
PQconninfoFree(connOptions); PQconninfoFree(connOptions);
connOptions = NULL; connOptions = NULL;
@ -4412,9 +4413,10 @@ conninfo_array_parse(const char *const * keywords, const char *const * values,
* *
* Defaults are obtained from a service file, environment variables, etc. * Defaults are obtained from a service file, environment variables, etc.
* *
* Returns TRUE if successful, otherwise FALSE; errorMessage is filled in * Returns TRUE if successful, otherwise FALSE; errorMessage, if supplied,
* upon failure. Note that failure to locate a default value is not an * is filled in upon failure. Note that failure to locate a default value
* error condition here --- we just leave the option's value as NULL. * is not an error condition here --- we just leave the option's value as
* NULL.
*/ */
static bool static bool
conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage) conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage)
@ -4424,9 +4426,10 @@ conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage)
/* /*
* If there's a service spec, use it to obtain any not-explicitly-given * If there's a service spec, use it to obtain any not-explicitly-given
* parameters. * parameters. Ignore error if no error message buffer is passed
* because there is no way to pass back the failure message.
*/ */
if (parseServiceInfo(options, errorMessage) != 0) if (parseServiceInfo(options, errorMessage) != 0 && errorMessage)
return false; return false;
/* /*
@ -4448,8 +4451,9 @@ conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage)
option->val = strdup(tmp); option->val = strdup(tmp);
if (!option->val) if (!option->val)
{ {
printfPQExpBuffer(errorMessage, if (errorMessage)
libpq_gettext("out of memory\n")); printfPQExpBuffer(errorMessage,
libpq_gettext("out of memory\n"));
return false; return false;
} }
continue; continue;
@ -4465,8 +4469,9 @@ conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage)
option->val = strdup(option->compiled); option->val = strdup(option->compiled);
if (!option->val) if (!option->val)
{ {
printfPQExpBuffer(errorMessage, if (errorMessage)
libpq_gettext("out of memory\n")); printfPQExpBuffer(errorMessage,
libpq_gettext("out of memory\n"));
return false; return false;
} }
continue; continue;
@ -4477,7 +4482,7 @@ conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage)
*/ */
if (strcmp(option->keyword, "user") == 0) if (strcmp(option->keyword, "user") == 0)
{ {
option->val = pg_fe_getauthname(errorMessage); option->val = pg_fe_getauthname();
continue; continue;
} }
} }