2008-05-17 09:28:26 +08:00
|
|
|
/*
|
2010-09-21 04:08:53 +08:00
|
|
|
* contrib/xml2/xslt_proc.c
|
2008-05-17 09:28:26 +08:00
|
|
|
*
|
2009-01-07 21:44:37 +08:00
|
|
|
* XSLT processing functions (requiring libxslt)
|
|
|
|
*
|
|
|
|
* John Gray, for Torchbox 2003-04-01
|
|
|
|
*/
|
2004-03-05 11:24:50 +08:00
|
|
|
#include "postgres.h"
|
2009-01-07 21:44:37 +08:00
|
|
|
|
2004-03-05 11:24:50 +08:00
|
|
|
#include "executor/spi.h"
|
2009-01-07 21:44:37 +08:00
|
|
|
#include "fmgr.h"
|
2004-03-05 11:24:50 +08:00
|
|
|
#include "funcapi.h"
|
|
|
|
#include "miscadmin.h"
|
2009-01-07 21:44:37 +08:00
|
|
|
#include "utils/builtins.h"
|
2010-03-04 03:10:22 +08:00
|
|
|
#include "utils/xml.h"
|
2004-03-05 11:24:50 +08:00
|
|
|
|
2010-03-02 02:07:59 +08:00
|
|
|
#ifdef USE_LIBXSLT
|
|
|
|
|
2004-03-05 11:24:50 +08:00
|
|
|
/* libxml includes */
|
|
|
|
|
|
|
|
#include <libxml/xpath.h>
|
|
|
|
#include <libxml/tree.h>
|
|
|
|
#include <libxml/xmlmemory.h>
|
|
|
|
|
|
|
|
/* libxslt includes */
|
|
|
|
|
|
|
|
#include <libxslt/xslt.h>
|
|
|
|
#include <libxslt/xsltInternals.h>
|
2012-08-15 06:28:29 +08:00
|
|
|
#include <libxslt/security.h>
|
2004-03-05 11:24:50 +08:00
|
|
|
#include <libxslt/transform.h>
|
|
|
|
#include <libxslt/xsltutils.h>
|
2010-07-07 03:19:02 +08:00
|
|
|
#endif /* USE_LIBXSLT */
|
2010-03-02 02:07:59 +08:00
|
|
|
|
2004-03-05 11:24:50 +08:00
|
|
|
|
2010-03-01 03:51:37 +08:00
|
|
|
/* externally accessible functions */
|
|
|
|
|
|
|
|
Datum xslt_process(PG_FUNCTION_ARGS);
|
|
|
|
|
2010-03-02 02:07:59 +08:00
|
|
|
#ifdef USE_LIBXSLT
|
|
|
|
|
2004-03-05 11:24:50 +08:00
|
|
|
/* declarations to come from xpath.c */
|
2011-07-21 01:03:12 +08:00
|
|
|
extern PgXmlErrorContext *pgxml_parser_init(PgXmlStrictness strictness);
|
2004-03-05 11:24:50 +08:00
|
|
|
|
|
|
|
/* local defs */
|
2010-08-11 07:02:00 +08:00
|
|
|
static const char **parse_params(text *paramstr);
|
2010-07-07 03:19:02 +08:00
|
|
|
#endif /* USE_LIBXSLT */
|
2010-03-02 02:07:59 +08:00
|
|
|
|
2004-03-05 11:24:50 +08:00
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(xslt_process);
|
|
|
|
|
2004-08-29 13:07:03 +08:00
|
|
|
Datum
|
|
|
|
xslt_process(PG_FUNCTION_ARGS)
|
|
|
|
{
|
2010-03-02 02:07:59 +08:00
|
|
|
#ifdef USE_LIBXSLT
|
|
|
|
|
2008-05-05 00:42:41 +08:00
|
|
|
text *doct = PG_GETARG_TEXT_P(0);
|
|
|
|
text *ssheet = PG_GETARG_TEXT_P(1);
|
2012-06-05 08:12:50 +08:00
|
|
|
text *result;
|
2008-05-05 00:42:41 +08:00
|
|
|
text *paramstr;
|
2010-08-11 07:02:00 +08:00
|
|
|
const char **params;
|
2011-07-21 01:03:12 +08:00
|
|
|
PgXmlErrorContext *xmlerrcxt;
|
|
|
|
volatile xsltStylesheetPtr stylesheet = NULL;
|
|
|
|
volatile xmlDocPtr doctree = NULL;
|
|
|
|
volatile xmlDocPtr restree = NULL;
|
2012-08-15 06:28:29 +08:00
|
|
|
volatile xsltSecurityPrefsPtr xslt_sec_prefs = NULL;
|
|
|
|
volatile xsltTransformContextPtr xslt_ctxt = NULL;
|
2011-07-21 01:03:12 +08:00
|
|
|
volatile int resstat = -1;
|
|
|
|
xmlChar *resstr = NULL;
|
|
|
|
int reslen = 0;
|
2004-08-29 13:07:03 +08:00
|
|
|
|
|
|
|
if (fcinfo->nargs == 3)
|
|
|
|
{
|
|
|
|
paramstr = PG_GETARG_TEXT_P(2);
|
2010-08-11 07:02:00 +08:00
|
|
|
params = parse_params(paramstr);
|
2004-08-29 13:07:03 +08:00
|
|
|
}
|
|
|
|
else
|
2010-08-11 07:02:00 +08:00
|
|
|
{
|
2005-10-15 10:49:52 +08:00
|
|
|
/* No parameters */
|
2010-08-11 07:02:00 +08:00
|
|
|
params = (const char **) palloc(sizeof(char *));
|
2004-08-29 13:07:03 +08:00
|
|
|
params[0] = NULL;
|
2010-08-11 07:02:00 +08:00
|
|
|
}
|
2004-08-29 13:07:03 +08:00
|
|
|
|
|
|
|
/* Setup parser */
|
2011-07-21 01:03:12 +08:00
|
|
|
xmlerrcxt = pgxml_parser_init(PG_XML_STRICTNESS_LEGACY);
|
2004-08-29 13:07:03 +08:00
|
|
|
|
2011-07-21 01:03:12 +08:00
|
|
|
PG_TRY();
|
|
|
|
{
|
2012-08-15 06:28:29 +08:00
|
|
|
xmlDocPtr ssdoc;
|
|
|
|
bool xslt_sec_prefs_error;
|
2004-08-29 13:07:03 +08:00
|
|
|
|
2012-08-15 06:28:29 +08:00
|
|
|
/* Parse document */
|
|
|
|
doctree = xmlParseMemory((char *) VARDATA(doct),
|
|
|
|
VARSIZE(doct) - VARHDRSZ);
|
2004-08-29 13:07:03 +08:00
|
|
|
|
2012-06-11 03:20:04 +08:00
|
|
|
if (doctree == NULL)
|
2011-07-21 01:03:12 +08:00
|
|
|
xml_ereport(xmlerrcxt, ERROR, ERRCODE_EXTERNAL_ROUTINE_EXCEPTION,
|
2012-06-11 03:20:04 +08:00
|
|
|
"error parsing XML document");
|
|
|
|
|
|
|
|
/* Same for stylesheet */
|
2012-08-15 06:28:29 +08:00
|
|
|
ssdoc = xmlParseMemory((char *) VARDATA(ssheet),
|
|
|
|
VARSIZE(ssheet) - VARHDRSZ);
|
|
|
|
|
|
|
|
if (ssdoc == NULL)
|
|
|
|
xml_ereport(xmlerrcxt, ERROR, ERRCODE_EXTERNAL_ROUTINE_EXCEPTION,
|
|
|
|
"error parsing stylesheet as XML document");
|
|
|
|
|
|
|
|
/* After this call we need not free ssdoc separately */
|
|
|
|
stylesheet = xsltParseStylesheetDoc(ssdoc);
|
2012-06-11 03:20:04 +08:00
|
|
|
|
|
|
|
if (stylesheet == NULL)
|
|
|
|
xml_ereport(xmlerrcxt, ERROR, ERRCODE_EXTERNAL_ROUTINE_EXCEPTION,
|
|
|
|
"failed to parse stylesheet");
|
2004-08-29 13:07:03 +08:00
|
|
|
|
2012-08-15 06:28:29 +08:00
|
|
|
xslt_ctxt = xsltNewTransformContext(stylesheet, doctree);
|
|
|
|
|
|
|
|
xslt_sec_prefs_error = false;
|
|
|
|
if ((xslt_sec_prefs = xsltNewSecurityPrefs()) == NULL)
|
|
|
|
xslt_sec_prefs_error = true;
|
|
|
|
|
|
|
|
if (xsltSetSecurityPrefs(xslt_sec_prefs, XSLT_SECPREF_READ_FILE,
|
|
|
|
xsltSecurityForbid) != 0)
|
|
|
|
xslt_sec_prefs_error = true;
|
|
|
|
if (xsltSetSecurityPrefs(xslt_sec_prefs, XSLT_SECPREF_WRITE_FILE,
|
|
|
|
xsltSecurityForbid) != 0)
|
|
|
|
xslt_sec_prefs_error = true;
|
|
|
|
if (xsltSetSecurityPrefs(xslt_sec_prefs, XSLT_SECPREF_CREATE_DIRECTORY,
|
|
|
|
xsltSecurityForbid) != 0)
|
|
|
|
xslt_sec_prefs_error = true;
|
|
|
|
if (xsltSetSecurityPrefs(xslt_sec_prefs, XSLT_SECPREF_READ_NETWORK,
|
|
|
|
xsltSecurityForbid) != 0)
|
|
|
|
xslt_sec_prefs_error = true;
|
|
|
|
if (xsltSetSecurityPrefs(xslt_sec_prefs, XSLT_SECPREF_WRITE_NETWORK,
|
|
|
|
xsltSecurityForbid) != 0)
|
|
|
|
xslt_sec_prefs_error = true;
|
|
|
|
if (xsltSetCtxtSecurityPrefs(xslt_sec_prefs, xslt_ctxt) != 0)
|
|
|
|
xslt_sec_prefs_error = true;
|
|
|
|
|
|
|
|
if (xslt_sec_prefs_error)
|
|
|
|
ereport(ERROR,
|
|
|
|
(errmsg("could not set libxslt security preferences")));
|
|
|
|
|
|
|
|
restree = xsltApplyStylesheetUser(stylesheet, doctree, params,
|
|
|
|
NULL, NULL, xslt_ctxt);
|
2012-06-05 08:12:50 +08:00
|
|
|
|
2012-06-11 03:20:04 +08:00
|
|
|
if (restree == NULL)
|
|
|
|
xml_ereport(xmlerrcxt, ERROR, ERRCODE_EXTERNAL_ROUTINE_EXCEPTION,
|
|
|
|
"failed to apply stylesheet");
|
2012-06-05 08:12:50 +08:00
|
|
|
|
2012-06-11 03:20:04 +08:00
|
|
|
resstat = xsltSaveResultToString(&resstr, &reslen, restree, stylesheet);
|
2011-07-21 01:03:12 +08:00
|
|
|
}
|
|
|
|
PG_CATCH();
|
|
|
|
{
|
|
|
|
if (stylesheet != NULL)
|
|
|
|
xsltFreeStylesheet(stylesheet);
|
|
|
|
if (restree != NULL)
|
|
|
|
xmlFreeDoc(restree);
|
|
|
|
if (doctree != NULL)
|
|
|
|
xmlFreeDoc(doctree);
|
2012-08-15 06:28:29 +08:00
|
|
|
if (xslt_sec_prefs != NULL)
|
|
|
|
xsltFreeSecurityPrefs(xslt_sec_prefs);
|
|
|
|
if (xslt_ctxt != NULL)
|
|
|
|
xsltFreeTransformContext(xslt_ctxt);
|
2011-07-21 01:03:12 +08:00
|
|
|
xsltCleanupGlobals();
|
|
|
|
|
|
|
|
pg_xml_done(xmlerrcxt, true);
|
|
|
|
|
|
|
|
PG_RE_THROW();
|
|
|
|
}
|
|
|
|
PG_END_TRY();
|
2004-08-29 13:07:03 +08:00
|
|
|
|
|
|
|
xsltFreeStylesheet(stylesheet);
|
|
|
|
xmlFreeDoc(restree);
|
|
|
|
xmlFreeDoc(doctree);
|
2012-08-15 06:28:29 +08:00
|
|
|
xsltFreeSecurityPrefs(xslt_sec_prefs);
|
|
|
|
xsltFreeTransformContext(xslt_ctxt);
|
2004-08-29 13:07:03 +08:00
|
|
|
xsltCleanupGlobals();
|
|
|
|
|
2011-07-21 01:03:12 +08:00
|
|
|
pg_xml_done(xmlerrcxt, false);
|
|
|
|
|
2012-06-05 08:12:50 +08:00
|
|
|
/* XXX this is pretty dubious, really ought to throw error instead */
|
2004-08-29 13:07:03 +08:00
|
|
|
if (resstat < 0)
|
|
|
|
PG_RETURN_NULL();
|
|
|
|
|
2012-06-05 08:12:50 +08:00
|
|
|
result = cstring_to_text_with_len((char *) resstr, reslen);
|
|
|
|
|
|
|
|
if (resstr)
|
|
|
|
xmlFree(resstr);
|
|
|
|
|
|
|
|
PG_RETURN_TEXT_P(result);
|
2010-07-07 03:19:02 +08:00
|
|
|
#else /* !USE_LIBXSLT */
|
2010-03-02 02:07:59 +08:00
|
|
|
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
|
|
|
|
errmsg("xslt_process() is not available without libxslt")));
|
|
|
|
PG_RETURN_NULL();
|
2010-07-07 03:19:02 +08:00
|
|
|
#endif /* USE_LIBXSLT */
|
2004-03-05 11:24:50 +08:00
|
|
|
}
|
|
|
|
|
2010-03-02 02:07:59 +08:00
|
|
|
#ifdef USE_LIBXSLT
|
2004-03-05 11:24:50 +08:00
|
|
|
|
2010-08-11 07:02:00 +08:00
|
|
|
static const char **
|
|
|
|
parse_params(text *paramstr)
|
2004-03-05 11:24:50 +08:00
|
|
|
{
|
2004-08-29 13:07:03 +08:00
|
|
|
char *pos;
|
|
|
|
char *pstr;
|
|
|
|
char *nvsep = "=";
|
|
|
|
char *itsep = ",";
|
2010-08-11 07:02:00 +08:00
|
|
|
const char **params;
|
|
|
|
int max_params;
|
|
|
|
int nparams;
|
2004-08-29 13:07:03 +08:00
|
|
|
|
2008-03-26 06:42:46 +08:00
|
|
|
pstr = text_to_cstring(paramstr);
|
2004-08-29 13:07:03 +08:00
|
|
|
|
2010-08-11 07:02:00 +08:00
|
|
|
max_params = 20; /* must be even! */
|
|
|
|
params = (const char **) palloc((max_params + 1) * sizeof(char *));
|
|
|
|
nparams = 0;
|
|
|
|
|
2004-08-29 13:07:03 +08:00
|
|
|
pos = pstr;
|
|
|
|
|
2010-08-11 07:02:00 +08:00
|
|
|
while (*pos != '\0')
|
2004-08-29 13:07:03 +08:00
|
|
|
{
|
2010-08-11 07:02:00 +08:00
|
|
|
if (nparams >= max_params)
|
|
|
|
{
|
|
|
|
max_params *= 2;
|
|
|
|
params = (const char **) repalloc(params,
|
2011-04-10 23:42:00 +08:00
|
|
|
(max_params + 1) * sizeof(char *));
|
2010-08-11 07:02:00 +08:00
|
|
|
}
|
|
|
|
params[nparams++] = pos;
|
2004-08-29 13:07:03 +08:00
|
|
|
pos = strstr(pos, nvsep);
|
|
|
|
if (pos != NULL)
|
|
|
|
{
|
|
|
|
*pos = '\0';
|
|
|
|
pos++;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2009-07-10 08:32:00 +08:00
|
|
|
/* No equal sign, so ignore this "parameter" */
|
2010-08-11 07:02:00 +08:00
|
|
|
nparams--;
|
2004-08-29 13:07:03 +08:00
|
|
|
break;
|
|
|
|
}
|
2010-08-11 07:02:00 +08:00
|
|
|
|
|
|
|
/* since max_params is even, we still have nparams < max_params */
|
|
|
|
params[nparams++] = pos;
|
2004-08-29 13:07:03 +08:00
|
|
|
pos = strstr(pos, itsep);
|
|
|
|
if (pos != NULL)
|
|
|
|
{
|
|
|
|
*pos = '\0';
|
|
|
|
pos++;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
break;
|
|
|
|
}
|
2009-07-10 08:32:00 +08:00
|
|
|
|
2010-08-11 07:02:00 +08:00
|
|
|
/* Add the terminator marker; we left room for it in the palloc's */
|
|
|
|
params[nparams] = NULL;
|
|
|
|
|
|
|
|
return params;
|
2004-03-05 11:24:50 +08:00
|
|
|
}
|
2010-03-02 02:07:59 +08:00
|
|
|
|
2010-07-07 03:19:02 +08:00
|
|
|
#endif /* USE_LIBXSLT */
|