postgres.c 92.1 KB
Newer Older
1 2
/*-------------------------------------------------------------------------
 *
3
 * postgres.c
4
 *	  POSTGRES C Backend Interface
5
 *
P
 
PostgreSQL Daemon 已提交
6
 * Portions Copyright (c) 1996-2005, PostgreSQL Global Development Group
B
Add:  
Bruce Momjian 已提交
7
 * Portions Copyright (c) 1994, Regents of the University of California
8 9 10
 *
 *
 * IDENTIFICATION
11
 *	  $PostgreSQL: pgsql/src/backend/tcop/postgres.c,v 1.468 2005/11/03 17:11:38 alvherre Exp $
12 13
 *
 * NOTES
14 15
 *	  this is the "main" module of the postgres backend and
 *	  hence the main module of the "traffic cop".
16 17 18
 *
 *-------------------------------------------------------------------------
 */
B
Bruce Momjian 已提交
19

20 21
#include "postgres.h"

B
Bruce Momjian 已提交
22
#include <unistd.h>
23
#include <signal.h>
B
Bruce Momjian 已提交
24
#include <fcntl.h>
25
#include <sys/socket.h>
26
#if HAVE_SYS_SELECT_H
27
#include <sys/select.h>
28
#endif
29
#ifdef HAVE_GETOPT_H
B
Bruce Momjian 已提交
30
#include <getopt.h>
31
#endif
32

33
#include "access/printtup.h"
34
#include "access/xlog.h"
35
#include "catalog/pg_type.h"
36
#include "commands/async.h"
37
#include "commands/prepare.h"
38
#include "commands/trigger.h"
39
#include "libpq/libpq.h"
40
#include "libpq/pqformat.h"
41
#include "libpq/pqsignal.h"
B
Bruce Momjian 已提交
42
#include "miscadmin.h"
43 44
#include "nodes/print.h"
#include "optimizer/cost.h"
45
#include "optimizer/planner.h"
46
#include "parser/analyze.h"
47
#include "parser/parser.h"
B
Bruce Momjian 已提交
48
#include "rewrite/rewriteHandler.h"
49
#include "storage/freespace.h"
50
#include "storage/ipc.h"
51
#include "storage/pg_shmem.h"
52
#include "storage/proc.h"
53
#include "storage/sinval.h"
54 55
#include "tcop/fastpath.h"
#include "tcop/pquery.h"
B
Bruce Momjian 已提交
56
#include "tcop/tcopprot.h"
57
#include "tcop/utility.h"
58
#include "utils/flatfiles.h"
59
#include "utils/guc.h"
60
#include "utils/lsyscache.h"
61
#include "utils/memutils.h"
M
 
Marc G. Fournier 已提交
62
#include "utils/ps_status.h"
B
Bruce Momjian 已提交
63
#include "mb/pg_wchar.h"
64

65
#include "pgstat.h"
M
 
Marc G. Fournier 已提交
66

67 68 69
extern int	optind;
extern char *optarg;

70
/* ----------------
71
 *		global variables
72 73
 * ----------------
 */
B
Bruce Momjian 已提交
74
const char *debug_query_string; /* for pgmonitor and log_min_error_statement */
75

76
/* Note: whereToSendOutput is initialized for the bootstrap/standalone case */
77
CommandDest whereToSendOutput = DestDebug;
78

79
/* flag for logging end of session */
B
Bruce Momjian 已提交
80
bool		Log_disconnections = false;
81

82 83
LogStmtLevel log_statement = LOGSTMT_NONE;

84 85 86 87
/* GUC variable for maximum stack depth (measured in kilobytes) */
int			max_stack_depth = 2048;


88 89 90 91
/* ----------------
 *		private variables
 * ----------------
 */
92

93
/* max_stack_depth converted to bytes for speed of checking */
B
Bruce Momjian 已提交
94
static int	max_stack_depth_bytes = 2048 * 1024;
95 96

/* stack base pointer (initialized by PostgresMain) */
97
/* Do not make static so PL/Java can modifiy it */
B
Bruce Momjian 已提交
98
char	   *stack_base_ptr = NULL;
99 100


101 102 103 104 105
/*
 * Flag to mark SIGHUP. Whenever the main loop comes around it
 * will reread the configuration file. (Better than doing the
 * reading in the signal handler, ey?)
 */
106
static volatile sig_atomic_t got_SIGHUP = false;
107

108 109 110 111 112 113
/*
 * Flag to keep track of whether we have started a transaction.
 * For extended query protocol this has to be remembered across messages.
 */
static bool xact_started = false;

114 115 116 117 118 119 120
/*
 * Flag to indicate that we are doing the outer loop's read-from-client,
 * as opposed to any random read from client that might happen within
 * commands like COPY FROM STDIN.
 */
static bool DoingCommandRead = false;

121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141
/*
 * Flags to implement skip-till-Sync-after-error behavior for messages of
 * the extended query protocol.
 */
static bool doing_extended_query_message = false;
static bool ignore_till_sync = false;

/*
 * If an unnamed prepared statement exists, it's stored here.
 * We keep it separate from the hashtable kept by commands/prepare.c
 * in order to reduce overhead for short-lived queries.
 */
static MemoryContext unnamed_stmt_context = NULL;
static PreparedStatement *unnamed_stmt_pstmt = NULL;


static bool EchoQuery = false;	/* default don't echo */

/*
 * people who want to use EOF should #define DONTUSENEWLINE in
 * tcop/tcopdebug.h
142 143
 */
#ifndef TCOP_DONTUSENEWLINE
B
Bruce Momjian 已提交
144
static int	UseNewLine = 1;		/* Use newlines query delimiters (the default) */
145
#else
146
static int	UseNewLine = 0;		/* Use EOF as query delimiters */
147
#endif   /* TCOP_DONTUSENEWLINE */
148 149 150


/* ----------------------------------------------------------------
151
 *		decls for routines only used in this file
152 153
 * ----------------------------------------------------------------
 */
154 155 156
static int	InteractiveBackend(StringInfo inBuf);
static int	SocketBackend(StringInfo inBuf);
static int	ReadCommand(StringInfo inBuf);
157 158
static bool log_after_parse(List *raw_parsetree_list,
				const char *query_string, char **prepare_string);
159
static List *pg_rewrite_queries(List *querytree_list);
160
static void start_xact_command(void);
161
static void finish_xact_command(void);
162
static void SigHupHandler(SIGNAL_ARGS);
163
static void log_disconnections(int code, Datum arg);
164 165 166


/* ----------------------------------------------------------------
167
 *		routines to obtain user input
168 169 170 171
 * ----------------------------------------------------------------
 */

/* ----------------
172
 *	InteractiveBackend() is called for user interactive connections
173 174 175
 *
 *	the string entered by the user is placed in its parameter inBuf,
 *	and we act like a Q message was received.
176
 *
177
 *	EOF is returned if end-of-file input is seen; time to shut down.
178 179 180
 * ----------------
 */

181
static int
182
InteractiveBackend(StringInfo inBuf)
183
{
184 185 186
	int			c;				/* character read from getc() */
	bool		end = false;	/* end-of-input flag */
	bool		backslashSeen = false;	/* have we seen a \ ? */
187

188 189
	/*
	 * display a prompt and obtain input from the user
190
	 */
191
	printf("backend> ");
192
	fflush(stdout);
193

194 195 196
	/* Reset inBuf to empty */
	inBuf->len = 0;
	inBuf->data[0] = '\0';
197
	inBuf->cursor = 0;
198

199 200 201 202
	for (;;)
	{
		if (UseNewLine)
		{
203
			/*
B
Bruce Momjian 已提交
204 205
			 * if we are using \n as a delimiter, then read characters until
			 * the \n.
206 207 208 209 210 211 212
			 */
			while ((c = getc(stdin)) != EOF)
			{
				if (c == '\n')
				{
					if (backslashSeen)
					{
213 214 215
						/* discard backslash from inBuf */
						inBuf->data[--inBuf->len] = '\0';
						backslashSeen = false;
216 217 218 219 220
						continue;
					}
					else
					{
						/* keep the newline character */
221
						appendStringInfoChar(inBuf, '\n');
222 223 224 225 226 227 228 229
						break;
					}
				}
				else if (c == '\\')
					backslashSeen = true;
				else
					backslashSeen = false;

230
				appendStringInfoChar(inBuf, (char) c);
231 232 233 234 235 236 237
			}

			if (c == EOF)
				end = true;
		}
		else
		{
238 239
			/*
			 * otherwise read characters until EOF.
240 241
			 */
			while ((c = getc(stdin)) != EOF)
242
				appendStringInfoChar(inBuf, (char) c);
243

244
			if (inBuf->len == 0)
245 246 247 248
				end = true;
		}

		if (end)
249
			return EOF;
250

251 252
		/*
		 * otherwise we have a user query so process it.
253 254 255 256
		 */
		break;
	}

257 258 259
	/* Add '\0' to make it look the same as message case. */
	appendStringInfoChar(inBuf, (char) '\0');

260 261
	/*
	 * if the query echo flag was given, print the query..
262 263
	 */
	if (EchoQuery)
264
		printf("statement: %s\n", inBuf->data);
265
	fflush(stdout);
266

267
	return 'Q';
268 269 270
}

/* ----------------
271
 *	SocketBackend()		Is called for frontend-backend connections
272
 *
273
 *	Returns the message type code, and loads message body data into inBuf.
274
 *
275
 *	EOF is returned if the connection is lost.
276 277
 * ----------------
 */
278
static int
279
SocketBackend(StringInfo inBuf)
280
{
281
	int			qtype;
282

283
	/*
284
	 * Get message type code from the frontend.
285
	 */
286
	qtype = pq_getbyte();
287

288 289
	if (qtype == EOF)			/* frontend disconnected */
	{
290 291 292
		ereport(COMMERROR,
				(errcode(ERRCODE_PROTOCOL_VIOLATION),
				 errmsg("unexpected EOF on client connection")));
293 294 295 296
		return qtype;
	}

	/*
B
Bruce Momjian 已提交
297 298 299
	 * Validate message type code before trying to read body; if we have lost
	 * sync, better to say "command unknown" than to run out of memory because
	 * we used garbage as a length word.
300
	 *
B
Bruce Momjian 已提交
301 302
	 * This also gives us a place to set the doing_extended_query_message flag as
	 * soon as possible.
303
	 */
304
	switch (qtype)
305
	{
306
		case 'Q':				/* simple query */
307
			doing_extended_query_message = false;
308 309 310 311 312
			if (PG_PROTOCOL_MAJOR(FrontendProtocol) < 3)
			{
				/* old style without length word; convert */
				if (pq_getstring(inBuf))
				{
313 314
					ereport(COMMERROR,
							(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
315
							 errmsg("unexpected EOF on client connection")));
316 317 318
					return EOF;
				}
			}
319 320
			break;

321
		case 'F':				/* fastpath function call */
322 323
			/* we let fastpath.c cope with old-style input of this */
			doing_extended_query_message = false;
324
			break;
325

326
		case 'X':				/* terminate */
327
			doing_extended_query_message = false;
328
			ignore_till_sync = false;
329 330 331 332 333 334 335 336 337 338 339
			break;

		case 'B':				/* bind */
		case 'C':				/* close */
		case 'D':				/* describe */
		case 'E':				/* execute */
		case 'H':				/* flush */
		case 'P':				/* parse */
			doing_extended_query_message = true;
			/* these are only legal in protocol 3 */
			if (PG_PROTOCOL_MAJOR(FrontendProtocol) < 3)
340 341
				ereport(FATAL,
						(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
342
						 errmsg("invalid frontend message type %d", qtype)));
343 344 345 346 347 348 349 350 351
			break;

		case 'S':				/* sync */
			/* stop any active skip-till-Sync */
			ignore_till_sync = false;
			/* mark not-extended, so that a new error doesn't begin skip */
			doing_extended_query_message = false;
			/* only legal in protocol 3 */
			if (PG_PROTOCOL_MAJOR(FrontendProtocol) < 3)
352 353
				ereport(FATAL,
						(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
354
						 errmsg("invalid frontend message type %d", qtype)));
355
			break;
356

357 358 359
		case 'd':				/* copy data */
		case 'c':				/* copy done */
		case 'f':				/* copy fail */
360 361 362
			doing_extended_query_message = false;
			/* these are only legal in protocol 3 */
			if (PG_PROTOCOL_MAJOR(FrontendProtocol) < 3)
363 364
				ereport(FATAL,
						(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
365
						 errmsg("invalid frontend message type %d", qtype)));
366
			break;
367

368
		default:
B
Bruce Momjian 已提交
369

370
			/*
B
Bruce Momjian 已提交
371 372 373
			 * Otherwise we got garbage from the frontend.	We treat this as
			 * fatal because we have probably lost message boundary sync, and
			 * there's no good way to recover.
374
			 */
375 376 377
			ereport(FATAL,
					(errcode(ERRCODE_PROTOCOL_VIOLATION),
					 errmsg("invalid frontend message type %d", qtype)));
378
			break;
379
	}
380

381
	/*
B
Bruce Momjian 已提交
382 383 384
	 * In protocol version 3, all frontend messages have a length word next
	 * after the type code; we can read the message contents independently of
	 * the type.
385 386 387 388 389 390 391
	 */
	if (PG_PROTOCOL_MAJOR(FrontendProtocol) >= 3)
	{
		if (pq_getmessage(inBuf, 0))
			return EOF;			/* suitable message already logged */
	}

392
	return qtype;
393 394 395
}

/* ----------------
396
 *		ReadCommand reads a command from either the frontend or
397 398 399
 *		standard input, places it in inBuf, and returns the
 *		message type code (first byte of the message).
 *		EOF is returned if end of file.
400 401
 * ----------------
 */
402
static int
403
ReadCommand(StringInfo inBuf)
404
{
405
	int			result;
406

407
	if (whereToSendOutput == DestRemote)
408
		result = SocketBackend(inBuf);
409
	else
410 411
		result = InteractiveBackend(inBuf);
	return result;
412 413
}

414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459
/*
 * prepare_for_client_read -- set up to possibly block on client input
 *
 * This must be called immediately before any low-level read from the
 * client connection.  It is necessary to do it at a sufficiently low level
 * that there won't be any other operations except the read kernel call
 * itself between this call and the subsequent client_read_ended() call.
 * In particular there mustn't be use of malloc() or other potentially
 * non-reentrant libc functions.  This restriction makes it safe for us
 * to allow interrupt service routines to execute nontrivial code while
 * we are waiting for input.
 */
void
prepare_for_client_read(void)
{
	if (DoingCommandRead)
	{
		/* Enable immediate processing of asynchronous signals */
		EnableNotifyInterrupt();
		EnableCatchupInterrupt();

		/* Allow "die" interrupt to be processed while waiting */
		ImmediateInterruptOK = true;

		/* And don't forget to detect one that already arrived */
		QueryCancelPending = false;
		CHECK_FOR_INTERRUPTS();
	}
}

/*
 * client_read_ended -- get out of the client-input state
 */
void
client_read_ended(void)
{
	if (DoingCommandRead)
	{
		ImmediateInterruptOK = false;
		QueryCancelPending = false;		/* forget any CANCEL signal */

		DisableNotifyInterrupt();
		DisableCatchupInterrupt();
	}
}

460 461 462 463 464 465

/*
 * Parse a query string and pass it through the rewriter.
 *
 * A list of Query nodes is returned, since the string might contain
 * multiple queries and/or the rewriter might expand one query to several.
466 467 468
 *
 * NOTE: this routine is no longer used for processing interactive queries,
 * but it is still needed for parsing of SQL function bodies.
469
 */
470
List *
B
Bruce Momjian 已提交
471
pg_parse_and_rewrite(const char *query_string,	/* string to execute */
472
					 Oid *paramTypes,	/* parameter types */
B
Bruce Momjian 已提交
473
					 int numParams)		/* number of parameters */
474
{
475
	List	   *raw_parsetree_list;
476
	List	   *querytree_list;
477
	ListCell   *list_item;
478

479 480
	/*
	 * (1) parse the request string into a list of raw parse trees.
481
	 */
482
	raw_parsetree_list = pg_parse_query(query_string);
483

484 485
	/*
	 * (2) Do parse analysis and rule rewrite.
486 487 488 489
	 */
	querytree_list = NIL;
	foreach(list_item, raw_parsetree_list)
	{
B
Bruce Momjian 已提交
490
		Node	   *parsetree = (Node *) lfirst(list_item);
491

492
		querytree_list = list_concat(querytree_list,
B
Bruce Momjian 已提交
493 494 495
									 pg_analyze_and_rewrite(parsetree,
															paramTypes,
															numParams));
496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513
	}

	return querytree_list;
}

/*
 * Do raw parsing (only).
 *
 * A list of parsetrees is returned, since there might be multiple
 * commands in the given string.
 *
 * NOTE: for interactive queries, it is important to keep this routine
 * separate from the analysis & rewrite stages.  Analysis and rewriting
 * cannot be done in an aborted transaction, since they require access to
 * database tables.  So, we rely on the raw parser to determine whether
 * we've seen a COMMIT or ABORT command; when we are in abort state, other
 * commands are not processed any further than the raw parse stage.
 */
514
List *
515
pg_parse_query(const char *query_string)
516
{
517
	List	   *raw_parsetree_list;
518

B
Rename:  
Bruce Momjian 已提交
519
	if (log_parser_stats)
520 521
		ResetUsage();

522
	raw_parsetree_list = raw_parser(query_string);
523

524 525 526 527 528 529 530 531
	if (log_parser_stats)
		ShowUsage("PARSER STATISTICS");

	return raw_parsetree_list;
}

static bool
log_after_parse(List *raw_parsetree_list, const char *query_string,
B
Bruce Momjian 已提交
532
				char **prepare_string)
533 534 535 536 537 538
{
	ListCell   *parsetree_item;
	bool		log_this_statement = (log_statement == LOGSTMT_ALL);

	*prepare_string = NULL;

B
Bruce Momjian 已提交
539
	/* Check if we need to log the statement, and get prepare_string. */
540
	foreach(parsetree_item, raw_parsetree_list)
541
	{
542 543 544 545 546 547 548 549 550 551
		Node	   *parsetree = (Node *) lfirst(parsetree_item);
		const char *commandTag;

		if (IsA(parsetree, ExplainStmt) &&
			((ExplainStmt *) parsetree)->analyze)
			parsetree = (Node *) (((ExplainStmt *) parsetree)->query);

		if (IsA(parsetree, PrepareStmt))
			parsetree = (Node *) (((PrepareStmt *) parsetree)->query);

552 553
		if (IsA(parsetree, SelectStmt) &&
			((SelectStmt *) parsetree)->into == NULL)
B
Bruce Momjian 已提交
554
			continue;			/* optimization for frequent command */
555 556 557 558 559 560 561

		if (log_statement == LOGSTMT_MOD &&
			(IsA(parsetree, InsertStmt) ||
			 IsA(parsetree, UpdateStmt) ||
			 IsA(parsetree, DeleteStmt) ||
			 IsA(parsetree, TruncateStmt) ||
			 (IsA(parsetree, CopyStmt) &&
B
Bruce Momjian 已提交
562
			  ((CopyStmt *) parsetree)->is_from)))		/* COPY FROM */
563 564 565 566 567 568
			log_this_statement = true;

		commandTag = CreateCommandTag(parsetree);
		if ((log_statement == LOGSTMT_MOD ||
			 log_statement == LOGSTMT_DDL) &&
			(strncmp(commandTag, "CREATE ", strlen("CREATE ")) == 0 ||
B
Bruce Momjian 已提交
569
			 IsA(parsetree, SelectStmt) ||		/* SELECT INTO, CREATE AS */
570 571
			 strncmp(commandTag, "ALTER ", strlen("ALTER ")) == 0 ||
			 strncmp(commandTag, "DROP ", strlen("DROP ")) == 0 ||
B
Bruce Momjian 已提交
572
			 IsA(parsetree, GrantStmt) ||		/* GRANT or REVOKE */
573 574 575 576
			 IsA(parsetree, CommentStmt)))
			log_this_statement = true;

		/*
B
Bruce Momjian 已提交
577 578
		 * For the first EXECUTE we find, record the client statement used by
		 * the PREPARE.
579 580
		 */
		if (IsA(parsetree, ExecuteStmt))
581
		{
582 583 584 585 586
			ExecuteStmt *stmt = (ExecuteStmt *) parsetree;
			PreparedStatement *entry;

			if ((entry = FetchPreparedStatement(stmt->name, false)) != NULL &&
				entry->query_string)
587
			{
588
				*prepare_string = palloc(strlen(entry->query_string) +
B
Bruce Momjian 已提交
589
									  strlen("  [client PREPARE:  %s]") - 1);
590
				sprintf(*prepare_string, "  [client PREPARE:  %s]",
B
Bruce Momjian 已提交
591
						entry->query_string);
592 593 594
			}
		}
	}
B
Bruce Momjian 已提交
595

596 597 598 599 600 601 602 603 604
	if (log_this_statement)
	{
		ereport(LOG,
				(errmsg("statement: %s%s", query_string,
						*prepare_string ? *prepare_string : "")));
		return true;
	}
	else
		return false;
605 606
}

607

608
/*
609 610
 * Given a raw parsetree (gram.y output), and optionally information about
 * types of parameter symbols ($n), perform parse analysis and rule rewriting.
611 612 613 614 615 616
 *
 * A list of Query nodes is returned, since either the analyzer or the
 * rewriter might expand one query to several.
 *
 * NOTE: for reasons mentioned above, this must be separate from raw parsing.
 */
617
List *
618
pg_analyze_and_rewrite(Node *parsetree, Oid *paramTypes, int numParams)
619 620 621
{
	List	   *querytree_list;

622 623
	/*
	 * (1) Perform parse analysis.
624
	 */
B
Rename:  
Bruce Momjian 已提交
625
	if (log_parser_stats)
626 627
		ResetUsage();

628
	querytree_list = parse_analyze(parsetree, paramTypes, numParams);
629

B
Rename:  
Bruce Momjian 已提交
630
	if (log_parser_stats)
631
		ShowUsage("PARSE ANALYSIS STATISTICS");
632

633 634
	/*
	 * (2) Rewrite the queries, as necessary
635 636 637 638 639 640 641 642
	 */
	querytree_list = pg_rewrite_queries(querytree_list);

	return querytree_list;
}

/*
 * Perform rewriting of a list of queries produced by parse analysis.
643 644 645
 *
 * Note: queries must just have come from the parser, because we do not do
 * AcquireRewriteLocks() on them.
646
 */
647
static List *
648 649 650
pg_rewrite_queries(List *querytree_list)
{
	List	   *new_list = NIL;
651
	ListCell   *list_item;
652 653 654 655 656

	if (log_parser_stats)
		ResetUsage();

	/*
B
Bruce Momjian 已提交
657 658
	 * rewritten queries are collected in new_list.  Note there may be more or
	 * fewer than in the original list.
659
	 */
660
	foreach(list_item, querytree_list)
661
	{
662
		Query	   *querytree = (Query *) lfirst(list_item);
663

664
		if (Debug_print_parse)
665
			elog_node_display(DEBUG1, "parse tree", querytree,
666
							  Debug_pretty_print);
667 668 669

		if (querytree->commandType == CMD_UTILITY)
		{
670 671
			/* don't rewrite utilities, just dump 'em into new_list */
			new_list = lappend(new_list, querytree);
672
		}
673
		else
674
		{
675
			/* rewrite regular queries */
676 677
			List	   *rewritten = QueryRewrite(querytree);

678
			new_list = list_concat(new_list, rewritten);
679 680 681 682 683
		}
	}

	querytree_list = new_list;

B
Rename:  
Bruce Momjian 已提交
684
	if (log_parser_stats)
685
		ShowUsage("REWRITER STATISTICS");
686

687
#ifdef COPY_PARSE_PLAN_TREES
B
Bruce Momjian 已提交
688 689

	/*
B
Bruce Momjian 已提交
690
	 * Optional debugging check: pass querytree output through copyObject()
B
Bruce Momjian 已提交
691
	 */
692 693
	new_list = (List *) copyObject(querytree_list);
	/* This checks both copyObject() and the equal() routines... */
B
Bruce Momjian 已提交
694
	if (!equal(new_list, querytree_list))
695
		elog(WARNING, "copyObject() failed to produce an equal parse tree");
696 697
	else
		querytree_list = new_list;
698 699
#endif

700
	if (Debug_print_rewritten)
701
		elog_node_display(DEBUG1, "rewritten parse tree", querytree_list,
702
						  Debug_pretty_print);
703

704 705
	return querytree_list;
}
706 707


708
/* Generate a plan for a single already-rewritten query. */
709
Plan *
710
pg_plan_query(Query *querytree, ParamListInfo boundParams)
711 712
{
	Plan	   *plan;
713

714 715 716
	/* Utility commands have no plans. */
	if (querytree->commandType == CMD_UTILITY)
		return NULL;
717

B
Rename:  
Bruce Momjian 已提交
718
	if (log_planner_stats)
719
		ResetUsage();
720

721
	/* call the optimizer */
722
	plan = planner(querytree, false, 0, boundParams);
723

B
Rename:  
Bruce Momjian 已提交
724
	if (log_planner_stats)
725
		ShowUsage("PLANNER STATISTICS");
726

727 728 729
#ifdef COPY_PARSE_PLAN_TREES
	/* Optional debugging check: pass plan output through copyObject() */
	{
B
Bruce Momjian 已提交
730
		Plan	   *new_plan = (Plan *) copyObject(plan);
731

B
Bruce Momjian 已提交
732
		/*
B
Bruce Momjian 已提交
733 734
		 * equal() currently does not have routines to compare Plan nodes, so
		 * don't try to test equality here.  Perhaps fix someday?
735 736 737
		 */
#ifdef NOT_USED
		/* This checks both copyObject() and the equal() routines... */
B
Bruce Momjian 已提交
738
		if (!equal(new_plan, plan))
739
			elog(WARNING, "copyObject() failed to produce an equal plan tree");
740 741 742 743 744 745
		else
#endif
			plan = new_plan;
	}
#endif

746 747
	/*
	 * Print plan if debugging.
748
	 */
749
	if (Debug_print_plan)
750
		elog_node_display(DEBUG1, "plan", plan, Debug_pretty_print);
751

752
	return plan;
753 754
}

755 756
/*
 * Generate plans for a list of already-rewritten queries.
757
 *
758 759 760 761 762 763 764 765 766
 * If needSnapshot is TRUE, we haven't yet set a snapshot for the current
 * query.  A snapshot must be set before invoking the planner, since it
 * might try to evaluate user-defined functions.  But we must not set a
 * snapshot if the list contains only utility statements, because some
 * utility statements depend on not having frozen the snapshot yet.
 * (We assume that such statements cannot appear together with plannable
 * statements in the rewriter's output.)
 */
List *
767 768
pg_plan_queries(List *querytrees, ParamListInfo boundParams,
				bool needSnapshot)
769 770
{
	List	   *plan_list = NIL;
771
	ListCell   *query_list;
772 773 774 775 776 777 778 779 780 781 782 783 784 785 786

	foreach(query_list, querytrees)
	{
		Query	   *query = (Query *) lfirst(query_list);
		Plan	   *plan;

		if (query->commandType == CMD_UTILITY)
		{
			/* Utility commands have no plans. */
			plan = NULL;
		}
		else
		{
			if (needSnapshot)
			{
787
				ActiveSnapshot = CopySnapshot(GetTransactionSnapshot());
788 789
				needSnapshot = false;
			}
790
			plan = pg_plan_query(query, boundParams);
791 792 793 794 795 796 797 798 799 800
		}

		plan_list = lappend(plan_list, plan);
	}

	return plan_list;
}


/*
801
 * exec_simple_query
802
 *
803
 * Execute a "simple Query" protocol message.
804
 */
805
static void
806
exec_simple_query(const char *query_string)
807
{
B
Bruce Momjian 已提交
808
	CommandDest dest = whereToSendOutput;
809
	MemoryContext oldcontext;
810 811
	List	   *parsetree_list;
	ListCell   *parsetree_item;
B
Bruce Momjian 已提交
812 813
	struct timeval start_t,
				stop_t;
B
Rename:  
Bruce Momjian 已提交
814
	bool		save_log_duration = log_duration;
B
Bruce Momjian 已提交
815
	int			save_log_min_duration_statement = log_min_duration_statement;
816
	bool		save_log_statement_stats = log_statement_stats;
B
Bruce Momjian 已提交
817
	char	   *prepare_string = NULL;
818
	bool		was_logged = false;
B
Bruce Momjian 已提交
819

820 821 822
	/*
	 * Report query to various monitoring facilities.
	 */
823
	debug_query_string = query_string;
824

825 826
	pgstat_report_activity(query_string);

827
	/*
B
Bruce Momjian 已提交
828
	 * We use save_log_* so "SET log_duration = true"  and "SET
B
Bruce Momjian 已提交
829 830 831
	 * log_min_duration_statement = true" don't report incorrect time because
	 * gettimeofday() wasn't called. Similarly, log_statement_stats has to be
	 * captured once.
832
	 */
833
	if (save_log_duration || save_log_min_duration_statement != -1)
834
		gettimeofday(&start_t, NULL);
835

836 837 838
	if (save_log_statement_stats)
		ResetUsage();

839
	/*
B
Bruce Momjian 已提交
840 841
	 * Start up a transaction command.	All queries generated by the
	 * query_string will be in this same command block, *unless* we find a
B
Bruce Momjian 已提交
842 843 844
	 * BEGIN/COMMIT/ABORT statement; we have to force a new xact command after
	 * one of those, else bad things will happen in xact.c. (Note that this
	 * will normally change current memory context.)
845 846
	 */
	start_xact_command();
847 848

	/*
B
Bruce Momjian 已提交
849 850 851 852
	 * Zap any pre-existing unnamed statement.	(While not strictly necessary,
	 * it seems best to define simple-Query mode as if it used the unnamed
	 * statement and portal; this ensures we recover any storage used by prior
	 * unnamed operations.)
853 854 855 856 857 858 859 860
	 */
	unnamed_stmt_pstmt = NULL;
	if (unnamed_stmt_context)
	{
		DropDependentPortals(unnamed_stmt_context);
		MemoryContextDelete(unnamed_stmt_context);
	}
	unnamed_stmt_context = NULL;
861

862 863 864
	/*
	 * Switch to appropriate context for constructing parsetrees.
	 */
865
	oldcontext = MemoryContextSwitchTo(MessageContext);
866

867 868
	QueryContext = CurrentMemoryContext;

B
Bruce Momjian 已提交
869
	/*
B
Bruce Momjian 已提交
870 871
	 * Do basic parsing of the query or queries (this should be safe even if
	 * we are in aborted transaction state!)
872
	 */
873
	parsetree_list = pg_parse_query(query_string);
874

875 876
	if (log_statement != LOGSTMT_NONE || save_log_min_duration_statement != -1)
		was_logged = log_after_parse(parsetree_list, query_string,
B
Bruce Momjian 已提交
877
									 &prepare_string);
878

879
	/*
880
	 * Switch back to transaction context to enter the loop.
881 882 883 884
	 */
	MemoryContextSwitchTo(oldcontext);

	/*
885
	 * Run through the raw parsetree(s) and process each one.
886
	 */
887
	foreach(parsetree_item, parsetree_list)
888
	{
B
Bruce Momjian 已提交
889
		Node	   *parsetree = (Node *) lfirst(parsetree_item);
890 891
		const char *commandTag;
		char		completionTag[COMPLETION_TAG_BUFSIZE];
B
Bruce Momjian 已提交
892
		List	   *querytree_list,
893 894
				   *plantree_list;
		Portal		portal;
895 896
		DestReceiver *receiver;
		int16		format;
897

898
		/*
B
Bruce Momjian 已提交
899 900 901 902
		 * Get the command name for use in status display (it also becomes the
		 * default completion tag, down inside PortalRun).	Set ps_status and
		 * do any special start-of-SQL-command processing needed by the
		 * destination.
903 904 905 906 907
		 */
		commandTag = CreateCommandTag(parsetree);

		set_ps_display(commandTag);

908
		BeginCommand(commandTag, dest);
909

910
		/*
911
		 * If we are in an aborted transaction, reject all commands except
B
Bruce Momjian 已提交
912 913 914 915 916
		 * COMMIT/ABORT.  It is important that this test occur before we try
		 * to do parse analysis, rewrite, or planning, since all those phases
		 * try to do database accesses, which may fail in abort state. (It
		 * might be safe to allow some additional utility commands in this
		 * state, but not many...)
917 918
		 */
		if (IsAbortedTransactionBlockState())
919
		{
B
Bruce Momjian 已提交
920
			bool		allowit = false;
921

922
			if (IsA(parsetree, TransactionStmt))
923 924 925
			{
				TransactionStmt *stmt = (TransactionStmt *) parsetree;

926
				if (stmt->kind == TRANS_STMT_COMMIT ||
927
					stmt->kind == TRANS_STMT_PREPARE ||
928 929
					stmt->kind == TRANS_STMT_ROLLBACK ||
					stmt->kind == TRANS_STMT_ROLLBACK_TO)
930
					allowit = true;
931
			}
932

B
Bruce Momjian 已提交
933
			if (!allowit)
934 935 936
				ereport(ERROR,
						(errcode(ERRCODE_IN_FAILED_SQL_TRANSACTION),
						 errmsg("current transaction is aborted, "
B
Bruce Momjian 已提交
937
						"commands ignored until end of transaction block")));
938
		}
939

940
		/* Make sure we are in a transaction command */
941
		start_xact_command();
942

943
		/* If we got a cancel signal in parsing or prior command, quit */
944
		CHECK_FOR_INTERRUPTS();
945 946

		/*
947
		 * OK to analyze, rewrite, and plan this query.
948
		 *
B
Bruce Momjian 已提交
949 950
		 * Switch to appropriate context for constructing querytrees (again,
		 * these must outlive the execution context).
951
		 */
952
		oldcontext = MemoryContextSwitchTo(MessageContext);
953

954
		querytree_list = pg_analyze_and_rewrite(parsetree, NULL, 0);
V
Vadim B. Mikheev 已提交
955

956
		plantree_list = pg_plan_queries(querytree_list, NULL, true);
957 958 959 960

		/* If we got a cancel signal in analysis or planning, quit */
		CHECK_FOR_INTERRUPTS();

961
		/*
B
Bruce Momjian 已提交
962 963
		 * Create unnamed portal to run the query or queries in. If there
		 * already is one, silently drop it.
964
		 */
965
		portal = CreatePortal("", true, true);
966

967 968 969 970 971 972
		PortalDefineQuery(portal,
						  query_string,
						  commandTag,
						  querytree_list,
						  plantree_list,
						  MessageContext);
973

974
		/*
975
		 * Start the portal.  No parameters here.
976
		 */
977
		PortalStart(portal, NULL, InvalidSnapshot);
978

979
		/*
B
Bruce Momjian 已提交
980 981 982 983
		 * Select the appropriate output format: text unless we are doing a
		 * FETCH from a binary cursor.	(Pretty grotty to have to do this here
		 * --- but it avoids grottiness in other places.  Ah, the joys of
		 * backward compatibility...)
984 985 986 987 988 989 990 991 992 993 994 995
		 */
		format = 0;				/* TEXT is default */
		if (IsA(parsetree, FetchStmt))
		{
			FetchStmt  *stmt = (FetchStmt *) parsetree;

			if (!stmt->ismove)
			{
				Portal		fportal = GetPortalByName(stmt->portalname);

				if (PortalIsValid(fportal) &&
					(fportal->cursorOptions & CURSOR_OPT_BINARY))
B
Bruce Momjian 已提交
996
					format = 1; /* BINARY */
997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011
			}
		}
		PortalSetResultFormat(portal, 1, &format);

		/*
		 * Now we can create the destination receiver object.
		 */
		receiver = CreateDestReceiver(dest, portal);

		/*
		 * Switch back to transaction context for execution.
		 */
		MemoryContextSwitchTo(oldcontext);

		/*
B
Bruce Momjian 已提交
1012
		 * Run the portal to completion, and then drop it (and the receiver).
1013
		 */
1014 1015
		(void) PortalRun(portal,
						 FETCH_ALL,
1016 1017
						 receiver,
						 receiver,
1018
						 completionTag);
1019

1020
		(*receiver->rDestroy) (receiver);
1021

1022
		PortalDrop(portal, false);
1023

1024
		if (IsA(parsetree, TransactionStmt))
1025
		{
1026
			/*
B
Bruce Momjian 已提交
1027 1028
			 * If this was a transaction control statement, commit it. We will
			 * start a new xact command for the next command (if any).
1029
			 */
1030
			finish_xact_command();
1031
		}
1032
		else if (lnext(parsetree_item) == NULL)
1033
		{
1034
			/*
B
Bruce Momjian 已提交
1035 1036 1037 1038 1039 1040 1041 1042
			 * If this is the last parsetree of the query string, close down
			 * transaction statement before reporting command-complete.  This
			 * is so that any end-of-transaction errors are reported before
			 * the command-complete message is issued, to avoid confusing
			 * clients who will expect either a command-complete message or an
			 * error, not one and then the other.  But for compatibility with
			 * historical Postgres behavior, we do not force a transaction
			 * boundary between queries appearing in a single query string.
1043
			 */
1044
			finish_xact_command();
1045
		}
1046
		else
1047
		{
1048
			/*
B
Bruce Momjian 已提交
1049 1050
			 * We need a CommandCounterIncrement after every query, except
			 * those that start or end a transaction block.
1051 1052
			 */
			CommandCounterIncrement();
1053
		}
1054 1055

		/*
B
Bruce Momjian 已提交
1056 1057 1058 1059
		 * Tell client that we're done with this query.  Note we emit exactly
		 * one EndCommand report for each raw parsetree, thus one for each SQL
		 * command the client sent, regardless of rewriting. (But a command
		 * aborted by error will not send an EndCommand report at all.)
1060
		 */
1061
		EndCommand(completionTag, dest);
B
Bruce Momjian 已提交
1062
	}							/* end loop over parsetrees */
1063

1064 1065 1066 1067 1068
	/*
	 * Close down transaction statement, if one is open.
	 */
	finish_xact_command();

1069 1070 1071
	/*
	 * If there were no parsetrees, return EmptyQueryResponse message.
	 */
1072
	if (!parsetree_list)
1073 1074
		NullCommand(dest);

1075
	QueryContext = NULL;
1076

1077
	/*
B
Bruce Momjian 已提交
1078 1079
	 * Combine processing here as we need to calculate the query duration in
	 * both instances.
1080
	 */
1081
	if (save_log_duration || save_log_min_duration_statement != -1)
1082
	{
B
Bruce Momjian 已提交
1083 1084
		long		usecs;

1085
		gettimeofday(&stop_t, NULL);
1086 1087 1088 1089 1090
		if (stop_t.tv_usec < start_t.tv_usec)
		{
			stop_t.tv_sec--;
			stop_t.tv_usec += 1000000;
		}
1091
		usecs = (long) (stop_t.tv_sec - start_t.tv_sec) * 1000000 +
B
Bruce Momjian 已提交
1092
			(long) (stop_t.tv_usec - start_t.tv_usec);
B
Bruce Momjian 已提交
1093

1094
		/* Only print duration if we previously printed the statement. */
1095
		if (was_logged && save_log_duration)
B
Bruce Momjian 已提交
1096 1097
			ereport(LOG,
					(errmsg("duration: %ld.%03ld ms",
B
Bruce Momjian 已提交
1098 1099 1100
							(long) ((stop_t.tv_sec - start_t.tv_sec) * 1000 +
								  (stop_t.tv_usec - start_t.tv_usec) / 1000),
						 (long) (stop_t.tv_usec - start_t.tv_usec) % 1000)));
B
Bruce Momjian 已提交
1101

B
Bruce Momjian 已提交
1102
		/*
B
Bruce Momjian 已提交
1103 1104
		 * Output a duration_statement to the log if the query has exceeded
		 * the min duration, or if we are to print all durations.
B
Bruce Momjian 已提交
1105
		 */
B
Bruce Momjian 已提交
1106 1107 1108
		if (save_log_min_duration_statement == 0 ||
			(save_log_min_duration_statement > 0 &&
			 usecs >= save_log_min_duration_statement * 1000))
1109
			ereport(LOG,
1110
					(errmsg("duration: %ld.%03ld ms  statement: %s%s",
B
Bruce Momjian 已提交
1111 1112 1113
							(long) ((stop_t.tv_sec - start_t.tv_sec) * 1000 +
								  (stop_t.tv_usec - start_t.tv_usec) / 1000),
							(long) (stop_t.tv_usec - start_t.tv_usec) % 1000,
1114 1115
							query_string,
							prepare_string ? prepare_string : "")));
1116
	}
1117

1118 1119 1120
	if (save_log_statement_stats)
		ShowUsage("QUERY STATISTICS");

1121 1122 1123
	if (prepare_string != NULL)
		pfree(prepare_string);

1124
	debug_query_string = NULL;
1125 1126
}

1127 1128 1129 1130 1131 1132 1133 1134
/*
 * exec_parse_message
 *
 * Execute a "Parse" protocol message.
 */
static void
exec_parse_message(const char *query_string,	/* string to execute */
				   const char *stmt_name,		/* name for prepared stmt */
B
Bruce Momjian 已提交
1135 1136
				   Oid *paramTypes,		/* parameter types */
				   int numParams)		/* number of parameters */
1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158
{
	MemoryContext oldcontext;
	List	   *parsetree_list;
	const char *commandTag;
	List	   *querytree_list,
			   *plantree_list,
			   *param_list;
	bool		is_named;
	bool		save_log_statement_stats = log_statement_stats;

	/*
	 * Report query to various monitoring facilities.
	 */
	debug_query_string = query_string;

	pgstat_report_activity(query_string);

	set_ps_display("PARSE");

	if (save_log_statement_stats)
		ResetUsage();

1159 1160
	if (log_statement == LOGSTMT_ALL)
		ereport(LOG,
1161 1162 1163
				(errmsg("statement: PREPARE %s AS %s",
						(*stmt_name != '\0') ? stmt_name : "<unnamed>",
						query_string)));
1164

1165
	/*
B
Bruce Momjian 已提交
1166 1167 1168
	 * Start up a transaction command so we can run parse analysis etc. (Note
	 * that this will normally change current memory context.) Nothing happens
	 * if we are already in one.
1169 1170 1171 1172 1173 1174
	 */
	start_xact_command();

	/*
	 * Switch to appropriate context for constructing parsetrees.
	 *
B
Bruce Momjian 已提交
1175 1176 1177
	 * We have two strategies depending on whether the prepared statement is
	 * named or not.  For a named prepared statement, we do parsing in
	 * MessageContext and copy the finished trees into the prepared
B
Bruce Momjian 已提交
1178 1179 1180 1181 1182 1183
	 * statement's private context; then the reset of MessageContext releases
	 * temporary space used by parsing and planning.  For an unnamed prepared
	 * statement, we assume the statement isn't going to hang around long, so
	 * getting rid of temp space quickly is probably not worth the costs of
	 * copying parse/plan trees.  So in this case, we set up a special context
	 * for the unnamed statement, and do all the parsing/planning therein.
1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213
	 */
	is_named = (stmt_name[0] != '\0');
	if (is_named)
	{
		/* Named prepared statement --- parse in MessageContext */
		oldcontext = MemoryContextSwitchTo(MessageContext);
	}
	else
	{
		/* Unnamed prepared statement --- release any prior unnamed stmt */
		unnamed_stmt_pstmt = NULL;
		if (unnamed_stmt_context)
		{
			DropDependentPortals(unnamed_stmt_context);
			MemoryContextDelete(unnamed_stmt_context);
		}
		unnamed_stmt_context = NULL;
		/* create context for parsing/planning */
		unnamed_stmt_context =
			AllocSetContextCreate(TopMemoryContext,
								  "unnamed prepared statement",
								  ALLOCSET_DEFAULT_MINSIZE,
								  ALLOCSET_DEFAULT_INITSIZE,
								  ALLOCSET_DEFAULT_MAXSIZE);
		oldcontext = MemoryContextSwitchTo(unnamed_stmt_context);
	}

	QueryContext = CurrentMemoryContext;

	/*
B
Bruce Momjian 已提交
1214 1215
	 * Do basic parsing of the query or queries (this should be safe even if
	 * we are in aborted transaction state!)
1216 1217 1218 1219
	 */
	parsetree_list = pg_parse_query(query_string);

	/*
B
Bruce Momjian 已提交
1220 1221 1222
	 * We only allow a single user statement in a prepared statement. This is
	 * mainly to keep the protocol simple --- otherwise we'd need to worry
	 * about multiple result tupdescs and things like that.
1223
	 */
1224
	if (list_length(parsetree_list) > 1)
1225 1226
		ereport(ERROR,
				(errcode(ERRCODE_SYNTAX_ERROR),
B
Bruce Momjian 已提交
1227
		errmsg("cannot insert multiple commands into a prepared statement")));
1228 1229 1230

	if (parsetree_list != NIL)
	{
1231
		Node	   *parsetree = (Node *) linitial(parsetree_list);
B
Bruce Momjian 已提交
1232
		int			i;
1233 1234 1235 1236 1237 1238 1239 1240

		/*
		 * Get the command name for possible use in status display.
		 */
		commandTag = CreateCommandTag(parsetree);

		/*
		 * If we are in an aborted transaction, reject all commands except
B
Bruce Momjian 已提交
1241 1242 1243 1244 1245
		 * COMMIT/ROLLBACK.  It is important that this test occur before we
		 * try to do parse analysis, rewrite, or planning, since all those
		 * phases try to do database accesses, which may fail in abort state.
		 * (It might be safe to allow some additional utility commands in this
		 * state, but not many...)
1246 1247 1248 1249 1250 1251 1252 1253 1254 1255
		 */
		if (IsAbortedTransactionBlockState())
		{
			bool		allowit = false;

			if (IsA(parsetree, TransactionStmt))
			{
				TransactionStmt *stmt = (TransactionStmt *) parsetree;

				if (stmt->kind == TRANS_STMT_COMMIT ||
1256
					stmt->kind == TRANS_STMT_PREPARE ||
1257 1258
					stmt->kind == TRANS_STMT_ROLLBACK ||
					stmt->kind == TRANS_STMT_ROLLBACK_TO)
1259 1260 1261 1262
					allowit = true;
			}

			if (!allowit)
1263 1264 1265
				ereport(ERROR,
						(errcode(ERRCODE_IN_FAILED_SQL_TRANSACTION),
						 errmsg("current transaction is aborted, "
B
Bruce Momjian 已提交
1266
						"commands ignored until end of transaction block")));
1267 1268 1269 1270
		}

		/*
		 * OK to analyze, rewrite, and plan this query.  Note that the
B
Bruce Momjian 已提交
1271 1272
		 * originally specified parameter set is not required to be complete,
		 * so we have to use parse_analyze_varparams().
1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287
		 */
		if (log_parser_stats)
			ResetUsage();

		querytree_list = parse_analyze_varparams(parsetree,
												 &paramTypes,
												 &numParams);

		/*
		 * Check all parameter types got determined, and convert array
		 * representation to a list for storage.
		 */
		param_list = NIL;
		for (i = 0; i < numParams; i++)
		{
B
Bruce Momjian 已提交
1288
			Oid			ptype = paramTypes[i];
1289 1290

			if (ptype == InvalidOid || ptype == UNKNOWNOID)
1291 1292
				ereport(ERROR,
						(errcode(ERRCODE_INDETERMINATE_DATATYPE),
B
Bruce Momjian 已提交
1293 1294
					 errmsg("could not determine data type of parameter $%d",
							i + 1)));
1295
			param_list = lappend_oid(param_list, ptype);
1296 1297 1298 1299 1300 1301 1302
		}

		if (log_parser_stats)
			ShowUsage("PARSE ANALYSIS STATISTICS");

		querytree_list = pg_rewrite_queries(querytree_list);

1303
		/*
B
Bruce Momjian 已提交
1304 1305
		 * If this is the unnamed statement and it has parameters, defer query
		 * planning until Bind.  Otherwise do it now.
1306 1307 1308 1309 1310
		 */
		if (!is_named && numParams > 0)
			plantree_list = NIL;
		else
			plantree_list = pg_plan_queries(querytree_list, NULL, true);
1311 1312 1313
	}
	else
	{
B
Bruce Momjian 已提交
1314
		/* Empty input string.	This is legal. */
1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357
		commandTag = NULL;
		querytree_list = NIL;
		plantree_list = NIL;
		param_list = NIL;
	}

	/* If we got a cancel signal in analysis or planning, quit */
	CHECK_FOR_INTERRUPTS();

	/*
	 * Store the query as a prepared statement.  See above comments.
	 */
	if (is_named)
	{
		StorePreparedStatement(stmt_name,
							   query_string,
							   commandTag,
							   querytree_list,
							   plantree_list,
							   param_list);
	}
	else
	{
		PreparedStatement *pstmt;

		pstmt = (PreparedStatement *) palloc0(sizeof(PreparedStatement));
		/* query_string needs to be copied into unnamed_stmt_context */
		pstmt->query_string = pstrdup(query_string);
		/* the rest is there already */
		pstmt->commandTag = commandTag;
		pstmt->query_list = querytree_list;
		pstmt->plan_list = plantree_list;
		pstmt->argtype_list = param_list;
		pstmt->context = unnamed_stmt_context;
		/* Now the unnamed statement is complete and valid */
		unnamed_stmt_pstmt = pstmt;
	}

	MemoryContextSwitchTo(oldcontext);

	QueryContext = NULL;

	/*
B
Bruce Momjian 已提交
1358 1359 1360
	 * We do NOT close the open transaction command here; that only happens
	 * when the client sends Sync.	Instead, do CommandCounterIncrement just
	 * in case something happened during parse/plan.
1361 1362 1363 1364 1365 1366
	 */
	CommandCounterIncrement();

	/*
	 * Send ParseComplete.
	 */
1367
	if (whereToSendOutput == DestRemote)
1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385
		pq_putemptymessage('1');

	if (save_log_statement_stats)
		ShowUsage("PARSE MESSAGE STATISTICS");

	debug_query_string = NULL;
}

/*
 * exec_bind_message
 *
 * Process a "Bind" message to create a portal from a prepared statement
 */
static void
exec_bind_message(StringInfo input_message)
{
	const char *portal_name;
	const char *stmt_name;
1386 1387
	int			numPFormats;
	int16	   *pformats = NULL;
1388
	int			numParams;
1389 1390 1391
	int			numRFormats;
	int16	   *rformats = NULL;
	int			i;
1392 1393 1394
	PreparedStatement *pstmt;
	Portal		portal;
	ParamListInfo params;
1395
	bool		isaborted = IsAbortedTransactionBlockState();
1396 1397 1398 1399 1400 1401

	pgstat_report_activity("<BIND>");

	set_ps_display("BIND");

	/*
B
Bruce Momjian 已提交
1402 1403 1404
	 * Start up a transaction command so we can call functions etc. (Note that
	 * this will normally change current memory context.) Nothing happens if
	 * we are already in one.
1405 1406 1407
	 */
	start_xact_command();

1408 1409 1410
	/* Switch back to message context */
	MemoryContextSwitchTo(MessageContext);

1411 1412 1413 1414
	/* Get the fixed part of the message */
	portal_name = pq_getmsgstring(input_message);
	stmt_name = pq_getmsgstring(input_message);

1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427
	/* Get the parameter format codes */
	numPFormats = pq_getmsgint(input_message, 2);
	if (numPFormats > 0)
	{
		pformats = (int16 *) palloc(numPFormats * sizeof(int16));
		for (i = 0; i < numPFormats; i++)
			pformats[i] = pq_getmsgint(input_message, 2);
	}

	/* Get the parameter value count */
	numParams = pq_getmsgint(input_message, 2);

	if (numPFormats > 1 && numPFormats != numParams)
1428 1429
		ereport(ERROR,
				(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
1430 1431
			errmsg("bind message has %d parameter formats but %d parameters",
				   numPFormats, numParams)));
1432 1433 1434 1435 1436 1437 1438 1439 1440

	/* Find prepared statement */
	if (stmt_name[0] != '\0')
		pstmt = FetchPreparedStatement(stmt_name, true);
	else
	{
		/* special-case the unnamed statement */
		pstmt = unnamed_stmt_pstmt;
		if (!pstmt)
1441 1442
			ereport(ERROR,
					(errcode(ERRCODE_UNDEFINED_PSTATEMENT),
B
Bruce Momjian 已提交
1443
					 errmsg("unnamed prepared statement does not exist")));
1444 1445
	}

1446
	if (numParams != list_length(pstmt->argtype_list))
1447 1448 1449
		ereport(ERROR,
				(errcode(ERRCODE_PROTOCOL_VIOLATION),
				 errmsg("bind message supplies %d parameters, but prepared statement \"%s\" requires %d",
B
Bruce Momjian 已提交
1450
				   numParams, stmt_name, list_length(pstmt->argtype_list))));
1451 1452

	/*
B
Bruce Momjian 已提交
1453 1454
	 * Create the portal.  Allow silent replacement of an existing portal only
	 * if the unnamed portal is specified.
1455 1456 1457 1458 1459 1460
	 */
	if (portal_name[0] == '\0')
		portal = CreatePortal(portal_name, true, true);
	else
		portal = CreatePortal(portal_name, false, false);

1461
	/* We need to output the parameter values someday */
1462 1463 1464 1465
	if (log_statement == LOGSTMT_ALL)
		ereport(LOG,
				(errmsg("statement: <BIND> %s", portal_name)));

1466 1467 1468
	/*
	 * Fetch parameters, if any, and store in the portal's memory context.
	 *
B
Bruce Momjian 已提交
1469 1470 1471
	 * In an aborted transaction, we can't risk calling user-defined functions,
	 * but we can't fail to Bind either, so bind all parameters to null
	 * values.
1472 1473 1474
	 */
	if (numParams > 0)
	{
1475
		ListCell   *l;
1476 1477 1478 1479 1480 1481 1482
		MemoryContext oldContext;

		oldContext = MemoryContextSwitchTo(PortalGetHeapMemory(portal));

		params = (ParamListInfo)
			palloc0((numParams + 1) * sizeof(ParamListInfoData));

1483
		i = 0;
1484 1485
		foreach(l, pstmt->argtype_list)
		{
1486
			Oid			ptype = lfirst_oid(l);
1487
			int32		plength;
1488 1489
			bool		isNull;

1490 1491 1492
			plength = pq_getmsgint(input_message, 4);
			isNull = (plength == -1);

1493 1494
			if (!isNull)
			{
1495
				const char *pvalue = pq_getmsgbytes(input_message, plength);
1496 1497

				if (isaborted)
1498 1499
				{
					/* We don't bother to check the format in this case */
1500
					isNull = true;
1501
				}
1502 1503
				else
				{
B
Bruce Momjian 已提交
1504
					int16		pformat;
1505
					StringInfoData pbuf;
B
Bruce Momjian 已提交
1506
					char		csave;
1507 1508 1509 1510 1511 1512

					if (numPFormats > 1)
						pformat = pformats[i];
					else if (numPFormats > 0)
						pformat = pformats[0];
					else
B
Bruce Momjian 已提交
1513
						pformat = 0;	/* default = text */
1514

1515
					/*
B
Bruce Momjian 已提交
1516 1517 1518 1519 1520 1521 1522
					 * Rather than copying data around, we just set up a phony
					 * StringInfo pointing to the correct portion of the
					 * message buffer.	We assume we can scribble on the
					 * message buffer so as to maintain the convention that
					 * StringInfos have a trailing null.  This is grotty but
					 * is a big win when dealing with very large parameter
					 * strings.
1523 1524 1525 1526 1527 1528 1529 1530 1531
					 */
					pbuf.data = (char *) pvalue;
					pbuf.maxlen = plength + 1;
					pbuf.len = plength;
					pbuf.cursor = 0;

					csave = pbuf.data[plength];
					pbuf.data[plength] = '\0';

1532 1533
					if (pformat == 0)
					{
1534 1535
						Oid			typinput;
						Oid			typioparam;
1536 1537
						char	   *pstring;

1538
						getTypeInputInfo(ptype, &typinput, &typioparam);
B
Bruce Momjian 已提交
1539

1540
						/*
B
Bruce Momjian 已提交
1541 1542
						 * We have to do encoding conversion before calling
						 * the typinput routine.
1543
						 */
1544
						pstring = pg_client_to_server(pbuf.data, plength);
1545
						params[i].value =
1546
							OidFunctionCall3(typinput,
1547
											 CStringGetDatum(pstring),
1548
											 ObjectIdGetDatum(typioparam),
1549 1550 1551 1552 1553 1554 1555
											 Int32GetDatum(-1));
						/* Free result of encoding conversion, if any */
						if (pstring != pbuf.data)
							pfree(pstring);
					}
					else if (pformat == 1)
					{
1556 1557
						Oid			typreceive;
						Oid			typioparam;
1558

B
Bruce Momjian 已提交
1559
						/*
B
Bruce Momjian 已提交
1560
						 * Call the parameter type's binary input converter
B
Bruce Momjian 已提交
1561
						 */
1562
						getTypeBinaryInputInfo(ptype, &typreceive, &typioparam);
1563 1564

						params[i].value =
1565
							OidFunctionCall3(typreceive,
1566
											 PointerGetDatum(&pbuf),
1567 1568
											 ObjectIdGetDatum(typioparam),
											 Int32GetDatum(-1));
1569 1570 1571

						/* Trouble if it didn't eat the whole buffer */
						if (pbuf.cursor != pbuf.len)
1572
							ereport(ERROR,
B
Bruce Momjian 已提交
1573 1574 1575
							 (errcode(ERRCODE_INVALID_BINARY_REPRESENTATION),
							  errmsg("incorrect binary data format in bind parameter %d",
									 i + 1)));
1576 1577 1578
					}
					else
					{
1579 1580 1581 1582
						ereport(ERROR,
								(errcode(ERRCODE_INVALID_PARAMETER_VALUE),
								 errmsg("unsupported format code: %d",
										pformat)));
1583
					}
1584 1585 1586

					/* Restore message buffer contents */
					pbuf.data[plength] = csave;
1587 1588
				}
			}
1589

1590 1591
			params[i].kind = PARAM_NUM;
			params[i].id = i + 1;
1592
			params[i].ptype = ptype;
1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604
			params[i].isnull = isNull;

			i++;
		}

		params[i].kind = PARAM_INVALID;

		MemoryContextSwitchTo(oldContext);
	}
	else
		params = NULL;

1605 1606 1607 1608 1609 1610 1611 1612 1613
	/* Get the result format codes */
	numRFormats = pq_getmsgint(input_message, 2);
	if (numRFormats > 0)
	{
		rformats = (int16 *) palloc(numRFormats * sizeof(int16));
		for (i = 0; i < numRFormats; i++)
			rformats[i] = pq_getmsgint(input_message, 2);
	}

1614 1615 1616
	pq_getmsgend(input_message);

	/*
B
Bruce Momjian 已提交
1617 1618
	 * If we didn't plan the query before, do it now.  This allows the planner
	 * to make use of the concrete parameter values we now have.
1619
	 *
B
Bruce Momjian 已提交
1620 1621
	 * This happens only for unnamed statements, and so switching into the
	 * statement context for planning is correct (see notes in
1622
	 * exec_parse_message).
1623
	 */
1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642
	if (pstmt->plan_list == NIL && pstmt->query_list != NIL &&
		!isaborted)
	{
		MemoryContext oldContext = MemoryContextSwitchTo(pstmt->context);

		pstmt->plan_list = pg_plan_queries(pstmt->query_list, params, true);
		MemoryContextSwitchTo(oldContext);
	}

	/*
	 * Define portal and start execution.
	 */
	PortalDefineQuery(portal,
					  pstmt->query_string,
					  pstmt->commandTag,
					  pstmt->query_list,
					  pstmt->plan_list,
					  pstmt->context);

1643
	PortalStart(portal, params, InvalidSnapshot);
1644

1645 1646 1647 1648 1649
	/*
	 * Apply the result format requests to the portal.
	 */
	PortalSetResultFormat(portal, numRFormats, rformats);

1650 1651 1652
	/*
	 * Send BindComplete.
	 */
1653
	if (whereToSendOutput == DestRemote)
1654 1655 1656 1657 1658 1659 1660 1661 1662
		pq_putemptymessage('2');
}

/*
 * exec_execute_message
 *
 * Process an "Execute" message for a portal
 */
static void
1663
exec_execute_message(const char *portal_name, long max_rows)
1664
{
B
Bruce Momjian 已提交
1665
	CommandDest dest;
1666
	DestReceiver *receiver;
1667 1668 1669 1670 1671
	Portal		portal;
	bool		is_trans_stmt = false;
	bool		is_trans_exit = false;
	bool		completed;
	char		completionTag[COMPLETION_TAG_BUFSIZE];
B
Bruce Momjian 已提交
1672 1673
	struct timeval start_t,
				stop_t;
1674 1675 1676
	bool		save_log_duration = log_duration;
	int			save_log_min_duration_statement = log_min_duration_statement;
	bool		save_log_statement_stats = log_statement_stats;
1677
	bool		execute_is_fetch = false;
1678 1679 1680

	/* Adjust destination to tell printtup.c what to do */
	dest = whereToSendOutput;
1681 1682
	if (dest == DestRemote)
		dest = DestRemoteExecute;
1683 1684 1685

	portal = GetPortalByName(portal_name);
	if (!PortalIsValid(portal))
1686 1687 1688
		ereport(ERROR,
				(errcode(ERRCODE_UNDEFINED_CURSOR),
				 errmsg("portal \"%s\" does not exist", portal_name)));
1689

1690
	/*
B
Bruce Momjian 已提交
1691 1692 1693 1694
	 * If we re-issue an Execute protocol request against an existing portal,
	 * then we are only fetching more rows rather than completely re-executing
	 * the query from the start. atStart is never reset for a v3 portal, so we
	 * are safe to use this check.
1695 1696 1697 1698
	 */
	if (!portal->atStart)
		execute_is_fetch = true;

1699
	/*
B
Bruce Momjian 已提交
1700 1701
	 * If the original query was a null string, just return
	 * EmptyQueryResponse.
1702 1703 1704 1705 1706 1707 1708 1709
	 */
	if (portal->commandTag == NULL)
	{
		Assert(portal->parseTrees == NIL);
		NullCommand(dest);
		return;
	}

1710 1711 1712 1713 1714 1715 1716
	/* Should we display the portal names here? */
	if (execute_is_fetch)
	{
		debug_query_string = "fetch message";
		pgstat_report_activity("<FETCH>");
	}
	else if (portal->sourceText)
1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728
	{
		debug_query_string = portal->sourceText;
		pgstat_report_activity(portal->sourceText);
	}
	else
	{
		debug_query_string = "execute message";
		pgstat_report_activity("<EXECUTE>");
	}

	set_ps_display(portal->commandTag);

1729 1730
	/*
	 * We use save_log_* so "SET log_duration = true"  and "SET
B
Bruce Momjian 已提交
1731 1732 1733
	 * log_min_duration_statement = true" don't report incorrect time because
	 * gettimeofday() wasn't called. Similarly, log_statement_stats has to be
	 * captured once.
1734 1735 1736 1737 1738 1739 1740 1741 1742 1743
	 */
	if (save_log_duration || save_log_min_duration_statement != -1)
		gettimeofday(&start_t, NULL);

	if (save_log_statement_stats)
		ResetUsage();

	if (log_statement == LOGSTMT_ALL)
		/* We have the portal, so output the source query. */
		ereport(LOG,
1744 1745
				(errmsg("statement: %sEXECUTE %s  [PREPARE:  %s]",
						(execute_is_fetch) ? "FETCH from " : "",
1746
						(*portal_name != '\0') ? portal_name : "<unnamed>",
1747 1748
						portal->sourceText ? portal->sourceText : "")));

1749 1750 1751
	BeginCommand(portal->commandTag, dest);

	/* Check for transaction-control commands */
1752
	if (list_length(portal->parseTrees) == 1)
1753
	{
1754
		Query	   *query = (Query *) linitial(portal->parseTrees);
1755 1756 1757 1758 1759 1760 1761 1762 1763

		if (query->commandType == CMD_UTILITY &&
			query->utilityStmt != NULL &&
			IsA(query->utilityStmt, TransactionStmt))
		{
			TransactionStmt *stmt = (TransactionStmt *) query->utilityStmt;

			is_trans_stmt = true;
			if (stmt->kind == TRANS_STMT_COMMIT ||
1764
				stmt->kind == TRANS_STMT_PREPARE ||
1765 1766
				stmt->kind == TRANS_STMT_ROLLBACK ||
				stmt->kind == TRANS_STMT_ROLLBACK_TO)
1767 1768 1769 1770
				is_trans_exit = true;
		}
	}

1771
	/*
B
Bruce Momjian 已提交
1772 1773
	 * Create dest receiver in MessageContext (we don't want it in transaction
	 * context, because that may get deleted if portal contains VACUUM).
1774 1775 1776
	 */
	receiver = CreateDestReceiver(dest, portal);

1777
	/*
B
Bruce Momjian 已提交
1778 1779
	 * Ensure we are in a transaction command (this should normally be the
	 * case already due to prior BIND).
1780 1781 1782 1783 1784 1785 1786 1787 1788 1789
	 */
	start_xact_command();

	/*
	 * If we are in aborted transaction state, the only portals we can
	 * actually run are those containing COMMIT or ROLLBACK commands.
	 */
	if (IsAbortedTransactionBlockState())
	{
		if (!is_trans_exit)
1790 1791 1792
			ereport(ERROR,
					(errcode(ERRCODE_IN_FAILED_SQL_TRANSACTION),
					 errmsg("current transaction is aborted, "
B
Bruce Momjian 已提交
1793
						"commands ignored until end of transaction block")));
1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806
	}

	/* Check for cancel signal before we start execution */
	CHECK_FOR_INTERRUPTS();

	/*
	 * Okay to run the portal.
	 */
	if (max_rows <= 0)
		max_rows = FETCH_ALL;

	completed = PortalRun(portal,
						  max_rows,
1807 1808
						  receiver,
						  receiver,
1809 1810
						  completionTag);

1811
	(*receiver->rDestroy) (receiver);
1812

1813 1814 1815 1816 1817
	if (completed)
	{
		if (is_trans_stmt)
		{
			/*
B
Bruce Momjian 已提交
1818
			 * If this was a transaction control statement, commit it.	We
B
Bruce Momjian 已提交
1819
			 * will start a new xact command for the next command (if any).
1820
			 */
1821
			finish_xact_command();
1822 1823 1824 1825
		}
		else
		{
			/*
B
Bruce Momjian 已提交
1826 1827
			 * We need a CommandCounterIncrement after every query, except
			 * those that start or end a transaction block.
1828 1829 1830 1831 1832 1833 1834 1835 1836 1837
			 */
			CommandCounterIncrement();
		}

		/* Send appropriate CommandComplete to client */
		EndCommand(completionTag, dest);
	}
	else
	{
		/* Portal run not complete, so send PortalSuspended */
1838
		if (whereToSendOutput == DestRemote)
1839 1840 1841
			pq_putemptymessage('s');
	}

1842
	/*
B
Bruce Momjian 已提交
1843 1844
	 * Combine processing here as we need to calculate the query duration in
	 * both instances.
1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856
	 */
	if (save_log_duration || save_log_min_duration_statement != -1)
	{
		long		usecs;

		gettimeofday(&stop_t, NULL);
		if (stop_t.tv_usec < start_t.tv_usec)
		{
			stop_t.tv_sec--;
			stop_t.tv_usec += 1000000;
		}
		usecs = (long) (stop_t.tv_sec - start_t.tv_sec) * 1000000 +
B
Bruce Momjian 已提交
1857
			(long) (stop_t.tv_usec - start_t.tv_usec);
1858 1859 1860 1861 1862

		/* Only print duration if we previously printed the statement. */
		if (log_statement == LOGSTMT_ALL && save_log_duration)
			ereport(LOG,
					(errmsg("duration: %ld.%03ld ms",
B
Bruce Momjian 已提交
1863 1864 1865
							(long) ((stop_t.tv_sec - start_t.tv_sec) * 1000 +
								  (stop_t.tv_usec - start_t.tv_usec) / 1000),
						 (long) (stop_t.tv_usec - start_t.tv_usec) % 1000)));
1866 1867

		/*
B
Bruce Momjian 已提交
1868 1869
		 * Output a duration_statement to the log if the query has exceeded
		 * the min duration, or if we are to print all durations.
1870 1871 1872 1873 1874
		 */
		if (save_log_min_duration_statement == 0 ||
			(save_log_min_duration_statement > 0 &&
			 usecs >= save_log_min_duration_statement * 1000))
			ereport(LOG,
1875
					(errmsg("duration: %ld.%03ld ms  statement: %sEXECUTE %s  [PREPARE:  %s]",
B
Bruce Momjian 已提交
1876 1877 1878
							(long) ((stop_t.tv_sec - start_t.tv_sec) * 1000 +
								  (stop_t.tv_usec - start_t.tv_usec) / 1000),
							(long) (stop_t.tv_usec - start_t.tv_usec) % 1000,
1879
							(execute_is_fetch) ? "FETCH from " : "",
B
Bruce Momjian 已提交
1880
						  (*portal_name != '\0') ? portal_name : "<unnamed>",
1881 1882 1883 1884 1885 1886
							portal->sourceText ? portal->sourceText : "")));
	}

	if (save_log_statement_stats)
		ShowUsage("QUERY STATISTICS");

1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898
	debug_query_string = NULL;
}

/*
 * exec_describe_statement_message
 *
 * Process a "Describe" message for a prepared statement
 */
static void
exec_describe_statement_message(const char *stmt_name)
{
	PreparedStatement *pstmt;
1899
	TupleDesc	tupdesc;
1900
	ListCell   *l;
1901 1902 1903 1904 1905 1906 1907 1908 1909 1910
	StringInfoData buf;

	/* Find prepared statement */
	if (stmt_name[0] != '\0')
		pstmt = FetchPreparedStatement(stmt_name, true);
	else
	{
		/* special-case the unnamed statement */
		pstmt = unnamed_stmt_pstmt;
		if (!pstmt)
1911 1912
			ereport(ERROR,
					(errcode(ERRCODE_UNDEFINED_PSTATEMENT),
B
Bruce Momjian 已提交
1913
					 errmsg("unnamed prepared statement does not exist")));
1914 1915
	}

1916
	if (whereToSendOutput != DestRemote)
1917 1918
		return;					/* can't actually do anything... */

1919 1920 1921
	/*
	 * First describe the parameters...
	 */
B
Bruce Momjian 已提交
1922
	pq_beginmessage(&buf, 't'); /* parameter description message type */
1923
	pq_sendint(&buf, list_length(pstmt->argtype_list), 2);
1924 1925 1926

	foreach(l, pstmt->argtype_list)
	{
1927
		Oid			ptype = lfirst_oid(l);
1928 1929 1930 1931

		pq_sendint(&buf, (int) ptype, 4);
	}
	pq_endmessage(&buf);
1932 1933 1934 1935 1936 1937

	/*
	 * Next send RowDescription or NoData to describe the result...
	 */
	tupdesc = FetchPreparedStatementResultDesc(pstmt);
	if (tupdesc)
1938 1939 1940
		SendRowDescriptionMessage(tupdesc,
								  FetchPreparedStatementTargetList(pstmt),
								  NULL);
1941 1942 1943
	else
		pq_putemptymessage('n');	/* NoData */

1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957
}

/*
 * exec_describe_portal_message
 *
 * Process a "Describe" message for a portal
 */
static void
exec_describe_portal_message(const char *portal_name)
{
	Portal		portal;

	portal = GetPortalByName(portal_name);
	if (!PortalIsValid(portal))
1958 1959 1960
		ereport(ERROR,
				(errcode(ERRCODE_UNDEFINED_CURSOR),
				 errmsg("portal \"%s\" does not exist", portal_name)));
1961

1962
	if (whereToSendOutput != DestRemote)
1963 1964 1965
		return;					/* can't actually do anything... */

	if (portal->tupDesc)
1966 1967
		SendRowDescriptionMessage(portal->tupDesc,
								  FetchPortalTargetList(portal),
1968
								  portal->formats);
1969 1970 1971 1972 1973
	else
		pq_putemptymessage('n');	/* NoData */
}


1974 1975 1976 1977 1978 1979
/*
 * Convenience routines for starting/committing a single command.
 */
static void
start_xact_command(void)
{
1980 1981
	if (!xact_started)
	{
1982 1983
		ereport(DEBUG3,
				(errmsg_internal("StartTransactionCommand")));
1984
		StartTransactionCommand();
1985 1986 1987 1988

		/* Set statement timeout running, if any */
		if (StatementTimeout > 0)
			enable_sig_alarm(StatementTimeout, true);
1989 1990
		else
			cancel_from_timeout = false;
B
Bruce Momjian 已提交
1991

1992 1993
		xact_started = true;
	}
1994 1995 1996
}

static void
1997
finish_xact_command(void)
1998
{
1999 2000 2001 2002
	if (xact_started)
	{
		/* Cancel any active statement timeout before committing */
		disable_sig_alarm(true);
2003

2004
		/* Now commit the command */
2005 2006
		ereport(DEBUG3,
				(errmsg_internal("CommitTransactionCommand")));
2007

2008
		CommitTransactionCommand();
2009

2010 2011 2012 2013 2014 2015
#ifdef MEMORY_CONTEXT_CHECKING
		/* Check all memory contexts that weren't freed during commit */
		/* (those that were, were checked before being deleted) */
		MemoryContextCheck(TopMemoryContext);
#endif

2016
#ifdef SHOW_MEMORY_STATS
2017
		/* Print mem stats after each commit for leak tracking */
2018 2019
		if (ShowStats)
			MemoryContextStats(TopMemoryContext);
2020
#endif
2021 2022 2023

		xact_started = false;
	}
2024 2025 2026
}


2027
/* --------------------------------
2028
 *		signal handler routines used in PostgresMain()
2029 2030 2031
 * --------------------------------
 */

2032
/*
T
Tom Lane 已提交
2033
 * quickdie() occurs when signalled SIGQUIT by the postmaster.
2034 2035 2036 2037
 *
 * Some backend has bought the farm,
 * so we need to stop what we're doing and exit.
 */
T
Tom Lane 已提交
2038
void
2039
quickdie(SIGNAL_ARGS)
2040
{
2041
	PG_SETMASK(&BlockSig);
B
Bruce Momjian 已提交
2042

2043
	/*
B
Bruce Momjian 已提交
2044 2045
	 * Ideally this should be ereport(FATAL), but then we'd not get control
	 * back...
2046 2047 2048
	 */
	ereport(WARNING,
			(errcode(ERRCODE_CRASH_SHUTDOWN),
B
Bruce Momjian 已提交
2049
			 errmsg("terminating connection because of crash of another server process"),
B
Bruce Momjian 已提交
2050 2051 2052 2053
	errdetail("The postmaster has commanded this server process to roll back"
			  " the current transaction and exit, because another"
			  " server process exited abnormally and possibly corrupted"
			  " shared memory."),
2054
			 errhint("In a moment you should be able to reconnect to the"
2055
					 " database and repeat your command.")));
B
Bruce Momjian 已提交
2056

2057
	/*
2058
	 * DO NOT proc_exit() -- we're here because shared memory may be
B
Bruce Momjian 已提交
2059 2060
	 * corrupted, so we don't want to try to clean up our transaction. Just
	 * nail the windows shut and get out of town.
2061
	 *
B
Bruce Momjian 已提交
2062 2063 2064 2065
	 * Note we do exit(1) not exit(0).	This is to force the postmaster into a
	 * system reset cycle if some idiot DBA sends a manual SIGQUIT to a random
	 * backend.  This is necessary precisely because we don't clean up our
	 * shared memory state.
2066
	 */
2067
	exit(1);
2068 2069
}

2070
/*
2071 2072
 * Shutdown signal from postmaster: abort transaction and exit
 * at soonest convenient time
2073
 */
2074
void
2075
die(SIGNAL_ARGS)
2076
{
2077 2078 2079
	int			save_errno = errno;

	/* Don't joggle the elbow of proc_exit */
B
Bruce Momjian 已提交
2080
	if (!proc_exit_inprogress)
2081
	{
2082
		InterruptPending = true;
2083
		ProcDiePending = true;
B
Bruce Momjian 已提交
2084

2085
		/*
B
Bruce Momjian 已提交
2086 2087
		 * If it's safe to interrupt, and we're waiting for input or a lock,
		 * service the interrupt immediately
2088
		 */
2089 2090
		if (ImmediateInterruptOK && InterruptHoldoffCount == 0 &&
			CritSectionCount == 0)
2091
		{
2092 2093 2094
			/* bump holdoff count to make ProcessInterrupts() a no-op */
			/* until we are done getting ready for it */
			InterruptHoldoffCount++;
2095
			DisableNotifyInterrupt();
2096
			DisableCatchupInterrupt();
2097
			/* Make sure CheckDeadLock won't run while shutting down... */
2098
			LockWaitCancel();
2099
			InterruptHoldoffCount--;
2100 2101
			ProcessInterrupts();
		}
2102
	}
2103 2104

	errno = save_errno;
2105 2106
}

2107
/*
2108
 * Timeout or shutdown signal from postmaster during client authentication.
2109
 * Simply exit(0).
2110 2111 2112
 *
 * XXX: possible future improvement: try to send a message indicating
 * why we are disconnecting.  Problem is to be sure we don't block while
2113
 * doing so, nor mess up the authentication message exchange.
2114 2115 2116 2117 2118 2119 2120
 */
void
authdie(SIGNAL_ARGS)
{
	exit(0);
}

2121
/*
2122 2123
 * Query-cancel signal from postmaster: abort current transaction
 * at soonest convenient time
2124
 */
2125
void
2126
StatementCancelHandler(SIGNAL_ARGS)
2127
{
2128 2129
	int			save_errno = errno;

B
Bruce Momjian 已提交
2130
	/*
2131
	 * Don't joggle the elbow of proc_exit
B
Bruce Momjian 已提交
2132
	 */
2133
	if (!proc_exit_inprogress)
2134
	{
2135 2136
		InterruptPending = true;
		QueryCancelPending = true;
B
Bruce Momjian 已提交
2137

2138
		/*
B
Bruce Momjian 已提交
2139 2140 2141
		 * If it's safe to interrupt, and we're waiting for a lock, service
		 * the interrupt immediately.  No point in interrupting if we're
		 * waiting for input, however.
2142
		 */
2143
		if (ImmediateInterruptOK && InterruptHoldoffCount == 0 &&
2144
			CritSectionCount == 0)
2145
		{
2146 2147 2148 2149 2150 2151
			/* bump holdoff count to make ProcessInterrupts() a no-op */
			/* until we are done getting ready for it */
			InterruptHoldoffCount++;
			if (LockWaitCancel())
			{
				DisableNotifyInterrupt();
2152
				DisableCatchupInterrupt();
T
Tom Lane 已提交
2153
				InterruptHoldoffCount--;
2154 2155 2156 2157
				ProcessInterrupts();
			}
			else
				InterruptHoldoffCount--;
2158
		}
2159 2160
	}

2161
	errno = save_errno;
2162 2163
}

2164
/* signal handler for floating point exception */
2165
void
2166 2167
FloatExceptionHandler(SIGNAL_ARGS)
{
2168 2169 2170
	ereport(ERROR,
			(errcode(ERRCODE_FLOATING_POINT_EXCEPTION),
			 errmsg("floating-point exception"),
B
Bruce Momjian 已提交
2171 2172 2173
			 errdetail("An invalid floating-point operation was signaled. "
					   "This probably means an out-of-range result or an "
					   "invalid operation, such as division by zero.")));
2174 2175
}

2176
/* SIGHUP: set flag to re-read config file at next convenient time */
2177
static void
2178
SigHupHandler(SIGNAL_ARGS)
2179
{
2180
	got_SIGHUP = true;
2181 2182
}

2183

2184 2185 2186 2187 2188 2189 2190 2191 2192 2193
/*
 * ProcessInterrupts: out-of-line portion of CHECK_FOR_INTERRUPTS() macro
 *
 * If an interrupt condition is pending, and it's safe to service it,
 * then clear the flag and accept the interrupt.  Called only when
 * InterruptPending is true.
 */
void
ProcessInterrupts(void)
{
2194 2195
	/* OK to accept interrupt now? */
	if (InterruptHoldoffCount != 0 || CritSectionCount != 0)
2196 2197 2198 2199 2200
		return;
	InterruptPending = false;
	if (ProcDiePending)
	{
		ProcDiePending = false;
B
Bruce Momjian 已提交
2201 2202
		QueryCancelPending = false;		/* ProcDie trumps QueryCancel */
		ImmediateInterruptOK = false;	/* not idle anymore */
2203
		DisableNotifyInterrupt();
2204
		DisableCatchupInterrupt();
2205 2206
		ereport(FATAL,
				(errcode(ERRCODE_ADMIN_SHUTDOWN),
B
Bruce Momjian 已提交
2207
			 errmsg("terminating connection due to administrator command")));
2208 2209 2210 2211
	}
	if (QueryCancelPending)
	{
		QueryCancelPending = false;
B
Bruce Momjian 已提交
2212
		ImmediateInterruptOK = false;	/* not idle anymore */
2213
		DisableNotifyInterrupt();
2214
		DisableCatchupInterrupt();
2215 2216 2217 2218 2219 2220 2221 2222
		if (cancel_from_timeout)
			ereport(ERROR,
					(errcode(ERRCODE_QUERY_CANCELED),
					 errmsg("canceling statement due to statement timeout")));
		else
			ereport(ERROR,
					(errcode(ERRCODE_QUERY_CANCELED),
					 errmsg("canceling statement due to user request")));
2223 2224 2225 2226
	}
	/* If we get here, do nothing (probably, QueryCancelPending was reset) */
}

2227

2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240
/*
 * check_stack_depth: check for excessively deep recursion
 *
 * This should be called someplace in any recursive routine that might possibly
 * recurse deep enough to overflow the stack.  Most Unixen treat stack
 * overflow as an unrecoverable SIGSEGV, so we want to error out ourselves
 * before hitting the hardware limit.  Unfortunately we have no direct way
 * to detect the hardware limit, so we have to rely on the admin to set a
 * GUC variable for it ...
 */
void
check_stack_depth(void)
{
B
Bruce Momjian 已提交
2241 2242
	char		stack_top_loc;
	int			stack_depth;
2243 2244 2245 2246

	/*
	 * Compute distance from PostgresMain's local variables to my own
	 *
B
Bruce Momjian 已提交
2247 2248 2249
	 * Note: in theory stack_depth should be ptrdiff_t or some such, but since
	 * the whole point of this code is to bound the value to something much
	 * less than integer-sized, int should work fine.
2250 2251
	 */
	stack_depth = (int) (stack_base_ptr - &stack_top_loc);
B
Bruce Momjian 已提交
2252

2253
	/*
B
Bruce Momjian 已提交
2254
	 * Take abs value, since stacks grow up on some machines, down on others
2255 2256 2257
	 */
	if (stack_depth < 0)
		stack_depth = -stack_depth;
B
Bruce Momjian 已提交
2258

2259 2260 2261
	/*
	 * Trouble?
	 *
B
Bruce Momjian 已提交
2262 2263 2264
	 * The test on stack_base_ptr prevents us from erroring out if called during
	 * process setup or in a non-backend process.  Logically it should be done
	 * first, but putting it here avoids wasting cycles during normal cases.
2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286
	 */
	if (stack_depth > max_stack_depth_bytes &&
		stack_base_ptr != NULL)
	{
		ereport(ERROR,
				(errcode(ERRCODE_STATEMENT_TOO_COMPLEX),
				 errmsg("stack depth limit exceeded"),
				 errhint("Increase the configuration parameter \"max_stack_depth\".")));
	}
}

/* GUC assign hook to update max_stack_depth_bytes from max_stack_depth */
bool
assign_max_stack_depth(int newval, bool doit, GucSource source)
{
	/* Range check was already handled by guc.c */
	if (doit)
		max_stack_depth_bytes = newval * 1024;
	return true;
}


2287
static void
2288
usage(const char *progname)
2289
{
2290
	printf(_("%s is the PostgreSQL stand-alone backend.  It is not\nintended to be used by normal users.\n\n"), progname);
2291

2292 2293
	printf(_("Usage:\n  %s [OPTION]... [DBNAME]\n\n"), progname);
	printf(_("Options:\n"));
M
 
Marc G. Fournier 已提交
2294
#ifdef USE_ASSERT_CHECKING
2295
	printf(_("  -A 1|0          enable/disable run-time assert checking\n"));
M
 
Marc G. Fournier 已提交
2296
#endif
2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318
	printf(_("  -B NBUFFERS     number of shared buffers\n"));
	printf(_("  -c NAME=VALUE   set run-time parameter\n"));
	printf(_("  -d 0-5          debugging level (0 is off)\n"));
	printf(_("  -D DATADIR      database directory\n"));
	printf(_("  -e              use European date input format (DMY)\n"));
	printf(_("  -E              echo query before execution\n"));
	printf(_("  -F              turn fsync off\n"));
	printf(_("  -N              do not use newline as interactive query delimiter\n"));
	printf(_("  -o FILENAME     send stdout and stderr to given file\n"));
	printf(_("  -P              disable system indexes\n"));
	printf(_("  -s              show statistics after each query\n"));
	printf(_("  -S WORK-MEM     set amount of memory for sorts (in kB)\n"));
	printf(_("  --describe-config  describe configuration parameters, then exit\n"));
	printf(_("  --help          show this help, then exit\n"));
	printf(_("  --version       output version information, then exit\n"));
	printf(_("\nDeveloper options:\n"));
	printf(_("  -f s|i|n|m|h    forbid use of some plan types\n"));
	printf(_("  -i              do not execute queries\n"));
	printf(_("  -O              allow system table structure changes\n"));
	printf(_("  -t pa|pl|ex     show timings after each query\n"));
	printf(_("  -W NUM          wait NUM seconds to allow attach from a debugger\n"));
	printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n"));
2319 2320
}

2321

2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356
/*
 * set_debug_options --- apply "-d N" command line option
 *
 * -d is not quite the same as setting log_min_messages because it enables
 * other output options.
 */
void
set_debug_options(int debug_flag, GucContext context, GucSource source)
{
	if (debug_flag > 0)
	{
		char		debugstr[64];

		sprintf(debugstr, "debug%d", debug_flag);
		SetConfigOption("log_min_messages", debugstr, context, source);
	}
	else
		SetConfigOption("log_min_messages", "notice", context, source);

	if (debug_flag >= 1 && context == PGC_POSTMASTER)
	{
		SetConfigOption("log_connections", "true", context, source);
		SetConfigOption("log_disconnections", "true", context, source);
	}
	if (debug_flag >= 2)
		SetConfigOption("log_statement", "all", context, source);
	if (debug_flag >= 3)
		SetConfigOption("debug_print_parse", "true", context, source);
	if (debug_flag >= 4)
		SetConfigOption("debug_print_plan", "true", context, source);
	if (debug_flag >= 5)
		SetConfigOption("debug_print_rewritten", "true", context, source);
}


2357
/* ----------------------------------------------------------------
2358
 * PostgresMain
B
Bruce Momjian 已提交
2359
 *	   postgres main loop -- all backends, interactive or otherwise start here
2360
 *
2361 2362 2363 2364
 * argc/argv are the command line arguments to be used.  (When being forked
 * by the postmaster, these are not the original argv array of the process.)
 * username is the (possibly authenticated) PostgreSQL user name to be used
 * for the session.
2365 2366 2367
 * ----------------------------------------------------------------
 */
int
2368
PostgresMain(int argc, char *argv[], const char *username)
2369
{
2370
	int			flag;
2371
	const char *dbname = NULL;
2372
	char	   *userDoption = NULL;
2373
	bool		secure;
2374
	int			errs = 0;
B
Bruce Momjian 已提交
2375 2376
	int			debug_flag = -1;	/* -1 means not given */
	List	   *guc_names = NIL;	/* for SUSET options */
2377 2378
	List	   *guc_values = NIL;
	GucContext	ctx;
2379
	GucSource	gucsource;
2380
	bool		am_superuser;
2381
	char	   *tmp;
2382
	int			firstchar;
2383
	char		stack_base;
B
Bruce Momjian 已提交
2384
	StringInfoData input_message;
2385
	sigjmp_buf	local_sigjmp_buf;
2386
	volatile bool send_rfq = true;
B
Bruce Momjian 已提交
2387

2388 2389 2390 2391
#define PendingConfigOption(name,val) \
	(guc_names = lappend(guc_names, pstrdup(name)), \
	 guc_values = lappend(guc_values, pstrdup(val)))

2392
	/*
B
Bruce Momjian 已提交
2393 2394
	 * Catch standard options before doing much else.  This even works on
	 * systems without getopt_long.
2395 2396 2397
	 */
	if (!IsUnderPostmaster && argc > 1)
	{
B
Bruce Momjian 已提交
2398
		if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
2399 2400 2401 2402
		{
			usage(argv[0]);
			exit(0);
		}
B
Bruce Momjian 已提交
2403
		if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
2404
		{
2405
			puts(PG_VERSIONSTR);
2406 2407
			exit(0);
		}
B
Bruce Momjian 已提交
2408
	}
2409

2410 2411 2412 2413 2414 2415
	/*
	 * initialize globals (already done if under postmaster, but not if
	 * standalone; cheap enough to do over)
	 */
	MyProcPid = getpid();

2416 2417 2418 2419 2420
	/*
	 * Fire up essential subsystems: error and memory management
	 *
	 * If we are running under the postmaster, this is done already.
	 */
2421
	if (!IsUnderPostmaster)
2422 2423
		MemoryContextInit();

2424 2425
	set_ps_display("startup");

2426 2427
	SetProcessingMode(InitProcessing);

2428 2429 2430
	/* Set up reference point for stack depth checking */
	stack_base_ptr = &stack_base;

2431 2432 2433 2434 2435 2436 2437
	/* Compute paths, if we didn't inherit them from postmaster */
	if (my_exec_path[0] == '\0')
	{
		if (find_my_exec(argv[0], my_exec_path) < 0)
			elog(FATAL, "%s: could not locate my own executable path",
				 argv[0]);
	}
B
Bruce Momjian 已提交
2438

B
Bruce Momjian 已提交
2439 2440
	if (pkglib_path[0] == '\0')
		get_pkglib_path(my_exec_path, pkglib_path);
2441

2442
	/*
2443
	 * Set default values for command-line options.
2444
	 */
2445
	EchoQuery = false;
2446

2447
	if (!IsUnderPostmaster)
2448
		InitializeGUCOptions();
2449

2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466
	/* ----------------
	 *	parse command line arguments
	 *
	 *	There are now two styles of command line layout for the backend:
	 *
	 *	For interactive use (not started from postmaster) the format is
	 *		postgres [switches] [databasename]
	 *	If the databasename is omitted it is taken to be the user name.
	 *
	 *	When started from the postmaster, the format is
	 *		postgres [secure switches] -p databasename [insecure switches]
	 *	Switches appearing after -p came from the client (via "options"
	 *	field of connection request).  For security reasons we restrict
	 *	what these switches can do.
	 * ----------------
	 */

2467 2468
	/* all options are allowed until '-p' */
	secure = true;
2469
	ctx = PGC_POSTMASTER;
2470
	gucsource = PGC_S_ARGV;		/* initial switches came from command line */
2471

2472
	while ((flag = getopt(argc, argv, "A:B:c:D:d:Eef:FiNOPo:p:S:st:v:W:-:")) != -1)
2473
	{
2474 2475
		switch (flag)
		{
M
 
Marc G. Fournier 已提交
2476 2477
			case 'A':
#ifdef USE_ASSERT_CHECKING
2478
				SetConfigOption("debug_assertions", optarg, ctx, gucsource);
M
 
Marc G. Fournier 已提交
2479
#else
2480 2481 2482
				ereport(WARNING,
						(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
						 errmsg("assert checking is not compiled in")));
M
 
Marc G. Fournier 已提交
2483 2484
#endif
				break;
2485

2486
			case 'B':
2487 2488 2489

				/*
				 * specify the size of buffer pool
2490
				 */
2491
				SetConfigOption("shared_buffers", optarg, ctx, gucsource);
2492
				break;
2493

2494
			case 'D':			/* PGDATA or config directory */
2495
				if (secure)
2496
					userDoption = optarg;
M
 
Marc G. Fournier 已提交
2497
				break;
2498

2499
			case 'd':			/* debug level */
2500
				debug_flag = atoi(optarg);
2501
				break;
2502 2503

			case 'E':
2504 2505 2506

				/*
				 * E - echo the query the user entered
2507
				 */
2508
				EchoQuery = true;
2509
				break;
2510 2511

			case 'e':
2512 2513

				/*
2514
				 * Use European date input format (DMY)
2515
				 */
2516
				SetConfigOption("datestyle", "euro", ctx, gucsource);
2517
				break;
2518 2519

			case 'F':
2520 2521 2522

				/*
				 * turn off fsync
2523
				 */
2524
				SetConfigOption("fsync", "false", ctx, gucsource);
2525
				break;
2526 2527

			case 'f':
2528 2529 2530

				/*
				 * f - forbid generation of certain plans
2531
				 */
2532
				tmp = NULL;
2533 2534 2535
				switch (optarg[0])
				{
					case 's':	/* seqscan */
2536
						tmp = "enable_seqscan";
2537 2538
						break;
					case 'i':	/* indexscan */
2539
						tmp = "enable_indexscan";
2540
						break;
2541 2542 2543
					case 'b':	/* bitmapscan */
						tmp = "enable_bitmapscan";
						break;
2544
					case 't':	/* tidscan */
2545
						tmp = "enable_tidscan";
2546 2547
						break;
					case 'n':	/* nestloop */
2548
						tmp = "enable_nestloop";
2549 2550
						break;
					case 'm':	/* mergejoin */
2551
						tmp = "enable_mergejoin";
2552 2553
						break;
					case 'h':	/* hashjoin */
2554
						tmp = "enable_hashjoin";
2555 2556 2557 2558
						break;
					default:
						errs++;
				}
2559
				if (tmp)
2560
					SetConfigOption(tmp, "false", ctx, gucsource);
2561 2562
				break;

2563
			case 'N':
2564 2565 2566

				/*
				 * N - Don't use newline as a query delimiter
2567 2568 2569
				 */
				UseNewLine = 0;
				break;
2570

2571
			case 'O':
2572 2573 2574

				/*
				 * allow system table structure modifications
2575
				 */
2576 2577
				if (secure)		/* XXX safe to allow from client??? */
					allowSystemTableMods = true;
2578 2579
				break;

H
Hiroshi Inoue 已提交
2580
			case 'P':
2581 2582 2583

				/*
				 * ignore system indexes
2584
				 *
B
Bruce Momjian 已提交
2585 2586 2587
				 * As of PG 7.4 this is safe to allow from the client, since it
				 * only disables reading the system indexes, not writing them.
				 * Worst case consequence is slowness.
H
Hiroshi Inoue 已提交
2588
				 */
2589
				IgnoreSystemIndexes(true);
H
Hiroshi Inoue 已提交
2590 2591
				break;

T
Tom Lane 已提交
2592
			case 'o':
2593 2594 2595

				/*
				 * o - send output (stdout and stderr) to the given file
T
Tom Lane 已提交
2596
				 */
2597 2598
				if (secure)
					StrNCpy(OutputFileName, optarg, MAXPGPATH);
T
Tom Lane 已提交
2599 2600
				break;

2601
			case 'p':
B
Bruce Momjian 已提交
2602

2603 2604 2605
				/*
				 * p - special flag passed if backend was forked by a
				 * postmaster.
2606
				 */
2607 2608
				if (secure)
				{
2609
					dbname = strdup(optarg);
2610

B
Bruce Momjian 已提交
2611
					secure = false;		/* subsequent switches are NOT secure */
2612
					ctx = PGC_BACKEND;
2613
					gucsource = PGC_S_CLIENT;
2614
				}
2615
				break;
2616

2617
			case 'S':
2618 2619 2620

				/*
				 * S - amount of sort memory to use in 1k bytes
2621
				 */
2622
				SetConfigOption("work_mem", optarg, ctx, gucsource);
2623
				break;
2624 2625

			case 's':
2626 2627 2628

				/*
				 * s - report usage statistics (timings) after each query
2629
				 *
B
Bruce Momjian 已提交
2630 2631
				 * Since log options are SUSET, we need to postpone unless still
				 * in secure context
2632
				 */
2633 2634 2635 2636 2637
				if (ctx == PGC_BACKEND)
					PendingConfigOption("log_statement_stats", "true");
				else
					SetConfigOption("log_statement_stats", "true",
									ctx, gucsource);
M
 
Marc G. Fournier 已提交
2638 2639
				break;

2640
			case 't':
2641
				/* ---------------
2642 2643 2644 2645 2646 2647 2648 2649 2650
				 *	tell postgres to report usage statistics (timings) for
				 *	each query
				 *
				 *	-tpa[rser] = print stats for parser time of each query
				 *	-tpl[anner] = print stats for planner time of each query
				 *	-te[xecutor] = print stats for executor time of each query
				 *	caution: -s can not be used together with -t.
				 * ----------------
				 */
2651
				tmp = NULL;
2652 2653 2654 2655
				switch (optarg[0])
				{
					case 'p':
						if (optarg[1] == 'a')
B
Rename:  
Bruce Momjian 已提交
2656
							tmp = "log_parser_stats";
2657
						else if (optarg[1] == 'l')
B
Rename:  
Bruce Momjian 已提交
2658
							tmp = "log_planner_stats";
2659 2660 2661 2662
						else
							errs++;
						break;
					case 'e':
2663
						tmp = "log_executor_stats";
2664 2665 2666 2667 2668
						break;
					default:
						errs++;
						break;
				}
2669
				if (tmp)
2670 2671 2672 2673 2674 2675
				{
					if (ctx == PGC_BACKEND)
						PendingConfigOption(tmp, "true");
					else
						SetConfigOption(tmp, "true", ctx, gucsource);
				}
2676 2677
				break;

2678
			case 'v':
2679 2680
				if (secure)
					FrontendProtocol = (ProtocolVersion) atoi(optarg);
2681 2682
				break;

M
 
Marc G. Fournier 已提交
2683
			case 'W':
2684 2685 2686

				/*
				 * wait N seconds to allow attach from a debugger
M
 
Marc G. Fournier 已提交
2687
				 */
B
Bruce Momjian 已提交
2688
				pg_usleep(atoi(optarg) * 1000000L);
M
 
Marc G. Fournier 已提交
2689 2690
				break;

2691
			case 'c':
2692
			case '-':
2693
				{
B
Bruce Momjian 已提交
2694 2695
					char	   *name,
							   *value;
2696

B
Bruce Momjian 已提交
2697 2698 2699 2700
					ParseLongOption(optarg, &name, &value);
					if (!value)
					{
						if (flag == '-')
2701 2702 2703 2704
							ereport(ERROR,
									(errcode(ERRCODE_SYNTAX_ERROR),
									 errmsg("--%s requires a value",
											optarg)));
B
Bruce Momjian 已提交
2705
						else
2706 2707 2708 2709
							ereport(ERROR,
									(errcode(ERRCODE_SYNTAX_ERROR),
									 errmsg("-c %s requires a value",
											optarg)));
B
Bruce Momjian 已提交
2710 2711
					}

2712
					/*
B
Bruce Momjian 已提交
2713 2714
					 * If a SUSET option, must postpone evaluation, unless we
					 * are still reading secure switches.
2715 2716 2717 2718 2719
					 */
					if (ctx == PGC_BACKEND && IsSuperuserConfigOption(name))
						PendingConfigOption(name, value);
					else
						SetConfigOption(name, value, ctx, gucsource);
B
Bruce Momjian 已提交
2720 2721 2722 2723 2724
					free(name);
					if (value)
						free(value);
					break;
				}
2725

2726 2727
			default:
				errs++;
T
Tom Lane 已提交
2728
				break;
2729
		}
2730 2731
	}

2732
	/*
B
Bruce Momjian 已提交
2733 2734
	 * Process any additional GUC variable settings passed in startup packet.
	 * These are handled exactly like command-line variables.
2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757
	 */
	if (MyProcPort != NULL)
	{
		ListCell   *gucopts = list_head(MyProcPort->guc_options);

		while (gucopts)
		{
			char	   *name;
			char	   *value;

			name = lfirst(gucopts);
			gucopts = lnext(gucopts);

			value = lfirst(gucopts);
			gucopts = lnext(gucopts);

			if (IsSuperuserConfigOption(name))
				PendingConfigOption(name, value);
			else
				SetConfigOption(name, value, PGC_BACKEND, PGC_S_CLIENT);
		}
	}

2758 2759
	/* Acquire configuration parameters, unless inherited from postmaster */
	if (!IsUnderPostmaster)
2760
	{
2761 2762
		if (!SelectConfigFiles(userDoption, argv[0]))
			proc_exit(1);
2763 2764
		/* If timezone is not set, determine what the OS uses */
		pg_timezone_initialize();
2765
	}
2766

2767
	/*
2768
	 * Set up signal handlers and masks.
2769
	 *
B
Bruce Momjian 已提交
2770 2771 2772
	 * Note that postmaster blocked all signals before forking child process, so
	 * there is no race condition whereby we might receive a signal before we
	 * have set up the handler.
T
Tom Lane 已提交
2773
	 *
B
Bruce Momjian 已提交
2774 2775 2776 2777 2778 2779
	 * Also note: it's best not to use any signals that are SIG_IGNored in the
	 * postmaster.	If such a signal arrives before we are able to change the
	 * handler to non-SIG_IGN, it'll get dropped.  Instead, make a dummy
	 * handler in the postmaster to reserve the signal. (Of course, this isn't
	 * an issue for signals that are locally generated, such as SIGALRM and
	 * SIGPIPE.)
2780
	 */
2781
	pqsignal(SIGHUP, SigHupHandler);	/* set flag to read config file */
B
Bruce Momjian 已提交
2782
	pqsignal(SIGINT, StatementCancelHandler);	/* cancel current query */
2783
	pqsignal(SIGTERM, die);		/* cancel current query and exit */
2784
	pqsignal(SIGQUIT, quickdie);	/* hard crash time */
B
Bruce Momjian 已提交
2785
	pqsignal(SIGALRM, handle_sig_alarm);		/* timeout conditions */
2786 2787 2788 2789

	/*
	 * Ignore failure to write to frontend. Note: if frontend closes
	 * connection, we will notice it and exit cleanly when control next
B
Bruce Momjian 已提交
2790 2791
	 * returns to outer loop.  This seems safer than forcing exit in the midst
	 * of output during who-knows-what operation...
2792 2793
	 */
	pqsignal(SIGPIPE, SIG_IGN);
2794 2795
	pqsignal(SIGUSR1, CatchupInterruptHandler);
	pqsignal(SIGUSR2, NotifyInterruptHandler);
2796
	pqsignal(SIGFPE, FloatExceptionHandler);
2797 2798

	/*
B
Bruce Momjian 已提交
2799
	 * Reset some signals that are accepted by postmaster but not by backend
2800
	 */
B
Bruce Momjian 已提交
2801
	pqsignal(SIGCHLD, SIG_DFL); /* system() requires this on some platforms */
2802

2803 2804
	pqinitmask();

T
Tom Lane 已提交
2805
	/* We allow SIGQUIT (quickdie) at all times */
2806
#ifdef HAVE_SIGPROCMASK
T
Tom Lane 已提交
2807
	sigdelset(&BlockSig, SIGQUIT);
2808
#else
T
Tom Lane 已提交
2809
	BlockSig &= ~(sigmask(SIGQUIT));
2810 2811
#endif

T
Tom Lane 已提交
2812
	PG_SETMASK(&BlockSig);		/* block everything except SIGQUIT */
2813

2814

2815
	if (IsUnderPostmaster)
2816
	{
2817
		/* noninteractive case: nothing should be left after switches */
2818
		if (errs || argc != optind || dbname == NULL)
2819
		{
2820 2821
			ereport(FATAL,
					(errcode(ERRCODE_SYNTAX_ERROR),
B
Bruce Momjian 已提交
2822 2823
				 errmsg("invalid command-line arguments for server process"),
			   errhint("Try \"%s --help\" for more information.", argv[0])));
2824
		}
2825

2826
		BaseInit();
2827
	}
2828
	else
2829
	{
2830 2831 2832
		/* interactive case: database name can be last arg on command line */
		if (errs || argc - optind > 1)
		{
2833 2834 2835 2836
			ereport(FATAL,
					(errcode(ERRCODE_SYNTAX_ERROR),
					 errmsg("%s: invalid command-line arguments",
							argv[0]),
B
Bruce Momjian 已提交
2837
			   errhint("Try \"%s --help\" for more information.", argv[0])));
2838 2839
		}
		else if (argc - optind == 1)
2840 2841
			dbname = argv[optind];
		else if ((dbname = username) == NULL)
2842
		{
2843 2844 2845 2846
			ereport(FATAL,
					(errcode(ERRCODE_INVALID_PARAMETER_VALUE),
					 errmsg("%s: no database nor user name specified",
							argv[0])));
2847
		}
2848

2849
		/*
B
Bruce Momjian 已提交
2850 2851
		 * Validate we have been given a reasonable-looking DataDir (if under
		 * postmaster, assume postmaster did this already).
2852
		 */
2853
		Assert(DataDir);
2854 2855
		ValidatePgVersion(DataDir);

2856 2857 2858
		/* Change into DataDir (if under postmaster, was done already) */
		ChangeToDataDir();

2859
		/*
2860
		 * Create lockfile for data directory.
2861
		 */
2862
		CreateDataDirLockFile(false);
2863

2864
		BaseInit();
2865 2866 2867 2868 2869

		/*
		 * Start up xlog for standalone backend, and register to have it
		 * closed down at exit.
		 */
2870
		StartupXLOG();
2871
		on_shmem_exit(ShutdownXLOG, 0);
2872 2873

		/*
B
Bruce Momjian 已提交
2874 2875
		 * Read any existing FSM cache file, and register to write one out at
		 * exit.
2876 2877 2878
		 */
		LoadFreeSpaceMap();
		on_shmem_exit(DumpFreeSpaceMap, 0);
2879 2880

		/*
B
Bruce Momjian 已提交
2881 2882
		 * We have to build the flat file for pg_database, but not for the
		 * user and group tables, since we won't try to do authentication.
2883 2884
		 */
		BuildFlatFiles(true);
2885 2886
	}

2887
	/*
2888 2889
	 * General initialization.
	 *
B
Bruce Momjian 已提交
2890 2891 2892
	 * NOTE: if you are tempted to add code in this vicinity, consider putting it
	 * inside InitPostgres() instead.  In particular, anything that involves
	 * database access should be there, not here.
2893
	 */
2894 2895
	ereport(DEBUG3,
			(errmsg_internal("InitPostgres")));
2896
	am_superuser = InitPostgres(dbname, username);
2897

2898
	SetProcessingMode(NormalProcessing);
2899

2900
	/*
2901 2902
	 * Now that we know if client is a superuser, we can try to apply SUSET
	 * GUC options that came from the client.
2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930
	 */
	ctx = am_superuser ? PGC_SUSET : PGC_USERSET;

	if (debug_flag >= 0)
		set_debug_options(debug_flag, ctx, PGC_S_CLIENT);

	if (guc_names != NIL)
	{
		ListCell   *namcell,
				   *valcell;

		forboth(namcell, guc_names, valcell, guc_values)
		{
			char	   *name = (char *) lfirst(namcell);
			char	   *value = (char *) lfirst(valcell);

			SetConfigOption(name, value, ctx, PGC_S_CLIENT);
			pfree(name);
			pfree(value);
		}
	}

	/*
	 * Now all GUC states are fully set up.  Report them to client if
	 * appropriate.
	 */
	BeginReportingGUCOptions();

2931
	/*
B
Bruce Momjian 已提交
2932 2933
	 * Also set up handler to log session end; we have to wait till now to be
	 * sure Log_disconnections has its final value.
2934 2935 2936 2937
	 */
	if (IsUnderPostmaster && Log_disconnections)
		on_proc_exit(log_disconnections, 0);

2938 2939
	/*
	 * Send this backend's cancellation info to the frontend.
2940
	 */
2941
	if (whereToSendOutput == DestRemote &&
M
 
Marc G. Fournier 已提交
2942 2943
		PG_PROTOCOL_MAJOR(FrontendProtocol) >= 2)
	{
2944
		StringInfoData buf;
B
Bruce Momjian 已提交
2945

2946
		pq_beginmessage(&buf, 'K');
2947 2948 2949
		pq_sendint(&buf, (int32) MyProcPid, sizeof(int32));
		pq_sendint(&buf, (int32) MyCancelKey, sizeof(int32));
		pq_endmessage(&buf);
M
 
Marc G. Fournier 已提交
2950 2951 2952
		/* Need not flush since ReadyForQuery will do it. */
	}

2953
	/* Welcome banner for standalone case */
2954
	if (whereToSendOutput == DestDebug)
2955
		printf("\nPostgreSQL stand-alone backend %s\n", PG_VERSION);
2956

2957 2958 2959
	/*
	 * Create the memory context we will use in the main loop.
	 *
2960 2961
	 * MessageContext is reset once per iteration of the main loop, ie, upon
	 * completion of processing of each command message from the client.
2962
	 */
2963 2964 2965 2966 2967
	MessageContext = AllocSetContextCreate(TopMemoryContext,
										   "MessageContext",
										   ALLOCSET_DEFAULT_MINSIZE,
										   ALLOCSET_DEFAULT_INITSIZE,
										   ALLOCSET_DEFAULT_MAXSIZE);
2968

2969
	/*
2970
	 * Remember stand-alone backend startup time
2971 2972
	 */
	if (!IsUnderPostmaster)
2973
		PgStartTime = GetCurrentTimestamp();
2974

2975 2976
	/*
	 * POSTGRES main processing loop begins here
2977
	 *
B
Bruce Momjian 已提交
2978 2979
	 * If an exception is encountered, processing resumes here so we abort the
	 * current transaction and start a new one.
2980
	 *
B
Bruce Momjian 已提交
2981 2982 2983 2984 2985 2986 2987
	 * You might wonder why this isn't coded as an infinite loop around a PG_TRY
	 * construct.  The reason is that this is the bottom of the exception
	 * stack, and so with PG_TRY there would be no exception handler in force
	 * at all during the CATCH part.  By leaving the outermost setjmp always
	 * active, we have at least some chance of recovering from an error during
	 * error recovery.	(If we get into an infinite loop thereby, it will soon
	 * be stopped by overflow of elog.c's internal state stack.)
2988 2989
	 */

2990
	if (sigsetjmp(local_sigjmp_buf, 1) != 0)
2991
	{
2992
		/*
2993
		 * NOTE: if you are tempted to add more code in this if-block,
2994
		 * consider the high probability that it should be in
B
Bruce Momjian 已提交
2995
		 * AbortTransaction() instead.	The only stuff done directly here
B
Bruce Momjian 已提交
2996 2997
		 * should be stuff that is guaranteed to apply *only* for outer-level
		 * error recovery, such as adjusting the FE/BE protocol status.
2998 2999 3000 3001 3002 3003 3004 3005 3006
		 */

		/* Since not using PG_TRY, must reset error stack by hand */
		error_context_stack = NULL;

		/* Prevent interrupts while cleaning up */
		HOLD_INTERRUPTS();

		/*
B
Bruce Momjian 已提交
3007 3008
		 * Forget any pending QueryCancel request, since we're returning to
		 * the idle loop anyway, and cancel the statement timer if running.
3009 3010
		 */
		QueryCancelPending = false;
3011
		disable_sig_alarm(true);
B
Bruce Momjian 已提交
3012
		QueryCancelPending = false;		/* again in case timeout occurred */
3013 3014

		/*
B
Bruce Momjian 已提交
3015 3016 3017
		 * Turn off these interrupts too.  This is only needed here and not in
		 * other exception-catching places since these interrupts are only
		 * enabled while we wait for client input.
3018
		 */
3019
		DoingCommandRead = false;
3020
		DisableNotifyInterrupt();
3021
		DisableCatchupInterrupt();
3022

3023 3024 3025
		/* Make sure libpq is in a good state */
		pq_comm_reset();

3026 3027
		/* Report the error to the client and/or server log */
		EmitErrorReport();
3028

3029
		/*
B
Bruce Momjian 已提交
3030 3031
		 * Make sure debug_query_string gets reset before we possibly clobber
		 * the storage it points at.
3032
		 */
3033
		debug_query_string = NULL;
3034

3035
		/*
3036
		 * Abort the current transaction in order to recover.
3037
		 */
3038
		AbortCurrentTransaction();
3039 3040

		/*
B
Bruce Momjian 已提交
3041 3042
		 * Now return to normal top-level context and clear ErrorContext for
		 * next time.
3043 3044
		 */
		MemoryContextSwitchTo(TopMemoryContext);
3045
		FlushErrorState();
3046
		QueryContext = NULL;
3047

3048
		/*
B
Bruce Momjian 已提交
3049 3050
		 * If we were handling an extended-query-protocol message, initiate
		 * skip till next Sync.  This also causes us not to issue
B
Bruce Momjian 已提交
3051
		 * ReadyForQuery (until we get Sync).
3052 3053 3054
		 */
		if (doing_extended_query_message)
			ignore_till_sync = true;
3055

3056 3057 3058 3059
		/* We don't have a transaction command open anymore */
		xact_started = false;

		/* Now we can allow interrupts again */
3060
		RESUME_INTERRUPTS();
3061
	}
3062

3063 3064
	/* We can now handle ereport(ERROR) */
	PG_exception_stack = &local_sigjmp_buf;
3065

3066
	PG_SETMASK(&UnBlockSig);
3067

3068 3069
	if (!ignore_till_sync)
		send_rfq = true;		/* initially, or after error */
3070

3071 3072
	/*
	 * Non-error queries loop here.
3073 3074 3075 3076
	 */

	for (;;)
	{
3077
		/*
B
Bruce Momjian 已提交
3078 3079
		 * At top of loop, reset extended-query-message flag, so that any
		 * errors encountered in "idle" state don't provoke skip.
3080 3081 3082
		 */
		doing_extended_query_message = false;

3083
		/*
B
Bruce Momjian 已提交
3084 3085
		 * Release storage left over from prior query cycle, and create a new
		 * query input buffer in the cleared MessageContext.
3086
		 */
3087 3088
		MemoryContextSwitchTo(MessageContext);
		MemoryContextResetAndDeleteChildren(MessageContext);
3089

3090
		initStringInfo(&input_message);
3091

3092
		/*
B
Bruce Momjian 已提交
3093 3094
		 * (1) If we've reached idle state, tell the frontend we're ready for
		 * a new query.
3095
		 *
3096
		 * Note: this includes fflush()'ing the last of the prior output.
3097 3098 3099
		 *
		 * This is also a good time to send collected statistics to the
		 * collector, and to update the PS stats display.  We avoid doing
B
Bruce Momjian 已提交
3100 3101 3102
		 * those every time through the message loop because it'd slow down
		 * processing of batched messages, and because we don't want to report
		 * uncommitted updates (that confuses autovacuum).
B
Bruce Momjian 已提交
3103
		 */
3104 3105
		if (send_rfq)
		{
3106
			if (IsTransactionOrTransactionBlock())
3107 3108 3109 3110 3111 3112
			{
				set_ps_display("idle in transaction");
				pgstat_report_activity("<IDLE> in transaction");
			}
			else
			{
3113 3114
				pgstat_report_tabstat();

3115 3116 3117
				set_ps_display("idle");
				pgstat_report_activity("<IDLE>");
			}
3118

3119 3120
			ReadyForQuery(whereToSendOutput);
			send_rfq = false;
3121
		}
3122

3123
		/*
B
Bruce Momjian 已提交
3124 3125 3126 3127
		 * (2) Allow asynchronous signals to be executed immediately if they
		 * come in while we are waiting for client input. (This must be
		 * conditional since we don't want, say, reads on behalf of COPY FROM
		 * STDIN doing the same thing.)
3128
		 */
3129 3130
		QueryCancelPending = false;		/* forget any earlier CANCEL signal */
		DoingCommandRead = true;
3131

3132 3133
		/*
		 * (3) read a command (loop blocks here)
3134
		 */
3135
		firstchar = ReadCommand(&input_message);
3136

3137 3138
		/*
		 * (4) disable async signal conditions again.
3139
		 */
3140
		DoingCommandRead = false;
3141

3142
		/*
B
Bruce Momjian 已提交
3143 3144
		 * (5) check for any other interesting events that happened while we
		 * slept.
3145 3146 3147 3148 3149 3150 3151
		 */
		if (got_SIGHUP)
		{
			got_SIGHUP = false;
			ProcessConfigFile(PGC_SIGHUP);
		}

3152
		/*
B
Bruce Momjian 已提交
3153 3154
		 * (6) process the command.  But ignore it if we're skipping till
		 * Sync.
3155
		 */
3156
		if (ignore_till_sync && firstchar != EOF)
3157 3158
			continue;

3159 3160
		switch (firstchar)
		{
3161
			case 'Q':			/* simple query */
3162
				{
3163 3164
					const char *query_string;

3165 3166
					query_string = pq_getmsgstring(&input_message);
					pq_getmsgend(&input_message);
3167

3168
					exec_simple_query(query_string);
3169

3170 3171
					send_rfq = true;
				}
3172 3173
				break;

3174 3175 3176 3177 3178 3179 3180
			case 'P':			/* parse */
				{
					const char *stmt_name;
					const char *query_string;
					int			numParams;
					Oid		   *paramTypes = NULL;

3181 3182 3183
					stmt_name = pq_getmsgstring(&input_message);
					query_string = pq_getmsgstring(&input_message);
					numParams = pq_getmsgint(&input_message, 2);
3184 3185
					if (numParams > 0)
					{
B
Bruce Momjian 已提交
3186
						int			i;
3187 3188 3189

						paramTypes = (Oid *) palloc(numParams * sizeof(Oid));
						for (i = 0; i < numParams; i++)
3190
							paramTypes[i] = pq_getmsgint(&input_message, 4);
3191
					}
3192
					pq_getmsgend(&input_message);
3193 3194 3195 3196 3197 3198 3199

					exec_parse_message(query_string, stmt_name,
									   paramTypes, numParams);
				}
				break;

			case 'B':			/* bind */
B
Bruce Momjian 已提交
3200

3201
				/*
B
Bruce Momjian 已提交
3202 3203
				 * this message is complex enough that it seems best to put
				 * the field extraction out-of-line
3204
				 */
3205
				exec_bind_message(&input_message);
3206 3207 3208 3209 3210
				break;

			case 'E':			/* execute */
				{
					const char *portal_name;
B
Bruce Momjian 已提交
3211
					int			max_rows;
3212

3213 3214 3215
					portal_name = pq_getmsgstring(&input_message);
					max_rows = pq_getmsgint(&input_message, 4);
					pq_getmsgend(&input_message);
3216

3217
					exec_execute_message(portal_name, max_rows);
3218 3219 3220
				}
				break;

3221 3222
			case 'F':			/* fastpath function call */
				/* Tell the collector what we're doing */
3223 3224
				pgstat_report_activity("<FASTPATH> function call");

3225
				/* start an xact for this function invocation */
3226
				start_xact_command();
3227

3228 3229 3230
				/* switch back to message context */
				MemoryContextSwitchTo(MessageContext);

3231 3232 3233
				/* set snapshot in case function needs one */
				ActiveSnapshot = CopySnapshot(GetTransactionSnapshot());

3234
				if (HandleFunctionRequest(&input_message) == EOF)
3235 3236
				{
					/* lost frontend connection during F message input */
B
Bruce Momjian 已提交
3237

3238
					/*
3239
					 * Reset whereToSendOutput to prevent ereport from
B
Bruce Momjian 已提交
3240
					 * attempting to send any more messages to client.
3241
					 */
3242 3243
					if (whereToSendOutput == DestRemote)
						whereToSendOutput = DestNone;
3244

3245
					proc_exit(0);
3246
				}
3247 3248

				/* commit the function-invocation transaction */
3249
				finish_xact_command();
3250

3251
				send_rfq = true;
3252 3253
				break;

B
Bruce Momjian 已提交
3254
			case 'C':			/* close */
3255
				{
B
Bruce Momjian 已提交
3256
					int			close_type;
3257 3258
					const char *close_target;

3259 3260 3261
					close_type = pq_getmsgbyte(&input_message);
					close_target = pq_getmsgstring(&input_message);
					pq_getmsgend(&input_message);
3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289

					switch (close_type)
					{
						case 'S':
							if (close_target[0] != '\0')
								DropPreparedStatement(close_target, false);
							else
							{
								/* special-case the unnamed statement */
								unnamed_stmt_pstmt = NULL;
								if (unnamed_stmt_context)
								{
									DropDependentPortals(unnamed_stmt_context);
									MemoryContextDelete(unnamed_stmt_context);
								}
								unnamed_stmt_context = NULL;
							}
							break;
						case 'P':
							{
								Portal		portal;

								portal = GetPortalByName(close_target);
								if (PortalIsValid(portal))
									PortalDrop(portal, false);
							}
							break;
						default:
3290 3291
							ereport(ERROR,
									(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
3292 3293
								   errmsg("invalid CLOSE message subtype %d",
										  close_type)));
3294 3295 3296
							break;
					}

3297
					if (whereToSendOutput == DestRemote)
B
Bruce Momjian 已提交
3298
						pq_putemptymessage('3');		/* CloseComplete */
3299 3300 3301 3302 3303
				}
				break;

			case 'D':			/* describe */
				{
B
Bruce Momjian 已提交
3304
					int			describe_type;
3305 3306
					const char *describe_target;

3307 3308 3309
					describe_type = pq_getmsgbyte(&input_message);
					describe_target = pq_getmsgstring(&input_message);
					pq_getmsgend(&input_message);
3310 3311 3312 3313 3314 3315 3316 3317 3318 3319

					switch (describe_type)
					{
						case 'S':
							exec_describe_statement_message(describe_target);
							break;
						case 'P':
							exec_describe_portal_message(describe_target);
							break;
						default:
3320 3321
							ereport(ERROR,
									(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
3322 3323
								errmsg("invalid DESCRIBE message subtype %d",
									   describe_type)));
3324 3325 3326 3327 3328
							break;
					}
				}
				break;

B
Bruce Momjian 已提交
3329
			case 'H':			/* flush */
3330
				pq_getmsgend(&input_message);
3331
				if (whereToSendOutput == DestRemote)
3332 3333 3334
					pq_flush();
				break;

B
Bruce Momjian 已提交
3335
			case 'S':			/* sync */
3336
				pq_getmsgend(&input_message);
3337
				finish_xact_command();
3338 3339 3340
				send_rfq = true;
				break;

3341
				/*
B
Bruce Momjian 已提交
3342 3343 3344
				 * 'X' means that the frontend is closing down the socket. EOF
				 * means unexpected loss of frontend connection. Either way,
				 * perform normal shutdown.
3345 3346
				 */
			case 'X':
3347
			case EOF:
B
Bruce Momjian 已提交
3348

3349
				/*
B
Bruce Momjian 已提交
3350 3351
				 * Reset whereToSendOutput to prevent ereport from attempting
				 * to send any more messages to client.
3352
				 */
3353 3354
				if (whereToSendOutput == DestRemote)
					whereToSendOutput = DestNone;
B
Bruce Momjian 已提交
3355

3356 3357
				/*
				 * NOTE: if you are tempted to add more code here, DON'T!
B
Bruce Momjian 已提交
3358
				 * Whatever you had in mind to do should be set up as an
B
Bruce Momjian 已提交
3359 3360 3361
				 * on_proc_exit or on_shmem_exit callback, instead. Otherwise
				 * it will fail to be called during other backend-shutdown
				 * scenarios.
3362
				 */
3363
				proc_exit(0);
3364

B
Bruce Momjian 已提交
3365 3366 3367 3368
			case 'd':			/* copy data */
			case 'c':			/* copy done */
			case 'f':			/* copy fail */

3369
				/*
B
Bruce Momjian 已提交
3370
				 * Accept but ignore these messages, per protocol spec; we
B
Bruce Momjian 已提交
3371 3372
				 * probably got here because a COPY failed, and the frontend
				 * is still sending data.
3373 3374 3375
				 */
				break;

3376
			default:
3377 3378 3379 3380
				ereport(FATAL,
						(errcode(ERRCODE_PROTOCOL_VIOLATION),
						 errmsg("invalid frontend message type %d",
								firstchar)));
3381
		}
3382
	}							/* end of input-reading loop */
3383

3384 3385
	/* can't get here because the above loop never exits */
	Assert(false);
3386

3387
	return 1;					/* keep compiler quiet */
3388 3389
}

3390
#ifndef HAVE_GETRUSAGE
B
Bruce Momjian 已提交
3391 3392
#include "rusagestub.h"
#else
3393
#include <sys/resource.h>
3394
#endif   /* HAVE_GETRUSAGE */
3395

3396 3397
static struct rusage Save_r;
static struct timeval Save_t;
3398 3399

void
3400
ResetUsage(void)
3401
{
3402
	getrusage(RUSAGE_SELF, &Save_r);
3403
	gettimeofday(&Save_t, NULL);
3404
	ResetBufferUsage();
3405
	/* ResetTupleCount(); */
3406 3407 3408
}

void
3409
ShowUsage(const char *title)
3410
{
3411
	StringInfoData str;
3412 3413 3414 3415
	struct timeval user,
				sys;
	struct timeval elapse_t;
	struct rusage r;
B
Bruce Momjian 已提交
3416
	char	   *bufusage;
3417 3418

	getrusage(RUSAGE_SELF, &r);
3419
	gettimeofday(&elapse_t, NULL);
3420 3421
	memcpy((char *) &user, (char *) &r.ru_utime, sizeof(user));
	memcpy((char *) &sys, (char *) &r.ru_stime, sizeof(sys));
3422 3423 3424 3425 3426 3427 3428 3429 3430 3431 3432 3433 3434 3435 3436 3437 3438 3439
	if (elapse_t.tv_usec < Save_t.tv_usec)
	{
		elapse_t.tv_sec--;
		elapse_t.tv_usec += 1000000;
	}
	if (r.ru_utime.tv_usec < Save_r.ru_utime.tv_usec)
	{
		r.ru_utime.tv_sec--;
		r.ru_utime.tv_usec += 1000000;
	}
	if (r.ru_stime.tv_usec < Save_r.ru_stime.tv_usec)
	{
		r.ru_stime.tv_sec--;
		r.ru_stime.tv_usec += 1000000;
	}

	/*
	 * the only stats we don't show here are for memory usage -- i can't
B
Bruce Momjian 已提交
3440 3441 3442 3443
	 * figure out how to interpret the relevant fields in the rusage struct,
	 * and they change names across o/s platforms, anyway. if you can figure
	 * out what the entries mean, you can somehow extract resident set size,
	 * shared text size, and unshared data and stack sizes.
3444
	 */
3445
	initStringInfo(&str);
3446

3447 3448
	appendStringInfo(&str, "! system usage stats:\n");
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3449
				"!\t%ld.%06ld elapsed %ld.%06ld user %ld.%06ld system sec\n",
3450 3451 3452
					 (long) (elapse_t.tv_sec - Save_t.tv_sec),
					 (long) (elapse_t.tv_usec - Save_t.tv_usec),
					 (long) (r.ru_utime.tv_sec - Save_r.ru_utime.tv_sec),
B
Bruce Momjian 已提交
3453
					 (long) (r.ru_utime.tv_usec - Save_r.ru_utime.tv_usec),
3454
					 (long) (r.ru_stime.tv_sec - Save_r.ru_stime.tv_sec),
B
Bruce Momjian 已提交
3455
					 (long) (r.ru_stime.tv_usec - Save_r.ru_stime.tv_usec));
3456
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3457
					 "!\t[%ld.%06ld user %ld.%06ld sys total]\n",
3458 3459 3460 3461
					 (long) user.tv_sec,
					 (long) user.tv_usec,
					 (long) sys.tv_sec,
					 (long) sys.tv_usec);
3462
/* BeOS has rusage but only has some fields, and not these... */
3463
#if defined(HAVE_GETRUSAGE)
3464
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3465 3466
					 "!\t%ld/%ld [%ld/%ld] filesystem blocks in/out\n",
					 r.ru_inblock - Save_r.ru_inblock,
3467
	/* they only drink coffee at dec */
B
Bruce Momjian 已提交
3468 3469
					 r.ru_oublock - Save_r.ru_oublock,
					 r.ru_inblock, r.ru_oublock);
3470
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3471
			  "!\t%ld/%ld [%ld/%ld] page faults/reclaims, %ld [%ld] swaps\n",
B
Bruce Momjian 已提交
3472 3473 3474 3475 3476
					 r.ru_majflt - Save_r.ru_majflt,
					 r.ru_minflt - Save_r.ru_minflt,
					 r.ru_majflt, r.ru_minflt,
					 r.ru_nswap - Save_r.ru_nswap,
					 r.ru_nswap);
3477
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3478
		 "!\t%ld [%ld] signals rcvd, %ld/%ld [%ld/%ld] messages rcvd/sent\n",
B
Bruce Momjian 已提交
3479 3480 3481 3482 3483
					 r.ru_nsignals - Save_r.ru_nsignals,
					 r.ru_nsignals,
					 r.ru_msgrcv - Save_r.ru_msgrcv,
					 r.ru_msgsnd - Save_r.ru_msgsnd,
					 r.ru_msgrcv, r.ru_msgsnd);
3484
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3485
			 "!\t%ld/%ld [%ld/%ld] voluntary/involuntary context switches\n",
B
Bruce Momjian 已提交
3486 3487 3488
					 r.ru_nvcsw - Save_r.ru_nvcsw,
					 r.ru_nivcsw - Save_r.ru_nivcsw,
					 r.ru_nvcsw, r.ru_nivcsw);
3489
#endif   /* HAVE_GETRUSAGE */
3490 3491

	bufusage = ShowBufferUsage();
3492
	appendStringInfo(&str, "! buffer usage stats:\n%s", bufusage);
3493 3494 3495
	pfree(bufusage);

	/* remove trailing newline */
B
Bruce Momjian 已提交
3496
	if (str.data[str.len - 1] == '\n')
3497 3498
		str.data[--str.len] = '\0';

3499 3500 3501
	ereport(LOG,
			(errmsg_internal("%s", title),
			 errdetail("%s", str.data)));
3502 3503

	pfree(str.data);
3504
}
3505 3506 3507 3508

/*
 * on_proc_exit handler to log end of session
 */
B
Bruce Momjian 已提交
3509
static void
3510
log_disconnections(int code, Datum arg)
3511
{
B
Bruce Momjian 已提交
3512 3513 3514 3515 3516
	Port	   *port = MyProcPort;
	struct timeval end;
	int			hours,
				minutes,
				seconds;
3517

3518
	gettimeofday(&end, NULL);
3519 3520 3521 3522 3523 3524 3525 3526
	if (end.tv_usec < port->session_start.tv_usec)
	{
		end.tv_sec--;
		end.tv_usec += 1000000;
	}
	end.tv_sec -= port->session_start.tv_sec;
	end.tv_usec -= port->session_start.tv_usec;

3527
	/* for stricter accuracy here we could round - this is close enough */
3528 3529 3530 3531
	hours = end.tv_sec / SECS_PER_HOUR;
	end.tv_sec %= SECS_PER_HOUR;
	minutes = end.tv_sec / SECS_PER_MINUTE;
	seconds = end.tv_sec % SECS_PER_MINUTE;
3532

3533 3534 3535 3536 3537
	ereport(LOG,
			(errmsg("disconnection: session time: %d:%02d:%02d.%02d "
					"user=%s database=%s host=%s%s%s",
					hours, minutes, seconds, (int) (end.tv_usec / 10000),
					port->user_name, port->database_name, port->remote_host,
B
Bruce Momjian 已提交
3538
				  port->remote_port[0] ? " port=" : "", port->remote_port)));
3539
}