Fix incorrect initialization of ProcGlobal->startupBufferPinWaitBufId.

tglsfdc · tglsfdc · commit 0f904c95a400 · 2011-08-02T13:24:06.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
@@ -2422,10 +2422,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
@@ -179,6 +179,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;
 
@@ -487,7 +490,6 @@ PublishStartupProcessInformation(void)
 
 	procglobal->startupProc = MyProc;
 	procglobal->startupProcPid = MyProcPid;
-	procglobal->startupBufferPinWaitBufId = 0;
 
 	SpinLockRelease(ProcStructLock);
 }
@@ -514,14 +516,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
@@ -145,7 +145,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
`@@ -2422,10 +2422,11 @@ LockBufferForCleanup(Buffer buffer)`
`2422`	`2422`	`/* Wait to be signaled by UnpinBuffer() */`
`2423`	`2423`	`if (InHotStandby)`
`2424`	`2424`	`{`
`2425`		`- /* Share the bufid that Startup process waits on */`
	`2425`	`+ /* Publish the bufid that Startup process waits on */`
`2426`	`2426`	`SetStartupBufferPinWaitBufId(buffer - 1);`
`2427`	`2427`	`/* Set alarm and then wait to be signaled by UnpinBuffer() */`
`2428`	`2428`	`ResolveRecoveryConflictWithBufferPin();`
	`2429`	`+ /* Reset the published bufid */`
`2429`	`2430`	`SetStartupBufferPinWaitBufId(-1);`
`2430`	`2431`	`}`
`2431`	`2432`	`else`