From 620ad368e8d16d9f6d0444587fc1721ca7d91563 Mon Sep 17 00:00:00 2001
From: benthecarman <benthecarman@live.com>
Date: Thu, 11 Jan 2024 12:06:49 +0000
Subject: [PATCH] Verify fedimint network

---
 mutiny-core/src/federation.rs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/mutiny-core/src/federation.rs b/mutiny-core/src/federation.rs
index ea4cf90dc..effa18018 100644
--- a/mutiny-core/src/federation.rs
+++ b/mutiny-core/src/federation.rs
@@ -39,7 +39,7 @@ use fedimint_ln_client::{
 };
 use fedimint_ln_common::LightningCommonInit;
 use fedimint_mint_client::MintClientInit;
-use fedimint_wallet_client::WalletClientInit;
+use fedimint_wallet_client::{WalletClientInit, WalletClientModule};
 use futures::future::{self};
 use futures_util::{pin_mut, StreamExt};
 use lightning::{log_debug, log_error, log_info, log_trace, log_warn, util::logger::Logger};
@@ -174,6 +174,17 @@ impl FederationClient {
             ))
             .await?;
 
+        // check federation is on expected network
+        let wallet_client = fedimint_client.get_first_module::<WalletClientModule>();
+        if network != wallet_client.get_network() {
+            log_error!(
+                logger,
+                "Fedimint on different network {}, expected: {network}",
+                wallet_client.get_network()
+            );
+            return Err(MutinyError::IncorrectNetwork(network));
+        }
+
         log_debug!(logger, "Built fedimint client");
         Ok(FederationClient {
             uuid,