pg_dump.c 128.5 KB
Newer Older
1 2
/*-------------------------------------------------------------------------
 *
3
 * pg_dump.c
4
 *	  pg_dump is a utility for dumping out a postgres database
B
Bruce Momjian 已提交
5
 *	  into a script file.
6 7 8
 *
 * Portions Copyright (c) 1996-2001, PostgreSQL Global Development Group
 * Portions Copyright (c) 1994, Regents of the University of California
9
 *
10 11 12 13 14 15
 *	pg_dump will read the system catalogs in a database and
 *	dump out a script that reproduces
 *	the schema of the database in terms of
 *		  user-defined types
 *		  user-defined functions
 *		  tables
16
 *		  indexes
17 18
 *		  aggregates
 *		  operators
19
 *		  privileges
20
 *
21
 * the output script is SQL that is understood by PostgreSQL
22 23 24
 *
 *
 * IDENTIFICATION
25
 *	  $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.232 2001/10/03 05:23:12 thomas Exp $
26
 *
27
 *-------------------------------------------------------------------------
28 29
 */

30 31 32 33 34 35 36
/*
 * Although this is not a backend module, we must include postgres.h anyway
 * so that we can include a bunch of backend include files.  pg_dump has
 * never pretended to be very independent of the backend anyhow ...
 */
#include "postgres.h"

37
#include <unistd.h>				/* for getopt() */
38
#include <ctype.h>
39 40 41
#ifdef ENABLE_NLS
#include <locale.h>
#endif
42 43 44 45 46 47 48
#ifdef HAVE_GETOPT_H
#include <getopt.h>
#endif
#ifdef HAVE_TERMIOS_H
#include <termios.h>
#endif

49 50 51 52
#ifndef HAVE_STRDUP
#include "strdup.h"
#endif

53
#include "access/attnum.h"
54
#include "access/htup.h"
55
#include "catalog/pg_class.h"
V
Vadim B. Mikheev 已提交
56
#include "catalog/pg_trigger.h"
57
#include "catalog/pg_type.h"
58

59
#include "libpq-fe.h"
60
#include "libpq/libpq-fs.h"
61 62

#include "pg_dump.h"
B
Bruce Momjian 已提交
63
#include "pg_backup.h"
64
#include "pg_backup_archiver.h"
65

B
Bruce Momjian 已提交
66 67 68 69 70 71 72 73
typedef enum _formatLiteralOptions
{
	CONV_ALL = 0,
	PASS_LFTAB = 3				/* NOTE: 1 and 2 are reserved in case we
								 * want to make a mask. */
	/* We could make this a bit mask for control chars, but I don't */
	/* see any value in making it more complex...the current code */
	/* only checks for 'opts == CONV_ALL' anyway. */
74 75
} formatLiteralOptions;

76 77
static void dumpComment(Archive *fout, const char *target, const char *oid,
						const char *classname, int subid,
78
						const char *((*deps)[]));
79
static void dumpSequence(Archive *fout, TableInfo tbinfo, const bool schemaOnly, const bool dataOnly);
B
Bruce Momjian 已提交
80 81
static void dumpACL(Archive *fout, TableInfo tbinfo);
static void dumpTriggers(Archive *fout, const char *tablename,
82
			 TableInfo *tblinfo, int numTables);
B
Bruce Momjian 已提交
83
static void dumpRules(Archive *fout, const char *tablename,
B
Bruce Momjian 已提交
84
		  TableInfo *tblinfo, int numTables);
85
static void formatStringLiteral(PQExpBuffer buf, const char *str, const formatLiteralOptions opts);
86
static void clearTableInfo(TableInfo *, int);
B
Bruce Momjian 已提交
87
static void dumpOneFunc(Archive *fout, FuncInfo *finfo, int i,
88
			TypeInfo *tinfo, int numTypes);
89 90
static Oid	findLastBuiltinOid_V71(const char *);
static Oid	findLastBuiltinOid_V70(void);
B
Bruce Momjian 已提交
91
static void setMaxOid(Archive *fout);
92

93
static void AddAcl(char *aclbuf, const char *keyword);
94
static char *GetPrivileges(Archive *AH, const char *s);
V
Vadim B. Mikheev 已提交
95

B
Bruce Momjian 已提交
96 97
static int	dumpBlobs(Archive *AH, char *, void *);
static int	dumpDatabase(Archive *AH);
98
static PQExpBuffer getPKconstraint(TableInfo *tblInfo, IndInfo *indInfo);
99
static const char *getAttrName(int attrnum, TableInfo *tblInfo);
100

B
Bruce Momjian 已提交
101
extern char *optarg;
102
extern int	optind,
B
Bruce Momjian 已提交
103
			opterr;
104 105

/* global decls */
106
bool		g_verbose;			/* User wants verbose narration of our
B
Bruce Momjian 已提交
107
								 * activities. */
108
Oid			g_last_builtin_oid; /* value of the last builtin oid */
B
Bruce Momjian 已提交
109
Archive    *g_fout;				/* the script file */
B
Bruce Momjian 已提交
110 111 112
PGconn	   *g_conn;				/* the database connection */

bool		force_quotes;		/* User wants to suppress double-quotes */
113 114 115 116
bool		dumpData;			/* dump data using proper insert strings */
bool		attrNames;			/* put attr names into insert strings */
bool		schemaOnly;
bool		dataOnly;
117
bool		aclsSkip;
118

B
Bruce Momjian 已提交
119
char		g_opaque_type[10];	/* name for the opaque type */
120 121

/* placeholders for the delimiters for comments */
122 123
char		g_comment_start[10];
char		g_comment_end[10];
124 125


B
Bruce Momjian 已提交
126 127 128 129
typedef struct _dumpContext
{
	TableInfo  *tblinfo;
	int			tblidx;
B
Bruce Momjian 已提交
130 131 132
	bool		oids;
} DumpContext;

133
static void
134
help(const char *progname)
135
{
136 137 138 139
	printf(gettext("%s dumps a database as a text file or to other formats.\n\n"), progname);
	puts(gettext("Usage:"));
	printf(gettext("  %s [options] dbname\n\n"), progname);
	puts(gettext("Options:"));
B
Hi,  
Bruce Momjian 已提交
140 141

#ifdef HAVE_GETOPT_LONG
142 143
	puts(gettext(
		"  -a, --data-only          dump only the data, not the schema\n"
144
		"  -b, --blobs              include large objects in dump\n"
145 146 147
		"  -c, --clean              clean (drop) schema prior to create\n"
		"  -C, --create             include commands to create database in dump\n"
		"  -d, --inserts            dump data as INSERT, rather than COPY, commands\n"
148
		"  -D, --column-inserts     dump data as INSERT commands with column names\n"
149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168
		"  -f, --file=FILENAME      output file name\n"
		"  -F, --format {c|t|p}     output file format (custom, tar, plain text)\n"
		"  -h, --host=HOSTNAME      database server host name\n"
		"  -i, --ignore-version     proceed even when server version mismatches\n"
		"                           pg_dump version\n"
		"  -n, --no-quotes          suppress most quotes around identifiers\n"
		"  -N, --quotes             enable most quotes around identifiers\n"
		"  -o, --oids               include oids in dump\n"
		"  -O, --no-owner           do not output \\connect commands in plain\n"
		"                           text format\n"
		"  -p, --port=PORT          database server port number\n"
		"  -R, --no-reconnect       disable ALL reconnections to the database in\n"
		"                           plain text format\n"
		"  -s, --schema-only        dump only the schema, no data\n"
		"  -S, --superuser=NAME     specify the superuser user name to use in\n"
		"                           plain text format\n"
		"  -t, --table=TABLE        dump this table only (* for all)\n"
		"  -U, --username=NAME      connect as specified database user\n"
		"  -v, --verbose            verbose mode\n"
		"  -W, --password           force password prompt (should happen automatically)\n"
169
		"  -x, --no-privileges      do not dump privileges (grant/revoke)\n"
170 171 172
		"  -X use-set-session-authorization, --use-set-session-authorization\n"
		"                           output SET SESSION AUTHORIZATION commands rather\n"
		"                           than \\connect commands\n"
173 174
		"  -Z, --compress {0-9}     compression level for compressed formats\n"
		));
B
Hi,  
Bruce Momjian 已提交
175
#else
176 177
	puts(gettext(
		"  -a                       dump only the data, not the schema\n"
178
		"  -b                       include large objects in dump\n"
179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203
		"  -c                       clean (drop) schema prior to create\n"
		"  -C                       include commands to create database in dump\n"
		"  -d                       dump data as INSERT, rather than COPY, commands\n"
		"  -D                       dump data as INSERT commands with column names\n"
		"  -f FILENAME              output file name\n"
		"  -F {c|t|p}               output file format (custom, tar, plain text)\n"
		"  -h HOSTNAME              database server host name\n"
		"  -i                       proceed even when server version mismatches\n"
		"                           pg_dump version\n"
		"  -n                       suppress most quotes around identifiers\n"
		"  -N                       enable most quotes around identifiers\n"
		"  -o                       include oids in dump\n"
		"  -O                       do not output \\connect commands in plain\n"
		"                           text format\n"
		"  -p PORT                  database server port number\n"
		"  -R                       disable ALL reconnections to the database in\n"
		"                           plain text format\n"
		"  -s                       dump only the schema, no data\n"
		"  -S NAME                  specify the superuser user name to use in\n"
		"                           plain text format\n"
		"  -t TABLE                 dump this table only (* for all)\n"
		"  -U NAME                  connect as specified database user\n"
		"  -v                       verbose mode\n"
		"  -W                       force password prompt (should happen automatically)\n"
		"  -x                       do not dump privileges (grant/revoke)\n"
204 205 206
		"  -X use-set-session-authorization\n"
		"                           output SET SESSION AUTHORIZATION commands rather\n"
		"                           than \\connect commands\n"
207 208
		"  -Z {0-9}                 compression level for compressed formats\n"
		));
B
Hi,  
Bruce Momjian 已提交
209
#endif
210 211 212
	puts(gettext("If no database name is not supplied, then the PGDATABASE environment\n"
				 "variable value is used.\n\n"
				 "Report bugs to <pgsql-bugs@postgresql.org>."));
213 214
}

215

216 217
void
exit_nicely(void)
218
{
219 220 221
	PQfinish(g_conn);
	if (g_verbose)
		write_msg(NULL, "*** aborted because of error\n");
222
	exit(1);
223 224 225
}


226 227
#define COPYBUFSIZ		8192

B
Bruce Momjian 已提交
228 229 230 231 232
/*
 *	Dump a table's contents for loading using the COPY command
 *	- this routine is called by the Archiver when it wants the table
 *	  to be dumped.
 */
233

B
Bruce Momjian 已提交
234 235
static int
dumpClasses_nodumpData(Archive *fout, char *oid, void *dctxv)
236
{
B
Bruce Momjian 已提交
237 238
	const DumpContext *dctx = (DumpContext *) dctxv;
	const char *classname = dctx->tblinfo[dctx->tblidx].relname;
239
	const bool	hasoids = dctx->tblinfo[dctx->tblidx].hasoids;
B
Bruce Momjian 已提交
240
	const bool	oids = dctx->oids;
241

242 243 244 245 246
	PGresult   *res;
	char		query[255];
	int			ret;
	bool		copydone;
	char		copybuf[COPYBUFSIZ];
247

B
Bruce Momjian 已提交
248
	if (g_verbose)
249
		write_msg(NULL, "dumping out the contents of table %s\n", classname);
250

251
	if (oids && hasoids)
252
	{
B
Bruce Momjian 已提交
253 254

		/*
255
		 * archprintf(fout, "COPY %s WITH OIDS FROM stdin;\n",
B
Bruce Momjian 已提交
256
		 * fmtId(classname, force_quotes));
257
		 *
B
Bruce Momjian 已提交
258
		 * - Not used as of V1.3 (needs to be in ArchiveEntry call)
259 260 261
		 *
		 */

262
		sprintf(query, "COPY %s WITH OIDS TO stdout;",
263
				fmtId(classname, force_quotes));
264 265 266
	}
	else
	{
B
Bruce Momjian 已提交
267 268 269 270

		/*
		 * archprintf(fout, "COPY %s FROM stdin;\n", fmtId(classname,
		 * force_quotes));
271 272 273 274 275
		 *
		 * - Not used as of V1.3 (needs to be in ArchiveEntry call)
		 *
		 */

276
		sprintf(query, "COPY %s TO stdout;", fmtId(classname, force_quotes));
277 278
	}
	res = PQexec(g_conn, query);
279 280
	if (!res ||
		PQresultStatus(res) == PGRES_FATAL_ERROR)
281
	{
282 283 284 285
		write_msg(NULL, "SQL command to dump the contents of table \"%s\" failed\n",
				  classname);
		write_msg(NULL, "Error message from server: %s", PQerrorMessage(g_conn));
		write_msg(NULL, "The command was: %s\n", query);
286
		exit_nicely();
287 288 289 290 291
	}
	else
	{
		if (PQresultStatus(res) != PGRES_COPY_OUT)
		{
292 293 294 295 296
			write_msg(NULL, "SQL command to dump the contents of table \"%s\" executed abnormally.\n",
					  classname);
			write_msg(NULL, "The server returned status %d when %d was expected.\n", 
					  PQresultStatus(res), PGRES_COPY_OUT);
			write_msg(NULL, "The command was: %s\n", query);
297
			exit_nicely();
298 299 300 301
		}
		else
		{
			copydone = false;
302

303 304
			while (!copydone)
			{
305
				ret = PQgetline(g_conn, copybuf, COPYBUFSIZ);
306 307 308 309 310 311 312 313 314

				if (copybuf[0] == '\\' &&
					copybuf[1] == '.' &&
					copybuf[2] == '\0')
				{
					copydone = true;	/* don't print this... */
				}
				else
				{
B
Bruce Momjian 已提交
315
					archputs(copybuf, fout);
316 317
					switch (ret)
					{
318 319 320 321
						case EOF:
							copydone = true;
							/* FALLTHROUGH */
						case 0:
B
Bruce Momjian 已提交
322
							archputc('\n', fout);
323 324 325
							break;
						case 1:
							break;
326 327
					}
				}
328

B
Bruce Momjian 已提交
329
				/*
330 331
				 * THROTTLE:
				 *
B
Bruce Momjian 已提交
332 333 334 335 336
				 * There was considerable discussion in late July, 2000
				 * regarding slowing down pg_dump when backing up large
				 * tables. Users with both slow & fast (muti-processor)
				 * machines experienced performance degradation when doing
				 * a backup.
337
				 *
B
Bruce Momjian 已提交
338 339 340 341 342
				 * Initial attempts based on sleeping for a number of ms for
				 * each ms of work were deemed too complex, then a simple
				 * 'sleep in each loop' implementation was suggested. The
				 * latter failed because the loop was too tight. Finally,
				 * the following was implemented:
343
				 *
B
Bruce Momjian 已提交
344 345 346
				 * If throttle is non-zero, then See how long since the last
				 * sleep. Work out how long to sleep (based on ratio). If
				 * sleep is more than 100ms, then sleep reset timer EndIf
347 348
				 * EndIf
				 *
B
Bruce Momjian 已提交
349 350
				 * where the throttle value was the number of ms to sleep per
				 * ms of work. The calculation was done in each loop.
351
				 *
B
Bruce Momjian 已提交
352 353 354 355 356
				 * Most of the hard work is done in the backend, and this
				 * solution still did not work particularly well: on slow
				 * machines, the ratio was 50:1, and on medium paced
				 * machines, 1:1, and on fast multi-processor machines, it
				 * had little or no effect, for reasons that were unclear.
357 358 359
				 *
				 * Further discussion ensued, and the proposal was dropped.
				 *
B
Bruce Momjian 已提交
360 361 362 363 364 365
				 * For those people who want this feature, it can be
				 * implemented using gettimeofday in each loop,
				 * calculating the time since last sleep, multiplying that
				 * by the sleep ratio, then if the result is more than a
				 * preset 'minimum sleep time' (say 100ms), call the
				 * 'select' function to sleep for a subsecond period ie.
366
				 *
B
Bruce Momjian 已提交
367
				 * select(0, NULL, NULL, NULL, &tvi);
368
				 *
B
Bruce Momjian 已提交
369 370 371 372
				 * This will return after the interval specified in the
				 * structure tvi. Fianally, call gettimeofday again to
				 * save the 'last sleep time'.
				 */
373
			}
B
Bruce Momjian 已提交
374
			archprintf(fout, "\\.\n");
375
		}
376
		ret = PQendcopy(g_conn);
377 378
		if (ret != 0)
		{
379 380 381
			write_msg(NULL, "SQL command to dump the contents of table \"%s\" failed: PQendcopy() failed.\n", classname);
			write_msg(NULL, "Error message from server: %s", PQerrorMessage(g_conn));
			write_msg(NULL, "The command was: %s\n", query);
B
Bruce Momjian 已提交
382
			PQclear(res);
383
			exit_nicely();
384 385
		}
	}
386

B
Bruce Momjian 已提交
387
	return 1;
388 389
}

B
Bruce Momjian 已提交
390
static int
B
Bruce Momjian 已提交
391
dumpClasses_dumpData(Archive *fout, char *oid, void *dctxv)
392
{
B
Bruce Momjian 已提交
393 394
	const DumpContext *dctx = (DumpContext *) dctxv;
	const char *classname = dctx->tblinfo[dctx->tblidx].relname;
B
Bruce Momjian 已提交
395

396 397
	PGresult   *res;
	PQExpBuffer q = createPQExpBuffer();
398 399
	int			tuple;
	int			field;
400

401 402
	if (fout->remoteVersion >= 70100)
	{
403
		appendPQExpBuffer(q, "DECLARE _pg_dump_cursor CURSOR FOR SELECT * FROM ONLY %s", fmtId(classname, force_quotes));
404
	} else {
405
		appendPQExpBuffer(q, "DECLARE _pg_dump_cursor CURSOR FOR SELECT * FROM %s", fmtId(classname, force_quotes));
406 407
	}

B
Hi, all  
Bruce Momjian 已提交
408
	res = PQexec(g_conn, q->data);
409
	if (!res ||
410
		PQresultStatus(res) != PGRES_COMMAND_OK)
411
	{
412 413 414
		write_msg(NULL, "dumpClasses(): SQL command failed\n");
		write_msg(NULL, "Error message from server: %s", PQerrorMessage(g_conn));
		write_msg(NULL, "The command was: %s\n", q->data);
415
		exit_nicely();
416
	}
417 418 419 420 421 422 423
	
	do {
		PQclear(res);

		res = PQexec(g_conn, "FETCH 100 FROM _pg_dump_cursor");
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
424
		{
425 426 427 428
			write_msg(NULL, "dumpClasses(): SQL command failed\n");
			write_msg(NULL, "Error message from server: %s", PQerrorMessage(g_conn));
			write_msg(NULL, "The command was: FETCH 100 FROM _pg_dump_cursor\n");
			exit_nicely();
429
		}
430 431

		for (tuple = 0; tuple < PQntuples(res); tuple++)
432
		{
433 434
			archprintf(fout, "INSERT INTO %s ", fmtId(classname, force_quotes));
			if (attrNames == true)
435
			{
436 437 438 439 440 441 442 443 444 445
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "(");
				for (field = 0; field < PQnfields(res); field++)
				{
					if (field > 0)
						appendPQExpBuffer(q, ",");
					appendPQExpBuffer(q, fmtId(PQfname(res, field), force_quotes));
				}
				appendPQExpBuffer(q, ") ");
				archprintf(fout, "%s", q->data);
446
			}
447 448
			archprintf(fout, "VALUES (");
			for (field = 0; field < PQnfields(res); field++)
449
			{
450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483
				if (field > 0)
					archprintf(fout, ",");
				if (PQgetisnull(res, tuple, field))
				{
					archprintf(fout, "NULL");
					continue;
				}
				switch (PQftype(res, field))
				{
					case INT2OID:
					case INT4OID:
					case OIDOID:	/* int types */
					case FLOAT4OID:
					case FLOAT8OID:/* float types */
						/* These types are printed without quotes */
						archprintf(fout, "%s",
								   PQgetvalue(res, tuple, field));
						break;
					case BITOID:
					case VARBITOID:
						archprintf(fout, "B'%s'",
								   PQgetvalue(res, tuple, field));
						break;
					default:
	
						/*
						 * All other types are printed as string literals,
						 * with appropriate escaping of special characters.
						 */
						resetPQExpBuffer(q);
						formatStringLiteral(q, PQgetvalue(res, tuple, field), CONV_ALL);
						archprintf(fout, "%s", q->data);
						break;
				}
484
			}
485
			archprintf(fout, ");\n");
486
		}
487 488 489 490 491 492 493 494 495 496 497 498

	} while( PQntuples(res) > 0 );
	PQclear(res);

	res = PQexec(g_conn, "CLOSE _pg_dump_cursor");
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
		write_msg(NULL, "dumpClasses(): SQL command failed\n");
		write_msg(NULL, "Error message from server: %s", PQerrorMessage(g_conn));
		write_msg(NULL, "The command was: CLOSE _pg_dump_cursor\n");
		exit_nicely();
499 500
	}
	PQclear(res);
501

502
	destroyPQExpBuffer(q);
B
Bruce Momjian 已提交
503
	return 1;
504 505
}

506 507 508 509 510 511 512 513
/*
 * Convert a string value to an SQL string literal,
 * with appropriate escaping of special characters.
 * Quote mark ' goes to '' per SQL standard, other
 * stuff goes to \ sequences.
 * The literal is appended to the given PQExpBuffer.
 */
static void
514
formatStringLiteral(PQExpBuffer buf, const char *str, const formatLiteralOptions opts)
515 516 517 518
{
	appendPQExpBufferChar(buf, '\'');
	while (*str)
	{
B
Bruce Momjian 已提交
519
		char		ch = *str++;
520 521 522

		if (ch == '\\' || ch == '\'')
		{
B
Bruce Momjian 已提交
523
			appendPQExpBufferChar(buf, ch);		/* double these */
524 525 526
			appendPQExpBufferChar(buf, ch);
		}
		else if ((unsigned char) ch < (unsigned char) ' ' &&
B
Bruce Momjian 已提交
527 528 529
				 (opts == CONV_ALL
				  || (ch != '\n' && ch != '\t')
				  ))
530
		{
B
Bruce Momjian 已提交
531 532 533 534 535

			/*
			 * generate octal escape for control chars other than
			 * whitespace
			 */
536 537 538 539 540 541 542 543 544 545 546
			appendPQExpBufferChar(buf, '\\');
			appendPQExpBufferChar(buf, ((ch >> 6) & 3) + '0');
			appendPQExpBufferChar(buf, ((ch >> 3) & 7) + '0');
			appendPQExpBufferChar(buf, (ch & 7) + '0');
		}
		else
			appendPQExpBufferChar(buf, ch);
	}
	appendPQExpBufferChar(buf, '\'');
}

547 548
/*
 * DumpClasses -
549
 *	  dump the contents of all the classes.
550 551
 */
static void
B
Bruce Momjian 已提交
552
dumpClasses(const TableInfo *tblinfo, const int numTables, Archive *fout,
B
Bruce Momjian 已提交
553
		 const char *onlytable, const bool oids, const bool force_quotes)
554
{
B
Bruce Momjian 已提交
555 556 557 558 559 560
	int			i;
	DataDumperPtr dumpFn;
	DumpContext *dumpCtx;
	char		copyBuf[512];
	char	   *copyStmt;

561
	if (g_verbose)
562 563 564 565 566 567
	{
		if (onlytable == NULL || (strlen(onlytable) == 0))
			write_msg(NULL, "preparing to dump the contents of all %d tables/sequences\n", numTables);
		else
			write_msg(NULL, "preparing to dump the contents of only one table/sequence\n");
	}
568

569 570
	for (i = 0; i < numTables; i++)
	{
571
		const char *classname = tblinfo[i].relname;
572 573

		/* Skip VIEW relations */
574
		if (tblinfo[i].viewdef != NULL)
575
			continue;
576 577

		if (tblinfo[i].sequence)/* already dumped */
V
Vadim B. Mikheev 已提交
578
			continue;
579

580
		if (!onlytable || (strcmp(classname, onlytable) == 0) || (strlen(onlytable) == 0))
581 582
		{
			if (g_verbose)
583
				write_msg(NULL, "preparing to dump the contents of table %s\n", classname);
584

585 586 587
#if 0
			becomeUser(fout, tblinfo[i].usename);
#endif
B
Bruce Momjian 已提交
588

B
Bruce Momjian 已提交
589 590
			dumpCtx = (DumpContext *) malloc(sizeof(DumpContext));
			dumpCtx->tblinfo = (TableInfo *) tblinfo;
B
Bruce Momjian 已提交
591 592
			dumpCtx->tblidx = i;
			dumpCtx->oids = oids;
593

594
			if (!dumpData)
595
			{
596
				/* Dump/restore using COPY */
B
Bruce Momjian 已提交
597
				dumpFn = dumpClasses_nodumpData;
598 599 600
				sprintf(copyBuf, "COPY %s %sFROM stdin;\n",
						fmtId(tblinfo[i].relname, force_quotes),
						(oids && tblinfo[i].hasoids) ? "WITH OIDS " : "");
601 602
				copyStmt = copyBuf;
			}
B
Bruce Momjian 已提交
603
			else
604
			{
605
				/* Restore using INSERT */
B
Bruce Momjian 已提交
606
				dumpFn = dumpClasses_dumpData;
607 608
				copyStmt = NULL;
			}
B
Bruce Momjian 已提交
609

P
Philip Warner 已提交
610
			ArchiveEntry(fout, tblinfo[i].oid, tblinfo[i].relname,
611 612
						 "TABLE DATA", NULL, "", "",
						 copyStmt, tblinfo[i].usename,
B
Bruce Momjian 已提交
613
						 dumpFn, dumpCtx);
614
		}
615
	}
616 617
}

618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641


static int
parse_version(const char* versionString)
{
	int			cnt;
	int			vmaj, vmin, vrev;

	cnt = sscanf(versionString, "%d.%d.%d", &vmaj, &vmin, &vrev);

	if (cnt < 2)
	{
		write_msg(NULL, "unable to parse version string \"%s\"\n", versionString);
		exit(1);
	}

	if (cnt == 2)
		vrev = 0;

	return (100 * vmaj + vmin) * 100 + vrev;
}



642
int
643
main(int argc, char **argv)
644
{
B
Bruce Momjian 已提交
645 646
	int			c;
	const char *filename = NULL;
B
Bruce Momjian 已提交
647
	const char *format = "p";
B
Bruce Momjian 已提交
648 649 650
	const char *dbname = NULL;
	const char *pghost = NULL;
	const char *pgport = NULL;
651
	const char *username = NULL;
B
Bruce Momjian 已提交
652
	char	   *tablename = NULL;
653
	bool		oids = false;
B
Bruce Momjian 已提交
654 655
	TableInfo  *tblinfo;
	int			numTables;
656
	bool		force_password = false;
B
Bruce Momjian 已提交
657
	int			compressLevel = -1;
658
	bool		ignore_version = false;
659 660
	int			plainText = 0;
	int			outputClean = 0;
661
	int			outputCreate = 0;
662
	int			outputBlobs = 0;
663 664
	int			outputNoOwner = 0;
	int			outputNoReconnect = 0;
665
	static int	use_setsessauth = 0;
B
Bruce Momjian 已提交
666
	char	   *outputSuperuser = NULL;
667

B
Bruce Momjian 已提交
668
	RestoreOptions *ropt;
669

B
Hi,  
Bruce Momjian 已提交
670 671 672
#ifdef HAVE_GETOPT_LONG
	static struct option long_options[] = {
		{"data-only", no_argument, NULL, 'a'},
B
Bruce Momjian 已提交
673
		{"blobs", no_argument, NULL, 'b'},
B
Hi,  
Bruce Momjian 已提交
674
		{"clean", no_argument, NULL, 'c'},
675
		{"create", no_argument, NULL, 'C'},
B
Bruce Momjian 已提交
676 677
		{"file", required_argument, NULL, 'f'},
		{"format", required_argument, NULL, 'F'},
678
		{"inserts", no_argument, NULL, 'd'},
679
		{"attribute-inserts", no_argument, NULL, 'D'},
680
		{"column-inserts", no_argument, NULL, 'D'},
B
Hi,  
Bruce Momjian 已提交
681
		{"host", required_argument, NULL, 'h'},
682
		{"ignore-version", no_argument, NULL, 'i'},
683
		{"no-reconnect", no_argument, NULL, 'R'},
B
Hi,  
Bruce Momjian 已提交
684 685
		{"no-quotes", no_argument, NULL, 'n'},
		{"quotes", no_argument, NULL, 'N'},
686
		{"oids", no_argument, NULL, 'o'},
687
		{"no-owner", no_argument, NULL, 'O'},
B
Hi,  
Bruce Momjian 已提交
688 689
		{"port", required_argument, NULL, 'p'},
		{"schema-only", no_argument, NULL, 's'},
690
		{"superuser", required_argument, NULL, 'S'},
B
Hi,  
Bruce Momjian 已提交
691
		{"table", required_argument, NULL, 't'},
692 693
		{"password", no_argument, NULL, 'W'},
		{"username", required_argument, NULL, 'U'},
B
Hi,  
Bruce Momjian 已提交
694
		{"verbose", no_argument, NULL, 'v'},
695
		{"no-privileges", no_argument, NULL, 'x'},
B
Hi,  
Bruce Momjian 已提交
696
		{"no-acl", no_argument, NULL, 'x'},
B
Bruce Momjian 已提交
697
		{"compress", required_argument, NULL, 'Z'},
B
Hi,  
Bruce Momjian 已提交
698
		{"help", no_argument, NULL, '?'},
699 700 701 702 703
		{"version", no_argument, NULL, 'V'},

		/* the following options don't have an equivalent short option
           letter, but are available as '-X long-name' */
		{"use-set-session-authorization", no_argument, &use_setsessauth, 1}
704
	};
705 706
	int			optindex;

B
Hi,  
Bruce Momjian 已提交
707 708
#endif

709 710 711 712 713 714
#ifdef ENABLE_NLS
	setlocale(LC_ALL, "");
	bindtextdomain("pg_dump", LOCALEDIR);
	textdomain("pg_dump");
#endif

715
	g_verbose = false;
716
	force_quotes = true;
717 718 719 720 721

	strcpy(g_comment_start, "-- ");
	g_comment_end[0] = '\0';
	strcpy(g_opaque_type, "opaque");

722
	dataOnly = schemaOnly = dumpData = attrNames = false;
723

724
	if (!strrchr(argv[0], '/'))
725 726
		progname = argv[0];
	else
727
		progname = strrchr(argv[0], '/') + 1;
728

729
	/* Set default options based on progname */
730 731 732
	if (strcmp(progname, "pg_backup") == 0)
	{
		format = "c";
P
Philip Warner 已提交
733
		outputBlobs = true;
734
	}
735

736 737
	if (argc > 1)
	{
B
Bruce Momjian 已提交
738
		if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
739 740 741 742
		{
			help(progname);
			exit(0);
		}
B
Bruce Momjian 已提交
743
		if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
744
		{
745
			puts("pg_dump (PostgreSQL) " PG_VERSION);
746 747 748 749
			exit(0);
		}
	}

B
Hi,  
Bruce Momjian 已提交
750
#ifdef HAVE_GETOPT_LONG
751
	while ((c = getopt_long(argc, argv, "abcCdDf:F:h:inNoOp:RsS:t:uU:vWxX:zZ:V?", long_options, &optindex)) != -1)
B
Hi,  
Bruce Momjian 已提交
752
#else
753
	while ((c = getopt(argc, argv, "abcCdDf:F:h:inNoOp:RsS:t:uU:vWxX:zZ:V?-")) != -1)
B
Hi,  
Bruce Momjian 已提交
754
#endif
755

756 757 758
	{
		switch (c)
		{
759
			case 'a':			/* Dump data only */
760
				dataOnly = true;
761
				break;
762

763 764 765 766
			case 'b':			/* Dump blobs */
				outputBlobs = true;
				break;

B
Bruce Momjian 已提交
767
			case 'c':			/* clean (i.e., drop) schema prior to
B
Bruce Momjian 已提交
768
								 * create */
B
Bruce Momjian 已提交
769
				outputClean = 1;
B
Bruce Momjian 已提交
770 771
				break;

772 773 774 775 776
			case 'C':			/* Create DB */

				outputCreate = 1;
				break;

777
			case 'd':			/* dump data as proper insert strings */
778
				dumpData = true;
779
				break;
780

781 782
			case 'D':			/* dump data as proper insert strings with
								 * attr names */
783 784
				dumpData = true;
				attrNames = true;
785
				break;
786

787
			case 'f':
788 789
				filename = optarg;
				break;
790

B
Bruce Momjian 已提交
791 792 793
			case 'F':
				format = optarg;
				break;
794

B
Bruce Momjian 已提交
795
			case 'h':			/* server host */
796 797
				pghost = optarg;
				break;
798

799 800 801
			case 'i':			/* ignore database version mismatch */
				ignore_version = true;
				break;
802

B
Bruce Momjian 已提交
803 804
			case 'n':			/* Do not force double-quotes on
								 * identifiers */
805
				force_quotes = false;
806
				break;
807

808
			case 'N':			/* Force double-quotes on identifiers */
809
				force_quotes = true;
810
				break;
811

812
			case 'o':			/* Dump oids */
813
				oids = true;
814
				break;
815 816


817 818 819
			case 'O':			/* Don't reconnect to match owner */
				outputNoOwner = 1;
				break;
820

821 822 823
			case 'p':			/* server port */
				pgport = optarg;
				break;
824

825 826 827
			case 'R':			/* No reconnect */
				outputNoReconnect = 1;
				break;
828

829
			case 's':			/* dump schema only */
830
				schemaOnly = true;
831
				break;
832

B
Bruce Momjian 已提交
833 834
			case 'S':			/* Username for superuser in plain text
								 * output */
835 836
				outputSuperuser = strdup(optarg);
				break;
837

838
			case 't':			/* Dump data for this table only */
839
				{
840
					int			i;
841 842

					tablename = strdup(optarg);
B
Bruce Momjian 已提交
843 844 845 846 847

					/*
					 * quoted string? Then strip quotes and preserve
					 * case...
					 */
848 849 850
					if (tablename[0] == '"')
					{
						strcpy(tablename, &tablename[1]);
B
Bruce Momjian 已提交
851 852
						if (*(tablename + strlen(tablename) - 1) == '"')
							*(tablename + strlen(tablename) - 1) = '\0';
853 854 855 856 857
					}
					/* otherwise, convert table name to lowercase... */
					else
					{
						for (i = 0; tablename[i]; i++)
858 859
							if (isupper((unsigned char) tablename[i]))
								tablename[i] = tolower((unsigned char) tablename[i]);
860

B
Bruce Momjian 已提交
861 862 863 864 865
						/*
						 * '*' is a special case meaning ALL tables, but
						 * only if unquoted
						 */
						if (strcmp(tablename, "*") == 0)
866 867
							tablename[0] = '\0';

868
					}
869
				}
870
				break;
871

872
			case 'u':
873
				force_password = true;
874
				username = simple_prompt("User name: ", 100, true);
875 876 877 878
				break;

			case 'U':
				username = optarg;
879
				break;
880

881 882
			case 'v':			/* verbose */
				g_verbose = true;
883
				break;
884

885 886 887 888
			case 'W':
				force_password = true;
				break;

889 890
			case 'x':			/* skip ACL dump */
				aclsSkip = true;
891
				break;
892

893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912
				/*
				 * Option letters were getting scarce, so I invented
				 * this new scheme: '-X feature' turns on some
				 * feature.  Compare to the -f option in GCC.  You
				 * should also add an equivalent GNU-style option
				 * --feature.  Features that require arguments should
				 * use '-X feature=foo'.
				 */
			case 'X':
				if (strcmp(optarg, "use-set-session-authorization")==0)
					use_setsessauth = 1;
				else
				{
					fprintf(stderr,
							gettext("%s: invalid -X option -- %s\n"),
							progname, optarg);
					fprintf(stderr, gettext("Try '%s --help' for more information.\n"), progname);
					exit(1);
				}
				break;
B
Bruce Momjian 已提交
913 914 915
			case 'Z':			/* Compression Level */
				compressLevel = atoi(optarg);
				break;
916

917 918
#ifndef HAVE_GETOPT_LONG
			case '-':
919 920 921 922
				fprintf(stderr,
						gettext("%s was compiled without support for long options.\n"
								"Use --help for help on invocation options.\n"),
						progname);
923 924
				exit(1);
				break;
925 926 927 928
#else
				/* This covers the long options equivalent to -X xxx. */
			case 0:
				break;
929
#endif
930
			default:
931
				fprintf(stderr, gettext("Try '%s --help' for more information.\n"), progname);
932
				exit(1);
933 934 935
		}
	}

B
Bruce Momjian 已提交
936 937
	if (optind < (argc - 1))
	{
P
Philip Warner 已提交
938
		fprintf(stderr,
939 940 941
				gettext("%s: too many command line options (first is '%s')\n"
						"Try '%s --help' for more information.\n"),
				progname, argv[optind + 1], progname);
P
Philip Warner 已提交
942 943 944
		exit(1);
	}

945 946 947 948 949 950 951
	/* Get the target database name */
	if (optind < argc)
		dbname = argv[optind];
	else
		dbname = getenv("PGDATABASE");
	if (!dbname)
	{
952
		write_msg(NULL, "no database name specified\n");
953 954 955
		exit(1);
	}

B
Bruce Momjian 已提交
956 957
	if (dataOnly && schemaOnly)
	{
958
		write_msg(NULL, "The options \"schema only\" (-s) and \"data only\" (-a) cannot be used together.\n");
B
Bruce Momjian 已提交
959 960 961
		exit(1);
	}

B
Bruce Momjian 已提交
962
	if (outputBlobs && tablename != NULL && strlen(tablename) > 0)
963
	{
964
		write_msg(NULL, "Large object output is not supported for a single table.\n");
965
		write_msg(NULL, "Use all tables or a full dump instead.\n");
966 967 968
		exit(1);
	}

969 970
	if (dumpData == true && oids == true)
	{
971 972
		write_msg(NULL, "INSERT (-d, -D) and OID (-o) options cannot be used together.\n");
		write_msg(NULL, "(The INSERT command cannot set oids.)\n");
973
		exit(1);
974
	}
975

B
Bruce Momjian 已提交
976
	if (outputBlobs == true && (format[0] == 'p' || format[0] == 'P'))
P
Philip Warner 已提交
977
	{
978
		write_msg(NULL, "large object output is not supported for plain text dump files.\n");
979
		write_msg(NULL, "(Use a different output format.)\n");
P
Philip Warner 已提交
980 981 982
		exit(1);
	}

983
	/* open the output file */
B
Bruce Momjian 已提交
984 985
	switch (format[0])
	{
B
Bruce Momjian 已提交
986 987 988 989 990 991 992 993 994 995 996 997 998 999

		case 'c':
		case 'C':
			g_fout = CreateArchive(filename, archCustom, compressLevel);
			break;

		case 'f':
		case 'F':
			g_fout = CreateArchive(filename, archFiles, compressLevel);
			break;

		case 'p':
		case 'P':
			plainText = 1;
1000 1001 1002 1003 1004 1005
			g_fout = CreateArchive(filename, archNull, 0);
			break;

		case 't':
		case 'T':
			g_fout = CreateArchive(filename, archTar, compressLevel);
B
Bruce Momjian 已提交
1006 1007 1008
			break;

		default:
1009
			write_msg(NULL, "invalid output format '%s' specified\n", format);
B
Bruce Momjian 已提交
1010
			exit(1);
B
Bruce Momjian 已提交
1011 1012 1013 1014
	}

	if (g_fout == NULL)
	{
1015
		write_msg(NULL, "could not open output file %s for writing\n", filename);
B
Bruce Momjian 已提交
1016
		exit(1);
1017 1018
	}

1019 1020
	/* Let the archiver know how noisy to be */
	g_fout->verbose = g_verbose;
1021

B
Bruce Momjian 已提交
1022 1023
	/*
	 * Open the database using the Archiver, so it knows about it. Errors
1024
	 * mean death.
B
Bruce Momjian 已提交
1025
	 */
1026 1027
	g_fout->minRemoteVersion = 70000; /* we can handle back to 7.0 */
	g_fout->maxRemoteVersion = parse_version(PG_VERSION);
1028
	g_conn = ConnectDatabase(g_fout, dbname, pghost, pgport, username, force_password, ignore_version);
1029

1030 1031 1032 1033 1034 1035 1036 1037
	/*
	 * Start serializable transaction to dump consistent data
	 */
	{
		PGresult   *res;

		res = PQexec(g_conn, "begin");
		if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
1038
			exit_horribly(g_fout, NULL, "BEGIN command failed: %s",
B
Bruce Momjian 已提交
1039
						  PQerrorMessage(g_conn));
1040

1041 1042 1043
		PQclear(res);
		res = PQexec(g_conn, "set transaction isolation level serializable");
		if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
1044
			exit_horribly(g_fout, NULL, "could not set transaction isolation level to serializable: %s",
B
Bruce Momjian 已提交
1045
						  PQerrorMessage(g_conn));
1046

1047 1048 1049
		PQclear(res);
	}

1050 1051 1052 1053
	if (g_fout->remoteVersion >= 70100)
		g_last_builtin_oid = findLastBuiltinOid_V71(dbname);
	else
		g_last_builtin_oid = findLastBuiltinOid_V70();
1054

1055 1056 1057 1058
	/* Dump the database definition */
	if (!dataOnly)
		dumpDatabase(g_fout);

1059
	if (oids == true)
1060
		setMaxOid(g_fout);
B
Bruce Momjian 已提交
1061 1062

	if (g_verbose)
1063
		write_msg(NULL, "last built-in oid is %u\n", g_last_builtin_oid);
B
Bruce Momjian 已提交
1064
	tblinfo = dumpSchema(g_fout, &numTables, tablename, aclsSkip, oids, schemaOnly, dataOnly);
1065

1066
	if (!schemaOnly)
B
Bruce Momjian 已提交
1067
		dumpClasses(tblinfo, numTables, g_fout, tablename, oids, force_quotes);
1068 1069

	if (outputBlobs)
B
Bruce Momjian 已提交
1070
		ArchiveEntry(g_fout, "0", "BLOBS", "BLOBS", NULL, "", "", "", "", dumpBlobs, 0);
1071

1072 1073
	if (!dataOnly)				/* dump indexes and triggers at the end
								 * for performance */
V
Vadim B. Mikheev 已提交
1074 1075
	{
		dumpTriggers(g_fout, tablename, tblinfo, numTables);
1076
		dumpRules(g_fout, tablename, tblinfo, numTables);
V
Vadim B. Mikheev 已提交
1077
	}
1078

1079 1080
	/* Now sort the output nicely */
	SortTocByOID(g_fout);
1081
	MoveToStart(g_fout, "DATABASE");
1082 1083 1084
	MoveToEnd(g_fout, "TABLE DATA");
	MoveToEnd(g_fout, "BLOBS");
	MoveToEnd(g_fout, "INDEX");
1085
	MoveToEnd(g_fout, "CONSTRAINT");
1086 1087
	MoveToEnd(g_fout, "TRIGGER");
	MoveToEnd(g_fout, "RULE");
1088
	MoveToEnd(g_fout, "SEQUENCE SET");
1089 1090 1091 1092 1093 1094
	/*
	 * Moving all comments to end is annoying, but must do it for comments
	 * on stuff we just moved, and we don't seem to have quite enough
	 * dependency structure to get it really right...
	 */
	MoveToEnd(g_fout, "COMMENT");
1095

B
Bruce Momjian 已提交
1096
	if (plainText)
B
Bruce Momjian 已提交
1097 1098
	{
		ropt = NewRestoreOptions();
B
Bruce Momjian 已提交
1099
		ropt->filename = (char *) filename;
B
Bruce Momjian 已提交
1100 1101
		ropt->dropSchema = outputClean;
		ropt->aclsSkip = aclsSkip;
1102 1103 1104 1105
		ropt->superuser = outputSuperuser;
		ropt->create = outputCreate;
		ropt->noOwner = outputNoOwner;
		ropt->noReconnect = outputNoReconnect;
1106
		ropt->use_setsessauth = use_setsessauth;
1107 1108 1109 1110 1111

		if (outputSuperuser)
			ropt->superuser = outputSuperuser;
		else
			ropt->superuser = PQuser(g_conn);
B
Bruce Momjian 已提交
1112 1113

		if (compressLevel == -1)
B
Bruce Momjian 已提交
1114
			ropt->compression = 0;
B
Bruce Momjian 已提交
1115
		else
B
Bruce Momjian 已提交
1116
			ropt->compression = compressLevel;
B
Bruce Momjian 已提交
1117

1118 1119
		ropt->suppressDumpWarnings = true; /* We've already shown them */

B
Bruce Momjian 已提交
1120 1121 1122 1123
		RestoreArchive(g_fout, ropt);
	}

	CloseArchive(g_fout);
1124

1125 1126 1127
	clearTableInfo(tblinfo, numTables);
	PQfinish(g_conn);
	exit(0);
1128 1129
}

1130 1131 1132 1133 1134
/*
 * dumpDatabase:
 *	dump the database definition
 *
 */
B
Bruce Momjian 已提交
1135
static int
1136 1137
dumpDatabase(Archive *AH)
{
B
Bruce Momjian 已提交
1138 1139 1140 1141 1142 1143
	PQExpBuffer dbQry = createPQExpBuffer();
	PQExpBuffer delQry = createPQExpBuffer();
	PQExpBuffer creaQry = createPQExpBuffer();
	PGresult   *res;
	int			ntups;
	int			i_dba;
1144 1145

	if (g_verbose)
1146
		write_msg(NULL, "saving database definition\n");
1147 1148

	/* Get the dba */
1149
	appendPQExpBuffer(dbQry, "select (select usename from pg_user where datdba = usesysid) as dba from pg_database"
B
Bruce Momjian 已提交
1150
					  " where datname = ");
1151
	formatStringLiteral(dbQry, PQdb(g_conn), CONV_ALL);
1152 1153 1154 1155 1156

	res = PQexec(g_conn, dbQry->data);
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1157 1158 1159
		write_msg(NULL, "SQL command failed\n");
		write_msg(NULL, "Error message from server: %s", PQerrorMessage(g_conn));
		write_msg(NULL, "The command was: %s\n", dbQry->data);
1160
		exit_nicely();
1161 1162 1163 1164
	}

	ntups = PQntuples(res);

1165 1166 1167
	if (ntups <= 0)
	{
		write_msg(NULL, "missing pg_database entry for database \"%s\"\n", PQdb(g_conn));
1168
		exit_nicely();
1169 1170
	}

1171 1172
	if (ntups != 1)
	{
1173 1174
		write_msg(NULL, "query returned more than one (%d) pg_database entry for database \"%s\"\n",
				  ntups, PQdb(g_conn));
1175
		exit_nicely();
1176 1177 1178 1179 1180 1181
	}

	appendPQExpBuffer(creaQry, "Create Database \"%s\";\n", PQdb(g_conn));
	appendPQExpBuffer(delQry, "Drop Database \"%s\";\n", PQdb(g_conn));
	i_dba = PQfnumber(res, "dba");

B
Bruce Momjian 已提交
1182 1183 1184 1185
	ArchiveEntry(AH, "0" /* OID */ , PQdb(g_conn) /* Name */ , "DATABASE", NULL,
				 creaQry->data /* Create */ , delQry->data /* Del */ ,
				 "" /* Copy */ , PQgetvalue(res, 0, i_dba) /* Owner */ ,
				 NULL /* Dumper */ , NULL /* Dumper Arg */ );
1186 1187 1188

	PQclear(res);

1189 1190 1191 1192
	destroyPQExpBuffer(dbQry);
	destroyPQExpBuffer(delQry);
	destroyPQExpBuffer(creaQry);

1193 1194 1195 1196
	return 1;
}


1197 1198 1199 1200 1201 1202
/*
 * dumpBlobs:
 *	dump all blobs
 *
 */

B
Bruce Momjian 已提交
1203
#define loBufSize 16384
1204 1205
#define loFetchSize 1000

B
Bruce Momjian 已提交
1206 1207
static int
dumpBlobs(Archive *AH, char *junkOid, void *junkVal)
1208
{
B
Bruce Momjian 已提交
1209 1210 1211 1212 1213 1214 1215
	PQExpBuffer oidQry = createPQExpBuffer();
	PQExpBuffer oidFetchQry = createPQExpBuffer();
	PGresult   *res;
	int			i;
	int			loFd;
	char		buf[loBufSize];
	int			cnt;
1216
	Oid			blobOid;
1217 1218

	if (g_verbose)
1219
		write_msg(NULL, "saving large objects\n");
1220 1221

	/* Cursor to get all BLOB tables */
1222 1223 1224 1225 1226 1227
	if (AH->remoteVersion >= 70100)
	{
		appendPQExpBuffer(oidQry, "Declare blobOid Cursor for SELECT DISTINCT loid FROM pg_largeobject");
	} else {
		appendPQExpBuffer(oidQry, "Declare blobOid Cursor for SELECT oid from pg_class where relkind = 'l'");
	}
1228 1229 1230 1231

	res = PQexec(g_conn, oidQry->data);
	if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
	{
1232
		write_msg(NULL, "dumpBlobs(): cursor declaration failed: %s", PQerrorMessage(g_conn));
1233
		exit_nicely();
1234 1235 1236 1237 1238
	}

	/* Fetch for cursor */
	appendPQExpBuffer(oidFetchQry, "Fetch %d in blobOid", loFetchSize);

B
Bruce Momjian 已提交
1239 1240
	do
	{
1241 1242 1243 1244 1245 1246
		/* Do a fetch */
		PQclear(res);
		res = PQexec(g_conn, oidFetchQry->data);

		if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
		{
1247 1248
			write_msg(NULL, "dumpBlobs(): fetch from cursor failed: %s",
					  PQerrorMessage(g_conn));
1249
			exit_nicely();
1250 1251 1252 1253 1254
		}

		/* Process the tuples, if any */
		for (i = 0; i < PQntuples(res); i++)
		{
1255
			blobOid = atooid(PQgetvalue(res, i, 0));
1256 1257 1258 1259
			/* Open the BLOB */
			loFd = lo_open(g_conn, blobOid, INV_READ);
			if (loFd == -1)
			{
1260 1261
				write_msg(NULL, "dumpBlobs(): could not open large object: %s",
						  PQerrorMessage(g_conn));
1262
				exit_nicely();
1263 1264 1265 1266 1267
			}

			StartBlob(AH, blobOid);

			/* Now read it in chunks, sending data to archive */
B
Bruce Momjian 已提交
1268 1269
			do
			{
1270
				cnt = lo_read(g_conn, loFd, buf, loBufSize);
B
Bruce Momjian 已提交
1271 1272
				if (cnt < 0)
				{
1273 1274
					write_msg(NULL, "dumpBlobs(): error reading large object: %s",
							  PQerrorMessage(g_conn));
1275
					exit_nicely();
1276 1277
				}

B
Bruce Momjian 已提交
1278
				WriteData(AH, buf, cnt);
1279 1280 1281 1282 1283 1284 1285 1286 1287 1288

			} while (cnt > 0);

			lo_close(g_conn, loFd);

			EndBlob(AH, blobOid);

		}
	} while (PQntuples(res) > 0);

1289 1290 1291
	destroyPQExpBuffer(oidQry);
	destroyPQExpBuffer(oidFetchQry);

1292 1293 1294
	return 1;
}

1295
/*
1296 1297
 * getTypes:
 *	  read all base types in the system catalogs and return them in the
1298 1299
 * TypeInfo* structure
 *
1300
 *	numTypes is set to the number of types read in
1301 1302
 *
 */
1303
TypeInfo   *
1304 1305
getTypes(int *numTypes)
{
B
Bruce Momjian 已提交
1306
	PGresult   *res;
1307 1308
	int			ntups;
	int			i;
1309
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1310
	TypeInfo   *tinfo;
1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324

	int			i_oid;
	int			i_typowner;
	int			i_typname;
	int			i_typlen;
	int			i_typprtlen;
	int			i_typinput;
	int			i_typoutput;
	int			i_typreceive;
	int			i_typsend;
	int			i_typelem;
	int			i_typdelim;
	int			i_typdefault;
	int			i_typrelid;
1325 1326
	int			i_typalign;
	int			i_typstorage;
1327
	int			i_typbyval;
1328
	int			i_typisdefined;
1329
	int			i_usename;
1330
	int			i_typedefn;
1331 1332 1333 1334 1335 1336

	/* find all base types */

	/*
	 * we include even the built-in types because those may be used as
	 * array elements by user-defined types
1337
	 *
1338 1339 1340
	 * we filter out the built-in types when we dump out the types
	 */

1341 1342 1343 1344
	if (g_fout->remoteVersion < 70100)
	{
		appendPQExpBuffer(query, "SELECT pg_type.oid, typowner, typname, typlen, typprtlen, "
						"typinput, typoutput, typreceive, typsend, typelem, typdelim, "
1345
						"typdefault, typrelid, typalign, 'p'::char as typstorage, typbyval, typisdefined, "
1346 1347 1348 1349 1350 1351
						"(select usename from pg_user where typowner = usesysid) as usename, "
						"typname as typedefn "
						"from pg_type");
	} else {
		appendPQExpBuffer(query, "SELECT pg_type.oid, typowner, typname, typlen, typprtlen, "
						"typinput, typoutput, typreceive, typsend, typelem, typdelim, "
1352
						"typdefault, typrelid, typalign, typstorage, typbyval, typisdefined, "
1353 1354 1355 1356
						"(select usename from pg_user where typowner = usesysid) as usename, "
						"format_type(pg_type.oid, NULL) as typedefn "
						"from pg_type");
	}
1357

B
Hi, all  
Bruce Momjian 已提交
1358
	res = PQexec(g_conn, query->data);
1359 1360 1361
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1362
		write_msg(NULL, "query to obtain list of data types failed: %s", PQerrorMessage(g_conn));
1363
		exit_nicely();
1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382
	}

	ntups = PQntuples(res);

	tinfo = (TypeInfo *) malloc(ntups * sizeof(TypeInfo));

	i_oid = PQfnumber(res, "oid");
	i_typowner = PQfnumber(res, "typowner");
	i_typname = PQfnumber(res, "typname");
	i_typlen = PQfnumber(res, "typlen");
	i_typprtlen = PQfnumber(res, "typprtlen");
	i_typinput = PQfnumber(res, "typinput");
	i_typoutput = PQfnumber(res, "typoutput");
	i_typreceive = PQfnumber(res, "typreceive");
	i_typsend = PQfnumber(res, "typsend");
	i_typelem = PQfnumber(res, "typelem");
	i_typdelim = PQfnumber(res, "typdelim");
	i_typdefault = PQfnumber(res, "typdefault");
	i_typrelid = PQfnumber(res, "typrelid");
1383 1384
	i_typalign = PQfnumber(res, "typalign");
	i_typstorage = PQfnumber(res, "typstorage");
1385
	i_typbyval = PQfnumber(res, "typbyval");
1386
	i_typisdefined = PQfnumber(res, "typisdefined");
1387
	i_usename = PQfnumber(res, "usename");
1388
	i_typedefn = PQfnumber(res, "typedefn");
1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402

	for (i = 0; i < ntups; i++)
	{
		tinfo[i].oid = strdup(PQgetvalue(res, i, i_oid));
		tinfo[i].typowner = strdup(PQgetvalue(res, i, i_typowner));
		tinfo[i].typname = strdup(PQgetvalue(res, i, i_typname));
		tinfo[i].typlen = strdup(PQgetvalue(res, i, i_typlen));
		tinfo[i].typprtlen = strdup(PQgetvalue(res, i, i_typprtlen));
		tinfo[i].typinput = strdup(PQgetvalue(res, i, i_typinput));
		tinfo[i].typoutput = strdup(PQgetvalue(res, i, i_typoutput));
		tinfo[i].typreceive = strdup(PQgetvalue(res, i, i_typreceive));
		tinfo[i].typsend = strdup(PQgetvalue(res, i, i_typsend));
		tinfo[i].typelem = strdup(PQgetvalue(res, i, i_typelem));
		tinfo[i].typdelim = strdup(PQgetvalue(res, i, i_typdelim));
1403 1404 1405 1406
		if (PQgetisnull(res, i, i_typdefault))
			tinfo[i].typdefault = NULL;
		else
			tinfo[i].typdefault = strdup(PQgetvalue(res, i, i_typdefault));
1407
		tinfo[i].typrelid = strdup(PQgetvalue(res, i, i_typrelid));
1408 1409
		tinfo[i].typalign = strdup(PQgetvalue(res, i, i_typalign));
		tinfo[i].typstorage = strdup(PQgetvalue(res, i, i_typstorage));
1410
		tinfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1411
		tinfo[i].typedefn = strdup(PQgetvalue(res, i, i_typedefn));
1412

1413
		if (strlen(tinfo[i].usename) == 0)
1414 1415
			write_msg(NULL, "WARNING: owner of data type %s appears to be invalid\n",
					  tinfo[i].typname);
1416

1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429
		if (strcmp(PQgetvalue(res, i, i_typbyval), "f") == 0)
			tinfo[i].passedbyvalue = 0;
		else
			tinfo[i].passedbyvalue = 1;

		/*
		 * check for user-defined array types, omit system generated ones
		 */
		if ((strcmp(tinfo[i].typelem, "0") != 0) &&
			tinfo[i].typname[0] != '_')
			tinfo[i].isArray = 1;
		else
			tinfo[i].isArray = 0;
1430 1431 1432 1433 1434

		if (strcmp(PQgetvalue(res, i, i_typisdefined), "f") == 0)
			tinfo[i].isDefined = 0;
		else
			tinfo[i].isDefined = 1;
1435 1436 1437 1438 1439 1440
	}

	*numTypes = ntups;

	PQclear(res);

1441 1442
	destroyPQExpBuffer(query);

1443
	return tinfo;
1444 1445 1446 1447
}

/*
 * getOperators:
1448
 *	  read all operators in the system catalogs and return them in the
1449 1450
 * OprInfo* structure
 *
1451
 *	numOprs is set to the number of operators read in
1452
 */
1453
OprInfo    *
1454 1455
getOperators(int *numOprs)
{
1456
	PGresult   *res;
1457 1458
	int			ntups;
	int			i;
1459
	PQExpBuffer query = createPQExpBuffer();
1460

B
Bruce Momjian 已提交
1461
	OprInfo    *oprinfo;
1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476

	int			i_oid;
	int			i_oprname;
	int			i_oprkind;
	int			i_oprcode;
	int			i_oprleft;
	int			i_oprright;
	int			i_oprcom;
	int			i_oprnegate;
	int			i_oprrest;
	int			i_oprjoin;
	int			i_oprcanhash;
	int			i_oprlsortop;
	int			i_oprrsortop;
	int			i_usename;
1477 1478 1479 1480 1481 1482

	/*
	 * find all operators, including builtin operators, filter out
	 * system-defined operators at dump-out time
	 */

B
Hi, all  
Bruce Momjian 已提交
1483
	appendPQExpBuffer(query, "SELECT pg_operator.oid, oprname, oprkind, oprcode, "
1484
			   "oprleft, oprright, oprcom, oprnegate, oprrest, oprjoin, "
B
Bruce Momjian 已提交
1485 1486 1487
					  "oprcanhash, oprlsortop, oprrsortop, "
	"(select usename from pg_user where oprowner = usesysid) as usename "
					  "from pg_operator");
1488

B
Hi, all  
Bruce Momjian 已提交
1489
	res = PQexec(g_conn, query->data);
1490 1491 1492
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1493
		write_msg(NULL, "query to obtain list of operators failed: %s", PQerrorMessage(g_conn));
1494
		exit_nicely();
1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532
	}

	ntups = PQntuples(res);
	*numOprs = ntups;

	oprinfo = (OprInfo *) malloc(ntups * sizeof(OprInfo));

	i_oid = PQfnumber(res, "oid");
	i_oprname = PQfnumber(res, "oprname");
	i_oprkind = PQfnumber(res, "oprkind");
	i_oprcode = PQfnumber(res, "oprcode");
	i_oprleft = PQfnumber(res, "oprleft");
	i_oprright = PQfnumber(res, "oprright");
	i_oprcom = PQfnumber(res, "oprcom");
	i_oprnegate = PQfnumber(res, "oprnegate");
	i_oprrest = PQfnumber(res, "oprrest");
	i_oprjoin = PQfnumber(res, "oprjoin");
	i_oprcanhash = PQfnumber(res, "oprcanhash");
	i_oprlsortop = PQfnumber(res, "oprlsortop");
	i_oprrsortop = PQfnumber(res, "oprrsortop");
	i_usename = PQfnumber(res, "usename");

	for (i = 0; i < ntups; i++)
	{
		oprinfo[i].oid = strdup(PQgetvalue(res, i, i_oid));
		oprinfo[i].oprname = strdup(PQgetvalue(res, i, i_oprname));
		oprinfo[i].oprkind = strdup(PQgetvalue(res, i, i_oprkind));
		oprinfo[i].oprcode = strdup(PQgetvalue(res, i, i_oprcode));
		oprinfo[i].oprleft = strdup(PQgetvalue(res, i, i_oprleft));
		oprinfo[i].oprright = strdup(PQgetvalue(res, i, i_oprright));
		oprinfo[i].oprcom = strdup(PQgetvalue(res, i, i_oprcom));
		oprinfo[i].oprnegate = strdup(PQgetvalue(res, i, i_oprnegate));
		oprinfo[i].oprrest = strdup(PQgetvalue(res, i, i_oprrest));
		oprinfo[i].oprjoin = strdup(PQgetvalue(res, i, i_oprjoin));
		oprinfo[i].oprcanhash = strdup(PQgetvalue(res, i, i_oprcanhash));
		oprinfo[i].oprlsortop = strdup(PQgetvalue(res, i, i_oprlsortop));
		oprinfo[i].oprrsortop = strdup(PQgetvalue(res, i, i_oprrsortop));
		oprinfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1533 1534

		if (strlen(oprinfo[i].usename) == 0)
1535 1536
			write_msg(NULL, "WARNING: owner of operator \"%s\" appears to be invalid\n",
					  oprinfo[i].oprname);
1537

1538 1539 1540 1541
	}

	PQclear(res);

1542 1543
	destroyPQExpBuffer(query);

1544
	return oprinfo;
1545 1546
}

1547
void
1548
clearTypeInfo(TypeInfo *tp, int numTypes)
1549
{
1550
	int			i;
1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579

	for (i = 0; i < numTypes; ++i)
	{
		if (tp[i].oid)
			free(tp[i].oid);
		if (tp[i].typowner)
			free(tp[i].typowner);
		if (tp[i].typname)
			free(tp[i].typname);
		if (tp[i].typlen)
			free(tp[i].typlen);
		if (tp[i].typprtlen)
			free(tp[i].typprtlen);
		if (tp[i].typinput)
			free(tp[i].typinput);
		if (tp[i].typoutput)
			free(tp[i].typoutput);
		if (tp[i].typreceive)
			free(tp[i].typreceive);
		if (tp[i].typsend)
			free(tp[i].typsend);
		if (tp[i].typelem)
			free(tp[i].typelem);
		if (tp[i].typdelim)
			free(tp[i].typdelim);
		if (tp[i].typdefault)
			free(tp[i].typdefault);
		if (tp[i].typrelid)
			free(tp[i].typrelid);
1580 1581 1582 1583
		if (tp[i].typalign)
			free(tp[i].typalign);
		if (tp[i].typstorage)
			free(tp[i].typstorage);
1584 1585
		if (tp[i].usename)
			free(tp[i].usename);
1586 1587
		if (tp[i].typedefn)
			free(tp[i].typedefn);
1588 1589
	}
	free(tp);
1590 1591 1592
}

void
1593
clearFuncInfo(FuncInfo *fun, int numFuncs)
1594
{
1595 1596
	int			i,
				a;
1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607

	if (!fun)
		return;
	for (i = 0; i < numFuncs; ++i)
	{
		if (fun[i].oid)
			free(fun[i].oid);
		if (fun[i].proname)
			free(fun[i].proname);
		if (fun[i].usename)
			free(fun[i].usename);
1608
		for (a = 0; a < FUNC_MAX_ARGS; ++a)
1609 1610 1611 1612 1613 1614 1615 1616 1617 1618
			if (fun[i].argtypes[a])
				free(fun[i].argtypes[a]);
		if (fun[i].prorettype)
			free(fun[i].prorettype);
		if (fun[i].prosrc)
			free(fun[i].prosrc);
		if (fun[i].probin)
			free(fun[i].probin);
	}
	free(fun);
1619 1620
}

1621
static void
1622
clearTableInfo(TableInfo *tblinfo, int numTables)
1623
{
1624 1625
	int			i,
				j;
1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650

	for (i = 0; i < numTables; ++i)
	{

		if (tblinfo[i].oid)
			free(tblinfo[i].oid);
		if (tblinfo[i].relacl)
			free(tblinfo[i].relacl);
		if (tblinfo[i].usename)
			free(tblinfo[i].usename);

		if (tblinfo[i].relname)
			free(tblinfo[i].relname);

		if (tblinfo[i].sequence)
			continue;

		/* Process Attributes */
		for (j = 0; j < tblinfo[i].numatts; j++)
		{
			if (tblinfo[i].attnames[j])
				free(tblinfo[i].attnames[j]);
			if (tblinfo[i].typnames[j])
				free(tblinfo[i].typnames[j]);
		}
B
Bruce Momjian 已提交
1651

B
Bruce Momjian 已提交
1652 1653 1654
		if (tblinfo[i].triggers)
		{
			for (j = 0; j < tblinfo[i].ntrig; j++)
B
Bruce Momjian 已提交
1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667
			{
				if (tblinfo[i].triggers[j].tgsrc)
					free(tblinfo[i].triggers[j].tgsrc);
				if (tblinfo[i].triggers[j].oid)
					free(tblinfo[i].triggers[j].oid);
				if (tblinfo[i].triggers[j].tgname)
					free(tblinfo[i].triggers[j].tgname);
				if (tblinfo[i].triggers[j].tgdel)
					free(tblinfo[i].triggers[j].tgdel);
			}
			free(tblinfo[i].triggers);
		}

1668 1669
		if (tblinfo[i].atttypmod)
			free((int *) tblinfo[i].atttypmod);
1670 1671
		if (tblinfo[i].inhAttrs)
			free((int *) tblinfo[i].inhAttrs);
1672 1673 1674 1675
		if (tblinfo[i].inhAttrDef)
			free((int *) tblinfo[i].inhAttrDef);
		if (tblinfo[i].inhNotNull)
			free((int *) tblinfo[i].inhNotNull);
1676 1677
		if (tblinfo[i].attnames)
			free(tblinfo[i].attnames);
1678 1679
		if (tblinfo[i].atttypedefns)
			free(tblinfo[i].atttypedefns);
1680 1681 1682 1683
		if (tblinfo[i].typnames)
			free(tblinfo[i].typnames);
		if (tblinfo[i].notnull)
			free(tblinfo[i].notnull);
1684 1685
		if (tblinfo[i].primary_key_name)
			free(tblinfo[i].primary_key_name);
1686 1687
	}
	free(tblinfo);
1688 1689
}

1690
void
1691
clearInhInfo(InhInfo *inh, int numInherits)
1692
{
1693
	int			i;
1694 1695 1696 1697 1698

	if (!inh)
		return;
	for (i = 0; i < numInherits; ++i)
	{
1699 1700
		if (inh[i].inhrelid)
			free(inh[i].inhrelid);
1701 1702 1703 1704
		if (inh[i].inhparent)
			free(inh[i].inhparent);
	}
	free(inh);
1705 1706 1707
}

void
1708
clearOprInfo(OprInfo *opr, int numOprs)
1709
{
1710
	int			i;
1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745

	if (!opr)
		return;
	for (i = 0; i < numOprs; ++i)
	{
		if (opr[i].oid)
			free(opr[i].oid);
		if (opr[i].oprname)
			free(opr[i].oprname);
		if (opr[i].oprkind)
			free(opr[i].oprkind);
		if (opr[i].oprcode)
			free(opr[i].oprcode);
		if (opr[i].oprleft)
			free(opr[i].oprleft);
		if (opr[i].oprright)
			free(opr[i].oprright);
		if (opr[i].oprcom)
			free(opr[i].oprcom);
		if (opr[i].oprnegate)
			free(opr[i].oprnegate);
		if (opr[i].oprrest)
			free(opr[i].oprrest);
		if (opr[i].oprjoin)
			free(opr[i].oprjoin);
		if (opr[i].oprcanhash)
			free(opr[i].oprcanhash);
		if (opr[i].oprlsortop)
			free(opr[i].oprlsortop);
		if (opr[i].oprrsortop)
			free(opr[i].oprrsortop);
		if (opr[i].usename)
			free(opr[i].usename);
	}
	free(opr);
1746 1747 1748
}

void
1749
clearIndInfo(IndInfo *ind, int numIndexes)
1750
{
1751 1752
	int			i,
				a;
1753 1754 1755

	if (!ind)
		return;
1756
	for (i = 0; i < numIndexes; ++i)
1757
	{
1758 1759 1760 1761
		if (ind[i].indexreloid)
			free(ind[i].indexreloid);
		if (ind[i].indreloid)
			free(ind[i].indreloid);
1762 1763 1764 1765
		if (ind[i].indexrelname)
			free(ind[i].indexrelname);
		if (ind[i].indrelname)
			free(ind[i].indrelname);
1766 1767
		if (ind[i].indexdef)
			free(ind[i].indexdef);
1768 1769
		if (ind[i].indisprimary)
			free(ind[i].indisprimary);
1770 1771 1772 1773 1774 1775 1776
		for (a = 0; a < INDEX_MAX_KEYS; ++a)
		{
			if (ind[i].indkey[a])
				free(ind[i].indkey[a]);
		}
	}
	free(ind);
1777 1778 1779
}

void
B
Bruce Momjian 已提交
1780
clearAggInfo(AggInfo *agginfo, int numArgs)
1781
{
1782
	int			i;
1783 1784 1785 1786 1787 1788 1789 1790 1791

	if (!agginfo)
		return;
	for (i = 0; i < numArgs; ++i)
	{
		if (agginfo[i].oid)
			free(agginfo[i].oid);
		if (agginfo[i].aggname)
			free(agginfo[i].aggname);
1792 1793
		if (agginfo[i].aggtransfn)
			free(agginfo[i].aggtransfn);
1794 1795
		if (agginfo[i].aggfinalfn)
			free(agginfo[i].aggfinalfn);
1796 1797
		if (agginfo[i].aggtranstype)
			free(agginfo[i].aggtranstype);
1798 1799
		if (agginfo[i].aggbasetype)
			free(agginfo[i].aggbasetype);
1800 1801
		if (agginfo[i].agginitval)
			free(agginfo[i].agginitval);
1802 1803 1804 1805
		if (agginfo[i].usename)
			free(agginfo[i].usename);
	}
	free(agginfo);
1806
}
1807 1808 1809

/*
 * getAggregates:
1810
 *	  read all the user-defined aggregates in the system catalogs and
1811 1812
 * return them in the AggInfo* structure
 *
1813
 * numAggs is set to the number of aggregates read in
1814
 */
1815
AggInfo    *
1816 1817
getAggregates(int *numAggs)
{
B
Bruce Momjian 已提交
1818
	PGresult   *res;
1819 1820
	int			ntups;
	int			i;
1821
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1822
	AggInfo    *agginfo;
1823 1824 1825

	int			i_oid;
	int			i_aggname;
1826
	int			i_aggtransfn;
1827
	int			i_aggfinalfn;
1828
	int			i_aggtranstype;
1829
	int			i_aggbasetype;
1830
	int			i_agginitval;
1831
	int			i_usename;
1832
	int			i_convertok;
1833 1834 1835

	/* find all user-defined aggregates */

1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851
	if (g_fout->remoteVersion < 70100)
	{
		appendPQExpBuffer(query, "SELECT pg_aggregate.oid, aggname, aggtransfn1 as aggtransfn, "
			"aggfinalfn, aggtranstype1 as aggtranstype, aggbasetype, "
			"agginitval1 as agginitval, "
			"(aggtransfn2 = 0 and aggtranstype2 = 0 and agginitval2 is null) as convertok, "
			"(select usename from pg_user where aggowner = usesysid) as usename "
			"from pg_aggregate");
	} else {
		appendPQExpBuffer(query, "SELECT pg_aggregate.oid, aggname, aggtransfn, "
			"aggfinalfn, aggtranstype, aggbasetype, "
			"agginitval, "
			"'t'::boolean as convertok, "
			"(select usename from pg_user where aggowner = usesysid) as usename "
			"from pg_aggregate");
	}
1852

B
Hi, all  
Bruce Momjian 已提交
1853
	res = PQexec(g_conn, query->data);
1854 1855 1856
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1857
		write_msg(NULL, "query to obtain list of aggregate functions failed: %s",
1858
				  PQerrorMessage(g_conn));
1859
		exit_nicely();
1860 1861 1862 1863 1864 1865 1866 1867 1868
	}

	ntups = PQntuples(res);
	*numAggs = ntups;

	agginfo = (AggInfo *) malloc(ntups * sizeof(AggInfo));

	i_oid = PQfnumber(res, "oid");
	i_aggname = PQfnumber(res, "aggname");
1869
	i_aggtransfn = PQfnumber(res, "aggtransfn");
1870
	i_aggfinalfn = PQfnumber(res, "aggfinalfn");
1871
	i_aggtranstype = PQfnumber(res, "aggtranstype");
1872
	i_aggbasetype = PQfnumber(res, "aggbasetype");
1873
	i_agginitval = PQfnumber(res, "agginitval");
1874
	i_usename = PQfnumber(res, "usename");
1875
	i_convertok = PQfnumber(res, "convertok");
1876 1877 1878 1879 1880

	for (i = 0; i < ntups; i++)
	{
		agginfo[i].oid = strdup(PQgetvalue(res, i, i_oid));
		agginfo[i].aggname = strdup(PQgetvalue(res, i, i_aggname));
1881
		agginfo[i].aggtransfn = strdup(PQgetvalue(res, i, i_aggtransfn));
1882
		agginfo[i].aggfinalfn = strdup(PQgetvalue(res, i, i_aggfinalfn));
1883
		agginfo[i].aggtranstype = strdup(PQgetvalue(res, i, i_aggtranstype));
1884
		agginfo[i].aggbasetype = strdup(PQgetvalue(res, i, i_aggbasetype));
1885
		agginfo[i].agginitval = strdup(PQgetvalue(res, i, i_agginitval));
1886
		agginfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1887
		if (strlen(agginfo[i].usename) == 0)
1888 1889
			write_msg(NULL, "WARNING: owner of aggregate function \"%s\" appears to be invalid\n",
					  agginfo[i].aggname);
1890

1891 1892
		agginfo[i].convertok = (PQgetvalue(res, i, i_convertok)[0] == 't');

1893 1894 1895 1896
	}

	PQclear(res);

1897 1898
	destroyPQExpBuffer(query);

1899
	return agginfo;
1900 1901 1902 1903
}

/*
 * getFuncs:
1904
 *	  read all the user-defined functions in the system catalogs and
1905 1906
 * return them in the FuncInfo* structure
 *
1907 1908
 * numFuncs is set to the number of functions read in
 *
1909 1910
 *
 */
1911
FuncInfo   *
1912 1913
getFuncs(int *numFuncs)
{
1914
	PGresult   *res;
1915 1916
	int			ntups;
	int			i;
1917 1918
	PQExpBuffer query = createPQExpBuffer();
	FuncInfo   *finfo;
1919 1920 1921 1922 1923 1924 1925 1926 1927 1928

	int			i_oid;
	int			i_proname;
	int			i_prolang;
	int			i_pronargs;
	int			i_proargtypes;
	int			i_prorettype;
	int			i_proretset;
	int			i_prosrc;
	int			i_probin;
1929
	int			i_iscachable;
1930
	int			i_isstrict;
1931
	int			i_usename;
1932 1933 1934

	/* find all user-defined funcs */

1935 1936 1937 1938 1939 1940
	if (g_fout->remoteVersion < 70100)
	{
		appendPQExpBuffer(query,
		   "SELECT pg_proc.oid, proname, prolang, pronargs, prorettype, "
					  "proretset, proargtypes, prosrc, probin, "
				"(select usename from pg_user where proowner = usesysid) as usename, "
1941
				"proiscachable, 'f'::boolean as proisstrict "
1942 1943 1944 1945 1946
				"from pg_proc "
				"where pg_proc.oid > '%u'::oid",
					  g_last_builtin_oid);
	} else {
		appendPQExpBuffer(query,
1947
		   "SELECT pg_proc.oid, proname, prolang, pronargs, prorettype, "
1948
					  "proretset, proargtypes, prosrc, probin, "
1949 1950 1951 1952
				"(select usename from pg_user where proowner = usesysid) as usename, "
				"proiscachable, proisstrict "
				"from pg_proc "
				"where pg_proc.oid > '%u'::oid",
1953
					  g_last_builtin_oid);
1954
	}
1955

B
Hi, all  
Bruce Momjian 已提交
1956
	res = PQexec(g_conn, query->data);
1957 1958 1959
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1960 1961
		write_msg(NULL, "query to obtain list of functions failed: %s",
				  PQerrorMessage(g_conn));
1962
		exit_nicely();
1963 1964 1965 1966 1967 1968 1969 1970
	}

	ntups = PQntuples(res);

	*numFuncs = ntups;

	finfo = (FuncInfo *) malloc(ntups * sizeof(FuncInfo));

1971 1972
	memset((char *) finfo, 0, ntups * sizeof(FuncInfo));

1973 1974 1975 1976 1977 1978 1979 1980 1981
	i_oid = PQfnumber(res, "oid");
	i_proname = PQfnumber(res, "proname");
	i_prolang = PQfnumber(res, "prolang");
	i_pronargs = PQfnumber(res, "pronargs");
	i_proargtypes = PQfnumber(res, "proargtypes");
	i_prorettype = PQfnumber(res, "prorettype");
	i_proretset = PQfnumber(res, "proretset");
	i_prosrc = PQfnumber(res, "prosrc");
	i_probin = PQfnumber(res, "probin");
1982
	i_iscachable = PQfnumber(res, "proiscachable");
1983
	i_isstrict = PQfnumber(res, "proisstrict");
1984 1985 1986 1987 1988 1989 1990
	i_usename = PQfnumber(res, "usename");

	for (i = 0; i < ntups; i++)
	{
		finfo[i].oid = strdup(PQgetvalue(res, i, i_oid));
		finfo[i].proname = strdup(PQgetvalue(res, i, i_proname));

1991
		finfo[i].prosrc = strdup(PQgetvalue(res, i, i_prosrc));
1992 1993 1994 1995 1996
		finfo[i].probin = strdup(PQgetvalue(res, i, i_probin));

		finfo[i].prorettype = strdup(PQgetvalue(res, i, i_prorettype));
		finfo[i].retset = (strcmp(PQgetvalue(res, i, i_proretset), "t") == 0);
		finfo[i].nargs = atoi(PQgetvalue(res, i, i_pronargs));
1997
		finfo[i].lang = atooid(PQgetvalue(res, i, i_prolang));
1998
		finfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
B
Bruce Momjian 已提交
1999 2000
		finfo[i].iscachable = (strcmp(PQgetvalue(res, i, i_iscachable), "t") == 0);
		finfo[i].isstrict = (strcmp(PQgetvalue(res, i, i_isstrict), "t") == 0);
2001 2002

		if (strlen(finfo[i].usename) == 0)
2003 2004
			write_msg(NULL, "WARNING: owner of function \"%s\" appears to be invalid\n",
					  finfo[i].proname);
2005

2006 2007
		if (finfo[i].nargs < 0 || finfo[i].nargs > FUNC_MAX_ARGS)
		{
2008 2009
			write_msg(NULL, "failed sanity check: function %s has more than %d (namely %d) arguments\n",
					  finfo[i].proname, FUNC_MAX_ARGS, finfo[i].nargs);
2010
			exit_nicely();
2011 2012 2013 2014
		}
		parseNumericArray(PQgetvalue(res, i, i_proargtypes),
						  finfo[i].argtypes,
						  finfo[i].nargs);
2015 2016 2017 2018 2019
		finfo[i].dumped = 0;
	}

	PQclear(res);

2020
	destroyPQExpBuffer(query);
2021

2022
	return finfo;
2023 2024 2025 2026
}

/*
 * getTables
2027
 *	  read all the user-defined tables (no indexes, no catalogs)
2028 2029
 * in the system catalogs return them in the TableInfo* structure
 *
2030
 * numTables is set to the number of tables read in
2031
 */
2032
TableInfo  *
V
Vadim B. Mikheev 已提交
2033
getTables(int *numTables, FuncInfo *finfo, int numFuncs)
2034
{
2035
	PGresult   *res;
2036 2037
	int			ntups;
	int			i;
2038
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
2039
	PQExpBuffer delqry = createPQExpBuffer();
2040
	TableInfo  *tblinfo;
2041

B
Bruce Momjian 已提交
2042
	int			i_reloid;
2043 2044 2045 2046
	int			i_relname;
	int			i_relkind;
	int			i_relacl;
	int			i_usename;
V
Vadim B. Mikheev 已提交
2047 2048
	int			i_relchecks;
	int			i_reltriggers;
2049
	int			i_relhasindex;
2050
	int			i_relhasoids;
2051 2052 2053 2054 2055

	char		relkindview[2];

	relkindview[0] = RELKIND_VIEW;
	relkindview[1] = '\0';
2056 2057

	/*
2058
	 * find all the user-defined tables (no indexes and no catalogs),
2059 2060 2061
	 * ordering by oid is important so that we always process the parent
	 * tables before the child tables when traversing the tblinfo*
	 *
2062
	 * we ignore tables that are not type 'r' (ordinary relation) or 'S'
2063
	 * (sequence) or 'v' (view).
2064 2065
	 */

2066
	if (g_fout->remoteVersion >= 70200)
2067 2068
	{
		appendPQExpBuffer(query,
2069 2070 2071 2072 2073 2074 2075 2076
						  "SELECT pg_class.oid, relname, relacl, relkind, "
						  "(select usename from pg_user where relowner = usesysid) as usename, "
						  "relchecks, reltriggers, relhasindex, relhasoids "
						  "from pg_class "
						  "where relname !~ '^pg_' "
						  "and relkind in ('%c', '%c', '%c') "
						  "order by oid",
						  RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW);
2077 2078 2079 2080 2081
	}
	else if (g_fout->remoteVersion >= 70100)
	{
		/* all tables have oids in 7.1 */
		appendPQExpBuffer(query,
2082 2083 2084 2085 2086 2087 2088 2089
						  "SELECT pg_class.oid, relname, relacl, relkind, "
						  "(select usename from pg_user where relowner = usesysid) as usename, "
						  "relchecks, reltriggers, relhasindex, 't'::bool as relhasoids "
						  "from pg_class "
						  "where relname !~ '^pg_' "
						  "and relkind in ('%c', '%c', '%c') "
						  "order by oid",
						  RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW);
2090
	} else {
2091
		/*
2092 2093
		 * Before 7.1, view relkind was not set to 'v', so we must check
		 * if we have a view by looking for a rule in pg_rewrite.
2094 2095
		 */
		appendPQExpBuffer(query,
2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109
						  "SELECT c.oid, relname, relacl, "
						  "CASE WHEN relhasrules and relkind = 'r' "
						  "  and EXISTS(SELECT rulename FROM pg_rewrite r WHERE "
						  "             r.ev_class = c.oid AND r.ev_type = '1') "
						  "THEN '%c'::\"char\" "
						  "ELSE relkind END AS relkind,"
						  "(select usename from pg_user where relowner = usesysid) as usename, "
						  "relchecks, reltriggers, relhasindex, 't'::bool as relhasoids "
						  "from pg_class c "
						  "where relname !~ '^pg_' "
						  "and relkind in ('%c', '%c', '%c') "
						  "order by oid",
						  RELKIND_VIEW,
						  RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW);
2110
	}
2111

B
Hi, all  
Bruce Momjian 已提交
2112
	res = PQexec(g_conn, query->data);
2113 2114 2115
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
2116 2117
		write_msg(NULL, "query to obtain list of tables failed: %s",
				  PQerrorMessage(g_conn));
2118
		exit_nicely();
2119 2120 2121 2122 2123 2124 2125 2126
	}

	ntups = PQntuples(res);

	*numTables = ntups;

	tblinfo = (TableInfo *) malloc(ntups * sizeof(TableInfo));

B
Bruce Momjian 已提交
2127
	i_reloid = PQfnumber(res, "oid");
2128 2129
	i_relname = PQfnumber(res, "relname");
	i_relacl = PQfnumber(res, "relacl");
2130
	i_relkind = PQfnumber(res, "relkind");
2131
	i_usename = PQfnumber(res, "usename");
V
Vadim B. Mikheev 已提交
2132 2133
	i_relchecks = PQfnumber(res, "relchecks");
	i_reltriggers = PQfnumber(res, "reltriggers");
B
Hi, all  
Bruce Momjian 已提交
2134
	i_relhasindex = PQfnumber(res, "relhasindex");
2135
	i_relhasoids = PQfnumber(res, "relhasoids");
2136 2137 2138

	for (i = 0; i < ntups; i++)
	{
B
Bruce Momjian 已提交
2139
		tblinfo[i].oid = strdup(PQgetvalue(res, i, i_reloid));
2140 2141 2142 2143
		tblinfo[i].relname = strdup(PQgetvalue(res, i, i_relname));
		tblinfo[i].relacl = strdup(PQgetvalue(res, i, i_relacl));
		tblinfo[i].sequence = (strcmp(PQgetvalue(res, i, i_relkind), "S") == 0);
		tblinfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
V
Vadim B. Mikheev 已提交
2144 2145
		tblinfo[i].ncheck = atoi(PQgetvalue(res, i, i_relchecks));
		tblinfo[i].ntrig = atoi(PQgetvalue(res, i, i_reltriggers));
2146

2147
		if (strlen(tblinfo[i].usename) == 0)
2148 2149
			write_msg(NULL, "WARNING: owner of table \"%s\" appears to be invalid\n",
					  tblinfo[i].relname);
2150

2151 2152 2153 2154 2155 2156
		/* Get view definition */
		if (strcmp(PQgetvalue(res, i, i_relkind), relkindview) == 0)
		{
			PGresult   *res2;

			resetPQExpBuffer(query);
2157 2158
			appendPQExpBuffer(query, "SELECT definition as viewdef, ");
			/* XXX 7.2 - replace with att from pg_views or some other generic source */
2159 2160 2161
			appendPQExpBuffer(query, "(select oid from pg_rewrite where "
										" rulename=('_RET' || viewname)::name) as view_oid"
										" from pg_views where viewname = ");
2162
			formatStringLiteral(query, tblinfo[i].relname, CONV_ALL);
2163 2164
			appendPQExpBuffer(query, ";");

2165 2166 2167
			res2 = PQexec(g_conn, query->data);
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
2168 2169
				write_msg(NULL, "query to obtain definition of view \"%s\" failed: %s",
						  tblinfo[i].relname, PQerrorMessage(g_conn));
2170
				exit_nicely();
2171 2172
			}

B
Bruce Momjian 已提交
2173
			if (PQntuples(res2) != 1)
2174 2175
			{
				if (PQntuples(res2) < 1)
2176 2177
					write_msg(NULL, "query to obtain definition of view \"%s\" returned no data\n",
							  tblinfo[i].relname);
B
Bruce Momjian 已提交
2178
				else
2179
					write_msg(NULL, "query to obtain definition of view \"%s\" returned more than one definition\n",
B
Bruce Momjian 已提交
2180
							tblinfo[i].relname);
2181
				exit_nicely();
2182 2183
			}

2184 2185
			if (PQgetisnull(res2, 0, 1))
			{
2186 2187
				write_msg(NULL, "query to obtain definition of view \"%s\" returned NULL oid\n",
						  tblinfo[i].relname);
2188
				exit_nicely();
2189 2190
			}

2191
			tblinfo[i].viewdef = strdup(PQgetvalue(res2, 0, 0));
2192
			tblinfo[i].viewoid = strdup(PQgetvalue(res2, 0, 1));
2193

B
Bruce Momjian 已提交
2194
			if (strlen(tblinfo[i].viewdef) == 0)
2195
			{
2196 2197
				write_msg(NULL, "definition of view \"%s\" appears to be empty (length zero)\n",
						  tblinfo[i].relname);
2198
				exit_nicely();
2199 2200 2201 2202 2203
			}
		}
		else
			tblinfo[i].viewdef = NULL;

2204
		/*
2205 2206
		 * Get non-inherited CHECK constraints, if any.
		 *
B
Bruce Momjian 已提交
2207 2208
		 * Exclude inherited CHECKs from CHECK constraints total. If a
		 * constraint matches by name and condition with a constraint
2209
		 * belonging to a parent class (OR conditions match and both
2210
		 * names start with '$', we assume it was inherited.
2211 2212
		 */
		if (tblinfo[i].ncheck > 0)
V
Vadim B. Mikheev 已提交
2213 2214
		{
			PGresult   *res2;
2215 2216
			int			i_rcname,
						i_rcsrc;
V
Vadim B. Mikheev 已提交
2217 2218
			int			ntups2;
			int			i2;
2219

V
Vadim B. Mikheev 已提交
2220
			if (g_verbose)
2221 2222
				write_msg(NULL, "finding CHECK constraints for table %s\n",
						  tblinfo[i].relname);
2223

B
Hi, all  
Bruce Momjian 已提交
2224 2225
			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "SELECT rcname, rcsrc from pg_relcheck "
2226
							  " where rcrelid = '%s'::oid "
2227
							  "   and not exists "
2228 2229 2230 2231 2232 2233 2234 2235
							  "  (select * from pg_relcheck as c, pg_inherits as i "
							  "    where i.inhrelid = pg_relcheck.rcrelid "
							  "      and (c.rcname = pg_relcheck.rcname "
							  "          or (    c.rcname[0] = '$' "
							  "              and pg_relcheck.rcname[0] = '$')"
							  "          )"
							  "      and c.rcsrc = pg_relcheck.rcsrc "
							  "      and c.rcrelid = i.inhparent) "
2236
							  " order by rcname ",
2237
							  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
2238
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
2239
			if (!res2 ||
2240
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2241
			{
2242
				write_msg(NULL, "query to obtain check constraints failed: %s", PQerrorMessage(g_conn));
2243
				exit_nicely();
V
Vadim B. Mikheev 已提交
2244 2245
			}
			ntups2 = PQntuples(res2);
2246
			if (ntups2 > tblinfo[i].ncheck)
V
Vadim B. Mikheev 已提交
2247
			{
2248 2249 2250
				write_msg(NULL, "expected %d check constraints on table \"%s\" but found %d\n",
						  tblinfo[i].ncheck, tblinfo[i].relname, ntups2);
				write_msg(NULL, "(The system catalogs might be corrupted.)\n");
2251
				exit_nicely();
V
Vadim B. Mikheev 已提交
2252
			}
2253 2254 2255 2256

			/* Set ncheck to the number of *non-inherited* CHECK constraints */
			tblinfo[i].ncheck = ntups2;

V
Vadim B. Mikheev 已提交
2257 2258
			i_rcname = PQfnumber(res2, "rcname");
			i_rcsrc = PQfnumber(res2, "rcsrc");
2259
			tblinfo[i].check_expr = (char **) malloc(ntups2 * sizeof(char *));
V
Vadim B. Mikheev 已提交
2260 2261
			for (i2 = 0; i2 < ntups2; i2++)
			{
B
Bruce Momjian 已提交
2262 2263
				const char *name = PQgetvalue(res2, i2, i_rcname);
				const char *expr = PQgetvalue(res2, i2, i_rcsrc);
2264

B
Hi, all  
Bruce Momjian 已提交
2265
				resetPQExpBuffer(query);
2266 2267
				if (name[0] != '$')
				{
2268 2269
					appendPQExpBuffer(query, "CONSTRAINT %s ",
									  fmtId(name, force_quotes));
B
Bruce Momjian 已提交
2270
				}
2271
				appendPQExpBuffer(query, "CHECK (%s)", expr);
B
Hi, all  
Bruce Momjian 已提交
2272
				tblinfo[i].check_expr[i2] = strdup(query->data);
V
Vadim B. Mikheev 已提交
2273 2274 2275 2276 2277
			}
			PQclear(res2);
		}
		else
			tblinfo[i].check_expr = NULL;
2278

2279 2280
		/* Get primary key */
		if (strcmp(PQgetvalue(res, i, i_relhasindex), "t") == 0)
2281
		{
2282
			PGresult   *res2;
B
Bruce Momjian 已提交
2283

B
Hi, all  
Bruce Momjian 已提交
2284
			resetPQExpBuffer(query);
B
Bruce Momjian 已提交
2285
			appendPQExpBuffer(query,
2286
							  "SELECT indexrelid FROM pg_index i WHERE i.indisprimary AND i.indrelid = '%s'::oid ",
2287
							  tblinfo[i].oid);
2288
			res2 = PQexec(g_conn, query->data);
B
Bruce Momjian 已提交
2289 2290
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
2291 2292
				write_msg(NULL, "query to obtain primary key of table \"%s\" failed: %s",
						  tblinfo[i].relname, PQerrorMessage(g_conn));
2293
				exit_nicely();
B
Bruce Momjian 已提交
2294
			}
2295

B
Bruce Momjian 已提交
2296 2297
			if (PQntuples(res2) > 1)
			{
2298
				write_msg(NULL, "query to obtain primary key of table \"%s\" produced more than one result\n",
2299
						tblinfo[i].relname);
2300
				exit_nicely();
2301 2302
			}

B
Bruce Momjian 已提交
2303
			if (PQntuples(res2) == 1)
2304
				tblinfo[i].pkIndexOid = strdup(PQgetvalue(res2, 0, 0));
B
Bruce Momjian 已提交
2305
			else
2306 2307
				tblinfo[i].pkIndexOid = NULL;

2308
		}
2309
		else
2310
			tblinfo[i].pkIndexOid = NULL;
B
Bruce Momjian 已提交
2311

2312 2313 2314
		/* Has it got OIDs? */
		tblinfo[i].hasoids = (strcmp(PQgetvalue(res, i, i_relhasoids), "t") == 0);

2315
		/* Get primary key name (if primary key exist) */
2316
		if (tblinfo[i].pkIndexOid != NULL)
2317 2318
		{
			PGresult   *res2;
B
Bruce Momjian 已提交
2319
			int			n;
2320 2321

			resetPQExpBuffer(query);
2322 2323
			appendPQExpBuffer(query,
							  "SELECT relname FROM pg_class "
2324
							  "WHERE oid = '%s'::oid",
2325
							  tblinfo[i].pkIndexOid);
2326

2327 2328 2329
			res2 = PQexec(g_conn, query->data);
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
2330 2331
				write_msg(NULL, "query to obtain name of primary key of table \"%s\" failed: %s",
						  tblinfo[i].relname, PQerrorMessage(g_conn));
2332
				exit_nicely();
2333 2334 2335 2336 2337
			}

			n = PQntuples(res2);
			if (n != 1)
			{
2338 2339
				write_msg(NULL, "query to obtain name of primary key of table \"%s\" did not return exactly one result\n",
						  tblinfo[i].relname);
2340
				exit_nicely();
2341 2342 2343 2344 2345 2346
			}

			tblinfo[i].primary_key_name =
				strdup(fmtId(PQgetvalue(res2, 0, 0), force_quotes));
			if (tblinfo[i].primary_key_name == NULL)
			{
2347
				write_msg(NULL, "out of memory\n");
2348
				exit_nicely();
2349 2350 2351 2352 2353
			}
		}
		else
			tblinfo[i].primary_key_name = NULL;

V
Vadim B. Mikheev 已提交
2354 2355 2356 2357
		/* Get Triggers */
		if (tblinfo[i].ntrig > 0)
		{
			PGresult   *res2;
B
Bruce,  
Bruce Momjian 已提交
2358
			int			i_tgoid,
2359
						i_tgname,
2360 2361 2362
						i_tgfoid,
						i_tgtype,
						i_tgnargs,
2363 2364 2365 2366
						i_tgargs,
						i_tgisconstraint,
						i_tgconstrname,
						i_tgdeferrable,
2367 2368
						i_tgconstrrelid,
						i_tgconstrrelname,
2369
						i_tginitdeferred;
V
Vadim B. Mikheev 已提交
2370 2371
			int			ntups2;
			int			i2;
2372

V
Vadim B. Mikheev 已提交
2373
			if (g_verbose)
2374
				write_msg(NULL, "finding triggers for table %s\n", tblinfo[i].relname);
2375

B
Hi, all  
Bruce Momjian 已提交
2376
			resetPQExpBuffer(query);
2377
			appendPQExpBuffer(query,
2378 2379 2380 2381 2382
								"SELECT tgname, tgfoid, tgtype, tgnargs, tgargs, "
								"tgisconstraint, tgconstrname, tgdeferrable, "
								"tgconstrrelid, tginitdeferred, oid, "
								"(select relname from pg_class where oid = tgconstrrelid) "
								"		as tgconstrrelname "
2383 2384 2385
							  "from pg_trigger "
							  "where tgrelid = '%s'::oid ",
							  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
2386
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
2387
			if (!res2 ||
2388
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2389
			{
2390
				write_msg(NULL, "query to obtain list of triggers failed: %s", PQerrorMessage(g_conn));
2391
				exit_nicely();
V
Vadim B. Mikheev 已提交
2392 2393 2394 2395
			}
			ntups2 = PQntuples(res2);
			if (ntups2 != tblinfo[i].ntrig)
			{
2396 2397
				write_msg(NULL, "expected %d triggers on table \"%s\" but found %d\n",
						  tblinfo[i].ntrig, tblinfo[i].relname, ntups2);
2398
				exit_nicely();
V
Vadim B. Mikheev 已提交
2399 2400 2401 2402 2403 2404
			}
			i_tgname = PQfnumber(res2, "tgname");
			i_tgfoid = PQfnumber(res2, "tgfoid");
			i_tgtype = PQfnumber(res2, "tgtype");
			i_tgnargs = PQfnumber(res2, "tgnargs");
			i_tgargs = PQfnumber(res2, "tgargs");
B
Bruce,  
Bruce Momjian 已提交
2405
			i_tgoid = PQfnumber(res2, "oid");
2406 2407 2408
			i_tgisconstraint = PQfnumber(res2, "tgisconstraint");
			i_tgconstrname = PQfnumber(res2, "tgconstrname");
			i_tgdeferrable = PQfnumber(res2, "tgdeferrable");
2409 2410
			i_tgconstrrelid = PQfnumber(res2, "tgconstrrelid");
			i_tgconstrrelname = PQfnumber(res2, "tgconstrrelname");
2411 2412
			i_tginitdeferred = PQfnumber(res2, "tginitdeferred");

B
Bruce Momjian 已提交
2413
			tblinfo[i].triggers = (TrigInfo *) malloc(ntups2 * sizeof(TrigInfo));
B
Hi, all  
Bruce Momjian 已提交
2414 2415
			resetPQExpBuffer(query);
			for (i2 = 0; i2 < ntups2; i2++)
V
Vadim B. Mikheev 已提交
2416
			{
2417
				const char *tgfuncoid = PQgetvalue(res2, i2, i_tgfoid);
2418
				char	   *tgfunc = NULL;
2419 2420
				int2		tgtype = atoi(PQgetvalue(res2, i2, i_tgtype));
				int			tgnargs = atoi(PQgetvalue(res2, i2, i_tgnargs));
B
Bruce Momjian 已提交
2421
				const char *tgargs = PQgetvalue(res2, i2, i_tgargs);
2422 2423 2424
				int			tgisconstraint;
				int			tgdeferrable;
				int			tginitdeferred;
2425 2426
				char	   *tgconstrrelid;
				char	   *tgname;
B
Bruce Momjian 已提交
2427
				const char *p;
2428 2429
				int			findx;

2430 2431
				tgname = PQgetvalue(res2, i2, i_tgname);

2432
				if (strcmp(PQgetvalue(res2, i2, i_tgisconstraint), "f") == 0)
2433
					tgisconstraint = 0;
2434
				else
2435
					tgisconstraint = 1;
2436 2437

				if (strcmp(PQgetvalue(res2, i2, i_tgdeferrable), "f") == 0)
2438
					tgdeferrable = 0;
2439
				else
2440
					tgdeferrable = 1;
2441 2442

				if (strcmp(PQgetvalue(res2, i2, i_tginitdeferred), "f") == 0)
2443
					tginitdeferred = 0;
2444
				else
2445
					tginitdeferred = 1;
2446

V
Vadim B. Mikheev 已提交
2447 2448
				for (findx = 0; findx < numFuncs; findx++)
				{
2449
					if (strcmp(finfo[findx].oid, tgfuncoid) == 0 &&
2450
						finfo[findx].nargs == 0 &&
V
Vadim B. Mikheev 已提交
2451 2452 2453
						strcmp(finfo[findx].prorettype, "0") == 0)
						break;
				}
2454

V
Vadim B. Mikheev 已提交
2455 2456
				if (findx == numFuncs)
				{
2457
					PGresult   *r;
2458
					int			numFuncs;
2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475

					/*
					 * the funcname is an oid which we use to find the
					 * name of the pg_proc.  We need to do this because
					 * getFuncs() only reads in the user-defined funcs not
					 * all the funcs.  We might not find what we want by
					 * looking in FuncInfo*
					 */
					resetPQExpBuffer(query);
					appendPQExpBuffer(query,
									  "SELECT proname from pg_proc "
									  "where pg_proc.oid = '%s'::oid",
									  tgfuncoid);

					r = PQexec(g_conn, query->data);
					if (!r || PQresultStatus(r) != PGRES_TUPLES_OK)
					{
2476 2477
						write_msg(NULL, "query to obtain procedure name for trigger \"%s\" failed: %s",
								  tgname, PQerrorMessage(g_conn));
2478
						exit_nicely();
2479 2480 2481 2482
					}

					/* Sanity: Check we got only one tuple */
					numFuncs = PQntuples(r);
B
Bruce Momjian 已提交
2483 2484
					if (numFuncs != 1)
					{
2485 2486
						write_msg(NULL, "query to obtain procedure name for trigger \"%s\" did not return exactly one result\n",
								  tgname);
2487
						exit_nicely();
2488
					}
2489

2490
					tgfunc = strdup(PQgetvalue(r, 0, PQfnumber(r, "proname")));
2491
					PQclear(r);
2492
				}
2493
				else
2494
					tgfunc = strdup(finfo[findx].proname);
B
Bruce Momjian 已提交
2495

2496
				appendPQExpBuffer(delqry, "DROP TRIGGER %s ", fmtId(tgname, force_quotes));
B
Bruce Momjian 已提交
2497
				appendPQExpBuffer(delqry, "ON %s;\n",
B
Bruce Momjian 已提交
2498
								fmtId(tblinfo[i].relname, force_quotes));
2499

B
Hi, all  
Bruce Momjian 已提交
2500
				resetPQExpBuffer(query);
2501 2502 2503 2504 2505
				if (tgisconstraint)
				{
					appendPQExpBuffer(query, "CREATE CONSTRAINT TRIGGER ");
					appendPQExpBuffer(query, fmtId(PQgetvalue(res2, i2, i_tgconstrname), force_quotes));
				}
2506 2507
				else
				{
2508
					appendPQExpBuffer(query, "CREATE TRIGGER ");
2509
					appendPQExpBuffer(query, fmtId(tgname, force_quotes));
2510
				}
B
Bruce Momjian 已提交
2511
				appendPQExpBufferChar(query, ' ');
V
Vadim B. Mikheev 已提交
2512 2513 2514
				/* Trigger type */
				findx = 0;
				if (TRIGGER_FOR_BEFORE(tgtype))
B
Hi, all  
Bruce Momjian 已提交
2515
					appendPQExpBuffer(query, "BEFORE");
V
Vadim B. Mikheev 已提交
2516
				else
B
Hi, all  
Bruce Momjian 已提交
2517
					appendPQExpBuffer(query, "AFTER");
V
Vadim B. Mikheev 已提交
2518 2519
				if (TRIGGER_FOR_INSERT(tgtype))
				{
B
Hi, all  
Bruce Momjian 已提交
2520
					appendPQExpBuffer(query, " INSERT");
V
Vadim B. Mikheev 已提交
2521 2522 2523 2524 2525
					findx++;
				}
				if (TRIGGER_FOR_DELETE(tgtype))
				{
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
2526
						appendPQExpBuffer(query, " OR DELETE");
V
Vadim B. Mikheev 已提交
2527
					else
B
Hi, all  
Bruce Momjian 已提交
2528
						appendPQExpBuffer(query, " DELETE");
V
Vadim B. Mikheev 已提交
2529 2530 2531
					findx++;
				}
				if (TRIGGER_FOR_UPDATE(tgtype))
2532
				{
V
Vadim B. Mikheev 已提交
2533
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
2534
						appendPQExpBuffer(query, " OR UPDATE");
V
Vadim B. Mikheev 已提交
2535
					else
B
Hi, all  
Bruce Momjian 已提交
2536
						appendPQExpBuffer(query, " UPDATE");
2537
				}
2538 2539
				appendPQExpBuffer(query, " ON %s ", fmtId(tblinfo[i].relname, force_quotes));

2540
				if (tgisconstraint)
2541
				{
2542 2543 2544 2545 2546
					tgconstrrelid = PQgetvalue(res2, i2, i_tgconstrrelid);

					if (strcmp(tgconstrrelid, "0") != 0) {

						if (PQgetisnull(res2, i2, i_tgconstrrelname))
2547
						{
2548
							write_msg(NULL, "query produced NULL referenced table name for foreign key trigger \"%s\" on table \"%s\" (oid of table: %s)\n",
2549
									  tgname, tblinfo[i].relname, tgconstrrelid);
2550
							exit_nicely();
2551
						}
2552

2553
						appendPQExpBuffer(query, " FROM %s",
2554 2555
										fmtId(PQgetvalue(res2, i2, i_tgconstrrelname), force_quotes));
					}
2556
					if (!tgdeferrable)
2557 2558 2559 2560
						appendPQExpBuffer(query, " NOT");
					appendPQExpBuffer(query, " DEFERRABLE INITIALLY ");
					if (tginitdeferred)
						appendPQExpBuffer(query, "DEFERRED");
2561
					else
2562
						appendPQExpBuffer(query, "IMMEDIATE");
2563

2564 2565 2566
				}

				appendPQExpBuffer(query, " FOR EACH ROW");
2567 2568
				appendPQExpBuffer(query, " EXECUTE PROCEDURE %s (",
								  fmtId(tgfunc, force_quotes));
V
Vadim B. Mikheev 已提交
2569 2570
				for (findx = 0; findx < tgnargs; findx++)
				{
2571
					const char *s;
2572 2573

					for (p = tgargs;;)
V
Vadim B. Mikheev 已提交
2574
					{
2575
						p = strchr(p, '\\');
V
Vadim B. Mikheev 已提交
2576 2577
						if (p == NULL)
						{
2578 2579 2580 2581
							write_msg(NULL, "bad argument string (%s) for trigger \"%s\" on table \"%s\"\n",
									  PQgetvalue(res2, i2, i_tgargs),
									  tgname,
									  tblinfo[i].relname);
2582
							exit_nicely();
V
Vadim B. Mikheev 已提交
2583 2584 2585 2586 2587 2588 2589
						}
						p++;
						if (*p == '\\')
						{
							p++;
							continue;
						}
2590
						if (p[0] == '0' && p[1] == '0' && p[2] == '0')
V
Vadim B. Mikheev 已提交
2591 2592 2593
							break;
					}
					p--;
2594
					appendPQExpBufferChar(query, '\'');
B
Bruce Momjian 已提交
2595
					for (s = tgargs; s < p;)
V
Vadim B. Mikheev 已提交
2596 2597
					{
						if (*s == '\'')
2598 2599
							appendPQExpBufferChar(query, '\\');
						appendPQExpBufferChar(query, *s++);
V
Vadim B. Mikheev 已提交
2600
					}
B
Bruce Momjian 已提交
2601 2602
					appendPQExpBufferChar(query, '\'');
					appendPQExpBuffer(query, (findx < tgnargs - 1) ? ", " : "");
V
Vadim B. Mikheev 已提交
2603 2604
					tgargs = p + 4;
				}
B
Hi, all  
Bruce Momjian 已提交
2605
				appendPQExpBuffer(query, ");\n");
2606

B
Bruce Momjian 已提交
2607
				tblinfo[i].triggers[i2].tgsrc = strdup(query->data);
B
Bruce,  
Bruce Momjian 已提交
2608 2609

				/*** Initialize trcomments and troids ***/
2610

B
Bruce,  
Bruce Momjian 已提交
2611
				resetPQExpBuffer(query);
2612
				appendPQExpBuffer(query, "TRIGGER %s ",
2613
					fmtId(tgname, force_quotes));
B
Bruce,  
Bruce Momjian 已提交
2614
				appendPQExpBuffer(query, "ON %s",
2615
								fmtId(tblinfo[i].relname, force_quotes));
B
Bruce Momjian 已提交
2616 2617
				tblinfo[i].triggers[i2].tgcomment = strdup(query->data);
				tblinfo[i].triggers[i2].oid = strdup(PQgetvalue(res2, i2, i_tgoid));
2618
				tblinfo[i].triggers[i2].tgname = strdup(fmtId(tgname, false));
B
Bruce Momjian 已提交
2619
				tblinfo[i].triggers[i2].tgdel = strdup(delqry->data);
B
Bruce,  
Bruce Momjian 已提交
2620

2621 2622
				if (tgfunc)
					free(tgfunc);
V
Vadim B. Mikheev 已提交
2623 2624 2625 2626 2627
			}
			PQclear(res2);
		}
		else
			tblinfo[i].triggers = NULL;
2628

2629 2630 2631 2632
	}

	PQclear(res);

2633 2634
	destroyPQExpBuffer(query);
	destroyPQExpBuffer(delqry);
2635

2636
	return tblinfo;
2637 2638 2639 2640
}

/*
 * getInherits
2641
 *	  read all the inheritance information
2642 2643
 * from the system catalogs return them in the InhInfo* structure
 *
2644
 * numInherits is set to the number of tables read in
2645
 */
2646
InhInfo    *
2647 2648
getInherits(int *numInherits)
{
2649
	PGresult   *res;
2650 2651
	int			ntups;
	int			i;
2652 2653
	PQExpBuffer query = createPQExpBuffer();
	InhInfo    *inhinfo;
2654

2655
	int			i_inhrelid;
2656
	int			i_inhparent;
2657 2658 2659

	/* find all the inheritance information */

B
Hi, all  
Bruce Momjian 已提交
2660
	appendPQExpBuffer(query, "SELECT inhrelid, inhparent from pg_inherits");
2661

B
Hi, all  
Bruce Momjian 已提交
2662
	res = PQexec(g_conn, query->data);
2663 2664 2665
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
2666 2667
		write_msg(NULL, "query to obtain inheritance relationships failed: %s",
				  PQerrorMessage(g_conn));
2668
		exit_nicely();
2669 2670 2671 2672 2673 2674 2675 2676
	}

	ntups = PQntuples(res);

	*numInherits = ntups;

	inhinfo = (InhInfo *) malloc(ntups * sizeof(InhInfo));

2677
	i_inhrelid = PQfnumber(res, "inhrelid");
2678 2679 2680 2681
	i_inhparent = PQfnumber(res, "inhparent");

	for (i = 0; i < ntups; i++)
	{
2682
		inhinfo[i].inhrelid = strdup(PQgetvalue(res, i, i_inhrelid));
2683 2684 2685 2686
		inhinfo[i].inhparent = strdup(PQgetvalue(res, i, i_inhparent));
	}

	PQclear(res);
2687 2688 2689

	destroyPQExpBuffer(query);

2690
	return inhinfo;
2691 2692 2693 2694
}

/*
 * getTableAttrs -
2695 2696
 *	  for each table in tblinfo, read its attributes types and names
 *
2697
 * this is implemented in a very inefficient way right now, looping
2698
 * through the tblinfo and doing a join per table to find the attrs and their
2699 2700
 * types
 *
2701
 *	modifies tblinfo
2702 2703
 */
void
2704
getTableAttrs(TableInfo *tblinfo, int numTables)
2705
{
2706 2707
	int			i,
				j;
2708
	PQExpBuffer q = createPQExpBuffer();
2709 2710
	int			i_attname;
	int			i_typname;
2711
	int			i_atttypmod;
2712
	int			i_attnotnull;
V
Vadim B. Mikheev 已提交
2713
	int			i_atthasdef;
2714
	int			i_atttypedefn;
2715
	PGresult   *res;
2716
	int			ntups;
2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730

	for (i = 0; i < numTables; i++)
	{
		if (tblinfo[i].sequence)
			continue;

		/* find all the user attributes and their types */
		/* we must read the attribute names in attribute number order! */

		/*
		 * because we will use the attnum to index into the attnames array
		 * later
		 */
		if (g_verbose)
2731 2732
			write_msg(NULL, "finding the columns and types for table %s\n",
					  tblinfo[i].relname);
2733

B
Hi, all  
Bruce Momjian 已提交
2734
		resetPQExpBuffer(q);
2735 2736 2737 2738

		if (g_fout->remoteVersion < 70100)
		{
			/* Fake the LOJ below */
2739
			appendPQExpBuffer(q,
2740
						"  SELECT a.attnum, a.attname, t.typname, a.atttypmod, "
2741 2742 2743 2744 2745
						"        a.attnotnull, a.atthasdef, NULL as atttypedefn "
						"    from pg_attribute a, pg_type t "
						"    where a.attrelid = '%s'::oid "
						"        and a.attnum > 0 "
						"        and a.atttypid = t.oid "
2746
						" UNION ALL SELECT a.attnum, a.attname, NULL as typname, a.atttypmod, "
2747 2748 2749 2750 2751 2752 2753 2754 2755
						"        a.attnotnull, a.atthasdef, NULL as atttypedefn "
						"    from pg_attribute a "
						"    where a.attrelid = '%s'::oid "
						"        and a.attnum > 0 "
						"        and Not Exists(Select * From pg_type t where a.atttypid = t.oid)"
						"    order by attnum",
						tblinfo[i].oid, tblinfo[i].oid);

		} else {
2756
			appendPQExpBuffer(q, "SELECT a.attnum, a.attname, t.typname, a.atttypmod, "
2757 2758 2759 2760 2761 2762 2763
								"a.attnotnull, a.atthasdef, format_type(a.atttypid, a.atttypmod) as atttypedefn "
								"from pg_attribute a LEFT OUTER JOIN pg_type t ON a.atttypid = t.oid "
								"where a.attrelid = '%s'::oid "
								"and a.attnum > 0 order by attnum",
								tblinfo[i].oid);
		}

B
Hi, all  
Bruce Momjian 已提交
2764
		res = PQexec(g_conn, q->data);
2765 2766 2767
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
2768
			write_msg(NULL, "query to get table columns failed: %s", PQerrorMessage(g_conn));
2769
			exit_nicely();
2770 2771 2772 2773 2774 2775
		}

		ntups = PQntuples(res);

		i_attname = PQfnumber(res, "attname");
		i_typname = PQfnumber(res, "typname");
2776
		i_atttypmod = PQfnumber(res, "atttypmod");
2777
		i_attnotnull = PQfnumber(res, "attnotnull");
V
Vadim B. Mikheev 已提交
2778
		i_atthasdef = PQfnumber(res, "atthasdef");
2779
		i_atttypedefn = PQfnumber(res, "atttypedefn");
2780 2781 2782

		tblinfo[i].numatts = ntups;
		tblinfo[i].attnames = (char **) malloc(ntups * sizeof(char *));
2783
		tblinfo[i].atttypedefns = (char **) malloc(ntups * sizeof(char *));
2784
		tblinfo[i].typnames = (char **) malloc(ntups * sizeof(char *));
2785
		tblinfo[i].atttypmod = (int *) malloc(ntups * sizeof(int));
2786
		tblinfo[i].inhAttrs = (int *) malloc(ntups * sizeof(int));
2787 2788
		tblinfo[i].inhAttrDef = (int *) malloc(ntups * sizeof(int));
		tblinfo[i].inhNotNull = (int *) malloc(ntups * sizeof(int));
2789
		tblinfo[i].notnull = (bool *) malloc(ntups * sizeof(bool));
V
Vadim B. Mikheev 已提交
2790
		tblinfo[i].adef_expr = (char **) malloc(ntups * sizeof(char *));
2791 2792 2793 2794
		tblinfo[i].parentRels = NULL;
		tblinfo[i].numParents = 0;
		for (j = 0; j < ntups; j++)
		{
2795 2796 2797
			/* Sanity check on LOJ */
			if (PQgetisnull(res, j, i_typname))
			{
2798 2799
				write_msg(NULL, "query produced NULL name for data type of column %d of table %s\n",
						  j+1, tblinfo[i].relname);
2800
				exit_nicely();
2801 2802
			}

2803
			tblinfo[i].attnames[j] = strdup(PQgetvalue(res, j, i_attname));
2804
			tblinfo[i].atttypedefns[j] = strdup(PQgetvalue(res, j, i_atttypedefn));
2805
			tblinfo[i].typnames[j] = strdup(PQgetvalue(res, j, i_typname));
2806
			tblinfo[i].atttypmod[j] = atoi(PQgetvalue(res, j, i_atttypmod));
2807 2808
			tblinfo[i].inhAttrs[j] = 0; /* this flag is set in
										 * flagInhAttrs() */
2809 2810 2811
			tblinfo[i].inhAttrDef[j] = 0;
			tblinfo[i].inhNotNull[j] = 0;

2812
			tblinfo[i].notnull[j] = (PQgetvalue(res, j, i_attnotnull)[0] == 't') ? true : false;
V
Vadim B. Mikheev 已提交
2813 2814 2815
			if (PQgetvalue(res, j, i_atthasdef)[0] == 't')
			{
				PGresult   *res2;
2816
				int			numAttr;
2817

V
Vadim B. Mikheev 已提交
2818
				if (g_verbose)
2819 2820
					write_msg(NULL, "finding DEFAULT expression for column %s\n",
							  tblinfo[i].attnames[j]);
2821

B
Hi, all  
Bruce Momjian 已提交
2822 2823
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "SELECT adsrc from pg_attrdef "
2824 2825
							 "where adrelid = '%s'::oid and adnum = %d ",
								  tblinfo[i].oid, j + 1);
B
Hi, all  
Bruce Momjian 已提交
2826
				res2 = PQexec(g_conn, q->data);
V
Vadim B. Mikheev 已提交
2827
				if (!res2 ||
2828
					PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2829
				{
2830 2831
					write_msg(NULL, "query to get column default value failed: %s",
							  PQerrorMessage(g_conn));
2832
					exit_nicely();
V
Vadim B. Mikheev 已提交
2833
				}
2834 2835 2836

				/* Sanity: Check we got only one tuple */
				numAttr = PQntuples(res2);
B
Bruce Momjian 已提交
2837 2838
				if (numAttr != 1)
				{
2839 2840
					write_msg(NULL, "query to get default value for column \"%s\" returned %d rows; expected 1\n",
							  tblinfo[i].attnames[j], numAttr);
2841
					exit_nicely();
2842 2843
				}

V
Vadim B. Mikheev 已提交
2844 2845 2846 2847 2848
				tblinfo[i].adef_expr[j] = strdup(PQgetvalue(res2, 0, PQfnumber(res2, "adsrc")));
				PQclear(res2);
			}
			else
				tblinfo[i].adef_expr[j] = NULL;
2849 2850 2851
		}
		PQclear(res);
	}
2852 2853

	destroyPQExpBuffer(q);
2854 2855 2856 2857
}


/*
2858 2859
 * getIndexes
 *	  read all the user-defined indexes information
2860 2861
 * from the system catalogs return them in the InhInfo* structure
 *
2862
 * numIndexes is set to the number of indexes read in
2863
 *
2864 2865
 *
 */
2866
IndInfo    *
2867
getIndexes(int *numIndexes)
2868
{
2869
	int			i;
2870
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
2871
	PGresult   *res;
2872
	int			ntups;
B
Bruce Momjian 已提交
2873
	IndInfo    *indinfo;
2874

2875 2876
	int			i_indexreloid;
	int			i_indreloid;
2877 2878
	int			i_indexrelname;
	int			i_indrelname;
2879
	int			i_indexdef;
2880
	int			i_indisprimary;
2881
	int			i_indkey;
2882 2883

	/*
2884
	 * find all the user-defined indexes.
2885
	 *
2886
	 * Notice we skip indexes on system classes
2887
	 *
2888
	 * XXXX: Use LOJ
2889 2890
	 */

B
Hi, all  
Bruce Momjian 已提交
2891
	appendPQExpBuffer(query,
2892 2893 2894
					  "SELECT i.indexrelid as indexreloid, "
					  "i.indrelid as indreloid, "
					  "t1.relname as indexrelname, t2.relname as indrelname, "
2895 2896 2897
					  "pg_get_indexdef(i.indexrelid) as indexdef, "
					  "i.indisprimary, i.indkey "
					  "from pg_index i, pg_class t1, pg_class t2 "
2898
					  "WHERE t1.oid = i.indexrelid and t2.oid = i.indrelid "
2899
					  "and i.indexrelid > '%u'::oid "
2900
					  "and t2.relname !~ '^pg_' ",
2901
					  g_last_builtin_oid);
2902

2903 2904 2905
	if (g_fout->remoteVersion < 70100)
		appendPQExpBuffer(query, " and t2.relkind != 'l'");

B
Hi, all  
Bruce Momjian 已提交
2906
	res = PQexec(g_conn, query->data);
2907 2908 2909
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
2910
		write_msg(NULL, "query to obtain list of indexes failed: %s", PQerrorMessage(g_conn));
2911
		exit_nicely();
2912 2913 2914 2915
	}

	ntups = PQntuples(res);

2916
	*numIndexes = ntups;
2917 2918 2919

	indinfo = (IndInfo *) malloc(ntups * sizeof(IndInfo));

2920 2921
	memset((char *) indinfo, 0, ntups * sizeof(IndInfo));

2922 2923
	i_indexreloid = PQfnumber(res, "indexreloid");
	i_indreloid = PQfnumber(res, "indreloid");
2924 2925
	i_indexrelname = PQfnumber(res, "indexrelname");
	i_indrelname = PQfnumber(res, "indrelname");
2926
	i_indexdef = PQfnumber(res, "indexdef");
2927
	i_indisprimary = PQfnumber(res, "indisprimary");
2928
	i_indkey = PQfnumber(res, "indkey");
2929 2930 2931

	for (i = 0; i < ntups; i++)
	{
2932 2933
		indinfo[i].indexreloid = strdup(PQgetvalue(res, i, i_indexreloid));
		indinfo[i].indreloid = strdup(PQgetvalue(res, i, i_indreloid));
2934 2935
		indinfo[i].indexrelname = strdup(PQgetvalue(res, i, i_indexrelname));
		indinfo[i].indrelname = strdup(PQgetvalue(res, i, i_indrelname));
2936 2937
		indinfo[i].indexdef = strdup(PQgetvalue(res, i, i_indexdef));
		indinfo[i].indisprimary = strdup(PQgetvalue(res, i, i_indisprimary));
2938 2939 2940
		parseNumericArray(PQgetvalue(res, i, i_indkey),
						  indinfo[i].indkey,
						  INDEX_MAX_KEYS);
2941 2942
	}
	PQclear(res);
2943 2944 2945

	destroyPQExpBuffer(query);

2946
	return indinfo;
2947 2948
}

B
Bruce,  
Bruce Momjian 已提交
2949
/*------------------------------------------------------------------
2950
 * dumpComment --
B
Bruce,  
Bruce Momjian 已提交
2951
 *
2952
 * This routine is used to dump any comments associated with the
B
Bruce,  
Bruce Momjian 已提交
2953
 * oid handed to this routine. The routine takes a constant character
2954 2955 2956 2957 2958 2959
 * string for the target part of the comment-creation command, plus
 * OID, class name, and subid which are the primary key for pg_description.
 * If a matching pg_description entry is found, it is dumped.
 * Additional dependencies can be passed for the comment, too --- this is
 * needed for VIEWs, whose comments are filed under the table OID but
 * which are dumped in order by their rule OID.
B
Bruce,  
Bruce Momjian 已提交
2960 2961 2962
 *------------------------------------------------------------------
*/

T
Tom Lane 已提交
2963
static void
2964
dumpComment(Archive *fout, const char *target, const char *oid,
2965
			const char *classname, int subid,
2966
			const char *((*deps)[]))
2967 2968
{
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
2969
	PQExpBuffer query;
2970
	int			i_description;
B
Bruce,  
Bruce Momjian 已提交
2971

2972 2973 2974 2975
	/* Comments are SCHEMA not data */
	if (dataOnly)
		return;

B
Bruce,  
Bruce Momjian 已提交
2976 2977 2978
	/*** Build query to find comment ***/

	query = createPQExpBuffer();
2979 2980 2981 2982

	if (fout->remoteVersion >= 70200)
	{
		appendPQExpBuffer(query, "SELECT description FROM pg_description "
2983
						  "WHERE objoid = '%s'::oid and classoid = "
2984 2985 2986 2987 2988 2989 2990
						  "(SELECT oid FROM pg_class where relname = '%s') "
						  "and objsubid = %d",
						  oid, classname, subid);
	}
	else
	{
		/* Note: this will fail to find attribute comments in pre-7.2... */
2991
		appendPQExpBuffer(query, "SELECT description FROM pg_description WHERE objoid = '%s'::oid", oid);
2992
	}
B
Bruce,  
Bruce Momjian 已提交
2993 2994 2995 2996

	/*** Execute query ***/

	res = PQexec(g_conn, query->data);
2997 2998
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
2999 3000
		write_msg(NULL, "query to get comment on oid %s failed: %s",
				   oid, PQerrorMessage(g_conn));
3001
		exit_nicely();
B
Bruce,  
Bruce Momjian 已提交
3002 3003 3004 3005
	}

	/*** If a comment exists, build COMMENT ON statement ***/

3006
	if (PQntuples(res) == 1)
3007
	{
B
Bruce,  
Bruce Momjian 已提交
3008
		i_description = PQfnumber(res, "description");
B
Bruce Momjian 已提交
3009
		resetPQExpBuffer(query);
3010
		appendPQExpBuffer(query, "COMMENT ON %s IS ", target);
3011 3012
		formatStringLiteral(query, PQgetvalue(res, 0, i_description),
							PASS_LFTAB);
3013
		appendPQExpBuffer(query, ";\n");
B
Bruce Momjian 已提交
3014

3015 3016
		ArchiveEntry(fout, oid, target, "COMMENT", deps,
					 query->data, "" /* Del */ ,
B
Bruce Momjian 已提交
3017
					 "" /* Copy */ , "" /* Owner */ , NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
3018 3019 3020 3021 3022
	}

	/*** Clear the statement buffer and return ***/

	PQclear(res);
3023
	destroyPQExpBuffer(query);
B
Bruce,  
Bruce Momjian 已提交
3024 3025 3026
}

/*------------------------------------------------------------------
3027
 * dumpDBComment --
B
Bruce,  
Bruce Momjian 已提交
3028
 *
3029 3030
 * This routine is used to dump any comments associated with the
 * database to which we are currently connected. If the user chose
B
Bruce,  
Bruce Momjian 已提交
3031 3032 3033 3034 3035
 * to dump the schema of the database, then this is the first
 * statement issued.
 *------------------------------------------------------------------
*/

3036
void
B
Bruce Momjian 已提交
3037
dumpDBComment(Archive *fout)
3038 3039
{
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
3040
	PQExpBuffer query;
3041
	int			i_oid;
B
Bruce,  
Bruce Momjian 已提交
3042 3043 3044 3045

	/*** Build query to find comment ***/

	query = createPQExpBuffer();
3046
	appendPQExpBuffer(query, "SELECT oid FROM pg_database WHERE datname = ");
3047
	formatStringLiteral(query, PQdb(g_conn), CONV_ALL);
B
Bruce,  
Bruce Momjian 已提交
3048 3049 3050 3051

	/*** Execute query ***/

	res = PQexec(g_conn, query->data);
3052 3053
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
3054 3055
		write_msg(NULL, "query to get database oid failed: %s",
				   PQerrorMessage(g_conn));
3056
		exit_nicely();
B
Bruce,  
Bruce Momjian 已提交
3057 3058 3059 3060
	}

	/*** If a comment exists, build COMMENT ON statement ***/

3061 3062
	if (PQntuples(res) != 0)
	{
B
Bruce,  
Bruce Momjian 已提交
3063 3064 3065
		i_oid = PQfnumber(res, "oid");
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "DATABASE %s", fmtId(PQdb(g_conn), force_quotes));
3066 3067
		dumpComment(fout, query->data, PQgetvalue(res, 0, i_oid),
					"pg_database", 0, NULL);
B
Bruce,  
Bruce Momjian 已提交
3068 3069 3070 3071 3072
	}

	/*** Clear the statement buffer and return ***/

	PQclear(res);
3073
	destroyPQExpBuffer(query);
B
Bruce,  
Bruce Momjian 已提交
3074 3075
}

3076 3077
/*
 * dumpTypes
3078
 *	  writes out to fout the queries to recreate all the user-defined types
3079 3080 3081
 *
 */
void
B
Bruce Momjian 已提交
3082
dumpTypes(Archive *fout, FuncInfo *finfo, int numFuncs,
3083
		  TypeInfo *tinfo, int numTypes)
3084
{
3085
	int			i;
3086
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3087
	PQExpBuffer delq = createPQExpBuffer();
3088
	int			funcInd;
3089
	const char *((*deps)[]);
3090
	int			depIdx;
3091 3092 3093 3094

	for (i = 0; i < numTypes; i++)
	{
		/* skip all the builtin types */
3095
		if (atooid(tinfo[i].oid) <= g_last_builtin_oid)
3096 3097 3098
			continue;

		/* skip relation types */
3099
		if (atooid(tinfo[i].typrelid) != 0)
3100 3101
			continue;

3102 3103 3104 3105
		/* skip undefined placeholder types */
		if (!tinfo[i].isDefined)
			continue;

3106 3107 3108 3109 3110
		/* skip all array types that start w/ underscore */
		if ((tinfo[i].typname[0] == '_') &&
			(strcmp(tinfo[i].typinput, "array_in") == 0))
			continue;

3111 3112 3113
		deps = malloc(sizeof(char*) * 10);
		depIdx = 0;

3114 3115 3116 3117 3118 3119
		/*
		 * before we create a type, we need to create the input and output
		 * functions for it, if they haven't been created already
		 */
		funcInd = findFuncByName(finfo, numFuncs, tinfo[i].typinput);
		if (funcInd != -1)
3120 3121
		{
			(*deps)[depIdx++] = strdup(finfo[funcInd].oid);
3122
			dumpOneFunc(fout, finfo, funcInd, tinfo, numTypes);
3123
		}
3124 3125 3126

		funcInd = findFuncByName(finfo, numFuncs, tinfo[i].typoutput);
		if (funcInd != -1)
3127 3128
		{
			(*deps)[depIdx++] = strdup(finfo[funcInd].oid);
3129
			dumpOneFunc(fout, finfo, funcInd, tinfo, numTypes);
3130
		}
3131

3132
		resetPQExpBuffer(delq);
B
Bruce Momjian 已提交
3133
		appendPQExpBuffer(delq, "DROP TYPE %s;\n", fmtId(tinfo[i].typname, force_quotes));
3134

B
Hi, all  
Bruce Momjian 已提交
3135 3136
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
3137
						  "CREATE TYPE %s "
3138
						  "( internallength = %s, externallength = %s,",
3139
						  fmtId(tinfo[i].typname, force_quotes),
3140 3141 3142 3143
						  (strcmp(tinfo[i].typlen, "-1") == 0) ?
						  "variable" : tinfo[i].typlen,
						  (strcmp(tinfo[i].typprtlen, "-1") == 0) ?
						  "variable" : tinfo[i].typprtlen);
3144 3145 3146 3147 3148 3149 3150
		/* cannot combine these because fmtId uses static result area */
		appendPQExpBuffer(q, " input = %s,",
						  fmtId(tinfo[i].typinput, force_quotes));
		appendPQExpBuffer(q, " output = %s,",
						  fmtId(tinfo[i].typoutput, force_quotes));
		appendPQExpBuffer(q, " send = %s,",
						  fmtId(tinfo[i].typsend, force_quotes));
3151
		appendPQExpBuffer(q, " receive = %s",
3152
						  fmtId(tinfo[i].typreceive, force_quotes));
3153 3154 3155 3156 3157 3158

		if (tinfo[i].typdefault != NULL)
		{
			appendPQExpBuffer(q, ", default = ");
			formatStringLiteral(q, tinfo[i].typdefault, CONV_ALL);
		}
3159 3160 3161

		if (tinfo[i].isArray)
		{
3162
			char	   *elemType;
3163

3164
			elemType = findTypeByOid(tinfo, numTypes, tinfo[i].typelem, zeroAsOpaque);
3165 3166
			if (elemType == NULL)
			{
3167
				write_msg(NULL, "notice: array type %s - type for elements (oid %s) is not dumped\n",
3168
						  tinfo[i].typname, tinfo[i].typelem);
3169 3170 3171
				continue;
			}

3172
			appendPQExpBuffer(q, ", element = %s, delimiter = ", elemType);
3173
			formatStringLiteral(q, tinfo[i].typdelim, CONV_ALL);
3174 3175

			(*deps)[depIdx++] = strdup(tinfo[i].typelem);
3176
		}
3177

3178 3179 3180 3181 3182
		if (strcmp(tinfo[i].typalign, "c") == 0)
			appendPQExpBuffer(q, ", alignment = char");
		else if (strcmp(tinfo[i].typalign, "s") == 0)
			appendPQExpBuffer(q, ", alignment = int2");
		else if (strcmp(tinfo[i].typalign, "i") == 0)
3183 3184 3185 3186 3187 3188
			appendPQExpBuffer(q, ", alignment = int4");
		else if (strcmp(tinfo[i].typalign, "d") == 0)
			appendPQExpBuffer(q, ", alignment = double");

		if (strcmp(tinfo[i].typstorage, "p") == 0)
			appendPQExpBuffer(q, ", storage = plain");
3189
		else if (strcmp(tinfo[i].typstorage, "e") == 0)
3190
			appendPQExpBuffer(q, ", storage = external");
3191
		else if (strcmp(tinfo[i].typstorage, "x") == 0)
3192
			appendPQExpBuffer(q, ", storage = extended");
3193
		else if (strcmp(tinfo[i].typstorage, "m") == 0)
3194 3195
			appendPQExpBuffer(q, ", storage = main");

3196
		if (tinfo[i].passedbyvalue)
3197
			appendPQExpBuffer(q, ", passedbyvalue);\n");
3198
		else
B
Hi, all  
Bruce Momjian 已提交
3199
			appendPQExpBuffer(q, ");\n");
3200

3201 3202 3203
		(*deps)[depIdx++] = NULL; /* End of List */

		ArchiveEntry(fout, tinfo[i].oid, tinfo[i].typname, "TYPE", deps,
B
Bruce Momjian 已提交
3204
				  q->data, delq->data, "", tinfo[i].usename, NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
3205 3206 3207 3208

		/*** Dump Type Comments ***/

		resetPQExpBuffer(q);
B
Bruce Momjian 已提交
3209

B
Bruce,  
Bruce Momjian 已提交
3210
		appendPQExpBuffer(q, "TYPE %s", fmtId(tinfo[i].typname, force_quotes));
3211
		dumpComment(fout, q->data, tinfo[i].oid, "pg_type", 0, NULL);
3212
	}
3213 3214 3215

	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
3216 3217
}

3218 3219
/*
 * dumpProcLangs
B
Bruce Momjian 已提交
3220
 *		  writes out to fout the queries to recreate user-defined procedural languages
3221 3222 3223
 *
 */
void
B
Bruce Momjian 已提交
3224
dumpProcLangs(Archive *fout, FuncInfo *finfo, int numFuncs,
B
Bruce Momjian 已提交
3225
			  TypeInfo *tinfo, int numTypes)
3226
{
3227 3228
	PGresult   *res;
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
3229 3230
	PQExpBuffer defqry = createPQExpBuffer();
	PQExpBuffer delqry = createPQExpBuffer();
3231
	int			ntups;
B
Bruce Momjian 已提交
3232
	int			i_oid;
3233 3234 3235 3236
	int			i_lanname;
	int			i_lanpltrusted;
	int			i_lanplcallfoid;
	int			i_lancompiler;
3237
	Oid			lanoid;
3238 3239 3240
	char	   *lanname;
	char	   *lancompiler;
	const char *lanplcallfoid;
3241 3242 3243
	int			i,
				fidx;

B
Bruce Momjian 已提交
3244
	appendPQExpBuffer(query, "SELECT oid, * FROM pg_language "
3245 3246
					  "WHERE lanispl "
					  "ORDER BY oid");
B
Hi, all  
Bruce Momjian 已提交
3247
	res = PQexec(g_conn, query->data);
3248 3249 3250
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
3251 3252
		write_msg(NULL, "query to obtain list of procedural languages failed: %s",
				   PQerrorMessage(g_conn));
3253
		exit_nicely();
3254 3255 3256
	}
	ntups = PQntuples(res);

B
Bruce Momjian 已提交
3257 3258 3259 3260
	i_lanname = PQfnumber(res, "lanname");
	i_lanpltrusted = PQfnumber(res, "lanpltrusted");
	i_lanplcallfoid = PQfnumber(res, "lanplcallfoid");
	i_lancompiler = PQfnumber(res, "lancompiler");
B
Bruce Momjian 已提交
3261
	i_oid = PQfnumber(res, "oid");
3262

B
Bruce Momjian 已提交
3263 3264
	for (i = 0; i < ntups; i++)
	{
3265
		lanoid = atooid(PQgetvalue(res, i, i_oid));
3266 3267 3268
		if (lanoid <= g_last_builtin_oid)
			continue;

3269
		lanplcallfoid = PQgetvalue(res, i, i_lanplcallfoid);
3270 3271


3272 3273 3274 3275 3276 3277 3278
		for (fidx = 0; fidx < numFuncs; fidx++)
		{
			if (!strcmp(finfo[fidx].oid, lanplcallfoid))
				break;
		}
		if (fidx >= numFuncs)
		{
3279 3280
			write_msg(NULL, "handler procedure for procedural language %s not found\n",
					   PQgetvalue(res, i, i_lanname));
3281
			exit_nicely();
3282 3283 3284 3285
		}

		dumpOneFunc(fout, finfo, fidx, tinfo, numTypes);

3286 3287
		lanname = PQgetvalue(res, i, i_lanname);
		lancompiler = PQgetvalue(res, i, i_lancompiler);
3288

3289
		appendPQExpBuffer(delqry, "DROP PROCEDURAL LANGUAGE ");
3290
		formatStringLiteral(delqry, lanname, CONV_ALL);
3291
		appendPQExpBuffer(delqry, ";\n");
3292

3293 3294 3295
		appendPQExpBuffer(defqry, "CREATE %sPROCEDURAL LANGUAGE ",
						  (PQgetvalue(res, i, i_lanpltrusted)[0] == 't') ?
						  "TRUSTED " : "");
3296
		formatStringLiteral(defqry, lanname, CONV_ALL);
3297 3298
		appendPQExpBuffer(defqry, " HANDLER %s LANCOMPILER ",
						  fmtId(finfo[fidx].proname, force_quotes));
3299
		formatStringLiteral(defqry, lancompiler, CONV_ALL);
3300
		appendPQExpBuffer(defqry, ";\n");
3301

B
Bruce Momjian 已提交
3302
		ArchiveEntry(fout, PQgetvalue(res, i, i_oid), lanname, "PROCEDURAL LANGUAGE",
B
Bruce Momjian 已提交
3303
				   NULL, defqry->data, delqry->data, "", "", NULL, NULL);
B
Bruce Momjian 已提交
3304

3305 3306
		resetPQExpBuffer(defqry);
		resetPQExpBuffer(delqry);
3307 3308 3309 3310
	}

	PQclear(res);

3311 3312 3313
	destroyPQExpBuffer(query);
	destroyPQExpBuffer(defqry);
	destroyPQExpBuffer(delqry);
3314 3315
}

3316 3317
/*
 * dumpFuncs
3318
 *	  writes out to fout the queries to recreate all the user-defined functions
3319 3320 3321
 *
 */
void
B
Bruce Momjian 已提交
3322
dumpFuncs(Archive *fout, FuncInfo *finfo, int numFuncs,
3323
		  TypeInfo *tinfo, int numTypes)
3324
{
3325
	int			i;
3326 3327 3328

	for (i = 0; i < numFuncs; i++)
		dumpOneFunc(fout, finfo, i, tinfo, numTypes);
3329 3330 3331 3332
}

/*
 * dumpOneFunc:
3333 3334
 *	  dump out only one function,  the index of which is given in the third
 *	argument
3335 3336 3337
 *
 */

3338
static void
B
Bruce Momjian 已提交
3339
dumpOneFunc(Archive *fout, FuncInfo *finfo, int i,
3340
			TypeInfo *tinfo, int numTypes)
3341
{
3342
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3343 3344
	PQExpBuffer fn = createPQExpBuffer();
	PQExpBuffer delqry = createPQExpBuffer();
B
Bruce,  
Bruce Momjian 已提交
3345
	PQExpBuffer fnlist = createPQExpBuffer();
3346
	PQExpBuffer asPart = createPQExpBuffer();
3347
	char		func_lang[NAMEDATALEN + 1];
3348 3349
	PGresult   *res;
	int			nlangs;
3350
	int			j;
3351 3352
	int			i_lanname;
	char		query[256];
3353

B
Bruce Momjian 已提交
3354 3355 3356 3357
	char	   *listSep;
	char	   *listSepComma = ",";
	char	   *listSepNone = "";
	char	   *rettypename;
3358

3359
	if (finfo[i].dumped)
3360 3361 3362
		goto done;

	finfo[i].dumped = 1;
3363

B
Bruce Momjian 已提交
3364
	/* becomeUser(fout, finfo[i].usename); */
3365

3366
	sprintf(query, "SELECT lanname FROM pg_language WHERE oid = '%u'::oid",
3367 3368 3369 3370
			finfo[i].lang);
	res = PQexec(g_conn, query);
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
B
Bruce Momjian 已提交
3371
	{
3372
		write_msg(NULL, "query to get name of procedural language failed: %s", PQerrorMessage(g_conn));
3373
		exit_nicely();
B
Bruce Momjian 已提交
3374
	}
3375 3376 3377
	nlangs = PQntuples(res);

	if (nlangs != 1)
B
Bruce Momjian 已提交
3378
	{
3379
		write_msg(NULL, "procedural language for function %s not found\n", finfo[i].proname);
3380
		exit_nicely();
B
Bruce Momjian 已提交
3381 3382
	}

3383
	i_lanname = PQfnumber(res, "lanname");
3384

3385
	/*
B
Bruce Momjian 已提交
3386 3387
	 * See backend/commands/define.c for details of how the 'AS' clause is
	 * used.
3388 3389 3390
	 */
	if (strcmp(finfo[i].probin, "-") != 0)
	{
3391
		appendPQExpBuffer(asPart, "AS ");
3392
		formatStringLiteral(asPart, finfo[i].probin, CONV_ALL);
3393
		if (strcmp(finfo[i].prosrc, "-") != 0)
3394 3395
		{
			appendPQExpBuffer(asPart, ", ");
3396
			formatStringLiteral(asPart, finfo[i].prosrc, PASS_LFTAB);
3397
		}
3398 3399 3400 3401
	}
	else
	{
		if (strcmp(finfo[i].prosrc, "-") != 0)
3402 3403
		{
			appendPQExpBuffer(asPart, "AS ");
3404
			formatStringLiteral(asPart, finfo[i].prosrc, PASS_LFTAB);
3405
		}
3406 3407
	}

3408
	strcpy(func_lang, PQgetvalue(res, 0, i_lanname));
B
Bruce Momjian 已提交
3409

3410
	PQclear(res);
B
Bruce Momjian 已提交
3411

B
Bruce Momjian 已提交
3412 3413
	resetPQExpBuffer(fn);
	appendPQExpBuffer(fn, "%s (", fmtId(finfo[i].proname, force_quotes));
3414 3415
	for (j = 0; j < finfo[i].nargs; j++)
	{
B
Bruce Momjian 已提交
3416
		char	   *typname;
3417

3418
		typname = findTypeByOid(tinfo, numTypes, finfo[i].argtypes[j], zeroAsOpaque);
3419 3420
		if (typname == NULL)
		{
3421 3422
			write_msg(NULL, "WARNING: function \"%s\" not dumped\n",
					  finfo[i].proname);
3423

3424
			write_msg(NULL, "reason: data type name of argument %d (oid %s) not found\n",
3425
					   j, finfo[i].argtypes[j]);
3426
			goto done;
3427 3428
		}

B
Bruce Momjian 已提交
3429 3430 3431
		appendPQExpBuffer(fn, "%s%s",
						  (j > 0) ? "," : "",
						  typname);
B
Bruce,  
Bruce Momjian 已提交
3432
		appendPQExpBuffer(fnlist, "%s%s",
B
Bruce Momjian 已提交
3433 3434
						  (j > 0) ? "," : "",
						  typname);
3435
	}
B
Bruce Momjian 已提交
3436 3437 3438
	appendPQExpBuffer(fn, ")");

	resetPQExpBuffer(delqry);
B
Bruce Momjian 已提交
3439
	appendPQExpBuffer(delqry, "DROP FUNCTION %s;\n", fn->data);
B
Bruce Momjian 已提交
3440

3441 3442 3443 3444
	rettypename = findTypeByOid(tinfo, numTypes, finfo[i].prorettype, zeroAsOpaque);

	if (rettypename == NULL)
	{
3445 3446
		write_msg(NULL, "WARNING: function \"%s\" not dumped\n",
				  finfo[i].proname);
3447

3448 3449
		write_msg(NULL, "reason: name of return data type (oid %s) not found\n",
				  finfo[i].prorettype);
3450
		goto done;
3451 3452
	}

B
Bruce Momjian 已提交
3453
	resetPQExpBuffer(q);
B
Bruce Momjian 已提交
3454
	appendPQExpBuffer(q, "CREATE FUNCTION %s ", fn->data);
3455 3456
	appendPQExpBuffer(q, "RETURNS %s%s %s LANGUAGE ",
					  (finfo[i].retset) ? "SETOF " : "",
3457
					  rettypename,
3458
					  asPart->data);
3459
	formatStringLiteral(q, func_lang, CONV_ALL);
3460

B
Bruce Momjian 已提交
3461
	if (finfo[i].iscachable || finfo[i].isstrict)		/* OR in new attrs here */
3462
	{
3463 3464 3465
		appendPQExpBuffer(q, " WITH (");
		listSep = listSepNone;

B
Bruce Momjian 已提交
3466 3467
		if (finfo[i].iscachable)
		{
3468 3469 3470 3471
			appendPQExpBuffer(q, "%s iscachable", listSep);
			listSep = listSepComma;
		}

B
Bruce Momjian 已提交
3472 3473
		if (finfo[i].isstrict)
		{
3474 3475 3476 3477
			appendPQExpBuffer(q, "%s isstrict", listSep);
			listSep = listSepComma;
		}
		appendPQExpBuffer(q, " )");
3478 3479 3480 3481
	}

	appendPQExpBuffer(q, ";\n");

B
Bruce Momjian 已提交
3482
	ArchiveEntry(fout, finfo[i].oid, fn->data, "FUNCTION", NULL, q->data, delqry->data,
B
Bruce Momjian 已提交
3483
				 "", finfo[i].usename, NULL, NULL);
3484

B
Bruce,  
Bruce Momjian 已提交
3485 3486 3487 3488 3489 3490
	/*** Dump Function Comments ***/

	resetPQExpBuffer(q);
	appendPQExpBuffer(q, "FUNCTION %s ",
					  fmtId(finfo[i].proname, force_quotes));
	appendPQExpBuffer(q, "( %s )", fnlist->data);
3491
	dumpComment(fout, q->data, finfo[i].oid, "pg_proc", 0, NULL);
3492 3493 3494 3495 3496 3497 3498

done:
	destroyPQExpBuffer(q);
	destroyPQExpBuffer(fn);
	destroyPQExpBuffer(delqry);
	destroyPQExpBuffer(fnlist);
	destroyPQExpBuffer(asPart);
3499 3500 3501 3502
}

/*
 * dumpOprs
3503
 *	  writes out to fout the queries to recreate all the user-defined operators
3504 3505
 *
 */
3506
void
B
Bruce Momjian 已提交
3507
dumpOprs(Archive *fout, OprInfo *oprinfo, int numOperators,
3508
		 TypeInfo *tinfo, int numTypes)
3509
{
B
Bruce Momjian 已提交
3510
	int			i;
3511
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3512
	PQExpBuffer delq = createPQExpBuffer();
3513 3514 3515 3516 3517 3518 3519 3520
	PQExpBuffer leftarg = createPQExpBuffer();
	PQExpBuffer rightarg = createPQExpBuffer();
	PQExpBuffer commutator = createPQExpBuffer();
	PQExpBuffer negator = createPQExpBuffer();
	PQExpBuffer restrictor = createPQExpBuffer();
	PQExpBuffer join = createPQExpBuffer();
	PQExpBuffer sort1 = createPQExpBuffer();
	PQExpBuffer sort2 = createPQExpBuffer();
3521 3522 3523

	for (i = 0; i < numOperators; i++)
	{
B
Bruce Momjian 已提交
3524
		char	   *name;
3525

B
Hi, all  
Bruce Momjian 已提交
3526 3527 3528 3529 3530 3531 3532 3533 3534
		resetPQExpBuffer(leftarg);
		resetPQExpBuffer(rightarg);
		resetPQExpBuffer(commutator);
		resetPQExpBuffer(negator);
		resetPQExpBuffer(restrictor);
		resetPQExpBuffer(join);
		resetPQExpBuffer(sort1);
		resetPQExpBuffer(sort2);

3535
		/* skip all the builtin oids */
3536
		if (atooid(oprinfo[i].oid) <= g_last_builtin_oid)
3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552
			continue;

		/*
		 * some operator are invalid because they were the result of user
		 * defining operators before commutators exist
		 */
		if (strcmp(oprinfo[i].oprcode, "-") == 0)
			continue;

		/*
		 * right unary means there's a left arg and left unary means
		 * there's a right arg
		 */
		if (strcmp(oprinfo[i].oprkind, "r") == 0 ||
			strcmp(oprinfo[i].oprkind, "b") == 0)
		{
B
Bruce Momjian 已提交
3553 3554
			name = findTypeByOid(tinfo, numTypes,
								 oprinfo[i].oprleft, zeroAsOpaque);
3555 3556
			if (name == NULL)
			{
3557 3558 3559 3560
				write_msg(NULL, "WARNING: operator \"%s\" (oid %s) not dumped\n",
						  oprinfo[i].oprname, oprinfo[i].oid);
				write_msg(NULL, "reason: oprleft (oid %s) not found\n",
						  oprinfo[i].oprleft);
3561 3562
				continue;
			}
B
Bruce Momjian 已提交
3563
			appendPQExpBuffer(leftarg, ",\n\tLEFTARG = %s ", name);
3564
		}
3565

3566 3567 3568
		if (strcmp(oprinfo[i].oprkind, "l") == 0 ||
			strcmp(oprinfo[i].oprkind, "b") == 0)
		{
B
Bruce Momjian 已提交
3569 3570
			name = findTypeByOid(tinfo, numTypes,
								 oprinfo[i].oprright, zeroAsOpaque);
3571 3572
			if (name == NULL)
			{
3573 3574 3575 3576
				write_msg(NULL, "WARNING: operator \"%s\" (oid %s) not dumped\n",
						  oprinfo[i].oprname, oprinfo[i].oid);
				write_msg(NULL, "reason: oprright (oid %s) not found\n",
						  oprinfo[i].oprright);
3577 3578 3579
				continue;
			}
			appendPQExpBuffer(rightarg, ",\n\tRIGHTARG = %s ", name);
3580
		}
3581

B
Hi, all  
Bruce Momjian 已提交
3582
		if (!(strcmp(oprinfo[i].oprcom, "0") == 0))
3583 3584 3585 3586
		{
			name = findOprByOid(oprinfo, numOperators, oprinfo[i].oprcom);
			if (name == NULL)
			{
3587 3588 3589 3590
				write_msg(NULL, "WARNING: operator \"%s\" (oid %s) not dumped\n",
						  oprinfo[i].oprname, oprinfo[i].oid);
				write_msg(NULL, "reason: oprcom (oid %s) not found\n",
						  oprinfo[i].oprcom);
3591 3592 3593 3594
				continue;
			}
			appendPQExpBuffer(commutator, ",\n\tCOMMUTATOR = %s ", name);
		}
3595

B
Hi, all  
Bruce Momjian 已提交
3596
		if (!(strcmp(oprinfo[i].oprnegate, "0") == 0))
3597 3598 3599 3600
		{
			name = findOprByOid(oprinfo, numOperators, oprinfo[i].oprnegate);
			if (name == NULL)
			{
3601 3602 3603 3604
				write_msg(NULL, "WARNING: operator \"%s\" (oid %s) not dumped\n",
						  oprinfo[i].oprname, oprinfo[i].oid);
				write_msg(NULL, "reason: oprnegate (oid %s) not found\n",
						  oprinfo[i].oprnegate);
3605 3606 3607 3608
				continue;
			}
			appendPQExpBuffer(negator, ",\n\tNEGATOR = %s ", name);
		}
3609

B
Hi, all  
Bruce Momjian 已提交
3610 3611
		if (!(strcmp(oprinfo[i].oprrest, "-") == 0))
			appendPQExpBuffer(restrictor, ",\n\tRESTRICT = %s ", oprinfo[i].oprrest);
3612

B
Hi, all  
Bruce Momjian 已提交
3613 3614
		if (!(strcmp(oprinfo[i].oprjoin, "-") == 0))
			appendPQExpBuffer(join, ",\n\tJOIN = %s ", oprinfo[i].oprjoin);
3615

B
Hi, all  
Bruce Momjian 已提交
3616
		if (!(strcmp(oprinfo[i].oprlsortop, "0") == 0))
3617
		{
B
Bruce Momjian 已提交
3618
			name = findOprByOid(oprinfo, numOperators, oprinfo[i].oprlsortop);
3619 3620
			if (name == NULL)
			{
3621 3622 3623 3624
				write_msg(NULL, "WARNING: operator \"%s\" (oid %s) not dumped\n",
						  oprinfo[i].oprname, oprinfo[i].oid);
				write_msg(NULL, "reason: oprlsortop (oid %s) not found\n",
						  oprinfo[i].oprlsortop);
3625 3626 3627 3628
				continue;
			}
			appendPQExpBuffer(sort1, ",\n\tSORT1 = %s ", name);
		}
3629

B
Hi, all  
Bruce Momjian 已提交
3630
		if (!(strcmp(oprinfo[i].oprrsortop, "0") == 0))
3631
		{
B
Bruce Momjian 已提交
3632
			name = findOprByOid(oprinfo, numOperators, oprinfo[i].oprrsortop);
3633 3634
			if (name == NULL)
			{
3635 3636 3637 3638
				write_msg(NULL, "WARNING: operator \"%s\" (oid %s) not dumped\n",
						  oprinfo[i].oprname, oprinfo[i].oid);
				write_msg(NULL, "reason: oprrsortop (oid %s) not found\n",
						  oprinfo[i].oprrsortop);
3639 3640 3641 3642
				continue;
			}
			appendPQExpBuffer(sort2, ",\n\tSORT2 = %s ", name);
		}
3643

B
Bruce Momjian 已提交
3644 3645
		resetPQExpBuffer(delq);
		appendPQExpBuffer(delq, "DROP OPERATOR %s (%s", oprinfo[i].oprname,
B
Bruce Momjian 已提交
3646
		findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft, zeroAsOpaque));
B
Bruce Momjian 已提交
3647
		appendPQExpBuffer(delq, ", %s);\n",
B
Bruce Momjian 已提交
3648
						  findTypeByOid(tinfo, numTypes, oprinfo[i].oprright, zeroAsOpaque));
3649

B
Hi, all  
Bruce Momjian 已提交
3650 3651
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
3652 3653 3654 3655 3656 3657 3658 3659 3660
						  "CREATE OPERATOR %s "
						  "(PROCEDURE = %s %s%s%s%s%s%s%s%s%s);\n",
						  oprinfo[i].oprname,
						  oprinfo[i].oprcode,
						  leftarg->data,
						  rightarg->data,
						  commutator->data,
						  negator->data,
						  restrictor->data,
B
Bruce Momjian 已提交
3661
		  (strcmp(oprinfo[i].oprcanhash, "t") == 0) ? ",\n\tHASHES" : "",
3662 3663 3664
						  join->data,
						  sort1->data,
						  sort2->data);
3665

B
Bruce Momjian 已提交
3666
		ArchiveEntry(fout, oprinfo[i].oid, oprinfo[i].oprname, "OPERATOR", NULL,
B
Bruce Momjian 已提交
3667
				q->data, delq->data, "", oprinfo[i].usename, NULL, NULL);
3668
	}
3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679

	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
	destroyPQExpBuffer(leftarg);
	destroyPQExpBuffer(rightarg);
	destroyPQExpBuffer(commutator);
	destroyPQExpBuffer(negator);
	destroyPQExpBuffer(restrictor);
	destroyPQExpBuffer(join);
	destroyPQExpBuffer(sort1);
	destroyPQExpBuffer(sort2);
3680 3681 3682 3683
}

/*
 * dumpAggs
3684
 *	  writes out to fout the queries to create all the user-defined aggregates
3685 3686 3687
 *
 */
void
B
Bruce Momjian 已提交
3688
dumpAggs(Archive *fout, AggInfo *agginfo, int numAggs,
3689
		 TypeInfo *tinfo, int numTypes)
3690
{
B
Bruce Momjian 已提交
3691
	int			i;
3692
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3693 3694
	PQExpBuffer delq = createPQExpBuffer();
	PQExpBuffer aggSig = createPQExpBuffer();
3695
	PQExpBuffer details = createPQExpBuffer();
3696 3697 3698

	for (i = 0; i < numAggs; i++)
	{
B
Bruce Momjian 已提交
3699
		char	   *name;
3700

3701
		resetPQExpBuffer(details);
B
Hi, all  
Bruce Momjian 已提交
3702

3703
		/* skip all the builtin oids */
3704 3705 3706 3707 3708 3709 3710 3711 3712
		if ( oidle( atooid(agginfo[i].oid), g_last_builtin_oid) )
			continue;

		resetPQExpBuffer(aggSig);
		appendPQExpBuffer(aggSig, "%s(%s)", agginfo[i].aggname,
						  findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsOpaque + useBaseTypeName));

		if (!agginfo[i].convertok)
		{
3713 3714 3715
			write_msg(NULL, "WARNING: aggregate function %s could not be dumped correctly for this database version; ignored\n",
					  aggSig->data);

3716
			resetPQExpBuffer(q);
3717
			appendPQExpBuffer(q, "-- WARNING: aggregate function %s could not be dumped correctly for this database version; ignored\n",
3718 3719 3720
									aggSig->data);
			ArchiveEntry(fout, agginfo[i].oid, aggSig->data, "WARNING", NULL,
					q->data, "" /* Del */, "", agginfo[i].usename, NULL, NULL);
3721
			continue;
3722
		}
3723

3724 3725 3726
		name = findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsAny + useBaseTypeName);
		if (name == NULL)
		{
3727 3728 3729 3730 3731 3732 3733 3734 3735 3736
			write_msg(NULL, "WARNING: aggregate function \"%s\" (oid %s) not dumped\n",
					  agginfo[i].aggname, agginfo[i].oid);
			write_msg(NULL, "reason: aggbasetype (oid %s) not found\n",
					  agginfo[i].aggbasetype);

			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "-- WARNING: aggregate function \"%s\" (oid %s) not dumped\n", agginfo[i].aggname, agginfo[i].oid);
			appendPQExpBuffer(q, "-- reason: aggbasetype (oid %s) not found\n", agginfo[i].aggbasetype);
			ArchiveEntry(fout, agginfo[i].oid, aggSig->data, "WARNING", NULL,
						 q->data, "" /* Del */, "", agginfo[i].usename, NULL, NULL);
3737 3738 3739
			continue;
		}
		appendPQExpBuffer(details, "BASETYPE = %s, ", name);
3740

3741 3742 3743
		name = findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype, zeroAsOpaque + useBaseTypeName);
		if (name == NULL)
		{
3744 3745 3746 3747 3748 3749 3750 3751 3752 3753
			write_msg(NULL, "WARNING: aggregate function \"%s\" (oid %s) not dumped\n",
					  agginfo[i].aggname, agginfo[i].oid);
			write_msg(NULL, "reason: aggtranstype (oid %s) not found\n",
					  agginfo[i].aggtranstype);

			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "-- WARNING: aggregate function \"%s\" (oid %s) not dumped\n", agginfo[i].aggname, agginfo[i].oid);
			appendPQExpBuffer(q, "-- reason: aggtranstype (oid %s) not found\n", agginfo[i].aggtranstype);
			ArchiveEntry(fout, agginfo[i].oid, aggSig->data, "WARNING", NULL,
						 q->data, "" /* Del */, "", agginfo[i].usename, NULL, NULL);
3754 3755
			continue;
		}
3756 3757
		appendPQExpBuffer(details,
						  "SFUNC = %s, STYPE = %s",
3758
						  agginfo[i].aggtransfn, name);
3759

3760
		if (agginfo[i].agginitval)
3761 3762
		{
			appendPQExpBuffer(details, ", INITCOND = ");
3763
			formatStringLiteral(details, agginfo[i].agginitval, CONV_ALL);
3764
		}
3765

B
Hi, all  
Bruce Momjian 已提交
3766
		if (!(strcmp(agginfo[i].aggfinalfn, "-") == 0))
3767 3768
			appendPQExpBuffer(details, ", FINALFUNC = %s",
							  agginfo[i].aggfinalfn);
3769

B
Bruce Momjian 已提交
3770 3771
		resetPQExpBuffer(delq);
		appendPQExpBuffer(delq, "DROP AGGREGATE %s;\n", aggSig->data);
3772

B
Hi, all  
Bruce Momjian 已提交
3773
		resetPQExpBuffer(q);
3774
		appendPQExpBuffer(q, "CREATE AGGREGATE %s ( %s );\n",
3775
						  agginfo[i].aggname,
3776
						  details->data);
3777

B
Bruce Momjian 已提交
3778
		ArchiveEntry(fout, agginfo[i].oid, aggSig->data, "AGGREGATE", NULL,
B
Bruce Momjian 已提交
3779
				q->data, delq->data, "", agginfo[i].usename, NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
3780 3781 3782 3783

		/*** Dump Aggregate Comments ***/

		resetPQExpBuffer(q);
3784
		appendPQExpBuffer(q, "AGGREGATE %s %s", agginfo[i].aggname,
3785
						  findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsOpaque + useBaseTypeName));
3786
		dumpComment(fout, q->data, agginfo[i].oid, "pg_aggregate", 0, NULL);
3787
	}
3788 3789 3790 3791 3792

	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
	destroyPQExpBuffer(aggSig);
	destroyPQExpBuffer(details);
3793 3794
}

3795 3796 3797 3798 3799
/*
 * These are some support functions to fix the acl problem of pg_dump
 *
 * Matthew C. Aycock 12/02/97
 */
3800 3801 3802

/* Append a keyword to a keyword list, inserting comma if needed.
 * Caller must make aclbuf big enough for all possible keywords.
3803
 */
3804
static void
B
Bruce Momjian 已提交
3805
AddAcl(char *aclbuf, const char *keyword)
3806
{
3807 3808 3809
	if (*aclbuf)
		strcat(aclbuf, ",");
	strcat(aclbuf, keyword);
3810
}
3811

3812
/*
3813
 * This will take a string of 'arwR' and return a malloced,
3814 3815
 * comma delimited string of SELECT,INSERT,UPDATE,DELETE,RULE
 */
V
Vadim B. Mikheev 已提交
3816
static char *
3817
GetPrivileges(Archive *AH, const char *s)
3818
{
3819
	char		aclbuf[100];
3820

3821
	aclbuf[0] = '\0';
3822

B
Bruce Momjian 已提交
3823
	if (strchr(s, 'a'))
3824
		AddAcl(aclbuf, "INSERT");
3825

B
Bruce Momjian 已提交
3826
	if (strchr(s, 'r'))
3827
		AddAcl(aclbuf, "SELECT");
3828

3829
	if (strchr(s, 'R'))
3830
		AddAcl(aclbuf, "RULE");
3831

3832 3833 3834 3835 3836 3837 3838 3839 3840 3841 3842 3843 3844 3845 3846 3847
	if (AH->remoteVersion >= 70200)
	{
		if (strchr(s, 'w'))
			AddAcl(aclbuf, "UPDATE");
		if (strchr(s, 'd'))
			AddAcl(aclbuf, "DELETE");
		if (strchr(s, 'x'))
			AddAcl(aclbuf, "REFERENCES");
		if (strchr(s, 't'))
			AddAcl(aclbuf, "TRIGGER");
	}
	else
	{
		if (strchr(s, 'w'))
			AddAcl(aclbuf, "UPDATE,DELETE");
	}
3848

3849
	return strdup(aclbuf);
3850
}
3851

B
Bruce Momjian 已提交
3852
/*
B
Bruce Momjian 已提交
3853
 * The name says it all; a function to append a string is the dest
B
Bruce Momjian 已提交
3854 3855
 * is big enough. If not, it does a realloc.
 */
B
Bruce Momjian 已提交
3856 3857
static void
strcatalloc(char **dest, int *dSize, char *src)
B
Bruce Momjian 已提交
3858
{
B
Bruce Momjian 已提交
3859 3860 3861 3862 3863 3864 3865 3866 3867 3868
	int			dLen = strlen(*dest);
	int			sLen = strlen(src);

	if ((dLen + sLen) >= *dSize)
	{
		*dSize = (dLen + sLen) * 2;
		*dest = realloc(*dest, *dSize);
	}
	strcpy(*dest + dLen, src);
}
B
Bruce Momjian 已提交
3869 3870


B
Bruce Momjian 已提交
3871 3872
/*
 * dumpACL:
3873 3874
 *	  Write out grant/revoke information
 *	  Called for sequences and tables
B
Bruce Momjian 已提交
3875 3876
 */

3877
static void
B
Bruce Momjian 已提交
3878
dumpACL(Archive *fout, TableInfo tbinfo)
B
Bruce Momjian 已提交
3879
{
B
Bruce Momjian 已提交
3880 3881
	const char *acls = tbinfo.relacl;
	char	   *aclbuf,
B
Bruce Momjian 已提交
3882 3883 3884
			   *tok,
			   *eqpos,
			   *priv;
3885
	char	   *objoid;
B
Bruce Momjian 已提交
3886
	char	   *sql;
B
Bruce Momjian 已提交
3887
	char		tmp[1024];
B
Bruce Momjian 已提交
3888
	int			sSize = 4096;
3889 3890 3891 3892

	if (strlen(acls) == 0)
		return;					/* table has default permissions */

B
Bruce Momjian 已提交
3893 3894 3895
	/*
	 * Allocate a larginsh buffer for the output SQL.
	 */
B
Bruce Momjian 已提交
3896
	sql = (char *) malloc(sSize);
B
Bruce Momjian 已提交
3897

B
Bruce Momjian 已提交
3898 3899 3900
	/*
	 * Revoke Default permissions for PUBLIC. Is this actually necessary,
	 * or is it just a waste of time?
3901
	 */
B
Bruce Momjian 已提交
3902
	sprintf(sql, "REVOKE ALL on %s from PUBLIC;\n",
3903
			fmtId(tbinfo.relname, force_quotes));
3904 3905 3906

	/* Make a working copy of acls so we can use strtok */
	aclbuf = strdup(acls);
B
Bruce Momjian 已提交
3907

3908 3909
	/* Scan comma-separated ACL items */
	for (tok = strtok(aclbuf, ","); tok != NULL; tok = strtok(NULL, ","))
3910
	{
3911

B
Bruce Momjian 已提交
3912 3913 3914
		/*
		 * Token may start with '{' and/or '"'.  Actually only the start
		 * of the string should have '{', but we don't verify that.
3915 3916 3917 3918 3919 3920 3921 3922
		 */
		if (*tok == '{')
			tok++;
		if (*tok == '"')
			tok++;

		/* User name is string up to = in tok */
		eqpos = strchr(tok, '=');
B
Bruce Momjian 已提交
3923
		if (!eqpos)
B
Bruce Momjian 已提交
3924
		{
3925 3926 3927
			write_msg(NULL, "could not parse ACL list ('%s') for relation %s\n",
					  acls, tbinfo.relname);
			exit_nicely();
B
Bruce Momjian 已提交
3928 3929
		}

B
Bruce Momjian 已提交
3930 3931 3932 3933
		/*
		 * Parse the privileges (right-hand side).	Skip if there are
		 * none.
		 */
3934
		priv = GetPrivileges(fout, eqpos + 1);
3935
		if (*priv)
3936
		{
B
Bruce Momjian 已提交
3937
			sprintf(tmp, "GRANT %s on %s to ",
3938
					priv, fmtId(tbinfo.relname, force_quotes));
B
Bruce Momjian 已提交
3939
			strcatalloc(&sql, &sSize, tmp);
B
Bruce Momjian 已提交
3940 3941 3942

			/*
			 * Note: fmtId() can only be called once per printf, so don't
3943 3944 3945 3946 3947
			 * try to merge printing of username into the above printf.
			 */
			if (eqpos == tok)
			{
				/* Empty left-hand side means "PUBLIC" */
B
Bruce Momjian 已提交
3948
				strcatalloc(&sql, &sSize, "PUBLIC;\n");
3949
			}
3950
			else
3951 3952
			{
				*eqpos = '\0';	/* it's ok to clobber aclbuf */
3953
				if (strncmp(tok, "group ", strlen("group ")) == 0)
B
Bruce Momjian 已提交
3954
					sprintf(tmp, "GROUP %s;\n",
3955 3956
							fmtId(tok + strlen("group "), force_quotes));
				else
B
Bruce Momjian 已提交
3957 3958
					sprintf(tmp, "%s;\n", fmtId(tok, force_quotes));
				strcatalloc(&sql, &sSize, tmp);
3959
			}
3960
		}
3961
		free(priv);
B
Bruce Momjian 已提交
3962
	}
3963 3964

	free(aclbuf);
B
Bruce Momjian 已提交
3965

3966 3967 3968 3969 3970 3971
	if (tbinfo.viewdef != NULL)
		objoid = tbinfo.viewoid;
	else
		objoid = tbinfo.oid;

	ArchiveEntry(fout, objoid, tbinfo.relname, "ACL", NULL, sql, "", "", "", NULL, NULL);
B
Bruce Momjian 已提交
3972

B
Bruce Momjian 已提交
3973 3974
}

3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004 4005 4006 4007 4008 4009 4010 4011 4012 4013 4014 4015 4016 4017 4018 4019 4020 4021 4022 4023 4024 4025 4026 4027 4028 4029
static void
_dumpTableAttr70(Archive *fout, TableInfo *tblinfo, int i, int j, PQExpBuffer q)
{
	int32				tmp_typmod;
	int					precision;
	int					scale;

	/* Show lengths on bpchar and varchar */
	if (!strcmp(tblinfo[i].typnames[j], "bpchar"))
	{
		int			len = (tblinfo[i].atttypmod[j] - VARHDRSZ);

		appendPQExpBuffer(q, "character");
		if (len > 1)
			appendPQExpBuffer(q, "(%d)",
					 tblinfo[i].atttypmod[j] - VARHDRSZ);
	}
	else if (!strcmp(tblinfo[i].typnames[j], "varchar"))
	{
		appendPQExpBuffer(q, "character varying");
		if (tblinfo[i].atttypmod[j] != -1)
		{
			appendPQExpBuffer(q, "(%d)",
					 tblinfo[i].atttypmod[j] - VARHDRSZ);
		}
	}
	else if (!strcmp(tblinfo[i].typnames[j], "numeric"))
	{
		appendPQExpBuffer(q, "numeric");
		if (tblinfo[i].atttypmod[j] != -1)
		{
			tmp_typmod = tblinfo[i].atttypmod[j] - VARHDRSZ;
			precision = (tmp_typmod >> 16) & 0xffff;
			scale = tmp_typmod & 0xffff;
			appendPQExpBuffer(q, "(%d,%d)",
							  precision, scale);
		}
	}

	/*
	 * char is an internal single-byte data type; Let's
	 * make sure we force it through with quotes. - thomas
	 * 1998-12-13
	 */
	else if (!strcmp(tblinfo[i].typnames[j], "char"))
	{
		appendPQExpBuffer(q, "%s",
					fmtId(tblinfo[i].typnames[j], true));
	}
	else
	{
		appendPQExpBuffer(q, "%s",
				   fmtId(tblinfo[i].typnames[j], false));
	}
}
4030

4031 4032
/*
 * dumpTables:
4033
 *	  write out to fout all the user-define tables
4034
 */
4035

4036
void
B
Bruce Momjian 已提交
4037
dumpTables(Archive *fout, TableInfo *tblinfo, int numTables,
4038
		   IndInfo *indinfo, int numIndexes,
4039 4040
		   InhInfo *inhinfo, int numInherits,
		   TypeInfo *tinfo, int numTypes, const char *tablename,
B
Bruce Momjian 已提交
4041 4042
		   const bool aclsSkip, const bool oids,
		   const bool schemaOnly, const bool dataOnly)
4043
{
4044 4045
	int			i,
				j,
B
Bruce Momjian 已提交
4046
				k;
4047
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
4048
	PQExpBuffer delq = createPQExpBuffer();
4049
	char	   *serialSeq = NULL;		/* implicit sequence name created
B
Bruce Momjian 已提交
4050 4051 4052 4053
										 * by SERIAL datatype */
	const char *serialSeqSuffix = "_id_seq";	/* suffix for implicit
												 * SERIAL sequences */
	char	  **parentRels;		/* list of names of parent relations */
4054
	int			numParents;
B
Bruce Momjian 已提交
4055
	int			actual_atts;	/* number of attrs in this CREATE statment */
4056
	char	   *reltypename;
4057
	char	   *objoid;
4058
	const char *((*commentDeps)[]);
4059

V
Vadim B. Mikheev 已提交
4060
	/* First - dump SEQUENCEs */
4061
	if (tablename && strlen(tablename) > 0)
B
Bruce Momjian 已提交
4062
	{
4063 4064
		/* XXX this code only works for serial columns named "id" */
		/* We really need dependency analysis! */
B
Bruce Momjian 已提交
4065 4066 4067 4068
		serialSeq = malloc(strlen(tablename) + strlen(serialSeqSuffix) + 1);
		strcpy(serialSeq, tablename);
		strcat(serialSeq, serialSeqSuffix);
	}
V
Vadim B. Mikheev 已提交
4069 4070 4071 4072
	for (i = 0; i < numTables; i++)
	{
		if (!(tblinfo[i].sequence))
			continue;
B
Bruce Momjian 已提交
4073
		if (!tablename || (!strcmp(tblinfo[i].relname, tablename))
B
Bruce Momjian 已提交
4074
			|| (serialSeq && !strcmp(tblinfo[i].relname, serialSeq)))
V
Vadim B. Mikheev 已提交
4075
		{
B
Bruce Momjian 已提交
4076
			/* becomeUser(fout, tblinfo[i].usename); */
4077
			dumpSequence(fout, tblinfo[i], schemaOnly, dataOnly);
4078
			if (!aclsSkip)
B
Bruce Momjian 已提交
4079
				dumpACL(fout, tblinfo[i]);
V
Vadim B. Mikheev 已提交
4080 4081
		}
	}
4082
	if (serialSeq)
B
Bruce Momjian 已提交
4083
		free(serialSeq);
4084

4085 4086
	for (i = 0; i < numTables; i++)
	{
4087
		if (tblinfo[i].sequence)/* already dumped */
V
Vadim B. Mikheev 已提交
4088
			continue;
4089

4090
		if (!tablename || (!strcmp(tblinfo[i].relname, tablename)) || (strlen(tablename) == 0))
4091 4092
		{

4093 4094
			resetPQExpBuffer(delq);
			resetPQExpBuffer(q);
4095

4096 4097 4098 4099
			/* Use the view definition if there is one */
			if (tblinfo[i].viewdef != NULL)
			{
				reltypename = "VIEW";
4100
				objoid = tblinfo[i].viewoid;
4101
				appendPQExpBuffer(delq, "DROP VIEW %s;\n", fmtId(tblinfo[i].relname, force_quotes));
4102
				appendPQExpBuffer(q, "CREATE VIEW %s as %s\n", fmtId(tblinfo[i].relname, force_quotes), tblinfo[i].viewdef);
4103 4104 4105
				commentDeps = malloc(sizeof(char*) * 2);
				(*commentDeps)[0] = strdup(objoid);
				(*commentDeps)[1] = NULL; /* end of list */
4106 4107
			}
			else
4108
			{
4109
				reltypename = "TABLE";
4110
				objoid = tblinfo[i].oid;
4111
				commentDeps = NULL;
4112 4113
				parentRels = tblinfo[i].parentRels;
				numParents = tblinfo[i].numParents;
B
Bruce Momjian 已提交
4114

4115
				appendPQExpBuffer(delq, "DROP TABLE %s;\n", fmtId(tblinfo[i].relname, force_quotes));
B
Bruce Momjian 已提交
4116

4117 4118 4119 4120 4121 4122
				appendPQExpBuffer(q, "CREATE TABLE %s (\n\t", fmtId(tblinfo[i].relname, force_quotes));
				actual_atts = 0;
				for (j = 0; j < tblinfo[i].numatts; j++)
				{
					/* Is this one of the table's own attrs ? */
					if (tblinfo[i].inhAttrs[j] == 0)
4123
					{
4124 4125 4126 4127 4128
						/* Format properly if not first attr */
						if (actual_atts > 0)
							appendPQExpBuffer(q, ",\n\t");

						/* Attr name & type */
4129 4130 4131 4132 4133 4134 4135 4136
						appendPQExpBuffer(q, "%s ", fmtId(tblinfo[i].attnames[j], force_quotes));

						if (g_fout->remoteVersion >= 70100)
						{
							appendPQExpBuffer(q, "%s", tblinfo[i].atttypedefns[j]);
						} else {
							_dumpTableAttr70(fout, tblinfo, i, j, q);
						}
4137 4138

						/* Default value */
4139
						if (tblinfo[i].adef_expr[j] != NULL && tblinfo[i].inhAttrDef[j] == 0)
4140
							appendPQExpBuffer(q, " DEFAULT %s",
B
Bruce Momjian 已提交
4141
											  tblinfo[i].adef_expr[j]);
4142 4143

						/* Not Null constraint */
4144
						if (tblinfo[i].notnull[j] && tblinfo[i].inhNotNull[j] == 0)
4145 4146 4147 4148
							appendPQExpBuffer(q, " NOT NULL");

						actual_atts++;
					}
4149
				}
B
Hi all  
Bruce Momjian 已提交
4150

4151 4152


4153 4154 4155 4156 4157 4158 4159
				/* put the CONSTRAINTS inside the table def */
				for (k = 0; k < tblinfo[i].ncheck; k++)
				{
					if (actual_atts + k > 0)
						appendPQExpBuffer(q, ",\n\t");

					appendPQExpBuffer(q, "%s",
B
Bruce Momjian 已提交
4160
									  tblinfo[i].check_expr[k]);
4161
				}
4162

4163 4164
				/* Primary Key */
				if (tblinfo[i].pkIndexOid != NULL)
4165
				{
B
Bruce Momjian 已提交
4166
					PQExpBuffer consDef;
4167 4168

					/* Find the corresponding index */
4169
					for (k = 0; k < numIndexes; k++)
4170
					{
4171 4172
						if (strcmp(indinfo[k].indexreloid,
								   tblinfo[i].pkIndexOid) == 0)
4173 4174 4175
							break;
					}

4176
					if (k >= numIndexes)
4177
					{
4178 4179
						write_msg(NULL, "dumpTables(): failed sanity check, could not find index (%s) for primary key constraint\n",
								   tblinfo[i].pkIndexOid);
4180
						exit_nicely();
4181 4182 4183 4184
					}

					consDef = getPKconstraint(&tblinfo[i], &indinfo[k]);

B
Bruce Momjian 已提交
4185
					if ((actual_atts + tblinfo[i].ncheck) > 0)
4186
						appendPQExpBuffer(q, ",\n\t");
4187 4188 4189 4190

					appendPQExpBuffer(q, "%s", consDef->data);

					destroyPQExpBuffer(consDef);
4191
				}
B
Bruce Momjian 已提交
4192

4193

4194
				appendPQExpBuffer(q, "\n)");
4195

4196
				if (numParents > 0)
B
Hi all  
Bruce Momjian 已提交
4197
				{
4198
					appendPQExpBuffer(q, "\nINHERITS (");
4199 4200 4201
					for (k = 0; k < numParents; k++)
					{
						appendPQExpBuffer(q, "%s%s",
B
Bruce Momjian 已提交
4202 4203
										  (k > 0) ? ", " : "",
									 fmtId(parentRels[k], force_quotes));
4204 4205
					}
					appendPQExpBuffer(q, ")");
B
Hi all  
Bruce Momjian 已提交
4206 4207
				}

4208 4209 4210
				if (!tblinfo[i].hasoids)
					appendPQExpBuffer(q, " WITHOUT OIDS");

4211 4212
				appendPQExpBuffer(q, ";\n");
			}
B
Bruce Momjian 已提交
4213

B
Bruce Momjian 已提交
4214 4215
			if (!dataOnly)
			{
4216

4217
				ArchiveEntry(fout, objoid, tblinfo[i].relname,
B
Bruce Momjian 已提交
4218 4219
							 reltypename, NULL, q->data, delq->data, "", tblinfo[i].usename,
							 NULL, NULL);
B
Bruce Momjian 已提交
4220

4221 4222 4223 4224
				if (!aclsSkip)
					dumpACL(fout, tblinfo[i]);

			}
4225

4226
			/* Dump Field Comments */
B
Bruce,  
Bruce Momjian 已提交
4227

4228 4229
			for (j = 0; j < tblinfo[i].numatts; j++)
			{
B
Bruce,  
Bruce Momjian 已提交
4230 4231 4232 4233
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "COLUMN %s", fmtId(tblinfo[i].relname, force_quotes));
				appendPQExpBuffer(q, ".");
				appendPQExpBuffer(q, "%s", fmtId(tblinfo[i].attnames[j], force_quotes));
4234 4235
				dumpComment(fout, q->data, tblinfo[i].oid,
							"pg_class", j+1, commentDeps);
B
Bruce,  
Bruce Momjian 已提交
4236
			}
4237

B
Bruce,  
Bruce Momjian 已提交
4238
			/* Dump Table Comments */
4239

B
Bruce,  
Bruce Momjian 已提交
4240
			resetPQExpBuffer(q);
4241
			appendPQExpBuffer(q, "%s %s", reltypename, fmtId(tblinfo[i].relname, force_quotes));
4242 4243
			dumpComment(fout, q->data, tblinfo[i].oid,
						"pg_class", 0, commentDeps);
4244

4245 4246
		}
	}
4247 4248 4249

	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
4250 4251
}

B
Bruce Momjian 已提交
4252 4253
static PQExpBuffer
getPKconstraint(TableInfo *tblInfo, IndInfo *indInfo)
4254
{
B
Bruce Momjian 已提交
4255 4256
	PQExpBuffer pkBuf = createPQExpBuffer();
	int			k;
4257 4258

	appendPQExpBuffer(pkBuf, "Constraint %s Primary Key (",
B
Bruce Momjian 已提交
4259
					  tblInfo->primary_key_name);
4260 4261 4262

	for (k = 0; k < INDEX_MAX_KEYS; k++)
	{
4263 4264
		int			indkey;
		const char *attname;
4265 4266 4267 4268

		indkey = atoi(indInfo->indkey[k]);
		if (indkey == InvalidAttrNumber)
			break;
4269
		attname = getAttrName(indkey, tblInfo);
4270 4271

		appendPQExpBuffer(pkBuf, "%s%s",
B
Bruce Momjian 已提交
4272 4273
						  (k == 0) ? "" : ", ",
						  fmtId(attname, force_quotes));
4274 4275 4276 4277 4278 4279 4280
	}

	appendPQExpBuffer(pkBuf, ")");

	return pkBuf;
}

4281 4282 4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302 4303 4304 4305 4306 4307 4308 4309
/*
 * getAttrName: extract the correct name for an attribute
 *
 * The array tblInfo->attnames[] only provides names of user attributes;
 * if a system attribute number is supplied, we have to fake it.
 * We also do a little bit of bounds checking for safety's sake.
 */
static const char *
getAttrName(int attrnum, TableInfo *tblInfo)
{
	if (attrnum > 0 && attrnum <= tblInfo->numatts)
		return tblInfo->attnames[attrnum-1];
	switch (attrnum)
	{
		case SelfItemPointerAttributeNumber:
			return "ctid";
		case ObjectIdAttributeNumber:
			return "oid";
		case MinTransactionIdAttributeNumber:
			return "xmin";
		case MinCommandIdAttributeNumber:
			return "cmin";
		case MaxTransactionIdAttributeNumber:
			return "xmax";
		case MaxCommandIdAttributeNumber:
			return "cmax";
		case TableOidAttributeNumber:
			return "tableoid";
	}
4310 4311
	write_msg(NULL, "getAttrName(): invalid column number %d for table %s\n",
			   attrnum, tblInfo->relname);
4312
	exit_nicely();
4313 4314 4315
	return NULL;				/* keep compiler quiet */
}

4316
/*
4317
 * dumpIndexes:
4318
 *	  write out to fout all the user-defined indexes
4319
 */
4320
void
4321
dumpIndexes(Archive *fout, IndInfo *indinfo, int numIndexes,
4322
			TableInfo *tblinfo, int numTables, const char *tablename)
4323
{
4324
	int			i;
4325
	int			tableInd;
4326 4327 4328
	PQExpBuffer q = createPQExpBuffer();
	PQExpBuffer delq = createPQExpBuffer();
	PQExpBuffer id1 = createPQExpBuffer();
4329

4330
	for (i = 0; i < numIndexes; i++)
4331
	{
4332 4333 4334 4335
		if (tablename && tablename[0] &&
			(strcmp(indinfo[i].indrelname, tablename) != 0))
			continue;

4336
		tableInd = findTableByName(tblinfo, numTables,
4337 4338 4339
								   indinfo[i].indrelname);
		if (tableInd < 0)
		{
4340 4341
			write_msg(NULL, "dumpIndexes(): failed sanity check, table %s was not found\n",
					   indinfo[i].indrelname);
4342
			exit_nicely();
4343
		}
4344

4345 4346 4347
		/* Handle PK indexes */
		if (strcmp(indinfo[i].indisprimary, "t") == 0)
		{
4348 4349 4350 4351 4352
#if 0
			/* PK: Enable this code when ALTER TABLE supports PK constraints. */

			PQExpBuffer consDef = getPKconstraint(&tblinfo[tableInd], &indinfo[i]);

4353
			resetPQExpBuffer(q);
4354

4355
			appendPQExpBuffer(q, "Alter Table %s Add %s;",
4356 4357 4358 4359
							  fmtId(tblinfo[tableInd].relname, force_quotes),
							  consDef->data);

			ArchiveEntry(fout, indinfo[i].oid, tblinfo[tableInd].primary_key_name,
4360
						 "CONSTRAINT", NULL, q->data, "",
4361 4362 4363 4364
						 "", tblinfo[tableInd].usename, NULL, NULL);

			destroyPQExpBuffer(consDef);
#endif
B
Bruce Momjian 已提交
4365 4366 4367 4368 4369

			/*
			 * Don't need to do anything else for this system-generated
			 * index
			 */
4370 4371 4372
			continue;
		}

4373 4374
		resetPQExpBuffer(id1);
		appendPQExpBuffer(id1, fmtId(indinfo[i].indexrelname, force_quotes));
4375

4376 4377
		resetPQExpBuffer(q);
		appendPQExpBuffer(q, "%s;\n", indinfo[i].indexdef);
B
Bruce,  
Bruce Momjian 已提交
4378

4379 4380
		resetPQExpBuffer(delq);
		appendPQExpBuffer(delq, "DROP INDEX %s;\n", id1->data);
B
Bruce Momjian 已提交
4381

4382 4383 4384 4385 4386 4387 4388 4389
		/*
		 * We make the index belong to the owner of its table, which
		 * is not necessarily right but should answer 99% of the time.
		 * Would have to add owner name to IndInfo to do it right.
		 */
		ArchiveEntry(fout, indinfo[i].indexreloid, id1->data,
					 "INDEX", NULL, q->data, delq->data,
					 "", tblinfo[tableInd].usename, NULL, NULL);
4390

4391 4392 4393 4394 4395
		/* Dump Index Comments */
		resetPQExpBuffer(q);
		appendPQExpBuffer(q, "INDEX %s", id1->data);
		dumpComment(fout, q->data, indinfo[i].indexreloid,
					"pg_class", 0, NULL);
4396
	}
4397

4398 4399 4400
	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
	destroyPQExpBuffer(id1);
4401
}
4402

4403
/*
B
Bruce Momjian 已提交
4404
 * dumpTuples
4405 4406 4407 4408 4409
 *	  prints out the tuples in ASCII representation. The output is a valid
 *	  input to COPY FROM stdin.
 *
 *	  We only need to do this for POSTGRES 4.2 databases since the
 *	  COPY TO statment doesn't escape newlines properly. It's been fixed
4410
 *	  in PostgreSQL.
4411 4412 4413 4414
 *
 * the attrmap passed in tells how to map the attributes copied in to the
 * attributes copied out
 */
4415
#ifdef NOT_USED
4416
void
4417
dumpTuples(PGresult *res, FILE *fout, int *attrmap)
4418
{
4419 4420 4421 4422
	int			j,
				k;
	int			m,
				n;
B
Bruce Momjian 已提交
4423
	char	  **outVals = NULL; /* values to copy out */
4424 4425 4426 4427 4428 4429 4430 4431 4432 4433 4434 4435 4436 4437 4438 4439 4440 4441 4442

	n = PQntuples(res);
	m = PQnfields(res);

	if (m > 0)
	{

		/*
		 * Print out the tuples but only print tuples with at least 1
		 * field.
		 */
		outVals = (char **) malloc(m * sizeof(char *));

		for (j = 0; j < n; j++)
		{
			for (k = 0; k < m; k++)
				outVals[attrmap[k]] = PQgetvalue(res, j, k);
			for (k = 0; k < m; k++)
			{
4443
				char	   *pval = outVals[k];
4444 4445 4446 4447 4448 4449 4450 4451 4452 4453 4454 4455 4456 4457 4458 4459 4460 4461 4462 4463

				if (k != 0)
					fputc('\t', fout);	/* delimiter for attribute */

				if (pval)
				{
					while (*pval != '\0')
					{
						/* escape tabs, newlines and backslashes */
						if (*pval == '\t' || *pval == '\n' || *pval == '\\')
							fputc('\\', fout);
						fputc(*pval, fout);
						pval++;
					}
				}
			}
			fputc('\n', fout);	/* delimiter for a tuple */
		}
		free(outVals);
	}
4464
}
4465

4466
#endif
4467

4468 4469 4470 4471
/*
 * setMaxOid -
 * find the maximum oid and generate a COPY statement to set it
*/
4472

4473
static void
B
Bruce Momjian 已提交
4474
setMaxOid(Archive *fout)
4475
{
B
Bruce Momjian 已提交
4476 4477
	PGresult   *res;
	Oid			max_oid;
B
Bruce Momjian 已提交
4478
	char		sql[1024];
4479

4480
	res = PQexec(g_conn, "CREATE TEMPORARY TABLE pgdump_oid (dummy int4)");
4481 4482 4483
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
4484
		write_msg(NULL, "could not create pgdump_oid table: %s", PQerrorMessage(g_conn));
4485
		exit_nicely();
4486 4487
	}
	PQclear(res);
4488
	res = PQexec(g_conn, "INSERT INTO pgdump_oid VALUES (0)");
4489 4490 4491
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
4492
		write_msg(NULL, "could not insert into pgdump_oid table: %s", PQerrorMessage(g_conn));
4493
		exit_nicely();
4494
	}
4495
	max_oid = PQoidValue(res);
4496 4497
	if (max_oid == 0)
	{
4498
		write_msg(NULL, "inserted invalid oid\n");
4499
		exit_nicely();
4500 4501
	}
	PQclear(res);
4502
	res = PQexec(g_conn, "DROP TABLE pgdump_oid;");
4503 4504 4505
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
4506
		write_msg(NULL, "could not drop pgdump_oid table: %s", PQerrorMessage(g_conn));
4507
		exit_nicely();
4508 4509 4510
	}
	PQclear(res);
	if (g_verbose)
4511
		write_msg(NULL, "maximum system oid is %u\n", max_oid);
4512 4513 4514
	snprintf(sql, 1024,
			 "CREATE TEMPORARY TABLE pgdump_oid (dummy int4);\n"
			 "COPY pgdump_oid WITH OIDS FROM stdin;\n"
4515
			 "%u\t0\n"
4516 4517 4518
			 "\\.\n"
			 "DROP TABLE pgdump_oid;\n",
			 max_oid);
B
Bruce Momjian 已提交
4519

4520
	ArchiveEntry(fout, "0", "Max OID", "<Init>", NULL, sql, "", "", "", NULL, NULL);
4521
}
4522 4523 4524

/*
 * findLastBuiltInOid -
4525
 * find the last built in oid
4526 4527
 * we do this by retrieving datlastsysoid from the pg_database entry for this database,
 */
4528

4529
static Oid
4530
findLastBuiltinOid_V71(const char *dbname)
4531
{
B
Bruce Momjian 已提交
4532
	PGresult   *res;
4533
	int			ntups;
4534
	Oid			last_oid;
4535 4536 4537
	PQExpBuffer query = createPQExpBuffer();

	resetPQExpBuffer(query);
4538
	appendPQExpBuffer(query, "SELECT datlastsysoid from pg_database where datname = ");
4539
	formatStringLiteral(query, dbname, CONV_ALL);
4540

4541
	res = PQexec(g_conn, query->data);
4542 4543 4544
	if (res == NULL ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
4545
		write_msg(NULL, "error in finding the last system oid: %s", PQerrorMessage(g_conn));
4546
		exit_nicely();
4547 4548
	}
	ntups = PQntuples(res);
4549
	if (ntups < 1)
4550
	{
4551
		write_msg(NULL, "missing pg_database entry for this database\n");
4552
		exit_nicely();
4553 4554 4555
	}
	if (ntups > 1)
	{
4556
		write_msg(NULL, "found more than one pg_database entry for this database\n");
4557
		exit_nicely();
4558
	}
4559
	last_oid = atooid(PQgetvalue(res, 0, PQfnumber(res, "datlastsysoid")));
4560
	PQclear(res);
4561
	destroyPQExpBuffer(query);
4562
	return last_oid;
4563 4564
}

4565 4566 4567 4568 4569 4570 4571
/*
 * findLastBuiltInOid -
 * find the last built in oid
 * we do this by looking up the oid of 'template1' in pg_database,
 * this is probably not foolproof but comes close
*/

4572
static Oid
4573 4574 4575 4576 4577 4578 4579 4580 4581 4582 4583
findLastBuiltinOid_V70(void)
{
	PGresult   *res;
	int			ntups;
	int			last_oid;

	res = PQexec(g_conn,
			  "SELECT oid from pg_database where datname = 'template1'");
	if (res == NULL ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
4584
		write_msg(NULL, "error in finding the template1 database: %s", PQerrorMessage(g_conn));
4585
		exit_nicely();
4586 4587 4588 4589
	}
	ntups = PQntuples(res);
	if (ntups < 1)
	{
4590
		write_msg(NULL, "could not find template1 database entry in the pg_database table\n");
4591
		exit_nicely();
4592 4593 4594
	}
	if (ntups > 1)
	{
4595
		write_msg(NULL, "found more than one template1 database entry in the pg_database table\n");
4596
		exit_nicely();
4597 4598 4599 4600 4601
	}
	last_oid = atooid(PQgetvalue(res, 0, PQfnumber(res, "oid")));
	PQclear(res);
	return last_oid;
}
4602

4603
static void
4604
dumpSequence(Archive *fout, TableInfo tbinfo, const bool schemaOnly, const bool dataOnly)
4605
{
4606
	PGresult   *res;
4607 4608 4609 4610 4611 4612
	char	   *last,
			   *incby,
			   *maxv,
			   *minv,
			   *cache;
	bool		cycled,
B
Bruce Momjian 已提交
4613
				called;
4614
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
4615
	PQExpBuffer delqry = createPQExpBuffer();
4616

B
Hi, all  
Bruce Momjian 已提交
4617
	appendPQExpBuffer(query,
4618
			"SELECT sequence_name, last_value, increment_by, max_value, "
4619 4620
				  "min_value, cache_value, is_cycled, is_called from %s",
					  fmtId(tbinfo.relname, force_quotes));
4621

B
Hi, all  
Bruce Momjian 已提交
4622
	res = PQexec(g_conn, query->data);
4623 4624
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
4625
		write_msg(NULL, "query to get data of sequence \"%s\" failed: %s", tbinfo.relname, PQerrorMessage(g_conn));
4626
		exit_nicely();
4627 4628 4629 4630
	}

	if (PQntuples(res) != 1)
	{
4631 4632
		write_msg(NULL, "query to get data of sequence \"%s\" returned %d rows (expected 1)\n",
				   tbinfo.relname, PQntuples(res));
4633
		exit_nicely();
4634 4635 4636 4637
	}

	if (strcmp(PQgetvalue(res, 0, 0), tbinfo.relname) != 0)
	{
4638 4639
		write_msg(NULL, "query to get data of sequence \"%s\" returned name \"%s\"\n",
				   tbinfo.relname, PQgetvalue(res, 0, 0));
4640
		exit_nicely();
4641 4642
	}

4643 4644 4645 4646 4647 4648 4649
	last = PQgetvalue(res, 0, 1);
	incby = PQgetvalue(res, 0, 2);
	maxv = PQgetvalue(res, 0, 3);
	minv = PQgetvalue(res, 0, 4);
	cache = PQgetvalue(res, 0, 5);
	cycled = (strcmp(PQgetvalue(res, 0, 6), "t") == 0);
	called = (strcmp(PQgetvalue(res, 0, 7), "t") == 0);
4650

4651
	/*
B
Bruce Momjian 已提交
4652 4653
	 * The logic we use for restoring sequences is as follows: -   Add a
	 * basic CREATE SEQUENCE statement (use last_val for start if called
4654
	 * is false, else use min_val for start_val).
4655 4656 4657
	 *
	 *	Add a 'SETVAL(seq, last_val, iscalled)' at restore-time iff
	 *  we load data
4658
	 */
4659

4660 4661 4662
	if (!dataOnly)
	{
		resetPQExpBuffer(delqry);
4663 4664
		appendPQExpBuffer(delqry, "DROP SEQUENCE %s;\n",
						  fmtId(tbinfo.relname, force_quotes));
4665

4666 4667
		resetPQExpBuffer(query);
		appendPQExpBuffer(query,
4668 4669
						  "CREATE SEQUENCE %s start %s increment %s "
						  "maxvalue %s minvalue %s cache %s%s;\n",
B
Bruce Momjian 已提交
4670
						  fmtId(tbinfo.relname, force_quotes),
4671
						  (called ? minv : last),
4672
						  incby, maxv, minv, cache,
4673
						  (cycled ? " cycle" : ""));
4674

P
Philip Warner 已提交
4675
		ArchiveEntry(fout, tbinfo.oid, tbinfo.relname, "SEQUENCE", NULL,
4676 4677
					 query->data, delqry->data, "", tbinfo.usename,
					 NULL, NULL);
4678
	}
4679

4680 4681 4682 4683 4684
	if (!schemaOnly)
	{
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "SELECT setval (");
		formatStringLiteral(query, fmtId(tbinfo.relname, force_quotes), CONV_ALL);
4685 4686
		appendPQExpBuffer(query, ", %s, %s);\n",
						  last, (called ? "true" : "false"));
4687

P
Philip Warner 已提交
4688
		ArchiveEntry(fout, tbinfo.oid, tbinfo.relname, "SEQUENCE SET", NULL,
B
Bruce Momjian 已提交
4689
					 query->data, "" /* Del */ , "", "", NULL, NULL);
4690
	}
B
Bruce,  
Bruce Momjian 已提交
4691

4692 4693 4694
	if (!dataOnly)
	{
		/* Dump Sequence Comments */
B
Bruce,  
Bruce Momjian 已提交
4695

4696 4697
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "SEQUENCE %s", fmtId(tbinfo.relname, force_quotes));
4698 4699
		dumpComment(fout, query->data, tbinfo.oid,
					"pg_class", 0, NULL);
4700
	}
4701

4702 4703
	PQclear(res);

4704 4705
	destroyPQExpBuffer(query);
	destroyPQExpBuffer(delqry);
4706
}
V
Vadim B. Mikheev 已提交
4707 4708


4709
static void
B
Bruce Momjian 已提交
4710
dumpTriggers(Archive *fout, const char *tablename,
4711
			 TableInfo *tblinfo, int numTables)
V
Vadim B. Mikheev 已提交
4712
{
4713 4714 4715
	int			i,
				j;

V
Vadim B. Mikheev 已提交
4716
	if (g_verbose)
4717
		write_msg(NULL, "dumping out triggers\n");
4718

V
Vadim B. Mikheev 已提交
4719 4720
	for (i = 0; i < numTables; i++)
	{
B
Bruce Momjian 已提交
4721
		if (tablename && (strcmp(tblinfo[i].relname, tablename) != 0) && (strlen(tablename) > 0))
V
Vadim B. Mikheev 已提交
4722
			continue;
4723

V
Vadim B. Mikheev 已提交
4724 4725
		for (j = 0; j < tblinfo[i].ntrig; j++)
		{
B
Bruce Momjian 已提交
4726
			ArchiveEntry(fout, tblinfo[i].triggers[j].oid, tblinfo[i].triggers[j].tgname,
B
Bruce Momjian 已提交
4727 4728
				   "TRIGGER", NULL, tblinfo[i].triggers[j].tgsrc, "", "",
						 tblinfo[i].usename, NULL, NULL);
4729 4730
			dumpComment(fout, tblinfo[i].triggers[j].tgcomment, tblinfo[i].triggers[j].oid,
						"pg_trigger", 0, NULL);
V
Vadim B. Mikheev 已提交
4731 4732 4733
		}
	}
}
4734 4735


4736
static void
B
Bruce Momjian 已提交
4737
dumpRules(Archive *fout, const char *tablename,
B
Bruce Momjian 已提交
4738
		  TableInfo *tblinfo, int numTables)
4739
{
B
Bruce Momjian 已提交
4740 4741 4742 4743
	PGresult   *res;
	int			nrules;
	int			i,
				t;
4744
	PQExpBuffer query = createPQExpBuffer();
4745

B
Bruce Momjian 已提交
4746
	int			i_definition;
4747
	int			i_oid;
4748
	int			i_owner;
4749
	int			i_rulename;
4750 4751

	if (g_verbose)
4752
		write_msg(NULL, "dumping out rules\n");
4753 4754 4755 4756 4757 4758

	/*
	 * For each table we dump
	 */
	for (t = 0; t < numTables; t++)
	{
B
Bruce Momjian 已提交
4759
		if (tablename && (strcmp(tblinfo[t].relname, tablename) != 0) && (strlen(tablename) > 0))
4760 4761 4762
			continue;

		/*
B
Bruce Momjian 已提交
4763 4764
		 * Get all rules defined for this table We include pg_rules in the
		 * cross since it filters out all view rules (pjw 15-Sep-2000).
4765 4766
		 *
		 * XXXX: Use LOJ here
4767
		 */
B
Hi, all  
Bruce Momjian 已提交
4768
		resetPQExpBuffer(query);
4769
		appendPQExpBuffer(query, "SELECT definition,"
4770
						  "   (select usename from pg_user where pg_class.relowner = usesysid) AS viewowner, "
4771 4772
						  "   pg_rewrite.oid, pg_rewrite.rulename "
						  "FROM pg_rewrite, pg_class, pg_rules "
4773
						  "WHERE pg_class.relname = ");
4774
		formatStringLiteral(query, tblinfo[t].relname, CONV_ALL);
4775
		appendPQExpBuffer(query,
4776 4777
						  "    AND pg_rewrite.ev_class = pg_class.oid "
						  "    AND pg_rules.tablename = pg_class.relname "
B
Bruce Momjian 已提交
4778
					   "    AND pg_rules.rulename = pg_rewrite.rulename "
4779
						  "ORDER BY pg_rewrite.oid");
B
Hi, all  
Bruce Momjian 已提交
4780
		res = PQexec(g_conn, query->data);
4781 4782 4783
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
4784 4785
			write_msg(NULL, "query to get rules associated with table \"%s\" failed: %s",
					   tblinfo[t].relname, PQerrorMessage(g_conn));
4786
			exit_nicely();
4787 4788 4789 4790
		}

		nrules = PQntuples(res);
		i_definition = PQfnumber(res, "definition");
4791
		i_owner = PQfnumber(res, "viewowner");
B
Bruce,  
Bruce Momjian 已提交
4792 4793
		i_oid = PQfnumber(res, "oid");
		i_rulename = PQfnumber(res, "rulename");
4794 4795 4796 4797

		/*
		 * Dump them out
		 */
B
Bruce,  
Bruce Momjian 已提交
4798

4799 4800
		for (i = 0; i < nrules; i++)
		{
B
Bruce Momjian 已提交
4801
			ArchiveEntry(fout, PQgetvalue(res, i, i_oid), PQgetvalue(res, i, i_rulename),
B
Bruce Momjian 已提交
4802 4803
						 "RULE", NULL, PQgetvalue(res, i, i_definition),
						 "", "", PQgetvalue(res, i, i_owner), NULL, NULL);
4804

B
Bruce,  
Bruce Momjian 已提交
4805 4806 4807 4808
			/* Dump rule comments */

			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "RULE %s", fmtId(PQgetvalue(res, i, i_rulename), force_quotes));
4809 4810
			dumpComment(fout, query->data, PQgetvalue(res, i, i_oid),
						"pg_rewrite", 0, NULL);
4811

B
Bruce,  
Bruce Momjian 已提交
4812 4813
		}

4814 4815
		PQclear(res);
	}
4816 4817

	destroyPQExpBuffer(query);
4818
}