diff --git a/contrib/xml2/pgxml.sql.in b/contrib/xml2/pgxml.sql.in index 9951d6c046131fdcbce88d8cc3babf351de70c20..f9d8bd0e31a716f3c2c8e9705e04293c3f541695 100644 --- a/contrib/xml2/pgxml.sql.in +++ b/contrib/xml2/pgxml.sql.in @@ -3,6 +3,9 @@ CREATE OR REPLACE FUNCTION xml_valid(text) RETURNS bool AS 'MODULE_PATHNAME' LANGUAGE 'c' WITH (isStrict); +CREATE OR REPLACE FUNCTION xml_encode_special_chars(text) RETURNS text + AS 'MODULE_PATHNAME' LANGUAGE 'c' WITH (isStrict); + CREATE OR REPLACE FUNCTION xpath_string(text,text) RETURNS text AS 'MODULE_PATHNAME' LANGUAGE 'c' WITH (isStrict); diff --git a/contrib/xml2/xpath.c b/contrib/xml2/xpath.c index ca95d41c22632f328acddfac50f1308124b6776c..d3ac76b9ef5b889301e9e6e934b73d094fac31bf 100644 --- a/contrib/xml2/xpath.c +++ b/contrib/xml2/xpath.c @@ -40,6 +40,7 @@ static xmlXPathObjectPtr pgxml_xpath(text *document, xmlChar * xpath); Datum xml_valid(PG_FUNCTION_ARGS); +Datum xml_encode_special_chars(PG_FUNCTION_ARGS); Datum xpath_nodeset(PG_FUNCTION_ARGS); Datum xpath_string(PG_FUNCTION_ARGS); Datum xpath_number(PG_FUNCTION_ARGS); @@ -186,6 +187,34 @@ xml_valid(PG_FUNCTION_ARGS) } +/* Encodes special characters (<, >, &, " and \r) as XML entities */ + +PG_FUNCTION_INFO_V1(xml_encode_special_chars); + +Datum +xml_encode_special_chars(PG_FUNCTION_ARGS) +{ + text *tin = PG_GETARG_TEXT_P(0); + text *tout; + int32 ressize; + xmlChar *ts, *tt; + + ts = pgxml_texttoxmlchar(tin); + + tt = xmlEncodeSpecialChars(NULL, ts); + + pfree(ts); + + ressize = strlen(tt); + tout = (text *) palloc(ressize + VARHDRSZ); + memcpy(VARDATA(tout), tt, ressize); + VARATT_SIZEP(tout) = ressize + VARHDRSZ; + + xmlFree(tt); + + PG_RETURN_TEXT_P(tout); +} + static xmlChar * pgxmlNodeSetToText(xmlNodeSetPtr nodeset,