Line data Source code
1 : #include "fd_funk_private.h"
2 : #include "../util/racesan/fd_racesan_target.h"
3 :
4 : /* Provide the actual record map implementation */
5 :
6 : #define POOL_NAME fd_funk_rec_pool
7 262362 : #define POOL_ELE_T fd_funk_rec_t
8 : #define POOL_IDX_T uint
9 259107 : #define POOL_NEXT map_next
10 : #define POOL_IMPL_STYLE 2
11 : #define POOL_LAZY 1
12 : #include "../util/tmpl/fd_pool_para.c"
13 :
14 : #define MAP_NAME fd_funk_rec_map
15 130339 : #define MAP_ELE_T fd_funk_rec_t
16 0 : #define MAP_KEY_T fd_funk_xid_key_pair_t
17 129191 : #define MAP_KEY pair
18 : #define MAP_KEY_EQ(k0,k1) fd_funk_xid_key_pair_eq((k0),(k1))
19 : #define MAP_KEY_HASH(k0,seed) fd_funk_xid_key_pair_hash((k0),(seed))
20 131549 : #define MAP_IDX_T uint
21 262203 : #define MAP_NEXT map_next
22 12 : #define MAP_MAGIC (0xf173da2ce77ecdb0UL) /* Firedancer rec db version 0 */
23 520530 : #define MAP_CNT_WIDTH FD_FUNK_REC_MAP_CNT_WIDTH
24 : #define MAP_IMPL_STYLE 2
25 : #define MAP_PEDANTIC 1
26 : #include "../util/tmpl/fd_map_chain_para.c"
27 :
28 : static fd_funk_txn_t *
29 : fd_funk_rec_txn_borrow( fd_funk_t const * funk,
30 : fd_funk_txn_xid_t const * xid,
31 0 : fd_funk_txn_map_query_t * query ) {
32 0 : memset( query, 0, sizeof(fd_funk_txn_map_query_t) );
33 0 : if( fd_funk_txn_xid_eq( xid, funk->shmem->last_publish ) ) return NULL;
34 :
35 0 : fd_funk_txn_map_query_t txn_query[1];
36 0 : for(;;) {
37 0 : int txn_query_err = fd_funk_txn_map_query_try( funk->txn_map, xid, NULL, txn_query, 0 );
38 0 : if( FD_UNLIKELY( txn_query_err==FD_MAP_ERR_AGAIN ) ) continue;
39 0 : if( FD_UNLIKELY( txn_query_err!=FD_MAP_SUCCESS ) ) {
40 0 : FD_LOG_CRIT(( "fd_funk_rec op failed: txn_map_query_try(%lu:%lu) error %i-%s",
41 0 : xid->ul[0], xid->ul[1],
42 0 : txn_query_err, fd_map_strerror( txn_query_err ) ));
43 0 : }
44 0 : break;
45 0 : }
46 0 : fd_funk_txn_t * txn = fd_funk_txn_map_query_ele( txn_query );
47 0 : uint txn_state = FD_VOLATILE_CONST( txn->state );
48 0 : if( FD_UNLIKELY( txn_state!=FD_FUNK_TXN_STATE_ACTIVE ) ) {
49 0 : FD_LOG_CRIT(( "fd_funk_rec op failed: txn %p %lu:%lu state is %u-%s",
50 0 : (void *)txn, xid->ul[0], xid->ul[1],
51 0 : txn_state, fd_funk_txn_state_str( txn_state ) ));
52 0 : }
53 0 : return txn;
54 0 : }
55 :
56 : static void
57 0 : fd_funk_rec_txn_release( fd_funk_txn_map_query_t const * query ) {
58 0 : if( !query->ele ) return;
59 0 : if( FD_UNLIKELY( fd_funk_txn_map_query_test( query )!=FD_MAP_SUCCESS ) ) {
60 0 : FD_LOG_CRIT(( "fd_funk_rec_txn_release: fd_funk_txn_map_query_test failed (data race detected?)" ));
61 0 : }
62 0 : }
63 :
64 : fd_funk_rec_t *
65 : fd_funk_rec_query_try( fd_funk_t * funk,
66 : fd_funk_txn_xid_t const * xid,
67 : fd_funk_rec_key_t const * key,
68 0 : fd_funk_rec_query_t * query ) {
69 0 : if( FD_UNLIKELY( !funk ) ) FD_LOG_CRIT(( "NULL funk" ));
70 0 : if( FD_UNLIKELY( !xid ) ) FD_LOG_CRIT(( "NULL xid" ));
71 0 : if( FD_UNLIKELY( !key ) ) FD_LOG_CRIT(( "NULL key" ));
72 0 : if( FD_UNLIKELY( !query ) ) FD_LOG_CRIT(( "NULL query" ));
73 :
74 0 : fd_funk_xid_key_pair_t pair[1];
75 0 : if( FD_UNLIKELY( fd_funk_txn_xid_eq( xid, funk->shmem->last_publish ) ) ) {
76 0 : fd_funk_txn_xid_set_root( pair->xid );
77 0 : } else {
78 0 : fd_funk_txn_xid_copy( pair->xid, xid );
79 0 : }
80 0 : fd_funk_rec_key_copy( pair->key, key );
81 :
82 0 : for(;;) {
83 0 : int err = fd_funk_rec_map_query_try( funk->rec_map, pair, NULL, query, 0 );
84 0 : if( err == FD_MAP_SUCCESS ) break;
85 0 : if( err == FD_MAP_ERR_KEY ) return NULL;
86 0 : if( err == FD_MAP_ERR_AGAIN ) continue;
87 0 : FD_LOG_CRIT(( "query returned err %d", err ));
88 0 : }
89 0 : return fd_funk_rec_map_query_ele( query );
90 0 : }
91 :
92 : fd_funk_rec_t *
93 : fd_funk_rec_prepare( fd_funk_t * funk,
94 : fd_funk_txn_xid_t const * xid,
95 : fd_funk_rec_key_t const * key,
96 : fd_funk_rec_prepare_t * prepare,
97 0 : int * opt_err ) {
98 0 : if( FD_UNLIKELY( !funk ) ) FD_LOG_CRIT(( "NULL funk" ));
99 0 : if( FD_UNLIKELY( !xid ) ) FD_LOG_CRIT(( "NULL xid" ));
100 0 : if( FD_UNLIKELY( !prepare ) ) FD_LOG_CRIT(( "NULL prepare" ));
101 :
102 0 : fd_funk_txn_map_query_t txn_query[1];
103 0 : fd_funk_txn_t * txn = fd_funk_rec_txn_borrow( funk, xid, txn_query );
104 :
105 0 : memset( prepare, 0, sizeof(fd_funk_rec_prepare_t) );
106 :
107 0 : if( !txn ) { /* Modifying last published */
108 0 : if( FD_UNLIKELY( fd_funk_last_publish_is_frozen( funk ) ) ) {
109 0 : fd_int_store_if( !!opt_err, opt_err, FD_FUNK_ERR_FROZEN );
110 0 : fd_funk_rec_txn_release( txn_query );
111 0 : return NULL;
112 0 : }
113 0 : } else {
114 0 : if( FD_UNLIKELY( fd_funk_txn_is_frozen( txn ) ) ) {
115 0 : fd_int_store_if( !!opt_err, opt_err, FD_FUNK_ERR_FROZEN );
116 0 : fd_funk_rec_txn_release( txn_query );
117 0 : return NULL;
118 0 : }
119 0 : }
120 :
121 0 : fd_funk_rec_t * rec = prepare->rec = fd_funk_rec_pool_acquire( funk->rec_pool, NULL, 1, opt_err );
122 0 : if( opt_err && *opt_err == FD_POOL_ERR_CORRUPT ) {
123 0 : FD_LOG_CRIT(( "corrupt element returned from funk rec pool" ));
124 0 : }
125 0 : if( FD_UNLIKELY( !rec ) ) {
126 0 : fd_int_store_if( !!opt_err, opt_err, FD_FUNK_ERR_REC );
127 0 : fd_funk_rec_txn_release( txn_query );
128 0 : return rec;
129 0 : }
130 :
131 0 : fd_funk_val_init( rec );
132 0 : if( txn == NULL ) {
133 0 : fd_funk_txn_xid_set_root( rec->pair.xid );
134 0 : } else {
135 0 : fd_funk_txn_xid_copy( rec->pair.xid, &txn->xid );
136 0 : prepare->rec_head_idx = &txn->rec_head_idx;
137 0 : prepare->rec_tail_idx = &txn->rec_tail_idx;
138 0 : }
139 0 : fd_funk_rec_key_copy( rec->pair.key, key );
140 0 : rec->tag = 0;
141 0 : rec->prev_idx = FD_FUNK_REC_IDX_NULL;
142 0 : rec->next_idx = FD_FUNK_REC_IDX_NULL;
143 0 : fd_funk_rec_txn_release( txn_query );
144 0 : return rec;
145 0 : }
146 :
147 : #if FD_HAS_ATOMIC
148 :
149 : static void
150 : fd_funk_rec_push_tail( fd_funk_t * funk,
151 128259 : fd_funk_rec_prepare_t * prepare ) {
152 128259 : fd_funk_rec_t * rec = prepare->rec;
153 128259 : uint rec_idx = (uint)( rec - funk->rec_pool->ele );
154 128259 : uint * rec_head_idx = prepare->rec_head_idx;
155 128259 : uint * rec_tail_idx = prepare->rec_tail_idx;
156 :
157 128259 : for(;;) {
158 :
159 : /* Doubly linked list append. Robust in the event of concurrent
160 : publishes. Iteration during publish not supported. Sequence:
161 : - Identify tail element
162 : - Set new element's prev and next pointers
163 : - Set tail element's next pointer
164 : - Set tail pointer */
165 :
166 128252 : uint rec_prev_idx = FD_VOLATILE_CONST( *rec_tail_idx );
167 128252 : rec->prev_idx = rec_prev_idx;
168 128252 : FD_COMPILER_MFENCE();
169 :
170 128252 : uint * next_idx_p;
171 128252 : if( fd_funk_rec_idx_is_null( rec_prev_idx ) ) {
172 5821 : next_idx_p = rec_head_idx;
173 122431 : } else {
174 122431 : next_idx_p = &funk->rec_pool->ele[ rec_prev_idx ].next_idx;
175 122431 : }
176 :
177 128252 : fd_racesan_hook( "funk_rec_push_tail:next_cas" );
178 128252 : if( FD_UNLIKELY( !__sync_bool_compare_and_swap( next_idx_p, FD_FUNK_REC_IDX_NULL, rec_idx ) ) ) {
179 : /* Another thread beat us to the punch */
180 0 : FD_SPIN_PAUSE();
181 0 : continue;
182 0 : }
183 :
184 128252 : fd_racesan_hook( "funk_rec_push_tail:tail_cas" );
185 128252 : if( FD_UNLIKELY( !__sync_bool_compare_and_swap( rec_tail_idx, rec_prev_idx, rec_idx ) ) ) {
186 : /* This CAS is guaranteed to succeed if the previous CAS passed. */
187 0 : FD_LOG_CRIT(( "Irrecoverable data race encountered while appending to txn rec list (invariant violation?): cas(%p,%u,%u)",
188 0 : (void *)rec_tail_idx, rec_prev_idx, rec_idx ));
189 0 : }
190 :
191 128252 : break;
192 128252 : }
193 128259 : }
194 :
195 : #else
196 :
197 : static void
198 : fd_funk_rec_push_tail( fd_funk_t * funk,
199 : fd_funk_rec_prepare_t * prepare ) {
200 : fd_funk_rec_t * rec = prepare->rec;
201 : uint rec_idx = (uint)( rec - funk->rec_pool->ele );
202 : uint * rec_head_idx = prepare->rec_head_idx;
203 : uint * rec_tail_idx = prepare->rec_tail_idx;
204 : uint rec_prev_idx = *rec_tail_idx;
205 : *rec_tail_idx = rec_idx;
206 : rec->prev_idx = rec_prev_idx;
207 : rec->next_idx = FD_FUNK_REC_IDX_NULL;
208 : if( fd_funk_rec_idx_is_null( rec_prev_idx ) ) {
209 : *rec_head_idx = rec_idx;
210 : } else {
211 : funk->rec_pool->ele[ rec_prev_idx ].next_idx = rec_idx;
212 : }
213 : }
214 :
215 : #endif
216 :
217 : void
218 : fd_funk_rec_publish( fd_funk_t * funk,
219 128254 : fd_funk_rec_prepare_t * prepare ) {
220 128254 : fd_funk_rec_t * rec = prepare->rec;
221 128254 : rec->prev_idx = FD_FUNK_REC_IDX_NULL;
222 128254 : rec->next_idx = FD_FUNK_REC_IDX_NULL;
223 :
224 128269 : if( prepare->rec_head_idx ) {
225 128269 : fd_funk_rec_push_tail( funk, prepare );
226 128269 : }
227 :
228 128254 : fd_racesan_hook( "funk_rec_publish:map_insert" );
229 128254 : int insert_err = fd_funk_rec_map_insert( funk->rec_map, rec, FD_MAP_FLAG_BLOCKING );
230 128254 : if( insert_err ) {
231 0 : FD_LOG_CRIT(( "fd_funk_rec_map_insert failed (%i-%s)", insert_err, fd_map_strerror( insert_err ) ));
232 0 : }
233 128254 : }
234 :
235 : int
236 0 : fd_funk_rec_verify( fd_funk_t * funk ) {
237 0 : fd_funk_rec_map_t * rec_map = funk->rec_map;
238 0 : fd_funk_rec_pool_t * rec_pool = funk->rec_pool;
239 0 : ulong rec_max = fd_funk_rec_pool_ele_max( rec_pool );
240 :
241 0 : # define TEST(c) do { \
242 0 : if( FD_UNLIKELY( !(c) ) ) { FD_LOG_WARNING(( "FAIL: %s", #c )); return FD_FUNK_ERR_INVAL; } \
243 0 : } while(0)
244 :
245 0 : TEST( !fd_funk_rec_map_verify( rec_map ) );
246 0 : TEST( !fd_funk_rec_pool_verify( rec_pool ) );
247 :
248 : /* Iterate (again) over all records in use */
249 :
250 0 : fd_funk_rec_map_shmem_private_chain_t * chain_tbl =
251 0 : fd_funk_rec_map_shmem_private_chain( rec_map->map, 0UL );
252 0 : ulong chain_cnt = fd_funk_rec_map_chain_cnt( rec_map );
253 0 : for( ulong chain_idx=0UL; chain_idx<chain_cnt; chain_idx++ ) {
254 0 : ulong chain_ele_cnt = fd_funk_rec_map_private_vcnt_cnt( chain_tbl[ chain_idx ].ver_cnt );
255 0 : ulong chain_ele_idx = 0UL;
256 0 : for( fd_funk_rec_map_iter_t iter = fd_funk_rec_map_iter( rec_map, chain_idx );
257 0 : !fd_funk_rec_map_iter_done( iter );
258 0 : iter = fd_funk_rec_map_iter_next( iter ), chain_ele_idx++ ) {
259 0 : fd_funk_rec_t const * rec = fd_funk_rec_map_iter_ele_const( iter );
260 :
261 : /* Make sure every record either links up with the last published
262 : transaction or an in-prep transaction and the flags are sane. */
263 :
264 0 : fd_funk_txn_xid_t const * xid = fd_funk_rec_xid( rec );
265 :
266 0 : if( fd_funk_txn_xid_eq_root( xid ) ) { /* This is a record from the last published transaction */
267 :
268 0 : TEST( fd_funk_txn_xid_eq_root( xid ) );
269 : /* No record linked list at the root txn */
270 0 : TEST( fd_funk_rec_idx_is_null( rec->prev_idx ) );
271 0 : TEST( fd_funk_rec_idx_is_null( rec->next_idx ) );
272 :
273 0 : } else { /* This is a record from an in-prep transaction */
274 :
275 0 : fd_funk_txn_map_query_t query[1];
276 0 : TEST( fd_funk_txn_map_query_try( funk->txn_map, xid, NULL, query, 0 )==FD_MAP_SUCCESS );
277 :
278 0 : }
279 0 : }
280 0 : TEST( chain_ele_idx==chain_ele_cnt );
281 0 : }
282 :
283 : /* Clear record tags and then verify membership */
284 :
285 0 : for( ulong rec_idx=0UL; rec_idx<rec_max; rec_idx++ ) rec_pool->ele[ rec_idx ].tag = 0;
286 :
287 0 : do {
288 0 : fd_funk_all_iter_t iter[1];
289 0 : for( fd_funk_all_iter_new( funk, iter ); !fd_funk_all_iter_done( iter ); fd_funk_all_iter_next( iter ) ) {
290 0 : fd_funk_rec_t * rec = fd_funk_all_iter_ele( iter );
291 0 : if( fd_funk_txn_xid_eq_root( rec->pair.xid ) ) {
292 0 : TEST( !rec->tag );
293 0 : rec->tag = 1;
294 0 : }
295 0 : }
296 :
297 0 : fd_funk_txn_all_iter_t txn_iter[1];
298 0 : for( fd_funk_txn_all_iter_new( funk, txn_iter ); !fd_funk_txn_all_iter_done( txn_iter ); fd_funk_txn_all_iter_next( txn_iter ) ) {
299 0 : fd_funk_txn_t const * txn = fd_funk_txn_all_iter_ele_const( txn_iter );
300 :
301 0 : uint rec_idx = txn->rec_head_idx;
302 0 : while( !fd_funk_rec_idx_is_null( rec_idx ) ) {
303 0 : TEST( (rec_idx<rec_max) && !rec_pool->ele[ rec_idx ].tag );
304 0 : rec_pool->ele[ rec_idx ].tag = 1;
305 0 : fd_funk_rec_query_t query[1];
306 0 : fd_funk_rec_t const * rec2 = fd_funk_rec_query_try( funk, &txn->xid, rec_pool->ele[ rec_idx ].pair.key, query );
307 0 : TEST( rec2 == rec_pool->ele + rec_idx );
308 0 : uint next_idx = rec_pool->ele[ rec_idx ].next_idx;
309 0 : if( !fd_funk_rec_idx_is_null( next_idx ) ) TEST( rec_pool->ele[ next_idx ].prev_idx==rec_idx );
310 0 : rec_idx = next_idx;
311 0 : }
312 0 : }
313 0 : } while(0);
314 :
315 0 : do {
316 0 : fd_funk_txn_all_iter_t txn_iter[1];
317 0 : for( fd_funk_txn_all_iter_new( funk, txn_iter ); !fd_funk_txn_all_iter_done( txn_iter ); fd_funk_txn_all_iter_next( txn_iter ) ) {
318 0 : fd_funk_txn_t const * txn = fd_funk_txn_all_iter_ele_const( txn_iter );
319 :
320 0 : uint rec_idx = txn->rec_tail_idx;
321 0 : while( !fd_funk_rec_idx_is_null( rec_idx ) ) {
322 0 : TEST( (rec_idx<rec_max) && rec_pool->ele[ rec_idx ].tag );
323 0 : rec_pool->ele[ rec_idx ].tag = 0;
324 0 : uint prev_idx = rec_pool->ele[ rec_idx ].prev_idx;
325 0 : if( !fd_funk_rec_idx_is_null( prev_idx ) ) TEST( rec_pool->ele[ prev_idx ].next_idx==rec_idx );
326 0 : rec_idx = prev_idx;
327 0 : }
328 0 : }
329 0 : } while(0);
330 :
331 0 : fd_funk_all_iter_t iter[1];
332 0 : for( fd_funk_all_iter_new( funk, iter ); !fd_funk_all_iter_done( iter ); fd_funk_all_iter_next( iter ) ) {
333 0 : fd_funk_rec_t const * rec = fd_funk_all_iter_ele( iter );
334 0 : FD_TEST( rec->tag == fd_funk_txn_xid_eq_root( rec->pair.xid ) );
335 0 : }
336 :
337 0 : # undef TEST
338 :
339 0 : return FD_FUNK_SUCCESS;
340 0 : }
|
