Revert "solfuzz: remove compute budget program fuzzer"

This reverts commit a3ad5b0.

Author: mjain-jump

src/flamenco/runtime/tests/Local.mk

@@ -1,10 +1,10 @@
-$(call add-hdrs,harness/generated/context.pb.h,harness/generated/elf.pb.h,harness/generated/invoke.pb.h,harness/generated/txn.pb.h,harness/generated/block.pb.h,harness/generated/vm.pb.h,harness/generated/type.pb.h,harness/generated/shred.pb.h harness/generated/metadata.pb.h)
-$(call add-objs,harness/generated/context.pb harness/generated/elf.pb harness/generated/invoke.pb harness/generated/txn.pb harness/generated/block.pb harness/generated/vm.pb harness/generated/type.pb harness/generated/shred.pb harness/generated/metadata.pb,fd_flamenco)
+$(call add-hdrs,harness/generated/context.pb.h,harness/generated/elf.pb.h,harness/generated/invoke.pb.h,harness/generated/txn.pb.h,harness/generated/block.pb.h,harness/generated/vm.pb.h,harness/generated/type.pb.h,harness/generated/shred.pb.h harness/generated/metadata.pb.h harness/generated/pack.pb.h)
+$(call add-objs,harness/generated/context.pb harness/generated/elf.pb harness/generated/invoke.pb harness/generated/txn.pb harness/generated/block.pb harness/generated/vm.pb harness/generated/type.pb harness/generated/shred.pb harness/generated/metadata.pb harness/generated/pack.pb,fd_flamenco)
 
 ifdef FD_HAS_INT128
 ifdef FD_HAS_SECP256K1
-$(call add-hdrs,harness/fd_elf_harness.h harness/fd_instr_harness.h harness/fd_txn_harness.h harness/fd_block_harness.h harness/fd_harness_common.h harness/fd_vm_harness.h harness/fd_types_harness.h)
-$(call add-objs,harness/fd_elf_harness harness/fd_instr_harness harness/fd_txn_harness harness/fd_block_harness harness/fd_harness_common harness/fd_vm_harness harness/fd_types_harness,fd_flamenco_test)
+$(call add-hdrs,harness/fd_elf_harness.h harness/fd_instr_harness.h harness/fd_txn_harness.h harness/fd_block_harness.h harness/fd_harness_common.h harness/fd_vm_harness.h harness/fd_pack_harness.h harness/fd_types_harness.h)
+$(call add-objs,harness/fd_elf_harness harness/fd_instr_harness harness/fd_txn_harness harness/fd_block_harness harness/fd_harness_common harness/fd_vm_harness harness/fd_pack_harness harness/fd_types_harness,fd_flamenco_test)
 $(call add-objs,harness/fd_exec_sol_compat,fd_flamenco_test)
 
 SOL_COMPAT_FLAGS:=-Wl,--undefined=fd_types_vt_by_name

src/flamenco/runtime/tests/harness/fd_exec_sol_compat.c

@@ -14,6 +14,7 @@
 #include "fd_block_harness.h"
 #include "fd_types_harness.h"
 #include "fd_vm_harness.h"
+#include "fd_pack_harness.h"
 #include "fd_elf_harness.h"
 
 #include "generated/elf.pb.h"
@@ -889,6 +890,38 @@ int sol_compat_shred_parse_v1( uchar *       out,
     return !!sol_compat_encode( out, out_sz, output, &fd_exec_test_accepts_shred_t_msg );
 }
 
+int
+sol_compat_pack_compute_budget_v1( uchar *       out,
+                                   ulong *       out_sz,
+                                   uchar const * in,
+                                   ulong         in_sz ) {
+  fd_runtime_fuzz_runner_t * runner = sol_compat_setup_runner( );
+
+  fd_exec_test_pack_compute_budget_context_t input[1] = {0};
+  void * res = sol_compat_decode( &input, in, in_sz, &fd_exec_test_pack_compute_budget_context_t_msg );
+  if( res==NULL ) {
+    sol_compat_cleanup_runner( runner );
+    return 0;
+  }
+
+  int ok = 0;
+  FD_SPAD_FRAME_BEGIN( runner->spad ) {
+  void * output = NULL;
+  sol_compat_execute_wrapper( runner, input, &output, fd_runtime_fuzz_pack_cpb_run );
+
+  if( output ) {
+    ok = !!sol_compat_encode( out, out_sz, output, &fd_exec_test_pack_compute_budget_effects_t_msg );
+  }
+  } FD_SPAD_FRAME_END;
+
+  pb_release( &fd_exec_test_pack_compute_budget_context_t_msg, input );
+  sol_compat_cleanup_runner( runner );
+
+  // Check wksp usage is 0
+  sol_compat_check_wksp_usage();
+  return ok;
+}
+
 int
 sol_compat_type_execute_v1( uchar *       out,
                             ulong *       out_sz,

src/flamenco/runtime/tests/harness/fd_pack_harness.c

@@ -0,0 +1,65 @@
+#include "fd_pack_harness.h"
+#include "../../../../disco/pack/fd_compute_budget_program.h"
+
+ulong
+fd_runtime_fuzz_pack_cpb_run( fd_runtime_fuzz_runner_t * _unused FD_PARAM_UNUSED,
+                              void const *               input_,
+                              void **                    output_,
+                              void *                     output_buf,
+                              ulong                      output_bufsz ){
+  fd_exec_test_pack_compute_budget_context_t const * input  = fd_type_pun_const( input_ );
+  fd_exec_test_pack_compute_budget_effects_t **      output = fd_type_pun( output_ );
+
+  ulong output_end = (ulong) output_buf + output_bufsz;
+  FD_SCRATCH_ALLOC_INIT( l, output_buf );
+
+  fd_exec_test_pack_compute_budget_effects_t * effects =
+    FD_SCRATCH_ALLOC_APPEND( l, alignof(fd_exec_test_pack_compute_budget_effects_t),
+                                sizeof (fd_exec_test_pack_compute_budget_effects_t) );
+  if( FD_UNLIKELY( _l > output_end ) ) {
+    return 0UL;
+  }
+  *effects = (fd_exec_test_pack_compute_budget_effects_t) FD_EXEC_TEST_PACK_COMPUTE_BUDGET_EFFECTS_INIT_ZERO;
+
+  fd_compute_budget_program_state_t cbp_state[1];
+  fd_compute_budget_program_init( cbp_state );
+do {
+
+  int ok = 1;
+  for( ulong i=0UL; i<input->instr_datas_count; ++i ){
+    pb_bytes_array_t * instr_data = input->instr_datas[i];
+    // Reject if any of the instructions fail to parse
+    if( !fd_compute_budget_program_parse( instr_data->bytes, instr_data->size, cbp_state ) ) {
+      ok = 0;
+      break;
+    };
+  }
+
+  if( !ok ) {
+    effects->is_empty = 1;
+    break;
+  }
+  ulong rewards;
+  uint compute_unit_limit;
+  ulong loaded_accounts_data_cost = 0UL;
+  fd_compute_budget_program_finalize( cbp_state,
+                                      input->instr_datas_count,
+                                      &rewards,
+                                      &compute_unit_limit,
+                                      &loaded_accounts_data_cost );
+  effects->rewards = rewards;
+  effects->compute_unit_limit = compute_unit_limit;
+
+  /*  If not set, use defaults.  See:
+      https://github.com/firedancer-io/firedancer/blob/688cb04408cf20b0600d900900cdbebebd181e5b/src/ballet/pack/fd_compute_budget_program.h#L64-L70
+      https://github.com/firedancer-io/agave/blob/838c1952595809a31520ff1603a13f2c9123aa51/runtime-transaction/src/compute_budget_instruction_details.rs#L49-L101
+  */
+  effects->heap_sz             = !!( cbp_state->flags & FD_COMPUTE_BUDGET_PROGRAM_FLAG_SET_HEAP )           ? cbp_state->heap_size : FD_VM_HEAP_DEFAULT;
+  effects->loaded_acct_data_sz = !!( cbp_state->flags & FD_COMPUTE_BUDGET_PROGRAM_FLAG_SET_LOADED_DATA_SZ ) ? cbp_state->loaded_acct_data_sz : FD_COMPUTE_BUDGET_MAX_LOADED_DATA_SZ;
+} while(0);
+
+  ulong actual_end = FD_SCRATCH_ALLOC_FINI( l, 1UL );
+
+  *output = effects;
+  return actual_end - (ulong) output_buf;
+}

src/flamenco/runtime/tests/harness/fd_pack_harness.h

@@ -0,0 +1,18 @@
+#ifndef HEADER_fd_src_flamenco_runtime_tests_fd_pack_harness_h
+#define HEADER_fd_src_flamenco_runtime_tests_fd_pack_harness_h
+
+#include "fd_instr_harness.h"
+#include "generated/pack.pb.h"
+
+FD_PROTOTYPES_BEGIN
+
+ulong
+fd_runtime_fuzz_pack_cpb_run( fd_runtime_fuzz_runner_t * _unused , /* fd_runtime_fuzz_runner_t */
+                              void const *               input_,
+                              void **                    output_,
+                              void *                     output_buf,
+                              ulong                      output_bufsz );
+
+FD_PROTOTYPES_END
+
+#endif /* HEADER_fd_src_flamenco_runtime_tests_fd_pack_harness_h */