Properly detoast data in brin_form_tuple
authorTomas Vondra <[email protected]>
Fri, 6 Nov 2020 23:41:52 +0000 (00:41 +0100)
committerTomas Vondra <[email protected]>
Fri, 6 Nov 2020 23:41:52 +0000 (00:41 +0100)
brin_form_tuple failed to consider the values may be toasted, inserting
the toast pointer into the index. This may easily result in index
corruption, as the toast data may be deleted and cleaned up by vacuum.
The cleanup however does not care about indexes, leaving invalid toast
pointers behind, which triggers errors like this:

  ERROR:  missing chunk number 0 for toast value 16433 in pg_toast_16426

A less severe consequence are inconsistent failures due to the index row
being too large, depending on whether brin_form_tuple operated on plain
or toasted version of the row. For example

    CREATE TABLE t (val TEXT);
    INSERT INTO t VALUES ('... long value ...')
    CREATE INDEX idx ON t USING brin (val);

would likely succeed, as the row would likely include toast pointer.
Switching the order of INSERT and CREATE INDEX would likely fail:

    ERROR:  index row size 8712 exceeds maximum 8152 for index "idx"

because this happens before the row values are toasted.

The bug exists since PostgreSQL 9.5 where BRIN indexes were introduced.
So backpatch all the way back.

Author: Tomas Vondra
Reviewed-by: Alvaro Herrera
Backpatch-through: 9.5
Discussion: https://postgr.es/m/20201001184133.oq5uq75sb45pu3aw@development
Discussion: https://postgr.es/m/20201104010544.zexj52mlldagzowv%40development

src/backend/access/brin/brin_tuple.c
src/test/regress/expected/brin.out
src/test/regress/sql/brin.sql

index 5a7462b59cfa4de1cd7ae8d9a4e3f492c2ff3be3..08243866a2951690094882e2b5eb6b43b43b6df8 100644 (file)
@@ -35,6 +35,7 @@
 #include "access/brin_tuple.h"
 #include "access/tupdesc.h"
 #include "access/tupmacs.h"
+#include "access/tuptoaster.h"
 #include "utils/datum.h"
 #include "utils/memutils.h"
 
@@ -100,6 +101,12 @@ brin_form_tuple(BrinDesc *brdesc, BlockNumber blkno, BrinMemTuple *tuple,
    Size        len,
                hoff,
                data_len;
+   int         i;
+
+#ifdef TOAST_INDEX_HACK
+   Datum      *untoasted_values;
+   int         nuntoasted = 0;
+#endif
 
    Assert(brdesc->bd_totalstored > 0);
 
@@ -108,6 +115,10 @@ brin_form_tuple(BrinDesc *brdesc, BlockNumber blkno, BrinMemTuple *tuple,
    phony_nullbitmap = (bits8 *)
        palloc(sizeof(bits8) * BITMAPLEN(brdesc->bd_totalstored));
 
+#ifdef TOAST_INDEX_HACK
+   untoasted_values = (Datum *) palloc(sizeof(Datum) * brdesc->bd_totalstored);
+#endif
+
    /*
     * Set up the values/nulls arrays for heap_fill_tuple
     */
@@ -139,10 +150,83 @@ brin_form_tuple(BrinDesc *brdesc, BlockNumber blkno, BrinMemTuple *tuple,
        if (tuple->bt_columns[keyno].bv_hasnulls)
            anynulls = true;
 
+       /*
+        * Now obtain the values of each stored datum.  Note that some values
+        * might be toasted, and we cannot rely on the original heap values
+        * sticking around forever, so we must detoast them.  Also try to
+        * compress them.
+        */
        for (datumno = 0;
             datumno < brdesc->bd_info[keyno]->oi_nstored;
             datumno++)
-           values[idxattno++] = tuple->bt_columns[keyno].bv_values[datumno];
+       {
+           Datum value = tuple->bt_columns[keyno].bv_values[datumno];
+
+#ifdef TOAST_INDEX_HACK
+
+           /* We must look at the stored type, not at the index descriptor. */
+           TypeCacheEntry  *atttype = brdesc->bd_info[keyno]->oi_typcache[datumno];
+
+           /* Do we need to free the value at the end? */
+           bool free_value = false;
+
+           /* For non-varlena types we don't need to do anything special */
+           if (atttype->typlen != -1)
+           {
+               values[idxattno++] = value;
+               continue;
+           }
+
+           /*
+            * Do nothing if value is not of varlena type. We don't need to
+            * care about NULL values here, thanks to bv_allnulls above.
+            *
+            * If value is stored EXTERNAL, must fetch it so we are not
+            * depending on outside storage.
+            *
+            * XXX Is this actually true? Could it be that the summary is
+            * NULL even for range with non-NULL data? E.g. degenerate bloom
+            * filter may be thrown away, etc.
+            */
+           if (VARATT_IS_EXTERNAL(DatumGetPointer(value)))
+           {
+               value = PointerGetDatum(heap_tuple_fetch_attr((struct varlena *)
+                                                             DatumGetPointer(value)));
+               free_value = true;
+           }
+
+           /*
+            * If value is above size target, and is of a compressible datatype,
+            * try to compress it in-line.
+            */
+           if (!VARATT_IS_EXTENDED(DatumGetPointer(value)) &&
+               VARSIZE(DatumGetPointer(value)) > TOAST_INDEX_TARGET &&
+               (atttype->typstorage == 'x' || atttype->typstorage == 'm'))
+           {
+               Datum       cvalue = toast_compress_datum(value);
+
+               if (DatumGetPointer(cvalue) != NULL)
+               {
+                   /* successful compression */
+                   if (free_value)
+                       pfree(DatumGetPointer(value));
+
+                   value = cvalue;
+                   free_value = true;
+               }
+           }
+
+           /*
+            * If we untoasted / compressed the value, we need to free it
+            * after forming the index tuple.
+            */
+           if (free_value)
+               untoasted_values[nuntoasted++] = value;
+
+#endif
+
+           values[idxattno++] = value;
+       }
    }
 
    /* Assert we did not overrun temp arrays */
@@ -194,6 +278,11 @@ brin_form_tuple(BrinDesc *brdesc, BlockNumber blkno, BrinMemTuple *tuple,
    pfree(nulls);
    pfree(phony_nullbitmap);
 
+#ifdef TOAST_INDEX_HACK
+   for (i = 0; i < nuntoasted; i++)
+       pfree(DatumGetPointer(untoasted_values[i]));
+#endif
+
    /*
     * Now fill in the real null bitmasks.  allnulls first.
     */
index abfc6746aa8a370afa57862c9de0eecaa7e2a071..1f8a817ab24d045dcf258ed8f2e79084cf8fb458 100644 (file)
@@ -418,3 +418,45 @@ SELECT brin_summarize_new_values('brinidx'); -- ok, no change expected
                          0
 (1 row)
 
+-- make sure data are properly de-toasted in BRIN index
+CREATE TABLE brintest_3 (a text, b text, c text, d text);
+-- long random strings (~2000 chars each, so ~6kB for min/max on two
+-- columns) to trigger toasting
+WITH rand_value AS (SELECT string_agg(md5(i::text),'') AS val FROM generate_series(1,60) s(i))
+INSERT INTO brintest_3
+SELECT val, val, val, val FROM rand_value;
+CREATE INDEX brin_test_toast_idx ON brintest_3 USING brin (b, c);
+DELETE FROM brintest_3;
+-- We need to wait a bit for all transactions to complete, so that the
+-- vacuum actually removes the TOAST rows. Creating an index concurrently
+-- is a one way to achieve that, because it does exactly such wait.
+CREATE INDEX CONCURRENTLY brin_test_temp_idx ON brintest_3(a);
+DROP INDEX brin_test_temp_idx;
+-- vacuum the table, to discard TOAST data
+VACUUM brintest_3;
+-- retry insert with a different random-looking (but deterministic) value
+-- the value is different, and so should replace either min or max in the
+-- brin summary
+WITH rand_value AS (SELECT string_agg(md5((-i)::text),'') AS val FROM generate_series(1,60) s(i))
+INSERT INTO brintest_3
+SELECT val, val, val, val FROM rand_value;
+-- now try some queries, accessing the brin index
+SET enable_seqscan = off;
+SET enable_bitmapscan = on;
+EXPLAIN (COSTS OFF)
+SELECT * FROM brintest_3 WHERE b < '0';
+                   QUERY PLAN                   
+------------------------------------------------
+ Bitmap Heap Scan on brintest_3
+   Recheck Cond: (b < '0'::text)
+   ->  Bitmap Index Scan on brin_test_toast_idx
+         Index Cond: (b < '0'::text)
+(4 rows)
+
+SELECT * FROM brintest_3 WHERE b < '0';
+ a | b | c | d 
+---+---+---+---
+(0 rows)
+
+DROP TABLE brintest_3;
+RESET enable_seqscan;
index 7398d12fb7488cc0b00e7eb69a5acdf6136f6af9..fb344eb0df26e7c948a8e9d4641c3b141f66a2a7 100644 (file)
@@ -421,3 +421,43 @@ UPDATE brintest SET textcol = '' WHERE textcol IS NOT NULL;
 SELECT brin_summarize_new_values('brintest'); -- error, not an index
 SELECT brin_summarize_new_values('tenk1_unique1'); -- error, not a BRIN index
 SELECT brin_summarize_new_values('brinidx'); -- ok, no change expected
+
+-- make sure data are properly de-toasted in BRIN index
+CREATE TABLE brintest_3 (a text, b text, c text, d text);
+
+-- long random strings (~2000 chars each, so ~6kB for min/max on two
+-- columns) to trigger toasting
+WITH rand_value AS (SELECT string_agg(md5(i::text),'') AS val FROM generate_series(1,60) s(i))
+INSERT INTO brintest_3
+SELECT val, val, val, val FROM rand_value;
+
+CREATE INDEX brin_test_toast_idx ON brintest_3 USING brin (b, c);
+DELETE FROM brintest_3;
+
+-- We need to wait a bit for all transactions to complete, so that the
+-- vacuum actually removes the TOAST rows. Creating an index concurrently
+-- is a one way to achieve that, because it does exactly such wait.
+CREATE INDEX CONCURRENTLY brin_test_temp_idx ON brintest_3(a);
+DROP INDEX brin_test_temp_idx;
+
+-- vacuum the table, to discard TOAST data
+VACUUM brintest_3;
+
+-- retry insert with a different random-looking (but deterministic) value
+-- the value is different, and so should replace either min or max in the
+-- brin summary
+WITH rand_value AS (SELECT string_agg(md5((-i)::text),'') AS val FROM generate_series(1,60) s(i))
+INSERT INTO brintest_3
+SELECT val, val, val, val FROM rand_value;
+
+-- now try some queries, accessing the brin index
+SET enable_seqscan = off;
+SET enable_bitmapscan = on;
+
+EXPLAIN (COSTS OFF)
+SELECT * FROM brintest_3 WHERE b < '0';
+
+SELECT * FROM brintest_3 WHERE b < '0';
+
+DROP TABLE brintest_3;
+RESET enable_seqscan;