pg_dump.c 131.9 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.222 2001/08/16 20:38:54 tgl 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 144 145 146 147
	puts(gettext(
		"  -a, --data-only          dump only the data, not the schema\n"
		"  -b, --blobs              include BLOB data in dump\n"
		"  -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
		"  -Z, --compress {0-9}     compression level for compressed formats\n"
		));
B
Hi,  
Bruce Momjian 已提交
172
#else
173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
	puts(gettext(
		"  -a                       dump only the data, not the schema\n"
		"  -b                       include BLOB data in dump\n"
		"  -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"
		"  -Z {0-9}                 compression level for compressed formats\n"
		));
B
Hi,  
Bruce Momjian 已提交
203
#endif
204 205 206
	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>."));
207 208
}

209

210 211
void
exit_nicely(void)
212
{
213 214 215
	PQfinish(g_conn);
	if (g_verbose)
		write_msg(NULL, "*** aborted because of error\n");
216
	exit(1);
217 218 219
}


220 221
#define COPYBUFSIZ		8192

B
Bruce Momjian 已提交
222 223 224 225 226
/*
 *	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.
 */
227

B
Bruce Momjian 已提交
228 229
static int
dumpClasses_nodumpData(Archive *fout, char *oid, void *dctxv)
230
{
B
Bruce Momjian 已提交
231 232
	const DumpContext *dctx = (DumpContext *) dctxv;
	const char *classname = dctx->tblinfo[dctx->tblidx].relname;
233
	const bool	hasoids = dctx->tblinfo[dctx->tblidx].hasoids;
B
Bruce Momjian 已提交
234
	const bool	oids = dctx->oids;
235

236 237 238 239 240
	PGresult   *res;
	char		query[255];
	int			ret;
	bool		copydone;
	char		copybuf[COPYBUFSIZ];
241

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

245
	if (oids && hasoids)
246
	{
B
Bruce Momjian 已提交
247 248

		/*
249
		 * archprintf(fout, "COPY %s WITH OIDS FROM stdin;\n",
B
Bruce Momjian 已提交
250
		 * fmtId(classname, force_quotes));
251
		 *
B
Bruce Momjian 已提交
252
		 * - Not used as of V1.3 (needs to be in ArchiveEntry call)
253 254 255
		 *
		 */

256
		sprintf(query, "COPY %s WITH OIDS TO stdout;",
257
				fmtId(classname, force_quotes));
258 259 260
	}
	else
	{
B
Bruce Momjian 已提交
261 262 263 264

		/*
		 * archprintf(fout, "COPY %s FROM stdin;\n", fmtId(classname,
		 * force_quotes));
265 266 267 268 269
		 *
		 * - Not used as of V1.3 (needs to be in ArchiveEntry call)
		 *
		 */

270
		sprintf(query, "COPY %s TO stdout;", fmtId(classname, force_quotes));
271 272
	}
	res = PQexec(g_conn, query);
273 274
	if (!res ||
		PQresultStatus(res) == PGRES_FATAL_ERROR)
275
	{
276 277 278 279
		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);
280
		exit_nicely();
281 282 283 284 285
	}
	else
	{
		if (PQresultStatus(res) != PGRES_COPY_OUT)
		{
286 287 288 289 290
			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);
291
			exit_nicely();
292 293 294 295
		}
		else
		{
			copydone = false;
296

297 298
			while (!copydone)
			{
299
				ret = PQgetline(g_conn, copybuf, COPYBUFSIZ);
300 301 302 303 304 305 306 307 308

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

B
Bruce Momjian 已提交
323
				/*
324 325
				 * THROTTLE:
				 *
B
Bruce Momjian 已提交
326 327 328 329 330
				 * 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.
331
				 *
B
Bruce Momjian 已提交
332 333 334 335 336
				 * 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:
337
				 *
B
Bruce Momjian 已提交
338 339 340
				 * 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
341 342
				 * EndIf
				 *
B
Bruce Momjian 已提交
343 344
				 * where the throttle value was the number of ms to sleep per
				 * ms of work. The calculation was done in each loop.
345
				 *
B
Bruce Momjian 已提交
346 347 348 349 350
				 * 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.
351 352 353
				 *
				 * Further discussion ensued, and the proposal was dropped.
				 *
B
Bruce Momjian 已提交
354 355 356 357 358 359
				 * 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.
360
				 *
B
Bruce Momjian 已提交
361
				 * select(0, NULL, NULL, NULL, &tvi);
362
				 *
B
Bruce Momjian 已提交
363 364 365 366
				 * This will return after the interval specified in the
				 * structure tvi. Fianally, call gettimeofday again to
				 * save the 'last sleep time'.
				 */
367
			}
B
Bruce Momjian 已提交
368
			archprintf(fout, "\\.\n");
369
		}
370
		ret = PQendcopy(g_conn);
371 372
		if (ret != 0)
		{
373 374 375
			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 已提交
376
			PQclear(res);
377
			exit_nicely();
378 379
		}
	}
380

B
Bruce Momjian 已提交
381
	return 1;
382 383
}

B
Bruce Momjian 已提交
384
static int
B
Bruce Momjian 已提交
385
dumpClasses_dumpData(Archive *fout, char *oid, void *dctxv)
386
{
B
Bruce Momjian 已提交
387 388
	const DumpContext *dctx = (DumpContext *) dctxv;
	const char *classname = dctx->tblinfo[dctx->tblidx].relname;
B
Bruce Momjian 已提交
389

390 391
	PGresult   *res;
	PQExpBuffer q = createPQExpBuffer();
392 393
	int			tuple;
	int			field;
394

395 396 397 398 399 400 401
	if (fout->remoteVersion >= 70100)
	{
		appendPQExpBuffer(q, "SELECT * FROM ONLY %s", fmtId(classname, force_quotes));
	} else {
		appendPQExpBuffer(q, "SELECT * FROM %s", fmtId(classname, force_quotes));
	}

B
Hi, all  
Bruce Momjian 已提交
402
	res = PQexec(g_conn, q->data);
403 404 405
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
406 407 408
		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);
409
		exit_nicely();
410
	}
411
	for (tuple = 0; tuple < PQntuples(res); tuple++)
412
	{
B
Bruce Momjian 已提交
413
		archprintf(fout, "INSERT INTO %s ", fmtId(classname, force_quotes));
414
		if (attrNames == true)
415
		{
B
Hi, all  
Bruce Momjian 已提交
416 417
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "(");
418
			for (field = 0; field < PQnfields(res); field++)
419
			{
420
				if (field > 0)
B
Hi, all  
Bruce Momjian 已提交
421 422
					appendPQExpBuffer(q, ",");
				appendPQExpBuffer(q, fmtId(PQfname(res, field), force_quotes));
423
			}
B
Hi, all  
Bruce Momjian 已提交
424
			appendPQExpBuffer(q, ") ");
B
Bruce Momjian 已提交
425
			archprintf(fout, "%s", q->data);
426
		}
B
Bruce Momjian 已提交
427
		archprintf(fout, "VALUES (");
428
		for (field = 0; field < PQnfields(res); field++)
429
		{
430
			if (field > 0)
B
Bruce Momjian 已提交
431
				archprintf(fout, ",");
432
			if (PQgetisnull(res, tuple, field))
433
			{
B
Bruce Momjian 已提交
434
				archprintf(fout, "NULL");
435 436 437
				continue;
			}
			switch (PQftype(res, field))
438
			{
439 440
				case INT2OID:
				case INT4OID:
441
				case OIDOID:	/* int types */
442
				case FLOAT4OID:
443
				case FLOAT8OID:/* float types */
444
					/* These types are printed without quotes */
B
Bruce Momjian 已提交
445
					archprintf(fout, "%s",
B
Bruce Momjian 已提交
446
							   PQgetvalue(res, tuple, field));
447
					break;
448
				case BITOID:
449 450 451 452
				case VARBITOID:
					archprintf(fout, "B'%s'",
							   PQgetvalue(res, tuple, field));
					break;
453
				default:
454

455 456
					/*
					 * All other types are printed as string literals,
457
					 * with appropriate escaping of special characters.
458
					 */
459
					resetPQExpBuffer(q);
460
					formatStringLiteral(q, PQgetvalue(res, tuple, field), CONV_ALL);
461
					archprintf(fout, "%s", q->data);
462
					break;
463
			}
464
		}
B
Bruce Momjian 已提交
465
		archprintf(fout, ");\n");
466 467
	}
	PQclear(res);
468
	destroyPQExpBuffer(q);
B
Bruce Momjian 已提交
469
	return 1;
470 471
}

472 473 474 475 476 477 478 479
/*
 * 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
480
formatStringLiteral(PQExpBuffer buf, const char *str, const formatLiteralOptions opts)
481 482 483 484
{
	appendPQExpBufferChar(buf, '\'');
	while (*str)
	{
B
Bruce Momjian 已提交
485
		char		ch = *str++;
486 487 488

		if (ch == '\\' || ch == '\'')
		{
B
Bruce Momjian 已提交
489
			appendPQExpBufferChar(buf, ch);		/* double these */
490 491 492
			appendPQExpBufferChar(buf, ch);
		}
		else if ((unsigned char) ch < (unsigned char) ' ' &&
B
Bruce Momjian 已提交
493 494 495
				 (opts == CONV_ALL
				  || (ch != '\n' && ch != '\t')
				  ))
496
		{
B
Bruce Momjian 已提交
497 498 499 500 501

			/*
			 * generate octal escape for control chars other than
			 * whitespace
			 */
502 503 504 505 506 507 508 509 510 511 512
			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, '\'');
}

513 514
/*
 * DumpClasses -
515
 *	  dump the contents of all the classes.
516 517
 */
static void
B
Bruce Momjian 已提交
518
dumpClasses(const TableInfo *tblinfo, const int numTables, Archive *fout,
B
Bruce Momjian 已提交
519
		 const char *onlytable, const bool oids, const bool force_quotes)
520
{
B
Bruce Momjian 已提交
521 522 523 524 525 526
	int			i;
	DataDumperPtr dumpFn;
	DumpContext *dumpCtx;
	char		copyBuf[512];
	char	   *copyStmt;

527
	if (g_verbose)
528 529 530 531 532 533
	{
		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");
	}
534

535 536
	for (i = 0; i < numTables; i++)
	{
537
		const char *classname = tblinfo[i].relname;
538 539

		/* Skip VIEW relations */
540
		if (tblinfo[i].viewdef != NULL)
541
			continue;
542 543

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

546
		if (!onlytable || (strcmp(classname, onlytable) == 0) || (strlen(onlytable) == 0))
547 548
		{
			if (g_verbose)
549
				write_msg(NULL, "preparing to dump the contents of table %s\n", classname);
550

551 552 553
#if 0
			becomeUser(fout, tblinfo[i].usename);
#endif
B
Bruce Momjian 已提交
554

B
Bruce Momjian 已提交
555 556
			dumpCtx = (DumpContext *) malloc(sizeof(DumpContext));
			dumpCtx->tblinfo = (TableInfo *) tblinfo;
B
Bruce Momjian 已提交
557 558
			dumpCtx->tblidx = i;
			dumpCtx->oids = oids;
559

560
			if (!dumpData)
561
			{
562
				/* Dump/restore using COPY */
B
Bruce Momjian 已提交
563
				dumpFn = dumpClasses_nodumpData;
564 565 566
				sprintf(copyBuf, "COPY %s %sFROM stdin;\n",
						fmtId(tblinfo[i].relname, force_quotes),
						(oids && tblinfo[i].hasoids) ? "WITH OIDS " : "");
567 568
				copyStmt = copyBuf;
			}
B
Bruce Momjian 已提交
569
			else
570
			{
571
				/* Restore using INSERT */
B
Bruce Momjian 已提交
572
				dumpFn = dumpClasses_dumpData;
573 574
				copyStmt = NULL;
			}
B
Bruce Momjian 已提交
575

P
Philip Warner 已提交
576
			ArchiveEntry(fout, tblinfo[i].oid, tblinfo[i].relname,
577 578
						 "TABLE DATA", NULL, "", "",
						 copyStmt, tblinfo[i].usename,
B
Bruce Momjian 已提交
579
						 dumpFn, dumpCtx);
580
		}
581
	}
582 583
}

584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607


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;
}



608
int
609
main(int argc, char **argv)
610
{
B
Bruce Momjian 已提交
611 612
	int			c;
	const char *filename = NULL;
B
Bruce Momjian 已提交
613
	const char *format = "p";
B
Bruce Momjian 已提交
614 615 616
	const char *dbname = NULL;
	const char *pghost = NULL;
	const char *pgport = NULL;
617
	const char *username = NULL;
B
Bruce Momjian 已提交
618
	char	   *tablename = NULL;
619
	bool		oids = false;
B
Bruce Momjian 已提交
620 621
	TableInfo  *tblinfo;
	int			numTables;
622
	bool		force_password = false;
B
Bruce Momjian 已提交
623
	int			compressLevel = -1;
624
	bool		ignore_version = false;
625 626
	int			plainText = 0;
	int			outputClean = 0;
627
	int			outputCreate = 0;
628
	int			outputBlobs = 0;
629 630
	int			outputNoOwner = 0;
	int			outputNoReconnect = 0;
B
Bruce Momjian 已提交
631
	char	   *outputSuperuser = NULL;
632

B
Bruce Momjian 已提交
633
	RestoreOptions *ropt;
634

B
Hi,  
Bruce Momjian 已提交
635 636 637
#ifdef HAVE_GETOPT_LONG
	static struct option long_options[] = {
		{"data-only", no_argument, NULL, 'a'},
B
Bruce Momjian 已提交
638
		{"blobs", no_argument, NULL, 'b'},
B
Hi,  
Bruce Momjian 已提交
639
		{"clean", no_argument, NULL, 'c'},
640
		{"create", no_argument, NULL, 'C'},
B
Bruce Momjian 已提交
641 642
		{"file", required_argument, NULL, 'f'},
		{"format", required_argument, NULL, 'F'},
643
		{"inserts", no_argument, NULL, 'd'},
644
		{"attribute-inserts", no_argument, NULL, 'D'},
645
		{"column-inserts", no_argument, NULL, 'D'},
B
Hi,  
Bruce Momjian 已提交
646
		{"host", required_argument, NULL, 'h'},
647
		{"ignore-version", no_argument, NULL, 'i'},
648
		{"no-reconnect", no_argument, NULL, 'R'},
B
Hi,  
Bruce Momjian 已提交
649 650
		{"no-quotes", no_argument, NULL, 'n'},
		{"quotes", no_argument, NULL, 'N'},
651
		{"oids", no_argument, NULL, 'o'},
652
		{"no-owner", no_argument, NULL, 'O'},
B
Hi,  
Bruce Momjian 已提交
653 654
		{"port", required_argument, NULL, 'p'},
		{"schema-only", no_argument, NULL, 's'},
655
		{"superuser", required_argument, NULL, 'S'},
B
Hi,  
Bruce Momjian 已提交
656
		{"table", required_argument, NULL, 't'},
657 658
		{"password", no_argument, NULL, 'W'},
		{"username", required_argument, NULL, 'U'},
B
Hi,  
Bruce Momjian 已提交
659
		{"verbose", no_argument, NULL, 'v'},
660
		{"no-privileges", no_argument, NULL, 'x'},
B
Hi,  
Bruce Momjian 已提交
661
		{"no-acl", no_argument, NULL, 'x'},
B
Bruce Momjian 已提交
662
		{"compress", required_argument, NULL, 'Z'},
B
Hi,  
Bruce Momjian 已提交
663
		{"help", no_argument, NULL, '?'},
664
		{"version", no_argument, NULL, 'V'}
665
	};
666 667
	int			optindex;

B
Hi,  
Bruce Momjian 已提交
668 669
#endif

670 671 672 673 674 675
#ifdef ENABLE_NLS
	setlocale(LC_ALL, "");
	bindtextdomain("pg_dump", LOCALEDIR);
	textdomain("pg_dump");
#endif

676
	g_verbose = false;
677
	force_quotes = true;
678 679 680 681 682

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

683
	dataOnly = schemaOnly = dumpData = attrNames = false;
684

685
	if (!strrchr(argv[0], '/'))
686 687
		progname = argv[0];
	else
688
		progname = strrchr(argv[0], '/') + 1;
689

690 691 692 693
	/* Set defaulty options based on progname */
	if (strcmp(progname, "pg_backup") == 0)
	{
		format = "c";
P
Philip Warner 已提交
694
		outputBlobs = true;
695
	}
696

697 698
	if (argc > 1)
	{
B
Bruce Momjian 已提交
699
		if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
700 701 702 703
		{
			help(progname);
			exit(0);
		}
B
Bruce Momjian 已提交
704
		if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
705
		{
706
			puts("pg_dump (PostgreSQL) " PG_VERSION);
707 708 709 710
			exit(0);
		}
	}

B
Hi,  
Bruce Momjian 已提交
711
#ifdef HAVE_GETOPT_LONG
712
	while ((c = getopt_long(argc, argv, "abcCdDf:F:h:inNoOp:RsS:t:uU:vWxzZ:V?", long_options, &optindex)) != -1)
B
Hi,  
Bruce Momjian 已提交
713
#else
714
	while ((c = getopt(argc, argv, "abcCdDf:F:h:inNoOp:RsS:t:uU:vWxzZ:V?-")) != -1)
B
Hi,  
Bruce Momjian 已提交
715
#endif
716

717 718 719
	{
		switch (c)
		{
720
			case 'a':			/* Dump data only */
721
				dataOnly = true;
722
				break;
723

724 725 726 727
			case 'b':			/* Dump blobs */
				outputBlobs = true;
				break;

B
Bruce Momjian 已提交
728
			case 'c':			/* clean (i.e., drop) schema prior to
B
Bruce Momjian 已提交
729
								 * create */
B
Bruce Momjian 已提交
730
				outputClean = 1;
B
Bruce Momjian 已提交
731 732
				break;

733 734 735 736 737
			case 'C':			/* Create DB */

				outputCreate = 1;
				break;

738
			case 'd':			/* dump data as proper insert strings */
739
				dumpData = true;
740
				break;
741

742 743
			case 'D':			/* dump data as proper insert strings with
								 * attr names */
744 745
				dumpData = true;
				attrNames = true;
746
				break;
747

748
			case 'f':
749 750
				filename = optarg;
				break;
751

B
Bruce Momjian 已提交
752 753 754
			case 'F':
				format = optarg;
				break;
755

B
Bruce Momjian 已提交
756
			case 'h':			/* server host */
757 758
				pghost = optarg;
				break;
759

760 761 762
			case 'i':			/* ignore database version mismatch */
				ignore_version = true;
				break;
763

B
Bruce Momjian 已提交
764 765
			case 'n':			/* Do not force double-quotes on
								 * identifiers */
766
				force_quotes = false;
767
				break;
768

769
			case 'N':			/* Force double-quotes on identifiers */
770
				force_quotes = true;
771
				break;
772

773
			case 'o':			/* Dump oids */
774
				oids = true;
775
				break;
776 777


778 779 780
			case 'O':			/* Don't reconnect to match owner */
				outputNoOwner = 1;
				break;
781

782 783 784
			case 'p':			/* server port */
				pgport = optarg;
				break;
785

786 787 788
			case 'R':			/* No reconnect */
				outputNoReconnect = 1;
				break;
789

790
			case 's':			/* dump schema only */
791
				schemaOnly = true;
792
				break;
793

B
Bruce Momjian 已提交
794 795
			case 'S':			/* Username for superuser in plain text
								 * output */
796 797
				outputSuperuser = strdup(optarg);
				break;
798

799
			case 't':			/* Dump data for this table only */
800
				{
801
					int			i;
802 803

					tablename = strdup(optarg);
B
Bruce Momjian 已提交
804 805 806 807 808

					/*
					 * quoted string? Then strip quotes and preserve
					 * case...
					 */
809 810 811
					if (tablename[0] == '"')
					{
						strcpy(tablename, &tablename[1]);
B
Bruce Momjian 已提交
812 813
						if (*(tablename + strlen(tablename) - 1) == '"')
							*(tablename + strlen(tablename) - 1) = '\0';
814 815 816 817 818
					}
					/* otherwise, convert table name to lowercase... */
					else
					{
						for (i = 0; tablename[i]; i++)
819 820
							if (isupper((unsigned char) tablename[i]))
								tablename[i] = tolower((unsigned char) tablename[i]);
821

B
Bruce Momjian 已提交
822 823 824 825 826
						/*
						 * '*' is a special case meaning ALL tables, but
						 * only if unquoted
						 */
						if (strcmp(tablename, "*") == 0)
827 828
							tablename[0] = '\0';

829
					}
830
				}
831
				break;
832

833
			case 'u':
834
				force_password = true;
835
				username = simple_prompt("User name: ", 100, true);
836 837 838 839
				break;

			case 'U':
				username = optarg;
840
				break;
841

842 843
			case 'v':			/* verbose */
				g_verbose = true;
844
				break;
845

846 847 848 849
			case 'W':
				force_password = true;
				break;

850 851
			case 'x':			/* skip ACL dump */
				aclsSkip = true;
852
				break;
853

B
Bruce Momjian 已提交
854 855 856
			case 'Z':			/* Compression Level */
				compressLevel = atoi(optarg);
				break;
857

858 859
#ifndef HAVE_GETOPT_LONG
			case '-':
860 861 862 863
				fprintf(stderr,
						gettext("%s was compiled without support for long options.\n"
								"Use --help for help on invocation options.\n"),
						progname);
864 865 866
				exit(1);
				break;
#endif
867
			default:
868
				fprintf(stderr, gettext("Try '%s --help' for more information.\n"), progname);
869
				exit(1);
870 871 872
		}
	}

B
Bruce Momjian 已提交
873 874
	if (optind < (argc - 1))
	{
P
Philip Warner 已提交
875
		fprintf(stderr,
876 877 878
				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 已提交
879 880 881
		exit(1);
	}

882 883 884 885 886 887 888
	/* Get the target database name */
	if (optind < argc)
		dbname = argv[optind];
	else
		dbname = getenv("PGDATABASE");
	if (!dbname)
	{
889
		write_msg(NULL, "no database name specified\n");
890 891 892
		exit(1);
	}

B
Bruce Momjian 已提交
893 894
	if (dataOnly && schemaOnly)
	{
895
		write_msg(NULL, "The options \"schema only\" (-s) and \"data only\" (-a) cannot be used together.\n");
B
Bruce Momjian 已提交
896 897 898
		exit(1);
	}

B
Bruce Momjian 已提交
899
	if (outputBlobs && tablename != NULL && strlen(tablename) > 0)
900
	{
901 902
		write_msg(NULL, "BLOB output is not supported for a single table.\n");
		write_msg(NULL, "Use all tables or a full dump instead.\n");
903 904 905
		exit(1);
	}

906 907
	if (dumpData == true && oids == true)
	{
908 909
		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");
910
		exit(1);
911
	}
912

B
Bruce Momjian 已提交
913
	if (outputBlobs == true && (format[0] == 'p' || format[0] == 'P'))
P
Philip Warner 已提交
914
	{
915 916
		write_msg(NULL, "BLOB output is not supported for plain text dump files.\n");
		write_msg(NULL, "(Use a different output format.)\n");
P
Philip Warner 已提交
917 918 919
		exit(1);
	}

920
	/* open the output file */
B
Bruce Momjian 已提交
921 922
	switch (format[0])
	{
B
Bruce Momjian 已提交
923 924 925 926 927 928 929 930 931 932 933 934 935 936

		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;
937 938 939 940 941 942
			g_fout = CreateArchive(filename, archNull, 0);
			break;

		case 't':
		case 'T':
			g_fout = CreateArchive(filename, archTar, compressLevel);
B
Bruce Momjian 已提交
943 944 945
			break;

		default:
946
			write_msg(NULL, "invalid output format '%s' specified\n", format);
B
Bruce Momjian 已提交
947
			exit(1);
B
Bruce Momjian 已提交
948 949 950 951
	}

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

956 957
	/* Let the archiver know how noisy to be */
	g_fout->verbose = g_verbose;
958

B
Bruce Momjian 已提交
959 960
	/*
	 * Open the database using the Archiver, so it knows about it. Errors
961
	 * mean death.
B
Bruce Momjian 已提交
962
	 */
963 964
	g_fout->minRemoteVersion = 70000; /* we can handle back to 7.0 */
	g_fout->maxRemoteVersion = parse_version(PG_VERSION);
965
	g_conn = ConnectDatabase(g_fout, dbname, pghost, pgport, username, force_password, ignore_version);
966

967 968 969 970 971 972 973 974
	/*
	 * Start serializable transaction to dump consistent data
	 */
	{
		PGresult   *res;

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

978 979 980
		PQclear(res);
		res = PQexec(g_conn, "set transaction isolation level serializable");
		if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
981
			exit_horribly(g_fout, NULL, "could not set transaction isolation level to serializable: %s",
B
Bruce Momjian 已提交
982
						  PQerrorMessage(g_conn));
983

984 985 986
		PQclear(res);
	}

987 988 989 990
	if (g_fout->remoteVersion >= 70100)
		g_last_builtin_oid = findLastBuiltinOid_V71(dbname);
	else
		g_last_builtin_oid = findLastBuiltinOid_V70();
991

992 993 994 995
	/* Dump the database definition */
	if (!dataOnly)
		dumpDatabase(g_fout);

996
	if (oids == true)
997
		setMaxOid(g_fout);
B
Bruce Momjian 已提交
998 999

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

1003
	if (!schemaOnly)
B
Bruce Momjian 已提交
1004
		dumpClasses(tblinfo, numTables, g_fout, tablename, oids, force_quotes);
1005 1006

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

1009 1010
	if (!dataOnly)				/* dump indexes and triggers at the end
								 * for performance */
V
Vadim B. Mikheev 已提交
1011 1012
	{
		dumpTriggers(g_fout, tablename, tblinfo, numTables);
1013
		dumpRules(g_fout, tablename, tblinfo, numTables);
V
Vadim B. Mikheev 已提交
1014
	}
1015

1016 1017
	/* Now sort the output nicely */
	SortTocByOID(g_fout);
1018
	MoveToStart(g_fout, "DATABASE");
1019 1020 1021
	MoveToEnd(g_fout, "TABLE DATA");
	MoveToEnd(g_fout, "BLOBS");
	MoveToEnd(g_fout, "INDEX");
1022
	MoveToEnd(g_fout, "CONSTRAINT");
1023 1024
	MoveToEnd(g_fout, "TRIGGER");
	MoveToEnd(g_fout, "RULE");
1025
	MoveToEnd(g_fout, "SEQUENCE SET");
1026 1027 1028 1029 1030 1031
	/*
	 * 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");
1032

B
Bruce Momjian 已提交
1033
	if (plainText)
B
Bruce Momjian 已提交
1034 1035
	{
		ropt = NewRestoreOptions();
B
Bruce Momjian 已提交
1036
		ropt->filename = (char *) filename;
B
Bruce Momjian 已提交
1037 1038
		ropt->dropSchema = outputClean;
		ropt->aclsSkip = aclsSkip;
1039 1040 1041 1042 1043 1044 1045 1046 1047
		ropt->superuser = outputSuperuser;
		ropt->create = outputCreate;
		ropt->noOwner = outputNoOwner;
		ropt->noReconnect = outputNoReconnect;

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

		if (compressLevel == -1)
B
Bruce Momjian 已提交
1050
			ropt->compression = 0;
B
Bruce Momjian 已提交
1051
		else
B
Bruce Momjian 已提交
1052
			ropt->compression = compressLevel;
B
Bruce Momjian 已提交
1053

1054 1055
		ropt->suppressDumpWarnings = true; /* We've already shown them */

B
Bruce Momjian 已提交
1056 1057 1058 1059
		RestoreArchive(g_fout, ropt);
	}

	CloseArchive(g_fout);
1060

1061 1062 1063
	clearTableInfo(tblinfo, numTables);
	PQfinish(g_conn);
	exit(0);
1064 1065
}

1066 1067 1068 1069 1070
/*
 * dumpDatabase:
 *	dump the database definition
 *
 */
B
Bruce Momjian 已提交
1071
static int
1072 1073
dumpDatabase(Archive *AH)
{
B
Bruce Momjian 已提交
1074 1075 1076 1077 1078 1079
	PQExpBuffer dbQry = createPQExpBuffer();
	PQExpBuffer delQry = createPQExpBuffer();
	PQExpBuffer creaQry = createPQExpBuffer();
	PGresult   *res;
	int			ntups;
	int			i_dba;
1080 1081

	if (g_verbose)
1082
		write_msg(NULL, "saving database definition\n");
1083 1084

	/* Get the dba */
1085
	appendPQExpBuffer(dbQry, "select (select usename from pg_user where datdba = usesysid) as dba from pg_database"
B
Bruce Momjian 已提交
1086
					  " where datname = ");
1087
	formatStringLiteral(dbQry, PQdb(g_conn), CONV_ALL);
1088 1089 1090 1091 1092

	res = PQexec(g_conn, dbQry->data);
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1093 1094 1095
		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);
1096
		exit_nicely();
1097 1098 1099 1100
	}

	ntups = PQntuples(res);

1101 1102 1103
	if (ntups <= 0)
	{
		write_msg(NULL, "missing pg_database entry for database \"%s\"\n", PQdb(g_conn));
1104
		exit_nicely();
1105 1106
	}

1107 1108
	if (ntups != 1)
	{
1109 1110
		write_msg(NULL, "query returned more than one (%d) pg_database entry for database \"%s\"\n",
				  ntups, PQdb(g_conn));
1111
		exit_nicely();
1112 1113 1114 1115 1116 1117
	}

	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 已提交
1118 1119 1120 1121
	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 */ );
1122 1123 1124

	PQclear(res);

1125 1126 1127 1128
	destroyPQExpBuffer(dbQry);
	destroyPQExpBuffer(delQry);
	destroyPQExpBuffer(creaQry);

1129 1130 1131 1132
	return 1;
}


1133 1134 1135 1136 1137 1138
/*
 * dumpBlobs:
 *	dump all blobs
 *
 */

B
Bruce Momjian 已提交
1139
#define loBufSize 16384
1140 1141
#define loFetchSize 1000

B
Bruce Momjian 已提交
1142 1143
static int
dumpBlobs(Archive *AH, char *junkOid, void *junkVal)
1144
{
B
Bruce Momjian 已提交
1145 1146 1147 1148 1149 1150 1151
	PQExpBuffer oidQry = createPQExpBuffer();
	PQExpBuffer oidFetchQry = createPQExpBuffer();
	PGresult   *res;
	int			i;
	int			loFd;
	char		buf[loBufSize];
	int			cnt;
1152
	Oid			blobOid;
1153 1154

	if (g_verbose)
1155
		write_msg(NULL, "saving BLOBs\n");
1156 1157

	/* Cursor to get all BLOB tables */
1158 1159 1160 1161 1162 1163
	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'");
	}
1164 1165 1166 1167

	res = PQexec(g_conn, oidQry->data);
	if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
	{
1168
		write_msg(NULL, "dumpBlobs(): cursor declaration failed: %s", PQerrorMessage(g_conn));
1169
		exit_nicely();
1170 1171 1172 1173 1174
	}

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

B
Bruce Momjian 已提交
1175 1176
	do
	{
1177 1178 1179 1180 1181 1182
		/* Do a fetch */
		PQclear(res);
		res = PQexec(g_conn, oidFetchQry->data);

		if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
		{
1183 1184
			write_msg(NULL, "dumpBlobs(): fetch from cursor failed: %s",
					  PQerrorMessage(g_conn));
1185
			exit_nicely();
1186 1187 1188 1189 1190
		}

		/* Process the tuples, if any */
		for (i = 0; i < PQntuples(res); i++)
		{
1191
			blobOid = atooid(PQgetvalue(res, i, 0));
1192 1193 1194 1195
			/* Open the BLOB */
			loFd = lo_open(g_conn, blobOid, INV_READ);
			if (loFd == -1)
			{
1196 1197
				write_msg(NULL, "dumpBlobs(): could not open large object: %s",
						  PQerrorMessage(g_conn));
1198
				exit_nicely();
1199 1200 1201 1202 1203
			}

			StartBlob(AH, blobOid);

			/* Now read it in chunks, sending data to archive */
B
Bruce Momjian 已提交
1204 1205
			do
			{
1206
				cnt = lo_read(g_conn, loFd, buf, loBufSize);
B
Bruce Momjian 已提交
1207 1208
				if (cnt < 0)
				{
1209 1210
					write_msg(NULL, "dumpBlobs(): error reading large object: %s",
							  PQerrorMessage(g_conn));
1211
					exit_nicely();
1212 1213
				}

B
Bruce Momjian 已提交
1214
				WriteData(AH, buf, cnt);
1215 1216 1217 1218 1219 1220 1221 1222 1223 1224

			} while (cnt > 0);

			lo_close(g_conn, loFd);

			EndBlob(AH, blobOid);

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

1225 1226 1227
	destroyPQExpBuffer(oidQry);
	destroyPQExpBuffer(oidFetchQry);

1228 1229 1230
	return 1;
}

1231
/*
1232 1233
 * getTypes:
 *	  read all base types in the system catalogs and return them in the
1234 1235
 * TypeInfo* structure
 *
1236
 *	numTypes is set to the number of types read in
1237 1238
 *
 */
1239
TypeInfo   *
1240 1241
getTypes(int *numTypes)
{
B
Bruce Momjian 已提交
1242
	PGresult   *res;
1243 1244
	int			ntups;
	int			i;
1245
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1246
	TypeInfo   *tinfo;
1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260

	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;
1261 1262
	int			i_typalign;
	int			i_typstorage;
1263
	int			i_typbyval;
1264
	int			i_typisdefined;
1265
	int			i_usename;
1266
	int			i_typedefn;
1267 1268 1269 1270 1271 1272

	/* find all base types */

	/*
	 * we include even the built-in types because those may be used as
	 * array elements by user-defined types
1273
	 *
1274 1275 1276
	 * we filter out the built-in types when we dump out the types
	 */

1277 1278 1279 1280
	if (g_fout->remoteVersion < 70100)
	{
		appendPQExpBuffer(query, "SELECT pg_type.oid, typowner, typname, typlen, typprtlen, "
						"typinput, typoutput, typreceive, typsend, typelem, typdelim, "
1281
						"typdefault, typrelid, typalign, 'p'::char as typstorage, typbyval, typisdefined, "
1282 1283 1284 1285 1286 1287
						"(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, "
1288
						"typdefault, typrelid, typalign, typstorage, typbyval, typisdefined, "
1289 1290 1291 1292
						"(select usename from pg_user where typowner = usesysid) as usename, "
						"format_type(pg_type.oid, NULL) as typedefn "
						"from pg_type");
	}
1293

B
Hi, all  
Bruce Momjian 已提交
1294
	res = PQexec(g_conn, query->data);
1295 1296 1297
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1298
		write_msg(NULL, "query to obtain list of data types failed: %s", PQerrorMessage(g_conn));
1299
		exit_nicely();
1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318
	}

	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");
1319 1320
	i_typalign = PQfnumber(res, "typalign");
	i_typstorage = PQfnumber(res, "typstorage");
1321
	i_typbyval = PQfnumber(res, "typbyval");
1322
	i_typisdefined = PQfnumber(res, "typisdefined");
1323
	i_usename = PQfnumber(res, "usename");
1324
	i_typedefn = PQfnumber(res, "typedefn");
1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340

	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));
		tinfo[i].typdefault = strdup(PQgetvalue(res, i, i_typdefault));
		tinfo[i].typrelid = strdup(PQgetvalue(res, i, i_typrelid));
1341 1342
		tinfo[i].typalign = strdup(PQgetvalue(res, i, i_typalign));
		tinfo[i].typstorage = strdup(PQgetvalue(res, i, i_typstorage));
1343
		tinfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1344
		tinfo[i].typedefn = strdup(PQgetvalue(res, i, i_typedefn));
1345

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

1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362
		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;
1363 1364 1365 1366 1367

		if (strcmp(PQgetvalue(res, i, i_typisdefined), "f") == 0)
			tinfo[i].isDefined = 0;
		else
			tinfo[i].isDefined = 1;
1368 1369 1370 1371 1372 1373
	}

	*numTypes = ntups;

	PQclear(res);

1374 1375
	destroyPQExpBuffer(query);

1376
	return tinfo;
1377 1378 1379 1380
}

/*
 * getOperators:
1381
 *	  read all operators in the system catalogs and return them in the
1382 1383
 * OprInfo* structure
 *
1384 1385
 *	numOprs is set to the number of operators read in
 *
1386 1387
 *
 */
1388
OprInfo    *
1389 1390
getOperators(int *numOprs)
{
1391
	PGresult   *res;
1392 1393
	int			ntups;
	int			i;
1394
	PQExpBuffer query = createPQExpBuffer();
1395

B
Bruce Momjian 已提交
1396
	OprInfo    *oprinfo;
1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411

	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;
1412 1413 1414 1415 1416 1417

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

B
Hi, all  
Bruce Momjian 已提交
1418
	appendPQExpBuffer(query, "SELECT pg_operator.oid, oprname, oprkind, oprcode, "
1419
			   "oprleft, oprright, oprcom, oprnegate, oprrest, oprjoin, "
B
Bruce Momjian 已提交
1420 1421 1422
					  "oprcanhash, oprlsortop, oprrsortop, "
	"(select usename from pg_user where oprowner = usesysid) as usename "
					  "from pg_operator");
1423

B
Hi, all  
Bruce Momjian 已提交
1424
	res = PQexec(g_conn, query->data);
1425 1426 1427
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1428
		write_msg(NULL, "query to obtain list of operators failed: %s", PQerrorMessage(g_conn));
1429
		exit_nicely();
1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467
	}

	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));
1468 1469

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

1473 1474 1475 1476
	}

	PQclear(res);

1477 1478
	destroyPQExpBuffer(query);

1479
	return oprinfo;
1480 1481
}

1482
void
1483
clearTypeInfo(TypeInfo *tp, int numTypes)
1484
{
1485
	int			i;
1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514

	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);
1515 1516 1517 1518
		if (tp[i].typalign)
			free(tp[i].typalign);
		if (tp[i].typstorage)
			free(tp[i].typstorage);
1519 1520
		if (tp[i].usename)
			free(tp[i].usename);
1521 1522
		if (tp[i].typedefn)
			free(tp[i].typedefn);
1523 1524
	}
	free(tp);
1525 1526 1527
}

void
1528
clearFuncInfo(FuncInfo *fun, int numFuncs)
1529
{
1530 1531
	int			i,
				a;
1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542

	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);
1543
		for (a = 0; a < FUNC_MAX_ARGS; ++a)
1544 1545 1546 1547 1548 1549 1550 1551 1552 1553
			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);
1554 1555
}

1556
static void
1557
clearTableInfo(TableInfo *tblinfo, int numTables)
1558
{
1559 1560
	int			i,
				j;
1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585

	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 已提交
1586

B
Bruce Momjian 已提交
1587 1588 1589
		if (tblinfo[i].triggers)
		{
			for (j = 0; j < tblinfo[i].ntrig; j++)
B
Bruce Momjian 已提交
1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602
			{
				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);
		}

1603 1604
		if (tblinfo[i].atttypmod)
			free((int *) tblinfo[i].atttypmod);
1605 1606
		if (tblinfo[i].inhAttrs)
			free((int *) tblinfo[i].inhAttrs);
1607 1608 1609 1610
		if (tblinfo[i].inhAttrDef)
			free((int *) tblinfo[i].inhAttrDef);
		if (tblinfo[i].inhNotNull)
			free((int *) tblinfo[i].inhNotNull);
1611 1612
		if (tblinfo[i].attnames)
			free(tblinfo[i].attnames);
1613 1614
		if (tblinfo[i].atttypedefns)
			free(tblinfo[i].atttypedefns);
1615 1616 1617 1618
		if (tblinfo[i].typnames)
			free(tblinfo[i].typnames);
		if (tblinfo[i].notnull)
			free(tblinfo[i].notnull);
1619 1620
		if (tblinfo[i].primary_key_name)
			free(tblinfo[i].primary_key_name);
1621 1622
	}
	free(tblinfo);
1623 1624
}

1625
void
1626
clearInhInfo(InhInfo *inh, int numInherits)
1627
{
1628
	int			i;
1629 1630 1631 1632 1633

	if (!inh)
		return;
	for (i = 0; i < numInherits; ++i)
	{
1634 1635
		if (inh[i].inhrelid)
			free(inh[i].inhrelid);
1636 1637 1638 1639
		if (inh[i].inhparent)
			free(inh[i].inhparent);
	}
	free(inh);
1640 1641 1642
}

void
1643
clearOprInfo(OprInfo *opr, int numOprs)
1644
{
1645
	int			i;
1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680

	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);
1681 1682 1683
}

void
1684
clearIndInfo(IndInfo *ind, int numIndexes)
1685
{
1686 1687
	int			i,
				a;
1688 1689 1690

	if (!ind)
		return;
1691
	for (i = 0; i < numIndexes; ++i)
1692
	{
1693 1694 1695 1696
		if (ind[i].indexreloid)
			free(ind[i].indexreloid);
		if (ind[i].indreloid)
			free(ind[i].indreloid);
1697 1698 1699 1700 1701 1702 1703 1704 1705 1706
		if (ind[i].indexrelname)
			free(ind[i].indexrelname);
		if (ind[i].indrelname)
			free(ind[i].indrelname);
		if (ind[i].indamname)
			free(ind[i].indamname);
		if (ind[i].indproc)
			free(ind[i].indproc);
		if (ind[i].indisunique)
			free(ind[i].indisunique);
1707 1708
		if (ind[i].indisprimary)
			free(ind[i].indisprimary);
1709 1710
		if (ind[i].indhaspred)
			free(ind[i].indhaspred);
1711 1712 1713 1714 1715 1716 1717 1718 1719
		for (a = 0; a < INDEX_MAX_KEYS; ++a)
		{
			if (ind[i].indkey[a])
				free(ind[i].indkey[a]);
			if (ind[i].indclass[a])
				free(ind[i].indclass[a]);
		}
	}
	free(ind);
1720 1721 1722
}

void
B
Bruce Momjian 已提交
1723
clearAggInfo(AggInfo *agginfo, int numArgs)
1724
{
1725
	int			i;
1726 1727 1728 1729 1730 1731 1732 1733 1734

	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);
1735 1736
		if (agginfo[i].aggtransfn)
			free(agginfo[i].aggtransfn);
1737 1738
		if (agginfo[i].aggfinalfn)
			free(agginfo[i].aggfinalfn);
1739 1740
		if (agginfo[i].aggtranstype)
			free(agginfo[i].aggtranstype);
1741 1742
		if (agginfo[i].aggbasetype)
			free(agginfo[i].aggbasetype);
1743 1744
		if (agginfo[i].agginitval)
			free(agginfo[i].agginitval);
1745 1746 1747 1748
		if (agginfo[i].usename)
			free(agginfo[i].usename);
	}
	free(agginfo);
1749
}
1750 1751 1752

/*
 * getAggregates:
1753
 *	  read all the user-defined aggregates in the system catalogs and
1754 1755
 * return them in the AggInfo* structure
 *
1756 1757
 * numAggs is set to the number of aggregates read in
 *
1758 1759
 *
 */
1760
AggInfo    *
1761 1762
getAggregates(int *numAggs)
{
B
Bruce Momjian 已提交
1763
	PGresult   *res;
1764 1765
	int			ntups;
	int			i;
1766
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1767
	AggInfo    *agginfo;
1768 1769 1770

	int			i_oid;
	int			i_aggname;
1771
	int			i_aggtransfn;
1772
	int			i_aggfinalfn;
1773
	int			i_aggtranstype;
1774
	int			i_aggbasetype;
1775
	int			i_agginitval;
1776
	int			i_usename;
1777
	int			i_convertok;
1778 1779 1780

	/* find all user-defined aggregates */

1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796
	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");
	}
1797

B
Hi, all  
Bruce Momjian 已提交
1798
	res = PQexec(g_conn, query->data);
1799 1800 1801
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1802
		write_msg(NULL, "query to obtain list of aggregate functions failed: %s",
1803
				  PQerrorMessage(g_conn));
1804
		exit_nicely();
1805 1806 1807 1808 1809 1810 1811 1812 1813
	}

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

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

	i_oid = PQfnumber(res, "oid");
	i_aggname = PQfnumber(res, "aggname");
1814
	i_aggtransfn = PQfnumber(res, "aggtransfn");
1815
	i_aggfinalfn = PQfnumber(res, "aggfinalfn");
1816
	i_aggtranstype = PQfnumber(res, "aggtranstype");
1817
	i_aggbasetype = PQfnumber(res, "aggbasetype");
1818
	i_agginitval = PQfnumber(res, "agginitval");
1819
	i_usename = PQfnumber(res, "usename");
1820
	i_convertok = PQfnumber(res, "convertok");
1821 1822 1823 1824 1825

	for (i = 0; i < ntups; i++)
	{
		agginfo[i].oid = strdup(PQgetvalue(res, i, i_oid));
		agginfo[i].aggname = strdup(PQgetvalue(res, i, i_aggname));
1826
		agginfo[i].aggtransfn = strdup(PQgetvalue(res, i, i_aggtransfn));
1827
		agginfo[i].aggfinalfn = strdup(PQgetvalue(res, i, i_aggfinalfn));
1828
		agginfo[i].aggtranstype = strdup(PQgetvalue(res, i, i_aggtranstype));
1829
		agginfo[i].aggbasetype = strdup(PQgetvalue(res, i, i_aggbasetype));
1830
		agginfo[i].agginitval = strdup(PQgetvalue(res, i, i_agginitval));
1831
		agginfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1832
		if (strlen(agginfo[i].usename) == 0)
1833 1834
			write_msg(NULL, "WARNING: owner of aggregate function \"%s\" appears to be invalid\n",
					  agginfo[i].aggname);
1835

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

1838 1839 1840 1841
	}

	PQclear(res);

1842 1843
	destroyPQExpBuffer(query);

1844
	return agginfo;
1845 1846 1847 1848
}

/*
 * getFuncs:
1849
 *	  read all the user-defined functions in the system catalogs and
1850 1851
 * return them in the FuncInfo* structure
 *
1852 1853
 * numFuncs is set to the number of functions read in
 *
1854 1855
 *
 */
1856
FuncInfo   *
1857 1858
getFuncs(int *numFuncs)
{
1859
	PGresult   *res;
1860 1861
	int			ntups;
	int			i;
1862 1863
	PQExpBuffer query = createPQExpBuffer();
	FuncInfo   *finfo;
1864 1865 1866 1867 1868 1869 1870 1871 1872 1873

	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;
1874
	int			i_iscachable;
1875
	int			i_isstrict;
1876
	int			i_usename;
1877 1878 1879

	/* find all user-defined funcs */

1880 1881 1882 1883 1884 1885
	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, "
1886
				"proiscachable, 'f'::boolean as proisstrict "
1887 1888 1889 1890 1891
				"from pg_proc "
				"where pg_proc.oid > '%u'::oid",
					  g_last_builtin_oid);
	} else {
		appendPQExpBuffer(query,
1892
		   "SELECT pg_proc.oid, proname, prolang, pronargs, prorettype, "
1893
					  "proretset, proargtypes, prosrc, probin, "
1894 1895 1896 1897
				"(select usename from pg_user where proowner = usesysid) as usename, "
				"proiscachable, proisstrict "
				"from pg_proc "
				"where pg_proc.oid > '%u'::oid",
1898
					  g_last_builtin_oid);
1899
	}
1900

B
Hi, all  
Bruce Momjian 已提交
1901
	res = PQexec(g_conn, query->data);
1902 1903 1904
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
1905 1906
		write_msg(NULL, "query to obtain list of functions failed: %s",
				  PQerrorMessage(g_conn));
1907
		exit_nicely();
1908 1909 1910 1911 1912 1913 1914 1915
	}

	ntups = PQntuples(res);

	*numFuncs = ntups;

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

1916 1917
	memset((char *) finfo, 0, ntups * sizeof(FuncInfo));

1918 1919 1920 1921 1922 1923 1924 1925 1926
	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");
1927
	i_iscachable = PQfnumber(res, "proiscachable");
1928
	i_isstrict = PQfnumber(res, "proisstrict");
1929 1930 1931 1932 1933 1934 1935
	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));

1936
		finfo[i].prosrc = strdup(PQgetvalue(res, i, i_prosrc));
1937 1938 1939 1940 1941
		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));
1942
		finfo[i].lang = atooid(PQgetvalue(res, i, i_prolang));
1943
		finfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
B
Bruce Momjian 已提交
1944 1945
		finfo[i].iscachable = (strcmp(PQgetvalue(res, i, i_iscachable), "t") == 0);
		finfo[i].isstrict = (strcmp(PQgetvalue(res, i, i_isstrict), "t") == 0);
1946 1947

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

1951 1952
		if (finfo[i].nargs < 0 || finfo[i].nargs > FUNC_MAX_ARGS)
		{
1953 1954
			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);
1955
			exit_nicely();
1956 1957 1958 1959
		}
		parseNumericArray(PQgetvalue(res, i, i_proargtypes),
						  finfo[i].argtypes,
						  finfo[i].nargs);
1960 1961 1962 1963 1964
		finfo[i].dumped = 0;
	}

	PQclear(res);

1965
	destroyPQExpBuffer(query);
1966

1967
	return finfo;
1968 1969 1970 1971
}

/*
 * getTables
1972
 *	  read all the user-defined tables (no indexes, no catalogs)
1973 1974
 * in the system catalogs return them in the TableInfo* structure
 *
1975 1976
 * numTables is set to the number of tables read in
 *
1977 1978
 *
 */
1979
TableInfo  *
V
Vadim B. Mikheev 已提交
1980
getTables(int *numTables, FuncInfo *finfo, int numFuncs)
1981
{
1982
	PGresult   *res;
1983 1984
	int			ntups;
	int			i;
1985
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1986
	PQExpBuffer delqry = createPQExpBuffer();
1987
	TableInfo  *tblinfo;
1988

B
Bruce Momjian 已提交
1989
	int			i_reloid;
1990 1991 1992 1993
	int			i_relname;
	int			i_relkind;
	int			i_relacl;
	int			i_usename;
V
Vadim B. Mikheev 已提交
1994 1995
	int			i_relchecks;
	int			i_reltriggers;
1996
	int			i_relhasindex;
1997
	int			i_relhasoids;
1998 1999 2000 2001 2002

	char		relkindview[2];

	relkindview[0] = RELKIND_VIEW;
	relkindview[1] = '\0';
2003 2004

	/*
2005
	 * find all the user-defined tables (no indexes and no catalogs),
2006 2007 2008
	 * ordering by oid is important so that we always process the parent
	 * tables before the child tables when traversing the tblinfo*
	 *
2009
	 * we ignore tables that are not type 'r' (ordinary relation) or 'S'
2010
	 * (sequence) or 'v' (view).
2011 2012
	 */

2013
	if (g_fout->remoteVersion >= 70200)
2014 2015
	{
		appendPQExpBuffer(query,
2016
					  "SELECT pg_class.oid, relname, relacl, relkind, "
2017
					  "(select usename from pg_user where relowner = usesysid) as usename, "
2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031
					  "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);
	}
	else if (g_fout->remoteVersion >= 70100)
	{
		/* all tables have oids in 7.1 */
		appendPQExpBuffer(query,
					  "SELECT pg_class.oid, relname, relacl, relkind, "
					  "(select usename from pg_user where relowner = usesysid) as usename, "
					  "relchecks, reltriggers, relhasindex, 't'::bool as relhasoids "
2032 2033
					  "from pg_class "
					  "where relname !~ '^pg_' "
2034 2035
					  "and relkind in ('%c', '%c', '%c') "
					  "order by oid",
B
Bruce Momjian 已提交
2036
					  RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW);
2037
	} else {
2038
		/*
2039 2040
		 * 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.
2041 2042 2043 2044
		 */
		appendPQExpBuffer(query,
					  "SELECT c.oid, relname, relacl, "
					  "CASE WHEN relhasrules and relkind = 'r' "
2045 2046 2047 2048 2049
					  "  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, "
2050
					  "relchecks, reltriggers, relhasindex, 't'::bool as relhasoids "
2051 2052 2053 2054
					  "from pg_class c "
					  "where relname !~ '^pg_' "
					  "and relkind in ('%c', '%c', '%c') "
					  "order by oid",
2055
					  RELKIND_VIEW,
2056 2057
					  RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW);
	}
2058

B
Hi, all  
Bruce Momjian 已提交
2059
	res = PQexec(g_conn, query->data);
2060 2061 2062
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
2063 2064
		write_msg(NULL, "query to obtain list of tables failed: %s",
				  PQerrorMessage(g_conn));
2065
		exit_nicely();
2066 2067 2068 2069 2070 2071 2072 2073
	}

	ntups = PQntuples(res);

	*numTables = ntups;

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

B
Bruce Momjian 已提交
2074
	i_reloid = PQfnumber(res, "oid");
2075 2076
	i_relname = PQfnumber(res, "relname");
	i_relacl = PQfnumber(res, "relacl");
2077
	i_relkind = PQfnumber(res, "relkind");
2078
	i_usename = PQfnumber(res, "usename");
V
Vadim B. Mikheev 已提交
2079 2080
	i_relchecks = PQfnumber(res, "relchecks");
	i_reltriggers = PQfnumber(res, "reltriggers");
B
Hi, all  
Bruce Momjian 已提交
2081
	i_relhasindex = PQfnumber(res, "relhasindex");
2082
	i_relhasoids = PQfnumber(res, "relhasoids");
2083 2084 2085

	for (i = 0; i < ntups; i++)
	{
B
Bruce Momjian 已提交
2086
		tblinfo[i].oid = strdup(PQgetvalue(res, i, i_reloid));
2087 2088 2089 2090
		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 已提交
2091 2092
		tblinfo[i].ncheck = atoi(PQgetvalue(res, i, i_relchecks));
		tblinfo[i].ntrig = atoi(PQgetvalue(res, i, i_reltriggers));
2093

2094
		if (strlen(tblinfo[i].usename) == 0)
2095 2096
			write_msg(NULL, "WARNING: owner of table \"%s\" appears to be invalid\n",
					  tblinfo[i].relname);
2097

2098 2099 2100 2101 2102 2103
		/* Get view definition */
		if (strcmp(PQgetvalue(res, i, i_relkind), relkindview) == 0)
		{
			PGresult   *res2;

			resetPQExpBuffer(query);
2104 2105
			appendPQExpBuffer(query, "SELECT definition as viewdef, ");
			/* XXX 7.2 - replace with att from pg_views or some other generic source */
2106 2107 2108
			appendPQExpBuffer(query, "(select oid from pg_rewrite where "
										" rulename=('_RET' || viewname)::name) as view_oid"
										" from pg_views where viewname = ");
2109
			formatStringLiteral(query, tblinfo[i].relname, CONV_ALL);
2110 2111
			appendPQExpBuffer(query, ";");

2112 2113 2114
			res2 = PQexec(g_conn, query->data);
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
2115 2116
				write_msg(NULL, "query to obtain definition of view \"%s\" failed: %s",
						  tblinfo[i].relname, PQerrorMessage(g_conn));
2117
				exit_nicely();
2118 2119
			}

B
Bruce Momjian 已提交
2120
			if (PQntuples(res2) != 1)
2121 2122
			{
				if (PQntuples(res2) < 1)
2123 2124
					write_msg(NULL, "query to obtain definition of view \"%s\" returned no data\n",
							  tblinfo[i].relname);
B
Bruce Momjian 已提交
2125
				else
2126
					write_msg(NULL, "query to obtain definition of view \"%s\" returned more than one definition\n",
B
Bruce Momjian 已提交
2127
							tblinfo[i].relname);
2128
				exit_nicely();
2129 2130
			}

2131 2132
			if (PQgetisnull(res2, 0, 1))
			{
2133 2134
				write_msg(NULL, "query to obtain definition of view \"%s\" returned NULL oid\n",
						  tblinfo[i].relname);
2135
				exit_nicely();
2136 2137
			}

2138
			tblinfo[i].viewdef = strdup(PQgetvalue(res2, 0, 0));
2139
			tblinfo[i].viewoid = strdup(PQgetvalue(res2, 0, 1));
2140

B
Bruce Momjian 已提交
2141
			if (strlen(tblinfo[i].viewdef) == 0)
2142
			{
2143 2144
				write_msg(NULL, "definition of view \"%s\" appears to be empty (length zero)\n",
						  tblinfo[i].relname);
2145
				exit_nicely();
2146 2147 2148 2149 2150
			}
		}
		else
			tblinfo[i].viewdef = NULL;

2151
		/*
2152 2153
		 * Get non-inherited CHECK constraints, if any.
		 *
B
Bruce Momjian 已提交
2154 2155
		 * Exclude inherited CHECKs from CHECK constraints total. If a
		 * constraint matches by name and condition with a constraint
2156
		 * belonging to a parent class (OR conditions match and both
2157
		 * names start with '$', we assume it was inherited.
2158 2159
		 */
		if (tblinfo[i].ncheck > 0)
V
Vadim B. Mikheev 已提交
2160 2161
		{
			PGresult   *res2;
2162 2163
			int			i_rcname,
						i_rcsrc;
V
Vadim B. Mikheev 已提交
2164 2165
			int			ntups2;
			int			i2;
2166

V
Vadim B. Mikheev 已提交
2167
			if (g_verbose)
2168 2169
				write_msg(NULL, "finding CHECK constraints for table %s\n",
						  tblinfo[i].relname);
2170

B
Hi, all  
Bruce Momjian 已提交
2171 2172
			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "SELECT rcname, rcsrc from pg_relcheck "
2173
							  " where rcrelid = '%s'::oid "
2174
							  "   and not exists "
2175 2176 2177 2178 2179 2180 2181 2182
							  "  (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) "
2183
							  " order by rcname ",
2184
							  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
2185
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
2186
			if (!res2 ||
2187
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2188
			{
2189
				write_msg(NULL, "query to obtain check constraints failed: %s", PQerrorMessage(g_conn));
2190
				exit_nicely();
V
Vadim B. Mikheev 已提交
2191 2192
			}
			ntups2 = PQntuples(res2);
2193
			if (ntups2 > tblinfo[i].ncheck)
V
Vadim B. Mikheev 已提交
2194
			{
2195 2196 2197
				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");
2198
				exit_nicely();
V
Vadim B. Mikheev 已提交
2199
			}
2200 2201 2202 2203

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

V
Vadim B. Mikheev 已提交
2204 2205
			i_rcname = PQfnumber(res2, "rcname");
			i_rcsrc = PQfnumber(res2, "rcsrc");
2206
			tblinfo[i].check_expr = (char **) malloc(ntups2 * sizeof(char *));
V
Vadim B. Mikheev 已提交
2207 2208
			for (i2 = 0; i2 < ntups2; i2++)
			{
B
Bruce Momjian 已提交
2209 2210
				const char *name = PQgetvalue(res2, i2, i_rcname);
				const char *expr = PQgetvalue(res2, i2, i_rcsrc);
2211

B
Hi, all  
Bruce Momjian 已提交
2212
				resetPQExpBuffer(query);
2213 2214
				if (name[0] != '$')
				{
2215 2216
					appendPQExpBuffer(query, "CONSTRAINT %s ",
									  fmtId(name, force_quotes));
B
Bruce Momjian 已提交
2217
				}
2218
				appendPQExpBuffer(query, "CHECK (%s)", expr);
B
Hi, all  
Bruce Momjian 已提交
2219
				tblinfo[i].check_expr[i2] = strdup(query->data);
V
Vadim B. Mikheev 已提交
2220 2221 2222 2223 2224
			}
			PQclear(res2);
		}
		else
			tblinfo[i].check_expr = NULL;
2225

2226 2227
		/* Get primary key */
		if (strcmp(PQgetvalue(res, i, i_relhasindex), "t") == 0)
2228
		{
2229
			PGresult   *res2;
B
Bruce Momjian 已提交
2230

B
Hi, all  
Bruce Momjian 已提交
2231
			resetPQExpBuffer(query);
B
Bruce Momjian 已提交
2232
			appendPQExpBuffer(query,
2233
							  "SELECT indexrelid FROM pg_index i WHERE i.indisprimary AND i.indrelid = %s ",
2234
							  tblinfo[i].oid);
2235
			res2 = PQexec(g_conn, query->data);
B
Bruce Momjian 已提交
2236 2237
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
2238 2239
				write_msg(NULL, "query to obtain primary key of table \"%s\" failed: %s",
						  tblinfo[i].relname, PQerrorMessage(g_conn));
2240
				exit_nicely();
B
Bruce Momjian 已提交
2241
			}
2242

B
Bruce Momjian 已提交
2243 2244
			if (PQntuples(res2) > 1)
			{
2245
				write_msg(NULL, "query to obtain primary key of table \"%s\" produced more than one result\n",
2246
						tblinfo[i].relname);
2247
				exit_nicely();
2248 2249
			}

B
Bruce Momjian 已提交
2250
			if (PQntuples(res2) == 1)
2251
				tblinfo[i].pkIndexOid = strdup(PQgetvalue(res2, 0, 0));
B
Bruce Momjian 已提交
2252
			else
2253 2254
				tblinfo[i].pkIndexOid = NULL;

2255
		}
2256
		else
2257
			tblinfo[i].pkIndexOid = NULL;
B
Bruce Momjian 已提交
2258

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

2262
		/* Get primary key name (if primary key exist) */
2263
		if (tblinfo[i].pkIndexOid != NULL)
2264 2265
		{
			PGresult   *res2;
B
Bruce Momjian 已提交
2266
			int			n;
2267 2268

			resetPQExpBuffer(query);
2269 2270 2271 2272
			appendPQExpBuffer(query,
							  "SELECT relname FROM pg_class "
							  "WHERE oid = %s",
							  tblinfo[i].pkIndexOid);
2273

2274 2275 2276
			res2 = PQexec(g_conn, query->data);
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
2277 2278
				write_msg(NULL, "query to obtain name of primary key of table \"%s\" failed: %s",
						  tblinfo[i].relname, PQerrorMessage(g_conn));
2279
				exit_nicely();
2280 2281 2282 2283 2284
			}

			n = PQntuples(res2);
			if (n != 1)
			{
2285 2286
				write_msg(NULL, "query to obtain name of primary key of table \"%s\" did not return exactly one result\n",
						  tblinfo[i].relname);
2287
				exit_nicely();
2288 2289 2290 2291 2292 2293
			}

			tblinfo[i].primary_key_name =
				strdup(fmtId(PQgetvalue(res2, 0, 0), force_quotes));
			if (tblinfo[i].primary_key_name == NULL)
			{
2294
				write_msg(NULL, "out of memory\n");
2295
				exit_nicely();
2296 2297 2298 2299 2300
			}
		}
		else
			tblinfo[i].primary_key_name = NULL;

V
Vadim B. Mikheev 已提交
2301 2302 2303 2304
		/* Get Triggers */
		if (tblinfo[i].ntrig > 0)
		{
			PGresult   *res2;
B
Bruce,  
Bruce Momjian 已提交
2305
			int			i_tgoid,
2306
						i_tgname,
2307 2308 2309
						i_tgfoid,
						i_tgtype,
						i_tgnargs,
2310 2311 2312 2313
						i_tgargs,
						i_tgisconstraint,
						i_tgconstrname,
						i_tgdeferrable,
2314 2315
						i_tgconstrrelid,
						i_tgconstrrelname,
2316
						i_tginitdeferred;
V
Vadim B. Mikheev 已提交
2317 2318
			int			ntups2;
			int			i2;
2319

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

B
Hi, all  
Bruce Momjian 已提交
2323
			resetPQExpBuffer(query);
2324
			appendPQExpBuffer(query,
2325 2326 2327 2328 2329
								"SELECT tgname, tgfoid, tgtype, tgnargs, tgargs, "
								"tgisconstraint, tgconstrname, tgdeferrable, "
								"tgconstrrelid, tginitdeferred, oid, "
								"(select relname from pg_class where oid = tgconstrrelid) "
								"		as tgconstrrelname "
2330 2331 2332
							  "from pg_trigger "
							  "where tgrelid = '%s'::oid ",
							  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
2333
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
2334
			if (!res2 ||
2335
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2336
			{
2337
				write_msg(NULL, "query to obtain list of triggers failed: %s", PQerrorMessage(g_conn));
2338
				exit_nicely();
V
Vadim B. Mikheev 已提交
2339 2340 2341 2342
			}
			ntups2 = PQntuples(res2);
			if (ntups2 != tblinfo[i].ntrig)
			{
2343 2344
				write_msg(NULL, "expected %d triggers on table \"%s\" but found %d\n",
						  tblinfo[i].ntrig, tblinfo[i].relname, ntups2);
2345
				exit_nicely();
V
Vadim B. Mikheev 已提交
2346 2347 2348 2349 2350 2351
			}
			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 已提交
2352
			i_tgoid = PQfnumber(res2, "oid");
2353 2354 2355
			i_tgisconstraint = PQfnumber(res2, "tgisconstraint");
			i_tgconstrname = PQfnumber(res2, "tgconstrname");
			i_tgdeferrable = PQfnumber(res2, "tgdeferrable");
2356 2357
			i_tgconstrrelid = PQfnumber(res2, "tgconstrrelid");
			i_tgconstrrelname = PQfnumber(res2, "tgconstrrelname");
2358 2359
			i_tginitdeferred = PQfnumber(res2, "tginitdeferred");

B
Bruce Momjian 已提交
2360
			tblinfo[i].triggers = (TrigInfo *) malloc(ntups2 * sizeof(TrigInfo));
B
Hi, all  
Bruce Momjian 已提交
2361 2362
			resetPQExpBuffer(query);
			for (i2 = 0; i2 < ntups2; i2++)
V
Vadim B. Mikheev 已提交
2363
			{
2364
				const char *tgfuncoid = PQgetvalue(res2, i2, i_tgfoid);
2365
				char	   *tgfunc = NULL;
2366 2367
				int2		tgtype = atoi(PQgetvalue(res2, i2, i_tgtype));
				int			tgnargs = atoi(PQgetvalue(res2, i2, i_tgnargs));
B
Bruce Momjian 已提交
2368
				const char *tgargs = PQgetvalue(res2, i2, i_tgargs);
2369 2370 2371
				int			tgisconstraint;
				int			tgdeferrable;
				int			tginitdeferred;
2372 2373
				char	   *tgconstrrelid;
				char	   *tgname;
B
Bruce Momjian 已提交
2374
				const char *p;
2375 2376
				int			findx;

2377 2378
				tgname = PQgetvalue(res2, i2, i_tgname);

2379
				if (strcmp(PQgetvalue(res2, i2, i_tgisconstraint), "f") == 0)
2380
					tgisconstraint = 0;
2381
				else
2382
					tgisconstraint = 1;
2383 2384

				if (strcmp(PQgetvalue(res2, i2, i_tgdeferrable), "f") == 0)
2385
					tgdeferrable = 0;
2386
				else
2387
					tgdeferrable = 1;
2388 2389

				if (strcmp(PQgetvalue(res2, i2, i_tginitdeferred), "f") == 0)
2390
					tginitdeferred = 0;
2391
				else
2392
					tginitdeferred = 1;
2393

V
Vadim B. Mikheev 已提交
2394 2395
				for (findx = 0; findx < numFuncs; findx++)
				{
2396
					if (strcmp(finfo[findx].oid, tgfuncoid) == 0 &&
2397
						finfo[findx].nargs == 0 &&
V
Vadim B. Mikheev 已提交
2398 2399 2400
						strcmp(finfo[findx].prorettype, "0") == 0)
						break;
				}
2401

V
Vadim B. Mikheev 已提交
2402 2403
				if (findx == numFuncs)
				{
2404
					PGresult   *r;
2405
					int			numFuncs;
2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422

					/*
					 * 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)
					{
2423 2424
						write_msg(NULL, "query to obtain procedure name for trigger \"%s\" failed: %s",
								  tgname, PQerrorMessage(g_conn));
2425
						exit_nicely();
2426 2427 2428 2429
					}

					/* Sanity: Check we got only one tuple */
					numFuncs = PQntuples(r);
B
Bruce Momjian 已提交
2430 2431
					if (numFuncs != 1)
					{
2432 2433
						write_msg(NULL, "query to obtain procedure name for trigger \"%s\" did not return exactly one result\n",
								  tgname);
2434
						exit_nicely();
2435
					}
2436

2437
					tgfunc = strdup(PQgetvalue(r, 0, PQfnumber(r, "proname")));
2438
					PQclear(r);
2439
				}
2440
				else
2441
					tgfunc = strdup(finfo[findx].proname);
B
Bruce Momjian 已提交
2442

2443
				appendPQExpBuffer(delqry, "DROP TRIGGER %s ", fmtId(tgname, force_quotes));
B
Bruce Momjian 已提交
2444
				appendPQExpBuffer(delqry, "ON %s;\n",
B
Bruce Momjian 已提交
2445
								fmtId(tblinfo[i].relname, force_quotes));
2446

B
Hi, all  
Bruce Momjian 已提交
2447
				resetPQExpBuffer(query);
2448 2449 2450 2451 2452
				if (tgisconstraint)
				{
					appendPQExpBuffer(query, "CREATE CONSTRAINT TRIGGER ");
					appendPQExpBuffer(query, fmtId(PQgetvalue(res2, i2, i_tgconstrname), force_quotes));
				}
2453 2454
				else
				{
2455
					appendPQExpBuffer(query, "CREATE TRIGGER ");
2456
					appendPQExpBuffer(query, fmtId(tgname, force_quotes));
2457
				}
B
Bruce Momjian 已提交
2458
				appendPQExpBufferChar(query, ' ');
V
Vadim B. Mikheev 已提交
2459 2460 2461
				/* Trigger type */
				findx = 0;
				if (TRIGGER_FOR_BEFORE(tgtype))
B
Hi, all  
Bruce Momjian 已提交
2462
					appendPQExpBuffer(query, "BEFORE");
V
Vadim B. Mikheev 已提交
2463
				else
B
Hi, all  
Bruce Momjian 已提交
2464
					appendPQExpBuffer(query, "AFTER");
V
Vadim B. Mikheev 已提交
2465 2466
				if (TRIGGER_FOR_INSERT(tgtype))
				{
B
Hi, all  
Bruce Momjian 已提交
2467
					appendPQExpBuffer(query, " INSERT");
V
Vadim B. Mikheev 已提交
2468 2469 2470 2471 2472
					findx++;
				}
				if (TRIGGER_FOR_DELETE(tgtype))
				{
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
2473
						appendPQExpBuffer(query, " OR DELETE");
V
Vadim B. Mikheev 已提交
2474
					else
B
Hi, all  
Bruce Momjian 已提交
2475
						appendPQExpBuffer(query, " DELETE");
V
Vadim B. Mikheev 已提交
2476 2477 2478
					findx++;
				}
				if (TRIGGER_FOR_UPDATE(tgtype))
2479
				{
V
Vadim B. Mikheev 已提交
2480
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
2481
						appendPQExpBuffer(query, " OR UPDATE");
V
Vadim B. Mikheev 已提交
2482
					else
B
Hi, all  
Bruce Momjian 已提交
2483
						appendPQExpBuffer(query, " UPDATE");
2484
				}
2485 2486
				appendPQExpBuffer(query, " ON %s ", fmtId(tblinfo[i].relname, force_quotes));

2487
				if (tgisconstraint)
2488
				{
2489 2490 2491 2492 2493
					tgconstrrelid = PQgetvalue(res2, i2, i_tgconstrrelid);

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

						if (PQgetisnull(res2, i2, i_tgconstrrelname))
2494
						{
2495 2496
							write_msg(NULL, "query produced NULL referenced table name for trigger \"%s\" on table \"%s\" (oid was %s)\n",
									  tgname, tblinfo[i].relname, tgconstrrelid);
2497
							exit_nicely();
2498
						}
2499

2500
						appendPQExpBuffer(query, " FROM %s",
2501 2502
										fmtId(PQgetvalue(res2, i2, i_tgconstrrelname), force_quotes));
					}
2503
					if (!tgdeferrable)
2504 2505 2506 2507
						appendPQExpBuffer(query, " NOT");
					appendPQExpBuffer(query, " DEFERRABLE INITIALLY ");
					if (tginitdeferred)
						appendPQExpBuffer(query, "DEFERRED");
2508
					else
2509
						appendPQExpBuffer(query, "IMMEDIATE");
2510

2511 2512 2513
				}

				appendPQExpBuffer(query, " FOR EACH ROW");
2514 2515
				appendPQExpBuffer(query, " EXECUTE PROCEDURE %s (",
								  fmtId(tgfunc, force_quotes));
V
Vadim B. Mikheev 已提交
2516 2517
				for (findx = 0; findx < tgnargs; findx++)
				{
2518
					const char *s;
2519 2520

					for (p = tgargs;;)
V
Vadim B. Mikheev 已提交
2521
					{
2522
						p = strchr(p, '\\');
V
Vadim B. Mikheev 已提交
2523 2524
						if (p == NULL)
						{
2525 2526 2527 2528
							write_msg(NULL, "bad argument string (%s) for trigger \"%s\" on table \"%s\"\n",
									  PQgetvalue(res2, i2, i_tgargs),
									  tgname,
									  tblinfo[i].relname);
2529
							exit_nicely();
V
Vadim B. Mikheev 已提交
2530 2531 2532 2533 2534 2535 2536
						}
						p++;
						if (*p == '\\')
						{
							p++;
							continue;
						}
2537
						if (p[0] == '0' && p[1] == '0' && p[2] == '0')
V
Vadim B. Mikheev 已提交
2538 2539 2540
							break;
					}
					p--;
2541
					appendPQExpBufferChar(query, '\'');
B
Bruce Momjian 已提交
2542
					for (s = tgargs; s < p;)
V
Vadim B. Mikheev 已提交
2543 2544
					{
						if (*s == '\'')
2545 2546
							appendPQExpBufferChar(query, '\\');
						appendPQExpBufferChar(query, *s++);
V
Vadim B. Mikheev 已提交
2547
					}
B
Bruce Momjian 已提交
2548 2549
					appendPQExpBufferChar(query, '\'');
					appendPQExpBuffer(query, (findx < tgnargs - 1) ? ", " : "");
V
Vadim B. Mikheev 已提交
2550 2551
					tgargs = p + 4;
				}
B
Hi, all  
Bruce Momjian 已提交
2552
				appendPQExpBuffer(query, ");\n");
2553

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

				/*** Initialize trcomments and troids ***/
2557

B
Bruce,  
Bruce Momjian 已提交
2558
				resetPQExpBuffer(query);
2559
				appendPQExpBuffer(query, "TRIGGER %s ",
2560
					fmtId(tgname, force_quotes));
B
Bruce,  
Bruce Momjian 已提交
2561
				appendPQExpBuffer(query, "ON %s",
2562
								fmtId(tblinfo[i].relname, force_quotes));
B
Bruce Momjian 已提交
2563 2564
				tblinfo[i].triggers[i2].tgcomment = strdup(query->data);
				tblinfo[i].triggers[i2].oid = strdup(PQgetvalue(res2, i2, i_tgoid));
2565
				tblinfo[i].triggers[i2].tgname = strdup(fmtId(tgname, false));
B
Bruce Momjian 已提交
2566
				tblinfo[i].triggers[i2].tgdel = strdup(delqry->data);
B
Bruce,  
Bruce Momjian 已提交
2567

2568 2569
				if (tgfunc)
					free(tgfunc);
V
Vadim B. Mikheev 已提交
2570 2571 2572 2573 2574
			}
			PQclear(res2);
		}
		else
			tblinfo[i].triggers = NULL;
2575

2576 2577 2578 2579
	}

	PQclear(res);

2580 2581
	destroyPQExpBuffer(query);
	destroyPQExpBuffer(delqry);
2582

2583
	return tblinfo;
2584 2585 2586 2587
}

/*
 * getInherits
2588
 *	  read all the inheritance information
2589 2590
 * from the system catalogs return them in the InhInfo* structure
 *
2591 2592
 * numInherits is set to the number of tables read in
 *
2593 2594
 *
 */
2595
InhInfo    *
2596 2597
getInherits(int *numInherits)
{
2598
	PGresult   *res;
2599 2600
	int			ntups;
	int			i;
2601 2602
	PQExpBuffer query = createPQExpBuffer();
	InhInfo    *inhinfo;
2603

2604
	int			i_inhrelid;
2605
	int			i_inhparent;
2606 2607 2608

	/* find all the inheritance information */

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

B
Hi, all  
Bruce Momjian 已提交
2611
	res = PQexec(g_conn, query->data);
2612 2613 2614
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
2615 2616
		write_msg(NULL, "query to obtain inheritance relationships failed: %s",
				  PQerrorMessage(g_conn));
2617
		exit_nicely();
2618 2619 2620 2621 2622 2623 2624 2625
	}

	ntups = PQntuples(res);

	*numInherits = ntups;

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

2626
	i_inhrelid = PQfnumber(res, "inhrelid");
2627 2628 2629 2630
	i_inhparent = PQfnumber(res, "inhparent");

	for (i = 0; i < ntups; i++)
	{
2631
		inhinfo[i].inhrelid = strdup(PQgetvalue(res, i, i_inhrelid));
2632 2633 2634 2635
		inhinfo[i].inhparent = strdup(PQgetvalue(res, i, i_inhparent));
	}

	PQclear(res);
2636 2637 2638

	destroyPQExpBuffer(query);

2639
	return inhinfo;
2640 2641 2642 2643
}

/*
 * getTableAttrs -
2644 2645
 *	  for each table in tblinfo, read its attributes types and names
 *
2646
 * this is implemented in a very inefficient way right now, looping
2647
 * through the tblinfo and doing a join per table to find the attrs and their
2648 2649
 * types
 *
2650
 *	modifies tblinfo
2651 2652
 */
void
2653
getTableAttrs(TableInfo *tblinfo, int numTables)
2654
{
2655 2656
	int			i,
				j;
2657
	PQExpBuffer q = createPQExpBuffer();
2658 2659
	int			i_attname;
	int			i_typname;
2660
	int			i_atttypmod;
2661
	int			i_attnotnull;
V
Vadim B. Mikheev 已提交
2662
	int			i_atthasdef;
2663
	int			i_atttypedefn;
2664
	PGresult   *res;
2665
	int			ntups;
2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679

	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)
2680 2681
			write_msg(NULL, "finding the columns and types for table %s\n",
					  tblinfo[i].relname);
2682

B
Hi, all  
Bruce Momjian 已提交
2683
		resetPQExpBuffer(q);
2684 2685 2686 2687

		if (g_fout->remoteVersion < 70100)
		{
			/* Fake the LOJ below */
2688
			appendPQExpBuffer(q,
2689
						"  SELECT a.attnum, a.attname, t.typname, a.atttypmod, "
2690 2691 2692 2693 2694
						"        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 "
2695
						" UNION ALL SELECT a.attnum, a.attname, NULL as typname, a.atttypmod, "
2696 2697 2698 2699 2700 2701 2702 2703 2704
						"        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 {
2705
			appendPQExpBuffer(q, "SELECT a.attnum, a.attname, t.typname, a.atttypmod, "
2706 2707 2708 2709 2710 2711 2712
								"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 已提交
2713
		res = PQexec(g_conn, q->data);
2714 2715 2716
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
2717
			write_msg(NULL, "query to get table columns failed: %s", PQerrorMessage(g_conn));
2718
			exit_nicely();
2719 2720 2721 2722 2723 2724
		}

		ntups = PQntuples(res);

		i_attname = PQfnumber(res, "attname");
		i_typname = PQfnumber(res, "typname");
2725
		i_atttypmod = PQfnumber(res, "atttypmod");
2726
		i_attnotnull = PQfnumber(res, "attnotnull");
V
Vadim B. Mikheev 已提交
2727
		i_atthasdef = PQfnumber(res, "atthasdef");
2728
		i_atttypedefn = PQfnumber(res, "atttypedefn");
2729 2730 2731

		tblinfo[i].numatts = ntups;
		tblinfo[i].attnames = (char **) malloc(ntups * sizeof(char *));
2732
		tblinfo[i].atttypedefns = (char **) malloc(ntups * sizeof(char *));
2733
		tblinfo[i].typnames = (char **) malloc(ntups * sizeof(char *));
2734
		tblinfo[i].atttypmod = (int *) malloc(ntups * sizeof(int));
2735
		tblinfo[i].inhAttrs = (int *) malloc(ntups * sizeof(int));
2736 2737
		tblinfo[i].inhAttrDef = (int *) malloc(ntups * sizeof(int));
		tblinfo[i].inhNotNull = (int *) malloc(ntups * sizeof(int));
2738
		tblinfo[i].notnull = (bool *) malloc(ntups * sizeof(bool));
V
Vadim B. Mikheev 已提交
2739
		tblinfo[i].adef_expr = (char **) malloc(ntups * sizeof(char *));
2740 2741 2742 2743
		tblinfo[i].parentRels = NULL;
		tblinfo[i].numParents = 0;
		for (j = 0; j < ntups; j++)
		{
2744 2745 2746
			/* Sanity check on LOJ */
			if (PQgetisnull(res, j, i_typname))
			{
2747 2748
				write_msg(NULL, "query produced NULL name for data type of column %d of table %s\n",
						  j+1, tblinfo[i].relname);
2749
				exit_nicely();
2750 2751
			}

2752
			tblinfo[i].attnames[j] = strdup(PQgetvalue(res, j, i_attname));
2753
			tblinfo[i].atttypedefns[j] = strdup(PQgetvalue(res, j, i_atttypedefn));
2754
			tblinfo[i].typnames[j] = strdup(PQgetvalue(res, j, i_typname));
2755
			tblinfo[i].atttypmod[j] = atoi(PQgetvalue(res, j, i_atttypmod));
2756 2757
			tblinfo[i].inhAttrs[j] = 0; /* this flag is set in
										 * flagInhAttrs() */
2758 2759 2760
			tblinfo[i].inhAttrDef[j] = 0;
			tblinfo[i].inhNotNull[j] = 0;

2761
			tblinfo[i].notnull[j] = (PQgetvalue(res, j, i_attnotnull)[0] == 't') ? true : false;
V
Vadim B. Mikheev 已提交
2762 2763 2764
			if (PQgetvalue(res, j, i_atthasdef)[0] == 't')
			{
				PGresult   *res2;
2765
				int			numAttr;
2766

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

B
Hi, all  
Bruce Momjian 已提交
2771 2772
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "SELECT adsrc from pg_attrdef "
2773 2774
							 "where adrelid = '%s'::oid and adnum = %d ",
								  tblinfo[i].oid, j + 1);
B
Hi, all  
Bruce Momjian 已提交
2775
				res2 = PQexec(g_conn, q->data);
V
Vadim B. Mikheev 已提交
2776
				if (!res2 ||
2777
					PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2778
				{
2779 2780
					write_msg(NULL, "query to get column default value failed: %s",
							  PQerrorMessage(g_conn));
2781
					exit_nicely();
V
Vadim B. Mikheev 已提交
2782
				}
2783 2784 2785

				/* Sanity: Check we got only one tuple */
				numAttr = PQntuples(res2);
B
Bruce Momjian 已提交
2786 2787
				if (numAttr != 1)
				{
2788 2789
					write_msg(NULL, "query to get default value for column \"%s\" returned %d rows; expected 1\n",
							  tblinfo[i].attnames[j], numAttr);
2790
					exit_nicely();
2791 2792
				}

V
Vadim B. Mikheev 已提交
2793 2794 2795 2796 2797
				tblinfo[i].adef_expr[j] = strdup(PQgetvalue(res2, 0, PQfnumber(res2, "adsrc")));
				PQclear(res2);
			}
			else
				tblinfo[i].adef_expr[j] = NULL;
2798 2799 2800
		}
		PQclear(res);
	}
2801 2802

	destroyPQExpBuffer(q);
2803 2804 2805 2806
}


/*
2807 2808
 * getIndexes
 *	  read all the user-defined indexes information
2809 2810
 * from the system catalogs return them in the InhInfo* structure
 *
2811
 * numIndexes is set to the number of indexes read in
2812
 *
2813 2814
 *
 */
2815
IndInfo    *
2816
getIndexes(int *numIndexes)
2817
{
2818
	int			i;
2819
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
2820
	PGresult   *res;
2821
	int			ntups;
B
Bruce Momjian 已提交
2822
	IndInfo    *indinfo;
2823

2824 2825
	int			i_indexreloid;
	int			i_indreloid;
2826 2827 2828 2829 2830 2831 2832
	int			i_indexrelname;
	int			i_indrelname;
	int			i_indamname;
	int			i_indproc;
	int			i_indkey;
	int			i_indclass;
	int			i_indisunique;
2833
	int			i_indisprimary;
2834
	int			i_indhaspred;
2835 2836

	/*
2837
	 * find all the user-defined indexes.
2838
	 *
2839
	 * Notice we skip indexes on system classes
2840 2841
	 *
	 * this is a 4-way join !!
2842 2843
	 *
	 * XXXX: Use LOJ
2844 2845
	 */

B
Hi, all  
Bruce Momjian 已提交
2846
	appendPQExpBuffer(query,
2847 2848 2849
					  "SELECT i.indexrelid as indexreloid, "
					  "i.indrelid as indreloid, "
					  "t1.relname as indexrelname, t2.relname as indrelname, "
2850
					  "i.indproc, i.indkey, i.indclass, "
2851 2852 2853 2854
					  "a.amname as indamname, i.indisunique, i.indisprimary, "
					  "length(i.indpred) > 0 as indhaspred "
					  "from pg_index i, pg_class t1, pg_class t2, pg_am a "
					  "WHERE t1.oid = i.indexrelid and t2.oid = i.indrelid "
2855
					  "and t1.relam = a.oid and i.indexrelid > '%u'::oid "
2856
					  "and t2.relname !~ '^pg_' ",
2857
					  g_last_builtin_oid);
2858

2859 2860 2861
	if (g_fout->remoteVersion < 70100)
		appendPQExpBuffer(query, " and t2.relkind != 'l'");

B
Hi, all  
Bruce Momjian 已提交
2862
	res = PQexec(g_conn, query->data);
2863 2864 2865
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
2866
		write_msg(NULL, "query to obtain list of indexes failed: %s", PQerrorMessage(g_conn));
2867
		exit_nicely();
2868 2869 2870 2871
	}

	ntups = PQntuples(res);

2872
	*numIndexes = ntups;
2873 2874 2875

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

2876 2877
	memset((char *) indinfo, 0, ntups * sizeof(IndInfo));

2878 2879
	i_indexreloid = PQfnumber(res, "indexreloid");
	i_indreloid = PQfnumber(res, "indreloid");
2880 2881 2882 2883 2884 2885 2886
	i_indexrelname = PQfnumber(res, "indexrelname");
	i_indrelname = PQfnumber(res, "indrelname");
	i_indamname = PQfnumber(res, "indamname");
	i_indproc = PQfnumber(res, "indproc");
	i_indkey = PQfnumber(res, "indkey");
	i_indclass = PQfnumber(res, "indclass");
	i_indisunique = PQfnumber(res, "indisunique");
2887
	i_indisprimary = PQfnumber(res, "indisprimary");
2888
	i_indhaspred = PQfnumber(res, "indhaspred");
2889 2890 2891

	for (i = 0; i < ntups; i++)
	{
2892 2893
		indinfo[i].indexreloid = strdup(PQgetvalue(res, i, i_indexreloid));
		indinfo[i].indreloid = strdup(PQgetvalue(res, i, i_indreloid));
2894 2895 2896 2897
		indinfo[i].indexrelname = strdup(PQgetvalue(res, i, i_indexrelname));
		indinfo[i].indrelname = strdup(PQgetvalue(res, i, i_indrelname));
		indinfo[i].indamname = strdup(PQgetvalue(res, i, i_indamname));
		indinfo[i].indproc = strdup(PQgetvalue(res, i, i_indproc));
2898 2899 2900 2901 2902 2903
		parseNumericArray(PQgetvalue(res, i, i_indkey),
						  indinfo[i].indkey,
						  INDEX_MAX_KEYS);
		parseNumericArray(PQgetvalue(res, i, i_indclass),
						  indinfo[i].indclass,
						  INDEX_MAX_KEYS);
2904
		indinfo[i].indisunique = strdup(PQgetvalue(res, i, i_indisunique));
2905
		indinfo[i].indisprimary = strdup(PQgetvalue(res, i, i_indisprimary));
2906
		indinfo[i].indhaspred = strdup(PQgetvalue(res, i, i_indhaspred));
2907 2908
	}
	PQclear(res);
2909 2910 2911

	destroyPQExpBuffer(query);

2912
	return indinfo;
2913 2914
}

B
Bruce,  
Bruce Momjian 已提交
2915
/*------------------------------------------------------------------
2916
 * dumpComment --
B
Bruce,  
Bruce Momjian 已提交
2917
 *
2918
 * This routine is used to dump any comments associated with the
B
Bruce,  
Bruce Momjian 已提交
2919
 * oid handed to this routine. The routine takes a constant character
2920 2921 2922 2923 2924 2925
 * 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 已提交
2926 2927 2928
 *------------------------------------------------------------------
*/

T
Tom Lane 已提交
2929
static void
2930
dumpComment(Archive *fout, const char *target, const char *oid,
2931
			const char *classname, int subid,
2932
			const char *((*deps)[]))
2933 2934
{
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
2935
	PQExpBuffer query;
2936
	int			i_description;
B
Bruce,  
Bruce Momjian 已提交
2937

2938 2939 2940 2941
	/* Comments are SCHEMA not data */
	if (dataOnly)
		return;

B
Bruce,  
Bruce Momjian 已提交
2942 2943 2944
	/*** Build query to find comment ***/

	query = createPQExpBuffer();
2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958

	if (fout->remoteVersion >= 70200)
	{
		appendPQExpBuffer(query, "SELECT description FROM pg_description "
						  "WHERE objoid = %s and classoid = "
						  "(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... */
		appendPQExpBuffer(query, "SELECT description FROM pg_description WHERE objoid = %s", oid);
	}
B
Bruce,  
Bruce Momjian 已提交
2959 2960 2961 2962

	/*** Execute query ***/

	res = PQexec(g_conn, query->data);
2963 2964
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
2965 2966
		write_msg(NULL, "query to get comment on oid %s failed: %s",
				   oid, PQerrorMessage(g_conn));
2967
		exit_nicely();
B
Bruce,  
Bruce Momjian 已提交
2968 2969 2970 2971
	}

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

2972
	if (PQntuples(res) == 1)
2973
	{
B
Bruce,  
Bruce Momjian 已提交
2974
		i_description = PQfnumber(res, "description");
B
Bruce Momjian 已提交
2975
		resetPQExpBuffer(query);
2976
		appendPQExpBuffer(query, "COMMENT ON %s IS ", target);
2977 2978
		formatStringLiteral(query, PQgetvalue(res, 0, i_description),
							PASS_LFTAB);
2979
		appendPQExpBuffer(query, ";\n");
B
Bruce Momjian 已提交
2980

2981 2982
		ArchiveEntry(fout, oid, target, "COMMENT", deps,
					 query->data, "" /* Del */ ,
B
Bruce Momjian 已提交
2983
					 "" /* Copy */ , "" /* Owner */ , NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
2984 2985 2986 2987 2988
	}

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

	PQclear(res);
2989
	destroyPQExpBuffer(query);
B
Bruce,  
Bruce Momjian 已提交
2990 2991 2992
}

/*------------------------------------------------------------------
2993
 * dumpDBComment --
B
Bruce,  
Bruce Momjian 已提交
2994
 *
2995 2996
 * 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 已提交
2997 2998 2999 3000 3001
 * to dump the schema of the database, then this is the first
 * statement issued.
 *------------------------------------------------------------------
*/

3002
void
B
Bruce Momjian 已提交
3003
dumpDBComment(Archive *fout)
3004 3005
{
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
3006
	PQExpBuffer query;
3007
	int			i_oid;
B
Bruce,  
Bruce Momjian 已提交
3008 3009 3010 3011

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

	query = createPQExpBuffer();
3012
	appendPQExpBuffer(query, "SELECT oid FROM pg_database WHERE datname = ");
3013
	formatStringLiteral(query, PQdb(g_conn), CONV_ALL);
B
Bruce,  
Bruce Momjian 已提交
3014 3015 3016 3017

	/*** Execute query ***/

	res = PQexec(g_conn, query->data);
3018 3019
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
3020 3021
		write_msg(NULL, "query to get database oid failed: %s",
				   PQerrorMessage(g_conn));
3022
		exit_nicely();
B
Bruce,  
Bruce Momjian 已提交
3023 3024 3025 3026
	}

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

3027 3028
	if (PQntuples(res) != 0)
	{
B
Bruce,  
Bruce Momjian 已提交
3029 3030 3031
		i_oid = PQfnumber(res, "oid");
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "DATABASE %s", fmtId(PQdb(g_conn), force_quotes));
3032 3033
		dumpComment(fout, query->data, PQgetvalue(res, 0, i_oid),
					"pg_database", 0, NULL);
B
Bruce,  
Bruce Momjian 已提交
3034 3035 3036 3037 3038
	}

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

	PQclear(res);
3039
	destroyPQExpBuffer(query);
B
Bruce,  
Bruce Momjian 已提交
3040 3041
}

3042 3043
/*
 * dumpTypes
3044
 *	  writes out to fout the queries to recreate all the user-defined types
3045 3046 3047
 *
 */
void
B
Bruce Momjian 已提交
3048
dumpTypes(Archive *fout, FuncInfo *finfo, int numFuncs,
3049
		  TypeInfo *tinfo, int numTypes)
3050
{
3051
	int			i;
3052
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3053
	PQExpBuffer delq = createPQExpBuffer();
3054
	int			funcInd;
3055
	const char *((*deps)[]);
3056
	int			depIdx;
3057 3058 3059 3060

	for (i = 0; i < numTypes; i++)
	{
		/* skip all the builtin types */
3061
		if (atooid(tinfo[i].oid) <= g_last_builtin_oid)
3062 3063 3064
			continue;

		/* skip relation types */
3065
		if (atooid(tinfo[i].typrelid) != 0)
3066 3067
			continue;

3068 3069 3070 3071
		/* skip undefined placeholder types */
		if (!tinfo[i].isDefined)
			continue;

3072 3073 3074 3075 3076
		/* skip all array types that start w/ underscore */
		if ((tinfo[i].typname[0] == '_') &&
			(strcmp(tinfo[i].typinput, "array_in") == 0))
			continue;

3077 3078 3079
		deps = malloc(sizeof(char*) * 10);
		depIdx = 0;

3080 3081 3082 3083 3084 3085
		/*
		 * 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)
3086 3087
		{
			(*deps)[depIdx++] = strdup(finfo[funcInd].oid);
3088
			dumpOneFunc(fout, finfo, funcInd, tinfo, numTypes);
3089
		}
3090 3091 3092

		funcInd = findFuncByName(finfo, numFuncs, tinfo[i].typoutput);
		if (funcInd != -1)
3093 3094
		{
			(*deps)[depIdx++] = strdup(finfo[funcInd].oid);
3095
			dumpOneFunc(fout, finfo, funcInd, tinfo, numTypes);
3096
		}
3097

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

B
Hi, all  
Bruce Momjian 已提交
3101 3102
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
3103
						  "CREATE TYPE %s "
3104
						  "( internallength = %s, externallength = %s,",
3105 3106
						  fmtId(tinfo[i].typname, force_quotes),
						  tinfo[i].typlen,
3107 3108 3109 3110 3111 3112 3113 3114 3115 3116
						  tinfo[i].typprtlen);
		/* 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));
		appendPQExpBuffer(q, " receive = %s, default = ",
						  fmtId(tinfo[i].typreceive, force_quotes));
3117
		formatStringLiteral(q, tinfo[i].typdefault, CONV_ALL);
3118 3119 3120

		if (tinfo[i].isArray)
		{
3121
			char	   *elemType;
3122

3123
			elemType = findTypeByOid(tinfo, numTypes, tinfo[i].typelem, zeroAsOpaque);
3124 3125
			if (elemType == NULL)
			{
3126
				write_msg(NULL, "notice: array type %s - type for elements (oid %s) is not dumped\n",
3127
						  tinfo[i].typname, tinfo[i].typelem);
3128 3129 3130
				continue;
			}

3131
			appendPQExpBuffer(q, ", element = %s, delimiter = ", elemType);
3132
			formatStringLiteral(q, tinfo[i].typdelim, CONV_ALL);
3133 3134

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

3137 3138 3139 3140 3141
		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)
3142 3143 3144 3145 3146 3147
			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");
3148
		else if (strcmp(tinfo[i].typstorage, "e") == 0)
3149
			appendPQExpBuffer(q, ", storage = external");
3150
		else if (strcmp(tinfo[i].typstorage, "x") == 0)
3151
			appendPQExpBuffer(q, ", storage = extended");
3152
		else if (strcmp(tinfo[i].typstorage, "m") == 0)
3153 3154
			appendPQExpBuffer(q, ", storage = main");

3155
		if (tinfo[i].passedbyvalue)
3156
			appendPQExpBuffer(q, ", passedbyvalue);\n");
3157
		else
B
Hi, all  
Bruce Momjian 已提交
3158
			appendPQExpBuffer(q, ");\n");
3159

3160 3161 3162
		(*deps)[depIdx++] = NULL; /* End of List */

		ArchiveEntry(fout, tinfo[i].oid, tinfo[i].typname, "TYPE", deps,
B
Bruce Momjian 已提交
3163
				  q->data, delq->data, "", tinfo[i].usename, NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
3164 3165 3166 3167

		/*** Dump Type Comments ***/

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

B
Bruce,  
Bruce Momjian 已提交
3169
		appendPQExpBuffer(q, "TYPE %s", fmtId(tinfo[i].typname, force_quotes));
3170
		dumpComment(fout, q->data, tinfo[i].oid, "pg_type", 0, NULL);
3171
	}
3172 3173 3174

	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
3175 3176
}

3177 3178
/*
 * dumpProcLangs
B
Bruce Momjian 已提交
3179
 *		  writes out to fout the queries to recreate user-defined procedural languages
3180 3181 3182
 *
 */
void
B
Bruce Momjian 已提交
3183
dumpProcLangs(Archive *fout, FuncInfo *finfo, int numFuncs,
B
Bruce Momjian 已提交
3184
			  TypeInfo *tinfo, int numTypes)
3185
{
3186 3187
	PGresult   *res;
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
3188 3189
	PQExpBuffer defqry = createPQExpBuffer();
	PQExpBuffer delqry = createPQExpBuffer();
3190
	int			ntups;
B
Bruce Momjian 已提交
3191
	int			i_oid;
3192 3193 3194 3195
	int			i_lanname;
	int			i_lanpltrusted;
	int			i_lanplcallfoid;
	int			i_lancompiler;
3196
	Oid			lanoid;
3197 3198 3199
	char	   *lanname;
	char	   *lancompiler;
	const char *lanplcallfoid;
3200 3201 3202
	int			i,
				fidx;

B
Bruce Momjian 已提交
3203
	appendPQExpBuffer(query, "SELECT oid, * FROM pg_language "
3204 3205
					  "WHERE lanispl "
					  "ORDER BY oid");
B
Hi, all  
Bruce Momjian 已提交
3206
	res = PQexec(g_conn, query->data);
3207 3208 3209
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
3210 3211
		write_msg(NULL, "query to obtain list of procedural languages failed: %s",
				   PQerrorMessage(g_conn));
3212
		exit_nicely();
3213 3214 3215
	}
	ntups = PQntuples(res);

B
Bruce Momjian 已提交
3216 3217 3218 3219
	i_lanname = PQfnumber(res, "lanname");
	i_lanpltrusted = PQfnumber(res, "lanpltrusted");
	i_lanplcallfoid = PQfnumber(res, "lanplcallfoid");
	i_lancompiler = PQfnumber(res, "lancompiler");
B
Bruce Momjian 已提交
3220
	i_oid = PQfnumber(res, "oid");
3221

B
Bruce Momjian 已提交
3222 3223
	for (i = 0; i < ntups; i++)
	{
3224
		lanoid = atooid(PQgetvalue(res, i, i_oid));
3225 3226 3227
		if (lanoid <= g_last_builtin_oid)
			continue;

3228
		lanplcallfoid = PQgetvalue(res, i, i_lanplcallfoid);
3229 3230


3231 3232 3233 3234 3235 3236 3237
		for (fidx = 0; fidx < numFuncs; fidx++)
		{
			if (!strcmp(finfo[fidx].oid, lanplcallfoid))
				break;
		}
		if (fidx >= numFuncs)
		{
3238 3239
			write_msg(NULL, "handler procedure for procedural language %s not found\n",
					   PQgetvalue(res, i, i_lanname));
3240
			exit_nicely();
3241 3242 3243 3244
		}

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

3245 3246
		lanname = PQgetvalue(res, i, i_lanname);
		lancompiler = PQgetvalue(res, i, i_lancompiler);
3247

3248
		appendPQExpBuffer(delqry, "DROP PROCEDURAL LANGUAGE ");
3249
		formatStringLiteral(delqry, lanname, CONV_ALL);
3250
		appendPQExpBuffer(delqry, ";\n");
3251

3252 3253 3254
		appendPQExpBuffer(defqry, "CREATE %sPROCEDURAL LANGUAGE ",
						  (PQgetvalue(res, i, i_lanpltrusted)[0] == 't') ?
						  "TRUSTED " : "");
3255
		formatStringLiteral(defqry, lanname, CONV_ALL);
3256 3257
		appendPQExpBuffer(defqry, " HANDLER %s LANCOMPILER ",
						  fmtId(finfo[fidx].proname, force_quotes));
3258
		formatStringLiteral(defqry, lancompiler, CONV_ALL);
3259
		appendPQExpBuffer(defqry, ";\n");
3260

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

3264 3265
		resetPQExpBuffer(defqry);
		resetPQExpBuffer(delqry);
3266 3267 3268 3269
	}

	PQclear(res);

3270 3271 3272
	destroyPQExpBuffer(query);
	destroyPQExpBuffer(defqry);
	destroyPQExpBuffer(delqry);
3273 3274
}

3275 3276
/*
 * dumpFuncs
3277
 *	  writes out to fout the queries to recreate all the user-defined functions
3278 3279 3280
 *
 */
void
B
Bruce Momjian 已提交
3281
dumpFuncs(Archive *fout, FuncInfo *finfo, int numFuncs,
3282
		  TypeInfo *tinfo, int numTypes)
3283
{
3284
	int			i;
3285 3286 3287

	for (i = 0; i < numFuncs; i++)
		dumpOneFunc(fout, finfo, i, tinfo, numTypes);
3288 3289 3290 3291
}

/*
 * dumpOneFunc:
3292 3293
 *	  dump out only one function,  the index of which is given in the third
 *	argument
3294 3295 3296
 *
 */

3297
static void
B
Bruce Momjian 已提交
3298
dumpOneFunc(Archive *fout, FuncInfo *finfo, int i,
3299
			TypeInfo *tinfo, int numTypes)
3300
{
3301
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3302 3303
	PQExpBuffer fn = createPQExpBuffer();
	PQExpBuffer delqry = createPQExpBuffer();
B
Bruce,  
Bruce Momjian 已提交
3304
	PQExpBuffer fnlist = createPQExpBuffer();
3305
	PQExpBuffer asPart = createPQExpBuffer();
3306
	char		func_lang[NAMEDATALEN + 1];
3307 3308
	PGresult   *res;
	int			nlangs;
3309
	int			j;
3310 3311
	int			i_lanname;
	char		query[256];
3312

B
Bruce Momjian 已提交
3313 3314 3315 3316
	char	   *listSep;
	char	   *listSepComma = ",";
	char	   *listSepNone = "";
	char	   *rettypename;
3317

3318
	if (finfo[i].dumped)
3319 3320 3321
		goto done;

	finfo[i].dumped = 1;
3322

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

3325 3326 3327 3328 3329
	sprintf(query, "SELECT lanname FROM pg_language WHERE oid = %u",
			finfo[i].lang);
	res = PQexec(g_conn, query);
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
B
Bruce Momjian 已提交
3330
	{
3331
		write_msg(NULL, "query to get name of procedural language failed: %s", PQerrorMessage(g_conn));
3332
		exit_nicely();
B
Bruce Momjian 已提交
3333
	}
3334 3335 3336
	nlangs = PQntuples(res);

	if (nlangs != 1)
B
Bruce Momjian 已提交
3337
	{
3338
		write_msg(NULL, "procedural language for function %s not found\n", finfo[i].proname);
3339
		exit_nicely();
B
Bruce Momjian 已提交
3340 3341
	}

3342
	i_lanname = PQfnumber(res, "lanname");
3343

3344
	/*
B
Bruce Momjian 已提交
3345 3346
	 * See backend/commands/define.c for details of how the 'AS' clause is
	 * used.
3347 3348 3349
	 */
	if (strcmp(finfo[i].probin, "-") != 0)
	{
3350
		appendPQExpBuffer(asPart, "AS ");
3351
		formatStringLiteral(asPart, finfo[i].probin, CONV_ALL);
3352
		if (strcmp(finfo[i].prosrc, "-") != 0)
3353 3354
		{
			appendPQExpBuffer(asPart, ", ");
3355
			formatStringLiteral(asPart, finfo[i].prosrc, PASS_LFTAB);
3356
		}
3357 3358 3359 3360
	}
	else
	{
		if (strcmp(finfo[i].prosrc, "-") != 0)
3361 3362
		{
			appendPQExpBuffer(asPart, "AS ");
3363
			formatStringLiteral(asPart, finfo[i].prosrc, PASS_LFTAB);
3364
		}
3365 3366
	}

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

3369
	PQclear(res);
B
Bruce Momjian 已提交
3370

B
Bruce Momjian 已提交
3371 3372
	resetPQExpBuffer(fn);
	appendPQExpBuffer(fn, "%s (", fmtId(finfo[i].proname, force_quotes));
3373 3374
	for (j = 0; j < finfo[i].nargs; j++)
	{
B
Bruce Momjian 已提交
3375
		char	   *typname;
3376

3377
		typname = findTypeByOid(tinfo, numTypes, finfo[i].argtypes[j], zeroAsOpaque);
3378 3379
		if (typname == NULL)
		{
3380 3381
			write_msg(NULL, "WARNING: function \"%s\" not dumped\n",
					  finfo[i].proname);
3382

3383
			write_msg(NULL, "reason: data type name of argument %d (oid %s) not found\n",
3384
					   j, finfo[i].argtypes[j]);
3385
			goto done;
3386 3387
		}

B
Bruce Momjian 已提交
3388 3389 3390
		appendPQExpBuffer(fn, "%s%s",
						  (j > 0) ? "," : "",
						  typname);
B
Bruce,  
Bruce Momjian 已提交
3391
		appendPQExpBuffer(fnlist, "%s%s",
B
Bruce Momjian 已提交
3392 3393
						  (j > 0) ? "," : "",
						  typname);
3394
	}
B
Bruce Momjian 已提交
3395 3396 3397
	appendPQExpBuffer(fn, ")");

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

3400 3401 3402 3403
	rettypename = findTypeByOid(tinfo, numTypes, finfo[i].prorettype, zeroAsOpaque);

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

3407 3408
		write_msg(NULL, "reason: name of return data type (oid %s) not found\n",
				  finfo[i].prorettype);
3409
		goto done;
3410 3411
	}

B
Bruce Momjian 已提交
3412
	resetPQExpBuffer(q);
B
Bruce Momjian 已提交
3413
	appendPQExpBuffer(q, "CREATE FUNCTION %s ", fn->data);
3414 3415
	appendPQExpBuffer(q, "RETURNS %s%s %s LANGUAGE ",
					  (finfo[i].retset) ? "SETOF " : "",
3416
					  rettypename,
3417
					  asPart->data);
3418
	formatStringLiteral(q, func_lang, CONV_ALL);
3419

B
Bruce Momjian 已提交
3420
	if (finfo[i].iscachable || finfo[i].isstrict)		/* OR in new attrs here */
3421
	{
3422 3423 3424
		appendPQExpBuffer(q, " WITH (");
		listSep = listSepNone;

B
Bruce Momjian 已提交
3425 3426
		if (finfo[i].iscachable)
		{
3427 3428 3429 3430
			appendPQExpBuffer(q, "%s iscachable", listSep);
			listSep = listSepComma;
		}

B
Bruce Momjian 已提交
3431 3432
		if (finfo[i].isstrict)
		{
3433 3434 3435 3436
			appendPQExpBuffer(q, "%s isstrict", listSep);
			listSep = listSepComma;
		}
		appendPQExpBuffer(q, " )");
3437 3438 3439 3440
	}

	appendPQExpBuffer(q, ";\n");

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

B
Bruce,  
Bruce Momjian 已提交
3444 3445 3446 3447 3448 3449
	/*** Dump Function Comments ***/

	resetPQExpBuffer(q);
	appendPQExpBuffer(q, "FUNCTION %s ",
					  fmtId(finfo[i].proname, force_quotes));
	appendPQExpBuffer(q, "( %s )", fnlist->data);
3450
	dumpComment(fout, q->data, finfo[i].oid, "pg_proc", 0, NULL);
3451 3452 3453 3454 3455 3456 3457

done:
	destroyPQExpBuffer(q);
	destroyPQExpBuffer(fn);
	destroyPQExpBuffer(delqry);
	destroyPQExpBuffer(fnlist);
	destroyPQExpBuffer(asPart);
3458 3459 3460 3461
}

/*
 * dumpOprs
3462
 *	  writes out to fout the queries to recreate all the user-defined operators
3463 3464
 *
 */
3465
void
B
Bruce Momjian 已提交
3466
dumpOprs(Archive *fout, OprInfo *oprinfo, int numOperators,
3467
		 TypeInfo *tinfo, int numTypes)
3468
{
B
Bruce Momjian 已提交
3469
	int			i;
3470
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3471
	PQExpBuffer delq = createPQExpBuffer();
3472 3473 3474 3475 3476 3477 3478 3479
	PQExpBuffer leftarg = createPQExpBuffer();
	PQExpBuffer rightarg = createPQExpBuffer();
	PQExpBuffer commutator = createPQExpBuffer();
	PQExpBuffer negator = createPQExpBuffer();
	PQExpBuffer restrictor = createPQExpBuffer();
	PQExpBuffer join = createPQExpBuffer();
	PQExpBuffer sort1 = createPQExpBuffer();
	PQExpBuffer sort2 = createPQExpBuffer();
3480 3481 3482

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

B
Hi, all  
Bruce Momjian 已提交
3485 3486 3487 3488 3489 3490 3491 3492 3493
		resetPQExpBuffer(leftarg);
		resetPQExpBuffer(rightarg);
		resetPQExpBuffer(commutator);
		resetPQExpBuffer(negator);
		resetPQExpBuffer(restrictor);
		resetPQExpBuffer(join);
		resetPQExpBuffer(sort1);
		resetPQExpBuffer(sort2);

3494
		/* skip all the builtin oids */
3495
		if (atooid(oprinfo[i].oid) <= g_last_builtin_oid)
3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509 3510 3511
			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 已提交
3512 3513
			name = findTypeByOid(tinfo, numTypes,
								 oprinfo[i].oprleft, zeroAsOpaque);
3514 3515
			if (name == NULL)
			{
3516 3517 3518 3519
				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);
3520 3521
				continue;
			}
B
Bruce Momjian 已提交
3522
			appendPQExpBuffer(leftarg, ",\n\tLEFTARG = %s ", name);
3523
		}
3524

3525 3526 3527
		if (strcmp(oprinfo[i].oprkind, "l") == 0 ||
			strcmp(oprinfo[i].oprkind, "b") == 0)
		{
B
Bruce Momjian 已提交
3528 3529
			name = findTypeByOid(tinfo, numTypes,
								 oprinfo[i].oprright, zeroAsOpaque);
3530 3531
			if (name == NULL)
			{
3532 3533 3534 3535
				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);
3536 3537 3538
				continue;
			}
			appendPQExpBuffer(rightarg, ",\n\tRIGHTARG = %s ", name);
3539
		}
3540

B
Hi, all  
Bruce Momjian 已提交
3541
		if (!(strcmp(oprinfo[i].oprcom, "0") == 0))
3542 3543 3544 3545
		{
			name = findOprByOid(oprinfo, numOperators, oprinfo[i].oprcom);
			if (name == NULL)
			{
3546 3547 3548 3549
				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);
3550 3551 3552 3553
				continue;
			}
			appendPQExpBuffer(commutator, ",\n\tCOMMUTATOR = %s ", name);
		}
3554

B
Hi, all  
Bruce Momjian 已提交
3555
		if (!(strcmp(oprinfo[i].oprnegate, "0") == 0))
3556 3557 3558 3559
		{
			name = findOprByOid(oprinfo, numOperators, oprinfo[i].oprnegate);
			if (name == NULL)
			{
3560 3561 3562 3563
				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);
3564 3565 3566 3567
				continue;
			}
			appendPQExpBuffer(negator, ",\n\tNEGATOR = %s ", name);
		}
3568

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

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

B
Hi, all  
Bruce Momjian 已提交
3575
		if (!(strcmp(oprinfo[i].oprlsortop, "0") == 0))
3576
		{
B
Bruce Momjian 已提交
3577
			name = findOprByOid(oprinfo, numOperators, oprinfo[i].oprlsortop);
3578 3579
			if (name == NULL)
			{
3580 3581 3582 3583
				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);
3584 3585 3586 3587
				continue;
			}
			appendPQExpBuffer(sort1, ",\n\tSORT1 = %s ", name);
		}
3588

B
Hi, all  
Bruce Momjian 已提交
3589
		if (!(strcmp(oprinfo[i].oprrsortop, "0") == 0))
3590
		{
B
Bruce Momjian 已提交
3591
			name = findOprByOid(oprinfo, numOperators, oprinfo[i].oprrsortop);
3592 3593
			if (name == NULL)
			{
3594 3595 3596 3597
				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);
3598 3599 3600 3601
				continue;
			}
			appendPQExpBuffer(sort2, ",\n\tSORT2 = %s ", name);
		}
3602

B
Bruce Momjian 已提交
3603 3604
		resetPQExpBuffer(delq);
		appendPQExpBuffer(delq, "DROP OPERATOR %s (%s", oprinfo[i].oprname,
B
Bruce Momjian 已提交
3605
		findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft, zeroAsOpaque));
B
Bruce Momjian 已提交
3606
		appendPQExpBuffer(delq, ", %s);\n",
B
Bruce Momjian 已提交
3607
						  findTypeByOid(tinfo, numTypes, oprinfo[i].oprright, zeroAsOpaque));
3608

B
Hi, all  
Bruce Momjian 已提交
3609 3610
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
3611 3612 3613 3614 3615 3616 3617 3618 3619
						  "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 已提交
3620
		  (strcmp(oprinfo[i].oprcanhash, "t") == 0) ? ",\n\tHASHES" : "",
3621 3622 3623
						  join->data,
						  sort1->data,
						  sort2->data);
3624

B
Bruce Momjian 已提交
3625
		ArchiveEntry(fout, oprinfo[i].oid, oprinfo[i].oprname, "OPERATOR", NULL,
B
Bruce Momjian 已提交
3626
				q->data, delq->data, "", oprinfo[i].usename, NULL, NULL);
3627
	}
3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638

	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
	destroyPQExpBuffer(leftarg);
	destroyPQExpBuffer(rightarg);
	destroyPQExpBuffer(commutator);
	destroyPQExpBuffer(negator);
	destroyPQExpBuffer(restrictor);
	destroyPQExpBuffer(join);
	destroyPQExpBuffer(sort1);
	destroyPQExpBuffer(sort2);
3639 3640 3641 3642
}

/*
 * dumpAggs
3643
 *	  writes out to fout the queries to create all the user-defined aggregates
3644 3645 3646
 *
 */
void
B
Bruce Momjian 已提交
3647
dumpAggs(Archive *fout, AggInfo *agginfo, int numAggs,
3648
		 TypeInfo *tinfo, int numTypes)
3649
{
B
Bruce Momjian 已提交
3650
	int			i;
3651
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3652 3653
	PQExpBuffer delq = createPQExpBuffer();
	PQExpBuffer aggSig = createPQExpBuffer();
3654
	PQExpBuffer details = createPQExpBuffer();
3655 3656 3657

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

3660
		resetPQExpBuffer(details);
B
Hi, all  
Bruce Momjian 已提交
3661

3662
		/* skip all the builtin oids */
3663 3664 3665 3666 3667 3668 3669 3670 3671
		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)
		{
3672 3673 3674
			write_msg(NULL, "WARNING: aggregate function %s could not be dumped correctly for this database version; ignored\n",
					  aggSig->data);

3675
			resetPQExpBuffer(q);
3676
			appendPQExpBuffer(q, "-- WARNING: aggregate function %s could not be dumped correctly for this database version; ignored\n",
3677 3678 3679
									aggSig->data);
			ArchiveEntry(fout, agginfo[i].oid, aggSig->data, "WARNING", NULL,
					q->data, "" /* Del */, "", agginfo[i].usename, NULL, NULL);
3680
			continue;
3681
		}
3682

3683 3684 3685
		name = findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsAny + useBaseTypeName);
		if (name == NULL)
		{
3686 3687 3688 3689 3690 3691 3692 3693 3694 3695
			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);
3696 3697 3698
			continue;
		}
		appendPQExpBuffer(details, "BASETYPE = %s, ", name);
3699

3700 3701 3702
		name = findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype, zeroAsOpaque + useBaseTypeName);
		if (name == NULL)
		{
3703 3704 3705 3706 3707 3708 3709 3710 3711 3712
			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);
3713 3714
			continue;
		}
3715 3716
		appendPQExpBuffer(details,
						  "SFUNC = %s, STYPE = %s",
3717
						  agginfo[i].aggtransfn, name);
3718

3719
		if (agginfo[i].agginitval)
3720 3721
		{
			appendPQExpBuffer(details, ", INITCOND = ");
3722
			formatStringLiteral(details, agginfo[i].agginitval, CONV_ALL);
3723
		}
3724

B
Hi, all  
Bruce Momjian 已提交
3725
		if (!(strcmp(agginfo[i].aggfinalfn, "-") == 0))
3726 3727
			appendPQExpBuffer(details, ", FINALFUNC = %s",
							  agginfo[i].aggfinalfn);
3728

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

B
Hi, all  
Bruce Momjian 已提交
3732
		resetPQExpBuffer(q);
3733
		appendPQExpBuffer(q, "CREATE AGGREGATE %s ( %s );\n",
3734
						  agginfo[i].aggname,
3735
						  details->data);
3736

B
Bruce Momjian 已提交
3737
		ArchiveEntry(fout, agginfo[i].oid, aggSig->data, "AGGREGATE", NULL,
B
Bruce Momjian 已提交
3738
				q->data, delq->data, "", agginfo[i].usename, NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
3739 3740 3741 3742

		/*** Dump Aggregate Comments ***/

		resetPQExpBuffer(q);
3743
		appendPQExpBuffer(q, "AGGREGATE %s %s", agginfo[i].aggname,
3744
						  findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsOpaque + useBaseTypeName));
3745
		dumpComment(fout, q->data, agginfo[i].oid, "pg_aggregate", 0, NULL);
3746
	}
3747 3748 3749 3750 3751

	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
	destroyPQExpBuffer(aggSig);
	destroyPQExpBuffer(details);
3752 3753
}

3754 3755 3756 3757 3758
/*
 * These are some support functions to fix the acl problem of pg_dump
 *
 * Matthew C. Aycock 12/02/97
 */
3759 3760 3761

/* Append a keyword to a keyword list, inserting comma if needed.
 * Caller must make aclbuf big enough for all possible keywords.
3762
 */
3763
static void
B
Bruce Momjian 已提交
3764
AddAcl(char *aclbuf, const char *keyword)
3765
{
3766 3767 3768
	if (*aclbuf)
		strcat(aclbuf, ",");
	strcat(aclbuf, keyword);
3769
}
3770

3771
/*
3772
 * This will take a string of 'arwR' and return a malloced,
3773 3774
 * comma delimited string of SELECT,INSERT,UPDATE,DELETE,RULE
 */
V
Vadim B. Mikheev 已提交
3775
static char *
3776
GetPrivileges(Archive *AH, const char *s)
3777
{
3778
	char		aclbuf[100];
3779

3780
	aclbuf[0] = '\0';
3781

B
Bruce Momjian 已提交
3782
	if (strchr(s, 'a'))
3783
		AddAcl(aclbuf, "INSERT");
3784

B
Bruce Momjian 已提交
3785
	if (strchr(s, 'r'))
3786
		AddAcl(aclbuf, "SELECT");
3787

3788
	if (strchr(s, 'R'))
3789
		AddAcl(aclbuf, "RULE");
3790

3791 3792 3793 3794 3795 3796 3797 3798 3799 3800 3801 3802 3803 3804 3805 3806
	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");
	}
3807

3808
	return strdup(aclbuf);
3809
}
3810

B
Bruce Momjian 已提交
3811
/*
B
Bruce Momjian 已提交
3812
 * The name says it all; a function to append a string is the dest
B
Bruce Momjian 已提交
3813 3814
 * is big enough. If not, it does a realloc.
 */
B
Bruce Momjian 已提交
3815 3816
static void
strcatalloc(char **dest, int *dSize, char *src)
B
Bruce Momjian 已提交
3817
{
B
Bruce Momjian 已提交
3818 3819 3820 3821 3822 3823 3824 3825 3826 3827
	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 已提交
3828 3829


B
Bruce Momjian 已提交
3830 3831
/*
 * dumpACL:
3832 3833
 *	  Write out grant/revoke information
 *	  Called for sequences and tables
B
Bruce Momjian 已提交
3834 3835
 */

3836
static void
B
Bruce Momjian 已提交
3837
dumpACL(Archive *fout, TableInfo tbinfo)
B
Bruce Momjian 已提交
3838
{
B
Bruce Momjian 已提交
3839 3840
	const char *acls = tbinfo.relacl;
	char	   *aclbuf,
B
Bruce Momjian 已提交
3841 3842 3843
			   *tok,
			   *eqpos,
			   *priv;
3844
	char	   *objoid;
B
Bruce Momjian 已提交
3845
	char	   *sql;
B
Bruce Momjian 已提交
3846
	char		tmp[1024];
B
Bruce Momjian 已提交
3847
	int			sSize = 4096;
3848 3849 3850 3851

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

B
Bruce Momjian 已提交
3852 3853 3854
	/*
	 * Allocate a larginsh buffer for the output SQL.
	 */
B
Bruce Momjian 已提交
3855
	sql = (char *) malloc(sSize);
B
Bruce Momjian 已提交
3856

B
Bruce Momjian 已提交
3857 3858 3859
	/*
	 * Revoke Default permissions for PUBLIC. Is this actually necessary,
	 * or is it just a waste of time?
3860
	 */
B
Bruce Momjian 已提交
3861
	sprintf(sql, "REVOKE ALL on %s from PUBLIC;\n",
3862
			fmtId(tbinfo.relname, force_quotes));
3863 3864 3865

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

3867 3868
	/* Scan comma-separated ACL items */
	for (tok = strtok(aclbuf, ","); tok != NULL; tok = strtok(NULL, ","))
3869
	{
3870

B
Bruce Momjian 已提交
3871 3872 3873
		/*
		 * Token may start with '{' and/or '"'.  Actually only the start
		 * of the string should have '{', but we don't verify that.
3874 3875 3876 3877 3878 3879 3880 3881
		 */
		if (*tok == '{')
			tok++;
		if (*tok == '"')
			tok++;

		/* User name is string up to = in tok */
		eqpos = strchr(tok, '=');
B
Bruce Momjian 已提交
3882
		if (!eqpos)
B
Bruce Momjian 已提交
3883
		{
3884 3885 3886
			write_msg(NULL, "could not parse ACL list ('%s') for relation %s\n",
					  acls, tbinfo.relname);
			exit_nicely();
B
Bruce Momjian 已提交
3887 3888
		}

B
Bruce Momjian 已提交
3889 3890 3891 3892
		/*
		 * Parse the privileges (right-hand side).	Skip if there are
		 * none.
		 */
3893
		priv = GetPrivileges(fout, eqpos + 1);
3894
		if (*priv)
3895
		{
B
Bruce Momjian 已提交
3896
			sprintf(tmp, "GRANT %s on %s to ",
3897
					priv, fmtId(tbinfo.relname, force_quotes));
B
Bruce Momjian 已提交
3898
			strcatalloc(&sql, &sSize, tmp);
B
Bruce Momjian 已提交
3899 3900 3901

			/*
			 * Note: fmtId() can only be called once per printf, so don't
3902 3903 3904 3905 3906
			 * try to merge printing of username into the above printf.
			 */
			if (eqpos == tok)
			{
				/* Empty left-hand side means "PUBLIC" */
B
Bruce Momjian 已提交
3907
				strcatalloc(&sql, &sSize, "PUBLIC;\n");
3908
			}
3909
			else
3910 3911
			{
				*eqpos = '\0';	/* it's ok to clobber aclbuf */
3912
				if (strncmp(tok, "group ", strlen("group ")) == 0)
B
Bruce Momjian 已提交
3913
					sprintf(tmp, "GROUP %s;\n",
3914 3915
							fmtId(tok + strlen("group "), force_quotes));
				else
B
Bruce Momjian 已提交
3916 3917
					sprintf(tmp, "%s;\n", fmtId(tok, force_quotes));
				strcatalloc(&sql, &sSize, tmp);
3918
			}
3919
		}
3920
		free(priv);
B
Bruce Momjian 已提交
3921
	}
3922 3923

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

3925 3926 3927 3928 3929 3930
	if (tbinfo.viewdef != NULL)
		objoid = tbinfo.viewoid;
	else
		objoid = tbinfo.oid;

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

B
Bruce Momjian 已提交
3932 3933
}

3934 3935 3936 3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950 3951 3952 3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985 3986 3987 3988
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));
	}
}
3989

3990 3991
/*
 * dumpTables:
3992
 *	  write out to fout all the user-define tables
3993
 */
3994

3995
void
B
Bruce Momjian 已提交
3996
dumpTables(Archive *fout, TableInfo *tblinfo, int numTables,
3997
		   IndInfo *indinfo, int numIndexes,
3998 3999
		   InhInfo *inhinfo, int numInherits,
		   TypeInfo *tinfo, int numTypes, const char *tablename,
B
Bruce Momjian 已提交
4000 4001
		   const bool aclsSkip, const bool oids,
		   const bool schemaOnly, const bool dataOnly)
4002
{
4003 4004
	int			i,
				j,
B
Bruce Momjian 已提交
4005
				k;
4006
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
4007
	PQExpBuffer delq = createPQExpBuffer();
4008
	char	   *serialSeq = NULL;		/* implicit sequence name created
B
Bruce Momjian 已提交
4009 4010 4011 4012
										 * by SERIAL datatype */
	const char *serialSeqSuffix = "_id_seq";	/* suffix for implicit
												 * SERIAL sequences */
	char	  **parentRels;		/* list of names of parent relations */
4013
	int			numParents;
B
Bruce Momjian 已提交
4014
	int			actual_atts;	/* number of attrs in this CREATE statment */
4015
	char	   *reltypename;
4016
	char	   *objoid;
4017
	const char *((*commentDeps)[]);
4018

V
Vadim B. Mikheev 已提交
4019
	/* First - dump SEQUENCEs */
4020
	if (tablename && strlen(tablename) > 0)
B
Bruce Momjian 已提交
4021
	{
4022 4023
		/* XXX this code only works for serial columns named "id" */
		/* We really need dependency analysis! */
B
Bruce Momjian 已提交
4024 4025 4026 4027
		serialSeq = malloc(strlen(tablename) + strlen(serialSeqSuffix) + 1);
		strcpy(serialSeq, tablename);
		strcat(serialSeq, serialSeqSuffix);
	}
V
Vadim B. Mikheev 已提交
4028 4029 4030 4031
	for (i = 0; i < numTables; i++)
	{
		if (!(tblinfo[i].sequence))
			continue;
B
Bruce Momjian 已提交
4032
		if (!tablename || (!strcmp(tblinfo[i].relname, tablename))
B
Bruce Momjian 已提交
4033
			|| (serialSeq && !strcmp(tblinfo[i].relname, serialSeq)))
V
Vadim B. Mikheev 已提交
4034
		{
B
Bruce Momjian 已提交
4035
			/* becomeUser(fout, tblinfo[i].usename); */
4036
			dumpSequence(fout, tblinfo[i], schemaOnly, dataOnly);
4037
			if (!aclsSkip)
B
Bruce Momjian 已提交
4038
				dumpACL(fout, tblinfo[i]);
V
Vadim B. Mikheev 已提交
4039 4040
		}
	}
4041
	if (serialSeq)
B
Bruce Momjian 已提交
4042
		free(serialSeq);
4043

4044 4045
	for (i = 0; i < numTables; i++)
	{
4046
		if (tblinfo[i].sequence)/* already dumped */
V
Vadim B. Mikheev 已提交
4047
			continue;
4048

4049
		if (!tablename || (!strcmp(tblinfo[i].relname, tablename)) || (strlen(tablename) == 0))
4050 4051
		{

4052 4053
			resetPQExpBuffer(delq);
			resetPQExpBuffer(q);
4054

4055 4056 4057 4058
			/* Use the view definition if there is one */
			if (tblinfo[i].viewdef != NULL)
			{
				reltypename = "VIEW";
4059
				objoid = tblinfo[i].viewoid;
4060
				appendPQExpBuffer(delq, "DROP VIEW %s;\n", fmtId(tblinfo[i].relname, force_quotes));
4061
				appendPQExpBuffer(q, "CREATE VIEW %s as %s\n", fmtId(tblinfo[i].relname, force_quotes), tblinfo[i].viewdef);
4062 4063 4064
				commentDeps = malloc(sizeof(char*) * 2);
				(*commentDeps)[0] = strdup(objoid);
				(*commentDeps)[1] = NULL; /* end of list */
4065 4066
			}
			else
4067
			{
4068
				reltypename = "TABLE";
4069
				objoid = tblinfo[i].oid;
4070
				commentDeps = NULL;
4071 4072
				parentRels = tblinfo[i].parentRels;
				numParents = tblinfo[i].numParents;
B
Bruce Momjian 已提交
4073

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

4076 4077 4078 4079 4080 4081
				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)
4082
					{
4083 4084 4085 4086 4087
						/* Format properly if not first attr */
						if (actual_atts > 0)
							appendPQExpBuffer(q, ",\n\t");

						/* Attr name & type */
4088 4089 4090 4091 4092 4093 4094 4095
						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);
						}
4096 4097

						/* Default value */
4098
						if (tblinfo[i].adef_expr[j] != NULL && tblinfo[i].inhAttrDef[j] == 0)
4099
							appendPQExpBuffer(q, " DEFAULT %s",
B
Bruce Momjian 已提交
4100
											  tblinfo[i].adef_expr[j]);
4101 4102

						/* Not Null constraint */
4103
						if (tblinfo[i].notnull[j] && tblinfo[i].inhNotNull[j] == 0)
4104 4105 4106 4107
							appendPQExpBuffer(q, " NOT NULL");

						actual_atts++;
					}
4108
				}
B
Hi all  
Bruce Momjian 已提交
4109

4110 4111


4112 4113 4114 4115 4116 4117 4118
				/* 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 已提交
4119
									  tblinfo[i].check_expr[k]);
4120
				}
4121

4122 4123
				/* Primary Key */
				if (tblinfo[i].pkIndexOid != NULL)
4124
				{
B
Bruce Momjian 已提交
4125
					PQExpBuffer consDef;
4126 4127

					/* Find the corresponding index */
4128
					for (k = 0; k < numIndexes; k++)
4129
					{
4130 4131
						if (strcmp(indinfo[k].indexreloid,
								   tblinfo[i].pkIndexOid) == 0)
4132 4133 4134
							break;
					}

4135
					if (k >= numIndexes)
4136
					{
4137 4138
						write_msg(NULL, "dumpTables(): failed sanity check, could not find index (%s) for primary key constraint\n",
								   tblinfo[i].pkIndexOid);
4139
						exit_nicely();
4140 4141 4142 4143
					}

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

B
Bruce Momjian 已提交
4144
					if ((actual_atts + tblinfo[i].ncheck) > 0)
4145
						appendPQExpBuffer(q, ",\n\t");
4146 4147 4148 4149

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

					destroyPQExpBuffer(consDef);
4150
				}
B
Bruce Momjian 已提交
4151

4152

4153
				appendPQExpBuffer(q, "\n)");
4154

4155
				if (numParents > 0)
B
Hi all  
Bruce Momjian 已提交
4156
				{
4157
					appendPQExpBuffer(q, "\nINHERITS (");
4158 4159 4160
					for (k = 0; k < numParents; k++)
					{
						appendPQExpBuffer(q, "%s%s",
B
Bruce Momjian 已提交
4161 4162
										  (k > 0) ? ", " : "",
									 fmtId(parentRels[k], force_quotes));
4163 4164
					}
					appendPQExpBuffer(q, ")");
B
Hi all  
Bruce Momjian 已提交
4165 4166
				}

4167 4168 4169
				if (!tblinfo[i].hasoids)
					appendPQExpBuffer(q, " WITHOUT OIDS");

4170 4171
				appendPQExpBuffer(q, ";\n");
			}
B
Bruce Momjian 已提交
4172

B
Bruce Momjian 已提交
4173 4174
			if (!dataOnly)
			{
4175

4176
				ArchiveEntry(fout, objoid, tblinfo[i].relname,
B
Bruce Momjian 已提交
4177 4178
							 reltypename, NULL, q->data, delq->data, "", tblinfo[i].usename,
							 NULL, NULL);
B
Bruce Momjian 已提交
4179

4180 4181 4182 4183
				if (!aclsSkip)
					dumpACL(fout, tblinfo[i]);

			}
4184

4185
			/* Dump Field Comments */
B
Bruce,  
Bruce Momjian 已提交
4186

4187 4188
			for (j = 0; j < tblinfo[i].numatts; j++)
			{
B
Bruce,  
Bruce Momjian 已提交
4189 4190 4191 4192
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "COLUMN %s", fmtId(tblinfo[i].relname, force_quotes));
				appendPQExpBuffer(q, ".");
				appendPQExpBuffer(q, "%s", fmtId(tblinfo[i].attnames[j], force_quotes));
4193 4194
				dumpComment(fout, q->data, tblinfo[i].oid,
							"pg_class", j+1, commentDeps);
B
Bruce,  
Bruce Momjian 已提交
4195
			}
4196

B
Bruce,  
Bruce Momjian 已提交
4197
			/* Dump Table Comments */
4198

B
Bruce,  
Bruce Momjian 已提交
4199
			resetPQExpBuffer(q);
4200
			appendPQExpBuffer(q, "%s %s", reltypename, fmtId(tblinfo[i].relname, force_quotes));
4201 4202
			dumpComment(fout, q->data, tblinfo[i].oid,
						"pg_class", 0, commentDeps);
4203

4204 4205
		}
	}
4206 4207 4208

	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
4209 4210
}

B
Bruce Momjian 已提交
4211 4212
static PQExpBuffer
getPKconstraint(TableInfo *tblInfo, IndInfo *indInfo)
4213
{
B
Bruce Momjian 已提交
4214 4215
	PQExpBuffer pkBuf = createPQExpBuffer();
	int			k;
4216 4217

	appendPQExpBuffer(pkBuf, "Constraint %s Primary Key (",
B
Bruce Momjian 已提交
4218
					  tblInfo->primary_key_name);
4219 4220 4221

	for (k = 0; k < INDEX_MAX_KEYS; k++)
	{
4222 4223
		int			indkey;
		const char *attname;
4224 4225 4226 4227

		indkey = atoi(indInfo->indkey[k]);
		if (indkey == InvalidAttrNumber)
			break;
4228
		attname = getAttrName(indkey, tblInfo);
4229 4230

		appendPQExpBuffer(pkBuf, "%s%s",
B
Bruce Momjian 已提交
4231 4232
						  (k == 0) ? "" : ", ",
						  fmtId(attname, force_quotes));
4233 4234 4235 4236 4237 4238 4239
	}

	appendPQExpBuffer(pkBuf, ")");

	return pkBuf;
}

4240 4241 4242 4243 4244 4245 4246 4247 4248 4249 4250 4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267 4268
/*
 * 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";
	}
4269 4270
	write_msg(NULL, "getAttrName(): invalid column number %d for table %s\n",
			   attrnum, tblInfo->relname);
4271
	exit_nicely();
4272 4273 4274
	return NULL;				/* keep compiler quiet */
}

4275
/*
4276
 * dumpIndexes:
4277
 *	  write out to fout all the user-defined indexes
4278
 */
4279
void
4280
dumpIndexes(Archive *fout, IndInfo *indinfo, int numIndexes,
4281
			TableInfo *tblinfo, int numTables, const char *tablename)
4282
{
4283 4284 4285
	int			i,
				k;
	int			tableInd;
4286
	PQExpBuffer attlist = createPQExpBuffer();
4287 4288 4289 4290
	PQExpBuffer q = createPQExpBuffer();
	PQExpBuffer delq = createPQExpBuffer();
	PQExpBuffer id1 = createPQExpBuffer();
	PQExpBuffer id2 = createPQExpBuffer();
B
Bruce Momjian 已提交
4291 4292 4293
	char	   *classname[INDEX_MAX_KEYS];
	char	   *funcname;		/* the name of the function to comput the
								 * index key from */
4294
	Oid			indclass;
4295 4296
	int			nclass;
	PGresult   *res;
4297

4298
	for (i = 0; i < numIndexes; i++)
4299 4300
	{
		tableInd = findTableByName(tblinfo, numTables,
4301 4302 4303
								   indinfo[i].indrelname);
		if (tableInd < 0)
		{
4304 4305
			write_msg(NULL, "dumpIndexes(): failed sanity check, table %s was not found\n",
					   indinfo[i].indrelname);
4306
			exit_nicely();
4307
		}
4308

4309 4310 4311
		/* Handle PK indexes */
		if (strcmp(indinfo[i].indisprimary, "t") == 0)
		{
4312 4313 4314 4315 4316 4317 4318 4319 4320 4321 4322 4323 4324 4325 4326 4327 4328
#if 0
			/* PK: Enable this code when ALTER TABLE supports PK constraints. */

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

			resetPQExpBuffer(attlist);

			appendPQExpBuffer(attlist, "Alter Table %s Add %s;",
							  fmtId(tblinfo[tableInd].relname, force_quotes),
							  consDef->data);

			ArchiveEntry(fout, indinfo[i].oid, tblinfo[tableInd].primary_key_name,
						 "CONSTRAINT", NULL, attlist->data, "",
						 "", tblinfo[tableInd].usename, NULL, NULL);

			destroyPQExpBuffer(consDef);
#endif
B
Bruce Momjian 已提交
4329 4330 4331 4332 4333

			/*
			 * Don't need to do anything else for this system-generated
			 * index
			 */
4334 4335 4336 4337
			continue;
		}


4338 4339 4340 4341
		if (strcmp(indinfo[i].indproc, "0") == 0)
			funcname = NULL;
		else
		{
B
Bruce Momjian 已提交
4342
			int			numFuncs;
4343 4344 4345 4346 4347 4348 4349

			/*
			 * 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*
			 */
B
Hi, all  
Bruce Momjian 已提交
4350 4351
			resetPQExpBuffer(q);
			appendPQExpBuffer(q,
4352 4353 4354
							  "SELECT proname from pg_proc "
							  "where pg_proc.oid = '%s'::oid",
							  indinfo[i].indproc);
B
Hi, all  
Bruce Momjian 已提交
4355
			res = PQexec(g_conn, q->data);
4356 4357
			if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
			{
4358 4359
				write_msg(NULL, "query to get function name of oid %s failed: %s",
						  indinfo[i].indproc, PQerrorMessage(g_conn));
4360
				exit_nicely();
4361
			}
4362 4363 4364

			/* Sanity: Check we got only one tuple */
			numFuncs = PQntuples(res);
B
Bruce Momjian 已提交
4365 4366
			if (numFuncs != 1)
			{
4367 4368
				write_msg(NULL, "query to get function name of oid %s returned %d rows; expected 1\n",
						  indinfo[i].indproc, numFuncs);
4369
				exit_nicely();
4370 4371
			}

B
Bruce Momjian 已提交
4372
			funcname = strdup(PQgetvalue(res, 0, PQfnumber(res, "proname")));
4373 4374 4375 4376 4377 4378
			PQclear(res);
		}

		/* convert opclass oid(s) into names */
		for (nclass = 0; nclass < INDEX_MAX_KEYS; nclass++)
		{
B
Bruce Momjian 已提交
4379
			int			numRows;
4380

4381
			indclass = atooid(indinfo[i].indclass[nclass]);
4382 4383
			if (indclass == 0)
				break;
B
Hi, all  
Bruce Momjian 已提交
4384 4385
			resetPQExpBuffer(q);
			appendPQExpBuffer(q,
4386 4387 4388
							  "SELECT opcname from pg_opclass "
							  "where pg_opclass.oid = '%u'::oid",
							  indclass);
B
Hi, all  
Bruce Momjian 已提交
4389
			res = PQexec(g_conn, q->data);
4390 4391
			if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
			{
4392 4393
				write_msg(NULL, "query to get operator class name of oid %u failed: %s",
						  indclass, PQerrorMessage(g_conn));
4394
				exit_nicely();
4395
			}
4396 4397 4398

			/* Sanity: Check we got only one tuple */
			numRows = PQntuples(res);
B
Bruce Momjian 已提交
4399 4400
			if (numRows != 1)
			{
4401 4402
				write_msg(NULL, "query to get operator class name of oid %u returned %d rows; expected 1\n",
						  indclass, numRows);
4403
				exit_nicely();
4404 4405
			}

B
Bruce Momjian 已提交
4406
			classname[nclass] = strdup(PQgetvalue(res, 0, PQfnumber(res, "opcname")));
4407 4408 4409 4410 4411
			PQclear(res);
		}

		if (funcname && nclass != 1)
		{
4412 4413
			write_msg(NULL, "There must be exactly one OpClass for functional index \"%s\".\n",
					  indinfo[i].indexrelname);
4414
			exit_nicely();
4415 4416 4417
		}

		/* convert attribute numbers into attribute list */
B
Hi, all  
Bruce Momjian 已提交
4418 4419
		resetPQExpBuffer(attlist);
		for (k = 0; k < INDEX_MAX_KEYS; k++)
4420
		{
4421 4422
			int			indkey;
			const char *attname;
4423 4424

			indkey = atoi(indinfo[i].indkey[k]);
B
Bruce Momjian 已提交
4425
			if (indkey == InvalidAttrNumber)
4426
				break;
4427
			attname = getAttrName(indkey, &tblinfo[tableInd]);
4428
			if (funcname)
B
Hi, all  
Bruce Momjian 已提交
4429
				appendPQExpBuffer(attlist, "%s%s",
4430 4431
								  (k == 0) ? "" : ", ",
								  fmtId(attname, force_quotes));
4432 4433 4434 4435
			else
			{
				if (k >= nclass)
				{
4436
					write_msg(NULL, "no operator class found for column \"%s\" of index \"%s\"\n",
4437
							attname, indinfo[i].indexrelname);
4438
					exit_nicely();
4439
				}
B
Hi, all  
Bruce Momjian 已提交
4440 4441 4442 4443 4444
				resetPQExpBuffer(id1);
				resetPQExpBuffer(id2);
				appendPQExpBuffer(id1, fmtId(attname, force_quotes));
				appendPQExpBuffer(id2, fmtId(classname[k], force_quotes));
				appendPQExpBuffer(attlist, "%s%s %s",
4445 4446
								  (k == 0) ? "" : ", ",
								  id1->data, id2->data);
4447 4448 4449 4450
				free(classname[k]);
			}
		}

B
Bruce Momjian 已提交
4451
		if (!tablename || (strcmp(indinfo[i].indrelname, tablename) == 0) || (strlen(tablename) == 0))
4452
		{
B
Hi, all  
Bruce Momjian 已提交
4453 4454 4455 4456
			resetPQExpBuffer(id1);
			resetPQExpBuffer(id2);
			appendPQExpBuffer(id1, fmtId(indinfo[i].indexrelname, force_quotes));
			appendPQExpBuffer(id2, fmtId(indinfo[i].indrelname, force_quotes));
4457

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

B
Bruce Momjian 已提交
4461 4462
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "CREATE %s INDEX %s on %s using %s (",
B
Bruce Momjian 已提交
4463 4464 4465 4466
			  (strcmp(indinfo[i].indisunique, "t") == 0) ? "UNIQUE" : "",
							  id1->data,
							  id2->data,
							  indinfo[i].indamname);
4467 4468
			if (funcname)
			{
4469
				/* need 2 printf's here cuz fmtId has static return area */
B
Bruce Momjian 已提交
4470
				appendPQExpBuffer(q, " %s", fmtId(funcname, false));
4471
				appendPQExpBuffer(q, " (%s) %s )", attlist->data,
B
Bruce Momjian 已提交
4472
								  fmtId(classname[0], force_quotes));
4473 4474 4475 4476
				free(funcname);
				free(classname[0]);
			}
			else
4477 4478
				appendPQExpBuffer(q, " %s )", attlist->data);
				
4479
			if (strcmp(indinfo[i].indhaspred, "t") == 0)
4480
			{
4481
				/* There is an index predicate, so fetch and dump it */
4482 4483 4484
				int numRows;
				PQExpBuffer pred = createPQExpBuffer();
				
4485 4486
				appendPQExpBuffer(pred, "SELECT pg_get_expr(indpred,indrelid) as pred FROM pg_index WHERE indexrelid = %s",
								  indinfo[i].indexreloid);
4487 4488 4489 4490 4491 4492 4493 4494 4495 4496 4497 4498 4499 4500 4501 4502 4503 4504 4505 4506 4507 4508 4509 4510 4511
				res = PQexec(g_conn, pred->data);
				if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
				{
					fprintf(stderr, "dumpIndices(): SELECT (indpred) failed.  "
							"Explanation from backend: '%s'.\n",
							PQerrorMessage(g_conn));
					exit_nicely();
				}
	
				/* Sanity: Check we got only one tuple */
				numRows = PQntuples(res);
				if (numRows != 1)
				{
					fprintf(stderr, "dumpIndices(): SELECT (indpred) for index %s returned %d tuples. Expected 1.\n",
							indinfo[i].indrelname, numRows);
					exit_nicely();
				}
	
				appendPQExpBuffer(q, " WHERE %s",
								  PQgetvalue(res, 0, PQfnumber(res, "pred")));

				PQclear(res);
				destroyPQExpBuffer( pred );
			}
			appendPQExpBuffer(q, ";\n");
B
Bruce,  
Bruce Momjian 已提交
4512

4513 4514 4515 4516 4517
			/*
			 * 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.
			 */
4518 4519
			ArchiveEntry(fout, indinfo[i].indexreloid, id1->data,
						 "INDEX", NULL, q->data, delq->data,
B
Bruce Momjian 已提交
4520
						 "", tblinfo[tableInd].usename, NULL, NULL);
B
Bruce Momjian 已提交
4521

4522
			/* Dump Index Comments */
B
Bruce,  
Bruce Momjian 已提交
4523 4524
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "INDEX %s", id1->data);
4525 4526
			dumpComment(fout, q->data, indinfo[i].indexreloid,
						"pg_class", 0, NULL);
4527

4528 4529
		}
	}
4530

4531 4532 4533 4534 4535
	destroyPQExpBuffer(attlist);
	destroyPQExpBuffer(q);
	destroyPQExpBuffer(delq);
	destroyPQExpBuffer(id1);
	destroyPQExpBuffer(id2);
4536
}
4537

4538
/*
B
Bruce Momjian 已提交
4539
 * dumpTuples
4540 4541 4542 4543 4544
 *	  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
4545
 *	  in PostgreSQL.
4546 4547 4548 4549
 *
 * the attrmap passed in tells how to map the attributes copied in to the
 * attributes copied out
 */
4550
#ifdef NOT_USED
4551
void
4552
dumpTuples(PGresult *res, FILE *fout, int *attrmap)
4553
{
4554 4555 4556 4557
	int			j,
				k;
	int			m,
				n;
B
Bruce Momjian 已提交
4558
	char	  **outVals = NULL; /* values to copy out */
4559 4560 4561 4562 4563 4564 4565 4566 4567 4568 4569 4570 4571 4572 4573 4574 4575 4576 4577

	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++)
			{
4578
				char	   *pval = outVals[k];
4579 4580 4581 4582 4583 4584 4585 4586 4587 4588 4589 4590 4591 4592 4593 4594 4595 4596 4597 4598

				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);
	}
4599
}
4600

4601
#endif
4602

4603 4604 4605 4606
/*
 * setMaxOid -
 * find the maximum oid and generate a COPY statement to set it
*/
4607

4608
static void
B
Bruce Momjian 已提交
4609
setMaxOid(Archive *fout)
4610
{
B
Bruce Momjian 已提交
4611 4612
	PGresult   *res;
	Oid			max_oid;
B
Bruce Momjian 已提交
4613
	char		sql[1024];
4614

4615
	res = PQexec(g_conn, "CREATE TEMPORARY TABLE pgdump_oid (dummy int4)");
4616 4617 4618
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
4619
		write_msg(NULL, "could not create pgdump_oid table: %s", PQerrorMessage(g_conn));
4620
		exit_nicely();
4621 4622
	}
	PQclear(res);
4623
	res = PQexec(g_conn, "INSERT INTO pgdump_oid VALUES (0)");
4624 4625 4626
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
4627
		write_msg(NULL, "could not insert into pgdump_oid table: %s", PQerrorMessage(g_conn));
4628
		exit_nicely();
4629
	}
4630
	max_oid = PQoidValue(res);
4631 4632
	if (max_oid == 0)
	{
4633
		write_msg(NULL, "inserted invalid oid\n");
4634
		exit_nicely();
4635 4636
	}
	PQclear(res);
4637
	res = PQexec(g_conn, "DROP TABLE pgdump_oid;");
4638 4639 4640
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
4641
		write_msg(NULL, "could not drop pgdump_oid table: %s", PQerrorMessage(g_conn));
4642
		exit_nicely();
4643 4644 4645
	}
	PQclear(res);
	if (g_verbose)
4646
		write_msg(NULL, "maximum system oid is %u\n", max_oid);
4647 4648 4649
	snprintf(sql, 1024,
			 "CREATE TEMPORARY TABLE pgdump_oid (dummy int4);\n"
			 "COPY pgdump_oid WITH OIDS FROM stdin;\n"
4650
			 "%u\t0\n"
4651 4652 4653
			 "\\.\n"
			 "DROP TABLE pgdump_oid;\n",
			 max_oid);
B
Bruce Momjian 已提交
4654

4655
	ArchiveEntry(fout, "0", "Max OID", "<Init>", NULL, sql, "", "", "", NULL, NULL);
4656
}
4657 4658 4659

/*
 * findLastBuiltInOid -
4660
 * find the last built in oid
4661 4662
 * we do this by retrieving datlastsysoid from the pg_database entry for this database,
 */
4663

4664
static Oid
4665
findLastBuiltinOid_V71(const char *dbname)
4666
{
B
Bruce Momjian 已提交
4667
	PGresult   *res;
4668
	int			ntups;
4669
	Oid			last_oid;
4670 4671 4672
	PQExpBuffer query = createPQExpBuffer();

	resetPQExpBuffer(query);
4673
	appendPQExpBuffer(query, "SELECT datlastsysoid from pg_database where datname = ");
4674
	formatStringLiteral(query, dbname, CONV_ALL);
4675

4676
	res = PQexec(g_conn, query->data);
4677 4678 4679
	if (res == NULL ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
4680
		write_msg(NULL, "error in finding the last system oid: %s", PQerrorMessage(g_conn));
4681
		exit_nicely();
4682 4683
	}
	ntups = PQntuples(res);
4684
	if (ntups < 1)
4685
	{
4686
		write_msg(NULL, "missing pg_database entry for this database\n");
4687
		exit_nicely();
4688 4689 4690
	}
	if (ntups > 1)
	{
4691
		write_msg(NULL, "found more than one pg_database entry for this database\n");
4692
		exit_nicely();
4693
	}
4694
	last_oid = atooid(PQgetvalue(res, 0, PQfnumber(res, "datlastsysoid")));
4695
	PQclear(res);
4696
	destroyPQExpBuffer(query);
4697
	return last_oid;
4698 4699
}

4700 4701 4702 4703 4704 4705 4706
/*
 * 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
*/

4707
static Oid
4708 4709 4710 4711 4712 4713 4714 4715 4716 4717 4718
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)
	{
4719
		write_msg(NULL, "error in finding the template1 database: %s", PQerrorMessage(g_conn));
4720
		exit_nicely();
4721 4722 4723 4724
	}
	ntups = PQntuples(res);
	if (ntups < 1)
	{
4725
		write_msg(NULL, "could not find template1 database entry in the pg_database table\n");
4726
		exit_nicely();
4727 4728 4729
	}
	if (ntups > 1)
	{
4730
		write_msg(NULL, "found more than one template1 database entry in the pg_database table\n");
4731
		exit_nicely();
4732 4733 4734 4735 4736
	}
	last_oid = atooid(PQgetvalue(res, 0, PQfnumber(res, "oid")));
	PQclear(res);
	return last_oid;
}
4737

4738
static void
4739
dumpSequence(Archive *fout, TableInfo tbinfo, const bool schemaOnly, const bool dataOnly)
4740
{
4741
	PGresult   *res;
4742 4743 4744 4745 4746 4747
	char	   *last,
			   *incby,
			   *maxv,
			   *minv,
			   *cache;
	bool		cycled,
B
Bruce Momjian 已提交
4748
				called;
4749
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
4750
	PQExpBuffer delqry = createPQExpBuffer();
4751

B
Hi, all  
Bruce Momjian 已提交
4752
	appendPQExpBuffer(query,
4753
			"SELECT sequence_name, last_value, increment_by, max_value, "
4754 4755
				  "min_value, cache_value, is_cycled, is_called from %s",
					  fmtId(tbinfo.relname, force_quotes));
4756

B
Hi, all  
Bruce Momjian 已提交
4757
	res = PQexec(g_conn, query->data);
4758 4759
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
4760
		write_msg(NULL, "query to get data of sequence \"%s\" failed: %s", tbinfo.relname, PQerrorMessage(g_conn));
4761
		exit_nicely();
4762 4763 4764 4765
	}

	if (PQntuples(res) != 1)
	{
4766 4767
		write_msg(NULL, "query to get data of sequence \"%s\" returned %d rows (expected 1)\n",
				   tbinfo.relname, PQntuples(res));
4768
		exit_nicely();
4769 4770 4771 4772
	}

	if (strcmp(PQgetvalue(res, 0, 0), tbinfo.relname) != 0)
	{
4773 4774
		write_msg(NULL, "query to get data of sequence \"%s\" returned name \"%s\"\n",
				   tbinfo.relname, PQgetvalue(res, 0, 0));
4775
		exit_nicely();
4776 4777
	}

4778 4779 4780 4781 4782 4783 4784
	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);
4785

4786
	/*
B
Bruce Momjian 已提交
4787 4788
	 * The logic we use for restoring sequences is as follows: -   Add a
	 * basic CREATE SEQUENCE statement (use last_val for start if called
4789
	 * is false, else use min_val for start_val).
4790 4791 4792
	 *
	 *	Add a 'SETVAL(seq, last_val, iscalled)' at restore-time iff
	 *  we load data
4793
	 */
4794

4795 4796 4797
	if (!dataOnly)
	{
		resetPQExpBuffer(delqry);
4798 4799
		appendPQExpBuffer(delqry, "DROP SEQUENCE %s;\n",
						  fmtId(tbinfo.relname, force_quotes));
4800

4801 4802
		resetPQExpBuffer(query);
		appendPQExpBuffer(query,
4803 4804
						  "CREATE SEQUENCE %s start %s increment %s "
						  "maxvalue %s minvalue %s cache %s%s;\n",
B
Bruce Momjian 已提交
4805
						  fmtId(tbinfo.relname, force_quotes),
4806
						  (called ? minv : last),
4807
						  incby, maxv, minv, cache,
4808
						  (cycled ? " cycle" : ""));
4809

P
Philip Warner 已提交
4810
		ArchiveEntry(fout, tbinfo.oid, tbinfo.relname, "SEQUENCE", NULL,
4811 4812
					 query->data, delqry->data, "", tbinfo.usename,
					 NULL, NULL);
4813
	}
4814

4815 4816 4817 4818 4819
	if (!schemaOnly)
	{
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "SELECT setval (");
		formatStringLiteral(query, fmtId(tbinfo.relname, force_quotes), CONV_ALL);
4820 4821
		appendPQExpBuffer(query, ", %s, %s);\n",
						  last, (called ? "true" : "false"));
4822

P
Philip Warner 已提交
4823
		ArchiveEntry(fout, tbinfo.oid, tbinfo.relname, "SEQUENCE SET", NULL,
B
Bruce Momjian 已提交
4824
					 query->data, "" /* Del */ , "", "", NULL, NULL);
4825
	}
B
Bruce,  
Bruce Momjian 已提交
4826

4827 4828 4829
	if (!dataOnly)
	{
		/* Dump Sequence Comments */
B
Bruce,  
Bruce Momjian 已提交
4830

4831 4832
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "SEQUENCE %s", fmtId(tbinfo.relname, force_quotes));
4833 4834
		dumpComment(fout, query->data, tbinfo.oid,
					"pg_class", 0, NULL);
4835
	}
4836

4837 4838
	PQclear(res);

4839 4840
	destroyPQExpBuffer(query);
	destroyPQExpBuffer(delqry);
4841
}
V
Vadim B. Mikheev 已提交
4842 4843


4844
static void
B
Bruce Momjian 已提交
4845
dumpTriggers(Archive *fout, const char *tablename,
4846
			 TableInfo *tblinfo, int numTables)
V
Vadim B. Mikheev 已提交
4847
{
4848 4849 4850
	int			i,
				j;

V
Vadim B. Mikheev 已提交
4851
	if (g_verbose)
4852
		write_msg(NULL, "dumping out triggers\n");
4853

V
Vadim B. Mikheev 已提交
4854 4855
	for (i = 0; i < numTables; i++)
	{
B
Bruce Momjian 已提交
4856
		if (tablename && (strcmp(tblinfo[i].relname, tablename) != 0) && (strlen(tablename) > 0))
V
Vadim B. Mikheev 已提交
4857
			continue;
4858

V
Vadim B. Mikheev 已提交
4859 4860
		for (j = 0; j < tblinfo[i].ntrig; j++)
		{
B
Bruce Momjian 已提交
4861
			ArchiveEntry(fout, tblinfo[i].triggers[j].oid, tblinfo[i].triggers[j].tgname,
B
Bruce Momjian 已提交
4862 4863
				   "TRIGGER", NULL, tblinfo[i].triggers[j].tgsrc, "", "",
						 tblinfo[i].usename, NULL, NULL);
4864 4865
			dumpComment(fout, tblinfo[i].triggers[j].tgcomment, tblinfo[i].triggers[j].oid,
						"pg_trigger", 0, NULL);
V
Vadim B. Mikheev 已提交
4866 4867 4868
		}
	}
}
4869 4870


4871
static void
B
Bruce Momjian 已提交
4872
dumpRules(Archive *fout, const char *tablename,
B
Bruce Momjian 已提交
4873
		  TableInfo *tblinfo, int numTables)
4874
{
B
Bruce Momjian 已提交
4875 4876 4877 4878
	PGresult   *res;
	int			nrules;
	int			i,
				t;
4879
	PQExpBuffer query = createPQExpBuffer();
4880

B
Bruce Momjian 已提交
4881
	int			i_definition;
4882
	int			i_oid;
4883
	int			i_owner;
4884
	int			i_rulename;
4885 4886

	if (g_verbose)
4887
		write_msg(NULL, "dumping out rules\n");
4888 4889 4890 4891 4892 4893

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

		/*
B
Bruce Momjian 已提交
4898 4899
		 * 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).
4900 4901
		 *
		 * XXXX: Use LOJ here
4902
		 */
B
Hi, all  
Bruce Momjian 已提交
4903
		resetPQExpBuffer(query);
4904
		appendPQExpBuffer(query, "SELECT definition,"
4905
						  "   (select usename from pg_user where pg_class.relowner = usesysid) AS viewowner, "
4906 4907
						  "   pg_rewrite.oid, pg_rewrite.rulename "
						  "FROM pg_rewrite, pg_class, pg_rules "
4908
						  "WHERE pg_class.relname = ");
4909
		formatStringLiteral(query, tblinfo[t].relname, CONV_ALL);
4910
		appendPQExpBuffer(query,
4911 4912
						  "    AND pg_rewrite.ev_class = pg_class.oid "
						  "    AND pg_rules.tablename = pg_class.relname "
B
Bruce Momjian 已提交
4913
					   "    AND pg_rules.rulename = pg_rewrite.rulename "
4914
						  "ORDER BY pg_rewrite.oid");
B
Hi, all  
Bruce Momjian 已提交
4915
		res = PQexec(g_conn, query->data);
4916 4917 4918
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
4919 4920
			write_msg(NULL, "query to get rules associated with table \"%s\" failed: %s",
					   tblinfo[t].relname, PQerrorMessage(g_conn));
4921
			exit_nicely();
4922 4923 4924 4925
		}

		nrules = PQntuples(res);
		i_definition = PQfnumber(res, "definition");
4926
		i_owner = PQfnumber(res, "viewowner");
B
Bruce,  
Bruce Momjian 已提交
4927 4928
		i_oid = PQfnumber(res, "oid");
		i_rulename = PQfnumber(res, "rulename");
4929 4930 4931 4932

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

4934 4935
		for (i = 0; i < nrules; i++)
		{
B
Bruce Momjian 已提交
4936
			ArchiveEntry(fout, PQgetvalue(res, i, i_oid), PQgetvalue(res, i, i_rulename),
B
Bruce Momjian 已提交
4937 4938
						 "RULE", NULL, PQgetvalue(res, i, i_definition),
						 "", "", PQgetvalue(res, i, i_owner), NULL, NULL);
4939

B
Bruce,  
Bruce Momjian 已提交
4940 4941 4942 4943
			/* Dump rule comments */

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

B
Bruce,  
Bruce Momjian 已提交
4947 4948
		}

4949 4950
		PQclear(res);
	}
4951 4952

	destroyPQExpBuffer(query);
4953
}