Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * fe-protocol3.c
4 : : * functions that are specific to frontend/backend protocol version 3
5 : : *
6 : : * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
7 : : * Portions Copyright (c) 1994, Regents of the University of California
8 : : *
9 : : *
10 : : * IDENTIFICATION
11 : : * src/interfaces/libpq/fe-protocol3.c
12 : : *
13 : : *-------------------------------------------------------------------------
14 : : */
15 : : #include "postgres_fe.h"
16 : :
17 : : #include <ctype.h>
18 : : #include <fcntl.h>
19 : : #include <limits.h>
20 : :
21 : : #ifdef WIN32
22 : : #include "win32.h"
23 : : #else
24 : : #include <unistd.h>
25 : : #include <netinet/tcp.h>
26 : : #endif
27 : :
28 : : #include "common/int.h"
29 : : #include "libpq-fe.h"
30 : : #include "libpq-int.h"
31 : : #include "mb/pg_wchar.h"
32 : : #include "port/pg_bswap.h"
33 : :
34 : : /*
35 : : * This macro lists the backend message types that could be "long" (more
36 : : * than a couple of kilobytes).
37 : : */
38 : : #define VALID_LONG_MESSAGE_TYPE(id) \
39 : : ((id) == PqMsg_CopyData || \
40 : : (id) == PqMsg_DataRow || \
41 : : (id) == PqMsg_ErrorResponse || \
42 : : (id) == PqMsg_FunctionCallResponse || \
43 : : (id) == PqMsg_NoticeResponse || \
44 : : (id) == PqMsg_NotificationResponse || \
45 : : (id) == PqMsg_RowDescription || \
46 : : (id) == PqMsg_ParameterDescription)
47 : :
48 : :
49 : : static void handleFatalError(PGconn *conn);
50 : : static void handleSyncLoss(PGconn *conn, char id, int msgLength);
51 : : static int getRowDescriptions(PGconn *conn, int msgLength);
52 : : static int getParamDescriptions(PGconn *conn, int msgLength);
53 : : static int getAnotherTuple(PGconn *conn, int msgLength);
54 : : static int getParameterStatus(PGconn *conn);
55 : : static int getBackendKeyData(PGconn *conn, int msgLength);
56 : : static int getNotify(PGconn *conn);
57 : : static int getCopyStart(PGconn *conn, ExecStatusType copytype);
58 : : static int getReadyForQuery(PGconn *conn);
59 : : static void reportErrorPosition(PQExpBuffer msg, const char *query,
60 : : int loc, int encoding);
61 : : static size_t build_startup_packet(const PGconn *conn, char *packet,
62 : : const PQEnvironmentOption *options);
63 : :
64 : :
65 : : /*
66 : : * parseInput: if appropriate, parse input data from backend
67 : : * until input is exhausted or a stopping state is reached.
68 : : * Note that this function will NOT attempt to read more data from the backend.
69 : : */
70 : : void
8481 tgl@sss.pgh.pa.us 71 :CBC 2179898 : pqParseInput3(PGconn *conn)
72 : : {
73 : : char id;
74 : : int msgLength;
75 : : int avail;
76 : :
77 : : /*
78 : : * Loop to parse successive complete messages available in the buffer.
79 : : */
80 : : for (;;)
81 : : {
82 : : /*
83 : : * Try to read a message. First get the type code and length. Return
84 : : * if not enough data.
85 : : */
86 : 7514167 : conn->inCursor = conn->inStart;
87 [ + + ]: 7514167 : if (pqGetc(&id, conn))
88 : 1623943 : return;
89 [ + + ]: 5890224 : if (pqGetInt(&msgLength, 4, conn))
90 : 1481 : return;
91 : :
92 : : /*
93 : : * Try to validate message type/length here. A length less than 4 is
94 : : * definitely broken. Large lengths should only be believed for a few
95 : : * message types.
96 : : */
97 [ - + ]: 5888743 : if (msgLength < 4)
98 : : {
8481 tgl@sss.pgh.pa.us 99 :UBC 0 : handleSyncLoss(conn, id, msgLength);
100 : 0 : return;
101 : : }
8278 tgl@sss.pgh.pa.us 102 [ + + + + :CBC 5888743 : if (msgLength > 30000 && !VALID_LONG_MESSAGE_TYPE(id))
+ + - + -
- - - - -
- - - - ]
103 : : {
8481 tgl@sss.pgh.pa.us 104 :UBC 0 : handleSyncLoss(conn, id, msgLength);
105 : 0 : return;
106 : : }
107 : :
108 : : /*
109 : : * Can't process if message body isn't all here yet.
110 : : */
8481 tgl@sss.pgh.pa.us 111 :CBC 5888743 : msgLength -= 4;
112 : 5888743 : avail = conn->inEnd - conn->inCursor;
113 [ + + ]: 5888743 : if (avail < msgLength)
114 : : {
115 : : /*
116 : : * Before returning, enlarge the input buffer if needed to hold
117 : : * the whole message. This is better than leaving it to
118 : : * pqReadData because we can avoid multiple cycles of realloc()
119 : : * when the message is large; also, we can implement a reasonable
120 : : * recovery strategy if we are unable to make the buffer big
121 : : * enough.
122 : : */
6664 123 [ - + ]: 76832 : if (pqCheckInBufferSpace(conn->inCursor + (size_t) msgLength,
124 : : conn))
125 : : {
126 : : /*
127 : : * Abandon the connection. There's not much else we can
128 : : * safely do; we can't just ignore the message or we could
129 : : * miss important changes to the connection state.
130 : : * pqCheckInBufferSpace() already reported the error.
131 : : */
370 heikki.linnakangas@i 132 :UBC 0 : handleFatalError(conn);
133 : : }
8481 tgl@sss.pgh.pa.us 134 :CBC 76832 : return;
135 : : }
136 : :
137 : : /*
138 : : * NOTIFY and NOTICE messages can happen in any state; always process
139 : : * them right away.
140 : : *
141 : : * Most other messages should only be processed while in BUSY state.
142 : : * (In particular, in READY state we hold off further parsing until
143 : : * the application collects the current PGresult.)
144 : : *
145 : : * However, if the state is IDLE then we got trouble; we need to deal
146 : : * with the unexpected message somehow.
147 : : *
148 : : * ParameterStatus ('S') messages are a special case: in IDLE state we
149 : : * must process 'em (this case could happen if a new value was adopted
150 : : * from config file due to SIGHUP), but otherwise we hold off until
151 : : * BUSY state.
152 : : */
1101 nathan@postgresql.or 153 [ + + ]: 5811911 : if (id == PqMsg_NotificationResponse)
154 : : {
8481 tgl@sss.pgh.pa.us 155 [ - + ]: 53 : if (getNotify(conn))
8481 tgl@sss.pgh.pa.us 156 :UBC 0 : return;
157 : : }
1101 nathan@postgresql.or 158 [ + + ]:CBC 5811858 : else if (id == PqMsg_NoticeResponse)
159 : : {
8481 tgl@sss.pgh.pa.us 160 [ - + ]: 17673 : if (pqGetErrorNotice3(conn, false))
8481 tgl@sss.pgh.pa.us 161 :UBC 0 : return;
162 : : }
8481 tgl@sss.pgh.pa.us 163 [ + + ]:CBC 5794185 : else if (conn->asyncStatus != PGASYNC_BUSY)
164 : : {
165 : : /* If not IDLE state, just wait ... */
166 [ + - ]: 477642 : if (conn->asyncStatus != PGASYNC_IDLE)
167 : 477642 : return;
168 : :
169 : : /*
170 : : * Unexpected message in IDLE state; need to recover somehow.
171 : : * ERROR messages are handled using the notice processor;
172 : : * ParameterStatus is handled normally; anything else is just
173 : : * dropped on the floor after displaying a suitable warning
174 : : * notice. (An ERROR is very possibly the backend telling us why
175 : : * it is about to close the connection, so we don't want to just
176 : : * discard it...)
177 : : */
1101 nathan@postgresql.or 178 [ # # ]:UBC 0 : if (id == PqMsg_ErrorResponse)
179 : : {
8424 bruce@momjian.us 180 [ # # ]: 0 : if (pqGetErrorNotice3(conn, false /* treat as notice */ ))
8481 tgl@sss.pgh.pa.us 181 : 0 : return;
182 : : }
1101 nathan@postgresql.or 183 [ # # ]: 0 : else if (id == PqMsg_ParameterStatus)
184 : : {
8481 tgl@sss.pgh.pa.us 185 [ # # ]: 0 : if (getParameterStatus(conn))
186 : 0 : return;
187 : : }
188 : : else
189 : : {
190 : : /* Any other case is unexpected and we summarily skip it */
8466 191 : 0 : pqInternalNotice(&conn->noticeHooks,
192 : : "message type 0x%02x arrived from server while idle",
193 : : id);
194 : : /* Discard the unexpected message */
8481 195 : 0 : conn->inCursor += msgLength;
196 : : }
197 : : }
198 : : else
199 : : {
200 : : /*
201 : : * In BUSY state, we can process everything.
202 : : */
8481 tgl@sss.pgh.pa.us 203 [ + + + + :CBC 5316543 : switch (id)
+ + + + +
+ + + + +
+ + + +
- ]
204 : : {
1101 nathan@postgresql.or 205 : 398242 : case PqMsg_CommandComplete:
8481 tgl@sss.pgh.pa.us 206 [ - + ]: 398242 : if (pqGets(&conn->workBuffer, conn))
8481 tgl@sss.pgh.pa.us 207 :UBC 0 : return;
1589 tgl@sss.pgh.pa.us 208 [ + + + - ]:CBC 398242 : if (!pgHavePendingResult(conn))
209 : : {
8481 210 : 198982 : conn->result = PQmakeEmptyPGresult(conn,
211 : : PGRES_COMMAND_OK);
7746 neilc@samurai.com 212 [ - + ]: 198982 : if (!conn->result)
213 : : {
1381 peter@eisentraut.org 214 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
4069 heikki.linnakangas@i 215 : 0 : pqSaveErrorResult(conn);
216 : : }
217 : : }
4069 heikki.linnakangas@i 218 [ + - ]:CBC 398242 : if (conn->result)
219 : 398242 : strlcpy(conn->result->cmdStatus, conn->workBuffer.data,
220 : : CMDSTATUS_LEN);
8481 tgl@sss.pgh.pa.us 221 : 398242 : conn->asyncStatus = PGASYNC_READY;
222 : 398242 : break;
1101 nathan@postgresql.or 223 : 31653 : case PqMsg_ErrorResponse:
8481 tgl@sss.pgh.pa.us 224 [ - + ]: 31653 : if (pqGetErrorNotice3(conn, true))
8481 tgl@sss.pgh.pa.us 225 :UBC 0 : return;
8481 tgl@sss.pgh.pa.us 226 :CBC 31653 : conn->asyncStatus = PGASYNC_READY;
227 : 31653 : break;
1101 nathan@postgresql.or 228 : 423999 : case PqMsg_ReadyForQuery:
8468 tgl@sss.pgh.pa.us 229 [ - + ]: 423999 : if (getReadyForQuery(conn))
8481 tgl@sss.pgh.pa.us 230 :UBC 0 : return;
1991 alvherre@alvh.no-ip. 231 [ + + ]:CBC 423999 : if (conn->pipelineStatus != PQ_PIPELINE_OFF)
232 : : {
233 : 385 : conn->result = PQmakeEmptyPGresult(conn,
234 : : PGRES_PIPELINE_SYNC);
235 [ - + ]: 385 : if (!conn->result)
236 : : {
1381 peter@eisentraut.org 237 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
1991 alvherre@alvh.no-ip. 238 : 0 : pqSaveErrorResult(conn);
239 : : }
240 : : else
241 : : {
1991 alvherre@alvh.no-ip. 242 :CBC 385 : conn->pipelineStatus = PQ_PIPELINE_ON;
243 : 385 : conn->asyncStatus = PGASYNC_READY;
244 : : }
245 : : }
246 : : else
247 : : {
248 : : /* Advance the command queue and set us idle */
996 249 : 423614 : pqCommandQueueAdvance(conn, true, false);
1991 250 : 423614 : conn->asyncStatus = PGASYNC_IDLE;
251 : : }
8481 tgl@sss.pgh.pa.us 252 : 423999 : break;
1101 nathan@postgresql.or 253 : 1116 : case PqMsg_EmptyQueryResponse:
1589 tgl@sss.pgh.pa.us 254 [ + - + - ]: 1116 : if (!pgHavePendingResult(conn))
255 : : {
8481 256 : 1116 : conn->result = PQmakeEmptyPGresult(conn,
257 : : PGRES_EMPTY_QUERY);
7746 neilc@samurai.com 258 [ - + ]: 1116 : if (!conn->result)
259 : : {
1381 peter@eisentraut.org 260 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
4069 heikki.linnakangas@i 261 : 0 : pqSaveErrorResult(conn);
262 : : }
263 : : }
8481 tgl@sss.pgh.pa.us 264 :CBC 1116 : conn->asyncStatus = PGASYNC_READY;
265 : 1116 : break;
1101 nathan@postgresql.or 266 : 4616 : case PqMsg_ParseComplete:
267 : : /* If we're doing PQprepare, we're done; else ignore */
1991 alvherre@alvh.no-ip. 268 [ + - ]: 4616 : if (conn->cmd_queue_head &&
269 [ + + ]: 4616 : conn->cmd_queue_head->queryclass == PGQUERY_PREPARE)
270 : : {
1589 tgl@sss.pgh.pa.us 271 [ + - + - ]: 1418 : if (!pgHavePendingResult(conn))
272 : : {
7983 273 : 1418 : conn->result = PQmakeEmptyPGresult(conn,
274 : : PGRES_COMMAND_OK);
7746 neilc@samurai.com 275 [ - + ]: 1418 : if (!conn->result)
276 : : {
1381 peter@eisentraut.org 277 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
4069 heikki.linnakangas@i 278 : 0 : pqSaveErrorResult(conn);
279 : : }
280 : : }
7983 tgl@sss.pgh.pa.us 281 :CBC 1418 : conn->asyncStatus = PGASYNC_READY;
282 : : }
283 : 4616 : break;
1101 nathan@postgresql.or 284 : 12118 : case PqMsg_BindComplete:
285 : : /* Nothing to do for this message type */
1150 michael@paquier.xyz 286 : 12118 : break;
1101 nathan@postgresql.or 287 : 21 : case PqMsg_CloseComplete:
288 : : /* If we're doing PQsendClose, we're done; else ignore */
1150 michael@paquier.xyz 289 [ + - ]: 21 : if (conn->cmd_queue_head &&
290 [ + - ]: 21 : conn->cmd_queue_head->queryclass == PGQUERY_CLOSE)
291 : : {
292 [ + - + - ]: 21 : if (!pgHavePendingResult(conn))
293 : : {
294 : 21 : conn->result = PQmakeEmptyPGresult(conn,
295 : : PGRES_COMMAND_OK);
296 [ - + ]: 21 : if (!conn->result)
297 : : {
1150 michael@paquier.xyz 298 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
299 : 0 : pqSaveErrorResult(conn);
300 : : }
301 : : }
1150 michael@paquier.xyz 302 :CBC 21 : conn->asyncStatus = PGASYNC_READY;
303 : : }
8468 tgl@sss.pgh.pa.us 304 : 21 : break;
1101 nathan@postgresql.or 305 : 241351 : case PqMsg_ParameterStatus:
8481 tgl@sss.pgh.pa.us 306 [ - + ]: 241351 : if (getParameterStatus(conn))
8481 tgl@sss.pgh.pa.us 307 :UBC 0 : return;
8481 tgl@sss.pgh.pa.us 308 :CBC 241351 : break;
1101 nathan@postgresql.or 309 : 15476 : case PqMsg_BackendKeyData:
310 : :
311 : : /*
312 : : * This is expected only during backend startup, but it's
313 : : * just as easy to handle it as part of the main loop.
314 : : * Save the data and continue processing.
315 : : */
512 heikki.linnakangas@i 316 [ - + ]: 15476 : if (getBackendKeyData(conn, msgLength))
8481 tgl@sss.pgh.pa.us 317 :UBC 0 : return;
8481 tgl@sss.pgh.pa.us 318 :CBC 15476 : break;
1101 nathan@postgresql.or 319 : 204389 : case PqMsg_RowDescription:
1651 tgl@sss.pgh.pa.us 320 [ + - ]: 204389 : if (conn->error_result ||
321 [ + + ]: 204389 : (conn->result != NULL &&
322 [ - + ]: 50 : conn->result->resultStatus == PGRES_FATAL_ERROR))
323 : : {
324 : : /*
325 : : * We've already choked for some reason. Just discard
326 : : * the data till we get to the end of the query.
327 : : */
3909 heikki.linnakangas@i 328 :UBC 0 : conn->inCursor += msgLength;
329 : : }
3909 heikki.linnakangas@i 330 [ + + ]:CBC 204389 : else if (conn->result == NULL ||
1991 alvherre@alvh.no-ip. 331 [ + - ]: 50 : (conn->cmd_queue_head &&
332 [ + - ]: 50 : conn->cmd_queue_head->queryclass == PGQUERY_DESCRIBE))
333 : : {
334 : : /* First 'T' in a query sequence */
5258 tgl@sss.pgh.pa.us 335 [ - + ]: 204389 : if (getRowDescriptions(conn, msgLength))
8481 tgl@sss.pgh.pa.us 336 :UBC 0 : return;
337 : : }
338 : : else
339 : : {
340 : : /*
341 : : * A new 'T' message is treated as the start of
342 : : * another PGresult. (It is not clear that this is
343 : : * really possible with the current backend.) We stop
344 : : * parsing until the application accepts the current
345 : : * result.
346 : : */
347 : 0 : conn->asyncStatus = PGASYNC_READY;
348 : 0 : return;
349 : : }
8481 tgl@sss.pgh.pa.us 350 :CBC 204389 : break;
1101 nathan@postgresql.or 351 : 5191 : case PqMsg_NoData:
352 : :
353 : : /*
354 : : * NoData indicates that we will not be seeing a
355 : : * RowDescription message because the statement or portal
356 : : * inquired about doesn't return rows.
357 : : *
358 : : * If we're doing a Describe, we have to pass something
359 : : * back to the client, so set up a COMMAND_OK result,
360 : : * instead of PGRES_TUPLES_OK. Otherwise we can just
361 : : * ignore this message.
362 : : */
1991 alvherre@alvh.no-ip. 363 [ + - ]: 5191 : if (conn->cmd_queue_head &&
364 [ + + ]: 5191 : conn->cmd_queue_head->queryclass == PGQUERY_DESCRIBE)
365 : : {
1589 tgl@sss.pgh.pa.us 366 [ - + - - ]: 8 : if (!pgHavePendingResult(conn))
367 : : {
6439 tgl@sss.pgh.pa.us 368 :UBC 0 : conn->result = PQmakeEmptyPGresult(conn,
369 : : PGRES_COMMAND_OK);
370 [ # # ]: 0 : if (!conn->result)
371 : : {
1381 peter@eisentraut.org 372 : 0 : libpq_append_conn_error(conn, "out of memory");
4069 heikki.linnakangas@i 373 : 0 : pqSaveErrorResult(conn);
374 : : }
375 : : }
7314 tgl@sss.pgh.pa.us 376 :CBC 8 : conn->asyncStatus = PGASYNC_READY;
377 : : }
378 : 5191 : break;
1101 nathan@postgresql.or 379 : 58 : case PqMsg_ParameterDescription:
3909 heikki.linnakangas@i 380 [ - + ]: 58 : if (getParamDescriptions(conn, msgLength))
7314 tgl@sss.pgh.pa.us 381 :UBC 0 : return;
1995 tgl@sss.pgh.pa.us 382 :CBC 58 : break;
1101 nathan@postgresql.or 383 : 3965761 : case PqMsg_DataRow:
8481 tgl@sss.pgh.pa.us 384 [ + - ]: 3965761 : if (conn->result != NULL &&
873 385 [ + + ]: 3965761 : (conn->result->resultStatus == PGRES_TUPLES_OK ||
386 [ + - ]: 123 : conn->result->resultStatus == PGRES_TUPLES_CHUNK))
387 : : {
388 : : /* Read another tuple of a normal query response */
8481 389 [ - + ]: 3965761 : if (getAnotherTuple(conn, msgLength))
8481 tgl@sss.pgh.pa.us 390 :UBC 0 : return;
391 : : }
1651 392 [ # # ]: 0 : else if (conn->error_result ||
393 [ # # ]: 0 : (conn->result != NULL &&
394 [ # # ]: 0 : conn->result->resultStatus == PGRES_FATAL_ERROR))
395 : : {
396 : : /*
397 : : * We've already choked for some reason. Just discard
398 : : * tuples till we get to the end of the query.
399 : : */
8481 400 : 0 : conn->inCursor += msgLength;
401 : : }
402 : : else
403 : : {
404 : : /* Set up to report error at end of query */
1381 peter@eisentraut.org 405 : 0 : libpq_append_conn_error(conn, "server sent data (\"D\" message) without prior row description (\"T\" message)");
8481 tgl@sss.pgh.pa.us 406 : 0 : pqSaveErrorResult(conn);
407 : : /* Discard the unexpected message */
408 : 0 : conn->inCursor += msgLength;
409 : : }
8481 tgl@sss.pgh.pa.us 410 :CBC 3965761 : break;
1101 nathan@postgresql.or 411 : 716 : case PqMsg_CopyInResponse:
8468 tgl@sss.pgh.pa.us 412 [ - + ]: 716 : if (getCopyStart(conn, PGRES_COPY_IN))
8481 tgl@sss.pgh.pa.us 413 :UBC 0 : return;
8481 tgl@sss.pgh.pa.us 414 :CBC 716 : conn->asyncStatus = PGASYNC_COPY_IN;
415 : 716 : break;
1101 nathan@postgresql.or 416 : 5321 : case PqMsg_CopyOutResponse:
8468 tgl@sss.pgh.pa.us 417 [ - + ]: 5321 : if (getCopyStart(conn, PGRES_COPY_OUT))
8481 tgl@sss.pgh.pa.us 418 :UBC 0 : return;
8481 tgl@sss.pgh.pa.us 419 :CBC 5321 : conn->asyncStatus = PGASYNC_COPY_OUT;
420 : 5321 : conn->copy_already_done = 0;
421 : 5321 : break;
1101 nathan@postgresql.or 422 : 815 : case PqMsg_CopyBothResponse:
5738 rhaas@postgresql.org 423 [ - + ]: 815 : if (getCopyStart(conn, PGRES_COPY_BOTH))
5738 rhaas@postgresql.org 424 :UBC 0 : return;
5738 rhaas@postgresql.org 425 :CBC 815 : conn->asyncStatus = PGASYNC_COPY_BOTH;
426 : 815 : conn->copy_already_done = 0;
427 : 815 : break;
1101 nathan@postgresql.or 428 : 1 : case PqMsg_CopyData:
429 : :
430 : : /*
431 : : * If we see Copy Data, just silently drop it. This would
432 : : * only occur if application exits COPY OUT mode too
433 : : * early.
434 : : */
8481 tgl@sss.pgh.pa.us 435 : 1 : conn->inCursor += msgLength;
436 : 1 : break;
1101 nathan@postgresql.or 437 : 5699 : case PqMsg_CopyDone:
438 : :
439 : : /*
440 : : * If we see Copy Done, just silently drop it. This is
441 : : * the normal case during PQendcopy. We will keep
442 : : * swallowing data, expecting to see command-complete for
443 : : * the COPY command.
444 : : */
8481 tgl@sss.pgh.pa.us 445 : 5699 : break;
8481 tgl@sss.pgh.pa.us 446 :UBC 0 : default:
1381 peter@eisentraut.org 447 : 0 : libpq_append_conn_error(conn, "unexpected response from server; first received character was \"%c\"", id);
448 : : /* build an error result holding the error message */
8481 tgl@sss.pgh.pa.us 449 : 0 : pqSaveErrorResult(conn);
450 : : /* not sure if we will see more, so go to ready state */
451 : 0 : conn->asyncStatus = PGASYNC_READY;
452 : : /* Discard the unexpected message */
453 : 0 : conn->inCursor += msgLength;
454 : 0 : break;
455 : : } /* switch on protocol character */
456 : : }
457 : : /* Successfully consumed this message */
8481 tgl@sss.pgh.pa.us 458 [ + - ]:CBC 5334269 : if (conn->inCursor == conn->inStart + 5 + msgLength)
459 : : {
460 : : /* Normal case: parsing agrees with specified length */
741 alvherre@alvh.no-ip. 461 : 5334269 : pqParseDone(conn, conn->inCursor);
462 : : }
370 heikki.linnakangas@i 463 [ # # # # ]:UBC 0 : else if (conn->error_result && conn->status == CONNECTION_BAD)
464 : : {
465 : : /* The connection was abandoned and we already reported it */
466 : 0 : return;
467 : : }
468 : : else
469 : : {
470 : : /* Trouble --- report it */
1381 peter@eisentraut.org 471 : 0 : libpq_append_conn_error(conn, "message contents do not agree with length in message type \"%c\"", id);
472 : : /* build an error result holding the error message */
8481 tgl@sss.pgh.pa.us 473 : 0 : pqSaveErrorResult(conn);
474 : 0 : conn->asyncStatus = PGASYNC_READY;
475 : : /* trust the specified message length as what to skip */
476 : 0 : conn->inStart += 5 + msgLength;
477 : : }
478 : : }
479 : : }
480 : :
481 : : /*
482 : : * handleFatalError: clean up after a nonrecoverable error
483 : : *
484 : : * This is for errors where we need to abandon the connection. The caller has
485 : : * already saved the error message in conn->errorMessage.
486 : : */
487 : : static void
370 heikki.linnakangas@i 488 : 0 : handleFatalError(PGconn *conn)
489 : : {
490 : : /* build an error result holding the error message */
8481 tgl@sss.pgh.pa.us 491 : 0 : pqSaveErrorResult(conn);
1991 alvherre@alvh.no-ip. 492 : 0 : conn->asyncStatus = PGASYNC_READY; /* drop out of PQgetResult wait loop */
493 : : /* flush input data since we're giving up on processing it */
3941 tgl@sss.pgh.pa.us 494 : 0 : pqDropConnection(conn, true);
3354 495 : 0 : conn->status = CONNECTION_BAD; /* No more connection to backend */
8481 496 : 0 : }
497 : :
498 : : /*
499 : : * handleSyncLoss: clean up after loss of message-boundary sync
500 : : *
501 : : * There isn't really a lot we can do here except abandon the connection.
502 : : */
503 : : static void
370 heikki.linnakangas@i 504 : 0 : handleSyncLoss(PGconn *conn, char id, int msgLength)
505 : : {
506 : 0 : libpq_append_conn_error(conn, "lost synchronization with server: got message type \"%c\", length %d",
507 : : id, msgLength);
508 : 0 : handleFatalError(conn);
509 : 0 : }
510 : :
511 : : /*
512 : : * parseInput subroutine to read a 'T' (row descriptions) message.
513 : : * We'll build a new PGresult structure (unless called for a Describe
514 : : * command for a prepared statement) containing the attribute data.
515 : : * Returns: 0 if processed message successfully, EOF to suspend parsing
516 : : * (the latter case is not actually used currently).
517 : : */
518 : : static int
5258 tgl@sss.pgh.pa.us 519 :CBC 204389 : getRowDescriptions(PGconn *conn, int msgLength)
520 : : {
521 : : PGresult *result;
522 : : int nfields;
523 : : const char *errmsg;
524 : : int i;
525 : :
526 : : /*
527 : : * When doing Describe for a prepared statement, there'll already be a
528 : : * PGresult created by getParamDescriptions, and we should fill data into
529 : : * that. Otherwise, create a new, empty PGresult.
530 : : */
1991 alvherre@alvh.no-ip. 531 [ + - ]: 204389 : if (!conn->cmd_queue_head ||
532 [ + - ]: 204389 : (conn->cmd_queue_head &&
533 [ + + ]: 204389 : conn->cmd_queue_head->queryclass == PGQUERY_DESCRIBE))
534 : : {
7314 tgl@sss.pgh.pa.us 535 [ + + ]: 51 : if (conn->result)
536 : 50 : result = conn->result;
537 : : else
538 : 1 : result = PQmakeEmptyPGresult(conn, PGRES_COMMAND_OK);
539 : : }
540 : : else
541 : 204338 : result = PQmakeEmptyPGresult(conn, PGRES_TUPLES_OK);
7746 neilc@samurai.com 542 [ - + ]: 204389 : if (!result)
543 : : {
5258 tgl@sss.pgh.pa.us 544 :UBC 0 : errmsg = NULL; /* means "out of memory", see below */
545 : 0 : goto advance_and_error;
546 : : }
547 : :
548 : : /* parseInput already read the 'T' label and message length. */
549 : : /* the next two bytes are the number of fields */
8481 tgl@sss.pgh.pa.us 550 [ - + ]:CBC 204389 : if (pqGetInt(&(result->numAttributes), 2, conn))
551 : : {
552 : : /* We should not run out of data here, so complain */
5258 tgl@sss.pgh.pa.us 553 :UBC 0 : errmsg = libpq_gettext("insufficient data in \"T\" message");
554 : 0 : goto advance_and_error;
555 : : }
8481 tgl@sss.pgh.pa.us 556 :CBC 204389 : nfields = result->numAttributes;
557 : :
558 : : /* allocate space for the attribute descriptors */
559 [ + + ]: 204389 : if (nfields > 0)
560 : : {
561 : 204159 : result->attDescs = (PGresAttDesc *)
3298 peter_e@gmx.net 562 : 204159 : pqResultAlloc(result, nfields * sizeof(PGresAttDesc), true);
7746 neilc@samurai.com 563 [ - + ]: 204159 : if (!result->attDescs)
564 : : {
5258 tgl@sss.pgh.pa.us 565 :UBC 0 : errmsg = NULL; /* means "out of memory", see below */
566 : 0 : goto advance_and_error;
567 : : }
7746 neilc@samurai.com 568 [ + - + - :CBC 2670003 : MemSet(result->attDescs, 0, nfields * sizeof(PGresAttDesc));
+ - + + +
+ ]
569 : : }
570 : :
571 : : /* result->binary is true only if ALL columns are binary */
8468 tgl@sss.pgh.pa.us 572 : 204389 : result->binary = (nfields > 0) ? 1 : 0;
573 : :
574 : : /* get type info */
8481 575 [ + + ]: 829420 : for (i = 0; i < nfields; i++)
576 : : {
577 : : int tableid;
578 : : int columnid;
579 : : int typid;
580 : : int typlen;
581 : : int atttypmod;
582 : : int format;
583 : :
584 [ + - + - ]: 1250062 : if (pqGets(&conn->workBuffer, conn) ||
585 [ + - ]: 1250062 : pqGetInt(&tableid, 4, conn) ||
586 [ + - ]: 1250062 : pqGetInt(&columnid, 2, conn) ||
587 [ + - ]: 1250062 : pqGetInt(&typid, 4, conn) ||
588 [ + - ]: 1250062 : pqGetInt(&typlen, 2, conn) ||
589 [ - + ]: 1250062 : pqGetInt(&atttypmod, 4, conn) ||
590 : 625031 : pqGetInt(&format, 2, conn))
591 : : {
592 : : /* We should not run out of data here, so complain */
5258 tgl@sss.pgh.pa.us 593 :UBC 0 : errmsg = libpq_gettext("insufficient data in \"T\" message");
594 : 0 : goto advance_and_error;
595 : : }
596 : :
597 : : /*
598 : : * Since pqGetInt treats 2-byte integers as unsigned, we need to
599 : : * coerce these results to signed form.
600 : : */
8481 tgl@sss.pgh.pa.us 601 :CBC 625031 : columnid = (int) ((int16) columnid);
602 : 625031 : typlen = (int) ((int16) typlen);
603 : 625031 : format = (int) ((int16) format);
604 : :
605 : 1250062 : result->attDescs[i].name = pqResultStrdup(result,
606 : 625031 : conn->workBuffer.data);
7746 neilc@samurai.com 607 [ - + ]: 625031 : if (!result->attDescs[i].name)
608 : : {
5258 tgl@sss.pgh.pa.us 609 :UBC 0 : errmsg = NULL; /* means "out of memory", see below */
610 : 0 : goto advance_and_error;
611 : : }
8468 tgl@sss.pgh.pa.us 612 :CBC 625031 : result->attDescs[i].tableid = tableid;
613 : 625031 : result->attDescs[i].columnid = columnid;
614 : 625031 : result->attDescs[i].format = format;
8481 615 : 625031 : result->attDescs[i].typid = typid;
616 : 625031 : result->attDescs[i].typlen = typlen;
617 : 625031 : result->attDescs[i].atttypmod = atttypmod;
618 : :
8468 619 [ + + ]: 625031 : if (format != 1)
620 : 624992 : result->binary = 0;
621 : : }
622 : :
623 : : /* Success! */
8481 624 : 204389 : conn->result = result;
625 : :
626 : : /*
627 : : * If we're doing a Describe, we're done, and ready to pass the result
628 : : * back to the client.
629 : : */
1991 alvherre@alvh.no-ip. 630 [ + - ]: 204389 : if ((!conn->cmd_queue_head) ||
631 [ + - ]: 204389 : (conn->cmd_queue_head &&
632 [ + + ]: 204389 : conn->cmd_queue_head->queryclass == PGQUERY_DESCRIBE))
633 : : {
5258 tgl@sss.pgh.pa.us 634 : 51 : conn->asyncStatus = PGASYNC_READY;
635 : 51 : return 0;
636 : : }
637 : :
638 : : /*
639 : : * We could perform additional setup for the new result set here, but for
640 : : * now there's nothing else to do.
641 : : */
642 : :
643 : : /* And we're done. */
5138 644 : 204338 : return 0;
645 : :
5258 tgl@sss.pgh.pa.us 646 :UBC 0 : advance_and_error:
647 : : /* Discard unsaved result, if any */
648 [ # # # # ]: 0 : if (result && result != conn->result)
7314 649 : 0 : PQclear(result);
650 : :
651 : : /*
652 : : * Replace partially constructed result with an error result. First
653 : : * discard the old result to try to win back some memory.
654 : : */
5258 655 : 0 : pqClearAsyncResult(conn);
656 : :
657 : : /*
658 : : * If preceding code didn't provide an error message, assume "out of
659 : : * memory" was meant. The advantage of having this special case is that
660 : : * freeing the old result first greatly improves the odds that gettext()
661 : : * will succeed in providing a translation.
662 : : */
663 [ # # ]: 0 : if (!errmsg)
664 : 0 : errmsg = libpq_gettext("out of memory for query result");
665 : :
2054 666 : 0 : appendPQExpBuffer(&conn->errorMessage, "%s\n", errmsg);
5258 667 : 0 : pqSaveErrorResult(conn);
668 : :
669 : : /*
670 : : * Show the message as fully consumed, else pqParseInput3 will overwrite
671 : : * our error with a complaint about that.
672 : : */
1995 673 : 0 : conn->inCursor = conn->inStart + 5 + msgLength;
674 : :
675 : : /*
676 : : * Return zero to allow input parsing to continue. Subsequent "D"
677 : : * messages will be ignored until we get to end of data, since an error
678 : : * result is already set up.
679 : : */
5258 680 : 0 : return 0;
681 : : }
682 : :
683 : : /*
684 : : * parseInput subroutine to read a 't' (ParameterDescription) message.
685 : : * We'll build a new PGresult structure containing the parameter data.
686 : : * Returns: 0 if processed message successfully, EOF to suspend parsing
687 : : * (the latter case is not actually used currently).
688 : : */
689 : : static int
3909 heikki.linnakangas@i 690 :CBC 58 : getParamDescriptions(PGconn *conn, int msgLength)
691 : : {
692 : : PGresult *result;
3800 tgl@sss.pgh.pa.us 693 : 58 : const char *errmsg = NULL; /* means "out of memory", see below */
694 : : int nparams;
695 : : int i;
696 : :
7314 697 : 58 : result = PQmakeEmptyPGresult(conn, PGRES_COMMAND_OK);
698 [ - + ]: 58 : if (!result)
3909 heikki.linnakangas@i 699 :UBC 0 : goto advance_and_error;
700 : :
701 : : /* parseInput already read the 't' label and message length. */
702 : : /* the next two bytes are the number of parameters */
7314 tgl@sss.pgh.pa.us 703 [ - + ]:CBC 58 : if (pqGetInt(&(result->numParameters), 2, conn))
3909 heikki.linnakangas@i 704 :UBC 0 : goto not_enough_data;
7314 tgl@sss.pgh.pa.us 705 :CBC 58 : nparams = result->numParameters;
706 : :
707 : : /* allocate space for the parameter descriptors */
708 [ + + ]: 58 : if (nparams > 0)
709 : : {
710 : 5 : result->paramDescs = (PGresParamDesc *)
3298 peter_e@gmx.net 711 : 5 : pqResultAlloc(result, nparams * sizeof(PGresParamDesc), true);
7314 tgl@sss.pgh.pa.us 712 [ - + ]: 5 : if (!result->paramDescs)
3909 heikki.linnakangas@i 713 :UBC 0 : goto advance_and_error;
7314 tgl@sss.pgh.pa.us 714 [ + - + + :CBC 9 : MemSet(result->paramDescs, 0, nparams * sizeof(PGresParamDesc));
+ - + - +
+ ]
715 : : }
716 : :
717 : : /* get parameter info */
718 [ + + ]: 67 : for (i = 0; i < nparams; i++)
719 : : {
720 : : int typid;
721 : :
722 [ - + ]: 9 : if (pqGetInt(&typid, 4, conn))
3909 heikki.linnakangas@i 723 :UBC 0 : goto not_enough_data;
7314 tgl@sss.pgh.pa.us 724 :CBC 9 : result->paramDescs[i].typid = typid;
725 : : }
726 : :
727 : : /* Success! */
728 : 58 : conn->result = result;
729 : :
730 : 58 : return 0;
731 : :
3909 heikki.linnakangas@i 732 :UBC 0 : not_enough_data:
1995 tgl@sss.pgh.pa.us 733 : 0 : errmsg = libpq_gettext("insufficient data in \"t\" message");
734 : :
3909 heikki.linnakangas@i 735 : 0 : advance_and_error:
736 : : /* Discard unsaved result, if any */
737 [ # # # # ]: 0 : if (result && result != conn->result)
738 : 0 : PQclear(result);
739 : :
740 : : /*
741 : : * Replace partially constructed result with an error result. First
742 : : * discard the old result to try to win back some memory.
743 : : */
744 : 0 : pqClearAsyncResult(conn);
745 : :
746 : : /*
747 : : * If preceding code didn't provide an error message, assume "out of
748 : : * memory" was meant. The advantage of having this special case is that
749 : : * freeing the old result first greatly improves the odds that gettext()
750 : : * will succeed in providing a translation.
751 : : */
752 [ # # ]: 0 : if (!errmsg)
753 : 0 : errmsg = libpq_gettext("out of memory");
2054 tgl@sss.pgh.pa.us 754 : 0 : appendPQExpBuffer(&conn->errorMessage, "%s\n", errmsg);
3909 heikki.linnakangas@i 755 : 0 : pqSaveErrorResult(conn);
756 : :
757 : : /*
758 : : * Show the message as fully consumed, else pqParseInput3 will overwrite
759 : : * our error with a complaint about that.
760 : : */
1995 tgl@sss.pgh.pa.us 761 : 0 : conn->inCursor = conn->inStart + 5 + msgLength;
762 : :
763 : : /*
764 : : * Return zero to allow input parsing to continue. Essentially, we've
765 : : * replaced the COMMAND_OK result with an error result, but since this
766 : : * doesn't affect the protocol state, it's fine.
767 : : */
3909 heikki.linnakangas@i 768 : 0 : return 0;
769 : : }
770 : :
771 : : /*
772 : : * parseInput subroutine to read a 'D' (row data) message.
773 : : * We fill rowbuf with column pointers and then call the row processor.
774 : : * Returns: 0 if processed message successfully, EOF to suspend parsing
775 : : * (the latter case is not actually used currently).
776 : : */
777 : : static int
8481 tgl@sss.pgh.pa.us 778 :CBC 3965761 : getAnotherTuple(PGconn *conn, int msgLength)
779 : : {
780 : 3965761 : PGresult *result = conn->result;
781 : 3965761 : int nfields = result->numAttributes;
782 : : const char *errmsg;
783 : : PGdataValue *rowbuf;
784 : : int tupnfields; /* # fields from tuple */
785 : : int vlen; /* length of the current field value */
786 : : int i;
787 : :
788 : : /* Get the field count and make sure it's what we expect */
789 [ - + ]: 3965761 : if (pqGetInt(&tupnfields, 2, conn))
790 : : {
791 : : /* We should not run out of data here, so complain */
5258 tgl@sss.pgh.pa.us 792 :UBC 0 : errmsg = libpq_gettext("insufficient data in \"D\" message");
793 : 0 : goto advance_and_error;
794 : : }
795 : :
8481 tgl@sss.pgh.pa.us 796 [ - + ]:CBC 3965761 : if (tupnfields != nfields)
797 : : {
5258 tgl@sss.pgh.pa.us 798 :UBC 0 : errmsg = libpq_gettext("unexpected field count in \"D\" message");
799 : 0 : goto advance_and_error;
800 : : }
801 : :
802 : : /* Resize row buffer if needed */
5258 tgl@sss.pgh.pa.us 803 :CBC 3965761 : rowbuf = conn->rowBuf;
804 [ + + ]: 3965761 : if (nfields > conn->rowBufLen)
805 : : {
806 : 215 : rowbuf = (PGdataValue *) realloc(rowbuf,
807 : : nfields * sizeof(PGdataValue));
808 [ - + ]: 215 : if (!rowbuf)
809 : : {
5258 tgl@sss.pgh.pa.us 810 :UBC 0 : errmsg = NULL; /* means "out of memory", see below */
811 : 0 : goto advance_and_error;
812 : : }
5258 tgl@sss.pgh.pa.us 813 :CBC 215 : conn->rowBuf = rowbuf;
814 : 215 : conn->rowBufLen = nfields;
815 : : }
816 : :
817 : : /* Scan the fields */
8481 818 [ + + ]: 24157934 : for (i = 0; i < nfields; i++)
819 : : {
820 : : /* get the value length */
821 [ - + ]: 20192173 : if (pqGetInt(&vlen, 4, conn))
822 : : {
823 : : /* We should not run out of data here, so complain */
5258 tgl@sss.pgh.pa.us 824 :UBC 0 : errmsg = libpq_gettext("insufficient data in \"D\" message");
825 : 0 : goto advance_and_error;
826 : : }
5258 tgl@sss.pgh.pa.us 827 :CBC 20192173 : rowbuf[i].len = vlen;
828 : :
829 : : /*
830 : : * rowbuf[i].value always points to the next address in the data
831 : : * buffer even if the value is NULL. This allows row processors to
832 : : * estimate data sizes more easily.
833 : : */
834 : 20192173 : rowbuf[i].value = conn->inBuffer + conn->inCursor;
835 : :
836 : : /* Skip over the data value */
8481 837 [ + + ]: 20192173 : if (vlen > 0)
838 : : {
5258 839 [ - + ]: 18857109 : if (pqSkipnchar(vlen, conn))
840 : : {
841 : : /* We should not run out of data here, so complain */
5258 tgl@sss.pgh.pa.us 842 :UBC 0 : errmsg = libpq_gettext("insufficient data in \"D\" message");
843 : 0 : goto advance_and_error;
844 : : }
845 : : }
846 : : }
847 : :
848 : : /* Process the collected row */
5258 tgl@sss.pgh.pa.us 849 :CBC 3965761 : errmsg = NULL;
5138 850 [ + - ]: 3965761 : if (pqRowProcessor(conn, &errmsg))
851 : 3965761 : return 0; /* normal, successful exit */
852 : :
853 : : /* pqRowProcessor failed, fall through to report it */
854 : :
5258 tgl@sss.pgh.pa.us 855 :UBC 0 : advance_and_error:
856 : :
857 : : /*
858 : : * Replace partially constructed result with an error result. First
859 : : * discard the old result to try to win back some memory.
860 : : */
8481 861 : 0 : pqClearAsyncResult(conn);
862 : :
863 : : /*
864 : : * If preceding code didn't provide an error message, assume "out of
865 : : * memory" was meant. The advantage of having this special case is that
866 : : * freeing the old result first greatly improves the odds that gettext()
867 : : * will succeed in providing a translation.
868 : : */
5258 869 [ # # ]: 0 : if (!errmsg)
870 : 0 : errmsg = libpq_gettext("out of memory for query result");
871 : :
2054 872 : 0 : appendPQExpBuffer(&conn->errorMessage, "%s\n", errmsg);
8481 873 : 0 : pqSaveErrorResult(conn);
874 : :
875 : : /*
876 : : * Show the message as fully consumed, else pqParseInput3 will overwrite
877 : : * our error with a complaint about that.
878 : : */
1995 879 : 0 : conn->inCursor = conn->inStart + 5 + msgLength;
880 : :
881 : : /*
882 : : * Return zero to allow input parsing to continue. Subsequent "D"
883 : : * messages will be ignored until we get to end of data, since an error
884 : : * result is already set up.
885 : : */
8481 886 : 0 : return 0;
887 : : }
888 : :
889 : :
890 : : /*
891 : : * Attempt to read an Error or Notice response message.
892 : : * This is possible in several places, so we break it out as a subroutine.
893 : : *
894 : : * Entry: 'E' or 'N' message type and length have already been consumed.
895 : : * Exit: returns 0 if successfully consumed message.
896 : : * returns EOF if not enough data.
897 : : */
898 : : int
8481 tgl@sss.pgh.pa.us 899 :CBC 49660 : pqGetErrorNotice3(PGconn *conn, bool isError)
900 : : {
7746 neilc@samurai.com 901 : 49660 : PGresult *res = NULL;
3798 tgl@sss.pgh.pa.us 902 : 49660 : bool have_position = false;
903 : : PQExpBufferData workBuf;
904 : : char id;
905 : :
906 : : /* If in pipeline mode, set error indicator for it */
1991 alvherre@alvh.no-ip. 907 [ + + + + ]: 49660 : if (isError && conn->pipelineStatus != PQ_PIPELINE_OFF)
908 : 98 : conn->pipelineStatus = PQ_PIPELINE_ABORTED;
909 : :
910 : : /*
911 : : * If this is an error message, pre-emptively clear any incomplete query
912 : : * result we may have. We'd just throw it away below anyway, and
913 : : * releasing it before collecting the error might avoid out-of-memory.
914 : : */
3058 tgl@sss.pgh.pa.us 915 [ + + ]: 49660 : if (isError)
916 : 31947 : pqClearAsyncResult(conn);
917 : :
918 : : /*
919 : : * Since the fields might be pretty long, we create a temporary
920 : : * PQExpBuffer rather than using conn->workBuffer. workBuffer is intended
921 : : * for stuff that is expected to be short. We shouldn't use
922 : : * conn->errorMessage either, since this might be only a notice.
923 : : */
8481 924 : 49660 : initPQExpBuffer(&workBuf);
925 : :
926 : : /*
927 : : * Make a PGresult to hold the accumulated fields. We temporarily lie
928 : : * about the result status, so that PQmakeEmptyPGresult doesn't uselessly
929 : : * copy conn->errorMessage.
930 : : *
931 : : * NB: This allocation can fail, if you run out of memory. The rest of the
932 : : * function handles that gracefully, and we still try to set the error
933 : : * message as the connection's error message.
934 : : */
7746 neilc@samurai.com 935 : 49660 : res = PQmakeEmptyPGresult(conn, PGRES_EMPTY_QUERY);
4069 heikki.linnakangas@i 936 [ + - ]: 49660 : if (res)
937 [ + + ]: 49660 : res->resultStatus = isError ? PGRES_FATAL_ERROR : PGRES_NONFATAL_ERROR;
938 : :
939 : : /*
940 : : * Read the fields and save into res.
941 : : *
942 : : * While at it, save the SQLSTATE in conn->last_sqlstate, and note whether
943 : : * we saw a PG_DIAG_STATEMENT_POSITION field.
944 : : */
945 : : for (;;)
946 : : {
8481 tgl@sss.pgh.pa.us 947 [ - + ]: 439058 : if (pqGetc(&id, conn))
8481 tgl@sss.pgh.pa.us 948 :UBC 0 : goto fail;
8481 tgl@sss.pgh.pa.us 949 [ + + ]:CBC 439058 : if (id == '\0')
950 : 49660 : break; /* terminator found */
951 [ - + ]: 389398 : if (pqGets(&workBuf, conn))
8481 tgl@sss.pgh.pa.us 952 :UBC 0 : goto fail;
8468 tgl@sss.pgh.pa.us 953 :CBC 389398 : pqSaveMessageField(res, id, workBuf.data);
3798 954 [ + + ]: 389398 : if (id == PG_DIAG_SQLSTATE)
955 : 49660 : strlcpy(conn->last_sqlstate, workBuf.data,
956 : : sizeof(conn->last_sqlstate));
957 [ + + ]: 339738 : else if (id == PG_DIAG_STATEMENT_POSITION)
958 : 8000 : have_position = true;
959 : : }
960 : :
961 : : /*
962 : : * Save the active query text, if any, into res as well; but only if we
963 : : * might need it for an error cursor display, which is only true if there
964 : : * is a PG_DIAG_STATEMENT_POSITION field.
965 : : */
1991 alvherre@alvh.no-ip. 966 [ + + + - : 49660 : if (have_position && res && conn->cmd_queue_head && conn->cmd_queue_head->query)
+ - + - ]
967 : 8000 : res->errQuery = pqResultStrdup(res, conn->cmd_queue_head->query);
968 : :
969 : : /*
970 : : * Now build the "overall" error message for PQresultErrorMessage.
971 : : */
8481 tgl@sss.pgh.pa.us 972 : 49660 : resetPQExpBuffer(&workBuf);
3798 973 : 49660 : pqBuildErrorMessage3(&workBuf, res, conn->verbosity, conn->show_context);
974 : :
975 : : /*
976 : : * Either save error as current async result, or just emit the notice.
977 : : */
978 [ + + ]: 49660 : if (isError)
979 : : {
3058 980 : 31947 : pqClearAsyncResult(conn); /* redundant, but be safe */
1651 981 [ + - ]: 31947 : if (res)
982 : : {
983 : 31947 : pqSetResultError(res, &workBuf, 0);
984 : 31947 : conn->result = res;
985 : : }
986 : : else
987 : : {
988 : : /* Fall back to using the internal-error processing paths */
1651 tgl@sss.pgh.pa.us 989 :UBC 0 : conn->error_result = true;
990 : : }
991 : :
3798 tgl@sss.pgh.pa.us 992 [ - + ]:CBC 31947 : if (PQExpBufferDataBroken(workBuf))
1381 peter@eisentraut.org 993 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
994 : : else
3798 tgl@sss.pgh.pa.us 995 :CBC 31947 : appendPQExpBufferStr(&conn->errorMessage, workBuf.data);
996 : : }
997 : : else
998 : : {
999 : : /* if we couldn't allocate the result set, just discard the NOTICE */
1000 [ + - ]: 17713 : if (res)
1001 : : {
1002 : : /*
1003 : : * We can cheat a little here and not copy the message. But if we
1004 : : * were unlucky enough to run out of memory while filling workBuf,
1005 : : * insert "out of memory", as in pqSetResultError.
1006 : : */
1855 1007 [ - + ]: 17713 : if (PQExpBufferDataBroken(workBuf))
1855 tgl@sss.pgh.pa.us 1008 :UBC 0 : res->errMsg = libpq_gettext("out of memory\n");
1009 : : else
1855 tgl@sss.pgh.pa.us 1010 :CBC 17713 : res->errMsg = workBuf.data;
3798 1011 [ + - ]: 17713 : if (res->noticeHooks.noticeRec != NULL)
3276 peter_e@gmx.net 1012 : 17713 : res->noticeHooks.noticeRec(res->noticeHooks.noticeRecArg, res);
3798 tgl@sss.pgh.pa.us 1013 : 17713 : PQclear(res);
1014 : : }
1015 : : }
1016 : :
1017 : 49660 : termPQExpBuffer(&workBuf);
1018 : 49660 : return 0;
1019 : :
3798 tgl@sss.pgh.pa.us 1020 :UBC 0 : fail:
1021 : 0 : PQclear(res);
1022 : 0 : termPQExpBuffer(&workBuf);
1023 : 0 : return EOF;
1024 : : }
1025 : :
1026 : : /*
1027 : : * Construct an error message from the fields in the given PGresult,
1028 : : * appending it to the contents of "msg".
1029 : : */
1030 : : void
3798 tgl@sss.pgh.pa.us 1031 :CBC 49663 : pqBuildErrorMessage3(PQExpBuffer msg, const PGresult *res,
1032 : : PGVerbosity verbosity, PGContextVisibility show_context)
1033 : : {
1034 : : const char *val;
1035 : 49663 : const char *querytext = NULL;
1036 : 49663 : int querypos = 0;
1037 : :
1038 : : /* If we couldn't allocate a PGresult, just say "out of memory" */
1039 [ - + ]: 49663 : if (res == NULL)
1040 : : {
2611 drowley@postgresql.o 1041 :UBC 0 : appendPQExpBufferStr(msg, libpq_gettext("out of memory\n"));
3798 tgl@sss.pgh.pa.us 1042 : 0 : return;
1043 : : }
1044 : :
1045 : : /*
1046 : : * If we don't have any broken-down fields, just return the base message.
1047 : : * This mainly applies if we're given a libpq-generated error result.
1048 : : */
3798 tgl@sss.pgh.pa.us 1049 [ - + ]:CBC 49663 : if (res->errFields == NULL)
1050 : : {
3798 tgl@sss.pgh.pa.us 1051 [ # # # # ]:UBC 0 : if (res->errMsg && res->errMsg[0])
1052 : 0 : appendPQExpBufferStr(msg, res->errMsg);
1053 : : else
2611 drowley@postgresql.o 1054 : 0 : appendPQExpBufferStr(msg, libpq_gettext("no error message available\n"));
3798 tgl@sss.pgh.pa.us 1055 : 0 : return;
1056 : : }
1057 : :
1058 : : /* Else build error message from relevant fields */
8401 peter_e@gmx.net 1059 :CBC 49663 : val = PQresultErrorField(res, PG_DIAG_SEVERITY);
8468 tgl@sss.pgh.pa.us 1060 [ + - ]: 49663 : if (val)
3798 1061 : 49663 : appendPQExpBuffer(msg, "%s: ", val);
1062 : :
2702 1063 [ + + ]: 49663 : if (verbosity == PQERRORS_SQLSTATE)
1064 : : {
1065 : : /*
1066 : : * If we have a SQLSTATE, print that and nothing else. If not (which
1067 : : * shouldn't happen for server-generated errors, but might possibly
1068 : : * happen for libpq-generated ones), fall back to TERSE format, as
1069 : : * that seems better than printing nothing at all.
1070 : : */
1071 : 115 : val = PQresultErrorField(res, PG_DIAG_SQLSTATE);
1072 [ + - ]: 115 : if (val)
1073 : : {
1074 : 115 : appendPQExpBuffer(msg, "%s\n", val);
1075 : 115 : return;
1076 : : }
2702 tgl@sss.pgh.pa.us 1077 :UBC 0 : verbosity = PQERRORS_TERSE;
1078 : : }
1079 : :
3798 tgl@sss.pgh.pa.us 1080 [ + + ]:CBC 49548 : if (verbosity == PQERRORS_VERBOSE)
1081 : : {
1082 : 3 : val = PQresultErrorField(res, PG_DIAG_SQLSTATE);
1083 [ + - ]: 3 : if (val)
1084 : 3 : appendPQExpBuffer(msg, "%s: ", val);
1085 : : }
8401 peter_e@gmx.net 1086 : 49548 : val = PQresultErrorField(res, PG_DIAG_MESSAGE_PRIMARY);
8468 tgl@sss.pgh.pa.us 1087 [ + - ]: 49548 : if (val)
3798 1088 : 49548 : appendPQExpBufferStr(msg, val);
8401 peter_e@gmx.net 1089 : 49548 : val = PQresultErrorField(res, PG_DIAG_STATEMENT_POSITION);
8468 tgl@sss.pgh.pa.us 1090 [ + + ]: 49548 : if (val)
1091 : : {
3798 1092 [ + + + - ]: 7999 : if (verbosity != PQERRORS_TERSE && res->errQuery != NULL)
1093 : : {
1094 : : /* emit position as a syntax cursor display */
1095 : 7995 : querytext = res->errQuery;
7471 1096 : 7995 : querypos = atoi(val);
1097 : : }
1098 : : else
1099 : : {
1100 : : /* emit position as text addition to primary message */
1101 : : /* translator: %s represents a digit string */
3798 1102 : 4 : appendPQExpBuffer(msg, libpq_gettext(" at character %s"),
1103 : : val);
1104 : : }
1105 : : }
1106 : : else
1107 : : {
8194 1108 : 41549 : val = PQresultErrorField(res, PG_DIAG_INTERNAL_POSITION);
1109 [ + + ]: 41549 : if (val)
1110 : : {
7471 1111 : 57 : querytext = PQresultErrorField(res, PG_DIAG_INTERNAL_QUERY);
3798 1112 [ + - + - ]: 57 : if (verbosity != PQERRORS_TERSE && querytext != NULL)
1113 : : {
1114 : : /* emit position as a syntax cursor display */
7471 1115 : 57 : querypos = atoi(val);
1116 : : }
1117 : : else
1118 : : {
1119 : : /* emit position as text addition to primary message */
1120 : : /* translator: %s represents a digit string */
3798 tgl@sss.pgh.pa.us 1121 :UBC 0 : appendPQExpBuffer(msg, libpq_gettext(" at character %s"),
1122 : : val);
1123 : : }
1124 : : }
1125 : : }
3798 tgl@sss.pgh.pa.us 1126 :CBC 49548 : appendPQExpBufferChar(msg, '\n');
1127 [ + + ]: 49548 : if (verbosity != PQERRORS_TERSE)
1128 : : {
7471 1129 [ + + + - ]: 49227 : if (querytext && querypos > 0)
3798 1130 : 8052 : reportErrorPosition(msg, querytext, querypos,
1131 : 8052 : res->client_encoding);
8401 peter_e@gmx.net 1132 : 49227 : val = PQresultErrorField(res, PG_DIAG_MESSAGE_DETAIL);
8468 tgl@sss.pgh.pa.us 1133 [ + + ]: 49227 : if (val)
3798 1134 : 8316 : appendPQExpBuffer(msg, libpq_gettext("DETAIL: %s\n"), val);
8401 peter_e@gmx.net 1135 : 49227 : val = PQresultErrorField(res, PG_DIAG_MESSAGE_HINT);
8468 tgl@sss.pgh.pa.us 1136 [ + + ]: 49227 : if (val)
3798 1137 : 3160 : appendPQExpBuffer(msg, libpq_gettext("HINT: %s\n"), val);
8194 1138 : 49227 : val = PQresultErrorField(res, PG_DIAG_INTERNAL_QUERY);
1139 [ + + ]: 49227 : if (val)
3798 1140 : 57 : appendPQExpBuffer(msg, libpq_gettext("QUERY: %s\n"), val);
1141 [ + + + + ]: 49227 : if (show_context == PQSHOW_CONTEXT_ALWAYS ||
1142 : 49012 : (show_context == PQSHOW_CONTEXT_ERRORS &&
1143 [ + + ]: 49012 : res->resultStatus == PGRES_FATAL_ERROR))
1144 : : {
4009 1145 : 31811 : val = PQresultErrorField(res, PG_DIAG_CONTEXT);
1146 [ + + ]: 31811 : if (val)
3798 1147 : 1620 : appendPQExpBuffer(msg, libpq_gettext("CONTEXT: %s\n"),
1148 : : val);
1149 : : }
1150 : : }
1151 [ + + ]: 49548 : if (verbosity == PQERRORS_VERBOSE)
1152 : : {
4958 1153 : 3 : val = PQresultErrorField(res, PG_DIAG_SCHEMA_NAME);
1154 [ - + ]: 3 : if (val)
3798 tgl@sss.pgh.pa.us 1155 :UBC 0 : appendPQExpBuffer(msg,
4958 1156 : 0 : libpq_gettext("SCHEMA NAME: %s\n"), val);
4958 tgl@sss.pgh.pa.us 1157 :CBC 3 : val = PQresultErrorField(res, PG_DIAG_TABLE_NAME);
1158 [ - + ]: 3 : if (val)
3798 tgl@sss.pgh.pa.us 1159 :UBC 0 : appendPQExpBuffer(msg,
4958 1160 : 0 : libpq_gettext("TABLE NAME: %s\n"), val);
4958 tgl@sss.pgh.pa.us 1161 :CBC 3 : val = PQresultErrorField(res, PG_DIAG_COLUMN_NAME);
1162 [ - + ]: 3 : if (val)
3798 tgl@sss.pgh.pa.us 1163 :UBC 0 : appendPQExpBuffer(msg,
4958 1164 : 0 : libpq_gettext("COLUMN NAME: %s\n"), val);
4958 tgl@sss.pgh.pa.us 1165 :CBC 3 : val = PQresultErrorField(res, PG_DIAG_DATATYPE_NAME);
1166 [ - + ]: 3 : if (val)
3798 tgl@sss.pgh.pa.us 1167 :UBC 0 : appendPQExpBuffer(msg,
4958 1168 : 0 : libpq_gettext("DATATYPE NAME: %s\n"), val);
4958 tgl@sss.pgh.pa.us 1169 :CBC 3 : val = PQresultErrorField(res, PG_DIAG_CONSTRAINT_NAME);
1170 [ - + ]: 3 : if (val)
3798 tgl@sss.pgh.pa.us 1171 :UBC 0 : appendPQExpBuffer(msg,
4958 1172 : 0 : libpq_gettext("CONSTRAINT NAME: %s\n"), val);
1173 : : }
3798 tgl@sss.pgh.pa.us 1174 [ + + ]:CBC 49548 : if (verbosity == PQERRORS_VERBOSE)
1175 : : {
1176 : : const char *valf;
1177 : : const char *vall;
1178 : :
8401 peter_e@gmx.net 1179 : 3 : valf = PQresultErrorField(res, PG_DIAG_SOURCE_FILE);
1180 : 3 : vall = PQresultErrorField(res, PG_DIAG_SOURCE_LINE);
1181 : 3 : val = PQresultErrorField(res, PG_DIAG_SOURCE_FUNCTION);
8468 tgl@sss.pgh.pa.us 1182 [ - + - - : 3 : if (val || valf || vall)
- - ]
1183 : : {
3798 1184 : 3 : appendPQExpBufferStr(msg, libpq_gettext("LOCATION: "));
8468 1185 [ + - ]: 3 : if (val)
3798 1186 : 3 : appendPQExpBuffer(msg, libpq_gettext("%s, "), val);
8468 1187 [ + - + - ]: 3 : if (valf && vall) /* unlikely we'd have just one */
3798 1188 : 3 : appendPQExpBuffer(msg, libpq_gettext("%s:%s"),
1189 : : valf, vall);
1190 : 3 : appendPQExpBufferChar(msg, '\n');
1191 : : }
1192 : : }
1193 : : }
1194 : :
1195 : : /*
1196 : : * Add an error-location display to the error message under construction.
1197 : : *
1198 : : * The cursor location is measured in logical characters; the query string
1199 : : * is presumed to be in the specified encoding.
1200 : : */
1201 : : static void
7471 1202 : 8052 : reportErrorPosition(PQExpBuffer msg, const char *query, int loc, int encoding)
1203 : : {
1204 : : #define DISPLAY_SIZE 60 /* screen width limit, in screen cols */
1205 : : #define MIN_RIGHT_CUT 10 /* try to keep this far away from EOL */
1206 : :
1207 : : char *wquery;
1208 : : int slen,
1209 : : cno,
1210 : : i,
1211 : : *qidx,
1212 : : *scridx,
1213 : : qoffset,
1214 : : scroffset,
1215 : : ibeg,
1216 : : iend,
1217 : : loc_line;
1218 : : bool mb_encoding,
1219 : : beg_trunc,
1220 : : end_trunc;
1221 : :
1222 : : /* Convert loc from 1-based to 0-based; no-op if out of range */
7270 1223 : 8052 : loc--;
1224 [ - + ]: 8052 : if (loc < 0)
7270 tgl@sss.pgh.pa.us 1225 :UBC 0 : return;
1226 : :
1227 : : /* Need a writable copy of the query */
7471 tgl@sss.pgh.pa.us 1228 :CBC 8052 : wquery = strdup(query);
1229 [ - + ]: 8052 : if (wquery == NULL)
7471 tgl@sss.pgh.pa.us 1230 :UBC 0 : return; /* fail silently if out of memory */
1231 : :
1232 : : /*
1233 : : * Each character might occupy multiple physical bytes in the string, and
1234 : : * in some Far Eastern character sets it might take more than one screen
1235 : : * column as well. We compute the starting byte offset and starting
1236 : : * screen column of each logical character, and store these in qidx[] and
1237 : : * scridx[] respectively.
1238 : : */
1239 : :
1240 : : /*
1241 : : * We need a safe allocation size.
1242 : : *
1243 : : * The only caller of reportErrorPosition() is pqBuildErrorMessage3(); it
1244 : : * gets its query from either a PQresultErrorField() or a PGcmdQueueEntry,
1245 : : * both of which must have fit into conn->inBuffer/outBuffer. So slen fits
1246 : : * inside an int, but we can't assume that (slen * sizeof(int)) fits
1247 : : * inside a size_t.
1248 : : */
7270 tgl@sss.pgh.pa.us 1249 :CBC 8052 : slen = strlen(wquery) + 1;
290 jchampion@postgresql 1250 [ - + ]: 8052 : if (slen > SIZE_MAX / sizeof(int))
1251 : : {
290 jchampion@postgresql 1252 :UBC 0 : free(wquery);
1253 : 0 : return;
1254 : : }
1255 : :
7471 tgl@sss.pgh.pa.us 1256 :CBC 8052 : qidx = (int *) malloc(slen * sizeof(int));
1257 [ - + ]: 8052 : if (qidx == NULL)
1258 : : {
7471 tgl@sss.pgh.pa.us 1259 :UBC 0 : free(wquery);
1260 : 0 : return;
1261 : : }
7471 tgl@sss.pgh.pa.us 1262 :CBC 8052 : scridx = (int *) malloc(slen * sizeof(int));
1263 [ - + ]: 8052 : if (scridx == NULL)
1264 : : {
7471 tgl@sss.pgh.pa.us 1265 :UBC 0 : free(qidx);
1266 : 0 : free(wquery);
1267 : 0 : return;
1268 : : }
1269 : :
1270 : : /* We can optimize a bit if it's a single-byte encoding */
7270 tgl@sss.pgh.pa.us 1271 :CBC 8052 : mb_encoding = (pg_encoding_max_length(encoding) != 1);
1272 : :
1273 : : /*
1274 : : * Within the scanning loop, cno is the current character's logical
1275 : : * number, qoffset is its offset in wquery, and scroffset is its starting
1276 : : * logical screen column (all indexed from 0). "loc" is the logical
1277 : : * character number of the error location. We scan to determine loc_line
1278 : : * (the 1-based line number containing loc) and ibeg/iend (first character
1279 : : * number and last+1 character number of the line containing loc). Note
1280 : : * that qidx[] and scridx[] are filled only as far as iend.
1281 : : */
7471 1282 : 8052 : qoffset = 0;
1283 : 8052 : scroffset = 0;
7270 1284 : 8052 : loc_line = 1;
1285 : 8052 : ibeg = 0;
1286 : 8052 : iend = -1; /* -1 means not set yet */
1287 : :
1288 [ + + ]: 473913 : for (cno = 0; wquery[qoffset] != '\0'; cno++)
1289 : : {
7267 bruce@momjian.us 1290 : 466851 : char ch = wquery[qoffset];
1291 : :
7270 tgl@sss.pgh.pa.us 1292 : 466851 : qidx[cno] = qoffset;
1293 : 466851 : scridx[cno] = scroffset;
1294 : :
1295 : : /*
1296 : : * Replace tabs with spaces in the writable copy. (Later we might
1297 : : * want to think about coping with their variable screen width, but
1298 : : * not today.)
1299 : : */
1300 [ + + ]: 466851 : if (ch == '\t')
1301 : 653 : wquery[qoffset] = ' ';
1302 : :
1303 : : /*
1304 : : * If end-of-line, count lines and mark positions. Each \r or \n
1305 : : * counts as a line except when \r \n appear together.
1306 : : */
1307 [ + - + + ]: 466198 : else if (ch == '\r' || ch == '\n')
1308 : : {
1309 [ + + ]: 3472 : if (cno < loc)
1310 : : {
1311 [ + - + + ]: 2482 : if (ch == '\r' ||
1312 : 2478 : cno == 0 ||
1313 [ + - ]: 2478 : wquery[qidx[cno - 1]] != '\r')
1314 : 2482 : loc_line++;
1315 : : /* extract beginning = last line start before loc. */
1316 : 2482 : ibeg = cno + 1;
1317 : : }
1318 : : else
1319 : : {
1320 : : /* set extract end. */
1321 : 990 : iend = cno;
1322 : : /* done scanning. */
1323 : 990 : break;
1324 : : }
1325 : : }
1326 : :
1327 : : /* Advance */
1328 [ + + ]: 465861 : if (mb_encoding)
1329 : : {
1330 : : int w;
1331 : :
1332 : 465673 : w = pg_encoding_dsplen(encoding, &wquery[qoffset]);
1333 : : /* treat any non-tab control chars as width 1 */
1334 [ + + ]: 465673 : if (w <= 0)
1335 : 2482 : w = 1;
1336 : 465673 : scroffset += w;
1907 1337 : 465673 : qoffset += PQmblenBounded(&wquery[qoffset], encoding);
1338 : : }
1339 : : else
1340 : : {
1341 : : /* We assume wide chars only exist in multibyte encodings */
7270 1342 : 188 : scroffset++;
1343 : 188 : qoffset++;
1344 : : }
1345 : : }
1346 : : /* Fix up if we didn't find an end-of-line after loc */
1347 [ + + ]: 8052 : if (iend < 0)
1348 : : {
1349 : 7062 : iend = cno; /* query length in chars, +1 */
1350 : 7062 : qidx[iend] = qoffset;
1351 : 7062 : scridx[iend] = scroffset;
1352 : : }
1353 : :
1354 : : /* Print only if loc is within computed query length */
1355 [ + + ]: 8052 : if (loc <= cno)
1356 : : {
1357 : : /* If the line extracted is too long, we truncate it. */
7471 1358 : 8040 : beg_trunc = false;
1359 : 8040 : end_trunc = false;
1360 [ + + ]: 8040 : if (scridx[iend] - scridx[ibeg] > DISPLAY_SIZE)
1361 : : {
1362 : : /*
1363 : : * We first truncate right if it is enough. This code might be
1364 : : * off a space or so on enforcing MIN_RIGHT_CUT if there's a wide
1365 : : * character right there, but that should be okay.
1366 : : */
1367 [ + + ]: 2293 : if (scridx[ibeg] + DISPLAY_SIZE >= scridx[loc] + MIN_RIGHT_CUT)
1368 : : {
1369 [ + + ]: 24627 : while (scridx[iend] - scridx[ibeg] > DISPLAY_SIZE)
1370 : 23224 : iend--;
1371 : 1403 : end_trunc = true;
1372 : : }
1373 : : else
1374 : : {
1375 : : /* Truncate right if not too close to loc. */
1376 [ + + ]: 11338 : while (scridx[loc] + MIN_RIGHT_CUT < scridx[iend])
1377 : : {
1378 : 10448 : iend--;
1379 : 10448 : end_trunc = true;
1380 : : }
1381 : :
1382 : : /* Truncate left if still too long. */
1383 [ + + ]: 19119 : while (scridx[iend] - scridx[ibeg] > DISPLAY_SIZE)
1384 : : {
1385 : 18229 : ibeg++;
1386 : 18229 : beg_trunc = true;
1387 : : }
1388 : : }
1389 : : }
1390 : :
1391 : : /* truncate working copy at desired endpoint */
1392 : 8040 : wquery[qidx[iend]] = '\0';
1393 : :
1394 : : /* Begin building the finished message. */
1395 : 8040 : i = msg->len;
1396 : 8040 : appendPQExpBuffer(msg, libpq_gettext("LINE %d: "), loc_line);
1397 [ + + ]: 8040 : if (beg_trunc)
1398 : 890 : appendPQExpBufferStr(msg, "...");
1399 : :
1400 : : /*
1401 : : * While we have the prefix in the msg buffer, compute its screen
1402 : : * width.
1403 : : */
1404 : 8040 : scroffset = 0;
1907 1405 [ + + ]: 75058 : for (; i < msg->len; i += PQmblenBounded(&msg->data[i], encoding))
1406 : : {
7267 bruce@momjian.us 1407 : 67018 : int w = pg_encoding_dsplen(encoding, &msg->data[i]);
1408 : :
7471 tgl@sss.pgh.pa.us 1409 [ - + ]: 67018 : if (w <= 0)
7471 tgl@sss.pgh.pa.us 1410 :UBC 0 : w = 1;
7471 tgl@sss.pgh.pa.us 1411 :CBC 67018 : scroffset += w;
1412 : : }
1413 : :
1414 : : /* Finish up the LINE message line. */
1415 : 8040 : appendPQExpBufferStr(msg, &wquery[qidx[ibeg]]);
1416 [ + + ]: 8040 : if (end_trunc)
1417 : 2046 : appendPQExpBufferStr(msg, "...");
1418 : 8040 : appendPQExpBufferChar(msg, '\n');
1419 : :
1420 : : /* Now emit the cursor marker line. */
1421 : 8040 : scroffset += scridx[loc] - scridx[ibeg];
1422 [ + + ]: 258278 : for (i = 0; i < scroffset; i++)
1423 : 250238 : appendPQExpBufferChar(msg, ' ');
1424 : 8040 : appendPQExpBufferChar(msg, '^');
1425 : 8040 : appendPQExpBufferChar(msg, '\n');
1426 : : }
1427 : :
1428 : : /* Clean up. */
1429 : 8052 : free(scridx);
1430 : 8052 : free(qidx);
1431 : 8052 : free(wquery);
1432 : : }
1433 : :
1434 : :
1435 : : /*
1436 : : * Attempt to read a NegotiateProtocolVersion message. Sets conn->pversion
1437 : : * to the version that's negotiated by the server.
1438 : : *
1439 : : * Entry: 'v' message type and length have already been consumed.
1440 : : * Exit: returns 0 if successfully consumed message.
1441 : : * returns 1 on failure. The error message is filled in.
1442 : : */
1443 : : int
1379 peter@eisentraut.org 1444 : 15811 : pqGetNegotiateProtocolVersion3(PGconn *conn)
1445 : : {
1446 : : int their_version;
1447 : : int num;
1448 : : bool found_test_protocol_negotiation;
1449 : : bool expect_test_protocol_negotiation;
1450 : :
1451 : : /*
1452 : : * During 19beta only, if protocol grease is in use, assume that it's the
1453 : : * cause of any invalid messages encountered below. We'll print extra
1454 : : * information for the end user in that case.
1455 : : */
185 jchampion@postgresql 1456 : 15811 : bool need_grease_info = (conn->max_pversion == PG_PROTOCOL_GREASE);
1457 : :
512 heikki.linnakangas@i 1458 [ - + ]: 15811 : if (pqGetInt(&their_version, 4, conn) != 0)
512 heikki.linnakangas@i 1459 :UBC 0 : goto eof;
1460 : :
1379 peter@eisentraut.org 1461 [ - + ]:CBC 15811 : if (pqGetInt(&num, 4, conn) != 0)
512 heikki.linnakangas@i 1462 :UBC 0 : goto eof;
1463 : :
1464 : : /*
1465 : : * Check the protocol version.
1466 : : *
1467 : : * PG_PROTOCOL_GREASE is intentionally unsupported and reserved. It's
1468 : : * higher than any real version, so check for that first, to get the most
1469 : : * specific error message. Then check the upper and lower bounds.
1470 : : */
202 jchampion@postgresql 1471 [ - + ]:CBC 15811 : if (their_version == PG_PROTOCOL_GREASE)
1472 : : {
202 jchampion@postgresql 1473 :UBC 0 : libpq_append_conn_error(conn, "received invalid protocol negotiation message: server requested \"grease\" protocol version 3.9999");
1474 : 0 : goto failure;
1475 : : }
1476 : :
512 heikki.linnakangas@i 1477 [ - + ]:CBC 15811 : if (their_version > conn->pversion)
1478 : : {
512 heikki.linnakangas@i 1479 :UBC 0 : libpq_append_conn_error(conn, "received invalid protocol negotiation message: server requested downgrade to a higher-numbered version");
1480 : 0 : goto failure;
1481 : : }
1482 : :
512 heikki.linnakangas@i 1483 [ - + ]:CBC 15811 : if (their_version < PG_PROTOCOL(3, 0))
1484 : : {
512 heikki.linnakangas@i 1485 :UBC 0 : libpq_append_conn_error(conn, "received invalid protocol negotiation message: server requested downgrade to pre-3.0 protocol version");
1486 : 0 : goto failure;
1487 : : }
1488 : :
1489 : : /* 3.1 never existed, we went straight from 3.0 to 3.2 */
216 jchampion@postgresql 1490 [ - + ]:CBC 15811 : if (their_version == PG_PROTOCOL_RESERVED_31)
1491 : : {
427 peter@eisentraut.org 1492 :UBC 0 : libpq_append_conn_error(conn, "received invalid protocol negotiation message: server requested downgrade to non-existent 3.1 protocol version");
512 heikki.linnakangas@i 1493 : 0 : goto failure;
1494 : : }
1495 : :
512 heikki.linnakangas@i 1496 [ - + ]:CBC 15811 : if (num < 0)
1497 : : {
512 heikki.linnakangas@i 1498 :UBC 0 : libpq_append_conn_error(conn, "received invalid protocol negotiation message: server reported negative number of unsupported parameters");
1499 : 0 : goto failure;
1500 : : }
1501 : :
512 heikki.linnakangas@i 1502 [ - + - - ]:CBC 15811 : if (their_version == conn->pversion && num == 0)
1503 : : {
512 heikki.linnakangas@i 1504 :UBC 0 : libpq_append_conn_error(conn, "received invalid protocol negotiation message: server negotiated but asks for no changes");
1505 : 0 : goto failure;
1506 : : }
1507 : :
512 heikki.linnakangas@i 1508 [ - + ]:CBC 15811 : if (their_version < conn->min_pversion)
1509 : : {
427 peter@eisentraut.org 1510 :UBC 0 : libpq_append_conn_error(conn, "server only supports protocol version %d.%d, but \"%s\" was set to %d.%d",
1511 : : PG_PROTOCOL_MAJOR(their_version),
1512 : : PG_PROTOCOL_MINOR(their_version),
1513 : : "min_protocol_version",
512 heikki.linnakangas@i 1514 : 0 : PG_PROTOCOL_MAJOR(conn->min_pversion),
1515 : 0 : PG_PROTOCOL_MINOR(conn->min_pversion));
1516 : :
185 jchampion@postgresql 1517 : 0 : need_grease_info = false; /* this is valid server behavior */
512 heikki.linnakangas@i 1518 : 0 : goto failure;
1519 : : }
1520 : :
1521 : : /* the version is acceptable */
512 heikki.linnakangas@i 1522 :CBC 15811 : conn->pversion = their_version;
1523 : :
1524 : : /*
1525 : : * Check that all expected unsupported parameters are reported by the
1526 : : * server.
1527 : : */
185 jchampion@postgresql 1528 : 15811 : found_test_protocol_negotiation = false;
1529 : 15811 : expect_test_protocol_negotiation = (conn->max_pversion == PG_PROTOCOL_GREASE);
1530 : :
512 heikki.linnakangas@i 1531 [ + + ]: 31622 : for (int i = 0; i < num; i++)
1532 : : {
1533 [ - + ]: 15811 : if (pqGets(&conn->workBuffer, conn))
1534 : : {
512 heikki.linnakangas@i 1535 :UBC 0 : goto eof;
1536 : : }
512 heikki.linnakangas@i 1537 [ - + ]:CBC 15811 : if (strncmp(conn->workBuffer.data, "_pq_.", 5) != 0)
1538 : : {
427 peter@eisentraut.org 1539 :UBC 0 : libpq_append_conn_error(conn, "received invalid protocol negotiation message: server reported unsupported parameter name without a \"%s\" prefix (\"%s\")", "_pq_.", conn->workBuffer.data);
512 heikki.linnakangas@i 1540 : 0 : goto failure;
1541 : : }
1542 : :
1543 : : /* Check if this is the expected test parameter */
185 jchampion@postgresql 1544 [ + - ]:CBC 15811 : if (expect_test_protocol_negotiation &&
1545 [ + - ]: 15811 : strcmp(conn->workBuffer.data, "_pq_.test_protocol_negotiation") == 0)
1546 : : {
1547 : 15811 : found_test_protocol_negotiation = true;
1548 : : }
1549 : : else
1550 : : {
185 jchampion@postgresql 1551 :UBC 0 : libpq_append_conn_error(conn, "received invalid protocol negotiation message: server reported an unsupported parameter that was not requested (\"%s\")",
1552 : : conn->workBuffer.data);
1553 : 0 : goto failure;
1554 : : }
1555 : : }
1556 : :
1557 : : /*
1558 : : * If we requested protocol grease, the server must report
1559 : : * _pq_.test_protocol_negotiation as unsupported. This ensures
1560 : : * comprehensive NegotiateProtocolVersion implementation.
1561 : : */
185 jchampion@postgresql 1562 [ + - - + ]:CBC 15811 : if (expect_test_protocol_negotiation && !found_test_protocol_negotiation)
1563 : : {
90 jchampion@postgresql 1564 :UBC 0 : libpq_append_conn_error(conn, "server did not report the unsupported \"%s\" parameter in its protocol negotiation message",
1565 : : "_pq_.test_protocol_negotiation");
512 heikki.linnakangas@i 1566 : 0 : goto failure;
1567 : : }
1568 : :
1379 peter@eisentraut.org 1569 :CBC 15811 : return 0;
1570 : :
512 heikki.linnakangas@i 1571 :UBC 0 : eof:
495 michael@paquier.xyz 1572 : 0 : libpq_append_conn_error(conn, "received invalid protocol negotiation message: message too short");
512 heikki.linnakangas@i 1573 : 0 : failure:
185 jchampion@postgresql 1574 [ # # ]: 0 : if (need_grease_info)
1575 : 0 : libpq_append_grease_info(conn);
512 heikki.linnakangas@i 1576 : 0 : conn->asyncStatus = PGASYNC_READY;
1577 : 0 : pqSaveErrorResult(conn);
1578 : 0 : return 1;
1579 : : }
1580 : :
1581 : :
1582 : : /*
1583 : : * Attempt to read a ParameterStatus message.
1584 : : * This is possible in several places, so we break it out as a subroutine.
1585 : : *
1586 : : * Entry: 'S' message type and length have already been consumed.
1587 : : * Exit: returns 0 if successfully consumed message.
1588 : : * returns EOF if not enough data.
1589 : : */
1590 : : static int
8481 tgl@sss.pgh.pa.us 1591 :CBC 241351 : getParameterStatus(PGconn *conn)
1592 : : {
1593 : : PQExpBufferData valueBuf;
1594 : :
1595 : : /* Get the parameter name */
1596 [ - + ]: 241351 : if (pqGets(&conn->workBuffer, conn))
8481 tgl@sss.pgh.pa.us 1597 :UBC 0 : return EOF;
1598 : : /* Get the parameter value (could be large) */
8481 tgl@sss.pgh.pa.us 1599 :CBC 241351 : initPQExpBuffer(&valueBuf);
1600 [ - + ]: 241351 : if (pqGets(&valueBuf, conn))
1601 : : {
8481 tgl@sss.pgh.pa.us 1602 :UBC 0 : termPQExpBuffer(&valueBuf);
1603 : 0 : return EOF;
1604 : : }
1605 : : /* And save it */
370 heikki.linnakangas@i 1606 [ - + ]:CBC 241351 : if (!pqSaveParameterStatus(conn, conn->workBuffer.data, valueBuf.data))
1607 : : {
370 heikki.linnakangas@i 1608 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
1609 : 0 : handleFatalError(conn);
1610 : : }
8481 tgl@sss.pgh.pa.us 1611 :CBC 241351 : termPQExpBuffer(&valueBuf);
1612 : 241351 : return 0;
1613 : : }
1614 : :
1615 : : /*
1616 : : * parseInput subroutine to read a BackendKeyData message.
1617 : : * Entry: 'K' message type and length have already been consumed.
1618 : : * Exit: returns 0 if successfully consumed message.
1619 : : * returns EOF if not enough data.
1620 : : */
1621 : : static int
512 heikki.linnakangas@i 1622 : 15476 : getBackendKeyData(PGconn *conn, int msgLength)
1623 : : {
1624 : : int cancel_key_len;
1625 : :
1626 [ - + ]: 15476 : if (conn->be_cancel_key)
1627 : : {
512 heikki.linnakangas@i 1628 :UBC 0 : free(conn->be_cancel_key);
1629 : 0 : conn->be_cancel_key = NULL;
1630 : 0 : conn->be_cancel_key_len = 0;
1631 : : }
1632 : :
512 heikki.linnakangas@i 1633 [ - + ]:CBC 15476 : if (pqGetInt(&(conn->be_pid), 4, conn))
512 heikki.linnakangas@i 1634 :UBC 0 : return EOF;
1635 : :
512 heikki.linnakangas@i 1636 :CBC 15476 : cancel_key_len = 5 + msgLength - (conn->inCursor - conn->inStart);
1637 : :
370 1638 [ + + - + ]: 15476 : if (cancel_key_len != 4 && conn->pversion == PG_PROTOCOL(3, 0))
1639 : : {
370 heikki.linnakangas@i 1640 :UBC 0 : libpq_append_conn_error(conn, "received invalid BackendKeyData message: cancel key with length %d not allowed in protocol version 3.0 (must be 4 bytes)", cancel_key_len);
1641 : 0 : handleFatalError(conn);
1642 : 0 : return 0;
1643 : : }
1644 : :
370 heikki.linnakangas@i 1645 [ - + ]:CBC 15476 : if (cancel_key_len < 4)
1646 : : {
370 heikki.linnakangas@i 1647 :UBC 0 : libpq_append_conn_error(conn, "received invalid BackendKeyData message: cancel key with length %d is too short (minimum 4 bytes)", cancel_key_len);
1648 : 0 : handleFatalError(conn);
1649 : 0 : return 0;
1650 : : }
1651 : :
370 heikki.linnakangas@i 1652 [ - + ]:CBC 15476 : if (cancel_key_len > 256)
1653 : : {
370 heikki.linnakangas@i 1654 :UBC 0 : libpq_append_conn_error(conn, "received invalid BackendKeyData message: cancel key with length %d is too long (maximum 256 bytes)", cancel_key_len);
1655 : 0 : handleFatalError(conn);
1656 : 0 : return 0;
1657 : : }
1658 : :
512 heikki.linnakangas@i 1659 :CBC 15476 : conn->be_cancel_key = malloc(cancel_key_len);
1660 [ - + ]: 15476 : if (conn->be_cancel_key == NULL)
1661 : : {
512 heikki.linnakangas@i 1662 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
370 1663 : 0 : handleFatalError(conn);
1664 : 0 : return 0;
1665 : : }
512 heikki.linnakangas@i 1666 [ - + ]:CBC 15476 : if (pqGetnchar(conn->be_cancel_key, cancel_key_len, conn))
1667 : : {
512 heikki.linnakangas@i 1668 :UBC 0 : free(conn->be_cancel_key);
1669 : 0 : conn->be_cancel_key = NULL;
1670 : 0 : return EOF;
1671 : : }
512 heikki.linnakangas@i 1672 :CBC 15476 : conn->be_cancel_key_len = cancel_key_len;
1673 : 15476 : return 0;
1674 : : }
1675 : :
1676 : :
1677 : : /*
1678 : : * Attempt to read a Notify response message.
1679 : : * This is possible in several places, so we break it out as a subroutine.
1680 : : *
1681 : : * Entry: 'A' message type and length have already been consumed.
1682 : : * Exit: returns 0 if successfully consumed Notify message.
1683 : : * returns EOF if not enough data.
1684 : : */
1685 : : static int
8481 tgl@sss.pgh.pa.us 1686 : 53 : getNotify(PGconn *conn)
1687 : : {
1688 : : int be_pid;
1689 : : char *svname;
1690 : : int nmlen;
1691 : : int extralen;
1692 : : PGnotify *newNotify;
1693 : :
1694 [ - + ]: 53 : if (pqGetInt(&be_pid, 4, conn))
8481 tgl@sss.pgh.pa.us 1695 :UBC 0 : return EOF;
8481 tgl@sss.pgh.pa.us 1696 [ - + ]:CBC 53 : if (pqGets(&conn->workBuffer, conn))
8481 tgl@sss.pgh.pa.us 1697 :UBC 0 : return EOF;
1698 : : /* must save name while getting extra string */
8468 tgl@sss.pgh.pa.us 1699 :CBC 53 : svname = strdup(conn->workBuffer.data);
1700 [ - + ]: 53 : if (!svname)
1701 : : {
1702 : : /*
1703 : : * Notify messages can arrive at any state, so we cannot associate the
1704 : : * error with any particular query. There's no way to return back an
1705 : : * "async error", so the best we can do is drop the connection. That
1706 : : * seems better than silently ignoring the notification.
1707 : : */
370 heikki.linnakangas@i 1708 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
1709 : 0 : handleFatalError(conn);
1710 : 0 : return 0;
1711 : : }
8468 tgl@sss.pgh.pa.us 1712 [ - + ]:CBC 53 : if (pqGets(&conn->workBuffer, conn))
1713 : : {
370 heikki.linnakangas@i 1714 :UBC 0 : free(svname);
8468 tgl@sss.pgh.pa.us 1715 : 0 : return EOF;
1716 : : }
1717 : :
1718 : : /*
1719 : : * Store the strings right after the PGnotify structure so it can all be
1720 : : * freed at once. We don't use NAMEDATALEN because we don't want to tie
1721 : : * this interface to a specific server name length.
1722 : : */
8468 tgl@sss.pgh.pa.us 1723 :CBC 53 : nmlen = strlen(svname);
1724 : 53 : extralen = strlen(conn->workBuffer.data);
1725 : 53 : newNotify = (PGnotify *) malloc(sizeof(PGnotify) + nmlen + extralen + 2);
370 heikki.linnakangas@i 1726 [ - + ]: 53 : if (!newNotify)
1727 : : {
370 heikki.linnakangas@i 1728 :UBC 0 : free(svname);
1729 : 0 : libpq_append_conn_error(conn, "out of memory");
1730 : 0 : handleFatalError(conn);
1731 : 0 : return 0;
1732 : : }
1733 : :
370 heikki.linnakangas@i 1734 :CBC 53 : newNotify->relname = (char *) newNotify + sizeof(PGnotify);
1735 : 53 : strcpy(newNotify->relname, svname);
1736 : 53 : newNotify->extra = newNotify->relname + nmlen + 1;
1737 : 53 : strcpy(newNotify->extra, conn->workBuffer.data);
1738 : 53 : newNotify->be_pid = be_pid;
1739 : 53 : newNotify->next = NULL;
1740 [ + + ]: 53 : if (conn->notifyTail)
1741 : 29 : conn->notifyTail->next = newNotify;
1742 : : else
1743 : 24 : conn->notifyHead = newNotify;
1744 : 53 : conn->notifyTail = newNotify;
1745 : :
8468 tgl@sss.pgh.pa.us 1746 : 53 : free(svname);
1747 : 53 : return 0;
1748 : : }
1749 : :
1750 : : /*
1751 : : * getCopyStart - process CopyInResponse, CopyOutResponse or
1752 : : * CopyBothResponse message
1753 : : *
1754 : : * parseInput already read the message type and length.
1755 : : */
1756 : : static int
1757 : 6852 : getCopyStart(PGconn *conn, ExecStatusType copytype)
1758 : : {
1759 : : PGresult *result;
1760 : : int nfields;
1761 : : int i;
1762 : :
1763 : 6852 : result = PQmakeEmptyPGresult(conn, copytype);
7746 neilc@samurai.com 1764 [ - + ]: 6852 : if (!result)
7746 neilc@samurai.com 1765 :UBC 0 : goto failure;
1766 : :
8468 tgl@sss.pgh.pa.us 1767 [ - + ]:CBC 6852 : if (pqGetc(&conn->copy_is_binary, conn))
7746 neilc@samurai.com 1768 :UBC 0 : goto failure;
8468 tgl@sss.pgh.pa.us 1769 :CBC 6852 : result->binary = conn->copy_is_binary;
1770 : : /* the next two bytes are the number of fields */
1771 [ - + ]: 6852 : if (pqGetInt(&(result->numAttributes), 2, conn))
7746 neilc@samurai.com 1772 :UBC 0 : goto failure;
8468 tgl@sss.pgh.pa.us 1773 :CBC 6852 : nfields = result->numAttributes;
1774 : :
1775 : : /* allocate space for the attribute descriptors */
1776 [ + + ]: 6852 : if (nfields > 0)
1777 : : {
1778 : 5698 : result->attDescs = (PGresAttDesc *)
3298 peter_e@gmx.net 1779 : 5698 : pqResultAlloc(result, nfields * sizeof(PGresAttDesc), true);
7746 neilc@samurai.com 1780 [ - + ]: 5698 : if (!result->attDescs)
7746 neilc@samurai.com 1781 :UBC 0 : goto failure;
7746 neilc@samurai.com 1782 [ + - + - :CBC 62454 : MemSet(result->attDescs, 0, nfields * sizeof(PGresAttDesc));
+ - + + +
+ ]
1783 : : }
1784 : :
8468 tgl@sss.pgh.pa.us 1785 [ + + ]: 25847 : for (i = 0; i < nfields; i++)
1786 : : {
1787 : : int format;
1788 : :
1789 [ - + ]: 18995 : if (pqGetInt(&format, 2, conn))
7746 neilc@samurai.com 1790 :UBC 0 : goto failure;
1791 : :
1792 : : /*
1793 : : * Since pqGetInt treats 2-byte integers as unsigned, we need to
1794 : : * coerce these results to signed form.
1795 : : */
8468 tgl@sss.pgh.pa.us 1796 :CBC 18995 : format = (int) ((int16) format);
1797 : 18995 : result->attDescs[i].format = format;
1798 : : }
1799 : :
1800 : : /* Success! */
1801 : 6852 : conn->result = result;
8481 1802 : 6852 : return 0;
1803 : :
7746 neilc@samurai.com 1804 :UBC 0 : failure:
1805 : 0 : PQclear(result);
1806 : 0 : return EOF;
1807 : : }
1808 : :
1809 : : /*
1810 : : * getReadyForQuery - process ReadyForQuery message
1811 : : */
1812 : : static int
8468 tgl@sss.pgh.pa.us 1813 :CBC 425342 : getReadyForQuery(PGconn *conn)
1814 : : {
1815 : : char xact_status;
1816 : :
1817 [ - + ]: 425342 : if (pqGetc(&xact_status, conn))
8468 tgl@sss.pgh.pa.us 1818 :UBC 0 : return EOF;
8468 tgl@sss.pgh.pa.us 1819 [ + + + - ]:CBC 425342 : switch (xact_status)
1820 : : {
1821 : 330882 : case 'I':
1822 : 330882 : conn->xactStatus = PQTRANS_IDLE;
1823 : 330882 : break;
1824 : 93260 : case 'T':
1825 : 93260 : conn->xactStatus = PQTRANS_INTRANS;
1826 : 93260 : break;
1827 : 1200 : case 'E':
1828 : 1200 : conn->xactStatus = PQTRANS_INERROR;
1829 : 1200 : break;
8468 tgl@sss.pgh.pa.us 1830 :UBC 0 : default:
1831 : 0 : conn->xactStatus = PQTRANS_UNKNOWN;
1832 : 0 : break;
1833 : : }
1834 : :
8468 tgl@sss.pgh.pa.us 1835 :CBC 425342 : return 0;
1836 : : }
1837 : :
1838 : : /*
1839 : : * getCopyDataMessage - fetch next CopyData message, process async messages
1840 : : *
1841 : : * Returns length word of CopyData message (> 0), or 0 if no complete
1842 : : * message available, -1 if end of copy, -2 if error.
1843 : : */
1844 : : static int
6800 1845 : 3150518 : getCopyDataMessage(PGconn *conn)
1846 : : {
1847 : : char id;
1848 : : int msgLength;
1849 : : int avail;
1850 : :
1851 : : for (;;)
1852 : : {
1853 : : /*
1854 : : * Do we have the next input message? To make life simpler for async
1855 : : * callers, we keep returning 0 until the next message is fully
1856 : : * available, even if it is not Copy Data.
1857 : : */
8468 1858 : 3150558 : conn->inCursor = conn->inStart;
1859 [ + + ]: 3150558 : if (pqGetc(&id, conn))
6800 1860 : 289560 : return 0;
8468 1861 [ + + ]: 2860998 : if (pqGetInt(&msgLength, 4, conn))
6800 1862 : 786 : return 0;
1863 [ - + ]: 2860212 : if (msgLength < 4)
1864 : : {
6800 tgl@sss.pgh.pa.us 1865 :UBC 0 : handleSyncLoss(conn, id, msgLength);
1866 : 0 : return -2;
1867 : : }
8468 tgl@sss.pgh.pa.us 1868 :CBC 2860212 : avail = conn->inEnd - conn->inCursor;
1869 [ + + ]: 2860212 : if (avail < msgLength - 4)
1870 : : {
1871 : : /*
1872 : : * Before returning, enlarge the input buffer if needed to hold
1873 : : * the whole message. See notes in parseInput.
1874 : : */
6664 1875 [ - + ]: 272376 : if (pqCheckInBufferSpace(conn->inCursor + (size_t) msgLength - 4,
1876 : : conn))
1877 : : {
1878 : : /*
1879 : : * Abandon the connection. There's not much else we can
1880 : : * safely do; we can't just ignore the message or we could
1881 : : * miss important changes to the connection state.
1882 : : * pqCheckInBufferSpace() already reported the error.
1883 : : */
370 heikki.linnakangas@i 1884 :UBC 0 : handleFatalError(conn);
6797 tgl@sss.pgh.pa.us 1885 : 0 : return -2;
1886 : : }
6800 tgl@sss.pgh.pa.us 1887 :CBC 272376 : return 0;
1888 : : }
1889 : :
1890 : : /*
1891 : : * If it's a legitimate async message type, process it. (NOTIFY
1892 : : * messages are not currently possible here, but we handle them for
1893 : : * completeness.) Otherwise, if it's anything except Copy Data,
1894 : : * report end-of-copy.
1895 : : */
1896 [ - + - + : 2587836 : switch (id)
+ + ]
1897 : : {
1101 nathan@postgresql.or 1898 :UBC 0 : case PqMsg_NotificationResponse:
6800 tgl@sss.pgh.pa.us 1899 [ # # ]: 0 : if (getNotify(conn))
1900 : 0 : return 0;
1901 : 0 : break;
1101 nathan@postgresql.or 1902 :CBC 40 : case PqMsg_NoticeResponse:
6800 tgl@sss.pgh.pa.us 1903 [ - + ]: 40 : if (pqGetErrorNotice3(conn, false))
6800 tgl@sss.pgh.pa.us 1904 :UBC 0 : return 0;
6800 tgl@sss.pgh.pa.us 1905 :CBC 40 : break;
1101 nathan@postgresql.or 1906 :UBC 0 : case PqMsg_ParameterStatus:
6800 tgl@sss.pgh.pa.us 1907 [ # # ]: 0 : if (getParameterStatus(conn))
1908 : 0 : return 0;
1909 : 0 : break;
1101 nathan@postgresql.or 1910 :CBC 2582233 : case PqMsg_CopyData:
6800 tgl@sss.pgh.pa.us 1911 : 2582233 : return msgLength;
1101 nathan@postgresql.or 1912 : 5500 : case PqMsg_CopyDone:
1913 : :
1914 : : /*
1915 : : * If this is a CopyDone message, exit COPY_OUT mode and let
1916 : : * caller read status with PQgetResult(). If we're in
1917 : : * COPY_BOTH mode, return to COPY_IN mode.
1918 : : */
4868 rhaas@postgresql.org 1919 [ + + ]: 5500 : if (conn->asyncStatus == PGASYNC_COPY_BOTH)
1920 : 14 : conn->asyncStatus = PGASYNC_COPY_IN;
1921 : : else
1922 : 5486 : conn->asyncStatus = PGASYNC_BUSY;
1923 : 5500 : return -1;
6800 tgl@sss.pgh.pa.us 1924 : 63 : default: /* treat as end of copy */
1925 : :
1926 : : /*
1927 : : * Any other message terminates either COPY_IN or COPY_BOTH
1928 : : * mode.
1929 : : */
4868 rhaas@postgresql.org 1930 : 63 : conn->asyncStatus = PGASYNC_BUSY;
6800 tgl@sss.pgh.pa.us 1931 : 63 : return -1;
1932 : : }
1933 : :
1934 : : /*
1935 : : * An error may have been triggered while processing the message,
1936 : : * report it if it's the case
1937 : : */
23 fujii@postgresql.org 1938 [ - + - - ]: 40 : if (conn->error_result && conn->status == CONNECTION_BAD)
23 fujii@postgresql.org 1939 :UBC 0 : return -2;
1940 : :
1941 : : /* Drop the processed message and loop around for another */
741 alvherre@alvh.no-ip. 1942 :CBC 40 : pqParseDone(conn, conn->inCursor);
1943 : : }
1944 : : }
1945 : :
1946 : : /*
1947 : : * PQgetCopyData - read a row of data from the backend during COPY OUT
1948 : : * or COPY BOTH
1949 : : *
1950 : : * If successful, sets *buffer to point to a malloc'd row of data, and
1951 : : * returns row length (always > 0) as result.
1952 : : * Returns 0 if no row available yet (only possible if async is true),
1953 : : * -1 if end of copy (consult PQgetResult), or -2 if error (consult
1954 : : * PQerrorMessage).
1955 : : */
1956 : : int
6800 tgl@sss.pgh.pa.us 1957 : 2958287 : pqGetCopyData3(PGconn *conn, char **buffer, int async)
1958 : : {
1959 : : int msgLength;
1960 : :
1961 : : for (;;)
1962 : : {
1963 : : /*
1964 : : * Collect the next input message. To make life simpler for async
1965 : : * callers, we keep returning 0 until the next message is fully
1966 : : * available, even if it is not Copy Data.
1967 : : */
1968 : 3150518 : msgLength = getCopyDataMessage(conn);
1969 [ + + ]: 3150518 : if (msgLength < 0)
6286 bruce@momjian.us 1970 : 5563 : return msgLength; /* end-of-copy or error */
6800 tgl@sss.pgh.pa.us 1971 [ + + ]: 3144955 : if (msgLength == 0)
1972 : : {
1973 : : /* Don't block if async read requested */
1974 [ + + ]: 562722 : if (async)
1975 : 370491 : return 0;
1976 : : /* Need to load more data */
3298 peter_e@gmx.net 1977 [ + - - + ]: 384462 : if (pqWait(true, false, conn) ||
6800 tgl@sss.pgh.pa.us 1978 : 192231 : pqReadData(conn) < 0)
6800 tgl@sss.pgh.pa.us 1979 :UBC 0 : return -2;
6800 tgl@sss.pgh.pa.us 1980 :CBC 192231 : continue;
1981 : : }
1982 : :
1983 : : /*
1984 : : * Drop zero-length messages (shouldn't happen anyway). Otherwise
1985 : : * pass the data back to the caller.
1986 : : */
8468 1987 : 2582233 : msgLength -= 4;
1988 [ + - ]: 2582233 : if (msgLength > 0)
1989 : : {
1990 : 2582233 : *buffer = (char *) malloc(msgLength + 1);
1991 [ - + ]: 2582233 : if (*buffer == NULL)
1992 : : {
1381 peter@eisentraut.org 1993 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
8468 tgl@sss.pgh.pa.us 1994 : 0 : return -2;
1995 : : }
8468 tgl@sss.pgh.pa.us 1996 :CBC 2582233 : memcpy(*buffer, &conn->inBuffer[conn->inCursor], msgLength);
3354 1997 : 2582233 : (*buffer)[msgLength] = '\0'; /* Add terminating null */
1998 : :
1999 : : /* Mark message consumed */
741 alvherre@alvh.no-ip. 2000 : 2582233 : pqParseDone(conn, conn->inCursor + msgLength);
2001 : :
8468 tgl@sss.pgh.pa.us 2002 : 2582233 : return msgLength;
2003 : : }
2004 : :
2005 : : /* Empty, so drop it and loop around for another */
741 alvherre@alvh.no-ip. 2006 :UBC 0 : pqParseDone(conn, conn->inCursor);
2007 : : }
2008 : : }
2009 : :
2010 : : /*
2011 : : * PQgetline - gets a newline-terminated string from the backend.
2012 : : *
2013 : : * See fe-exec.c for documentation.
2014 : : */
2015 : : int
8481 tgl@sss.pgh.pa.us 2016 : 0 : pqGetline3(PGconn *conn, char *s, int maxlen)
2017 : : {
2018 : : int status;
2019 : :
4516 bruce@momjian.us 2020 [ # # ]: 0 : if (conn->sock == PGINVALID_SOCKET ||
4871 rhaas@postgresql.org 2021 [ # # ]: 0 : (conn->asyncStatus != PGASYNC_COPY_OUT &&
2022 [ # # ]: 0 : conn->asyncStatus != PGASYNC_COPY_BOTH) ||
8481 tgl@sss.pgh.pa.us 2023 [ # # ]: 0 : conn->copy_is_binary)
2024 : : {
1381 peter@eisentraut.org 2025 : 0 : libpq_append_conn_error(conn, "PQgetline: not doing text COPY OUT");
8481 tgl@sss.pgh.pa.us 2026 : 0 : *s = '\0';
2027 : 0 : return EOF;
2028 : : }
2029 : :
8424 bruce@momjian.us 2030 [ # # ]: 0 : while ((status = PQgetlineAsync(conn, s, maxlen - 1)) == 0)
2031 : : {
2032 : : /* need to load more data */
3298 peter_e@gmx.net 2033 [ # # # # ]: 0 : if (pqWait(true, false, conn) ||
8481 tgl@sss.pgh.pa.us 2034 : 0 : pqReadData(conn) < 0)
2035 : : {
2036 : 0 : *s = '\0';
2037 : 0 : return EOF;
2038 : : }
2039 : : }
2040 : :
2041 [ # # ]: 0 : if (status < 0)
2042 : : {
2043 : : /* End of copy detected; gin up old-style terminator */
2044 : 0 : strcpy(s, "\\.");
2045 : 0 : return 0;
2046 : : }
2047 : :
2048 : : /* Add null terminator, and strip trailing \n if present */
8424 bruce@momjian.us 2049 [ # # ]: 0 : if (s[status - 1] == '\n')
2050 : : {
2051 : 0 : s[status - 1] = '\0';
8481 tgl@sss.pgh.pa.us 2052 : 0 : return 0;
2053 : : }
2054 : : else
2055 : : {
2056 : 0 : s[status] = '\0';
2057 : 0 : return 1;
2058 : : }
2059 : : }
2060 : :
2061 : : /*
2062 : : * PQgetlineAsync - gets a COPY data row without blocking.
2063 : : *
2064 : : * See fe-exec.c for documentation.
2065 : : */
2066 : : int
2067 : 0 : pqGetlineAsync3(PGconn *conn, char *buffer, int bufsize)
2068 : : {
2069 : : int msgLength;
2070 : : int avail;
2071 : :
4871 rhaas@postgresql.org 2072 [ # # ]: 0 : if (conn->asyncStatus != PGASYNC_COPY_OUT
2073 [ # # ]: 0 : && conn->asyncStatus != PGASYNC_COPY_BOTH)
8481 tgl@sss.pgh.pa.us 2074 : 0 : return -1; /* we are not doing a copy... */
2075 : :
2076 : : /*
2077 : : * Recognize the next input message. To make life simpler for async
2078 : : * callers, we keep returning 0 until the next message is fully available
2079 : : * even if it is not Copy Data. This should keep PQendcopy from blocking.
2080 : : * (Note: unlike pqGetCopyData3, we do not change asyncStatus here.)
2081 : : */
6800 2082 : 0 : msgLength = getCopyDataMessage(conn);
2083 [ # # ]: 0 : if (msgLength < 0)
2084 : 0 : return -1; /* end-of-copy or error */
2085 [ # # ]: 0 : if (msgLength == 0)
2086 : 0 : return 0; /* no data yet */
2087 : :
2088 : : /*
2089 : : * Move data from libpq's buffer to the caller's. In the case where a
2090 : : * prior call found the caller's buffer too small, we use
2091 : : * conn->copy_already_done to remember how much of the row was already
2092 : : * returned to the caller.
2093 : : */
8481 2094 : 0 : conn->inCursor += conn->copy_already_done;
2095 : 0 : avail = msgLength - 4 - conn->copy_already_done;
2096 [ # # ]: 0 : if (avail <= bufsize)
2097 : : {
2098 : : /* Able to consume the whole message */
2099 : 0 : memcpy(buffer, &conn->inBuffer[conn->inCursor], avail);
2100 : : /* Mark message consumed */
2101 : 0 : conn->inStart = conn->inCursor + avail;
2102 : : /* Reset state for next time */
2103 : 0 : conn->copy_already_done = 0;
2104 : 0 : return avail;
2105 : : }
2106 : : else
2107 : : {
2108 : : /* We must return a partial message */
2109 : 0 : memcpy(buffer, &conn->inBuffer[conn->inCursor], bufsize);
2110 : : /* The message is NOT consumed from libpq's buffer */
2111 : 0 : conn->copy_already_done += bufsize;
2112 : 0 : return bufsize;
2113 : : }
2114 : : }
2115 : :
2116 : : /*
2117 : : * PQendcopy
2118 : : *
2119 : : * See fe-exec.c for documentation.
2120 : : */
2121 : : int
8481 tgl@sss.pgh.pa.us 2122 :CBC 206 : pqEndcopy3(PGconn *conn)
2123 : : {
2124 : : PGresult *result;
2125 : :
2126 [ + + ]: 206 : if (conn->asyncStatus != PGASYNC_COPY_IN &&
4871 rhaas@postgresql.org 2127 [ - + ]: 199 : conn->asyncStatus != PGASYNC_COPY_OUT &&
4871 rhaas@postgresql.org 2128 [ # # ]:UBC 0 : conn->asyncStatus != PGASYNC_COPY_BOTH)
2129 : : {
1381 peter@eisentraut.org 2130 : 0 : libpq_append_conn_error(conn, "no COPY in progress");
8481 tgl@sss.pgh.pa.us 2131 : 0 : return 1;
2132 : : }
2133 : :
2134 : : /* Send the CopyDone message if needed */
4871 rhaas@postgresql.org 2135 [ + + ]:CBC 206 : if (conn->asyncStatus == PGASYNC_COPY_IN ||
2136 [ - + ]: 199 : conn->asyncStatus == PGASYNC_COPY_BOTH)
2137 : : {
1101 nathan@postgresql.or 2138 [ + - - + ]: 14 : if (pqPutMsgStart(PqMsg_CopyDone, conn) < 0 ||
8481 tgl@sss.pgh.pa.us 2139 : 7 : pqPutMsgEnd(conn) < 0)
8481 tgl@sss.pgh.pa.us 2140 :UBC 0 : return 1;
2141 : :
2142 : : /*
2143 : : * If we sent the COPY command in extended-query mode, we must issue a
2144 : : * Sync as well.
2145 : : */
1991 alvherre@alvh.no-ip. 2146 [ + - ]:CBC 7 : if (conn->cmd_queue_head &&
2147 [ - + ]: 7 : conn->cmd_queue_head->queryclass != PGQUERY_SIMPLE)
2148 : : {
1101 nathan@postgresql.or 2149 [ # # # # ]:UBC 0 : if (pqPutMsgStart(PqMsg_Sync, conn) < 0 ||
8415 tgl@sss.pgh.pa.us 2150 : 0 : pqPutMsgEnd(conn) < 0)
2151 : 0 : return 1;
2152 : : }
2153 : : }
2154 : :
2155 : : /*
2156 : : * make sure no data is waiting to be sent, abort if we are non-blocking
2157 : : * and the flush fails
2158 : : */
8481 tgl@sss.pgh.pa.us 2159 [ - + - - ]:CBC 206 : if (pqFlush(conn) && pqIsnonblocking(conn))
7533 neilc@samurai.com 2160 :UBC 0 : return 1;
2161 : :
2162 : : /* Return to active duty */
8481 tgl@sss.pgh.pa.us 2163 :CBC 206 : conn->asyncStatus = PGASYNC_BUSY;
2164 : :
2165 : : /*
2166 : : * Non blocking connections may have to abort at this point. If everyone
2167 : : * played the game there should be no problem, but in error scenarios the
2168 : : * expected messages may not have arrived yet. (We are assuming that the
2169 : : * backend's packetizing will ensure that CommandComplete arrives along
2170 : : * with the CopyDone; are there corner cases where that doesn't happen?)
2171 : : */
2172 [ - + - - ]: 206 : if (pqIsnonblocking(conn) && PQisBusy(conn))
7533 neilc@samurai.com 2173 :UBC 0 : return 1;
2174 : :
2175 : : /* Wait for the completion response */
8481 tgl@sss.pgh.pa.us 2176 :CBC 206 : result = PQgetResult(conn);
2177 : :
2178 : : /* Expecting a successful result */
2179 [ + - + - ]: 206 : if (result && result->resultStatus == PGRES_COMMAND_OK)
2180 : : {
2181 : 206 : PQclear(result);
2182 : 206 : return 0;
2183 : : }
2184 : :
2185 : : /*
2186 : : * Trouble. For backwards-compatibility reasons, we issue the error
2187 : : * message as if it were a notice (would be nice to get rid of this
2188 : : * silliness, but too many apps probably don't handle errors from
2189 : : * PQendcopy reasonably). Note that the app can still obtain the error
2190 : : * status from the PGconn object.
2191 : : */
8481 tgl@sss.pgh.pa.us 2192 [ # # ]:UBC 0 : if (conn->errorMessage.len > 0)
2193 : : {
2194 : : /* We have to strip the trailing newline ... pain in neck... */
8424 bruce@momjian.us 2195 : 0 : char svLast = conn->errorMessage.data[conn->errorMessage.len - 1];
2196 : :
8468 tgl@sss.pgh.pa.us 2197 [ # # ]: 0 : if (svLast == '\n')
8424 bruce@momjian.us 2198 : 0 : conn->errorMessage.data[conn->errorMessage.len - 1] = '\0';
8466 tgl@sss.pgh.pa.us 2199 : 0 : pqInternalNotice(&conn->noticeHooks, "%s", conn->errorMessage.data);
8424 bruce@momjian.us 2200 : 0 : conn->errorMessage.data[conn->errorMessage.len - 1] = svLast;
2201 : : }
2202 : :
8481 tgl@sss.pgh.pa.us 2203 : 0 : PQclear(result);
2204 : :
2205 : 0 : return 1;
2206 : : }
2207 : :
2208 : :
2209 : : /*
2210 : : * PQnfn - Send a function call to the POSTGRES backend.
2211 : : *
2212 : : * See fe-exec.c for documentation.
2213 : : */
2214 : : PGresult *
8481 tgl@sss.pgh.pa.us 2215 :CBC 1343 : pqFunctionCall3(PGconn *conn, Oid fnid,
2216 : : int *result_buf, int buf_size, int *actual_result_len,
2217 : : int result_is_int,
2218 : : const PQArgBlock *args, int nargs)
2219 : : {
2220 : 1343 : bool needInput = false;
2221 : 1343 : ExecStatusType status = PGRES_FATAL_ERROR;
2222 : : char id;
2223 : : int msgLength;
2224 : : int avail;
2225 : : int i;
2226 : :
2227 : : /* already validated by PQnfn */
1991 alvherre@alvh.no-ip. 2228 [ - + ]: 1343 : Assert(conn->pipelineStatus == PQ_PIPELINE_OFF);
2229 : :
2230 : : /* PQnfn already validated connection state */
2231 : :
1101 nathan@postgresql.or 2232 [ + - + - ]: 2686 : if (pqPutMsgStart(PqMsg_FunctionCall, conn) < 0 ||
8424 bruce@momjian.us 2233 [ + - ]: 2686 : pqPutInt(fnid, 4, conn) < 0 || /* function id */
3354 tgl@sss.pgh.pa.us 2234 [ + - ]: 2686 : pqPutInt(1, 2, conn) < 0 || /* # of format codes */
2235 [ - + ]: 2686 : pqPutInt(1, 2, conn) < 0 || /* format code: BINARY */
8424 bruce@momjian.us 2236 : 1343 : pqPutInt(nargs, 2, conn) < 0) /* # of args */
2237 : : {
2238 : : /* error message should be set up already */
8481 tgl@sss.pgh.pa.us 2239 :UBC 0 : return NULL;
2240 : : }
2241 : :
8481 tgl@sss.pgh.pa.us 2242 [ + + ]:CBC 3906 : for (i = 0; i < nargs; ++i)
2243 : : { /* len.int4 + contents */
2244 [ - + ]: 2563 : if (pqPutInt(args[i].len, 4, conn))
8481 tgl@sss.pgh.pa.us 2245 :UBC 0 : return NULL;
8481 tgl@sss.pgh.pa.us 2246 [ - + ]:CBC 2563 : if (args[i].len == -1)
8481 tgl@sss.pgh.pa.us 2247 :UBC 0 : continue; /* it's NULL */
2248 : :
8481 tgl@sss.pgh.pa.us 2249 [ + + ]:CBC 2563 : if (args[i].isint)
2250 : : {
2251 [ - + ]: 1906 : if (pqPutInt(args[i].u.integer, args[i].len, conn))
8481 tgl@sss.pgh.pa.us 2252 :UBC 0 : return NULL;
2253 : : }
2254 : : else
2255 : : {
476 heikki.linnakangas@i 2256 [ - + ]:CBC 657 : if (pqPutnchar(args[i].u.ptr, args[i].len, conn))
8481 tgl@sss.pgh.pa.us 2257 :UBC 0 : return NULL;
2258 : : }
2259 : : }
2260 : :
3354 tgl@sss.pgh.pa.us 2261 [ - + ]:CBC 1343 : if (pqPutInt(1, 2, conn) < 0) /* result format code: BINARY */
8481 tgl@sss.pgh.pa.us 2262 :UBC 0 : return NULL;
2263 : :
8481 tgl@sss.pgh.pa.us 2264 [ + - + - ]:CBC 2686 : if (pqPutMsgEnd(conn) < 0 ||
2265 : 1343 : pqFlush(conn))
8481 tgl@sss.pgh.pa.us 2266 :UBC 0 : return NULL;
2267 : :
2268 : : for (;;)
2269 : : {
8481 tgl@sss.pgh.pa.us 2270 [ + + ]:CBC 4229 : if (needInput)
2271 : : {
2272 : : /* Wait for some data to arrive (or for the channel to close) */
3298 peter_e@gmx.net 2273 [ + - + - ]: 3086 : if (pqWait(true, false, conn) ||
8481 tgl@sss.pgh.pa.us 2274 : 1543 : pqReadData(conn) < 0)
2275 : : break;
2276 : : }
2277 : :
2278 : : /*
2279 : : * Scan the message. If we run out of data, loop around to try again.
2280 : : */
2281 : 4229 : needInput = true;
2282 : :
2283 : 4229 : conn->inCursor = conn->inStart;
2284 [ + + ]: 4229 : if (pqGetc(&id, conn))
2285 : 1343 : continue;
2286 [ - + ]: 2886 : if (pqGetInt(&msgLength, 4, conn))
8481 tgl@sss.pgh.pa.us 2287 :UBC 0 : continue;
2288 : :
2289 : : /*
2290 : : * Try to validate message type/length here. A length less than 4 is
2291 : : * definitely broken. Large lengths should only be believed for a few
2292 : : * message types.
2293 : : */
8481 tgl@sss.pgh.pa.us 2294 [ - + ]:CBC 2886 : if (msgLength < 4)
2295 : : {
8481 tgl@sss.pgh.pa.us 2296 :UBC 0 : handleSyncLoss(conn, id, msgLength);
2297 : 0 : break;
2298 : : }
8278 tgl@sss.pgh.pa.us 2299 [ - + - - :CBC 2886 : if (msgLength > 30000 && !VALID_LONG_MESSAGE_TYPE(id))
- - - - -
- - - - -
- - - - ]
2300 : : {
8481 tgl@sss.pgh.pa.us 2301 :UBC 0 : handleSyncLoss(conn, id, msgLength);
2302 : 0 : break;
2303 : : }
2304 : :
2305 : : /*
2306 : : * Can't process if message body isn't all here yet.
2307 : : */
8481 tgl@sss.pgh.pa.us 2308 :CBC 2886 : msgLength -= 4;
2309 : 2886 : avail = conn->inEnd - conn->inCursor;
2310 [ + + ]: 2886 : if (avail < msgLength)
2311 : : {
2312 : : /*
2313 : : * Before looping, enlarge the input buffer if needed to hold the
2314 : : * whole message. See notes in parseInput.
2315 : : */
6664 2316 [ - + ]: 200 : if (pqCheckInBufferSpace(conn->inCursor + (size_t) msgLength,
2317 : : conn))
2318 : : {
2319 : : /*
2320 : : * Abandon the connection. There's not much else we can
2321 : : * safely do; we can't just ignore the message or we could
2322 : : * miss important changes to the connection state.
2323 : : * pqCheckInBufferSpace() already reported the error.
2324 : : */
370 heikki.linnakangas@i 2325 :UBC 0 : handleFatalError(conn);
8481 tgl@sss.pgh.pa.us 2326 : 0 : break;
2327 : : }
8481 tgl@sss.pgh.pa.us 2328 :CBC 200 : continue;
2329 : : }
2330 : :
2331 : : /*
2332 : : * We should see V or E response to the command, but might get N
2333 : : * and/or A notices first. We also need to swallow the final Z before
2334 : : * returning.
2335 : : */
2336 [ + - - - : 2686 : switch (id)
+ - - ]
2337 : : {
367 nathan@postgresql.or 2338 : 1343 : case PqMsg_FunctionCallResponse:
8481 tgl@sss.pgh.pa.us 2339 [ - + ]: 1343 : if (pqGetInt(actual_result_len, 4, conn))
8481 tgl@sss.pgh.pa.us 2340 :UBC 0 : continue;
8481 tgl@sss.pgh.pa.us 2341 [ + - ]:CBC 1343 : if (*actual_result_len != -1)
2342 : : {
2343 [ + + ]: 1343 : if (result_is_int)
2344 : : {
2345 [ - + ]: 878 : if (pqGetInt(result_buf, *actual_result_len, conn))
8481 tgl@sss.pgh.pa.us 2346 :UBC 0 : continue;
2347 : : }
2348 : : else
2349 : : {
2350 : : /*
2351 : : * If the server returned too much data for the
2352 : : * buffer, something fishy is going on. Abandon ship.
2353 : : */
108 nathan@postgresql.or 2354 [ + - - + ]:CBC 465 : if (buf_size != -1 && *actual_result_len > buf_size)
2355 : : {
108 nathan@postgresql.or 2356 :UBC 0 : libpq_append_conn_error(conn, "server returned too much data");
2357 : 0 : handleFatalError(conn);
2358 : 0 : return pqPrepareAsyncResult(conn);
2359 : : }
2360 : :
476 heikki.linnakangas@i 2361 [ - + ]:CBC 465 : if (pqGetnchar(result_buf,
8481 tgl@sss.pgh.pa.us 2362 : 465 : *actual_result_len,
2363 : : conn))
8481 tgl@sss.pgh.pa.us 2364 :UBC 0 : continue;
2365 : : }
2366 : : }
2367 : : /* correctly finished function result message */
8481 tgl@sss.pgh.pa.us 2368 :CBC 1343 : status = PGRES_COMMAND_OK;
2369 : 1343 : break;
367 nathan@postgresql.or 2370 :UBC 0 : case PqMsg_ErrorResponse:
8481 tgl@sss.pgh.pa.us 2371 [ # # ]: 0 : if (pqGetErrorNotice3(conn, true))
2372 : 0 : continue;
2373 : 0 : status = PGRES_FATAL_ERROR;
2374 : 0 : break;
367 nathan@postgresql.or 2375 : 0 : case PqMsg_NotificationResponse:
2376 : : /* handle notify and go back to processing return values */
8481 tgl@sss.pgh.pa.us 2377 [ # # ]: 0 : if (getNotify(conn))
2378 : 0 : continue;
2379 : 0 : break;
367 nathan@postgresql.or 2380 : 0 : case PqMsg_NoticeResponse:
2381 : : /* handle notice and go back to processing return values */
8481 tgl@sss.pgh.pa.us 2382 [ # # ]: 0 : if (pqGetErrorNotice3(conn, false))
2383 : 0 : continue;
2384 : 0 : break;
367 nathan@postgresql.or 2385 :CBC 1343 : case PqMsg_ReadyForQuery:
8468 tgl@sss.pgh.pa.us 2386 [ - + ]: 1343 : if (getReadyForQuery(conn))
8481 tgl@sss.pgh.pa.us 2387 :UBC 0 : continue;
2388 : :
2389 : : /* consume the message */
741 alvherre@alvh.no-ip. 2390 :CBC 1343 : pqParseDone(conn, conn->inStart + 5 + msgLength);
2391 : :
2392 : : /*
2393 : : * If we already have a result object (probably an error), use
2394 : : * that. Otherwise, if we saw a function result message,
2395 : : * report COMMAND_OK. Otherwise, the backend violated the
2396 : : * protocol, so complain.
2397 : : */
1589 tgl@sss.pgh.pa.us 2398 [ + - + - ]: 1343 : if (!pgHavePendingResult(conn))
2399 : : {
1651 2400 [ + - ]: 1343 : if (status == PGRES_COMMAND_OK)
2401 : : {
2402 : 1343 : conn->result = PQmakeEmptyPGresult(conn, status);
2403 [ - + ]: 1343 : if (!conn->result)
2404 : : {
1381 peter@eisentraut.org 2405 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
1651 tgl@sss.pgh.pa.us 2406 : 0 : pqSaveErrorResult(conn);
2407 : : }
2408 : : }
2409 : : else
2410 : : {
1381 peter@eisentraut.org 2411 : 0 : libpq_append_conn_error(conn, "protocol error: no function result");
1651 tgl@sss.pgh.pa.us 2412 : 0 : pqSaveErrorResult(conn);
2413 : : }
2414 : : }
2415 : : /* and we're out */
1651 tgl@sss.pgh.pa.us 2416 :CBC 1343 : return pqPrepareAsyncResult(conn);
367 nathan@postgresql.or 2417 :UBC 0 : case PqMsg_ParameterStatus:
8481 tgl@sss.pgh.pa.us 2418 [ # # ]: 0 : if (getParameterStatus(conn))
2419 : 0 : continue;
2420 : 0 : break;
2421 : 0 : default:
2422 : : /* The backend violates the protocol. */
1381 peter@eisentraut.org 2423 : 0 : libpq_append_conn_error(conn, "protocol error: id=0x%x", id);
8481 tgl@sss.pgh.pa.us 2424 : 0 : pqSaveErrorResult(conn);
2425 : :
2426 : : /*
2427 : : * We can't call parsing done due to the protocol violation
2428 : : * (so message tracing wouldn't work), but trust the specified
2429 : : * message length as what to skip.
2430 : : */
2431 : 0 : conn->inStart += 5 + msgLength;
2432 : 0 : return pqPrepareAsyncResult(conn);
2433 : : }
2434 : :
2435 : : /*
2436 : : * An error may have been triggered while processing the message, bail
2437 : : * out
2438 : : */
23 fujii@postgresql.org 2439 [ - + - - ]:CBC 1343 : if (conn->error_result && conn->status == CONNECTION_BAD)
23 fujii@postgresql.org 2440 :UBC 0 : return pqPrepareAsyncResult(conn);
2441 : :
2442 : : /* Completed parsing this message, keep going */
741 alvherre@alvh.no-ip. 2443 :CBC 1343 : pqParseDone(conn, conn->inStart + 5 + msgLength);
8481 tgl@sss.pgh.pa.us 2444 : 1343 : needInput = false;
2445 : : }
2446 : :
2447 : : /*
2448 : : * We fall out of the loop only upon failing to read data.
2449 : : * conn->errorMessage has been set by pqWait or pqReadData. We want to
2450 : : * append it to any already-received error message.
2451 : : */
8481 tgl@sss.pgh.pa.us 2452 :UBC 0 : pqSaveErrorResult(conn);
2453 : 0 : return pqPrepareAsyncResult(conn);
2454 : : }
2455 : :
2456 : :
2457 : : /*
2458 : : * Construct startup packet
2459 : : *
2460 : : * Returns a malloc'd packet buffer, or NULL if out of memory
2461 : : */
2462 : : char *
8481 tgl@sss.pgh.pa.us 2463 :CBC 15843 : pqBuildStartupPacket3(PGconn *conn, int *packetlen,
2464 : : const PQEnvironmentOption *options)
2465 : : {
2466 : : char *startpacket;
2467 : : size_t len;
2468 : :
290 jchampion@postgresql 2469 : 15843 : len = build_startup_packet(conn, NULL, options);
2470 [ + - - + ]: 15843 : if (len == 0 || len > INT_MAX)
290 jchampion@postgresql 2471 :UBC 0 : return NULL;
2472 : :
290 jchampion@postgresql 2473 :CBC 15843 : *packetlen = len;
8481 tgl@sss.pgh.pa.us 2474 : 15843 : startpacket = (char *) malloc(*packetlen);
2475 [ - + ]: 15843 : if (!startpacket)
8481 tgl@sss.pgh.pa.us 2476 :UBC 0 : return NULL;
2477 : :
290 jchampion@postgresql 2478 :CBC 15843 : len = build_startup_packet(conn, startpacket, options);
2479 [ - + ]: 15843 : Assert(*packetlen == len);
2480 : :
8481 tgl@sss.pgh.pa.us 2481 : 15843 : return startpacket;
2482 : : }
2483 : :
2484 : : /*
2485 : : * Build a startup packet given a filled-in PGconn structure.
2486 : : *
2487 : : * We need to figure out how much space is needed, then fill it in.
2488 : : * To avoid duplicate logic, this routine is called twice: the first time
2489 : : * (with packet == NULL) just counts the space needed, the second time
2490 : : * (with packet == allocated space) fills it in. Return value is the number
2491 : : * of bytes used, or zero in the unlikely event of size_t overflow.
2492 : : */
2493 : : static size_t
2494 : 31686 : build_startup_packet(const PGconn *conn, char *packet,
2495 : : const PQEnvironmentOption *options)
2496 : : {
290 jchampion@postgresql 2497 : 31686 : size_t packet_len = 0;
2498 : : const PQEnvironmentOption *next_eo;
2499 : : const char *val;
2500 : :
2501 : : /* Protocol version comes first. */
8481 tgl@sss.pgh.pa.us 2502 [ + + ]: 31686 : if (packet)
2503 : : {
3252 andres@anarazel.de 2504 : 15843 : ProtocolVersion pv = pg_hton32(conn->pversion);
2505 : :
8481 tgl@sss.pgh.pa.us 2506 : 15843 : memcpy(packet + packet_len, &pv, sizeof(ProtocolVersion));
2507 : : }
2508 : 31686 : packet_len += sizeof(ProtocolVersion);
2509 : :
2510 : : /* Add user name, database name, options */
2511 : :
2512 : : #define ADD_STARTUP_OPTION(optname, optval) \
2513 : : do { \
2514 : : if (packet) \
2515 : : strcpy(packet + packet_len, optname); \
2516 : : if (pg_add_size_overflow(packet_len, strlen(optname) + 1, &packet_len)) \
2517 : : return 0; \
2518 : : if (packet) \
2519 : : strcpy(packet + packet_len, optval); \
2520 : : if (pg_add_size_overflow(packet_len, strlen(optval) + 1, &packet_len)) \
2521 : : return 0; \
2522 : : } while(0)
2523 : :
2524 [ + - + - ]: 31686 : if (conn->pguser && conn->pguser[0])
6112 2525 [ + + - + : 31686 : ADD_STARTUP_OPTION("user", conn->pguser);
+ + - + ]
8481 2526 [ + - + - ]: 31686 : if (conn->dbName && conn->dbName[0])
6112 2527 [ + + - + : 31686 : ADD_STARTUP_OPTION("database", conn->dbName);
+ + - + ]
5965 magnus@hagander.net 2528 [ + + + + ]: 31686 : if (conn->replication && conn->replication[0])
6068 heikki.linnakangas@i 2529 [ + + - + : 3518 : ADD_STARTUP_OPTION("replication", conn->replication);
+ + - + ]
8481 tgl@sss.pgh.pa.us 2530 [ + - + + ]: 31686 : if (conn->pgoptions && conn->pgoptions[0])
6112 2531 [ + + - + : 9012 : ADD_STARTUP_OPTION("options", conn->pgoptions);
+ + - + ]
2532 [ + - ]: 31686 : if (conn->send_appname)
2533 : : {
2534 : : /* Use appname if present, otherwise use fallback */
2535 [ + + ]: 31686 : val = conn->appname ? conn->appname : conn->fbappname;
2536 [ + + + - ]: 31686 : if (val && val[0])
2537 [ + + - + : 31678 : ADD_STARTUP_OPTION("application_name", val);
+ + - + ]
2538 : : }
2539 : :
5668 peter_e@gmx.net 2540 [ + + + - ]: 31686 : if (conn->client_encoding_initial && conn->client_encoding_initial[0])
2541 [ + + - + : 1884 : ADD_STARTUP_OPTION("client_encoding", conn->client_encoding_initial);
+ + - + ]
2542 : :
2543 : : /*
2544 : : * Add the test_protocol_negotiation option when greasing, to test that
2545 : : * servers properly report unsupported protocol options in addition to
2546 : : * unsupported minor versions.
2547 : : */
185 jchampion@postgresql 2548 [ + + ]: 31686 : if (conn->pversion == PG_PROTOCOL_GREASE)
2549 [ + + - + : 31644 : ADD_STARTUP_OPTION("_pq_.test_protocol_negotiation", "");
+ + - + ]
2550 : :
2551 : : /* Add any environment-driven GUC settings needed */
8481 tgl@sss.pgh.pa.us 2552 [ + + ]: 126744 : for (next_eo = options; next_eo->envName; next_eo++)
2553 : : {
2554 [ + + ]: 95058 : if ((val = getenv(next_eo->envName)) != NULL)
2555 : : {
8147 2556 [ + - ]: 10884 : if (pg_strcasecmp(val, "default") != 0)
6112 2557 [ + + - + : 10884 : ADD_STARTUP_OPTION(next_eo->pgName, val);
+ + - + ]
2558 : : }
2559 : : }
2560 : :
2561 : : /* Add trailing terminator */
8481 2562 [ + + ]: 31686 : if (packet)
2563 : 15843 : packet[packet_len] = '\0';
276 jchampion@postgresql 2564 [ - + ]: 31686 : if (pg_add_size_overflow(packet_len, 1, &packet_len))
290 jchampion@postgresql 2565 :UBC 0 : return 0;
2566 : :
8481 tgl@sss.pgh.pa.us 2567 :CBC 31686 : return packet_len;
2568 : : }
|
