pg_dump.c 97.1 KB
Newer Older
1 2
/*-------------------------------------------------------------------------
 *
3
 * pg_dump.c
4
 *	  pg_dump is an utility for dumping out a postgres database
5 6
 * into a script file.
 *
7 8 9 10 11 12 13 14 15 16
 *	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
 *		  indices
 *		  aggregates
 *		  operators
 *		  ACL - grant/revoke
17
 *
18
 * the output script is SQL that is understood by PostgreSQL
19
 *
B
Add:  
Bruce Momjian 已提交
20 21
 * Portions Copyright (c) 1996-2000, PostgreSQL, Inc
 * Portions Copyright (c) 1994, Regents of the University of California
22 23 24
 *
 *
 * IDENTIFICATION
25
 *	  $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.152 2000/06/14 18:17:50 petere Exp $
26
 *
27
 * Modifications - 6/10/96 - dave@bensoft.com - version 1.13.dhb
28
 *
29 30 31 32
 *	 Applied 'insert string' patch from "Marc G. Fournier" <scrappy@ki.net>
 *	 Added '-t table' option
 *	 Added '-a' option
 *	 Added '-da' option
33
 *
34 35
 * Modifications - 6/12/96 - dave@bensoft.com - version 1.13.dhb.2
 *
36 37 38
 *	 - Fixed dumpTable output to output lengths for char and varchar types!
 *	 - Added single. quote to twin single quote expansion for 'insert' string
 *	   mode.
39
 *
40 41
 * Modifications - 7/26/96 - asussman@vidya.com
 *
42
 *	 - Fixed ouput lengths for char and varchar type where the length is variable (-1)
43
 *
44 45
 * Modifications - 6/1/97 - igor@sba.miami.edu
 * - Added functions to free allocated memory used for retrieving
46 47
 *	 indices,tables,inheritance,types,functions and aggregates.
 *	 No more leaks reported by Purify.
48 49 50
 *
 *
 * Modifications - 1/26/98 - pjlobo@euitt.upm.es
51
 *		 - Added support for password authentication
52 53 54
 *-------------------------------------------------------------------------
 */

55
#include <unistd.h>				/* for getopt() */
56
#include <ctype.h>
57 58

#include "postgres.h"
59 60 61 62 63 64 65 66 67

#ifdef HAVE_GETOPT_H
#include <getopt.h>
#endif
#ifdef HAVE_TERMIOS_H
#include <termios.h>
#endif

#include "access/attnum.h"
68
#include "access/htup.h"
69
#include "catalog/pg_index.h"
70
#include "catalog/pg_language.h"
V
Vadim B. Mikheev 已提交
71
#include "catalog/pg_trigger.h"
72
#include "catalog/pg_type.h"
73
#include "version.h"
74

75
#include "libpq-fe.h"
76
#ifndef HAVE_STRDUP
77 78
#include "strdup.h"
#endif
79 80 81

#include "pg_dump.h"

B
Bruce,  
Bruce Momjian 已提交
82
static void dumpComment(FILE *outfile, const char *target, const char *oid);
83
static void dumpSequence(FILE *fout, TableInfo tbinfo);
B
Bruce Momjian 已提交
84
static void dumpACL(FILE *fout, TableInfo tbinfo);
85
static void dumpTriggers(FILE *fout, const char *tablename,
86
			 TableInfo *tblinfo, int numTables);
87
static void dumpRules(FILE *fout, const char *tablename,
B
Bruce Momjian 已提交
88
		  TableInfo *tblinfo, int numTables);
89 90
static char *checkForQuote(const char *s);
static void clearTableInfo(TableInfo *, int);
91
static void dumpOneFunc(FILE *fout, FuncInfo *finfo, int i,
92
			TypeInfo *tinfo, int numTypes);
93 94
static int	findLastBuiltinOid(void);
static bool isViewRule(char *relname);
95
static void setMaxOid(FILE *fout);
96

97 98
static void AddAcl(char *aclbuf, const char *keyword);
static char *GetPrivileges(const char *s);
99
static void becomeUser(FILE *fout, const char *username);
V
Vadim B. Mikheev 已提交
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 108 109 110 111 112
								 * activities. */
int			g_last_builtin_oid; /* value of the last builtin oid */
FILE	   *g_fout;				/* the script file */
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
bool		dropSchema;
119

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

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


static void
128
help(const char *progname)
129
{
P
Peter Eisentraut 已提交
130
	printf("%s dumps a database as a text file.\n\n", progname);
131 132 133
	puts("Usage:");
	printf("  %s [options] dbname\n\n", progname);
	puts("Options:");
B
Hi,  
Bruce Momjian 已提交
134 135

#ifdef HAVE_GETOPT_LONG
136
	puts(
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
	"  -a, --data-only          dump out only the data, not the schema\n"
	   "  -c, --clean              clean (drop) schema prior to create\n"
		 "  -d, --inserts            dump data as INSERT, rather than COPY, commands\n"
		 "  -D, --attribute-inserts  dump data as INSERT commands with attribute names\n"
		 "  -h, --host <hostname>    server host name\n"
		 "  -i, --ignore-version     proceed when database version != pg_dump version\n"
	"  -n, --no-quotes          suppress most quotes around identifiers\n"
	 "  -N, --quotes             enable most quotes around identifiers\n"
		 "  -o, --oids               dump object ids (oids)\n"
		 "  -p, --port <port>        server port number\n"
		 "  -s, --schema-only        dump out only the schema, no data\n"
		 "  -t, --table <table>      dump for this table only\n"
		 "  -u, --password           use password authentication\n"
		 "  -v, --verbose            verbose\n"
		 "  -x, --no-acl             do not dump ACL's (grant/revoke)\n"
		);
B
Hi,  
Bruce Momjian 已提交
153
#else
154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170
	puts(
		 "  -a                       dump out only the data, no schema\n"
	   "  -c                       clean (drop) schema prior to create\n"
		 "  -d                       dump data as INSERT, rather than COPY, commands\n"
		 "  -D                       dump data as INSERT commands with attribute names\n"
		 "  -h <hostname>            server host name\n"
		 "  -i                       proceed when database version != pg_dump version\n"
	"  -n                       suppress most quotes around identifiers\n"
	 "  -N                       enable most quotes around identifiers\n"
		 "  -o                       dump object ids (oids)\n"
		 "  -p <port>                server port number\n"
		 "  -s                       dump out only the schema, no data\n"
		 "  -t <table>               dump for this table only\n"
		 "  -u                       use password authentication\n"
		 "  -v                       verbose\n"
		 "  -x                       do not dump ACL's (grant/revoke)\n"
		);
B
Hi,  
Bruce Momjian 已提交
171
#endif
172 173
	puts("If no database name is not supplied, then the PGDATABASE environment\nvariable value is used.\n");
	puts("Report bugs to <pgsql-bugs@postgresql.org>.");
174
}
175

176

177 178 179
static void
version(void)
{
180 181 182 183
	puts("pg_dump (PostgreSQL) " PG_RELEASE "." PG_VERSION "." PG_SUBVERSION);
	puts("Portions Copyright (c) 1996-2000, PostgreSQL, Inc");
	puts("Portions Copyright (C) 1996 Regents of the University of California");
	puts("Read the file COPYRIGHT to see the usage and distribution terms.");
184 185
}

186

187
static void
188
exit_nicely(PGconn *conn)
189
{
190 191
	PQfinish(conn);
	exit(1);
192 193 194
}


M
Marc G. Fournier 已提交
195 196
/*
 * isViewRule
197
 *				Determine if the relation is a VIEW
M
Marc G. Fournier 已提交
198 199
 *
 */
200
static bool
M
Marc G. Fournier 已提交
201 202
isViewRule(char *relname)
{
B
Bruce Momjian 已提交
203 204
	PGresult   *res;
	int			ntups;
205
	PQExpBuffer query = createPQExpBuffer();
206

B
Hi, all  
Bruce Momjian 已提交
207 208 209 210
	appendPQExpBuffer(query, "select relname from pg_class, pg_rewrite ");
	appendPQExpBuffer(query, "where pg_class.oid = ev_class ");
	appendPQExpBuffer(query, "and pg_rewrite.ev_type = '1' ");
	appendPQExpBuffer(query, "and rulename = '_RET%s'", relname);
211

B
Hi, all  
Bruce Momjian 已提交
212
	res = PQexec(g_conn, query->data);
213 214 215
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
216
		fprintf(stderr, "isViewRule(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
217 218 219 220 221 222 223
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	PQclear(res);
	return ntups > 0 ? TRUE : FALSE;
M
Marc G. Fournier 已提交
224
}
225

226 227 228 229
#define COPYBUFSIZ		8192


static void
230
dumpClasses_nodumpData(FILE *fout, const char *classname, const bool oids)
231 232
{

233 234 235 236 237
	PGresult   *res;
	char		query[255];
	int			ret;
	bool		copydone;
	char		copybuf[COPYBUFSIZ];
238

239
	if (oids == true)
240
	{
241
		fprintf(fout, "COPY %s WITH OIDS FROM stdin;\n",
242
				fmtId(classname, force_quotes));
243
		sprintf(query, "COPY %s WITH OIDS TO stdout;\n",
244
				fmtId(classname, force_quotes));
245 246 247
	}
	else
	{
248 249
		fprintf(fout, "COPY %s FROM stdin;\n", fmtId(classname, force_quotes));
		sprintf(query, "COPY %s TO stdout;\n", fmtId(classname, force_quotes));
250 251
	}
	res = PQexec(g_conn, query);
252 253
	if (!res ||
		PQresultStatus(res) == PGRES_FATAL_ERROR)
254
	{
255
		fprintf(stderr, "SQL query to dump the contents of Table '%s' "
256 257 258 259 260 261 262 263 264
				"did not execute.  Explanation from backend: '%s'.\n"
				"The query was: '%s'.\n",
				classname, PQerrorMessage(g_conn), query);
		exit_nicely(g_conn);
	}
	else
	{
		if (PQresultStatus(res) != PGRES_COPY_OUT)
		{
265
			fprintf(stderr, "SQL query to dump the contents of Table '%s' "
266 267 268 269 270 271 272 273 274 275 276
					"executed abnormally.\n"
					"PQexec() returned status %d when %d was expected.\n"
					"The query was: '%s'.\n",
				  classname, PQresultStatus(res), PGRES_COPY_OUT, query);
			exit_nicely(g_conn);
		}
		else
		{
			copydone = false;
			while (!copydone)
			{
277
				ret = PQgetline(g_conn, copybuf, COPYBUFSIZ);
278 279 280 281 282 283 284 285 286 287 288 289

				if (copybuf[0] == '\\' &&
					copybuf[1] == '.' &&
					copybuf[2] == '\0')
				{
					copydone = true;	/* don't print this... */
				}
				else
				{
					fputs(copybuf, fout);
					switch (ret)
					{
290 291 292 293 294 295 296 297
						case EOF:
							copydone = true;
							/* FALLTHROUGH */
						case 0:
							fputc('\n', fout);
							break;
						case 1:
							break;
298 299 300 301 302
					}
				}
			}
			fprintf(fout, "\\.\n");
		}
303
		ret = PQendcopy(g_conn);
304 305
		if (ret != 0)
		{
306
			fprintf(stderr, "SQL query to dump the contents of Table '%s' "
307 308 309 310 311
					"did not execute correctly.  After we read all the "
				 "table contents from the backend, PQendcopy() failed.  "
					"Explanation from backend: '%s'.\n"
					"The query was: '%s'.\n",
					classname, PQerrorMessage(g_conn), query);
B
Bruce Momjian 已提交
312
			PQclear(res);
313 314 315
			exit_nicely(g_conn);
		}
	}
316 317 318 319 320
}



static void
B
Bruce Momjian 已提交
321
dumpClasses_dumpData(FILE *fout, const char *classname)
322
{
323 324
	PGresult   *res;
	PQExpBuffer q = createPQExpBuffer();
325 326
	int			tuple;
	int			field;
327
	const char *expsrc;
328

B
Hi, all  
Bruce Momjian 已提交
329 330
	appendPQExpBuffer(q, "SELECT * FROM %s", fmtId(classname, force_quotes));
	res = PQexec(g_conn, q->data);
331 332 333
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
334
		fprintf(stderr, "dumpClasses(): command failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
335 336
		exit_nicely(g_conn);
	}
337
	for (tuple = 0; tuple < PQntuples(res); tuple++)
338
	{
339
		fprintf(fout, "INSERT INTO %s ", fmtId(classname, force_quotes));
340
		if (attrNames == true)
341
		{
B
Hi, all  
Bruce Momjian 已提交
342 343
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "(");
344
			for (field = 0; field < PQnfields(res); field++)
345
			{
346
				if (field > 0)
B
Hi, all  
Bruce Momjian 已提交
347 348
					appendPQExpBuffer(q, ",");
				appendPQExpBuffer(q, fmtId(PQfname(res, field), force_quotes));
349
			}
B
Hi, all  
Bruce Momjian 已提交
350 351
			appendPQExpBuffer(q, ") ");
			fprintf(fout, "%s", q->data);
352
		}
353
		fprintf(fout, "VALUES (");
354
		for (field = 0; field < PQnfields(res); field++)
355
		{
356 357
			if (field > 0)
				fprintf(fout, ",");
358
			if (PQgetisnull(res, tuple, field))
359
			{
360
				fprintf(fout, "NULL");
361 362 363
				continue;
			}
			switch (PQftype(res, field))
364
			{
365 366
				case INT2OID:
				case INT4OID:
367
				case OIDOID:	/* int types */
368
				case FLOAT4OID:
369
				case FLOAT8OID:/* float types */
370 371 372 373 374
					/* These types are printed without quotes */
					fprintf(fout, "%s",
							PQgetvalue(res, tuple, field));
					break;
				default:
375

376 377
					/*
					 * All other types are printed as string literals,
378 379 380
					 * with appropriate escaping of special characters.
					 * Quote mark ' goes to '' per SQL standard, other
					 * stuff goes to \ sequences.
381 382 383 384 385 386 387 388
					 */
					putc('\'', fout);
					expsrc = PQgetvalue(res, tuple, field);
					while (*expsrc)
					{
						char		ch = *expsrc++;

						if (ch == '\\' || ch == '\'')
389
						{
390
							putc(ch, fout);		/* double these */
391
							putc(ch, fout);
392
						}
393 394 395 396 397 398 399 400 401 402 403 404 405
						else if (ch < '\040')
						{
							/* generate octal escape for control chars */
							putc('\\', fout);
							putc(((ch >> 6) & 3) + '0', fout);
							putc(((ch >> 3) & 7) + '0', fout);
							putc((ch & 7) + '0', fout);
						}
						else
							putc(ch, fout);
					}
					putc('\'', fout);
					break;
406
			}
407
		}
408 409 410
		fprintf(fout, ");\n");
	}
	PQclear(res);
411 412 413 414 415 416
}



/*
 * DumpClasses -
417
 *	  dump the contents of all the classes.
418 419
 */
static void
420
dumpClasses(const TableInfo *tblinfo, const int numTables, FILE *fout,
421 422 423
			const char *onlytable, const bool oids)
{

424
	int			i;
B
Bruce Momjian 已提交
425
	char	   *all_only;
426 427 428 429

	if (onlytable == NULL)
		all_only = "all";
	else
430
		all_only = "only";
431 432

	if (g_verbose)
433 434 435
		fprintf(stderr, "%s dumping out the contents of %s %d table%s/sequence%s %s\n",
				g_comment_start, all_only,
				(onlytable == NULL) ? numTables : 1,
436
		  (onlytable == NULL) ? "s" : "", (onlytable == NULL) ? "s" : "",
437
				g_comment_end);
438

V
Vadim B. Mikheev 已提交
439 440 441 442 443 444 445 446 447 448
	/* Dump SEQUENCEs first (if dataOnly) */
	if (dataOnly)
	{
		for (i = 0; i < numTables; i++)
		{
			if (!(tblinfo[i].sequence))
				continue;
			if (!onlytable || (!strcmp(tblinfo[i].relname, onlytable)))
			{
				if (g_verbose)
449
					fprintf(stderr, "%s dumping out schema of sequence '%s' %s\n",
450
					 g_comment_start, tblinfo[i].relname, g_comment_end);
451
				becomeUser(fout, tblinfo[i].usename);
V
Vadim B. Mikheev 已提交
452 453 454 455
				dumpSequence(fout, tblinfo[i]);
			}
		}
	}
456 457 458

	for (i = 0; i < numTables; i++)
	{
459
		const char *classname = tblinfo[i].relname;
460 461 462 463

		/* Skip VIEW relations */
		if (isViewRule(tblinfo[i].relname))
			continue;
464 465

		if (tblinfo[i].sequence)/* already dumped */
V
Vadim B. Mikheev 已提交
466
			continue;
467 468 469 470

		if (!onlytable || (!strcmp(classname, onlytable)))
		{
			if (g_verbose)
471
				fprintf(stderr, "%s dumping out the contents of Table '%s' %s\n",
472 473
						g_comment_start, classname, g_comment_end);

474 475
			becomeUser(fout, tblinfo[i].usename);

476 477 478
			if (!dumpData)
				dumpClasses_nodumpData(fout, classname, oids);
			else
B
Bruce Momjian 已提交
479
				dumpClasses_dumpData(fout, classname);
480
		}
481
	}
482 483 484
}


485 486 487
static void
prompt_for_password(char *username, char *password)
{
488
	char		buf[512];
B
Bruce Momjian 已提交
489
	int			length;
490 491

#ifdef HAVE_TERMIOS_H
B
Bruce Momjian 已提交
492
	struct termios t_orig,
493
				   t;
494 495
#endif

496
	fprintf(stderr, "Username: ");
497
	fflush(stderr);
498
	fgets(username, 100, stdin);
499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516
	length = strlen(username);
	/* skip rest of the line */
	if (length > 0 && username[length - 1] != '\n')
	{
		do
		{
			fgets(buf, 512, stdin);
		} while (buf[strlen(buf) - 1] != '\n');
	}
	if (length > 0 && username[length - 1] == '\n')
		username[length - 1] = '\0';

#ifdef HAVE_TERMIOS_H
	tcgetattr(0, &t);
	t_orig = t;
	t.c_lflag &= ~ECHO;
	tcsetattr(0, TCSADRAIN, &t);
#endif
517 518
	fprintf(stderr, "Password: ");
	fflush(stderr);
519
	fgets(password, 100, stdin);
520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535
#ifdef HAVE_TERMIOS_H
	tcsetattr(0, TCSADRAIN, &t_orig);
#endif

	length = strlen(password);
	/* skip rest of the line */
	if (length > 0 && password[length - 1] != '\n')
	{
		do
		{
			fgets(buf, 512, stdin);
		} while (buf[strlen(buf) - 1] != '\n');
	}
	if (length > 0 && password[length - 1] == '\n')
		password[length - 1] = '\0';

536
	fprintf(stderr, "\n\n");
537 538
}

539

540
static void
541
check_database_version(bool ignoreVersion)
542
{
543 544 545 546
	PGresult   *res;
	const char *dbversion;
	const char *myversion = "PostgreSQL " PG_RELEASE "." PG_VERSION;
	int			myversionlen = strlen(myversion);
547 548 549 550 551 552 553 554 555 556 557 558 559

	res = PQexec(g_conn, "SELECT version()");
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK ||
		PQntuples(res) != 1)
	{
		fprintf(stderr, "check_database_version(): command failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
		exit_nicely(g_conn);
	}
	dbversion = PQgetvalue(res, 0, 0);
	if (strncmp(dbversion, myversion, myversionlen) != 0)
	{
		fprintf(stderr, "Database version: %s\npg_dump version: %s\n",
560
				dbversion, PG_RELEASE "." PG_VERSION);
561 562 563 564 565 566 567 568 569 570 571 572 573
		if (ignoreVersion)
			fprintf(stderr, "Proceeding despite version mismatch.\n");
		else
		{
			fprintf(stderr, "Aborting because of version mismatch.\n"
					"Use --ignore-version if you think it's safe to proceed anyway.\n");
			exit_nicely(g_conn);
		}
	}
	PQclear(res);
}


574
int
575
main(int argc, char **argv)
576
{
B
Bruce Momjian 已提交
577 578 579 580 581 582 583
	int			c;
	const char *progname;
	const char *filename = NULL;
	const char *dbname = NULL;
	const char *pghost = NULL;
	const char *pgport = NULL;
	char	   *tablename = NULL;
584
	bool		oids = false;
B
Bruce Momjian 已提交
585 586
	TableInfo  *tblinfo;
	int			numTables;
587 588
	char		connect_string[512] = "";
	char		tmp_string[128];
589 590
	char		username[100];
	char		password[100];
591
	bool		use_password = false;
592
	bool		ignore_version = false;
593

B
Hi,  
Bruce Momjian 已提交
594 595 596 597
#ifdef HAVE_GETOPT_LONG
	static struct option long_options[] = {
		{"data-only", no_argument, NULL, 'a'},
		{"clean", no_argument, NULL, 'c'},
598
		{"inserts", no_argument, NULL, 'd'},
599
		{"attribute-inserts", no_argument, NULL, 'D'},
B
Hi,  
Bruce Momjian 已提交
600
		{"host", required_argument, NULL, 'h'},
601
		{"ignore-version", no_argument, NULL, 'i'},
B
Hi,  
Bruce Momjian 已提交
602 603
		{"no-quotes", no_argument, NULL, 'n'},
		{"quotes", no_argument, NULL, 'N'},
604
		{"oids", no_argument, NULL, 'o'},
B
Hi,  
Bruce Momjian 已提交
605 606 607 608 609 610 611
		{"port", required_argument, NULL, 'p'},
		{"schema-only", no_argument, NULL, 's'},
		{"table", required_argument, NULL, 't'},
		{"password", no_argument, NULL, 'u'},
		{"verbose", no_argument, NULL, 'v'},
		{"no-acl", no_argument, NULL, 'x'},
		{"help", no_argument, NULL, '?'},
612
		{"version", no_argument, NULL, 'V'}
613
	};
614 615
	int			optindex;

B
Hi,  
Bruce Momjian 已提交
616 617
#endif

618
	g_verbose = false;
619
	force_quotes = true;
620
	dropSchema = false;
621 622 623 624 625

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

626
	dataOnly = schemaOnly = dumpData = attrNames = false;
627

628 629 630 631
	if (!strrchr(argv[0], SEP_CHAR))
		progname = argv[0];
	else
		progname = strrchr(argv[0], SEP_CHAR) + 1;
632

633

B
Hi,  
Bruce Momjian 已提交
634
#ifdef HAVE_GETOPT_LONG
635
	while ((c = getopt_long(argc, argv, "acdDf:h:inNop:st:uvxzV?", long_options, &optindex)) != -1)
B
Hi,  
Bruce Momjian 已提交
636
#else
637
	while ((c = getopt(argc, argv, "acdDf:h:inNop:st:uvxzV?-")) != -1)
B
Hi,  
Bruce Momjian 已提交
638
#endif
639 640 641
	{
		switch (c)
		{
642
			case 'a':			/* Dump data only */
643
				dataOnly = true;
644
				break;
B
Bruce Momjian 已提交
645 646
			case 'c':			/* clean (i.e., drop) schema prior to
								 * create */
647
				dropSchema = true;
648
				break;
649
			case 'd':			/* dump data as proper insert strings */
650
				dumpData = true;
651 652 653
				break;
			case 'D':			/* dump data as proper insert strings with
								 * attr names */
654 655
				dumpData = true;
				attrNames = true;
656
				break;
657
			case 'f':
658 659
				filename = optarg;
				break;
B
Bruce Momjian 已提交
660
			case 'h':			/* server host */
661 662
				pghost = optarg;
				break;
663 664 665
			case 'i':			/* ignore database version mismatch */
				ignore_version = true;
				break;
B
Bruce Momjian 已提交
666 667
			case 'n':			/* Do not force double-quotes on
								 * identifiers */
668
				force_quotes = false;
669
				break;
670
			case 'N':			/* Force double-quotes on identifiers */
671
				force_quotes = true;
672
				break;
673
			case 'o':			/* Dump oids */
674
				oids = true;
675
				break;
676 677 678
			case 'p':			/* server port */
				pgport = optarg;
				break;
679
			case 's':			/* dump schema only */
680
				schemaOnly = true;
681 682
				break;
			case 't':			/* Dump data for this table only */
683
				{
684
					int			i;
685 686

					tablename = strdup(optarg);
B
Bruce Momjian 已提交
687 688 689 690 691

					/*
					 * quoted string? Then strip quotes and preserve
					 * case...
					 */
692 693 694
					if (tablename[0] == '"')
					{
						strcpy(tablename, &tablename[1]);
B
Bruce Momjian 已提交
695 696
						if (*(tablename + strlen(tablename) - 1) == '"')
							*(tablename + strlen(tablename) - 1) = '\0';
697 698 699 700 701
					}
					/* otherwise, convert table name to lowercase... */
					else
					{
						for (i = 0; tablename[i]; i++)
702 703
							if (isascii((int) tablename[i]) &&
								isupper((int) tablename[i]))
704 705
								tablename[i] = tolower(tablename[i]);
					}
706
				}
707
				break;
708 709 710
			case 'u':
				use_password = true;
				break;
711 712
			case 'v':			/* verbose */
				g_verbose = true;
713
				break;
714 715
			case 'x':			/* skip ACL dump */
				aclsSkip = true;
716
				break;
717 718 719 720
			case 'V':
				version();
				exit(0);
				break;
721
			case '?':
722 723 724 725 726 727 728 729 730 731 732 733 734 735 736

				/*
				 * getopt returns '?' on unknown argument. That's not
				 * quite what we want
				 */
				if (strcmp(argv[optind - 1], "-?") == 0 || strcmp(argv[optind - 1], "--help") == 0)
				{
					help(progname);
					exit(1);
				}
				else
				{
					fputs("Try -? for help.\n", stderr);
					exit(1);
				}
737
				break;
738 739 740
#ifndef HAVE_GETOPT_LONG
			case '-':
				fprintf(stderr, "%s was compiled without support for long options.\n"
741
				   "Use -? for help on invocation options.\n", progname);
742 743 744
				exit(1);
				break;
#endif
745
			default:
746 747
				fprintf(stderr, "%s: unknown option -%c\nTry -? for help.\n", progname, c);
				exit(1);
748 749 750
		}
	}

751 752 753
	if (dumpData == true && oids == true)
	{
		fprintf(stderr,
754
				"%s: INSERT's can not set oids, so INSERT and OID options can not be used together.\n",
755
				progname);
756
		exit(1);
757
	}
758

759 760 761 762 763
	/* open the output file */
	if (filename == NULL)
		g_fout = stdout;
	else
	{
764
		g_fout = fopen(filename, PG_BINARY_W);
765 766 767 768 769
		if (g_fout == NULL)
		{
			fprintf(stderr,
				 "%s: could not open output file named %s for writing\n",
					progname, filename);
770
			exit(1);
771 772 773 774 775
		}
	}

	/* find database */
	if (!(dbname = argv[optind]) &&
776
		!(dbname = getenv("PGDATABASE")))
777 778
	{
		fprintf(stderr, "%s: no database name specified\n", progname);
779
		exit(1);
780 781
	}

782 783 784 785 786
	/* g_conn = PQsetdb(pghost, pgport, NULL, NULL, dbname); */
	if (pghost != NULL)
	{
		sprintf(tmp_string, "host=%s ", pghost);
		strcat(connect_string, tmp_string);
787
	}
788 789 790 791
	if (pgport != NULL)
	{
		sprintf(tmp_string, "port=%s ", pgport);
		strcat(connect_string, tmp_string);
792
	}
793 794 795 796
	if (dbname != NULL)
	{
		sprintf(tmp_string, "dbname=%s ", dbname);
		strcat(connect_string, tmp_string);
797
	}
798 799 800 801 802 803 804 805
	if (use_password)
	{
		prompt_for_password(username, password);
		strcat(connect_string, "authtype=password ");
		sprintf(tmp_string, "user=%s ", username);
		strcat(connect_string, tmp_string);
		sprintf(tmp_string, "password=%s ", password);
		strcat(connect_string, tmp_string);
806 807
		MemSet(tmp_string, 0, sizeof(tmp_string));
		MemSet(password, 0, sizeof(password));
808 809
	}
	g_conn = PQconnectdb(connect_string);
810
	MemSet(connect_string, 0, sizeof(connect_string));
811 812 813 814 815 816 817 818
	/* check to see that the backend connection was successfully made */
	if (PQstatus(g_conn) == CONNECTION_BAD)
	{
		fprintf(stderr, "Connection to database '%s' failed.\n", dbname);
		fprintf(stderr, "%s\n", PQerrorMessage(g_conn));
		exit_nicely(g_conn);
	}

819 820 821
	/* check for version mismatch */
	check_database_version(ignore_version);

822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843
	/*
	 * Start serializable transaction to dump consistent data
	 */
	{
		PGresult   *res;

		res = PQexec(g_conn, "begin");
		if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
		{
			fprintf(stderr, "BEGIN command failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
			exit_nicely(g_conn);
		}
		PQclear(res);
		res = PQexec(g_conn, "set transaction isolation level serializable");
		if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
		{
			fprintf(stderr, "SET TRANSACTION command failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
			exit_nicely(g_conn);
		}
		PQclear(res);
	}

844 845
	g_last_builtin_oid = findLastBuiltinOid();

846
	if (oids == true)
847 848 849 850
		setMaxOid(g_fout);
	if (!dataOnly)
	{
		if (g_verbose)
851
			fprintf(stderr, "%s last builtin oid is %u %s\n",
852
					g_comment_start, g_last_builtin_oid, g_comment_end);
853
		tblinfo = dumpSchema(g_fout, &numTables, tablename, aclsSkip);
854 855
	}
	else
856
		tblinfo = dumpSchema(NULL, &numTables, tablename, aclsSkip);
857

858 859
	if (!schemaOnly)
	{
860 861 862
		if (dataOnly)
			fprintf(g_fout, "UPDATE \"pg_class\" SET \"reltriggers\" = 0 WHERE \"relname\" !~ '^pg_';\n");

863 864
		dumpClasses(tblinfo, numTables, g_fout, tablename, oids);

865 866
		if (dataOnly)
		{
867 868 869 870 871 872 873 874
			fprintf(g_fout, "BEGIN TRANSACTION;\n");
			fprintf(g_fout, "CREATE TEMP TABLE \"tr\" (\"tmp_relname\" name, \"tmp_reltriggers\" smallint);\n");
			fprintf(g_fout, "INSERT INTO \"tr\" SELECT C.\"relname\", count(T.\"oid\") FROM \"pg_class\" C, \"pg_trigger\" T WHERE C.\"oid\" = T.\"tgrelid\" AND C.\"relname\" !~ '^pg_' GROUP BY 1;\n");
			fprintf(g_fout, "UPDATE \"pg_class\" SET \"reltriggers\" = TMP.\"tmp_reltriggers\" FROM \"tr\" TMP WHERE \"pg_class\".\"relname\" = TMP.\"tmp_relname\";\n");
			fprintf(g_fout, "COMMIT TRANSACTION;\n");
		}
	}

875 876
	if (!dataOnly)				/* dump indexes and triggers at the end
								 * for performance */
V
Vadim B. Mikheev 已提交
877 878 879
	{
		dumpSchemaIdx(g_fout, tablename, tblinfo, numTables);
		dumpTriggers(g_fout, tablename, tblinfo, numTables);
880
		dumpRules(g_fout, tablename, tblinfo, numTables);
V
Vadim B. Mikheev 已提交
881
	}
882 883

	fflush(g_fout);
884 885 886
	if (g_fout != stdout)
		fclose(g_fout);

887 888 889
	clearTableInfo(tblinfo, numTables);
	PQfinish(g_conn);
	exit(0);
890 891 892
}

/*
893 894
 * getTypes:
 *	  read all base types in the system catalogs and return them in the
895 896
 * TypeInfo* structure
 *
897
 *	numTypes is set to the number of types read in
898 899
 *
 */
900
TypeInfo   *
901 902
getTypes(int *numTypes)
{
B
Bruce Momjian 已提交
903
	PGresult   *res;
904 905
	int			ntups;
	int			i;
906
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
907
	TypeInfo   *tinfo;
908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923

	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;
	int			i_typbyval;
	int			i_usename;
924 925 926 927 928 929 930 931 932 933 934 935

	/* find all base types */

	/*
	 * we include even the built-in types because those may be used as
	 * array elements by user-defined types
	 */

	/*
	 * we filter out the built-in types when we dump out the types
	 */

B
Hi, all  
Bruce Momjian 已提交
936
	appendPQExpBuffer(query, "SELECT pg_type.oid, typowner, typname, typlen, typprtlen, "
937
		  "typinput, typoutput, typreceive, typsend, typelem, typdelim, "
938 939
		 "typdefault, typrelid, typbyval, usename from pg_type, pg_user "
					  "where typowner = usesysid");
940

B
Hi, all  
Bruce Momjian 已提交
941
	res = PQexec(g_conn, query->data);
942 943 944
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
945
		fprintf(stderr, "getTypes(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005
		exit_nicely(g_conn);
	}

	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");
	i_typbyval = PQfnumber(res, "typbyval");
	i_usename = PQfnumber(res, "usename");

	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));
		tinfo[i].usename = strdup(PQgetvalue(res, i, i_usename));

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

	*numTypes = ntups;

	PQclear(res);

	return tinfo;
1006 1007 1008 1009
}

/*
 * getOperators:
1010
 *	  read all operators in the system catalogs and return them in the
1011 1012
 * OprInfo* structure
 *
1013 1014
 *	numOprs is set to the number of operators read in
 *
1015 1016
 *
 */
1017
OprInfo    *
1018 1019
getOperators(int *numOprs)
{
1020
	PGresult   *res;
1021 1022
	int			ntups;
	int			i;
1023
	PQExpBuffer query = createPQExpBuffer();
1024

B
Bruce Momjian 已提交
1025
	OprInfo    *oprinfo;
1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040

	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;
1041 1042 1043 1044 1045 1046

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

B
Hi, all  
Bruce Momjian 已提交
1047
	appendPQExpBuffer(query, "SELECT pg_operator.oid, oprname, oprkind, oprcode, "
1048 1049 1050 1051
			   "oprleft, oprright, oprcom, oprnegate, oprrest, oprjoin, "
					  "oprcanhash, oprlsortop, oprrsortop, usename "
					  "from pg_operator, pg_user "
					  "where oprowner = usesysid");
1052

B
Hi, all  
Bruce Momjian 已提交
1053
	res = PQexec(g_conn, query->data);
1054 1055 1056
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1057
		fprintf(stderr, "getOperators(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
		exit_nicely(g_conn);
	}

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

	PQclear(res);

	return oprinfo;
1102 1103
}

1104
void
1105
clearTypeInfo(TypeInfo *tp, int numTypes)
1106
{
1107
	int			i;
1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140

	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);
		if (tp[i].usename)
			free(tp[i].usename);
	}
	free(tp);
1141 1142 1143
}

void
1144
clearFuncInfo(FuncInfo *fun, int numFuncs)
1145
{
1146 1147
	int			i,
				a;
1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158

	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);
1159
		for (a = 0; a < FUNC_MAX_ARGS; ++a)
1160 1161 1162 1163 1164 1165 1166 1167 1168 1169
			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);
1170 1171
}

1172
static void
1173
clearTableInfo(TableInfo *tblinfo, int numTables)
1174
{
1175 1176
	int			i,
				j;
1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196

	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++)
		{
B
Bruce,  
Bruce Momjian 已提交
1197 1198
			if (tblinfo[i].attoids[j])
				free(tblinfo[i].attoids[j]);
1199 1200 1201 1202 1203
			if (tblinfo[i].attnames[j])
				free(tblinfo[i].attnames[j]);
			if (tblinfo[i].typnames[j])
				free(tblinfo[i].typnames[j]);
		}
1204 1205
		if (tblinfo[i].atttypmod)
			free((int *) tblinfo[i].atttypmod);
1206 1207 1208 1209 1210 1211 1212 1213
		if (tblinfo[i].inhAttrs)
			free((int *) tblinfo[i].inhAttrs);
		if (tblinfo[i].attnames)
			free(tblinfo[i].attnames);
		if (tblinfo[i].typnames)
			free(tblinfo[i].typnames);
		if (tblinfo[i].notnull)
			free(tblinfo[i].notnull);
1214 1215
		if (tblinfo[i].primary_key)
			free(tblinfo[i].primary_key);
1216 1217
	}
	free(tblinfo);
1218 1219
}

1220
void
1221
clearInhInfo(InhInfo *inh, int numInherits)
1222
{
1223
	int			i;
1224 1225 1226 1227 1228

	if (!inh)
		return;
	for (i = 0; i < numInherits; ++i)
	{
1229 1230
		if (inh[i].inhrelid)
			free(inh[i].inhrelid);
1231 1232 1233 1234
		if (inh[i].inhparent)
			free(inh[i].inhparent);
	}
	free(inh);
1235 1236 1237
}

void
1238
clearOprInfo(OprInfo *opr, int numOprs)
1239
{
1240
	int			i;
1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275

	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);
1276 1277 1278
}

void
1279
clearIndInfo(IndInfo *ind, int numIndices)
1280
{
1281 1282
	int			i,
				a;
1283 1284 1285 1286 1287

	if (!ind)
		return;
	for (i = 0; i < numIndices; ++i)
	{
B
Bruce,  
Bruce Momjian 已提交
1288 1289
		if (ind[i].indoid)
			free(ind[i].indoid);
1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308
		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);
		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);
1309 1310 1311
}

void
B
Bruce Momjian 已提交
1312
clearAggInfo(AggInfo *agginfo, int numArgs)
1313
{
1314
	int			i;
1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343

	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);
		if (agginfo[i].aggtransfn1)
			free(agginfo[i].aggtransfn1);
		if (agginfo[i].aggtransfn2)
			free(agginfo[i].aggtransfn2);
		if (agginfo[i].aggfinalfn)
			free(agginfo[i].aggfinalfn);
		if (agginfo[i].aggtranstype1)
			free(agginfo[i].aggtranstype1);
		if (agginfo[i].aggbasetype)
			free(agginfo[i].aggbasetype);
		if (agginfo[i].aggtranstype2)
			free(agginfo[i].aggtranstype2);
		if (agginfo[i].agginitval1)
			free(agginfo[i].agginitval1);
		if (agginfo[i].agginitval2)
			free(agginfo[i].agginitval2);
		if (agginfo[i].usename)
			free(agginfo[i].usename);
	}
	free(agginfo);
1344
}
1345 1346 1347

/*
 * getAggregates:
1348
 *	  read all the user-defined aggregates in the system catalogs and
1349 1350
 * return them in the AggInfo* structure
 *
1351 1352
 * numAggs is set to the number of aggregates read in
 *
1353 1354
 *
 */
1355
AggInfo    *
1356 1357
getAggregates(int *numAggs)
{
B
Bruce Momjian 已提交
1358
	PGresult   *res;
1359 1360
	int			ntups;
	int			i;
1361
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1362
	AggInfo    *agginfo;
1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374

	int			i_oid;
	int			i_aggname;
	int			i_aggtransfn1;
	int			i_aggtransfn2;
	int			i_aggfinalfn;
	int			i_aggtranstype1;
	int			i_aggbasetype;
	int			i_aggtranstype2;
	int			i_agginitval1;
	int			i_agginitval2;
	int			i_usename;
1375 1376 1377

	/* find all user-defined aggregates */

B
Hi, all  
Bruce Momjian 已提交
1378
	appendPQExpBuffer(query,
1379 1380
		   "SELECT pg_aggregate.oid, aggname, aggtransfn1, aggtransfn2, "
				"aggfinalfn, aggtranstype1, aggbasetype, aggtranstype2, "
1381
		  "agginitval1, agginitval2, usename from pg_aggregate, pg_user "
1382
					  "where aggowner = usesysid");
1383

B
Hi, all  
Bruce Momjian 已提交
1384
	res = PQexec(g_conn, query->data);
1385 1386 1387
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1388
		fprintf(stderr, "getAggregates(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426
		exit_nicely(g_conn);
	}

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

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

	i_oid = PQfnumber(res, "oid");
	i_aggname = PQfnumber(res, "aggname");
	i_aggtransfn1 = PQfnumber(res, "aggtransfn1");
	i_aggtransfn2 = PQfnumber(res, "aggtransfn2");
	i_aggfinalfn = PQfnumber(res, "aggfinalfn");
	i_aggtranstype1 = PQfnumber(res, "aggtranstype1");
	i_aggbasetype = PQfnumber(res, "aggbasetype");
	i_aggtranstype2 = PQfnumber(res, "aggtranstype2");
	i_agginitval1 = PQfnumber(res, "agginitval1");
	i_agginitval2 = PQfnumber(res, "agginitval2");
	i_usename = PQfnumber(res, "usename");

	for (i = 0; i < ntups; i++)
	{
		agginfo[i].oid = strdup(PQgetvalue(res, i, i_oid));
		agginfo[i].aggname = strdup(PQgetvalue(res, i, i_aggname));
		agginfo[i].aggtransfn1 = strdup(PQgetvalue(res, i, i_aggtransfn1));
		agginfo[i].aggtransfn2 = strdup(PQgetvalue(res, i, i_aggtransfn2));
		agginfo[i].aggfinalfn = strdup(PQgetvalue(res, i, i_aggfinalfn));
		agginfo[i].aggtranstype1 = strdup(PQgetvalue(res, i, i_aggtranstype1));
		agginfo[i].aggbasetype = strdup(PQgetvalue(res, i, i_aggbasetype));
		agginfo[i].aggtranstype2 = strdup(PQgetvalue(res, i, i_aggtranstype2));
		agginfo[i].agginitval1 = strdup(PQgetvalue(res, i, i_agginitval1));
		agginfo[i].agginitval2 = strdup(PQgetvalue(res, i, i_agginitval2));
		agginfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
	}

	PQclear(res);

	return agginfo;
1427 1428 1429 1430
}

/*
 * getFuncs:
1431
 *	  read all the user-defined functions in the system catalogs and
1432 1433
 * return them in the FuncInfo* structure
 *
1434 1435
 * numFuncs is set to the number of functions read in
 *
1436 1437
 *
 */
1438
FuncInfo   *
1439 1440
getFuncs(int *numFuncs)
{
1441
	PGresult   *res;
1442 1443
	int			ntups;
	int			i;
1444 1445
	PQExpBuffer query = createPQExpBuffer();
	FuncInfo   *finfo;
1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456

	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;
	int			i_usename;
1457 1458 1459

	/* find all user-defined funcs */

B
Hi, all  
Bruce Momjian 已提交
1460
	appendPQExpBuffer(query,
1461 1462 1463 1464 1465
		   "SELECT pg_proc.oid, proname, prolang, pronargs, prorettype, "
					  "proretset, proargtypes, prosrc, probin, usename "
					  "from pg_proc, pg_user "
				 "where pg_proc.oid > '%u'::oid and proowner = usesysid",
					  g_last_builtin_oid);
1466

B
Hi, all  
Bruce Momjian 已提交
1467
	res = PQexec(g_conn, query->data);
1468 1469 1470
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1471
		fprintf(stderr, "getFuncs(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
1472 1473 1474 1475 1476 1477 1478 1479 1480
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numFuncs = ntups;

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

1481 1482
	memset((char *) finfo, 0, ntups * sizeof(FuncInfo));

1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504
	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");
	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));

		finfo[i].prosrc = checkForQuote(PQgetvalue(res, i, i_prosrc));
		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));
1505
		finfo[i].lang = atoi(PQgetvalue(res, i, i_prolang));
1506
		finfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1507 1508 1509 1510
		if (finfo[i].nargs < 0 || finfo[i].nargs > FUNC_MAX_ARGS)
		{
			fprintf(stderr, "failed sanity check: %s has %d args\n",
					finfo[i].proname, finfo[i].nargs);
1511
			exit(1);
1512 1513 1514 1515
		}
		parseNumericArray(PQgetvalue(res, i, i_proargtypes),
						  finfo[i].argtypes,
						  finfo[i].nargs);
1516 1517 1518 1519 1520 1521
		finfo[i].dumped = 0;
	}

	PQclear(res);

	return finfo;
1522 1523 1524 1525 1526

}

/*
 * getTables
1527
 *	  read all the user-defined tables (no indices, no catalogs)
1528 1529
 * in the system catalogs return them in the TableInfo* structure
 *
1530 1531
 * numTables is set to the number of tables read in
 *
1532 1533
 *
 */
1534
TableInfo  *
V
Vadim B. Mikheev 已提交
1535
getTables(int *numTables, FuncInfo *finfo, int numFuncs)
1536
{
1537
	PGresult   *res;
1538 1539
	int			ntups;
	int			i;
1540 1541
	PQExpBuffer query = createPQExpBuffer();
	TableInfo  *tblinfo;
1542 1543 1544 1545 1546 1547

	int			i_oid;
	int			i_relname;
	int			i_relkind;
	int			i_relacl;
	int			i_usename;
V
Vadim B. Mikheev 已提交
1548 1549
	int			i_relchecks;
	int			i_reltriggers;
1550
	int			i_relhasindex;
1551 1552 1553 1554 1555 1556

	/*
	 * find all the user-defined tables (no indices and no catalogs),
	 * ordering by oid is important so that we always process the parent
	 * tables before the child tables when traversing the tblinfo*
	 *
1557 1558 1559
	 * we ignore tables that are not type 'r' (ordinary relation) or 'S'
	 * (sequence) --- in particular, Large Object relations (type 'l') are
	 * ignored.
1560 1561
	 */

B
Hi, all  
Bruce Momjian 已提交
1562
	appendPQExpBuffer(query,
1563 1564 1565 1566 1567 1568
			   "SELECT pg_class.oid, relname, relkind, relacl, usename, "
					  "relchecks, reltriggers, relhasindex "
					  "from pg_class, pg_user "
					  "where relowner = usesysid and "
				"(relkind = 'r' or relkind = 'S') and relname !~ '^pg_' "
					  "order by oid");
1569

B
Hi, all  
Bruce Momjian 已提交
1570
	res = PQexec(g_conn, query->data);
1571 1572 1573
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1574
		fprintf(stderr, "getTables(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numTables = ntups;

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

	i_oid = PQfnumber(res, "oid");
	i_relname = PQfnumber(res, "relname");
	i_relkind = PQfnumber(res, "relkind");
	i_relacl = PQfnumber(res, "relacl");
	i_usename = PQfnumber(res, "usename");
V
Vadim B. Mikheev 已提交
1589 1590
	i_relchecks = PQfnumber(res, "relchecks");
	i_reltriggers = PQfnumber(res, "reltriggers");
B
Hi, all  
Bruce Momjian 已提交
1591
	i_relhasindex = PQfnumber(res, "relhasindex");
1592 1593 1594 1595 1596 1597 1598 1599

	for (i = 0; i < ntups; i++)
	{
		tblinfo[i].oid = strdup(PQgetvalue(res, i, i_oid));
		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 已提交
1600 1601
		tblinfo[i].ncheck = atoi(PQgetvalue(res, i, i_relchecks));
		tblinfo[i].ntrig = atoi(PQgetvalue(res, i, i_reltriggers));
1602

B
Bruce Momjian 已提交
1603 1604 1605
		/*
		 * Exclude inherited CHECKs from CHECK constraints total. If a
		 * constraint matches by name and condition with a constraint
1606 1607 1608 1609
		 * belonging to a parent class, we assume it was inherited.
		 */
		if (tblinfo[i].ncheck > 0)
		{
B
Bruce Momjian 已提交
1610
			PGresult   *res2;
1611 1612
			int			ntups2;

B
Bruce Momjian 已提交
1613
			if (g_verbose)
1614 1615 1616 1617 1618 1619
				fprintf(stderr, "%s excluding inherited CHECK constraints "
						"for relation: '%s' %s\n",
						g_comment_start,
						tblinfo[i].relname,
						g_comment_end);

B
Hi, all  
Bruce Momjian 已提交
1620 1621
			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "SELECT rcname from pg_relcheck, pg_inherits as i "
1622 1623 1624 1625 1626 1627 1628 1629
							  "where rcrelid = '%s'::oid "
							  " and rcrelid = i.inhrelid"
							  " and exists "
							  "  (select * from pg_relcheck as c "
							  "    where c.rcname = pg_relcheck.rcname "
							  "      and c.rcsrc = pg_relcheck.rcsrc "
							  "      and c.rcrelid = i.inhparent) ",
							  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
1630
			res2 = PQexec(g_conn, query->data);
1631 1632 1633
			if (!res2 ||
				PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
B
Bruce Momjian 已提交
1634
				fprintf(stderr, "getTables(): SELECT (for inherited CHECK) failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649
				exit_nicely(g_conn);
			}
			ntups2 = PQntuples(res2);
			tblinfo[i].ncheck -= ntups2;
			if (tblinfo[i].ncheck < 0)
			{
				fprintf(stderr, "getTables(): found more inherited CHECKs than total for "
						"relation %s\n",
						tblinfo[i].relname);
				exit_nicely(g_conn);
			}
			PQclear(res2);
		}

		/* Get non-inherited CHECK constraints, if any */
V
Vadim B. Mikheev 已提交
1650 1651 1652
		if (tblinfo[i].ncheck > 0)
		{
			PGresult   *res2;
1653 1654
			int			i_rcname,
						i_rcsrc;
V
Vadim B. Mikheev 已提交
1655 1656
			int			ntups2;
			int			i2;
1657

V
Vadim B. Mikheev 已提交
1658
			if (g_verbose)
1659
				fprintf(stderr, "%s finding CHECK constraints for relation: '%s' %s\n",
V
Vadim B. Mikheev 已提交
1660 1661 1662
						g_comment_start,
						tblinfo[i].relname,
						g_comment_end);
1663

B
Hi, all  
Bruce Momjian 已提交
1664 1665
			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "SELECT rcname, rcsrc from pg_relcheck "
1666 1667 1668 1669 1670 1671 1672 1673
							  "where rcrelid = '%s'::oid "
							  "   and not exists "
				   "  (select * from pg_relcheck as c, pg_inherits as i "
							  "   where i.inhrelid = pg_relcheck.rcrelid "
							  "     and c.rcname = pg_relcheck.rcname "
							  "     and c.rcsrc = pg_relcheck.rcsrc "
							  "     and c.rcrelid = i.inhparent) ",
							  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
1674
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
1675
			if (!res2 ||
1676
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
1677
			{
B
Bruce Momjian 已提交
1678
				fprintf(stderr, "getTables(): SELECT (for CHECK) failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
V
Vadim B. Mikheev 已提交
1679 1680 1681 1682 1683
				exit_nicely(g_conn);
			}
			ntups2 = PQntuples(res2);
			if (ntups2 != tblinfo[i].ncheck)
			{
1684
				fprintf(stderr, "getTables(): relation '%s': %d CHECKs were expected, but got %d\n",
1685
						tblinfo[i].relname, tblinfo[i].ncheck, ntups2);
V
Vadim B. Mikheev 已提交
1686 1687 1688 1689
				exit_nicely(g_conn);
			}
			i_rcname = PQfnumber(res2, "rcname");
			i_rcsrc = PQfnumber(res2, "rcsrc");
1690
			tblinfo[i].check_expr = (char **) malloc(ntups2 * sizeof(char *));
V
Vadim B. Mikheev 已提交
1691 1692
			for (i2 = 0; i2 < ntups2; i2++)
			{
B
Bruce Momjian 已提交
1693 1694
				const char *name = PQgetvalue(res2, i2, i_rcname);
				const char *expr = PQgetvalue(res2, i2, i_rcsrc);
1695

B
Hi, all  
Bruce Momjian 已提交
1696
				resetPQExpBuffer(query);
1697 1698
				if (name[0] != '$')
				{
1699 1700
					appendPQExpBuffer(query, "CONSTRAINT %s ",
									  fmtId(name, force_quotes));
B
Bruce Momjian 已提交
1701
				}
1702
				appendPQExpBuffer(query, "CHECK (%s)", expr);
B
Hi, all  
Bruce Momjian 已提交
1703
				tblinfo[i].check_expr[i2] = strdup(query->data);
V
Vadim B. Mikheev 已提交
1704 1705 1706 1707 1708
			}
			PQclear(res2);
		}
		else
			tblinfo[i].check_expr = NULL;
1709

1710 1711
		/* Get primary key */
		if (strcmp(PQgetvalue(res, i, i_relhasindex), "t") == 0)
1712
		{
1713 1714 1715
			PGresult   *res2;
			char		str[INDEX_MAX_KEYS * (NAMEDATALEN * 2 + 4) + 1];
			int			j;
B
Bruce Momjian 已提交
1716

B
Hi, all  
Bruce Momjian 已提交
1717
			resetPQExpBuffer(query);
B
Bruce Momjian 已提交
1718
			appendPQExpBuffer(query,
1719 1720 1721 1722 1723 1724
							  "SELECT a.attname "
						   "FROM pg_index i, pg_class c, pg_attribute a "
							  "WHERE i.indisprimary AND i.indrelid = %s "
							  "  AND i.indexrelid = c.oid AND a.attnum > 0 AND a.attrelid = c.oid "
							  "ORDER BY a.attnum ",
							  tblinfo[i].oid);
1725
			res2 = PQexec(g_conn, query->data);
B
Bruce Momjian 已提交
1726 1727 1728
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
				fprintf(stderr, "getTables(): SELECT (for PRIMARY KEY) failed.  Explanation from backend: %s",
1729
						PQerrorMessage(g_conn));
B
Bruce Momjian 已提交
1730 1731
				exit_nicely(g_conn);
			}
1732

1733 1734 1735
			str[0] = '\0';
			for (j = 0; j < PQntuples(res2); j++)
			{
1736
				if (strlen(str) > 0)
1737 1738 1739 1740
					strcat(str, ", ");
				strcat(str, fmtId(PQgetvalue(res2, j, 0), force_quotes));
			}

1741 1742
			if (strlen(str) > 0)
			{
1743
				tblinfo[i].primary_key = strdup(str);
1744 1745
				if (tblinfo[i].primary_key == NULL)
				{
1746 1747 1748 1749 1750 1751
					perror("strdup");
					exit(1);
				}
			}
			else
				tblinfo[i].primary_key = NULL;
1752
		}
1753 1754
		else
			tblinfo[i].primary_key = NULL;
B
Bruce Momjian 已提交
1755

V
Vadim B. Mikheev 已提交
1756 1757 1758 1759
		/* Get Triggers */
		if (tblinfo[i].ntrig > 0)
		{
			PGresult   *res2;
B
Bruce,  
Bruce Momjian 已提交
1760
			int			i_tgoid,
1761
						i_tgname,
1762 1763 1764
						i_tgfoid,
						i_tgtype,
						i_tgnargs,
1765 1766 1767 1768 1769
						i_tgargs,
						i_tgisconstraint,
						i_tgconstrname,
						i_tgdeferrable,
						i_tginitdeferred;
V
Vadim B. Mikheev 已提交
1770 1771
			int			ntups2;
			int			i2;
1772

V
Vadim B. Mikheev 已提交
1773
			if (g_verbose)
1774
				fprintf(stderr, "%s finding Triggers for relation: '%s' %s\n",
V
Vadim B. Mikheev 已提交
1775 1776 1777
						g_comment_start,
						tblinfo[i].relname,
						g_comment_end);
1778

B
Hi, all  
Bruce Momjian 已提交
1779
			resetPQExpBuffer(query);
1780
			appendPQExpBuffer(query, "SELECT tgname, tgfoid, tgtype, tgnargs, tgargs, tgisconstraint, tgconstrname, tgdeferrable, tginitdeferred, oid "
1781 1782 1783
							  "from pg_trigger "
							  "where tgrelid = '%s'::oid ",
							  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
1784
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
1785
			if (!res2 ||
1786
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
1787
			{
B
Bruce Momjian 已提交
1788
				fprintf(stderr, "getTables(): SELECT (for TRIGGER) failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
V
Vadim B. Mikheev 已提交
1789 1790 1791 1792 1793
				exit_nicely(g_conn);
			}
			ntups2 = PQntuples(res2);
			if (ntups2 != tblinfo[i].ntrig)
			{
1794
				fprintf(stderr, "getTables(): relation '%s': %d Triggers were expected, but got %d\n",
1795
						tblinfo[i].relname, tblinfo[i].ntrig, ntups2);
V
Vadim B. Mikheev 已提交
1796 1797 1798 1799 1800 1801 1802
				exit_nicely(g_conn);
			}
			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 已提交
1803
			i_tgoid = PQfnumber(res2, "oid");
1804 1805 1806 1807 1808
			i_tgisconstraint = PQfnumber(res2, "tgisconstraint");
			i_tgconstrname = PQfnumber(res2, "tgconstrname");
			i_tgdeferrable = PQfnumber(res2, "tgdeferrable");
			i_tginitdeferred = PQfnumber(res2, "tginitdeferred");

1809
			tblinfo[i].triggers = (char **) malloc(ntups2 * sizeof(char *));
B
Bruce,  
Bruce Momjian 已提交
1810 1811
			tblinfo[i].trcomments = (char **) malloc(ntups2 * sizeof(char *));
			tblinfo[i].troids = (char **) malloc(ntups2 * sizeof(char *));
B
Hi, all  
Bruce Momjian 已提交
1812 1813
			resetPQExpBuffer(query);
			for (i2 = 0; i2 < ntups2; i2++)
V
Vadim B. Mikheev 已提交
1814
			{
1815
				const char *tgfuncoid = PQgetvalue(res2, i2, i_tgfoid);
1816
				char	   *tgfunc = NULL;
1817 1818
				int2		tgtype = atoi(PQgetvalue(res2, i2, i_tgtype));
				int			tgnargs = atoi(PQgetvalue(res2, i2, i_tgnargs));
B
Bruce Momjian 已提交
1819
				const char *tgargs = PQgetvalue(res2, i2, i_tgargs);
1820 1821 1822
				int			tgisconstraint;
				int			tgdeferrable;
				int			tginitdeferred;
B
Bruce Momjian 已提交
1823
				const char *p;
1824 1825
				int			findx;

1826
				if (strcmp(PQgetvalue(res2, i2, i_tgisconstraint), "f") == 0)
1827
					tgisconstraint = 0;
1828
				else
1829
					tgisconstraint = 1;
1830 1831

				if (strcmp(PQgetvalue(res2, i2, i_tgdeferrable), "f") == 0)
1832
					tgdeferrable = 0;
1833
				else
1834
					tgdeferrable = 1;
1835 1836

				if (strcmp(PQgetvalue(res2, i2, i_tginitdeferred), "f") == 0)
1837
					tginitdeferred = 0;
1838
				else
1839
					tginitdeferred = 1;
1840

V
Vadim B. Mikheev 已提交
1841 1842
				for (findx = 0; findx < numFuncs; findx++)
				{
1843
					if (strcmp(finfo[findx].oid, tgfuncoid) == 0 &&
1844
						finfo[findx].nargs == 0 &&
V
Vadim B. Mikheev 已提交
1845 1846 1847
						strcmp(finfo[findx].prorettype, "0") == 0)
						break;
				}
1848

V
Vadim B. Mikheev 已提交
1849 1850
				if (findx == numFuncs)
				{
1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871
					PGresult   *r;

					/*
					 * 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)
					{
						fprintf(stderr, "getTables(): SELECT (funcname) failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
						exit_nicely(g_conn);
					}
1872
					tgfunc = strdup(PQgetvalue(r, 0, PQfnumber(r, "proname")));
1873
					PQclear(r);
1874
				}
1875
				else
1876
					tgfunc = strdup(finfo[findx].proname);
B
Bruce Momjian 已提交
1877
#if 0
1878
				/* XXX - how to emit this DROP TRIGGER? */
1879
				if (dropSchema)
B
Bruce Momjian 已提交
1880
				{
B
Hi, all  
Bruce Momjian 已提交
1881
					resetPQExpBuffer(query);
1882
					appendPQExpBuffer(query, "DROP TRIGGER %s ",
1883 1884
									fmtId(PQgetvalue(res2, i2, i_tgname),
										  force_quotes));
1885
					appendPQExpBuffer(query, "ON %s;\n",
1886
								fmtId(tblinfo[i].relname, force_quotes));
B
Hi, all  
Bruce Momjian 已提交
1887
					fputs(query->data, fout);
B
Bruce Momjian 已提交
1888
				}
1889 1890
#endif

B
Hi, all  
Bruce Momjian 已提交
1891
				resetPQExpBuffer(query);
1892 1893 1894 1895 1896
				if (tgisconstraint)
				{
					appendPQExpBuffer(query, "CREATE CONSTRAINT TRIGGER ");
					appendPQExpBuffer(query, fmtId(PQgetvalue(res2, i2, i_tgconstrname), force_quotes));
				}
1897 1898
				else
				{
1899 1900 1901
					appendPQExpBuffer(query, "CREATE TRIGGER ");
					appendPQExpBuffer(query, fmtId(PQgetvalue(res2, i2, i_tgname), force_quotes));
				}
B
Bruce Momjian 已提交
1902
				appendPQExpBufferChar(query, ' ');
V
Vadim B. Mikheev 已提交
1903 1904 1905
				/* Trigger type */
				findx = 0;
				if (TRIGGER_FOR_BEFORE(tgtype))
B
Hi, all  
Bruce Momjian 已提交
1906
					appendPQExpBuffer(query, "BEFORE");
V
Vadim B. Mikheev 已提交
1907
				else
B
Hi, all  
Bruce Momjian 已提交
1908
					appendPQExpBuffer(query, "AFTER");
V
Vadim B. Mikheev 已提交
1909 1910
				if (TRIGGER_FOR_INSERT(tgtype))
				{
B
Hi, all  
Bruce Momjian 已提交
1911
					appendPQExpBuffer(query, " INSERT");
V
Vadim B. Mikheev 已提交
1912 1913 1914 1915 1916
					findx++;
				}
				if (TRIGGER_FOR_DELETE(tgtype))
				{
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
1917
						appendPQExpBuffer(query, " OR DELETE");
V
Vadim B. Mikheev 已提交
1918
					else
B
Hi, all  
Bruce Momjian 已提交
1919
						appendPQExpBuffer(query, " DELETE");
V
Vadim B. Mikheev 已提交
1920 1921 1922
					findx++;
				}
				if (TRIGGER_FOR_UPDATE(tgtype))
1923
				{
V
Vadim B. Mikheev 已提交
1924
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
1925
						appendPQExpBuffer(query, " OR UPDATE");
V
Vadim B. Mikheev 已提交
1926
					else
B
Hi, all  
Bruce Momjian 已提交
1927
						appendPQExpBuffer(query, " UPDATE");
1928
				}
1929 1930
				appendPQExpBuffer(query, " ON %s ", fmtId(tblinfo[i].relname, force_quotes));

1931
				if (tgisconstraint)
1932
				{
1933
					if (!tgdeferrable)
1934 1935 1936 1937
						appendPQExpBuffer(query, " NOT");
					appendPQExpBuffer(query, " DEFERRABLE INITIALLY ");
					if (tginitdeferred)
						appendPQExpBuffer(query, "DEFERRED");
1938
					else
1939
						appendPQExpBuffer(query, "IMMEDIATE");
1940

1941 1942 1943
				}

				appendPQExpBuffer(query, " FOR EACH ROW");
1944 1945
				appendPQExpBuffer(query, " EXECUTE PROCEDURE %s (",
								  fmtId(tgfunc, force_quotes));
V
Vadim B. Mikheev 已提交
1946 1947
				for (findx = 0; findx < tgnargs; findx++)
				{
1948
					const char *s;
1949 1950

					for (p = tgargs;;)
V
Vadim B. Mikheev 已提交
1951
					{
1952
						p = strchr(p, '\\');
V
Vadim B. Mikheev 已提交
1953 1954
						if (p == NULL)
						{
1955
							fprintf(stderr, "getTables(): relation '%s': bad argument string (%s) for trigger '%s'\n",
1956 1957 1958
									tblinfo[i].relname,
									PQgetvalue(res2, i2, i_tgargs),
									PQgetvalue(res2, i2, i_tgname));
V
Vadim B. Mikheev 已提交
1959 1960 1961 1962 1963 1964 1965 1966
							exit_nicely(g_conn);
						}
						p++;
						if (*p == '\\')
						{
							p++;
							continue;
						}
1967
						if (p[0] == '0' && p[1] == '0' && p[2] == '0')
V
Vadim B. Mikheev 已提交
1968 1969 1970
							break;
					}
					p--;
1971
					appendPQExpBufferChar(query, '\'');
B
Bruce Momjian 已提交
1972
					for (s = tgargs; s < p;)
V
Vadim B. Mikheev 已提交
1973 1974
					{
						if (*s == '\'')
1975 1976
							appendPQExpBufferChar(query, '\\');
						appendPQExpBufferChar(query, *s++);
V
Vadim B. Mikheev 已提交
1977
					}
B
Bruce Momjian 已提交
1978 1979
					appendPQExpBufferChar(query, '\'');
					appendPQExpBuffer(query, (findx < tgnargs - 1) ? ", " : "");
V
Vadim B. Mikheev 已提交
1980 1981
					tgargs = p + 4;
				}
B
Hi, all  
Bruce Momjian 已提交
1982
				appendPQExpBuffer(query, ");\n");
1983

B
Hi, all  
Bruce Momjian 已提交
1984
				tblinfo[i].triggers[i2] = strdup(query->data);
B
Bruce,  
Bruce Momjian 已提交
1985 1986

				/*** Initialize trcomments and troids ***/
1987

B
Bruce,  
Bruce Momjian 已提交
1988
				resetPQExpBuffer(query);
1989 1990
				appendPQExpBuffer(query, "TRIGGER %s ",
					fmtId(PQgetvalue(res2, i2, i_tgname), force_quotes));
B
Bruce,  
Bruce Momjian 已提交
1991
				appendPQExpBuffer(query, "ON %s",
1992
								fmtId(tblinfo[i].relname, force_quotes));
B
Bruce,  
Bruce Momjian 已提交
1993 1994 1995
				tblinfo[i].trcomments[i2] = strdup(query->data);
				tblinfo[i].troids[i2] = strdup(PQgetvalue(res2, i2, i_tgoid));

1996 1997
				if (tgfunc)
					free(tgfunc);
V
Vadim B. Mikheev 已提交
1998 1999 2000 2001
			}
			PQclear(res2);
		}
		else
B
Bruce,  
Bruce Momjian 已提交
2002
		{
V
Vadim B. Mikheev 已提交
2003
			tblinfo[i].triggers = NULL;
B
Bruce,  
Bruce Momjian 已提交
2004 2005 2006
			tblinfo[i].trcomments = NULL;
			tblinfo[i].troids = NULL;
		}
2007

2008 2009 2010 2011 2012
	}

	PQclear(res);

	return tblinfo;
2013 2014 2015 2016 2017

}

/*
 * getInherits
2018
 *	  read all the inheritance information
2019 2020
 * from the system catalogs return them in the InhInfo* structure
 *
2021 2022
 * numInherits is set to the number of tables read in
 *
2023 2024
 *
 */
2025
InhInfo    *
2026 2027
getInherits(int *numInherits)
{
2028
	PGresult   *res;
2029 2030
	int			ntups;
	int			i;
2031 2032
	PQExpBuffer query = createPQExpBuffer();
	InhInfo    *inhinfo;
2033

2034
	int			i_inhrelid;
2035
	int			i_inhparent;
2036 2037 2038

	/* find all the inheritance information */

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

B
Hi, all  
Bruce Momjian 已提交
2041
	res = PQexec(g_conn, query->data);
2042 2043 2044
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
2045
		fprintf(stderr, "getInherits(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2046 2047 2048 2049 2050 2051 2052 2053 2054
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numInherits = ntups;

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

2055
	i_inhrelid = PQfnumber(res, "inhrelid");
2056 2057 2058 2059
	i_inhparent = PQfnumber(res, "inhparent");

	for (i = 0; i < ntups; i++)
	{
2060
		inhinfo[i].inhrelid = strdup(PQgetvalue(res, i, i_inhrelid));
2061 2062 2063 2064 2065
		inhinfo[i].inhparent = strdup(PQgetvalue(res, i, i_inhparent));
	}

	PQclear(res);
	return inhinfo;
2066 2067 2068 2069
}

/*
 * getTableAttrs -
2070 2071
 *	  for each table in tblinfo, read its attributes types and names
 *
2072
 * this is implemented in a very inefficient way right now, looping
2073
 * through the tblinfo and doing a join per table to find the attrs and their
2074 2075
 * types
 *
2076
 *	modifies tblinfo
2077 2078
 */
void
2079
getTableAttrs(TableInfo *tblinfo, int numTables)
2080
{
2081 2082
	int			i,
				j;
2083
	PQExpBuffer q = createPQExpBuffer();
2084 2085
	int			i_attname;
	int			i_typname;
2086
	int			i_atttypmod;
2087
	int			i_attnotnull;
V
Vadim B. Mikheev 已提交
2088
	int			i_atthasdef;
2089 2090
	int			i_attoid;
	PGresult   *res;
2091
	int			ntups;
2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106

	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)
2107
			fprintf(stderr, "%s finding the attrs and types for table: '%s' %s\n",
2108 2109 2110 2111
					g_comment_start,
					tblinfo[i].relname,
					g_comment_end);

B
Hi, all  
Bruce Momjian 已提交
2112
		resetPQExpBuffer(q);
B
Bruce,  
Bruce Momjian 已提交
2113
		appendPQExpBuffer(q, "SELECT a.oid as attoid, a.attnum, a.attname, t.typname, a.atttypmod, "
2114 2115 2116 2117 2118
						  "a.attnotnull, a.atthasdef "
						  "from pg_attribute a, pg_type t "
				   "where a.attrelid = '%s'::oid and a.atttypid = t.oid "
						  "and a.attnum > 0 order by attnum",
						  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
2119
		res = PQexec(g_conn, q->data);
2120 2121 2122
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
B
Bruce Momjian 已提交
2123
			fprintf(stderr, "getTableAttrs(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2124 2125 2126 2127 2128
			exit_nicely(g_conn);
		}

		ntups = PQntuples(res);

B
Bruce,  
Bruce Momjian 已提交
2129
		i_attoid = PQfnumber(res, "attoid");
2130 2131
		i_attname = PQfnumber(res, "attname");
		i_typname = PQfnumber(res, "typname");
2132
		i_atttypmod = PQfnumber(res, "atttypmod");
2133
		i_attnotnull = PQfnumber(res, "attnotnull");
V
Vadim B. Mikheev 已提交
2134
		i_atthasdef = PQfnumber(res, "atthasdef");
2135 2136

		tblinfo[i].numatts = ntups;
B
Bruce,  
Bruce Momjian 已提交
2137
		tblinfo[i].attoids = (char **) malloc(ntups * sizeof(char *));
2138 2139
		tblinfo[i].attnames = (char **) malloc(ntups * sizeof(char *));
		tblinfo[i].typnames = (char **) malloc(ntups * sizeof(char *));
2140
		tblinfo[i].atttypmod = (int *) malloc(ntups * sizeof(int));
2141 2142
		tblinfo[i].inhAttrs = (int *) malloc(ntups * sizeof(int));
		tblinfo[i].notnull = (bool *) malloc(ntups * sizeof(bool));
V
Vadim B. Mikheev 已提交
2143
		tblinfo[i].adef_expr = (char **) malloc(ntups * sizeof(char *));
2144 2145 2146 2147
		tblinfo[i].parentRels = NULL;
		tblinfo[i].numParents = 0;
		for (j = 0; j < ntups; j++)
		{
B
Bruce,  
Bruce Momjian 已提交
2148
			tblinfo[i].attoids[j] = strdup(PQgetvalue(res, j, i_attoid));
2149 2150
			tblinfo[i].attnames[j] = strdup(PQgetvalue(res, j, i_attname));
			tblinfo[i].typnames[j] = strdup(PQgetvalue(res, j, i_typname));
2151
			tblinfo[i].atttypmod[j] = atoi(PQgetvalue(res, j, i_atttypmod));
2152 2153 2154
			tblinfo[i].inhAttrs[j] = 0; /* this flag is set in
										 * flagInhAttrs() */
			tblinfo[i].notnull[j] = (PQgetvalue(res, j, i_attnotnull)[0] == 't') ? true : false;
V
Vadim B. Mikheev 已提交
2155 2156 2157
			if (PQgetvalue(res, j, i_atthasdef)[0] == 't')
			{
				PGresult   *res2;
2158

V
Vadim B. Mikheev 已提交
2159
				if (g_verbose)
2160
					fprintf(stderr, "%s finding DEFAULT expression for attr: '%s' %s\n",
V
Vadim B. Mikheev 已提交
2161 2162 2163
							g_comment_start,
							tblinfo[i].attnames[j],
							g_comment_end);
2164

B
Hi, all  
Bruce Momjian 已提交
2165 2166
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "SELECT adsrc from pg_attrdef "
2167 2168
							 "where adrelid = '%s'::oid and adnum = %d ",
								  tblinfo[i].oid, j + 1);
B
Hi, all  
Bruce Momjian 已提交
2169
				res2 = PQexec(g_conn, q->data);
V
Vadim B. Mikheev 已提交
2170
				if (!res2 ||
2171
					PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2172
				{
B
Bruce Momjian 已提交
2173
					fprintf(stderr, "getTableAttrs(): SELECT (for DEFAULT) failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
V
Vadim B. Mikheev 已提交
2174 2175 2176 2177 2178 2179 2180
					exit_nicely(g_conn);
				}
				tblinfo[i].adef_expr[j] = strdup(PQgetvalue(res2, 0, PQfnumber(res2, "adsrc")));
				PQclear(res2);
			}
			else
				tblinfo[i].adef_expr[j] = NULL;
2181 2182 2183
		}
		PQclear(res);
	}
2184 2185 2186 2187 2188
}


/*
 * getIndices
2189
 *	  read all the user-defined indices information
2190 2191
 * from the system catalogs return them in the InhInfo* structure
 *
2192 2193
 * numIndices is set to the number of indices read in
 *
2194 2195
 *
 */
2196
IndInfo    *
2197 2198
getIndices(int *numIndices)
{
2199
	int			i;
2200
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
2201
	PGresult   *res;
2202
	int			ntups;
B
Bruce Momjian 已提交
2203
	IndInfo    *indinfo;
2204 2205 2206 2207 2208 2209 2210 2211

	int			i_indexrelname;
	int			i_indrelname;
	int			i_indamname;
	int			i_indproc;
	int			i_indkey;
	int			i_indclass;
	int			i_indisunique;
2212
	int			i_indoid;
2213 2214 2215 2216 2217

	/*
	 * find all the user-defined indices. We do not handle partial
	 * indices.
	 *
2218
	 * Notice we skip indices on inversion objects (relkind 'l')
2219 2220 2221 2222
	 *
	 * this is a 4-way join !!
	 */

B
Hi, all  
Bruce Momjian 已提交
2223
	appendPQExpBuffer(query,
2224 2225 2226 2227 2228 2229 2230 2231
					  "SELECT t1.oid as indoid, t1.relname as indexrelname, t2.relname as indrelname, "
					  "i.indproc, i.indkey, i.indclass, "
					  "a.amname as indamname, i.indisunique "
					"from pg_index i, pg_class t1, pg_class t2, pg_am a "
				   "WHERE t1.oid = i.indexrelid and t2.oid = i.indrelid "
					  "and t1.relam = a.oid and i.indexrelid > '%u'::oid "
					  "and t2.relname !~ '^pg_' and t2.relkind != 'l' and not i.indisprimary",
					  g_last_builtin_oid);
2232

B
Hi, all  
Bruce Momjian 已提交
2233
	res = PQexec(g_conn, query->data);
2234 2235 2236
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
2237
		fprintf(stderr, "getIndices(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2238 2239 2240 2241 2242 2243 2244 2245 2246
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numIndices = ntups;

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

2247 2248
	memset((char *) indinfo, 0, ntups * sizeof(IndInfo));

B
Bruce,  
Bruce Momjian 已提交
2249
	i_indoid = PQfnumber(res, "indoid");
2250 2251 2252 2253 2254 2255 2256 2257 2258 2259
	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");

	for (i = 0; i < ntups; i++)
	{
B
Bruce,  
Bruce Momjian 已提交
2260
		indinfo[i].indoid = strdup(PQgetvalue(res, i, i_indoid));
2261 2262 2263 2264
		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));
2265 2266 2267 2268 2269 2270
		parseNumericArray(PQgetvalue(res, i, i_indkey),
						  indinfo[i].indkey,
						  INDEX_MAX_KEYS);
		parseNumericArray(PQgetvalue(res, i, i_indclass),
						  indinfo[i].indclass,
						  INDEX_MAX_KEYS);
2271 2272 2273 2274
		indinfo[i].indisunique = strdup(PQgetvalue(res, i, i_indisunique));
	}
	PQclear(res);
	return indinfo;
2275 2276
}

B
Bruce,  
Bruce Momjian 已提交
2277
/*------------------------------------------------------------------
2278
 * dumpComments --
B
Bruce,  
Bruce Momjian 已提交
2279
 *
2280
 * This routine is used to dump any comments associated with the
B
Bruce,  
Bruce Momjian 已提交
2281 2282 2283 2284
 * oid handed to this routine. The routine takes a constant character
 * string for the target part of the object and the oid of the object
 * whose comments are to be dumped. It is perfectly acceptable
 * to hand an oid to this routine which has not been commented. In
2285
 * addition, the routine takes the stdio FILE handle to which the
B
Bruce,  
Bruce Momjian 已提交
2286 2287 2288 2289
 * output should be written.
 *------------------------------------------------------------------
*/

T
Tom Lane 已提交
2290
static void
2291 2292
dumpComment(FILE *fout, const char *target, const char *oid)
{
B
Bruce,  
Bruce Momjian 已提交
2293

2294
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
2295
	PQExpBuffer query;
2296
	int			i_description;
B
Bruce,  
Bruce Momjian 已提交
2297 2298 2299 2300 2301 2302 2303 2304 2305 2306

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

	query = createPQExpBuffer();
	appendPQExpBuffer(query, "SELECT description FROM pg_description WHERE objoid = ");
	appendPQExpBuffer(query, oid);

	/*** Execute query ***/

	res = PQexec(g_conn, query->data);
2307 2308
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce,  
Bruce Momjian 已提交
2309 2310 2311 2312 2313 2314 2315
		fprintf(stderr, "DumpComment: SELECT failed: '%s'.\n",
				PQerrorMessage(g_conn));
		exit_nicely(g_conn);
	}

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

2316 2317
	if (PQntuples(res) != 0)
	{
B
Bruce,  
Bruce Momjian 已提交
2318
		i_description = PQfnumber(res, "description");
2319
		fprintf(fout, "COMMENT ON %s IS '%s';\n",
B
Bruce,  
Bruce Momjian 已提交
2320 2321 2322 2323 2324 2325
				target, checkForQuote(PQgetvalue(res, 0, i_description)));
	}

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

	PQclear(res);
2326

B
Bruce,  
Bruce Momjian 已提交
2327 2328 2329
}

/*------------------------------------------------------------------
2330
 * dumpDBComment --
B
Bruce,  
Bruce Momjian 已提交
2331
 *
2332 2333
 * 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 已提交
2334 2335 2336 2337 2338
 * to dump the schema of the database, then this is the first
 * statement issued.
 *------------------------------------------------------------------
*/

2339 2340 2341
void
dumpDBComment(FILE *fout)
{
B
Bruce,  
Bruce Momjian 已提交
2342

2343
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
2344
	PQExpBuffer query;
2345
	int			i_oid;
B
Bruce,  
Bruce Momjian 已提交
2346 2347 2348 2349 2350 2351 2352 2353 2354 2355

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

	query = createPQExpBuffer();
	appendPQExpBuffer(query, "SELECT oid FROM pg_database WHERE datname = '%s'",
					  PQdb(g_conn));

	/*** Execute query ***/

	res = PQexec(g_conn, query->data);
2356 2357
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce,  
Bruce Momjian 已提交
2358 2359 2360 2361 2362 2363 2364
		fprintf(stderr, "dumpDBComment: SELECT failed: '%s'.\n",
				PQerrorMessage(g_conn));
		exit_nicely(g_conn);
	}

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

2365 2366
	if (PQntuples(res) != 0)
	{
B
Bruce,  
Bruce Momjian 已提交
2367 2368 2369 2370 2371 2372 2373 2374 2375
		i_oid = PQfnumber(res, "oid");
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "DATABASE %s", fmtId(PQdb(g_conn), force_quotes));
		dumpComment(fout, query->data, PQgetvalue(res, 0, i_oid));
	}

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

	PQclear(res);
2376

B
Bruce,  
Bruce Momjian 已提交
2377 2378
}

2379 2380
/*
 * dumpTypes
2381
 *	  writes out to fout the queries to recreate all the user-defined types
2382 2383 2384
 *
 */
void
2385 2386
dumpTypes(FILE *fout, FuncInfo *finfo, int numFuncs,
		  TypeInfo *tinfo, int numTypes)
2387
{
2388
	int			i;
2389
	PQExpBuffer q = createPQExpBuffer();
2390
	int			funcInd;
2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419

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

		/* skip all the builtin types */
		if (atoi(tinfo[i].oid) < g_last_builtin_oid)
			continue;

		/* skip relation types */
		if (atoi(tinfo[i].typrelid) != 0)
			continue;

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

		/*
		 * 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)
			dumpOneFunc(fout, finfo, funcInd, tinfo, numTypes);

		funcInd = findFuncByName(finfo, numFuncs, tinfo[i].typoutput);
		if (funcInd != -1)
			dumpOneFunc(fout, finfo, funcInd, tinfo, numTypes);

2420
		becomeUser(fout, tinfo[i].usename);
2421

2422
		if (dropSchema)
B
Bruce Momjian 已提交
2423
		{
B
Hi, all  
Bruce Momjian 已提交
2424 2425 2426
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "DROP TYPE %s;\n", fmtId(tinfo[i].typname, force_quotes));
			fputs(q->data, fout);
B
Bruce Momjian 已提交
2427
		}
2428

B
Hi, all  
Bruce Momjian 已提交
2429 2430
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441
						  "CREATE TYPE %s "
			   "( internallength = %s, externallength = %s, input = %s, "
				  "output = %s, send = %s, receive = %s, default = '%s'",
						  fmtId(tinfo[i].typname, force_quotes),
						  tinfo[i].typlen,
						  tinfo[i].typprtlen,
						  tinfo[i].typinput,
						  tinfo[i].typoutput,
						  tinfo[i].typsend,
						  tinfo[i].typreceive,
						  tinfo[i].typdefault);
2442 2443 2444

		if (tinfo[i].isArray)
		{
2445
			char	   *elemType;
2446 2447 2448

			elemType = findTypeByOid(tinfo, numTypes, tinfo[i].typelem);

B
Hi, all  
Bruce Momjian 已提交
2449
			appendPQExpBuffer(q, ", element = %s, delimiter = '%s'",
2450
							  elemType, tinfo[i].typdelim);
2451 2452
		}
		if (tinfo[i].passedbyvalue)
B
Hi, all  
Bruce Momjian 已提交
2453
			appendPQExpBuffer(q, ",passedbyvalue);\n");
2454
		else
B
Hi, all  
Bruce Momjian 已提交
2455
			appendPQExpBuffer(q, ");\n");
2456

B
Hi, all  
Bruce Momjian 已提交
2457
		fputs(q->data, fout);
B
Bruce,  
Bruce Momjian 已提交
2458 2459 2460 2461 2462 2463 2464

		/*** Dump Type Comments ***/

		resetPQExpBuffer(q);
		appendPQExpBuffer(q, "TYPE %s", fmtId(tinfo[i].typname, force_quotes));
		dumpComment(fout, q->data, tinfo[i].oid);

2465
	}
2466 2467
}

2468 2469
/*
 * dumpProcLangs
B
Bruce Momjian 已提交
2470
 *		  writes out to fout the queries to recreate user-defined procedural languages
2471 2472 2473 2474
 *
 */
void
dumpProcLangs(FILE *fout, FuncInfo *finfo, int numFuncs,
B
Bruce Momjian 已提交
2475
			  TypeInfo *tinfo, int numTypes)
2476
{
2477 2478
	PGresult   *res;
	PQExpBuffer query = createPQExpBuffer();
2479 2480 2481 2482 2483
	int			ntups;
	int			i_lanname;
	int			i_lanpltrusted;
	int			i_lanplcallfoid;
	int			i_lancompiler;
2484 2485 2486
	char	   *lanname;
	char	   *lancompiler;
	const char *lanplcallfoid;
2487 2488 2489
	int			i,
				fidx;

B
Hi, all  
Bruce Momjian 已提交
2490
	appendPQExpBuffer(query, "SELECT * FROM pg_language "
2491 2492
					  "WHERE lanispl "
					  "ORDER BY oid");
B
Hi, all  
Bruce Momjian 已提交
2493
	res = PQexec(g_conn, query->data);
2494 2495 2496
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
2497
		fprintf(stderr, "dumpProcLangs(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2498 2499 2500 2501
		exit_nicely(g_conn);
	}
	ntups = PQntuples(res);

B
Bruce Momjian 已提交
2502 2503 2504 2505
	i_lanname = PQfnumber(res, "lanname");
	i_lanpltrusted = PQfnumber(res, "lanpltrusted");
	i_lanplcallfoid = PQfnumber(res, "lanplcallfoid");
	i_lancompiler = PQfnumber(res, "lancompiler");
2506

B
Bruce Momjian 已提交
2507 2508
	for (i = 0; i < ntups; i++)
	{
2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525
		lanplcallfoid = PQgetvalue(res, i, i_lanplcallfoid);
		for (fidx = 0; fidx < numFuncs; fidx++)
		{
			if (!strcmp(finfo[fidx].oid, lanplcallfoid))
				break;
		}
		if (fidx >= numFuncs)
		{
			fprintf(stderr, "dumpProcLangs(): handler procedure for language %s not found\n", PQgetvalue(res, i, i_lanname));
			exit_nicely(g_conn);
		}

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

		lanname = checkForQuote(PQgetvalue(res, i, i_lanname));
		lancompiler = checkForQuote(PQgetvalue(res, i, i_lancompiler));

2526
		if (dropSchema)
B
Bruce Momjian 已提交
2527
			fprintf(fout, "DROP PROCEDURAL LANGUAGE '%s';\n", lanname);
2528

2529
		fprintf(fout, "CREATE %sPROCEDURAL LANGUAGE '%s' "
B
Bruce Momjian 已提交
2530 2531 2532 2533 2534
				"HANDLER %s LANCOMPILER '%s';\n",
		(PQgetvalue(res, i, i_lanpltrusted)[0] == 't') ? "TRUSTED " : "",
				lanname,
				fmtId(finfo[fidx].proname, force_quotes),
				lancompiler);
2535 2536 2537 2538 2539 2540 2541 2542 2543

		free(lanname);
		free(lancompiler);
	}

	PQclear(res);

}

2544 2545
/*
 * dumpFuncs
2546
 *	  writes out to fout the queries to recreate all the user-defined functions
2547 2548 2549
 *
 */
void
2550 2551
dumpFuncs(FILE *fout, FuncInfo *finfo, int numFuncs,
		  TypeInfo *tinfo, int numTypes)
2552
{
2553
	int			i;
2554 2555 2556

	for (i = 0; i < numFuncs; i++)
		dumpOneFunc(fout, finfo, i, tinfo, numTypes);
2557 2558 2559 2560
}

/*
 * dumpOneFunc:
2561 2562
 *	  dump out only one function,  the index of which is given in the third
 *	argument
2563 2564 2565
 *
 */

2566
static void
2567 2568
dumpOneFunc(FILE *fout, FuncInfo *finfo, int i,
			TypeInfo *tinfo, int numTypes)
2569
{
2570
	PQExpBuffer q = createPQExpBuffer();
B
Bruce,  
Bruce Momjian 已提交
2571
	PQExpBuffer fnlist = createPQExpBuffer();
B
Bruce Momjian 已提交
2572
	int			j;
2573
	char	   *func_def;
2574
	char		func_lang[NAMEDATALEN + 1];
2575 2576 2577 2578
	PGresult   *res;
	int			nlangs;
	int			i_lanname;
	char		query[256];
2579 2580 2581 2582 2583 2584

	if (finfo[i].dumped)
		return;
	else
		finfo[i].dumped = 1;

2585
	becomeUser(fout, finfo[i].usename);
2586

2587 2588 2589 2590 2591
	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)
2592
	{
2593 2594
		fprintf(stderr, "dumpOneFunc(): SELECT for procedural language failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
		exit_nicely(g_conn);
2595
	}
2596 2597 2598
	nlangs = PQntuples(res);

	if (nlangs != 1)
2599
	{
2600 2601
		fprintf(stderr, "dumpOneFunc(): procedural language for function %s not found\n", finfo[i].proname);
		exit_nicely(g_conn);
2602 2603
	}

2604
	i_lanname = PQfnumber(res, "lanname");
2605

2606 2607
	func_def = finfo[i].prosrc;
	strcpy(func_lang, PQgetvalue(res, 0, i_lanname));
2608

2609
	PQclear(res);
2610

2611
	if (dropSchema)
B
Bruce Momjian 已提交
2612
	{
B
Hi, all  
Bruce Momjian 已提交
2613 2614
		resetPQExpBuffer(q);
		appendPQExpBuffer(q, "DROP FUNCTION %s (", fmtId(finfo[i].proname, force_quotes));
B
Bruce Momjian 已提交
2615 2616 2617 2618 2619
		for (j = 0; j < finfo[i].nargs; j++)
		{
			char	   *typname;

			typname = findTypeByOid(tinfo, numTypes, finfo[i].argtypes[j]);
B
Hi, all  
Bruce Momjian 已提交
2620
			appendPQExpBuffer(q, "%s%s",
2621 2622
							  (j > 0) ? "," : "",
							  fmtId(typname, false));
B
Bruce Momjian 已提交
2623
		}
B
Hi, all  
Bruce Momjian 已提交
2624 2625
		appendPQExpBuffer(q, ");\n");
		fputs(q->data, fout);
B
Bruce Momjian 已提交
2626
	}
2627

B
Hi, all  
Bruce Momjian 已提交
2628 2629
	resetPQExpBuffer(q);
	appendPQExpBuffer(q, "CREATE FUNCTION %s (", fmtId(finfo[i].proname, force_quotes));
2630 2631
	for (j = 0; j < finfo[i].nargs; j++)
	{
2632
		char	   *typname;
2633 2634

		typname = findTypeByOid(tinfo, numTypes, finfo[i].argtypes[j]);
B
Hi, all  
Bruce Momjian 已提交
2635
		appendPQExpBuffer(q, "%s%s",
2636 2637
						  (j > 0) ? "," : "",
						  fmtId(typname, false));
B
Bruce,  
Bruce Momjian 已提交
2638 2639 2640
		appendPQExpBuffer(fnlist, "%s%s",
						  (j > 0) ? "," : "",
						  fmtId(typname, false));
2641
	}
B
Hi, all  
Bruce Momjian 已提交
2642
	appendPQExpBuffer(q, " ) RETURNS %s%s AS '%s' LANGUAGE '%s';\n",
2643 2644 2645
					  (finfo[i].retset) ? " SETOF " : "",
	   fmtId(findTypeByOid(tinfo, numTypes, finfo[i].prorettype), false),
					  func_def, func_lang);
2646

B
Hi, all  
Bruce Momjian 已提交
2647
	fputs(q->data, fout);
2648

B
Bruce,  
Bruce Momjian 已提交
2649 2650 2651 2652 2653 2654 2655 2656
	/*** Dump Function Comments ***/

	resetPQExpBuffer(q);
	appendPQExpBuffer(q, "FUNCTION %s ",
					  fmtId(finfo[i].proname, force_quotes));
	appendPQExpBuffer(q, "( %s )", fnlist->data);
	dumpComment(fout, q->data, finfo[i].oid);

2657 2658 2659 2660
}

/*
 * dumpOprs
2661
 *	  writes out to fout the queries to recreate all the user-defined operators
2662 2663
 *
 */
2664
void
2665 2666
dumpOprs(FILE *fout, OprInfo *oprinfo, int numOperators,
		 TypeInfo *tinfo, int numTypes)
2667
{
B
Bruce Momjian 已提交
2668
	int			i;
2669 2670 2671 2672 2673 2674 2675 2676 2677
	PQExpBuffer q = createPQExpBuffer();
	PQExpBuffer leftarg = createPQExpBuffer();
	PQExpBuffer rightarg = createPQExpBuffer();
	PQExpBuffer commutator = createPQExpBuffer();
	PQExpBuffer negator = createPQExpBuffer();
	PQExpBuffer restrictor = createPQExpBuffer();
	PQExpBuffer join = createPQExpBuffer();
	PQExpBuffer sort1 = createPQExpBuffer();
	PQExpBuffer sort2 = createPQExpBuffer();
2678 2679 2680 2681

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

B
Hi, all  
Bruce Momjian 已提交
2682 2683 2684 2685 2686 2687 2688 2689 2690
		resetPQExpBuffer(leftarg);
		resetPQExpBuffer(rightarg);
		resetPQExpBuffer(commutator);
		resetPQExpBuffer(negator);
		resetPQExpBuffer(restrictor);
		resetPQExpBuffer(join);
		resetPQExpBuffer(sort1);
		resetPQExpBuffer(sort2);

2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708
		/* skip all the builtin oids */
		if (atoi(oprinfo[i].oid) < g_last_builtin_oid)
			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
Hi, all  
Bruce Momjian 已提交
2709
			appendPQExpBuffer(leftarg, ",\n\tLEFTARG = %s ",
2710
							  fmtId(findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft), false));
2711 2712 2713 2714
		}
		if (strcmp(oprinfo[i].oprkind, "l") == 0 ||
			strcmp(oprinfo[i].oprkind, "b") == 0)
		{
B
Hi, all  
Bruce Momjian 已提交
2715
			appendPQExpBuffer(rightarg, ",\n\tRIGHTARG = %s ",
2716
							  fmtId(findTypeByOid(tinfo, numTypes, oprinfo[i].oprright), false));
2717
		}
B
Hi, all  
Bruce Momjian 已提交
2718 2719
		if (!(strcmp(oprinfo[i].oprcom, "0") == 0))
			appendPQExpBuffer(commutator, ",\n\tCOMMUTATOR = %s ",
2720 2721
				 findOprByOid(oprinfo, numOperators, oprinfo[i].oprcom));

B
Hi, all  
Bruce Momjian 已提交
2722 2723
		if (!(strcmp(oprinfo[i].oprnegate, "0") == 0))
			appendPQExpBuffer(negator, ",\n\tNEGATOR = %s ",
2724 2725
			  findOprByOid(oprinfo, numOperators, oprinfo[i].oprnegate));

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

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

B
Hi, all  
Bruce Momjian 已提交
2732 2733
		if (!(strcmp(oprinfo[i].oprlsortop, "0") == 0))
			appendPQExpBuffer(sort1, ",\n\tSORT1 = %s ",
B
Bruce Momjian 已提交
2734
			 findOprByOid(oprinfo, numOperators, oprinfo[i].oprlsortop));
2735

B
Hi, all  
Bruce Momjian 已提交
2736 2737
		if (!(strcmp(oprinfo[i].oprrsortop, "0") == 0))
			appendPQExpBuffer(sort2, ",\n\tSORT2 = %s ",
B
Bruce Momjian 已提交
2738
			 findOprByOid(oprinfo, numOperators, oprinfo[i].oprrsortop));
2739

2740
		becomeUser(fout, oprinfo[i].usename);
2741

2742
		if (dropSchema)
B
Bruce Momjian 已提交
2743
		{
B
Hi, all  
Bruce Momjian 已提交
2744
			resetPQExpBuffer(q);
2745
			appendPQExpBuffer(q, "DROP OPERATOR %s (%s", oprinfo[i].oprname,
2746 2747
				fmtId(findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft),
					  false));
2748
			appendPQExpBuffer(q, ", %s);\n",
2749 2750
			   fmtId(findTypeByOid(tinfo, numTypes, oprinfo[i].oprright),
					 false));
B
Hi, all  
Bruce Momjian 已提交
2751
			fputs(q->data, fout);
B
Bruce Momjian 已提交
2752
		}
2753

B
Hi, all  
Bruce Momjian 已提交
2754 2755
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
2756 2757 2758 2759 2760 2761 2762 2763 2764
						  "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 已提交
2765
		  (strcmp(oprinfo[i].oprcanhash, "t") == 0) ? ",\n\tHASHES" : "",
2766 2767 2768
						  join->data,
						  sort1->data,
						  sort2->data);
2769

B
Hi, all  
Bruce Momjian 已提交
2770
		fputs(q->data, fout);
2771
	}
2772 2773 2774 2775
}

/*
 * dumpAggs
2776
 *	  writes out to fout the queries to create all the user-defined aggregates
2777 2778 2779
 *
 */
void
2780 2781
dumpAggs(FILE *fout, AggInfo *agginfo, int numAggs,
		 TypeInfo *tinfo, int numTypes)
2782
{
B
Bruce Momjian 已提交
2783
	int			i;
2784 2785 2786 2787 2788
	PQExpBuffer q = createPQExpBuffer();
	PQExpBuffer sfunc1 = createPQExpBuffer();
	PQExpBuffer sfunc2 = createPQExpBuffer();
	PQExpBuffer basetype = createPQExpBuffer();
	PQExpBuffer finalfunc = createPQExpBuffer();
2789
	char		comma1[2],
B
Bruce Momjian 已提交
2790
				comma2[2];
2791 2792 2793

	for (i = 0; i < numAggs; i++)
	{
B
Hi, all  
Bruce Momjian 已提交
2794 2795 2796 2797 2798 2799

		resetPQExpBuffer(sfunc1);
		resetPQExpBuffer(sfunc2);
		resetPQExpBuffer(basetype);
		resetPQExpBuffer(finalfunc);

2800 2801 2802
		/* skip all the builtin oids */
		if (atoi(agginfo[i].oid) < g_last_builtin_oid)
			continue;
2803

B
Hi, all  
Bruce Momjian 已提交
2804
		appendPQExpBuffer(basetype,
2805 2806
						  "BASETYPE = %s, ",
						  fmtId(findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype), false));
2807

B
Hi, all  
Bruce Momjian 已提交
2808
		if (!(strcmp(agginfo[i].aggtransfn1, "-") == 0))
2809
		{
B
Hi, all  
Bruce Momjian 已提交
2810
			appendPQExpBuffer(sfunc1,
2811 2812 2813
							  "SFUNC1 = %s, STYPE1 = %s",
							  agginfo[i].aggtransfn1,
							  fmtId(findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype1), false));
2814
			if (agginfo[i].agginitval1)
B
Hi, all  
Bruce Momjian 已提交
2815
				appendPQExpBuffer(sfunc1, ", INITCOND1 = '%s'",
2816
								  agginfo[i].agginitval1);
2817 2818 2819

		}

B
Hi, all  
Bruce Momjian 已提交
2820
		if (!(strcmp(agginfo[i].aggtransfn2, "-") == 0))
2821
		{
B
Hi, all  
Bruce Momjian 已提交
2822
			appendPQExpBuffer(sfunc2,
2823 2824 2825
							  "SFUNC2 = %s, STYPE2 = %s",
							  agginfo[i].aggtransfn2,
							  fmtId(findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype2), false));
2826
			if (agginfo[i].agginitval2)
B
Hi, all  
Bruce Momjian 已提交
2827
				appendPQExpBuffer(sfunc2, ", INITCOND2 = '%s'",
2828
								  agginfo[i].agginitval2);
2829 2830
		}

B
Hi, all  
Bruce Momjian 已提交
2831 2832 2833
		if (!(strcmp(agginfo[i].aggfinalfn, "-") == 0))
			appendPQExpBuffer(finalfunc, "FINALFUNC = %s", agginfo[i].aggfinalfn);
		if (sfunc1->data[0] != '\0' && sfunc2->data[0] != '\0')
2834 2835 2836 2837 2838 2839 2840
		{
			comma1[0] = ',';
			comma1[1] = '\0';
		}
		else
			comma1[0] = '\0';

B
Hi, all  
Bruce Momjian 已提交
2841
		if (finalfunc->data[0] != '\0' && (sfunc1->data[0] != '\0' || sfunc2->data[0] != '\0'))
2842 2843 2844 2845 2846 2847 2848
		{
			comma2[0] = ',';
			comma2[1] = '\0';
		}
		else
			comma2[0] = '\0';

2849
		becomeUser(fout, agginfo[i].usename);
2850

2851
		if (dropSchema)
B
Bruce Momjian 已提交
2852
		{
B
Hi, all  
Bruce Momjian 已提交
2853 2854
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "DROP AGGREGATE %s %s;\n", agginfo[i].aggname,
2855
							  fmtId(findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype), false));
B
Hi, all  
Bruce Momjian 已提交
2856
			fputs(q->data, fout);
B
Bruce Momjian 已提交
2857
		}
2858

B
Hi, all  
Bruce Momjian 已提交
2859 2860
		resetPQExpBuffer(q);
		appendPQExpBuffer(q, "CREATE AGGREGATE %s ( %s %s%s %s%s %s );\n",
2861 2862 2863 2864 2865 2866 2867
						  agginfo[i].aggname,
						  basetype->data,
						  sfunc1->data,
						  comma1,
						  sfunc2->data,
						  comma2,
						  finalfunc->data);
2868

B
Hi, all  
Bruce Momjian 已提交
2869
		fputs(q->data, fout);
B
Bruce,  
Bruce Momjian 已提交
2870 2871 2872 2873

		/*** Dump Aggregate Comments ***/

		resetPQExpBuffer(q);
2874
		appendPQExpBuffer(q, "AGGREGATE %s %s", agginfo[i].aggname,
B
Bruce,  
Bruce Momjian 已提交
2875 2876 2877
						  fmtId(findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype), false));
		dumpComment(fout, q->data, agginfo[i].oid);

2878
	}
2879 2880
}

2881 2882 2883 2884 2885
/*
 * These are some support functions to fix the acl problem of pg_dump
 *
 * Matthew C. Aycock 12/02/97
 */
2886 2887 2888

/* Append a keyword to a keyword list, inserting comma if needed.
 * Caller must make aclbuf big enough for all possible keywords.
2889
 */
2890
static void
B
Bruce Momjian 已提交
2891
AddAcl(char *aclbuf, const char *keyword)
2892
{
2893 2894 2895
	if (*aclbuf)
		strcat(aclbuf, ",");
	strcat(aclbuf, keyword);
2896
}
2897

2898
/*
2899
 * This will take a string of 'arwR' and return a malloced,
2900 2901
 * comma delimited string of SELECT,INSERT,UPDATE,DELETE,RULE
 */
V
Vadim B. Mikheev 已提交
2902
static char *
2903
GetPrivileges(const char *s)
2904
{
2905
	char		aclbuf[100];
2906

2907
	aclbuf[0] = '\0';
2908

B
Bruce Momjian 已提交
2909
	if (strchr(s, 'a'))
2910
		AddAcl(aclbuf, "INSERT");
2911

B
Bruce Momjian 已提交
2912
	if (strchr(s, 'w'))
2913
		AddAcl(aclbuf, "UPDATE,DELETE");
B
Bruce Momjian 已提交
2914 2915

	if (strchr(s, 'r'))
2916
		AddAcl(aclbuf, "SELECT");
2917

2918
	if (strchr(s, 'R'))
2919
		AddAcl(aclbuf, "RULE");
2920

2921 2922 2923
	/* Special-case when they're all there */
	if (strcmp(aclbuf, "INSERT,UPDATE,DELETE,SELECT,RULE") == 0)
		return strdup("ALL");
2924

2925
	return strdup(aclbuf);
2926
}
2927

B
Bruce Momjian 已提交
2928 2929
/*
 * dumpACL:
2930 2931
 *	  Write out grant/revoke information
 *	  Called for sequences and tables
B
Bruce Momjian 已提交
2932 2933
 */

2934
static void
B
Bruce Momjian 已提交
2935 2936
dumpACL(FILE *fout, TableInfo tbinfo)
{
2937 2938
	const char *acls = tbinfo.relacl;
	char	   *aclbuf,
B
Bruce Momjian 已提交
2939 2940 2941
			   *tok,
			   *eqpos,
			   *priv;
2942 2943 2944 2945

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

B
Bruce Momjian 已提交
2946 2947 2948
	/*
	 * Revoke Default permissions for PUBLIC. Is this actually necessary,
	 * or is it just a waste of time?
2949 2950 2951
	 */
	fprintf(fout,
			"REVOKE ALL on %s from PUBLIC;\n",
2952
			fmtId(tbinfo.relname, force_quotes));
2953 2954 2955

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

2957 2958
	/* Scan comma-separated ACL items */
	for (tok = strtok(aclbuf, ","); tok != NULL; tok = strtok(NULL, ","))
2959
	{
2960

B
Bruce Momjian 已提交
2961 2962 2963
		/*
		 * Token may start with '{' and/or '"'.  Actually only the start
		 * of the string should have '{', but we don't verify that.
2964 2965 2966 2967 2968 2969 2970 2971
		 */
		if (*tok == '{')
			tok++;
		if (*tok == '"')
			tok++;

		/* User name is string up to = in tok */
		eqpos = strchr(tok, '=');
B
Bruce Momjian 已提交
2972
		if (!eqpos)
B
Bruce Momjian 已提交
2973
		{
2974
			fprintf(stderr, "Could not parse ACL list for '%s'...Exiting!\n",
B
Bruce Momjian 已提交
2975 2976 2977 2978
					tbinfo.relname);
			exit_nicely(g_conn);
		}

B
Bruce Momjian 已提交
2979 2980 2981 2982
		/*
		 * Parse the privileges (right-hand side).	Skip if there are
		 * none.
		 */
2983 2984
		priv = GetPrivileges(eqpos + 1);
		if (*priv)
2985
		{
2986 2987
			fprintf(fout,
					"GRANT %s on %s to ",
2988
					priv, fmtId(tbinfo.relname, force_quotes));
B
Bruce Momjian 已提交
2989 2990 2991

			/*
			 * Note: fmtId() can only be called once per printf, so don't
2992 2993 2994 2995 2996
			 * try to merge printing of username into the above printf.
			 */
			if (eqpos == tok)
			{
				/* Empty left-hand side means "PUBLIC" */
2997
				fprintf(fout, "PUBLIC;\n");
2998
			}
2999
			else
3000 3001
			{
				*eqpos = '\0';	/* it's ok to clobber aclbuf */
3002 3003 3004 3005 3006
				if (strncmp(tok, "group ", strlen("group ")) == 0)
					fprintf(fout, "GROUP %s;\n",
							fmtId(tok + strlen("group "), force_quotes));
				else
					fprintf(fout, "%s;\n", fmtId(tok, force_quotes));
3007
			}
3008
		}
3009
		free(priv);
B
Bruce Momjian 已提交
3010
	}
3011 3012

	free(aclbuf);
B
Bruce Momjian 已提交
3013 3014
}

3015

3016 3017
/*
 * dumpTables:
3018
 *	  write out to fout all the user-define tables
3019
 */
3020

3021
void
3022 3023 3024
dumpTables(FILE *fout, TableInfo *tblinfo, int numTables,
		   InhInfo *inhinfo, int numInherits,
		   TypeInfo *tinfo, int numTypes, const char *tablename,
3025
		   const bool aclsSkip)
3026
{
3027 3028
	int			i,
				j,
B
Bruce Momjian 已提交
3029
				k;
3030 3031
	PQExpBuffer q = createPQExpBuffer();
	char	   *serialSeq = NULL;		/* implicit sequence name created
B
Bruce Momjian 已提交
3032 3033 3034 3035
										 * by SERIAL datatype */
	const char *serialSeqSuffix = "_id_seq";	/* suffix for implicit
												 * SERIAL sequences */
	char	  **parentRels;		/* list of names of parent relations */
3036
	int			numParents;
B
Bruce Momjian 已提交
3037 3038
	int			actual_atts;	/* number of attrs in this CREATE statment */
	int32		tmp_typmod;
3039 3040 3041
	int			precision;
	int			scale;

3042

V
Vadim B. Mikheev 已提交
3043
	/* First - dump SEQUENCEs */
B
Bruce Momjian 已提交
3044
	if (tablename)
B
Bruce Momjian 已提交
3045 3046 3047 3048 3049
	{
		serialSeq = malloc(strlen(tablename) + strlen(serialSeqSuffix) + 1);
		strcpy(serialSeq, tablename);
		strcat(serialSeq, serialSeqSuffix);
	}
V
Vadim B. Mikheev 已提交
3050 3051 3052 3053
	for (i = 0; i < numTables; i++)
	{
		if (!(tblinfo[i].sequence))
			continue;
B
Bruce Momjian 已提交
3054
		if (!tablename || (!strcmp(tblinfo[i].relname, tablename))
B
Bruce Momjian 已提交
3055
			|| (serialSeq && !strcmp(tblinfo[i].relname, serialSeq)))
V
Vadim B. Mikheev 已提交
3056
		{
3057
			becomeUser(fout, tblinfo[i].usename);
V
Vadim B. Mikheev 已提交
3058
			dumpSequence(fout, tblinfo[i]);
3059
			if (!aclsSkip)
B
Bruce Momjian 已提交
3060
				dumpACL(fout, tblinfo[i]);
V
Vadim B. Mikheev 已提交
3061 3062
		}
	}
B
Bruce Momjian 已提交
3063
	if (tablename)
B
Bruce Momjian 已提交
3064
		free(serialSeq);
3065

3066 3067
	for (i = 0; i < numTables; i++)
	{
3068
		if (tblinfo[i].sequence)/* already dumped */
V
Vadim B. Mikheev 已提交
3069
			continue;
3070 3071 3072 3073

		if (!tablename || (!strcmp(tblinfo[i].relname, tablename)))
		{

3074
			/* Skip VIEW relations */
B
Bruce Momjian 已提交
3075

3076
			/*
B
Bruce Momjian 已提交
3077 3078
			 * if (isViewRule(tblinfo[i].relname)) continue;
			 */
3079

3080 3081
			parentRels = tblinfo[i].parentRels;
			numParents = tblinfo[i].numParents;
3082

3083
			becomeUser(fout, tblinfo[i].usename);
3084

3085
			if (dropSchema)
B
Bruce Momjian 已提交
3086
			{
B
Hi, all  
Bruce Momjian 已提交
3087 3088 3089
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "DROP TABLE %s;\n", fmtId(tblinfo[i].relname, force_quotes));
				fputs(q->data, fout);
B
Bruce Momjian 已提交
3090
			}
3091

B
Hi, all  
Bruce Momjian 已提交
3092 3093
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "CREATE TABLE %s (\n\t", fmtId(tblinfo[i].relname, force_quotes));
3094 3095 3096 3097
			actual_atts = 0;
			for (j = 0; j < tblinfo[i].numatts; j++)
			{
				if (tblinfo[i].inhAttrs[j] == 0)
B
Hi all  
Bruce Momjian 已提交
3098
				{
3099
					if (actual_atts > 0)
B
Hi, all  
Bruce Momjian 已提交
3100 3101
						appendPQExpBuffer(q, ",\n\t");
					appendPQExpBuffer(q, "%s ",
3102
							fmtId(tblinfo[i].attnames[j], force_quotes));
3103 3104 3105

					/* Show lengths on bpchar and varchar */
					if (!strcmp(tblinfo[i].typnames[j], "bpchar"))
3106
					{
B
Bruce Momjian 已提交
3107 3108
						int			len = (tblinfo[i].atttypmod[j] - VARHDRSZ);

B
Hi, all  
Bruce Momjian 已提交
3109
						appendPQExpBuffer(q, "character");
3110
						if (len > 1)
B
Hi, all  
Bruce Momjian 已提交
3111
							appendPQExpBuffer(q, "(%d)",
3112
									 tblinfo[i].atttypmod[j] - VARHDRSZ);
3113
					}
3114 3115
					else if (!strcmp(tblinfo[i].typnames[j], "varchar"))
					{
B
Hi, all  
Bruce Momjian 已提交
3116
						appendPQExpBuffer(q, "character varying");
3117 3118
						if (tblinfo[i].atttypmod[j] != -1)
						{
B
Hi, all  
Bruce Momjian 已提交
3119
							appendPQExpBuffer(q, "(%d)",
3120
									 tblinfo[i].atttypmod[j] - VARHDRSZ);
3121
						}
3122
					}
3123 3124
					else if (!strcmp(tblinfo[i].typnames[j], "numeric"))
					{
B
Hi, all  
Bruce Momjian 已提交
3125
						appendPQExpBuffer(q, "numeric");
3126 3127 3128 3129
						if (tblinfo[i].atttypmod[j] != -1)
						{
							tmp_typmod = tblinfo[i].atttypmod[j] - VARHDRSZ;
							precision = (tmp_typmod >> 16) & 0xffff;
B
Bruce Momjian 已提交
3130
							scale = tmp_typmod & 0xffff;
B
Hi, all  
Bruce Momjian 已提交
3131
							appendPQExpBuffer(q, "(%d,%d)",
3132
											  precision, scale);
3133 3134
						}
					}
B
Bruce Momjian 已提交
3135 3136 3137 3138 3139

					/*
					 * char is an internal single-byte data type; Let's
					 * make sure we force it through with quotes. - thomas
					 * 1998-12-13
3140 3141 3142
					 */
					else if (!strcmp(tblinfo[i].typnames[j], "char"))
					{
B
Hi, all  
Bruce Momjian 已提交
3143
						appendPQExpBuffer(q, "%s",
3144
									fmtId(tblinfo[i].typnames[j], true));
3145
					}
3146 3147
					else
					{
B
Hi, all  
Bruce Momjian 已提交
3148
						appendPQExpBuffer(q, "%s",
3149
								   fmtId(tblinfo[i].typnames[j], false));
3150 3151
					}
					if (tblinfo[i].adef_expr[j] != NULL)
B
Hi, all  
Bruce Momjian 已提交
3152
						appendPQExpBuffer(q, " DEFAULT %s",
3153
										  tblinfo[i].adef_expr[j]);
3154
					if (tblinfo[i].notnull[j])
B
Hi, all  
Bruce Momjian 已提交
3155
						appendPQExpBuffer(q, " NOT NULL");
3156
					actual_atts++;
3157
				}
3158
			}
B
Hi all  
Bruce Momjian 已提交
3159

3160 3161 3162
			/* put the CONSTRAINTS inside the table def */
			for (k = 0; k < tblinfo[i].ncheck; k++)
			{
3163
				if (actual_atts + k > 0)
B
Hi, all  
Bruce Momjian 已提交
3164 3165
					appendPQExpBuffer(q, ",\n\t");
				appendPQExpBuffer(q, "%s",
3166
								  tblinfo[i].check_expr[k]);
3167
			}
3168

3169 3170 3171 3172 3173 3174 3175
			/* PRIMARY KEY */
			if (tblinfo[i].primary_key)
			{
				if (actual_atts + tblinfo[i].ncheck > 0)
					appendPQExpBuffer(q, ",\n\t");
				appendPQExpBuffer(q, "PRIMARY KEY (%s)", tblinfo[i].primary_key);
			}
B
Bruce Momjian 已提交
3176

3177
			appendPQExpBuffer(q, "\n)");
3178 3179 3180

			if (numParents > 0)
			{
B
Hi, all  
Bruce Momjian 已提交
3181
				appendPQExpBuffer(q, "\ninherits (");
3182
				for (k = 0; k < numParents; k++)
B
Hi all  
Bruce Momjian 已提交
3183
				{
B
Hi, all  
Bruce Momjian 已提交
3184
					appendPQExpBuffer(q, "%s%s",
3185 3186
									  (k > 0) ? ", " : "",
									  fmtId(parentRels[k], force_quotes));
B
Hi all  
Bruce Momjian 已提交
3187
				}
B
Hi, all  
Bruce Momjian 已提交
3188
				appendPQExpBuffer(q, ")");
3189
			}
B
Hi all  
Bruce Momjian 已提交
3190

B
Hi, all  
Bruce Momjian 已提交
3191 3192
			appendPQExpBuffer(q, ";\n");
			fputs(q->data, fout);
3193
			if (!aclsSkip)
B
Bruce Momjian 已提交
3194
				dumpACL(fout, tblinfo[i]);
3195

3196
			/* Dump Field Comments */
B
Bruce,  
Bruce Momjian 已提交
3197

3198 3199
			for (j = 0; j < tblinfo[i].numatts; j++)
			{
B
Bruce,  
Bruce Momjian 已提交
3200 3201 3202 3203 3204 3205
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "COLUMN %s", fmtId(tblinfo[i].relname, force_quotes));
				appendPQExpBuffer(q, ".");
				appendPQExpBuffer(q, "%s", fmtId(tblinfo[i].attnames[j], force_quotes));
				dumpComment(fout, q->data, tblinfo[i].attoids[j]);
			}
3206

B
Bruce,  
Bruce Momjian 已提交
3207
			/* Dump Table Comments */
3208

B
Bruce,  
Bruce Momjian 已提交
3209 3210 3211
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "TABLE %s", fmtId(tblinfo[i].relname, force_quotes));
			dumpComment(fout, q->data, tblinfo[i].oid);
3212

3213 3214
		}
	}
3215 3216 3217 3218
}

/*
 * dumpIndices:
3219
 *	  write out to fout all the user-define indices
3220
 */
3221
void
3222 3223
dumpIndices(FILE *fout, IndInfo *indinfo, int numIndices,
			TableInfo *tblinfo, int numTables, const char *tablename)
3224
{
3225 3226 3227
	int			i,
				k;
	int			tableInd;
3228
	PQExpBuffer attlist = createPQExpBuffer();
B
Bruce Momjian 已提交
3229 3230 3231
	char	   *classname[INDEX_MAX_KEYS];
	char	   *funcname;		/* the name of the function to comput the
								 * index key from */
3232 3233 3234
	int			indkey,
				indclass;
	int			nclass;
3235

3236
	PQExpBuffer q = createPQExpBuffer(),
B
Hi, all  
Bruce Momjian 已提交
3237 3238
				id1 = createPQExpBuffer(),
				id2 = createPQExpBuffer();
3239
	PGresult   *res;
3240 3241 3242 3243

	for (i = 0; i < numIndices; i++)
	{
		tableInd = findTableByName(tblinfo, numTables,
3244 3245 3246 3247 3248
								   indinfo[i].indrelname);
		if (tableInd < 0)
		{
			fprintf(stderr, "failed sanity check, table %s was not found\n",
					indinfo[i].indrelname);
3249
			exit(1);
3250
		}
3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262

		if (strcmp(indinfo[i].indproc, "0") == 0)
			funcname = NULL;
		else
		{

			/*
			 * 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 已提交
3263 3264
			resetPQExpBuffer(q);
			appendPQExpBuffer(q,
3265 3266 3267
							  "SELECT proname from pg_proc "
							  "where pg_proc.oid = '%s'::oid",
							  indinfo[i].indproc);
B
Hi, all  
Bruce Momjian 已提交
3268
			res = PQexec(g_conn, q->data);
3269 3270
			if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
			{
B
Bruce Momjian 已提交
3271
				fprintf(stderr, "dumpIndices(): SELECT (funcname) failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284
				exit_nicely(g_conn);
			}
			funcname = strdup(PQgetvalue(res, 0,
										 PQfnumber(res, "proname")));
			PQclear(res);
		}

		/* convert opclass oid(s) into names */
		for (nclass = 0; nclass < INDEX_MAX_KEYS; nclass++)
		{
			indclass = atoi(indinfo[i].indclass[nclass]);
			if (indclass == 0)
				break;
B
Hi, all  
Bruce Momjian 已提交
3285 3286
			resetPQExpBuffer(q);
			appendPQExpBuffer(q,
3287 3288 3289
							  "SELECT opcname from pg_opclass "
							  "where pg_opclass.oid = '%u'::oid",
							  indclass);
B
Hi, all  
Bruce Momjian 已提交
3290
			res = PQexec(g_conn, q->data);
3291 3292
			if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
			{
B
Bruce Momjian 已提交
3293
				fprintf(stderr, "dumpIndices(): SELECT (classname) failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3294 3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308
				exit_nicely(g_conn);
			}
			classname[nclass] = strdup(PQgetvalue(res, 0,
											 PQfnumber(res, "opcname")));
			PQclear(res);
		}

		if (funcname && nclass != 1)
		{
			fprintf(stderr, "dumpIndices(): Must be exactly one OpClass "
					"for functional index %s\n", indinfo[i].indexrelname);
			exit_nicely(g_conn);
		}

		/* convert attribute numbers into attribute list */
B
Hi, all  
Bruce Momjian 已提交
3309 3310
		resetPQExpBuffer(attlist);
		for (k = 0; k < INDEX_MAX_KEYS; k++)
3311
		{
3312
			char	   *attname;
3313 3314

			indkey = atoi(indinfo[i].indkey[k]);
B
Bruce Momjian 已提交
3315
			if (indkey == InvalidAttrNumber)
3316 3317 3318 3319 3320 3321 3322
				break;
			indkey--;
			if (indkey == ObjectIdAttributeNumber - 1)
				attname = "oid";
			else
				attname = tblinfo[tableInd].attnames[indkey];
			if (funcname)
B
Hi, all  
Bruce Momjian 已提交
3323
				appendPQExpBuffer(attlist, "%s%s",
B
Bruce Momjian 已提交
3324
					 (k == 0) ? "" : ", ", fmtId(attname, force_quotes));
3325 3326 3327 3328 3329
			else
			{
				if (k >= nclass)
				{
					fprintf(stderr, "dumpIndices(): OpClass not found for "
3330
							"attribute '%s' of index '%s'\n",
3331 3332 3333
							attname, indinfo[i].indexrelname);
					exit_nicely(g_conn);
				}
B
Hi, all  
Bruce Momjian 已提交
3334 3335 3336 3337 3338
				resetPQExpBuffer(id1);
				resetPQExpBuffer(id2);
				appendPQExpBuffer(id1, fmtId(attname, force_quotes));
				appendPQExpBuffer(id2, fmtId(classname[k], force_quotes));
				appendPQExpBuffer(attlist, "%s%s %s",
3339
							 (k == 0) ? "" : ", ", id1->data, id2->data);
3340 3341 3342 3343 3344 3345
				free(classname[k]);
			}
		}

		if (!tablename || (!strcmp(indinfo[i].indrelname, tablename)))
		{
B
Bruce Momjian 已提交
3346 3347 3348 3349 3350

			/*
			 * 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.
3351 3352
			 */
			becomeUser(fout, tblinfo[tableInd].usename);
3353

B
Hi, all  
Bruce Momjian 已提交
3354 3355 3356 3357
			resetPQExpBuffer(id1);
			resetPQExpBuffer(id2);
			appendPQExpBuffer(id1, fmtId(indinfo[i].indexrelname, force_quotes));
			appendPQExpBuffer(id2, fmtId(indinfo[i].indrelname, force_quotes));
3358

3359
			if (dropSchema)
B
Bruce Momjian 已提交
3360
			{
B
Hi, all  
Bruce Momjian 已提交
3361 3362 3363
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "DROP INDEX %s;\n", id1->data);
				fputs(q->data, fout);
B
Bruce Momjian 已提交
3364
			}
3365

3366
			fprintf(fout, "CREATE %s INDEX %s on %s using %s (",
3367
			  (strcmp(indinfo[i].indisunique, "t") == 0) ? "UNIQUE" : "",
B
Hi, all  
Bruce Momjian 已提交
3368 3369
					id1->data,
					id2->data,
3370 3371 3372
					indinfo[i].indamname);
			if (funcname)
			{
3373
				/* need 2 printf's here cuz fmtId has static return area */
3374
				fprintf(fout, " %s", fmtId(funcname, false));
B
Hi, all  
Bruce Momjian 已提交
3375
				fprintf(fout, " (%s) %s );\n", attlist->data, fmtId(classname[0], force_quotes));
3376 3377 3378 3379
				free(funcname);
				free(classname[0]);
			}
			else
B
Hi, all  
Bruce Momjian 已提交
3380
				fprintf(fout, " %s );\n", attlist->data);
B
Bruce,  
Bruce Momjian 已提交
3381 3382 3383 3384 3385 3386

			/* Dump Index Comments */

			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "INDEX %s", id1->data);
			dumpComment(fout, q->data, indinfo[i].indoid);
3387

3388 3389
		}
	}
3390 3391

}
3392

3393
/*
B
Bruce Momjian 已提交
3394
 * dumpTuples
3395 3396 3397 3398 3399
 *	  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
3400
 *	  in PostgreSQL.
3401 3402 3403 3404
 *
 * the attrmap passed in tells how to map the attributes copied in to the
 * attributes copied out
 */
3405
#ifdef NOT_USED
3406
void
3407
dumpTuples(PGresult *res, FILE *fout, int *attrmap)
3408
{
3409 3410 3411 3412
	int			j,
				k;
	int			m,
				n;
B
Bruce Momjian 已提交
3413
	char	  **outVals = NULL; /* values to copy out */
3414 3415 3416 3417 3418 3419 3420 3421 3422 3423 3424 3425 3426 3427 3428 3429 3430 3431 3432

	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++)
			{
3433
				char	   *pval = outVals[k];
3434 3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446 3447 3448 3449 3450 3451 3452 3453

				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);
	}
3454
}
3455

3456
#endif
3457

3458 3459 3460 3461
/*
 * setMaxOid -
 * find the maximum oid and generate a COPY statement to set it
*/
3462

3463
static void
3464
setMaxOid(FILE *fout)
3465
{
B
Bruce Momjian 已提交
3466
	PGresult   *res;
3467
	Oid			max_oid;
3468 3469 3470 3471 3472

	res = PQexec(g_conn, "CREATE TABLE pgdump_oid (dummy int4)");
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
B
Bruce Momjian 已提交
3473
		fprintf(stderr, "Can not create pgdump_oid table.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3474 3475 3476 3477 3478 3479 3480
		exit_nicely(g_conn);
	}
	PQclear(res);
	res = PQexec(g_conn, "INSERT INTO pgdump_oid VALUES (0)");
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
B
Bruce Momjian 已提交
3481
		fprintf(stderr, "Can not insert into pgdump_oid table.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3482 3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494
		exit_nicely(g_conn);
	}
	max_oid = atol(PQoidStatus(res));
	if (max_oid == 0)
	{
		fprintf(stderr, "Invalid max id in setMaxOid\n");
		exit_nicely(g_conn);
	}
	PQclear(res);
	res = PQexec(g_conn, "DROP TABLE pgdump_oid;");
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
B
Bruce Momjian 已提交
3495
		fprintf(stderr, "Can not drop pgdump_oid table.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3496 3497 3498 3499
		exit_nicely(g_conn);
	}
	PQclear(res);
	if (g_verbose)
3500
		fprintf(stderr, "%s maximum system oid is %u %s\n",
3501 3502 3503 3504 3505 3506 3507
				g_comment_start, max_oid, g_comment_end);
	fprintf(fout, "CREATE TABLE pgdump_oid (dummy int4);\n");
	fprintf(fout, "COPY pgdump_oid WITH OIDS FROM stdin;\n");
	fprintf(fout, "%-d\t0\n", max_oid);
	fprintf(fout, "\\.\n");
	fprintf(fout, "DROP TABLE pgdump_oid;\n");
}
3508 3509 3510

/*
 * findLastBuiltInOid -
3511
 * find the last built in oid
3512
 * we do this by looking up the oid of 'template1' in pg_database,
3513
 * this is probably not foolproof but comes close
3514 3515
*/

3516
static int
3517
findLastBuiltinOid(void)
3518
{
B
Bruce Momjian 已提交
3519
	PGresult   *res;
3520 3521
	int			ntups;
	int			last_oid;
3522 3523 3524 3525 3526 3527

	res = PQexec(g_conn,
			  "SELECT oid from pg_database where datname = 'template1'");
	if (res == NULL ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
3528 3529
		fprintf(stderr, "pg_dump error in finding the template1 database.");
		fprintf(stderr, "Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3530 3531 3532
		exit_nicely(g_conn);
	}
	ntups = PQntuples(res);
3533
	if (ntups < 1)
3534
	{
3535 3536
		fprintf(stderr, "pg_dump: couldn't find the template1 database.\n");
		fprintf(stderr, "There is no 'template1' entry in the 'pg_database' table.\n");
3537 3538 3539 3540
		exit_nicely(g_conn);
	}
	if (ntups > 1)
	{
3541 3542
		fprintf(stderr, "pg_dump: found more than one template1 database.\n");
		fprintf(stderr, "There is more than one 'template1' entry in the 'pg_database' table\n");
3543 3544 3545 3546 3547
		exit_nicely(g_conn);
	}
	last_oid = atoi(PQgetvalue(res, 0, PQfnumber(res, "oid")));
	PQclear(res);
	return last_oid;
3548 3549 3550 3551 3552
}


/*
 * checkForQuote:
3553
 *	  checks a string for quote characters and quotes them
3554
 */
3555
static char *
3556
checkForQuote(const char *s)
3557
{
B
Bruce Momjian 已提交
3558
	char	   *r;
3559
	char		c;
B
Bruce Momjian 已提交
3560
	char	   *result;
3561

3562
	int			j = 0;
3563

3564
	r = malloc(strlen(s) * 3 + 1);		/* definitely long enough */
3565

3566 3567
	while ((c = *s) != '\0')
	{
3568

3569 3570 3571 3572 3573 3574 3575 3576 3577 3578 3579
		if (c == '\'')
		{
			r[j++] = '\'';		/* quote the single quotes */
		}
		r[j++] = c;
		s++;
	}
	r[j] = '\0';

	result = strdup(r);
	free(r);
3580

3581
	return result;
3582 3583

}
3584 3585


3586
static void
3587
dumpSequence(FILE *fout, TableInfo tbinfo)
3588
{
3589
	PGresult   *res;
B
Bruce Momjian 已提交
3590
	int4		last,
3591 3592 3593 3594
				incby,
				maxv,
				minv,
				cache;
B
Bruce Momjian 已提交
3595
	char		cycled,
B
Bruce Momjian 已提交
3596
				called;
3597 3598
	const char *t;
	PQExpBuffer query = createPQExpBuffer();
3599

B
Hi, all  
Bruce Momjian 已提交
3600
	appendPQExpBuffer(query,
3601
			"SELECT sequence_name, last_value, increment_by, max_value, "
3602 3603
				  "min_value, cache_value, is_cycled, is_called from %s",
					  fmtId(tbinfo.relname, force_quotes));
3604

B
Hi, all  
Bruce Momjian 已提交
3605
	res = PQexec(g_conn, query->data);
3606 3607
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
3608
		fprintf(stderr, "dumpSequence(%s): SELECT failed.  Explanation from backend: '%s'.\n", tbinfo.relname, PQerrorMessage(g_conn));
3609 3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622 3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638 3639
		exit_nicely(g_conn);
	}

	if (PQntuples(res) != 1)
	{
		fprintf(stderr, "dumpSequence(%s): %d (!= 1) tuples returned by SELECT\n",
				tbinfo.relname, PQntuples(res));
		exit_nicely(g_conn);
	}

	if (strcmp(PQgetvalue(res, 0, 0), tbinfo.relname) != 0)
	{
		fprintf(stderr, "dumpSequence(%s): different sequence name "
				"returned by SELECT: %s\n",
				tbinfo.relname, PQgetvalue(res, 0, 0));
		exit_nicely(g_conn);
	}


	last = atoi(PQgetvalue(res, 0, 1));
	incby = atoi(PQgetvalue(res, 0, 2));
	maxv = atoi(PQgetvalue(res, 0, 3));
	minv = atoi(PQgetvalue(res, 0, 4));
	cache = atoi(PQgetvalue(res, 0, 5));
	t = PQgetvalue(res, 0, 6);
	cycled = *t;
	t = PQgetvalue(res, 0, 7);
	called = *t;

	PQclear(res);

3640
	if (dropSchema)
B
Bruce Momjian 已提交
3641
	{
B
Hi, all  
Bruce Momjian 已提交
3642 3643 3644
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "DROP SEQUENCE %s;\n", fmtId(tbinfo.relname, force_quotes));
		fputs(query->data, fout);
B
Bruce Momjian 已提交
3645
	}
3646

B
Hi, all  
Bruce Momjian 已提交
3647 3648
	resetPQExpBuffer(query);
	appendPQExpBuffer(query,
3649 3650
				  "CREATE SEQUENCE %s start %d increment %d maxvalue %d "
					  "minvalue %d  cache %d %s;\n",
B
Bruce Momjian 已提交
3651
	 fmtId(tbinfo.relname, force_quotes), last, incby, maxv, minv, cache,
3652
					  (cycled == 't') ? "cycle" : "");
3653

B
Hi, all  
Bruce Momjian 已提交
3654
	fputs(query->data, fout);
3655

B
Bruce,  
Bruce Momjian 已提交
3656 3657 3658 3659 3660 3661
	/* Dump Sequence Comments */

	resetPQExpBuffer(query);
	appendPQExpBuffer(query, "SEQUENCE %s", fmtId(tbinfo.relname, force_quotes));
	dumpComment(fout, query->data, tbinfo.oid);

3662 3663 3664
	if (called == 'f')
		return;					/* nothing to do more */

B
Hi, all  
Bruce Momjian 已提交
3665 3666 3667
	resetPQExpBuffer(query);
	appendPQExpBuffer(query, "SELECT nextval ('%s');\n", fmtId(tbinfo.relname, force_quotes));
	fputs(query->data, fout);
3668 3669

}
V
Vadim B. Mikheev 已提交
3670 3671


3672 3673 3674
static void
dumpTriggers(FILE *fout, const char *tablename,
			 TableInfo *tblinfo, int numTables)
V
Vadim B. Mikheev 已提交
3675
{
3676 3677 3678
	int			i,
				j;

V
Vadim B. Mikheev 已提交
3679 3680 3681
	if (g_verbose)
		fprintf(stderr, "%s dumping out triggers %s\n",
				g_comment_start, g_comment_end);
3682

V
Vadim B. Mikheev 已提交
3683 3684 3685 3686 3687 3688
	for (i = 0; i < numTables; i++)
	{
		if (tablename && strcmp(tblinfo[i].relname, tablename))
			continue;
		for (j = 0; j < tblinfo[i].ntrig; j++)
		{
3689
			becomeUser(fout, tblinfo[i].usename);
V
Vadim B. Mikheev 已提交
3690
			fputs(tblinfo[i].triggers[j], fout);
B
Bruce,  
Bruce Momjian 已提交
3691
			dumpComment(fout, tblinfo[i].trcomments[j], tblinfo[i].troids[j]);
V
Vadim B. Mikheev 已提交
3692 3693 3694
		}
	}
}
3695 3696


3697 3698
static void
dumpRules(FILE *fout, const char *tablename,
B
Bruce Momjian 已提交
3699
		  TableInfo *tblinfo, int numTables)
3700
{
B
Bruce Momjian 已提交
3701 3702 3703 3704
	PGresult   *res;
	int			nrules;
	int			i,
				t;
3705
	PQExpBuffer query = createPQExpBuffer();
3706

B
Bruce Momjian 已提交
3707
	int			i_definition;
3708 3709
	int			i_oid;
	int			i_rulename;
3710 3711 3712 3713 3714 3715 3716 3717 3718 3719 3720 3721 3722 3723 3724 3725

	if (g_verbose)
		fprintf(stderr, "%s dumping out rules %s\n",
				g_comment_start, g_comment_end);

	/*
	 * For each table we dump
	 */
	for (t = 0; t < numTables; t++)
	{
		if (tablename && strcmp(tblinfo[t].relname, tablename))
			continue;

		/*
		 * Get all rules defined for this table
		 */
B
Hi, all  
Bruce Momjian 已提交
3726 3727
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "SELECT pg_get_ruledef(pg_rewrite.rulename) "
3728 3729 3730 3731 3732
						  "AS definition, pg_rewrite.oid, pg_rewrite.rulename FROM pg_rewrite, pg_class "
						  "WHERE pg_class.relname = '%s' "
						  "AND pg_rewrite.ev_class = pg_class.oid "
						  "ORDER BY pg_rewrite.oid",
						  tblinfo[t].relname);
B
Hi, all  
Bruce Momjian 已提交
3733
		res = PQexec(g_conn, query->data);
3734 3735 3736
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
B
Bruce Momjian 已提交
3737 3738
			fprintf(stderr, "dumpRules(): SELECT failed for table %s.  Explanation from backend: '%s'.\n",
					tblinfo[t].relname, PQerrorMessage(g_conn));
3739 3740 3741 3742 3743
			exit_nicely(g_conn);
		}

		nrules = PQntuples(res);
		i_definition = PQfnumber(res, "definition");
B
Bruce,  
Bruce Momjian 已提交
3744 3745
		i_oid = PQfnumber(res, "oid");
		i_rulename = PQfnumber(res, "rulename");
3746 3747 3748 3749

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

3751 3752 3753
		for (i = 0; i < nrules; i++)
		{

3754 3755
			fprintf(fout, "%s\n", PQgetvalue(res, i, i_definition));

B
Bruce,  
Bruce Momjian 已提交
3756 3757 3758 3759 3760
			/* Dump rule comments */

			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "RULE %s", fmtId(PQgetvalue(res, i, i_rulename), force_quotes));
			dumpComment(fout, query->data, PQgetvalue(res, i, i_oid));
3761

B
Bruce,  
Bruce Momjian 已提交
3762 3763
		}

3764 3765 3766 3767 3768
		PQclear(res);
	}
}


3769 3770 3771 3772 3773 3774
/* Issue a psql \connect command to become the specified user.
 * We want to do this only if we are dumping ACLs,
 * and only if the new username is different from the last one
 * (to avoid the overhead of useless backend launches).
 */

3775 3776
static void
becomeUser(FILE *fout, const char *username)
3777 3778 3779
{
	static const char *lastusername = "";

3780
	if (aclsSkip)
3781 3782 3783 3784 3785 3786 3787 3788 3789
		return;

	if (strcmp(lastusername, username) == 0)
		return;

	fprintf(fout, "\\connect - %s\n", username);

	lastusername = username;
}