Ignore attempts to \gset into specially treated variables.

If an interactive psql session used \gset when querying a compromised
server, the attacker could execute arbitrary code as the operating
system account running psql.  Using a prefix not found among specially
treated variables, e.g. every lowercase string, precluded the attack.
Fix by issuing a warning and setting no variable for the column in
question.  Users wanting the old behavior can use a prefix and then a
meta-command like "\set HISTSIZE :prefix_HISTSIZE".  Back-patch to 9.5
(all supported versions).

Reviewed by Robert Haas.  Reported by Nick Cleaton.

Security: CVE-2020-25696

diff --git a/src/bin/psql/common.c b/src/bin/psql/common.c
index d04a7943d6a9436fb83b3febeb85f1e42096a119..4e342c180a6c187990c9c5bcf028ab0ed0323d0e 100644 (file)
--- a/src/bin/psql/common.c
+++ b/src/bin/psql/common.c
@@ -894,6 +894,13 @@ StoreQueryTuple(const PGresult *result)
                        /* concatenate prefix and column name */
                        varname = psprintf("%s%s", pset.gset_prefix, colname);
 
+                       if (VariableHasHook(pset.vars, varname))
+                       {
+                               psql_error("attempt to \\gset into specially treated variable \"%s\" ignored\n",
+                                                  varname);
+                               continue;
+                       }
+
                        if (!PQgetisnull(result, 0, i))
                                value = PQgetvalue(result, 0, i);
                        else

diff --git a/src/bin/psql/variables.c b/src/bin/psql/variables.c
index 120b25c6966195c6d01df9cfa88ef27672ad2438..0d28ba9c92bb6e1abe528a8eee9f3cf05932ea7e 100644 (file)
--- a/src/bin/psql/variables.c
+++ b/src/bin/psql/variables.c
@@ -360,6 +360,32 @@ SetVariableHooks(VariableSpace space, const char *name,
                (void) (*ahook) (current->value);
 }
 
+/*
+ * Return true iff the named variable has substitute and/or assign hook
+ * functions.
+ */
+bool
+VariableHasHook(VariableSpace space, const char *name)
+{
+       struct _variable *current;
+
+       Assert(space);
+       Assert(name);
+
+       for (current = space->next; current; current = current->next)
+       {
+               int                     cmp = strcmp(current->name, name);
+
+               if (cmp == 0)
+                       return (current->substitute_hook != NULL ||
+                                       current->assign_hook != NULL);
+               if (cmp > 0)
+                       break;                          /* it's not there */
+       }
+
+       return false;
+}
+
 /*
  * Convenience function to set a variable's value to "on".
  */

diff --git a/src/bin/psql/variables.h b/src/bin/psql/variables.h
index 02d85b1bc2e45b158128b1ccfd5387844357fe56..8dc5c20ee8fc50b03d03e809f906ad6ebafe0de4 100644 (file)
--- a/src/bin/psql/variables.h
+++ b/src/bin/psql/variables.h
@@ -90,6 +90,7 @@ bool          DeleteVariable(VariableSpace space, const char *name);
 void SetVariableHooks(VariableSpace space, const char *name,
                                 VariableSubstituteHook shook,
                                 VariableAssignHook ahook);
+bool           VariableHasHook(VariableSpace space, const char *name);
 
 void           PsqlVarEnumError(const char *name, const char *value, const char *suggestions);
 

diff --git a/src/test/regress/expected/psql.out b/src/test/regress/expected/psql.out
index 0c94144575afe3c57450eddc1d14cf0b87e4eac2..1ae81912c734d5dc53545ddb58fb613e99cc75a6 100644 (file)
--- a/src/test/regress/expected/psql.out
+++ b/src/test/regress/expected/psql.out
@@ -84,6 +84,10 @@ select 10 as test01, 20 as test02, 'Hello' as test03 \gset pref01_
 select 10 as "bad name"
 \gset
 invalid variable name: "bad name"
+select 97 as "EOF", 'ok' as _foo \gset IGNORE
+attempt to \gset into specially treated variable "IGNOREEOF" ignored
+\echo :IGNORE_foo :IGNOREEOF
+ok 0
 -- multiple backslash commands in one line
 select 1 as x, 2 as y \gset pref01_ \\ \echo :pref01_x
 1

diff --git a/src/test/regress/sql/psql.sql b/src/test/regress/sql/psql.sql
index 4a676c3119558d3c2d01f23fde42598cdfe01226..7f8ab2e5c218e9d851d7347ee97fa6ce886f1abd 100644 (file)
--- a/src/test/regress/sql/psql.sql
+++ b/src/test/regress/sql/psql.sql
@@ -48,6 +48,9 @@ select 10 as test01, 20 as test02, 'Hello' as test03 \gset pref01_
 select 10 as "bad name"
 \gset
 
+select 97 as "EOF", 'ok' as _foo \gset IGNORE
+\echo :IGNORE_foo :IGNOREEOF
+
 -- multiple backslash commands in one line
 select 1 as x, 2 as y \gset pref01_ \\ \echo :pref01_x
 select 3 as x, 4 as y \gset pref01_ \echo :pref01_x \echo :pref01_y