Line data Source code
1 : #ifndef HEADER_fd_src_flamenco_leaders_fd_leaders_h
2 : #define HEADER_fd_src_flamenco_leaders_fd_leaders_h
3 :
4 : /* fd_leaders provides APIs for the Solana leader schedule.
5 : Logic is compatible with Solana mainnet as of 2023-Jul.
6 :
7 : Every slot is assigned a leader (identified by a "node identity"
8 : public key). The sequence of leaders for all slots in an epoch is
9 : called the "leader schedule". The main responsibility of the leader
10 : is to produce a block for the slot.
11 :
12 : The leader schedule is divided into sched_cnt rotations. Each
13 : rotation spans one or more slots, such that
14 :
15 : slots_per_epoch = sched_cnt * slots_per_rotation
16 :
17 : The leader can only change between rotations. An example leader
18 : schedule looks as follows (where A, B, C are node identities and each
19 : column is a slot, with 4 slots per rotation)
20 :
21 : A A A A B B B B B B B B C C C C A A A A
22 : ^ ^ ^ ^ ^
23 : rotation rotation rotation rotation rotation
24 :
25 : The mainnet epoch duration is quite long (currently 432000 slots, for
26 : more information see fd_sysvar_epoch_schedule.h). To save space, we
27 : dedup pubkeys into a lookup table and only store an index for each
28 : rotation. */
29 :
30 : #include "fd_leaders_base.h"
31 : #include "../../ballet/wsample/fd_wsample.h"
32 :
33 24 : #define FD_ULONG_MAX( a, b ) (__builtin_choose_expr( __builtin_constant_p( a ) & __builtin_constant_p( b ), \
34 24 : ((ulong )(a))>=((ulong )(b)) ? ((ulong )(a)) : ((ulong )(b)), \
35 24 : fd_ulong_max( (a), (b) ) ))
36 :
37 : /* FD_EPOCH_LEADERS_{ALIGN,FOOTPRINT} are compile-time-friendly versions
38 : of the fd_epoch_leaders_{align,footprint} functions. */
39 :
40 0 : #define FD_EPOCH_LEADERS_ALIGN (64UL)
41 : #define FD_EPOCH_LEADERS_FOOTPRINT( pub_cnt, slot_cnt ) \
42 6351 : ( FD_LAYOUT_FINI( FD_LAYOUT_APPEND( FD_LAYOUT_APPEND( \
43 6351 : FD_LAYOUT_INIT, \
44 6351 : alignof(fd_epoch_leaders_t), sizeof(fd_epoch_leaders_t) ), \
45 6351 : alignof(uint), ( \
46 6351 : (slot_cnt+FD_EPOCH_SLOTS_PER_ROTATION-1UL)/FD_EPOCH_SLOTS_PER_ROTATION*sizeof(uint) \
47 6351 : ) ), \
48 6351 : FD_EPOCH_LEADERS_ALIGN ) + \
49 6351 : FD_ULONG_ALIGN_UP( FD_ULONG_MAX( 32UL*((pub_cnt)+1UL), \
50 6351 : FD_WSAMPLE_FOOTPRINT( pub_cnt, 0 ) ), 64UL ) )
51 :
52 6814017 : #define FD_EPOCH_SLOTS_PER_ROTATION (4UL)
53 :
54 : /* fd_epoch_leaders_t contains the leader schedule of a Solana epoch. */
55 :
56 : struct fd_epoch_leaders {
57 : /* This struct contains the schedule for epoch `epoch` which spans
58 : slots [slot0, slot0+slot_cnt). */
59 : ulong epoch;
60 : ulong slot0;
61 : ulong slot_cnt;
62 :
63 : /* pub is a lookup table for node public keys with length pub_cnt */
64 : fd_pubkey_t * pub;
65 : ulong pub_cnt;
66 :
67 : /* sched contains the leader schedule in the form of indexes into
68 : the pub array. For sched_cnt, refer to below. */
69 : uint * sched;
70 : ulong sched_cnt;
71 : };
72 : typedef struct fd_epoch_leaders fd_epoch_leaders_t;
73 :
74 : FD_PROTOTYPES_BEGIN
75 :
76 : /* fd_epoch_leaders_{align,footprint} describe the required footprint
77 : and alignment of the leader schedule object. pub_cnt is the number
78 : of unique public keys. slot_cnt is the number of slots in the
79 : epoch. */
80 :
81 : FD_FN_CONST ulong
82 : fd_epoch_leaders_align( void );
83 :
84 : FD_FN_CONST ulong
85 : fd_epoch_leaders_footprint( ulong pub_cnt,
86 : ulong slot_cnt );
87 :
88 : /* fd_epoch_leaders_new formats a memory region for use as a leader
89 : schedule object. shmem points to the first byte of a memory region
90 : with matching alignment and footprint requirements. The leader
91 : schedule object will contain the leader schedule for epoch `epoch`
92 : which spans slots [slot0, slot0+slot_cnt). `slot0` must be the first
93 : slot in the epoch, but slot_cnt can be less than the length of the
94 : epoch to derive only the first portion of the leader schedule.
95 : pub_cnt is the number of unique public keys in this schedule.
96 : `stakes` points to the first entry of pub_cnt entries of stake
97 : weights sorted by tuple (stake, pubkey) in descending order.
98 : `vote_keyed_lsched` is either 0 or 1, when 1 the leader schedule
99 : is computed by vote accounts (see SIMD-0180).
100 :
101 : If `stakes` does not include all staked nodes, e.g. in the case of an
102 : attack that swamps the network with fake validators, `stakes` should
103 : contain the first `pub_cnt` of them in the normal sort order, and the
104 : sum of the remaining stake must be provided in excluded_stake,
105 : measured in lamports.
106 :
107 : Does NOT retain a read interest in stakes upon return.
108 : The caller is not joined to the object on return. */
109 : void *
110 : fd_epoch_leaders_new( void * shmem,
111 : ulong epoch,
112 : ulong slot0,
113 : ulong slot_cnt,
114 : ulong pub_cnt,
115 : fd_vote_stake_weight_t * stakes, /* indexed [0, pub_cnt) */
116 : ulong excluded_stake,
117 : ulong vote_keyed_lsched );
118 :
119 : /* fd_epoch_leaders_join joins the caller to the leader schedule object.
120 : fd_epoch_leaders_leave undoes an existing join. */
121 :
122 : fd_epoch_leaders_t *
123 : fd_epoch_leaders_join( void * shleaders );
124 :
125 : void *
126 : fd_epoch_leaders_leave( fd_epoch_leaders_t * leaders );
127 :
128 : /* fd_epoch_leaders_delete unformats a memory region and returns owner-
129 : ship back to the caller. */
130 :
131 : void *
132 : fd_epoch_leaders_delete( void * shleaders );
133 :
134 : /* FD_INDETERMINATE_LEADER has base58 encoding
135 : 1111111111indeterminateLeader9QSxFYNqsXA. In hex, this pubkey ends
136 : with 0x0badf00d0badf00d. */
137 6339 : #define FD_INDETERMINATE_LEADER 0x00U,0x00U,0x00U,0x00U,0x00U,0x00U,0x00U,0x00U,0x00U,0x00U,0x99U,0xf6U,0x0fU,0x96U,0x2cU,0xddU,\
138 6339 : 0x38U,0x21U,0xf3U,0x0cU,0x16U,0x1dU,0xe3U,0x0aU,0x0bU,0xadU,0xf0U,0x0dU,0x0bU,0xadU,0xf0U,0x0dU
139 :
140 : /* fd_epoch_leaders_get returns a pointer to the selected public key
141 : given a slot. Returns NULL if slot is not in [slot0, slot0+slot_cnt)
142 : given the values supplied in fd_epoch_leaders_new.
143 :
144 : If a non-zero value was provided for excluded_stake in
145 : fd_epoch_leaders_new and a validator included in the excluded_stake
146 : is the leader for the requested slot, instead of returning the
147 : correct value (which is not known), fd_epoch_leaders_get will return
148 : a pointer to a pubkey with value FD_INDETERMINATE_LEADER. */
149 :
150 : FD_FN_PURE static inline fd_pubkey_t const *
151 : fd_epoch_leaders_get( fd_epoch_leaders_t const * leaders,
152 6801855 : ulong slot ) {
153 6801855 : if( FD_UNLIKELY( leaders==NULL ) ) return NULL;
154 6801855 : ulong slot_delta = slot - leaders->slot0;
155 6801855 : if( FD_UNLIKELY( slot < leaders->slot0 ) ) return NULL;
156 6801723 : if( FD_UNLIKELY( slot_delta>=leaders->slot_cnt ) ) return NULL;
157 6801345 : return (fd_pubkey_t const *)( leaders->pub + leaders->sched[ slot_delta/FD_EPOCH_SLOTS_PER_ROTATION ] );
158 6801723 : }
159 :
160 : FD_PROTOTYPES_END
161 :
162 : #endif /* HEADER_fd_src_flamenco_leaders_fd_leaders_h */
|