Fix incorrect initialization of ProcGlobal->startupBufferPinWaitBufId.

tglsfdc · tglsfdc · commit 0dd6a09e3d12 · 2011-08-02T13:24:00.000-04:00
It was initialized in the wrong place and to the wrong value.  With bad
luck this could result in incorrect query-cancellation failures in hot
standby sessions, should a HS backend be holding pin on buffer number 1
while trying to acquire a lock.
diff --git a/src/backend/storage/buffer/bufmgr.c b/src/backend/storage/buffer/bufmgr.c
@@ -2449,10 +2449,11 @@ LockBufferForCleanup(Buffer buffer)
 		/* Wait to be signaled by UnpinBuffer() */
 		if (InHotStandby)
 		{
-			/* Share the bufid that Startup process waits on */
+			/* Publish the bufid that Startup process waits on */
 			SetStartupBufferPinWaitBufId(buffer - 1);
 			/* Set alarm and then wait to be signaled by UnpinBuffer() */
 			ResolveRecoveryConflictWithBufferPin();
+			/* Reset the published bufid */
 			SetStartupBufferPinWaitBufId(-1);
 		}
 		else
diff --git a/src/backend/storage/lmgr/proc.c b/src/backend/storage/lmgr/proc.c
@@ -180,6 +180,9 @@ InitProcGlobal(void)
 	 */
 	ProcGlobal->freeProcs = NULL;
 	ProcGlobal->autovacFreeProcs = NULL;
+	ProcGlobal->startupProc = NULL;
+	ProcGlobal->startupProcPid = 0;
+	ProcGlobal->startupBufferPinWaitBufId = -1;
 
 	ProcGlobal->spins_per_delay = DEFAULT_SPINS_PER_DELAY;
 
@@ -499,7 +502,6 @@ PublishStartupProcessInformation(void)
 
 	procglobal->startupProc = MyProc;
 	procglobal->startupProcPid = MyProcPid;
-	procglobal->startupBufferPinWaitBufId = 0;
 
 	SpinLockRelease(ProcStructLock);
 }
@@ -526,14 +528,10 @@ SetStartupBufferPinWaitBufId(int bufid)
 int
 GetStartupBufferPinWaitBufId(void)
 {
-	int			bufid;
-
 	/* use volatile pointer to prevent code rearrangement */
 	volatile PROC_HDR *procglobal = ProcGlobal;
 
-	bufid = procglobal->startupBufferPinWaitBufId;
-
-	return bufid;
+	return procglobal->startupBufferPinWaitBufId;
 }
 
 /*
diff --git a/src/include/storage/proc.h b/src/include/storage/proc.h
@@ -159,7 +159,7 @@ typedef struct PROC_HDR
 	/* The proc of the Startup process, since not in ProcArray */
 	PGPROC	   *startupProc;
 	int			startupProcPid;
-	/* Buffer id of the buffer that Startup process waits for pin on */
+	/* Buffer id of the buffer that Startup process waits for pin on, or -1 */
 	int			startupBufferPinWaitBufId;
 } PROC_HDR;
 

Original file line number	Diff line number	Diff line change
`@@ -2449,10 +2449,11 @@ LockBufferForCleanup(Buffer buffer)`
`2449`	`2449`	`/* Wait to be signaled by UnpinBuffer() */`
`2450`	`2450`	`if (InHotStandby)`
`2451`	`2451`	`{`
`2452`		`- /* Share the bufid that Startup process waits on */`
	`2452`	`+ /* Publish the bufid that Startup process waits on */`
`2453`	`2453`	`SetStartupBufferPinWaitBufId(buffer - 1);`
`2454`	`2454`	`/* Set alarm and then wait to be signaled by UnpinBuffer() */`
`2455`	`2455`	`ResolveRecoveryConflictWithBufferPin();`
	`2456`	`+ /* Reset the published bufid */`
`2456`	`2457`	`SetStartupBufferPinWaitBufId(-1);`
`2457`	`2458`	`}`
`2458`	`2459`	`else`