Line data Source code
1 : /*-------------------------------------------------------------------------
2 : *
3 : * initdb --- initialize a PostgreSQL installation
4 : *
5 : * initdb creates (initializes) a PostgreSQL database cluster (site,
6 : * instance, installation, whatever). A database cluster is a
7 : * collection of PostgreSQL databases all managed by the same server.
8 : *
9 : * To create the database cluster, we create the directory that contains
10 : * all its data, create the files that hold the global tables, create
11 : * a few other control files for it, and create three databases: the
12 : * template databases "template0" and "template1", and a default user
13 : * database "postgres".
14 : *
15 : * The template databases are ordinary PostgreSQL databases. template0
16 : * is never supposed to change after initdb, whereas template1 can be
17 : * changed to add site-local standard data. Either one can be copied
18 : * to produce a new database.
19 : *
20 : * For largely-historical reasons, the template1 database is the one built
21 : * by the basic bootstrap process. After it is complete, template0 and
22 : * the default database, postgres, are made just by copying template1.
23 : *
24 : * To create template1, we run the postgres (backend) program in bootstrap
25 : * mode and feed it data from the postgres.bki library file. After this
26 : * initial bootstrap phase, some additional stuff is created by normal
27 : * SQL commands fed to a standalone backend. Some of those commands are
28 : * just embedded into this program (yeah, it's ugly), but larger chunks
29 : * are taken from script files.
30 : *
31 : *
32 : * Note:
33 : * The program has some memory leakage - it isn't worth cleaning it up.
34 : *
35 : * This is a C implementation of the previous shell script for setting up a
36 : * PostgreSQL cluster location, and should be highly compatible with it.
37 : * author of C translation: Andrew Dunstan mailto:andrew@dunslane.net
38 : *
39 : * This code is released under the terms of the PostgreSQL License.
40 : *
41 : * Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
42 : * Portions Copyright (c) 1994, Regents of the University of California
43 : *
44 : * src/bin/initdb/initdb.c
45 : *
46 : *-------------------------------------------------------------------------
47 : */
48 :
49 : #include "postgres_fe.h"
50 :
51 : #include <dirent.h>
52 : #include <fcntl.h>
53 : #include <netdb.h>
54 : #include <sys/socket.h>
55 : #include <sys/stat.h>
56 : #ifdef USE_ICU
57 : #include <unicode/ucol.h>
58 : #endif
59 : #include <unistd.h>
60 : #include <signal.h>
61 : #include <time.h>
62 :
63 : #ifdef HAVE_SHM_OPEN
64 : #include "sys/mman.h"
65 : #endif
66 :
67 : #include "access/xlog_internal.h"
68 : #include "catalog/pg_authid_d.h"
69 : #include "catalog/pg_class_d.h" /* pgrminclude ignore */
70 : #include "catalog/pg_collation_d.h"
71 : #include "catalog/pg_database_d.h" /* pgrminclude ignore */
72 : #include "common/file_perm.h"
73 : #include "common/file_utils.h"
74 : #include "common/logging.h"
75 : #include "common/pg_prng.h"
76 : #include "common/restricted_token.h"
77 : #include "common/string.h"
78 : #include "common/username.h"
79 : #include "fe_utils/option_utils.h"
80 : #include "fe_utils/string_utils.h"
81 : #include "getopt_long.h"
82 : #include "mb/pg_wchar.h"
83 : #include "miscadmin.h"
84 :
85 :
86 : /* Ideally this would be in a .h file, but it hardly seems worth the trouble */
87 : extern const char *select_default_timezone(const char *share_path);
88 :
89 : /* simple list of strings */
90 : typedef struct _stringlist
91 : {
92 : char *str;
93 : struct _stringlist *next;
94 : } _stringlist;
95 :
96 : static const char *const auth_methods_host[] = {
97 : "trust", "reject", "scram-sha-256", "md5", "password", "ident", "radius",
98 : #ifdef ENABLE_GSS
99 : "gss",
100 : #endif
101 : #ifdef ENABLE_SSPI
102 : "sspi",
103 : #endif
104 : #ifdef USE_PAM
105 : "pam", "pam ",
106 : #endif
107 : #ifdef USE_BSD_AUTH
108 : "bsd",
109 : #endif
110 : #ifdef USE_LDAP
111 : "ldap",
112 : #endif
113 : #ifdef USE_SSL
114 : "cert",
115 : #endif
116 : NULL
117 : };
118 : static const char *const auth_methods_local[] = {
119 : "trust", "reject", "scram-sha-256", "md5", "password", "peer", "radius",
120 : #ifdef USE_PAM
121 : "pam", "pam ",
122 : #endif
123 : #ifdef USE_BSD_AUTH
124 : "bsd",
125 : #endif
126 : #ifdef USE_LDAP
127 : "ldap",
128 : #endif
129 : NULL
130 : };
131 :
132 : /*
133 : * these values are passed in by makefile defines
134 : */
135 : static char *share_path = NULL;
136 :
137 : /* values to be obtained from arguments */
138 : static char *pg_data = NULL;
139 : static char *encoding = NULL;
140 : static char *locale = NULL;
141 : static char *lc_collate = NULL;
142 : static char *lc_ctype = NULL;
143 : static char *lc_monetary = NULL;
144 : static char *lc_numeric = NULL;
145 : static char *lc_time = NULL;
146 : static char *lc_messages = NULL;
147 : static char locale_provider = COLLPROVIDER_LIBC;
148 : static bool builtin_locale_specified = false;
149 : static char *datlocale = NULL;
150 : static bool icu_locale_specified = false;
151 : static char *icu_rules = NULL;
152 : static const char *default_text_search_config = NULL;
153 : static char *username = NULL;
154 : static bool pwprompt = false;
155 : static char *pwfilename = NULL;
156 : static char *superuser_password = NULL;
157 : static const char *authmethodhost = NULL;
158 : static const char *authmethodlocal = NULL;
159 : static _stringlist *extra_guc_names = NULL;
160 : static _stringlist *extra_guc_values = NULL;
161 : static bool debug = false;
162 : static bool noclean = false;
163 : static bool noinstructions = false;
164 : static bool do_sync = true;
165 : static bool sync_only = false;
166 : static bool show_setting = false;
167 : static bool data_checksums = false;
168 : static char *xlog_dir = NULL;
169 : static int wal_segment_size_mb = (DEFAULT_XLOG_SEG_SIZE) / (1024 * 1024);
170 : static DataDirSyncMethod sync_method = DATA_DIR_SYNC_METHOD_FSYNC;
171 :
172 :
173 : /* internal vars */
174 : static const char *progname;
175 : static int encodingid;
176 : static char *bki_file;
177 : static char *hba_file;
178 : static char *ident_file;
179 : static char *conf_file;
180 : static char *dictionary_file;
181 : static char *info_schema_file;
182 : static char *features_file;
183 : static char *system_constraints_file;
184 : static char *system_functions_file;
185 : static char *system_views_file;
186 : static bool success = false;
187 : static bool made_new_pgdata = false;
188 : static bool found_existing_pgdata = false;
189 : static bool made_new_xlogdir = false;
190 : static bool found_existing_xlogdir = false;
191 : static char infoversion[100];
192 : static bool caught_signal = false;
193 : static bool output_failed = false;
194 : static int output_errno = 0;
195 : static char *pgdata_native;
196 :
197 : /* defaults */
198 : static int n_connections = 10;
199 : static int n_buffers = 50;
200 : static const char *dynamic_shared_memory_type = NULL;
201 : static const char *default_timezone = NULL;
202 :
203 : /*
204 : * Warning messages for authentication methods
205 : */
206 : #define AUTHTRUST_WARNING \
207 : "# CAUTION: Configuring the system for local \"trust\" authentication\n" \
208 : "# allows any local user to connect as any PostgreSQL user, including\n" \
209 : "# the database superuser. If you do not trust all your local users,\n" \
210 : "# use another authentication method.\n"
211 : static bool authwarning = false;
212 :
213 : /*
214 : * Centralized knowledge of switches to pass to backend
215 : *
216 : * Note: we run the backend with -F (fsync disabled) and then do a single
217 : * pass of fsync'ing at the end. This is faster than fsync'ing each step.
218 : *
219 : * Note: in the shell-script version, we also passed PGDATA as a -D switch,
220 : * but here it is more convenient to pass it as an environment variable
221 : * (no quoting to worry about).
222 : */
223 : static const char *const boot_options = "-F -c log_checkpoints=false";
224 : static const char *const backend_options = "--single -F -O -j -c search_path=pg_catalog -c exit_on_error=true -c log_checkpoints=false";
225 :
226 : /* Additional switches to pass to backend (either boot or standalone) */
227 : static char *extra_options = "";
228 :
229 : static const char *const subdirs[] = {
230 : "global",
231 : "pg_wal/archive_status",
232 : "pg_wal/summaries",
233 : "pg_commit_ts",
234 : "pg_dynshmem",
235 : "pg_notify",
236 : "pg_serial",
237 : "pg_snapshots",
238 : "pg_subtrans",
239 : "pg_twophase",
240 : "pg_multixact",
241 : "pg_multixact/members",
242 : "pg_multixact/offsets",
243 : "base",
244 : "base/1",
245 : "pg_replslot",
246 : "pg_tblspc",
247 : "pg_stat",
248 : "pg_stat_tmp",
249 : "pg_xact",
250 : "pg_logical",
251 : "pg_logical/snapshots",
252 : "pg_logical/mappings"
253 : };
254 :
255 :
256 : /* path to 'initdb' binary directory */
257 : static char bin_path[MAXPGPATH];
258 : static char backend_exec[MAXPGPATH];
259 :
260 : static char **replace_token(char **lines,
261 : const char *token, const char *replacement);
262 : static char **replace_guc_value(char **lines,
263 : const char *guc_name, const char *guc_value,
264 : bool mark_as_comment);
265 : static bool guc_value_requires_quotes(const char *guc_value);
266 : static char **readfile(const char *path);
267 : static void writefile(char *path, char **lines);
268 : static FILE *popen_check(const char *command, const char *mode);
269 : static char *get_id(void);
270 : static int get_encoding_id(const char *encoding_name);
271 : static void set_input(char **dest, const char *filename);
272 : static void check_input(char *path);
273 : static void write_version_file(const char *extrapath);
274 : static void set_null_conf(void);
275 : static void test_config_settings(void);
276 : static bool test_specific_config_settings(int test_conns, int test_buffs);
277 : static void setup_config(void);
278 : static void bootstrap_template1(void);
279 : static void setup_auth(FILE *cmdfd);
280 : static void get_su_pwd(void);
281 : static void setup_depend(FILE *cmdfd);
282 : static void setup_run_file(FILE *cmdfd, const char *filename);
283 : static void setup_description(FILE *cmdfd);
284 : static void setup_collation(FILE *cmdfd);
285 : static void setup_privileges(FILE *cmdfd);
286 : static void set_info_version(void);
287 : static void setup_schema(FILE *cmdfd);
288 : static void load_plpgsql(FILE *cmdfd);
289 : static void vacuum_db(FILE *cmdfd);
290 : static void make_template0(FILE *cmdfd);
291 : static void make_postgres(FILE *cmdfd);
292 : static void trapsig(SIGNAL_ARGS);
293 : static void check_ok(void);
294 : static char *escape_quotes(const char *src);
295 : static char *escape_quotes_bki(const char *src);
296 : static int locale_date_order(const char *locale);
297 : static void check_locale_name(int category, const char *locale,
298 : char **canonname);
299 : static bool check_locale_encoding(const char *locale, int user_enc);
300 : static void setlocales(void);
301 : static void usage(const char *progname);
302 : void setup_pgdata(void);
303 : void setup_bin_paths(const char *argv0);
304 : void setup_data_file_paths(void);
305 : void setup_locale_encoding(void);
306 : void setup_signals(void);
307 : void setup_text_search(void);
308 : void create_data_directory(void);
309 : void create_xlog_or_symlink(void);
310 : void warn_on_mount_point(int error);
311 : void initialize_data_directory(void);
312 :
313 : /*
314 : * macros for running pipes to postgres
315 : */
316 : #define PG_CMD_DECL FILE *cmdfd
317 :
318 : #define PG_CMD_OPEN(cmd) \
319 : do { \
320 : cmdfd = popen_check(cmd, "w"); \
321 : if (cmdfd == NULL) \
322 : exit(1); /* message already printed by popen_check */ \
323 : } while (0)
324 :
325 : #define PG_CMD_CLOSE() \
326 : do { \
327 : if (pclose_check(cmdfd)) \
328 : exit(1); /* message already printed by pclose_check */ \
329 : } while (0)
330 :
331 : #define PG_CMD_PUTS(line) \
332 : do { \
333 : if (fputs(line, cmdfd) < 0 || fflush(cmdfd) < 0) \
334 : output_failed = true, output_errno = errno; \
335 : } while (0)
336 :
337 : #define PG_CMD_PRINTF(fmt, ...) \
338 : do { \
339 : if (fprintf(cmdfd, fmt, __VA_ARGS__) < 0 || fflush(cmdfd) < 0) \
340 : output_failed = true, output_errno = errno; \
341 : } while (0)
342 :
343 : /*
344 : * Escape single quotes and backslashes, suitably for insertions into
345 : * configuration files or SQL E'' strings.
346 : */
347 : static char *
348 978 : escape_quotes(const char *src)
349 : {
350 978 : char *result = escape_single_quotes_ascii(src);
351 :
352 978 : if (!result)
353 0 : pg_fatal("out of memory");
354 978 : return result;
355 : }
356 :
357 : /*
358 : * Escape a field value to be inserted into the BKI data.
359 : * Run the value through escape_quotes (which will be inverted
360 : * by the backend's DeescapeQuotedString() function), then wrap
361 : * the value in single quotes, even if that isn't strictly necessary.
362 : */
363 : static char *
364 264 : escape_quotes_bki(const char *src)
365 : {
366 : char *result;
367 264 : char *data = escape_quotes(src);
368 : char *resultp;
369 : char *datap;
370 :
371 264 : result = (char *) pg_malloc(strlen(data) + 3);
372 264 : resultp = result;
373 264 : *resultp++ = '\'';
374 2066 : for (datap = data; *datap; datap++)
375 1802 : *resultp++ = *datap;
376 264 : *resultp++ = '\'';
377 264 : *resultp = '\0';
378 :
379 264 : free(data);
380 264 : return result;
381 : }
382 :
383 : /*
384 : * Add an item at the end of a stringlist.
385 : */
386 : static void
387 20 : add_stringlist_item(_stringlist **listhead, const char *str)
388 : {
389 20 : _stringlist *newentry = pg_malloc(sizeof(_stringlist));
390 : _stringlist *oldentry;
391 :
392 20 : newentry->str = pg_strdup(str);
393 20 : newentry->next = NULL;
394 20 : if (*listhead == NULL)
395 12 : *listhead = newentry;
396 : else
397 : {
398 12 : for (oldentry = *listhead; oldentry->next; oldentry = oldentry->next)
399 : /* skip */ ;
400 8 : oldentry->next = newentry;
401 : }
402 20 : }
403 :
404 : /*
405 : * Modify the array of lines, replacing "token" by "replacement"
406 : * the first time it occurs on each line.
407 : *
408 : * The array must be a malloc'd array of individually malloc'd strings.
409 : * We free any discarded strings.
410 : *
411 : * This does most of what sed was used for in the shell script, but
412 : * doesn't need any regexp stuff.
413 : */
414 : static char **
415 1230 : replace_token(char **lines, const char *token, const char *replacement)
416 : {
417 : int toklen,
418 : replen,
419 : diff;
420 :
421 1230 : toklen = strlen(token);
422 1230 : replen = strlen(replacement);
423 1230 : diff = replen - toklen;
424 :
425 10576606 : for (int i = 0; lines[i]; i++)
426 : {
427 : char *where;
428 : char *newline;
429 : int pre;
430 :
431 : /* nothing to do if no change needed */
432 10575376 : if ((where = strstr(lines[i], token)) == NULL)
433 10572588 : continue;
434 :
435 : /* if we get here a change is needed - set up new line */
436 :
437 2788 : newline = (char *) pg_malloc(strlen(lines[i]) + diff + 1);
438 :
439 2788 : pre = where - lines[i];
440 :
441 2788 : memcpy(newline, lines[i], pre);
442 :
443 2788 : memcpy(newline + pre, replacement, replen);
444 :
445 2788 : strcpy(newline + pre + replen, lines[i] + pre + toklen);
446 :
447 2788 : free(lines[i]);
448 2788 : lines[i] = newline;
449 : }
450 :
451 1230 : return lines;
452 : }
453 :
454 : /*
455 : * Modify the array of lines, replacing the possibly-commented-out
456 : * assignment of parameter guc_name with a live assignment of guc_value.
457 : * The value will be suitably quoted.
458 : *
459 : * If mark_as_comment is true, the replacement line is prefixed with '#'.
460 : * This is used for fixing up cases where the effective default might not
461 : * match what is in postgresql.conf.sample.
462 : *
463 : * We assume there's at most one matching assignment. If we find no match,
464 : * append a new line with the desired assignment.
465 : *
466 : * The array must be a malloc'd array of individually malloc'd strings.
467 : * We free any discarded strings.
468 : */
469 : static char **
470 1414 : replace_guc_value(char **lines, const char *guc_name, const char *guc_value,
471 : bool mark_as_comment)
472 : {
473 1414 : int namelen = strlen(guc_name);
474 1414 : PQExpBuffer newline = createPQExpBuffer();
475 : int i;
476 :
477 : /* prepare the replacement line, except for possible comment and newline */
478 1414 : if (mark_as_comment)
479 328 : appendPQExpBufferChar(newline, '#');
480 1414 : appendPQExpBuffer(newline, "%s = ", guc_name);
481 1414 : if (guc_value_requires_quotes(guc_value))
482 554 : appendPQExpBuffer(newline, "'%s'", escape_quotes(guc_value));
483 : else
484 860 : appendPQExpBufferStr(newline, guc_value);
485 :
486 613058 : for (i = 0; lines[i]; i++)
487 : {
488 : const char *where;
489 : const char *namestart;
490 :
491 : /*
492 : * Look for a line assigning to guc_name. Typically it will be
493 : * preceded by '#', but that might not be the case if a -c switch
494 : * overrides a previous assignment. We allow leading whitespace too,
495 : * although normally there wouldn't be any.
496 : */
497 613056 : where = lines[i];
498 2587414 : while (*where == '#' || isspace((unsigned char) *where))
499 1974358 : where++;
500 613056 : if (pg_strncasecmp(where, guc_name, namelen) != 0)
501 611644 : continue;
502 1412 : namestart = where;
503 1412 : where += namelen;
504 2824 : while (isspace((unsigned char) *where))
505 1412 : where++;
506 1412 : if (*where != '=')
507 0 : continue;
508 :
509 : /* found it -- let's use the canonical casing shown in the file */
510 1412 : memcpy(&newline->data[mark_as_comment ? 1 : 0], namestart, namelen);
511 :
512 : /* now append the original comment if any */
513 1412 : where = strrchr(where, '#');
514 1412 : if (where)
515 : {
516 : /*
517 : * We try to preserve original indentation, which is tedious.
518 : * oldindent and newindent are measured in de-tab-ified columns.
519 : */
520 : const char *ptr;
521 918 : int oldindent = 0;
522 : int newindent;
523 :
524 23572 : for (ptr = lines[i]; ptr < where; ptr++)
525 : {
526 22654 : if (*ptr == '\t')
527 2432 : oldindent += 8 - (oldindent % 8);
528 : else
529 20222 : oldindent++;
530 : }
531 : /* ignore the possibility of tabs in guc_value */
532 918 : newindent = newline->len;
533 : /* append appropriate tabs and spaces, forcing at least one */
534 918 : oldindent = Max(oldindent, newindent + 1);
535 3356 : while (newindent < oldindent)
536 : {
537 2438 : int newindent_if_tab = newindent + 8 - (newindent % 8);
538 :
539 2438 : if (newindent_if_tab <= oldindent)
540 : {
541 2438 : appendPQExpBufferChar(newline, '\t');
542 2438 : newindent = newindent_if_tab;
543 : }
544 : else
545 : {
546 0 : appendPQExpBufferChar(newline, ' ');
547 0 : newindent++;
548 : }
549 : }
550 : /* and finally append the old comment */
551 918 : appendPQExpBufferStr(newline, where);
552 : /* we'll have appended the original newline; don't add another */
553 : }
554 : else
555 494 : appendPQExpBufferChar(newline, '\n');
556 :
557 1412 : free(lines[i]);
558 1412 : lines[i] = newline->data;
559 :
560 1412 : break; /* assume there's only one match */
561 : }
562 :
563 1414 : if (lines[i] == NULL)
564 : {
565 : /*
566 : * No match, so append a new entry. (We rely on the bootstrap server
567 : * to complain if it's not a valid GUC name.)
568 : */
569 2 : appendPQExpBufferChar(newline, '\n');
570 2 : lines = pg_realloc_array(lines, char *, i + 2);
571 2 : lines[i++] = newline->data;
572 2 : lines[i] = NULL; /* keep the array null-terminated */
573 : }
574 :
575 1414 : free(newline); /* but don't free newline->data */
576 :
577 1414 : return lines;
578 : }
579 :
580 : /*
581 : * Decide if we should quote a replacement GUC value. We aren't too tense
582 : * here, but we'd like to avoid quoting simple identifiers and numbers
583 : * with units, which are common cases.
584 : */
585 : static bool
586 1414 : guc_value_requires_quotes(const char *guc_value)
587 : {
588 : /* Don't use <ctype.h> macros here, they might accept too much */
589 : #define LETTERS "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
590 : #define DIGITS "0123456789"
591 :
592 1414 : if (*guc_value == '\0')
593 0 : return true; /* empty string must be quoted */
594 1414 : if (strchr(LETTERS, *guc_value))
595 : {
596 742 : if (strspn(guc_value, LETTERS DIGITS) == strlen(guc_value))
597 270 : return false; /* it's an identifier */
598 472 : return true; /* nope */
599 : }
600 672 : if (strchr(DIGITS, *guc_value))
601 : {
602 : /* skip over digits */
603 590 : guc_value += strspn(guc_value, DIGITS);
604 : /* there can be zero or more unit letters after the digits */
605 590 : if (strspn(guc_value, LETTERS) == strlen(guc_value))
606 590 : return false; /* it's a number, possibly with units */
607 0 : return true; /* nope */
608 : }
609 82 : return true; /* all else must be quoted */
610 : }
611 :
612 : /*
613 : * get the lines from a text file
614 : *
615 : * The result is a malloc'd array of individually malloc'd strings.
616 : */
617 : static char **
618 728 : readfile(const char *path)
619 : {
620 : char **result;
621 : FILE *infile;
622 : StringInfoData line;
623 : int maxlines;
624 : int n;
625 :
626 728 : if ((infile = fopen(path, "r")) == NULL)
627 0 : pg_fatal("could not open file \"%s\" for reading: %m", path);
628 :
629 728 : initStringInfo(&line);
630 :
631 728 : maxlines = 1024;
632 728 : result = (char **) pg_malloc(maxlines * sizeof(char *));
633 :
634 728 : n = 0;
635 1572160 : while (pg_get_line_buf(infile, &line))
636 : {
637 : /* make sure there will be room for a trailing NULL pointer */
638 1571432 : if (n >= maxlines - 1)
639 : {
640 648 : maxlines *= 2;
641 648 : result = (char **) pg_realloc(result, maxlines * sizeof(char *));
642 : }
643 :
644 1571432 : result[n++] = pg_strdup(line.data);
645 : }
646 728 : result[n] = NULL;
647 :
648 728 : pfree(line.data);
649 :
650 728 : fclose(infile);
651 :
652 728 : return result;
653 : }
654 :
655 : /*
656 : * write an array of lines to a file
657 : *
658 : * "lines" must be a malloc'd array of individually malloc'd strings.
659 : * All that data is freed here.
660 : *
661 : * This is only used to write text files. Use fopen "w" not PG_BINARY_W
662 : * so that the resulting configuration files are nicely editable on Windows.
663 : */
664 : static void
665 328 : writefile(char *path, char **lines)
666 : {
667 : FILE *out_file;
668 : char **line;
669 :
670 328 : if ((out_file = fopen(path, "w")) == NULL)
671 0 : pg_fatal("could not open file \"%s\" for writing: %m", path);
672 85446 : for (line = lines; *line != NULL; line++)
673 : {
674 85118 : if (fputs(*line, out_file) < 0)
675 0 : pg_fatal("could not write file \"%s\": %m", path);
676 85118 : free(*line);
677 : }
678 328 : if (fclose(out_file))
679 0 : pg_fatal("could not close file \"%s\": %m", path);
680 328 : free(lines);
681 328 : }
682 :
683 : /*
684 : * Open a subcommand with suitable error messaging
685 : */
686 : static FILE *
687 162 : popen_check(const char *command, const char *mode)
688 : {
689 : FILE *cmdfd;
690 :
691 162 : fflush(NULL);
692 162 : errno = 0;
693 162 : cmdfd = popen(command, mode);
694 162 : if (cmdfd == NULL)
695 0 : pg_log_error("could not execute command \"%s\": %m", command);
696 162 : return cmdfd;
697 : }
698 :
699 : /*
700 : * clean up any files we created on failure
701 : * if we created the data directory remove it too
702 : */
703 : static void
704 106 : cleanup_directories_atexit(void)
705 : {
706 106 : if (success)
707 76 : return;
708 :
709 30 : if (!noclean)
710 : {
711 30 : if (made_new_pgdata)
712 : {
713 10 : pg_log_info("removing data directory \"%s\"", pg_data);
714 10 : if (!rmtree(pg_data, true))
715 0 : pg_log_error("failed to remove data directory");
716 : }
717 20 : else if (found_existing_pgdata)
718 : {
719 0 : pg_log_info("removing contents of data directory \"%s\"",
720 : pg_data);
721 0 : if (!rmtree(pg_data, false))
722 0 : pg_log_error("failed to remove contents of data directory");
723 : }
724 :
725 30 : if (made_new_xlogdir)
726 : {
727 0 : pg_log_info("removing WAL directory \"%s\"", xlog_dir);
728 0 : if (!rmtree(xlog_dir, true))
729 0 : pg_log_error("failed to remove WAL directory");
730 : }
731 30 : else if (found_existing_xlogdir)
732 : {
733 0 : pg_log_info("removing contents of WAL directory \"%s\"", xlog_dir);
734 0 : if (!rmtree(xlog_dir, false))
735 0 : pg_log_error("failed to remove contents of WAL directory");
736 : }
737 : /* otherwise died during startup, do nothing! */
738 : }
739 : else
740 : {
741 0 : if (made_new_pgdata || found_existing_pgdata)
742 0 : pg_log_info("data directory \"%s\" not removed at user's request",
743 : pg_data);
744 :
745 0 : if (made_new_xlogdir || found_existing_xlogdir)
746 0 : pg_log_info("WAL directory \"%s\" not removed at user's request",
747 : xlog_dir);
748 : }
749 : }
750 :
751 : /*
752 : * find the current user
753 : *
754 : * on unix make sure it isn't root
755 : */
756 : static char *
757 100 : get_id(void)
758 : {
759 : const char *username;
760 :
761 : #ifndef WIN32
762 100 : if (geteuid() == 0) /* 0 is root's uid */
763 : {
764 0 : pg_log_error("cannot be run as root");
765 0 : pg_log_error_hint("Please log in (using, e.g., \"su\") as the (unprivileged) user that will own the server process.");
766 0 : exit(1);
767 : }
768 : #endif
769 :
770 100 : username = get_user_name_or_exit(progname);
771 :
772 100 : return pg_strdup(username);
773 : }
774 :
775 : static char *
776 82 : encodingid_to_string(int enc)
777 : {
778 : char result[20];
779 :
780 82 : sprintf(result, "%d", enc);
781 82 : return pg_strdup(result);
782 : }
783 :
784 : /*
785 : * get the encoding id for a given encoding name
786 : */
787 : static int
788 32 : get_encoding_id(const char *encoding_name)
789 : {
790 : int enc;
791 :
792 32 : if (encoding_name && *encoding_name)
793 : {
794 32 : if ((enc = pg_valid_server_encoding(encoding_name)) >= 0)
795 32 : return enc;
796 : }
797 0 : pg_fatal("\"%s\" is not a valid server encoding name",
798 : encoding_name ? encoding_name : "(null)");
799 : }
800 :
801 : /*
802 : * Support for determining the best default text search configuration.
803 : * We key this off the first part of LC_CTYPE (ie, the language name).
804 : */
805 : struct tsearch_config_match
806 : {
807 : const char *tsconfname;
808 : const char *langname;
809 : };
810 :
811 : static const struct tsearch_config_match tsearch_config_languages[] =
812 : {
813 : {"arabic", "ar"},
814 : {"arabic", "Arabic"},
815 : {"armenian", "hy"},
816 : {"armenian", "Armenian"},
817 : {"basque", "eu"},
818 : {"basque", "Basque"},
819 : {"catalan", "ca"},
820 : {"catalan", "Catalan"},
821 : {"danish", "da"},
822 : {"danish", "Danish"},
823 : {"dutch", "nl"},
824 : {"dutch", "Dutch"},
825 : {"english", "C"},
826 : {"english", "POSIX"},
827 : {"english", "en"},
828 : {"english", "English"},
829 : {"finnish", "fi"},
830 : {"finnish", "Finnish"},
831 : {"french", "fr"},
832 : {"french", "French"},
833 : {"german", "de"},
834 : {"german", "German"},
835 : {"greek", "el"},
836 : {"greek", "Greek"},
837 : {"hindi", "hi"},
838 : {"hindi", "Hindi"},
839 : {"hungarian", "hu"},
840 : {"hungarian", "Hungarian"},
841 : {"indonesian", "id"},
842 : {"indonesian", "Indonesian"},
843 : {"irish", "ga"},
844 : {"irish", "Irish"},
845 : {"italian", "it"},
846 : {"italian", "Italian"},
847 : {"lithuanian", "lt"},
848 : {"lithuanian", "Lithuanian"},
849 : {"nepali", "ne"},
850 : {"nepali", "Nepali"},
851 : {"norwegian", "no"},
852 : {"norwegian", "Norwegian"},
853 : {"portuguese", "pt"},
854 : {"portuguese", "Portuguese"},
855 : {"romanian", "ro"},
856 : {"russian", "ru"},
857 : {"russian", "Russian"},
858 : {"serbian", "sr"},
859 : {"serbian", "Serbian"},
860 : {"spanish", "es"},
861 : {"spanish", "Spanish"},
862 : {"swedish", "sv"},
863 : {"swedish", "Swedish"},
864 : {"tamil", "ta"},
865 : {"tamil", "Tamil"},
866 : {"turkish", "tr"},
867 : {"turkish", "Turkish"},
868 : {"yiddish", "yi"},
869 : {"yiddish", "Yiddish"},
870 : {NULL, NULL} /* end marker */
871 : };
872 :
873 : /*
874 : * Look for a text search configuration matching lc_ctype, and return its
875 : * name; return NULL if no match.
876 : */
877 : static const char *
878 88 : find_matching_ts_config(const char *lc_type)
879 : {
880 : int i;
881 : char *langname,
882 : *ptr;
883 :
884 : /*
885 : * Convert lc_ctype to a language name by stripping everything after an
886 : * underscore (usual case) or a hyphen (Windows "locale name"; see
887 : * comments at IsoLocaleName()).
888 : *
889 : * XXX Should ' ' be a stop character? This would select "norwegian" for
890 : * the Windows locale "Norwegian (Nynorsk)_Norway.1252". If we do so, we
891 : * should also accept the "nn" and "nb" Unix locales.
892 : *
893 : * Just for paranoia, we also stop at '.' or '@'.
894 : */
895 88 : if (lc_type == NULL)
896 0 : langname = pg_strdup("");
897 : else
898 : {
899 88 : ptr = langname = pg_strdup(lc_type);
900 226 : while (*ptr &&
901 188 : *ptr != '_' && *ptr != '-' && *ptr != '.' && *ptr != '@')
902 138 : ptr++;
903 88 : *ptr = '\0';
904 : }
905 :
906 1244 : for (i = 0; tsearch_config_languages[i].tsconfname; i++)
907 : {
908 1244 : if (pg_strcasecmp(tsearch_config_languages[i].langname, langname) == 0)
909 : {
910 88 : free(langname);
911 88 : return tsearch_config_languages[i].tsconfname;
912 : }
913 : }
914 :
915 0 : free(langname);
916 0 : return NULL;
917 : }
918 :
919 :
920 : /*
921 : * set name of given input file variable under data directory
922 : */
923 : static void
924 980 : set_input(char **dest, const char *filename)
925 : {
926 980 : *dest = psprintf("%s/%s", share_path, filename);
927 980 : }
928 :
929 : /*
930 : * check that given input file exists
931 : */
932 : static void
933 980 : check_input(char *path)
934 : {
935 : struct stat statbuf;
936 :
937 980 : if (stat(path, &statbuf) != 0)
938 : {
939 0 : if (errno == ENOENT)
940 : {
941 0 : pg_log_error("file \"%s\" does not exist", path);
942 0 : pg_log_error_hint("This might mean you have a corrupted installation or identified the wrong directory with the invocation option -L.");
943 : }
944 : else
945 : {
946 0 : pg_log_error("could not access file \"%s\": %m", path);
947 0 : pg_log_error_hint("This might mean you have a corrupted installation or identified the wrong directory with the invocation option -L.");
948 : }
949 0 : exit(1);
950 : }
951 980 : if (!S_ISREG(statbuf.st_mode))
952 : {
953 0 : pg_log_error("file \"%s\" is not a regular file", path);
954 0 : pg_log_error_hint("This might mean you have a corrupted installation or identified the wrong directory with the invocation option -L.");
955 0 : exit(1);
956 : }
957 980 : }
958 :
959 : /*
960 : * write out the PG_VERSION file in the data dir, or its subdirectory
961 : * if extrapath is not NULL
962 : */
963 : static void
964 162 : write_version_file(const char *extrapath)
965 : {
966 : FILE *version_file;
967 : char *path;
968 :
969 162 : if (extrapath == NULL)
970 82 : path = psprintf("%s/PG_VERSION", pg_data);
971 : else
972 80 : path = psprintf("%s/%s/PG_VERSION", pg_data, extrapath);
973 :
974 162 : if ((version_file = fopen(path, PG_BINARY_W)) == NULL)
975 0 : pg_fatal("could not open file \"%s\" for writing: %m", path);
976 324 : if (fprintf(version_file, "%s\n", PG_MAJORVERSION) < 0 ||
977 162 : fclose(version_file))
978 0 : pg_fatal("could not write file \"%s\": %m", path);
979 162 : free(path);
980 162 : }
981 :
982 : /*
983 : * set up an empty config file so we can check config settings by launching
984 : * a test backend
985 : */
986 : static void
987 82 : set_null_conf(void)
988 : {
989 : FILE *conf_file;
990 : char *path;
991 :
992 82 : path = psprintf("%s/postgresql.conf", pg_data);
993 82 : conf_file = fopen(path, PG_BINARY_W);
994 82 : if (conf_file == NULL)
995 0 : pg_fatal("could not open file \"%s\" for writing: %m", path);
996 82 : if (fclose(conf_file))
997 0 : pg_fatal("could not write file \"%s\": %m", path);
998 82 : free(path);
999 82 : }
1000 :
1001 : /*
1002 : * Determine which dynamic shared memory implementation should be used on
1003 : * this platform. POSIX shared memory is preferable because the default
1004 : * allocation limits are much higher than the limits for System V on most
1005 : * systems that support both, but the fact that a platform has shm_open
1006 : * doesn't guarantee that that call will succeed when attempted. So, we
1007 : * attempt to reproduce what the postmaster will do when allocating a POSIX
1008 : * segment in dsm_impl.c; if it doesn't work, we assume it won't work for
1009 : * the postmaster either, and configure the cluster for System V shared
1010 : * memory instead.
1011 : *
1012 : * We avoid choosing Solaris's implementation of shm_open() by default. It
1013 : * can sleep and fail spuriously under contention.
1014 : */
1015 : static const char *
1016 82 : choose_dsm_implementation(void)
1017 : {
1018 : #if defined(HAVE_SHM_OPEN) && !defined(__sun__)
1019 82 : int ntries = 10;
1020 : pg_prng_state prng_state;
1021 :
1022 : /* Initialize prng; this function is its only user in this program. */
1023 82 : pg_prng_seed(&prng_state, (uint64) (getpid() ^ time(NULL)));
1024 :
1025 82 : while (ntries > 0)
1026 : {
1027 : uint32 handle;
1028 : char name[64];
1029 : int fd;
1030 :
1031 82 : handle = pg_prng_uint32(&prng_state);
1032 82 : snprintf(name, 64, "/PostgreSQL.%u", handle);
1033 82 : if ((fd = shm_open(name, O_CREAT | O_RDWR | O_EXCL, 0600)) != -1)
1034 : {
1035 82 : close(fd);
1036 82 : shm_unlink(name);
1037 82 : return "posix";
1038 : }
1039 0 : if (errno != EEXIST)
1040 0 : break;
1041 0 : --ntries;
1042 : }
1043 : #endif
1044 :
1045 : #ifdef WIN32
1046 : return "windows";
1047 : #else
1048 0 : return "sysv";
1049 : #endif
1050 : }
1051 :
1052 : /*
1053 : * Determine platform-specific config settings
1054 : *
1055 : * Use reasonable values if kernel will let us, else scale back.
1056 : */
1057 : static void
1058 82 : test_config_settings(void)
1059 : {
1060 : /*
1061 : * This macro defines the minimum shared_buffers we want for a given
1062 : * max_connections value. The arrays show the settings to try.
1063 : */
1064 : #define MIN_BUFS_FOR_CONNS(nconns) ((nconns) * 10)
1065 :
1066 : static const int trial_conns[] = {
1067 : 100, 50, 40, 30, 20
1068 : };
1069 : static const int trial_bufs[] = {
1070 : 16384, 8192, 4096, 3584, 3072, 2560, 2048, 1536,
1071 : 1000, 900, 800, 700, 600, 500,
1072 : 400, 300, 200, 100, 50
1073 : };
1074 :
1075 82 : const int connslen = sizeof(trial_conns) / sizeof(int);
1076 82 : const int bufslen = sizeof(trial_bufs) / sizeof(int);
1077 : int i,
1078 : test_conns,
1079 : test_buffs,
1080 82 : ok_buffers = 0;
1081 :
1082 : /*
1083 : * Need to determine working DSM implementation first so that subsequent
1084 : * tests don't fail because DSM setting doesn't work.
1085 : */
1086 82 : printf(_("selecting dynamic shared memory implementation ... "));
1087 82 : fflush(stdout);
1088 82 : dynamic_shared_memory_type = choose_dsm_implementation();
1089 82 : printf("%s\n", dynamic_shared_memory_type);
1090 :
1091 : /*
1092 : * Probe for max_connections before shared_buffers, since it is subject to
1093 : * more constraints than shared_buffers.
1094 : */
1095 82 : printf(_("selecting default max_connections ... "));
1096 82 : fflush(stdout);
1097 :
1098 92 : for (i = 0; i < connslen; i++)
1099 : {
1100 90 : test_conns = trial_conns[i];
1101 90 : test_buffs = MIN_BUFS_FOR_CONNS(test_conns);
1102 :
1103 90 : if (test_specific_config_settings(test_conns, test_buffs))
1104 : {
1105 80 : ok_buffers = test_buffs;
1106 80 : break;
1107 : }
1108 : }
1109 82 : if (i >= connslen)
1110 2 : i = connslen - 1;
1111 82 : n_connections = trial_conns[i];
1112 :
1113 82 : printf("%d\n", n_connections);
1114 :
1115 82 : printf(_("selecting default shared_buffers ... "));
1116 82 : fflush(stdout);
1117 :
1118 120 : for (i = 0; i < bufslen; i++)
1119 : {
1120 : /* Use same amount of memory, independent of BLCKSZ */
1121 118 : test_buffs = (trial_bufs[i] * 8192) / BLCKSZ;
1122 118 : if (test_buffs <= ok_buffers)
1123 : {
1124 0 : test_buffs = ok_buffers;
1125 0 : break;
1126 : }
1127 :
1128 118 : if (test_specific_config_settings(n_connections, test_buffs))
1129 80 : break;
1130 : }
1131 82 : n_buffers = test_buffs;
1132 :
1133 82 : if ((n_buffers * (BLCKSZ / 1024)) % 1024 == 0)
1134 80 : printf("%dMB\n", (n_buffers * (BLCKSZ / 1024)) / 1024);
1135 : else
1136 2 : printf("%dkB\n", n_buffers * (BLCKSZ / 1024));
1137 :
1138 82 : printf(_("selecting default time zone ... "));
1139 82 : fflush(stdout);
1140 82 : default_timezone = select_default_timezone(share_path);
1141 82 : printf("%s\n", default_timezone ? default_timezone : "GMT");
1142 82 : }
1143 :
1144 : /*
1145 : * Test a specific combination of configuration settings.
1146 : */
1147 : static bool
1148 208 : test_specific_config_settings(int test_conns, int test_buffs)
1149 : {
1150 : PQExpBufferData cmd;
1151 : _stringlist *gnames,
1152 : *gvalues;
1153 : int status;
1154 :
1155 208 : initPQExpBuffer(&cmd);
1156 :
1157 : /* Set up the test postmaster invocation */
1158 208 : printfPQExpBuffer(&cmd,
1159 : "\"%s\" --check %s %s "
1160 : "-c max_connections=%d "
1161 : "-c shared_buffers=%d "
1162 : "-c dynamic_shared_memory_type=%s",
1163 : backend_exec, boot_options, extra_options,
1164 : test_conns, test_buffs,
1165 : dynamic_shared_memory_type);
1166 :
1167 : /* Add any user-given setting overrides */
1168 272 : for (gnames = extra_guc_names, gvalues = extra_guc_values;
1169 : gnames != NULL; /* assume lists have the same length */
1170 64 : gnames = gnames->next, gvalues = gvalues->next)
1171 : {
1172 64 : appendPQExpBuffer(&cmd, " -c %s=", gnames->str);
1173 64 : appendShellString(&cmd, gvalues->str);
1174 : }
1175 :
1176 208 : appendPQExpBuffer(&cmd,
1177 : " < \"%s\" > \"%s\" 2>&1",
1178 : DEVNULL, DEVNULL);
1179 :
1180 208 : fflush(NULL);
1181 208 : status = system(cmd.data);
1182 :
1183 208 : termPQExpBuffer(&cmd);
1184 :
1185 208 : return (status == 0);
1186 : }
1187 :
1188 : /*
1189 : * Calculate the default wal_size with a "pretty" unit.
1190 : */
1191 : static char *
1192 164 : pretty_wal_size(int segment_count)
1193 : {
1194 164 : int sz = wal_segment_size_mb * segment_count;
1195 164 : char *result = pg_malloc(14);
1196 :
1197 164 : if ((sz % 1024) == 0)
1198 72 : snprintf(result, 14, "%dGB", sz / 1024);
1199 : else
1200 92 : snprintf(result, 14, "%dMB", sz);
1201 :
1202 164 : return result;
1203 : }
1204 :
1205 : /*
1206 : * set up all the config files
1207 : */
1208 : static void
1209 82 : setup_config(void)
1210 : {
1211 : char **conflines;
1212 : char repltok[MAXPGPATH];
1213 : char path[MAXPGPATH];
1214 : _stringlist *gnames,
1215 : *gvalues;
1216 :
1217 82 : fputs(_("creating configuration files ... "), stdout);
1218 82 : fflush(stdout);
1219 :
1220 : /* postgresql.conf */
1221 :
1222 82 : conflines = readfile(conf_file);
1223 :
1224 82 : snprintf(repltok, sizeof(repltok), "%d", n_connections);
1225 82 : conflines = replace_guc_value(conflines, "max_connections",
1226 : repltok, false);
1227 :
1228 82 : if ((n_buffers * (BLCKSZ / 1024)) % 1024 == 0)
1229 80 : snprintf(repltok, sizeof(repltok), "%dMB",
1230 80 : (n_buffers * (BLCKSZ / 1024)) / 1024);
1231 : else
1232 2 : snprintf(repltok, sizeof(repltok), "%dkB",
1233 : n_buffers * (BLCKSZ / 1024));
1234 82 : conflines = replace_guc_value(conflines, "shared_buffers",
1235 : repltok, false);
1236 :
1237 82 : conflines = replace_guc_value(conflines, "lc_messages",
1238 : lc_messages, false);
1239 :
1240 82 : conflines = replace_guc_value(conflines, "lc_monetary",
1241 : lc_monetary, false);
1242 :
1243 82 : conflines = replace_guc_value(conflines, "lc_numeric",
1244 : lc_numeric, false);
1245 :
1246 82 : conflines = replace_guc_value(conflines, "lc_time",
1247 : lc_time, false);
1248 :
1249 82 : switch (locale_date_order(lc_time))
1250 : {
1251 0 : case DATEORDER_YMD:
1252 0 : strcpy(repltok, "iso, ymd");
1253 0 : break;
1254 0 : case DATEORDER_DMY:
1255 0 : strcpy(repltok, "iso, dmy");
1256 0 : break;
1257 82 : case DATEORDER_MDY:
1258 : default:
1259 82 : strcpy(repltok, "iso, mdy");
1260 82 : break;
1261 : }
1262 82 : conflines = replace_guc_value(conflines, "datestyle",
1263 : repltok, false);
1264 :
1265 82 : snprintf(repltok, sizeof(repltok), "pg_catalog.%s",
1266 : default_text_search_config);
1267 82 : conflines = replace_guc_value(conflines, "default_text_search_config",
1268 : repltok, false);
1269 :
1270 82 : if (default_timezone)
1271 : {
1272 82 : conflines = replace_guc_value(conflines, "timezone",
1273 : default_timezone, false);
1274 82 : conflines = replace_guc_value(conflines, "log_timezone",
1275 : default_timezone, false);
1276 : }
1277 :
1278 82 : conflines = replace_guc_value(conflines, "dynamic_shared_memory_type",
1279 : dynamic_shared_memory_type, false);
1280 :
1281 : /* Caution: these depend on wal_segment_size_mb, they're not constants */
1282 82 : conflines = replace_guc_value(conflines, "min_wal_size",
1283 82 : pretty_wal_size(DEFAULT_MIN_WAL_SEGS), false);
1284 :
1285 82 : conflines = replace_guc_value(conflines, "max_wal_size",
1286 82 : pretty_wal_size(DEFAULT_MAX_WAL_SEGS), false);
1287 :
1288 : /*
1289 : * Fix up various entries to match the true compile-time defaults. Since
1290 : * these are indeed defaults, keep the postgresql.conf lines commented.
1291 : */
1292 82 : conflines = replace_guc_value(conflines, "unix_socket_directories",
1293 : DEFAULT_PGSOCKET_DIR, true);
1294 :
1295 82 : conflines = replace_guc_value(conflines, "port",
1296 : DEF_PGPORT_STR, true);
1297 :
1298 : #if DEFAULT_BACKEND_FLUSH_AFTER > 0
1299 : snprintf(repltok, sizeof(repltok), "%dkB",
1300 : DEFAULT_BACKEND_FLUSH_AFTER * (BLCKSZ / 1024));
1301 : conflines = replace_guc_value(conflines, "backend_flush_after",
1302 : repltok, true);
1303 : #endif
1304 :
1305 : #if DEFAULT_BGWRITER_FLUSH_AFTER > 0
1306 82 : snprintf(repltok, sizeof(repltok), "%dkB",
1307 : DEFAULT_BGWRITER_FLUSH_AFTER * (BLCKSZ / 1024));
1308 82 : conflines = replace_guc_value(conflines, "bgwriter_flush_after",
1309 : repltok, true);
1310 : #endif
1311 :
1312 : #if DEFAULT_CHECKPOINT_FLUSH_AFTER > 0
1313 82 : snprintf(repltok, sizeof(repltok), "%dkB",
1314 : DEFAULT_CHECKPOINT_FLUSH_AFTER * (BLCKSZ / 1024));
1315 82 : conflines = replace_guc_value(conflines, "checkpoint_flush_after",
1316 : repltok, true);
1317 : #endif
1318 :
1319 : #ifndef USE_PREFETCH
1320 : conflines = replace_guc_value(conflines, "effective_io_concurrency",
1321 : "0", true);
1322 : #endif
1323 :
1324 : #ifdef WIN32
1325 : conflines = replace_guc_value(conflines, "update_process_title",
1326 : "off", true);
1327 : #endif
1328 :
1329 : /*
1330 : * Change password_encryption setting to md5 if md5 was chosen as an
1331 : * authentication method, unless scram-sha-256 was also chosen.
1332 : */
1333 82 : if ((strcmp(authmethodlocal, "md5") == 0 &&
1334 0 : strcmp(authmethodhost, "scram-sha-256") != 0) ||
1335 82 : (strcmp(authmethodhost, "md5") == 0 &&
1336 0 : strcmp(authmethodlocal, "scram-sha-256") != 0))
1337 : {
1338 0 : conflines = replace_guc_value(conflines, "password_encryption",
1339 : "md5", false);
1340 : }
1341 :
1342 : /*
1343 : * If group access has been enabled for the cluster then it makes sense to
1344 : * ensure that the log files also allow group access. Otherwise a backup
1345 : * from a user in the group would fail if the log files were not
1346 : * relocated.
1347 : */
1348 82 : if (pg_dir_create_mode == PG_DIR_MODE_GROUP)
1349 : {
1350 10 : conflines = replace_guc_value(conflines, "log_file_mode",
1351 : "0640", false);
1352 : }
1353 :
1354 : /*
1355 : * Now replace anything that's overridden via -c switches.
1356 : */
1357 92 : for (gnames = extra_guc_names, gvalues = extra_guc_values;
1358 : gnames != NULL; /* assume lists have the same length */
1359 10 : gnames = gnames->next, gvalues = gvalues->next)
1360 : {
1361 10 : conflines = replace_guc_value(conflines, gnames->str,
1362 10 : gvalues->str, false);
1363 : }
1364 :
1365 : /* ... and write out the finished postgresql.conf file */
1366 82 : snprintf(path, sizeof(path), "%s/postgresql.conf", pg_data);
1367 :
1368 82 : writefile(path, conflines);
1369 82 : if (chmod(path, pg_file_create_mode) != 0)
1370 0 : pg_fatal("could not change permissions of \"%s\": %m", path);
1371 :
1372 :
1373 : /* postgresql.auto.conf */
1374 :
1375 82 : conflines = pg_malloc_array(char *, 3);
1376 82 : conflines[0] = pg_strdup("# Do not edit this file manually!\n");
1377 82 : conflines[1] = pg_strdup("# It will be overwritten by the ALTER SYSTEM command.\n");
1378 82 : conflines[2] = NULL;
1379 :
1380 82 : sprintf(path, "%s/postgresql.auto.conf", pg_data);
1381 :
1382 82 : writefile(path, conflines);
1383 82 : if (chmod(path, pg_file_create_mode) != 0)
1384 0 : pg_fatal("could not change permissions of \"%s\": %m", path);
1385 :
1386 :
1387 : /* pg_hba.conf */
1388 :
1389 82 : conflines = readfile(hba_file);
1390 :
1391 82 : conflines = replace_token(conflines, "@remove-line-for-nolocal@", "");
1392 :
1393 :
1394 : /*
1395 : * Probe to see if there is really any platform support for IPv6, and
1396 : * comment out the relevant pg_hba line if not. This avoids runtime
1397 : * warnings if getaddrinfo doesn't actually cope with IPv6. Particularly
1398 : * useful on Windows, where executables built on a machine with IPv6 may
1399 : * have to run on a machine without.
1400 : */
1401 : {
1402 : struct addrinfo *gai_result;
1403 : struct addrinfo hints;
1404 82 : int err = 0;
1405 :
1406 : #ifdef WIN32
1407 : /* need to call WSAStartup before calling getaddrinfo */
1408 : WSADATA wsaData;
1409 :
1410 : err = WSAStartup(MAKEWORD(2, 2), &wsaData);
1411 : #endif
1412 :
1413 : /* for best results, this code should match parse_hba_line() */
1414 82 : hints.ai_flags = AI_NUMERICHOST;
1415 82 : hints.ai_family = AF_UNSPEC;
1416 82 : hints.ai_socktype = 0;
1417 82 : hints.ai_protocol = 0;
1418 82 : hints.ai_addrlen = 0;
1419 82 : hints.ai_canonname = NULL;
1420 82 : hints.ai_addr = NULL;
1421 82 : hints.ai_next = NULL;
1422 :
1423 164 : if (err != 0 ||
1424 82 : getaddrinfo("::1", NULL, &hints, &gai_result) != 0)
1425 : {
1426 0 : conflines = replace_token(conflines,
1427 : "host all all ::1",
1428 : "#host all all ::1");
1429 0 : conflines = replace_token(conflines,
1430 : "host replication all ::1",
1431 : "#host replication all ::1");
1432 : }
1433 : }
1434 :
1435 : /* Replace default authentication methods */
1436 82 : conflines = replace_token(conflines,
1437 : "@authmethodhost@",
1438 : authmethodhost);
1439 82 : conflines = replace_token(conflines,
1440 : "@authmethodlocal@",
1441 : authmethodlocal);
1442 :
1443 82 : conflines = replace_token(conflines,
1444 : "@authcomment@",
1445 82 : (strcmp(authmethodlocal, "trust") == 0 || strcmp(authmethodhost, "trust") == 0) ? AUTHTRUST_WARNING : "");
1446 :
1447 82 : snprintf(path, sizeof(path), "%s/pg_hba.conf", pg_data);
1448 :
1449 82 : writefile(path, conflines);
1450 82 : if (chmod(path, pg_file_create_mode) != 0)
1451 0 : pg_fatal("could not change permissions of \"%s\": %m", path);
1452 :
1453 :
1454 : /* pg_ident.conf */
1455 :
1456 82 : conflines = readfile(ident_file);
1457 :
1458 82 : snprintf(path, sizeof(path), "%s/pg_ident.conf", pg_data);
1459 :
1460 82 : writefile(path, conflines);
1461 82 : if (chmod(path, pg_file_create_mode) != 0)
1462 0 : pg_fatal("could not change permissions of \"%s\": %m", path);
1463 :
1464 82 : check_ok();
1465 82 : }
1466 :
1467 :
1468 : /*
1469 : * run the BKI script in bootstrap mode to create template1
1470 : */
1471 : static void
1472 82 : bootstrap_template1(void)
1473 : {
1474 : PG_CMD_DECL;
1475 : PQExpBufferData cmd;
1476 : char **line;
1477 : char **bki_lines;
1478 : char headerline[MAXPGPATH];
1479 : char buf[64];
1480 :
1481 82 : printf(_("running bootstrap script ... "));
1482 82 : fflush(stdout);
1483 :
1484 82 : bki_lines = readfile(bki_file);
1485 :
1486 : /* Check that bki file appears to be of the right version */
1487 :
1488 82 : snprintf(headerline, sizeof(headerline), "# PostgreSQL %s\n",
1489 : PG_MAJORVERSION);
1490 :
1491 82 : if (strcmp(headerline, *bki_lines) != 0)
1492 : {
1493 0 : pg_log_error("input file \"%s\" does not belong to PostgreSQL %s",
1494 : bki_file, PG_VERSION);
1495 0 : pg_log_error_hint("Specify the correct path using the option -L.");
1496 0 : exit(1);
1497 : }
1498 :
1499 : /* Substitute for various symbols used in the BKI file */
1500 :
1501 82 : sprintf(buf, "%d", NAMEDATALEN);
1502 82 : bki_lines = replace_token(bki_lines, "NAMEDATALEN", buf);
1503 :
1504 82 : sprintf(buf, "%d", (int) sizeof(Pointer));
1505 82 : bki_lines = replace_token(bki_lines, "SIZEOF_POINTER", buf);
1506 :
1507 82 : bki_lines = replace_token(bki_lines, "ALIGNOF_POINTER",
1508 : (sizeof(Pointer) == 4) ? "i" : "d");
1509 :
1510 82 : bki_lines = replace_token(bki_lines, "FLOAT8PASSBYVAL",
1511 : FLOAT8PASSBYVAL ? "true" : "false");
1512 :
1513 82 : bki_lines = replace_token(bki_lines, "POSTGRES",
1514 82 : escape_quotes_bki(username));
1515 :
1516 82 : bki_lines = replace_token(bki_lines, "ENCODING",
1517 82 : encodingid_to_string(encodingid));
1518 :
1519 82 : bki_lines = replace_token(bki_lines, "LC_COLLATE",
1520 82 : escape_quotes_bki(lc_collate));
1521 :
1522 82 : bki_lines = replace_token(bki_lines, "LC_CTYPE",
1523 82 : escape_quotes_bki(lc_ctype));
1524 :
1525 82 : bki_lines = replace_token(bki_lines, "DATLOCALE",
1526 82 : datlocale ? escape_quotes_bki(datlocale) : "_null_");
1527 :
1528 82 : bki_lines = replace_token(bki_lines, "ICU_RULES",
1529 82 : icu_rules ? escape_quotes_bki(icu_rules) : "_null_");
1530 :
1531 82 : sprintf(buf, "%c", locale_provider);
1532 82 : bki_lines = replace_token(bki_lines, "LOCALE_PROVIDER", buf);
1533 :
1534 : /* Also ensure backend isn't confused by this environment var: */
1535 82 : unsetenv("PGCLIENTENCODING");
1536 :
1537 82 : initPQExpBuffer(&cmd);
1538 :
1539 82 : printfPQExpBuffer(&cmd, "\"%s\" --boot %s %s", backend_exec, boot_options, extra_options);
1540 82 : appendPQExpBuffer(&cmd, " -X %d", wal_segment_size_mb * (1024 * 1024));
1541 82 : if (data_checksums)
1542 2 : appendPQExpBuffer(&cmd, " -k");
1543 82 : if (debug)
1544 0 : appendPQExpBuffer(&cmd, " -d 5");
1545 :
1546 :
1547 82 : PG_CMD_OPEN(cmd.data);
1548 :
1549 957842 : for (line = bki_lines; *line != NULL; line++)
1550 : {
1551 957760 : PG_CMD_PUTS(*line);
1552 957760 : free(*line);
1553 : }
1554 :
1555 82 : PG_CMD_CLOSE();
1556 :
1557 80 : termPQExpBuffer(&cmd);
1558 80 : free(bki_lines);
1559 :
1560 80 : check_ok();
1561 80 : }
1562 :
1563 : /*
1564 : * set up the shadow password table
1565 : */
1566 : static void
1567 80 : setup_auth(FILE *cmdfd)
1568 : {
1569 : /*
1570 : * The authid table shouldn't be readable except through views, to ensure
1571 : * passwords are not publicly visible.
1572 : */
1573 80 : PG_CMD_PUTS("REVOKE ALL ON pg_authid FROM public;\n\n");
1574 :
1575 80 : if (superuser_password)
1576 0 : PG_CMD_PRINTF("ALTER USER \"%s\" WITH PASSWORD E'%s';\n\n",
1577 : username, escape_quotes(superuser_password));
1578 80 : }
1579 :
1580 : /*
1581 : * get the superuser password if required
1582 : */
1583 : static void
1584 0 : get_su_pwd(void)
1585 : {
1586 : char *pwd1;
1587 :
1588 0 : if (pwprompt)
1589 : {
1590 : /*
1591 : * Read password from terminal
1592 : */
1593 : char *pwd2;
1594 :
1595 0 : printf("\n");
1596 0 : fflush(stdout);
1597 0 : pwd1 = simple_prompt("Enter new superuser password: ", false);
1598 0 : pwd2 = simple_prompt("Enter it again: ", false);
1599 0 : if (strcmp(pwd1, pwd2) != 0)
1600 : {
1601 0 : fprintf(stderr, _("Passwords didn't match.\n"));
1602 0 : exit(1);
1603 : }
1604 0 : free(pwd2);
1605 : }
1606 : else
1607 : {
1608 : /*
1609 : * Read password from file
1610 : *
1611 : * Ideally this should insist that the file not be world-readable.
1612 : * However, this option is mainly intended for use on Windows where
1613 : * file permissions may not exist at all, so we'll skip the paranoia
1614 : * for now.
1615 : */
1616 0 : FILE *pwf = fopen(pwfilename, "r");
1617 :
1618 0 : if (!pwf)
1619 0 : pg_fatal("could not open file \"%s\" for reading: %m",
1620 : pwfilename);
1621 0 : pwd1 = pg_get_line(pwf, NULL);
1622 0 : if (!pwd1)
1623 : {
1624 0 : if (ferror(pwf))
1625 0 : pg_fatal("could not read password from file \"%s\": %m",
1626 : pwfilename);
1627 : else
1628 0 : pg_fatal("password file \"%s\" is empty",
1629 : pwfilename);
1630 : }
1631 0 : fclose(pwf);
1632 :
1633 0 : (void) pg_strip_crlf(pwd1);
1634 : }
1635 :
1636 0 : superuser_password = pwd1;
1637 0 : }
1638 :
1639 : /*
1640 : * set up pg_depend
1641 : */
1642 : static void
1643 80 : setup_depend(FILE *cmdfd)
1644 : {
1645 : /*
1646 : * Advance the OID counter so that subsequently-created objects aren't
1647 : * pinned.
1648 : */
1649 80 : PG_CMD_PUTS("SELECT pg_stop_making_pinned_objects();\n\n");
1650 80 : }
1651 :
1652 : /*
1653 : * Run external file
1654 : */
1655 : static void
1656 400 : setup_run_file(FILE *cmdfd, const char *filename)
1657 : {
1658 : char **lines;
1659 :
1660 400 : lines = readfile(filename);
1661 :
1662 529120 : for (char **line = lines; *line != NULL; line++)
1663 : {
1664 528720 : PG_CMD_PUTS(*line);
1665 528720 : free(*line);
1666 : }
1667 :
1668 400 : PG_CMD_PUTS("\n\n");
1669 :
1670 400 : free(lines);
1671 400 : }
1672 :
1673 : /*
1674 : * fill in extra description data
1675 : */
1676 : static void
1677 80 : setup_description(FILE *cmdfd)
1678 : {
1679 : /* Create default descriptions for operator implementation functions */
1680 80 : PG_CMD_PUTS("WITH funcdescs AS ( "
1681 : "SELECT p.oid as p_oid, o.oid as o_oid, oprname "
1682 : "FROM pg_proc p JOIN pg_operator o ON oprcode = p.oid ) "
1683 : "INSERT INTO pg_description "
1684 : " SELECT p_oid, 'pg_proc'::regclass, 0, "
1685 : " 'implementation of ' || oprname || ' operator' "
1686 : " FROM funcdescs "
1687 : " WHERE NOT EXISTS (SELECT 1 FROM pg_description "
1688 : " WHERE objoid = p_oid AND classoid = 'pg_proc'::regclass) "
1689 : " AND NOT EXISTS (SELECT 1 FROM pg_description "
1690 : " WHERE objoid = o_oid AND classoid = 'pg_operator'::regclass"
1691 : " AND description LIKE 'deprecated%');\n\n");
1692 80 : }
1693 :
1694 : /*
1695 : * populate pg_collation
1696 : */
1697 : static void
1698 80 : setup_collation(FILE *cmdfd)
1699 : {
1700 : /*
1701 : * Set the collation version for collations defined in pg_collation.dat,
1702 : * but not the ones where we know that the collation behavior will never
1703 : * change.
1704 : */
1705 80 : PG_CMD_PUTS("UPDATE pg_collation SET collversion = pg_collation_actual_version(oid) WHERE collname = 'unicode';\n\n");
1706 :
1707 : /* Import all collations we can find in the operating system */
1708 80 : PG_CMD_PUTS("SELECT pg_import_system_collations('pg_catalog');\n\n");
1709 80 : }
1710 :
1711 : /*
1712 : * Set up privileges
1713 : *
1714 : * We mark most system catalogs as world-readable. We don't currently have
1715 : * to touch functions, languages, or databases, because their default
1716 : * permissions are OK.
1717 : *
1718 : * Some objects may require different permissions by default, so we
1719 : * make sure we don't overwrite privilege sets that have already been
1720 : * set (NOT NULL).
1721 : *
1722 : * Also populate pg_init_privs to save what the privileges are at init
1723 : * time. This is used by pg_dump to allow users to change privileges
1724 : * on catalog objects and to have those privilege changes preserved
1725 : * across dump/reload and pg_upgrade.
1726 : *
1727 : * Note that pg_init_privs is only for per-database objects and therefore
1728 : * we don't include databases or tablespaces.
1729 : */
1730 : static void
1731 80 : setup_privileges(FILE *cmdfd)
1732 : {
1733 80 : PG_CMD_PRINTF("UPDATE pg_class "
1734 : " SET relacl = (SELECT array_agg(a.acl) FROM "
1735 : " (SELECT E'=r/\"%s\"' as acl "
1736 : " UNION SELECT unnest(pg_catalog.acldefault("
1737 : " CASE WHEN relkind = " CppAsString2(RELKIND_SEQUENCE) " THEN 's' "
1738 : " ELSE 'r' END::\"char\"," CppAsString2(BOOTSTRAP_SUPERUSERID) "::oid))"
1739 : " ) as a) "
1740 : " WHERE relkind IN (" CppAsString2(RELKIND_RELATION) ", "
1741 : CppAsString2(RELKIND_VIEW) ", " CppAsString2(RELKIND_MATVIEW) ", "
1742 : CppAsString2(RELKIND_SEQUENCE) ")"
1743 : " AND relacl IS NULL;\n\n",
1744 : escape_quotes(username));
1745 80 : PG_CMD_PUTS("GRANT USAGE ON SCHEMA pg_catalog, public TO PUBLIC;\n\n");
1746 80 : PG_CMD_PUTS("REVOKE ALL ON pg_largeobject FROM PUBLIC;\n\n");
1747 80 : PG_CMD_PUTS("INSERT INTO pg_init_privs "
1748 : " (objoid, classoid, objsubid, initprivs, privtype)"
1749 : " SELECT"
1750 : " oid,"
1751 : " (SELECT oid FROM pg_class WHERE relname = 'pg_class'),"
1752 : " 0,"
1753 : " relacl,"
1754 : " 'i'"
1755 : " FROM"
1756 : " pg_class"
1757 : " WHERE"
1758 : " relacl IS NOT NULL"
1759 : " AND relkind IN (" CppAsString2(RELKIND_RELATION) ", "
1760 : CppAsString2(RELKIND_VIEW) ", " CppAsString2(RELKIND_MATVIEW) ", "
1761 : CppAsString2(RELKIND_SEQUENCE) ");\n\n");
1762 80 : PG_CMD_PUTS("INSERT INTO pg_init_privs "
1763 : " (objoid, classoid, objsubid, initprivs, privtype)"
1764 : " SELECT"
1765 : " pg_class.oid,"
1766 : " (SELECT oid FROM pg_class WHERE relname = 'pg_class'),"
1767 : " pg_attribute.attnum,"
1768 : " pg_attribute.attacl,"
1769 : " 'i'"
1770 : " FROM"
1771 : " pg_class"
1772 : " JOIN pg_attribute ON (pg_class.oid = pg_attribute.attrelid)"
1773 : " WHERE"
1774 : " pg_attribute.attacl IS NOT NULL"
1775 : " AND pg_class.relkind IN (" CppAsString2(RELKIND_RELATION) ", "
1776 : CppAsString2(RELKIND_VIEW) ", " CppAsString2(RELKIND_MATVIEW) ", "
1777 : CppAsString2(RELKIND_SEQUENCE) ");\n\n");
1778 80 : PG_CMD_PUTS("INSERT INTO pg_init_privs "
1779 : " (objoid, classoid, objsubid, initprivs, privtype)"
1780 : " SELECT"
1781 : " oid,"
1782 : " (SELECT oid FROM pg_class WHERE relname = 'pg_proc'),"
1783 : " 0,"
1784 : " proacl,"
1785 : " 'i'"
1786 : " FROM"
1787 : " pg_proc"
1788 : " WHERE"
1789 : " proacl IS NOT NULL;\n\n");
1790 80 : PG_CMD_PUTS("INSERT INTO pg_init_privs "
1791 : " (objoid, classoid, objsubid, initprivs, privtype)"
1792 : " SELECT"
1793 : " oid,"
1794 : " (SELECT oid FROM pg_class WHERE relname = 'pg_type'),"
1795 : " 0,"
1796 : " typacl,"
1797 : " 'i'"
1798 : " FROM"
1799 : " pg_type"
1800 : " WHERE"
1801 : " typacl IS NOT NULL;\n\n");
1802 80 : PG_CMD_PUTS("INSERT INTO pg_init_privs "
1803 : " (objoid, classoid, objsubid, initprivs, privtype)"
1804 : " SELECT"
1805 : " oid,"
1806 : " (SELECT oid FROM pg_class WHERE relname = 'pg_language'),"
1807 : " 0,"
1808 : " lanacl,"
1809 : " 'i'"
1810 : " FROM"
1811 : " pg_language"
1812 : " WHERE"
1813 : " lanacl IS NOT NULL;\n\n");
1814 80 : PG_CMD_PUTS("INSERT INTO pg_init_privs "
1815 : " (objoid, classoid, objsubid, initprivs, privtype)"
1816 : " SELECT"
1817 : " oid,"
1818 : " (SELECT oid FROM pg_class WHERE "
1819 : " relname = 'pg_largeobject_metadata'),"
1820 : " 0,"
1821 : " lomacl,"
1822 : " 'i'"
1823 : " FROM"
1824 : " pg_largeobject_metadata"
1825 : " WHERE"
1826 : " lomacl IS NOT NULL;\n\n");
1827 80 : PG_CMD_PUTS("INSERT INTO pg_init_privs "
1828 : " (objoid, classoid, objsubid, initprivs, privtype)"
1829 : " SELECT"
1830 : " oid,"
1831 : " (SELECT oid FROM pg_class WHERE relname = 'pg_namespace'),"
1832 : " 0,"
1833 : " nspacl,"
1834 : " 'i'"
1835 : " FROM"
1836 : " pg_namespace"
1837 : " WHERE"
1838 : " nspacl IS NOT NULL;\n\n");
1839 80 : PG_CMD_PUTS("INSERT INTO pg_init_privs "
1840 : " (objoid, classoid, objsubid, initprivs, privtype)"
1841 : " SELECT"
1842 : " oid,"
1843 : " (SELECT oid FROM pg_class WHERE "
1844 : " relname = 'pg_foreign_data_wrapper'),"
1845 : " 0,"
1846 : " fdwacl,"
1847 : " 'i'"
1848 : " FROM"
1849 : " pg_foreign_data_wrapper"
1850 : " WHERE"
1851 : " fdwacl IS NOT NULL;\n\n");
1852 80 : PG_CMD_PUTS("INSERT INTO pg_init_privs "
1853 : " (objoid, classoid, objsubid, initprivs, privtype)"
1854 : " SELECT"
1855 : " oid,"
1856 : " (SELECT oid FROM pg_class "
1857 : " WHERE relname = 'pg_foreign_server'),"
1858 : " 0,"
1859 : " srvacl,"
1860 : " 'i'"
1861 : " FROM"
1862 : " pg_foreign_server"
1863 : " WHERE"
1864 : " srvacl IS NOT NULL;\n\n");
1865 80 : }
1866 :
1867 : /*
1868 : * extract the strange version of version required for information schema
1869 : * (09.08.0007abc)
1870 : */
1871 : static void
1872 98 : set_info_version(void)
1873 : {
1874 : char *letterversion;
1875 98 : long major = 0,
1876 98 : minor = 0,
1877 98 : micro = 0;
1878 : char *endptr;
1879 98 : char *vstr = pg_strdup(PG_VERSION);
1880 : char *ptr;
1881 :
1882 98 : ptr = vstr + (strlen(vstr) - 1);
1883 588 : while (ptr != vstr && (*ptr < '0' || *ptr > '9'))
1884 490 : ptr--;
1885 98 : letterversion = ptr + 1;
1886 98 : major = strtol(vstr, &endptr, 10);
1887 98 : if (*endptr)
1888 98 : minor = strtol(endptr + 1, &endptr, 10);
1889 98 : if (*endptr)
1890 98 : micro = strtol(endptr + 1, &endptr, 10);
1891 98 : snprintf(infoversion, sizeof(infoversion), "%02ld.%02ld.%04ld%s",
1892 : major, minor, micro, letterversion);
1893 98 : }
1894 :
1895 : /*
1896 : * load info schema and populate from features file
1897 : */
1898 : static void
1899 80 : setup_schema(FILE *cmdfd)
1900 : {
1901 80 : setup_run_file(cmdfd, info_schema_file);
1902 :
1903 80 : PG_CMD_PRINTF("UPDATE information_schema.sql_implementation_info "
1904 : " SET character_value = '%s' "
1905 : " WHERE implementation_info_name = 'DBMS VERSION';\n\n",
1906 : infoversion);
1907 :
1908 80 : PG_CMD_PRINTF("COPY information_schema.sql_features "
1909 : " (feature_id, feature_name, sub_feature_id, "
1910 : " sub_feature_name, is_supported, comments) "
1911 : " FROM E'%s';\n\n",
1912 : escape_quotes(features_file));
1913 80 : }
1914 :
1915 : /*
1916 : * load PL/pgSQL server-side language
1917 : */
1918 : static void
1919 80 : load_plpgsql(FILE *cmdfd)
1920 : {
1921 80 : PG_CMD_PUTS("CREATE EXTENSION plpgsql;\n\n");
1922 80 : }
1923 :
1924 : /*
1925 : * clean everything up in template1
1926 : */
1927 : static void
1928 80 : vacuum_db(FILE *cmdfd)
1929 : {
1930 : /* Run analyze before VACUUM so the statistics are frozen. */
1931 80 : PG_CMD_PUTS("ANALYZE;\n\nVACUUM FREEZE;\n\n");
1932 80 : }
1933 :
1934 : /*
1935 : * copy template1 to template0
1936 : */
1937 : static void
1938 80 : make_template0(FILE *cmdfd)
1939 : {
1940 : /*
1941 : * pg_upgrade tries to preserve database OIDs across upgrades. It's smart
1942 : * enough to drop and recreate a conflicting database with the same name,
1943 : * but if the same OID were used for one system-created database in the
1944 : * old cluster and a different system-created database in the new cluster,
1945 : * it would fail. To avoid that, assign a fixed OID to template0 rather
1946 : * than letting the server choose one.
1947 : *
1948 : * (Note that, while the user could have dropped and recreated these
1949 : * objects in the old cluster, the problem scenario only exists if the OID
1950 : * that is in use in the old cluster is also used in the new cluster - and
1951 : * the new cluster should be the result of a fresh initdb.)
1952 : *
1953 : * We use "STRATEGY = file_copy" here because checkpoints during initdb
1954 : * are cheap. "STRATEGY = wal_log" would generate more WAL, which would be
1955 : * a little bit slower and make the new cluster a little bit bigger.
1956 : */
1957 80 : PG_CMD_PUTS("CREATE DATABASE template0 IS_TEMPLATE = true ALLOW_CONNECTIONS = false"
1958 : " OID = " CppAsString2(Template0DbOid)
1959 : " STRATEGY = file_copy;\n\n");
1960 :
1961 : /*
1962 : * template0 shouldn't have any collation-dependent objects, so unset the
1963 : * collation version. This disables collation version checks when making
1964 : * a new database from it.
1965 : */
1966 80 : PG_CMD_PUTS("UPDATE pg_database SET datcollversion = NULL WHERE datname = 'template0';\n\n");
1967 :
1968 : /*
1969 : * While we are here, do set the collation version on template1.
1970 : */
1971 80 : PG_CMD_PUTS("UPDATE pg_database SET datcollversion = pg_database_collation_actual_version(oid) WHERE datname = 'template1';\n\n");
1972 :
1973 : /*
1974 : * Explicitly revoke public create-schema and create-temp-table privileges
1975 : * in template1 and template0; else the latter would be on by default
1976 : */
1977 80 : PG_CMD_PUTS("REVOKE CREATE,TEMPORARY ON DATABASE template1 FROM public;\n\n");
1978 80 : PG_CMD_PUTS("REVOKE CREATE,TEMPORARY ON DATABASE template0 FROM public;\n\n");
1979 :
1980 80 : PG_CMD_PUTS("COMMENT ON DATABASE template0 IS 'unmodifiable empty database';\n\n");
1981 :
1982 : /*
1983 : * Finally vacuum to clean up dead rows in pg_database
1984 : */
1985 80 : PG_CMD_PUTS("VACUUM pg_database;\n\n");
1986 80 : }
1987 :
1988 : /*
1989 : * copy template1 to postgres
1990 : */
1991 : static void
1992 80 : make_postgres(FILE *cmdfd)
1993 : {
1994 : /*
1995 : * Just as we did for template0, and for the same reasons, assign a fixed
1996 : * OID to postgres and select the file_copy strategy.
1997 : */
1998 80 : PG_CMD_PUTS("CREATE DATABASE postgres OID = " CppAsString2(PostgresDbOid)
1999 : " STRATEGY = file_copy;\n\n");
2000 80 : PG_CMD_PUTS("COMMENT ON DATABASE postgres IS 'default administrative connection database';\n\n");
2001 80 : }
2002 :
2003 : /*
2004 : * signal handler in case we are interrupted.
2005 : *
2006 : * The Windows runtime docs at
2007 : * https://docs.microsoft.com/en-us/cpp/c-runtime-library/reference/signal
2008 : * specifically forbid a number of things being done from a signal handler,
2009 : * including IO, memory allocation and system calls, and only allow jmpbuf
2010 : * if you are handling SIGFPE.
2011 : *
2012 : * I avoided doing the forbidden things by setting a flag instead of calling
2013 : * exit() directly.
2014 : *
2015 : * Also note the behaviour of Windows with SIGINT, which says this:
2016 : * SIGINT is not supported for any Win32 application. When a CTRL+C interrupt
2017 : * occurs, Win32 operating systems generate a new thread to specifically
2018 : * handle that interrupt. This can cause a single-thread application, such as
2019 : * one in UNIX, to become multithreaded and cause unexpected behavior.
2020 : *
2021 : * I have no idea how to handle this. (Strange they call UNIX an application!)
2022 : * So this will need some testing on Windows.
2023 : */
2024 : static void
2025 0 : trapsig(SIGNAL_ARGS)
2026 : {
2027 : /* handle systems that reset the handler, like Windows (grr) */
2028 0 : pqsignal(postgres_signal_arg, trapsig);
2029 0 : caught_signal = true;
2030 0 : }
2031 :
2032 : /*
2033 : * call exit() if we got a signal, or else output "ok".
2034 : */
2035 : static void
2036 416 : check_ok(void)
2037 : {
2038 416 : if (caught_signal)
2039 : {
2040 0 : printf(_("caught signal\n"));
2041 0 : fflush(stdout);
2042 0 : exit(1);
2043 : }
2044 416 : else if (output_failed)
2045 : {
2046 0 : printf(_("could not write to child process: %s\n"),
2047 : strerror(output_errno));
2048 0 : fflush(stdout);
2049 0 : exit(1);
2050 : }
2051 : else
2052 : {
2053 : /* all seems well */
2054 416 : printf(_("ok\n"));
2055 416 : fflush(stdout);
2056 : }
2057 416 : }
2058 :
2059 : /* Hack to suppress a warning about %x from some versions of gcc */
2060 : static inline size_t
2061 82 : my_strftime(char *s, size_t max, const char *fmt, const struct tm *tm)
2062 : {
2063 82 : return strftime(s, max, fmt, tm);
2064 : }
2065 :
2066 : /*
2067 : * Determine likely date order from locale
2068 : */
2069 : static int
2070 82 : locale_date_order(const char *locale)
2071 : {
2072 : struct tm testtime;
2073 : char buf[128];
2074 : char *posD;
2075 : char *posM;
2076 : char *posY;
2077 : char *save;
2078 : size_t res;
2079 : int result;
2080 :
2081 82 : result = DATEORDER_MDY; /* default */
2082 :
2083 82 : save = setlocale(LC_TIME, NULL);
2084 82 : if (!save)
2085 0 : return result;
2086 82 : save = pg_strdup(save);
2087 :
2088 82 : setlocale(LC_TIME, locale);
2089 :
2090 82 : memset(&testtime, 0, sizeof(testtime));
2091 82 : testtime.tm_mday = 22;
2092 82 : testtime.tm_mon = 10; /* November, should come out as "11" */
2093 82 : testtime.tm_year = 133; /* 2033 */
2094 :
2095 82 : res = my_strftime(buf, sizeof(buf), "%x", &testtime);
2096 :
2097 82 : setlocale(LC_TIME, save);
2098 82 : free(save);
2099 :
2100 82 : if (res == 0)
2101 0 : return result;
2102 :
2103 82 : posM = strstr(buf, "11");
2104 82 : posD = strstr(buf, "22");
2105 82 : posY = strstr(buf, "33");
2106 :
2107 82 : if (!posM || !posD || !posY)
2108 0 : return result;
2109 :
2110 82 : if (posY < posM && posM < posD)
2111 0 : result = DATEORDER_YMD;
2112 82 : else if (posD < posM)
2113 0 : result = DATEORDER_DMY;
2114 : else
2115 82 : result = DATEORDER_MDY;
2116 :
2117 82 : return result;
2118 : }
2119 :
2120 : /*
2121 : * Verify that locale name is valid for the locale category.
2122 : *
2123 : * If successful, and canonname isn't NULL, a malloc'd copy of the locale's
2124 : * canonical name is stored there. This is especially useful for figuring out
2125 : * what locale name "" means (ie, the environment value). (Actually,
2126 : * it seems that on most implementations that's the only thing it's good for;
2127 : * we could wish that setlocale gave back a canonically spelled version of
2128 : * the locale name, but typically it doesn't.)
2129 : *
2130 : * this should match the backend's check_locale() function
2131 : */
2132 : static void
2133 588 : check_locale_name(int category, const char *locale, char **canonname)
2134 : {
2135 : char *save;
2136 : char *res;
2137 :
2138 588 : if (canonname)
2139 588 : *canonname = NULL; /* in case of failure */
2140 :
2141 588 : save = setlocale(category, NULL);
2142 588 : if (!save)
2143 0 : pg_fatal("setlocale() failed");
2144 :
2145 : /* save may be pointing at a modifiable scratch variable, so copy it. */
2146 588 : save = pg_strdup(save);
2147 :
2148 : /* for setlocale() call */
2149 588 : if (!locale)
2150 368 : locale = "";
2151 :
2152 : /* set the locale with setlocale, to see if it accepts it. */
2153 588 : res = setlocale(category, locale);
2154 :
2155 : /* save canonical name if requested. */
2156 588 : if (res && canonname)
2157 588 : *canonname = pg_strdup(res);
2158 :
2159 : /* restore old value. */
2160 588 : if (!setlocale(category, save))
2161 0 : pg_fatal("failed to restore old locale \"%s\"", save);
2162 588 : free(save);
2163 :
2164 : /* complain if locale wasn't valid */
2165 588 : if (res == NULL)
2166 : {
2167 0 : if (*locale)
2168 : {
2169 0 : pg_log_error("invalid locale name \"%s\"", locale);
2170 0 : pg_log_error_hint("If the locale name is specific to ICU, use --icu-locale.");
2171 0 : exit(1);
2172 : }
2173 : else
2174 : {
2175 : /*
2176 : * If no relevant switch was given on command line, locale is an
2177 : * empty string, which is not too helpful to report. Presumably
2178 : * setlocale() found something it did not like in the environment.
2179 : * Ideally we'd report the bad environment variable, but since
2180 : * setlocale's behavior is implementation-specific, it's hard to
2181 : * be sure what it didn't like. Print a safe generic message.
2182 : */
2183 0 : pg_fatal("invalid locale settings; check LANG and LC_* environment variables");
2184 : }
2185 : }
2186 588 : }
2187 :
2188 : /*
2189 : * check if the chosen encoding matches the encoding required by the locale
2190 : *
2191 : * this should match the similar check in the backend createdb() function
2192 : */
2193 : static bool
2194 184 : check_locale_encoding(const char *locale, int user_enc)
2195 : {
2196 : int locale_enc;
2197 :
2198 184 : locale_enc = pg_get_encoding_from_locale(locale, true);
2199 :
2200 : /* See notes in createdb() to understand these tests */
2201 192 : if (!(locale_enc == user_enc ||
2202 8 : locale_enc == PG_SQL_ASCII ||
2203 : locale_enc == -1 ||
2204 : #ifdef WIN32
2205 : user_enc == PG_UTF8 ||
2206 : #endif
2207 : user_enc == PG_SQL_ASCII))
2208 : {
2209 0 : pg_log_error("encoding mismatch");
2210 0 : pg_log_error_detail("The encoding you selected (%s) and the encoding that the "
2211 : "selected locale uses (%s) do not match. This would lead to "
2212 : "misbehavior in various character string processing functions.",
2213 : pg_encoding_to_char(user_enc),
2214 : pg_encoding_to_char(locale_enc));
2215 0 : pg_log_error_hint("Rerun %s and either do not specify an encoding explicitly, "
2216 : "or choose a matching combination.",
2217 : progname);
2218 0 : return false;
2219 : }
2220 184 : return true;
2221 : }
2222 :
2223 : /*
2224 : * check if the chosen encoding matches is supported by ICU
2225 : *
2226 : * this should match the similar check in the backend createdb() function
2227 : */
2228 : static bool
2229 12 : check_icu_locale_encoding(int user_enc)
2230 : {
2231 12 : if (!(is_encoding_supported_by_icu(user_enc)))
2232 : {
2233 2 : pg_log_error("encoding mismatch");
2234 2 : pg_log_error_detail("The encoding you selected (%s) is not supported with the ICU provider.",
2235 : pg_encoding_to_char(user_enc));
2236 2 : pg_log_error_hint("Rerun %s and either do not specify an encoding explicitly, "
2237 : "or choose a matching combination.",
2238 : progname);
2239 2 : return false;
2240 : }
2241 10 : return true;
2242 : }
2243 :
2244 : /*
2245 : * Convert to canonical BCP47 language tag. Must be consistent with
2246 : * icu_language_tag().
2247 : */
2248 : static char *
2249 14 : icu_language_tag(const char *loc_str)
2250 : {
2251 : #ifdef USE_ICU
2252 : UErrorCode status;
2253 : char *langtag;
2254 14 : size_t buflen = 32; /* arbitrary starting buffer size */
2255 14 : const bool strict = true;
2256 :
2257 : /*
2258 : * A BCP47 language tag doesn't have a clearly-defined upper limit (cf.
2259 : * RFC5646 section 4.4). Additionally, in older ICU versions,
2260 : * uloc_toLanguageTag() doesn't always return the ultimate length on the
2261 : * first call, necessitating a loop.
2262 : */
2263 14 : langtag = pg_malloc(buflen);
2264 : while (true)
2265 : {
2266 14 : status = U_ZERO_ERROR;
2267 14 : uloc_toLanguageTag(loc_str, langtag, buflen, strict, &status);
2268 :
2269 : /* try again if the buffer is not large enough */
2270 14 : if (status == U_BUFFER_OVERFLOW_ERROR ||
2271 14 : status == U_STRING_NOT_TERMINATED_WARNING)
2272 : {
2273 0 : buflen = buflen * 2;
2274 0 : langtag = pg_realloc(langtag, buflen);
2275 0 : continue;
2276 : }
2277 :
2278 14 : break;
2279 : }
2280 :
2281 14 : if (U_FAILURE(status))
2282 : {
2283 0 : pg_free(langtag);
2284 :
2285 0 : pg_fatal("could not convert locale name \"%s\" to language tag: %s",
2286 : loc_str, u_errorName(status));
2287 : }
2288 :
2289 14 : return langtag;
2290 : #else
2291 : pg_fatal("ICU is not supported in this build");
2292 : return NULL; /* keep compiler quiet */
2293 : #endif
2294 : }
2295 :
2296 : /*
2297 : * Perform best-effort check that the locale is a valid one. Should be
2298 : * consistent with pg_locale.c, except that it doesn't need to open the
2299 : * collator (that will happen during post-bootstrap initialization).
2300 : */
2301 : static void
2302 14 : icu_validate_locale(const char *loc_str)
2303 : {
2304 : #ifdef USE_ICU
2305 : UErrorCode status;
2306 : char lang[ULOC_LANG_CAPACITY];
2307 14 : bool found = false;
2308 :
2309 : /* validate that we can extract the language */
2310 14 : status = U_ZERO_ERROR;
2311 14 : uloc_getLanguage(loc_str, lang, ULOC_LANG_CAPACITY, &status);
2312 14 : if (U_FAILURE(status))
2313 : {
2314 0 : pg_fatal("could not get language from locale \"%s\": %s",
2315 : loc_str, u_errorName(status));
2316 : return;
2317 : }
2318 :
2319 : /* check for special language name */
2320 14 : if (strcmp(lang, "") == 0 ||
2321 8 : strcmp(lang, "root") == 0 || strcmp(lang, "und") == 0)
2322 6 : found = true;
2323 :
2324 : /* search for matching language within ICU */
2325 2364 : for (int32_t i = 0; !found && i < uloc_countAvailable(); i++)
2326 : {
2327 2350 : const char *otherloc = uloc_getAvailable(i);
2328 : char otherlang[ULOC_LANG_CAPACITY];
2329 :
2330 2350 : status = U_ZERO_ERROR;
2331 2350 : uloc_getLanguage(otherloc, otherlang, ULOC_LANG_CAPACITY, &status);
2332 2350 : if (U_FAILURE(status))
2333 0 : continue;
2334 :
2335 2350 : if (strcmp(lang, otherlang) == 0)
2336 6 : found = true;
2337 : }
2338 :
2339 14 : if (!found)
2340 2 : pg_fatal("locale \"%s\" has unknown language \"%s\"",
2341 : loc_str, lang);
2342 : #else
2343 : pg_fatal("ICU is not supported in this build");
2344 : #endif
2345 : }
2346 :
2347 : /*
2348 : * set up the locale variables
2349 : *
2350 : * assumes we have called setlocale(LC_ALL, "") -- see set_pglocale_pgservice
2351 : */
2352 : static void
2353 98 : setlocales(void)
2354 : {
2355 : char *canonname;
2356 :
2357 : /* set empty lc_* and datlocale values to locale config if set */
2358 :
2359 98 : if (locale)
2360 : {
2361 34 : if (!lc_ctype)
2362 30 : lc_ctype = locale;
2363 34 : if (!lc_collate)
2364 32 : lc_collate = locale;
2365 34 : if (!lc_numeric)
2366 32 : lc_numeric = locale;
2367 34 : if (!lc_time)
2368 32 : lc_time = locale;
2369 34 : if (!lc_monetary)
2370 32 : lc_monetary = locale;
2371 34 : if (!lc_messages)
2372 32 : lc_messages = locale;
2373 34 : if (!datlocale && locale_provider != COLLPROVIDER_LIBC)
2374 6 : datlocale = locale;
2375 : }
2376 :
2377 : /*
2378 : * canonicalize locale names, and obtain any missing values from our
2379 : * current environment
2380 : */
2381 98 : check_locale_name(LC_CTYPE, lc_ctype, &canonname);
2382 98 : lc_ctype = canonname;
2383 98 : check_locale_name(LC_COLLATE, lc_collate, &canonname);
2384 98 : lc_collate = canonname;
2385 98 : check_locale_name(LC_NUMERIC, lc_numeric, &canonname);
2386 98 : lc_numeric = canonname;
2387 98 : check_locale_name(LC_TIME, lc_time, &canonname);
2388 98 : lc_time = canonname;
2389 98 : check_locale_name(LC_MONETARY, lc_monetary, &canonname);
2390 98 : lc_monetary = canonname;
2391 : #if defined(LC_MESSAGES) && !defined(WIN32)
2392 98 : check_locale_name(LC_MESSAGES, lc_messages, &canonname);
2393 98 : lc_messages = canonname;
2394 : #else
2395 : /* when LC_MESSAGES is not available, use the LC_CTYPE setting */
2396 : check_locale_name(LC_CTYPE, lc_messages, &canonname);
2397 : lc_messages = canonname;
2398 : #endif
2399 :
2400 98 : if (locale_provider != COLLPROVIDER_LIBC && datlocale == NULL)
2401 4 : pg_fatal("locale must be specified if provider is %s",
2402 : collprovider_name(locale_provider));
2403 :
2404 94 : if (locale_provider == COLLPROVIDER_BUILTIN)
2405 : {
2406 10 : if (strcmp(datlocale, "C") == 0)
2407 4 : canonname = "C";
2408 6 : else if (strcmp(datlocale, "C.UTF-8") == 0 ||
2409 0 : strcmp(datlocale, "C.UTF8") == 0)
2410 6 : canonname = "C.UTF-8";
2411 : else
2412 0 : pg_fatal("invalid locale name \"%s\" for builtin provider",
2413 : datlocale);
2414 :
2415 10 : datlocale = canonname;
2416 : }
2417 84 : else if (locale_provider == COLLPROVIDER_ICU)
2418 : {
2419 : char *langtag;
2420 :
2421 : /* canonicalize to a language tag */
2422 14 : langtag = icu_language_tag(datlocale);
2423 14 : printf(_("Using language tag \"%s\" for ICU locale \"%s\".\n"),
2424 : langtag, datlocale);
2425 14 : pg_free(datlocale);
2426 14 : datlocale = langtag;
2427 :
2428 14 : icu_validate_locale(datlocale);
2429 :
2430 : /*
2431 : * In supported builds, the ICU locale ID will be opened during
2432 : * post-bootstrap initialization, which will perform extra checks.
2433 : */
2434 : #ifndef USE_ICU
2435 : pg_fatal("ICU is not supported in this build");
2436 : #endif
2437 : }
2438 92 : }
2439 :
2440 : /*
2441 : * print help text
2442 : */
2443 : static void
2444 2 : usage(const char *progname)
2445 : {
2446 2 : printf(_("%s initializes a PostgreSQL database cluster.\n\n"), progname);
2447 2 : printf(_("Usage:\n"));
2448 2 : printf(_(" %s [OPTION]... [DATADIR]\n"), progname);
2449 2 : printf(_("\nOptions:\n"));
2450 2 : printf(_(" -A, --auth=METHOD default authentication method for local connections\n"));
2451 2 : printf(_(" --auth-host=METHOD default authentication method for local TCP/IP connections\n"));
2452 2 : printf(_(" --auth-local=METHOD default authentication method for local-socket connections\n"));
2453 2 : printf(_(" [-D, --pgdata=]DATADIR location for this database cluster\n"));
2454 2 : printf(_(" -E, --encoding=ENCODING set default encoding for new databases\n"));
2455 2 : printf(_(" -g, --allow-group-access allow group read/execute on data directory\n"));
2456 2 : printf(_(" --icu-locale=LOCALE set ICU locale ID for new databases\n"));
2457 2 : printf(_(" --icu-rules=RULES set additional ICU collation rules for new databases\n"));
2458 2 : printf(_(" -k, --data-checksums use data page checksums\n"));
2459 2 : printf(_(" --locale=LOCALE set default locale for new databases\n"));
2460 2 : printf(_(" --lc-collate=, --lc-ctype=, --lc-messages=LOCALE\n"
2461 : " --lc-monetary=, --lc-numeric=, --lc-time=LOCALE\n"
2462 : " set default locale in the respective category for\n"
2463 : " new databases (default taken from environment)\n"));
2464 2 : printf(_(" --no-locale equivalent to --locale=C\n"));
2465 2 : printf(_(" --builtin-locale=LOCALE\n"
2466 : " set builtin locale name for new databases\n"));
2467 2 : printf(_(" --locale-provider={builtin|libc|icu}\n"
2468 : " set default locale provider for new databases\n"));
2469 2 : printf(_(" --pwfile=FILE read password for the new superuser from file\n"));
2470 2 : printf(_(" -T, --text-search-config=CFG\n"
2471 : " default text search configuration\n"));
2472 2 : printf(_(" -U, --username=NAME database superuser name\n"));
2473 2 : printf(_(" -W, --pwprompt prompt for a password for the new superuser\n"));
2474 2 : printf(_(" -X, --waldir=WALDIR location for the write-ahead log directory\n"));
2475 2 : printf(_(" --wal-segsize=SIZE size of WAL segments, in megabytes\n"));
2476 2 : printf(_("\nLess commonly used options:\n"));
2477 2 : printf(_(" -c, --set NAME=VALUE override default setting for server parameter\n"));
2478 2 : printf(_(" -d, --debug generate lots of debugging output\n"));
2479 2 : printf(_(" --discard-caches set debug_discard_caches=1\n"));
2480 2 : printf(_(" -L DIRECTORY where to find the input files\n"));
2481 2 : printf(_(" -n, --no-clean do not clean up after errors\n"));
2482 2 : printf(_(" -N, --no-sync do not wait for changes to be written safely to disk\n"));
2483 2 : printf(_(" --no-instructions do not print instructions for next steps\n"));
2484 2 : printf(_(" -s, --show show internal settings\n"));
2485 2 : printf(_(" --sync-method=METHOD set method for syncing files to disk\n"));
2486 2 : printf(_(" -S, --sync-only only sync database files to disk, then exit\n"));
2487 2 : printf(_("\nOther options:\n"));
2488 2 : printf(_(" -V, --version output version information, then exit\n"));
2489 2 : printf(_(" -?, --help show this help, then exit\n"));
2490 2 : printf(_("\nIf the data directory is not specified, the environment variable PGDATA\n"
2491 : "is used.\n"));
2492 2 : printf(_("\nReport bugs to <%s>.\n"), PACKAGE_BUGREPORT);
2493 2 : printf(_("%s home page: <%s>\n"), PACKAGE_NAME, PACKAGE_URL);
2494 2 : }
2495 :
2496 : static void
2497 200 : check_authmethod_unspecified(const char **authmethod)
2498 : {
2499 200 : if (*authmethod == NULL)
2500 : {
2501 88 : authwarning = true;
2502 88 : *authmethod = "trust";
2503 : }
2504 200 : }
2505 :
2506 : static void
2507 200 : check_authmethod_valid(const char *authmethod, const char *const *valid_methods, const char *conntype)
2508 : {
2509 : const char *const *p;
2510 :
2511 200 : for (p = valid_methods; *p; p++)
2512 : {
2513 200 : if (strcmp(authmethod, *p) == 0)
2514 200 : return;
2515 : /* with space = param */
2516 0 : if (strchr(authmethod, ' '))
2517 0 : if (strncmp(authmethod, *p, (authmethod - strchr(authmethod, ' '))) == 0)
2518 0 : return;
2519 : }
2520 :
2521 0 : pg_fatal("invalid authentication method \"%s\" for \"%s\" connections",
2522 : authmethod, conntype);
2523 : }
2524 :
2525 : static void
2526 100 : check_need_password(const char *authmethodlocal, const char *authmethodhost)
2527 : {
2528 100 : if ((strcmp(authmethodlocal, "md5") == 0 ||
2529 100 : strcmp(authmethodlocal, "password") == 0 ||
2530 100 : strcmp(authmethodlocal, "scram-sha-256") == 0) &&
2531 0 : (strcmp(authmethodhost, "md5") == 0 ||
2532 0 : strcmp(authmethodhost, "password") == 0 ||
2533 0 : strcmp(authmethodhost, "scram-sha-256") == 0) &&
2534 0 : !(pwprompt || pwfilename))
2535 0 : pg_fatal("must specify a password for the superuser to enable password authentication");
2536 100 : }
2537 :
2538 :
2539 : void
2540 106 : setup_pgdata(void)
2541 : {
2542 : char *pgdata_get_env;
2543 :
2544 106 : if (!pg_data)
2545 : {
2546 0 : pgdata_get_env = getenv("PGDATA");
2547 0 : if (pgdata_get_env && strlen(pgdata_get_env))
2548 : {
2549 : /* PGDATA found */
2550 0 : pg_data = pg_strdup(pgdata_get_env);
2551 : }
2552 : else
2553 : {
2554 0 : pg_log_error("no data directory specified");
2555 0 : pg_log_error_hint("You must identify the directory where the data for this database system "
2556 : "will reside. Do this with either the invocation option -D or the "
2557 : "environment variable PGDATA.");
2558 0 : exit(1);
2559 : }
2560 : }
2561 :
2562 106 : pgdata_native = pg_strdup(pg_data);
2563 106 : canonicalize_path(pg_data);
2564 :
2565 : /*
2566 : * we have to set PGDATA for postgres rather than pass it on the command
2567 : * line to avoid dumb quoting problems on Windows, and we would especially
2568 : * need quotes otherwise on Windows because paths there are most likely to
2569 : * have embedded spaces.
2570 : */
2571 106 : if (setenv("PGDATA", pg_data, 1) != 0)
2572 0 : pg_fatal("could not set environment");
2573 106 : }
2574 :
2575 :
2576 : void
2577 100 : setup_bin_paths(const char *argv0)
2578 : {
2579 : int ret;
2580 :
2581 100 : if ((ret = find_other_exec(argv0, "postgres", PG_BACKEND_VERSIONSTR,
2582 : backend_exec)) < 0)
2583 : {
2584 : char full_path[MAXPGPATH];
2585 :
2586 0 : if (find_my_exec(argv0, full_path) < 0)
2587 0 : strlcpy(full_path, progname, sizeof(full_path));
2588 :
2589 0 : if (ret == -1)
2590 0 : pg_fatal("program \"%s\" is needed by %s but was not found in the same directory as \"%s\"",
2591 : "postgres", progname, full_path);
2592 : else
2593 0 : pg_fatal("program \"%s\" was found by \"%s\" but was not the same version as %s",
2594 : "postgres", full_path, progname);
2595 : }
2596 :
2597 : /* store binary directory */
2598 100 : strcpy(bin_path, backend_exec);
2599 100 : *last_dir_separator(bin_path) = '\0';
2600 100 : canonicalize_path(bin_path);
2601 :
2602 100 : if (!share_path)
2603 : {
2604 100 : share_path = pg_malloc(MAXPGPATH);
2605 100 : get_share_path(backend_exec, share_path);
2606 : }
2607 0 : else if (!is_absolute_path(share_path))
2608 0 : pg_fatal("input file location must be an absolute path");
2609 :
2610 100 : canonicalize_path(share_path);
2611 100 : }
2612 :
2613 : void
2614 98 : setup_locale_encoding(void)
2615 : {
2616 98 : setlocales();
2617 :
2618 92 : if (locale_provider == COLLPROVIDER_LIBC &&
2619 70 : strcmp(lc_ctype, lc_collate) == 0 &&
2620 70 : strcmp(lc_ctype, lc_time) == 0 &&
2621 70 : strcmp(lc_ctype, lc_numeric) == 0 &&
2622 70 : strcmp(lc_ctype, lc_monetary) == 0 &&
2623 70 : strcmp(lc_ctype, lc_messages) == 0 &&
2624 28 : (!datlocale || strcmp(lc_ctype, datlocale) == 0))
2625 28 : printf(_("The database cluster will be initialized with locale \"%s\".\n"), lc_ctype);
2626 : else
2627 : {
2628 64 : printf(_("The database cluster will be initialized with this locale configuration:\n"));
2629 64 : printf(_(" locale provider: %s\n"), collprovider_name(locale_provider));
2630 64 : if (locale_provider != COLLPROVIDER_LIBC)
2631 22 : printf(_(" default collation: %s\n"), datlocale);
2632 64 : printf(_(" LC_COLLATE: %s\n"
2633 : " LC_CTYPE: %s\n"
2634 : " LC_MESSAGES: %s\n"
2635 : " LC_MONETARY: %s\n"
2636 : " LC_NUMERIC: %s\n"
2637 : " LC_TIME: %s\n"),
2638 : lc_collate,
2639 : lc_ctype,
2640 : lc_messages,
2641 : lc_monetary,
2642 : lc_numeric,
2643 : lc_time);
2644 : }
2645 :
2646 92 : if (!encoding)
2647 : {
2648 : int ctype_enc;
2649 :
2650 60 : ctype_enc = pg_get_encoding_from_locale(lc_ctype, true);
2651 :
2652 : /*
2653 : * If ctype_enc=SQL_ASCII, it's compatible with any encoding. ICU does
2654 : * not support SQL_ASCII, so select UTF-8 instead.
2655 : */
2656 60 : if (locale_provider == COLLPROVIDER_ICU && ctype_enc == PG_SQL_ASCII)
2657 2 : ctype_enc = PG_UTF8;
2658 :
2659 60 : if (ctype_enc == -1)
2660 : {
2661 : /* Couldn't recognize the locale's codeset */
2662 0 : pg_log_error("could not find suitable encoding for locale \"%s\"",
2663 : lc_ctype);
2664 0 : pg_log_error_hint("Rerun %s with the -E option.", progname);
2665 0 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
2666 0 : exit(1);
2667 : }
2668 60 : else if (!pg_valid_server_encoding_id(ctype_enc))
2669 : {
2670 : /*
2671 : * We recognized it, but it's not a legal server encoding. On
2672 : * Windows, UTF-8 works with any locale, so we can fall back to
2673 : * UTF-8.
2674 : */
2675 : #ifdef WIN32
2676 : encodingid = PG_UTF8;
2677 : printf(_("Encoding \"%s\" implied by locale is not allowed as a server-side encoding.\n"
2678 : "The default database encoding will be set to \"%s\" instead.\n"),
2679 : pg_encoding_to_char(ctype_enc),
2680 : pg_encoding_to_char(encodingid));
2681 : #else
2682 0 : pg_log_error("locale \"%s\" requires unsupported encoding \"%s\"",
2683 : lc_ctype, pg_encoding_to_char(ctype_enc));
2684 0 : pg_log_error_detail("Encoding \"%s\" is not allowed as a server-side encoding.",
2685 : pg_encoding_to_char(ctype_enc));
2686 0 : pg_log_error_hint("Rerun %s with a different locale selection.",
2687 : progname);
2688 0 : exit(1);
2689 : #endif
2690 : }
2691 : else
2692 : {
2693 60 : encodingid = ctype_enc;
2694 60 : printf(_("The default database encoding has accordingly been set to \"%s\".\n"),
2695 : pg_encoding_to_char(encodingid));
2696 : }
2697 : }
2698 : else
2699 32 : encodingid = get_encoding_id(encoding);
2700 :
2701 92 : if (!check_locale_encoding(lc_ctype, encodingid) ||
2702 92 : !check_locale_encoding(lc_collate, encodingid))
2703 0 : exit(1); /* check_locale_encoding printed the error */
2704 :
2705 92 : if (locale_provider == COLLPROVIDER_BUILTIN)
2706 : {
2707 10 : if (strcmp(datlocale, "C.UTF-8") == 0 && encodingid != PG_UTF8)
2708 2 : pg_fatal("builtin provider locale \"%s\" requires encoding \"%s\"",
2709 : datlocale, "UTF-8");
2710 : }
2711 :
2712 90 : if (locale_provider == COLLPROVIDER_ICU &&
2713 12 : !check_icu_locale_encoding(encodingid))
2714 2 : exit(1);
2715 88 : }
2716 :
2717 :
2718 : void
2719 98 : setup_data_file_paths(void)
2720 : {
2721 98 : set_input(&bki_file, "postgres.bki");
2722 98 : set_input(&hba_file, "pg_hba.conf.sample");
2723 98 : set_input(&ident_file, "pg_ident.conf.sample");
2724 98 : set_input(&conf_file, "postgresql.conf.sample");
2725 98 : set_input(&dictionary_file, "snowball_create.sql");
2726 98 : set_input(&info_schema_file, "information_schema.sql");
2727 98 : set_input(&features_file, "sql_features.txt");
2728 98 : set_input(&system_constraints_file, "system_constraints.sql");
2729 98 : set_input(&system_functions_file, "system_functions.sql");
2730 98 : set_input(&system_views_file, "system_views.sql");
2731 :
2732 98 : if (show_setting || debug)
2733 : {
2734 0 : fprintf(stderr,
2735 : "VERSION=%s\n"
2736 : "PGDATA=%s\nshare_path=%s\nPGPATH=%s\n"
2737 : "POSTGRES_SUPERUSERNAME=%s\nPOSTGRES_BKI=%s\n"
2738 : "POSTGRESQL_CONF_SAMPLE=%s\n"
2739 : "PG_HBA_SAMPLE=%s\nPG_IDENT_SAMPLE=%s\n",
2740 : PG_VERSION,
2741 : pg_data, share_path, bin_path,
2742 : username, bki_file,
2743 : conf_file,
2744 : hba_file, ident_file);
2745 0 : if (show_setting)
2746 0 : exit(0);
2747 : }
2748 :
2749 98 : check_input(bki_file);
2750 98 : check_input(hba_file);
2751 98 : check_input(ident_file);
2752 98 : check_input(conf_file);
2753 98 : check_input(dictionary_file);
2754 98 : check_input(info_schema_file);
2755 98 : check_input(features_file);
2756 98 : check_input(system_constraints_file);
2757 98 : check_input(system_functions_file);
2758 98 : check_input(system_views_file);
2759 98 : }
2760 :
2761 :
2762 : void
2763 88 : setup_text_search(void)
2764 : {
2765 88 : if (!default_text_search_config)
2766 : {
2767 86 : default_text_search_config = find_matching_ts_config(lc_ctype);
2768 86 : if (!default_text_search_config)
2769 : {
2770 0 : pg_log_info("could not find suitable text search configuration for locale \"%s\"",
2771 : lc_ctype);
2772 0 : default_text_search_config = "simple";
2773 : }
2774 : }
2775 : else
2776 : {
2777 2 : const char *checkmatch = find_matching_ts_config(lc_ctype);
2778 :
2779 2 : if (checkmatch == NULL)
2780 : {
2781 0 : pg_log_warning("suitable text search configuration for locale \"%s\" is unknown",
2782 : lc_ctype);
2783 : }
2784 2 : else if (strcmp(checkmatch, default_text_search_config) != 0)
2785 : {
2786 2 : pg_log_warning("specified text search configuration \"%s\" might not match locale \"%s\"",
2787 : default_text_search_config, lc_ctype);
2788 : }
2789 : }
2790 :
2791 88 : printf(_("The default text search configuration will be set to \"%s\".\n"),
2792 : default_text_search_config);
2793 88 : }
2794 :
2795 :
2796 : void
2797 88 : setup_signals(void)
2798 : {
2799 : /* some of these are not valid on Windows */
2800 : #ifdef SIGHUP
2801 88 : pqsignal(SIGHUP, trapsig);
2802 : #endif
2803 : #ifdef SIGINT
2804 88 : pqsignal(SIGINT, trapsig);
2805 : #endif
2806 : #ifdef SIGQUIT
2807 88 : pqsignal(SIGQUIT, trapsig);
2808 : #endif
2809 : #ifdef SIGTERM
2810 88 : pqsignal(SIGTERM, trapsig);
2811 : #endif
2812 :
2813 : /* Ignore SIGPIPE when writing to backend, so we can clean up */
2814 : #ifdef SIGPIPE
2815 88 : pqsignal(SIGPIPE, SIG_IGN);
2816 : #endif
2817 :
2818 : /* Prevent SIGSYS so we can probe for kernel calls that might not work */
2819 : #ifdef SIGSYS
2820 88 : pqsignal(SIGSYS, SIG_IGN);
2821 : #endif
2822 88 : }
2823 :
2824 :
2825 : void
2826 88 : create_data_directory(void)
2827 : {
2828 : int ret;
2829 :
2830 88 : switch ((ret = pg_check_dir(pg_data)))
2831 : {
2832 84 : case 0:
2833 : /* PGDATA not there, must create it */
2834 84 : printf(_("creating directory %s ... "),
2835 : pg_data);
2836 84 : fflush(stdout);
2837 :
2838 84 : if (pg_mkdir_p(pg_data, pg_dir_create_mode) != 0)
2839 0 : pg_fatal("could not create directory \"%s\": %m", pg_data);
2840 : else
2841 84 : check_ok();
2842 :
2843 84 : made_new_pgdata = true;
2844 84 : break;
2845 :
2846 2 : case 1:
2847 : /* Present but empty, fix permissions and use it */
2848 2 : printf(_("fixing permissions on existing directory %s ... "),
2849 : pg_data);
2850 2 : fflush(stdout);
2851 :
2852 2 : if (chmod(pg_data, pg_dir_create_mode) != 0)
2853 0 : pg_fatal("could not change permissions of directory \"%s\": %m",
2854 : pg_data);
2855 : else
2856 2 : check_ok();
2857 :
2858 2 : found_existing_pgdata = true;
2859 2 : break;
2860 :
2861 2 : case 2:
2862 : case 3:
2863 : case 4:
2864 : /* Present and not empty */
2865 2 : pg_log_error("directory \"%s\" exists but is not empty", pg_data);
2866 2 : if (ret != 4)
2867 0 : warn_on_mount_point(ret);
2868 : else
2869 2 : pg_log_error_hint("If you want to create a new database system, either remove or empty "
2870 : "the directory \"%s\" or run %s "
2871 : "with an argument other than \"%s\".",
2872 : pg_data, progname, pg_data);
2873 2 : exit(1); /* no further message needed */
2874 :
2875 0 : default:
2876 : /* Trouble accessing directory */
2877 0 : pg_fatal("could not access directory \"%s\": %m", pg_data);
2878 : }
2879 86 : }
2880 :
2881 :
2882 : /* Create WAL directory, and symlink if required */
2883 : void
2884 86 : create_xlog_or_symlink(void)
2885 : {
2886 : char *subdirloc;
2887 :
2888 : /* form name of the place for the subdirectory or symlink */
2889 86 : subdirloc = psprintf("%s/pg_wal", pg_data);
2890 :
2891 86 : if (xlog_dir)
2892 : {
2893 : int ret;
2894 :
2895 : /* clean up xlog directory name, check it's absolute */
2896 6 : canonicalize_path(xlog_dir);
2897 6 : if (!is_absolute_path(xlog_dir))
2898 2 : pg_fatal("WAL directory location must be an absolute path");
2899 :
2900 : /* check if the specified xlog directory exists/is empty */
2901 4 : switch ((ret = pg_check_dir(xlog_dir)))
2902 : {
2903 0 : case 0:
2904 : /* xlog directory not there, must create it */
2905 0 : printf(_("creating directory %s ... "),
2906 : xlog_dir);
2907 0 : fflush(stdout);
2908 :
2909 0 : if (pg_mkdir_p(xlog_dir, pg_dir_create_mode) != 0)
2910 0 : pg_fatal("could not create directory \"%s\": %m",
2911 : xlog_dir);
2912 : else
2913 0 : check_ok();
2914 :
2915 0 : made_new_xlogdir = true;
2916 0 : break;
2917 :
2918 2 : case 1:
2919 : /* Present but empty, fix permissions and use it */
2920 2 : printf(_("fixing permissions on existing directory %s ... "),
2921 : xlog_dir);
2922 2 : fflush(stdout);
2923 :
2924 2 : if (chmod(xlog_dir, pg_dir_create_mode) != 0)
2925 0 : pg_fatal("could not change permissions of directory \"%s\": %m",
2926 : xlog_dir);
2927 : else
2928 2 : check_ok();
2929 :
2930 2 : found_existing_xlogdir = true;
2931 2 : break;
2932 :
2933 2 : case 2:
2934 : case 3:
2935 : case 4:
2936 : /* Present and not empty */
2937 2 : pg_log_error("directory \"%s\" exists but is not empty", xlog_dir);
2938 2 : if (ret != 4)
2939 2 : warn_on_mount_point(ret);
2940 : else
2941 0 : pg_log_error_hint("If you want to store the WAL there, either remove or empty the directory \"%s\".",
2942 : xlog_dir);
2943 2 : exit(1);
2944 :
2945 0 : default:
2946 : /* Trouble accessing directory */
2947 0 : pg_fatal("could not access directory \"%s\": %m", xlog_dir);
2948 : }
2949 :
2950 2 : if (symlink(xlog_dir, subdirloc) != 0)
2951 0 : pg_fatal("could not create symbolic link \"%s\": %m",
2952 : subdirloc);
2953 : }
2954 : else
2955 : {
2956 : /* Without -X option, just make the subdirectory normally */
2957 80 : if (mkdir(subdirloc, pg_dir_create_mode) < 0)
2958 0 : pg_fatal("could not create directory \"%s\": %m",
2959 : subdirloc);
2960 : }
2961 :
2962 82 : free(subdirloc);
2963 82 : }
2964 :
2965 :
2966 : void
2967 2 : warn_on_mount_point(int error)
2968 : {
2969 2 : if (error == 2)
2970 0 : pg_log_error_detail("It contains a dot-prefixed/invisible file, perhaps due to it being a mount point.");
2971 2 : else if (error == 3)
2972 2 : pg_log_error_detail("It contains a lost+found directory, perhaps due to it being a mount point.");
2973 :
2974 2 : pg_log_error_hint("Using a mount point directly as the data directory is not recommended.\n"
2975 : "Create a subdirectory under the mount point.");
2976 2 : }
2977 :
2978 :
2979 : void
2980 88 : initialize_data_directory(void)
2981 : {
2982 : PG_CMD_DECL;
2983 : PQExpBufferData cmd;
2984 : int i;
2985 :
2986 88 : setup_signals();
2987 :
2988 : /*
2989 : * Set mask based on requested PGDATA permissions. pg_mode_mask, and
2990 : * friends like pg_dir_create_mode, are set to owner-only by default and
2991 : * then updated if -g is passed in by calling SetDataDirectoryCreatePerm()
2992 : * when parsing our options (see above).
2993 : */
2994 88 : umask(pg_mode_mask);
2995 :
2996 88 : create_data_directory();
2997 :
2998 86 : create_xlog_or_symlink();
2999 :
3000 : /* Create required subdirectories (other than pg_wal) */
3001 82 : printf(_("creating subdirectories ... "));
3002 82 : fflush(stdout);
3003 :
3004 1968 : for (i = 0; i < lengthof(subdirs); i++)
3005 : {
3006 : char *path;
3007 :
3008 1886 : path = psprintf("%s/%s", pg_data, subdirs[i]);
3009 :
3010 : /*
3011 : * The parent directory already exists, so we only need mkdir() not
3012 : * pg_mkdir_p() here, which avoids some failure modes; cf bug #13853.
3013 : */
3014 1886 : if (mkdir(path, pg_dir_create_mode) < 0)
3015 0 : pg_fatal("could not create directory \"%s\": %m", path);
3016 :
3017 1886 : free(path);
3018 : }
3019 :
3020 82 : check_ok();
3021 :
3022 : /* Top level PG_VERSION is checked by bootstrapper, so make it first */
3023 82 : write_version_file(NULL);
3024 :
3025 : /* Select suitable configuration settings */
3026 82 : set_null_conf();
3027 82 : test_config_settings();
3028 :
3029 : /* Now create all the text config files */
3030 82 : setup_config();
3031 :
3032 : /* Bootstrap template1 */
3033 82 : bootstrap_template1();
3034 :
3035 : /*
3036 : * Make the per-database PG_VERSION for template1 only after init'ing it
3037 : */
3038 80 : write_version_file("base/1");
3039 :
3040 : /*
3041 : * Create the stuff we don't need to use bootstrap mode for, using a
3042 : * backend running in simple standalone mode.
3043 : */
3044 80 : fputs(_("performing post-bootstrap initialization ... "), stdout);
3045 80 : fflush(stdout);
3046 :
3047 80 : initPQExpBuffer(&cmd);
3048 80 : printfPQExpBuffer(&cmd, "\"%s\" %s %s template1 >%s",
3049 : backend_exec, backend_options, extra_options, DEVNULL);
3050 :
3051 80 : PG_CMD_OPEN(cmd.data);
3052 :
3053 80 : setup_auth(cmdfd);
3054 :
3055 80 : setup_run_file(cmdfd, system_constraints_file);
3056 :
3057 80 : setup_run_file(cmdfd, system_functions_file);
3058 :
3059 80 : setup_depend(cmdfd);
3060 :
3061 : /*
3062 : * Note that no objects created after setup_depend() will be "pinned".
3063 : * They are all droppable at the whim of the DBA.
3064 : */
3065 :
3066 80 : setup_run_file(cmdfd, system_views_file);
3067 :
3068 80 : setup_description(cmdfd);
3069 :
3070 80 : setup_collation(cmdfd);
3071 :
3072 80 : setup_run_file(cmdfd, dictionary_file);
3073 :
3074 80 : setup_privileges(cmdfd);
3075 :
3076 80 : setup_schema(cmdfd);
3077 :
3078 80 : load_plpgsql(cmdfd);
3079 :
3080 80 : vacuum_db(cmdfd);
3081 :
3082 80 : make_template0(cmdfd);
3083 :
3084 80 : make_postgres(cmdfd);
3085 :
3086 80 : PG_CMD_CLOSE();
3087 76 : termPQExpBuffer(&cmd);
3088 :
3089 76 : check_ok();
3090 76 : }
3091 :
3092 :
3093 : int
3094 140 : main(int argc, char *argv[])
3095 : {
3096 : static struct option long_options[] = {
3097 : {"pgdata", required_argument, NULL, 'D'},
3098 : {"encoding", required_argument, NULL, 'E'},
3099 : {"locale", required_argument, NULL, 1},
3100 : {"lc-collate", required_argument, NULL, 2},
3101 : {"lc-ctype", required_argument, NULL, 3},
3102 : {"lc-monetary", required_argument, NULL, 4},
3103 : {"lc-numeric", required_argument, NULL, 5},
3104 : {"lc-time", required_argument, NULL, 6},
3105 : {"lc-messages", required_argument, NULL, 7},
3106 : {"no-locale", no_argument, NULL, 8},
3107 : {"text-search-config", required_argument, NULL, 'T'},
3108 : {"auth", required_argument, NULL, 'A'},
3109 : {"auth-local", required_argument, NULL, 10},
3110 : {"auth-host", required_argument, NULL, 11},
3111 : {"pwprompt", no_argument, NULL, 'W'},
3112 : {"pwfile", required_argument, NULL, 9},
3113 : {"username", required_argument, NULL, 'U'},
3114 : {"help", no_argument, NULL, '?'},
3115 : {"version", no_argument, NULL, 'V'},
3116 : {"debug", no_argument, NULL, 'd'},
3117 : {"show", no_argument, NULL, 's'},
3118 : {"noclean", no_argument, NULL, 'n'}, /* for backwards compatibility */
3119 : {"no-clean", no_argument, NULL, 'n'},
3120 : {"nosync", no_argument, NULL, 'N'}, /* for backwards compatibility */
3121 : {"no-sync", no_argument, NULL, 'N'},
3122 : {"no-instructions", no_argument, NULL, 13},
3123 : {"set", required_argument, NULL, 'c'},
3124 : {"sync-only", no_argument, NULL, 'S'},
3125 : {"waldir", required_argument, NULL, 'X'},
3126 : {"wal-segsize", required_argument, NULL, 12},
3127 : {"data-checksums", no_argument, NULL, 'k'},
3128 : {"allow-group-access", no_argument, NULL, 'g'},
3129 : {"discard-caches", no_argument, NULL, 14},
3130 : {"locale-provider", required_argument, NULL, 15},
3131 : {"builtin-locale", required_argument, NULL, 16},
3132 : {"icu-locale", required_argument, NULL, 17},
3133 : {"icu-rules", required_argument, NULL, 18},
3134 : {"sync-method", required_argument, NULL, 19},
3135 : {NULL, 0, NULL, 0}
3136 : };
3137 :
3138 : /*
3139 : * options with no short version return a low integer, the rest return
3140 : * their short version value
3141 : */
3142 : int c;
3143 : int option_index;
3144 : char *effective_user;
3145 : PQExpBuffer start_db_cmd;
3146 : char pg_ctl_path[MAXPGPATH];
3147 :
3148 : /*
3149 : * Ensure that buffering behavior of stdout matches what it is in
3150 : * interactive usage (at least on most platforms). This prevents
3151 : * unexpected output ordering when, eg, output is redirected to a file.
3152 : * POSIX says we must do this before any other usage of these files.
3153 : */
3154 140 : setvbuf(stdout, NULL, PG_IOLBF, 0);
3155 :
3156 140 : pg_logging_init(argv[0]);
3157 140 : progname = get_progname(argv[0]);
3158 140 : set_pglocale_pgservice(argv[0], PG_TEXTDOMAIN("initdb"));
3159 :
3160 140 : if (argc > 1)
3161 : {
3162 140 : if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
3163 : {
3164 2 : usage(progname);
3165 2 : exit(0);
3166 : }
3167 138 : if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
3168 : {
3169 22 : puts("initdb (PostgreSQL) " PG_VERSION);
3170 22 : exit(0);
3171 : }
3172 : }
3173 :
3174 : /* process command-line options */
3175 :
3176 548 : while ((c = getopt_long(argc, argv, "A:c:dD:E:gkL:nNsST:U:WX:",
3177 : long_options, &option_index)) != -1)
3178 : {
3179 436 : switch (c)
3180 : {
3181 56 : case 'A':
3182 56 : authmethodlocal = authmethodhost = pg_strdup(optarg);
3183 :
3184 : /*
3185 : * When ident is specified, use peer for local connections.
3186 : * Mirrored, when peer is specified, use ident for TCP/IP
3187 : * connections.
3188 : */
3189 56 : if (strcmp(authmethodhost, "ident") == 0)
3190 0 : authmethodlocal = "peer";
3191 56 : else if (strcmp(authmethodlocal, "peer") == 0)
3192 0 : authmethodhost = "ident";
3193 56 : break;
3194 0 : case 10:
3195 0 : authmethodlocal = pg_strdup(optarg);
3196 0 : break;
3197 0 : case 11:
3198 0 : authmethodhost = pg_strdup(optarg);
3199 0 : break;
3200 10 : case 'c':
3201 : {
3202 10 : char *buf = pg_strdup(optarg);
3203 10 : char *equals = strchr(buf, '=');
3204 :
3205 10 : if (!equals)
3206 : {
3207 0 : pg_log_error("-c %s requires a value", buf);
3208 0 : pg_log_error_hint("Try \"%s --help\" for more information.",
3209 : progname);
3210 0 : exit(1);
3211 : }
3212 10 : *equals++ = '\0'; /* terminate variable name */
3213 10 : add_stringlist_item(&extra_guc_names, buf);
3214 10 : add_stringlist_item(&extra_guc_values, equals);
3215 10 : pfree(buf);
3216 : }
3217 10 : break;
3218 56 : case 'D':
3219 56 : pg_data = pg_strdup(optarg);
3220 56 : break;
3221 32 : case 'E':
3222 32 : encoding = pg_strdup(optarg);
3223 32 : break;
3224 0 : case 'W':
3225 0 : pwprompt = true;
3226 0 : break;
3227 8 : case 'U':
3228 8 : username = pg_strdup(optarg);
3229 8 : break;
3230 0 : case 'd':
3231 0 : debug = true;
3232 0 : printf(_("Running in debug mode.\n"));
3233 0 : break;
3234 8 : case 'n':
3235 8 : noclean = true;
3236 8 : printf(_("Running in no-clean mode. Mistakes will not be cleaned up.\n"));
3237 8 : break;
3238 96 : case 'N':
3239 96 : do_sync = false;
3240 96 : break;
3241 6 : case 'S':
3242 6 : sync_only = true;
3243 6 : break;
3244 2 : case 'k':
3245 2 : data_checksums = true;
3246 2 : break;
3247 0 : case 'L':
3248 0 : share_path = pg_strdup(optarg);
3249 0 : break;
3250 30 : case 1:
3251 30 : locale = pg_strdup(optarg);
3252 30 : break;
3253 8 : case 2:
3254 8 : lc_collate = pg_strdup(optarg);
3255 8 : break;
3256 10 : case 3:
3257 10 : lc_ctype = pg_strdup(optarg);
3258 10 : break;
3259 2 : case 4:
3260 2 : lc_monetary = pg_strdup(optarg);
3261 2 : break;
3262 2 : case 5:
3263 2 : lc_numeric = pg_strdup(optarg);
3264 2 : break;
3265 2 : case 6:
3266 2 : lc_time = pg_strdup(optarg);
3267 2 : break;
3268 6 : case 7:
3269 6 : lc_messages = pg_strdup(optarg);
3270 6 : break;
3271 4 : case 8:
3272 4 : locale = "C";
3273 4 : break;
3274 0 : case 9:
3275 0 : pwfilename = pg_strdup(optarg);
3276 0 : break;
3277 0 : case 's':
3278 0 : show_setting = true;
3279 0 : break;
3280 2 : case 'T':
3281 2 : default_text_search_config = pg_strdup(optarg);
3282 2 : break;
3283 6 : case 'X':
3284 6 : xlog_dir = pg_strdup(optarg);
3285 6 : break;
3286 10 : case 12:
3287 10 : if (!option_parse_int(optarg, "--wal-segsize", 1, 1024, &wal_segment_size_mb))
3288 0 : exit(1);
3289 10 : break;
3290 4 : case 13:
3291 4 : noinstructions = true;
3292 4 : break;
3293 10 : case 'g':
3294 10 : SetDataDirectoryCreatePerm(PG_DIR_MODE_GROUP);
3295 10 : break;
3296 0 : case 14:
3297 0 : extra_options = psprintf("%s %s",
3298 : extra_options,
3299 : "-c debug_discard_caches=1");
3300 0 : break;
3301 38 : case 15:
3302 38 : if (strcmp(optarg, "builtin") == 0)
3303 16 : locale_provider = COLLPROVIDER_BUILTIN;
3304 22 : else if (strcmp(optarg, "icu") == 0)
3305 16 : locale_provider = COLLPROVIDER_ICU;
3306 6 : else if (strcmp(optarg, "libc") == 0)
3307 4 : locale_provider = COLLPROVIDER_LIBC;
3308 : else
3309 2 : pg_fatal("unrecognized locale provider: %s", optarg);
3310 36 : break;
3311 6 : case 16:
3312 6 : datlocale = pg_strdup(optarg);
3313 6 : builtin_locale_specified = true;
3314 6 : break;
3315 16 : case 17:
3316 16 : datlocale = pg_strdup(optarg);
3317 16 : icu_locale_specified = true;
3318 16 : break;
3319 2 : case 18:
3320 2 : icu_rules = pg_strdup(optarg);
3321 2 : break;
3322 2 : case 19:
3323 2 : if (!parse_sync_method(optarg, &sync_method))
3324 0 : exit(1);
3325 2 : break;
3326 2 : default:
3327 : /* getopt_long already emitted a complaint */
3328 2 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
3329 2 : exit(1);
3330 : }
3331 : }
3332 :
3333 :
3334 : /*
3335 : * Non-option argument specifies data directory as long as it wasn't
3336 : * already specified with -D / --pgdata
3337 : */
3338 112 : if (optind < argc && !pg_data)
3339 : {
3340 56 : pg_data = pg_strdup(argv[optind]);
3341 56 : optind++;
3342 : }
3343 :
3344 112 : if (optind < argc)
3345 : {
3346 0 : pg_log_error("too many command-line arguments (first is \"%s\")",
3347 : argv[optind]);
3348 0 : pg_log_error_hint("Try \"%s --help\" for more information.", progname);
3349 0 : exit(1);
3350 : }
3351 :
3352 112 : if (builtin_locale_specified && locale_provider != COLLPROVIDER_BUILTIN)
3353 0 : pg_fatal("%s cannot be specified unless locale provider \"%s\" is chosen",
3354 : "--builtin-locale", "builtin");
3355 :
3356 112 : if (icu_locale_specified && locale_provider != COLLPROVIDER_ICU)
3357 4 : pg_fatal("%s cannot be specified unless locale provider \"%s\" is chosen",
3358 : "--icu-locale", "icu");
3359 :
3360 108 : if (icu_rules && locale_provider != COLLPROVIDER_ICU)
3361 2 : pg_fatal("%s cannot be specified unless locale provider \"%s\" is chosen",
3362 : "--icu-rules", "icu");
3363 :
3364 106 : atexit(cleanup_directories_atexit);
3365 :
3366 : /* If we only need to sync, just do it and exit */
3367 106 : if (sync_only)
3368 : {
3369 6 : setup_pgdata();
3370 :
3371 : /* must check that directory is readable */
3372 6 : if (pg_check_dir(pg_data) <= 0)
3373 2 : pg_fatal("could not access directory \"%s\": %m", pg_data);
3374 :
3375 4 : fputs(_("syncing data to disk ... "), stdout);
3376 4 : fflush(stdout);
3377 4 : sync_pgdata(pg_data, PG_VERSION_NUM, sync_method);
3378 4 : check_ok();
3379 4 : return 0;
3380 : }
3381 :
3382 100 : if (pwprompt && pwfilename)
3383 0 : pg_fatal("password prompt and password file cannot be specified together");
3384 :
3385 100 : check_authmethod_unspecified(&authmethodlocal);
3386 100 : check_authmethod_unspecified(&authmethodhost);
3387 :
3388 100 : check_authmethod_valid(authmethodlocal, auth_methods_local, "local");
3389 100 : check_authmethod_valid(authmethodhost, auth_methods_host, "host");
3390 :
3391 100 : check_need_password(authmethodlocal, authmethodhost);
3392 :
3393 100 : if (!IsValidWalSegSize(wal_segment_size_mb * 1024 * 1024))
3394 0 : pg_fatal("argument of %s must be a power of two between 1 and 1024", "--wal-segsize");
3395 :
3396 100 : get_restricted_token();
3397 :
3398 100 : setup_pgdata();
3399 :
3400 100 : setup_bin_paths(argv[0]);
3401 :
3402 100 : effective_user = get_id();
3403 100 : if (!username)
3404 92 : username = effective_user;
3405 :
3406 100 : if (strncmp(username, "pg_", 3) == 0)
3407 2 : pg_fatal("superuser name \"%s\" is disallowed; role names cannot begin with \"pg_\"", username);
3408 :
3409 98 : printf(_("The files belonging to this database system will be owned "
3410 : "by user \"%s\".\n"
3411 : "This user must also own the server process.\n\n"),
3412 : effective_user);
3413 :
3414 98 : set_info_version();
3415 :
3416 98 : setup_data_file_paths();
3417 :
3418 98 : setup_locale_encoding();
3419 :
3420 88 : setup_text_search();
3421 :
3422 88 : printf("\n");
3423 :
3424 88 : if (data_checksums)
3425 2 : printf(_("Data page checksums are enabled.\n"));
3426 : else
3427 86 : printf(_("Data page checksums are disabled.\n"));
3428 :
3429 88 : if (pwprompt || pwfilename)
3430 0 : get_su_pwd();
3431 :
3432 88 : printf("\n");
3433 :
3434 88 : initialize_data_directory();
3435 :
3436 76 : if (do_sync)
3437 : {
3438 4 : fputs(_("syncing data to disk ... "), stdout);
3439 4 : fflush(stdout);
3440 4 : sync_pgdata(pg_data, PG_VERSION_NUM, sync_method);
3441 4 : check_ok();
3442 : }
3443 : else
3444 72 : printf(_("\nSync to disk skipped.\nThe data directory might become corrupt if the operating system crashes.\n"));
3445 :
3446 76 : if (authwarning)
3447 : {
3448 20 : printf("\n");
3449 20 : pg_log_warning("enabling \"trust\" authentication for local connections");
3450 20 : pg_log_warning_hint("You can change this by editing pg_hba.conf or using the option -A, or "
3451 : "--auth-local and --auth-host, the next time you run initdb.");
3452 : }
3453 :
3454 76 : if (!noinstructions)
3455 : {
3456 : /*
3457 : * Build up a shell command to tell the user how to start the server
3458 : */
3459 72 : start_db_cmd = createPQExpBuffer();
3460 :
3461 : /* Get directory specification used to start initdb ... */
3462 72 : strlcpy(pg_ctl_path, argv[0], sizeof(pg_ctl_path));
3463 72 : canonicalize_path(pg_ctl_path);
3464 72 : get_parent_directory(pg_ctl_path);
3465 : /* ... and tag on pg_ctl instead */
3466 72 : join_path_components(pg_ctl_path, pg_ctl_path, "pg_ctl");
3467 :
3468 : /* Convert the path to use native separators */
3469 72 : make_native_path(pg_ctl_path);
3470 :
3471 : /* path to pg_ctl, properly quoted */
3472 72 : appendShellString(start_db_cmd, pg_ctl_path);
3473 :
3474 : /* add -D switch, with properly quoted data directory */
3475 72 : appendPQExpBufferStr(start_db_cmd, " -D ");
3476 72 : appendShellString(start_db_cmd, pgdata_native);
3477 :
3478 : /* add suggested -l switch and "start" command */
3479 : /* translator: This is a placeholder in a shell command. */
3480 72 : appendPQExpBuffer(start_db_cmd, " -l %s start", _("logfile"));
3481 :
3482 72 : printf(_("\nSuccess. You can now start the database server using:\n\n"
3483 : " %s\n\n"),
3484 : start_db_cmd->data);
3485 :
3486 72 : destroyPQExpBuffer(start_db_cmd);
3487 : }
3488 :
3489 :
3490 76 : success = true;
3491 76 : return 0;
3492 : }
|