#include "postgres_fe.h"
#include <signal.h>
#include "oauth-utils.h"

Include dependency graph for oauth-utils.c:

Macros
#define	SOCK_ERRNO errno

#define	SOCK_ERRNO_SET(e) (errno = (e))

Functions
void	libpq_oauth_init (libpq_gettext_func gettext_impl)

bool	oauth_unsafe_debugging_enabled (void)

int	pq_block_sigpipe (sigset_t osigset, bool sigpipe_pending)

void	pq_reset_sigpipe (sigset_t *osigset, bool sigpipe_pending, bool got_epipe)

Variables
pgthreadlock_t	pg_g_threadlock

static libpq_gettext_func	libpq_gettext_impl

Macro Definition Documentation

◆ SOCK_ERRNO

#define SOCK_ERRNO errno

Definition at line 97 of file oauth-utils.c.

◆ SOCK_ERRNO_SET

#define SOCK_ERRNO_SET ( e ) (errno = (e))

Definition at line 98 of file oauth-utils.c.

Function Documentation

◆ libpq_oauth_init()

void libpq_oauth_init ( libpq_gettext_func gettext_impl )

Definition at line 46 of file oauth-utils.c.

{
    pg_g_threadlock = PQgetThreadLock();
    libpq_gettext_impl = gettext_impl;
}

References fb(), libpq_gettext_impl, pg_g_threadlock, and PQgetThreadLock().

◆ oauth_unsafe_debugging_enabled()

bool oauth_unsafe_debugging_enabled ( void )

Definition at line 82 of file oauth-utils.c.

{
    const char *env = getenv("PGOAUTHDEBUG");
 
    return (env && strcmp(env, "UNSAFE") == 0);
}

References fb().

Referenced by pg_start_oauthbearer().

◆ pq_block_sigpipe()

int pq_block_sigpipe	(	sigset_t *	osigset,
		bool *	sigpipe_pending
	)

Definition at line 105 of file oauth-utils.c.

{
    sigset_t    sigpipe_sigset;
    sigset_t    sigset;
 
    sigemptyset(&sigpipe_sigset);
    sigaddset(&sigpipe_sigset, SIGPIPE);
 
    /* Block SIGPIPE and save previous mask for later reset */
    SOCK_ERRNO_SET(pthread_sigmask(SIG_BLOCK, &sigpipe_sigset, osigset));
    if (SOCK_ERRNO)
        return -1;
 
    /* We can have a pending SIGPIPE only if it was blocked before */
    if (sigismember(osigset, SIGPIPE))
    {
        /* Is there a pending SIGPIPE? */
        if (sigpending(&sigset) != 0)
            return -1;
 
        if (sigismember(&sigset, SIGPIPE))
            *sigpipe_pending = true;
        else
            *sigpipe_pending = false;
    }
    else
        *sigpipe_pending = false;
 
    return 0;
}

References fb(), SIGPIPE, SOCK_ERRNO, and SOCK_ERRNO_SET.

Referenced by pg_fe_run_oauth_flow(), and PQprint().

◆ pq_reset_sigpipe()

void pq_reset_sigpipe	(	sigset_t *	osigset,
		bool	sigpipe_pending,
		bool	got_epipe
	)

Definition at line 141 of file oauth-utils.c.

{
    int         save_errno = SOCK_ERRNO;
    int         signo;
    sigset_t    sigset;
 
    /* Clear SIGPIPE only if none was pending */
    if (got_epipe && !sigpipe_pending)
    {
        if (sigpending(&sigset) == 0 &&
            sigismember(&sigset, SIGPIPE))
        {
            sigset_t    sigpipe_sigset;
 
            sigemptyset(&sigpipe_sigset);
            sigaddset(&sigpipe_sigset, SIGPIPE);
 
            sigwait(&sigpipe_sigset, &signo);
        }
    }
 
    /* Restore saved block mask */
    pthread_sigmask(SIG_SETMASK, osigset, NULL);
 
    SOCK_ERRNO_SET(save_errno);
}