PostgreSQL Source Code  git master
pg_backup_db.c File Reference
#include "postgres_fe.h"
#include <unistd.h>
#include <ctype.h>
#include "common/connect.h"
#include "common/string.h"
#include "dumputils.h"
#include "fe_utils/string_utils.h"
#include "parallel.h"
#include "pg_backup_archiver.h"
#include "pg_backup_db.h"
#include "pg_backup_utils.h"
Include dependency graph for pg_backup_db.c:

Go to the source code of this file.

Functions

static void _check_database_version (ArchiveHandle *AH)
 
static void notice_processor (void *arg, const char *message)
 
void ReconnectToServer (ArchiveHandle *AH, const char *dbname)
 
void ConnectDatabase (Archive *AHX, const ConnParams *cparams, bool isReconnect)
 
void DisconnectDatabase (Archive *AHX)
 
PGconnGetConnection (Archive *AHX)
 
static void die_on_query_failure (ArchiveHandle *AH, const char *query)
 
void ExecuteSqlStatement (Archive *AHX, const char *query)
 
PGresultExecuteSqlQuery (Archive *AHX, const char *query, ExecStatusType status)
 
PGresultExecuteSqlQueryForSingleRow (Archive *fout, const char *query)
 
static void ExecuteSqlCommand (ArchiveHandle *AH, const char *qry, const char *desc)
 
static void ExecuteSimpleCommands (ArchiveHandle *AH, const char *buf, size_t bufLen)
 
int ExecuteSqlCommandBuf (Archive *AHX, const char *buf, size_t bufLen)
 
void EndDBCopyMode (Archive *AHX, const char *tocEntryTag)
 
void StartTransaction (Archive *AHX)
 
void CommitTransaction (Archive *AHX)
 
void DropBlobIfExists (ArchiveHandle *AH, Oid oid)
 

Function Documentation

◆ _check_database_version()

static void _check_database_version ( ArchiveHandle AH)
static

Definition at line 33 of file pg_backup_db.c.

References _archiveHandle::archiveRemoteVersion, _archiveHandle::connection, ExecuteSqlQueryForSingleRow(), fatal, Archive::isStandby, Archive::maxRemoteVersion, pg_log_error, pg_strdup(), PQclear(), PQgetvalue(), PQparameterStatus(), PQserverVersion(), progname, _archiveHandle::public, Archive::remoteVersion, and Archive::remoteVersionStr.

Referenced by ConnectDatabase().

34 {
35  const char *remoteversion_str;
36  int remoteversion;
37  PGresult *res;
38 
39  remoteversion_str = PQparameterStatus(AH->connection, "server_version");
40  remoteversion = PQserverVersion(AH->connection);
41  if (remoteversion == 0 || !remoteversion_str)
42  fatal("could not get server_version from libpq");
43 
44  AH->public.remoteVersionStr = pg_strdup(remoteversion_str);
45  AH->public.remoteVersion = remoteversion;
46  if (!AH->archiveRemoteVersion)
48 
49  if (remoteversion != PG_VERSION_NUM
50  && (remoteversion < AH->public.minRemoteVersion ||
51  remoteversion > AH->public.maxRemoteVersion))
52  {
53  pg_log_error("server version: %s; %s version: %s",
54  remoteversion_str, progname, PG_VERSION);
55  fatal("aborting because of server version mismatch");
56  }
57 
58  /*
59  * When running against 9.0 or later, check if we are in recovery mode,
60  * which means we are on a hot standby.
61  */
62  if (remoteversion >= 90000)
63  {
64  res = ExecuteSqlQueryForSingleRow((Archive *) AH, "SELECT pg_catalog.pg_is_in_recovery()");
65 
66  AH->public.isStandby = (strcmp(PQgetvalue(res, 0, 0), "t") == 0);
67  PQclear(res);
68  }
69  else
70  AH->public.isStandby = false;
71 }
char * PQgetvalue(const PGresult *res, int tup_num, int field_num)
Definition: fe-exec.c:3163
const char * PQparameterStatus(const PGconn *conn, const char *paramName)
Definition: fe-connect.c:6634
#define pg_log_error(...)
Definition: logging.h:80
int PQserverVersion(const PGconn *conn)
Definition: fe-connect.c:6659
const char * progname
Definition: pg_standby.c:36
int maxRemoteVersion
Definition: pg_backup.h:200
PGresult * ExecuteSqlQueryForSingleRow(Archive *fout, const char *query)
Definition: pg_backup_db.c:311
bool isStandby
Definition: pg_backup.h:197
char * pg_strdup(const char *in)
Definition: fe_memutils.c:85
void PQclear(PGresult *res)
Definition: fe-exec.c:694
#define fatal(...)
char * remoteVersionStr
Definition: pg_backup.h:195
int remoteVersion
Definition: pg_backup.h:196

◆ CommitTransaction()

void CommitTransaction ( Archive AHX)

Definition at line 543 of file pg_backup_db.c.

References ExecuteSqlCommand().

544 {
545  ArchiveHandle *AH = (ArchiveHandle *) AHX;
546 
547  ExecuteSqlCommand(AH, "COMMIT", "could not commit database transaction");
548 }
static void ExecuteSqlCommand(ArchiveHandle *AH, const char *qry, const char *desc)
Definition: pg_backup_db.c:334

◆ ConnectDatabase()

void ConnectDatabase ( Archive AHX,
const ConnParams cparams,
bool  isReconnect 
)

Definition at line 114 of file pg_backup_db.c.

References _check_database_version(), ALWAYS_SECURE_SEARCH_PATH_SQL, Assert, _archiveHandle::connection, CONNECTION_BAD, _connParams::dbname, ExecuteSqlQueryForSingleRow(), fatal, free, i, lengthof, notice_processor(), _connParams::override_dbname, password, pg_strdup(), _connParams::pghost, _connParams::pgport, PQclear(), PQconnectdbParams(), PQconnectionNeedsPassword(), PQconnectionUsedPassword(), PQdb(), PQerrorMessage(), PQfinish(), PQpass(), PQsetNoticeProcessor(), PQstatus(), progname, _connParams::promptPassword, _archiveHandle::savedPassword, set_archive_cancel_info(), simple_prompt(), TRI_NO, TRI_YES, _connParams::username, and values.

Referenced by CloneArchive(), main(), ReconnectToServer(), restore_toc_entries_postfork(), and RestoreArchive().

117 {
118  ArchiveHandle *AH = (ArchiveHandle *) AHX;
119  trivalue prompt_password;
120  char *password;
121  bool new_pass;
122 
123  if (AH->connection)
124  fatal("already connected to a database");
125 
126  /* Never prompt for a password during a reconnection */
127  prompt_password = isReconnect ? TRI_NO : cparams->promptPassword;
128 
129  password = AH->savedPassword;
130 
131  if (prompt_password == TRI_YES && password == NULL)
132  password = simple_prompt("Password: ", false);
133 
134  /*
135  * Start the connection. Loop until we have a password if requested by
136  * backend.
137  */
138  do
139  {
140  const char *keywords[8];
141  const char *values[8];
142  int i = 0;
143 
144  /*
145  * If dbname is a connstring, its entries can override the other
146  * values obtained from cparams; but in turn, override_dbname can
147  * override the dbname component of it.
148  */
149  keywords[i] = "host";
150  values[i++] = cparams->pghost;
151  keywords[i] = "port";
152  values[i++] = cparams->pgport;
153  keywords[i] = "user";
154  values[i++] = cparams->username;
155  keywords[i] = "password";
156  values[i++] = password;
157  keywords[i] = "dbname";
158  values[i++] = cparams->dbname;
159  if (cparams->override_dbname)
160  {
161  keywords[i] = "dbname";
162  values[i++] = cparams->override_dbname;
163  }
164  keywords[i] = "fallback_application_name";
165  values[i++] = progname;
166  keywords[i] = NULL;
167  values[i++] = NULL;
168  Assert(i <= lengthof(keywords));
169 
170  new_pass = false;
171  AH->connection = PQconnectdbParams(keywords, values, true);
172 
173  if (!AH->connection)
174  fatal("could not connect to database");
175 
176  if (PQstatus(AH->connection) == CONNECTION_BAD &&
178  password == NULL &&
179  prompt_password != TRI_NO)
180  {
181  PQfinish(AH->connection);
182  password = simple_prompt("Password: ", false);
183  new_pass = true;
184  }
185  } while (new_pass);
186 
187  /* check to see that the backend connection was successfully made */
188  if (PQstatus(AH->connection) == CONNECTION_BAD)
189  {
190  if (isReconnect)
191  fatal("reconnection to database \"%s\" failed: %s",
192  PQdb(AH->connection) ? PQdb(AH->connection) : "",
194  else
195  fatal("connection to database \"%s\" failed: %s",
196  PQdb(AH->connection) ? PQdb(AH->connection) : "",
198  }
199 
200  /* Start strict; later phases may override this. */
203 
204  if (password && password != AH->savedPassword)
205  free(password);
206 
207  /*
208  * We want to remember connection's actual password, whether or not we got
209  * it by prompting. So we don't just store the password variable.
210  */
212  {
213  if (AH->savedPassword)
214  free(AH->savedPassword);
216  }
217 
218  /* check for version mismatch */
220 
222 
223  /* arrange for SIGINT to issue a query cancel on this connection */
225 }
char * PQerrorMessage(const PGconn *conn)
Definition: fe-connect.c:6669
PQnoticeProcessor PQsetNoticeProcessor(PGconn *conn, PQnoticeProcessor proc, void *arg)
Definition: fe-connect.c:6845
char * pgport
Definition: pg_backup.h:66
void PQfinish(PGconn *conn)
Definition: fe-connect.c:4174
char * simple_prompt(const char *prompt, bool echo)
Definition: sprompt.c:38
#define lengthof(array)
Definition: c.h:730
const char * progname
Definition: pg_standby.c:36
PGconn * PQconnectdbParams(const char *const *keywords, const char *const *values, int expand_dbname)
Definition: fe-connect.c:647
trivalue promptPassword
Definition: pg_backup.h:69
char * dbname
Definition: pg_backup.h:65
PGresult * ExecuteSqlQueryForSingleRow(Archive *fout, const char *query)
Definition: pg_backup_db.c:311
char * pg_strdup(const char *in)
Definition: fe_memutils.c:85
char * PQpass(const PGconn *conn)
Definition: fe-connect.c:6535
static char * password
Definition: streamutil.c:53
static void notice_processor(void *arg, const char *message)
Definition: pg_backup_db.c:269
char * override_dbname
Definition: pg_backup.h:72
char * pghost
Definition: pg_backup.h:67
char * username
Definition: pg_backup.h:68
trivalue
Definition: vacuumlo.c:34
void PQclear(PGresult *res)
Definition: fe-exec.c:694
#define free(a)
Definition: header.h:65
char * PQdb(const PGconn *conn)
Definition: fe-connect.c:6519
#define Assert(condition)
Definition: c.h:800
#define fatal(...)
#define ALWAYS_SECURE_SEARCH_PATH_SQL
Definition: connect.h:25
static Datum values[MAXATTR]
Definition: bootstrap.c:165
static void _check_database_version(ArchiveHandle *AH)
Definition: pg_backup_db.c:33
int PQconnectionUsedPassword(const PGconn *conn)
Definition: fe-connect.c:6718
int PQconnectionNeedsPassword(const PGconn *conn)
Definition: fe-connect.c:6703
void set_archive_cancel_info(ArchiveHandle *AH, PGconn *conn)
Definition: parallel.c:735
int i
ConnStatusType PQstatus(const PGconn *conn)
Definition: fe-connect.c:6616

◆ die_on_query_failure()

static void die_on_query_failure ( ArchiveHandle AH,
const char *  query 
)
static

Definition at line 276 of file pg_backup_db.c.

References _archiveHandle::connection, fatal, pg_log_error, and PQerrorMessage().

Referenced by ExecuteSqlQuery(), and ExecuteSqlStatement().

277 {
278  pg_log_error("query failed: %s",
280  fatal("query was: %s", query);
281 }
char * PQerrorMessage(const PGconn *conn)
Definition: fe-connect.c:6669
#define pg_log_error(...)
Definition: logging.h:80
#define fatal(...)

◆ DisconnectDatabase()

void DisconnectDatabase ( Archive AHX)

Definition at line 232 of file pg_backup_db.c.

References _archiveHandle::connCancel, _archiveHandle::connection, PQcancel(), PQfinish(), PQTRANS_ACTIVE, PQtransactionStatus(), and set_archive_cancel_info().

Referenced by archive_close_connection(), restore_toc_entries_prefork(), RestoreArchive(), and RunWorker().

233 {
234  ArchiveHandle *AH = (ArchiveHandle *) AHX;
235  char errbuf[1];
236 
237  if (!AH->connection)
238  return;
239 
240  if (AH->connCancel)
241  {
242  /*
243  * If we have an active query, send a cancel before closing, ignoring
244  * any errors. This is of no use for a normal exit, but might be
245  * helpful during fatal().
246  */
248  (void) PQcancel(AH->connCancel, errbuf, sizeof(errbuf));
249 
250  /*
251  * Prevent signal handler from sending a cancel after this.
252  */
253  set_archive_cancel_info(AH, NULL);
254  }
255 
256  PQfinish(AH->connection);
257  AH->connection = NULL;
258 }
void PQfinish(PGconn *conn)
Definition: fe-connect.c:4174
PGcancel *volatile connCancel
PGTransactionStatusType PQtransactionStatus(const PGconn *conn)
Definition: fe-connect.c:6624
int PQcancel(PGcancel *cancel, char *errbuf, int errbufsize)
Definition: fe-connect.c:4450
void set_archive_cancel_info(ArchiveHandle *AH, PGconn *conn)
Definition: parallel.c:735

◆ DropBlobIfExists()

void DropBlobIfExists ( ArchiveHandle AH,
Oid  oid 
)

Definition at line 551 of file pg_backup_db.c.

References ahprintf(), _archiveHandle::connection, and PQserverVersion().

Referenced by _StartBlob(), RestoreArchive(), and StartRestoreBlob().

552 {
553  /*
554  * If we are not restoring to a direct database connection, we have to
555  * guess about how to detect whether the blob exists. Assume new-style.
556  */
557  if (AH->connection == NULL ||
558  PQserverVersion(AH->connection) >= 90000)
559  {
560  ahprintf(AH,
561  "SELECT pg_catalog.lo_unlink(oid) "
562  "FROM pg_catalog.pg_largeobject_metadata "
563  "WHERE oid = '%u';\n",
564  oid);
565  }
566  else
567  {
568  /* Restoring to pre-9.0 server, so do it the old way */
569  ahprintf(AH,
570  "SELECT CASE WHEN EXISTS("
571  "SELECT 1 FROM pg_catalog.pg_largeobject WHERE loid = '%u'"
572  ") THEN pg_catalog.lo_unlink('%u') END;\n",
573  oid, oid);
574  }
575 }
int PQserverVersion(const PGconn *conn)
Definition: fe-connect.c:6659
int ahprintf(ArchiveHandle *AH, const char *fmt,...)

◆ EndDBCopyMode()

void EndDBCopyMode ( Archive AHX,
const char *  tocEntryTag 
)

Definition at line 506 of file pg_backup_db.c.

References _archiveHandle::connection, fatal, pg_log_warning, _archiveHandle::pgCopyIn, PGRES_COMMAND_OK, PQclear(), PQerrorMessage(), PQgetResult(), PQputCopyEnd(), PQresultStatus(), and warn_or_exit_horribly().

Referenced by restore_toc_entry().

507 {
508  ArchiveHandle *AH = (ArchiveHandle *) AHX;
509 
510  if (AH->pgCopyIn)
511  {
512  PGresult *res;
513 
514  if (PQputCopyEnd(AH->connection, NULL) <= 0)
515  fatal("error returned by PQputCopyEnd: %s",
517 
518  /* Check command status and return to normal libpq state */
519  res = PQgetResult(AH->connection);
520  if (PQresultStatus(res) != PGRES_COMMAND_OK)
521  warn_or_exit_horribly(AH, "COPY failed for table \"%s\": %s",
522  tocEntryTag, PQerrorMessage(AH->connection));
523  PQclear(res);
524 
525  /* Do this to ensure we've pumped libpq back to idle state */
526  if (PQgetResult(AH->connection) != NULL)
527  pg_log_warning("unexpected extra results during COPY of table \"%s\"",
528  tocEntryTag);
529 
530  AH->pgCopyIn = false;
531  }
532 }
char * PQerrorMessage(const PGconn *conn)
Definition: fe-connect.c:6669
void warn_or_exit_horribly(ArchiveHandle *AH, const char *fmt,...)
int PQputCopyEnd(PGconn *conn, const char *errormsg)
Definition: fe-exec.c:2384
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
void PQclear(PGresult *res)
Definition: fe-exec.c:694
#define fatal(...)
#define pg_log_warning(...)
Definition: pgfnames.c:24
PGresult * PQgetResult(PGconn *conn)
Definition: fe-exec.c:1778

◆ ExecuteSimpleCommands()

static void ExecuteSimpleCommands ( ArchiveHandle AH,
const char *  buf,
size_t  bufLen 
)
static

Definition at line 386 of file pg_backup_db.c.

References appendPQExpBufferChar(), sqlparseInfo::backSlash, buf, createPQExpBuffer(), sqlparseInfo::curCmd, PQExpBufferData::data, ExecuteSqlCommand(), PQExpBufferData::len, _archiveHandle::public, resetPQExpBuffer(), SQL_IN_DOUBLE_QUOTE, SQL_IN_SINGLE_QUOTE, SQL_SCAN, _archiveHandle::sqlparse, sqlparseInfo::state, and Archive::std_strings.

Referenced by ExecuteSqlCommandBuf().

387 {
388  const char *qry = buf;
389  const char *eos = buf + bufLen;
390 
391  /* initialize command buffer if first time through */
392  if (AH->sqlparse.curCmd == NULL)
394 
395  for (; qry < eos; qry++)
396  {
397  char ch = *qry;
398 
399  /* For neatness, we skip any newlines between commands */
400  if (!(ch == '\n' && AH->sqlparse.curCmd->len == 0))
402 
403  switch (AH->sqlparse.state)
404  {
405  case SQL_SCAN: /* Default state == 0, set in _allocAH */
406  if (ch == ';')
407  {
408  /*
409  * We've found the end of a statement. Send it and reset
410  * the buffer.
411  */
413  "could not execute query");
415  }
416  else if (ch == '\'')
417  {
419  AH->sqlparse.backSlash = false;
420  }
421  else if (ch == '"')
422  {
424  }
425  break;
426 
427  case SQL_IN_SINGLE_QUOTE:
428  /* We needn't handle '' specially */
429  if (ch == '\'' && !AH->sqlparse.backSlash)
430  AH->sqlparse.state = SQL_SCAN;
431  else if (ch == '\\' && !AH->public.std_strings)
433  else
434  AH->sqlparse.backSlash = false;
435  break;
436 
437  case SQL_IN_DOUBLE_QUOTE:
438  /* We needn't handle "" specially */
439  if (ch == '"')
440  AH->sqlparse.state = SQL_SCAN;
441  break;
442  }
443  }
444 }
PQExpBuffer curCmd
sqlparseState state
static void ExecuteSqlCommand(ArchiveHandle *AH, const char *qry, const char *desc)
Definition: pg_backup_db.c:334
static char * buf
Definition: pg_test_fsync.c:68
sqlparseInfo sqlparse
PQExpBuffer createPQExpBuffer(void)
Definition: pqexpbuffer.c:74
void appendPQExpBufferChar(PQExpBuffer str, char ch)
Definition: pqexpbuffer.c:380
void resetPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:148
bool std_strings
Definition: pg_backup.h:207

◆ ExecuteSqlCommand()

static void ExecuteSqlCommand ( ArchiveHandle AH,
const char *  qry,
const char *  desc 
)
static

Definition at line 334 of file pg_backup_db.c.

References conn, _archiveHandle::connection, fprintf, _archiveHandle::pgCopyIn, PGRES_COMMAND_OK, PGRES_COPY_IN, PGRES_EMPTY_QUERY, PGRES_TUPLES_OK, PQclear(), PQerrorMessage(), PQexec(), PQresultStatus(), and warn_or_exit_horribly().

Referenced by CommitTransaction(), ExecuteSimpleCommands(), ExecuteSqlCommandBuf(), and StartTransaction().

335 {
336  PGconn *conn = AH->connection;
337  PGresult *res;
338 
339 #ifdef NOT_USED
340  fprintf(stderr, "Executing: '%s'\n\n", qry);
341 #endif
342  res = PQexec(conn, qry);
343 
344  switch (PQresultStatus(res))
345  {
346  case PGRES_COMMAND_OK:
347  case PGRES_TUPLES_OK:
348  case PGRES_EMPTY_QUERY:
349  /* A-OK */
350  break;
351  case PGRES_COPY_IN:
352  /* Assume this is an expected result */
353  AH->pgCopyIn = true;
354  break;
355  default:
356  /* trouble */
357  warn_or_exit_horribly(AH, "%s: %sCommand was: %s",
358  desc, PQerrorMessage(conn), qry);
359  break;
360  }
361 
362  PQclear(res);
363 }
char * PQerrorMessage(const PGconn *conn)
Definition: fe-connect.c:6669
void warn_or_exit_horribly(ArchiveHandle *AH, const char *fmt,...)
#define fprintf
Definition: port.h:219
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
PGconn * conn
Definition: streamutil.c:54
void PQclear(PGresult *res)
Definition: fe-exec.c:694
PGresult * PQexec(PGconn *conn, const char *query)
Definition: fe-exec.c:1939

◆ ExecuteSqlCommandBuf()

int ExecuteSqlCommandBuf ( Archive AHX,
const char *  buf,
size_t  bufLen 
)

Definition at line 451 of file pg_backup_db.c.

References _archiveHandle::connection, ExecuteSimpleCommands(), ExecuteSqlCommand(), fatal, free, OUTPUT_COPYDATA, OUTPUT_OTHERDATA, _archiveHandle::outputKind, pg_malloc(), _archiveHandle::pgCopyIn, PQerrorMessage(), PQputCopyData(), and generate_unaccent_rules::str.

Referenced by ahwrite().

452 {
453  ArchiveHandle *AH = (ArchiveHandle *) AHX;
454 
455  if (AH->outputKind == OUTPUT_COPYDATA)
456  {
457  /*
458  * COPY data.
459  *
460  * We drop the data on the floor if libpq has failed to enter COPY
461  * mode; this allows us to behave reasonably when trying to continue
462  * after an error in a COPY command.
463  */
464  if (AH->pgCopyIn &&
465  PQputCopyData(AH->connection, buf, bufLen) <= 0)
466  fatal("error returned by PQputCopyData: %s",
468  }
469  else if (AH->outputKind == OUTPUT_OTHERDATA)
470  {
471  /*
472  * Table data expressed as INSERT commands; or, in old dump files,
473  * BLOB COMMENTS data (which is expressed as COMMENT ON commands).
474  */
475  ExecuteSimpleCommands(AH, buf, bufLen);
476  }
477  else
478  {
479  /*
480  * General SQL commands; we assume that commands will not be split
481  * across calls.
482  *
483  * In most cases the data passed to us will be a null-terminated
484  * string, but if it's not, we have to add a trailing null.
485  */
486  if (buf[bufLen] == '\0')
487  ExecuteSqlCommand(AH, buf, "could not execute query");
488  else
489  {
490  char *str = (char *) pg_malloc(bufLen + 1);
491 
492  memcpy(str, buf, bufLen);
493  str[bufLen] = '\0';
494  ExecuteSqlCommand(AH, str, "could not execute query");
495  free(str);
496  }
497  }
498 
499  return bufLen;
500 }
int PQputCopyData(PGconn *conn, const char *buffer, int nbytes)
Definition: fe-exec.c:2317
char * PQerrorMessage(const PGconn *conn)
Definition: fe-connect.c:6669
void * pg_malloc(size_t size)
Definition: fe_memutils.c:47
static void ExecuteSqlCommand(ArchiveHandle *AH, const char *qry, const char *desc)
Definition: pg_backup_db.c:334
static char * buf
Definition: pg_test_fsync.c:68
static void ExecuteSimpleCommands(ArchiveHandle *AH, const char *buf, size_t bufLen)
Definition: pg_backup_db.c:386
#define free(a)
Definition: header.h:65
#define fatal(...)
ArchiverOutput outputKind

◆ ExecuteSqlQuery()

PGresult* ExecuteSqlQuery ( Archive AHX,
const char *  query,
ExecStatusType  status 
)

Definition at line 296 of file pg_backup_db.c.

References _archiveHandle::connection, die_on_query_failure(), PQexec(), and PQresultStatus().

Referenced by append_depends_on_extension(), buildMatViewRefreshDependencies(), collectComments(), collectSecLabels(), createViewAsClause(), dumpBlobs(), dumpCompositeType(), dumpCompositeTypeColComments(), dumpDatabase(), dumpDatabaseConfig(), dumpEnumType(), dumpOpclass(), dumpOpfamily(), dumpRule(), dumpSequence(), dumpSequenceData(), dumpTable(), dumpTableData_copy(), dumpTableData_insert(), dumpTSConfig(), dumpUserMappings(), ExecuteSqlQueryForSingleRow(), expand_foreign_server_name_patterns(), expand_schema_name_patterns(), expand_table_name_patterns(), getAccessMethods(), getAggregates(), getBlobs(), getCasts(), getCollations(), getConstraints(), getConversions(), getDefaultACLs(), getDependencies(), getDomainConstraints(), getEventTriggers(), getExtendedStatistics(), getExtensionMembership(), getExtensions(), getForeignDataWrappers(), getForeignServers(), getFuncs(), getIndexes(), getInherits(), getNamespaces(), getOpclasses(), getOperators(), getOpfamilies(), getPolicies(), getProcLangs(), getPublications(), getPublicationTables(), getRules(), getSubscriptions(), getTableAttrs(), getTables(), getTransforms(), getTriggers(), getTSConfigurations(), getTSDictionaries(), getTSParsers(), getTSTemplates(), getTypes(), and processExtensionTables().

297 {
298  ArchiveHandle *AH = (ArchiveHandle *) AHX;
299  PGresult *res;
300 
301  res = PQexec(AH->connection, query);
302  if (PQresultStatus(res) != status)
303  die_on_query_failure(AH, query);
304  return res;
305 }
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
static void die_on_query_failure(ArchiveHandle *AH, const char *query)
Definition: pg_backup_db.c:276
PGresult * PQexec(PGconn *conn, const char *query)
Definition: fe-exec.c:1939
static void static void status(const char *fmt,...) pg_attribute_printf(1
Definition: pg_regress.c:227

◆ ExecuteSqlQueryForSingleRow()

PGresult* ExecuteSqlQueryForSingleRow ( Archive fout,
const char *  query 
)

Definition at line 311 of file pg_backup_db.c.

References ExecuteSqlQuery(), fatal, ngettext, PGRES_TUPLES_OK, and PQntuples().

Referenced by _check_database_version(), binary_upgrade_set_pg_class_oids(), binary_upgrade_set_type_oids_by_rel_oid(), binary_upgrade_set_type_oids_by_type_oid(), ConnectDatabase(), convertTSFunction(), dumpAgg(), dumpBaseType(), dumpCollation(), dumpConversion(), dumpDatabase(), dumpDomain(), dumpForeignServer(), dumpFunc(), dumpOpclass(), dumpOpfamily(), dumpOpr(), dumpRangeType(), dumpSearchPath(), dumpStatisticsExt(), dumpTableSchema(), dumpTSConfig(), dumpTSDictionary(), expand_table_name_patterns(), findLastBuiltinOid_V71(), get_language_name(), get_synchronized_snapshot(), getFormattedTypeName(), and setup_connection().

312 {
313  PGresult *res;
314  int ntups;
315 
316  res = ExecuteSqlQuery(fout, query, PGRES_TUPLES_OK);
317 
318  /* Expecting a single result only */
319  ntups = PQntuples(res);
320  if (ntups != 1)
321  fatal(ngettext("query returned %d row instead of one: %s",
322  "query returned %d rows instead of one: %s",
323  ntups),
324  ntups, query);
325 
326  return res;
327 }
int PQntuples(const PGresult *res)
Definition: fe-exec.c:2769
#define ngettext(s, p, n)
Definition: c.h:1178
#define fatal(...)
PGresult * ExecuteSqlQuery(Archive *AHX, const char *query, ExecStatusType status)
Definition: pg_backup_db.c:296

◆ ExecuteSqlStatement()

void ExecuteSqlStatement ( Archive AHX,
const char *  query 
)

Definition at line 284 of file pg_backup_db.c.

References _archiveHandle::connection, die_on_query_failure(), PGRES_COMMAND_OK, PQclear(), PQexec(), and PQresultStatus().

Referenced by dumpBlobs(), dumpTableData_insert(), expand_table_name_patterns(), getTables(), and setup_connection().

285 {
286  ArchiveHandle *AH = (ArchiveHandle *) AHX;
287  PGresult *res;
288 
289  res = PQexec(AH->connection, query);
290  if (PQresultStatus(res) != PGRES_COMMAND_OK)
291  die_on_query_failure(AH, query);
292  PQclear(res);
293 }
ExecStatusType PQresultStatus(const PGresult *res)
Definition: fe-exec.c:2692
static void die_on_query_failure(ArchiveHandle *AH, const char *query)
Definition: pg_backup_db.c:276
void PQclear(PGresult *res)
Definition: fe-exec.c:694
PGresult * PQexec(PGconn *conn, const char *query)
Definition: fe-exec.c:1939

◆ GetConnection()

PGconn* GetConnection ( Archive AHX)

Definition at line 261 of file pg_backup_db.c.

References _archiveHandle::connection.

262 {
263  ArchiveHandle *AH = (ArchiveHandle *) AHX;
264 
265  return AH->connection;
266 }

◆ notice_processor()

static void notice_processor ( void *  arg,
const char *  message 
)
static

Definition at line 269 of file pg_backup_db.c.

References pg_log_generic(), and PG_LOG_INFO.

Referenced by ConnectDatabase().

270 {
271  pg_log_generic(PG_LOG_INFO, "%s", message);
272 }
void pg_log_generic(enum pg_log_level level, const char *pg_restrict fmt,...)
Definition: logging.c:197

◆ ReconnectToServer()

void ReconnectToServer ( ArchiveHandle AH,
const char *  dbname 
)

Definition at line 78 of file pg_backup_db.c.

References ConnectDatabase(), _archiveHandle::connection, _restoreOptions::cparams, _connParams::override_dbname, pg_strdup(), PQfinish(), _archiveHandle::public, and Archive::ropt.

Referenced by _reconnectToDB().

79 {
80  PGconn *oldConn = AH->connection;
81  RestoreOptions *ropt = AH->public.ropt;
82 
83  /*
84  * Save the dbname, if given, in override_dbname so that it will also
85  * affect any later reconnection attempt.
86  */
87  if (dbname)
89 
90  /*
91  * Note: we want to establish the new connection, and in particular update
92  * ArchiveHandle's connCancel, before closing old connection. Otherwise
93  * an ill-timed SIGINT could try to access a dead connection.
94  */
95  AH->connection = NULL; /* dodge error check in ConnectDatabase */
96 
97  ConnectDatabase((Archive *) AH, &ropt->cparams, true);
98 
99  PQfinish(oldConn);
100 }
RestoreOptions * ropt
Definition: pg_backup.h:192
void PQfinish(PGconn *conn)
Definition: fe-connect.c:4174
char * pg_strdup(const char *in)
Definition: fe_memutils.c:85
char * override_dbname
Definition: pg_backup.h:72
ConnParams cparams
Definition: pg_backup.h:124
void ConnectDatabase(Archive *AHX, const ConnParams *cparams, bool isReconnect)
Definition: pg_backup_db.c:114
char * dbname
Definition: streamutil.c:51

◆ StartTransaction()

void StartTransaction ( Archive AHX)

Definition at line 535 of file pg_backup_db.c.

References ExecuteSqlCommand().

536 {
537  ArchiveHandle *AH = (ArchiveHandle *) AHX;
538 
539  ExecuteSqlCommand(AH, "BEGIN", "could not start database transaction");
540 }
static void ExecuteSqlCommand(ArchiveHandle *AH, const char *qry, const char *desc)
Definition: pg_backup_db.c:334