program: whitelist lighthouse for end of tx (#1429)

* program: whitelist lighthouse for end of tx * cargo fmt -- * CHANGELOG
drift-labs · Jan 20, 2025 · 1ab525d · 1ab525d
1 parent d15ff6c
commit 1ab525d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Features
 
+- program: allow lighthouse at end of swap ([#1429](https://github.com/drift-labs/protocol-v2/pull/1429))
+
 ### Fixes
 
 ### Breaking

diff --git a/programs/drift/src/ids.rs b/programs/drift/src/ids.rs
@@ -82,3 +82,8 @@ pub mod swift_server {
     #[cfg(feature = "anchor-test")]
     declare_id!("DpaEdAPW3ZX67fnczT14AoX12Lx9VMkxvtT81nCHy3Nv");
 }
+
+pub mod lighthouse {
+    use solana_program::declare_id;
+    declare_id!("L2TExMFKdjpN9kozasaurPirfHy9P8sbXoAN1qA3S95");
+}
diff --git a/programs/drift/src/instructions/user.rs b/programs/drift/src/instructions/user.rs
@@ -19,7 +19,8 @@ use crate::controller::spot_position::{
 };
 use crate::error::ErrorCode;
 use crate::ids::{
-    jupiter_mainnet_3, jupiter_mainnet_4, jupiter_mainnet_6, marinade_mainnet, serum_program,
+    jupiter_mainnet_3, jupiter_mainnet_4, jupiter_mainnet_6, lighthouse, marinade_mainnet,
+    serum_program,
 };
 use crate::instructions::constraints::*;
 use crate::instructions::optional_accounts::{
@@ -2683,6 +2684,10 @@ pub fn handle_begin_swap<'c: 'info, 'info>(
             }
         } else {
             if found_end {
+                if ix.program_id == lighthouse::ID {
+                    continue;
+                }
+
                 for meta in ix.accounts.iter() {
                     validate!(
                         meta.is_writable == false,