Branch data Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * parse_coerce.c
4 : : * handle type coercions/conversions for parser
5 : : *
6 : : * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
7 : : * Portions Copyright (c) 1994, Regents of the University of California
8 : : *
9 : : *
10 : : * IDENTIFICATION
11 : : * src/backend/parser/parse_coerce.c
12 : : *
13 : : *-------------------------------------------------------------------------
14 : : */
15 : : #include "postgres.h"
16 : :
17 : : #include "access/htup_details.h"
18 : : #include "catalog/pg_cast.h"
19 : : #include "catalog/pg_class.h"
20 : : #include "catalog/pg_inherits.h"
21 : : #include "catalog/pg_proc.h"
22 : : #include "catalog/pg_type.h"
23 : : #include "nodes/makefuncs.h"
24 : : #include "nodes/nodeFuncs.h"
25 : : #include "parser/parse_coerce.h"
26 : : #include "parser/parse_relation.h"
27 : : #include "parser/parse_type.h"
28 : : #include "utils/builtins.h"
29 : : #include "utils/datum.h" /* needed for datumIsEqual() */
30 : : #include "utils/fmgroids.h"
31 : : #include "utils/lsyscache.h"
32 : : #include "utils/syscache.h"
33 : : #include "utils/typcache.h"
34 : :
35 : :
36 : : static Node *coerce_type_typmod(Node *node,
37 : : Oid targetTypeId, int32 targetTypMod,
38 : : CoercionContext ccontext, CoercionForm cformat,
39 : : int location,
40 : : bool hideInputCoercion);
41 : : static void hide_coercion_node(Node *node);
42 : : static Node *build_coercion_expression(Node *node,
43 : : CoercionPathType pathtype,
44 : : Oid funcId,
45 : : Oid targetTypeId, int32 targetTypMod,
46 : : CoercionContext ccontext, CoercionForm cformat,
47 : : int location);
48 : : static Node *coerce_record_to_complex(ParseState *pstate, Node *node,
49 : : Oid targetTypeId,
50 : : CoercionContext ccontext,
51 : : CoercionForm cformat,
52 : : int location);
53 : : static bool is_complex_array(Oid typid);
54 : : static bool typeIsOfTypedTable(Oid reltypeId, Oid reloftypeId);
55 : :
56 : :
57 : : /*
58 : : * coerce_to_target_type()
59 : : * Convert an expression to a target type and typmod.
60 : : *
61 : : * This is the general-purpose entry point for arbitrary type coercion
62 : : * operations. Direct use of the component operations can_coerce_type,
63 : : * coerce_type, and coerce_type_typmod should be restricted to special
64 : : * cases (eg, when the conversion is expected to succeed).
65 : : *
66 : : * Returns the possibly-transformed expression tree, or NULL if the type
67 : : * conversion is not possible. (We do this, rather than ereport'ing directly,
68 : : * so that callers can generate custom error messages indicating context.)
69 : : *
70 : : * pstate - parse state (can be NULL, see coerce_type)
71 : : * expr - input expression tree (already transformed by transformExpr)
72 : : * exprtype - result type of expr
73 : : * targettype - desired result type
74 : : * targettypmod - desired result typmod
75 : : * ccontext, cformat - context indicators to control coercions
76 : : * location - parse location of the coercion request, or -1 if unknown/implicit
77 : : */
78 : : Node *
79 : 52269 : coerce_to_target_type(ParseState *pstate, Node *expr, Oid exprtype,
80 : : Oid targettype, int32 targettypmod,
81 : : CoercionContext ccontext,
82 : : CoercionForm cformat,
83 : : int location)
84 : : {
85 : 52269 : Node *result;
86 : 52269 : Node *origexpr;
87 : :
88 [ + + ]: 52269 : if (!can_coerce_type(1, &exprtype, &targettype, ccontext))
89 : 54 : return NULL;
90 : :
91 : : /*
92 : : * If the input has a CollateExpr at the top, strip it off, perform the
93 : : * coercion, and put a new one back on. This is annoying since it
94 : : * duplicates logic in coerce_type, but if we don't do this then it's too
95 : : * hard to tell whether coerce_type actually changed anything, and we
96 : : * *must* know that to avoid possibly calling hide_coercion_node on
97 : : * something that wasn't generated by coerce_type. Note that if there are
98 : : * multiple stacked CollateExprs, we just discard all but the topmost.
99 : : * Also, if the target type isn't collatable, we discard the CollateExpr.
100 : : */
101 : 52215 : origexpr = expr;
102 [ - + + + ]: 52225 : while (expr && IsA(expr, CollateExpr))
103 : 10 : expr = (Node *) ((CollateExpr *) expr)->arg;
104 : :
105 : 104430 : result = coerce_type(pstate, expr, exprtype,
106 : 52215 : targettype, targettypmod,
107 : 52215 : ccontext, cformat, location);
108 : :
109 : : /*
110 : : * If the target is a fixed-length type, it may need a length coercion as
111 : : * well as a type coercion. If we find ourselves adding both, force the
112 : : * inner coercion node to implicit display form.
113 : : */
114 : 84240 : result = coerce_type_typmod(result,
115 : 52215 : targettype, targettypmod,
116 : 52215 : ccontext, cformat, location,
117 [ + + ]: 52215 : (result != expr && !IsA(result, Const)));
118 : :
119 [ + + + + ]: 52215 : if (expr != origexpr && type_is_collatable(targettype))
120 : : {
121 : : /* Reinstall top CollateExpr */
122 : 8 : CollateExpr *coll = (CollateExpr *) origexpr;
123 : 8 : CollateExpr *newcoll = makeNode(CollateExpr);
124 : :
125 : 8 : newcoll->arg = (Expr *) result;
126 : 8 : newcoll->collOid = coll->collOid;
127 : 8 : newcoll->location = coll->location;
128 : 8 : result = (Node *) newcoll;
129 : 8 : }
130 : :
131 : 52215 : return result;
132 : 52269 : }
133 : :
134 : :
135 : : /*
136 : : * coerce_type()
137 : : * Convert an expression to a different type.
138 : : *
139 : : * The caller should already have determined that the coercion is possible;
140 : : * see can_coerce_type.
141 : : *
142 : : * Normally, no coercion to a typmod (length) is performed here. The caller
143 : : * must call coerce_type_typmod as well, if a typmod constraint is wanted.
144 : : * (But if the target type is a domain, it may internally contain a
145 : : * typmod constraint, which will be applied inside coerce_to_domain.)
146 : : * In some cases pg_cast specifies a type coercion function that also
147 : : * applies length conversion, and in those cases only, the result will
148 : : * already be properly coerced to the specified typmod.
149 : : *
150 : : * pstate is only used in the case that we are able to resolve the type of
151 : : * a previously UNKNOWN Param. It is okay to pass pstate = NULL if the
152 : : * caller does not want type information updated for Params.
153 : : *
154 : : * Note: this function must not modify the given expression tree, only add
155 : : * decoration on top of it. See transformSetOperationTree, for example.
156 : : */
157 : : Node *
158 : 120295 : coerce_type(ParseState *pstate, Node *node,
159 : : Oid inputTypeId, Oid targetTypeId, int32 targetTypeMod,
160 : : CoercionContext ccontext, CoercionForm cformat, int location)
161 : : {
162 : 120295 : Node *result;
163 : 120295 : CoercionPathType pathtype;
164 : 120295 : Oid funcId;
165 : :
166 [ + + - + ]: 120295 : if (targetTypeId == inputTypeId ||
167 : 101063 : node == NULL)
168 : : {
169 : : /* no conversion needed */
170 : 19232 : return node;
171 : : }
172 [ + + ]: 101063 : if (targetTypeId == ANYOID ||
173 [ + + ]: 98985 : targetTypeId == ANYELEMENTOID ||
174 [ + + ]: 98005 : targetTypeId == ANYNONARRAYOID ||
175 [ + - - + ]: 96088 : targetTypeId == ANYCOMPATIBLEOID ||
176 : 96088 : targetTypeId == ANYCOMPATIBLENONARRAYOID)
177 : : {
178 : : /*
179 : : * Assume can_coerce_type verified that implicit coercion is okay.
180 : : *
181 : : * Note: by returning the unmodified node here, we are saying that
182 : : * it's OK to treat an UNKNOWN constant as a valid input for a
183 : : * function accepting one of these pseudotypes. This should be all
184 : : * right, since an UNKNOWN value is still a perfectly valid Datum.
185 : : *
186 : : * NB: we do NOT want a RelabelType here: the exposed type of the
187 : : * function argument must be its actual type, not the polymorphic
188 : : * pseudotype.
189 : : */
190 : 4975 : return node;
191 : : }
192 [ + + ]: 96088 : if (targetTypeId == ANYARRAYOID ||
193 [ + + ]: 92905 : targetTypeId == ANYENUMOID ||
194 [ + + ]: 92839 : targetTypeId == ANYRANGEOID ||
195 [ + + ]: 91567 : targetTypeId == ANYMULTIRANGEOID ||
196 [ + - ]: 90689 : targetTypeId == ANYCOMPATIBLEARRAYOID ||
197 [ + - - + ]: 90689 : targetTypeId == ANYCOMPATIBLERANGEOID ||
198 : 90689 : targetTypeId == ANYCOMPATIBLEMULTIRANGEOID)
199 : : {
200 : : /*
201 : : * Assume can_coerce_type verified that implicit coercion is okay.
202 : : *
203 : : * These cases are unlike the ones above because the exposed type of
204 : : * the argument must be an actual array, enum, range, or multirange
205 : : * type. In particular the argument must *not* be an UNKNOWN
206 : : * constant. If it is, we just fall through; below, we'll call the
207 : : * pseudotype's input function, which will produce an error. Also, if
208 : : * what we have is a domain over array, enum, range, or multirange, we
209 : : * have to relabel it to its base type.
210 : : *
211 : : * Note: currently, we can't actually see a domain-over-enum here,
212 : : * since the other functions in this file will not match such a
213 : : * parameter to ANYENUM. But that should get changed eventually.
214 : : */
215 [ + + ]: 5399 : if (inputTypeId != UNKNOWNOID)
216 : : {
217 : 5391 : Oid baseTypeId = getBaseType(inputTypeId);
218 : :
219 [ + + ]: 5391 : if (baseTypeId != inputTypeId)
220 : : {
221 : 18 : RelabelType *r = makeRelabelType((Expr *) node,
222 : 9 : baseTypeId, -1,
223 : : InvalidOid,
224 : 9 : cformat);
225 : :
226 : 9 : r->location = location;
227 : 9 : return (Node *) r;
228 : 9 : }
229 : : /* Not a domain type, so return it as-is */
230 : 5382 : return node;
231 : 5391 : }
232 : 8 : }
233 [ + + + + ]: 90697 : if (inputTypeId == UNKNOWNOID && IsA(node, Const))
234 : : {
235 : : /*
236 : : * Input is a string constant with previously undetermined type. Apply
237 : : * the target type's typinput function to it to produce a constant of
238 : : * the target type.
239 : : *
240 : : * NOTE: this case cannot be folded together with the other
241 : : * constant-input case, since the typinput function does not
242 : : * necessarily behave the same as a type conversion function. For
243 : : * example, int4's typinput function will reject "1.2", whereas
244 : : * float-to-int type conversion will round to integer.
245 : : *
246 : : * XXX if the typinput function is not immutable, we really ought to
247 : : * postpone evaluation of the function call until runtime. But there
248 : : * is no way to represent a typinput function call as an expression
249 : : * tree, because C-string values are not Datums. (XXX This *is*
250 : : * possible as of 7.3, do we want to do it?)
251 : : */
252 : 66630 : Const *con = (Const *) node;
253 : 66630 : Const *newcon = makeNode(Const);
254 : 66630 : Oid baseTypeId;
255 : 66630 : int32 baseTypeMod;
256 : 66630 : int32 inputTypeMod;
257 : 66630 : Type baseType;
258 : 66630 : ParseCallbackState pcbstate;
259 : :
260 : : /*
261 : : * If the target type is a domain, we want to call its base type's
262 : : * input routine, not domain_in(). This is to avoid premature failure
263 : : * when the domain applies a typmod: existing input routines follow
264 : : * implicit-coercion semantics for length checks, which is not always
265 : : * what we want here. The needed check will be applied properly
266 : : * inside coerce_to_domain().
267 : : */
268 : 66630 : baseTypeMod = targetTypeMod;
269 : 66630 : baseTypeId = getBaseTypeAndTypmod(targetTypeId, &baseTypeMod);
270 : :
271 : : /*
272 : : * For most types we pass typmod -1 to the input routine, because
273 : : * existing input routines follow implicit-coercion semantics for
274 : : * length checks, which is not always what we want here. Any length
275 : : * constraint will be applied later by our caller. An exception
276 : : * however is the INTERVAL type, for which we *must* pass the typmod
277 : : * or it won't be able to obey the bizarre SQL-spec input rules. (Ugly
278 : : * as sin, but so is this part of the spec...)
279 : : */
280 [ + + ]: 66630 : if (baseTypeId == INTERVALOID)
281 : 839 : inputTypeMod = baseTypeMod;
282 : : else
283 : 65791 : inputTypeMod = -1;
284 : :
285 : 66630 : baseType = typeidType(baseTypeId);
286 : :
287 : 66630 : newcon->consttype = baseTypeId;
288 : 66630 : newcon->consttypmod = inputTypeMod;
289 : 66630 : newcon->constcollid = typeTypeCollation(baseType);
290 : 66630 : newcon->constlen = typeLen(baseType);
291 : 66630 : newcon->constbyval = typeByVal(baseType);
292 : 66630 : newcon->constisnull = con->constisnull;
293 : :
294 : : /*
295 : : * We use the original literal's location regardless of the position
296 : : * of the coercion. This is a change from pre-9.2 behavior, meant to
297 : : * simplify life for pg_stat_statements.
298 : : */
299 : 66630 : newcon->location = con->location;
300 : :
301 : : /*
302 : : * Set up to point at the constant's text if the input routine throws
303 : : * an error.
304 : : */
305 : 66630 : setup_parser_errposition_callback(&pcbstate, pstate, con->location);
306 : :
307 : : /*
308 : : * We assume here that UNKNOWN's internal representation is the same
309 : : * as CSTRING.
310 : : */
311 [ + + ]: 66630 : if (!con->constisnull)
312 : 120610 : newcon->constvalue = stringTypeDatum(baseType,
313 : 60305 : DatumGetCString(con->constvalue),
314 : 60305 : inputTypeMod);
315 : : else
316 : 12650 : newcon->constvalue = stringTypeDatum(baseType,
317 : : NULL,
318 : 6325 : inputTypeMod);
319 : :
320 : : /*
321 : : * If it's a varlena value, force it to be in non-expanded
322 : : * (non-toasted) format; this avoids any possible dependency on
323 : : * external values and improves consistency of representation.
324 : : */
325 [ + + + + ]: 66630 : if (!con->constisnull && newcon->constlen == -1)
326 : 33133 : newcon->constvalue =
327 : 33133 : PointerGetDatum(PG_DETOAST_DATUM(newcon->constvalue));
328 : :
329 : : #ifdef RANDOMIZE_ALLOCATED_MEMORY
330 : :
331 : : /*
332 : : * For pass-by-reference data types, repeat the conversion to see if
333 : : * the input function leaves any uninitialized bytes in the result. We
334 : : * can only detect that reliably if RANDOMIZE_ALLOCATED_MEMORY is
335 : : * enabled, so we don't bother testing otherwise. The reason we don't
336 : : * want any instability in the input function is that comparison of
337 : : * Const nodes relies on bytewise comparison of the datums, so if the
338 : : * input function leaves garbage then subexpressions that should be
339 : : * identical may not get recognized as such. See pgsql-hackers
340 : : * discussion of 2008-04-04.
341 : : */
342 : : if (!con->constisnull && !newcon->constbyval)
343 : : {
344 : : Datum val2;
345 : :
346 : : val2 = stringTypeDatum(baseType,
347 : : DatumGetCString(con->constvalue),
348 : : inputTypeMod);
349 : : if (newcon->constlen == -1)
350 : : val2 = PointerGetDatum(PG_DETOAST_DATUM(val2));
351 : : if (!datumIsEqual(newcon->constvalue, val2, false, newcon->constlen))
352 : : elog(WARNING, "type %s has unstable input conversion for \"%s\"",
353 : : typeTypeName(baseType), DatumGetCString(con->constvalue));
354 : : }
355 : : #endif
356 : :
357 : 66630 : cancel_parser_errposition_callback(&pcbstate);
358 : :
359 : 66630 : result = (Node *) newcon;
360 : :
361 : : /* If target is a domain, apply constraints. */
362 [ + + ]: 66630 : if (baseTypeId != targetTypeId)
363 : 976 : result = coerce_to_domain(result,
364 : 488 : baseTypeId, baseTypeMod,
365 : 488 : targetTypeId,
366 : 488 : ccontext, cformat, location,
367 : : false);
368 : :
369 : 66630 : ReleaseSysCache(baseType);
370 : :
371 : 66630 : return result;
372 : 66630 : }
373 [ + + ]: 24067 : if (IsA(node, Param) &&
374 [ + + + + ]: 957 : pstate != NULL && pstate->p_coerce_param_hook != NULL)
375 : : {
376 : : /*
377 : : * Allow the CoerceParamHook to decide what happens. It can return a
378 : : * transformed node (very possibly the same Param node), or return
379 : : * NULL to indicate we should proceed with normal coercion.
380 : : */
381 : 202 : result = pstate->p_coerce_param_hook(pstate,
382 : 101 : (Param *) node,
383 : 101 : targetTypeId,
384 : 101 : targetTypeMod,
385 : 101 : location);
386 [ + + ]: 101 : if (result)
387 : 93 : return result;
388 : 8 : }
389 [ + + ]: 23974 : if (IsA(node, CollateExpr))
390 : : {
391 : : /*
392 : : * If we have a COLLATE clause, we have to push the coercion
393 : : * underneath the COLLATE; or discard the COLLATE if the target type
394 : : * isn't collatable. This is really ugly, but there is little choice
395 : : * because the above hacks on Consts and Params wouldn't happen
396 : : * otherwise. This kluge has consequences in coerce_to_target_type.
397 : : */
398 : 1250 : CollateExpr *coll = (CollateExpr *) node;
399 : :
400 : 2500 : result = coerce_type(pstate, (Node *) coll->arg,
401 : 1250 : inputTypeId, targetTypeId, targetTypeMod,
402 : 1250 : ccontext, cformat, location);
403 [ - + ]: 1250 : if (type_is_collatable(targetTypeId))
404 : : {
405 : 1250 : CollateExpr *newcoll = makeNode(CollateExpr);
406 : :
407 : 1250 : newcoll->arg = (Expr *) result;
408 : 1250 : newcoll->collOid = coll->collOid;
409 : 1250 : newcoll->location = coll->location;
410 : 1250 : result = (Node *) newcoll;
411 : 1250 : }
412 : 1250 : return result;
413 : 1250 : }
414 : 22724 : pathtype = find_coercion_pathway(targetTypeId, inputTypeId, ccontext,
415 : : &funcId);
416 [ + + ]: 22724 : if (pathtype != COERCION_PATH_NONE)
417 : : {
418 : 22229 : Oid baseTypeId;
419 : 22229 : int32 baseTypeMod;
420 : :
421 : 22229 : baseTypeMod = targetTypeMod;
422 : 22229 : baseTypeId = getBaseTypeAndTypmod(targetTypeId, &baseTypeMod);
423 : :
424 [ + + ]: 22229 : if (pathtype != COERCION_PATH_RELABELTYPE)
425 : : {
426 : : /*
427 : : * Generate an expression tree representing run-time application
428 : : * of the conversion function. If we are dealing with a domain
429 : : * target type, the conversion function will yield the base type,
430 : : * and we need to extract the correct typmod to use from the
431 : : * domain's typtypmod.
432 : : */
433 : 21406 : result = build_coercion_expression(node, pathtype, funcId,
434 : 10703 : baseTypeId, baseTypeMod,
435 : 10703 : ccontext, cformat, location);
436 : :
437 : : /*
438 : : * If domain, coerce to the domain type and relabel with domain
439 : : * type ID, hiding the previous coercion node.
440 : : */
441 [ + + ]: 10703 : if (targetTypeId != baseTypeId)
442 : 74 : result = coerce_to_domain(result, baseTypeId, baseTypeMod,
443 : 37 : targetTypeId,
444 : 37 : ccontext, cformat, location,
445 : : true);
446 : 10703 : }
447 : : else
448 : : {
449 : : /*
450 : : * We don't need to do a physical conversion, but we do need to
451 : : * attach a RelabelType node so that the expression will be seen
452 : : * to have the intended type when inspected by higher-level code.
453 : : *
454 : : * Also, domains may have value restrictions beyond the base type
455 : : * that must be accounted for. If the destination is a domain
456 : : * then we won't need a RelabelType node.
457 : : */
458 : 23052 : result = coerce_to_domain(node, baseTypeId, baseTypeMod,
459 : 11526 : targetTypeId,
460 : 11526 : ccontext, cformat, location,
461 : : false);
462 [ + + ]: 11526 : if (result == node)
463 : : {
464 : : /*
465 : : * XXX could we label result with exprTypmod(node) instead of
466 : : * default -1 typmod, to save a possible length-coercion
467 : : * later? Would work if both types have same interpretation of
468 : : * typmod, which is likely but not certain.
469 : : */
470 : 21686 : RelabelType *r = makeRelabelType((Expr *) result,
471 : 10843 : targetTypeId, -1,
472 : : InvalidOid,
473 : 10843 : cformat);
474 : :
475 : 10843 : r->location = location;
476 : 10843 : result = (Node *) r;
477 : 10843 : }
478 : : }
479 : 22229 : return result;
480 : 22229 : }
481 [ + + - + ]: 495 : if (inputTypeId == RECORDOID &&
482 : 260 : ISCOMPLEX(targetTypeId))
483 : : {
484 : : /* Coerce a RECORD to a specific complex type */
485 : 520 : return coerce_record_to_complex(pstate, node, targetTypeId,
486 : 260 : ccontext, cformat, location);
487 : : }
488 [ + + - + ]: 235 : if (targetTypeId == RECORDOID &&
489 : 222 : ISCOMPLEX(inputTypeId))
490 : : {
491 : : /* Coerce a specific complex type to RECORD */
492 : : /* NB: we do NOT want a RelabelType here */
493 : 222 : return node;
494 : : }
495 : : #ifdef NOT_USED
496 : : if (inputTypeId == RECORDARRAYOID &&
497 : : is_complex_array(targetTypeId))
498 : : {
499 : : /* Coerce record[] to a specific complex array type */
500 : : /* not implemented yet ... */
501 : : }
502 : : #endif
503 [ + + - + ]: 13 : if (targetTypeId == RECORDARRAYOID &&
504 : 2 : is_complex_array(inputTypeId))
505 : : {
506 : : /* Coerce a specific complex array type to record[] */
507 : : /* NB: we do NOT want a RelabelType here */
508 : 2 : return node;
509 : : }
510 : 11 : if (typeInheritsFrom(inputTypeId, targetTypeId)
511 [ + + + - ]: 11 : || typeIsOfTypedTable(inputTypeId, targetTypeId))
512 : : {
513 : : /*
514 : : * Input class type is a subclass of target, so generate an
515 : : * appropriate runtime conversion (removing unneeded columns and
516 : : * possibly rearranging the ones that are wanted).
517 : : *
518 : : * We will also get here when the input is a domain over a subclass of
519 : : * the target type. To keep life simple for the executor, we define
520 : : * ConvertRowtypeExpr as only working between regular composite types;
521 : : * therefore, in such cases insert a RelabelType to smash the input
522 : : * expression down to its base type.
523 : : */
524 : 11 : Oid baseTypeId = getBaseType(inputTypeId);
525 : 11 : ConvertRowtypeExpr *r = makeNode(ConvertRowtypeExpr);
526 : :
527 [ + - ]: 11 : if (baseTypeId != inputTypeId)
528 : : {
529 : 0 : RelabelType *rt = makeRelabelType((Expr *) node,
530 : 0 : baseTypeId, -1,
531 : : InvalidOid,
532 : : COERCE_IMPLICIT_CAST);
533 : :
534 : 0 : rt->location = location;
535 : 0 : node = (Node *) rt;
536 : 0 : }
537 : 11 : r->arg = (Expr *) node;
538 : 11 : r->resulttype = targetTypeId;
539 : 11 : r->convertformat = cformat;
540 : 11 : r->location = location;
541 : 11 : return (Node *) r;
542 : 11 : }
543 : : /* If we get here, caller blew it */
544 [ # # # # ]: 0 : elog(ERROR, "failed to find conversion function from %s to %s",
545 : : format_type_be(inputTypeId), format_type_be(targetTypeId));
546 : 0 : return NULL; /* keep compiler quiet */
547 : 120295 : }
548 : :
549 : :
550 : : /*
551 : : * can_coerce_type()
552 : : * Can input_typeids be coerced to target_typeids?
553 : : *
554 : : * We must be told the context (CAST construct, assignment, implicit coercion)
555 : : * as this determines the set of available casts.
556 : : */
557 : : bool
558 : 155890 : can_coerce_type(int nargs, const Oid *input_typeids, const Oid *target_typeids,
559 : : CoercionContext ccontext)
560 : : {
561 : 155890 : bool have_generics = false;
562 : 155890 : int i;
563 : :
564 : : /* run through argument list... */
565 [ + + ]: 280952 : for (i = 0; i < nargs; i++)
566 : : {
567 : 181335 : Oid inputTypeId = input_typeids[i];
568 : 181335 : Oid targetTypeId = target_typeids[i];
569 : 181335 : CoercionPathType pathtype;
570 : 181335 : Oid funcId;
571 : :
572 : : /* no problem if same type */
573 [ + + ]: 181335 : if (inputTypeId == targetTypeId)
574 : 28440 : continue;
575 : :
576 : : /* accept if target is ANY */
577 [ + + ]: 152895 : if (targetTypeId == ANYOID)
578 : 2120 : continue;
579 : :
580 : : /* accept if target is polymorphic, for now */
581 [ + + + + : 150775 : if (IsPolymorphicType(targetTypeId))
+ + + + +
+ + + + +
+ + + + +
+ + + ]
582 : : {
583 : 18380 : have_generics = true; /* do more checking later */
584 : 18380 : continue;
585 : : }
586 : :
587 : : /*
588 : : * If input is an untyped string constant, assume we can convert it to
589 : : * anything.
590 : : */
591 [ + + ]: 132395 : if (inputTypeId == UNKNOWNOID)
592 : 54722 : continue;
593 : :
594 : : /*
595 : : * If pg_cast shows that we can coerce, accept. This test now covers
596 : : * both binary-compatible and coercion-function cases.
597 : : */
598 : 77673 : pathtype = find_coercion_pathway(targetTypeId, inputTypeId, ccontext,
599 : : &funcId);
600 [ + + ]: 77673 : if (pathtype != COERCION_PATH_NONE)
601 : 20958 : continue;
602 : :
603 : : /*
604 : : * If input is RECORD and target is a composite type, assume we can
605 : : * coerce (may need tighter checking here)
606 : : */
607 [ + + + + ]: 56715 : if (inputTypeId == RECORDOID &&
608 : 283 : ISCOMPLEX(targetTypeId))
609 : 260 : continue;
610 : :
611 : : /*
612 : : * If input is a composite type and target is RECORD, accept
613 : : */
614 [ + + + + ]: 56455 : if (targetTypeId == RECORDOID &&
615 : 916 : ISCOMPLEX(inputTypeId))
616 : 171 : continue;
617 : :
618 : : #ifdef NOT_USED /* not implemented yet */
619 : :
620 : : /*
621 : : * If input is record[] and target is a composite array type, assume
622 : : * we can coerce (may need tighter checking here)
623 : : */
624 : : if (inputTypeId == RECORDARRAYOID &&
625 : : is_complex_array(targetTypeId))
626 : : continue;
627 : : #endif
628 : :
629 : : /*
630 : : * If input is a composite array type and target is record[], accept
631 : : */
632 [ + + + - ]: 56284 : if (targetTypeId == RECORDARRAYOID &&
633 : 2 : is_complex_array(inputTypeId))
634 : 0 : continue;
635 : :
636 : : /*
637 : : * If input is a class type that inherits from target, accept
638 : : */
639 : 56284 : if (typeInheritsFrom(inputTypeId, targetTypeId)
640 [ + + + + ]: 56284 : || typeIsOfTypedTable(inputTypeId, targetTypeId))
641 : 11 : continue;
642 : :
643 : : /*
644 : : * Else, cannot coerce at this argument position
645 : : */
646 : 56273 : return false;
647 [ + + ]: 181335 : }
648 : :
649 : : /* If we found any generic argument types, cross-check them */
650 [ + + ]: 99617 : if (have_generics)
651 : : {
652 [ + + + + ]: 26388 : if (!check_generic_type_consistency(input_typeids, target_typeids,
653 : 13194 : nargs))
654 : 6683 : return false;
655 : 6511 : }
656 : :
657 : 92934 : return true;
658 : 155890 : }
659 : :
660 : :
661 : : /*
662 : : * Create an expression tree to represent coercion to a domain type.
663 : : *
664 : : * 'arg': input expression
665 : : * 'baseTypeId': base type of domain
666 : : * 'baseTypeMod': base type typmod of domain
667 : : * 'typeId': target type to coerce to
668 : : * 'ccontext': context indicator to control coercions
669 : : * 'cformat': coercion display format
670 : : * 'location': coercion request location
671 : : * 'hideInputCoercion': if true, hide the input coercion under this one.
672 : : *
673 : : * If the target type isn't a domain, the given 'arg' is returned as-is.
674 : : */
675 : : Node *
676 : 12155 : coerce_to_domain(Node *arg, Oid baseTypeId, int32 baseTypeMod, Oid typeId,
677 : : CoercionContext ccontext, CoercionForm cformat, int location,
678 : : bool hideInputCoercion)
679 : : {
680 : 12155 : CoerceToDomain *result;
681 : :
682 : : /* We now require the caller to supply correct baseTypeId/baseTypeMod */
683 [ + - ]: 12155 : Assert(OidIsValid(baseTypeId));
684 : :
685 : : /* If it isn't a domain, return the node as it was passed in */
686 [ + + ]: 12155 : if (baseTypeId == typeId)
687 : 10843 : return arg;
688 : :
689 : : /* Suppress display of nested coercion steps */
690 [ + + ]: 1312 : if (hideInputCoercion)
691 : 37 : hide_coercion_node(arg);
692 : :
693 : : /*
694 : : * If the domain applies a typmod to its base type, build the appropriate
695 : : * coercion step. Mark it implicit for display purposes, because we don't
696 : : * want it shown separately by ruleutils.c; but the isExplicit flag passed
697 : : * to the conversion function depends on the manner in which the domain
698 : : * coercion is invoked, so that the semantics of implicit and explicit
699 : : * coercion differ. (Is that really the behavior we want?)
700 : : *
701 : : * NOTE: because we apply this as part of the fixed expression structure,
702 : : * ALTER DOMAIN cannot alter the typtypmod. But it's unclear that that
703 : : * would be safe to do anyway, without lots of knowledge about what the
704 : : * base type thinks the typmod means.
705 : : */
706 : 2624 : arg = coerce_type_typmod(arg, baseTypeId, baseTypeMod,
707 : 1312 : ccontext, COERCE_IMPLICIT_CAST, location,
708 : : false);
709 : :
710 : : /*
711 : : * Now build the domain coercion node. This represents run-time checking
712 : : * of any constraints currently attached to the domain. This also ensures
713 : : * that the expression is properly labeled as to result type.
714 : : */
715 : 1312 : result = makeNode(CoerceToDomain);
716 : 1312 : result->arg = (Expr *) arg;
717 : 1312 : result->resulttype = typeId;
718 : 1312 : result->resulttypmod = -1; /* currently, always -1 for domains */
719 : : /* resultcollid will be set by parse_collate.c */
720 : 1312 : result->coercionformat = cformat;
721 : 1312 : result->location = location;
722 : :
723 : 1312 : return (Node *) result;
724 : 12155 : }
725 : :
726 : :
727 : : /*
728 : : * coerce_type_typmod()
729 : : * Force a value to a particular typmod, if meaningful and possible.
730 : : *
731 : : * This is applied to values that are going to be stored in a relation
732 : : * (where we have an atttypmod for the column) as well as values being
733 : : * explicitly CASTed (where the typmod comes from the target type spec).
734 : : *
735 : : * The caller must have already ensured that the value is of the correct
736 : : * type, typically by applying coerce_type.
737 : : *
738 : : * ccontext may affect semantics, depending on whether the length coercion
739 : : * function pays attention to the isExplicit flag it's passed.
740 : : *
741 : : * cformat determines the display properties of the generated node (if any).
742 : : *
743 : : * If hideInputCoercion is true *and* we generate a node, the input node is
744 : : * forced to IMPLICIT display form, so that only the typmod coercion node will
745 : : * be visible when displaying the expression.
746 : : *
747 : : * NOTE: this does not need to work on domain types, because any typmod
748 : : * coercion for a domain is considered to be part of the type coercion
749 : : * needed to produce the domain value in the first place. So, no getBaseType.
750 : : */
751 : : static Node *
752 : 52737 : coerce_type_typmod(Node *node, Oid targetTypeId, int32 targetTypMod,
753 : : CoercionContext ccontext, CoercionForm cformat,
754 : : int location,
755 : : bool hideInputCoercion)
756 : : {
757 : 52737 : CoercionPathType pathtype;
758 : 52737 : Oid funcId;
759 : :
760 : : /* Skip coercion if already done */
761 [ + + ]: 52737 : if (targetTypMod == exprTypmod(node))
762 : 50419 : return node;
763 : :
764 : : /* Suppress display of nested coercion steps */
765 [ + + ]: 2318 : if (hideInputCoercion)
766 : 101 : hide_coercion_node(node);
767 : :
768 : : /*
769 : : * A negative typmod means that no actual coercion is needed, but we still
770 : : * want a RelabelType to ensure that the expression exposes the intended
771 : : * typmod.
772 : : */
773 [ + + ]: 2318 : if (targetTypMod < 0)
774 : 5 : pathtype = COERCION_PATH_NONE;
775 : : else
776 : 2313 : pathtype = find_typmod_coercion_function(targetTypeId, &funcId);
777 : :
778 [ + + ]: 2318 : if (pathtype != COERCION_PATH_NONE)
779 : : {
780 : 4622 : node = build_coercion_expression(node, pathtype, funcId,
781 : 2311 : targetTypeId, targetTypMod,
782 : 2311 : ccontext, cformat, location);
783 : 2311 : }
784 : : else
785 : : {
786 : : /*
787 : : * We don't need to perform any actual coercion step, but we should
788 : : * apply a RelabelType to ensure that the expression exposes the
789 : : * intended typmod.
790 : : */
791 : 14 : node = applyRelabelType(node, targetTypeId, targetTypMod,
792 : 7 : exprCollation(node),
793 : 7 : cformat, location, false);
794 : : }
795 : :
796 : 2318 : return node;
797 : 52737 : }
798 : :
799 : : /*
800 : : * Mark a coercion node as IMPLICIT so it will never be displayed by
801 : : * ruleutils.c. We use this when we generate a nest of coercion nodes
802 : : * to implement what is logically one conversion; the inner nodes are
803 : : * forced to IMPLICIT_CAST format. This does not change their semantics,
804 : : * only display behavior.
805 : : *
806 : : * It is caller error to call this on something that doesn't have a
807 : : * CoercionForm field.
808 : : */
809 : : static void
810 : 138 : hide_coercion_node(Node *node)
811 : : {
812 [ + + ]: 138 : if (IsA(node, FuncExpr))
813 : 58 : ((FuncExpr *) node)->funcformat = COERCE_IMPLICIT_CAST;
814 [ + + ]: 80 : else if (IsA(node, RelabelType))
815 : 41 : ((RelabelType *) node)->relabelformat = COERCE_IMPLICIT_CAST;
816 [ + + ]: 39 : else if (IsA(node, CoerceViaIO))
817 : 37 : ((CoerceViaIO *) node)->coerceformat = COERCE_IMPLICIT_CAST;
818 [ + - ]: 2 : else if (IsA(node, ArrayCoerceExpr))
819 : 2 : ((ArrayCoerceExpr *) node)->coerceformat = COERCE_IMPLICIT_CAST;
820 [ # # ]: 0 : else if (IsA(node, ConvertRowtypeExpr))
821 : 0 : ((ConvertRowtypeExpr *) node)->convertformat = COERCE_IMPLICIT_CAST;
822 [ # # ]: 0 : else if (IsA(node, RowExpr))
823 : 0 : ((RowExpr *) node)->row_format = COERCE_IMPLICIT_CAST;
824 [ # # ]: 0 : else if (IsA(node, CoerceToDomain))
825 : 0 : ((CoerceToDomain *) node)->coercionformat = COERCE_IMPLICIT_CAST;
826 : : else
827 [ # # # # ]: 0 : elog(ERROR, "unsupported node type: %d", (int) nodeTag(node));
828 : 138 : }
829 : :
830 : : /*
831 : : * build_coercion_expression()
832 : : * Construct an expression tree for applying a pg_cast entry.
833 : : *
834 : : * This is used for both type-coercion and length-coercion operations,
835 : : * since there is no difference in terms of the calling convention.
836 : : */
837 : : static Node *
838 : 13014 : build_coercion_expression(Node *node,
839 : : CoercionPathType pathtype,
840 : : Oid funcId,
841 : : Oid targetTypeId, int32 targetTypMod,
842 : : CoercionContext ccontext, CoercionForm cformat,
843 : : int location)
844 : : {
845 : 13014 : int nargs = 0;
846 : :
847 [ + + ]: 13014 : if (OidIsValid(funcId))
848 : : {
849 : 9294 : HeapTuple tp;
850 : 9294 : Form_pg_proc procstruct;
851 : :
852 : 9294 : tp = SearchSysCache1(PROCOID, ObjectIdGetDatum(funcId));
853 [ + - ]: 9294 : if (!HeapTupleIsValid(tp))
854 [ # # # # ]: 0 : elog(ERROR, "cache lookup failed for function %u", funcId);
855 : 9294 : procstruct = (Form_pg_proc) GETSTRUCT(tp);
856 : :
857 : : /*
858 : : * These Asserts essentially check that function is a legal coercion
859 : : * function. We can't make the seemingly obvious tests on prorettype
860 : : * and proargtypes[0], even in the COERCION_PATH_FUNC case, because of
861 : : * various binary-compatibility cases.
862 : : */
863 : : /* Assert(targetTypeId == procstruct->prorettype); */
864 [ + - ]: 9294 : Assert(!procstruct->proretset);
865 [ + - ]: 9294 : Assert(procstruct->prokind == PROKIND_FUNCTION);
866 : 9294 : nargs = procstruct->pronargs;
867 [ + - ]: 9294 : Assert(nargs >= 1 && nargs <= 3);
868 : : /* Assert(procstruct->proargtypes.values[0] == exprType(node)); */
869 [ + + + - ]: 9294 : Assert(nargs < 2 || procstruct->proargtypes.values[1] == INT4OID);
870 [ + + + - ]: 9294 : Assert(nargs < 3 || procstruct->proargtypes.values[2] == BOOLOID);
871 : :
872 : 9294 : ReleaseSysCache(tp);
873 : 9294 : }
874 : :
875 [ + + ]: 13014 : if (pathtype == COERCION_PATH_FUNC)
876 : : {
877 : : /* We build an ordinary FuncExpr with special arguments */
878 : 9281 : FuncExpr *fexpr;
879 : 9281 : List *args;
880 : 9281 : Const *cons;
881 : :
882 [ + - ]: 9281 : Assert(OidIsValid(funcId));
883 : :
884 : 9281 : args = list_make1(node);
885 : :
886 [ + + ]: 9281 : if (nargs >= 2)
887 : : {
888 : : /* Pass target typmod as an int4 constant */
889 : 2510 : cons = makeConst(INT4OID,
890 : : -1,
891 : : InvalidOid,
892 : : sizeof(int32),
893 : 2510 : Int32GetDatum(targetTypMod),
894 : : false,
895 : : true);
896 : :
897 : 2510 : args = lappend(args, cons);
898 : 2510 : }
899 : :
900 [ + + ]: 9281 : if (nargs == 3)
901 : : {
902 : : /* Pass it a boolean isExplicit parameter, too */
903 : 1535 : cons = makeConst(BOOLOID,
904 : : -1,
905 : : InvalidOid,
906 : : sizeof(bool),
907 : 1535 : BoolGetDatum(ccontext == COERCION_EXPLICIT),
908 : : false,
909 : : true);
910 : :
911 : 1535 : args = lappend(args, cons);
912 : 1535 : }
913 : :
914 : 18562 : fexpr = makeFuncExpr(funcId, targetTypeId, args,
915 : 9281 : InvalidOid, InvalidOid, cformat);
916 : 9281 : fexpr->location = location;
917 : 9281 : return (Node *) fexpr;
918 : 9281 : }
919 [ + + ]: 3733 : else if (pathtype == COERCION_PATH_ARRAYCOERCE)
920 : : {
921 : : /* We need to build an ArrayCoerceExpr */
922 : 675 : ArrayCoerceExpr *acoerce = makeNode(ArrayCoerceExpr);
923 : 675 : CaseTestExpr *ctest = makeNode(CaseTestExpr);
924 : 675 : Oid sourceBaseTypeId;
925 : 675 : int32 sourceBaseTypeMod;
926 : 675 : Oid targetElementType;
927 : 675 : Node *elemexpr;
928 : :
929 : : /*
930 : : * Look through any domain over the source array type. Note we don't
931 : : * expect that the target type is a domain; it must be a plain array.
932 : : * (To get to a domain target type, we'll do coerce_to_domain later.)
933 : : */
934 : 675 : sourceBaseTypeMod = exprTypmod(node);
935 : 675 : sourceBaseTypeId = getBaseTypeAndTypmod(exprType(node),
936 : : &sourceBaseTypeMod);
937 : :
938 : : /*
939 : : * Set up a CaseTestExpr representing one element of the source array.
940 : : * This is an abuse of CaseTestExpr, but it's OK as long as there
941 : : * can't be any CaseExpr or ArrayCoerceExpr within the completed
942 : : * elemexpr.
943 : : */
944 : 675 : ctest->typeId = get_element_type(sourceBaseTypeId);
945 [ + - ]: 675 : Assert(OidIsValid(ctest->typeId));
946 : 675 : ctest->typeMod = sourceBaseTypeMod;
947 : 675 : ctest->collation = InvalidOid; /* Assume coercions don't care */
948 : :
949 : : /* And coerce it to the target element type */
950 : 675 : targetElementType = get_element_type(targetTypeId);
951 [ + - ]: 675 : Assert(OidIsValid(targetElementType));
952 : :
953 : 675 : elemexpr = coerce_to_target_type(NULL,
954 : 675 : (Node *) ctest,
955 : 675 : ctest->typeId,
956 : 675 : targetElementType,
957 : 675 : targetTypMod,
958 : 675 : ccontext,
959 : 675 : cformat,
960 : 675 : location);
961 [ + - ]: 675 : if (elemexpr == NULL) /* shouldn't happen */
962 [ # # # # ]: 0 : elog(ERROR, "failed to coerce array element type as expected");
963 : :
964 : 675 : acoerce->arg = (Expr *) node;
965 : 675 : acoerce->elemexpr = (Expr *) elemexpr;
966 : 675 : acoerce->resulttype = targetTypeId;
967 : :
968 : : /*
969 : : * Label the output as having a particular element typmod only if we
970 : : * ended up with a per-element expression that is labeled that way.
971 : : */
972 : 675 : acoerce->resulttypmod = exprTypmod(elemexpr);
973 : : /* resultcollid will be set by parse_collate.c */
974 : 675 : acoerce->coerceformat = cformat;
975 : 675 : acoerce->location = location;
976 : :
977 : 675 : return (Node *) acoerce;
978 : 675 : }
979 [ + - ]: 3058 : else if (pathtype == COERCION_PATH_COERCEVIAIO)
980 : : {
981 : : /* We need to build a CoerceViaIO node */
982 : 3058 : CoerceViaIO *iocoerce = makeNode(CoerceViaIO);
983 : :
984 [ + - ]: 3058 : Assert(!OidIsValid(funcId));
985 : :
986 : 3058 : iocoerce->arg = (Expr *) node;
987 : 3058 : iocoerce->resulttype = targetTypeId;
988 : : /* resultcollid will be set by parse_collate.c */
989 : 3058 : iocoerce->coerceformat = cformat;
990 : 3058 : iocoerce->location = location;
991 : :
992 : 3058 : return (Node *) iocoerce;
993 : 3058 : }
994 : : else
995 : : {
996 [ # # # # ]: 0 : elog(ERROR, "unsupported pathtype %d in build_coercion_expression",
997 : : (int) pathtype);
998 : 0 : return NULL; /* keep compiler quiet */
999 : : }
1000 : 13014 : }
1001 : :
1002 : :
1003 : : /*
1004 : : * coerce_record_to_complex
1005 : : * Coerce a RECORD to a specific composite type.
1006 : : *
1007 : : * Currently we only support this for inputs that are RowExprs or whole-row
1008 : : * Vars.
1009 : : */
1010 : : static Node *
1011 : 260 : coerce_record_to_complex(ParseState *pstate, Node *node,
1012 : : Oid targetTypeId,
1013 : : CoercionContext ccontext,
1014 : : CoercionForm cformat,
1015 : : int location)
1016 : : {
1017 : 260 : RowExpr *rowexpr;
1018 : 260 : Oid baseTypeId;
1019 : 260 : int32 baseTypeMod = -1;
1020 : 260 : TupleDesc tupdesc;
1021 : 260 : List *args = NIL;
1022 : 260 : List *newargs;
1023 : 260 : int i;
1024 : 260 : int ucolno;
1025 : 260 : ListCell *arg;
1026 : :
1027 [ + - - + ]: 260 : if (node && IsA(node, RowExpr))
1028 : : {
1029 : : /*
1030 : : * Since the RowExpr must be of type RECORD, we needn't worry about it
1031 : : * containing any dropped columns.
1032 : : */
1033 : 260 : args = ((RowExpr *) node)->args;
1034 : 260 : }
1035 [ # # ]: 0 : else if (node && IsA(node, Var) &&
1036 : 0 : ((Var *) node)->varattno == InvalidAttrNumber)
1037 : : {
1038 : 0 : int rtindex = ((Var *) node)->varno;
1039 : 0 : int sublevels_up = ((Var *) node)->varlevelsup;
1040 : 0 : int vlocation = ((Var *) node)->location;
1041 : 0 : ParseNamespaceItem *nsitem;
1042 : :
1043 : 0 : nsitem = GetNSItemByRangeTablePosn(pstate, rtindex, sublevels_up);
1044 : 0 : args = expandNSItemVars(pstate, nsitem, sublevels_up, vlocation, NULL);
1045 : 0 : }
1046 : : else
1047 [ # # # # ]: 0 : ereport(ERROR,
1048 : : (errcode(ERRCODE_CANNOT_COERCE),
1049 : : errmsg("cannot cast type %s to %s",
1050 : : format_type_be(RECORDOID),
1051 : : format_type_be(targetTypeId)),
1052 : : parser_coercion_errposition(pstate, location, node)));
1053 : :
1054 : : /*
1055 : : * Look up the composite type, accounting for possibility that what we are
1056 : : * given is a domain over composite.
1057 : : */
1058 : 260 : baseTypeId = getBaseTypeAndTypmod(targetTypeId, &baseTypeMod);
1059 : 260 : tupdesc = lookup_rowtype_tupdesc(baseTypeId, baseTypeMod);
1060 : :
1061 : : /* Process the fields */
1062 : 260 : newargs = NIL;
1063 : 260 : ucolno = 1;
1064 : 260 : arg = list_head(args);
1065 [ + + ]: 838 : for (i = 0; i < tupdesc->natts; i++)
1066 : : {
1067 : 578 : Node *expr;
1068 : 578 : Node *cexpr;
1069 : 578 : Oid exprtype;
1070 : 578 : Form_pg_attribute attr = TupleDescAttr(tupdesc, i);
1071 : :
1072 : : /* Fill in NULLs for dropped columns in rowtype */
1073 [ - + ]: 578 : if (attr->attisdropped)
1074 : : {
1075 : : /*
1076 : : * can't use atttypid here, but it doesn't really matter what type
1077 : : * the Const claims to be.
1078 : : */
1079 : 0 : newargs = lappend(newargs,
1080 : 0 : makeNullConst(INT4OID, -1, InvalidOid));
1081 : 0 : continue;
1082 : : }
1083 : :
1084 [ + - ]: 578 : if (arg == NULL)
1085 [ # # # # ]: 0 : ereport(ERROR,
1086 : : (errcode(ERRCODE_CANNOT_COERCE),
1087 : : errmsg("cannot cast type %s to %s",
1088 : : format_type_be(RECORDOID),
1089 : : format_type_be(targetTypeId)),
1090 : : errdetail("Input has too few columns."),
1091 : : parser_coercion_errposition(pstate, location, node)));
1092 : 578 : expr = (Node *) lfirst(arg);
1093 : 578 : exprtype = exprType(expr);
1094 : :
1095 : 1156 : cexpr = coerce_to_target_type(pstate,
1096 : 578 : expr, exprtype,
1097 : 578 : attr->atttypid,
1098 : 578 : attr->atttypmod,
1099 : 578 : ccontext,
1100 : : COERCE_IMPLICIT_CAST,
1101 : : -1);
1102 [ + - ]: 578 : if (cexpr == NULL)
1103 [ # # # # ]: 0 : ereport(ERROR,
1104 : : (errcode(ERRCODE_CANNOT_COERCE),
1105 : : errmsg("cannot cast type %s to %s",
1106 : : format_type_be(RECORDOID),
1107 : : format_type_be(targetTypeId)),
1108 : : errdetail("Cannot cast type %s to %s in column %d.",
1109 : : format_type_be(exprtype),
1110 : : format_type_be(attr->atttypid),
1111 : : ucolno),
1112 : : parser_coercion_errposition(pstate, location, expr)));
1113 : 578 : newargs = lappend(newargs, cexpr);
1114 : 578 : ucolno++;
1115 : 578 : arg = lnext(args, arg);
1116 [ - - + ]: 578 : }
1117 [ + - ]: 260 : if (arg != NULL)
1118 [ # # # # ]: 0 : ereport(ERROR,
1119 : : (errcode(ERRCODE_CANNOT_COERCE),
1120 : : errmsg("cannot cast type %s to %s",
1121 : : format_type_be(RECORDOID),
1122 : : format_type_be(targetTypeId)),
1123 : : errdetail("Input has too many columns."),
1124 : : parser_coercion_errposition(pstate, location, node)));
1125 : :
1126 [ - + ]: 260 : ReleaseTupleDesc(tupdesc);
1127 : :
1128 : 260 : rowexpr = makeNode(RowExpr);
1129 : 260 : rowexpr->args = newargs;
1130 : 260 : rowexpr->row_typeid = baseTypeId;
1131 : 260 : rowexpr->row_format = cformat;
1132 : 260 : rowexpr->colnames = NIL; /* not needed for named target type */
1133 : 260 : rowexpr->location = location;
1134 : :
1135 : : /* If target is a domain, apply constraints */
1136 [ + + ]: 260 : if (baseTypeId != targetTypeId)
1137 : : {
1138 : 21 : rowexpr->row_format = COERCE_IMPLICIT_CAST;
1139 : 42 : return coerce_to_domain((Node *) rowexpr,
1140 : 21 : baseTypeId, baseTypeMod,
1141 : 21 : targetTypeId,
1142 : 21 : ccontext, cformat, location,
1143 : : false);
1144 : : }
1145 : :
1146 : 239 : return (Node *) rowexpr;
1147 : 260 : }
1148 : :
1149 : : /*
1150 : : * coerce_to_boolean()
1151 : : * Coerce an argument of a construct that requires boolean input
1152 : : * (AND, OR, NOT, etc). Also check that input is not a set.
1153 : : *
1154 : : * Returns the possibly-transformed node tree.
1155 : : *
1156 : : * As with coerce_type, pstate may be NULL if no special unknown-Param
1157 : : * processing is wanted.
1158 : : */
1159 : : Node *
1160 : 65726 : coerce_to_boolean(ParseState *pstate, Node *node,
1161 : : const char *constructName)
1162 : : {
1163 : 65726 : Oid inputTypeId = exprType(node);
1164 : :
1165 [ + + ]: 65726 : if (inputTypeId != BOOLOID)
1166 : : {
1167 : 12 : Node *newnode;
1168 : :
1169 : 12 : newnode = coerce_to_target_type(pstate, node, inputTypeId,
1170 : : BOOLOID, -1,
1171 : : COERCION_ASSIGNMENT,
1172 : : COERCE_IMPLICIT_CAST,
1173 : : -1);
1174 [ + + ]: 12 : if (newnode == NULL)
1175 [ + - + - ]: 1 : ereport(ERROR,
1176 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
1177 : : /* translator: first %s is name of a SQL construct, eg WHERE */
1178 : : errmsg("argument of %s must be type %s, not type %s",
1179 : : constructName, "boolean",
1180 : : format_type_be(inputTypeId)),
1181 : : parser_errposition(pstate, exprLocation(node))));
1182 : 11 : node = newnode;
1183 : 11 : }
1184 : :
1185 [ + - ]: 65725 : if (expression_returns_set(node))
1186 [ # # # # ]: 0 : ereport(ERROR,
1187 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
1188 : : /* translator: %s is name of a SQL construct, eg WHERE */
1189 : : errmsg("argument of %s must not return a set",
1190 : : constructName),
1191 : : parser_errposition(pstate, exprLocation(node))));
1192 : :
1193 : 131450 : return node;
1194 : 65725 : }
1195 : :
1196 : : /*
1197 : : * coerce_to_specific_type_typmod()
1198 : : * Coerce an argument of a construct that requires a specific data type,
1199 : : * with a specific typmod. Also check that input is not a set.
1200 : : *
1201 : : * Returns the possibly-transformed node tree.
1202 : : *
1203 : : * As with coerce_type, pstate may be NULL if no special unknown-Param
1204 : : * processing is wanted.
1205 : : */
1206 : : Node *
1207 : 1383 : coerce_to_specific_type_typmod(ParseState *pstate, Node *node,
1208 : : Oid targetTypeId, int32 targetTypmod,
1209 : : const char *constructName)
1210 : : {
1211 : 1383 : Oid inputTypeId = exprType(node);
1212 : :
1213 [ + + ]: 1383 : if (inputTypeId != targetTypeId)
1214 : : {
1215 : 1131 : Node *newnode;
1216 : :
1217 : 2262 : newnode = coerce_to_target_type(pstate, node, inputTypeId,
1218 : 1131 : targetTypeId, targetTypmod,
1219 : : COERCION_ASSIGNMENT,
1220 : : COERCE_IMPLICIT_CAST,
1221 : : -1);
1222 [ + + ]: 1131 : if (newnode == NULL)
1223 [ + - + - ]: 1 : ereport(ERROR,
1224 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
1225 : : /* translator: first %s is name of a SQL construct, eg LIMIT */
1226 : : errmsg("argument of %s must be type %s, not type %s",
1227 : : constructName,
1228 : : format_type_be(targetTypeId),
1229 : : format_type_be(inputTypeId)),
1230 : : parser_errposition(pstate, exprLocation(node))));
1231 : 1130 : node = newnode;
1232 : 1130 : }
1233 : :
1234 [ + - ]: 1382 : if (expression_returns_set(node))
1235 [ # # # # ]: 0 : ereport(ERROR,
1236 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
1237 : : /* translator: %s is name of a SQL construct, eg LIMIT */
1238 : : errmsg("argument of %s must not return a set",
1239 : : constructName),
1240 : : parser_errposition(pstate, exprLocation(node))));
1241 : :
1242 : 2764 : return node;
1243 : 1382 : }
1244 : :
1245 : : /*
1246 : : * coerce_to_specific_type()
1247 : : * Coerce an argument of a construct that requires a specific data type.
1248 : : * Also check that input is not a set.
1249 : : *
1250 : : * Returns the possibly-transformed node tree.
1251 : : *
1252 : : * As with coerce_type, pstate may be NULL if no special unknown-Param
1253 : : * processing is wanted.
1254 : : */
1255 : : Node *
1256 : 1374 : coerce_to_specific_type(ParseState *pstate, Node *node,
1257 : : Oid targetTypeId,
1258 : : const char *constructName)
1259 : : {
1260 : 2748 : return coerce_to_specific_type_typmod(pstate, node,
1261 : 1374 : targetTypeId, -1,
1262 : 1374 : constructName);
1263 : : }
1264 : :
1265 : : /*
1266 : : * coerce_null_to_domain()
1267 : : * Build a NULL constant, then wrap it in CoerceToDomain
1268 : : * if the desired type is a domain type. This allows any
1269 : : * NOT NULL domain constraint to be enforced at runtime.
1270 : : */
1271 : : Node *
1272 : 1894 : coerce_null_to_domain(Oid typid, int32 typmod, Oid collation,
1273 : : int typlen, bool typbyval)
1274 : : {
1275 : 1894 : Node *result;
1276 : 1894 : Oid baseTypeId;
1277 : 1894 : int32 baseTypeMod = typmod;
1278 : :
1279 : : /*
1280 : : * The constant must appear to have the domain's base type/typmod, else
1281 : : * coerce_to_domain() will apply a length coercion which is useless.
1282 : : */
1283 : 1894 : baseTypeId = getBaseTypeAndTypmod(typid, &baseTypeMod);
1284 : 3788 : result = (Node *) makeConst(baseTypeId,
1285 : 1894 : baseTypeMod,
1286 : 1894 : collation,
1287 : 1894 : typlen,
1288 : : (Datum) 0,
1289 : : true, /* isnull */
1290 : 1894 : typbyval);
1291 [ + + ]: 1894 : if (typid != baseTypeId)
1292 : 22 : result = coerce_to_domain(result,
1293 : 11 : baseTypeId, baseTypeMod,
1294 : 11 : typid,
1295 : : COERCION_IMPLICIT,
1296 : : COERCE_IMPLICIT_CAST,
1297 : : -1,
1298 : : false);
1299 : 3788 : return result;
1300 : 1894 : }
1301 : :
1302 : : /*
1303 : : * parser_coercion_errposition - report coercion error location, if possible
1304 : : *
1305 : : * We prefer to point at the coercion request (CAST, ::, etc) if possible;
1306 : : * but there may be no such location in the case of an implicit coercion.
1307 : : * In that case point at the input expression.
1308 : : *
1309 : : * XXX possibly this is more generally useful than coercion errors;
1310 : : * if so, should rename and place with parser_errposition.
1311 : : */
1312 : : int
1313 : 4 : parser_coercion_errposition(ParseState *pstate,
1314 : : int coerce_location,
1315 : : Node *input_expr)
1316 : : {
1317 [ + - ]: 4 : if (coerce_location >= 0)
1318 : 4 : return parser_errposition(pstate, coerce_location);
1319 : : else
1320 : 0 : return parser_errposition(pstate, exprLocation(input_expr));
1321 : 4 : }
1322 : :
1323 : :
1324 : : /*
1325 : : * select_common_type()
1326 : : * Determine the common supertype of a list of input expressions.
1327 : : * This is used for determining the output type of CASE, UNION,
1328 : : * and similar constructs.
1329 : : *
1330 : : * 'exprs' is a *nonempty* list of expressions. Note that earlier items
1331 : : * in the list will be preferred if there is doubt.
1332 : : * 'context' is a phrase to use in the error message if we fail to select
1333 : : * a usable type. Pass NULL to have the routine return InvalidOid
1334 : : * rather than throwing an error on failure.
1335 : : * 'which_expr': if not NULL, receives a pointer to the particular input
1336 : : * expression from which the result type was taken.
1337 : : *
1338 : : * Caution: "failure" just means that there were inputs of different type
1339 : : * categories. It is not guaranteed that all the inputs are coercible to the
1340 : : * selected type; caller must check that (see verify_common_type).
1341 : : */
1342 : : Oid
1343 : 12669 : select_common_type(ParseState *pstate, List *exprs, const char *context,
1344 : : Node **which_expr)
1345 : : {
1346 : 12669 : Node *pexpr;
1347 : 12669 : Oid ptype;
1348 : 12669 : TYPCATEGORY pcategory;
1349 : 12669 : bool pispreferred;
1350 : 12669 : ListCell *lc;
1351 : :
1352 [ + - ]: 12669 : Assert(exprs != NIL);
1353 : 12669 : pexpr = (Node *) linitial(exprs);
1354 : 12669 : lc = list_second_cell(exprs);
1355 : 12669 : ptype = exprType(pexpr);
1356 : :
1357 : : /*
1358 : : * If all input types are valid and exactly the same, just pick that type.
1359 : : * This is the only way that we will resolve the result as being a domain
1360 : : * type; otherwise domains are smashed to their base types for comparison.
1361 : : */
1362 [ + + ]: 12669 : if (ptype != UNKNOWNOID)
1363 : : {
1364 [ + - + + : 20334 : for_each_cell(lc, exprs, lc)
+ + ]
1365 : : {
1366 : 10888 : Node *nexpr = (Node *) lfirst(lc);
1367 : 10888 : Oid ntype = exprType(nexpr);
1368 : :
1369 [ + + ]: 10888 : if (ntype != ptype)
1370 : 3705 : break;
1371 [ + + ]: 10888 : }
1372 [ + + ]: 9446 : if (lc == NULL) /* got to the end of the list? */
1373 : : {
1374 [ + + ]: 5741 : if (which_expr)
1375 : 3000 : *which_expr = pexpr;
1376 : 5741 : return ptype;
1377 : : }
1378 : 3705 : }
1379 : :
1380 : : /*
1381 : : * Nope, so set up for the full algorithm. Note that at this point, lc
1382 : : * points to the first list item with type different from pexpr's; we need
1383 : : * not re-examine any items the previous loop advanced over.
1384 : : */
1385 : 6928 : ptype = getBaseType(ptype);
1386 : 6928 : get_type_category_preferred(ptype, &pcategory, &pispreferred);
1387 : :
1388 [ + - + + : 18726 : for_each_cell(lc, exprs, lc)
+ + - + ]
1389 : : {
1390 : 11800 : Node *nexpr = (Node *) lfirst(lc);
1391 : 11800 : Oid ntype = getBaseType(exprType(nexpr));
1392 : :
1393 : : /* move on to next one if no new information... */
1394 [ + + + + ]: 11800 : if (ntype != UNKNOWNOID && ntype != ptype)
1395 : : {
1396 : 2339 : TYPCATEGORY ncategory;
1397 : 2339 : bool nispreferred;
1398 : :
1399 : 2339 : get_type_category_preferred(ntype, &ncategory, &nispreferred);
1400 [ + + ]: 2339 : if (ptype == UNKNOWNOID)
1401 : : {
1402 : : /* so far, only unknowns so take anything... */
1403 : 1847 : pexpr = nexpr;
1404 : 1847 : ptype = ntype;
1405 : 1847 : pcategory = ncategory;
1406 : 1847 : pispreferred = nispreferred;
1407 : 1847 : }
1408 [ + + ]: 492 : else if (ncategory != pcategory)
1409 : : {
1410 : : /*
1411 : : * both types in different categories? then not much hope...
1412 : : */
1413 [ - + ]: 2 : if (context == NULL)
1414 : 0 : return InvalidOid;
1415 [ + - + - ]: 2 : ereport(ERROR,
1416 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
1417 : : /*------
1418 : : translator: first %s is name of a SQL construct, eg CASE */
1419 : : errmsg("%s types %s and %s cannot be matched",
1420 : : context,
1421 : : format_type_be(ptype),
1422 : : format_type_be(ntype)),
1423 : : parser_errposition(pstate, exprLocation(nexpr))));
1424 : 0 : }
1425 [ + + ]: 490 : else if (!pispreferred &&
1426 [ + + + + ]: 185 : can_coerce_type(1, &ptype, &ntype, COERCION_IMPLICIT) &&
1427 : 53 : !can_coerce_type(1, &ntype, &ptype, COERCION_IMPLICIT))
1428 : : {
1429 : : /*
1430 : : * take new type if can coerce to it implicitly but not the
1431 : : * other way; but if we have a preferred type, stay on it.
1432 : : */
1433 : 41 : pexpr = nexpr;
1434 : 41 : ptype = ntype;
1435 : 41 : pcategory = ncategory;
1436 : 41 : pispreferred = nispreferred;
1437 : 41 : }
1438 [ - + ]: 2337 : }
1439 [ - + ]: 11798 : }
1440 : :
1441 : : /*
1442 : : * If all the inputs were UNKNOWN type --- ie, unknown-type literals ---
1443 : : * then resolve as type TEXT. This situation comes up with constructs
1444 : : * like SELECT (CASE WHEN foo THEN 'bar' ELSE 'baz' END); SELECT 'foo'
1445 : : * UNION SELECT 'bar'; It might seem desirable to leave the construct's
1446 : : * output type as UNKNOWN, but that really doesn't work, because we'd
1447 : : * probably end up needing a runtime coercion from UNKNOWN to something
1448 : : * else, and we usually won't have it. We need to coerce the unknown
1449 : : * literals while they are still literals, so a decision has to be made
1450 : : * now.
1451 : : */
1452 [ + + ]: 6926 : if (ptype == UNKNOWNOID)
1453 : 1376 : ptype = TEXTOID;
1454 : :
1455 [ + + ]: 6926 : if (which_expr)
1456 : 2112 : *which_expr = pexpr;
1457 : 6926 : return ptype;
1458 : 12667 : }
1459 : :
1460 : : /*
1461 : : * select_common_type_from_oids()
1462 : : * Determine the common supertype of an array of type OIDs.
1463 : : *
1464 : : * This is the same logic as select_common_type(), but working from
1465 : : * an array of type OIDs not a list of expressions. As in that function,
1466 : : * earlier entries in the array have some preference over later ones.
1467 : : * On failure, return InvalidOid if noerror is true, else throw an error.
1468 : : *
1469 : : * Caution: "failure" just means that there were inputs of different type
1470 : : * categories. It is not guaranteed that all the inputs are coercible to the
1471 : : * selected type; caller must check that (see verify_common_type_from_oids).
1472 : : *
1473 : : * Note: neither caller will pass any UNKNOWNOID entries, so the tests
1474 : : * for that in this function are dead code. However, they don't cost much,
1475 : : * and it seems better to keep this logic as close to select_common_type()
1476 : : * as possible.
1477 : : */
1478 : : static Oid
1479 : 951 : select_common_type_from_oids(int nargs, const Oid *typeids, bool noerror)
1480 : : {
1481 : 951 : Oid ptype;
1482 : 951 : TYPCATEGORY pcategory;
1483 : 951 : bool pispreferred;
1484 : 951 : int i = 1;
1485 : :
1486 [ + - ]: 951 : Assert(nargs > 0);
1487 : 951 : ptype = typeids[0];
1488 : :
1489 : : /* If all input types are valid and exactly the same, pick that type. */
1490 [ - + ]: 951 : if (ptype != UNKNOWNOID)
1491 : : {
1492 [ + + ]: 1500 : for (; i < nargs; i++)
1493 : : {
1494 [ + + ]: 734 : if (typeids[i] != ptype)
1495 : 185 : break;
1496 : 549 : }
1497 [ + + ]: 951 : if (i == nargs)
1498 : 766 : return ptype;
1499 : 185 : }
1500 : :
1501 : : /*
1502 : : * Nope, so set up for the full algorithm. Note that at this point, we
1503 : : * can skip array entries before "i"; they are all equal to ptype.
1504 : : */
1505 : 185 : ptype = getBaseType(ptype);
1506 : 185 : get_type_category_preferred(ptype, &pcategory, &pispreferred);
1507 : :
1508 [ + + ]: 269 : for (; i < nargs; i++)
1509 : : {
1510 : 194 : Oid ntype = getBaseType(typeids[i]);
1511 : :
1512 : : /* move on to next one if no new information... */
1513 [ + - + + ]: 194 : if (ntype != UNKNOWNOID && ntype != ptype)
1514 : : {
1515 : 192 : TYPCATEGORY ncategory;
1516 : 192 : bool nispreferred;
1517 : :
1518 : 192 : get_type_category_preferred(ntype, &ncategory, &nispreferred);
1519 [ - + ]: 192 : if (ptype == UNKNOWNOID)
1520 : : {
1521 : : /* so far, only unknowns so take anything... */
1522 : 0 : ptype = ntype;
1523 : 0 : pcategory = ncategory;
1524 : 0 : pispreferred = nispreferred;
1525 : 0 : }
1526 [ + + ]: 192 : else if (ncategory != pcategory)
1527 : : {
1528 : : /*
1529 : : * both types in different categories? then not much hope...
1530 : : */
1531 [ + - ]: 110 : if (noerror)
1532 : 110 : return InvalidOid;
1533 [ # # # # ]: 0 : ereport(ERROR,
1534 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
1535 : : errmsg("argument types %s and %s cannot be matched",
1536 : : format_type_be(ptype),
1537 : : format_type_be(ntype))));
1538 : 0 : }
1539 [ + + ]: 82 : else if (!pispreferred &&
1540 [ + + - + ]: 70 : can_coerce_type(1, &ptype, &ntype, COERCION_IMPLICIT) &&
1541 : 45 : !can_coerce_type(1, &ntype, &ptype, COERCION_IMPLICIT))
1542 : : {
1543 : : /*
1544 : : * take new type if can coerce to it implicitly but not the
1545 : : * other way; but if we have a preferred type, stay on it.
1546 : : */
1547 : 45 : ptype = ntype;
1548 : 45 : pcategory = ncategory;
1549 : 45 : pispreferred = nispreferred;
1550 : 45 : }
1551 [ + + ]: 192 : }
1552 [ + + ]: 194 : }
1553 : :
1554 : : /* Like select_common_type(), choose TEXT if all inputs were UNKNOWN */
1555 [ + - ]: 75 : if (ptype == UNKNOWNOID)
1556 : 0 : ptype = TEXTOID;
1557 : :
1558 : 75 : return ptype;
1559 : 951 : }
1560 : :
1561 : : /*
1562 : : * coerce_to_common_type()
1563 : : * Coerce an expression to the given type.
1564 : : *
1565 : : * This is used following select_common_type() to coerce the individual
1566 : : * expressions to the desired type. 'context' is a phrase to use in the
1567 : : * error message if we fail to coerce.
1568 : : *
1569 : : * As with coerce_type, pstate may be NULL if no special unknown-Param
1570 : : * processing is wanted.
1571 : : */
1572 : : Node *
1573 : 30146 : coerce_to_common_type(ParseState *pstate, Node *node,
1574 : : Oid targetTypeId, const char *context)
1575 : : {
1576 : 30146 : Oid inputTypeId = exprType(node);
1577 : :
1578 [ + + ]: 30146 : if (inputTypeId == targetTypeId)
1579 : 17370 : return node; /* no work */
1580 [ + - ]: 12776 : if (can_coerce_type(1, &inputTypeId, &targetTypeId, COERCION_IMPLICIT))
1581 : 12776 : node = coerce_type(pstate, node, inputTypeId, targetTypeId, -1,
1582 : : COERCION_IMPLICIT, COERCE_IMPLICIT_CAST, -1);
1583 : : else
1584 [ # # # # ]: 0 : ereport(ERROR,
1585 : : (errcode(ERRCODE_CANNOT_COERCE),
1586 : : /* translator: first %s is name of a SQL construct, eg CASE */
1587 : : errmsg("%s could not convert type %s to %s",
1588 : : context,
1589 : : format_type_be(inputTypeId),
1590 : : format_type_be(targetTypeId)),
1591 : : parser_errposition(pstate, exprLocation(node))));
1592 : 12776 : return node;
1593 : 30146 : }
1594 : :
1595 : : /*
1596 : : * verify_common_type()
1597 : : * Verify that all input types can be coerced to a proposed common type.
1598 : : * Return true if so, false if not all coercions are possible.
1599 : : *
1600 : : * Most callers of select_common_type() don't need to do this explicitly
1601 : : * because the checks will happen while trying to convert input expressions
1602 : : * to the right type, e.g. in coerce_to_common_type(). However, if a separate
1603 : : * check step is needed to validate the applicability of the common type, call
1604 : : * this.
1605 : : */
1606 : : bool
1607 : 914 : verify_common_type(Oid common_type, List *exprs)
1608 : : {
1609 : 914 : ListCell *lc;
1610 : :
1611 [ + - + + : 4757 : foreach(lc, exprs)
+ + + + ]
1612 : : {
1613 : 3843 : Node *nexpr = (Node *) lfirst(lc);
1614 : 3843 : Oid ntype = exprType(nexpr);
1615 : :
1616 [ + + ]: 3843 : if (!can_coerce_type(1, &ntype, &common_type, COERCION_IMPLICIT))
1617 : 1 : return false;
1618 [ + + ]: 3843 : }
1619 : 913 : return true;
1620 : 914 : }
1621 : :
1622 : : /*
1623 : : * verify_common_type_from_oids()
1624 : : * As above, but work from an array of type OIDs.
1625 : : */
1626 : : static bool
1627 : 841 : verify_common_type_from_oids(Oid common_type, int nargs, const Oid *typeids)
1628 : : {
1629 [ + + - + : 2315 : for (int i = 0; i < nargs; i++)
+ ]
1630 : : {
1631 [ + + ]: 1474 : if (!can_coerce_type(1, &typeids[i], &common_type, COERCION_IMPLICIT))
1632 : 2 : return false;
1633 : 1472 : }
1634 : 839 : return true;
1635 : 841 : }
1636 : :
1637 : : /*
1638 : : * select_common_typmod()
1639 : : * Determine the common typmod of a list of input expressions.
1640 : : *
1641 : : * common_type is the selected common type of the expressions, typically
1642 : : * computed using select_common_type().
1643 : : */
1644 : : int32
1645 : 7124 : select_common_typmod(ParseState *pstate, List *exprs, Oid common_type)
1646 : : {
1647 : 7124 : ListCell *lc;
1648 : 7124 : bool first = true;
1649 : 7124 : int32 result = -1;
1650 : :
1651 [ + - + + : 22979 : foreach(lc, exprs)
+ + + + ]
1652 : : {
1653 : 15855 : Node *expr = (Node *) lfirst(lc);
1654 : :
1655 : : /* Types must match */
1656 [ + + ]: 15855 : if (exprType(expr) != common_type)
1657 : 30 : return -1;
1658 [ + + ]: 15825 : else if (first)
1659 : : {
1660 : 7109 : result = exprTypmod(expr);
1661 : 7109 : first = false;
1662 : 7109 : }
1663 : : else
1664 : : {
1665 : : /* As soon as we see a non-matching typmod, fall back to -1 */
1666 [ + + ]: 8716 : if (result != exprTypmod(expr))
1667 : 6 : return -1;
1668 : : }
1669 [ + + ]: 15855 : }
1670 : :
1671 : 7088 : return result;
1672 : 7124 : }
1673 : :
1674 : : /*
1675 : : * check_generic_type_consistency()
1676 : : * Are the actual arguments potentially compatible with a
1677 : : * polymorphic function?
1678 : : *
1679 : : * The argument consistency rules are:
1680 : : *
1681 : : * 1) All arguments declared ANYELEMENT must have the same datatype.
1682 : : * 2) All arguments declared ANYARRAY must have the same datatype,
1683 : : * which must be a varlena array type.
1684 : : * 3) All arguments declared ANYRANGE must be the same range type.
1685 : : * Similarly, all arguments declared ANYMULTIRANGE must be the same
1686 : : * multirange type; and if both of these appear, the ANYRANGE type
1687 : : * must be the element type of the ANYMULTIRANGE type.
1688 : : * 4) If there are arguments of more than one of these polymorphic types,
1689 : : * the array element type and/or range subtype must be the same as each
1690 : : * other and the same as the ANYELEMENT type.
1691 : : * 5) ANYENUM is treated the same as ANYELEMENT except that if it is used
1692 : : * (alone or in combination with plain ANYELEMENT), we add the extra
1693 : : * condition that the ANYELEMENT type must be an enum.
1694 : : * 6) ANYNONARRAY is treated the same as ANYELEMENT except that if it is used,
1695 : : * we add the extra condition that the ANYELEMENT type must not be an array.
1696 : : * (This is a no-op if used in combination with ANYARRAY or ANYENUM, but
1697 : : * is an extra restriction if not.)
1698 : : * 7) All arguments declared ANYCOMPATIBLE must be implicitly castable
1699 : : * to a common supertype (chosen as per select_common_type's rules).
1700 : : * ANYCOMPATIBLENONARRAY works like ANYCOMPATIBLE but also requires the
1701 : : * common supertype to not be an array. If there are ANYCOMPATIBLEARRAY
1702 : : * or ANYCOMPATIBLERANGE or ANYCOMPATIBLEMULTIRANGE arguments, their element
1703 : : * types or subtypes are included while making the choice of common supertype.
1704 : : * 8) The resolved type of ANYCOMPATIBLEARRAY arguments will be the array
1705 : : * type over the common supertype (which might not be the same array type
1706 : : * as any of the original arrays).
1707 : : * 9) All ANYCOMPATIBLERANGE arguments must be the exact same range type
1708 : : * (after domain flattening), since we have no preference rule that would
1709 : : * let us choose one over another. Furthermore, that range's subtype
1710 : : * must exactly match the common supertype chosen by rule 7.
1711 : : * 10) All ANYCOMPATIBLEMULTIRANGE arguments must be the exact same multirange
1712 : : * type (after domain flattening), since we have no preference rule that
1713 : : * would let us choose one over another. Furthermore, if ANYCOMPATIBLERANGE
1714 : : * also appears, that range type must be the multirange's element type;
1715 : : * otherwise, the multirange's range's subtype must exactly match the
1716 : : * common supertype chosen by rule 7.
1717 : : *
1718 : : * Domains over arrays match ANYARRAY, and are immediately flattened to their
1719 : : * base type. (Thus, for example, we will consider it a match if one ANYARRAY
1720 : : * argument is a domain over int4[] while another one is just int4[].) Also
1721 : : * notice that such a domain does *not* match ANYNONARRAY. The same goes
1722 : : * for ANYCOMPATIBLEARRAY and ANYCOMPATIBLENONARRAY.
1723 : : *
1724 : : * Similarly, domains over ranges match ANYRANGE or ANYCOMPATIBLERANGE,
1725 : : * and are immediately flattened to their base type. Likewise, domains
1726 : : * over multiranges match ANYMULTIRANGE or ANYCOMPATIBLEMULTIRANGE and are
1727 : : * immediately flattened to their base type.
1728 : : *
1729 : : * Note that domains aren't currently considered to match ANYENUM,
1730 : : * even if their base type would match.
1731 : : *
1732 : : * If we have UNKNOWN input (ie, an untyped literal) for any polymorphic
1733 : : * argument, assume it is okay.
1734 : : *
1735 : : * We do not ereport here, but just return false if a rule is violated.
1736 : : */
1737 : : bool
1738 : 13194 : check_generic_type_consistency(const Oid *actual_arg_types,
1739 : : const Oid *declared_arg_types,
1740 : : int nargs)
1741 : : {
1742 : 13194 : Oid elem_typeid = InvalidOid;
1743 : 13194 : Oid array_typeid = InvalidOid;
1744 : 13194 : Oid range_typeid = InvalidOid;
1745 : 13194 : Oid multirange_typeid = InvalidOid;
1746 : 13194 : Oid anycompatible_range_typeid = InvalidOid;
1747 : 13194 : Oid anycompatible_range_typelem = InvalidOid;
1748 : 13194 : Oid anycompatible_multirange_typeid = InvalidOid;
1749 : 13194 : Oid anycompatible_multirange_typelem = InvalidOid;
1750 : 13194 : Oid range_typelem = InvalidOid;
1751 : 13194 : bool have_anynonarray = false;
1752 : 13194 : bool have_anyenum = false;
1753 : 13194 : bool have_anycompatible_nonarray = false;
1754 : 13194 : int n_anycompatible_args = 0;
1755 : 13194 : Oid anycompatible_actual_types[FUNC_MAX_ARGS];
1756 : :
1757 : : /*
1758 : : * Loop through the arguments to see if we have any that are polymorphic.
1759 : : * If so, require the actual types to be consistent.
1760 : : */
1761 [ + - ]: 13194 : Assert(nargs <= FUNC_MAX_ARGS);
1762 [ + + + + ]: 33645 : for (int j = 0; j < nargs; j++)
1763 : : {
1764 : 20451 : Oid decl_type = declared_arg_types[j];
1765 : 20451 : Oid actual_type = actual_arg_types[j];
1766 : :
1767 [ + + ]: 20451 : if (decl_type == ANYELEMENTOID ||
1768 [ + + + + ]: 19457 : decl_type == ANYNONARRAYOID ||
1769 : 17522 : decl_type == ANYENUMOID)
1770 : : {
1771 [ + + ]: 4520 : if (decl_type == ANYNONARRAYOID)
1772 : 1935 : have_anynonarray = true;
1773 [ + + ]: 2585 : else if (decl_type == ANYENUMOID)
1774 : 1591 : have_anyenum = true;
1775 [ + + ]: 4520 : if (actual_type == UNKNOWNOID)
1776 : 266 : continue;
1777 [ + + + + ]: 4254 : if (OidIsValid(elem_typeid) && actual_type != elem_typeid)
1778 : 494 : return false;
1779 : 3760 : elem_typeid = actual_type;
1780 : 3760 : }
1781 [ + + ]: 15931 : else if (decl_type == ANYARRAYOID)
1782 : : {
1783 [ + + ]: 6038 : if (actual_type == UNKNOWNOID)
1784 : 223 : continue;
1785 : 5815 : actual_type = getBaseType(actual_type); /* flatten domains */
1786 [ + + + + ]: 5815 : if (OidIsValid(array_typeid) && actual_type != array_typeid)
1787 : 527 : return false;
1788 : 5288 : array_typeid = actual_type;
1789 : 5288 : }
1790 [ + + ]: 9893 : else if (decl_type == ANYRANGEOID)
1791 : : {
1792 [ + + ]: 3029 : if (actual_type == UNKNOWNOID)
1793 : 257 : continue;
1794 : 2772 : actual_type = getBaseType(actual_type); /* flatten domains */
1795 [ + + + + ]: 2772 : if (OidIsValid(range_typeid) && actual_type != range_typeid)
1796 : 662 : return false;
1797 : 2110 : range_typeid = actual_type;
1798 : 2110 : }
1799 [ + + ]: 6864 : else if (decl_type == ANYMULTIRANGEOID)
1800 : : {
1801 [ + + ]: 3325 : if (actual_type == UNKNOWNOID)
1802 : 260 : continue;
1803 : 3065 : actual_type = getBaseType(actual_type); /* flatten domains */
1804 [ + + + + ]: 3065 : if (OidIsValid(multirange_typeid) && actual_type != multirange_typeid)
1805 : 661 : return false;
1806 : 2404 : multirange_typeid = actual_type;
1807 : 2404 : }
1808 [ + + + + ]: 3539 : else if (decl_type == ANYCOMPATIBLEOID ||
1809 : 3062 : decl_type == ANYCOMPATIBLENONARRAYOID)
1810 : : {
1811 [ + + ]: 483 : if (decl_type == ANYCOMPATIBLENONARRAYOID)
1812 : 6 : have_anycompatible_nonarray = true;
1813 [ + + ]: 483 : if (actual_type == UNKNOWNOID)
1814 : 114 : continue;
1815 : : /* collect the actual types of non-unknown COMPATIBLE args */
1816 : 369 : anycompatible_actual_types[n_anycompatible_args++] = actual_type;
1817 : 369 : }
1818 [ + + ]: 3056 : else if (decl_type == ANYCOMPATIBLEARRAYOID)
1819 : : {
1820 : 715 : Oid elem_type;
1821 : :
1822 [ + + ]: 715 : if (actual_type == UNKNOWNOID)
1823 : 116 : continue;
1824 : 599 : actual_type = getBaseType(actual_type); /* flatten domains */
1825 : 599 : elem_type = get_element_type(actual_type);
1826 [ + + ]: 599 : if (!OidIsValid(elem_type))
1827 : 224 : return false; /* not an array */
1828 : : /* collect the element type for common-supertype choice */
1829 : 375 : anycompatible_actual_types[n_anycompatible_args++] = elem_type;
1830 [ + + ]: 715 : }
1831 [ + + ]: 2341 : else if (decl_type == ANYCOMPATIBLERANGEOID)
1832 : : {
1833 [ + + ]: 31 : if (actual_type == UNKNOWNOID)
1834 : 2 : continue;
1835 : 29 : actual_type = getBaseType(actual_type); /* flatten domains */
1836 [ + + ]: 29 : if (OidIsValid(anycompatible_range_typeid))
1837 : : {
1838 : : /* All ANYCOMPATIBLERANGE arguments must be the same type */
1839 [ + + ]: 2 : if (anycompatible_range_typeid != actual_type)
1840 : 1 : return false;
1841 : 1 : }
1842 : : else
1843 : : {
1844 : 27 : anycompatible_range_typeid = actual_type;
1845 : 27 : anycompatible_range_typelem = get_range_subtype(actual_type);
1846 [ + + ]: 27 : if (!OidIsValid(anycompatible_range_typelem))
1847 : 1 : return false; /* not a range type */
1848 : : /* collect the subtype for common-supertype choice */
1849 : 26 : anycompatible_actual_types[n_anycompatible_args++] = anycompatible_range_typelem;
1850 : : }
1851 : 27 : }
1852 [ + + ]: 2310 : else if (decl_type == ANYCOMPATIBLEMULTIRANGEOID)
1853 : : {
1854 [ + + ]: 23 : if (actual_type == UNKNOWNOID)
1855 : 3 : continue;
1856 : 20 : actual_type = getBaseType(actual_type); /* flatten domains */
1857 [ + + ]: 20 : if (OidIsValid(anycompatible_multirange_typeid))
1858 : : {
1859 : : /* All ANYCOMPATIBLEMULTIRANGE arguments must be the same type */
1860 [ + + ]: 2 : if (anycompatible_multirange_typeid != actual_type)
1861 : 1 : return false;
1862 : 1 : }
1863 : : else
1864 : : {
1865 : 18 : anycompatible_multirange_typeid = actual_type;
1866 : 18 : anycompatible_multirange_typelem = get_multirange_range(actual_type);
1867 [ + + ]: 18 : if (!OidIsValid(anycompatible_multirange_typelem))
1868 : 1 : return false; /* not a multirange type */
1869 : : /* we'll consider the subtype below */
1870 : : }
1871 : 18 : }
1872 [ + + + ]: 20451 : }
1873 : :
1874 : : /* Get the element type based on the array type, if we have one */
1875 [ + + ]: 10622 : if (OidIsValid(array_typeid))
1876 : : {
1877 [ - + ]: 4616 : if (array_typeid == ANYARRAYOID)
1878 : : {
1879 : : /*
1880 : : * Special case for matching ANYARRAY input to an ANYARRAY
1881 : : * argument: allow it for now. enforce_generic_type_consistency()
1882 : : * might complain later, depending on the presence of other
1883 : : * polymorphic arguments or results, but it will deliver a less
1884 : : * surprising error message than "function does not exist".
1885 : : *
1886 : : * (If you think to change this, note that can_coerce_type will
1887 : : * consider such a situation as a match, so that we might not even
1888 : : * get here.)
1889 : : */
1890 : 0 : }
1891 : : else
1892 : : {
1893 : 4616 : Oid array_typelem;
1894 : :
1895 : 4616 : array_typelem = get_element_type(array_typeid);
1896 [ + + ]: 4616 : if (!OidIsValid(array_typelem))
1897 : 1995 : return false; /* should be an array, but isn't */
1898 : :
1899 [ + + ]: 2621 : if (!OidIsValid(elem_typeid))
1900 : : {
1901 : : /*
1902 : : * if we don't have an element type yet, use the one we just
1903 : : * got
1904 : : */
1905 : 2601 : elem_typeid = array_typelem;
1906 : 2601 : }
1907 [ + + ]: 20 : else if (array_typelem != elem_typeid)
1908 : : {
1909 : : /* otherwise, they better match */
1910 : 8 : return false;
1911 : : }
1912 [ + + ]: 4616 : }
1913 : 2613 : }
1914 : :
1915 : : /* Deduce range type from multirange type, or check that they agree */
1916 [ + + ]: 8619 : if (OidIsValid(multirange_typeid))
1917 : : {
1918 : 1529 : Oid multirange_typelem;
1919 : :
1920 : 1529 : multirange_typelem = get_multirange_range(multirange_typeid);
1921 [ + + ]: 1529 : if (!OidIsValid(multirange_typelem))
1922 : 1167 : return false; /* should be a multirange, but isn't */
1923 : :
1924 [ + + ]: 362 : if (!OidIsValid(range_typeid))
1925 : : {
1926 : : /* If we don't have a range type yet, use the one we just got */
1927 : 274 : range_typeid = multirange_typelem;
1928 : 274 : range_typelem = get_range_subtype(multirange_typelem);
1929 [ - + ]: 274 : if (!OidIsValid(range_typelem))
1930 : 0 : return false; /* should be a range, but isn't */
1931 : 274 : }
1932 [ + + ]: 88 : else if (multirange_typelem != range_typeid)
1933 : : {
1934 : : /* otherwise, they better match */
1935 : 51 : return false;
1936 : : }
1937 [ + + ]: 1529 : }
1938 : :
1939 : : /* Get the element type based on the range type, if we have one */
1940 [ + + ]: 7401 : if (OidIsValid(range_typeid))
1941 : : {
1942 : 1228 : range_typelem = get_range_subtype(range_typeid);
1943 [ + + ]: 1228 : if (!OidIsValid(range_typelem))
1944 : 447 : return false; /* should be a range, but isn't */
1945 : :
1946 [ + + ]: 781 : if (!OidIsValid(elem_typeid))
1947 : : {
1948 : : /*
1949 : : * If we don't have an element type yet, use the one we just got
1950 : : */
1951 : 714 : elem_typeid = range_typelem;
1952 : 714 : }
1953 [ + + ]: 67 : else if (range_typelem != elem_typeid)
1954 : : {
1955 : : /* otherwise, they better match */
1956 : 29 : return false;
1957 : : }
1958 : 752 : }
1959 : :
1960 [ + + ]: 6925 : if (have_anynonarray)
1961 : : {
1962 : : /* require the element type to not be an array or domain over array */
1963 [ + + ]: 1935 : if (type_is_array_domain(elem_typeid))
1964 : 15 : return false;
1965 : 1920 : }
1966 : :
1967 [ + + ]: 6910 : if (have_anyenum)
1968 : : {
1969 : : /* require the element type to be an enum */
1970 [ + + ]: 337 : if (!type_is_enum(elem_typeid))
1971 : 278 : return false;
1972 : 59 : }
1973 : :
1974 : : /* Deduce range type from multirange type, or check that they agree */
1975 [ + + ]: 6632 : if (OidIsValid(anycompatible_multirange_typeid))
1976 : : {
1977 [ + + ]: 16 : if (OidIsValid(anycompatible_range_typeid))
1978 : : {
1979 [ + + + + ]: 4 : if (anycompatible_multirange_typelem !=
1980 : 2 : anycompatible_range_typeid)
1981 : 1 : return false;
1982 : 1 : }
1983 : : else
1984 : : {
1985 : 14 : anycompatible_range_typeid = anycompatible_multirange_typelem;
1986 : 14 : anycompatible_range_typelem = get_range_subtype(anycompatible_range_typeid);
1987 [ + - ]: 14 : if (!OidIsValid(anycompatible_range_typelem))
1988 : 0 : return false; /* not a range type */
1989 : : /* collect the subtype for common-supertype choice */
1990 : 14 : anycompatible_actual_types[n_anycompatible_args++] =
1991 : 14 : anycompatible_range_typelem;
1992 : : }
1993 : 15 : }
1994 : :
1995 : : /* Check matching of ANYCOMPATIBLE-family arguments, if any */
1996 [ + + ]: 6631 : if (n_anycompatible_args > 0)
1997 : : {
1998 : 428 : Oid anycompatible_typeid;
1999 : :
2000 : 428 : anycompatible_typeid =
2001 : 856 : select_common_type_from_oids(n_anycompatible_args,
2002 : 428 : anycompatible_actual_types,
2003 : : true);
2004 : :
2005 [ + + ]: 428 : if (!OidIsValid(anycompatible_typeid))
2006 : 110 : return false; /* there's definitely no common supertype */
2007 : :
2008 : : /* We have to verify that the selected type actually works */
2009 [ + + + + ]: 636 : if (!verify_common_type_from_oids(anycompatible_typeid,
2010 : 318 : n_anycompatible_args,
2011 : 318 : anycompatible_actual_types))
2012 : 2 : return false;
2013 : :
2014 [ + + ]: 316 : if (have_anycompatible_nonarray)
2015 : : {
2016 : : /*
2017 : : * require the anycompatible type to not be an array or domain
2018 : : * over array
2019 : : */
2020 [ + + ]: 3 : if (type_is_array_domain(anycompatible_typeid))
2021 : 1 : return false;
2022 : 2 : }
2023 : :
2024 : : /*
2025 : : * The anycompatible type must exactly match the range element type,
2026 : : * if we were able to identify one. This checks compatibility for
2027 : : * anycompatiblemultirange too since that also sets
2028 : : * anycompatible_range_typelem above.
2029 : : */
2030 [ + + + + ]: 315 : if (OidIsValid(anycompatible_range_typelem) &&
2031 : 38 : anycompatible_range_typelem != anycompatible_typeid)
2032 : 7 : return false;
2033 [ + + ]: 428 : }
2034 : :
2035 : : /* Looks valid */
2036 : 6511 : return true;
2037 : 13194 : }
2038 : :
2039 : : /*
2040 : : * enforce_generic_type_consistency()
2041 : : * Make sure a polymorphic function is legally callable, and
2042 : : * deduce actual argument and result types.
2043 : : *
2044 : : * If any polymorphic pseudotype is used in a function's arguments or
2045 : : * return type, we make sure the actual data types are consistent with
2046 : : * each other. The argument consistency rules are shown above for
2047 : : * check_generic_type_consistency().
2048 : : *
2049 : : * If we have UNKNOWN input (ie, an untyped literal) for any polymorphic
2050 : : * argument, we attempt to deduce the actual type it should have. If
2051 : : * successful, we alter that position of declared_arg_types[] so that
2052 : : * make_fn_arguments will coerce the literal to the right thing.
2053 : : *
2054 : : * If we have polymorphic arguments of the ANYCOMPATIBLE family,
2055 : : * we similarly alter declared_arg_types[] entries to show the resolved
2056 : : * common supertype, so that make_fn_arguments will coerce the actual
2057 : : * arguments to the proper type.
2058 : : *
2059 : : * Rules are applied to the function's return type (possibly altering it)
2060 : : * if it is declared as a polymorphic type and there is at least one
2061 : : * polymorphic argument type:
2062 : : *
2063 : : * 1) If return type is ANYELEMENT, and any argument is ANYELEMENT, use the
2064 : : * argument's actual type as the function's return type.
2065 : : * 2) If return type is ANYARRAY, and any argument is ANYARRAY, use the
2066 : : * argument's actual type as the function's return type.
2067 : : * 3) Similarly, if return type is ANYRANGE or ANYMULTIRANGE, and any
2068 : : * argument is ANYRANGE or ANYMULTIRANGE, use that argument's actual type
2069 : : * (or the corresponding range or multirange type) as the function's return
2070 : : * type.
2071 : : * 4) Otherwise, if return type is ANYELEMENT or ANYARRAY, and there is
2072 : : * at least one ANYELEMENT, ANYARRAY, ANYRANGE, or ANYMULTIRANGE input,
2073 : : * deduce the return type from those inputs, or throw error if we can't.
2074 : : * 5) Otherwise, if return type is ANYRANGE or ANYMULTIRANGE, throw error.
2075 : : * (We have no way to select a specific range type if the arguments don't
2076 : : * include ANYRANGE or ANYMULTIRANGE.)
2077 : : * 6) ANYENUM is treated the same as ANYELEMENT except that if it is used
2078 : : * (alone or in combination with plain ANYELEMENT), we add the extra
2079 : : * condition that the ANYELEMENT type must be an enum.
2080 : : * 7) ANYNONARRAY is treated the same as ANYELEMENT except that if it is used,
2081 : : * we add the extra condition that the ANYELEMENT type must not be an array.
2082 : : * (This is a no-op if used in combination with ANYARRAY or ANYENUM, but
2083 : : * is an extra restriction if not.)
2084 : : * 8) ANYCOMPATIBLE, ANYCOMPATIBLEARRAY, and ANYCOMPATIBLENONARRAY are handled
2085 : : * by resolving the common supertype of those arguments (or their element
2086 : : * types, for array inputs), and then coercing all those arguments to the
2087 : : * common supertype, or the array type over the common supertype for
2088 : : * ANYCOMPATIBLEARRAY.
2089 : : * 9) For ANYCOMPATIBLERANGE and ANYCOMPATIBLEMULTIRANGE, there must be at
2090 : : * least one non-UNKNOWN input matching those arguments, and all such
2091 : : * inputs must be the same range type (or its multirange type, as
2092 : : * appropriate), since we cannot deduce a range type from non-range types.
2093 : : * Furthermore, the range type's subtype is included while choosing the
2094 : : * common supertype for ANYCOMPATIBLE et al, and it must exactly match
2095 : : * that common supertype.
2096 : : *
2097 : : * Domains over arrays or ranges match ANYARRAY or ANYRANGE arguments,
2098 : : * respectively, and are immediately flattened to their base type. (In
2099 : : * particular, if the return type is also ANYARRAY or ANYRANGE, we'll set
2100 : : * it to the base type not the domain type.) The same is true for
2101 : : * ANYMULTIRANGE, ANYCOMPATIBLEARRAY, ANYCOMPATIBLERANGE, and
2102 : : * ANYCOMPATIBLEMULTIRANGE.
2103 : : *
2104 : : * When allow_poly is false, we are not expecting any of the actual_arg_types
2105 : : * to be polymorphic, and we should not return a polymorphic result type
2106 : : * either. When allow_poly is true, it is okay to have polymorphic "actual"
2107 : : * arg types, and we can return a matching polymorphic type as the result.
2108 : : * (This case is currently used only to check compatibility of an aggregate's
2109 : : * declaration with the underlying transfn.)
2110 : : *
2111 : : * A special case is that we could see ANYARRAY as an actual_arg_type even
2112 : : * when allow_poly is false (this is possible only because pg_statistic has
2113 : : * columns shown as anyarray in the catalogs). We allow this to match a
2114 : : * declared ANYARRAY argument, but only if there is no other polymorphic
2115 : : * argument that we would need to match it with, and no need to determine
2116 : : * the element type to infer the result type. Note this means that functions
2117 : : * taking ANYARRAY had better behave sanely if applied to the pg_statistic
2118 : : * columns; they can't just assume that successive inputs are of the same
2119 : : * actual element type. There is no similar logic for ANYCOMPATIBLEARRAY;
2120 : : * there isn't a need for it since there are no catalog columns of that type,
2121 : : * so we won't see it as input. We could consider matching an actual ANYARRAY
2122 : : * input to an ANYCOMPATIBLEARRAY argument, but at present that seems useless
2123 : : * as well, since there's no value in using ANYCOMPATIBLEARRAY unless there's
2124 : : * at least one other ANYCOMPATIBLE-family argument or result.
2125 : : *
2126 : : * Also, if there are no arguments declared to be of polymorphic types,
2127 : : * we'll return the rettype unmodified even if it's polymorphic. This should
2128 : : * never occur for user-declared functions, because CREATE FUNCTION prevents
2129 : : * it. But it does happen for some built-in functions, such as array_in().
2130 : : */
2131 : : Oid
2132 : 91716 : enforce_generic_type_consistency(const Oid *actual_arg_types,
2133 : : Oid *declared_arg_types,
2134 : : int nargs,
2135 : : Oid rettype,
2136 : : bool allow_poly)
2137 : : {
2138 : 91716 : bool have_poly_anycompatible = false;
2139 : 91716 : bool have_poly_unknowns = false;
2140 : 91716 : Oid elem_typeid = InvalidOid;
2141 : 91716 : Oid array_typeid = InvalidOid;
2142 : 91716 : Oid range_typeid = InvalidOid;
2143 : 91716 : Oid multirange_typeid = InvalidOid;
2144 : 91716 : Oid anycompatible_typeid = InvalidOid;
2145 : 91716 : Oid anycompatible_array_typeid = InvalidOid;
2146 : 91716 : Oid anycompatible_range_typeid = InvalidOid;
2147 : 91716 : Oid anycompatible_range_typelem = InvalidOid;
2148 : 91716 : Oid anycompatible_multirange_typeid = InvalidOid;
2149 : 91716 : Oid anycompatible_multirange_typelem = InvalidOid;
2150 : 91716 : bool have_anynonarray = (rettype == ANYNONARRAYOID);
2151 : 91716 : bool have_anyenum = (rettype == ANYENUMOID);
2152 : 91716 : bool have_anymultirange = (rettype == ANYMULTIRANGEOID);
2153 : 91716 : bool have_anycompatible_nonarray = (rettype == ANYCOMPATIBLENONARRAYOID);
2154 : 91716 : bool have_anycompatible_array = (rettype == ANYCOMPATIBLEARRAYOID);
2155 : 91716 : bool have_anycompatible_range = (rettype == ANYCOMPATIBLERANGEOID);
2156 : 91716 : bool have_anycompatible_multirange = (rettype == ANYCOMPATIBLEMULTIRANGEOID);
2157 : 91716 : int n_poly_args = 0; /* this counts all family-1 arguments */
2158 : 91716 : int n_anycompatible_args = 0; /* this counts only non-unknowns */
2159 : 91716 : Oid anycompatible_actual_types[FUNC_MAX_ARGS];
2160 : :
2161 : : /*
2162 : : * Loop through the arguments to see if we have any that are polymorphic.
2163 : : * If so, require the actual types to be consistent.
2164 : : */
2165 [ + - ]: 91716 : Assert(nargs <= FUNC_MAX_ARGS);
2166 [ + + ]: 264667 : for (int j = 0; j < nargs; j++)
2167 : : {
2168 : 172951 : Oid decl_type = declared_arg_types[j];
2169 : 172951 : Oid actual_type = actual_arg_types[j];
2170 : :
2171 [ + + ]: 172951 : if (decl_type == ANYELEMENTOID ||
2172 [ + + + + ]: 171883 : decl_type == ANYNONARRAYOID ||
2173 : 169966 : decl_type == ANYENUMOID)
2174 : : {
2175 : 3100 : n_poly_args++;
2176 [ + + ]: 3100 : if (decl_type == ANYNONARRAYOID)
2177 : 1917 : have_anynonarray = true;
2178 [ + + ]: 1183 : else if (decl_type == ANYENUMOID)
2179 : 115 : have_anyenum = true;
2180 [ + + ]: 3100 : if (actual_type == UNKNOWNOID)
2181 : : {
2182 : 70 : have_poly_unknowns = true;
2183 : 70 : continue;
2184 : : }
2185 [ + + + + ]: 3030 : if (allow_poly && decl_type == actual_type)
2186 : 22 : continue; /* no new information here */
2187 [ + + + - ]: 3008 : if (OidIsValid(elem_typeid) && actual_type != elem_typeid)
2188 [ # # # # ]: 0 : ereport(ERROR,
2189 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2190 : : errmsg("arguments declared \"%s\" are not all alike", "anyelement"),
2191 : : errdetail("%s versus %s",
2192 : : format_type_be(elem_typeid),
2193 : : format_type_be(actual_type))));
2194 : 3008 : elem_typeid = actual_type;
2195 : 3008 : }
2196 [ + + ]: 169851 : else if (decl_type == ANYARRAYOID)
2197 : : {
2198 : 3916 : n_poly_args++;
2199 [ + + ]: 3916 : if (actual_type == UNKNOWNOID)
2200 : : {
2201 : 667 : have_poly_unknowns = true;
2202 : 667 : continue;
2203 : : }
2204 [ + + + + ]: 3249 : if (allow_poly && decl_type == actual_type)
2205 : 13 : continue; /* no new information here */
2206 : 3236 : actual_type = getBaseType(actual_type); /* flatten domains */
2207 [ + + + - ]: 3236 : if (OidIsValid(array_typeid) && actual_type != array_typeid)
2208 [ # # # # ]: 0 : ereport(ERROR,
2209 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2210 : : errmsg("arguments declared \"%s\" are not all alike", "anyarray"),
2211 : : errdetail("%s versus %s",
2212 : : format_type_be(array_typeid),
2213 : : format_type_be(actual_type))));
2214 : 3236 : array_typeid = actual_type;
2215 : 3236 : }
2216 [ + + ]: 165935 : else if (decl_type == ANYRANGEOID)
2217 : : {
2218 : 1228 : n_poly_args++;
2219 [ + + ]: 1228 : if (actual_type == UNKNOWNOID)
2220 : : {
2221 : 126 : have_poly_unknowns = true;
2222 : 126 : continue;
2223 : : }
2224 [ - + # # ]: 1102 : if (allow_poly && decl_type == actual_type)
2225 : 0 : continue; /* no new information here */
2226 : 1102 : actual_type = getBaseType(actual_type); /* flatten domains */
2227 [ + + - + ]: 1102 : if (OidIsValid(range_typeid) && actual_type != range_typeid)
2228 [ # # # # ]: 0 : ereport(ERROR,
2229 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2230 : : errmsg("arguments declared \"%s\" are not all alike", "anyrange"),
2231 : : errdetail("%s versus %s",
2232 : : format_type_be(range_typeid),
2233 : : format_type_be(actual_type))));
2234 : 1102 : range_typeid = actual_type;
2235 : 1102 : }
2236 [ + + ]: 164707 : else if (decl_type == ANYMULTIRANGEOID)
2237 : : {
2238 : 858 : n_poly_args++;
2239 : 858 : have_anymultirange = true;
2240 [ + + ]: 858 : if (actual_type == UNKNOWNOID)
2241 : : {
2242 : 44 : have_poly_unknowns = true;
2243 : 44 : continue;
2244 : : }
2245 [ - + # # ]: 814 : if (allow_poly && decl_type == actual_type)
2246 : 0 : continue; /* no new information here */
2247 : 814 : actual_type = getBaseType(actual_type); /* flatten domains */
2248 [ + + + - ]: 814 : if (OidIsValid(multirange_typeid) && actual_type != multirange_typeid)
2249 [ # # # # ]: 0 : ereport(ERROR,
2250 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2251 : : errmsg("arguments declared \"%s\" are not all alike", "anymultirange"),
2252 : : errdetail("%s versus %s",
2253 : : format_type_be(multirange_typeid),
2254 : : format_type_be(actual_type))));
2255 : 814 : multirange_typeid = actual_type;
2256 : 814 : }
2257 [ + + + + ]: 163849 : else if (decl_type == ANYCOMPATIBLEOID ||
2258 : 163629 : decl_type == ANYCOMPATIBLENONARRAYOID)
2259 : : {
2260 : 224 : have_poly_anycompatible = true;
2261 [ + + ]: 224 : if (decl_type == ANYCOMPATIBLENONARRAYOID)
2262 : 4 : have_anycompatible_nonarray = true;
2263 [ + + ]: 224 : if (actual_type == UNKNOWNOID)
2264 : 68 : continue;
2265 [ + + + + ]: 156 : if (allow_poly && decl_type == actual_type)
2266 : 1 : continue; /* no new information here */
2267 : : /* collect the actual types of non-unknown COMPATIBLE args */
2268 : 155 : anycompatible_actual_types[n_anycompatible_args++] = actual_type;
2269 : 155 : }
2270 [ + + ]: 163625 : else if (decl_type == ANYCOMPATIBLEARRAYOID)
2271 : : {
2272 : 786 : Oid anycompatible_elem_type;
2273 : :
2274 : 786 : have_poly_anycompatible = true;
2275 : 786 : have_anycompatible_array = true;
2276 [ + + ]: 786 : if (actual_type == UNKNOWNOID)
2277 : 5 : continue;
2278 [ + + + + ]: 781 : if (allow_poly && decl_type == actual_type)
2279 : 1 : continue; /* no new information here */
2280 : 780 : actual_type = getBaseType(actual_type); /* flatten domains */
2281 : 780 : anycompatible_elem_type = get_element_type(actual_type);
2282 [ + - ]: 780 : if (!OidIsValid(anycompatible_elem_type))
2283 [ # # # # ]: 0 : ereport(ERROR,
2284 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2285 : : errmsg("argument declared %s is not an array but type %s",
2286 : : "anycompatiblearray",
2287 : : format_type_be(actual_type))));
2288 : : /* collect the element type for common-supertype choice */
2289 : 780 : anycompatible_actual_types[n_anycompatible_args++] = anycompatible_elem_type;
2290 [ + + ]: 786 : }
2291 [ + + ]: 162839 : else if (decl_type == ANYCOMPATIBLERANGEOID)
2292 : : {
2293 : 23 : have_poly_anycompatible = true;
2294 : 23 : have_anycompatible_range = true;
2295 [ + + ]: 23 : if (actual_type == UNKNOWNOID)
2296 : 2 : continue;
2297 [ - + # # ]: 21 : if (allow_poly && decl_type == actual_type)
2298 : 0 : continue; /* no new information here */
2299 : 21 : actual_type = getBaseType(actual_type); /* flatten domains */
2300 [ + + ]: 21 : if (OidIsValid(anycompatible_range_typeid))
2301 : : {
2302 : : /* All ANYCOMPATIBLERANGE arguments must be the same type */
2303 [ + - ]: 1 : if (anycompatible_range_typeid != actual_type)
2304 [ # # # # ]: 0 : ereport(ERROR,
2305 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2306 : : errmsg("arguments declared \"%s\" are not all alike", "anycompatiblerange"),
2307 : : errdetail("%s versus %s",
2308 : : format_type_be(anycompatible_range_typeid),
2309 : : format_type_be(actual_type))));
2310 : 1 : }
2311 : : else
2312 : : {
2313 : 20 : anycompatible_range_typeid = actual_type;
2314 : 20 : anycompatible_range_typelem = get_range_subtype(actual_type);
2315 [ + - ]: 20 : if (!OidIsValid(anycompatible_range_typelem))
2316 [ # # # # ]: 0 : ereport(ERROR,
2317 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2318 : : errmsg("argument declared %s is not a range type but type %s",
2319 : : "anycompatiblerange",
2320 : : format_type_be(actual_type))));
2321 : : /* collect the subtype for common-supertype choice */
2322 : 20 : anycompatible_actual_types[n_anycompatible_args++] = anycompatible_range_typelem;
2323 : : }
2324 : 21 : }
2325 [ + + ]: 162816 : else if (decl_type == ANYCOMPATIBLEMULTIRANGEOID)
2326 : : {
2327 : 16 : have_poly_anycompatible = true;
2328 : 16 : have_anycompatible_multirange = true;
2329 [ + + ]: 16 : if (actual_type == UNKNOWNOID)
2330 : 3 : continue;
2331 [ - + # # ]: 13 : if (allow_poly && decl_type == actual_type)
2332 : 0 : continue; /* no new information here */
2333 : 13 : actual_type = getBaseType(actual_type); /* flatten domains */
2334 [ + + ]: 13 : if (OidIsValid(anycompatible_multirange_typeid))
2335 : : {
2336 : : /* All ANYCOMPATIBLEMULTIRANGE arguments must be the same type */
2337 [ - + ]: 1 : if (anycompatible_multirange_typeid != actual_type)
2338 [ # # # # ]: 0 : ereport(ERROR,
2339 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2340 : : errmsg("arguments declared \"%s\" are not all alike", "anycompatiblemultirange"),
2341 : : errdetail("%s versus %s",
2342 : : format_type_be(anycompatible_multirange_typeid),
2343 : : format_type_be(actual_type))));
2344 : 1 : }
2345 : : else
2346 : : {
2347 : 12 : anycompatible_multirange_typeid = actual_type;
2348 : 12 : anycompatible_multirange_typelem = get_multirange_range(actual_type);
2349 [ + - ]: 12 : if (!OidIsValid(anycompatible_multirange_typelem))
2350 [ # # # # ]: 0 : ereport(ERROR,
2351 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2352 : : errmsg("argument declared %s is not a multirange type but type %s",
2353 : : "anycompatiblemultirange",
2354 : : format_type_be(actual_type))));
2355 : : /* we'll consider the subtype below */
2356 : : }
2357 : 13 : }
2358 [ + + ]: 172951 : }
2359 : :
2360 : : /*
2361 : : * Fast Track: if none of the arguments are polymorphic, return the
2362 : : * unmodified rettype. Not our job to resolve it if it's polymorphic.
2363 : : */
2364 [ + + + + ]: 91716 : if (n_poly_args == 0 && !have_poly_anycompatible)
2365 : 83912 : return rettype;
2366 : :
2367 : : /* Check matching of family-1 polymorphic arguments, if any */
2368 [ + + ]: 7804 : if (n_poly_args)
2369 : : {
2370 : : /* Get the element type based on the array type, if we have one */
2371 [ + + ]: 7285 : if (OidIsValid(array_typeid))
2372 : : {
2373 : 3190 : Oid array_typelem;
2374 : :
2375 [ + + ]: 3190 : if (array_typeid == ANYARRAYOID)
2376 : : {
2377 : : /*
2378 : : * Special case for matching ANYARRAY input to an ANYARRAY
2379 : : * argument: allow it iff no other arguments are family-1
2380 : : * polymorphics (otherwise we couldn't be sure whether the
2381 : : * array element type matches up) and the result type doesn't
2382 : : * require us to infer a specific element type.
2383 : : */
2384 [ + + ]: 8 : if (n_poly_args != 1 ||
2385 [ + + ]: 5 : (rettype != ANYARRAYOID &&
2386 : 1 : IsPolymorphicTypeFamily1(rettype)))
2387 [ + - + - ]: 2 : ereport(ERROR,
2388 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2389 : : errmsg("cannot determine element type of \"anyarray\" argument")));
2390 : 5 : array_typelem = ANYELEMENTOID;
2391 : 5 : }
2392 : : else
2393 : : {
2394 : 3183 : array_typelem = get_element_type(array_typeid);
2395 [ + - ]: 3183 : if (!OidIsValid(array_typelem))
2396 [ # # # # ]: 0 : ereport(ERROR,
2397 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2398 : : errmsg("argument declared %s is not an array but type %s",
2399 : : "anyarray", format_type_be(array_typeid))));
2400 : : }
2401 : :
2402 [ + + ]: 3188 : if (!OidIsValid(elem_typeid))
2403 : : {
2404 : : /*
2405 : : * if we don't have an element type yet, use the one we just
2406 : : * got
2407 : : */
2408 : 3176 : elem_typeid = array_typelem;
2409 : 3176 : }
2410 [ + - ]: 12 : else if (array_typelem != elem_typeid)
2411 : : {
2412 : : /* otherwise, they better match */
2413 [ # # # # ]: 0 : ereport(ERROR,
2414 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2415 : : errmsg("argument declared %s is not consistent with argument declared %s",
2416 : : "anyarray", "anyelement"),
2417 : : errdetail("%s versus %s",
2418 : : format_type_be(array_typeid),
2419 : : format_type_be(elem_typeid))));
2420 : 0 : }
2421 : 3188 : }
2422 : :
2423 : : /* Deduce range type from multirange type, or vice versa */
2424 [ + + ]: 7283 : if (OidIsValid(multirange_typeid))
2425 : : {
2426 : 597 : Oid multirange_typelem;
2427 : :
2428 : 597 : multirange_typelem = get_multirange_range(multirange_typeid);
2429 [ + - ]: 597 : if (!OidIsValid(multirange_typelem))
2430 [ # # # # ]: 0 : ereport(ERROR,
2431 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2432 : : errmsg("argument declared %s is not a multirange type but type %s",
2433 : : "anymultirange",
2434 : : format_type_be(multirange_typeid))));
2435 : :
2436 [ + + ]: 597 : if (!OidIsValid(range_typeid))
2437 : : {
2438 : : /* if we don't have a range type yet, use the one we just got */
2439 : 387 : range_typeid = multirange_typelem;
2440 : 387 : }
2441 [ + - ]: 210 : else if (multirange_typelem != range_typeid)
2442 : : {
2443 : : /* otherwise, they better match */
2444 [ # # # # ]: 0 : ereport(ERROR,
2445 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2446 : : errmsg("argument declared %s is not consistent with argument declared %s",
2447 : : "anymultirange", "anyrange"),
2448 : : errdetail("%s versus %s",
2449 : : format_type_be(multirange_typeid),
2450 : : format_type_be(range_typeid))));
2451 : 0 : }
2452 : 597 : }
2453 [ + + + + ]: 6686 : else if (have_anymultirange && OidIsValid(range_typeid))
2454 : : {
2455 : 40 : multirange_typeid = get_range_multirange(range_typeid);
2456 : : /* We'll complain below if that didn't work */
2457 : 40 : }
2458 : :
2459 : : /* Get the element type based on the range type, if we have one */
2460 [ + + ]: 7283 : if (OidIsValid(range_typeid))
2461 : : {
2462 : 1202 : Oid range_typelem;
2463 : :
2464 : 1202 : range_typelem = get_range_subtype(range_typeid);
2465 [ + - ]: 1202 : if (!OidIsValid(range_typelem))
2466 [ # # # # ]: 0 : ereport(ERROR,
2467 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2468 : : errmsg("argument declared %s is not a range type but type %s",
2469 : : "anyrange",
2470 : : format_type_be(range_typeid))));
2471 : :
2472 [ + + ]: 1202 : if (!OidIsValid(elem_typeid))
2473 : : {
2474 : : /*
2475 : : * if we don't have an element type yet, use the one we just
2476 : : * got
2477 : : */
2478 : 1121 : elem_typeid = range_typelem;
2479 : 1121 : }
2480 [ + - ]: 81 : else if (range_typelem != elem_typeid)
2481 : : {
2482 : : /* otherwise, they better match */
2483 [ # # # # ]: 0 : ereport(ERROR,
2484 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2485 : : errmsg("argument declared %s is not consistent with argument declared %s",
2486 : : "anyrange", "anyelement"),
2487 : : errdetail("%s versus %s",
2488 : : format_type_be(range_typeid),
2489 : : format_type_be(elem_typeid))));
2490 : 0 : }
2491 : 1202 : }
2492 : :
2493 [ + + ]: 7283 : if (!OidIsValid(elem_typeid))
2494 : : {
2495 [ + + ]: 29 : if (allow_poly)
2496 : : {
2497 : 25 : elem_typeid = ANYELEMENTOID;
2498 : 25 : array_typeid = ANYARRAYOID;
2499 : 25 : range_typeid = ANYRANGEOID;
2500 : 25 : multirange_typeid = ANYMULTIRANGEOID;
2501 : 25 : }
2502 : : else
2503 : : {
2504 : : /*
2505 : : * Only way to get here is if all the family-1 polymorphic
2506 : : * arguments have UNKNOWN inputs.
2507 : : */
2508 [ + - + - ]: 4 : ereport(ERROR,
2509 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2510 : : errmsg("could not determine polymorphic type because input has type %s",
2511 : : "unknown")));
2512 : : }
2513 : 25 : }
2514 : :
2515 [ + + - + ]: 7279 : if (have_anynonarray && elem_typeid != ANYELEMENTOID)
2516 : : {
2517 : : /*
2518 : : * require the element type to not be an array or domain over
2519 : : * array
2520 : : */
2521 [ + - ]: 1917 : if (type_is_array_domain(elem_typeid))
2522 [ # # # # ]: 0 : ereport(ERROR,
2523 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2524 : : errmsg("type matched to anynonarray is an array type: %s",
2525 : : format_type_be(elem_typeid))));
2526 : 1917 : }
2527 : :
2528 [ + + - + ]: 7279 : if (have_anyenum && elem_typeid != ANYELEMENTOID)
2529 : : {
2530 : : /* require the element type to be an enum */
2531 [ + - ]: 83 : if (!type_is_enum(elem_typeid))
2532 [ # # # # ]: 0 : ereport(ERROR,
2533 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2534 : : errmsg("type matched to anyenum is not an enum type: %s",
2535 : : format_type_be(elem_typeid))));
2536 : 83 : }
2537 : 7279 : }
2538 : :
2539 : : /* Check matching of family-2 polymorphic arguments, if any */
2540 [ + + ]: 7798 : if (have_poly_anycompatible)
2541 : : {
2542 : : /* Deduce range type from multirange type, or vice versa */
2543 [ + + ]: 528 : if (OidIsValid(anycompatible_multirange_typeid))
2544 : : {
2545 [ + + ]: 12 : if (OidIsValid(anycompatible_range_typeid))
2546 : : {
2547 [ + - + - ]: 2 : if (anycompatible_multirange_typelem !=
2548 : 1 : anycompatible_range_typeid)
2549 [ # # # # ]: 0 : ereport(ERROR,
2550 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2551 : : errmsg("argument declared %s is not consistent with argument declared %s",
2552 : : "anycompatiblemultirange",
2553 : : "anycompatiblerange"),
2554 : : errdetail("%s versus %s",
2555 : : format_type_be(anycompatible_multirange_typeid),
2556 : : format_type_be(anycompatible_range_typeid))));
2557 : 1 : }
2558 : : else
2559 : : {
2560 : 11 : anycompatible_range_typeid = anycompatible_multirange_typelem;
2561 : 11 : anycompatible_range_typelem = get_range_subtype(anycompatible_range_typeid);
2562 [ + - ]: 11 : if (!OidIsValid(anycompatible_range_typelem))
2563 [ # # # # ]: 0 : ereport(ERROR,
2564 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2565 : : errmsg("argument declared %s is not a multirange type but type %s",
2566 : : "anycompatiblemultirange",
2567 : : format_type_be(anycompatible_multirange_typeid))));
2568 : : /* this enables element type matching check below */
2569 : 11 : have_anycompatible_range = true;
2570 : : /* collect the subtype for common-supertype choice */
2571 : 11 : anycompatible_actual_types[n_anycompatible_args++] =
2572 : 11 : anycompatible_range_typelem;
2573 : : }
2574 : 12 : }
2575 [ + + + + ]: 516 : else if (have_anycompatible_multirange &&
2576 : 5 : OidIsValid(anycompatible_range_typeid))
2577 : : {
2578 : 1 : anycompatible_multirange_typeid = get_range_multirange(anycompatible_range_typeid);
2579 : : /* We'll complain below if that didn't work */
2580 : 1 : }
2581 : :
2582 [ + + ]: 528 : if (n_anycompatible_args > 0)
2583 : : {
2584 : 523 : anycompatible_typeid =
2585 : 1046 : select_common_type_from_oids(n_anycompatible_args,
2586 : 523 : anycompatible_actual_types,
2587 : : false);
2588 : :
2589 : : /* We have to verify that the selected type actually works */
2590 [ + - + - ]: 1046 : if (!verify_common_type_from_oids(anycompatible_typeid,
2591 : 523 : n_anycompatible_args,
2592 : 523 : anycompatible_actual_types))
2593 [ # # # # ]: 0 : ereport(ERROR,
2594 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2595 : : errmsg("arguments of anycompatible family cannot be cast to a common type")));
2596 : :
2597 [ + + ]: 523 : if (have_anycompatible_array)
2598 : : {
2599 : 490 : anycompatible_array_typeid = get_array_type(anycompatible_typeid);
2600 [ + - ]: 490 : if (!OidIsValid(anycompatible_array_typeid))
2601 [ # # # # ]: 0 : ereport(ERROR,
2602 : : (errcode(ERRCODE_UNDEFINED_OBJECT),
2603 : : errmsg("could not find array type for data type %s",
2604 : : format_type_be(anycompatible_typeid))));
2605 : 490 : }
2606 : :
2607 [ + + ]: 523 : if (have_anycompatible_range)
2608 : : {
2609 : : /* we can't infer a range type from the others */
2610 [ + + ]: 32 : if (!OidIsValid(anycompatible_range_typeid))
2611 [ + - + - ]: 1 : ereport(ERROR,
2612 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2613 : : errmsg("could not determine polymorphic type %s because input has type %s",
2614 : : "anycompatiblerange", "unknown")));
2615 : :
2616 : : /*
2617 : : * the anycompatible type must exactly match the range element
2618 : : * type
2619 : : */
2620 [ + - ]: 31 : if (anycompatible_range_typelem != anycompatible_typeid)
2621 [ # # # # ]: 0 : ereport(ERROR,
2622 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2623 : : errmsg("anycompatiblerange type %s does not match anycompatible type %s",
2624 : : format_type_be(anycompatible_range_typeid),
2625 : : format_type_be(anycompatible_typeid))));
2626 : 31 : }
2627 : :
2628 [ + + ]: 522 : if (have_anycompatible_multirange)
2629 : : {
2630 : : /* we can't infer a multirange type from the others */
2631 [ + + ]: 14 : if (!OidIsValid(anycompatible_multirange_typeid))
2632 [ + - + - ]: 1 : ereport(ERROR,
2633 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2634 : : errmsg("could not determine polymorphic type %s because input has type %s",
2635 : : "anycompatiblemultirange", "unknown")));
2636 : :
2637 : : /*
2638 : : * the anycompatible type must exactly match the multirange
2639 : : * element type
2640 : : */
2641 [ + - ]: 13 : if (anycompatible_range_typelem != anycompatible_typeid)
2642 [ # # # # ]: 0 : ereport(ERROR,
2643 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2644 : : errmsg("anycompatiblemultirange type %s does not match anycompatible type %s",
2645 : : format_type_be(anycompatible_multirange_typeid),
2646 : : format_type_be(anycompatible_typeid))));
2647 : 13 : }
2648 : :
2649 [ + + ]: 521 : if (have_anycompatible_nonarray)
2650 : : {
2651 : : /*
2652 : : * require the element type to not be an array or domain over
2653 : : * array
2654 : : */
2655 [ + - ]: 2 : if (type_is_array_domain(anycompatible_typeid))
2656 [ # # # # ]: 0 : ereport(ERROR,
2657 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2658 : : errmsg("type matched to anycompatiblenonarray is an array type: %s",
2659 : : format_type_be(anycompatible_typeid))));
2660 : 2 : }
2661 : 521 : }
2662 : : else
2663 : : {
2664 [ + + ]: 5 : if (allow_poly)
2665 : : {
2666 : 1 : anycompatible_typeid = ANYCOMPATIBLEOID;
2667 : 1 : anycompatible_array_typeid = ANYCOMPATIBLEARRAYOID;
2668 : 1 : anycompatible_range_typeid = ANYCOMPATIBLERANGEOID;
2669 : 1 : anycompatible_multirange_typeid = ANYCOMPATIBLEMULTIRANGEOID;
2670 : 1 : }
2671 : : else
2672 : : {
2673 : : /*
2674 : : * Only way to get here is if all the family-2 polymorphic
2675 : : * arguments have UNKNOWN inputs. Resolve to TEXT as
2676 : : * select_common_type() would do. That doesn't license us to
2677 : : * use TEXTRANGE or TEXTMULTIRANGE, though.
2678 : : */
2679 : 4 : anycompatible_typeid = TEXTOID;
2680 : 4 : anycompatible_array_typeid = TEXTARRAYOID;
2681 [ + + ]: 4 : if (have_anycompatible_range)
2682 [ + - + - ]: 2 : ereport(ERROR,
2683 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2684 : : errmsg("could not determine polymorphic type %s because input has type %s",
2685 : : "anycompatiblerange", "unknown")));
2686 [ + + ]: 2 : if (have_anycompatible_multirange)
2687 [ + - + - ]: 1 : ereport(ERROR,
2688 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2689 : : errmsg("could not determine polymorphic type %s because input has type %s",
2690 : : "anycompatiblemultirange", "unknown")));
2691 : : }
2692 : : }
2693 : :
2694 : : /* replace family-2 polymorphic types by selected types */
2695 [ + + ]: 1588 : for (int j = 0; j < nargs; j++)
2696 : : {
2697 : 1065 : Oid decl_type = declared_arg_types[j];
2698 : :
2699 [ + + + + ]: 1065 : if (decl_type == ANYCOMPATIBLEOID ||
2700 : 847 : decl_type == ANYCOMPATIBLENONARRAYOID)
2701 : 222 : declared_arg_types[j] = anycompatible_typeid;
2702 [ + + ]: 843 : else if (decl_type == ANYCOMPATIBLEARRAYOID)
2703 : 786 : declared_arg_types[j] = anycompatible_array_typeid;
2704 [ + + ]: 57 : else if (decl_type == ANYCOMPATIBLERANGEOID)
2705 : 21 : declared_arg_types[j] = anycompatible_range_typeid;
2706 [ + + ]: 36 : else if (decl_type == ANYCOMPATIBLEMULTIRANGEOID)
2707 : 13 : declared_arg_types[j] = anycompatible_multirange_typeid;
2708 : 1065 : }
2709 : 523 : }
2710 : :
2711 : : /*
2712 : : * If we had any UNKNOWN inputs for family-1 polymorphic arguments,
2713 : : * re-scan to assign correct types to them.
2714 : : *
2715 : : * Note: we don't have to consider unknown inputs that were matched to
2716 : : * family-2 polymorphic arguments, because we forcibly updated their
2717 : : * declared_arg_types[] positions just above.
2718 : : */
2719 [ + + ]: 7793 : if (have_poly_unknowns)
2720 : : {
2721 [ + + ]: 2745 : for (int j = 0; j < nargs; j++)
2722 : : {
2723 : 1845 : Oid decl_type = declared_arg_types[j];
2724 : 1845 : Oid actual_type = actual_arg_types[j];
2725 : :
2726 [ + + ]: 1845 : if (actual_type != UNKNOWNOID)
2727 : 938 : continue;
2728 : :
2729 [ + + ]: 907 : if (decl_type == ANYELEMENTOID ||
2730 [ + - + + ]: 863 : decl_type == ANYNONARRAYOID ||
2731 : 863 : decl_type == ANYENUMOID)
2732 : 69 : declared_arg_types[j] = elem_typeid;
2733 [ + + ]: 838 : else if (decl_type == ANYARRAYOID)
2734 : : {
2735 [ + + ]: 667 : if (!OidIsValid(array_typeid))
2736 : : {
2737 : 5 : array_typeid = get_array_type(elem_typeid);
2738 [ + - ]: 5 : if (!OidIsValid(array_typeid))
2739 [ # # # # ]: 0 : ereport(ERROR,
2740 : : (errcode(ERRCODE_UNDEFINED_OBJECT),
2741 : : errmsg("could not find array type for data type %s",
2742 : : format_type_be(elem_typeid))));
2743 : 5 : }
2744 : 667 : declared_arg_types[j] = array_typeid;
2745 : 667 : }
2746 [ + + ]: 171 : else if (decl_type == ANYRANGEOID)
2747 : : {
2748 [ + - ]: 125 : if (!OidIsValid(range_typeid))
2749 : : {
2750 : : /* we can't infer a range type from the others */
2751 [ # # # # ]: 0 : ereport(ERROR,
2752 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2753 : : errmsg("could not determine polymorphic type %s because input has type %s",
2754 : : "anyrange", "unknown")));
2755 : 0 : }
2756 : 125 : declared_arg_types[j] = range_typeid;
2757 : 125 : }
2758 [ + + ]: 46 : else if (decl_type == ANYMULTIRANGEOID)
2759 : : {
2760 [ + - ]: 42 : if (!OidIsValid(multirange_typeid))
2761 : : {
2762 : : /* we can't infer a multirange type from the others */
2763 [ # # # # ]: 0 : ereport(ERROR,
2764 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2765 : : errmsg("could not determine polymorphic type %s because input has type %s",
2766 : : "anymultirange", "unknown")));
2767 : 0 : }
2768 : 42 : declared_arg_types[j] = multirange_typeid;
2769 : 42 : }
2770 [ + + ]: 1845 : }
2771 : 900 : }
2772 : :
2773 : : /* if we return ANYELEMENT use the appropriate argument type */
2774 [ + + ]: 7793 : if (rettype == ANYELEMENTOID ||
2775 [ + - + + ]: 6521 : rettype == ANYNONARRAYOID ||
2776 : 6521 : rettype == ANYENUMOID)
2777 : 1310 : return elem_typeid;
2778 : :
2779 : : /* if we return ANYARRAY use the appropriate argument type */
2780 [ + + ]: 6483 : if (rettype == ANYARRAYOID)
2781 : : {
2782 [ + + ]: 2007 : if (!OidIsValid(array_typeid))
2783 : : {
2784 : 1845 : array_typeid = get_array_type(elem_typeid);
2785 [ + - ]: 1845 : if (!OidIsValid(array_typeid))
2786 [ # # # # ]: 0 : ereport(ERROR,
2787 : : (errcode(ERRCODE_UNDEFINED_OBJECT),
2788 : : errmsg("could not find array type for data type %s",
2789 : : format_type_be(elem_typeid))));
2790 : 1845 : }
2791 : 2007 : return array_typeid;
2792 : : }
2793 : :
2794 : : /* if we return ANYRANGE use the appropriate argument type */
2795 [ + + ]: 4476 : if (rettype == ANYRANGEOID)
2796 : : {
2797 : : /* this error is unreachable if the function signature is valid: */
2798 [ + - ]: 73 : if (!OidIsValid(range_typeid))
2799 [ # # # # ]: 0 : ereport(ERROR,
2800 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2801 : : errmsg_internal("could not determine polymorphic type %s because input has type %s",
2802 : : "anyrange", "unknown")));
2803 : 73 : return range_typeid;
2804 : : }
2805 : :
2806 : : /* if we return ANYMULTIRANGE use the appropriate argument type */
2807 [ + + ]: 4403 : if (rettype == ANYMULTIRANGEOID)
2808 : : {
2809 : : /* this error is unreachable if the function signature is valid: */
2810 [ + - ]: 155 : if (!OidIsValid(multirange_typeid))
2811 [ # # # # ]: 0 : ereport(ERROR,
2812 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2813 : : errmsg_internal("could not determine polymorphic type %s because input has type %s",
2814 : : "anymultirange", "unknown")));
2815 : 155 : return multirange_typeid;
2816 : : }
2817 : :
2818 : : /* if we return ANYCOMPATIBLE use the appropriate type */
2819 [ + + - + ]: 4248 : if (rettype == ANYCOMPATIBLEOID ||
2820 : 4222 : rettype == ANYCOMPATIBLENONARRAYOID)
2821 : : {
2822 : : /* this error is unreachable if the function signature is valid: */
2823 [ + - ]: 26 : if (!OidIsValid(anycompatible_typeid))
2824 [ # # # # ]: 0 : ereport(ERROR,
2825 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2826 : : errmsg_internal("could not identify anycompatible type")));
2827 : 26 : return anycompatible_typeid;
2828 : : }
2829 : :
2830 : : /* if we return ANYCOMPATIBLEARRAY use the appropriate type */
2831 [ + + ]: 4222 : if (rettype == ANYCOMPATIBLEARRAYOID)
2832 : : {
2833 : : /* this error is unreachable if the function signature is valid: */
2834 [ + - ]: 436 : if (!OidIsValid(anycompatible_array_typeid))
2835 [ # # # # ]: 0 : ereport(ERROR,
2836 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2837 : : errmsg_internal("could not identify anycompatiblearray type")));
2838 : 436 : return anycompatible_array_typeid;
2839 : : }
2840 : :
2841 : : /* if we return ANYCOMPATIBLERANGE use the appropriate argument type */
2842 [ + + ]: 3786 : if (rettype == ANYCOMPATIBLERANGEOID)
2843 : : {
2844 : : /* this error is unreachable if the function signature is valid: */
2845 [ + - ]: 7 : if (!OidIsValid(anycompatible_range_typeid))
2846 [ # # # # ]: 0 : ereport(ERROR,
2847 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2848 : : errmsg_internal("could not identify anycompatiblerange type")));
2849 : 7 : return anycompatible_range_typeid;
2850 : : }
2851 : :
2852 : : /* if we return ANYCOMPATIBLEMULTIRANGE use the appropriate argument type */
2853 [ + + ]: 3779 : if (rettype == ANYCOMPATIBLEMULTIRANGEOID)
2854 : : {
2855 : : /* this error is unreachable if the function signature is valid: */
2856 [ + - ]: 5 : if (!OidIsValid(anycompatible_multirange_typeid))
2857 [ # # # # ]: 0 : ereport(ERROR,
2858 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2859 : : errmsg_internal("could not identify anycompatiblemultirange type")));
2860 : 5 : return anycompatible_multirange_typeid;
2861 : : }
2862 : :
2863 : : /* we don't return a generic type; send back the original return type */
2864 : 3774 : return rettype;
2865 : 91705 : }
2866 : :
2867 : : /*
2868 : : * check_valid_polymorphic_signature()
2869 : : * Is a proposed function signature valid per polymorphism rules?
2870 : : *
2871 : : * Returns NULL if the signature is valid (either ret_type is not polymorphic,
2872 : : * or it can be deduced from the given declared argument types). Otherwise,
2873 : : * returns a palloc'd, already translated errdetail string saying why not.
2874 : : */
2875 : : char *
2876 : 1705 : check_valid_polymorphic_signature(Oid ret_type,
2877 : : const Oid *declared_arg_types,
2878 : : int nargs)
2879 : : {
2880 [ + + + + ]: 1705 : if (ret_type == ANYRANGEOID || ret_type == ANYMULTIRANGEOID)
2881 : : {
2882 : : /*
2883 : : * ANYRANGE and ANYMULTIRANGE require an ANYRANGE or ANYMULTIRANGE
2884 : : * input, else we can't tell which of several range types with the
2885 : : * same element type to use.
2886 : : */
2887 [ + + + + ]: 45 : for (int i = 0; i < nargs; i++)
2888 : : {
2889 [ + + + + ]: 23 : if (declared_arg_types[i] == ANYRANGEOID ||
2890 : 18 : declared_arg_types[i] == ANYMULTIRANGEOID)
2891 : 10 : return NULL; /* OK */
2892 : 13 : }
2893 : 24 : return psprintf(_("A result of type %s requires at least one input of type anyrange or anymultirange."),
2894 : 12 : format_type_be(ret_type));
2895 : : }
2896 [ + + + + ]: 1683 : else if (ret_type == ANYCOMPATIBLERANGEOID || ret_type == ANYCOMPATIBLEMULTIRANGEOID)
2897 : : {
2898 : : /*
2899 : : * ANYCOMPATIBLERANGE and ANYCOMPATIBLEMULTIRANGE require an
2900 : : * ANYCOMPATIBLERANGE or ANYCOMPATIBLEMULTIRANGE input, else we can't
2901 : : * tell which of several range types with the same element type to
2902 : : * use.
2903 : : */
2904 [ + + + + ]: 32 : for (int i = 0; i < nargs; i++)
2905 : : {
2906 [ + + + + ]: 17 : if (declared_arg_types[i] == ANYCOMPATIBLERANGEOID ||
2907 : 12 : declared_arg_types[i] == ANYCOMPATIBLEMULTIRANGEOID)
2908 : 8 : return NULL; /* OK */
2909 : 9 : }
2910 : 14 : return psprintf(_("A result of type %s requires at least one input of type anycompatiblerange or anycompatiblemultirange."),
2911 : 7 : format_type_be(ret_type));
2912 : : }
2913 [ + + + + : 1668 : else if (IsPolymorphicTypeFamily1(ret_type))
+ - + - +
- - + ]
2914 : : {
2915 : : /* Otherwise, any family-1 type can be deduced from any other */
2916 [ + + + + ]: 253 : for (int i = 0; i < nargs; i++)
2917 : : {
2918 [ + + + + : 125 : if (IsPolymorphicTypeFamily1(declared_arg_types[i]))
+ - + - +
+ + + ]
2919 : 109 : return NULL; /* OK */
2920 : 16 : }
2921 : : /* Keep this list in sync with IsPolymorphicTypeFamily1! */
2922 : 38 : return psprintf(_("A result of type %s requires at least one input of type anyelement, anyarray, anynonarray, anyenum, anyrange, or anymultirange."),
2923 : 19 : format_type_be(ret_type));
2924 : : }
2925 [ + + + + : 1540 : else if (IsPolymorphicTypeFamily2(ret_type))
+ - + - -
+ ]
2926 : : {
2927 : : /* Otherwise, any family-2 type can be deduced from any other */
2928 [ + + + + ]: 68 : for (int i = 0; i < nargs; i++)
2929 : : {
2930 [ + + + + : 37 : if (IsPolymorphicTypeFamily2(declared_arg_types[i]))
+ + + + +
+ ]
2931 : 30 : return NULL; /* OK */
2932 : 7 : }
2933 : : /* Keep this list in sync with IsPolymorphicTypeFamily2! */
2934 : 2 : return psprintf(_("A result of type %s requires at least one input of type anycompatible, anycompatiblearray, anycompatiblenonarray, anycompatiblerange, or anycompatiblemultirange."),
2935 : 1 : format_type_be(ret_type));
2936 : : }
2937 : : else
2938 : 1509 : return NULL; /* OK, ret_type is not polymorphic */
2939 : 1705 : }
2940 : :
2941 : : /*
2942 : : * check_valid_internal_signature()
2943 : : * Is a proposed function signature valid per INTERNAL safety rules?
2944 : : *
2945 : : * Returns NULL if OK, or a suitable error message if ret_type is INTERNAL but
2946 : : * none of the declared arg types are. (It's unsafe to create such a function
2947 : : * since it would allow invocation of INTERNAL-consuming functions directly
2948 : : * from SQL.) It's overkill to return the error detail message, since there
2949 : : * is only one possibility, but we do it like this to keep the API similar to
2950 : : * check_valid_polymorphic_signature().
2951 : : */
2952 : : char *
2953 : 1554 : check_valid_internal_signature(Oid ret_type,
2954 : : const Oid *declared_arg_types,
2955 : : int nargs)
2956 : : {
2957 [ + + ]: 1554 : if (ret_type == INTERNALOID)
2958 : : {
2959 [ + - - + : 8 : for (int i = 0; i < nargs; i++)
- ]
2960 : : {
2961 [ + - ]: 4 : if (declared_arg_types[i] == ret_type)
2962 : 4 : return NULL; /* OK */
2963 : 0 : }
2964 : 0 : return pstrdup(_("A result of type internal requires at least one input of type internal."));
2965 : : }
2966 : : else
2967 : 1550 : return NULL; /* OK, ret_type is not INTERNAL */
2968 : 1554 : }
2969 : :
2970 : :
2971 : : /* TypeCategory()
2972 : : * Assign a category to the specified type OID.
2973 : : *
2974 : : * NB: this must not return TYPCATEGORY_INVALID.
2975 : : */
2976 : : TYPCATEGORY
2977 : 16653 : TypeCategory(Oid type)
2978 : : {
2979 : 16653 : char typcategory;
2980 : 16653 : bool typispreferred;
2981 : :
2982 : 16653 : get_type_category_preferred(type, &typcategory, &typispreferred);
2983 [ + - ]: 16653 : Assert(typcategory != TYPCATEGORY_INVALID);
2984 : 33306 : return (TYPCATEGORY) typcategory;
2985 : 16653 : }
2986 : :
2987 : :
2988 : : /* IsPreferredType()
2989 : : * Check if this type is a preferred type for the given category.
2990 : : *
2991 : : * If category is TYPCATEGORY_INVALID, then we'll return true for preferred
2992 : : * types of any category; otherwise, only for preferred types of that
2993 : : * category.
2994 : : */
2995 : : bool
2996 : 2335 : IsPreferredType(TYPCATEGORY category, Oid type)
2997 : : {
2998 : 2335 : char typcategory;
2999 : 2335 : bool typispreferred;
3000 : :
3001 : 2335 : get_type_category_preferred(type, &typcategory, &typispreferred);
3002 [ + + - + ]: 2335 : if (category == typcategory || category == TYPCATEGORY_INVALID)
3003 : 1646 : return typispreferred;
3004 : : else
3005 : 689 : return false;
3006 : 2335 : }
3007 : :
3008 : :
3009 : : /* IsBinaryCoercible()
3010 : : * Check if srctype is binary-coercible to targettype.
3011 : : *
3012 : : * This notion allows us to cheat and directly exchange values without
3013 : : * going through the trouble of calling a conversion function. Note that
3014 : : * in general, this should only be an implementation shortcut. Before 7.4,
3015 : : * this was also used as a heuristic for resolving overloaded functions and
3016 : : * operators, but that's basically a bad idea.
3017 : : *
3018 : : * As of 7.3, binary coercibility isn't hardwired into the code anymore.
3019 : : * We consider two types binary-coercible if there is an implicitly
3020 : : * invokable, no-function-needed pg_cast entry. Also, a domain is always
3021 : : * binary-coercible to its base type, though *not* vice versa (in the other
3022 : : * direction, one must apply domain constraint checks before accepting the
3023 : : * value as legitimate). We also need to special-case various polymorphic
3024 : : * types.
3025 : : *
3026 : : * This function replaces IsBinaryCompatible(), which was an inherently
3027 : : * symmetric test. Since the pg_cast entries aren't necessarily symmetric,
3028 : : * the order of the operands is now significant.
3029 : : */
3030 : : bool
3031 : 136171 : IsBinaryCoercible(Oid srctype, Oid targettype)
3032 : : {
3033 : 136171 : Oid castoid;
3034 : :
3035 : 272342 : return IsBinaryCoercibleWithCast(srctype, targettype, &castoid);
3036 : 136171 : }
3037 : :
3038 : : /* IsBinaryCoercibleWithCast()
3039 : : * Check if srctype is binary-coercible to targettype.
3040 : : *
3041 : : * This variant also returns the OID of the pg_cast entry if one is involved.
3042 : : * *castoid is set to InvalidOid if no binary-coercible cast exists, or if
3043 : : * there is a hard-wired rule for it rather than a pg_cast entry.
3044 : : */
3045 : : bool
3046 : 136181 : IsBinaryCoercibleWithCast(Oid srctype, Oid targettype,
3047 : : Oid *castoid)
3048 : : {
3049 : 136181 : HeapTuple tuple;
3050 : 136181 : Form_pg_cast castForm;
3051 : 136181 : bool result;
3052 : :
3053 : 136181 : *castoid = InvalidOid;
3054 : :
3055 : : /* Fast path if same type */
3056 [ + + ]: 136181 : if (srctype == targettype)
3057 : 25155 : return true;
3058 : :
3059 : : /* Anything is coercible to ANY or ANYELEMENT or ANYCOMPATIBLE */
3060 [ + + + + : 111026 : if (targettype == ANYOID || targettype == ANYELEMENTOID ||
- + ]
3061 : 111008 : targettype == ANYCOMPATIBLEOID)
3062 : 18 : return true;
3063 : :
3064 : : /* If srctype is a domain, reduce to its base type */
3065 [ - + ]: 111008 : if (OidIsValid(srctype))
3066 : 111008 : srctype = getBaseType(srctype);
3067 : :
3068 : : /* Somewhat-fast path for domain -> base type case */
3069 [ + + ]: 111008 : if (srctype == targettype)
3070 : 2 : return true;
3071 : :
3072 : : /* Also accept any array type as coercible to ANY[COMPATIBLE]ARRAY */
3073 [ + + - + ]: 111006 : if (targettype == ANYARRAYOID || targettype == ANYCOMPATIBLEARRAYOID)
3074 [ + + ]: 6416 : if (type_is_array(srctype))
3075 : 106 : return true;
3076 : :
3077 : : /* Also accept any non-array type as coercible to ANY[COMPATIBLE]NONARRAY */
3078 [ + - - + ]: 110900 : if (targettype == ANYNONARRAYOID || targettype == ANYCOMPATIBLENONARRAYOID)
3079 [ # # ]: 0 : if (!type_is_array(srctype))
3080 : 0 : return true;
3081 : :
3082 : : /* Also accept any enum type as coercible to ANYENUM */
3083 [ + + ]: 110900 : if (targettype == ANYENUMOID)
3084 [ + + ]: 5938 : if (type_is_enum(srctype))
3085 : 8 : return true;
3086 : :
3087 : : /* Also accept any range type as coercible to ANY[COMPATIBLE]RANGE */
3088 [ + + - + ]: 110892 : if (targettype == ANYRANGEOID || targettype == ANYCOMPATIBLERANGEOID)
3089 [ + + ]: 1689 : if (type_is_range(srctype))
3090 : 385 : return true;
3091 : :
3092 : : /* Also, any multirange type is coercible to ANY[COMPATIBLE]MULTIRANGE */
3093 [ + + - + ]: 110507 : if (targettype == ANYMULTIRANGEOID || targettype == ANYCOMPATIBLEMULTIRANGEOID)
3094 [ + + ]: 2379 : if (type_is_multirange(srctype))
3095 : 54 : return true;
3096 : :
3097 : : /* Also accept any composite type as coercible to RECORD */
3098 [ + + ]: 110453 : if (targettype == RECORDOID)
3099 [ + + ]: 1271 : if (ISCOMPLEX(srctype))
3100 : 64 : return true;
3101 : :
3102 : : /* Also accept any composite array type as coercible to RECORD[] */
3103 [ + - ]: 110389 : if (targettype == RECORDARRAYOID)
3104 [ # # ]: 0 : if (is_complex_array(srctype))
3105 : 0 : return true;
3106 : :
3107 : : /* Else look in pg_cast */
3108 : 110389 : tuple = SearchSysCache2(CASTSOURCETARGET,
3109 : 110389 : ObjectIdGetDatum(srctype),
3110 : 110389 : ObjectIdGetDatum(targettype));
3111 [ + + ]: 110389 : if (!HeapTupleIsValid(tuple))
3112 : 79476 : return false; /* no cast */
3113 : 30913 : castForm = (Form_pg_cast) GETSTRUCT(tuple);
3114 : :
3115 [ + + ]: 32510 : result = (castForm->castmethod == COERCION_METHOD_BINARY &&
3116 : 1597 : castForm->castcontext == COERCION_CODE_IMPLICIT);
3117 : :
3118 [ + + ]: 30913 : if (result)
3119 : 1133 : *castoid = castForm->oid;
3120 : :
3121 : 30913 : ReleaseSysCache(tuple);
3122 : :
3123 : 30913 : return result;
3124 : 136181 : }
3125 : :
3126 : :
3127 : : /*
3128 : : * find_coercion_pathway
3129 : : * Look for a coercion pathway between two types.
3130 : : *
3131 : : * Currently, this deals only with scalar-type cases; it does not consider
3132 : : * polymorphic types nor casts between composite types. (Perhaps fold
3133 : : * those in someday?)
3134 : : *
3135 : : * ccontext determines the set of available casts.
3136 : : *
3137 : : * The possible result codes are:
3138 : : * COERCION_PATH_NONE: failed to find any coercion pathway
3139 : : * *funcid is set to InvalidOid
3140 : : * COERCION_PATH_FUNC: apply the coercion function returned in *funcid
3141 : : * COERCION_PATH_RELABELTYPE: binary-compatible cast, no function needed
3142 : : * *funcid is set to InvalidOid
3143 : : * COERCION_PATH_ARRAYCOERCE: need an ArrayCoerceExpr node
3144 : : * *funcid is set to InvalidOid
3145 : : * COERCION_PATH_COERCEVIAIO: need a CoerceViaIO node
3146 : : * *funcid is set to InvalidOid
3147 : : *
3148 : : * Note: COERCION_PATH_RELABELTYPE does not necessarily mean that no work is
3149 : : * needed to do the coercion; if the target is a domain then we may need to
3150 : : * apply domain constraint checking. If you want to check for a zero-effort
3151 : : * conversion then use IsBinaryCoercible().
3152 : : */
3153 : : CoercionPathType
3154 : 101721 : find_coercion_pathway(Oid targetTypeId, Oid sourceTypeId,
3155 : : CoercionContext ccontext,
3156 : : Oid *funcid)
3157 : : {
3158 : 101721 : CoercionPathType result = COERCION_PATH_NONE;
3159 : 101721 : HeapTuple tuple;
3160 : :
3161 : 101721 : *funcid = InvalidOid;
3162 : :
3163 : : /* Perhaps the types are domains; if so, look at their base types */
3164 [ - + ]: 101721 : if (OidIsValid(sourceTypeId))
3165 : 101721 : sourceTypeId = getBaseType(sourceTypeId);
3166 [ - + ]: 101721 : if (OidIsValid(targetTypeId))
3167 : 101721 : targetTypeId = getBaseType(targetTypeId);
3168 : :
3169 : : /* Domains are always coercible to and from their base type */
3170 [ + + ]: 101721 : if (sourceTypeId == targetTypeId)
3171 : 2594 : return COERCION_PATH_RELABELTYPE;
3172 : :
3173 : : /* Look in pg_cast */
3174 : 99127 : tuple = SearchSysCache2(CASTSOURCETARGET,
3175 : 99127 : ObjectIdGetDatum(sourceTypeId),
3176 : 99127 : ObjectIdGetDatum(targetTypeId));
3177 : :
3178 [ + + ]: 99127 : if (HeapTupleIsValid(tuple))
3179 : : {
3180 : 41476 : Form_pg_cast castForm = (Form_pg_cast) GETSTRUCT(tuple);
3181 : 41476 : CoercionContext castcontext;
3182 : :
3183 : : /* convert char value for castcontext to CoercionContext enum */
3184 [ + + + - ]: 41476 : switch (castForm->castcontext)
3185 : : {
3186 : : case COERCION_CODE_IMPLICIT:
3187 : 32546 : castcontext = COERCION_IMPLICIT;
3188 : 32546 : break;
3189 : : case COERCION_CODE_ASSIGNMENT:
3190 : 7866 : castcontext = COERCION_ASSIGNMENT;
3191 : 7866 : break;
3192 : : case COERCION_CODE_EXPLICIT:
3193 : 1064 : castcontext = COERCION_EXPLICIT;
3194 : 1064 : break;
3195 : : default:
3196 [ # # # # ]: 0 : elog(ERROR, "unrecognized castcontext: %d",
3197 : : (int) castForm->castcontext);
3198 : 0 : castcontext = 0; /* keep compiler quiet */
3199 : 0 : break;
3200 : : }
3201 : :
3202 : : /* Rely on ordering of enum for correct behavior here */
3203 [ + + ]: 41476 : if (ccontext >= castcontext)
3204 : : {
3205 [ + + + - ]: 34603 : switch (castForm->castmethod)
3206 : : {
3207 : : case COERCION_METHOD_FUNCTION:
3208 : 16589 : result = COERCION_PATH_FUNC;
3209 : 16589 : *funcid = castForm->castfunc;
3210 : 16589 : break;
3211 : : case COERCION_METHOD_INOUT:
3212 : 124 : result = COERCION_PATH_COERCEVIAIO;
3213 : 124 : break;
3214 : : case COERCION_METHOD_BINARY:
3215 : 17890 : result = COERCION_PATH_RELABELTYPE;
3216 : 17890 : break;
3217 : : default:
3218 [ # # # # ]: 0 : elog(ERROR, "unrecognized castmethod: %d",
3219 : : (int) castForm->castmethod);
3220 : 0 : break;
3221 : : }
3222 : 34603 : }
3223 : :
3224 : 41476 : ReleaseSysCache(tuple);
3225 : 41476 : }
3226 : : else
3227 : : {
3228 : : /*
3229 : : * If there's no pg_cast entry, perhaps we are dealing with a pair of
3230 : : * array types. If so, and if their element types have a conversion
3231 : : * pathway, report that we can coerce with an ArrayCoerceExpr.
3232 : : *
3233 : : * Hack: disallow coercions to oidvector and int2vector, which
3234 : : * otherwise tend to capture coercions that should go to "real" array
3235 : : * types. We want those types to be considered "real" arrays for many
3236 : : * purposes, but not this one. (Also, ArrayCoerceExpr isn't
3237 : : * guaranteed to produce an output that meets the restrictions of
3238 : : * these datatypes, such as being 1-dimensional.)
3239 : : */
3240 [ + + - + ]: 57651 : if (targetTypeId != OIDVECTOROID && targetTypeId != INT2VECTOROID)
3241 : : {
3242 : 56913 : Oid targetElem;
3243 : 56913 : Oid sourceElem;
3244 : :
3245 [ + + + + ]: 56913 : if ((targetElem = get_element_type(targetTypeId)) != InvalidOid &&
3246 : 1445 : (sourceElem = get_element_type(sourceTypeId)) != InvalidOid)
3247 : : {
3248 : 1278 : CoercionPathType elempathtype;
3249 : 1278 : Oid elemfuncid;
3250 : :
3251 : 2556 : elempathtype = find_coercion_pathway(targetElem,
3252 : 1278 : sourceElem,
3253 : 1278 : ccontext,
3254 : : &elemfuncid);
3255 [ + + ]: 1278 : if (elempathtype != COERCION_PATH_NONE)
3256 : : {
3257 : 1260 : result = COERCION_PATH_ARRAYCOERCE;
3258 : 1260 : }
3259 : 1278 : }
3260 : 56913 : }
3261 : :
3262 : : /*
3263 : : * If we still haven't found a possibility, consider automatic casting
3264 : : * using I/O functions. We allow assignment casts to string types and
3265 : : * explicit casts from string types to be handled this way. (The
3266 : : * CoerceViaIO mechanism is a lot more general than that, but this is
3267 : : * all we want to allow in the absence of a pg_cast entry.) It would
3268 : : * probably be better to insist on explicit casts in both directions,
3269 : : * but this is a compromise to preserve something of the pre-8.3
3270 : : * behavior that many types had implicit (yipes!) casts to text.
3271 : : */
3272 [ + + ]: 57651 : if (result == COERCION_PATH_NONE)
3273 : : {
3274 [ + + + + ]: 56391 : if (ccontext >= COERCION_ASSIGNMENT &&
3275 : 6595 : TypeCategory(targetTypeId) == TYPCATEGORY_STRING)
3276 : 5227 : result = COERCION_PATH_COERCEVIAIO;
3277 [ + + + + ]: 51164 : else if (ccontext >= COERCION_EXPLICIT &&
3278 : 1255 : TypeCategory(sourceTypeId) == TYPCATEGORY_STRING)
3279 : 772 : result = COERCION_PATH_COERCEVIAIO;
3280 : 56391 : }
3281 : : }
3282 : :
3283 : : /*
3284 : : * When parsing PL/pgSQL assignments, allow an I/O cast to be used
3285 : : * whenever no normal coercion is available.
3286 : : */
3287 [ + + + + ]: 99127 : if (result == COERCION_PATH_NONE &&
3288 : 57265 : ccontext == COERCION_PLPGSQL)
3289 : 32 : result = COERCION_PATH_COERCEVIAIO;
3290 : :
3291 : 99127 : return result;
3292 : 101721 : }
3293 : :
3294 : :
3295 : : /*
3296 : : * find_typmod_coercion_function -- does the given type need length coercion?
3297 : : *
3298 : : * If the target type possesses a pg_cast function from itself to itself,
3299 : : * it must need length coercion.
3300 : : *
3301 : : * "bpchar" (ie, char(N)) and "numeric" are examples of such types.
3302 : : *
3303 : : * If the given type is a varlena array type, we do not look for a coercion
3304 : : * function associated directly with the array type, but instead look for
3305 : : * one associated with the element type. An ArrayCoerceExpr node must be
3306 : : * used to apply such a function. (Note: currently, it's pointless to
3307 : : * return the funcid in this case, because it'll just get looked up again
3308 : : * in the recursive construction of the ArrayCoerceExpr's elemexpr.)
3309 : : *
3310 : : * We use the same result enum as find_coercion_pathway, but the only possible
3311 : : * result codes are:
3312 : : * COERCION_PATH_NONE: no length coercion needed
3313 : : * COERCION_PATH_FUNC: apply the function returned in *funcid
3314 : : * COERCION_PATH_ARRAYCOERCE: apply the function using ArrayCoerceExpr
3315 : : */
3316 : : CoercionPathType
3317 : 2313 : find_typmod_coercion_function(Oid typeId,
3318 : : Oid *funcid)
3319 : : {
3320 : 2313 : CoercionPathType result;
3321 : 2313 : Type targetType;
3322 : 2313 : Form_pg_type typeForm;
3323 : 2313 : HeapTuple tuple;
3324 : :
3325 : 2313 : *funcid = InvalidOid;
3326 : 2313 : result = COERCION_PATH_FUNC;
3327 : :
3328 : 2313 : targetType = typeidType(typeId);
3329 : 2313 : typeForm = (Form_pg_type) GETSTRUCT(targetType);
3330 : :
3331 : : /* Check for a "true" array type */
3332 [ + + - + ]: 2313 : if (IsTrueArrayType(typeForm))
3333 : : {
3334 : : /* Yes, switch our attention to the element type */
3335 : 13 : typeId = typeForm->typelem;
3336 : 13 : result = COERCION_PATH_ARRAYCOERCE;
3337 : 13 : }
3338 : 2313 : ReleaseSysCache(targetType);
3339 : :
3340 : : /* Look in pg_cast */
3341 : 2313 : tuple = SearchSysCache2(CASTSOURCETARGET,
3342 : 2313 : ObjectIdGetDatum(typeId),
3343 : 2313 : ObjectIdGetDatum(typeId));
3344 : :
3345 [ + + ]: 2313 : if (HeapTupleIsValid(tuple))
3346 : : {
3347 : 2311 : Form_pg_cast castForm = (Form_pg_cast) GETSTRUCT(tuple);
3348 : :
3349 : 2311 : *funcid = castForm->castfunc;
3350 : 2311 : ReleaseSysCache(tuple);
3351 : 2311 : }
3352 : :
3353 [ + + ]: 2313 : if (!OidIsValid(*funcid))
3354 : 2 : result = COERCION_PATH_NONE;
3355 : :
3356 : 4626 : return result;
3357 : 2313 : }
3358 : :
3359 : : /*
3360 : : * is_complex_array
3361 : : * Is this type an array of composite?
3362 : : *
3363 : : * Note: this will not return true for record[]; check for RECORDARRAYOID
3364 : : * separately if needed.
3365 : : */
3366 : : static bool
3367 : 4 : is_complex_array(Oid typid)
3368 : : {
3369 : 4 : Oid elemtype = get_element_type(typid);
3370 : :
3371 [ + + ]: 4 : return (OidIsValid(elemtype) && ISCOMPLEX(elemtype));
3372 : 4 : }
3373 : :
3374 : :
3375 : : /*
3376 : : * Check whether reltypeId is the row type of a typed table of type
3377 : : * reloftypeId, or is a domain over such a row type. (This is conceptually
3378 : : * similar to the subtype relationship checked by typeInheritsFrom().)
3379 : : */
3380 : : static bool
3381 : 56275 : typeIsOfTypedTable(Oid reltypeId, Oid reloftypeId)
3382 : : {
3383 : 56275 : Oid relid = typeOrDomainTypeRelid(reltypeId);
3384 : 56275 : bool result = false;
3385 : :
3386 [ + + ]: 56275 : if (relid)
3387 : : {
3388 : 2126 : HeapTuple tp;
3389 : 2126 : Form_pg_class reltup;
3390 : :
3391 : 2126 : tp = SearchSysCache1(RELOID, ObjectIdGetDatum(relid));
3392 [ + - ]: 2126 : if (!HeapTupleIsValid(tp))
3393 [ # # # # ]: 0 : elog(ERROR, "cache lookup failed for relation %u", relid);
3394 : :
3395 : 2126 : reltup = (Form_pg_class) GETSTRUCT(tp);
3396 [ + + ]: 2126 : if (reltup->reloftype == reloftypeId)
3397 : 2 : result = true;
3398 : :
3399 : 2126 : ReleaseSysCache(tp);
3400 : 2126 : }
3401 : :
3402 : 112550 : return result;
3403 : 56275 : }
|
