Line data Source code
1 : /*-------------------------------------------------------------------------
2 : *
3 : * readfuncs.c
4 : * Reader functions for Postgres tree nodes.
5 : *
6 : * Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
7 : * Portions Copyright (c) 1994, Regents of the University of California
8 : *
9 : *
10 : * IDENTIFICATION
11 : * src/backend/nodes/readfuncs.c
12 : *
13 : * NOTES
14 : * Parse location fields are written out by outfuncs.c, but only for
15 : * debugging use. When reading a location field, we normally discard
16 : * the stored value and set the location field to -1 (ie, "unknown").
17 : * This is because nodes coming from a stored rule should not be thought
18 : * to have a known location in the current query's text.
19 : *
20 : * However, if restore_location_fields is true, we do restore location
21 : * fields from the string. This is currently intended only for use by the
22 : * WRITE_READ_PARSE_PLAN_TREES test code, which doesn't want to cause
23 : * any change in the node contents.
24 : *
25 : *-------------------------------------------------------------------------
26 : */
27 : #include "postgres.h"
28 :
29 : #include <math.h>
30 :
31 : #include "miscadmin.h"
32 : #include "nodes/bitmapset.h"
33 : #include "nodes/readfuncs.h"
34 :
35 :
36 : /*
37 : * Macros to simplify reading of different kinds of fields. Use these
38 : * wherever possible to reduce the chance for silly typos. Note that these
39 : * hard-wire conventions about the names of the local variables in a Read
40 : * routine.
41 : */
42 :
43 : /* Macros for declaring appropriate local variables */
44 :
45 : /* A few guys need only local_node */
46 : #define READ_LOCALS_NO_FIELDS(nodeTypeName) \
47 : nodeTypeName *local_node = makeNode(nodeTypeName)
48 :
49 : /* And a few guys need only the pg_strtok support fields */
50 : #define READ_TEMP_LOCALS() \
51 : const char *token; \
52 : int length
53 :
54 : /* ... but most need both */
55 : #define READ_LOCALS(nodeTypeName) \
56 : READ_LOCALS_NO_FIELDS(nodeTypeName); \
57 : READ_TEMP_LOCALS()
58 :
59 : /* Read an integer field (anything written as ":fldname %d") */
60 : #define READ_INT_FIELD(fldname) \
61 : token = pg_strtok(&length); /* skip :fldname */ \
62 : token = pg_strtok(&length); /* get field value */ \
63 : local_node->fldname = atoi(token)
64 :
65 : /* Read an unsigned integer field (anything written as ":fldname %u") */
66 : #define READ_UINT_FIELD(fldname) \
67 : token = pg_strtok(&length); /* skip :fldname */ \
68 : token = pg_strtok(&length); /* get field value */ \
69 : local_node->fldname = atoui(token)
70 :
71 : /* Read an unsigned integer field (anything written using UINT64_FORMAT) */
72 : #define READ_UINT64_FIELD(fldname) \
73 : token = pg_strtok(&length); /* skip :fldname */ \
74 : token = pg_strtok(&length); /* get field value */ \
75 : local_node->fldname = strtou64(token, NULL, 10)
76 :
77 : /* Read a long integer field (anything written as ":fldname %ld") */
78 : #define READ_LONG_FIELD(fldname) \
79 : token = pg_strtok(&length); /* skip :fldname */ \
80 : token = pg_strtok(&length); /* get field value */ \
81 : local_node->fldname = atol(token)
82 :
83 : /* Read an OID field (don't hard-wire assumption that OID is same as uint) */
84 : #define READ_OID_FIELD(fldname) \
85 : token = pg_strtok(&length); /* skip :fldname */ \
86 : token = pg_strtok(&length); /* get field value */ \
87 : local_node->fldname = atooid(token)
88 :
89 : /* Read a char field (ie, one ascii character) */
90 : #define READ_CHAR_FIELD(fldname) \
91 : token = pg_strtok(&length); /* skip :fldname */ \
92 : token = pg_strtok(&length); /* get field value */ \
93 : /* avoid overhead of calling debackslash() for one char */ \
94 : local_node->fldname = (length == 0) ? '\0' : (token[0] == '\\' ? token[1] : token[0])
95 :
96 : /* Read an enumerated-type field that was written as an integer code */
97 : #define READ_ENUM_FIELD(fldname, enumtype) \
98 : token = pg_strtok(&length); /* skip :fldname */ \
99 : token = pg_strtok(&length); /* get field value */ \
100 : local_node->fldname = (enumtype) atoi(token)
101 :
102 : /* Read a float field */
103 : #define READ_FLOAT_FIELD(fldname) \
104 : token = pg_strtok(&length); /* skip :fldname */ \
105 : token = pg_strtok(&length); /* get field value */ \
106 : local_node->fldname = atof(token)
107 :
108 : /* Read a boolean field */
109 : #define READ_BOOL_FIELD(fldname) \
110 : token = pg_strtok(&length); /* skip :fldname */ \
111 : token = pg_strtok(&length); /* get field value */ \
112 : local_node->fldname = strtobool(token)
113 :
114 : /* Read a character-string field */
115 : #define READ_STRING_FIELD(fldname) \
116 : token = pg_strtok(&length); /* skip :fldname */ \
117 : token = pg_strtok(&length); /* get field value */ \
118 : local_node->fldname = nullable_string(token, length)
119 :
120 : /* Read a parse location field (and possibly throw away the value) */
121 : #ifdef WRITE_READ_PARSE_PLAN_TREES
122 : #define READ_LOCATION_FIELD(fldname) \
123 : token = pg_strtok(&length); /* skip :fldname */ \
124 : token = pg_strtok(&length); /* get field value */ \
125 : local_node->fldname = restore_location_fields ? atoi(token) : -1
126 : #else
127 : #define READ_LOCATION_FIELD(fldname) \
128 : token = pg_strtok(&length); /* skip :fldname */ \
129 : token = pg_strtok(&length); /* get field value */ \
130 : (void) token; /* in case not used elsewhere */ \
131 : local_node->fldname = -1 /* set field to "unknown" */
132 : #endif
133 :
134 : /* Read a Node field */
135 : #define READ_NODE_FIELD(fldname) \
136 : token = pg_strtok(&length); /* skip :fldname */ \
137 : (void) token; /* in case not used elsewhere */ \
138 : local_node->fldname = nodeRead(NULL, 0)
139 :
140 : /* Read a bitmapset field */
141 : #define READ_BITMAPSET_FIELD(fldname) \
142 : token = pg_strtok(&length); /* skip :fldname */ \
143 : (void) token; /* in case not used elsewhere */ \
144 : local_node->fldname = _readBitmapset()
145 :
146 : /* Read an attribute number array */
147 : #define READ_ATTRNUMBER_ARRAY(fldname, len) \
148 : token = pg_strtok(&length); /* skip :fldname */ \
149 : local_node->fldname = readAttrNumberCols(len)
150 :
151 : /* Read an oid array */
152 : #define READ_OID_ARRAY(fldname, len) \
153 : token = pg_strtok(&length); /* skip :fldname */ \
154 : local_node->fldname = readOidCols(len)
155 :
156 : /* Read an int array */
157 : #define READ_INT_ARRAY(fldname, len) \
158 : token = pg_strtok(&length); /* skip :fldname */ \
159 : local_node->fldname = readIntCols(len)
160 :
161 : /* Read a bool array */
162 : #define READ_BOOL_ARRAY(fldname, len) \
163 : token = pg_strtok(&length); /* skip :fldname */ \
164 : local_node->fldname = readBoolCols(len)
165 :
166 : /* Routine exit */
167 : #define READ_DONE() \
168 : return local_node
169 :
170 :
171 : /*
172 : * NOTE: use atoi() to read values written with %d, or atoui() to read
173 : * values written with %u in outfuncs.c. An exception is OID values,
174 : * for which use atooid(). (As of 7.1, outfuncs.c writes OIDs as %u,
175 : * but this will probably change in the future.)
176 : */
177 : #define atoui(x) ((unsigned int) strtoul((x), NULL, 10))
178 :
179 : #define strtobool(x) ((*(x) == 't') ? true : false)
180 :
181 : static char *
182 14518546 : nullable_string(const char *token, int length)
183 : {
184 : /* outToken emits <> for NULL, and pg_strtok makes that an empty string */
185 14518546 : if (length == 0)
186 6795394 : return NULL;
187 : /* outToken emits "" for empty string */
188 7723152 : if (length == 2 && token[0] == '"' && token[1] == '"')
189 156 : return pstrdup("");
190 : /* otherwise, we must remove protective backslashes added by outToken */
191 7722996 : return debackslash(token, length);
192 : }
193 :
194 :
195 : /*
196 : * _readBitmapset
197 : *
198 : * Note: this code is used in contexts where we know that a Bitmapset
199 : * is expected. There is equivalent code in nodeRead() that can read a
200 : * Bitmapset when we come across one in other contexts.
201 : */
202 : static Bitmapset *
203 16999128 : _readBitmapset(void)
204 : {
205 16999128 : Bitmapset *result = NULL;
206 :
207 : READ_TEMP_LOCALS();
208 :
209 16999128 : token = pg_strtok(&length);
210 16999128 : if (token == NULL)
211 0 : elog(ERROR, "incomplete Bitmapset structure");
212 16999128 : if (length != 1 || token[0] != '(')
213 0 : elog(ERROR, "unrecognized token: \"%.*s\"", length, token);
214 :
215 16999128 : token = pg_strtok(&length);
216 16999128 : if (token == NULL)
217 0 : elog(ERROR, "incomplete Bitmapset structure");
218 16999128 : if (length != 1 || token[0] != 'b')
219 0 : elog(ERROR, "unrecognized token: \"%.*s\"", length, token);
220 :
221 : for (;;)
222 4636106 : {
223 : int val;
224 : char *endptr;
225 :
226 21635234 : token = pg_strtok(&length);
227 21635234 : if (token == NULL)
228 0 : elog(ERROR, "unterminated Bitmapset structure");
229 21635234 : if (length == 1 && token[0] == ')')
230 16999128 : break;
231 4636106 : val = (int) strtol(token, &endptr, 10);
232 4636106 : if (endptr != token + length)
233 0 : elog(ERROR, "unrecognized integer: \"%.*s\"", length, token);
234 4636106 : result = bms_add_member(result, val);
235 : }
236 :
237 16999128 : return result;
238 : }
239 :
240 : /*
241 : * We export this function for use by extensions that define extensible nodes.
242 : * That's somewhat historical, though, because calling nodeRead() will work.
243 : */
244 : Bitmapset *
245 0 : readBitmapset(void)
246 : {
247 0 : return _readBitmapset();
248 : }
249 :
250 : #include "readfuncs.funcs.c"
251 :
252 :
253 : /*
254 : * Support functions for nodes with custom_read_write attribute or
255 : * special_read_write attribute
256 : */
257 :
258 : static Const *
259 2622844 : _readConst(void)
260 : {
261 2622844 : READ_LOCALS(Const);
262 :
263 2622844 : READ_OID_FIELD(consttype);
264 2622844 : READ_INT_FIELD(consttypmod);
265 2622844 : READ_OID_FIELD(constcollid);
266 2622844 : READ_INT_FIELD(constlen);
267 2622844 : READ_BOOL_FIELD(constbyval);
268 2622844 : READ_BOOL_FIELD(constisnull);
269 2622844 : READ_LOCATION_FIELD(location);
270 :
271 2622844 : token = pg_strtok(&length); /* skip :constvalue */
272 2622844 : if (local_node->constisnull)
273 215322 : token = pg_strtok(&length); /* skip "<>" */
274 : else
275 2407522 : local_node->constvalue = readDatum(local_node->constbyval);
276 :
277 2622844 : READ_DONE();
278 : }
279 :
280 : static BoolExpr *
281 340128 : _readBoolExpr(void)
282 : {
283 340128 : READ_LOCALS(BoolExpr);
284 :
285 : /* do-it-yourself enum representation */
286 340128 : token = pg_strtok(&length); /* skip :boolop */
287 340128 : token = pg_strtok(&length); /* get field value */
288 340128 : if (length == 3 && strncmp(token, "and", 3) == 0)
289 258454 : local_node->boolop = AND_EXPR;
290 81674 : else if (length == 2 && strncmp(token, "or", 2) == 0)
291 42120 : local_node->boolop = OR_EXPR;
292 39554 : else if (length == 3 && strncmp(token, "not", 3) == 0)
293 39554 : local_node->boolop = NOT_EXPR;
294 : else
295 0 : elog(ERROR, "unrecognized boolop \"%.*s\"", length, token);
296 :
297 340128 : READ_NODE_FIELD(args);
298 340128 : READ_LOCATION_FIELD(location);
299 :
300 340128 : READ_DONE();
301 : }
302 :
303 : static A_Const *
304 1259380 : _readA_Const(void)
305 : {
306 1259380 : READ_LOCALS(A_Const);
307 :
308 : /* We expect either NULL or :val here */
309 1259380 : token = pg_strtok(&length);
310 1259380 : if (length == 4 && strncmp(token, "NULL", 4) == 0)
311 71606 : local_node->isnull = true;
312 : else
313 : {
314 1187774 : union ValUnion *tmp = nodeRead(NULL, 0);
315 :
316 : /* To forestall valgrind complaints, copy only the valid data */
317 1187774 : switch (nodeTag(tmp))
318 : {
319 453908 : case T_Integer:
320 453908 : memcpy(&local_node->val, tmp, sizeof(Integer));
321 453908 : break;
322 11042 : case T_Float:
323 11042 : memcpy(&local_node->val, tmp, sizeof(Float));
324 11042 : break;
325 57418 : case T_Boolean:
326 57418 : memcpy(&local_node->val, tmp, sizeof(Boolean));
327 57418 : break;
328 661338 : case T_String:
329 661338 : memcpy(&local_node->val, tmp, sizeof(String));
330 661338 : break;
331 4068 : case T_BitString:
332 4068 : memcpy(&local_node->val, tmp, sizeof(BitString));
333 4068 : break;
334 0 : default:
335 0 : elog(ERROR, "unrecognized node type: %d",
336 : (int) nodeTag(tmp));
337 : break;
338 : }
339 : }
340 :
341 1259380 : READ_LOCATION_FIELD(location);
342 :
343 1259380 : READ_DONE();
344 : }
345 :
346 : static RangeTblEntry *
347 1568058 : _readRangeTblEntry(void)
348 : {
349 1568058 : READ_LOCALS(RangeTblEntry);
350 :
351 1568058 : READ_NODE_FIELD(alias);
352 1568058 : READ_NODE_FIELD(eref);
353 1568058 : READ_ENUM_FIELD(rtekind, RTEKind);
354 :
355 1568058 : switch (local_node->rtekind)
356 : {
357 951930 : case RTE_RELATION:
358 951930 : READ_OID_FIELD(relid);
359 951930 : READ_BOOL_FIELD(inh);
360 951930 : READ_CHAR_FIELD(relkind);
361 951930 : READ_INT_FIELD(rellockmode);
362 951930 : READ_UINT_FIELD(perminfoindex);
363 951930 : READ_NODE_FIELD(tablesample);
364 951930 : break;
365 114386 : case RTE_SUBQUERY:
366 114386 : READ_NODE_FIELD(subquery);
367 114386 : READ_BOOL_FIELD(security_barrier);
368 : /* we re-use these RELATION fields, too: */
369 114386 : READ_OID_FIELD(relid);
370 114386 : READ_BOOL_FIELD(inh);
371 114386 : READ_CHAR_FIELD(relkind);
372 114386 : READ_INT_FIELD(rellockmode);
373 114386 : READ_UINT_FIELD(perminfoindex);
374 114386 : break;
375 173554 : case RTE_JOIN:
376 173554 : READ_ENUM_FIELD(jointype, JoinType);
377 173554 : READ_INT_FIELD(joinmergedcols);
378 173554 : READ_NODE_FIELD(joinaliasvars);
379 173554 : READ_NODE_FIELD(joinleftcols);
380 173554 : READ_NODE_FIELD(joinrightcols);
381 173554 : READ_NODE_FIELD(join_using_alias);
382 173554 : break;
383 89232 : case RTE_FUNCTION:
384 89232 : READ_NODE_FIELD(functions);
385 89232 : READ_BOOL_FIELD(funcordinality);
386 89232 : break;
387 1254 : case RTE_TABLEFUNC:
388 1254 : READ_NODE_FIELD(tablefunc);
389 : /* The RTE must have a copy of the column type info, if any */
390 1254 : if (local_node->tablefunc)
391 : {
392 706 : TableFunc *tf = local_node->tablefunc;
393 :
394 706 : local_node->coltypes = tf->coltypes;
395 706 : local_node->coltypmods = tf->coltypmods;
396 706 : local_node->colcollations = tf->colcollations;
397 : }
398 1254 : break;
399 18550 : case RTE_VALUES:
400 18550 : READ_NODE_FIELD(values_lists);
401 18550 : READ_NODE_FIELD(coltypes);
402 18550 : READ_NODE_FIELD(coltypmods);
403 18550 : READ_NODE_FIELD(colcollations);
404 18550 : break;
405 9976 : case RTE_CTE:
406 9976 : READ_STRING_FIELD(ctename);
407 9976 : READ_UINT_FIELD(ctelevelsup);
408 9976 : READ_BOOL_FIELD(self_reference);
409 9976 : READ_NODE_FIELD(coltypes);
410 9976 : READ_NODE_FIELD(coltypmods);
411 9976 : READ_NODE_FIELD(colcollations);
412 9976 : break;
413 876 : case RTE_NAMEDTUPLESTORE:
414 876 : READ_STRING_FIELD(enrname);
415 876 : READ_FLOAT_FIELD(enrtuples);
416 876 : READ_NODE_FIELD(coltypes);
417 876 : READ_NODE_FIELD(coltypmods);
418 876 : READ_NODE_FIELD(colcollations);
419 : /* we re-use these RELATION fields, too: */
420 876 : READ_OID_FIELD(relid);
421 876 : break;
422 208300 : case RTE_RESULT:
423 : /* no extra fields */
424 208300 : break;
425 0 : default:
426 0 : elog(ERROR, "unrecognized RTE kind: %d",
427 : (int) local_node->rtekind);
428 : break;
429 : }
430 :
431 1568058 : READ_BOOL_FIELD(lateral);
432 1568058 : READ_BOOL_FIELD(inFromCl);
433 1568058 : READ_NODE_FIELD(securityQuals);
434 :
435 1568058 : READ_DONE();
436 : }
437 :
438 : static A_Expr *
439 594386 : _readA_Expr(void)
440 : {
441 594386 : READ_LOCALS(A_Expr);
442 :
443 594386 : token = pg_strtok(&length);
444 :
445 594386 : if (length == 3 && strncmp(token, "ANY", 3) == 0)
446 : {
447 14174 : local_node->kind = AEXPR_OP_ANY;
448 14174 : READ_NODE_FIELD(name);
449 : }
450 580212 : else if (length == 3 && strncmp(token, "ALL", 3) == 0)
451 : {
452 144 : local_node->kind = AEXPR_OP_ALL;
453 144 : READ_NODE_FIELD(name);
454 : }
455 580068 : else if (length == 8 && strncmp(token, "DISTINCT", 8) == 0)
456 : {
457 860 : local_node->kind = AEXPR_DISTINCT;
458 860 : READ_NODE_FIELD(name);
459 : }
460 579208 : else if (length == 12 && strncmp(token, "NOT_DISTINCT", 12) == 0)
461 : {
462 68 : local_node->kind = AEXPR_NOT_DISTINCT;
463 68 : READ_NODE_FIELD(name);
464 : }
465 579140 : else if (length == 6 && strncmp(token, "NULLIF", 6) == 0)
466 : {
467 346 : local_node->kind = AEXPR_NULLIF;
468 346 : READ_NODE_FIELD(name);
469 : }
470 578794 : else if (length == 2 && strncmp(token, "IN", 2) == 0)
471 : {
472 21734 : local_node->kind = AEXPR_IN;
473 21734 : READ_NODE_FIELD(name);
474 : }
475 557060 : else if (length == 4 && strncmp(token, "LIKE", 4) == 0)
476 : {
477 2170 : local_node->kind = AEXPR_LIKE;
478 2170 : READ_NODE_FIELD(name);
479 : }
480 554890 : else if (length == 5 && strncmp(token, "ILIKE", 5) == 0)
481 : {
482 204 : local_node->kind = AEXPR_ILIKE;
483 204 : READ_NODE_FIELD(name);
484 : }
485 554686 : else if (length == 7 && strncmp(token, "SIMILAR", 7) == 0)
486 : {
487 70 : local_node->kind = AEXPR_SIMILAR;
488 70 : READ_NODE_FIELD(name);
489 : }
490 554616 : else if (length == 7 && strncmp(token, "BETWEEN", 7) == 0)
491 : {
492 558 : local_node->kind = AEXPR_BETWEEN;
493 558 : READ_NODE_FIELD(name);
494 : }
495 554058 : else if (length == 11 && strncmp(token, "NOT_BETWEEN", 11) == 0)
496 : {
497 12 : local_node->kind = AEXPR_NOT_BETWEEN;
498 12 : READ_NODE_FIELD(name);
499 : }
500 554046 : else if (length == 11 && strncmp(token, "BETWEEN_SYM", 11) == 0)
501 : {
502 12 : local_node->kind = AEXPR_BETWEEN_SYM;
503 12 : READ_NODE_FIELD(name);
504 : }
505 554034 : else if (length == 15 && strncmp(token, "NOT_BETWEEN_SYM", 15) == 0)
506 : {
507 12 : local_node->kind = AEXPR_NOT_BETWEEN_SYM;
508 12 : READ_NODE_FIELD(name);
509 : }
510 554022 : else if (length == 5 && strncmp(token, ":name", 5) == 0)
511 : {
512 554022 : local_node->kind = AEXPR_OP;
513 554022 : local_node->name = nodeRead(NULL, 0);
514 : }
515 : else
516 0 : elog(ERROR, "unrecognized A_Expr kind: \"%.*s\"", length, token);
517 :
518 594386 : READ_NODE_FIELD(lexpr);
519 594386 : READ_NODE_FIELD(rexpr);
520 594386 : READ_LOCATION_FIELD(location);
521 :
522 594386 : READ_DONE();
523 : }
524 :
525 : static ExtensibleNode *
526 0 : _readExtensibleNode(void)
527 : {
528 : const ExtensibleNodeMethods *methods;
529 : ExtensibleNode *local_node;
530 : const char *extnodename;
531 :
532 : READ_TEMP_LOCALS();
533 :
534 0 : token = pg_strtok(&length); /* skip :extnodename */
535 0 : token = pg_strtok(&length); /* get extnodename */
536 :
537 0 : extnodename = nullable_string(token, length);
538 0 : if (!extnodename)
539 0 : elog(ERROR, "extnodename has to be supplied");
540 0 : methods = GetExtensibleNodeMethods(extnodename, false);
541 :
542 0 : local_node = (ExtensibleNode *) newNode(methods->node_size,
543 : T_ExtensibleNode);
544 0 : local_node->extnodename = extnodename;
545 :
546 : /* deserialize the private fields */
547 0 : methods->nodeRead(local_node);
548 :
549 0 : READ_DONE();
550 : }
551 :
552 :
553 : /*
554 : * parseNodeString
555 : *
556 : * Given a character string representing a node tree, parseNodeString creates
557 : * the internal node structure.
558 : *
559 : * The string to be read must already have been loaded into pg_strtok().
560 : */
561 : Node *
562 44195014 : parseNodeString(void)
563 : {
564 : READ_TEMP_LOCALS();
565 :
566 : /* Guard against stack overflow due to overly complex expressions */
567 44195014 : check_stack_depth();
568 :
569 44195014 : token = pg_strtok(&length);
570 :
571 : #define MATCH(tokname, namelen) \
572 : (length == namelen && memcmp(token, tokname, namelen) == 0)
573 :
574 : #include "readfuncs.switch.c"
575 :
576 0 : elog(ERROR, "badly formatted node string \"%.32s\"...", token);
577 : return NULL; /* keep compiler quiet */
578 : }
579 :
580 :
581 : /*
582 : * readDatum
583 : *
584 : * Given a string representation of a constant, recreate the appropriate
585 : * Datum. The string representation embeds length info, but not byValue,
586 : * so we must be told that.
587 : */
588 : Datum
589 2407522 : readDatum(bool typbyval)
590 : {
591 : Size length,
592 : i;
593 : int tokenLength;
594 : const char *token;
595 : Datum res;
596 : char *s;
597 :
598 : /*
599 : * read the actual length of the value
600 : */
601 2407522 : token = pg_strtok(&tokenLength);
602 2407522 : length = atoui(token);
603 :
604 2407522 : token = pg_strtok(&tokenLength); /* read the '[' */
605 2407522 : if (token == NULL || token[0] != '[')
606 0 : elog(ERROR, "expected \"[\" to start datum, but got \"%s\"; length = %zu",
607 : token ? token : "[NULL]", length);
608 :
609 2407522 : if (typbyval)
610 : {
611 1635912 : if (length > (Size) sizeof(Datum))
612 0 : elog(ERROR, "byval datum but length = %zu", length);
613 1635912 : res = (Datum) 0;
614 1635912 : s = (char *) (&res);
615 14723208 : for (i = 0; i < (Size) sizeof(Datum); i++)
616 : {
617 13087296 : token = pg_strtok(&tokenLength);
618 13087296 : s[i] = (char) atoi(token);
619 : }
620 : }
621 771610 : else if (length <= 0)
622 0 : res = (Datum) NULL;
623 : else
624 : {
625 771610 : s = (char *) palloc(length);
626 65845774 : for (i = 0; i < length; i++)
627 : {
628 65074164 : token = pg_strtok(&tokenLength);
629 65074164 : s[i] = (char) atoi(token);
630 : }
631 771610 : res = PointerGetDatum(s);
632 : }
633 :
634 2407522 : token = pg_strtok(&tokenLength); /* read the ']' */
635 2407522 : if (token == NULL || token[0] != ']')
636 0 : elog(ERROR, "expected \"]\" to end datum, but got \"%s\"; length = %zu",
637 : token ? token : "[NULL]", length);
638 :
639 2407522 : return res;
640 : }
641 :
642 : /*
643 : * common implementation for scalar-array-reading functions
644 : *
645 : * The data format is either "<>" for a NULL pointer (in which case numCols
646 : * is ignored) or "(item item item)" where the number of items must equal
647 : * numCols. The convfunc must be okay with stopping at whitespace or a
648 : * right parenthesis, since pg_strtok won't null-terminate the token.
649 : */
650 : #define READ_SCALAR_ARRAY(fnname, datatype, convfunc) \
651 : datatype * \
652 : fnname(int numCols) \
653 : { \
654 : datatype *vals; \
655 : READ_TEMP_LOCALS(); \
656 : token = pg_strtok(&length); \
657 : if (token == NULL) \
658 : elog(ERROR, "incomplete scalar array"); \
659 : if (length == 0) \
660 : return NULL; /* it was "<>", so return NULL pointer */ \
661 : if (length != 1 || token[0] != '(') \
662 : elog(ERROR, "unrecognized token: \"%.*s\"", length, token); \
663 : vals = (datatype *) palloc(numCols * sizeof(datatype)); \
664 : for (int i = 0; i < numCols; i++) \
665 : { \
666 : token = pg_strtok(&length); \
667 : if (token == NULL || token[0] == ')') \
668 : elog(ERROR, "incomplete scalar array"); \
669 : vals[i] = convfunc(token); \
670 : } \
671 : token = pg_strtok(&length); \
672 : if (token == NULL || length != 1 || token[0] != ')') \
673 : elog(ERROR, "incomplete scalar array"); \
674 : return vals; \
675 : }
676 :
677 : /*
678 : * Note: these functions are exported in nodes.h for possible use by
679 : * extensions, so don't mess too much with their names or API.
680 : */
681 400994 : READ_SCALAR_ARRAY(readAttrNumberCols, int16, atoi)
682 515620 : READ_SCALAR_ARRAY(readOidCols, Oid, atooid)
683 : /* outfuncs.c has writeIndexCols, but we don't yet need that here */
684 : /* READ_SCALAR_ARRAY(readIndexCols, Index, atoui) */
685 22484 : READ_SCALAR_ARRAY(readIntCols, int, atoi)
686 170552 : READ_SCALAR_ARRAY(readBoolCols, bool, strtobool)
|