pg_dump.c 111.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.180 2000/11/14 18:37:45 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 142
static char *checkForQuote(const char *s);
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
	const char *expsrc;
438

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

487 488
					/*
					 * All other types are printed as string literals,
489 490 491
					 * with appropriate escaping of special characters.
					 * Quote mark ' goes to '' per SQL standard, other
					 * stuff goes to \ sequences.
492
					 */
B
Bruce Momjian 已提交
493
					archputc('\'', fout);
494 495 496 497 498 499
					expsrc = PQgetvalue(res, tuple, field);
					while (*expsrc)
					{
						char		ch = *expsrc++;

						if (ch == '\\' || ch == '\'')
500
						{
B
Bruce Momjian 已提交
501 502
							archputc(ch, fout);			/* double these */
							archputc(ch, fout);
503
						}
504 505 506
						else if (ch < '\040')
						{
							/* generate octal escape for control chars */
B
Bruce Momjian 已提交
507 508 509 510
							archputc('\\', fout);
							archputc(((ch >> 6) & 3) + '0', fout);
							archputc(((ch >> 3) & 7) + '0', fout);
							archputc((ch & 7) + '0', fout);
511 512
						}
						else
B
Bruce Momjian 已提交
513
							archputc(ch, fout);
514
					}
B
Bruce Momjian 已提交
515
					archputc('\'', fout);
516
					break;
517
			}
518
		}
B
Bruce Momjian 已提交
519
		archprintf(fout, ");\n");
520 521
	}
	PQclear(res);
B
Bruce Momjian 已提交
522
	return 1;
523 524 525 526
}

/*
 * DumpClasses -
527
 *	  dump the contents of all the classes.
528 529
 */
static void
B
Bruce Momjian 已提交
530
dumpClasses(const TableInfo *tblinfo, const int numTables, Archive *fout,
531
			const char *onlytable, const bool oids, const bool force_quotes)
532 533
{

B
Bruce Momjian 已提交
534 535 536 537
	int				i;
	char	   		*all_only;
	DataDumperPtr	dumpFn;
	DumpContext		*dumpCtx;
538 539 540
	char			*oidsPart;
	char			copyBuf[512];
	char			*copyStmt;
541 542 543 544

	if (onlytable == NULL)
		all_only = "all";
	else
545
		all_only = "only";
546

547 548 549 550 551 552
	if (oids == true)
		oidsPart = "WITH OIDS ";
	else
		oidsPart = "";


553
	if (g_verbose)
554
		fprintf(stderr, "%s preparing to dump out the contents of %s %d table%s/sequence%s %s\n",
555 556
				g_comment_start, all_only,
				(onlytable == NULL) ? numTables : 1,
557
		  (onlytable == NULL) ? "s" : "", (onlytable == NULL) ? "s" : "",
558
				g_comment_end);
559

V
Vadim B. Mikheev 已提交
560 561 562 563 564 565 566 567 568 569
	/* 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)
570
					fprintf(stderr, "%s dumping out schema of sequence '%s' %s\n",
571
					 g_comment_start, tblinfo[i].relname, g_comment_end);
B
Bruce Momjian 已提交
572
				/* becomeUser(fout, tblinfo[i].usename); */
V
Vadim B. Mikheev 已提交
573 574 575 576
				dumpSequence(fout, tblinfo[i]);
			}
		}
	}
577 578 579

	for (i = 0; i < numTables; i++)
	{
580
		const char *classname = tblinfo[i].relname;
581 582

		/* Skip VIEW relations */
583
		if (tblinfo[i].viewdef != NULL)
584
			continue;
585 586

		if (tblinfo[i].sequence)/* already dumped */
V
Vadim B. Mikheev 已提交
587
			continue;
588 589 590 591

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

B
Bruce Momjian 已提交
595 596 597 598 599 600
			/* becomeUser(fout, tblinfo[i].usename); */

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

602 603
			if (!dumpData) /* Dump/restore using COPY */
			{
B
Bruce Momjian 已提交
604 605
				dumpFn = dumpClasses_nodumpData;
				/* dumpClasses_nodumpData(fout, classname, oids); */
606 607 608 609 610 611
				sprintf(copyBuf, "COPY %s %s FROM stdin;\n", fmtId(tblinfo[i].relname, force_quotes),
						oidsPart);
				copyStmt = copyBuf;
			}
			else /* Restore using INSERT */
			{
B
Bruce Momjian 已提交
612 613
				dumpFn = dumpClasses_dumpData;
				/* dumpClasses_dumpData(fout, classname); */
614 615
				copyStmt = NULL;
			}
B
Bruce Momjian 已提交
616 617

			ArchiveEntry(fout, tblinfo[i].oid, fmtId(tblinfo[i].relname, false),
618
							"TABLE DATA", NULL, "", "", copyStmt, tblinfo[i].usename,
B
Bruce Momjian 已提交
619
							dumpFn, dumpCtx);
620
		}
621
	}
622 623
}

624
int
625
main(int argc, char **argv)
626
{
B
Bruce Momjian 已提交
627 628 629
	int			c;
	const char *progname;
	const char *filename = NULL;
B
Bruce Momjian 已提交
630
	const char *format = "p";
B
Bruce Momjian 已提交
631 632 633 634
	const char *dbname = NULL;
	const char *pghost = NULL;
	const char *pgport = NULL;
	char	   *tablename = NULL;
635
	bool		oids = false;
B
Bruce Momjian 已提交
636 637
	TableInfo  *tblinfo;
	int			numTables;
638
	bool		use_password = false;
B
Bruce Momjian 已提交
639
	int			compressLevel = -1;
640
	bool		ignore_version = false;
641 642
	int			plainText = 0;
	int			outputClean = 0;
643
	int			outputCreate = 0;
644
	int			outputBlobs = 0;
645 646 647
	int			outputNoOwner = 0;
	int			outputNoReconnect = 0;
	char		*outputSuperuser = NULL;
648

B
Bruce Momjian 已提交
649
	RestoreOptions	*ropt;
650

B
Hi,  
Bruce Momjian 已提交
651 652 653
#ifdef HAVE_GETOPT_LONG
	static struct option long_options[] = {
		{"data-only", no_argument, NULL, 'a'},
654
		{"blobs", no_argument, NULL, 'b' },
B
Hi,  
Bruce Momjian 已提交
655
		{"clean", no_argument, NULL, 'c'},
656
		{"create", no_argument, NULL, 'C'},
B
Bruce Momjian 已提交
657 658
		{"file", required_argument, NULL, 'f'},
		{"format", required_argument, NULL, 'F'},
659
		{"inserts", no_argument, NULL, 'd'},
660
		{"attribute-inserts", no_argument, NULL, 'D'},
B
Hi,  
Bruce Momjian 已提交
661
		{"host", required_argument, NULL, 'h'},
662
		{"ignore-version", no_argument, NULL, 'i'},
663
		{"no-reconnect", no_argument, NULL, 'R'},
B
Hi,  
Bruce Momjian 已提交
664 665
		{"no-quotes", no_argument, NULL, 'n'},
		{"quotes", no_argument, NULL, 'N'},
666
		{"oids", no_argument, NULL, 'o'},
667
		{"no-owner", no_argument, NULL, 'O'},
B
Hi,  
Bruce Momjian 已提交
668 669
		{"port", required_argument, NULL, 'p'},
		{"schema-only", no_argument, NULL, 's'},
670
		{"superuser", required_argument, NULL, 'S'},
B
Hi,  
Bruce Momjian 已提交
671 672 673 674
		{"table", required_argument, NULL, 't'},
		{"password", no_argument, NULL, 'u'},
		{"verbose", no_argument, NULL, 'v'},
		{"no-acl", no_argument, NULL, 'x'},
B
Bruce Momjian 已提交
675
		{"compress", required_argument, NULL, 'Z'},
B
Hi,  
Bruce Momjian 已提交
676
		{"help", no_argument, NULL, '?'},
677
		{"version", no_argument, NULL, 'V'}
678
	};
679 680
	int			optindex;

B
Hi,  
Bruce Momjian 已提交
681 682
#endif

683
	g_verbose = false;
684
	force_quotes = true;
685 686 687 688 689

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

690
	dataOnly = schemaOnly = dumpData = attrNames = false;
691

692 693 694 695
	if (!strrchr(argv[0], SEP_CHAR))
		progname = argv[0];
	else
		progname = strrchr(argv[0], SEP_CHAR) + 1;
696

697 698 699 700
	/* Set defaulty options based on progname */
	if (strcmp(progname, "pg_backup") == 0)
	{
		format = "c";
P
Philip Warner 已提交
701
		outputBlobs = true;
702
	}
703

B
Hi,  
Bruce Momjian 已提交
704
#ifdef HAVE_GETOPT_LONG
705
	while ((c = getopt_long(argc, argv, "abcCdDf:F:h:inNoOp:sS:t:uvxzZ:V?", long_options, &optindex)) != -1)
B
Hi,  
Bruce Momjian 已提交
706
#else
707
	while ((c = getopt(argc, argv, "abcCdDf:F:h:inNoOp:sS:t:uvxzZ:V?-")) != -1)
B
Hi,  
Bruce Momjian 已提交
708
#endif
709

710 711 712
	{
		switch (c)
		{
713
			case 'a':			/* Dump data only */
714
				dataOnly = true;
715
				break;
716

717 718 719 720
			case 'b':			/* Dump blobs */
				outputBlobs = true;
				break;

B
Bruce Momjian 已提交
721
			case 'c':			/* clean (i.e., drop) schema prior to
B
Bruce Momjian 已提交
722 723 724 725
 								 * create */
				outputClean = 1;
 				break;
 
726 727 728 729 730
			case 'C':			/* Create DB */

				outputCreate = 1;
				break;

731
			case 'd':			/* dump data as proper insert strings */
732
				dumpData = true;
733
				break;
734

735 736
			case 'D':			/* dump data as proper insert strings with
								 * attr names */
737 738
				dumpData = true;
				attrNames = true;
739
				break;
740

741
			case 'f':
742 743
				filename = optarg;
				break;
744

B
Bruce Momjian 已提交
745 746 747
			case 'F':
				format = optarg;
				break;
748

B
Bruce Momjian 已提交
749
			case 'h':			/* server host */
750 751
				pghost = optarg;
				break;
752

753 754 755
			case 'i':			/* ignore database version mismatch */
				ignore_version = true;
				break;
756

B
Bruce Momjian 已提交
757 758
			case 'n':			/* Do not force double-quotes on
								 * identifiers */
759
				force_quotes = false;
760
				break;
761

762
			case 'N':			/* Force double-quotes on identifiers */
763
				force_quotes = true;
764
				break;
765

766
			case 'o':			/* Dump oids */
767
				oids = true;
768
				break;
769 770


771 772 773
			case 'O':			/* Don't reconnect to match owner */
				outputNoOwner = 1;
				break;
774

775 776 777
			case 'p':			/* server port */
				pgport = optarg;
				break;
778

779 780 781
			case 'R':			/* No reconnect */
				outputNoReconnect = 1;
				break;
782

783
			case 's':			/* dump schema only */
784
				schemaOnly = true;
785
				break;
786

787 788 789
			case 'S':			/* Username for superuser in plain text output */
				outputSuperuser = strdup(optarg);
				break;
790

791
			case 't':			/* Dump data for this table only */
792
				{
793
					int			i;
794 795

					tablename = strdup(optarg);
B
Bruce Momjian 已提交
796 797 798 799 800

					/*
					 * quoted string? Then strip quotes and preserve
					 * case...
					 */
801 802 803
					if (tablename[0] == '"')
					{
						strcpy(tablename, &tablename[1]);
B
Bruce Momjian 已提交
804 805
						if (*(tablename + strlen(tablename) - 1) == '"')
							*(tablename + strlen(tablename) - 1) = '\0';
806 807 808 809 810
					}
					/* otherwise, convert table name to lowercase... */
					else
					{
						for (i = 0; tablename[i]; i++)
811 812
							if (isascii((int) tablename[i]) &&
								isupper((int) tablename[i]))
813 814
								tablename[i] = tolower(tablename[i]);
					}
815
				}
816
				break;
817

818 819 820
			case 'u':
				use_password = true;
				break;
821

822 823
			case 'v':			/* verbose */
				g_verbose = true;
824
				break;
825

826 827
			case 'x':			/* skip ACL dump */
				aclsSkip = true;
828
				break;
829

B
Bruce Momjian 已提交
830 831 832
			case 'Z':			/* Compression Level */
				compressLevel = atoi(optarg);
				break;
833

834 835 836 837
			case 'V':
				version();
				exit(0);
				break;
838

839
			case '?':
840 841 842 843 844 845 846 847 848 849 850 851 852 853 854

				/*
				 * 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);
				}
855
				break;
856 857 858
#ifndef HAVE_GETOPT_LONG
			case '-':
				fprintf(stderr, "%s was compiled without support for long options.\n"
859
				   "Use -? for help on invocation options.\n", progname);
860 861 862
				exit(1);
				break;
#endif
863
			default:
864 865
				fprintf(stderr, "%s: unknown option -%c\nTry -? for help.\n", progname, c);
				exit(1);
866 867 868
		}
	}

P
Philip Warner 已提交
869 870 871 872 873 874 875 876
	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 已提交
877 878 879 880 881 882 883 884
	if (dataOnly && schemaOnly)
	{
		fprintf(stderr,
				"%s: 'Schema Only' and 'Data Only' are incompatible options.\n",
				progname);
		exit(1);
	}

885 886 887 888 889 890 891 892
	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);
	}

893 894 895
	if (dumpData == true && oids == true)
	{
		fprintf(stderr,
896
				"%s: INSERT's can not set oids, so INSERT and OID options can not be used together.\n",
897
				progname);
898
		exit(1);
899
	}
900

P
Philip Warner 已提交
901 902 903 904 905 906 907 908
	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);
	}

909
	/* open the output file */
B
Bruce Momjian 已提交
910 911 912 913 914 915 916 917 918 919 920 921 922 923 924
	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;
925 926 927 928 929 930
			g_fout = CreateArchive(filename, archNull, 0);
			break;

		case 't':
		case 'T':
			g_fout = CreateArchive(filename, archTar, compressLevel);
B
Bruce Momjian 已提交
931 932 933
			break;

		default:
934
			fprintf(stderr,
B
Bruce Momjian 已提交
935 936 937 938 939 940 941 942 943 944
				"%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);
945 946
	}

947 948
	/* Let the archiver know how noisy to be */
	g_fout->verbose = g_verbose;
949

950
	dbname = argv[optind];
951

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

955 956 957 958 959 960 961 962
	/*
	 * Start serializable transaction to dump consistent data
	 */
	{
		PGresult   *res;

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

966 967 968
		PQclear(res);
		res = PQexec(g_conn, "set transaction isolation level serializable");
		if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
969 970 971
		    exit_horribly(g_fout, "SET TRANSACTION command failed. Explanation from backend: '%s'.\n",
				PQerrorMessage(g_conn));

972 973 974
		PQclear(res);
	}

975
	g_last_builtin_oid = findLastBuiltinOid(dbname);
976

977 978 979 980
	/* Dump the database definition */
	if (!dataOnly)
		dumpDatabase(g_fout);

981
	if (oids == true)
982
		setMaxOid(g_fout);
B
Bruce Momjian 已提交
983 984

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

989
	if (!schemaOnly)
990 991 992 993 994 995
	{
	    dumpClasses(tblinfo, numTables, g_fout, tablename, oids, force_quotes);
	}

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

997 998
	if (!dataOnly)				/* dump indexes and triggers at the end
								 * for performance */
V
Vadim B. Mikheev 已提交
999 1000 1001
	{
		dumpSchemaIdx(g_fout, tablename, tblinfo, numTables);
		dumpTriggers(g_fout, tablename, tblinfo, numTables);
1002
		dumpRules(g_fout, tablename, tblinfo, numTables);
V
Vadim B. Mikheev 已提交
1003
	}
1004

1005 1006
	/* Now sort the output nicely */
	SortTocByOID(g_fout);
1007
	MoveToStart(g_fout, "DATABASE");
1008 1009 1010 1011 1012
	MoveToEnd(g_fout, "TABLE DATA");
	MoveToEnd(g_fout, "BLOBS");
	MoveToEnd(g_fout, "INDEX");
	MoveToEnd(g_fout, "TRIGGER");
	MoveToEnd(g_fout, "RULE");
1013
	MoveToEnd(g_fout, "SEQUENCE SET");
1014

B
Bruce Momjian 已提交
1015 1016 1017 1018 1019 1020
	if (plainText) 
	{
		ropt = NewRestoreOptions();
		ropt->filename = (char*)filename;
		ropt->dropSchema = outputClean;
		ropt->aclsSkip = aclsSkip;
1021 1022 1023 1024 1025 1026 1027 1028 1029
		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 已提交
1030 1031 1032 1033 1034 1035 1036 1037 1038 1039

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

		RestoreArchive(g_fout, ropt);
	}

	CloseArchive(g_fout);
1040

1041 1042 1043
	clearTableInfo(tblinfo, numTables);
	PQfinish(g_conn);
	exit(0);
1044 1045
}

1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064
/*
 * 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 */
1065
	appendPQExpBuffer(dbQry, "select (select usename from pg_user where datdba = usesysid) as dba from pg_database"
1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099
							" where datname = '%s'", PQdb(g_conn));

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


1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124
/*
 * 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 */
1125
    appendPQExpBuffer(oidQry, "Declare blobOid Cursor for SELECT DISTINCT loid FROM pg_largeobject");
1126 1127 1128 1129

	res = PQexec(g_conn, oidQry->data);
	if (!res || PQresultStatus(res) != PGRES_COMMAND_OK)
	{
1130 1131
		fprintf(stderr, "dumpBlobs(): Declare Cursor failed.  Explanation from backend: '%s'.\n", 
										PQerrorMessage(g_conn));
1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 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
		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;
}

1187
/*
1188 1189
 * getTypes:
 *	  read all base types in the system catalogs and return them in the
1190 1191
 * TypeInfo* structure
 *
1192
 *	numTypes is set to the number of types read in
1193 1194
 *
 */
1195
TypeInfo   *
1196 1197
getTypes(int *numTypes)
{
B
Bruce Momjian 已提交
1198
	PGresult   *res;
1199 1200
	int			ntups;
	int			i;
1201
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1202
	TypeInfo   *tinfo;
1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218

	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;
1219
	int			i_typedefn;
1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231

	/* 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 已提交
1232
	appendPQExpBuffer(query, "SELECT pg_type.oid, typowner, typname, typlen, typprtlen, "
1233 1234 1235 1236 1237
		"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" );
1238

B
Hi, all  
Bruce Momjian 已提交
1239
	res = PQexec(g_conn, query->data);
1240 1241 1242
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1243
		fprintf(stderr, "getTypes(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265
		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");
1266
	i_typedefn = PQfnumber(res, "typedefn");
1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283

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

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

1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308
		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;
1309 1310 1311 1312
}

/*
 * getOperators:
1313
 *	  read all operators in the system catalogs and return them in the
1314 1315
 * OprInfo* structure
 *
1316 1317
 *	numOprs is set to the number of operators read in
 *
1318 1319
 *
 */
1320
OprInfo    *
1321 1322
getOperators(int *numOprs)
{
1323
	PGresult   *res;
1324 1325
	int			ntups;
	int			i;
1326
	PQExpBuffer query = createPQExpBuffer();
1327

B
Bruce Momjian 已提交
1328
	OprInfo    *oprinfo;
1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343

	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;
1344 1345 1346 1347 1348 1349

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

B
Hi, all  
Bruce Momjian 已提交
1350
	appendPQExpBuffer(query, "SELECT pg_operator.oid, oprname, oprkind, oprcode, "
1351
			   "oprleft, oprright, oprcom, oprnegate, oprrest, oprjoin, "
1352 1353 1354
					"oprcanhash, oprlsortop, oprrsortop, "
					"(select usename from pg_user where oprowner = usesysid) as usename "
					"from pg_operator");
1355

B
Hi, all  
Bruce Momjian 已提交
1356
	res = PQexec(g_conn, query->data);
1357 1358 1359
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1360
		fprintf(stderr, "getOperators(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399
		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));
1400 1401 1402 1403 1404

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

1405 1406 1407 1408 1409
	}

	PQclear(res);

	return oprinfo;
1410 1411
}

1412
void
1413
clearTypeInfo(TypeInfo *tp, int numTypes)
1414
{
1415
	int			i;
1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448

	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);
1449 1450 1451
}

void
1452
clearFuncInfo(FuncInfo *fun, int numFuncs)
1453
{
1454 1455
	int			i,
				a;
1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466

	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);
1467
		for (a = 0; a < FUNC_MAX_ARGS; ++a)
1468 1469 1470 1471 1472 1473 1474 1475 1476 1477
			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);
1478 1479
}

1480
static void
1481
clearTableInfo(TableInfo *tblinfo, int numTables)
1482
{
1483 1484
	int			i,
				j;
1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504

	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 已提交
1505 1506
			if (tblinfo[i].attoids[j])
				free(tblinfo[i].attoids[j]);
1507 1508 1509 1510 1511
			if (tblinfo[i].attnames[j])
				free(tblinfo[i].attnames[j]);
			if (tblinfo[i].typnames[j])
				free(tblinfo[i].typnames[j]);
		}
B
Bruce Momjian 已提交
1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527

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

1528 1529
		if (tblinfo[i].atttypmod)
			free((int *) tblinfo[i].atttypmod);
1530 1531 1532 1533
		if (tblinfo[i].inhAttrs)
			free((int *) tblinfo[i].inhAttrs);
		if (tblinfo[i].attnames)
			free(tblinfo[i].attnames);
1534 1535
		if (tblinfo[i].atttypedefns)
			free(tblinfo[i].atttypedefns);
1536 1537 1538 1539
		if (tblinfo[i].typnames)
			free(tblinfo[i].typnames);
		if (tblinfo[i].notnull)
			free(tblinfo[i].notnull);
1540 1541
		if (tblinfo[i].primary_key)
			free(tblinfo[i].primary_key);
1542 1543
	}
	free(tblinfo);
1544 1545
}

1546
void
1547
clearInhInfo(InhInfo *inh, int numInherits)
1548
{
1549
	int			i;
1550 1551 1552 1553 1554

	if (!inh)
		return;
	for (i = 0; i < numInherits; ++i)
	{
1555 1556
		if (inh[i].inhrelid)
			free(inh[i].inhrelid);
1557 1558 1559 1560
		if (inh[i].inhparent)
			free(inh[i].inhparent);
	}
	free(inh);
1561 1562 1563
}

void
1564
clearOprInfo(OprInfo *opr, int numOprs)
1565
{
1566
	int			i;
1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601

	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);
1602 1603 1604
}

void
1605
clearIndInfo(IndInfo *ind, int numIndices)
1606
{
1607 1608
	int			i,
				a;
1609 1610 1611 1612 1613

	if (!ind)
		return;
	for (i = 0; i < numIndices; ++i)
	{
B
Bruce,  
Bruce Momjian 已提交
1614 1615
		if (ind[i].indoid)
			free(ind[i].indoid);
1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634
		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);
1635 1636 1637
}

void
B
Bruce Momjian 已提交
1638
clearAggInfo(AggInfo *agginfo, int numArgs)
1639
{
1640
	int			i;
1641 1642 1643 1644 1645 1646 1647 1648 1649

	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);
1650 1651
		if (agginfo[i].aggtransfn)
			free(agginfo[i].aggtransfn);
1652 1653
		if (agginfo[i].aggfinalfn)
			free(agginfo[i].aggfinalfn);
1654 1655
		if (agginfo[i].aggtranstype)
			free(agginfo[i].aggtranstype);
1656 1657
		if (agginfo[i].aggbasetype)
			free(agginfo[i].aggbasetype);
1658 1659
		if (agginfo[i].agginitval)
			free(agginfo[i].agginitval);
1660 1661 1662 1663
		if (agginfo[i].usename)
			free(agginfo[i].usename);
	}
	free(agginfo);
1664
}
1665 1666 1667

/*
 * getAggregates:
1668
 *	  read all the user-defined aggregates in the system catalogs and
1669 1670
 * return them in the AggInfo* structure
 *
1671 1672
 * numAggs is set to the number of aggregates read in
 *
1673 1674
 *
 */
1675
AggInfo    *
1676 1677
getAggregates(int *numAggs)
{
B
Bruce Momjian 已提交
1678
	PGresult   *res;
1679 1680
	int			ntups;
	int			i;
1681
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1682
	AggInfo    *agginfo;
1683 1684 1685

	int			i_oid;
	int			i_aggname;
1686
	int			i_aggtransfn;
1687
	int			i_aggfinalfn;
1688
	int			i_aggtranstype;
1689
	int			i_aggbasetype;
1690
	int			i_agginitval;
1691
	int			i_usename;
1692 1693 1694

	/* find all user-defined aggregates */

B
Hi, all  
Bruce Momjian 已提交
1695
	appendPQExpBuffer(query,
1696 1697 1698 1699 1700
						"SELECT pg_aggregate.oid, aggname, aggtransfn, "
						"aggfinalfn, aggtranstype, aggbasetype, "
						"agginitval, "
						"(select usename from pg_user where aggowner = usesysid) as usename "
						"from pg_aggregate" );
1701

B
Hi, all  
Bruce Momjian 已提交
1702
	res = PQexec(g_conn, query->data);
1703 1704 1705
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1706 1707
		fprintf(stderr, "getAggregates(): SELECT failed.  Explanation from backend: '%s'.\n",
					PQerrorMessage(g_conn));
1708 1709 1710 1711 1712 1713 1714 1715 1716 1717
		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");
1718
	i_aggtransfn = PQfnumber(res, "aggtransfn");
1719
	i_aggfinalfn = PQfnumber(res, "aggfinalfn");
1720
	i_aggtranstype = PQfnumber(res, "aggtranstype");
1721
	i_aggbasetype = PQfnumber(res, "aggbasetype");
1722
	i_agginitval = PQfnumber(res, "agginitval");
1723 1724 1725 1726 1727 1728
	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));
1729
		agginfo[i].aggtransfn = strdup(PQgetvalue(res, i, i_aggtransfn));
1730
		agginfo[i].aggfinalfn = strdup(PQgetvalue(res, i, i_aggfinalfn));
1731
		agginfo[i].aggtranstype = strdup(PQgetvalue(res, i, i_aggtranstype));
1732
		agginfo[i].aggbasetype = strdup(PQgetvalue(res, i, i_aggbasetype));
1733
		agginfo[i].agginitval = strdup(PQgetvalue(res, i, i_agginitval));
1734
		agginfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1735 1736 1737 1738
		if (strlen(agginfo[i].usename) == 0)
			fprintf(stderr, "WARNING: owner of aggregate '%s' appears to be invalid\n",
						agginfo[i].aggname);

1739 1740 1741 1742 1743
	}

	PQclear(res);

	return agginfo;
1744 1745 1746 1747
}

/*
 * getFuncs:
1748
 *	  read all the user-defined functions in the system catalogs and
1749 1750
 * return them in the FuncInfo* structure
 *
1751 1752
 * numFuncs is set to the number of functions read in
 *
1753 1754
 *
 */
1755
FuncInfo   *
1756 1757
getFuncs(int *numFuncs)
{
1758
	PGresult   *res;
1759 1760
	int			ntups;
	int			i;
1761 1762
	PQExpBuffer query = createPQExpBuffer();
	FuncInfo   *finfo;
1763 1764 1765 1766 1767 1768 1769 1770 1771 1772

	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;
1773
	int			i_iscachable;
1774
	int			i_isstrict;
1775
	int			i_usename;
1776 1777 1778

	/* find all user-defined funcs */

B
Hi, all  
Bruce Momjian 已提交
1779
	appendPQExpBuffer(query,
1780
		   "SELECT pg_proc.oid, proname, prolang, pronargs, prorettype, "
1781 1782
					  "proretset, proargtypes, prosrc, probin, "
					  "(select usename from pg_user where proowner = usesysid) as usename, "
1783
					  "proiscachable, proisstrict "
1784 1785
					  "from pg_proc "
				 "where pg_proc.oid > '%u'::oid",
1786
					  g_last_builtin_oid);
1787

B
Hi, all  
Bruce Momjian 已提交
1788
	res = PQexec(g_conn, query->data);
1789 1790 1791
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1792 1793
		fprintf(stderr, "getFuncs(): SELECT failed.  Explanation from backend: '%s'.\n", 
					PQerrorMessage(g_conn));
1794 1795 1796 1797 1798 1799 1800 1801 1802
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numFuncs = ntups;

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

1803 1804
	memset((char *) finfo, 0, ntups * sizeof(FuncInfo));

1805 1806 1807 1808 1809 1810 1811 1812 1813
	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");
1814
	i_iscachable = PQfnumber(res, "proiscachable");
1815
	i_isstrict = PQfnumber(res, "proisstrict");
1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828
	i_usename = PQfnumber(res, "usename");

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

		finfo[i].prosrc = checkForQuote(PQgetvalue(res, i, i_prosrc));
		finfo[i].probin = strdup(PQgetvalue(res, i, i_probin));

		finfo[i].prorettype = strdup(PQgetvalue(res, i, i_prorettype));
		finfo[i].retset = (strcmp(PQgetvalue(res, i, i_proretset), "t") == 0);
		finfo[i].nargs = atoi(PQgetvalue(res, i, i_pronargs));
1829
		finfo[i].lang = atoi(PQgetvalue(res, i, i_prolang));
1830
		finfo[i].usename = strdup(PQgetvalue(res, i, i_usename));
1831
		finfo[i].iscachable = (strcmp(PQgetvalue(res, i, i_iscachable),"t") == 0);
1832
		finfo[i].isstrict = (strcmp(PQgetvalue(res, i, i_isstrict),"t") == 0);
1833 1834 1835 1836 1837

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

1838 1839 1840 1841
		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);
1842
			exit(1);
1843 1844 1845 1846
		}
		parseNumericArray(PQgetvalue(res, i, i_proargtypes),
						  finfo[i].argtypes,
						  finfo[i].nargs);
1847 1848 1849 1850 1851 1852
		finfo[i].dumped = 0;
	}

	PQclear(res);

	return finfo;
1853 1854 1855 1856 1857

}

/*
 * getTables
1858
 *	  read all the user-defined tables (no indices, no catalogs)
1859 1860
 * in the system catalogs return them in the TableInfo* structure
 *
1861 1862
 * numTables is set to the number of tables read in
 *
1863 1864
 *
 */
1865
TableInfo  *
V
Vadim B. Mikheev 已提交
1866
getTables(int *numTables, FuncInfo *finfo, int numFuncs)
1867
{
1868
	PGresult   *res;
1869 1870
	int			ntups;
	int			i;
1871
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
1872
	PQExpBuffer delqry = createPQExpBuffer();
1873
	TableInfo  *tblinfo;
1874

B
Bruce Momjian 已提交
1875
	int			i_reloid;
1876 1877 1878 1879
	int			i_relname;
	int			i_relkind;
	int			i_relacl;
	int			i_usename;
V
Vadim B. Mikheev 已提交
1880 1881
	int			i_relchecks;
	int			i_reltriggers;
1882
	int			i_relhasindex;
1883 1884 1885 1886 1887

	char		relkindview[2];

	relkindview[0] = RELKIND_VIEW;
	relkindview[1] = '\0';
1888 1889 1890 1891 1892 1893

	/*
	 * 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*
	 *
1894
	 * we ignore tables that are not type 'r' (ordinary relation) or 'S'
1895
	 * (sequence) or 'v' (view).
1896 1897
	 */

B
Hi, all  
Bruce Momjian 已提交
1898
	appendPQExpBuffer(query,
1899 1900
			   "SELECT pg_class.oid, relname, relkind, relacl, "
					  "(select usename from pg_user where relowner = usesysid) as usename, "
1901
					  "relchecks, reltriggers, relhasindex "
1902 1903
					  "from pg_class "
					  "where relname !~ '^pg_' "
1904 1905 1906
					  "and relkind in ('%c', '%c', '%c') "
					  "order by oid",
				RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW);
1907

B
Hi, all  
Bruce Momjian 已提交
1908
	res = PQexec(g_conn, query->data);
1909 1910 1911
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
1912 1913
		fprintf(stderr, "getTables(): SELECT failed.  Explanation from backend: '%s'.\n", 
					PQerrorMessage(g_conn));
1914 1915 1916 1917 1918 1919 1920 1921 1922
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numTables = ntups;

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

B
Bruce Momjian 已提交
1923
	i_reloid = PQfnumber(res, "oid");
1924 1925 1926 1927
	i_relname = PQfnumber(res, "relname");
	i_relkind = PQfnumber(res, "relkind");
	i_relacl = PQfnumber(res, "relacl");
	i_usename = PQfnumber(res, "usename");
V
Vadim B. Mikheev 已提交
1928 1929
	i_relchecks = PQfnumber(res, "relchecks");
	i_reltriggers = PQfnumber(res, "reltriggers");
B
Hi, all  
Bruce Momjian 已提交
1930
	i_relhasindex = PQfnumber(res, "relhasindex");
1931 1932 1933

	for (i = 0; i < ntups; i++)
	{
B
Bruce Momjian 已提交
1934
		tblinfo[i].oid = strdup(PQgetvalue(res, i, i_reloid));
1935 1936 1937 1938
		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 已提交
1939 1940
		tblinfo[i].ncheck = atoi(PQgetvalue(res, i, i_relchecks));
		tblinfo[i].ntrig = atoi(PQgetvalue(res, i, i_reltriggers));
1941

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

1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986
		/* Get view definition */
		if (strcmp(PQgetvalue(res, i, i_relkind), relkindview) == 0)
		{
			PGresult   *res2;

			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "SELECT pg_get_viewdef('%s') as viewdef ", tblinfo[i].relname);
			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 已提交
1987 1988 1989
		/*
		 * Exclude inherited CHECKs from CHECK constraints total. If a
		 * constraint matches by name and condition with a constraint
1990 1991 1992 1993
		 * belonging to a parent class, we assume it was inherited.
		 */
		if (tblinfo[i].ncheck > 0)
		{
B
Bruce Momjian 已提交
1994
			PGresult   *res2;
1995 1996
			int			ntups2;

B
Bruce Momjian 已提交
1997
			if (g_verbose)
1998 1999 2000 2001 2002 2003
				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 已提交
2004 2005
			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "SELECT rcname from pg_relcheck, pg_inherits as i "
2006 2007 2008 2009 2010 2011 2012 2013
							  "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 已提交
2014
			res2 = PQexec(g_conn, query->data);
2015 2016 2017
			if (!res2 ||
				PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
B
Bruce Momjian 已提交
2018 2019
				fprintf(stderr, "getTables(): SELECT (for inherited CHECK) failed.  "
							"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034
				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 已提交
2035 2036 2037
		if (tblinfo[i].ncheck > 0)
		{
			PGresult   *res2;
2038 2039
			int			i_rcname,
						i_rcsrc;
V
Vadim B. Mikheev 已提交
2040 2041
			int			ntups2;
			int			i2;
2042

V
Vadim B. Mikheev 已提交
2043
			if (g_verbose)
2044
				fprintf(stderr, "%s finding CHECK constraints for relation: '%s' %s\n",
V
Vadim B. Mikheev 已提交
2045 2046 2047
						g_comment_start,
						tblinfo[i].relname,
						g_comment_end);
2048

B
Hi, all  
Bruce Momjian 已提交
2049 2050
			resetPQExpBuffer(query);
			appendPQExpBuffer(query, "SELECT rcname, rcsrc from pg_relcheck "
2051 2052 2053 2054 2055 2056 2057 2058
							  "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 已提交
2059
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
2060
			if (!res2 ||
2061
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2062
			{
B
Bruce Momjian 已提交
2063 2064
				fprintf(stderr, "getTables(): SELECT (for CHECK) failed.  "
						"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
V
Vadim B. Mikheev 已提交
2065 2066 2067 2068 2069
				exit_nicely(g_conn);
			}
			ntups2 = PQntuples(res2);
			if (ntups2 != tblinfo[i].ncheck)
			{
2070
				fprintf(stderr, "getTables(): relation '%s': %d CHECKs were expected, but got %d\n",
2071
						tblinfo[i].relname, tblinfo[i].ncheck, ntups2);
V
Vadim B. Mikheev 已提交
2072 2073 2074 2075
				exit_nicely(g_conn);
			}
			i_rcname = PQfnumber(res2, "rcname");
			i_rcsrc = PQfnumber(res2, "rcsrc");
2076
			tblinfo[i].check_expr = (char **) malloc(ntups2 * sizeof(char *));
V
Vadim B. Mikheev 已提交
2077 2078
			for (i2 = 0; i2 < ntups2; i2++)
			{
B
Bruce Momjian 已提交
2079 2080
				const char *name = PQgetvalue(res2, i2, i_rcname);
				const char *expr = PQgetvalue(res2, i2, i_rcsrc);
2081

B
Hi, all  
Bruce Momjian 已提交
2082
				resetPQExpBuffer(query);
2083 2084
				if (name[0] != '$')
				{
2085 2086
					appendPQExpBuffer(query, "CONSTRAINT %s ",
									  fmtId(name, force_quotes));
B
Bruce Momjian 已提交
2087
				}
2088
				appendPQExpBuffer(query, "CHECK (%s)", expr);
B
Hi, all  
Bruce Momjian 已提交
2089
				tblinfo[i].check_expr[i2] = strdup(query->data);
V
Vadim B. Mikheev 已提交
2090 2091 2092 2093 2094
			}
			PQclear(res2);
		}
		else
			tblinfo[i].check_expr = NULL;
2095

2096 2097
		/* Get primary key */
		if (strcmp(PQgetvalue(res, i, i_relhasindex), "t") == 0)
2098
		{
2099 2100 2101
			PGresult   *res2;
			char		str[INDEX_MAX_KEYS * (NAMEDATALEN * 2 + 4) + 1];
			int			j;
B
Bruce Momjian 已提交
2102

B
Hi, all  
Bruce Momjian 已提交
2103
			resetPQExpBuffer(query);
B
Bruce Momjian 已提交
2104
			appendPQExpBuffer(query,
2105 2106 2107 2108 2109 2110
							  "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);
2111
			res2 = PQexec(g_conn, query->data);
B
Bruce Momjian 已提交
2112 2113 2114
			if (!res2 || PQresultStatus(res2) != PGRES_TUPLES_OK)
			{
				fprintf(stderr, "getTables(): SELECT (for PRIMARY KEY) failed.  Explanation from backend: %s",
2115
						PQerrorMessage(g_conn));
B
Bruce Momjian 已提交
2116 2117
				exit_nicely(g_conn);
			}
2118

2119 2120 2121
			str[0] = '\0';
			for (j = 0; j < PQntuples(res2); j++)
			{
2122
				if (strlen(str) > 0)
2123 2124 2125 2126
					strcat(str, ", ");
				strcat(str, fmtId(PQgetvalue(res2, j, 0), force_quotes));
			}

2127 2128
			if (strlen(str) > 0)
			{
2129
				tblinfo[i].primary_key = strdup(str);
2130 2131
				if (tblinfo[i].primary_key == NULL)
				{
2132 2133 2134 2135 2136 2137
					perror("strdup");
					exit(1);
				}
			}
			else
				tblinfo[i].primary_key = NULL;
2138
		}
2139 2140
		else
			tblinfo[i].primary_key = NULL;
B
Bruce Momjian 已提交
2141

V
Vadim B. Mikheev 已提交
2142 2143 2144 2145
		/* Get Triggers */
		if (tblinfo[i].ntrig > 0)
		{
			PGresult   *res2;
B
Bruce,  
Bruce Momjian 已提交
2146
			int			i_tgoid,
2147
						i_tgname,
2148 2149 2150
						i_tgfoid,
						i_tgtype,
						i_tgnargs,
2151 2152 2153 2154 2155
						i_tgargs,
						i_tgisconstraint,
						i_tgconstrname,
						i_tgdeferrable,
						i_tginitdeferred;
V
Vadim B. Mikheev 已提交
2156 2157
			int			ntups2;
			int			i2;
2158

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

B
Hi, all  
Bruce Momjian 已提交
2165
			resetPQExpBuffer(query);
2166
			appendPQExpBuffer(query, "SELECT tgname, tgfoid, tgtype, tgnargs, tgargs, tgisconstraint, tgconstrname, tgdeferrable, tginitdeferred, oid "
2167 2168 2169
							  "from pg_trigger "
							  "where tgrelid = '%s'::oid ",
							  tblinfo[i].oid);
B
Hi, all  
Bruce Momjian 已提交
2170
			res2 = PQexec(g_conn, query->data);
V
Vadim B. Mikheev 已提交
2171
			if (!res2 ||
2172
				PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2173
			{
B
Bruce Momjian 已提交
2174 2175
				fprintf(stderr, "getTables(): SELECT (for TRIGGER) failed.  "
							"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
V
Vadim B. Mikheev 已提交
2176 2177 2178 2179 2180
				exit_nicely(g_conn);
			}
			ntups2 = PQntuples(res2);
			if (ntups2 != tblinfo[i].ntrig)
			{
2181
				fprintf(stderr, "getTables(): relation '%s': %d Triggers were expected, but got %d\n",
2182
						tblinfo[i].relname, tblinfo[i].ntrig, ntups2);
V
Vadim B. Mikheev 已提交
2183 2184 2185 2186 2187 2188 2189
				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 已提交
2190
			i_tgoid = PQfnumber(res2, "oid");
2191 2192 2193 2194 2195
			i_tgisconstraint = PQfnumber(res2, "tgisconstraint");
			i_tgconstrname = PQfnumber(res2, "tgconstrname");
			i_tgdeferrable = PQfnumber(res2, "tgdeferrable");
			i_tginitdeferred = PQfnumber(res2, "tginitdeferred");

B
Bruce Momjian 已提交
2196
			tblinfo[i].triggers = (TrigInfo*) malloc(ntups2 * sizeof(TrigInfo));
B
Hi, all  
Bruce Momjian 已提交
2197 2198
			resetPQExpBuffer(query);
			for (i2 = 0; i2 < ntups2; i2++)
V
Vadim B. Mikheev 已提交
2199
			{
2200
				const char *tgfuncoid = PQgetvalue(res2, i2, i_tgfoid);
2201
				char	   *tgfunc = NULL;
2202 2203
				int2		tgtype = atoi(PQgetvalue(res2, i2, i_tgtype));
				int			tgnargs = atoi(PQgetvalue(res2, i2, i_tgnargs));
B
Bruce Momjian 已提交
2204
				const char *tgargs = PQgetvalue(res2, i2, i_tgargs);
2205 2206 2207
				int			tgisconstraint;
				int			tgdeferrable;
				int			tginitdeferred;
B
Bruce Momjian 已提交
2208
				const char *p;
2209 2210
				int			findx;

2211
				if (strcmp(PQgetvalue(res2, i2, i_tgisconstraint), "f") == 0)
2212
					tgisconstraint = 0;
2213
				else
2214
					tgisconstraint = 1;
2215 2216

				if (strcmp(PQgetvalue(res2, i2, i_tgdeferrable), "f") == 0)
2217
					tgdeferrable = 0;
2218
				else
2219
					tgdeferrable = 1;
2220 2221

				if (strcmp(PQgetvalue(res2, i2, i_tginitdeferred), "f") == 0)
2222
					tginitdeferred = 0;
2223
				else
2224
					tginitdeferred = 1;
2225

V
Vadim B. Mikheev 已提交
2226 2227
				for (findx = 0; findx < numFuncs; findx++)
				{
2228
					if (strcmp(finfo[findx].oid, tgfuncoid) == 0 &&
2229
						finfo[findx].nargs == 0 &&
V
Vadim B. Mikheev 已提交
2230 2231 2232
						strcmp(finfo[findx].prorettype, "0") == 0)
						break;
				}
2233

V
Vadim B. Mikheev 已提交
2234 2235
				if (findx == numFuncs)
				{
2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256
					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);
					}
2257
					tgfunc = strdup(PQgetvalue(r, 0, PQfnumber(r, "proname")));
2258
					PQclear(r);
2259
				}
2260
				else
2261
					tgfunc = strdup(finfo[findx].proname);
B
Bruce Momjian 已提交
2262 2263

				appendPQExpBuffer(delqry, "DROP TRIGGER %s ",
2264
									fmtId(PQgetvalue(res2, i2, i_tgname),
B
Bruce Momjian 已提交
2265 2266 2267
									force_quotes));
				appendPQExpBuffer(delqry, "ON %s;\n",
									fmtId(tblinfo[i].relname, force_quotes));
2268

B
Hi, all  
Bruce Momjian 已提交
2269
				resetPQExpBuffer(query);
2270 2271 2272 2273 2274
				if (tgisconstraint)
				{
					appendPQExpBuffer(query, "CREATE CONSTRAINT TRIGGER ");
					appendPQExpBuffer(query, fmtId(PQgetvalue(res2, i2, i_tgconstrname), force_quotes));
				}
2275 2276
				else
				{
2277 2278 2279
					appendPQExpBuffer(query, "CREATE TRIGGER ");
					appendPQExpBuffer(query, fmtId(PQgetvalue(res2, i2, i_tgname), force_quotes));
				}
B
Bruce Momjian 已提交
2280
				appendPQExpBufferChar(query, ' ');
V
Vadim B. Mikheev 已提交
2281 2282 2283
				/* Trigger type */
				findx = 0;
				if (TRIGGER_FOR_BEFORE(tgtype))
B
Hi, all  
Bruce Momjian 已提交
2284
					appendPQExpBuffer(query, "BEFORE");
V
Vadim B. Mikheev 已提交
2285
				else
B
Hi, all  
Bruce Momjian 已提交
2286
					appendPQExpBuffer(query, "AFTER");
V
Vadim B. Mikheev 已提交
2287 2288
				if (TRIGGER_FOR_INSERT(tgtype))
				{
B
Hi, all  
Bruce Momjian 已提交
2289
					appendPQExpBuffer(query, " INSERT");
V
Vadim B. Mikheev 已提交
2290 2291 2292 2293 2294
					findx++;
				}
				if (TRIGGER_FOR_DELETE(tgtype))
				{
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
2295
						appendPQExpBuffer(query, " OR DELETE");
V
Vadim B. Mikheev 已提交
2296
					else
B
Hi, all  
Bruce Momjian 已提交
2297
						appendPQExpBuffer(query, " DELETE");
V
Vadim B. Mikheev 已提交
2298 2299 2300
					findx++;
				}
				if (TRIGGER_FOR_UPDATE(tgtype))
2301
				{
V
Vadim B. Mikheev 已提交
2302
					if (findx > 0)
B
Hi, all  
Bruce Momjian 已提交
2303
						appendPQExpBuffer(query, " OR UPDATE");
V
Vadim B. Mikheev 已提交
2304
					else
B
Hi, all  
Bruce Momjian 已提交
2305
						appendPQExpBuffer(query, " UPDATE");
2306
				}
2307 2308
				appendPQExpBuffer(query, " ON %s ", fmtId(tblinfo[i].relname, force_quotes));

2309
				if (tgisconstraint)
2310
				{
2311
					if (!tgdeferrable)
2312 2313 2314 2315
						appendPQExpBuffer(query, " NOT");
					appendPQExpBuffer(query, " DEFERRABLE INITIALLY ");
					if (tginitdeferred)
						appendPQExpBuffer(query, "DEFERRED");
2316
					else
2317
						appendPQExpBuffer(query, "IMMEDIATE");
2318

2319 2320 2321
				}

				appendPQExpBuffer(query, " FOR EACH ROW");
2322 2323
				appendPQExpBuffer(query, " EXECUTE PROCEDURE %s (",
								  fmtId(tgfunc, force_quotes));
V
Vadim B. Mikheev 已提交
2324 2325
				for (findx = 0; findx < tgnargs; findx++)
				{
2326
					const char *s;
2327 2328

					for (p = tgargs;;)
V
Vadim B. Mikheev 已提交
2329
					{
2330
						p = strchr(p, '\\');
V
Vadim B. Mikheev 已提交
2331 2332
						if (p == NULL)
						{
B
Bruce Momjian 已提交
2333 2334
							fprintf(stderr, "getTables(): relation '%s': bad argument "
									"string (%s) for trigger '%s'\n",
2335 2336 2337
									tblinfo[i].relname,
									PQgetvalue(res2, i2, i_tgargs),
									PQgetvalue(res2, i2, i_tgname));
V
Vadim B. Mikheev 已提交
2338 2339 2340 2341 2342 2343 2344 2345
							exit_nicely(g_conn);
						}
						p++;
						if (*p == '\\')
						{
							p++;
							continue;
						}
2346
						if (p[0] == '0' && p[1] == '0' && p[2] == '0')
V
Vadim B. Mikheev 已提交
2347 2348 2349
							break;
					}
					p--;
2350
					appendPQExpBufferChar(query, '\'');
B
Bruce Momjian 已提交
2351
					for (s = tgargs; s < p;)
V
Vadim B. Mikheev 已提交
2352 2353
					{
						if (*s == '\'')
2354 2355
							appendPQExpBufferChar(query, '\\');
						appendPQExpBufferChar(query, *s++);
V
Vadim B. Mikheev 已提交
2356
					}
B
Bruce Momjian 已提交
2357 2358
					appendPQExpBufferChar(query, '\'');
					appendPQExpBuffer(query, (findx < tgnargs - 1) ? ", " : "");
V
Vadim B. Mikheev 已提交
2359 2360
					tgargs = p + 4;
				}
B
Hi, all  
Bruce Momjian 已提交
2361
				appendPQExpBuffer(query, ");\n");
2362

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

				/*** Initialize trcomments and troids ***/
2366

B
Bruce,  
Bruce Momjian 已提交
2367
				resetPQExpBuffer(query);
2368 2369
				appendPQExpBuffer(query, "TRIGGER %s ",
					fmtId(PQgetvalue(res2, i2, i_tgname), force_quotes));
B
Bruce,  
Bruce Momjian 已提交
2370
				appendPQExpBuffer(query, "ON %s",
2371
								fmtId(tblinfo[i].relname, force_quotes));
B
Bruce Momjian 已提交
2372 2373 2374 2375
				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 已提交
2376

2377 2378
				if (tgfunc)
					free(tgfunc);
V
Vadim B. Mikheev 已提交
2379 2380 2381 2382
			}
			PQclear(res2);
		}
		else
B
Bruce,  
Bruce Momjian 已提交
2383
		{
V
Vadim B. Mikheev 已提交
2384
			tblinfo[i].triggers = NULL;
B
Bruce,  
Bruce Momjian 已提交
2385
		}
2386

2387 2388 2389 2390 2391
	}

	PQclear(res);

	return tblinfo;
2392 2393 2394 2395 2396

}

/*
 * getInherits
2397
 *	  read all the inheritance information
2398 2399
 * from the system catalogs return them in the InhInfo* structure
 *
2400 2401
 * numInherits is set to the number of tables read in
 *
2402 2403
 *
 */
2404
InhInfo    *
2405 2406
getInherits(int *numInherits)
{
2407
	PGresult   *res;
2408 2409
	int			ntups;
	int			i;
2410 2411
	PQExpBuffer query = createPQExpBuffer();
	InhInfo    *inhinfo;
2412

2413
	int			i_inhrelid;
2414
	int			i_inhparent;
2415 2416 2417

	/* find all the inheritance information */

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

B
Hi, all  
Bruce Momjian 已提交
2420
	res = PQexec(g_conn, query->data);
2421 2422 2423
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
2424 2425
		fprintf(stderr, "getInherits(): SELECT failed.  Explanation from backend: '%s'.\n", 
				PQerrorMessage(g_conn));
2426 2427 2428 2429 2430 2431 2432 2433 2434
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numInherits = ntups;

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

2435
	i_inhrelid = PQfnumber(res, "inhrelid");
2436 2437 2438 2439
	i_inhparent = PQfnumber(res, "inhparent");

	for (i = 0; i < ntups; i++)
	{
2440
		inhinfo[i].inhrelid = strdup(PQgetvalue(res, i, i_inhrelid));
2441 2442 2443 2444 2445
		inhinfo[i].inhparent = strdup(PQgetvalue(res, i, i_inhparent));
	}

	PQclear(res);
	return inhinfo;
2446 2447 2448 2449
}

/*
 * getTableAttrs -
2450 2451
 *	  for each table in tblinfo, read its attributes types and names
 *
2452
 * this is implemented in a very inefficient way right now, looping
2453
 * through the tblinfo and doing a join per table to find the attrs and their
2454 2455
 * types
 *
2456
 *	modifies tblinfo
2457 2458
 */
void
2459
getTableAttrs(TableInfo *tblinfo, int numTables)
2460
{
2461 2462
	int			i,
				j;
2463
	PQExpBuffer q = createPQExpBuffer();
2464 2465
	int			i_attname;
	int			i_typname;
2466
	int			i_atttypmod;
2467
	int			i_attnotnull;
V
Vadim B. Mikheev 已提交
2468
	int			i_atthasdef;
2469
	int			i_attoid;
2470
	int			i_atttypedefn;
2471
	PGresult   *res;
2472
	int			ntups;
2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487

	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)
2488
			fprintf(stderr, "%s finding the attrs and types for table: '%s' %s\n",
2489 2490 2491 2492
					g_comment_start,
					tblinfo[i].relname,
					g_comment_end);

B
Hi, all  
Bruce Momjian 已提交
2493
		resetPQExpBuffer(q);
B
Bruce,  
Bruce Momjian 已提交
2494
		appendPQExpBuffer(q, "SELECT a.oid as attoid, a.attnum, a.attname, t.typname, a.atttypmod, "
2495
						  "a.attnotnull, a.atthasdef, format_type(a.atttypid, a.atttypmod) as atttypedefn "
2496 2497 2498 2499
						  "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 已提交
2500
		res = PQexec(g_conn, q->data);
2501 2502 2503
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
B
Bruce Momjian 已提交
2504 2505
			fprintf(stderr, "getTableAttrs(): SELECT failed.  "
					"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2506 2507 2508 2509 2510
			exit_nicely(g_conn);
		}

		ntups = PQntuples(res);

B
Bruce,  
Bruce Momjian 已提交
2511
		i_attoid = PQfnumber(res, "attoid");
2512 2513
		i_attname = PQfnumber(res, "attname");
		i_typname = PQfnumber(res, "typname");
2514
		i_atttypmod = PQfnumber(res, "atttypmod");
2515
		i_attnotnull = PQfnumber(res, "attnotnull");
V
Vadim B. Mikheev 已提交
2516
		i_atthasdef = PQfnumber(res, "atthasdef");
2517
		i_atttypedefn = PQfnumber(res, "atttypedefn");
2518 2519

		tblinfo[i].numatts = ntups;
B
Bruce,  
Bruce Momjian 已提交
2520
		tblinfo[i].attoids = (char **) malloc(ntups * sizeof(char *));
2521
		tblinfo[i].attnames = (char **) malloc(ntups * sizeof(char *));
2522
		tblinfo[i].atttypedefns = (char **) malloc(ntups * sizeof(char *));
2523
		tblinfo[i].typnames = (char **) malloc(ntups * sizeof(char *));
2524
		tblinfo[i].atttypmod = (int *) malloc(ntups * sizeof(int));
2525 2526
		tblinfo[i].inhAttrs = (int *) malloc(ntups * sizeof(int));
		tblinfo[i].notnull = (bool *) malloc(ntups * sizeof(bool));
V
Vadim B. Mikheev 已提交
2527
		tblinfo[i].adef_expr = (char **) malloc(ntups * sizeof(char *));
2528 2529 2530 2531
		tblinfo[i].parentRels = NULL;
		tblinfo[i].numParents = 0;
		for (j = 0; j < ntups; j++)
		{
B
Bruce,  
Bruce Momjian 已提交
2532
			tblinfo[i].attoids[j] = strdup(PQgetvalue(res, j, i_attoid));
2533
			tblinfo[i].attnames[j] = strdup(PQgetvalue(res, j, i_attname));
2534
			tblinfo[i].atttypedefns[j] = strdup(PQgetvalue(res, j, i_atttypedefn));
2535
			tblinfo[i].typnames[j] = strdup(PQgetvalue(res, j, i_typname));
2536
			tblinfo[i].atttypmod[j] = atoi(PQgetvalue(res, j, i_atttypmod));
2537 2538 2539
			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 已提交
2540 2541 2542
			if (PQgetvalue(res, j, i_atthasdef)[0] == 't')
			{
				PGresult   *res2;
2543

V
Vadim B. Mikheev 已提交
2544
				if (g_verbose)
2545
					fprintf(stderr, "%s finding DEFAULT expression for attr: '%s' %s\n",
V
Vadim B. Mikheev 已提交
2546 2547 2548
							g_comment_start,
							tblinfo[i].attnames[j],
							g_comment_end);
2549

B
Hi, all  
Bruce Momjian 已提交
2550 2551
				resetPQExpBuffer(q);
				appendPQExpBuffer(q, "SELECT adsrc from pg_attrdef "
2552 2553
							 "where adrelid = '%s'::oid and adnum = %d ",
								  tblinfo[i].oid, j + 1);
B
Hi, all  
Bruce Momjian 已提交
2554
				res2 = PQexec(g_conn, q->data);
V
Vadim B. Mikheev 已提交
2555
				if (!res2 ||
2556
					PQresultStatus(res2) != PGRES_TUPLES_OK)
V
Vadim B. Mikheev 已提交
2557
				{
B
Bruce Momjian 已提交
2558 2559
					fprintf(stderr, "getTableAttrs(): SELECT (for DEFAULT) failed.  "
							"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
V
Vadim B. Mikheev 已提交
2560 2561 2562 2563 2564 2565 2566
					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;
2567 2568 2569
		}
		PQclear(res);
	}
2570 2571 2572 2573 2574
}


/*
 * getIndices
2575
 *	  read all the user-defined indices information
2576 2577
 * from the system catalogs return them in the InhInfo* structure
 *
2578 2579
 * numIndices is set to the number of indices read in
 *
2580 2581
 *
 */
2582
IndInfo    *
2583 2584
getIndices(int *numIndices)
{
2585
	int			i;
2586
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
2587
	PGresult   *res;
2588
	int			ntups;
B
Bruce Momjian 已提交
2589
	IndInfo    *indinfo;
2590 2591 2592 2593 2594 2595 2596 2597

	int			i_indexrelname;
	int			i_indrelname;
	int			i_indamname;
	int			i_indproc;
	int			i_indkey;
	int			i_indclass;
	int			i_indisunique;
2598
	int			i_indoid;
2599 2600 2601 2602 2603

	/*
	 * find all the user-defined indices. We do not handle partial
	 * indices.
	 *
2604
	 * Notice we skip indices on system classes
2605 2606 2607 2608
	 *
	 * this is a 4-way join !!
	 */

B
Hi, all  
Bruce Momjian 已提交
2609
	appendPQExpBuffer(query,
2610 2611 2612 2613 2614 2615
					  "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 "
2616 2617
					  "and t2.relname !~ '^pg_' and not i.indisprimary",
					  g_last_builtin_oid);
2618

B
Hi, all  
Bruce Momjian 已提交
2619
	res = PQexec(g_conn, query->data);
2620 2621 2622
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
2623 2624
		fprintf(stderr, "getIndices(): SELECT failed.  "
					"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2625 2626 2627 2628 2629 2630 2631 2632 2633
		exit_nicely(g_conn);
	}

	ntups = PQntuples(res);

	*numIndices = ntups;

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

2634 2635
	memset((char *) indinfo, 0, ntups * sizeof(IndInfo));

B
Bruce,  
Bruce Momjian 已提交
2636
	i_indoid = PQfnumber(res, "indoid");
2637 2638 2639 2640 2641 2642 2643 2644 2645 2646
	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 已提交
2647
		indinfo[i].indoid = strdup(PQgetvalue(res, i, i_indoid));
2648 2649 2650 2651
		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));
2652 2653 2654 2655 2656 2657
		parseNumericArray(PQgetvalue(res, i, i_indkey),
						  indinfo[i].indkey,
						  INDEX_MAX_KEYS);
		parseNumericArray(PQgetvalue(res, i, i_indclass),
						  indinfo[i].indclass,
						  INDEX_MAX_KEYS);
2658 2659 2660 2661
		indinfo[i].indisunique = strdup(PQgetvalue(res, i, i_indisunique));
	}
	PQclear(res);
	return indinfo;
2662 2663
}

B
Bruce,  
Bruce Momjian 已提交
2664
/*------------------------------------------------------------------
2665
 * dumpComments --
B
Bruce,  
Bruce Momjian 已提交
2666
 *
2667
 * This routine is used to dump any comments associated with the
B
Bruce,  
Bruce Momjian 已提交
2668 2669 2670 2671
 * 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
2672
 * addition, the routine takes the stdio FILE handle to which the
B
Bruce,  
Bruce Momjian 已提交
2673 2674 2675 2676
 * output should be written.
 *------------------------------------------------------------------
*/

T
Tom Lane 已提交
2677
static void
B
Bruce Momjian 已提交
2678
dumpComment(Archive *fout, const char *target, const char *oid)
2679
{
B
Bruce,  
Bruce Momjian 已提交
2680

2681
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
2682
	PQExpBuffer query;
2683
	int			i_description;
B
Bruce,  
Bruce Momjian 已提交
2684 2685 2686 2687 2688 2689 2690 2691 2692 2693

	/*** 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);
2694 2695
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce,  
Bruce Momjian 已提交
2696 2697 2698 2699 2700 2701 2702
		fprintf(stderr, "DumpComment: SELECT failed: '%s'.\n",
				PQerrorMessage(g_conn));
		exit_nicely(g_conn);
	}

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

2703 2704
	if (PQntuples(res) != 0)
	{
B
Bruce,  
Bruce Momjian 已提交
2705
		i_description = PQfnumber(res, "description");
B
Bruce Momjian 已提交
2706 2707 2708 2709 2710
		resetPQExpBuffer(query);
		appendPQExpBuffer(query, "COMMENT ON %s IS '%s';\n",
							target, checkForQuote(PQgetvalue(res, 0, i_description)));

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

B
Bruce,  
Bruce Momjian 已提交
2713 2714 2715 2716 2717
	}

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

	PQclear(res);
2718

B
Bruce,  
Bruce Momjian 已提交
2719 2720 2721
}

/*------------------------------------------------------------------
2722
 * dumpDBComment --
B
Bruce,  
Bruce Momjian 已提交
2723
 *
2724 2725
 * 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 已提交
2726 2727 2728 2729 2730
 * to dump the schema of the database, then this is the first
 * statement issued.
 *------------------------------------------------------------------
*/

2731
void
B
Bruce Momjian 已提交
2732
dumpDBComment(Archive *fout)
2733
{
B
Bruce,  
Bruce Momjian 已提交
2734

2735
	PGresult   *res;
B
Bruce,  
Bruce Momjian 已提交
2736
	PQExpBuffer query;
2737
	int			i_oid;
B
Bruce,  
Bruce Momjian 已提交
2738 2739 2740 2741 2742 2743 2744 2745 2746 2747

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

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

	/*** Execute query ***/

	res = PQexec(g_conn, query->data);
2748 2749
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce,  
Bruce Momjian 已提交
2750 2751 2752 2753 2754 2755 2756
		fprintf(stderr, "dumpDBComment: SELECT failed: '%s'.\n",
				PQerrorMessage(g_conn));
		exit_nicely(g_conn);
	}

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

2757 2758
	if (PQntuples(res) != 0)
	{
B
Bruce,  
Bruce Momjian 已提交
2759 2760 2761 2762 2763 2764 2765 2766 2767
		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);
2768

B
Bruce,  
Bruce Momjian 已提交
2769 2770
}

2771 2772
/*
 * dumpTypes
2773
 *	  writes out to fout the queries to recreate all the user-defined types
2774 2775 2776
 *
 */
void
B
Bruce Momjian 已提交
2777
dumpTypes(Archive *fout, FuncInfo *finfo, int numFuncs,
2778
		  TypeInfo *tinfo, int numTypes)
2779
{
2780
	int			i;
2781
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
2782
	PQExpBuffer	delq = createPQExpBuffer();
2783
	int			funcInd;
2784 2785 2786 2787 2788

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

		/* skip all the builtin types */
2789
		if (atooid(tinfo[i].oid) <= g_last_builtin_oid)
2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805 2806 2807 2808 2809 2810 2811 2812
			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 已提交
2813
		appendPQExpBuffer(delq, "DROP TYPE %s;\n", fmtId(tinfo[i].typname, force_quotes));
2814

B
Hi, all  
Bruce Momjian 已提交
2815 2816
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827
						  "CREATE TYPE %s "
			   "( internallength = %s, externallength = %s, input = %s, "
				  "output = %s, send = %s, receive = %s, default = '%s'",
						  fmtId(tinfo[i].typname, force_quotes),
						  tinfo[i].typlen,
						  tinfo[i].typprtlen,
						  tinfo[i].typinput,
						  tinfo[i].typoutput,
						  tinfo[i].typsend,
						  tinfo[i].typreceive,
						  tinfo[i].typdefault);
2828 2829 2830

		if (tinfo[i].isArray)
		{
2831
			char	   *elemType;
2832

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

B
Hi, all  
Bruce Momjian 已提交
2835
			appendPQExpBuffer(q, ", element = %s, delimiter = '%s'",
2836
							  elemType, tinfo[i].typdelim);
2837 2838
		}
		if (tinfo[i].passedbyvalue)
B
Hi, all  
Bruce Momjian 已提交
2839
			appendPQExpBuffer(q, ",passedbyvalue);\n");
2840
		else
B
Hi, all  
Bruce Momjian 已提交
2841
			appendPQExpBuffer(q, ");\n");
2842

B
Bruce Momjian 已提交
2843
		ArchiveEntry(fout, tinfo[i].oid, fmtId(tinfo[i].typname, force_quotes), "TYPE", NULL,
2844
						q->data, delq->data, "", tinfo[i].usename, NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
2845 2846 2847 2848

		/*** Dump Type Comments ***/

		resetPQExpBuffer(q);
B
Bruce Momjian 已提交
2849 2850
		resetPQExpBuffer(delq);

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

B
Bruce Momjian 已提交
2854
		resetPQExpBuffer(q);
2855
	}
2856 2857
}

2858 2859
/*
 * dumpProcLangs
B
Bruce Momjian 已提交
2860
 *		  writes out to fout the queries to recreate user-defined procedural languages
2861 2862 2863
 *
 */
void
B
Bruce Momjian 已提交
2864
dumpProcLangs(Archive *fout, FuncInfo *finfo, int numFuncs,
B
Bruce Momjian 已提交
2865
			  TypeInfo *tinfo, int numTypes)
2866
{
2867 2868
	PGresult   *res;
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
2869 2870
	PQExpBuffer defqry = createPQExpBuffer();
	PQExpBuffer delqry = createPQExpBuffer();
2871
	int			ntups;
B
Bruce Momjian 已提交
2872
	int			i_oid;
2873 2874 2875 2876
	int			i_lanname;
	int			i_lanpltrusted;
	int			i_lanplcallfoid;
	int			i_lancompiler;
2877
	Oid			lanoid;
2878 2879 2880
	char	   *lanname;
	char	   *lancompiler;
	const char *lanplcallfoid;
2881 2882 2883
	int			i,
				fidx;

B
Bruce Momjian 已提交
2884
	appendPQExpBuffer(query, "SELECT oid, * FROM pg_language "
2885 2886
					  "WHERE lanispl "
					  "ORDER BY oid");
B
Hi, all  
Bruce Momjian 已提交
2887
	res = PQexec(g_conn, query->data);
2888 2889 2890
	if (!res ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
2891
		fprintf(stderr, "dumpProcLangs(): SELECT failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
2892 2893 2894 2895
		exit_nicely(g_conn);
	}
	ntups = PQntuples(res);

B
Bruce Momjian 已提交
2896 2897 2898 2899
	i_lanname = PQfnumber(res, "lanname");
	i_lanpltrusted = PQfnumber(res, "lanpltrusted");
	i_lanplcallfoid = PQfnumber(res, "lanplcallfoid");
	i_lancompiler = PQfnumber(res, "lancompiler");
B
Bruce Momjian 已提交
2900
	i_oid = PQfnumber(res, "oid");
2901

B
Bruce Momjian 已提交
2902 2903
	for (i = 0; i < ntups; i++)
	{
2904
		lanoid = atooid(PQgetvalue(res, i, i_oid));
2905 2906 2907
		if (lanoid <= g_last_builtin_oid)
			continue;

2908
		lanplcallfoid = PQgetvalue(res, i, i_lanplcallfoid);
2909 2910


2911 2912 2913 2914 2915 2916 2917
		for (fidx = 0; fidx < numFuncs; fidx++)
		{
			if (!strcmp(finfo[fidx].oid, lanplcallfoid))
				break;
		}
		if (fidx >= numFuncs)
		{
B
Bruce Momjian 已提交
2918 2919
			fprintf(stderr, "dumpProcLangs(): handler procedure for "
						"language %s not found\n", PQgetvalue(res, i, i_lanname));
2920 2921 2922 2923 2924 2925 2926 2927
			exit_nicely(g_conn);
		}

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

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

B
Bruce Momjian 已提交
2928
		appendPQExpBuffer(delqry, "DROP PROCEDURAL LANGUAGE '%s';\n", lanname);
2929

B
Bruce Momjian 已提交
2930
		appendPQExpBuffer(defqry, "CREATE %sPROCEDURAL LANGUAGE '%s' "
B
Bruce Momjian 已提交
2931
				"HANDLER %s LANCOMPILER '%s';\n",
B
Bruce Momjian 已提交
2932
				(PQgetvalue(res, i, i_lanpltrusted)[0] == 't') ? "TRUSTED " : "",
B
Bruce Momjian 已提交
2933 2934 2935
				lanname,
				fmtId(finfo[fidx].proname, force_quotes),
				lancompiler);
2936

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

2940 2941
		free(lanname);
		free(lancompiler);
2942 2943 2944

		resetPQExpBuffer(defqry);
		resetPQExpBuffer(delqry);
2945 2946 2947 2948 2949 2950
	}

	PQclear(res);

}

2951 2952
/*
 * dumpFuncs
2953
 *	  writes out to fout the queries to recreate all the user-defined functions
2954 2955 2956
 *
 */
void
B
Bruce Momjian 已提交
2957
dumpFuncs(Archive *fout, FuncInfo *finfo, int numFuncs,
2958
		  TypeInfo *tinfo, int numTypes)
2959
{
2960
	int			i;
2961 2962 2963

	for (i = 0; i < numFuncs; i++)
		dumpOneFunc(fout, finfo, i, tinfo, numTypes);
2964 2965 2966 2967
}

/*
 * dumpOneFunc:
2968 2969
 *	  dump out only one function,  the index of which is given in the third
 *	argument
2970 2971 2972
 *
 */

2973
static void
B
Bruce Momjian 已提交
2974
dumpOneFunc(Archive *fout, FuncInfo *finfo, int i,
2975
			TypeInfo *tinfo, int numTypes)
2976
{
2977
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
2978 2979
	PQExpBuffer fn = createPQExpBuffer();
	PQExpBuffer delqry = createPQExpBuffer();
B
Bruce,  
Bruce Momjian 已提交
2980
	PQExpBuffer fnlist = createPQExpBuffer();
B
Bruce Momjian 已提交
2981
	int			j;
2982
	PQExpBuffer asPart = createPQExpBuffer();
2983
	char		func_lang[NAMEDATALEN + 1];
2984 2985 2986 2987
	PGresult   *res;
	int			nlangs;
	int			i_lanname;
	char		query[256];
2988

2989 2990 2991 2992
	char		*listSep;
	char		*listSepComma = ",";
	char		*listSepNone = "";

2993 2994 2995 2996 2997
	if (finfo[i].dumped)
		return;
	else
		finfo[i].dumped = 1;

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

3000 3001 3002 3003 3004
	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 已提交
3005
  	{
3006 3007
		fprintf(stderr, "dumpOneFunc(): SELECT for procedural language failed.  Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
		exit_nicely(g_conn);
B
Bruce Momjian 已提交
3008
  	}
3009 3010 3011
	nlangs = PQntuples(res);

	if (nlangs != 1)
B
Bruce Momjian 已提交
3012
  	{
3013 3014
		fprintf(stderr, "dumpOneFunc(): procedural language for function %s not found\n", finfo[i].proname);
		exit_nicely(g_conn);
B
Bruce Momjian 已提交
3015 3016
  	}
  
3017
	i_lanname = PQfnumber(res, "lanname");
3018

3019
	/*
3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032 3033 3034 3035
	 * See backend/commands/define.c for details of how the 'AS' clause
	 * is used.
	 */
	if (strcmp(finfo[i].probin, "-") != 0)
	{
		if (strcmp(finfo[i].prosrc, "-") != 0)
			appendPQExpBuffer(asPart, "AS '%s', '%s'", finfo[i].probin, finfo[i].prosrc);
		else
			appendPQExpBuffer(asPart, "AS '%s'", finfo[i].probin);
	}
	else
	{
		if (strcmp(finfo[i].prosrc, "-") != 0)
			appendPQExpBuffer(asPart, "AS '%s'", finfo[i].prosrc);
	}

3036
	strcpy(func_lang, PQgetvalue(res, 0, i_lanname));
B
Bruce Momjian 已提交
3037
  
3038
	PQclear(res);
B
Bruce Momjian 已提交
3039 3040 3041
 
	resetPQExpBuffer(fn);
	appendPQExpBuffer(fn, "%s (", fmtId(finfo[i].proname, force_quotes));
3042 3043
	for (j = 0; j < finfo[i].nargs; j++)
	{
B
Bruce Momjian 已提交
3044
		char			*typname;
3045

3046 3047 3048 3049
		typname = findTypeByOid(tinfo, numTypes, finfo[i].argtypes[j], zeroAsOpaque);
		appendPQExpBuffer(fn, "%s%s", 
							(j > 0) ? "," : "", 
							typname);
B
Bruce,  
Bruce Momjian 已提交
3050
		appendPQExpBuffer(fnlist, "%s%s",
3051 3052
							(j > 0) ? "," : "", 
							typname );
3053
	}
B
Bruce Momjian 已提交
3054 3055 3056 3057 3058 3059 3060
	appendPQExpBuffer(fn, ")");

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

	resetPQExpBuffer(q);
	appendPQExpBuffer(q, "CREATE FUNCTION %s ", fn->data );
3061
	appendPQExpBuffer(q, "RETURNS %s%s %s LANGUAGE '%s'",
3062
					  (finfo[i].retset) ? " SETOF " : "",
3063
					  findTypeByOid(tinfo, numTypes, finfo[i].prorettype, zeroAsOpaque),
3064
					  asPart->data, func_lang);
3065

3066
	if (finfo[i].iscachable || finfo[i].isstrict) /* OR in new attrs here */
3067
	{
3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080
		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, " )");
3081 3082 3083 3084
	}

	appendPQExpBuffer(q, ";\n");

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

B
Bruce,  
Bruce Momjian 已提交
3088 3089 3090 3091 3092 3093 3094 3095
	/*** 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);

3096 3097 3098 3099
}

/*
 * dumpOprs
3100
 *	  writes out to fout the queries to recreate all the user-defined operators
3101 3102
 *
 */
3103
void
B
Bruce Momjian 已提交
3104
dumpOprs(Archive *fout, OprInfo *oprinfo, int numOperators,
3105
		 TypeInfo *tinfo, int numTypes)
3106
{
B
Bruce Momjian 已提交
3107
	int			i;
3108
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3109
	PQExpBuffer delq = createPQExpBuffer();
3110 3111 3112 3113 3114 3115 3116 3117
	PQExpBuffer leftarg = createPQExpBuffer();
	PQExpBuffer rightarg = createPQExpBuffer();
	PQExpBuffer commutator = createPQExpBuffer();
	PQExpBuffer negator = createPQExpBuffer();
	PQExpBuffer restrictor = createPQExpBuffer();
	PQExpBuffer join = createPQExpBuffer();
	PQExpBuffer sort1 = createPQExpBuffer();
	PQExpBuffer sort2 = createPQExpBuffer();
3118 3119 3120 3121

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

B
Hi, all  
Bruce Momjian 已提交
3122 3123 3124 3125 3126 3127 3128 3129 3130
		resetPQExpBuffer(leftarg);
		resetPQExpBuffer(rightarg);
		resetPQExpBuffer(commutator);
		resetPQExpBuffer(negator);
		resetPQExpBuffer(restrictor);
		resetPQExpBuffer(join);
		resetPQExpBuffer(sort1);
		resetPQExpBuffer(sort2);

3131
		/* skip all the builtin oids */
3132
		if (atooid(oprinfo[i].oid) <= g_last_builtin_oid)
3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148
			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 已提交
3149
			appendPQExpBuffer(leftarg, ",\n\tLEFTARG = %s ",
3150
								findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft, zeroAsOpaque) );
3151 3152 3153 3154
		}
		if (strcmp(oprinfo[i].oprkind, "l") == 0 ||
			strcmp(oprinfo[i].oprkind, "b") == 0)
		{
B
Hi, all  
Bruce Momjian 已提交
3155
			appendPQExpBuffer(rightarg, ",\n\tRIGHTARG = %s ",
3156
							  findTypeByOid(tinfo, numTypes, oprinfo[i].oprright, zeroAsOpaque) );
3157
		}
B
Hi, all  
Bruce Momjian 已提交
3158 3159
		if (!(strcmp(oprinfo[i].oprcom, "0") == 0))
			appendPQExpBuffer(commutator, ",\n\tCOMMUTATOR = %s ",
3160 3161
				 findOprByOid(oprinfo, numOperators, oprinfo[i].oprcom));

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

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

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

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

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

B
Bruce Momjian 已提交
3180 3181
		resetPQExpBuffer(delq);
		appendPQExpBuffer(delq, "DROP OPERATOR %s (%s", oprinfo[i].oprname,
3182
				findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft, zeroAsOpaque) );
B
Bruce Momjian 已提交
3183
		appendPQExpBuffer(delq, ", %s);\n",
3184
					findTypeByOid(tinfo, numTypes, oprinfo[i].oprright, zeroAsOpaque) );
3185

B
Hi, all  
Bruce Momjian 已提交
3186 3187
		resetPQExpBuffer(q);
		appendPQExpBuffer(q,
3188 3189 3190 3191 3192 3193 3194 3195 3196
						  "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 已提交
3197
						(strcmp(oprinfo[i].oprcanhash, "t") == 0) ? ",\n\tHASHES" : "",
3198 3199 3200
						  join->data,
						  sort1->data,
						  sort2->data);
3201

B
Bruce Momjian 已提交
3202
		ArchiveEntry(fout, oprinfo[i].oid, oprinfo[i].oprname, "OPERATOR", NULL,
3203
						q->data, delq->data, "", oprinfo[i].usename, NULL, NULL);
3204
	}
3205 3206 3207 3208
}

/*
 * dumpAggs
3209
 *	  writes out to fout the queries to create all the user-defined aggregates
3210 3211 3212
 *
 */
void
B
Bruce Momjian 已提交
3213
dumpAggs(Archive *fout, AggInfo *agginfo, int numAggs,
3214
		 TypeInfo *tinfo, int numTypes)
3215
{
B
Bruce Momjian 已提交
3216
	int			i;
3217
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3218 3219
	PQExpBuffer delq = createPQExpBuffer();
	PQExpBuffer aggSig = createPQExpBuffer();
3220
	PQExpBuffer details = createPQExpBuffer();
3221 3222 3223

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

3226
		/* skip all the builtin oids */
3227
		if (atooid(agginfo[i].oid) <= g_last_builtin_oid)
3228
			continue;
3229

3230
		appendPQExpBuffer(details,
3231
						  "BASETYPE = %s, ",
3232
						  findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsAny + useBaseTypeName));
3233

3234 3235 3236
		appendPQExpBuffer(details,
						  "SFUNC = %s, STYPE = %s",
						  agginfo[i].aggtransfn,
3237
						  findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype, zeroAsOpaque + useBaseTypeName));
3238

3239 3240 3241
		if (agginfo[i].agginitval)
			appendPQExpBuffer(details, ", INITCOND = '%s'",
							  agginfo[i].agginitval);
3242

B
Hi, all  
Bruce Momjian 已提交
3243
		if (!(strcmp(agginfo[i].aggfinalfn, "-") == 0))
3244 3245
			appendPQExpBuffer(details, ", FINALFUNC = %s",
							  agginfo[i].aggfinalfn);
3246

B
Bruce Momjian 已提交
3247 3248
		resetPQExpBuffer(aggSig);
		appendPQExpBuffer(aggSig, "%s %s", agginfo[i].aggname,
3249
							findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype, zeroAsOpaque + useBaseTypeName));
3250

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

B
Hi, all  
Bruce Momjian 已提交
3254
		resetPQExpBuffer(q);
3255
		appendPQExpBuffer(q, "CREATE AGGREGATE %s ( %s );\n",
3256
						  agginfo[i].aggname,
3257
						  details->data);
3258

B
Bruce Momjian 已提交
3259
		ArchiveEntry(fout, agginfo[i].oid, aggSig->data, "AGGREGATE", NULL,
3260
						q->data, delq->data, "", agginfo[i].usename, NULL, NULL);
B
Bruce,  
Bruce Momjian 已提交
3261 3262 3263 3264

		/*** Dump Aggregate Comments ***/

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

3269
	}
3270 3271
}

3272 3273 3274 3275 3276
/*
 * These are some support functions to fix the acl problem of pg_dump
 *
 * Matthew C. Aycock 12/02/97
 */
3277 3278 3279

/* Append a keyword to a keyword list, inserting comma if needed.
 * Caller must make aclbuf big enough for all possible keywords.
3280
 */
3281
static void
B
Bruce Momjian 已提交
3282
AddAcl(char *aclbuf, const char *keyword)
3283
{
3284 3285 3286
	if (*aclbuf)
		strcat(aclbuf, ",");
	strcat(aclbuf, keyword);
3287
}
3288

3289
/*
3290
 * This will take a string of 'arwR' and return a malloced,
3291 3292
 * comma delimited string of SELECT,INSERT,UPDATE,DELETE,RULE
 */
V
Vadim B. Mikheev 已提交
3293
static char *
3294
GetPrivileges(const char *s)
3295
{
3296
	char		aclbuf[100];
3297

3298
	aclbuf[0] = '\0';
3299

B
Bruce Momjian 已提交
3300
	if (strchr(s, 'a'))
3301
		AddAcl(aclbuf, "INSERT");
3302

B
Bruce Momjian 已提交
3303
	if (strchr(s, 'w'))
3304
		AddAcl(aclbuf, "UPDATE,DELETE");
B
Bruce Momjian 已提交
3305 3306

	if (strchr(s, 'r'))
3307
		AddAcl(aclbuf, "SELECT");
3308

3309
	if (strchr(s, 'R'))
3310
		AddAcl(aclbuf, "RULE");
3311

3312 3313 3314
	/* Special-case when they're all there */
	if (strcmp(aclbuf, "INSERT,UPDATE,DELETE,SELECT,RULE") == 0)
		return strdup("ALL");
3315

3316
	return strdup(aclbuf);
3317
}
3318

B
Bruce Momjian 已提交
3319 3320 3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334
/*
 * 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 已提交
3335 3336
/*
 * dumpACL:
3337 3338
 *	  Write out grant/revoke information
 *	  Called for sequences and tables
B
Bruce Momjian 已提交
3339 3340
 */

3341
static void
B
Bruce Momjian 已提交
3342
dumpACL(Archive *fout, TableInfo tbinfo)
B
Bruce Momjian 已提交
3343
{
B
Bruce Momjian 已提交
3344 3345
	const char 	*acls = tbinfo.relacl;
	char	   	*aclbuf,
B
Bruce Momjian 已提交
3346 3347 3348
			   *tok,
			   *eqpos,
			   *priv;
B
Bruce Momjian 已提交
3349 3350 3351
	char		*sql;
	char		tmp[1024];
	int		sSize = 4096;
3352 3353 3354 3355

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

B
Bruce Momjian 已提交
3356 3357 3358 3359 3360
	/*
	 * Allocate a larginsh buffer for the output SQL.
	 */
	sql = (char*)malloc(sSize);

B
Bruce Momjian 已提交
3361 3362 3363
	/*
	 * Revoke Default permissions for PUBLIC. Is this actually necessary,
	 * or is it just a waste of time?
3364
	 */
B
Bruce Momjian 已提交
3365
	sprintf(sql,	"REVOKE ALL on %s from PUBLIC;\n",
3366
			fmtId(tbinfo.relname, force_quotes));
3367 3368 3369

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

3371 3372
	/* Scan comma-separated ACL items */
	for (tok = strtok(aclbuf, ","); tok != NULL; tok = strtok(NULL, ","))
3373
	{
3374

B
Bruce Momjian 已提交
3375 3376 3377
		/*
		 * Token may start with '{' and/or '"'.  Actually only the start
		 * of the string should have '{', but we don't verify that.
3378 3379 3380 3381 3382 3383 3384 3385
		 */
		if (*tok == '{')
			tok++;
		if (*tok == '"')
			tok++;

		/* User name is string up to = in tok */
		eqpos = strchr(tok, '=');
B
Bruce Momjian 已提交
3386
		if (!eqpos)
B
Bruce Momjian 已提交
3387
		{
3388 3389
			fprintf(stderr, "Could not parse ACL list ('%s') for '%s'...Exiting!\n",
					acls, tbinfo.relname);
B
Bruce Momjian 已提交
3390 3391 3392
			exit_nicely(g_conn);
		}

B
Bruce Momjian 已提交
3393 3394 3395 3396
		/*
		 * Parse the privileges (right-hand side).	Skip if there are
		 * none.
		 */
3397 3398
		priv = GetPrivileges(eqpos + 1);
		if (*priv)
3399
		{
B
Bruce Momjian 已提交
3400
			sprintf(tmp,	"GRANT %s on %s to ",
3401
					priv, fmtId(tbinfo.relname, force_quotes));
B
Bruce Momjian 已提交
3402
			strcatalloc(&sql, &sSize, tmp);
B
Bruce Momjian 已提交
3403 3404 3405

			/*
			 * Note: fmtId() can only be called once per printf, so don't
3406 3407 3408 3409 3410
			 * try to merge printing of username into the above printf.
			 */
			if (eqpos == tok)
			{
				/* Empty left-hand side means "PUBLIC" */
B
Bruce Momjian 已提交
3411
				strcatalloc(&sql, &sSize, "PUBLIC;\n");
3412
			}
3413
			else
3414 3415
			{
				*eqpos = '\0';	/* it's ok to clobber aclbuf */
3416
				if (strncmp(tok, "group ", strlen("group ")) == 0)
B
Bruce Momjian 已提交
3417
					sprintf(tmp, "GROUP %s;\n",
3418 3419
							fmtId(tok + strlen("group "), force_quotes));
				else
B
Bruce Momjian 已提交
3420 3421
					sprintf(tmp, "%s;\n", fmtId(tok, force_quotes));
				strcatalloc(&sql, &sSize, tmp);
3422
			}
3423
		}
3424
		free(priv);
B
Bruce Momjian 已提交
3425
	}
3426 3427

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

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

B
Bruce Momjian 已提交
3431 3432
}

3433

3434 3435
/*
 * dumpTables:
3436
 *	  write out to fout all the user-define tables
3437
 */
3438

3439
void
B
Bruce Momjian 已提交
3440
dumpTables(Archive *fout, TableInfo *tblinfo, int numTables,
3441 3442
		   InhInfo *inhinfo, int numInherits,
		   TypeInfo *tinfo, int numTypes, const char *tablename,
B
Bruce Momjian 已提交
3443 3444
		   const bool aclsSkip, const bool oids,
		   const bool schemaOnly, const bool dataOnly)
3445
{
3446 3447
	int			i,
				j,
B
Bruce Momjian 已提交
3448
				k;
3449
	PQExpBuffer q = createPQExpBuffer();
B
Bruce Momjian 已提交
3450
	PQExpBuffer delq = createPQExpBuffer();
3451
	char	   *serialSeq = NULL;		/* implicit sequence name created
B
Bruce Momjian 已提交
3452 3453 3454 3455
										 * by SERIAL datatype */
	const char *serialSeqSuffix = "_id_seq";	/* suffix for implicit
												 * SERIAL sequences */
	char	  **parentRels;		/* list of names of parent relations */
3456
	int			numParents;
B
Bruce Momjian 已提交
3457
	int			actual_atts;	/* number of attrs in this CREATE statment */
3458
	char	   *reltypename;
3459

V
Vadim B. Mikheev 已提交
3460
	/* First - dump SEQUENCEs */
B
Bruce Momjian 已提交
3461
	if (tablename)
B
Bruce Momjian 已提交
3462 3463 3464 3465 3466
	{
		serialSeq = malloc(strlen(tablename) + strlen(serialSeqSuffix) + 1);
		strcpy(serialSeq, tablename);
		strcat(serialSeq, serialSeqSuffix);
	}
V
Vadim B. Mikheev 已提交
3467 3468 3469 3470
	for (i = 0; i < numTables; i++)
	{
		if (!(tblinfo[i].sequence))
			continue;
B
Bruce Momjian 已提交
3471
		if (!tablename || (!strcmp(tblinfo[i].relname, tablename))
B
Bruce Momjian 已提交
3472
			|| (serialSeq && !strcmp(tblinfo[i].relname, serialSeq)))
V
Vadim B. Mikheev 已提交
3473
		{
B
Bruce Momjian 已提交
3474
			/* becomeUser(fout, tblinfo[i].usename); */
V
Vadim B. Mikheev 已提交
3475
			dumpSequence(fout, tblinfo[i]);
3476
			if (!aclsSkip)
B
Bruce Momjian 已提交
3477
				dumpACL(fout, tblinfo[i]);
V
Vadim B. Mikheev 已提交
3478 3479
		}
	}
B
Bruce Momjian 已提交
3480
	if (tablename)
B
Bruce Momjian 已提交
3481
		free(serialSeq);
3482

3483 3484
	for (i = 0; i < numTables; i++)
	{
3485
		if (tblinfo[i].sequence)/* already dumped */
V
Vadim B. Mikheev 已提交
3486
			continue;
3487 3488 3489 3490

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

3491 3492
			resetPQExpBuffer(delq);
			resetPQExpBuffer(q);
3493

3494 3495 3496 3497
			/* Use the view definition if there is one */
			if (tblinfo[i].viewdef != NULL)
			{
				reltypename = "VIEW";
3498

3499 3500
				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);
3501

3502 3503
			}
			else
3504
			{
3505
				reltypename = "TABLE";
3506

3507 3508
				parentRels = tblinfo[i].parentRels;
				numParents = tblinfo[i].numParents;
B
Bruce Momjian 已提交
3509

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

3512 3513 3514 3515 3516 3517
				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)
3518
					{
3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530
						/* 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",
3531
										  tblinfo[i].adef_expr[j]);
3532 3533 3534 3535 3536 3537 3538

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

						actual_atts++;
					}
3539
				}
B
Hi all  
Bruce Momjian 已提交
3540

3541 3542 3543 3544 3545 3546 3547
				/* 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",
3548
								  tblinfo[i].check_expr[k]);
3549
				}
3550

3551 3552 3553 3554 3555 3556 3557
				/* PRIMARY KEY */
				if (tblinfo[i].primary_key)
				{
					if (actual_atts + tblinfo[i].ncheck > 0)
						appendPQExpBuffer(q, ",\n\t");
					appendPQExpBuffer(q, "PRIMARY KEY (%s)", tblinfo[i].primary_key);
				}
B
Bruce Momjian 已提交
3558

3559
				appendPQExpBuffer(q, "\n)");
3560

3561
				if (numParents > 0)
B
Hi all  
Bruce Momjian 已提交
3562
				{
3563 3564 3565 3566
					appendPQExpBuffer(q, "\ninherits (");
					for (k = 0; k < numParents; k++)
					{
						appendPQExpBuffer(q, "%s%s",
3567 3568
									  (k > 0) ? ", " : "",
									  fmtId(parentRels[k], force_quotes));
3569 3570
					}
					appendPQExpBuffer(q, ")");
B
Hi all  
Bruce Momjian 已提交
3571 3572
				}

3573 3574
				appendPQExpBuffer(q, ";\n");
			}
B
Bruce Momjian 已提交
3575 3576

			if (!dataOnly) {
3577 3578
					ArchiveEntry(fout, tblinfo[i].oid, fmtId(tblinfo[i].relname, false),
								reltypename, NULL, q->data, delq->data, "", tblinfo[i].usename,
B
Bruce Momjian 已提交
3579 3580 3581 3582
								NULL, NULL);
			}

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

3585
			/* Dump Field Comments */
B
Bruce,  
Bruce Momjian 已提交
3586

3587 3588
			for (j = 0; j < tblinfo[i].numatts; j++)
			{
B
Bruce,  
Bruce Momjian 已提交
3589 3590 3591 3592 3593 3594
				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]);
			}
3595

B
Bruce,  
Bruce Momjian 已提交
3596
			/* Dump Table Comments */
3597

B
Bruce,  
Bruce Momjian 已提交
3598
			resetPQExpBuffer(q);
3599
			appendPQExpBuffer(q, "%s %s", reltypename, fmtId(tblinfo[i].relname, force_quotes));
B
Bruce,  
Bruce Momjian 已提交
3600
			dumpComment(fout, q->data, tblinfo[i].oid);
3601

3602 3603
		}
	}
3604 3605 3606 3607
}

/*
 * dumpIndices:
3608
 *	  write out to fout all the user-define indices
3609
 */
3610
void
B
Bruce Momjian 已提交
3611
dumpIndices(Archive *fout, IndInfo *indinfo, int numIndices,
3612
			TableInfo *tblinfo, int numTables, const char *tablename)
3613
{
3614 3615 3616
	int			i,
				k;
	int			tableInd;
3617
	PQExpBuffer attlist = createPQExpBuffer();
B
Bruce Momjian 已提交
3618 3619 3620
	char	   *classname[INDEX_MAX_KEYS];
	char	   *funcname;		/* the name of the function to comput the
								 * index key from */
3621 3622 3623
	int			indkey,
				indclass;
	int			nclass;
3624

3625
	PQExpBuffer q = createPQExpBuffer(),
B
Bruce Momjian 已提交
3626
				delq = createPQExpBuffer(),
B
Hi, all  
Bruce Momjian 已提交
3627 3628
				id1 = createPQExpBuffer(),
				id2 = createPQExpBuffer();
3629
	PGresult   *res;
3630 3631 3632 3633

	for (i = 0; i < numIndices; i++)
	{
		tableInd = findTableByName(tblinfo, numTables,
3634 3635 3636 3637 3638
								   indinfo[i].indrelname);
		if (tableInd < 0)
		{
			fprintf(stderr, "failed sanity check, table %s was not found\n",
					indinfo[i].indrelname);
3639
			exit(1);
3640
		}
3641 3642 3643 3644 3645 3646 3647 3648 3649 3650 3651 3652

		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 已提交
3653 3654
			resetPQExpBuffer(q);
			appendPQExpBuffer(q,
3655 3656 3657
							  "SELECT proname from pg_proc "
							  "where pg_proc.oid = '%s'::oid",
							  indinfo[i].indproc);
B
Hi, all  
Bruce Momjian 已提交
3658
			res = PQexec(g_conn, q->data);
3659 3660
			if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
			{
B
Bruce Momjian 已提交
3661 3662
				fprintf(stderr, "dumpIndices(): SELECT (funcname) failed.  "
						"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3663 3664
				exit_nicely(g_conn);
			}
B
Bruce Momjian 已提交
3665
			funcname = strdup(PQgetvalue(res, 0, PQfnumber(res, "proname")));
3666 3667 3668 3669 3670 3671 3672 3673 3674
			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 已提交
3675 3676
			resetPQExpBuffer(q);
			appendPQExpBuffer(q,
3677 3678 3679
							  "SELECT opcname from pg_opclass "
							  "where pg_opclass.oid = '%u'::oid",
							  indclass);
B
Hi, all  
Bruce Momjian 已提交
3680
			res = PQexec(g_conn, q->data);
3681 3682
			if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
			{
B
Bruce Momjian 已提交
3683 3684
				fprintf(stderr, "dumpIndices(): SELECT (classname) failed.  "
									"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3685 3686
				exit_nicely(g_conn);
			}
B
Bruce Momjian 已提交
3687
			classname[nclass] = strdup(PQgetvalue(res, 0, PQfnumber(res, "opcname")));
3688 3689 3690 3691 3692 3693 3694 3695 3696 3697 3698
			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 已提交
3699 3700
		resetPQExpBuffer(attlist);
		for (k = 0; k < INDEX_MAX_KEYS; k++)
3701
		{
3702
			char	   *attname;
3703 3704

			indkey = atoi(indinfo[i].indkey[k]);
B
Bruce Momjian 已提交
3705
			if (indkey == InvalidAttrNumber)
3706 3707 3708 3709 3710 3711 3712
				break;
			indkey--;
			if (indkey == ObjectIdAttributeNumber - 1)
				attname = "oid";
			else
				attname = tblinfo[tableInd].attnames[indkey];
			if (funcname)
B
Hi, all  
Bruce Momjian 已提交
3713
				appendPQExpBuffer(attlist, "%s%s",
B
Bruce Momjian 已提交
3714
					 (k == 0) ? "" : ", ", fmtId(attname, force_quotes));
3715 3716 3717 3718 3719
			else
			{
				if (k >= nclass)
				{
					fprintf(stderr, "dumpIndices(): OpClass not found for "
3720
							"attribute '%s' of index '%s'\n",
3721 3722 3723
							attname, indinfo[i].indexrelname);
					exit_nicely(g_conn);
				}
B
Hi, all  
Bruce Momjian 已提交
3724 3725 3726 3727 3728
				resetPQExpBuffer(id1);
				resetPQExpBuffer(id2);
				appendPQExpBuffer(id1, fmtId(attname, force_quotes));
				appendPQExpBuffer(id2, fmtId(classname[k], force_quotes));
				appendPQExpBuffer(attlist, "%s%s %s",
3729
							 (k == 0) ? "" : ", ", id1->data, id2->data);
3730 3731 3732 3733 3734 3735
				free(classname[k]);
			}
		}

		if (!tablename || (!strcmp(indinfo[i].indrelname, tablename)))
		{
B
Bruce Momjian 已提交
3736 3737 3738 3739 3740

			/*
			 * 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.
3741
			 */
3742

B
Hi, all  
Bruce Momjian 已提交
3743 3744 3745 3746
			resetPQExpBuffer(id1);
			resetPQExpBuffer(id2);
			appendPQExpBuffer(id1, fmtId(indinfo[i].indexrelname, force_quotes));
			appendPQExpBuffer(id2, fmtId(indinfo[i].indrelname, force_quotes));
3747

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

B
Bruce Momjian 已提交
3751 3752 3753
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "CREATE %s INDEX %s on %s using %s (",
					(strcmp(indinfo[i].indisunique, "t") == 0) ? "UNIQUE" : "",
B
Hi, all  
Bruce Momjian 已提交
3754 3755
					id1->data,
					id2->data,
3756 3757 3758
					indinfo[i].indamname);
			if (funcname)
			{
3759
				/* need 2 printf's here cuz fmtId has static return area */
B
Bruce Momjian 已提交
3760 3761 3762
				appendPQExpBuffer(q, " %s", fmtId(funcname, false));
				appendPQExpBuffer(q, " (%s) %s );\n", attlist->data, 
									fmtId(classname[0], force_quotes));
3763 3764 3765 3766
				free(funcname);
				free(classname[0]);
			}
			else
B
Bruce Momjian 已提交
3767
				appendPQExpBuffer(q, " %s );\n", attlist->data);
B
Bruce,  
Bruce Momjian 已提交
3768 3769 3770

			/* Dump Index Comments */

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

B
Bruce,  
Bruce Momjian 已提交
3774 3775 3776
			resetPQExpBuffer(q);
			appendPQExpBuffer(q, "INDEX %s", id1->data);
			dumpComment(fout, q->data, indinfo[i].indoid);
3777

3778 3779
		}
	}
3780 3781

}
3782

3783
/*
B
Bruce Momjian 已提交
3784
 * dumpTuples
3785 3786 3787 3788 3789
 *	  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
3790
 *	  in PostgreSQL.
3791 3792 3793 3794
 *
 * the attrmap passed in tells how to map the attributes copied in to the
 * attributes copied out
 */
3795
#ifdef NOT_USED
3796
void
3797
dumpTuples(PGresult *res, FILE *fout, int *attrmap)
3798
{
3799 3800 3801 3802
	int			j,
				k;
	int			m,
				n;
B
Bruce Momjian 已提交
3803
	char	  **outVals = NULL; /* values to copy out */
3804 3805 3806 3807 3808 3809 3810 3811 3812 3813 3814 3815 3816 3817 3818 3819 3820 3821 3822

	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++)
			{
3823
				char	   *pval = outVals[k];
3824 3825 3826 3827 3828 3829 3830 3831 3832 3833 3834 3835 3836 3837 3838 3839 3840 3841 3842 3843

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

3846
#endif
3847

3848 3849 3850 3851
/*
 * setMaxOid -
 * find the maximum oid and generate a COPY statement to set it
*/
3852

3853
static void
B
Bruce Momjian 已提交
3854
setMaxOid(Archive *fout)
3855
{
B
Bruce Momjian 已提交
3856 3857 3858 3859
	PGresult   	*res;
	Oid		max_oid;
	char		sql[1024];
	int		pos;
3860

3861
	res = PQexec(g_conn, "CREATE TEMPORARY TABLE pgdump_oid (dummy int4)");
3862 3863 3864
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
3865
		fprintf(stderr, "Can not create pgdump_oid table.  "
B
Bruce Momjian 已提交
3866
					"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3867 3868 3869
		exit_nicely(g_conn);
	}
	PQclear(res);
3870
	res = PQexec(g_conn, "INSERT INTO pgdump_oid VALUES (0)");
3871 3872 3873
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
3874
		fprintf(stderr, "Can not insert into pgdump_oid table.  "
B
Bruce Momjian 已提交
3875
					"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3876 3877 3878 3879 3880 3881 3882 3883 3884
		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);
3885
	res = PQexec(g_conn, "DROP TABLE pgdump_oid;");
3886 3887 3888
	if (!res ||
		PQresultStatus(res) != PGRES_COMMAND_OK)
	{
3889
		fprintf(stderr, "Can not drop pgdump_oid table.  "
B
Bruce Momjian 已提交
3890
							"Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3891 3892 3893 3894
		exit_nicely(g_conn);
	}
	PQclear(res);
	if (g_verbose)
3895
		fprintf(stderr, "%s maximum system oid is %u %s\n",
3896
				g_comment_start, max_oid, g_comment_end);
3897 3898
	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 已提交
3899 3900
	pos = pos + snprintf(sql+pos, 1024-pos, "%-d\t0\n", max_oid);
	pos = pos + snprintf(sql+pos, 1024-pos, "\\.\n");
3901
	pos = pos + snprintf(sql+pos, 1024-pos, "DROP TABLE pgdump_oid;\n");
B
Bruce Momjian 已提交
3902

3903
	ArchiveEntry(fout, "0", "Max OID", "<Init>", NULL, sql, "", "", "", NULL, NULL);
3904
}
3905 3906 3907

/*
 * findLastBuiltInOid -
3908
 * find the last built in oid
3909 3910
 * we do this by retrieving datlastsysoid from the pg_database entry for this database,
 */
3911

3912
static Oid
3913
findLastBuiltinOid(const char* dbname)
3914
{
B
Bruce Momjian 已提交
3915
	PGresult   *res;
3916
	int			ntups;
3917
	Oid			last_oid;
3918 3919 3920 3921
	PQExpBuffer query = createPQExpBuffer();

	resetPQExpBuffer(query);
	appendPQExpBuffer(query, "SELECT datlastsysoid from pg_database where datname = '%s'", dbname);
3922

3923
	res = PQexec(g_conn, query->data);
3924 3925 3926
	if (res == NULL ||
		PQresultStatus(res) != PGRES_TUPLES_OK)
	{
3927
		fprintf(stderr, "pg_dump: error in finding the last system OID");
3928
		fprintf(stderr, "Explanation from backend: '%s'.\n", PQerrorMessage(g_conn));
3929 3930 3931
		exit_nicely(g_conn);
	}
	ntups = PQntuples(res);
3932
	if (ntups < 1)
3933
	{
3934 3935
		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");
3936 3937 3938 3939
		exit_nicely(g_conn);
	}
	if (ntups > 1)
	{
3940 3941
		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");
3942 3943
		exit_nicely(g_conn);
	}
3944
	last_oid = atoi(PQgetvalue(res, 0, PQfnumber(res, "datlastsysoid")));
3945 3946
	PQclear(res);
	return last_oid;
3947 3948 3949 3950 3951
}


/*
 * checkForQuote:
3952
 *	  checks a string for quote characters and quotes them
3953
 */
3954
static char *
3955
checkForQuote(const char *s)
3956
{
B
Bruce Momjian 已提交
3957
	char	   *r;
3958
	char		c;
B
Bruce Momjian 已提交
3959
	char	   *result;
3960

3961
	int			j = 0;
3962

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

3965 3966
	while ((c = *s) != '\0')
	{
3967

3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978
		if (c == '\'')
		{
			r[j++] = '\'';		/* quote the single quotes */
		}
		r[j++] = c;
		s++;
	}
	r[j] = '\0';

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

3980
	return result;
3981 3982

}
3983 3984


3985
static void
B
Bruce Momjian 已提交
3986
dumpSequence(Archive *fout, TableInfo tbinfo)
3987
{
3988
	PGresult   *res;
B
Bruce Momjian 已提交
3989
	int4		last,
3990 3991 3992 3993
				incby,
				maxv,
				minv,
				cache;
B
Bruce Momjian 已提交
3994
	char		cycled,
B
Bruce Momjian 已提交
3995
				called;
3996 3997
	const char *t;
	PQExpBuffer query = createPQExpBuffer();
B
Bruce Momjian 已提交
3998
	PQExpBuffer delqry = createPQExpBuffer();
3999

B
Hi, all  
Bruce Momjian 已提交
4000
	appendPQExpBuffer(query,
4001
			"SELECT sequence_name, last_value, increment_by, max_value, "
4002 4003
				  "min_value, cache_value, is_cycled, is_called from %s",
					  fmtId(tbinfo.relname, force_quotes));
4004

B
Hi, all  
Bruce Momjian 已提交
4005
	res = PQexec(g_conn, query->data);
4006 4007
	if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
	{
B
Bruce Momjian 已提交
4008 4009
		fprintf(stderr, "dumpSequence(%s): SELECT failed.  "
					"Explanation from backend: '%s'.\n", tbinfo.relname, PQerrorMessage(g_conn));
4010 4011 4012 4013 4014 4015 4016 4017 4018 4019 4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040
		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 已提交
4041 4042
	resetPQExpBuffer(delqry);
	appendPQExpBuffer(delqry, "DROP SEQUENCE %s;\n", fmtId(tbinfo.relname, force_quotes));
4043

4044 4045 4046 4047 4048 4049
	/*
	 * 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 已提交
4050 4051
	resetPQExpBuffer(query);
	appendPQExpBuffer(query,
4052 4053
				  "CREATE SEQUENCE %s start %d increment %d maxvalue %d "
					  "minvalue %d  cache %d %s;\n",
4054 4055 4056
					fmtId(tbinfo.relname, force_quotes), 
					  (called == 't') ? minv : last,
					  incby, maxv, minv, cache,
4057
					  (cycled == 't') ? "cycle" : "");
4058

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

4062 4063 4064 4065 4066 4067 4068 4069

	resetPQExpBuffer(query);
	appendPQExpBuffer(query, "SELECT setval ('%s', %d, '%c');\n", 
						fmtId(tbinfo.relname, force_quotes), last, called);

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

B
Bruce,  
Bruce Momjian 已提交
4070 4071 4072 4073 4074 4075
	/* Dump Sequence Comments */

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

4076
}
V
Vadim B. Mikheev 已提交
4077 4078


4079
static void
B
Bruce Momjian 已提交
4080
dumpTriggers(Archive *fout, const char *tablename,
4081
			 TableInfo *tblinfo, int numTables)
V
Vadim B. Mikheev 已提交
4082
{
4083 4084 4085
	int			i,
				j;

V
Vadim B. Mikheev 已提交
4086 4087 4088
	if (g_verbose)
		fprintf(stderr, "%s dumping out triggers %s\n",
				g_comment_start, g_comment_end);
4089

V
Vadim B. Mikheev 已提交
4090 4091 4092 4093 4094 4095
	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 已提交
4096
			ArchiveEntry(fout, tblinfo[i].triggers[j].oid, tblinfo[i].triggers[j].tgname,
4097
						"TRIGGER", NULL, tblinfo[i].triggers[j].tgsrc, "", "", 
B
Bruce Momjian 已提交
4098 4099
						tblinfo[i].usename, NULL, NULL);
			dumpComment(fout, tblinfo[i].triggers[j].tgcomment, tblinfo[i].triggers[j].oid);
V
Vadim B. Mikheev 已提交
4100 4101 4102
		}
	}
}
4103 4104


4105
static void
B
Bruce Momjian 已提交
4106
dumpRules(Archive *fout, const char *tablename,
B
Bruce Momjian 已提交
4107
		  TableInfo *tblinfo, int numTables)
4108
{
B
Bruce Momjian 已提交
4109 4110 4111 4112
	PGresult   *res;
	int			nrules;
	int			i,
				t;
4113
	PQExpBuffer query = createPQExpBuffer();
4114

B
Bruce Momjian 已提交
4115
	int			i_definition;
4116
	int			i_oid;
4117
	int			i_owner;
4118
	int			i_rulename;
4119 4120 4121 4122 4123 4124 4125 4126 4127 4128 4129 4130 4131 4132 4133

	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
4134 4135
		 * We include pg_rules in the cross since it filters out
		 * all view rules (pjw 15-Sep-2000).
4136
		 */
B
Hi, all  
Bruce Momjian 已提交
4137
		resetPQExpBuffer(query);
4138
		appendPQExpBuffer(query, "SELECT definition,"
4139
						  "   (select usename from pg_user where pg_class.relowner = usesysid) AS viewowner, "
4140 4141
						  "   pg_rewrite.oid, pg_rewrite.rulename "
						  "FROM pg_rewrite, pg_class, pg_rules "
4142
						  "WHERE pg_class.relname = '%s' "
4143 4144 4145
						  "    AND pg_rewrite.ev_class = pg_class.oid "
						  "    AND pg_rules.tablename = pg_class.relname "
						  "    AND pg_rules.rulename = pg_rewrite.rulename "
4146 4147
						  "ORDER BY pg_rewrite.oid",
						  tblinfo[t].relname);
B
Hi, all  
Bruce Momjian 已提交
4148
		res = PQexec(g_conn, query->data);
4149 4150 4151
		if (!res ||
			PQresultStatus(res) != PGRES_TUPLES_OK)
		{
4152
			fprintf(stderr, "dumpRules(): SELECT failed for rules associated with table \"%s\".\n\tExplanation from backend: '%s'.\n",
B
Bruce Momjian 已提交
4153
					tblinfo[t].relname, PQerrorMessage(g_conn));
4154 4155 4156 4157 4158
			exit_nicely(g_conn);
		}

		nrules = PQntuples(res);
		i_definition = PQfnumber(res, "definition");
4159
		i_owner = PQfnumber(res, "viewowner");
B
Bruce,  
Bruce Momjian 已提交
4160 4161
		i_oid = PQfnumber(res, "oid");
		i_rulename = PQfnumber(res, "rulename");
4162 4163 4164 4165

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

4167 4168
		for (i = 0; i < nrules; i++)
		{
B
Bruce Momjian 已提交
4169 4170
			ArchiveEntry(fout, PQgetvalue(res, i, i_oid), PQgetvalue(res, i, i_rulename),
							"RULE", NULL, PQgetvalue(res, i, i_definition),
4171
							"", "", PQgetvalue(res, i, i_owner), NULL, NULL);
4172

B
Bruce,  
Bruce Momjian 已提交
4173 4174 4175 4176 4177
			/* 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));
4178

B
Bruce,  
Bruce Momjian 已提交
4179 4180
		}

4181 4182 4183 4184
		PQclear(res);
	}
}