Add enable-capitalization-change flag

ledger-tool/src/main.rs

@@ -1295,6 +1295,12 @@ fn main() {
                         .takes_value(true)
                         .help("Snapshot archive format to use.")
                         .conflicts_with("no_snapshot"),
+                )
+                .arg(
+                    Arg::with_name("enable_capitalization_change")
+                        .long("enable-capitalization-change")
+                        .takes_value(false)
+                        .help("If snapshot creation should succeed with a capitalization delta."),
                 ),
         )
         .subcommand(
@@ -1806,6 +1812,9 @@ fn main() {
                         None
                     };
 
+                    let enable_capitalization_change =
+                        arg_matches.is_present("enable_capitalization_change");
+
                     let snapshot_type_str = if is_incremental {
                         "incremental "
                     } else if is_minimized {
@@ -2052,13 +2061,21 @@ fn main() {
                     bank.set_capitalization();
 
                     let post_capitalization = bank.capitalization();
+
                     if pre_capitalization != post_capitalization {
                         let amount = if pre_capitalization > post_capitalization {
                             format!("-{}", pre_capitalization - post_capitalization)
                         } else {
                             (post_capitalization - pre_capitalization).to_string()
                         };
-                        warn!("Capitalization change: {amount} lamports");
+                        let msg = format!("Capitalization change: {amount} lamports");
+                        warn!("{msg}");
+                        if !enable_capitalization_change {
+                            eprintln!(
+                                "{msg}\nBut `--enable-capitalization-change flag not provided"
+                            );
+                            exit(1);
+                        }
                     }
 
                     let bank = if let Some(warp_slot) = warp_slot {