chore: check epochs when we ::connect() to the sortition DB against the given epochs, and replace them if need be

Author: jcnelson

stackslib/src/chainstate/burn/db/sortdb.rs

@@ -2848,6 +2848,35 @@ impl SortitionDB {
         Ok(())
     }
 
+    /// Validates given StacksEpochs (will runtime panic if there is any invalid StacksEpoch structuring) and
+    /// replaces them into the SortitionDB's epochs table
+    fn validate_and_replace_epochs(
+        db_tx: &Transaction,
+        epochs: &[StacksEpoch],
+    ) -> Result<(), db_error> {
+        let epochs = StacksEpoch::validate_epochs(epochs);
+        let existing_epochs = Self::get_stacks_epochs(db_tx)?;
+        if existing_epochs == epochs {
+            return Ok(());
+        }
+        info!("Replace existing epochs with new epochs");
+        db_tx.execute("DELETE FROM epochs;", NO_PARAMS)?;
+        for epoch in epochs.into_iter() {
+            let args: &[&dyn ToSql] = &[
+                &(epoch.epoch_id as u32),
+                &u64_to_sql(epoch.start_height)?,
+                &u64_to_sql(epoch.end_height)?,
+                &epoch.block_limit,
+                &epoch.network_epoch,
+            ];
+            db_tx.execute(
+                "INSERT INTO epochs (epoch_id,start_block_height,end_block_height,block_limit,network_epoch) VALUES (?1,?2,?3,?4,?5)",
+                args
+            )?;
+        }
+        Ok(())
+    }
+
     /// Get a block commit by its content-addressed location in a specific sortition.
     pub fn get_block_commit(
         conn: &Connection,
@@ -3322,6 +3351,10 @@ impl SortitionDB {
 
                         self.apply_schema_8_migration(migrator.take())?;
                     } else if version == expected_version {
+                        let tx = self.tx_begin()?;
+                        SortitionDB::validate_and_replace_epochs(&tx, epochs)?;
+                        tx.commit()?;
+
                         return Ok(());
                     } else {
                         panic!("The schema version of the sortition DB is invalid.")
@@ -10629,4 +10662,51 @@ pub mod tests {
             good_ops_2[3]
         );
     }
+
+    #[test]
+    fn test_validate_and_replace_epochs() {
+        use crate::core::STACKS_EPOCHS_MAINNET;
+
+        let path_root = "/tmp/test_validate_and_replace_epochs";
+        if fs::metadata(path_root).is_ok() {
+            fs::remove_dir_all(path_root).unwrap();
+        }
+
+        fs::create_dir_all(path_root).unwrap();
+
+        let mut bad_epochs = STACKS_EPOCHS_MAINNET.to_vec();
+        let idx = bad_epochs.len() - 2;
+        bad_epochs[idx].end_height += 1;
+        bad_epochs[idx + 1].start_height += 1;
+
+        let sortdb = SortitionDB::connect(
+            &format!("{}/sortdb.sqlite", &path_root),
+            0,
+            &BurnchainHeaderHash([0x00; 32]),
+            0,
+            &bad_epochs,
+            PoxConstants::mainnet_default(),
+            None,
+            true,
+        )
+        .unwrap();
+
+        let db_epochs = SortitionDB::get_stacks_epochs(sortdb.conn()).unwrap();
+        assert_eq!(db_epochs, bad_epochs);
+
+        let fixed_sortdb = SortitionDB::connect(
+            &format!("{}/sortdb.sqlite", &path_root),
+            0,
+            &BurnchainHeaderHash([0x00; 32]),
+            0,
+            &STACKS_EPOCHS_MAINNET.to_vec(),
+            PoxConstants::mainnet_default(),
+            None,
+            true,
+        )
+        .unwrap();
+
+        let db_epochs = SortitionDB::get_stacks_epochs(sortdb.conn()).unwrap();
+        assert_eq!(db_epochs, STACKS_EPOCHS_MAINNET.to_vec());
+    }
 }