LCOV - code coverage report
Current view: top level - src/bin/pg_upgrade - pg_upgrade.c (source / functions) Hit Total Coverage
Test: PostgreSQL 15devel Lines: 178 199 89.4 %
Date: 2021-12-09 04:09:06 Functions: 10 10 100.0 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : /*
       2             :  *  pg_upgrade.c
       3             :  *
       4             :  *  main source file
       5             :  *
       6             :  *  Copyright (c) 2010-2021, 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
      19             :  *  in a cluster, they can diverge due to CLUSTER, REINDEX, or VACUUM
      20             :  *  FULL.  In the new cluster, pg_class.oid and pg_class.relfilenode will
      21             :  *  be the same and will match the old pg_class.oid value.  Because of
      22             :  *  this, old/new pg_class.relfilenode values will not match if CLUSTER,
      23             :  *  REINDEX, or VACUUM FULL have been performed in the old cluster.
      24             :  *
      25             :  *  We control all assignments of pg_type.oid because these oids are stored
      26             :  *  in user composite type values.
      27             :  *
      28             :  *  We control all assignments of pg_enum.oid because these oids are stored
      29             :  *  in user tables as enum values.
      30             :  *
      31             :  *  We control all assignments of pg_authid.oid for historical reasons (the
      32             :  *  oids used to be stored in pg_largeobject_metadata, which is now copied via
      33             :  *  SQL commands), that might change at some point in the future.
      34             :  */
      35             : 
      36             : 
      37             : 
      38             : #include "postgres_fe.h"
      39             : 
      40             : #ifdef HAVE_LANGINFO_H
      41             : #include <langinfo.h>
      42             : #endif
      43             : 
      44             : #include "catalog/pg_class_d.h"
      45             : #include "common/file_perm.h"
      46             : #include "common/logging.h"
      47             : #include "common/restricted_token.h"
      48             : #include "fe_utils/string_utils.h"
      49             : #include "pg_upgrade.h"
      50             : 
      51             : static void prepare_new_cluster(void);
      52             : static void prepare_new_globals(void);
      53             : static void create_new_objects(void);
      54             : static void copy_xact_xlog_xid(void);
      55             : static void set_frozenxids(bool minmxid_only);
      56             : static void setup(char *argv0, bool *live_check);
      57             : static void cleanup(void);
      58             : 
      59             : ClusterInfo old_cluster,
      60             :             new_cluster;
      61             : OSInfo      os_info;
      62             : 
      63             : char       *output_files[] = {
      64             :     SERVER_LOG_FILE,
      65             : #ifdef WIN32
      66             :     /* unique file for pg_ctl start */
      67             :     SERVER_START_LOG_FILE,
      68             : #endif
      69             :     UTILITY_LOG_FILE,
      70             :     INTERNAL_LOG_FILE,
      71             :     NULL
      72             : };
      73             : 
      74             : 
      75             : int
      76           2 : main(int argc, char **argv)
      77             : {
      78           2 :     char       *deletion_script_file_name = NULL;
      79           2 :     bool        live_check = false;
      80             : 
      81           2 :     pg_logging_init(argv[0]);
      82           2 :     set_pglocale_pgservice(argv[0], PG_TEXTDOMAIN("pg_upgrade"));
      83             : 
      84             :     /* Set default restrictive mask until new cluster permissions are read */
      85           2 :     umask(PG_MODE_MASK_OWNER);
      86             : 
      87           2 :     parseCommandLine(argc, argv);
      88             : 
      89           2 :     get_restricted_token();
      90             : 
      91           2 :     adjust_data_dir(&old_cluster);
      92           2 :     adjust_data_dir(&new_cluster);
      93             : 
      94           2 :     setup(argv[0], &live_check);
      95             : 
      96           2 :     output_check_banner(live_check);
      97             : 
      98           2 :     check_cluster_versions();
      99             : 
     100           2 :     get_sock_dir(&old_cluster, live_check);
     101           2 :     get_sock_dir(&new_cluster, false);
     102             : 
     103           2 :     check_cluster_compatibility(live_check);
     104             : 
     105             :     /* Set mask based on PGDATA permissions */
     106           2 :     if (!GetDataDirectoryCreatePerm(new_cluster.pgdata))
     107           0 :         pg_fatal("could not read permissions of directory \"%s\": %s\n",
     108           0 :                  new_cluster.pgdata, strerror(errno));
     109             : 
     110           2 :     umask(pg_mode_mask);
     111             : 
     112           2 :     check_and_dump_old_cluster(live_check);
     113             : 
     114             : 
     115             :     /* -- NEW -- */
     116           2 :     start_postmaster(&new_cluster, true);
     117             : 
     118           2 :     check_new_cluster();
     119           2 :     report_clusters_compatible();
     120             : 
     121           2 :     pg_log(PG_REPORT,
     122             :            "\n"
     123             :            "Performing Upgrade\n"
     124             :            "------------------\n");
     125             : 
     126           2 :     prepare_new_cluster();
     127             : 
     128           2 :     stop_postmaster(false);
     129             : 
     130             :     /*
     131             :      * Destructive Changes to New Cluster
     132             :      */
     133             : 
     134           2 :     copy_xact_xlog_xid();
     135             : 
     136             :     /* New now using xids of the old system */
     137             : 
     138             :     /* -- NEW -- */
     139           2 :     start_postmaster(&new_cluster, true);
     140             : 
     141           2 :     prepare_new_globals();
     142             : 
     143           2 :     create_new_objects();
     144             : 
     145           2 :     stop_postmaster(false);
     146             : 
     147             :     /*
     148             :      * Most failures happen in create_new_objects(), which has completed at
     149             :      * this point.  We do this here because it is just before linking, which
     150             :      * will link the old and new cluster data files, preventing the old
     151             :      * cluster from being safely started once the new cluster is started.
     152             :      */
     153           2 :     if (user_opts.transfer_mode == TRANSFER_MODE_LINK)
     154           0 :         disable_old_cluster();
     155             : 
     156           2 :     transfer_all_new_tablespaces(&old_cluster.dbarr, &new_cluster.dbarr,
     157             :                                  old_cluster.pgdata, new_cluster.pgdata);
     158             : 
     159             :     /*
     160             :      * Assuming OIDs are only used in system tables, there is no need to
     161             :      * restore the OID counter because we have not transferred any OIDs from
     162             :      * the old system, but we do it anyway just in case.  We do it late here
     163             :      * because there is no need to have the schema load use new oids.
     164             :      */
     165           2 :     prep_status("Setting next OID for new cluster");
     166           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     167             :               "\"%s/pg_resetwal\" -o %u \"%s\"",
     168             :               new_cluster.bindir, old_cluster.controldata.chkpnt_nxtoid,
     169             :               new_cluster.pgdata);
     170           2 :     check_ok();
     171             : 
     172           2 :     prep_status("Sync data directory to disk");
     173           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     174             :               "\"%s/initdb\" --sync-only \"%s\"", new_cluster.bindir,
     175             :               new_cluster.pgdata);
     176           2 :     check_ok();
     177             : 
     178           2 :     create_script_for_old_cluster_deletion(&deletion_script_file_name);
     179             : 
     180           2 :     issue_warnings_and_set_wal_level();
     181             : 
     182           2 :     pg_log(PG_REPORT,
     183             :            "\n"
     184             :            "Upgrade Complete\n"
     185             :            "----------------\n");
     186             : 
     187           2 :     output_completion_banner(deletion_script_file_name);
     188             : 
     189           2 :     pg_free(deletion_script_file_name);
     190             : 
     191           2 :     cleanup();
     192             : 
     193           2 :     return 0;
     194             : }
     195             : 
     196             : 
     197             : static void
     198           2 : setup(char *argv0, bool *live_check)
     199             : {
     200             :     /*
     201             :      * make sure the user has a clean environment, otherwise, we may confuse
     202             :      * libpq when we connect to one (or both) of the servers.
     203             :      */
     204           2 :     check_pghost_envvar();
     205             : 
     206             :     /*
     207             :      * In case the user hasn't specified the directory for the new binaries
     208             :      * with -B, default to using the path of the currently executed pg_upgrade
     209             :      * binary.
     210             :      */
     211           2 :     if (!new_cluster.bindir)
     212             :     {
     213             :         char        exec_path[MAXPGPATH];
     214             : 
     215           2 :         if (find_my_exec(argv0, exec_path) < 0)
     216           0 :             pg_fatal("%s: could not find own program executable\n", argv0);
     217             :         /* Trim off program name and keep just path */
     218           2 :         *last_dir_separator(exec_path) = '\0';
     219           2 :         canonicalize_path(exec_path);
     220           2 :         new_cluster.bindir = pg_strdup(exec_path);
     221             :     }
     222             : 
     223           2 :     verify_directories();
     224             : 
     225             :     /* no postmasters should be running, except for a live check */
     226           2 :     if (pid_lock_file_exists(old_cluster.pgdata))
     227             :     {
     228             :         /*
     229             :          * If we have a postmaster.pid file, try to start the server.  If it
     230             :          * starts, the pid file was stale, so stop the server.  If it doesn't
     231             :          * start, assume the server is running.  If the pid file is left over
     232             :          * from a server crash, this also allows any committed transactions
     233             :          * stored in the WAL to be replayed so they are not lost, because WAL
     234             :          * files are not transferred from old to new servers.  We later check
     235             :          * for a clean shutdown.
     236             :          */
     237           0 :         if (start_postmaster(&old_cluster, false))
     238           0 :             stop_postmaster(false);
     239             :         else
     240             :         {
     241           0 :             if (!user_opts.check)
     242           0 :                 pg_fatal("There seems to be a postmaster servicing the old cluster.\n"
     243             :                          "Please shutdown that postmaster and try again.\n");
     244             :             else
     245           0 :                 *live_check = true;
     246             :         }
     247             :     }
     248             : 
     249             :     /* same goes for the new postmaster */
     250           2 :     if (pid_lock_file_exists(new_cluster.pgdata))
     251             :     {
     252           0 :         if (start_postmaster(&new_cluster, false))
     253           0 :             stop_postmaster(false);
     254             :         else
     255           0 :             pg_fatal("There seems to be a postmaster servicing the new cluster.\n"
     256             :                      "Please shutdown that postmaster and try again.\n");
     257             :     }
     258           2 : }
     259             : 
     260             : 
     261             : static void
     262           2 : prepare_new_cluster(void)
     263             : {
     264             :     /*
     265             :      * It would make more sense to freeze after loading the schema, but that
     266             :      * would cause us to lose the frozenxids restored by the load. We use
     267             :      * --analyze so autovacuum doesn't update statistics later
     268             :      */
     269           2 :     prep_status("Analyzing all rows in the new cluster");
     270           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     271             :               "\"%s/vacuumdb\" %s --all --analyze %s",
     272             :               new_cluster.bindir, cluster_conn_opts(&new_cluster),
     273           2 :               log_opts.verbose ? "--verbose" : "");
     274           2 :     check_ok();
     275             : 
     276             :     /*
     277             :      * We do freeze after analyze so pg_statistic is also frozen. template0 is
     278             :      * not frozen here, but data rows were frozen by initdb, and we set its
     279             :      * datfrozenxid, relfrozenxids, and relminmxid later to match the new xid
     280             :      * counter later.
     281             :      */
     282           2 :     prep_status("Freezing all rows in the new cluster");
     283           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     284             :               "\"%s/vacuumdb\" %s --all --freeze %s",
     285             :               new_cluster.bindir, cluster_conn_opts(&new_cluster),
     286           2 :               log_opts.verbose ? "--verbose" : "");
     287           2 :     check_ok();
     288           2 : }
     289             : 
     290             : 
     291             : static void
     292           2 : prepare_new_globals(void)
     293             : {
     294             :     /*
     295             :      * Before we restore anything, set frozenxids of initdb-created tables.
     296             :      */
     297           2 :     set_frozenxids(false);
     298             : 
     299             :     /*
     300             :      * Now restore global objects (roles and tablespaces).
     301             :      */
     302           2 :     prep_status("Restoring global objects in the new cluster");
     303             : 
     304           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     305             :               "\"%s/psql\" " EXEC_PSQL_ARGS " %s -f \"%s\"",
     306             :               new_cluster.bindir, cluster_conn_opts(&new_cluster),
     307             :               GLOBALS_DUMP_FILE);
     308           2 :     check_ok();
     309           2 : }
     310             : 
     311             : 
     312             : static void
     313           2 : create_new_objects(void)
     314             : {
     315             :     int         dbnum;
     316             : 
     317           2 :     prep_status("Restoring database schemas in the new cluster\n");
     318             : 
     319             :     /*
     320             :      * We cannot process the template1 database concurrently with others,
     321             :      * because when it's transiently dropped, connection attempts would fail.
     322             :      * So handle it in a separate non-parallelized pass.
     323             :      */
     324          12 :     for (dbnum = 0; dbnum < old_cluster.dbarr.ndbs; dbnum++)
     325             :     {
     326             :         char        sql_file_name[MAXPGPATH],
     327             :                     log_file_name[MAXPGPATH];
     328          12 :         DbInfo     *old_db = &old_cluster.dbarr.dbs[dbnum];
     329             :         const char *create_opts;
     330             : 
     331             :         /* Process only template1 in this pass */
     332          12 :         if (strcmp(old_db->db_name, "template1") != 0)
     333          10 :             continue;
     334             : 
     335           2 :         pg_log(PG_STATUS, "%s", old_db->db_name);
     336           2 :         snprintf(sql_file_name, sizeof(sql_file_name), DB_DUMP_FILE_MASK, old_db->db_oid);
     337           2 :         snprintf(log_file_name, sizeof(log_file_name), DB_DUMP_LOG_FILE_MASK, old_db->db_oid);
     338             : 
     339             :         /*
     340             :          * template1 database will already exist in the target installation,
     341             :          * so tell pg_restore to drop and recreate it; otherwise we would fail
     342             :          * to propagate its database-level properties.
     343             :          */
     344           2 :         create_opts = "--clean --create";
     345             : 
     346           2 :         exec_prog(log_file_name,
     347             :                   NULL,
     348             :                   true,
     349             :                   true,
     350             :                   "\"%s/pg_restore\" %s %s --exit-on-error --verbose "
     351             :                   "--dbname postgres \"%s\"",
     352             :                   new_cluster.bindir,
     353             :                   cluster_conn_opts(&new_cluster),
     354             :                   create_opts,
     355             :                   sql_file_name);
     356             : 
     357           2 :         break;                  /* done once we've processed template1 */
     358             :     }
     359             : 
     360          14 :     for (dbnum = 0; dbnum < old_cluster.dbarr.ndbs; dbnum++)
     361             :     {
     362             :         char        sql_file_name[MAXPGPATH],
     363             :                     log_file_name[MAXPGPATH];
     364          12 :         DbInfo     *old_db = &old_cluster.dbarr.dbs[dbnum];
     365             :         const char *create_opts;
     366             : 
     367             :         /* Skip template1 in this pass */
     368          12 :         if (strcmp(old_db->db_name, "template1") == 0)
     369           2 :             continue;
     370             : 
     371          10 :         pg_log(PG_STATUS, "%s", old_db->db_name);
     372          10 :         snprintf(sql_file_name, sizeof(sql_file_name), DB_DUMP_FILE_MASK, old_db->db_oid);
     373          10 :         snprintf(log_file_name, sizeof(log_file_name), DB_DUMP_LOG_FILE_MASK, old_db->db_oid);
     374             : 
     375             :         /*
     376             :          * postgres database will already exist in the target installation, so
     377             :          * tell pg_restore to drop and recreate it; otherwise we would fail to
     378             :          * propagate its database-level properties.
     379             :          */
     380          10 :         if (strcmp(old_db->db_name, "postgres") == 0)
     381           2 :             create_opts = "--clean --create";
     382             :         else
     383           8 :             create_opts = "--create";
     384             : 
     385          10 :         parallel_exec_prog(log_file_name,
     386             :                            NULL,
     387             :                            "\"%s/pg_restore\" %s %s --exit-on-error --verbose "
     388             :                            "--dbname template1 \"%s\"",
     389             :                            new_cluster.bindir,
     390             :                            cluster_conn_opts(&new_cluster),
     391             :                            create_opts,
     392             :                            sql_file_name);
     393             :     }
     394             : 
     395             :     /* reap all children */
     396           2 :     while (reap_child(true) == true)
     397             :         ;
     398             : 
     399           2 :     end_progress_output();
     400           2 :     check_ok();
     401             : 
     402             :     /*
     403             :      * We don't have minmxids for databases or relations in pre-9.3 clusters,
     404             :      * so set those after we have restored the schema.
     405             :      */
     406           2 :     if (GET_MAJOR_VERSION(old_cluster.major_version) <= 902)
     407           0 :         set_frozenxids(true);
     408             : 
     409             :     /* update new_cluster info now that we have objects in the databases */
     410           2 :     get_db_and_rel_infos(&new_cluster);
     411           2 : }
     412             : 
     413             : /*
     414             :  * Delete the given subdirectory contents from the new cluster
     415             :  */
     416             : static void
     417           6 : remove_new_subdir(const char *subdir, bool rmtopdir)
     418             : {
     419             :     char        new_path[MAXPGPATH];
     420             : 
     421           6 :     prep_status("Deleting files from new %s", subdir);
     422             : 
     423           6 :     snprintf(new_path, sizeof(new_path), "%s/%s", new_cluster.pgdata, subdir);
     424           6 :     if (!rmtree(new_path, rmtopdir))
     425           0 :         pg_fatal("could not delete directory \"%s\"\n", new_path);
     426             : 
     427           6 :     check_ok();
     428           6 : }
     429             : 
     430             : /*
     431             :  * Copy the files from the old cluster into it
     432             :  */
     433             : static void
     434           6 : copy_subdir_files(const char *old_subdir, const char *new_subdir)
     435             : {
     436             :     char        old_path[MAXPGPATH];
     437             :     char        new_path[MAXPGPATH];
     438             : 
     439           6 :     remove_new_subdir(new_subdir, true);
     440             : 
     441           6 :     snprintf(old_path, sizeof(old_path), "%s/%s", old_cluster.pgdata, old_subdir);
     442           6 :     snprintf(new_path, sizeof(new_path), "%s/%s", new_cluster.pgdata, new_subdir);
     443             : 
     444           6 :     prep_status("Copying old %s to new server", old_subdir);
     445             : 
     446           6 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     447             : #ifndef WIN32
     448             :               "cp -Rf \"%s\" \"%s\"",
     449             : #else
     450             :     /* flags: everything, no confirm, quiet, overwrite read-only */
     451             :               "xcopy /e /y /q /r \"%s\" \"%s\\\"",
     452             : #endif
     453             :               old_path, new_path);
     454             : 
     455           6 :     check_ok();
     456           6 : }
     457             : 
     458             : static void
     459           2 : copy_xact_xlog_xid(void)
     460             : {
     461             :     /*
     462             :      * Copy old commit logs to new data dir. pg_clog has been renamed to
     463             :      * pg_xact in post-10 clusters.
     464             :      */
     465           2 :     copy_subdir_files(GET_MAJOR_VERSION(old_cluster.major_version) <= 906 ?
     466             :                       "pg_clog" : "pg_xact",
     467           2 :                       GET_MAJOR_VERSION(new_cluster.major_version) <= 906 ?
     468             :                       "pg_clog" : "pg_xact");
     469             : 
     470           2 :     prep_status("Setting oldest XID for new cluster");
     471           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     472             :               "\"%s/pg_resetwal\" -f -u %u \"%s\"",
     473             :               new_cluster.bindir, old_cluster.controldata.chkpnt_oldstxid,
     474             :               new_cluster.pgdata);
     475           2 :     check_ok();
     476             : 
     477             :     /* set the next transaction id and epoch of the new cluster */
     478           2 :     prep_status("Setting next transaction ID and epoch for new cluster");
     479           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     480             :               "\"%s/pg_resetwal\" -f -x %u \"%s\"",
     481             :               new_cluster.bindir, old_cluster.controldata.chkpnt_nxtxid,
     482             :               new_cluster.pgdata);
     483           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     484             :               "\"%s/pg_resetwal\" -f -e %u \"%s\"",
     485             :               new_cluster.bindir, old_cluster.controldata.chkpnt_nxtepoch,
     486             :               new_cluster.pgdata);
     487             :     /* must reset commit timestamp limits also */
     488           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     489             :               "\"%s/pg_resetwal\" -f -c %u,%u \"%s\"",
     490             :               new_cluster.bindir,
     491             :               old_cluster.controldata.chkpnt_nxtxid,
     492             :               old_cluster.controldata.chkpnt_nxtxid,
     493             :               new_cluster.pgdata);
     494           2 :     check_ok();
     495             : 
     496             :     /*
     497             :      * If the old server is before the MULTIXACT_FORMATCHANGE_CAT_VER change
     498             :      * (see pg_upgrade.h) and the new server is after, then we don't copy
     499             :      * pg_multixact files, but we need to reset pg_control so that the new
     500             :      * server doesn't attempt to read multis older than the cutoff value.
     501             :      */
     502           2 :     if (old_cluster.controldata.cat_ver >= MULTIXACT_FORMATCHANGE_CAT_VER &&
     503           2 :         new_cluster.controldata.cat_ver >= MULTIXACT_FORMATCHANGE_CAT_VER)
     504             :     {
     505           2 :         copy_subdir_files("pg_multixact/offsets", "pg_multixact/offsets");
     506           2 :         copy_subdir_files("pg_multixact/members", "pg_multixact/members");
     507             : 
     508           2 :         prep_status("Setting next multixact ID and offset for new cluster");
     509             : 
     510             :         /*
     511             :          * we preserve all files and contents, so we must preserve both "next"
     512             :          * counters here and the oldest multi present on system.
     513             :          */
     514           2 :         exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     515             :                   "\"%s/pg_resetwal\" -O %u -m %u,%u \"%s\"",
     516             :                   new_cluster.bindir,
     517             :                   old_cluster.controldata.chkpnt_nxtmxoff,
     518             :                   old_cluster.controldata.chkpnt_nxtmulti,
     519             :                   old_cluster.controldata.chkpnt_oldstMulti,
     520             :                   new_cluster.pgdata);
     521           2 :         check_ok();
     522             :     }
     523           0 :     else if (new_cluster.controldata.cat_ver >= MULTIXACT_FORMATCHANGE_CAT_VER)
     524             :     {
     525             :         /*
     526             :          * Remove offsets/0000 file created by initdb that no longer matches
     527             :          * the new multi-xid value.  "members" starts at zero so no need to
     528             :          * remove it.
     529             :          */
     530           0 :         remove_new_subdir("pg_multixact/offsets", false);
     531             : 
     532           0 :         prep_status("Setting oldest multixact ID in new cluster");
     533             : 
     534             :         /*
     535             :          * We don't preserve files in this case, but it's important that the
     536             :          * oldest multi is set to the latest value used by the old system, so
     537             :          * that multixact.c returns the empty set for multis that might be
     538             :          * present on disk.  We set next multi to the value following that; it
     539             :          * might end up wrapped around (i.e. 0) if the old cluster had
     540             :          * next=MaxMultiXactId, but multixact.c can cope with that just fine.
     541             :          */
     542           0 :         exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     543             :                   "\"%s/pg_resetwal\" -m %u,%u \"%s\"",
     544             :                   new_cluster.bindir,
     545           0 :                   old_cluster.controldata.chkpnt_nxtmulti + 1,
     546             :                   old_cluster.controldata.chkpnt_nxtmulti,
     547             :                   new_cluster.pgdata);
     548           0 :         check_ok();
     549             :     }
     550             : 
     551             :     /* now reset the wal archives in the new cluster */
     552           2 :     prep_status("Resetting WAL archives");
     553           2 :     exec_prog(UTILITY_LOG_FILE, NULL, true, true,
     554             :     /* use timeline 1 to match controldata and no WAL history file */
     555             :               "\"%s/pg_resetwal\" -l 00000001%s \"%s\"", new_cluster.bindir,
     556             :               old_cluster.controldata.nextxlogfile + 8,
     557             :               new_cluster.pgdata);
     558           2 :     check_ok();
     559           2 : }
     560             : 
     561             : 
     562             : /*
     563             :  *  set_frozenxids()
     564             :  *
     565             :  * This is called on the new cluster before we restore anything, with
     566             :  * minmxid_only = false.  Its purpose is to ensure that all initdb-created
     567             :  * vacuumable tables have relfrozenxid/relminmxid matching the old cluster's
     568             :  * xid/mxid counters.  We also initialize the datfrozenxid/datminmxid of the
     569             :  * built-in databases to match.
     570             :  *
     571             :  * As we create user tables later, their relfrozenxid/relminmxid fields will
     572             :  * be restored properly by the binary-upgrade restore script.  Likewise for
     573             :  * user-database datfrozenxid/datminmxid.  However, if we're upgrading from a
     574             :  * pre-9.3 database, which does not store per-table or per-DB minmxid, then
     575             :  * the relminmxid/datminmxid values filled in by the restore script will just
     576             :  * be zeroes.
     577             :  *
     578             :  * Hence, with a pre-9.3 source database, a second call occurs after
     579             :  * everything is restored, with minmxid_only = true.  This pass will
     580             :  * initialize all tables and databases, both those made by initdb and user
     581             :  * objects, with the desired minmxid value.  frozenxid values are left alone.
     582             :  */
     583             : static void
     584           2 : set_frozenxids(bool minmxid_only)
     585             : {
     586             :     int         dbnum;
     587             :     PGconn     *conn,
     588             :                *conn_template1;
     589             :     PGresult   *dbres;
     590             :     int         ntups;
     591             :     int         i_datname;
     592             :     int         i_datallowconn;
     593             : 
     594           2 :     if (!minmxid_only)
     595           2 :         prep_status("Setting frozenxid and minmxid counters in new cluster");
     596             :     else
     597           0 :         prep_status("Setting minmxid counter in new cluster");
     598             : 
     599           2 :     conn_template1 = connectToServer(&new_cluster, "template1");
     600             : 
     601           2 :     if (!minmxid_only)
     602             :         /* set pg_database.datfrozenxid */
     603           2 :         PQclear(executeQueryOrDie(conn_template1,
     604             :                                   "UPDATE pg_catalog.pg_database "
     605             :                                   "SET datfrozenxid = '%u'",
     606             :                                   old_cluster.controldata.chkpnt_nxtxid));
     607             : 
     608             :     /* set pg_database.datminmxid */
     609           2 :     PQclear(executeQueryOrDie(conn_template1,
     610             :                               "UPDATE pg_catalog.pg_database "
     611             :                               "SET datminmxid = '%u'",
     612             :                               old_cluster.controldata.chkpnt_nxtmulti));
     613             : 
     614             :     /* get database names */
     615           2 :     dbres = executeQueryOrDie(conn_template1,
     616             :                               "SELECT  datname, datallowconn "
     617             :                               "FROM    pg_catalog.pg_database");
     618             : 
     619           2 :     i_datname = PQfnumber(dbres, "datname");
     620           2 :     i_datallowconn = PQfnumber(dbres, "datallowconn");
     621             : 
     622           2 :     ntups = PQntuples(dbres);
     623           8 :     for (dbnum = 0; dbnum < ntups; dbnum++)
     624             :     {
     625           6 :         char       *datname = PQgetvalue(dbres, dbnum, i_datname);
     626           6 :         char       *datallowconn = PQgetvalue(dbres, dbnum, i_datallowconn);
     627             : 
     628             :         /*
     629             :          * We must update databases where datallowconn = false, e.g.
     630             :          * template0, because autovacuum increments their datfrozenxids,
     631             :          * relfrozenxids, and relminmxid even if autovacuum is turned off, and
     632             :          * even though all the data rows are already frozen.  To enable this,
     633             :          * we temporarily change datallowconn.
     634             :          */
     635           6 :         if (strcmp(datallowconn, "f") == 0)
     636           2 :             PQclear(executeQueryOrDie(conn_template1,
     637             :                                       "ALTER DATABASE %s ALLOW_CONNECTIONS = true",
     638             :                                       quote_identifier(datname)));
     639             : 
     640           6 :         conn = connectToServer(&new_cluster, datname);
     641             : 
     642           6 :         if (!minmxid_only)
     643             :             /* set pg_class.relfrozenxid */
     644           6 :             PQclear(executeQueryOrDie(conn,
     645             :                                       "UPDATE  pg_catalog.pg_class "
     646             :                                       "SET relfrozenxid = '%u' "
     647             :             /* only heap, materialized view, and TOAST are vacuumed */
     648             :                                       "WHERE   relkind IN ("
     649             :                                       CppAsString2(RELKIND_RELATION) ", "
     650             :                                       CppAsString2(RELKIND_MATVIEW) ", "
     651             :                                       CppAsString2(RELKIND_TOASTVALUE) ")",
     652             :                                       old_cluster.controldata.chkpnt_nxtxid));
     653             : 
     654             :         /* set pg_class.relminmxid */
     655           6 :         PQclear(executeQueryOrDie(conn,
     656             :                                   "UPDATE  pg_catalog.pg_class "
     657             :                                   "SET relminmxid = '%u' "
     658             :         /* only heap, materialized view, and TOAST are vacuumed */
     659             :                                   "WHERE   relkind IN ("
     660             :                                   CppAsString2(RELKIND_RELATION) ", "
     661             :                                   CppAsString2(RELKIND_MATVIEW) ", "
     662             :                                   CppAsString2(RELKIND_TOASTVALUE) ")",
     663             :                                   old_cluster.controldata.chkpnt_nxtmulti));
     664           6 :         PQfinish(conn);
     665             : 
     666             :         /* Reset datallowconn flag */
     667           6 :         if (strcmp(datallowconn, "f") == 0)
     668           2 :             PQclear(executeQueryOrDie(conn_template1,
     669             :                                       "ALTER DATABASE %s ALLOW_CONNECTIONS = false",
     670             :                                       quote_identifier(datname)));
     671             :     }
     672             : 
     673           2 :     PQclear(dbres);
     674             : 
     675           2 :     PQfinish(conn_template1);
     676             : 
     677           2 :     check_ok();
     678           2 : }
     679             : 
     680             : 
     681             : static void
     682           2 : cleanup(void)
     683             : {
     684           2 :     fclose(log_opts.internal);
     685             : 
     686             :     /* Remove dump and log files? */
     687           2 :     if (!log_opts.retain)
     688             :     {
     689             :         int         dbnum;
     690             :         char      **filename;
     691             : 
     692           8 :         for (filename = output_files; *filename != NULL; filename++)
     693           6 :             unlink(*filename);
     694             : 
     695             :         /* remove dump files */
     696           2 :         unlink(GLOBALS_DUMP_FILE);
     697             : 
     698           2 :         if (old_cluster.dbarr.dbs)
     699          14 :             for (dbnum = 0; dbnum < old_cluster.dbarr.ndbs; dbnum++)
     700             :             {
     701             :                 char        sql_file_name[MAXPGPATH],
     702             :                             log_file_name[MAXPGPATH];
     703          12 :                 DbInfo     *old_db = &old_cluster.dbarr.dbs[dbnum];
     704             : 
     705          12 :                 snprintf(sql_file_name, sizeof(sql_file_name), DB_DUMP_FILE_MASK, old_db->db_oid);
     706          12 :                 unlink(sql_file_name);
     707             : 
     708          12 :                 snprintf(log_file_name, sizeof(log_file_name), DB_DUMP_LOG_FILE_MASK, old_db->db_oid);
     709          12 :                 unlink(log_file_name);
     710             :             }
     711             :     }
     712           2 : }

Generated by: LCOV version 1.14