PostgreSQL Source Code  git master
libpq-fe.h File Reference
#include <stdio.h>
#include "postgres_ext.h"
Include dependency graph for libpq-fe.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Data Structures

struct  pgNotify
 
struct  _PQprintOpt
 
struct  _PQconninfoOption
 
struct  PQArgBlock
 
struct  pgresAttDesc
 

Macros

#define PG_COPYRES_ATTRS   0x01
 
#define PG_COPYRES_TUPLES   0x02 /* Implies PG_COPYRES_ATTRS */
 
#define PG_COPYRES_EVENTS   0x04
 
#define PG_COPYRES_NOTICEHOOKS   0x08
 
#define PQsetdb(M_PGHOST, M_PGPORT, M_PGOPT, M_PGTTY, M_DBNAME)   PQsetdbLogin(M_PGHOST, M_PGPORT, M_PGOPT, M_PGTTY, M_DBNAME, NULL, NULL)
 
#define PQfreeNotify(ptr)   PQfreemem(ptr)
 
#define PQnoPasswordSupplied   "fe_sendauth: no password supplied\n"
 

Typedefs

typedef struct pg_conn PGconn
 
typedef struct pg_result PGresult
 
typedef struct pg_cancel PGcancel
 
typedef struct pgNotify PGnotify
 
typedef void(* PQnoticeReceiver) (void *arg, const PGresult *res)
 
typedef void(* PQnoticeProcessor) (void *arg, const char *message)
 
typedef char pqbool
 
typedef struct _PQprintOpt PQprintOpt
 
typedef struct _PQconninfoOption PQconninfoOption
 
typedef struct pgresAttDesc PGresAttDesc
 
typedef void(* pgthreadlock_t) (int acquire)
 
typedef int(* PQsslKeyPassHook_OpenSSL_type) (char *buf, int size, PGconn *conn)
 

Enumerations

enum  ConnStatusType {
  CONNECTION_OK, CONNECTION_BAD, CONNECTION_STARTED, CONNECTION_MADE,
  CONNECTION_AWAITING_RESPONSE, CONNECTION_AUTH_OK, CONNECTION_SETENV, CONNECTION_SSL_STARTUP,
  CONNECTION_NEEDED, CONNECTION_CHECK_WRITABLE, CONNECTION_CONSUME, CONNECTION_GSS_STARTUP,
  CONNECTION_CHECK_TARGET
}
 
enum  PostgresPollingStatusType {
  PGRES_POLLING_FAILED = 0, PGRES_POLLING_READING, PGRES_POLLING_WRITING, PGRES_POLLING_OK,
  PGRES_POLLING_ACTIVE
}
 
enum  ExecStatusType {
  PGRES_EMPTY_QUERY = 0, PGRES_COMMAND_OK, PGRES_TUPLES_OK, PGRES_COPY_OUT,
  PGRES_COPY_IN, PGRES_BAD_RESPONSE, PGRES_NONFATAL_ERROR, PGRES_FATAL_ERROR,
  PGRES_COPY_BOTH, PGRES_SINGLE_TUPLE
}
 
enum  PGTransactionStatusType {
  PQTRANS_IDLE, PQTRANS_ACTIVE, PQTRANS_INTRANS, PQTRANS_INERROR,
  PQTRANS_UNKNOWN
}
 
enum  PGVerbosity { PQERRORS_TERSE, PQERRORS_DEFAULT, PQERRORS_VERBOSE, PQERRORS_SQLSTATE }
 
enum  PGContextVisibility { PQSHOW_CONTEXT_NEVER, PQSHOW_CONTEXT_ERRORS, PQSHOW_CONTEXT_ALWAYS }
 
enum  PGPing { PQPING_OK, PQPING_REJECT, PQPING_NO_RESPONSE, PQPING_NO_ATTEMPT }
 

Functions

PGconnPQconnectStart (const char *conninfo)
 
PGconnPQconnectStartParams (const char *const *keywords, const char *const *values, int expand_dbname)
 
PostgresPollingStatusType PQconnectPoll (PGconn *conn)
 
PGconnPQconnectdb (const char *conninfo)
 
PGconnPQconnectdbParams (const char *const *keywords, const char *const *values, int expand_dbname)
 
PGconnPQsetdbLogin (const char *pghost, const char *pgport, const char *pgoptions, const char *pgtty, const char *dbName, const char *login, const char *pwd)
 
void PQfinish (PGconn *conn)
 
PQconninfoOptionPQconndefaults (void)
 
PQconninfoOptionPQconninfoParse (const char *conninfo, char **errmsg)
 
PQconninfoOptionPQconninfo (PGconn *conn)
 
void PQconninfoFree (PQconninfoOption *connOptions)
 
int PQresetStart (PGconn *conn)
 
PostgresPollingStatusType PQresetPoll (PGconn *conn)
 
void PQreset (PGconn *conn)
 
PGcancelPQgetCancel (PGconn *conn)
 
void PQfreeCancel (PGcancel *cancel)
 
int PQcancel (PGcancel *cancel, char *errbuf, int errbufsize)
 
int PQrequestCancel (PGconn *conn)
 
char * PQdb (const PGconn *conn)
 
char * PQuser (const PGconn *conn)
 
char * PQpass (const PGconn *conn)
 
char * PQhost (const PGconn *conn)
 
char * PQhostaddr (const PGconn *conn)
 
char * PQport (const PGconn *conn)
 
char * PQtty (const PGconn *conn)
 
char * PQoptions (const PGconn *conn)
 
ConnStatusType PQstatus (const PGconn *conn)
 
PGTransactionStatusType PQtransactionStatus (const PGconn *conn)
 
const char * PQparameterStatus (const PGconn *conn, const char *paramName)
 
int PQprotocolVersion (const PGconn *conn)
 
int PQserverVersion (const PGconn *conn)
 
char * PQerrorMessage (const PGconn *conn)
 
int PQsocket (const PGconn *conn)
 
int PQbackendPID (const PGconn *conn)
 
int PQconnectionNeedsPassword (const PGconn *conn)
 
int PQconnectionUsedPassword (const PGconn *conn)
 
int PQclientEncoding (const PGconn *conn)
 
int PQsetClientEncoding (PGconn *conn, const char *encoding)
 
int PQsslInUse (PGconn *conn)
 
void * PQsslStruct (PGconn *conn, const char *struct_name)
 
const char * PQsslAttribute (PGconn *conn, const char *attribute_name)
 
const char *const * PQsslAttributeNames (PGconn *conn)
 
void * PQgetssl (PGconn *conn)
 
void PQinitSSL (int do_init)
 
void PQinitOpenSSL (int do_ssl, int do_crypto)
 
int PQgssEncInUse (PGconn *conn)
 
void * PQgetgssctx (PGconn *conn)
 
PGVerbosity PQsetErrorVerbosity (PGconn *conn, PGVerbosity verbosity)
 
PGContextVisibility PQsetErrorContextVisibility (PGconn *conn, PGContextVisibility show_context)
 
void PQtrace (PGconn *conn, FILE *debug_port)
 
void PQuntrace (PGconn *conn)
 
PQnoticeReceiver PQsetNoticeReceiver (PGconn *conn, PQnoticeReceiver proc, void *arg)
 
PQnoticeProcessor PQsetNoticeProcessor (PGconn *conn, PQnoticeProcessor proc, void *arg)
 
pgthreadlock_t PQregisterThreadLock (pgthreadlock_t newhandler)
 
PGresultPQexec (PGconn *conn, const char *query)
 
PGresultPQexecParams (PGconn *conn, const char *command, int nParams, const Oid *paramTypes, const char *const *paramValues, const int *paramLengths, const int *paramFormats, int resultFormat)
 
PGresultPQprepare (PGconn *conn, const char *stmtName, const char *query, int nParams, const Oid *paramTypes)
 
PGresultPQexecPrepared (PGconn *conn, const char *stmtName, int nParams, const char *const *paramValues, const int *paramLengths, const int *paramFormats, int resultFormat)
 
int PQsendQuery (PGconn *conn, const char *query)
 
int PQsendQueryParams (PGconn *conn, const char *command, int nParams, const Oid *paramTypes, const char *const *paramValues, const int *paramLengths, const int *paramFormats, int resultFormat)
 
int PQsendPrepare (PGconn *conn, const char *stmtName, const char *query, int nParams, const Oid *paramTypes)
 
int PQsendQueryPrepared (PGconn *conn, const char *stmtName, int nParams, const char *const *paramValues, const int *paramLengths, const int *paramFormats, int resultFormat)
 
int PQsetSingleRowMode (PGconn *conn)
 
PGresultPQgetResult (PGconn *conn)
 
int PQisBusy (PGconn *conn)
 
int PQconsumeInput (PGconn *conn)
 
PGnotifyPQnotifies (PGconn *conn)
 
int PQputCopyData (PGconn *conn, const char *buffer, int nbytes)
 
int PQputCopyEnd (PGconn *conn, const char *errormsg)
 
int PQgetCopyData (PGconn *conn, char **buffer, int async)
 
int PQgetline (PGconn *conn, char *string, int length)
 
int PQputline (PGconn *conn, const char *string)
 
int PQgetlineAsync (PGconn *conn, char *buffer, int bufsize)
 
int PQputnbytes (PGconn *conn, const char *buffer, int nbytes)
 
int PQendcopy (PGconn *conn)
 
int PQsetnonblocking (PGconn *conn, int arg)
 
int PQisnonblocking (const PGconn *conn)
 
int PQisthreadsafe (void)
 
PGPing PQping (const char *conninfo)
 
PGPing PQpingParams (const char *const *keywords, const char *const *values, int expand_dbname)
 
int PQflush (PGconn *conn)
 
PGresultPQfn (PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
 
ExecStatusType PQresultStatus (const PGresult *res)
 
char * PQresStatus (ExecStatusType status)
 
char * PQresultErrorMessage (const PGresult *res)
 
char * PQresultVerboseErrorMessage (const PGresult *res, PGVerbosity verbosity, PGContextVisibility show_context)
 
char * PQresultErrorField (const PGresult *res, int fieldcode)
 
int PQntuples (const PGresult *res)
 
int PQnfields (const PGresult *res)
 
int PQbinaryTuples (const PGresult *res)
 
char * PQfname (const PGresult *res, int field_num)
 
int PQfnumber (const PGresult *res, const char *field_name)
 
Oid PQftable (const PGresult *res, int field_num)
 
int PQftablecol (const PGresult *res, int field_num)
 
int PQfformat (const PGresult *res, int field_num)
 
Oid PQftype (const PGresult *res, int field_num)
 
int PQfsize (const PGresult *res, int field_num)
 
int PQfmod (const PGresult *res, int field_num)
 
char * PQcmdStatus (PGresult *res)
 
char * PQoidStatus (const PGresult *res)
 
Oid PQoidValue (const PGresult *res)
 
char * PQcmdTuples (PGresult *res)
 
char * PQgetvalue (const PGresult *res, int tup_num, int field_num)
 
int PQgetlength (const PGresult *res, int tup_num, int field_num)
 
int PQgetisnull (const PGresult *res, int tup_num, int field_num)
 
int PQnparams (const PGresult *res)
 
Oid PQparamtype (const PGresult *res, int param_num)
 
PGresultPQdescribePrepared (PGconn *conn, const char *stmt)
 
PGresultPQdescribePortal (PGconn *conn, const char *portal)
 
int PQsendDescribePrepared (PGconn *conn, const char *stmt)
 
int PQsendDescribePortal (PGconn *conn, const char *portal)
 
void PQclear (PGresult *res)
 
void PQfreemem (void *ptr)
 
PGresultPQmakeEmptyPGresult (PGconn *conn, ExecStatusType status)
 
PGresultPQcopyResult (const PGresult *src, int flags)
 
int PQsetResultAttrs (PGresult *res, int numAttributes, PGresAttDesc *attDescs)
 
void * PQresultAlloc (PGresult *res, size_t nBytes)
 
size_t PQresultMemorySize (const PGresult *res)
 
int PQsetvalue (PGresult *res, int tup_num, int field_num, char *value, int len)
 
size_t PQescapeStringConn (PGconn *conn, char *to, const char *from, size_t length, int *error)
 
char * PQescapeLiteral (PGconn *conn, const char *str, size_t len)
 
char * PQescapeIdentifier (PGconn *conn, const char *str, size_t len)
 
unsigned char * PQescapeByteaConn (PGconn *conn, const unsigned char *from, size_t from_length, size_t *to_length)
 
unsigned char * PQunescapeBytea (const unsigned char *strtext, size_t *retbuflen)
 
size_t PQescapeString (char *to, const char *from, size_t length)
 
unsigned char * PQescapeBytea (const unsigned char *from, size_t from_length, size_t *to_length)
 
void PQprint (FILE *fout, const PGresult *res, const PQprintOpt *ps)
 
void PQdisplayTuples (const PGresult *res, FILE *fp, int fillAlign, const char *fieldSep, int printHeader, int quiet)
 
void PQprintTuples (const PGresult *res, FILE *fout, int PrintAttNames, int TerseOutput, int colWidth)
 
int lo_open (PGconn *conn, Oid lobjId, int mode)
 
int lo_close (PGconn *conn, int fd)
 
int lo_read (PGconn *conn, int fd, char *buf, size_t len)
 
int lo_write (PGconn *conn, int fd, const char *buf, size_t len)
 
int lo_lseek (PGconn *conn, int fd, int offset, int whence)
 
pg_int64 lo_lseek64 (PGconn *conn, int fd, pg_int64 offset, int whence)
 
Oid lo_creat (PGconn *conn, int mode)
 
Oid lo_create (PGconn *conn, Oid lobjId)
 
int lo_tell (PGconn *conn, int fd)
 
pg_int64 lo_tell64 (PGconn *conn, int fd)
 
int lo_truncate (PGconn *conn, int fd, size_t len)
 
int lo_truncate64 (PGconn *conn, int fd, pg_int64 len)
 
int lo_unlink (PGconn *conn, Oid lobjId)
 
Oid lo_import (PGconn *conn, const char *filename)
 
Oid lo_import_with_oid (PGconn *conn, const char *filename, Oid lobjId)
 
int lo_export (PGconn *conn, Oid lobjId, const char *filename)
 
int PQlibVersion (void)
 
int PQmblen (const char *s, int encoding)
 
int PQdsplen (const char *s, int encoding)
 
int PQenv2encoding (void)
 
char * PQencryptPassword (const char *passwd, const char *user)
 
char * PQencryptPasswordConn (PGconn *conn, const char *passwd, const char *user, const char *algorithm)
 
int pg_char_to_encoding (const char *name)
 
const char * pg_encoding_to_char (int encoding)
 
int pg_valid_server_encoding_id (int encoding)
 
PQsslKeyPassHook_OpenSSL_type PQgetSSLKeyPassHook_OpenSSL (void)
 
void PQsetSSLKeyPassHook_OpenSSL (PQsslKeyPassHook_OpenSSL_type hook)
 
int PQdefaultSSLKeyPassHook_OpenSSL (char *buf, int size, PGconn *conn)
 

Macro Definition Documentation

◆ PG_COPYRES_ATTRS

#define PG_COPYRES_ATTRS   0x01

Definition at line 34 of file libpq-fe.h.

Referenced by PQcopyResult(), and pqRowProcessor().

◆ PG_COPYRES_EVENTS

#define PG_COPYRES_EVENTS   0x04

Definition at line 36 of file libpq-fe.h.

Referenced by PQcopyResult(), and pqRowProcessor().

◆ PG_COPYRES_NOTICEHOOKS

#define PG_COPYRES_NOTICEHOOKS   0x08

Definition at line 37 of file libpq-fe.h.

Referenced by PQcopyResult(), and pqRowProcessor().

◆ PG_COPYRES_TUPLES

#define PG_COPYRES_TUPLES   0x02 /* Implies PG_COPYRES_ATTRS */

Definition at line 35 of file libpq-fe.h.

Referenced by PQcopyResult().

◆ PQfreeNotify

#define PQfreeNotify (   ptr)    PQfreemem(ptr)

Definition at line 518 of file libpq-fe.h.

◆ PQnoPasswordSupplied

#define PQnoPasswordSupplied   "fe_sendauth: no password supplied\n"

Definition at line 522 of file libpq-fe.h.

Referenced by pg_fe_sendauth(), and pg_SASL_init().

◆ PQsetdb

#define PQsetdb (   M_PGHOST,
  M_PGPORT,
  M_PGOPT,
  M_PGTTY,
  M_DBNAME 
)    PQsetdbLogin(M_PGHOST, M_PGPORT, M_PGOPT, M_PGTTY, M_DBNAME, NULL, NULL)

Definition at line 272 of file libpq-fe.h.

Referenced by main().

Typedef Documentation

◆ PGcancel

typedef struct pg_cancel PGcancel

Definition at line 156 of file libpq-fe.h.

◆ PGconn

typedef struct pg_conn PGconn

Definition at line 143 of file libpq-fe.h.

◆ PGnotify

typedef struct pgNotify PGnotify

◆ PGresAttDesc

typedef struct pgresAttDesc PGresAttDesc

◆ PGresult

typedef struct pg_result PGresult

Definition at line 150 of file libpq-fe.h.

◆ pgthreadlock_t

typedef void(* pgthreadlock_t) (int acquire)

Definition at line 384 of file libpq-fe.h.

◆ pqbool

typedef char pqbool

Definition at line 178 of file libpq-fe.h.

◆ PQconninfoOption

◆ PQnoticeProcessor

typedef void(* PQnoticeProcessor) (void *arg, const char *message)

Definition at line 175 of file libpq-fe.h.

◆ PQnoticeReceiver

typedef void(* PQnoticeReceiver) (void *arg, const PGresult *res)

Definition at line 174 of file libpq-fe.h.

◆ PQprintOpt

typedef struct _PQprintOpt PQprintOpt

◆ PQsslKeyPassHook_OpenSSL_type

typedef int(* PQsslKeyPassHook_OpenSSL_type) (char *buf, int size, PGconn *conn)

Definition at line 623 of file libpq-fe.h.

Enumeration Type Documentation

◆ ConnStatusType

Enumerator
CONNECTION_OK 
CONNECTION_BAD 
CONNECTION_STARTED 
CONNECTION_MADE 
CONNECTION_AWAITING_RESPONSE 
CONNECTION_AUTH_OK 
CONNECTION_SETENV 
CONNECTION_SSL_STARTUP 
CONNECTION_NEEDED 
CONNECTION_CHECK_WRITABLE 
CONNECTION_CONSUME 
CONNECTION_GSS_STARTUP 
CONNECTION_CHECK_TARGET 

Definition at line 47 of file libpq-fe.h.

48 {
51  /* Non-blocking mode only below here */
52 
53  /*
54  * The existence of these should never be relied upon - they should only
55  * be used for user feedback or similar purposes.
56  */
57  CONNECTION_STARTED, /* Waiting for connection to be made. */
58  CONNECTION_MADE, /* Connection OK; waiting to send. */
59  CONNECTION_AWAITING_RESPONSE, /* Waiting for a response from the
60  * postmaster. */
61  CONNECTION_AUTH_OK, /* Received authentication; waiting for
62  * backend startup. */
63  CONNECTION_SETENV, /* Negotiating environment. */
64  CONNECTION_SSL_STARTUP, /* Negotiating SSL. */
65  CONNECTION_NEEDED, /* Internal state: connect() needed */
66  CONNECTION_CHECK_WRITABLE, /* Check if we could make a writable
67  * connection. */
68  CONNECTION_CONSUME, /* Wait for any pending message and consume
69  * them. */
70  CONNECTION_GSS_STARTUP, /* Negotiating GSSAPI. */
71  CONNECTION_CHECK_TARGET /* Check if we have a proper target connection */
ConnStatusType
Definition: libpq-fe.h:47

◆ ExecStatusType

Enumerator
PGRES_EMPTY_QUERY 
PGRES_COMMAND_OK 
PGRES_TUPLES_OK 
PGRES_COPY_OUT 
PGRES_COPY_IN 
PGRES_BAD_RESPONSE 
PGRES_NONFATAL_ERROR 
PGRES_FATAL_ERROR 
PGRES_COPY_BOTH 
PGRES_SINGLE_TUPLE 

Definition at line 84 of file libpq-fe.h.

85 {
86  PGRES_EMPTY_QUERY = 0, /* empty query string was executed */
87  PGRES_COMMAND_OK, /* a query command that doesn't return
88  * anything was executed properly by the
89  * backend */
90  PGRES_TUPLES_OK, /* a query command that returns tuples was
91  * executed properly by the backend, PGresult
92  * contains the result tuples */
93  PGRES_COPY_OUT, /* Copy Out data transfer in progress */
94  PGRES_COPY_IN, /* Copy In data transfer in progress */
95  PGRES_BAD_RESPONSE, /* an unexpected response was recv'd from the
96  * backend */
97  PGRES_NONFATAL_ERROR, /* notice or warning message */
98  PGRES_FATAL_ERROR, /* query failed */
99  PGRES_COPY_BOTH, /* Copy In/Out data transfer in progress */
100  PGRES_SINGLE_TUPLE /* single tuple from larger resultset */
ExecStatusType
Definition: libpq-fe.h:84

◆ PGContextVisibility

Enumerator
PQSHOW_CONTEXT_NEVER 
PQSHOW_CONTEXT_ERRORS 
PQSHOW_CONTEXT_ALWAYS 

Definition at line 120 of file libpq-fe.h.

121 {
122  PQSHOW_CONTEXT_NEVER, /* never show CONTEXT field */
123  PQSHOW_CONTEXT_ERRORS, /* show CONTEXT for errors only (default) */
124  PQSHOW_CONTEXT_ALWAYS /* always show CONTEXT field */
PGContextVisibility
Definition: libpq-fe.h:120

◆ PGPing

enum PGPing
Enumerator
PQPING_OK 
PQPING_REJECT 
PQPING_NO_RESPONSE 
PQPING_NO_ATTEMPT 

Definition at line 132 of file libpq-fe.h.

133 {
134  PQPING_OK, /* server is accepting connections */
135  PQPING_REJECT, /* server is alive but rejecting connections */
136  PQPING_NO_RESPONSE, /* could not establish connection */
137  PQPING_NO_ATTEMPT /* connection not attempted (bad params) */
138 } PGPing;
PGPing
Definition: libpq-fe.h:132

◆ PGTransactionStatusType

Enumerator
PQTRANS_IDLE 
PQTRANS_ACTIVE 
PQTRANS_INTRANS 
PQTRANS_INERROR 
PQTRANS_UNKNOWN 

Definition at line 103 of file libpq-fe.h.

104 {
105  PQTRANS_IDLE, /* connection idle */
106  PQTRANS_ACTIVE, /* command in progress */
107  PQTRANS_INTRANS, /* idle, within transaction block */
108  PQTRANS_INERROR, /* idle, within failed transaction */
109  PQTRANS_UNKNOWN /* cannot determine status */
PGTransactionStatusType
Definition: libpq-fe.h:103

◆ PGVerbosity

Enumerator
PQERRORS_TERSE 
PQERRORS_DEFAULT 
PQERRORS_VERBOSE 
PQERRORS_SQLSTATE 

Definition at line 112 of file libpq-fe.h.

113 {
114  PQERRORS_TERSE, /* single-line error messages */
115  PQERRORS_DEFAULT, /* recommended style */
116  PQERRORS_VERBOSE, /* all the facts, ma'am */
117  PQERRORS_SQLSTATE /* only error severity and SQLSTATE code */
118 } PGVerbosity;
PGVerbosity
Definition: libpq-fe.h:112

◆ PostgresPollingStatusType

Enumerator
PGRES_POLLING_FAILED 
PGRES_POLLING_READING 
PGRES_POLLING_WRITING 
PGRES_POLLING_OK 
PGRES_POLLING_ACTIVE 

Definition at line 74 of file libpq-fe.h.

75 {
77  PGRES_POLLING_READING, /* These two indicate that one may */
78  PGRES_POLLING_WRITING, /* use select before polling again. */
80  PGRES_POLLING_ACTIVE /* unused; keep for awhile for backwards
81  * compatibility */
PostgresPollingStatusType
Definition: libpq-fe.h:74

Function Documentation

◆ lo_close()

int lo_close ( PGconn conn,
int  fd 
)

Definition at line 99 of file fe-lobj.c.

References fd(), pgLobjfuncs::fn_lo_close, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, lo_initialize(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), and PQArgBlock::u.

Referenced by dumpBlobs(), EndRestoreBlob(), exportFile(), importFile(), lo_export(), lo_import_internal(), my_truncate(), overwrite(), and pickout().

100 {
101  PQArgBlock argv[1];
102  PGresult *res;
103  int retval;
104  int result_len;
105 
106  if (conn == NULL || conn->lobjfuncs == NULL)
107  {
108  if (lo_initialize(conn) < 0)
109  return -1;
110  }
111 
112  argv[0].isint = 1;
113  argv[0].len = 4;
114  argv[0].u.integer = fd;
115  res = PQfn(conn, conn->lobjfuncs->fn_lo_close,
116  &retval, &result_len, 1, argv, 1);
117  if (PQresultStatus(res) == PGRES_COMMAND_OK)
118  {
119  PQclear(res);
120  return retval;
121  }
122  else
123  {
124  PQclear(res);
125  return -1;
126  }
127 }
union PQArgBlock::@161 u
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
void PQclear(PGresult *res)
Definition: fe-exec.c:694
Oid fn_lo_close
Definition: libpq-int.h:271
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_creat()

Oid lo_creat ( PGconn conn,
int  mode 
)

Definition at line 465 of file fe-lobj.c.

References pgLobjfuncs::fn_lo_creat, PQArgBlock::integer, InvalidOid, PQArgBlock::isint, PQArgBlock::len, lo_initialize(), pg_conn::lobjfuncs, mode, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), and PQArgBlock::u.

Referenced by importFile(), and lo_import_internal().

466 {
467  PQArgBlock argv[1];
468  PGresult *res;
469  int retval;
470  int result_len;
471 
472  if (conn == NULL || conn->lobjfuncs == NULL)
473  {
474  if (lo_initialize(conn) < 0)
475  return InvalidOid;
476  }
477 
478  argv[0].isint = 1;
479  argv[0].len = 4;
480  argv[0].u.integer = mode;
481  res = PQfn(conn, conn->lobjfuncs->fn_lo_creat,
482  &retval, &result_len, 1, argv, 1);
483  if (PQresultStatus(res) == PGRES_COMMAND_OK)
484  {
485  PQclear(res);
486  return (Oid) retval;
487  }
488  else
489  {
490  PQclear(res);
491  return InvalidOid;
492  }
493 }
static PgChecksumMode mode
Definition: pg_checksums.c:61
union PQArgBlock::@161 u
unsigned int Oid
Definition: postgres_ext.h:31
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
#define InvalidOid
Definition: postgres_ext.h:36
void PQclear(PGresult *res)
Definition: fe-exec.c:694
Oid fn_lo_creat
Definition: libpq-int.h:272
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_create()

Oid lo_create ( PGconn conn,
Oid  lobjId 
)

Definition at line 504 of file fe-lobj.c.

References pg_conn::errorMessage, pgLobjfuncs::fn_lo_create, PQArgBlock::integer, InvalidOid, PQArgBlock::isint, PQArgBlock::len, libpq_gettext, lo_initialize(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), printfPQExpBuffer(), and PQArgBlock::u.

Referenced by lo_import_internal(), and StartRestoreBlob().

505 {
506  PQArgBlock argv[1];
507  PGresult *res;
508  int retval;
509  int result_len;
510 
511  if (conn == NULL || conn->lobjfuncs == NULL)
512  {
513  if (lo_initialize(conn) < 0)
514  return InvalidOid;
515  }
516 
517  /* Must check this on-the-fly because it's not there pre-8.1 */
518  if (conn->lobjfuncs->fn_lo_create == 0)
519  {
521  libpq_gettext("cannot determine OID of function lo_create\n"));
522  return InvalidOid;
523  }
524 
525  argv[0].isint = 1;
526  argv[0].len = 4;
527  argv[0].u.integer = lobjId;
528  res = PQfn(conn, conn->lobjfuncs->fn_lo_create,
529  &retval, &result_len, 1, argv, 1);
530  if (PQresultStatus(res) == PGRES_COMMAND_OK)
531  {
532  PQclear(res);
533  return (Oid) retval;
534  }
535  else
536  {
537  PQclear(res);
538  return InvalidOid;
539  }
540 }
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
union PQArgBlock::@161 u
Oid fn_lo_create
Definition: libpq-int.h:273
unsigned int Oid
Definition: postgres_ext.h:31
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
PQExpBufferData errorMessage
Definition: libpq-int.h:526
#define InvalidOid
Definition: postgres_ext.h:36
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
#define libpq_gettext(x)
Definition: libpq-int.h:805
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_export()

int lo_export ( PGconn conn,
Oid  lobjId,
const char *  filename 
)

Definition at line 784 of file fe-lobj.c.

References buf, close, pg_conn::errorMessage, fd(), INV_READ, libpq_gettext, LO_BUFSIZE, lo_close(), lo_open(), lo_read(), PG_BINARY, PG_STRERROR_R_BUFLEN, printfPQExpBuffer(), strerror_r, and write.

Referenced by do_lo_export(), and main().

785 {
786  int result = 1;
787  int fd;
788  int nbytes,
789  tmp;
790  char buf[LO_BUFSIZE];
791  int lobj;
792  char sebuf[PG_STRERROR_R_BUFLEN];
793 
794  /*
795  * open the large object.
796  */
797  lobj = lo_open(conn, lobjId, INV_READ);
798  if (lobj == -1)
799  {
800  /* we assume lo_open() already set a suitable error message */
801  return -1;
802  }
803 
804  /*
805  * create the file to be written to
806  */
807  fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC | PG_BINARY, 0666);
808  if (fd < 0)
809  {
810  /* We must do lo_close before setting the errorMessage */
811  int save_errno = errno;
812 
813  (void) lo_close(conn, lobj);
815  libpq_gettext("could not open file \"%s\": %s\n"),
816  filename,
817  strerror_r(save_errno, sebuf, sizeof(sebuf)));
818  return -1;
819  }
820 
821  /*
822  * read in from the large object and write to the file
823  */
824  while ((nbytes = lo_read(conn, lobj, buf, LO_BUFSIZE)) > 0)
825  {
826  tmp = write(fd, buf, nbytes);
827  if (tmp != nbytes)
828  {
829  /* We must do lo_close before setting the errorMessage */
830  int save_errno = errno;
831 
832  (void) lo_close(conn, lobj);
833  (void) close(fd);
835  libpq_gettext("could not write to file \"%s\": %s\n"),
836  filename,
837  strerror_r(save_errno, sebuf, sizeof(sebuf)));
838  return -1;
839  }
840  }
841 
842  /*
843  * If lo_read() failed, we are now in an aborted transaction so there's no
844  * need for lo_close(); furthermore, if we tried it we'd overwrite the
845  * useful error result with a useless one. So skip lo_close() if we got a
846  * failure result.
847  */
848  if (nbytes < 0 ||
849  lo_close(conn, lobj) != 0)
850  {
851  /* assume lo_read() or lo_close() left a suitable error message */
852  result = -1;
853  }
854 
855  /* if we already failed, don't overwrite that msg with a close error */
856  if (close(fd) != 0 && result >= 0)
857  {
859  libpq_gettext("could not write to file \"%s\": %s\n"),
860  filename, strerror_r(errno, sebuf, sizeof(sebuf)));
861  result = -1;
862  }
863 
864  return result;
865 }
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
#define write(a, b, c)
Definition: win32.h:14
#define PG_STRERROR_R_BUFLEN
Definition: port.h:211
static int fd(const char *x, int i)
Definition: preproc-init.c:105
#define PG_BINARY
Definition: c.h:1234
#define INV_READ
Definition: libpq-fs.h:22
int lo_close(PGconn *conn, int fd)
Definition: fe-lobj.c:99
static char * buf
Definition: pg_test_fsync.c:67
int lo_open(PGconn *conn, Oid lobjId, int mode)
Definition: fe-lobj.c:57
PQExpBufferData errorMessage
Definition: libpq-int.h:526
#define LO_BUFSIZE
Definition: fe-lobj.c:42
#define strerror_r
Definition: port.h:210
static char * filename
Definition: pg_dumpall.c:90
int lo_read(PGconn *conn, int fd, char *buf, size_t len)
Definition: fe-lobj.c:258
#define close(a)
Definition: win32.h:12
#define libpq_gettext(x)
Definition: libpq-int.h:805

◆ lo_import()

Oid lo_import ( PGconn conn,
const char *  filename 
)

Definition at line 668 of file fe-lobj.c.

References InvalidOid, and lo_import_internal().

Referenced by do_lo_import(), and main().

669 {
670  return lo_import_internal(conn, filename, InvalidOid);
671 }
static Oid lo_import_internal(PGconn *conn, const char *filename, Oid oid)
Definition: fe-lobj.c:689
#define InvalidOid
Definition: postgres_ext.h:36
static char * filename
Definition: pg_dumpall.c:90

◆ lo_import_with_oid()

Oid lo_import_with_oid ( PGconn conn,
const char *  filename,
Oid  lobjId 
)

Definition at line 683 of file fe-lobj.c.

References lo_import_internal().

684 {
685  return lo_import_internal(conn, filename, lobjId);
686 }
static Oid lo_import_internal(PGconn *conn, const char *filename, Oid oid)
Definition: fe-lobj.c:689
static char * filename
Definition: pg_dumpall.c:90

◆ lo_lseek()

int lo_lseek ( PGconn conn,
int  fd,
int  offset,
int  whence 
)

Definition at line 365 of file fe-lobj.c.

References fd(), pgLobjfuncs::fn_lo_lseek, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, lo_initialize(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), and PQArgBlock::u.

Referenced by overwrite(), and pickout().

366 {
367  PQArgBlock argv[3];
368  PGresult *res;
369  int retval;
370  int result_len;
371 
372  if (conn == NULL || conn->lobjfuncs == NULL)
373  {
374  if (lo_initialize(conn) < 0)
375  return -1;
376  }
377 
378  argv[0].isint = 1;
379  argv[0].len = 4;
380  argv[0].u.integer = fd;
381 
382  argv[1].isint = 1;
383  argv[1].len = 4;
384  argv[1].u.integer = offset;
385 
386  argv[2].isint = 1;
387  argv[2].len = 4;
388  argv[2].u.integer = whence;
389 
390  res = PQfn(conn, conn->lobjfuncs->fn_lo_lseek,
391  &retval, &result_len, 1, argv, 3);
392  if (PQresultStatus(res) == PGRES_COMMAND_OK)
393  {
394  PQclear(res);
395  return retval;
396  }
397  else
398  {
399  PQclear(res);
400  return -1;
401  }
402 }
union PQArgBlock::@161 u
Oid fn_lo_lseek
Definition: libpq-int.h:275
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_lseek64()

pg_int64 lo_lseek64 ( PGconn conn,
int  fd,
pg_int64  offset,
int  whence 
)

Definition at line 409 of file fe-lobj.c.

References pg_conn::errorMessage, fd(), pgLobjfuncs::fn_lo_lseek64, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, libpq_gettext, lo_hton64(), lo_initialize(), lo_ntoh64(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), printfPQExpBuffer(), PQArgBlock::ptr, and PQArgBlock::u.

Referenced by overwrite(), and pickout().

410 {
411  PQArgBlock argv[3];
412  PGresult *res;
413  pg_int64 retval;
414  int result_len;
415 
416  if (conn == NULL || conn->lobjfuncs == NULL)
417  {
418  if (lo_initialize(conn) < 0)
419  return -1;
420  }
421 
422  if (conn->lobjfuncs->fn_lo_lseek64 == 0)
423  {
425  libpq_gettext("cannot determine OID of function lo_lseek64\n"));
426  return -1;
427  }
428 
429  argv[0].isint = 1;
430  argv[0].len = 4;
431  argv[0].u.integer = fd;
432 
433  offset = lo_hton64(offset);
434  argv[1].isint = 0;
435  argv[1].len = 8;
436  argv[1].u.ptr = (int *) &offset;
437 
438  argv[2].isint = 1;
439  argv[2].len = 4;
440  argv[2].u.integer = whence;
441 
442  res = PQfn(conn, conn->lobjfuncs->fn_lo_lseek64,
443  (void *) &retval, &result_len, 0, argv, 3);
444  if (PQresultStatus(res) == PGRES_COMMAND_OK && result_len == 8)
445  {
446  PQclear(res);
447  return lo_ntoh64(retval);
448  }
449  else
450  {
451  PQclear(res);
452  return -1;
453  }
454 }
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
union PQArgBlock::@161 u
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
PG_INT64_TYPE pg_int64
Definition: postgres_ext.h:47
static pg_int64 lo_hton64(pg_int64 host64)
Definition: fe-lobj.c:1061
PQExpBufferData errorMessage
Definition: libpq-int.h:526
void PQclear(PGresult *res)
Definition: fe-exec.c:694
Oid fn_lo_lseek64
Definition: libpq-int.h:276
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
int * ptr
Definition: libpq-fe.h:229
#define libpq_gettext(x)
Definition: libpq-int.h:805
int integer
Definition: libpq-fe.h:230
static pg_int64 lo_ntoh64(pg_int64 net64)
Definition: fe-lobj.c:1086
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_open()

int lo_open ( PGconn conn,
Oid  lobjId,
int  mode 
)

Definition at line 57 of file fe-lobj.c.

References fd(), pgLobjfuncs::fn_lo_open, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, lo_initialize(), pg_conn::lobjfuncs, mode, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), and PQArgBlock::u.

Referenced by dumpBlobs(), exportFile(), importFile(), lo_export(), lo_import_internal(), my_truncate(), overwrite(), pickout(), and StartRestoreBlob().

58 {
59  int fd;
60  int result_len;
61  PQArgBlock argv[2];
62  PGresult *res;
63 
64  if (conn == NULL || conn->lobjfuncs == NULL)
65  {
66  if (lo_initialize(conn) < 0)
67  return -1;
68  }
69 
70  argv[0].isint = 1;
71  argv[0].len = 4;
72  argv[0].u.integer = lobjId;
73 
74  argv[1].isint = 1;
75  argv[1].len = 4;
76  argv[1].u.integer = mode;
77 
78  res = PQfn(conn, conn->lobjfuncs->fn_lo_open, &fd, &result_len, 1, argv, 2);
79  if (PQresultStatus(res) == PGRES_COMMAND_OK)
80  {
81  PQclear(res);
82  return fd;
83  }
84  else
85  {
86  PQclear(res);
87  return -1;
88  }
89 }
static PgChecksumMode mode
Definition: pg_checksums.c:61
union PQArgBlock::@161 u
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
Oid fn_lo_open
Definition: libpq-int.h:270
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_read()

int lo_read ( PGconn conn,
int  fd,
char *  buf,
size_t  len 
)

Definition at line 258 of file fe-lobj.c.

References pg_conn::errorMessage, fd(), pgLobjfuncs::fn_lo_read, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, libpq_gettext, lo_initialize(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), printfPQExpBuffer(), and PQArgBlock::u.

Referenced by lo_export().

259 {
260  PQArgBlock argv[2];
261  PGresult *res;
262  int result_len;
263 
264  if (conn == NULL || conn->lobjfuncs == NULL)
265  {
266  if (lo_initialize(conn) < 0)
267  return -1;
268  }
269 
270  /*
271  * Long ago, somebody thought it'd be a good idea to declare this function
272  * as taking size_t ... but the underlying backend function only accepts a
273  * signed int32 length. So throw error if the given value overflows
274  * int32.
275  */
276  if (len > (size_t) INT_MAX)
277  {
279  libpq_gettext("argument of lo_read exceeds integer range\n"));
280  return -1;
281  }
282 
283  argv[0].isint = 1;
284  argv[0].len = 4;
285  argv[0].u.integer = fd;
286 
287  argv[1].isint = 1;
288  argv[1].len = 4;
289  argv[1].u.integer = (int) len;
290 
291  res = PQfn(conn, conn->lobjfuncs->fn_lo_read,
292  (void *) buf, &result_len, 0, argv, 2);
293  if (PQresultStatus(res) == PGRES_COMMAND_OK)
294  {
295  PQclear(res);
296  return result_len;
297  }
298  else
299  {
300  PQclear(res);
301  return -1;
302  }
303 }
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
union PQArgBlock::@161 u
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
Oid fn_lo_read
Definition: libpq-int.h:281
static char * buf
Definition: pg_test_fsync.c:67
PQExpBufferData errorMessage
Definition: libpq-int.h:526
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
#define libpq_gettext(x)
Definition: libpq-int.h:805
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_tell()

int lo_tell ( PGconn conn,
int  fd 
)

Definition at line 548 of file fe-lobj.c.

References fd(), pgLobjfuncs::fn_lo_tell, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, lo_initialize(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), and PQArgBlock::u.

549 {
550  int retval;
551  PQArgBlock argv[1];
552  PGresult *res;
553  int result_len;
554 
555  if (conn == NULL || conn->lobjfuncs == NULL)
556  {
557  if (lo_initialize(conn) < 0)
558  return -1;
559  }
560 
561  argv[0].isint = 1;
562  argv[0].len = 4;
563  argv[0].u.integer = fd;
564 
565  res = PQfn(conn, conn->lobjfuncs->fn_lo_tell,
566  &retval, &result_len, 1, argv, 1);
567  if (PQresultStatus(res) == PGRES_COMMAND_OK)
568  {
569  PQclear(res);
570  return retval;
571  }
572  else
573  {
574  PQclear(res);
575  return -1;
576  }
577 }
union PQArgBlock::@161 u
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
Oid fn_lo_tell
Definition: libpq-int.h:277
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_tell64()

pg_int64 lo_tell64 ( PGconn conn,
int  fd 
)

Definition at line 584 of file fe-lobj.c.

References pg_conn::errorMessage, fd(), pgLobjfuncs::fn_lo_tell64, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, libpq_gettext, lo_initialize(), lo_ntoh64(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), printfPQExpBuffer(), and PQArgBlock::u.

Referenced by pickout().

585 {
586  pg_int64 retval;
587  PQArgBlock argv[1];
588  PGresult *res;
589  int result_len;
590 
591  if (conn == NULL || conn->lobjfuncs == NULL)
592  {
593  if (lo_initialize(conn) < 0)
594  return -1;
595  }
596 
597  if (conn->lobjfuncs->fn_lo_tell64 == 0)
598  {
600  libpq_gettext("cannot determine OID of function lo_tell64\n"));
601  return -1;
602  }
603 
604  argv[0].isint = 1;
605  argv[0].len = 4;
606  argv[0].u.integer = fd;
607 
608  res = PQfn(conn, conn->lobjfuncs->fn_lo_tell64,
609  (void *) &retval, &result_len, 0, argv, 1);
610  if (PQresultStatus(res) == PGRES_COMMAND_OK && result_len == 8)
611  {
612  PQclear(res);
613  return lo_ntoh64(retval);
614  }
615  else
616  {
617  PQclear(res);
618  return -1;
619  }
620 }
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
union PQArgBlock::@161 u
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
PG_INT64_TYPE pg_int64
Definition: postgres_ext.h:47
Oid fn_lo_tell64
Definition: libpq-int.h:278
PQExpBufferData errorMessage
Definition: libpq-int.h:526
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
#define libpq_gettext(x)
Definition: libpq-int.h:805
int integer
Definition: libpq-fe.h:230
static pg_int64 lo_ntoh64(pg_int64 net64)
Definition: fe-lobj.c:1086
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_truncate()

int lo_truncate ( PGconn conn,
int  fd,
size_t  len 
)

Definition at line 137 of file fe-lobj.c.

References pg_conn::errorMessage, fd(), pgLobjfuncs::fn_lo_truncate, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, libpq_gettext, lo_initialize(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), printfPQExpBuffer(), and PQArgBlock::u.

138 {
139  PQArgBlock argv[2];
140  PGresult *res;
141  int retval;
142  int result_len;
143 
144  if (conn == NULL || conn->lobjfuncs == NULL)
145  {
146  if (lo_initialize(conn) < 0)
147  return -1;
148  }
149 
150  /* Must check this on-the-fly because it's not there pre-8.3 */
151  if (conn->lobjfuncs->fn_lo_truncate == 0)
152  {
154  libpq_gettext("cannot determine OID of function lo_truncate\n"));
155  return -1;
156  }
157 
158  /*
159  * Long ago, somebody thought it'd be a good idea to declare this function
160  * as taking size_t ... but the underlying backend function only accepts a
161  * signed int32 length. So throw error if the given value overflows
162  * int32. (A possible alternative is to automatically redirect the call
163  * to lo_truncate64; but if the caller wanted to rely on that backend
164  * function being available, he could have called lo_truncate64 for
165  * himself.)
166  */
167  if (len > (size_t) INT_MAX)
168  {
170  libpq_gettext("argument of lo_truncate exceeds integer range\n"));
171  return -1;
172  }
173 
174  argv[0].isint = 1;
175  argv[0].len = 4;
176  argv[0].u.integer = fd;
177 
178  argv[1].isint = 1;
179  argv[1].len = 4;
180  argv[1].u.integer = (int) len;
181 
182  res = PQfn(conn, conn->lobjfuncs->fn_lo_truncate,
183  &retval, &result_len, 1, argv, 2);
184 
185  if (PQresultStatus(res) == PGRES_COMMAND_OK)
186  {
187  PQclear(res);
188  return retval;
189  }
190  else
191  {
192  PQclear(res);
193  return -1;
194  }
195 }
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
union PQArgBlock::@161 u
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
Oid fn_lo_truncate
Definition: libpq-int.h:279
PQExpBufferData errorMessage
Definition: libpq-int.h:526
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
#define libpq_gettext(x)
Definition: libpq-int.h:805
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_truncate64()

int lo_truncate64 ( PGconn conn,
int  fd,
pg_int64  len 
)

Definition at line 205 of file fe-lobj.c.

References pg_conn::errorMessage, fd(), pgLobjfuncs::fn_lo_truncate64, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, libpq_gettext, lo_hton64(), lo_initialize(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), printfPQExpBuffer(), PQArgBlock::ptr, and PQArgBlock::u.

Referenced by my_truncate().

206 {
207  PQArgBlock argv[2];
208  PGresult *res;
209  int retval;
210  int result_len;
211 
212  if (conn == NULL || conn->lobjfuncs == NULL)
213  {
214  if (lo_initialize(conn) < 0)
215  return -1;
216  }
217 
218  if (conn->lobjfuncs->fn_lo_truncate64 == 0)
219  {
221  libpq_gettext("cannot determine OID of function lo_truncate64\n"));
222  return -1;
223  }
224 
225  argv[0].isint = 1;
226  argv[0].len = 4;
227  argv[0].u.integer = fd;
228 
229  len = lo_hton64(len);
230  argv[1].isint = 0;
231  argv[1].len = 8;
232  argv[1].u.ptr = (int *) &len;
233 
234  res = PQfn(conn, conn->lobjfuncs->fn_lo_truncate64,
235  &retval, &result_len, 1, argv, 2);
236 
237  if (PQresultStatus(res) == PGRES_COMMAND_OK)
238  {
239  PQclear(res);
240  return retval;
241  }
242  else
243  {
244  PQclear(res);
245  return -1;
246  }
247 }
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
union PQArgBlock::@161 u
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
static pg_int64 lo_hton64(pg_int64 host64)
Definition: fe-lobj.c:1061
Oid fn_lo_truncate64
Definition: libpq-int.h:280
PQExpBufferData errorMessage
Definition: libpq-int.h:526
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
int * ptr
Definition: libpq-fe.h:229
#define libpq_gettext(x)
Definition: libpq-int.h:805
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_unlink()

int lo_unlink ( PGconn conn,
Oid  lobjId 
)

Definition at line 628 of file fe-lobj.c.

References pgLobjfuncs::fn_lo_unlink, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, lo_initialize(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), and PQArgBlock::u.

Referenced by do_lo_unlink(), and vacuumlo().

629 {
630  PQArgBlock argv[1];
631  PGresult *res;
632  int result_len;
633  int retval;
634 
635  if (conn == NULL || conn->lobjfuncs == NULL)
636  {
637  if (lo_initialize(conn) < 0)
638  return -1;
639  }
640 
641  argv[0].isint = 1;
642  argv[0].len = 4;
643  argv[0].u.integer = lobjId;
644 
645  res = PQfn(conn, conn->lobjfuncs->fn_lo_unlink,
646  &retval, &result_len, 1, argv, 1);
647  if (PQresultStatus(res) == PGRES_COMMAND_OK)
648  {
649  PQclear(res);
650  return retval;
651  }
652  else
653  {
654  PQclear(res);
655  return -1;
656  }
657 }
union PQArgBlock::@161 u
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
void PQclear(PGresult *res)
Definition: fe-exec.c:694
Oid fn_lo_unlink
Definition: libpq-int.h:274
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ lo_write()

int lo_write ( PGconn conn,
int  fd,
const char *  buf,
size_t  len 
)

Definition at line 312 of file fe-lobj.c.

References pg_conn::errorMessage, fd(), pgLobjfuncs::fn_lo_write, PQArgBlock::integer, PQArgBlock::isint, PQArgBlock::len, libpq_gettext, lo_initialize(), pg_conn::lobjfuncs, PGRES_COMMAND_OK, PQclear(), PQfn(), PQresultStatus(), printfPQExpBuffer(), PQArgBlock::ptr, PQArgBlock::u, and unconstify.

Referenced by lo_import_internal().

313 {
314  PQArgBlock argv[2];
315  PGresult *res;
316  int result_len;
317  int retval;
318 
319  if (conn == NULL || conn->lobjfuncs == NULL)
320  {
321  if (lo_initialize(conn) < 0)
322  return -1;
323  }
324 
325  /*
326  * Long ago, somebody thought it'd be a good idea to declare this function
327  * as taking size_t ... but the underlying backend function only accepts a
328  * signed int32 length. So throw error if the given value overflows
329  * int32.
330  */
331  if (len > (size_t) INT_MAX)
332  {
334  libpq_gettext("argument of lo_write exceeds integer range\n"));
335  return -1;
336  }
337 
338  argv[0].isint = 1;
339  argv[0].len = 4;
340  argv[0].u.integer = fd;
341 
342  argv[1].isint = 0;
343  argv[1].len = (int) len;
344  argv[1].u.ptr = (int *) unconstify(char *, buf);
345 
346  res = PQfn(conn, conn->lobjfuncs->fn_lo_write,
347  &retval, &result_len, 1, argv, 2);
348  if (PQresultStatus(res) == PGRES_COMMAND_OK)
349  {
350  PQclear(res);
351  return retval;
352  }
353  else
354  {
355  PQclear(res);
356  return -1;
357  }
358 }
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
union PQArgBlock::@161 u
Oid fn_lo_write
Definition: libpq-int.h:282
PGresult * PQfn(PGconn *conn, int fnid, int *result_buf, int *result_len, int result_is_int, const PQArgBlock *args, int nargs)
Definition: fe-exec.c:2652
static int fd(const char *x, int i)
Definition: preproc-init.c:105
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
static char * buf
Definition: pg_test_fsync.c:67
#define unconstify(underlying_type, expr)
Definition: c.h:1206
PQExpBufferData errorMessage
Definition: libpq-int.h:526
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:441
int * ptr
Definition: libpq-fe.h:229
#define libpq_gettext(x)
Definition: libpq-int.h:805
int integer
Definition: libpq-fe.h:230
static int lo_initialize(PGconn *conn)
Definition: fe-lobj.c:876

◆ pg_char_to_encoding()

int pg_char_to_encoding ( const char *  name)

Definition at line 550 of file encnames.c.

References clean_encoding_name(), sort-test::key, lengthof, NAMEDATALEN, and pg_encname_tbl.

Referenced by CreateConversionCommand(), length_in_encoding(), main(), PG_char_to_encoding(), pg_convert(), pg_get_utf8_id(), pg_valid_client_encoding(), pg_valid_server_encoding(), PQenv2encoding(), pqSaveParameterStatus(), ProcessCopyOptions(), processEncodingEntry(), and to_ascii_encname().

551 {
552  unsigned int nel = lengthof(pg_encname_tbl);
553  const pg_encname *base = pg_encname_tbl,
554  *last = base + nel - 1,
555  *position;
556  int result;
557  char buff[NAMEDATALEN],
558  *key;
559 
560  if (name == NULL || *name == '\0')
561  return -1;
562 
563  if (strlen(name) >= NAMEDATALEN)
564  return -1; /* it's certainly not in the table */
565 
566  key = clean_encoding_name(name, buff);
567 
568  while (last >= base)
569  {
570  position = base + ((last - base) >> 1);
571  result = key[0] - position->name[0];
572 
573  if (result == 0)
574  {
575  result = strcmp(key, position->name);
576  if (result == 0)
577  return position->encoding;
578  }
579  if (result < 0)
580  last = position - 1;
581  else
582  base = position + 1;
583  }
584  return -1;
585 }
#define lengthof(array)
Definition: c.h:668
#define NAMEDATALEN
static const pg_encname pg_encname_tbl[]
Definition: encnames.c:39
const char * name
Definition: encode.c:555
static char * clean_encoding_name(const char *key, char *newkey)
Definition: encnames.c:525

◆ pg_encoding_to_char()

◆ pg_valid_server_encoding_id()

int pg_valid_server_encoding_id ( int  encoding)

Definition at line 514 of file encnames.c.

References PG_VALID_BE_ENCODING.

Referenced by setup_locale_encoding().

515 {
517 }
#define PG_VALID_BE_ENCODING(_enc)
Definition: pg_wchar.h:295
int32 encoding
Definition: pg_database.h:41

◆ PQbackendPID()

int PQbackendPID ( const PGconn conn)

Definition at line 6713 of file fe-connect.c.

References pg_conn::be_pid, CONNECTION_OK, and pg_conn::status.

Referenced by get_prompt(), libpqrcv_get_backend_pid(), main(), and StartLogStreamer().

6714 {
6715  if (!conn || conn->status != CONNECTION_OK)
6716  return 0;
6717  return conn->be_pid;
6718 }
ConnStatusType status
Definition: libpq-int.h:388
int be_pid
Definition: libpq-int.h:434

◆ PQbinaryTuples()

int PQbinaryTuples ( const PGresult res)

Definition at line 2785 of file fe-exec.c.

References pg_result::binary.

Referenced by ProcessResult().

2786 {
2787  if (!res)
2788  return 0;
2789  return res->binary;
2790 }
int binary
Definition: libpq-int.h:179

◆ PQcancel()

int PQcancel ( PGcancel cancel,
char *  errbuf,
int  errbufsize 
)

Definition at line 4461 of file fe-connect.c.

References pg_cancel::be_key, pg_cancel::be_pid, internal_cancel(), pg_cancel::raddr, and strlcpy().

Referenced by dblink_cancel_query(), disconnectDatabase(), DisconnectDatabase(), handle_sigint(), pgfdw_cancel_query(), setup_cancel_handler(), ShutdownWorkersHard(), sigTermHandler(), and try_complete_step().

4462 {
4463  if (!cancel)
4464  {
4465  strlcpy(errbuf, "PQcancel() -- no cancel object supplied", errbufsize);
4466  return false;
4467  }
4468 
4469  return internal_cancel(&cancel->raddr, cancel->be_pid, cancel->be_key,
4470  errbuf, errbufsize);
4471 }
static int internal_cancel(SockAddr *raddr, int be_pid, int be_key, char *errbuf, int errbufsize)
Definition: fe-connect.c:4357
int be_pid
Definition: libpq-int.h:539
int be_key
Definition: libpq-int.h:540
size_t strlcpy(char *dst, const char *src, size_t siz)
Definition: strlcpy.c:45
SockAddr raddr
Definition: libpq-int.h:538

◆ PQclear()

void PQclear ( PGresult res)

Definition at line 694 of file fe-exec.c.

References pg_result::attDescs, pg_result::curBlock, pg_result::errFields, pg_result::events, free, i, PGEvent::name, pg_result::nEvents, pgresult_data::next, pg_result::paramDescs, PGEvent::passThrough, PGEVT_RESULTDESTROY, PGEvent::proc, PGEventResultDestroy::result, PGEvent::resultInitialized, and pg_result::tuples.

Referenced by _check_database_version(), _doSetSessionAuth(), _selectOutputSchema(), _selectTableAccessMethod(), _selectTablespace(), add_tablespace_footer(), append_depends_on_extension(), appendQualifiedRelation(), BaseBackup(), binary_upgrade_set_pg_class_oids(), binary_upgrade_set_type_oids_by_rel_oid(), binary_upgrade_set_type_oids_by_type_oid(), buildMatViewRefreshDependencies(), buildShSecLabels(), check_for_data_type_usage(), check_for_isn_and_int8_passing_mismatch(), check_for_jsonb_9_4_usage(), check_for_pg_role_prefix(), check_for_prepared_transactions(), check_for_reg_data_type_usage(), check_for_tables_with_oids(), check_is_install_user(), check_loadable_libraries(), check_prepare_conn(), check_proper_datallowconn(), ClearOrSaveResult(), close_cursor(), cluster_all_databases(), connectDatabase(), ConnectDatabase(), connectToServer(), convertTSFunction(), create_cursor(), CreateReplicationSlot(), createViewAsClause(), dblink_close(), dblink_exec(), dblink_fetch(), dblink_open(), dblink_res_error(), deallocate_one(), describeAccessMethods(), describeAggregates(), describeFunctions(), describeOneTableDetails(), describeOneTSConfig(), describeOneTSParser(), describeOperators(), describePublications(), DescribeQuery(), describeRoles(), describeSubscriptions(), describeTableDetails(), describeTablespaces(), describeTypes(), descriptor_free(), do_lo_import(), do_lo_list(), do_sql_command(), dropDBs(), DropReplicationSlot(), dropRoles(), dropTablespaces(), dumpAgg(), dumpBaseType(), dumpBlobs(), dumpCollation(), dumpCompositeType(), dumpCompositeTypeColComments(), dumpConversion(), dumpDatabase(), dumpDatabaseConfig(), dumpDatabases(), dumpDomain(), dumpEnumType(), dumpFunc(), dumpGroups(), dumpOpclass(), dumpOpfamily(), dumpOpr(), dumpRangeType(), dumpRoleMembership(), dumpRoles(), dumpRule(), dumpSearchPath(), dumpSequence(), dumpSequenceData(), dumpStatisticsExt(), dumpTable(), dumpTableData_copy(), dumpTableData_insert(), dumpTableSchema(), dumpTablespaces(), dumpTSConfig(), dumpTSDictionary(), dumpUserConfig(), dumpUserMappings(), ecpg_autostart_transaction(), ecpg_check_PQresult(), ecpg_is_type_an_array(), ecpg_process_output(), ECPGdescribe(), ECPGsetcommit(), ECPGtrans(), EndDBCopyMode(), exec_command_password(), ExecQueryUsingCursor(), execute_foreign_modify(), executeCommand(), executeMaintenanceCommand(), executeQueryOrDie(), ExecuteSqlCommand(), ExecuteSqlStatement(), executeStatement(), expand_dbname_patterns(), expand_foreign_server_name_patterns(), expand_schema_name_patterns(), expand_table_name_patterns(), fail_lo_xact(), fetch_more_data(), findLastBuiltinOid_V71(), finish_foreign_modify(), finish_lo_xact(), get_create_object_cmd(), get_db_infos(), get_language_name(), get_loadable_libraries(), get_parallel_object_list(), get_rel_infos(), get_remote_estimate(), get_returning_data(), get_synchronized_snapshot(), get_tablespace_paths(), getAccessMethods(), getAggregates(), getBlobs(), getCasts(), getCollations(), GetConnection(), getConstraints(), getConversions(), getCopyStart(), getDefaultACLs(), getDependencies(), getDomainConstraints(), getEventTriggers(), getExtendedStatistics(), getExtensionMembership(), getExtensions(), getForeignDataWrappers(), getForeignServers(), getFormattedTypeName(), getFuncs(), getIndexes(), getInherits(), getNamespaces(), getOpclasses(), getOperators(), getOpfamilies(), getParamDescriptions(), getPolicies(), getProcLangs(), getPublications(), getPublicationTables(), getRowDescriptions(), getRules(), getSubscriptions(), getTableAttrs(), GetTableInfo(), getTables(), getTransforms(), getTriggers(), getTSConfigurations(), getTSDictionaries(), getTSParsers(), getTSTemplates(), getTypes(), handleCopyIn(), HandleEndOfCopyStream(), initGenerateDataClientSide(), libpq_executeFileMap(), libpqConnect(), libpqGetFile(), libpqProcessFileList(), libpqrcv_create_slot(), libpqrcv_endstreaming(), libpqrcv_exec(), libpqrcv_identify_system(), libpqrcv_PQexec(), libpqrcv_readtimelinehistoryfile(), libpqrcv_receive(), libpqrcv_startstreaming(), listAllDbs(), listCasts(), listCollations(), listConversions(), listDbRoleSettings(), listDefaultACLs(), listDomains(), listEventTriggers(), listExtensionContents(), listExtensions(), listForeignDataWrappers(), listForeignServers(), listForeignTables(), listLanguages(), listOneExtensionContents(), listOperatorClasses(), listOperatorFamilies(), listOpFamilyFunctions(), listOpFamilyOperators(), listPartitionedTables(), listPublications(), listSchemas(), listTables(), listTSConfigs(), listTSConfigsVerbose(), listTSDictionaries(), listTSParsers(), listTSParsersVerbose(), listTSTemplates(), listUserMappings(), lo_close(), lo_creat(), lo_create(), lo_initialize(), lo_lseek(), lo_lseek64(), lo_open(), lo_read(), lo_tell(), lo_tell64(), lo_truncate(), lo_truncate64(), lo_unlink(), lo_write(), lockTableForWorker(), lookup_object_oid(), main(), materializeResult(), new_9_0_populate_pg_largeobject_metadata(), objectDescription(), old_9_6_invalidate_hash_indexes(), permissionsList(), pg_attribute_noreturn(), pgfdw_cancel_query(), pgfdw_exec_cleanup_query(), pgfdw_get_cleanup_result(), pgfdw_get_result(), pgfdw_report_error(), pgfdw_xact_callback(), postgresAcquireSampleRowsFunc(), postgresAnalyzeForeignTable(), postgresEndDirectModify(), postgresImportForeignSchema(), postgresReScanForeignScan(), pqClearAsyncResult(), PQconnectPoll(), PQcopyResult(), PQencryptPasswordConn(), pqEndcopy2(), pqEndcopy3(), PQexecFinish(), PQexecStart(), pqGetErrorNotice2(), pqGetErrorNotice3(), pqInternalNotice(), PQmakeEmptyPGresult(), pqRowProcessor(), PQsetClientEncoding(), pqSetenvPoll(), prepare_common(), prepare_foreign_modify(), processExtensionTables(), processQueryResult(), ProcessResult(), PSQLexecWatch(), readCommandResponse(), ReceiveCopyData(), receiveFileChunks(), ReceiveXlogStream(), ReconnectToServer(), reindex_all_databases(), RetrieveDataDirCreatePerm(), RetrieveWalSegSize(), run_permutation(), run_simple_command(), run_simple_query(), RunIdentifySystem(), sendCommand(), SendQuery(), set_frozenxids(), setup_connection(), sql_conn(), sql_exec(), start_lo_xact(), store_returning_result(), storeQueryResult(), StreamLogicalLog(), try_complete_step(), tryExecuteStatement(), vacuum_all_databases(), vacuum_one_database(), and vacuumlo().

695 {
696  PGresult_data *block;
697  int i;
698 
699  if (!res)
700  return;
701 
702  for (i = 0; i < res->nEvents; i++)
703  {
704  /* only send DESTROY to successfully-initialized event procs */
705  if (res->events[i].resultInitialized)
706  {
708 
709  evt.result = res;
710  (void) res->events[i].proc(PGEVT_RESULTDESTROY, &evt,
711  res->events[i].passThrough);
712  }
713  free(res->events[i].name);
714  }
715 
716  if (res->events)
717  free(res->events);
718 
719  /* Free all the subsidiary blocks */
720  while ((block = res->curBlock) != NULL)
721  {
722  res->curBlock = block->next;
723  free(block);
724  }
725 
726  /* Free the top-level tuple pointer array */
727  if (res->tuples)
728  free(res->tuples);
729 
730  /* zero out the pointer fields to catch programming errors */
731  res->attDescs = NULL;
732  res->tuples = NULL;
733  res->paramDescs = NULL;
734  res->errFields = NULL;
735  res->events = NULL;
736  res->nEvents = 0;
737  /* res->curBlock was zeroed out earlier */
738 
739  /* Free the PGresult structure itself */
740  free(res);
741 }
PGMessageField * errFields
Definition: libpq-int.h:197
bool resultInitialized
Definition: libpq-int.h:164
PGresult_data * next
Definition: libpq-int.h:105
int nEvents
Definition: libpq-int.h:188
PGresAttDesc * attDescs
Definition: libpq-int.h:171
PGresAttValue ** tuples
Definition: libpq-int.h:172
PGresult_data * curBlock
Definition: libpq-int.h:208
#define free(a)
Definition: header.h:65
PGresParamDesc * paramDescs
Definition: libpq-int.h:176
PGEventProc proc
Definition: libpq-int.h:160
int i
void * passThrough
Definition: libpq-int.h:162
PGEvent * events
Definition: libpq-int.h:187
char * name
Definition: libpq-int.h:161

◆ PQclientEncoding()

int PQclientEncoding ( const PGconn conn)

◆ PQcmdStatus()

char* PQcmdStatus ( PGresult res)

Definition at line 3040 of file fe-exec.c.

References pg_result::cmdStatus.

Referenced by dblink_exec(), ecpg_process_output(), materializeResult(), PrintQueryResults(), PrintQueryStatus(), PSQLexecWatch(), and SendQuery().

3041 {
3042  if (!res)
3043  return NULL;
3044  return res->cmdStatus;
3045 }
char cmdStatus[CMDSTATUS_LEN]
Definition: libpq-int.h:178

◆ PQcmdTuples()

char* PQcmdTuples ( PGresult res)

Definition at line 3110 of file fe-exec.c.

References pg_result::cmdStatus, pg_result::noticeHooks, and pqInternalNotice().

Referenced by ecpg_process_output(), execute_dml_stmt(), execute_foreign_modify(), and SetResultVariables().

3111 {
3112  char *p,
3113  *c;
3114 
3115  if (!res)
3116  return "";
3117 
3118  if (strncmp(res->cmdStatus, "INSERT ", 7) == 0)
3119  {
3120  p = res->cmdStatus + 7;
3121  /* INSERT: skip oid and space */
3122  while (*p && *p != ' ')
3123  p++;
3124  if (*p == 0)
3125  goto interpret_error; /* no space? */
3126  p++;
3127  }
3128  else if (strncmp(res->cmdStatus, "SELECT ", 7) == 0 ||
3129  strncmp(res->cmdStatus, "DELETE ", 7) == 0 ||
3130  strncmp(res->cmdStatus, "UPDATE ", 7) == 0)
3131  p = res->cmdStatus + 7;
3132  else if (strncmp(res->cmdStatus, "FETCH ", 6) == 0)
3133  p = res->cmdStatus + 6;
3134  else if (strncmp(res->cmdStatus, "MOVE ", 5) == 0 ||
3135  strncmp(res->cmdStatus, "COPY ", 5) == 0)
3136  p = res->cmdStatus + 5;
3137  else
3138  return "";
3139 
3140  /* check that we have an integer (at least one digit, nothing else) */
3141  for (c = p; *c; c++)
3142  {
3143  if (!isdigit((unsigned char) *c))
3144  goto interpret_error;
3145  }
3146  if (c == p)
3147  goto interpret_error;
3148 
3149  return p;
3150 
3151 interpret_error:
3153  "could not interpret result from server: %s",
3154  res->cmdStatus);
3155  return "";
3156 }
PGNoticeHooks noticeHooks
Definition: libpq-int.h:186
char * c
void pqInternalNotice(const PGNoticeHooks *hooks, const char *fmt,...)
Definition: fe-exec.c:871
char cmdStatus[CMDSTATUS_LEN]
Definition: libpq-int.h:178

◆ PQconndefaults()

PQconninfoOption* PQconndefaults ( void  )

Definition at line 1434 of file fe-connect.c.

References conninfo_add_defaults(), conninfo_init(), initPQExpBuffer(), PQconninfoFree(), PQExpBufferDataBroken, and termPQExpBuffer().

Referenced by check_pghost_envvar(), dblink_fdw_validator(), get_connect_string(), InitPgFdwOptions(), and main().

1435 {
1436  PQExpBufferData errorBuf;
1437  PQconninfoOption *connOptions;
1438 
1439  /* We don't actually report any errors here, but callees want a buffer */
1440  initPQExpBuffer(&errorBuf);
1441  if (PQExpBufferDataBroken(errorBuf))
1442  return NULL; /* out of memory already :-( */
1443 
1444  connOptions = conninfo_init(&errorBuf);
1445  if (connOptions != NULL)
1446  {
1447  /* pass NULL errorBuf to ignore errors */
1448  if (!conninfo_add_defaults(connOptions, NULL))
1449  {
1450  PQconninfoFree(connOptions);
1451  connOptions = NULL;
1452  }
1453  }
1454 
1455  termPQExpBuffer(&errorBuf);
1456  return connOptions;
1457 }
void termPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:131
static PQconninfoOption * conninfo_init(PQExpBuffer errorMessage)
Definition: fe-connect.c:5311
void PQconninfoFree(PQconninfoOption *connOptions)
Definition: fe-connect.c:6519
#define PQExpBufferDataBroken(buf)
Definition: pqexpbuffer.h:67
static bool conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage)
Definition: fe-connect.c:5747
void initPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:92

◆ PQconnectdb()

PGconn* PQconnectdb ( const char *  conninfo)

Definition at line 696 of file fe-connect.c.

References conn, connectDBComplete(), CONNECTION_BAD, PQconnectStart(), and pg_conn::status.

Referenced by dblink_connect(), dblink_get_conn(), get_db_conn(), libpqConnect(), and main().

697 {
698  PGconn *conn = PQconnectStart(conninfo);
699 
700  if (conn && conn->status != CONNECTION_BAD)
701  (void) connectDBComplete(conn);
702 
703  return conn;
704 }
PGconn * conn
Definition: streamutil.c:54
static int connectDBComplete(PGconn *conn)
Definition: fe-connect.c:2072
PGconn * PQconnectStart(const char *conninfo)
Definition: fe-connect.c:822
ConnStatusType status
Definition: libpq-int.h:388

◆ PQconnectdbParams()

PGconn* PQconnectdbParams ( const char *const *  keywords,
const char *const *  values,
int  expand_dbname 
)

Definition at line 640 of file fe-connect.c.

References conn, connectDBComplete(), CONNECTION_BAD, PQconnectStartParams(), and pg_conn::status.

Referenced by _connectDB(), connect_pg_server(), connectDatabase(), ConnectDatabase(), do_connect(), doConnect(), ECPGconnect(), GetConnection(), main(), sql_conn(), and vacuumlo().

643 {
644  PGconn *conn = PQconnectStartParams(keywords, values, expand_dbname);
645 
646  if (conn && conn->status != CONNECTION_BAD)
647  (void) connectDBComplete(conn);
648 
649  return conn;
650 
651 }
PGconn * conn
Definition: streamutil.c:54
static int connectDBComplete(PGconn *conn)
Definition: fe-connect.c:2072
ConnStatusType status
Definition: libpq-int.h:388
static Datum values[MAXATTR]
Definition: bootstrap.c:167
PGconn * PQconnectStartParams(const char *const *keywords, const char *const *values, int expand_dbname)
Definition: fe-connect.c:743

◆ PQconnectionNeedsPassword()

int PQconnectionNeedsPassword ( const PGconn conn)

Definition at line 6721 of file fe-connect.c.

References password, pg_conn::password_needed, and PQpass().

Referenced by _connectDB(), connectDatabase(), ConnectDatabase(), do_connect(), doConnect(), GetConnection(), main(), sql_conn(), and vacuumlo().

6722 {
6723  char *password;
6724 
6725  if (!conn)
6726  return false;
6727  password = PQpass(conn);
6728  if (conn->password_needed &&
6729  (password == NULL || password[0] == '\0'))
6730  return true;
6731  else
6732  return false;
6733 }
static char password[100]
Definition: streamutil.c:53
bool password_needed
Definition: libpq-int.h:417
char * PQpass(const PGconn *conn)
Definition: fe-connect.c:6553

◆ PQconnectionUsedPassword()

int PQconnectionUsedPassword ( const PGconn conn)

Definition at line 6736 of file fe-connect.c.

References pg_conn::password_needed.

Referenced by _connectDB(), connect_pg_server(), ConnectDatabase(), and dblink_security_check().

6737 {
6738  if (!conn)
6739  return false;
6740  if (conn->password_needed)
6741  return true;
6742  else
6743  return false;
6744 }
bool password_needed
Definition: libpq-int.h:417

◆ PQconnectPoll()

PostgresPollingStatusType PQconnectPoll ( PGconn conn)

Definition at line 2254 of file fe-connect.c.

References SockAddr::addr, pg_conn::addr_cur, pg_conn::addrlist, pg_conn::addrlist_family, addrinfo::ai_addr, addrinfo::ai_addrlen, addrinfo::ai_family, addrinfo::ai_flags, addrinfo::ai_next, AI_NUMERICHOST, addrinfo::ai_socktype, appendPQExpBuffer(), appendPQExpBufferChar(), appendPQExpBufferStr(), pg_conn::appname, Assert, pg_conn::asyncStatus, AUTH_REQ_MD5, AUTH_REQ_OK, pg_conn::auth_req_received, CHT_HOST_ADDRESS, CHT_HOST_NAME, CHT_UNIX_SOCKET, connect, connectFailureMessage(), CONNECTION_AUTH_OK, CONNECTION_AWAITING_RESPONSE, CONNECTION_BAD, CONNECTION_CHECK_TARGET, CONNECTION_CHECK_WRITABLE, CONNECTION_CONSUME, CONNECTION_GSS_STARTUP, CONNECTION_MADE, CONNECTION_NEEDED, CONNECTION_OK, CONNECTION_SETENV, CONNECTION_SSL_STARTUP, CONNECTION_STARTED, connectNoDelay(), pg_conn::connhost, pg_conn::connip, PQExpBufferData::data, EINPROGRESS, EINTR, EnvironmentOptions, ERRCODE_APPNAME_UNKNOWN, pg_conn::errorMessage, EWOULDBLOCK, pg_conn::fbappname, free, gai_strerror, getHostaddr(), getpeereid(), pg_conn::gssencmode, pg_conn_host::host, pg_conn_host::hostaddr, pg_conn::inCursor, pg_conn::inEnd, pg_conn::inStart, IS_AF_UNIX, pg_conn::laddr, PQExpBufferData::len, libpq_gettext, MAXPGPATH, MemSet, pg_conn::nconnhost, NEGOTIATE_GSS_CODE, NEGOTIATE_SSL_CODE, pg_conn::next_eo, NI_MAXHOST, parse_int_param(), PG_DIAG_SQLSTATE, pg_fe_sendauth(), pg_getaddrinfo_all(), pg_GSS_have_cred_cache(), pg_hton32, PG_PROTOCOL, PG_PROTOCOL_MAJOR, pg_set_noblock(), PG_STRERROR_R_BUFLEN, PGASYNC_BUSY, PGASYNC_IDLE, PGINVALID_SOCKET, pgpassfileWarning(), PGRES_FATAL_ERROR, PGRES_POLLING_FAILED, PGRES_POLLING_OK, PGRES_POLLING_READING, PGRES_POLLING_WRITING, PGRES_TUPLES_OK, pg_conn_host::port, pqBuildStartupPacket2(), pqBuildStartupPacket3(), pqCheckInBufferSpace(), PQclear(), pqClearAsyncResult(), PQconsumeInput(), pqDropConnection(), pqDropServerData(), pqFlush(), pqGetc(), pqGetErrorNotice3(), pqGetInt(), pqGetpwuid(), PQgetResult(), pqGets_append(), PQgetvalue(), PQisBusy(), PQntuples(), pqPacketSend(), pqReadData(), PQresultErrorField(), PQresultStatus(), pqsecure_initialize(), pqsecure_open_client(), pqsecure_open_gss(), PQsendQuery(), pqSetenvPoll(), PQTRANS_IDLE, printfPQExpBuffer(), pg_conn::pversion, pg_conn::raddr, release_conn_addrinfo(), pg_conn::requirepeer, restoreErrorMessage(), pg_result::resultStatus, SockAddr::salen, saveErrorMessage(), pg_conn::send_appname, sendTerminateConn(), pg_conn::setenv_state, SETENV_STATE_CLIENT_ENCODING_SEND, setKeepalivesCount(), setKeepalivesIdle(), setKeepalivesInterval(), setTCPUserTimeout(), pg_conn::sigpipe_flag, pg_conn::sigpipe_so, snprintf, pg_conn::sock, SOCK_ERRNO, SOCK_STRERROR, socket, pg_conn::ssl_in_use, pg_conn::sslmode, pg_conn::status, STATUS_OK, strerror_r, pg_conn::sversion, pg_conn::target_session_attrs, termPQExpBuffer(), pg_conn::try_next_addr, pg_conn::try_next_host, pg_conn_host::type, UNIXSOCK_PATH, UNIXSOCK_PATH_BUFLEN, useKeepalives(), _internalPQconninfoOption::val, pg_conn::whichhost, and pg_conn::xactStatus.

Referenced by connectDBComplete(), connectDBStart(), libpqrcv_connect(), and PQresetPoll().

2255 {
2256  bool reset_connection_state_machine = false;
2257  bool need_new_connection = false;
2258  PGresult *res;
2259  char sebuf[PG_STRERROR_R_BUFLEN];
2260  int optval;
2261  PQExpBufferData savedMessage;
2262 
2263  if (conn == NULL)
2264  return PGRES_POLLING_FAILED;
2265 
2266  /* Get the new data */
2267  switch (conn->status)
2268  {
2269  /*
2270  * We really shouldn't have been polled in these two cases, but we
2271  * can handle it.
2272  */
2273  case CONNECTION_BAD:
2274  return PGRES_POLLING_FAILED;
2275  case CONNECTION_OK:
2276  return PGRES_POLLING_OK;
2277 
2278  /* These are reading states */
2280  case CONNECTION_AUTH_OK:
2281  {
2282  /* Load waiting data */
2283  int n = pqReadData(conn);
2284 
2285  if (n < 0)
2286  goto error_return;
2287  if (n == 0)
2288  return PGRES_POLLING_READING;
2289 
2290  break;
2291  }
2292 
2293  /* These are writing states, so we just proceed. */
2294  case CONNECTION_STARTED:
2295  case CONNECTION_MADE:
2296  break;
2297 
2298  /* We allow pqSetenvPoll to decide whether to proceed. */
2299  case CONNECTION_SETENV:
2300  break;
2301 
2302  /* Special cases: proceed without waiting. */
2304  case CONNECTION_NEEDED:
2306  case CONNECTION_CONSUME:
2308  break;
2309 
2310  default:
2312  libpq_gettext("invalid connection state, probably indicative of memory corruption\n"));
2313  goto error_return;
2314  }
2315 
2316 
2317 keep_going: /* We will come back to here until there is
2318  * nothing left to do. */
2319 
2320  /* Time to advance to next address, or next host if no more addresses? */
2321  if (conn->try_next_addr)
2322  {
2323  if (conn->addr_cur && conn->addr_cur->ai_next)
2324  {
2325  conn->addr_cur = conn->addr_cur->ai_next;
2326  reset_connection_state_machine = true;
2327  }
2328  else
2329  conn->try_next_host = true;
2330  conn->try_next_addr = false;
2331  }
2332 
2333  /* Time to advance to next connhost[] entry? */
2334  if (conn->try_next_host)
2335  {
2336  pg_conn_host *ch;
2337  struct addrinfo hint;
2338  int thisport;
2339  int ret;
2340  char portstr[MAXPGPATH];
2341 
2342  if (conn->whichhost + 1 >= conn->nconnhost)
2343  {
2344  /*
2345  * Oops, no more hosts. An appropriate error message is already
2346  * set up, so just set the right status.
2347  */
2348  goto error_return;
2349  }
2350  conn->whichhost++;
2351 
2352  /* Drop any address info for previous host */
2353  release_conn_addrinfo(conn);
2354 
2355  /*
2356  * Look up info for the new host. On failure, log the problem in
2357  * conn->errorMessage, then loop around to try the next host. (Note
2358  * we don't clear try_next_host until we've succeeded.)
2359  */
2360  ch = &conn->connhost[conn->whichhost];
2361 
2362  /* Initialize hint structure */
2363  MemSet(&hint, 0, sizeof(hint));
2364  hint.ai_socktype = SOCK_STREAM;
2365  conn->addrlist_family = hint.ai_family = AF_UNSPEC;
2366 
2367  /* Figure out the port number we're going to use. */
2368  if (ch->port == NULL || ch->port[0] == '\0')
2369  thisport = DEF_PGPORT;
2370  else
2371  {
2372  if (!parse_int_param(ch->port, &thisport, conn, "port"))
2373  goto error_return;
2374 
2375  if (thisport < 1 || thisport > 65535)
2376  {
2378  libpq_gettext("invalid port number: \"%s\"\n"),
2379  ch->port);
2380  goto keep_going;
2381  }
2382  }
2383  snprintf(portstr, sizeof(portstr), "%d", thisport);
2384 
2385  /* Use pg_getaddrinfo_all() to resolve the address */
2386  switch (ch->type)
2387  {
2388  case CHT_HOST_NAME:
2389  ret = pg_getaddrinfo_all(ch->host, portstr, &hint,
2390  &conn->addrlist);
2391  if (ret || !conn->addrlist)
2392  {
2394  libpq_gettext("could not translate host name \"%s\" to address: %s\n"),
2395  ch->host, gai_strerror(ret));
2396  goto keep_going;
2397  }
2398  break;
2399 
2400  case CHT_HOST_ADDRESS:
2401  hint.ai_flags = AI_NUMERICHOST;
2402  ret = pg_getaddrinfo_all(ch->hostaddr, portstr, &hint,
2403  &conn->addrlist);
2404  if (ret || !conn->addrlist)
2405  {
2407  libpq_gettext("could not parse network address \"%s\": %s\n"),
2408  ch->hostaddr, gai_strerror(ret));
2409  goto keep_going;
2410  }
2411  break;
2412 
2413  case CHT_UNIX_SOCKET:
2414 #ifdef HAVE_UNIX_SOCKETS
2415  conn->addrlist_family = hint.ai_family = AF_UNIX;
2416  UNIXSOCK_PATH(portstr, thisport, ch->host);
2417  if (strlen(portstr) >= UNIXSOCK_PATH_BUFLEN)
2418  {
2420  libpq_gettext("Unix-domain socket path \"%s\" is too long (maximum %d bytes)\n"),
2421  portstr,
2422  (int) (UNIXSOCK_PATH_BUFLEN - 1));
2423  goto keep_going;
2424  }
2425 
2426  /*
2427  * NULL hostname tells pg_getaddrinfo_all to parse the service
2428  * name as a Unix-domain socket path.
2429  */
2430  ret = pg_getaddrinfo_all(NULL, portstr, &hint,
2431  &conn->addrlist);
2432  if (ret || !conn->addrlist)
2433  {
2435  libpq_gettext("could not translate Unix-domain socket path \"%s\" to address: %s\n"),
2436  portstr, gai_strerror(ret));
2437  goto keep_going;
2438  }
2439 #else
2440  Assert(false);
2441 #endif
2442  break;
2443  }
2444 
2445  /* OK, scan this addrlist for a working server address */
2446  conn->addr_cur = conn->addrlist;
2447  reset_connection_state_machine = true;
2448  conn->try_next_host = false;
2449  }
2450 
2451  /* Reset connection state machine? */
2452  if (reset_connection_state_machine)
2453  {
2454  /*
2455  * (Re) initialize our connection control variables for a set of
2456  * connection attempts to a single server address. These variables
2457  * must persist across individual connection attempts, but we must
2458  * reset them when we start to consider a new server.
2459  */
2460  conn->pversion = PG_PROTOCOL(3, 0);
2461  conn->send_appname = true;
2462 #ifdef USE_SSL
2463  /* initialize these values based on SSL mode */
2464  conn->allow_ssl_try = (conn->sslmode[0] != 'd'); /* "disable" */
2465  conn->wait_ssl_try = (conn->sslmode[0] == 'a'); /* "allow" */
2466 #endif
2467 
2468  reset_connection_state_machine = false;
2469  need_new_connection = true;
2470  }
2471 
2472  /* Force a new connection (perhaps to the same server as before)? */
2473  if (need_new_connection)
2474  {
2475  /* Drop any existing connection */
2476  pqDropConnection(conn, true);
2477 
2478  /* Reset all state obtained from old server */
2479  pqDropServerData(conn);
2480 
2481  /* Drop any PGresult we might have, too */
2482  conn->asyncStatus = PGASYNC_IDLE;
2483  conn->xactStatus = PQTRANS_IDLE;
2484  pqClearAsyncResult(conn);
2485 
2486  /* Reset conn->status to put the state machine in the right state */
2487  conn->status = CONNECTION_NEEDED;
2488 
2489  need_new_connection = false;
2490  }
2491 
2492  /* Now try to advance the state machine for this connection */
2493  switch (conn->status)
2494  {
2495  case CONNECTION_NEEDED:
2496  {
2497  /*
2498  * Try to initiate a connection to one of the addresses
2499  * returned by pg_getaddrinfo_all(). conn->addr_cur is the
2500  * next one to try.
2501  *
2502  * The extra level of braces here is historical. It's not
2503  * worth reindenting this whole switch case to remove 'em.
2504  */
2505  {
2506  struct addrinfo *addr_cur = conn->addr_cur;
2507  char host_addr[NI_MAXHOST];
2508 
2509  /*
2510  * Advance to next possible host, if we've tried all of
2511  * the addresses for the current host.
2512  */
2513  if (addr_cur == NULL)
2514  {
2515  conn->try_next_host = true;
2516  goto keep_going;
2517  }
2518 
2519  /* Remember current address for possible error msg */
2520  memcpy(&conn->raddr.addr, addr_cur->ai_addr,
2521  addr_cur->ai_addrlen);
2522  conn->raddr.salen = addr_cur->ai_addrlen;
2523 
2524  /* set connip */
2525  if (conn->connip != NULL)
2526  {
2527  free(conn->connip);
2528  conn->connip = NULL;
2529  }
2530 
2531  getHostaddr(conn, host_addr, NI_MAXHOST);
2532  if (strlen(host_addr) > 0)
2533  conn->connip = strdup(host_addr);
2534 
2535  /*
2536  * purposely ignore strdup failure; not a big problem if
2537  * it fails anyway.
2538  */
2539 
2540  conn->sock = socket(addr_cur->ai_family, SOCK_STREAM, 0);
2541  if (conn->sock == PGINVALID_SOCKET)
2542  {
2543  /*
2544  * Silently ignore socket() failure if we have more
2545  * addresses to try; this reduces useless chatter in
2546  * cases where the address list includes both IPv4 and
2547  * IPv6 but kernel only accepts one family.
2548  */
2549  if (addr_cur->ai_next != NULL ||
2550  conn->whichhost + 1 < conn->nconnhost)
2551  {
2552  conn->try_next_addr = true;
2553  goto keep_going;
2554  }
2556  libpq_gettext("could not create socket: %s\n"),
2557  SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
2558  goto error_return;
2559  }
2560 
2561  /*
2562  * Select socket options: no delay of outgoing data for
2563  * TCP sockets, nonblock mode, close-on-exec. Try the
2564  * next address if any of this fails.
2565  */
2566  if (!IS_AF_UNIX(addr_cur->ai_family))
2567  {
2568  if (!connectNoDelay(conn))
2569  {
2570  /* error message already created */
2571  conn->try_next_addr = true;
2572  goto keep_going;
2573  }
2574  }
2575  if (!pg_set_noblock(conn->sock))
2576  {
2578  libpq_gettext("could not set socket to nonblocking mode: %s\n"),
2579  SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
2580  conn->try_next_addr = true;
2581  goto keep_going;
2582  }
2583 
2584 #ifdef F_SETFD
2585  if (fcntl(conn->sock, F_SETFD, FD_CLOEXEC) == -1)
2586  {
2588  libpq_gettext("could not set socket to close-on-exec mode: %s\n"),
2589  SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
2590  conn->try_next_addr = true;
2591  goto keep_going;
2592  }
2593 #endif /* F_SETFD */
2594 
2595  if (!IS_AF_UNIX(addr_cur->ai_family))
2596  {
2597 #ifndef WIN32
2598  int on = 1;
2599 #endif
2600  int usekeepalives = useKeepalives(conn);
2601  int err = 0;
2602 
2603  if (usekeepalives < 0)
2604  {
2606  libpq_gettext("keepalives parameter must be an integer\n"));
2607  err = 1;
2608  }
2609  else if (usekeepalives == 0)
2610  {
2611  /* Do nothing */
2612  }
2613 #ifndef WIN32
2614  else if (setsockopt(conn->sock,
2615  SOL_SOCKET, SO_KEEPALIVE,
2616  (char *) &on, sizeof(on)) < 0)
2617  {
2619  libpq_gettext("setsockopt(%s) failed: %s\n"),
2620  "SO_KEEPALIVE",
2621  SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
2622  err = 1;
2623  }
2624  else if (!setKeepalivesIdle(conn)
2625  || !setKeepalivesInterval(conn)
2626  || !setKeepalivesCount(conn))
2627  err = 1;
2628 #else /* WIN32 */
2629 #ifdef SIO_KEEPALIVE_VALS
2630  else if (!setKeepalivesWin32(conn))
2631  err = 1;
2632 #endif /* SIO_KEEPALIVE_VALS */
2633 #endif /* WIN32 */
2634  else if (!setTCPUserTimeout(conn))
2635  err = 1;
2636 
2637  if (err)
2638  {
2639  conn->try_next_addr = true;
2640  goto keep_going;
2641  }
2642  }
2643 
2644  /*----------
2645  * We have three methods of blocking SIGPIPE during
2646  * send() calls to this socket:
2647  *
2648  * - setsockopt(sock, SO_NOSIGPIPE)
2649  * - send(sock, ..., MSG_NOSIGNAL)
2650  * - setting the signal mask to SIG_IGN during send()
2651  *
2652  * The third method requires three syscalls per send,
2653  * so we prefer either of the first two, but they are
2654  * less portable. The state is tracked in the following
2655  * members of PGconn:
2656  *
2657  * conn->sigpipe_so - we have set up SO_NOSIGPIPE
2658  * conn->sigpipe_flag - we're specifying MSG_NOSIGNAL
2659  *
2660  * If we can use SO_NOSIGPIPE, then set sigpipe_so here
2661  * and we're done. Otherwise, set sigpipe_flag so that
2662  * we will try MSG_NOSIGNAL on sends. If we get an error
2663  * with MSG_NOSIGNAL, we'll clear that flag and revert to
2664  * signal masking.
2665  *----------
2666  */
2667  conn->sigpipe_so = false;
2668 #ifdef MSG_NOSIGNAL
2669  conn->sigpipe_flag = true;
2670 #else
2671  conn->sigpipe_flag = false;
2672 #endif /* MSG_NOSIGNAL */
2673 
2674 #ifdef SO_NOSIGPIPE
2675  optval = 1;
2676  if (setsockopt(conn->sock, SOL_SOCKET, SO_NOSIGPIPE,
2677  (char *) &optval, sizeof(optval)) == 0)
2678  {
2679  conn->sigpipe_so = true;
2680  conn->sigpipe_flag = false;
2681  }
2682 #endif /* SO_NOSIGPIPE */
2683 
2684  /*
2685  * Start/make connection. This should not block, since we
2686  * are in nonblock mode. If it does, well, too bad.
2687  */
2688  if (connect(conn->sock, addr_cur->ai_addr,
2689  addr_cur->ai_addrlen) < 0)
2690  {
2691  if (SOCK_ERRNO == EINPROGRESS ||
2692 #ifdef WIN32
2693  SOCK_ERRNO == EWOULDBLOCK ||
2694 #endif
2695  SOCK_ERRNO == EINTR)
2696  {
2697  /*
2698  * This is fine - we're in non-blocking mode, and
2699  * the connection is in progress. Tell caller to
2700  * wait for write-ready on socket.
2701  */
2702  conn->status = CONNECTION_STARTED;
2703  return PGRES_POLLING_WRITING;
2704  }
2705  /* otherwise, trouble */
2706  }
2707  else
2708  {
2709  /*
2710  * Hm, we're connected already --- seems the "nonblock
2711  * connection" wasn't. Advance the state machine and
2712  * go do the next stuff.
2713  */
2714  conn->status = CONNECTION_STARTED;
2715  goto keep_going;
2716  }
2717 
2718  /*
2719  * This connection failed. Add the error report to
2720  * conn->errorMessage, then try the next address if any.
2721  */
2723  conn->try_next_addr = true;
2724  goto keep_going;
2725  }
2726  }
2727 
2728  case CONNECTION_STARTED:
2729  {
2730  ACCEPT_TYPE_ARG3 optlen = sizeof(optval);
2731 
2732  /*
2733  * Write ready, since we've made it here, so the connection
2734  * has been made ... or has failed.
2735  */
2736 
2737  /*
2738  * Now check (using getsockopt) that there is not an error
2739  * state waiting for us on the socket.
2740  */
2741 
2742  if (getsockopt(conn->sock, SOL_SOCKET, SO_ERROR,
2743  (char *) &optval, &optlen) == -1)
2744  {
2746  libpq_gettext("could not get socket error status: %s\n"),
2747  SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
2748  goto error_return;
2749  }
2750  else if (optval != 0)
2751  {
2752  /*
2753  * When using a nonblocking connect, we will typically see
2754  * connect failures at this point, so provide a friendly
2755  * error message.
2756  */
2757  connectFailureMessage(conn, optval);
2758 
2759  /*
2760  * Try the next address if any, just as in the case where
2761  * connect() returned failure immediately.
2762  */
2763  conn->try_next_addr = true;
2764  goto keep_going;
2765  }
2766 
2767  /* Fill in the client address */
2768  conn->laddr.salen = sizeof(conn->laddr.addr);
2769  if (getsockname(conn->sock,
2770  (struct sockaddr *) &conn->laddr.addr,
2771  &conn->laddr.salen) < 0)
2772  {
2774  libpq_gettext("could not get client address from socket: %s\n"),
2775  SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
2776  goto error_return;
2777  }
2778 
2779  /*
2780  * Make sure we can write before advancing to next step.
2781  */
2782  conn->status = CONNECTION_MADE;
2783  return PGRES_POLLING_WRITING;
2784  }
2785 
2786  case CONNECTION_MADE:
2787  {
2788  char *startpacket;
2789  int packetlen;
2790 
2791  /*
2792  * Implement requirepeer check, if requested and it's a
2793  * Unix-domain socket.
2794  */
2795  if (conn->requirepeer && conn->requirepeer[0] &&
2796  IS_AF_UNIX(conn->raddr.addr.ss_family))
2797  {
2798 #ifndef WIN32
2799  char pwdbuf[BUFSIZ];
2800  struct passwd pass_buf;
2801  struct passwd *pass;
2802  int passerr;
2803 #endif
2804  uid_t uid;
2805  gid_t gid;
2806 
2807  errno = 0;
2808  if (getpeereid(conn->sock, &uid, &gid) != 0)
2809  {
2810  /*
2811  * Provide special error message if getpeereid is a
2812  * stub
2813  */
2814  if (errno == ENOSYS)
2816  libpq_gettext("requirepeer parameter is not supported on this platform\n"));
2817  else
2819  libpq_gettext("could not get peer credentials: %s\n"),
2820  strerror_r(errno, sebuf, sizeof(sebuf)));
2821  goto error_return;
2822  }
2823 
2824 #ifndef WIN32
2825  passerr = pqGetpwuid(uid, &pass_buf, pwdbuf, sizeof(pwdbuf), &pass);
2826  if (pass == NULL)
2827  {
2828  if (passerr != 0)
2830  libpq_gettext("could not look up local user ID %d: %s\n"),
2831  (int) uid,
2832  strerror_r(passerr, sebuf, sizeof(sebuf)));
2833  else
2835  libpq_gettext("local user with ID %d does not exist\n"),
2836  (int) uid);
2837  goto error_return;
2838  }
2839 
2840  if (strcmp(pass->pw_name, conn->requirepeer) != 0)
2841  {
2843  libpq_gettext("requirepeer specifies \"%s\", but actual peer user name is \"%s\"\n"),
2844  conn->requirepeer, pass->pw_name);
2845  goto error_return;
2846  }
2847 #else /* WIN32 */
2848  /* should have failed with ENOSYS above */
2849  Assert(false);
2850 #endif /* WIN32 */
2851  }
2852 
2853  if (IS_AF_UNIX(conn->raddr.addr.ss_family))
2854  {
2855  /* Don't request SSL or GSSAPI over Unix sockets */
2856 #ifdef USE_SSL
2857  conn->allow_ssl_try = false;
2858 #endif
2859 #ifdef ENABLE_GSS
2860  conn->try_gss = false;
2861 #endif
2862  }
2863 
2864 #ifdef ENABLE_GSS
2865 
2866  /*
2867  * If GSSAPI encryption is enabled, then call
2868  * pg_GSS_have_cred_cache() which will return true if we can
2869  * acquire credentials (and give us a handle to use in
2870  * conn->gcred), and then send a packet to the server asking
2871  * for GSSAPI Encryption (and skip past SSL negotiation and
2872  * regular startup below).
2873  */
2874  if (conn->try_gss && !conn->gctx)
2875  conn->try_gss = pg_GSS_have_cred_cache(&conn->gcred);
2876  if (conn->try_gss && !conn->gctx)
2877  {
2879 
2880  if (pqPacketSend(conn, 0, &pv, sizeof(pv)) != STATUS_OK)
2881  {
2883  libpq_gettext("could not send GSSAPI negotiation packet: %s\n"),
2884  SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
2885  goto error_return;
2886  }
2887 
2888  /* Ok, wait for response */
2890  return PGRES_POLLING_READING;
2891  }
2892  else if (!conn->gctx && conn->gssencmode[0] == 'r')
2893  {
2895  libpq_gettext("GSSAPI encryption required but was impossible (possibly no credential cache, no server support, or using a local socket)\n"));
2896  goto error_return;
2897  }
2898 #endif
2899 
2900 #ifdef USE_SSL
2901 
2902  /*
2903  * If SSL is enabled and we haven't already got it running,
2904  * request it instead of sending the startup message.
2905  */
2906  if (conn->allow_ssl_try && !conn->wait_ssl_try &&
2907  !conn->ssl_in_use)
2908  {
2909  ProtocolVersion pv;
2910 
2911  /*
2912  * Send the SSL request packet.
2913  *
2914  * Theoretically, this could block, but it really
2915  * shouldn't since we only got here if the socket is
2916  * write-ready.
2917  */
2919  if (pqPacketSend(conn, 0, &pv, sizeof(pv)) != STATUS_OK)
2920  {
2922  libpq_gettext("could not send SSL negotiation packet: %s\n"),
2923  SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
2924  goto error_return;
2925  }
2926  /* Ok, wait for response */
2928  return PGRES_POLLING_READING;
2929  }
2930 #endif /* USE_SSL */
2931 
2932  /*
2933  * Build the startup packet.
2934  */
2935  if (PG_PROTOCOL_MAJOR(conn->pversion) >= 3)
2936  startpacket = pqBuildStartupPacket3(conn, &packetlen,
2938  else
2939  startpacket = pqBuildStartupPacket2(conn, &packetlen,
2941  if (!startpacket)
2942  {
2943  /*
2944  * will not appendbuffer here, since it's likely to also
2945  * run out of memory
2946  */
2948  libpq_gettext("out of memory\n"));
2949  goto error_return;
2950  }
2951 
2952  /*
2953  * Send the startup packet.
2954  *
2955  * Theoretically, this could block, but it really shouldn't
2956  * since we only got here if the socket is write-ready.
2957  */
2958  if (pqPacketSend(conn, 0, startpacket, packetlen) != STATUS_OK)
2959  {
2961  libpq_gettext("could not send startup packet: %s\n"),
2962  SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
2963  free(startpacket);
2964  goto error_return;
2965  }
2966 
2967  free(startpacket);
2968 
2970  return PGRES_POLLING_READING;
2971  }
2972 
2973  /*
2974  * Handle SSL negotiation: wait for postmaster messages and
2975  * respond as necessary.
2976  */
2978  {
2979 #ifdef USE_SSL
2980  PostgresPollingStatusType pollres;
2981 
2982  /*
2983  * On first time through, get the postmaster's response to our
2984  * SSL negotiation packet.
2985  */
2986  if (!conn->ssl_in_use)
2987  {
2988  /*
2989  * We use pqReadData here since it has the logic to
2990  * distinguish no-data-yet from connection closure. Since
2991  * conn->ssl isn't set, a plain recv() will occur.
2992  */
2993  char SSLok;
2994  int rdresult;
2995 
2996  rdresult = pqReadData(conn);
2997  if (rdresult < 0)
2998  {
2999  /* errorMessage is already filled in */
3000  goto error_return;
3001  }
3002  if (rdresult == 0)
3003  {
3004  /* caller failed to wait for data */
3005  return PGRES_POLLING_READING;
3006  }
3007  if (pqGetc(&SSLok, conn) < 0)
3008  {
3009  /* should not happen really */
3010  return PGRES_POLLING_READING;
3011  }
3012  if (SSLok == 'S')
3013  {
3014  /* mark byte consumed */
3015  conn->inStart = conn->inCursor;
3016  /* Set up global SSL state if required */
3017  if (pqsecure_initialize(conn) != 0)
3018  goto error_return;
3019  }
3020  else if (SSLok == 'N')
3021  {
3022  /* mark byte consumed */
3023  conn->inStart = conn->inCursor;
3024  /* OK to do without SSL? */
3025  if (conn->sslmode[0] == 'r' || /* "require" */
3026  conn->sslmode[0] == 'v') /* "verify-ca" or
3027  * "verify-full" */
3028  {
3029  /* Require SSL, but server does not want it */
3031  libpq_gettext("server does not support SSL, but SSL was required\n"));
3032  goto error_return;
3033  }
3034  /* Otherwise, proceed with normal startup */
3035  conn->allow_ssl_try = false;
3036  conn->status = CONNECTION_MADE;
3037  return PGRES_POLLING_WRITING;
3038  }
3039  else if (SSLok == 'E')
3040  {
3041  /*
3042  * Server failure of some sort, such as failure to
3043  * fork a backend process. We need to process and
3044  * report the error message, which might be formatted
3045  * according to either protocol 2 or protocol 3.
3046  * Rather than duplicate the code for that, we flip
3047  * into AWAITING_RESPONSE state and let the code there
3048  * deal with it. Note we have *not* consumed the "E"
3049  * byte here.
3050  */
3052  goto keep_going;
3053  }
3054  else
3055  {
3057  libpq_gettext("received invalid response to SSL negotiation: %c\n"),
3058  SSLok);
3059  goto error_return;
3060  }
3061  }
3062 
3063  /*
3064  * Begin or continue the SSL negotiation process.
3065  */
3066  pollres = pqsecure_open_client(conn);
3067  if (pollres == PGRES_POLLING_OK)
3068  {
3069  /* SSL handshake done, ready to send startup packet */
3070  conn->status = CONNECTION_MADE;
3071  return PGRES_POLLING_WRITING;
3072  }
3073  if (pollres == PGRES_POLLING_FAILED)
3074  {
3075  /*
3076  * Failed ... if sslmode is "prefer" then do a non-SSL
3077  * retry
3078  */
3079  if (conn->sslmode[0] == 'p' /* "prefer" */
3080  && conn->allow_ssl_try /* redundant? */
3081  && !conn->wait_ssl_try) /* redundant? */
3082  {
3083  /* only retry once */
3084  conn->allow_ssl_try = false;
3085  need_new_connection = true;
3086  goto keep_going;
3087  }
3088  /* Else it's a hard failure */
3089  goto error_return;
3090  }
3091  /* Else, return POLLING_READING or POLLING_WRITING status */
3092  return pollres;
3093 #else /* !USE_SSL */
3094  /* can't get here */
3095  goto error_return;
3096 #endif /* USE_SSL */
3097  }
3098 
3100  {
3101 #ifdef ENABLE_GSS
3102  PostgresPollingStatusType pollres;
3103 
3104  /*
3105  * If we haven't yet, get the postmaster's response to our
3106  * negotiation packet
3107  */
3108  if (conn->try_gss && !conn->gctx)
3109  {
3110  char gss_ok;
3111  int rdresult = pqReadData(conn);
3112 
3113  if (rdresult < 0)
3114  /* pqReadData fills in error message */
3115  goto error_return;
3116  else if (rdresult == 0)
3117  /* caller failed to wait for data */
3118  return PGRES_POLLING_READING;
3119  if (pqGetc(&gss_ok, conn) < 0)
3120  /* shouldn't happen... */
3121  return PGRES_POLLING_READING;
3122 
3123  if (gss_ok == 'E')
3124  {
3125  /*
3126  * Server failure of some sort. Assume it's a
3127  * protocol version support failure, and let's see if
3128  * we can't recover (if it's not, we'll get a better
3129  * error message on retry). Server gets fussy if we
3130  * don't hang up the socket, though.
3131  */
3132  conn->try_gss = false;
3133  pqDropConnection(conn, true);
3134  conn->status = CONNECTION_NEEDED;
3135  goto keep_going;
3136  }
3137 
3138  /* mark byte consumed */
3139  conn->inStart = conn->inCursor;
3140 
3141  if (gss_ok == 'N')
3142  {
3143  /* Server doesn't want GSSAPI; fall back if we can */
3144  if (conn->gssencmode[0] == 'r')
3145  {
3147  libpq_gettext("server doesn't support GSSAPI encryption, but it was required\n"));
3148  goto error_return;
3149  }
3150 
3151  conn->try_gss = false;
3152  conn->status = CONNECTION_MADE;
3153  return PGRES_POLLING_WRITING;
3154  }
3155  else if (gss_ok != 'G')
3156  {
3158  libpq_gettext("received invalid response to GSSAPI negotiation: %c\n"),
3159  gss_ok);
3160  goto error_return;
3161  }
3162  }
3163 
3164  /* Begin or continue GSSAPI negotiation */
3165  pollres = pqsecure_open_gss(conn);
3166  if (pollres == PGRES_POLLING_OK)
3167  {
3168  /* All set for startup packet */
3169  conn->status = CONNECTION_MADE;
3170  return PGRES_POLLING_WRITING;
3171  }
3172  else if (pollres == PGRES_POLLING_FAILED &&
3173  conn->gssencmode[0] == 'p')
3174  {
3175  /*
3176  * We failed, but we can retry on "prefer". Have to drop
3177  * the current connection to do so, though.
3178  */
3179  conn->try_gss = false;
3180  pqDropConnection(conn, true);
3181  conn->status = CONNECTION_NEEDED;
3182  goto keep_going;
3183  }
3184  return pollres;
3185 #else /* !ENABLE_GSS */
3186  /* unreachable */
3187  goto error_return;
3188 #endif /* ENABLE_GSS */
3189  }
3190 
3191  /*
3192  * Handle authentication exchange: wait for postmaster messages
3193  * and respond as necessary.
3194  */
3196  {
3197  char beresp;
3198  int msgLength;
3199  int avail;
3200  AuthRequest areq;
3201  int res;
3202 
3203  /*
3204  * Scan the message from current point (note that if we find
3205  * the message is incomplete, we will return without advancing
3206  * inStart, and resume here next time).
3207  */
3208  conn->inCursor = conn->inStart;
3209 
3210  /* Read type byte */
3211  if (pqGetc(&beresp, conn))
3212  {
3213  /* We'll come back when there is more data */
3214  return PGRES_POLLING_READING;
3215  }
3216 
3217  /*
3218  * Validate message type: we expect only an authentication
3219  * request or an error here. Anything else probably means
3220  * it's not Postgres on the other end at all.
3221  */
3222  if (!(beresp == 'R' || beresp == 'E'))
3223  {
3225  libpq_gettext("expected authentication request from server, but received %c\n"),
3226  beresp);
3227  goto error_return;
3228  }
3229 
3230  if (PG_PROTOCOL_MAJOR(conn->pversion) >= 3)
3231  {
3232  /* Read message length word */
3233  if (pqGetInt(&msgLength, 4, conn))
3234  {
3235  /* We'll come back when there is more data */
3236  return PGRES_POLLING_READING;
3237  }
3238  }
3239  else
3240  {
3241  /* Set phony message length to disable checks below */
3242  msgLength = 8;
3243  }
3244 
3245  /*
3246  * Try to validate message length before using it.
3247  * Authentication requests can't be very large, although GSS
3248  * auth requests may not be that small. Errors can be a
3249  * little larger, but not huge. If we see a large apparent
3250  * length in an error, it means we're really talking to a
3251  * pre-3.0-protocol server; cope.
3252  */
3253  if (beresp == 'R' && (msgLength < 8 || msgLength > 2000))
3254  {
3256  libpq_gettext("expected authentication request from server, but received %c\n"),
3257  beresp);
3258  goto error_return;
3259  }
3260 
3261  if (beresp == 'E' && (msgLength < 8 || msgLength > 30000))
3262  {
3263  /* Handle error from a pre-3.0 server */
3264  conn->inCursor = conn->inStart + 1; /* reread data */
3265  if (pqGets_append(&conn->errorMessage, conn))
3266  {
3267  /* We'll come back when there is more data */
3268  return PGRES_POLLING_READING;
3269  }
3270  /* OK, we read the message; mark data consumed */
3271  conn->inStart = conn->inCursor;
3272 
3273  /*
3274  * The postmaster typically won't end its message with a
3275  * newline, so add one to conform to libpq conventions.
3276  */
3277  appendPQExpBufferChar(&conn->errorMessage, '\n');
3278 
3279  /*
3280  * If we tried to open the connection in 3.0 protocol,
3281  * fall back to 2.0 protocol.
3282  */
3283  if (PG_PROTOCOL_MAJOR(conn->pversion) >= 3)
3284  {
3285  conn->pversion = PG_PROTOCOL(2, 0);
3286  need_new_connection = true;
3287  goto keep_going;
3288  }
3289 
3290  goto error_return;
3291  }
3292 
3293  /*
3294  * Can't process if message body isn't all here yet.
3295  *
3296  * (In protocol 2.0 case, we are assuming messages carry at
3297  * least 4 bytes of data.)
3298  */
3299  msgLength -= 4;
3300  avail = conn->inEnd - conn->inCursor;
3301  if (avail < msgLength)
3302  {
3303  /*
3304  * Before returning, try to enlarge the input buffer if
3305  * needed to hold the whole message; see notes in
3306  * pqParseInput3.
3307  */
3308  if (pqCheckInBufferSpace(conn->inCursor + (size_t) msgLength,
3309  conn))
3310  goto error_return;
3311  /* We'll come back when there is more data */
3312  return PGRES_POLLING_READING;
3313  }
3314 
3315  /* Handle errors. */
3316  if (beresp == 'E')
3317  {
3318  if (PG_PROTOCOL_MAJOR(conn->pversion) >= 3)
3319  {
3320  if (pqGetErrorNotice3(conn, true))
3321  {
3322  /* We'll come back when there is more data */
3323  return PGRES_POLLING_READING;
3324  }
3325  }
3326  else
3327  {
3328  if (pqGets_append(&conn->errorMessage, conn))
3329  {
3330  /* We'll come back when there is more data */
3331  return PGRES_POLLING_READING;
3332  }
3333  }
3334  /* OK, we read the message; mark data consumed */
3335  conn->inStart = conn->inCursor;
3336 
3337  /* Check to see if we should mention pgpassfile */
3338  pgpassfileWarning(conn);
3339 
3340 #ifdef ENABLE_GSS
3341 
3342  /*
3343  * If gssencmode is "prefer" and we're using GSSAPI, retry
3344  * without it.
3345  */
3346  if (conn->gssenc && conn->gssencmode[0] == 'p')
3347  {
3348  OM_uint32 minor;
3349 
3350  /* postmaster expects us to drop the connection */
3351  conn->try_gss = false;
3352  conn->gssenc = false;
3353  gss_delete_sec_context(&minor, &conn->gctx, NULL);
3354  pqDropConnection(conn, true);
3355  conn->status = CONNECTION_NEEDED;
3356  goto keep_going;
3357  }
3358 #endif
3359 
3360 #ifdef USE_SSL
3361 
3362  /*
3363  * if sslmode is "allow" and we haven't tried an SSL
3364  * connection already, then retry with an SSL connection
3365  */
3366  if (conn->sslmode[0] == 'a' /* "allow" */
3367  && !conn->ssl_in_use
3368  && conn->allow_ssl_try
3369  && conn->wait_ssl_try)
3370  {
3371  /* only retry once */
3372  conn->wait_ssl_try = false;
3373  need_new_connection = true;
3374  goto keep_going;
3375  }
3376 
3377  /*
3378  * if sslmode is "prefer" and we're in an SSL connection,
3379  * then do a non-SSL retry
3380  */
3381  if (conn->sslmode[0] == 'p' /* "prefer" */
3382  && conn->ssl_in_use
3383  && conn->allow_ssl_try /* redundant? */
3384  && !conn->wait_ssl_try) /* redundant? */
3385  {
3386  /* only retry once */
3387  conn->allow_ssl_try = false;
3388  need_new_connection = true;
3389  goto keep_going;
3390  }
3391 #endif
3392 
3393  goto error_return;
3394  }
3395 
3396  /* It is an authentication request. */
3397  conn->auth_req_received = true;
3398 
3399  /* Get the type of request. */
3400  if (pqGetInt((int *) &areq, 4, conn))
3401  {
3402  /* We'll come back when there are more data */
3403  return PGRES_POLLING_READING;
3404  }
3405  msgLength -= 4;
3406 
3407  /*
3408  * Ensure the password salt is in the input buffer, if it's an
3409  * MD5 request. All the other authentication methods that
3410  * contain extra data in the authentication request are only
3411  * supported in protocol version 3, in which case we already
3412  * read the whole message above.
3413  */
3414  if (areq == AUTH_REQ_MD5 && PG_PROTOCOL_MAJOR(conn->pversion) < 3)
3415  {
3416  msgLength += 4;
3417 
3418  avail = conn->inEnd - conn->inCursor;
3419  if (avail < 4)
3420  {
3421  /*
3422  * Before returning, try to enlarge the input buffer
3423  * if needed to hold the whole message; see notes in
3424  * pqParseInput3.
3425  */
3426  if (pqCheckInBufferSpace(conn->inCursor + (size_t) 4,
3427  conn))
3428  goto error_return;
3429  /* We'll come back when there is more data */
3430  return PGRES_POLLING_READING;
3431  }
3432  }
3433 
3434  /*
3435  * Process the rest of the authentication request message, and
3436  * respond to it if necessary.
3437  *
3438  * Note that conn->pghost must be non-NULL if we are going to
3439  * avoid the Kerberos code doing a hostname look-up.
3440  */
3441  res = pg_fe_sendauth(areq, msgLength, conn);
3442  conn->errorMessage.len = strlen(conn->errorMessage.data);
3443 
3444  /* OK, we have processed the message; mark data consumed */
3445  conn->inStart = conn->inCursor;
3446 
3447  if (res != STATUS_OK)
3448  goto error_return;
3449 
3450  /*
3451  * Just make sure that any data sent by pg_fe_sendauth is
3452  * flushed out. Although this theoretically could block, it
3453  * really shouldn't since we don't send large auth responses.
3454  */
3455  if (pqFlush(conn))
3456  goto error_return;
3457 
3458  if (areq == AUTH_REQ_OK)
3459  {
3460  /* We are done with authentication exchange */
3461  conn->status = CONNECTION_AUTH_OK;
3462 
3463  /*
3464  * Set asyncStatus so that PQgetResult will think that
3465  * what comes back next is the result of a query. See
3466  * below.
3467  */
3468  conn->asyncStatus = PGASYNC_BUSY;
3469  }
3470 
3471  /* Look to see if we have more data yet. */
3472  goto keep_going;
3473  }
3474 
3475  case CONNECTION_AUTH_OK:
3476  {
3477  /*
3478  * Now we expect to hear from the backend. A ReadyForQuery
3479  * message indicates that startup is successful, but we might
3480  * also get an Error message indicating failure. (Notice
3481  * messages indicating nonfatal warnings are also allowed by
3482  * the protocol, as are ParameterStatus and BackendKeyData
3483  * messages.) Easiest way to handle this is to let
3484  * PQgetResult() read the messages. We just have to fake it
3485  * out about the state of the connection, by setting
3486  * asyncStatus = PGASYNC_BUSY (done above).
3487  */
3488 
3489  if (PQisBusy(conn))
3490  return PGRES_POLLING_READING;
3491 
3492  res = PQgetResult(conn);
3493 
3494  /*
3495  * NULL return indicating we have gone to IDLE state is
3496  * expected
3497  */
3498  if (res)
3499  {
3500  if (res->resultStatus != PGRES_FATAL_ERROR)
3502  libpq_gettext("unexpected message from server during startup\n"));
3503  else if (conn->send_appname &&
3504  (conn->appname || conn->fbappname))
3505  {
3506  /*
3507  * If we tried to send application_name, check to see
3508  * if the error is about that --- pre-9.0 servers will
3509  * reject it at this stage of the process. If so,
3510  * close the connection and retry without sending
3511  * application_name. We could possibly get a false
3512  * SQLSTATE match here and retry uselessly, but there
3513  * seems no great harm in that; we'll just get the
3514  * same error again if it's unrelated.
3515  */
3516  const char *sqlstate;
3517 
3518  sqlstate = PQresultErrorField(res, PG_DIAG_SQLSTATE);
3519  if (sqlstate &&
3520  strcmp(sqlstate, ERRCODE_APPNAME_UNKNOWN) == 0)
3521  {
3522  PQclear(res);
3523  conn->send_appname = false;
3524  need_new_connection = true;
3525  goto keep_going;
3526  }
3527  }
3528 
3529  /*
3530  * if the resultStatus is FATAL, then conn->errorMessage
3531  * already has a copy of the error; needn't copy it back.
3532  * But add a newline if it's not there already, since
3533  * postmaster error messages may not have one.
3534  */
3535  if (conn->errorMessage.len <= 0 ||
3536  conn->errorMessage.data[conn->errorMessage.len - 1] != '\n')
3537  appendPQExpBufferChar(&conn->errorMessage, '\n');
3538  PQclear(res);
3539  goto error_return;
3540  }
3541 
3542  /* Fire up post-connection housekeeping if needed */
3543  if (PG_PROTOCOL_MAJOR(conn->pversion) < 3)
3544  {
3545  conn->status = CONNECTION_SETENV;
3547  conn->next_eo = EnvironmentOptions;
3548  return PGRES_POLLING_WRITING;
3549  }
3550 
3551  /* Almost there now ... */
3553  goto keep_going;
3554  }
3555 
3557  {
3558  /*
3559  * If a read-write connection is required, see if we have one.
3560  *
3561  * Servers before 7.4 lack the transaction_read_only GUC, but
3562  * by the same token they don't have any read-only mode, so we
3563  * may just skip the test in that case.
3564  */
3565  if (conn->sversion >= 70400 &&
3566  conn->target_session_attrs != NULL &&
3567  strcmp(conn->target_session_attrs, "read-write") == 0)
3568  {
3569  /*
3570  * Save existing error messages across the PQsendQuery
3571  * attempt. This is necessary because PQsendQuery is
3572  * going to reset conn->errorMessage, so we would lose
3573  * error messages related to previous hosts we have tried
3574  * and failed to connect to.
3575  */
3576  if (!saveErrorMessage(conn, &savedMessage))
3577  goto error_return;
3578 
3579  conn->status = CONNECTION_OK;
3580  if (!PQsendQuery(conn,
3581  "SHOW transaction_read_only"))
3582  {
3583  restoreErrorMessage(conn, &savedMessage);
3584  goto error_return;
3585  }
3587  restoreErrorMessage(conn, &savedMessage);
3588  return PGRES_POLLING_READING;
3589  }
3590 
3591  /* We can release the address list now. */
3592  release_conn_addrinfo(conn);
3593 
3594  /* We are open for business! */
3595  conn->status = CONNECTION_OK;
3596  return PGRES_POLLING_OK;
3597  }
3598 
3599  case CONNECTION_SETENV:
3600  {
3601  /*
3602  * Do post-connection housekeeping (only needed in protocol
3603  * 2.0).
3604  *
3605  * We pretend that the connection is OK for the duration of
3606  * these queries.
3607  */
3608  conn->status = CONNECTION_OK;
3609 
3610  switch (pqSetenvPoll(conn))
3611  {
3612  case PGRES_POLLING_OK: /* Success */
3613  break;
3614 
3615  case PGRES_POLLING_READING: /* Still going */
3616  conn->status = CONNECTION_SETENV;
3617  return PGRES_POLLING_READING;
3618 
3619  case PGRES_POLLING_WRITING: /* Still going */
3620  conn->status = CONNECTION_SETENV;
3621  return PGRES_POLLING_WRITING;
3622 
3623  default:
3624  goto error_return;
3625  }
3626 
3627  /* Almost there now ... */
3629  goto keep_going;
3630  }
3631 
3632  case CONNECTION_CONSUME:
3633  {
3634  conn->status = CONNECTION_OK;
3635  if (!PQconsumeInput(conn))
3636  goto error_return;
3637 
3638  if (PQisBusy(conn))
3639  {
3640  conn->status = CONNECTION_CONSUME;
3641  return PGRES_POLLING_READING;
3642  }
3643 
3644  /*
3645  * Call PQgetResult() again to consume NULL result.
3646  */
3647  res = PQgetResult(conn);
3648  if (res != NULL)
3649  {
3650  PQclear(res);
3651  conn->status = CONNECTION_CONSUME;
3652  goto keep_going;
3653  }
3654 
3655  /* We can release the address list now. */
3656  release_conn_addrinfo(conn);
3657 
3658  /* We are open for business! */
3659  conn->status = CONNECTION_OK;
3660  return PGRES_POLLING_OK;
3661  }
3663  {
3664  const char *displayed_host;
3665  const char *displayed_port;
3666 
3667  if (!saveErrorMessage(conn, &savedMessage))
3668  goto error_return;
3669 
3670  conn->status = CONNECTION_OK;
3671  if (!PQconsumeInput(conn))
3672  {
3673  restoreErrorMessage(conn, &savedMessage);
3674  goto error_return;
3675  }
3676 
3677  if (PQisBusy(conn))
3678  {
3680  restoreErrorMessage(conn, &savedMessage);
3681  return PGRES_POLLING_READING;
3682  }
3683 
3684  res = PQgetResult(conn);
3685  if (res && (PQresultStatus(res) == PGRES_TUPLES_OK) &&
3686  PQntuples(res) == 1)
3687  {
3688  char *val;
3689 
3690  val = PQgetvalue(res, 0, 0);
3691  if (strncmp(val, "on", 2) == 0)
3692  {
3693  /* Not writable; fail this connection. */
3694  PQclear(res);
3695  restoreErrorMessage(conn, &savedMessage);
3696 
3697  /* Append error report to conn->errorMessage. */
3698  if (conn->connhost[conn->whichhost].type == CHT_HOST_ADDRESS)
3699  displayed_host = conn->connhost[conn->whichhost].hostaddr;
3700  else
3701  displayed_host = conn->connhost[conn->whichhost].host;
3702  displayed_port = conn->connhost[conn->whichhost].port;
3703  if (displayed_port == NULL || displayed_port[0] == '\0')
3704  displayed_port = DEF_PGPORT_STR;
3705 
3707  libpq_gettext("could not make a writable "
3708  "connection to server "
3709  "\"%s:%s\"\n"),
3710  displayed_host, displayed_port);
3711 
3712  /* Close connection politely. */
3713  conn->status = CONNECTION_OK;
3714  sendTerminateConn(conn);
3715 
3716  /*
3717  * Try next host if any, but we don't want to consider
3718  * additional addresses for this host.
3719  */
3720  conn->try_next_host = true;
3721  goto keep_going;
3722  }
3723 
3724  /* Session is read-write, so we're good. */
3725  PQclear(res);
3726  termPQExpBuffer(&savedMessage);
3727 
3728  /*
3729  * Finish reading any remaining messages before being
3730  * considered as ready.
3731  */
3732  conn->status = CONNECTION_CONSUME;
3733  goto keep_going;
3734  }
3735 
3736  /*
3737  * Something went wrong with "SHOW transaction_read_only". We
3738  * should try next addresses.
3739  */
3740  if (res)
3741  PQclear(res);
3742  restoreErrorMessage(conn, &savedMessage);
3743 
3744  /* Append error report to conn->errorMessage. */
3745  if (conn->connhost[conn->whichhost].type == CHT_HOST_ADDRESS)
3746  displayed_host = conn->connhost[conn->whichhost].hostaddr;
3747  else
3748  displayed_host = conn->connhost[conn->whichhost].host;
3749  displayed_port = conn->connhost[conn->whichhost].port;
3750  if (displayed_port == NULL || displayed_port[0] == '\0')
3751  displayed_port = DEF_PGPORT_STR;
3753  libpq_gettext("test \"SHOW transaction_read_only\" failed "
3754  "on server \"%s:%s\"\n"),
3755  displayed_host, displayed_port);
3756 
3757  /* Close connection politely. */
3758  conn->status = CONNECTION_OK;
3759  sendTerminateConn(conn);
3760 
3761  /* Try next address */
3762  conn->try_next_addr = true;
3763  goto keep_going;
3764  }
3765 
3766  default:
3768  libpq_gettext("invalid connection state %d, "
3769  "probably indicative of memory corruption\n"),
3770  conn->status);
3771  goto error_return;
3772  }
3773 
3774  /* Unreachable */
3775 
3776 error_return:
3777 
3778  /*
3779  * We used to close the socket at this point, but that makes it awkward
3780  * for those above us if they wish to remove this socket from their own
3781  * records (an fd_set for example). We'll just have this socket closed
3782  * when PQfinish is called (which is compulsory even after an error, since
3783  * the connection structure must be freed).
3784  */
3785  conn->status = CONNECTION_BAD;
3786  return PGRES_POLLING_FAILED;
3787 }
int pqFlush(PGconn *conn)
Definition: fe-misc.c:996
bool sigpipe_flag
Definition: libpq-int.h:419
char * gssencmode
Definition: libpq-int.h:366
#define NEGOTIATE_GSS_CODE
Definition: pqcomm.h:206
int inEnd
Definition: libpq-int.h:448
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
#define UNIXSOCK_PATH(path, port, sockdir)
Definition: pqcomm.h:70
int inStart
Definition: libpq-int.h:446
static void release_conn_addrinfo(PGconn *conn)
Definition: fe-connect.c:4108
struct addrinfo * addr_cur
Definition: libpq-int.h:427
char * PQgetvalue(const PGresult *res, int tup_num, int field_num)
Definition: fe-exec.c:3163
static int connectNoDelay(PGconn *conn)
Definition: fe-connect.c:1601
int pg_fe_sendauth(AuthRequest areq, int payloadlen, PGconn *conn)
Definition: fe-auth.c:863
void termPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:131
int getpeereid(int sock, uid_t *uid, gid_t *gid)
Definition: getpeereid.c:35
#define UNIXSOCK_PATH_BUFLEN
Definition: pqcomm.h:86
bool sigpipe_so
Definition: libpq-int.h:418
char * pqBuildStartupPacket2(PGconn *conn, int *packetlen, const PQEnvironmentOption *options)
void appendPQExpBufferStr(PQExpBuffer str, const char *data)
Definition: pqexpbuffer.c:369
#define AUTH_REQ_OK
Definition: pqcomm.h:165
int pqCheckInBufferSpace(size_t bytes_needed, PGconn *conn)
Definition: fe-misc.c:407
#define PG_STRERROR_R_BUFLEN
Definition: port.h:211
static void pgpassfileWarning(PGconn *conn)
Definition: fe-connect.c:7105
char * requirepeer
Definition: libpq-int.h:365
char * host
Definition: libpq-int.h:312
struct sockaddr_storage addr
Definition: pqcomm.h:64
static int setKeepalivesIdle(PGconn *conn)
Definition: fe-connect.c:1799
#define MemSet(start, val, len)
Definition: c.h:971
int pqGetInt(int *result, size_t bytes, PGconn *conn)
Definition: fe-misc.c:266
static bool saveErrorMessage(PGconn *conn, PQExpBuffer savedMessage)
Definition: fe-connect.c:2191
#define connect(s, name, namelen)
Definition: win32_port.h:435
static void sendTerminateConn(PGconn *conn)
Definition: fe-connect.c:4123
void pqDropConnection(PGconn *conn, bool flushInput)
Definition: fe-connect.c:458
#define PG_PROTOCOL_MAJOR(v)
Definition: pqcomm.h:104
#define AI_NUMERICHOST
Definition: getaddrinfo.h:73
int PQntuples(const PGresult *res)
Definition: fe-exec.c:2769
#define PG_DIAG_SQLSTATE
Definition: postgres_ext.h:57
uint32 AuthRequest
Definition: pqcomm.h:179
struct addrinfo * addrlist
Definition: libpq-int.h:426
#define AUTH_REQ_MD5
Definition: pqcomm.h:170
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
pg_conn_host_type type
Definition: libpq-int.h:311
PostgresPollingStatusType pqSetenvPoll(PGconn *conn)
Definition: fe-protocol2.c:47
static void getHostaddr(PGconn *conn, char *host_addr, int host_addr_len)
Definition: fe-connect.c:1628
#define gai_strerror
Definition: getaddrinfo.h:146
int pg_getaddrinfo_all(const char *hostname, const char *servname, const struct addrinfo *hintp, struct addrinfo **result)
Definition: ip.c:57
static bool parse_int_param(const char *value, int *result, PGconn *conn, const char *context)
Definition: fe-connect.c:1753
int sversion
Definition: libpq-int.h:415
#define SOCK_STRERROR
Definition: libpq-int.h:819
int PQsendQuery(PGconn *conn, const char *query)
Definition: fe-exec.c:1234
PGAsyncStatusType asyncStatus
Definition: libpq-int.h:389
#define NI_MAXHOST
Definition: getaddrinfo.h:88
PGSetenvStatusType setenv_state
Definition: libpq-int.h:429
static int setTCPUserTimeout(PGconn *conn)
Definition: fe-connect.c:1954
static int useKeepalives(PGconn *conn)
Definition: fe-connect.c:1734
#define pg_hton32(x)
Definition: pg_bswap.h:121
#define IS_AF_UNIX(fam)
Definition: ip.h:24
#define SOCK_ERRNO
Definition: libpq-int.h:818
int pqGetErrorNotice3(PGconn *conn, bool isError)
Definition: fe-protocol3.c:871
#define MAXPGPATH
void appendPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:267
bool try_next_host
Definition: libpq-int.h:425
#define ERRCODE_APPNAME_UNKNOWN
Definition: fe-connect.c:91
char * hostaddr
Definition: libpq-int.h:313
pg_conn_host * connhost
Definition: libpq-int.h:406
int pqReadData(PGconn *conn)
Definition: fe-misc.c:628
bool ssl_in_use
Definition: libpq-int.h:472
const PQEnvironmentOption * next_eo
Definition: libpq-int.h:430
static void restoreErrorMessage(PGconn *conn, PQExpBuffer savedMessage)
Definition: fe-connect.c:2213
char * appname
Definition: libpq-int.h:343
PostgresPollingStatusType pqsecure_open_client(PGconn *conn)
Definition: fe-secure.c:183
ACCEPT_TYPE_ARG3 salen
Definition: pqcomm.h:65
char * target_session_attrs
Definition: libpq-int.h:374
static void pqDropServerData(PGconn *conn)
Definition: fe-connect.c:547
#define STATUS_OK
Definition: c.h:1133
pgsocket sock
Definition: libpq-int.h:410
uint32 ProtocolVersion
Definition: pqcomm.h:113
int pqGetc(char *result, PGconn *conn)
Definition: fe-misc.c:93
static void connectFailureMessage(PGconn *conn, int errorno)
Definition: fe-connect.c:1660
#define socket(af, type, protocol)
Definition: win32_port.h:431
SockAddr raddr
Definition: libpq-int.h:413
#define PGINVALID_SOCKET
Definition: port.h:33
int addrlist_family
Definition: libpq-int.h:428
int pqsecure_initialize(PGconn *conn)
Definition: fe-secure.c:168
int PQconsumeInput(PGconn *conn)
Definition: fe-exec.c:1704
void appendPQExpBufferChar(PQExpBuffer str, char ch)
Definition: pqexpbuffer.c:380
char * sslmode
Definition: libpq-int.h:358
PQExpBufferData errorMessage
Definition: libpq-int.h:526
bool try_next_addr
Definition: libpq-int.h:424
void PQclear(PGresult *res)
Definition: fe-exec.c:694
#define free(a)
Definition: header.h:65
char * PQresultErrorField(const PGresult *res, int fieldcode)
Definition: fe-exec.c:2754
#define Assert(condition)
Definition: c.h:738
bool pg_set_noblock(pgsocket sock)
Definition: noblock.c:25
char * connip
Definition: libpq-int.h:407
int PQisBusy(PGconn *conn)
Definition: fe-exec.c:1754
char * pqBuildStartupPacket3(PGconn *conn, int *packetlen, const PQEnvironmentOption *options)
#define strerror_r
Definition: port.h:210
ProtocolVersion pversion
Definition: libpq-int.h:414
ConnStatusType status
Definition: libpq-int.h:388
bool auth_req_received
Definition: libpq-int.h:416
int uid_t
Definition: win32_port.h:230
static int setKeepalivesInterval(PGconn *conn)
Definition: fe-connect.c:1833
SockAddr laddr
Definition: libpq-int.h:412
PostgresPollingStatusType
Definition: libpq-fe.h:74
struct addrinfo * ai_next
Definition: getaddrinfo.h:107
int pqGetpwuid(uid_t uid, struct passwd *resultbuf, char *buffer, size_t buflen, struct passwd **result)
Definition: thread.c:68
bool pg_GSS_have_cred_cache(gss_cred_id_t *cred_out)
int pqPacketSend(PGconn *conn, char pack_type, const void *buf, size_t buf_len)
Definition: fe-connect.c:4530
int gid_t
Definition: win32_port.h:231
#define EINPROGRESS
Definition: win32_port.h:335
char * fbappname
Definition: libpq-int.h:344
#define EWOULDBLOCK
Definition: win32_port.h:329
size_t ai_addrlen
Definition: getaddrinfo.h:104
PostgresPollingStatusType pqsecure_open_gss(PGconn *conn)
void pqClearAsyncResult(PGconn *conn)
Definition: fe-exec.c:749
int inCursor
Definition: libpq-int.h:447
#define EINTR
Definition: win32_port.h:323
char * port
Definition: libpq-int.h:314
#define NEGOTIATE_SSL_CODE
Definition: pqcomm.h:205
int nconnhost
Definition: libpq-int.h:404
#define snprintf
Definition: port.h:193
#define PG_PROTOCOL(m, n)
Definition: pqcomm.h:106
int pqGets_append(PQExpBuffer buf, PGconn *conn)
Definition: fe-misc.c:168
long val
Definition: informix.c:664
PGresult * PQgetResult(PGconn *conn)
Definition: fe-exec.c:1778
struct sockaddr * ai_addr
Definition: getaddrinfo.h:105
static int setKeepalivesCount(PGconn *conn)
Definition: fe-connect.c:1868
int whichhost
Definition: libpq-int.h:405
#define libpq_gettext(x)
Definition: libpq-int.h:805
bool send_appname
Definition: libpq-int.h:431
PGTransactionStatusType xactStatus
Definition: libpq-int.h:390
static const PQEnvironmentOption EnvironmentOptions[]
Definition: fe-connect.c:362
int ai_family
Definition: getaddrinfo.h:101

◆ PQconnectStart()

PGconn* PQconnectStart ( const char *  conninfo)

Definition at line 822 of file fe-connect.c.

References conn, connectDBStart(), CONNECTION_BAD, connectOptions1(), connectOptions2(), makeEmptyPGconn(), and pg_conn::status.

Referenced by PQconnectdb(), and PQping().

823 {
824  PGconn *conn;
825 
826  /*
827  * Allocate memory for the conn structure
828  */
829  conn = makeEmptyPGconn();
830  if (conn == NULL)
831  return NULL;
832 
833  /*
834  * Parse the conninfo string
835  */
836  if (!connectOptions1(conn, conninfo))
837  return conn;
838 
839  /*
840  * Compute derived options
841  */
842  if (!connectOptions2(conn))
843  return conn;
844 
845  /*
846  * Connect to the database
847  */
848  if (!connectDBStart(conn))
849  {
850  /* Just in case we failed to set it in connectDBStart */
851  conn->status = CONNECTION_BAD;
852  }
853 
854  return conn;
855 }
static bool connectOptions1(PGconn *conn, const char *conninfo)
Definition: fe-connect.c:908
PGconn * conn
Definition: streamutil.c:54
static int connectDBStart(PGconn *conn)
Definition: fe-connect.c:1993
static PGconn * makeEmptyPGconn(void)
Definition: fe-connect.c:3862
ConnStatusType status
Definition: libpq-int.h:388
static bool connectOptions2(PGconn *conn)
Definition: fe-connect.c:1006

◆ PQconnectStartParams()

PGconn* PQconnectStartParams ( const char *const *  keywords,
const char *const *  values,
int  expand_dbname 
)

Definition at line 743 of file fe-connect.c.

References conn, connectDBStart(), CONNECTION_BAD, connectOptions2(), conninfo_array_parse(), pg_conn::errorMessage, fillPGconn(), makeEmptyPGconn(), PQconninfoFree(), and pg_conn::status.

Referenced by libpqrcv_connect(), PQconnectdbParams(), and PQpingParams().

746 {
747  PGconn *conn;
748  PQconninfoOption *connOptions;
749 
750  /*
751  * Allocate memory for the conn structure
752  */
753  conn = makeEmptyPGconn();
754  if (conn == NULL)
755  return NULL;
756 
757  /*
758  * Parse the conninfo arrays
759  */
760  connOptions = conninfo_array_parse(keywords, values,
761  &conn->errorMessage,
762  true, expand_dbname);
763  if (connOptions == NULL)
764  {
765  conn->status = CONNECTION_BAD;
766  /* errorMessage is already set */
767  return conn;
768  }
769 
770  /*
771  * Move option values into conn structure
772  */
773  if (!fillPGconn(conn, connOptions))
774  {
775  PQconninfoFree(connOptions);
776  return conn;
777  }
778 
779  /*
780  * Free the option info - all is in conn now
781  */
782  PQconninfoFree(connOptions);
783 
784  /*
785  * Compute derived options
786  */
787  if (!connectOptions2(conn))
788  return conn;
789 
790  /*
791  * Connect to the database
792  */
793  if (!connectDBStart(conn))
794  {
795  /* Just in case we failed to set it in connectDBStart */
796  conn->status = CONNECTION_BAD;
797  }
798 
799  return conn;
800 }
PGconn * conn
Definition: streamutil.c:54
static int connectDBStart(PGconn *conn)
Definition: fe-connect.c:1993
void PQconninfoFree(PQconninfoOption *connOptions)
Definition: fe-connect.c:6519
static PGconn * makeEmptyPGconn(void)
Definition: fe-connect.c:3862
PQExpBufferData errorMessage
Definition: libpq-int.h:526
static bool fillPGconn(PGconn *conn, PQconninfoOption *connOptions)
Definition: fe-connect.c:866
ConnStatusType status
Definition: libpq-int.h:388
static Datum values[MAXATTR]
Definition: bootstrap.c:167
static bool connectOptions2(PGconn *conn)
Definition: fe-connect.c:1006
static PQconninfoOption * conninfo_array_parse(const char *const *keywords, const char *const *values, PQExpBuffer errorMessage, bool use_defaults, int expand_dbname)
Definition: fe-connect.c:5583

◆ PQconninfo()

PQconninfoOption* PQconninfo ( PGconn conn)

Definition at line 6478 of file fe-connect.c.

References conninfo_init(), conninfo_storeval(), _internalPQconninfoOption::connofs, initPQExpBuffer(), _internalPQconninfoOption::keyword, PQExpBufferDataBroken, and termPQExpBuffer().

Referenced by GenerateRecoveryConfig(), has_hostaddr(), and libpqrcv_get_conninfo().

6479 {
6480  PQExpBufferData errorBuf;
6481  PQconninfoOption *connOptions;
6482 
6483  if (conn == NULL)
6484  return NULL;
6485 
6486  /* We don't actually report any errors here, but callees want a buffer */
6487  initPQExpBuffer(&errorBuf);
6488  if (PQExpBufferDataBroken(errorBuf))
6489  return NULL; /* out of memory already :-( */
6490 
6491  connOptions = conninfo_init(&errorBuf);
6492 
6493  if (connOptions != NULL)
6494  {
6496 
6497  for (option = PQconninfoOptions; option->keyword; option++)
6498  {
6499  char **connmember;
6500 
6501  if (option->connofs < 0)
6502  continue;
6503 
6504  connmember = (char **) ((char *) conn + option->connofs);
6505 
6506  if (*connmember)
6507  conninfo_storeval(connOptions, option->keyword, *connmember,
6508  &errorBuf, true, false);
6509  }
6510  }
6511 
6512  termPQExpBuffer(&errorBuf);
6513 
6514  return connOptions;
6515 }
static PQconninfoOption * conninfo_storeval(PQconninfoOption *connOptions, const char *keyword, const char *value, PQExpBuffer errorMessage, bool ignoreMissing, bool uri_decode)
Definition: fe-connect.c:6395
void termPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:131
static PQconninfoOption * conninfo_init(PQExpBuffer errorMessage)
Definition: fe-connect.c:5311
static const internalPQconninfoOption PQconninfoOptions[]
Definition: fe-connect.c:193
#define PQExpBufferDataBroken(buf)
Definition: pqexpbuffer.h:67
void initPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:92

◆ PQconninfoFree()

void PQconninfoFree ( PQconninfoOption connOptions)

Definition at line 6519 of file fe-connect.c.

References free, _PQconninfoOption::keyword, and _PQconninfoOption::val.

Referenced by check_pghost_envvar(), connectDatabase(), connectOptions1(), conninfo_array_parse(), conninfo_parse(), conninfo_uri_parse(), dblink_connstr_check(), GenerateRecoveryConfig(), GetConnection(), has_hostaddr(), libpqrcv_check_conninfo(), libpqrcv_get_conninfo(), PQconndefaults(), and PQconnectStartParams().

6520 {
6522 
6523  if (connOptions == NULL)
6524  return;
6525 
6526  for (option = connOptions; option->keyword != NULL; option++)
6527  {
6528  if (option->val != NULL)
6529  free(option->val);
6530  }
6531  free(connOptions);
6532 }
#define free(a)
Definition: header.h:65

◆ PQconninfoParse()

PQconninfoOption* PQconninfoParse ( const char *  conninfo,
char **  errmsg 
)

Definition at line 5289 of file fe-connect.c.

References PQExpBufferData::data, initPQExpBuffer(), parse_connection_string(), PQExpBufferDataBroken, and termPQExpBuffer().

Referenced by connectDatabase(), dblink_connstr_check(), GetConnection(), libpqrcv_check_conninfo(), and main().

5290 {
5291  PQExpBufferData errorBuf;
5292  PQconninfoOption *connOptions;
5293 
5294  if (errmsg)
5295  *errmsg = NULL; /* default */
5296  initPQExpBuffer(&errorBuf);
5297  if (PQExpBufferDataBroken(errorBuf))
5298  return NULL; /* out of memory already :-( */
5299  connOptions = parse_connection_string(conninfo, &errorBuf, false);
5300  if (connOptions == NULL && errmsg)
5301  *errmsg = errorBuf.data;
5302  else
5303  termPQExpBuffer(&errorBuf);
5304  return connOptions;
5305 }
void termPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:131
static PQconninfoOption * parse_connection_string(const char *conninfo, PQExpBuffer errorMessage, bool use_defaults)
Definition: fe-connect.c:5351
#define PQExpBufferDataBroken(buf)
Definition: pqexpbuffer.h:67
int errmsg(const char *fmt,...)
Definition: elog.c:824
void initPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:92

◆ PQconsumeInput()

int PQconsumeInput ( PGconn conn)

Definition at line 1704 of file fe-exec.c.

References pqFlush(), pqIsnonblocking, and pqReadData().

Referenced by advanceConnectionState(), CopyStreamReceive(), dblink_get_notify(), dblink_is_busy(), ecpg_process_output(), libpqrcv_PQgetResult(), libpqrcv_receive(), main(), ParallelSlotsGetIdle(), pgfdw_get_cleanup_result(), pgfdw_get_result(), PQconnectPoll(), PrintNotifications(), StreamLogicalLog(), and try_complete_step().

1705 {
1706  if (!conn)
1707  return 0;
1708 
1709  /*
1710  * for non-blocking connections try to flush the send-queue, otherwise we
1711  * may never get a response for something that may not have already been
1712  * sent because it's in our write buffer!
1713  */
1714  if (pqIsnonblocking(conn))
1715  {
1716  if (pqFlush(conn) < 0)
1717  return 0;
1718  }
1719 
1720  /*
1721  * Load more data, if available. We do this no matter what state we are
1722  * in, since we are probably getting called because the application wants
1723  * to get rid of a read-select condition. Note that we will NOT block
1724  * waiting for more input.
1725  */
1726  if (pqReadData(conn) < 0)
1727  return 0;
1728 
1729  /* Parsing of the data waits till later. */
1730  return 1;
1731 }
int pqFlush(PGconn *conn)
Definition: fe-misc.c:996
#define pqIsnonblocking(conn)
Definition: libpq-int.h:799
int pqReadData(PGconn *conn)
Definition: fe-misc.c:628

◆ PQcopyResult()

PGresult* PQcopyResult ( const PGresult src,
int  flags 
)

Definition at line 294 of file fe-exec.c.

References pg_result::attDescs, pg_result::client_encoding, pg_result::cmdStatus, PGEventResultCopy::dest, generate_unaccent_rules::dest, dupEvents(), pg_result::events, i, pgresAttValue::len, pg_result::memorySize, pg_result::nEvents, pg_result::noticeHooks, pg_result::ntups, pg_result::numAttributes, PGEvent::passThrough, PG_COPYRES_ATTRS, PG_COPYRES_EVENTS, PG_COPYRES_NOTICEHOOKS, PG_COPYRES_TUPLES, PGEVT_RESULTCOPY, PGRES_TUPLES_OK, PQclear(), PQmakeEmptyPGresult(), PQsetResultAttrs(), PQsetvalue(), PGEvent::proc, PGEvent::resultInitialized, PGEventResultCopy::src, pg_result::tuples, and pgresAttValue::value.

Referenced by pqRowProcessor().

295 {
296  PGresult *dest;
297  int i;
298 
299  if (!src)
300  return NULL;
301 
302  dest = PQmakeEmptyPGresult(NULL, PGRES_TUPLES_OK);
303  if (!dest)
304  return NULL;
305 
306  /* Always copy these over. Is cmdStatus really useful here? */
307  dest->client_encoding = src->client_encoding;
308  strcpy(dest->cmdStatus, src->cmdStatus);
309 
310  /* Wants attrs? */
311  if (flags & (PG_COPYRES_ATTRS | PG_COPYRES_TUPLES))
312  {
313  if (!PQsetResultAttrs(dest, src->numAttributes, src->attDescs))
314  {
315  PQclear(dest);
316  return NULL;
317  }
318  }
319 
320  /* Wants to copy tuples? */
321  if (flags & PG_COPYRES_TUPLES)
322  {
323  int tup,
324  field;
325 
326  for (tup = 0; tup < src->ntups; tup++)
327  {
328  for (field = 0; field < src->numAttributes; field++)
329  {
330  if (!PQsetvalue(dest, tup, field,
331  src->tuples[tup][field].value,
332  src->tuples[tup][field].len))
333  {
334  PQclear(dest);
335  return NULL;
336  }
337  }
338  }
339  }
340 
341  /* Wants to copy notice hooks? */
342  if (flags & PG_COPYRES_NOTICEHOOKS)
343  dest->noticeHooks = src->noticeHooks;
344 
345  /* Wants to copy PGEvents? */
346  if ((flags & PG_COPYRES_EVENTS) && src->nEvents > 0)
347  {
348  dest->events = dupEvents(src->events, src->nEvents,
349  &dest->memorySize);
350  if (!dest->events)
351  {
352  PQclear(dest);
353  return NULL;
354  }
355  dest->nEvents = src->nEvents;
356  }
357 
358  /* Okay, trigger PGEVT_RESULTCOPY event */
359  for (i = 0; i < dest->nEvents; i++)
360  {
361  if (src->events[i].resultInitialized)
362  {
363  PGEventResultCopy evt;
364 
365  evt.src = src;
366  evt.dest = dest;
367  if (!dest->events[i].proc(PGEVT_RESULTCOPY, &evt,
368  dest->events[i].passThrough))
369  {
370  PQclear(dest);
371  return NULL;
372  }
373  dest->events[i].resultInitialized = true;
374  }
375  }
376 
377  return dest;
378 }
size_t memorySize
Definition: libpq-int.h:212
const PGresult * src
Definition: libpq-events.h:60
bool resultInitialized
Definition: libpq-int.h:164
int PQsetvalue(PGresult *res, int tup_num, int field_num, char *value, int len)
Definition: fe-exec.c:431
int nEvents
Definition: libpq-int.h:188
PGresAttDesc * attDescs
Definition: libpq-int.h:171
#define PG_COPYRES_TUPLES
Definition: libpq-fe.h:35
PGresAttValue ** tuples
Definition: libpq-int.h:172
#define PG_COPYRES_EVENTS
Definition: libpq-fe.h:36
PGNoticeHooks noticeHooks
Definition: libpq-int.h:186
#define PG_COPYRES_NOTICEHOOKS
Definition: libpq-fe.h:37
int PQsetResultAttrs(PGresult *res, int numAttributes, PGresAttDesc *attDescs)
Definition: fe-exec.c:229
PGresult * PQmakeEmptyPGresult(PGconn *conn, ExecStatusType status)
Definition: fe-exec.c:141
int numAttributes
Definition: libpq-int.h:170
#define PG_COPYRES_ATTRS
Definition: libpq-fe.h:34
void PQclear(PGresult *res)
Definition: fe-exec.c:694
int ntups
Definition: libpq-int.h:169
PGEventProc proc
Definition: libpq-int.h:160
char cmdStatus[CMDSTATUS_LEN]
Definition: libpq-int.h:178
int i
void * passThrough
Definition: libpq-int.h:162
char * value
Definition: libpq-int.h:138
static PGEvent * dupEvents(PGEvent *events, int count, size_t *memSize)
Definition: fe-exec.c:387
PGEvent * events
Definition: libpq-int.h:187
PGresult * dest
Definition: libpq-events.h:61
int client_encoding
Definition: libpq-int.h:189

◆ PQdb()

◆ PQdefaultSSLKeyPassHook_OpenSSL()

int PQdefaultSSLKeyPassHook_OpenSSL ( char *  buf,
int  size,
PGconn conn 
)

Definition at line 1672 of file fe-secure-openssl.c.

References fprintf, libpq_gettext, and pg_conn::sslpassword.

Referenced by PQssl_passwd_cb().

1673 {
1674  if (conn->sslpassword)
1675  {
1676  if (strlen(conn->sslpassword) + 1 > size)
1677  fprintf(stderr, libpq_gettext("WARNING: sslpassword truncated\n"));
1678  strncpy(buf, conn->sslpassword, size);
1679  buf[size - 1] = '\0';
1680  return strlen(buf);
1681  }
1682  else
1683  {
1684  buf[0] = '\0';
1685  return 0;
1686  }
1687 }
#define fprintf
Definition: port.h:197
static char * buf
Definition: pg_test_fsync.c:67
char * sslpassword
Definition: libpq-int.h:362
#define libpq_gettext(x)
Definition: libpq-int.h:805

◆ PQdescribePortal()

PGresult* PQdescribePortal ( PGconn conn,
const char *  portal 
)

Definition at line 2175 of file fe-exec.c.

References PQexecFinish(), PQexecStart(), and PQsendDescribe().

2176 {
2177  if (!PQexecStart(conn))
2178  return NULL;
2179  if (!PQsendDescribe(conn, 'P', portal))
2180  return NULL;
2181  return PQexecFinish(conn);
2182 }
static int PQsendDescribe(PGconn *conn, char desc_type, const char *desc_target)
Definition: fe-exec.c:2220
static PGresult * PQexecFinish(PGconn *conn)
Definition: fe-exec.c:2095
static bool PQexecStart(PGconn *conn)
Definition: fe-exec.c:2021

◆ PQdescribePrepared()

PGresult* PQdescribePrepared ( PGconn conn,
const char *  stmt 
)

Definition at line 2156 of file fe-exec.c.

References PQexecFinish(), PQexecStart(), and PQsendDescribe().

Referenced by DescribeQuery(), and ECPGdescribe().

2157 {
2158  if (!PQexecStart(conn))
2159  return NULL;
2160  if (!PQsendDescribe(conn, 'S', stmt))
2161  return NULL;
2162  return PQexecFinish(conn);
2163 }
static int PQsendDescribe(PGconn *conn, char desc_type, const char *desc_target)
Definition: fe-exec.c:2220
static PGresult * PQexecFinish(PGconn *conn)
Definition: fe-exec.c:2095
static bool PQexecStart(PGconn *conn)
Definition: fe-exec.c:2021

◆ PQdisplayTuples()

void PQdisplayTuples ( const PGresult res,
FILE *  fp,
int  fillAlign,
const char *  fieldSep,
int  printHeader,
int  quiet 
)

Definition at line 578 of file fe-print.c.

References DEFAULT_FIELD_SEP, fill(), fprintf, free, i, libpq_gettext, malloc, PQfname(), PQgetlength(), PQgetvalue(), PQnfields(), PQntuples(), and generate_unaccent_rules::stdout.

585 {
586 #define DEFAULT_FIELD_SEP " "
587 
588  int i,
589  j;
590  int nFields;
591  int nTuples;
592  int *fLength = NULL;
593 
594  if (fieldSep == NULL)
595  fieldSep = DEFAULT_FIELD_SEP;
596 
597  /* Get some useful info about the results */
598  nFields = PQnfields(res);
599  nTuples = PQntuples(res);
600 
601  if (fp == NULL)
602  fp = stdout;
603 
604  /* Figure the field lengths to align to */
605  /* will be somewhat time consuming for very large results */
606  if (fillAlign)
607  {
608  fLength = (int *) malloc(nFields * sizeof(int));
609  if (!fLength)
610  {
611  fprintf(stderr, libpq_gettext("out of memory\n"));
612  abort();
613  }
614 
615  for (j = 0; j < nFields; j++)
616  {
617  fLength[j] = strlen(PQfname(res, j));
618  for (i = 0; i < nTuples; i++)
619  {
620  int flen = PQgetlength(res, i, j);
621 
622  if (flen > fLength[j])
623  fLength[j] = flen;
624  }
625  }
626  }
627 
628  if (printHeader)
629  {
630  /* first, print out the attribute names */
631  for (i = 0; i < nFields; i++)
632  {
633  fputs(PQfname(res, i), fp);
634  if (fillAlign)
635  fill(strlen(PQfname(res, i)), fLength[i], ' ', fp);
636  fputs(fieldSep, fp);
637  }
638  fprintf(fp, "\n");
639 
640  /* Underline the attribute names */
641  for (i = 0; i < nFields; i++)
642  {
643  if (fillAlign)
644  fill(0, fLength[i], '-', fp);
645  fputs(fieldSep, fp);
646  }
647  fprintf(fp, "\n");
648  }
649 
650  /* next, print out the instances */
651  for (i = 0; i < nTuples; i++)
652  {
653  for (j = 0; j < nFields; j++)
654  {
655  fprintf(fp, "%s", PQgetvalue(res, i, j));
656  if (fillAlign)
657  fill(strlen(PQgetvalue(res, i, j)), fLength[j], ' ', fp);
658  fputs(fieldSep, fp);
659  }
660  fprintf(fp, "\n");
661  }
662 
663  if (!quiet)
664  fprintf(fp, "\nQuery returned %d row%s.\n", PQntuples(res),
665  (PQntuples(res) == 1) ? "" : "s");
666 
667  fflush(fp);
668 
669  if (fLength)
670  free(fLength);
671 }
static void fill(int length, int max, char filler, FILE *fp)
Definition: fe-print.c:761
int PQgetlength(const PGresult *res, int tup_num, int field_num)
Definition: fe-exec.c:3174
int PQnfields(const PGresult *res)
Definition: fe-exec.c:2777
char * PQgetvalue(const PGresult *res, int tup_num, int field_num)
Definition: fe-exec.c:3163
char * PQfname(const PGresult *res, int field_num)
Definition: fe-exec.c:2855
int PQntuples(const PGresult *res)
Definition: fe-exec.c:2769
#define fprintf
Definition: port.h:197
#define malloc(a)
Definition: header.h:50
#define free(a)
Definition: header.h:65
#define DEFAULT_FIELD_SEP
int i
#define libpq_gettext(x)
Definition: libpq-int.h:805

◆ PQdsplen()

int PQdsplen ( const char *  s,
int  encoding 
)

Definition at line 1225 of file fe-misc.c.

References pg_encoding_dsplen().

Referenced by get_prompt(), pg_wcsformat(), pg_wcssize(), pg_wcswidth(), and strlen_max_width().

1226 {
1227  return pg_encoding_dsplen(encoding, s);
1228 }
int pg_encoding_dsplen(int encoding, const char *mbstr)
Definition: wchar.c:1565
int32 encoding
Definition: pg_database.h:41

◆ PQencryptPassword()

char* PQencryptPassword ( const char *  passwd,
const char *  user 
)

Definition at line 1147 of file fe-auth.c.

References free, malloc, MD5_PASSWD_LEN, and pg_md5_encrypt().

1148 {
1149  char *crypt_pwd;
1150 
1151  crypt_pwd = malloc(MD5_PASSWD_LEN + 1);
1152  if (!crypt_pwd)
1153  return NULL;
1154 
1155  if (!pg_md5_encrypt(passwd, user, strlen(user), crypt_pwd))
1156  {
1157  free(crypt_pwd);
1158  return NULL;
1159  }
1160 
1161  return crypt_pwd;
1162 }
#define malloc(a)
Definition: header.h:50
bool pg_md5_encrypt(const char *passwd, const char *salt, size_t salt_len, char *buf)
Definition: md5.c:323
#define MD5_PASSWD_LEN
Definition: md5.h:20
#define free(a)
Definition: header.h:65
static char * user
Definition: pg_regress.c:93

◆ PQencryptPasswordConn()

char* PQencryptPasswordConn ( PGconn conn,
const char *  passwd,
const char *  user,
const char *  algorithm 
)

Definition at line 1189 of file fe-auth.c.

References pg_conn::errorMessage, free, libpq_gettext, malloc, MAX_ALGORITHM_NAME_LEN, MD5_PASSWD_LEN, pg_fe_scram_build_secret(), pg_md5_encrypt(), PGRES_TUPLES_OK, PQclear(), PQexec(), PQgetvalue(), PQnfields(), PQntuples(), PQresultStatus(), printfPQExpBuffer(), and val.

Referenced by exec_command_password(), and main().

1191 {
1192 #define MAX_ALGORITHM_NAME_LEN 50
1193  char algobuf[MAX_ALGORITHM_NAME_LEN + 1];
1194  char *crypt_pwd = NULL;
1195 
1196  if (!conn)
1197  return NULL;
1198 
1199  /* If no algorithm was given, ask the server. */
1200  if (algorithm == NULL)
1201  {
1202  PGresult *res;
1203  char *val;
1204 
1205  res = PQexec(conn, "show password_encryption");
1206  if (res == NULL)
1207  {
1208  /* PQexec() should've set conn->errorMessage already */
1209  return NULL;
1210  }
1211  if (PQresultStatus(res) != PGRES_TUPLES_OK)
1212  {
1213  /* PQexec() should've set conn->errorMessage already */
1214  PQclear(res);
1215  return NULL;
1216  }
1217  if (PQntuples(res) != 1 || PQnfields(res) != 1)
1218  {
1219  PQclear(res);
1221  libpq_gettext("unexpected shape of result set returned for SHOW\n"));
1222  return NULL;
1223  }
1224  val = PQgetvalue(res, 0, 0);
1225 
1226  if (strlen(val) > MAX_ALGORITHM_NAME_LEN)
1227  {
1228  PQclear(res);
1230  libpq_gettext("password_encryption value too long\n"));
1231  return NULL;
1232  }
1233  strcpy(algobuf, val);
1234  PQclear(res);
1235 
1236  algorithm = algobuf;
1237  }
1238 
1239  /*
1240  * Also accept "on" and "off" as aliases for "md5", because
1241  * password_encryption was a boolean before PostgreSQL 10. We refuse to
1242  * send the password in plaintext even if it was "off".
1243  */
1244  if (strcmp(algorithm, "on") == 0 ||
1245  strcmp(algorithm, "off") == 0)
1246  algorithm = "md5";
1247 
1248  /*
1249  * Ok, now we know what algorithm to use
1250  */
1251  if (strcmp(algorithm, "scram-sha-256") == 0)
1252  {
1253  crypt_pwd = pg_fe_scram_build_secret(passwd);
1254  }
1255  else if (strcmp(algorithm, "md5") == 0)
1256  {
1257  crypt_pwd = malloc(MD5_PASSWD_LEN + 1);
1258  if (crypt_pwd)
1259  {
1260  if (!pg_md5_encrypt(passwd, user, strlen(user), crypt_pwd))
1261  {
1262  free(crypt_pwd);
1263  crypt_pwd = NULL;
1264  }
1265  }
1266  }
1267  else
1268  {
1270  libpq_gettext("unrecognized password encryption algorithm \"%s\"\n"),
1271  algorithm);
1272  return NULL;
1273  }
1274 
1275  if (!crypt_pwd)
1277  libpq_gettext("out of memory\n"));
1278 
1279  return crypt_pwd;
1280 }
int PQnfields(const PGresult *res)
Definition: fe-exec.c:2777
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:237
char * PQgetvalue(const PGresult *res, int tup_num, int field_num)
Definition: fe-exec.c:3163
int PQntuples(const PGresult *res)
Definition: fe-exec.c:2769
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
#define malloc(a)
Definition: header.h:50
#define MAX_ALGORITHM_NAME_LEN
bool pg_md5_encrypt(const char *passwd, const char *salt, size_t salt_len, char *buf)
Definition: md5.c:323
char * pg_fe_scram_build_secret(const char *password)
PQExpBufferData errorMessage
Definition: libpq-int.h:526
void PQclear(PGresult *res)
Definition: fe-exec.c:694
#define MD5_PASSWD_LEN
Definition: md5.h:20
#define free(a)
Definition: header.h:65
static char * user
Definition: pg_regress.c:93
PGresult * PQexec(PGconn *conn, const char *query)
Definition: fe-exec.c:1939
long val
Definition: informix.c:664
#define libpq_gettext(x)
Definition: libpq-int.h:805

◆ PQendcopy()

int PQendcopy ( PGconn conn)

Definition at line 2618 of file fe-exec.c.

References PG_PROTOCOL_MAJOR, pqEndcopy2(), pqEndcopy3(), and pg_conn::pversion.

Referenced by ecpg_check_PQresult(), initGenerateDataClientSide(), and libpqrcv_endstreaming().

2619 {
2620  if (!conn)
2621  return 0;
2622 
2623  if (PG_PROTOCOL_MAJOR(conn->pversion) >= 3)
2624  return pqEndcopy3(conn);
2625  else
2626  return pqEndcopy2(conn);
2627 }
#define PG_PROTOCOL_MAJOR(v)
Definition: pqcomm.h:104
int pqEndcopy3(PGconn *conn)
ProtocolVersion pversion
Definition: libpq-int.h:414
int pqEndcopy2(PGconn *conn)

◆ PQenv2encoding()

int PQenv2encoding ( void  )

Definition at line 1234 of file fe-misc.c.

References dgettext, dngettext, encoding, libpq_gettext, libpq_ngettext, pg_char_to_encoding(), PG_SQL_ASCII, PG_TEXTDOMAIN, and generate_unaccent_rules::str.

Referenced by main().

1235 {
1236  char *str;
1237  int encoding = PG_SQL_ASCII;