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_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_type PQgetSSLKeyPassHook (void)
 
void PQsetSSLKeyPassHook (PQsslKeyPassHook_type hook)
 
int PQdefaultSSLKeyPassHook (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_type

typedef int(* PQsslKeyPassHook_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 }
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
union PQArgBlock::@163 u
Oid fn_lo_close
Definition: libpq-int.h:271
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
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
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
union PQArgBlock::@163 u
Oid fn_lo_creat
Definition: libpq-int.h:272
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
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
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:524
#define InvalidOid
Definition: postgres_ext.h:36
void PQclear(PGresult *res)
Definition: fe-exec.c:694
union PQArgBlock::@163 u
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
#define libpq_gettext(x)
Definition: libpq-int.h:803
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:210
static int fd(const char *x, int i)
Definition: preproc-init.c:105
#define PG_BINARY
Definition: c.h:1221
#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:524
#define LO_BUFSIZE
Definition: fe-lobj.c:42
#define strerror_r
Definition: port.h:209
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:803

◆ 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 }
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
union PQArgBlock::@163 u
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
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
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:524
void PQclear(PGresult *res)
Definition: fe-exec.c:694
union PQArgBlock::@163 u
Oid fn_lo_lseek64
Definition: libpq-int.h:276
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
int * ptr
Definition: libpq-fe.h:229
#define libpq_gettext(x)
Definition: libpq-int.h:803
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
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
union PQArgBlock::@163 u
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
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
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:524
void PQclear(PGresult *res)
Definition: fe-exec.c:694
union PQArgBlock::@163 u
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
#define libpq_gettext(x)
Definition: libpq-int.h:803
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 }
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
union PQArgBlock::@163 u
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
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
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:524
void PQclear(PGresult *res)
Definition: fe-exec.c:694
union PQArgBlock::@163 u
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
#define libpq_gettext(x)
Definition: libpq-int.h:803
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
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:524
void PQclear(PGresult *res)
Definition: fe-exec.c:694
union PQArgBlock::@163 u
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
#define libpq_gettext(x)
Definition: libpq-int.h:803
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
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:524
void PQclear(PGresult *res)
Definition: fe-exec.c:694
union PQArgBlock::@163 u
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
int * ptr
Definition: libpq-fe.h:229
#define libpq_gettext(x)
Definition: libpq-int.h:803
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 }
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
union PQArgBlock::@163 u
Oid fn_lo_unlink
Definition: libpq-int.h:274
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
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
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:1193
PQExpBufferData errorMessage
Definition: libpq-int.h:524
void PQclear(PGresult *res)
Definition: fe-exec.c:694
union PQArgBlock::@163 u
int isint
Definition: libpq-fe.h:226
PGlobjfuncs * lobjfuncs
Definition: libpq-int.h:439
int * ptr
Definition: libpq-fe.h:229
#define libpq_gettext(x)
Definition: libpq-int.h:803
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:669
#define NAMEDATALEN
static const pg_encname pg_encname_tbl[]
Definition: encnames.c:39
const char * name
Definition: encode.c:521
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 6657 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().

6658 {
6659  if (!conn || conn->status != CONNECTION_OK)
6660  return 0;
6661  return conn->be_pid;
6662 }
ConnStatusType status
Definition: libpq-int.h:386
int be_pid
Definition: libpq-int.h:432

◆ 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 4409 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().

4410 {
4411  if (!cancel)
4412  {
4413  strlcpy(errbuf, "PQcancel() -- no cancel object supplied", errbufsize);
4414  return false;
4415  }
4416 
4417  return internal_cancel(&cancel->raddr, cancel->be_pid, cancel->be_key,
4418  errbuf, errbufsize);
4419 }
static int internal_cancel(SockAddr *raddr, int be_pid, int be_key, char *errbuf, int errbufsize)
Definition: fe-connect.c:4305
int be_pid
Definition: libpq-int.h:537
int be_key
Definition: libpq-int.h:538
size_t strlcpy(char *dst, const char *src, size_t siz)
Definition: strlcpy.c:45
SockAddr raddr
Definition: libpq-int.h:536

◆ 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(), 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_schema_name_patterns(), expand_table_name_patterns(), fail_lo_xact(), fetch_more_data(), findLastBuiltinOid_V71(), finish_foreign_modify(), finish_lo_xact(), get_create_object_cmd(), get_db_infos(), get_language_name(), get_loadable_libraries(), get_parallel_object_list(), get_rel_infos(), get_remote_estimate(), get_returning_data(), get_synchronized_snapshot(), get_tablespace_paths(), getAccessMethods(), getAggregates(), getBlobs(), getCasts(), getCollations(), GetConnection(), getConstraints(), getConversions(), getCopyStart(), getDefaultACLs(), getDependencies(), getDomainConstraints(), getEventTriggers(), getExtendedStatistics(), getExtensionMembership(), getExtensions(), getForeignDataWrappers(), getForeignServers(), getFormattedTypeName(), getFuncs(), getIndexes(), getInherits(), getNamespaces(), getOpclasses(), getOperators(), getOpfamilies(), getParamDescriptions(), getPolicies(), getProcLangs(), getPublications(), getPublicationTables(), getRowDescriptions(), getRules(), getSubscriptions(), getTableAttrs(), GetTableInfo(), getTables(), getTransforms(), getTriggers(), getTSConfigurations(), getTSDictionaries(), getTSParsers(), getTSTemplates(), getTypes(), handleCopyIn(), HandleEndOfCopyStream(), initGenerateDataClientSide(), libpq_executeFileMap(), libpqConnect(), libpqGetFile(), libpqProcessFileList(), libpqrcv_create_slot(), libpqrcv_endstreaming(), libpqrcv_exec(), libpqrcv_identify_system(), libpqrcv_PQexec(), libpqrcv_readtimelinehistoryfile(), libpqrcv_receive(), libpqrcv_startstreaming(), listAllDbs(), listCasts(), listCollations(), listConversions(), listDbRoleSettings(), listDefaultACLs(), listDomains(), listEventTriggers(), listExtensionContents(), listExtensions(), listForeignDataWrappers(), listForeignServers(), listForeignTables(), listLanguages(), listOneExtensionContents(), 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 1382 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().

1383 {
1384  PQExpBufferData errorBuf;
1385  PQconninfoOption *connOptions;
1386 
1387  /* We don't actually report any errors here, but callees want a buffer */
1388  initPQExpBuffer(&errorBuf);
1389  if (PQExpBufferDataBroken(errorBuf))
1390  return NULL; /* out of memory already :-( */
1391 
1392  connOptions = conninfo_init(&errorBuf);
1393  if (connOptions != NULL)
1394  {
1395  /* pass NULL errorBuf to ignore errors */
1396  if (!conninfo_add_defaults(connOptions, NULL))
1397  {
1398  PQconninfoFree(connOptions);
1399  connOptions = NULL;
1400  }
1401  }
1402 
1403  termPQExpBuffer(&errorBuf);
1404  return connOptions;
1405 }
void termPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:131
static PQconninfoOption * conninfo_init(PQExpBuffer errorMessage)
Definition: fe-connect.c:5255
void PQconninfoFree(PQconninfoOption *connOptions)
Definition: fe-connect.c:6463
#define PQExpBufferDataBroken(buf)
Definition: pqexpbuffer.h:67
static bool conninfo_add_defaults(PQconninfoOption *options, PQExpBuffer errorMessage)
Definition: fe-connect.c:5691
void initPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:92

◆ PQconnectdb()

PGconn* PQconnectdb ( const char *  conninfo)

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

687 {
688  PGconn *conn = PQconnectStart(conninfo);
689 
690  if (conn && conn->status != CONNECTION_BAD)
691  (void) connectDBComplete(conn);
692 
693  return conn;
694 }
PGconn * conn
Definition: streamutil.c:54
static int connectDBComplete(PGconn *conn)
Definition: fe-connect.c:2020
PGconn * PQconnectStart(const char *conninfo)
Definition: fe-connect.c:812
ConnStatusType status
Definition: libpq-int.h:386

◆ PQconnectdbParams()

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

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

633 {
634  PGconn *conn = PQconnectStartParams(keywords, values, expand_dbname);
635 
636  if (conn && conn->status != CONNECTION_BAD)
637  (void) connectDBComplete(conn);
638 
639  return conn;
640 
641 }
PGconn * conn
Definition: streamutil.c:54
static int connectDBComplete(PGconn *conn)
Definition: fe-connect.c:2020
ConnStatusType status
Definition: libpq-int.h:386
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:733

◆ PQconnectionNeedsPassword()

int PQconnectionNeedsPassword ( const PGconn conn)

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

6666 {
6667  char *password;
6668 
6669  if (!conn)
6670  return false;
6671  password = PQpass(conn);
6672  if (conn->password_needed &&
6673  (password == NULL || password[0] == '\0'))
6674  return true;
6675  else
6676  return false;
6677 }
static char password[100]
Definition: streamutil.c:53
bool password_needed
Definition: libpq-int.h:415
char * PQpass(const PGconn *conn)
Definition: fe-connect.c:6497

◆ PQconnectionUsedPassword()

int PQconnectionUsedPassword ( const PGconn conn)

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

References pg_conn::password_needed.

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

6681 {
6682  if (!conn)
6683  return false;
6684  if (conn->password_needed)
6685  return true;
6686  else
6687  return false;
6688 }
bool password_needed
Definition: libpq-int.h:415

◆ PQconnectPoll()

PostgresPollingStatusType PQconnectPoll ( PGconn conn)

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

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

◆ PQconnectStart()

PGconn* PQconnectStart ( const char *  conninfo)

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

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

Referenced by PQconnectdb(), and PQping().

813 {
814  PGconn *conn;
815 
816  /*
817  * Allocate memory for the conn structure
818  */
819  conn = makeEmptyPGconn();
820  if (conn == NULL)
821  return NULL;
822 
823  /*
824  * Parse the conninfo string
825  */
826  if (!connectOptions1(conn, conninfo))
827  return conn;
828 
829  /*
830  * Compute derived options
831  */
832  if (!connectOptions2(conn))
833  return conn;
834 
835  /*
836  * Connect to the database
837  */
838  if (!connectDBStart(conn))
839  {
840  /* Just in case we failed to set it in connectDBStart */
841  conn->status = CONNECTION_BAD;
842  }
843 
844  return conn;
845 }
static bool connectOptions1(PGconn *conn, const char *conninfo)
Definition: fe-connect.c:898
PGconn * conn
Definition: streamutil.c:54
static int connectDBStart(PGconn *conn)
Definition: fe-connect.c:1941
static PGconn * makeEmptyPGconn(void)
Definition: fe-connect.c:3817
ConnStatusType status
Definition: libpq-int.h:386
static bool connectOptions2(PGconn *conn)
Definition: fe-connect.c:996

◆ PQconnectStartParams()

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

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

736 {
737  PGconn *conn;
738  PQconninfoOption *connOptions;
739 
740  /*
741  * Allocate memory for the conn structure
742  */
743  conn = makeEmptyPGconn();
744  if (conn == NULL)
745  return NULL;
746 
747  /*
748  * Parse the conninfo arrays
749  */
750  connOptions = conninfo_array_parse(keywords, values,
751  &conn->errorMessage,
752  true, expand_dbname);
753  if (connOptions == NULL)
754  {
755  conn->status = CONNECTION_BAD;
756  /* errorMessage is already set */
757  return conn;
758  }
759 
760  /*
761  * Move option values into conn structure
762  */
763  if (!fillPGconn(conn, connOptions))
764  {
765  PQconninfoFree(connOptions);
766  return conn;
767  }
768 
769  /*
770  * Free the option info - all is in conn now
771  */
772  PQconninfoFree(connOptions);
773 
774  /*
775  * Compute derived options
776  */
777  if (!connectOptions2(conn))
778  return conn;
779 
780  /*
781  * Connect to the database
782  */
783  if (!connectDBStart(conn))
784  {
785  /* Just in case we failed to set it in connectDBStart */
786  conn->status = CONNECTION_BAD;
787  }
788 
789  return conn;
790 }
PGconn * conn
Definition: streamutil.c:54
static int connectDBStart(PGconn *conn)
Definition: fe-connect.c:1941
void PQconninfoFree(PQconninfoOption *connOptions)
Definition: fe-connect.c:6463
static PGconn * makeEmptyPGconn(void)
Definition: fe-connect.c:3817
PQExpBufferData errorMessage
Definition: libpq-int.h:524
static bool fillPGconn(PGconn *conn, PQconninfoOption *connOptions)
Definition: fe-connect.c:856
ConnStatusType status
Definition: libpq-int.h:386
static Datum values[MAXATTR]
Definition: bootstrap.c:167
static bool connectOptions2(PGconn *conn)
Definition: fe-connect.c:996
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:5527

◆ PQconninfo()

PQconninfoOption* PQconninfo ( PGconn conn)

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

6423 {
6424  PQExpBufferData errorBuf;
6425  PQconninfoOption *connOptions;
6426 
6427  if (conn == NULL)
6428  return NULL;
6429 
6430  /* We don't actually report any errors here, but callees want a buffer */
6431  initPQExpBuffer(&errorBuf);
6432  if (PQExpBufferDataBroken(errorBuf))
6433  return NULL; /* out of memory already :-( */
6434 
6435  connOptions = conninfo_init(&errorBuf);
6436 
6437  if (connOptions != NULL)
6438  {
6440 
6441  for (option = PQconninfoOptions; option->keyword; option++)
6442  {
6443  char **connmember;
6444 
6445  if (option->connofs < 0)
6446  continue;
6447 
6448  connmember = (char **) ((char *) conn + option->connofs);
6449 
6450  if (*connmember)
6451  conninfo_storeval(connOptions, option->keyword, *connmember,
6452  &errorBuf, true, false);
6453  }
6454  }
6455 
6456  termPQExpBuffer(&errorBuf);
6457 
6458  return connOptions;
6459 }
static PQconninfoOption * conninfo_storeval(PQconninfoOption *connOptions, const char *keyword, const char *value, PQExpBuffer errorMessage, bool ignoreMissing, bool uri_decode)
Definition: fe-connect.c:6339
void termPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:131
static PQconninfoOption * conninfo_init(PQExpBuffer errorMessage)
Definition: fe-connect.c:5255
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 6463 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().

6464 {
6466 
6467  if (connOptions == NULL)
6468  return;
6469 
6470  for (option = connOptions; option->keyword != NULL; option++)
6471  {
6472  if (option->val != NULL)
6473  free(option->val);
6474  }
6475  free(connOptions);
6476 }
#define free(a)
Definition: header.h:65

◆ PQconninfoParse()

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

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

5234 {
5235  PQExpBufferData errorBuf;
5236  PQconninfoOption *connOptions;
5237 
5238  if (errmsg)
5239  *errmsg = NULL; /* default */
5240  initPQExpBuffer(&errorBuf);
5241  if (PQExpBufferDataBroken(errorBuf))
5242  return NULL; /* out of memory already :-( */
5243  connOptions = parse_connection_string(conninfo, &errorBuf, false);
5244  if (connOptions == NULL && errmsg)
5245  *errmsg = errorBuf.data;
5246  else
5247  termPQExpBuffer(&errorBuf);
5248  return connOptions;
5249 }
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:5295
#define PQExpBufferDataBroken(buf)
Definition: pqexpbuffer.h:67
int errmsg(const char *fmt,...)
Definition: elog.c:822
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:997
#define pqIsnonblocking(conn)
Definition: libpq-int.h:797
int pqReadData(PGconn *conn)
Definition: fe-misc.c:628

◆ PQcopyResult()

PGresult* PQcopyResult ( const PGresult src,
int  flags 
)

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

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

Referenced by pqRowProcessor().

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

◆ PQdb()

◆ PQdefaultSSLKeyPassHook()

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

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

References fprintf, libpq_gettext, and pg_conn::sslpassword.

Referenced by PQssl_passwd_cb().

1619 {
1620  if (conn->sslpassword)
1621  {
1622  if (strlen(conn->sslpassword) + 1 > size)
1623  fprintf(stderr, libpq_gettext("WARNING: sslpassword truncated"));
1624  strncpy(buf, conn->sslpassword, size);
1625  buf[size-1] = '\0';
1626  return strlen(buf);
1627  }
1628  else
1629  {
1630  buf[0] = '\0';
1631  return 0;
1632  }
1633 }
#define fprintf
Definition: port.h:196
static char * buf
Definition: pg_test_fsync.c:67
char * sslpassword
Definition: libpq-int.h:362
#define libpq_gettext(x)
Definition: libpq-int.h:803

◆ 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:196
#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:803

◆ PQdsplen()

int PQdsplen ( const char *  s,
int  encoding 
)

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

References pg_encoding_dsplen().

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

1227 {
1228  return pg_encoding_dsplen(encoding, s);
1229 }
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:94

◆ 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:524
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:94
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:803

◆ 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:412
int pqEndcopy2(PGconn *conn)

◆ PQenv2encoding()

int PQenv2encoding ( void  )

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

1236 {
1237  char *