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