diff -durpN postgresql/doc/src/sgml/libpq.sgml postgresql.1/doc/src/sgml/libpq.sgml --- postgresql/doc/src/sgml/libpq.sgml 2012-08-03 09:39:30.114266570 +0200 +++ postgresql.1/doc/src/sgml/libpq.sgml 2012-10-10 14:09:31.924890230 +0200 @@ -496,6 +496,37 @@ typedef struct + + PQconninfoPQconninfo + + + Returns the connection options used by a live connection. + +PQconninfoOption *PQconninfo(PGconn *conn, bool for_replication); + + + + + Returns a connection options array. This can be used to determine + all possible PQconnectdb options and their + current values that were used to connect to the server. The return + value points to an array of PQconninfoOption + structures, which ends with an entry having a null keyword + pointer. Every notes above for PQconndefaults also apply. + + + + The for_replication argument can be used to exclude some + options from the list which are used by the walreceiver module. + pg_basebackup uses this pre-filtered list + to construct primary_conninfo in the automatically generated + recovery.conf file. + + + + + + PQconninfoParsePQconninfoParse diff -durpN postgresql/src/interfaces/libpq/exports.txt postgresql.1/src/interfaces/libpq/exports.txt --- postgresql/src/interfaces/libpq/exports.txt 2012-10-09 09:58:14.342782974 +0200 +++ postgresql.1/src/interfaces/libpq/exports.txt 2012-10-10 14:09:56.052057835 +0200 @@ -164,3 +164,4 @@ PQsetSingleRowMode 161 lo_lseek64 162 lo_tell64 163 lo_truncate64 164 +PQconninfo 165 diff -durpN postgresql/src/interfaces/libpq/fe-connect.c postgresql.1/src/interfaces/libpq/fe-connect.c --- postgresql/src/interfaces/libpq/fe-connect.c 2012-09-09 08:11:09.470401480 +0200 +++ postgresql.1/src/interfaces/libpq/fe-connect.c 2012-10-10 21:53:05.502966398 +0200 @@ -137,6 +137,9 @@ static int ldapServiceLookup(const char * PQconninfoOptions[] *must* be NULL. In a working copy, non-null "val" * fields point to malloc'd strings that should be freed when the working * array is freed (see PQconninfoFree). + * + * If you add a new connection option to this list, remember to add it to + * PQconninfoMappings[] below. * ---------- */ static const PQconninfoOption PQconninfoOptions[] = { @@ -264,6 +267,66 @@ static const PQconninfoOption PQconninfo NULL, NULL, 0} }; +/* + * We need a mapping between the PQconninfoOptions[] array + * and PGconn members. We have to keep it separate from + * PQconninfoOptions[] to not leak info about PGconn members + * to clients. + */ +typedef struct PQconninfoMapping { + char *keyword; + size_t member_offset; + bool for_replication; + /* + * Special and simplistic value mapping between + * PQconninfoOption and PGconn. Only used by "requiressl". + */ + char *conninfoValue; + char *connValue; +} PQconninfoMapping; +#define PGCONNMEMBERADDR(conn, mapping) ((char **)((char *)conn + mapping->member_offset)) + +static const PQconninfoMapping PQconninfoMappings[] = +{ + /* "authtype" is not used anymore, there is no mapping to PGconn */ + /* there is no mapping of "service" to PGconn */ + { "user", offsetof(struct pg_conn, pguser), true, NULL, NULL }, + { "password", offsetof(struct pg_conn, pgpass), true, NULL, NULL }, + { "connect_timeout", offsetof(struct pg_conn, connect_timeout), true, NULL, NULL }, + { "dbname", offsetof(struct pg_conn, dbName), false, NULL, NULL }, + { "host", offsetof(struct pg_conn, pghost), true, NULL, NULL }, + { "hostaddr", offsetof(struct pg_conn, pghostaddr), true, NULL, NULL }, + { "port", offsetof(struct pg_conn, pgport), true, NULL, NULL }, + { "client_encoding", offsetof(struct pg_conn, client_encoding_initial), false, NULL, NULL }, + { "tty", offsetof(struct pg_conn, pgtty), false, NULL, NULL }, + { "options", offsetof(struct pg_conn, pgoptions), true, NULL, NULL }, + { "application_name", offsetof(struct pg_conn, appname), false, NULL, NULL }, + { "fallback_application_name", offsetof(struct pg_conn, fbappname), false, NULL, NULL }, + { "keepalives", offsetof(struct pg_conn, keepalives), true, NULL, NULL }, + { "keepalives_idle", offsetof(struct pg_conn, keepalives_idle), true, NULL, NULL }, + { "keepalives_interval", offsetof(struct pg_conn, keepalives_interval), true, NULL, NULL }, + { "keepalives_count", offsetof(struct pg_conn, keepalives_count), true, NULL, NULL }, +#ifdef USE_SSL + { "requiressl", offsetof(struct pg_conn, sslmode), false, "1", "require" }, +#endif + { "sslmode", offsetof(struct pg_conn, sslmode), true, NULL, NULL }, + { "sslcompression", offsetof(struct pg_conn, sslcompression), true, NULL, NULL }, + { "sslcert", offsetof(struct pg_conn, sslcert), true, NULL, NULL }, + { "sslkey", offsetof(struct pg_conn, sslkey), true, NULL, NULL }, + { "sslrootcert", offsetof(struct pg_conn, sslrootcert), true, NULL, NULL }, + { "sslcrl", offsetof(struct pg_conn, sslcrl), true, NULL, NULL }, + { "requirepeer", offsetof(struct pg_conn, requirepeer), true, NULL, NULL }, +#if defined(KRB5) || defined(ENABLE_GSS) || defined(ENABLE_SSPI) + { "krbsrvname", offsetof(struct pg_conn, krbsrvname), true, NULL, NULL }, +#endif +#if defined(ENABLE_GSS) && defined(ENABLE_SSPI) + { "gsslib", offsetof(struct pg_conn, requirepeer), true, NULL, NULL }, +#endif + { "replication", offsetof(struct pg_conn, replication), false, NULL, NULL }, + /* Terminating entry --- MUST BE LAST */ + { NULL, 0, false, NULL, NULL } +}; + static const PQEnvironmentOption EnvironmentOptions[] = { /* common user-interface settings */ @@ -295,7 +358,8 @@ static PGconn *makeEmptyPGconn(void); static void fillPGconn(PGconn *conn, PQconninfoOption *connOptions); static void freePGconn(PGconn *conn); static void closePGconn(PGconn *conn); -static PQconninfoOption *conninfo_init(PQExpBuffer errorMessage); +static PQconninfoOption *conninfo_init(PQExpBuffer errorMessage, + bool for_replication); static PQconninfoOption *parse_connection_string(const char *conninfo, PQExpBuffer errorMessage, bool use_defaults); static int uri_prefix_length(const char *connstr); @@ -318,6 +382,8 @@ static char *conninfo_uri_decode(const c static bool get_hexdigit(char digit, int *value); static const char *conninfo_getval(PQconninfoOption *connOptions, const char *keyword); +static void conninfo_setval(PQconninfoOption *connOptions, + const char *keyword, const char *val); static PQconninfoOption *conninfo_storeval(PQconninfoOption *connOptions, const char *keyword, const char *value, PQExpBuffer errorMessage, bool ignoreMissing, bool uri_decode); @@ -627,7 +693,9 @@ PQconnectStart(const char *conninfo) static void fillPGconn(PGconn *conn, PQconninfoOption *connOptions) { + const PQconninfoMapping *mapping; const char *tmp; + char **memberaddr; /* * Move option values into conn structure @@ -637,72 +705,24 @@ fillPGconn(PGconn *conn, PQconninfoOptio * * XXX: probably worth checking strdup() return value here... */ - tmp = conninfo_getval(connOptions, "hostaddr"); - conn->pghostaddr = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "host"); - conn->pghost = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "port"); - conn->pgport = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "tty"); - conn->pgtty = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "options"); - conn->pgoptions = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "application_name"); - conn->appname = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "fallback_application_name"); - conn->fbappname = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "dbname"); - conn->dbName = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "user"); - conn->pguser = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "password"); - conn->pgpass = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "connect_timeout"); - conn->connect_timeout = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "client_encoding"); - conn->client_encoding_initial = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "keepalives"); - conn->keepalives = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "keepalives_idle"); - conn->keepalives_idle = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "keepalives_interval"); - conn->keepalives_interval = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "keepalives_count"); - conn->keepalives_count = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "sslmode"); - conn->sslmode = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "sslcompression"); - conn->sslcompression = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "sslkey"); - conn->sslkey = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "sslcert"); - conn->sslcert = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "sslrootcert"); - conn->sslrootcert = tmp ? strdup(tmp) : NULL; - tmp = conninfo_getval(connOptions, "sslcrl"); - conn->sslcrl = tmp ? strdup(tmp) : NULL; -#ifdef USE_SSL - tmp = conninfo_getval(connOptions, "requiressl"); - if (tmp && tmp[0] == '1') + for (mapping = PQconninfoMappings; mapping->keyword; mapping++) { - /* here warn that the requiressl option is deprecated? */ - if (conn->sslmode) - free(conn->sslmode); - conn->sslmode = strdup("require"); + tmp = conninfo_getval(connOptions, mapping->keyword); + memberaddr = PGCONNMEMBERADDR(conn, mapping); + + if (mapping->conninfoValue && mapping->connValue) + { + size_t len = strlen(mapping->conninfoValue); + if (tmp && strncmp(tmp, mapping->conninfoValue, len) == 0) + { + if (*memberaddr) + free(*memberaddr); + *memberaddr = strdup(mapping->connValue); + } + } + else + *memberaddr = tmp ? strdup(tmp) : NULL; } -#endif - tmp = conninfo_getval(connOptions, "requirepeer"); - conn->requirepeer = tmp ? strdup(tmp) : NULL; -#if defined(KRB5) || defined(ENABLE_GSS) || defined(ENABLE_SSPI) - tmp = conninfo_getval(connOptions, "krbsrvname"); - conn->krbsrvname = tmp ? strdup(tmp) : NULL; -#endif -#if defined(ENABLE_GSS) && defined(ENABLE_SSPI) - tmp = conninfo_getval(connOptions, "gsslib"); - conn->gsslib = tmp ? strdup(tmp) : NULL; -#endif - tmp = conninfo_getval(connOptions, "replication"); - conn->replication = tmp ? strdup(tmp) : NULL; } /* @@ -884,7 +904,7 @@ PQconndefaults(void) if (PQExpBufferDataBroken(errorBuf)) return NULL; /* out of memory already :-( */ - connOptions = conninfo_init(&errorBuf); + connOptions = conninfo_init(&errorBuf, false); if (connOptions != NULL) { if (!conninfo_add_defaults(connOptions, &errorBuf)) @@ -4006,9 +4026,11 @@ PQconninfoParse(const char *conninfo, ch /* * Build a working copy of the constant PQconninfoOptions array. + * If for_replication is true, only return the options that are + * not added by libpqwalreceiver. */ static PQconninfoOption * -conninfo_init(PQExpBuffer errorMessage) +conninfo_init(PQExpBuffer errorMessage, bool for_replication) { PQconninfoOption *options; @@ -4019,7 +4041,28 @@ conninfo_init(PQExpBuffer errorMessage) libpq_gettext("out of memory\n")); return NULL; } - memcpy(options, PQconninfoOptions, sizeof(PQconninfoOptions)); + if (for_replication) + { + const PQconninfoMapping *mapping = PQconninfoMappings; + PQconninfoOption *opt_dest = options; + + while (mapping->keyword) + { + PQconninfoOption *opt_src = conninfo_find((PQconninfoOption *)PQconninfoOptions, mapping->keyword); + + if (opt_src && mapping->for_replication) + { + memcpy(opt_dest, opt_src, sizeof(PQconninfoOption)); + opt_dest++; + } + + opt_src++; + mapping++; + } + MemSet(opt_dest, 0, sizeof(PQconninfoOption)); + } + else + memcpy(options, PQconninfoOptions, sizeof(PQconninfoOptions)); return options; } @@ -4095,7 +4138,7 @@ conninfo_parse(const char *conninfo, PQE PQconninfoOption *options; /* Make a working copy of PQconninfoOptions */ - options = conninfo_init(errorMessage); + options = conninfo_init(errorMessage, false); if (options == NULL) return NULL; @@ -4295,7 +4338,7 @@ conninfo_array_parse(const char *const * } /* Make a working copy of PQconninfoOptions */ - options = conninfo_init(errorMessage); + options = conninfo_init(errorMessage, false); if (options == NULL) { PQconninfoFree(dbname_options); @@ -4485,7 +4528,7 @@ conninfo_uri_parse(const char *uri, PQEx PQconninfoOption *options; /* Make a working copy of PQconninfoOptions */ - options = conninfo_init(errorMessage); + options = conninfo_init(errorMessage, false); if (options == NULL) return NULL; @@ -4985,6 +5028,24 @@ conninfo_getval(PQconninfoOption *connOp } /* + * Set an option value corresponding to the keyword in the connOptions array. + */ +static void +conninfo_setval(PQconninfoOption *connOptions, const char *keyword, + const char *val) +{ + PQconninfoOption *option; + + option = conninfo_find(connOptions, keyword); + if (option) + { + if (option->val) + free(option->val); + option->val = val ? strdup(val) : NULL; + } +} + +/* * Store a (new) value for an option corresponding to the keyword in * connOptions array. * @@ -5066,6 +5127,50 @@ conninfo_find(PQconninfoOption *connOpti } +/* + * Return the connection options used for the connections + */ +PQconninfoOption * +PQconninfo(PGconn *conn, bool for_replication) +{ + PQExpBufferData errorBuf; + PQconninfoOption *connOptions; + + if (conn == NULL) + return NULL; + + /* We don't actually report any errors here, but callees want a buffer */ + initPQExpBuffer(&errorBuf); + if (PQExpBufferDataBroken(errorBuf)) + return NULL; /* out of memory already :-( */ + + connOptions = conninfo_init(&errorBuf, for_replication); + + termPQExpBuffer(&errorBuf); + + if (connOptions != NULL) + { + const PQconninfoMapping *mapping; + + for (mapping = PQconninfoMappings; mapping->keyword; mapping++) + { + char **memberaddr = PGCONNMEMBERADDR(conn, mapping); + + if (mapping->conninfoValue && mapping->connValue) + { + size_t len = strlen(mapping->connValue); + if (*memberaddr && strncmp(*memberaddr, mapping->connValue, len) == 0) + conninfo_setval(connOptions, mapping->keyword, mapping->conninfoValue); + } + else + conninfo_setval(connOptions, mapping->keyword, *memberaddr); + } + } + + return connOptions; +} + + void PQconninfoFree(PQconninfoOption *connOptions) { diff -durpN postgresql/src/interfaces/libpq/libpq-fe.h postgresql.1/src/interfaces/libpq/libpq-fe.h --- postgresql/src/interfaces/libpq/libpq-fe.h 2012-10-09 09:58:14.343782980 +0200 +++ postgresql.1/src/interfaces/libpq/libpq-fe.h 2012-10-10 14:09:31.936890314 +0200 @@ -262,6 +262,9 @@ extern PQconninfoOption *PQconndefaults( /* parse connection options in same way as PQconnectdb */ extern PQconninfoOption *PQconninfoParse(const char *conninfo, char **errmsg); +/* return the connection options used by a live connection */ +extern PQconninfoOption *PQconninfo(PGconn *conn, bool for_replication); + /* free the data structure returned by PQconndefaults() or PQconninfoParse() */ extern void PQconninfoFree(PQconninfoOption *connOptions);