mirror of
https://github.com/postgres/postgres.git
synced 2026-02-28 12:20:43 -05:00
61ad93cdd4
This commit adds fmtIdEnc() and fmtQualifiedIdEnc(), which allow to specify the encoding as an explicit argument. Additionally setFmtEncoding() is provided, which defines the encoding when no explicit encoding is provided, to avoid breaking all code using fmtId(). All users of fmtId()/fmtQualifiedId() are either converted to the explicit version or a call to setFmtEncoding() has been added. This commit does not yet utilize the now well-defined encoding, that will happen in a subsequent commit. Reviewed-by: Noah Misch <noah@leadboat.com> Reviewed-by: Tom Lane <tgl@sss.pgh.pa.us> Backpatch-through: 13 Security: CVE-2025-1094
329 lines
9 KiB
C
329 lines
9 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* createdb
|
|
*
|
|
* Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
|
|
* Portions Copyright (c) 1994, Regents of the University of California
|
|
*
|
|
* src/bin/scripts/createdb.c
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
#include "postgres_fe.h"
|
|
|
|
#include "common.h"
|
|
#include "common/logging.h"
|
|
#include "fe_utils/option_utils.h"
|
|
#include "fe_utils/string_utils.h"
|
|
|
|
|
|
static void help(const char *progname);
|
|
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
static struct option long_options[] = {
|
|
{"host", required_argument, NULL, 'h'},
|
|
{"port", required_argument, NULL, 'p'},
|
|
{"username", required_argument, NULL, 'U'},
|
|
{"no-password", no_argument, NULL, 'w'},
|
|
{"password", no_argument, NULL, 'W'},
|
|
{"echo", no_argument, NULL, 'e'},
|
|
{"owner", required_argument, NULL, 'O'},
|
|
{"tablespace", required_argument, NULL, 'D'},
|
|
{"template", required_argument, NULL, 'T'},
|
|
{"encoding", required_argument, NULL, 'E'},
|
|
{"strategy", required_argument, NULL, 'S'},
|
|
{"lc-collate", required_argument, NULL, 1},
|
|
{"lc-ctype", required_argument, NULL, 2},
|
|
{"locale", required_argument, NULL, 'l'},
|
|
{"maintenance-db", required_argument, NULL, 3},
|
|
{"locale-provider", required_argument, NULL, 4},
|
|
{"builtin-locale", required_argument, NULL, 5},
|
|
{"icu-locale", required_argument, NULL, 6},
|
|
{"icu-rules", required_argument, NULL, 7},
|
|
{NULL, 0, NULL, 0}
|
|
};
|
|
|
|
const char *progname;
|
|
int optindex;
|
|
int c;
|
|
|
|
const char *dbname = NULL;
|
|
const char *maintenance_db = NULL;
|
|
char *comment = NULL;
|
|
char *host = NULL;
|
|
char *port = NULL;
|
|
char *username = NULL;
|
|
enum trivalue prompt_password = TRI_DEFAULT;
|
|
ConnParams cparams;
|
|
bool echo = false;
|
|
char *owner = NULL;
|
|
char *tablespace = NULL;
|
|
char *template = NULL;
|
|
char *encoding = NULL;
|
|
char *strategy = NULL;
|
|
char *lc_collate = NULL;
|
|
char *lc_ctype = NULL;
|
|
char *locale = NULL;
|
|
char *locale_provider = NULL;
|
|
char *builtin_locale = NULL;
|
|
char *icu_locale = NULL;
|
|
char *icu_rules = NULL;
|
|
|
|
PQExpBufferData sql;
|
|
|
|
PGconn *conn;
|
|
PGresult *result;
|
|
|
|
pg_logging_init(argv[0]);
|
|
progname = get_progname(argv[0]);
|
|
set_pglocale_pgservice(argv[0], PG_TEXTDOMAIN("pgscripts"));
|
|
|
|
handle_help_version_opts(argc, argv, "createdb", help);
|
|
|
|
while ((c = getopt_long(argc, argv, "D:eE:h:l:O:p:S:T:U:wW", long_options, &optindex)) != -1)
|
|
{
|
|
switch (c)
|
|
{
|
|
case 'D':
|
|
tablespace = pg_strdup(optarg);
|
|
break;
|
|
case 'e':
|
|
echo = true;
|
|
break;
|
|
case 'E':
|
|
encoding = pg_strdup(optarg);
|
|
break;
|
|
case 'h':
|
|
host = pg_strdup(optarg);
|
|
break;
|
|
case 'l':
|
|
locale = pg_strdup(optarg);
|
|
break;
|
|
case 'O':
|
|
owner = pg_strdup(optarg);
|
|
break;
|
|
case 'p':
|
|
port = pg_strdup(optarg);
|
|
break;
|
|
case 'S':
|
|
strategy = pg_strdup(optarg);
|
|
break;
|
|
case 'T':
|
|
template = pg_strdup(optarg);
|
|
break;
|
|
case 'U':
|
|
username = pg_strdup(optarg);
|
|
break;
|
|
case 'w':
|
|
prompt_password = TRI_NO;
|
|
break;
|
|
case 'W':
|
|
prompt_password = TRI_YES;
|
|
break;
|
|
case 1:
|
|
lc_collate = pg_strdup(optarg);
|
|
break;
|
|
case 2:
|
|
lc_ctype = pg_strdup(optarg);
|
|
break;
|
|
case 3:
|
|
maintenance_db = pg_strdup(optarg);
|
|
break;
|
|
case 4:
|
|
locale_provider = pg_strdup(optarg);
|
|
break;
|
|
case 5:
|
|
builtin_locale = pg_strdup(optarg);
|
|
break;
|
|
case 6:
|
|
icu_locale = pg_strdup(optarg);
|
|
break;
|
|
case 7:
|
|
icu_rules = pg_strdup(optarg);
|
|
break;
|
|
default:
|
|
/* getopt_long already emitted a complaint */
|
|
pg_log_error_hint("Try \"%s --help\" for more information.", progname);
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
switch (argc - optind)
|
|
{
|
|
case 0:
|
|
break;
|
|
case 1:
|
|
dbname = argv[optind];
|
|
break;
|
|
case 2:
|
|
dbname = argv[optind];
|
|
comment = argv[optind + 1];
|
|
break;
|
|
default:
|
|
pg_log_error("too many command-line arguments (first is \"%s\")",
|
|
argv[optind + 2]);
|
|
pg_log_error_hint("Try \"%s --help\" for more information.", progname);
|
|
exit(1);
|
|
}
|
|
|
|
if (encoding)
|
|
{
|
|
if (pg_char_to_encoding(encoding) < 0)
|
|
pg_fatal("\"%s\" is not a valid encoding name", encoding);
|
|
}
|
|
|
|
if (dbname == NULL)
|
|
{
|
|
if (getenv("PGDATABASE"))
|
|
dbname = getenv("PGDATABASE");
|
|
else if (getenv("PGUSER"))
|
|
dbname = getenv("PGUSER");
|
|
else
|
|
dbname = get_user_name_or_exit(progname);
|
|
}
|
|
|
|
/* No point in trying to use postgres db when creating postgres db. */
|
|
if (maintenance_db == NULL && strcmp(dbname, "postgres") == 0)
|
|
maintenance_db = "template1";
|
|
|
|
cparams.dbname = maintenance_db;
|
|
cparams.pghost = host;
|
|
cparams.pgport = port;
|
|
cparams.pguser = username;
|
|
cparams.prompt_password = prompt_password;
|
|
cparams.override_dbname = NULL;
|
|
|
|
conn = connectMaintenanceDatabase(&cparams, progname, echo);
|
|
|
|
setFmtEncoding(PQclientEncoding(conn));
|
|
|
|
initPQExpBuffer(&sql);
|
|
|
|
appendPQExpBuffer(&sql, "CREATE DATABASE %s",
|
|
fmtId(dbname));
|
|
|
|
if (owner)
|
|
appendPQExpBuffer(&sql, " OWNER %s", fmtId(owner));
|
|
if (tablespace)
|
|
appendPQExpBuffer(&sql, " TABLESPACE %s", fmtId(tablespace));
|
|
if (encoding)
|
|
{
|
|
appendPQExpBufferStr(&sql, " ENCODING ");
|
|
appendStringLiteralConn(&sql, encoding, conn);
|
|
}
|
|
if (strategy)
|
|
appendPQExpBuffer(&sql, " STRATEGY %s", fmtId(strategy));
|
|
if (template)
|
|
appendPQExpBuffer(&sql, " TEMPLATE %s", fmtId(template));
|
|
if (locale)
|
|
{
|
|
appendPQExpBufferStr(&sql, " LOCALE ");
|
|
appendStringLiteralConn(&sql, locale, conn);
|
|
}
|
|
if (builtin_locale)
|
|
{
|
|
appendPQExpBufferStr(&sql, " BUILTIN_LOCALE ");
|
|
appendStringLiteralConn(&sql, builtin_locale, conn);
|
|
}
|
|
if (lc_collate)
|
|
{
|
|
appendPQExpBufferStr(&sql, " LC_COLLATE ");
|
|
appendStringLiteralConn(&sql, lc_collate, conn);
|
|
}
|
|
if (lc_ctype)
|
|
{
|
|
appendPQExpBufferStr(&sql, " LC_CTYPE ");
|
|
appendStringLiteralConn(&sql, lc_ctype, conn);
|
|
}
|
|
if (locale_provider)
|
|
appendPQExpBuffer(&sql, " LOCALE_PROVIDER %s", fmtId(locale_provider));
|
|
if (icu_locale)
|
|
{
|
|
appendPQExpBufferStr(&sql, " ICU_LOCALE ");
|
|
appendStringLiteralConn(&sql, icu_locale, conn);
|
|
}
|
|
if (icu_rules)
|
|
{
|
|
appendPQExpBufferStr(&sql, " ICU_RULES ");
|
|
appendStringLiteralConn(&sql, icu_rules, conn);
|
|
}
|
|
|
|
appendPQExpBufferChar(&sql, ';');
|
|
|
|
if (echo)
|
|
printf("%s\n", sql.data);
|
|
result = PQexec(conn, sql.data);
|
|
|
|
if (PQresultStatus(result) != PGRES_COMMAND_OK)
|
|
{
|
|
pg_log_error("database creation failed: %s", PQerrorMessage(conn));
|
|
PQfinish(conn);
|
|
exit(1);
|
|
}
|
|
|
|
PQclear(result);
|
|
|
|
if (comment)
|
|
{
|
|
printfPQExpBuffer(&sql, "COMMENT ON DATABASE %s IS ", fmtId(dbname));
|
|
appendStringLiteralConn(&sql, comment, conn);
|
|
appendPQExpBufferChar(&sql, ';');
|
|
|
|
if (echo)
|
|
printf("%s\n", sql.data);
|
|
result = PQexec(conn, sql.data);
|
|
|
|
if (PQresultStatus(result) != PGRES_COMMAND_OK)
|
|
{
|
|
pg_log_error("comment creation failed (database was created): %s",
|
|
PQerrorMessage(conn));
|
|
PQfinish(conn);
|
|
exit(1);
|
|
}
|
|
|
|
PQclear(result);
|
|
}
|
|
|
|
PQfinish(conn);
|
|
|
|
exit(0);
|
|
}
|
|
|
|
|
|
static void
|
|
help(const char *progname)
|
|
{
|
|
printf(_("%s creates a PostgreSQL database.\n\n"), progname);
|
|
printf(_("Usage:\n"));
|
|
printf(_(" %s [OPTION]... [DBNAME] [DESCRIPTION]\n"), progname);
|
|
printf(_("\nOptions:\n"));
|
|
printf(_(" -D, --tablespace=TABLESPACE default tablespace for the database\n"));
|
|
printf(_(" -e, --echo show the commands being sent to the server\n"));
|
|
printf(_(" -E, --encoding=ENCODING encoding for the database\n"));
|
|
printf(_(" -l, --locale=LOCALE locale settings for the database\n"));
|
|
printf(_(" --lc-collate=LOCALE LC_COLLATE setting for the database\n"));
|
|
printf(_(" --lc-ctype=LOCALE LC_CTYPE setting for the database\n"));
|
|
printf(_(" --builtin-locale=LOCALE builtin locale setting for the database\n"));
|
|
printf(_(" --icu-locale=LOCALE ICU locale setting for the database\n"));
|
|
printf(_(" --icu-rules=RULES ICU rules setting for the database\n"));
|
|
printf(_(" --locale-provider={builtin|libc|icu}\n"
|
|
" locale provider for the database's default collation\n"));
|
|
printf(_(" -O, --owner=OWNER database user to own the new database\n"));
|
|
printf(_(" -S, --strategy=STRATEGY database creation strategy wal_log or file_copy\n"));
|
|
printf(_(" -T, --template=TEMPLATE template database to copy\n"));
|
|
printf(_(" -V, --version output version information, then exit\n"));
|
|
printf(_(" -?, --help show this help, then exit\n"));
|
|
printf(_("\nConnection options:\n"));
|
|
printf(_(" -h, --host=HOSTNAME database server host or socket directory\n"));
|
|
printf(_(" -p, --port=PORT database server port\n"));
|
|
printf(_(" -U, --username=USERNAME user name to connect as\n"));
|
|
printf(_(" -w, --no-password never prompt for password\n"));
|
|
printf(_(" -W, --password force password prompt\n"));
|
|
printf(_(" --maintenance-db=DBNAME alternate maintenance database\n"));
|
|
printf(_("\nBy default, a database with the same name as the current user is created.\n"));
|
|
printf(_("\nReport bugs to <%s>.\n"), PACKAGE_BUGREPORT);
|
|
printf(_("%s home page: <%s>\n"), PACKAGE_NAME, PACKAGE_URL);
|
|
}
|