Line data    Source code

       1             : #include "fd_features.h"
       2             : #include "../runtime/fd_bank.h"
       3             : #include "../runtime/fd_system_ids.h"
       4             : #include "../runtime/sysvar/fd_sysvar_epoch_schedule.h"
       5             : #include "../accdb/fd_accdb_sync.h"
       6             : 
       7             : FD_STATIC_ASSERT( sizeof  ( fd_feature_t                  )==9UL, layout );
       8             : FD_STATIC_ASSERT( offsetof( fd_feature_t, is_active       )==0UL, layout );
       9             : FD_STATIC_ASSERT( offsetof( fd_feature_t, activation_slot )==1UL, layout );
      10             : 
      11             : fd_feature_t *
      12             : fd_feature_decode( fd_feature_t * feature,
      13             :                    uchar const *  data,
      14          21 :                    ulong          data_sz ) {
      15          21 :   if( FD_UNLIKELY( data_sz < sizeof(fd_feature_t) ) ) return NULL;
      16          18 :   *feature = FD_LOAD( fd_feature_t, data );
      17          18 :   if( FD_UNLIKELY( feature->is_active>1 ) ) return NULL;
      18          18 :   return feature;
      19          18 : }
      20             : 
      21             : void
      22           3 : fd_features_enable_all( fd_features_t * f ) {
      23           3 :   for( fd_feature_id_t const * id = fd_feature_iter_init();
      24         810 :        !fd_feature_iter_done( id );
      25         807 :        id = fd_feature_iter_next( id ) ) {
      26         807 :     fd_features_set( f, id, 0UL );
      27         807 :   }
      28           3 : }
      29             : 
      30             : void
      31         540 : fd_features_disable_all( fd_features_t * f ) {
      32         540 :   for( fd_feature_id_t const * id = fd_feature_iter_init();
      33      145800 :        !fd_feature_iter_done( id );
      34      145260 :        id = fd_feature_iter_next( id ) ) {
      35      145260 :     fd_features_set( f, id, FD_FEATURE_DISABLED );
      36      145260 :   }
      37         540 : }
      38             : 
      39             : void
      40         219 : fd_features_enable_cleaned_up( fd_features_t * f ) {
      41         219 :   for( fd_feature_id_t const * id = fd_feature_iter_init();
      42       59130 :        !fd_feature_iter_done( id );
      43       58911 :        id = fd_feature_iter_next( id ) ) {
      44       58911 :     if( FD_LIKELY( id->cleaned_up ) ) {
      45       43362 :       fd_features_set( f, id, 0UL );
      46       43362 :     } else {
      47       15549 :       fd_features_set( f, id, FD_FEATURE_DISABLED );
      48       15549 :     }
      49       58911 :   }
      50         219 : }
      51             : 
      52             : void
      53          18 : fd_features_enable_one_offs( fd_features_t * f, char const * * one_offs, uint one_offs_cnt, ulong slot ) {
      54          18 :   uchar pubkey[32];
      55          72 :   for( uint i=0U; i<one_offs_cnt; i++ ) {
      56          54 :     fd_base58_decode_32( one_offs[i], pubkey );
      57          54 :     for( fd_feature_id_t const * id = fd_feature_iter_init();
      58       13230 :          !fd_feature_iter_done( id );
      59       13230 :          id = fd_feature_iter_next( id ) ) {
      60       13230 :       if( !memcmp( &id->id, pubkey, sizeof(fd_pubkey_t) ) ) {
      61          54 :         fd_features_set( f, id, slot );
      62          54 :         break;
      63          54 :       }
      64       13230 :     }
      65          54 :   }
      66          18 : }
      67             : 
      68             : static void
      69             : fd_feature_restore( fd_bank_t *               bank,
      70             :                     fd_accdb_user_t *         accdb,
      71             :                     fd_funk_txn_xid_t const * xid,
      72             :                     fd_feature_id_t const *   id,
      73       30666 :                     fd_pubkey_t const *       addr ) {
      74             : 
      75       30666 :   fd_features_t *             features       = &bank->f.features;
      76       30666 :   fd_epoch_schedule_t const * epoch_schedule = &bank->f.epoch_schedule;
      77       30666 :   ulong                       slot           = bank->f.slot;
      78             : 
      79             :   /* Skip reverted features */
      80       30666 :   if( FD_UNLIKELY( id->reverted ) ) return;
      81             : 
      82       29412 :   fd_accdb_ro_t ro[1];
      83       29412 :   if( FD_UNLIKELY( !fd_accdb_open_ro( accdb, ro, xid, addr ) ) )  {
      84       29388 :     return;
      85       29388 :   }
      86             : 
      87             :   /* Skip accounts that are not owned by the feature program
      88             :      https://github.com/anza-xyz/solana-sdk/blob/6512aca61167088ce10f2b545c35c9bcb1400e70/feature-gate-interface/src/lib.rs#L42-L44 */
      89          24 :   if( FD_UNLIKELY( !fd_pubkey_eq( fd_accdb_ref_owner( ro ), &fd_solana_feature_program_id ) ) ) {
      90           6 :     fd_accdb_close_ro( accdb, ro );
      91           6 :     return;
      92           6 :   }
      93             : 
      94             :   /* Account data size must be >= FD_FEATURE_SIZEOF (9 bytes)
      95             :      https://github.com/anza-xyz/solana-sdk/blob/6512aca61167088ce10f2b545c35c9bcb1400e70/feature-gate-interface/src/lib.rs#L45-L47 */
      96          18 :   if( FD_UNLIKELY( fd_accdb_ref_data_sz( ro ) < sizeof(fd_feature_t) ) ) {
      97           6 :     fd_accdb_close_ro( accdb, ro );
      98           6 :     return;
      99           6 :   }
     100             : 
     101             :   /* Deserialize the feature account data
     102             :      https://github.com/anza-xyz/solana-sdk/blob/6512aca61167088ce10f2b545c35c9bcb1400e70/feature-gate-interface/src/lib.rs#L48-L50 */
     103          12 :   fd_feature_t feature[1];
     104          12 :   if( FD_UNLIKELY( !fd_feature_decode(
     105          12 :       feature,
     106          12 :       fd_accdb_ref_data_const( ro ),
     107          12 :       fd_accdb_ref_data_sz   ( ro ) ) ) ) {
     108           0 :     fd_accdb_close_ro( accdb, ro );
     109           0 :     return;
     110           0 :   }
     111          12 :   fd_accdb_close_ro( accdb, ro );
     112             : 
     113          12 :   FD_BASE58_ENCODE_32_BYTES( addr->uc, addr_b58 );
     114          12 :   if( feature->is_active ) {
     115           9 :     FD_LOG_DEBUG(( "feature %s activated at slot %lu", addr_b58, feature->activation_slot ));
     116           9 :     fd_features_set( features, id, feature->activation_slot );
     117           9 :   } else if( fd_slot_to_epoch( epoch_schedule, slot, NULL )!=fd_slot_to_epoch( epoch_schedule, slot+1UL, NULL ) ) {
     118           3 :     ulong activation_slot = slot+1UL;
     119           3 :     FD_LOG_DEBUG(( "feature %s pending, pre-populating activation at slot %lu", addr_b58, activation_slot ));
     120           3 :     fd_features_set( features, id, activation_slot );
     121           3 :   } else {
     122           0 :     FD_LOG_DEBUG(( "feature %s not activated at slot %lu", addr_b58, feature->activation_slot ));
     123           0 :   }
     124          12 : }
     125             : 
     126             : void
     127             : fd_features_restore( fd_bank_t *               bank,
     128             :                      fd_accdb_user_t *         accdb,
     129         114 :                      fd_funk_txn_xid_t const * xid ) {
     130             : 
     131         114 :   for( fd_feature_id_t const * id = fd_feature_iter_init();
     132       30780 :                                    !fd_feature_iter_done( id );
     133       30666 :                                id = fd_feature_iter_next( id ) ) {
     134       30666 :     fd_feature_restore( bank, accdb, xid, id, &id->id );
     135       30666 :   }
     136         114 : }