#include "postgres.h"
#include "libpq/pqsignal.h"

Include dependency graph for signal.c:

Functions
static DWORD WINAPI	pg_signal_thread (LPVOID param)

static BOOL WINAPI	pg_console_handler (DWORD dwCtrlType)

void	pg_usleep (long microsec)

void	pgwin32_signal_initialize (void)

void	pgwin32_dispatch_queued_signals (void)

int	pqsigprocmask (int how, const sigset_t set, sigset_t oset)

int	pqsigaction (int signum, const struct sigaction act, struct sigaction oldact)

HANDLE	pgwin32_create_signal_listener (pid_t pid)

void	pg_queue_signal (int signum)

Variables
volatile int	pg_signal_queue

int	pg_signal_mask

HANDLE	pgwin32_signal_event

HANDLE	pgwin32_initial_signal_pipe = INVALID_HANDLE_VALUE

static CRITICAL_SECTION	pg_signal_crit_sec

static struct sigaction	pg_signal_array [PG_SIGNAL_COUNT]

static pqsigfunc	pg_signal_defaults [PG_SIGNAL_COUNT]

Function Documentation

◆ pg_console_handler()

static BOOL WINAPI pg_console_handler ( DWORD dwCtrlType )

static

Definition at line 377 of file signal.c.

{
    if (dwCtrlType == CTRL_C_EVENT ||
        dwCtrlType == CTRL_BREAK_EVENT ||
        dwCtrlType == CTRL_CLOSE_EVENT ||
        dwCtrlType == CTRL_SHUTDOWN_EVENT)
    {
        pg_queue_signal(SIGINT);
        return TRUE;
    }
    return FALSE;
}

References pg_queue_signal().

Referenced by pgwin32_signal_initialize().

◆ pg_queue_signal()

void pg_queue_signal ( int signum )

Definition at line 259 of file signal.c.

{
    Assert(pgwin32_signal_event != NULL);
    if (signum >= PG_SIGNAL_COUNT || signum <= 0)
        return;                 /* ignore any bad signal number */
 
    EnterCriticalSection(&pg_signal_crit_sec);
    pg_signal_queue |= sigmask(signum);
    LeaveCriticalSection(&pg_signal_crit_sec);
 
    SetEvent(pgwin32_signal_event);
}

References Assert(), PG_SIGNAL_COUNT, pg_signal_crit_sec, pg_signal_queue, pgwin32_signal_event, and sigmask.

Referenced by pg_console_handler(), pg_signal_thread(), and pg_timer_thread().

◆ pg_signal_thread()

static DWORD WINAPI pg_signal_thread ( LPVOID param )

static

Definition at line 274 of file signal.c.

{
    char        pipename[128];
    HANDLE      pipe = pgwin32_initial_signal_pipe;
 
    /* Set up pipe name, in case we have to re-create the pipe. */
    snprintf(pipename, sizeof(pipename), "\\\\.\\pipe\\pgsignal_%lu", GetCurrentProcessId());
 
    for (;;)
    {
        BOOL        fConnected;
 
        /* Create a new pipe instance if we don't have one. */
        if (pipe == INVALID_HANDLE_VALUE)
        {
            pipe = CreateNamedPipe(pipename, PIPE_ACCESS_DUPLEX,
                                   PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE | PIPE_WAIT,
                                   PIPE_UNLIMITED_INSTANCES, 16, 16, 1000, NULL);
 
            if (pipe == INVALID_HANDLE_VALUE)
            {
                write_stderr("could not create signal listener pipe: error code %lu; retrying\n", GetLastError());
                SleepEx(500, FALSE);
                continue;
            }
        }
 
        /*
         * Wait for a client to connect.  If something connects before we
         * reach here, we'll get back a "failure" with ERROR_PIPE_CONNECTED,
         * which is actually a success (way to go, Microsoft).
         */
        fConnected = ConnectNamedPipe(pipe, NULL) ? TRUE : (GetLastError() == ERROR_PIPE_CONNECTED);
        if (fConnected)
        {
            /*
             * We have a connection from a would-be signal sender. Process it.
             */
            BYTE        sigNum;
            DWORD       bytes;
 
            if (ReadFile(pipe, &sigNum, 1, &bytes, NULL) &&
                bytes == 1)
            {
                /*
                 * Queue the signal before responding to the client.  In this
                 * way, it's guaranteed that once kill() has returned in the
                 * signal sender, the next CHECK_FOR_INTERRUPTS() in the
                 * signal recipient will see the signal.  (This is a stronger
                 * guarantee than POSIX makes; maybe we don't need it?  But
                 * without it, we've seen timing bugs on Windows that do not
                 * manifest on any known Unix.)
                 */
                pg_queue_signal(sigNum);
 
                /*
                 * Write something back to the client, allowing its
                 * CallNamedPipe() call to terminate.
                 */
                WriteFile(pipe, &sigNum, 1, &bytes, NULL);  /* Don't care if it
                                                             * works or not */
 
                /*
                 * We must wait for the client to read the data before we can
                 * disconnect, else the data will be lost.  (If the WriteFile
                 * call failed, there'll be nothing in the buffer, so this
                 * shouldn't block.)
                 */
                FlushFileBuffers(pipe);
            }
            else
            {
                /*
                 * If we fail to read a byte from the client, assume it's the
                 * client's problem and do nothing.  Perhaps it'd be better to
                 * force a pipe close and reopen?
                 */
            }
 
            /* Disconnect from client so that we can re-use the pipe. */
            DisconnectNamedPipe(pipe);
        }
        else
        {
            /*
             * Connection failed.  Cleanup and try again.
             *
             * This should never happen.  If it does, there's a window where
             * we'll miss signals until we manage to re-create the pipe.
             * However, just trying to use the same pipe again is probably not
             * going to work, so we have little choice.
             */
            CloseHandle(pipe);
            pipe = INVALID_HANDLE_VALUE;
        }
    }
    return 0;
}

References pg_queue_signal(), pgwin32_initial_signal_pipe, snprintf, and write_stderr.

Referenced by pgwin32_signal_initialize().

◆ pg_usleep()

void pg_usleep ( long microsec )

Definition at line 53 of file signal.c.

{
    if (unlikely(pgwin32_signal_event == NULL))
    {
        /*
         * If we're reached by pgwin32_open_handle() early in startup before
         * the signal event is set up, just fall back to a regular
         * non-interruptible sleep.
         */
        SleepEx((microsec < 500 ? 1 : (microsec + 500) / 1000), FALSE);
        return;
    }
 
    if (WaitForSingleObject(pgwin32_signal_event,
                            (microsec < 500 ? 1 : (microsec + 500) / 1000))
        == WAIT_OBJECT_0)
    {
        pgwin32_dispatch_queued_signals();
        errno = EINTR;
        return;
    }
}

References EINTR, pgwin32_dispatch_queued_signals(), pgwin32_signal_event, and unlikely.

Referenced by _bt_pendingfsm_finalize(), AcceptConnection(), auth_delay_checks(), AutoVacLauncherMain(), AutoVacWorkerMain(), BackendInitialize(), BackgroundWorkerMain(), BackgroundWriterMain(), CheckpointerMain(), ConditionalXactLockTableWait(), CountOtherDBBackends(), CreateCheckPoint(), do_watch(), exec_prog(), FileReadV(), FileWriteV(), get_controlfile_by_exact_path(), InitPostgres(), main(), perform_spin_delay(), pgarch_ArchiverCopyLoop(), pgwin32_recv(), read_local_xlog_page_guts(), regression_main(), RequestCheckpoint(), ResolveRecoveryConflictWithDatabase(), ResolveRecoveryConflictWithVirtualXIDs(), StartupXLOG(), threadRun(), vacuum_delay_point(), wait_for_connection_state(), wait_for_end_recovery(), wait_for_postmaster_promote(), wait_for_postmaster_start(), wait_for_postmaster_stop(), wait_pid(), WaitExceedsMaxStandbyDelay(), WALDumpOpenSegment(), WalSndWaitStopping(), WalWriterMain(), XactLockTableWait(), and XLogFlush().

◆ pgwin32_create_signal_listener()

HANDLE pgwin32_create_signal_listener ( pid_t pid )

Definition at line 227 of file signal.c.

{
    char        pipename[128];
    HANDLE      pipe;
 
    snprintf(pipename, sizeof(pipename), "\\\\.\\pipe\\pgsignal_%u", (int) pid);
 
    pipe = CreateNamedPipe(pipename, PIPE_ACCESS_DUPLEX,
                           PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE | PIPE_WAIT,
                           PIPE_UNLIMITED_INSTANCES, 16, 16, 1000, NULL);
 
    if (pipe == INVALID_HANDLE_VALUE)
        ereport(ERROR,
                (errmsg("could not create signal listener pipe for PID %d: error code %lu",
                        (int) pid, GetLastError())));
 
    return pipe;
}

References ereport, errmsg(), ERROR, and snprintf.

◆ pgwin32_dispatch_queued_signals()

void pgwin32_dispatch_queued_signals ( void )

Definition at line 120 of file signal.c.

{
    int         exec_mask;
 
    Assert(pgwin32_signal_event != NULL);
    EnterCriticalSection(&pg_signal_crit_sec);
    while ((exec_mask = UNBLOCKED_SIGNAL_QUEUE()) != 0)
    {
        /* One or more unblocked signals queued for execution */
        int         i;
 
        for (i = 1; i < PG_SIGNAL_COUNT; i++)
        {
            if (exec_mask & sigmask(i))
            {
                /* Execute this signal */
                struct sigaction *act = &pg_signal_array[i];
                pqsigfunc   sig = act->sa_handler;
 
                if (sig == SIG_DFL)
                    sig = pg_signal_defaults[i];
                pg_signal_queue &= ~sigmask(i);
                if (sig != SIG_ERR && sig != SIG_IGN && sig != SIG_DFL)
                {
                    sigset_t    block_mask;
                    sigset_t    save_mask;
 
                    LeaveCriticalSection(&pg_signal_crit_sec);
 
                    block_mask = act->sa_mask;
                    if ((act->sa_flags & SA_NODEFER) == 0)
                        block_mask |= sigmask(i);
 
                    sigprocmask(SIG_BLOCK, &block_mask, &save_mask);
                    sig(i);
                    sigprocmask(SIG_SETMASK, &save_mask, NULL);
 
                    EnterCriticalSection(&pg_signal_crit_sec);
                    break;      /* Restart outer loop, in case signal mask or
                                 * queue has been modified inside signal
                                 * handler */
                }
            }
        }
    }
    ResetEvent(pgwin32_signal_event);
    LeaveCriticalSection(&pg_signal_crit_sec);
}

References Assert(), i, pg_signal_array, PG_SIGNAL_COUNT, pg_signal_crit_sec, pg_signal_defaults, pg_signal_queue, pgwin32_signal_event, sig, sigmask, and UNBLOCKED_SIGNAL_QUEUE.

Referenced by pg_usleep(), PGSemaphoreLock(), pgwin32_poll_signals(), pgwin32_select(), pgwin32_waitforsinglesocket(), pqsigprocmask(), and WaitEventSetWait().

◆ pgwin32_signal_initialize()

void pgwin32_signal_initialize ( void )

Definition at line 79 of file signal.c.

{
    int         i;
    HANDLE      signal_thread_handle;
 
    InitializeCriticalSection(&pg_signal_crit_sec);
 
    for (i = 0; i < PG_SIGNAL_COUNT; i++)
    {
        pg_signal_array[i].sa_handler = SIG_DFL;
        pg_signal_array[i].sa_mask = 0;
        pg_signal_array[i].sa_flags = 0;
        pg_signal_defaults[i] = SIG_IGN;
    }
    pg_signal_mask = 0;
    pg_signal_queue = 0;
 
    /* Create the global event handle used to flag signals */
    pgwin32_signal_event = CreateEvent(NULL, TRUE, FALSE, NULL);
    if (pgwin32_signal_event == NULL)
        ereport(FATAL,
                (errmsg_internal("could not create signal event: error code %lu", GetLastError())));
 
    /* Create thread for handling signals */
    signal_thread_handle = CreateThread(NULL, 0, pg_signal_thread, NULL, 0, NULL);
    if (signal_thread_handle == NULL)
        ereport(FATAL,
                (errmsg_internal("could not create signal handler thread")));
 
    /* Create console control handle to pick up Ctrl-C etc */
    if (!SetConsoleCtrlHandler(pg_console_handler, TRUE))
        ereport(FATAL,
                (errmsg_internal("could not set console control handler")));
}

References ereport, errmsg_internal(), FATAL, i, pg_console_handler(), pg_signal_array, PG_SIGNAL_COUNT, pg_signal_crit_sec, pg_signal_defaults, pg_signal_mask, pg_signal_queue, pg_signal_thread(), and pgwin32_signal_event.

Referenced by InitPostmasterChild(), InitStandaloneProcess(), and PostmasterMain().

◆ pqsigaction()

int pqsigaction	(	int	signum,
		const struct sigaction *	act,
		struct sigaction *	oldact
	)

Definition at line 210 of file signal.c.

{
    if (signum >= PG_SIGNAL_COUNT || signum < 0)
    {
        errno = EINVAL;
        return -1;
    }
    if (oldact)
        *oldact = pg_signal_array[signum];
    if (act)
        pg_signal_array[signum] = *act;
    return 0;
}

References pg_signal_array, and PG_SIGNAL_COUNT.

◆ pqsigprocmask()

int pqsigprocmask	(	int	how,
		const sigset_t *	set,
		sigset_t *	oset
	)

Definition at line 171 of file signal.c.

{
    if (oset)
        *oset = pg_signal_mask;
 
    if (!set)
        return 0;
 
    switch (how)
    {
        case SIG_BLOCK:
            pg_signal_mask |= *set;
            break;
        case SIG_UNBLOCK:
            pg_signal_mask &= ~*set;
            break;
        case SIG_SETMASK:
            pg_signal_mask = *set;
            break;
        default:
            errno = EINVAL;
            return -1;
    }
 
    /*
     * Dispatch any signals queued up right away, in case we have unblocked
     * one or more signals previously queued
     */
    pgwin32_dispatch_queued_signals();
 
    return 0;
}