]> git.ipfire.org Git - thirdparty/postgresql.git/commitdiff
Fix connection string handling in src/bin/scripts/ programs.
authorTom Lane <tgl@sss.pgh.pa.us>
Mon, 19 Oct 2020 23:03:46 +0000 (19:03 -0400)
committerTom Lane <tgl@sss.pgh.pa.us>
Mon, 19 Oct 2020 23:03:46 +0000 (19:03 -0400)
When told to process all databases, clusterdb, reindexdb, and vacuumdb
would reconnect by replacing their --maintenance-db parameter with the
name of the target database.  If that parameter is a connstring (which
has been allowed for a long time, though we failed to document that
before this patch), we'd lose any other options it might specify, for
example SSL or GSS parameters, possibly resulting in failure to connect.
Thus, this is the same bug as commit a45bc8a4f fixed in pg_dump and
pg_restore.  We can fix it in the same way, by using libpq's rules for
handling multiple "dbname" parameters to add the target database name
separately.  I chose to apply the same refactoring approach as in that
patch, with a struct to handle the command line parameters that need to
be passed through to connectDatabase.  (Maybe someday we can unify the
very similar functions here and in pg_dump/pg_restore.)

Per Peter Eisentraut's comments on bug #16604.  Back-patch to all
supported branches.

Discussion: https://postgr.es/m/16604-933f4b8791227b15@postgresql.org

16 files changed:
doc/src/sgml/ref/clusterdb.sgml
doc/src/sgml/ref/createdb.sgml
doc/src/sgml/ref/dropdb.sgml
doc/src/sgml/ref/reindexdb.sgml
doc/src/sgml/ref/vacuumdb.sgml
src/bin/scripts/clusterdb.c
src/bin/scripts/common.c
src/bin/scripts/common.h
src/bin/scripts/createdb.c
src/bin/scripts/createuser.c
src/bin/scripts/dropdb.c
src/bin/scripts/dropuser.c
src/bin/scripts/reindexdb.c
src/bin/scripts/scripts_parallel.c
src/bin/scripts/scripts_parallel.h
src/bin/scripts/vacuumdb.c

index c53bacf8648c08e36b23813b173fa529b05b0519..c838b22c4405349928125534867592a5b1fab1d8 100644 (file)
@@ -90,9 +90,9 @@ PostgreSQL documentation
       <term><option><optional>--dbname=</optional><replaceable class="parameter">dbname</replaceable></option></term>
       <listitem>
        <para>
-        Specifies the name of the database to be clustered.
-        If this is not specified and <option>-a</option> (or
-        <option>--all</option>) is not used, the database name is read
+        Specifies the name of the database to be clustered,
+        when <option>-a</option>/<option>--all</option> is not used.
+        If this is not specified, the database name is read
         from the environment variable <envar>PGDATABASE</envar>.  If
         that is not set, the user name specified for the connection is
         used.  The <replaceable>dbname</replaceable> can be a <link
@@ -249,10 +249,16 @@ PostgreSQL documentation
       <term><option>--maintenance-db=<replaceable class="parameter">dbname</replaceable></option></term>
       <listitem>
        <para>
-         Specifies the name of the database to connect to discover what other
-         databases should be clustered. If not specified, the
-         <literal>postgres</literal> database will be used,
-         and if that does not exist, <literal>template1</literal> will be used.
+        Specifies the name of the database to connect to to discover which
+        databases should be clustered,
+        when <option>-a</option>/<option>--all</option> is used.
+        If not specified, the <literal>postgres</literal> database will be used,
+        or if that does not exist, <literal>template1</literal> will be used.
+        This can be a <link linkend="libpq-connstring">connection
+        string</link>.  If so, connection string parameters will override any
+        conflicting command line options.  Also, connection string parameters
+        other than the database name itself will be re-used when connecting
+        to other databases.
        </para>
       </listitem>
      </varlistentry>
index d3c92943f0714f46a1ce3c915aaebd0671858e6e..c4f7e12a0862de3ff768a48722948cdbe5ef14fc 100644 (file)
@@ -284,6 +284,9 @@ PostgreSQL documentation
          database will be used; if that does not exist (or if it is the name
          of the new database being created), <literal>template1</literal> will
          be used.
+         This can be a <link linkend="libpq-connstring">connection
+         string</link>.  If so, connection string parameters will override any
+         conflicting command line options.
        </para>
       </listitem>
      </varlistentry>
index ded85b0e232dfd05f949bd213307cc340287d498..70dbf615b61e246f140ac2c546aa6211974c5367 100644 (file)
@@ -217,6 +217,9 @@ PostgreSQL documentation
          target database. If not specified, the <literal>postgres</literal>
          database will be used; if that does not exist (or is the database
          being dropped), <literal>template1</literal> will be used.
+         This can be a <link linkend="libpq-connstring">connection
+         string</link>.  If so, connection string parameters will override any
+         conflicting command line options.
        </para>
       </listitem>
      </varlistentry>
index 0f3f12bfbfc65fb31a4022c52b40a86693a87064..09ad8cd6c41701044bd88c1578ae357231b7bfca 100644 (file)
@@ -134,9 +134,9 @@ PostgreSQL documentation
       <term><option><optional>--dbname=</optional><replaceable class="parameter">dbname</replaceable></option></term>
       <listitem>
        <para>
-        Specifies the name of the database to be reindexed.
-        If this is not specified and <option>-a</option> (or
-        <option>--all</option>) is not used, the database name is read
+        Specifies the name of the database to be reindexed,
+        when <option>-a</option>/<option>--all</option> is not used.
+        If this is not specified, the database name is read
         from the environment variable <envar>PGDATABASE</envar>.  If
         that is not set, the user name specified for the connection is
         used.  The <replaceable>dbname</replaceable> can be a <link
@@ -351,10 +351,16 @@ PostgreSQL documentation
       <term><option>--maintenance-db=<replaceable class="parameter">dbname</replaceable></option></term>
       <listitem>
        <para>
-         Specifies the name of the database to connect to discover what other
-         databases should be reindexed. If not specified, the
-         <literal>postgres</literal> database will be used,
-         and if that does not exist, <literal>template1</literal> will be used.
+        Specifies the name of the database to connect to to discover which
+        databases should be reindexed,
+        when <option>-a</option>/<option>--all</option> is used.
+        If not specified, the <literal>postgres</literal> database will be used,
+        or if that does not exist, <literal>template1</literal> will be used.
+        This can be a <link linkend="libpq-connstring">connection
+        string</link>.  If so, connection string parameters will override any
+        conflicting command line options.  Also, connection string parameters
+        other than the database name itself will be re-used when connecting
+        to other databases.
        </para>
       </listitem>
      </varlistentry>
index e8e3166024a0a1d8baff22c4b6cd3771f28464a1..ace5d2a6e489356bacb066adbfb1226d21bc3385 100644 (file)
@@ -92,9 +92,9 @@ PostgreSQL documentation
       <term><option><optional>--dbname=</optional><replaceable class="parameter">dbname</replaceable></option></term>
       <listitem>
        <para>
-        Specifies the name of the database to be cleaned or analyzed.
-        If this is not specified and <option>-a</option> (or
-        <option>--all</option>) is not used, the database name is read
+        Specifies the name of the database to be cleaned or analyzed,
+        when <option>-a</option>/<option>--all</option> is not used.
+        If this is not specified, the database name is read
         from the environment variable <envar>PGDATABASE</envar>.  If
         that is not set, the user name specified for the connection is
         used.  The <replaceable>dbname</replaceable> can be a <link
@@ -444,10 +444,16 @@ PostgreSQL documentation
       <term><option>--maintenance-db=<replaceable class="parameter">dbname</replaceable></option></term>
       <listitem>
        <para>
-         Specifies the name of the database to connect to discover what other
-         databases should be vacuumed. If not specified, the
-         <literal>postgres</literal> database will be used,
-         and if that does not exist, <literal>template1</literal> will be used.
+        Specifies the name of the database to connect to to discover which
+        databases should be vacuumed,
+        when <option>-a</option>/<option>--all</option> is used.
+        If not specified, the <literal>postgres</literal> database will be used,
+        or if that does not exist, <literal>template1</literal> will be used.
+        This can be a <link linkend="libpq-connstring">connection
+        string</link>.  If so, connection string parameters will override any
+        conflicting command line options.  Also, connection string parameters
+        other than the database name itself will be re-used when connecting
+        to other databases.
        </para>
       </listitem>
      </varlistentry>
index 12972de0e91e74c1789743f4a3bb93664b4de148..2f786e61037be7470ea5ff2f73a77c447fd3e96d 100644 (file)
 #include "fe_utils/string_utils.h"
 
 
-static void cluster_one_database(const char *dbname, bool verbose, const char *table,
-                                                                const char *host, const char *port,
-                                                                const char *username, enum trivalue prompt_password,
-                                                                const char *progname, bool echo);
-static void cluster_all_databases(bool verbose, const char *maintenance_db,
-                                                                 const char *host, const char *port,
-                                                                 const char *username, enum trivalue prompt_password,
-                                                                 const char *progname, bool echo, bool quiet);
-
+static void cluster_one_database(const ConnParams *cparams, const char *table,
+                                                                const char *progname, bool verbose, bool echo);
+static void cluster_all_databases(ConnParams *cparams, const char *progname,
+                                                                 bool verbose, bool echo, bool quiet);
 static void help(const char *progname);
 
 
@@ -58,6 +53,7 @@ main(int argc, char *argv[])
        char       *port = NULL;
        char       *username = NULL;
        enum trivalue prompt_password = TRI_DEFAULT;
+       ConnParams      cparams;
        bool            echo = false;
        bool            quiet = false;
        bool            alldb = false;
@@ -134,6 +130,13 @@ main(int argc, char *argv[])
                exit(1);
        }
 
+       /* fill cparams except for dbname, which is set below */
+       cparams.pghost = host;
+       cparams.pgport = port;
+       cparams.pguser = username;
+       cparams.prompt_password = prompt_password;
+       cparams.override_dbname = NULL;
+
        setup_cancel_handler(NULL);
 
        if (alldb)
@@ -150,8 +153,9 @@ main(int argc, char *argv[])
                        exit(1);
                }
 
-               cluster_all_databases(verbose, maintenance_db, host, port, username, prompt_password,
-                                                         progname, echo, quiet);
+               cparams.dbname = maintenance_db;
+
+               cluster_all_databases(&cparams, progname, verbose, echo, quiet);
        }
        else
        {
@@ -165,21 +169,21 @@ main(int argc, char *argv[])
                                dbname = get_user_name_or_exit(progname);
                }
 
+               cparams.dbname = dbname;
+
                if (tables.head != NULL)
                {
                        SimpleStringListCell *cell;
 
                        for (cell = tables.head; cell; cell = cell->next)
                        {
-                               cluster_one_database(dbname, verbose, cell->val,
-                                                                        host, port, username, prompt_password,
-                                                                        progname, echo);
+                               cluster_one_database(&cparams, cell->val,
+                                                                        progname, verbose, echo);
                        }
                }
                else
-                       cluster_one_database(dbname, verbose, NULL,
-                                                                host, port, username, prompt_password,
-                                                                progname, echo);
+                       cluster_one_database(&cparams, NULL,
+                                                                progname, verbose, echo);
        }
 
        exit(0);
@@ -187,17 +191,14 @@ main(int argc, char *argv[])
 
 
 static void
-cluster_one_database(const char *dbname, bool verbose, const char *table,
-                                        const char *host, const char *port,
-                                        const char *username, enum trivalue prompt_password,
-                                        const char *progname, bool echo)
+cluster_one_database(const ConnParams *cparams, const char *table,
+                                        const char *progname, bool verbose, bool echo)
 {
        PQExpBufferData sql;
 
        PGconn     *conn;
 
-       conn = connectDatabase(dbname, host, port, username, prompt_password,
-                                                  progname, echo, false, false);
+       conn = connectDatabase(cparams, progname, echo, false, false);
 
        initPQExpBuffer(&sql);
 
@@ -228,22 +229,17 @@ cluster_one_database(const char *dbname, bool verbose, const char *table,
 
 
 static void
-cluster_all_databases(bool verbose, const char *maintenance_db,
-                                         const char *host, const char *port,
-                                         const char *username, enum trivalue prompt_password,
-                                         const char *progname, bool echo, bool quiet)
+cluster_all_databases(ConnParams *cparams, const char *progname,
+                                         bool verbose, bool echo, bool quiet)
 {
        PGconn     *conn;
        PGresult   *result;
-       PQExpBufferData connstr;
        int                     i;
 
-       conn = connectMaintenanceDatabase(maintenance_db, host, port, username,
-                                                                         prompt_password, progname, echo);
+       conn = connectMaintenanceDatabase(cparams, progname, echo);
        result = executeQuery(conn, "SELECT datname FROM pg_database WHERE datallowconn ORDER BY 1;", echo);
        PQfinish(conn);
 
-       initPQExpBuffer(&connstr);
        for (i = 0; i < PQntuples(result); i++)
        {
                char       *dbname = PQgetvalue(result, i, 0);
@@ -254,15 +250,10 @@ cluster_all_databases(bool verbose, const char *maintenance_db,
                        fflush(stdout);
                }
 
-               resetPQExpBuffer(&connstr);
-               appendPQExpBufferStr(&connstr, "dbname=");
-               appendConnStrVal(&connstr, dbname);
+               cparams->override_dbname = dbname;
 
-               cluster_one_database(connstr.data, verbose, NULL,
-                                                        host, port, username, prompt_password,
-                                                        progname, echo);
+               cluster_one_database(cparams, NULL, progname, verbose, echo);
        }
-       termPQExpBuffer(&connstr);
 
        PQclear(result);
 }
index 420d0d11a5a1595212fdf916374a51eb39d5dcc1..afa62e4da73fb98fa9cb9fafbb3992523ab526db 100644 (file)
@@ -53,7 +53,7 @@ handle_help_version_opts(int argc, char *argv[],
  * Make a database connection with the given parameters.
  *
  * An interactive password prompt is automatically issued if needed and
- * allowed by prompt_password.
+ * allowed by cparams->prompt_password.
  *
  * If allow_password_reuse is true, we will try to re-use any password
  * given during previous calls to this routine.  (Callers should not pass
@@ -61,9 +61,7 @@ handle_help_version_opts(int argc, char *argv[],
  * as before, else we might create password exposure hazards.)
  */
 PGconn *
-connectDatabase(const char *dbname, const char *pghost,
-                               const char *pgport, const char *pguser,
-                               enum trivalue prompt_password, const char *progname,
+connectDatabase(const ConnParams *cparams, const char *progname,
                                bool echo, bool fail_ok, bool allow_password_reuse)
 {
        PGconn     *conn;
@@ -71,10 +69,13 @@ connectDatabase(const char *dbname, const char *pghost,
        static bool have_password = false;
        static char password[100];
 
+       /* Callers must supply at least dbname; other params can be NULL */
+       Assert(cparams->dbname);
+
        if (!allow_password_reuse)
                have_password = false;
 
-       if (!have_password && prompt_password == TRI_YES)
+       if (cparams->prompt_password == TRI_YES && !have_password)
        {
                simple_prompt("Password: ", password, sizeof(password), false);
                have_password = true;
@@ -86,23 +87,35 @@ connectDatabase(const char *dbname, const char *pghost,
         */
        do
        {
-               const char *keywords[7];
-               const char *values[7];
-
-               keywords[0] = "host";
-               values[0] = pghost;
-               keywords[1] = "port";
-               values[1] = pgport;
-               keywords[2] = "user";
-               values[2] = pguser;
-               keywords[3] = "password";
-               values[3] = have_password ? password : NULL;
-               keywords[4] = "dbname";
-               values[4] = dbname;
-               keywords[5] = "fallback_application_name";
-               values[5] = progname;
-               keywords[6] = NULL;
-               values[6] = NULL;
+               const char *keywords[8];
+               const char *values[8];
+               int                     i = 0;
+
+               /*
+                * If dbname is a connstring, its entries can override the other
+                * values obtained from cparams; but in turn, override_dbname can
+                * override the dbname component of it.
+                */
+               keywords[i] = "host";
+               values[i++] = cparams->pghost;
+               keywords[i] = "port";
+               values[i++] = cparams->pgport;
+               keywords[i] = "user";
+               values[i++] = cparams->pguser;
+               keywords[i] = "password";
+               values[i++] = have_password ? password : NULL;
+               keywords[i] = "dbname";
+               values[i++] = cparams->dbname;
+               if (cparams->override_dbname)
+               {
+                       keywords[i] = "dbname";
+                       values[i++] = cparams->override_dbname;
+               }
+               keywords[i] = "fallback_application_name";
+               values[i++] = progname;
+               keywords[i] = NULL;
+               values[i++] = NULL;
+               Assert(i <= lengthof(keywords));
 
                new_pass = false;
                conn = PQconnectdbParams(keywords, values, true);
@@ -110,7 +123,7 @@ connectDatabase(const char *dbname, const char *pghost,
                if (!conn)
                {
                        pg_log_error("could not connect to database %s: out of memory",
-                                                dbname);
+                                                cparams->dbname);
                        exit(1);
                }
 
@@ -119,7 +132,7 @@ connectDatabase(const char *dbname, const char *pghost,
                 */
                if (PQstatus(conn) == CONNECTION_BAD &&
                        PQconnectionNeedsPassword(conn) &&
-                       prompt_password != TRI_NO)
+                       cparams->prompt_password != TRI_NO)
                {
                        PQfinish(conn);
                        simple_prompt("Password: ", password, sizeof(password), false);
@@ -137,10 +150,11 @@ connectDatabase(const char *dbname, const char *pghost,
                        return NULL;
                }
                pg_log_error("could not connect to database %s: %s",
-                                        dbname, PQerrorMessage(conn));
+                                        cparams->dbname, PQerrorMessage(conn));
                exit(1);
        }
 
+       /* Start strict; callers may override this. */
        PQclear(executeQuery(conn, ALWAYS_SECURE_SEARCH_PATH_SQL, echo));
 
        return conn;
@@ -148,27 +162,30 @@ connectDatabase(const char *dbname, const char *pghost,
 
 /*
  * Try to connect to the appropriate maintenance database.
+ *
+ * This differs from connectDatabase only in that it has a rule for
+ * inserting a default "dbname" if none was given (which is why cparams
+ * is not const).  Note that cparams->dbname should typically come from
+ * a --maintenance-db command line parameter.
  */
 PGconn *
-connectMaintenanceDatabase(const char *maintenance_db,
-                                                  const char *pghost, const char *pgport,
-                                                  const char *pguser, enum trivalue prompt_password,
+connectMaintenanceDatabase(ConnParams *cparams,
                                                   const char *progname, bool echo)
 {
        PGconn     *conn;
 
        /* If a maintenance database name was specified, just connect to it. */
-       if (maintenance_db)
-               return connectDatabase(maintenance_db, pghost, pgport, pguser,
-                                                          prompt_password, progname, echo, false, false);
+       if (cparams->dbname)
+               return connectDatabase(cparams, progname, echo, false, false);
 
        /* Otherwise, try postgres first and then template1. */
-       conn = connectDatabase("postgres", pghost, pgport, pguser, prompt_password,
-                                                  progname, echo, true, false);
+       cparams->dbname = "postgres";
+       conn = connectDatabase(cparams, progname, echo, true, false);
        if (!conn)
-               conn = connectDatabase("template1", pghost, pgport, pguser,
-                                                          prompt_password, progname, echo, false, false);
-
+       {
+               cparams->dbname = "template1";
+               conn = connectDatabase(cparams, progname, echo, false, false);
+       }
        return conn;
 }
 
index ddf6320b47c8ffdf173258e785b4c57464b078e0..9ec57cdd87c0e466a45b2e3b4ac2d8123ca06099 100644 (file)
@@ -21,20 +21,32 @@ enum trivalue
        TRI_YES
 };
 
+/* Parameters needed by connectDatabase/connectMaintenanceDatabase */
+typedef struct _connParams
+{
+       /* These fields record the actual command line parameters */
+       const char *dbname;                     /* this may be a connstring! */
+       const char *pghost;
+       const char *pgport;
+       const char *pguser;
+       enum trivalue prompt_password;
+       /* If not NULL, this overrides the dbname obtained from command line */
+       /* (but *only* the DB name, not anything else in the connstring) */
+       const char *override_dbname;
+} ConnParams;
+
 typedef void (*help_handler) (const char *progname);
 
 extern void handle_help_version_opts(int argc, char *argv[],
                                                                         const char *fixed_progname,
                                                                         help_handler hlp);
 
-extern PGconn *connectDatabase(const char *dbname, const char *pghost,
-                                                          const char *pgport, const char *pguser,
-                                                          enum trivalue prompt_password, const char *progname,
-                                                          bool echo, bool fail_ok, bool allow_password_reuse);
+extern PGconn *connectDatabase(const ConnParams *cparams,
+                                                          const char *progname,
+                                                          bool echo, bool fail_ok,
+                                                          bool allow_password_reuse);
 
-extern PGconn *connectMaintenanceDatabase(const char *maintenance_db,
-                                                                                 const char *pghost, const char *pgport,
-                                                                                 const char *pguser, enum trivalue prompt_password,
+extern PGconn *connectMaintenanceDatabase(ConnParams *cparams,
                                                                                  const char *progname, bool echo);
 
 extern void disconnectDatabase(PGconn *conn);
index 1353af97c49ebe66ad6ca937fff776fd8875a18e..91e6e2194bd7700fa963c25153ab17d914cc723a 100644 (file)
@@ -51,6 +51,7 @@ main(int argc, char *argv[])
        char       *port = NULL;
        char       *username = NULL;
        enum trivalue prompt_password = TRI_DEFAULT;
+       ConnParams      cparams;
        bool            echo = false;
        char       *owner = NULL;
        char       *tablespace = NULL;
@@ -180,8 +181,14 @@ main(int argc, char *argv[])
        if (maintenance_db == NULL && strcmp(dbname, "postgres") == 0)
                maintenance_db = "template1";
 
-       conn = connectMaintenanceDatabase(maintenance_db, host, port, username,
-                                                                         prompt_password, progname, echo);
+       cparams.dbname = maintenance_db;
+       cparams.pghost = host;
+       cparams.pgport = port;
+       cparams.pguser = username;
+       cparams.prompt_password = prompt_password;
+       cparams.override_dbname = NULL;
+
+       conn = connectMaintenanceDatabase(&cparams, progname, echo);
 
        initPQExpBuffer(&sql);
 
index 9ced079ac7598f564f1f172eff9ef0b7cc39e15d..a15d21f26434243b4b6a5fb044b53b173277558c 100644 (file)
@@ -58,6 +58,7 @@ main(int argc, char *argv[])
        char       *username = NULL;
        SimpleStringList roles = {NULL, NULL};
        enum trivalue prompt_password = TRI_DEFAULT;
+       ConnParams      cparams;
        bool            echo = false;
        bool            interactive = false;
        int                     conn_limit = -2;        /* less than minimum valid value */
@@ -256,8 +257,14 @@ main(int argc, char *argv[])
        if (login == 0)
                login = TRI_YES;
 
-       conn = connectDatabase("postgres", host, port, username, prompt_password,
-                                                  progname, echo, false, false);
+       cparams.dbname = NULL;          /* this program lacks any dbname option... */
+       cparams.pghost = host;
+       cparams.pgport = port;
+       cparams.pguser = username;
+       cparams.prompt_password = prompt_password;
+       cparams.override_dbname = NULL;
+
+       conn = connectMaintenanceDatabase(&cparams, progname, echo);
 
        initPQExpBuffer(&sql);
 
index 581c7749c86a14642840806d0b0f48c1a123c859..ccbf78e91a86437e8ada464648d7ff4dc6a81efa 100644 (file)
@@ -48,6 +48,7 @@ main(int argc, char *argv[])
        char       *port = NULL;
        char       *username = NULL;
        enum trivalue prompt_password = TRI_DEFAULT;
+       ConnParams      cparams;
        bool            echo = false;
        bool            interactive = false;
        bool            force = false;
@@ -137,9 +138,14 @@ main(int argc, char *argv[])
        if (maintenance_db == NULL && strcmp(dbname, "postgres") == 0)
                maintenance_db = "template1";
 
-       conn = connectMaintenanceDatabase(maintenance_db,
-                                                                         host, port, username, prompt_password,
-                                                                         progname, echo);
+       cparams.dbname = maintenance_db;
+       cparams.pghost = host;
+       cparams.pgport = port;
+       cparams.pguser = username;
+       cparams.prompt_password = prompt_password;
+       cparams.override_dbname = NULL;
+
+       conn = connectMaintenanceDatabase(&cparams, progname, echo);
 
        if (echo)
                printf("%s\n", sql.data);
index fee270d4f6d11ff0977fc2f9d25b9009131c8d4d..ef4fe4cd18da590f7099e730e85a92a0d5b29cb6 100644 (file)
@@ -45,6 +45,7 @@ main(int argc, char *argv[])
        char       *port = NULL;
        char       *username = NULL;
        enum trivalue prompt_password = TRI_DEFAULT;
+       ConnParams      cparams;
        bool            echo = false;
        bool            interactive = false;
        char            dropuser_buf[128];
@@ -131,13 +132,19 @@ main(int argc, char *argv[])
                        exit(0);
        }
 
+       cparams.dbname = NULL;          /* this program lacks any dbname option... */
+       cparams.pghost = host;
+       cparams.pgport = port;
+       cparams.pguser = username;
+       cparams.prompt_password = prompt_password;
+       cparams.override_dbname = NULL;
+
+       conn = connectMaintenanceDatabase(&cparams, progname, echo);
+
        initPQExpBuffer(&sql);
        appendPQExpBuffer(&sql, "DROP ROLE %s%s;",
                                          (if_exists ? "IF EXISTS " : ""), fmtId(dropuser));
 
-       conn = connectDatabase("postgres", host, port, username, prompt_password,
-                                                  progname, echo, false, false);
-
        if (echo)
                printf("%s\n", sql.data);
        result = PQexec(conn, sql.data);
index 40dcbc928332416f0fb7e11431be6612ecb55271..40003380aef5fb2b4e2b1da9d2c0e21ae2473a4f 100644 (file)
@@ -34,15 +34,12 @@ static SimpleStringList *get_parallel_object_list(PGconn *conn,
                                                                                                  ReindexType type,
                                                                                                  SimpleStringList *user_list,
                                                                                                  bool echo);
-static void reindex_one_database(const char *dbname, ReindexType type,
-                                                                SimpleStringList *user_list, const char *host,
-                                                                const char *port, const char *username,
-                                                                enum trivalue prompt_password, const char *progname,
+static void reindex_one_database(const ConnParams *cparams, ReindexType type,
+                                                                SimpleStringList *user_list,
+                                                                const char *progname,
                                                                 bool echo, bool verbose, bool concurrently,
                                                                 int concurrentCons);
-static void reindex_all_databases(const char *maintenance_db,
-                                                                 const char *host, const char *port,
-                                                                 const char *username, enum trivalue prompt_password,
+static void reindex_all_databases(ConnParams *cparams,
                                                                  const char *progname, bool echo,
                                                                  bool quiet, bool verbose, bool concurrently,
                                                                  int concurrentCons);
@@ -86,6 +83,7 @@ main(int argc, char *argv[])
        const char *port = NULL;
        const char *username = NULL;
        enum trivalue prompt_password = TRI_DEFAULT;
+       ConnParams      cparams;
        bool            syscatalog = false;
        bool            alldb = false;
        bool            echo = false;
@@ -188,6 +186,13 @@ main(int argc, char *argv[])
                exit(1);
        }
 
+       /* fill cparams except for dbname, which is set below */
+       cparams.pghost = host;
+       cparams.pgport = port;
+       cparams.pguser = username;
+       cparams.prompt_password = prompt_password;
+       cparams.override_dbname = NULL;
+
        setup_cancel_handler(NULL);
 
        if (alldb)
@@ -218,8 +223,9 @@ main(int argc, char *argv[])
                        exit(1);
                }
 
-               reindex_all_databases(maintenance_db, host, port, username,
-                                                         prompt_password, progname, echo, quiet, verbose,
+               cparams.dbname = maintenance_db;
+
+               reindex_all_databases(&cparams, progname, echo, quiet, verbose,
                                                          concurrently, concurrentCons);
        }
        else if (syscatalog)
@@ -256,9 +262,11 @@ main(int argc, char *argv[])
                                dbname = get_user_name_or_exit(progname);
                }
 
-               reindex_one_database(dbname, REINDEX_SYSTEM, NULL, host,
-                                                        port, username, prompt_password, progname,
-                                                        echo, verbose, concurrently, 1);
+               cparams.dbname = dbname;
+
+               reindex_one_database(&cparams, REINDEX_SYSTEM, NULL,
+                                                        progname, echo, verbose,
+                                                        concurrently, 1);
        }
        else
        {
@@ -283,40 +291,40 @@ main(int argc, char *argv[])
                                dbname = get_user_name_or_exit(progname);
                }
 
+               cparams.dbname = dbname;
+
                if (schemas.head != NULL)
-                       reindex_one_database(dbname, REINDEX_SCHEMA, &schemas, host,
-                                                                port, username, prompt_password, progname,
-                                                                echo, verbose, concurrently, concurrentCons);
+                       reindex_one_database(&cparams, REINDEX_SCHEMA, &schemas,
+                                                                progname, echo, verbose,
+                                                                concurrently, concurrentCons);
 
                if (indexes.head != NULL)
-                       reindex_one_database(dbname, REINDEX_INDEX, &indexes, host,
-                                                                port, username, prompt_password, progname,
-                                                                echo, verbose, concurrently, 1);
+                       reindex_one_database(&cparams, REINDEX_INDEX, &indexes,
+                                                                progname, echo, verbose,
+                                                                concurrently, 1);
 
                if (tables.head != NULL)
-                       reindex_one_database(dbname, REINDEX_TABLE, &tables, host,
-                                                                port, username, prompt_password, progname,
-                                                                echo, verbose, concurrently,
-                                                                concurrentCons);
+                       reindex_one_database(&cparams, REINDEX_TABLE, &tables,
+                                                                progname, echo, verbose,
+                                                                concurrently, concurrentCons);
 
                /*
                 * reindex database only if neither index nor table nor schema is
                 * specified
                 */
                if (indexes.head == NULL && tables.head == NULL && schemas.head == NULL)
-                       reindex_one_database(dbname, REINDEX_DATABASE, NULL, host,
-                                                                port, username, prompt_password, progname,
-                                                                echo, verbose, concurrently, concurrentCons);
+                       reindex_one_database(&cparams, REINDEX_DATABASE, NULL,
+                                                                progname, echo, verbose,
+                                                                concurrently, concurrentCons);
        }
 
        exit(0);
 }
 
 static void
-reindex_one_database(const char *dbname, ReindexType type,
-                                        SimpleStringList *user_list, const char *host,
-                                        const char *port, const char *username,
-                                        enum trivalue prompt_password, const char *progname, bool echo,
+reindex_one_database(const ConnParams *cparams, ReindexType type,
+                                        SimpleStringList *user_list,
+                                        const char *progname, bool echo,
                                         bool verbose, bool concurrently, int concurrentCons)
 {
        PGconn     *conn;
@@ -328,8 +336,7 @@ reindex_one_database(const char *dbname, ReindexType type,
        bool            failed = false;
        int                     items_count = 0;
 
-       conn = connectDatabase(dbname, host, port, username, prompt_password,
-                                                  progname, echo, false, false);
+       conn = connectDatabase(cparams, progname, echo, false, false);
 
        if (concurrently && PQserverVersion(conn) < 120000)
        {
@@ -436,8 +443,7 @@ reindex_one_database(const char *dbname, ReindexType type,
 
        Assert(process_list != NULL);
 
-       slots = ParallelSlotsSetup(dbname, host, port, username, prompt_password,
-                                                          progname, echo, conn, concurrentCons);
+       slots = ParallelSlotsSetup(cparams, progname, echo, conn, concurrentCons);
 
        cell = process_list->head;
        do
@@ -705,23 +711,18 @@ get_parallel_object_list(PGconn *conn, ReindexType type,
 }
 
 static void
-reindex_all_databases(const char *maintenance_db,
-                                         const char *host, const char *port,
-                                         const char *username, enum trivalue prompt_password,
+reindex_all_databases(ConnParams *cparams,
                                          const char *progname, bool echo, bool quiet, bool verbose,
                                          bool concurrently, int concurrentCons)
 {
        PGconn     *conn;
        PGresult   *result;
-       PQExpBufferData connstr;
        int                     i;
 
-       conn = connectMaintenanceDatabase(maintenance_db, host, port, username,
-                                                                         prompt_password, progname, echo);
+       conn = connectMaintenanceDatabase(cparams, progname, echo);
        result = executeQuery(conn, "SELECT datname FROM pg_database WHERE datallowconn ORDER BY 1;", echo);
        PQfinish(conn);
 
-       initPQExpBuffer(&connstr);
        for (i = 0; i < PQntuples(result); i++)
        {
                char       *dbname = PQgetvalue(result, i, 0);
@@ -732,16 +733,12 @@ reindex_all_databases(const char *maintenance_db,
                        fflush(stdout);
                }
 
-               resetPQExpBuffer(&connstr);
-               appendPQExpBufferStr(&connstr, "dbname=");
-               appendConnStrVal(&connstr, dbname);
+               cparams->override_dbname = dbname;
 
-               reindex_one_database(connstr.data, REINDEX_DATABASE, NULL, host,
-                                                        port, username, prompt_password,
+               reindex_one_database(cparams, REINDEX_DATABASE, NULL,
                                                         progname, echo, verbose, concurrently,
                                                         concurrentCons);
        }
-       termPQExpBuffer(&connstr);
 
        PQclear(result);
 }
index 01bc6dfeffc90165349e3979b6da59ed80301f0e..ec264a269a7d7a506c347112af6be183b2aec9ce 100644 (file)
@@ -205,8 +205,7 @@ ParallelSlotsGetIdle(ParallelSlot *slots, int numslots)
  * set.
  */
 ParallelSlot *
-ParallelSlotsSetup(const char *dbname, const char *host, const char *port,
-                                  const char *username, bool prompt_password,
+ParallelSlotsSetup(const ConnParams *cparams,
                                   const char *progname, bool echo,
                                   PGconn *conn, int numslots)
 {
@@ -221,8 +220,7 @@ ParallelSlotsSetup(const char *dbname, const char *host, const char *port,
        {
                for (i = 1; i < numslots; i++)
                {
-                       conn = connectDatabase(dbname, host, port, username, prompt_password,
-                                                                  progname, echo, false, true);
+                       conn = connectDatabase(cparams, progname, echo, false, true);
 
                        /*
                         * Fail and exit immediately if trying to use a socket in an
index cf20449ce3e29053447b63c2747987113a636567..c9d9f0623e949bd93d7f9ef6da6ff45577a30a56 100644 (file)
@@ -12,6 +12,7 @@
 #ifndef SCRIPTS_PARALLEL_H
 #define SCRIPTS_PARALLEL_H
 
+#include "common.h"
 #include "libpq-fe.h"
 
 
@@ -23,10 +24,7 @@ typedef struct ParallelSlot
 
 extern ParallelSlot *ParallelSlotsGetIdle(ParallelSlot *slots, int numslots);
 
-extern ParallelSlot *ParallelSlotsSetup(const char *dbname, const char *host,
-                                                                               const char *port,
-                                                                               const char *username,
-                                                                               bool prompt_password,
+extern ParallelSlot *ParallelSlotsSetup(const ConnParams *cparams,
                                                                                const char *progname, bool echo,
                                                                                PGconn *conn, int numslots);
 
index dc67df45e6daa658c97a28be6b3d7f346710ecba..3e57fa9a8614f50ed63a35ff0fa57412b497e1f0 100644 (file)
@@ -40,19 +40,16 @@ typedef struct vacuumingOptions
 } vacuumingOptions;
 
 
-static void vacuum_one_database(const char *dbname, vacuumingOptions *vacopts,
+static void vacuum_one_database(const ConnParams *cparams,
+                                                               vacuumingOptions *vacopts,
                                                                int stage,
                                                                SimpleStringList *tables,
-                                                               const char *host, const char *port,
-                                                               const char *username, enum trivalue prompt_password,
                                                                int concurrentCons,
                                                                const char *progname, bool echo, bool quiet);
 
-static void vacuum_all_databases(vacuumingOptions *vacopts,
+static void vacuum_all_databases(ConnParams *cparams,
+                                                                vacuumingOptions *vacopts,
                                                                 bool analyze_in_stages,
-                                                                const char *maintenance_db,
-                                                                const char *host, const char *port,
-                                                                const char *username, enum trivalue prompt_password,
                                                                 int concurrentCons,
                                                                 const char *progname, bool echo, bool quiet);
 
@@ -108,6 +105,7 @@ main(int argc, char *argv[])
        char       *port = NULL;
        char       *username = NULL;
        enum trivalue prompt_password = TRI_DEFAULT;
+       ConnParams      cparams;
        bool            echo = false;
        bool            quiet = false;
        vacuumingOptions vacopts;
@@ -287,6 +285,13 @@ main(int argc, char *argv[])
                }
        }
 
+       /* fill cparams except for dbname, which is set below */
+       cparams.pghost = host;
+       cparams.pgport = port;
+       cparams.pguser = username;
+       cparams.prompt_password = prompt_password;
+       cparams.override_dbname = NULL;
+
        setup_cancel_handler(NULL);
 
        /* Avoid opening extra connections. */
@@ -306,10 +311,10 @@ main(int argc, char *argv[])
                        exit(1);
                }
 
-               vacuum_all_databases(&vacopts,
+               cparams.dbname = maintenance_db;
+
+               vacuum_all_databases(&cparams, &vacopts,
                                                         analyze_in_stages,
-                                                        maintenance_db,
-                                                        host, port, username, prompt_password,
                                                         concurrentCons,
                                                         progname, echo, quiet);
        }
@@ -325,25 +330,25 @@ main(int argc, char *argv[])
                                dbname = get_user_name_or_exit(progname);
                }
 
+               cparams.dbname = dbname;
+
                if (analyze_in_stages)
                {
                        int                     stage;
 
                        for (stage = 0; stage < ANALYZE_NUM_STAGES; stage++)
                        {
-                               vacuum_one_database(dbname, &vacopts,
+                               vacuum_one_database(&cparams, &vacopts,
                                                                        stage,
                                                                        &tables,
-                                                                       host, port, username, prompt_password,
                                                                        concurrentCons,
                                                                        progname, echo, quiet);
                        }
                }
                else
-                       vacuum_one_database(dbname, &vacopts,
+                       vacuum_one_database(&cparams, &vacopts,
                                                                ANALYZE_NO_STAGE,
                                                                &tables,
-                                                               host, port, username, prompt_password,
                                                                concurrentCons,
                                                                progname, echo, quiet);
        }
@@ -365,11 +370,10 @@ main(int argc, char *argv[])
  * a list of tables from the database.
  */
 static void
-vacuum_one_database(const char *dbname, vacuumingOptions *vacopts,
+vacuum_one_database(const ConnParams *cparams,
+                                       vacuumingOptions *vacopts,
                                        int stage,
                                        SimpleStringList *tables,
-                                       const char *host, const char *port,
-                                       const char *username, enum trivalue prompt_password,
                                        int concurrentCons,
                                        const char *progname, bool echo, bool quiet)
 {
@@ -401,8 +405,7 @@ vacuum_one_database(const char *dbname, vacuumingOptions *vacopts,
        Assert(stage == ANALYZE_NO_STAGE ||
                   (stage >= 0 && stage < ANALYZE_NUM_STAGES));
 
-       conn = connectDatabase(dbname, host, port, username, prompt_password,
-                                                  progname, echo, false, true);
+       conn = connectDatabase(cparams, progname, echo, false, true);
 
        if (vacopts->disable_page_skipping && PQserverVersion(conn) < 90600)
        {
@@ -630,8 +633,7 @@ vacuum_one_database(const char *dbname, vacuumingOptions *vacopts,
        if (concurrentCons <= 0)
                concurrentCons = 1;
 
-       slots = ParallelSlotsSetup(dbname, host, port, username, prompt_password,
-                                                          progname, echo, conn, concurrentCons);
+       slots = ParallelSlotsSetup(cparams, progname, echo, conn, concurrentCons);
 
        /*
         * Prepare all the connections to run the appropriate analyze stage, if
@@ -703,28 +705,23 @@ finish:
  * quickly everywhere before generating more detailed ones.
  */
 static void
-vacuum_all_databases(vacuumingOptions *vacopts,
+vacuum_all_databases(ConnParams *cparams,
+                                        vacuumingOptions *vacopts,
                                         bool analyze_in_stages,
-                                        const char *maintenance_db, const char *host,
-                                        const char *port, const char *username,
-                                        enum trivalue prompt_password,
                                         int concurrentCons,
                                         const char *progname, bool echo, bool quiet)
 {
        PGconn     *conn;
        PGresult   *result;
-       PQExpBufferData connstr;
        int                     stage;
        int                     i;
 
-       conn = connectMaintenanceDatabase(maintenance_db, host, port, username,
-                                                                         prompt_password, progname, echo);
+       conn = connectMaintenanceDatabase(cparams, progname, echo);
        result = executeQuery(conn,
                                                  "SELECT datname FROM pg_database WHERE datallowconn ORDER BY 1;",
                                                  echo);
        PQfinish(conn);
 
-       initPQExpBuffer(&connstr);
        if (analyze_in_stages)
        {
                /*
@@ -739,14 +736,11 @@ vacuum_all_databases(vacuumingOptions *vacopts,
                {
                        for (i = 0; i < PQntuples(result); i++)
                        {
-                               resetPQExpBuffer(&connstr);
-                               appendPQExpBufferStr(&connstr, "dbname=");
-                               appendConnStrVal(&connstr, PQgetvalue(result, i, 0));
+                               cparams->override_dbname = PQgetvalue(result, i, 0);
 
-                               vacuum_one_database(connstr.data, vacopts,
+                               vacuum_one_database(cparams, vacopts,
                                                                        stage,
                                                                        NULL,
-                                                                       host, port, username, prompt_password,
                                                                        concurrentCons,
                                                                        progname, echo, quiet);
                        }
@@ -756,19 +750,15 @@ vacuum_all_databases(vacuumingOptions *vacopts,
        {
                for (i = 0; i < PQntuples(result); i++)
                {
-                       resetPQExpBuffer(&connstr);
-                       appendPQExpBufferStr(&connstr, "dbname=");
-                       appendConnStrVal(&connstr, PQgetvalue(result, i, 0));
+                       cparams->override_dbname = PQgetvalue(result, i, 0);
 
-                       vacuum_one_database(connstr.data, vacopts,
+                       vacuum_one_database(cparams, vacopts,
                                                                ANALYZE_NO_STAGE,
                                                                NULL,
-                                                               host, port, username, prompt_password,
                                                                concurrentCons,
                                                                progname, echo, quiet);
                }
        }
-       termPQExpBuffer(&connstr);
 
        PQclear(result);
 }