pg_dump.c 113.0 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.184 2001/01/04 01:23:47 tgl 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
B
Bruce Momjian 已提交
52 53 54 55 56 57 58 59 60 61 62 63
 *
 * Modifications - 28-Jun-2000 - Philip Warner pjw@rhyme.com.au
 *		 - Used custom IO routines to allow for more
 *		   output formats and simple rearrangement of order.
 *		 - Discouraged operations more appropriate to the 'restore'
 *		   operation. (eg. -c "clear schema" - now always dumps
 *		   commands, but pg_restore can be told not to output them). 
 *		 - Added RI warnings to the 'as insert strings' output mode
 *		 - Added a small number of comments
 *		 - Added a -Z option for compression level on compressed formats
 *		 - Restored '-f' in usage output
 *
64 65 66 67 68
 *
 * Modifications - 17-Jul-2000 - Philip Warner pjw@rhyme.com.au
 *		 - Support for BLOB output.
 *		 - Sort archive by OID, put some items at end (out of OID order)
 *
69 70 71 72 73 74 75 76 77 78 79 80 81
 * Modifications - 28-Jul-2000 - pjw@rhyme.com.au (1.45)
 *
 * 		Added --create, --no-owner, --superuser, --no-reconnect (pg_dump & pg_restore)
 *		Added code to dump 'Create Schema' statement (pg_dump)
 *		Don't bother to disable/enable triggers if we don't have a superuser (pg_restore)
 *		Cleaned up code for reconnecting to database.
 *		Force a reconnect as superuser before enabling/disabling triggers.
 *
 * Modifications - 31-Jul-2000 - pjw@rhyme.com.au (1.46, 1.47)
 *		Added & Removed --throttle (pg_dump)
 *		Fixed minor bug in language dumping code: expbuffres were not being reset.
 *		Fixed version number initialization in _allocAH (pg_backup_archiver.c)
 *
82 83 84 85 86 87 88 89
 * Modifications - 14-Sep-2000 - pjw@rhyme.com.au 
 *		Use symbols for tests on relkind (ie. use RELKIND_VIEW, not 'v')
 * 		Support for relkind = RELKIND_VIEW.
 * 		Fix bug in support for -b option (== --blobs).
 *		Dump views as views (using 'create view').
 *		Remove 'isViewRule' since we check the relkind when getting tables.
 *		Now uses temp table 'pgdump_oid' rather than 'pg_dump_oid' (errors otherwise).
 *
90 91 92 93 94 95 96 97 98
 * Modifications - 02-Oct-2000 - pjw@rhyme.com.au
 *
 *	  - Be more paranoid when getting views: call get_viewdef in separate statement
 *		so we can be more informative in error messages.
 *	  - Support for 'isstrict' procedure attribute. 
 *	  - Disable --blobs and --table since (a) it's a pain to get ONLY the blobs for the 
 *		table with the currently implementation, and (b) it's not clear how to restore
 *		a partial BLOB backup (given the current OID-based BLOB implementation).
 *
99
 *-------------------------------------------------------------------------
100 101
 */

102
#include <unistd.h>				/* for getopt() */
103
#include <ctype.h>
104

105 106
#include "pg_backup.h"

107
#include "postgres.h"
108 109 110 111 112 113 114 115 116

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

#include "access/attnum.h"
117
#include "access/htup.h"
118
#include "catalog/pg_index.h"
119
#include "catalog/pg_language.h"
V
Vadim B. Mikheev 已提交
120
#include "catalog/pg_trigger.h"
121
#include "catalog/pg_type.h"
122

123
#include "libpq-fe.h"
124
#include <libpq/libpq-fs.h>
125
#ifndef HAVE_STRDUP
126 127
#include "strdup.h"
#endif
128 129

#include "pg_dump.h"
B
Bruce Momjian 已提交
130
#include "pg_backup.h"
131

132 133
#define atooid(x)  ((Oid) strtoul((x), NULL, 10))

B
Bruce Momjian 已提交
134 135 136 137
static void dumpComment(Archive *outfile, const char *target, const char *oid);
static void dumpSequence(Archive *fout, TableInfo tbinfo);
static void dumpACL(Archive *fout, TableInfo tbinfo);
static void dumpTriggers(Archive *fout, const char *tablename,
138
			 TableInfo *tblinfo, int numTables);
B
Bruce Momjian 已提交
139
static void dumpRules(Archive *fout, const char *tablename,
B
Bruce Momjian 已提交
140
		  TableInfo *tblinfo, int numTables);
141
static void formatStringLiteral(PQExpBuffer buf, const char *str);
142
static void clearTableInfo(TableInfo *, int);
B
Bruce Momjian 已提交
143
static void dumpOneFunc(Archive *fout, FuncInfo *finfo, int i,
144
			TypeInfo *tinfo, int numTypes);
145
static Oid	findLastBuiltinOid(const char*);
B
Bruce Momjian 已提交
146
static void setMaxOid(Archive *fout);
147

148 149
static void AddAcl(char *aclbuf, const char *keyword);
static char *GetPrivileges(const char *s);
V
Vadim B. Mikheev 已提交
150

151
static int dumpBlobs(Archive *AH, char*, void*);
152
static int dumpDatabase(Archive *AH);
153

B
Bruce Momjian 已提交
154
extern char *optarg;
155
extern int	optind,
B
Bruce Momjian 已提交
156
			opterr;
157 158

/* global decls */
159
bool		g_verbose;			/* User wants verbose narration of our
B
Bruce Momjian 已提交
160
								 * activities. */
161
Oid			g_last_builtin_oid; /* value of the last builtin oid */
B
Bruce Momjian 已提交
162
Archive	   *g_fout;				/* the script file */
B
Bruce Momjian 已提交
163 164 165
PGconn	   *g_conn;				/* the database connection */

bool		force_quotes;		/* User wants to suppress double-quotes */
166 167 168 169
bool		dumpData;			/* dump data using proper insert strings */
bool		attrNames;			/* put attr names into insert strings */
bool		schemaOnly;
bool		dataOnly;
170
bool		aclsSkip;
171

B
Bruce Momjian 已提交
172
char		g_opaque_type[10];	/* name for the opaque type */
173 174

/* placeholders for the delimiters for comments */
175 176
char		g_comment_start[10];
char		g_comment_end[10];
177 178


B
Bruce Momjian 已提交
179 180 181 182 183 184
typedef struct _dumpContext {
	TableInfo	*tblinfo;
	int		tblidx;
	bool		oids;
} DumpContext;

185
static void
186
help(const char *progname)
187
{
P
Peter Eisentraut 已提交
188
	printf("%s dumps a database as a text file.\n\n", progname);
189 190 191
	puts("Usage:");
	printf("  %s [options] dbname\n\n", progname);
	puts("Options:");
B
Hi,  
Bruce Momjian 已提交
192 193

#ifdef HAVE_GETOPT_LONG
194
	puts(
B
Bruce Momjian 已提交
195
		"  -a, --data-only          dump out only the data, not the schema\n"
196
		"  -b, --blobs				dump out blob data\n"
B
Bruce Momjian 已提交
197
	   	"  -c, --clean              clean (drop) schema prior to create\n"
198
		"  -C, --create             output commands to create database\n"
B
Bruce Momjian 已提交
199 200 201 202 203 204 205 206 207
		"  -d, --inserts            dump data as INSERT, rather than COPY, commands\n"
		"  -D, --attribute-inserts  dump data as INSERT commands with attribute names\n"
		"  -f, --file               specify output file name\n"
		"  -F, --format {c|f|p}     output file format (custom, files, plain text)\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"
208
		"  -O, --no-owner           don't output \\connect commands in plain text format\n"
B
Bruce Momjian 已提交
209
		"  -p, --port <port>        server port number\n"
210
		"  -R, --no-reconnect       disable ALL reconnections to the database in plain text format\n"
B
Bruce Momjian 已提交
211
		"  -s, --schema-only        dump out only the schema, no data\n"
212
		"  -S, --superuser <name>   specify the superuser username to use in plain text format\n"
B
Bruce Momjian 已提交
213 214 215 216 217
		"  -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"
		"  -Z, --compress {0-9}     compression level for compressed formats\n"
218
		);
B
Hi,  
Bruce Momjian 已提交
219
#else
220
	puts(
B
Bruce Momjian 已提交
221
		"  -a                       dump out only the data, no schema\n"
222
		"  -b                       dump out blob data\n"
B
Bruce Momjian 已提交
223
		"  -c                       clean (drop) schema prior to create\n"
224
		"  -C                       output commands to create database\n"
B
Bruce Momjian 已提交
225 226 227 228 229 230 231 232 233
		"  -d                       dump data as INSERT, rather than COPY, commands\n"
		"  -D                       dump data as INSERT commands with attribute names\n"
		"  -f                       specify output file name\n"
		"  -F {c|f|p}               output file format (custom, files, plain text)\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"
234
		"  -O                       don't output \\connect commands in plain text format\n"
B
Bruce Momjian 已提交
235
		"  -p <port>                server port number\n"
236
		"  -R                       disable ALL reconnections to the database in plain text format\n"
B
Bruce Momjian 已提交
237
		"  -s                       dump out only the schema, no data\n"
238
		"  -S <name>                specify the superuser username to use in plain text format\n"
B
Bruce Momjian 已提交
239 240 241 242 243
		"  -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"
		"  -Z {0-9}                 compression level for compressed formats\n"
244
		);
B
Hi,  
Bruce Momjian 已提交
245
#endif
246 247
	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>.");
248
}
249

250

251 252 253
static void
version(void)
{
254
	puts("pg_dump (PostgreSQL) " PG_VERSION);
255 256 257
	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.");
258 259
}

260

261
static void
262
exit_nicely(PGconn *conn)
263
{
264 265
	PQfinish(conn);
	exit(1);
266 267 268
}


269 270
#define COPYBUFSIZ		8192

B
Bruce Momjian 已提交
271 272 273 274 275
/*
 *	Dump a table's contents for loading using the COPY command
 *	- this routine is called by the Archiver when it wants the table
 *	  to be dumped.
 */
276

B
Bruce Momjian 已提交
277 278
static int 
dumpClasses_nodumpData(Archive *fout, char* oid, void *dctxv)
279
{
B
Bruce Momjian 已提交
280 281 282
	const DumpContext	*dctx = (DumpContext*)dctxv; 	
	const char 	*classname = dctx->tblinfo[dctx->tblidx].relname;
	const bool	oids = dctx->oids;
283

284 285 286 287 288
	PGresult   *res;
	char		query[255];
	int			ret;
	bool		copydone;
	char		copybuf[COPYBUFSIZ];
289

290 291 292
    if (g_verbose)
        fprintf(stderr, "%s dumping out the contents of table %s\n", g_comment_start, classname);

293
	if (oids == true)
294
	{
295 296 297 298 299 300 301 302
		/* 
		 * archprintf(fout, "COPY %s WITH OIDS FROM stdin;\n",
		 *		fmtId(classname, force_quotes));
		 *
		 *  - Not used as of V1.3 (needs to be in ArchiveEntry call)
		 *
		 */

303
		sprintf(query, "COPY %s WITH OIDS TO stdout;\n",
304
				fmtId(classname, force_quotes));
305 306 307
	}
	else
	{
308 309 310 311 312 313 314
		/* 
		 *archprintf(fout, "COPY %s FROM stdin;\n", fmtId(classname, force_quotes));
		 *
		 * - Not used as of V1.3 (needs to be in ArchiveEntry call)
		 *
		 */

315
		sprintf(query, "COPY %s TO stdout;\n", fmtId(classname, force_quotes));
316 317
	}
	res = PQexec(g_conn, query);
318 319
	if (!res ||
		PQresultStatus(res) == PGRES_FATAL_ERROR)
320
	{
321
		fprintf(stderr, "SQL query to dump the contents of Table '%s' "
322 323 324 325 326 327 328 329 330
				"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)
		{
331
			fprintf(stderr, "SQL query to dump the contents of Table '%s' "
332 333 334 335 336 337 338 339 340
					"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;
341

342 343
			while (!copydone)
			{
344
				ret = PQgetline(g_conn, copybuf, COPYBUFSIZ);
345 346 347 348 349 350 351 352 353

				if (copybuf[0] == '\\' &&
					copybuf[1] == '.' &&
					copybuf[2] == '\0')
				{
					copydone = true;	/* don't print this... */
				}
				else
				{
B
Bruce Momjian 已提交
354
					archputs(copybuf, fout);
355 356
					switch (ret)
					{
357 358 359 360
						case EOF:
							copydone = true;
							/* FALLTHROUGH */
						case 0:
B
Bruce Momjian 已提交
361
							archputc('\n', fout);
362 363 364
							break;
						case 1:
							break;
365 366
					}
				}
367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406

				/* 
				 * THROTTLE:
				 *
				 * There was considerable discussion in late July, 2000 regarding slowing down
				 * pg_dump when backing up large tables. Users with both slow & fast (muti-processor)
				 * machines experienced performance degradation when doing a backup.
				 *
				 * Initial attempts based on sleeping for a number of ms for each ms of work were deemed
				 * too complex, then a simple 'sleep in each loop' implementation was suggested. The latter
				 * failed because the loop was too tight. Finally, the following was implemented:
				 *
				 * If throttle is non-zero, then
				 * 		See how long since the last sleep.
				 *		Work out how long to sleep (based on ratio).
				 * 		If sleep is more than 100ms, then 
			     *			sleep 
				 *			reset timer
				 *		EndIf
				 * EndIf
				 *
				 * where the throttle value was the number of ms to sleep per ms of work. The calculation was 
				 * done in each loop.
				 *
				 * Most of the hard work is done in the backend, and this solution still did not work 
				 * particularly well: on slow machines, the ratio was 50:1, and on medium paced machines, 1:1,
				 * and on fast multi-processor machines, it had little or no effect, for reasons that were unclear.
				 *
				 * Further discussion ensued, and the proposal was dropped.
				 *
				 * For those people who want this feature, it can be implemented using gettimeofday in each
				 * loop, calculating the time since last sleep, multiplying that by the sleep ratio, then
				 * if the result is more than a preset 'minimum sleep time' (say 100ms), call the 'select'
				 * function to sleep for a subsecond period ie.
				 *
				 *        select(0, NULL, NULL, NULL, &tvi);
				 *
				 * This will return after the interval specified in the structure tvi. Fianally, call
				 * gettimeofday again to save the 'last sleep time'.
				 */	
407
			}
B
Bruce Momjian 已提交
408
			archprintf(fout, "\\.\n");
409
		}
410
		ret = PQendcopy(g_conn);
411 412
		if (ret != 0)
		{
413
			fprintf(stderr, "SQL query to dump the contents of Table '%s' "
414 415 416 417 418
					"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 已提交
419
			PQclear(res);
420 421 422
			exit_nicely(g_conn);
		}
	}
423

B
Bruce Momjian 已提交
424
	return 1;
425 426
}

B
Bruce Momjian 已提交
427 428
static int
dumpClasses_dumpData(Archive *fout, char* oid, void *dctxv)
429
{
B
Bruce Momjian 已提交
430 431 432
	const DumpContext	*dctx = (DumpContext*)dctxv;
	const char			*classname = dctx->tblinfo[dctx->tblidx].relname;

433 434
	PGresult   *res;
	PQExpBuffer q = createPQExpBuffer();
435 436
	int			tuple;
	int			field;
437

B
Hi, all  
Bruce Momjian 已提交
438 439
	appendPQExpBuffer(q, "SELECT * FROM %s", fmtId(classname, force_quotes));
	res = PQexec(g_conn, q->data);
440 441 442
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
443 444
		fprintf(stderr, "dumpClasses(): command failed.  Explanation from backend: '%s'.\n", 
					PQerrorMessage(g_conn));
445 446
		exit_nicely(g_conn);
	}
447
	for (tuple = 0; tuple < PQntuples(res); tuple++)
448
	{
B
Bruce Momjian 已提交
449
		archprintf(fout, "INSERT INTO %s ", fmtId(classname, force_quotes));
450
		if (attrNames == true)
451
		{
B
Hi, all  
Bruce Momjian 已提交
452 453
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "(");
454
			for (field = 0; field < PQnfields(res); field++)
455
			{
456
				if (field > 0)
B
Hi, all  
Bruce Momjian 已提交
457 458
					appendPQExpBuffer(q, ",");
				appendPQExpBuffer(q, fmtId(PQfname(res, field), force_quotes));
459
			}
B
Hi, all  
Bruce Momjian 已提交
460
			appendPQExpBuffer(q, ") ");
B
Bruce Momjian 已提交
461
			archprintf(fout, "%s", q->data);
462
		}
B
Bruce Momjian 已提交
463
		archprintf(fout, "VALUES (");
464
		for (field = 0; field < PQnfields(res); field++)
465
		{
466
			if (field > 0)
B
Bruce Momjian 已提交
467
				archprintf(fout, ",");
468
			if (PQgetisnull(res, tuple, field))
469
			{
B
Bruce Momjian 已提交
470
				archprintf(fout, "NULL");
471 472 473
				continue;
			}
			switch (PQftype(res, field))
474
			{
475 476
				case INT2OID:
				case INT4OID:
477
				case OIDOID:	/* int types */
478
				case FLOAT4OID:
479
				case FLOAT8OID:/* float types */
480
					/* These types are printed without quotes */
B
Bruce Momjian 已提交
481
					archprintf(fout, "%s",
482 483
							PQgetvalue(res, tuple, field));
					break;
484 485 486 487 488
				case ZPBITOID:
				case VARBITOID:
					archprintf(fout, "B'%s'",
							   PQgetvalue(res, tuple, field));
					break;
489
				default:
490

491 492
					/*
					 * All other types are printed as string literals,
493
					 * with appropriate escaping of special characters.
494
					 */
495 496 497
					resetPQExpBuffer(q);
					formatStringLiteral(q, PQgetvalue(res, tuple, field));
					archprintf(fout, "%s", q->data);
498
					break;
499
			}
500
		}
B
Bruce Momjian 已提交
501
		archprintf(fout, ");\n");
502 503
	}
	PQclear(res);
B
Bruce Momjian 已提交
504
	return 1;
505 506
}

507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541
/*
 * Convert a string value to an SQL string literal,
 * with appropriate escaping of special characters.
 * Quote mark ' goes to '' per SQL standard, other
 * stuff goes to \ sequences.
 * The literal is appended to the given PQExpBuffer.
 */
static void
formatStringLiteral(PQExpBuffer buf, const char *str)
{
	appendPQExpBufferChar(buf, '\'');
	while (*str)
	{
		char	ch = *str++;

		if (ch == '\\' || ch == '\'')
		{
			appendPQExpBufferChar(buf, ch);	/* double these */
			appendPQExpBufferChar(buf, ch);
		}
		else if ((unsigned char) ch < (unsigned char) ' ' &&
				 ch != '\n' && ch != '\t')
		{
			/* generate octal escape for control chars other than whitespace */
			appendPQExpBufferChar(buf, '\\');
			appendPQExpBufferChar(buf, ((ch >> 6) & 3) + '0');
			appendPQExpBufferChar(buf, ((ch >> 3) & 7) + '0');
			appendPQExpBufferChar(buf, (ch & 7) + '0');
		}
		else
			appendPQExpBufferChar(buf, ch);
	}
	appendPQExpBufferChar(buf, '\'');
}

542 543
/*
 * DumpClasses -
544
 *	  dump the contents of all the classes.
545 546
 */
static void
B
Bruce Momjian 已提交
547
dumpClasses(const TableInfo *tblinfo, const int numTables, Archive *fout,
548
			const char *onlytable, const bool oids, const bool force_quotes)
549 550
{

B
Bruce Momjian 已提交
551 552 553 554
	int				i;
	char	   		*all_only;
	DataDumperPtr	dumpFn;
	DumpContext		*dumpCtx;
555 556 557
	char			*oidsPart;
	char			copyBuf[512];
	char			*copyStmt;
558 559 560 561

	if (onlytable == NULL)
		all_only = "all";
	else
562
		all_only = "only";
563

564 565 566 567 568 569
	if (oids == true)
		oidsPart = "WITH OIDS ";
	else
		oidsPart = "";


570
	if (g_verbose)
571
		fprintf(stderr, "%s preparing to dump out the contents of %s %d table%s/sequence%s %s\n",
572 573
				g_comment_start, all_only,
				(onlytable == NULL) ? numTables : 1,
574
		  (onlytable == NULL) ? "s" : "", (onlytable == NULL) ? "s" : "",
575
				g_comment_end);
576

V
Vadim B. Mikheev 已提交
577 578 579 580 581 582 583 584 585 586
	/* 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)
587
					fprintf(stderr, "%s dumping out schema of sequence '%s' %s\n",
588
					 g_comment_start, tblinfo[i].relname, g_comment_end);
B
Bruce Momjian 已提交
589
				/* becomeUser(fout, tblinfo[i].usename); */
V
Vadim B. Mikheev 已提交
590 591 592 593
				dumpSequence(fout, tblinfo[i]);
			}
		}
	}
594 595 596

	for (i = 0; i < numTables; i++)
	{
597
		const char *classname = tblinfo[i].relname;
598 599

		/* Skip VIEW relations */
600
		if (tblinfo[i].viewdef != NULL)
601
			continue;
602 603

		if (tblinfo[i].sequence)/* already dumped */
V
Vadim B. Mikheev 已提交
604
			continue;
605 606 607 608

		if (!onlytable || (!strcmp(classname, onlytable)))
		{
			if (g_verbose)
609
				fprintf(stderr, "%s preparing to dump out the contents of Table '%s' %s\n",
610 611
						g_comment_start, classname, g_comment_end);

B
Bruce Momjian 已提交
612 613 614 615 616 617
			/* becomeUser(fout, tblinfo[i].usename); */

			dumpCtx = (DumpContext*)malloc(sizeof(DumpContext));
			dumpCtx->tblinfo = (TableInfo*)tblinfo;
			dumpCtx->tblidx = i;
			dumpCtx->oids = oids;
618

619 620
			if (!dumpData) /* Dump/restore using COPY */
			{
B
Bruce Momjian 已提交
621 622
				dumpFn = dumpClasses_nodumpData;
				/* dumpClasses_nodumpData(fout, classname, oids); */
623 624 625 626 627 628
				sprintf(copyBuf, "COPY %s %s FROM stdin;\n", fmtId(tblinfo[i].relname, force_quotes),
						oidsPart);
				copyStmt = copyBuf;
			}
			else /* Restore using INSERT */
			{
B
Bruce Momjian 已提交
629 630
				dumpFn = dumpClasses_dumpData;
				/* dumpClasses_dumpData(fout, classname); */
631 632
				copyStmt = NULL;
			}
B
Bruce Momjian 已提交
633 634

			ArchiveEntry(fout, tblinfo[i].oid, fmtId(tblinfo[i].relname, false),
635
							"TABLE DATA", NULL, "", "", copyStmt, tblinfo[i].usename,
B
Bruce Momjian 已提交
636
							dumpFn, dumpCtx);
637
		}
638
	}
639 640
}

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

B
Bruce Momjian 已提交
666
	RestoreOptions	*ropt;
667

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

B
Hi,  
Bruce Momjian 已提交
698 699
#endif

700
	g_verbose = false;
701
	force_quotes = true;
702 703 704 705 706

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

707
	dataOnly = schemaOnly = dumpData = attrNames = false;
708

709 710 711 712
	if (!strrchr(argv[0], SEP_CHAR))
		progname = argv[0];
	else
		progname = strrchr(argv[0], SEP_CHAR) + 1;
713

714 715 716 717
	/* Set defaulty options based on progname */
	if (strcmp(progname, "pg_backup") == 0)
	{
		format = "c";
P
Philip Warner 已提交
718
		outputBlobs = true;
719
	}
720

B
Hi,  
Bruce Momjian 已提交
721
#ifdef HAVE_GETOPT_LONG
722
	while ((c = getopt_long(argc, argv, "abcCdDf:F:h:inNoOp:sS:t:uvxzZ:V?", long_options, &optindex)) != -1)
B
Hi,  
Bruce Momjian 已提交
723
#else
724
	while ((c = getopt(argc, argv, "abcCdDf:F:h:inNoOp:sS:t:uvxzZ:V?-")) != -1)
B
Hi,  
Bruce Momjian 已提交
725
#endif
726

727 728 729
	{
		switch (c)
		{
730
			case 'a':			/* Dump data only */
731
				dataOnly = true;
732
				break;
733

734 735 736 737
			case 'b':			/* Dump blobs */
				outputBlobs = true;
				break;

B
Bruce Momjian 已提交
738
			case 'c':			/* clean (i.e., drop) schema prior to
B
Bruce Momjian 已提交
739 740 741 742
 								 * create */
				outputClean = 1;
 				break;
 
743 744 745 746 747
			case 'C':			/* Create DB */

				outputCreate = 1;
				break;

748
			case 'd':			/* dump data as proper insert strings */
749
				dumpData = true;
750
				break;
751

752 753
			case 'D':			/* dump data as proper insert strings with
								 * attr names */
754 755
				dumpData = true;
				attrNames = true;
756
				break;
757

758
			case 'f':
759 760
				filename = optarg;
				break;
761

B
Bruce Momjian 已提交
762 763 764
			case 'F':
				format = optarg;
				break;
765

B
Bruce Momjian 已提交
766
			case 'h':			/* server host */
767 768
				pghost = optarg;
				break;
769

770 771 772
			case 'i':			/* ignore database version mismatch */
				ignore_version = true;
				break;
773

B
Bruce Momjian 已提交
774 775
			case 'n':			/* Do not force double-quotes on
								 * identifiers */
776
				force_quotes = false;
777
				break;
778

779
			case 'N':			/* Force double-quotes on identifiers */
780
				force_quotes = true;
781
				break;
782

783
			case 'o':			/* Dump oids */
784
				oids = true;
785
				break;
786 787


788 789 790
			case 'O':			/* Don't reconnect to match owner */
				outputNoOwner = 1;
				break;
791

792 793 794
			case 'p':			/* server port */
				pgport = optarg;
				break;
795

796 797 798
			case 'R':			/* No reconnect */
				outputNoReconnect = 1;
				break;
799

800
			case 's':			/* dump schema only */
801
				schemaOnly = true;
802
				break;
803

804 805 806
			case 'S':			/* Username for superuser in plain text output */
				outputSuperuser = strdup(optarg);
				break;
807

808
			case 't':			/* Dump data for this table only */
809
				{
810
					int			i;
811 812

					tablename = strdup(optarg);
B
Bruce Momjian 已提交
813 814 815 816 817

					/*
					 * quoted string? Then strip quotes and preserve
					 * case...
					 */
818 819 820
					if (tablename[0] == '"')
					{
						strcpy(tablename, &tablename[1]);
B
Bruce Momjian 已提交
821 822
						if (*(tablename + strlen(tablename) - 1) == '"')
							*(tablename + strlen(tablename) - 1) = '\0';
823 824 825 826 827
					}
					/* otherwise, convert table name to lowercase... */
					else
					{
						for (i = 0; tablename[i]; i++)
828 829
							if (isupper((unsigned char) tablename[i]))
								tablename[i] = tolower((unsigned char) tablename[i]);
830
					}
831
				}
832
				break;
833

834 835 836
			case 'u':
				use_password = true;
				break;
837

838 839
			case 'v':			/* verbose */
				g_verbose = true;
840
				break;
841

842 843
			case 'x':			/* skip ACL dump */
				aclsSkip = true;
844
				break;
845

B
Bruce Momjian 已提交
846 847 848
			case 'Z':			/* Compression Level */
				compressLevel = atoi(optarg);
				break;
849

850 851 852 853
			case 'V':
				version();
				exit(0);
				break;
854

855
			case '?':
856 857 858 859 860 861 862 863 864 865 866 867 868 869 870

				/*
				 * 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);
				}
871
				break;
872 873 874
#ifndef HAVE_GETOPT_LONG
			case '-':
				fprintf(stderr, "%s was compiled without support for long options.\n"
875
				   "Use -? for help on invocation options.\n", progname);
876 877 878
				exit(1);
				break;
#endif
879
			default:
880 881
				fprintf(stderr, "%s: unknown option -%c\nTry -? for help.\n", progname, c);
				exit(1);
882 883 884
		}
	}

P
Philip Warner 已提交
885 886 887 888 889 890 891 892
	if (optind < (argc - 1)) {
		fprintf(stderr,
				"%s: extra parameters found on command line after '%s' (first is '%s').\n"
			    "Please respecify command.\nUse -? for help on invocation options.\n",
				progname, argv[optind], argv[optind+1]);
		exit(1);
	}

B
Bruce Momjian 已提交
893 894 895 896 897 898 899 900
	if (dataOnly && schemaOnly)
	{
		fprintf(stderr,
				"%s: 'Schema Only' and 'Data Only' are incompatible options.\n",
				progname);
		exit(1);
	}

901 902 903 904 905 906 907 908
	if (outputBlobs && (tablename != NULL) )
	{
		fprintf(stderr,
				"%s: BLOB output is not supported for a single table. Use a full dump instead.\n",
				progname);
		exit(1);
	}

909 910 911
	if (dumpData == true && oids == true)
	{
		fprintf(stderr,
912
				"%s: INSERT's can not set oids, so INSERT and OID options can not be used together.\n",
913
				progname);
914
		exit(1);
915
	}
916

P
Philip Warner 已提交
917 918 919 920 921 922 923 924
	if (outputBlobs == true && (format[0] == 'p' || format[0] == 'P') )
	{
		fprintf(stderr,
			"%s: BLOB output is not supported for plain text dump files. Use a different output format.\n",
			progname);
		exit(1);
	}

925
	/* open the output file */
B
Bruce Momjian 已提交
926 927 928 929 930 931 932 933 934 935 936 937 938 939 940
	switch (format[0]) {

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

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

		case 'p':
		case 'P':
			plainText = 1;
941 942 943 944 945 946
			g_fout = CreateArchive(filename, archNull, 0);
			break;

		case 't':
		case 'T':
			g_fout = CreateArchive(filename, archTar, compressLevel);
B
Bruce Momjian 已提交
947 948 949
			break;

		default:
950
			fprintf(stderr,
B
Bruce Momjian 已提交
951 952 953 954 955 956 957 958 959 960
				"%s: invalid output format '%s' specified\n", progname, format);
		    exit(1);	
	}

	if (g_fout == NULL)
	{
		fprintf(stderr,
			 "%s: could not open output file named %s for writing\n",
				progname, filename);
		exit(1);
961 962
	}

963 964
	/* Let the archiver know how noisy to be */
	g_fout->verbose = g_verbose;
965

966
	dbname = argv[optind];
967

968
	/* Open the database using the Archiver, so it knows about it. Errors mean death */
969
	g_conn = ConnectDatabase(g_fout, dbname, pghost, pgport, use_password, ignore_version);
970

971 972 973 974 975 976 977 978
	/*
	 * Start serializable transaction to dump consistent data
	 */
	{
		PGresult   *res;

		res = PQexec(g_conn, "begin");
		if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
979 980 981
		    exit_horribly(g_fout, "BEGIN command failed. Explanation from backend: '%s'.\n",
				PQerrorMessage(g_conn));

982 983 984
		PQclear(res);
		res = PQexec(g_conn, "set transaction isolation level serializable");
		if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
985 986 987
		    exit_horribly(g_fout, "SET TRANSACTION command failed. Explanation from backend: '%s'.\n",
				PQerrorMessage(g_conn));

988 989 990
		PQclear(res);
	}

991
	g_last_builtin_oid = findLastBuiltinOid(dbname);
992

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

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

	if (g_verbose)
1001
			fprintf(stderr, "%s last builtin oid is %u %s\n",
1002
					g_comment_start, g_last_builtin_oid, g_comment_end);
B
Bruce Momjian 已提交
1003
	tblinfo = dumpSchema(g_fout, &numTables, tablename, aclsSkip, oids, schemaOnly, dataOnly);
1004

1005
	if (!schemaOnly)
1006 1007 1008 1009 1010 1011
	{
	    dumpClasses(tblinfo, numTables, g_fout, tablename, oids, force_quotes);
	}

	if (outputBlobs)
		ArchiveEntry(g_fout, "0", "BLOBS", "BLOBS", NULL, "", "", "", "", dumpBlobs, 0); 
1012

1013 1014
	if (!dataOnly)				/* dump indexes and triggers at the end
								 * for performance */
V
Vadim B. Mikheev 已提交
1015 1016 1017
	{
		dumpSchemaIdx(g_fout, tablename, tblinfo, numTables);
		dumpTriggers(g_fout, tablename, tblinfo, numTables);
1018
		dumpRules(g_fout, tablename, tblinfo, numTables);
V
Vadim B. Mikheev 已提交
1019
	}
1020

1021 1022
	/* Now sort the output nicely */
	SortTocByOID(g_fout);
1023
	MoveToStart(g_fout, "DATABASE");
1024 1025 1026 1027 1028
	MoveToEnd(g_fout, "TABLE DATA");
	MoveToEnd(g_fout, "BLOBS");
	MoveToEnd(g_fout, "INDEX");
	MoveToEnd(g_fout, "TRIGGER");
	MoveToEnd(g_fout, "RULE");
1029
	MoveToEnd(g_fout, "SEQUENCE SET");
1030

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

		if (outputSuperuser)
			ropt->superuser = outputSuperuser;
		else
			ropt->superuser = PQuser(g_conn);
B
Bruce Momjian 已提交
1046 1047 1048 1049 1050 1051 1052 1053 1054 1055

		if (compressLevel == -1)
		    ropt->compression = 0;
		else
		    ropt->compression = compressLevel;

		RestoreArchive(g_fout, ropt);
	}

	CloseArchive(g_fout);
1056

1057 1058 1059
	clearTableInfo(tblinfo, numTables);
	PQfinish(g_conn);
	exit(0);
1060 1061
}

1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080
/*
 * dumpDatabase:
 *	dump the database definition
 *
 */
static int 
dumpDatabase(Archive *AH)
{
	PQExpBuffer		dbQry = createPQExpBuffer();
	PQExpBuffer		delQry = createPQExpBuffer();
	PQExpBuffer		creaQry = createPQExpBuffer();
	PGresult		*res;
	int				ntups;
	int				i_dba;

	if (g_verbose)
		fprintf(stderr, "%s saving database definition\n", g_comment_start);

	/* Get the dba */
1081
	appendPQExpBuffer(dbQry, "select (select usename from pg_user where datdba = usesysid) as dba from pg_database"
1082 1083
							" where datname = ");
	formatStringLiteral(dbQry, PQdb(g_conn));
1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116

	res = PQexec(g_conn, dbQry->data);
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
		fprintf(stderr, "getDatabase(): SELECT failed.  Explanation from backend: '%s'.\n", 
						PQerrorMessage(g_conn));
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	if (ntups != 1)
	{
		fprintf(stderr, "getDatabase(): SELECT returned %d databases.\n", ntups);
		exit_nicely(g_conn);
	}

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

	ArchiveEntry(AH, "0" /* OID */, PQdb(g_conn) /* Name */, "DATABASE", NULL, 
						creaQry->data /* Create */, delQry->data /*Del*/,
						"" /* Copy */, PQgetvalue(res, 0, i_dba) /*Owner*/, 
						NULL /* Dumper */, NULL /* Dumper Arg */);	

	PQclear(res);

	return 1;
}


1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141
/*
 * dumpBlobs:
 *	dump all blobs
 *
 */

#define loBufSize 16384 
#define loFetchSize 1000

static int 
dumpBlobs(Archive *AH, char* junkOid, void *junkVal)
{
	PQExpBuffer		oidQry = createPQExpBuffer();
	PQExpBuffer		oidFetchQry = createPQExpBuffer();
	PGresult		*res;
	int				i;
	int				loFd;
	char			buf[loBufSize];
	int				cnt;
	int				blobOid;

	if (g_verbose)
		fprintf(stderr, "%s saving BLOBs\n", g_comment_start);

	/* Cursor to get all BLOB tables */
1142
    appendPQExpBuffer(oidQry, "Declare blobOid Cursor for SELECT DISTINCT loid FROM pg_largeobject");
1143 1144 1145 1146

	res = PQexec(g_conn, oidQry->data);
	if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
	{
1147 1148
		fprintf(stderr, "dumpBlobs(): Declare Cursor failed.  Explanation from backend: '%s'.\n", 
										PQerrorMessage(g_conn));
1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203
		exit_nicely(g_conn);
	}

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

	do {
		/* Do a fetch */
		PQclear(res);
		res = PQexec(g_conn, oidFetchQry->data);

		if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
		{
		    fprintf(stderr, "dumpBlobs(): Fetch Cursor failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
		    exit_nicely(g_conn);
		}

		/* Process the tuples, if any */
		for (i = 0; i < PQntuples(res); i++)
		{
			blobOid = atoi(PQgetvalue(res, i, 0));
			/* Open the BLOB */
			loFd = lo_open(g_conn, blobOid, INV_READ);
			if (loFd == -1)
			{
				fprintf(stderr, "dumpBlobs(): Could not open large object.  "
									"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
				exit_nicely(g_conn);
			}

			StartBlob(AH, blobOid);

			/* Now read it in chunks, sending data to archive */
			do {
				cnt = lo_read(g_conn, loFd, buf, loBufSize);
				if (cnt < 0) {
					fprintf(stderr, "dumpBlobs(): Error reading large object. "
								" Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
					exit_nicely(g_conn);
				}

				WriteData(AH, buf, cnt); 

			} while (cnt > 0);

			lo_close(g_conn, loFd);

			EndBlob(AH, blobOid);

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

	return 1;
}

1204
/*
1205 1206
 * getTypes:
 *	  read all base types in the system catalogs and return them in the
1207 1208
 * TypeInfo* structure
 *
1209
 *	numTypes is set to the number of types read in
1210 1211
 *
 */
1212
TypeInfo   *
1213 1214
getTypes(int *numTypes)
{
B
Bruce Momjian 已提交
1215
	PGresult   *res;
1216 1217
	int			ntups;
	int			i;
1218
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1219
	TypeInfo   *tinfo;
1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235

	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;
1236
	int			i_typedefn;
1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248

	/* 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 已提交
1249
	appendPQExpBuffer(query, "SELECT pg_type.oid, typowner, typname, typlen, typprtlen, "
1250 1251 1252 1253 1254
		"typinput, typoutput, typreceive, typsend, typelem, typdelim, "
		"typdefault, typrelid, typbyval, "
		"(select usename from pg_user where typowner = usesysid) as usename, "
		"format_type(pg_type.oid, NULL) as typedefn "
		"from pg_type" );
1255

B
Hi, all  
Bruce Momjian 已提交
1256
	res = PQexec(g_conn, query->data);
1257 1258 1259
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1260
		fprintf(stderr, "getTypes(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282
		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");
1283
	i_typedefn = PQfnumber(res, "typedefn");
1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300

	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));
1301
		tinfo[i].typedefn = strdup(PQgetvalue(res, i, i_typedefn));
1302

1303 1304 1305
		if (strlen(tinfo[i].usename) == 0)
			fprintf(stderr, "WARNING: owner of type '%s' appears to be invalid\n",tinfo[i].typname);

1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325
		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;
1326 1327 1328 1329
}

/*
 * getOperators:
1330
 *	  read all operators in the system catalogs and return them in the
1331 1332
 * OprInfo* structure
 *
1333 1334
 *	numOprs is set to the number of operators read in
 *
1335 1336
 *
 */
1337
OprInfo    *
1338 1339
getOperators(int *numOprs)
{
1340
	PGresult   *res;
1341 1342
	int			ntups;
	int			i;
1343
	PQExpBuffer query = createPQExpBuffer();
1344

B
Bruce Momjian 已提交
1345
	OprInfo    *oprinfo;
1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360

	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;
1361 1362 1363 1364 1365 1366

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

B
Hi, all  
Bruce Momjian 已提交
1367
	appendPQExpBuffer(query, "SELECT pg_operator.oid, oprname, oprkind, oprcode, "
1368
			   "oprleft, oprright, oprcom, oprnegate, oprrest, oprjoin, "
1369 1370 1371
					"oprcanhash, oprlsortop, oprrsortop, "
					"(select usename from pg_user where oprowner = usesysid) as usename "
					"from pg_operator");
1372

B
Hi, all  
Bruce Momjian 已提交
1373
	res = PQexec(g_conn, query->data);
1374 1375 1376
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1377
		fprintf(stderr, "getOperators(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 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
		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));
1417 1418 1419 1420 1421

		if (strlen(oprinfo[i].usename) == 0)
			fprintf(stderr, "WARNING: owner of operator '%s' appears to be invalid\n",
						oprinfo[i].oprname);

1422 1423 1424 1425 1426
	}

	PQclear(res);

	return oprinfo;
1427 1428
}

1429
void
1430
clearTypeInfo(TypeInfo *tp, int numTypes)
1431
{
1432
	int			i;
1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465

	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);
1466 1467 1468
}

void
1469
clearFuncInfo(FuncInfo *fun, int numFuncs)
1470
{
1471 1472
	int			i,
				a;
1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483

	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);
1484
		for (a = 0; a < FUNC_MAX_ARGS; ++a)
1485 1486 1487 1488 1489 1490 1491 1492 1493 1494
			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);
1495 1496
}

1497
static void
1498
clearTableInfo(TableInfo *tblinfo, int numTables)
1499
{
1500 1501
	int			i,
				j;
1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521

	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 已提交
1522 1523
			if (tblinfo[i].attoids[j])
				free(tblinfo[i].attoids[j]);
1524 1525 1526 1527 1528
			if (tblinfo[i].attnames[j])
				free(tblinfo[i].attnames[j]);
			if (tblinfo[i].typnames[j])
				free(tblinfo[i].typnames[j]);
		}
B
Bruce Momjian 已提交
1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544

		if (tblinfo[i].triggers) {
			for (j = 0; j < tblinfo[i].ntrig ; j++)
			{
				if (tblinfo[i].triggers[j].tgsrc)
					free(tblinfo[i].triggers[j].tgsrc);
				if (tblinfo[i].triggers[j].oid)
					free(tblinfo[i].triggers[j].oid);
				if (tblinfo[i].triggers[j].tgname)
					free(tblinfo[i].triggers[j].tgname);
				if (tblinfo[i].triggers[j].tgdel)
					free(tblinfo[i].triggers[j].tgdel);
			}
			free(tblinfo[i].triggers);
		}

1545 1546
		if (tblinfo[i].atttypmod)
			free((int *) tblinfo[i].atttypmod);
1547 1548 1549 1550
		if (tblinfo[i].inhAttrs)
			free((int *) tblinfo[i].inhAttrs);
		if (tblinfo[i].attnames)
			free(tblinfo[i].attnames);
1551 1552
		if (tblinfo[i].atttypedefns)
			free(tblinfo[i].atttypedefns);
1553 1554 1555 1556
		if (tblinfo[i].typnames)
			free(tblinfo[i].typnames);
		if (tblinfo[i].notnull)
			free(tblinfo[i].notnull);
1557 1558
		if (tblinfo[i].primary_key)
			free(tblinfo[i].primary_key);
1559 1560
		if (tblinfo[i].primary_key_name)
			free(tblinfo[i].primary_key_name);
1561 1562
	}
	free(tblinfo);
1563 1564
}

1565
void
1566
clearInhInfo(InhInfo *inh, int numInherits)
1567
{
1568
	int			i;
1569 1570 1571 1572 1573

	if (!inh)
		return;
	for (i = 0; i < numInherits; ++i)
	{
1574 1575
		if (inh[i].inhrelid)
			free(inh[i].inhrelid);
1576 1577 1578 1579
		if (inh[i].inhparent)
			free(inh[i].inhparent);
	}
	free(inh);
1580 1581 1582
}

void
1583
clearOprInfo(OprInfo *opr, int numOprs)
1584
{
1585
	int			i;
1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620

	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);
1621 1622 1623
}

void
1624
clearIndInfo(IndInfo *ind, int numIndices)
1625
{
1626 1627
	int			i,
				a;
1628 1629 1630 1631 1632

	if (!ind)
		return;
	for (i = 0; i < numIndices; ++i)
	{
B
Bruce,  
Bruce Momjian 已提交
1633 1634
		if (ind[i].indoid)
			free(ind[i].indoid);
1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653
		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);
1654 1655 1656
}

void
B
Bruce Momjian 已提交
1657
clearAggInfo(AggInfo *agginfo, int numArgs)
1658
{
1659
	int			i;
1660 1661 1662 1663 1664 1665 1666 1667 1668

	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);
1669 1670
		if (agginfo[i].aggtransfn)
			free(agginfo[i].aggtransfn);
1671 1672
		if (agginfo[i].aggfinalfn)
			free(agginfo[i].aggfinalfn);
1673 1674
		if (agginfo[i].aggtranstype)
			free(agginfo[i].aggtranstype);
1675 1676
		if (agginfo[i].aggbasetype)
			free(agginfo[i].aggbasetype);
1677 1678
		if (agginfo[i].agginitval)
			free(agginfo[i].agginitval);
1679 1680 1681 1682
		if (agginfo[i].usename)
			free(agginfo[i].usename);
	}
	free(agginfo);
1683
}
1684 1685 1686

/*
 * getAggregates:
1687
 *	  read all the user-defined aggregates in the system catalogs and
1688 1689
 * return them in the AggInfo* structure
 *
1690 1691
 * numAggs is set to the number of aggregates read in
 *
1692 1693
 *
 */
1694
AggInfo    *
1695 1696
getAggregates(int *numAggs)
{
B
Bruce Momjian 已提交
1697
	PGresult   *res;
1698 1699
	int			ntups;
	int			i;
1700
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1701
	AggInfo    *agginfo;
1702 1703 1704

	int			i_oid;
	int			i_aggname;
1705
	int			i_aggtransfn;
1706
	int			i_aggfinalfn;
1707
	int			i_aggtranstype;
1708
	int			i_aggbasetype;
1709
	int			i_agginitval;
1710
	int			i_usename;
1711 1712 1713

	/* find all user-defined aggregates */

B
Hi, all  
Bruce Momjian 已提交
1714
	appendPQExpBuffer(query,
1715 1716 1717 1718 1719
						"SELECT pg_aggregate.oid, aggname, aggtransfn, "
						"aggfinalfn, aggtranstype, aggbasetype, "
						"agginitval, "
						"(select usename from pg_user where aggowner = usesysid) as usename "
						"from pg_aggregate" );
1720

B
Hi, all  
Bruce Momjian 已提交
1721
	res = PQexec(g_conn, query->data);
1722 1723 1724
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1725 1726
		fprintf(stderr, "getAggregates(): SELECT failed.  Explanation from backend: '%s'.\n",
					PQerrorMessage(g_conn));
1727 1728 1729 1730 1731 1732 1733 1734 1735 1736
		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");
1737
	i_aggtransfn = PQfnumber(res, "aggtransfn");
1738
	i_aggfinalfn = PQfnumber(res, "aggfinalfn");
1739
	i_aggtranstype = PQfnumber(res, "aggtranstype");
1740
	i_aggbasetype = PQfnumber(res, "aggbasetype");
1741
	i_agginitval = PQfnumber(res, "agginitval");
1742 1743 1744 1745 1746 1747
	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));
1748
		agginfo[i].aggtransfn = strdup(PQgetvalue(res, i, i_aggtransfn));
1749
		agginfo[i].aggfinalfn = strdup(PQgetvalue(res, i, i_aggfinalfn));
1750
		agginfo[i].aggtranstype = strdup(PQgetvalue(res, i, i_aggtranstype));
1751
		agginfo[i].aggbasetype = strdup(PQgetvalue(res, i, i_aggbasetype));
1752
		agginfo[i].agginitval = strdup(PQgetvalue(res, i, i_agginitval));
1753
		agginfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1754 1755 1756 1757
		if (strlen(agginfo[i].usename) == 0)
			fprintf(stderr, "WARNING: owner of aggregate '%s' appears to be invalid\n",
						agginfo[i].aggname);

1758 1759 1760 1761 1762
	}

	PQclear(res);

	return agginfo;
1763 1764 1765 1766
}

/*
 * getFuncs:
1767
 *	  read all the user-defined functions in the system catalogs and
1768 1769
 * return them in the FuncInfo* structure
 *
1770 1771
 * numFuncs is set to the number of functions read in
 *
1772 1773
 *
 */
1774
FuncInfo   *
1775 1776
getFuncs(int *numFuncs)
{
1777
	PGresult   *res;
1778 1779
	int			ntups;
	int			i;
1780 1781
	PQExpBuffer query = createPQExpBuffer();
	FuncInfo   *finfo;
1782 1783 1784 1785 1786 1787 1788 1789 1790 1791

	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;
1792
	int			i_iscachable;
1793
	int			i_isstrict;
1794
	int			i_usename;
1795 1796 1797

	/* find all user-defined funcs */

B
Hi, all  
Bruce Momjian 已提交
1798
	appendPQExpBuffer(query,
1799
		   "SELECT pg_proc.oid, proname, prolang, pronargs, prorettype, "
1800 1801
					  "proretset, proargtypes, prosrc, probin, "
					  "(select usename from pg_user where proowner = usesysid) as usename, "
1802
					  "proiscachable, proisstrict "
1803 1804
					  "from pg_proc "
				 "where pg_proc.oid > '%u'::oid",
1805
					  g_last_builtin_oid);
1806

B
Hi, all  
Bruce Momjian 已提交
1807
	res = PQexec(g_conn, query->data);
1808 1809 1810
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1811 1812
		fprintf(stderr, "getFuncs(): SELECT failed.  Explanation from backend: '%s'.\n", 
					PQerrorMessage(g_conn));
1813 1814 1815 1816 1817 1818 1819 1820 1821
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numFuncs = ntups;

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

1822 1823
	memset((char *) finfo, 0, ntups * sizeof(FuncInfo));

1824 1825 1826 1827 1828 1829 1830 1831 1832
	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");
1833
	i_iscachable = PQfnumber(res, "proiscachable");
1834
	i_isstrict = PQfnumber(res, "proisstrict");
1835 1836 1837 1838 1839 1840 1841
	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));

1842
		finfo[i].prosrc = strdup(PQgetvalue(res, i, i_prosrc));
1843 1844 1845 1846 1847
		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));
1848
		finfo[i].lang = atoi(PQgetvalue(res, i, i_prolang));
1849
		finfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1850
		finfo[i].iscachable = (strcmp(PQgetvalue(res, i, i_iscachable),"t") == 0);
1851
		finfo[i].isstrict = (strcmp(PQgetvalue(res, i, i_isstrict),"t") == 0);
1852 1853 1854 1855 1856

		if (strlen(finfo[i].usename) == 0)
			fprintf(stderr, "WARNING: owner of function '%s' appears to be invalid\n",
						finfo[i].proname);

1857 1858 1859 1860
		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);
1861
			exit(1);
1862 1863 1864 1865
		}
		parseNumericArray(PQgetvalue(res, i, i_proargtypes),
						  finfo[i].argtypes,
						  finfo[i].nargs);
1866 1867 1868 1869 1870 1871
		finfo[i].dumped = 0;
	}

	PQclear(res);

	return finfo;
1872 1873 1874 1875 1876

}

/*
 * getTables
1877
 *	  read all the user-defined tables (no indices, no catalogs)
1878 1879
 * in the system catalogs return them in the TableInfo* structure
 *
1880 1881
 * numTables is set to the number of tables read in
 *
1882 1883
 *
 */
1884
TableInfo  *
V
Vadim B. Mikheev 已提交
1885
getTables(int *numTables, FuncInfo *finfo, int numFuncs)
1886
{
1887
	PGresult   *res;
1888 1889
	int			ntups;
	int			i;
1890
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1891
	PQExpBuffer delqry = createPQExpBuffer();
1892
	TableInfo  *tblinfo;
1893

B
Bruce Momjian 已提交
1894
	int			i_reloid;
1895 1896 1897 1898
	int			i_relname;
	int			i_relkind;
	int			i_relacl;
	int			i_usename;
V
Vadim B. Mikheev 已提交
1899 1900
	int			i_relchecks;
	int			i_reltriggers;
1901
	int			i_relhasindex;
1902 1903 1904 1905 1906

	char		relkindview[2];

	relkindview[0] = RELKIND_VIEW;
	relkindview[1] = '\0';
1907 1908 1909 1910 1911 1912

	/*
	 * 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*
	 *
1913
	 * we ignore tables that are not type 'r' (ordinary relation) or 'S'
1914
	 * (sequence) or 'v' (view).
1915 1916
	 */

B
Hi, all  
Bruce Momjian 已提交
1917
	appendPQExpBuffer(query,
1918 1919
			   "SELECT pg_class.oid, relname, relkind, relacl, "
					  "(select usename from pg_user where relowner = usesysid) as usename, "
1920
					  "relchecks, reltriggers, relhasindex "
1921 1922
					  "from pg_class "
					  "where relname !~ '^pg_' "
1923 1924 1925
					  "and relkind in ('%c', '%c', '%c') "
					  "order by oid",
				RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW);
1926

B
Hi, all  
Bruce Momjian 已提交
1927
	res = PQexec(g_conn, query->data);
1928 1929 1930
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1931 1932
		fprintf(stderr, "getTables(): SELECT failed.  Explanation from backend: '%s'.\n", 
					PQerrorMessage(g_conn));
1933 1934 1935 1936 1937 1938 1939 1940 1941
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numTables = ntups;

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

B
Bruce Momjian 已提交
1942
	i_reloid = PQfnumber(res, "oid");
1943 1944 1945 1946
	i_relname = PQfnumber(res, "relname");
	i_relkind = PQfnumber(res, "relkind");
	i_relacl = PQfnumber(res, "relacl");
	i_usename = PQfnumber(res, "usename");
V
Vadim B. Mikheev 已提交
1947 1948
	i_relchecks = PQfnumber(res, "relchecks");
	i_reltriggers = PQfnumber(res, "reltriggers");
B
Hi, all  
Bruce Momjian 已提交
1949
	i_relhasindex = PQfnumber(res, "relhasindex");
1950 1951 1952

	for (i = 0; i < ntups; i++)
	{
B
Bruce Momjian 已提交
1953
		tblinfo[i].oid = strdup(PQgetvalue(res, i, i_reloid));
1954 1955 1956 1957
		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 已提交
1958 1959
		tblinfo[i].ncheck = atoi(PQgetvalue(res, i, i_relchecks));
		tblinfo[i].ntrig = atoi(PQgetvalue(res, i, i_reltriggers));
1960

1961 1962 1963 1964
		if (strlen(tblinfo[i].usename) == 0)
			fprintf(stderr, "WARNING: owner of table '%s' appears to be invalid\n",
						tblinfo[i].relname);

1965 1966 1967 1968 1969 1970
		/* Get view definition */
		if (strcmp(PQgetvalue(res, i, i_relkind), relkindview) == 0)
		{
			PGresult   *res2;

			resetPQExpBuffer(query);
1971 1972 1973
			appendPQExpBuffer(query, "SELECT pg_get_viewdef(");
			formatStringLiteral(query, tblinfo[i].relname);
			appendPQExpBuffer(query, ") as viewdef");
1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007
			res2 = PQexec(g_conn, query->data);
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
				fprintf(stderr, "getTables(): SELECT (for VIEW DEFINITION) failed.  "
							"Explanation from backend: %s",
							PQerrorMessage(g_conn));
				exit_nicely(g_conn);
			}

			if (PQntuples(res2) != 1) 
			{
				if (PQntuples(res2) < 1)
				{
					fprintf(stderr, "getTables(): SELECT (for VIEW %s) returned no definitions",
								tblinfo[i].relname);
				} else {
					fprintf(stderr, "getTables(): SELECT (for VIEW %s) returned more than 1 definition",
								tblinfo[i].relname);
				}
				exit_nicely(g_conn);
			}

			tblinfo[i].viewdef = strdup(PQgetvalue(res2, 0, 0));

			if (strlen(tblinfo[i].viewdef) == 0) 
			{
				fprintf(stderr, "getTables(): SELECT (for VIEW %s) returned empty definition",
							tblinfo[i].relname);
				exit_nicely(g_conn);
			}
		}
		else
			tblinfo[i].viewdef = NULL;

B
Bruce Momjian 已提交
2008 2009 2010
		/*
		 * Exclude inherited CHECKs from CHECK constraints total. If a
		 * constraint matches by name and condition with a constraint
2011 2012 2013 2014
		 * belonging to a parent class, we assume it was inherited.
		 */
		if (tblinfo[i].ncheck > 0)
		{
B
Bruce Momjian 已提交
2015
			PGresult   *res2;
2016 2017
			int			ntups2;

B
Bruce Momjian 已提交
2018
			if (g_verbose)
2019 2020 2021 2022 2023 2024
				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 已提交
2025 2026
			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "SELECT rcname from pg_relcheck, pg_inherits as i "
2027 2028 2029 2030 2031 2032 2033 2034
							  "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 已提交
2035
			res2 = PQexec(g_conn, query->data);
2036 2037 2038
			if (!res2 ||
				PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
B
Bruce Momjian 已提交
2039 2040
				fprintf(stderr, "getTables(): SELECT (for inherited CHECK) failed.  "
							"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055
				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 已提交
2056 2057 2058
		if (tblinfo[i].ncheck > 0)
		{
			PGresult   *res2;
2059 2060
			int			i_rcname,
						i_rcsrc;
V
Vadim B. Mikheev 已提交
2061 2062
			int			ntups2;
			int			i2;
2063

V
Vadim B. Mikheev 已提交
2064
			if (g_verbose)
2065
				fprintf(stderr, "%s finding CHECK constraints for relation: '%s' %s\n",
V
Vadim B. Mikheev 已提交
2066 2067 2068
						g_comment_start,
						tblinfo[i].relname,
						g_comment_end);
2069

B
Hi, all  
Bruce Momjian 已提交
2070 2071
			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "SELECT rcname, rcsrc from pg_relcheck "
2072 2073 2074 2075 2076 2077 2078 2079
							  "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 已提交
2080
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
2081
			if (!res2 ||
2082
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2083
			{
B
Bruce Momjian 已提交
2084 2085
				fprintf(stderr, "getTables(): SELECT (for CHECK) failed.  "
						"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
V
Vadim B. Mikheev 已提交
2086 2087 2088 2089 2090
				exit_nicely(g_conn);
			}
			ntups2 = PQntuples(res2);
			if (ntups2 != tblinfo[i].ncheck)
			{
2091
				fprintf(stderr, "getTables(): relation '%s': %d CHECKs were expected, but got %d\n",
2092
						tblinfo[i].relname, tblinfo[i].ncheck, ntups2);
V
Vadim B. Mikheev 已提交
2093 2094 2095 2096
				exit_nicely(g_conn);
			}
			i_rcname = PQfnumber(res2, "rcname");
			i_rcsrc = PQfnumber(res2, "rcsrc");
2097
			tblinfo[i].check_expr = (char **) malloc(ntups2 * sizeof(char *));
V
Vadim B. Mikheev 已提交
2098 2099
			for (i2 = 0; i2 < ntups2; i2++)
			{
B
Bruce Momjian 已提交
2100 2101
				const char *name = PQgetvalue(res2, i2, i_rcname);
				const char *expr = PQgetvalue(res2, i2, i_rcsrc);
2102

B
Hi, all  
Bruce Momjian 已提交
2103
				resetPQExpBuffer(query);
2104 2105
				if (name[0] != '$')
				{
2106 2107
					appendPQExpBuffer(query, "CONSTRAINT %s ",
									  fmtId(name, force_quotes));
B
Bruce Momjian 已提交
2108
				}
2109
				appendPQExpBuffer(query, "CHECK (%s)", expr);
B
Hi, all  
Bruce Momjian 已提交
2110
				tblinfo[i].check_expr[i2] = strdup(query->data);
V
Vadim B. Mikheev 已提交
2111 2112 2113 2114 2115
			}
			PQclear(res2);
		}
		else
			tblinfo[i].check_expr = NULL;
2116

2117 2118
		/* Get primary key */
		if (strcmp(PQgetvalue(res, i, i_relhasindex), "t") == 0)
2119
		{
2120 2121 2122
			PGresult   *res2;
			char		str[INDEX_MAX_KEYS * (NAMEDATALEN * 2 + 4) + 1];
			int			j;
B
Bruce Momjian 已提交
2123

B
Hi, all  
Bruce Momjian 已提交
2124
			resetPQExpBuffer(query);
B
Bruce Momjian 已提交
2125
			appendPQExpBuffer(query,
2126 2127 2128 2129 2130 2131
							  "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);
2132
			res2 = PQexec(g_conn, query->data);
B
Bruce Momjian 已提交
2133 2134 2135
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
				fprintf(stderr, "getTables(): SELECT (for PRIMARY KEY) failed.  Explanation from backend: %s",
2136
						PQerrorMessage(g_conn));
B
Bruce Momjian 已提交
2137 2138
				exit_nicely(g_conn);
			}
2139

2140 2141 2142
			str[0] = '\0';
			for (j = 0; j < PQntuples(res2); j++)
			{
2143
				if (strlen(str) > 0)
2144 2145 2146 2147
					strcat(str, ", ");
				strcat(str, fmtId(PQgetvalue(res2, j, 0), force_quotes));
			}

2148 2149
			if (strlen(str) > 0)
			{
2150
				tblinfo[i].primary_key = strdup(str);
2151 2152
				if (tblinfo[i].primary_key == NULL)
				{
2153 2154 2155 2156 2157 2158
					perror("strdup");
					exit(1);
				}
			}
			else
				tblinfo[i].primary_key = NULL;
2159
		}
2160 2161
		else
			tblinfo[i].primary_key = NULL;
B
Bruce Momjian 已提交
2162

2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205
		/* Get primary key name (if primary key exist) */
		if (tblinfo[i].primary_key)
		{
			PGresult   *res2;
			int		   n;

			resetPQExpBuffer(query);
			appendPQExpBuffer(query,
							  "SELECT c.relname "
							  "FROM pg_index i, pg_class c "
							  "WHERE i.indrelid = %s"
							  "AND   i.indisprimary "
							  "AND   c.oid = i.indexrelid ",
							  tblinfo[i].oid);
			res2 = PQexec(g_conn, query->data);
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
				fprintf(stderr, "getTables(): SELECT (for PRIMARY KEY NAME) failed.  Explanation from backend: %s",
						PQerrorMessage(g_conn));
				exit_nicely(g_conn);
			}

			n = PQntuples(res2);
			if (n != 1)
			{
				fprintf(stderr,
						"getTables(): SELECT (for PRIMARY KEY NAME) failed. This is impossible but object with OID == %s have %d primary keys.\n",
						tblinfo[i].oid,
						n);
				exit_nicely(g_conn);
			}

			tblinfo[i].primary_key_name =
				strdup(fmtId(PQgetvalue(res2, 0, 0), force_quotes));
			if (tblinfo[i].primary_key_name == NULL)
			{
				perror("strdup");
				exit(1);
			}
		}
		else
			tblinfo[i].primary_key_name = NULL;

V
Vadim B. Mikheev 已提交
2206 2207 2208 2209
		/* Get Triggers */
		if (tblinfo[i].ntrig > 0)
		{
			PGresult   *res2;
B
Bruce,  
Bruce Momjian 已提交
2210
			int			i_tgoid,
2211
						i_tgname,
2212 2213 2214
						i_tgfoid,
						i_tgtype,
						i_tgnargs,
2215 2216 2217 2218 2219
						i_tgargs,
						i_tgisconstraint,
						i_tgconstrname,
						i_tgdeferrable,
						i_tginitdeferred;
V
Vadim B. Mikheev 已提交
2220 2221
			int			ntups2;
			int			i2;
2222

V
Vadim B. Mikheev 已提交
2223
			if (g_verbose)
2224
				fprintf(stderr, "%s finding Triggers for relation: '%s' %s\n",
V
Vadim B. Mikheev 已提交
2225 2226 2227
						g_comment_start,
						tblinfo[i].relname,
						g_comment_end);
2228

B
Hi, all  
Bruce Momjian 已提交
2229
			resetPQExpBuffer(query);
2230
			appendPQExpBuffer(query, "SELECT tgname, tgfoid, tgtype, tgnargs, tgargs, tgisconstraint, tgconstrname, tgdeferrable, tginitdeferred, oid "
2231 2232 2233
							  "from pg_trigger "
							  "where tgrelid = '%s'::oid ",
							  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
2234
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
2235
			if (!res2 ||
2236
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2237
			{
B
Bruce Momjian 已提交
2238 2239
				fprintf(stderr, "getTables(): SELECT (for TRIGGER) failed.  "
							"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
V
Vadim B. Mikheev 已提交
2240 2241 2242 2243 2244
				exit_nicely(g_conn);
			}
			ntups2 = PQntuples(res2);
			if (ntups2 != tblinfo[i].ntrig)
			{
2245
				fprintf(stderr, "getTables(): relation '%s': %d Triggers were expected, but got %d\n",
2246
						tblinfo[i].relname, tblinfo[i].ntrig, ntups2);
V
Vadim B. Mikheev 已提交
2247 2248 2249 2250 2251 2252 2253
				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 已提交
2254
			i_tgoid = PQfnumber(res2, "oid");
2255 2256 2257 2258 2259
			i_tgisconstraint = PQfnumber(res2, "tgisconstraint");
			i_tgconstrname = PQfnumber(res2, "tgconstrname");
			i_tgdeferrable = PQfnumber(res2, "tgdeferrable");
			i_tginitdeferred = PQfnumber(res2, "tginitdeferred");

B
Bruce Momjian 已提交
2260
			tblinfo[i].triggers = (TrigInfo*) malloc(ntups2 * sizeof(TrigInfo));
B
Hi, all  
Bruce Momjian 已提交
2261 2262
			resetPQExpBuffer(query);
			for (i2 = 0; i2 < ntups2; i2++)
V
Vadim B. Mikheev 已提交
2263
			{
2264
				const char *tgfuncoid = PQgetvalue(res2, i2, i_tgfoid);
2265
				char	   *tgfunc = NULL;
2266 2267
				int2		tgtype = atoi(PQgetvalue(res2, i2, i_tgtype));
				int			tgnargs = atoi(PQgetvalue(res2, i2, i_tgnargs));
B
Bruce Momjian 已提交
2268
				const char *tgargs = PQgetvalue(res2, i2, i_tgargs);
2269 2270 2271
				int			tgisconstraint;
				int			tgdeferrable;
				int			tginitdeferred;
B
Bruce Momjian 已提交
2272
				const char *p;
2273 2274
				int			findx;

2275
				if (strcmp(PQgetvalue(res2, i2, i_tgisconstraint), "f") == 0)
2276
					tgisconstraint = 0;
2277
				else
2278
					tgisconstraint = 1;
2279 2280

				if (strcmp(PQgetvalue(res2, i2, i_tgdeferrable), "f") == 0)
2281
					tgdeferrable = 0;
2282
				else
2283
					tgdeferrable = 1;
2284 2285

				if (strcmp(PQgetvalue(res2, i2, i_tginitdeferred), "f") == 0)
2286
					tginitdeferred = 0;
2287
				else
2288
					tginitdeferred = 1;
2289

V
Vadim B. Mikheev 已提交
2290 2291
				for (findx = 0; findx < numFuncs; findx++)
				{
2292
					if (strcmp(finfo[findx].oid, tgfuncoid) == 0 &&
2293
						finfo[findx].nargs == 0 &&
V
Vadim B. Mikheev 已提交
2294 2295 2296
						strcmp(finfo[findx].prorettype, "0") == 0)
						break;
				}
2297

V
Vadim B. Mikheev 已提交
2298 2299
				if (findx == numFuncs)
				{
2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320
					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);
					}
2321
					tgfunc = strdup(PQgetvalue(r, 0, PQfnumber(r, "proname")));
2322
					PQclear(r);
2323
				}
2324
				else
2325
					tgfunc = strdup(finfo[findx].proname);
B
Bruce Momjian 已提交
2326 2327

				appendPQExpBuffer(delqry, "DROP TRIGGER %s ",
2328
									fmtId(PQgetvalue(res2, i2, i_tgname),
B
Bruce Momjian 已提交
2329 2330 2331
									force_quotes));
				appendPQExpBuffer(delqry, "ON %s;\n",
									fmtId(tblinfo[i].relname, force_quotes));
2332

B
Hi, all  
Bruce Momjian 已提交
2333
				resetPQExpBuffer(query);
2334 2335 2336 2337 2338
				if (tgisconstraint)
				{
					appendPQExpBuffer(query, "CREATE CONSTRAINT TRIGGER ");
					appendPQExpBuffer(query, fmtId(PQgetvalue(res2, i2, i_tgconstrname), force_quotes));
				}
2339 2340
				else
				{
2341 2342 2343
					appendPQExpBuffer(query, "CREATE TRIGGER ");
					appendPQExpBuffer(query, fmtId(PQgetvalue(res2, i2, i_tgname), force_quotes));
				}
B
Bruce Momjian 已提交
2344
				appendPQExpBufferChar(query, ' ');
V
Vadim B. Mikheev 已提交
2345 2346 2347
				/* Trigger type */
				findx = 0;
				if (TRIGGER_FOR_BEFORE(tgtype))
B
Hi, all  
Bruce Momjian 已提交
2348
					appendPQExpBuffer(query, "BEFORE");
V
Vadim B. Mikheev 已提交
2349
				else
B
Hi, all  
Bruce Momjian 已提交
2350
					appendPQExpBuffer(query, "AFTER");
V
Vadim B. Mikheev 已提交
2351 2352
				if (TRIGGER_FOR_INSERT(tgtype))
				{
B
Hi, all  
Bruce Momjian 已提交
2353
					appendPQExpBuffer(query, " INSERT");
V
Vadim B. Mikheev 已提交
2354 2355 2356 2357 2358
					findx++;
				}
				if (TRIGGER_FOR_DELETE(tgtype))
				{
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
2359
						appendPQExpBuffer(query, " OR DELETE");
V
Vadim B. Mikheev 已提交
2360
					else
B
Hi, all  
Bruce Momjian 已提交
2361
						appendPQExpBuffer(query, " DELETE");
V
Vadim B. Mikheev 已提交
2362 2363 2364
					findx++;
				}
				if (TRIGGER_FOR_UPDATE(tgtype))
2365
				{
V
Vadim B. Mikheev 已提交
2366
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
2367
						appendPQExpBuffer(query, " OR UPDATE");
V
Vadim B. Mikheev 已提交
2368
					else
B
Hi, all  
Bruce Momjian 已提交
2369
						appendPQExpBuffer(query, " UPDATE");
2370
				}
2371 2372
				appendPQExpBuffer(query, " ON %s ", fmtId(tblinfo[i].relname, force_quotes));

2373
				if (tgisconstraint)
2374
				{
2375
					if (!tgdeferrable)
2376 2377 2378 2379
						appendPQExpBuffer(query, " NOT");
					appendPQExpBuffer(query, " DEFERRABLE INITIALLY ");
					if (tginitdeferred)
						appendPQExpBuffer(query, "DEFERRED");
2380
					else
2381
						appendPQExpBuffer(query, "IMMEDIATE");
2382

2383 2384 2385
				}

				appendPQExpBuffer(query, " FOR EACH ROW");
2386 2387
				appendPQExpBuffer(query, " EXECUTE PROCEDURE %s (",
								  fmtId(tgfunc, force_quotes));
V
Vadim B. Mikheev 已提交
2388 2389
				for (findx = 0; findx < tgnargs; findx++)
				{
2390
					const char *s;
2391 2392

					for (p = tgargs;;)
V
Vadim B. Mikheev 已提交
2393
					{
2394
						p = strchr(p, '\\');
V
Vadim B. Mikheev 已提交
2395 2396
						if (p == NULL)
						{
B
Bruce Momjian 已提交
2397 2398
							fprintf(stderr, "getTables(): relation '%s': bad argument "
									"string (%s) for trigger '%s'\n",
2399 2400 2401
									tblinfo[i].relname,
									PQgetvalue(res2, i2, i_tgargs),
									PQgetvalue(res2, i2, i_tgname));
V
Vadim B. Mikheev 已提交
2402 2403 2404 2405 2406 2407 2408 2409
							exit_nicely(g_conn);
						}
						p++;
						if (*p == '\\')
						{
							p++;
							continue;
						}
2410
						if (p[0] == '0' && p[1] == '0' && p[2] == '0')
V
Vadim B. Mikheev 已提交
2411 2412 2413
							break;
					}
					p--;
2414
					appendPQExpBufferChar(query, '\'');
B
Bruce Momjian 已提交
2415
					for (s = tgargs; s < p;)
V
Vadim B. Mikheev 已提交
2416 2417
					{
						if (*s == '\'')
2418 2419
							appendPQExpBufferChar(query, '\\');
						appendPQExpBufferChar(query, *s++);
V
Vadim B. Mikheev 已提交
2420
					}
B
Bruce Momjian 已提交
2421 2422
					appendPQExpBufferChar(query, '\'');
					appendPQExpBuffer(query, (findx < tgnargs - 1) ? ", " : "");
V
Vadim B. Mikheev 已提交
2423 2424
					tgargs = p + 4;
				}
B
Hi, all  
Bruce Momjian 已提交
2425
				appendPQExpBuffer(query, ");\n");
2426

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

				/*** Initialize trcomments and troids ***/
2430

B
Bruce,  
Bruce Momjian 已提交
2431
				resetPQExpBuffer(query);
2432 2433
				appendPQExpBuffer(query, "TRIGGER %s ",
					fmtId(PQgetvalue(res2, i2, i_tgname), force_quotes));
B
Bruce,  
Bruce Momjian 已提交
2434
				appendPQExpBuffer(query, "ON %s",
2435
								fmtId(tblinfo[i].relname, force_quotes));
B
Bruce Momjian 已提交
2436 2437 2438 2439
				tblinfo[i].triggers[i2].tgcomment = strdup(query->data);
				tblinfo[i].triggers[i2].oid = strdup(PQgetvalue(res2, i2, i_tgoid));
				tblinfo[i].triggers[i2].tgname = strdup(fmtId(PQgetvalue(res2, i2, i_tgname),false));
				tblinfo[i].triggers[i2].tgdel = strdup(delqry->data);
B
Bruce,  
Bruce Momjian 已提交
2440

2441 2442
				if (tgfunc)
					free(tgfunc);
V
Vadim B. Mikheev 已提交
2443 2444 2445 2446
			}
			PQclear(res2);
		}
		else
B
Bruce,  
Bruce Momjian 已提交
2447
		{
V
Vadim B. Mikheev 已提交
2448
			tblinfo[i].triggers = NULL;
B
Bruce,  
Bruce Momjian 已提交
2449
		}
2450

2451 2452 2453 2454 2455
	}

	PQclear(res);

	return tblinfo;
2456 2457 2458 2459 2460

}

/*
 * getInherits
2461
 *	  read all the inheritance information
2462 2463
 * from the system catalogs return them in the InhInfo* structure
 *
2464 2465
 * numInherits is set to the number of tables read in
 *
2466 2467
 *
 */
2468
InhInfo    *
2469 2470
getInherits(int *numInherits)
{
2471
	PGresult   *res;
2472 2473
	int			ntups;
	int			i;
2474 2475
	PQExpBuffer query = createPQExpBuffer();
	InhInfo    *inhinfo;
2476

2477
	int			i_inhrelid;
2478
	int			i_inhparent;
2479 2480 2481

	/* find all the inheritance information */

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

B
Hi, all  
Bruce Momjian 已提交
2484
	res = PQexec(g_conn, query->data);
2485 2486 2487
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
2488 2489
		fprintf(stderr, "getInherits(): SELECT failed.  Explanation from backend: '%s'.\n", 
				PQerrorMessage(g_conn));
2490 2491 2492 2493 2494 2495 2496 2497 2498
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numInherits = ntups;

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

2499
	i_inhrelid = PQfnumber(res, "inhrelid");
2500 2501 2502 2503
	i_inhparent = PQfnumber(res, "inhparent");

	for (i = 0; i < ntups; i++)
	{
2504
		inhinfo[i].inhrelid = strdup(PQgetvalue(res, i, i_inhrelid));
2505 2506 2507 2508 2509
		inhinfo[i].inhparent = strdup(PQgetvalue(res, i, i_inhparent));
	}

	PQclear(res);
	return inhinfo;
2510 2511 2512 2513
}

/*
 * getTableAttrs -
2514 2515
 *	  for each table in tblinfo, read its attributes types and names
 *
2516
 * this is implemented in a very inefficient way right now, looping
2517
 * through the tblinfo and doing a join per table to find the attrs and their
2518 2519
 * types
 *
2520
 *	modifies tblinfo
2521 2522
 */
void
2523
getTableAttrs(TableInfo *tblinfo, int numTables)
2524
{
2525 2526
	int			i,
				j;
2527
	PQExpBuffer q = createPQExpBuffer();
2528 2529
	int			i_attname;
	int			i_typname;
2530
	int			i_atttypmod;
2531
	int			i_attnotnull;
V
Vadim B. Mikheev 已提交
2532
	int			i_atthasdef;
2533
	int			i_attoid;
2534
	int			i_atttypedefn;
2535
	PGresult   *res;
2536
	int			ntups;
2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551

	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)
2552
			fprintf(stderr, "%s finding the attrs and types for table: '%s' %s\n",
2553 2554 2555 2556
					g_comment_start,
					tblinfo[i].relname,
					g_comment_end);

B
Hi, all  
Bruce Momjian 已提交
2557
		resetPQExpBuffer(q);
B
Bruce,  
Bruce Momjian 已提交
2558
		appendPQExpBuffer(q, "SELECT a.oid as attoid, a.attnum, a.attname, t.typname, a.atttypmod, "
2559
						  "a.attnotnull, a.atthasdef, format_type(a.atttypid, a.atttypmod) as atttypedefn "
2560 2561 2562 2563
						  "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 已提交
2564
		res = PQexec(g_conn, q->data);
2565 2566 2567
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
B
Bruce Momjian 已提交
2568 2569
			fprintf(stderr, "getTableAttrs(): SELECT failed.  "
					"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2570 2571 2572 2573 2574
			exit_nicely(g_conn);
		}

		ntups = PQntuples(res);

B
Bruce,  
Bruce Momjian 已提交
2575
		i_attoid = PQfnumber(res, "attoid");
2576 2577
		i_attname = PQfnumber(res, "attname");
		i_typname = PQfnumber(res, "typname");
2578
		i_atttypmod = PQfnumber(res, "atttypmod");
2579
		i_attnotnull = PQfnumber(res, "attnotnull");
V
Vadim B. Mikheev 已提交
2580
		i_atthasdef = PQfnumber(res, "atthasdef");
2581
		i_atttypedefn = PQfnumber(res, "atttypedefn");
2582 2583

		tblinfo[i].numatts = ntups;
B
Bruce,  
Bruce Momjian 已提交
2584
		tblinfo[i].attoids = (char **) malloc(ntups * sizeof(char *));
2585
		tblinfo[i].attnames = (char **) malloc(ntups * sizeof(char *));
2586
		tblinfo[i].atttypedefns = (char **) malloc(ntups * sizeof(char *));
2587
		tblinfo[i].typnames = (char **) malloc(ntups * sizeof(char *));
2588
		tblinfo[i].atttypmod = (int *) malloc(ntups * sizeof(int));
2589 2590
		tblinfo[i].inhAttrs = (int *) malloc(ntups * sizeof(int));
		tblinfo[i].notnull = (bool *) malloc(ntups * sizeof(bool));
V
Vadim B. Mikheev 已提交
2591
		tblinfo[i].adef_expr = (char **) malloc(ntups * sizeof(char *));
2592 2593 2594 2595
		tblinfo[i].parentRels = NULL;
		tblinfo[i].numParents = 0;
		for (j = 0; j < ntups; j++)
		{
B
Bruce,  
Bruce Momjian 已提交
2596
			tblinfo[i].attoids[j] = strdup(PQgetvalue(res, j, i_attoid));
2597
			tblinfo[i].attnames[j] = strdup(PQgetvalue(res, j, i_attname));
2598
			tblinfo[i].atttypedefns[j] = strdup(PQgetvalue(res, j, i_atttypedefn));
2599
			tblinfo[i].typnames[j] = strdup(PQgetvalue(res, j, i_typname));
2600
			tblinfo[i].atttypmod[j] = atoi(PQgetvalue(res, j, i_atttypmod));
2601 2602 2603
			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 已提交
2604 2605 2606
			if (PQgetvalue(res, j, i_atthasdef)[0] == 't')
			{
				PGresult   *res2;
2607

V
Vadim B. Mikheev 已提交
2608
				if (g_verbose)
2609
					fprintf(stderr, "%s finding DEFAULT expression for attr: '%s' %s\n",
V
Vadim B. Mikheev 已提交
2610 2611 2612
							g_comment_start,
							tblinfo[i].attnames[j],
							g_comment_end);
2613

B
Hi, all  
Bruce Momjian 已提交
2614 2615
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "SELECT adsrc from pg_attrdef "
2616 2617
							 "where adrelid = '%s'::oid and adnum = %d ",
								  tblinfo[i].oid, j + 1);
B
Hi, all  
Bruce Momjian 已提交
2618
				res2 = PQexec(g_conn, q->data);
V
Vadim B. Mikheev 已提交
2619
				if (!res2 ||
2620
					PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2621
				{
B
Bruce Momjian 已提交
2622 2623
					fprintf(stderr, "getTableAttrs(): SELECT (for DEFAULT) failed.  "
							"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
V
Vadim B. Mikheev 已提交
2624 2625 2626 2627 2628 2629 2630
					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;
2631 2632 2633
		}
		PQclear(res);
	}
2634 2635 2636 2637 2638
}


/*
 * getIndices
2639
 *	  read all the user-defined indices information
2640 2641
 * from the system catalogs return them in the InhInfo* structure
 *
2642 2643
 * numIndices is set to the number of indices read in
 *
2644 2645
 *
 */
2646
IndInfo    *
2647 2648
getIndices(int *numIndices)
{
2649
	int			i;
2650
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
2651
	PGresult   *res;
2652
	int			ntups;
B
Bruce Momjian 已提交
2653
	IndInfo    *indinfo;
2654 2655 2656 2657 2658 2659 2660 2661

	int			i_indexrelname;
	int			i_indrelname;
	int			i_indamname;
	int			i_indproc;
	int			i_indkey;
	int			i_indclass;
	int			i_indisunique;
2662
	int			i_indoid;
2663 2664 2665 2666 2667

	/*
	 * find all the user-defined indices. We do not handle partial
	 * indices.
	 *
2668
	 * Notice we skip indices on system classes
2669 2670 2671 2672
	 *
	 * this is a 4-way join !!
	 */

B
Hi, all  
Bruce Momjian 已提交
2673
	appendPQExpBuffer(query,
2674 2675 2676 2677 2678 2679
					  "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 "
2680 2681
					  "and t2.relname !~ '^pg_' and not i.indisprimary",
					  g_last_builtin_oid);
2682

B
Hi, all  
Bruce Momjian 已提交
2683
	res = PQexec(g_conn, query->data);
2684 2685 2686
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
2687 2688
		fprintf(stderr, "getIndices(): SELECT failed.  "
					"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2689 2690 2691 2692 2693 2694 2695 2696 2697
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numIndices = ntups;

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

2698 2699
	memset((char *) indinfo, 0, ntups * sizeof(IndInfo));

B
Bruce,  
Bruce Momjian 已提交
2700
	i_indoid = PQfnumber(res, "indoid");
2701 2702 2703 2704 2705 2706 2707 2708 2709 2710
	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 已提交
2711
		indinfo[i].indoid = strdup(PQgetvalue(res, i, i_indoid));
2712 2713 2714 2715
		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));
2716 2717 2718 2719 2720 2721
		parseNumericArray(PQgetvalue(res, i, i_indkey),
						  indinfo[i].indkey,
						  INDEX_MAX_KEYS);
		parseNumericArray(PQgetvalue(res, i, i_indclass),
						  indinfo[i].indclass,
						  INDEX_MAX_KEYS);
2722 2723 2724 2725
		indinfo[i].indisunique = strdup(PQgetvalue(res, i, i_indisunique));
	}
	PQclear(res);
	return indinfo;
2726 2727
}

B
Bruce,  
Bruce Momjian 已提交
2728
/*------------------------------------------------------------------
2729
 * dumpComments --
B
Bruce,  
Bruce Momjian 已提交
2730
 *
2731
 * This routine is used to dump any comments associated with the
B
Bruce,  
Bruce Momjian 已提交
2732 2733 2734 2735
 * 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
2736
 * addition, the routine takes the stdio FILE handle to which the
B
Bruce,  
Bruce Momjian 已提交
2737 2738 2739 2740
 * output should be written.
 *------------------------------------------------------------------
*/

T
Tom Lane 已提交
2741
static void
B
Bruce Momjian 已提交
2742
dumpComment(Archive *fout, const char *target, const char *oid)
2743
{
B
Bruce,  
Bruce Momjian 已提交
2744

2745
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
2746
	PQExpBuffer query;
2747
	int			i_description;
B
Bruce,  
Bruce Momjian 已提交
2748 2749 2750 2751 2752 2753 2754 2755 2756 2757

	/*** 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);
2758 2759
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce,  
Bruce Momjian 已提交
2760 2761 2762 2763 2764 2765 2766
		fprintf(stderr, "DumpComment: SELECT failed: '%s'.\n",
				PQerrorMessage(g_conn));
		exit_nicely(g_conn);
	}

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

2767 2768
	if (PQntuples(res) != 0)
	{
B
Bruce,  
Bruce Momjian 已提交
2769
		i_description = PQfnumber(res, "description");
B
Bruce Momjian 已提交
2770
		resetPQExpBuffer(query);
2771 2772 2773
		appendPQExpBuffer(query, "COMMENT ON %s IS ", target);
		formatStringLiteral(query, PQgetvalue(res, 0, i_description));
		appendPQExpBuffer(query, ";\n");
B
Bruce Momjian 已提交
2774 2775

		ArchiveEntry(fout, oid, target, "COMMENT", NULL, query->data, "" /*Del*/,
2776
					   "" /* Copy */, "" /*Owner*/, NULL, NULL);	
B
Bruce Momjian 已提交
2777

B
Bruce,  
Bruce Momjian 已提交
2778 2779 2780 2781 2782
	}

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

	PQclear(res);
2783

B
Bruce,  
Bruce Momjian 已提交
2784 2785 2786
}

/*------------------------------------------------------------------
2787
 * dumpDBComment --
B
Bruce,  
Bruce Momjian 已提交
2788
 *
2789 2790
 * 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 已提交
2791 2792 2793 2794 2795
 * to dump the schema of the database, then this is the first
 * statement issued.
 *------------------------------------------------------------------
*/

2796
void
B
Bruce Momjian 已提交
2797
dumpDBComment(Archive *fout)
2798
{
B
Bruce,  
Bruce Momjian 已提交
2799

2800
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
2801
	PQExpBuffer query;
2802
	int			i_oid;
B
Bruce,  
Bruce Momjian 已提交
2803 2804 2805 2806

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

	query = createPQExpBuffer();
2807 2808
	appendPQExpBuffer(query, "SELECT oid FROM pg_database WHERE datname = ");
	formatStringLiteral(query, PQdb(g_conn));
B
Bruce,  
Bruce Momjian 已提交
2809 2810 2811 2812

	/*** Execute query ***/

	res = PQexec(g_conn, query->data);
2813 2814
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce,  
Bruce Momjian 已提交
2815 2816 2817 2818 2819 2820 2821
		fprintf(stderr, "dumpDBComment: SELECT failed: '%s'.\n",
				PQerrorMessage(g_conn));
		exit_nicely(g_conn);
	}

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

2822 2823
	if (PQntuples(res) != 0)
	{
B
Bruce,  
Bruce Momjian 已提交
2824 2825 2826 2827 2828 2829 2830 2831 2832
		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);
2833

B
Bruce,  
Bruce Momjian 已提交
2834 2835
}

2836 2837
/*
 * dumpTypes
2838
 *	  writes out to fout the queries to recreate all the user-defined types
2839 2840 2841
 *
 */
void
B
Bruce Momjian 已提交
2842
dumpTypes(Archive *fout, FuncInfo *finfo, int numFuncs,
2843
		  TypeInfo *tinfo, int numTypes)
2844
{
2845
	int			i;
2846
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
2847
	PQExpBuffer	delq = createPQExpBuffer();
2848
	int			funcInd;
2849 2850 2851 2852 2853

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

		/* skip all the builtin types */
2854
		if (atooid(tinfo[i].oid) <= g_last_builtin_oid)
2855 2856 2857 2858 2859 2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877
			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);

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

B
Hi, all  
Bruce Momjian 已提交
2880 2881
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
2882
						  "CREATE TYPE %s "
2883
						  "( internallength = %s, externallength = %s,",
2884 2885
						  fmtId(tinfo[i].typname, force_quotes),
						  tinfo[i].typlen,
2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896
						  tinfo[i].typprtlen);
		/* cannot combine these because fmtId uses static result area */
		appendPQExpBuffer(q, " input = %s,",
						  fmtId(tinfo[i].typinput, force_quotes));
		appendPQExpBuffer(q, " output = %s,",
						  fmtId(tinfo[i].typoutput, force_quotes));
		appendPQExpBuffer(q, " send = %s,",
						  fmtId(tinfo[i].typsend, force_quotes));
		appendPQExpBuffer(q, " receive = %s, default = ",
						  fmtId(tinfo[i].typreceive, force_quotes));
		formatStringLiteral(q, tinfo[i].typdefault);
2897 2898 2899

		if (tinfo[i].isArray)
		{
2900
			char	   *elemType;
2901

2902
			elemType = findTypeByOid(tinfo, numTypes, tinfo[i].typelem, zeroAsOpaque);
2903 2904
			appendPQExpBuffer(q, ", element = %s, delimiter = ", elemType);
			formatStringLiteral(q, tinfo[i].typdelim);
2905 2906
		}
		if (tinfo[i].passedbyvalue)
B
Hi, all  
Bruce Momjian 已提交
2907
			appendPQExpBuffer(q, ",passedbyvalue);\n");
2908
		else
B
Hi, all  
Bruce Momjian 已提交
2909
			appendPQExpBuffer(q, ");\n");
2910

B
Bruce Momjian 已提交
2911
		ArchiveEntry(fout, tinfo[i].oid, fmtId(tinfo[i].typname, force_quotes), "TYPE", NULL,
2912
						q->data, delq->data, "", tinfo[i].usename, NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
2913 2914 2915 2916

		/*** Dump Type Comments ***/

		resetPQExpBuffer(q);
B
Bruce Momjian 已提交
2917 2918
		resetPQExpBuffer(delq);

B
Bruce,  
Bruce Momjian 已提交
2919 2920 2921
		appendPQExpBuffer(q, "TYPE %s", fmtId(tinfo[i].typname, force_quotes));
		dumpComment(fout, q->data, tinfo[i].oid);

B
Bruce Momjian 已提交
2922
		resetPQExpBuffer(q);
2923
	}
2924 2925
}

2926 2927
/*
 * dumpProcLangs
B
Bruce Momjian 已提交
2928
 *		  writes out to fout the queries to recreate user-defined procedural languages
2929 2930 2931
 *
 */
void
B
Bruce Momjian 已提交
2932
dumpProcLangs(Archive *fout, FuncInfo *finfo, int numFuncs,
B
Bruce Momjian 已提交
2933
			  TypeInfo *tinfo, int numTypes)
2934
{
2935 2936
	PGresult   *res;
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
2937 2938
	PQExpBuffer defqry = createPQExpBuffer();
	PQExpBuffer delqry = createPQExpBuffer();
2939
	int			ntups;
B
Bruce Momjian 已提交
2940
	int			i_oid;
2941 2942 2943 2944
	int			i_lanname;
	int			i_lanpltrusted;
	int			i_lanplcallfoid;
	int			i_lancompiler;
2945
	Oid			lanoid;
2946 2947 2948
	char	   *lanname;
	char	   *lancompiler;
	const char *lanplcallfoid;
2949 2950 2951
	int			i,
				fidx;

B
Bruce Momjian 已提交
2952
	appendPQExpBuffer(query, "SELECT oid, * FROM pg_language "
2953 2954
					  "WHERE lanispl "
					  "ORDER BY oid");
B
Hi, all  
Bruce Momjian 已提交
2955
	res = PQexec(g_conn, query->data);
2956 2957 2958
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
2959
		fprintf(stderr, "dumpProcLangs(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2960 2961 2962 2963
		exit_nicely(g_conn);
	}
	ntups = PQntuples(res);

B
Bruce Momjian 已提交
2964 2965 2966 2967
	i_lanname = PQfnumber(res, "lanname");
	i_lanpltrusted = PQfnumber(res, "lanpltrusted");
	i_lanplcallfoid = PQfnumber(res, "lanplcallfoid");
	i_lancompiler = PQfnumber(res, "lancompiler");
B
Bruce Momjian 已提交
2968
	i_oid = PQfnumber(res, "oid");
2969

B
Bruce Momjian 已提交
2970 2971
	for (i = 0; i < ntups; i++)
	{
2972
		lanoid = atooid(PQgetvalue(res, i, i_oid));
2973 2974 2975
		if (lanoid <= g_last_builtin_oid)
			continue;

2976
		lanplcallfoid = PQgetvalue(res, i, i_lanplcallfoid);
2977 2978


2979 2980 2981 2982 2983 2984 2985
		for (fidx = 0; fidx < numFuncs; fidx++)
		{
			if (!strcmp(finfo[fidx].oid, lanplcallfoid))
				break;
		}
		if (fidx >= numFuncs)
		{
B
Bruce Momjian 已提交
2986 2987
			fprintf(stderr, "dumpProcLangs(): handler procedure for "
						"language %s not found\n", PQgetvalue(res, i, i_lanname));
2988 2989 2990 2991 2992
			exit_nicely(g_conn);
		}

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

2993 2994
		lanname = PQgetvalue(res, i, i_lanname);
		lancompiler = PQgetvalue(res, i, i_lancompiler);
2995

2996 2997 2998
		appendPQExpBuffer(delqry, "DROP PROCEDURAL LANGUAGE ");
		formatStringLiteral(delqry, lanname);
		appendPQExpBuffer(delqry, ";\n");
2999

3000 3001 3002 3003 3004 3005 3006 3007
		appendPQExpBuffer(defqry, "CREATE %sPROCEDURAL LANGUAGE ",
						  (PQgetvalue(res, i, i_lanpltrusted)[0] == 't') ?
						  "TRUSTED " : "");
		formatStringLiteral(defqry, lanname);
		appendPQExpBuffer(defqry, " HANDLER %s LANCOMPILER ",
						  fmtId(finfo[fidx].proname, force_quotes));
		formatStringLiteral(defqry, lancompiler);
		appendPQExpBuffer(defqry, ";\n");
3008

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

3012 3013
		resetPQExpBuffer(defqry);
		resetPQExpBuffer(delqry);
3014 3015 3016 3017 3018 3019
	}

	PQclear(res);

}

3020 3021
/*
 * dumpFuncs
3022
 *	  writes out to fout the queries to recreate all the user-defined functions
3023 3024 3025
 *
 */
void
B
Bruce Momjian 已提交
3026
dumpFuncs(Archive *fout, FuncInfo *finfo, int numFuncs,
3027
		  TypeInfo *tinfo, int numTypes)
3028
{
3029
	int			i;
3030 3031 3032

	for (i = 0; i < numFuncs; i++)
		dumpOneFunc(fout, finfo, i, tinfo, numTypes);
3033 3034 3035 3036
}

/*
 * dumpOneFunc:
3037 3038
 *	  dump out only one function,  the index of which is given in the third
 *	argument
3039 3040 3041
 *
 */

3042
static void
B
Bruce Momjian 已提交
3043
dumpOneFunc(Archive *fout, FuncInfo *finfo, int i,
3044
			TypeInfo *tinfo, int numTypes)
3045
{
3046
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3047 3048
	PQExpBuffer fn = createPQExpBuffer();
	PQExpBuffer delqry = createPQExpBuffer();
B
Bruce,  
Bruce Momjian 已提交
3049
	PQExpBuffer fnlist = createPQExpBuffer();
B
Bruce Momjian 已提交
3050
	int			j;
3051
	PQExpBuffer asPart = createPQExpBuffer();
3052
	char		func_lang[NAMEDATALEN + 1];
3053 3054 3055 3056
	PGresult   *res;
	int			nlangs;
	int			i_lanname;
	char		query[256];
3057

3058 3059 3060 3061
	char		*listSep;
	char		*listSepComma = ",";
	char		*listSepNone = "";

3062 3063 3064 3065 3066
	if (finfo[i].dumped)
		return;
	else
		finfo[i].dumped = 1;

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

3069 3070 3071 3072 3073
	sprintf(query, "SELECT lanname FROM pg_language WHERE oid = %u",
			finfo[i].lang);
	res = PQexec(g_conn, query);
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
B
Bruce Momjian 已提交
3074
  	{
3075 3076
		fprintf(stderr, "dumpOneFunc(): SELECT for procedural language failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
		exit_nicely(g_conn);
B
Bruce Momjian 已提交
3077
  	}
3078 3079 3080
	nlangs = PQntuples(res);

	if (nlangs != 1)
B
Bruce Momjian 已提交
3081
  	{
3082 3083
		fprintf(stderr, "dumpOneFunc(): procedural language for function %s not found\n", finfo[i].proname);
		exit_nicely(g_conn);
B
Bruce Momjian 已提交
3084 3085
  	}
  
3086
	i_lanname = PQfnumber(res, "lanname");
3087

3088
	/*
3089 3090 3091 3092 3093
	 * See backend/commands/define.c for details of how the 'AS' clause
	 * is used.
	 */
	if (strcmp(finfo[i].probin, "-") != 0)
	{
3094 3095
		appendPQExpBuffer(asPart, "AS ");
		formatStringLiteral(asPart, finfo[i].probin);
3096
		if (strcmp(finfo[i].prosrc, "-") != 0)
3097 3098 3099 3100
		{
			appendPQExpBuffer(asPart, ", ");
			formatStringLiteral(asPart, finfo[i].prosrc);
		}
3101 3102 3103 3104
	}
	else
	{
		if (strcmp(finfo[i].prosrc, "-") != 0)
3105 3106 3107 3108
		{
			appendPQExpBuffer(asPart, "AS ");
			formatStringLiteral(asPart, finfo[i].prosrc);
		}
3109 3110
	}

3111
	strcpy(func_lang, PQgetvalue(res, 0, i_lanname));
B
Bruce Momjian 已提交
3112
  
3113
	PQclear(res);
B
Bruce Momjian 已提交
3114 3115 3116
 
	resetPQExpBuffer(fn);
	appendPQExpBuffer(fn, "%s (", fmtId(finfo[i].proname, force_quotes));
3117 3118
	for (j = 0; j < finfo[i].nargs; j++)
	{
B
Bruce Momjian 已提交
3119
		char			*typname;
3120

3121 3122 3123 3124
		typname = findTypeByOid(tinfo, numTypes, finfo[i].argtypes[j], zeroAsOpaque);
		appendPQExpBuffer(fn, "%s%s", 
							(j > 0) ? "," : "", 
							typname);
B
Bruce,  
Bruce Momjian 已提交
3125
		appendPQExpBuffer(fnlist, "%s%s",
3126 3127
							(j > 0) ? "," : "", 
							typname );
3128
	}
B
Bruce Momjian 已提交
3129 3130 3131 3132 3133 3134 3135
	appendPQExpBuffer(fn, ")");

	resetPQExpBuffer(delqry);
	appendPQExpBuffer(delqry, "DROP FUNCTION %s;\n", fn->data );

	resetPQExpBuffer(q);
	appendPQExpBuffer(q, "CREATE FUNCTION %s ", fn->data );
3136 3137
	appendPQExpBuffer(q, "RETURNS %s%s %s LANGUAGE ",
					  (finfo[i].retset) ? "SETOF " : "",
3138
					  findTypeByOid(tinfo, numTypes, finfo[i].prorettype, zeroAsOpaque),
3139 3140
					  asPart->data);
	formatStringLiteral(q, func_lang);
3141

3142
	if (finfo[i].iscachable || finfo[i].isstrict) /* OR in new attrs here */
3143
	{
3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156
		appendPQExpBuffer(q, " WITH (");
		listSep = listSepNone;

		if (finfo[i].iscachable) {
			appendPQExpBuffer(q, "%s iscachable", listSep);
			listSep = listSepComma;
		}

		if (finfo[i].isstrict) {
			appendPQExpBuffer(q, "%s isstrict", listSep);
			listSep = listSepComma;
		}
		appendPQExpBuffer(q, " )");
3157 3158 3159 3160
	}

	appendPQExpBuffer(q, ";\n");

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

B
Bruce,  
Bruce Momjian 已提交
3164 3165 3166 3167 3168 3169 3170 3171
	/*** 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);

3172 3173 3174 3175
}

/*
 * dumpOprs
3176
 *	  writes out to fout the queries to recreate all the user-defined operators
3177 3178
 *
 */
3179
void
B
Bruce Momjian 已提交
3180
dumpOprs(Archive *fout, OprInfo *oprinfo, int numOperators,
3181
		 TypeInfo *tinfo, int numTypes)
3182
{
B
Bruce Momjian 已提交
3183
	int			i;
3184
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3185
	PQExpBuffer delq = createPQExpBuffer();
3186 3187 3188 3189 3190 3191 3192 3193
	PQExpBuffer leftarg = createPQExpBuffer();
	PQExpBuffer rightarg = createPQExpBuffer();
	PQExpBuffer commutator = createPQExpBuffer();
	PQExpBuffer negator = createPQExpBuffer();
	PQExpBuffer restrictor = createPQExpBuffer();
	PQExpBuffer join = createPQExpBuffer();
	PQExpBuffer sort1 = createPQExpBuffer();
	PQExpBuffer sort2 = createPQExpBuffer();
3194 3195 3196 3197

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

B
Hi, all  
Bruce Momjian 已提交
3198 3199 3200 3201 3202 3203 3204 3205 3206
		resetPQExpBuffer(leftarg);
		resetPQExpBuffer(rightarg);
		resetPQExpBuffer(commutator);
		resetPQExpBuffer(negator);
		resetPQExpBuffer(restrictor);
		resetPQExpBuffer(join);
		resetPQExpBuffer(sort1);
		resetPQExpBuffer(sort2);

3207
		/* skip all the builtin oids */
3208
		if (atooid(oprinfo[i].oid) <= g_last_builtin_oid)
3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224
			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 已提交
3225
			appendPQExpBuffer(leftarg, ",\n\tLEFTARG = %s ",
3226
								findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft, zeroAsOpaque) );
3227 3228 3229 3230
		}
		if (strcmp(oprinfo[i].oprkind, "l") == 0 ||
			strcmp(oprinfo[i].oprkind, "b") == 0)
		{
B
Hi, all  
Bruce Momjian 已提交
3231
			appendPQExpBuffer(rightarg, ",\n\tRIGHTARG = %s ",
3232
							  findTypeByOid(tinfo, numTypes, oprinfo[i].oprright, zeroAsOpaque) );
3233
		}
B
Hi, all  
Bruce Momjian 已提交
3234 3235
		if (!(strcmp(oprinfo[i].oprcom, "0") == 0))
			appendPQExpBuffer(commutator, ",\n\tCOMMUTATOR = %s ",
3236 3237
				 findOprByOid(oprinfo, numOperators, oprinfo[i].oprcom));

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

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

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

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

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

B
Bruce Momjian 已提交
3256 3257
		resetPQExpBuffer(delq);
		appendPQExpBuffer(delq, "DROP OPERATOR %s (%s", oprinfo[i].oprname,
3258
				findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft, zeroAsOpaque) );
B
Bruce Momjian 已提交
3259
		appendPQExpBuffer(delq, ", %s);\n",
3260
					findTypeByOid(tinfo, numTypes, oprinfo[i].oprright, zeroAsOpaque) );
3261

B
Hi, all  
Bruce Momjian 已提交
3262 3263
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
3264 3265 3266 3267 3268 3269 3270 3271 3272
						  "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 已提交
3273
						(strcmp(oprinfo[i].oprcanhash, "t") == 0) ? ",\n\tHASHES" : "",
3274 3275 3276
						  join->data,
						  sort1->data,
						  sort2->data);
3277

B
Bruce Momjian 已提交
3278
		ArchiveEntry(fout, oprinfo[i].oid, oprinfo[i].oprname, "OPERATOR", NULL,
3279
						q->data, delq->data, "", oprinfo[i].usename, NULL, NULL);
3280
	}
3281 3282 3283 3284
}

/*
 * dumpAggs
3285
 *	  writes out to fout the queries to create all the user-defined aggregates
3286 3287 3288
 *
 */
void
B
Bruce Momjian 已提交
3289
dumpAggs(Archive *fout, AggInfo *agginfo, int numAggs,
3290
		 TypeInfo *tinfo, int numTypes)
3291
{
B
Bruce Momjian 已提交
3292
	int			i;
3293
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3294 3295
	PQExpBuffer delq = createPQExpBuffer();
	PQExpBuffer aggSig = createPQExpBuffer();
3296
	PQExpBuffer details = createPQExpBuffer();
3297 3298 3299

	for (i = 0; i < numAggs; i++)
	{
3300
		resetPQExpBuffer(details);
B
Hi, all  
Bruce Momjian 已提交
3301

3302
		/* skip all the builtin oids */
3303
		if (atooid(agginfo[i].oid) <= g_last_builtin_oid)
3304
			continue;
3305

3306
		appendPQExpBuffer(details,
3307
						  "BASETYPE = %s, ",
3308
						  findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsAny + useBaseTypeName));
3309

3310 3311 3312
		appendPQExpBuffer(details,
						  "SFUNC = %s, STYPE = %s",
						  agginfo[i].aggtransfn,
3313
						  findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype, zeroAsOpaque + useBaseTypeName));
3314

3315
		if (agginfo[i].agginitval)
3316 3317 3318 3319
		{
			appendPQExpBuffer(details, ", INITCOND = ");
			formatStringLiteral(details, agginfo[i].agginitval);
		}
3320

B
Hi, all  
Bruce Momjian 已提交
3321
		if (!(strcmp(agginfo[i].aggfinalfn, "-") == 0))
3322 3323
			appendPQExpBuffer(details, ", FINALFUNC = %s",
							  agginfo[i].aggfinalfn);
3324

B
Bruce Momjian 已提交
3325 3326
		resetPQExpBuffer(aggSig);
		appendPQExpBuffer(aggSig, "%s %s", agginfo[i].aggname,
3327
							findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsOpaque + useBaseTypeName));
3328

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

B
Hi, all  
Bruce Momjian 已提交
3332
		resetPQExpBuffer(q);
3333
		appendPQExpBuffer(q, "CREATE AGGREGATE %s ( %s );\n",
3334
						  agginfo[i].aggname,
3335
						  details->data);
3336

B
Bruce Momjian 已提交
3337
		ArchiveEntry(fout, agginfo[i].oid, aggSig->data, "AGGREGATE", NULL,
3338
						q->data, delq->data, "", agginfo[i].usename, NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
3339 3340 3341 3342

		/*** Dump Aggregate Comments ***/

		resetPQExpBuffer(q);
3343
		appendPQExpBuffer(q, "AGGREGATE %s %s", agginfo[i].aggname,
3344
						  findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsOpaque + useBaseTypeName));
B
Bruce,  
Bruce Momjian 已提交
3345 3346
		dumpComment(fout, q->data, agginfo[i].oid);

3347
	}
3348 3349
}

3350 3351 3352 3353 3354
/*
 * These are some support functions to fix the acl problem of pg_dump
 *
 * Matthew C. Aycock 12/02/97
 */
3355 3356 3357

/* Append a keyword to a keyword list, inserting comma if needed.
 * Caller must make aclbuf big enough for all possible keywords.
3358
 */
3359
static void
B
Bruce Momjian 已提交
3360
AddAcl(char *aclbuf, const char *keyword)
3361
{
3362 3363 3364
	if (*aclbuf)
		strcat(aclbuf, ",");
	strcat(aclbuf, keyword);
3365
}
3366

3367
/*
3368
 * This will take a string of 'arwR' and return a malloced,
3369 3370
 * comma delimited string of SELECT,INSERT,UPDATE,DELETE,RULE
 */
V
Vadim B. Mikheev 已提交
3371
static char *
3372
GetPrivileges(const char *s)
3373
{
3374
	char		aclbuf[100];
3375

3376
	aclbuf[0] = '\0';
3377

B
Bruce Momjian 已提交
3378
	if (strchr(s, 'a'))
3379
		AddAcl(aclbuf, "INSERT");
3380

B
Bruce Momjian 已提交
3381
	if (strchr(s, 'w'))
3382
		AddAcl(aclbuf, "UPDATE,DELETE");
B
Bruce Momjian 已提交
3383 3384

	if (strchr(s, 'r'))
3385
		AddAcl(aclbuf, "SELECT");
3386

3387
	if (strchr(s, 'R'))
3388
		AddAcl(aclbuf, "RULE");
3389

3390 3391 3392
	/* Special-case when they're all there */
	if (strcmp(aclbuf, "INSERT,UPDATE,DELETE,SELECT,RULE") == 0)
		return strdup("ALL");
3393

3394
	return strdup(aclbuf);
3395
}
3396

B
Bruce Momjian 已提交
3397 3398 3399 3400 3401 3402 3403 3404 3405 3406 3407 3408 3409 3410 3411 3412
/*
 * The name says it all; a function to append a string is the dest 
 * is big enough. If not, it does a realloc.
 */
static void strcatalloc(char **dest, int *dSize, char *src)
{
    int dLen = strlen(*dest);
    int sLen = strlen(src);
    if ( (dLen + sLen) >= *dSize) {
	*dSize = (dLen + sLen) * 2;
	*dest = realloc(*dest, *dSize);
    }
    strcpy(*dest + dLen, src);
} 


B
Bruce Momjian 已提交
3413 3414
/*
 * dumpACL:
3415 3416
 *	  Write out grant/revoke information
 *	  Called for sequences and tables
B
Bruce Momjian 已提交
3417 3418
 */

3419
static void
B
Bruce Momjian 已提交
3420
dumpACL(Archive *fout, TableInfo tbinfo)
B
Bruce Momjian 已提交
3421
{
B
Bruce Momjian 已提交
3422 3423
	const char 	*acls = tbinfo.relacl;
	char	   	*aclbuf,
B
Bruce Momjian 已提交
3424 3425 3426
			   *tok,
			   *eqpos,
			   *priv;
B
Bruce Momjian 已提交
3427 3428 3429
	char		*sql;
	char		tmp[1024];
	int		sSize = 4096;
3430 3431 3432 3433

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

B
Bruce Momjian 已提交
3434 3435 3436 3437 3438
	/*
	 * Allocate a larginsh buffer for the output SQL.
	 */
	sql = (char*)malloc(sSize);

B
Bruce Momjian 已提交
3439 3440 3441
	/*
	 * Revoke Default permissions for PUBLIC. Is this actually necessary,
	 * or is it just a waste of time?
3442
	 */
B
Bruce Momjian 已提交
3443
	sprintf(sql,	"REVOKE ALL on %s from PUBLIC;\n",
3444
			fmtId(tbinfo.relname, force_quotes));
3445 3446 3447

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

3449 3450
	/* Scan comma-separated ACL items */
	for (tok = strtok(aclbuf, ","); tok != NULL; tok = strtok(NULL, ","))
3451
	{
3452

B
Bruce Momjian 已提交
3453 3454 3455
		/*
		 * Token may start with '{' and/or '"'.  Actually only the start
		 * of the string should have '{', but we don't verify that.
3456 3457 3458 3459 3460 3461 3462 3463
		 */
		if (*tok == '{')
			tok++;
		if (*tok == '"')
			tok++;

		/* User name is string up to = in tok */
		eqpos = strchr(tok, '=');
B
Bruce Momjian 已提交
3464
		if (!eqpos)
B
Bruce Momjian 已提交
3465
		{
3466 3467
			fprintf(stderr, "Could not parse ACL list ('%s') for '%s'...Exiting!\n",
					acls, tbinfo.relname);
B
Bruce Momjian 已提交
3468 3469 3470
			exit_nicely(g_conn);
		}

B
Bruce Momjian 已提交
3471 3472 3473 3474
		/*
		 * Parse the privileges (right-hand side).	Skip if there are
		 * none.
		 */
3475 3476
		priv = GetPrivileges(eqpos + 1);
		if (*priv)
3477
		{
B
Bruce Momjian 已提交
3478
			sprintf(tmp,	"GRANT %s on %s to ",
3479
					priv, fmtId(tbinfo.relname, force_quotes));
B
Bruce Momjian 已提交
3480
			strcatalloc(&sql, &sSize, tmp);
B
Bruce Momjian 已提交
3481 3482 3483

			/*
			 * Note: fmtId() can only be called once per printf, so don't
3484 3485 3486 3487 3488
			 * try to merge printing of username into the above printf.
			 */
			if (eqpos == tok)
			{
				/* Empty left-hand side means "PUBLIC" */
B
Bruce Momjian 已提交
3489
				strcatalloc(&sql, &sSize, "PUBLIC;\n");
3490
			}
3491
			else
3492 3493
			{
				*eqpos = '\0';	/* it's ok to clobber aclbuf */
3494
				if (strncmp(tok, "group ", strlen("group ")) == 0)
B
Bruce Momjian 已提交
3495
					sprintf(tmp, "GROUP %s;\n",
3496 3497
							fmtId(tok + strlen("group "), force_quotes));
				else
B
Bruce Momjian 已提交
3498 3499
					sprintf(tmp, "%s;\n", fmtId(tok, force_quotes));
				strcatalloc(&sql, &sSize, tmp);
3500
			}
3501
		}
3502
		free(priv);
B
Bruce Momjian 已提交
3503
	}
3504 3505

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

3507
	ArchiveEntry(fout, tbinfo.oid, tbinfo.relname, "ACL", NULL, sql, "", "", "", NULL, NULL);
B
Bruce Momjian 已提交
3508

B
Bruce Momjian 已提交
3509 3510
}

3511

3512 3513
/*
 * dumpTables:
3514
 *	  write out to fout all the user-define tables
3515
 */
3516

3517
void
B
Bruce Momjian 已提交
3518
dumpTables(Archive *fout, TableInfo *tblinfo, int numTables,
3519 3520
		   InhInfo *inhinfo, int numInherits,
		   TypeInfo *tinfo, int numTypes, const char *tablename,
B
Bruce Momjian 已提交
3521 3522
		   const bool aclsSkip, const bool oids,
		   const bool schemaOnly, const bool dataOnly)
3523
{
3524 3525
	int			i,
				j,
B
Bruce Momjian 已提交
3526
				k;
3527
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3528
	PQExpBuffer delq = createPQExpBuffer();
3529
	char	   *serialSeq = NULL;		/* implicit sequence name created
B
Bruce Momjian 已提交
3530 3531 3532 3533
										 * by SERIAL datatype */
	const char *serialSeqSuffix = "_id_seq";	/* suffix for implicit
												 * SERIAL sequences */
	char	  **parentRels;		/* list of names of parent relations */
3534
	int			numParents;
B
Bruce Momjian 已提交
3535
	int			actual_atts;	/* number of attrs in this CREATE statment */
3536
	char	   *reltypename;
3537

V
Vadim B. Mikheev 已提交
3538
	/* First - dump SEQUENCEs */
B
Bruce Momjian 已提交
3539
	if (tablename)
B
Bruce Momjian 已提交
3540 3541 3542 3543 3544
	{
		serialSeq = malloc(strlen(tablename) + strlen(serialSeqSuffix) + 1);
		strcpy(serialSeq, tablename);
		strcat(serialSeq, serialSeqSuffix);
	}
V
Vadim B. Mikheev 已提交
3545 3546 3547 3548
	for (i = 0; i < numTables; i++)
	{
		if (!(tblinfo[i].sequence))
			continue;
B
Bruce Momjian 已提交
3549
		if (!tablename || (!strcmp(tblinfo[i].relname, tablename))
B
Bruce Momjian 已提交
3550
			|| (serialSeq && !strcmp(tblinfo[i].relname, serialSeq)))
V
Vadim B. Mikheev 已提交
3551
		{
B
Bruce Momjian 已提交
3552
			/* becomeUser(fout, tblinfo[i].usename); */
V
Vadim B. Mikheev 已提交
3553
			dumpSequence(fout, tblinfo[i]);
3554
			if (!aclsSkip)
B
Bruce Momjian 已提交
3555
				dumpACL(fout, tblinfo[i]);
V
Vadim B. Mikheev 已提交
3556 3557
		}
	}
B
Bruce Momjian 已提交
3558
	if (tablename)
B
Bruce Momjian 已提交
3559
		free(serialSeq);
3560

3561 3562
	for (i = 0; i < numTables; i++)
	{
3563
		if (tblinfo[i].sequence)/* already dumped */
V
Vadim B. Mikheev 已提交
3564
			continue;
3565 3566 3567 3568

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

3569 3570
			resetPQExpBuffer(delq);
			resetPQExpBuffer(q);
3571

3572 3573 3574 3575
			/* Use the view definition if there is one */
			if (tblinfo[i].viewdef != NULL)
			{
				reltypename = "VIEW";
3576

3577 3578
				appendPQExpBuffer(delq, "DROP VIEW %s;\n", fmtId(tblinfo[i].relname, force_quotes));
				appendPQExpBuffer(q, "CREATE VIEW %s as %s", fmtId(tblinfo[i].relname, force_quotes), tblinfo[i].viewdef);
3579

3580 3581
			}
			else
3582
			{
3583
				reltypename = "TABLE";
3584

3585 3586
				parentRels = tblinfo[i].parentRels;
				numParents = tblinfo[i].numParents;
B
Bruce Momjian 已提交
3587

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

3590 3591 3592 3593 3594 3595
				appendPQExpBuffer(q, "CREATE TABLE %s (\n\t", fmtId(tblinfo[i].relname, force_quotes));
				actual_atts = 0;
				for (j = 0; j < tblinfo[i].numatts; j++)
				{
					/* Is this one of the table's own attrs ? */
					if (tblinfo[i].inhAttrs[j] == 0)
3596
					{
3597 3598 3599 3600 3601 3602 3603 3604 3605 3606 3607 3608
						/* Format properly if not first attr */
						if (actual_atts > 0)
							appendPQExpBuffer(q, ",\n\t");

						/* Attr name & type */
						appendPQExpBuffer(q, "%s %s",
							fmtId(tblinfo[i].attnames[j], force_quotes),
							tblinfo[i].atttypedefns[j]);

						/* Default value */
						if (tblinfo[i].adef_expr[j] != NULL)
							appendPQExpBuffer(q, " DEFAULT %s",
3609
										  tblinfo[i].adef_expr[j]);
3610 3611 3612 3613 3614 3615 3616

						/* Not Null constraint */
						if (tblinfo[i].notnull[j])
							appendPQExpBuffer(q, " NOT NULL");

						actual_atts++;
					}
3617
				}
B
Hi all  
Bruce Momjian 已提交
3618

3619 3620 3621 3622 3623 3624 3625
				/* put the CONSTRAINTS inside the table def */
				for (k = 0; k < tblinfo[i].ncheck; k++)
				{
					if (actual_atts + k > 0)
						appendPQExpBuffer(q, ",\n\t");

					appendPQExpBuffer(q, "%s",
3626
								  tblinfo[i].check_expr[k]);
3627
				}
3628

3629 3630 3631 3632 3633
				/* PRIMARY KEY */
				if (tblinfo[i].primary_key)
				{
					if (actual_atts + tblinfo[i].ncheck > 0)
						appendPQExpBuffer(q, ",\n\t");
3634 3635 3636 3637
					appendPQExpBuffer(q,
									  "CONSTRAINT %s PRIMARY KEY (%s)",
									  tblinfo[i].primary_key_name,
									  tblinfo[i].primary_key);
3638
				}
B
Bruce Momjian 已提交
3639

3640
				appendPQExpBuffer(q, "\n)");
3641

3642
				if (numParents > 0)
B
Hi all  
Bruce Momjian 已提交
3643
				{
3644 3645 3646 3647
					appendPQExpBuffer(q, "\ninherits (");
					for (k = 0; k < numParents; k++)
					{
						appendPQExpBuffer(q, "%s%s",
3648 3649
									  (k > 0) ? ", " : "",
									  fmtId(parentRels[k], force_quotes));
3650 3651
					}
					appendPQExpBuffer(q, ")");
B
Hi all  
Bruce Momjian 已提交
3652 3653
				}

3654 3655
				appendPQExpBuffer(q, ";\n");
			}
B
Bruce Momjian 已提交
3656 3657

			if (!dataOnly) {
3658 3659
					ArchiveEntry(fout, tblinfo[i].oid, fmtId(tblinfo[i].relname, false),
								reltypename, NULL, q->data, delq->data, "", tblinfo[i].usename,
B
Bruce Momjian 已提交
3660 3661 3662 3663
								NULL, NULL);
			}

			if (!dataOnly && !aclsSkip)
B
Bruce Momjian 已提交
3664
				dumpACL(fout, tblinfo[i]);
3665

3666
			/* Dump Field Comments */
B
Bruce,  
Bruce Momjian 已提交
3667

3668 3669
			for (j = 0; j < tblinfo[i].numatts; j++)
			{
B
Bruce,  
Bruce Momjian 已提交
3670 3671 3672 3673 3674 3675
				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]);
			}
3676

B
Bruce,  
Bruce Momjian 已提交
3677
			/* Dump Table Comments */
3678

B
Bruce,  
Bruce Momjian 已提交
3679
			resetPQExpBuffer(q);
3680
			appendPQExpBuffer(q, "%s %s", reltypename, fmtId(tblinfo[i].relname, force_quotes));
B
Bruce,  
Bruce Momjian 已提交
3681
			dumpComment(fout, q->data, tblinfo[i].oid);
3682

3683 3684
		}
	}
3685 3686 3687 3688
}

/*
 * dumpIndices:
3689
 *	  write out to fout all the user-define indices
3690
 */
3691
void
B
Bruce Momjian 已提交
3692
dumpIndices(Archive *fout, IndInfo *indinfo, int numIndices,
3693
			TableInfo *tblinfo, int numTables, const char *tablename)
3694
{
3695 3696 3697
	int			i,
				k;
	int			tableInd;
3698
	PQExpBuffer attlist = createPQExpBuffer();
B
Bruce Momjian 已提交
3699 3700 3701
	char	   *classname[INDEX_MAX_KEYS];
	char	   *funcname;		/* the name of the function to comput the
								 * index key from */
3702 3703 3704
	int			indkey,
				indclass;
	int			nclass;
3705

3706
	PQExpBuffer q = createPQExpBuffer(),
B
Bruce Momjian 已提交
3707
				delq = createPQExpBuffer(),
B
Hi, all  
Bruce Momjian 已提交
3708 3709
				id1 = createPQExpBuffer(),
				id2 = createPQExpBuffer();
3710
	PGresult   *res;
3711 3712 3713 3714

	for (i = 0; i < numIndices; i++)
	{
		tableInd = findTableByName(tblinfo, numTables,
3715 3716 3717 3718 3719
								   indinfo[i].indrelname);
		if (tableInd < 0)
		{
			fprintf(stderr, "failed sanity check, table %s was not found\n",
					indinfo[i].indrelname);
3720
			exit(1);
3721
		}
3722 3723 3724 3725 3726 3727 3728 3729 3730 3731 3732 3733

		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 已提交
3734 3735
			resetPQExpBuffer(q);
			appendPQExpBuffer(q,
3736 3737 3738
							  "SELECT proname from pg_proc "
							  "where pg_proc.oid = '%s'::oid",
							  indinfo[i].indproc);
B
Hi, all  
Bruce Momjian 已提交
3739
			res = PQexec(g_conn, q->data);
3740 3741
			if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
			{
B
Bruce Momjian 已提交
3742 3743
				fprintf(stderr, "dumpIndices(): SELECT (funcname) failed.  "
						"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3744 3745
				exit_nicely(g_conn);
			}
B
Bruce Momjian 已提交
3746
			funcname = strdup(PQgetvalue(res, 0, PQfnumber(res, "proname")));
3747 3748 3749 3750 3751 3752 3753 3754 3755
			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 已提交
3756 3757
			resetPQExpBuffer(q);
			appendPQExpBuffer(q,
3758 3759 3760
							  "SELECT opcname from pg_opclass "
							  "where pg_opclass.oid = '%u'::oid",
							  indclass);
B
Hi, all  
Bruce Momjian 已提交
3761
			res = PQexec(g_conn, q->data);
3762 3763
			if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
			{
B
Bruce Momjian 已提交
3764 3765
				fprintf(stderr, "dumpIndices(): SELECT (classname) failed.  "
									"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3766 3767
				exit_nicely(g_conn);
			}
B
Bruce Momjian 已提交
3768
			classname[nclass] = strdup(PQgetvalue(res, 0, PQfnumber(res, "opcname")));
3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779
			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 已提交
3780 3781
		resetPQExpBuffer(attlist);
		for (k = 0; k < INDEX_MAX_KEYS; k++)
3782
		{
3783
			char	   *attname;
3784 3785

			indkey = atoi(indinfo[i].indkey[k]);
B
Bruce Momjian 已提交
3786
			if (indkey == InvalidAttrNumber)
3787 3788 3789 3790 3791 3792 3793
				break;
			indkey--;
			if (indkey == ObjectIdAttributeNumber - 1)
				attname = "oid";
			else
				attname = tblinfo[tableInd].attnames[indkey];
			if (funcname)
B
Hi, all  
Bruce Momjian 已提交
3794
				appendPQExpBuffer(attlist, "%s%s",
B
Bruce Momjian 已提交
3795
					 (k == 0) ? "" : ", ", fmtId(attname, force_quotes));
3796 3797 3798 3799 3800
			else
			{
				if (k >= nclass)
				{
					fprintf(stderr, "dumpIndices(): OpClass not found for "
3801
							"attribute '%s' of index '%s'\n",
3802 3803 3804
							attname, indinfo[i].indexrelname);
					exit_nicely(g_conn);
				}
B
Hi, all  
Bruce Momjian 已提交
3805 3806 3807 3808 3809
				resetPQExpBuffer(id1);
				resetPQExpBuffer(id2);
				appendPQExpBuffer(id1, fmtId(attname, force_quotes));
				appendPQExpBuffer(id2, fmtId(classname[k], force_quotes));
				appendPQExpBuffer(attlist, "%s%s %s",
3810
							 (k == 0) ? "" : ", ", id1->data, id2->data);
3811 3812 3813 3814 3815 3816
				free(classname[k]);
			}
		}

		if (!tablename || (!strcmp(indinfo[i].indrelname, tablename)))
		{
B
Bruce Momjian 已提交
3817 3818 3819 3820 3821

			/*
			 * 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.
3822
			 */
3823

B
Hi, all  
Bruce Momjian 已提交
3824 3825 3826 3827
			resetPQExpBuffer(id1);
			resetPQExpBuffer(id2);
			appendPQExpBuffer(id1, fmtId(indinfo[i].indexrelname, force_quotes));
			appendPQExpBuffer(id2, fmtId(indinfo[i].indrelname, force_quotes));
3828

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

B
Bruce Momjian 已提交
3832 3833 3834
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "CREATE %s INDEX %s on %s using %s (",
					(strcmp(indinfo[i].indisunique, "t") == 0) ? "UNIQUE" : "",
B
Hi, all  
Bruce Momjian 已提交
3835 3836
					id1->data,
					id2->data,
3837 3838 3839
					indinfo[i].indamname);
			if (funcname)
			{
3840
				/* need 2 printf's here cuz fmtId has static return area */
B
Bruce Momjian 已提交
3841 3842 3843
				appendPQExpBuffer(q, " %s", fmtId(funcname, false));
				appendPQExpBuffer(q, " (%s) %s );\n", attlist->data, 
									fmtId(classname[0], force_quotes));
3844 3845 3846 3847
				free(funcname);
				free(classname[0]);
			}
			else
B
Bruce Momjian 已提交
3848
				appendPQExpBuffer(q, " %s );\n", attlist->data);
B
Bruce,  
Bruce Momjian 已提交
3849 3850 3851

			/* Dump Index Comments */

B
Bruce Momjian 已提交
3852
			ArchiveEntry(fout, tblinfo[tableInd].oid, id1->data, "INDEX", NULL, q->data, delq->data,
3853
							"", tblinfo[tableInd].usename, NULL, NULL);
B
Bruce Momjian 已提交
3854

B
Bruce,  
Bruce Momjian 已提交
3855 3856 3857
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "INDEX %s", id1->data);
			dumpComment(fout, q->data, indinfo[i].indoid);
3858

3859 3860
		}
	}
3861 3862

}
3863

3864
/*
B
Bruce Momjian 已提交
3865
 * dumpTuples
3866 3867 3868 3869 3870
 *	  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
3871
 *	  in PostgreSQL.
3872 3873 3874 3875
 *
 * the attrmap passed in tells how to map the attributes copied in to the
 * attributes copied out
 */
3876
#ifdef NOT_USED
3877
void
3878
dumpTuples(PGresult *res, FILE *fout, int *attrmap)
3879
{
3880 3881 3882 3883
	int			j,
				k;
	int			m,
				n;
B
Bruce Momjian 已提交
3884
	char	  **outVals = NULL; /* values to copy out */
3885 3886 3887 3888 3889 3890 3891 3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903

	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++)
			{
3904
				char	   *pval = outVals[k];
3905 3906 3907 3908 3909 3910 3911 3912 3913 3914 3915 3916 3917 3918 3919 3920 3921 3922 3923 3924

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

3927
#endif
3928

3929 3930 3931 3932
/*
 * setMaxOid -
 * find the maximum oid and generate a COPY statement to set it
*/
3933

3934
static void
B
Bruce Momjian 已提交
3935
setMaxOid(Archive *fout)
3936
{
B
Bruce Momjian 已提交
3937 3938 3939 3940
	PGresult   	*res;
	Oid		max_oid;
	char		sql[1024];
	int		pos;
3941

3942
	res = PQexec(g_conn, "CREATE TEMPORARY TABLE pgdump_oid (dummy int4)");
3943 3944 3945
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
3946
		fprintf(stderr, "Can not create pgdump_oid table.  "
B
Bruce Momjian 已提交
3947
					"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3948 3949 3950
		exit_nicely(g_conn);
	}
	PQclear(res);
3951
	res = PQexec(g_conn, "INSERT INTO pgdump_oid VALUES (0)");
3952 3953 3954
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
3955
		fprintf(stderr, "Can not insert into pgdump_oid table.  "
B
Bruce Momjian 已提交
3956
					"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3957 3958 3959 3960 3961 3962 3963 3964 3965
		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);
3966
	res = PQexec(g_conn, "DROP TABLE pgdump_oid;");
3967 3968 3969
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
3970
		fprintf(stderr, "Can not drop pgdump_oid table.  "
B
Bruce Momjian 已提交
3971
							"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3972 3973 3974 3975
		exit_nicely(g_conn);
	}
	PQclear(res);
	if (g_verbose)
3976
		fprintf(stderr, "%s maximum system oid is %u %s\n",
3977
				g_comment_start, max_oid, g_comment_end);
3978 3979
	pos = snprintf(sql, 1024, "CREATE TEMPORARY TABLE pgdump_oid (dummy int4);\n");
	pos = pos + snprintf(sql+pos, 1024-pos, "COPY pgdump_oid WITH OIDS FROM stdin;\n");
B
Bruce Momjian 已提交
3980 3981
	pos = pos + snprintf(sql+pos, 1024-pos, "%-d\t0\n", max_oid);
	pos = pos + snprintf(sql+pos, 1024-pos, "\\.\n");
3982
	pos = pos + snprintf(sql+pos, 1024-pos, "DROP TABLE pgdump_oid;\n");
B
Bruce Momjian 已提交
3983

3984
	ArchiveEntry(fout, "0", "Max OID", "<Init>", NULL, sql, "", "", "", NULL, NULL);
3985
}
3986 3987 3988

/*
 * findLastBuiltInOid -
3989
 * find the last built in oid
3990 3991
 * we do this by retrieving datlastsysoid from the pg_database entry for this database,
 */
3992

3993
static Oid
3994
findLastBuiltinOid(const char* dbname)
3995
{
B
Bruce Momjian 已提交
3996
	PGresult   *res;
3997
	int			ntups;
3998
	Oid			last_oid;
3999 4000 4001
	PQExpBuffer query = createPQExpBuffer();

	resetPQExpBuffer(query);
4002 4003
	appendPQExpBuffer(query, "SELECT datlastsysoid from pg_database where datname = ");
	formatStringLiteral(query, dbname);
4004

4005
	res = PQexec(g_conn, query->data);
4006 4007 4008
	if (res == NULL ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
4009
		fprintf(stderr, "pg_dump: error in finding the last system OID");
4010
		fprintf(stderr, "Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
4011 4012 4013
		exit_nicely(g_conn);
	}
	ntups = PQntuples(res);
4014
	if (ntups < 1)
4015
	{
4016 4017
		fprintf(stderr, "pg_dump: couldn't find the pg_database entry.\n");
		fprintf(stderr, "There is no entry in the 'pg_database' table for this database.\n");
4018 4019 4020 4021
		exit_nicely(g_conn);
	}
	if (ntups > 1)
	{
4022 4023
		fprintf(stderr, "pg_dump: found more than one matching database.\n");
		fprintf(stderr, "There is more than one entry for this database in the 'pg_database' table\n");
4024 4025
		exit_nicely(g_conn);
	}
4026
	last_oid = atoi(PQgetvalue(res, 0, PQfnumber(res, "datlastsysoid")));
4027 4028
	PQclear(res);
	return last_oid;
4029 4030 4031
}


4032
static void
B
Bruce Momjian 已提交
4033
dumpSequence(Archive *fout, TableInfo tbinfo)
4034
{
4035
	PGresult   *res;
B
Bruce Momjian 已提交
4036
	int4		last,
4037 4038 4039 4040
				incby,
				maxv,
				minv,
				cache;
B
Bruce Momjian 已提交
4041
	char		cycled,
B
Bruce Momjian 已提交
4042
				called;
4043 4044
	const char *t;
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
4045
	PQExpBuffer delqry = createPQExpBuffer();
4046

B
Hi, all  
Bruce Momjian 已提交
4047
	appendPQExpBuffer(query,
4048
			"SELECT sequence_name, last_value, increment_by, max_value, "
4049 4050
				  "min_value, cache_value, is_cycled, is_called from %s",
					  fmtId(tbinfo.relname, force_quotes));
4051

B
Hi, all  
Bruce Momjian 已提交
4052
	res = PQexec(g_conn, query->data);
4053 4054
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
4055 4056
		fprintf(stderr, "dumpSequence(%s): SELECT failed.  "
					"Explanation from backend: '%s'.\n", tbinfo.relname, PQerrorMessage(g_conn));
4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069 4070 4071 4072 4073 4074 4075 4076 4077 4078 4079 4080 4081 4082 4083 4084 4085 4086 4087
		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);

B
Bruce Momjian 已提交
4088 4089
	resetPQExpBuffer(delqry);
	appendPQExpBuffer(delqry, "DROP SEQUENCE %s;\n", fmtId(tbinfo.relname, force_quotes));
4090

4091 4092 4093 4094 4095 4096
	/*
	 * The logic we use for restoring sequences is as follows:
	 *		- 	Add a basic CREATE SEQUENCE statement 
	 *			(use last_val for start if called == 'f', else use min_val for start_val).
	 *		-	Add a 'SETVAL(seq, last_val, iscalled)' at restore-time iff we load data
	 */
B
Hi, all  
Bruce Momjian 已提交
4097 4098
	resetPQExpBuffer(query);
	appendPQExpBuffer(query,
4099 4100
				  "CREATE SEQUENCE %s start %d increment %d maxvalue %d "
					  "minvalue %d  cache %d %s;\n",
4101 4102 4103
					fmtId(tbinfo.relname, force_quotes), 
					  (called == 't') ? minv : last,
					  incby, maxv, minv, cache,
4104
					  (cycled == 't') ? "cycle" : "");
4105

B
Bruce Momjian 已提交
4106
	ArchiveEntry(fout, tbinfo.oid, fmtId(tbinfo.relname, force_quotes), "SEQUENCE", NULL,
4107
					query->data, delqry->data, "", tbinfo.usename, NULL, NULL);
4108

4109 4110

	resetPQExpBuffer(query);
4111 4112 4113
	appendPQExpBuffer(query, "SELECT setval (");
	formatStringLiteral(query, fmtId(tbinfo.relname, force_quotes));
	appendPQExpBuffer(query, ", %d, '%c');\n", last, called);
4114 4115 4116 4117

	ArchiveEntry(fout, tbinfo.oid, fmtId(tbinfo.relname, force_quotes), "SEQUENCE SET", NULL,
					query->data, "" /* Del */, "", "", NULL, NULL);

B
Bruce,  
Bruce Momjian 已提交
4118 4119 4120 4121 4122 4123
	/* Dump Sequence Comments */

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

4124
}
V
Vadim B. Mikheev 已提交
4125 4126


4127
static void
B
Bruce Momjian 已提交
4128
dumpTriggers(Archive *fout, const char *tablename,
4129
			 TableInfo *tblinfo, int numTables)
V
Vadim B. Mikheev 已提交
4130
{
4131 4132 4133
	int			i,
				j;

V
Vadim B. Mikheev 已提交
4134 4135 4136
	if (g_verbose)
		fprintf(stderr, "%s dumping out triggers %s\n",
				g_comment_start, g_comment_end);
4137

V
Vadim B. Mikheev 已提交
4138 4139 4140 4141 4142 4143
	for (i = 0; i < numTables; i++)
	{
		if (tablename && strcmp(tblinfo[i].relname, tablename))
			continue;
		for (j = 0; j < tblinfo[i].ntrig; j++)
		{
B
Bruce Momjian 已提交
4144
			ArchiveEntry(fout, tblinfo[i].triggers[j].oid, tblinfo[i].triggers[j].tgname,
4145
						"TRIGGER", NULL, tblinfo[i].triggers[j].tgsrc, "", "", 
B
Bruce Momjian 已提交
4146 4147
						tblinfo[i].usename, NULL, NULL);
			dumpComment(fout, tblinfo[i].triggers[j].tgcomment, tblinfo[i].triggers[j].oid);
V
Vadim B. Mikheev 已提交
4148 4149 4150
		}
	}
}
4151 4152


4153
static void
B
Bruce Momjian 已提交
4154
dumpRules(Archive *fout, const char *tablename,
B
Bruce Momjian 已提交
4155
		  TableInfo *tblinfo, int numTables)
4156
{
B
Bruce Momjian 已提交
4157 4158 4159 4160
	PGresult   *res;
	int			nrules;
	int			i,
				t;
4161
	PQExpBuffer query = createPQExpBuffer();
4162

B
Bruce Momjian 已提交
4163
	int			i_definition;
4164
	int			i_oid;
4165
	int			i_owner;
4166
	int			i_rulename;
4167 4168 4169 4170 4171 4172 4173 4174 4175 4176 4177 4178 4179 4180 4181

	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
4182 4183
		 * We include pg_rules in the cross since it filters out
		 * all view rules (pjw 15-Sep-2000).
4184
		 */
B
Hi, all  
Bruce Momjian 已提交
4185
		resetPQExpBuffer(query);
4186
		appendPQExpBuffer(query, "SELECT definition,"
4187
						  "   (select usename from pg_user where pg_class.relowner = usesysid) AS viewowner, "
4188 4189
						  "   pg_rewrite.oid, pg_rewrite.rulename "
						  "FROM pg_rewrite, pg_class, pg_rules "
4190 4191 4192
						  "WHERE pg_class.relname = ");
		formatStringLiteral(query, tblinfo[t].relname);
		appendPQExpBuffer(query,
4193 4194 4195
						  "    AND pg_rewrite.ev_class = pg_class.oid "
						  "    AND pg_rules.tablename = pg_class.relname "
						  "    AND pg_rules.rulename = pg_rewrite.rulename "
4196
						  "ORDER BY pg_rewrite.oid");
B
Hi, all  
Bruce Momjian 已提交
4197
		res = PQexec(g_conn, query->data);
4198 4199 4200
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
4201
			fprintf(stderr, "dumpRules(): SELECT failed for rules associated with table \"%s\".\n\tExplanation from backend: '%s'.\n",
B
Bruce Momjian 已提交
4202
					tblinfo[t].relname, PQerrorMessage(g_conn));
4203 4204 4205 4206 4207
			exit_nicely(g_conn);
		}

		nrules = PQntuples(res);
		i_definition = PQfnumber(res, "definition");
4208
		i_owner = PQfnumber(res, "viewowner");
B
Bruce,  
Bruce Momjian 已提交
4209 4210
		i_oid = PQfnumber(res, "oid");
		i_rulename = PQfnumber(res, "rulename");
4211 4212 4213 4214

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

4216 4217
		for (i = 0; i < nrules; i++)
		{
B
Bruce Momjian 已提交
4218 4219
			ArchiveEntry(fout, PQgetvalue(res, i, i_oid), PQgetvalue(res, i, i_rulename),
							"RULE", NULL, PQgetvalue(res, i, i_definition),
4220
							"", "", PQgetvalue(res, i, i_owner), NULL, NULL);
4221

B
Bruce,  
Bruce Momjian 已提交
4222 4223 4224 4225 4226
			/* 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));
4227

B
Bruce,  
Bruce Momjian 已提交
4228 4229
		}

4230 4231 4232 4233
		PQclear(res);
	}
}