9 #include "snapshot-interface.h"
10 #include "threads-model.h"
13 memory_order orders[6] = {
14 memory_order_relaxed, memory_order_consume, memory_order_acquire,
15 memory_order_release, memory_order_acq_rel, memory_order_seq_cst,
18 static void ensureModel() {
20 snapshot_system_init(10000, 1024, 1024, 40000);
21 model = new ModelChecker();
22 model->startChecker();
26 /* --- helper functions --- */
27 uint64_t model_rmwrcas_action_helper(void *obj, int atomic_index, uint64_t oldval, int size, const char *position) {
29 return model->switch_thread(new ModelAction(ATOMIC_RMWRCAS, position, orders[atomic_index], obj, oldval, size));
32 uint64_t model_rmwr_action_helper(void *obj, int atomic_index, const char *position) {
34 return model->switch_thread(new ModelAction(ATOMIC_RMWR, position, orders[atomic_index], obj));
37 void model_rmw_action_helper(void *obj, uint64_t val, int atomic_index, const char * position) {
39 model->switch_thread(new ModelAction(ATOMIC_RMW, position, orders[atomic_index], obj, val));
42 void model_rmwc_action_helper(void *obj, int atomic_index, const char *position) {
44 model->switch_thread(new ModelAction(ATOMIC_RMWC, position, orders[atomic_index], obj));
48 #define VOLATILELOAD(size) \
49 uint ## size ## _t cds_volatile_load ## size(void * obj, const char * position) { \
51 return (uint ## size ## _t)model->switch_thread(new ModelAction(ATOMIC_READ, position, memory_order_volatile_load, obj)); \
59 // cds volatile stores
60 #define VOLATILESTORE(size) \
61 void cds_volatile_store ## size (void * obj, uint ## size ## _t val, const char * position) { \
63 model->switch_thread(new ModelAction(ATOMIC_WRITE, position, memory_order_volatile_store, obj, (uint64_t) val)); \
64 *((volatile uint ## size ## _t *)obj) = val; \
65 thread_id_t tid = thread_current_id(); \
66 for(int i=0;i < size / 8;i++) { \
67 atomraceCheckWrite(tid, (void *)(((char *)obj)+i)); \
77 #define CDSATOMICINT(size) \
78 void cds_atomic_init ## size (void * obj, uint ## size ## _t val, const char * position) { \
80 model->switch_thread(new ModelAction(ATOMIC_INIT, position, memory_order_relaxed, obj, (uint64_t) val)); \
81 *((volatile uint ## size ## _t *)obj) = val; \
82 thread_id_t tid = thread_current_id(); \
83 for(int i=0;i < size / 8;i++) { \
84 atomraceCheckWrite(tid, (void *)(((char *)obj)+i)); \
94 #define CDSATOMICLOAD(size) \
95 uint ## size ## _t cds_atomic_load ## size(void * obj, int atomic_index, const char * position) { \
97 uint ## size ## _t val = (uint ## size ## _t)model->switch_thread( \
98 new ModelAction(ATOMIC_READ, position, orders[atomic_index], obj)); \
99 thread_id_t tid = thread_current_id(); \
100 for(int i=0;i < size / 8;i++) { \
101 atomraceCheckRead(tid, (void *)(((char *)obj)+i)); \
112 #define CDSATOMICSTORE(size) \
113 void cds_atomic_store ## size(void * obj, uint ## size ## _t val, int atomic_index, const char * position) { \
115 model->switch_thread(new ModelAction(ATOMIC_WRITE, position, orders[atomic_index], obj, (uint64_t) val)); \
116 *((volatile uint ## size ## _t *)obj) = val; \
117 thread_id_t tid = thread_current_id(); \
118 for(int i=0;i < size / 8;i++) { \
119 atomraceCheckWrite(tid, (void *)(((char *)obj)+i)); \
129 #define _ATOMIC_RMW_(__op__, size, addr, val, atomic_index, position) \
131 uint ## size ## _t _old = model_rmwr_action_helper(addr, atomic_index, position); \
132 uint ## size ## _t _copy = _old; \
133 uint ## size ## _t _val = val; \
135 model_rmw_action_helper(addr, (uint64_t) _copy, atomic_index, position); \
136 *((volatile uint ## size ## _t *)addr) = _copy; \
137 thread_id_t tid = thread_current_id(); \
138 for(int i=0;i < size / 8;i++) { \
139 atomraceCheckRead(tid, (void *)(((char *)addr)+i)); \
140 recordWrite(tid, (void *)(((char *)addr)+i)); \
145 // cds atomic exchange
146 #define CDSATOMICEXCHANGE(size) \
147 uint ## size ## _t cds_atomic_exchange ## size(void* addr, uint ## size ## _t val, int atomic_index, const char * position) { \
148 _ATOMIC_RMW_( =, size, addr, val, atomic_index, position); \
152 CDSATOMICEXCHANGE(16)
153 CDSATOMICEXCHANGE(32)
154 CDSATOMICEXCHANGE(64)
156 // cds atomic fetch add
157 #define CDSATOMICADD(size) \
158 uint ## size ## _t cds_atomic_fetch_add ## size(void* addr, uint ## size ## _t val, int atomic_index, const char * position) { \
159 _ATOMIC_RMW_( +=, size, addr, val, atomic_index, position); \
167 // cds atomic fetch sub
168 #define CDSATOMICSUB(size) \
169 uint ## size ## _t cds_atomic_fetch_sub ## size(void* addr, uint ## size ## _t val, int atomic_index, const char * position) { \
170 _ATOMIC_RMW_( -=, size, addr, val, atomic_index, position); \
178 // cds atomic fetch and
179 #define CDSATOMICAND(size) \
180 uint ## size ## _t cds_atomic_fetch_and ## size(void* addr, uint ## size ## _t val, int atomic_index, const char * position) { \
181 _ATOMIC_RMW_( &=, size, addr, val, atomic_index, position); \
189 // cds atomic fetch or
190 #define CDSATOMICOR(size) \
191 uint ## size ## _t cds_atomic_fetch_or ## size(void* addr, uint ## size ## _t val, int atomic_index, const char * position) { \
192 _ATOMIC_RMW_( |=, size, addr, val, atomic_index, position); \
200 // cds atomic fetch xor
201 #define CDSATOMICXOR(size) \
202 uint ## size ## _t cds_atomic_fetch_xor ## size(void* addr, uint ## size ## _t val, int atomic_index, const char * position) { \
203 _ATOMIC_RMW_( ^=, size, addr, val, atomic_index, position); \
211 // cds atomic compare and exchange
212 // In order to accomodate the LLVM PASS, the return values are not true or false.
214 #define _ATOMIC_CMPSWP_WEAK_ _ATOMIC_CMPSWP_
215 #define _ATOMIC_CMPSWP_(size, addr, expected, desired, atomic_index, position) \
217 uint ## size ## _t _desired = desired; \
218 uint ## size ## _t _expected = expected; \
219 uint ## size ## _t _old = model_rmwrcas_action_helper(addr, atomic_index, _expected, sizeof(_expected), position); \
220 if (_old == _expected) { \
221 model_rmw_action_helper(addr, (uint64_t) _desired, atomic_index, position); \
222 *((volatile uint ## size ## _t *)addr) = desired; \
223 thread_id_t tid = thread_current_id(); \
224 for(int i=0;i < size / 8;i++) { \
225 recordWrite(tid, (void *)(((char *)addr)+i)); \
227 return _expected; } \
229 model_rmwc_action_helper(addr, atomic_index, position); _expected = _old; return _old; } \
232 // atomic_compare_exchange version 1: the CmpOperand (corresponds to expected)
233 // extracted from LLVM IR is an integer type.
234 #define CDSATOMICCASV1(size) \
235 uint ## size ## _t cds_atomic_compare_exchange ## size ## _v1(void* addr, uint ## size ## _t expected, uint ## size ## _t desired, int atomic_index_succ, int atomic_index_fail, const char *position) { \
236 _ATOMIC_CMPSWP_(size, addr, expected, desired, atomic_index_succ, position); \
244 // atomic_compare_exchange version 2
245 #define CDSATOMICCASV2(size) \
246 bool cds_atomic_compare_exchange ## size ## _v2(void* addr, uint ## size ## _t* expected, uint ## size ## _t desired, int atomic_index_succ, int atomic_index_fail, const char *position) { \
247 uint ## size ## _t ret = cds_atomic_compare_exchange ## size ## _v1(addr, *expected, desired, atomic_index_succ, atomic_index_fail, position); \
248 if (ret == *expected) {return true;} else {return false;} \
256 // cds atomic thread fence
258 void cds_atomic_thread_fence(int atomic_index, const char * position) {
259 model->switch_thread(
260 new ModelAction(ATOMIC_FENCE, position, orders[atomic_index], FENCE_LOCATION)
265 #define _ATOMIC_CMPSWP_( __a__, __e__, __m__, __x__ ) \
266 ({ volatile __typeof__((__a__)->__f__)* __p__ = & ((__a__)->__f__); \
267 __typeof__(__e__) __q__ = (__e__); \
268 __typeof__(__m__) __v__ = (__m__); \
270 __typeof__((__a__)->__f__) __t__=(__typeof__((__a__)->__f__)) model_rmwr_action((void *)__p__, __x__); \
271 if (__t__ == * __q__ ) { \
272 model_rmw_action((void *)__p__, __x__, (uint64_t) __v__); __r__ = true; } \
273 else { model_rmwc_action((void *)__p__, __x__); *__q__ = __t__; __r__ = false;} \
276 #define _ATOMIC_FENCE_( __x__ ) \
277 ({ model_fence_action(__x__);})
282 #define _ATOMIC_MODIFY_( __a__, __o__, __m__, __x__ ) \
283 ({ volatile __typeof__((__a__)->__f__)* __p__ = & ((__a__)->__f__); \
284 __typeof__((__a__)->__f__) __old__=(__typeof__((__a__)->__f__)) model_rmwr_action((void *)__p__, __x__); \
285 __typeof__(__m__) __v__ = (__m__); \
286 __typeof__((__a__)->__f__) __copy__= __old__; \
287 __copy__ __o__ __v__; \
288 model_rmw_action((void *)__p__, __x__, (uint64_t) __copy__); \
289 __old__ = __old__; Silence clang (-Wunused-value) \
293 void cds_func_entry(const char * funcName) {
296 thread_id_t tid = thread_current_id();
299 ModelHistory *history = model->get_history();
300 if ( !history->getFuncMap()->contains(funcName) ) {
301 // add func id to func map
302 func_id = history->get_func_counter();
303 history->incr_func_counter();
304 history->getFuncMap()->put(funcName, func_id);
306 // add func id to reverse func map
307 ModelVector<const char *> * func_map_rev = history->getFuncMapRev();
308 if ( func_map_rev->size() <= func_id )
309 func_map_rev->resize( func_id + 1 );
311 func_map_rev->at(func_id) = funcName;
313 func_id = history->getFuncMap()->get(funcName);
316 history->enter_function(func_id, tid);
320 void cds_func_exit(const char * funcName) {
323 thread_id_t tid = thread_current_id();
326 ModelHistory *history = model->get_history();
327 func_id = history->getFuncMap()->get(funcName);
329 * func_id not found; this could happen in the case where a function calls cds_func_entry
330 * when the model has been defined yet, but then an atomic inside the function initializes
331 * the model. And then cds_func_exit is called upon the function exiting.
337 history->exit_function(func_id, tid);