LCOV - differential code coverage report
Current view: top level - src/bin/pg_upgrade - pg_upgrade.c (source / functions) Coverage Total Hit UNC UBC GNC CBC DUB DCB
Current: 7a15cff1f11193467898da1c1fabf06fd2caee04 vs 84a3778c79c2d28b4dc281d03ef2ab019b16483b Lines: 82.9 % 334 277 14 43 18 259 4 4
Current Date: 2025-12-15 18:36:29 -0500 Functions: 100.0 % 14 14 3 11
Baseline: lcov-20251216-010103-baseline Branches: 58.2 % 134 78 9 47 11 67
Baseline Date: 2025-12-15 13:30:48 -0800 Line coverage date bins:
Legend: Lines:     hit not hit
Branches: + taken - not taken # not executed
(1,7] days: 26.3 % 19 5 14 5
(30,360] days: 94.4 % 36 34 2 13 21
(360..) days: 85.3 % 279 238 41 238
Function coverage date bins:
(30,360] days: 100.0 % 2 2 1 1
(360..) days: 100.0 % 12 12 2 10
Branch coverage date bins:
(1,7] days: 25.0 % 12 3 9 3
(30,360] days: 83.3 % 24 20 4 8 12
(360..) days: 56.1 % 98 55 43 55

 Age         Owner                    Branch data    TLA  Line data    Source code
                                  1                 :                : /*
                                  2                 :                :  *  pg_upgrade.c
                                  3                 :                :  *
                                  4                 :                :  *  main source file
                                  5                 :                :  *
                                  6                 :                :  *  Copyright (c) 2010-2025, PostgreSQL Global Development Group
                                  7                 :                :  *  src/bin/pg_upgrade/pg_upgrade.c
                                  8                 :                :  */
                                  9                 :                : 
                                 10                 :                : /*
                                 11                 :                :  *  To simplify the upgrade process, we force certain system values to be
                                 12                 :                :  *  identical between old and new clusters:
                                 13                 :                :  *
                                 14                 :                :  *  We control all assignments of pg_class.oid (and relfilenode) so toast
                                 15                 :                :  *  oids are the same between old and new clusters.  This is important
                                 16                 :                :  *  because toast oids are stored as toast pointers in user tables.
                                 17                 :                :  *
                                 18                 :                :  *  While pg_class.oid and pg_class.relfilenode are initially the same in a
                                 19                 :                :  *  cluster, they can diverge due to CLUSTER, REINDEX, or VACUUM FULL. We
                                 20                 :                :  *  control assignments of pg_class.relfilenode because we want the filenames
                                 21                 :                :  *  to match between the old and new cluster.
                                 22                 :                :  *
                                 23                 :                :  *  We control assignment of pg_tablespace.oid because we want the oid to match
                                 24                 :                :  *  between the old and new cluster.
                                 25                 :                :  *
                                 26                 :                :  *  We control all assignments of pg_type.oid because these oids are stored
                                 27                 :                :  *  in user composite type values.
                                 28                 :                :  *
                                 29                 :                :  *  We control all assignments of pg_enum.oid because these oids are stored
                                 30                 :                :  *  in user tables as enum values.
                                 31                 :                :  *
                                 32                 :                :  *  We control all assignments of pg_authid.oid because the oids are stored in
                                 33                 :                :  *  pg_largeobject_metadata, which is copied via file transfer for upgrades
                                 34                 :                :  *  from v16 and newer.
                                 35                 :                :  *
                                 36                 :                :  *  We control all assignments of pg_database.oid because we want the directory
                                 37                 :                :  *  names to match between the old and new cluster.
                                 38                 :                :  */
                                 39                 :                : 
                                 40                 :                : 
                                 41                 :                : 
                                 42                 :                : #include "postgres_fe.h"
                                 43                 :                : 
                                 44                 :                : #include <time.h>
                                 45                 :                : 
                                 46                 :                : #include "access/multixact.h"
                                 47                 :                : #include "catalog/pg_class_d.h"
                                 48                 :                : #include "common/file_perm.h"
                                 49                 :                : #include "common/logging.h"
                                 50                 :                : #include "common/restricted_token.h"
                                 51                 :                : #include "fe_utils/string_utils.h"
                                 52                 :                : #include "pg_upgrade.h"
                                 53                 :                : 
                                 54                 :                : /*
                                 55                 :                :  * Maximum number of pg_restore actions (TOC entries) to process within one
                                 56                 :                :  * transaction.  At some point we might want to make this user-controllable,
                                 57                 :                :  * but for now a hard-wired setting will suffice.
                                 58                 :                :  */
                                 59                 :                : #define RESTORE_TRANSACTION_SIZE 1000
                                 60                 :                : 
                                 61                 :                : static void set_new_cluster_char_signedness(void);
                                 62                 :                : static void set_locale_and_encoding(void);
                                 63                 :                : static void prepare_new_cluster(void);
                                 64                 :                : static void prepare_new_globals(void);
                                 65                 :                : static void create_new_objects(void);
                                 66                 :                : static void copy_xact_xlog_xid(void);
                                 67                 :                : static void set_frozenxids(bool minmxid_only);
                                 68                 :                : static void make_outputdirs(char *pgdata);
                                 69                 :                : static void setup(char *argv0);
                                 70                 :                : static void create_logical_replication_slots(void);
                                 71                 :                : static void create_conflict_detection_slot(void);
                                 72                 :                : 
                                 73                 :                : ClusterInfo old_cluster,
                                 74                 :                :             new_cluster;
                                 75                 :                : OSInfo      os_info;
                                 76                 :                : 
                                 77                 :                : char       *output_files[] = {
                                 78                 :                :     SERVER_LOG_FILE,
                                 79                 :                : #ifdef WIN32
                                 80                 :                :     /* unique file for pg_ctl start */
                                 81                 :                :     SERVER_START_LOG_FILE,
                                 82                 :                : #endif
                                 83                 :                :     UTILITY_LOG_FILE,
                                 84                 :                :     INTERNAL_LOG_FILE,
                                 85                 :                :     NULL
                                 86                 :                : };
                                 87                 :                : 
                                 88                 :                : 
                                 89                 :                : int
 5697 bruce@momjian.us           90                 :CBC          22 : main(int argc, char **argv)
                                 91                 :                : {
                                 92                 :             22 :     char       *deletion_script_file_name = NULL;
                                 93                 :                :     bool        migrate_logical_slots;
                                 94                 :                : 
                                 95                 :                :     /*
                                 96                 :                :      * pg_upgrade doesn't currently use common/logging.c, but initialize it
                                 97                 :                :      * anyway because we might call common code that does.
                                 98                 :                :      */
 2451 peter@eisentraut.org       99                 :             22 :     pg_logging_init(argv[0]);
 3350 peter_e@gmx.net           100                 :             22 :     set_pglocale_pgservice(argv[0], PG_TEXTDOMAIN("pg_upgrade"));
                                101                 :                : 
                                102                 :                :     /* Set default restrictive mask until new cluster permissions are read */
 2810 sfrost@snowman.net        103                 :             22 :     umask(PG_MODE_MASK_OWNER);
                                104                 :                : 
 5537 bruce@momjian.us          105                 :             22 :     parseCommandLine(argc, argv);
                                106                 :                : 
 2451 peter@eisentraut.org      107                 :             19 :     get_restricted_token();
                                108                 :                : 
 5184 bruce@momjian.us          109                 :             19 :     adjust_data_dir(&old_cluster);
                                110                 :             19 :     adjust_data_dir(&new_cluster);
                                111                 :                : 
                                112                 :                :     /*
                                113                 :                :      * Set mask based on PGDATA permissions, needed for the creation of the
                                114                 :                :      * output directories with correct permissions.
                                115                 :                :      */
 1409 michael@paquier.xyz       116         [ -  + ]:             19 :     if (!GetDataDirectoryCreatePerm(new_cluster.pgdata))
  644 michael@paquier.xyz       117                 :UBC           0 :         pg_fatal("could not read permissions of directory \"%s\": %m",
                                118                 :                :                  new_cluster.pgdata);
                                119                 :                : 
 1409 michael@paquier.xyz       120                 :CBC          19 :     umask(pg_mode_mask);
                                121                 :                : 
                                122                 :                :     /*
                                123                 :                :      * This needs to happen after adjusting the data directory of the new
                                124                 :                :      * cluster in adjust_data_dir().
                                125                 :                :      */
                                126                 :             19 :     make_outputdirs(new_cluster.pgdata);
                                127                 :                : 
  508 nathan@postgresql.or      128                 :             19 :     setup(argv[0]);
                                129                 :                : 
                                130                 :             18 :     output_check_banner();
                                131                 :                : 
 5537 bruce@momjian.us          132                 :             18 :     check_cluster_versions();
                                133                 :                : 
  508 nathan@postgresql.or      134                 :             17 :     get_sock_dir(&old_cluster);
                                135                 :             17 :     get_sock_dir(&new_cluster);
                                136                 :                : 
                                137                 :             17 :     check_cluster_compatibility();
                                138                 :                : 
                                139                 :             17 :     check_and_dump_old_cluster();
                                140                 :                : 
                                141                 :                : 
                                142                 :                :     /* -- NEW -- */
 4709 bruce@momjian.us          143                 :             14 :     start_postmaster(&new_cluster, true);
                                144                 :                : 
 5537                           145                 :             14 :     check_new_cluster();
                                146                 :             10 :     report_clusters_compatible();
                                147                 :                : 
 3038 peter_e@gmx.net           148                 :              9 :     pg_log(PG_REPORT,
                                149                 :                :            "\n"
                                150                 :                :            "Performing Upgrade\n"
                                151                 :                :            "------------------");
                                152                 :                : 
 1013 jdavis@postgresql.or      153                 :              9 :     set_locale_and_encoding();
                                154                 :                : 
 5537 bruce@momjian.us          155                 :              9 :     prepare_new_cluster();
                                156                 :                : 
 5349                           157                 :              9 :     stop_postmaster(false);
                                158                 :                : 
                                159                 :                :     /*
                                160                 :                :      * Destructive Changes to New Cluster
                                161                 :                :      */
                                162                 :                : 
 3196 rhaas@postgresql.org      163                 :              9 :     copy_xact_xlog_xid();
  298 msawada@postgresql.o      164                 :              9 :     set_new_cluster_char_signedness();
                                165                 :                : 
                                166                 :                :     /* New now using xids of the old system */
                                167                 :                : 
                                168                 :                :     /* -- NEW -- */
 4709 bruce@momjian.us          169                 :              9 :     start_postmaster(&new_cluster, true);
                                170                 :                : 
 2885 tgl@sss.pgh.pa.us         171                 :              9 :     prepare_new_globals();
                                172                 :                : 
 5537 bruce@momjian.us          173                 :              9 :     create_new_objects();
                                174                 :                : 
 5349                           175                 :              9 :     stop_postmaster(false);
                                176                 :                : 
                                177                 :                :     /*
                                178                 :                :      * Most failures happen in create_new_objects(), which has completed at
                                179                 :                :      * this point.  We do this here because it is just before file transfer,
                                180                 :                :      * which for --link will make it unsafe to start the old cluster once the
                                181                 :                :      * new cluster is started, and for --swap will make it unsafe to start the
                                182                 :                :      * old cluster at all.
                                183                 :                :      */
  266 nathan@postgresql.or      184         [ +  + ]:              9 :     if (user_opts.transfer_mode == TRANSFER_MODE_LINK ||
                                185         [ +  + ]:              8 :         user_opts.transfer_mode == TRANSFER_MODE_SWAP)
                                186                 :              2 :         disable_old_cluster(user_opts.transfer_mode);
                                187                 :                : 
 4724 bruce@momjian.us          188                 :              9 :     transfer_all_new_tablespaces(&old_cluster.dbarr, &new_cluster.dbarr,
                                189                 :                :                                  old_cluster.pgdata, new_cluster.pgdata);
                                190                 :                : 
                                191                 :                :     /*
                                192                 :                :      * Assuming OIDs are only used in system tables, there is no need to
                                193                 :                :      * restore the OID counter because we have not transferred any OIDs from
                                194                 :                :      * the old system, but we do it anyway just in case.  We do it late here
                                195                 :                :      * because there is no need to have the schema load use new oids.
                                196                 :                :      */
 5271 peter_e@gmx.net           197                 :              9 :     prep_status("Setting next OID for new cluster");
 2899 bruce@momjian.us          198                 :              9 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                199                 :                :               "\"%s/pg_resetwal\" -o %u \"%s\"",
                                200                 :                :               new_cluster.bindir, old_cluster.controldata.chkpnt_nxtoid,
                                201                 :                :               new_cluster.pgdata);
 5537                           202                 :              9 :     check_ok();
                                203                 :                : 
  146 akapila@postgresql.o      204                 :GNC           9 :     migrate_logical_slots = count_old_cluster_logical_slots();
                                205                 :                : 
                                206                 :                :     /*
                                207                 :                :      * Migrate replication slots to the new cluster.
                                208                 :                :      *
                                209                 :                :      * Note that we must migrate logical slots after resetting WAL because
                                210                 :                :      * otherwise the required WAL would be removed and slots would become
                                211                 :                :      * unusable.  There is a possibility that background processes might
                                212                 :                :      * generate some WAL before we could create the slots in the new cluster
                                213                 :                :      * but we can ignore that WAL as that won't be required downstream.
                                214                 :                :      *
                                215                 :                :      * The conflict detection slot is not affected by concerns related to WALs
                                216                 :                :      * as it only retains the dead tuples. It is created here for consistency.
                                217                 :                :      * Note that the new conflict detection slot uses the latest transaction
                                218                 :                :      * ID as xmin, so it cannot protect dead tuples that existed before the
                                219                 :                :      * upgrade. Additionally, commit timestamps and origin data are not
                                220                 :                :      * preserved during the upgrade. So, even after creating the slot, the
                                221                 :                :      * upgraded subscriber may be unable to detect conflicts or log relevant
                                222                 :                :      * commit timestamps and origins when applying changes from the publisher
                                223                 :                :      * occurred before the upgrade especially if those changes were not
                                224                 :                :      * replicated. It can only protect tuples that might be deleted after the
                                225                 :                :      * new cluster starts.
                                226                 :                :      */
                                227   [ +  +  +  + ]:              9 :     if (migrate_logical_slots || old_cluster.sub_retain_dead_tuples)
                                228                 :                :     {
  782 akapila@postgresql.o      229                 :CBC           2 :         start_postmaster(&new_cluster, true);
                                230                 :                : 
  146 akapila@postgresql.o      231         [ +  + ]:GNC           2 :         if (migrate_logical_slots)
                                232                 :              1 :             create_logical_replication_slots();
                                233                 :                : 
                                234         [ +  + ]:              2 :         if (old_cluster.sub_retain_dead_tuples)
                                235                 :              1 :             create_conflict_detection_slot();
                                236                 :                : 
  782 akapila@postgresql.o      237                 :CBC           2 :         stop_postmaster(false);
                                238                 :                :     }
                                239                 :                : 
 1459 michael@paquier.xyz       240         [ -  + ]:              9 :     if (user_opts.do_sync)
                                241                 :                :     {
 1459 michael@paquier.xyz       242                 :UBC           0 :         prep_status("Sync data directory to disk");
                                243                 :              0 :         exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                244                 :                :                   "\"%s/initdb\" --sync-only %s \"%s\" --sync-method %s",
                                245                 :                :                   new_cluster.bindir,
  266 nathan@postgresql.or      246         [ #  # ]:              0 :                   (user_opts.transfer_mode == TRANSFER_MODE_SWAP) ?
                                247                 :                :                   "--no-sync-data-files" : "",
                                248                 :                :                   new_cluster.pgdata,
                                249                 :                :                   user_opts.sync_method);
 1459 michael@paquier.xyz       250                 :              0 :         check_ok();
                                251                 :                :     }
                                252                 :                : 
 5537 bruce@momjian.us          253                 :CBC           9 :     create_script_for_old_cluster_deletion(&deletion_script_file_name);
                                254                 :                : 
 3101                           255                 :              9 :     issue_warnings_and_set_wal_level();
                                256                 :                : 
 3038 peter_e@gmx.net           257                 :              9 :     pg_log(PG_REPORT,
                                258                 :                :            "\n"
                                259                 :                :            "Upgrade Complete\n"
                                260                 :                :            "----------------");
                                261                 :                : 
 1863 magnus@hagander.net       262                 :              9 :     output_completion_banner(deletion_script_file_name);
                                263                 :                : 
 5697 bruce@momjian.us          264                 :              9 :     pg_free(deletion_script_file_name);
                                265                 :                : 
 1287 michael@paquier.xyz       266                 :              9 :     cleanup_output_dirs();
                                267                 :                : 
 5697 bruce@momjian.us          268                 :              9 :     return 0;
                                269                 :                : }
                                270                 :                : 
                                271                 :                : /*
                                272                 :                :  * Create and assign proper permissions to the set of output directories
                                273                 :                :  * used to store any data generated internally, filling in log_opts in
                                274                 :                :  * the process.
                                275                 :                :  */
                                276                 :                : static void
 1409 michael@paquier.xyz       277                 :             19 : make_outputdirs(char *pgdata)
                                278                 :                : {
                                279                 :                :     FILE       *fp;
                                280                 :                :     char      **filename;
                                281                 :             19 :     time_t      run_time = time(NULL);
                                282                 :                :     char        filename_path[MAXPGPATH];
                                283                 :                :     char        timebuf[128];
                                284                 :                :     struct timeval time;
                                285                 :                :     time_t      tt;
                                286                 :                :     int         len;
                                287                 :                : 
 1287                           288                 :             19 :     log_opts.rootdir = (char *) pg_malloc0(MAXPGPATH);
                                289                 :             19 :     len = snprintf(log_opts.rootdir, MAXPGPATH, "%s/%s", pgdata, BASE_OUTPUTDIR);
                                290         [ -  + ]:             19 :     if (len >= MAXPGPATH)
 1253 tgl@sss.pgh.pa.us         291                 :UBC           0 :         pg_fatal("directory path for new cluster is too long");
                                292                 :                : 
                                293                 :                :     /* BASE_OUTPUTDIR/$timestamp/ */
 1287 michael@paquier.xyz       294                 :CBC          19 :     gettimeofday(&time, NULL);
                                295                 :             19 :     tt = (time_t) time.tv_sec;
                                296                 :             19 :     strftime(timebuf, sizeof(timebuf), "%Y%m%dT%H%M%S", localtime(&tt));
                                297                 :                :     /* append milliseconds */
                                298                 :             19 :     snprintf(timebuf + strlen(timebuf), sizeof(timebuf) - strlen(timebuf),
                                299                 :             19 :              ".%03d", (int) (time.tv_usec / 1000));
                                300                 :             19 :     log_opts.basedir = (char *) pg_malloc0(MAXPGPATH);
                                301                 :             19 :     len = snprintf(log_opts.basedir, MAXPGPATH, "%s/%s", log_opts.rootdir,
                                302                 :                :                    timebuf);
                                303         [ -  + ]:             19 :     if (len >= MAXPGPATH)
 1253 tgl@sss.pgh.pa.us         304                 :UBC           0 :         pg_fatal("directory path for new cluster is too long");
                                305                 :                : 
                                306                 :                :     /* BASE_OUTPUTDIR/$timestamp/dump/ */
 1287 michael@paquier.xyz       307                 :CBC          19 :     log_opts.dumpdir = (char *) pg_malloc0(MAXPGPATH);
                                308                 :             19 :     len = snprintf(log_opts.dumpdir, MAXPGPATH, "%s/%s/%s", log_opts.rootdir,
                                309                 :                :                    timebuf, DUMP_OUTPUTDIR);
                                310         [ -  + ]:             19 :     if (len >= MAXPGPATH)
 1253 tgl@sss.pgh.pa.us         311                 :UBC           0 :         pg_fatal("directory path for new cluster is too long");
                                312                 :                : 
                                313                 :                :     /* BASE_OUTPUTDIR/$timestamp/log/ */
 1287 michael@paquier.xyz       314                 :CBC          19 :     log_opts.logdir = (char *) pg_malloc0(MAXPGPATH);
                                315                 :             19 :     len = snprintf(log_opts.logdir, MAXPGPATH, "%s/%s/%s", log_opts.rootdir,
                                316                 :                :                    timebuf, LOG_OUTPUTDIR);
                                317         [ -  + ]:             19 :     if (len >= MAXPGPATH)
 1253 tgl@sss.pgh.pa.us         318                 :UBC           0 :         pg_fatal("directory path for new cluster is too long");
                                319                 :                : 
                                320                 :                :     /*
                                321                 :                :      * Ignore the error case where the root path exists, as it is kept the
                                322                 :                :      * same across runs.
                                323                 :                :      */
 1287 michael@paquier.xyz       324   [ +  +  -  + ]:CBC          19 :     if (mkdir(log_opts.rootdir, pg_dir_create_mode) < 0 && errno != EEXIST)
 1253 tgl@sss.pgh.pa.us         325                 :UBC           0 :         pg_fatal("could not create directory \"%s\": %m", log_opts.rootdir);
 1287 michael@paquier.xyz       326         [ -  + ]:CBC          19 :     if (mkdir(log_opts.basedir, pg_dir_create_mode) < 0)
 1253 tgl@sss.pgh.pa.us         327                 :UBC           0 :         pg_fatal("could not create directory \"%s\": %m", log_opts.basedir);
 1287 michael@paquier.xyz       328         [ -  + ]:CBC          19 :     if (mkdir(log_opts.dumpdir, pg_dir_create_mode) < 0)
 1253 tgl@sss.pgh.pa.us         329                 :UBC           0 :         pg_fatal("could not create directory \"%s\": %m", log_opts.dumpdir);
 1287 michael@paquier.xyz       330         [ -  + ]:CBC          19 :     if (mkdir(log_opts.logdir, pg_dir_create_mode) < 0)
 1253 tgl@sss.pgh.pa.us         331                 :UBC           0 :         pg_fatal("could not create directory \"%s\": %m", log_opts.logdir);
                                332                 :                : 
 1282 tgl@sss.pgh.pa.us         333                 :CBC          19 :     len = snprintf(filename_path, sizeof(filename_path), "%s/%s",
                                334                 :                :                    log_opts.logdir, INTERNAL_LOG_FILE);
                                335         [ -  + ]:             19 :     if (len >= sizeof(filename_path))
 1253 tgl@sss.pgh.pa.us         336                 :UBC           0 :         pg_fatal("directory path for new cluster is too long");
                                337                 :                : 
 1409 michael@paquier.xyz       338         [ -  + ]:CBC          19 :     if ((log_opts.internal = fopen_priv(filename_path, "a")) == NULL)
 1253 tgl@sss.pgh.pa.us         339                 :UBC           0 :         pg_fatal("could not open log file \"%s\": %m", filename_path);
                                340                 :                : 
                                341                 :                :     /* label start of upgrade in logfiles */
 1409 michael@paquier.xyz       342         [ +  + ]:CBC          76 :     for (filename = output_files; *filename != NULL; filename++)
                                343                 :                :     {
 1282 tgl@sss.pgh.pa.us         344                 :             57 :         len = snprintf(filename_path, sizeof(filename_path), "%s/%s",
                                345                 :                :                        log_opts.logdir, *filename);
                                346         [ -  + ]:             57 :         if (len >= sizeof(filename_path))
 1253 tgl@sss.pgh.pa.us         347                 :UBC           0 :             pg_fatal("directory path for new cluster is too long");
 1409 michael@paquier.xyz       348         [ -  + ]:CBC          57 :         if ((fp = fopen_priv(filename_path, "a")) == NULL)
 1253 tgl@sss.pgh.pa.us         349                 :UBC           0 :             pg_fatal("could not write to log file \"%s\": %m", filename_path);
                                350                 :                : 
 1282 tgl@sss.pgh.pa.us         351                 :CBC          57 :         fprintf(fp,
                                352                 :                :                 "-----------------------------------------------------------------\n"
                                353                 :                :                 "  pg_upgrade run on %s"
                                354                 :                :                 "-----------------------------------------------------------------\n\n",
                                355                 :                :                 ctime(&run_time));
 1409 michael@paquier.xyz       356                 :             57 :         fclose(fp);
                                357                 :                :     }
                                358                 :             19 : }
                                359                 :                : 
                                360                 :                : 
                                361                 :                : static void
  508 nathan@postgresql.or      362                 :             19 : setup(char *argv0)
                                363                 :                : {
                                364                 :                :     /*
                                365                 :                :      * make sure the user has a clean environment, otherwise, we may confuse
                                366                 :                :      * libpq when we connect to one (or both) of the servers.
                                367                 :                :      */
 5328 bruce@momjian.us          368                 :             19 :     check_pghost_envvar();
                                369                 :                : 
                                370                 :                :     /*
                                371                 :                :      * In case the user hasn't specified the directory for the new binaries
                                372                 :                :      * with -B, default to using the path of the currently executed pg_upgrade
                                373                 :                :      * binary.
                                374                 :                :      */
 2334 peter@eisentraut.org      375         [ -  + ]:             19 :     if (!new_cluster.bindir)
                                376                 :                :     {
                                377                 :                :         char        exec_path[MAXPGPATH];
                                378                 :                : 
 2334 peter@eisentraut.org      379         [ #  # ]:UBC           0 :         if (find_my_exec(argv0, exec_path) < 0)
 1253 tgl@sss.pgh.pa.us         380                 :              0 :             pg_fatal("%s: could not find own program executable", argv0);
                                381                 :                :         /* Trim off program name and keep just path */
 2334 peter@eisentraut.org      382                 :              0 :         *last_dir_separator(exec_path) = '\0';
                                383                 :              0 :         canonicalize_path(exec_path);
                                384                 :              0 :         new_cluster.bindir = pg_strdup(exec_path);
                                385                 :                :     }
                                386                 :                : 
 5537 bruce@momjian.us          387                 :CBC          19 :     verify_directories();
                                388                 :                : 
                                389                 :                :     /* no postmasters should be running, except for a live check */
 4709                           390         [ -  + ]:             18 :     if (pid_lock_file_exists(old_cluster.pgdata))
                                391                 :                :     {
                                392                 :                :         /*
                                393                 :                :          * If we have a postmaster.pid file, try to start the server.  If it
                                394                 :                :          * starts, the pid file was stale, so stop the server.  If it doesn't
                                395                 :                :          * start, assume the server is running.  If the pid file is left over
                                396                 :                :          * from a server crash, this also allows any committed transactions
                                397                 :                :          * stored in the WAL to be replayed so they are not lost, because WAL
                                398                 :                :          * files are not transferred from old to new servers.  We later check
                                399                 :                :          * for a clean shutdown.
                                400                 :                :          */
 4709 bruce@momjian.us          401         [ #  # ]:UBC           0 :         if (start_postmaster(&old_cluster, false))
                                402                 :              0 :             stop_postmaster(false);
                                403                 :                :         else
                                404                 :                :         {
                                405         [ #  # ]:              0 :             if (!user_opts.check)
 4459 peter_e@gmx.net           406                 :              0 :                 pg_fatal("There seems to be a postmaster servicing the old cluster.\n"
                                407                 :                :                          "Please shutdown that postmaster and try again.");
                                408                 :                :             else
  508 nathan@postgresql.or      409                 :              0 :                 user_opts.live_check = true;
                                410                 :                :         }
                                411                 :                :     }
                                412                 :                : 
                                413                 :                :     /* same goes for the new postmaster */
 4709 bruce@momjian.us          414         [ -  + ]:CBC          18 :     if (pid_lock_file_exists(new_cluster.pgdata))
                                415                 :                :     {
 4709 bruce@momjian.us          416         [ #  # ]:UBC           0 :         if (start_postmaster(&new_cluster, false))
                                417                 :              0 :             stop_postmaster(false);
                                418                 :                :         else
 4459 peter_e@gmx.net           419                 :              0 :             pg_fatal("There seems to be a postmaster servicing the new cluster.\n"
                                420                 :                :                      "Please shutdown that postmaster and try again.");
                                421                 :                :     }
 5697 bruce@momjian.us          422                 :CBC          18 : }
                                423                 :                : 
                                424                 :                : /*
                                425                 :                :  * Set the new cluster's default char signedness using the old cluster's
                                426                 :                :  * value.
                                427                 :                :  */
                                428                 :                : static void
  298 msawada@postgresql.o      429                 :              9 : set_new_cluster_char_signedness(void)
                                430                 :                : {
                                431                 :                :     bool        new_char_signedness;
                                432                 :                : 
                                433                 :                :     /*
                                434                 :                :      * Use the specified char signedness if specified. Otherwise we inherit
                                435                 :                :      * the source database's signedness.
                                436                 :                :      */
                                437         [ -  + ]:              9 :     if (user_opts.char_signedness != -1)
  298 msawada@postgresql.o      438                 :UBC           0 :         new_char_signedness = (user_opts.char_signedness == 1);
                                439                 :                :     else
  298 msawada@postgresql.o      440                 :CBC           9 :         new_char_signedness = old_cluster.controldata.default_char_signedness;
                                441                 :                : 
                                442                 :                :     /* Change the char signedness of the new cluster, if necessary */
                                443         [ +  + ]:              9 :     if (new_cluster.controldata.default_char_signedness != new_char_signedness)
                                444                 :                :     {
                                445                 :              1 :         prep_status("Setting the default char signedness for new cluster");
                                446                 :                : 
                                447         [ -  + ]:              1 :         exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                448                 :                :                   "\"%s/pg_resetwal\" --char-signedness %s \"%s\"",
                                449                 :                :                   new_cluster.bindir,
                                450                 :                :                   new_char_signedness ? "signed" : "unsigned",
                                451                 :                :                   new_cluster.pgdata);
                                452                 :                : 
                                453                 :              1 :         check_ok();
                                454                 :                :     }
                                455                 :              9 : }
                                456                 :                : 
                                457                 :                : /*
                                458                 :                :  * Copy locale and encoding information into the new cluster's template0.
                                459                 :                :  *
                                460                 :                :  * We need to copy the encoding, datlocprovider, datcollate, datctype, and
                                461                 :                :  * datlocale. We don't need datcollversion because that's never set for
                                462                 :                :  * template0.
                                463                 :                :  */
                                464                 :                : static void
 1013 jdavis@postgresql.or      465                 :              9 : set_locale_and_encoding(void)
                                466                 :                : {
                                467                 :                :     PGconn     *conn_new_template1;
                                468                 :                :     char       *datcollate_literal;
                                469                 :                :     char       *datctype_literal;
  647                           470                 :              9 :     char       *datlocale_literal = NULL;
 1013                           471                 :              9 :     DbLocaleInfo *locale = old_cluster.template0;
                                472                 :                : 
                                473                 :              9 :     prep_status("Setting locale and encoding for new cluster");
                                474                 :                : 
                                475                 :                :     /* escape literals with respect to new cluster */
                                476                 :              9 :     conn_new_template1 = connectToServer(&new_cluster, "template1");
                                477                 :                : 
                                478                 :              9 :     datcollate_literal = PQescapeLiteral(conn_new_template1,
                                479                 :              9 :                                          locale->db_collate,
                                480                 :              9 :                                          strlen(locale->db_collate));
                                481                 :              9 :     datctype_literal = PQescapeLiteral(conn_new_template1,
                                482                 :              9 :                                        locale->db_ctype,
                                483                 :              9 :                                        strlen(locale->db_ctype));
                                484                 :                : 
  257                           485         [ +  + ]:              9 :     if (locale->db_locale)
                                486                 :              1 :         datlocale_literal = PQescapeLiteral(conn_new_template1,
                                487                 :              1 :                                             locale->db_locale,
                                488                 :              1 :                                             strlen(locale->db_locale));
                                489                 :                :     else
                                490                 :              8 :         datlocale_literal = "NULL";
                                491                 :                : 
                                492                 :                :     /* update template0 in new cluster */
  647                           493         [ +  - ]:              9 :     if (GET_MAJOR_VERSION(new_cluster.major_version) >= 1700)
                                494                 :              9 :         PQclear(executeQueryOrDie(conn_new_template1,
                                495                 :                :                                   "UPDATE pg_catalog.pg_database "
                                496                 :                :                                   "  SET encoding = %d, "
                                497                 :                :                                   "      datlocprovider = '%c', "
                                498                 :                :                                   "      datcollate = %s, "
                                499                 :                :                                   "      datctype = %s, "
                                500                 :                :                                   "      datlocale = %s "
                                501                 :                :                                   "  WHERE datname = 'template0' ",
                                502                 :                :                                   locale->db_encoding,
                                503                 :              9 :                                   locale->db_collprovider,
                                504                 :                :                                   datcollate_literal,
                                505                 :                :                                   datctype_literal,
                                506                 :                :                                   datlocale_literal));
  647 jdavis@postgresql.or      507         [ #  # ]:UBC           0 :     else if (GET_MAJOR_VERSION(new_cluster.major_version) >= 1500)
 1013                           508                 :              0 :         PQclear(executeQueryOrDie(conn_new_template1,
                                509                 :                :                                   "UPDATE pg_catalog.pg_database "
                                510                 :                :                                   "  SET encoding = %d, "
                                511                 :                :                                   "      datlocprovider = '%c', "
                                512                 :                :                                   "      datcollate = %s, "
                                513                 :                :                                   "      datctype = %s, "
                                514                 :                :                                   "      daticulocale = %s "
                                515                 :                :                                   "  WHERE datname = 'template0' ",
                                516                 :                :                                   locale->db_encoding,
                                517                 :              0 :                                   locale->db_collprovider,
                                518                 :                :                                   datcollate_literal,
                                519                 :                :                                   datctype_literal,
                                520                 :                :                                   datlocale_literal));
                                521                 :                :     else
                                522                 :              0 :         PQclear(executeQueryOrDie(conn_new_template1,
                                523                 :                :                                   "UPDATE pg_catalog.pg_database "
                                524                 :                :                                   "  SET encoding = %d, "
                                525                 :                :                                   "      datcollate = %s, "
                                526                 :                :                                   "      datctype = %s "
                                527                 :                :                                   "  WHERE datname = 'template0' ",
                                528                 :                :                                   locale->db_encoding,
                                529                 :                :                                   datcollate_literal,
                                530                 :                :                                   datctype_literal));
                                531                 :                : 
 1013 jdavis@postgresql.or      532                 :CBC           9 :     PQfreemem(datcollate_literal);
                                533                 :              9 :     PQfreemem(datctype_literal);
  257                           534         [ +  + ]:              9 :     if (locale->db_locale)
                                535                 :              1 :         PQfreemem(datlocale_literal);
                                536                 :                : 
 1013                           537                 :              9 :     PQfinish(conn_new_template1);
                                538                 :                : 
                                539                 :              9 :     check_ok();
                                540                 :              9 : }
                                541                 :                : 
                                542                 :                : 
                                543                 :                : static void
 5537 bruce@momjian.us          544                 :              9 : prepare_new_cluster(void)
                                545                 :                : {
                                546                 :                :     /*
                                547                 :                :      * It would make more sense to freeze after loading the schema, but that
                                548                 :                :      * would cause us to lose the frozenxids restored by the load. We use
                                549                 :                :      * --analyze so autovacuum doesn't update statistics later
                                550                 :                :      */
                                551                 :              9 :     prep_status("Analyzing all rows in the new cluster");
 2899                           552                 :              9 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                553                 :                :               "\"%s/vacuumdb\" %s --all --analyze %s",
                                554                 :                :               new_cluster.bindir, cluster_conn_opts(&new_cluster),
 4859 alvherre@alvh.no-ip.      555         [ -  + ]:              9 :               log_opts.verbose ? "--verbose" : "");
 5537 bruce@momjian.us          556                 :              9 :     check_ok();
                                557                 :                : 
                                558                 :                :     /*
                                559                 :                :      * We do freeze after analyze so pg_statistic is also frozen. template0 is
                                560                 :                :      * not frozen here, but data rows were frozen by initdb, and we set its
                                561                 :                :      * datfrozenxid, relfrozenxids, and relminmxid later to match the new xid
                                562                 :                :      * counter later.
                                563                 :                :      */
 3020 peter_e@gmx.net           564                 :              9 :     prep_status("Freezing all rows in the new cluster");
 2899 bruce@momjian.us          565                 :              9 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                566                 :                :               "\"%s/vacuumdb\" %s --all --freeze %s",
                                567                 :                :               new_cluster.bindir, cluster_conn_opts(&new_cluster),
 4859 alvherre@alvh.no-ip.      568         [ -  + ]:              9 :               log_opts.verbose ? "--verbose" : "");
 5537 bruce@momjian.us          569                 :              9 :     check_ok();
 5697                           570                 :              9 : }
                                571                 :                : 
                                572                 :                : 
                                573                 :                : static void
 2885 tgl@sss.pgh.pa.us         574                 :              9 : prepare_new_globals(void)
                                575                 :                : {
                                576                 :                :     /*
                                577                 :                :      * Before we restore anything, set frozenxids of initdb-created tables.
                                578                 :                :      */
 4185 bruce@momjian.us          579                 :              9 :     set_frozenxids(false);
                                580                 :                : 
                                581                 :                :     /*
                                582                 :                :      * Now restore global objects (roles and tablespaces).
                                583                 :                :      */
 4764                           584                 :              9 :     prep_status("Restoring global objects in the new cluster");
                                585                 :                : 
 2899                           586                 :              9 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                587                 :                :               "\"%s/psql\" " EXEC_PSQL_ARGS " %s -f \"%s/%s\"",
                                588                 :                :               new_cluster.bindir, cluster_conn_opts(&new_cluster),
                                589                 :                :               log_opts.dumpdir,
                                590                 :                :               GLOBALS_DUMP_FILE);
 5537                           591                 :              9 :     check_ok();
 5697                           592                 :              9 : }
                                593                 :                : 
                                594                 :                : 
                                595                 :                : static void
 5537                           596                 :              9 : create_new_objects(void)
                                597                 :                : {
                                598                 :                :     int         dbnum;
                                599                 :                :     PGconn     *conn_new_template1;
                                600                 :                : 
 1394 andres@anarazel.de        601                 :              9 :     prep_status_progress("Restoring database schemas in the new cluster");
                                602                 :                : 
                                603                 :                :     /*
                                604                 :                :      * Ensure that any changes to template0 are fully written out to disk
                                605                 :                :      * prior to restoring the databases.  This is necessary because we use the
                                606                 :                :      * FILE_COPY strategy to create the databases (which testing has shown to
                                607                 :                :      * be faster), and when the server is in binary upgrade mode, it skips the
                                608                 :                :      * checkpoints this strategy ordinarily performs.
                                609                 :                :      */
  526 nathan@postgresql.or      610                 :              9 :     conn_new_template1 = connectToServer(&new_cluster, "template1");
                                611                 :              9 :     PQclear(executeQueryOrDie(conn_new_template1, "CHECKPOINT"));
                                612                 :              9 :     PQfinish(conn_new_template1);
                                613                 :                : 
                                614                 :                :     /*
                                615                 :                :      * We cannot process the template1 database concurrently with others,
                                616                 :                :      * because when it's transiently dropped, connection attempts would fail.
                                617                 :                :      * So handle it in a separate non-parallelized pass.
                                618                 :                :      */
 4764 bruce@momjian.us          619         [ +  - ]:              9 :     for (dbnum = 0; dbnum < old_cluster.dbarr.ndbs; dbnum++)
                                620                 :                :     {
                                621                 :                :         char        sql_file_name[MAXPGPATH],
                                622                 :                :                     log_file_name[MAXPGPATH];
 4584                           623                 :              9 :         DbInfo     *old_db = &old_cluster.dbarr.dbs[dbnum];
                                624                 :                :         const char *create_opts;
                                625                 :                : 
                                626                 :                :         /* Process only template1 in this pass */
 2851 tgl@sss.pgh.pa.us         627         [ -  + ]:              9 :         if (strcmp(old_db->db_name, "template1") != 0)
 2851 tgl@sss.pgh.pa.us         628                 :UBC           0 :             continue;
                                629                 :                : 
 4757 bruce@momjian.us          630                 :CBC           9 :         pg_log(PG_STATUS, "%s", old_db->db_name);
 4738                           631                 :              9 :         snprintf(sql_file_name, sizeof(sql_file_name), DB_DUMP_FILE_MASK, old_db->db_oid);
                                632                 :              9 :         snprintf(log_file_name, sizeof(log_file_name), DB_DUMP_LOG_FILE_MASK, old_db->db_oid);
                                633                 :                : 
                                634                 :                :         /*
                                635                 :                :          * template1 database will already exist in the target installation,
                                636                 :                :          * so tell pg_restore to drop and recreate it; otherwise we would fail
                                637                 :                :          * to propagate its database-level properties.
                                638                 :                :          */
 2851 tgl@sss.pgh.pa.us         639                 :              9 :         create_opts = "--clean --create";
                                640                 :                : 
                                641                 :              9 :         exec_prog(log_file_name,
                                642                 :                :                   NULL,
                                643                 :                :                   true,
                                644                 :                :                   true,
                                645                 :                :                   "\"%s/pg_restore\" %s %s --exit-on-error --verbose "
                                646                 :                :                   "--transaction-size=%d "
                                647                 :                :                   "--dbname postgres \"%s/%s\"",
                                648                 :                :                   new_cluster.bindir,
                                649                 :                :                   cluster_conn_opts(&new_cluster),
                                650                 :                :                   create_opts,
                                651                 :                :                   RESTORE_TRANSACTION_SIZE,
                                652                 :                :                   log_opts.dumpdir,
                                653                 :                :                   sql_file_name);
                                654                 :                : 
                                655                 :              9 :         break;                  /* done once we've processed template1 */
                                656                 :                :     }
                                657                 :                : 
                                658         [ +  + ]:             39 :     for (dbnum = 0; dbnum < old_cluster.dbarr.ndbs; dbnum++)
                                659                 :                :     {
                                660                 :                :         char        sql_file_name[MAXPGPATH],
                                661                 :                :                     log_file_name[MAXPGPATH];
                                662                 :             30 :         DbInfo     *old_db = &old_cluster.dbarr.dbs[dbnum];
                                663                 :                :         const char *create_opts;
                                664                 :                :         int         txn_size;
                                665                 :                : 
                                666                 :                :         /* Skip template1 in this pass */
 2885                           667         [ +  + ]:             30 :         if (strcmp(old_db->db_name, "template1") == 0)
 2851                           668                 :              9 :             continue;
                                669                 :                : 
                                670                 :             21 :         pg_log(PG_STATUS, "%s", old_db->db_name);
                                671                 :             21 :         snprintf(sql_file_name, sizeof(sql_file_name), DB_DUMP_FILE_MASK, old_db->db_oid);
                                672                 :             21 :         snprintf(log_file_name, sizeof(log_file_name), DB_DUMP_LOG_FILE_MASK, old_db->db_oid);
                                673                 :                : 
                                674                 :                :         /*
                                675                 :                :          * postgres database will already exist in the target installation, so
                                676                 :                :          * tell pg_restore to drop and recreate it; otherwise we would fail to
                                677                 :                :          * propagate its database-level properties.
                                678                 :                :          */
                                679         [ +  + ]:             21 :         if (strcmp(old_db->db_name, "postgres") == 0)
                                680                 :              9 :             create_opts = "--clean --create";
                                681                 :                :         else
                                682                 :             12 :             create_opts = "--create";
                                683                 :                : 
                                684                 :                :         /*
                                685                 :                :          * In parallel mode, reduce the --transaction-size of each restore job
                                686                 :                :          * so that the total number of locks that could be held across all the
                                687                 :                :          * jobs stays in bounds.
                                688                 :                :          */
  624                           689                 :             21 :         txn_size = RESTORE_TRANSACTION_SIZE;
                                690         [ -  + ]:             21 :         if (user_opts.jobs > 1)
                                691                 :                :         {
  624 tgl@sss.pgh.pa.us         692                 :UBC           0 :             txn_size /= user_opts.jobs;
                                693                 :                :             /* Keep some sanity if -j is huge */
                                694                 :              0 :             txn_size = Max(txn_size, 10);
                                695                 :                :         }
                                696                 :                : 
 4581 sfrost@snowman.net        697                 :CBC          21 :         parallel_exec_prog(log_file_name,
                                698                 :                :                            NULL,
                                699                 :                :                            "\"%s/pg_restore\" %s %s --exit-on-error --verbose "
                                700                 :                :                            "--transaction-size=%d "
                                701                 :                :                            "--dbname template1 \"%s/%s\"",
                                702                 :                :                            new_cluster.bindir,
                                703                 :                :                            cluster_conn_opts(&new_cluster),
                                704                 :                :                            create_opts,
                                705                 :                :                            txn_size,
                                706                 :                :                            log_opts.dumpdir,
                                707                 :                :                            sql_file_name);
                                708                 :                :     }
                                709                 :                : 
                                710                 :                :     /* reap all children */
 4738 bruce@momjian.us          711         [ -  + ]:              9 :     while (reap_child(true) == true)
                                712                 :                :         ;
                                713                 :                : 
 4764                           714                 :              9 :     end_progress_output();
 5537                           715                 :              9 :     check_ok();
                                716                 :                : 
                                717                 :                :     /*
                                718                 :                :      * We don't have minmxids for databases or relations in pre-9.3 clusters,
                                719                 :                :      * so set those after we have restored the schema.
                                720                 :                :      */
 1897                           721         [ -  + ]:              9 :     if (GET_MAJOR_VERSION(old_cluster.major_version) <= 902)
 4185 bruce@momjian.us          722                 :UBC           0 :         set_frozenxids(true);
                                723                 :                : 
                                724                 :                :     /* update new_cluster info now that we have objects in the databases */
  508 nathan@postgresql.or      725                 :CBC           9 :     get_db_rel_and_slot_infos(&new_cluster);
 5697 bruce@momjian.us          726                 :              9 : }
                                727                 :                : 
                                728                 :                : /*
                                729                 :                :  * Delete the given subdirectory contents from the new cluster
                                730                 :                :  */
                                731                 :                : static void
 2968 peter_e@gmx.net           732                 :             27 : remove_new_subdir(const char *subdir, bool rmtopdir)
                                733                 :                : {
                                734                 :                :     char        new_path[MAXPGPATH];
                                735                 :                : 
 4912 alvherre@alvh.no-ip.      736                 :             27 :     prep_status("Deleting files from new %s", subdir);
                                737                 :                : 
                                738                 :             27 :     snprintf(new_path, sizeof(new_path), "%s/%s", new_cluster.pgdata, subdir);
 4193 bruce@momjian.us          739         [ -  + ]:             27 :     if (!rmtree(new_path, rmtopdir))
 1253 tgl@sss.pgh.pa.us         740                 :UBC           0 :         pg_fatal("could not delete directory \"%s\"", new_path);
                                741                 :                : 
 5537 bruce@momjian.us          742                 :CBC          27 :     check_ok();
 4193                           743                 :             27 : }
                                744                 :                : 
                                745                 :                : /*
                                746                 :                :  * Copy the files from the old cluster into it
                                747                 :                :  */
                                748                 :                : static void
 2968 peter_e@gmx.net           749                 :             27 : copy_subdir_files(const char *old_subdir, const char *new_subdir)
                                750                 :                : {
                                751                 :                :     char        old_path[MAXPGPATH];
                                752                 :                :     char        new_path[MAXPGPATH];
                                753                 :                : 
 3196 rhaas@postgresql.org      754                 :             27 :     remove_new_subdir(new_subdir, true);
                                755                 :                : 
                                756                 :             27 :     snprintf(old_path, sizeof(old_path), "%s/%s", old_cluster.pgdata, old_subdir);
                                757                 :             27 :     snprintf(new_path, sizeof(new_path), "%s/%s", new_cluster.pgdata, new_subdir);
                                758                 :                : 
                                759                 :             27 :     prep_status("Copying old %s to new server", old_subdir);
                                760                 :                : 
 2899 bruce@momjian.us          761                 :             27 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                762                 :                : #ifndef WIN32
                                763                 :                :               "cp -Rf \"%s\" \"%s\"",
                                764                 :                : #else
                                765                 :                :     /* flags: everything, no confirm, quiet, overwrite read-only */
                                766                 :                :               "xcopy /e /y /q /r \"%s\" \"%s\\\"",
                                767                 :                : #endif
                                768                 :                :               old_path, new_path);
                                769                 :                : 
 5537                           770                 :             27 :     check_ok();
 4912 alvherre@alvh.no-ip.      771                 :             27 : }
                                772                 :                : 
                                773                 :                : static void
 3196 rhaas@postgresql.org      774                 :              9 : copy_xact_xlog_xid(void)
                                775                 :                : {
                                776                 :                :     /*
                                777                 :                :      * Copy old commit logs to new data dir. pg_clog has been renamed to
                                778                 :                :      * pg_xact in post-10 clusters.
                                779                 :                :      */
 1897 bruce@momjian.us          780         [ -  + ]:              9 :     copy_subdir_files(GET_MAJOR_VERSION(old_cluster.major_version) <= 906 ?
                                781                 :                :                       "pg_clog" : "pg_xact",
                                782         [ -  + ]:              9 :                       GET_MAJOR_VERSION(new_cluster.major_version) <= 906 ?
                                783                 :                :                       "pg_clog" : "pg_xact");
                                784                 :                : 
 1604                           785                 :              9 :     prep_status("Setting oldest XID for new cluster");
                                786                 :              9 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                787                 :                :               "\"%s/pg_resetwal\" -f -u %u \"%s\"",
                                788                 :                :               new_cluster.bindir, old_cluster.controldata.chkpnt_oldstxid,
                                789                 :                :               new_cluster.pgdata);
                                790                 :              9 :     check_ok();
                                791                 :                : 
                                792                 :                :     /* set the next transaction id and epoch of the new cluster */
 4120                           793                 :              9 :     prep_status("Setting next transaction ID and epoch for new cluster");
 2899                           794                 :              9 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                795                 :                :               "\"%s/pg_resetwal\" -f -x %u \"%s\"",
                                796                 :                :               new_cluster.bindir, old_cluster.controldata.chkpnt_nxtxid,
                                797                 :                :               new_cluster.pgdata);
                                798                 :              9 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                799                 :                :               "\"%s/pg_resetwal\" -f -e %u \"%s\"",
                                800                 :                :               new_cluster.bindir, old_cluster.controldata.chkpnt_nxtepoch,
                                801                 :                :               new_cluster.pgdata);
                                802                 :                :     /* must reset commit timestamp limits also */
                                803                 :              9 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                804                 :                :               "\"%s/pg_resetwal\" -f -c %u,%u \"%s\"",
                                805                 :                :               new_cluster.bindir,
                                806                 :                :               old_cluster.controldata.chkpnt_nxtxid,
                                807                 :                :               old_cluster.controldata.chkpnt_nxtxid,
                                808                 :                :               new_cluster.pgdata);
 5537                           809                 :              9 :     check_ok();
                                810                 :                : 
                                811                 :                :     /* Copy or convert pg_multixact files */
    7 heikki.linnakangas@i      812         [ -  + ]:GNC           9 :     Assert(new_cluster.controldata.cat_ver >= MULTIXACT_FORMATCHANGE_CAT_VER);
                                813         [ -  + ]:              9 :     Assert(new_cluster.controldata.cat_ver >= MULTIXACTOFFSET_FORMATCHANGE_CAT_VER);
                                814         [ +  - ]:              9 :     if (old_cluster.controldata.cat_ver >= MULTIXACTOFFSET_FORMATCHANGE_CAT_VER)
                                815                 :                :     {
                                816                 :                :         /* No change in multixact format, just copy the files */
                                817                 :              9 :         MultiXactId new_nxtmulti = old_cluster.controldata.chkpnt_nxtmulti;
                                818                 :              9 :         MultiXactOffset new_nxtmxoff = old_cluster.controldata.chkpnt_nxtmxoff;
                                819                 :                : 
 3196 rhaas@postgresql.org      820                 :CBC           9 :         copy_subdir_files("pg_multixact/offsets", "pg_multixact/offsets");
                                821                 :              9 :         copy_subdir_files("pg_multixact/members", "pg_multixact/members");
                                822                 :                : 
 4710 alvherre@alvh.no-ip.      823                 :              9 :         prep_status("Setting next multixact ID and offset for new cluster");
                                824                 :                : 
                                825                 :                :         /*
                                826                 :                :          * we preserve all files and contents, so we must preserve both "next"
                                827                 :                :          * counters here and the oldest multi present on system.
                                828                 :                :          */
 2899 bruce@momjian.us          829                 :              9 :         exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                830                 :                :                   "\"%s/pg_resetwal\" -O %" PRIu64 " -m %u,%u \"%s\"",
                                831                 :                :                   new_cluster.bindir, new_nxtmxoff, new_nxtmulti,
                                832                 :                :                   old_cluster.controldata.chkpnt_oldstMulti,
                                833                 :                :                   new_cluster.pgdata);
 4710 alvherre@alvh.no-ip.      834                 :              9 :         check_ok();
                                835                 :                :     }
                                836                 :                :     else
                                837                 :                :     {
                                838                 :                :         /* Conversion is needed */
                                839                 :                :         MultiXactId nxtmulti;
                                840                 :                :         MultiXactId oldstMulti;
                                841                 :                :         MultiXactOffset nxtmxoff;
                                842                 :                : 
                                843                 :                :         /*
                                844                 :                :          * Determine the range of multixacts to convert.
                                845                 :                :          */
    7 heikki.linnakangas@i      846                 :UNC           0 :         nxtmulti = old_cluster.controldata.chkpnt_nxtmulti;
                                847         [ #  # ]:              0 :         if (old_cluster.controldata.cat_ver >= MULTIXACT_FORMATCHANGE_CAT_VER)
                                848                 :                :         {
                                849                 :                :             /* Versions 9.3 - 18: convert all multixids  */
                                850                 :              0 :             oldstMulti = old_cluster.controldata.chkpnt_oldstMulti;
                                851                 :                :         }
                                852                 :                :         else
                                853                 :                :         {
                                854                 :                :             /*
                                855                 :                :              * In PostgreSQL 9.2 and below, multitransactions were only used
                                856                 :                :              * for row locking, and as such don't need to be preserved during
                                857                 :                :              * upgrade.  In that case, we utilize rewrite_multixacts() just to
                                858                 :                :              * initialize new, empty files in the new format.
                                859                 :                :              *
                                860                 :                :              * It's important that the oldest multi is set to the latest value
                                861                 :                :              * used by the old system, so that multixact.c returns the empty
                                862                 :                :              * set for multis that might be present on disk.
                                863                 :                :              */
                                864                 :              0 :             oldstMulti = nxtmulti;
                                865                 :                :         }
                                866                 :                :         /* handle wraparound */
                                867         [ #  # ]:              0 :         if (nxtmulti < FirstMultiXactId)
                                868                 :              0 :             nxtmulti = FirstMultiXactId;
                                869         [ #  # ]:              0 :         if (oldstMulti < FirstMultiXactId)
                                870                 :              0 :             oldstMulti = FirstMultiXactId;
                                871                 :                : 
                                872                 :                :         /*
                                873                 :                :          * Remove the files created by initdb in the new cluster.
                                874                 :                :          * rewrite_multixacts() will create new ones.
                                875                 :                :          */
                                876                 :              0 :         remove_new_subdir("pg_multixact/members", false);
                                877                 :              0 :         remove_new_subdir("pg_multixact/offsets", false);
                                878                 :                : 
                                879                 :                :         /*
                                880                 :                :          * Create new pg_multixact files, converting old ones if needed.
                                881                 :                :          */
                                882                 :              0 :         prep_status("Converting pg_multixact files");
                                883                 :              0 :         nxtmxoff = rewrite_multixacts(oldstMulti, nxtmulti);
                                884                 :              0 :         check_ok();
                                885                 :                : 
                                886                 :              0 :         prep_status("Setting next multixact ID and offset for new cluster");
 2899 bruce@momjian.us          887                 :UBC           0 :         exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                888                 :                :                   "\"%s/pg_resetwal\" -O %" PRIu64 " -m %u,%u \"%s\"",
                                889                 :                :                   new_cluster.bindir,
                                890                 :                :                   nxtmxoff, nxtmulti, oldstMulti,
                                891                 :                :                   new_cluster.pgdata);
 4710 alvherre@alvh.no-ip.      892                 :              0 :         check_ok();
                                893                 :                :     }
                                894                 :                : 
                                895                 :                :     /* now reset the wal archives in the new cluster */
 5537 bruce@momjian.us          896                 :CBC           9 :     prep_status("Resetting WAL archives");
 2899                           897                 :              9 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
                                898                 :                :     /* use timeline 1 to match controldata and no WAL history file */
                                899                 :                :               "\"%s/pg_resetwal\" -l 00000001%s \"%s\"", new_cluster.bindir,
                                900                 :                :               old_cluster.controldata.nextxlogfile + 8,
                                901                 :                :               new_cluster.pgdata);
 5537                           902                 :              9 :     check_ok();
 5697                           903                 :              9 : }
                                904                 :                : 
                                905                 :                : 
                                906                 :                : /*
                                907                 :                :  *  set_frozenxids()
                                908                 :                :  *
                                909                 :                :  * This is called on the new cluster before we restore anything, with
                                910                 :                :  * minmxid_only = false.  Its purpose is to ensure that all initdb-created
                                911                 :                :  * vacuumable tables have relfrozenxid/relminmxid matching the old cluster's
                                912                 :                :  * xid/mxid counters.  We also initialize the datfrozenxid/datminmxid of the
                                913                 :                :  * built-in databases to match.
                                914                 :                :  *
                                915                 :                :  * As we create user tables later, their relfrozenxid/relminmxid fields will
                                916                 :                :  * be restored properly by the binary-upgrade restore script.  Likewise for
                                917                 :                :  * user-database datfrozenxid/datminmxid.  However, if we're upgrading from a
                                918                 :                :  * pre-9.3 database, which does not store per-table or per-DB minmxid, then
                                919                 :                :  * the relminmxid/datminmxid values filled in by the restore script will just
                                920                 :                :  * be zeroes.
                                921                 :                :  *
                                922                 :                :  * Hence, with a pre-9.3 source database, a second call occurs after
                                923                 :                :  * everything is restored, with minmxid_only = true.  This pass will
                                924                 :                :  * initialize all tables and databases, both those made by initdb and user
                                925                 :                :  * objects, with the desired minmxid value.  frozenxid values are left alone.
                                926                 :                :  */
                                927                 :                : static void
 4185                           928                 :              9 : set_frozenxids(bool minmxid_only)
                                929                 :                : {
                                930                 :                :     int         dbnum;
                                931                 :                :     PGconn     *conn,
                                932                 :                :                *conn_template1;
                                933                 :                :     PGresult   *dbres;
                                934                 :                :     int         ntups;
                                935                 :                :     int         i_datname;
                                936                 :                :     int         i_datallowconn;
                                937                 :                : 
                                938         [ +  - ]:              9 :     if (!minmxid_only)
                                939                 :              9 :         prep_status("Setting frozenxid and minmxid counters in new cluster");
                                940                 :                :     else
 4185 bruce@momjian.us          941                 :UBC           0 :         prep_status("Setting minmxid counter in new cluster");
                                942                 :                : 
 5463 bruce@momjian.us          943                 :CBC           9 :     conn_template1 = connectToServer(&new_cluster, "template1");
                                944                 :                : 
 4185                           945         [ +  - ]:              9 :     if (!minmxid_only)
                                946                 :                :         /* set pg_database.datfrozenxid */
                                947                 :              9 :         PQclear(executeQueryOrDie(conn_template1,
                                948                 :                :                                   "UPDATE pg_catalog.pg_database "
                                949                 :                :                                   "SET datfrozenxid = '%u'",
                                950                 :                :                                   old_cluster.controldata.chkpnt_nxtxid));
                                951                 :                : 
                                952                 :                :     /* set pg_database.datminmxid */
 5537                           953                 :              9 :     PQclear(executeQueryOrDie(conn_template1,
                                954                 :                :                               "UPDATE pg_catalog.pg_database "
                                955                 :                :                               "SET datminmxid = '%u'",
                                956                 :                :                               old_cluster.controldata.chkpnt_nxtmulti));
                                957                 :                : 
                                958                 :                :     /* get database names */
                                959                 :              9 :     dbres = executeQueryOrDie(conn_template1,
                                960                 :                :                               "SELECT  datname, datallowconn "
                                961                 :                :                               "FROM    pg_catalog.pg_database");
                                962                 :                : 
 5690                           963                 :              9 :     i_datname = PQfnumber(dbres, "datname");
                                964                 :              9 :     i_datallowconn = PQfnumber(dbres, "datallowconn");
                                965                 :                : 
 5697                           966                 :              9 :     ntups = PQntuples(dbres);
                                967         [ +  + ]:             36 :     for (dbnum = 0; dbnum < ntups; dbnum++)
                                968                 :                :     {
 5642                           969                 :             27 :         char       *datname = PQgetvalue(dbres, dbnum, i_datname);
                                970                 :             27 :         char       *datallowconn = PQgetvalue(dbres, dbnum, i_datallowconn);
                                971                 :                : 
                                972                 :                :         /*
                                973                 :                :          * We must update databases where datallowconn = false, e.g.
                                974                 :                :          * template0, because autovacuum increments their datfrozenxids,
                                975                 :                :          * relfrozenxids, and relminmxid even if autovacuum is turned off, and
                                976                 :                :          * even though all the data rows are already frozen.  To enable this,
                                977                 :                :          * we temporarily change datallowconn.
                                978                 :                :          */
 5690                           979         [ +  + ]:             27 :         if (strcmp(datallowconn, "f") == 0)
 5537                           980                 :              9 :             PQclear(executeQueryOrDie(conn_template1,
                                981                 :                :                                       "ALTER DATABASE %s ALLOW_CONNECTIONS = true",
                                982                 :                :                                       quote_identifier(datname)));
                                983                 :                : 
 5463                           984                 :             27 :         conn = connectToServer(&new_cluster, datname);
                                985                 :                : 
 4185                           986         [ +  - ]:             27 :         if (!minmxid_only)
                                987                 :                :             /* set pg_class.relfrozenxid */
                                988                 :             27 :             PQclear(executeQueryOrDie(conn,
                                989                 :                :                                       "UPDATE  pg_catalog.pg_class "
                                990                 :                :                                       "SET relfrozenxid = '%u' "
                                991                 :                :             /* only heap, materialized view, and TOAST are vacuumed */
                                992                 :                :                                       "WHERE   relkind IN ("
                                993                 :                :                                       CppAsString2(RELKIND_RELATION) ", "
                                994                 :                :                                       CppAsString2(RELKIND_MATVIEW) ", "
                                995                 :                :                                       CppAsString2(RELKIND_TOASTVALUE) ")",
                                996                 :                :                                       old_cluster.controldata.chkpnt_nxtxid));
                                997                 :                : 
                                998                 :                :         /* set pg_class.relminmxid */
 5537                           999                 :             27 :         PQclear(executeQueryOrDie(conn,
                               1000                 :                :                                   "UPDATE  pg_catalog.pg_class "
                               1001                 :                :                                   "SET relminmxid = '%u' "
                               1002                 :                :         /* only heap, materialized view, and TOAST are vacuumed */
                               1003                 :                :                                   "WHERE   relkind IN ("
                               1004                 :                :                                   CppAsString2(RELKIND_RELATION) ", "
                               1005                 :                :                                   CppAsString2(RELKIND_MATVIEW) ", "
                               1006                 :                :                                   CppAsString2(RELKIND_TOASTVALUE) ")",
                               1007                 :                :                                   old_cluster.controldata.chkpnt_nxtmulti));
 5697                          1008                 :             27 :         PQfinish(conn);
                               1009                 :                : 
                               1010                 :                :         /* Reset datallowconn flag */
 5690                          1011         [ +  + ]:             27 :         if (strcmp(datallowconn, "f") == 0)
 5537                          1012                 :              9 :             PQclear(executeQueryOrDie(conn_template1,
                               1013                 :                :                                       "ALTER DATABASE %s ALLOW_CONNECTIONS = false",
                               1014                 :                :                                       quote_identifier(datname)));
                               1015                 :                :     }
                               1016                 :                : 
 5697                          1017                 :              9 :     PQclear(dbres);
                               1018                 :                : 
 5690                          1019                 :              9 :     PQfinish(conn_template1);
                               1020                 :                : 
 5537                          1021                 :              9 :     check_ok();
 5697                          1022                 :              9 : }
                               1023                 :                : 
                               1024                 :                : /*
                               1025                 :                :  * create_logical_replication_slots()
                               1026                 :                :  *
                               1027                 :                :  * Similar to create_new_objects() but only restores logical replication slots.
                               1028                 :                :  */
                               1029                 :                : static void
  782 akapila@postgresql.o     1030                 :              1 : create_logical_replication_slots(void)
                               1031                 :                : {
                               1032                 :              1 :     prep_status_progress("Restoring logical replication slots in the new cluster");
                               1033                 :                : 
                               1034         [ +  + ]:              3 :     for (int dbnum = 0; dbnum < old_cluster.dbarr.ndbs; dbnum++)
                               1035                 :                :     {
                               1036                 :              2 :         DbInfo     *old_db = &old_cluster.dbarr.dbs[dbnum];
                               1037                 :              2 :         LogicalSlotInfoArr *slot_arr = &old_db->slot_arr;
                               1038                 :                :         PGconn     *conn;
                               1039                 :                :         PQExpBuffer query;
                               1040                 :                : 
                               1041                 :                :         /* Skip this database if there are no slots */
                               1042         [ +  + ]:              2 :         if (slot_arr->nslots == 0)
                               1043                 :              1 :             continue;
                               1044                 :                : 
                               1045                 :              1 :         conn = connectToServer(&new_cluster, old_db->db_name);
                               1046                 :              1 :         query = createPQExpBuffer();
                               1047                 :                : 
                               1048                 :              1 :         pg_log(PG_STATUS, "%s", old_db->db_name);
                               1049                 :                : 
                               1050         [ +  + ]:              2 :         for (int slotnum = 0; slotnum < slot_arr->nslots; slotnum++)
                               1051                 :                :         {
                               1052                 :              1 :             LogicalSlotInfo *slot_info = &slot_arr->slots[slotnum];
                               1053                 :                : 
                               1054                 :                :             /* Constructs a query for creating logical replication slots */
  242 drowley@postgresql.o     1055                 :              1 :             appendPQExpBufferStr(query,
                               1056                 :                :                                  "SELECT * FROM "
                               1057                 :                :                                  "pg_catalog.pg_create_logical_replication_slot(");
  782 akapila@postgresql.o     1058                 :              1 :             appendStringLiteralConn(query, slot_info->slotname, conn);
  242 drowley@postgresql.o     1059                 :              1 :             appendPQExpBufferStr(query, ", ");
  782 akapila@postgresql.o     1060                 :              1 :             appendStringLiteralConn(query, slot_info->plugin, conn);
                               1061                 :                : 
  691                          1062                 :              2 :             appendPQExpBuffer(query, ", false, %s, %s);",
                               1063         [ +  - ]:              1 :                               slot_info->two_phase ? "true" : "false",
                               1064         [ +  - ]:              1 :                               slot_info->failover ? "true" : "false");
                               1065                 :                : 
  782                          1066                 :              1 :             PQclear(executeQueryOrDie(conn, "%s", query->data));
                               1067                 :                : 
                               1068                 :              1 :             resetPQExpBuffer(query);
                               1069                 :                :         }
                               1070                 :                : 
                               1071                 :              1 :         PQfinish(conn);
                               1072                 :                : 
                               1073                 :              1 :         destroyPQExpBuffer(query);
                               1074                 :                :     }
                               1075                 :                : 
                               1076                 :              1 :     end_progress_output();
                               1077                 :              1 :     check_ok();
                               1078                 :                : 
                               1079                 :              1 :     return;
                               1080                 :                : }
                               1081                 :                : 
                               1082                 :                : /*
                               1083                 :                :  * create_conflict_detection_slot()
                               1084                 :                :  *
                               1085                 :                :  * Create a replication slot to retain information necessary for conflict
                               1086                 :                :  * detection such as dead tuples, commit timestamps, and origins, for migrated
                               1087                 :                :  * subscriptions with retain_dead_tuples enabled.
                               1088                 :                :  */
                               1089                 :                : static void
  146 akapila@postgresql.o     1090                 :GNC           1 : create_conflict_detection_slot(void)
                               1091                 :                : {
                               1092                 :                :     PGconn     *conn_new_template1;
                               1093                 :                : 
                               1094                 :              1 :     prep_status("Creating the replication conflict detection slot");
                               1095                 :                : 
                               1096                 :              1 :     conn_new_template1 = connectToServer(&new_cluster, "template1");
                               1097                 :              1 :     PQclear(executeQueryOrDie(conn_new_template1, "SELECT pg_catalog.binary_upgrade_create_conflict_detection_slot()"));
                               1098                 :              1 :     PQfinish(conn_new_template1);
                               1099                 :                : 
                               1100                 :              1 :     check_ok();
                               1101                 :              1 : }
        

Generated by: LCOV version 2.4-beta