1 /*-------------------------------------------------------------------------
4 * functions related to sending a query down to the backend
6 * Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group
7 * Portions Copyright (c) 1994, Regents of the University of California
11 * src/interfaces/libpq/fe-exec.c
13 *-------------------------------------------------------------------------
15 #include "postgres_fe.h"
28 #include "libpq-int.h"
29 #include "mb/pg_wchar.h"
31 /* keep this in same order as ExecStatusType in libpq-fe.h */
32 char *const pgresStatus
[] = {
39 "PGRES_NONFATAL_ERROR",
43 "PGRES_PIPELINE_SYNC",
44 "PGRES_PIPELINE_ABORTED",
48 /* We return this if we're unable to make a PGresult at all */
49 static const PGresult OOM_result
= {
50 .resultStatus
= PGRES_FATAL_ERROR
,
51 .client_encoding
= PG_SQL_ASCII
,
52 .errMsg
= "out of memory\n",
56 * static state needed by PQescapeString and PQescapeBytea; initialize to
57 * values that result in backward-compatible behavior
59 static int static_client_encoding
= PG_SQL_ASCII
;
60 static bool static_std_strings
= false;
63 static PGEvent
*dupEvents(PGEvent
*events
, int count
, size_t *memSize
);
64 static bool pqAddTuple(PGresult
*res
, PGresAttValue
*tup
,
65 const char **errmsgp
);
66 static int PQsendQueryInternal(PGconn
*conn
, const char *query
, bool newQuery
);
67 static bool PQsendQueryStart(PGconn
*conn
, bool newQuery
);
68 static int PQsendQueryGuts(PGconn
*conn
,
72 const Oid
*paramTypes
,
73 const char *const *paramValues
,
74 const int *paramLengths
,
75 const int *paramFormats
,
77 static void parseInput(PGconn
*conn
);
78 static PGresult
*getCopyResult(PGconn
*conn
, ExecStatusType copytype
);
79 static bool PQexecStart(PGconn
*conn
);
80 static PGresult
*PQexecFinish(PGconn
*conn
);
81 static int PQsendTypedCommand(PGconn
*conn
, char command
, char type
,
83 static int check_field_number(const PGresult
*res
, int field_num
);
84 static void pqPipelineProcessQueue(PGconn
*conn
);
85 static int pqPipelineSyncInternal(PGconn
*conn
, bool immediate_flush
);
86 static int pqPipelineFlush(PGconn
*conn
);
90 * Space management for PGresult.
92 * Formerly, libpq did a separate malloc() for each field of each tuple
93 * returned by a query. This was remarkably expensive --- malloc/free
94 * consumed a sizable part of the application's runtime. And there is
95 * no real need to keep track of the fields separately, since they will
96 * all be freed together when the PGresult is released. So now, we grab
97 * large blocks of storage from malloc and allocate space for query data
98 * within these blocks, using a trivially simple allocator. This reduces
99 * the number of malloc/free calls dramatically, and it also avoids
100 * fragmentation of the malloc storage arena.
101 * The PGresult structure itself is still malloc'd separately. We could
102 * combine it with the first allocation block, but that would waste space
103 * for the common case that no extra storage is actually needed (that is,
104 * the SQL command did not return tuples).
106 * We also malloc the top-level array of tuple pointers separately, because
107 * we need to be able to enlarge it via realloc, and our trivial space
108 * allocator doesn't handle that effectively. (Too bad the FE/BE protocol
109 * doesn't tell us up front how many tuples will be returned.)
110 * All other subsidiary storage for a PGresult is kept in PGresult_data blocks
111 * of size PGRESULT_DATA_BLOCKSIZE. The overhead at the start of each block
112 * is just a link to the next one, if any. Free-space management info is
113 * kept in the owning PGresult.
114 * A query returning a small amount of data will thus require three malloc
115 * calls: one for the PGresult, one for the tuples pointer array, and one
116 * PGresult_data block.
118 * Only the most recently allocated PGresult_data block is a candidate to
119 * have more stuff added to it --- any extra space left over in older blocks
120 * is wasted. We could be smarter and search the whole chain, but the point
121 * here is to be simple and fast. Typical applications do not keep a PGresult
122 * around very long anyway, so some wasted space within one is not a problem.
124 * Tuning constants for the space allocator are:
125 * PGRESULT_DATA_BLOCKSIZE: size of a standard allocation block, in bytes
126 * PGRESULT_ALIGN_BOUNDARY: assumed alignment requirement for binary data
127 * PGRESULT_SEP_ALLOC_THRESHOLD: objects bigger than this are given separate
128 * blocks, instead of being crammed into a regular allocation block.
129 * Requirements for correct function are:
130 * PGRESULT_ALIGN_BOUNDARY must be a multiple of the alignment requirements
131 * of all machine data types. (Currently this is set from configure
132 * tests, so it should be OK automatically.)
133 * PGRESULT_SEP_ALLOC_THRESHOLD + PGRESULT_BLOCK_OVERHEAD <=
134 * PGRESULT_DATA_BLOCKSIZE
135 * pqResultAlloc assumes an object smaller than the threshold will fit
137 * The amount of space wasted at the end of a block could be as much as
138 * PGRESULT_SEP_ALLOC_THRESHOLD, so it doesn't pay to make that too large.
142 #define PGRESULT_DATA_BLOCKSIZE 2048
143 #define PGRESULT_ALIGN_BOUNDARY MAXIMUM_ALIGNOF /* from configure */
144 #define PGRESULT_BLOCK_OVERHEAD Max(sizeof(PGresult_data), PGRESULT_ALIGN_BOUNDARY)
145 #define PGRESULT_SEP_ALLOC_THRESHOLD (PGRESULT_DATA_BLOCKSIZE / 2)
149 * PQmakeEmptyPGresult
150 * returns a newly allocated, initialized PGresult with given status.
151 * If conn is not NULL and status indicates an error, the conn's
152 * errorMessage is copied. Also, any PGEvents are copied from the conn.
154 * Note: the logic to copy the conn's errorMessage is now vestigial;
155 * no internal caller uses it. However, that behavior is documented for
156 * outside callers, so we'd better keep it.
159 PQmakeEmptyPGresult(PGconn
*conn
, ExecStatusType status
)
163 result
= (PGresult
*) malloc(sizeof(PGresult
));
168 result
->numAttributes
= 0;
169 result
->attDescs
= NULL
;
170 result
->tuples
= NULL
;
171 result
->tupArrSize
= 0;
172 result
->numParameters
= 0;
173 result
->paramDescs
= NULL
;
174 result
->resultStatus
= status
;
175 result
->cmdStatus
[0] = '\0';
177 result
->events
= NULL
;
179 result
->errMsg
= NULL
;
180 result
->errFields
= NULL
;
181 result
->errQuery
= NULL
;
182 result
->null_field
[0] = '\0';
183 result
->curBlock
= NULL
;
184 result
->curOffset
= 0;
185 result
->spaceLeft
= 0;
186 result
->memorySize
= sizeof(PGresult
);
190 /* copy connection data we might need for operations on PGresult */
191 result
->noticeHooks
= conn
->noticeHooks
;
192 result
->client_encoding
= conn
->client_encoding
;
194 /* consider copying conn's errorMessage */
197 case PGRES_EMPTY_QUERY
:
198 case PGRES_COMMAND_OK
:
199 case PGRES_TUPLES_OK
:
202 case PGRES_COPY_BOTH
:
203 case PGRES_SINGLE_TUPLE
:
204 case PGRES_TUPLES_CHUNK
:
205 /* non-error cases */
208 /* we intentionally do not use or modify errorReported here */
209 pqSetResultError(result
, &conn
->errorMessage
, 0);
213 /* copy events last; result must be valid if we need to PQclear */
214 if (conn
->nEvents
> 0)
216 result
->events
= dupEvents(conn
->events
, conn
->nEvents
,
217 &result
->memorySize
);
223 result
->nEvents
= conn
->nEvents
;
229 result
->noticeHooks
.noticeRec
= NULL
;
230 result
->noticeHooks
.noticeRecArg
= NULL
;
231 result
->noticeHooks
.noticeProc
= NULL
;
232 result
->noticeHooks
.noticeProcArg
= NULL
;
233 result
->client_encoding
= PG_SQL_ASCII
;
242 * Set the attributes for a given result. This function fails if there are
243 * already attributes contained in the provided result. The call is
244 * ignored if numAttributes is zero or attDescs is NULL. If the
245 * function fails, it returns zero. If the function succeeds, it
246 * returns a non-zero value.
249 PQsetResultAttrs(PGresult
*res
, int numAttributes
, PGresAttDesc
*attDescs
)
253 /* Fail if argument is NULL or OOM_result */
254 if (!res
|| (const PGresult
*) res
== &OOM_result
)
257 /* If attrs already exist, they cannot be overwritten. */
258 if (res
->numAttributes
> 0)
261 /* ignore no-op request */
262 if (numAttributes
<= 0 || !attDescs
)
265 res
->attDescs
= (PGresAttDesc
*)
266 PQresultAlloc(res
, numAttributes
* sizeof(PGresAttDesc
));
271 res
->numAttributes
= numAttributes
;
272 memcpy(res
->attDescs
, attDescs
, numAttributes
* sizeof(PGresAttDesc
));
274 /* deep-copy the attribute names, and determine format */
276 for (i
= 0; i
< res
->numAttributes
; i
++)
278 if (res
->attDescs
[i
].name
)
279 res
->attDescs
[i
].name
= pqResultStrdup(res
, res
->attDescs
[i
].name
);
281 res
->attDescs
[i
].name
= res
->null_field
;
283 if (!res
->attDescs
[i
].name
)
286 if (res
->attDescs
[i
].format
== 0)
296 * Returns a deep copy of the provided 'src' PGresult, which cannot be NULL.
297 * The 'flags' argument controls which portions of the result will or will
298 * NOT be copied. The created result is always put into the
299 * PGRES_TUPLES_OK status. The source result error message is not copied,
300 * although cmdStatus is.
302 * To set custom attributes, use PQsetResultAttrs. That function requires
303 * that there are no attrs contained in the result, so to use that
304 * function you cannot use the PG_COPYRES_ATTRS or PG_COPYRES_TUPLES
305 * options with this function.
308 * PG_COPYRES_ATTRS - Copy the source result's attributes
310 * PG_COPYRES_TUPLES - Copy the source result's tuples. This implies
311 * copying the attrs, seeing how the attrs are needed by the tuples.
313 * PG_COPYRES_EVENTS - Copy the source result's events.
315 * PG_COPYRES_NOTICEHOOKS - Copy the source result's notice hooks.
318 PQcopyResult(const PGresult
*src
, int flags
)
326 dest
= PQmakeEmptyPGresult(NULL
, PGRES_TUPLES_OK
);
330 /* Always copy these over. Is cmdStatus really useful here? */
331 dest
->client_encoding
= src
->client_encoding
;
332 strcpy(dest
->cmdStatus
, src
->cmdStatus
);
335 if (flags
& (PG_COPYRES_ATTRS
| PG_COPYRES_TUPLES
))
337 if (!PQsetResultAttrs(dest
, src
->numAttributes
, src
->attDescs
))
344 /* Wants to copy tuples? */
345 if (flags
& PG_COPYRES_TUPLES
)
350 for (tup
= 0; tup
< src
->ntups
; tup
++)
352 for (field
= 0; field
< src
->numAttributes
; field
++)
354 if (!PQsetvalue(dest
, tup
, field
,
355 src
->tuples
[tup
][field
].value
,
356 src
->tuples
[tup
][field
].len
))
365 /* Wants to copy notice hooks? */
366 if (flags
& PG_COPYRES_NOTICEHOOKS
)
367 dest
->noticeHooks
= src
->noticeHooks
;
369 /* Wants to copy PGEvents? */
370 if ((flags
& PG_COPYRES_EVENTS
) && src
->nEvents
> 0)
372 dest
->events
= dupEvents(src
->events
, src
->nEvents
,
379 dest
->nEvents
= src
->nEvents
;
382 /* Okay, trigger PGEVT_RESULTCOPY event */
383 for (i
= 0; i
< dest
->nEvents
; i
++)
385 /* We don't fire events that had some previous failure */
386 if (src
->events
[i
].resultInitialized
)
388 PGEventResultCopy evt
;
392 if (dest
->events
[i
].proc(PGEVT_RESULTCOPY
, &evt
,
393 dest
->events
[i
].passThrough
))
394 dest
->events
[i
].resultInitialized
= true;
402 * Copy an array of PGEvents (with no extra space for more).
403 * Does not duplicate the event instance data, sets this to NULL.
404 * Also, the resultInitialized flags are all cleared.
405 * The total space allocated is added to *memSize.
408 dupEvents(PGEvent
*events
, int count
, size_t *memSize
)
414 if (!events
|| count
<= 0)
417 msize
= count
* sizeof(PGEvent
);
418 newEvents
= (PGEvent
*) malloc(msize
);
422 for (i
= 0; i
< count
; i
++)
424 newEvents
[i
].proc
= events
[i
].proc
;
425 newEvents
[i
].passThrough
= events
[i
].passThrough
;
426 newEvents
[i
].data
= NULL
;
427 newEvents
[i
].resultInitialized
= false;
428 newEvents
[i
].name
= strdup(events
[i
].name
);
429 if (!newEvents
[i
].name
)
432 free(newEvents
[i
].name
);
436 msize
+= strlen(events
[i
].name
) + 1;
445 * Sets the value for a tuple field. The tup_num must be less than or
446 * equal to PQntuples(res). If it is equal, a new tuple is created and
447 * added to the result.
448 * Returns a non-zero value for success and zero for failure.
449 * (On failure, we report the specific problem via pqInternalNotice.)
452 PQsetvalue(PGresult
*res
, int tup_num
, int field_num
, char *value
, int len
)
454 PGresAttValue
*attval
;
455 const char *errmsg
= NULL
;
457 /* Fail if argument is NULL or OOM_result */
458 if (!res
|| (const PGresult
*) res
== &OOM_result
)
461 /* Invalid field_num? */
462 if (!check_field_number(res
, field_num
))
465 /* Invalid tup_num, must be <= ntups */
466 if (tup_num
< 0 || tup_num
> res
->ntups
)
468 pqInternalNotice(&res
->noticeHooks
,
469 "row number %d is out of range 0..%d",
470 tup_num
, res
->ntups
);
474 /* need to allocate a new tuple? */
475 if (tup_num
== res
->ntups
)
480 tup
= (PGresAttValue
*)
481 pqResultAlloc(res
, res
->numAttributes
* sizeof(PGresAttValue
),
487 /* initialize each column to NULL */
488 for (i
= 0; i
< res
->numAttributes
; i
++)
490 tup
[i
].len
= NULL_LEN
;
491 tup
[i
].value
= res
->null_field
;
494 /* add it to the array */
495 if (!pqAddTuple(res
, tup
, &errmsg
))
499 attval
= &res
->tuples
[tup_num
][field_num
];
501 /* treat either NULL_LEN or NULL value pointer as a NULL field */
502 if (len
== NULL_LEN
|| value
== NULL
)
504 attval
->len
= NULL_LEN
;
505 attval
->value
= res
->null_field
;
510 attval
->value
= res
->null_field
;
514 attval
->value
= (char *) pqResultAlloc(res
, len
+ 1, true);
518 memcpy(attval
->value
, value
, len
);
519 attval
->value
[len
] = '\0';
525 * Report failure via pqInternalNotice. If preceding code didn't provide
526 * an error message, assume "out of memory" was meant.
530 errmsg
= libpq_gettext("out of memory");
531 pqInternalNotice(&res
->noticeHooks
, "%s", errmsg
);
537 * pqResultAlloc - exported routine to allocate local storage in a PGresult.
539 * We force all such allocations to be maxaligned, since we don't know
540 * whether the value might be binary.
543 PQresultAlloc(PGresult
*res
, size_t nBytes
)
545 /* Fail if argument is NULL or OOM_result */
546 if (!res
|| (const PGresult
*) res
== &OOM_result
)
549 return pqResultAlloc(res
, nBytes
, true);
554 * Allocate subsidiary storage for a PGresult.
556 * nBytes is the amount of space needed for the object.
557 * If isBinary is true, we assume that we need to align the object on
558 * a machine allocation boundary.
559 * If isBinary is false, we assume the object is a char string and can
560 * be allocated on any byte boundary.
563 pqResultAlloc(PGresult
*res
, size_t nBytes
, bool isBinary
)
566 PGresult_data
*block
;
572 return res
->null_field
;
575 * If alignment is needed, round up the current position to an alignment
580 int offset
= res
->curOffset
% PGRESULT_ALIGN_BOUNDARY
;
584 res
->curOffset
+= PGRESULT_ALIGN_BOUNDARY
- offset
;
585 res
->spaceLeft
-= PGRESULT_ALIGN_BOUNDARY
- offset
;
589 /* If there's enough space in the current block, no problem. */
590 if (nBytes
<= (size_t) res
->spaceLeft
)
592 space
= res
->curBlock
->space
+ res
->curOffset
;
593 res
->curOffset
+= nBytes
;
594 res
->spaceLeft
-= nBytes
;
599 * If the requested object is very large, give it its own block; this
600 * avoids wasting what might be most of the current block to start a new
601 * block. (We'd have to special-case requests bigger than the block size
602 * anyway.) The object is always given binary alignment in this case.
604 if (nBytes
>= PGRESULT_SEP_ALLOC_THRESHOLD
)
606 size_t alloc_size
= nBytes
+ PGRESULT_BLOCK_OVERHEAD
;
608 block
= (PGresult_data
*) malloc(alloc_size
);
611 res
->memorySize
+= alloc_size
;
612 space
= block
->space
+ PGRESULT_BLOCK_OVERHEAD
;
616 * Tuck special block below the active block, so that we don't
617 * have to waste the free space in the active block.
619 block
->next
= res
->curBlock
->next
;
620 res
->curBlock
->next
= block
;
624 /* Must set up the new block as the first active block. */
626 res
->curBlock
= block
;
627 res
->spaceLeft
= 0; /* be sure it's marked full */
632 /* Otherwise, start a new block. */
633 block
= (PGresult_data
*) malloc(PGRESULT_DATA_BLOCKSIZE
);
636 res
->memorySize
+= PGRESULT_DATA_BLOCKSIZE
;
637 block
->next
= res
->curBlock
;
638 res
->curBlock
= block
;
641 /* object needs full alignment */
642 res
->curOffset
= PGRESULT_BLOCK_OVERHEAD
;
643 res
->spaceLeft
= PGRESULT_DATA_BLOCKSIZE
- PGRESULT_BLOCK_OVERHEAD
;
647 /* we can cram it right after the overhead pointer */
648 res
->curOffset
= sizeof(PGresult_data
);
649 res
->spaceLeft
= PGRESULT_DATA_BLOCKSIZE
- sizeof(PGresult_data
);
652 space
= block
->space
+ res
->curOffset
;
653 res
->curOffset
+= nBytes
;
654 res
->spaceLeft
-= nBytes
;
659 * PQresultMemorySize -
660 * Returns total space allocated for the PGresult.
663 PQresultMemorySize(const PGresult
*res
)
667 return res
->memorySize
;
672 * Like strdup, but the space is subsidiary PGresult space.
675 pqResultStrdup(PGresult
*res
, const char *str
)
677 char *space
= (char *) pqResultAlloc(res
, strlen(str
) + 1, false);
686 * assign a new error message to a PGresult
688 * Copy text from errorMessage buffer beginning at given offset
689 * (it's caller's responsibility that offset is valid)
692 pqSetResultError(PGresult
*res
, PQExpBuffer errorMessage
, int offset
)
700 * We handle two OOM scenarios here. The errorMessage buffer might be
701 * marked "broken" due to having previously failed to allocate enough
702 * memory for the message, or it might be fine but pqResultStrdup fails
703 * and returns NULL. In either case, just make res->errMsg point directly
704 * at a constant "out of memory" string.
706 if (!PQExpBufferBroken(errorMessage
))
707 msg
= pqResultStrdup(res
, errorMessage
->data
+ offset
);
713 res
->errMsg
= libpq_gettext("out of memory\n");
718 * free's the memory associated with a PGresult
721 PQclear(PGresult
*res
)
723 PGresult_data
*block
;
726 /* As a convenience, do nothing for a NULL pointer */
729 /* Also, do nothing if the argument is OOM_result */
730 if ((const PGresult
*) res
== &OOM_result
)
733 /* Close down any events we may have */
734 for (i
= 0; i
< res
->nEvents
; i
++)
736 /* only send DESTROY to successfully-initialized event procs */
737 if (res
->events
[i
].resultInitialized
)
739 PGEventResultDestroy evt
;
742 (void) res
->events
[i
].proc(PGEVT_RESULTDESTROY
, &evt
,
743 res
->events
[i
].passThrough
);
745 free(res
->events
[i
].name
);
750 /* Free all the subsidiary blocks */
751 while ((block
= res
->curBlock
) != NULL
)
753 res
->curBlock
= block
->next
;
757 /* Free the top-level tuple pointer array */
760 /* zero out the pointer fields to catch programming errors */
761 res
->attDescs
= NULL
;
763 res
->paramDescs
= NULL
;
764 res
->errFields
= NULL
;
767 /* res->curBlock was zeroed out earlier */
769 /* Free the PGresult structure itself */
774 * Handy subroutine to deallocate any partially constructed async result.
776 * Any "saved" result gets cleared too.
779 pqClearAsyncResult(PGconn
*conn
)
781 PQclear(conn
->result
);
783 conn
->error_result
= false;
784 PQclear(conn
->saved_result
);
785 conn
->saved_result
= NULL
;
789 * pqSaveErrorResult -
790 * remember that we have an error condition
792 * In much of libpq, reporting an error just requires appending text to
793 * conn->errorMessage and returning a failure code to one's caller.
794 * Where returning a failure code is impractical, instead call this
795 * function to remember that an error needs to be reported.
797 * (It might seem that appending text to conn->errorMessage should be
798 * sufficient, but we can't rely on that working under out-of-memory
799 * conditions. The OOM hazard is also why we don't try to make a new
800 * PGresult right here.)
803 pqSaveErrorResult(PGconn
*conn
)
805 /* Drop any pending result ... */
806 pqClearAsyncResult(conn
);
807 /* ... and set flag to remember to make an error result later */
808 conn
->error_result
= true;
813 * report a write failure
815 * As above, after appending conn->write_err_msg to whatever other error we
816 * have. This is used when we've detected a write failure and have exhausted
817 * our chances of reporting something else instead.
820 pqSaveWriteError(PGconn
*conn
)
823 * If write_err_msg is null because of previous strdup failure, do what we
824 * can. (It's likely our machinations here will get OOM failures as well,
825 * but might as well try.)
827 if (conn
->write_err_msg
)
829 appendPQExpBufferStr(&conn
->errorMessage
, conn
->write_err_msg
);
830 /* Avoid possibly appending the same message twice */
831 conn
->write_err_msg
[0] = '\0';
834 libpq_append_conn_error(conn
, "write to server failed");
836 pqSaveErrorResult(conn
);
840 * pqPrepareAsyncResult -
841 * prepare the current async result object for return to the caller
843 * If there is not already an async result object, build an error object
844 * using whatever is in conn->errorMessage. In any case, clear the async
845 * result storage, and update our notion of how much error text has been
846 * returned to the application.
848 * Note that in no case (not even OOM) do we return NULL.
851 pqPrepareAsyncResult(PGconn
*conn
)
859 * If the pre-existing result is an ERROR (presumably something
860 * received from the server), assume that it represents whatever is in
861 * conn->errorMessage, and advance errorReported.
863 if (res
->resultStatus
== PGRES_FATAL_ERROR
)
864 conn
->errorReported
= conn
->errorMessage
.len
;
869 * We get here after internal-to-libpq errors. We should probably
870 * always have error_result = true, but if we don't, gin up some error
873 if (!conn
->error_result
)
874 libpq_append_conn_error(conn
, "no error text available");
876 /* Paranoia: be sure errorReported offset is sane */
877 if (conn
->errorReported
< 0 ||
878 conn
->errorReported
>= conn
->errorMessage
.len
)
879 conn
->errorReported
= 0;
882 * Make a PGresult struct for the error. We temporarily lie about the
883 * result status, so that PQmakeEmptyPGresult doesn't uselessly copy
884 * all of conn->errorMessage.
886 res
= PQmakeEmptyPGresult(conn
, PGRES_EMPTY_QUERY
);
890 * Report whatever new error text we have, and advance
893 res
->resultStatus
= PGRES_FATAL_ERROR
;
894 pqSetResultError(res
, &conn
->errorMessage
, conn
->errorReported
);
895 conn
->errorReported
= conn
->errorMessage
.len
;
900 * Ouch, not enough memory for a PGresult. Fortunately, we have a
901 * card up our sleeve: we can use the static OOM_result. Casting
902 * away const here is a bit ugly, but it seems best to declare
903 * OOM_result as const, in hopes it will be allocated in read-only
906 res
= unconstify(PGresult
*, &OOM_result
);
909 * Don't advance errorReported. Perhaps we'll be able to report
916 * Replace conn->result with saved_result, if any. In the normal case
917 * there isn't a saved result and we're just dropping ownership of the
918 * current result. In partial-result mode this restores the situation to
919 * what it was before we created the current partial result.
921 conn
->result
= conn
->saved_result
;
922 conn
->error_result
= false; /* saved_result is never an error */
923 conn
->saved_result
= NULL
;
929 * pqInternalNotice - produce an internally-generated notice message
931 * A format string and optional arguments can be passed. Note that we do
932 * libpq_gettext() here, so callers need not.
934 * The supplied text is taken as primary message (ie., it should not include
935 * a trailing newline, and should not be more than one line).
938 pqInternalNotice(const PGNoticeHooks
*hooks
, const char *fmt
,...)
944 if (hooks
->noticeRec
== NULL
)
945 return; /* nobody home to receive notice? */
947 /* Format the message */
949 vsnprintf(msgBuf
, sizeof(msgBuf
), libpq_gettext(fmt
), args
);
951 msgBuf
[sizeof(msgBuf
) - 1] = '\0'; /* make real sure it's terminated */
953 /* Make a PGresult to pass to the notice receiver */
954 res
= PQmakeEmptyPGresult(NULL
, PGRES_NONFATAL_ERROR
);
957 res
->noticeHooks
= *hooks
;
960 * Set up fields of notice.
962 pqSaveMessageField(res
, PG_DIAG_MESSAGE_PRIMARY
, msgBuf
);
963 pqSaveMessageField(res
, PG_DIAG_SEVERITY
, libpq_gettext("NOTICE"));
964 pqSaveMessageField(res
, PG_DIAG_SEVERITY_NONLOCALIZED
, "NOTICE");
965 /* XXX should provide a SQLSTATE too? */
968 * Result text is always just the primary message + newline. If we can't
969 * allocate it, substitute "out of memory", as in pqSetResultError.
971 res
->errMsg
= (char *) pqResultAlloc(res
, strlen(msgBuf
) + 2, false);
973 sprintf(res
->errMsg
, "%s\n", msgBuf
);
975 res
->errMsg
= libpq_gettext("out of memory\n");
978 * Pass to receiver, then free it.
980 res
->noticeHooks
.noticeRec(res
->noticeHooks
.noticeRecArg
, res
);
986 * add a row pointer to the PGresult structure, growing it if necessary
987 * Returns true if OK, false if an error prevented adding the row
989 * On error, *errmsgp can be set to an error string to be returned.
990 * If it is left NULL, the error is presumed to be "out of memory".
993 pqAddTuple(PGresult
*res
, PGresAttValue
*tup
, const char **errmsgp
)
995 if (res
->ntups
>= res
->tupArrSize
)
998 * Try to grow the array.
1000 * We can use realloc because shallow copying of the structure is
1001 * okay. Note that the first time through, res->tuples is NULL. While
1002 * ANSI says that realloc() should act like malloc() in that case,
1003 * some old C libraries (like SunOS 4.1.x) coredump instead. On
1004 * failure realloc is supposed to return NULL without damaging the
1005 * existing allocation. Note that the positions beyond res->ntups are
1006 * garbage, not necessarily NULL.
1009 PGresAttValue
**newTuples
;
1012 * Since we use integers for row numbers, we can't support more than
1013 * INT_MAX rows. Make sure we allow that many, though.
1015 if (res
->tupArrSize
<= INT_MAX
/ 2)
1016 newSize
= (res
->tupArrSize
> 0) ? res
->tupArrSize
* 2 : 128;
1017 else if (res
->tupArrSize
< INT_MAX
)
1021 *errmsgp
= libpq_gettext("PGresult cannot support more than INT_MAX tuples");
1026 * Also, on 32-bit platforms we could, in theory, overflow size_t even
1027 * before newSize gets to INT_MAX. (In practice we'd doubtless hit
1028 * OOM long before that, but let's check.)
1030 #if INT_MAX >= (SIZE_MAX / 2)
1031 if (newSize
> SIZE_MAX
/ sizeof(PGresAttValue
*))
1033 *errmsgp
= libpq_gettext("size_t overflow");
1038 if (res
->tuples
== NULL
)
1039 newTuples
= (PGresAttValue
**)
1040 malloc(newSize
* sizeof(PGresAttValue
*));
1042 newTuples
= (PGresAttValue
**)
1043 realloc(res
->tuples
, newSize
* sizeof(PGresAttValue
*));
1045 return false; /* malloc or realloc failed */
1047 (newSize
- res
->tupArrSize
) * sizeof(PGresAttValue
*);
1048 res
->tupArrSize
= newSize
;
1049 res
->tuples
= newTuples
;
1051 res
->tuples
[res
->ntups
] = tup
;
1057 * pqSaveMessageField - save one field of an error or notice message
1060 pqSaveMessageField(PGresult
*res
, char code
, const char *value
)
1062 PGMessageField
*pfield
;
1064 pfield
= (PGMessageField
*)
1066 offsetof(PGMessageField
, contents
) +
1070 return; /* out of memory? */
1071 pfield
->code
= code
;
1072 strcpy(pfield
->contents
, value
);
1073 pfield
->next
= res
->errFields
;
1074 res
->errFields
= pfield
;
1078 * pqSaveParameterStatus - remember parameter status sent by backend
1081 pqSaveParameterStatus(PGconn
*conn
, const char *name
, const char *value
)
1083 pgParameterStatus
*pstatus
;
1084 pgParameterStatus
*prev
;
1087 * Forget any old information about the parameter
1089 for (pstatus
= conn
->pstatus
, prev
= NULL
;
1091 prev
= pstatus
, pstatus
= pstatus
->next
)
1093 if (strcmp(pstatus
->name
, name
) == 0)
1096 prev
->next
= pstatus
->next
;
1098 conn
->pstatus
= pstatus
->next
;
1099 free(pstatus
); /* frees name and value strings too */
1105 * Store new info as a single malloc block
1107 pstatus
= (pgParameterStatus
*) malloc(sizeof(pgParameterStatus
) +
1108 strlen(name
) + strlen(value
) + 2);
1113 ptr
= ((char *) pstatus
) + sizeof(pgParameterStatus
);
1114 pstatus
->name
= ptr
;
1116 ptr
+= strlen(name
) + 1;
1117 pstatus
->value
= ptr
;
1119 pstatus
->next
= conn
->pstatus
;
1120 conn
->pstatus
= pstatus
;
1124 * Save values of settings that are of interest to libpq in fields of the
1125 * PGconn object. We keep client_encoding and standard_conforming_strings
1126 * in static variables as well, so that PQescapeString and PQescapeBytea
1127 * can behave somewhat sanely (at least in single-connection-using
1130 if (strcmp(name
, "client_encoding") == 0)
1132 conn
->client_encoding
= pg_char_to_encoding(value
);
1133 /* if we don't recognize the encoding name, fall back to SQL_ASCII */
1134 if (conn
->client_encoding
< 0)
1135 conn
->client_encoding
= PG_SQL_ASCII
;
1136 static_client_encoding
= conn
->client_encoding
;
1138 else if (strcmp(name
, "standard_conforming_strings") == 0)
1140 conn
->std_strings
= (strcmp(value
, "on") == 0);
1141 static_std_strings
= conn
->std_strings
;
1143 else if (strcmp(name
, "server_version") == 0)
1145 /* We convert the server version to numeric form. */
1151 cnt
= sscanf(value
, "%d.%d.%d", &vmaj
, &vmin
, &vrev
);
1155 /* old style, e.g. 9.6.1 */
1156 conn
->sversion
= (100 * vmaj
+ vmin
) * 100 + vrev
;
1162 /* new style, e.g. 10.1 */
1163 conn
->sversion
= 100 * 100 * vmaj
+ vmin
;
1167 /* old style without minor version, e.g. 9.6devel */
1168 conn
->sversion
= (100 * vmaj
+ vmin
) * 100;
1173 /* new style without minor version, e.g. 10devel */
1174 conn
->sversion
= 100 * 100 * vmaj
;
1177 conn
->sversion
= 0; /* unknown */
1179 else if (strcmp(name
, "default_transaction_read_only") == 0)
1181 conn
->default_transaction_read_only
=
1182 (strcmp(value
, "on") == 0) ? PG_BOOL_YES
: PG_BOOL_NO
;
1184 else if (strcmp(name
, "in_hot_standby") == 0)
1186 conn
->in_hot_standby
=
1187 (strcmp(value
, "on") == 0) ? PG_BOOL_YES
: PG_BOOL_NO
;
1189 else if (strcmp(name
, "scram_iterations") == 0)
1191 conn
->scram_sha_256_iterations
= atoi(value
);
1198 * Add the received row to the current async result (conn->result).
1199 * Returns 1 if OK, 0 if error occurred.
1201 * On error, *errmsgp can be set to an error string to be returned.
1202 * (Such a string should already be translated via libpq_gettext().)
1203 * If it is left NULL, the error is presumed to be "out of memory".
1206 pqRowProcessor(PGconn
*conn
, const char **errmsgp
)
1208 PGresult
*res
= conn
->result
;
1209 int nfields
= res
->numAttributes
;
1210 const PGdataValue
*columns
= conn
->rowBuf
;
1215 * In partial-result mode, if we don't already have a partial PGresult
1216 * then make one by cloning conn->result (which should hold the correct
1217 * result metadata by now). Then the original conn->result is moved over
1218 * to saved_result so that we can re-use it as a reference for future
1219 * partial results. The saved result will become active again after
1220 * pqPrepareAsyncResult() returns the partial result to the application.
1222 if (conn
->partialResMode
&& conn
->saved_result
== NULL
)
1224 /* Copy everything that should be in the result at this point */
1225 res
= PQcopyResult(res
,
1226 PG_COPYRES_ATTRS
| PG_COPYRES_EVENTS
|
1227 PG_COPYRES_NOTICEHOOKS
);
1230 /* Change result status to appropriate special value */
1231 res
->resultStatus
= (conn
->singleRowMode
? PGRES_SINGLE_TUPLE
: PGRES_TUPLES_CHUNK
);
1232 /* And stash it as the active result */
1233 conn
->saved_result
= conn
->result
;
1238 * Basically we just allocate space in the PGresult for each field and
1239 * copy the data over.
1241 * Note: on malloc failure, we return 0 leaving *errmsgp still NULL, which
1242 * caller will take to mean "out of memory". This is preferable to trying
1243 * to set up such a message here, because evidently there's not enough
1244 * memory for gettext() to do anything.
1246 tup
= (PGresAttValue
*)
1247 pqResultAlloc(res
, nfields
* sizeof(PGresAttValue
), true);
1251 for (i
= 0; i
< nfields
; i
++)
1253 int clen
= columns
[i
].len
;
1258 tup
[i
].len
= NULL_LEN
;
1259 tup
[i
].value
= res
->null_field
;
1263 bool isbinary
= (res
->attDescs
[i
].format
!= 0);
1266 val
= (char *) pqResultAlloc(res
, clen
+ 1, isbinary
);
1270 /* copy and zero-terminate the data (even if it's binary) */
1271 memcpy(val
, columns
[i
].value
, clen
);
1279 /* And add the tuple to the PGresult's tuple array */
1280 if (!pqAddTuple(res
, tup
, errmsgp
))
1284 * Success. In partial-result mode, if we have enough rows then make the
1285 * result available to the client immediately.
1287 if (conn
->partialResMode
&& res
->ntups
>= conn
->maxChunkSize
)
1288 conn
->asyncStatus
= PGASYNC_READY_MORE
;
1295 * pqAllocCmdQueueEntry
1296 * Get a command queue entry for caller to fill.
1298 * If the recycle queue has a free element, that is returned; if not, a
1299 * fresh one is allocated. Caller is responsible for adding it to the
1300 * command queue (pqAppendCmdQueueEntry) once the struct is filled in, or
1301 * releasing the memory (pqRecycleCmdQueueEntry) if an error occurs.
1303 * If allocation fails, sets the error message and returns NULL.
1305 static PGcmdQueueEntry
*
1306 pqAllocCmdQueueEntry(PGconn
*conn
)
1308 PGcmdQueueEntry
*entry
;
1310 if (conn
->cmd_queue_recycle
== NULL
)
1312 entry
= (PGcmdQueueEntry
*) malloc(sizeof(PGcmdQueueEntry
));
1315 libpq_append_conn_error(conn
, "out of memory");
1321 entry
= conn
->cmd_queue_recycle
;
1322 conn
->cmd_queue_recycle
= entry
->next
;
1325 entry
->query
= NULL
;
1331 * pqAppendCmdQueueEntry
1332 * Append a caller-allocated entry to the command queue, and update
1333 * conn->asyncStatus to account for it.
1335 * The query itself must already have been put in the output buffer by the
1339 pqAppendCmdQueueEntry(PGconn
*conn
, PGcmdQueueEntry
*entry
)
1341 Assert(entry
->next
== NULL
);
1343 if (conn
->cmd_queue_head
== NULL
)
1344 conn
->cmd_queue_head
= entry
;
1346 conn
->cmd_queue_tail
->next
= entry
;
1348 conn
->cmd_queue_tail
= entry
;
1350 switch (conn
->pipelineStatus
)
1352 case PQ_PIPELINE_OFF
:
1353 case PQ_PIPELINE_ON
:
1356 * When not in pipeline aborted state, if there's a result ready
1357 * to be consumed, let it be so (that is, don't change away from
1358 * READY or READY_MORE); otherwise set us busy to wait for
1359 * something to arrive from the server.
1361 if (conn
->asyncStatus
== PGASYNC_IDLE
)
1362 conn
->asyncStatus
= PGASYNC_BUSY
;
1365 case PQ_PIPELINE_ABORTED
:
1368 * In aborted pipeline state, we don't expect anything from the
1369 * server (since we don't send any queries that are queued).
1370 * Therefore, if IDLE then do what PQgetResult would do to let
1371 * itself consume commands from the queue; if we're in any other
1372 * state, we don't have to do anything.
1374 if (conn
->asyncStatus
== PGASYNC_IDLE
||
1375 conn
->asyncStatus
== PGASYNC_PIPELINE_IDLE
)
1376 pqPipelineProcessQueue(conn
);
1382 * pqRecycleCmdQueueEntry
1383 * Push a command queue entry onto the freelist.
1386 pqRecycleCmdQueueEntry(PGconn
*conn
, PGcmdQueueEntry
*entry
)
1391 /* recyclable entries should not have a follow-on command */
1392 Assert(entry
->next
== NULL
);
1397 entry
->query
= NULL
;
1400 entry
->next
= conn
->cmd_queue_recycle
;
1401 conn
->cmd_queue_recycle
= entry
;
1407 * Submit a query, but don't wait for it to finish
1409 * Returns: 1 if successfully submitted
1410 * 0 if error (conn->errorMessage is set)
1412 * PQsendQueryContinue is a non-exported version that behaves identically
1413 * except that it doesn't reset conn->errorMessage.
1416 PQsendQuery(PGconn
*conn
, const char *query
)
1418 return PQsendQueryInternal(conn
, query
, true);
1422 PQsendQueryContinue(PGconn
*conn
, const char *query
)
1424 return PQsendQueryInternal(conn
, query
, false);
1428 PQsendQueryInternal(PGconn
*conn
, const char *query
, bool newQuery
)
1430 PGcmdQueueEntry
*entry
= NULL
;
1432 if (!PQsendQueryStart(conn
, newQuery
))
1435 /* check the argument */
1438 libpq_append_conn_error(conn
, "command string is a null pointer");
1442 if (conn
->pipelineStatus
!= PQ_PIPELINE_OFF
)
1444 libpq_append_conn_error(conn
, "%s not allowed in pipeline mode",
1449 entry
= pqAllocCmdQueueEntry(conn
);
1451 return 0; /* error msg already set */
1453 /* Send the query message(s) */
1454 /* construct the outgoing Query message */
1455 if (pqPutMsgStart(PqMsg_Query
, conn
) < 0 ||
1456 pqPuts(query
, conn
) < 0 ||
1457 pqPutMsgEnd(conn
) < 0)
1459 /* error message should be set up already */
1460 pqRecycleCmdQueueEntry(conn
, entry
);
1464 /* remember we are using simple query protocol */
1465 entry
->queryclass
= PGQUERY_SIMPLE
;
1466 /* and remember the query text too, if possible */
1467 entry
->query
= strdup(query
);
1470 * Give the data a push. In nonblock mode, don't complain if we're unable
1471 * to send it all; PQgetResult() will do any additional flushing needed.
1473 if (pqFlush(conn
) < 0)
1476 /* OK, it's launched! */
1477 pqAppendCmdQueueEntry(conn
, entry
);
1482 pqRecycleCmdQueueEntry(conn
, entry
);
1483 /* error message should be set up already */
1489 * Like PQsendQuery, but use extended query protocol so we can pass parameters
1492 PQsendQueryParams(PGconn
*conn
,
1493 const char *command
,
1495 const Oid
*paramTypes
,
1496 const char *const *paramValues
,
1497 const int *paramLengths
,
1498 const int *paramFormats
,
1501 if (!PQsendQueryStart(conn
, true))
1504 /* check the arguments */
1507 libpq_append_conn_error(conn
, "command string is a null pointer");
1510 if (nParams
< 0 || nParams
> PQ_QUERY_PARAM_MAX_LIMIT
)
1512 libpq_append_conn_error(conn
, "number of parameters must be between 0 and %d",
1513 PQ_QUERY_PARAM_MAX_LIMIT
);
1517 return PQsendQueryGuts(conn
,
1519 "", /* use unnamed statement */
1530 * Submit a Parse message, but don't wait for it to finish
1532 * Returns: 1 if successfully submitted
1533 * 0 if error (conn->errorMessage is set)
1536 PQsendPrepare(PGconn
*conn
,
1537 const char *stmtName
, const char *query
,
1538 int nParams
, const Oid
*paramTypes
)
1540 PGcmdQueueEntry
*entry
= NULL
;
1542 if (!PQsendQueryStart(conn
, true))
1545 /* check the arguments */
1548 libpq_append_conn_error(conn
, "statement name is a null pointer");
1553 libpq_append_conn_error(conn
, "command string is a null pointer");
1556 if (nParams
< 0 || nParams
> PQ_QUERY_PARAM_MAX_LIMIT
)
1558 libpq_append_conn_error(conn
, "number of parameters must be between 0 and %d",
1559 PQ_QUERY_PARAM_MAX_LIMIT
);
1563 entry
= pqAllocCmdQueueEntry(conn
);
1565 return 0; /* error msg already set */
1567 /* construct the Parse message */
1568 if (pqPutMsgStart(PqMsg_Parse
, conn
) < 0 ||
1569 pqPuts(stmtName
, conn
) < 0 ||
1570 pqPuts(query
, conn
) < 0)
1573 if (nParams
> 0 && paramTypes
)
1577 if (pqPutInt(nParams
, 2, conn
) < 0)
1579 for (i
= 0; i
< nParams
; i
++)
1581 if (pqPutInt(paramTypes
[i
], 4, conn
) < 0)
1587 if (pqPutInt(0, 2, conn
) < 0)
1590 if (pqPutMsgEnd(conn
) < 0)
1593 /* Add a Sync, unless in pipeline mode. */
1594 if (conn
->pipelineStatus
== PQ_PIPELINE_OFF
)
1596 if (pqPutMsgStart(PqMsg_Sync
, conn
) < 0 ||
1597 pqPutMsgEnd(conn
) < 0)
1601 /* remember we are doing just a Parse */
1602 entry
->queryclass
= PGQUERY_PREPARE
;
1604 /* and remember the query text too, if possible */
1605 /* if insufficient memory, query just winds up NULL */
1606 entry
->query
= strdup(query
);
1609 * Give the data a push (in pipeline mode, only if we're past the size
1610 * threshold). In nonblock mode, don't complain if we're unable to send
1611 * it all; PQgetResult() will do any additional flushing needed.
1613 if (pqPipelineFlush(conn
) < 0)
1616 /* OK, it's launched! */
1617 pqAppendCmdQueueEntry(conn
, entry
);
1622 pqRecycleCmdQueueEntry(conn
, entry
);
1623 /* error message should be set up already */
1628 * PQsendQueryPrepared
1629 * Like PQsendQuery, but execute a previously prepared statement,
1630 * using extended query protocol so we can pass parameters
1633 PQsendQueryPrepared(PGconn
*conn
,
1634 const char *stmtName
,
1636 const char *const *paramValues
,
1637 const int *paramLengths
,
1638 const int *paramFormats
,
1641 if (!PQsendQueryStart(conn
, true))
1644 /* check the arguments */
1647 libpq_append_conn_error(conn
, "statement name is a null pointer");
1650 if (nParams
< 0 || nParams
> PQ_QUERY_PARAM_MAX_LIMIT
)
1652 libpq_append_conn_error(conn
, "number of parameters must be between 0 and %d",
1653 PQ_QUERY_PARAM_MAX_LIMIT
);
1657 return PQsendQueryGuts(conn
,
1658 NULL
, /* no command to parse */
1661 NULL
, /* no param types */
1670 * Common startup code for PQsendQuery and sibling routines
1673 PQsendQueryStart(PGconn
*conn
, bool newQuery
)
1679 * If this is the beginning of a query cycle, reset the error state.
1680 * However, in pipeline mode with something already queued, the error
1681 * buffer belongs to that command and we shouldn't clear it.
1683 if (newQuery
&& conn
->cmd_queue_head
== NULL
)
1684 pqClearConnErrorState(conn
);
1686 /* Don't try to send if we know there's no live connection. */
1687 if (conn
->status
!= CONNECTION_OK
)
1689 libpq_append_conn_error(conn
, "no connection to the server");
1693 /* Can't send while already busy, either, unless enqueuing for later */
1694 if (conn
->asyncStatus
!= PGASYNC_IDLE
&&
1695 conn
->pipelineStatus
== PQ_PIPELINE_OFF
)
1697 libpq_append_conn_error(conn
, "another command is already in progress");
1701 if (conn
->pipelineStatus
!= PQ_PIPELINE_OFF
)
1704 * When enqueuing commands we don't change much of the connection
1705 * state since it's already in use for the current command. The
1706 * connection state will get updated when pqPipelineProcessQueue()
1707 * advances to start processing the queued message.
1709 * Just make sure we can safely enqueue given the current connection
1710 * state. We can enqueue behind another queue item, or behind a
1711 * non-queue command (one that sends its own sync), but we can't
1712 * enqueue if the connection is in a copy state.
1714 switch (conn
->asyncStatus
)
1717 case PGASYNC_PIPELINE_IDLE
:
1719 case PGASYNC_READY_MORE
:
1724 case PGASYNC_COPY_IN
:
1725 case PGASYNC_COPY_OUT
:
1726 case PGASYNC_COPY_BOTH
:
1727 libpq_append_conn_error(conn
, "cannot queue commands during COPY");
1734 * This command's results will come in immediately. Initialize async
1735 * result-accumulation state
1737 pqClearAsyncResult(conn
);
1739 /* reset partial-result mode */
1740 conn
->partialResMode
= false;
1741 conn
->singleRowMode
= false;
1742 conn
->maxChunkSize
= 0;
1745 /* ready to send command message */
1751 * Common code for sending a query with extended query protocol
1752 * PQsendQueryStart should be done already
1754 * command may be NULL to indicate we use an already-prepared statement
1757 PQsendQueryGuts(PGconn
*conn
,
1758 const char *command
,
1759 const char *stmtName
,
1761 const Oid
*paramTypes
,
1762 const char *const *paramValues
,
1763 const int *paramLengths
,
1764 const int *paramFormats
,
1768 PGcmdQueueEntry
*entry
;
1770 entry
= pqAllocCmdQueueEntry(conn
);
1772 return 0; /* error msg already set */
1775 * We will send Parse (if needed), Bind, Describe Portal, Execute, Sync
1776 * (if not in pipeline mode), using specified statement name and the
1782 /* construct the Parse message */
1783 if (pqPutMsgStart(PqMsg_Parse
, conn
) < 0 ||
1784 pqPuts(stmtName
, conn
) < 0 ||
1785 pqPuts(command
, conn
) < 0)
1787 if (nParams
> 0 && paramTypes
)
1789 if (pqPutInt(nParams
, 2, conn
) < 0)
1791 for (i
= 0; i
< nParams
; i
++)
1793 if (pqPutInt(paramTypes
[i
], 4, conn
) < 0)
1799 if (pqPutInt(0, 2, conn
) < 0)
1802 if (pqPutMsgEnd(conn
) < 0)
1806 /* Construct the Bind message */
1807 if (pqPutMsgStart(PqMsg_Bind
, conn
) < 0 ||
1808 pqPuts("", conn
) < 0 ||
1809 pqPuts(stmtName
, conn
) < 0)
1812 /* Send parameter formats */
1813 if (nParams
> 0 && paramFormats
)
1815 if (pqPutInt(nParams
, 2, conn
) < 0)
1817 for (i
= 0; i
< nParams
; i
++)
1819 if (pqPutInt(paramFormats
[i
], 2, conn
) < 0)
1825 if (pqPutInt(0, 2, conn
) < 0)
1829 if (pqPutInt(nParams
, 2, conn
) < 0)
1832 /* Send parameters */
1833 for (i
= 0; i
< nParams
; i
++)
1835 if (paramValues
&& paramValues
[i
])
1839 if (paramFormats
&& paramFormats
[i
] != 0)
1841 /* binary parameter */
1843 nbytes
= paramLengths
[i
];
1846 libpq_append_conn_error(conn
, "length must be given for binary parameter");
1852 /* text parameter, do not use paramLengths */
1853 nbytes
= strlen(paramValues
[i
]);
1855 if (pqPutInt(nbytes
, 4, conn
) < 0 ||
1856 pqPutnchar(paramValues
[i
], nbytes
, conn
) < 0)
1861 /* take the param as NULL */
1862 if (pqPutInt(-1, 4, conn
) < 0)
1866 if (pqPutInt(1, 2, conn
) < 0 ||
1867 pqPutInt(resultFormat
, 2, conn
))
1869 if (pqPutMsgEnd(conn
) < 0)
1872 /* construct the Describe Portal message */
1873 if (pqPutMsgStart(PqMsg_Describe
, conn
) < 0 ||
1874 pqPutc('P', conn
) < 0 ||
1875 pqPuts("", conn
) < 0 ||
1876 pqPutMsgEnd(conn
) < 0)
1879 /* construct the Execute message */
1880 if (pqPutMsgStart(PqMsg_Execute
, conn
) < 0 ||
1881 pqPuts("", conn
) < 0 ||
1882 pqPutInt(0, 4, conn
) < 0 ||
1883 pqPutMsgEnd(conn
) < 0)
1886 /* construct the Sync message if not in pipeline mode */
1887 if (conn
->pipelineStatus
== PQ_PIPELINE_OFF
)
1889 if (pqPutMsgStart(PqMsg_Sync
, conn
) < 0 ||
1890 pqPutMsgEnd(conn
) < 0)
1894 /* remember we are using extended query protocol */
1895 entry
->queryclass
= PGQUERY_EXTENDED
;
1897 /* and remember the query text too, if possible */
1898 /* if insufficient memory, query just winds up NULL */
1900 entry
->query
= strdup(command
);
1903 * Give the data a push (in pipeline mode, only if we're past the size
1904 * threshold). In nonblock mode, don't complain if we're unable to send
1905 * it all; PQgetResult() will do any additional flushing needed.
1907 if (pqPipelineFlush(conn
) < 0)
1910 /* OK, it's launched! */
1911 pqAppendCmdQueueEntry(conn
, entry
);
1916 pqRecycleCmdQueueEntry(conn
, entry
);
1917 /* error message should be set up already */
1922 * Is it OK to change partial-result mode now?
1925 canChangeResultMode(PGconn
*conn
)
1928 * Only allow changing the mode when we have launched a query and not yet
1929 * received any results.
1933 if (conn
->asyncStatus
!= PGASYNC_BUSY
)
1935 if (!conn
->cmd_queue_head
||
1936 (conn
->cmd_queue_head
->queryclass
!= PGQUERY_SIMPLE
&&
1937 conn
->cmd_queue_head
->queryclass
!= PGQUERY_EXTENDED
))
1939 if (pgHavePendingResult(conn
))
1945 * Select row-by-row processing mode
1948 PQsetSingleRowMode(PGconn
*conn
)
1950 if (canChangeResultMode(conn
))
1952 conn
->partialResMode
= true;
1953 conn
->singleRowMode
= true;
1954 conn
->maxChunkSize
= 1;
1962 * Select chunked results processing mode
1965 PQsetChunkedRowsMode(PGconn
*conn
, int chunkSize
)
1967 if (chunkSize
> 0 && canChangeResultMode(conn
))
1969 conn
->partialResMode
= true;
1970 conn
->singleRowMode
= false;
1971 conn
->maxChunkSize
= chunkSize
;
1979 * Consume any available input from the backend
1980 * 0 return: some kind of trouble
1981 * 1 return: no problem
1984 PQconsumeInput(PGconn
*conn
)
1990 * for non-blocking connections try to flush the send-queue, otherwise we
1991 * may never get a response for something that may not have already been
1992 * sent because it's in our write buffer!
1994 if (pqIsnonblocking(conn
))
1996 if (pqFlush(conn
) < 0)
2001 * Load more data, if available. We do this no matter what state we are
2002 * in, since we are probably getting called because the application wants
2003 * to get rid of a read-select condition. Note that we will NOT block
2004 * waiting for more input.
2006 if (pqReadData(conn
) < 0)
2009 /* Parsing of the data waits till later. */
2015 * parseInput: if appropriate, parse input data from backend
2016 * until input is exhausted or a stopping state is reached.
2017 * Note that this function will NOT attempt to read more data from the backend.
2020 parseInput(PGconn
*conn
)
2022 pqParseInput3(conn
);
2027 * Return true if PQgetResult would block waiting for input.
2031 PQisBusy(PGconn
*conn
)
2036 /* Parse any available data, if our state permits. */
2040 * PQgetResult will return immediately in all states except BUSY. Also,
2041 * if we've detected read EOF and dropped the connection, we can expect
2042 * that PQgetResult will fail immediately. Note that we do *not* check
2043 * conn->write_failed here --- once that's become set, we know we have
2044 * trouble, but we need to keep trying to read until we have a complete
2045 * server message or detect read EOF.
2047 return conn
->asyncStatus
== PGASYNC_BUSY
&& conn
->status
!= CONNECTION_BAD
;
2052 * Get the next PGresult produced by a query. Returns NULL if no
2053 * query work remains or an error has occurred (e.g. out of
2056 * In pipeline mode, once all the result of a query have been returned,
2057 * PQgetResult returns NULL to let the user know that the next
2058 * query is being processed. At the end of the pipeline, returns a
2059 * result with PQresultStatus(result) == PGRES_PIPELINE_SYNC.
2062 PQgetResult(PGconn
*conn
)
2069 /* Parse any available data, if our state permits. */
2072 /* If not ready to return something, block until we are. */
2073 while (conn
->asyncStatus
== PGASYNC_BUSY
)
2078 * If data remains unsent, send it. Else we might be waiting for the
2079 * result of a command the backend hasn't even got yet.
2081 while ((flushResult
= pqFlush(conn
)) > 0)
2083 if (pqWait(false, true, conn
))
2091 * Wait for some more data, and load it. (Note: if the connection has
2092 * been lost, pqWait should return immediately because the socket
2093 * should be read-ready, either with the last server data or with an
2094 * EOF indication. We expect therefore that this won't result in any
2095 * undue delay in reporting a previous write failure.)
2098 pqWait(true, false, conn
) ||
2099 pqReadData(conn
) < 0)
2101 /* Report the error saved by pqWait or pqReadData */
2102 pqSaveErrorResult(conn
);
2103 conn
->asyncStatus
= PGASYNC_IDLE
;
2104 return pqPrepareAsyncResult(conn
);
2111 * If we had a write error, but nothing above obtained a query result
2112 * or detected a read error, report the write error.
2114 if (conn
->write_failed
&& conn
->asyncStatus
== PGASYNC_BUSY
)
2116 pqSaveWriteError(conn
);
2117 conn
->asyncStatus
= PGASYNC_IDLE
;
2118 return pqPrepareAsyncResult(conn
);
2122 /* Return the appropriate thing. */
2123 switch (conn
->asyncStatus
)
2126 res
= NULL
; /* query is complete */
2128 case PGASYNC_PIPELINE_IDLE
:
2129 Assert(conn
->pipelineStatus
!= PQ_PIPELINE_OFF
);
2132 * We're about to return the NULL that terminates the round of
2133 * results from the current query; prepare to send the results of
2134 * the next query, if any, when we're called next. If there's no
2135 * next element in the command queue, this gets us in IDLE state.
2137 pqPipelineProcessQueue(conn
);
2138 res
= NULL
; /* query is complete */
2142 res
= pqPrepareAsyncResult(conn
);
2145 * Normally pqPrepareAsyncResult will have left conn->result
2146 * empty. Otherwise, "res" must be a not-full PGRES_TUPLES_CHUNK
2147 * result, which we want to return to the caller while staying in
2148 * PGASYNC_READY state. Then the next call here will return the
2149 * empty PGRES_TUPLES_OK result that was restored from
2150 * saved_result, after which we can proceed.
2154 Assert(res
->resultStatus
== PGRES_TUPLES_CHUNK
);
2158 /* Advance the queue as appropriate */
2159 pqCommandQueueAdvance(conn
, false,
2160 res
->resultStatus
== PGRES_PIPELINE_SYNC
);
2162 if (conn
->pipelineStatus
!= PQ_PIPELINE_OFF
)
2165 * We're about to send the results of the current query. Set
2166 * us idle now, and ...
2168 conn
->asyncStatus
= PGASYNC_PIPELINE_IDLE
;
2171 * ... in cases when we're sending a pipeline-sync result,
2172 * move queue processing forwards immediately, so that next
2173 * time we're called, we're prepared to return the next result
2174 * received from the server. In all other cases, leave the
2175 * queue state change for next time, so that a terminating
2176 * NULL result is sent.
2178 * (In other words: we don't return a NULL after a pipeline
2181 if (res
->resultStatus
== PGRES_PIPELINE_SYNC
)
2182 pqPipelineProcessQueue(conn
);
2186 /* Set the state back to BUSY, allowing parsing to proceed. */
2187 conn
->asyncStatus
= PGASYNC_BUSY
;
2190 case PGASYNC_READY_MORE
:
2191 res
= pqPrepareAsyncResult(conn
);
2192 /* Set the state back to BUSY, allowing parsing to proceed. */
2193 conn
->asyncStatus
= PGASYNC_BUSY
;
2195 case PGASYNC_COPY_IN
:
2196 res
= getCopyResult(conn
, PGRES_COPY_IN
);
2198 case PGASYNC_COPY_OUT
:
2199 res
= getCopyResult(conn
, PGRES_COPY_OUT
);
2201 case PGASYNC_COPY_BOTH
:
2202 res
= getCopyResult(conn
, PGRES_COPY_BOTH
);
2205 libpq_append_conn_error(conn
, "unexpected asyncStatus: %d", (int) conn
->asyncStatus
);
2206 pqSaveErrorResult(conn
);
2207 conn
->asyncStatus
= PGASYNC_IDLE
; /* try to restore valid state */
2208 res
= pqPrepareAsyncResult(conn
);
2212 /* Time to fire PGEVT_RESULTCREATE events, if there are any */
2213 if (res
&& res
->nEvents
> 0)
2214 (void) PQfireResultCreateEvents(conn
, res
);
2221 * Helper for PQgetResult: generate result for COPY-in-progress cases
2224 getCopyResult(PGconn
*conn
, ExecStatusType copytype
)
2227 * If the server connection has been lost, don't pretend everything is
2228 * hunky-dory; instead return a PGRES_FATAL_ERROR result, and reset the
2229 * asyncStatus to idle (corresponding to what we'd do if we'd detected I/O
2230 * error in the earlier steps in PQgetResult). The text returned in the
2231 * result is whatever is in conn->errorMessage; we hope that was filled
2232 * with something relevant when the lost connection was detected.
2234 if (conn
->status
!= CONNECTION_OK
)
2236 pqSaveErrorResult(conn
);
2237 conn
->asyncStatus
= PGASYNC_IDLE
;
2238 return pqPrepareAsyncResult(conn
);
2241 /* If we have an async result for the COPY, return that */
2242 if (conn
->result
&& conn
->result
->resultStatus
== copytype
)
2243 return pqPrepareAsyncResult(conn
);
2245 /* Otherwise, invent a suitable PGresult */
2246 return PQmakeEmptyPGresult(conn
, copytype
);
2252 * send a query to the backend and package up the result in a PGresult
2254 * If the query was not even sent, return NULL; conn->errorMessage is set to
2255 * a relevant message.
2256 * If the query was sent, a new PGresult is returned (which could indicate
2257 * either success or failure).
2258 * The user is responsible for freeing the PGresult via PQclear()
2259 * when done with it.
2262 PQexec(PGconn
*conn
, const char *query
)
2264 if (!PQexecStart(conn
))
2266 if (!PQsendQuery(conn
, query
))
2268 return PQexecFinish(conn
);
2273 * Like PQexec, but use extended query protocol so we can pass parameters
2276 PQexecParams(PGconn
*conn
,
2277 const char *command
,
2279 const Oid
*paramTypes
,
2280 const char *const *paramValues
,
2281 const int *paramLengths
,
2282 const int *paramFormats
,
2285 if (!PQexecStart(conn
))
2287 if (!PQsendQueryParams(conn
, command
,
2288 nParams
, paramTypes
, paramValues
, paramLengths
,
2289 paramFormats
, resultFormat
))
2291 return PQexecFinish(conn
);
2296 * Creates a prepared statement by issuing a Parse message.
2298 * If the query was not even sent, return NULL; conn->errorMessage is set to
2299 * a relevant message.
2300 * If the query was sent, a new PGresult is returned (which could indicate
2301 * either success or failure).
2302 * The user is responsible for freeing the PGresult via PQclear()
2303 * when done with it.
2306 PQprepare(PGconn
*conn
,
2307 const char *stmtName
, const char *query
,
2308 int nParams
, const Oid
*paramTypes
)
2310 if (!PQexecStart(conn
))
2312 if (!PQsendPrepare(conn
, stmtName
, query
, nParams
, paramTypes
))
2314 return PQexecFinish(conn
);
2319 * Like PQexec, but execute a previously prepared statement,
2320 * using extended query protocol so we can pass parameters
2323 PQexecPrepared(PGconn
*conn
,
2324 const char *stmtName
,
2326 const char *const *paramValues
,
2327 const int *paramLengths
,
2328 const int *paramFormats
,
2331 if (!PQexecStart(conn
))
2333 if (!PQsendQueryPrepared(conn
, stmtName
,
2334 nParams
, paramValues
, paramLengths
,
2335 paramFormats
, resultFormat
))
2337 return PQexecFinish(conn
);
2341 * Common code for PQexec and sibling routines: prepare to send command
2344 PQexecStart(PGconn
*conn
)
2352 * Since this is the beginning of a query cycle, reset the error state.
2353 * However, in pipeline mode with something already queued, the error
2354 * buffer belongs to that command and we shouldn't clear it.
2356 if (conn
->cmd_queue_head
== NULL
)
2357 pqClearConnErrorState(conn
);
2359 if (conn
->pipelineStatus
!= PQ_PIPELINE_OFF
)
2361 libpq_append_conn_error(conn
, "synchronous command execution functions are not allowed in pipeline mode");
2366 * Silently discard any prior query result that application didn't eat.
2367 * This is probably poor design, but it's here for backward compatibility.
2369 while ((result
= PQgetResult(conn
)) != NULL
)
2371 ExecStatusType resultStatus
= result
->resultStatus
;
2373 PQclear(result
); /* only need its status */
2374 if (resultStatus
== PGRES_COPY_IN
)
2376 /* get out of a COPY IN state */
2377 if (PQputCopyEnd(conn
,
2378 libpq_gettext("COPY terminated by new PQexec")) < 0)
2380 /* keep waiting to swallow the copy's failure message */
2382 else if (resultStatus
== PGRES_COPY_OUT
)
2385 * Get out of a COPY OUT state: we just switch back to BUSY and
2386 * allow the remaining COPY data to be dropped on the floor.
2388 conn
->asyncStatus
= PGASYNC_BUSY
;
2389 /* keep waiting to swallow the copy's completion message */
2391 else if (resultStatus
== PGRES_COPY_BOTH
)
2393 /* We don't allow PQexec during COPY BOTH */
2394 libpq_append_conn_error(conn
, "PQexec not allowed during COPY BOTH");
2397 /* check for loss of connection, too */
2398 if (conn
->status
== CONNECTION_BAD
)
2402 /* OK to send a command */
2407 * Common code for PQexec and sibling routines: wait for command result
2410 PQexecFinish(PGconn
*conn
)
2413 PGresult
*lastResult
;
2416 * For backwards compatibility, return the last result if there are more
2417 * than one. (We used to have logic here to concatenate successive error
2418 * messages, but now that happens automatically, since conn->errorMessage
2419 * will continue to accumulate errors throughout this loop.)
2421 * We have to stop if we see copy in/out/both, however. We will resume
2422 * parsing after application performs the data transfer.
2424 * Also stop if the connection is lost (else we'll loop infinitely).
2427 while ((result
= PQgetResult(conn
)) != NULL
)
2429 PQclear(lastResult
);
2430 lastResult
= result
;
2431 if (result
->resultStatus
== PGRES_COPY_IN
||
2432 result
->resultStatus
== PGRES_COPY_OUT
||
2433 result
->resultStatus
== PGRES_COPY_BOTH
||
2434 conn
->status
== CONNECTION_BAD
)
2442 * PQdescribePrepared
2443 * Obtain information about a previously prepared statement
2445 * If the query was not even sent, return NULL; conn->errorMessage is set to
2446 * a relevant message.
2447 * If the query was sent, a new PGresult is returned (which could indicate
2448 * either success or failure). On success, the PGresult contains status
2449 * PGRES_COMMAND_OK, and its parameter and column-heading fields describe
2450 * the statement's inputs and outputs respectively.
2451 * The user is responsible for freeing the PGresult via PQclear()
2452 * when done with it.
2455 PQdescribePrepared(PGconn
*conn
, const char *stmt
)
2457 if (!PQexecStart(conn
))
2459 if (!PQsendTypedCommand(conn
, PqMsg_Describe
, 'S', stmt
))
2461 return PQexecFinish(conn
);
2466 * Obtain information about a previously created portal
2468 * This is much like PQdescribePrepared, except that no parameter info is
2469 * returned. Note that at the moment, libpq doesn't really expose portals
2470 * to the client; but this can be used with a portal created by a SQL
2471 * DECLARE CURSOR command.
2474 PQdescribePortal(PGconn
*conn
, const char *portal
)
2476 if (!PQexecStart(conn
))
2478 if (!PQsendTypedCommand(conn
, PqMsg_Describe
, 'P', portal
))
2480 return PQexecFinish(conn
);
2484 * PQsendDescribePrepared
2485 * Submit a Describe Statement command, but don't wait for it to finish
2487 * Returns: 1 if successfully submitted
2488 * 0 if error (conn->errorMessage is set)
2491 PQsendDescribePrepared(PGconn
*conn
, const char *stmt
)
2493 return PQsendTypedCommand(conn
, PqMsg_Describe
, 'S', stmt
);
2497 * PQsendDescribePortal
2498 * Submit a Describe Portal command, but don't wait for it to finish
2500 * Returns: 1 if successfully submitted
2501 * 0 if error (conn->errorMessage is set)
2504 PQsendDescribePortal(PGconn
*conn
, const char *portal
)
2506 return PQsendTypedCommand(conn
, PqMsg_Describe
, 'P', portal
);
2511 * Close a previously prepared statement
2513 * If the query was not even sent, return NULL; conn->errorMessage is set to
2514 * a relevant message.
2515 * If the query was sent, a new PGresult is returned (which could indicate
2516 * either success or failure). On success, the PGresult contains status
2517 * PGRES_COMMAND_OK. The user is responsible for freeing the PGresult via
2518 * PQclear() when done with it.
2521 PQclosePrepared(PGconn
*conn
, const char *stmt
)
2523 if (!PQexecStart(conn
))
2525 if (!PQsendTypedCommand(conn
, PqMsg_Close
, 'S', stmt
))
2527 return PQexecFinish(conn
);
2532 * Close a previously created portal
2534 * This is exactly like PQclosePrepared, but for portals. Note that at the
2535 * moment, libpq doesn't really expose portals to the client; but this can be
2536 * used with a portal created by a SQL DECLARE CURSOR command.
2539 PQclosePortal(PGconn
*conn
, const char *portal
)
2541 if (!PQexecStart(conn
))
2543 if (!PQsendTypedCommand(conn
, PqMsg_Close
, 'P', portal
))
2545 return PQexecFinish(conn
);
2549 * PQsendClosePrepared
2550 * Submit a Close Statement command, but don't wait for it to finish
2552 * Returns: 1 if successfully submitted
2553 * 0 if error (conn->errorMessage is set)
2556 PQsendClosePrepared(PGconn
*conn
, const char *stmt
)
2558 return PQsendTypedCommand(conn
, PqMsg_Close
, 'S', stmt
);
2563 * Submit a Close Portal command, but don't wait for it to finish
2565 * Returns: 1 if successfully submitted
2566 * 0 if error (conn->errorMessage is set)
2569 PQsendClosePortal(PGconn
*conn
, const char *portal
)
2571 return PQsendTypedCommand(conn
, PqMsg_Close
, 'P', portal
);
2575 * PQsendTypedCommand
2576 * Common code to send a Describe or Close command
2578 * Available options for "command" are
2579 * PqMsg_Close for Close; or
2580 * PqMsg_Describe for Describe.
2582 * Available options for "type" are
2583 * 'S' to run a command on a prepared statement; or
2584 * 'P' to run a command on a portal.
2586 * Returns 1 on success and 0 on failure.
2589 PQsendTypedCommand(PGconn
*conn
, char command
, char type
, const char *target
)
2591 PGcmdQueueEntry
*entry
= NULL
;
2593 /* Treat null target as empty string */
2597 if (!PQsendQueryStart(conn
, true))
2600 entry
= pqAllocCmdQueueEntry(conn
);
2602 return 0; /* error msg already set */
2604 /* construct the Close message */
2605 if (pqPutMsgStart(command
, conn
) < 0 ||
2606 pqPutc(type
, conn
) < 0 ||
2607 pqPuts(target
, conn
) < 0 ||
2608 pqPutMsgEnd(conn
) < 0)
2611 /* construct the Sync message */
2612 if (conn
->pipelineStatus
== PQ_PIPELINE_OFF
)
2614 if (pqPutMsgStart(PqMsg_Sync
, conn
) < 0 ||
2615 pqPutMsgEnd(conn
) < 0)
2619 /* remember if we are doing a Close or a Describe */
2620 if (command
== PqMsg_Close
)
2622 entry
->queryclass
= PGQUERY_CLOSE
;
2624 else if (command
== PqMsg_Describe
)
2626 entry
->queryclass
= PGQUERY_DESCRIBE
;
2630 libpq_append_conn_error(conn
, "unrecognized message type \"%c\"", command
);
2635 * Give the data a push (in pipeline mode, only if we're past the size
2636 * threshold). In nonblock mode, don't complain if we're unable to send
2637 * it all; PQgetResult() will do any additional flushing needed.
2639 if (pqPipelineFlush(conn
) < 0)
2642 /* OK, it's launched! */
2643 pqAppendCmdQueueEntry(conn
, entry
);
2648 pqRecycleCmdQueueEntry(conn
, entry
);
2649 /* error message should be set up already */
2655 * returns a PGnotify* structure of the latest async notification
2656 * that has not yet been handled
2658 * returns NULL, if there is currently
2659 * no unhandled async notification from the backend
2661 * the CALLER is responsible for FREE'ing the structure returned
2663 * Note that this function does not read any new data from the socket;
2664 * so usually, caller should call PQconsumeInput() first.
2667 PQnotifies(PGconn
*conn
)
2674 /* Parse any available data to see if we can extract NOTIFY messages. */
2677 event
= conn
->notifyHead
;
2680 conn
->notifyHead
= event
->next
;
2681 if (!conn
->notifyHead
)
2682 conn
->notifyTail
= NULL
;
2683 event
->next
= NULL
; /* don't let app see the internal state */
2689 * PQputCopyData - send some data to the backend during COPY IN or COPY BOTH
2691 * Returns 1 if successful, 0 if data could not be sent (only possible
2692 * in nonblock mode), or -1 if an error occurs.
2695 PQputCopyData(PGconn
*conn
, const char *buffer
, int nbytes
)
2699 if (conn
->asyncStatus
!= PGASYNC_COPY_IN
&&
2700 conn
->asyncStatus
!= PGASYNC_COPY_BOTH
)
2702 libpq_append_conn_error(conn
, "no COPY in progress");
2707 * Process any NOTICE or NOTIFY messages that might be pending in the
2708 * input buffer. Since the server might generate many notices during the
2709 * COPY, we want to clean those out reasonably promptly to prevent
2710 * indefinite expansion of the input buffer. (Note: the actual read of
2711 * input data into the input buffer happens down inside pqSendSome, but
2712 * it's not authorized to get rid of the data again.)
2719 * Try to flush any previously sent data in preference to growing the
2720 * output buffer. If we can't enlarge the buffer enough to hold the
2721 * data, return 0 in the nonblock case, else hard error. (For
2722 * simplicity, always assume 5 bytes of overhead.)
2724 if ((conn
->outBufSize
- conn
->outCount
- 5) < nbytes
)
2726 if (pqFlush(conn
) < 0)
2728 if (pqCheckOutBufferSpace(conn
->outCount
+ 5 + (size_t) nbytes
,
2730 return pqIsnonblocking(conn
) ? 0 : -1;
2732 /* Send the data (too simple to delegate to fe-protocol files) */
2733 if (pqPutMsgStart(PqMsg_CopyData
, conn
) < 0 ||
2734 pqPutnchar(buffer
, nbytes
, conn
) < 0 ||
2735 pqPutMsgEnd(conn
) < 0)
2742 * PQputCopyEnd - send EOF indication to the backend during COPY IN
2744 * After calling this, use PQgetResult() to check command completion status.
2746 * Returns 1 if successful, or -1 if an error occurs.
2749 PQputCopyEnd(PGconn
*conn
, const char *errormsg
)
2753 if (conn
->asyncStatus
!= PGASYNC_COPY_IN
&&
2754 conn
->asyncStatus
!= PGASYNC_COPY_BOTH
)
2756 libpq_append_conn_error(conn
, "no COPY in progress");
2761 * Send the COPY END indicator. This is simple enough that we don't
2762 * bother delegating it to the fe-protocol files.
2766 /* Send COPY FAIL */
2767 if (pqPutMsgStart(PqMsg_CopyFail
, conn
) < 0 ||
2768 pqPuts(errormsg
, conn
) < 0 ||
2769 pqPutMsgEnd(conn
) < 0)
2774 /* Send COPY DONE */
2775 if (pqPutMsgStart(PqMsg_CopyDone
, conn
) < 0 ||
2776 pqPutMsgEnd(conn
) < 0)
2781 * If we sent the COPY command in extended-query mode, we must issue a
2784 if (conn
->cmd_queue_head
&&
2785 conn
->cmd_queue_head
->queryclass
!= PGQUERY_SIMPLE
)
2787 if (pqPutMsgStart(PqMsg_Sync
, conn
) < 0 ||
2788 pqPutMsgEnd(conn
) < 0)
2792 /* Return to active duty */
2793 if (conn
->asyncStatus
== PGASYNC_COPY_BOTH
)
2794 conn
->asyncStatus
= PGASYNC_COPY_OUT
;
2796 conn
->asyncStatus
= PGASYNC_BUSY
;
2798 /* Try to flush data */
2799 if (pqFlush(conn
) < 0)
2806 * PQgetCopyData - read a row of data from the backend during COPY OUT
2809 * If successful, sets *buffer to point to a malloc'd row of data, and
2810 * returns row length (always > 0) as result.
2811 * Returns 0 if no row available yet (only possible if async is true),
2812 * -1 if end of copy (consult PQgetResult), or -2 if error (consult
2816 PQgetCopyData(PGconn
*conn
, char **buffer
, int async
)
2818 *buffer
= NULL
; /* for all failure cases */
2821 if (conn
->asyncStatus
!= PGASYNC_COPY_OUT
&&
2822 conn
->asyncStatus
!= PGASYNC_COPY_BOTH
)
2824 libpq_append_conn_error(conn
, "no COPY in progress");
2827 return pqGetCopyData3(conn
, buffer
, async
);
2831 * PQgetline - gets a newline-terminated string from the backend.
2833 * Chiefly here so that applications can use "COPY <rel> to stdout"
2834 * and read the output string. Returns a null-terminated string in `buffer`.
2836 * XXX this routine is now deprecated, because it can't handle binary data.
2837 * If called during a COPY BINARY we return EOF.
2839 * PQgetline reads up to `length`-1 characters (like fgets(3)) but strips
2840 * the terminating \n (like gets(3)).
2842 * CAUTION: the caller is responsible for detecting the end-of-copy signal
2843 * (a line containing just "\.") when using this routine.
2846 * EOF if error (eg, invalid arguments are given)
2847 * 0 if EOL is reached (i.e., \n has been read)
2848 * (this is required for backward-compatibility -- this
2849 * routine used to always return EOF or 0, assuming that
2850 * the line ended within `length` bytes.)
2851 * 1 in other cases (i.e., the buffer was filled before \n is reached)
2854 PQgetline(PGconn
*conn
, char *buffer
, int length
)
2856 if (!buffer
|| length
<= 0)
2859 /* length must be at least 3 to hold the \. terminator! */
2866 return pqGetline3(conn
, buffer
, length
);
2870 * PQgetlineAsync - gets a COPY data row without blocking.
2872 * This routine is for applications that want to do "COPY <rel> to stdout"
2873 * asynchronously, that is without blocking. Having issued the COPY command
2874 * and gotten a PGRES_COPY_OUT response, the app should call PQconsumeInput
2875 * and this routine until the end-of-data signal is detected. Unlike
2876 * PQgetline, this routine takes responsibility for detecting end-of-data.
2878 * On each call, PQgetlineAsync will return data if a complete data row
2879 * is available in libpq's input buffer. Otherwise, no data is returned
2880 * until the rest of the row arrives.
2882 * If -1 is returned, the end-of-data signal has been recognized (and removed
2883 * from libpq's input buffer). The caller *must* next call PQendcopy and
2884 * then return to normal processing.
2887 * -1 if the end-of-copy-data marker has been recognized
2888 * 0 if no data is available
2889 * >0 the number of bytes returned.
2891 * The data returned will not extend beyond a data-row boundary. If possible
2892 * a whole row will be returned at one time. But if the buffer offered by
2893 * the caller is too small to hold a row sent by the backend, then a partial
2894 * data row will be returned. In text mode this can be detected by testing
2895 * whether the last returned byte is '\n' or not.
2897 * The returned data is *not* null-terminated.
2901 PQgetlineAsync(PGconn
*conn
, char *buffer
, int bufsize
)
2906 return pqGetlineAsync3(conn
, buffer
, bufsize
);
2910 * PQputline -- sends a string to the backend during COPY IN.
2911 * Returns 0 if OK, EOF if not.
2913 * This is deprecated primarily because the return convention doesn't allow
2914 * caller to tell the difference between a hard error and a nonblock-mode
2918 PQputline(PGconn
*conn
, const char *string
)
2920 return PQputnbytes(conn
, string
, strlen(string
));
2924 * PQputnbytes -- like PQputline, but buffer need not be null-terminated.
2925 * Returns 0 if OK, EOF if not.
2928 PQputnbytes(PGconn
*conn
, const char *buffer
, int nbytes
)
2930 if (PQputCopyData(conn
, buffer
, nbytes
) > 0)
2938 * After completing the data transfer portion of a copy in/out,
2939 * the application must call this routine to finish the command protocol.
2941 * This is deprecated; it's cleaner to use PQgetResult to get the transfer
2949 PQendcopy(PGconn
*conn
)
2954 return pqEndcopy3(conn
);
2959 * PQfn - Send a function call to the POSTGRES backend.
2961 * conn : backend connection
2962 * fnid : OID of function to be called
2963 * result_buf : pointer to result buffer
2964 * result_len : actual length of result is returned here
2965 * result_is_int : If the result is an integer, this must be 1,
2966 * otherwise this should be 0
2967 * args : pointer to an array of function arguments
2968 * (each has length, if integer, and value/pointer)
2969 * nargs : # of arguments in args array.
2972 * PGresult with status = PGRES_COMMAND_OK if successful.
2973 * *result_len is > 0 if there is a return value, 0 if not.
2974 * PGresult with status = PGRES_FATAL_ERROR if backend returns an error.
2975 * NULL on communications failure. conn->errorMessage will be set.
2985 const PQArgBlock
*args
,
2994 * Since this is the beginning of a query cycle, reset the error state.
2995 * However, in pipeline mode with something already queued, the error
2996 * buffer belongs to that command and we shouldn't clear it.
2998 if (conn
->cmd_queue_head
== NULL
)
2999 pqClearConnErrorState(conn
);
3001 if (conn
->pipelineStatus
!= PQ_PIPELINE_OFF
)
3003 libpq_append_conn_error(conn
, "%s not allowed in pipeline mode", "PQfn");
3007 if (conn
->sock
== PGINVALID_SOCKET
|| conn
->asyncStatus
!= PGASYNC_IDLE
||
3008 pgHavePendingResult(conn
))
3010 libpq_append_conn_error(conn
, "connection in wrong state");
3014 return pqFunctionCall3(conn
, fnid
,
3015 result_buf
, result_len
,
3020 /* ====== Pipeline mode support ======== */
3023 * PQenterPipelineMode
3024 * Put an idle connection in pipeline mode.
3026 * Returns 1 on success. On failure, errorMessage is set and 0 is returned.
3028 * Commands submitted after this can be pipelined on the connection;
3029 * there's no requirement to wait for one to finish before the next is
3032 * Queuing of a new query or syncing during COPY is not allowed.
3034 * A set of commands is terminated by a PQpipelineSync. Multiple sync
3035 * points can be established while in pipeline mode. Pipeline mode can
3036 * be exited by calling PQexitPipelineMode() once all results are processed.
3038 * This doesn't actually send anything on the wire, it just puts libpq
3039 * into a state where it can pipeline work.
3042 PQenterPipelineMode(PGconn
*conn
)
3047 /* succeed with no action if already in pipeline mode */
3048 if (conn
->pipelineStatus
!= PQ_PIPELINE_OFF
)
3051 if (conn
->asyncStatus
!= PGASYNC_IDLE
)
3053 libpq_append_conn_error(conn
, "cannot enter pipeline mode, connection not idle");
3057 conn
->pipelineStatus
= PQ_PIPELINE_ON
;
3063 * PQexitPipelineMode
3064 * End pipeline mode and return to normal command mode.
3066 * Returns 1 in success (pipeline mode successfully ended, or not in pipeline
3069 * Returns 0 if in pipeline mode and cannot be ended yet. Error message will
3073 PQexitPipelineMode(PGconn
*conn
)
3078 if (conn
->pipelineStatus
== PQ_PIPELINE_OFF
&&
3079 (conn
->asyncStatus
== PGASYNC_IDLE
||
3080 conn
->asyncStatus
== PGASYNC_PIPELINE_IDLE
) &&
3081 conn
->cmd_queue_head
== NULL
)
3084 switch (conn
->asyncStatus
)
3087 case PGASYNC_READY_MORE
:
3088 /* there are some uncollected results */
3089 libpq_append_conn_error(conn
, "cannot exit pipeline mode with uncollected results");
3093 libpq_append_conn_error(conn
, "cannot exit pipeline mode while busy");
3097 case PGASYNC_PIPELINE_IDLE
:
3101 case PGASYNC_COPY_IN
:
3102 case PGASYNC_COPY_OUT
:
3103 case PGASYNC_COPY_BOTH
:
3104 libpq_append_conn_error(conn
, "cannot exit pipeline mode while in COPY");
3107 /* still work to process */
3108 if (conn
->cmd_queue_head
!= NULL
)
3110 libpq_append_conn_error(conn
, "cannot exit pipeline mode with uncollected results");
3114 conn
->pipelineStatus
= PQ_PIPELINE_OFF
;
3115 conn
->asyncStatus
= PGASYNC_IDLE
;
3117 /* Flush any pending data in out buffer */
3118 if (pqFlush(conn
) < 0)
3119 return 0; /* error message is setup already */
3124 * pqCommandQueueAdvance
3125 * Remove one query from the command queue, if appropriate.
3127 * If we have received all results corresponding to the head element
3128 * in the command queue, remove it.
3130 * In simple query protocol we must not advance the command queue until the
3131 * ReadyForQuery message has been received. This is because in simple mode a
3132 * command can have multiple queries, and we must process result for all of
3133 * them before moving on to the next command.
3135 * Another consideration is synchronization during error processing in
3136 * extended query protocol: we refuse to advance the queue past a SYNC queue
3137 * element, unless the result we've received is also a SYNC. In particular
3138 * this protects us from advancing when an error is received at an
3139 * inappropriate moment.
3142 pqCommandQueueAdvance(PGconn
*conn
, bool isReadyForQuery
, bool gotSync
)
3144 PGcmdQueueEntry
*prevquery
;
3146 if (conn
->cmd_queue_head
== NULL
)
3150 * If processing a query of simple query protocol, we only advance the
3151 * queue when we receive the ReadyForQuery message for it.
3153 if (conn
->cmd_queue_head
->queryclass
== PGQUERY_SIMPLE
&& !isReadyForQuery
)
3157 * If we're waiting for a SYNC, don't advance the queue until we get one.
3159 if (conn
->cmd_queue_head
->queryclass
== PGQUERY_SYNC
&& !gotSync
)
3162 /* delink element from queue */
3163 prevquery
= conn
->cmd_queue_head
;
3164 conn
->cmd_queue_head
= conn
->cmd_queue_head
->next
;
3166 /* If the queue is now empty, reset the tail too */
3167 if (conn
->cmd_queue_head
== NULL
)
3168 conn
->cmd_queue_tail
= NULL
;
3170 /* and make the queue element recyclable */
3171 prevquery
->next
= NULL
;
3172 pqRecycleCmdQueueEntry(conn
, prevquery
);
3176 * pqPipelineProcessQueue: subroutine for PQgetResult
3177 * In pipeline mode, start processing the results of the next query in the queue.
3180 pqPipelineProcessQueue(PGconn
*conn
)
3182 switch (conn
->asyncStatus
)
3184 case PGASYNC_COPY_IN
:
3185 case PGASYNC_COPY_OUT
:
3186 case PGASYNC_COPY_BOTH
:
3188 case PGASYNC_READY_MORE
:
3190 /* client still has to process current query or results */
3196 * If we're in IDLE mode and there's some command in the queue,
3197 * get us into PIPELINE_IDLE mode and process normally. Otherwise
3198 * there's nothing for us to do.
3200 if (conn
->cmd_queue_head
!= NULL
)
3202 conn
->asyncStatus
= PGASYNC_PIPELINE_IDLE
;
3207 case PGASYNC_PIPELINE_IDLE
:
3208 Assert(conn
->pipelineStatus
!= PQ_PIPELINE_OFF
);
3209 /* next query please */
3214 * Reset partial-result mode. (Client has to set it up for each query, if
3217 conn
->partialResMode
= false;
3218 conn
->singleRowMode
= false;
3219 conn
->maxChunkSize
= 0;
3222 * If there are no further commands to process in the queue, get us in
3223 * "real idle" mode now.
3225 if (conn
->cmd_queue_head
== NULL
)
3227 conn
->asyncStatus
= PGASYNC_IDLE
;
3232 * Reset the error state. This and the next couple of steps correspond to
3233 * what PQsendQueryStart didn't do for this query.
3235 pqClearConnErrorState(conn
);
3237 /* Initialize async result-accumulation state */
3238 pqClearAsyncResult(conn
);
3240 if (conn
->pipelineStatus
== PQ_PIPELINE_ABORTED
&&
3241 conn
->cmd_queue_head
->queryclass
!= PGQUERY_SYNC
)
3244 * In an aborted pipeline we don't get anything from the server for
3245 * each result; we're just discarding commands from the queue until we
3246 * get to the next sync from the server.
3248 * The PGRES_PIPELINE_ABORTED results tell the client that its queries
3251 conn
->result
= PQmakeEmptyPGresult(conn
, PGRES_PIPELINE_ABORTED
);
3254 libpq_append_conn_error(conn
, "out of memory");
3255 pqSaveErrorResult(conn
);
3258 conn
->asyncStatus
= PGASYNC_READY
;
3262 /* allow parsing to continue */
3263 conn
->asyncStatus
= PGASYNC_BUSY
;
3269 * Send a Sync message as part of a pipeline, and flush to server
3272 PQpipelineSync(PGconn
*conn
)
3274 return pqPipelineSyncInternal(conn
, true);
3278 * PQsendPipelineSync
3279 * Send a Sync message as part of a pipeline, without flushing to server
3282 PQsendPipelineSync(PGconn
*conn
)
3284 return pqPipelineSyncInternal(conn
, false);
3288 * Workhorse function for PQpipelineSync and PQsendPipelineSync.
3290 * immediate_flush controls if the flush happens immediately after sending the
3291 * Sync message or not.
3294 pqPipelineSyncInternal(PGconn
*conn
, bool immediate_flush
)
3296 PGcmdQueueEntry
*entry
;
3301 if (conn
->pipelineStatus
== PQ_PIPELINE_OFF
)
3303 libpq_append_conn_error(conn
, "cannot send pipeline when not in pipeline mode");
3307 switch (conn
->asyncStatus
)
3309 case PGASYNC_COPY_IN
:
3310 case PGASYNC_COPY_OUT
:
3311 case PGASYNC_COPY_BOTH
:
3312 /* should be unreachable */
3313 appendPQExpBufferStr(&conn
->errorMessage
,
3314 "internal error: cannot send pipeline while in COPY\n");
3317 case PGASYNC_READY_MORE
:
3320 case PGASYNC_PIPELINE_IDLE
:
3321 /* OK to send sync */
3325 entry
= pqAllocCmdQueueEntry(conn
);
3327 return 0; /* error msg already set */
3329 entry
->queryclass
= PGQUERY_SYNC
;
3330 entry
->query
= NULL
;
3332 /* construct the Sync message */
3333 if (pqPutMsgStart(PqMsg_Sync
, conn
) < 0 ||
3334 pqPutMsgEnd(conn
) < 0)
3338 * Give the data a push. In nonblock mode, don't complain if we're unable
3339 * to send it all; PQgetResult() will do any additional flushing needed.
3340 * If immediate_flush is disabled, the data is pushed if we are past the
3343 if (immediate_flush
)
3345 if (pqFlush(conn
) < 0)
3350 if (pqPipelineFlush(conn
) < 0)
3354 /* OK, it's launched! */
3355 pqAppendCmdQueueEntry(conn
, entry
);
3360 pqRecycleCmdQueueEntry(conn
, entry
);
3361 /* error message should be set up already */
3366 * PQsendFlushRequest
3367 * Send request for server to flush its buffer. Useful in pipeline
3368 * mode when a sync point is not desired.
3371 PQsendFlushRequest(PGconn
*conn
)
3376 /* Don't try to send if we know there's no live connection. */
3377 if (conn
->status
!= CONNECTION_OK
)
3379 libpq_append_conn_error(conn
, "no connection to the server");
3383 /* Can't send while already busy, either, unless enqueuing for later */
3384 if (conn
->asyncStatus
!= PGASYNC_IDLE
&&
3385 conn
->pipelineStatus
== PQ_PIPELINE_OFF
)
3387 libpq_append_conn_error(conn
, "another command is already in progress");
3391 if (pqPutMsgStart(PqMsg_Flush
, conn
) < 0 ||
3392 pqPutMsgEnd(conn
) < 0)
3398 * Give the data a push (in pipeline mode, only if we're past the size
3399 * threshold). In nonblock mode, don't complain if we're unable to send
3400 * it all; PQgetResult() will do any additional flushing needed.
3402 if (pqPipelineFlush(conn
) < 0)
3408 /* ====== accessor funcs for PGresult ======== */
3411 PQresultStatus(const PGresult
*res
)
3414 return PGRES_FATAL_ERROR
;
3415 return res
->resultStatus
;
3419 PQresStatus(ExecStatusType status
)
3421 if ((unsigned int) status
>= lengthof(pgresStatus
))
3422 return libpq_gettext("invalid ExecStatusType code");
3423 return pgresStatus
[status
];
3427 PQresultErrorMessage(const PGresult
*res
)
3429 if (!res
|| !res
->errMsg
)
3435 PQresultVerboseErrorMessage(const PGresult
*res
,
3436 PGVerbosity verbosity
,
3437 PGContextVisibility show_context
)
3439 PQExpBufferData workBuf
;
3442 * Because the caller is expected to free the result string, we must
3443 * strdup any constant result. We use plain strdup and document that
3444 * callers should expect NULL if out-of-memory.
3447 (res
->resultStatus
!= PGRES_FATAL_ERROR
&&
3448 res
->resultStatus
!= PGRES_NONFATAL_ERROR
))
3449 return strdup(libpq_gettext("PGresult is not an error result\n"));
3451 initPQExpBuffer(&workBuf
);
3453 pqBuildErrorMessage3(&workBuf
, res
, verbosity
, show_context
);
3455 /* If insufficient memory to format the message, fail cleanly */
3456 if (PQExpBufferDataBroken(workBuf
))
3458 termPQExpBuffer(&workBuf
);
3459 return strdup(libpq_gettext("out of memory\n"));
3462 return workBuf
.data
;
3466 PQresultErrorField(const PGresult
*res
, int fieldcode
)
3468 PGMessageField
*pfield
;
3472 for (pfield
= res
->errFields
; pfield
!= NULL
; pfield
= pfield
->next
)
3474 if (pfield
->code
== fieldcode
)
3475 return pfield
->contents
;
3481 PQntuples(const PGresult
*res
)
3489 PQnfields(const PGresult
*res
)
3493 return res
->numAttributes
;
3497 PQbinaryTuples(const PGresult
*res
)
3505 * Helper routines to range-check field numbers and tuple numbers.
3506 * Return true if OK, false if not
3510 check_field_number(const PGresult
*res
, int field_num
)
3513 return false; /* no way to display error message... */
3514 if (field_num
< 0 || field_num
>= res
->numAttributes
)
3516 pqInternalNotice(&res
->noticeHooks
,
3517 "column number %d is out of range 0..%d",
3518 field_num
, res
->numAttributes
- 1);
3525 check_tuple_field_number(const PGresult
*res
,
3526 int tup_num
, int field_num
)
3529 return false; /* no way to display error message... */
3530 if (tup_num
< 0 || tup_num
>= res
->ntups
)
3532 pqInternalNotice(&res
->noticeHooks
,
3533 "row number %d is out of range 0..%d",
3534 tup_num
, res
->ntups
- 1);
3537 if (field_num
< 0 || field_num
>= res
->numAttributes
)
3539 pqInternalNotice(&res
->noticeHooks
,
3540 "column number %d is out of range 0..%d",
3541 field_num
, res
->numAttributes
- 1);
3548 check_param_number(const PGresult
*res
, int param_num
)
3551 return false; /* no way to display error message... */
3552 if (param_num
< 0 || param_num
>= res
->numParameters
)
3554 pqInternalNotice(&res
->noticeHooks
,
3555 "parameter number %d is out of range 0..%d",
3556 param_num
, res
->numParameters
- 1);
3564 * returns NULL if the field_num is invalid
3567 PQfname(const PGresult
*res
, int field_num
)
3569 if (!check_field_number(res
, field_num
))
3572 return res
->attDescs
[field_num
].name
;
3578 * PQfnumber: find column number given column name
3580 * The column name is parsed as if it were in a SQL statement, including
3581 * case-folding and double-quote processing. But note a possible gotcha:
3582 * downcasing in the frontend might follow different locale rules than
3583 * downcasing in the backend...
3585 * Returns -1 if no match. In the present backend it is also possible
3586 * to have multiple matches, in which case the first one is found.
3589 PQfnumber(const PGresult
*res
, const char *field_name
)
3593 bool all_lower
= true;
3602 * Note: it is correct to reject a zero-length input string; the proper
3603 * input to match a zero-length field name would be "".
3605 if (field_name
== NULL
||
3606 field_name
[0] == '\0' ||
3607 res
->attDescs
== NULL
)
3611 * Check if we can avoid the strdup() and related work because the
3612 * passed-in string wouldn't be changed before we do the check anyway.
3614 for (iptr
= field_name
; *iptr
; iptr
++)
3618 if (c
== '"' || c
!= pg_tolower((unsigned char) c
))
3626 for (i
= 0; i
< res
->numAttributes
; i
++)
3627 if (strcmp(field_name
, res
->attDescs
[i
].name
) == 0)
3630 /* Fall through to the normal check if that didn't work out. */
3633 * Note: this code will not reject partially quoted strings, eg
3634 * foo"BAR"foo will become fooBARfoo when it probably ought to be an error
3637 field_case
= strdup(field_name
);
3638 if (field_case
== NULL
)
3639 return -1; /* grotty */
3643 for (iptr
= field_case
; *iptr
; iptr
++)
3653 /* doubled quotes become a single quote */
3667 c
= pg_tolower((unsigned char) c
);
3673 for (i
= 0; i
< res
->numAttributes
; i
++)
3675 if (strcmp(field_case
, res
->attDescs
[i
].name
) == 0)
3686 PQftable(const PGresult
*res
, int field_num
)
3688 if (!check_field_number(res
, field_num
))
3691 return res
->attDescs
[field_num
].tableid
;
3697 PQftablecol(const PGresult
*res
, int field_num
)
3699 if (!check_field_number(res
, field_num
))
3702 return res
->attDescs
[field_num
].columnid
;
3708 PQfformat(const PGresult
*res
, int field_num
)
3710 if (!check_field_number(res
, field_num
))
3713 return res
->attDescs
[field_num
].format
;
3719 PQftype(const PGresult
*res
, int field_num
)
3721 if (!check_field_number(res
, field_num
))
3724 return res
->attDescs
[field_num
].typid
;
3730 PQfsize(const PGresult
*res
, int field_num
)
3732 if (!check_field_number(res
, field_num
))
3735 return res
->attDescs
[field_num
].typlen
;
3741 PQfmod(const PGresult
*res
, int field_num
)
3743 if (!check_field_number(res
, field_num
))
3746 return res
->attDescs
[field_num
].atttypmod
;
3752 PQcmdStatus(PGresult
*res
)
3756 return res
->cmdStatus
;
3761 * if the last command was an INSERT, return the oid string
3765 PQoidStatus(const PGresult
*res
)
3768 * This must be enough to hold the result. Don't laugh, this is better
3769 * than what this function used to do.
3771 static char buf
[24];
3775 if (!res
|| strncmp(res
->cmdStatus
, "INSERT ", 7) != 0)
3778 len
= strspn(res
->cmdStatus
+ 7, "0123456789");
3779 if (len
> sizeof(buf
) - 1)
3780 len
= sizeof(buf
) - 1;
3781 memcpy(buf
, res
->cmdStatus
+ 7, len
);
3789 * a perhaps preferable form of the above which just returns
3793 PQoidValue(const PGresult
*res
)
3795 char *endptr
= NULL
;
3796 unsigned long result
;
3799 strncmp(res
->cmdStatus
, "INSERT ", 7) != 0 ||
3800 res
->cmdStatus
[7] < '0' ||
3801 res
->cmdStatus
[7] > '9')
3804 result
= strtoul(res
->cmdStatus
+ 7, &endptr
, 10);
3806 if (!endptr
|| (*endptr
!= ' ' && *endptr
!= '\0'))
3809 return (Oid
) result
;
3815 * If the last command was INSERT/UPDATE/DELETE/MERGE/MOVE/FETCH/COPY,
3816 * return a string containing the number of inserted/affected tuples.
3817 * If not, return "".
3819 * XXX: this should probably return an int
3822 PQcmdTuples(PGresult
*res
)
3830 if (strncmp(res
->cmdStatus
, "INSERT ", 7) == 0)
3832 p
= res
->cmdStatus
+ 7;
3833 /* INSERT: skip oid and space */
3834 while (*p
&& *p
!= ' ')
3837 goto interpret_error
; /* no space? */
3840 else if (strncmp(res
->cmdStatus
, "SELECT ", 7) == 0 ||
3841 strncmp(res
->cmdStatus
, "DELETE ", 7) == 0 ||
3842 strncmp(res
->cmdStatus
, "UPDATE ", 7) == 0)
3843 p
= res
->cmdStatus
+ 7;
3844 else if (strncmp(res
->cmdStatus
, "FETCH ", 6) == 0 ||
3845 strncmp(res
->cmdStatus
, "MERGE ", 6) == 0)
3846 p
= res
->cmdStatus
+ 6;
3847 else if (strncmp(res
->cmdStatus
, "MOVE ", 5) == 0 ||
3848 strncmp(res
->cmdStatus
, "COPY ", 5) == 0)
3849 p
= res
->cmdStatus
+ 5;
3853 /* check that we have an integer (at least one digit, nothing else) */
3854 for (c
= p
; *c
; c
++)
3856 if (!isdigit((unsigned char) *c
))
3857 goto interpret_error
;
3860 goto interpret_error
;
3865 pqInternalNotice(&res
->noticeHooks
,
3866 "could not interpret result from server: %s",
3873 * return the value of field 'field_num' of row 'tup_num'
3876 PQgetvalue(const PGresult
*res
, int tup_num
, int field_num
)
3878 if (!check_tuple_field_number(res
, tup_num
, field_num
))
3880 return res
->tuples
[tup_num
][field_num
].value
;
3884 * returns the actual length of a field value in bytes.
3887 PQgetlength(const PGresult
*res
, int tup_num
, int field_num
)
3889 if (!check_tuple_field_number(res
, tup_num
, field_num
))
3891 if (res
->tuples
[tup_num
][field_num
].len
!= NULL_LEN
)
3892 return res
->tuples
[tup_num
][field_num
].len
;
3898 * returns the null status of a field value.
3901 PQgetisnull(const PGresult
*res
, int tup_num
, int field_num
)
3903 if (!check_tuple_field_number(res
, tup_num
, field_num
))
3904 return 1; /* pretend it is null */
3905 if (res
->tuples
[tup_num
][field_num
].len
== NULL_LEN
)
3912 * returns the number of input parameters of a prepared statement.
3915 PQnparams(const PGresult
*res
)
3919 return res
->numParameters
;
3923 * returns type Oid of the specified statement parameter.
3926 PQparamtype(const PGresult
*res
, int param_num
)
3928 if (!check_param_number(res
, param_num
))
3930 if (res
->paramDescs
)
3931 return res
->paramDescs
[param_num
].typid
;
3937 /* PQsetnonblocking:
3938 * sets the PGconn's database connection non-blocking if the arg is true
3939 * or makes it blocking if the arg is false, this will not protect
3940 * you from PQexec(), you'll only be safe when using the non-blocking API.
3941 * Needs to be called only on a connected database connection.
3944 PQsetnonblocking(PGconn
*conn
, int arg
)
3948 if (!conn
|| conn
->status
== CONNECTION_BAD
)
3951 barg
= (arg
? true : false);
3953 /* early out if the socket is already in the state requested */
3954 if (barg
== conn
->nonblocking
)
3958 * to guarantee constancy for flushing/query/result-polling behavior we
3959 * need to flush the send queue at this point in order to guarantee proper
3960 * behavior. this is ok because either they are making a transition _from_
3961 * or _to_ blocking mode, either way we can block them.
3963 * Clear error state in case pqFlush adds to it, unless we're actively
3964 * pipelining, in which case it seems best not to.
3966 if (conn
->cmd_queue_head
== NULL
)
3967 pqClearConnErrorState(conn
);
3969 /* if we are going from blocking to non-blocking flush here */
3973 conn
->nonblocking
= barg
;
3979 * return the blocking status of the database connection
3980 * true == nonblocking, false == blocking
3983 PQisnonblocking(const PGconn
*conn
)
3985 if (!conn
|| conn
->status
== CONNECTION_BAD
)
3987 return pqIsnonblocking(conn
);
3990 /* libpq is thread-safe? */
3992 PQisthreadsafe(void)
3998 /* try to force data out, really only useful for non-blocking users */
4000 PQflush(PGconn
*conn
)
4002 if (!conn
|| conn
->status
== CONNECTION_BAD
)
4004 return pqFlush(conn
);
4010 * In pipeline mode, data will be flushed only when the out buffer reaches the
4011 * threshold value. In non-pipeline mode, it behaves as stock pqFlush.
4013 * Returns 0 on success.
4016 pqPipelineFlush(PGconn
*conn
)
4018 if ((conn
->pipelineStatus
!= PQ_PIPELINE_ON
) ||
4019 (conn
->outCount
>= OUTBUFFER_THRESHOLD
))
4020 return pqFlush(conn
);
4026 * PQfreemem - safely frees memory allocated
4028 * Needed mostly by Win32, unless multithreaded DLL (/MD in VC6)
4029 * Used for freeing memory from PQescapeBytea()/PQunescapeBytea()
4032 PQfreemem(void *ptr
)
4038 * PQfreeNotify - free's the memory associated with a PGnotify
4040 * This function is here only for binary backward compatibility.
4041 * New code should use PQfreemem(). A macro will automatically map
4042 * calls to PQfreemem. It should be removed in the future. bjm 2003-03-24
4046 void PQfreeNotify(PGnotify
*notify
);
4049 PQfreeNotify(PGnotify
*notify
)
4056 * Escaping arbitrary strings to get valid SQL literal strings.
4058 * Replaces "'" with "''", and if not std_strings, replaces "\" with "\\".
4060 * length is the length of the source string. (Note: if a terminating NUL
4061 * is encountered sooner, PQescapeString stops short of "length"; the behavior
4062 * is thus rather like strncpy.)
4064 * For safety the buffer at "to" must be at least 2*length + 1 bytes long.
4065 * A terminating NUL character is added to the output string, whether the
4066 * input is NUL-terminated or not.
4068 * Returns the actual length of the output (not counting the terminating NUL).
4071 PQescapeStringInternal(PGconn
*conn
,
4072 char *to
, const char *from
, size_t length
,
4074 int encoding
, bool std_strings
)
4076 const char *source
= from
;
4078 size_t remaining
= length
;
4083 while (remaining
> 0 && *source
!= '\0')
4089 /* Fast path for plain ASCII */
4090 if (!IS_HIGHBIT_SET(c
))
4092 /* Apply quoting if needed */
4093 if (SQL_STR_DOUBLE(c
, !std_strings
))
4095 /* Copy the character */
4102 /* Slow path for possible multibyte characters */
4103 len
= pg_encoding_mblen(encoding
, source
);
4105 /* Copy the character */
4106 for (i
= 0; i
< len
; i
++)
4108 if (remaining
== 0 || *source
== '\0')
4110 *target
++ = *source
++;
4115 * If we hit premature end of string (ie, incomplete multibyte
4116 * character), try to pad out to the correct length with spaces. We
4117 * may not be able to pad completely, but we will always be able to
4118 * insert at least one pad space (since we'd not have quoted a
4119 * multibyte character). This should be enough to make a string that
4120 * the server will error out on.
4127 libpq_append_conn_error(conn
, "incomplete multibyte character");
4128 for (; i
< len
; i
++)
4130 if (((size_t) (target
- to
)) / 2 >= length
)
4138 /* Write the terminating NUL character. */
4145 PQescapeStringConn(PGconn
*conn
,
4146 char *to
, const char *from
, size_t length
,
4151 /* force empty-string result */
4158 if (conn
->cmd_queue_head
== NULL
)
4159 pqClearConnErrorState(conn
);
4161 return PQescapeStringInternal(conn
, to
, from
, length
, error
,
4162 conn
->client_encoding
,
4167 PQescapeString(char *to
, const char *from
, size_t length
)
4169 return PQescapeStringInternal(NULL
, to
, from
, length
, NULL
,
4170 static_client_encoding
,
4171 static_std_strings
);
4176 * Escape arbitrary strings. If as_ident is true, we escape the result
4177 * as an identifier; if false, as a literal. The result is returned in
4178 * a newly allocated buffer. If we fail due to an encoding violation or out
4179 * of memory condition, we return NULL, storing an error message into conn.
4182 PQescapeInternal(PGconn
*conn
, const char *str
, size_t len
, bool as_ident
)
4187 int num_quotes
= 0; /* single or double, depending on as_ident */
4188 int num_backslashes
= 0;
4191 char quote_char
= as_ident
? '"' : '\'';
4193 /* We must have a connection, else fail immediately. */
4197 if (conn
->cmd_queue_head
== NULL
)
4198 pqClearConnErrorState(conn
);
4200 /* Scan the string for characters that must be escaped. */
4201 for (s
= str
; (s
- str
) < len
&& *s
!= '\0'; ++s
)
4203 if (*s
== quote_char
)
4205 else if (*s
== '\\')
4207 else if (IS_HIGHBIT_SET(*s
))
4211 /* Slow path for possible multibyte characters */
4212 charlen
= pg_encoding_mblen(conn
->client_encoding
, s
);
4214 /* Multibyte character overruns allowable length. */
4215 if ((s
- str
) + charlen
> len
|| memchr(s
, 0, charlen
) != NULL
)
4217 libpq_append_conn_error(conn
, "incomplete multibyte character");
4221 /* Adjust s, bearing in mind that for loop will increment it. */
4226 /* Allocate output buffer. */
4227 input_len
= s
- str
;
4228 result_size
= input_len
+ num_quotes
+ 3; /* two quotes, plus a NUL */
4229 if (!as_ident
&& num_backslashes
> 0)
4230 result_size
+= num_backslashes
+ 2;
4231 result
= rp
= (char *) malloc(result_size
);
4234 libpq_append_conn_error(conn
, "out of memory");
4239 * If we are escaping a literal that contains backslashes, we use the
4240 * escape string syntax so that the result is correct under either value
4241 * of standard_conforming_strings. We also emit a leading space in this
4242 * case, to guard against the possibility that the result might be
4243 * interpolated immediately following an identifier.
4245 if (!as_ident
&& num_backslashes
> 0)
4251 /* Opening quote. */
4255 * Use fast path if possible.
4257 * We've already verified that the input string is well-formed in the
4258 * current encoding. If it contains no quotes and, in the case of
4259 * literal-escaping, no backslashes, then we can just copy it directly to
4260 * the output buffer, adding the necessary quotes.
4262 * If not, we must rescan the input and process each character
4265 if (num_quotes
== 0 && (num_backslashes
== 0 || as_ident
))
4267 memcpy(rp
, str
, input_len
);
4272 for (s
= str
; s
- str
< input_len
; ++s
)
4274 if (*s
== quote_char
|| (!as_ident
&& *s
== '\\'))
4279 else if (!IS_HIGHBIT_SET(*s
))
4283 int i
= pg_encoding_mblen(conn
->client_encoding
, s
);
4290 ++s
; /* for loop will provide the final increment */
4296 /* Closing quote and terminating NUL. */
4304 PQescapeLiteral(PGconn
*conn
, const char *str
, size_t len
)
4306 return PQescapeInternal(conn
, str
, len
, false);
4310 PQescapeIdentifier(PGconn
*conn
, const char *str
, size_t len
)
4312 return PQescapeInternal(conn
, str
, len
, true);
4315 /* HEX encoding support for bytea */
4316 static const char hextbl
[] = "0123456789abcdef";
4318 static const int8 hexlookup
[128] = {
4319 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
4320 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
4321 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
4322 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, -1, -1, -1, -1, -1, -1,
4323 -1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1, -1,
4324 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
4325 -1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1, -1,
4326 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
4334 if (c
> 0 && c
< 127)
4335 res
= hexlookup
[(unsigned char) c
];
4342 * PQescapeBytea - converts from binary string to the
4343 * minimal encoding necessary to include the string in an SQL
4344 * INSERT statement with a bytea type column as the target.
4346 * We can use either hex or escape (traditional) encoding.
4347 * In escape mode, the following transformations are applied:
4348 * '\0' == ASCII 0 == \000
4349 * '\'' == ASCII 39 == ''
4350 * '\\' == ASCII 92 == \\
4351 * anything < 0x20, or > 0x7e ---> \ooo
4352 * (where ooo is an octal expression)
4354 * If not std_strings, all backslashes sent to the output are doubled.
4356 static unsigned char *
4357 PQescapeByteaInternal(PGconn
*conn
,
4358 const unsigned char *from
, size_t from_length
,
4359 size_t *to_length
, bool std_strings
, bool use_hex
)
4361 const unsigned char *vp
;
4363 unsigned char *result
;
4366 size_t bslash_len
= (std_strings
? 1 : 2);
4369 * empty string has 1 char ('\0')
4375 len
+= bslash_len
+ 1 + 2 * from_length
;
4380 for (i
= from_length
; i
> 0; i
--, vp
++)
4382 if (*vp
< 0x20 || *vp
> 0x7e)
4383 len
+= bslash_len
+ 3;
4384 else if (*vp
== '\'')
4386 else if (*vp
== '\\')
4387 len
+= bslash_len
+ bslash_len
;
4394 rp
= result
= (unsigned char *) malloc(len
);
4398 libpq_append_conn_error(conn
, "out of memory");
4411 for (i
= from_length
; i
> 0; i
--, vp
++)
4413 unsigned char c
= *vp
;
4417 *rp
++ = hextbl
[(c
>> 4) & 0xF];
4418 *rp
++ = hextbl
[c
& 0xF];
4420 else if (c
< 0x20 || c
> 0x7e)
4425 *rp
++ = (c
>> 6) + '0';
4426 *rp
++ = ((c
>> 3) & 07) + '0';
4427 *rp
++ = (c
& 07) + '0';
4453 PQescapeByteaConn(PGconn
*conn
,
4454 const unsigned char *from
, size_t from_length
,
4460 if (conn
->cmd_queue_head
== NULL
)
4461 pqClearConnErrorState(conn
);
4463 return PQescapeByteaInternal(conn
, from
, from_length
, to_length
,
4465 (conn
->sversion
>= 90000));
4469 PQescapeBytea(const unsigned char *from
, size_t from_length
, size_t *to_length
)
4471 return PQescapeByteaInternal(NULL
, from
, from_length
, to_length
,
4473 false /* can't use hex */ );
4477 #define ISFIRSTOCTDIGIT(CH) ((CH) >= '0' && (CH) <= '3')
4478 #define ISOCTDIGIT(CH) ((CH) >= '0' && (CH) <= '7')
4479 #define OCTVAL(CH) ((CH) - '0')
4482 * PQunescapeBytea - converts the null terminated string representation
4483 * of a bytea, strtext, into binary, filling a buffer. It returns a
4484 * pointer to the buffer (or NULL on error), and the size of the
4485 * buffer in retbuflen. The pointer may subsequently be used as an
4486 * argument to the function PQfreemem.
4488 * The following transformations are made:
4489 * \\ == ASCII 92 == \
4490 * \ooo == a byte whose value = ooo (ooo is an octal number)
4491 * \x == x (x is any character not matched by the above transformations)
4494 PQunescapeBytea(const unsigned char *strtext
, size_t *retbuflen
)
4498 unsigned char *buffer
,
4503 if (strtext
== NULL
)
4506 strtextlen
= strlen((const char *) strtext
);
4508 if (strtext
[0] == '\\' && strtext
[1] == 'x')
4510 const unsigned char *s
;
4513 buflen
= (strtextlen
- 2) / 2;
4514 /* Avoid unportable malloc(0) */
4515 buffer
= (unsigned char *) malloc(buflen
> 0 ? buflen
: 1);
4527 * Bad input is silently ignored. Note that this includes
4528 * whitespace between hex pairs, which is allowed by byteain.
4531 if (!*s
|| v1
== (char) -1)
4534 if (v2
!= (char) -1)
4535 *p
++ = (v1
<< 4) | v2
;
4538 buflen
= p
- buffer
;
4543 * Length of input is max length of output, but add one to avoid
4544 * unportable malloc(0) if input is zero-length.
4546 buffer
= (unsigned char *) malloc(strtextlen
+ 1);
4550 for (i
= j
= 0; i
< strtextlen
;)
4556 if (strtext
[i
] == '\\')
4557 buffer
[j
++] = strtext
[i
++];
4560 if ((ISFIRSTOCTDIGIT(strtext
[i
])) &&
4561 (ISOCTDIGIT(strtext
[i
+ 1])) &&
4562 (ISOCTDIGIT(strtext
[i
+ 2])))
4566 byte
= OCTVAL(strtext
[i
++]);
4567 byte
= (byte
<< 3) + OCTVAL(strtext
[i
++]);
4568 byte
= (byte
<< 3) + OCTVAL(strtext
[i
++]);
4574 * Note: if we see '\' followed by something that isn't a
4575 * recognized escape sequence, we loop around having done
4576 * nothing except advance i. Therefore the something will
4577 * be emitted as ordinary data on the next cycle. Corner
4578 * case: '\' at end of string will just be discarded.
4583 buffer
[j
++] = strtext
[i
++];
4587 buflen
= j
; /* buflen is the length of the dequoted data */
4590 /* Shrink the buffer to be no larger than necessary */
4591 /* +1 avoids unportable behavior when buflen==0 */
4592 tmpbuf
= realloc(buffer
, buflen
+ 1);
4594 /* It would only be a very brain-dead realloc that could fail, but... */
4601 *retbuflen
= buflen
;