Add publisher check register_v2_callbacks in token pools

Author: JohnChangUK

bindings/ccip_token_pools/managed_token_pool/managed_token_pool/managed_token_pool.go

@@ -137,6 +137,10 @@ module burn_mint_token_pool::burn_mint_token_pool {
     }
 
     public fun register_v2_callbacks(publisher: &signer) {
+        assert!(
+            signer::address_of(publisher) == @burn_mint_token_pool,
+            error::permission_denied(E_NOT_PUBLISHER)
+        );
         token_admin_registry::register_pool_v2(
             publisher,
             @burn_mint_local_token,

bindings/ccip_token_pools/regulated_token_pool/regulated_token_pool/regulated_token_pool.go

@@ -160,6 +160,10 @@ module lock_release_token_pool::lock_release_token_pool {
     }
 
     public fun register_v2_callbacks(publisher: &signer) {
+        assert!(
+            signer::address_of(publisher) == @lock_release_token_pool,
+            error::permission_denied(E_NOT_PUBLISHER)
+        );
         token_admin_registry::register_pool_v2(
             publisher,
             @lock_release_local_token,

contracts/ccip/ccip_token_pools/burn_mint_token_pool/sources/burn_mint_token_pool.move

@@ -29,6 +29,7 @@ module managed_token_pool::managed_token_pool {
 
     const E_INVALID_ARGUMENTS: u64 = 1;
     const E_UNKNOWN_FUNCTION: u64 = 2;
+    const E_NOT_PUBLISHER: u64 = 3;
 
     // ================================================================
     // |                             Init                             |
@@ -85,6 +86,10 @@ module managed_token_pool::managed_token_pool {
     }
 
     public fun register_v2_callbacks(publisher: &signer) {
+        assert!(
+            signer::address_of(publisher) == @managed_token_pool,
+            error::permission_denied(E_NOT_PUBLISHER)
+        );
         let managed_token_address = managed_token::token_metadata();
         token_admin_registry::register_pool_v2(
             publisher,

contracts/ccip/ccip_token_pools/lock_release_token_pool/sources/lock_release_token_pool.move

@@ -29,6 +29,7 @@ module regulated_token_pool::regulated_token_pool {
 
     const E_INVALID_ARGUMENTS: u64 = 1;
     const E_UNKNOWN_FUNCTION: u64 = 2;
+    const E_NOT_PUBLISHER: u64 = 3;
 
     // ================================================================
     // |                             Init                             |
@@ -83,6 +84,10 @@ module regulated_token_pool::regulated_token_pool {
     }
 
     public fun register_v2_callbacks(publisher: &signer) {
+        assert!(
+            signer::address_of(publisher) == @regulated_token_pool,
+            error::permission_denied(E_NOT_PUBLISHER)
+        );
         let regulated_token_address = regulated_token::token_address();
         token_admin_registry::register_pool_v2(
             publisher,

contracts/ccip/ccip_token_pools/managed_token_pool/sources/managed_token_pool.move

@@ -139,6 +139,10 @@ module usdc_token_pool::usdc_token_pool {
     }
 
     public fun register_v2_callbacks(publisher: &signer) {
+        assert!(
+            signer::address_of(publisher) == @usdc_token_pool,
+            error::permission_denied(E_NOT_PUBLISHER)
+        );
         token_admin_registry::register_pool_v2(
             publisher,
             @local_token,