Age Owner Branch data TLA Line data Source code
1 : : /*
2 : : * psql - the PostgreSQL interactive terminal
3 : : *
4 : : * Copyright (c) 2000-2024, PostgreSQL Global Development Group
5 : : *
6 : : * src/bin/psql/common.c
7 : : */
8 : : #include "postgres_fe.h"
9 : :
10 : : #include <ctype.h>
11 : : #include <limits.h>
12 : : #include <math.h>
13 : : #include <pwd.h>
14 : : #include <signal.h>
15 : : #ifndef WIN32
16 : : #include <unistd.h> /* for write() */
17 : : #else
18 : : #include <io.h> /* for _write() */
19 : : #include <win32.h>
20 : : #endif
21 : :
22 : : #include "command.h"
23 : : #include "common.h"
24 : : #include "common/logging.h"
25 : : #include "copy.h"
26 : : #include "crosstabview.h"
27 : : #include "fe_utils/cancel.h"
28 : : #include "fe_utils/mbprint.h"
29 : : #include "fe_utils/string_utils.h"
30 : : #include "portability/instr_time.h"
31 : : #include "settings.h"
32 : :
33 : : static bool DescribeQuery(const char *query, double *elapsed_msec);
34 : : static int ExecQueryAndProcessResults(const char *query,
35 : : double *elapsed_msec,
36 : : bool *svpt_gone_p,
37 : : bool is_watch,
38 : : int min_rows,
39 : : const printQueryOpt *opt,
40 : : FILE *printQueryFout);
41 : : static bool command_no_begin(const char *query);
42 : :
43 : :
44 : : /*
45 : : * openQueryOutputFile --- attempt to open a query output file
46 : : *
47 : : * fname == NULL selects stdout, else an initial '|' selects a pipe,
48 : : * else plain file.
49 : : *
50 : : * Returns output file pointer into *fout, and is-a-pipe flag into *is_pipe.
51 : : * Caller is responsible for adjusting SIGPIPE state if it's a pipe.
52 : : *
53 : : * On error, reports suitable error message and returns false.
54 : : */
55 : : bool
3055 tgl@sss.pgh.pa.us 56 :CBC 7974 : openQueryOutputFile(const char *fname, FILE **fout, bool *is_pipe)
57 : : {
8928 bruce@momjian.us 58 [ + + - + ]: 7974 : if (!fname || fname[0] == '\0')
59 : : {
3055 tgl@sss.pgh.pa.us 60 : 7952 : *fout = stdout;
61 : 7952 : *is_pipe = false;
62 : : }
8928 bruce@momjian.us 63 [ + + ]: 22 : else if (*fname == '|')
64 : : {
594 tgl@sss.pgh.pa.us 65 :GBC 4 : fflush(NULL);
3055 66 : 4 : *fout = popen(fname + 1, "w");
67 : 4 : *is_pipe = true;
68 : : }
69 : : else
70 : : {
3055 tgl@sss.pgh.pa.us 71 :CBC 18 : *fout = fopen(fname, "w");
72 : 18 : *is_pipe = false;
73 : : }
74 : :
75 [ - + ]: 7974 : if (*fout == NULL)
76 : : {
1840 peter@eisentraut.org 77 :UBC 0 : pg_log_error("%s: %m", fname);
3055 tgl@sss.pgh.pa.us 78 : 0 : return false;
79 : : }
80 : :
3055 tgl@sss.pgh.pa.us 81 :CBC 7974 : return true;
82 : : }
83 : :
84 : : /*
85 : : * Check if an output stream for \g needs to be opened, and if yes,
86 : : * open it and update the caller's gfile_fout and is_pipe state variables.
87 : : * Return true if OK, false if an error occurred.
88 : : */
89 : : static bool
8 tgl@sss.pgh.pa.us 90 :GNC 61839 : SetupGOutput(FILE **gfile_fout, bool *is_pipe)
91 : : {
92 : : /* If there is a \g file or program, and it's not already open, open it */
93 [ + + + + ]: 61839 : if (pset.gfname != NULL && *gfile_fout == NULL)
94 : : {
95 [ + - ]: 16 : if (openQueryOutputFile(pset.gfname, gfile_fout, is_pipe))
96 : : {
97 [ + + ]: 16 : if (*is_pipe)
98 : 4 : disable_sigpipe_trap();
99 : : }
100 : : else
8 tgl@sss.pgh.pa.us 101 :UNC 0 : return false;
102 : : }
8 tgl@sss.pgh.pa.us 103 :GNC 61839 : return true;
104 : : }
105 : :
106 : : /*
107 : : * Close the output stream for \g, if we opened it.
108 : : */
109 : : static void
110 : 159982 : CloseGOutput(FILE *gfile_fout, bool is_pipe)
111 : : {
112 [ + + ]: 159982 : if (gfile_fout)
113 : : {
114 [ + + ]: 16 : if (is_pipe)
115 : : {
116 : 4 : SetShellResultVariables(pclose(gfile_fout));
117 : 4 : restore_sigpipe_trap();
118 : : }
119 : : else
120 : 12 : fclose(gfile_fout);
121 : : }
122 : 159982 : }
123 : :
124 : : /*
125 : : * setQFout
126 : : * -- handler for -o command line option and \o command
127 : : *
128 : : * On success, updates pset with the new output file and returns true.
129 : : * On failure, returns false without changing pset state.
130 : : */
131 : : bool
3055 tgl@sss.pgh.pa.us 132 :CBC 7958 : setQFout(const char *fname)
133 : : {
134 : : FILE *fout;
135 : : bool is_pipe;
136 : :
137 : : /* First make sure we can open the new output file/pipe */
138 [ - + ]: 7958 : if (!openQueryOutputFile(fname, &fout, &is_pipe))
3055 tgl@sss.pgh.pa.us 139 :UBC 0 : return false;
140 : :
141 : : /* Close old file/pipe */
3055 tgl@sss.pgh.pa.us 142 [ + - + + :CBC 7958 : if (pset.queryFout && pset.queryFout != stdout && pset.queryFout != stderr)
+ - ]
143 : : {
144 [ - + ]: 6 : if (pset.queryFoutPipe)
374 tgl@sss.pgh.pa.us 145 :UBC 0 : SetShellResultVariables(pclose(pset.queryFout));
146 : : else
3055 tgl@sss.pgh.pa.us 147 :CBC 6 : fclose(pset.queryFout);
148 : : }
149 : :
150 : 7958 : pset.queryFout = fout;
151 : 7958 : pset.queryFoutPipe = is_pipe;
152 : :
153 : : /* Adjust SIGPIPE handling appropriately: ignore signal if is_pipe */
154 : 7958 : set_sigpipe_trap_state(is_pipe);
155 : 7958 : restore_sigpipe_trap();
156 : :
157 : 7958 : return true;
158 : : }
159 : :
160 : :
161 : : /*
162 : : * Variable-fetching callback for flex lexer
163 : : *
164 : : * If the specified variable exists, return its value as a string (malloc'd
165 : : * and expected to be freed by the caller); else return NULL.
166 : : *
167 : : * If "quote" isn't PQUOTE_PLAIN, then return the value suitably quoted and
168 : : * escaped for the specified quoting requirement. (Failure in escaping
169 : : * should lead to printing an error and returning NULL.)
170 : : *
171 : : * "passthrough" is the pointer previously given to psql_scan_set_passthrough.
172 : : * In psql, passthrough points to a ConditionalStack, which we check to
173 : : * determine whether variable expansion is allowed.
174 : : */
175 : : char *
2570 176 : 1789 : psql_get_variable(const char *varname, PsqlScanQuoteType quote,
177 : : void *passthrough)
178 : : {
179 : 1789 : char *result = NULL;
180 : : const char *value;
181 : :
182 : : /* In an inactive \if branch, suppress all variable substitutions */
2572 183 [ + - + + ]: 1789 : if (passthrough && !conditional_active((ConditionalStack) passthrough))
184 : 36 : return NULL;
185 : :
2949 186 : 1753 : value = GetVariable(pset.vars, varname);
187 [ + + ]: 1753 : if (!value)
188 : 226 : return NULL;
189 : :
2570 190 [ + + - - ]: 1527 : switch (quote)
191 : : {
192 : 1136 : case PQUOTE_PLAIN:
193 : 1136 : result = pg_strdup(value);
194 : 1136 : break;
195 : 391 : case PQUOTE_SQL_LITERAL:
196 : : case PQUOTE_SQL_IDENT:
197 : : {
198 : : /*
199 : : * For these cases, we use libpq's quoting functions, which
200 : : * assume the string is in the connection's client encoding.
201 : : */
202 : : char *escaped_value;
203 : :
204 [ - + ]: 391 : if (!pset.db)
205 : : {
1840 peter@eisentraut.org 206 :UBC 0 : pg_log_error("cannot escape without active connection");
2570 tgl@sss.pgh.pa.us 207 : 0 : return NULL;
208 : : }
209 : :
2570 tgl@sss.pgh.pa.us 210 [ + + ]:CBC 391 : if (quote == PQUOTE_SQL_LITERAL)
211 : : escaped_value =
212 : 378 : PQescapeLiteral(pset.db, value, strlen(value));
213 : : else
214 : : escaped_value =
215 : 13 : PQescapeIdentifier(pset.db, value, strlen(value));
216 : :
217 [ - + ]: 391 : if (escaped_value == NULL)
218 : : {
2570 tgl@sss.pgh.pa.us 219 :UBC 0 : const char *error = PQerrorMessage(pset.db);
220 : :
1840 peter@eisentraut.org 221 : 0 : pg_log_info("%s", error);
2570 tgl@sss.pgh.pa.us 222 : 0 : return NULL;
223 : : }
224 : :
225 : : /*
226 : : * Rather than complicate the lexer's API with a notion of
227 : : * which free() routine to use, just pay the price of an extra
228 : : * strdup().
229 : : */
2570 tgl@sss.pgh.pa.us 230 :CBC 391 : result = pg_strdup(escaped_value);
231 : 391 : PQfreemem(escaped_value);
232 : 391 : break;
233 : : }
2570 tgl@sss.pgh.pa.us 234 :UBC 0 : case PQUOTE_SHELL_ARG:
235 : : {
236 : : /*
237 : : * For this we use appendShellStringNoError, which is
238 : : * encoding-agnostic, which is fine since the shell probably
239 : : * is too. In any case, the only special character is "'",
240 : : * which is not known to appear in valid multibyte characters.
241 : : */
242 : : PQExpBufferData buf;
243 : :
244 : 0 : initPQExpBuffer(&buf);
245 [ # # ]: 0 : if (!appendShellStringNoError(&buf, value))
246 : : {
1840 peter@eisentraut.org 247 : 0 : pg_log_error("shell command argument contains a newline or carriage return: \"%s\"",
248 : : value);
2570 tgl@sss.pgh.pa.us 249 : 0 : free(buf.data);
250 : 0 : return NULL;
251 : : }
252 : 0 : result = buf.data;
253 : 0 : break;
254 : : }
255 : :
256 : : /* No default: we want a compiler warning for missing cases */
257 : : }
258 : :
2949 tgl@sss.pgh.pa.us 259 :CBC 1527 : return result;
260 : : }
261 : :
262 : :
263 : : /*
264 : : * for backend Notice messages (INFO, WARNING, etc)
265 : : */
266 : : void
8768 bruce@momjian.us 267 : 76514 : NoticeProcessor(void *arg, const char *message)
268 : : {
269 : : (void) arg; /* not used */
1840 peter@eisentraut.org 270 : 76514 : pg_log_info("%s", message);
8853 peter_e@gmx.net 271 : 76514 : }
272 : :
273 : :
274 : :
275 : : /*
276 : : * Code to support query cancellation
277 : : *
278 : : * Before we start a query, we enable the SIGINT signal catcher to send a
279 : : * cancel request to the backend.
280 : : *
281 : : * SIGINT is supposed to abort all long-running psql operations, not only
282 : : * database queries. In most places, this is accomplished by checking
283 : : * cancel_pressed during long-running loops. However, that won't work when
284 : : * blocked on user input (in readline() or fgets()). In those places, we
285 : : * set sigint_interrupt_enabled true while blocked, instructing the signal
286 : : * catcher to longjmp through sigint_interrupt_jmp. We assume readline and
287 : : * fgets are coded to handle possible interruption.
288 : : *
289 : : * On Windows, currently this does not work, so control-C is less useful
290 : : * there.
291 : : */
292 : : volatile sig_atomic_t sigint_interrupt_enabled = false;
293 : :
294 : : sigjmp_buf sigint_interrupt_jmp;
295 : :
296 : : static void
1595 michael@paquier.xyz 297 : 1 : psql_cancel_callback(void)
298 : : {
299 : : #ifndef WIN32
300 : : /* if we are waiting for input, longjmp out of it */
6514 tgl@sss.pgh.pa.us 301 [ - + ]: 1 : if (sigint_interrupt_enabled)
302 : : {
6514 tgl@sss.pgh.pa.us 303 :UBC 0 : sigint_interrupt_enabled = false;
304 : 0 : siglongjmp(sigint_interrupt_jmp, 1);
305 : : }
306 : : #endif
307 : :
308 : : /* else, set cancel flag to stop any long-running loops */
1580 michael@paquier.xyz 309 :CBC 1 : cancel_pressed = true;
7106 tgl@sss.pgh.pa.us 310 : 1 : }
311 : :
312 : : void
1595 michael@paquier.xyz 313 : 7959 : psql_setup_cancel_handler(void)
314 : : {
315 : 7959 : setup_cancel_handler(psql_cancel_callback);
7106 tgl@sss.pgh.pa.us 316 : 7959 : }
317 : :
318 : :
319 : : /* ConnectionUp
320 : : *
321 : : * Returns whether our backend connection is still there.
322 : : */
323 : : static bool
7126 neilc@samurai.com 324 : 179226 : ConnectionUp(void)
325 : : {
7696 bruce@momjian.us 326 : 179226 : return PQstatus(pset.db) != CONNECTION_BAD;
327 : : }
328 : :
329 : :
330 : :
331 : : /* CheckConnection
332 : : *
333 : : * Verify that we still have a good connection to the backend, and if not,
334 : : * see if it can be restored.
335 : : *
336 : : * Returns true if either the connection was still there, or it could be
337 : : * restored successfully; false otherwise. If, however, there was no
338 : : * connection and the session is non-interactive, this will exit the program
339 : : * with a code of EXIT_BADCONN.
340 : : */
341 : : static bool
7559 tgl@sss.pgh.pa.us 342 : 179226 : CheckConnection(void)
343 : : {
344 : : bool OK;
345 : :
7696 bruce@momjian.us 346 : 179226 : OK = ConnectionUp();
347 [ + + ]: 179226 : if (!OK)
348 : : {
349 [ + - ]: 13 : if (!pset.cur_cmd_interactive)
350 : : {
737 tgl@sss.pgh.pa.us 351 : 13 : pg_log_error("connection to server was lost");
7696 bruce@momjian.us 352 : 13 : exit(EXIT_BADCONN);
353 : : }
354 : :
1840 peter@eisentraut.org 355 :UBC 0 : fprintf(stderr, _("The connection to the server was lost. Attempting reset: "));
7696 bruce@momjian.us 356 : 0 : PQreset(pset.db);
357 : 0 : OK = ConnectionUp();
358 [ # # ]: 0 : if (!OK)
359 : : {
1840 peter@eisentraut.org 360 : 0 : fprintf(stderr, _("Failed.\n"));
361 : :
362 : : /*
363 : : * Transition to having no connection; but stash away the failed
364 : : * connection so that we can still refer to its parameters in a
365 : : * later \connect attempt. Keep the state cleanup here in sync
366 : : * with do_connect().
367 : : */
1269 tgl@sss.pgh.pa.us 368 [ # # ]: 0 : if (pset.dead_conn)
369 : 0 : PQfinish(pset.dead_conn);
370 : 0 : pset.dead_conn = pset.db;
7696 bruce@momjian.us 371 : 0 : pset.db = NULL;
372 : 0 : ResetCancelConn();
7596 tgl@sss.pgh.pa.us 373 : 0 : UnsyncVariables();
374 : : }
375 : : else
376 : : {
1840 peter@eisentraut.org 377 : 0 : fprintf(stderr, _("Succeeded.\n"));
378 : :
379 : : /*
380 : : * Re-sync, just in case anything changed. Keep this in sync with
381 : : * do_connect().
382 : : */
1686 tgl@sss.pgh.pa.us 383 : 0 : SyncVariables();
384 : 0 : connection_warnings(false); /* Must be after SyncVariables */
385 : : }
386 : : }
387 : :
7696 bruce@momjian.us 388 :CBC 179213 : return OK;
389 : : }
390 : :
391 : :
392 : :
393 : :
394 : : /*
395 : : * AcceptResult
396 : : *
397 : : * Checks whether a result is valid, giving an error message if necessary;
398 : : * and ensures that the connection to the backend is still up.
399 : : *
400 : : * Returns true for valid result, false for error state.
401 : : */
402 : : static bool
741 peter@eisentraut.org 403 : 177208 : AcceptResult(const PGresult *result, bool show_error)
404 : : {
405 : : bool OK;
406 : :
7696 bruce@momjian.us 407 [ - + ]: 177208 : if (!result)
7559 bruce@momjian.us 408 :UBC 0 : OK = false;
409 : : else
7559 bruce@momjian.us 410 [ + + + ]:CBC 177208 : switch (PQresultStatus(result))
411 : : {
412 : 157964 : case PGRES_COMMAND_OK:
413 : : case PGRES_TUPLES_OK:
414 : : case PGRES_TUPLES_CHUNK:
415 : : case PGRES_EMPTY_QUERY:
416 : : case PGRES_COPY_IN:
417 : : case PGRES_COPY_OUT:
418 : : /* Fine, do nothing */
4463 alvherre@alvh.no-ip. 419 : 157964 : OK = true;
420 : 157964 : break;
421 : :
422 : 19243 : case PGRES_BAD_RESPONSE:
423 : : case PGRES_NONFATAL_ERROR:
424 : : case PGRES_FATAL_ERROR:
425 : 19243 : OK = false;
7559 bruce@momjian.us 426 : 19243 : break;
427 : :
428 : 1 : default:
429 : 1 : OK = false;
1840 peter@eisentraut.org 430 : 1 : pg_log_error("unexpected PQresultStatus: %d",
431 : : PQresultStatus(result));
7559 bruce@momjian.us 432 : 1 : break;
433 : : }
434 : :
741 peter@eisentraut.org 435 [ + + + + ]: 177208 : if (!OK && show_error)
436 : : {
6758 bruce@momjian.us 437 : 3 : const char *error = PQerrorMessage(pset.db);
438 : :
439 [ + - ]: 3 : if (strlen(error))
1840 peter@eisentraut.org 440 : 3 : pg_log_info("%s", error);
441 : :
7559 tgl@sss.pgh.pa.us 442 : 3 : CheckConnection();
443 : : }
444 : :
7696 bruce@momjian.us 445 : 177208 : return OK;
446 : : }
447 : :
448 : :
449 : : /*
450 : : * Set special variables from a query result
451 : : * - ERROR: true/false, whether an error occurred on this query
452 : : * - SQLSTATE: code of error, or "00000" if no error, or "" if unknown
453 : : * - ROW_COUNT: how many rows were returned or affected, or "0"
454 : : * - LAST_ERROR_SQLSTATE: same for last error
455 : : * - LAST_ERROR_MESSAGE: message of last error
456 : : *
457 : : * Note: current policy is to apply this only to the results of queries
458 : : * entered by the user, not queries generated by slash commands.
459 : : */
460 : : static void
794 peter@eisentraut.org 461 : 159987 : SetResultVariables(PGresult *result, bool success)
462 : : {
2406 tgl@sss.pgh.pa.us 463 [ + + ]: 159987 : if (success)
464 : : {
794 peter@eisentraut.org 465 : 140611 : const char *ntuples = PQcmdTuples(result);
466 : :
2406 tgl@sss.pgh.pa.us 467 : 140611 : SetVariable(pset.vars, "ERROR", "false");
468 : 140611 : SetVariable(pset.vars, "SQLSTATE", "00000");
469 [ + + ]: 140611 : SetVariable(pset.vars, "ROW_COUNT", *ntuples ? ntuples : "0");
470 : : }
471 : : else
472 : : {
794 peter@eisentraut.org 473 : 19376 : const char *code = PQresultErrorField(result, PG_DIAG_SQLSTATE);
474 : 19376 : const char *mesg = PQresultErrorField(result, PG_DIAG_MESSAGE_PRIMARY);
475 : :
2406 tgl@sss.pgh.pa.us 476 : 19376 : SetVariable(pset.vars, "ERROR", "true");
477 : :
478 : : /*
479 : : * If there is no SQLSTATE code, use an empty string. This can happen
480 : : * for libpq-detected errors (e.g., lost connection, ENOMEM).
481 : : */
482 [ + + ]: 19376 : if (code == NULL)
483 : 50 : code = "";
484 : 19376 : SetVariable(pset.vars, "SQLSTATE", code);
485 : 19376 : SetVariable(pset.vars, "ROW_COUNT", "0");
486 : 19376 : SetVariable(pset.vars, "LAST_ERROR_SQLSTATE", code);
487 [ + + ]: 19376 : SetVariable(pset.vars, "LAST_ERROR_MESSAGE", mesg ? mesg : "");
488 : : }
489 : 159987 : }
490 : :
491 : :
492 : : /*
493 : : * Set special variables from a shell command result
494 : : * - SHELL_ERROR: true/false, whether command returned exit code 0
495 : : * - SHELL_EXIT_CODE: exit code according to shell conventions
496 : : *
497 : : * The argument is a wait status as returned by wait(2) or waitpid(2),
498 : : * which also applies to pclose(3) and system(3).
499 : : */
500 : : void
374 501 : 4 : SetShellResultVariables(int wait_result)
502 : : {
503 : : char buf[32];
504 : :
505 [ + - ]: 4 : SetVariable(pset.vars, "SHELL_ERROR",
506 : : (wait_result == 0) ? "false" : "true");
507 : 4 : snprintf(buf, sizeof(buf), "%d", wait_result_to_exit_code(wait_result));
508 : 4 : SetVariable(pset.vars, "SHELL_EXIT_CODE", buf);
509 : 4 : }
510 : :
511 : :
512 : : /*
513 : : * ClearOrSaveResult
514 : : *
515 : : * If the result represents an error, remember it for possible display by
516 : : * \errverbose. Otherwise, just PQclear() it.
517 : : *
518 : : * Note: current policy is to apply this to the results of all queries,
519 : : * including "back door" queries, for debugging's sake. It's OK to use
520 : : * PQclear() directly on results known to not be error results, however.
521 : : */
522 : : static void
2933 523 : 160959 : ClearOrSaveResult(PGresult *result)
524 : : {
525 [ + + ]: 160959 : if (result)
526 : : {
527 [ + + ]: 160699 : switch (PQresultStatus(result))
528 : : {
529 : 19327 : case PGRES_NONFATAL_ERROR:
530 : : case PGRES_FATAL_ERROR:
651 peter@eisentraut.org 531 : 19327 : PQclear(pset.last_error_result);
2933 tgl@sss.pgh.pa.us 532 : 19327 : pset.last_error_result = result;
533 : 19327 : break;
534 : :
535 : 141372 : default:
536 : 141372 : PQclear(result);
537 : 141372 : break;
538 : : }
539 : : }
540 : 160959 : }
541 : :
542 : :
543 : : /*
544 : : * Consume all results
545 : : */
546 : : static void
741 peter@eisentraut.org 547 :UBC 0 : ClearOrSaveAllResults(void)
548 : : {
549 : : PGresult *result;
550 : :
551 [ # # ]: 0 : while ((result = PQgetResult(pset.db)) != NULL)
552 : 0 : ClearOrSaveResult(result);
553 : 0 : }
554 : :
555 : :
556 : : /*
557 : : * Print microtiming output. Always print raw milliseconds; if the interval
558 : : * is >= 1 second, also break it down into days/hours/minutes/seconds.
559 : : */
560 : : static void
2780 tgl@sss.pgh.pa.us 561 :CBC 2 : PrintTiming(double elapsed_msec)
562 : : {
563 : : double seconds;
564 : : double minutes;
565 : : double hours;
566 : : double days;
567 : :
568 [ + - ]: 2 : if (elapsed_msec < 1000.0)
569 : : {
570 : : /* This is the traditional (pre-v10) output format */
571 : 2 : printf(_("Time: %.3f ms\n"), elapsed_msec);
572 : 2 : return;
573 : : }
574 : :
575 : : /*
576 : : * Note: we could print just seconds, in a format like %06.3f, when the
577 : : * total is less than 1min. But that's hard to interpret unless we tack
578 : : * on "s" or otherwise annotate it. Forcing the display to include
579 : : * minutes seems like a better solution.
580 : : */
2780 tgl@sss.pgh.pa.us 581 :UBC 0 : seconds = elapsed_msec / 1000.0;
582 : 0 : minutes = floor(seconds / 60.0);
583 : 0 : seconds -= 60.0 * minutes;
584 [ # # ]: 0 : if (minutes < 60.0)
585 : : {
586 : 0 : printf(_("Time: %.3f ms (%02d:%06.3f)\n"),
587 : : elapsed_msec, (int) minutes, seconds);
588 : 0 : return;
589 : : }
590 : :
591 : 0 : hours = floor(minutes / 60.0);
592 : 0 : minutes -= 60.0 * hours;
593 [ # # ]: 0 : if (hours < 24.0)
594 : : {
595 : 0 : printf(_("Time: %.3f ms (%02d:%02d:%06.3f)\n"),
596 : : elapsed_msec, (int) hours, (int) minutes, seconds);
597 : 0 : return;
598 : : }
599 : :
600 : 0 : days = floor(hours / 24.0);
601 : 0 : hours -= 24.0 * days;
602 : 0 : printf(_("Time: %.3f ms (%.0f d %02d:%02d:%06.3f)\n"),
603 : : elapsed_msec, days, (int) hours, (int) minutes, seconds);
604 : : }
605 : :
606 : :
607 : : /*
608 : : * PSQLexec
609 : : *
610 : : * This is the way to send "backdoor" queries (those not directly entered
611 : : * by the user). It is subject to -E but not -e.
612 : : *
613 : : * Caller is responsible for handling the ensuing processing if a COPY
614 : : * command is sent.
615 : : *
616 : : * Note: we don't bother to check PQclientEncoding; it is assumed that no
617 : : * caller uses this path to issue "SET CLIENT_ENCODING".
618 : : */
619 : : PGresult *
3461 fujii@postgresql.org 620 :CBC 16399 : PSQLexec(const char *query)
621 : : {
622 : : PGresult *res;
623 : :
8857 peter_e@gmx.net 624 [ - + ]: 16399 : if (!pset.db)
625 : : {
1840 peter@eisentraut.org 626 :UBC 0 : pg_log_error("You are currently not connected to a database.");
8928 bruce@momjian.us 627 : 0 : return NULL;
628 : : }
629 : :
6438 tgl@sss.pgh.pa.us 630 [ - + ]:CBC 16399 : if (pset.echo_hidden != PSQL_ECHO_HIDDEN_OFF)
631 : : {
263 nathan@postgresql.or 632 :UNC 0 : printf(_("/******** QUERY *********/\n"
633 : : "%s\n"
634 : : "/************************/\n\n"), query);
8928 bruce@momjian.us 635 :UBC 0 : fflush(stdout);
6879 636 [ # # ]: 0 : if (pset.logfile)
637 : : {
6767 peter_e@gmx.net 638 : 0 : fprintf(pset.logfile,
263 nathan@postgresql.or 639 :UNC 0 : _("/******** QUERY *********/\n"
640 : : "%s\n"
641 : : "/************************/\n\n"), query);
6879 bruce@momjian.us 642 :UBC 0 : fflush(pset.logfile);
643 : : }
644 : :
6438 tgl@sss.pgh.pa.us 645 [ # # ]: 0 : if (pset.echo_hidden == PSQL_ECHO_HIDDEN_NOEXEC)
7596 646 : 0 : return NULL;
647 : : }
648 : :
1595 michael@paquier.xyz 649 :CBC 16399 : SetCancelConn(pset.db);
650 : :
7596 tgl@sss.pgh.pa.us 651 : 16399 : res = PQexec(pset.db, query);
652 : :
6438 653 : 16399 : ResetCancelConn();
654 : :
741 peter@eisentraut.org 655 [ - + ]: 16399 : if (!AcceptResult(res, true))
656 : : {
2933 tgl@sss.pgh.pa.us 657 :UBC 0 : ClearOrSaveResult(res);
7696 bruce@momjian.us 658 : 0 : res = NULL;
659 : : }
660 : :
7696 bruce@momjian.us 661 :CBC 16399 : return res;
662 : : }
663 : :
664 : :
665 : : /*
666 : : * PSQLexecWatch
667 : : *
668 : : * This function is used for \watch command to send the query to
669 : : * the server and print out the result.
670 : : *
671 : : * Returns 1 if the query executed successfully, 0 if it cannot be repeated,
672 : : * e.g., because of the interrupt, -1 on error.
673 : : */
674 : : int
229 dgustafsson@postgres 675 :GNC 8 : PSQLexecWatch(const char *query, const printQueryOpt *opt, FILE *printQueryFout, int min_rows)
676 : : {
915 peter@eisentraut.org 677 :CBC 8 : bool timing = pset.timing;
3249 bruce@momjian.us 678 : 8 : double elapsed_msec = 0;
679 : : int res;
680 : :
3510 fujii@postgresql.org 681 [ - + ]: 8 : if (!pset.db)
682 : : {
1840 peter@eisentraut.org 683 :UBC 0 : pg_log_error("You are currently not connected to a database.");
3510 fujii@postgresql.org 684 : 0 : return 0;
685 : : }
686 : :
1595 michael@paquier.xyz 687 :CBC 8 : SetCancelConn(pset.db);
688 : :
229 dgustafsson@postgres 689 :GNC 8 : res = ExecQueryAndProcessResults(query, &elapsed_msec, NULL, true, min_rows, opt, printQueryFout);
690 : :
3510 fujii@postgresql.org 691 :CBC 7 : ResetCancelConn();
692 : :
693 : : /* Possible microtiming output */
915 peter@eisentraut.org 694 [ - + ]: 7 : if (timing)
2780 tgl@sss.pgh.pa.us 695 :UBC 0 : PrintTiming(elapsed_msec);
696 : :
741 peter@eisentraut.org 697 :CBC 7 : return res;
698 : : }
699 : :
700 : :
701 : : /*
702 : : * PrintNotifications: check for asynchronous notifications, and print them out
703 : : */
704 : : static void
7696 bruce@momjian.us 705 : 160009 : PrintNotifications(void)
706 : : {
707 : : PGnotify *notify;
708 : :
2004 tgl@sss.pgh.pa.us 709 : 160009 : PQconsumeInput(pset.db);
710 [ + + ]: 160011 : while ((notify = PQnotifies(pset.db)) != NULL)
711 : : {
712 : : /* for backward compatibility, only show payload if nonempty */
5171 713 [ + + ]: 2 : if (notify->extra[0])
714 : 1 : fprintf(pset.queryFout, _("Asynchronous notification \"%s\" with payload \"%s\" received from server process with PID %d.\n"),
715 : : notify->relname, notify->extra, notify->be_pid);
716 : : else
717 : 1 : fprintf(pset.queryFout, _("Asynchronous notification \"%s\" received from server process with PID %d.\n"),
718 : : notify->relname, notify->be_pid);
7696 bruce@momjian.us 719 : 2 : fflush(pset.queryFout);
7516 tgl@sss.pgh.pa.us 720 : 2 : PQfreemem(notify);
2004 721 : 2 : PQconsumeInput(pset.db);
722 : : }
7696 bruce@momjian.us 723 : 160009 : }
724 : :
725 : :
726 : : /*
727 : : * PrintQueryTuples: assuming query result is OK, print its tuples
728 : : *
729 : : * We use the options given by opt unless that's NULL, in which case
730 : : * we use pset.popt.
731 : : *
732 : : * Output is to printQueryFout unless that's NULL, in which case
733 : : * we use pset.queryFout.
734 : : *
735 : : * Returns true if successful, false otherwise.
736 : : */
737 : : static bool
559 tgl@sss.pgh.pa.us 738 : 61393 : PrintQueryTuples(const PGresult *result, const printQueryOpt *opt,
739 : : FILE *printQueryFout)
740 : : {
794 peter@eisentraut.org 741 : 61393 : bool ok = true;
559 tgl@sss.pgh.pa.us 742 [ + + ]: 61393 : FILE *fout = printQueryFout ? printQueryFout : pset.queryFout;
743 : :
744 [ + + ]: 61393 : printQuery(result, opt ? opt : &pset.popt, fout, false, pset.logfile);
745 : 61393 : fflush(fout);
746 [ - + ]: 61393 : if (ferror(fout))
747 : : {
559 tgl@sss.pgh.pa.us 748 :UBC 0 : pg_log_error("could not print result table: %m");
749 : 0 : ok = false;
750 : : }
751 : :
794 peter@eisentraut.org 752 :CBC 61393 : return ok;
753 : : }
754 : :
755 : :
756 : : /*
757 : : * StoreQueryTuple: assuming query result is OK, save data into variables
758 : : *
759 : : * Returns true if successful, false otherwise.
760 : : */
761 : : static bool
4089 tgl@sss.pgh.pa.us 762 : 329 : StoreQueryTuple(const PGresult *result)
763 : : {
764 : 329 : bool success = true;
765 : :
766 [ + + ]: 329 : if (PQntuples(result) < 1)
767 : : {
1840 peter@eisentraut.org 768 : 9 : pg_log_error("no rows returned for \\gset");
4089 tgl@sss.pgh.pa.us 769 : 9 : success = false;
770 : : }
771 [ + + ]: 320 : else if (PQntuples(result) > 1)
772 : : {
1840 peter@eisentraut.org 773 : 6 : pg_log_error("more than one row returned for \\gset");
4089 tgl@sss.pgh.pa.us 774 : 6 : success = false;
775 : : }
776 : : else
777 : : {
778 : : int i;
779 : :
780 [ + + ]: 700 : for (i = 0; i < PQnfields(result); i++)
781 : : {
782 : 389 : char *colname = PQfname(result, i);
783 : : char *varname;
784 : : char *value;
785 : :
786 : : /* concatenate prefix and column name */
3827 787 : 389 : varname = psprintf("%s%s", pset.gset_prefix, colname);
788 : :
1252 noah@leadboat.com 789 [ + + ]: 389 : if (VariableHasHook(pset.vars, varname))
790 : : {
791 : 3 : pg_log_warning("attempt to \\gset into specially treated variable \"%s\" ignored",
792 : : varname);
793 : 3 : continue;
794 : : }
795 : :
4089 tgl@sss.pgh.pa.us 796 [ + + ]: 386 : if (!PQgetisnull(result, 0, i))
797 : 380 : value = PQgetvalue(result, 0, i);
798 : : else
799 : : {
800 : : /* for NULL value, unset rather than set the variable */
801 : 6 : value = NULL;
802 : : }
803 : :
804 [ + + ]: 386 : if (!SetVariable(pset.vars, varname, value))
805 : : {
806 : 3 : free(varname);
807 : 3 : success = false;
808 : 3 : break;
809 : : }
810 : :
811 : 383 : free(varname);
812 : : }
813 : : }
814 : :
815 : 329 : return success;
816 : : }
817 : :
818 : :
819 : : /*
820 : : * ExecQueryTuples: assuming query result is OK, execute each query
821 : : * result field as a SQL statement
822 : : *
823 : : * Returns true if successful, false otherwise.
824 : : */
825 : : static bool
2932 826 : 22 : ExecQueryTuples(const PGresult *result)
827 : : {
828 : 22 : bool success = true;
829 : 22 : int nrows = PQntuples(result);
830 : 22 : int ncolumns = PQnfields(result);
831 : : int r,
832 : : c;
833 : :
834 : : /*
835 : : * We must turn off gexec_flag to avoid infinite recursion.
836 : : */
837 : 22 : pset.gexec_flag = false;
838 : :
839 [ + + ]: 234 : for (r = 0; r < nrows; r++)
840 : : {
841 [ + + ]: 433 : for (c = 0; c < ncolumns; c++)
842 : : {
843 [ + + ]: 221 : if (!PQgetisnull(result, r, c))
844 : : {
845 : 218 : const char *query = PQgetvalue(result, r, c);
846 : :
847 : : /* Abandon execution if cancel_pressed */
1580 michael@paquier.xyz 848 [ - + ]: 218 : if (cancel_pressed)
2932 tgl@sss.pgh.pa.us 849 :UBC 0 : goto loop_exit;
850 : :
851 : : /*
852 : : * ECHO_ALL mode should echo these queries, but SendQuery
853 : : * assumes that MainLoop did that, so we have to do it here.
854 : : */
2932 tgl@sss.pgh.pa.us 855 [ + + + - ]:CBC 218 : if (pset.echo == PSQL_ECHO_ALL && !pset.singlestep)
856 : : {
857 : 215 : puts(query);
858 : 215 : fflush(stdout);
859 : : }
860 : :
861 [ + + ]: 218 : if (!SendQuery(query))
862 : : {
863 : : /* Error - abandon execution if ON_ERROR_STOP */
864 : 3 : success = false;
865 [ - + ]: 3 : if (pset.on_error_stop)
2932 tgl@sss.pgh.pa.us 866 :UBC 0 : goto loop_exit;
867 : : }
868 : : }
869 : : }
870 : : }
871 : :
2932 tgl@sss.pgh.pa.us 872 :CBC 22 : loop_exit:
873 : :
874 : : /*
875 : : * Restore state. We know gexec_flag was on, else we'd not be here. (We
876 : : * also know it'll get turned off at end of command, but that's not ours
877 : : * to do here.)
878 : : */
879 : 22 : pset.gexec_flag = true;
880 : :
881 : : /* Return true if all queries were successful */
882 : 22 : return success;
883 : : }
884 : :
885 : :
886 : : /*
887 : : * Marshal the COPY data. Either path will get the
888 : : * connection out of its COPY state, then call PQresultStatus()
889 : : * once and report any error. Return whether all was ok.
890 : : *
891 : : * For COPY OUT, direct the output to copystream, or discard if that's NULL.
892 : : * For COPY IN, use pset.copyStream as data source if it's set,
893 : : * otherwise cur_cmd_source.
894 : : *
895 : : * Update *resultp if further processing is necessary; set to NULL otherwise.
896 : : * Return a result when queryFout can safely output a result status: on COPY
897 : : * IN, or on COPY OUT if written to something other than pset.queryFout.
898 : : * Returning NULL prevents the command status from being printed, which we
899 : : * want if the status line doesn't get taken as part of the COPY data.
900 : : */
901 : : static bool
559 902 : 680 : HandleCopyResult(PGresult **resultp, FILE *copystream)
903 : : {
904 : : bool success;
905 : : PGresult *copy_result;
744 peter@eisentraut.org 906 : 680 : ExecStatusType result_status = PQresultStatus(*resultp);
907 : :
908 [ + + - + ]: 680 : Assert(result_status == PGRES_COPY_OUT ||
909 : : result_status == PGRES_COPY_IN);
910 : :
911 : 680 : SetCancelConn(pset.db);
912 : :
913 [ + + ]: 680 : if (result_status == PGRES_COPY_OUT)
914 : : {
915 : 249 : success = handleCopyOut(pset.db,
916 : : copystream,
917 : : ©_result)
918 [ + + + - ]: 249 : && (copystream != NULL);
919 : :
920 : : /*
921 : : * Suppress status printing if the report would go to the same place
922 : : * as the COPY data just went. Note this doesn't prevent error
923 : : * reporting, since handleCopyOut did that.
924 : : */
925 [ + + ]: 249 : if (copystream == pset.queryFout)
926 : : {
927 : 236 : PQclear(copy_result);
928 : 236 : copy_result = NULL;
929 : : }
930 : : }
931 : : else
932 : : {
933 : : /* COPY IN */
934 : : /* Ignore the copystream argument passed to the function */
935 [ + + ]: 431 : copystream = pset.copyStream ? pset.copyStream : pset.cur_cmd_source;
936 : 431 : success = handleCopyIn(pset.db,
937 : : copystream,
938 : 431 : PQbinaryTuples(*resultp),
939 : : ©_result);
940 : : }
941 : 680 : ResetCancelConn();
942 : :
943 : : /*
944 : : * Replace the PGRES_COPY_OUT/IN result with COPY command's exit status,
945 : : * or with NULL if we want to suppress printing anything.
946 : : */
947 : 680 : PQclear(*resultp);
948 : 680 : *resultp = copy_result;
949 : :
950 : 680 : return success;
951 : : }
952 : :
953 : : /*
954 : : * PrintQueryStatus: report command status as required
955 : : */
956 : : static void
741 957 : 141144 : PrintQueryStatus(PGresult *result, FILE *printQueryFout)
958 : : {
959 : : char buf[16];
6 tgl@sss.pgh.pa.us 960 :GNC 141144 : const char *cmdstatus = PQcmdStatus(result);
741 peter@eisentraut.org 961 [ + - ]:CBC 141144 : FILE *fout = printQueryFout ? printQueryFout : pset.queryFout;
962 : :
963 : : /* Do nothing if it's a TUPLES_OK result that isn't from RETURNING */
6 tgl@sss.pgh.pa.us 964 [ + + ]:GNC 141144 : if (PQresultStatus(result) == PGRES_TUPLES_OK)
965 : : {
966 [ + + ]: 61834 : if (!(strncmp(cmdstatus, "INSERT", 6) == 0 ||
967 [ + + ]: 61622 : strncmp(cmdstatus, "UPDATE", 6) == 0 ||
968 [ + + ]: 61420 : strncmp(cmdstatus, "DELETE", 6) == 0 ||
969 [ + + ]: 61329 : strncmp(cmdstatus, "MERGE", 5) == 0))
970 : 61293 : return;
971 : : }
972 : :
6438 tgl@sss.pgh.pa.us 973 [ + + ]:CBC 79851 : if (!pset.quiet)
974 : : {
6454 975 [ - + ]: 417 : if (pset.popt.topt.format == PRINT_HTML)
976 : : {
741 peter@eisentraut.org 977 :UBC 0 : fputs("<p>", fout);
6 tgl@sss.pgh.pa.us 978 :UNC 0 : html_escaped_print(cmdstatus, fout);
741 peter@eisentraut.org 979 :UBC 0 : fputs("</p>\n", fout);
980 : : }
981 : : else
6 tgl@sss.pgh.pa.us 982 :GNC 417 : fprintf(fout, "%s\n", cmdstatus);
559 tgl@sss.pgh.pa.us 983 :CBC 417 : fflush(fout);
984 : : }
985 : :
6454 986 [ - + ]: 79851 : if (pset.logfile)
6 tgl@sss.pgh.pa.us 987 :UNC 0 : fprintf(pset.logfile, "%s\n", cmdstatus);
988 : :
794 peter@eisentraut.org 989 :CBC 79851 : snprintf(buf, sizeof(buf), "%u", (unsigned int) PQoidValue(result));
6454 tgl@sss.pgh.pa.us 990 : 79851 : SetVariable(pset.vars, "LASTOID", buf);
6454 tgl@sss.pgh.pa.us 991 :ECB (74324) : }
992 : :
993 : :
994 : : /*
995 : : * PrintQueryResult: print out (or store or execute) query result as required
996 : : *
997 : : * last is true if this is the last result of a command string.
998 : : * opt and printQueryFout are defined as for PrintQueryTuples.
999 : : * printStatusFout is where to send command status; NULL means pset.queryFout.
1000 : : *
1001 : : * Returns true if the query executed successfully, false otherwise.
1002 : : */
1003 : : static bool
559 tgl@sss.pgh.pa.us 1004 :CBC 141185 : PrintQueryResult(PGresult *result, bool last,
1005 : : const printQueryOpt *opt, FILE *printQueryFout,
1006 : : FILE *printStatusFout)
1007 : : {
1008 : : bool success;
1009 : :
794 peter@eisentraut.org 1010 [ - + ]: 141185 : if (!result)
7496 tgl@sss.pgh.pa.us 1011 :UBC 0 : return false;
1012 : :
794 peter@eisentraut.org 1013 [ + + + - :CBC 141185 : switch (PQresultStatus(result))
- - ]
1014 : : {
7496 tgl@sss.pgh.pa.us 1015 : 61817 : case PGRES_TUPLES_OK:
1016 : : /* store or execute or print the data ... */
741 peter@eisentraut.org 1017 [ + + + + ]: 61817 : if (last && pset.gset_prefix)
794 1018 : 329 : success = StoreQueryTuple(result);
741 1019 [ + + + + ]: 61488 : else if (last && pset.gexec_flag)
794 1020 : 22 : success = ExecQueryTuples(result);
741 1021 [ + + + + ]: 61466 : else if (last && pset.crosstab_flag)
794 1022 : 66 : success = PrintResultInCrosstab(result);
741 1023 [ + + + + ]: 61400 : else if (last || pset.show_all_results)
1024 : 61393 : success = PrintQueryTuples(result, opt, printQueryFout);
1025 : : else
1026 : 7 : success = true;
1027 : :
1028 : : /*
1029 : : * If it's INSERT/UPDATE/DELETE/MERGE RETURNING, also print
1030 : : * status.
1031 : : */
1032 [ + + + + ]: 61817 : if (last || pset.show_all_results)
6 tgl@sss.pgh.pa.us 1033 :GNC 61810 : PrintQueryStatus(result, printStatusFout);
1034 : :
7496 tgl@sss.pgh.pa.us 1035 :CBC 61817 : break;
1036 : :
7559 bruce@momjian.us 1037 : 79310 : case PGRES_COMMAND_OK:
741 peter@eisentraut.org 1038 [ + + + - ]: 79310 : if (last || pset.show_all_results)
559 tgl@sss.pgh.pa.us 1039 : 79310 : PrintQueryStatus(result, printStatusFout);
6454 1040 : 79310 : success = true;
1041 : 79310 : break;
1042 : :
7496 1043 : 58 : case PGRES_EMPTY_QUERY:
1044 : 58 : success = true;
7559 bruce@momjian.us 1045 : 58 : break;
1046 : :
7496 tgl@sss.pgh.pa.us 1047 :UBC 0 : case PGRES_COPY_OUT:
1048 : : case PGRES_COPY_IN:
1049 : : /* nothing to do here: already processed */
1050 : 0 : success = true;
7559 bruce@momjian.us 1051 : 0 : break;
1052 : :
4463 alvherre@alvh.no-ip. 1053 :LBC (78) : case PGRES_BAD_RESPONSE:
1054 : : case PGRES_NONFATAL_ERROR:
1055 : : case PGRES_FATAL_ERROR:
1056 : (78) : success = false;
1057 : (78) : break;
1058 : :
7696 bruce@momjian.us 1059 :UBC 0 : default:
4463 alvherre@alvh.no-ip. 1060 : 0 : success = false;
1840 peter@eisentraut.org 1061 : 0 : pg_log_error("unexpected PQresultStatus: %d",
1062 : : PQresultStatus(result));
7559 bruce@momjian.us 1063 : 0 : break;
1064 : : }
1065 : :
7696 bruce@momjian.us 1066 :CBC 141185 : return success;
1067 : : }
1068 : :
1069 : : /*
1070 : : * SendQuery: send the query string to the backend
1071 : : * (and print out result)
1072 : : *
1073 : : * Note: This is the "front door" way to send a query. That is, use it to
1074 : : * send queries actually entered by the user. These queries will be subject to
1075 : : * single step mode.
1076 : : * To send "back door" queries (generated by slash commands, etc.) in a
1077 : : * controlled way, use PSQLexec().
1078 : : *
1079 : : * Returns true if the query executed successfully, false otherwise.
1080 : : */
1081 : : bool
1082 : 160021 : SendQuery(const char *query)
1083 : : {
915 peter@eisentraut.org 1084 : 160021 : bool timing = pset.timing;
1085 : : PGTransactionStatusType transaction_status;
6438 tgl@sss.pgh.pa.us 1086 : 160021 : double elapsed_msec = 0;
4089 1087 : 160021 : bool OK = false;
1088 : : int i;
1089 : 160021 : bool on_error_rollback_savepoint = false;
745 peter@eisentraut.org 1090 : 160021 : bool svpt_gone = false;
1091 : :
7696 bruce@momjian.us 1092 [ - + ]: 160021 : if (!pset.db)
1093 : : {
1840 peter@eisentraut.org 1094 :UBC 0 : pg_log_error("You are currently not connected to a database.");
4089 tgl@sss.pgh.pa.us 1095 : 0 : goto sendquery_cleanup;
1096 : : }
1097 : :
6438 tgl@sss.pgh.pa.us 1098 [ - + ]:CBC 160021 : if (pset.singlestep)
1099 : : {
1100 : : char buf[3];
1101 : :
2932 tgl@sss.pgh.pa.us 1102 :UBC 0 : fflush(stderr);
263 nathan@postgresql.or 1103 :UNC 0 : printf(_("/**(Single step mode: verify command)******************************************/\n"
1104 : : "%s\n"
1105 : : "/**(press return to proceed or enter x and return to cancel)*******************/\n"),
1106 : : query);
7696 bruce@momjian.us 1107 :UBC 0 : fflush(stdout);
1108 [ # # ]: 0 : if (fgets(buf, sizeof(buf), stdin) != NULL)
1109 [ # # ]: 0 : if (buf[0] == 'x')
4089 tgl@sss.pgh.pa.us 1110 : 0 : goto sendquery_cleanup;
1580 michael@paquier.xyz 1111 [ # # ]: 0 : if (cancel_pressed)
2932 tgl@sss.pgh.pa.us 1112 : 0 : goto sendquery_cleanup;
1113 : : }
6438 tgl@sss.pgh.pa.us 1114 [ + + ]:CBC 160021 : else if (pset.echo == PSQL_ECHO_QUERIES)
1115 : : {
7696 bruce@momjian.us 1116 : 15 : puts(query);
7563 1117 : 15 : fflush(stdout);
1118 : : }
1119 : :
6879 1120 [ - + ]: 160021 : if (pset.logfile)
1121 : : {
6767 peter_e@gmx.net 1122 :UBC 0 : fprintf(pset.logfile,
263 nathan@postgresql.or 1123 :UNC 0 : _("/******** QUERY *********/\n"
1124 : : "%s\n"
1125 : : "/************************/\n\n"), query);
6879 bruce@momjian.us 1126 :UBC 0 : fflush(pset.logfile);
1127 : : }
1128 : :
1595 michael@paquier.xyz 1129 :CBC 160021 : SetCancelConn(pset.db);
1130 : :
6926 bruce@momjian.us 1131 : 160021 : transaction_status = PQtransactionStatus(pset.db);
1132 : :
1133 [ + + ]: 160021 : if (transaction_status == PQTRANS_IDLE &&
6438 tgl@sss.pgh.pa.us 1134 [ + + ]: 146422 : !pset.autocommit &&
7146 1135 [ + + ]: 42 : !command_no_begin(query))
1136 : : {
1137 : : PGresult *result;
1138 : :
794 peter@eisentraut.org 1139 : 36 : result = PQexec(pset.db, "BEGIN");
1140 [ - + ]: 36 : if (PQresultStatus(result) != PGRES_COMMAND_OK)
1141 : : {
1840 peter@eisentraut.org 1142 :UBC 0 : pg_log_info("%s", PQerrorMessage(pset.db));
794 1143 : 0 : ClearOrSaveResult(result);
4089 tgl@sss.pgh.pa.us 1144 : 0 : goto sendquery_cleanup;
1145 : : }
794 peter@eisentraut.org 1146 :CBC 36 : ClearOrSaveResult(result);
6781 bruce@momjian.us 1147 : 36 : transaction_status = PQtransactionStatus(pset.db);
1148 : : }
1149 : :
1150 [ + + ]: 160021 : if (transaction_status == PQTRANS_INTRANS &&
6438 tgl@sss.pgh.pa.us 1151 [ + + ]: 13105 : pset.on_error_rollback != PSQL_ERROR_ROLLBACK_OFF &&
6781 bruce@momjian.us 1152 [ + - ]: 78 : (pset.cur_cmd_interactive ||
6438 tgl@sss.pgh.pa.us 1153 [ + - ]: 78 : pset.on_error_rollback == PSQL_ERROR_ROLLBACK_ON))
1154 : : {
1155 : : PGresult *result;
1156 : :
794 peter@eisentraut.org 1157 : 78 : result = PQexec(pset.db, "SAVEPOINT pg_psql_temporary_savepoint");
1158 [ - + ]: 78 : if (PQresultStatus(result) != PGRES_COMMAND_OK)
1159 : : {
850 tgl@sss.pgh.pa.us 1160 :UBC 0 : pg_log_info("%s", PQerrorMessage(pset.db));
794 peter@eisentraut.org 1161 : 0 : ClearOrSaveResult(result);
850 tgl@sss.pgh.pa.us 1162 : 0 : goto sendquery_cleanup;
1163 : : }
794 peter@eisentraut.org 1164 :CBC 78 : ClearOrSaveResult(result);
850 tgl@sss.pgh.pa.us 1165 : 78 : on_error_rollback_savepoint = true;
1166 : : }
1167 : :
2413 1168 [ + + ]: 160021 : if (pset.gdesc_flag)
1169 : : {
1170 : : /* Describe query's result columns, without executing it */
1171 : 34 : OK = DescribeQuery(query, &elapsed_msec);
1172 : : }
1173 : : else
1174 : : {
1175 : : /* Default fetch-and-print mode */
8 tgl@sss.pgh.pa.us 1176 :GNC 159987 : OK = (ExecQueryAndProcessResults(query, &elapsed_msec, &svpt_gone, false, 0, NULL, NULL) > 0);
1177 : : }
1178 : :
3566 fujii@postgresql.org 1179 [ + + + + ]:CBC 160009 : if (!OK && pset.echo == PSQL_ECHO_ERRORS)
1840 peter@eisentraut.org 1180 : 3 : pg_log_info("STATEMENT: %s", query);
1181 : :
1182 : : /* If we made a temporary savepoint, possibly release/rollback */
6926 bruce@momjian.us 1183 [ + + ]: 160009 : if (on_error_rollback_savepoint)
1184 : : {
4463 alvherre@alvh.no-ip. 1185 : 78 : const char *svptcmd = NULL;
1186 : :
6926 bruce@momjian.us 1187 : 78 : transaction_status = PQtransactionStatus(pset.db);
1188 : :
4463 alvherre@alvh.no-ip. 1189 [ + + + - ]: 78 : switch (transaction_status)
1190 : : {
1191 : 21 : case PQTRANS_INERROR:
1192 : : /* We always rollback on an error */
1193 : 21 : svptcmd = "ROLLBACK TO pg_psql_temporary_savepoint";
1194 : 21 : break;
1195 : :
1196 : 21 : case PQTRANS_IDLE:
1197 : : /* If they are no longer in a transaction, then do nothing */
1198 : 21 : break;
1199 : :
1200 : 36 : case PQTRANS_INTRANS:
1201 : :
1202 : : /*
1203 : : * Release our savepoint, but do nothing if they are messing
1204 : : * with savepoints themselves
1205 : : */
745 peter@eisentraut.org 1206 [ + + ]: 36 : if (!svpt_gone)
4463 alvherre@alvh.no-ip. 1207 : 33 : svptcmd = "RELEASE pg_psql_temporary_savepoint";
1208 : 36 : break;
1209 : :
4463 alvherre@alvh.no-ip. 1210 :UBC 0 : case PQTRANS_ACTIVE:
1211 : : case PQTRANS_UNKNOWN:
1212 : : default:
1213 : 0 : OK = false;
1214 : : /* PQTRANS_UNKNOWN is expected given a broken connection. */
4421 rhaas@postgresql.org 1215 [ # # # # ]: 0 : if (transaction_status != PQTRANS_UNKNOWN || ConnectionUp())
1840 peter@eisentraut.org 1216 : 0 : pg_log_error("unexpected transaction status (%d)",
1217 : : transaction_status);
4463 alvherre@alvh.no-ip. 1218 : 0 : break;
1219 : : }
1220 : :
5720 tgl@sss.pgh.pa.us 1221 [ + + ]:CBC 78 : if (svptcmd)
1222 : : {
1223 : : PGresult *svptres;
1224 : :
1225 : 54 : svptres = PQexec(pset.db, svptcmd);
1226 [ - + ]: 54 : if (PQresultStatus(svptres) != PGRES_COMMAND_OK)
1227 : : {
1840 peter@eisentraut.org 1228 :UBC 0 : pg_log_info("%s", PQerrorMessage(pset.db));
2933 tgl@sss.pgh.pa.us 1229 : 0 : ClearOrSaveResult(svptres);
4089 1230 : 0 : OK = false;
1231 : :
1232 : 0 : goto sendquery_cleanup;
1233 : : }
6498 alvherre@alvh.no-ip. 1234 :CBC 54 : PQclear(svptres);
1235 : : }
1236 : : }
1237 : :
1238 : : /* Possible microtiming output */
915 peter@eisentraut.org 1239 [ + + ]: 160009 : if (timing)
2780 tgl@sss.pgh.pa.us 1240 : 2 : PrintTiming(elapsed_msec);
1241 : :
1242 : : /* check for events that may occur during query execution */
1243 : :
7516 1244 [ + + + - ]: 160011 : if (pset.encoding != PQclientEncoding(pset.db) &&
1245 : 2 : PQclientEncoding(pset.db) >= 0)
1246 : : {
1247 : : /* track effects of SET CLIENT_ENCODING */
1248 : 2 : pset.encoding = PQclientEncoding(pset.db);
1249 : 2 : pset.popt.topt.encoding = pset.encoding;
1250 : 2 : SetVariable(pset.vars, "ENCODING",
1251 : : pg_encoding_to_char(pset.encoding));
1252 : : }
1253 : :
7696 bruce@momjian.us 1254 : 160009 : PrintNotifications();
1255 : :
1256 : : /* perform cleanup that should occur after any attempted query */
1257 : :
4089 tgl@sss.pgh.pa.us 1258 : 160009 : sendquery_cleanup:
1259 : :
1260 : : /* global cancellation reset */
741 peter@eisentraut.org 1261 : 160009 : ResetCancelConn();
1262 : :
1263 : : /* reset \g's output-to-filename trigger */
4089 tgl@sss.pgh.pa.us 1264 [ + + ]: 160009 : if (pset.gfname)
1265 : : {
1266 : 16 : free(pset.gfname);
1267 : 16 : pset.gfname = NULL;
1268 : : }
1269 : :
1270 : : /* restore print settings if \g changed them */
1468 1271 [ + + ]: 160009 : if (pset.gsavepopt)
1272 : : {
1273 : 18 : restorePsetInfo(&pset.popt, pset.gsavepopt);
1274 : 18 : pset.gsavepopt = NULL;
1275 : : }
1276 : :
1277 : : /* clean up after \bind */
516 peter@eisentraut.org 1278 [ + + ]: 160009 : if (pset.bind_flag)
1279 : : {
1280 [ + + ]: 30 : for (i = 0; i < pset.bind_nparams; i++)
1281 : 12 : free(pset.bind_params[i]);
1282 : 18 : free(pset.bind_params);
1283 : 18 : pset.bind_params = NULL;
1284 : 18 : pset.bind_flag = false;
1285 : : }
1286 : :
1287 : : /* reset \gset trigger */
4089 tgl@sss.pgh.pa.us 1288 [ + + ]: 160009 : if (pset.gset_prefix)
1289 : : {
1290 : 329 : free(pset.gset_prefix);
1291 : 329 : pset.gset_prefix = NULL;
1292 : : }
1293 : :
1294 : : /* reset \gdesc trigger */
2413 1295 : 160009 : pset.gdesc_flag = false;
1296 : :
1297 : : /* reset \gexec trigger */
2932 1298 : 160009 : pset.gexec_flag = false;
1299 : :
1300 : : /* reset \crosstabview trigger */
2928 alvherre@alvh.no-ip. 1301 : 160009 : pset.crosstab_flag = false;
2922 tgl@sss.pgh.pa.us 1302 [ + + ]: 800045 : for (i = 0; i < lengthof(pset.ctv_args); i++)
1303 : : {
1304 : 640036 : pg_free(pset.ctv_args[i]);
1305 : 640036 : pset.ctv_args[i] = NULL;
1306 : : }
1307 : :
7696 bruce@momjian.us 1308 : 160009 : return OK;
1309 : : }
1310 : :
1311 : :
1312 : : /*
1313 : : * DescribeQuery: describe the result columns of a query, without executing it
1314 : : *
1315 : : * Returns true if the operation executed successfully, false otherwise.
1316 : : *
1317 : : * If pset.timing is on, total query time (exclusive of result-printing) is
1318 : : * stored into *elapsed_msec.
1319 : : */
1320 : : static bool
2413 tgl@sss.pgh.pa.us 1321 : 34 : DescribeQuery(const char *query, double *elapsed_msec)
1322 : : {
915 peter@eisentraut.org 1323 : 34 : bool timing = pset.timing;
1324 : : PGresult *result;
1325 : : bool OK;
1326 : : instr_time before,
1327 : : after;
1328 : :
2413 tgl@sss.pgh.pa.us 1329 : 34 : *elapsed_msec = 0;
1330 : :
915 peter@eisentraut.org 1331 [ - + ]: 34 : if (timing)
2413 tgl@sss.pgh.pa.us 1332 :UBC 0 : INSTR_TIME_SET_CURRENT(before);
1333 : : else
450 andres@anarazel.de 1334 :CBC 34 : INSTR_TIME_SET_ZERO(before);
1335 : :
1336 : : /*
1337 : : * To parse the query but not execute it, we prepare it, using the unnamed
1338 : : * prepared statement. This is invisible to psql users, since there's no
1339 : : * way to access the unnamed prepared statement from psql user space. The
1340 : : * next Parse or Query protocol message would overwrite the statement
1341 : : * anyway. (So there's no great need to clear it when done, which is a
1342 : : * good thing because libpq provides no easy way to do that.)
1343 : : */
794 peter@eisentraut.org 1344 : 34 : result = PQprepare(pset.db, "", query, 0, NULL);
1345 [ + + ]: 34 : if (PQresultStatus(result) != PGRES_COMMAND_OK)
1346 : : {
1840 1347 : 7 : pg_log_info("%s", PQerrorMessage(pset.db));
794 1348 : 7 : SetResultVariables(result, false);
1349 : 7 : ClearOrSaveResult(result);
2413 tgl@sss.pgh.pa.us 1350 : 7 : return false;
1351 : : }
794 peter@eisentraut.org 1352 : 27 : PQclear(result);
1353 : :
1354 : 27 : result = PQdescribePrepared(pset.db, "");
741 1355 [ + - + - ]: 54 : OK = AcceptResult(result, true) &&
794 1356 : 27 : (PQresultStatus(result) == PGRES_COMMAND_OK);
1357 [ + - + - ]: 27 : if (OK && result)
1358 : : {
1359 [ + + ]: 27 : if (PQnfields(result) > 0)
1360 : : {
1361 : : PQExpBufferData buf;
1362 : : int i;
1363 : :
2413 tgl@sss.pgh.pa.us 1364 : 18 : initPQExpBuffer(&buf);
1365 : :
1366 : 18 : printfPQExpBuffer(&buf,
1367 : : "SELECT name AS \"%s\", pg_catalog.format_type(tp, tpm) AS \"%s\"\n"
1368 : : "FROM (VALUES ",
1369 : : gettext_noop("Column"),
1370 : : gettext_noop("Type"));
1371 : :
794 peter@eisentraut.org 1372 [ + + ]: 81 : for (i = 0; i < PQnfields(result); i++)
1373 : : {
1374 : : const char *name;
1375 : : char *escname;
1376 : :
2413 tgl@sss.pgh.pa.us 1377 [ + + ]: 63 : if (i > 0)
1378 : 45 : appendPQExpBufferStr(&buf, ",");
1379 : :
794 peter@eisentraut.org 1380 : 63 : name = PQfname(result, i);
2413 tgl@sss.pgh.pa.us 1381 : 63 : escname = PQescapeLiteral(pset.db, name, strlen(name));
1382 : :
1383 [ - + ]: 63 : if (escname == NULL)
1384 : : {
1840 peter@eisentraut.org 1385 :UBC 0 : pg_log_info("%s", PQerrorMessage(pset.db));
794 1386 : 0 : PQclear(result);
2413 tgl@sss.pgh.pa.us 1387 : 0 : termPQExpBuffer(&buf);
1388 : 0 : return false;
1389 : : }
1390 : :
2413 tgl@sss.pgh.pa.us 1391 :CBC 63 : appendPQExpBuffer(&buf, "(%s, '%u'::pg_catalog.oid, %d)",
1392 : : escname,
1393 : : PQftype(result, i),
1394 : : PQfmod(result, i));
1395 : :
1396 : 63 : PQfreemem(escname);
1397 : : }
1398 : :
1399 : 18 : appendPQExpBufferStr(&buf, ") s(name, tp, tpm)");
794 peter@eisentraut.org 1400 : 18 : PQclear(result);
1401 : :
1402 : 18 : result = PQexec(pset.db, buf.data);
741 1403 : 18 : OK = AcceptResult(result, true);
1404 : :
915 1405 [ - + ]: 18 : if (timing)
1406 : : {
2413 tgl@sss.pgh.pa.us 1407 :UBC 0 : INSTR_TIME_SET_CURRENT(after);
1408 : 0 : INSTR_TIME_SUBTRACT(after, before);
1409 : 0 : *elapsed_msec += INSTR_TIME_GET_MILLISEC(after);
1410 : : }
1411 : :
794 peter@eisentraut.org 1412 [ + - + - ]:CBC 18 : if (OK && result)
559 tgl@sss.pgh.pa.us 1413 : 18 : OK = PrintQueryResult(result, true, NULL, NULL, NULL);
1414 : :
2413 1415 : 18 : termPQExpBuffer(&buf);
1416 : : }
1417 : : else
1418 : 9 : fprintf(pset.queryFout,
1419 : 9 : _("The command has no result, or the result has no columns.\n"));
1420 : : }
1421 : :
794 peter@eisentraut.org 1422 : 27 : SetResultVariables(result, OK);
1423 : 27 : ClearOrSaveResult(result);
1424 : :
2413 tgl@sss.pgh.pa.us 1425 : 27 : return OK;
1426 : : }
1427 : :
1428 : :
1429 : : /*
1430 : : * ExecQueryAndProcessResults: utility function for use by SendQuery()
1431 : : * and PSQLexecWatch().
1432 : : *
1433 : : * Sends query and cycles through PGresult objects.
1434 : : *
1435 : : * If our command string contained a COPY FROM STDIN or COPY TO STDOUT, the
1436 : : * PGresult associated with these commands must be processed by providing an
1437 : : * input or output stream. In that event, we'll marshal data for the COPY.
1438 : : *
1439 : : * For other commands, the results are processed normally, depending on their
1440 : : * status.
1441 : : *
1442 : : * When invoked from \watch, is_watch is true and min_rows is the value
1443 : : * of that option, or 0 if it wasn't set.
1444 : : *
1445 : : * Returns 1 on complete success, 0 on interrupt and -1 or errors. Possible
1446 : : * failure modes include purely client-side problems; check the transaction
1447 : : * status for the server-side opinion.
1448 : : *
1449 : : * Note that on a combined query, failure does not mean that nothing was
1450 : : * committed.
1451 : : */
1452 : : static int
559 1453 : 159995 : ExecQueryAndProcessResults(const char *query,
1454 : : double *elapsed_msec, bool *svpt_gone_p,
1455 : : bool is_watch, int min_rows,
1456 : : const printQueryOpt *opt, FILE *printQueryFout)
1457 : : {
745 peter@eisentraut.org 1458 : 159995 : bool timing = pset.timing;
1459 : : bool success;
229 dgustafsson@postgres 1460 :GNC 159995 : bool return_early = false;
1461 : : instr_time before,
1462 : : after;
1463 : : PGresult *result;
559 tgl@sss.pgh.pa.us 1464 :CBC 159995 : FILE *gfile_fout = NULL;
1465 : 159995 : bool gfile_is_pipe = false;
1466 : :
745 peter@eisentraut.org 1467 [ + + ]: 159995 : if (timing)
1468 : 2 : INSTR_TIME_SET_CURRENT(before);
1469 : : else
450 andres@anarazel.de 1470 : 159993 : INSTR_TIME_SET_ZERO(before);
1471 : :
516 peter@eisentraut.org 1472 [ + + ]: 159995 : if (pset.bind_flag)
331 tgl@sss.pgh.pa.us 1473 : 18 : success = PQsendQueryParams(pset.db, query, pset.bind_nparams, NULL, (const char *const *) pset.bind_params, NULL, NULL, 0);
1474 : : else
516 peter@eisentraut.org 1475 : 159977 : success = PQsendQuery(pset.db, query);
1476 : :
741 1477 [ - + ]: 159995 : if (!success)
1478 : : {
741 peter@eisentraut.org 1479 :UBC 0 : const char *error = PQerrorMessage(pset.db);
1480 : :
1481 [ # # ]: 0 : if (strlen(error))
1482 : 0 : pg_log_info("%s", error);
1483 : :
1484 : 0 : CheckConnection();
1485 : :
1486 : 0 : return -1;
1487 : : }
1488 : :
1489 : : /*
1490 : : * Fetch the result in chunks if FETCH_COUNT is set, except when:
1491 : : *
1492 : : * * SHOW_ALL_RESULTS is false, since that requires us to complete the
1493 : : * query before we can tell if its results should be displayed.
1494 : : *
1495 : : * * We're doing \crosstab, which likewise needs to see all the rows at
1496 : : * once.
1497 : : *
1498 : : * * We're doing \gexec: we must complete the data fetch to make the
1499 : : * connection free for issuing the resulting commands.
1500 : : *
1501 : : * * We're doing \gset: only one result row is allowed anyway.
1502 : : *
1503 : : * * We're doing \watch: users probably don't want us to force use of the
1504 : : * pager for that, plus chunking could break the min_rows check.
1505 : : */
8 tgl@sss.pgh.pa.us 1506 [ + + + - ]:GNC 159995 : if (pset.fetch_count > 0 && pset.show_all_results &&
1507 [ + - + + ]: 55 : !pset.crosstab_flag && !pset.gexec_flag &&
1508 [ + + + - ]: 52 : !pset.gset_prefix && !is_watch)
1509 : : {
1510 [ - + ]: 40 : if (!PQsetChunkedRowsMode(pset.db, pset.fetch_count))
8 tgl@sss.pgh.pa.us 1511 :UNC 0 : pg_log_warning("fetching results in chunked mode failed");
1512 : : }
1513 : :
1514 : : /*
1515 : : * If SIGINT is sent while the query is processing, the interrupt will be
1516 : : * consumed. The user's intention, though, is to cancel the entire watch
1517 : : * process, so detect a sent cancellation request and exit in this case.
1518 : : */
741 peter@eisentraut.org 1519 [ + + - + ]:CBC 159995 : if (is_watch && cancel_pressed)
1520 : : {
741 peter@eisentraut.org 1521 :UBC 0 : ClearOrSaveAllResults();
1522 : 0 : return 0;
1523 : : }
1524 : :
1525 : : /* first result */
741 peter@eisentraut.org 1526 :CBC 159995 : result = PQgetResult(pset.db);
229 dgustafsson@postgres 1527 [ + + + - ]:GNC 159995 : if (min_rows > 0 && PQntuples(result) < min_rows)
1528 : : {
1529 : 1 : return_early = true;
1530 : : }
1531 : :
741 peter@eisentraut.org 1532 [ + + ]:CBC 320743 : while (result != NULL)
1533 : : {
1534 : : ExecStatusType result_status;
8 tgl@sss.pgh.pa.us 1535 :GNC 160761 : bool is_chunked_result = false;
1536 : : PGresult *next_result;
1537 : : bool last;
1538 : :
741 peter@eisentraut.org 1539 [ + + ]:CBC 160761 : if (!AcceptResult(result, false))
1540 : 19228 : {
1541 : : /*
1542 : : * Some error occurred, either a server-side failure or a failure
1543 : : * to submit the command string. Record that.
1544 : : */
1545 : 19241 : const char *error = PQresultErrorMessage(result);
1546 : :
1547 [ + + ]: 19241 : if (strlen(error))
1548 : 19240 : pg_log_info("%s", error);
1549 : :
1550 : 19241 : CheckConnection();
1551 [ + + ]: 19228 : if (!is_watch)
1552 : 19227 : SetResultVariables(result, false);
1553 : :
1554 : : /* keep the result status before clearing it */
1555 : 19228 : result_status = PQresultStatus(result);
1556 : 19228 : ClearOrSaveResult(result);
1557 : 19228 : success = false;
1558 : :
1559 : : /*
1560 : : * switch to next result
1561 : : */
1562 [ + + + - ]: 19228 : if (result_status == PGRES_COPY_BOTH ||
1563 [ - + ]: 19227 : result_status == PGRES_COPY_OUT ||
1564 : : result_status == PGRES_COPY_IN)
1565 : :
1566 : : /*
1567 : : * For some obscure reason PQgetResult does *not* return a
1568 : : * NULL in copy cases despite the result having been cleared,
1569 : : * but keeps returning an "empty" result that we have to
1570 : : * ignore manually.
1571 : : */
1572 : 1 : result = NULL;
1573 : : else
1574 : 19227 : result = PQgetResult(pset.db);
1575 : :
1576 : : /*
1577 : : * Get current timing measure in case an error occurs
1578 : : */
692 1579 [ + + ]: 19228 : if (timing)
1580 : : {
1581 : 1 : INSTR_TIME_SET_CURRENT(after);
1582 : 1 : INSTR_TIME_SUBTRACT(after, before);
1583 : 1 : *elapsed_msec = INSTR_TIME_GET_MILLISEC(after);
1584 : : }
1585 : :
741 1586 : 19228 : continue;
1587 : : }
1588 [ + + + + ]: 141520 : else if (svpt_gone_p && !*svpt_gone_p)
1589 : : {
1590 : : /*
1591 : : * Check if the user ran any command that would destroy our
1592 : : * internal savepoint: If the user did COMMIT AND CHAIN, RELEASE
1593 : : * or ROLLBACK, our savepoint is gone. If they issued a SAVEPOINT,
1594 : : * releasing ours would remove theirs.
1595 : : */
1596 : 141483 : const char *cmd = PQcmdStatus(result);
1597 : :
1598 : 423661 : *svpt_gone_p = (strcmp(cmd, "COMMIT") == 0 ||
1599 [ + + ]: 140695 : strcmp(cmd, "SAVEPOINT") == 0 ||
1600 [ + + + + ]: 422230 : strcmp(cmd, "RELEASE") == 0 ||
1601 [ + + ]: 140052 : strcmp(cmd, "ROLLBACK") == 0);
1602 : : }
1603 : :
1604 : 141520 : result_status = PQresultStatus(result);
1605 : :
1606 : : /* must handle COPY before changing the current result */
1607 [ - + ]: 141520 : Assert(result_status != PGRES_COPY_BOTH);
1608 [ + + + + ]: 141520 : if (result_status == PGRES_COPY_IN ||
1609 : : result_status == PGRES_COPY_OUT)
1610 : : {
559 tgl@sss.pgh.pa.us 1611 : 680 : FILE *copy_stream = NULL;
1612 : :
1613 : : /*
1614 : : * For COPY OUT, direct the output to the default place (probably
1615 : : * a pager pipe) for \watch, or to pset.copyStream for \copy,
1616 : : * otherwise to pset.gfname if that's set, otherwise to
1617 : : * pset.queryFout.
1618 : : */
1619 [ + + ]: 680 : if (result_status == PGRES_COPY_OUT)
1620 : : {
1621 [ - + ]: 249 : if (is_watch)
1622 : : {
1623 : : /* invoked by \watch */
559 tgl@sss.pgh.pa.us 1624 [ # # ]:UBC 0 : copy_stream = printQueryFout ? printQueryFout : pset.queryFout;
1625 : : }
559 tgl@sss.pgh.pa.us 1626 [ + + ]:CBC 249 : else if (pset.copyStream)
1627 : : {
1628 : : /* invoked by \copy */
1629 : 27 : copy_stream = pset.copyStream;
1630 : : }
1631 [ + + ]: 222 : else if (pset.gfname)
1632 : : {
1633 : : /* COPY followed by \g filename or \g |program */
8 tgl@sss.pgh.pa.us 1634 :GNC 13 : success &= SetupGOutput(&gfile_fout, &gfile_is_pipe);
1635 [ + - ]: 13 : if (gfile_fout)
559 tgl@sss.pgh.pa.us 1636 :CBC 13 : copy_stream = gfile_fout;
1637 : : }
1638 : : else
1639 : : {
1640 : : /* fall back to the generic query output stream */
1641 : 209 : copy_stream = pset.queryFout;
1642 : : }
1643 : : }
1644 : :
1645 : : /*
1646 : : * Even if the output stream could not be opened, we call
1647 : : * HandleCopyResult() with a NULL output stream to collect and
1648 : : * discard the COPY data.
1649 : : */
1650 : 680 : success &= HandleCopyResult(&result, copy_stream);
1651 : : }
1652 : :
1653 : : /* If we have a chunked result, collect and print all chunks */
8 tgl@sss.pgh.pa.us 1654 [ + + ]:GNC 141520 : if (result_status == PGRES_TUPLES_CHUNK)
1655 : : {
6 1656 [ + - ]: 27 : FILE *tuples_fout = printQueryFout ? printQueryFout : pset.queryFout;
1657 [ - + ]: 27 : printQueryOpt my_popt = opt ? *opt : pset.popt;
8 1658 : 27 : int64 total_tuples = 0;
1659 : 27 : bool is_pager = false;
1660 : 27 : int flush_error = 0;
1661 : :
1662 : : /* initialize print options for partial table output */
1663 : 27 : my_popt.topt.start_table = true;
1664 : 27 : my_popt.topt.stop_table = false;
1665 : 27 : my_popt.topt.prior_records = 0;
1666 : :
1667 : : /* open \g file if needed */
1668 : 27 : success &= SetupGOutput(&gfile_fout, &gfile_is_pipe);
1669 [ - + ]: 27 : if (gfile_fout)
8 tgl@sss.pgh.pa.us 1670 :UNC 0 : tuples_fout = gfile_fout;
1671 : :
1672 : : /* force use of pager for any chunked resultset going to stdout */
8 tgl@sss.pgh.pa.us 1673 [ + - + - ]:GNC 27 : if (success && tuples_fout == stdout)
1674 : : {
1675 : 27 : tuples_fout = PageOutput(INT_MAX, &(my_popt.topt));
1676 : 27 : is_pager = true;
1677 : : }
1678 : :
1679 : : do
1680 : : {
1681 : : /*
1682 : : * Display the current chunk of results, unless the output
1683 : : * stream stopped working or we got cancelled. We skip use of
1684 : : * PrintQueryResult and go directly to printQuery, so that we
1685 : : * can pass the correct is_pager value and because we don't
1686 : : * want PrintQueryStatus to happen yet. Above, we rejected
1687 : : * use of chunking for all cases in which PrintQueryResult
1688 : : * would send the result to someplace other than printQuery.
1689 : : */
1690 [ + - + - : 39 : if (success && !flush_error && !cancel_pressed)
+ - ]
1691 : : {
1692 : 39 : printQuery(result, &my_popt, tuples_fout, is_pager, pset.logfile);
1693 : 39 : flush_error = fflush(tuples_fout);
1694 : : }
1695 : :
1696 : : /* after the first result set, disallow header decoration */
1697 : 39 : my_popt.topt.start_table = false;
1698 : :
1699 : : /* count tuples before dropping the result */
1700 : 39 : my_popt.topt.prior_records += PQntuples(result);
1701 : 39 : total_tuples += PQntuples(result);
1702 : :
1703 : 39 : ClearOrSaveResult(result);
1704 : :
1705 : : /* get the next result, loop if it's PGRES_TUPLES_CHUNK */
1706 : 39 : result = PQgetResult(pset.db);
1707 [ + + ]: 39 : } while (PQresultStatus(result) == PGRES_TUPLES_CHUNK);
1708 : :
1709 : : /* We expect an empty PGRES_TUPLES_OK, else there's a problem */
1710 [ + + ]: 27 : if (PQresultStatus(result) == PGRES_TUPLES_OK)
1711 : : {
1712 : : char buf[32];
1713 : :
1714 [ - + ]: 24 : Assert(PQntuples(result) == 0);
1715 : :
1716 : : /* Display the footer using the empty result */
1717 [ + - + - : 24 : if (success && !flush_error && !cancel_pressed)
+ - ]
1718 : : {
1719 : 24 : my_popt.topt.stop_table = true;
1720 : 24 : printQuery(result, &my_popt, tuples_fout, is_pager, pset.logfile);
1721 : 24 : fflush(tuples_fout);
1722 : : }
1723 : :
1724 [ + - ]: 24 : if (is_pager)
1725 : 24 : ClosePager(tuples_fout);
1726 : :
1727 : : /*
1728 : : * It's possible the data is from a RETURNING clause, in which
1729 : : * case we need to print query status.
1730 : : */
6 1731 : 24 : PrintQueryStatus(result, printQueryFout);
1732 : :
1733 : : /*
1734 : : * We must do a fake SetResultVariables(), since we don't have
1735 : : * a PGresult corresponding to the whole query.
1736 : : */
8 1737 : 24 : SetVariable(pset.vars, "ERROR", "false");
1738 : 24 : SetVariable(pset.vars, "SQLSTATE", "00000");
1739 : 24 : snprintf(buf, sizeof(buf), INT64_FORMAT, total_tuples);
1740 : 24 : SetVariable(pset.vars, "ROW_COUNT", buf);
1741 : : /* Prevent SetResultVariables call below */
1742 : 24 : is_chunked_result = true;
1743 : :
1744 : : /* Clear the empty result so it isn't printed below */
1745 : 24 : ClearOrSaveResult(result);
1746 : 24 : result = NULL;
1747 : : }
1748 : : else
1749 : : {
1750 : : /* Probably an error report, so close the pager and print it */
1751 [ + - ]: 3 : if (is_pager)
1752 : 3 : ClosePager(tuples_fout);
1753 : :
1754 : 3 : success &= AcceptResult(result, true);
1755 : : /* SetResultVariables and ClearOrSaveResult happen below */
1756 : : }
1757 : : }
1758 : :
1759 : : /*
1760 : : * Check PQgetResult() again. In the typical case of a single-command
1761 : : * string, it will return NULL. Otherwise, we'll have other results
1762 : : * to process. We need to do that to check whether this is the last.
1763 : : */
741 peter@eisentraut.org 1764 :CBC 141520 : next_result = PQgetResult(pset.db);
1765 : 141520 : last = (next_result == NULL);
1766 : :
1767 : : /*
1768 : : * Update current timing measure.
1769 : : *
1770 : : * It will include the display of previous results, if any. This
1771 : : * cannot be helped because the server goes on processing further
1772 : : * queries anyway while the previous ones are being displayed. The
1773 : : * parallel execution of the client display hides the server time when
1774 : : * it is shorter.
1775 : : *
1776 : : * With combined queries, timing must be understood as an upper bound
1777 : : * of the time spent processing them.
1778 : : */
692 1779 [ + + ]: 141520 : if (timing)
1780 : : {
741 1781 : 1 : INSTR_TIME_SET_CURRENT(after);
1782 : 1 : INSTR_TIME_SUBTRACT(after, before);
1783 : 1 : *elapsed_msec = INSTR_TIME_GET_MILLISEC(after);
1784 : : }
1785 : :
1786 : : /* this may or may not print something depending on settings */
1787 [ + + ]: 141520 : if (result != NULL)
1788 : : {
1789 : : /*
1790 : : * If results need to be printed into the file specified by \g,
1791 : : * open it, unless we already did. Note that when pset.gfname is
1792 : : * set, the passed-in value of printQueryFout is not used for
1793 : : * tuple output, but it's still used for status output.
1794 : : */
559 tgl@sss.pgh.pa.us 1795 : 141260 : FILE *tuples_fout = printQueryFout;
1796 : :
8 tgl@sss.pgh.pa.us 1797 [ + + ]:GNC 141260 : if (PQresultStatus(result) == PGRES_TUPLES_OK)
1798 : 61799 : success &= SetupGOutput(&gfile_fout, &gfile_is_pipe);
1799 [ + + ]: 141260 : if (gfile_fout)
559 tgl@sss.pgh.pa.us 1800 :CBC 30 : tuples_fout = gfile_fout;
8 tgl@sss.pgh.pa.us 1801 [ + + ]:GNC 141260 : if (success)
559 tgl@sss.pgh.pa.us 1802 :CBC 141167 : success &= PrintQueryResult(result, last, opt,
1803 : : tuples_fout, printQueryFout);
1804 : : }
1805 : :
1806 : : /* set variables from last result, unless dealt with elsewhere */
8 tgl@sss.pgh.pa.us 1807 [ + + + + :GNC 141520 : if (last && !is_watch && !is_chunked_result)
+ + ]
195 michael@paquier.xyz 1808 : 140726 : SetResultVariables(result, success);
1809 : :
741 peter@eisentraut.org 1810 :CBC 141520 : ClearOrSaveResult(result);
1811 : 141520 : result = next_result;
1812 : :
1813 [ - + ]: 141520 : if (cancel_pressed)
1814 : : {
1815 : : /* drop this next result, as well as any others not yet read */
6 tgl@sss.pgh.pa.us 1816 :UBC 0 : ClearOrSaveResult(result);
741 peter@eisentraut.org 1817 : 0 : ClearOrSaveAllResults();
1818 : 0 : break;
1819 : : }
1820 : : }
1821 : :
1822 : : /* close \g file if we opened it */
8 tgl@sss.pgh.pa.us 1823 :GNC 159982 : CloseGOutput(gfile_fout, gfile_is_pipe);
1824 : :
1825 : : /* may need this to recover from conn loss during COPY */
741 peter@eisentraut.org 1826 [ - + ]:CBC 159982 : if (!CheckConnection())
741 peter@eisentraut.org 1827 :UBC 0 : return -1;
1828 : :
229 dgustafsson@postgres 1829 [ + + + + ]:GNC 159982 : if (cancel_pressed || return_early)
1830 : 2 : return 0;
1831 : :
1832 [ + + ]: 159980 : return success ? 1 : -1;
1833 : : }
1834 : :
1835 : :
1836 : : /*
1837 : : * Advance the given char pointer over white space and SQL comments.
1838 : : */
1839 : : static const char *
7146 tgl@sss.pgh.pa.us 1840 :CBC 54 : skip_white_space(const char *query)
1841 : : {
6756 bruce@momjian.us 1842 : 54 : int cnestlevel = 0; /* slash-star comment nest level */
1843 : :
7596 tgl@sss.pgh.pa.us 1844 [ + - ]: 66 : while (*query)
1845 : : {
1042 1846 : 66 : int mblen = PQmblenBounded(query, pset.encoding);
1847 : :
1848 : : /*
1849 : : * Note: we assume the encoding is a superset of ASCII, so that for
1850 : : * example "query[0] == '/'" is meaningful. However, we do NOT assume
1851 : : * that the second and subsequent bytes of a multibyte character
1852 : : * couldn't look like ASCII characters; so it is critical to advance
1853 : : * by mblen, not 1, whenever we haven't exactly identified the
1854 : : * character we are skipping over.
1855 : : */
7596 1856 [ + + ]: 66 : if (isspace((unsigned char) *query))
7146 1857 : 12 : query += mblen;
1858 [ - + - - ]: 54 : else if (query[0] == '/' && query[1] == '*')
1859 : : {
7146 tgl@sss.pgh.pa.us 1860 :UBC 0 : cnestlevel++;
7596 1861 : 0 : query += 2;
1862 : : }
7146 tgl@sss.pgh.pa.us 1863 [ - + - - :CBC 54 : else if (cnestlevel > 0 && query[0] == '*' && query[1] == '/')
- - ]
1864 : : {
7146 tgl@sss.pgh.pa.us 1865 :UBC 0 : cnestlevel--;
1866 : 0 : query += 2;
1867 : : }
7146 tgl@sss.pgh.pa.us 1868 [ + - - + :CBC 54 : else if (cnestlevel == 0 && query[0] == '-' && query[1] == '-')
- - ]
1869 : : {
7596 tgl@sss.pgh.pa.us 1870 :UBC 0 : query += 2;
1871 : :
1872 : : /*
1873 : : * We have to skip to end of line since any slash-star inside the
1874 : : * -- comment does NOT start a slash-star comment.
1875 : : */
1876 [ # # ]: 0 : while (*query)
1877 : : {
7146 1878 [ # # ]: 0 : if (*query == '\n')
1879 : : {
1880 : 0 : query++;
7596 1881 : 0 : break;
1882 : : }
1042 1883 : 0 : query += PQmblenBounded(query, pset.encoding);
1884 : : }
1885 : : }
7146 tgl@sss.pgh.pa.us 1886 [ - + ]:CBC 54 : else if (cnestlevel > 0)
7146 tgl@sss.pgh.pa.us 1887 :UBC 0 : query += mblen;
1888 : : else
7596 tgl@sss.pgh.pa.us 1889 :CBC 54 : break; /* found first token */
1890 : : }
1891 : :
7146 1892 : 54 : return query;
1893 : : }
1894 : :
1895 : :
1896 : : /*
1897 : : * Check whether a command is one of those for which we should NOT start
1898 : : * a new transaction block (ie, send a preceding BEGIN).
1899 : : *
1900 : : * These include the transaction control statements themselves, plus
1901 : : * certain statements that the backend disallows inside transaction blocks.
1902 : : */
1903 : : static bool
1904 : 42 : command_no_begin(const char *query)
1905 : : {
1906 : : int wordlen;
1907 : :
1908 : : /*
1909 : : * First we must advance over any whitespace and comments.
1910 : : */
1911 : 42 : query = skip_white_space(query);
1912 : :
1913 : : /*
1914 : : * Check word length (since "beginx" is not "begin").
1915 : : */
7596 1916 : 42 : wordlen = 0;
1917 [ + + ]: 282 : while (isalpha((unsigned char) query[wordlen]))
1042 1918 : 240 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
1919 : :
1920 : : /*
1921 : : * Transaction control commands. These should include every keyword that
1922 : : * gives rise to a TransactionStmt in the backend grammar, except for the
1923 : : * savepoint-related commands.
1924 : : *
1925 : : * (We assume that START must be START TRANSACTION, since there is
1926 : : * presently no other "START foo" command.)
1927 : : */
7146 1928 [ + + - + ]: 42 : if (wordlen == 5 && pg_strncasecmp(query, "abort", 5) == 0)
7146 tgl@sss.pgh.pa.us 1929 :UBC 0 : return true;
7282 tgl@sss.pgh.pa.us 1930 [ + + + - ]:CBC 42 : if (wordlen == 5 && pg_strncasecmp(query, "begin", 5) == 0)
7596 1931 : 6 : return true;
7146 1932 [ - + - - ]: 36 : if (wordlen == 5 && pg_strncasecmp(query, "start", 5) == 0)
7146 tgl@sss.pgh.pa.us 1933 :UBC 0 : return true;
7282 tgl@sss.pgh.pa.us 1934 [ + + - + ]:CBC 36 : if (wordlen == 6 && pg_strncasecmp(query, "commit", 6) == 0)
7596 tgl@sss.pgh.pa.us 1935 :UBC 0 : return true;
7146 tgl@sss.pgh.pa.us 1936 [ - + - - ]:CBC 36 : if (wordlen == 3 && pg_strncasecmp(query, "end", 3) == 0)
7596 tgl@sss.pgh.pa.us 1937 :UBC 0 : return true;
7146 tgl@sss.pgh.pa.us 1938 [ - + - - ]:CBC 36 : if (wordlen == 8 && pg_strncasecmp(query, "rollback", 8) == 0)
7596 tgl@sss.pgh.pa.us 1939 :UBC 0 : return true;
6876 tgl@sss.pgh.pa.us 1940 [ - + - - ]:CBC 36 : if (wordlen == 7 && pg_strncasecmp(query, "prepare", 7) == 0)
1941 : : {
1942 : : /* PREPARE TRANSACTION is a TC command, PREPARE foo is not */
6876 tgl@sss.pgh.pa.us 1943 :UBC 0 : query += wordlen;
1944 : :
1945 : 0 : query = skip_white_space(query);
1946 : :
1947 : 0 : wordlen = 0;
1948 [ # # ]: 0 : while (isalpha((unsigned char) query[wordlen]))
1042 1949 : 0 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
1950 : :
6876 1951 [ # # # # ]: 0 : if (wordlen == 11 && pg_strncasecmp(query, "transaction", 11) == 0)
1952 : 0 : return true;
1953 : 0 : return false;
1954 : : }
1955 : :
1956 : : /*
1957 : : * Commands not allowed within transactions. The statements checked for
1958 : : * here should be exactly those that call PreventInTransactionBlock() in
1959 : : * the backend.
1960 : : */
7146 tgl@sss.pgh.pa.us 1961 [ + + - + ]:CBC 36 : if (wordlen == 6 && pg_strncasecmp(query, "vacuum", 6) == 0)
7596 tgl@sss.pgh.pa.us 1962 :UBC 0 : return true;
7146 tgl@sss.pgh.pa.us 1963 [ - + - - ]:CBC 36 : if (wordlen == 7 && pg_strncasecmp(query, "cluster", 7) == 0)
1964 : : {
1965 : : /* CLUSTER with any arguments is allowed in transactions */
6442 tgl@sss.pgh.pa.us 1966 :UBC 0 : query += wordlen;
1967 : :
1968 : 0 : query = skip_white_space(query);
1969 : :
1970 [ # # ]: 0 : if (isalpha((unsigned char) query[0]))
1971 : 0 : return false; /* has additional words */
1972 : 0 : return true; /* it's CLUSTER without arguments */
1973 : : }
1974 : :
6442 tgl@sss.pgh.pa.us 1975 [ + + + + ]:CBC 36 : if (wordlen == 6 && pg_strncasecmp(query, "create", 6) == 0)
1976 : : {
1977 : 6 : query += wordlen;
1978 : :
1979 : 6 : query = skip_white_space(query);
1980 : :
1981 : 6 : wordlen = 0;
1982 [ + + ]: 36 : while (isalpha((unsigned char) query[wordlen]))
1042 1983 : 30 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
1984 : :
6442 1985 [ - + - - ]: 6 : if (wordlen == 8 && pg_strncasecmp(query, "database", 8) == 0)
6442 tgl@sss.pgh.pa.us 1986 :UBC 0 : return true;
6442 tgl@sss.pgh.pa.us 1987 [ - + - - ]:CBC 6 : if (wordlen == 10 && pg_strncasecmp(query, "tablespace", 10) == 0)
6442 tgl@sss.pgh.pa.us 1988 :UBC 0 : return true;
1989 : :
1990 : : /* CREATE [UNIQUE] INDEX CONCURRENTLY isn't allowed in xacts */
6442 tgl@sss.pgh.pa.us 1991 [ - + - - ]:CBC 6 : if (wordlen == 6 && pg_strncasecmp(query, "unique", 6) == 0)
1992 : : {
6442 tgl@sss.pgh.pa.us 1993 :UBC 0 : query += wordlen;
1994 : :
1995 : 0 : query = skip_white_space(query);
1996 : :
1997 : 0 : wordlen = 0;
1998 [ # # ]: 0 : while (isalpha((unsigned char) query[wordlen]))
1042 1999 : 0 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
2000 : : }
2001 : :
6442 tgl@sss.pgh.pa.us 2002 [ + - - + ]:CBC 6 : if (wordlen == 5 && pg_strncasecmp(query, "index", 5) == 0)
2003 : : {
6442 tgl@sss.pgh.pa.us 2004 :UBC 0 : query += wordlen;
2005 : :
2006 : 0 : query = skip_white_space(query);
2007 : :
2008 : 0 : wordlen = 0;
2009 [ # # ]: 0 : while (isalpha((unsigned char) query[wordlen]))
1042 2010 : 0 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
2011 : :
6442 2012 [ # # # # ]: 0 : if (wordlen == 12 && pg_strncasecmp(query, "concurrently", 12) == 0)
2013 : 0 : return true;
2014 : : }
2015 : :
6442 tgl@sss.pgh.pa.us 2016 :CBC 6 : return false;
2017 : : }
2018 : :
3574 fujii@postgresql.org 2019 [ - + - - ]: 30 : if (wordlen == 5 && pg_strncasecmp(query, "alter", 5) == 0)
2020 : : {
3574 fujii@postgresql.org 2021 :UBC 0 : query += wordlen;
2022 : :
2023 : 0 : query = skip_white_space(query);
2024 : :
2025 : 0 : wordlen = 0;
2026 [ # # ]: 0 : while (isalpha((unsigned char) query[wordlen]))
1042 tgl@sss.pgh.pa.us 2027 : 0 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
2028 : :
2029 : : /* ALTER SYSTEM isn't allowed in xacts */
3574 fujii@postgresql.org 2030 [ # # # # ]: 0 : if (wordlen == 6 && pg_strncasecmp(query, "system", 6) == 0)
2031 : 0 : return true;
2032 : :
2033 : 0 : return false;
2034 : : }
2035 : :
2036 : : /*
2037 : : * Note: these tests will match DROP SYSTEM and REINDEX TABLESPACE, which
2038 : : * aren't really valid commands so we don't care much. The other four
2039 : : * possible matches are correct.
2040 : : */
6442 tgl@sss.pgh.pa.us 2041 [ + + - + :CBC 30 : if ((wordlen == 4 && pg_strncasecmp(query, "drop", 4) == 0) ||
- + ]
7146 tgl@sss.pgh.pa.us 2042 [ # # ]:UBC 0 : (wordlen == 7 && pg_strncasecmp(query, "reindex", 7) == 0))
2043 : : {
7146 tgl@sss.pgh.pa.us 2044 :CBC 3 : query += wordlen;
2045 : :
2046 : 3 : query = skip_white_space(query);
2047 : :
2048 : 3 : wordlen = 0;
2049 [ + + ]: 18 : while (isalpha((unsigned char) query[wordlen]))
1042 2050 : 15 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
2051 : :
7146 2052 [ - + - - ]: 3 : if (wordlen == 8 && pg_strncasecmp(query, "database", 8) == 0)
7146 tgl@sss.pgh.pa.us 2053 :UBC 0 : return true;
6871 tgl@sss.pgh.pa.us 2054 [ - + - - ]:CBC 3 : if (wordlen == 6 && pg_strncasecmp(query, "system", 6) == 0)
6871 tgl@sss.pgh.pa.us 2055 :UBC 0 : return true;
7146 tgl@sss.pgh.pa.us 2056 [ - + - - ]:CBC 3 : if (wordlen == 10 && pg_strncasecmp(query, "tablespace", 10) == 0)
7146 tgl@sss.pgh.pa.us 2057 :UBC 0 : return true;
1843 peter@eisentraut.org 2058 [ + - + - :CBC 6 : if (wordlen == 5 && (pg_strncasecmp(query, "index", 5) == 0 ||
+ - ]
2059 : 3 : pg_strncasecmp(query, "table", 5) == 0))
2060 : : {
2061 : 3 : query += wordlen;
2062 : 3 : query = skip_white_space(query);
2063 : 3 : wordlen = 0;
2064 [ + + ]: 12 : while (isalpha((unsigned char) query[wordlen]))
1042 tgl@sss.pgh.pa.us 2065 : 9 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
2066 : :
2067 : : /*
2068 : : * REINDEX [ TABLE | INDEX ] CONCURRENTLY are not allowed in
2069 : : * xacts.
2070 : : */
1843 peter@eisentraut.org 2071 [ - + - - ]: 3 : if (wordlen == 12 && pg_strncasecmp(query, "concurrently", 12) == 0)
1843 peter@eisentraut.org 2072 :UBC 0 : return true;
2073 : : }
2074 : :
2075 : : /* DROP INDEX CONCURRENTLY isn't allowed in xacts */
3314 bruce@momjian.us 2076 [ - + - - ]:CBC 3 : if (wordlen == 5 && pg_strncasecmp(query, "index", 5) == 0)
2077 : : {
3314 bruce@momjian.us 2078 :UBC 0 : query += wordlen;
2079 : :
2080 : 0 : query = skip_white_space(query);
2081 : :
2082 : 0 : wordlen = 0;
2083 [ # # ]: 0 : while (isalpha((unsigned char) query[wordlen]))
1042 tgl@sss.pgh.pa.us 2084 : 0 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
2085 : :
3314 bruce@momjian.us 2086 [ # # # # ]: 0 : if (wordlen == 12 && pg_strncasecmp(query, "concurrently", 12) == 0)
2087 : 0 : return true;
2088 : :
2089 : 0 : return false;
2090 : : }
2091 : :
4947 tgl@sss.pgh.pa.us 2092 :CBC 3 : return false;
2093 : : }
2094 : :
2095 : : /* DISCARD ALL isn't allowed in xacts, but other variants are allowed. */
itagaki.takahiro@gma 2096 [ - + - - ]: 27 : if (wordlen == 7 && pg_strncasecmp(query, "discard", 7) == 0)
2097 : : {
4947 itagaki.takahiro@gma 2098 :UBC 0 : query += wordlen;
2099 : :
2100 : 0 : query = skip_white_space(query);
2101 : :
2102 : 0 : wordlen = 0;
2103 [ # # ]: 0 : while (isalpha((unsigned char) query[wordlen]))
1042 tgl@sss.pgh.pa.us 2104 : 0 : wordlen += PQmblenBounded(&query[wordlen], pset.encoding);
2105 : :
4947 itagaki.takahiro@gma 2106 [ # # # # ]: 0 : if (wordlen == 3 && pg_strncasecmp(query, "all", 3) == 0)
2107 : 0 : return true;
tgl@sss.pgh.pa.us 2108 : 0 : return false;
2109 : : }
2110 : :
7596 tgl@sss.pgh.pa.us 2111 :CBC 27 : return false;
2112 : : }
2113 : :
2114 : :
2115 : : /*
2116 : : * Test if the current user is a database superuser.
2117 : : */
2118 : : bool
2119 : 54 : is_superuser(void)
2120 : : {
2121 : : const char *val;
2122 : :
2123 [ - + ]: 54 : if (!pset.db)
7596 tgl@sss.pgh.pa.us 2124 :UBC 0 : return false;
2125 : :
7596 tgl@sss.pgh.pa.us 2126 :CBC 54 : val = PQparameterStatus(pset.db, "is_superuser");
2127 : :
2128 [ + - + - ]: 54 : if (val && strcmp(val, "on") == 0)
2129 : 54 : return true;
2130 : :
7596 tgl@sss.pgh.pa.us 2131 :UBC 0 : return false;
2132 : : }
2133 : :
2134 : :
2135 : : /*
2136 : : * Test if the current session uses standard string literals.
2137 : : */
2138 : : bool
6614 bruce@momjian.us 2139 :CBC 304127 : standard_strings(void)
2140 : : {
2141 : : const char *val;
2142 : :
2143 [ - + ]: 304127 : if (!pset.db)
6614 bruce@momjian.us 2144 :UBC 0 : return false;
2145 : :
6614 bruce@momjian.us 2146 :CBC 304127 : val = PQparameterStatus(pset.db, "standard_conforming_strings");
2147 : :
2148 [ + - + + ]: 304127 : if (val && strcmp(val, "on") == 0)
2149 : 304031 : return true;
2150 : :
2151 : 96 : return false;
2152 : : }
2153 : :
2154 : :
2155 : : /*
2156 : : * Return the session user of the current connection.
2157 : : */
2158 : : const char *
7529 peter_e@gmx.net 2159 :UBC 0 : session_username(void)
2160 : : {
2161 : : const char *val;
2162 : :
2163 [ # # ]: 0 : if (!pset.db)
2164 : 0 : return NULL;
2165 : :
2166 : 0 : val = PQparameterStatus(pset.db, "session_authorization");
2167 [ # # ]: 0 : if (val)
2168 : 0 : return val;
2169 : : else
2170 : 0 : return PQuser(pset.db);
2171 : : }
2172 : :
2173 : :
2174 : : /* expand_tilde
2175 : : *
2176 : : * substitute '~' with HOME or '~username' with username's home dir
2177 : : *
2178 : : */
2179 : : void
7401 bruce@momjian.us 2180 :CBC 86 : expand_tilde(char **filename)
2181 : : {
2182 [ + - + + ]: 86 : if (!filename || !(*filename))
4028 2183 : 6 : return;
2184 : :
2185 : : /*
2186 : : * WIN32 doesn't use tilde expansion for file names. Also, it uses tilde
2187 : : * for short versions of long file names, though the tilde is usually
2188 : : * toward the end, not at the beginning.
2189 : : */
2190 : : #ifndef WIN32
2191 : :
2192 : : /* try tilde expansion */
7401 2193 [ - + ]: 80 : if (**filename == '~')
2194 : : {
2195 : : char *fn;
2196 : : char oldp,
2197 : : *p;
2198 : : struct passwd *pw;
2199 : : char home[MAXPGPATH];
2200 : :
7401 bruce@momjian.us 2201 :UBC 0 : fn = *filename;
7179 2202 : 0 : *home = '\0';
2203 : :
7401 2204 : 0 : p = fn + 1;
2205 [ # # # # ]: 0 : while (*p != '/' && *p != '\0')
2206 : 0 : p++;
2207 : :
2208 : 0 : oldp = *p;
2209 : 0 : *p = '\0';
2210 : :
2211 [ # # ]: 0 : if (*(fn + 1) == '\0')
6883 2212 : 0 : get_home_path(home); /* ~ or ~/ only */
7401 2213 [ # # ]: 0 : else if ((pw = getpwnam(fn + 1)) != NULL)
5995 2214 : 0 : strlcpy(home, pw->pw_dir, sizeof(home)); /* ~user */
2215 : :
7401 2216 : 0 : *p = oldp;
7179 2217 [ # # ]: 0 : if (strlen(home) != 0)
2218 : : {
2219 : : char *newfn;
2220 : :
3827 tgl@sss.pgh.pa.us 2221 : 0 : newfn = psprintf("%s%s", home, p);
7401 bruce@momjian.us 2222 : 0 : free(fn);
2223 : 0 : *filename = newfn;
2224 : : }
2225 : : }
2226 : : #endif
2227 : : }
2228 : :
2229 : : /*
2230 : : * Checks if connection string starts with either of the valid URI prefix
2231 : : * designators.
2232 : : *
2233 : : * Returns the URI prefix length, 0 if the string doesn't contain a URI prefix.
2234 : : *
2235 : : * XXX This is a duplicate of the eponymous libpq function.
2236 : : */
2237 : : static int
3300 alvherre@alvh.no-ip. 2238 :CBC 14 : uri_prefix_length(const char *connstr)
2239 : : {
2240 : : /* The connection URI must start with either of the following designators: */
2241 : : static const char uri_designator[] = "postgresql://";
2242 : : static const char short_uri_designator[] = "postgres://";
2243 : :
2244 [ - + ]: 14 : if (strncmp(connstr, uri_designator,
2245 : : sizeof(uri_designator) - 1) == 0)
3300 alvherre@alvh.no-ip. 2246 :UBC 0 : return sizeof(uri_designator) - 1;
2247 : :
3300 alvherre@alvh.no-ip. 2248 [ - + ]:CBC 14 : if (strncmp(connstr, short_uri_designator,
2249 : : sizeof(short_uri_designator) - 1) == 0)
3300 alvherre@alvh.no-ip. 2250 :UBC 0 : return sizeof(short_uri_designator) - 1;
2251 : :
3300 alvherre@alvh.no-ip. 2252 :CBC 14 : return 0;
2253 : : }
2254 : :
2255 : : /*
2256 : : * Recognized connection string either starts with a valid URI prefix or
2257 : : * contains a "=" in it.
2258 : : *
2259 : : * Must be consistent with parse_connection_string: anything for which this
2260 : : * returns true should at least look like it's parseable by that routine.
2261 : : *
2262 : : * XXX This is a duplicate of the eponymous libpq function.
2263 : : */
2264 : : bool
2265 : 14 : recognized_connection_string(const char *connstr)
2266 : : {
2267 [ + - + + ]: 14 : return uri_prefix_length(connstr) != 0 || strchr(connstr, '=') != NULL;
2268 : : }
|
