postgres.c 40.5 KB
Newer Older
1 2 3
/*-------------------------------------------------------------------------
 *
 * postgres.c--
4
 *	  POSTGRES C Backend Interface
5 6 7 8 9
 *
 * Copyright (c) 1994, Regents of the University of California
 *
 *
 * IDENTIFICATION
10
 *	  $Header: /cvsroot/pgsql/src/backend/tcop/postgres.c,v 1.88 1998/09/01 03:25:41 momjian Exp $
11 12
 *
 * NOTES
13 14
 *	  this is the "main" module of the postgres backend and
 *	  hence the main module of the "traffic cop".
15 16 17
 *
 *-------------------------------------------------------------------------
 */
B
Bruce Momjian 已提交
18 19

#include <unistd.h>
20 21
#include <stdio.h>
#include <string.h>
22
#include <signal.h>
23
#include <time.h>
B
Bruce Momjian 已提交
24
#include <setjmp.h>
25
#include <sys/time.h>
B
Bruce Momjian 已提交
26 27
#include <sys/types.h>
#include <fcntl.h>
28
#include <sys/param.h>			/* for MAXHOSTNAMELEN on most */
29
#ifndef MAXHOSTNAMELEN
30
#include <netdb.h>				/* for MAXHOSTNAMELEN on some */
M
Fixes:  
Marc G. Fournier 已提交
31
#endif
32
#ifndef MAXHOSTNAMELEN			/* for MAXHOSTNAMELEN under sco3.2v5.0.2 */
33 34
#include <sys/socket.h>
#endif
35
#include <errno.h>
36
#if HAVE_SYS_SELECT_H
37
#include <sys/select.h>
38
#endif							/* aix */
M
 
Marc G. Fournier 已提交
39 40 41
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
42 43 44

#include "postgres.h"
#include "miscadmin.h"
45
#include "fmgr.h"
46

47 48
#include "access/xact.h"
#include "catalog/catname.h"
49
#include "commands/async.h"
50 51 52
#include "executor/execdebug.h"
#include "executor/executor.h"
#include "libpq/libpq.h"
53
#include "libpq/libpq-be.h"
54 55 56 57
#include "libpq/pqsignal.h"
#include "nodes/pg_list.h"
#include "nodes/print.h"
#include "optimizer/cost.h"
58
#include "optimizer/planner.h"
59
#include "optimizer/prep.h"
60
#include "parser/parser.h"
61 62 63 64 65
#include "rewrite/rewriteHandler.h"		/* for QueryRewrite() */
#include "storage/bufmgr.h"
#include "tcop/dest.h"
#include "tcop/fastpath.h"
#include "tcop/pquery.h"
66
#include "tcop/tcopdebug.h"
67 68 69 70
#include "tcop/tcopprot.h"		/* where declarations for this file go */
#include "tcop/utility.h"
#include "utils/mcxt.h"
#include "utils/rel.h"
M
 
Marc G. Fournier 已提交
71
#include "utils/ps_status.h"
72

73
#if FALSE
74
#include "nodes/relation.h"
75
#endif
76 77 78 79

#if 0
#include "optimizer/xfunc.h"
#endif
80

81
#if FALSE
82
#include "nodes/plannodes.h"
83
#endif
84

85
#if FALSE
86
#include "nodes/memnodes.h"
87
#endif
88

M
 
Marc G. Fournier 已提交
89 90
#include "utils/trace.h"

91
#ifdef MULTIBYTE
M
 
Marc G. Fournier 已提交
92
#include "mb/pg_wchar.h"
93 94
#endif

M
 
Marc G. Fournier 已提交
95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
/*
 * Trace flags, see backend/utils/misc/trace.c
 */
#define Verbose				pg_options[TRACE_VERBOSE]
#define DebugPrintQuery		pg_options[TRACE_QUERY]
#define DebugPrintPlan		pg_options[TRACE_PLAN]
#define DebugPrintParse		pg_options[TRACE_PARSE]
#define ShowParserStats		pg_options[TRACE_PARSERSTATS]
#define ShowPlannerStats	pg_options[TRACE_PLANNERSTATS]
#define ShowExecutorStats	pg_options[TRACE_EXECUTORSTATS]
#define DebugPrintRewrittenParsetree \
							pg_options[TRACE_REWRITTEN]
#ifdef LOCK_MGR_DEBUG
#define LockDebug			pg_options[TRACE_LOCKS]
#endif

#define DeadlockCheckTimer	pg_options[OPT_DEADLOCKTIMEOUT]
#define HostnameLookup		pg_options[OPT_HOSTLOOKUP]
#define ShowPortNumber		pg_options[OPT_SHOWPORTNUMBER]

115
/* ----------------
116
 *		global variables
117 118
 * ----------------
 */
119 120

/*static bool	EnableRewrite = true; , never changes why have it*/
121
CommandDest whereToSendOutput;
122

M
 
Marc G. Fournier 已提交
123 124
/* Define status buffer needed by PS_SET_STATUS */
PS_DEFINE_BUFFER;
125

126 127
extern int	lockingOff;
extern int	NBuffers;
128

129 130 131
int			dontExecute = 0;
static int	ShowStats;
static bool IsEmptyQuery = false;
132

133
char		relname[80];		/* current relation name */
134

135
#if defined(nextstep)
136
jmp_buf		Warn_restart;
137 138

#define sigsetjmp(x,y)	setjmp(x)
139 140
#define siglongjmp longjmp
#else
141
sigjmp_buf	Warn_restart;
142 143

#endif							/* defined(nextstep) */
144
bool			InError;
145

146
extern int	NBuffers;
147

148 149 150 151
static int	EchoQuery = 0;		/* default don't echo */
time_t		tim;
char		pg_pathname[256];
FILE	   *StatFp;
152

153
/* ----------------
154 155
 *		people who want to use EOF should #define DONTUSENEWLINE in
 *		tcop/tcopdebug.h
156 157 158
 * ----------------
 */
#ifndef TCOP_DONTUSENEWLINE
159
int			UseNewLine = 1;		/* Use newlines query delimiters (the
160 161
								 * default) */

162
#else
163
int			UseNewLine = 0;		/* Use EOF as query delimiters */
164 165

#endif							/* TCOP_DONTUSENEWLINE */
166 167

/* ----------------
168 169
 *		bushy tree plan flag: if true planner will generate bushy-tree
 *		plans
170 171
 * ----------------
 */
172 173
int			BushyPlanFlag = 0;	/* default to false -- consider only
								 * left-deep trees */
174 175 176 177

/*
** Flags for expensive function optimization -- JMH 3/9/92
*/
178
int			XfuncMode = 0;
179 180 181

/*
 * ----------------
182 183 184 185 186 187 188
 *	 Note: _exec_repeat_ defaults to 1 but may be changed
 *		   by a DEBUG command.	 If you set this to a large
 *		   number N, run a single query, and then set it
 *		   back to 1 and run N queries, you can get an idea
 *		   of how much time is being spent in the parser and
 *		   planner b/c in the first case this overhead only
 *		   happens once.  -cim 6/9/91
189 190
 * ----------------
*/
191
int			_exec_repeat_ = 1;
192 193

/* ----------------------------------------------------------------
194
 *		decls for routines only used in this file
195 196
 * ----------------------------------------------------------------
 */
197
static char InteractiveBackend(char *inBuf);
198 199
static char SocketBackend(char *inBuf);
static char ReadCommand(char *inBuf);
200 201 202


/* ----------------------------------------------------------------
203
 *		routines to obtain user input
204 205 206 207
 * ----------------------------------------------------------------
 */

/* ----------------
208 209
 *	InteractiveBackend() is called for user interactive connections
 *	the string entered by the user is placed in its parameter inBuf.
210 211 212 213 214 215
 * ----------------
 */

static char
InteractiveBackend(char *inBuf)
{
216 217 218 219
	char	   *stuff = inBuf;	/* current place in input buffer */
	int			c;				/* character read from getc() */
	bool		end = false;	/* end-of-input flag */
	bool		backslashSeen = false;	/* have we seen a \ ? */
220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278

	/* ----------------
	 *	display a prompt and obtain input from the user
	 * ----------------
	 */
	printf("> ");

	for (;;)
	{
		if (UseNewLine)
		{
			/* ----------------
			 *	if we are using \n as a delimiter, then read
			 *	characters until the \n.
			 * ----------------
			 */
			while ((c = getc(stdin)) != EOF)
			{
				if (c == '\n')
				{
					if (backslashSeen)
					{
						stuff--;
						continue;
					}
					else
					{
						/* keep the newline character */
						*stuff++ = '\n';
						*stuff++ = '\0';
						break;
					}
				}
				else if (c == '\\')
					backslashSeen = true;
				else
					backslashSeen = false;

				*stuff++ = (char) c;
			}

			if (c == EOF)
				end = true;
		}
		else
		{
			/* ----------------
			 *	otherwise read characters until EOF.
			 * ----------------
			 */
			while ((c = getc(stdin)) != EOF)
				*stuff++ = (char) c;

			if (stuff == inBuf)
				end = true;
		}

		if (end)
		{
M
 
Marc G. Fournier 已提交
279
			if (Verbose)
280 281
				puts("EOF");
			IsEmptyQuery = true;
282
			proc_exit(0);
283 284 285 286 287 288 289 290 291 292 293 294 295 296
		}

		/* ----------------
		 *	otherwise we have a user query so process it.
		 * ----------------
		 */
		break;
	}

	/* ----------------
	 *	if the query echo flag was given, print the query..
	 * ----------------
	 */
	if (EchoQuery)
M
 
Marc G. Fournier 已提交
297
		printf("query: %s\n", inBuf);
298

299
	return 'Q';
300 301 302
}

/* ----------------
303
 *	SocketBackend()		Is called for frontend-backend connections
304
 *
305 306
 *	If the input is a query (case 'Q') then the string entered by
 *	the user is placed in its parameter inBuf.
307
 *
308 309 310
 *	If the input is a fastpath function call (case 'F') then
 *	the function call is processed in HandleFunctionRequest().
 *	(now called from PostgresMain())
311 312 313 314
 * ----------------
 */

static char
315
SocketBackend(char *inBuf)
316
{
317 318
	char		qtype[2];
	char		result = '\0';
319 320 321 322 323 324 325 326 327 328 329 330

	/* ----------------
	 *	get input from the frontend
	 * ----------------
	 */
	strcpy(qtype, "?");
	if (pq_getnchar(qtype, 0, 1) == EOF)
	{
		/* ------------
		 *	when front-end applications quits/dies
		 * ------------
		 */
331
		proc_exit(0);
332 333 334 335
	}

	switch (*qtype)
	{
336 337 338 339 340 341 342 343
			/* ----------------
			 *	'Q': user entered a query
			 * ----------------
			 */
		case 'Q':
			pq_getstr(inBuf, MAX_PARSE_BUFFER);
			result = 'Q';
			break;
344

345 346 347 348 349 350
			/* ----------------
			 *	'F':  calling user/system functions
			 * ----------------
			 */
		case 'F':
			pq_getstr(inBuf, MAX_PARSE_BUFFER); /* ignore the rest of the
351
												 * line */
352 353
			result = 'F';
			break;
354

355 356 357 358 359 360 361
			/* ----------------
			 *	'X':  frontend is exiting
			 * ----------------
			 */
		case 'X':
			result = 'X';
			break;
362

363 364 365 366 367 368 369 370 371 372
			/* ----------------
			 *	otherwise we got garbage from the frontend.
			 *
			 *	XXX are we certain that we want to do an elog(FATAL) here?
			 *		-cim 1/24/90
			 * ----------------
			 */
		default:
			elog(FATAL, "Socket command type %c unknown\n", *qtype);
			break;
373 374
	}
	return result;
375 376 377
}

/* ----------------
378 379 380 381
 *		ReadCommand reads a command from either the frontend or
 *		standard input, places it in inBuf, and returns a char
 *		representing whether the string is a 'Q'uery or a 'F'astpath
 *		call.
382 383 384
 * ----------------
 */
static char
385
ReadCommand(char *inBuf)
386
{
387 388
	if (IsUnderPostmaster)
		return SocketBackend(inBuf);
389 390
	else
		return InteractiveBackend(inBuf);
391 392
}

393 394 395 396 397 398
List *
pg_parse_and_plan(char *query_string,	/* string to execute */
				  Oid *typev,	/* argument types */
				  int nargs,	/* number of arguments */
				  QueryTreeList **queryListP,	/* pointer to the parse
												 * trees */
399 400
				  CommandDest dest,		/* where results should go */
				  bool aclOverride)
401
{
402 403 404 405 406 407 408 409
	QueryTreeList *querytree_list;
	int			i;
	List	   *plan_list = NIL;
	Plan	   *plan;
	int			j;
	QueryTreeList *new_list;
	List	   *rewritten = NIL;
	Query	   *querytree;
410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439

	/* ----------------
	 *	(1) parse the request string into a list of parse trees
	 * ----------------
	 */
	if (ShowParserStats)
		ResetUsage();

	querytree_list = parser(query_string, typev, nargs);

	if (ShowParserStats)
	{
		fprintf(stderr, "! Parser Stats:\n");
		ShowUsage();
	}

	/* new_list holds the rewritten queries */
	new_list = (QueryTreeList *) malloc(sizeof(QueryTreeList));
	new_list->len = querytree_list->len;
	new_list->qtrees = (Query **) malloc(new_list->len * sizeof(Query *));

	/* ----------------
	 *	(2) rewrite the queries, as necessary
	 * ----------------
	 */
	j = 0;						/* counter for the new_list, new_list can
								 * be longer than old list as a result of
								 * rewrites */
	for (i = 0; i < querytree_list->len; i++)
	{
440 441 442 443
		List	   *union_result,
				   *union_list,
				   *rewritten_list;

444 445
		querytree = querytree_list->qtrees[i];

M
 
Marc G. Fournier 已提交
446
		if (DebugPrintQuery)
447
		{
M
 
Marc G. Fournier 已提交
448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480
			if (DebugPrintQuery > 3) {
				/* Print the query string as is if query debug level > 3 */
				TPRINTF(TRACE_QUERY, "query: %s",query_string);
			} else {
				/* Print condensed query string to fit in one log line */
				char	buff[8192+1];
				char	c,
						*s,
						*d;
				int 	n,
						is_space=1;

				for (s=query_string,d=buff,n=0; (c=*s) && (n<8192); s++) {
					switch (c) {
						case '\r':
						case '\n':
						case '\t':
							c = ' ';
							/* fall through */
						case ' ':
							if (is_space) continue;
							is_space = 1;
							break;
						default:
							is_space = 0;
							break;
					}
					*d++ = c;
					n++;
				}
				*d = '\0';
				TPRINTF(TRACE_QUERY, "query: %s",buff);
			}
481
		}
482 483 484 485 486 487 488 489

		/* don't rewrite utilites */
		if (querytree->commandType == CMD_UTILITY)
		{
			new_list->qtrees[j++] = querytree;
			continue;
		}

M
 
Marc G. Fournier 已提交
490
		if (DebugPrintParse)
491
		{
M
 
Marc G. Fournier 已提交
492
			TPRINTF(TRACE_PARSE, "parser outputs:");
493 494 495 496 497
			nodeDisplay(querytree);
		}

		/* rewrite queries (retrieve, append, delete, replace) */
		rewritten = QueryRewrite(querytree);
B
Bruce Momjian 已提交
498

499
		if (rewritten != NIL)
500
		{
501 502
			int			len,
						k;
503

504 505 506 507 508 509 510 511 512 513 514 515 516
			/*
			 * Rewrite the UNIONS.
			 */
			foreach(rewritten_list, rewritten)
			{
				Query	   *qry = (Query *) lfirst(rewritten_list);

				union_result = NIL;
				foreach(union_list, qry->unionClause)
					union_result = nconc(union_result, QueryRewrite((Query *) lfirst(union_list)));
				qry->unionClause = union_result;
			}

517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534
			len = length(rewritten);
			if (len == 1)
				new_list->qtrees[j++] = (Query *) lfirst(rewritten);
			else
			{
				/* rewritten queries are longer than original query */
				/* grow the new_list to accommodate */
				new_list->len += len - 1;		/* - 1 because originally
												 * we allocated one space
												 * for the query */
				new_list->qtrees = realloc(new_list->qtrees,
										new_list->len * sizeof(Query *));
				for (k = 0; k < len; k++)
					new_list->qtrees[j++] = (Query *) nth(k, rewritten);
			}
		}
	}

535 536 537 538 539 540 541 542
	/* ----------
	 * Due to rewriting, the new list could also have been
	 * shrunk (do instead nothing). Forget obsolete queries
	 * at the end.
	 * ----------
	 */
	new_list->len = j;

543 544 545 546 547 548
	/* we're done with the original lists, free it */
	free(querytree_list->qtrees);
	free(querytree_list);

	querytree_list = new_list;

549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568
	/*
	 * Override ACL checking if requested
	 */
	if (aclOverride) {
		for (i = 0; i < querytree_list->len; i++) {
			RangeTblEntry	*rte;
			List		*l;

			querytree = querytree_list->qtrees[i];
			if (querytree->commandType == CMD_UTILITY)
				continue;

			foreach (l, querytree->rtable) {
				rte = (RangeTblEntry *)lfirst(l);

				rte->skipAcl = TRUE;
			}
		}
	}

M
 
Marc G. Fournier 已提交
569
	if (DebugPrintRewrittenParsetree)
570
	{
M
 
Marc G. Fournier 已提交
571
		TPRINTF(TRACE_REWRITTEN, "after rewriting:");
572 573 574

		for (i = 0; i < querytree_list->len; i++)
		{
575
			nodeDisplay(querytree_list->qtrees[i]);
576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598
			printf("\n");
		}
	}

	for (i = 0; i < querytree_list->len; i++)
	{
		querytree = querytree_list->qtrees[i];

		/*
		 * For each query that isn't a utility invocation, generate a
		 * plan.
		 */

		if (querytree->commandType != CMD_UTILITY)
		{

			if (IsAbortedTransactionBlockState())
			{
				/* ----------------
				 *	 the EndCommand() stuff is to tell the frontend
				 *	 that the command ended. -cim 6/1/90
				 * ----------------
				 */
599
				char	   *tag = "*ABORT STATE*";
600 601 602 603 604 605

				EndCommand(tag, dest);

				elog(NOTICE, "(transaction aborted): %s",
					 "queries ignored until END");

606 607 608 609
				free(querytree_list->qtrees);
				free(querytree_list);
				if (queryListP)
					*queryListP = (QueryTreeList *) NULL;
610 611 612 613 614
				return (List *) NULL;
			}

			if (ShowPlannerStats)
				ResetUsage();
615 616

			/* call that optimizer */
617
			plan = planner(querytree);
618

619 620 621 622 623 624
			if (ShowPlannerStats)
			{
				fprintf(stderr, "! Planner Stats:\n");
				ShowUsage();
			}
			plan_list = lappend(plan_list, plan);
M
Fixes:  
Marc G. Fournier 已提交
625
#ifdef INDEXSCAN_PATCH
626 627 628 629 630 631
			/* ----------------
			 *	Print plan if debugging.
			 *	This has been moved here to get debugging output
			 *	also for queries in functions.	DZ - 27-8-1996
			 * ----------------
			 */
M
 
Marc G. Fournier 已提交
632
			if (DebugPrintPlan)
633
			{
M
 
Marc G. Fournier 已提交
634
				TPRINTF(TRACE_PLAN, "plan:");
635 636
				nodeDisplay(plan);
			}
M
Fixes:  
Marc G. Fournier 已提交
637
#endif
638
		}
639
#ifdef FUNC_UTIL_PATCH
640 641 642 643 644 645 646 647

		/*
		 * If the command is an utility append a null plan. This is needed
		 * to keep the plan_list aligned with the querytree_list or the
		 * function executor will crash.  DZ - 30-8-1996
		 */
		else
			plan_list = lappend(plan_list, NULL);
648
#endif
649 650
	}

651 652 653 654 655 656 657 658 659 660
	/* ----------
	 * Check if the rewriting had thrown away anything
	 * ----------
	 */
	if (querytree_list->len == 0) {
		free(querytree_list->qtrees);
		free(querytree_list);
		querytree_list = NULL;
	}

661 662 663
	if (queryListP)
		*queryListP = querytree_list;

664
	return plan_list;
665 666 667
}

/* ----------------------------------------------------------------
668
 *		pg_exec_query()
669 670 671 672 673 674
 *
 *		Takes a querystring, runs the parser/utilities or
 *		parser/planner/executor over it as necessary
 *		Begin Transaction Should have been called before this
 *		and CommitTransaction After this is called
 *		This is strictly because we do not allow for nested xactions.
675
 *
676 677 678 679
 *		NON-OBVIOUS-RESTRICTIONS
 *		this function _MUST_ allocate a new "parsetree" each time,
 *		since it may be stored in a named portal and should not
 *		change its value.
680 681 682 683 684
 *
 * ----------------------------------------------------------------
 */

void
685
pg_exec_query(char *query_string)
686
{
687 688 689 690 691 692 693
	pg_exec_query_dest(query_string, whereToSendOutput, FALSE);
}

void
pg_exec_query_acl_override(char *query_string)
{
	pg_exec_query_dest(query_string, whereToSendOutput, TRUE);
694 695 696
}

void
697
pg_exec_query_dest(char *query_string,	/* string to execute */
698 699 700 701 702
				   CommandDest dest,	/* where results should go */
				   bool aclOverride)	/* to give utility
				   			 * commands power of
							 * superusers
							 */
703
{
704 705 706 707 708 709
	List	   *plan_list;
	Plan	   *plan;
	Query	   *querytree;
	int			i,
				j;
	QueryTreeList *querytree_list;
710 711

	/* plan the queries */
712
	plan_list = pg_parse_and_plan(query_string, NULL, 0, &querytree_list, dest, aclOverride);
713

714 715 716
	if (QueryCancel)
		CancelQuery();
		
717
	/* pg_parse_and_plan could have failed */
718 719 720 721 722 723 724
	if (querytree_list == NULL)
		return;

	for (i = 0; i < querytree_list->len; i++)
	{
		querytree = querytree_list->qtrees[i];

725
#ifdef FUNC_UTIL_PATCH
726 727 728 729 730 731 732

		/*
		 * Advance on the plan_list in every case.	Now the plan_list has
		 * the same length of the querytree_list.  DZ - 30-8-1996
		 */
		plan = (Plan *) lfirst(plan_list);
		plan_list = lnext(plan_list);
733
#endif
734 735 736 737 738 739 740 741 742
		if (querytree->commandType == CMD_UTILITY)
		{
			/* ----------------
			 *	 process utility functions (create, destroy, etc..)
			 *
			 *	 Note: we do not check for the transaction aborted state
			 *	 because that is done in ProcessUtility.
			 * ----------------
			 */
M
 
Marc G. Fournier 已提交
743 744 745 746 747
			if (DebugPrintQuery) {
				TPRINTF(TRACE_QUERY, "ProcessUtility: %s", query_string);
			} else if (Verbose) {
				TPRINTF(TRACE_VERBOSE, "ProcessUtility");
			} 
748 749 750 751 752 753

			ProcessUtility(querytree->utilityStmt, dest);

		}
		else
		{
754
#ifndef FUNC_UTIL_PATCH
755 756 757 758 759 760

			/*
			 * Moved before the if.  DZ - 30-8-1996
			 */
			plan = (Plan *) lfirst(plan_list);
			plan_list = lnext(plan_list);
761
#endif
762

M
Fixes:  
Marc G. Fournier 已提交
763
#ifdef INDEXSCAN_PATCH
764 765

			/*
766
			 * Print moved in pg_parse_and_plan.	DZ - 27-8-1996
767
			 */
M
Fixes:  
Marc G. Fournier 已提交
768
#else
769 770 771 772
			/* ----------------
			 *	print plan if debugging
			 * ----------------
			 */
M
 
Marc G. Fournier 已提交
773
			if (DebugPrintPlan)
774
			{
M
 
Marc G. Fournier 已提交
775
				TPRINTF(TRACE_PLAN, "plan:");
776 777
				nodeDisplay(plan);
			}
M
Fixes:  
Marc G. Fournier 已提交
778
#endif
779 780 781 782 783 784 785 786 787 788

			/* ----------------
			 *	 execute the plan
			 *
			 */
			if (ShowExecutorStats)
				ResetUsage();

			for (j = 0; j < _exec_repeat_; j++)
			{
M
 
Marc G. Fournier 已提交
789
				if (Verbose)
790
				{
M
 
Marc G. Fournier 已提交
791
					TPRINTF(TRACE_VERBOSE, "ProcessQuery");
792
				}
793
				ProcessQuery(querytree, plan, dest);
794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814
			}

			if (ShowExecutorStats)
			{
				fprintf(stderr, "! Executor Stats:\n");
				ShowUsage();
			}
		}

		/*
		 * In a query block, we want to increment the command counter
		 * between queries so that the effects of early queries are
		 * visible to subsequent ones.
		 */

		if (querytree_list)
			CommandCounterIncrement();
	}

	free(querytree_list->qtrees);
	free(querytree_list);
815 816 817
}

/* --------------------------------
818
 *		signal handler routines used in PostgresMain()
819
 *
M
 
Marc G. Fournier 已提交
820
 *		handle_warn() is used to catch kill(getpid(),SIGQUIT) which
821
 *		occurs when elog(ERROR) is called.
822
 *
823 824 825
 *		quickdie() occurs when signalled by the postmaster.
 *		Some backend has bought the farm,
 *		so we need to stop what we're doing and exit.
826
 *
827
 *		die() preforms an orderly cleanup via ExitPostgres()
828 829 830 831
 * --------------------------------
 */

void
832
handle_warn(SIGNAL_ARGS)
833
{
834
	siglongjmp(Warn_restart, 1);
835 836
}

837
void
838
quickdie(SIGNAL_ARGS)
839
{
840
	elog(NOTICE, "Message from PostgreSQL backend:"
841 842 843 844 845
		 "\n\tThe Postmaster has informed me that some other backend"
		 " died abnormally and possibly corrupted shared memory."
		 "\n\tI have rolled back the current transaction and am"
		 " going to terminate your database system connection and exit."
	"\n\tPlease reconnect to the database system and repeat your query.");
846

847 848 849 850 851 852 853 854

	/*
	 * DO NOT ExitPostgres(0) -- we're here because shared memory may be
	 * corrupted, so we don't want to flush any shared state to stable
	 * storage.  Just nail the windows shut and get out of town.
	 */

	exit(0);
855 856 857
}

void
858
die(SIGNAL_ARGS)
859
{
860
	ExitPostgres(0);
861 862 863
}

/* signal handler for floating point exception */
864
void
865
FloatExceptionHandler(SIGNAL_ARGS)
866
{
867
	elog(ERROR, "floating point exception!"
868 869
		 " The last floating point operation either exceeded legal ranges"
		 " or was a divide by zero");
870 871 872
}


M
 
Marc G. Fournier 已提交
873
/* signal handler for query cancel signal from postmaster */
874 875 876 877 878 879 880 881 882
static void
QueryCancelHandler(SIGNAL_ARGS)
{
	QueryCancel = true;
}

void
CancelQuery(void)
{
M
 
Marc G. Fournier 已提交
883 884 885
	/* QueryCancel flag will be reset in main loop, which we reach by
	 * longjmp from elog().
	 */
886 887 888 889
	elog(ERROR, "Query was cancelled.");
}


890 891
static void
usage(char *progname)
892
{
893
	fprintf(stderr,
894
			"Usage: %s [options] [dbname]\n", progname);
M
 
Marc G. Fournier 已提交
895 896 897
#ifdef USE_ASSERT_CHECKING
	fprintf(stderr, "    A: enable/disable assert checking\n");
#endif
898 899 900 901 902
	fprintf(stderr, "\t-B buffers\tset number of buffers in buffer pool\n");
	fprintf(stderr, "\t-C \t\tsupress version info\n");
	fprintf(stderr, "\t-D dir\t\tdata directory\n");
	fprintf(stderr, "\t-E \t\techo query before execution\n");
	fprintf(stderr, "\t-F \t\tturn off fsync\n");
903
#ifdef LOCK_MGR_DEBUG
904
	fprintf(stderr, "\t-K \t\tset locking debug level [0|1|2]\n");
905
#endif
906 907 908 909 910 911 912 913
	fprintf(stderr, "\t-P port\t\tset port file descriptor\n");
	fprintf(stderr, "\t-Q \t\tsuppress informational messages\n");
	fprintf(stderr, "\t-S buffers\tset amount of sort memory available\n");
	fprintf(stderr, "\t-d [1|2|3]\tset debug level\n");
	fprintf(stderr, "\t-e \t\tturn on European date format\n");
	fprintf(stderr, "\t-o file\t\tsend stdout and stderr to given filename \n");
	fprintf(stderr, "\t-s \t\tshow stats after each query\n");
	fprintf(stderr, "\t-v version\tset protocol version being used by frontend\n");
M
 
Marc G. Fournier 已提交
914
	fprintf(stderr, "\t-W \t\twait N seconds to allow attach from a debugger\n");
915 916 917
}

/* ----------------------------------------------------------------
918 919 920
 *		PostgresMain
 *		  postgres main loop
 *		all backends, interactive or otherwise start here
921 922 923
 * ----------------------------------------------------------------
 */
int
924
PostgresMain(int argc, char *argv[], int real_argc, char *real_argv[])
925
{
926 927 928 929 930
	bool			flagC = false,
					flagQ = false,
					flagE = false,
					flagEu = false;
	int				flag;
931

M
 
Marc G. Fournier 已提交
932 933
	char	   		*DBName = NULL;
	int				errs = 0;
934

M
 
Marc G. Fournier 已提交
935 936 937
	char			firstchar;
	char			parser_input[MAX_PARSE_BUFFER];
	char	   		*userName;
M
 
Marc G. Fournier 已提交
938 939 940 941 942

	/* Used if verbose is set, must be initialized */
	char	   		*remote_info = "interactive";
	char	   		*remote_host = "";
	unsigned short 	remote_port  = 0;
943

M
 
Marc G. Fournier 已提交
944 945 946 947
	char	   		*DBDate = NULL;
	extern int		optind;
	extern char 	*optarg;
	extern short 	DebugLvl;
948 949 950 951 952

	/* ----------------
	 *	parse command line arguments
	 * ----------------
	 */
953 954 955 956

	/*
	 * Set default values.
	 */
957
	ShowStats = 0;
958
	ShowParserStats = ShowPlannerStats = ShowExecutorStats = 0;
M
 
Marc G. Fournier 已提交
959
	DeadlockCheckTimer = DEADLOCK_CHECK_TIMER;
960
#ifdef LOCK_MGR_DEBUG
M
 
Marc G. Fournier 已提交
961
	LockDebug = 0;
962 963
#endif

964
	/*
965 966
	 * get hostname is either the environment variable PGHOST or NULL NULL
	 * means Unix-socket only
967
	 */
968
	DataDir = getenv("PGDATA");
969

970
	/*
971 972 973
	 * Try to get initial values for date styles and formats. Does not do
	 * a complete job, but should be good enough for backend. Cannot call
	 * parse_date() since palloc/pfree memory is not set up yet.
974 975 976 977 978 979 980 981 982 983 984 985 986 987 988
	 */
	DBDate = getenv("PGDATESTYLE");
	if (DBDate != NULL)
	{
		if (strcasecmp(DBDate, "ISO") == 0)
			DateStyle = USE_ISO_DATES;
		else if (strcasecmp(DBDate, "SQL") == 0)
			DateStyle = USE_SQL_DATES;
		else if (strcasecmp(DBDate, "POSTGRES") == 0)
			DateStyle = USE_POSTGRES_DATES;
		else if (strcasecmp(DBDate, "GERMAN") == 0)
		{
			DateStyle = USE_GERMAN_DATES;
			EuroDates = TRUE;
		}
989

990 991 992 993 994
		if (strcasecmp(DBDate, "NONEURO") == 0)
			EuroDates = FALSE;
		else if (strcasecmp(DBDate, "EURO") == 0)
			EuroDates = TRUE;
	}
M
 
Marc G. Fournier 已提交
995 996 997 998 999
	
	/*
	 * Read default pg_options from file $DATADIR/pg_options.
	 */
	read_pg_options(0);
1000

1001 1002
    optind = 1; /* reset after postmaster usage */
	
M
 
Marc G. Fournier 已提交
1003 1004
	while ((flag = getopt(argc, argv,
						  "A:B:bCD:d:Eef:iK:Lm:MNo:P:pQS:st:v:x:FW:"))
1005 1006 1007
		   != EOF)
		switch (flag)
		{
M
 
Marc G. Fournier 已提交
1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018
			case 'A':
				/* ----------------
				 *  enable/disable assert checking.
				 * ----------------
				 */
#ifdef USE_ASSERT_CHECKING
				assert_enabled = atoi(optarg);
#else
				fprintf(stderr, "Assert checking is not enabled\n");
#endif
				break;
1019

1020 1021 1022 1023 1024 1025 1026
			case 'b':
				/* ----------------
				 *	set BushyPlanFlag to true.
				 * ----------------
				 */
				BushyPlanFlag = 1;
				break;
M
 
Marc G. Fournier 已提交
1027

1028 1029 1030 1031 1032 1033 1034
			case 'B':
				/* ----------------
				 *	specify the size of buffer pool
				 * ----------------
				 */
				NBuffers = atoi(optarg);
				break;
1035

1036 1037 1038 1039 1040
			case 'C':
				/* ----------------
				 *	don't print version string (don't know why this is 'C' --mao)
				 * ----------------
				 */
1041
				flagC = true;
1042
				break;
1043

1044 1045
			case 'D':			/* PGDATA directory */
				DataDir = optarg;
1046

1047
			case 'd':			/* debug level */
1048
				flagQ = false;
1049
				DebugLvl = (short) atoi(optarg);
M
 
Marc G. Fournier 已提交
1050 1051 1052 1053 1054 1055 1056 1057 1058
				if (DebugLvl >= 1)
				{
					Verbose = DebugLvl;
				}
				if (DebugLvl >= 2)
				{
  					DebugPrintQuery = true;
				}
				if (DebugLvl >= 3)
1059
				{
M
 
Marc G. Fournier 已提交
1060 1061 1062 1063
					DebugPrintQuery = DebugLvl;
				}
				if (DebugLvl >= 4)
  				{
1064 1065 1066 1067
					DebugPrintParse = true;
					DebugPrintPlan = true;
					DebugPrintRewrittenParsetree = true;
				}
1068
				break;
1069 1070 1071 1072 1073 1074

			case 'E':
				/* ----------------
				 *	E - echo the query the user entered
				 * ----------------
				 */
1075
				flagE = true;
1076
				break;
1077 1078 1079 1080 1081 1082

			case 'e':
				/* --------------------------
				 * Use european date formats.
				 * --------------------------
				 */
1083
				flagEu = true;
1084
				break;
1085 1086 1087 1088 1089 1090 1091

			case 'F':
				/* --------------------
				 *	turn off fsync
				 * --------------------
				 */
				fsyncOff = 1;
1092
				break;
1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110

			case 'f':
				/* -----------------
				 *	  f - forbid generation of certain plans
				 * -----------------
				 */
				switch (optarg[0])
				{
					case 's':	/* seqscan */
						_enable_seqscan_ = false;
						break;
					case 'i':	/* indexscan */
						_enable_indexscan_ = false;
						break;
					case 'n':	/* nestloop */
						_enable_nestloop_ = false;
						break;
					case 'm':	/* mergejoin */
1111
						_enable_mergejoin_ = false;
1112 1113 1114 1115 1116 1117 1118
						break;
					case 'h':	/* hashjoin */
						_enable_hashjoin_ = false;
						break;
					default:
						errs++;
				}
1119 1120
				break;

1121 1122 1123
			case 'i':
				dontExecute = 1;
				break;
1124

1125
			case 'K':
1126
#ifdef LOCK_MGR_DEBUG
M
 
Marc G. Fournier 已提交
1127
				LockDebug = atoi(optarg);
1128
#else
1129
				fprintf(stderr, "Lock debug not compiled in\n");
1130
#endif
1131
				break;
1132

1133 1134 1135 1136 1137 1138 1139
			case 'L':
				/* --------------------
				 *	turn off locking
				 * --------------------
				 */
				lockingOff = 1;
				break;
1140

1141
			case 'm':
1142
				/* Multiplexed backends are no longer supported. */
1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153
				break;
			case 'M':
				exit(PostmasterMain(argc, argv));
				break;
			case 'N':
				/* ----------------
				 *	N - Don't use newline as a query delimiter
				 * ----------------
				 */
				UseNewLine = 0;
				break;
1154

1155 1156 1157 1158 1159
			case 'o':
				/* ----------------
				 *	o - send output (stdout and stderr) to the given file
				 * ----------------
				 */
1160
				StrNCpy(OutputFileName, optarg, MAXPGPATH);
1161
				break;
1162

1163 1164 1165 1166 1167 1168 1169 1170
			case 'p':			/* started by postmaster */
				/* ----------------
				 *	p - special flag passed if backend was forked
				 *		by a postmaster.
				 * ----------------
				 */
				IsUnderPostmaster = true;
				break;
1171

1172 1173 1174 1175 1176 1177 1178 1179 1180
			case 'P':
				/* ----------------
				 *	P - Use the passed file descriptor number as the port
				 *	  on which to communicate with the user.  This is ONLY
				 *	  useful for debugging when fired up by the postmaster.
				 * ----------------
				 */
				Portfd = atoi(optarg);
				break;
1181

1182 1183 1184 1185 1186
			case 'Q':
				/* ----------------
				 *	Q - set Quiet mode (reduce debugging output)
				 * ----------------
				 */
1187
				flagQ = true;
M
 
Marc G. Fournier 已提交
1188
				Verbose = 0;
1189
				break;
1190

1191 1192
			case 'S':
				/* ----------------
V
Vadim B. Mikheev 已提交
1193
				 *	S - amount of sort memory to use in 1k bytes
1194 1195
				 * ----------------
				 */
1196
				{
1197 1198
					int			S;

V
Vadim B. Mikheev 已提交
1199
					S = atoi(optarg);
1200
					if (S >= 4 * BLCKSZ / 1024)
V
Vadim B. Mikheev 已提交
1201
						SortMem = S;
1202
				}
1203
				break;
1204 1205 1206 1207 1208 1209 1210 1211

			case 's':
				/* ----------------
				 *	  s - report usage statistics (timings) after each query
				 * ----------------
				 */
				ShowStats = 1;
				StatFp = stderr;
1212
				break;
1213

M
 
Marc G. Fournier 已提交
1214 1215 1216 1217
	    	case 'T':
				parse_options(optarg);
				break;

1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246
			case 't':
				/* ----------------
				 *	tell postgres to report usage statistics (timings) for
				 *	each query
				 *
				 *	-tpa[rser] = print stats for parser time of each query
				 *	-tpl[anner] = print stats for planner time of each query
				 *	-te[xecutor] = print stats for executor time of each query
				 *	caution: -s can not be used together with -t.
				 * ----------------
				 */
				StatFp = stderr;
				switch (optarg[0])
				{
					case 'p':
						if (optarg[1] == 'a')
							ShowParserStats = 1;
						else if (optarg[1] == 'l')
							ShowPlannerStats = 1;
						else
							errs++;
						break;
					case 'e':
						ShowExecutorStats = 1;
						break;
					default:
						errs++;
						break;
				}
1247 1248
				break;

1249
			case 'v':
1250
				FrontendProtocol = (ProtocolVersion) atoi(optarg);
1251 1252
				break;

M
 
Marc G. Fournier 已提交
1253 1254 1255 1256 1257 1258 1259 1260
			case 'W':
				/* ----------------
				 *  wait N seconds to allow attach from a debugger
				 * ----------------
				 */
				sleep(atoi(optarg));
				break;

1261
			case 'x':
1262
#if 0							/* planner/xfunc.h */
1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290

				/*
				 * control joey hellerstein's expensive function
				 * optimization
				 */
				if (XfuncMode != 0)
				{
					fprintf(stderr, "only one -x flag is allowed\n");
					errs++;
					break;
				}
				if (strcmp(optarg, "off") == 0)
					XfuncMode = XFUNC_OFF;
				else if (strcmp(optarg, "nor") == 0)
					XfuncMode = XFUNC_NOR;
				else if (strcmp(optarg, "nopull") == 0)
					XfuncMode = XFUNC_NOPULL;
				else if (strcmp(optarg, "nopm") == 0)
					XfuncMode = XFUNC_NOPM;
				else if (strcmp(optarg, "pullall") == 0)
					XfuncMode = XFUNC_PULLALL;
				else if (strcmp(optarg, "wait") == 0)
					XfuncMode = XFUNC_WAIT;
				else
				{
					fprintf(stderr, "use -x {off,nor,nopull,nopm,pullall,wait}\n");
					errs++;
				}
1291
#endif
1292
				break;
1293

1294 1295 1296 1297 1298 1299
			default:
				/* ----------------
				 *	default: bad command line option
				 * ----------------
				 */
				errs++;
1300 1301 1302 1303 1304 1305 1306 1307 1308
		}

	/* ----------------
	 *	get user name and pathname and check command line validity
	 * ----------------
	 */
	SetPgUserName();
	userName = GetPgUserName();

1309
#ifdef CYR_RECODE
1310
	SetCharSet();
1311 1312
#endif

1313
	if (FindExec(pg_pathname, argv[0], "postgres") < 0)
1314 1315 1316 1317 1318 1319
		elog(FATAL, "%s: could not locate executable, bailing out...",
			 argv[0]);

	if (errs || argc - optind > 1)
	{
		usage(argv[0]);
1320
		proc_exit(1);
1321 1322 1323 1324 1325 1326 1327
	}
	else if (argc - optind == 1)
		DBName = argv[optind];
	else if ((DBName = userName) == NULL)
	{
		fprintf(stderr, "%s: USER undefined and no database specified\n",
				argv[0]);
1328
		proc_exit(1);
1329 1330 1331 1332 1333 1334
	}

	if (ShowStats &&
		(ShowParserStats || ShowPlannerStats || ShowExecutorStats))
	{
		fprintf(stderr, "-s can not be used together with -t.\n");
1335
		proc_exit(1);
1336 1337 1338 1339 1340 1341 1342 1343 1344
	}

	if (!DataDir)
	{
		fprintf(stderr, "%s does not know where to find the database system "
				"data.  You must specify the directory that contains the "
				"database system either by specifying the -D invocation "
			 "option or by setting the PGDATA environment variable.\n\n",
				argv[0]);
1345
		proc_exit(1);
1346 1347 1348 1349 1350 1351
	}

	Noversion = flagC;
	EchoQuery = flagE;
	EuroDates = flagEu;

M
 
Marc G. Fournier 已提交
1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386
	/*
	 * Find remote host name or address.
	 */
	if (IsUnderPostmaster) {
		switch (MyProcPort->raddr.sa.sa_family) {
			struct hostent *host_ent;

		    case AF_INET:
				remote_info = remote_host = malloc(48);
				remote_port = ntohs(MyProcPort->raddr.in.sin_port);
				strcpy(remote_host, inet_ntoa(MyProcPort->raddr.in.sin_addr));
				if (HostnameLookup) {
					host_ent = \
						gethostbyaddr((char *)&MyProcPort->raddr.in.sin_addr,
									  sizeof(MyProcPort->raddr.in.sin_addr),
									  AF_INET);
					if (host_ent) {
						strncpy(remote_host, host_ent->h_name, 48);
						*(remote_host+47) = '\0';
					}
				}
				if (ShowPortNumber) {
					remote_info = malloc(strlen(remote_host)+6);
					sprintf(remote_info, "%s:%d", remote_host, remote_port);
				}
				break;
		    case AF_UNIX:
				remote_info = remote_host = "localhost";
				break;
		    default:
				remote_info = remote_host = "unknown";
				break;
		}
	}

1387
	/* ----------------
M
 
Marc G. Fournier 已提交
1388
	 *	set process params for ps
1389 1390
	 * ----------------
	 */
M
 
Marc G. Fournier 已提交
1391 1392 1393 1394
	if (IsUnderPostmaster) {
		PS_INIT_STATUS(real_argc, real_argv, argv[0], 
					   remote_info, userName, DBName);
		PS_SET_STATUS("idle");
1395 1396
	}

1397
	/* ----------------
M
 
Marc G. Fournier 已提交
1398
	 *	print flags
1399 1400
	 * ----------------
	 */
M
 
Marc G. Fournier 已提交
1401
	if (Verbose)
1402
	{
M
 
Marc G. Fournier 已提交
1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420
		if (Verbose == 1) {
			TPRINTF(TRACE_VERBOSE, "started: host=%s user=%s database=%s",
					remote_host, userName, DBName);
		} else {
			TPRINTF(TRACE_VERBOSE, "debug info:");
			TPRINTF(TRACE_VERBOSE, "\tUser         = %s", userName);
			TPRINTF(TRACE_VERBOSE, "\tRemoteHost   = %s", remote_host);
			TPRINTF(TRACE_VERBOSE, "\tRemotePort   = %d", remote_port);
			TPRINTF(TRACE_VERBOSE, "\tDatabaseName = %s", DBName);
			TPRINTF(TRACE_VERBOSE, "\tVerbose      = %d", Verbose);
			TPRINTF(TRACE_VERBOSE, "\tNoversion    = %c", Noversion ? 't' : 'f');
			TPRINTF(TRACE_VERBOSE, "\ttimings      = %c", ShowStats ? 't' : 'f');
			TPRINTF(TRACE_VERBOSE, "\tdates        = %s",
					EuroDates ? "European" : "Normal");
			TPRINTF(TRACE_VERBOSE, "\tbufsize      = %d", NBuffers);
			TPRINTF(TRACE_VERBOSE, "\tsortmem      = %d", SortMem);
			TPRINTF(TRACE_VERBOSE, "\tquery echo   = %c", EchoQuery ? 't' : 'f');
		}
1421 1422
	}

1423 1424 1425 1426
	/* ----------------
	 *	initialize portal file descriptors
	 * ----------------
	 */
1427
	if (IsUnderPostmaster)
1428 1429 1430 1431 1432 1433 1434 1435 1436
	{
		if (Portfd < 0)
		{
			fprintf(stderr,
					"Postmaster flag set: no port number specified, use /dev/null\n");
			Portfd = open(NULL_DEV, O_RDWR, 0666);
		}
		pq_init(Portfd);
		whereToSendOutput = Remote;
1437
	}
1438 1439 1440 1441 1442 1443
	else
		whereToSendOutput = Debug;

	SetProcessingMode(InitProcessing);

	/* initialize */
M
 
Marc G. Fournier 已提交
1444 1445
	if (Verbose)
		TPRINTF(TRACE_VERBOSE, "InitPostgres");
1446 1447 1448

	InitPostgres(DBName);

1449
#ifdef MULTIBYTE
1450
	/* set default client encoding */
M
 
Marc G. Fournier 已提交
1451
	if (Verbose)
1452 1453 1454 1455
	{
		puts("\treset_client_encoding()..");
	}
	reset_client_encoding();
M
 
Marc G. Fournier 已提交
1456
	if (Verbose)
1457 1458 1459 1460 1461
	{
		puts("\treset_client_encoding() done.");
	}
#endif

M
 
Marc G. Fournier 已提交
1462 1463 1464 1465 1466 1467 1468
	/* ----------------
	 *  Set up handler for cancel-request signal, and
	 *	send this backend's cancellation info to the frontend.
	 *	This should not be done until we are sure startup is successful.
	 * ----------------
	 */

M
 
Marc G. Fournier 已提交
1469 1470 1471 1472 1473 1474 1475 1476 1477
	pqsignal(SIGHUP,  read_pg_options);		/* upate pg_options from file */
	pqsignal(SIGINT,  QueryCancelHandler);	/* cancel current query */
	pqsignal(SIGQUIT, handle_warn);			/* handle error */
	pqsignal(SIGTERM, die);
	pqsignal(SIGPIPE, die);
	pqsignal(SIGUSR1, quickdie);
	pqsignal(SIGUSR2, Async_NotifyHandler);	/* flush also sinval cache */
	pqsignal(SIGCHLD, SIG_IGN);				/* ignored, sent by LockOwners */
	pqsignal(SIGFPE,  FloatExceptionHandler);
M
 
Marc G. Fournier 已提交
1478 1479 1480 1481 1482 1483 1484 1485 1486 1487

	if (whereToSendOutput == Remote &&
		PG_PROTOCOL_MAJOR(FrontendProtocol) >= 2)
	{
		pq_putnchar("K", 1);
		pq_putint((int32) MyProcPid, sizeof(int32));
		pq_putint((int32) MyCancelKey, sizeof(int32));
		/* Need not flush since ReadyForQuery will do it. */
	}

1488 1489 1490 1491 1492 1493 1494
	/* ----------------
	 *	if an exception is encountered, processing resumes here
	 *	so we abort the current transaction and start a new one.
	 *	This must be done after we initialize the slave backends
	 *	so that the slaves signal the master to abort the transaction
	 *	rather than calling AbortCurrentTransaction() themselves.
	 *
M
 
Marc G. Fournier 已提交
1495 1496
	 *	Note:  elog(ERROR) causes a kill(getpid(),SIGQUIT) to occur
	 *		   sending us back here.
1497 1498 1499 1500 1501
	 * ----------------
	 */

	if (sigsetjmp(Warn_restart, 1) != 0)
	{
1502
		InError = true;
1503 1504 1505

		time(&tim);

M
 
Marc G. Fournier 已提交
1506 1507
		if (Verbose)
			TPRINTF(TRACE_VERBOSE, "AbortCurrentTransaction");
1508

B
Bruce Momjian 已提交
1509
		MemSet(parser_input, 0, MAX_PARSE_BUFFER);
1510 1511

		AbortCurrentTransaction();
1512

1513
	}
1514 1515

	InError = false;
1516 1517 1518 1519 1520

	/* ----------------
	 *	POSTGRES main processing loop begins here
	 * ----------------
	 */
1521
	if (!IsUnderPostmaster)
1522 1523
	{
		puts("\nPOSTGRES backend interactive interface");
1524
		puts("$Revision: 1.88 $ $Date: 1998/09/01 03:25:41 $");
1525 1526 1527 1528 1529 1530 1531 1532 1533
	}

	/* ----------------
	 * if stable main memory is assumed (-S(old) flag is set), it is necessary
	 * to flush all dirty shared buffers before exit
	 * plai 8/7/90
	 * ----------------
	 */
	if (!TransactionFlushEnabled())
1534
		on_shmem_exit(FlushBufferPool, NULL);
1535 1536 1537

	for (;;)
	{
B
Bruce Momjian 已提交
1538 1539 1540 1541
		/* ----------------
		 *	 (0) tell the frontend we're ready for a new query.
		 * ----------------
		 */
1542
		ReadyForQuery(whereToSendOutput);
B
Bruce Momjian 已提交
1543

1544 1545 1546 1547
		/* ----------------
		 *	 (1) read a command.
		 * ----------------
		 */
B
Bruce Momjian 已提交
1548
		MemSet(parser_input, 0, MAX_PARSE_BUFFER);
1549

1550
		firstchar = ReadCommand(parser_input);
1551 1552 1553

		QueryCancel = false;

1554 1555 1556 1557
		/* process the command */
		switch (firstchar)
		{
				/* ----------------
1558 1559
				 *	'F' indicates a fastpath call.
				 *		XXX HandleFunctionRequest
1560 1561
				 * ----------------
				 */
1562
			case 'F':
1563 1564
				IsEmptyQuery = false;

1565
				/* start an xact for this function invocation */
M
 
Marc G. Fournier 已提交
1566
				if (Verbose)
1567
				{
M
 
Marc G. Fournier 已提交
1568
					TPRINTF(TRACE_VERBOSE, "StartTransactionCommand");
1569
				}
1570

1571
				StartTransactionCommand();
1572
				HandleFunctionRequest();
M
 
Marc G. Fournier 已提交
1573
				PS_SET_STATUS("idle");
1574
				break;
1575

1576 1577 1578 1579 1580 1581
				/* ----------------
				 *	'Q' indicates a user query
				 * ----------------
				 */
			case 'Q':
				fflush(stdout);
1582

1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602
				if (strspn(parser_input, " \t\n") == strlen(parser_input))
				{
					/* ----------------
					 *	if there is nothing in the input buffer, don't bother
					 *	trying to parse and execute anything..
					 * ----------------
					 */
					IsEmptyQuery = true;
				}
				else
				{
					/* ----------------
					 *	otherwise, process the input string.
					 * ----------------
					 */
					IsEmptyQuery = false;
					if (ShowStats)
						ResetUsage();

					/* start an xact for this query */
M
 
Marc G. Fournier 已提交
1603
					if (Verbose)
1604
					{
M
 
Marc G. Fournier 已提交
1605
						TPRINTF(TRACE_VERBOSE, "StartTransactionCommand");
1606 1607
					}
					StartTransactionCommand();
1608

1609 1610
					pg_exec_query(parser_input);

M
 
Marc G. Fournier 已提交
1611
					PS_SET_STATUS("idle");
1612

1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623
					if (ShowStats)
						ShowUsage();
				}
				break;

				/* ----------------
				 *	'X' means that the frontend is closing down the socket
				 * ----------------
				 */
			case 'X':
				pq_close();
1624
				proc_exit(0);
1625 1626 1627
				break;

			default:
M
 
Marc G. Fournier 已提交
1628
				elog(ERROR, "unknown frontend message was received");
1629 1630 1631 1632 1633 1634
		}

		/* ----------------
		 *	 (3) commit the current transaction
		 *
		 *	 Note: if we had an empty input buffer, then we didn't
1635
		 *	 call pg_exec_query, so we don't bother to commit this transaction.
1636 1637 1638 1639
		 * ----------------
		 */
		if (!IsEmptyQuery)
		{
M
 
Marc G. Fournier 已提交
1640
			if (Verbose)
1641
			{
M
 
Marc G. Fournier 已提交
1642
				TPRINTF(TRACE_VERBOSE, "CommitTransactionCommand");
1643
			}
M
 
Marc G. Fournier 已提交
1644
			PS_SET_STATUS("commit");
1645
			CommitTransactionCommand();
M
 
Marc G. Fournier 已提交
1646
			PS_SET_STATUS("idle");
1647 1648 1649 1650

		}
		else
		{
1651
			if (IsUnderPostmaster)
1652 1653 1654 1655
				NullCommand(Remote);
		}

	}							/* infinite for-loop */
1656
	proc_exit(0);
1657
	return 1;
1658 1659
}

1660
#ifndef HAVE_GETRUSAGE
1661
#include "rusagestub.h"
1662
#else							/* HAVE_GETRUSAGE */
1663
#include <sys/resource.h>
1664
#endif							/* HAVE_GETRUSAGE */
1665

1666 1667
struct rusage Save_r;
struct timeval Save_t;
1668 1669

void
1670
ResetUsage(void)
1671
{
1672 1673 1674 1675 1676 1677
	struct timezone tz;

	getrusage(RUSAGE_SELF, &Save_r);
	gettimeofday(&Save_t, &tz);
	ResetBufferUsage();
/*	  ResetTupleCount(); */
1678 1679 1680
}

void
1681
ShowUsage(void)
1682
{
1683 1684 1685
	struct timeval user,
				sys;
	struct timeval elapse_t;
1686
	struct timezone tz;
1687
	struct rusage r;
1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732

	getrusage(RUSAGE_SELF, &r);
	gettimeofday(&elapse_t, &tz);
	memmove((char *) &user, (char *) &r.ru_utime, sizeof(user));
	memmove((char *) &sys, (char *) &r.ru_stime, sizeof(sys));
	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
	 * 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.
	 */

	fprintf(StatFp, "! system usage stats:\n");
	fprintf(StatFp,
			"!\t%ld.%06ld elapsed %ld.%06ld user %ld.%06ld system sec\n",
			(long int) elapse_t.tv_sec - Save_t.tv_sec,
			(long int) elapse_t.tv_usec - Save_t.tv_usec,
			(long int) r.ru_utime.tv_sec - Save_r.ru_utime.tv_sec,
			(long int) r.ru_utime.tv_usec - Save_r.ru_utime.tv_usec,
			(long int) r.ru_stime.tv_sec - Save_r.ru_stime.tv_sec,
			(long int) r.ru_stime.tv_usec - Save_r.ru_stime.tv_usec);
	fprintf(StatFp,
			"!\t[%ld.%06ld user %ld.%06ld sys total]\n",
			(long int) user.tv_sec,
			(long int) user.tv_usec,
			(long int) sys.tv_sec,
			(long int) sys.tv_usec);
1733
#ifdef HAVE_GETRUSAGE
1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762
	fprintf(StatFp,
			"!\t%ld/%ld [%ld/%ld] filesystem blocks in/out\n",
			r.ru_inblock - Save_r.ru_inblock,
	/* they only drink coffee at dec */
			r.ru_oublock - Save_r.ru_oublock,
			r.ru_inblock, r.ru_oublock);
	fprintf(StatFp,
		  "!\t%ld/%ld [%ld/%ld] page faults/reclaims, %ld [%ld] swaps\n",
			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);
	fprintf(StatFp,
	 "!\t%ld [%ld] signals rcvd, %ld/%ld [%ld/%ld] messages rcvd/sent\n",
			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);
	fprintf(StatFp,
		 "!\t%ld/%ld [%ld/%ld] voluntary/involuntary context switches\n",
			r.ru_nvcsw - Save_r.ru_nvcsw,
			r.ru_nivcsw - Save_r.ru_nivcsw,
			r.ru_nvcsw, r.ru_nivcsw);
#endif							/* HAVE_GETRUSAGE */
	fprintf(StatFp, "! postgres usage stats:\n");
	PrintBufferUsage(StatFp);
/*	   DisplayTupleCount(StatFp); */
1763
}
M
 
Marc G. Fournier 已提交
1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789

#ifdef USE_ASSERT_CHECKING
int
assertEnable(int val)
{
	assert_enabled = val;
	return val;
}

#ifdef ASSERT_CHECKING_TEST
int
assertTest(int val)
{
	Assert(val == 0);

	if (assert_enabled) {
		/* val != 0 should be trapped by previous Assert */
		elog(NOTICE, "Assert test successfull (val = %d)", val);
	} else {
		elog(NOTICE, "Assert checking is disabled (val = %d)", val);
	}

	return val;
}
#endif
#endif