Line data    Source code

       1             : #include "fd_exec_instr_ctx.h"
       2             : #include "../fd_runtime.h"
       3             : #include "../fd_borrowed_account.h"
       4             : 
       5             : int
       6             : fd_exec_instr_ctx_find_idx_of_instr_account( fd_exec_instr_ctx_t const * ctx,
       7         126 :                                              fd_pubkey_t const *         pubkey ) {
       8         252 :   for( int i=0; i<ctx->instr->acct_cnt; i++ ) {
       9         252 :     ushort idx_in_txn = ctx->instr->accounts[ i ].index_in_transaction;
      10         252 :     if( memcmp( pubkey->uc, ctx->txn_out->accounts.keys[ idx_in_txn ].uc, sizeof(fd_pubkey_t) )==0 ) {
      11         126 :       return i;
      12         126 :     }
      13         252 :   }
      14           0 :   return -1;
      15         126 : }
      16             : 
      17             : int
      18             : fd_exec_instr_ctx_get_key_of_account_at_index( fd_exec_instr_ctx_t const * ctx,
      19             :                                                ushort                      idx_in_instr,
      20           6 :                                                fd_pubkey_t const * *       key ) {
      21           6 :   ushort idx_in_txn;
      22           6 :   int err = fd_exec_instr_ctx_get_index_of_instr_account_in_transaction( ctx,
      23           6 :                                                                          idx_in_instr,
      24           6 :                                                                          &idx_in_txn );
      25           6 :   if( FD_UNLIKELY( err ) ) {
      26           0 :     return err;
      27           0 :   }
      28             : 
      29           6 :   return fd_runtime_get_key_of_account_at_index( ctx->txn_out,
      30           6 :                                                       idx_in_txn,
      31           6 :                                                       key );
      32           6 : }
      33             : 
      34             : int
      35             : fd_exec_instr_ctx_get_last_program_key( fd_exec_instr_ctx_t const * ctx,
      36         630 :                                         fd_pubkey_t const * *       key ) {
      37         630 :   return fd_runtime_get_key_of_account_at_index( ctx->txn_out,
      38         630 :                                                       ctx->instr->program_id,
      39         630 :                                                       key );
      40         630 : }
      41             : 
      42             : int
      43             : fd_exec_instr_ctx_try_borrow_account( fd_exec_instr_ctx_t const * ctx,
      44             :                                       ushort                      idx_in_instr,
      45             :                                       ushort                      idx_in_txn,
      46         846 :                                       fd_borrowed_account_t *     account ) {
      47             :   /* Get the account from the transaction context using idx_in_txn.
      48             :      https://github.com/anza-xyz/agave/blob/v2.1.14/sdk/src/transaction_context.rs#L600-L602 */
      49         846 :   int err = fd_runtime_get_account_at_index( ctx->txn_in,
      50         846 :                                              ctx->txn_out,
      51         846 :                                              idx_in_txn,
      52         846 :                                              NULL );
      53         846 :   if( FD_UNLIKELY( err ) ) {
      54             :     /* Return a MissingAccount error if the account is not found.
      55             :        https://github.com/anza-xyz/agave/blob/v2.1.14/sdk/src/transaction_context.rs#L603 */
      56           0 :     FD_TXN_ERR_FOR_LOG_INSTR( ctx->txn_out, FD_EXECUTOR_INSTR_ERR_MISSING_ACC, ctx->txn_out->err.exec_err_idx );
      57           0 :     return FD_EXECUTOR_INSTR_ERR_MISSING_ACC;
      58           0 :   }
      59             : 
      60         846 :   fd_account_meta_t * meta = ctx->txn_out->accounts.account[idx_in_txn].meta;
      61             : 
      62             :   /* Return an AccountBorrowFailed error if the write is not acquirable.
      63             :      https://github.com/anza-xyz/agave/blob/v2.1.14/sdk/src/transaction_context.rs#L605 */
      64         846 :   if( FD_UNLIKELY( ctx->runtime->accounts.refcnt[idx_in_txn]!=0UL ) ) {
      65           0 :     return FD_EXECUTOR_INSTR_ERR_ACC_BORROW_FAILED;
      66           0 :   }
      67         846 :   ctx->runtime->accounts.refcnt[idx_in_txn]++;
      68             : 
      69             :   /* Create a BorrowedAccount upon success.
      70             :      https://github.com/anza-xyz/agave/blob/v2.1.14/sdk/src/transaction_context.rs#L606 */
      71         846 :   fd_borrowed_account_init( account,
      72         846 :                             &ctx->txn_out->accounts.keys[idx_in_txn],
      73         846 :                             meta,
      74         846 :                             ctx,
      75         846 :                             idx_in_instr,
      76         846 :                             &ctx->runtime->accounts.refcnt[idx_in_txn] );
      77         846 :   return FD_EXECUTOR_INSTR_SUCCESS;
      78         846 : }
      79             : 
      80             : int
      81             : fd_exec_instr_ctx_try_borrow_instr_account( fd_exec_instr_ctx_t const * ctx,
      82             :                                             ushort                      idx,
      83         840 :                                             fd_borrowed_account_t *     account ) {
      84             :   /* Find the index of the account in the transaction context.
      85             :      https://github.com/anza-xyz/agave/blob/v2.1.14/sdk/src/transaction_context.rs#L649-L650 */
      86         840 :   ushort idx_in_txn;
      87         840 :   int err = fd_exec_instr_ctx_get_index_of_instr_account_in_transaction( ctx,
      88         840 :                                                                          idx,
      89         840 :                                                                          &idx_in_txn );
      90         840 :   if( FD_UNLIKELY( err ) ) {
      91           0 :     return err;
      92           0 :   }
      93             : 
      94         840 :   return fd_exec_instr_ctx_try_borrow_account( ctx,
      95         840 :                                                idx,
      96         840 :                                                idx_in_txn,
      97         840 :                                                account );
      98         840 : }
      99             : 
     100             : int
     101             : fd_exec_instr_ctx_try_borrow_instr_account_with_key( fd_exec_instr_ctx_t const * ctx,
     102             :                                                      fd_pubkey_t const *         pubkey,
     103           0 :                                                      fd_borrowed_account_t *     account ) {
     104           0 :   for( ushort i=0; i<ctx->instr->acct_cnt; i++ ) {
     105           0 :     ushort idx_in_txn = ctx->instr->accounts[ i ].index_in_transaction;
     106           0 :     if( memcmp( pubkey->uc, ctx->txn_out->accounts.keys[ idx_in_txn ].uc, sizeof(fd_pubkey_t) )==0 ) {
     107           0 :       return fd_exec_instr_ctx_try_borrow_instr_account( ctx, i, account );
     108           0 :     }
     109           0 :   }
     110             : 
     111             :   /* Return a NotEnoughAccountKeys error if the account is not found
     112             :      in the instruction context to match the error code returned by
     113             :      fd_exec_instr_ctx_try_borrow_instr_account. */
     114           0 :   return FD_EXECUTOR_INSTR_ERR_MISSING_ACC;
     115           0 : }
     116             : 
     117             : int
     118             : fd_exec_instr_ctx_try_borrow_last_program_account( fd_exec_instr_ctx_t const * ctx,
     119           6 :                                                    fd_borrowed_account_t *     account ) {
     120             :   /* The index_in_instruction for a borrowed program account is invalid,
     121             :      so it is set to a sentinel value of USHORT_MAX. */
     122           6 :   return fd_exec_instr_ctx_try_borrow_account( ctx,
     123           6 :                                                USHORT_MAX,
     124           6 :                                                ctx->instr->program_id,
     125           6 :                                                account );
     126           6 : }
     127             : 
     128             : int
     129             : fd_exec_instr_ctx_get_signers( fd_exec_instr_ctx_t const * ctx,
     130             :                                fd_pubkey_t const *         signers[static FD_TXN_SIG_MAX],
     131          12 :                                ulong *                     signers_cnt ) {
     132          12 :   ulong j = 0UL;
     133          60 :   for( ushort i=0; i<ctx->instr->acct_cnt; i++ ) {
     134          48 :     if( fd_instr_acc_is_signer_idx( ctx->instr, i, NULL ) ) {
     135          24 :       ushort idx_in_txn = ctx->instr->accounts[i].index_in_transaction;
     136          24 :       fd_pubkey_t const * pubkey = NULL;
     137          24 :       int err = fd_runtime_get_key_of_account_at_index( ctx->txn_out,
     138          24 :                                                         idx_in_txn,
     139          24 :                                                         &pubkey );
     140          24 :       if( FD_UNLIKELY( err ) ) {
     141           0 :         return err;
     142           0 :       }
     143             : 
     144             :       /* Skip if duplicate signer */
     145          24 :       if( FD_UNLIKELY( fd_signers_contains( signers, j, pubkey ) ) ) {
     146           0 :         continue;
     147           0 :       }
     148             : 
     149             :       /* This should never be possible */
     150          24 :       if( FD_UNLIKELY( j>=FD_TXN_SIG_MAX ) ) {
     151           0 :         FD_LOG_CRIT(( "invariant violation: too many signers (cnt=%lu, FD_TXN_SIG_MAX=%lu)", j, (ulong)FD_TXN_SIG_MAX ));
     152           0 :       }
     153             : 
     154          24 :       signers[j++] = pubkey;
     155          24 :     }
     156          48 :   }
     157             : 
     158          12 :   *signers_cnt = j;
     159          12 :   return FD_EXECUTOR_INSTR_SUCCESS;
     160          12 : }
     161             : 
     162             : int
     163             : fd_exec_instr_ctx_any_signed( fd_exec_instr_ctx_t const * ctx,
     164          24 :                               fd_pubkey_t const *         pubkey ) {
     165          24 :   int is_signer = 0;
     166          48 :   for( ushort j=0; j<ctx->instr->acct_cnt; j++ ) {
     167          24 :     ushort idx_in_txn = ctx->instr->accounts[ j ].index_in_transaction;
     168          24 :     is_signer |=
     169             :         ( ( !!fd_instr_acc_is_signer_idx( ctx->instr, j, NULL ) ) &
     170          24 :         ( 0==memcmp( pubkey->key, ctx->txn_out->accounts.keys[ idx_in_txn ].key, sizeof(fd_pubkey_t) ) ) );
     171          24 :   }
     172          24 :   return is_signer;
     173          24 : }