#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 (pgthreadlock_t threadlock_impl, libpq_gettext_func gettext_impl, conn_errorMessage_func errmsg_impl, conn_oauth_client_id_func clientid_impl, conn_oauth_client_secret_func clientsecret_impl, conn_oauth_discovery_uri_func discoveryuri_impl, conn_oauth_issuer_id_func issuerid_impl, conn_oauth_scope_func scope_impl, conn_sasl_state_func saslstate_impl, set_conn_altsock_func setaltsock_impl, set_conn_oauth_token_func settoken_impl)

void	libpq_append_conn_error (PGconn conn, const char fmt,...)

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

conn_errorMessage_func	conn_errorMessage

conn_oauth_client_id_func	conn_oauth_client_id

conn_oauth_client_secret_func	conn_oauth_client_secret

conn_oauth_discovery_uri_func	conn_oauth_discovery_uri

conn_oauth_issuer_id_func	conn_oauth_issuer_id

conn_oauth_scope_func	conn_oauth_scope

conn_sasl_state_func	conn_sasl_state

set_conn_altsock_func	set_conn_altsock

set_conn_oauth_token_func	set_conn_oauth_token

Macro Definition Documentation

◆ SOCK_ERRNO

#define SOCK_ERRNO errno

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

◆ SOCK_ERRNO_SET

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

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

Function Documentation

◆ libpq_append_conn_error()

void libpq_append_conn_error	(	PGconn *	conn,
		const char *	fmt,
			...
	)

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

{
    int         save_errno = errno;
    bool        done;
    va_list     args;
    PQExpBuffer errorMessage = conn_errorMessage(conn);
 
    Assert(fmt[strlen(fmt) - 1] != '\n');
 
    if (PQExpBufferBroken(errorMessage))
        return;                 /* already failed */
 
    /* Loop in case we have to retry after enlarging the buffer. */
    do
    {
        errno = save_errno;
        va_start(args, fmt);
        done = appendPQExpBufferVA(errorMessage, libpq_gettext(fmt), args);
        va_end(args);
    } while (!done);
 
    appendPQExpBufferChar(errorMessage, '\n');
}

References appendPQExpBufferChar(), appendPQExpBufferVA(), generate_unaccent_rules::args, Assert(), conn, conn_errorMessage, libpq_gettext, and PQExpBufferBroken.

Referenced by build_client_final_message(), build_client_first_message(), check_expected_areq(), client_initial_response(), connectFailureMessage(), connectNoDelay(), fillPGconn(), free_async_ctx(), getBackendKeyData(), handle_oauth_sasl_error(), handleSyncLoss(), init_allowed_encryption_methods(), initialize_curl(), initialize_SSL(), issuer_from_well_known_uri(), lo_create(), lo_export(), lo_import_internal(), lo_initialize(), lo_lseek64(), lo_read(), lo_tell64(), lo_truncate(), lo_truncate64(), lo_write(), oauth_exchange(), open_client_SSL(), openssl_verify_peer_name_matches_certificate_ip(), openssl_verify_peer_name_matches_certificate_name(), pg_fe_run_oauth_flow_impl(), pg_fe_sendauth(), pg_GSS_load_servicename(), pg_GSS_read(), pg_GSS_write(), pg_password_sendauth(), pg_SASL_continue(), pg_SASL_init(), pgpassfileWarning(), pgtls_get_peer_certificate_hash(), pgtls_read(), pgtls_write(), pq_verify_peer_name_matches_certificate(), pq_verify_peer_name_matches_certificate_ip(), pq_verify_peer_name_matches_certificate_name(), pqAllocCmdQueueEntry(), PQcancelCreate(), PQcancelPoll(), PQcancelStart(), pqConnectOptions2(), PQconnectPoll(), pqCopyPGconn(), PQencryptPasswordConn(), pqEndcopy3(), PQenterPipelineMode(), PQescapeByteaInternal(), PQescapeInternal(), PQescapeStringInternal(), PQexecStart(), PQexitPipelineMode(), PQfn(), pqFunctionCall3(), PQgetCopyData(), pqGetCopyData3(), pqGetErrorNotice3(), pqGetline3(), pqGetNegotiateProtocolVersion3(), PQgetResult(), pqParseInput3(), pqParseIntParam(), pqParseProtocolVersion(), pqPipelineProcessQueue(), pqPipelineSyncInternal(), pqPrepareAsyncResult(), PQputCopyData(), PQputCopyEnd(), pqSaveWriteError(), pqsecure_open_gss(), pqsecure_raw_read(), PQsendFlushRequest(), PQsendPrepare(), PQsendQueryGuts(), PQsendQueryInternal(), PQsendQueryParams(), PQsendQueryPrepared(), PQsendQueryStart(), PQsendTypedCommand(), PQsetdbLogin(), read_server_final_message(), read_server_first_message(), run_user_oauth_flow(), scram_exchange(), select_next_encryption_method(), setKeepalivesCount(), setKeepalivesIdle(), setKeepalivesInterval(), setTCPUserTimeout(), setup_oauth_parameters(), setup_token_request(), and store_conn_addrinfo().

◆ libpq_oauth_init()

void libpq_oauth_init	(	pgthreadlock_t	threadlock_impl,
		libpq_gettext_func	gettext_impl,
		conn_errorMessage_func	errmsg_impl,
		conn_oauth_client_id_func	clientid_impl,
		conn_oauth_client_secret_func	clientsecret_impl,
		conn_oauth_discovery_uri_func	discoveryuri_impl,
		conn_oauth_issuer_id_func	issuerid_impl,
		conn_oauth_scope_func	scope_impl,
		conn_sasl_state_func	saslstate_impl,
		set_conn_altsock_func	setaltsock_impl,
		set_conn_oauth_token_func	settoken_impl
	)

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

{
    pg_g_threadlock = threadlock_impl;
    libpq_gettext_impl = gettext_impl;
    conn_errorMessage = errmsg_impl;
    conn_oauth_client_id = clientid_impl;
    conn_oauth_client_secret = clientsecret_impl;
    conn_oauth_discovery_uri = discoveryuri_impl;
    conn_oauth_issuer_id = issuerid_impl;
    conn_oauth_scope = scope_impl;
    conn_sasl_state = saslstate_impl;
    set_conn_altsock = setaltsock_impl;
    set_conn_oauth_token = settoken_impl;
}

References conn_errorMessage, conn_oauth_client_id, conn_oauth_client_secret, conn_oauth_discovery_uri, conn_oauth_issuer_id, conn_oauth_scope, conn_sasl_state, libpq_gettext_impl, pg_g_threadlock, set_conn_altsock, and set_conn_oauth_token.

◆ oauth_unsafe_debugging_enabled()

bool oauth_unsafe_debugging_enabled ( void )

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

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

Referenced by pg_fe_run_oauth_flow_impl().

◆ pq_block_sigpipe()

int pq_block_sigpipe	(	sigset_t *	osigset,
		bool *	sigpipe_pending
	)

Definition at line 172 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 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 208 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);
}