plperl.c 64.9 KB
Newer Older
1 2 3
/**********************************************************************
 * plperl.c - perl as a procedural language for PostgreSQL
 *
4
 *	  $PostgreSQL: pgsql/src/pl/plperl/plperl.c,v 1.126 2007/02/01 19:10:29 momjian Exp $
5
 *
6 7
 **********************************************************************/

8
#include "postgres.h"
9
/* Defined by Perl */
10
#undef _
11 12

/* system stuff */
13
#include <ctype.h>
14
#include <fcntl.h>
15
#include <unistd.h>
A
 
Andrew Dunstan 已提交
16
#include <locale.h>
17 18

/* postgreSQL stuff */
19 20
#include "commands/trigger.h"
#include "executor/spi.h"
21
#include "funcapi.h"
22 23 24 25
#include "mb/pg_wchar.h"
#include "miscadmin.h"
#include "nodes/makefuncs.h"
#include "parser/parse_type.h"
26
#include "utils/guc.h"
27
#include "utils/lsyscache.h"
28
#include "utils/memutils.h"
29
#include "utils/typcache.h"
30
#include "utils/hsearch.h"
31 32

/* perl stuff */
A
 
Andrew Dunstan 已提交
33
#include "plperl.h"
34

35 36
PG_MODULE_MAGIC;

37 38 39 40 41 42
/**********************************************************************
 * The information we cache about loaded procedures
 **********************************************************************/
typedef struct plperl_proc_desc
{
	char	   *proname;
43 44
	TransactionId fn_xmin;
	CommandId	fn_cmin;
45
	bool		fn_readonly;
46
	bool		lanpltrusted;
47
	bool		fn_retistuple;	/* true, if function returns tuple */
B
Bruce Momjian 已提交
48
	bool		fn_retisset;	/* true, if function returns set */
B
Bruce Momjian 已提交
49
	bool		fn_retisarray;	/* true if function returns array */
50
	Oid			result_oid;		/* Oid of result type */
B
Bruce Momjian 已提交
51
	FmgrInfo	result_in_func; /* I/O function and arg for result type */
52
	Oid			result_typioparam;
53 54
	int			nargs;
	FmgrInfo	arg_out_func[FUNC_MAX_ARGS];
55
	bool		arg_is_rowtype[FUNC_MAX_ARGS];
56
	SV		   *reference;
57
} plperl_proc_desc;
58

59 60 61 62 63 64 65 66
/* hash table entry for proc desc  */

typedef struct plperl_proc_entry
{
	char proc_name[NAMEDATALEN];
	plperl_proc_desc *proc_data;
} plperl_proc_entry;

67 68 69 70 71 72 73
/*
 * The information we cache for the duration of a single call to a
 * function.
 */
typedef struct plperl_call_data
{
	plperl_proc_desc *prodesc;
B
Bruce Momjian 已提交
74 75 76 77 78
	FunctionCallInfo fcinfo;
	Tuplestorestate *tuple_store;
	TupleDesc	ret_tdesc;
	AttInMetadata *attinmeta;
	MemoryContext tmp_cxt;
79 80
} plperl_call_data;

A
 
Andrew Dunstan 已提交
81 82 83 84 85 86 87 88 89 90 91 92
/**********************************************************************
 * The information we cache about prepared and saved plans
 **********************************************************************/
typedef struct plperl_query_desc
{
	char		qname[sizeof(long) * 2 + 1];
	void	   *plan;
	int			nargs;
	Oid		   *argtypes;
	FmgrInfo   *arginfuncs;
	Oid		   *argtypioparams;
} plperl_query_desc;
93

94 95 96 97 98 99 100 101
/* hash table entry for query desc  */

typedef struct plperl_query_entry
{
	char query_name[NAMEDATALEN];
	plperl_query_desc *query_data;
} plperl_query_entry;

102 103 104
/**********************************************************************
 * Global data
 **********************************************************************/
105 106 107 108 109 110 111 112 113 114 115 116 117

typedef enum
{
	INTERP_NONE,
	INTERP_HELD,
	INTERP_TRUSTED,
	INTERP_UNTRUSTED,
	INTERP_BOTH
} InterpState;

static InterpState interp_state = INTERP_NONE;
static bool can_run_two = false;

118
static bool plperl_safe_init_done = false;
119 120 121 122 123 124
static PerlInterpreter *plperl_trusted_interp = NULL;
static PerlInterpreter *plperl_untrusted_interp = NULL;
static PerlInterpreter *plperl_held_interp = NULL;
static bool trusted_context;
static HTAB  *plperl_proc_hash = NULL;
static HTAB  *plperl_query_hash = NULL;
125

126 127
static bool plperl_use_strict = false;

128 129
/* this is saved and restored by plperl_call_handler */
static plperl_call_data *current_call_data = NULL;
130

131 132 133
/**********************************************************************
 * Forward declarations
 **********************************************************************/
B
Bruce Momjian 已提交
134
Datum		plperl_call_handler(PG_FUNCTION_ARGS);
135
Datum		plperl_validator(PG_FUNCTION_ARGS);
136 137 138
void		_PG_init(void);

static void plperl_init_interp(void);
139

140
static Datum plperl_func_handler(PG_FUNCTION_ARGS);
141

142
static Datum plperl_trigger_handler(PG_FUNCTION_ARGS);
143 144
static plperl_proc_desc *compile_plperl_function(Oid fn_oid, bool is_trigger);

145
static SV  *plperl_hash_from_tuple(HeapTuple tuple, TupleDesc tupdesc);
146
static void plperl_init_shared_libs(pTHX);
147
static HV  *plperl_spi_execute_fetch_result(SPITupleTable *, int, int);
148 149 150
static SV  *newSVstring(const char *str);
static SV **hv_store_string(HV *hv, const char *key, SV *val);
static SV **hv_fetch_string(HV *hv, const char *key);
151

152 153 154 155 156 157 158
/*
 * This routine is a crock, and so is everyplace that calls it.  The problem
 * is that the cached form of plperl functions/queries is allocated permanently
 * (mostly via malloc()) and never released until backend exit.  Subsidiary
 * data structures such as fmgr info records therefore must live forever
 * as well.  A better implementation would store all this stuff in a per-
 * function memory context that could be reclaimed at need.  In the meantime,
159 160 161
 * fmgr_info_cxt must be called specifying TopMemoryContext so that whatever
 * it might allocate, and whatever the eventual function might allocate using
 * fn_mcxt, will live forever too.
162 163 164 165
 */
static void
perm_fmgr_info(Oid functionId, FmgrInfo *finfo)
{
166
	fmgr_info_cxt(functionId, finfo, TopMemoryContext);
167 168
}

169

170 171 172 173 174
/*
 * _PG_init()			- library load-time initialization
 *
 * DO NOT make this static nor change its name!
 */
175
void
176
_PG_init(void)
177
{
178 179
	/* Be sure we do initialization only once (should be redundant now) */
	static bool inited = false;
180
    HASHCTL     hash_ctl;
181 182

	if (inited)
183 184
		return;

185
	DefineCustomBoolVariable("plperl.use_strict",
B
Bruce Momjian 已提交
186 187 188 189 190
	  "If true, will compile trusted and untrusted perl code in strict mode",
							 NULL,
							 &plperl_use_strict,
							 PGC_USERSET,
							 NULL, NULL);
191 192

	EmitWarningsOnPlaceholders("plperl");
193

194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
	MemSet(&hash_ctl, 0, sizeof(hash_ctl));

	hash_ctl.keysize = NAMEDATALEN;
	hash_ctl.entrysize = sizeof(plperl_proc_entry);

	plperl_proc_hash = hash_create("PLPerl Procedures",
								   32,
								   &hash_ctl,
								   HASH_ELEM);

	hash_ctl.entrysize = sizeof(plperl_query_entry);
	plperl_query_hash = hash_create("PLPerl Queries",
									32,
									&hash_ctl,
									HASH_ELEM);

210 211
	plperl_init_interp();

212
	inited = true;
213 214
}

215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241
/* Each of these macros must represent a single string literal */

#define PERLBOOT \
	"SPI::bootstrap(); use vars qw(%_SHARED);" \
	"sub ::plperl_warn { my $msg = shift; " \
	"       $msg =~ s/\\(eval \\d+\\) //g; &elog(&NOTICE, $msg); } " \
	"$SIG{__WARN__} = \\&::plperl_warn; " \
	"sub ::plperl_die { my $msg = shift; " \
	"       $msg =~ s/\\(eval \\d+\\) //g; die $msg; } " \
	"$SIG{__DIE__} = \\&::plperl_die; " \
	"sub ::mkunsafefunc {" \
	"      my $ret = eval(qq[ sub { $_[0] $_[1] } ]); " \
	"      $@ =~ s/\\(eval \\d+\\) //g if $@; return $ret; }" \
	"use strict; " \
	"sub ::mk_strict_unsafefunc {" \
	"      my $ret = eval(qq[ sub { use strict; $_[0] $_[1] } ]); " \
	"      $@ =~ s/\\(eval \\d+\\) //g if $@; return $ret; } " \
	"sub ::_plperl_to_pg_array {" \
	"  my $arg = shift; ref $arg eq 'ARRAY' || return $arg; " \
	"  my $res = ''; my $first = 1; " \
	"  foreach my $elem (@$arg) " \
	"  { " \
	"    $res .= ', ' unless $first; $first = undef; " \
	"    if (ref $elem) " \
	"    { " \
	"      $res .= _plperl_to_pg_array($elem); " \
	"    } " \
A
 
Andrew Dunstan 已提交
242
	"    elsif (defined($elem)) " \
243 244 245 246 247
	"    { " \
	"      my $str = qq($elem); " \
	"      $str =~ s/([\"\\\\])/\\\\$1/g; " \
	"      $res .= qq(\"$str\"); " \
	"    } " \
A
 
Andrew Dunstan 已提交
248 249 250 251
	"    else " \
	"    { "\
	"      $res .= 'NULL' ; " \
	"    } "\
252 253 254 255 256 257 258 259 260 261 262 263
	"  } " \
	"  return qq({$res}); " \
	"} "

#define SAFE_MODULE \
	"require Safe; $Safe::VERSION"

#define SAFE_OK \
	"use vars qw($PLContainer); $PLContainer = new Safe('PLPerl');" \
	"$PLContainer->permit_only(':default');" \
	"$PLContainer->permit(qw[:base_math !:base_io sort time]);" \
	"$PLContainer->share(qw[&elog &spi_exec_query &return_next " \
A
 
Andrew Dunstan 已提交
264 265
	"&spi_query &spi_fetchrow &spi_cursor_close " \
	"&spi_prepare &spi_exec_prepared &spi_query_prepared &spi_freeplan " \
266 267 268 269 270 271 272 273 274
	"&_plperl_to_pg_array " \
	"&DEBUG &LOG &INFO &NOTICE &WARNING &ERROR %_SHARED ]);" \
	"sub ::mksafefunc {" \
	"      my $ret = $PLContainer->reval(qq[sub { $_[0] $_[1] }]); " \
	"      $@ =~ s/\\(eval \\d+\\) //g if $@; return $ret; }" \
	"$PLContainer->permit('require'); $PLContainer->reval('use strict;');" \
	"$PLContainer->deny('require');" \
	"sub ::mk_strict_safefunc {" \
	"      my $ret = $PLContainer->reval(qq[sub { BEGIN { strict->import(); } $_[0] $_[1] }]); " \
B
Bruce Momjian 已提交
275
	"      $@ =~ s/\\(eval \\d+\\) //g if $@; return $ret; }"
276 277 278 279 280 281 282 283 284 285 286 287

#define SAFE_BAD \
	"use vars qw($PLContainer); $PLContainer = new Safe('PLPerl');" \
	"$PLContainer->permit_only(':default');" \
	"$PLContainer->share(qw[&elog &ERROR ]);" \
	"sub ::mksafefunc { return $PLContainer->reval(qq[sub { " \
	"      elog(ERROR,'trusted Perl functions disabled - " \
	"      please upgrade Perl Safe module to version 2.09 or later');}]); }" \
	"sub ::mk_strict_safefunc { return $PLContainer->reval(qq[sub { " \
	"      elog(ERROR,'trusted Perl functions disabled - " \
	"      please upgrade Perl Safe module to version 2.09 or later');}]); }"

288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352
#define TEST_FOR_MULTI \
	"use Config; " \
	"$Config{usemultiplicity} eq 'define' or "  \
    "($Config{usethreads} eq 'define' " \
	" and $Config{useithreads} eq 'define')"


/********************************************************************
 *
 * We start out by creating a "held" interpreter that we can use in
 * trusted or untrusted mode (but not both) as the need arises. Later, we
 * assign that interpreter if it is available to either the trusted or 
 * untrusted interpreter. If it has already been assigned, and we need to
 * create the other interpreter, we do that if we can, or error out.
 * We detect if it is safe to run two interpreters during the setup of the
 * dummy interpreter.
 */


static void 
check_interp(bool trusted)
{
	if (interp_state == INTERP_HELD)
	{
		if (trusted)
		{
			plperl_trusted_interp = plperl_held_interp;
			interp_state = INTERP_TRUSTED;
		}
		else
		{
			plperl_untrusted_interp = plperl_held_interp;
			interp_state = INTERP_UNTRUSTED;
		}
		plperl_held_interp = NULL;
		trusted_context = trusted;
	}
	else if (interp_state == INTERP_BOTH || 
			 (trusted && interp_state == INTERP_TRUSTED) ||
			 (!trusted && interp_state == INTERP_UNTRUSTED))
	{
		if (trusted_context != trusted)
		{
			if (trusted)
				PERL_SET_CONTEXT(plperl_trusted_interp);
			else
				PERL_SET_CONTEXT(plperl_untrusted_interp);
			trusted_context = trusted;
		}
	}
	else if (can_run_two)
	{
		PERL_SET_CONTEXT(plperl_held_interp);
		plperl_init_interp();
		if (trusted)
			plperl_trusted_interp = plperl_held_interp;
		else
			plperl_untrusted_interp = plperl_held_interp;
		interp_state = INTERP_BOTH;
		plperl_held_interp = NULL;
		trusted_context = trusted;
	}
	else
	{
		elog(ERROR, 
353
			 "cannot allocate second Perl interpreter on this platform");
354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371

	}
	
}


static void
restore_context (bool old_context)
{
	if (trusted_context != old_context)
	{
		if (old_context)
			PERL_SET_CONTEXT(plperl_trusted_interp);
		else
			PERL_SET_CONTEXT(plperl_untrusted_interp);
		trusted_context = old_context;
	}
}
372 373

static void
374
plperl_init_interp(void)
375
{
B
Bruce Momjian 已提交
376
	static char *embedding[3] = {
377
		"", "-e", PERLBOOT
378 379
	};

A
 
Andrew Dunstan 已提交
380 381
#ifdef WIN32

B
Bruce Momjian 已提交
382
	/*
A
 
Andrew Dunstan 已提交
383
	 * The perl library on startup does horrible things like call
B
Bruce Momjian 已提交
384 385 386 387 388
	 * setlocale(LC_ALL,""). We have protected against that on most platforms
	 * by setting the environment appropriately. However, on Windows,
	 * setlocale() does not consult the environment, so we need to save the
	 * existing locale settings before perl has a chance to mangle them and
	 * restore them after its dirty deeds are done.
A
 
Andrew Dunstan 已提交
389 390 391 392 393 394 395 396
	 *
	 * MSDN ref:
	 * http://msdn.microsoft.com/library/en-us/vclib/html/_crt_locale.asp
	 *
	 * It appears that we only need to do this on interpreter startup, and
	 * subsequent calls to the interpreter don't mess with the locale
	 * settings.
	 *
B
Bruce Momjian 已提交
397 398
	 * We restore them using Perl's POSIX::setlocale() function so that Perl
	 * doesn't have a different idea of the locale from Postgres.
A
 
Andrew Dunstan 已提交
399 400 401
	 *
	 */

B
Bruce Momjian 已提交
402 403 404 405 406 407 408
	char	   *loc;
	char	   *save_collate,
			   *save_ctype,
			   *save_monetary,
			   *save_numeric,
			   *save_time;
	char		buf[1024];
A
 
Andrew Dunstan 已提交
409

B
Bruce Momjian 已提交
410
	loc = setlocale(LC_COLLATE, NULL);
A
 
Andrew Dunstan 已提交
411
	save_collate = loc ? pstrdup(loc) : NULL;
B
Bruce Momjian 已提交
412
	loc = setlocale(LC_CTYPE, NULL);
A
 
Andrew Dunstan 已提交
413
	save_ctype = loc ? pstrdup(loc) : NULL;
B
Bruce Momjian 已提交
414
	loc = setlocale(LC_MONETARY, NULL);
A
 
Andrew Dunstan 已提交
415
	save_monetary = loc ? pstrdup(loc) : NULL;
B
Bruce Momjian 已提交
416
	loc = setlocale(LC_NUMERIC, NULL);
A
 
Andrew Dunstan 已提交
417
	save_numeric = loc ? pstrdup(loc) : NULL;
B
Bruce Momjian 已提交
418
	loc = setlocale(LC_TIME, NULL);
A
 
Andrew Dunstan 已提交
419 420 421
	save_time = loc ? pstrdup(loc) : NULL;
#endif

422 423 424

	plperl_held_interp = perl_alloc();
	if (!plperl_held_interp)
425
		elog(ERROR, "could not allocate Perl interpreter");
426

427 428 429 430
	perl_construct(plperl_held_interp);
	perl_parse(plperl_held_interp, plperl_init_shared_libs, 
			   3, embedding, NULL);
	perl_run(plperl_held_interp);
431

432 433 434 435 436 437 438 439
	if (interp_state == INTERP_NONE)
	{
		SV *res;

		res = eval_pv(TEST_FOR_MULTI,TRUE);
		can_run_two = SvIV(res); 
		interp_state = INTERP_HELD;
	}
A
 
Andrew Dunstan 已提交
440 441 442

#ifdef WIN32

B
Bruce Momjian 已提交
443
	eval_pv("use POSIX qw(locale_h);", TRUE);	/* croak on failure */
A
 
Andrew Dunstan 已提交
444 445 446

	if (save_collate != NULL)
	{
B
Bruce Momjian 已提交
447 448 449
		snprintf(buf, sizeof(buf), "setlocale(%s,'%s');",
				 "LC_COLLATE", save_collate);
		eval_pv(buf, TRUE);
A
 
Andrew Dunstan 已提交
450 451 452 453
		pfree(save_collate);
	}
	if (save_ctype != NULL)
	{
B
Bruce Momjian 已提交
454 455 456
		snprintf(buf, sizeof(buf), "setlocale(%s,'%s');",
				 "LC_CTYPE", save_ctype);
		eval_pv(buf, TRUE);
A
 
Andrew Dunstan 已提交
457 458 459 460
		pfree(save_ctype);
	}
	if (save_monetary != NULL)
	{
B
Bruce Momjian 已提交
461 462 463
		snprintf(buf, sizeof(buf), "setlocale(%s,'%s');",
				 "LC_MONETARY", save_monetary);
		eval_pv(buf, TRUE);
A
 
Andrew Dunstan 已提交
464 465 466 467
		pfree(save_monetary);
	}
	if (save_numeric != NULL)
	{
B
Bruce Momjian 已提交
468 469 470
		snprintf(buf, sizeof(buf), "setlocale(%s,'%s');",
				 "LC_NUMERIC", save_numeric);
		eval_pv(buf, TRUE);
A
 
Andrew Dunstan 已提交
471 472 473 474
		pfree(save_numeric);
	}
	if (save_time != NULL)
	{
B
Bruce Momjian 已提交
475 476 477
		snprintf(buf, sizeof(buf), "setlocale(%s,'%s');",
				 "LC_TIME", save_time);
		eval_pv(buf, TRUE);
A
 
Andrew Dunstan 已提交
478 479 480 481
		pfree(save_time);
	}
#endif

482 483
}

484 485 486 487

static void
plperl_safe_init(void)
{
B
Bruce Momjian 已提交
488
	SV		   *res;
489
	double		safe_version;
490

491
	res = eval_pv(SAFE_MODULE, FALSE);	/* TRUE = croak if failure */
492 493 494

	safe_version = SvNV(res);

495 496 497 498 499
	/*
	 * We actually want to reject safe_version < 2.09, but it's risky to
	 * assume that floating-point comparisons are exact, so use a slightly
	 * smaller comparison value.
	 */
B
Bruce Momjian 已提交
500
	if (safe_version < 2.0899)
501 502
	{
		/* not safe, so disallow all trusted funcs */
503
		eval_pv(SAFE_BAD, FALSE);
504 505 506
	}
	else
	{
507
		eval_pv(SAFE_OK, FALSE);
508
	}
509 510 511 512

	plperl_safe_init_done = true;
}

513 514 515 516 517 518
/*
 * Perl likes to put a newline after its error messages; clean up such
 */
static char *
strip_trailing_ws(const char *msg)
{
B
Bruce Momjian 已提交
519 520
	char	   *res = pstrdup(msg);
	int			len = strlen(res);
521

B
Bruce Momjian 已提交
522
	while (len > 0 && isspace((unsigned char) res[len - 1]))
523 524 525 526 527
		res[--len] = '\0';
	return res;
}


528 529
/* Build a tuple from a hash. */

530
static HeapTuple
531
plperl_build_tuple_result(HV *perlhash, AttInMetadata *attinmeta)
532
{
533 534 535 536 537 538
	TupleDesc	td = attinmeta->tupdesc;
	char	  **values;
	SV		   *val;
	char	   *key;
	I32			klen;
	HeapTuple	tup;
539

540
	values = (char **) palloc0(td->natts * sizeof(char *));
541

542 543 544
	hv_iterinit(perlhash);
	while ((val = hv_iternextsv(perlhash, &key, &klen)))
	{
B
Bruce Momjian 已提交
545
		int			attn = SPI_fnumber(td, key);
546

547
		if (attn <= 0 || td->attrs[attn - 1]->attisdropped)
548 549 550 551
			ereport(ERROR,
					(errcode(ERRCODE_UNDEFINED_COLUMN),
					 errmsg("Perl hash contains nonexistent column \"%s\"",
							key)));
552
		if (SvOK(val) && SvTYPE(val) != SVt_NULL)
553
			values[attn - 1] = SvPV(val, PL_na);
554
	}
555 556 557 558 559
	hv_iterinit(perlhash);

	tup = BuildTupleFromCStrings(attinmeta, values);
	pfree(values);
	return tup;
560 561
}

562 563 564
/*
 * convert perl array to postgres string representation
 */
B
Bruce Momjian 已提交
565
static SV  *
566
plperl_convert_to_pg_array(SV *src)
567
{
B
Bruce Momjian 已提交
568 569 570 571
	SV		   *rv;
	int			count;

	dSP;
572

B
Bruce Momjian 已提交
573
	PUSHMARK(SP);
574
	XPUSHs(src);
B
Bruce Momjian 已提交
575
	PUTBACK;
576

577
	count = call_pv("::_plperl_to_pg_array", G_SCALAR);
578

B
Bruce Momjian 已提交
579
	SPAGAIN;
580 581

	if (count != 1)
582
		elog(ERROR, "unexpected _plperl_to_pg_array failure");
583 584 585

	rv = POPs;

B
Bruce Momjian 已提交
586 587 588
	PUTBACK;

	return rv;
589 590
}

591

592 593
/* Set up the arguments for a trigger call. */

594 595 596 597 598
static SV  *
plperl_trigger_build_args(FunctionCallInfo fcinfo)
{
	TriggerData *tdata;
	TupleDesc	tupdesc;
599
	int			i;
600 601 602 603 604
	char	   *level;
	char	   *event;
	char	   *relid;
	char	   *when;
	HV		   *hv;
605

606
	hv = newHV();
607 608 609 610

	tdata = (TriggerData *) fcinfo->context;
	tupdesc = tdata->tg_relation->rd_att;

611
	relid = DatumGetCString(
B
Bruce Momjian 已提交
612 613 614 615
							DirectFunctionCall1(oidout,
								  ObjectIdGetDatum(tdata->tg_relation->rd_id)
												)
		);
616

617 618
	hv_store_string(hv, "name", newSVstring(tdata->tg_trigger->tgname));
	hv_store_string(hv, "relid", newSVstring(relid));
619 620 621

	if (TRIGGER_FIRED_BY_INSERT(tdata->tg_event))
	{
622
		event = "INSERT";
623
		if (TRIGGER_FIRED_FOR_ROW(tdata->tg_event))
624 625 626
			hv_store_string(hv, "new",
							plperl_hash_from_tuple(tdata->tg_trigtuple,
												   tupdesc));
627 628 629
	}
	else if (TRIGGER_FIRED_BY_DELETE(tdata->tg_event))
	{
630
		event = "DELETE";
631
		if (TRIGGER_FIRED_FOR_ROW(tdata->tg_event))
632 633 634
			hv_store_string(hv, "old",
							plperl_hash_from_tuple(tdata->tg_trigtuple,
												   tupdesc));
635 636 637
	}
	else if (TRIGGER_FIRED_BY_UPDATE(tdata->tg_event))
	{
638
		event = "UPDATE";
639 640
		if (TRIGGER_FIRED_FOR_ROW(tdata->tg_event))
		{
641 642 643 644 645 646
			hv_store_string(hv, "old",
							plperl_hash_from_tuple(tdata->tg_trigtuple,
												   tupdesc));
			hv_store_string(hv, "new",
							plperl_hash_from_tuple(tdata->tg_newtuple,
												   tupdesc));
647
		}
648
	}
649
	else
650
		event = "UNKNOWN";
651

652 653
	hv_store_string(hv, "event", newSVstring(event));
	hv_store_string(hv, "argc", newSViv(tdata->tg_trigger->tgnargs));
654

655
	if (tdata->tg_trigger->tgnargs > 0)
656
	{
B
Bruce Momjian 已提交
657 658 659
		AV		   *av = newAV();

		for (i = 0; i < tdata->tg_trigger->tgnargs; i++)
660 661
			av_push(av, newSVstring(tdata->tg_trigger->tgargs[i]));
		hv_store_string(hv, "args", newRV_noinc((SV *) av));
662
	}
663

664 665
	hv_store_string(hv, "relname",
					newSVstring(SPI_getrelname(tdata->tg_relation)));
666

667 668
	hv_store_string(hv, "table_name",
					newSVstring(SPI_getrelname(tdata->tg_relation)));
A
 
Andrew Dunstan 已提交
669

670 671
	hv_store_string(hv, "table_schema",
					newSVstring(SPI_getnspname(tdata->tg_relation)));
A
 
Andrew Dunstan 已提交
672

673
	if (TRIGGER_FIRED_BEFORE(tdata->tg_event))
674
		when = "BEFORE";
675
	else if (TRIGGER_FIRED_AFTER(tdata->tg_event))
676
		when = "AFTER";
677
	else
678
		when = "UNKNOWN";
679
	hv_store_string(hv, "when", newSVstring(when));
680 681

	if (TRIGGER_FIRED_FOR_ROW(tdata->tg_event))
682
		level = "ROW";
683
	else if (TRIGGER_FIRED_FOR_STATEMENT(tdata->tg_event))
684
		level = "STATEMENT";
685
	else
686
		level = "UNKNOWN";
687
	hv_store_string(hv, "level", newSVstring(level));
688

B
Bruce Momjian 已提交
689
	return newRV_noinc((SV *) hv);
690 691 692
}


693
/* Set up the new tuple returned from a trigger. */
694

695
static HeapTuple
696
plperl_modify_tuple(HV *hvTD, TriggerData *tdata, HeapTuple otup)
697 698 699 700
{
	SV		  **svp;
	HV		   *hvNew;
	HeapTuple	rtup;
701 702 703 704 705 706 707 708
	SV		   *val;
	char	   *key;
	I32			klen;
	int			slotsused;
	int		   *modattrs;
	Datum	   *modvalues;
	char	   *modnulls;

709 710 711 712
	TupleDesc	tupdesc;

	tupdesc = tdata->tg_relation->rd_att;

713
	svp = hv_fetch_string(hvTD, "new");
714
	if (!svp)
715 716 717
		ereport(ERROR,
				(errcode(ERRCODE_UNDEFINED_COLUMN),
				 errmsg("$_TD->{new} does not exist")));
718
	if (!SvOK(*svp) || SvTYPE(*svp) != SVt_RV || SvTYPE(SvRV(*svp)) != SVt_PVHV)
719 720 721
		ereport(ERROR,
				(errcode(ERRCODE_DATATYPE_MISMATCH),
				 errmsg("$_TD->{new} is not a hash reference")));
722 723
	hvNew = (HV *) SvRV(*svp);

724 725 726 727
	modattrs = palloc(tupdesc->natts * sizeof(int));
	modvalues = palloc(tupdesc->natts * sizeof(Datum));
	modnulls = palloc(tupdesc->natts * sizeof(char));
	slotsused = 0;
728

729 730
	hv_iterinit(hvNew);
	while ((val = hv_iternextsv(hvNew, &key, &klen)))
731
	{
732
		int			attn = SPI_fnumber(tupdesc, key);
733 734 735 736
		Oid			typinput;
		Oid			typioparam;
		int32		atttypmod;
		FmgrInfo	finfo;
737

738
		if (attn <= 0 || tupdesc->attrs[attn - 1]->attisdropped)
739 740 741 742
			ereport(ERROR,
					(errcode(ERRCODE_UNDEFINED_COLUMN),
					 errmsg("Perl hash contains nonexistent column \"%s\"",
							key)));
743 744 745 746 747
		/* XXX would be better to cache these lookups */
		getTypeInputInfo(tupdesc->attrs[attn - 1]->atttypid,
						 &typinput, &typioparam);
		fmgr_info(typinput, &finfo);
		atttypmod = tupdesc->attrs[attn - 1]->atttypmod;
748
		if (SvOK(val) && SvTYPE(val) != SVt_NULL)
749
		{
750 751 752 753
			modvalues[slotsused] = InputFunctionCall(&finfo,
													 SvPV(val, PL_na),
													 typioparam,
													 atttypmod);
754
			modnulls[slotsused] = ' ';
755 756 757
		}
		else
		{
758 759 760 761
			modvalues[slotsused] = InputFunctionCall(&finfo,
													 NULL,
													 typioparam,
													 atttypmod);
762
			modnulls[slotsused] = 'n';
763
		}
764 765
		modattrs[slotsused] = attn;
		slotsused++;
766
	}
767 768 769 770
	hv_iterinit(hvNew);

	rtup = SPI_modifytuple(tdata->tg_relation, otup, slotsused,
						   modattrs, modvalues, modnulls);
771 772 773 774

	pfree(modattrs);
	pfree(modvalues);
	pfree(modnulls);
775

776
	if (rtup == NULL)
777
		elog(ERROR, "SPI_modifytuple failed: %s",
778
			 SPI_result_code_string(SPI_result));
779 780 781

	return rtup;
}
782

783

784 785
/*
 * This is the only externally-visible part of the plperl call interface.
786
 * The Postgres function and trigger managers call it to execute a
787 788
 * perl function.
 */
789
PG_FUNCTION_INFO_V1(plperl_call_handler);
790 791

Datum
792
plperl_call_handler(PG_FUNCTION_ARGS)
793
{
B
Bruce Momjian 已提交
794
	Datum		retval;
795
	plperl_call_data *save_call_data;
796

797
	save_call_data = current_call_data;
798 799 800 801 802 803 804 805 806
	PG_TRY();
	{
		if (CALLED_AS_TRIGGER(fcinfo))
			retval = PointerGetDatum(plperl_trigger_handler(fcinfo));
		else
			retval = plperl_func_handler(fcinfo);
	}
	PG_CATCH();
	{
807
		current_call_data = save_call_data;
808 809 810 811
		PG_RE_THROW();
	}
	PG_END_TRY();

812
	current_call_data = save_call_data;
813 814 815
	return retval;
}

816 817 818 819 820 821 822 823 824 825 826 827
/*
 * This is the other externally visible function - it is called when CREATE
 * FUNCTION is issued to validate the function being created/replaced.
 */
PG_FUNCTION_INFO_V1(plperl_validator);

Datum
plperl_validator(PG_FUNCTION_ARGS)
{
	Oid			funcoid = PG_GETARG_OID(0);
	HeapTuple	tuple;
	Form_pg_proc proc;
828
	char		functyptype;
829 830 831 832
	int			numargs;
	Oid		   *argtypes;
	char	  **argnames;
	char	   *argmodes;
833
	bool		istrigger = false;
834
	int			i;
835 836 837 838 839 840 841 842 843

	/* Get the new function's pg_proc entry */
	tuple = SearchSysCache(PROCOID,
						   ObjectIdGetDatum(funcoid),
						   0, 0, 0);
	if (!HeapTupleIsValid(tuple))
		elog(ERROR, "cache lookup failed for function %u", funcoid);
	proc = (Form_pg_proc) GETSTRUCT(tuple);

844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861
	functyptype = get_typtype(proc->prorettype);

	/* Disallow pseudotype result */
	/* except for TRIGGER, RECORD, or VOID */
	if (functyptype == 'p')
	{
		/* we assume OPAQUE with no arguments means a trigger */
		if (proc->prorettype == TRIGGEROID ||
			(proc->prorettype == OPAQUEOID && proc->pronargs == 0))
			istrigger = true;
		else if (proc->prorettype != RECORDOID &&
				 proc->prorettype != VOIDOID)
			ereport(ERROR,
					(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
					 errmsg("plperl functions cannot return type %s",
							format_type_be(proc->prorettype))));
	}

862 863 864 865 866 867 868 869 870 871 872 873
	/* Disallow pseudotypes in arguments (either IN or OUT) */
	numargs = get_func_arg_info(tuple,
								&argtypes, &argnames, &argmodes);
	for (i = 0; i < numargs; i++)
	{
		if (get_typtype(argtypes[i]) == 'p')
			ereport(ERROR,
					(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
					 errmsg("plperl functions cannot take type %s",
							format_type_be(argtypes[i]))));
	}

874 875
	ReleaseSysCache(tuple);

876 877 878
	/* Postpone body checks if !check_function_bodies */
	if (check_function_bodies)
	{
879
		(void) compile_plperl_function(funcoid, istrigger);
880
	}
881 882 883 884 885

	/* the result of a validator is ignored */
	PG_RETURN_VOID();
}

886

887 888 889 890
/*
 * Uses mksafefunc/mkunsafefunc to create an anonymous sub whose text is
 * supplied in s, and returns a reference to the closure.
 */
B
Bruce Momjian 已提交
891
static SV  *
892
plperl_create_sub(char *s, bool trusted)
893
{
894
	dSP;
895
	SV		   *subref;
B
Bruce Momjian 已提交
896
	int			count;
B
Bruce Momjian 已提交
897
	char	   *compile_sub;
898

B
Bruce Momjian 已提交
899
	if (trusted && !plperl_safe_init_done)
900
	{
901
		plperl_safe_init();
902 903
		SPAGAIN;
	}
904

905 906 907
	ENTER;
	SAVETMPS;
	PUSHMARK(SP);
908 909
	XPUSHs(sv_2mortal(newSVstring("our $_TD; local $_TD=$_[0]; shift;")));
	XPUSHs(sv_2mortal(newSVstring(s)));
B
Bruce Momjian 已提交
910
	PUTBACK;
B
Bruce Momjian 已提交
911

912 913
	/*
	 * G_KEEPERR seems to be needed here, else we don't recognize compile
B
Bruce Momjian 已提交
914 915
	 * errors properly.  Perhaps it's because there's another level of eval
	 * inside mksafefunc?
916
	 */
917 918 919 920 921 922 923 924 925 926 927

	if (trusted && plperl_use_strict)
		compile_sub = "::mk_strict_safefunc";
	else if (plperl_use_strict)
		compile_sub = "::mk_strict_unsafefunc";
	else if (trusted)
		compile_sub = "::mksafefunc";
	else
		compile_sub = "::mkunsafefunc";

	count = perl_call_pv(compile_sub, G_SCALAR | G_EVAL | G_KEEPERR);
928 929
	SPAGAIN;

930 931 932 933 934
	if (count != 1)
	{
		PUTBACK;
		FREETMPS;
		LEAVE;
935
		elog(ERROR, "didn't get a return item from mksafefunc");
936 937
	}

938
	if (SvTRUE(ERRSV))
939
	{
940
		(void) POPs;
941 942 943
		PUTBACK;
		FREETMPS;
		LEAVE;
944 945 946 947
		ereport(ERROR,
				(errcode(ERRCODE_SYNTAX_ERROR),
				 errmsg("creation of Perl function failed: %s",
						strip_trailing_ws(SvPV(ERRSV, PL_na)))));
948 949 950
	}

	/*
951 952
	 * need to make a deep copy of the return. it comes off the stack as a
	 * temporary.
953 954 955
	 */
	subref = newSVsv(POPs);

956
	if (!SvROK(subref) || SvTYPE(SvRV(subref)) != SVt_PVCV)
957
	{
958 959 960
		PUTBACK;
		FREETMPS;
		LEAVE;
961

962 963 964 965
		/*
		 * subref is our responsibility because it is not mortal
		 */
		SvREFCNT_dec(subref);
966
		elog(ERROR, "didn't get a code ref");
967 968 969 970 971
	}

	PUTBACK;
	FREETMPS;
	LEAVE;
972

973 974 975
	return subref;
}

976

977
/**********************************************************************
978
 * plperl_init_shared_libs()		-
979 980 981 982
 *
 * We cannot use the DynaLoader directly to get at the Opcode
 * module (used by Safe.pm). So, we link Opcode into ourselves
 * and do the initialization behind perl's back.
983
 *
984 985
 **********************************************************************/

986 987
EXTERN_C void boot_DynaLoader(pTHX_ CV *cv);
EXTERN_C void boot_SPI(pTHX_ CV *cv);
988

989
static void
990
plperl_init_shared_libs(pTHX)
991
{
992 993
	char	   *file = __FILE__;

994
	newXS("DynaLoader::boot_DynaLoader", boot_DynaLoader, file);
995
	newXS("SPI::bootstrap", boot_SPI, file);
996 997
}

998

B
Bruce Momjian 已提交
999
static SV  *
1000
plperl_call_perl_func(plperl_proc_desc *desc, FunctionCallInfo fcinfo)
1001 1002
{
	dSP;
1003 1004 1005
	SV		   *retval;
	int			i;
	int			count;
B
Bruce Momjian 已提交
1006
	SV		   *sv;
1007 1008 1009 1010

	ENTER;
	SAVETMPS;

1011
	PUSHMARK(SP);
1012

B
Bruce Momjian 已提交
1013
	XPUSHs(&PL_sv_undef);		/* no trigger data */
1014

1015 1016
	for (i = 0; i < desc->nargs; i++)
	{
1017 1018 1019
		if (fcinfo->argnull[i])
			XPUSHs(&PL_sv_undef);
		else if (desc->arg_is_rowtype[i])
1020
		{
1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036
			HeapTupleHeader td;
			Oid			tupType;
			int32		tupTypmod;
			TupleDesc	tupdesc;
			HeapTupleData tmptup;
			SV		   *hashref;

			td = DatumGetHeapTupleHeader(fcinfo->arg[i]);
			/* Extract rowtype info and find a tupdesc */
			tupType = HeapTupleHeaderGetTypeId(td);
			tupTypmod = HeapTupleHeaderGetTypMod(td);
			tupdesc = lookup_rowtype_tupdesc(tupType, tupTypmod);
			/* Build a temporary HeapTuple control structure */
			tmptup.t_len = HeapTupleHeaderGetDatumLength(td);
			tmptup.t_data = td;

1037 1038
			hashref = plperl_hash_from_tuple(&tmptup, tupdesc);
			XPUSHs(sv_2mortal(hashref));
1039
			ReleaseTupleDesc(tupdesc);
1040 1041 1042
		}
		else
		{
1043 1044
			char	   *tmp;

1045 1046
			tmp = OutputFunctionCall(&(desc->arg_out_func[i]),
									 fcinfo->arg[i]);
1047
			sv = newSVstring(tmp);
1048
			XPUSHs(sv_2mortal(sv));
1049
			pfree(tmp);
1050 1051 1052
		}
	}
	PUTBACK;
1053 1054 1055

	/* Do NOT use G_KEEPERR here */
	count = perl_call_sv(desc->reference, G_SCALAR | G_EVAL);
1056 1057 1058

	SPAGAIN;

1059 1060 1061 1062
	if (count != 1)
	{
		PUTBACK;
		FREETMPS;
1063
		LEAVE;
1064
		elog(ERROR, "didn't get a return item from function");
1065 1066
	}

1067
	if (SvTRUE(ERRSV))
1068
	{
1069
		(void) POPs;
1070 1071
		PUTBACK;
		FREETMPS;
1072
		LEAVE;
1073 1074 1075 1076
		/* XXX need to find a way to assign an errcode here */
		ereport(ERROR,
				(errmsg("error from Perl function: %s",
						strip_trailing_ws(SvPV(ERRSV, PL_na)))));
1077 1078 1079 1080
	}

	retval = newSVsv(POPs);

1081 1082 1083
	PUTBACK;
	FREETMPS;
	LEAVE;
1084 1085 1086 1087

	return retval;
}

1088

1089
static SV  *
1090 1091
plperl_call_perl_trigger_func(plperl_proc_desc *desc, FunctionCallInfo fcinfo,
							  SV *td)
1092 1093 1094
{
	dSP;
	SV		   *retval;
1095
	Trigger    *tg_trigger;
1096 1097 1098 1099 1100 1101 1102
	int			i;
	int			count;

	ENTER;
	SAVETMPS;

	PUSHMARK(sp);
1103

1104
	XPUSHs(td);
1105

1106 1107
	tg_trigger = ((TriggerData *) fcinfo->context)->tg_trigger;
	for (i = 0; i < tg_trigger->tgnargs; i++)
1108
		XPUSHs(sv_2mortal(newSVstring(tg_trigger->tgargs[i])));
1109 1110
	PUTBACK;

1111 1112
	/* Do NOT use G_KEEPERR here */
	count = perl_call_sv(desc->reference, G_SCALAR | G_EVAL);
1113 1114 1115 1116 1117 1118 1119 1120

	SPAGAIN;

	if (count != 1)
	{
		PUTBACK;
		FREETMPS;
		LEAVE;
1121
		elog(ERROR, "didn't get a return item from trigger function");
1122 1123 1124 1125
	}

	if (SvTRUE(ERRSV))
	{
1126
		(void) POPs;
1127 1128 1129
		PUTBACK;
		FREETMPS;
		LEAVE;
1130 1131 1132 1133
		/* XXX need to find a way to assign an errcode here */
		ereport(ERROR,
				(errmsg("error from Perl trigger function: %s",
						strip_trailing_ws(SvPV(ERRSV, PL_na)))));
1134 1135 1136 1137 1138 1139 1140 1141 1142 1143
	}

	retval = newSVsv(POPs);

	PUTBACK;
	FREETMPS;
	LEAVE;

	return retval;
}
1144

1145

1146
static Datum
1147
plperl_func_handler(PG_FUNCTION_ARGS)
1148 1149
{
	plperl_proc_desc *prodesc;
1150 1151
	SV		   *perlret;
	Datum		retval;
1152
	ReturnSetInfo *rsi;
B
Bruce Momjian 已提交
1153
	SV		   *array_ret = NULL;
1154
	bool       oldcontext = trusted_context;
1155

1156
	/*
B
Bruce Momjian 已提交
1157 1158
	 * Create the call_data beforing connecting to SPI, so that it is not
	 * allocated in the SPI memory context
1159 1160 1161 1162
	 */
	current_call_data = (plperl_call_data *) palloc0(sizeof(plperl_call_data));
	current_call_data->fcinfo = fcinfo;

1163 1164 1165
	if (SPI_connect() != SPI_OK_CONNECT)
		elog(ERROR, "could not connect to SPI manager");

1166
	prodesc = compile_plperl_function(fcinfo->flinfo->fn_oid, false);
1167
	current_call_data->prodesc = prodesc;
1168

B
Bruce Momjian 已提交
1169
	rsi = (ReturnSetInfo *) fcinfo->resultinfo;
B
Bruce Momjian 已提交
1170

T
Tom Lane 已提交
1171
	if (prodesc->fn_retisset)
1172
	{
T
Tom Lane 已提交
1173 1174 1175 1176 1177 1178 1179 1180
		/* Check context before allowing the call to go through */
		if (!rsi || !IsA(rsi, ReturnSetInfo) ||
			(rsi->allowedModes & SFRM_Materialize) == 0 ||
			rsi->expectedDesc == NULL)
			ereport(ERROR,
					(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
					 errmsg("set-valued function called in context that "
							"cannot accept a set")));
1181 1182
	}

1183 1184
	check_interp(prodesc->lanpltrusted);

1185
	perlret = plperl_call_perl_func(prodesc, fcinfo);
1186 1187 1188 1189 1190 1191 1192 1193

	/************************************************************
	 * Disconnect from SPI manager and then create the return
	 * values datum (if the input function does a palloc for it
	 * this must not be allocated in the SPI memory context
	 * because SPI_finish would free it).
	 ************************************************************/
	if (SPI_finish() != SPI_OK_FINISH)
1194
		elog(ERROR, "SPI_finish() failed");
1195

T
Tom Lane 已提交
1196
	if (prodesc->fn_retisset)
1197
	{
T
Tom Lane 已提交
1198 1199
		/*
		 * If the Perl function returned an arrayref, we pretend that it
B
Bruce Momjian 已提交
1200 1201 1202
		 * called return_next() for each element of the array, to handle old
		 * SRFs that didn't know about return_next(). Any other sort of return
		 * value is an error.
T
Tom Lane 已提交
1203
		 */
1204 1205
		if (SvTYPE(perlret) == SVt_RV &&
			SvTYPE(SvRV(perlret)) == SVt_PVAV)
1206
		{
B
Bruce Momjian 已提交
1207 1208 1209 1210 1211
			int			i = 0;
			SV		  **svp = 0;
			AV		   *rav = (AV *) SvRV(perlret);

			while ((svp = av_fetch(rav, i, FALSE)) != NULL)
1212
			{
1213 1214 1215
				plperl_return_next(*svp);
				i++;
			}
1216
		}
1217
		else if (SvTYPE(perlret) != SVt_NULL)
1218
		{
1219 1220
			ereport(ERROR,
					(errcode(ERRCODE_DATATYPE_MISMATCH),
1221 1222
					 errmsg("set-returning Perl function must return "
							"reference to array or use return_next")));
1223
		}
B
Bruce Momjian 已提交
1224

1225
		rsi->returnMode = SFRM_Materialize;
1226
		if (current_call_data->tuple_store)
1227
		{
1228 1229
			rsi->setResult = current_call_data->tuple_store;
			rsi->setDesc = current_call_data->ret_tdesc;
1230
		}
B
Bruce Momjian 已提交
1231
		retval = (Datum) 0;
1232 1233 1234 1235 1236 1237
	}
	else if (SvTYPE(perlret) == SVt_NULL)
	{
		/* Return NULL if Perl code returned undef */
		if (rsi && IsA(rsi, ReturnSetInfo))
			rsi->isDone = ExprEndResult;
1238 1239
		retval = InputFunctionCall(&prodesc->result_in_func, NULL,
								   prodesc->result_typioparam, -1);
1240
		fcinfo->isnull = true;
B
Bruce Momjian 已提交
1241
	}
1242
	else if (prodesc->fn_retistuple)
1243
	{
1244
		/* Return a perl hash converted to a Datum */
B
Bruce Momjian 已提交
1245
		TupleDesc	td;
1246
		AttInMetadata *attinmeta;
B
Bruce Momjian 已提交
1247
		HeapTuple	tup;
1248

1249 1250 1251
		if (!SvOK(perlret) || SvTYPE(perlret) != SVt_RV ||
			SvTYPE(SvRV(perlret)) != SVt_PVHV)
		{
1252 1253
			ereport(ERROR,
					(errcode(ERRCODE_DATATYPE_MISMATCH),
1254 1255 1256
					 errmsg("composite-returning Perl function "
							"must return reference to hash")));
		}
1257

1258 1259 1260 1261 1262 1263 1264 1265
		/* XXX should cache the attinmeta data instead of recomputing */
		if (get_call_result_type(fcinfo, NULL, &td) != TYPEFUNC_COMPOSITE)
		{
			ereport(ERROR,
					(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
					 errmsg("function returning record called in context "
							"that cannot accept type record")));
		}
1266

1267
		attinmeta = TupleDescGetAttInMetadata(td);
B
Bruce Momjian 已提交
1268
		tup = plperl_build_tuple_result((HV *) SvRV(perlret), attinmeta);
1269 1270 1271 1272
		retval = HeapTupleGetDatum(tup);
	}
	else
	{
B
Bruce Momjian 已提交
1273 1274 1275 1276
		/* Return a perl string converted to a Datum */
		char	   *val;

		if (prodesc->fn_retisarray && SvROK(perlret) &&
1277
			SvTYPE(SvRV(perlret)) == SVt_PVAV)
B
Bruce Momjian 已提交
1278 1279 1280 1281 1282
		{
			array_ret = plperl_convert_to_pg_array(perlret);
			SvREFCNT_dec(perlret);
			perlret = array_ret;
		}
1283 1284 1285

		val = SvPV(perlret, PL_na);

1286 1287
		retval = InputFunctionCall(&prodesc->result_in_func, val,
								   prodesc->result_typioparam, -1);
1288
	}
1289

1290
	if (array_ret == NULL)
B
Bruce Momjian 已提交
1291
		SvREFCNT_dec(perlret);
1292

1293
	current_call_data = NULL;
1294 1295
	restore_context(oldcontext);

1296 1297 1298
	return retval;
}

1299

1300 1301 1302 1303 1304 1305 1306 1307
static Datum
plperl_trigger_handler(PG_FUNCTION_ARGS)
{
	plperl_proc_desc *prodesc;
	SV		   *perlret;
	Datum		retval;
	SV		   *svTD;
	HV		   *hvTD;
1308
	bool       oldcontext = trusted_context;
1309

1310
	/*
B
Bruce Momjian 已提交
1311 1312
	 * Create the call_data beforing connecting to SPI, so that it is not
	 * allocated in the SPI memory context
1313 1314 1315 1316
	 */
	current_call_data = (plperl_call_data *) palloc0(sizeof(plperl_call_data));
	current_call_data->fcinfo = fcinfo;

1317 1318 1319 1320
	/* Connect to SPI manager */
	if (SPI_connect() != SPI_OK_CONNECT)
		elog(ERROR, "could not connect to SPI manager");

1321 1322
	/* Find or compile the function */
	prodesc = compile_plperl_function(fcinfo->flinfo->fn_oid, true);
1323
	current_call_data->prodesc = prodesc;
1324

1325 1326
	check_interp(prodesc->lanpltrusted);

1327 1328
	svTD = plperl_trigger_build_args(fcinfo);
	perlret = plperl_call_perl_trigger_func(prodesc, fcinfo, svTD);
1329
	hvTD = (HV *) SvRV(svTD);
1330 1331 1332 1333 1334 1335 1336 1337

	/************************************************************
	* Disconnect from SPI manager and then create the return
	* values datum (if the input function does a palloc for it
	* this must not be allocated in the SPI memory context
	* because SPI_finish would free it).
	************************************************************/
	if (SPI_finish() != SPI_OK_FINISH)
1338
		elog(ERROR, "SPI_finish() failed");
1339

1340
	if (!(perlret && SvOK(perlret) && SvTYPE(perlret) != SVt_NULL))
1341
	{
1342
		/* undef result means go ahead with original tuple */
1343 1344 1345 1346 1347 1348 1349 1350
		TriggerData *trigdata = ((TriggerData *) fcinfo->context);

		if (TRIGGER_FIRED_BY_INSERT(trigdata->tg_event))
			retval = (Datum) trigdata->tg_trigtuple;
		else if (TRIGGER_FIRED_BY_UPDATE(trigdata->tg_event))
			retval = (Datum) trigdata->tg_newtuple;
		else if (TRIGGER_FIRED_BY_DELETE(trigdata->tg_event))
			retval = (Datum) trigdata->tg_trigtuple;
1351
		else
B
Bruce Momjian 已提交
1352
			retval = (Datum) 0; /* can this happen? */
1353 1354 1355
	}
	else
	{
1356 1357
		HeapTuple	trv;
		char	   *tmp;
1358

1359
		tmp = SvPV(perlret, PL_na);
1360

1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372
		if (pg_strcasecmp(tmp, "SKIP") == 0)
			trv = NULL;
		else if (pg_strcasecmp(tmp, "MODIFY") == 0)
		{
			TriggerData *trigdata = (TriggerData *) fcinfo->context;

			if (TRIGGER_FIRED_BY_INSERT(trigdata->tg_event))
				trv = plperl_modify_tuple(hvTD, trigdata,
										  trigdata->tg_trigtuple);
			else if (TRIGGER_FIRED_BY_UPDATE(trigdata->tg_event))
				trv = plperl_modify_tuple(hvTD, trigdata,
										  trigdata->tg_newtuple);
1373 1374
			else
			{
1375 1376
				ereport(WARNING,
						(errcode(ERRCODE_E_R_I_E_TRIGGER_PROTOCOL_VIOLATED),
B
Bruce Momjian 已提交
1377
					   errmsg("ignoring modified tuple in DELETE trigger")));
1378 1379 1380
				trv = NULL;
			}
		}
1381
		else
1382
		{
1383 1384
			ereport(ERROR,
					(errcode(ERRCODE_E_R_I_E_TRIGGER_PROTOCOL_VIOLATED),
1385 1386
					 errmsg("result of Perl trigger function must be undef, "
							"\"SKIP\" or \"MODIFY\"")));
1387 1388 1389
			trv = NULL;
		}
		retval = PointerGetDatum(trv);
1390 1391
	}

1392 1393 1394
	SvREFCNT_dec(svTD);
	if (perlret)
		SvREFCNT_dec(perlret);
1395

1396
	current_call_data = NULL;
1397
	restore_context(oldcontext);
1398 1399
	return retval;
}
1400

1401

1402 1403
static plperl_proc_desc *
compile_plperl_function(Oid fn_oid, bool is_trigger)
1404
{
1405 1406 1407 1408
	HeapTuple	procTup;
	Form_pg_proc procStruct;
	char		internal_proname[64];
	plperl_proc_desc *prodesc = NULL;
1409
	int			i;
1410 1411 1412
	plperl_proc_entry *hash_entry;
	bool found;
	bool oldcontext = trusted_context;
1413

1414 1415 1416 1417 1418
	/* We'll need the pg_proc tuple in any case... */
	procTup = SearchSysCache(PROCOID,
							 ObjectIdGetDatum(fn_oid),
							 0, 0, 0);
	if (!HeapTupleIsValid(procTup))
1419
		elog(ERROR, "cache lookup failed for function %u", fn_oid);
1420
	procStruct = (Form_pg_proc) GETSTRUCT(procTup);
1421 1422

	/************************************************************
1423
	 * Build our internal proc name from the function's Oid
1424
	 ************************************************************/
1425 1426 1427 1428
	if (!is_trigger)
		sprintf(internal_proname, "__PLPerl_proc_%u", fn_oid);
	else
		sprintf(internal_proname, "__PLPerl_proc_%u_trigger", fn_oid);
1429

1430 1431 1432
	/************************************************************
	 * Lookup the internal proc name in the hashtable
	 ************************************************************/
1433 1434 1435 1436
	hash_entry = hash_search(plperl_proc_hash, internal_proname, 
							 HASH_FIND, NULL);

	if (hash_entry)
1437
	{
1438 1439
		bool		uptodate;

1440
		prodesc = hash_entry->proc_data;
1441

1442
		/************************************************************
1443 1444 1445
		 * If it's present, must check whether it's still up to date.
		 * This is needed because CREATE OR REPLACE FUNCTION can modify the
		 * function's pg_proc entry without changing its OID.
1446
		 ************************************************************/
1447
		uptodate = (prodesc->fn_xmin == HeapTupleHeaderGetXmin(procTup->t_data) &&
B
Bruce Momjian 已提交
1448
				prodesc->fn_cmin == HeapTupleHeaderGetCmin(procTup->t_data));
1449 1450 1451

		if (!uptodate)
		{
1452
			free(prodesc); /* are we leaking memory here? */
1453
			prodesc = NULL;
1454 1455
			hash_search(plperl_proc_hash, internal_proname,
						HASH_REMOVE,NULL);
1456 1457 1458 1459 1460
		}
	}

	/************************************************************
	 * If we haven't found it in the hashtable, we analyze
1461
	 * the function's arguments and return type and store
1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472
	 * the in-/out-functions in the prodesc block and create
	 * a new hashtable entry for it.
	 *
	 * Then we load the procedure into the Perl interpreter.
	 ************************************************************/
	if (prodesc == NULL)
	{
		HeapTuple	langTup;
		HeapTuple	typeTup;
		Form_pg_language langStruct;
		Form_pg_type typeStruct;
1473 1474
		Datum		prosrcdatum;
		bool		isnull;
1475 1476 1477 1478 1479 1480
		char	   *proc_source;

		/************************************************************
		 * Allocate a new procedure description block
		 ************************************************************/
		prodesc = (plperl_proc_desc *) malloc(sizeof(plperl_proc_desc));
1481
		if (prodesc == NULL)
1482 1483 1484
			ereport(ERROR,
					(errcode(ERRCODE_OUT_OF_MEMORY),
					 errmsg("out of memory")));
1485 1486
		MemSet(prodesc, 0, sizeof(plperl_proc_desc));
		prodesc->proname = strdup(internal_proname);
1487 1488
		prodesc->fn_xmin = HeapTupleHeaderGetXmin(procTup->t_data);
		prodesc->fn_cmin = HeapTupleHeaderGetCmin(procTup->t_data);
1489

1490 1491 1492 1493
		/* Remember if function is STABLE/IMMUTABLE */
		prodesc->fn_readonly =
			(procStruct->provolatile != PROVOLATILE_VOLATILE);

1494
		/************************************************************
1495
		 * Lookup the pg_language tuple by Oid
1496
		 ************************************************************/
1497 1498
		langTup = SearchSysCache(LANGOID,
								 ObjectIdGetDatum(procStruct->prolang),
1499
								 0, 0, 0);
1500
		if (!HeapTupleIsValid(langTup))
1501 1502 1503
		{
			free(prodesc->proname);
			free(prodesc);
1504
			elog(ERROR, "cache lookup failed for language %u",
1505
				 procStruct->prolang);
1506
		}
1507 1508 1509
		langStruct = (Form_pg_language) GETSTRUCT(langTup);
		prodesc->lanpltrusted = langStruct->lanpltrusted;
		ReleaseSysCache(langTup);
1510 1511

		/************************************************************
1512 1513
		 * Get the required information for input conversion of the
		 * return value.
1514
		 ************************************************************/
1515 1516 1517
		if (!is_trigger)
		{
			typeTup = SearchSysCache(TYPEOID,
B
Bruce Momjian 已提交
1518
									 ObjectIdGetDatum(procStruct->prorettype),
1519 1520 1521 1522 1523
									 0, 0, 0);
			if (!HeapTupleIsValid(typeTup))
			{
				free(prodesc->proname);
				free(prodesc);
1524
				elog(ERROR, "cache lookup failed for type %u",
1525
					 procStruct->prorettype);
1526 1527 1528
			}
			typeStruct = (Form_pg_type) GETSTRUCT(typeTup);

1529
			/* Disallow pseudotype result, except VOID or RECORD */
1530 1531
			if (typeStruct->typtype == 'p')
			{
1532 1533
				if (procStruct->prorettype == VOIDOID ||
					procStruct->prorettype == RECORDOID)
B
Bruce Momjian 已提交
1534
					 /* okay */ ;
1535
				else if (procStruct->prorettype == TRIGGEROID)
1536 1537 1538
				{
					free(prodesc->proname);
					free(prodesc);
1539 1540
					ereport(ERROR,
							(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1541
							 errmsg("trigger functions can only be called "
1542
									"as triggers")));
1543 1544 1545 1546 1547
				}
				else
				{
					free(prodesc->proname);
					free(prodesc);
1548 1549
					ereport(ERROR,
							(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
B
Bruce Momjian 已提交
1550 1551
							 errmsg("plperl functions cannot return type %s",
									format_type_be(procStruct->prorettype))));
1552 1553 1554
				}
			}

1555 1556 1557 1558
			prodesc->result_oid = procStruct->prorettype;
			prodesc->fn_retisset = procStruct->proretset;
			prodesc->fn_retistuple = (typeStruct->typtype == 'c' ||
									  procStruct->prorettype == RECORDOID);
1559

B
Bruce Momjian 已提交
1560 1561
			prodesc->fn_retisarray =
				(typeStruct->typlen == -1 && typeStruct->typelem);
1562

1563
			perm_fmgr_info(typeStruct->typinput, &(prodesc->result_in_func));
1564
			prodesc->result_typioparam = getTypeIOParam(typeTup);
1565 1566 1567

			ReleaseSysCache(typeTup);
		}
1568 1569

		/************************************************************
1570 1571
		 * Get the required information for output conversion
		 * of all procedure arguments
1572
		 ************************************************************/
1573 1574 1575 1576 1577 1578
		if (!is_trigger)
		{
			prodesc->nargs = procStruct->pronargs;
			for (i = 0; i < prodesc->nargs; i++)
			{
				typeTup = SearchSysCache(TYPEOID,
B
Bruce Momjian 已提交
1579
						 ObjectIdGetDatum(procStruct->proargtypes.values[i]),
1580 1581 1582 1583 1584
										 0, 0, 0);
				if (!HeapTupleIsValid(typeTup))
				{
					free(prodesc->proname);
					free(prodesc);
1585
					elog(ERROR, "cache lookup failed for type %u",
1586
						 procStruct->proargtypes.values[i]);
1587 1588 1589
				}
				typeStruct = (Form_pg_type) GETSTRUCT(typeTup);

1590 1591 1592 1593 1594
				/* Disallow pseudotype argument */
				if (typeStruct->typtype == 'p')
				{
					free(prodesc->proname);
					free(prodesc);
1595 1596
					ereport(ERROR,
							(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
B
Bruce Momjian 已提交
1597 1598
							 errmsg("plperl functions cannot take type %s",
						format_type_be(procStruct->proargtypes.values[i]))));
1599 1600
				}

1601 1602
				if (typeStruct->typtype == 'c')
					prodesc->arg_is_rowtype[i] = true;
1603
				else
1604 1605 1606 1607 1608
				{
					prodesc->arg_is_rowtype[i] = false;
					perm_fmgr_info(typeStruct->typoutput,
								   &(prodesc->arg_out_func[i]));
				}
1609 1610 1611 1612

				ReleaseSysCache(typeTup);
			}
		}
1613

1614 1615 1616 1617 1618
		/************************************************************
		 * create the text of the anonymous subroutine.
		 * we do not use a named subroutine so that we can call directly
		 * through the reference.
		 ************************************************************/
1619 1620 1621 1622
		prosrcdatum = SysCacheGetAttr(PROCOID, procTup,
									  Anum_pg_proc_prosrc, &isnull);
		if (isnull)
			elog(ERROR, "null prosrc");
1623
		proc_source = DatumGetCString(DirectFunctionCall1(textout,
1624
														  prosrcdatum));
1625 1626

		/************************************************************
1627
		 * Create the procedure in the interpreter
1628
		 ************************************************************/
1629 1630 1631

		check_interp(prodesc->lanpltrusted);

1632
		prodesc->reference = plperl_create_sub(proc_source, prodesc->lanpltrusted);
1633 1634 1635

		restore_context(oldcontext);

1636
		pfree(proc_source);
B
Bruce Momjian 已提交
1637
		if (!prodesc->reference)	/* can this happen? */
1638 1639 1640
		{
			free(prodesc->proname);
			free(prodesc);
1641
			elog(ERROR, "could not create internal procedure \"%s\"",
1642
				 internal_proname);
1643 1644
		}

1645 1646 1647
		hash_entry = hash_search(plperl_proc_hash, internal_proname,
								 HASH_ENTER, &found);
		hash_entry->proc_data = prodesc;
1648 1649
	}

1650
	ReleaseSysCache(procTup);
1651

1652 1653
	return prodesc;
}
1654 1655


1656 1657
/* Build a hash from all attributes of a given tuple. */

B
Bruce Momjian 已提交
1658
static SV  *
1659
plperl_hash_from_tuple(HeapTuple tuple, TupleDesc tupdesc)
1660
{
1661
	HV		   *hv;
1662
	int			i;
1663

1664
	hv = newHV();
1665 1666 1667

	for (i = 0; i < tupdesc->natts; i++)
	{
1668 1669 1670 1671 1672 1673 1674
		Datum		attr;
		bool		isnull;
		char	   *attname;
		char	   *outputstr;
		Oid			typoutput;
		bool		typisvarlena;

1675 1676 1677
		if (tupdesc->attrs[i]->attisdropped)
			continue;

1678
		attname = NameStr(tupdesc->attrs[i]->attname);
1679 1680
		attr = heap_getattr(tuple, i + 1, tupdesc, &isnull);

B
Bruce Momjian 已提交
1681 1682
		if (isnull)
		{
1683
			/* Store (attname => undef) and move on. */
1684
			hv_store_string(hv, attname, newSV(0));
1685 1686 1687
			continue;
		}

1688 1689
		/* XXX should have a way to cache these lookups */
		getTypeOutputInfo(tupdesc->attrs[i]->atttypid,
1690
						  &typoutput, &typisvarlena);
1691

1692
		outputstr = OidOutputFunctionCall(typoutput, attr);
1693

1694
		hv_store_string(hv, attname, newSVstring(outputstr));
1695 1696

		pfree(outputstr);
1697
	}
1698

1699
	return newRV_noinc((SV *) hv);
1700
}
1701 1702 1703 1704 1705 1706 1707


HV *
plperl_spi_exec(char *query, int limit)
{
	HV		   *ret_hv;

1708
	/*
B
Bruce Momjian 已提交
1709 1710
	 * Execute the query inside a sub-transaction, so we can cope with errors
	 * sanely
1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722
	 */
	MemoryContext oldcontext = CurrentMemoryContext;
	ResourceOwner oldowner = CurrentResourceOwner;

	BeginInternalSubTransaction(NULL);
	/* Want to run inside function's memory context */
	MemoryContextSwitchTo(oldcontext);

	PG_TRY();
	{
		int			spi_rv;

1723
		spi_rv = SPI_execute(query, current_call_data->prodesc->fn_readonly,
1724 1725 1726 1727 1728 1729 1730 1731
							 limit);
		ret_hv = plperl_spi_execute_fetch_result(SPI_tuptable, SPI_processed,
												 spi_rv);

		/* Commit the inner transaction, return to outer xact context */
		ReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;
B
Bruce Momjian 已提交
1732

1733
		/*
B
Bruce Momjian 已提交
1734 1735
		 * AtEOSubXact_SPI() should not have popped any SPI context, but just
		 * in case it did, make sure we remain connected.
1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753
		 */
		SPI_restore_connection();
	}
	PG_CATCH();
	{
		ErrorData  *edata;

		/* Save error info */
		MemoryContextSwitchTo(oldcontext);
		edata = CopyErrorData();
		FlushErrorState();

		/* Abort the inner transaction */
		RollbackAndReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;

		/*
B
Bruce Momjian 已提交
1754 1755 1756
		 * If AtEOSubXact_SPI() popped any SPI context of the subxact, it will
		 * have left us in a disconnected state.  We need this hack to return
		 * to connected state.
1757 1758 1759 1760 1761 1762 1763 1764 1765 1766
		 */
		SPI_restore_connection();

		/* Punt the error to Perl */
		croak("%s", edata->message);

		/* Can't get here, but keep compiler quiet */
		return NULL;
	}
	PG_END_TRY();
1767 1768 1769 1770

	return ret_hv;
}

1771

1772
static HV  *
1773 1774
plperl_spi_execute_fetch_result(SPITupleTable *tuptable, int processed,
								int status)
1775 1776 1777 1778 1779
{
	HV		   *result;

	result = newHV();

1780 1781 1782 1783
	hv_store_string(result, "status",
					newSVstring(SPI_result_code_string(status)));
	hv_store_string(result, "processed",
					newSViv(processed));
1784

1785
	if (status > 0 && tuptable)
1786
	{
1787
		AV		   *rows;
1788
		SV		   *row;
1789
		int			i;
1790

1791 1792 1793 1794
		rows = newAV();
		for (i = 0; i < processed; i++)
		{
			row = plperl_hash_from_tuple(tuptable->vals[i], tuptable->tupdesc);
1795
			av_push(rows, row);
1796
		}
1797 1798
		hv_store_string(result, "rows",
						newRV_noinc((SV *) rows));
1799 1800 1801 1802 1803 1804
	}

	SPI_freetuptable(tuptable);

	return result;
}
1805 1806


1807 1808
/*
 * Note: plperl_return_next is called both in Postgres and Perl contexts.
1809
 * We report any errors in Postgres fashion (via ereport).	If called in
1810 1811 1812 1813 1814
 * Perl context, it is SPI.xs's responsibility to catch the error and
 * convert to a Perl error.  We assume (perhaps without adequate justification)
 * that we need not abort the current transaction if the Perl code traps the
 * error.
 */
1815
void
1816
plperl_return_next(SV *sv)
1817
{
1818 1819 1820 1821
	plperl_proc_desc *prodesc;
	FunctionCallInfo fcinfo;
	ReturnSetInfo *rsi;
	MemoryContext old_cxt;
B
Bruce Momjian 已提交
1822
	HeapTuple	tuple;
1823 1824 1825 1826

	if (!sv)
		return;

1827 1828 1829 1830
	prodesc = current_call_data->prodesc;
	fcinfo = current_call_data->fcinfo;
	rsi = (ReturnSetInfo *) fcinfo->resultinfo;

1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842
	if (!prodesc->fn_retisset)
		ereport(ERROR,
				(errcode(ERRCODE_SYNTAX_ERROR),
				 errmsg("cannot use return_next in a non-SETOF function")));

	if (prodesc->fn_retistuple &&
		!(SvOK(sv) && SvTYPE(sv) == SVt_RV && SvTYPE(SvRV(sv)) == SVt_PVHV))
		ereport(ERROR,
				(errcode(ERRCODE_DATATYPE_MISMATCH),
				 errmsg("setof-composite-returning Perl function "
						"must call return_next with reference to hash")));

1843 1844
	if (!current_call_data->ret_tdesc)
	{
B
Bruce Momjian 已提交
1845
		TupleDesc	tupdesc;
1846 1847 1848

		Assert(!current_call_data->tuple_store);
		Assert(!current_call_data->attinmeta);
1849

1850
		/*
B
Bruce Momjian 已提交
1851 1852
		 * This is the first call to return_next in the current PL/Perl
		 * function call, so memoize some lookups
1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866
		 */
		if (prodesc->fn_retistuple)
			(void) get_call_result_type(fcinfo, NULL, &tupdesc);
		else
			tupdesc = rsi->expectedDesc;

		/*
		 * Make sure the tuple_store and ret_tdesc are sufficiently
		 * long-lived.
		 */
		old_cxt = MemoryContextSwitchTo(rsi->econtext->ecxt_per_query_memory);

		current_call_data->ret_tdesc = CreateTupleDescCopy(tupdesc);
		current_call_data->tuple_store =
1867
			tuplestore_begin_heap(true, false, work_mem);
1868 1869 1870 1871 1872
		if (prodesc->fn_retistuple)
		{
			current_call_data->attinmeta =
				TupleDescGetAttInMetadata(current_call_data->ret_tdesc);
		}
1873

1874
		MemoryContextSwitchTo(old_cxt);
B
Bruce Momjian 已提交
1875
	}
1876 1877 1878

	/*
	 * Producing the tuple we want to return requires making plenty of
B
Bruce Momjian 已提交
1879 1880 1881
	 * palloc() allocations that are not cleaned up. Since this function can
	 * be called many times before the current memory context is reset, we
	 * need to do those allocations in a temporary context.
1882 1883
	 */
	if (!current_call_data->tmp_cxt)
1884
	{
1885 1886 1887 1888 1889 1890 1891 1892 1893
		current_call_data->tmp_cxt =
			AllocSetContextCreate(rsi->econtext->ecxt_per_tuple_memory,
								  "PL/Perl return_next temporary cxt",
								  ALLOCSET_DEFAULT_MINSIZE,
								  ALLOCSET_DEFAULT_INITSIZE,
								  ALLOCSET_DEFAULT_MAXSIZE);
	}

	old_cxt = MemoryContextSwitchTo(current_call_data->tmp_cxt);
1894

1895 1896 1897 1898
	if (prodesc->fn_retistuple)
	{
		tuple = plperl_build_tuple_result((HV *) SvRV(sv),
										  current_call_data->attinmeta);
1899 1900 1901
	}
	else
	{
1902 1903
		Datum		ret;
		bool		isNull;
1904 1905 1906

		if (SvOK(sv) && SvTYPE(sv) != SVt_NULL)
		{
1907
			char	   *val = SvPV(sv, PL_na);
B
Bruce Momjian 已提交
1908

1909 1910
			ret = InputFunctionCall(&prodesc->result_in_func, val,
									prodesc->result_typioparam, -1);
1911 1912
			isNull = false;
		}
1913 1914 1915 1916 1917 1918
		else
		{
			ret = InputFunctionCall(&prodesc->result_in_func, NULL,
									prodesc->result_typioparam, -1);
			isNull = true;
		}
1919

1920
		tuple = heap_form_tuple(current_call_data->ret_tdesc, &ret, &isNull);
1921 1922
	}

1923 1924 1925 1926
	/* Make sure to store the tuple in a long-lived memory context */
	MemoryContextSwitchTo(rsi->econtext->ecxt_per_query_memory);
	tuplestore_puttuple(current_call_data->tuple_store, tuple);
	MemoryContextSwitchTo(old_cxt);
1927

1928
	MemoryContextReset(current_call_data->tmp_cxt);
1929
}
1930 1931 1932 1933 1934


SV *
plperl_spi_query(char *query)
{
B
Bruce Momjian 已提交
1935
	SV		   *cursor;
1936

1937 1938 1939 1940
	/*
	 * Execute the query inside a sub-transaction, so we can cope with errors
	 * sanely
	 */
1941 1942 1943 1944
	MemoryContext oldcontext = CurrentMemoryContext;
	ResourceOwner oldowner = CurrentResourceOwner;

	BeginInternalSubTransaction(NULL);
1945
	/* Want to run inside function's memory context */
1946 1947 1948 1949
	MemoryContextSwitchTo(oldcontext);

	PG_TRY();
	{
B
Bruce Momjian 已提交
1950
		void	   *plan;
A
 
Andrew Dunstan 已提交
1951
		Portal		portal;
1952

1953
		/* Create a cursor for the query */
1954
		plan = SPI_prepare(query, 0, NULL);
B
Bruce Momjian 已提交
1955
		if (plan == NULL)
A
 
Andrew Dunstan 已提交
1956
			elog(ERROR, "SPI_prepare() failed:%s",
B
Bruce Momjian 已提交
1957
				 SPI_result_code_string(SPI_result));
A
 
Andrew Dunstan 已提交
1958 1959

		portal = SPI_cursor_open(NULL, plan, NULL, NULL, false);
B
Bruce Momjian 已提交
1960 1961
		SPI_freeplan(plan);
		if (portal == NULL)
A
 
Andrew Dunstan 已提交
1962
			elog(ERROR, "SPI_cursor_open() failed:%s",
B
Bruce Momjian 已提交
1963
				 SPI_result_code_string(SPI_result));
1964
		cursor = newSVstring(portal->name);
1965

1966
		/* Commit the inner transaction, return to outer xact context */
1967 1968 1969
		ReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;
1970 1971 1972 1973 1974

		/*
		 * AtEOSubXact_SPI() should not have popped any SPI context, but just
		 * in case it did, make sure we remain connected.
		 */
1975 1976 1977 1978 1979 1980
		SPI_restore_connection();
	}
	PG_CATCH();
	{
		ErrorData  *edata;

1981
		/* Save error info */
1982 1983 1984 1985
		MemoryContextSwitchTo(oldcontext);
		edata = CopyErrorData();
		FlushErrorState();

1986
		/* Abort the inner transaction */
1987 1988 1989 1990
		RollbackAndReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;

1991 1992 1993 1994 1995
		/*
		 * If AtEOSubXact_SPI() popped any SPI context of the subxact, it will
		 * have left us in a disconnected state.  We need this hack to return
		 * to connected state.
		 */
1996
		SPI_restore_connection();
1997 1998

		/* Punt the error to Perl */
1999
		croak("%s", edata->message);
2000 2001

		/* Can't get here, but keep compiler quiet */
2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012
		return NULL;
	}
	PG_END_TRY();

	return cursor;
}


SV *
plperl_spi_fetchrow(char *cursor)
{
2013 2014 2015 2016 2017 2018 2019 2020
	SV		   *row;

	/*
	 * Execute the FETCH inside a sub-transaction, so we can cope with errors
	 * sanely
	 */
	MemoryContext oldcontext = CurrentMemoryContext;
	ResourceOwner oldowner = CurrentResourceOwner;
2021

2022 2023 2024
	BeginInternalSubTransaction(NULL);
	/* Want to run inside function's memory context */
	MemoryContextSwitchTo(oldcontext);
2025

2026
	PG_TRY();
B
Bruce Momjian 已提交
2027
	{
2028 2029 2030
		Portal		p = SPI_cursor_find(cursor);

		if (!p)
A
 
Andrew Dunstan 已提交
2031 2032 2033
		{
			row = &PL_sv_undef;
		}
2034 2035 2036 2037 2038 2039
		else
		{
			SPI_cursor_fetch(p, true, 1);
			if (SPI_processed == 0)
			{
				SPI_cursor_close(p);
A
 
Andrew Dunstan 已提交
2040
				row = &PL_sv_undef;
2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059
			}
			else
			{
				row = plperl_hash_from_tuple(SPI_tuptable->vals[0],
											 SPI_tuptable->tupdesc);
			}
			SPI_freetuptable(SPI_tuptable);
		}

		/* Commit the inner transaction, return to outer xact context */
		ReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;

		/*
		 * AtEOSubXact_SPI() should not have popped any SPI context, but just
		 * in case it did, make sure we remain connected.
		 */
		SPI_restore_connection();
2060
	}
2061 2062 2063
	PG_CATCH();
	{
		ErrorData  *edata;
2064

2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088
		/* Save error info */
		MemoryContextSwitchTo(oldcontext);
		edata = CopyErrorData();
		FlushErrorState();

		/* Abort the inner transaction */
		RollbackAndReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;

		/*
		 * If AtEOSubXact_SPI() popped any SPI context of the subxact, it will
		 * have left us in a disconnected state.  We need this hack to return
		 * to connected state.
		 */
		SPI_restore_connection();

		/* Punt the error to Perl */
		croak("%s", edata->message);

		/* Can't get here, but keep compiler quiet */
		return NULL;
	}
	PG_END_TRY();
2089 2090 2091

	return row;
}
A
 
Andrew Dunstan 已提交
2092 2093 2094 2095

void
plperl_spi_cursor_close(char *cursor)
{
B
Bruce Momjian 已提交
2096 2097
	Portal		p = SPI_cursor_find(cursor);

A
 
Andrew Dunstan 已提交
2098 2099 2100 2101 2102
	if (p)
		SPI_cursor_close(p);
}

SV *
B
Bruce Momjian 已提交
2103
plperl_spi_prepare(char *query, int argc, SV **argv)
A
 
Andrew Dunstan 已提交
2104 2105
{
	plperl_query_desc *qdesc;
2106 2107
	plperl_query_entry *hash_entry;
	bool        found;
A
 
Andrew Dunstan 已提交
2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121
	void	   *plan;
	int			i;

	MemoryContext oldcontext = CurrentMemoryContext;
	ResourceOwner oldowner = CurrentResourceOwner;

	BeginInternalSubTransaction(NULL);
	MemoryContextSwitchTo(oldcontext);

	/************************************************************
	 * Allocate the new querydesc structure
	 ************************************************************/
	qdesc = (plperl_query_desc *) malloc(sizeof(plperl_query_desc));
	MemSet(qdesc, 0, sizeof(plperl_query_desc));
B
Bruce Momjian 已提交
2122 2123 2124 2125 2126
	snprintf(qdesc->qname, sizeof(qdesc->qname), "%lx", (long) qdesc);
	qdesc->nargs = argc;
	qdesc->argtypes = (Oid *) malloc(argc * sizeof(Oid));
	qdesc->arginfuncs = (FmgrInfo *) malloc(argc * sizeof(FmgrInfo));
	qdesc->argtypioparams = (Oid *) malloc(argc * sizeof(Oid));
A
 
Andrew Dunstan 已提交
2127 2128 2129 2130

	PG_TRY();
	{
		/************************************************************
2131 2132 2133
		 * Resolve argument type names and then look them up by oid 
         * in the system cache, and remember the required information 
         * for input conversion.
A
 
Andrew Dunstan 已提交
2134 2135 2136
		 ************************************************************/
		for (i = 0; i < argc; i++)
		{
2137 2138 2139 2140 2141 2142 2143 2144
			Oid         typId, typInput, typIOParam;
            int32       typmod;

			parseTypeString(SvPV(argv[i], PL_na), &typId, &typmod);

			getTypeInputInfo(typId, &typInput, &typIOParam);

			qdesc->argtypes[i] = typId;
2145
			perm_fmgr_info(typInput, &(qdesc->arginfuncs[i]));
2146
			qdesc->argtypioparams[i] = typIOParam;
A
 
Andrew Dunstan 已提交
2147 2148 2149 2150 2151 2152 2153 2154 2155
		}

		/************************************************************
		 * Prepare the plan and check for errors
		 ************************************************************/
		plan = SPI_prepare(query, argc, qdesc->argtypes);

		if (plan == NULL)
			elog(ERROR, "SPI_prepare() failed:%s",
B
Bruce Momjian 已提交
2156
				 SPI_result_code_string(SPI_result));
A
 
Andrew Dunstan 已提交
2157 2158 2159 2160 2161 2162 2163

		/************************************************************
		 * Save the plan into permanent memory (right now it's in the
		 * SPI procCxt, which will go away at function end).
		 ************************************************************/
		qdesc->plan = SPI_saveplan(plan);
		if (qdesc->plan == NULL)
B
Bruce Momjian 已提交
2164 2165
			elog(ERROR, "SPI_saveplan() failed: %s",
				 SPI_result_code_string(SPI_result));
A
 
Andrew Dunstan 已提交
2166 2167 2168 2169 2170 2171 2172 2173

		/* Release the procCxt copy to avoid within-function memory leak */
		SPI_freeplan(plan);

		/* Commit the inner transaction, return to outer xact context */
		ReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;
B
Bruce Momjian 已提交
2174

A
 
Andrew Dunstan 已提交
2175
		/*
B
Bruce Momjian 已提交
2176 2177
		 * AtEOSubXact_SPI() should not have popped any SPI context, but just
		 * in case it did, make sure we remain connected.
A
 
Andrew Dunstan 已提交
2178 2179 2180 2181 2182 2183
		 */
		SPI_restore_connection();
	}
	PG_CATCH();
	{
		ErrorData  *edata;
B
Bruce Momjian 已提交
2184 2185 2186 2187

		free(qdesc->argtypes);
		free(qdesc->arginfuncs);
		free(qdesc->argtypioparams);
A
 
Andrew Dunstan 已提交
2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200
		free(qdesc);

		/* Save error info */
		MemoryContextSwitchTo(oldcontext);
		edata = CopyErrorData();
		FlushErrorState();

		/* Abort the inner transaction */
		RollbackAndReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;

		/*
B
Bruce Momjian 已提交
2201 2202 2203
		 * If AtEOSubXact_SPI() popped any SPI context of the subxact, it will
		 * have left us in a disconnected state.  We need this hack to return
		 * to connected state.
A
 
Andrew Dunstan 已提交
2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218
		 */
		SPI_restore_connection();

		/* Punt the error to Perl */
		croak("%s", edata->message);

		/* Can't get here, but keep compiler quiet */
		return NULL;
	}
	PG_END_TRY();

	/************************************************************
	 * Insert a hashtable entry for the plan and return
	 * the key to the caller.
	 ************************************************************/
2219 2220 2221 2222

	hash_entry = hash_search(plperl_query_hash, qdesc->qname,
							 HASH_ENTER,&found);
	hash_entry->query_data = qdesc;
A
 
Andrew Dunstan 已提交
2223

2224
	return newSVstring(qdesc->qname);
B
Bruce Momjian 已提交
2225
}
A
 
Andrew Dunstan 已提交
2226 2227

HV *
B
Bruce Momjian 已提交
2228
plperl_spi_exec_prepared(char *query, HV *attr, int argc, SV **argv)
A
 
Andrew Dunstan 已提交
2229 2230
{
	HV		   *ret_hv;
B
Bruce Momjian 已提交
2231 2232 2233 2234 2235
	SV		  **sv;
	int			i,
				limit,
				spi_rv;
	char	   *nulls;
A
 
Andrew Dunstan 已提交
2236 2237
	Datum	   *argvalues;
	plperl_query_desc *qdesc;
2238
	plperl_query_entry *hash_entry;
A
 
Andrew Dunstan 已提交
2239 2240

	/*
B
Bruce Momjian 已提交
2241 2242
	 * Execute the query inside a sub-transaction, so we can cope with errors
	 * sanely
A
 
Andrew Dunstan 已提交
2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255
	 */
	MemoryContext oldcontext = CurrentMemoryContext;
	ResourceOwner oldowner = CurrentResourceOwner;

	BeginInternalSubTransaction(NULL);
	/* Want to run inside function's memory context */
	MemoryContextSwitchTo(oldcontext);

	PG_TRY();
	{
		/************************************************************
		 * Fetch the saved plan descriptor, see if it's o.k.
		 ************************************************************/
2256 2257 2258 2259

		hash_entry = hash_search(plperl_query_hash, query,
										 HASH_FIND,NULL);
		if (hash_entry == NULL)
A
 
Andrew Dunstan 已提交
2260 2261
			elog(ERROR, "spi_exec_prepared: Invalid prepared query passed");

2262 2263
		qdesc = hash_entry->query_data;

B
Bruce Momjian 已提交
2264
		if (qdesc == NULL)
A
 
Andrew Dunstan 已提交
2265 2266
			elog(ERROR, "spi_exec_prepared: panic - plperl_query_hash value vanished");

B
Bruce Momjian 已提交
2267 2268 2269 2270
		if (qdesc->nargs != argc)
			elog(ERROR, "spi_exec_prepared: expected %d argument(s), %d passed",
				 qdesc->nargs, argc);

A
 
Andrew Dunstan 已提交
2271 2272 2273 2274
		/************************************************************
		 * Parse eventual attributes
		 ************************************************************/
		limit = 0;
B
Bruce Momjian 已提交
2275
		if (attr != NULL)
A
 
Andrew Dunstan 已提交
2276
		{
2277
			sv = hv_fetch_string(attr, "limit");
B
Bruce Momjian 已提交
2278 2279
			if (*sv && SvIOK(*sv))
				limit = SvIV(*sv);
A
 
Andrew Dunstan 已提交
2280 2281 2282 2283
		}
		/************************************************************
		 * Set up arguments
		 ************************************************************/
B
Bruce Momjian 已提交
2284
		if (argc > 0)
A
 
Andrew Dunstan 已提交
2285
		{
2286
			nulls = (char *) palloc(argc);
A
 
Andrew Dunstan 已提交
2287
			argvalues = (Datum *) palloc(argc * sizeof(Datum));
B
Bruce Momjian 已提交
2288 2289
		}
		else
A
 
Andrew Dunstan 已提交
2290 2291 2292 2293 2294
		{
			nulls = NULL;
			argvalues = NULL;
		}

B
Bruce Momjian 已提交
2295
		for (i = 0; i < argc; i++)
A
 
Andrew Dunstan 已提交
2296
		{
B
Bruce Momjian 已提交
2297
			if (SvTYPE(argv[i]) != SVt_NULL)
A
 
Andrew Dunstan 已提交
2298
			{
2299 2300 2301 2302
				argvalues[i] = InputFunctionCall(&qdesc->arginfuncs[i],
												 SvPV(argv[i], PL_na),
												 qdesc->argtypioparams[i],
												 -1);
A
 
Andrew Dunstan 已提交
2303
				nulls[i] = ' ';
B
Bruce Momjian 已提交
2304 2305
			}
			else
A
 
Andrew Dunstan 已提交
2306
			{
2307 2308 2309 2310
				argvalues[i] = InputFunctionCall(&qdesc->arginfuncs[i],
												 NULL,
												 qdesc->argtypioparams[i],
												 -1);
A
 
Andrew Dunstan 已提交
2311 2312 2313 2314 2315 2316 2317
				nulls[i] = 'n';
			}
		}

		/************************************************************
		 * go
		 ************************************************************/
B
Bruce Momjian 已提交
2318
		spi_rv = SPI_execute_plan(qdesc->plan, argvalues, nulls,
A
 
Andrew Dunstan 已提交
2319 2320 2321
							 current_call_data->prodesc->fn_readonly, limit);
		ret_hv = plperl_spi_execute_fetch_result(SPI_tuptable, SPI_processed,
												 spi_rv);
B
Bruce Momjian 已提交
2322
		if (argc > 0)
A
 
Andrew Dunstan 已提交
2323
		{
B
Bruce Momjian 已提交
2324 2325
			pfree(argvalues);
			pfree(nulls);
A
 
Andrew Dunstan 已提交
2326 2327 2328 2329 2330 2331
		}

		/* Commit the inner transaction, return to outer xact context */
		ReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;
B
Bruce Momjian 已提交
2332

A
 
Andrew Dunstan 已提交
2333
		/*
B
Bruce Momjian 已提交
2334 2335
		 * AtEOSubXact_SPI() should not have popped any SPI context, but just
		 * in case it did, make sure we remain connected.
A
 
Andrew Dunstan 已提交
2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353
		 */
		SPI_restore_connection();
	}
	PG_CATCH();
	{
		ErrorData  *edata;

		/* Save error info */
		MemoryContextSwitchTo(oldcontext);
		edata = CopyErrorData();
		FlushErrorState();

		/* Abort the inner transaction */
		RollbackAndReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;

		/*
B
Bruce Momjian 已提交
2354 2355 2356
		 * If AtEOSubXact_SPI() popped any SPI context of the subxact, it will
		 * have left us in a disconnected state.  We need this hack to return
		 * to connected state.
A
 
Andrew Dunstan 已提交
2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371
		 */
		SPI_restore_connection();

		/* Punt the error to Perl */
		croak("%s", edata->message);

		/* Can't get here, but keep compiler quiet */
		return NULL;
	}
	PG_END_TRY();

	return ret_hv;
}

SV *
B
Bruce Momjian 已提交
2372
plperl_spi_query_prepared(char *query, int argc, SV **argv)
A
 
Andrew Dunstan 已提交
2373
{
B
Bruce Momjian 已提交
2374 2375
	int			i;
	char	   *nulls;
A
 
Andrew Dunstan 已提交
2376 2377
	Datum	   *argvalues;
	plperl_query_desc *qdesc;
2378
	plperl_query_entry *hash_entry;
B
Bruce Momjian 已提交
2379 2380
	SV		   *cursor;
	Portal		portal = NULL;
A
 
Andrew Dunstan 已提交
2381 2382

	/*
B
Bruce Momjian 已提交
2383 2384
	 * Execute the query inside a sub-transaction, so we can cope with errors
	 * sanely
A
 
Andrew Dunstan 已提交
2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397
	 */
	MemoryContext oldcontext = CurrentMemoryContext;
	ResourceOwner oldowner = CurrentResourceOwner;

	BeginInternalSubTransaction(NULL);
	/* Want to run inside function's memory context */
	MemoryContextSwitchTo(oldcontext);

	PG_TRY();
	{
		/************************************************************
		 * Fetch the saved plan descriptor, see if it's o.k.
		 ************************************************************/
2398 2399 2400 2401 2402 2403
		hash_entry = hash_search(plperl_query_hash, query,
										 HASH_FIND,NULL);
		if (hash_entry == NULL)
			elog(ERROR, "spi_exec_prepared: Invalid prepared query passed");

		qdesc = hash_entry->query_data;
A
 
Andrew Dunstan 已提交
2404

B
Bruce Momjian 已提交
2405
		if (qdesc == NULL)
A
 
Andrew Dunstan 已提交
2406 2407
			elog(ERROR, "spi_query_prepared: panic - plperl_query_hash value vanished");

B
Bruce Momjian 已提交
2408 2409 2410 2411
		if (qdesc->nargs != argc)
			elog(ERROR, "spi_query_prepared: expected %d argument(s), %d passed",
				 qdesc->nargs, argc);

A
 
Andrew Dunstan 已提交
2412 2413 2414
		/************************************************************
		 * Set up arguments
		 ************************************************************/
B
Bruce Momjian 已提交
2415
		if (argc > 0)
A
 
Andrew Dunstan 已提交
2416
		{
2417
			nulls = (char *) palloc(argc);
A
 
Andrew Dunstan 已提交
2418
			argvalues = (Datum *) palloc(argc * sizeof(Datum));
B
Bruce Momjian 已提交
2419 2420
		}
		else
A
 
Andrew Dunstan 已提交
2421 2422 2423 2424 2425
		{
			nulls = NULL;
			argvalues = NULL;
		}

B
Bruce Momjian 已提交
2426
		for (i = 0; i < argc; i++)
A
 
Andrew Dunstan 已提交
2427
		{
B
Bruce Momjian 已提交
2428
			if (SvTYPE(argv[i]) != SVt_NULL)
A
 
Andrew Dunstan 已提交
2429
			{
2430 2431 2432 2433
				argvalues[i] = InputFunctionCall(&qdesc->arginfuncs[i],
												 SvPV(argv[i], PL_na),
												 qdesc->argtypioparams[i],
												 -1);
A
 
Andrew Dunstan 已提交
2434
				nulls[i] = ' ';
B
Bruce Momjian 已提交
2435 2436
			}
			else
A
 
Andrew Dunstan 已提交
2437
			{
2438 2439 2440 2441
				argvalues[i] = InputFunctionCall(&qdesc->arginfuncs[i],
												 NULL,
												 qdesc->argtypioparams[i],
												 -1);
A
 
Andrew Dunstan 已提交
2442 2443 2444 2445 2446 2447 2448
				nulls[i] = 'n';
			}
		}

		/************************************************************
		 * go
		 ************************************************************/
B
Bruce Momjian 已提交
2449 2450 2451
		portal = SPI_cursor_open(NULL, qdesc->plan, argvalues, nulls,
								 current_call_data->prodesc->fn_readonly);
		if (argc > 0)
A
 
Andrew Dunstan 已提交
2452
		{
B
Bruce Momjian 已提交
2453 2454
			pfree(argvalues);
			pfree(nulls);
A
 
Andrew Dunstan 已提交
2455
		}
B
Bruce Momjian 已提交
2456
		if (portal == NULL)
A
 
Andrew Dunstan 已提交
2457
			elog(ERROR, "SPI_cursor_open() failed:%s",
B
Bruce Momjian 已提交
2458
				 SPI_result_code_string(SPI_result));
A
 
Andrew Dunstan 已提交
2459

2460
		cursor = newSVstring(portal->name);
A
 
Andrew Dunstan 已提交
2461 2462 2463 2464 2465

		/* Commit the inner transaction, return to outer xact context */
		ReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;
B
Bruce Momjian 已提交
2466

A
 
Andrew Dunstan 已提交
2467
		/*
B
Bruce Momjian 已提交
2468 2469
		 * AtEOSubXact_SPI() should not have popped any SPI context, but just
		 * in case it did, make sure we remain connected.
A
 
Andrew Dunstan 已提交
2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487
		 */
		SPI_restore_connection();
	}
	PG_CATCH();
	{
		ErrorData  *edata;

		/* Save error info */
		MemoryContextSwitchTo(oldcontext);
		edata = CopyErrorData();
		FlushErrorState();

		/* Abort the inner transaction */
		RollbackAndReleaseCurrentSubTransaction();
		MemoryContextSwitchTo(oldcontext);
		CurrentResourceOwner = oldowner;

		/*
B
Bruce Momjian 已提交
2488 2489 2490
		 * If AtEOSubXact_SPI() popped any SPI context of the subxact, it will
		 * have left us in a disconnected state.  We need this hack to return
		 * to connected state.
A
 
Andrew Dunstan 已提交
2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507
		 */
		SPI_restore_connection();

		/* Punt the error to Perl */
		croak("%s", edata->message);

		/* Can't get here, but keep compiler quiet */
		return NULL;
	}
	PG_END_TRY();

	return cursor;
}

void
plperl_spi_freeplan(char *query)
{
B
Bruce Momjian 已提交
2508
	void	   *plan;
A
 
Andrew Dunstan 已提交
2509
	plperl_query_desc *qdesc;
2510
	plperl_query_entry *hash_entry;
A
 
Andrew Dunstan 已提交
2511

2512 2513 2514 2515 2516 2517
	hash_entry = hash_search(plperl_query_hash, query,
										 HASH_FIND,NULL);
	if (hash_entry == NULL)
		elog(ERROR, "spi_exec_prepared: Invalid prepared query passed");

	qdesc = hash_entry->query_data;
A
 
Andrew Dunstan 已提交
2518

B
Bruce Momjian 已提交
2519
	if (qdesc == NULL)
A
 
Andrew Dunstan 已提交
2520 2521 2522
		elog(ERROR, "spi_exec_freeplan: panic - plperl_query_hash value vanished");

	/*
B
Bruce Momjian 已提交
2523 2524 2525
	 * free all memory before SPI_freeplan, so if it dies, nothing will be
	 * left over
	 */
2526 2527 2528
	hash_search(plperl_query_hash, query, 
				HASH_REMOVE,NULL);

B
Bruce Momjian 已提交
2529 2530 2531 2532
	plan = qdesc->plan;
	free(qdesc->argtypes);
	free(qdesc->arginfuncs);
	free(qdesc->argtypioparams);
A
 
Andrew Dunstan 已提交
2533 2534
	free(qdesc);

B
Bruce Momjian 已提交
2535
	SPI_freeplan(plan);
A
 
Andrew Dunstan 已提交
2536
}
2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592

/*
 * Create a new SV from a string assumed to be in the current database's
 * encoding.
 */
static SV *
newSVstring(const char *str)
{
	SV		   *sv;

	sv = newSVpv(str, 0);
#if PERL_BCDVERSION >= 0x5006000L
	if (GetDatabaseEncoding() == PG_UTF8)
		SvUTF8_on(sv);
#endif
	return sv;
}

/*
 * Store an SV into a hash table under a key that is a string assumed to be
 * in the current database's encoding.
 */
static SV **
hv_store_string(HV *hv, const char *key, SV *val)
{
	int32	klen = strlen(key);

	/*
	 * This seems nowhere documented, but under Perl 5.8.0 and up,
	 * hv_store() recognizes a negative klen parameter as meaning
	 * a UTF-8 encoded key.  It does not appear that hashes track
	 * UTF-8-ness of keys at all in Perl 5.6.
	 */
#if PERL_BCDVERSION >= 0x5008000L
	if (GetDatabaseEncoding() == PG_UTF8)
		klen = -klen;
#endif
	return hv_store(hv, key, klen, val, 0);
}

/*
 * Fetch an SV from a hash table under a key that is a string assumed to be
 * in the current database's encoding.
 */
static SV **
hv_fetch_string(HV *hv, const char *key)
{
	int32	klen = strlen(key);

	/* See notes in hv_store_string */
#if PERL_BCDVERSION >= 0x5008000L
	if (GetDatabaseEncoding() == PG_UTF8)
		klen = -klen;
#endif
	return hv_fetch(hv, key, klen, 0);
}