proc.c 26.8 KB
Newer Older
1 2
/*-------------------------------------------------------------------------
 *
3
 * proc.c
4
 *	  routines to manage per-process shared memory data structure
5
 *
B
Bruce Momjian 已提交
6
 * Portions Copyright (c) 1996-2002, PostgreSQL Global Development Group
B
Add:  
Bruce Momjian 已提交
7
 * Portions Copyright (c) 1994, Regents of the University of California
8 9 10
 *
 *
 * IDENTIFICATION
B
Bruce Momjian 已提交
11
 *	  $Header: /cvsroot/pgsql/src/backend/storage/lmgr/proc.c,v 1.121 2002/06/20 20:29:35 momjian Exp $
12 13 14 15 16
 *
 *-------------------------------------------------------------------------
 */
/*
 * Interface (a):
17
 *		ProcSleep(), ProcWakeup(),
18 19
 *		ProcQueueAlloc() -- create a shm queue for sleeping processes
 *		ProcQueueInit() -- create a queue without allocing memory
20 21 22 23 24 25 26 27
 *
 * Locking and waiting for buffers can cause the backend to be
 * put to sleep.  Whoever releases the lock, etc. wakes the
 * process up again (and gives it an error code so it knows
 * whether it was awoken on an error condition).
 *
 * Interface (b):
 *
28 29
 * ProcReleaseLocks -- frees the locks associated with current transaction
 *
30
 * ProcKill -- destroys the shared memory state (and locks)
31
 *		associated with the process.
32 33
 *
 * 5/15/91 -- removed the buffer pool based lock chain in favor
34 35 36 37 38 39
 *		of a shared memory lock chain.	The write-protection is
 *		more expensive if the lock chain is in the buffer pool.
 *		The only reason I kept the lock chain in the buffer pool
 *		in the first place was to allow the lock table to grow larger
 *		than available shared memory and that isn't going to work
 *		without a lot of unimplemented support anyway.
40
 */
41 42
#include "postgres.h"

43
#include <errno.h>
44
#include <signal.h>
45 46
#include <unistd.h>
#include <sys/time.h>
M
Marc G. Fournier 已提交
47

48
#include "miscadmin.h"
49
#include "access/xact.h"
50
#include "storage/ipc.h"
51
#include "storage/proc.h"
52
#include "storage/sinval.h"
53
#include "storage/spin.h"
54

55

B
Bruce Momjian 已提交
56
int			DeadlockTimeout = 1000;
M
 
Marc G. Fournier 已提交
57

J
Jan Wieck 已提交
58
PGPROC	   *MyProc = NULL;
59 60

/*
J
Jan Wieck 已提交
61
 * This spinlock protects the freelist of recycled PGPROC structures.
62
 * We cannot use an LWLock because the LWLock manager depends on already
J
Jan Wieck 已提交
63
 * having a PGPROC and a wait semaphore!  But these structures are touched
64 65
 * relatively infrequently (only at backend startup or shutdown) and not for
 * very long, so a spinlock is okay.
66
 */
67
static slock_t *ProcStructLock = NULL;
68 69 70

static PROC_HDR *ProcGlobal = NULL;

J
Jan Wieck 已提交
71
static PGPROC *DummyProc = NULL;
72

73
static bool waitingForLock = false;
74
static bool waitingForSignal = false;
75 76

static void ProcKill(void);
77
static void DummyProcKill(void);
78

V
Vadim B. Mikheev 已提交
79

80 81 82 83 84 85 86 87 88 89
/*
 * Report number of semaphores needed by InitProcGlobal.
 */
int
ProcGlobalSemas(int maxBackends)
{
	/* We need a sema per backend, plus one for the dummy process. */
	return maxBackends + 1;
}

90 91
/*
 * InitProcGlobal -
92
 *	  initializes the global process table. We put it here so that
93
 *	  the postmaster can do this initialization.
94
 *
95
 *	  We also create all the per-process semaphores we will need to support
96 97 98 99 100 101 102 103 104
 *	  the requested number of backends.  We used to allocate semaphores
 *	  only when backends were actually started up, but that is bad because
 *	  it lets Postgres fail under load --- a lot of Unix systems are
 *	  (mis)configured with small limits on the number of semaphores, and
 *	  running out when trying to start another backend is a common failure.
 *	  So, now we grab enough semaphores to support the desired max number
 *	  of backends immediately at initialization --- if the sysadmin has set
 *	  MaxBackends higher than his kernel will support, he'll find out sooner
 *	  rather than later.
105 106 107 108
 *
 *	  Another reason for creating semaphores here is that the semaphore
 *	  implementation typically requires us to create semaphores in the
 *	  postmaster, not in backends.
109 110
 */
void
111
InitProcGlobal(int maxBackends)
112
{
113
	bool		found = false;
114

115
	/* Create or attach to the ProcGlobal shared structure */
116
	ProcGlobal = (PROC_HDR *)
117
		ShmemInitStruct("Proc Header", sizeof(PROC_HDR), &found);
118

119 120
	/* --------------------
	 * We're the first - initialize.
121 122
	 * XXX if found should ever be true, it is a sign of impending doom ...
	 * ought to complain if so?
123 124 125
	 * --------------------
	 */
	if (!found)
126
	{
127
		int			i;
128

129
		ProcGlobal->freeProcs = INVALID_OFFSET;
130

B
Bruce Momjian 已提交
131
		/*
J
Jan Wieck 已提交
132
		 * Pre-create the PGPROC structures and create a semaphore for each.
133
		 */
134
		for (i = 0; i < maxBackends; i++)
135
		{
J
Jan Wieck 已提交
136
			PGPROC   *proc;
137

J
Jan Wieck 已提交
138
			proc = (PGPROC *) ShmemAlloc(sizeof(PGPROC));
139 140
			if (!proc)
				elog(FATAL, "cannot create new proc: out of memory");
J
Jan Wieck 已提交
141
			MemSet(proc, 0, sizeof(PGPROC));
142 143 144
			PGSemaphoreCreate(&proc->sem);
			proc->links.next = ProcGlobal->freeProcs;
			ProcGlobal->freeProcs = MAKE_OFFSET(proc);
145
		}
146 147

		/*
J
Jan Wieck 已提交
148
		 * Pre-allocate a PGPROC structure for dummy (checkpoint) processes,
149
		 * too.  This does not get linked into the freeProcs list.
150
		 */
J
Jan Wieck 已提交
151
		DummyProc = (PGPROC *) ShmemAlloc(sizeof(PGPROC));
152 153
		if (!DummyProc)
			elog(FATAL, "cannot create new proc: out of memory");
J
Jan Wieck 已提交
154
		MemSet(DummyProc, 0, sizeof(PGPROC));
155
		DummyProc->pid = 0;		/* marks DummyProc as not in use */
156
		PGSemaphoreCreate(&DummyProc->sem);
157 158 159 160

		/* Create ProcStructLock spinlock, too */
		ProcStructLock = (slock_t *) ShmemAlloc(sizeof(slock_t));
		SpinLockInit(ProcStructLock);
161 162 163
	}
}

164
/*
165
 * InitProcess -- initialize a per-process data structure for this backend
166 167
 */
void
168
InitProcess(void)
169
{
170
	SHMEM_OFFSET myOffset;
171 172
	/* use volatile pointer to prevent code rearrangement */
	volatile PROC_HDR *procglobal = ProcGlobal;
173 174

	/*
175 176
	 * ProcGlobal should be set by a previous call to InitProcGlobal (if
	 * we are a backend, we inherit this by fork() from the postmaster).
177
	 */
178
	if (procglobal == NULL)
179
		elog(PANIC, "InitProcess: Proc Header uninitialized");
180 181 182

	if (MyProc != NULL)
		elog(ERROR, "InitProcess: you already exist");
183

184
	/*
185
	 * Try to get a proc struct from the free list.  If this fails,
J
Jan Wieck 已提交
186
	 * we must be out of PGPROC structures (not to mention semaphores).
187
	 */
188
	SpinLockAcquire(ProcStructLock);
189

190
	myOffset = procglobal->freeProcs;
191 192

	if (myOffset != INVALID_OFFSET)
193
	{
J
Jan Wieck 已提交
194
		MyProc = (PGPROC *) MAKE_PTR(myOffset);
195
		procglobal->freeProcs = MyProc->links.next;
196
		SpinLockRelease(ProcStructLock);
197 198 199 200
	}
	else
	{
		/*
J
Jan Wieck 已提交
201
		 * If we reach here, all the PGPROCs are in use.  This is one of
202 203
		 * the possible places to detect "too many backends", so give the
		 * standard error message.
204
		 */
205
		SpinLockRelease(ProcStructLock);
206
		elog(FATAL, "Sorry, too many clients already");
207
	}
208

209
	/*
210 211
	 * Initialize all fields of MyProc, except for the semaphore which
	 * was prepared for us by InitProcGlobal.
212
	 */
213
	SHMQueueElemInit(&(MyProc->links));
214
	MyProc->errType = STATUS_OK;
215
	MyProc->xid = InvalidTransactionId;
216
	MyProc->xmin = InvalidTransactionId;
217 218
	MyProc->pid = MyProcPid;
	MyProc->databaseId = MyDatabaseId;
219
	MyProc->logRec.xrecoff = 0;
220 221 222
	MyProc->lwWaiting = false;
	MyProc->lwExclusive = false;
	MyProc->lwWaitLink = NULL;
223 224 225
	MyProc->waitLock = NULL;
	MyProc->waitHolder = NULL;
	SHMQueueInit(&(MyProc->procHolders));
226

227
	/*
228
	 * Arrange to clean up at backend exit.
229
	 */
230
	on_shmem_exit(ProcKill, 0);
231

232
	/*
233
	 * We might be reusing a semaphore that belonged to a failed process.
234 235
	 * So be careful and reinitialize its value here.
	 */
236
	PGSemaphoreReset(&MyProc->sem);
237

238
	/*
J
Jan Wieck 已提交
239
	 * Now that we have a PGPROC, we could try to acquire locks, so
B
Bruce Momjian 已提交
240
	 * initialize the deadlock checker.
241 242
	 */
	InitDeadLockChecking();
243 244
}

245 246 247 248
/*
 * InitDummyProcess -- create a dummy per-process data structure
 *
 * This is called by checkpoint processes so that they will have a MyProc
J
Jan Wieck 已提交
249
 * value that's real enough to let them wait for LWLocks.  The PGPROC and
250 251 252 253 254 255
 * sema that are assigned are the extra ones created during InitProcGlobal.
 */
void
InitDummyProcess(void)
{
	/*
256 257
	 * ProcGlobal should be set by a previous call to InitProcGlobal (we
	 * inherit this by fork() from the postmaster).
258 259
	 */
	if (ProcGlobal == NULL || DummyProc == NULL)
260
		elog(PANIC, "InitDummyProcess: Proc Header uninitialized");
261 262 263 264 265 266 267 268 269 270 271 272 273

	if (MyProc != NULL)
		elog(ERROR, "InitDummyProcess: you already exist");

	/*
	 * DummyProc should not presently be in use by anyone else
	 */
	if (DummyProc->pid != 0)
		elog(FATAL, "InitDummyProcess: DummyProc is in use by PID %d",
			 DummyProc->pid);
	MyProc = DummyProc;

	/*
274 275
	 * Initialize all fields of MyProc, except MyProc->sem which was set
	 * up by InitProcGlobal.
276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299
	 */
	MyProc->pid = MyProcPid;	/* marks DummyProc as in use by me */
	SHMQueueElemInit(&(MyProc->links));
	MyProc->errType = STATUS_OK;
	MyProc->xid = InvalidTransactionId;
	MyProc->xmin = InvalidTransactionId;
	MyProc->databaseId = MyDatabaseId;
	MyProc->logRec.xrecoff = 0;
	MyProc->lwWaiting = false;
	MyProc->lwExclusive = false;
	MyProc->lwWaitLink = NULL;
	MyProc->waitLock = NULL;
	MyProc->waitHolder = NULL;
	SHMQueueInit(&(MyProc->procHolders));

	/*
	 * Arrange to clean up at process exit.
	 */
	on_shmem_exit(DummyProcKill, 0);

	/*
	 * We might be reusing a semaphore that belonged to a failed process.
	 * So be careful and reinitialize its value here.
	 */
300
	PGSemaphoreReset(&MyProc->sem);
301 302
}

303 304 305
/*
 * Cancel any pending wait for lock, when aborting a transaction.
 *
306 307
 * Returns true if we had been waiting for a lock, else false.
 *
308 309 310 311
 * (Normally, this would only happen if we accept a cancel/die
 * interrupt while waiting; but an elog(ERROR) while waiting is
 * within the realm of possibility, too.)
 */
312
bool
313 314 315 316
LockWaitCancel(void)
{
	/* Nothing to do if we weren't waiting for a lock */
	if (!waitingForLock)
317 318
		return false;

319 320 321
	waitingForLock = false;

	/* Turn off the deadlock timer, if it's still running (see ProcSleep) */
322
	disable_sigalrm_interrupt();
323 324

	/* Unlink myself from the wait queue, if on it (might not be anymore!) */
325
	LWLockAcquire(LockMgrLock, LW_EXCLUSIVE);
326 327
	if (MyProc->links.next != INVALID_OFFSET)
		RemoveFromWaitQueue(MyProc);
328
	LWLockRelease(LockMgrLock);
H
Hiroshi Inoue 已提交
329

330 331 332
	/*
	 * Reset the proc wait semaphore to zero.  This is necessary in the
	 * scenario where someone else granted us the lock we wanted before we
B
Bruce Momjian 已提交
333 334 335 336 337
	 * were able to remove ourselves from the wait-list.  The semaphore
	 * will have been bumped to 1 by the would-be grantor, and since we
	 * are no longer going to wait on the sema, we have to force it back
	 * to zero. Otherwise, our next attempt to wait for a lock will fall
	 * through prematurely.
338
	 */
339
	PGSemaphoreReset(&MyProc->sem);
340 341

	/*
B
Bruce Momjian 已提交
342 343
	 * Return true even if we were kicked off the lock before we were able
	 * to remove ourselves.
344 345
	 */
	return true;
H
Hiroshi Inoue 已提交
346
}
347

348

349
/*
350 351 352 353 354 355 356 357
 * ProcReleaseLocks() -- release locks associated with current transaction
 *			at transaction commit or abort
 *
 * At commit, we release only locks tagged with the current transaction's XID,
 * leaving those marked with XID 0 (ie, session locks) undisturbed.  At abort,
 * we release all locks including XID 0, because we need to clean up after
 * a failure.  This logic will need extension if we ever support nested
 * transactions.
358
 *
359
 * Note that user locks are not released in either case.
360 361
 */
void
362
ProcReleaseLocks(bool isCommit)
363
{
364 365
	if (!MyProc)
		return;
366 367 368
	/* If waiting, get off wait queue (should only be needed after error) */
	LockWaitCancel();
	/* Release locks */
369 370
	LockReleaseAll(DEFAULT_LOCKMETHOD, MyProc,
				   !isCommit, GetCurrentTransactionId());
371 372 373 374 375
}


/*
 * ProcKill() -- Destroy the per-proc data structure for
376
 *		this process. Release any of its held LW locks.
377 378
 */
static void
379
ProcKill(void)
380
{
381 382 383
	/* use volatile pointer to prevent code rearrangement */
	volatile PROC_HDR *procglobal = ProcGlobal;

384
	Assert(MyProc != NULL);
385

386 387 388 389 390
	/* Release any LW locks I am holding */
	LWLockReleaseAll();

	/* Abort any buffer I/O in progress */
	AbortBufferIO();
391

392 393
	/* Get off any wait queue I might be on */
	LockWaitCancel();
394

395
	/* Remove from the standard lock table */
396
	LockReleaseAll(DEFAULT_LOCKMETHOD, MyProc, true, InvalidTransactionId);
397

398 399
#ifdef USER_LOCKS
	/* Remove from the user lock table */
400
	LockReleaseAll(USER_LOCKMETHOD, MyProc, true, InvalidTransactionId);
401
#endif
402

403
	SpinLockAcquire(ProcStructLock);
404

J
Jan Wieck 已提交
405
	/* Return PGPROC structure (and semaphore) to freelist */
406 407
	MyProc->links.next = procglobal->freeProcs;
	procglobal->freeProcs = MAKE_OFFSET(MyProc);
408

J
Jan Wieck 已提交
409
	/* PGPROC struct isn't mine anymore */
410
	MyProc = NULL;
411

412 413 414 415 416
	SpinLockRelease(ProcStructLock);
}

/*
 * DummyProcKill() -- Cut-down version of ProcKill for dummy (checkpoint)
J
Jan Wieck 已提交
417
 *		processes.	The PGPROC and sema are not released, only marked
418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435
 *		as not-in-use.
 */
static void
DummyProcKill(void)
{
	Assert(MyProc != NULL && MyProc == DummyProc);

	/* Release any LW locks I am holding */
	LWLockReleaseAll();

	/* Abort any buffer I/O in progress */
	AbortBufferIO();

	/* I can't be on regular lock queues, so needn't check */

	/* Mark DummyProc no longer in use */
	MyProc->pid = 0;

J
Jan Wieck 已提交
436
	/* PGPROC struct isn't mine anymore */
437
	MyProc = NULL;
438 439
}

440

441 442
/*
 * ProcQueue package: routines for putting processes to sleep
443
 *		and  waking them up
444 445 446 447 448 449 450 451
 */

/*
 * ProcQueueAlloc -- alloc/attach to a shared memory process queue
 *
 * Returns: a pointer to the queue or NULL
 * Side Effects: Initializes the queue if we allocated one
 */
452
#ifdef NOT_USED
453
PROC_QUEUE *
454 455
ProcQueueAlloc(char *name)
{
456 457
	bool		found;
	PROC_QUEUE *queue = (PROC_QUEUE *)
B
Bruce Momjian 已提交
458
	ShmemInitStruct(name, sizeof(PROC_QUEUE), &found);
459 460

	if (!queue)
461
		return NULL;
462 463
	if (!found)
		ProcQueueInit(queue);
464
	return queue;
465
}
466
#endif
467 468 469 470 471

/*
 * ProcQueueInit -- initialize a shared memory process queue
 */
void
472
ProcQueueInit(PROC_QUEUE *queue)
473
{
474 475
	SHMQueueInit(&(queue->links));
	queue->size = 0;
476 477 478 479 480 481
}


/*
 * ProcSleep -- put a process to sleep
 *
482 483
 * Caller must have set MyProc->heldLocks to reflect locks already held
 * on the lockable object by this process (under all XIDs).
484
 *
485
 * Locktable's masterLock must be held at entry, and will be held
486
 * at exit.
487
 *
488
 * Result: STATUS_OK if we acquired the lock, STATUS_ERROR if not (deadlock).
489
 *
490
 * ASSUME: that no one will fiddle with the queue until after
491
 *		we release the masterLock.
492 493
 *
 * NOTES: The process queue is now a priority queue for locking.
494 495 496
 *
 * P() on the semaphore should put us to sleep.  The process
 * semaphore is normally zero, so when we try to acquire it, we sleep.
497 498
 */
int
499
ProcSleep(LOCKMETHODTABLE *lockMethodTable,
500 501
		  LOCKMODE lockmode,
		  LOCK *lock,
502
		  HOLDER *holder)
503
{
504
	LOCKMETHODCTL *lockctl = lockMethodTable->ctl;
505
	LWLockId	masterLock = lockctl->masterLock;
506 507
	PROC_QUEUE *waitQueue = &(lock->waitProcs);
	int			myHeldLocks = MyProc->heldLocks;
508
	bool		early_deadlock = false;
J
Jan Wieck 已提交
509
	PGPROC	   *proc;
510
	int			i;
511

512
	/*
513 514 515 516 517 518
	 * Determine where to add myself in the wait queue.
	 *
	 * Normally I should go at the end of the queue.  However, if I already
	 * hold locks that conflict with the request of any previous waiter,
	 * put myself in the queue just in front of the first such waiter.
	 * This is not a necessary step, since deadlock detection would move
519 520 521
	 * me to before that waiter anyway; but it's relatively cheap to
	 * detect such a conflict immediately, and avoid delaying till
	 * deadlock timeout.
522
	 *
523 524
	 * Special case: if I find I should go in front of some waiter, check to
	 * see if I conflict with already-held locks or the requests before
525 526
	 * that waiter.  If not, then just grant myself the requested lock
	 * immediately.  This is the same as the test for immediate grant in
527 528
	 * LockAcquire, except we are only considering the part of the wait
	 * queue before my insertion point.
529 530
	 */
	if (myHeldLocks != 0)
V
Vadim B. Mikheev 已提交
531
	{
532 533
		int			aheadRequests = 0;

J
Jan Wieck 已提交
534
		proc = (PGPROC *) MAKE_PTR(waitQueue->links.next);
535
		for (i = 0; i < waitQueue->size; i++)
V
Vadim B. Mikheev 已提交
536
		{
537 538
			/* Must he wait for me? */
			if (lockctl->conflictTab[proc->waitLockMode] & myHeldLocks)
V
Vadim B. Mikheev 已提交
539
			{
540 541 542
				/* Must I wait for him ? */
				if (lockctl->conflictTab[lockmode] & proc->heldLocks)
				{
543
					/*
544 545 546 547 548
					 * Yes, so we have a deadlock.	Easiest way to clean
					 * up correctly is to call RemoveFromWaitQueue(), but
					 * we can't do that until we are *on* the wait queue.
					 * So, set a flag to check below, and break out of
					 * loop.
549 550 551
					 */
					early_deadlock = true;
					break;
552
				}
553 554 555 556 557 558 559 560
				/* I must go before this waiter.  Check special case. */
				if ((lockctl->conflictTab[lockmode] & aheadRequests) == 0 &&
					LockCheckConflicts(lockMethodTable,
									   lockmode,
									   lock,
									   holder,
									   MyProc,
									   NULL) == STATUS_OK)
561
				{
562 563 564
					/* Skip the wait and just grant myself the lock. */
					GrantLock(lock, holder, lockmode);
					return STATUS_OK;
565 566
				}
				/* Break out of loop to put myself before him */
V
Vadim B. Mikheev 已提交
567
				break;
568
			}
569 570
			/* Nope, so advance to next waiter */
			aheadRequests |= (1 << proc->waitLockMode);
J
Jan Wieck 已提交
571
			proc = (PGPROC *) MAKE_PTR(proc->links.next);
V
Vadim B. Mikheev 已提交
572
		}
B
Bruce Momjian 已提交
573

574 575 576 577
		/*
		 * If we fall out of loop normally, proc points to waitQueue head,
		 * so we will insert at tail of queue as desired.
		 */
578 579 580 581
	}
	else
	{
		/* I hold no locks, so I can't push in front of anyone. */
J
Jan Wieck 已提交
582
		proc = (PGPROC *) &(waitQueue->links);
V
Vadim B. Mikheev 已提交
583
	}
584

585 586 587
	/*
	 * Insert self into queue, ahead of the given proc (or at tail of
	 * queue).
588
	 */
589
	SHMQueueInsertBefore(&(proc->links), &(MyProc->links));
B
Bruce Momjian 已提交
590
	waitQueue->size++;
591

592
	lock->waitMask |= (1 << lockmode);
593

J
Jan Wieck 已提交
594
	/* Set up wait information in PGPROC object, too */
595 596 597 598
	MyProc->waitLock = lock;
	MyProc->waitHolder = holder;
	MyProc->waitLockMode = lockmode;

599
	MyProc->errType = STATUS_OK;	/* initialize result for success */
600 601 602

	/*
	 * If we detected deadlock, give up without waiting.  This must agree
603 604
	 * with HandleDeadLock's recovery code, except that we shouldn't
	 * release the semaphore since we haven't tried to lock it yet.
605 606 607 608 609 610 611
	 */
	if (early_deadlock)
	{
		RemoveFromWaitQueue(MyProc);
		MyProc->errType = STATUS_ERROR;
		return STATUS_ERROR;
	}
612

613 614 615
	/* mark that we are waiting for a lock */
	waitingForLock = true;

616
	/*
617
	 * Release the locktable's masterLock.
618
	 *
619 620 621 622
	 * NOTE: this may also cause us to exit critical-section state, possibly
	 * allowing a cancel/die interrupt to be accepted. This is OK because
	 * we have recorded the fact that we are waiting for a lock, and so
	 * LockWaitCancel will clean up if cancel/die happens.
623
	 */
624
	LWLockRelease(masterLock);
625

626
	/*
627 628 629 630 631 632 633
	 * Set timer so we can wake up after awhile and check for a deadlock.
	 * If a deadlock is detected, the handler releases the process's
	 * semaphore and sets MyProc->errType = STATUS_ERROR, allowing us to
	 * know that we must report failure rather than success.
	 *
	 * By delaying the check until we've waited for a bit, we can avoid
	 * running the rather expensive deadlock-check code in most cases.
634
	 */
635
	if (!enable_sigalrm_interrupt(DeadlockTimeout))
636
		elog(FATAL, "ProcSleep: Unable to set timer for process wakeup");
637

638
	/*
639 640
	 * If someone wakes us between LWLockRelease and PGSemaphoreLock,
	 * PGSemaphoreLock will not block.  The wakeup is "saved" by the
641
	 * semaphore implementation.  Note also that if HandleDeadLock is
642
	 * invoked but does not detect a deadlock, PGSemaphoreLock() will
643 644
	 * continue to wait.  There used to be a loop here, but it was useless
	 * code...
645 646 647 648 649 650
	 *
	 * We pass interruptOK = true, which eliminates a window in which
	 * cancel/die interrupts would be held off undesirably.  This is a
	 * promise that we don't mind losing control to a cancel/die interrupt
	 * here.  We don't, because we have no state-change work to do after
	 * being granted the lock (the grantor did it all).
651
	 */
652
	PGSemaphoreLock(&MyProc->sem, true);
653

654
	/*
655
	 * Disable the timer, if it's still running
B
Bruce Momjian 已提交
656
	 */
657
	if (!disable_sigalrm_interrupt())
658
		elog(FATAL, "ProcSleep: Unable to disable timer for process wakeup");
B
Bruce Momjian 已提交
659

660 661 662 663 664
	/*
	 * Now there is nothing for LockWaitCancel to do.
	 */
	waitingForLock = false;

665
	/*
666
	 * Re-acquire the locktable's masterLock.
667
	 */
668
	LWLockAcquire(masterLock, LW_EXCLUSIVE);
669

670 671 672 673
	/*
	 * We don't have to do anything else, because the awaker did all the
	 * necessary update of the lock table and MyProc.
	 */
674
	return MyProc->errType;
675 676 677 678 679 680
}


/*
 * ProcWakeup -- wake up a process by releasing its private semaphore.
 *
681
 *	 Also remove the process from the wait queue and set its links invalid.
682
 *	 RETURN: the next process in the wait queue.
683 684 685 686
 *
 * XXX: presently, this code is only used for the "success" case, and only
 * works correctly for that case.  To clean up in failure case, would need
 * to twiddle the lock's request counts too --- see RemoveFromWaitQueue.
687
 */
J
Jan Wieck 已提交
688 689
PGPROC *
ProcWakeup(PGPROC *proc, int errType)
690
{
J
Jan Wieck 已提交
691
	PGPROC	   *retProc;
692

693
	/* assume that masterLock has been acquired */
694

695
	/* Proc should be sleeping ... */
696 697
	if (proc->links.prev == INVALID_OFFSET ||
		proc->links.next == INVALID_OFFSET)
J
Jan Wieck 已提交
698
		return (PGPROC *) NULL;
699

700
	/* Save next process before we zap the list link */
J
Jan Wieck 已提交
701
	retProc = (PGPROC *) MAKE_PTR(proc->links.next);
702

703
	/* Remove process from wait queue */
704
	SHMQueueDelete(&(proc->links));
705
	(proc->waitLock->waitProcs.size)--;
706

707 708 709
	/* Clean up process' state and pass it the ok/fail signal */
	proc->waitLock = NULL;
	proc->waitHolder = NULL;
710 711
	proc->errType = errType;

712
	/* And awaken it */
713
	PGSemaphoreUnlock(&proc->sem);
714 715

	return retProc;
716 717 718 719
}

/*
 * ProcLockWakeup -- routine for waking up processes when a lock is
720 721
 *		released (or a prior waiter is aborted).  Scan all waiters
 *		for lock, waken any that are no longer blocked.
722
 */
723 724
void
ProcLockWakeup(LOCKMETHODTABLE *lockMethodTable, LOCK *lock)
725
{
726 727 728
	LOCKMETHODCTL *lockctl = lockMethodTable->ctl;
	PROC_QUEUE *waitQueue = &(lock->waitProcs);
	int			queue_size = waitQueue->size;
J
Jan Wieck 已提交
729
	PGPROC	   *proc;
730
	int			aheadRequests = 0;
M
 
Marc G. Fournier 已提交
731

732
	Assert(queue_size >= 0);
733

734 735
	if (queue_size == 0)
		return;
736

J
Jan Wieck 已提交
737
	proc = (PGPROC *) MAKE_PTR(waitQueue->links.next);
738

739 740
	while (queue_size-- > 0)
	{
B
Bruce Momjian 已提交
741
		LOCKMODE	lockmode = proc->waitLockMode;
M
 
Marc G. Fournier 已提交
742 743

		/*
744 745
		 * Waken if (a) doesn't conflict with requests of earlier waiters,
		 * and (b) doesn't conflict with already-held locks.
M
 
Marc G. Fournier 已提交
746
		 */
747
		if ((lockctl->conflictTab[lockmode] & aheadRequests) == 0 &&
748 749 750 751 752 753
			LockCheckConflicts(lockMethodTable,
							   lockmode,
							   lock,
							   proc->waitHolder,
							   proc,
							   NULL) == STATUS_OK)
M
 
Marc G. Fournier 已提交
754
		{
755 756 757
			/* OK to waken */
			GrantLock(lock, proc->waitHolder, lockmode);
			proc = ProcWakeup(proc, STATUS_OK);
B
Bruce Momjian 已提交
758

759
			/*
B
Bruce Momjian 已提交
760 761 762
			 * ProcWakeup removes proc from the lock's waiting process
			 * queue and returns the next proc in chain; don't use proc's
			 * next-link, because it's been cleared.
763
			 */
M
 
Marc G. Fournier 已提交
764
		}
765
		else
766
		{
B
Bruce Momjian 已提交
767 768 769 770
			/*
			 * Cannot wake this guy. Remember his request for later
			 * checks.
			 */
771
			aheadRequests |= (1 << lockmode);
J
Jan Wieck 已提交
772
			proc = (PGPROC *) MAKE_PTR(proc->links.next);
773
		}
M
 
Marc G. Fournier 已提交
774
	}
775 776

	Assert(waitQueue->size >= 0);
777 778 779
}

/* --------------------
780
 * We only get to this routine if we got SIGALRM after DeadlockTimeout
781 782 783 784
 * while waiting for a lock to be released by some other process.  Look
 * to see if there's a deadlock; if not, just return and continue waiting.
 * If we have a real deadlock, remove ourselves from the lock's wait queue
 * and signal an error to ProcSleep.
785 786
 * --------------------
 */
787
void
788
HandleDeadLock(SIGNAL_ARGS)
789
{
790
	int			save_errno = errno;
791

792
	/*
B
Bruce Momjian 已提交
793 794
	 * Acquire locktable lock.	Note that the SIGALRM interrupt had better
	 * not be enabled anywhere that this process itself holds the
795
	 * locktable lock, else this will wait forever.  Also note that
796 797
	 * LWLockAcquire creates a critical section, so that this routine
	 * cannot be interrupted by cancel/die interrupts.
798
	 */
799
	LWLockAcquire(LockMgrLock, LW_EXCLUSIVE);
800

801
	/*
802 803 804
	 * Check to see if we've been awoken by anyone in the interim.
	 *
	 * If we have we can return and resume our transaction -- happy day.
805 806
	 * Before we are awoken the process releasing the lock grants it to us
	 * so we know that we don't have to wait anymore.
807
	 *
808
	 * We check by looking to see if we've been unlinked from the wait queue.
809 810 811 812
	 * This is quicker than checking our semaphore's state, since no
	 * kernel call is needed, and it is safe because we hold the locktable
	 * lock.
	 *
813 814 815 816
	 */
	if (MyProc->links.prev == INVALID_OFFSET ||
		MyProc->links.next == INVALID_OFFSET)
	{
817
		LWLockRelease(LockMgrLock);
818
		errno = save_errno;
819 820 821
		return;
	}

822
#ifdef LOCK_DEBUG
B
Bruce Momjian 已提交
823 824
	if (Debug_deadlocks)
		DumpAllLocks();
825 826
#endif

827
	if (!DeadLockCheck(MyProc))
B
Bruce Momjian 已提交
828
	{
829
		/* No deadlock, so keep waiting */
830
		LWLockRelease(LockMgrLock);
831
		errno = save_errno;
B
Bruce Momjian 已提交
832 833 834
		return;
	}

835
	/*
836 837 838
	 * Oops.  We have a deadlock.
	 *
	 * Get this process out of wait state.
839
	 */
840 841
	RemoveFromWaitQueue(MyProc);

842 843 844
	/*
	 * Set MyProc->errType to STATUS_ERROR so that ProcSleep will report
	 * an error after we return from this signal handler.
845 846
	 */
	MyProc->errType = STATUS_ERROR;
847

848 849 850
	/*
	 * Unlock my semaphore so that the interrupted ProcSleep() call can
	 * finish.
851
	 */
852
	PGSemaphoreUnlock(&MyProc->sem);
853

854 855 856 857 858 859 860 861 862
	/*
	 * We're done here.  Transaction abort caused by the error that
	 * ProcSleep will raise will cause any other locks we hold to be
	 * released, thus allowing other processes to wake up; we don't need
	 * to do that here. NOTE: an exception is that releasing locks we hold
	 * doesn't consider the possibility of waiters that were blocked
	 * behind us on the lock we just failed to get, and might now be
	 * wakable because we're not in front of them anymore.  However,
	 * RemoveFromWaitQueue took care of waking up any such processes.
863
	 */
864
	LWLockRelease(LockMgrLock);
865
	errno = save_errno;
866 867 868
}


869 870 871 872 873 874 875 876 877 878 879 880
/*
 * ProcWaitForSignal - wait for a signal from another backend.
 *
 * This can share the semaphore normally used for waiting for locks,
 * since a backend could never be waiting for a lock and a signal at
 * the same time.  As with locks, it's OK if the signal arrives just
 * before we actually reach the waiting state.
 */
void
ProcWaitForSignal(void)
{
	waitingForSignal = true;
881
	PGSemaphoreLock(&MyProc->sem, true);
882 883 884 885 886 887 888 889 890 891 892 893 894
	waitingForSignal = false;
}

/*
 * ProcCancelWaitForSignal - clean up an aborted wait for signal
 *
 * We need this in case the signal arrived after we aborted waiting,
 * or if it arrived but we never reached ProcWaitForSignal() at all.
 * Caller should call this after resetting the signal request status.
 */
void
ProcCancelWaitForSignal(void)
{
895
	PGSemaphoreReset(&MyProc->sem);
896 897 898 899 900 901 902 903 904
	waitingForSignal = false;
}

/*
 * ProcSendSignal - send a signal to a backend identified by BackendId
 */
void
ProcSendSignal(BackendId procId)
{
J
Jan Wieck 已提交
905
	PGPROC	   *proc = BackendIdGetProc(procId);
906 907

	if (proc != NULL)
908
		PGSemaphoreUnlock(&proc->sem);
909 910 911
}


912 913 914 915 916 917 918 919 920
/*****************************************************************************
 * SIGALRM interrupt support
 *
 * Maybe these should be in pqsignal.c?
 *****************************************************************************/

/*
 * Enable the SIGALRM interrupt to fire after the specified delay
 *
921
 * Delay is given in milliseconds.	Caller should be sure a SIGALRM
922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941
 * signal handler is installed before this is called.
 *
 * Returns TRUE if okay, FALSE on failure.
 */
bool
enable_sigalrm_interrupt(int delayms)
{
#ifndef __BEOS__
	struct itimerval timeval,
				dummy;

	MemSet(&timeval, 0, sizeof(struct itimerval));
	timeval.it_value.tv_sec = delayms / 1000;
	timeval.it_value.tv_usec = (delayms % 1000) * 1000;
	if (setitimer(ITIMER_REAL, &timeval, &dummy))
		return false;
#else
	/* BeOS doesn't have setitimer, but has set_alarm */
	bigtime_t	time_interval;

942
	time_interval = delayms * 1000;		/* usecs */
943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972
	if (set_alarm(time_interval, B_ONE_SHOT_RELATIVE_ALARM) < 0)
		return false;
#endif

	return true;
}

/*
 * Disable the SIGALRM interrupt, if it has not yet fired
 *
 * Returns TRUE if okay, FALSE on failure.
 */
bool
disable_sigalrm_interrupt(void)
{
#ifndef __BEOS__
	struct itimerval timeval,
				dummy;

	MemSet(&timeval, 0, sizeof(struct itimerval));
	if (setitimer(ITIMER_REAL, &timeval, &dummy))
		return false;
#else
	/* BeOS doesn't have setitimer, but has set_alarm */
	if (set_alarm(B_INFINITE_TIMEOUT, B_PERIODIC_ALARM) < 0)
		return false;
#endif

	return true;
}