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