From 2c1d548898c164b141d10945f815c6b1721bcc57 Mon Sep 17 00:00:00 2001
From: Kunal Bhargava <kbhargava@jumptrading.com>
Date: Fri, 16 May 2025 22:24:23 +0000
Subject: [PATCH] flamenco: activate precompiles at activation

---
 src/flamenco/runtime/fd_runtime.c             |  8 +++++
 .../runtime/program/fd_builtin_programs.c     | 29 +++++++++++++++++++
 .../runtime/program/fd_builtin_programs.h     | 13 +++++++++
 3 files changed, 50 insertions(+)

diff --git a/src/flamenco/runtime/fd_runtime.c b/src/flamenco/runtime/fd_runtime.c
index 925f6b2a2f..901fecaf12 100644
--- a/src/flamenco/runtime/fd_runtime.c
+++ b/src/flamenco/runtime/fd_runtime.c
@@ -2660,6 +2660,14 @@ fd_apply_builtin_program_feature_transitions( fd_exec_slot_ctx_t * slot_ctx,
     }
   }
 
+  /* https://github.com/anza-xyz/agave/blob/c1080de464cfb578c301e975f498964b5d5313db/runtime/src/bank.rs#L6795-L6805 */
+  fd_precompile_program_t const * precompiles = fd_precompiles();
+  for( ulong i=0UL; i<fd_num_precompiles(); i++ ) {
+    if( FD_FEATURE_JUST_ACTIVATED_OFFSET( slot_ctx, precompiles[i].feature_offset ) ) {
+      fd_write_builtin_account( slot_ctx, *precompiles[i].pubkey, "", 0 );
+    }
+  }
+
   } FD_SPAD_FRAME_END;
 }
 
diff --git a/src/flamenco/runtime/program/fd_builtin_programs.c b/src/flamenco/runtime/program/fd_builtin_programs.c
index 5ef4bf7c77..35081d6ccb 100644
--- a/src/flamenco/runtime/program/fd_builtin_programs.c
+++ b/src/flamenco/runtime/program/fd_builtin_programs.c
@@ -1,4 +1,5 @@
 #include "fd_builtin_programs.h"
+#include "fd_precompiles.h"
 #include "../fd_runtime.h"
 #include "../fd_acc_mgr.h"
 #include "../fd_system_ids.h"
@@ -26,6 +27,13 @@
         builtin_program_id                                                                                                     \
     }
 
+#define PRECOMPILE(program_id, feature_offset, verify_fn) \
+    {                                                     \
+        program_id,                                       \
+        feature_offset,                                   \
+        verify_fn                                         \
+    }
+
 #define NO_CORE_BPF_MIGRATION_CONFIG NULL
 
 #define DEFINE_CORE_BPF_MIGRATION_CONFIG(name, buffer_address, feature_offset, program_id) \
@@ -57,12 +65,23 @@ DEFINE_CORE_BPF_MIGRATION_CONFIG(STATELESS_TO_CORE_BPF_FEATURE_GATE_PROGRAM_CONF
 
 #define FEATURE_PROGRAM_BUILTIN               STATELESS_BUILTIN(&fd_solana_feature_program_id, MIGRATE_STATELESS_TO_CORE_BPF_FEATURE_GATE_PROGRAM_CONFIG)
 
+#define SECP256R1_PROGRAM_BUILTIN             PRECOMPILE(&fd_solana_secp256r1_program_id,          offsetof(fd_features_t, enable_secp256r1_precompile), fd_precompile_secp256r1_verify)
+#define KECCAK_SECP_PROGRAM_BUILTIN           PRECOMPILE(&fd_solana_keccak_secp_256k_program_id,   NO_ENABLE_FEATURE_ID,                                 fd_precompile_secp256k1_verify)
+#define ED25519_SV_PROGRAM_BUILTIN            PRECOMPILE(&fd_solana_ed25519_sig_verify_program_id, NO_ENABLE_FEATURE_ID,                                 fd_precompile_ed25519_verify)
+
 /* https://github.com/anza-xyz/agave/blob/v2.1.0/runtime/src/bank/builtins/mod.rs#L133-L143 */
 static const fd_stateless_builtin_program_t stateless_programs_builtins[] = {
     FEATURE_PROGRAM_BUILTIN
 };
 #define STATELESS_BUILTINS_COUNT (sizeof(stateless_programs_builtins) / sizeof(fd_stateless_builtin_program_t))
 
+static const fd_precompile_program_t precompiles[] = {
+    SECP256R1_PROGRAM_BUILTIN,
+    KECCAK_SECP_PROGRAM_BUILTIN,
+    ED25519_SV_PROGRAM_BUILTIN
+};
+#define PRECOMPILE_PROGRAMS_COUNT (sizeof(precompiles) / sizeof(fd_precompile_program_t))
+
 /* https://github.com/anza-xyz/agave/blob/v2.1.0/runtime/src/bank/builtins/mod.rs#L34-L131 */
 static fd_builtin_program_t const builtin_programs[] = {
     SYSTEM_PROGRAM_BUILTIN,
@@ -250,6 +269,16 @@ fd_num_stateless_builtins( void ) {
   return STATELESS_BUILTINS_COUNT;
 }
 
+fd_precompile_program_t const *
+fd_precompiles( void ) {
+  return precompiles;
+}
+
+ulong
+fd_num_precompiles( void ) {
+  return PRECOMPILE_PROGRAMS_COUNT;
+}
+
 uchar
 fd_is_migrating_builtin_program( fd_exec_txn_ctx_t const * txn_ctx,
                                  fd_pubkey_t const *       pubkey,
diff --git a/src/flamenco/runtime/program/fd_builtin_programs.h b/src/flamenco/runtime/program/fd_builtin_programs.h
index 64c00b3f16..0c26460688 100644
--- a/src/flamenco/runtime/program/fd_builtin_programs.h
+++ b/src/flamenco/runtime/program/fd_builtin_programs.h
@@ -39,6 +39,13 @@ struct fd_stateless_builtin_program {
 };
 typedef struct fd_stateless_builtin_program fd_stateless_builtin_program_t;
 
+struct fd_precompile_program {
+  fd_pubkey_t const * pubkey;
+  ulong               feature_offset;
+  int                 (*verify_fn)(fd_exec_instr_ctx_t*);
+};
+typedef struct fd_precompile_program fd_precompile_program_t;
+
 FD_PROTOTYPES_BEGIN
 
 /* Initialize the builtin program accounts */
@@ -79,6 +86,12 @@ fd_is_migrating_builtin_program( fd_exec_txn_ctx_t const * txn_ctx,
 uchar
 fd_is_non_migrating_builtin_program( fd_pubkey_t const * pubkey );
 
+fd_precompile_program_t const *
+fd_precompiles( void );
+
+ulong
+fd_num_precompiles( void );
+
 FD_PROTOTYPES_END
 
 #endif /* HEADER_fd_src_flamenco_runtime_program_fd_buildin_programs_h */