LCOV - code coverage report
Current view: top level - src/backend/access/brin - brin_validate.c (source / functions) Hit Total Coverage
Test: PostgreSQL 18devel Lines: 69 89 77.5 %
Date: 2025-02-22 07:14:56 Functions: 1 1 100.0 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : /*-------------------------------------------------------------------------
       2             :  *
       3             :  * brin_validate.c
       4             :  *    Opclass validator for BRIN.
       5             :  *
       6             :  * Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group
       7             :  * Portions Copyright (c) 1994, Regents of the University of California
       8             :  *
       9             :  * IDENTIFICATION
      10             :  *    src/backend/access/brin/brin_validate.c
      11             :  *
      12             :  *-------------------------------------------------------------------------
      13             :  */
      14             : #include "postgres.h"
      15             : 
      16             : #include "access/amvalidate.h"
      17             : #include "access/brin_internal.h"
      18             : #include "access/htup_details.h"
      19             : #include "catalog/pg_amop.h"
      20             : #include "catalog/pg_amproc.h"
      21             : #include "catalog/pg_opclass.h"
      22             : #include "catalog/pg_type.h"
      23             : #include "utils/builtins.h"
      24             : #include "utils/lsyscache.h"
      25             : #include "utils/regproc.h"
      26             : #include "utils/syscache.h"
      27             : 
      28             : /*
      29             :  * Validator for a BRIN opclass.
      30             :  *
      31             :  * Some of the checks done here cover the whole opfamily, and therefore are
      32             :  * redundant when checking each opclass in a family.  But they don't run long
      33             :  * enough to be much of a problem, so we accept the duplication rather than
      34             :  * complicate the amvalidate API.
      35             :  */
      36             : bool
      37         432 : brinvalidate(Oid opclassoid)
      38             : {
      39         432 :     bool        result = true;
      40             :     HeapTuple   classtup;
      41             :     Form_pg_opclass classform;
      42             :     Oid         opfamilyoid;
      43             :     Oid         opcintype;
      44             :     char       *opclassname;
      45             :     char       *opfamilyname;
      46             :     CatCList   *proclist,
      47             :                *oprlist;
      48         432 :     uint64      allfuncs = 0;
      49         432 :     uint64      allops = 0;
      50             :     List       *grouplist;
      51             :     OpFamilyOpFuncGroup *opclassgroup;
      52             :     int         i;
      53             :     ListCell   *lc;
      54             : 
      55             :     /* Fetch opclass information */
      56         432 :     classtup = SearchSysCache1(CLAOID, ObjectIdGetDatum(opclassoid));
      57         432 :     if (!HeapTupleIsValid(classtup))
      58           0 :         elog(ERROR, "cache lookup failed for operator class %u", opclassoid);
      59         432 :     classform = (Form_pg_opclass) GETSTRUCT(classtup);
      60             : 
      61         432 :     opfamilyoid = classform->opcfamily;
      62         432 :     opcintype = classform->opcintype;
      63         432 :     opclassname = NameStr(classform->opcname);
      64             : 
      65             :     /* Fetch opfamily information */
      66         432 :     opfamilyname = get_opfamily_name(opfamilyoid, false);
      67             : 
      68             :     /* Fetch all operators and support functions of the opfamily */
      69         432 :     oprlist = SearchSysCacheList1(AMOPSTRATEGY, ObjectIdGetDatum(opfamilyoid));
      70         432 :     proclist = SearchSysCacheList1(AMPROCNUM, ObjectIdGetDatum(opfamilyoid));
      71             : 
      72             :     /* Check individual support functions */
      73        4068 :     for (i = 0; i < proclist->n_members; i++)
      74             :     {
      75        3636 :         HeapTuple   proctup = &proclist->members[i]->tuple;
      76        3636 :         Form_pg_amproc procform = (Form_pg_amproc) GETSTRUCT(proctup);
      77             :         bool        ok;
      78             : 
      79             :         /* Check procedure numbers and function signatures */
      80        3636 :         switch (procform->amprocnum)
      81             :         {
      82         684 :             case BRIN_PROCNUM_OPCINFO:
      83         684 :                 ok = check_amproc_signature(procform->amproc, INTERNALOID, true,
      84             :                                             1, 1, INTERNALOID);
      85         684 :                 break;
      86         684 :             case BRIN_PROCNUM_ADDVALUE:
      87         684 :                 ok = check_amproc_signature(procform->amproc, BOOLOID, true,
      88             :                                             4, 4, INTERNALOID, INTERNALOID,
      89             :                                             INTERNALOID, INTERNALOID);
      90         684 :                 break;
      91         684 :             case BRIN_PROCNUM_CONSISTENT:
      92         684 :                 ok = check_amproc_signature(procform->amproc, BOOLOID, true,
      93             :                                             3, 4, INTERNALOID, INTERNALOID,
      94             :                                             INTERNALOID, INT4OID);
      95         684 :                 break;
      96         684 :             case BRIN_PROCNUM_UNION:
      97         684 :                 ok = check_amproc_signature(procform->amproc, BOOLOID, true,
      98             :                                             3, 3, INTERNALOID, INTERNALOID,
      99             :                                             INTERNALOID);
     100         684 :                 break;
     101         426 :             case BRIN_PROCNUM_OPTIONS:
     102         426 :                 ok = check_amoptsproc_signature(procform->amproc);
     103         426 :                 break;
     104         474 :             default:
     105             :                 /* Complain if it's not a valid optional proc number */
     106         474 :                 if (procform->amprocnum < BRIN_FIRST_OPTIONAL_PROCNUM ||
     107         474 :                     procform->amprocnum > BRIN_LAST_OPTIONAL_PROCNUM)
     108             :                 {
     109           0 :                     ereport(INFO,
     110             :                             (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
     111             :                              errmsg("operator family \"%s\" of access method %s contains function %s with invalid support number %d",
     112             :                                     opfamilyname, "brin",
     113             :                                     format_procedure(procform->amproc),
     114             :                                     procform->amprocnum)));
     115           0 :                     result = false;
     116           0 :                     continue;   /* omit bad proc numbers from allfuncs */
     117             :                 }
     118             :                 /* Can't check signatures of optional procs, so assume OK */
     119         474 :                 ok = true;
     120         474 :                 break;
     121             :         }
     122             : 
     123        3636 :         if (!ok)
     124             :         {
     125           0 :             ereport(INFO,
     126             :                     (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
     127             :                      errmsg("operator family \"%s\" of access method %s contains function %s with wrong signature for support number %d",
     128             :                             opfamilyname, "brin",
     129             :                             format_procedure(procform->amproc),
     130             :                             procform->amprocnum)));
     131           0 :             result = false;
     132             :         }
     133             : 
     134             :         /* Track all valid procedure numbers seen in opfamily */
     135        3636 :         allfuncs |= ((uint64) 1) << procform->amprocnum;
     136             :     }
     137             : 
     138             :     /* Check individual operators */
     139        5448 :     for (i = 0; i < oprlist->n_members; i++)
     140             :     {
     141        5016 :         HeapTuple   oprtup = &oprlist->members[i]->tuple;
     142        5016 :         Form_pg_amop oprform = (Form_pg_amop) GETSTRUCT(oprtup);
     143             : 
     144             :         /* Check that only allowed strategy numbers exist */
     145        5016 :         if (oprform->amopstrategy < 1 || oprform->amopstrategy > 63)
     146             :         {
     147           0 :             ereport(INFO,
     148             :                     (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
     149             :                      errmsg("operator family \"%s\" of access method %s contains operator %s with invalid strategy number %d",
     150             :                             opfamilyname, "brin",
     151             :                             format_operator(oprform->amopopr),
     152             :                             oprform->amopstrategy)));
     153           0 :             result = false;
     154             :         }
     155             :         else
     156             :         {
     157             :             /*
     158             :              * The set of operators supplied varies across BRIN opfamilies.
     159             :              * Our plan is to identify all operator strategy numbers used in
     160             :              * the opfamily and then complain about datatype combinations that
     161             :              * are missing any operator(s).  However, consider only numbers
     162             :              * that appear in some non-cross-type case, since cross-type
     163             :              * operators may have unique strategies.  (This is not a great
     164             :              * heuristic, in particular an erroneous number used in a
     165             :              * cross-type operator will not get noticed; but the core BRIN
     166             :              * opfamilies are messy enough to make it necessary.)
     167             :              */
     168        5016 :             if (oprform->amoplefttype == oprform->amoprighttype)
     169        2604 :                 allops |= ((uint64) 1) << oprform->amopstrategy;
     170             :         }
     171             : 
     172             :         /* brin doesn't support ORDER BY operators */
     173        5016 :         if (oprform->amoppurpose != AMOP_SEARCH ||
     174        5016 :             OidIsValid(oprform->amopsortfamily))
     175             :         {
     176           0 :             ereport(INFO,
     177             :                     (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
     178             :                      errmsg("operator family \"%s\" of access method %s contains invalid ORDER BY specification for operator %s",
     179             :                             opfamilyname, "brin",
     180             :                             format_operator(oprform->amopopr))));
     181           0 :             result = false;
     182             :         }
     183             : 
     184             :         /* Check operator signature --- same for all brin strategies */
     185        5016 :         if (!check_amop_signature(oprform->amopopr, BOOLOID,
     186             :                                   oprform->amoplefttype,
     187             :                                   oprform->amoprighttype))
     188             :         {
     189           0 :             ereport(INFO,
     190             :                     (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
     191             :                      errmsg("operator family \"%s\" of access method %s contains operator %s with wrong signature",
     192             :                             opfamilyname, "brin",
     193             :                             format_operator(oprform->amopopr))));
     194           0 :             result = false;
     195             :         }
     196             :     }
     197             : 
     198             :     /* Now check for inconsistent groups of operators/functions */
     199         432 :     grouplist = identify_opfamily_groups(oprlist, proclist);
     200         432 :     opclassgroup = NULL;
     201        1608 :     foreach(lc, grouplist)
     202             :     {
     203        1176 :         OpFamilyOpFuncGroup *thisgroup = (OpFamilyOpFuncGroup *) lfirst(lc);
     204             : 
     205             :         /* Remember the group exactly matching the test opclass */
     206        1176 :         if (thisgroup->lefttype == opcintype &&
     207         612 :             thisgroup->righttype == opcintype)
     208         432 :             opclassgroup = thisgroup;
     209             : 
     210             :         /*
     211             :          * Some BRIN opfamilies expect cross-type support functions to exist,
     212             :          * and some don't.  We don't know exactly which are which, so if we
     213             :          * find a cross-type operator for which there are no support functions
     214             :          * at all, let it pass.  (Don't expect that all operators exist for
     215             :          * such cross-type cases, either.)
     216             :          */
     217        1176 :         if (thisgroup->functionset == 0 &&
     218         492 :             thisgroup->lefttype != thisgroup->righttype)
     219         492 :             continue;
     220             : 
     221             :         /*
     222             :          * Else complain if there seems to be an incomplete set of either
     223             :          * operators or support functions for this datatype pair.
     224             :          */
     225         684 :         if (thisgroup->operatorset != allops)
     226             :         {
     227           0 :             ereport(INFO,
     228             :                     (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
     229             :                      errmsg("operator family \"%s\" of access method %s is missing operator(s) for types %s and %s",
     230             :                             opfamilyname, "brin",
     231             :                             format_type_be(thisgroup->lefttype),
     232             :                             format_type_be(thisgroup->righttype))));
     233           0 :             result = false;
     234             :         }
     235         684 :         if (thisgroup->functionset != allfuncs)
     236             :         {
     237           0 :             ereport(INFO,
     238             :                     (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
     239             :                      errmsg("operator family \"%s\" of access method %s is missing support function(s) for types %s and %s",
     240             :                             opfamilyname, "brin",
     241             :                             format_type_be(thisgroup->lefttype),
     242             :                             format_type_be(thisgroup->righttype))));
     243           0 :             result = false;
     244             :         }
     245             :     }
     246             : 
     247             :     /* Check that the originally-named opclass is complete */
     248         432 :     if (!opclassgroup || opclassgroup->operatorset != allops)
     249             :     {
     250           0 :         ereport(INFO,
     251             :                 (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
     252             :                  errmsg("operator class \"%s\" of access method %s is missing operator(s)",
     253             :                         opclassname, "brin")));
     254           0 :         result = false;
     255             :     }
     256        2160 :     for (i = 1; i <= BRIN_MANDATORY_NPROCS; i++)
     257             :     {
     258        1728 :         if (opclassgroup &&
     259        1728 :             (opclassgroup->functionset & (((int64) 1) << i)) != 0)
     260        1728 :             continue;           /* got it */
     261           0 :         ereport(INFO,
     262             :                 (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
     263             :                  errmsg("operator class \"%s\" of access method %s is missing support function %d",
     264             :                         opclassname, "brin", i)));
     265           0 :         result = false;
     266             :     }
     267             : 
     268         432 :     ReleaseCatCacheList(proclist);
     269         432 :     ReleaseCatCacheList(oprlist);
     270         432 :     ReleaseSysCache(classtup);
     271             : 
     272         432 :     return result;
     273             : }

Generated by: LCOV version 1.14