Line data Source code
1 : /*-------------------------------------------------------------------------
2 : *
3 : * xml.c
4 : * XML data type support.
5 : *
6 : *
7 : * Portions Copyright (c) 1996-2021, PostgreSQL Global Development Group
8 : * Portions Copyright (c) 1994, Regents of the University of California
9 : *
10 : * src/backend/utils/adt/xml.c
11 : *
12 : *-------------------------------------------------------------------------
13 : */
14 :
15 : /*
16 : * Generally, XML type support is only available when libxml use was
17 : * configured during the build. But even if that is not done, the
18 : * type and all the functions are available, but most of them will
19 : * fail. For one thing, this avoids having to manage variant catalog
20 : * installations. But it also has nice effects such as that you can
21 : * dump a database containing XML type data even if the server is not
22 : * linked with libxml. Thus, make sure xml_out() works even if nothing
23 : * else does.
24 : */
25 :
26 : /*
27 : * Notes on memory management:
28 : *
29 : * Sometimes libxml allocates global structures in the hope that it can reuse
30 : * them later on. This makes it impractical to change the xmlMemSetup
31 : * functions on-the-fly; that is likely to lead to trying to pfree() chunks
32 : * allocated with malloc() or vice versa. Since libxml might be used by
33 : * loadable modules, eg libperl, our only safe choices are to change the
34 : * functions at postmaster/backend launch or not at all. Since we'd rather
35 : * not activate libxml in sessions that might never use it, the latter choice
36 : * is the preferred one. However, for debugging purposes it can be awfully
37 : * handy to constrain libxml's allocations to be done in a specific palloc
38 : * context, where they're easy to track. Therefore there is code here that
39 : * can be enabled in debug builds to redirect libxml's allocations into a
40 : * special context LibxmlContext. It's not recommended to turn this on in
41 : * a production build because of the possibility of bad interactions with
42 : * external modules.
43 : */
44 : /* #define USE_LIBXMLCONTEXT */
45 :
46 : #include "postgres.h"
47 :
48 : #ifdef USE_LIBXML
49 : #include <libxml/chvalid.h>
50 : #include <libxml/parser.h>
51 : #include <libxml/parserInternals.h>
52 : #include <libxml/tree.h>
53 : #include <libxml/uri.h>
54 : #include <libxml/xmlerror.h>
55 : #include <libxml/xmlversion.h>
56 : #include <libxml/xmlwriter.h>
57 : #include <libxml/xpath.h>
58 : #include <libxml/xpathInternals.h>
59 :
60 : /*
61 : * We used to check for xmlStructuredErrorContext via a configure test; but
62 : * that doesn't work on Windows, so instead use this grottier method of
63 : * testing the library version number.
64 : */
65 : #if LIBXML_VERSION >= 20704
66 : #define HAVE_XMLSTRUCTUREDERRORCONTEXT 1
67 : #endif
68 : #endif /* USE_LIBXML */
69 :
70 : #include "access/htup_details.h"
71 : #include "access/table.h"
72 : #include "catalog/namespace.h"
73 : #include "catalog/pg_class.h"
74 : #include "catalog/pg_type.h"
75 : #include "commands/dbcommands.h"
76 : #include "executor/spi.h"
77 : #include "executor/tablefunc.h"
78 : #include "fmgr.h"
79 : #include "lib/stringinfo.h"
80 : #include "libpq/pqformat.h"
81 : #include "mb/pg_wchar.h"
82 : #include "miscadmin.h"
83 : #include "nodes/execnodes.h"
84 : #include "nodes/nodeFuncs.h"
85 : #include "utils/array.h"
86 : #include "utils/builtins.h"
87 : #include "utils/date.h"
88 : #include "utils/datetime.h"
89 : #include "utils/lsyscache.h"
90 : #include "utils/memutils.h"
91 : #include "utils/rel.h"
92 : #include "utils/syscache.h"
93 : #include "utils/xml.h"
94 :
95 :
96 : /* GUC variables */
97 : int xmlbinary;
98 : int xmloption;
99 :
100 : #ifdef USE_LIBXML
101 :
102 : /* random number to identify PgXmlErrorContext */
103 : #define ERRCXT_MAGIC 68275028
104 :
105 : struct PgXmlErrorContext
106 : {
107 : int magic;
108 : /* strictness argument passed to pg_xml_init */
109 : PgXmlStrictness strictness;
110 : /* current error status and accumulated message, if any */
111 : bool err_occurred;
112 : StringInfoData err_buf;
113 : /* previous libxml error handling state (saved by pg_xml_init) */
114 : xmlStructuredErrorFunc saved_errfunc;
115 : void *saved_errcxt;
116 : /* previous libxml entity handler (saved by pg_xml_init) */
117 : xmlExternalEntityLoader saved_entityfunc;
118 : };
119 :
120 : static xmlParserInputPtr xmlPgEntityLoader(const char *URL, const char *ID,
121 : xmlParserCtxtPtr ctxt);
122 : static void xml_errorHandler(void *data, xmlErrorPtr error);
123 : static void xml_ereport_by_code(int level, int sqlcode,
124 : const char *msg, int errcode);
125 : static void chopStringInfoNewlines(StringInfo str);
126 : static void appendStringInfoLineSeparator(StringInfo str);
127 :
128 : #ifdef USE_LIBXMLCONTEXT
129 :
130 : static MemoryContext LibxmlContext = NULL;
131 :
132 : static void xml_memory_init(void);
133 : static void *xml_palloc(size_t size);
134 : static void *xml_repalloc(void *ptr, size_t size);
135 : static void xml_pfree(void *ptr);
136 : static char *xml_pstrdup(const char *string);
137 : #endif /* USE_LIBXMLCONTEXT */
138 :
139 : static xmlChar *xml_text2xmlChar(text *in);
140 : static int parse_xml_decl(const xmlChar *str, size_t *lenp,
141 : xmlChar **version, xmlChar **encoding, int *standalone);
142 : static bool print_xml_decl(StringInfo buf, const xmlChar *version,
143 : pg_enc encoding, int standalone);
144 : static bool xml_doctype_in_content(const xmlChar *str);
145 : static xmlDocPtr xml_parse(text *data, XmlOptionType xmloption_arg,
146 : bool preserve_whitespace, int encoding);
147 : static text *xml_xmlnodetoxmltype(xmlNodePtr cur, PgXmlErrorContext *xmlerrcxt);
148 : static int xml_xpathobjtoxmlarray(xmlXPathObjectPtr xpathobj,
149 : ArrayBuildState *astate,
150 : PgXmlErrorContext *xmlerrcxt);
151 : static xmlChar *pg_xmlCharStrndup(const char *str, size_t len);
152 : #endif /* USE_LIBXML */
153 :
154 : static void xmldata_root_element_start(StringInfo result, const char *eltname,
155 : const char *xmlschema, const char *targetns,
156 : bool top_level);
157 : static void xmldata_root_element_end(StringInfo result, const char *eltname);
158 : static StringInfo query_to_xml_internal(const char *query, char *tablename,
159 : const char *xmlschema, bool nulls, bool tableforest,
160 : const char *targetns, bool top_level);
161 : static const char *map_sql_table_to_xmlschema(TupleDesc tupdesc, Oid relid,
162 : bool nulls, bool tableforest, const char *targetns);
163 : static const char *map_sql_schema_to_xmlschema_types(Oid nspid,
164 : List *relid_list, bool nulls,
165 : bool tableforest, const char *targetns);
166 : static const char *map_sql_catalog_to_xmlschema_types(List *nspid_list,
167 : bool nulls, bool tableforest,
168 : const char *targetns);
169 : static const char *map_sql_type_to_xml_name(Oid typeoid, int typmod);
170 : static const char *map_sql_typecoll_to_xmlschema_types(List *tupdesc_list);
171 : static const char *map_sql_type_to_xmlschema_type(Oid typeoid, int typmod);
172 : static void SPI_sql_row_to_xmlelement(uint64 rownum, StringInfo result,
173 : char *tablename, bool nulls, bool tableforest,
174 : const char *targetns, bool top_level);
175 :
176 : /* XMLTABLE support */
177 : #ifdef USE_LIBXML
178 : /* random number to identify XmlTableContext */
179 : #define XMLTABLE_CONTEXT_MAGIC 46922182
180 : typedef struct XmlTableBuilderData
181 : {
182 : int magic;
183 : int natts;
184 : long int row_count;
185 : PgXmlErrorContext *xmlerrcxt;
186 : xmlParserCtxtPtr ctxt;
187 : xmlDocPtr doc;
188 : xmlXPathContextPtr xpathcxt;
189 : xmlXPathCompExprPtr xpathcomp;
190 : xmlXPathObjectPtr xpathobj;
191 : xmlXPathCompExprPtr *xpathscomp;
192 : } XmlTableBuilderData;
193 : #endif
194 :
195 : static void XmlTableInitOpaque(struct TableFuncScanState *state, int natts);
196 : static void XmlTableSetDocument(struct TableFuncScanState *state, Datum value);
197 : static void XmlTableSetNamespace(struct TableFuncScanState *state, const char *name,
198 : const char *uri);
199 : static void XmlTableSetRowFilter(struct TableFuncScanState *state, const char *path);
200 : static void XmlTableSetColumnFilter(struct TableFuncScanState *state,
201 : const char *path, int colnum);
202 : static bool XmlTableFetchRow(struct TableFuncScanState *state);
203 : static Datum XmlTableGetValue(struct TableFuncScanState *state, int colnum,
204 : Oid typid, int32 typmod, bool *isnull);
205 : static void XmlTableDestroyOpaque(struct TableFuncScanState *state);
206 :
207 : const TableFuncRoutine XmlTableRoutine =
208 : {
209 : XmlTableInitOpaque,
210 : XmlTableSetDocument,
211 : XmlTableSetNamespace,
212 : XmlTableSetRowFilter,
213 : XmlTableSetColumnFilter,
214 : XmlTableFetchRow,
215 : XmlTableGetValue,
216 : XmlTableDestroyOpaque
217 : };
218 :
219 : #define NO_XML_SUPPORT() \
220 : ereport(ERROR, \
221 : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED), \
222 : errmsg("unsupported XML feature"), \
223 : errdetail("This functionality requires the server to be built with libxml support."), \
224 : errhint("You need to rebuild PostgreSQL using --with-libxml.")))
225 :
226 :
227 : /* from SQL/XML:2008 section 4.9 */
228 : #define NAMESPACE_XSD "http://www.w3.org/2001/XMLSchema"
229 : #define NAMESPACE_XSI "http://www.w3.org/2001/XMLSchema-instance"
230 : #define NAMESPACE_SQLXML "http://standards.iso.org/iso/9075/2003/sqlxml"
231 :
232 :
233 : #ifdef USE_LIBXML
234 :
235 : static int
236 0 : xmlChar_to_encoding(const xmlChar *encoding_name)
237 : {
238 0 : int encoding = pg_char_to_encoding((const char *) encoding_name);
239 :
240 0 : if (encoding < 0)
241 0 : ereport(ERROR,
242 : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
243 : errmsg("invalid encoding name \"%s\"",
244 : (const char *) encoding_name)));
245 0 : return encoding;
246 : }
247 : #endif
248 :
249 :
250 : /*
251 : * xml_in uses a plain C string to VARDATA conversion, so for the time being
252 : * we use the conversion function for the text datatype.
253 : *
254 : * This is only acceptable so long as xmltype and text use the same
255 : * representation.
256 : */
257 : Datum
258 438 : xml_in(PG_FUNCTION_ARGS)
259 : {
260 : #ifdef USE_LIBXML
261 438 : char *s = PG_GETARG_CSTRING(0);
262 : xmltype *vardata;
263 : xmlDocPtr doc;
264 :
265 438 : vardata = (xmltype *) cstring_to_text(s);
266 :
267 : /*
268 : * Parse the data to check if it is well-formed XML data. Assume that
269 : * ERROR occurred if parsing failed.
270 : */
271 438 : doc = xml_parse(vardata, xmloption, true, GetDatabaseEncoding());
272 406 : xmlFreeDoc(doc);
273 :
274 406 : PG_RETURN_XML_P(vardata);
275 : #else
276 : NO_XML_SUPPORT();
277 : return 0;
278 : #endif
279 : }
280 :
281 :
282 : #define PG_XML_DEFAULT_VERSION "1.0"
283 :
284 :
285 : /*
286 : * xml_out_internal uses a plain VARDATA to C string conversion, so for the
287 : * time being we use the conversion function for the text datatype.
288 : *
289 : * This is only acceptable so long as xmltype and text use the same
290 : * representation.
291 : */
292 : static char *
293 14586 : xml_out_internal(xmltype *x, pg_enc target_encoding)
294 : {
295 14586 : char *str = text_to_cstring((text *) x);
296 :
297 : #ifdef USE_LIBXML
298 14586 : size_t len = strlen(str);
299 : xmlChar *version;
300 : int standalone;
301 : int res_code;
302 :
303 14586 : if ((res_code = parse_xml_decl((xmlChar *) str,
304 : &len, &version, NULL, &standalone)) == 0)
305 : {
306 : StringInfoData buf;
307 :
308 14586 : initStringInfo(&buf);
309 :
310 14586 : if (!print_xml_decl(&buf, version, target_encoding, standalone))
311 : {
312 : /*
313 : * If we are not going to produce an XML declaration, eat a single
314 : * newline in the original string to prevent empty first lines in
315 : * the output.
316 : */
317 14554 : if (*(str + len) == '\n')
318 4 : len += 1;
319 : }
320 14586 : appendStringInfoString(&buf, str + len);
321 :
322 14586 : pfree(str);
323 :
324 14586 : return buf.data;
325 : }
326 :
327 0 : xml_ereport_by_code(WARNING, ERRCODE_INTERNAL_ERROR,
328 : "could not parse XML declaration in stored value",
329 : res_code);
330 : #endif
331 0 : return str;
332 : }
333 :
334 :
335 : Datum
336 14410 : xml_out(PG_FUNCTION_ARGS)
337 : {
338 14410 : xmltype *x = PG_GETARG_XML_P(0);
339 :
340 : /*
341 : * xml_out removes the encoding property in all cases. This is because we
342 : * cannot control from here whether the datum will be converted to a
343 : * different client encoding, so we'd do more harm than good by including
344 : * it.
345 : */
346 14410 : PG_RETURN_CSTRING(xml_out_internal(x, 0));
347 : }
348 :
349 :
350 : Datum
351 0 : xml_recv(PG_FUNCTION_ARGS)
352 : {
353 : #ifdef USE_LIBXML
354 0 : StringInfo buf = (StringInfo) PG_GETARG_POINTER(0);
355 : xmltype *result;
356 : char *str;
357 : char *newstr;
358 : int nbytes;
359 : xmlDocPtr doc;
360 0 : xmlChar *encodingStr = NULL;
361 : int encoding;
362 :
363 : /*
364 : * Read the data in raw format. We don't know yet what the encoding is, as
365 : * that information is embedded in the xml declaration; so we have to
366 : * parse that before converting to server encoding.
367 : */
368 0 : nbytes = buf->len - buf->cursor;
369 0 : str = (char *) pq_getmsgbytes(buf, nbytes);
370 :
371 : /*
372 : * We need a null-terminated string to pass to parse_xml_decl(). Rather
373 : * than make a separate copy, make the temporary result one byte bigger
374 : * than it needs to be.
375 : */
376 0 : result = palloc(nbytes + 1 + VARHDRSZ);
377 0 : SET_VARSIZE(result, nbytes + VARHDRSZ);
378 0 : memcpy(VARDATA(result), str, nbytes);
379 0 : str = VARDATA(result);
380 0 : str[nbytes] = '\0';
381 :
382 0 : parse_xml_decl((const xmlChar *) str, NULL, NULL, &encodingStr, NULL);
383 :
384 : /*
385 : * If encoding wasn't explicitly specified in the XML header, treat it as
386 : * UTF-8, as that's the default in XML. This is different from xml_in(),
387 : * where the input has to go through the normal client to server encoding
388 : * conversion.
389 : */
390 0 : encoding = encodingStr ? xmlChar_to_encoding(encodingStr) : PG_UTF8;
391 :
392 : /*
393 : * Parse the data to check if it is well-formed XML data. Assume that
394 : * xml_parse will throw ERROR if not.
395 : */
396 0 : doc = xml_parse(result, xmloption, true, encoding);
397 0 : xmlFreeDoc(doc);
398 :
399 : /* Now that we know what we're dealing with, convert to server encoding */
400 0 : newstr = pg_any_to_server(str, nbytes, encoding);
401 :
402 0 : if (newstr != str)
403 : {
404 0 : pfree(result);
405 0 : result = (xmltype *) cstring_to_text(newstr);
406 0 : pfree(newstr);
407 : }
408 :
409 0 : PG_RETURN_XML_P(result);
410 : #else
411 : NO_XML_SUPPORT();
412 : return 0;
413 : #endif
414 : }
415 :
416 :
417 : Datum
418 0 : xml_send(PG_FUNCTION_ARGS)
419 : {
420 0 : xmltype *x = PG_GETARG_XML_P(0);
421 : char *outval;
422 : StringInfoData buf;
423 :
424 : /*
425 : * xml_out_internal doesn't convert the encoding, it just prints the right
426 : * declaration. pq_sendtext will do the conversion.
427 : */
428 0 : outval = xml_out_internal(x, pg_get_client_encoding());
429 :
430 0 : pq_begintypsend(&buf);
431 0 : pq_sendtext(&buf, outval, strlen(outval));
432 0 : pfree(outval);
433 0 : PG_RETURN_BYTEA_P(pq_endtypsend(&buf));
434 : }
435 :
436 :
437 : #ifdef USE_LIBXML
438 : static void
439 108 : appendStringInfoText(StringInfo str, const text *t)
440 : {
441 108 : appendBinaryStringInfo(str, VARDATA_ANY(t), VARSIZE_ANY_EXHDR(t));
442 108 : }
443 : #endif
444 :
445 :
446 : static xmltype *
447 13946 : stringinfo_to_xmltype(StringInfo buf)
448 : {
449 13946 : return (xmltype *) cstring_to_text_with_len(buf->data, buf->len);
450 : }
451 :
452 :
453 : static xmltype *
454 52 : cstring_to_xmltype(const char *string)
455 : {
456 52 : return (xmltype *) cstring_to_text(string);
457 : }
458 :
459 :
460 : #ifdef USE_LIBXML
461 : static xmltype *
462 13982 : xmlBuffer_to_xmltype(xmlBufferPtr buf)
463 : {
464 13982 : return (xmltype *) cstring_to_text_with_len((const char *) xmlBufferContent(buf),
465 : xmlBufferLength(buf));
466 : }
467 : #endif
468 :
469 :
470 : Datum
471 28 : xmlcomment(PG_FUNCTION_ARGS)
472 : {
473 : #ifdef USE_LIBXML
474 28 : text *arg = PG_GETARG_TEXT_PP(0);
475 28 : char *argdata = VARDATA_ANY(arg);
476 28 : int len = VARSIZE_ANY_EXHDR(arg);
477 : StringInfoData buf;
478 : int i;
479 :
480 : /* check for "--" in string or "-" at the end */
481 120 : for (i = 1; i < len; i++)
482 : {
483 96 : if (argdata[i] == '-' && argdata[i - 1] == '-')
484 4 : ereport(ERROR,
485 : (errcode(ERRCODE_INVALID_XML_COMMENT),
486 : errmsg("invalid XML comment")));
487 : }
488 24 : if (len > 0 && argdata[len - 1] == '-')
489 4 : ereport(ERROR,
490 : (errcode(ERRCODE_INVALID_XML_COMMENT),
491 : errmsg("invalid XML comment")));
492 :
493 20 : initStringInfo(&buf);
494 20 : appendStringInfoString(&buf, "<!--");
495 20 : appendStringInfoText(&buf, arg);
496 20 : appendStringInfoString(&buf, "-->");
497 :
498 20 : PG_RETURN_XML_P(stringinfo_to_xmltype(&buf));
499 : #else
500 : NO_XML_SUPPORT();
501 : return 0;
502 : #endif
503 : }
504 :
505 :
506 :
507 : /*
508 : * TODO: xmlconcat needs to merge the notations and unparsed entities
509 : * of the argument values. Not very important in practice, though.
510 : */
511 : xmltype *
512 13778 : xmlconcat(List *args)
513 : {
514 : #ifdef USE_LIBXML
515 13778 : int global_standalone = 1;
516 13778 : xmlChar *global_version = NULL;
517 13778 : bool global_version_no_value = false;
518 : StringInfoData buf;
519 : ListCell *v;
520 :
521 13778 : initStringInfo(&buf);
522 41338 : foreach(v, args)
523 : {
524 27560 : xmltype *x = DatumGetXmlP(PointerGetDatum(lfirst(v)));
525 : size_t len;
526 : xmlChar *version;
527 : int standalone;
528 : char *str;
529 :
530 27560 : len = VARSIZE(x) - VARHDRSZ;
531 27560 : str = text_to_cstring((text *) x);
532 :
533 27560 : parse_xml_decl((xmlChar *) str, &len, &version, NULL, &standalone);
534 :
535 27560 : if (standalone == 0 && global_standalone == 1)
536 0 : global_standalone = 0;
537 27560 : if (standalone < 0)
538 27552 : global_standalone = -1;
539 :
540 27560 : if (!version)
541 27548 : global_version_no_value = true;
542 12 : else if (!global_version)
543 8 : global_version = version;
544 4 : else if (xmlStrcmp(version, global_version) != 0)
545 0 : global_version_no_value = true;
546 :
547 27560 : appendStringInfoString(&buf, str + len);
548 27560 : pfree(str);
549 : }
550 :
551 13778 : if (!global_version_no_value || global_standalone >= 0)
552 : {
553 : StringInfoData buf2;
554 :
555 4 : initStringInfo(&buf2);
556 :
557 4 : print_xml_decl(&buf2,
558 4 : (!global_version_no_value) ? global_version : NULL,
559 : 0,
560 : global_standalone);
561 :
562 4 : appendBinaryStringInfo(&buf2, buf.data, buf.len);
563 4 : buf = buf2;
564 : }
565 :
566 13778 : return stringinfo_to_xmltype(&buf);
567 : #else
568 : NO_XML_SUPPORT();
569 : return NULL;
570 : #endif
571 : }
572 :
573 :
574 : /*
575 : * XMLAGG support
576 : */
577 : Datum
578 13762 : xmlconcat2(PG_FUNCTION_ARGS)
579 : {
580 13762 : if (PG_ARGISNULL(0))
581 : {
582 12 : if (PG_ARGISNULL(1))
583 0 : PG_RETURN_NULL();
584 : else
585 12 : PG_RETURN_XML_P(PG_GETARG_XML_P(1));
586 : }
587 13750 : else if (PG_ARGISNULL(1))
588 0 : PG_RETURN_XML_P(PG_GETARG_XML_P(0));
589 : else
590 13750 : PG_RETURN_XML_P(xmlconcat(list_make2(PG_GETARG_XML_P(0),
591 : PG_GETARG_XML_P(1))));
592 : }
593 :
594 :
595 : Datum
596 4 : texttoxml(PG_FUNCTION_ARGS)
597 : {
598 4 : text *data = PG_GETARG_TEXT_PP(0);
599 :
600 4 : PG_RETURN_XML_P(xmlparse(data, xmloption, true));
601 : }
602 :
603 :
604 : Datum
605 0 : xmltotext(PG_FUNCTION_ARGS)
606 : {
607 0 : xmltype *data = PG_GETARG_XML_P(0);
608 :
609 : /* It's actually binary compatible. */
610 0 : PG_RETURN_TEXT_P((text *) data);
611 : }
612 :
613 :
614 : text *
615 16 : xmltotext_with_xmloption(xmltype *data, XmlOptionType xmloption_arg)
616 : {
617 16 : if (xmloption_arg == XMLOPTION_DOCUMENT && !xml_is_document(data))
618 4 : ereport(ERROR,
619 : (errcode(ERRCODE_NOT_AN_XML_DOCUMENT),
620 : errmsg("not an XML document")));
621 :
622 : /* It's actually binary compatible, save for the above check. */
623 12 : return (text *) data;
624 : }
625 :
626 :
627 : xmltype *
628 13866 : xmlelement(XmlExpr *xexpr,
629 : Datum *named_argvalue, bool *named_argnull,
630 : Datum *argvalue, bool *argnull)
631 : {
632 : #ifdef USE_LIBXML
633 : xmltype *result;
634 : List *named_arg_strings;
635 : List *arg_strings;
636 : int i;
637 : ListCell *arg;
638 : ListCell *narg;
639 : PgXmlErrorContext *xmlerrcxt;
640 13866 : volatile xmlBufferPtr buf = NULL;
641 13866 : volatile xmlTextWriterPtr writer = NULL;
642 :
643 : /*
644 : * All arguments are already evaluated, and their values are passed in the
645 : * named_argvalue/named_argnull or argvalue/argnull arrays. This avoids
646 : * issues if one of the arguments involves a call to some other function
647 : * or subsystem that wants to use libxml on its own terms. We examine the
648 : * original XmlExpr to identify the numbers and types of the arguments.
649 : */
650 13866 : named_arg_strings = NIL;
651 13866 : i = 0;
652 13898 : foreach(arg, xexpr->named_args)
653 : {
654 36 : Expr *e = (Expr *) lfirst(arg);
655 : char *str;
656 :
657 36 : if (named_argnull[i])
658 0 : str = NULL;
659 : else
660 36 : str = map_sql_value_to_xml_value(named_argvalue[i],
661 : exprType((Node *) e),
662 : false);
663 32 : named_arg_strings = lappend(named_arg_strings, str);
664 32 : i++;
665 : }
666 :
667 13862 : arg_strings = NIL;
668 13862 : i = 0;
669 27708 : foreach(arg, xexpr->args)
670 : {
671 13846 : Expr *e = (Expr *) lfirst(arg);
672 : char *str;
673 :
674 : /* here we can just forget NULL elements immediately */
675 13846 : if (!argnull[i])
676 : {
677 13846 : str = map_sql_value_to_xml_value(argvalue[i],
678 : exprType((Node *) e),
679 : true);
680 13846 : arg_strings = lappend(arg_strings, str);
681 : }
682 13846 : i++;
683 : }
684 :
685 13862 : xmlerrcxt = pg_xml_init(PG_XML_STRICTNESS_ALL);
686 :
687 13862 : PG_TRY();
688 : {
689 13862 : buf = xmlBufferCreate();
690 13862 : if (buf == NULL || xmlerrcxt->err_occurred)
691 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
692 : "could not allocate xmlBuffer");
693 13862 : writer = xmlNewTextWriterMemory(buf, 0);
694 13862 : if (writer == NULL || xmlerrcxt->err_occurred)
695 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
696 : "could not allocate xmlTextWriter");
697 :
698 13862 : xmlTextWriterStartElement(writer, (xmlChar *) xexpr->name);
699 :
700 13894 : forboth(arg, named_arg_strings, narg, xexpr->arg_names)
701 : {
702 32 : char *str = (char *) lfirst(arg);
703 32 : char *argname = strVal(lfirst(narg));
704 :
705 32 : if (str)
706 32 : xmlTextWriterWriteAttribute(writer,
707 : (xmlChar *) argname,
708 : (xmlChar *) str);
709 : }
710 :
711 27708 : foreach(arg, arg_strings)
712 : {
713 13846 : char *str = (char *) lfirst(arg);
714 :
715 13846 : xmlTextWriterWriteRaw(writer, (xmlChar *) str);
716 : }
717 :
718 13862 : xmlTextWriterEndElement(writer);
719 :
720 : /* we MUST do this now to flush data out to the buffer ... */
721 13862 : xmlFreeTextWriter(writer);
722 13862 : writer = NULL;
723 :
724 13862 : result = xmlBuffer_to_xmltype(buf);
725 : }
726 0 : PG_CATCH();
727 : {
728 0 : if (writer)
729 0 : xmlFreeTextWriter(writer);
730 0 : if (buf)
731 0 : xmlBufferFree(buf);
732 :
733 0 : pg_xml_done(xmlerrcxt, true);
734 :
735 0 : PG_RE_THROW();
736 : }
737 13862 : PG_END_TRY();
738 :
739 13862 : xmlBufferFree(buf);
740 :
741 13862 : pg_xml_done(xmlerrcxt, false);
742 :
743 13862 : return result;
744 : #else
745 : NO_XML_SUPPORT();
746 : return NULL;
747 : #endif
748 : }
749 :
750 :
751 : xmltype *
752 92 : xmlparse(text *data, XmlOptionType xmloption_arg, bool preserve_whitespace)
753 : {
754 : #ifdef USE_LIBXML
755 : xmlDocPtr doc;
756 :
757 92 : doc = xml_parse(data, xmloption_arg, preserve_whitespace,
758 : GetDatabaseEncoding());
759 60 : xmlFreeDoc(doc);
760 :
761 60 : return (xmltype *) data;
762 : #else
763 : NO_XML_SUPPORT();
764 : return NULL;
765 : #endif
766 : }
767 :
768 :
769 : xmltype *
770 48 : xmlpi(const char *target, text *arg, bool arg_is_null, bool *result_is_null)
771 : {
772 : #ifdef USE_LIBXML
773 : xmltype *result;
774 : StringInfoData buf;
775 :
776 48 : if (pg_strcasecmp(target, "xml") == 0)
777 8 : ereport(ERROR,
778 : (errcode(ERRCODE_SYNTAX_ERROR), /* really */
779 : errmsg("invalid XML processing instruction"),
780 : errdetail("XML processing instruction target name cannot be \"%s\".", target)));
781 :
782 : /*
783 : * Following the SQL standard, the null check comes after the syntax check
784 : * above.
785 : */
786 40 : *result_is_null = arg_is_null;
787 40 : if (*result_is_null)
788 8 : return NULL;
789 :
790 32 : initStringInfo(&buf);
791 :
792 32 : appendStringInfo(&buf, "<?%s", target);
793 :
794 32 : if (arg != NULL)
795 : {
796 : char *string;
797 :
798 16 : string = text_to_cstring(arg);
799 16 : if (strstr(string, "?>") != NULL)
800 4 : ereport(ERROR,
801 : (errcode(ERRCODE_INVALID_XML_PROCESSING_INSTRUCTION),
802 : errmsg("invalid XML processing instruction"),
803 : errdetail("XML processing instruction cannot contain \"?>\".")));
804 :
805 12 : appendStringInfoChar(&buf, ' ');
806 12 : appendStringInfoString(&buf, string + strspn(string, " "));
807 12 : pfree(string);
808 : }
809 28 : appendStringInfoString(&buf, "?>");
810 :
811 28 : result = stringinfo_to_xmltype(&buf);
812 28 : pfree(buf.data);
813 28 : return result;
814 : #else
815 : NO_XML_SUPPORT();
816 : return NULL;
817 : #endif
818 : }
819 :
820 :
821 : xmltype *
822 40 : xmlroot(xmltype *data, text *version, int standalone)
823 : {
824 : #ifdef USE_LIBXML
825 : char *str;
826 : size_t len;
827 : xmlChar *orig_version;
828 : int orig_standalone;
829 : StringInfoData buf;
830 :
831 40 : len = VARSIZE(data) - VARHDRSZ;
832 40 : str = text_to_cstring((text *) data);
833 :
834 40 : parse_xml_decl((xmlChar *) str, &len, &orig_version, NULL, &orig_standalone);
835 :
836 40 : if (version)
837 16 : orig_version = xml_text2xmlChar(version);
838 : else
839 24 : orig_version = NULL;
840 :
841 40 : switch (standalone)
842 : {
843 12 : case XML_STANDALONE_YES:
844 12 : orig_standalone = 1;
845 12 : break;
846 8 : case XML_STANDALONE_NO:
847 8 : orig_standalone = 0;
848 8 : break;
849 8 : case XML_STANDALONE_NO_VALUE:
850 8 : orig_standalone = -1;
851 8 : break;
852 12 : case XML_STANDALONE_OMITTED:
853 : /* leave original value */
854 12 : break;
855 : }
856 :
857 40 : initStringInfo(&buf);
858 40 : print_xml_decl(&buf, orig_version, 0, orig_standalone);
859 40 : appendStringInfoString(&buf, str + len);
860 :
861 40 : return stringinfo_to_xmltype(&buf);
862 : #else
863 : NO_XML_SUPPORT();
864 : return NULL;
865 : #endif
866 : }
867 :
868 :
869 : /*
870 : * Validate document (given as string) against DTD (given as external link)
871 : *
872 : * This has been removed because it is a security hole: unprivileged users
873 : * should not be able to use Postgres to fetch arbitrary external files,
874 : * which unfortunately is exactly what libxml is willing to do with the DTD
875 : * parameter.
876 : */
877 : Datum
878 0 : xmlvalidate(PG_FUNCTION_ARGS)
879 : {
880 0 : ereport(ERROR,
881 : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
882 : errmsg("xmlvalidate is not implemented")));
883 : return 0;
884 : }
885 :
886 :
887 : bool
888 20 : xml_is_document(xmltype *arg)
889 : {
890 : #ifdef USE_LIBXML
891 : bool result;
892 20 : volatile xmlDocPtr doc = NULL;
893 20 : MemoryContext ccxt = CurrentMemoryContext;
894 :
895 : /* We want to catch ereport(INVALID_XML_DOCUMENT) and return false */
896 20 : PG_TRY();
897 : {
898 20 : doc = xml_parse((text *) arg, XMLOPTION_DOCUMENT, true,
899 : GetDatabaseEncoding());
900 8 : result = true;
901 : }
902 12 : PG_CATCH();
903 : {
904 : ErrorData *errdata;
905 : MemoryContext ecxt;
906 :
907 12 : ecxt = MemoryContextSwitchTo(ccxt);
908 12 : errdata = CopyErrorData();
909 12 : if (errdata->sqlerrcode == ERRCODE_INVALID_XML_DOCUMENT)
910 : {
911 12 : FlushErrorState();
912 12 : result = false;
913 : }
914 : else
915 : {
916 0 : MemoryContextSwitchTo(ecxt);
917 0 : PG_RE_THROW();
918 : }
919 : }
920 20 : PG_END_TRY();
921 :
922 20 : if (doc)
923 8 : xmlFreeDoc(doc);
924 :
925 20 : return result;
926 : #else /* not USE_LIBXML */
927 : NO_XML_SUPPORT();
928 : return false;
929 : #endif /* not USE_LIBXML */
930 : }
931 :
932 :
933 : #ifdef USE_LIBXML
934 :
935 : /*
936 : * pg_xml_init_library --- set up for use of libxml
937 : *
938 : * This should be called by each function that is about to use libxml
939 : * facilities but doesn't require error handling. It initializes libxml
940 : * and verifies compatibility with the loaded libxml version. These are
941 : * once-per-session activities.
942 : *
943 : * TODO: xmlChar is utf8-char, make proper tuning (initdb with enc!=utf8 and
944 : * check)
945 : */
946 : void
947 58104 : pg_xml_init_library(void)
948 : {
949 : static bool first_time = true;
950 :
951 58104 : if (first_time)
952 : {
953 : /* Stuff we need do only once per session */
954 :
955 : /*
956 : * Currently, we have no pure UTF-8 support for internals -- check if
957 : * we can work.
958 : */
959 : if (sizeof(char) != sizeof(xmlChar))
960 : ereport(ERROR,
961 : (errmsg("could not initialize XML library"),
962 : errdetail("libxml2 has incompatible char type: sizeof(char)=%u, sizeof(xmlChar)=%u.",
963 : (int) sizeof(char), (int) sizeof(xmlChar))));
964 :
965 : #ifdef USE_LIBXMLCONTEXT
966 : /* Set up libxml's memory allocation our way */
967 : xml_memory_init();
968 : #endif
969 :
970 : /* Check library compatibility */
971 18 : LIBXML_TEST_VERSION;
972 :
973 18 : first_time = false;
974 : }
975 58104 : }
976 :
977 : /*
978 : * pg_xml_init --- set up for use of libxml and register an error handler
979 : *
980 : * This should be called by each function that is about to use libxml
981 : * facilities and requires error handling. It initializes libxml with
982 : * pg_xml_init_library() and establishes our libxml error handler.
983 : *
984 : * strictness determines which errors are reported and which are ignored.
985 : *
986 : * Calls to this function MUST be followed by a PG_TRY block that guarantees
987 : * that pg_xml_done() is called during either normal or error exit.
988 : *
989 : * This is exported for use by contrib/xml2, as well as other code that might
990 : * wish to share use of this module's libxml error handler.
991 : */
992 : PgXmlErrorContext *
993 15068 : pg_xml_init(PgXmlStrictness strictness)
994 : {
995 : PgXmlErrorContext *errcxt;
996 : void *new_errcxt;
997 :
998 : /* Do one-time setup if needed */
999 15068 : pg_xml_init_library();
1000 :
1001 : /* Create error handling context structure */
1002 15068 : errcxt = (PgXmlErrorContext *) palloc(sizeof(PgXmlErrorContext));
1003 15068 : errcxt->magic = ERRCXT_MAGIC;
1004 15068 : errcxt->strictness = strictness;
1005 15068 : errcxt->err_occurred = false;
1006 15068 : initStringInfo(&errcxt->err_buf);
1007 :
1008 : /*
1009 : * Save original error handler and install ours. libxml originally didn't
1010 : * distinguish between the contexts for generic and for structured error
1011 : * handlers. If we're using an old libxml version, we must thus save the
1012 : * generic error context, even though we're using a structured error
1013 : * handler.
1014 : */
1015 15068 : errcxt->saved_errfunc = xmlStructuredError;
1016 :
1017 : #ifdef HAVE_XMLSTRUCTUREDERRORCONTEXT
1018 15068 : errcxt->saved_errcxt = xmlStructuredErrorContext;
1019 : #else
1020 : errcxt->saved_errcxt = xmlGenericErrorContext;
1021 : #endif
1022 :
1023 15068 : xmlSetStructuredErrorFunc((void *) errcxt, xml_errorHandler);
1024 :
1025 : /*
1026 : * Verify that xmlSetStructuredErrorFunc set the context variable we
1027 : * expected it to. If not, the error context pointer we just saved is not
1028 : * the correct thing to restore, and since that leaves us without a way to
1029 : * restore the context in pg_xml_done, we must fail.
1030 : *
1031 : * The only known situation in which this test fails is if we compile with
1032 : * headers from a libxml2 that doesn't track the structured error context
1033 : * separately (< 2.7.4), but at runtime use a version that does, or vice
1034 : * versa. The libxml2 authors did not treat that change as constituting
1035 : * an ABI break, so the LIBXML_TEST_VERSION test in pg_xml_init_library
1036 : * fails to protect us from this.
1037 : */
1038 :
1039 : #ifdef HAVE_XMLSTRUCTUREDERRORCONTEXT
1040 15068 : new_errcxt = xmlStructuredErrorContext;
1041 : #else
1042 : new_errcxt = xmlGenericErrorContext;
1043 : #endif
1044 :
1045 15068 : if (new_errcxt != (void *) errcxt)
1046 0 : ereport(ERROR,
1047 : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1048 : errmsg("could not set up XML error handler"),
1049 : errhint("This probably indicates that the version of libxml2"
1050 : " being used is not compatible with the libxml2"
1051 : " header files that PostgreSQL was built with.")));
1052 :
1053 : /*
1054 : * Also, install an entity loader to prevent unwanted fetches of external
1055 : * files and URLs.
1056 : */
1057 15068 : errcxt->saved_entityfunc = xmlGetExternalEntityLoader();
1058 15068 : xmlSetExternalEntityLoader(xmlPgEntityLoader);
1059 :
1060 15068 : return errcxt;
1061 : }
1062 :
1063 :
1064 : /*
1065 : * pg_xml_done --- restore previous libxml error handling
1066 : *
1067 : * Resets libxml's global error-handling state to what it was before
1068 : * pg_xml_init() was called.
1069 : *
1070 : * This routine verifies that all pending errors have been dealt with
1071 : * (in assert-enabled builds, anyway).
1072 : */
1073 : void
1074 15068 : pg_xml_done(PgXmlErrorContext *errcxt, bool isError)
1075 : {
1076 : void *cur_errcxt;
1077 :
1078 : /* An assert seems like enough protection here */
1079 : Assert(errcxt->magic == ERRCXT_MAGIC);
1080 :
1081 : /*
1082 : * In a normal exit, there should be no un-handled libxml errors. But we
1083 : * shouldn't try to enforce this during error recovery, since the longjmp
1084 : * could have been thrown before xml_ereport had a chance to run.
1085 : */
1086 : Assert(!errcxt->err_occurred || isError);
1087 :
1088 : /*
1089 : * Check that libxml's global state is correct, warn if not. This is a
1090 : * real test and not an Assert because it has a higher probability of
1091 : * happening.
1092 : */
1093 : #ifdef HAVE_XMLSTRUCTUREDERRORCONTEXT
1094 15068 : cur_errcxt = xmlStructuredErrorContext;
1095 : #else
1096 : cur_errcxt = xmlGenericErrorContext;
1097 : #endif
1098 :
1099 15068 : if (cur_errcxt != (void *) errcxt)
1100 0 : elog(WARNING, "libxml error handling state is out of sync with xml.c");
1101 :
1102 : /* Restore the saved handlers */
1103 15068 : xmlSetStructuredErrorFunc(errcxt->saved_errcxt, errcxt->saved_errfunc);
1104 15068 : xmlSetExternalEntityLoader(errcxt->saved_entityfunc);
1105 :
1106 : /*
1107 : * Mark the struct as invalid, just in case somebody somehow manages to
1108 : * call xml_errorHandler or xml_ereport with it.
1109 : */
1110 15068 : errcxt->magic = 0;
1111 :
1112 : /* Release memory */
1113 15068 : pfree(errcxt->err_buf.data);
1114 15068 : pfree(errcxt);
1115 15068 : }
1116 :
1117 :
1118 : /*
1119 : * pg_xml_error_occurred() --- test the error flag
1120 : */
1121 : bool
1122 0 : pg_xml_error_occurred(PgXmlErrorContext *errcxt)
1123 : {
1124 0 : return errcxt->err_occurred;
1125 : }
1126 :
1127 :
1128 : /*
1129 : * SQL/XML allows storing "XML documents" or "XML content". "XML
1130 : * documents" are specified by the XML specification and are parsed
1131 : * easily by libxml. "XML content" is specified by SQL/XML as the
1132 : * production "XMLDecl? content". But libxml can only parse the
1133 : * "content" part, so we have to parse the XML declaration ourselves
1134 : * to complete this.
1135 : */
1136 :
1137 : #define CHECK_XML_SPACE(p) \
1138 : do { \
1139 : if (!xmlIsBlank_ch(*(p))) \
1140 : return XML_ERR_SPACE_REQUIRED; \
1141 : } while (0)
1142 :
1143 : #define SKIP_XML_SPACE(p) \
1144 : while (xmlIsBlank_ch(*(p))) (p)++
1145 :
1146 : /* Letter | Digit | '.' | '-' | '_' | ':' | CombiningChar | Extender */
1147 : /* Beware of multiple evaluations of argument! */
1148 : #define PG_XMLISNAMECHAR(c) \
1149 : (xmlIsBaseChar_ch(c) || xmlIsIdeographicQ(c) \
1150 : || xmlIsDigit_ch(c) \
1151 : || c == '.' || c == '-' || c == '_' || c == ':' \
1152 : || xmlIsCombiningQ(c) \
1153 : || xmlIsExtender_ch(c))
1154 :
1155 : /* pnstrdup, but deal with xmlChar not char; len is measured in xmlChars */
1156 : static xmlChar *
1157 108 : xml_pnstrdup(const xmlChar *str, size_t len)
1158 : {
1159 : xmlChar *result;
1160 :
1161 108 : result = (xmlChar *) palloc((len + 1) * sizeof(xmlChar));
1162 108 : memcpy(result, str, len * sizeof(xmlChar));
1163 108 : result[len] = 0;
1164 108 : return result;
1165 : }
1166 :
1167 : /* Ditto, except input is char* */
1168 : static xmlChar *
1169 1616 : pg_xmlCharStrndup(const char *str, size_t len)
1170 : {
1171 : xmlChar *result;
1172 :
1173 1616 : result = (xmlChar *) palloc((len + 1) * sizeof(xmlChar));
1174 1616 : memcpy(result, str, len);
1175 1616 : result[len] = '\0';
1176 :
1177 1616 : return result;
1178 : }
1179 :
1180 : /*
1181 : * Copy xmlChar string to PostgreSQL-owned memory, freeing the input.
1182 : *
1183 : * The input xmlChar is freed regardless of success of the copy.
1184 : */
1185 : static char *
1186 69500 : xml_pstrdup_and_free(xmlChar *str)
1187 : {
1188 : char *result;
1189 :
1190 69500 : if (str)
1191 : {
1192 69500 : PG_TRY();
1193 : {
1194 69500 : result = pstrdup((char *) str);
1195 : }
1196 0 : PG_FINALLY();
1197 : {
1198 69500 : xmlFree(str);
1199 : }
1200 69500 : PG_END_TRY();
1201 : }
1202 : else
1203 0 : result = NULL;
1204 :
1205 69500 : return result;
1206 : }
1207 :
1208 : /*
1209 : * str is the null-terminated input string. Remaining arguments are
1210 : * output arguments; each can be NULL if value is not wanted.
1211 : * version and encoding are returned as locally-palloc'd strings.
1212 : * Result is 0 if OK, an error code if not.
1213 : */
1214 : static int
1215 43036 : parse_xml_decl(const xmlChar *str, size_t *lenp,
1216 : xmlChar **version, xmlChar **encoding, int *standalone)
1217 : {
1218 : const xmlChar *p;
1219 : const xmlChar *save_p;
1220 : size_t len;
1221 : int utf8char;
1222 : int utf8len;
1223 :
1224 : /*
1225 : * Only initialize libxml. We don't need error handling here, but we do
1226 : * need to make sure libxml is initialized before calling any of its
1227 : * functions. Note that this is safe (and a no-op) if caller has already
1228 : * done pg_xml_init().
1229 : */
1230 43036 : pg_xml_init_library();
1231 :
1232 : /* Initialize output arguments to "not present" */
1233 43036 : if (version)
1234 42668 : *version = NULL;
1235 43036 : if (encoding)
1236 0 : *encoding = NULL;
1237 43036 : if (standalone)
1238 42668 : *standalone = -1;
1239 :
1240 43036 : p = str;
1241 :
1242 43036 : if (xmlStrncmp(p, (xmlChar *) "<?xml", 5) != 0)
1243 42916 : goto finished;
1244 :
1245 : /*
1246 : * If next char is a name char, it's a PI like <?xml-stylesheet ...?>
1247 : * rather than an XMLDecl, so we have done what we came to do and found no
1248 : * XMLDecl.
1249 : *
1250 : * We need an input length value for xmlGetUTF8Char, but there's no need
1251 : * to count the whole document size, so use strnlen not strlen.
1252 : */
1253 120 : utf8len = strnlen((const char *) (p + 5), MAX_MULTIBYTE_CHAR_LEN);
1254 120 : utf8char = xmlGetUTF8Char(p + 5, &utf8len);
1255 120 : if (PG_XMLISNAMECHAR(utf8char))
1256 8 : goto finished;
1257 :
1258 112 : p += 5;
1259 :
1260 : /* version */
1261 112 : CHECK_XML_SPACE(p);
1262 224 : SKIP_XML_SPACE(p);
1263 112 : if (xmlStrncmp(p, (xmlChar *) "version", 7) != 0)
1264 0 : return XML_ERR_VERSION_MISSING;
1265 112 : p += 7;
1266 112 : SKIP_XML_SPACE(p);
1267 112 : if (*p != '=')
1268 0 : return XML_ERR_VERSION_MISSING;
1269 112 : p += 1;
1270 112 : SKIP_XML_SPACE(p);
1271 :
1272 112 : if (*p == '\'' || *p == '"')
1273 112 : {
1274 : const xmlChar *q;
1275 :
1276 112 : q = xmlStrchr(p + 1, *p);
1277 112 : if (!q)
1278 0 : return XML_ERR_VERSION_MISSING;
1279 :
1280 112 : if (version)
1281 108 : *version = xml_pnstrdup(p + 1, q - p - 1);
1282 112 : p = q + 1;
1283 : }
1284 : else
1285 0 : return XML_ERR_VERSION_MISSING;
1286 :
1287 : /* encoding */
1288 112 : save_p = p;
1289 192 : SKIP_XML_SPACE(p);
1290 112 : if (xmlStrncmp(p, (xmlChar *) "encoding", 8) == 0)
1291 : {
1292 16 : CHECK_XML_SPACE(save_p);
1293 16 : p += 8;
1294 16 : SKIP_XML_SPACE(p);
1295 16 : if (*p != '=')
1296 0 : return XML_ERR_MISSING_ENCODING;
1297 16 : p += 1;
1298 16 : SKIP_XML_SPACE(p);
1299 :
1300 16 : if (*p == '\'' || *p == '"')
1301 16 : {
1302 : const xmlChar *q;
1303 :
1304 16 : q = xmlStrchr(p + 1, *p);
1305 16 : if (!q)
1306 0 : return XML_ERR_MISSING_ENCODING;
1307 :
1308 16 : if (encoding)
1309 0 : *encoding = xml_pnstrdup(p + 1, q - p - 1);
1310 16 : p = q + 1;
1311 : }
1312 : else
1313 0 : return XML_ERR_MISSING_ENCODING;
1314 : }
1315 : else
1316 : {
1317 96 : p = save_p;
1318 : }
1319 :
1320 : /* standalone */
1321 112 : save_p = p;
1322 176 : SKIP_XML_SPACE(p);
1323 112 : if (xmlStrncmp(p, (xmlChar *) "standalone", 10) == 0)
1324 : {
1325 64 : CHECK_XML_SPACE(save_p);
1326 64 : p += 10;
1327 64 : SKIP_XML_SPACE(p);
1328 64 : if (*p != '=')
1329 0 : return XML_ERR_STANDALONE_VALUE;
1330 64 : p += 1;
1331 64 : SKIP_XML_SPACE(p);
1332 128 : if (xmlStrncmp(p, (xmlChar *) "'yes'", 5) == 0 ||
1333 64 : xmlStrncmp(p, (xmlChar *) "\"yes\"", 5) == 0)
1334 : {
1335 40 : if (standalone)
1336 40 : *standalone = 1;
1337 40 : p += 5;
1338 : }
1339 48 : else if (xmlStrncmp(p, (xmlChar *) "'no'", 4) == 0 ||
1340 24 : xmlStrncmp(p, (xmlChar *) "\"no\"", 4) == 0)
1341 : {
1342 24 : if (standalone)
1343 24 : *standalone = 0;
1344 24 : p += 4;
1345 : }
1346 : else
1347 0 : return XML_ERR_STANDALONE_VALUE;
1348 : }
1349 : else
1350 : {
1351 48 : p = save_p;
1352 : }
1353 :
1354 112 : SKIP_XML_SPACE(p);
1355 112 : if (xmlStrncmp(p, (xmlChar *) "?>", 2) != 0)
1356 0 : return XML_ERR_XMLDECL_NOT_FINISHED;
1357 112 : p += 2;
1358 :
1359 43036 : finished:
1360 43036 : len = p - str;
1361 :
1362 46764 : for (p = str; p < str + len; p++)
1363 3728 : if (*p > 127)
1364 0 : return XML_ERR_INVALID_CHAR;
1365 :
1366 43036 : if (lenp)
1367 43036 : *lenp = len;
1368 :
1369 43036 : return XML_ERR_OK;
1370 : }
1371 :
1372 :
1373 : /*
1374 : * Write an XML declaration. On output, we adjust the XML declaration
1375 : * as follows. (These rules are the moral equivalent of the clause
1376 : * "Serialization of an XML value" in the SQL standard.)
1377 : *
1378 : * We try to avoid generating an XML declaration if possible. This is
1379 : * so that you don't get trivial things like xml '<foo/>' resulting in
1380 : * '<?xml version="1.0"?><foo/>', which would surely be annoying. We
1381 : * must provide a declaration if the standalone property is specified
1382 : * or if we include an encoding declaration. If we have a
1383 : * declaration, we must specify a version (XML requires this).
1384 : * Otherwise we only make a declaration if the version is not "1.0",
1385 : * which is the default version specified in SQL:2003.
1386 : */
1387 : static bool
1388 14630 : print_xml_decl(StringInfo buf, const xmlChar *version,
1389 : pg_enc encoding, int standalone)
1390 : {
1391 14630 : if ((version && strcmp((const char *) version, PG_XML_DEFAULT_VERSION) != 0)
1392 14606 : || (encoding && encoding != PG_UTF8)
1393 14606 : || standalone != -1)
1394 : {
1395 64 : appendStringInfoString(buf, "<?xml");
1396 :
1397 64 : if (version)
1398 48 : appendStringInfo(buf, " version=\"%s\"", version);
1399 : else
1400 16 : appendStringInfo(buf, " version=\"%s\"", PG_XML_DEFAULT_VERSION);
1401 :
1402 64 : if (encoding && encoding != PG_UTF8)
1403 : {
1404 : /*
1405 : * XXX might be useful to convert this to IANA names (ISO-8859-1
1406 : * instead of LATIN1 etc.); needs field experience
1407 : */
1408 0 : appendStringInfo(buf, " encoding=\"%s\"",
1409 : pg_encoding_to_char(encoding));
1410 : }
1411 :
1412 64 : if (standalone == 1)
1413 32 : appendStringInfoString(buf, " standalone=\"yes\"");
1414 32 : else if (standalone == 0)
1415 16 : appendStringInfoString(buf, " standalone=\"no\"");
1416 64 : appendStringInfoString(buf, "?>");
1417 :
1418 64 : return true;
1419 : }
1420 : else
1421 14566 : return false;
1422 : }
1423 :
1424 : /*
1425 : * Test whether an input that is to be parsed as CONTENT contains a DTD.
1426 : *
1427 : * The SQL/XML:2003 definition of CONTENT ("XMLDecl? content") is not
1428 : * satisfied by a document with a DTD, which is a bit of a wart, as it means
1429 : * the CONTENT type is not a proper superset of DOCUMENT. SQL/XML:2006 and
1430 : * later fix that, by redefining content with reference to the "more
1431 : * permissive" Document Node of the XQuery/XPath Data Model, such that any
1432 : * DOCUMENT value is indeed also a CONTENT value. That definition is more
1433 : * useful, as CONTENT becomes usable for parsing input of unknown form (think
1434 : * pg_restore).
1435 : *
1436 : * As used below in parse_xml when parsing for CONTENT, libxml does not give
1437 : * us the 2006+ behavior, but only the 2003; it will choke if the input has
1438 : * a DTD. But we can provide the 2006+ definition of CONTENT easily enough,
1439 : * by detecting this case first and simply doing the parse as DOCUMENT.
1440 : *
1441 : * A DTD can be found arbitrarily far in, but that would be a contrived case;
1442 : * it will ordinarily start within a few dozen characters. The only things
1443 : * that can precede it are an XMLDecl (here, the caller will have called
1444 : * parse_xml_decl already), whitespace, comments, and processing instructions.
1445 : * This function need only return true if it sees a valid sequence of such
1446 : * things leading to <!DOCTYPE. It can simply return false in any other
1447 : * cases, including malformed input; that will mean the input gets parsed as
1448 : * CONTENT as originally planned, with libxml reporting any errors.
1449 : *
1450 : * This is only to be called from xml_parse, when pg_xml_init has already
1451 : * been called. The input is already in UTF8 encoding.
1452 : */
1453 : static bool
1454 482 : xml_doctype_in_content(const xmlChar *str)
1455 : {
1456 482 : const xmlChar *p = str;
1457 :
1458 : for (;;)
1459 24 : {
1460 : const xmlChar *e;
1461 :
1462 542 : SKIP_XML_SPACE(p);
1463 506 : if (*p != '<')
1464 100 : return false;
1465 406 : p++;
1466 :
1467 406 : if (*p == '!')
1468 : {
1469 36 : p++;
1470 :
1471 : /* if we see <!DOCTYPE, we can return true */
1472 36 : if (xmlStrncmp(p, (xmlChar *) "DOCTYPE", 7) == 0)
1473 16 : return true;
1474 :
1475 : /* otherwise, if it's not a comment, fail */
1476 20 : if (xmlStrncmp(p, (xmlChar *) "--", 2) != 0)
1477 0 : return false;
1478 : /* find end of comment: find -- and a > must follow */
1479 20 : p = xmlStrstr(p + 2, (xmlChar *) "--");
1480 20 : if (!p || p[2] != '>')
1481 0 : return false;
1482 : /* advance over comment, and keep scanning */
1483 20 : p += 3;
1484 20 : continue;
1485 : }
1486 :
1487 : /* otherwise, if it's not a PI <?target something?>, fail */
1488 370 : if (*p != '?')
1489 366 : return false;
1490 4 : p++;
1491 :
1492 : /* find end of PI (the string ?> is forbidden within a PI) */
1493 4 : e = xmlStrstr(p, (xmlChar *) "?>");
1494 4 : if (!e)
1495 0 : return false;
1496 :
1497 : /* advance over PI, keep scanning */
1498 4 : p = e + 2;
1499 : }
1500 : }
1501 :
1502 :
1503 : /*
1504 : * Convert a C string to XML internal representation
1505 : *
1506 : * Note: it is caller's responsibility to xmlFreeDoc() the result,
1507 : * else a permanent memory leak will ensue!
1508 : *
1509 : * TODO maybe libxml2's xmlreader is better? (do not construct DOM,
1510 : * yet do not use SAX - see xmlreader.c)
1511 : */
1512 : static xmlDocPtr
1513 626 : xml_parse(text *data, XmlOptionType xmloption_arg, bool preserve_whitespace,
1514 : int encoding)
1515 : {
1516 : int32 len;
1517 : xmlChar *string;
1518 : xmlChar *utf8string;
1519 : PgXmlErrorContext *xmlerrcxt;
1520 626 : volatile xmlParserCtxtPtr ctxt = NULL;
1521 626 : volatile xmlDocPtr doc = NULL;
1522 :
1523 626 : len = VARSIZE_ANY_EXHDR(data); /* will be useful later */
1524 626 : string = xml_text2xmlChar(data);
1525 :
1526 626 : utf8string = pg_do_encoding_conversion(string,
1527 : len,
1528 : encoding,
1529 : PG_UTF8);
1530 :
1531 : /* Start up libxml and its parser */
1532 626 : xmlerrcxt = pg_xml_init(PG_XML_STRICTNESS_WELLFORMED);
1533 :
1534 : /* Use a TRY block to ensure we clean up correctly */
1535 626 : PG_TRY();
1536 : {
1537 626 : bool parse_as_document = false;
1538 : int res_code;
1539 626 : size_t count = 0;
1540 626 : xmlChar *version = NULL;
1541 626 : int standalone = 0;
1542 :
1543 626 : xmlInitParser();
1544 :
1545 626 : ctxt = xmlNewParserCtxt();
1546 626 : if (ctxt == NULL || xmlerrcxt->err_occurred)
1547 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
1548 : "could not allocate parser context");
1549 :
1550 : /* Decide whether to parse as document or content */
1551 626 : if (xmloption_arg == XMLOPTION_DOCUMENT)
1552 144 : parse_as_document = true;
1553 : else
1554 : {
1555 : /* Parse and skip over the XML declaration, if any */
1556 482 : res_code = parse_xml_decl(utf8string,
1557 : &count, &version, NULL, &standalone);
1558 482 : if (res_code != 0)
1559 0 : xml_ereport_by_code(ERROR, ERRCODE_INVALID_XML_CONTENT,
1560 : "invalid XML content: invalid XML declaration",
1561 : res_code);
1562 :
1563 : /* Is there a DOCTYPE element? */
1564 482 : if (xml_doctype_in_content(utf8string + count))
1565 16 : parse_as_document = true;
1566 : }
1567 :
1568 626 : if (parse_as_document)
1569 : {
1570 : /*
1571 : * Note, that here we try to apply DTD defaults
1572 : * (XML_PARSE_DTDATTR) according to SQL/XML:2008 GR 10.16.7.d:
1573 : * 'Default values defined by internal DTD are applied'. As for
1574 : * external DTDs, we try to support them too, (see SQL/XML:2008 GR
1575 : * 10.16.7.e)
1576 : */
1577 160 : doc = xmlCtxtReadDoc(ctxt, utf8string,
1578 : NULL,
1579 : "UTF-8",
1580 : XML_PARSE_NOENT | XML_PARSE_DTDATTR
1581 : | (preserve_whitespace ? 0 : XML_PARSE_NOBLANKS));
1582 160 : if (doc == NULL || xmlerrcxt->err_occurred)
1583 : {
1584 : /* Use original option to decide which error code to throw */
1585 80 : if (xmloption_arg == XMLOPTION_DOCUMENT)
1586 76 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_INVALID_XML_DOCUMENT,
1587 : "invalid XML document");
1588 : else
1589 4 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_INVALID_XML_CONTENT,
1590 : "invalid XML content");
1591 : }
1592 : }
1593 : else
1594 : {
1595 466 : doc = xmlNewDoc(version);
1596 : Assert(doc->encoding == NULL);
1597 466 : doc->encoding = xmlStrdup((const xmlChar *) "UTF-8");
1598 466 : doc->standalone = standalone;
1599 :
1600 : /* allow empty content */
1601 466 : if (*(utf8string + count))
1602 : {
1603 1386 : res_code = xmlParseBalancedChunkMemory(doc, NULL, NULL, 0,
1604 462 : utf8string + count, NULL);
1605 462 : if (res_code != 0 || xmlerrcxt->err_occurred)
1606 32 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_INVALID_XML_CONTENT,
1607 : "invalid XML content");
1608 : }
1609 : }
1610 : }
1611 112 : PG_CATCH();
1612 : {
1613 112 : if (doc != NULL)
1614 32 : xmlFreeDoc(doc);
1615 112 : if (ctxt != NULL)
1616 112 : xmlFreeParserCtxt(ctxt);
1617 :
1618 112 : pg_xml_done(xmlerrcxt, true);
1619 :
1620 112 : PG_RE_THROW();
1621 : }
1622 514 : PG_END_TRY();
1623 :
1624 514 : xmlFreeParserCtxt(ctxt);
1625 :
1626 514 : pg_xml_done(xmlerrcxt, false);
1627 :
1628 514 : return doc;
1629 : }
1630 :
1631 :
1632 : /*
1633 : * xmlChar<->text conversions
1634 : */
1635 : static xmlChar *
1636 642 : xml_text2xmlChar(text *in)
1637 : {
1638 642 : return (xmlChar *) text_to_cstring(in);
1639 : }
1640 :
1641 :
1642 : #ifdef USE_LIBXMLCONTEXT
1643 :
1644 : /*
1645 : * Manage the special context used for all libxml allocations (but only
1646 : * in special debug builds; see notes at top of file)
1647 : */
1648 : static void
1649 : xml_memory_init(void)
1650 : {
1651 : /* Create memory context if not there already */
1652 : if (LibxmlContext == NULL)
1653 : LibxmlContext = AllocSetContextCreate(TopMemoryContext,
1654 : "Libxml context",
1655 : ALLOCSET_DEFAULT_SIZES);
1656 :
1657 : /* Re-establish the callbacks even if already set */
1658 : xmlMemSetup(xml_pfree, xml_palloc, xml_repalloc, xml_pstrdup);
1659 : }
1660 :
1661 : /*
1662 : * Wrappers for memory management functions
1663 : */
1664 : static void *
1665 : xml_palloc(size_t size)
1666 : {
1667 : return MemoryContextAlloc(LibxmlContext, size);
1668 : }
1669 :
1670 :
1671 : static void *
1672 : xml_repalloc(void *ptr, size_t size)
1673 : {
1674 : return repalloc(ptr, size);
1675 : }
1676 :
1677 :
1678 : static void
1679 : xml_pfree(void *ptr)
1680 : {
1681 : /* At least some parts of libxml assume xmlFree(NULL) is allowed */
1682 : if (ptr)
1683 : pfree(ptr);
1684 : }
1685 :
1686 :
1687 : static char *
1688 : xml_pstrdup(const char *string)
1689 : {
1690 : return MemoryContextStrdup(LibxmlContext, string);
1691 : }
1692 : #endif /* USE_LIBXMLCONTEXT */
1693 :
1694 :
1695 : /*
1696 : * xmlPgEntityLoader --- entity loader callback function
1697 : *
1698 : * Silently prevent any external entity URL from being loaded. We don't want
1699 : * to throw an error, so instead make the entity appear to expand to an empty
1700 : * string.
1701 : *
1702 : * We would prefer to allow loading entities that exist in the system's
1703 : * global XML catalog; but the available libxml2 APIs make that a complex
1704 : * and fragile task. For now, just shut down all external access.
1705 : */
1706 : static xmlParserInputPtr
1707 28 : xmlPgEntityLoader(const char *URL, const char *ID,
1708 : xmlParserCtxtPtr ctxt)
1709 : {
1710 28 : return xmlNewStringInputStream(ctxt, (const xmlChar *) "");
1711 : }
1712 :
1713 :
1714 : /*
1715 : * xml_ereport --- report an XML-related error
1716 : *
1717 : * The "msg" is the SQL-level message; some can be adopted from the SQL/XML
1718 : * standard. This function adds libxml's native error message, if any, as
1719 : * detail.
1720 : *
1721 : * This is exported for modules that want to share the core libxml error
1722 : * handler. Note that pg_xml_init() *must* have been called previously.
1723 : */
1724 : void
1725 120 : xml_ereport(PgXmlErrorContext *errcxt, int level, int sqlcode, const char *msg)
1726 : {
1727 : char *detail;
1728 :
1729 : /* Defend against someone passing us a bogus context struct */
1730 120 : if (errcxt->magic != ERRCXT_MAGIC)
1731 0 : elog(ERROR, "xml_ereport called with invalid PgXmlErrorContext");
1732 :
1733 : /* Flag that the current libxml error has been reported */
1734 120 : errcxt->err_occurred = false;
1735 :
1736 : /* Include detail only if we have some text from libxml */
1737 120 : if (errcxt->err_buf.len > 0)
1738 120 : detail = errcxt->err_buf.data;
1739 : else
1740 0 : detail = NULL;
1741 :
1742 120 : ereport(level,
1743 : (errcode(sqlcode),
1744 : errmsg_internal("%s", msg),
1745 : detail ? errdetail_internal("%s", detail) : 0));
1746 0 : }
1747 :
1748 :
1749 : /*
1750 : * Error handler for libxml errors and warnings
1751 : */
1752 : static void
1753 222 : xml_errorHandler(void *data, xmlErrorPtr error)
1754 : {
1755 222 : PgXmlErrorContext *xmlerrcxt = (PgXmlErrorContext *) data;
1756 222 : xmlParserCtxtPtr ctxt = (xmlParserCtxtPtr) error->ctxt;
1757 222 : xmlParserInputPtr input = (ctxt != NULL) ? ctxt->input : NULL;
1758 222 : xmlNodePtr node = error->node;
1759 222 : const xmlChar *name = (node != NULL &&
1760 222 : node->type == XML_ELEMENT_NODE) ? node->name : NULL;
1761 222 : int domain = error->domain;
1762 222 : int level = error->level;
1763 : StringInfo errorBuf;
1764 :
1765 : /*
1766 : * Defend against someone passing us a bogus context struct.
1767 : *
1768 : * We force a backend exit if this check fails because longjmp'ing out of
1769 : * libxml would likely render it unsafe to use further.
1770 : */
1771 222 : if (xmlerrcxt->magic != ERRCXT_MAGIC)
1772 0 : elog(FATAL, "xml_errorHandler called with invalid PgXmlErrorContext");
1773 :
1774 : /*----------
1775 : * Older libxml versions report some errors differently.
1776 : * First, some errors were previously reported as coming from the parser
1777 : * domain but are now reported as coming from the namespace domain.
1778 : * Second, some warnings were upgraded to errors.
1779 : * We attempt to compensate for that here.
1780 : *----------
1781 : */
1782 222 : switch (error->code)
1783 : {
1784 20 : case XML_WAR_NS_URI:
1785 20 : level = XML_ERR_ERROR;
1786 20 : domain = XML_FROM_NAMESPACE;
1787 20 : break;
1788 :
1789 36 : case XML_ERR_NS_DECL_ERROR:
1790 : case XML_WAR_NS_URI_RELATIVE:
1791 : case XML_WAR_NS_COLUMN:
1792 : case XML_NS_ERR_XML_NAMESPACE:
1793 : case XML_NS_ERR_UNDEFINED_NAMESPACE:
1794 : case XML_NS_ERR_QNAME:
1795 : case XML_NS_ERR_ATTRIBUTE_REDEFINED:
1796 : case XML_NS_ERR_EMPTY:
1797 36 : domain = XML_FROM_NAMESPACE;
1798 36 : break;
1799 : }
1800 :
1801 : /* Decide whether to act on the error or not */
1802 222 : switch (domain)
1803 : {
1804 166 : case XML_FROM_PARSER:
1805 : case XML_FROM_NONE:
1806 : case XML_FROM_MEMORY:
1807 : case XML_FROM_IO:
1808 :
1809 : /*
1810 : * Suppress warnings about undeclared entities. We need to do
1811 : * this to avoid problems due to not loading DTD definitions.
1812 : */
1813 166 : if (error->code == XML_WAR_UNDECLARED_ENTITY)
1814 4 : return;
1815 :
1816 : /* Otherwise, accept error regardless of the parsing purpose */
1817 162 : break;
1818 :
1819 56 : default:
1820 : /* Ignore error if only doing well-formedness check */
1821 56 : if (xmlerrcxt->strictness == PG_XML_STRICTNESS_WELLFORMED)
1822 44 : return;
1823 12 : break;
1824 : }
1825 :
1826 : /* Prepare error message in errorBuf */
1827 174 : errorBuf = makeStringInfo();
1828 :
1829 174 : if (error->line > 0)
1830 174 : appendStringInfo(errorBuf, "line %d: ", error->line);
1831 174 : if (name != NULL)
1832 0 : appendStringInfo(errorBuf, "element %s: ", name);
1833 174 : if (error->message != NULL)
1834 174 : appendStringInfoString(errorBuf, error->message);
1835 : else
1836 0 : appendStringInfoString(errorBuf, "(no message provided)");
1837 :
1838 : /*
1839 : * Append context information to errorBuf.
1840 : *
1841 : * xmlParserPrintFileContext() uses libxml's "generic" error handler to
1842 : * write the context. Since we don't want to duplicate libxml
1843 : * functionality here, we set up a generic error handler temporarily.
1844 : *
1845 : * We use appendStringInfo() directly as libxml's generic error handler.
1846 : * This should work because it has essentially the same signature as
1847 : * libxml expects, namely (void *ptr, const char *msg, ...).
1848 : */
1849 174 : if (input != NULL)
1850 : {
1851 174 : xmlGenericErrorFunc errFuncSaved = xmlGenericError;
1852 174 : void *errCtxSaved = xmlGenericErrorContext;
1853 :
1854 174 : xmlSetGenericErrorFunc((void *) errorBuf,
1855 : (xmlGenericErrorFunc) appendStringInfo);
1856 :
1857 : /* Add context information to errorBuf */
1858 174 : appendStringInfoLineSeparator(errorBuf);
1859 :
1860 174 : xmlParserPrintFileContext(input);
1861 :
1862 : /* Restore generic error func */
1863 174 : xmlSetGenericErrorFunc(errCtxSaved, errFuncSaved);
1864 : }
1865 :
1866 : /* Get rid of any trailing newlines in errorBuf */
1867 174 : chopStringInfoNewlines(errorBuf);
1868 :
1869 : /*
1870 : * Legacy error handling mode. err_occurred is never set, we just add the
1871 : * message to err_buf. This mode exists because the xml2 contrib module
1872 : * uses our error-handling infrastructure, but we don't want to change its
1873 : * behaviour since it's deprecated anyway. This is also why we don't
1874 : * distinguish between notices, warnings and errors here --- the old-style
1875 : * generic error handler wouldn't have done that either.
1876 : */
1877 174 : if (xmlerrcxt->strictness == PG_XML_STRICTNESS_LEGACY)
1878 : {
1879 2 : appendStringInfoLineSeparator(&xmlerrcxt->err_buf);
1880 2 : appendBinaryStringInfo(&xmlerrcxt->err_buf, errorBuf->data,
1881 : errorBuf->len);
1882 :
1883 2 : pfree(errorBuf->data);
1884 2 : pfree(errorBuf);
1885 2 : return;
1886 : }
1887 :
1888 : /*
1889 : * We don't want to ereport() here because that'd probably leave libxml in
1890 : * an inconsistent state. Instead, we remember the error and ereport()
1891 : * from xml_ereport().
1892 : *
1893 : * Warnings and notices can be reported immediately since they won't cause
1894 : * a longjmp() out of libxml.
1895 : */
1896 172 : if (level >= XML_ERR_ERROR)
1897 : {
1898 168 : appendStringInfoLineSeparator(&xmlerrcxt->err_buf);
1899 168 : appendBinaryStringInfo(&xmlerrcxt->err_buf, errorBuf->data,
1900 : errorBuf->len);
1901 :
1902 168 : xmlerrcxt->err_occurred = true;
1903 : }
1904 4 : else if (level >= XML_ERR_WARNING)
1905 : {
1906 4 : ereport(WARNING,
1907 : (errmsg_internal("%s", errorBuf->data)));
1908 : }
1909 : else
1910 : {
1911 0 : ereport(NOTICE,
1912 : (errmsg_internal("%s", errorBuf->data)));
1913 : }
1914 :
1915 172 : pfree(errorBuf->data);
1916 172 : pfree(errorBuf);
1917 : }
1918 :
1919 :
1920 : /*
1921 : * Wrapper for "ereport" function for XML-related errors. The "msg"
1922 : * is the SQL-level message; some can be adopted from the SQL/XML
1923 : * standard. This function uses "code" to create a textual detail
1924 : * message. At the moment, we only need to cover those codes that we
1925 : * may raise in this file.
1926 : */
1927 : static void
1928 0 : xml_ereport_by_code(int level, int sqlcode,
1929 : const char *msg, int code)
1930 : {
1931 : const char *det;
1932 :
1933 0 : switch (code)
1934 : {
1935 0 : case XML_ERR_INVALID_CHAR:
1936 0 : det = gettext_noop("Invalid character value.");
1937 0 : break;
1938 0 : case XML_ERR_SPACE_REQUIRED:
1939 0 : det = gettext_noop("Space required.");
1940 0 : break;
1941 0 : case XML_ERR_STANDALONE_VALUE:
1942 0 : det = gettext_noop("standalone accepts only 'yes' or 'no'.");
1943 0 : break;
1944 0 : case XML_ERR_VERSION_MISSING:
1945 0 : det = gettext_noop("Malformed declaration: missing version.");
1946 0 : break;
1947 0 : case XML_ERR_MISSING_ENCODING:
1948 0 : det = gettext_noop("Missing encoding in text declaration.");
1949 0 : break;
1950 0 : case XML_ERR_XMLDECL_NOT_FINISHED:
1951 0 : det = gettext_noop("Parsing XML declaration: '?>' expected.");
1952 0 : break;
1953 0 : default:
1954 0 : det = gettext_noop("Unrecognized libxml error code: %d.");
1955 0 : break;
1956 : }
1957 :
1958 0 : ereport(level,
1959 : (errcode(sqlcode),
1960 : errmsg_internal("%s", msg),
1961 : errdetail(det, code)));
1962 0 : }
1963 :
1964 :
1965 : /*
1966 : * Remove all trailing newlines from a StringInfo string
1967 : */
1968 : static void
1969 518 : chopStringInfoNewlines(StringInfo str)
1970 : {
1971 866 : while (str->len > 0 && str->data[str->len - 1] == '\n')
1972 348 : str->data[--str->len] = '\0';
1973 518 : }
1974 :
1975 :
1976 : /*
1977 : * Append a newline after removing any existing trailing newlines
1978 : */
1979 : static void
1980 344 : appendStringInfoLineSeparator(StringInfo str)
1981 : {
1982 344 : chopStringInfoNewlines(str);
1983 344 : if (str->len > 0)
1984 222 : appendStringInfoChar(str, '\n');
1985 344 : }
1986 :
1987 :
1988 : /*
1989 : * Convert one char in the current server encoding to a Unicode codepoint.
1990 : */
1991 : static pg_wchar
1992 12094 : sqlchar_to_unicode(const char *s)
1993 : {
1994 : char *utf8string;
1995 : pg_wchar ret[2]; /* need space for trailing zero */
1996 :
1997 : /* note we're not assuming s is null-terminated */
1998 12094 : utf8string = pg_server_to_any(s, pg_mblen(s), PG_UTF8);
1999 :
2000 12094 : pg_encoding_mb2wchar_with_len(PG_UTF8, utf8string, ret,
2001 : pg_encoding_mblen(PG_UTF8, utf8string));
2002 :
2003 12094 : if (utf8string != s)
2004 0 : pfree(utf8string);
2005 :
2006 12094 : return ret[0];
2007 : }
2008 :
2009 :
2010 : static bool
2011 2394 : is_valid_xml_namefirst(pg_wchar c)
2012 : {
2013 : /* (Letter | '_' | ':') */
2014 2398 : return (xmlIsBaseCharQ(c) || xmlIsIdeographicQ(c)
2015 4792 : || c == '_' || c == ':');
2016 : }
2017 :
2018 :
2019 : static bool
2020 9700 : is_valid_xml_namechar(pg_wchar c)
2021 : {
2022 : /* Letter | Digit | '.' | '-' | '_' | ':' | CombiningChar | Extender */
2023 10294 : return (xmlIsBaseCharQ(c) || xmlIsIdeographicQ(c)
2024 594 : || xmlIsDigitQ(c)
2025 170 : || c == '.' || c == '-' || c == '_' || c == ':'
2026 8 : || xmlIsCombiningQ(c)
2027 20588 : || xmlIsExtenderQ(c));
2028 : }
2029 : #endif /* USE_LIBXML */
2030 :
2031 :
2032 : /*
2033 : * Map SQL identifier to XML name; see SQL/XML:2008 section 9.1.
2034 : */
2035 : char *
2036 2404 : map_sql_identifier_to_xml_name(const char *ident, bool fully_escaped,
2037 : bool escape_period)
2038 : {
2039 : #ifdef USE_LIBXML
2040 : StringInfoData buf;
2041 : const char *p;
2042 :
2043 : /*
2044 : * SQL/XML doesn't make use of this case anywhere, so it's probably a
2045 : * mistake.
2046 : */
2047 : Assert(fully_escaped || !escape_period);
2048 :
2049 2404 : initStringInfo(&buf);
2050 :
2051 14512 : for (p = ident; *p; p += pg_mblen(p))
2052 : {
2053 12108 : if (*p == ':' && (p == ident || fully_escaped))
2054 10 : appendStringInfoString(&buf, "_x003A_");
2055 12098 : else if (*p == '_' && *(p + 1) == 'x')
2056 4 : appendStringInfoString(&buf, "_x005F_");
2057 14250 : else if (fully_escaped && p == ident &&
2058 2156 : pg_strncasecmp(p, "xml", 3) == 0)
2059 : {
2060 0 : if (*p == 'x')
2061 0 : appendStringInfoString(&buf, "_x0078_");
2062 : else
2063 0 : appendStringInfoString(&buf, "_x0058_");
2064 : }
2065 12094 : else if (escape_period && *p == '.')
2066 0 : appendStringInfoString(&buf, "_x002E_");
2067 : else
2068 : {
2069 12094 : pg_wchar u = sqlchar_to_unicode(p);
2070 :
2071 24188 : if ((p == ident)
2072 2394 : ? !is_valid_xml_namefirst(u)
2073 9700 : : !is_valid_xml_namechar(u))
2074 12 : appendStringInfo(&buf, "_x%04X_", (unsigned int) u);
2075 : else
2076 12082 : appendBinaryStringInfo(&buf, p, pg_mblen(p));
2077 : }
2078 : }
2079 :
2080 2404 : return buf.data;
2081 : #else /* not USE_LIBXML */
2082 : NO_XML_SUPPORT();
2083 : return NULL;
2084 : #endif /* not USE_LIBXML */
2085 : }
2086 :
2087 :
2088 : /*
2089 : * Map XML name to SQL identifier; see SQL/XML:2008 section 9.3.
2090 : */
2091 : char *
2092 80 : map_xml_name_to_sql_identifier(const char *name)
2093 : {
2094 : StringInfoData buf;
2095 : const char *p;
2096 :
2097 80 : initStringInfo(&buf);
2098 :
2099 440 : for (p = name; *p; p += pg_mblen(p))
2100 : {
2101 360 : if (*p == '_' && *(p + 1) == 'x'
2102 10 : && isxdigit((unsigned char) *(p + 2))
2103 10 : && isxdigit((unsigned char) *(p + 3))
2104 10 : && isxdigit((unsigned char) *(p + 4))
2105 10 : && isxdigit((unsigned char) *(p + 5))
2106 10 : && *(p + 6) == '_')
2107 10 : {
2108 : char cbuf[MAX_UNICODE_EQUIVALENT_STRING + 1];
2109 : unsigned int u;
2110 :
2111 10 : sscanf(p + 2, "%X", &u);
2112 10 : pg_unicode_to_server(u, (unsigned char *) cbuf);
2113 10 : appendStringInfoString(&buf, cbuf);
2114 10 : p += 6;
2115 : }
2116 : else
2117 350 : appendBinaryStringInfo(&buf, p, pg_mblen(p));
2118 : }
2119 :
2120 80 : return buf.data;
2121 : }
2122 :
2123 : /*
2124 : * Map SQL value to XML value; see SQL/XML:2008 section 9.8.
2125 : *
2126 : * When xml_escape_strings is true, then certain characters in string
2127 : * values are replaced by entity references (< etc.), as specified
2128 : * in SQL/XML:2008 section 9.8 GR 9) a) iii). This is normally what is
2129 : * wanted. The false case is mainly useful when the resulting value
2130 : * is used with xmlTextWriterWriteAttribute() to write out an
2131 : * attribute, because that function does the escaping itself.
2132 : */
2133 : char *
2134 83098 : map_sql_value_to_xml_value(Datum value, Oid type, bool xml_escape_strings)
2135 : {
2136 83098 : if (type_is_array_domain(type))
2137 : {
2138 : ArrayType *array;
2139 : Oid elmtype;
2140 : int16 elmlen;
2141 : bool elmbyval;
2142 : char elmalign;
2143 : int num_elems;
2144 : Datum *elem_values;
2145 : bool *elem_nulls;
2146 : StringInfoData buf;
2147 : int i;
2148 :
2149 4 : array = DatumGetArrayTypeP(value);
2150 4 : elmtype = ARR_ELEMTYPE(array);
2151 4 : get_typlenbyvalalign(elmtype, &elmlen, &elmbyval, &elmalign);
2152 :
2153 4 : deconstruct_array(array, elmtype,
2154 : elmlen, elmbyval, elmalign,
2155 : &elem_values, &elem_nulls,
2156 : &num_elems);
2157 :
2158 4 : initStringInfo(&buf);
2159 :
2160 16 : for (i = 0; i < num_elems; i++)
2161 : {
2162 12 : if (elem_nulls[i])
2163 0 : continue;
2164 12 : appendStringInfoString(&buf, "<element>");
2165 12 : appendStringInfoString(&buf,
2166 12 : map_sql_value_to_xml_value(elem_values[i],
2167 : elmtype, true));
2168 12 : appendStringInfoString(&buf, "</element>");
2169 : }
2170 :
2171 4 : pfree(elem_values);
2172 4 : pfree(elem_nulls);
2173 :
2174 4 : return buf.data;
2175 : }
2176 : else
2177 : {
2178 : Oid typeOut;
2179 : bool isvarlena;
2180 : char *str;
2181 :
2182 : /*
2183 : * Flatten domains; the special-case treatments below should apply to,
2184 : * eg, domains over boolean not just boolean.
2185 : */
2186 83094 : type = getBaseType(type);
2187 :
2188 : /*
2189 : * Special XSD formatting for some data types
2190 : */
2191 83094 : switch (type)
2192 : {
2193 44 : case BOOLOID:
2194 44 : if (DatumGetBool(value))
2195 40 : return "true";
2196 : else
2197 4 : return "false";
2198 :
2199 32 : case DATEOID:
2200 : {
2201 : DateADT date;
2202 : struct pg_tm tm;
2203 : char buf[MAXDATELEN + 1];
2204 :
2205 32 : date = DatumGetDateADT(value);
2206 : /* XSD doesn't support infinite values */
2207 32 : if (DATE_NOT_FINITE(date))
2208 0 : ereport(ERROR,
2209 : (errcode(ERRCODE_DATETIME_VALUE_OUT_OF_RANGE),
2210 : errmsg("date out of range"),
2211 : errdetail("XML does not support infinite date values.")));
2212 32 : j2date(date + POSTGRES_EPOCH_JDATE,
2213 : &(tm.tm_year), &(tm.tm_mon), &(tm.tm_mday));
2214 32 : EncodeDateOnly(&tm, USE_XSD_DATES, buf);
2215 :
2216 32 : return pstrdup(buf);
2217 : }
2218 :
2219 24 : case TIMESTAMPOID:
2220 : {
2221 : Timestamp timestamp;
2222 : struct pg_tm tm;
2223 : fsec_t fsec;
2224 : char buf[MAXDATELEN + 1];
2225 :
2226 24 : timestamp = DatumGetTimestamp(value);
2227 :
2228 : /* XSD doesn't support infinite values */
2229 24 : if (TIMESTAMP_NOT_FINITE(timestamp))
2230 4 : ereport(ERROR,
2231 : (errcode(ERRCODE_DATETIME_VALUE_OUT_OF_RANGE),
2232 : errmsg("timestamp out of range"),
2233 : errdetail("XML does not support infinite timestamp values.")));
2234 20 : else if (timestamp2tm(timestamp, NULL, &tm, &fsec, NULL, NULL) == 0)
2235 20 : EncodeDateTime(&tm, fsec, false, 0, NULL, USE_XSD_DATES, buf);
2236 : else
2237 0 : ereport(ERROR,
2238 : (errcode(ERRCODE_DATETIME_VALUE_OUT_OF_RANGE),
2239 : errmsg("timestamp out of range")));
2240 :
2241 20 : return pstrdup(buf);
2242 : }
2243 :
2244 0 : case TIMESTAMPTZOID:
2245 : {
2246 : TimestampTz timestamp;
2247 : struct pg_tm tm;
2248 : int tz;
2249 : fsec_t fsec;
2250 0 : const char *tzn = NULL;
2251 : char buf[MAXDATELEN + 1];
2252 :
2253 0 : timestamp = DatumGetTimestamp(value);
2254 :
2255 : /* XSD doesn't support infinite values */
2256 0 : if (TIMESTAMP_NOT_FINITE(timestamp))
2257 0 : ereport(ERROR,
2258 : (errcode(ERRCODE_DATETIME_VALUE_OUT_OF_RANGE),
2259 : errmsg("timestamp out of range"),
2260 : errdetail("XML does not support infinite timestamp values.")));
2261 0 : else if (timestamp2tm(timestamp, &tz, &tm, &fsec, &tzn, NULL) == 0)
2262 0 : EncodeDateTime(&tm, fsec, true, tz, tzn, USE_XSD_DATES, buf);
2263 : else
2264 0 : ereport(ERROR,
2265 : (errcode(ERRCODE_DATETIME_VALUE_OUT_OF_RANGE),
2266 : errmsg("timestamp out of range")));
2267 :
2268 0 : return pstrdup(buf);
2269 : }
2270 :
2271 : #ifdef USE_LIBXML
2272 24 : case BYTEAOID:
2273 : {
2274 24 : bytea *bstr = DatumGetByteaPP(value);
2275 : PgXmlErrorContext *xmlerrcxt;
2276 24 : volatile xmlBufferPtr buf = NULL;
2277 24 : volatile xmlTextWriterPtr writer = NULL;
2278 : char *result;
2279 :
2280 24 : xmlerrcxt = pg_xml_init(PG_XML_STRICTNESS_ALL);
2281 :
2282 24 : PG_TRY();
2283 : {
2284 24 : buf = xmlBufferCreate();
2285 24 : if (buf == NULL || xmlerrcxt->err_occurred)
2286 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
2287 : "could not allocate xmlBuffer");
2288 24 : writer = xmlNewTextWriterMemory(buf, 0);
2289 24 : if (writer == NULL || xmlerrcxt->err_occurred)
2290 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
2291 : "could not allocate xmlTextWriter");
2292 :
2293 24 : if (xmlbinary == XMLBINARY_BASE64)
2294 20 : xmlTextWriterWriteBase64(writer, VARDATA_ANY(bstr),
2295 20 : 0, VARSIZE_ANY_EXHDR(bstr));
2296 : else
2297 4 : xmlTextWriterWriteBinHex(writer, VARDATA_ANY(bstr),
2298 4 : 0, VARSIZE_ANY_EXHDR(bstr));
2299 :
2300 : /* we MUST do this now to flush data out to the buffer */
2301 24 : xmlFreeTextWriter(writer);
2302 24 : writer = NULL;
2303 :
2304 24 : result = pstrdup((const char *) xmlBufferContent(buf));
2305 : }
2306 0 : PG_CATCH();
2307 : {
2308 0 : if (writer)
2309 0 : xmlFreeTextWriter(writer);
2310 0 : if (buf)
2311 0 : xmlBufferFree(buf);
2312 :
2313 0 : pg_xml_done(xmlerrcxt, true);
2314 :
2315 0 : PG_RE_THROW();
2316 : }
2317 24 : PG_END_TRY();
2318 :
2319 24 : xmlBufferFree(buf);
2320 :
2321 24 : pg_xml_done(xmlerrcxt, false);
2322 :
2323 24 : return result;
2324 : }
2325 : #endif /* USE_LIBXML */
2326 :
2327 : }
2328 :
2329 : /*
2330 : * otherwise, just use the type's native text representation
2331 : */
2332 82970 : getTypeOutputInfo(type, &typeOut, &isvarlena);
2333 82970 : str = OidOutputFunctionCall(typeOut, value);
2334 :
2335 : /* ... exactly as-is for XML, and when escaping is not wanted */
2336 82970 : if (type == XMLOID || !xml_escape_strings)
2337 13806 : return str;
2338 :
2339 : /* otherwise, translate special characters as needed */
2340 69164 : return escape_xml(str);
2341 : }
2342 : }
2343 :
2344 :
2345 : /*
2346 : * Escape characters in text that have special meanings in XML.
2347 : *
2348 : * Returns a palloc'd string.
2349 : *
2350 : * NB: this is intentionally not dependent on libxml.
2351 : */
2352 : char *
2353 69332 : escape_xml(const char *str)
2354 : {
2355 : StringInfoData buf;
2356 : const char *p;
2357 :
2358 69332 : initStringInfo(&buf);
2359 420136 : for (p = str; *p; p++)
2360 : {
2361 350804 : switch (*p)
2362 : {
2363 0 : case '&':
2364 0 : appendStringInfoString(&buf, "&");
2365 0 : break;
2366 24 : case '<':
2367 24 : appendStringInfoString(&buf, "<");
2368 24 : break;
2369 16 : case '>':
2370 16 : appendStringInfoString(&buf, ">");
2371 16 : break;
2372 0 : case '\r':
2373 0 : appendStringInfoString(&buf, "
");
2374 0 : break;
2375 350764 : default:
2376 350764 : appendStringInfoCharMacro(&buf, *p);
2377 350764 : break;
2378 : }
2379 : }
2380 69332 : return buf.data;
2381 : }
2382 :
2383 :
2384 : static char *
2385 16 : _SPI_strdup(const char *s)
2386 : {
2387 16 : size_t len = strlen(s) + 1;
2388 16 : char *ret = SPI_palloc(len);
2389 :
2390 16 : memcpy(ret, s, len);
2391 16 : return ret;
2392 : }
2393 :
2394 :
2395 : /*
2396 : * SQL to XML mapping functions
2397 : *
2398 : * What follows below was at one point intentionally organized so that
2399 : * you can read along in the SQL/XML standard. The functions are
2400 : * mostly split up the way the clauses lay out in the standards
2401 : * document, and the identifiers are also aligned with the standard
2402 : * text. Unfortunately, SQL/XML:2006 reordered the clauses
2403 : * differently than SQL/XML:2003, so the order below doesn't make much
2404 : * sense anymore.
2405 : *
2406 : * There are many things going on there:
2407 : *
2408 : * There are two kinds of mappings: Mapping SQL data (table contents)
2409 : * to XML documents, and mapping SQL structure (the "schema") to XML
2410 : * Schema. And there are functions that do both at the same time.
2411 : *
2412 : * Then you can map a database, a schema, or a table, each in both
2413 : * ways. This breaks down recursively: Mapping a database invokes
2414 : * mapping schemas, which invokes mapping tables, which invokes
2415 : * mapping rows, which invokes mapping columns, although you can't
2416 : * call the last two from the outside. Because of this, there are a
2417 : * number of xyz_internal() functions which are to be called both from
2418 : * the function manager wrapper and from some upper layer in a
2419 : * recursive call.
2420 : *
2421 : * See the documentation about what the common function arguments
2422 : * nulls, tableforest, and targetns mean.
2423 : *
2424 : * Some style guidelines for XML output: Use double quotes for quoting
2425 : * XML attributes. Indent XML elements by two spaces, but remember
2426 : * that a lot of code is called recursively at different levels, so
2427 : * it's better not to indent rather than create output that indents
2428 : * and outdents weirdly. Add newlines to make the output look nice.
2429 : */
2430 :
2431 :
2432 : /*
2433 : * Visibility of objects for XML mappings; see SQL/XML:2008 section
2434 : * 4.10.8.
2435 : */
2436 :
2437 : /*
2438 : * Given a query, which must return type oid as first column, produce
2439 : * a list of Oids with the query results.
2440 : */
2441 : static List *
2442 24 : query_to_oid_list(const char *query)
2443 : {
2444 : uint64 i;
2445 24 : List *list = NIL;
2446 : int spi_result;
2447 :
2448 24 : spi_result = SPI_execute(query, true, 0);
2449 24 : if (spi_result != SPI_OK_SELECT)
2450 0 : elog(ERROR, "SPI_execute returned %s for %s",
2451 : SPI_result_code_string(spi_result), query);
2452 :
2453 72 : for (i = 0; i < SPI_processed; i++)
2454 : {
2455 : Datum oid;
2456 : bool isnull;
2457 :
2458 48 : oid = SPI_getbinval(SPI_tuptable->vals[i],
2459 48 : SPI_tuptable->tupdesc,
2460 : 1,
2461 : &isnull);
2462 48 : if (!isnull)
2463 48 : list = lappend_oid(list, DatumGetObjectId(oid));
2464 : }
2465 :
2466 24 : return list;
2467 : }
2468 :
2469 :
2470 : static List *
2471 24 : schema_get_xml_visible_tables(Oid nspid)
2472 : {
2473 : StringInfoData query;
2474 :
2475 24 : initStringInfo(&query);
2476 24 : appendStringInfo(&query, "SELECT oid FROM pg_catalog.pg_class"
2477 : " WHERE relnamespace = %u AND relkind IN ("
2478 : CppAsString2(RELKIND_RELATION) ","
2479 : CppAsString2(RELKIND_MATVIEW) ","
2480 : CppAsString2(RELKIND_VIEW) ")"
2481 : " AND pg_catalog.has_table_privilege (oid, 'SELECT')"
2482 : " ORDER BY relname;", nspid);
2483 :
2484 24 : return query_to_oid_list(query.data);
2485 : }
2486 :
2487 :
2488 : /*
2489 : * Including the system schemas is probably not useful for a database
2490 : * mapping.
2491 : */
2492 : #define XML_VISIBLE_SCHEMAS_EXCLUDE "(nspname ~ '^pg_' OR nspname = 'information_schema')"
2493 :
2494 : #define XML_VISIBLE_SCHEMAS "SELECT oid FROM pg_catalog.pg_namespace WHERE pg_catalog.has_schema_privilege (oid, 'USAGE') AND NOT " XML_VISIBLE_SCHEMAS_EXCLUDE
2495 :
2496 :
2497 : static List *
2498 0 : database_get_xml_visible_schemas(void)
2499 : {
2500 0 : return query_to_oid_list(XML_VISIBLE_SCHEMAS " ORDER BY nspname;");
2501 : }
2502 :
2503 :
2504 : static List *
2505 0 : database_get_xml_visible_tables(void)
2506 : {
2507 : /* At the moment there is no order required here. */
2508 0 : return query_to_oid_list("SELECT oid FROM pg_catalog.pg_class"
2509 : " WHERE relkind IN ("
2510 : CppAsString2(RELKIND_RELATION) ","
2511 : CppAsString2(RELKIND_MATVIEW) ","
2512 : CppAsString2(RELKIND_VIEW) ")"
2513 : " AND pg_catalog.has_table_privilege(pg_class.oid, 'SELECT')"
2514 : " AND relnamespace IN (" XML_VISIBLE_SCHEMAS ");");
2515 : }
2516 :
2517 :
2518 : /*
2519 : * Map SQL table to XML and/or XML Schema document; see SQL/XML:2008
2520 : * section 9.11.
2521 : */
2522 :
2523 : static StringInfo
2524 64 : table_to_xml_internal(Oid relid,
2525 : const char *xmlschema, bool nulls, bool tableforest,
2526 : const char *targetns, bool top_level)
2527 : {
2528 : StringInfoData query;
2529 :
2530 64 : initStringInfo(&query);
2531 64 : appendStringInfo(&query, "SELECT * FROM %s",
2532 64 : DatumGetCString(DirectFunctionCall1(regclassout,
2533 : ObjectIdGetDatum(relid))));
2534 64 : return query_to_xml_internal(query.data, get_rel_name(relid),
2535 : xmlschema, nulls, tableforest,
2536 : targetns, top_level);
2537 : }
2538 :
2539 :
2540 : Datum
2541 24 : table_to_xml(PG_FUNCTION_ARGS)
2542 : {
2543 24 : Oid relid = PG_GETARG_OID(0);
2544 24 : bool nulls = PG_GETARG_BOOL(1);
2545 24 : bool tableforest = PG_GETARG_BOOL(2);
2546 24 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2547 :
2548 24 : PG_RETURN_XML_P(stringinfo_to_xmltype(table_to_xml_internal(relid, NULL,
2549 : nulls, tableforest,
2550 : targetns, true)));
2551 : }
2552 :
2553 :
2554 : Datum
2555 8 : query_to_xml(PG_FUNCTION_ARGS)
2556 : {
2557 8 : char *query = text_to_cstring(PG_GETARG_TEXT_PP(0));
2558 8 : bool nulls = PG_GETARG_BOOL(1);
2559 8 : bool tableforest = PG_GETARG_BOOL(2);
2560 8 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2561 :
2562 8 : PG_RETURN_XML_P(stringinfo_to_xmltype(query_to_xml_internal(query, NULL,
2563 : NULL, nulls, tableforest,
2564 : targetns, true)));
2565 : }
2566 :
2567 :
2568 : Datum
2569 8 : cursor_to_xml(PG_FUNCTION_ARGS)
2570 : {
2571 8 : char *name = text_to_cstring(PG_GETARG_TEXT_PP(0));
2572 8 : int32 count = PG_GETARG_INT32(1);
2573 8 : bool nulls = PG_GETARG_BOOL(2);
2574 8 : bool tableforest = PG_GETARG_BOOL(3);
2575 8 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(4));
2576 :
2577 : StringInfoData result;
2578 : Portal portal;
2579 : uint64 i;
2580 :
2581 8 : initStringInfo(&result);
2582 :
2583 8 : if (!tableforest)
2584 : {
2585 4 : xmldata_root_element_start(&result, "table", NULL, targetns, true);
2586 4 : appendStringInfoChar(&result, '\n');
2587 : }
2588 :
2589 8 : SPI_connect();
2590 8 : portal = SPI_cursor_find(name);
2591 8 : if (portal == NULL)
2592 0 : ereport(ERROR,
2593 : (errcode(ERRCODE_UNDEFINED_CURSOR),
2594 : errmsg("cursor \"%s\" does not exist", name)));
2595 :
2596 8 : SPI_cursor_fetch(portal, true, count);
2597 32 : for (i = 0; i < SPI_processed; i++)
2598 24 : SPI_sql_row_to_xmlelement(i, &result, NULL, nulls,
2599 : tableforest, targetns, true);
2600 :
2601 8 : SPI_finish();
2602 :
2603 8 : if (!tableforest)
2604 4 : xmldata_root_element_end(&result, "table");
2605 :
2606 8 : PG_RETURN_XML_P(stringinfo_to_xmltype(&result));
2607 : }
2608 :
2609 :
2610 : /*
2611 : * Write the start tag of the root element of a data mapping.
2612 : *
2613 : * top_level means that this is the very top level of the eventual
2614 : * output. For example, when the user calls table_to_xml, then a call
2615 : * with a table name to this function is the top level. When the user
2616 : * calls database_to_xml, then a call with a schema name to this
2617 : * function is not the top level. If top_level is false, then the XML
2618 : * namespace declarations are omitted, because they supposedly already
2619 : * appeared earlier in the output. Repeating them is not wrong, but
2620 : * it looks ugly.
2621 : */
2622 : static void
2623 160 : xmldata_root_element_start(StringInfo result, const char *eltname,
2624 : const char *xmlschema, const char *targetns,
2625 : bool top_level)
2626 : {
2627 : /* This isn't really wrong but currently makes no sense. */
2628 : Assert(top_level || !xmlschema);
2629 :
2630 160 : appendStringInfo(result, "<%s", eltname);
2631 160 : if (top_level)
2632 : {
2633 120 : appendStringInfoString(result, " xmlns:xsi=\"" NAMESPACE_XSI "\"");
2634 120 : if (strlen(targetns) > 0)
2635 20 : appendStringInfo(result, " xmlns=\"%s\"", targetns);
2636 : }
2637 160 : if (xmlschema)
2638 : {
2639 : /* FIXME: better targets */
2640 12 : if (strlen(targetns) > 0)
2641 4 : appendStringInfo(result, " xsi:schemaLocation=\"%s #\"", targetns);
2642 : else
2643 8 : appendStringInfoString(result, " xsi:noNamespaceSchemaLocation=\"#\"");
2644 : }
2645 160 : appendStringInfoString(result, ">\n");
2646 160 : }
2647 :
2648 :
2649 : static void
2650 160 : xmldata_root_element_end(StringInfo result, const char *eltname)
2651 : {
2652 160 : appendStringInfo(result, "</%s>\n", eltname);
2653 160 : }
2654 :
2655 :
2656 : static StringInfo
2657 76 : query_to_xml_internal(const char *query, char *tablename,
2658 : const char *xmlschema, bool nulls, bool tableforest,
2659 : const char *targetns, bool top_level)
2660 : {
2661 : StringInfo result;
2662 : char *xmltn;
2663 : uint64 i;
2664 :
2665 76 : if (tablename)
2666 64 : xmltn = map_sql_identifier_to_xml_name(tablename, true, false);
2667 : else
2668 12 : xmltn = "table";
2669 :
2670 76 : result = makeStringInfo();
2671 :
2672 76 : SPI_connect();
2673 76 : if (SPI_execute(query, true, 0) != SPI_OK_SELECT)
2674 0 : ereport(ERROR,
2675 : (errcode(ERRCODE_DATA_EXCEPTION),
2676 : errmsg("invalid query")));
2677 :
2678 76 : if (!tableforest)
2679 : {
2680 36 : xmldata_root_element_start(result, xmltn, xmlschema,
2681 : targetns, top_level);
2682 36 : appendStringInfoChar(result, '\n');
2683 : }
2684 :
2685 76 : if (xmlschema)
2686 20 : appendStringInfo(result, "%s\n\n", xmlschema);
2687 :
2688 264 : for (i = 0; i < SPI_processed; i++)
2689 188 : SPI_sql_row_to_xmlelement(i, result, tablename, nulls,
2690 : tableforest, targetns, top_level);
2691 :
2692 76 : if (!tableforest)
2693 36 : xmldata_root_element_end(result, xmltn);
2694 :
2695 76 : SPI_finish();
2696 :
2697 76 : return result;
2698 : }
2699 :
2700 :
2701 : Datum
2702 20 : table_to_xmlschema(PG_FUNCTION_ARGS)
2703 : {
2704 20 : Oid relid = PG_GETARG_OID(0);
2705 20 : bool nulls = PG_GETARG_BOOL(1);
2706 20 : bool tableforest = PG_GETARG_BOOL(2);
2707 20 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2708 : const char *result;
2709 : Relation rel;
2710 :
2711 20 : rel = table_open(relid, AccessShareLock);
2712 20 : result = map_sql_table_to_xmlschema(rel->rd_att, relid, nulls,
2713 : tableforest, targetns);
2714 20 : table_close(rel, NoLock);
2715 :
2716 20 : PG_RETURN_XML_P(cstring_to_xmltype(result));
2717 : }
2718 :
2719 :
2720 : Datum
2721 4 : query_to_xmlschema(PG_FUNCTION_ARGS)
2722 : {
2723 4 : char *query = text_to_cstring(PG_GETARG_TEXT_PP(0));
2724 4 : bool nulls = PG_GETARG_BOOL(1);
2725 4 : bool tableforest = PG_GETARG_BOOL(2);
2726 4 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2727 : const char *result;
2728 : SPIPlanPtr plan;
2729 : Portal portal;
2730 :
2731 4 : SPI_connect();
2732 :
2733 4 : if ((plan = SPI_prepare(query, 0, NULL)) == NULL)
2734 0 : elog(ERROR, "SPI_prepare(\"%s\") failed", query);
2735 :
2736 4 : if ((portal = SPI_cursor_open(NULL, plan, NULL, NULL, true)) == NULL)
2737 0 : elog(ERROR, "SPI_cursor_open(\"%s\") failed", query);
2738 :
2739 4 : result = _SPI_strdup(map_sql_table_to_xmlschema(portal->tupDesc,
2740 : InvalidOid, nulls,
2741 : tableforest, targetns));
2742 4 : SPI_cursor_close(portal);
2743 4 : SPI_finish();
2744 :
2745 4 : PG_RETURN_XML_P(cstring_to_xmltype(result));
2746 : }
2747 :
2748 :
2749 : Datum
2750 8 : cursor_to_xmlschema(PG_FUNCTION_ARGS)
2751 : {
2752 8 : char *name = text_to_cstring(PG_GETARG_TEXT_PP(0));
2753 8 : bool nulls = PG_GETARG_BOOL(1);
2754 8 : bool tableforest = PG_GETARG_BOOL(2);
2755 8 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2756 : const char *xmlschema;
2757 : Portal portal;
2758 :
2759 8 : SPI_connect();
2760 8 : portal = SPI_cursor_find(name);
2761 8 : if (portal == NULL)
2762 0 : ereport(ERROR,
2763 : (errcode(ERRCODE_UNDEFINED_CURSOR),
2764 : errmsg("cursor \"%s\" does not exist", name)));
2765 :
2766 8 : xmlschema = _SPI_strdup(map_sql_table_to_xmlschema(portal->tupDesc,
2767 : InvalidOid, nulls,
2768 : tableforest, targetns));
2769 8 : SPI_finish();
2770 :
2771 8 : PG_RETURN_XML_P(cstring_to_xmltype(xmlschema));
2772 : }
2773 :
2774 :
2775 : Datum
2776 16 : table_to_xml_and_xmlschema(PG_FUNCTION_ARGS)
2777 : {
2778 16 : Oid relid = PG_GETARG_OID(0);
2779 16 : bool nulls = PG_GETARG_BOOL(1);
2780 16 : bool tableforest = PG_GETARG_BOOL(2);
2781 16 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2782 : Relation rel;
2783 : const char *xmlschema;
2784 :
2785 16 : rel = table_open(relid, AccessShareLock);
2786 16 : xmlschema = map_sql_table_to_xmlschema(rel->rd_att, relid, nulls,
2787 : tableforest, targetns);
2788 16 : table_close(rel, NoLock);
2789 :
2790 16 : PG_RETURN_XML_P(stringinfo_to_xmltype(table_to_xml_internal(relid,
2791 : xmlschema, nulls, tableforest,
2792 : targetns, true)));
2793 : }
2794 :
2795 :
2796 : Datum
2797 4 : query_to_xml_and_xmlschema(PG_FUNCTION_ARGS)
2798 : {
2799 4 : char *query = text_to_cstring(PG_GETARG_TEXT_PP(0));
2800 4 : bool nulls = PG_GETARG_BOOL(1);
2801 4 : bool tableforest = PG_GETARG_BOOL(2);
2802 4 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2803 :
2804 : const char *xmlschema;
2805 : SPIPlanPtr plan;
2806 : Portal portal;
2807 :
2808 4 : SPI_connect();
2809 :
2810 4 : if ((plan = SPI_prepare(query, 0, NULL)) == NULL)
2811 0 : elog(ERROR, "SPI_prepare(\"%s\") failed", query);
2812 :
2813 4 : if ((portal = SPI_cursor_open(NULL, plan, NULL, NULL, true)) == NULL)
2814 0 : elog(ERROR, "SPI_cursor_open(\"%s\") failed", query);
2815 :
2816 4 : xmlschema = _SPI_strdup(map_sql_table_to_xmlschema(portal->tupDesc,
2817 : InvalidOid, nulls, tableforest, targetns));
2818 4 : SPI_cursor_close(portal);
2819 4 : SPI_finish();
2820 :
2821 4 : PG_RETURN_XML_P(stringinfo_to_xmltype(query_to_xml_internal(query, NULL,
2822 : xmlschema, nulls, tableforest,
2823 : targetns, true)));
2824 : }
2825 :
2826 :
2827 : /*
2828 : * Map SQL schema to XML and/or XML Schema document; see SQL/XML:2008
2829 : * sections 9.13, 9.14.
2830 : */
2831 :
2832 : static StringInfo
2833 12 : schema_to_xml_internal(Oid nspid, const char *xmlschema, bool nulls,
2834 : bool tableforest, const char *targetns, bool top_level)
2835 : {
2836 : StringInfo result;
2837 : char *xmlsn;
2838 : List *relid_list;
2839 : ListCell *cell;
2840 :
2841 12 : xmlsn = map_sql_identifier_to_xml_name(get_namespace_name(nspid),
2842 : true, false);
2843 12 : result = makeStringInfo();
2844 :
2845 12 : xmldata_root_element_start(result, xmlsn, xmlschema, targetns, top_level);
2846 12 : appendStringInfoChar(result, '\n');
2847 :
2848 12 : if (xmlschema)
2849 4 : appendStringInfo(result, "%s\n\n", xmlschema);
2850 :
2851 12 : SPI_connect();
2852 :
2853 12 : relid_list = schema_get_xml_visible_tables(nspid);
2854 :
2855 36 : foreach(cell, relid_list)
2856 : {
2857 24 : Oid relid = lfirst_oid(cell);
2858 : StringInfo subres;
2859 :
2860 24 : subres = table_to_xml_internal(relid, NULL, nulls, tableforest,
2861 : targetns, false);
2862 :
2863 24 : appendBinaryStringInfo(result, subres->data, subres->len);
2864 24 : appendStringInfoChar(result, '\n');
2865 : }
2866 :
2867 12 : SPI_finish();
2868 :
2869 12 : xmldata_root_element_end(result, xmlsn);
2870 :
2871 12 : return result;
2872 : }
2873 :
2874 :
2875 : Datum
2876 8 : schema_to_xml(PG_FUNCTION_ARGS)
2877 : {
2878 8 : Name name = PG_GETARG_NAME(0);
2879 8 : bool nulls = PG_GETARG_BOOL(1);
2880 8 : bool tableforest = PG_GETARG_BOOL(2);
2881 8 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2882 :
2883 : char *schemaname;
2884 : Oid nspid;
2885 :
2886 8 : schemaname = NameStr(*name);
2887 8 : nspid = LookupExplicitNamespace(schemaname, false);
2888 :
2889 8 : PG_RETURN_XML_P(stringinfo_to_xmltype(schema_to_xml_internal(nspid, NULL,
2890 : nulls, tableforest, targetns, true)));
2891 : }
2892 :
2893 :
2894 : /*
2895 : * Write the start element of the root element of an XML Schema mapping.
2896 : */
2897 : static void
2898 64 : xsd_schema_element_start(StringInfo result, const char *targetns)
2899 : {
2900 64 : appendStringInfoString(result,
2901 : "<xsd:schema\n"
2902 : " xmlns:xsd=\"" NAMESPACE_XSD "\"");
2903 64 : if (strlen(targetns) > 0)
2904 12 : appendStringInfo(result,
2905 : "\n"
2906 : " targetNamespace=\"%s\"\n"
2907 : " elementFormDefault=\"qualified\"",
2908 : targetns);
2909 64 : appendStringInfoString(result,
2910 : ">\n\n");
2911 64 : }
2912 :
2913 :
2914 : static void
2915 64 : xsd_schema_element_end(StringInfo result)
2916 : {
2917 64 : appendStringInfoString(result, "</xsd:schema>");
2918 64 : }
2919 :
2920 :
2921 : static StringInfo
2922 12 : schema_to_xmlschema_internal(const char *schemaname, bool nulls,
2923 : bool tableforest, const char *targetns)
2924 : {
2925 : Oid nspid;
2926 : List *relid_list;
2927 : List *tupdesc_list;
2928 : ListCell *cell;
2929 : StringInfo result;
2930 :
2931 12 : result = makeStringInfo();
2932 :
2933 12 : nspid = LookupExplicitNamespace(schemaname, false);
2934 :
2935 12 : xsd_schema_element_start(result, targetns);
2936 :
2937 12 : SPI_connect();
2938 :
2939 12 : relid_list = schema_get_xml_visible_tables(nspid);
2940 :
2941 12 : tupdesc_list = NIL;
2942 36 : foreach(cell, relid_list)
2943 : {
2944 : Relation rel;
2945 :
2946 24 : rel = table_open(lfirst_oid(cell), AccessShareLock);
2947 24 : tupdesc_list = lappend(tupdesc_list, CreateTupleDescCopy(rel->rd_att));
2948 24 : table_close(rel, NoLock);
2949 : }
2950 :
2951 12 : appendStringInfoString(result,
2952 : map_sql_typecoll_to_xmlschema_types(tupdesc_list));
2953 :
2954 12 : appendStringInfoString(result,
2955 : map_sql_schema_to_xmlschema_types(nspid, relid_list,
2956 : nulls, tableforest, targetns));
2957 :
2958 12 : xsd_schema_element_end(result);
2959 :
2960 12 : SPI_finish();
2961 :
2962 12 : return result;
2963 : }
2964 :
2965 :
2966 : Datum
2967 8 : schema_to_xmlschema(PG_FUNCTION_ARGS)
2968 : {
2969 8 : Name name = PG_GETARG_NAME(0);
2970 8 : bool nulls = PG_GETARG_BOOL(1);
2971 8 : bool tableforest = PG_GETARG_BOOL(2);
2972 8 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2973 :
2974 8 : PG_RETURN_XML_P(stringinfo_to_xmltype(schema_to_xmlschema_internal(NameStr(*name),
2975 : nulls, tableforest, targetns)));
2976 : }
2977 :
2978 :
2979 : Datum
2980 4 : schema_to_xml_and_xmlschema(PG_FUNCTION_ARGS)
2981 : {
2982 4 : Name name = PG_GETARG_NAME(0);
2983 4 : bool nulls = PG_GETARG_BOOL(1);
2984 4 : bool tableforest = PG_GETARG_BOOL(2);
2985 4 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(3));
2986 : char *schemaname;
2987 : Oid nspid;
2988 : StringInfo xmlschema;
2989 :
2990 4 : schemaname = NameStr(*name);
2991 4 : nspid = LookupExplicitNamespace(schemaname, false);
2992 :
2993 4 : xmlschema = schema_to_xmlschema_internal(schemaname, nulls,
2994 : tableforest, targetns);
2995 :
2996 4 : PG_RETURN_XML_P(stringinfo_to_xmltype(schema_to_xml_internal(nspid,
2997 : xmlschema->data, nulls,
2998 : tableforest, targetns, true)));
2999 : }
3000 :
3001 :
3002 : /*
3003 : * Map SQL database to XML and/or XML Schema document; see SQL/XML:2008
3004 : * sections 9.16, 9.17.
3005 : */
3006 :
3007 : static StringInfo
3008 0 : database_to_xml_internal(const char *xmlschema, bool nulls,
3009 : bool tableforest, const char *targetns)
3010 : {
3011 : StringInfo result;
3012 : List *nspid_list;
3013 : ListCell *cell;
3014 : char *xmlcn;
3015 :
3016 0 : xmlcn = map_sql_identifier_to_xml_name(get_database_name(MyDatabaseId),
3017 : true, false);
3018 0 : result = makeStringInfo();
3019 :
3020 0 : xmldata_root_element_start(result, xmlcn, xmlschema, targetns, true);
3021 0 : appendStringInfoChar(result, '\n');
3022 :
3023 0 : if (xmlschema)
3024 0 : appendStringInfo(result, "%s\n\n", xmlschema);
3025 :
3026 0 : SPI_connect();
3027 :
3028 0 : nspid_list = database_get_xml_visible_schemas();
3029 :
3030 0 : foreach(cell, nspid_list)
3031 : {
3032 0 : Oid nspid = lfirst_oid(cell);
3033 : StringInfo subres;
3034 :
3035 0 : subres = schema_to_xml_internal(nspid, NULL, nulls,
3036 : tableforest, targetns, false);
3037 :
3038 0 : appendBinaryStringInfo(result, subres->data, subres->len);
3039 0 : appendStringInfoChar(result, '\n');
3040 : }
3041 :
3042 0 : SPI_finish();
3043 :
3044 0 : xmldata_root_element_end(result, xmlcn);
3045 :
3046 0 : return result;
3047 : }
3048 :
3049 :
3050 : Datum
3051 0 : database_to_xml(PG_FUNCTION_ARGS)
3052 : {
3053 0 : bool nulls = PG_GETARG_BOOL(0);
3054 0 : bool tableforest = PG_GETARG_BOOL(1);
3055 0 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(2));
3056 :
3057 0 : PG_RETURN_XML_P(stringinfo_to_xmltype(database_to_xml_internal(NULL, nulls,
3058 : tableforest, targetns)));
3059 : }
3060 :
3061 :
3062 : static StringInfo
3063 0 : database_to_xmlschema_internal(bool nulls, bool tableforest,
3064 : const char *targetns)
3065 : {
3066 : List *relid_list;
3067 : List *nspid_list;
3068 : List *tupdesc_list;
3069 : ListCell *cell;
3070 : StringInfo result;
3071 :
3072 0 : result = makeStringInfo();
3073 :
3074 0 : xsd_schema_element_start(result, targetns);
3075 :
3076 0 : SPI_connect();
3077 :
3078 0 : relid_list = database_get_xml_visible_tables();
3079 0 : nspid_list = database_get_xml_visible_schemas();
3080 :
3081 0 : tupdesc_list = NIL;
3082 0 : foreach(cell, relid_list)
3083 : {
3084 : Relation rel;
3085 :
3086 0 : rel = table_open(lfirst_oid(cell), AccessShareLock);
3087 0 : tupdesc_list = lappend(tupdesc_list, CreateTupleDescCopy(rel->rd_att));
3088 0 : table_close(rel, NoLock);
3089 : }
3090 :
3091 0 : appendStringInfoString(result,
3092 : map_sql_typecoll_to_xmlschema_types(tupdesc_list));
3093 :
3094 0 : appendStringInfoString(result,
3095 : map_sql_catalog_to_xmlschema_types(nspid_list, nulls, tableforest, targetns));
3096 :
3097 0 : xsd_schema_element_end(result);
3098 :
3099 0 : SPI_finish();
3100 :
3101 0 : return result;
3102 : }
3103 :
3104 :
3105 : Datum
3106 0 : database_to_xmlschema(PG_FUNCTION_ARGS)
3107 : {
3108 0 : bool nulls = PG_GETARG_BOOL(0);
3109 0 : bool tableforest = PG_GETARG_BOOL(1);
3110 0 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(2));
3111 :
3112 0 : PG_RETURN_XML_P(stringinfo_to_xmltype(database_to_xmlschema_internal(nulls,
3113 : tableforest, targetns)));
3114 : }
3115 :
3116 :
3117 : Datum
3118 0 : database_to_xml_and_xmlschema(PG_FUNCTION_ARGS)
3119 : {
3120 0 : bool nulls = PG_GETARG_BOOL(0);
3121 0 : bool tableforest = PG_GETARG_BOOL(1);
3122 0 : const char *targetns = text_to_cstring(PG_GETARG_TEXT_PP(2));
3123 : StringInfo xmlschema;
3124 :
3125 0 : xmlschema = database_to_xmlschema_internal(nulls, tableforest, targetns);
3126 :
3127 0 : PG_RETURN_XML_P(stringinfo_to_xmltype(database_to_xml_internal(xmlschema->data,
3128 : nulls, tableforest, targetns)));
3129 : }
3130 :
3131 :
3132 : /*
3133 : * Map a multi-part SQL name to an XML name; see SQL/XML:2008 section
3134 : * 9.2.
3135 : */
3136 : static char *
3137 256 : map_multipart_sql_identifier_to_xml_name(const char *a, const char *b, const char *c, const char *d)
3138 : {
3139 : StringInfoData result;
3140 :
3141 256 : initStringInfo(&result);
3142 :
3143 256 : if (a)
3144 256 : appendStringInfoString(&result,
3145 256 : map_sql_identifier_to_xml_name(a, true, true));
3146 256 : if (b)
3147 256 : appendStringInfo(&result, ".%s",
3148 : map_sql_identifier_to_xml_name(b, true, true));
3149 256 : if (c)
3150 256 : appendStringInfo(&result, ".%s",
3151 : map_sql_identifier_to_xml_name(c, true, true));
3152 256 : if (d)
3153 244 : appendStringInfo(&result, ".%s",
3154 : map_sql_identifier_to_xml_name(d, true, true));
3155 :
3156 256 : return result.data;
3157 : }
3158 :
3159 :
3160 : /*
3161 : * Map an SQL table to an XML Schema document; see SQL/XML:2008
3162 : * section 9.11.
3163 : *
3164 : * Map an SQL table to XML Schema data types; see SQL/XML:2008 section
3165 : * 9.9.
3166 : */
3167 : static const char *
3168 52 : map_sql_table_to_xmlschema(TupleDesc tupdesc, Oid relid, bool nulls,
3169 : bool tableforest, const char *targetns)
3170 : {
3171 : int i;
3172 : char *xmltn;
3173 : char *tabletypename;
3174 : char *rowtypename;
3175 : StringInfoData result;
3176 :
3177 52 : initStringInfo(&result);
3178 :
3179 52 : if (OidIsValid(relid))
3180 : {
3181 : HeapTuple tuple;
3182 : Form_pg_class reltuple;
3183 :
3184 36 : tuple = SearchSysCache1(RELOID, ObjectIdGetDatum(relid));
3185 36 : if (!HeapTupleIsValid(tuple))
3186 0 : elog(ERROR, "cache lookup failed for relation %u", relid);
3187 36 : reltuple = (Form_pg_class) GETSTRUCT(tuple);
3188 :
3189 36 : xmltn = map_sql_identifier_to_xml_name(NameStr(reltuple->relname),
3190 : true, false);
3191 :
3192 36 : tabletypename = map_multipart_sql_identifier_to_xml_name("TableType",
3193 36 : get_database_name(MyDatabaseId),
3194 36 : get_namespace_name(reltuple->relnamespace),
3195 36 : NameStr(reltuple->relname));
3196 :
3197 36 : rowtypename = map_multipart_sql_identifier_to_xml_name("RowType",
3198 36 : get_database_name(MyDatabaseId),
3199 36 : get_namespace_name(reltuple->relnamespace),
3200 36 : NameStr(reltuple->relname));
3201 :
3202 36 : ReleaseSysCache(tuple);
3203 : }
3204 : else
3205 : {
3206 16 : if (tableforest)
3207 8 : xmltn = "row";
3208 : else
3209 8 : xmltn = "table";
3210 :
3211 16 : tabletypename = "TableType";
3212 16 : rowtypename = "RowType";
3213 : }
3214 :
3215 52 : xsd_schema_element_start(&result, targetns);
3216 :
3217 52 : appendStringInfoString(&result,
3218 52 : map_sql_typecoll_to_xmlschema_types(list_make1(tupdesc)));
3219 :
3220 52 : appendStringInfo(&result,
3221 : "<xsd:complexType name=\"%s\">\n"
3222 : " <xsd:sequence>\n",
3223 : rowtypename);
3224 :
3225 208 : for (i = 0; i < tupdesc->natts; i++)
3226 : {
3227 156 : Form_pg_attribute att = TupleDescAttr(tupdesc, i);
3228 :
3229 156 : if (att->attisdropped)
3230 4 : continue;
3231 304 : appendStringInfo(&result,
3232 : " <xsd:element name=\"%s\" type=\"%s\"%s></xsd:element>\n",
3233 152 : map_sql_identifier_to_xml_name(NameStr(att->attname),
3234 : true, false),
3235 : map_sql_type_to_xml_name(att->atttypid, -1),
3236 : nulls ? " nillable=\"true\"" : " minOccurs=\"0\"");
3237 : }
3238 :
3239 52 : appendStringInfoString(&result,
3240 : " </xsd:sequence>\n"
3241 : "</xsd:complexType>\n\n");
3242 :
3243 52 : if (!tableforest)
3244 : {
3245 28 : appendStringInfo(&result,
3246 : "<xsd:complexType name=\"%s\">\n"
3247 : " <xsd:sequence>\n"
3248 : " <xsd:element name=\"row\" type=\"%s\" minOccurs=\"0\" maxOccurs=\"unbounded\"/>\n"
3249 : " </xsd:sequence>\n"
3250 : "</xsd:complexType>\n\n",
3251 : tabletypename, rowtypename);
3252 :
3253 28 : appendStringInfo(&result,
3254 : "<xsd:element name=\"%s\" type=\"%s\"/>\n\n",
3255 : xmltn, tabletypename);
3256 : }
3257 : else
3258 24 : appendStringInfo(&result,
3259 : "<xsd:element name=\"%s\" type=\"%s\"/>\n\n",
3260 : xmltn, rowtypename);
3261 :
3262 52 : xsd_schema_element_end(&result);
3263 :
3264 52 : return result.data;
3265 : }
3266 :
3267 :
3268 : /*
3269 : * Map an SQL schema to XML Schema data types; see SQL/XML:2008
3270 : * section 9.12.
3271 : */
3272 : static const char *
3273 12 : map_sql_schema_to_xmlschema_types(Oid nspid, List *relid_list, bool nulls,
3274 : bool tableforest, const char *targetns)
3275 : {
3276 : char *dbname;
3277 : char *nspname;
3278 : char *xmlsn;
3279 : char *schematypename;
3280 : StringInfoData result;
3281 : ListCell *cell;
3282 :
3283 12 : dbname = get_database_name(MyDatabaseId);
3284 12 : nspname = get_namespace_name(nspid);
3285 :
3286 12 : initStringInfo(&result);
3287 :
3288 12 : xmlsn = map_sql_identifier_to_xml_name(nspname, true, false);
3289 :
3290 12 : schematypename = map_multipart_sql_identifier_to_xml_name("SchemaType",
3291 : dbname,
3292 : nspname,
3293 : NULL);
3294 :
3295 12 : appendStringInfo(&result,
3296 : "<xsd:complexType name=\"%s\">\n", schematypename);
3297 12 : if (!tableforest)
3298 4 : appendStringInfoString(&result,
3299 : " <xsd:all>\n");
3300 : else
3301 8 : appendStringInfoString(&result,
3302 : " <xsd:sequence>\n");
3303 :
3304 36 : foreach(cell, relid_list)
3305 : {
3306 24 : Oid relid = lfirst_oid(cell);
3307 24 : char *relname = get_rel_name(relid);
3308 24 : char *xmltn = map_sql_identifier_to_xml_name(relname, true, false);
3309 24 : char *tabletypename = map_multipart_sql_identifier_to_xml_name(tableforest ? "RowType" : "TableType",
3310 : dbname,
3311 : nspname,
3312 : relname);
3313 :
3314 24 : if (!tableforest)
3315 8 : appendStringInfo(&result,
3316 : " <xsd:element name=\"%s\" type=\"%s\"/>\n",
3317 : xmltn, tabletypename);
3318 : else
3319 16 : appendStringInfo(&result,
3320 : " <xsd:element name=\"%s\" type=\"%s\" minOccurs=\"0\" maxOccurs=\"unbounded\"/>\n",
3321 : xmltn, tabletypename);
3322 : }
3323 :
3324 12 : if (!tableforest)
3325 4 : appendStringInfoString(&result,
3326 : " </xsd:all>\n");
3327 : else
3328 8 : appendStringInfoString(&result,
3329 : " </xsd:sequence>\n");
3330 12 : appendStringInfoString(&result,
3331 : "</xsd:complexType>\n\n");
3332 :
3333 12 : appendStringInfo(&result,
3334 : "<xsd:element name=\"%s\" type=\"%s\"/>\n\n",
3335 : xmlsn, schematypename);
3336 :
3337 12 : return result.data;
3338 : }
3339 :
3340 :
3341 : /*
3342 : * Map an SQL catalog to XML Schema data types; see SQL/XML:2008
3343 : * section 9.15.
3344 : */
3345 : static const char *
3346 0 : map_sql_catalog_to_xmlschema_types(List *nspid_list, bool nulls,
3347 : bool tableforest, const char *targetns)
3348 : {
3349 : char *dbname;
3350 : char *xmlcn;
3351 : char *catalogtypename;
3352 : StringInfoData result;
3353 : ListCell *cell;
3354 :
3355 0 : dbname = get_database_name(MyDatabaseId);
3356 :
3357 0 : initStringInfo(&result);
3358 :
3359 0 : xmlcn = map_sql_identifier_to_xml_name(dbname, true, false);
3360 :
3361 0 : catalogtypename = map_multipart_sql_identifier_to_xml_name("CatalogType",
3362 : dbname,
3363 : NULL,
3364 : NULL);
3365 :
3366 0 : appendStringInfo(&result,
3367 : "<xsd:complexType name=\"%s\">\n", catalogtypename);
3368 0 : appendStringInfoString(&result,
3369 : " <xsd:all>\n");
3370 :
3371 0 : foreach(cell, nspid_list)
3372 : {
3373 0 : Oid nspid = lfirst_oid(cell);
3374 0 : char *nspname = get_namespace_name(nspid);
3375 0 : char *xmlsn = map_sql_identifier_to_xml_name(nspname, true, false);
3376 0 : char *schematypename = map_multipart_sql_identifier_to_xml_name("SchemaType",
3377 : dbname,
3378 : nspname,
3379 : NULL);
3380 :
3381 0 : appendStringInfo(&result,
3382 : " <xsd:element name=\"%s\" type=\"%s\"/>\n",
3383 : xmlsn, schematypename);
3384 : }
3385 :
3386 0 : appendStringInfoString(&result,
3387 : " </xsd:all>\n");
3388 0 : appendStringInfoString(&result,
3389 : "</xsd:complexType>\n\n");
3390 :
3391 0 : appendStringInfo(&result,
3392 : "<xsd:element name=\"%s\" type=\"%s\"/>\n\n",
3393 : xmlcn, catalogtypename);
3394 :
3395 0 : return result.data;
3396 : }
3397 :
3398 :
3399 : /*
3400 : * Map an SQL data type to an XML name; see SQL/XML:2008 section 9.4.
3401 : */
3402 : static const char *
3403 500 : map_sql_type_to_xml_name(Oid typeoid, int typmod)
3404 : {
3405 : StringInfoData result;
3406 :
3407 500 : initStringInfo(&result);
3408 :
3409 500 : switch (typeoid)
3410 : {
3411 20 : case BPCHAROID:
3412 20 : if (typmod == -1)
3413 20 : appendStringInfoString(&result, "CHAR");
3414 : else
3415 0 : appendStringInfo(&result, "CHAR_%d", typmod - VARHDRSZ);
3416 20 : break;
3417 36 : case VARCHAROID:
3418 36 : if (typmod == -1)
3419 36 : appendStringInfoString(&result, "VARCHAR");
3420 : else
3421 0 : appendStringInfo(&result, "VARCHAR_%d", typmod - VARHDRSZ);
3422 36 : break;
3423 20 : case NUMERICOID:
3424 20 : if (typmod == -1)
3425 20 : appendStringInfoString(&result, "NUMERIC");
3426 : else
3427 0 : appendStringInfo(&result, "NUMERIC_%d_%d",
3428 0 : ((typmod - VARHDRSZ) >> 16) & 0xffff,
3429 0 : (typmod - VARHDRSZ) & 0xffff);
3430 20 : break;
3431 116 : case INT4OID:
3432 116 : appendStringInfoString(&result, "INTEGER");
3433 116 : break;
3434 20 : case INT2OID:
3435 20 : appendStringInfoString(&result, "SMALLINT");
3436 20 : break;
3437 20 : case INT8OID:
3438 20 : appendStringInfoString(&result, "BIGINT");
3439 20 : break;
3440 20 : case FLOAT4OID:
3441 20 : appendStringInfoString(&result, "REAL");
3442 20 : break;
3443 0 : case FLOAT8OID:
3444 0 : appendStringInfoString(&result, "DOUBLE");
3445 0 : break;
3446 20 : case BOOLOID:
3447 20 : appendStringInfoString(&result, "BOOLEAN");
3448 20 : break;
3449 20 : case TIMEOID:
3450 20 : if (typmod == -1)
3451 20 : appendStringInfoString(&result, "TIME");
3452 : else
3453 0 : appendStringInfo(&result, "TIME_%d", typmod);
3454 20 : break;
3455 0 : case TIMETZOID:
3456 0 : if (typmod == -1)
3457 0 : appendStringInfoString(&result, "TIME_WTZ");
3458 : else
3459 0 : appendStringInfo(&result, "TIME_WTZ_%d", typmod);
3460 0 : break;
3461 20 : case TIMESTAMPOID:
3462 20 : if (typmod == -1)
3463 20 : appendStringInfoString(&result, "TIMESTAMP");
3464 : else
3465 0 : appendStringInfo(&result, "TIMESTAMP_%d", typmod);
3466 20 : break;
3467 0 : case TIMESTAMPTZOID:
3468 0 : if (typmod == -1)
3469 0 : appendStringInfoString(&result, "TIMESTAMP_WTZ");
3470 : else
3471 0 : appendStringInfo(&result, "TIMESTAMP_WTZ_%d", typmod);
3472 0 : break;
3473 20 : case DATEOID:
3474 20 : appendStringInfoString(&result, "DATE");
3475 20 : break;
3476 20 : case XMLOID:
3477 20 : appendStringInfoString(&result, "XML");
3478 20 : break;
3479 148 : default:
3480 : {
3481 : HeapTuple tuple;
3482 : Form_pg_type typtuple;
3483 :
3484 148 : tuple = SearchSysCache1(TYPEOID, ObjectIdGetDatum(typeoid));
3485 148 : if (!HeapTupleIsValid(tuple))
3486 0 : elog(ERROR, "cache lookup failed for type %u", typeoid);
3487 148 : typtuple = (Form_pg_type) GETSTRUCT(tuple);
3488 :
3489 148 : appendStringInfoString(&result,
3490 148 : map_multipart_sql_identifier_to_xml_name((typtuple->typtype == TYPTYPE_DOMAIN) ? "Domain" : "UDT",
3491 148 : get_database_name(MyDatabaseId),
3492 148 : get_namespace_name(typtuple->typnamespace),
3493 148 : NameStr(typtuple->typname)));
3494 :
3495 148 : ReleaseSysCache(tuple);
3496 : }
3497 : }
3498 :
3499 500 : return result.data;
3500 : }
3501 :
3502 :
3503 : /*
3504 : * Map a collection of SQL data types to XML Schema data types; see
3505 : * SQL/XML:2008 section 9.7.
3506 : */
3507 : static const char *
3508 64 : map_sql_typecoll_to_xmlschema_types(List *tupdesc_list)
3509 : {
3510 64 : List *uniquetypes = NIL;
3511 : int i;
3512 : StringInfoData result;
3513 : ListCell *cell0;
3514 :
3515 : /* extract all column types used in the set of TupleDescs */
3516 140 : foreach(cell0, tupdesc_list)
3517 : {
3518 76 : TupleDesc tupdesc = (TupleDesc) lfirst(cell0);
3519 :
3520 436 : for (i = 0; i < tupdesc->natts; i++)
3521 : {
3522 360 : Form_pg_attribute att = TupleDescAttr(tupdesc, i);
3523 :
3524 360 : if (att->attisdropped)
3525 16 : continue;
3526 344 : uniquetypes = list_append_unique_oid(uniquetypes, att->atttypid);
3527 : }
3528 : }
3529 :
3530 : /* add base types of domains */
3531 396 : foreach(cell0, uniquetypes)
3532 : {
3533 332 : Oid typid = lfirst_oid(cell0);
3534 332 : Oid basetypid = getBaseType(typid);
3535 :
3536 332 : if (basetypid != typid)
3537 16 : uniquetypes = list_append_unique_oid(uniquetypes, basetypid);
3538 : }
3539 :
3540 : /* Convert to textual form */
3541 64 : initStringInfo(&result);
3542 :
3543 396 : foreach(cell0, uniquetypes)
3544 : {
3545 332 : appendStringInfo(&result, "%s\n",
3546 : map_sql_type_to_xmlschema_type(lfirst_oid(cell0),
3547 : -1));
3548 : }
3549 :
3550 64 : return result.data;
3551 : }
3552 :
3553 :
3554 : /*
3555 : * Map an SQL data type to a named XML Schema data type; see
3556 : * SQL/XML:2008 sections 9.5 and 9.6.
3557 : *
3558 : * (The distinction between 9.5 and 9.6 is basically that 9.6 adds
3559 : * a name attribute, which this function does. The name-less version
3560 : * 9.5 doesn't appear to be required anywhere.)
3561 : */
3562 : static const char *
3563 332 : map_sql_type_to_xmlschema_type(Oid typeoid, int typmod)
3564 : {
3565 : StringInfoData result;
3566 332 : const char *typename = map_sql_type_to_xml_name(typeoid, typmod);
3567 :
3568 332 : initStringInfo(&result);
3569 :
3570 332 : if (typeoid == XMLOID)
3571 : {
3572 16 : appendStringInfoString(&result,
3573 : "<xsd:complexType mixed=\"true\">\n"
3574 : " <xsd:sequence>\n"
3575 : " <xsd:any name=\"element\" minOccurs=\"0\" maxOccurs=\"unbounded\" processContents=\"skip\"/>\n"
3576 : " </xsd:sequence>\n"
3577 : "</xsd:complexType>\n");
3578 : }
3579 : else
3580 : {
3581 316 : appendStringInfo(&result,
3582 : "<xsd:simpleType name=\"%s\">\n", typename);
3583 :
3584 316 : switch (typeoid)
3585 : {
3586 92 : case BPCHAROID:
3587 : case VARCHAROID:
3588 : case TEXTOID:
3589 92 : appendStringInfoString(&result,
3590 : " <xsd:restriction base=\"xsd:string\">\n");
3591 92 : if (typmod != -1)
3592 0 : appendStringInfo(&result,
3593 : " <xsd:maxLength value=\"%d\"/>\n",
3594 : typmod - VARHDRSZ);
3595 92 : appendStringInfoString(&result, " </xsd:restriction>\n");
3596 92 : break;
3597 :
3598 16 : case BYTEAOID:
3599 16 : appendStringInfo(&result,
3600 : " <xsd:restriction base=\"xsd:%s\">\n"
3601 : " </xsd:restriction>\n",
3602 16 : xmlbinary == XMLBINARY_BASE64 ? "base64Binary" : "hexBinary");
3603 16 : break;
3604 :
3605 16 : case NUMERICOID:
3606 16 : if (typmod != -1)
3607 0 : appendStringInfo(&result,
3608 : " <xsd:restriction base=\"xsd:decimal\">\n"
3609 : " <xsd:totalDigits value=\"%d\"/>\n"
3610 : " <xsd:fractionDigits value=\"%d\"/>\n"
3611 : " </xsd:restriction>\n",
3612 0 : ((typmod - VARHDRSZ) >> 16) & 0xffff,
3613 0 : (typmod - VARHDRSZ) & 0xffff);
3614 16 : break;
3615 :
3616 16 : case INT2OID:
3617 16 : appendStringInfo(&result,
3618 : " <xsd:restriction base=\"xsd:short\">\n"
3619 : " <xsd:maxInclusive value=\"%d\"/>\n"
3620 : " <xsd:minInclusive value=\"%d\"/>\n"
3621 : " </xsd:restriction>\n",
3622 : SHRT_MAX, SHRT_MIN);
3623 16 : break;
3624 :
3625 64 : case INT4OID:
3626 64 : appendStringInfo(&result,
3627 : " <xsd:restriction base=\"xsd:int\">\n"
3628 : " <xsd:maxInclusive value=\"%d\"/>\n"
3629 : " <xsd:minInclusive value=\"%d\"/>\n"
3630 : " </xsd:restriction>\n",
3631 : INT_MAX, INT_MIN);
3632 64 : break;
3633 :
3634 16 : case INT8OID:
3635 16 : appendStringInfo(&result,
3636 : " <xsd:restriction base=\"xsd:long\">\n"
3637 : " <xsd:maxInclusive value=\"" INT64_FORMAT "\"/>\n"
3638 : " <xsd:minInclusive value=\"" INT64_FORMAT "\"/>\n"
3639 : " </xsd:restriction>\n",
3640 : (((uint64) 1) << (sizeof(int64) * 8 - 1)) - 1,
3641 : (((uint64) 1) << (sizeof(int64) * 8 - 1)));
3642 16 : break;
3643 :
3644 16 : case FLOAT4OID:
3645 16 : appendStringInfoString(&result,
3646 : " <xsd:restriction base=\"xsd:float\"></xsd:restriction>\n");
3647 16 : break;
3648 :
3649 0 : case FLOAT8OID:
3650 0 : appendStringInfoString(&result,
3651 : " <xsd:restriction base=\"xsd:double\"></xsd:restriction>\n");
3652 0 : break;
3653 :
3654 16 : case BOOLOID:
3655 16 : appendStringInfoString(&result,
3656 : " <xsd:restriction base=\"xsd:boolean\"></xsd:restriction>\n");
3657 16 : break;
3658 :
3659 16 : case TIMEOID:
3660 : case TIMETZOID:
3661 : {
3662 16 : const char *tz = (typeoid == TIMETZOID ? "(+|-)\\p{Nd}{2}:\\p{Nd}{2}" : "");
3663 :
3664 16 : if (typmod == -1)
3665 16 : appendStringInfo(&result,
3666 : " <xsd:restriction base=\"xsd:time\">\n"
3667 : " <xsd:pattern value=\"\\p{Nd}{2}:\\p{Nd}{2}:\\p{Nd}{2}(.\\p{Nd}+)?%s\"/>\n"
3668 : " </xsd:restriction>\n", tz);
3669 0 : else if (typmod == 0)
3670 0 : appendStringInfo(&result,
3671 : " <xsd:restriction base=\"xsd:time\">\n"
3672 : " <xsd:pattern value=\"\\p{Nd}{2}:\\p{Nd}{2}:\\p{Nd}{2}%s\"/>\n"
3673 : " </xsd:restriction>\n", tz);
3674 : else
3675 0 : appendStringInfo(&result,
3676 : " <xsd:restriction base=\"xsd:time\">\n"
3677 : " <xsd:pattern value=\"\\p{Nd}{2}:\\p{Nd}{2}:\\p{Nd}{2}.\\p{Nd}{%d}%s\"/>\n"
3678 : " </xsd:restriction>\n", typmod - VARHDRSZ, tz);
3679 16 : break;
3680 : }
3681 :
3682 16 : case TIMESTAMPOID:
3683 : case TIMESTAMPTZOID:
3684 : {
3685 16 : const char *tz = (typeoid == TIMESTAMPTZOID ? "(+|-)\\p{Nd}{2}:\\p{Nd}{2}" : "");
3686 :
3687 16 : if (typmod == -1)
3688 16 : appendStringInfo(&result,
3689 : " <xsd:restriction base=\"xsd:dateTime\">\n"
3690 : " <xsd:pattern value=\"\\p{Nd}{4}-\\p{Nd}{2}-\\p{Nd}{2}T\\p{Nd}{2}:\\p{Nd}{2}:\\p{Nd}{2}(.\\p{Nd}+)?%s\"/>\n"
3691 : " </xsd:restriction>\n", tz);
3692 0 : else if (typmod == 0)
3693 0 : appendStringInfo(&result,
3694 : " <xsd:restriction base=\"xsd:dateTime\">\n"
3695 : " <xsd:pattern value=\"\\p{Nd}{4}-\\p{Nd}{2}-\\p{Nd}{2}T\\p{Nd}{2}:\\p{Nd}{2}:\\p{Nd}{2}%s\"/>\n"
3696 : " </xsd:restriction>\n", tz);
3697 : else
3698 0 : appendStringInfo(&result,
3699 : " <xsd:restriction base=\"xsd:dateTime\">\n"
3700 : " <xsd:pattern value=\"\\p{Nd}{4}-\\p{Nd}{2}-\\p{Nd}{2}T\\p{Nd}{2}:\\p{Nd}{2}:\\p{Nd}{2}.\\p{Nd}{%d}%s\"/>\n"
3701 : " </xsd:restriction>\n", typmod - VARHDRSZ, tz);
3702 16 : break;
3703 : }
3704 :
3705 16 : case DATEOID:
3706 16 : appendStringInfoString(&result,
3707 : " <xsd:restriction base=\"xsd:date\">\n"
3708 : " <xsd:pattern value=\"\\p{Nd}{4}-\\p{Nd}{2}-\\p{Nd}{2}\"/>\n"
3709 : " </xsd:restriction>\n");
3710 16 : break;
3711 :
3712 16 : default:
3713 16 : if (get_typtype(typeoid) == TYPTYPE_DOMAIN)
3714 : {
3715 : Oid base_typeoid;
3716 16 : int32 base_typmod = -1;
3717 :
3718 16 : base_typeoid = getBaseTypeAndTypmod(typeoid, &base_typmod);
3719 :
3720 16 : appendStringInfo(&result,
3721 : " <xsd:restriction base=\"%s\"/>\n",
3722 : map_sql_type_to_xml_name(base_typeoid, base_typmod));
3723 : }
3724 16 : break;
3725 : }
3726 316 : appendStringInfoString(&result, "</xsd:simpleType>\n");
3727 : }
3728 :
3729 332 : return result.data;
3730 : }
3731 :
3732 :
3733 : /*
3734 : * Map an SQL row to an XML element, taking the row from the active
3735 : * SPI cursor. See also SQL/XML:2008 section 9.10.
3736 : */
3737 : static void
3738 212 : SPI_sql_row_to_xmlelement(uint64 rownum, StringInfo result, char *tablename,
3739 : bool nulls, bool tableforest,
3740 : const char *targetns, bool top_level)
3741 : {
3742 : int i;
3743 : char *xmltn;
3744 :
3745 212 : if (tablename)
3746 152 : xmltn = map_sql_identifier_to_xml_name(tablename, true, false);
3747 : else
3748 : {
3749 60 : if (tableforest)
3750 24 : xmltn = "row";
3751 : else
3752 36 : xmltn = "table";
3753 : }
3754 :
3755 212 : if (tableforest)
3756 108 : xmldata_root_element_start(result, xmltn, NULL, targetns, top_level);
3757 : else
3758 104 : appendStringInfoString(result, "<row>\n");
3759 :
3760 824 : for (i = 1; i <= SPI_tuptable->tupdesc->natts; i++)
3761 : {
3762 : char *colname;
3763 : Datum colval;
3764 : bool isnull;
3765 :
3766 612 : colname = map_sql_identifier_to_xml_name(SPI_fname(SPI_tuptable->tupdesc, i),
3767 : true, false);
3768 612 : colval = SPI_getbinval(SPI_tuptable->vals[rownum],
3769 612 : SPI_tuptable->tupdesc,
3770 : i,
3771 : &isnull);
3772 612 : if (isnull)
3773 : {
3774 76 : if (nulls)
3775 40 : appendStringInfo(result, " <%s xsi:nil=\"true\"/>\n", colname);
3776 : }
3777 : else
3778 536 : appendStringInfo(result, " <%s>%s</%s>\n",
3779 : colname,
3780 : map_sql_value_to_xml_value(colval,
3781 536 : SPI_gettypeid(SPI_tuptable->tupdesc, i), true),
3782 : colname);
3783 : }
3784 :
3785 212 : if (tableforest)
3786 : {
3787 108 : xmldata_root_element_end(result, xmltn);
3788 108 : appendStringInfoChar(result, '\n');
3789 : }
3790 : else
3791 104 : appendStringInfoString(result, "</row>\n\n");
3792 212 : }
3793 :
3794 :
3795 : /*
3796 : * XPath related functions
3797 : */
3798 :
3799 : #ifdef USE_LIBXML
3800 :
3801 : /*
3802 : * Convert XML node to text.
3803 : *
3804 : * For attribute and text nodes, return the escaped text. For anything else,
3805 : * dump the whole subtree.
3806 : */
3807 : static text *
3808 148 : xml_xmlnodetoxmltype(xmlNodePtr cur, PgXmlErrorContext *xmlerrcxt)
3809 : {
3810 148 : xmltype *result = NULL;
3811 :
3812 148 : if (cur->type != XML_ATTRIBUTE_NODE && cur->type != XML_TEXT_NODE)
3813 120 : {
3814 120 : void (*volatile nodefree) (xmlNodePtr) = NULL;
3815 120 : volatile xmlBufferPtr buf = NULL;
3816 120 : volatile xmlNodePtr cur_copy = NULL;
3817 :
3818 120 : PG_TRY();
3819 : {
3820 : int bytes;
3821 :
3822 120 : buf = xmlBufferCreate();
3823 120 : if (buf == NULL || xmlerrcxt->err_occurred)
3824 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
3825 : "could not allocate xmlBuffer");
3826 :
3827 : /*
3828 : * Produce a dump of the node that we can serialize. xmlNodeDump
3829 : * does that, but the result of that function won't contain
3830 : * namespace definitions from ancestor nodes, so we first do a
3831 : * xmlCopyNode() which duplicates the node along with its required
3832 : * namespace definitions.
3833 : *
3834 : * Some old libxml2 versions such as 2.7.6 produce partially
3835 : * broken XML_DOCUMENT_NODE nodes (unset content field) when
3836 : * copying them. xmlNodeDump of such a node works fine, but
3837 : * xmlFreeNode crashes; set us up to call xmlFreeDoc instead.
3838 : */
3839 120 : cur_copy = xmlCopyNode(cur, 1);
3840 120 : if (cur_copy == NULL || xmlerrcxt->err_occurred)
3841 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
3842 : "could not copy node");
3843 240 : nodefree = (cur_copy->type == XML_DOCUMENT_NODE) ?
3844 120 : (void (*) (xmlNodePtr)) xmlFreeDoc : xmlFreeNode;
3845 :
3846 120 : bytes = xmlNodeDump(buf, NULL, cur_copy, 0, 0);
3847 120 : if (bytes == -1 || xmlerrcxt->err_occurred)
3848 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
3849 : "could not dump node");
3850 :
3851 120 : result = xmlBuffer_to_xmltype(buf);
3852 : }
3853 0 : PG_FINALLY();
3854 : {
3855 120 : if (nodefree)
3856 120 : nodefree(cur_copy);
3857 120 : if (buf)
3858 120 : xmlBufferFree(buf);
3859 : }
3860 120 : PG_END_TRY();
3861 : }
3862 : else
3863 : {
3864 : xmlChar *str;
3865 :
3866 28 : str = xmlXPathCastNodeToString(cur);
3867 28 : PG_TRY();
3868 : {
3869 : /* Here we rely on XML having the same representation as TEXT */
3870 28 : char *escaped = escape_xml((char *) str);
3871 :
3872 28 : result = (xmltype *) cstring_to_text(escaped);
3873 28 : pfree(escaped);
3874 : }
3875 0 : PG_FINALLY();
3876 : {
3877 28 : xmlFree(str);
3878 : }
3879 28 : PG_END_TRY();
3880 : }
3881 :
3882 148 : return result;
3883 : }
3884 :
3885 : /*
3886 : * Convert an XML XPath object (the result of evaluating an XPath expression)
3887 : * to an array of xml values, which are appended to astate. The function
3888 : * result value is the number of elements in the array.
3889 : *
3890 : * If "astate" is NULL then we don't generate the array value, but we still
3891 : * return the number of elements it would have had.
3892 : *
3893 : * Nodesets are converted to an array containing the nodes' textual
3894 : * representations. Primitive values (float, double, string) are converted
3895 : * to a single-element array containing the value's string representation.
3896 : */
3897 : static int
3898 360 : xml_xpathobjtoxmlarray(xmlXPathObjectPtr xpathobj,
3899 : ArrayBuildState *astate,
3900 : PgXmlErrorContext *xmlerrcxt)
3901 : {
3902 360 : int result = 0;
3903 : Datum datum;
3904 : Oid datumtype;
3905 : char *result_str;
3906 :
3907 360 : switch (xpathobj->type)
3908 : {
3909 332 : case XPATH_NODESET:
3910 332 : if (xpathobj->nodesetval != NULL)
3911 : {
3912 236 : result = xpathobj->nodesetval->nodeNr;
3913 236 : if (astate != NULL)
3914 : {
3915 : int i;
3916 :
3917 112 : for (i = 0; i < result; i++)
3918 : {
3919 60 : datum = PointerGetDatum(xml_xmlnodetoxmltype(xpathobj->nodesetval->nodeTab[i],
3920 : xmlerrcxt));
3921 60 : (void) accumArrayResult(astate, datum, false,
3922 : XMLOID, CurrentMemoryContext);
3923 : }
3924 : }
3925 : }
3926 332 : return result;
3927 :
3928 8 : case XPATH_BOOLEAN:
3929 8 : if (astate == NULL)
3930 0 : return 1;
3931 8 : datum = BoolGetDatum(xpathobj->boolval);
3932 8 : datumtype = BOOLOID;
3933 8 : break;
3934 :
3935 12 : case XPATH_NUMBER:
3936 12 : if (astate == NULL)
3937 8 : return 1;
3938 4 : datum = Float8GetDatum(xpathobj->floatval);
3939 4 : datumtype = FLOAT8OID;
3940 4 : break;
3941 :
3942 8 : case XPATH_STRING:
3943 8 : if (astate == NULL)
3944 0 : return 1;
3945 8 : datum = CStringGetDatum((char *) xpathobj->stringval);
3946 8 : datumtype = CSTRINGOID;
3947 8 : break;
3948 :
3949 0 : default:
3950 0 : elog(ERROR, "xpath expression result type %d is unsupported",
3951 : xpathobj->type);
3952 : return 0; /* keep compiler quiet */
3953 : }
3954 :
3955 : /* Common code for scalar-value cases */
3956 20 : result_str = map_sql_value_to_xml_value(datum, datumtype, true);
3957 20 : datum = PointerGetDatum(cstring_to_xmltype(result_str));
3958 20 : (void) accumArrayResult(astate, datum, false,
3959 : XMLOID, CurrentMemoryContext);
3960 20 : return 1;
3961 : }
3962 :
3963 :
3964 : /*
3965 : * Common code for xpath() and xmlexists()
3966 : *
3967 : * Evaluate XPath expression and return number of nodes in res_nitems
3968 : * and array of XML values in astate. Either of those pointers can be
3969 : * NULL if the corresponding result isn't wanted.
3970 : *
3971 : * It is up to the user to ensure that the XML passed is in fact
3972 : * an XML document - XPath doesn't work easily on fragments without
3973 : * a context node being known.
3974 : */
3975 : static void
3976 372 : xpath_internal(text *xpath_expr_text, xmltype *data, ArrayType *namespaces,
3977 : int *res_nitems, ArrayBuildState *astate)
3978 : {
3979 : PgXmlErrorContext *xmlerrcxt;
3980 372 : volatile xmlParserCtxtPtr ctxt = NULL;
3981 372 : volatile xmlDocPtr doc = NULL;
3982 372 : volatile xmlXPathContextPtr xpathctx = NULL;
3983 372 : volatile xmlXPathCompExprPtr xpathcomp = NULL;
3984 372 : volatile xmlXPathObjectPtr xpathobj = NULL;
3985 : char *datastr;
3986 : int32 len;
3987 : int32 xpath_len;
3988 : xmlChar *string;
3989 : xmlChar *xpath_expr;
3990 372 : size_t xmldecl_len = 0;
3991 : int i;
3992 : int ndim;
3993 : Datum *ns_names_uris;
3994 : bool *ns_names_uris_nulls;
3995 : int ns_count;
3996 :
3997 : /*
3998 : * Namespace mappings are passed as text[]. If an empty array is passed
3999 : * (ndim = 0, "0-dimensional"), then there are no namespace mappings.
4000 : * Else, a 2-dimensional array with length of the second axis being equal
4001 : * to 2 should be passed, i.e., every subarray contains 2 elements, the
4002 : * first element defining the name, the second one the URI. Example:
4003 : * ARRAY[ARRAY['myns', 'http://example.com'], ARRAY['myns2',
4004 : * 'http://example2.com']].
4005 : */
4006 372 : ndim = namespaces ? ARR_NDIM(namespaces) : 0;
4007 372 : if (ndim != 0)
4008 : {
4009 : int *dims;
4010 :
4011 84 : dims = ARR_DIMS(namespaces);
4012 :
4013 84 : if (ndim != 2 || dims[1] != 2)
4014 0 : ereport(ERROR,
4015 : (errcode(ERRCODE_DATA_EXCEPTION),
4016 : errmsg("invalid array for XML namespace mapping"),
4017 : errdetail("The array must be two-dimensional with length of the second axis equal to 2.")));
4018 :
4019 : Assert(ARR_ELEMTYPE(namespaces) == TEXTOID);
4020 :
4021 84 : deconstruct_array(namespaces, TEXTOID, -1, false, TYPALIGN_INT,
4022 : &ns_names_uris, &ns_names_uris_nulls,
4023 : &ns_count);
4024 :
4025 : Assert((ns_count % 2) == 0); /* checked above */
4026 84 : ns_count /= 2; /* count pairs only */
4027 : }
4028 : else
4029 : {
4030 288 : ns_names_uris = NULL;
4031 288 : ns_names_uris_nulls = NULL;
4032 288 : ns_count = 0;
4033 : }
4034 :
4035 372 : datastr = VARDATA(data);
4036 372 : len = VARSIZE(data) - VARHDRSZ;
4037 372 : xpath_len = VARSIZE_ANY_EXHDR(xpath_expr_text);
4038 372 : if (xpath_len == 0)
4039 4 : ereport(ERROR,
4040 : (errcode(ERRCODE_DATA_EXCEPTION),
4041 : errmsg("empty XPath expression")));
4042 :
4043 368 : string = pg_xmlCharStrndup(datastr, len);
4044 368 : xpath_expr = pg_xmlCharStrndup(VARDATA_ANY(xpath_expr_text), xpath_len);
4045 :
4046 : /*
4047 : * In a UTF8 database, skip any xml declaration, which might assert
4048 : * another encoding. Ignore parse_xml_decl() failure, letting
4049 : * xmlCtxtReadMemory() report parse errors. Documentation disclaims
4050 : * xpath() support for non-ASCII data in non-UTF8 databases, so leave
4051 : * those scenarios bug-compatible with historical behavior.
4052 : */
4053 368 : if (GetDatabaseEncoding() == PG_UTF8)
4054 368 : parse_xml_decl(string, &xmldecl_len, NULL, NULL, NULL);
4055 :
4056 368 : xmlerrcxt = pg_xml_init(PG_XML_STRICTNESS_ALL);
4057 :
4058 368 : PG_TRY();
4059 : {
4060 368 : xmlInitParser();
4061 :
4062 : /*
4063 : * redundant XML parsing (two parsings for the same value during one
4064 : * command execution are possible)
4065 : */
4066 368 : ctxt = xmlNewParserCtxt();
4067 368 : if (ctxt == NULL || xmlerrcxt->err_occurred)
4068 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
4069 : "could not allocate parser context");
4070 736 : doc = xmlCtxtReadMemory(ctxt, (char *) string + xmldecl_len,
4071 368 : len - xmldecl_len, NULL, NULL, 0);
4072 368 : if (doc == NULL || xmlerrcxt->err_occurred)
4073 8 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_INVALID_XML_DOCUMENT,
4074 : "could not parse XML document");
4075 360 : xpathctx = xmlXPathNewContext(doc);
4076 360 : if (xpathctx == NULL || xmlerrcxt->err_occurred)
4077 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
4078 : "could not allocate XPath context");
4079 360 : xpathctx->node = (xmlNodePtr) doc;
4080 :
4081 : /* register namespaces, if any */
4082 360 : if (ns_count > 0)
4083 : {
4084 168 : for (i = 0; i < ns_count; i++)
4085 : {
4086 : char *ns_name;
4087 : char *ns_uri;
4088 :
4089 84 : if (ns_names_uris_nulls[i * 2] ||
4090 84 : ns_names_uris_nulls[i * 2 + 1])
4091 0 : ereport(ERROR,
4092 : (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
4093 : errmsg("neither namespace name nor URI may be null")));
4094 84 : ns_name = TextDatumGetCString(ns_names_uris[i * 2]);
4095 84 : ns_uri = TextDatumGetCString(ns_names_uris[i * 2 + 1]);
4096 84 : if (xmlXPathRegisterNs(xpathctx,
4097 : (xmlChar *) ns_name,
4098 : (xmlChar *) ns_uri) != 0)
4099 0 : ereport(ERROR, /* is this an internal error??? */
4100 : (errmsg("could not register XML namespace with name \"%s\" and URI \"%s\"",
4101 : ns_name, ns_uri)));
4102 : }
4103 : }
4104 :
4105 360 : xpathcomp = xmlXPathCompile(xpath_expr);
4106 360 : if (xpathcomp == NULL || xmlerrcxt->err_occurred)
4107 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_INTERNAL_ERROR,
4108 : "invalid XPath expression");
4109 :
4110 : /*
4111 : * Version 2.6.27 introduces a function named
4112 : * xmlXPathCompiledEvalToBoolean, which would be enough for xmlexists,
4113 : * but we can derive the existence by whether any nodes are returned,
4114 : * thereby preventing a library version upgrade and keeping the code
4115 : * the same.
4116 : */
4117 360 : xpathobj = xmlXPathCompiledEval(xpathcomp, xpathctx);
4118 360 : if (xpathobj == NULL || xmlerrcxt->err_occurred)
4119 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_INTERNAL_ERROR,
4120 : "could not create XPath object");
4121 :
4122 : /*
4123 : * Extract the results as requested.
4124 : */
4125 360 : if (res_nitems != NULL)
4126 288 : *res_nitems = xml_xpathobjtoxmlarray(xpathobj, astate, xmlerrcxt);
4127 : else
4128 72 : (void) xml_xpathobjtoxmlarray(xpathobj, astate, xmlerrcxt);
4129 : }
4130 8 : PG_CATCH();
4131 : {
4132 8 : if (xpathobj)
4133 0 : xmlXPathFreeObject(xpathobj);
4134 8 : if (xpathcomp)
4135 0 : xmlXPathFreeCompExpr(xpathcomp);
4136 8 : if (xpathctx)
4137 0 : xmlXPathFreeContext(xpathctx);
4138 8 : if (doc)
4139 8 : xmlFreeDoc(doc);
4140 8 : if (ctxt)
4141 8 : xmlFreeParserCtxt(ctxt);
4142 :
4143 8 : pg_xml_done(xmlerrcxt, true);
4144 :
4145 8 : PG_RE_THROW();
4146 : }
4147 360 : PG_END_TRY();
4148 :
4149 360 : xmlXPathFreeObject(xpathobj);
4150 360 : xmlXPathFreeCompExpr(xpathcomp);
4151 360 : xmlXPathFreeContext(xpathctx);
4152 360 : xmlFreeDoc(doc);
4153 360 : xmlFreeParserCtxt(ctxt);
4154 :
4155 360 : pg_xml_done(xmlerrcxt, false);
4156 360 : }
4157 : #endif /* USE_LIBXML */
4158 :
4159 : /*
4160 : * Evaluate XPath expression and return array of XML values.
4161 : *
4162 : * As we have no support of XQuery sequences yet, this function seems
4163 : * to be the most useful one (array of XML functions plays a role of
4164 : * some kind of substitution for XQuery sequences).
4165 : */
4166 : Datum
4167 84 : xpath(PG_FUNCTION_ARGS)
4168 : {
4169 : #ifdef USE_LIBXML
4170 84 : text *xpath_expr_text = PG_GETARG_TEXT_PP(0);
4171 84 : xmltype *data = PG_GETARG_XML_P(1);
4172 84 : ArrayType *namespaces = PG_GETARG_ARRAYTYPE_P(2);
4173 : ArrayBuildState *astate;
4174 :
4175 84 : astate = initArrayResult(XMLOID, CurrentMemoryContext, true);
4176 84 : xpath_internal(xpath_expr_text, data, namespaces,
4177 : NULL, astate);
4178 72 : PG_RETURN_ARRAYTYPE_P(makeArrayResult(astate, CurrentMemoryContext));
4179 : #else
4180 : NO_XML_SUPPORT();
4181 : return 0;
4182 : #endif
4183 : }
4184 :
4185 : /*
4186 : * Determines if the node specified by the supplied XPath exists
4187 : * in a given XML document, returning a boolean.
4188 : */
4189 : Datum
4190 132 : xmlexists(PG_FUNCTION_ARGS)
4191 : {
4192 : #ifdef USE_LIBXML
4193 132 : text *xpath_expr_text = PG_GETARG_TEXT_PP(0);
4194 132 : xmltype *data = PG_GETARG_XML_P(1);
4195 : int res_nitems;
4196 :
4197 132 : xpath_internal(xpath_expr_text, data, NULL,
4198 : &res_nitems, NULL);
4199 :
4200 132 : PG_RETURN_BOOL(res_nitems > 0);
4201 : #else
4202 : NO_XML_SUPPORT();
4203 : return 0;
4204 : #endif
4205 : }
4206 :
4207 : /*
4208 : * Determines if the node specified by the supplied XPath exists
4209 : * in a given XML document, returning a boolean. Differs from
4210 : * xmlexists as it supports namespaces and is not defined in SQL/XML.
4211 : */
4212 : Datum
4213 156 : xpath_exists(PG_FUNCTION_ARGS)
4214 : {
4215 : #ifdef USE_LIBXML
4216 156 : text *xpath_expr_text = PG_GETARG_TEXT_PP(0);
4217 156 : xmltype *data = PG_GETARG_XML_P(1);
4218 156 : ArrayType *namespaces = PG_GETARG_ARRAYTYPE_P(2);
4219 : int res_nitems;
4220 :
4221 156 : xpath_internal(xpath_expr_text, data, namespaces,
4222 : &res_nitems, NULL);
4223 :
4224 156 : PG_RETURN_BOOL(res_nitems > 0);
4225 : #else
4226 : NO_XML_SUPPORT();
4227 : return 0;
4228 : #endif
4229 : }
4230 :
4231 : /*
4232 : * Functions for checking well-formed-ness
4233 : */
4234 :
4235 : #ifdef USE_LIBXML
4236 : static bool
4237 76 : wellformed_xml(text *data, XmlOptionType xmloption_arg)
4238 : {
4239 : bool result;
4240 76 : volatile xmlDocPtr doc = NULL;
4241 :
4242 : /* We want to catch any exceptions and return false */
4243 76 : PG_TRY();
4244 : {
4245 76 : doc = xml_parse(data, xmloption_arg, true, GetDatabaseEncoding());
4246 40 : result = true;
4247 : }
4248 36 : PG_CATCH();
4249 : {
4250 36 : FlushErrorState();
4251 36 : result = false;
4252 : }
4253 76 : PG_END_TRY();
4254 :
4255 76 : if (doc)
4256 40 : xmlFreeDoc(doc);
4257 :
4258 76 : return result;
4259 : }
4260 : #endif
4261 :
4262 : Datum
4263 60 : xml_is_well_formed(PG_FUNCTION_ARGS)
4264 : {
4265 : #ifdef USE_LIBXML
4266 60 : text *data = PG_GETARG_TEXT_PP(0);
4267 :
4268 60 : PG_RETURN_BOOL(wellformed_xml(data, xmloption));
4269 : #else
4270 : NO_XML_SUPPORT();
4271 : return 0;
4272 : #endif /* not USE_LIBXML */
4273 : }
4274 :
4275 : Datum
4276 8 : xml_is_well_formed_document(PG_FUNCTION_ARGS)
4277 : {
4278 : #ifdef USE_LIBXML
4279 8 : text *data = PG_GETARG_TEXT_PP(0);
4280 :
4281 8 : PG_RETURN_BOOL(wellformed_xml(data, XMLOPTION_DOCUMENT));
4282 : #else
4283 : NO_XML_SUPPORT();
4284 : return 0;
4285 : #endif /* not USE_LIBXML */
4286 : }
4287 :
4288 : Datum
4289 8 : xml_is_well_formed_content(PG_FUNCTION_ARGS)
4290 : {
4291 : #ifdef USE_LIBXML
4292 8 : text *data = PG_GETARG_TEXT_PP(0);
4293 :
4294 8 : PG_RETURN_BOOL(wellformed_xml(data, XMLOPTION_CONTENT));
4295 : #else
4296 : NO_XML_SUPPORT();
4297 : return 0;
4298 : #endif /* not USE_LIBXML */
4299 : }
4300 :
4301 : /*
4302 : * support functions for XMLTABLE
4303 : *
4304 : */
4305 : #ifdef USE_LIBXML
4306 :
4307 : /*
4308 : * Returns private data from executor state. Ensure validity by check with
4309 : * MAGIC number.
4310 : */
4311 : static inline XmlTableBuilderData *
4312 99142 : GetXmlTableBuilderPrivateData(TableFuncScanState *state, const char *fname)
4313 : {
4314 : XmlTableBuilderData *result;
4315 :
4316 99142 : if (!IsA(state, TableFuncScanState))
4317 0 : elog(ERROR, "%s called with invalid TableFuncScanState", fname);
4318 99142 : result = (XmlTableBuilderData *) state->opaque;
4319 99142 : if (result->magic != XMLTABLE_CONTEXT_MAGIC)
4320 0 : elog(ERROR, "%s called with invalid TableFuncScanState", fname);
4321 :
4322 99142 : return result;
4323 : }
4324 : #endif
4325 :
4326 : /*
4327 : * XmlTableInitOpaque
4328 : * Fill in TableFuncScanState->opaque for XmlTable processor; initialize
4329 : * the XML parser.
4330 : *
4331 : * Note: Because we call pg_xml_init() here and pg_xml_done() in
4332 : * XmlTableDestroyOpaque, it is critical for robustness that no other
4333 : * executor nodes run until this node is processed to completion. Caller
4334 : * must execute this to completion (probably filling a tuplestore to exhaust
4335 : * this node in a single pass) instead of using row-per-call mode.
4336 : */
4337 : static void
4338 176 : XmlTableInitOpaque(TableFuncScanState *state, int natts)
4339 : {
4340 : #ifdef USE_LIBXML
4341 176 : volatile xmlParserCtxtPtr ctxt = NULL;
4342 : XmlTableBuilderData *xtCxt;
4343 : PgXmlErrorContext *xmlerrcxt;
4344 :
4345 176 : xtCxt = palloc0(sizeof(XmlTableBuilderData));
4346 176 : xtCxt->magic = XMLTABLE_CONTEXT_MAGIC;
4347 176 : xtCxt->natts = natts;
4348 176 : xtCxt->xpathscomp = palloc0(sizeof(xmlXPathCompExprPtr) * natts);
4349 :
4350 176 : xmlerrcxt = pg_xml_init(PG_XML_STRICTNESS_ALL);
4351 :
4352 176 : PG_TRY();
4353 : {
4354 176 : xmlInitParser();
4355 :
4356 176 : ctxt = xmlNewParserCtxt();
4357 176 : if (ctxt == NULL || xmlerrcxt->err_occurred)
4358 0 : xml_ereport(xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
4359 : "could not allocate parser context");
4360 : }
4361 0 : PG_CATCH();
4362 : {
4363 0 : if (ctxt != NULL)
4364 0 : xmlFreeParserCtxt(ctxt);
4365 :
4366 0 : pg_xml_done(xmlerrcxt, true);
4367 :
4368 0 : PG_RE_THROW();
4369 : }
4370 176 : PG_END_TRY();
4371 :
4372 176 : xtCxt->xmlerrcxt = xmlerrcxt;
4373 176 : xtCxt->ctxt = ctxt;
4374 :
4375 176 : state->opaque = xtCxt;
4376 : #else
4377 : NO_XML_SUPPORT();
4378 : #endif /* not USE_LIBXML */
4379 176 : }
4380 :
4381 : /*
4382 : * XmlTableSetDocument
4383 : * Install the input document
4384 : */
4385 : static void
4386 176 : XmlTableSetDocument(TableFuncScanState *state, Datum value)
4387 : {
4388 : #ifdef USE_LIBXML
4389 : XmlTableBuilderData *xtCxt;
4390 176 : xmltype *xmlval = DatumGetXmlP(value);
4391 : char *str;
4392 : xmlChar *xstr;
4393 : int length;
4394 176 : volatile xmlDocPtr doc = NULL;
4395 176 : volatile xmlXPathContextPtr xpathcxt = NULL;
4396 :
4397 176 : xtCxt = GetXmlTableBuilderPrivateData(state, "XmlTableSetDocument");
4398 :
4399 : /*
4400 : * Use out function for casting to string (remove encoding property). See
4401 : * comment in xml_out.
4402 : */
4403 176 : str = xml_out_internal(xmlval, 0);
4404 :
4405 176 : length = strlen(str);
4406 176 : xstr = pg_xmlCharStrndup(str, length);
4407 :
4408 176 : PG_TRY();
4409 : {
4410 176 : doc = xmlCtxtReadMemory(xtCxt->ctxt, (char *) xstr, length, NULL, NULL, 0);
4411 176 : if (doc == NULL || xtCxt->xmlerrcxt->err_occurred)
4412 0 : xml_ereport(xtCxt->xmlerrcxt, ERROR, ERRCODE_INVALID_XML_DOCUMENT,
4413 : "could not parse XML document");
4414 176 : xpathcxt = xmlXPathNewContext(doc);
4415 176 : if (xpathcxt == NULL || xtCxt->xmlerrcxt->err_occurred)
4416 0 : xml_ereport(xtCxt->xmlerrcxt, ERROR, ERRCODE_OUT_OF_MEMORY,
4417 : "could not allocate XPath context");
4418 176 : xpathcxt->node = (xmlNodePtr) doc;
4419 : }
4420 0 : PG_CATCH();
4421 : {
4422 0 : if (xpathcxt != NULL)
4423 0 : xmlXPathFreeContext(xpathcxt);
4424 0 : if (doc != NULL)
4425 0 : xmlFreeDoc(doc);
4426 :
4427 0 : PG_RE_THROW();
4428 : }
4429 176 : PG_END_TRY();
4430 :
4431 176 : xtCxt->doc = doc;
4432 176 : xtCxt->xpathcxt = xpathcxt;
4433 : #else
4434 : NO_XML_SUPPORT();
4435 : #endif /* not USE_LIBXML */
4436 176 : }
4437 :
4438 : /*
4439 : * XmlTableSetNamespace
4440 : * Add a namespace declaration
4441 : */
4442 : static void
4443 12 : XmlTableSetNamespace(TableFuncScanState *state, const char *name, const char *uri)
4444 : {
4445 : #ifdef USE_LIBXML
4446 : XmlTableBuilderData *xtCxt;
4447 :
4448 12 : if (name == NULL)
4449 4 : ereport(ERROR,
4450 : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
4451 : errmsg("DEFAULT namespace is not supported")));
4452 8 : xtCxt = GetXmlTableBuilderPrivateData(state, "XmlTableSetNamespace");
4453 :
4454 8 : if (xmlXPathRegisterNs(xtCxt->xpathcxt,
4455 8 : pg_xmlCharStrndup(name, strlen(name)),
4456 8 : pg_xmlCharStrndup(uri, strlen(uri))))
4457 0 : xml_ereport(xtCxt->xmlerrcxt, ERROR, ERRCODE_DATA_EXCEPTION,
4458 : "could not set XML namespace");
4459 : #else
4460 : NO_XML_SUPPORT();
4461 : #endif /* not USE_LIBXML */
4462 8 : }
4463 :
4464 : /*
4465 : * XmlTableSetRowFilter
4466 : * Install the row-filter Xpath expression.
4467 : */
4468 : static void
4469 172 : XmlTableSetRowFilter(TableFuncScanState *state, const char *path)
4470 : {
4471 : #ifdef USE_LIBXML
4472 : XmlTableBuilderData *xtCxt;
4473 : xmlChar *xstr;
4474 :
4475 172 : xtCxt = GetXmlTableBuilderPrivateData(state, "XmlTableSetRowFilter");
4476 :
4477 172 : if (*path == '\0')
4478 0 : ereport(ERROR,
4479 : (errcode(ERRCODE_DATA_EXCEPTION),
4480 : errmsg("row path filter must not be empty string")));
4481 :
4482 172 : xstr = pg_xmlCharStrndup(path, strlen(path));
4483 :
4484 172 : xtCxt->xpathcomp = xmlXPathCompile(xstr);
4485 172 : if (xtCxt->xpathcomp == NULL || xtCxt->xmlerrcxt->err_occurred)
4486 0 : xml_ereport(xtCxt->xmlerrcxt, ERROR, ERRCODE_SYNTAX_ERROR,
4487 : "invalid XPath expression");
4488 : #else
4489 : NO_XML_SUPPORT();
4490 : #endif /* not USE_LIBXML */
4491 172 : }
4492 :
4493 : /*
4494 : * XmlTableSetColumnFilter
4495 : * Install the column-filter Xpath expression, for the given column.
4496 : */
4497 : static void
4498 516 : XmlTableSetColumnFilter(TableFuncScanState *state, const char *path, int colnum)
4499 : {
4500 : #ifdef USE_LIBXML
4501 : XmlTableBuilderData *xtCxt;
4502 : xmlChar *xstr;
4503 :
4504 : AssertArg(PointerIsValid(path));
4505 :
4506 516 : xtCxt = GetXmlTableBuilderPrivateData(state, "XmlTableSetColumnFilter");
4507 :
4508 516 : if (*path == '\0')
4509 0 : ereport(ERROR,
4510 : (errcode(ERRCODE_DATA_EXCEPTION),
4511 : errmsg("column path filter must not be empty string")));
4512 :
4513 516 : xstr = pg_xmlCharStrndup(path, strlen(path));
4514 :
4515 516 : xtCxt->xpathscomp[colnum] = xmlXPathCompile(xstr);
4516 516 : if (xtCxt->xpathscomp[colnum] == NULL || xtCxt->xmlerrcxt->err_occurred)
4517 0 : xml_ereport(xtCxt->xmlerrcxt, ERROR, ERRCODE_DATA_EXCEPTION,
4518 : "invalid XPath expression");
4519 : #else
4520 : NO_XML_SUPPORT();
4521 : #endif /* not USE_LIBXML */
4522 516 : }
4523 :
4524 : /*
4525 : * XmlTableFetchRow
4526 : * Prepare the next "current" tuple for upcoming GetValue calls.
4527 : * Returns false if the row-filter expression returned no more rows.
4528 : */
4529 : static bool
4530 14254 : XmlTableFetchRow(TableFuncScanState *state)
4531 : {
4532 : #ifdef USE_LIBXML
4533 : XmlTableBuilderData *xtCxt;
4534 :
4535 14254 : xtCxt = GetXmlTableBuilderPrivateData(state, "XmlTableFetchRow");
4536 :
4537 : /* Propagate our own error context to libxml2 */
4538 14254 : xmlSetStructuredErrorFunc((void *) xtCxt->xmlerrcxt, xml_errorHandler);
4539 :
4540 14254 : if (xtCxt->xpathobj == NULL)
4541 : {
4542 172 : xtCxt->xpathobj = xmlXPathCompiledEval(xtCxt->xpathcomp, xtCxt->xpathcxt);
4543 172 : if (xtCxt->xpathobj == NULL || xtCxt->xmlerrcxt->err_occurred)
4544 0 : xml_ereport(xtCxt->xmlerrcxt, ERROR, ERRCODE_INTERNAL_ERROR,
4545 : "could not create XPath object");
4546 :
4547 172 : xtCxt->row_count = 0;
4548 : }
4549 :
4550 14254 : if (xtCxt->xpathobj->type == XPATH_NODESET)
4551 : {
4552 14254 : if (xtCxt->xpathobj->nodesetval != NULL)
4553 : {
4554 14254 : if (xtCxt->row_count++ < xtCxt->xpathobj->nodesetval->nodeNr)
4555 14090 : return true;
4556 : }
4557 : }
4558 :
4559 164 : return false;
4560 : #else
4561 : NO_XML_SUPPORT();
4562 : return false;
4563 : #endif /* not USE_LIBXML */
4564 : }
4565 :
4566 : /*
4567 : * XmlTableGetValue
4568 : * Return the value for column number 'colnum' for the current row. If
4569 : * column -1 is requested, return representation of the whole row.
4570 : *
4571 : * This leaks memory, so be sure to reset often the context in which it's
4572 : * called.
4573 : */
4574 : static Datum
4575 83840 : XmlTableGetValue(TableFuncScanState *state, int colnum,
4576 : Oid typid, int32 typmod, bool *isnull)
4577 : {
4578 : #ifdef USE_LIBXML
4579 : XmlTableBuilderData *xtCxt;
4580 83840 : Datum result = (Datum) 0;
4581 : xmlNodePtr cur;
4582 83840 : char *cstr = NULL;
4583 83840 : volatile xmlXPathObjectPtr xpathobj = NULL;
4584 :
4585 83840 : xtCxt = GetXmlTableBuilderPrivateData(state, "XmlTableGetValue");
4586 :
4587 : Assert(xtCxt->xpathobj &&
4588 : xtCxt->xpathobj->type == XPATH_NODESET &&
4589 : xtCxt->xpathobj->nodesetval != NULL);
4590 :
4591 : /* Propagate our own error context to libxml2 */
4592 83840 : xmlSetStructuredErrorFunc((void *) xtCxt->xmlerrcxt, xml_errorHandler);
4593 :
4594 83840 : *isnull = false;
4595 :
4596 83840 : cur = xtCxt->xpathobj->nodesetval->nodeTab[xtCxt->row_count - 1];
4597 :
4598 : Assert(xtCxt->xpathscomp[colnum] != NULL);
4599 :
4600 83840 : PG_TRY();
4601 : {
4602 : /* Set current node as entry point for XPath evaluation */
4603 83840 : xtCxt->xpathcxt->node = cur;
4604 :
4605 : /* Evaluate column path */
4606 83840 : xpathobj = xmlXPathCompiledEval(xtCxt->xpathscomp[colnum], xtCxt->xpathcxt);
4607 83840 : if (xpathobj == NULL || xtCxt->xmlerrcxt->err_occurred)
4608 0 : xml_ereport(xtCxt->xmlerrcxt, ERROR, ERRCODE_INTERNAL_ERROR,
4609 : "could not create XPath object");
4610 :
4611 : /*
4612 : * There are four possible cases, depending on the number of nodes
4613 : * returned by the XPath expression and the type of the target column:
4614 : * a) XPath returns no nodes. b) The target type is XML (return all
4615 : * as XML). For non-XML return types: c) One node (return content).
4616 : * d) Multiple nodes (error).
4617 : */
4618 83840 : if (xpathobj->type == XPATH_NODESET)
4619 : {
4620 83820 : int count = 0;
4621 :
4622 83820 : if (xpathobj->nodesetval != NULL)
4623 83680 : count = xpathobj->nodesetval->nodeNr;
4624 :
4625 83820 : if (xpathobj->nodesetval == NULL || count == 0)
4626 : {
4627 14276 : *isnull = true;
4628 : }
4629 : else
4630 : {
4631 69544 : if (typid == XMLOID)
4632 : {
4633 : text *textstr;
4634 : StringInfoData str;
4635 :
4636 : /* Concatenate serialized values */
4637 48 : initStringInfo(&str);
4638 136 : for (int i = 0; i < count; i++)
4639 : {
4640 : textstr =
4641 88 : xml_xmlnodetoxmltype(xpathobj->nodesetval->nodeTab[i],
4642 : xtCxt->xmlerrcxt);
4643 :
4644 88 : appendStringInfoText(&str, textstr);
4645 : }
4646 48 : cstr = str.data;
4647 : }
4648 : else
4649 : {
4650 : xmlChar *str;
4651 :
4652 69496 : if (count > 1)
4653 4 : ereport(ERROR,
4654 : (errcode(ERRCODE_CARDINALITY_VIOLATION),
4655 : errmsg("more than one value returned by column XPath expression")));
4656 :
4657 69492 : str = xmlXPathCastNodeSetToString(xpathobj->nodesetval);
4658 69492 : cstr = str ? xml_pstrdup_and_free(str) : "";
4659 : }
4660 : }
4661 : }
4662 20 : else if (xpathobj->type == XPATH_STRING)
4663 : {
4664 : /* Content should be escaped when target will be XML */
4665 12 : if (typid == XMLOID)
4666 4 : cstr = escape_xml((char *) xpathobj->stringval);
4667 : else
4668 8 : cstr = (char *) xpathobj->stringval;
4669 : }
4670 8 : else if (xpathobj->type == XPATH_BOOLEAN)
4671 : {
4672 : char typcategory;
4673 : bool typispreferred;
4674 : xmlChar *str;
4675 :
4676 : /* Allow implicit casting from boolean to numbers */
4677 4 : get_type_category_preferred(typid, &typcategory, &typispreferred);
4678 :
4679 4 : if (typcategory != TYPCATEGORY_NUMERIC)
4680 4 : str = xmlXPathCastBooleanToString(xpathobj->boolval);
4681 : else
4682 0 : str = xmlXPathCastNumberToString(xmlXPathCastBooleanToNumber(xpathobj->boolval));
4683 :
4684 4 : cstr = xml_pstrdup_and_free(str);
4685 : }
4686 4 : else if (xpathobj->type == XPATH_NUMBER)
4687 : {
4688 : xmlChar *str;
4689 :
4690 4 : str = xmlXPathCastNumberToString(xpathobj->floatval);
4691 4 : cstr = xml_pstrdup_and_free(str);
4692 : }
4693 : else
4694 0 : elog(ERROR, "unexpected XPath object type %u", xpathobj->type);
4695 :
4696 : /*
4697 : * By here, either cstr contains the result value, or the isnull flag
4698 : * has been set.
4699 : */
4700 : Assert(cstr || *isnull);
4701 :
4702 83836 : if (!*isnull)
4703 69560 : result = InputFunctionCall(&state->in_functions[colnum],
4704 : cstr,
4705 69560 : state->typioparams[colnum],
4706 : typmod);
4707 : }
4708 4 : PG_FINALLY();
4709 : {
4710 83840 : if (xpathobj != NULL)
4711 83840 : xmlXPathFreeObject(xpathobj);
4712 : }
4713 83840 : PG_END_TRY();
4714 :
4715 83836 : return result;
4716 : #else
4717 : NO_XML_SUPPORT();
4718 : return 0;
4719 : #endif /* not USE_LIBXML */
4720 : }
4721 :
4722 : /*
4723 : * XmlTableDestroyOpaque
4724 : * Release all libxml2 resources
4725 : */
4726 : static void
4727 176 : XmlTableDestroyOpaque(TableFuncScanState *state)
4728 : {
4729 : #ifdef USE_LIBXML
4730 : XmlTableBuilderData *xtCxt;
4731 :
4732 176 : xtCxt = GetXmlTableBuilderPrivateData(state, "XmlTableDestroyOpaque");
4733 :
4734 : /* Propagate our own error context to libxml2 */
4735 176 : xmlSetStructuredErrorFunc((void *) xtCxt->xmlerrcxt, xml_errorHandler);
4736 :
4737 176 : if (xtCxt->xpathscomp != NULL)
4738 : {
4739 : int i;
4740 :
4741 744 : for (i = 0; i < xtCxt->natts; i++)
4742 568 : if (xtCxt->xpathscomp[i] != NULL)
4743 516 : xmlXPathFreeCompExpr(xtCxt->xpathscomp[i]);
4744 : }
4745 :
4746 176 : if (xtCxt->xpathobj != NULL)
4747 172 : xmlXPathFreeObject(xtCxt->xpathobj);
4748 176 : if (xtCxt->xpathcomp != NULL)
4749 172 : xmlXPathFreeCompExpr(xtCxt->xpathcomp);
4750 176 : if (xtCxt->xpathcxt != NULL)
4751 176 : xmlXPathFreeContext(xtCxt->xpathcxt);
4752 176 : if (xtCxt->doc != NULL)
4753 176 : xmlFreeDoc(xtCxt->doc);
4754 176 : if (xtCxt->ctxt != NULL)
4755 176 : xmlFreeParserCtxt(xtCxt->ctxt);
4756 :
4757 176 : pg_xml_done(xtCxt->xmlerrcxt, true);
4758 :
4759 : /* not valid anymore */
4760 176 : xtCxt->magic = 0;
4761 176 : state->opaque = NULL;
4762 :
4763 : #else
4764 : NO_XML_SUPPORT();
4765 : #endif /* not USE_LIBXML */
4766 176 : }
|
