solana: add missing endpoint constraint (#190)

Co-authored-by: A5 Pickle <[email protected]>
wormhole-foundation · Aug 9, 2024 · 3e0d3e0 · 3e0d3e0
1 parent d003059
commit 3e0d3e0
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/solana/programs/matching-engine/src/processor/admin/router_endpoint/update/cctp.rs b/solana/programs/matching-engine/src/processor/admin/router_endpoint/update/cctp.rs
@@ -12,6 +12,17 @@ use common::wormhole_cctp_solana::cctp::token_messenger_minter_program::{
 pub struct UpdateCctpRouterEndpoint<'info> {
     admin: OwnerOnly<'info>,
 
+    #[account(
+        constraint = {
+            require_eq!(
+                args.chain,
+                router_endpoint.chain,
+                crate::error::MatchingEngineError::InvalidEndpoint,
+            );
+
+            true
+        }
+    )]
     router_endpoint: ExistingMutRouterEndpoint<'info>,
 
     /// CHECK: Seeds must be \["remote_token_messenger"\, remote_domain.to_string()] (CCTP Token