mirrors/postgres
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

On Windows, syslogger runs in two threads. The main thread processes config

Browse Source 
reload and rotation signals, and a helper thread reads messages from the
pipe and writes them to the log file. However, server code isn't generally
thread-safe, so if both try to do e.g palloc()/pfree() at the same time,
bad things will happen. To fix that, use a critical section (which is like
a mutex) to enforce that only one the threads are active at a time.
This commit is contained in:
Heikki Linnakangas 2010-04-16 09:51:54 +00:00
parent 9df6127bcf
commit b2bc12af46
1 changed files with 28 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
src/backend/postmaster/syslogger.c
View File

@ -18,7 +18,7 @@
* *
* *
* IDENTIFICATION * IDENTIFICATION
* $PostgreSQL: pgsql/src/backend/postmaster/syslogger.c,v 1.51.2.2 2010/04/01 20:12:28 heikki Exp $ * $PostgreSQL: pgsql/src/backend/postmaster/syslogger.c,v 1.51.2.3 2010/04/16 09:51:54 heikki Exp $
* *
*------------------------------------------------------------------------- *-------------------------------------------------------------------------
*/ */
@ -117,7 +117,7 @@ HANDLE syslogPipe[2] = {0, 0};
#ifdef WIN32 #ifdef WIN32
static HANDLE threadHandle = 0; static HANDLE threadHandle = 0;
static CRITICAL_SECTION sysfileSection; static CRITICAL_SECTION sysloggerSection;
#endif #endif
/* /*
@ -268,7 +268,8 @@ SysLoggerMain(int argc, char *argv[])
#ifdef WIN32 #ifdef WIN32
/* Fire up separate data transfer thread */ /* Fire up separate data transfer thread */
InitializeCriticalSection(&sysfileSection); InitializeCriticalSection(&sysloggerSection);
EnterCriticalSection(&sysloggerSection);
threadHandle = (HANDLE) _beginthreadex(NULL, 0, pipeThread, NULL, 0, NULL); threadHandle = (HANDLE) _beginthreadex(NULL, 0, pipeThread, NULL, 0, NULL);
if (threadHandle == 0) if (threadHandle == 0)
@ -423,8 +424,16 @@ SysLoggerMain(int argc, char *argv[])
* On Windows we leave it to a separate thread to transfer data and * On Windows we leave it to a separate thread to transfer data and
* detect pipe EOF. The main thread just wakes up once a second to * detect pipe EOF. The main thread just wakes up once a second to
* check for SIGHUP and rotation conditions. * check for SIGHUP and rotation conditions.
*
* Server code isn't generally thread-safe, so we ensure that only
* one of the threads is active at a time by entering the critical
* section whenever we're not sleeping.
*/ */
LeaveCriticalSection(&sysloggerSection);
pg_usleep(1000000L); pg_usleep(1000000L);
EnterCriticalSection(&sysloggerSection);
#endif /* WIN32 */ #endif /* WIN32 */
if (pipe_eof_seen) if (pipe_eof_seen)
@ -911,17 +920,9 @@ write_syslogger_file(const char *buffer, int count, int destination)
if (destination == LOG_DESTINATION_CSVLOG && csvlogFile == NULL) if (destination == LOG_DESTINATION_CSVLOG && csvlogFile == NULL)
open_csvlogfile(); open_csvlogfile();
#ifdef WIN32
EnterCriticalSection(&sysfileSection);
#endif
logfile = destination == LOG_DESTINATION_CSVLOG ? csvlogFile : syslogFile; logfile = destination == LOG_DESTINATION_CSVLOG ? csvlogFile : syslogFile;
rc = fwrite(buffer, 1, count, logfile); rc = fwrite(buffer, 1, count, logfile);
#ifdef WIN32
LeaveCriticalSection(&sysfileSection);
#endif
/* can't use ereport here because of possible recursion */ /* can't use ereport here because of possible recursion */
if (rc != count) if (rc != count)
write_stderr("could not write to log file: %s\n", strerror(errno)); write_stderr("could not write to log file: %s\n", strerror(errno));
@ -945,11 +946,21 @@ pipeThread(void *arg)
for (;;) for (;;)
{ {
DWORD bytesRead; DWORD bytesRead;
BOOL result;
if (!ReadFile(syslogPipe[0], result = ReadFile(syslogPipe[0],
logbuffer + bytes_in_logbuffer, logbuffer + bytes_in_logbuffer,
sizeof(logbuffer) - bytes_in_logbuffer, sizeof(logbuffer) - bytes_in_logbuffer,
&bytesRead, 0)) &bytesRead, 0);
/*
* Enter critical section before doing anything that might touch
* global state shared by the main thread. Anything that uses
* palloc()/pfree() in particular are not safe outside the critical
* section.
*/
EnterCriticalSection(&sysloggerSection);
if (!result)
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
@ -966,6 +977,7 @@ pipeThread(void *arg)
bytes_in_logbuffer += bytesRead; bytes_in_logbuffer += bytesRead;
process_pipe_input(logbuffer, &bytes_in_logbuffer); process_pipe_input(logbuffer, &bytes_in_logbuffer);
} }
LeaveCriticalSection(&sysloggerSection);
} }
/* We exit the above loop only upon detecting pipe EOF */ /* We exit the above loop only upon detecting pipe EOF */
@ -974,6 +986,7 @@ pipeThread(void *arg)
/* if there's any data left then force it out now */ /* if there's any data left then force it out now */
flush_pipe_input(logbuffer, &bytes_in_logbuffer); flush_pipe_input(logbuffer, &bytes_in_logbuffer);
LeaveCriticalSection(&sysloggerSection);
_endthread(); _endthread();
return 0; return 0;
} }
@ -1097,18 +1110,9 @@ logfile_rotate(bool time_based_rotation, int size_rotation_for)
_setmode(_fileno(fh), _O_TEXT); /* use CRLF line endings on Windows */ _setmode(_fileno(fh), _O_TEXT); /* use CRLF line endings on Windows */
#endif #endif
/* On Windows, need to interlock against data-transfer thread */
#ifdef WIN32
EnterCriticalSection(&sysfileSection);
#endif
fclose(syslogFile); fclose(syslogFile);
syslogFile = fh; syslogFile = fh;
#ifdef WIN32
LeaveCriticalSection(&sysfileSection);
#endif
/* instead of pfree'ing filename, remember it for next time */ /* instead of pfree'ing filename, remember it for next time */
if (last_file_name != NULL) if (last_file_name != NULL)
pfree(last_file_name); pfree(last_file_name);
@ -1164,18 +1168,9 @@ logfile_rotate(bool time_based_rotation, int size_rotation_for)
_setmode(_fileno(fh), _O_TEXT); /* use CRLF line endings on Windows */ _setmode(_fileno(fh), _O_TEXT); /* use CRLF line endings on Windows */
#endif #endif
/* On Windows, need to interlock against data-transfer thread */
#ifdef WIN32
EnterCriticalSection(&sysfileSection);
#endif
fclose(csvlogFile); fclose(csvlogFile);
csvlogFile = fh; csvlogFile = fh;
#ifdef WIN32
LeaveCriticalSection(&sysfileSection);
#endif
/* instead of pfree'ing filename, remember it for next time */ /* instead of pfree'ing filename, remember it for next time */
if (last_csv_file_name != NULL) if (last_csv_file_name != NULL)
pfree(last_csv_file_name); pfree(last_csv_file_name);