postgres.c 92.2 KB
Newer Older
1 2
/*-------------------------------------------------------------------------
 *
3
 * postgres.c
4
 *	  POSTGRES C Backend Interface
5
 *
6
 * Portions Copyright (c) 1996-2006, 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.494 2006/08/04 18:53:46 momjian 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/xact.h"
35
#include "catalog/pg_type.h"
36
#include "commands/async.h"
37
#include "commands/prepare.h"
38
#include "libpq/libpq.h"
39
#include "libpq/pqformat.h"
40
#include "libpq/pqsignal.h"
B
Bruce Momjian 已提交
41
#include "miscadmin.h"
42
#include "nodes/print.h"
43
#include "optimizer/planner.h"
44
#include "parser/analyze.h"
45
#include "parser/parser.h"
B
Bruce Momjian 已提交
46
#include "rewrite/rewriteHandler.h"
47
#include "storage/freespace.h"
48 49
#include "storage/ipc.h"
#include "storage/proc.h"
50
#include "storage/sinval.h"
51 52
#include "tcop/fastpath.h"
#include "tcop/pquery.h"
B
Bruce Momjian 已提交
53
#include "tcop/tcopprot.h"
54
#include "tcop/utility.h"
55
#include "utils/flatfiles.h"
56
#include "utils/lsyscache.h"
57
#include "utils/memutils.h"
M
 
Marc G. Fournier 已提交
58
#include "utils/ps_status.h"
B
Bruce Momjian 已提交
59
#include "mb/pg_wchar.h"
60

61
#include "pgstat.h"
M
 
Marc G. Fournier 已提交
62

63 64 65
extern int	optind;
extern char *optarg;

66
/* ----------------
67
 *		global variables
68 69
 * ----------------
 */
B
Bruce Momjian 已提交
70
const char *debug_query_string; /* for pgmonitor and log_min_error_statement */
71

72
/* Note: whereToSendOutput is initialized for the bootstrap/standalone case */
73
CommandDest whereToSendOutput = DestDebug;
74

75
/* flag for logging end of session */
B
Bruce Momjian 已提交
76
bool		Log_disconnections = false;
77

78 79
LogStmtLevel log_statement = LOGSTMT_NONE;

80 81 82
/* GUC variable for maximum stack depth (measured in kilobytes) */
int			max_stack_depth = 2048;

83 84 85 86
/* wait N seconds to allow attach from a debugger */
int			PostAuthDelay = 0;


87

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

N
Neil Conway 已提交
96 97 98 99
/*
 * Stack base pointer -- initialized by PostgresMain. This is not static
 * so that PL/Java can modify it.
 */
B
Bruce Momjian 已提交
100
char	   *stack_base_ptr = NULL;
101 102


103 104 105 106 107
/*
 * 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?)
 */
108
static volatile sig_atomic_t got_SIGHUP = false;
109

110 111 112 113 114 115
/*
 * 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;

116 117 118 119 120 121 122
/*
 * 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;

123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143
/*
 * 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
144 145
 */
#ifndef TCOP_DONTUSENEWLINE
B
Bruce Momjian 已提交
146
static int	UseNewLine = 1;		/* Use newlines query delimiters (the default) */
147
#else
148
static int	UseNewLine = 0;		/* Use EOF as query delimiters */
149
#endif   /* TCOP_DONTUSENEWLINE */
150 151 152


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


/* ----------------------------------------------------------------
172
 *		routines to obtain user input
173 174 175 176
 * ----------------------------------------------------------------
 */

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

186
static int
187
InteractiveBackend(StringInfo inBuf)
188
{
189 190 191
	int			c;				/* character read from getc() */
	bool		end = false;	/* end-of-input flag */
	bool		backslashSeen = false;	/* have we seen a \ ? */
192

193 194
	/*
	 * display a prompt and obtain input from the user
195
	 */
196
	printf("backend> ");
197
	fflush(stdout);
198

199 200 201
	/* Reset inBuf to empty */
	inBuf->len = 0;
	inBuf->data[0] = '\0';
202
	inBuf->cursor = 0;
203

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

235
				appendStringInfoChar(inBuf, (char) c);
236 237 238 239 240 241 242
			}

			if (c == EOF)
				end = true;
		}
		else
		{
243 244
			/*
			 * otherwise read characters until EOF.
245 246
			 */
			while ((c = getc(stdin)) != EOF)
247
				appendStringInfoChar(inBuf, (char) c);
248

249
			if (inBuf->len == 0)
250 251 252 253
				end = true;
		}

		if (end)
254
			return EOF;
255

256 257
		/*
		 * otherwise we have a user query so process it.
258 259 260 261
		 */
		break;
	}

262 263 264
	/* Add '\0' to make it look the same as message case. */
	appendStringInfoChar(inBuf, (char) '\0');

265 266
	/*
	 * if the query echo flag was given, print the query..
267 268
	 */
	if (EchoQuery)
269
		printf("statement: %s\n", inBuf->data);
270
	fflush(stdout);
271

272
	return 'Q';
273 274 275
}

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

288
	/*
289
	 * Get message type code from the frontend.
290
	 */
291
	qtype = pq_getbyte();
292

293 294
	if (qtype == EOF)			/* frontend disconnected */
	{
295 296 297
		ereport(COMMERROR,
				(errcode(ERRCODE_PROTOCOL_VIOLATION),
				 errmsg("unexpected EOF on client connection")));
298 299 300 301
		return qtype;
	}

	/*
B
Bruce Momjian 已提交
302 303 304
	 * 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.
305
	 *
306 307
	 * This also gives us a place to set the doing_extended_query_message flag
	 * as soon as possible.
308
	 */
309
	switch (qtype)
310
	{
311
		case 'Q':				/* simple query */
312
			doing_extended_query_message = false;
313 314 315 316 317
			if (PG_PROTOCOL_MAJOR(FrontendProtocol) < 3)
			{
				/* old style without length word; convert */
				if (pq_getstring(inBuf))
				{
318 319
					ereport(COMMERROR,
							(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
320
							 errmsg("unexpected EOF on client connection")));
321 322 323
					return EOF;
				}
			}
324 325
			break;

326
		case 'F':				/* fastpath function call */
327 328
			/* we let fastpath.c cope with old-style input of this */
			doing_extended_query_message = false;
329
			break;
330

331
		case 'X':				/* terminate */
332
			doing_extended_query_message = false;
333
			ignore_till_sync = false;
334 335 336 337 338 339 340 341 342 343 344
			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)
345 346
				ereport(FATAL,
						(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
347
						 errmsg("invalid frontend message type %d", qtype)));
348 349 350 351 352 353 354 355 356
			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)
357 358
				ereport(FATAL,
						(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
359
						 errmsg("invalid frontend message type %d", qtype)));
360
			break;
361

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

373
		default:
B
Bruce Momjian 已提交
374

375
			/*
B
Bruce Momjian 已提交
376 377 378
			 * 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.
379
			 */
380 381 382
			ereport(FATAL,
					(errcode(ERRCODE_PROTOCOL_VIOLATION),
					 errmsg("invalid frontend message type %d", qtype)));
383
			break;
384
	}
385

386
	/*
B
Bruce Momjian 已提交
387 388 389
	 * 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.
390 391 392 393 394 395 396
	 */
	if (PG_PROTOCOL_MAJOR(FrontendProtocol) >= 3)
	{
		if (pq_getmessage(inBuf, 0))
			return EOF;			/* suitable message already logged */
	}

397
	return qtype;
398 399 400
}

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

412
	if (whereToSendOutput == DestRemote)
413
		result = SocketBackend(inBuf);
414
	else
415 416
		result = InteractiveBackend(inBuf);
	return result;
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 460 461 462 463 464
/*
 * 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();
	}
}

465 466 467 468 469 470

/*
 * 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.
471 472 473
 *
 * NOTE: this routine is no longer used for processing interactive queries,
 * but it is still needed for parsing of SQL function bodies.
474
 */
475
List *
B
Bruce Momjian 已提交
476
pg_parse_and_rewrite(const char *query_string,	/* string to execute */
477
					 Oid *paramTypes,	/* parameter types */
B
Bruce Momjian 已提交
478
					 int numParams)		/* number of parameters */
479
{
480
	List	   *raw_parsetree_list;
481
	List	   *querytree_list;
482
	ListCell   *list_item;
483

484 485
	/*
	 * (1) parse the request string into a list of raw parse trees.
486
	 */
487
	raw_parsetree_list = pg_parse_query(query_string);
488

489 490
	/*
	 * (2) Do parse analysis and rule rewrite.
491 492 493 494
	 */
	querytree_list = NIL;
	foreach(list_item, raw_parsetree_list)
	{
B
Bruce Momjian 已提交
495
		Node	   *parsetree = (Node *) lfirst(list_item);
496

497
		querytree_list = list_concat(querytree_list,
B
Bruce Momjian 已提交
498
									 pg_analyze_and_rewrite(parsetree,
499
															query_string,
B
Bruce Momjian 已提交
500 501
															paramTypes,
															numParams));
502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519
	}

	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.
 */
520
List *
521
pg_parse_query(const char *query_string)
522
{
523
	List	   *raw_parsetree_list;
524

B
Rename:  
Bruce Momjian 已提交
525
	if (log_parser_stats)
526 527
		ResetUsage();

528
	raw_parsetree_list = raw_parser(query_string);
529

530 531 532 533 534 535 536 537
	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 已提交
538
				char **prepare_string)
539 540 541 542 543 544
{
	ListCell   *parsetree_item;
	bool		log_this_statement = (log_statement == LOGSTMT_ALL);

	*prepare_string = NULL;

B
Bruce Momjian 已提交
545
	/* Check if we need to log the statement, and get prepare_string. */
546
	foreach(parsetree_item, raw_parsetree_list)
547
	{
548 549 550 551 552 553 554 555 556 557
		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);

558 559
		if (IsA(parsetree, SelectStmt) &&
			((SelectStmt *) parsetree)->into == NULL)
B
Bruce Momjian 已提交
560
			continue;			/* optimization for frequent command */
561 562 563 564 565 566 567

		if (log_statement == LOGSTMT_MOD &&
			(IsA(parsetree, InsertStmt) ||
			 IsA(parsetree, UpdateStmt) ||
			 IsA(parsetree, DeleteStmt) ||
			 IsA(parsetree, TruncateStmt) ||
			 (IsA(parsetree, CopyStmt) &&
B
Bruce Momjian 已提交
568
			  ((CopyStmt *) parsetree)->is_from)))		/* COPY FROM */
569 570 571 572 573 574
			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 已提交
575
			 IsA(parsetree, SelectStmt) ||		/* SELECT INTO, CREATE AS */
576 577
			 strncmp(commandTag, "ALTER ", strlen("ALTER ")) == 0 ||
			 strncmp(commandTag, "DROP ", strlen("DROP ")) == 0 ||
B
Bruce Momjian 已提交
578
			 IsA(parsetree, GrantStmt) ||		/* GRANT or REVOKE */
579 580 581 582
			 IsA(parsetree, CommentStmt)))
			log_this_statement = true;

		/*
B
Bruce Momjian 已提交
583
		 * For the first EXECUTE we find, record the client statement used by
584 585
		 * the PREPARE.  PREPARE doesn't save the parse tree so we have no
		 * way to conditionally output based on the type of query prepared.
586 587
		 */
		if (IsA(parsetree, ExecuteStmt))
588
		{
589 590 591
			ExecuteStmt *stmt = (ExecuteStmt *) parsetree;
			PreparedStatement *entry;

592 593
			if (*prepare_string == NULL &&
				(entry = FetchPreparedStatement(stmt->name, false)) != NULL &&
594
				entry->query_string)
595
			{
596
				*prepare_string = palloc(strlen(entry->query_string) +
597 598
									  strlen("  [PREPARE:  %s]") - 2 + 1);
				sprintf(*prepare_string, "  [PREPARE:  %s]",
B
Bruce Momjian 已提交
599
						entry->query_string);
600 601 602
			}
		}
	}
B
Bruce Momjian 已提交
603

604 605 606 607 608 609 610 611 612
	if (log_this_statement)
	{
		ereport(LOG,
				(errmsg("statement: %s%s", query_string,
						*prepare_string ? *prepare_string : "")));
		return true;
	}
	else
		return false;
613 614
}

615

616
/*
617 618
 * Given a raw parsetree (gram.y output), and optionally information about
 * types of parameter symbols ($n), perform parse analysis and rule rewriting.
619 620 621 622 623 624
 *
 * 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.
 */
625
List *
626 627
pg_analyze_and_rewrite(Node *parsetree, const char *query_string,
					   Oid *paramTypes, int numParams)
628 629 630
{
	List	   *querytree_list;

631 632
	/*
	 * (1) Perform parse analysis.
633
	 */
B
Rename:  
Bruce Momjian 已提交
634
	if (log_parser_stats)
635 636
		ResetUsage();

637 638
	querytree_list = parse_analyze(parsetree, query_string,
								   paramTypes, numParams);
639

B
Rename:  
Bruce Momjian 已提交
640
	if (log_parser_stats)
641
		ShowUsage("PARSE ANALYSIS STATISTICS");
642

643 644
	/*
	 * (2) Rewrite the queries, as necessary
645 646 647 648 649 650 651 652
	 */
	querytree_list = pg_rewrite_queries(querytree_list);

	return querytree_list;
}

/*
 * Perform rewriting of a list of queries produced by parse analysis.
653 654 655
 *
 * Note: queries must just have come from the parser, because we do not do
 * AcquireRewriteLocks() on them.
656
 */
657
static List *
658 659 660
pg_rewrite_queries(List *querytree_list)
{
	List	   *new_list = NIL;
661
	ListCell   *list_item;
662 663 664 665 666

	if (log_parser_stats)
		ResetUsage();

	/*
B
Bruce Momjian 已提交
667 668
	 * rewritten queries are collected in new_list.  Note there may be more or
	 * fewer than in the original list.
669
	 */
670
	foreach(list_item, querytree_list)
671
	{
672
		Query	   *querytree = (Query *) lfirst(list_item);
673

674
		if (Debug_print_parse)
675
			elog_node_display(DEBUG1, "parse tree", querytree,
676
							  Debug_pretty_print);
677 678 679

		if (querytree->commandType == CMD_UTILITY)
		{
680 681
			/* don't rewrite utilities, just dump 'em into new_list */
			new_list = lappend(new_list, querytree);
682
		}
683
		else
684
		{
685
			/* rewrite regular queries */
686 687
			List	   *rewritten = QueryRewrite(querytree);

688
			new_list = list_concat(new_list, rewritten);
689 690 691 692 693
		}
	}

	querytree_list = new_list;

B
Rename:  
Bruce Momjian 已提交
694
	if (log_parser_stats)
695
		ShowUsage("REWRITER STATISTICS");
696

697
#ifdef COPY_PARSE_PLAN_TREES
B
Bruce Momjian 已提交
698 699

	/*
B
Bruce Momjian 已提交
700
	 * Optional debugging check: pass querytree output through copyObject()
B
Bruce Momjian 已提交
701
	 */
702 703
	new_list = (List *) copyObject(querytree_list);
	/* This checks both copyObject() and the equal() routines... */
B
Bruce Momjian 已提交
704
	if (!equal(new_list, querytree_list))
705
		elog(WARNING, "copyObject() failed to produce an equal parse tree");
706 707
	else
		querytree_list = new_list;
708 709
#endif

710
	if (Debug_print_rewritten)
711
		elog_node_display(DEBUG1, "rewritten parse tree", querytree_list,
712
						  Debug_pretty_print);
713

714 715
	return querytree_list;
}
716 717


718
/* Generate a plan for a single already-rewritten query. */
719
Plan *
720
pg_plan_query(Query *querytree, ParamListInfo boundParams)
721 722
{
	Plan	   *plan;
723

724 725 726
	/* Utility commands have no plans. */
	if (querytree->commandType == CMD_UTILITY)
		return NULL;
727

B
Rename:  
Bruce Momjian 已提交
728
	if (log_planner_stats)
729
		ResetUsage();
730

731
	/* call the optimizer */
732
	plan = planner(querytree, false, 0, boundParams);
733

B
Rename:  
Bruce Momjian 已提交
734
	if (log_planner_stats)
735
		ShowUsage("PLANNER STATISTICS");
736

737 738 739
#ifdef COPY_PARSE_PLAN_TREES
	/* Optional debugging check: pass plan output through copyObject() */
	{
B
Bruce Momjian 已提交
740
		Plan	   *new_plan = (Plan *) copyObject(plan);
741

B
Bruce Momjian 已提交
742
		/*
B
Bruce Momjian 已提交
743 744
		 * equal() currently does not have routines to compare Plan nodes, so
		 * don't try to test equality here.  Perhaps fix someday?
745 746 747
		 */
#ifdef NOT_USED
		/* This checks both copyObject() and the equal() routines... */
B
Bruce Momjian 已提交
748
		if (!equal(new_plan, plan))
749
			elog(WARNING, "copyObject() failed to produce an equal plan tree");
750 751 752 753 754 755
		else
#endif
			plan = new_plan;
	}
#endif

756 757
	/*
	 * Print plan if debugging.
758
	 */
759
	if (Debug_print_plan)
760
		elog_node_display(DEBUG1, "plan", plan, Debug_pretty_print);
761

762
	return plan;
763 764
}

765 766
/*
 * Generate plans for a list of already-rewritten queries.
767
 *
768 769 770 771 772 773 774 775 776
 * 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 *
777 778
pg_plan_queries(List *querytrees, ParamListInfo boundParams,
				bool needSnapshot)
779 780
{
	List	   *plan_list = NIL;
781
	ListCell   *query_list;
782 783 784 785 786 787 788 789 790 791 792 793 794 795 796

	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)
			{
797
				ActiveSnapshot = CopySnapshot(GetTransactionSnapshot());
798 799
				needSnapshot = false;
			}
800
			plan = pg_plan_query(query, boundParams);
801 802 803 804 805 806 807 808 809 810
		}

		plan_list = lappend(plan_list, plan);
	}

	return plan_list;
}


/*
811
 * exec_simple_query
812
 *
813
 * Execute a "simple Query" protocol message.
814
 */
815
static void
816
exec_simple_query(const char *query_string)
817
{
B
Bruce Momjian 已提交
818
	CommandDest dest = whereToSendOutput;
819
	MemoryContext oldcontext;
820 821
	List	   *parsetree_list;
	ListCell   *parsetree_item;
822
	bool		save_log_statement_stats = log_statement_stats;
B
Bruce Momjian 已提交
823
	char	   *prepare_string = NULL;
824
	bool		was_logged = false;
B
Bruce Momjian 已提交
825

826 827 828
	/*
	 * Report query to various monitoring facilities.
	 */
829
	debug_query_string = query_string;
830

831 832
	pgstat_report_activity(query_string);

833
	/*
834 835
	 * We use save_log_statement_stats so ShowUsage doesn't report incorrect
	 * results because ResetUsage wasn't called.
836
	 */
837 838 839
	if (save_log_statement_stats)
		ResetUsage();

840
	/*
B
Bruce Momjian 已提交
841 842
	 * 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 已提交
843 844 845
	 * 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.)
846
	 */
847
	start_xact_command();
848 849

	/*
B
Bruce Momjian 已提交
850 851 852 853
	 * 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.)
854 855 856 857 858 859 860 861
	 */
	unnamed_stmt_pstmt = NULL;
	if (unnamed_stmt_context)
	{
		DropDependentPortals(unnamed_stmt_context);
		MemoryContextDelete(unnamed_stmt_context);
	}
	unnamed_stmt_context = NULL;
862

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

868 869
	QueryContext = CurrentMemoryContext;

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

876 877
	/* Log immediately if dictated by log_statement */
	if (log_statement != LOGSTMT_NONE)
878
		was_logged = log_after_parse(parsetree_list, query_string,
B
Bruce Momjian 已提交
879
									 &prepare_string);
880

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

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

900
		/*
B
Bruce Momjian 已提交
901 902 903 904
		 * 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.
905 906 907
		 */
		commandTag = CreateCommandTag(parsetree);

908
		set_ps_display(commandTag, false);
909

910
		BeginCommand(commandTag, dest);
911

912
		/*
913
		 * If we are in an aborted transaction, reject all commands except
B
Bruce Momjian 已提交
914 915 916 917 918
		 * 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...)
919
		 */
920 921 922 923 924
		if (IsAbortedTransactionBlockState() &&
			!IsTransactionExitStmt(parsetree))
			ereport(ERROR,
					(errcode(ERRCODE_IN_FAILED_SQL_TRANSACTION),
					 errmsg("current transaction is aborted, "
B
Bruce Momjian 已提交
925
						"commands ignored until end of transaction block")));
926

927
		/* Make sure we are in a transaction command */
928
		start_xact_command();
929

930
		/* If we got a cancel signal in parsing or prior command, quit */
931
		CHECK_FOR_INTERRUPTS();
932 933

		/*
934
		 * OK to analyze, rewrite, and plan this query.
935
		 *
B
Bruce Momjian 已提交
936 937
		 * Switch to appropriate context for constructing querytrees (again,
		 * these must outlive the execution context).
938
		 */
939
		oldcontext = MemoryContextSwitchTo(MessageContext);
940

941 942
		querytree_list = pg_analyze_and_rewrite(parsetree, query_string,
												NULL, 0);
V
Vadim B. Mikheev 已提交
943

944
		plantree_list = pg_plan_queries(querytree_list, NULL, true);
945 946 947 948

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

949
		/*
B
Bruce Momjian 已提交
950 951
		 * Create unnamed portal to run the query or queries in. If there
		 * already is one, silently drop it.
952
		 */
953
		portal = CreatePortal("", true, true);
954 955
		/* Don't display the portal in pg_cursors */
		portal->visible = false;
956

957 958 959 960 961 962
		PortalDefineQuery(portal,
						  query_string,
						  commandTag,
						  querytree_list,
						  plantree_list,
						  MessageContext);
963

964
		/*
965
		 * Start the portal.  No parameters here.
966
		 */
967
		PortalStart(portal, NULL, InvalidSnapshot);
968

969
		/*
B
Bruce Momjian 已提交
970 971 972 973
		 * 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...)
974 975 976 977 978 979 980 981 982 983 984 985
		 */
		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 已提交
986
					format = 1; /* BINARY */
987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001
			}
		}
		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 已提交
1002
		 * Run the portal to completion, and then drop it (and the receiver).
1003
		 */
1004 1005
		(void) PortalRun(portal,
						 FETCH_ALL,
1006 1007
						 receiver,
						 receiver,
1008
						 completionTag);
1009

1010
		(*receiver->rDestroy) (receiver);
1011

1012
		PortalDrop(portal, false);
1013

1014
		if (IsA(parsetree, TransactionStmt))
1015
		{
1016
			/*
B
Bruce Momjian 已提交
1017 1018
			 * If this was a transaction control statement, commit it. We will
			 * start a new xact command for the next command (if any).
1019
			 */
1020
			finish_xact_command();
1021
		}
1022
		else if (lnext(parsetree_item) == NULL)
1023
		{
1024
			/*
B
Bruce Momjian 已提交
1025 1026 1027 1028 1029 1030 1031 1032
			 * 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.
1033
			 */
1034
			finish_xact_command();
1035
		}
1036
		else
1037
		{
1038
			/*
B
Bruce Momjian 已提交
1039 1040
			 * We need a CommandCounterIncrement after every query, except
			 * those that start or end a transaction block.
1041 1042
			 */
			CommandCounterIncrement();
1043
		}
1044 1045

		/*
B
Bruce Momjian 已提交
1046 1047 1048 1049
		 * 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.)
1050
		 */
1051
		EndCommand(completionTag, dest);
B
Bruce Momjian 已提交
1052
	}							/* end loop over parsetrees */
1053

1054 1055 1056
	/*
	 * Close down transaction statement, if one is open.
	 */
1057
	finish_xact_command();
1058

1059 1060 1061
	/*
	 * If there were no parsetrees, return EmptyQueryResponse message.
	 */
1062
	if (!parsetree_list)
1063 1064
		NullCommand(dest);

1065
	QueryContext = NULL;
1066

1067
	/*
1068
	 * Emit duration logging if appropriate.
1069
	 */
1070
	if (log_duration || log_min_duration_statement >= 0)
1071
	{
1072 1073 1074
		long		secs;
		int			usecs;
		int			msecs;
B
Bruce Momjian 已提交
1075

1076 1077 1078 1079
		TimestampDifference(GetCurrentStatementStartTimestamp(),
							GetCurrentTimestamp(),
							&secs, &usecs);
		msecs = usecs / 1000;
B
Bruce Momjian 已提交
1080

B
Bruce Momjian 已提交
1081
		/*
1082 1083 1084 1085
		 * The odd-looking test for log_min_duration_statement being
		 * exceeded is designed to avoid integer overflow with very
		 * long durations: don't compute secs * 1000 until we've
		 * verified it will fit in int.
B
Bruce Momjian 已提交
1086
		 */
1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103
		if (log_duration ||
			log_min_duration_statement == 0 ||
			(log_min_duration_statement > 0 &&
			 (secs > log_min_duration_statement / 1000 ||
			  secs * 1000 + msecs >= log_min_duration_statement)))
		{
			if (was_logged)
				ereport(LOG,
						(errmsg("duration: %ld.%03d ms",
								secs, msecs)));
			else
				ereport(LOG,
						(errmsg("duration: %ld.%03d ms  statement: %s%s",
								secs, msecs,
								query_string,
								prepare_string ? prepare_string : "")));
		}
1104
	}
1105

1106 1107 1108
	if (save_log_statement_stats)
		ShowUsage("QUERY STATISTICS");

1109 1110 1111
	if (prepare_string != NULL)
		pfree(prepare_string);

1112
	debug_query_string = NULL;
1113 1114
}

1115 1116 1117 1118 1119 1120 1121 1122
/*
 * 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 已提交
1123 1124
				   Oid *paramTypes,		/* parameter types */
				   int numParams)		/* number of parameters */
1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141
{
	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);

1142
	set_ps_display("PARSE", false);
1143 1144 1145 1146

	if (save_log_statement_stats)
		ResetUsage();

1147 1148
	if (log_statement == LOGSTMT_ALL)
		ereport(LOG,
1149 1150
				(errmsg("statement: <protocol> PREPARE %s AS %s",
						*stmt_name ? stmt_name : "<unnamed>",
1151
						query_string)));
1152

1153
	/*
B
Bruce Momjian 已提交
1154 1155 1156
	 * 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.
1157
	 */
1158
	start_xact_command();
1159 1160 1161 1162

	/*
	 * Switch to appropriate context for constructing parsetrees.
	 *
B
Bruce Momjian 已提交
1163 1164 1165
	 * 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 已提交
1166 1167 1168 1169 1170 1171
	 * 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.
1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201
	 */
	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 已提交
1202 1203
	 * Do basic parsing of the query or queries (this should be safe even if
	 * we are in aborted transaction state!)
1204 1205 1206 1207
	 */
	parsetree_list = pg_parse_query(query_string);

	/*
B
Bruce Momjian 已提交
1208 1209 1210
	 * 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.
1211
	 */
1212
	if (list_length(parsetree_list) > 1)
1213 1214
		ereport(ERROR,
				(errcode(ERRCODE_SYNTAX_ERROR),
B
Bruce Momjian 已提交
1215
		errmsg("cannot insert multiple commands into a prepared statement")));
1216 1217 1218

	if (parsetree_list != NIL)
	{
1219
		Node	   *parsetree = (Node *) linitial(parsetree_list);
B
Bruce Momjian 已提交
1220
		int			i;
1221 1222 1223 1224 1225 1226 1227 1228

		/*
		 * 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 已提交
1229 1230 1231 1232 1233
		 * 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...)
1234
		 */
1235 1236 1237 1238 1239
		if (IsAbortedTransactionBlockState() &&
			!IsTransactionExitStmt(parsetree))
			ereport(ERROR,
					(errcode(ERRCODE_IN_FAILED_SQL_TRANSACTION),
					 errmsg("current transaction is aborted, "
B
Bruce Momjian 已提交
1240
						"commands ignored until end of transaction block")));
1241 1242 1243

		/*
		 * OK to analyze, rewrite, and plan this query.  Note that the
B
Bruce Momjian 已提交
1244 1245
		 * originally specified parameter set is not required to be complete,
		 * so we have to use parse_analyze_varparams().
1246 1247 1248 1249 1250
		 */
		if (log_parser_stats)
			ResetUsage();

		querytree_list = parse_analyze_varparams(parsetree,
1251
												 query_string,
1252 1253 1254 1255 1256 1257 1258 1259 1260 1261
												 &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 已提交
1262
			Oid			ptype = paramTypes[i];
1263 1264

			if (ptype == InvalidOid || ptype == UNKNOWNOID)
1265 1266
				ereport(ERROR,
						(errcode(ERRCODE_INDETERMINATE_DATATYPE),
B
Bruce Momjian 已提交
1267 1268
					 errmsg("could not determine data type of parameter $%d",
							i + 1)));
1269
			param_list = lappend_oid(param_list, ptype);
1270 1271 1272 1273 1274 1275 1276
		}

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

		querytree_list = pg_rewrite_queries(querytree_list);

1277
		/*
B
Bruce Momjian 已提交
1278 1279
		 * If this is the unnamed statement and it has parameters, defer query
		 * planning until Bind.  Otherwise do it now.
1280 1281 1282 1283 1284
		 */
		if (!is_named && numParams > 0)
			plantree_list = NIL;
		else
			plantree_list = pg_plan_queries(querytree_list, NULL, true);
1285 1286 1287
	}
	else
	{
B
Bruce Momjian 已提交
1288
		/* Empty input string.	This is legal. */
1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307
		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,
1308 1309
							   param_list,
							   false);
1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322
	}
	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;
1323
		pstmt->from_sql = false;
1324 1325 1326 1327 1328 1329 1330 1331 1332 1333
		pstmt->context = unnamed_stmt_context;
		/* Now the unnamed statement is complete and valid */
		unnamed_stmt_pstmt = pstmt;
	}

	MemoryContextSwitchTo(oldcontext);

	QueryContext = NULL;

	/*
B
Bruce Momjian 已提交
1334 1335 1336
	 * 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.
1337 1338 1339 1340 1341 1342
	 */
	CommandCounterIncrement();

	/*
	 * Send ParseComplete.
	 */
1343
	if (whereToSendOutput == DestRemote)
1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361
		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;
1362 1363
	int			numPFormats;
	int16	   *pformats = NULL;
1364
	int			numParams;
1365 1366 1367
	int			numRFormats;
	int16	   *rformats = NULL;
	int			i;
1368 1369 1370 1371 1372 1373
	PreparedStatement *pstmt;
	Portal		portal;
	ParamListInfo params;

	pgstat_report_activity("<BIND>");

1374
	set_ps_display("BIND", false);
1375 1376

	/*
B
Bruce Momjian 已提交
1377 1378 1379
	 * 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.
1380
	 */
1381
	start_xact_command();
1382

1383 1384 1385
	/* Switch back to message context */
	MemoryContextSwitchTo(MessageContext);

1386 1387 1388 1389
	/* Get the fixed part of the message */
	portal_name = pq_getmsgstring(input_message);
	stmt_name = pq_getmsgstring(input_message);

1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
	/* 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)
1403 1404
		ereport(ERROR,
				(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
1405 1406
			errmsg("bind message has %d parameter formats but %d parameters",
				   numPFormats, numParams)));
1407 1408 1409 1410 1411 1412 1413 1414 1415

	/* 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)
1416 1417
			ereport(ERROR,
					(errcode(ERRCODE_UNDEFINED_PSTATEMENT),
B
Bruce Momjian 已提交
1418
					 errmsg("unnamed prepared statement does not exist")));
1419 1420
	}

1421
	if (numParams != list_length(pstmt->argtype_list))
1422 1423 1424
		ereport(ERROR,
				(errcode(ERRCODE_PROTOCOL_VIOLATION),
				 errmsg("bind message supplies %d parameters, but prepared statement \"%s\" requires %d",
B
Bruce Momjian 已提交
1425
				   numParams, stmt_name, list_length(pstmt->argtype_list))));
1426

1427 1428
	/*
	 * If we are in aborted transaction state, the only portals we can
1429 1430 1431 1432 1433
	 * actually run are those containing COMMIT or ROLLBACK commands. We
	 * disallow binding anything else to avoid problems with infrastructure
	 * that expects to run inside a valid transaction.	We also disallow
	 * binding any parameters, since we can't risk calling user-defined I/O
	 * functions.
1434 1435 1436 1437 1438 1439 1440 1441 1442
	 */
	if (IsAbortedTransactionBlockState() &&
		(!IsTransactionExitStmtList(pstmt->query_list) ||
		 numParams != 0))
		ereport(ERROR,
				(errcode(ERRCODE_IN_FAILED_SQL_TRANSACTION),
				 errmsg("current transaction is aborted, "
						"commands ignored until end of transaction block")));

1443
	/*
B
Bruce Momjian 已提交
1444 1445
	 * Create the portal.  Allow silent replacement of an existing portal only
	 * if the unnamed portal is specified.
1446 1447 1448 1449 1450 1451
	 */
	if (portal_name[0] == '\0')
		portal = CreatePortal(portal_name, true, true);
	else
		portal = CreatePortal(portal_name, false, false);

1452
	/* We need to output the parameter values someday */
1453 1454
	if (log_statement == LOGSTMT_ALL)
		ereport(LOG,
1455 1456 1457
				(errmsg("statement: <protocol> <BIND> %s  [PREPARE:  %s]",
						*portal_name ? portal_name : "<unnamed>",
						portal->sourceText ? portal->sourceText : "")));
1458

1459 1460 1461 1462 1463
	/*
	 * Fetch parameters, if any, and store in the portal's memory context.
	 */
	if (numParams > 0)
	{
1464
		ListCell   *l;
1465 1466 1467 1468
		MemoryContext oldContext;

		oldContext = MemoryContextSwitchTo(PortalGetHeapMemory(portal));

1469 1470 1471 1472
		/* sizeof(ParamListInfoData) includes the first array element */
		params = (ParamListInfo) palloc(sizeof(ParamListInfoData) +
								(numParams - 1) * sizeof(ParamExternData));
		params->numParams = numParams;
1473

1474
		i = 0;
1475 1476
		foreach(l, pstmt->argtype_list)
		{
1477
			Oid			ptype = lfirst_oid(l);
1478
			int32		plength;
1479
			bool		isNull;
1480 1481 1482
			StringInfoData pbuf;
			char		csave;
			int16		pformat;
1483

1484 1485 1486
			plength = pq_getmsgint(input_message, 4);
			isNull = (plength == -1);

1487 1488
			if (!isNull)
			{
1489
				const char *pvalue = pq_getmsgbytes(input_message, plength);
1490 1491 1492

				/*
				 * Rather than copying data around, we just set up a phony
1493 1494 1495 1496 1497
				 * 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.
1498 1499 1500 1501 1502 1503 1504 1505
				 */
				pbuf.data = (char *) pvalue;
				pbuf.maxlen = plength + 1;
				pbuf.len = plength;
				pbuf.cursor = 0;

				csave = pbuf.data[plength];
				pbuf.data[plength] = '\0';
1506 1507 1508 1509 1510 1511
			}
			else
			{
				pbuf.data = NULL;		/* keep compiler quiet */
				csave = 0;
			}
1512

1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524
			if (numPFormats > 1)
				pformat = pformats[i];
			else if (numPFormats > 0)
				pformat = pformats[0];
			else
				pformat = 0;	/* default = text */

			if (pformat == 0)
			{
				Oid			typinput;
				Oid			typioparam;
				char	   *pstring;
1525

1526
				getTypeInputInfo(ptype, &typinput, &typioparam);
1527

1528 1529 1530 1531 1532 1533 1534
				/*
				 * We have to do encoding conversion before calling the
				 * typinput routine.
				 */
				if (isNull)
					pstring = NULL;
				else
1535
					pstring = pg_client_to_server(pbuf.data, plength);
1536

1537 1538 1539 1540
				params->params[i].value = OidInputFunctionCall(typinput,
															   pstring,
															   typioparam,
															   -1);
1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557
				/* Free result of encoding conversion, if any */
				if (pstring && pstring != pbuf.data)
					pfree(pstring);
			}
			else if (pformat == 1)
			{
				Oid			typreceive;
				Oid			typioparam;
				StringInfo	bufptr;

				/*
				 * Call the parameter type's binary input converter
				 */
				getTypeBinaryInputInfo(ptype, &typreceive, &typioparam);

				if (isNull)
					bufptr = NULL;
1558
				else
1559 1560
					bufptr = &pbuf;

1561 1562 1563 1564
				params->params[i].value = OidReceiveFunctionCall(typreceive,
																 bufptr,
																 typioparam,
																 -1);
1565 1566 1567

				/* Trouble if it didn't eat the whole buffer */
				if (!isNull && pbuf.cursor != pbuf.len)
1568
					ereport(ERROR,
1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579
							(errcode(ERRCODE_INVALID_BINARY_REPRESENTATION),
							 errmsg("incorrect binary data format in bind parameter %d",
									i + 1)));
			}
			else
			{
				ereport(ERROR,
						(errcode(ERRCODE_INVALID_PARAMETER_VALUE),
						 errmsg("unsupported format code: %d",
								pformat)));
			}
1580

1581 1582
			/* Restore message buffer contents */
			if (!isNull)
1583
				pbuf.data[plength] = csave;
1584

1585 1586
			params->params[i].isnull = isNull;
			params->params[i].ptype = ptype;
1587 1588 1589 1590 1591 1592 1593 1594 1595

			i++;
		}

		MemoryContextSwitchTo(oldContext);
	}
	else
		params = NULL;

1596 1597 1598 1599 1600 1601 1602 1603 1604
	/* 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);
	}

1605 1606 1607
	pq_getmsgend(input_message);

	/*
B
Bruce Momjian 已提交
1608 1609
	 * 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.
1610
	 *
B
Bruce Momjian 已提交
1611 1612
	 * This happens only for unnamed statements, and so switching into the
	 * statement context for planning is correct (see notes in
1613
	 * exec_parse_message).
1614
	 */
1615
	if (pstmt->plan_list == NIL && pstmt->query_list != NIL)
1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632
	{
		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);

1633
	PortalStart(portal, params, InvalidSnapshot);
1634

1635 1636 1637 1638 1639
	/*
	 * Apply the result format requests to the portal.
	 */
	PortalSetResultFormat(portal, numRFormats, rformats);

1640 1641 1642
	/*
	 * Send BindComplete.
	 */
1643
	if (whereToSendOutput == DestRemote)
1644 1645 1646 1647 1648 1649 1650 1651 1652
		pq_putemptymessage('2');
}

/*
 * exec_execute_message
 *
 * Process an "Execute" message for a portal
 */
static void
1653
exec_execute_message(const char *portal_name, long max_rows)
1654
{
B
Bruce Momjian 已提交
1655
	CommandDest dest;
1656
	DestReceiver *receiver;
1657 1658 1659
	Portal		portal;
	bool		completed;
	char		completionTag[COMPLETION_TAG_BUFSIZE];
1660
	bool		save_log_statement_stats = log_statement_stats;
1661
	bool		execute_is_fetch = false;
1662 1663 1664

	/* Adjust destination to tell printtup.c what to do */
	dest = whereToSendOutput;
1665 1666
	if (dest == DestRemote)
		dest = DestRemoteExecute;
1667 1668 1669

	portal = GetPortalByName(portal_name);
	if (!PortalIsValid(portal))
1670 1671 1672
		ereport(ERROR,
				(errcode(ERRCODE_UNDEFINED_CURSOR),
				 errmsg("portal \"%s\" does not exist", portal_name)));
1673

1674
	/*
B
Bruce Momjian 已提交
1675 1676 1677 1678
	 * 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.
1679 1680 1681 1682
	 */
	if (!portal->atStart)
		execute_is_fetch = true;

1683
	/*
B
Bruce Momjian 已提交
1684 1685
	 * If the original query was a null string, just return
	 * EmptyQueryResponse.
1686 1687 1688 1689 1690 1691 1692 1693
	 */
	if (portal->commandTag == NULL)
	{
		Assert(portal->parseTrees == NIL);
		NullCommand(dest);
		return;
	}

1694 1695 1696 1697 1698 1699 1700
	/* Should we display the portal names here? */
	if (execute_is_fetch)
	{
		debug_query_string = "fetch message";
		pgstat_report_activity("<FETCH>");
	}
	else if (portal->sourceText)
1701 1702 1703 1704 1705 1706 1707 1708 1709 1710
	{
		debug_query_string = portal->sourceText;
		pgstat_report_activity(portal->sourceText);
	}
	else
	{
		debug_query_string = "execute message";
		pgstat_report_activity("<EXECUTE>");
	}

1711
	set_ps_display(portal->commandTag, false);
1712

1713
	/*
1714 1715
	 * We use save_log_statement_stats so ShowUsage doesn't report incorrect
	 * results because ResetUsage wasn't called.
1716 1717 1718 1719 1720 1721 1722
	 */
	if (save_log_statement_stats)
		ResetUsage();

	if (log_statement == LOGSTMT_ALL)
		/* We have the portal, so output the source query. */
		ereport(LOG,
1723
				(errmsg("statement: <protocol> %sEXECUTE %s  [PREPARE:  %s]",
1724
						execute_is_fetch ? "FETCH from " : "",
1725
						*portal_name ? portal_name : "<unnamed>",
1726 1727
						portal->sourceText ? portal->sourceText : "")));

1728 1729
	BeginCommand(portal->commandTag, dest);

1730
	/*
B
Bruce Momjian 已提交
1731 1732
	 * Create dest receiver in MessageContext (we don't want it in transaction
	 * context, because that may get deleted if portal contains VACUUM).
1733 1734 1735
	 */
	receiver = CreateDestReceiver(dest, portal);

1736
	/*
B
Bruce Momjian 已提交
1737 1738
	 * Ensure we are in a transaction command (this should normally be the
	 * case already due to prior BIND).
1739
	 */
1740
	start_xact_command();
1741 1742 1743 1744 1745

	/*
	 * If we are in aborted transaction state, the only portals we can
	 * actually run are those containing COMMIT or ROLLBACK commands.
	 */
1746 1747 1748 1749 1750
	if (IsAbortedTransactionBlockState() &&
		!IsTransactionExitStmtList(portal->parseTrees))
		ereport(ERROR,
				(errcode(ERRCODE_IN_FAILED_SQL_TRANSACTION),
				 errmsg("current transaction is aborted, "
B
Bruce Momjian 已提交
1751
						"commands ignored until end of transaction block")));
1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763

	/* 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,
1764 1765
						  receiver,
						  receiver,
1766 1767
						  completionTag);

1768
	(*receiver->rDestroy) (receiver);
1769

1770 1771
	if (completed)
	{
1772
		if (IsTransactionStmtList(portal->parseTrees))
1773 1774
		{
			/*
B
Bruce Momjian 已提交
1775
			 * If this was a transaction control statement, commit it.	We
B
Bruce Momjian 已提交
1776
			 * will start a new xact command for the next command (if any).
1777
			 */
1778
			finish_xact_command();
1779 1780 1781 1782
		}
		else
		{
			/*
B
Bruce Momjian 已提交
1783 1784
			 * We need a CommandCounterIncrement after every query, except
			 * those that start or end a transaction block.
1785 1786 1787 1788 1789 1790 1791 1792 1793 1794
			 */
			CommandCounterIncrement();
		}

		/* Send appropriate CommandComplete to client */
		EndCommand(completionTag, dest);
	}
	else
	{
		/* Portal run not complete, so send PortalSuspended */
1795
		if (whereToSendOutput == DestRemote)
1796 1797 1798
			pq_putemptymessage('s');
	}

1799
	/*
1800
	 * Emit duration logging if appropriate.
1801
	 */
1802
	if (log_duration || log_min_duration_statement >= 0)
1803
	{
1804 1805 1806
		long		secs;
		int			usecs;
		int			msecs;
1807

1808 1809 1810 1811
		TimestampDifference(GetCurrentStatementStartTimestamp(),
							GetCurrentTimestamp(),
							&secs, &usecs);
		msecs = usecs / 1000;
1812 1813

		/*
1814 1815 1816 1817
		 * The odd-looking test for log_min_duration_statement being
		 * exceeded is designed to avoid integer overflow with very
		 * long durations: don't compute secs * 1000 until we've
		 * verified it will fit in int.
1818
		 */
1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830
		if (log_duration ||
			log_min_duration_statement == 0 ||
			(log_min_duration_statement > 0 &&
			 (secs > log_min_duration_statement / 1000 ||
			  secs * 1000 + msecs >= log_min_duration_statement)))
		{
			if (log_statement == LOGSTMT_ALL)	/* already logged? */
				ereport(LOG,
						(errmsg("duration: %ld.%03d ms",
								secs, msecs)));
			else
				ereport(LOG,
1831
						(errmsg("duration: %ld.%03d ms  statement: <protocol> %sEXECUTE %s  [PREPARE:  %s]",
1832 1833
								secs, msecs,
								execute_is_fetch ? "FETCH from " : "",
1834 1835
								*portal_name ? portal_name : "<unnamed>",
								portal->sourceText ? portal->sourceText : "")));
1836
		}
1837 1838 1839 1840 1841
	}

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

1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853
	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;
1854
	TupleDesc	tupdesc;
1855
	ListCell   *l;
1856 1857
	StringInfoData buf;

1858 1859 1860 1861
	/*
	 * Start up a transaction command. (Note that this will normally change
	 * current memory context.) Nothing happens if we are already in one.
	 */
1862
	start_xact_command();
1863 1864 1865 1866

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

1867 1868 1869 1870 1871 1872 1873 1874
	/* 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)
1875 1876
			ereport(ERROR,
					(errcode(ERRCODE_UNDEFINED_PSTATEMENT),
B
Bruce Momjian 已提交
1877
					 errmsg("unnamed prepared statement does not exist")));
1878 1879
	}

1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895
	/*
	 * If we are in aborted transaction state, we can't safely create a result
	 * tupledesc, because that needs catalog accesses.  Hence, refuse to
	 * Describe statements that return data.  (We shouldn't just refuse all
	 * Describes, since that might break the ability of some clients to issue
	 * COMMIT or ROLLBACK commands, if they use code that blindly Describes
	 * whatever it does.)  We can Describe parameters without doing anything
	 * dangerous, so we don't restrict that.
	 */
	if (IsAbortedTransactionBlockState() &&
		PreparedStatementReturnsTuples(pstmt))
		ereport(ERROR,
				(errcode(ERRCODE_IN_FAILED_SQL_TRANSACTION),
				 errmsg("current transaction is aborted, "
						"commands ignored until end of transaction block")));

1896
	if (whereToSendOutput != DestRemote)
1897 1898
		return;					/* can't actually do anything... */

1899 1900 1901
	/*
	 * First describe the parameters...
	 */
B
Bruce Momjian 已提交
1902
	pq_beginmessage(&buf, 't'); /* parameter description message type */
1903
	pq_sendint(&buf, list_length(pstmt->argtype_list), 2);
1904 1905 1906

	foreach(l, pstmt->argtype_list)
	{
1907
		Oid			ptype = lfirst_oid(l);
1908 1909 1910 1911

		pq_sendint(&buf, (int) ptype, 4);
	}
	pq_endmessage(&buf);
1912 1913 1914 1915 1916 1917

	/*
	 * Next send RowDescription or NoData to describe the result...
	 */
	tupdesc = FetchPreparedStatementResultDesc(pstmt);
	if (tupdesc)
1918 1919 1920
		SendRowDescriptionMessage(tupdesc,
								  FetchPreparedStatementTargetList(pstmt),
								  NULL);
1921 1922 1923
	else
		pq_putemptymessage('n');	/* NoData */

1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935
}

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

1936 1937 1938 1939
	/*
	 * Start up a transaction command. (Note that this will normally change
	 * current memory context.) Nothing happens if we are already in one.
	 */
1940
	start_xact_command();
1941 1942 1943 1944

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

1945 1946
	portal = GetPortalByName(portal_name);
	if (!PortalIsValid(portal))
1947 1948 1949
		ereport(ERROR,
				(errcode(ERRCODE_UNDEFINED_CURSOR),
				 errmsg("portal \"%s\" does not exist", portal_name)));
1950

1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965
	/*
	 * If we are in aborted transaction state, we can't run
	 * SendRowDescriptionMessage(), because that needs catalog accesses.
	 * Hence, refuse to Describe portals that return data.  (We shouldn't just
	 * refuse all Describes, since that might break the ability of some
	 * clients to issue COMMIT or ROLLBACK commands, if they use code that
	 * blindly Describes whatever it does.)
	 */
	if (IsAbortedTransactionBlockState() &&
		portal->tupDesc)
		ereport(ERROR,
				(errcode(ERRCODE_IN_FAILED_SQL_TRANSACTION),
				 errmsg("current transaction is aborted, "
						"commands ignored until end of transaction block")));

1966
	if (whereToSendOutput != DestRemote)
1967 1968 1969
		return;					/* can't actually do anything... */

	if (portal->tupDesc)
1970 1971
		SendRowDescriptionMessage(portal->tupDesc,
								  FetchPortalTargetList(portal),
1972
								  portal->formats);
1973 1974 1975 1976 1977
	else
		pq_putemptymessage('n');	/* NoData */
}


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

		/* Set statement timeout running, if any */
1991
		/* NB: this mustn't be enabled until we are within an xact */
1992 1993
		if (StatementTimeout > 0)
			enable_sig_alarm(StatementTimeout, true);
1994 1995
		else
			cancel_from_timeout = false;
B
Bruce Momjian 已提交
1996

1997 1998 1999 2000 2001 2002 2003 2004 2005
		xact_started = true;
	}
}

static void
finish_xact_command(void)
{
	if (xact_started)
	{
2006 2007 2008 2009
		/* Cancel any active statement timeout before committing */
		disable_sig_alarm(true);

		/* Now commit the command */
2010 2011
		ereport(DEBUG3,
				(errmsg_internal("CommitTransactionCommand")));
2012

2013
		CommitTransactionCommand();
2014

2015 2016 2017 2018 2019 2020
#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

2021
#ifdef SHOW_MEMORY_STATS
2022
		/* Print mem stats after each commit for leak tracking */
2023 2024
		if (ShowStats)
			MemoryContextStats(TopMemoryContext);
2025
#endif
2026 2027 2028

		xact_started = false;
	}
2029 2030 2031
}


2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081
/*
 * Convenience routines for checking whether a statement is one of the
 * ones that we allow in transaction-aborted state.
 */

static bool
IsTransactionExitStmt(Node *parsetree)
{
	if (parsetree && IsA(parsetree, TransactionStmt))
	{
		TransactionStmt *stmt = (TransactionStmt *) parsetree;

		if (stmt->kind == TRANS_STMT_COMMIT ||
			stmt->kind == TRANS_STMT_PREPARE ||
			stmt->kind == TRANS_STMT_ROLLBACK ||
			stmt->kind == TRANS_STMT_ROLLBACK_TO)
			return true;
	}
	return false;
}

static bool
IsTransactionExitStmtList(List *parseTrees)
{
	if (list_length(parseTrees) == 1)
	{
		Query	   *query = (Query *) linitial(parseTrees);

		if (query->commandType == CMD_UTILITY &&
			IsTransactionExitStmt(query->utilityStmt))
			return true;
	}
	return false;
}

static bool
IsTransactionStmtList(List *parseTrees)
{
	if (list_length(parseTrees) == 1)
	{
		Query	   *query = (Query *) linitial(parseTrees);

		if (query->commandType == CMD_UTILITY &&
			query->utilityStmt && IsA(query->utilityStmt, TransactionStmt))
			return true;
	}
	return false;
}


2082
/* --------------------------------
2083
 *		signal handler routines used in PostgresMain()
2084 2085 2086
 * --------------------------------
 */

2087
/*
T
Tom Lane 已提交
2088
 * quickdie() occurs when signalled SIGQUIT by the postmaster.
2089 2090 2091 2092
 *
 * Some backend has bought the farm,
 * so we need to stop what we're doing and exit.
 */
T
Tom Lane 已提交
2093
void
2094
quickdie(SIGNAL_ARGS)
2095
{
2096
	PG_SETMASK(&BlockSig);
B
Bruce Momjian 已提交
2097

2098
	/*
B
Bruce Momjian 已提交
2099 2100
	 * Ideally this should be ereport(FATAL), but then we'd not get control
	 * back...
2101 2102 2103
	 */
	ereport(WARNING,
			(errcode(ERRCODE_CRASH_SHUTDOWN),
B
Bruce Momjian 已提交
2104
			 errmsg("terminating connection because of crash of another server process"),
B
Bruce Momjian 已提交
2105 2106 2107 2108
	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."),
2109
			 errhint("In a moment you should be able to reconnect to the"
2110
					 " database and repeat your command.")));
B
Bruce Momjian 已提交
2111

2112
	/*
2113
	 * DO NOT proc_exit() -- we're here because shared memory may be
B
Bruce Momjian 已提交
2114 2115
	 * corrupted, so we don't want to try to clean up our transaction. Just
	 * nail the windows shut and get out of town.
2116
	 *
B
Bruce Momjian 已提交
2117 2118 2119 2120
	 * 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.
2121
	 */
2122
	exit(1);
2123 2124
}

2125
/*
2126 2127
 * Shutdown signal from postmaster: abort transaction and exit
 * at soonest convenient time
2128
 */
2129
void
2130
die(SIGNAL_ARGS)
2131
{
2132 2133 2134
	int			save_errno = errno;

	/* Don't joggle the elbow of proc_exit */
B
Bruce Momjian 已提交
2135
	if (!proc_exit_inprogress)
2136
	{
2137
		InterruptPending = true;
2138
		ProcDiePending = true;
B
Bruce Momjian 已提交
2139

2140
		/*
B
Bruce Momjian 已提交
2141 2142
		 * If it's safe to interrupt, and we're waiting for input or a lock,
		 * service the interrupt immediately
2143
		 */
2144 2145
		if (ImmediateInterruptOK && InterruptHoldoffCount == 0 &&
			CritSectionCount == 0)
2146
		{
2147 2148 2149
			/* bump holdoff count to make ProcessInterrupts() a no-op */
			/* until we are done getting ready for it */
			InterruptHoldoffCount++;
2150
			DisableNotifyInterrupt();
2151
			DisableCatchupInterrupt();
2152
			/* Make sure CheckDeadLock won't run while shutting down... */
2153
			LockWaitCancel();
2154
			InterruptHoldoffCount--;
2155 2156
			ProcessInterrupts();
		}
2157
	}
2158 2159

	errno = save_errno;
2160 2161
}

2162
/*
2163
 * Timeout or shutdown signal from postmaster during client authentication.
2164
 * Simply exit(0).
2165 2166 2167
 *
 * XXX: possible future improvement: try to send a message indicating
 * why we are disconnecting.  Problem is to be sure we don't block while
2168
 * doing so, nor mess up the authentication message exchange.
2169 2170 2171 2172 2173 2174 2175
 */
void
authdie(SIGNAL_ARGS)
{
	exit(0);
}

2176
/*
2177 2178
 * Query-cancel signal from postmaster: abort current transaction
 * at soonest convenient time
2179
 */
2180
void
2181
StatementCancelHandler(SIGNAL_ARGS)
2182
{
2183 2184
	int			save_errno = errno;

B
Bruce Momjian 已提交
2185
	/*
2186
	 * Don't joggle the elbow of proc_exit
B
Bruce Momjian 已提交
2187
	 */
2188
	if (!proc_exit_inprogress)
2189
	{
2190 2191
		InterruptPending = true;
		QueryCancelPending = true;
B
Bruce Momjian 已提交
2192

2193
		/*
B
Bruce Momjian 已提交
2194 2195 2196
		 * 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.
2197
		 */
2198
		if (ImmediateInterruptOK && InterruptHoldoffCount == 0 &&
2199
			CritSectionCount == 0)
2200
		{
2201 2202 2203 2204 2205 2206
			/* bump holdoff count to make ProcessInterrupts() a no-op */
			/* until we are done getting ready for it */
			InterruptHoldoffCount++;
			if (LockWaitCancel())
			{
				DisableNotifyInterrupt();
2207
				DisableCatchupInterrupt();
T
Tom Lane 已提交
2208
				InterruptHoldoffCount--;
2209 2210 2211 2212
				ProcessInterrupts();
			}
			else
				InterruptHoldoffCount--;
2213
		}
2214 2215
	}

2216
	errno = save_errno;
2217 2218
}

2219
/* signal handler for floating point exception */
2220
void
2221 2222
FloatExceptionHandler(SIGNAL_ARGS)
{
2223 2224 2225
	ereport(ERROR,
			(errcode(ERRCODE_FLOATING_POINT_EXCEPTION),
			 errmsg("floating-point exception"),
B
Bruce Momjian 已提交
2226 2227 2228
			 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.")));
2229 2230
}

2231
/* SIGHUP: set flag to re-read config file at next convenient time */
2232
static void
2233
SigHupHandler(SIGNAL_ARGS)
2234
{
2235
	got_SIGHUP = true;
2236 2237
}

2238

2239 2240 2241 2242 2243 2244 2245 2246 2247 2248
/*
 * 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)
{
2249 2250
	/* OK to accept interrupt now? */
	if (InterruptHoldoffCount != 0 || CritSectionCount != 0)
2251 2252 2253 2254 2255
		return;
	InterruptPending = false;
	if (ProcDiePending)
	{
		ProcDiePending = false;
B
Bruce Momjian 已提交
2256 2257
		QueryCancelPending = false;		/* ProcDie trumps QueryCancel */
		ImmediateInterruptOK = false;	/* not idle anymore */
2258
		DisableNotifyInterrupt();
2259
		DisableCatchupInterrupt();
2260 2261
		ereport(FATAL,
				(errcode(ERRCODE_ADMIN_SHUTDOWN),
B
Bruce Momjian 已提交
2262
			 errmsg("terminating connection due to administrator command")));
2263 2264 2265 2266
	}
	if (QueryCancelPending)
	{
		QueryCancelPending = false;
B
Bruce Momjian 已提交
2267
		ImmediateInterruptOK = false;	/* not idle anymore */
2268
		DisableNotifyInterrupt();
2269
		DisableCatchupInterrupt();
2270 2271 2272 2273 2274 2275 2276 2277
		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")));
2278 2279 2280 2281
	}
	/* If we get here, do nothing (probably, QueryCancelPending was reset) */
}

2282

2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295
/*
 * 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 已提交
2296 2297
	char		stack_top_loc;
	int			stack_depth;
2298 2299 2300 2301

	/*
	 * Compute distance from PostgresMain's local variables to my own
	 *
B
Bruce Momjian 已提交
2302 2303 2304
	 * 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.
2305 2306
	 */
	stack_depth = (int) (stack_base_ptr - &stack_top_loc);
B
Bruce Momjian 已提交
2307

2308
	/*
B
Bruce Momjian 已提交
2309
	 * Take abs value, since stacks grow up on some machines, down on others
2310 2311 2312
	 */
	if (stack_depth < 0)
		stack_depth = -stack_depth;
B
Bruce Momjian 已提交
2313

2314 2315 2316
	/*
	 * Trouble?
	 *
2317 2318 2319 2320
	 * 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.
2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342
	 */
	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;
}


2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377
/*
 * 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);
}


2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437
bool
set_plan_disabling_options(const char *arg, GucContext context, GucSource source)
{
	char *tmp = NULL;

	switch (arg[0])
	{
		case 's':	/* seqscan */
			tmp = "enable_seqscan";
			break;
		case 'i':	/* indexscan */
			tmp = "enable_indexscan";
			break;
		case 'b':	/* bitmapscan */
			tmp = "enable_bitmapscan";
			break;
		case 't':	/* tidscan */
			tmp = "enable_tidscan";
			break;
		case 'n':	/* nestloop */
			tmp = "enable_nestloop";
			break;
		case 'm':	/* mergejoin */
			tmp = "enable_mergejoin";
			break;
		case 'h':	/* hashjoin */
			tmp = "enable_hashjoin";
			break;
	}
	if (tmp)
	{
		SetConfigOption(tmp, "false", context, source);
		return true;
	}
	else
		return false;
}


const char *
get_stats_option_name(const char *arg)
{
	switch (arg[0])
	{
		case 'p':
			if (optarg[1] == 'a') /* "parser" */
				return "log_parser_stats";
			else if (optarg[1] == 'l') /* "planner" */
				return "log_planner_stats";
			break;

		case 'e':	/* "executor" */
			return "log_executor_stats";
			break;
	}

	return NULL;
}


2438
/* ----------------------------------------------------------------
2439
 * PostgresMain
B
Bruce Momjian 已提交
2440
 *	   postgres main loop -- all backends, interactive or otherwise start here
2441
 *
2442 2443 2444 2445
 * 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.
2446 2447 2448
 * ----------------------------------------------------------------
 */
int
2449
PostgresMain(int argc, char *argv[], const char *username)
2450
{
2451
	int			flag;
2452
	const char *dbname = NULL;
2453
	char	   *userDoption = NULL;
2454
	bool		secure;
2455
	int			errs = 0;
B
Bruce Momjian 已提交
2456 2457
	int			debug_flag = -1;	/* -1 means not given */
	List	   *guc_names = NIL;	/* for SUSET options */
2458 2459
	List	   *guc_values = NIL;
	GucContext	ctx;
2460
	GucSource	gucsource;
2461
	bool		am_superuser;
2462
	int			firstchar;
2463
	char		stack_base;
B
Bruce Momjian 已提交
2464
	StringInfoData input_message;
2465
	sigjmp_buf	local_sigjmp_buf;
2466
	volatile bool send_ready_for_query = true;
B
Bruce Momjian 已提交
2467

2468 2469 2470 2471
#define PendingConfigOption(name,val) \
	(guc_names = lappend(guc_names, pstrdup(name)), \
	 guc_values = lappend(guc_values, pstrdup(val)))

2472 2473 2474 2475 2476 2477
	/*
	 * initialize globals (already done if under postmaster, but not if
	 * standalone; cheap enough to do over)
	 */
	MyProcPid = getpid();

2478 2479 2480 2481 2482
	/*
	 * Fire up essential subsystems: error and memory management
	 *
	 * If we are running under the postmaster, this is done already.
	 */
2483
	if (!IsUnderPostmaster)
2484 2485
		MemoryContextInit();

2486
	set_ps_display("startup", false);
2487

2488 2489
	SetProcessingMode(InitProcessing);

2490 2491 2492
	/* Set up reference point for stack depth checking */
	stack_base_ptr = &stack_base;

2493 2494 2495 2496 2497 2498 2499
	/* 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 已提交
2500

B
Bruce Momjian 已提交
2501 2502
	if (pkglib_path[0] == '\0')
		get_pkglib_path(my_exec_path, pkglib_path);
2503

2504
	/*
2505
	 * Set default values for command-line options.
2506
	 */
2507
	EchoQuery = false;
2508

2509
	if (!IsUnderPostmaster)
2510
		InitializeGUCOptions();
2511

2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528
	/* ----------------
	 *	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.
	 * ----------------
	 */

2529 2530 2531 2532 2533 2534 2535
	/* Ignore the initial --single argument, if present */
	if (argc > 1 && strcmp(argv[1], "--single") == 0)
	{
		argv++;
		argc--;
	}

2536 2537
	/* all options are allowed until '-p' */
	secure = true;
2538
	ctx = PGC_POSTMASTER;
2539
	gucsource = PGC_S_ARGV;		/* initial switches came from command line */
2540

2541
	while ((flag = getopt(argc, argv, "A:B:c:D:d:EeFf:h:ijk:lN:nOo:Pp:r:S:sTt:v:W:y:-:")) != -1)
2542
	{
2543 2544
		switch (flag)
		{
M
 
Marc G. Fournier 已提交
2545
			case 'A':
2546
				SetConfigOption("debug_assertions", optarg, ctx, gucsource);
M
 
Marc G. Fournier 已提交
2547
				break;
2548

2549
			case 'B':
2550
				SetConfigOption("shared_buffers", optarg, ctx, gucsource);
2551
				break;
2552

2553
			case 'D':
2554
				if (secure)
2555
					userDoption = optarg;
M
 
Marc G. Fournier 已提交
2556
				break;
2557

2558
			case 'd':
2559
				debug_flag = atoi(optarg);
2560
				break;
2561 2562

			case 'E':
2563
				EchoQuery = true;
2564
				break;
2565 2566

			case 'e':
2567
				SetConfigOption("datestyle", "euro", ctx, gucsource);
2568
				break;
2569 2570

			case 'F':
2571
				SetConfigOption("fsync", "false", ctx, gucsource);
2572
				break;
2573 2574

			case 'f':
2575 2576 2577
				if (!set_plan_disabling_options(optarg, ctx, gucsource))
					errs++;
				break;
2578

2579 2580
			case 'h':
				SetConfigOption("listen_addresses", optarg, ctx, gucsource);
2581 2582
				break;

2583 2584 2585
			case 'i':
				SetConfigOption("listen_addresses", "*", ctx, gucsource);
				break;
2586

2587
			case 'j':
2588 2589
				UseNewLine = 0;
				break;
2590

2591 2592 2593
			case 'k':
				SetConfigOption("unix_socket_directory", optarg, ctx, gucsource);
				break;
2594

2595 2596
			case 'l':
				SetConfigOption("ssl", "true", ctx, gucsource);
2597 2598
				break;

2599 2600 2601
			case 'N':
				SetConfigOption("max_connections", optarg, ctx, gucsource);
				break;
2602

2603 2604 2605 2606 2607 2608
			case 'n':
				/* ignored for consistency with postmaster */
				break;

			case 'O':
				SetConfigOption("allow_system_table_mods", "true", ctx, gucsource);
H
Hiroshi Inoue 已提交
2609 2610
				break;

T
Tom Lane 已提交
2611
			case 'o':
2612 2613
				errs++;
				break;
2614

2615 2616
			case 'P':
				SetConfigOption("ignore_system_indexes", "true", ctx, gucsource);
T
Tom Lane 已提交
2617 2618
				break;

2619
			case 'p':
2620 2621
				SetConfigOption("port", optarg, ctx, gucsource);
				break;
B
Bruce Momjian 已提交
2622

2623 2624
			case 'r':
				/* send output (stdout and stderr) to the given file */
2625
				if (secure)
2626
					StrNCpy(OutputFileName, optarg, MAXPGPATH);
2627
				break;
2628

2629
			case 'S':
2630
				SetConfigOption("work_mem", optarg, ctx, gucsource);
2631
				break;
2632 2633

			case 's':
2634
				/*
2635 2636
				 * Since log options are SUSET, we need to postpone unless
				 * still in secure context
2637
				 */
2638 2639 2640 2641 2642
				if (ctx == PGC_BACKEND)
					PendingConfigOption("log_statement_stats", "true");
				else
					SetConfigOption("log_statement_stats", "true",
									ctx, gucsource);
M
 
Marc G. Fournier 已提交
2643 2644
				break;

2645 2646 2647 2648
			case 'T':
				/* ignored for consistency with postmaster */
				break;

2649
			case 't':
2650 2651
			{
				const char *tmp = get_stats_option_name(optarg);
2652
				if (tmp)
2653 2654 2655 2656 2657 2658
				{
					if (ctx == PGC_BACKEND)
						PendingConfigOption(tmp, "true");
					else
						SetConfigOption(tmp, "true", ctx, gucsource);
				}
2659 2660
				else
					errs++;
2661
				break;
2662
			}
2663

2664
			case 'v':
2665 2666
				if (secure)
					FrontendProtocol = (ProtocolVersion) atoi(optarg);
2667 2668
				break;

M
 
Marc G. Fournier 已提交
2669
			case 'W':
2670 2671
				SetConfigOption("post_auth_delay", optarg, ctx, gucsource);
				break;
2672

2673 2674

			case 'y':
2675
				/*
2676 2677
				 * y - special flag passed if backend was forked by a
				 * postmaster.
M
 
Marc G. Fournier 已提交
2678
				 */
2679 2680 2681 2682 2683 2684 2685 2686
				if (secure)
				{
					dbname = strdup(optarg);

					secure = false;		/* subsequent switches are NOT secure */
					ctx = PGC_BACKEND;
					gucsource = PGC_S_CLIENT;
				}
M
 
Marc G. Fournier 已提交
2687 2688
				break;

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

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

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

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

2730
	/*
B
Bruce Momjian 已提交
2731 2732
	 * Process any additional GUC variable settings passed in startup packet.
	 * These are handled exactly like command-line variables.
2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755
	 */
	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);
		}
	}

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

2767 2768 2769
	if (PostAuthDelay)
		pg_usleep(PostAuthDelay * 1000000L);

2770
	/*
2771
	 * Set up signal handlers and masks.
2772
	 *
2773 2774 2775
	 * 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 已提交
2776
	 *
B
Bruce Momjian 已提交
2777 2778 2779 2780 2781 2782
	 * 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.)
2783
	 */
2784
	pqsignal(SIGHUP, SigHupHandler);	/* set flag to read config file */
B
Bruce Momjian 已提交
2785
	pqsignal(SIGINT, StatementCancelHandler);	/* cancel current query */
2786
	pqsignal(SIGTERM, die);		/* cancel current query and exit */
2787
	pqsignal(SIGQUIT, quickdie);	/* hard crash time */
B
Bruce Momjian 已提交
2788
	pqsignal(SIGALRM, handle_sig_alarm);		/* timeout conditions */
2789 2790 2791 2792

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

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

2806 2807
	pqinitmask();

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

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

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

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

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

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

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

2866
		BaseInit();
2867 2868 2869 2870 2871

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

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

		/*
B
Bruce Momjian 已提交
2883 2884
		 * 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.
2885 2886
		 */
		BuildFlatFiles(true);
2887 2888
	}

2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901
	/*
	 * Create a per-backend PGPROC struct in shared memory, except in
	 * the EXEC_BACKEND case where this was done in SubPostmasterMain.
	 * We must do this before we can use LWLocks (and in the EXEC_BACKEND
	 * case we already had to do some stuff with LWLocks).
	 */
#ifdef EXEC_BACKEND
	if (!IsUnderPostmaster)
		InitProcess();
#else
	InitProcess();
#endif

2902
	/*
2903 2904
	 * General initialization.
	 *
2905 2906 2907
	 * 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.
2908
	 */
2909 2910
	ereport(DEBUG3,
			(errmsg_internal("InitPostgres")));
2911
	am_superuser = InitPostgres(dbname, username);
2912

2913
	SetProcessingMode(NormalProcessing);
2914

2915
	/*
2916 2917
	 * Now that we know if client is a superuser, we can try to apply SUSET
	 * GUC options that came from the client.
2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945
	 */
	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();

2946
	/*
B
Bruce Momjian 已提交
2947 2948
	 * Also set up handler to log session end; we have to wait till now to be
	 * sure Log_disconnections has its final value.
2949 2950 2951 2952
	 */
	if (IsUnderPostmaster && Log_disconnections)
		on_proc_exit(log_disconnections, 0);

2953 2954
	/*
	 * Send this backend's cancellation info to the frontend.
2955
	 */
2956
	if (whereToSendOutput == DestRemote &&
M
 
Marc G. Fournier 已提交
2957 2958
		PG_PROTOCOL_MAJOR(FrontendProtocol) >= 2)
	{
2959
		StringInfoData buf;
B
Bruce Momjian 已提交
2960

2961
		pq_beginmessage(&buf, 'K');
2962 2963 2964
		pq_sendint(&buf, (int32) MyProcPid, sizeof(int32));
		pq_sendint(&buf, (int32) MyCancelKey, sizeof(int32));
		pq_endmessage(&buf);
M
 
Marc G. Fournier 已提交
2965 2966 2967
		/* Need not flush since ReadyForQuery will do it. */
	}

2968
	/* Welcome banner for standalone case */
2969
	if (whereToSendOutput == DestDebug)
2970
		printf("\nPostgreSQL stand-alone backend %s\n", PG_VERSION);
2971

2972 2973 2974
	/*
	 * Create the memory context we will use in the main loop.
	 *
2975 2976
	 * MessageContext is reset once per iteration of the main loop, ie, upon
	 * completion of processing of each command message from the client.
2977
	 */
2978 2979 2980 2981 2982
	MessageContext = AllocSetContextCreate(TopMemoryContext,
										   "MessageContext",
										   ALLOCSET_DEFAULT_MINSIZE,
										   ALLOCSET_DEFAULT_INITSIZE,
										   ALLOCSET_DEFAULT_MAXSIZE);
2983

2984
	/*
2985
	 * Remember stand-alone backend startup time
2986 2987
	 */
	if (!IsUnderPostmaster)
2988
		PgStartTime = GetCurrentTimestamp();
2989

2990 2991
	/*
	 * POSTGRES main processing loop begins here
2992
	 *
B
Bruce Momjian 已提交
2993 2994
	 * If an exception is encountered, processing resumes here so we abort the
	 * current transaction and start a new one.
2995
	 *
2996 2997 2998 2999 3000 3001 3002
	 * 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.)
3003 3004
	 */

3005
	if (sigsetjmp(local_sigjmp_buf, 1) != 0)
3006
	{
3007
		/*
3008
		 * NOTE: if you are tempted to add more code in this if-block,
3009
		 * consider the high probability that it should be in
B
Bruce Momjian 已提交
3010
		 * AbortTransaction() instead.	The only stuff done directly here
B
Bruce Momjian 已提交
3011 3012
		 * should be stuff that is guaranteed to apply *only* for outer-level
		 * error recovery, such as adjusting the FE/BE protocol status.
3013 3014 3015 3016 3017 3018 3019 3020 3021
		 */

		/* 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 已提交
3022 3023
		 * Forget any pending QueryCancel request, since we're returning to
		 * the idle loop anyway, and cancel the statement timer if running.
3024 3025
		 */
		QueryCancelPending = false;
3026
		disable_sig_alarm(true);
B
Bruce Momjian 已提交
3027
		QueryCancelPending = false;		/* again in case timeout occurred */
3028 3029

		/*
B
Bruce Momjian 已提交
3030 3031 3032
		 * 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.
3033
		 */
3034
		DoingCommandRead = false;
3035
		DisableNotifyInterrupt();
3036
		DisableCatchupInterrupt();
3037

3038 3039 3040
		/* Make sure libpq is in a good state */
		pq_comm_reset();

3041 3042
		/* Report the error to the client and/or server log */
		EmitErrorReport();
3043

3044
		/*
B
Bruce Momjian 已提交
3045 3046
		 * Make sure debug_query_string gets reset before we possibly clobber
		 * the storage it points at.
3047
		 */
3048
		debug_query_string = NULL;
3049

3050
		/*
3051
		 * Abort the current transaction in order to recover.
3052
		 */
3053
		AbortCurrentTransaction();
3054 3055

		/*
B
Bruce Momjian 已提交
3056 3057
		 * Now return to normal top-level context and clear ErrorContext for
		 * next time.
3058 3059
		 */
		MemoryContextSwitchTo(TopMemoryContext);
3060
		FlushErrorState();
3061
		QueryContext = NULL;
3062

3063
		/*
B
Bruce Momjian 已提交
3064 3065
		 * If we were handling an extended-query-protocol message, initiate
		 * skip till next Sync.  This also causes us not to issue
B
Bruce Momjian 已提交
3066
		 * ReadyForQuery (until we get Sync).
3067 3068 3069
		 */
		if (doing_extended_query_message)
			ignore_till_sync = true;
3070

3071 3072
		/* We don't have a transaction command open anymore */
		xact_started = false;
3073

3074
		/* Now we can allow interrupts again */
3075
		RESUME_INTERRUPTS();
3076
	}
3077

3078 3079
	/* We can now handle ereport(ERROR) */
	PG_exception_stack = &local_sigjmp_buf;
3080

3081
	PG_SETMASK(&UnBlockSig);
3082

3083
	if (!ignore_till_sync)
3084
		send_ready_for_query = true;		/* initially, or after error */
3085

3086 3087
	/*
	 * Non-error queries loop here.
3088 3089 3090 3091
	 */

	for (;;)
	{
3092
		/*
B
Bruce Momjian 已提交
3093 3094
		 * At top of loop, reset extended-query-message flag, so that any
		 * errors encountered in "idle" state don't provoke skip.
3095 3096 3097
		 */
		doing_extended_query_message = false;

3098
		/*
B
Bruce Momjian 已提交
3099 3100
		 * Release storage left over from prior query cycle, and create a new
		 * query input buffer in the cleared MessageContext.
3101
		 */
3102 3103
		MemoryContextSwitchTo(MessageContext);
		MemoryContextResetAndDeleteChildren(MessageContext);
3104

3105
		initStringInfo(&input_message);
3106

3107
		/*
B
Bruce Momjian 已提交
3108 3109
		 * (1) If we've reached idle state, tell the frontend we're ready for
		 * a new query.
3110
		 *
3111
		 * Note: this includes fflush()'ing the last of the prior output.
3112 3113 3114
		 *
		 * 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 已提交
3115 3116 3117
		 * 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 已提交
3118
		 */
3119
		if (send_ready_for_query)
3120
		{
3121
			if (IsTransactionOrTransactionBlock())
3122
			{
3123
				set_ps_display("idle in transaction", false);
3124 3125 3126 3127
				pgstat_report_activity("<IDLE> in transaction");
			}
			else
			{
3128 3129
				pgstat_report_tabstat();

3130
				set_ps_display("idle", false);
3131 3132
				pgstat_report_activity("<IDLE>");
			}
3133

3134
			ReadyForQuery(whereToSendOutput);
3135
			send_ready_for_query = false;
3136
		}
3137

3138
		/*
B
Bruce Momjian 已提交
3139 3140 3141 3142
		 * (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.)
3143
		 */
3144 3145
		QueryCancelPending = false;		/* forget any earlier CANCEL signal */
		DoingCommandRead = true;
3146

3147 3148
		/*
		 * (3) read a command (loop blocks here)
3149
		 */
3150
		firstchar = ReadCommand(&input_message);
3151

3152 3153
		/*
		 * (4) disable async signal conditions again.
3154
		 */
3155
		DoingCommandRead = false;
3156

3157
		/*
B
Bruce Momjian 已提交
3158 3159
		 * (5) check for any other interesting events that happened while we
		 * slept.
3160 3161 3162 3163 3164 3165 3166
		 */
		if (got_SIGHUP)
		{
			got_SIGHUP = false;
			ProcessConfigFile(PGC_SIGHUP);
		}

3167
		/*
B
Bruce Momjian 已提交
3168 3169
		 * (6) process the command.  But ignore it if we're skipping till
		 * Sync.
3170
		 */
3171
		if (ignore_till_sync && firstchar != EOF)
3172 3173
			continue;

3174 3175
		switch (firstchar)
		{
3176
			case 'Q':			/* simple query */
3177
				{
3178 3179
					const char *query_string;

3180 3181 3182
					/* Set statement_timestamp() */
					SetCurrentStatementStartTimestamp();

3183 3184
					query_string = pq_getmsgstring(&input_message);
					pq_getmsgend(&input_message);
3185

3186
					exec_simple_query(query_string);
3187

3188
					send_ready_for_query = true;
3189
				}
3190 3191
				break;

3192 3193 3194 3195 3196 3197 3198
			case 'P':			/* parse */
				{
					const char *stmt_name;
					const char *query_string;
					int			numParams;
					Oid		   *paramTypes = NULL;

3199 3200 3201
					/* Set statement_timestamp() */
					SetCurrentStatementStartTimestamp();

3202 3203 3204
					stmt_name = pq_getmsgstring(&input_message);
					query_string = pq_getmsgstring(&input_message);
					numParams = pq_getmsgint(&input_message, 2);
3205 3206
					if (numParams > 0)
					{
B
Bruce Momjian 已提交
3207
						int			i;
3208 3209 3210

						paramTypes = (Oid *) palloc(numParams * sizeof(Oid));
						for (i = 0; i < numParams; i++)
3211
							paramTypes[i] = pq_getmsgint(&input_message, 4);
3212
					}
3213
					pq_getmsgend(&input_message);
3214 3215 3216 3217 3218 3219 3220

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

			case 'B':			/* bind */
3221 3222
				/* Set statement_timestamp() */
				SetCurrentStatementStartTimestamp();
B
Bruce Momjian 已提交
3223

3224
				/*
B
Bruce Momjian 已提交
3225 3226
				 * this message is complex enough that it seems best to put
				 * the field extraction out-of-line
3227
				 */
3228
				exec_bind_message(&input_message);
3229 3230 3231 3232 3233
				break;

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

3236 3237 3238
					/* Set statement_timestamp() */
					SetCurrentStatementStartTimestamp();

3239 3240 3241
					portal_name = pq_getmsgstring(&input_message);
					max_rows = pq_getmsgint(&input_message, 4);
					pq_getmsgend(&input_message);
3242

3243
					exec_execute_message(portal_name, max_rows);
3244 3245 3246
				}
				break;

3247
			case 'F':			/* fastpath function call */
3248 3249 3250
				/* Set statement_timestamp() */
				SetCurrentStatementStartTimestamp();

3251
				/* Tell the collector what we're doing */
3252 3253
				pgstat_report_activity("<FASTPATH> function call");

3254
				/* start an xact for this function invocation */
3255
				start_xact_command();
3256

3257 3258 3259 3260 3261 3262 3263 3264 3265
				/*
				 * Note: we may at this point be inside an aborted
				 * transaction.  We can't throw error for that until
				 * we've finished reading the function-call message, so
				 * HandleFunctionRequest() must check for it after doing so.
				 * Be careful not to do anything that assumes we're inside a
				 * valid transaction here.
				 */

3266 3267 3268
				/* switch back to message context */
				MemoryContextSwitchTo(MessageContext);

3269
				if (HandleFunctionRequest(&input_message) == EOF)
3270 3271
				{
					/* lost frontend connection during F message input */
B
Bruce Momjian 已提交
3272

3273
					/*
3274
					 * Reset whereToSendOutput to prevent ereport from
B
Bruce Momjian 已提交
3275
					 * attempting to send any more messages to client.
3276
					 */
3277 3278
					if (whereToSendOutput == DestRemote)
						whereToSendOutput = DestNone;
3279

3280
					proc_exit(0);
3281
				}
3282 3283

				/* commit the function-invocation transaction */
3284
				finish_xact_command();
3285

3286
				send_ready_for_query = true;
3287 3288
				break;

B
Bruce Momjian 已提交
3289
			case 'C':			/* close */
3290
				{
B
Bruce Momjian 已提交
3291
					int			close_type;
3292 3293
					const char *close_target;

3294 3295 3296
					close_type = pq_getmsgbyte(&input_message);
					close_target = pq_getmsgstring(&input_message);
					pq_getmsgend(&input_message);
3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316 3317 3318 3319 3320 3321 3322 3323 3324

					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:
3325 3326
							ereport(ERROR,
									(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
3327 3328
								   errmsg("invalid CLOSE message subtype %d",
										  close_type)));
3329 3330 3331
							break;
					}

3332
					if (whereToSendOutput == DestRemote)
B
Bruce Momjian 已提交
3333
						pq_putemptymessage('3');		/* CloseComplete */
3334 3335 3336 3337 3338
				}
				break;

			case 'D':			/* describe */
				{
B
Bruce Momjian 已提交
3339
					int			describe_type;
3340 3341
					const char *describe_target;

3342 3343 3344
					/* Set statement_timestamp() (needed for xact) */
					SetCurrentStatementStartTimestamp();

3345 3346 3347
					describe_type = pq_getmsgbyte(&input_message);
					describe_target = pq_getmsgstring(&input_message);
					pq_getmsgend(&input_message);
3348 3349 3350 3351 3352 3353 3354 3355 3356 3357

					switch (describe_type)
					{
						case 'S':
							exec_describe_statement_message(describe_target);
							break;
						case 'P':
							exec_describe_portal_message(describe_target);
							break;
						default:
3358 3359
							ereport(ERROR,
									(errcode(ERRCODE_PROTOCOL_VIOLATION),
B
Bruce Momjian 已提交
3360 3361
								errmsg("invalid DESCRIBE message subtype %d",
									   describe_type)));
3362 3363 3364 3365 3366
							break;
					}
				}
				break;

B
Bruce Momjian 已提交
3367
			case 'H':			/* flush */
3368
				pq_getmsgend(&input_message);
3369
				if (whereToSendOutput == DestRemote)
3370 3371 3372
					pq_flush();
				break;

B
Bruce Momjian 已提交
3373
			case 'S':			/* sync */
3374
				pq_getmsgend(&input_message);
3375
				finish_xact_command();
3376
				send_ready_for_query = true;
3377 3378
				break;

3379
				/*
B
Bruce Momjian 已提交
3380 3381 3382
				 * 'X' means that the frontend is closing down the socket. EOF
				 * means unexpected loss of frontend connection. Either way,
				 * perform normal shutdown.
3383 3384
				 */
			case 'X':
3385
			case EOF:
B
Bruce Momjian 已提交
3386

3387
				/*
B
Bruce Momjian 已提交
3388 3389
				 * Reset whereToSendOutput to prevent ereport from attempting
				 * to send any more messages to client.
3390
				 */
3391 3392
				if (whereToSendOutput == DestRemote)
					whereToSendOutput = DestNone;
B
Bruce Momjian 已提交
3393

3394 3395
				/*
				 * NOTE: if you are tempted to add more code here, DON'T!
B
Bruce Momjian 已提交
3396
				 * Whatever you had in mind to do should be set up as an
B
Bruce Momjian 已提交
3397 3398 3399
				 * on_proc_exit or on_shmem_exit callback, instead. Otherwise
				 * it will fail to be called during other backend-shutdown
				 * scenarios.
3400
				 */
3401
				proc_exit(0);
3402

B
Bruce Momjian 已提交
3403 3404 3405 3406
			case 'd':			/* copy data */
			case 'c':			/* copy done */
			case 'f':			/* copy fail */

3407
				/*
B
Bruce Momjian 已提交
3408
				 * Accept but ignore these messages, per protocol spec; we
B
Bruce Momjian 已提交
3409 3410
				 * probably got here because a COPY failed, and the frontend
				 * is still sending data.
3411 3412 3413
				 */
				break;

3414
			default:
3415 3416 3417 3418
				ereport(FATAL,
						(errcode(ERRCODE_PROTOCOL_VIOLATION),
						 errmsg("invalid frontend message type %d",
								firstchar)));
3419
		}
3420
	}							/* end of input-reading loop */
3421

3422 3423
	/* can't get here because the above loop never exits */
	Assert(false);
3424

3425
	return 1;					/* keep compiler quiet */
3426 3427
}

3428
#ifndef HAVE_GETRUSAGE
B
Bruce Momjian 已提交
3429 3430
#include "rusagestub.h"
#else
3431
#include <sys/resource.h>
3432
#endif   /* HAVE_GETRUSAGE */
3433

3434 3435
static struct rusage Save_r;
static struct timeval Save_t;
3436 3437

void
3438
ResetUsage(void)
3439
{
3440
	getrusage(RUSAGE_SELF, &Save_r);
3441
	gettimeofday(&Save_t, NULL);
3442
	ResetBufferUsage();
3443
	/* ResetTupleCount(); */
3444 3445 3446
}

void
3447
ShowUsage(const char *title)
3448
{
3449
	StringInfoData str;
3450 3451 3452 3453
	struct timeval user,
				sys;
	struct timeval elapse_t;
	struct rusage r;
B
Bruce Momjian 已提交
3454
	char	   *bufusage;
3455 3456

	getrusage(RUSAGE_SELF, &r);
3457
	gettimeofday(&elapse_t, NULL);
3458 3459
	memcpy((char *) &user, (char *) &r.ru_utime, sizeof(user));
	memcpy((char *) &sys, (char *) &r.ru_stime, sizeof(sys));
3460 3461 3462 3463 3464 3465 3466 3467 3468 3469 3470 3471 3472 3473 3474 3475 3476 3477
	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 已提交
3478 3479 3480 3481
	 * 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.
3482
	 */
3483
	initStringInfo(&str);
3484

3485 3486
	appendStringInfo(&str, "! system usage stats:\n");
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3487
				"!\t%ld.%06ld elapsed %ld.%06ld user %ld.%06ld system sec\n",
3488 3489 3490
					 (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 已提交
3491
					 (long) (r.ru_utime.tv_usec - Save_r.ru_utime.tv_usec),
3492
					 (long) (r.ru_stime.tv_sec - Save_r.ru_stime.tv_sec),
B
Bruce Momjian 已提交
3493
					 (long) (r.ru_stime.tv_usec - Save_r.ru_stime.tv_usec));
3494
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3495
					 "!\t[%ld.%06ld user %ld.%06ld sys total]\n",
3496 3497 3498 3499
					 (long) user.tv_sec,
					 (long) user.tv_usec,
					 (long) sys.tv_sec,
					 (long) sys.tv_usec);
3500
#if defined(HAVE_GETRUSAGE)
3501
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3502 3503
					 "!\t%ld/%ld [%ld/%ld] filesystem blocks in/out\n",
					 r.ru_inblock - Save_r.ru_inblock,
3504
	/* they only drink coffee at dec */
B
Bruce Momjian 已提交
3505 3506
					 r.ru_oublock - Save_r.ru_oublock,
					 r.ru_inblock, r.ru_oublock);
3507
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3508
			  "!\t%ld/%ld [%ld/%ld] page faults/reclaims, %ld [%ld] swaps\n",
B
Bruce Momjian 已提交
3509 3510 3511 3512 3513
					 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);
3514
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3515
		 "!\t%ld [%ld] signals rcvd, %ld/%ld [%ld/%ld] messages rcvd/sent\n",
B
Bruce Momjian 已提交
3516 3517 3518 3519 3520
					 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);
3521
	appendStringInfo(&str,
B
Bruce Momjian 已提交
3522
			 "!\t%ld/%ld [%ld/%ld] voluntary/involuntary context switches\n",
B
Bruce Momjian 已提交
3523 3524 3525
					 r.ru_nvcsw - Save_r.ru_nvcsw,
					 r.ru_nivcsw - Save_r.ru_nivcsw,
					 r.ru_nvcsw, r.ru_nivcsw);
3526
#endif   /* HAVE_GETRUSAGE */
3527 3528

	bufusage = ShowBufferUsage();
3529
	appendStringInfo(&str, "! buffer usage stats:\n%s", bufusage);
3530 3531 3532
	pfree(bufusage);

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

3536 3537 3538
	ereport(LOG,
			(errmsg_internal("%s", title),
			 errdetail("%s", str.data)));
3539 3540

	pfree(str.data);
3541
}
3542 3543 3544 3545

/*
 * on_proc_exit handler to log end of session
 */
B
Bruce Momjian 已提交
3546
static void
3547
log_disconnections(int code, Datum arg)
3548
{
B
Bruce Momjian 已提交
3549
	Port	   *port = MyProcPort;
3550 3551 3552
	long		secs;
	int			usecs;
	int			msecs;
B
Bruce Momjian 已提交
3553 3554 3555
	int			hours,
				minutes,
				seconds;
3556

3557 3558 3559 3560
	TimestampDifference(port->SessionStartTime,
						GetCurrentTimestamp(),
						&secs, &usecs);
	msecs = usecs / 1000;
3561

3562 3563 3564 3565
	hours = secs / SECS_PER_HOUR;
	secs %= SECS_PER_HOUR;
	minutes = secs / SECS_PER_MINUTE;
	seconds = secs % SECS_PER_MINUTE;
3566

3567
	ereport(LOG,
3568
			(errmsg("disconnection: session time: %d:%02d:%02d.%03d "
3569
					"user=%s database=%s host=%s%s%s",
3570
					hours, minutes, seconds, msecs,
3571
					port->user_name, port->database_name, port->remote_host,
3572
					port->remote_port[0] ? " port=" : "", port->remote_port)));
3573
}