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:1211
#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:1183
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:675
#define NAMEDATALEN
static const pg_encname pg_encname_tbl[]
Definition: encnames.c:39
const char * name
Definition: encode.c:561
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 6698 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().

6699 {
6700  if (!conn || conn->status != CONNECTION_OK)
6701  return 0;
6702  return conn->be_pid;
6703 }
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 4446 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().

4447 {
4448  if (!cancel)
4449  {
4450  strlcpy(errbuf, "PQcancel() -- no cancel object supplied", errbufsize);
4451  return false;
4452  }
4453 
4454  return internal_cancel(&cancel->raddr, cancel->be_pid, cancel->be_key,
4455  errbuf, errbufsize);
4456 }
static int internal_cancel(SockAddr *raddr, int be_pid, int be_key, char *errbuf, int errbufsize)
Definition: fe-connect.c:4342
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_connect(), 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 1444 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().

1445 {
1446  PQExpBufferData errorBuf;
1447  PQconninfoOption *connOptions;
1448 
1449  /* We don't actually report any errors here, but callees want a buffer */
1450  initPQExpBuffer(&errorBuf);
1451  if (PQExpBufferDataBroken(errorBuf))
1452  return NULL; /* out of memory already :-( */
1453 
1454  connOptions = conninfo_init(&errorBuf);
1455  if (connOptions != NULL)
1456  {
1457  /* pass NULL errorBuf to ignore errors */
1458  if (!conninfo_add_defaults(connOptions, NULL))
1459  {
1460  PQconninfoFree(connOptions);
1461  connOptions = NULL;
1462  }
1463  }
1464 
1465  termPQExpBuffer(&errorBuf);
1466  return connOptions;
1467 }
void termPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:131
static PQconninfoOption * conninfo_init(PQExpBuffer errorMessage)
Definition: fe-connect.c:5296
void PQconninfoFree(PQconninfoOption *connOptions)
Definition: fe-connect.c:6504
#define PQExpBufferDataBroken(buf)
Definition: pqexpbuffer.h:67
static bool conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage)
Definition: fe-connect.c:5732
void initPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:92

◆ PQconnectdb()

PGconn* PQconnectdb ( const char *  conninfo)

Definition at line 702 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().

703 {
704  PGconn *conn = PQconnectStart(conninfo);
705 
706  if (conn && conn->status != CONNECTION_BAD)
707  (void) connectDBComplete(conn);
708 
709  return conn;
710 }
PGconn * conn
Definition: streamutil.c:54
static int connectDBComplete(PGconn *conn)
Definition: fe-connect.c:2077
PGconn * PQconnectStart(const char *conninfo)
Definition: fe-connect.c:828
ConnStatusType status
Definition: libpq-int.h:388

◆ PQconnectdbParams()

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

Definition at line 646 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().

649 {
650  PGconn *conn = PQconnectStartParams(keywords, values, expand_dbname);
651 
652  if (conn && conn->status != CONNECTION_BAD)
653  (void) connectDBComplete(conn);
654 
655  return conn;
656 
657 }
PGconn * conn
Definition: streamutil.c:54
static int connectDBComplete(PGconn *conn)
Definition: fe-connect.c:2077
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:749

◆ PQconnectionNeedsPassword()

int PQconnectionNeedsPassword ( const PGconn conn)

Definition at line 6706 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().

6707 {
6708  char *password;
6709 
6710  if (!conn)
6711  return false;
6712  password = PQpass(conn);
6713  if (conn->password_needed &&
6714  (password == NULL || password[0] == '\0'))
6715  return true;
6716  else
6717  return false;
6718 }
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:6538

◆ PQconnectionUsedPassword()

int PQconnectionUsedPassword ( const PGconn conn)

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

References pg_conn::password_needed.

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

6722 {
6723  if (!conn)
6724  return false;
6725  if (conn->password_needed)
6726  return true;
6727  else
6728  return false;
6729 }
bool password_needed
Definition: libpq-int.h:417

◆ PQconnectPoll()

PostgresPollingStatusType PQconnectPoll ( PGconn conn)

Definition at line 2259 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().

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

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

Referenced by PQconnectdb(), and PQping().

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

◆ PQconnectStartParams()

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

Definition at line 749 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().

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

◆ PQconninfo()

PQconninfoOption* PQconninfo ( PGconn conn)

Definition at line 6463 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().

6464 {
6465  PQExpBufferData errorBuf;
6466  PQconninfoOption *connOptions;
6467 
6468  if (conn == NULL)
6469  return NULL;
6470 
6471  /* We don't actually report any errors here, but callees want a buffer */
6472  initPQExpBuffer(&errorBuf);
6473  if (PQExpBufferDataBroken(errorBuf))
6474  return NULL; /* out of memory already :-( */
6475 
6476  connOptions = conninfo_init(&errorBuf);
6477 
6478  if (connOptions != NULL)
6479  {
6481 
6482  for (option = PQconninfoOptions; option->keyword; option++)
6483  {
6484  char **connmember;
6485 
6486  if (option->connofs < 0)
6487  continue;
6488 
6489  connmember = (char **) ((char *) conn + option->connofs);
6490 
6491  if (*connmember)
6492  conninfo_storeval(connOptions, option->keyword, *connmember,
6493  &errorBuf, true, false);
6494  }
6495  }
6496 
6497  termPQExpBuffer(&errorBuf);
6498 
6499  return connOptions;
6500 }
static PQconninfoOption * conninfo_storeval(PQconninfoOption *connOptions, const char *keyword, const char *value, PQExpBuffer errorMessage, bool ignoreMissing, bool uri_decode)
Definition: fe-connect.c:6380
void termPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:131
static PQconninfoOption * conninfo_init(PQExpBuffer errorMessage)
Definition: fe-connect.c:5296
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 6504 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().

6505 {
6507 
6508  if (connOptions == NULL)
6509  return;
6510 
6511  for (option = connOptions; option->keyword != NULL; option++)
6512  {
6513  if (option->val != NULL)
6514  free(option->val);
6515  }
6516  free(connOptions);
6517 }
#define free(a)
Definition: header.h:65

◆ PQconninfoParse()

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

Definition at line 5274 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().

5275 {
5276  PQExpBufferData errorBuf;
5277  PQconninfoOption *connOptions;
5278 
5279  if (errmsg)
5280  *errmsg = NULL; /* default */
5281  initPQExpBuffer(&errorBuf);
5282  if (PQExpBufferDataBroken(errorBuf))
5283  return NULL; /* out of memory already :-( */
5284  connOptions = parse_connection_string(conninfo, &errorBuf, false);
5285  if (connOptions == NULL && errmsg)
5286  *errmsg = errorBuf.data;
5287  else
5288  termPQExpBuffer(&errorBuf);
5289  return connOptions;
5290 }
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:5336
#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:1002
#define pqIsnonblocking(conn)
Definition: libpq-int.h:799
int pqReadData(PGconn *conn)
Definition: fe-misc.c:615

◆ 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 1713 of file fe-secure-openssl.c.

References fprintf, libpq_gettext, and pg_conn::sslpassword.

Referenced by PQssl_passwd_cb().

1714 {
1715  if (conn->sslpassword)
1716  {
1717  if (strlen(conn->sslpassword) + 1 > size)
1718  fprintf(stderr, libpq_gettext("WARNING: sslpassword truncated\n"));
1719  strncpy(buf, conn->sslpassword, size);
1720  buf[size - 1] = '\0';
1721  return strlen(buf);
1722  }
1723  else
1724  {
1725  buf[0] = '\0';
1726  return 0;
1727  }
1728 }
#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 1231 of file fe-misc.c.

References pg_encoding_dsplen().

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

1232 {
1233  return pg_encoding_dsplen(encoding, s);
1234 }
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 1240 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().

1241 {
1242  char *str;
1243  int encoding =