LCOV - differential code coverage report
Current view: top level - src/bin/initdb - initdb.c (source / functions) Coverage Total Hit UNC LBC UBC GBC GIC GNC CBC DUB DCB
Current: Differential Code Coverage 16@8cea358b128 vs 17@8cea358b128 Lines: 81.5 % 1226 999 7 220 9 66 924 3 41
Current Date: 2024-04-14 14:21:10 Functions: 96.8 % 62 60 2 7 53
Baseline: 16@8cea358b128 Branches: 64.8 % 733 475 17 1 240 7 2 51 415
Baseline Date: 2024-04-14 14:21:09 Line coverage date bins:
Legend: Lines: hit not hit | Branches: + taken - not taken # not executed [..60] days: 93.2 % 44 41 3 2 36 3
(60,120] days: 100.0 % 4 4 4
(180,240] days: 72.7 % 11 8 3 8
(240..) days: 81.1 % 1167 946 1 220 7 22 917
Function coverage date bins:
(240..) days: 96.8 % 62 60 2 7 53
Branch coverage date bins:
[..60] days: 83.3 % 48 40 8 36 4
(180,240] days: 50.0 % 12 6 6 6
(240..) days: 63.7 % 673 429 3 1 240 7 2 9 411

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

Generated by: LCOV version 2.1-beta2-3-g6141622