added in no_entrypoint

jito-foundation · Oct 31, 2024 · 0936626 · 0936626
1 parent 928b9f7
commit 0936626
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 14 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -73,9 +73,9 @@ jito-account-traits-derive = { git = "https://github.com/jito-foundation/restaki
 jito-jsm-core = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3" }
 jito-restaking-client = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3" }
 jito-restaking-core = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3" }
-jito-restaking-program = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3" }
+jito-restaking-program = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3", features = ["no-entrypoint"] }
 jito-restaking-sdk = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3" }
 jito-vault-client = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3" }
 jito-vault-core = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3" }
-jito-vault-program = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3" }
+jito-vault-program = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3", features = ["no-entrypoint"] }
 jito-vault-sdk = { git = "https://github.com/jito-foundation/restaking.git", version = "=0.0.3" }
diff --git a/program/src/finalize_weight_table.rs b/program/src/finalize_weight_table.rs
@@ -30,10 +30,10 @@ pub fn process_finalize_weight_table(
     load_signer(weight_table_admin, true)?;
     WeightTable::load(program_id, weight_table, ncn, ncn_epoch, true)?;
 
-    // if restaking_program_id.key.ne(&jito_restaking_program()) {
-    //     msg!("Incorrect restaking program ID");
-    //     return Err(ProgramError::InvalidAccountData);
-    // }
+    if restaking_program_id.key.ne(&jito_restaking_program::id()) {
+        msg!("Incorrect restaking program ID");
+        return Err(ProgramError::InvalidAccountData);
+    }
 
     if ncn_weight_table_admin.ne(weight_table_admin.key) {
         msg!("Vault update delegations ticket is not at the correct PDA");

diff --git a/program/src/initialize_weight_table.rs b/program/src/initialize_weight_table.rs
@@ -45,10 +45,10 @@ pub fn process_initialize_weight_table(
     load_signer(weight_table_admin, true)?;
     load_system_program(system_program)?;
 
-    // if restaking_program_id.key.ne(&jito_restaking_program::id()) {
-    //     msg!("Incorrect restaking program ID");
-    //     return Err(ProgramError::InvalidAccountData);
-    // }
+    if restaking_program_id.key.ne(&jito_restaking_program::id()) {
+        msg!("Incorrect restaking program ID");
+        return Err(ProgramError::InvalidAccountData);
+    }
 
     if ncn_weight_table_admin.ne(weight_table_admin.key) {
         msg!("Vault update delegations ticket is not at the correct PDA");

diff --git a/program/src/update_weight_table.rs b/program/src/update_weight_table.rs
@@ -33,10 +33,10 @@ pub fn process_update_weight_table(
     load_token_mint(mint)?;
     WeightTable::load(program_id, weight_table, ncn, ncn_epoch, true)?;
 
-    // if restaking_program_id.key.ne(&jito_restaking_program::id()) {
-    //     msg!("Incorrect restaking program ID");
-    //     return Err(ProgramError::InvalidAccountData);
-    // }
+    if restaking_program_id.key.ne(&jito_restaking_program::id()) {
+        msg!("Incorrect restaking program ID");
+        return Err(ProgramError::InvalidAccountData);
+    }
 
     if ncn_weight_table_admin.ne(weight_table_admin.key) {
         msg!("Vault update delegations ticket is not at the correct PDA");