fuzz: hack entrypoint

openbook-dex · Jan 24, 2024 · 2e83992 · 2e83992
1 parent 9fbff36
commit 2e83992
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/programs/openbook-v2/fuzz/src/accounts_state.rs b/programs/openbook-v2/fuzz/src/accounts_state.rs
@@ -75,7 +75,7 @@ impl AccountsState {
         infos
     }
 
-    pub fn update(&mut self, infos: Vec<AccountInfo>) {
+    pub fn update(&mut self, infos: &[AccountInfo]) {
         infos.iter().for_each(|info| {
             let account = self.0.get_mut(info.key).unwrap();
             let new_data = info.data.borrow();

diff --git a/programs/openbook-v2/fuzz/src/processor.rs b/programs/openbook-v2/fuzz/src/processor.rs
@@ -57,11 +57,18 @@ impl program_stubs::SyscallStubs for TestSyscallStubs {
                 &new_account_infos,
                 &instruction.data,
             ),
-            id if id == openbook_v2::ID => openbook_v2::entry(
-                &instruction.program_id,
-                &new_account_infos,
-                &instruction.data,
-            ),
+            id if id == openbook_v2::ID => {
+                let extended_lifetime_accs = unsafe {
+                    core::mem::transmute::<&[AccountInfo], &[AccountInfo<'_>]>(
+                        new_account_infos.as_ref(),
+                    )
+                };
+                openbook_v2::entry(
+                    &instruction.program_id,
+                    &extended_lifetime_accs,
+                    &instruction.data,
+                )
+            }
             _ => Err(ProgramError::IncorrectProgramId),
         }
     }
@@ -99,7 +106,7 @@ pub fn process_instruction(
     );
 
     if res.is_ok() {
-        state.update(account_infos);
+        state.update(&account_infos);
     }
 
     res