postgres.c 42.0 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
B
Bruce Momjian 已提交
10
 *	  $Header: /cvsroot/pgsql/src/backend/tcop/postgres.c,v 1.122 1999/07/16 03:13:50 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 24
#include <time.h>
#include <sys/time.h>
B
Bruce Momjian 已提交
25 26
#include <sys/types.h>
#include <fcntl.h>
B
Bruce Momjian 已提交
27
#include <sys/param.h>
28
#ifndef MAXHOSTNAMELEN
B
Bruce Momjian 已提交
29
#include <netdb.h>
M
Fixes:  
Marc G. Fournier 已提交
30
#endif
31
#ifndef MAXHOSTNAMELEN			/* for MAXHOSTNAMELEN under sco3.2v5.0.2 */
32 33
#include <sys/socket.h>
#endif
34
#include <errno.h>
35
#if HAVE_SYS_SELECT_H
36
#include <sys/select.h>
37
#endif	 /* aix */
M
 
Marc G. Fournier 已提交
38 39 40
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
41 42
#ifdef __CYGWIN32__
#endif
43 44 45 46 47

#include "postgres.h"
#include "miscadmin.h"

#include "commands/async.h"
48
#include "libpq/libpq.h"
49
#include "libpq/pqformat.h"
50 51 52
#include "libpq/pqsignal.h"
#include "nodes/print.h"
#include "optimizer/cost.h"
53
#include "optimizer/planner.h"
54
#include "parser/parser.h"
B
Bruce Momjian 已提交
55
#include "rewrite/rewriteHandler.h"
56 57
#include "tcop/fastpath.h"
#include "tcop/pquery.h"
B
Bruce Momjian 已提交
58
#include "tcop/tcopprot.h"
59
#include "tcop/utility.h"
M
 
Marc G. Fournier 已提交
60
#include "utils/ps_status.h"
61
#include "utils/temprel.h"
62
#include "../backend/parser/parse.h"
63

M
 
Marc G. Fournier 已提交
64
#ifdef NOT_USED
65
#endif
66

M
 
Marc G. Fournier 已提交
67
#ifdef NOT_USED
68
#endif
69

M
 
Marc G. Fournier 已提交
70
#ifdef NOT_USED
71
#endif
72

M
 
Marc G. Fournier 已提交
73 74
#include "utils/trace.h"

75
#ifdef MULTIBYTE
76 77
#endif

M
 
Marc G. Fournier 已提交
78 79 80 81 82 83 84
/*
 * 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]
J
Jan Wieck 已提交
85 86 87 88 89 90
#define DebugPrintRewrittenParsetree \
							pg_options[TRACE_REWRITTEN]
#define DebugPPrintPlan		pg_options[TRACE_PRETTY_PLAN]
#define DebugPPrintParse	pg_options[TRACE_PRETTY_PARSE]
#define DebugPPrintRewrittenParsetree \
							pg_options[TRACE_PRETTY_REWRITTEN]
M
 
Marc G. Fournier 已提交
91 92 93 94 95 96 97 98 99 100 101
#define ShowParserStats		pg_options[TRACE_PARSERSTATS]
#define ShowPlannerStats	pg_options[TRACE_PLANNERSTATS]
#define ShowExecutorStats	pg_options[TRACE_EXECUTORSTATS]
#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]

102
/* ----------------
103
 *		global variables
104 105
 * ----------------
 */
106 107

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

M
 
Marc G. Fournier 已提交
110 111
/* Define status buffer needed by PS_SET_STATUS */
PS_DEFINE_BUFFER;
112

113 114
extern int	lockingOff;
extern int	NBuffers;
115

116 117 118
int			dontExecute = 0;
static int	ShowStats;
static bool IsEmptyQuery = false;
119

120
char		relname[80];		/* current relation name */
121

122
/* note: these declarations had better match tcopprot.h */
B
Bruce Momjian 已提交
123
DLLIMPORT sigjmp_buf Warn_restart;
124
bool		InError;
125

126
extern int	NBuffers;
127

128 129 130 131
static int	EchoQuery = 0;		/* default don't echo */
time_t		tim;
char		pg_pathname[256];
FILE	   *StatFp;
132

133
/* ----------------
134 135
 *		people who want to use EOF should #define DONTUSENEWLINE in
 *		tcop/tcopdebug.h
136 137 138
 * ----------------
 */
#ifndef TCOP_DONTUSENEWLINE
139
int			UseNewLine = 1;		/* Use newlines query delimiters (the
140 141
								 * default) */

142
#else
143
int			UseNewLine = 0;		/* Use EOF as query delimiters */
144

145
#endif	 /* TCOP_DONTUSENEWLINE */
146 147 148 149

/*
** Flags for expensive function optimization -- JMH 3/9/92
*/
150
int			XfuncMode = 0;
151 152 153

/*
 * ----------------
154 155 156 157 158 159 160
 *	 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
161 162
 * ----------------
*/
163
int			_exec_repeat_ = 1;
164 165

/* ----------------------------------------------------------------
166
 *		decls for routines only used in this file
167 168
 * ----------------------------------------------------------------
 */
169
static char InteractiveBackend(char *inBuf);
170 171
static char SocketBackend(char *inBuf);
static char ReadCommand(char *inBuf);
172
static void pg_exec_query(char *query_string);
173 174 175


/* ----------------------------------------------------------------
176
 *		routines to obtain user input
177 178 179 180
 * ----------------------------------------------------------------
 */

/* ----------------
181 182
 *	InteractiveBackend() is called for user interactive connections
 *	the string entered by the user is placed in its parameter inBuf.
183 184 185 186 187 188
 * ----------------
 */

static char
InteractiveBackend(char *inBuf)
{
189 190 191 192
	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 \ ? */
193 194 195 196 197

	/* ----------------
	 *	display a prompt and obtain input from the user
	 * ----------------
	 */
198
	printf("backend> ");
199
	fflush(stdout);
200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252

	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 已提交
253
			if (Verbose)
254 255
				puts("EOF");
			IsEmptyQuery = true;
256
			proc_exit(0);
257 258 259 260 261 262 263 264 265 266 267 268 269 270
		}

		/* ----------------
		 *	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 已提交
271
		printf("query: %s\n", inBuf);
272
	fflush(stdout);
273

274
	return 'Q';
275 276 277
}

/* ----------------
278
 *	SocketBackend()		Is called for frontend-backend connections
279
 *
280 281
 *	If the input is a query (case 'Q') then the string entered by
 *	the user is placed in its parameter inBuf.
282
 *
283 284 285
 *	If the input is a fastpath function call (case 'F') then
 *	the function call is processed in HandleFunctionRequest().
 *	(now called from PostgresMain())
286 287 288 289
 * ----------------
 */

static char
290
SocketBackend(char *inBuf)
291
{
292
	char		qtype;
293
	char		result = '\0';
294 295 296 297 298

	/* ----------------
	 *	get input from the frontend
	 * ----------------
	 */
299 300
	qtype = '?';
	if (pq_getbytes(&qtype, 1) == EOF)
301 302 303 304 305
	{
		/* ------------
		 *	when front-end applications quits/dies
		 * ------------
		 */
306
		proc_exit(0);
307 308
	}

309
	switch (qtype)
310
	{
311 312 313 314 315 316 317 318
			/* ----------------
			 *	'Q': user entered a query
			 * ----------------
			 */
		case 'Q':
			pq_getstr(inBuf, MAX_PARSE_BUFFER);
			result = 'Q';
			break;
319

320 321 322 323 324 325
			/* ----------------
			 *	'F':  calling user/system functions
			 * ----------------
			 */
		case 'F':
			pq_getstr(inBuf, MAX_PARSE_BUFFER); /* ignore the rest of the
326
												 * line */
327 328
			result = 'F';
			break;
329

330 331 332 333 334 335 336
			/* ----------------
			 *	'X':  frontend is exiting
			 * ----------------
			 */
		case 'X':
			result = 'X';
			break;
337

338 339 340 341 342 343 344 345
			/* ----------------
			 *	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:
346
			elog(FATAL, "Socket command type %c unknown", qtype);
347
			break;
348 349
	}
	return result;
350 351 352
}

/* ----------------
353 354 355 356
 *		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.
357 358 359
 * ----------------
 */
static char
360
ReadCommand(char *inBuf)
361
{
362 363
	if (IsUnderPostmaster)
		return SocketBackend(inBuf);
364 365
	else
		return InteractiveBackend(inBuf);
366 367
}

368 369 370 371
List *
pg_parse_and_plan(char *query_string,	/* string to execute */
				  Oid *typev,	/* argument types */
				  int nargs,	/* number of arguments */
B
Bruce Momjian 已提交
372 373
				  List **queryListP,	/* returned pointer to the parse
										 * trees */
374 375
				  CommandDest dest,		/* where results should go */
				  bool aclOverride)
376
{
377
	List	   *querytree_list = NIL;
378
	List	   *plan_list = NIL;
379
	List	   *querytree_list_item;
380
	Query	   *querytree;
381 382 383
	Plan	   *plan;
	List	   *new_list;
	List	   *rewritten;
384

385 386 387
	if (DebugPrintQuery)
	{
		if (DebugPrintQuery > 3)
B
Bruce Momjian 已提交
388
		{
389
			/* Print the query string as is if query debug level > 3 */
B
Bruce Momjian 已提交
390
			TPRINTF(TRACE_QUERY, "query: %s", query_string);
391 392 393 394 395 396
		}
		else
		{
			/* Print condensed query string to fit in one log line */
			char		buff[MAX_QUERY_SIZE + 1];
			char		c,
B
Bruce Momjian 已提交
397 398
					   *s,
					   *d;
399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427
			int			n,
						is_space = 1;

			for (s = query_string, d = buff, n = 0; (c = *s) && (n < MAX_QUERY_SIZE); 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);
		}
	}

428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444
	/* ----------------
	 *	(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();
	}

	/* ----------------
	 *	(2) rewrite the queries, as necessary
445
	 *
B
Bruce Momjian 已提交
446 447
	 *	rewritten queries are collected in new_list.  Note there may be
	 *	more or fewer than in the original list.
448 449
	 * ----------------
	 */
450
	new_list = NIL;
B
Bruce Momjian 已提交
451
	foreach(querytree_list_item, querytree_list)
452
	{
453
		querytree = (Query *) lfirst(querytree_list_item);
454

J
Jan Wieck 已提交
455
		if (DebugPrintParse || DebugPPrintParse)
456
		{
B
Bruce Momjian 已提交
457 458
			if (DebugPPrintParse)
			{
J
Jan Wieck 已提交
459 460
				TPRINTF(TRACE_PRETTY_PARSE, "parser outputs:");
				nodeDisplay(querytree);
B
Bruce Momjian 已提交
461 462 463
			}
			else
			{
J
Jan Wieck 已提交
464 465 466
				TPRINTF(TRACE_PARSE, "parser outputs:");
				printf("\n%s\n\n", nodeToString(querytree));
			}
467
		}
468 469 470

		if (querytree->commandType == CMD_UTILITY)
		{
471 472
			/* don't rewrite utilities, just dump 'em into new_list */
			new_list = lappend(new_list, querytree);
473
		}
474
		else
475
		{
476 477 478
			/* rewrite regular queries */
			rewritten = QueryRewrite(querytree);
			new_list = nconc(new_list, rewritten);
479 480 481 482 483
		}
	}

	querytree_list = new_list;

484 485 486
	/*
	 * Override ACL checking if requested
	 */
487 488
	if (aclOverride)
	{
B
Bruce Momjian 已提交
489
		foreach(querytree_list_item, querytree_list)
490 491
		{
			List	   *l;
492

493 494
			querytree = (Query *) lfirst(querytree_list_item);

495 496 497
			if (querytree->commandType == CMD_UTILITY)
				continue;

498 499
			foreach(l, querytree->rtable)
			{
500
				RangeTblEntry *rte = (RangeTblEntry *) lfirst(l);
501 502 503 504 505 506

				rte->skipAcl = TRUE;
			}
		}
	}

J
Jan Wieck 已提交
507
	if (DebugPrintRewrittenParsetree || DebugPPrintRewrittenParsetree)
508
	{
B
Bruce Momjian 已提交
509 510
		if (DebugPPrintRewrittenParsetree)
		{
J
Jan Wieck 已提交
511
			TPRINTF(TRACE_PRETTY_REWRITTEN, "after rewriting:");
512

B
Bruce Momjian 已提交
513
			foreach(querytree_list_item, querytree_list)
J
Jan Wieck 已提交
514
			{
515 516
				querytree = (Query *) lfirst(querytree_list_item);
				nodeDisplay(querytree);
J
Jan Wieck 已提交
517 518
				printf("\n");
			}
B
Bruce Momjian 已提交
519 520 521
		}
		else
		{
J
Jan Wieck 已提交
522 523
			TPRINTF(TRACE_REWRITTEN, "after rewriting:");

B
Bruce Momjian 已提交
524
			foreach(querytree_list_item, querytree_list)
J
Jan Wieck 已提交
525
			{
526 527
				querytree = (Query *) lfirst(querytree_list_item);
				printf("\n%s\n\n", nodeToString(querytree));
J
Jan Wieck 已提交
528
			}
529 530 531
		}
	}

B
Bruce Momjian 已提交
532
	foreach(querytree_list_item, querytree_list)
533
	{
534
		querytree = (Query *) lfirst(querytree_list_item);
535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550

		/*
		 * 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
				 * ----------------
				 */
551
				char	   *tag = "*ABORT STATE*";
552 553 554 555 556 557

				EndCommand(tag, dest);

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

558
				if (queryListP)
559 560
					*queryListP = NIL;
				return NIL;
561 562 563 564
			}

			if (ShowPlannerStats)
				ResetUsage();
565 566

			/* call that optimizer */
567
			plan = planner(querytree);
568

569 570 571 572 573 574
			if (ShowPlannerStats)
			{
				fprintf(stderr, "! Planner Stats:\n");
				ShowUsage();
			}
			plan_list = lappend(plan_list, plan);
M
Fixes:  
Marc G. Fournier 已提交
575
#ifdef INDEXSCAN_PATCH
576 577 578 579 580 581
			/* ----------------
			 *	Print plan if debugging.
			 *	This has been moved here to get debugging output
			 *	also for queries in functions.	DZ - 27-8-1996
			 * ----------------
			 */
J
Jan Wieck 已提交
582
			if (DebugPrintPlan || DebugPPrintPlan)
583
			{
B
Bruce Momjian 已提交
584 585
				if (DebugPPrintPlan)
				{
J
Jan Wieck 已提交
586 587
					TPRINTF(TRACE_PRETTY_PLAN, "plan:");
					nodeDisplay(plan);
B
Bruce Momjian 已提交
588 589 590
				}
				else
				{
J
Jan Wieck 已提交
591 592 593
					TPRINTF(TRACE_PLAN, "plan:");
					printf("\n%s\n\n", nodeToString(plan));
				}
594
			}
M
Fixes:  
Marc G. Fournier 已提交
595
#endif
596
		}
B
Bruce Momjian 已提交
597

598 599 600 601 602 603 604
		/*
		 * 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);
605 606
	}

607 608 609
	if (queryListP)
		*queryListP = querytree_list;

610
	return plan_list;
611 612 613
}

/* ----------------------------------------------------------------
614
 *		pg_exec_query()
615 616 617 618 619 620
 *
 *		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.
621
 *
622 623 624 625
 *		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.
626 627 628 629
 *
 * ----------------------------------------------------------------
 */

630
static void
631
pg_exec_query(char *query_string)
632
{
633 634 635 636 637 638 639
	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);
640 641 642
}

void
643
pg_exec_query_dest(char *query_string,	/* string to execute */
644
				   CommandDest dest,	/* where results should go */
645 646
				   bool aclOverride)	/* to give utility commands power
										 * of superusers */
647
{
648
	List	   *querytree_list;
649 650
	List	   *plan_list;
	Query	   *querytree;
651 652
	Plan	   *plan;
	int			j;
653 654

	/* plan the queries */
655 656
	plan_list = pg_parse_and_plan(query_string, NULL, 0,
								  &querytree_list, dest, aclOverride);
657

658
	/* if we got a cancel signal whilst planning, quit */
659 660
	if (QueryCancel)
		CancelQuery();
661

662
	/* OK, do it to it! */
663

B
Bruce Momjian 已提交
664 665 666 667
	/*
	 * NOTE: we do not use "foreach" here because we want to be sure the
	 * list pointers have been advanced before the query is executed. We
	 * need to do that because VACUUM has a nasty little habit of doing
668 669 670 671
	 * CommitTransactionCommand at startup, and that will release the
	 * memory holding our parse/plan lists :-(.  This needs a better
	 * solution --- currently, the code will crash if someone submits
	 * "vacuum; something-else" in a single query string.  But memory
B
Bruce Momjian 已提交
672 673
	 * allocation needs redesigned anyway, so this will have to do for
	 * now.
674
	 */
675

676 677 678 679
	while (querytree_list)
	{
		querytree = (Query *) lfirst(querytree_list);
		querytree_list = lnext(querytree_list);
680 681
		plan = (Plan *) lfirst(plan_list);
		plan_list = lnext(plan_list);
682

683 684 685 686 687 688 689 690 691
		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.
			 * ----------------
			 */
692
			if (DebugPrintQuery)
M
 
Marc G. Fournier 已提交
693
				TPRINTF(TRACE_QUERY, "ProcessUtility: %s", query_string);
694
			else if (Verbose)
M
 
Marc G. Fournier 已提交
695
				TPRINTF(TRACE_VERBOSE, "ProcessUtility");
696

697 698 699 700 701 702 703
			/*
			 * We have to set query SnapShot in the case of FETCH or COPY TO.
			 */
			if (nodeTag(querytree->utilityStmt) == T_FetchStmt ||
				(nodeTag(querytree->utilityStmt) == T_CopyStmt && 
				((CopyStmt *)(querytree->utilityStmt))->direction != FROM))
				SetQuerySnapshot();
704 705 706 707
			ProcessUtility(querytree->utilityStmt, dest);
		}
		else
		{
M
Fixes:  
Marc G. Fournier 已提交
708
#ifdef INDEXSCAN_PATCH
709 710

			/*
711
			 * Print moved in pg_parse_and_plan.	DZ - 27-8-1996
712
			 */
M
Fixes:  
Marc G. Fournier 已提交
713
#else
714 715 716 717
			/* ----------------
			 *	print plan if debugging
			 * ----------------
			 */
J
Jan Wieck 已提交
718
			if (DebugPrintPlan || DebugPPrintPlan)
719
			{
B
Bruce Momjian 已提交
720 721
				if (DebugPPrintPlan)
				{
J
Jan Wieck 已提交
722 723
					TPRINTF(TRACE_PRETTY_PLAN, "plan:");
					nodeDisplay(plan);
B
Bruce Momjian 已提交
724 725 726
				}
				else
				{
J
Jan Wieck 已提交
727 728 729
					TPRINTF(TRACE_PLAN, "plan:");
					printf("\n%s\n\n", nodeToString(plan));
				}
730
			}
M
Fixes:  
Marc G. Fournier 已提交
731
#endif
732

V
Vadim B. Mikheev 已提交
733 734 735
			SetQuerySnapshot();

			/*
B
Bruce Momjian 已提交
736
			 * execute the plan
737 738 739 740 741 742
			 */
			if (ShowExecutorStats)
				ResetUsage();

			for (j = 0; j < _exec_repeat_; j++)
			{
M
 
Marc G. Fournier 已提交
743 744
				if (Verbose)
					TPRINTF(TRACE_VERBOSE, "ProcessQuery");
745
				ProcessQuery(querytree, plan, dest);
746 747 748 749 750 751 752 753 754 755 756 757 758 759 760
			}

			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.
		 */

761
		CommandCounterIncrement();
762
	}
763 764 765
}

/* --------------------------------
766
 *		signal handler routines used in PostgresMain()
767
 *
B
Bruce Momjian 已提交
768
 *		handle_warn() catches SIGQUIT.	It forces control back to the main
769 770 771
 *		loop, just as if an internal error (elog(ERROR,...)) had occurred.
 *		elog() used to actually use kill(2) to induce a SIGQUIT to get here!
 *		But that's not 100% reliable on some systems, so now it does its own
B
Bruce Momjian 已提交
772
 *		siglongjmp() instead.
773
 *		We still provide the signal catcher so that an error quit can be
B
Bruce Momjian 已提交
774
 *		forced externally.	This should be done only with great caution,
775 776
 *		however, since an asynchronous signal could leave the system in
 *		who-knows-what inconsistent state.
777
 *
778 779 780
 *		quickdie() occurs when signalled by the postmaster.
 *		Some backend has bought the farm,
 *		so we need to stop what we're doing and exit.
781
 *
782
 *		die() performs an orderly cleanup via ExitPostgres()
783 784 785 786
 * --------------------------------
 */

void
787
handle_warn(SIGNAL_ARGS)
788
{
789
	siglongjmp(Warn_restart, 1);
790 791
}

792
void
793
quickdie(SIGNAL_ARGS)
794
{
795
	elog(NOTICE, "Message from PostgreSQL backend:"
796 797 798 799 800
		 "\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.");
801

802 803 804 805 806 807 808 809

	/*
	 * 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);
810 811 812
}

void
813
die(SIGNAL_ARGS)
814
{
815
	ExitPostgres(0);
816 817 818
}

/* signal handler for floating point exception */
819
void
820
FloatExceptionHandler(SIGNAL_ARGS)
821
{
822
	elog(ERROR, "floating point exception!"
823 824
		 " The last floating point operation either exceeded legal ranges"
		 " or was a divide by zero");
825 826 827
}


M
 
Marc G. Fournier 已提交
828
/* signal handler for query cancel signal from postmaster */
829 830 831 832 833 834 835 836 837
static void
QueryCancelHandler(SIGNAL_ARGS)
{
	QueryCancel = true;
}

void
CancelQuery(void)
{
B
Bruce Momjian 已提交
838

839 840
	/*
	 * QueryCancel flag will be reset in main loop, which we reach by
M
 
Marc G. Fournier 已提交
841 842
	 * longjmp from elog().
	 */
843 844 845 846
	elog(ERROR, "Query was cancelled.");
}


847 848
static void
usage(char *progname)
849
{
850
	fprintf(stderr,
851
			"Usage: %s [options] [dbname]\n", progname);
M
 
Marc G. Fournier 已提交
852
#ifdef USE_ASSERT_CHECKING
T
Tom Lane 已提交
853
	fprintf(stderr, "\t-A on\t\tenable/disable assert checking\n");
M
 
Marc G. Fournier 已提交
854
#endif
855
	fprintf(stderr, "\t-B buffers\tset number of buffers in buffer pool\n");
T
Tom Lane 已提交
856
	fprintf(stderr, "\t-C \t\tsuppress version info\n");
857 858 859
	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");
860
#ifdef LOCK_MGR_DEBUG
T
Tom Lane 已提交
861
	fprintf(stderr, "\t-K lev\t\tset locking debug level [0|1|2]\n");
862
#endif
863 864
	fprintf(stderr, "\t-L \t\tturn off locking\n");
	fprintf(stderr, "\t-N \t\tdon't use newline as interactive query delimiter\n");
865
	fprintf(stderr, "\t-O \t\tallow system table structure changes\n");
866
	fprintf(stderr, "\t-Q \t\tsuppress informational messages\n");
867
	fprintf(stderr, "\t-S kbytes\tset amount of memory for sorts (in kbytes)\n");
T
Tom Lane 已提交
868 869
	fprintf(stderr, "\t-T options\tspecify pg_options\n");
	fprintf(stderr, "\t-W sec\t\twait N seconds to allow attach from a debugger\n");
870 871
	fprintf(stderr, "\t-d [1|2|3]\tset debug level\n");
	fprintf(stderr, "\t-e \t\tturn on European date format\n");
T
Tom Lane 已提交
872
	fprintf(stderr, "\t-f [s|i|n|m|h]\tforbid use of some plan types\n");
873 874 875
	fprintf(stderr, "\t-i \t\tdon't execute queries\n");
	fprintf(stderr, "\t-o file\t\tsend stdout and stderr to given filename\n");
	fprintf(stderr, "\t-p database\tbackend is started under a postmaster\n");
876
	fprintf(stderr, "\t-s \t\tshow stats after each query\n");
877
	fprintf(stderr, "\t-t [pa|pl|ex]\tshow timings after each query\n");
878
	fprintf(stderr, "\t-v version\tset protocol version being used by frontend\n");
879 880 881
}

/* ----------------------------------------------------------------
882 883
 *	PostgresMain
 *		postgres main loop
884
 *		all backends, interactive or otherwise start here
885 886 887 888 889
 *
 *	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.
 *	real_argc/real_argv point to the original argv array, which is needed by
 *	PS_INIT_STATUS on some platforms.
890 891 892
 * ----------------------------------------------------------------
 */
int
893
PostgresMain(int argc, char *argv[], int real_argc, char *real_argv[])
894
{
895
	int			flag;
896

897
	char	   *DBName = NULL;
898
	bool		secure = true;
899
	int			errs = 0;
900

901 902 903
	char		firstchar;
	char		parser_input[MAX_PARSE_BUFFER];
	char	   *userName;
M
 
Marc G. Fournier 已提交
904 905

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

910 911 912 913
	char	   *DBDate = NULL;
	extern int	optind;
	extern char *optarg;
	extern short DebugLvl;
914

915
	/*
916
	 * Set default values for command-line options.
917
	 */
918
	IsUnderPostmaster = false;
919
	ShowStats = 0;
920
	ShowParserStats = ShowPlannerStats = ShowExecutorStats = 0;
M
 
Marc G. Fournier 已提交
921
	DeadlockCheckTimer = DEADLOCK_CHECK_TIMER;
922 923
	Noversion = false;
	EchoQuery = false;
924
#ifdef LOCK_MGR_DEBUG
M
 
Marc G. Fournier 已提交
925
	LockDebug = 0;
926
#endif
927
	DataDir = getenv("PGDATA");
928

929
	/*
930 931 932
	 * 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.
933 934 935 936 937 938 939 940 941 942 943 944 945 946 947
	 */
	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;
		}
948
		else if (strcasecmp(DBDate, "NONEURO") == 0)
949 950 951 952
			EuroDates = FALSE;
		else if (strcasecmp(DBDate, "EURO") == 0)
			EuroDates = TRUE;
	}
953

M
 
Marc G. Fournier 已提交
954 955 956
	/*
	 * Read default pg_options from file $DATADIR/pg_options.
	 */
957
	if (DataDir)
M
 
Marc G. Fournier 已提交
958
		read_pg_options(0);
959

960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977
	/* ----------------
	 *	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.
	 * ----------------
	 */

	optind = 1;					/* reset after postmaster's usage */
978

M
 
Marc G. Fournier 已提交
979
	while ((flag = getopt(argc, argv,
980
						  "A:B:CD:d:EeFf:iK:LNOo:p:QS:sT:t:v:W:x:"))
981 982 983
		   != EOF)
		switch (flag)
		{
M
 
Marc G. Fournier 已提交
984 985
			case 'A':
				/* ----------------
986
				 *	enable/disable assert checking.
M
 
Marc G. Fournier 已提交
987 988 989 990 991 992 993 994
				 * ----------------
				 */
#ifdef USE_ASSERT_CHECKING
				assert_enabled = atoi(optarg);
#else
				fprintf(stderr, "Assert checking is not enabled\n");
#endif
				break;
995

996 997 998 999 1000
			case 'B':
				/* ----------------
				 *	specify the size of buffer pool
				 * ----------------
				 */
1001 1002
				if (secure)
					NBuffers = atoi(optarg);
1003
				break;
1004

1005 1006
			case 'C':
				/* ----------------
1007
				 *	don't print version string
1008 1009
				 * ----------------
				 */
1010
				Noversion = true;
1011
				break;
1012

1013
			case 'D':			/* PGDATA directory */
1014 1015 1016 1017 1018 1019 1020 1021 1022
				if (secure)
				{
					if (!DataDir)
					{
						DataDir = optarg;
						/* must be done after DataDir is defined */
						read_pg_options(0);
					}
					DataDir = optarg;
1023
				}
M
 
Marc G. Fournier 已提交
1024
				break;
1025

1026 1027
			case 'd':			/* debug level */
				DebugLvl = (short) atoi(optarg);
M
 
Marc G. Fournier 已提交
1028 1029 1030
				if (DebugLvl >= 1)
					Verbose = DebugLvl;
				if (DebugLvl >= 2)
1031
					DebugPrintQuery = true;
M
 
Marc G. Fournier 已提交
1032 1033 1034
				if (DebugLvl >= 3)
					DebugPrintQuery = DebugLvl;
				if (DebugLvl >= 4)
1035
				{
1036 1037 1038 1039
					DebugPrintParse = true;
					DebugPrintPlan = true;
					DebugPrintRewrittenParsetree = true;
				}
J
Jan Wieck 已提交
1040 1041 1042 1043 1044 1045
				if (DebugLvl >= 5)
				{
					DebugPPrintParse = true;
					DebugPPrintPlan = true;
					DebugPPrintRewrittenParsetree = true;
				}
1046
				break;
1047 1048 1049 1050 1051 1052

			case 'E':
				/* ----------------
				 *	E - echo the query the user entered
				 * ----------------
				 */
1053
				EchoQuery = true;
1054
				break;
1055 1056 1057 1058 1059 1060

			case 'e':
				/* --------------------------
				 * Use european date formats.
				 * --------------------------
				 */
1061
				EuroDates = true;
1062
				break;
1063 1064 1065 1066 1067 1068

			case 'F':
				/* --------------------
				 *	turn off fsync
				 * --------------------
				 */
1069 1070
				if (secure)
					disableFsync = true;
1071
				break;
1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089

			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 */
1090
						_enable_mergejoin_ = false;
1091 1092 1093 1094 1095 1096 1097
						break;
					case 'h':	/* hashjoin */
						_enable_hashjoin_ = false;
						break;
					default:
						errs++;
				}
1098 1099
				break;

1100 1101 1102
			case 'i':
				dontExecute = 1;
				break;
1103

1104
			case 'K':
1105
#ifdef LOCK_MGR_DEBUG
M
 
Marc G. Fournier 已提交
1106
				LockDebug = atoi(optarg);
1107
#else
1108
				fprintf(stderr, "Lock debug not compiled in\n");
1109
#endif
1110
				break;
1111

1112 1113 1114 1115 1116
			case 'L':
				/* --------------------
				 *	turn off locking
				 * --------------------
				 */
1117 1118
				if (secure)
					lockingOff = 1;
T
Tom Lane 已提交
1119 1120
				break;

1121 1122 1123 1124 1125 1126 1127
			case 'N':
				/* ----------------
				 *	N - Don't use newline as a query delimiter
				 * ----------------
				 */
				UseNewLine = 0;
				break;
1128

1129 1130 1131 1132 1133
			case 'O':
				/* --------------------
				 *	allow system table structure modifications
				 * --------------------
				 */
1134 1135
				if (secure)		/* XXX safe to allow from client??? */
					allowSystemTableMods = true;
1136 1137
				break;

T
Tom Lane 已提交
1138 1139 1140 1141 1142
			case 'o':
				/* ----------------
				 *	o - send output (stdout and stderr) to the given file
				 * ----------------
				 */
1143 1144
				if (secure)
					StrNCpy(OutputFileName, optarg, MAXPGPATH);
T
Tom Lane 已提交
1145 1146
				break;

1147
			case 'p':
1148 1149 1150 1151 1152
				/* ----------------
				 *	p - special flag passed if backend was forked
				 *		by a postmaster.
				 * ----------------
				 */
1153 1154 1155 1156
				if (secure)
				{
					IsUnderPostmaster = true;
					DBName = optarg;
B
Bruce Momjian 已提交
1157 1158
					secure = false;		/* subsequent switches are NOT
										 * secure */
1159
				}
1160
				break;
1161

1162 1163 1164 1165 1166
			case 'Q':
				/* ----------------
				 *	Q - set Quiet mode (reduce debugging output)
				 * ----------------
				 */
M
 
Marc G. Fournier 已提交
1167
				Verbose = 0;
1168
				break;
1169

1170 1171
			case 'S':
				/* ----------------
V
Vadim B. Mikheev 已提交
1172
				 *	S - amount of sort memory to use in 1k bytes
1173 1174
				 * ----------------
				 */
1175
				{
1176 1177
					int			S;

V
Vadim B. Mikheev 已提交
1178
					S = atoi(optarg);
1179
					if (S >= 4 * BLCKSZ / 1024)
V
Vadim B. Mikheev 已提交
1180
						SortMem = S;
1181
				}
1182
				break;
1183 1184 1185 1186 1187 1188 1189 1190

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

1193
			case 'T':
1194 1195 1196 1197 1198
				/* ----------------
				 *	T - tracing options
				 * ----------------
				 */
				parse_options(optarg, secure);
M
 
Marc G. Fournier 已提交
1199 1200
				break;

1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229
			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;
				}
1230 1231
				break;

1232
			case 'v':
1233 1234
				if (secure)
					FrontendProtocol = (ProtocolVersion) atoi(optarg);
1235 1236
				break;

M
 
Marc G. Fournier 已提交
1237 1238
			case 'W':
				/* ----------------
1239
				 *	wait N seconds to allow attach from a debugger
M
 
Marc G. Fournier 已提交
1240 1241 1242 1243 1244
				 * ----------------
				 */
				sleep(atoi(optarg));
				break;

1245
			case 'x':
B
Bruce Momjian 已提交
1246
#ifdef NOT_USED					/* planner/xfunc.h */
1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274

				/*
				 * 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++;
				}
1275
#endif
1276
				break;
1277

1278 1279 1280 1281 1282 1283
			default:
				/* ----------------
				 *	default: bad command line option
				 * ----------------
				 */
				errs++;
T
Tom Lane 已提交
1284
				break;
1285 1286 1287
		}

	/* ----------------
1288 1289
	 *	get user name (needed now in case it is the default database name)
	 *	and check command line validity
1290 1291 1292 1293 1294
	 * ----------------
	 */
	SetPgUserName();
	userName = GetPgUserName();

1295
	if (IsUnderPostmaster)
1296
	{
1297 1298 1299 1300 1301 1302
		/* noninteractive case: nothing should be left after switches */
		if (errs || argc != optind || DBName == NULL)
		{
			usage(argv[0]);
			proc_exit(1);
		}
1303
	}
1304
	else
1305
	{
1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319
		/* interactive case: database name can be last arg on command line */
		if (errs || argc - optind > 1)
		{
			usage(argv[0]);
			proc_exit(1);
		}
		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]);
			proc_exit(1);
		}
1320 1321 1322 1323 1324 1325
	}

	if (ShowStats &&
		(ShowParserStats || ShowPlannerStats || ShowExecutorStats))
	{
		fprintf(stderr, "-s can not be used together with -t.\n");
1326
		proc_exit(1);
1327 1328 1329 1330 1331 1332 1333
	}

	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 "
B
Bruce Momjian 已提交
1334
			 "option or by setting the PGDATA environment variable.\n\n",
1335
				argv[0]);
1336
		proc_exit(1);
1337 1338
	}

1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349
	/*
	 * Set up additional info.
	 */

#ifdef CYR_RECODE
	SetCharSet();
#endif

	if (FindExec(pg_pathname, argv[0], "postgres") < 0)
		elog(FATAL, "%s: could not locate executable, bailing out...",
			 argv[0]);
1350

M
 
Marc G. Fournier 已提交
1351 1352 1353
	/*
	 * Find remote host name or address.
	 */
1354 1355 1356 1357 1358
	if (IsUnderPostmaster)
	{
		switch (MyProcPort->raddr.sa.sa_family)
		{
				struct hostent *host_ent;
M
 
Marc G. Fournier 已提交
1359

1360
			case AF_INET:
M
 
Marc G. Fournier 已提交
1361 1362 1363
				remote_info = remote_host = malloc(48);
				remote_port = ntohs(MyProcPort->raddr.in.sin_port);
				strcpy(remote_host, inet_ntoa(MyProcPort->raddr.in.sin_addr));
1364 1365
				if (HostnameLookup)
				{
M
 
Marc G. Fournier 已提交
1366
					host_ent = \
1367 1368
						gethostbyaddr((char *) &MyProcPort->raddr.in.sin_addr,
								   sizeof(MyProcPort->raddr.in.sin_addr),
M
 
Marc G. Fournier 已提交
1369
									  AF_INET);
1370 1371
					if (host_ent)
					{
M
 
Marc G. Fournier 已提交
1372
						strncpy(remote_host, host_ent->h_name, 48);
1373
						*(remote_host + 47) = '\0';
M
 
Marc G. Fournier 已提交
1374 1375
					}
				}
1376 1377 1378
				if (ShowPortNumber)
				{
					remote_info = malloc(strlen(remote_host) + 6);
M
 
Marc G. Fournier 已提交
1379 1380 1381
					sprintf(remote_info, "%s:%d", remote_host, remote_port);
				}
				break;
1382
			case AF_UNIX:
M
 
Marc G. Fournier 已提交
1383 1384
				remote_info = remote_host = "localhost";
				break;
1385
			default:
M
 
Marc G. Fournier 已提交
1386 1387 1388 1389 1390
				remote_info = remote_host = "unknown";
				break;
		}
	}

1391
	/* ----------------
M
 
Marc G. Fournier 已提交
1392
	 *	set process params for ps
1393 1394
	 * ----------------
	 */
1395 1396 1397
	if (IsUnderPostmaster)
	{
		PS_INIT_STATUS(real_argc, real_argv, argv[0],
M
 
Marc G. Fournier 已提交
1398
					   remote_info, userName, DBName);
1399
		PS_SET_STATUS("startup");
1400 1401
	}

1402
	/* ----------------
M
 
Marc G. Fournier 已提交
1403
	 *	print flags
1404 1405
	 * ----------------
	 */
M
 
Marc G. Fournier 已提交
1406
	if (Verbose)
1407
	{
1408 1409
		if (Verbose == 1)
		{
M
 
Marc G. Fournier 已提交
1410 1411
			TPRINTF(TRACE_VERBOSE, "started: host=%s user=%s database=%s",
					remote_host, userName, DBName);
1412 1413 1414
		}
		else
		{
M
 
Marc G. Fournier 已提交
1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428
			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');
		}
1429 1430
	}

1431
	/* ----------------
1432
	 *	initialize I/O
1433 1434
	 * ----------------
	 */
1435
	if (IsUnderPostmaster)
1436
	{
1437
		pq_init();				/* initialize libpq at backend startup */
1438
		whereToSendOutput = Remote;
1439
	}
1440 1441 1442
	else
		whereToSendOutput = Debug;

1443 1444 1445 1446
	/* ----------------
	 *	general initialization
	 * ----------------
	 */
1447 1448
	SetProcessingMode(InitProcessing);

M
 
Marc G. Fournier 已提交
1449 1450
	if (Verbose)
		TPRINTF(TRACE_VERBOSE, "InitPostgres");
1451 1452 1453

	InitPostgres(DBName);

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

1463 1464
	on_shmem_exit(remove_all_temp_relations, NULL);

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

1472
	pqsignal(SIGHUP, read_pg_options);	/* update pg_options from file */
1473 1474
	pqsignal(SIGINT, QueryCancelHandler);		/* cancel current query */
	pqsignal(SIGQUIT, handle_warn);		/* handle error */
M
 
Marc G. Fournier 已提交
1475
	pqsignal(SIGTERM, die);
B
Bruce Momjian 已提交
1476 1477 1478 1479 1480 1481 1482
	pqsignal(SIGPIPE, SIG_IGN); /* ignore failure to write to frontend */

	/*
	 * Note: if frontend closes connection, we will notice it and exit
	 * cleanly when control next returns to outer loop.  This seems safer
	 * than forcing exit in the midst of output during who-knows-what
	 * operation...
1483
	 */
M
 
Marc G. Fournier 已提交
1484
	pqsignal(SIGUSR1, quickdie);
1485 1486 1487
	pqsignal(SIGUSR2, Async_NotifyHandler);		/* flush also sinval cache */
	pqsignal(SIGCHLD, SIG_IGN); /* ignored, sent by LockOwners */
	pqsignal(SIGFPE, FloatExceptionHandler);
M
 
Marc G. Fournier 已提交
1488 1489 1490 1491

	if (whereToSendOutput == Remote &&
		PG_PROTOCOL_MAJOR(FrontendProtocol) >= 2)
	{
1492
		StringInfoData buf;
B
Bruce Momjian 已提交
1493

1494 1495 1496 1497 1498
		pq_beginmessage(&buf);
		pq_sendbyte(&buf, 'K');
		pq_sendint(&buf, (int32) MyProcPid, sizeof(int32));
		pq_sendint(&buf, (int32) MyCancelKey, sizeof(int32));
		pq_endmessage(&buf);
M
 
Marc G. Fournier 已提交
1499 1500 1501
		/* Need not flush since ReadyForQuery will do it. */
	}

1502 1503 1504
	if (!IsUnderPostmaster)
	{
		puts("\nPOSTGRES backend interactive interface ");
B
Bruce Momjian 已提交
1505
		puts("$Revision: 1.122 $ $Date: 1999/07/16 03:13:50 $\n");
1506 1507
	}

1508
	/* ----------------
1509 1510
	 *	POSTGRES main processing loop begins here
	 *
1511 1512 1513
	 *	if an exception is encountered, processing resumes here
	 *	so we abort the current transaction and start a new one.
	 *
1514
	 *	Note:  elog(ERROR) does a siglongjmp() to transfer control here.
1515 1516 1517 1518 1519
	 * ----------------
	 */

	if (sigsetjmp(Warn_restart, 1) != 0)
	{
1520
		InError = true;
1521 1522 1523

		time(&tim);

M
 
Marc G. Fournier 已提交
1524 1525
		if (Verbose)
			TPRINTF(TRACE_VERBOSE, "AbortCurrentTransaction");
1526 1527 1528

		AbortCurrentTransaction();
	}
1529 1530

	InError = false;
1531

1532 1533
	/*
	 * Non-error queries loop here.
1534 1535 1536 1537
	 */

	for (;;)
	{
1538 1539
		PS_SET_STATUS("idle");

B
Bruce Momjian 已提交
1540
		/* ----------------
1541 1542
		 *	 (1) tell the frontend we're ready for a new query.
		 *
B
Bruce Momjian 已提交
1543
		 *	 Note: this includes fflush()'ing the last of the prior output.
B
Bruce Momjian 已提交
1544 1545
		 * ----------------
		 */
1546
		ReadyForQuery(whereToSendOutput);
B
Bruce Momjian 已提交
1547

1548
		/* ----------------
1549
		 *	 (2) deal with pending asynchronous NOTIFY from other backends,
B
Bruce Momjian 已提交
1550
		 *	 and enable async.c's signal handler to execute NOTIFY directly.
1551 1552 1553 1554 1555 1556 1557 1558
		 * ----------------
		 */
		QueryCancel = false;	/* forget any earlier CANCEL signal */

		EnableNotifyInterrupt();

		/* ----------------
		 *	 (3) read a command.
1559 1560
		 * ----------------
		 */
B
Bruce Momjian 已提交
1561
		MemSet(parser_input, 0, MAX_PARSE_BUFFER);
1562

1563
		firstchar = ReadCommand(parser_input);
1564

1565
		QueryCancel = false;	/* forget any earlier CANCEL signal */
1566

1567
		/* ----------------
1568 1569 1570 1571 1572 1573 1574
		 *	 (4) disable async.c's signal handler.
		 * ----------------
		 */
		DisableNotifyInterrupt();

		/* ----------------
		 *	 (5) process the command.
1575 1576
		 * ----------------
		 */
1577 1578 1579
		switch (firstchar)
		{
				/* ----------------
1580 1581
				 *	'F' indicates a fastpath call.
				 *		XXX HandleFunctionRequest
1582 1583
				 * ----------------
				 */
1584
			case 'F':
1585 1586
				IsEmptyQuery = false;

1587
				/* start an xact for this function invocation */
M
 
Marc G. Fournier 已提交
1588 1589
				if (Verbose)
					TPRINTF(TRACE_VERBOSE, "StartTransactionCommand");
1590
				StartTransactionCommand();
1591

1592 1593
				HandleFunctionRequest();
				break;
1594

1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619
				/* ----------------
				 *	'Q' indicates a user query
				 * ----------------
				 */
			case 'Q':
				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 已提交
1620 1621
					if (Verbose)
						TPRINTF(TRACE_VERBOSE, "StartTransactionCommand");
1622
					StartTransactionCommand();
1623

1624 1625
					pg_exec_query(parser_input);

1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636
					if (ShowStats)
						ShowUsage();
				}
				break;

				/* ----------------
				 *	'X' means that the frontend is closing down the socket
				 * ----------------
				 */
			case 'X':
				pq_close();
1637
				proc_exit(0);
1638 1639 1640
				break;

			default:
M
 
Marc G. Fournier 已提交
1641
				elog(ERROR, "unknown frontend message was received");
1642 1643 1644
		}

		/* ----------------
1645
		 *	 (6) commit the current transaction
1646 1647
		 *
		 *	 Note: if we had an empty input buffer, then we didn't
1648
		 *	 call pg_exec_query, so we don't bother to commit this transaction.
1649 1650 1651 1652
		 * ----------------
		 */
		if (!IsEmptyQuery)
		{
M
 
Marc G. Fournier 已提交
1653 1654 1655
			if (Verbose)
				TPRINTF(TRACE_VERBOSE, "CommitTransactionCommand");
			PS_SET_STATUS("commit");
1656 1657 1658 1659
			CommitTransactionCommand();
		}
		else
		{
1660
			if (IsUnderPostmaster)
1661 1662 1663
				NullCommand(Remote);
		}
	}							/* infinite for-loop */
1664 1665

	proc_exit(0);				/* shouldn't get here... */
1666
	return 1;
1667 1668
}

1669
#ifndef HAVE_GETRUSAGE
1670
#else							/* HAVE_GETRUSAGE */
1671
#include <sys/resource.h>
1672
#endif	 /* HAVE_GETRUSAGE */
1673

1674 1675
struct rusage Save_r;
struct timeval Save_t;
1676 1677

void
1678
ResetUsage(void)
1679
{
1680 1681 1682 1683 1684 1685
	struct timezone tz;

	getrusage(RUSAGE_SELF, &Save_r);
	gettimeofday(&Save_t, &tz);
	ResetBufferUsage();
/*	  ResetTupleCount(); */
1686 1687 1688
}

void
1689
ShowUsage(void)
1690
{
1691 1692 1693
	struct timeval user,
				sys;
	struct timeval elapse_t;
1694
	struct timezone tz;
1695
	struct rusage r;
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 1733 1734 1735 1736 1737 1738 1739 1740

	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);
1741
#ifdef HAVE_GETRUSAGE
1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766
	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);
1767
#endif	 /* HAVE_GETRUSAGE */
1768 1769 1770
	fprintf(StatFp, "! postgres usage stats:\n");
	PrintBufferUsage(StatFp);
/*	   DisplayTupleCount(StatFp); */
1771
}
M
 
Marc G. Fournier 已提交
1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786

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

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

1787 1788
	if (assert_enabled)
	{
M
 
Marc G. Fournier 已提交
1789 1790 1791
		/* val != 0 should be trapped by previous Assert */
		elog(NOTICE, "Assert test successfull (val = %d)", val);
	}
1792 1793
	else
		elog(NOTICE, "Assert checking is disabled (val = %d)", val);
M
 
Marc G. Fournier 已提交
1794 1795 1796

	return val;
}
1797

M
 
Marc G. Fournier 已提交
1798 1799
#endif
#endif