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