Line data Source code
1 : /*
2 : * oid2name, a PostgreSQL app to map OIDs on the filesystem
3 : * to table and database names.
4 : *
5 : * Originally by
6 : * B. Palmer, bpalmer@crimelabs.net 1-17-2001
7 : *
8 : * contrib/oid2name/oid2name.c
9 : */
10 : #include "postgres_fe.h"
11 :
12 : #include "catalog/pg_class_d.h"
13 : #include "common/connect.h"
14 : #include "common/logging.h"
15 : #include "common/string.h"
16 : #include "getopt_long.h"
17 : #include "libpq-fe.h"
18 : #include "pg_getopt.h"
19 :
20 : /* an extensible array to keep track of elements to show */
21 : typedef struct
22 : {
23 : char **array;
24 : int num;
25 : int alloc;
26 : } eary;
27 :
28 : /* these are the opts structures for command line params */
29 : struct options
30 : {
31 : eary *tables;
32 : eary *oids;
33 : eary *filenumbers;
34 :
35 : bool quiet;
36 : bool systables;
37 : bool indexes;
38 : bool nodb;
39 : bool extended;
40 : bool tablespaces;
41 :
42 : char *dbname;
43 : char *hostname;
44 : char *port;
45 : char *username;
46 : const char *progname;
47 : };
48 :
49 : /* function prototypes */
50 : static void help(const char *progname);
51 : void get_opts(int argc, char **argv, struct options *my_opts);
52 : void add_one_elt(char *eltname, eary *eary);
53 : char *get_comma_elts(eary *eary);
54 : PGconn *sql_conn(struct options *my_opts);
55 : int sql_exec(PGconn *conn, const char *todo, bool quiet);
56 : void sql_exec_dumpalldbs(PGconn *conn, struct options *opts);
57 : void sql_exec_dumpalltables(PGconn *conn, struct options *opts);
58 : void sql_exec_searchtables(PGconn *conn, struct options *opts);
59 : void sql_exec_dumpalltbspc(PGconn *conn, struct options *opts);
60 :
61 : /* function to parse command line options and check for some usage errors. */
62 : void
63 0 : get_opts(int argc, char **argv, struct options *my_opts)
64 : {
65 : static const struct option long_options[] = {
66 : {"dbname", required_argument, NULL, 'd'},
67 : {"host", required_argument, NULL, 'h'},
68 : {"host", required_argument, NULL, 'H'}, /* deprecated */
69 : {"filenode", required_argument, NULL, 'f'},
70 : {"indexes", no_argument, NULL, 'i'},
71 : {"oid", required_argument, NULL, 'o'},
72 : {"port", required_argument, NULL, 'p'},
73 : {"quiet", no_argument, NULL, 'q'},
74 : {"tablespaces", no_argument, NULL, 's'},
75 : {"system-objects", no_argument, NULL, 'S'},
76 : {"table", required_argument, NULL, 't'},
77 : {"username", required_argument, NULL, 'U'},
78 : {"version", no_argument, NULL, 'V'},
79 : {"extended", no_argument, NULL, 'x'},
80 : {"help", no_argument, NULL, '?'},
81 : {NULL, 0, NULL, 0}
82 : };
83 :
84 0 : int c;
85 0 : const char *progname;
86 0 : int optindex;
87 :
88 0 : pg_logging_init(argv[0]);
89 0 : progname = get_progname(argv[0]);
90 :
91 : /* set the defaults */
92 0 : my_opts->quiet = false;
93 0 : my_opts->systables = false;
94 0 : my_opts->indexes = false;
95 0 : my_opts->nodb = false;
96 0 : my_opts->extended = false;
97 0 : my_opts->tablespaces = false;
98 0 : my_opts->dbname = NULL;
99 0 : my_opts->hostname = NULL;
100 0 : my_opts->port = NULL;
101 0 : my_opts->username = NULL;
102 0 : my_opts->progname = progname;
103 :
104 0 : if (argc > 1)
105 : {
106 0 : if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
107 : {
108 0 : help(progname);
109 0 : exit(0);
110 : }
111 0 : if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
112 : {
113 0 : puts("oid2name (PostgreSQL) " PG_VERSION);
114 0 : exit(0);
115 : }
116 0 : }
117 :
118 : /* get opts */
119 0 : while ((c = getopt_long(argc, argv, "d:f:h:H:io:p:qsSt:U:x", long_options, &optindex)) != -1)
120 : {
121 0 : switch (c)
122 : {
123 : /* specify the database */
124 : case 'd':
125 0 : my_opts->dbname = pg_strdup(optarg);
126 0 : break;
127 :
128 : /* specify one filenumber to show */
129 : case 'f':
130 0 : add_one_elt(optarg, my_opts->filenumbers);
131 0 : break;
132 :
133 : /* host to connect to */
134 : case 'H': /* deprecated */
135 : case 'h':
136 0 : my_opts->hostname = pg_strdup(optarg);
137 0 : break;
138 :
139 : /* also display indexes */
140 : case 'i':
141 0 : my_opts->indexes = true;
142 0 : break;
143 :
144 : /* specify one Oid to show */
145 : case 'o':
146 0 : add_one_elt(optarg, my_opts->oids);
147 0 : break;
148 :
149 : /* port to connect to on remote host */
150 : case 'p':
151 0 : my_opts->port = pg_strdup(optarg);
152 0 : break;
153 :
154 : /* don't show headers */
155 : case 'q':
156 0 : my_opts->quiet = true;
157 0 : break;
158 :
159 : /* dump tablespaces only */
160 : case 's':
161 0 : my_opts->tablespaces = true;
162 0 : break;
163 :
164 : /* display system tables */
165 : case 'S':
166 0 : my_opts->systables = true;
167 0 : break;
168 :
169 : /* specify one tablename to show */
170 : case 't':
171 0 : add_one_elt(optarg, my_opts->tables);
172 0 : break;
173 :
174 : /* username */
175 : case 'U':
176 0 : my_opts->username = pg_strdup(optarg);
177 0 : break;
178 :
179 : /* display extra columns */
180 : case 'x':
181 0 : my_opts->extended = true;
182 0 : break;
183 :
184 : default:
185 : /* getopt_long already emitted a complaint */
186 0 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
187 0 : exit(1);
188 : }
189 : }
190 :
191 0 : if (optind < argc)
192 : {
193 0 : pg_log_error("too many command-line arguments (first is \"%s\")",
194 : argv[optind]);
195 0 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
196 0 : exit(1);
197 : }
198 0 : }
199 :
200 : static void
201 0 : help(const char *progname)
202 : {
203 0 : printf("%s helps examining the file structure used by PostgreSQL.\n\n"
204 : "Usage:\n"
205 : " %s [OPTION]...\n"
206 : "\nOptions:\n"
207 : " -f, --filenode=FILENODE show info for table with given file node\n"
208 : " -i, --indexes show indexes and sequences too\n"
209 : " -o, --oid=OID show info for table with given OID\n"
210 : " -q, --quiet quiet (don't show headers)\n"
211 : " -s, --tablespaces show all tablespaces\n"
212 : " -S, --system-objects show system objects too\n"
213 : " -t, --table=TABLE show info for named table\n"
214 : " -V, --version output version information, then exit\n"
215 : " -x, --extended extended (show additional columns)\n"
216 : " -?, --help show this help, then exit\n"
217 : "\nConnection options:\n"
218 : " -d, --dbname=DBNAME database to connect to\n"
219 : " -h, --host=HOSTNAME database server host or socket directory\n"
220 : " -H (same as -h, deprecated)\n"
221 : " -p, --port=PORT database server port number\n"
222 : " -U, --username=USERNAME connect as specified database user\n"
223 : "\nThe default action is to show all database OIDs.\n\n"
224 : "Report bugs to <%s>.\n"
225 : "%s home page: <%s>\n",
226 : progname, progname, PACKAGE_BUGREPORT, PACKAGE_NAME, PACKAGE_URL);
227 0 : }
228 :
229 : /*
230 : * add_one_elt
231 : *
232 : * Add one element to a (possibly empty) eary struct.
233 : */
234 : void
235 0 : add_one_elt(char *eltname, eary *eary)
236 : {
237 0 : if (eary->alloc == 0)
238 : {
239 0 : eary ->alloc = 8;
240 0 : eary ->array = (char **) pg_malloc(8 * sizeof(char *));
241 0 : }
242 0 : else if (eary->num >= eary->alloc)
243 : {
244 0 : eary ->alloc *= 2;
245 0 : eary ->array = (char **) pg_realloc(eary->array,
246 0 : eary->alloc * sizeof(char *));
247 0 : }
248 :
249 0 : eary ->array[eary->num] = pg_strdup(eltname);
250 0 : eary ->num++;
251 0 : }
252 :
253 : /*
254 : * get_comma_elts
255 : *
256 : * Return the elements of an eary as a (freshly allocated) single string, in
257 : * single quotes, separated by commas and properly escaped for insertion in an
258 : * SQL statement.
259 : */
260 : char *
261 0 : get_comma_elts(eary *eary)
262 : {
263 0 : char *ret,
264 : *ptr;
265 0 : int i,
266 0 : length = 0;
267 :
268 0 : if (eary->num == 0)
269 0 : return pg_strdup("");
270 :
271 : /*
272 : * PQescapeString wants 2 * length + 1 bytes of breath space. Add two
273 : * chars per element for the single quotes and one for the comma.
274 : */
275 0 : for (i = 0; i < eary->num; i++)
276 0 : length += strlen(eary->array[i]);
277 :
278 0 : ret = (char *) pg_malloc(length * 2 + 4 * eary->num);
279 0 : ptr = ret;
280 :
281 0 : for (i = 0; i < eary->num; i++)
282 : {
283 0 : if (i != 0)
284 0 : sprintf(ptr++, ",");
285 0 : sprintf(ptr++, "'");
286 0 : ptr += PQescapeString(ptr, eary->array[i], strlen(eary->array[i]));
287 0 : sprintf(ptr++, "'");
288 0 : }
289 :
290 0 : return ret;
291 0 : }
292 :
293 : /* establish connection with database. */
294 : PGconn *
295 0 : sql_conn(struct options *my_opts)
296 : {
297 0 : PGconn *conn;
298 0 : char *password = NULL;
299 0 : bool new_pass;
300 0 : PGresult *res;
301 :
302 : /*
303 : * Start the connection. Loop until we have a password if requested by
304 : * backend.
305 : */
306 0 : do
307 : {
308 : #define PARAMS_ARRAY_SIZE 7
309 :
310 0 : const char *keywords[PARAMS_ARRAY_SIZE];
311 0 : const char *values[PARAMS_ARRAY_SIZE];
312 :
313 0 : keywords[0] = "host";
314 0 : values[0] = my_opts->hostname;
315 0 : keywords[1] = "port";
316 0 : values[1] = my_opts->port;
317 0 : keywords[2] = "user";
318 0 : values[2] = my_opts->username;
319 0 : keywords[3] = "password";
320 0 : values[3] = password;
321 0 : keywords[4] = "dbname";
322 0 : values[4] = my_opts->dbname;
323 0 : keywords[5] = "fallback_application_name";
324 0 : values[5] = my_opts->progname;
325 0 : keywords[6] = NULL;
326 0 : values[6] = NULL;
327 :
328 0 : new_pass = false;
329 0 : conn = PQconnectdbParams(keywords, values, true);
330 :
331 0 : if (!conn)
332 0 : pg_fatal("could not connect to database %s",
333 : my_opts->dbname);
334 :
335 0 : if (PQstatus(conn) == CONNECTION_BAD &&
336 0 : PQconnectionNeedsPassword(conn) &&
337 0 : !password)
338 : {
339 0 : PQfinish(conn);
340 0 : password = simple_prompt("Password: ", false);
341 0 : new_pass = true;
342 0 : }
343 0 : } while (new_pass);
344 :
345 : /* check to see that the backend connection was successfully made */
346 0 : if (PQstatus(conn) == CONNECTION_BAD)
347 : {
348 0 : pg_log_error("%s", PQerrorMessage(conn));
349 0 : PQfinish(conn);
350 0 : exit(1);
351 : }
352 :
353 0 : res = PQexec(conn, ALWAYS_SECURE_SEARCH_PATH_SQL);
354 0 : if (PQresultStatus(res) != PGRES_TUPLES_OK)
355 : {
356 0 : pg_log_error("could not clear \"search_path\": %s",
357 : PQerrorMessage(conn));
358 0 : PQclear(res);
359 0 : PQfinish(conn);
360 0 : exit(1);
361 : }
362 0 : PQclear(res);
363 :
364 : /* return the conn if good */
365 0 : return conn;
366 0 : }
367 :
368 : /*
369 : * Actual code to make call to the database and print the output data.
370 : */
371 : int
372 0 : sql_exec(PGconn *conn, const char *todo, bool quiet)
373 : {
374 0 : PGresult *res;
375 :
376 0 : int nfields;
377 0 : int nrows;
378 0 : int i,
379 : j,
380 : l;
381 0 : int *length;
382 0 : char *pad;
383 :
384 : /* make the call */
385 0 : res = PQexec(conn, todo);
386 :
387 : /* check and deal with errors */
388 0 : if (!res || PQresultStatus(res) > 2)
389 : {
390 0 : pg_log_error("query failed: %s", PQerrorMessage(conn));
391 0 : pg_log_error_detail("Query was: %s", todo);
392 :
393 0 : PQclear(res);
394 0 : PQfinish(conn);
395 0 : exit(1);
396 : }
397 :
398 : /* get the number of fields */
399 0 : nrows = PQntuples(res);
400 0 : nfields = PQnfields(res);
401 :
402 : /* for each field, get the needed width */
403 0 : length = (int *) pg_malloc(sizeof(int) * nfields);
404 0 : for (j = 0; j < nfields; j++)
405 0 : length[j] = strlen(PQfname(res, j));
406 :
407 0 : for (i = 0; i < nrows; i++)
408 : {
409 0 : for (j = 0; j < nfields; j++)
410 : {
411 0 : l = strlen(PQgetvalue(res, i, j));
412 0 : if (l > length[j])
413 0 : length[j] = strlen(PQgetvalue(res, i, j));
414 0 : }
415 0 : }
416 :
417 : /* print a header */
418 0 : if (!quiet)
419 : {
420 0 : for (j = 0, l = 0; j < nfields; j++)
421 : {
422 0 : fprintf(stdout, "%*s", length[j] + 2, PQfname(res, j));
423 0 : l += length[j] + 2;
424 0 : }
425 0 : fprintf(stdout, "\n");
426 0 : pad = (char *) pg_malloc(l + 1);
427 0 : memset(pad, '-', l);
428 0 : pad[l] = '\0';
429 0 : fprintf(stdout, "%s\n", pad);
430 0 : free(pad);
431 0 : }
432 :
433 : /* for each row, dump the information */
434 0 : for (i = 0; i < nrows; i++)
435 : {
436 0 : for (j = 0; j < nfields; j++)
437 0 : fprintf(stdout, "%*s", length[j] + 2, PQgetvalue(res, i, j));
438 0 : fprintf(stdout, "\n");
439 0 : }
440 :
441 : /* cleanup */
442 0 : PQclear(res);
443 0 : free(length);
444 :
445 0 : return 0;
446 0 : }
447 :
448 : /*
449 : * Dump all databases. There are no system objects to worry about.
450 : */
451 : void
452 0 : sql_exec_dumpalldbs(PGconn *conn, struct options *opts)
453 : {
454 0 : char todo[1024];
455 :
456 : /* get the oid and database name from the system pg_database table */
457 0 : snprintf(todo, sizeof(todo),
458 : "SELECT d.oid AS \"Oid\", datname AS \"Database Name\", "
459 : "spcname AS \"Tablespace\" FROM pg_catalog.pg_database d JOIN pg_catalog.pg_tablespace t ON "
460 : "(dattablespace = t.oid) ORDER BY 2");
461 :
462 0 : sql_exec(conn, todo, opts->quiet);
463 0 : }
464 :
465 : /*
466 : * Dump all tables, indexes and sequences in the current database.
467 : */
468 : void
469 0 : sql_exec_dumpalltables(PGconn *conn, struct options *opts)
470 : {
471 0 : char todo[1024];
472 0 : char *addfields = ",c.oid AS \"Oid\", nspname AS \"Schema\", spcname as \"Tablespace\" ";
473 :
474 0 : snprintf(todo, sizeof(todo),
475 : "SELECT pg_catalog.pg_relation_filenode(c.oid) as \"Filenode\", relname as \"Table Name\" %s "
476 : "FROM pg_catalog.pg_class c "
477 : " LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace "
478 : " LEFT JOIN pg_catalog.pg_database d ON d.datname = pg_catalog.current_database(),"
479 : " pg_catalog.pg_tablespace t "
480 : "WHERE relkind IN (" CppAsString2(RELKIND_RELATION) ","
481 : CppAsString2(RELKIND_MATVIEW) "%s%s) AND "
482 : " %s"
483 : " t.oid = CASE"
484 : " WHEN reltablespace <> 0 THEN reltablespace"
485 : " ELSE dattablespace"
486 : " END "
487 : "ORDER BY relname",
488 0 : opts->extended ? addfields : "",
489 0 : opts->indexes ? "," CppAsString2(RELKIND_INDEX) "," CppAsString2(RELKIND_SEQUENCE) : "",
490 0 : opts->systables ? "," CppAsString2(RELKIND_TOASTVALUE) : "",
491 0 : opts->systables ? "" : "n.nspname NOT IN ('pg_catalog', 'information_schema') AND n.nspname !~ '^pg_toast' AND");
492 :
493 0 : sql_exec(conn, todo, opts->quiet);
494 0 : }
495 :
496 : /*
497 : * Show oid, filenumber, name, schema and tablespace for each of the
498 : * given objects in the current database.
499 : */
500 : void
501 0 : sql_exec_searchtables(PGconn *conn, struct options *opts)
502 : {
503 0 : char *todo;
504 0 : char *qualifiers,
505 : *ptr;
506 0 : char *comma_oids,
507 : *comma_filenumbers,
508 : *comma_tables;
509 0 : bool written = false;
510 0 : char *addfields = ",c.oid AS \"Oid\", nspname AS \"Schema\", spcname as \"Tablespace\" ";
511 :
512 : /* get tables qualifiers, whether names, filenumbers, or OIDs */
513 0 : comma_oids = get_comma_elts(opts->oids);
514 0 : comma_tables = get_comma_elts(opts->tables);
515 0 : comma_filenumbers = get_comma_elts(opts->filenumbers);
516 :
517 : /* 80 extra chars for SQL expression */
518 0 : qualifiers = (char *) pg_malloc(strlen(comma_oids) + strlen(comma_tables) +
519 0 : strlen(comma_filenumbers) + 80);
520 0 : ptr = qualifiers;
521 :
522 0 : if (opts->oids->num > 0)
523 : {
524 0 : ptr += sprintf(ptr, "c.oid IN (%s)", comma_oids);
525 0 : written = true;
526 0 : }
527 0 : if (opts->filenumbers->num > 0)
528 : {
529 0 : if (written)
530 0 : ptr += sprintf(ptr, " OR ");
531 0 : ptr += sprintf(ptr, "pg_catalog.pg_relation_filenode(c.oid) IN (%s)",
532 0 : comma_filenumbers);
533 0 : written = true;
534 0 : }
535 0 : if (opts->tables->num > 0)
536 : {
537 0 : if (written)
538 0 : ptr += sprintf(ptr, " OR ");
539 0 : sprintf(ptr, "c.relname ~~ ANY (ARRAY[%s])", comma_tables);
540 0 : }
541 0 : free(comma_oids);
542 0 : free(comma_tables);
543 0 : free(comma_filenumbers);
544 :
545 : /* now build the query */
546 0 : todo = psprintf("SELECT pg_catalog.pg_relation_filenode(c.oid) as \"Filenode\", relname as \"Table Name\" %s\n"
547 : "FROM pg_catalog.pg_class c\n"
548 : " LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace\n"
549 : " LEFT JOIN pg_catalog.pg_database d ON d.datname = pg_catalog.current_database(),\n"
550 : " pg_catalog.pg_tablespace t\n"
551 : "WHERE relkind IN (" CppAsString2(RELKIND_RELATION) ","
552 : CppAsString2(RELKIND_MATVIEW) ","
553 : CppAsString2(RELKIND_INDEX) ","
554 : CppAsString2(RELKIND_SEQUENCE) ","
555 : CppAsString2(RELKIND_TOASTVALUE) ") AND\n"
556 : " t.oid = CASE\n"
557 : " WHEN reltablespace <> 0 THEN reltablespace\n"
558 : " ELSE dattablespace\n"
559 : " END AND\n"
560 : " (%s)\n"
561 : "ORDER BY relname\n",
562 0 : opts->extended ? addfields : "",
563 0 : qualifiers);
564 :
565 0 : free(qualifiers);
566 :
567 0 : sql_exec(conn, todo, opts->quiet);
568 0 : }
569 :
570 : void
571 0 : sql_exec_dumpalltbspc(PGconn *conn, struct options *opts)
572 : {
573 0 : char todo[1024];
574 :
575 0 : snprintf(todo, sizeof(todo),
576 : "SELECT oid AS \"Oid\", spcname as \"Tablespace Name\"\n"
577 : "FROM pg_catalog.pg_tablespace");
578 :
579 0 : sql_exec(conn, todo, opts->quiet);
580 0 : }
581 :
582 : int
583 0 : main(int argc, char **argv)
584 : {
585 0 : struct options *my_opts;
586 0 : PGconn *pgconn;
587 :
588 0 : my_opts = (struct options *) pg_malloc(sizeof(struct options));
589 :
590 0 : my_opts->oids = (eary *) pg_malloc(sizeof(eary));
591 0 : my_opts->tables = (eary *) pg_malloc(sizeof(eary));
592 0 : my_opts->filenumbers = (eary *) pg_malloc(sizeof(eary));
593 :
594 0 : my_opts->oids->num = my_opts->oids->alloc = 0;
595 0 : my_opts->tables->num = my_opts->tables->alloc = 0;
596 0 : my_opts->filenumbers->num = my_opts->filenumbers->alloc = 0;
597 :
598 : /* parse the opts */
599 0 : get_opts(argc, argv, my_opts);
600 :
601 0 : if (my_opts->dbname == NULL)
602 : {
603 0 : my_opts->dbname = "postgres";
604 0 : my_opts->nodb = true;
605 0 : }
606 0 : pgconn = sql_conn(my_opts);
607 :
608 : /* display only tablespaces */
609 0 : if (my_opts->tablespaces)
610 : {
611 0 : if (!my_opts->quiet)
612 0 : printf("All tablespaces:\n");
613 0 : sql_exec_dumpalltbspc(pgconn, my_opts);
614 :
615 0 : PQfinish(pgconn);
616 0 : exit(0);
617 : }
618 :
619 : /* display the given elements in the database */
620 0 : if (my_opts->oids->num > 0 ||
621 0 : my_opts->tables->num > 0 ||
622 0 : my_opts->filenumbers->num > 0)
623 : {
624 0 : if (!my_opts->quiet)
625 0 : printf("From database \"%s\":\n", my_opts->dbname);
626 0 : sql_exec_searchtables(pgconn, my_opts);
627 :
628 0 : PQfinish(pgconn);
629 0 : exit(0);
630 : }
631 :
632 : /* no elements given; dump the given database */
633 0 : if (my_opts->dbname && !my_opts->nodb)
634 : {
635 0 : if (!my_opts->quiet)
636 0 : printf("From database \"%s\":\n", my_opts->dbname);
637 0 : sql_exec_dumpalltables(pgconn, my_opts);
638 :
639 0 : PQfinish(pgconn);
640 0 : exit(0);
641 : }
642 :
643 : /* no database either; dump all databases */
644 0 : if (!my_opts->quiet)
645 0 : printf("All databases:\n");
646 0 : sql_exec_dumpalldbs(pgconn, my_opts);
647 :
648 0 : PQfinish(pgconn);
649 0 : return 0;
650 0 : }
|
