✨ Initialize svm with current time

Author: lukacan

src/accounts_db.rs

@@ -1,5 +1,4 @@
 use std::collections::HashMap;
-use std::time::UNIX_EPOCH;
 
 use serde::de::DeserializeOwned;
 
@@ -10,6 +9,8 @@ use solana_sdk::pubkey::Pubkey;
 use solana_sdk::sysvar::Sysvar;
 use solana_sdk::sysvar::SysvarId;
 
+use crate::utils::get_current_timestamp;
+
 #[derive(Default)]
 pub struct AccountsDB {
     accounts: HashMap<Pubkey, AccountSharedData>,
@@ -83,10 +84,7 @@ impl AccountsDB {
         let mut clock: Clock =
             bincode::deserialize(self.sysvars.get(&Clock::id()).unwrap().data()).unwrap();
 
-        let current_timestamp = std::time::SystemTime::now()
-            .duration_since(UNIX_EPOCH)
-            .expect("Time went backwards!")
-            .as_secs();
+        let current_timestamp = get_current_timestamp();
 
         // current time is always greater than last clock update
         let time_since_last_update =
@@ -126,10 +124,7 @@ pub struct SysvarTracker {
 
 impl SysvarTracker {
     pub fn refresh_last_clock_update(&mut self) {
-        self.last_clock_update = std::time::SystemTime::now()
-            .duration_since(UNIX_EPOCH)
-            .expect("Time went backwards!")
-            .as_secs();
+        self.last_clock_update = get_current_timestamp();
     }
 }
 

src/trident_svm_methods.rs

@@ -49,6 +49,7 @@ use crate::log::turn_off_solana_logging;
 use crate::native::BUILTINS;
 use crate::trident_fork_graphs::TridentForkGraph;
 use crate::trident_svm::TridentSVM;
+use crate::utils::get_current_timestamp;
 use crate::utils::ProgramEntrypoint;
 use crate::utils::SBFTarget;
 use crate::utils::TridentAccountSharedData;
@@ -97,7 +98,11 @@ impl TridentSVM {
         self
     }
     fn with_sysvars(mut self) -> Self {
-        self.set_sysvar(&Clock::default());
+        let clock = Clock {
+            unix_timestamp: get_current_timestamp() as i64,
+            ..Default::default()
+        };
+        self.set_sysvar(&clock);
         self.set_sysvar(&EpochRewards::default());
         self.set_sysvar(&EpochSchedule::default());
         #[allow(deprecated)]

src/utils.rs

@@ -1,3 +1,5 @@
+use std::time::UNIX_EPOCH;
+
 use solana_program_runtime::invoke_context::BuiltinFunctionWithContext;
 
 use solana_sdk::account::AccountSharedData;
@@ -46,3 +48,10 @@ impl TridentAccountSharedData {
         Self { address, account }
     }
 }
+
+pub(crate) fn get_current_timestamp() -> u64 {
+    std::time::SystemTime::now()
+        .duration_since(UNIX_EPOCH)
+        .expect("Time went backwards!")
+        .as_secs()
+}