mirror of https://github.com/postgres/postgres
ConnectDatabase is used by pg_dumpall, pg_restore and pg_dump so move common code to new file. new file name: connectdb.c Author: Mahendra Singh Thalor <mahi6run@gmail.com>pull/210/head
Andrew Dunstan
6 months ago
parent
ff3a7f0b68
commit
c1da728106
@ -0,0 +1,294 @@ |
||||
/*-------------------------------------------------------------------------
|
||||
* |
||||
* connectdb.c |
||||
* This is a common file connection to the database. |
||||
* |
||||
* Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group |
||||
* Portions Copyright (c) 1994, Regents of the University of California |
||||
* |
||||
* IDENTIFICATION |
||||
* src/bin/pg_dump/connectdb.c |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
|
||||
#include "postgres_fe.h" |
||||
|
||||
#include "common/connect.h" |
||||
#include "common/logging.h" |
||||
#include "common/string.h" |
||||
#include "connectdb.h" |
||||
#include "dumputils.h" |
||||
#include "fe_utils/string_utils.h" |
||||
|
||||
static char *constructConnStr(const char **keywords, const char **values); |
||||
|
||||
/*
|
||||
* ConnectDatabase |
||||
* |
||||
* Make a database connection with the given parameters. An |
||||
* interactive password prompt is automatically issued if required. |
||||
* |
||||
* If fail_on_error is false, we return NULL without printing any message |
||||
* on failure, but preserve any prompted password for the next try. |
||||
* |
||||
* On success, the 'connstr' is set to a connection string containing |
||||
* the options used and 'server_version' is set to version so that caller |
||||
* can use them. |
||||
*/ |
||||
PGconn * |
||||
ConnectDatabase(const char *dbname, const char *connection_string, |
||||
const char *pghost, const char *pgport, const char *pguser, |
||||
trivalue prompt_password, bool fail_on_error, const char *progname, |
||||
const char **connstr, int *server_version, char *password, |
||||
char *override_dbname) |
||||
{ |
||||
PGconn *conn; |
||||
bool new_pass; |
||||
const char *remoteversion_str; |
||||
int my_version; |
||||
const char **keywords = NULL; |
||||
const char **values = NULL; |
||||
PQconninfoOption *conn_opts = NULL; |
||||
int server_version_temp; |
||||
|
||||
if (prompt_password == TRI_YES && !password) |
||||
password = simple_prompt("Password: ", false); |
||||
|
||||
/*
|
||||
* Start the connection. Loop until we have a password if requested by |
||||
* backend. |
||||
*/ |
||||
do |
||||
{ |
||||
int argcount = 8; |
||||
PQconninfoOption *conn_opt; |
||||
char *err_msg = NULL; |
||||
int i = 0; |
||||
|
||||
free(keywords); |
||||
free(values); |
||||
PQconninfoFree(conn_opts); |
||||
|
||||
/*
|
||||
* Merge the connection info inputs given in form of connection string |
||||
* and other options. Explicitly discard any dbname value in the |
||||
* connection string; otherwise, PQconnectdbParams() would interpret |
||||
* that value as being itself a connection string. |
||||
*/ |
||||
if (connection_string) |
||||
{ |
||||
conn_opts = PQconninfoParse(connection_string, &err_msg); |
||||
if (conn_opts == NULL) |
||||
pg_fatal("%s", err_msg); |
||||
|
||||
for (conn_opt = conn_opts; conn_opt->keyword != NULL; conn_opt++) |
||||
{ |
||||
if (conn_opt->val != NULL && conn_opt->val[0] != '\0' && |
||||
strcmp(conn_opt->keyword, "dbname") != 0) |
||||
argcount++; |
||||
} |
||||
|
||||
keywords = pg_malloc0((argcount + 1) * sizeof(*keywords)); |
||||
values = pg_malloc0((argcount + 1) * sizeof(*values)); |
||||
|
||||
for (conn_opt = conn_opts; conn_opt->keyword != NULL; conn_opt++) |
||||
{ |
||||
if (conn_opt->val != NULL && conn_opt->val[0] != '\0' && |
||||
strcmp(conn_opt->keyword, "dbname") != 0) |
||||
{ |
||||
keywords[i] = conn_opt->keyword; |
||||
values[i] = conn_opt->val; |
||||
i++; |
||||
} |
||||
} |
||||
} |
||||
else |
||||
{ |
||||
keywords = pg_malloc0((argcount + 1) * sizeof(*keywords)); |
||||
values = pg_malloc0((argcount + 1) * sizeof(*values)); |
||||
} |
||||
|
||||
if (pghost) |
||||
{ |
||||
keywords[i] = "host"; |
||||
values[i] = pghost; |
||||
i++; |
||||
} |
||||
if (pgport) |
||||
{ |
||||
keywords[i] = "port"; |
||||
values[i] = pgport; |
||||
i++; |
||||
} |
||||
if (pguser) |
||||
{ |
||||
keywords[i] = "user"; |
||||
values[i] = pguser; |
||||
i++; |
||||
} |
||||
if (password) |
||||
{ |
||||
keywords[i] = "password"; |
||||
values[i] = password; |
||||
i++; |
||||
} |
||||
if (dbname) |
||||
{ |
||||
keywords[i] = "dbname"; |
||||
values[i] = dbname; |
||||
i++; |
||||
} |
||||
if (override_dbname) |
||||
{ |
||||
keywords[i] = "dbname"; |
||||
values[i++] = override_dbname; |
||||
} |
||||
|
||||
keywords[i] = "fallback_application_name"; |
||||
values[i] = progname; |
||||
i++; |
||||
|
||||
new_pass = false; |
||||
conn = PQconnectdbParams(keywords, values, true); |
||||
|
||||
if (!conn) |
||||
pg_fatal("could not connect to database \"%s\"", dbname); |
||||
|
||||
if (PQstatus(conn) == CONNECTION_BAD && |
||||
PQconnectionNeedsPassword(conn) && |
||||
!password && |
||||
prompt_password != TRI_NO) |
||||
{ |
||||
PQfinish(conn); |
||||
password = simple_prompt("Password: ", false); |
||||
new_pass = true; |
||||
} |
||||
} while (new_pass); |
||||
|
||||
/* check to see that the backend connection was successfully made */ |
||||
if (PQstatus(conn) == CONNECTION_BAD) |
||||
{ |
||||
if (fail_on_error) |
||||
pg_fatal("%s", PQerrorMessage(conn)); |
||||
else |
||||
{ |
||||
PQfinish(conn); |
||||
|
||||
free(keywords); |
||||
free(values); |
||||
PQconninfoFree(conn_opts); |
||||
|
||||
return NULL; |
||||
} |
||||
} |
||||
|
||||
/*
|
||||
* Ok, connected successfully. If requested, remember the options used, in |
||||
* the form of a connection string. |
||||
*/ |
||||
if (connstr) |
||||
*connstr = constructConnStr(keywords, values); |
||||
|
||||
free(keywords); |
||||
free(values); |
||||
PQconninfoFree(conn_opts); |
||||
|
||||
/* Check version */ |
||||
remoteversion_str = PQparameterStatus(conn, "server_version"); |
||||
if (!remoteversion_str) |
||||
pg_fatal("could not get server version"); |
||||
|
||||
server_version_temp = PQserverVersion(conn); |
||||
if (server_version_temp == 0) |
||||
pg_fatal("could not parse server version \"%s\"", |
||||
remoteversion_str); |
||||
|
||||
/* If requested, then copy server version to out variable. */ |
||||
if (server_version) |
||||
*server_version = server_version_temp; |
||||
|
||||
my_version = PG_VERSION_NUM; |
||||
|
||||
/*
|
||||
* We allow the server to be back to 9.2, and up to any minor release of |
||||
* our own major version. (See also version check in pg_dump.c.) |
||||
*/ |
||||
if (my_version != server_version_temp |
||||
&& (server_version_temp < 90200 || |
||||
(server_version_temp / 100) > (my_version / 100))) |
||||
{ |
||||
pg_log_error("aborting because of server version mismatch"); |
||||
pg_log_error_detail("server version: %s; %s version: %s", |
||||
remoteversion_str, progname, PG_VERSION); |
||||
exit_nicely(1); |
||||
} |
||||
|
||||
PQclear(executeQuery(conn, ALWAYS_SECURE_SEARCH_PATH_SQL)); |
||||
|
||||
return conn; |
||||
} |
||||
|
||||
/*
|
||||
* constructConnStr |
||||
* |
||||
* Construct a connection string from the given keyword/value pairs. It is |
||||
* used to pass the connection options to the pg_dump subprocess. |
||||
* |
||||
* The following parameters are excluded: |
||||
* dbname - varies in each pg_dump invocation |
||||
* password - it's not secure to pass a password on the command line |
||||
* fallback_application_name - we'll let pg_dump set it |
||||
*/ |
||||
static char * |
||||
constructConnStr(const char **keywords, const char **values) |
||||
{ |
||||
PQExpBuffer buf = createPQExpBuffer(); |
||||
char *connstr; |
||||
int i; |
||||
bool firstkeyword = true; |
||||
|
||||
/* Construct a new connection string in key='value' format. */ |
||||
for (i = 0; keywords[i] != NULL; i++) |
||||
{ |
||||
if (strcmp(keywords[i], "dbname") == 0 || |
||||
strcmp(keywords[i], "password") == 0 || |
||||
strcmp(keywords[i], "fallback_application_name") == 0) |
||||
continue; |
||||
|
||||
if (!firstkeyword) |
||||
appendPQExpBufferChar(buf, ' '); |
||||
firstkeyword = false; |
||||
appendPQExpBuffer(buf, "%s=", keywords[i]); |
||||
appendConnStrVal(buf, values[i]); |
||||
} |
||||
|
||||
connstr = pg_strdup(buf->data); |
||||
destroyPQExpBuffer(buf); |
||||
return connstr; |
||||
} |
||||
|
||||
/*
|
||||
* executeQuery |
||||
* |
||||
* Run a query, return the results, exit program on failure. |
||||
*/ |
||||
PGresult * |
||||
executeQuery(PGconn *conn, const char *query) |
||||
{ |
||||
PGresult *res; |
||||
|
||||
pg_log_info("executing %s", query); |
||||
|
||||
res = PQexec(conn, query); |
||||
if (!res || |
||||
PQresultStatus(res) != PGRES_TUPLES_OK) |
||||
{ |
||||
pg_log_error("query failed: %s", PQerrorMessage(conn)); |
||||
pg_log_error_detail("Query was: %s", query); |
||||
PQfinish(conn); |
||||
exit_nicely(1); |
||||
} |
||||
|
||||
return res; |
||||
} |
||||
@ -0,0 +1,26 @@ |
||||
/*-------------------------------------------------------------------------
|
||||
* |
||||
* connectdb.h |
||||
* Common header file for connection to the database. |
||||
* |
||||
* Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group |
||||
* Portions Copyright (c) 1994, Regents of the University of California |
||||
* |
||||
* IDENTIFICATION |
||||
* src/bin/pg_dump/connectdb.h |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
#ifndef CONNECTDB_H |
||||
#define CONNECTDB_H |
||||
|
||||
#include "pg_backup.h" |
||||
#include "pg_backup_utils.h" |
||||
|
||||
extern PGconn *ConnectDatabase(const char *dbname, const char *connection_string, const char *pghost, |
||||
const char *pgport, const char *pguser, |
||||
trivalue prompt_password, bool fail_on_error, |
||||
const char *progname, const char **connstr, int *server_version, |
||||
char *password, char *override_dbname); |
||||
extern PGresult *executeQuery(PGconn *conn, const char *query); |
||||
#endif /* CONNECTDB_H */ |
||||
Loading…
Reference in new issue