Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * pg_dumpall.c
4 : : *
5 : : * Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
6 : : * Portions Copyright (c) 1994, Regents of the University of California
7 : : *
8 : : * pg_dumpall forces all pg_dump output to be text, since it also outputs
9 : : * text into the same output stream.
10 : : *
11 : : * src/bin/pg_dump/pg_dumpall.c
12 : : *
13 : : *-------------------------------------------------------------------------
14 : : */
15 : :
16 : : #include "postgres_fe.h"
17 : :
18 : : #include <time.h>
19 : : #include <unistd.h>
20 : :
21 : : #include "catalog/pg_authid_d.h"
22 : : #include "common/connect.h"
23 : : #include "common/file_utils.h"
24 : : #include "common/hashfn_unstable.h"
25 : : #include "common/logging.h"
26 : : #include "common/string.h"
27 : : #include "dumputils.h"
28 : : #include "fe_utils/string_utils.h"
29 : : #include "filter.h"
30 : : #include "getopt_long.h"
31 : : #include "pg_backup.h"
32 : :
33 : : /* version string we expect back from pg_dump */
34 : : #define PGDUMP_VERSIONSTR "pg_dump (PostgreSQL) " PG_VERSION "\n"
35 : :
36 : : typedef struct
37 : : {
38 : : uint32 status;
39 : : uint32 hashval;
40 : : char *rolename;
41 : : } RoleNameEntry;
42 : :
43 : : #define SH_PREFIX rolename
44 : : #define SH_ELEMENT_TYPE RoleNameEntry
45 : : #define SH_KEY_TYPE char *
46 : : #define SH_KEY rolename
47 : : #define SH_HASH_KEY(tb, key) hash_string(key)
48 : : #define SH_EQUAL(tb, a, b) (strcmp(a, b) == 0)
49 : : #define SH_STORE_HASH
50 : : #define SH_GET_HASH(tb, a) (a)->hashval
51 : : #define SH_SCOPE static inline
52 : : #define SH_RAW_ALLOCATOR pg_malloc0
53 : : #define SH_DECLARE
54 : : #define SH_DEFINE
55 : : #include "lib/simplehash.h"
56 : :
57 : : static void help(void);
58 : :
59 : : static void dropRoles(PGconn *conn);
60 : : static void dumpRoles(PGconn *conn);
61 : : static void dumpRoleMembership(PGconn *conn);
62 : : static void dumpRoleGUCPrivs(PGconn *conn);
63 : : static void dropTablespaces(PGconn *conn);
64 : : static void dumpTablespaces(PGconn *conn);
65 : : static void dropDBs(PGconn *conn);
66 : : static void dumpUserConfig(PGconn *conn, const char *username);
67 : : static void dumpDatabases(PGconn *conn);
68 : : static void dumpTimestamp(const char *msg);
69 : : static int runPgDump(const char *dbname, const char *create_opts);
70 : : static void buildShSecLabels(PGconn *conn,
71 : : const char *catalog_name, Oid objectId,
72 : : const char *objtype, const char *objname,
73 : : PQExpBuffer buffer);
74 : : static PGconn *connectDatabase(const char *dbname,
75 : : const char *connection_string, const char *pghost,
76 : : const char *pgport, const char *pguser,
77 : : trivalue prompt_password, bool fail_on_error);
78 : : static char *constructConnStr(const char **keywords, const char **values);
79 : : static PGresult *executeQuery(PGconn *conn, const char *query);
80 : : static void executeCommand(PGconn *conn, const char *query);
81 : : static void expand_dbname_patterns(PGconn *conn, SimpleStringList *patterns,
82 : : SimpleStringList *names);
83 : : static void read_dumpall_filters(const char *filename, SimpleStringList *patterns);
84 : :
85 : : static char pg_dump_bin[MAXPGPATH];
86 : : static const char *progname;
87 : : static PQExpBuffer pgdumpopts;
88 : : static char *connstr = "";
89 : : static bool output_clean = false;
90 : : static bool skip_acls = false;
91 : : static bool verbose = false;
92 : : static bool dosync = true;
93 : :
94 : : static int binary_upgrade = 0;
95 : : static int column_inserts = 0;
96 : : static int disable_dollar_quoting = 0;
97 : : static int disable_triggers = 0;
98 : : static int if_exists = 0;
99 : : static int inserts = 0;
100 : : static int no_table_access_method = 0;
101 : : static int no_tablespaces = 0;
102 : : static int use_setsessauth = 0;
103 : : static int no_comments = 0;
104 : : static int no_publications = 0;
105 : : static int no_security_labels = 0;
106 : : static int no_subscriptions = 0;
107 : : static int no_toast_compression = 0;
108 : : static int no_unlogged_table_data = 0;
109 : : static int no_role_passwords = 0;
110 : : static int server_version;
111 : : static int load_via_partition_root = 0;
112 : : static int on_conflict_do_nothing = 0;
113 : :
114 : : static char role_catalog[10];
115 : : #define PG_AUTHID "pg_authid"
116 : : #define PG_ROLES "pg_roles "
117 : :
118 : : static FILE *OPF;
119 : : static char *filename = NULL;
120 : :
121 : : static SimpleStringList database_exclude_patterns = {NULL, NULL};
122 : : static SimpleStringList database_exclude_names = {NULL, NULL};
123 : :
124 : : #define exit_nicely(code) exit(code)
125 : :
126 : : int
7901 peter_e@gmx.net 127 :CBC 48 : main(int argc, char *argv[])
128 : : {
129 : : static struct option long_options[] = {
130 : : {"data-only", no_argument, NULL, 'a'},
131 : : {"clean", no_argument, NULL, 'c'},
132 : : {"encoding", required_argument, NULL, 'E'},
133 : : {"file", required_argument, NULL, 'f'},
134 : : {"globals-only", no_argument, NULL, 'g'},
135 : : {"host", required_argument, NULL, 'h'},
136 : : {"dbname", required_argument, NULL, 'd'},
137 : : {"database", required_argument, NULL, 'l'},
138 : : {"no-owner", no_argument, NULL, 'O'},
139 : : {"port", required_argument, NULL, 'p'},
140 : : {"roles-only", no_argument, NULL, 'r'},
141 : : {"schema-only", no_argument, NULL, 's'},
142 : : {"superuser", required_argument, NULL, 'S'},
143 : : {"tablespaces-only", no_argument, NULL, 't'},
144 : : {"username", required_argument, NULL, 'U'},
145 : : {"verbose", no_argument, NULL, 'v'},
146 : : {"no-password", no_argument, NULL, 'w'},
147 : : {"password", no_argument, NULL, 'W'},
148 : : {"no-privileges", no_argument, NULL, 'x'},
149 : : {"no-acl", no_argument, NULL, 'x'},
150 : :
151 : : /*
152 : : * the following options don't have an equivalent short option letter
153 : : */
154 : : {"attribute-inserts", no_argument, &column_inserts, 1},
155 : : {"binary-upgrade", no_argument, &binary_upgrade, 1},
156 : : {"column-inserts", no_argument, &column_inserts, 1},
157 : : {"disable-dollar-quoting", no_argument, &disable_dollar_quoting, 1},
158 : : {"disable-triggers", no_argument, &disable_triggers, 1},
159 : : {"exclude-database", required_argument, NULL, 6},
160 : : {"extra-float-digits", required_argument, NULL, 5},
161 : : {"if-exists", no_argument, &if_exists, 1},
162 : : {"inserts", no_argument, &inserts, 1},
163 : : {"lock-wait-timeout", required_argument, NULL, 2},
164 : : {"no-table-access-method", no_argument, &no_table_access_method, 1},
165 : : {"no-tablespaces", no_argument, &no_tablespaces, 1},
166 : : {"quote-all-identifiers", no_argument, "e_all_identifiers, 1},
167 : : {"load-via-partition-root", no_argument, &load_via_partition_root, 1},
168 : : {"role", required_argument, NULL, 3},
169 : : {"use-set-session-authorization", no_argument, &use_setsessauth, 1},
170 : : {"no-comments", no_argument, &no_comments, 1},
171 : : {"no-publications", no_argument, &no_publications, 1},
172 : : {"no-role-passwords", no_argument, &no_role_passwords, 1},
173 : : {"no-security-labels", no_argument, &no_security_labels, 1},
174 : : {"no-subscriptions", no_argument, &no_subscriptions, 1},
175 : : {"no-sync", no_argument, NULL, 4},
176 : : {"no-toast-compression", no_argument, &no_toast_compression, 1},
177 : : {"no-unlogged-table-data", no_argument, &no_unlogged_table_data, 1},
178 : : {"on-conflict-do-nothing", no_argument, &on_conflict_do_nothing, 1},
179 : : {"rows-per-insert", required_argument, NULL, 7},
180 : : {"filter", required_argument, NULL, 8},
181 : :
182 : : {NULL, 0, NULL, 0}
183 : : };
184 : :
4153 bruce@momjian.us 185 : 48 : char *pghost = NULL;
186 : 48 : char *pgport = NULL;
187 : 48 : char *pguser = NULL;
188 : 48 : char *pgdb = NULL;
189 : 48 : char *use_role = NULL;
2417 rhaas@postgresql.org 190 : 48 : const char *dumpencoding = NULL;
3470 alvherre@alvh.no-ip. 191 : 48 : trivalue prompt_password = TRI_DEFAULT;
4153 bruce@momjian.us 192 : 48 : bool data_only = false;
193 : 48 : bool globals_only = false;
194 : 48 : bool roles_only = false;
195 : 48 : bool tablespaces_only = false;
196 : : PGconn *conn;
197 : : int encoding;
198 : : const char *std_strings;
199 : : int c,
200 : : ret;
201 : : int optindex;
202 : :
1840 peter@eisentraut.org 203 : 48 : pg_logging_init(argv[0]);
204 : 48 : pg_logging_set_level(PG_LOG_WARNING);
5603 peter_e@gmx.net 205 : 48 : set_pglocale_pgservice(argv[0], PG_TEXTDOMAIN("pg_dump"));
7681 bruce@momjian.us 206 : 48 : progname = get_progname(argv[0]);
207 : :
7901 peter_e@gmx.net 208 [ + - ]: 48 : if (argc > 1)
209 : : {
210 [ + + - + ]: 48 : if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
211 : : {
212 : 1 : help();
4441 rhaas@postgresql.org 213 : 1 : exit_nicely(0);
214 : : }
7901 peter_e@gmx.net 215 [ + + + + ]: 47 : if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
216 : : {
217 : 10 : puts("pg_dumpall (PostgreSQL) " PG_VERSION);
4441 rhaas@postgresql.org 218 : 10 : exit_nicely(0);
219 : : }
220 : : }
221 : :
6537 peter_e@gmx.net 222 [ - + ]: 37 : if ((ret = find_other_exec(argv[0], "pg_dump", PGDUMP_VERSIONSTR,
223 : : pg_dump_bin)) < 0)
224 : : {
225 : : char full_path[MAXPGPATH];
226 : :
7121 bruce@momjian.us 227 [ # # ]:UBC 0 : if (find_my_exec(argv[0], full_path) < 0)
6273 peter_e@gmx.net 228 : 0 : strlcpy(full_path, progname, sizeof(full_path));
229 : :
7278 bruce@momjian.us 230 [ # # ]: 0 : if (ret == -1)
737 tgl@sss.pgh.pa.us 231 : 0 : pg_fatal("program \"%s\" is needed by %s but was not found in the same directory as \"%s\"",
232 : : "pg_dump", progname, full_path);
233 : : else
234 : 0 : pg_fatal("program \"%s\" was found by \"%s\" but was not the same version as %s",
235 : : "pg_dump", full_path, progname);
236 : : }
237 : :
7901 peter_e@gmx.net 238 :CBC 37 : pgdumpopts = createPQExpBuffer();
239 : :
1788 michael@paquier.xyz 240 [ + + ]: 158 : while ((c = getopt_long(argc, argv, "acd:E:f:gh:l:Op:rsS:tU:vwWx", long_options, &optindex)) != -1)
241 : : {
7901 peter_e@gmx.net 242 [ - + + - : 125 : switch (c)
+ + + + -
+ + - - +
+ + - - -
+ - - + -
+ - + + ]
243 : : {
7625 tgl@sss.pgh.pa.us 244 :UBC 0 : case 'a':
245 : 0 : data_only = true;
3800 heikki.linnakangas@i 246 : 0 : appendPQExpBufferStr(pgdumpopts, " -a");
7625 tgl@sss.pgh.pa.us 247 : 0 : break;
248 : :
7901 peter_e@gmx.net 249 :CBC 1 : case 'c':
250 : 1 : output_clean = true;
251 : 1 : break;
252 : :
4066 heikki.linnakangas@i 253 : 8 : case 'd':
254 : 8 : connstr = pg_strdup(optarg);
255 : 8 : break;
256 : :
2417 rhaas@postgresql.org 257 :UBC 0 : case 'E':
258 : 0 : dumpencoding = pg_strdup(optarg);
259 : 0 : appendPQExpBufferStr(pgdumpopts, " -E ");
260 : 0 : appendShellString(pgdumpopts, optarg);
261 : 0 : break;
262 : :
6289 bruce@momjian.us 263 :CBC 26 : case 'f':
4202 264 : 26 : filename = pg_strdup(optarg);
3800 heikki.linnakangas@i 265 : 26 : appendPQExpBufferStr(pgdumpopts, " -f ");
2806 noah@leadboat.com 266 : 26 : appendShellString(pgdumpopts, filename);
6289 bruce@momjian.us 267 : 26 : break;
268 : :
7901 peter_e@gmx.net 269 : 11 : case 'g':
270 : 11 : globals_only = true;
271 : 11 : break;
272 : :
273 : 4 : case 'h':
4202 bruce@momjian.us 274 : 4 : pghost = pg_strdup(optarg);
7901 peter_e@gmx.net 275 : 4 : break;
276 : :
6289 bruce@momjian.us 277 : 1 : case 'l':
4202 278 : 1 : pgdb = pg_strdup(optarg);
6289 279 : 1 : break;
280 : :
7216 bruce@momjian.us 281 :UBC 0 : case 'O':
3800 heikki.linnakangas@i 282 : 0 : appendPQExpBufferStr(pgdumpopts, " -O");
7216 bruce@momjian.us 283 : 0 : break;
284 : :
7901 peter_e@gmx.net 285 :CBC 12 : case 'p':
4202 bruce@momjian.us 286 : 12 : pgport = pg_strdup(optarg);
7901 peter_e@gmx.net 287 : 12 : break;
288 : :
6289 bruce@momjian.us 289 : 7 : case 'r':
290 : 7 : roles_only = true;
291 : 7 : break;
292 : :
7625 tgl@sss.pgh.pa.us 293 :UBC 0 : case 's':
3800 heikki.linnakangas@i 294 : 0 : appendPQExpBufferStr(pgdumpopts, " -s");
7625 tgl@sss.pgh.pa.us 295 : 0 : break;
296 : :
7216 bruce@momjian.us 297 : 0 : case 'S':
3800 heikki.linnakangas@i 298 : 0 : appendPQExpBufferStr(pgdumpopts, " -S ");
2806 noah@leadboat.com 299 : 0 : appendShellString(pgdumpopts, optarg);
7216 bruce@momjian.us 300 : 0 : break;
301 : :
6289 bruce@momjian.us 302 :CBC 2 : case 't':
303 : 2 : tablespaces_only = true;
304 : 2 : break;
305 : :
7901 peter_e@gmx.net 306 : 11 : case 'U':
4202 bruce@momjian.us 307 : 11 : pguser = pg_strdup(optarg);
7901 peter_e@gmx.net 308 : 11 : break;
309 : :
310 : 2 : case 'v':
311 : 2 : verbose = true;
1305 tgl@sss.pgh.pa.us 312 : 2 : pg_logging_increase_verbosity();
3800 heikki.linnakangas@i 313 : 2 : appendPQExpBufferStr(pgdumpopts, " -v");
7901 peter_e@gmx.net 314 : 2 : break;
315 : :
5526 peter_e@gmx.net 316 :UBC 0 : case 'w':
317 : 0 : prompt_password = TRI_NO;
3800 heikki.linnakangas@i 318 : 0 : appendPQExpBufferStr(pgdumpopts, " -w");
5526 peter_e@gmx.net 319 : 0 : break;
320 : :
7901 321 : 0 : case 'W':
5526 322 : 0 : prompt_password = TRI_YES;
3800 heikki.linnakangas@i 323 : 0 : appendPQExpBufferStr(pgdumpopts, " -W");
7901 peter_e@gmx.net 324 : 0 : break;
325 : :
7625 tgl@sss.pgh.pa.us 326 : 0 : case 'x':
327 : 0 : skip_acls = true;
3800 heikki.linnakangas@i 328 : 0 : appendPQExpBufferStr(pgdumpopts, " -x");
7625 tgl@sss.pgh.pa.us 329 : 0 : break;
330 : :
7216 bruce@momjian.us 331 :CBC 12 : case 0:
332 : 12 : break;
333 : :
5707 alvherre@alvh.no-ip. 334 :UBC 0 : case 2:
3800 heikki.linnakangas@i 335 : 0 : appendPQExpBufferStr(pgdumpopts, " --lock-wait-timeout ");
2806 noah@leadboat.com 336 : 0 : appendShellString(pgdumpopts, optarg);
5578 tgl@sss.pgh.pa.us 337 : 0 : break;
338 : :
339 : 0 : case 3:
4202 bruce@momjian.us 340 : 0 : use_role = pg_strdup(optarg);
3800 heikki.linnakangas@i 341 : 0 : appendPQExpBufferStr(pgdumpopts, " --role ");
2806 noah@leadboat.com 342 : 0 : appendShellString(pgdumpopts, use_role);
5707 alvherre@alvh.no-ip. 343 : 0 : break;
344 : :
2580 andrew@dunslane.net 345 :CBC 17 : case 4:
346 : 17 : dosync = false;
347 : 17 : appendPQExpBufferStr(pgdumpopts, " --no-sync");
348 : 17 : break;
349 : :
1882 andrew@dunslane.net 350 :UBC 0 : case 5:
351 : 0 : appendPQExpBufferStr(pgdumpopts, " --extra-float-digits ");
352 : 0 : appendShellString(pgdumpopts, optarg);
353 : 0 : break;
354 : :
1871 andrew@dunslane.net 355 :CBC 5 : case 6:
356 : 5 : simple_string_list_append(&database_exclude_patterns, optarg);
357 : 5 : break;
358 : :
1766 alvherre@alvh.no-ip. 359 :UBC 0 : case 7:
360 : 0 : appendPQExpBufferStr(pgdumpopts, " --rows-per-insert ");
361 : 0 : appendShellString(pgdumpopts, optarg);
362 : 0 : break;
363 : :
137 dgustafsson@postgres 364 :GNC 5 : case 8:
365 : 5 : read_dumpall_filters(optarg, &database_exclude_patterns);
366 : 2 : break;
367 : :
7893 bruce@momjian.us 368 :CBC 1 : default:
369 : : /* getopt_long already emitted a complaint */
737 tgl@sss.pgh.pa.us 370 : 1 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
4441 rhaas@postgresql.org 371 : 1 : exit_nicely(1);
372 : : }
373 : : }
374 : :
375 : : /* Complain if any arguments remain */
7893 bruce@momjian.us 376 [ + + ]: 33 : if (optind < argc)
377 : : {
1840 peter@eisentraut.org 378 : 1 : pg_log_error("too many command-line arguments (first is \"%s\")",
379 : : argv[optind]);
737 tgl@sss.pgh.pa.us 380 : 1 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
4441 rhaas@postgresql.org 381 : 1 : exit_nicely(1);
382 : : }
383 : :
1871 andrew@dunslane.net 384 [ + + + + ]: 32 : if (database_exclude_patterns.head != NULL &&
385 [ + - - + ]: 5 : (globals_only || roles_only || tablespaces_only))
386 : : {
1749 peter@eisentraut.org 387 : 2 : pg_log_error("option --exclude-database cannot be used together with -g/--globals-only, -r/--roles-only, or -t/--tablespaces-only");
737 tgl@sss.pgh.pa.us 388 : 2 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
1871 andrew@dunslane.net 389 : 2 : exit_nicely(1);
390 : : }
391 : :
392 : : /* Make sure the user hasn't specified a mix of globals-only options */
6289 bruce@momjian.us 393 [ + + + + ]: 30 : if (globals_only && roles_only)
394 : : {
1840 peter@eisentraut.org 395 : 1 : pg_log_error("options -g/--globals-only and -r/--roles-only cannot be used together");
737 tgl@sss.pgh.pa.us 396 : 1 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
4441 rhaas@postgresql.org 397 : 1 : exit_nicely(1);
398 : : }
399 : :
6289 bruce@momjian.us 400 [ + + + + ]: 29 : if (globals_only && tablespaces_only)
401 : : {
1840 peter@eisentraut.org 402 : 1 : pg_log_error("options -g/--globals-only and -t/--tablespaces-only cannot be used together");
737 tgl@sss.pgh.pa.us 403 : 1 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
4441 rhaas@postgresql.org 404 : 1 : exit_nicely(1);
405 : : }
406 : :
3695 alvherre@alvh.no-ip. 407 [ + + + - ]: 28 : if (if_exists && !output_clean)
737 tgl@sss.pgh.pa.us 408 : 1 : pg_fatal("option --if-exists requires option -c/--clean");
409 : :
6289 bruce@momjian.us 410 [ + + + + ]: 27 : if (roles_only && tablespaces_only)
411 : : {
1840 peter@eisentraut.org 412 : 1 : pg_log_error("options -r/--roles-only and -t/--tablespaces-only cannot be used together");
737 tgl@sss.pgh.pa.us 413 : 1 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
4441 rhaas@postgresql.org 414 : 1 : exit_nicely(1);
415 : : }
416 : :
417 : : /*
418 : : * If password values are not required in the dump, switch to using
419 : : * pg_roles which is equally useful, just more likely to have unrestricted
420 : : * access than pg_authid.
421 : : */
2595 simon@2ndQuadrant.co 422 [ - + ]: 26 : if (no_role_passwords)
2595 simon@2ndQuadrant.co 423 :UBC 0 : sprintf(role_catalog, "%s", PG_ROLES);
424 : : else
2595 simon@2ndQuadrant.co 425 :CBC 26 : sprintf(role_catalog, "%s", PG_AUTHID);
426 : :
427 : : /* Add long options to the pg_dump argument list */
4993 tgl@sss.pgh.pa.us 428 [ + + ]: 26 : if (binary_upgrade)
3800 heikki.linnakangas@i 429 : 4 : appendPQExpBufferStr(pgdumpopts, " --binary-upgrade");
4993 tgl@sss.pgh.pa.us 430 [ - + ]: 26 : if (column_inserts)
3800 heikki.linnakangas@i 431 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --column-inserts");
4993 tgl@sss.pgh.pa.us 432 [ - + ]:CBC 26 : if (disable_dollar_quoting)
3800 heikki.linnakangas@i 433 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --disable-dollar-quoting");
4993 tgl@sss.pgh.pa.us 434 [ - + ]:CBC 26 : if (disable_triggers)
3800 heikki.linnakangas@i 435 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --disable-triggers");
4993 tgl@sss.pgh.pa.us 436 [ - + ]:CBC 26 : if (inserts)
3800 heikki.linnakangas@i 437 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --inserts");
818 michael@paquier.xyz 438 [ - + ]:CBC 26 : if (no_table_access_method)
818 michael@paquier.xyz 439 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --no-table-access-method");
4993 tgl@sss.pgh.pa.us 440 [ - + ]:CBC 26 : if (no_tablespaces)
3800 heikki.linnakangas@i 441 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --no-tablespaces");
4993 tgl@sss.pgh.pa.us 442 [ + + ]:CBC 26 : if (quote_all_identifiers)
3800 heikki.linnakangas@i 443 : 4 : appendPQExpBufferStr(pgdumpopts, " --quote-all-identifiers");
2435 rhaas@postgresql.org 444 [ - + ]: 26 : if (load_via_partition_root)
2435 rhaas@postgresql.org 445 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --load-via-partition-root");
4993 tgl@sss.pgh.pa.us 446 [ - + ]:CBC 26 : if (use_setsessauth)
3800 heikki.linnakangas@i 447 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --use-set-session-authorization");
2271 tgl@sss.pgh.pa.us 448 [ - + ]:CBC 26 : if (no_comments)
2271 tgl@sss.pgh.pa.us 449 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --no-comments");
2529 peter_e@gmx.net 450 [ - + ]:CBC 26 : if (no_publications)
2529 peter_e@gmx.net 451 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --no-publications");
4714 peter_e@gmx.net 452 [ - + ]:CBC 26 : if (no_security_labels)
3800 heikki.linnakangas@i 453 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --no-security-labels");
2532 peter_e@gmx.net 454 [ - + ]:CBC 26 : if (no_subscriptions)
2532 peter_e@gmx.net 455 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --no-subscriptions");
1061 michael@paquier.xyz 456 [ - + ]:CBC 26 : if (no_toast_compression)
1061 michael@paquier.xyz 457 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --no-toast-compression");
4855 rhaas@postgresql.org 458 [ + + ]:CBC 26 : if (no_unlogged_table_data)
3800 heikki.linnakangas@i 459 : 3 : appendPQExpBufferStr(pgdumpopts, " --no-unlogged-table-data");
2102 tmunro@postgresql.or 460 [ - + ]: 26 : if (on_conflict_do_nothing)
2102 tmunro@postgresql.or 461 :UBC 0 : appendPQExpBufferStr(pgdumpopts, " --on-conflict-do-nothing");
462 : :
463 : : /*
464 : : * If there was a database specified on the command line, use that,
465 : : * otherwise try to connect to database "postgres", and failing that
466 : : * "template1".
467 : : */
6289 bruce@momjian.us 468 [ + + ]:CBC 26 : if (pgdb)
469 : : {
4066 heikki.linnakangas@i 470 : 1 : conn = connectDatabase(pgdb, connstr, pghost, pgport, pguser,
471 : : prompt_password, false);
472 : :
6289 bruce@momjian.us 473 [ - + ]: 1 : if (!conn)
737 tgl@sss.pgh.pa.us 474 :UBC 0 : pg_fatal("could not connect to database \"%s\"", pgdb);
475 : : }
476 : : else
477 : : {
4066 heikki.linnakangas@i 478 :CBC 25 : conn = connectDatabase("postgres", connstr, pghost, pgport, pguser,
479 : : prompt_password, false);
6289 bruce@momjian.us 480 [ - + ]: 25 : if (!conn)
4066 heikki.linnakangas@i 481 :UBC 0 : conn = connectDatabase("template1", connstr, pghost, pgport, pguser,
482 : : prompt_password, true);
483 : :
6289 bruce@momjian.us 484 [ - + ]:CBC 25 : if (!conn)
485 : : {
1840 peter@eisentraut.org 486 :UBC 0 : pg_log_error("could not connect to databases \"postgres\" or \"template1\"\n"
487 : : "Please specify an alternative database.");
737 tgl@sss.pgh.pa.us 488 : 0 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
4441 rhaas@postgresql.org 489 : 0 : exit_nicely(1);
490 : : }
491 : : }
492 : :
493 : : /*
494 : : * Get a list of database names that match the exclude patterns
495 : : */
1871 andrew@dunslane.net 496 :CBC 26 : expand_dbname_patterns(conn, &database_exclude_patterns,
497 : : &database_exclude_names);
498 : :
499 : : /*
500 : : * Open the output file if required, otherwise use stdout
501 : : */
6289 bruce@momjian.us 502 [ + + ]: 24 : if (filename)
503 : : {
504 : 22 : OPF = fopen(filename, PG_BINARY_W);
505 [ - + ]: 22 : if (!OPF)
737 tgl@sss.pgh.pa.us 506 :UBC 0 : pg_fatal("could not open output file \"%s\": %m",
507 : : filename);
508 : : }
509 : : else
6289 bruce@momjian.us 510 :CBC 2 : OPF = stdout;
511 : :
512 : : /*
513 : : * Set the client encoding if requested.
514 : : */
2417 rhaas@postgresql.org 515 [ - + ]: 24 : if (dumpencoding)
516 : : {
2417 rhaas@postgresql.org 517 [ # # ]:UBC 0 : if (PQsetClientEncoding(conn, dumpencoding) < 0)
737 tgl@sss.pgh.pa.us 518 : 0 : pg_fatal("invalid client encoding \"%s\" specified",
519 : : dumpencoding);
520 : : }
521 : :
522 : : /*
523 : : * Get the active encoding and the standard_conforming_strings setting, so
524 : : * we know how to escape strings.
525 : : */
6531 tgl@sss.pgh.pa.us 526 :CBC 24 : encoding = PQclientEncoding(conn);
527 : 24 : std_strings = PQparameterStatus(conn, "standard_conforming_strings");
528 [ - + ]: 24 : if (!std_strings)
6531 tgl@sss.pgh.pa.us 529 :UBC 0 : std_strings = "off";
530 : :
531 : : /* Set the role if requested */
852 tgl@sss.pgh.pa.us 532 [ - + ]:CBC 24 : if (use_role)
533 : : {
5578 tgl@sss.pgh.pa.us 534 :UBC 0 : PQExpBuffer query = createPQExpBuffer();
535 : :
536 : 0 : appendPQExpBuffer(query, "SET ROLE %s", fmtId(use_role));
537 : 0 : executeCommand(conn, query->data);
538 : 0 : destroyPQExpBuffer(query);
539 : : }
540 : :
541 : : /* Force quoting of all identifiers if requested. */
852 tgl@sss.pgh.pa.us 542 [ + + ]:CBC 24 : if (quote_all_identifiers)
5015 rhaas@postgresql.org 543 : 4 : executeCommand(conn, "SET quote_all_identifiers = true");
544 : :
6289 bruce@momjian.us 545 : 24 : fprintf(OPF, "--\n-- PostgreSQL database cluster dump\n--\n\n");
7251 546 [ + + ]: 24 : if (verbose)
7168 547 : 2 : dumpTimestamp("Started on");
548 : :
549 : : /*
550 : : * We used to emit \connect postgres here, but that served no purpose
551 : : * other than to break things for installations without a postgres
552 : : * database. Everything we're restoring here is a global, so whichever
553 : : * database we're connected to at the moment is fine.
554 : : */
555 : :
556 : : /* Restore will need to write to the target cluster */
3788 kgrittn@postgresql.o 557 : 24 : fprintf(OPF, "SET default_transaction_read_only = off;\n\n");
558 : :
559 : : /* Replicate encoding and std_strings in output */
5482 tgl@sss.pgh.pa.us 560 : 24 : fprintf(OPF, "SET client_encoding = '%s';\n",
561 : : pg_encoding_to_char(encoding));
562 : 24 : fprintf(OPF, "SET standard_conforming_strings = %s;\n", std_strings);
563 [ - + ]: 24 : if (strcmp(std_strings, "off") == 0)
5482 tgl@sss.pgh.pa.us 564 :UBC 0 : fprintf(OPF, "SET escape_string_warning = off;\n");
5482 tgl@sss.pgh.pa.us 565 :CBC 24 : fprintf(OPF, "\n");
566 : :
7625 567 [ + - ]: 24 : if (!data_only)
568 : : {
569 : : /*
570 : : * If asked to --clean, do that first. We can avoid detailed
571 : : * dependency analysis because databases never depend on each other,
572 : : * and tablespaces never depend on each other. Roles could have
573 : : * grants to each other, but DROP ROLE will clean those up silently.
574 : : */
5482 575 [ + + ]: 24 : if (output_clean)
576 : : {
577 [ - + - - : 1 : if (!globals_only && !roles_only && !tablespaces_only)
- - ]
5482 tgl@sss.pgh.pa.us 578 :UBC 0 : dropDBs(conn);
579 : :
5482 tgl@sss.pgh.pa.us 580 [ + - + - ]:CBC 1 : if (!roles_only && !no_tablespaces)
2741 581 : 1 : dropTablespaces(conn);
582 : :
5482 583 [ + - ]: 1 : if (!tablespaces_only)
584 : 1 : dropRoles(conn);
585 : : }
586 : :
587 : : /*
588 : : * Now create objects as requested. Be careful that option logic here
589 : : * is the same as for drops above.
590 : : */
6289 bruce@momjian.us 591 [ + - ]: 24 : if (!tablespaces_only)
592 : : {
593 : : /* Dump roles (users) */
594 : 24 : dumpRoles(conn);
595 : :
596 : : /* Dump role memberships */
852 tgl@sss.pgh.pa.us 597 : 24 : dumpRoleMembership(conn);
598 : :
599 : : /* Dump role GUC privileges */
739 600 [ + - + - ]: 24 : if (server_version >= 150000 && !skip_acls)
601 : 24 : dumpRoleGUCPrivs(conn);
602 : : }
603 : :
604 : : /* Dump tablespaces */
5869 605 [ + + + - ]: 24 : if (!roles_only && !no_tablespaces)
2741 606 : 19 : dumpTablespaces(conn);
607 : : }
608 : :
6289 bruce@momjian.us 609 [ + + + + : 24 : if (!globals_only && !roles_only && !tablespaces_only)
+ - ]
7625 tgl@sss.pgh.pa.us 610 : 12 : dumpDatabases(conn);
611 : :
7901 peter_e@gmx.net 612 : 23 : PQfinish(conn);
613 : :
7251 bruce@momjian.us 614 [ + + ]: 23 : if (verbose)
615 : 2 : dumpTimestamp("Completed on");
6289 616 : 23 : fprintf(OPF, "--\n-- PostgreSQL database cluster dump complete\n--\n\n");
617 : :
618 [ + + ]: 23 : if (filename)
619 : : {
620 : 21 : fclose(OPF);
621 : :
622 : : /* sync the resulting file, errors are not fatal */
2580 andrew@dunslane.net 623 [ + + ]: 21 : if (dosync)
1840 peter@eisentraut.org 624 : 6 : (void) fsync_fname(filename, false);
625 : : }
626 : :
4441 rhaas@postgresql.org 627 : 23 : exit_nicely(0);
628 : : }
629 : :
630 : :
631 : : static void
7901 peter_e@gmx.net 632 : 1 : help(void)
633 : : {
634 : 1 : printf(_("%s extracts a PostgreSQL database cluster into an SQL script file.\n\n"), progname);
635 : 1 : printf(_("Usage:\n"));
7849 636 : 1 : printf(_(" %s [OPTION]...\n"), progname);
637 : :
7209 bruce@momjian.us 638 : 1 : printf(_("\nGeneral options:\n"));
4349 peter_e@gmx.net 639 : 1 : printf(_(" -f, --file=FILENAME output file name\n"));
2668 sfrost@snowman.net 640 : 1 : printf(_(" -v, --verbose verbose mode\n"));
4318 peter_e@gmx.net 641 : 1 : printf(_(" -V, --version output version information, then exit\n"));
4349 642 : 1 : printf(_(" --lock-wait-timeout=TIMEOUT fail after waiting TIMEOUT for a table lock\n"));
4318 643 : 1 : printf(_(" -?, --help show this help, then exit\n"));
7209 bruce@momjian.us 644 : 1 : printf(_("\nOptions controlling the output content:\n"));
4349 peter_e@gmx.net 645 : 1 : printf(_(" -a, --data-only dump only the data, not the schema\n"));
646 : 1 : printf(_(" -c, --clean clean (drop) databases before recreating\n"));
2417 rhaas@postgresql.org 647 : 1 : printf(_(" -E, --encoding=ENCODING dump the data in encoding ENCODING\n"));
4349 peter_e@gmx.net 648 : 1 : printf(_(" -g, --globals-only dump only global objects, no databases\n"));
649 : 1 : printf(_(" -O, --no-owner skip restoration of object ownership\n"));
650 : 1 : printf(_(" -r, --roles-only dump only roles, no databases or tablespaces\n"));
651 : 1 : printf(_(" -s, --schema-only dump only the schema, no data\n"));
652 : 1 : printf(_(" -S, --superuser=NAME superuser user name to use in the dump\n"));
653 : 1 : printf(_(" -t, --tablespaces-only dump only tablespaces, no databases or roles\n"));
654 : 1 : printf(_(" -x, --no-privileges do not dump privileges (grant/revoke)\n"));
655 : 1 : printf(_(" --binary-upgrade for use by upgrade utilities only\n"));
656 : 1 : printf(_(" --column-inserts dump data as INSERT commands with column names\n"));
657 : 1 : printf(_(" --disable-dollar-quoting disable dollar quoting, use SQL standard quoting\n"));
658 : 1 : printf(_(" --disable-triggers disable triggers during data-only restore\n"));
1871 andrew@dunslane.net 659 : 1 : printf(_(" --exclude-database=PATTERN exclude databases whose name matches PATTERN\n"));
1882 660 : 1 : printf(_(" --extra-float-digits=NUM override default setting for extra_float_digits\n"));
137 dgustafsson@postgres 661 :GNC 1 : printf(_(" --filter=FILENAME exclude databases specified in FILENAME\n"));
3695 alvherre@alvh.no-ip. 662 :CBC 1 : printf(_(" --if-exists use IF EXISTS when dropping objects\n"));
4349 peter_e@gmx.net 663 : 1 : printf(_(" --inserts dump data as INSERT commands, rather than COPY\n"));
2141 664 : 1 : printf(_(" --load-via-partition-root load partitions via the root table\n"));
2271 tgl@sss.pgh.pa.us 665 : 1 : printf(_(" --no-comments do not dump comments\n"));
2529 peter_e@gmx.net 666 : 1 : printf(_(" --no-publications do not dump publications\n"));
2502 667 : 1 : printf(_(" --no-role-passwords do not dump passwords for roles\n"));
4349 668 : 1 : printf(_(" --no-security-labels do not dump security label assignments\n"));
2532 669 : 1 : printf(_(" --no-subscriptions do not dump subscriptions\n"));
2580 andrew@dunslane.net 670 : 1 : printf(_(" --no-sync do not wait for changes to be written safely to disk\n"));
818 michael@paquier.xyz 671 : 1 : printf(_(" --no-table-access-method do not dump table access methods\n"));
4349 peter_e@gmx.net 672 : 1 : printf(_(" --no-tablespaces do not dump tablespace assignments\n"));
1061 michael@paquier.xyz 673 : 1 : printf(_(" --no-toast-compression do not dump TOAST compression methods\n"));
4349 peter_e@gmx.net 674 : 1 : printf(_(" --no-unlogged-table-data do not dump unlogged table data\n"));
2102 tmunro@postgresql.or 675 : 1 : printf(_(" --on-conflict-do-nothing add ON CONFLICT DO NOTHING to INSERT commands\n"));
4349 peter_e@gmx.net 676 : 1 : printf(_(" --quote-all-identifiers quote all identifiers, even if not key words\n"));
1766 alvherre@alvh.no-ip. 677 : 1 : printf(_(" --rows-per-insert=NROWS number of rows per INSERT; implies --inserts\n"));
6399 peter_e@gmx.net 678 : 1 : printf(_(" --use-set-session-authorization\n"
679 : : " use SET SESSION AUTHORIZATION commands instead of\n"
680 : : " ALTER OWNER commands to set ownership\n"));
681 : :
7849 682 : 1 : printf(_("\nConnection options:\n"));
4066 heikki.linnakangas@i 683 : 1 : printf(_(" -d, --dbname=CONNSTR connect using connection string\n"));
7613 bruce@momjian.us 684 : 1 : printf(_(" -h, --host=HOSTNAME database server host or socket directory\n"));
5527 peter_e@gmx.net 685 : 1 : printf(_(" -l, --database=DBNAME alternative default database\n"));
7849 686 : 1 : printf(_(" -p, --port=PORT database server port number\n"));
687 : 1 : printf(_(" -U, --username=NAME connect as specified database user\n"));
5526 688 : 1 : printf(_(" -w, --no-password never prompt for password\n"));
7849 689 : 1 : printf(_(" -W, --password force password prompt (should happen automatically)\n"));
4708 690 : 1 : printf(_(" --role=ROLENAME do SET ROLE before dump\n"));
691 : :
5527 692 : 1 : printf(_("\nIf -f/--file is not used, then the SQL script will be written to the standard\n"
693 : : "output.\n\n"));
1507 peter@eisentraut.org 694 : 1 : printf(_("Report bugs to <%s>.\n"), PACKAGE_BUGREPORT);
695 : 1 : printf(_("%s home page: <%s>\n"), PACKAGE_NAME, PACKAGE_URL);
7901 peter_e@gmx.net 696 : 1 : }
697 : :
698 : :
699 : : /*
700 : : * Drop roles
701 : : */
702 : : static void
5482 tgl@sss.pgh.pa.us 703 : 1 : dropRoles(PGconn *conn)
704 : : {
2595 simon@2ndQuadrant.co 705 : 1 : PQExpBuffer buf = createPQExpBuffer();
706 : : PGresult *res;
707 : : int i_rolname;
708 : : int i;
709 : :
2882 sfrost@snowman.net 710 [ + - ]: 1 : if (server_version >= 90600)
2595 simon@2ndQuadrant.co 711 : 1 : printfPQExpBuffer(buf,
712 : : "SELECT rolname "
713 : : "FROM %s "
714 : : "WHERE rolname !~ '^pg_' "
715 : : "ORDER BY 1", role_catalog);
716 : : else
2595 simon@2ndQuadrant.co 717 :UBC 0 : printfPQExpBuffer(buf,
718 : : "SELECT rolname "
719 : : "FROM %s "
720 : : "ORDER BY 1", role_catalog);
721 : :
2595 simon@2ndQuadrant.co 722 :CBC 1 : res = executeQuery(conn, buf->data);
723 : :
5482 tgl@sss.pgh.pa.us 724 : 1 : i_rolname = PQfnumber(res, "rolname");
725 : :
726 [ + - ]: 1 : if (PQntuples(res) > 0)
727 : 1 : fprintf(OPF, "--\n-- Drop roles\n--\n\n");
728 : :
729 [ + + ]: 4 : for (i = 0; i < PQntuples(res); i++)
730 : : {
731 : : const char *rolename;
732 : :
733 : 3 : rolename = PQgetvalue(res, i, i_rolname);
734 : :
3695 alvherre@alvh.no-ip. 735 :UBC 0 : fprintf(OPF, "DROP ROLE %s%s;\n",
3695 alvherre@alvh.no-ip. 736 [ - + ]:CBC 3 : if_exists ? "IF EXISTS " : "",
737 : : fmtId(rolename));
738 : : }
739 : :
5482 tgl@sss.pgh.pa.us 740 : 1 : PQclear(res);
2595 simon@2ndQuadrant.co 741 : 1 : destroyPQExpBuffer(buf);
742 : :
5482 tgl@sss.pgh.pa.us 743 : 1 : fprintf(OPF, "\n\n");
744 : 1 : }
745 : :
746 : : /*
747 : : * Dump roles
748 : : */
749 : : static void
6761 750 : 24 : dumpRoles(PGconn *conn)
751 : : {
6855 752 : 24 : PQExpBuffer buf = createPQExpBuffer();
753 : : PGresult *res;
754 : : int i_oid,
755 : : i_rolname,
756 : : i_rolsuper,
757 : : i_rolinherit,
758 : : i_rolcreaterole,
759 : : i_rolcreatedb,
760 : : i_rolcanlogin,
761 : : i_rolconnlimit,
762 : : i_rolpassword,
763 : : i_rolvaliduntil,
764 : : i_rolreplication,
765 : : i_rolbypassrls,
766 : : i_rolcomment,
767 : : i_is_current_user;
768 : : int i;
769 : :
770 : : /*
771 : : * Notes: rolconfig is dumped later, and pg_authid must be used for
772 : : * extracting rolcomment regardless of role_catalog.
773 : : */
2928 sfrost@snowman.net 774 [ + - ]: 24 : if (server_version >= 90600)
3495 775 : 24 : printfPQExpBuffer(buf,
776 : : "SELECT oid, rolname, rolsuper, rolinherit, "
777 : : "rolcreaterole, rolcreatedb, "
778 : : "rolcanlogin, rolconnlimit, rolpassword, "
779 : : "rolvaliduntil, rolreplication, rolbypassrls, "
780 : : "pg_catalog.shobj_description(oid, 'pg_authid') as rolcomment, "
781 : : "rolname = current_user AS is_current_user "
782 : : "FROM %s "
783 : : "WHERE rolname !~ '^pg_' "
784 : : "ORDER BY 2", role_catalog);
2900 sfrost@snowman.net 785 [ # # ]:UBC 0 : else if (server_version >= 90500)
786 : 0 : printfPQExpBuffer(buf,
787 : : "SELECT oid, rolname, rolsuper, rolinherit, "
788 : : "rolcreaterole, rolcreatedb, "
789 : : "rolcanlogin, rolconnlimit, rolpassword, "
790 : : "rolvaliduntil, rolreplication, rolbypassrls, "
791 : : "pg_catalog.shobj_description(oid, 'pg_authid') as rolcomment, "
792 : : "rolname = current_user AS is_current_user "
793 : : "FROM %s "
794 : : "ORDER BY 2", role_catalog);
795 : : else
6832 tgl@sss.pgh.pa.us 796 : 0 : printfPQExpBuffer(buf,
797 : : "SELECT oid, rolname, rolsuper, rolinherit, "
798 : : "rolcreaterole, rolcreatedb, "
799 : : "rolcanlogin, rolconnlimit, rolpassword, "
800 : : "rolvaliduntil, rolreplication, "
801 : : "false as rolbypassrls, "
802 : : "pg_catalog.shobj_description(oid, 'pg_authid') as rolcomment, "
803 : : "rolname = current_user AS is_current_user "
804 : : "FROM %s "
805 : : "ORDER BY 2", role_catalog);
806 : :
6832 tgl@sss.pgh.pa.us 807 :CBC 24 : res = executeQuery(conn, buf->data);
808 : :
4846 bruce@momjian.us 809 : 24 : i_oid = PQfnumber(res, "oid");
6832 tgl@sss.pgh.pa.us 810 : 24 : i_rolname = PQfnumber(res, "rolname");
811 : 24 : i_rolsuper = PQfnumber(res, "rolsuper");
812 : 24 : i_rolinherit = PQfnumber(res, "rolinherit");
813 : 24 : i_rolcreaterole = PQfnumber(res, "rolcreaterole");
814 : 24 : i_rolcreatedb = PQfnumber(res, "rolcreatedb");
815 : 24 : i_rolcanlogin = PQfnumber(res, "rolcanlogin");
816 : 24 : i_rolconnlimit = PQfnumber(res, "rolconnlimit");
817 : 24 : i_rolpassword = PQfnumber(res, "rolpassword");
818 : 24 : i_rolvaliduntil = PQfnumber(res, "rolvaliduntil");
4855 magnus@hagander.net 819 : 24 : i_rolreplication = PQfnumber(res, "rolreplication");
3495 sfrost@snowman.net 820 : 24 : i_rolbypassrls = PQfnumber(res, "rolbypassrls");
6636 bruce@momjian.us 821 : 24 : i_rolcomment = PQfnumber(res, "rolcomment");
4150 822 : 24 : i_is_current_user = PQfnumber(res, "is_current_user");
823 : :
6761 tgl@sss.pgh.pa.us 824 [ + - ]: 24 : if (PQntuples(res) > 0)
6289 bruce@momjian.us 825 : 24 : fprintf(OPF, "--\n-- Roles\n--\n\n");
826 : :
7901 peter_e@gmx.net 827 [ + + ]: 88 : for (i = 0; i < PQntuples(res); i++)
828 : : {
829 : : const char *rolename;
830 : : Oid auth_oid;
831 : :
4652 rhaas@postgresql.org 832 : 64 : auth_oid = atooid(PQgetvalue(res, i, i_oid));
6832 tgl@sss.pgh.pa.us 833 : 64 : rolename = PQgetvalue(res, i, i_rolname);
834 : :
2866 rhaas@postgresql.org 835 [ - + ]: 64 : if (strncmp(rolename, "pg_", 3) == 0)
836 : : {
1840 peter@eisentraut.org 837 :UBC 0 : pg_log_warning("role name starting with \"pg_\" skipped (%s)",
838 : : rolename);
2928 sfrost@snowman.net 839 : 0 : continue;
840 : : }
841 : :
6761 tgl@sss.pgh.pa.us 842 :CBC 64 : resetPQExpBuffer(buf);
843 : :
4846 bruce@momjian.us 844 [ + + ]: 64 : if (binary_upgrade)
845 : : {
3800 heikki.linnakangas@i 846 : 4 : appendPQExpBufferStr(buf, "\n-- For binary upgrade, must preserve pg_authid.oid\n");
4846 bruce@momjian.us 847 : 4 : appendPQExpBuffer(buf,
848 : : "SELECT pg_catalog.binary_upgrade_set_next_pg_authid_oid('%u'::pg_catalog.oid);\n\n",
849 : : auth_oid);
850 : : }
851 : :
852 : : /*
853 : : * We dump CREATE ROLE followed by ALTER ROLE to ensure that the role
854 : : * will acquire the right properties even if it already exists (ie, it
855 : : * won't hurt for the CREATE to fail). This is particularly important
856 : : * for the role we are connected as, since even with --clean we will
857 : : * have failed to drop it. binary_upgrade cannot generate any errors,
858 : : * so we assume the current role is already created.
859 : : */
4150 860 [ + + ]: 64 : if (!binary_upgrade ||
861 [ - + ]: 4 : strcmp(PQgetvalue(res, i, i_is_current_user), "f") == 0)
4153 862 : 60 : appendPQExpBuffer(buf, "CREATE ROLE %s;\n", fmtId(rolename));
6761 tgl@sss.pgh.pa.us 863 : 64 : appendPQExpBuffer(buf, "ALTER ROLE %s WITH", fmtId(rolename));
864 : :
6832 865 [ + + ]: 64 : if (strcmp(PQgetvalue(res, i, i_rolsuper), "t") == 0)
3800 heikki.linnakangas@i 866 : 52 : appendPQExpBufferStr(buf, " SUPERUSER");
867 : : else
868 : 12 : appendPQExpBufferStr(buf, " NOSUPERUSER");
869 : :
6832 tgl@sss.pgh.pa.us 870 [ + - ]: 64 : if (strcmp(PQgetvalue(res, i, i_rolinherit), "t") == 0)
3800 heikki.linnakangas@i 871 : 64 : appendPQExpBufferStr(buf, " INHERIT");
872 : : else
3800 heikki.linnakangas@i 873 :UBC 0 : appendPQExpBufferStr(buf, " NOINHERIT");
874 : :
6832 tgl@sss.pgh.pa.us 875 [ + + ]:CBC 64 : if (strcmp(PQgetvalue(res, i, i_rolcreaterole), "t") == 0)
3800 heikki.linnakangas@i 876 : 52 : appendPQExpBufferStr(buf, " CREATEROLE");
877 : : else
878 : 12 : appendPQExpBufferStr(buf, " NOCREATEROLE");
879 : :
6832 tgl@sss.pgh.pa.us 880 [ + + ]: 64 : if (strcmp(PQgetvalue(res, i, i_rolcreatedb), "t") == 0)
3800 heikki.linnakangas@i 881 : 52 : appendPQExpBufferStr(buf, " CREATEDB");
882 : : else
883 : 12 : appendPQExpBufferStr(buf, " NOCREATEDB");
884 : :
6832 tgl@sss.pgh.pa.us 885 [ + + ]: 64 : if (strcmp(PQgetvalue(res, i, i_rolcanlogin), "t") == 0)
3800 heikki.linnakangas@i 886 : 53 : appendPQExpBufferStr(buf, " LOGIN");
887 : : else
888 : 11 : appendPQExpBufferStr(buf, " NOLOGIN");
889 : :
4855 magnus@hagander.net 890 [ + + ]: 64 : if (strcmp(PQgetvalue(res, i, i_rolreplication), "t") == 0)
3800 heikki.linnakangas@i 891 : 24 : appendPQExpBufferStr(buf, " REPLICATION");
892 : : else
893 : 40 : appendPQExpBufferStr(buf, " NOREPLICATION");
894 : :
3495 sfrost@snowman.net 895 [ + + ]: 64 : if (strcmp(PQgetvalue(res, i, i_rolbypassrls), "t") == 0)
896 : 24 : appendPQExpBufferStr(buf, " BYPASSRLS");
897 : : else
898 : 40 : appendPQExpBufferStr(buf, " NOBYPASSRLS");
899 : :
6832 tgl@sss.pgh.pa.us 900 [ - + ]: 64 : if (strcmp(PQgetvalue(res, i, i_rolconnlimit), "-1") != 0)
6832 tgl@sss.pgh.pa.us 901 :UBC 0 : appendPQExpBuffer(buf, " CONNECTION LIMIT %s",
902 : : PQgetvalue(res, i, i_rolconnlimit));
903 : :
904 : :
2595 simon@2ndQuadrant.co 905 [ - + - - ]:CBC 64 : if (!PQgetisnull(res, i, i_rolpassword) && !no_role_passwords)
906 : : {
3800 heikki.linnakangas@i 907 :UBC 0 : appendPQExpBufferStr(buf, " PASSWORD ");
6531 tgl@sss.pgh.pa.us 908 : 0 : appendStringLiteralConn(buf, PQgetvalue(res, i, i_rolpassword), conn);
909 : : }
910 : :
6832 tgl@sss.pgh.pa.us 911 [ - + ]:CBC 64 : if (!PQgetisnull(res, i, i_rolvaliduntil))
7625 tgl@sss.pgh.pa.us 912 :UBC 0 : appendPQExpBuffer(buf, " VALID UNTIL '%s'",
913 : : PQgetvalue(res, i, i_rolvaliduntil));
914 : :
3800 heikki.linnakangas@i 915 :CBC 64 : appendPQExpBufferStr(buf, ";\n");
916 : :
2271 tgl@sss.pgh.pa.us 917 [ + - - + ]: 64 : if (!no_comments && !PQgetisnull(res, i, i_rolcomment))
918 : : {
6636 bruce@momjian.us 919 :UBC 0 : appendPQExpBuffer(buf, "COMMENT ON ROLE %s IS ", fmtId(rolename));
6531 tgl@sss.pgh.pa.us 920 : 0 : appendStringLiteralConn(buf, PQgetvalue(res, i, i_rolcomment), conn);
3800 heikki.linnakangas@i 921 : 0 : appendPQExpBufferStr(buf, ";\n");
922 : : }
923 : :
852 tgl@sss.pgh.pa.us 924 [ + - ]:CBC 64 : if (!no_security_labels)
4652 rhaas@postgresql.org 925 : 64 : buildShSecLabels(conn, "pg_authid", auth_oid,
926 : : "ROLE", rolename,
927 : : buf);
928 : :
6289 bruce@momjian.us 929 : 64 : fprintf(OPF, "%s", buf->data);
930 : : }
931 : :
932 : : /*
933 : : * Dump configuration settings for roles after all roles have been dumped.
934 : : * We do it this way because config settings for roles could mention the
935 : : * names of other roles.
936 : : */
852 tgl@sss.pgh.pa.us 937 [ + - ]: 24 : if (PQntuples(res) > 0)
938 : 24 : fprintf(OPF, "\n--\n-- User Configurations\n--\n");
939 : :
2741 940 [ + + ]: 88 : for (i = 0; i < PQntuples(res); i++)
941 : 64 : dumpUserConfig(conn, PQgetvalue(res, i, i_rolname));
942 : :
7901 peter_e@gmx.net 943 : 24 : PQclear(res);
944 : :
6289 bruce@momjian.us 945 : 24 : fprintf(OPF, "\n\n");
946 : :
6855 tgl@sss.pgh.pa.us 947 : 24 : destroyPQExpBuffer(buf);
7901 peter_e@gmx.net 948 : 24 : }
949 : :
950 : :
951 : : /*
952 : : * Dump role memberships.
953 : : *
954 : : * Note: we expect dumpRoles already created all the roles, but there is
955 : : * no membership yet.
956 : : */
957 : : static void
6832 tgl@sss.pgh.pa.us 958 : 24 : dumpRoleMembership(PGconn *conn)
959 : : {
2595 simon@2ndQuadrant.co 960 : 24 : PQExpBuffer buf = createPQExpBuffer();
331 tgl@sss.pgh.pa.us 961 : 24 : PQExpBuffer optbuf = createPQExpBuffer();
962 : : PGresult *res;
601 rhaas@postgresql.org 963 : 24 : int start = 0,
964 : : end,
965 : : total;
966 : : bool dump_grantors;
967 : : bool dump_grant_options;
968 : : int i_inherit_option;
969 : : int i_set_option;
970 : :
971 : : /*
972 : : * Previous versions of PostgreSQL didn't used to track the grantor very
973 : : * carefully in the backend, and the grantor could be any user even if
974 : : * they didn't have ADMIN OPTION on the role, or a user that no longer
975 : : * existed. To avoid dump and restore failures, don't dump the grantor
976 : : * when talking to an old server version.
977 : : */
978 : 24 : dump_grantors = (PQserverVersion(conn) >= 160000);
979 : :
980 : : /*
981 : : * Previous versions of PostgreSQL also did not have grant-level options.
982 : : */
513 983 : 24 : dump_grant_options = (server_version >= 160000);
984 : :
985 : : /* Generate and execute query. */
601 986 : 24 : printfPQExpBuffer(buf, "SELECT ur.rolname AS role, "
987 : : "um.rolname AS member, "
988 : : "ug.oid AS grantorid, "
989 : : "ug.rolname AS grantor, "
990 : : "a.admin_option");
513 991 [ + - ]: 24 : if (dump_grant_options)
992 : 24 : appendPQExpBufferStr(buf, ", a.inherit_option, a.set_option");
598 993 : 24 : appendPQExpBuffer(buf, " FROM pg_auth_members a "
994 : : "LEFT JOIN %s ur on ur.oid = a.roleid "
995 : : "LEFT JOIN %s um on um.oid = a.member "
996 : : "LEFT JOIN %s ug on ug.oid = a.grantor "
997 : : "WHERE NOT (ur.rolname ~ '^pg_' AND um.rolname ~ '^pg_')"
998 : : "ORDER BY 1,2,4", role_catalog, role_catalog, role_catalog);
2595 simon@2ndQuadrant.co 999 : 24 : res = executeQuery(conn, buf->data);
598 rhaas@postgresql.org 1000 : 24 : i_inherit_option = PQfnumber(res, "inherit_option");
513 1001 : 24 : i_set_option = PQfnumber(res, "set_option");
1002 : :
6761 tgl@sss.pgh.pa.us 1003 [ - + ]: 24 : if (PQntuples(res) > 0)
6289 bruce@momjian.us 1004 :UBC 0 : fprintf(OPF, "--\n-- Role memberships\n--\n\n");
1005 : :
1006 : : /*
1007 : : * We can't dump these GRANT commands in arbitrary order, because a role
1008 : : * that is named as a grantor must already have ADMIN OPTION on the role
1009 : : * for which it is granting permissions, except for the bootstrap
1010 : : * superuser, who can always be named as the grantor.
1011 : : *
1012 : : * We handle this by considering these grants role by role. For each role,
1013 : : * we initially consider the only allowable grantor to be the bootstrap
1014 : : * superuser. Every time we grant ADMIN OPTION on the role to some user,
1015 : : * that user also becomes an allowable grantor. We make repeated passes
1016 : : * over the grants for the role, each time dumping those whose grantors
1017 : : * are allowable and which we haven't done yet. Eventually this should let
1018 : : * us dump all the grants.
1019 : : */
601 rhaas@postgresql.org 1020 :CBC 24 : total = PQntuples(res);
1021 [ - + ]: 24 : while (start < total)
1022 : : {
601 rhaas@postgresql.org 1023 :UBC 0 : char *role = PQgetvalue(res, start, 0);
1024 : : int i;
1025 : : bool *done;
1026 : : int remaining;
1027 : 0 : int prev_remaining = 0;
1028 : : rolename_hash *ht;
1029 : :
1030 : : /* All memberships for a single role should be adjacent. */
1031 [ # # ]: 0 : for (end = start; end < total; ++end)
1032 : : {
1033 : : char *otherrole;
1034 : :
1035 : 0 : otherrole = PQgetvalue(res, end, 0);
1036 [ # # ]: 0 : if (strcmp(role, otherrole) != 0)
1037 : 0 : break;
1038 : : }
1039 : :
1040 : 0 : role = PQgetvalue(res, start, 0);
1041 : 0 : remaining = end - start;
1042 : 0 : done = pg_malloc0(remaining * sizeof(bool));
1043 : 0 : ht = rolename_create(remaining, NULL);
1044 : :
1045 : : /*
1046 : : * Make repeated passes over the grants for this role until all have
1047 : : * been dumped.
1048 : : */
1049 [ # # ]: 0 : while (remaining > 0)
1050 : : {
1051 : : /*
1052 : : * We should make progress on every iteration, because a notional
1053 : : * graph whose vertices are grants and whose edges point from
1054 : : * grantors to members should be connected and acyclic. If we fail
1055 : : * to make progress, either we or the server have messed up.
1056 : : */
1057 [ # # ]: 0 : if (remaining == prev_remaining)
1058 : : {
1059 : 0 : pg_log_error("could not find a legal dump ordering for memberships in role \"%s\"",
1060 : : role);
1061 : 0 : PQfinish(conn);
1062 : 0 : exit_nicely(1);
1063 : : }
354 dgustafsson@postgres 1064 : 0 : prev_remaining = remaining;
1065 : :
1066 : : /* Make one pass over the grants for this role. */
601 rhaas@postgresql.org 1067 [ # # ]: 0 : for (i = start; i < end; ++i)
1068 : : {
1069 : : char *member;
1070 : : char *admin_option;
1071 : : char *grantorid;
1072 : : char *grantor;
513 1073 : 0 : char *set_option = "true";
1074 : : bool found;
1075 : :
1076 : : /* If we already did this grant, don't do it again. */
601 1077 [ # # ]: 0 : if (done[i - start])
1078 : 0 : continue;
1079 : :
1080 : 0 : member = PQgetvalue(res, i, 1);
1081 : 0 : grantorid = PQgetvalue(res, i, 2);
1082 : 0 : grantor = PQgetvalue(res, i, 3);
1083 : 0 : admin_option = PQgetvalue(res, i, 4);
513 1084 [ # # ]: 0 : if (dump_grant_options)
1085 : 0 : set_option = PQgetvalue(res, i, i_set_option);
1086 : :
1087 : : /*
1088 : : * If we're not dumping grantors or if the grantor is the
1089 : : * bootstrap superuser, it's fine to dump this now. Otherwise,
1090 : : * it's got to be someone who has already been granted ADMIN
1091 : : * OPTION.
1092 : : */
601 1093 [ # # ]: 0 : if (dump_grantors &&
1094 [ # # # # ]: 0 : atooid(grantorid) != BOOTSTRAP_SUPERUSERID &&
1095 : 0 : rolename_lookup(ht, grantor) == NULL)
1096 : 0 : continue;
1097 : :
1098 : : /* Remember that we did this so that we don't do it again. */
1099 : 0 : done[i - start] = true;
1100 : 0 : --remaining;
1101 : :
1102 : : /*
1103 : : * If ADMIN OPTION is being granted, remember that grants
1104 : : * listing this member as the grantor can now be dumped.
1105 : : */
1106 [ # # ]: 0 : if (*admin_option == 't')
1107 : 0 : rolename_insert(ht, member, &found);
1108 : :
1109 : : /* Generate the actual GRANT statement. */
598 1110 : 0 : resetPQExpBuffer(optbuf);
601 1111 : 0 : fprintf(OPF, "GRANT %s", fmtId(role));
1112 : 0 : fprintf(OPF, " TO %s", fmtId(member));
1113 [ # # ]: 0 : if (*admin_option == 't')
598 1114 : 0 : appendPQExpBufferStr(optbuf, "ADMIN OPTION");
513 1115 [ # # ]: 0 : if (dump_grant_options)
1116 : : {
1117 : : char *inherit_option;
1118 : :
598 1119 [ # # ]: 0 : if (optbuf->data[0] != '\0')
1120 : 0 : appendPQExpBufferStr(optbuf, ", ");
1121 : 0 : inherit_option = PQgetvalue(res, i, i_inherit_option);
1122 : 0 : appendPQExpBuffer(optbuf, "INHERIT %s",
1123 [ # # ]: 0 : *inherit_option == 't' ?
1124 : : "TRUE" : "FALSE");
1125 : : }
513 1126 [ # # ]: 0 : if (*set_option != 't')
1127 : : {
1128 [ # # ]: 0 : if (optbuf->data[0] != '\0')
1129 : 0 : appendPQExpBufferStr(optbuf, ", ");
1130 : 0 : appendPQExpBuffer(optbuf, "SET FALSE");
1131 : : }
598 1132 [ # # ]: 0 : if (optbuf->data[0] != '\0')
1133 : 0 : fprintf(OPF, " WITH %s", optbuf->data);
601 1134 [ # # ]: 0 : if (dump_grantors)
1135 : 0 : fprintf(OPF, " GRANTED BY %s", fmtId(grantor));
1136 : 0 : fprintf(OPF, ";\n");
1137 : : }
1138 : : }
1139 : :
1140 : 0 : rolename_destroy(ht);
1141 : 0 : pg_free(done);
1142 : 0 : start = end;
1143 : : }
1144 : :
6832 tgl@sss.pgh.pa.us 1145 :CBC 24 : PQclear(res);
2595 simon@2ndQuadrant.co 1146 : 24 : destroyPQExpBuffer(buf);
1147 : :
6289 bruce@momjian.us 1148 : 24 : fprintf(OPF, "\n\n");
6832 tgl@sss.pgh.pa.us 1149 : 24 : }
1150 : :
1151 : :
1152 : : /*
1153 : : * Dump role configuration parameter privileges. This code is used for 15.0
1154 : : * and later servers.
1155 : : *
1156 : : * Note: we expect dumpRoles already created all the roles, but there are
1157 : : * no per-role configuration parameter privileges yet.
1158 : : */
1159 : : static void
739 1160 : 24 : dumpRoleGUCPrivs(PGconn *conn)
1161 : : {
1162 : : PGresult *res;
1163 : : int i;
1164 : :
1165 : : /*
1166 : : * Get all parameters that have non-default acls defined.
1167 : : */
1168 : 24 : res = executeQuery(conn, "SELECT parname, "
1169 : : "pg_catalog.pg_get_userbyid(" CppAsString2(BOOTSTRAP_SUPERUSERID) ") AS parowner, "
1170 : : "paracl, "
1171 : : "pg_catalog.acldefault('p', " CppAsString2(BOOTSTRAP_SUPERUSERID) ") AS acldefault "
1172 : : "FROM pg_catalog.pg_parameter_acl "
1173 : : "ORDER BY 1");
1174 : :
1175 [ + + ]: 24 : if (PQntuples(res) > 0)
1176 : 1 : fprintf(OPF, "--\n-- Role privileges on configuration parameters\n--\n\n");
1177 : :
1178 [ + + ]: 27 : for (i = 0; i < PQntuples(res); i++)
1179 : : {
1180 : 3 : PQExpBuffer buf = createPQExpBuffer();
1181 : 3 : char *parname = PQgetvalue(res, i, 0);
1182 : 3 : char *parowner = PQgetvalue(res, i, 1);
1183 : 3 : char *paracl = PQgetvalue(res, i, 2);
1184 : 3 : char *acldefault = PQgetvalue(res, i, 3);
1185 : : char *fparname;
1186 : :
1187 : : /* needed for buildACLCommands() */
1188 : 3 : fparname = pg_strdup(fmtId(parname));
1189 : :
1190 [ - + ]: 3 : if (!buildACLCommands(fparname, NULL, NULL, "PARAMETER",
1191 : : paracl, acldefault,
1192 : : parowner, "", server_version, buf))
1193 : : {
739 tgl@sss.pgh.pa.us 1194 :UBC 0 : pg_log_error("could not parse ACL list (%s) for parameter \"%s\"",
1195 : : paracl, parname);
1196 : 0 : PQfinish(conn);
1197 : 0 : exit_nicely(1);
1198 : : }
1199 : :
739 tgl@sss.pgh.pa.us 1200 :CBC 3 : fprintf(OPF, "%s", buf->data);
1201 : :
1202 : 3 : free(fparname);
1203 : 3 : destroyPQExpBuffer(buf);
1204 : : }
1205 : :
1206 : 24 : PQclear(res);
1207 : 24 : fprintf(OPF, "\n\n");
1208 : 24 : }
1209 : :
1210 : :
1211 : : /*
1212 : : * Drop tablespaces.
1213 : : */
1214 : : static void
5482 1215 : 1 : dropTablespaces(PGconn *conn)
1216 : : {
1217 : : PGresult *res;
1218 : : int i;
1219 : :
1220 : : /*
1221 : : * Get all tablespaces except built-in ones (which we assume are named
1222 : : * pg_xxx)
1223 : : */
1224 : 1 : res = executeQuery(conn, "SELECT spcname "
1225 : : "FROM pg_catalog.pg_tablespace "
1226 : : "WHERE spcname !~ '^pg_' "
1227 : : "ORDER BY 1");
1228 : :
1229 [ + - ]: 1 : if (PQntuples(res) > 0)
5482 tgl@sss.pgh.pa.us 1230 :GBC 1 : fprintf(OPF, "--\n-- Drop tablespaces\n--\n\n");
1231 : :
5482 tgl@sss.pgh.pa.us 1232 [ + + ]:CBC 2 : for (i = 0; i < PQntuples(res); i++)
1233 : : {
5482 tgl@sss.pgh.pa.us 1234 :GBC 1 : char *spcname = PQgetvalue(res, i, 0);
1235 : :
3695 alvherre@alvh.no-ip. 1236 :UBC 0 : fprintf(OPF, "DROP TABLESPACE %s%s;\n",
3695 alvherre@alvh.no-ip. 1237 [ - + ]:GBC 1 : if_exists ? "IF EXISTS " : "",
1238 : : fmtId(spcname));
1239 : : }
1240 : :
5482 tgl@sss.pgh.pa.us 1241 :CBC 1 : PQclear(res);
1242 : :
1243 : 1 : fprintf(OPF, "\n\n");
1244 : 1 : }
1245 : :
1246 : : /*
1247 : : * Dump tablespaces.
1248 : : */
1249 : : static void
7240 1250 : 19 : dumpTablespaces(PGconn *conn)
1251 : : {
1252 : : PGresult *res;
1253 : : int i;
1254 : :
1255 : : /*
1256 : : * Get all tablespaces except built-in ones (which we assume are named
1257 : : * pg_xxx)
1258 : : */
852 1259 : 19 : res = executeQuery(conn, "SELECT oid, spcname, "
1260 : : "pg_catalog.pg_get_userbyid(spcowner) AS spcowner, "
1261 : : "pg_catalog.pg_tablespace_location(oid), "
1262 : : "spcacl, acldefault('t', spcowner) AS acldefault, "
1263 : : "array_to_string(spcoptions, ', '),"
1264 : : "pg_catalog.shobj_description(oid, 'pg_tablespace') "
1265 : : "FROM pg_catalog.pg_tablespace "
1266 : : "WHERE spcname !~ '^pg_' "
1267 : : "ORDER BY 1");
1268 : :
7209 bruce@momjian.us 1269 [ + + ]: 19 : if (PQntuples(res) > 0)
6289 bruce@momjian.us 1270 :GBC 5 : fprintf(OPF, "--\n-- Tablespaces\n--\n\n");
1271 : :
7240 tgl@sss.pgh.pa.us 1272 [ + + ]:CBC 24 : for (i = 0; i < PQntuples(res); i++)
1273 : : {
7240 tgl@sss.pgh.pa.us 1274 :GBC 5 : PQExpBuffer buf = createPQExpBuffer();
2239 1275 : 5 : Oid spcoid = atooid(PQgetvalue(res, i, 0));
4652 rhaas@postgresql.org 1276 : 5 : char *spcname = PQgetvalue(res, i, 1);
1277 : 5 : char *spcowner = PQgetvalue(res, i, 2);
1278 : 5 : char *spclocation = PQgetvalue(res, i, 3);
1279 : 5 : char *spcacl = PQgetvalue(res, i, 4);
860 tgl@sss.pgh.pa.us 1280 : 5 : char *acldefault = PQgetvalue(res, i, 5);
2828 sfrost@snowman.net 1281 : 5 : char *spcoptions = PQgetvalue(res, i, 6);
1282 : 5 : char *spccomment = PQgetvalue(res, i, 7);
1283 : : char *fspcname;
1284 : :
1285 : : /* needed for buildACLCommands() */
4524 bruce@momjian.us 1286 : 5 : fspcname = pg_strdup(fmtId(spcname));
1287 : :
787 rhaas@postgresql.org 1288 [ - + ]: 5 : if (binary_upgrade)
1289 : : {
787 rhaas@postgresql.org 1290 :UBC 0 : appendPQExpBufferStr(buf, "\n-- For binary upgrade, must preserve pg_tablespace oid\n");
1291 : 0 : appendPQExpBuffer(buf, "SELECT pg_catalog.binary_upgrade_set_next_pg_tablespace_oid('%u'::pg_catalog.oid);\n", spcoid);
1292 : : }
1293 : :
7240 tgl@sss.pgh.pa.us 1294 :GBC 5 : appendPQExpBuffer(buf, "CREATE TABLESPACE %s", fspcname);
1295 : 5 : appendPQExpBuffer(buf, " OWNER %s", fmtId(spcowner));
1296 : :
3800 heikki.linnakangas@i 1297 : 5 : appendPQExpBufferStr(buf, " LOCATION ");
1298 : :
1299 : : /*
1300 : : * In-place tablespaces use a relative path, and need to be dumped
1301 : : * with an empty string as location.
1302 : : */
249 michael@paquier.xyz 1303 [ - + ]:GNC 5 : if (is_absolute_path(spclocation))
249 michael@paquier.xyz 1304 :UNC 0 : appendStringLiteralConn(buf, spclocation, conn);
1305 : : else
249 michael@paquier.xyz 1306 :GNC 5 : appendStringLiteralConn(buf, "", conn);
1307 : :
3800 heikki.linnakangas@i 1308 :GBC 5 : appendPQExpBufferStr(buf, ";\n");
1309 : :
5213 rhaas@postgresql.org 1310 [ + - - + ]: 5 : if (spcoptions && spcoptions[0] != '\0')
5213 rhaas@postgresql.org 1311 :UBC 0 : appendPQExpBuffer(buf, "ALTER TABLESPACE %s SET (%s);\n",
1312 : : fspcname, spcoptions);
1313 : :
1314 : : /* tablespaces can't have initprivs */
1315 : :
7240 tgl@sss.pgh.pa.us 1316 [ + - ]:GBC 5 : if (!skip_acls &&
2239 1317 [ - + ]: 5 : !buildACLCommands(fspcname, NULL, NULL, "TABLESPACE",
1318 : : spcacl, acldefault,
1319 : : spcowner, "", server_version, buf))
1320 : : {
1840 peter@eisentraut.org 1321 :UBC 0 : pg_log_error("could not parse ACL list (%s) for tablespace \"%s\"",
1322 : : spcacl, spcname);
7240 tgl@sss.pgh.pa.us 1323 : 0 : PQfinish(conn);
4441 rhaas@postgresql.org 1324 : 0 : exit_nicely(1);
1325 : : }
1326 : :
2271 tgl@sss.pgh.pa.us 1327 [ + - + - :GBC 5 : if (!no_comments && spccomment && spccomment[0] != '\0')
- + ]
1328 : : {
6636 bruce@momjian.us 1329 :UBC 0 : appendPQExpBuffer(buf, "COMMENT ON TABLESPACE %s IS ", fspcname);
6531 tgl@sss.pgh.pa.us 1330 : 0 : appendStringLiteralConn(buf, spccomment, conn);
3800 heikki.linnakangas@i 1331 : 0 : appendPQExpBufferStr(buf, ";\n");
1332 : : }
1333 : :
852 tgl@sss.pgh.pa.us 1334 [ + - ]:GBC 5 : if (!no_security_labels)
4652 rhaas@postgresql.org 1335 : 5 : buildShSecLabels(conn, "pg_tablespace", spcoid,
1336 : : "TABLESPACE", spcname,
1337 : : buf);
1338 : :
6289 bruce@momjian.us 1339 : 5 : fprintf(OPF, "%s", buf->data);
1340 : :
7240 tgl@sss.pgh.pa.us 1341 : 5 : free(fspcname);
1342 : 5 : destroyPQExpBuffer(buf);
1343 : : }
1344 : :
7240 tgl@sss.pgh.pa.us 1345 :CBC 19 : PQclear(res);
6289 bruce@momjian.us 1346 : 19 : fprintf(OPF, "\n\n");
7240 tgl@sss.pgh.pa.us 1347 : 19 : }
1348 : :
1349 : :
1350 : : /*
1351 : : * Dump commands to drop each database.
1352 : : */
1353 : : static void
5482 tgl@sss.pgh.pa.us 1354 :UBC 0 : dropDBs(PGconn *conn)
1355 : : {
1356 : : PGresult *res;
1357 : : int i;
1358 : :
1359 : : /*
1360 : : * Skip databases marked not datallowconn, since we'd be unable to connect
1361 : : * to them anyway. This must agree with dumpDatabases().
1362 : : */
2741 1363 : 0 : res = executeQuery(conn,
1364 : : "SELECT datname "
1365 : : "FROM pg_database d "
1366 : : "WHERE datallowconn AND datconnlimit != -2 "
1367 : : "ORDER BY datname");
1368 : :
5482 1369 [ # # ]: 0 : if (PQntuples(res) > 0)
2274 1370 : 0 : fprintf(OPF, "--\n-- Drop databases (except postgres and template1)\n--\n\n");
1371 : :
5482 1372 [ # # ]: 0 : for (i = 0; i < PQntuples(res); i++)
1373 : : {
1374 : 0 : char *dbname = PQgetvalue(res, i, 0);
1375 : :
1376 : : /*
1377 : : * Skip "postgres" and "template1"; dumpDatabases() will deal with
1378 : : * them specially. Also, be sure to skip "template0", even if for
1379 : : * some reason it's not marked !datallowconn.
1380 : : */
1381 [ # # ]: 0 : if (strcmp(dbname, "template1") != 0 &&
2274 1382 [ # # ]: 0 : strcmp(dbname, "template0") != 0 &&
5482 1383 [ # # ]: 0 : strcmp(dbname, "postgres") != 0)
1384 : : {
3695 alvherre@alvh.no-ip. 1385 : 0 : fprintf(OPF, "DROP DATABASE %s%s;\n",
1386 [ # # ]: 0 : if_exists ? "IF EXISTS " : "",
1387 : : fmtId(dbname));
1388 : : }
1389 : : }
1390 : :
5482 tgl@sss.pgh.pa.us 1391 : 0 : PQclear(res);
1392 : :
1393 : 0 : fprintf(OPF, "\n\n");
1394 : 0 : }
1395 : :
1396 : :
1397 : : /*
1398 : : * Dump user-specific configuration
1399 : : */
1400 : : static void
7900 peter_e@gmx.net 1401 :CBC 64 : dumpUserConfig(PGconn *conn, const char *username)
1402 : : {
1403 : 64 : PQExpBuffer buf = createPQExpBuffer();
1404 : : PGresult *res;
1405 : :
333 akorotkov@postgresql 1406 : 64 : printfPQExpBuffer(buf, "SELECT unnest(setconfig) FROM pg_db_role_setting "
1407 : : "WHERE setdatabase = 0 AND setrole = "
1408 : : "(SELECT oid FROM %s WHERE rolname = ",
1409 : : role_catalog);
852 tgl@sss.pgh.pa.us 1410 : 64 : appendStringLiteralConn(buf, username, conn);
1411 : 64 : appendPQExpBufferChar(buf, ')');
1412 : :
1413 : 64 : res = executeQuery(conn, buf->data);
1414 : :
1415 [ - + ]: 64 : if (PQntuples(res) > 0)
852 tgl@sss.pgh.pa.us 1416 :UBC 0 : fprintf(OPF, "\n--\n-- User Config \"%s\"\n--\n\n", username);
1417 : :
852 tgl@sss.pgh.pa.us 1418 [ - + ]:CBC 64 : for (int i = 0; i < PQntuples(res); i++)
1419 : : {
852 tgl@sss.pgh.pa.us 1420 :UBC 0 : resetPQExpBuffer(buf);
333 akorotkov@postgresql 1421 : 0 : makeAlterConfigCommand(conn, PQgetvalue(res, i, 0),
1422 : : "ROLE", username, NULL, NULL,
1423 : : buf);
852 tgl@sss.pgh.pa.us 1424 : 0 : fprintf(OPF, "%s", buf->data);
1425 : : }
1426 : :
852 tgl@sss.pgh.pa.us 1427 :CBC 64 : PQclear(res);
1428 : :
7900 peter_e@gmx.net 1429 : 64 : destroyPQExpBuffer(buf);
1430 : 64 : }
1431 : :
1432 : : /*
1433 : : * Find a list of database names that match the given patterns.
1434 : : * See also expand_table_name_patterns() in pg_dump.c
1435 : : */
1436 : : static void
1871 andrew@dunslane.net 1437 : 26 : expand_dbname_patterns(PGconn *conn,
1438 : : SimpleStringList *patterns,
1439 : : SimpleStringList *names)
1440 : : {
1441 : : PQExpBuffer query;
1442 : : PGresult *res;
1443 : :
1444 [ + + ]: 26 : if (patterns->head == NULL)
1445 : 21 : return; /* nothing to do */
1446 : :
1447 : 5 : query = createPQExpBuffer();
1448 : :
1449 : : /*
1450 : : * The loop below runs multiple SELECTs, which might sometimes result in
1451 : : * duplicate entries in the name list, but we don't care, since all we're
1452 : : * going to do is test membership of the list.
1453 : : */
1454 : :
1455 [ + + ]: 8 : for (SimpleStringListCell *cell = patterns->head; cell; cell = cell->next)
1456 : : {
1457 : : int dotcnt;
1458 : :
1746 drowley@postgresql.o 1459 : 5 : appendPQExpBufferStr(query,
1460 : : "SELECT datname FROM pg_catalog.pg_database n\n");
1871 andrew@dunslane.net 1461 : 5 : processSQLNamePattern(conn, query, cell->val, false,
1462 : : false, NULL, "datname", NULL, NULL, NULL,
1463 : : &dotcnt);
1464 : :
725 rhaas@postgresql.org 1465 [ + + ]: 5 : if (dotcnt > 0)
1466 : : {
1467 : 2 : pg_log_error("improper qualified name (too many dotted names): %s",
1468 : : cell->val);
1469 : 2 : PQfinish(conn);
1470 : 2 : exit_nicely(1);
1471 : : }
1472 : :
1871 andrew@dunslane.net 1473 : 3 : res = executeQuery(conn, query->data);
1474 [ + + ]: 7 : for (int i = 0; i < PQntuples(res); i++)
1475 : : {
1476 : 4 : simple_string_list_append(names, PQgetvalue(res, i, 0));
1477 : : }
1478 : :
1479 : 3 : PQclear(res);
1480 : 3 : resetPQExpBuffer(query);
1481 : : }
1482 : :
1483 : 3 : destroyPQExpBuffer(query);
1484 : : }
1485 : :
1486 : : /*
1487 : : * Dump contents of databases.
1488 : : */
1489 : : static void
7901 peter_e@gmx.net 1490 : 12 : dumpDatabases(PGconn *conn)
1491 : : {
1492 : : PGresult *res;
1493 : : int i;
1494 : :
1495 : : /*
1496 : : * Skip databases marked not datallowconn, since we'd be unable to connect
1497 : : * to them anyway. This must agree with dropDBs().
1498 : : *
1499 : : * We arrange for template1 to be processed first, then we process other
1500 : : * DBs in alphabetical order. If we just did them all alphabetically, we
1501 : : * might find ourselves trying to drop the "postgres" database while still
1502 : : * connected to it. This makes trying to run the restore script while
1503 : : * connected to "template1" a bad idea, but there's no fixed order that
1504 : : * doesn't have some failure mode with --clean.
1505 : : */
2274 tgl@sss.pgh.pa.us 1506 : 12 : res = executeQuery(conn,
1507 : : "SELECT datname "
1508 : : "FROM pg_database d "
1509 : : "WHERE datallowconn AND datconnlimit != -2 "
1510 : : "ORDER BY (datname <> 'template1'), datname");
1511 : :
1871 andrew@dunslane.net 1512 [ + - ]: 12 : if (PQntuples(res) > 0)
1513 : 12 : fprintf(OPF, "--\n-- Databases\n--\n\n");
1514 : :
7901 peter_e@gmx.net 1515 [ + + ]: 65 : for (i = 0; i < PQntuples(res); i++)
1516 : : {
2274 tgl@sss.pgh.pa.us 1517 : 54 : char *dbname = PQgetvalue(res, i, 0);
1518 : : const char *create_opts;
1519 : : int ret;
1520 : :
1521 : : /* Skip template0, even if it's not marked !datallowconn. */
1522 [ - + ]: 54 : if (strcmp(dbname, "template0") == 0)
2274 tgl@sss.pgh.pa.us 1523 :UBC 0 : continue;
1524 : :
1525 : : /* Skip any explicitly excluded database */
1871 andrew@dunslane.net 1526 [ + + ]:CBC 54 : if (simple_string_list_member(&database_exclude_names, dbname))
1527 : : {
1741 peter@eisentraut.org 1528 : 4 : pg_log_info("excluding database \"%s\"", dbname);
1871 andrew@dunslane.net 1529 : 4 : continue;
1530 : : }
1531 : :
1741 peter@eisentraut.org 1532 : 50 : pg_log_info("dumping database \"%s\"", dbname);
1533 : :
1871 andrew@dunslane.net 1534 : 50 : fprintf(OPF, "--\n-- Database \"%s\" dump\n--\n\n", dbname);
1535 : :
1536 : : /*
1537 : : * We assume that "template1" and "postgres" already exist in the
1538 : : * target installation. dropDBs() won't have removed them, for fear
1539 : : * of removing the DB the restore script is initially connected to. If
1540 : : * --clean was specified, tell pg_dump to drop and recreate them;
1541 : : * otherwise we'll merely restore their contents. Other databases
1542 : : * should simply be created.
1543 : : */
2274 tgl@sss.pgh.pa.us 1544 [ + + + + ]: 50 : if (strcmp(dbname, "template1") == 0 || strcmp(dbname, "postgres") == 0)
1545 : : {
1546 [ - + ]: 22 : if (output_clean)
2274 tgl@sss.pgh.pa.us 1547 :UBC 0 : create_opts = "--clean --create";
1548 : : else
1549 : : {
2274 tgl@sss.pgh.pa.us 1550 :CBC 22 : create_opts = "";
1551 : : /* Since pg_dump won't emit a \connect command, we must */
1552 : 22 : fprintf(OPF, "\\connect %s\n\n", dbname);
1553 : : }
1554 : : }
1555 : : else
1556 : 28 : create_opts = "--create";
1557 : :
6289 bruce@momjian.us 1558 [ + + ]: 50 : if (filename)
1559 : 44 : fclose(OPF);
1560 : :
2274 tgl@sss.pgh.pa.us 1561 : 50 : ret = runPgDump(dbname, create_opts);
7901 peter_e@gmx.net 1562 [ - + ]: 49 : if (ret != 0)
737 tgl@sss.pgh.pa.us 1563 :UBC 0 : pg_fatal("pg_dump failed on database \"%s\", exiting", dbname);
1564 : :
6289 bruce@momjian.us 1565 [ + + ]:CBC 49 : if (filename)
1566 : : {
1567 : 43 : OPF = fopen(filename, PG_BINARY_A);
1568 [ - + ]: 43 : if (!OPF)
737 tgl@sss.pgh.pa.us 1569 :UBC 0 : pg_fatal("could not re-open the output file \"%s\": %m",
1570 : : filename);
1571 : : }
1572 : : }
1573 : :
7901 peter_e@gmx.net 1574 :CBC 11 : PQclear(res);
1575 : 11 : }
1576 : :
1577 : :
1578 : :
1579 : : /*
1580 : : * Run pg_dump on dbname, with specified options.
1581 : : */
1582 : : static int
2274 tgl@sss.pgh.pa.us 1583 : 50 : runPgDump(const char *dbname, const char *create_opts)
1584 : : {
1585 : : PQExpBufferData connstrbuf;
1586 : : PQExpBufferData cmd;
1587 : : int ret;
1588 : :
284 peter@eisentraut.org 1589 :GNC 50 : initPQExpBuffer(&connstrbuf);
1590 : 50 : initPQExpBuffer(&cmd);
1591 : :
1592 : 50 : printfPQExpBuffer(&cmd, "\"%s\" %s %s", pg_dump_bin,
2274 tgl@sss.pgh.pa.us 1593 :CBC 50 : pgdumpopts->data, create_opts);
1594 : :
1595 : : /*
1596 : : * If we have a filename, use the undocumented plain-append pg_dump
1597 : : * format.
1598 : : */
6289 bruce@momjian.us 1599 [ + + ]: 50 : if (filename)
284 peter@eisentraut.org 1600 :GNC 44 : appendPQExpBufferStr(&cmd, " -Fa ");
1601 : : else
1602 : 6 : appendPQExpBufferStr(&cmd, " -Fp ");
1603 : :
1604 : : /*
1605 : : * Append the database name to the already-constructed stem of connection
1606 : : * string.
1607 : : */
1608 : 50 : appendPQExpBuffer(&connstrbuf, "%s dbname=", connstr);
1609 : 50 : appendConnStrVal(&connstrbuf, dbname);
1610 : :
1611 : 50 : appendShellString(&cmd, connstrbuf.data);
1612 : :
1613 : 49 : pg_log_info("running \"%s\"", cmd.data);
1614 : :
594 tgl@sss.pgh.pa.us 1615 :CBC 49 : fflush(NULL);
1616 : :
284 peter@eisentraut.org 1617 :GNC 49 : ret = system(cmd.data);
1618 : :
1619 : 49 : termPQExpBuffer(&cmd);
1620 : 49 : termPQExpBuffer(&connstrbuf);
1621 : :
7901 peter_e@gmx.net 1622 :CBC 49 : return ret;
1623 : : }
1624 : :
1625 : : /*
1626 : : * buildShSecLabels
1627 : : *
1628 : : * Build SECURITY LABEL command(s) for a shared object
1629 : : *
1630 : : * The caller has to provide object type and identity in two separate formats:
1631 : : * catalog_name (e.g., "pg_database") and object OID, as well as
1632 : : * type name (e.g., "DATABASE") and object name (not pre-quoted).
1633 : : *
1634 : : * The command(s) are appended to "buffer".
1635 : : */
1636 : : static void
2239 tgl@sss.pgh.pa.us 1637 : 69 : buildShSecLabels(PGconn *conn, const char *catalog_name, Oid objectId,
1638 : : const char *objtype, const char *objname,
1639 : : PQExpBuffer buffer)
1640 : : {
4326 bruce@momjian.us 1641 : 69 : PQExpBuffer sql = createPQExpBuffer();
1642 : : PGresult *res;
1643 : :
1328 peter@eisentraut.org 1644 : 69 : buildShSecLabelQuery(catalog_name, objectId, sql);
4652 rhaas@postgresql.org 1645 : 69 : res = executeQuery(conn, sql->data);
2239 tgl@sss.pgh.pa.us 1646 : 69 : emitShSecLabels(conn, res, buffer, objtype, objname);
1647 : :
4652 rhaas@postgresql.org 1648 : 69 : PQclear(res);
1649 : 69 : destroyPQExpBuffer(sql);
1650 : 69 : }
1651 : :
1652 : : /*
1653 : : * Make a database connection with the given parameters. An
1654 : : * interactive password prompt is automatically issued if required.
1655 : : *
1656 : : * If fail_on_error is false, we return NULL without printing any message
1657 : : * on failure, but preserve any prompted password for the next try.
1658 : : *
1659 : : * On success, the global variable 'connstr' is set to a connection string
1660 : : * containing the options used.
1661 : : */
1662 : : static PGconn *
4066 heikki.linnakangas@i 1663 : 26 : connectDatabase(const char *dbname, const char *connection_string,
1664 : : const char *pghost, const char *pgport, const char *pguser,
1665 : : trivalue prompt_password, bool fail_on_error)
1666 : : {
1667 : : PGconn *conn;
1668 : : bool new_pass;
1669 : : const char *remoteversion_str;
1670 : : int my_version;
1671 : 26 : const char **keywords = NULL;
1672 : 26 : const char **values = NULL;
1673 : 26 : PQconninfoOption *conn_opts = NULL;
1674 : : static char *password = NULL;
1675 : :
1319 tgl@sss.pgh.pa.us 1676 [ - + - - ]: 26 : if (prompt_password == TRI_YES && !password)
1319 tgl@sss.pgh.pa.us 1677 :UBC 0 : password = simple_prompt("Password: ", false);
1678 : :
1679 : : /*
1680 : : * Start the connection. Loop until we have a password if requested by
1681 : : * backend.
1682 : : */
1683 : : do
1684 : : {
4066 heikki.linnakangas@i 1685 :CBC 26 : int argcount = 6;
1686 : : PQconninfoOption *conn_opt;
1687 : 26 : char *err_msg = NULL;
1688 : 26 : int i = 0;
1689 : :
668 peter@eisentraut.org 1690 : 26 : free(keywords);
1691 : 26 : free(values);
651 1692 : 26 : PQconninfoFree(conn_opts);
1693 : :
1694 : : /*
1695 : : * Merge the connection info inputs given in form of connection string
1696 : : * and other options. Explicitly discard any dbname value in the
1697 : : * connection string; otherwise, PQconnectdbParams() would interpret
1698 : : * that value as being itself a connection string.
1699 : : */
4066 heikki.linnakangas@i 1700 [ + - ]: 26 : if (connection_string)
1701 : : {
1702 : 26 : conn_opts = PQconninfoParse(connection_string, &err_msg);
1703 [ - + ]: 26 : if (conn_opts == NULL)
737 tgl@sss.pgh.pa.us 1704 :UBC 0 : pg_fatal("%s", err_msg);
1705 : :
4066 heikki.linnakangas@i 1706 [ + + ]:CBC 1092 : for (conn_opt = conn_opts; conn_opt->keyword != NULL; conn_opt++)
1707 : : {
2806 noah@leadboat.com 1708 [ + + + - ]: 1066 : if (conn_opt->val != NULL && conn_opt->val[0] != '\0' &&
1709 [ + + ]: 24 : strcmp(conn_opt->keyword, "dbname") != 0)
4066 heikki.linnakangas@i 1710 : 16 : argcount++;
1711 : : }
1712 : :
1713 : 26 : keywords = pg_malloc0((argcount + 1) * sizeof(*keywords));
1714 : 26 : values = pg_malloc0((argcount + 1) * sizeof(*values));
1715 : :
1716 [ + + ]: 1092 : for (conn_opt = conn_opts; conn_opt->keyword != NULL; conn_opt++)
1717 : : {
2806 noah@leadboat.com 1718 [ + + + - ]: 1066 : if (conn_opt->val != NULL && conn_opt->val[0] != '\0' &&
1719 [ + + ]: 24 : strcmp(conn_opt->keyword, "dbname") != 0)
1720 : : {
4066 heikki.linnakangas@i 1721 : 16 : keywords[i] = conn_opt->keyword;
1722 : 16 : values[i] = conn_opt->val;
1723 : 16 : i++;
1724 : : }
1725 : : }
1726 : : }
1727 : : else
1728 : : {
4066 heikki.linnakangas@i 1729 :UBC 0 : keywords = pg_malloc0((argcount + 1) * sizeof(*keywords));
1730 : 0 : values = pg_malloc0((argcount + 1) * sizeof(*values));
1731 : : }
1732 : :
4066 heikki.linnakangas@i 1733 [ + + ]:CBC 26 : if (pghost)
1734 : : {
1735 : 4 : keywords[i] = "host";
1736 : 4 : values[i] = pghost;
1737 : 4 : i++;
1738 : : }
1739 [ + + ]: 26 : if (pgport)
1740 : : {
1741 : 8 : keywords[i] = "port";
1742 : 8 : values[i] = pgport;
1743 : 8 : i++;
1744 : : }
1745 [ + + ]: 26 : if (pguser)
1746 : : {
1747 : 11 : keywords[i] = "user";
1748 : 11 : values[i] = pguser;
1749 : 11 : i++;
1750 : : }
1319 tgl@sss.pgh.pa.us 1751 [ - + ]: 26 : if (password)
1752 : : {
4066 heikki.linnakangas@i 1753 :UBC 0 : keywords[i] = "password";
1754 : 0 : values[i] = password;
1755 : 0 : i++;
1756 : : }
4066 heikki.linnakangas@i 1757 [ + - ]:CBC 26 : if (dbname)
1758 : : {
1759 : 26 : keywords[i] = "dbname";
1760 : 26 : values[i] = dbname;
1761 : 26 : i++;
1762 : : }
1763 : 26 : keywords[i] = "fallback_application_name";
1764 : 26 : values[i] = progname;
1765 : 26 : i++;
1766 : :
6125 tgl@sss.pgh.pa.us 1767 : 26 : new_pass = false;
5182 mail@joeconway.com 1768 : 26 : conn = PQconnectdbParams(keywords, values, true);
1769 : :
7901 peter_e@gmx.net 1770 [ - + ]: 26 : if (!conn)
737 tgl@sss.pgh.pa.us 1771 :UBC 0 : pg_fatal("could not connect to database \"%s\"", dbname);
1772 : :
7901 peter_e@gmx.net 1773 [ - + - - ]:CBC 26 : if (PQstatus(conn) == CONNECTION_BAD &&
5971 tgl@sss.pgh.pa.us 1774 :UBC 0 : PQconnectionNeedsPassword(conn) &&
1319 1775 [ # # # # ]: 0 : !password &&
1776 : : prompt_password != TRI_NO)
1777 : : {
7901 peter_e@gmx.net 1778 : 0 : PQfinish(conn);
1319 tgl@sss.pgh.pa.us 1779 : 0 : password = simple_prompt("Password: ", false);
6125 1780 : 0 : new_pass = true;
1781 : : }
6125 tgl@sss.pgh.pa.us 1782 [ - + ]:CBC 26 : } while (new_pass);
1783 : :
1784 : : /* check to see that the backend connection was successfully made */
7901 peter_e@gmx.net 1785 [ - + ]: 26 : if (PQstatus(conn) == CONNECTION_BAD)
1786 : : {
6855 tgl@sss.pgh.pa.us 1787 [ # # ]:UBC 0 : if (fail_on_error)
737 1788 : 0 : pg_fatal("%s", PQerrorMessage(conn));
1789 : : else
1790 : : {
6855 1791 : 0 : PQfinish(conn);
1792 : :
4066 heikki.linnakangas@i 1793 : 0 : free(keywords);
1794 : 0 : free(values);
1795 : 0 : PQconninfoFree(conn_opts);
1796 : :
6855 tgl@sss.pgh.pa.us 1797 : 0 : return NULL;
1798 : : }
1799 : : }
1800 : :
1801 : : /*
1802 : : * Ok, connected successfully. Remember the options used, in the form of a
1803 : : * connection string.
1804 : : */
4066 heikki.linnakangas@i 1805 :CBC 26 : connstr = constructConnStr(keywords, values);
1806 : :
1807 : 26 : free(keywords);
1808 : 26 : free(values);
1809 : 26 : PQconninfoFree(conn_opts);
1810 : :
1811 : : /* Check version */
7602 tgl@sss.pgh.pa.us 1812 : 26 : remoteversion_str = PQparameterStatus(conn, "server_version");
1813 [ - + ]: 26 : if (!remoteversion_str)
737 tgl@sss.pgh.pa.us 1814 :UBC 0 : pg_fatal("could not get server version");
4037 heikki.linnakangas@i 1815 :CBC 26 : server_version = PQserverVersion(conn);
1816 [ - + ]: 26 : if (server_version == 0)
737 tgl@sss.pgh.pa.us 1817 :UBC 0 : pg_fatal("could not parse server version \"%s\"",
1818 : : remoteversion_str);
1819 : :
4037 heikki.linnakangas@i 1820 :CBC 26 : my_version = PG_VERSION_NUM;
1821 : :
1822 : : /*
1823 : : * We allow the server to be back to 9.2, and up to any minor release of
1824 : : * our own major version. (See also version check in pg_dump.c.)
1825 : : */
6936 tgl@sss.pgh.pa.us 1826 [ - + ]: 26 : if (my_version != server_version
852 tgl@sss.pgh.pa.us 1827 [ # # ]:UBC 0 : && (server_version < 90200 ||
5845 1828 [ # # ]: 0 : (server_version / 100) > (my_version / 100)))
1829 : : {
1840 peter@eisentraut.org 1830 : 0 : pg_log_error("aborting because of server version mismatch");
737 tgl@sss.pgh.pa.us 1831 : 0 : pg_log_error_detail("server version: %s; %s version: %s",
1832 : : remoteversion_str, progname, PG_VERSION);
4441 rhaas@postgresql.org 1833 : 0 : exit_nicely(1);
1834 : : }
1835 : :
2239 noah@leadboat.com 1836 :CBC 26 : PQclear(executeQuery(conn, ALWAYS_SECURE_SEARCH_PATH_SQL));
1837 : :
7901 peter_e@gmx.net 1838 : 26 : return conn;
1839 : : }
1840 : :
1841 : : /* ----------
1842 : : * Construct a connection string from the given keyword/value pairs. It is
1843 : : * used to pass the connection options to the pg_dump subprocess.
1844 : : *
1845 : : * The following parameters are excluded:
1846 : : * dbname - varies in each pg_dump invocation
1847 : : * password - it's not secure to pass a password on the command line
1848 : : * fallback_application_name - we'll let pg_dump set it
1849 : : * ----------
1850 : : */
1851 : : static char *
4066 heikki.linnakangas@i 1852 : 26 : constructConnStr(const char **keywords, const char **values)
1853 : : {
1854 : 26 : PQExpBuffer buf = createPQExpBuffer();
1855 : : char *connstr;
1856 : : int i;
1857 : 26 : bool firstkeyword = true;
1858 : :
1859 : : /* Construct a new connection string in key='value' format. */
1860 [ + + ]: 117 : for (i = 0; keywords[i] != NULL; i++)
1861 : : {
1862 [ + + ]: 91 : if (strcmp(keywords[i], "dbname") == 0 ||
1863 [ + - ]: 65 : strcmp(keywords[i], "password") == 0 ||
1864 [ + + ]: 65 : strcmp(keywords[i], "fallback_application_name") == 0)
1865 : 52 : continue;
1866 : :
1867 [ + + ]: 39 : if (!firstkeyword)
1868 : 20 : appendPQExpBufferChar(buf, ' ');
1869 : 39 : firstkeyword = false;
1870 : 39 : appendPQExpBuffer(buf, "%s=", keywords[i]);
2806 noah@leadboat.com 1871 : 39 : appendConnStrVal(buf, values[i]);
1872 : : }
1873 : :
4066 heikki.linnakangas@i 1874 : 26 : connstr = pg_strdup(buf->data);
1875 : 26 : destroyPQExpBuffer(buf);
1876 : 26 : return connstr;
1877 : : }
1878 : :
1879 : : /*
1880 : : * Run a query, return the results, exit program on failure.
1881 : : */
1882 : : static PGresult *
7901 peter_e@gmx.net 1883 : 267 : executeQuery(PGconn *conn, const char *query)
1884 : : {
1885 : : PGresult *res;
1886 : :
1840 peter@eisentraut.org 1887 : 267 : pg_log_info("executing %s", query);
1888 : :
7901 peter_e@gmx.net 1889 : 267 : res = PQexec(conn, query);
1890 [ + - - + ]: 534 : if (!res ||
1891 : 267 : PQresultStatus(res) != PGRES_TUPLES_OK)
1892 : : {
1840 peter@eisentraut.org 1893 :UBC 0 : pg_log_error("query failed: %s", PQerrorMessage(conn));
737 tgl@sss.pgh.pa.us 1894 : 0 : pg_log_error_detail("Query was: %s", query);
7901 peter_e@gmx.net 1895 : 0 : PQfinish(conn);
4441 rhaas@postgresql.org 1896 : 0 : exit_nicely(1);
1897 : : }
1898 : :
7901 peter_e@gmx.net 1899 :CBC 267 : return res;
1900 : : }
1901 : :
1902 : : /*
1903 : : * As above for a SQL command (which returns nothing).
1904 : : */
1905 : : static void
6832 tgl@sss.pgh.pa.us 1906 : 4 : executeCommand(PGconn *conn, const char *query)
1907 : : {
1908 : : PGresult *res;
1909 : :
1840 peter@eisentraut.org 1910 : 4 : pg_log_info("executing %s", query);
1911 : :
6832 tgl@sss.pgh.pa.us 1912 : 4 : res = PQexec(conn, query);
1913 [ + - - + ]: 8 : if (!res ||
1914 : 4 : PQresultStatus(res) != PGRES_COMMAND_OK)
1915 : : {
1840 peter@eisentraut.org 1916 :UBC 0 : pg_log_error("query failed: %s", PQerrorMessage(conn));
737 tgl@sss.pgh.pa.us 1917 : 0 : pg_log_error_detail("Query was: %s", query);
6832 1918 : 0 : PQfinish(conn);
4441 rhaas@postgresql.org 1919 : 0 : exit_nicely(1);
1920 : : }
1921 : :
6832 tgl@sss.pgh.pa.us 1922 :CBC 4 : PQclear(res);
1923 : 4 : }
1924 : :
1925 : :
1926 : : /*
1927 : : * dumpTimestamp
1928 : : */
1929 : : static void
3458 1930 : 4 : dumpTimestamp(const char *msg)
1931 : : {
1932 : : char buf[64];
7168 bruce@momjian.us 1933 : 4 : time_t now = time(NULL);
1934 : :
3458 tgl@sss.pgh.pa.us 1935 [ + - ]: 4 : if (strftime(buf, sizeof(buf), PGDUMP_STRFTIME_FMT, localtime(&now)) != 0)
6289 bruce@momjian.us 1936 : 4 : fprintf(OPF, "-- %s %s\n\n", msg, buf);
7251 1937 : 4 : }
1938 : :
1939 : : /*
1940 : : * read_dumpall_filters - retrieve database identifier patterns from file
1941 : : *
1942 : : * Parse the specified filter file for include and exclude patterns, and add
1943 : : * them to the relevant lists. If the filename is "-" then filters will be
1944 : : * read from STDIN rather than a file.
1945 : : *
1946 : : * At the moment, the only allowed filter is for database exclusion.
1947 : : */
1948 : : static void
137 dgustafsson@postgres 1949 :GNC 5 : read_dumpall_filters(const char *filename, SimpleStringList *pattern)
1950 : : {
1951 : : FilterStateData fstate;
1952 : : char *objname;
1953 : : FilterCommandType comtype;
1954 : : FilterObjectType objtype;
1955 : :
1956 : 5 : filter_init(&fstate, filename, exit);
1957 : :
1958 [ + + ]: 12 : while (filter_read_item(&fstate, &objname, &comtype, &objtype))
1959 : : {
1960 [ - + ]: 3 : if (comtype == FILTER_COMMAND_TYPE_INCLUDE)
1961 : : {
136 dgustafsson@postgres 1962 :UNC 0 : pg_log_filter_error(&fstate, _("%s filter for \"%s\" is not allowed"),
1963 : : "include",
1964 : : filter_object_type_name(objtype));
137 1965 : 0 : exit_nicely(1);
1966 : : }
1967 : :
137 dgustafsson@postgres 1968 [ - + + - ]:GNC 3 : switch (objtype)
1969 : : {
137 dgustafsson@postgres 1970 :UNC 0 : case FILTER_OBJECT_TYPE_NONE:
1971 : 0 : break;
137 dgustafsson@postgres 1972 :GNC 1 : case FILTER_OBJECT_TYPE_FUNCTION:
1973 : : case FILTER_OBJECT_TYPE_INDEX:
1974 : : case FILTER_OBJECT_TYPE_TABLE_DATA:
1975 : : case FILTER_OBJECT_TYPE_TABLE_DATA_AND_CHILDREN:
1976 : : case FILTER_OBJECT_TYPE_TRIGGER:
1977 : : case FILTER_OBJECT_TYPE_EXTENSION:
1978 : : case FILTER_OBJECT_TYPE_FOREIGN_DATA:
1979 : : case FILTER_OBJECT_TYPE_SCHEMA:
1980 : : case FILTER_OBJECT_TYPE_TABLE:
1981 : : case FILTER_OBJECT_TYPE_TABLE_AND_CHILDREN:
136 1982 : 1 : pg_log_filter_error(&fstate, _("unsupported filter object"));
137 1983 : 1 : exit_nicely(1);
1984 : : break;
1985 : :
1986 : 2 : case FILTER_OBJECT_TYPE_DATABASE:
1987 : 2 : simple_string_list_append(pattern, objname);
1988 : 2 : break;
1989 : : }
1990 : :
1991 [ - + ]: 2 : if (objname)
1992 : 2 : free(objname);
1993 : : }
1994 : :
1995 : 2 : filter_free(&fstate);
137 dgustafsson@postgres 1996 :GIC 2 : }
|
