Prevent satisfies_hash_partition from crashing with VARIADIC NULL.

Commit f3b0897a1213f46b4d3a99a7f8ef3a4b32e03572 fixed some
related problems, but overlooked this one. That commit first
appeared in PostgreSQL 11, so back-patch to all supported branches.

Backpatch-through: 14
Discussion: http://postgr.es/m/CA+TgmobsvQw3F+KRYT83=N3teh8D2t-oPR=U06QDZJE3viCJRg@mail.gmail.com
Reviewed-by: Tender Wang <tndrwang@gmail.com>
Reviewed-by: Ewan Young <kdbase.hack@gmail.com>

diff --git a/src/backend/partitioning/partbounds.c b/src/backend/partitioning/partbounds.c
index c28639d2e3f4f91ac1f9d575d78ef08402549d24..63e89fd4ef8dc6aad7feb3c1d225f167985464af 100644 (file)
--- a/src/backend/partitioning/partbounds.c
+++ b/src/backend/partitioning/partbounds.c
@@ -4868,6 +4868,12 @@ satisfies_hash_partition(PG_FUNCTION_ARGS)
                                                           fcinfo->flinfo->fn_mcxt);
                        }
                }
+               else if (PG_ARGISNULL(3))
+               {
+                       /* Special case for VARIADIC NULL::sometype[] */
+                       relation_close(parent, NoLock);
+                       PG_RETURN_BOOL(false);
+               }
                else
                {
                        ArrayType  *variadic_array = PG_GETARG_ARRAYTYPE_P(3);
@@ -4938,12 +4944,18 @@ satisfies_hash_partition(PG_FUNCTION_ARGS)
        }
        else
        {
-               ArrayType  *variadic_array = PG_GETARG_ARRAYTYPE_P(3);
+               ArrayType  *variadic_array;
                int                     i;
                int                     nelems;
                Datum      *datum;
                bool       *isnull;
 
+               /* Special case for VARIADIC NULL::sometype[] */
+               if (PG_ARGISNULL(3))
+                       PG_RETURN_BOOL(false);
+
+               variadic_array = PG_GETARG_ARRAYTYPE_P(3);
+
                deconstruct_array(variadic_array,
                                                  my_extra->variadic_type,
                                                  my_extra->variadic_typlen,

diff --git a/src/test/regress/expected/hash_part.out b/src/test/regress/expected/hash_part.out
index cb39161f867220628458385390116b71418c21f2..44b6e461ffe177f0e3948e9d659d2b0dca633e79 100644 (file)
--- a/src/test/regress/expected/hash_part.out
+++ b/src/test/regress/expected/hash_part.out
@@ -40,6 +40,13 @@ SELECT satisfies_hash_partition('mchash'::regclass, 4, NULL, NULL);
  f
 (1 row)
 
+-- variadic null
+SELECT satisfies_hash_partition('mchash'::regclass, 4, 0, VARIADIC NULL::int[]);
+ satisfies_hash_partition 
+--------------------------
+ f
+(1 row)
+
 -- too many arguments
 SELECT satisfies_hash_partition('mchash'::regclass, 4, 0, NULL::int, NULL::text, NULL::json);
 ERROR:  number of partitioning columns (2) does not match number of partition keys provided (3)

diff --git a/src/test/regress/sql/hash_part.sql b/src/test/regress/sql/hash_part.sql
index 6e2c1f21bfc5af4b3ac849eab7ac20ba80d7c54a..7243299d962e1df9f9d1054d6e9a2b44650ea399 100644 (file)
--- a/src/test/regress/sql/hash_part.sql
+++ b/src/test/regress/sql/hash_part.sql
@@ -35,6 +35,9 @@ SELECT satisfies_hash_partition('mchash'::regclass, NULL, 0, NULL);
 -- remainder is null
 SELECT satisfies_hash_partition('mchash'::regclass, 4, NULL, NULL);
 
+-- variadic null
+SELECT satisfies_hash_partition('mchash'::regclass, 4, 0, VARIADIC NULL::int[]);
+
 -- too many arguments
 SELECT satisfies_hash_partition('mchash'::regclass, 4, 0, NULL::int, NULL::text, NULL::json);