Line data    Source code

       1             : #include "fd_instr_info.h"
       2             : #include "../context/fd_exec_txn_ctx.h"
       3             : #include "../../../util/bits/fd_uwide.h"
       4             : 
       5             : void
       6             : fd_instr_info_accumulate_starting_lamports( fd_instr_info_t *         instr,
       7             :                                             fd_exec_txn_ctx_t const * txn_ctx,
       8             :                                             ushort                    idx_in_callee,
       9           0 :                                             ushort                    idx_in_txn ) {
      10           0 :   if( FD_LIKELY( !instr->is_duplicate[ idx_in_callee ] ) ) {
      11           0 :     fd_txn_account_t const * account = &txn_ctx->accounts[ idx_in_txn ];
      12           0 :     if( fd_txn_account_get_meta( account ) ) {
      13           0 :       fd_uwide_inc(
      14           0 :         &instr->starting_lamports_h, &instr->starting_lamports_l,
      15           0 :         instr->starting_lamports_h, instr->starting_lamports_l,
      16           0 :         fd_txn_account_get_lamports( account ) );
      17           0 :     }
      18           0 :   }
      19           0 : }
      20             : 
      21             : void
      22             : fd_instr_info_init_from_txn_instr( fd_instr_info_t *      instr,
      23             :                                    fd_exec_txn_ctx_t *    txn_ctx,
      24           0 :                                    fd_txn_instr_t const * txn_instr ) {
      25             : 
      26           0 :   fd_txn_t const * txn_descriptor = TXN( &txn_ctx->txn );
      27           0 :   uchar *          instr_acc_idxs = (uchar *)txn_ctx->txn.payload + txn_instr->acct_off;
      28             : 
      29           0 :   instr->program_id = txn_instr->program_id;
      30             : 
      31             :   /* See note in fd_instr_info.h. TLDR capping this value at 256 should have
      32             :      literally 0 effect on program execution, down to the error codes. This
      33             :      is purely for the sake of not increasing the overall memory footprint of the
      34             :      transaction context. If this change causes issues, we may need to increase
      35             :      the array sizes in the instr info. */
      36           0 :   instr->acct_cnt = fd_ushort_min( txn_instr->acct_cnt, FD_INSTR_ACCT_MAX );
      37           0 :   instr->data_sz  = txn_instr->data_sz;
      38           0 :   instr->data     = (uchar *)txn_ctx->txn.payload + txn_instr->data_off;
      39             : 
      40           0 :   uchar acc_idx_seen[ FD_INSTR_ACCT_MAX ];
      41           0 :   memset(acc_idx_seen, 0, FD_INSTR_ACCT_MAX);
      42             : 
      43           0 :   for( ushort i=0; i<instr->acct_cnt; i++ ) {
      44           0 :     ushort acc_idx = instr_acc_idxs[i];
      45             : 
      46           0 :     fd_instr_info_setup_instr_account( instr,
      47           0 :                                        acc_idx_seen,
      48           0 :                                        acc_idx,
      49           0 :                                        acc_idx,
      50           0 :                                        i,
      51           0 :                                        (uchar)fd_exec_txn_ctx_account_is_writable_idx( txn_ctx, instr_acc_idxs[i] ),
      52           0 :                                        (uchar)fd_txn_is_signer( txn_descriptor, instr_acc_idxs[i] ) );
      53             : 
      54           0 :   }
      55           0 : }
      56             : 
      57             : int
      58             : fd_instr_info_sum_account_lamports( fd_instr_info_t const * instr,
      59             :                                     fd_exec_txn_ctx_t *     txn_ctx,
      60             :                                     ulong *                 total_lamports_h,
      61           0 :                                     ulong *                 total_lamports_l ) {
      62           0 :   *total_lamports_h = 0UL;
      63           0 :   *total_lamports_l = 0UL;
      64           0 :   for( ulong i=0UL; i<instr->acct_cnt; ++i ) {
      65           0 :     ushort idx_in_txn = instr->accounts[i].index_in_transaction;
      66           0 :     fd_txn_account_t const * account = &txn_ctx->accounts[ idx_in_txn ];
      67             : 
      68           0 :     if( !fd_txn_account_get_meta( account ) ||
      69           0 :         instr->is_duplicate[i] ) {
      70           0 :       continue;
      71           0 :     }
      72             : 
      73             :     /* Perform a checked add on a fd_uwide */
      74           0 :     ulong tmp_total_lamports_h = 0UL;
      75           0 :     ulong tmp_total_lamports_l = 0UL;
      76             : 
      77           0 :     fd_uwide_inc( &tmp_total_lamports_h, &tmp_total_lamports_l, *total_lamports_h, *total_lamports_l,
      78           0 :                   fd_txn_account_get_lamports( account ) );
      79             : 
      80           0 :     if( tmp_total_lamports_h < *total_lamports_h ) {
      81           0 :       return FD_EXECUTOR_INSTR_ERR_ARITHMETIC_OVERFLOW;
      82           0 :     }
      83             : 
      84           0 :     *total_lamports_h = tmp_total_lamports_h;
      85           0 :     *total_lamports_l = tmp_total_lamports_l;
      86           0 :   }
      87             : 
      88           0 :   return FD_EXECUTOR_INSTR_SUCCESS;
      89           0 : }