Fix connection string handling in src/bin/scripts/ programs.
authorTom Lane <[email protected]>
Mon, 19 Oct 2020 23:03:47 +0000 (19:03 -0400)
committerTom Lane <[email protected]>
Mon, 19 Oct 2020 23:03:47 +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/createlang.c
src/bin/scripts/createuser.c
src/bin/scripts/dropdb.c
src/bin/scripts/droplang.c
src/bin/scripts/dropuser.c
src/bin/scripts/reindexdb.c
src/bin/scripts/vacuumdb.c

index 8e4f9c9cbbff3099efb421b4114153cb4a00ad8f..60edd33a3708bf14d56a42d66abbbd10b78ec204 100644 (file)
@@ -90,9 +90,9 @@ PostgreSQL documentation
       <term><option><optional>--dbname=</><replaceable class="parameter">dbname</replaceable></></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></></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 c363bd4a5654e772dd6ab3b27f72f37a4d7b9295..943f86171ad20ab2124145dde5137e3d595b0dc1 100644 (file)
@@ -286,6 +286,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 16c49e7928bcb739beda3387894686d88f0e2995..0d7170d66552771b0d86564566237763b146227c 100644 (file)
@@ -205,6 +205,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 2f4b4618edcabf4f82d1659f7cc22d6dff84c186..a426a32e9c8d4b381011e64dec59dae10227d5f7 100644 (file)
@@ -123,9 +123,9 @@ PostgreSQL documentation
       <term><option><optional>--dbname=</><replaceable class="parameter">dbname</replaceable></></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
@@ -317,10 +317,16 @@ PostgreSQL documentation
       <term><option>--maintenance-db=<replaceable class="parameter">dbname</replaceable></></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 f8e2e1a9447ed49e00d94e96d66b9eca09217a5f..54fd1956099c193d33410d0f45932ad1d55d0a82 100644 (file)
@@ -92,9 +92,9 @@ PostgreSQL documentation
       <term><option><optional>--dbname=</><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
@@ -342,10 +342,16 @@ PostgreSQL documentation
       <term><option>--maintenance-db=<replaceable class="parameter">dbname</replaceable></></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 a6fc0449978969552e61689ccb5dc8c4725f0521..3e32160f27302c40f04e1e1d62b867bb6dca0909 100644 (file)
 #include "dumputils.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);
 
 
@@ -55,6 +50,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;
@@ -130,6 +126,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();
 
    if (alldb)
@@ -148,8 +151,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
    {
@@ -163,21 +167,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);
@@ -185,17 +189,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);
 
@@ -226,22 +227,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;", progname, echo);
    PQfinish(conn);
 
-   initPQExpBuffer(&connstr);
    for (i = 0; i < PQntuples(result); i++)
    {
        char       *dbname = PQgetvalue(result, i, 0);
@@ -252,15 +248,10 @@ cluster_all_databases(bool verbose, const char *maintenance_db,
            fflush(stdout);
        }
 
-       resetPQExpBuffer(&connstr);
-       appendPQExpBuffer(&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 c5e004bff2945a16df553e1c6b2d972f476ee3bd..21083cab33f9d539a5f13d5bb1fa442e4c04b48a 100644 (file)
@@ -57,7 +57,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
@@ -65,15 +65,16 @@ 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;
    static char *password = NULL;
    bool        new_pass;
 
+   /* Callers must supply at least dbname; other params can be NULL */
+   Assert(cparams->dbname);
+
    if (!allow_password_reuse)
    {
        if (password)
@@ -81,7 +82,7 @@ connectDatabase(const char *dbname, const char *pghost,
        password = NULL;
    }
 
-   if (password == NULL && prompt_password == TRI_YES)
+   if (cparams->prompt_password == TRI_YES && password == NULL)
        password = simple_prompt("Password: ", 100, false);
 
    /*
@@ -90,23 +91,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] = password;
-       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++] = password;
+       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);
@@ -114,7 +127,7 @@ connectDatabase(const char *dbname, const char *pghost,
        if (!conn)
        {
            fprintf(stderr, _("%s: could not connect to database %s: out of memory\n"),
-                   progname, dbname);
+                   progname, cparams->dbname);
            exit(1);
        }
 
@@ -123,7 +136,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);
            if (password)
@@ -142,10 +155,11 @@ connectDatabase(const char *dbname, const char *pghost,
            return NULL;
        }
        fprintf(stderr, _("%s: could not connect to database %s: %s"),
-               progname, dbname, PQerrorMessage(conn));
+               progname, cparams->dbname, PQerrorMessage(conn));
        exit(1);
    }
 
+   /* Start strict; callers may override this. */
    PQclear(executeQuery(conn, ALWAYS_SECURE_SEARCH_PATH_SQL,
                         progname, echo));
 
@@ -154,27 +168,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 c74ca4e2b5af6e2dcd730bd9ef210e0910190b64..161e06cb49db2049abdad951f08214144fb4a729 100644 (file)
@@ -23,20 +23,32 @@ enum trivalue
 
 extern bool CancelRequested;
 
+/* 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 PGresult *executeQuery(PGconn *conn, const char *query,
index 413573097693fd9f4fab5c33f37cf15651521b97..0719ff9124584f5f63067cab2d1b788a7595321c 100644 (file)
@@ -50,6 +50,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;
@@ -181,8 +182,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 34bb9e1f355da4c1d0feef6330af227cf267f5f7..aa2ae53d167ad41664924118fc9a5f145852a11e 100644 (file)
@@ -42,6 +42,7 @@ main(int argc, char *argv[])
    char       *port = NULL;
    char       *username = NULL;
    enum trivalue prompt_password = TRI_DEFAULT;
+   ConnParams  cparams;
    bool        echo = false;
    char       *langname = NULL;
 
@@ -130,6 +131,13 @@ main(int argc, char *argv[])
            dbname = get_user_name_or_exit(progname);
    }
 
+   cparams.dbname = dbname;
+   cparams.pghost = host;
+   cparams.pgport = port;
+   cparams.pguser = username;
+   cparams.prompt_password = prompt_password;
+   cparams.override_dbname = NULL;
+
    initPQExpBuffer(&sql);
 
    /*
@@ -140,7 +148,7 @@ main(int argc, char *argv[])
        printQueryOpt popt;
        static const bool translate_columns[] = {false, true};
 
-       conn = connectDatabase(dbname, host, port, username, prompt_password,
+       conn = connectDatabase(&cparams,
                               progname, echo, false, false);
 
        printfPQExpBuffer(&sql, "SELECT lanname as \"%s\", "
@@ -180,7 +188,7 @@ main(int argc, char *argv[])
        if (*p >= 'A' && *p <= 'Z')
            *p += ('a' - 'A');
 
-   conn = connectDatabase(dbname, host, port, username, prompt_password,
+   conn = connectDatabase(&cparams,
                           progname, echo, false, false);
 
    /*
index b3c622b387c487b319e91023e0b66e00de8be089..2962da54e1d4430378963f9e705722158aa59aa1 100644 (file)
@@ -60,6 +60,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 */
@@ -259,8 +260,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 8809aacf998516e0fe4728a5e8b6f0152f0fda0d..5292e1b28ee40bc4b4b3cf682b2c9fff32665427 100644 (file)
@@ -46,6 +46,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;
 
@@ -128,9 +129,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 a029813e53bab774026f9a6ca58a1341c3efd372..cdad1c4ac62f1c4fd4740f8753b2fe489ac61ff7 100644 (file)
@@ -44,6 +44,7 @@ main(int argc, char *argv[])
    char       *port = NULL;
    char       *username = NULL;
    enum trivalue prompt_password = TRI_DEFAULT;
+   ConnParams  cparams;
    bool        echo = false;
    char       *langname = NULL;
    char       *p;
@@ -129,6 +130,13 @@ main(int argc, char *argv[])
            dbname = get_user_name_or_exit(progname);
    }
 
+   cparams.dbname = dbname;
+   cparams.pghost = host;
+   cparams.pgport = port;
+   cparams.pguser = username;
+   cparams.prompt_password = prompt_password;
+   cparams.override_dbname = NULL;
+
    initPQExpBuffer(&sql);
 
    /*
@@ -139,7 +147,7 @@ main(int argc, char *argv[])
        printQueryOpt popt;
        static const bool translate_columns[] = {false, true};
 
-       conn = connectDatabase(dbname, host, port, username, prompt_password,
+       conn = connectDatabase(&cparams,
                               progname, echo, false, false);
 
        printfPQExpBuffer(&sql, "SELECT lanname as \"%s\", "
@@ -181,7 +189,7 @@ main(int argc, char *argv[])
        if (*p >= 'A' && *p <= 'Z')
            *p += ('a' - 'A');
 
-   conn = connectDatabase(dbname, host, port, username, prompt_password,
+   conn = connectDatabase(&cparams,
                           progname, echo, false, false);
 
    /*
index a90fcb2aabb7ced72c2b5a583676c25bfb1a676d..5fdff0d68e1a0e1211c049574e6cd62993425b9f 100644 (file)
@@ -44,6 +44,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;
 
@@ -124,13 +125,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 f71b9b3e95a7423e180f345b129f0b19f7772aa9..0f0a7fe50409761eaf74daf8b5999b85a8912e64 100644 (file)
 #include "dumputils.h"
 
 
-static void reindex_one_database(const char *name, const char *dbname,
-                    const char *type, const char *host,
-                    const char *port, const char *username,
-                    enum trivalue prompt_password, const char *progname,
+static void reindex_one_database(const ConnParams *cparams,
+                                const char *type, const char *name,
+                                const char *progname,
                     bool echo, bool verbose);
-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);
-static void reindex_system_catalogs(const char *dbname,
-                       const char *host, const char *port,
-                       const char *username, enum trivalue prompt_password,
+static void reindex_system_catalogs(const ConnParams *cparams,
                        const char *progname, bool echo, bool verbose);
 static void help(const char *progname);
 
@@ -62,6 +57,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;
@@ -150,6 +146,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();
 
    if (alldb)
@@ -180,8 +183,10 @@ 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);
    }
    else if (syscatalog)
    {
@@ -211,7 +216,9 @@ main(int argc, char *argv[])
                dbname = get_user_name_or_exit(progname);
        }
 
-       reindex_system_catalogs(dbname, host, port, username, prompt_password,
+       cparams.dbname = dbname;
+
+       reindex_system_catalogs(&cparams,
                                progname, echo, verbose);
    }
    else
@@ -226,14 +233,16 @@ main(int argc, char *argv[])
                dbname = get_user_name_or_exit(progname);
        }
 
+       cparams.dbname = dbname;
+
        if (schemas.head != NULL)
        {
            SimpleStringListCell *cell;
 
            for (cell = schemas.head; cell; cell = cell->next)
            {
-               reindex_one_database(cell->val, dbname, "SCHEMA", host, port,
-                        username, prompt_password, progname, echo, verbose);
+               reindex_one_database(&cparams, "SCHEMA", cell->val,
+                                    progname, echo, verbose);
            }
        }
 
@@ -243,8 +252,8 @@ main(int argc, char *argv[])
 
            for (cell = indexes.head; cell; cell = cell->next)
            {
-               reindex_one_database(cell->val, dbname, "INDEX", host, port,
-                        username, prompt_password, progname, echo, verbose);
+               reindex_one_database(&cparams, "INDEX", cell->val,
+                                    progname, echo, verbose);
            }
        }
        if (tables.head != NULL)
@@ -253,8 +262,8 @@ main(int argc, char *argv[])
 
            for (cell = tables.head; cell; cell = cell->next)
            {
-               reindex_one_database(cell->val, dbname, "TABLE", host, port,
-                        username, prompt_password, progname, echo, verbose);
+               reindex_one_database(&cparams, "TABLE", cell->val,
+                                    progname, echo, verbose);
            }
        }
 
@@ -263,25 +272,24 @@ main(int argc, char *argv[])
         * specified
         */
        if (indexes.head == NULL && tables.head == NULL && schemas.head == NULL)
-           reindex_one_database(NULL, dbname, "DATABASE", host, port,
-                        username, prompt_password, progname, echo, verbose);
+           reindex_one_database(&cparams, "DATABASE", NULL,
+                                progname, echo, verbose);
    }
 
    exit(0);
 }
 
 static void
-reindex_one_database(const char *name, const char *dbname, const char *type,
-                    const char *host, const char *port, const char *username,
-             enum trivalue prompt_password, const char *progname, bool echo,
-                    bool verbose)
+reindex_one_database(const ConnParams *cparams,
+                    const char *type, const char *name,
+                    const char *progname,
+                    bool echo, bool verbose)
 {
    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);
 
@@ -324,22 +332,17 @@ reindex_one_database(const char *name, const char *dbname, const char *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)
 {
    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;", progname, echo);
    PQfinish(conn);
 
-   initPQExpBuffer(&connstr);
    for (i = 0; i < PQntuples(result); i++)
    {
        char       *dbname = PQgetvalue(result, i, 0);
@@ -350,29 +353,23 @@ reindex_all_databases(const char *maintenance_db,
            fflush(stdout);
        }
 
-       resetPQExpBuffer(&connstr);
-       appendPQExpBuffer(&connstr, "dbname=");
-       appendConnStrVal(&connstr, dbname);
+       cparams->override_dbname = dbname;
 
-       reindex_one_database(NULL, connstr.data, "DATABASE", host,
-                            port, username, prompt_password,
+       reindex_one_database(cparams, "DATABASE", NULL,
                             progname, echo, verbose);
    }
-   termPQExpBuffer(&connstr);
 
    PQclear(result);
 }
 
 static void
-reindex_system_catalogs(const char *dbname, const char *host, const char *port,
-                       const char *username, enum trivalue prompt_password,
+reindex_system_catalogs(const ConnParams *cparams,
                        const char *progname, bool echo, bool verbose)
 {
    PGconn     *conn;
    PQExpBufferData sql;
 
-   conn = connectDatabase(dbname, host, port, username, prompt_password,
-                          progname, echo, false, false);
+   conn = connectDatabase(cparams, progname, echo, false, false);
 
    initPQExpBuffer(&sql);
 
index 068ae7a189e1e8851652e1be2262df12d2ff156d..16db22d2b555954f3dfa167e045b0561eefa90ca 100644 (file)
@@ -44,19 +44,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);
 
@@ -123,6 +120,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;
@@ -249,6 +247,13 @@ main(int argc, char *argv[])
        /* allow 'and_analyze' with 'analyze_only' */
    }
 
+   /* 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();
 
    /* Avoid opening extra connections. */
@@ -270,10 +275,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);
    }
@@ -289,25 +294,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);
    }
@@ -329,11 +334,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)
 {
@@ -359,8 +363,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 (!quiet)
    {
@@ -432,8 +435,7 @@ vacuum_one_database(const char *dbname, vacuumingOptions *vacopts,
    {
        for (i = 1; i < concurrentCons; 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
@@ -556,28 +558,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;",
                          progname, echo);
    PQfinish(conn);
 
-   initPQExpBuffer(&connstr);
    if (analyze_in_stages)
    {
        /*
@@ -592,14 +589,11 @@ vacuum_all_databases(vacuumingOptions *vacopts,
        {
            for (i = 0; i < PQntuples(result); i++)
            {
-               resetPQExpBuffer(&connstr);
-               appendPQExpBuffer(&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);
            }
@@ -609,19 +603,15 @@ vacuum_all_databases(vacuumingOptions *vacopts,
    {
        for (i = 0; i < PQntuples(result); i++)
        {
-           resetPQExpBuffer(&connstr);
-           appendPQExpBuffer(&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);
 }