Branch data Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * common.c
4 : : * Common support routines for bin/scripts/
5 : : *
6 : : *
7 : : * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
8 : : * Portions Copyright (c) 1994, Regents of the University of California
9 : : *
10 : : * src/bin/scripts/common.c
11 : : *
12 : : *-------------------------------------------------------------------------
13 : : */
14 : :
15 : : #include "postgres_fe.h"
16 : :
17 : : #include <signal.h>
18 : : #include <unistd.h>
19 : :
20 : : #include "common.h"
21 : : #include "common/connect.h"
22 : : #include "common/logging.h"
23 : : #include "common/string.h"
24 : : #include "fe_utils/query_utils.h"
25 : : #include "fe_utils/string_utils.h"
26 : :
27 : : /*
28 : : * Split TABLE[(COLUMNS)] into TABLE and [(COLUMNS)] portions. When you
29 : : * finish using them, pg_free(*table). *columns is a pointer into "spec",
30 : : * possibly to its NUL terminator.
31 : : */
32 : : void
33 : 64 : splitTableColumnsSpec(const char *spec, int encoding,
34 : : char **table, const char **columns)
35 : : {
36 : 64 : bool inquotes = false;
37 : 64 : const char *cp = spec;
38 : :
39 : : /*
40 : : * Find the first '(' not identifier-quoted. Based on
41 : : * dequote_downcase_identifier().
42 : : */
43 [ + + + + : 890 : while (*cp && (*cp != '(' || inquotes))
+ + ]
44 : : {
45 [ + + ]: 826 : if (*cp == '"')
46 : : {
47 [ + + + + ]: 3 : if (inquotes && cp[1] == '"')
48 : 1 : cp++; /* pair does not affect quoting */
49 : : else
50 : 2 : inquotes = !inquotes;
51 : 3 : cp++;
52 : : }
53 : : else
54 : 823 : cp += PQmblenBounded(cp, encoding);
55 : : }
56 : 64 : *table = pnstrdup(spec, cp - spec);
57 : 64 : *columns = cp;
58 : 64 : }
59 : :
60 : : /*
61 : : * Break apart TABLE[(COLUMNS)] of "spec". With the reset_val of search_path
62 : : * in effect, have regclassin() interpret the TABLE portion. Append to "buf"
63 : : * the qualified name of TABLE, followed by any (COLUMNS). Exit on failure.
64 : : * We use this to interpret --table=foo under the search path psql would get,
65 : : * in advance of "ANALYZE public.foo" under the always-secure search path.
66 : : */
67 : : void
68 : 42 : appendQualifiedRelation(PQExpBuffer buf, const char *spec,
69 : : PGconn *conn, bool echo)
70 : : {
71 : : char *table;
72 : : const char *columns;
73 : : PQExpBufferData sql;
74 : : PGresult *res;
75 : : int ntups;
76 : :
77 : 42 : splitTableColumnsSpec(spec, PQclientEncoding(conn), &table, &columns);
78 : :
79 : : /*
80 : : * Query must remain ABSOLUTELY devoid of unqualified names. This would
81 : : * be unnecessary given a regclassin() variant taking a search_path
82 : : * argument.
83 : : */
84 : 42 : initPQExpBuffer(&sql);
85 : 42 : appendPQExpBufferStr(&sql,
86 : : "SELECT c.relname, ns.nspname\n"
87 : : " FROM pg_catalog.pg_class c,"
88 : : " pg_catalog.pg_namespace ns\n"
89 : : " WHERE c.relnamespace OPERATOR(pg_catalog.=) ns.oid\n"
90 : : " AND c.oid OPERATOR(pg_catalog.=) ");
91 : 42 : appendStringLiteralConn(&sql, table, conn);
92 : 42 : appendPQExpBufferStr(&sql, "::pg_catalog.regclass;");
93 : :
94 : 42 : executeCommand(conn, "RESET search_path;", echo);
95 : :
96 : : /*
97 : : * One row is a typical result, as is a nonexistent relation ERROR.
98 : : * regclassin() unconditionally accepts all-digits input as an OID; if no
99 : : * relation has that OID; this query returns no rows. Catalog corruption
100 : : * might elicit other row counts.
101 : : */
102 : 42 : res = executeQuery(conn, sql.data, echo);
103 : 41 : ntups = PQntuples(res);
104 [ - + ]: 41 : if (ntups != 1)
105 : : {
106 : 0 : pg_log_error(ngettext("query returned %d row instead of one: %s",
107 : : "query returned %d rows instead of one: %s",
108 : : ntups),
109 : : ntups, sql.data);
110 : 0 : PQfinish(conn);
111 : 0 : exit(1);
112 : : }
113 : 82 : appendPQExpBufferStr(buf,
114 : 41 : fmtQualifiedIdEnc(PQgetvalue(res, 0, 1),
115 : 41 : PQgetvalue(res, 0, 0),
116 : : PQclientEncoding(conn)));
117 : 41 : appendPQExpBufferStr(buf, columns);
118 : 41 : PQclear(res);
119 : 41 : termPQExpBuffer(&sql);
120 : 41 : pg_free(table);
121 : :
122 : 41 : PQclear(executeQuery(conn, ALWAYS_SECURE_SEARCH_PATH_SQL, echo));
123 : 41 : }
124 : :
125 : :
126 : : /*
127 : : * Check yes/no answer in a localized way. 1=yes, 0=no, -1=neither.
128 : : */
129 : :
130 : : /* translator: abbreviation for "yes" */
131 : : #define PG_YESLETTER gettext_noop("y")
132 : : /* translator: abbreviation for "no" */
133 : : #define PG_NOLETTER gettext_noop("n")
134 : :
135 : : bool
136 : 0 : yesno_prompt(const char *question)
137 : : {
138 : : char prompt[256];
139 : :
140 : : /*------
141 : : translator: This is a question followed by the translated options for
142 : : "yes" and "no". */
143 : 0 : snprintf(prompt, sizeof(prompt), _("%s (%s/%s) "),
144 : : _(question), _(PG_YESLETTER), _(PG_NOLETTER));
145 : :
146 : : for (;;)
147 : 0 : {
148 : : char *resp;
149 : :
150 : 0 : resp = simple_prompt(prompt, true);
151 : :
152 [ # # ]: 0 : if (strcmp(resp, _(PG_YESLETTER)) == 0)
153 : : {
154 : 0 : free(resp);
155 : 0 : return true;
156 : : }
157 [ # # ]: 0 : if (strcmp(resp, _(PG_NOLETTER)) == 0)
158 : : {
159 : 0 : free(resp);
160 : 0 : return false;
161 : : }
162 : 0 : free(resp);
163 : :
164 : 0 : printf(_("Please answer \"%s\" or \"%s\".\n"),
165 : : _(PG_YESLETTER), _(PG_NOLETTER));
166 : : }
167 : : }
|
