+++ /dev/null
-/* -----------------------------------------------------------------------------
- * (c) The GHC Team 1998-2005
- *
- * STM implementation.
- *
- * Overview
- * --------
- *
- * See the PPoPP 2005 paper "Composable memory transactions". In summary,
- * each transcation has a TRec (transaction record) holding entries for each of the
- * TVars (transactional variables) that it has accessed. Each entry records
- * (a) the TVar, (b) the expected value seen in the TVar, (c) the new value that
- * the transaction wants to write to the TVar, (d) during commit, the identity of
- * the TRec that wrote the expected value.
- *
- * Separate TRecs are used for each level in a nest of transactions. This allows
- * a nested transaction to be aborted without condemning its enclosing transactions.
- * This is needed in the implementation of catchRetry. Note that the "expected value"
- * in a nested transaction's TRec is the value expected to be *held in memory* if
- * the transaction commits -- not the "new value" stored in one of the enclosing
- * transactions. This means that validation can be done without searching through
- * a nest of TRecs.
- *
- * Concurrency control
- * -------------------
- *
- * Three different concurrency control schemes can be built according to the settings
- * in STM.h:
- *
- * STM_UNIPROC assumes that the caller serialises invocations on the STM interface.
- * In the Haskell RTS this means it is suitable only for non-THREADED_RTS builds.
- *
- * STM_CG_LOCK uses coarse-grained locking -- a single 'stm lock' is acquired during
- * an invocation on the STM interface. Note that this does not mean that
- * transactions are simply serialized -- the lock is only held *within* the
- * implementation of stmCommitTransaction, stmWait etc.
- *
- * STM_FG_LOCKS uses fine-grained locking -- locking is done on a per-TVar basis
- * and, when committing a transaction, no locks are acquired for TVars that have
- * been read but not updated.
- *
- * Concurrency control is implemented in the functions:
- *
- * lock_stm
- * unlock_stm
- * lock_tvar / cond_lock_tvar
- * unlock_tvar
- *
- * The choice between STM_UNIPROC / STM_CG_LOCK / STM_FG_LOCKS affects the
- * implementation of these functions.
- *
- * lock_stm & unlock_stm are straightforward : they acquire a simple spin-lock
- * using STM_CG_LOCK, and otherwise they are no-ops.
- *
- * lock_tvar / cond_lock_tvar and unlock_tvar are more complex because they
- * have other effects (present in STM_UNIPROC and STM_CG_LOCK builds) as well
- * as the actual business of maniupultaing a lock (present only in STM_FG_LOCKS
- * builds). This is because locking a TVar is implemented by writing the lock
- * holder's TRec into the TVar's current_value field:
- *
- * lock_tvar - lock a specified TVar (STM_FG_LOCKS only), returning the value
- * it contained.
- *
- * cond_lock_tvar - lock a specified TVar (STM_FG_LOCKS only) if it
- * contains a specified value. Return TRUE if this succeeds,
- * FALSE otherwise.
- *
- * unlock_tvar - release the lock on a specified TVar (STM_FG_LOCKS only),
- * storing a specified value in place of the lock entry.
- *
- * Using these operations, the typcial pattern of a commit/validate/wait operation
- * is to (a) lock the STM, (b) lock all the TVars being updated, (c) check that
- * the TVars that were only read from still contain their expected values,
- * (d) release the locks on the TVars, writing updates to them in the case of a
- * commit, (e) unlock the STM.
- *
- * Queues of waiting threads hang off the first_wait_queue_entry field of each
- * TVar. This may only be manipulated when holding that TVar's lock. In
- * particular, when a thread is putting itself to sleep, it mustn't release
- * the TVar's lock until it has added itself to the wait queue and marked its
- * TSO as BlockedOnSTM -- this makes sure that other threads will know to wake it.
- *
- * ---------------------------------------------------------------------------*/
-
-#include "PosixSource.h"
-#include "Rts.h"
-#include "RtsFlags.h"
-#include "RtsUtils.h"
-#include "Schedule.h"
-#include "SMP.h"
-#include "STM.h"
-#include "Storage.h"
-
-#include <stdlib.h>
-#include <stdio.h>
-
-#define TRUE 1
-#define FALSE 0
-
-// ACQ_ASSERT is used for assertions which are only required for
-// THREADED_RTS builds with fine-grained locking.
-
-#if defined(STM_FG_LOCKS)
-#define ACQ_ASSERT(_X) ASSERT(_X)
-#define NACQ_ASSERT(_X) /*Nothing*/
-#else
-#define ACQ_ASSERT(_X) /*Nothing*/
-#define NACQ_ASSERT(_X) ASSERT(_X)
-#endif
-
-/*......................................................................*/
-
-// If SHAKE is defined then validation will sometime spuriously fail. They helps test
-// unusualy code paths if genuine contention is rare
-
-#if defined(DEBUG)
-#define SHAKE
-#if defined(THREADED_RTS)
-#define TRACE(_x...) IF_DEBUG(stm, debugBelch("STM (task %p): ", (void *)(unsigned long)(unsigned int)osThreadId()); debugBelch ( _x ))
-#else
-#define TRACE(_x...) IF_DEBUG(stm, debugBelch ( _x ))
-#endif
-#else
-#define TRACE(_x...) /*Nothing*/
-#endif
-
-#ifdef SHAKE
-static const int do_shake = TRUE;
-#else
-static const int do_shake = FALSE;
-#endif
-static int shake_ctr = 0;
-static int shake_lim = 1;
-
-static int shake(void) {
- if (do_shake) {
- if (((shake_ctr++) % shake_lim) == 0) {
- shake_ctr = 1;
- shake_lim ++;
- return TRUE;
- }
- return FALSE;
- } else {
- return FALSE;
- }
-}
-
-/*......................................................................*/
-
-// Helper macros for iterating over entries within a transaction
-// record
-
-#define FOR_EACH_ENTRY(_t,_x,CODE) do { \
- StgTRecHeader *__t = (_t); \
- StgTRecChunk *__c = __t -> current_chunk; \
- StgWord __limit = __c -> next_entry_idx; \
- TRACE("%p : FOR_EACH_ENTRY, current_chunk=%p limit=%ld\n", __t, __c, __limit); \
- while (__c != END_STM_CHUNK_LIST) { \
- StgWord __i; \
- for (__i = 0; __i < __limit; __i ++) { \
- TRecEntry *_x = &(__c -> entries[__i]); \
- do { CODE } while (0); \
- } \
- __c = __c -> prev_chunk; \
- __limit = TREC_CHUNK_NUM_ENTRIES; \
- } \
- exit_for_each: \
- if (FALSE) goto exit_for_each; \
-} while (0)
-
-#define BREAK_FOR_EACH goto exit_for_each
-
-/*......................................................................*/
-
-// if REUSE_MEMORY is defined then attempt to re-use descriptors, log chunks,
-// and wait queue entries without GC
-
-#define REUSE_MEMORY
-
-/*......................................................................*/
-
-#define IF_STM_UNIPROC(__X) do { } while (0)
-#define IF_STM_CG_LOCK(__X) do { } while (0)
-#define IF_STM_FG_LOCKS(__X) do { } while (0)
-
-#if defined(STM_UNIPROC)
-#undef IF_STM_UNIPROC
-#define IF_STM_UNIPROC(__X) do { __X } while (0)
-static const StgBool use_read_phase = FALSE;
-
-static void lock_stm(StgTRecHeader *trec STG_UNUSED) {
- TRACE("%p : lock_stm()\n", trec);
-}
-
-static void unlock_stm(StgTRecHeader *trec STG_UNUSED) {
- TRACE("%p : unlock_stm()\n", trec);
-}
-
-static StgClosure *lock_tvar(StgTRecHeader *trec STG_UNUSED,
- StgTVar *s STG_UNUSED) {
- StgClosure *result;
- TRACE("%p : lock_tvar(%p)\n", trec, s);
- result = s -> current_value;
- return result;
-}
-
-static void unlock_tvar(StgTRecHeader *trec STG_UNUSED,
- StgTVar *s STG_UNUSED,
- StgClosure *c,
- StgBool force_update) {
- TRACE("%p : unlock_tvar(%p)\n", trec, s);
- if (force_update) {
- s -> current_value = c;
- }
-}
-
-static StgBool cond_lock_tvar(StgTRecHeader *trec STG_UNUSED,
- StgTVar *s STG_UNUSED,
- StgClosure *expected) {
- StgClosure *result;
- TRACE("%p : cond_lock_tvar(%p, %p)\n", trec, s, expected);
- result = s -> current_value;
- TRACE("%p : %s\n", trec, (result == expected) ? "success" : "failure");
- return (result == expected);
-}
-#endif
-
-#if defined(STM_CG_LOCK) /*........................................*/
-
-#undef IF_STM_CG_LOCK
-#define IF_STM_CG_LOCK(__X) do { __X } while (0)
-static const StgBool use_read_phase = FALSE;
-static volatile StgTRecHeader *smp_locked = NULL;
-
-static void lock_stm(StgTRecHeader *trec) {
- while (cas(&smp_locked, NULL, trec) != NULL) { }
- TRACE("%p : lock_stm()\n", trec);
-}
-
-static void unlock_stm(StgTRecHeader *trec STG_UNUSED) {
- TRACE("%p : unlock_stm()\n", trec);
- ASSERT (smp_locked == trec);
- smp_locked = 0;
-}
-
-static StgClosure *lock_tvar(StgTRecHeader *trec STG_UNUSED,
- StgTVar *s STG_UNUSED) {
- StgClosure *result;
- TRACE("%p : lock_tvar(%p)\n", trec, s);
- ASSERT (smp_locked == trec);
- result = s -> current_value;
- return result;
-}
-
-static void *unlock_tvar(StgTRecHeader *trec STG_UNUSED,
- StgTVar *s STG_UNUSED,
- StgClosure *c,
- StgBool force_update) {
- TRACE("%p : unlock_tvar(%p, %p)\n", trec, s, c);
- ASSERT (smp_locked == trec);
- if (force_update) {
- s -> current_value = c;
- }
-}
-
-static StgBool cond_lock_tvar(StgTRecHeader *trec STG_UNUSED,
- StgTVar *s STG_UNUSED,
- StgClosure *expected) {
- StgClosure *result;
- TRACE("%p : cond_lock_tvar(%p, %p)\n", trec, s, expected);
- ASSERT (smp_locked == trec);
- result = s -> current_value;
- TRACE("%p : %d\n", result ? "success" : "failure");
- return (result == expected);
-}
-#endif
-
-#if defined(STM_FG_LOCKS) /*...................................*/
-
-#undef IF_STM_FG_LOCKS
-#define IF_STM_FG_LOCKS(__X) do { __X } while (0)
-static const StgBool use_read_phase = TRUE;
-
-static void lock_stm(StgTRecHeader *trec STG_UNUSED) {
- TRACE("%p : lock_stm()\n", trec);
-}
-
-static void unlock_stm(StgTRecHeader *trec STG_UNUSED) {
- TRACE("%p : unlock_stm()\n", trec);
-}
-
-static StgClosure *lock_tvar(StgTRecHeader *trec,
- StgTVar *s STG_UNUSED) {
- StgClosure *result;
- TRACE("%p : lock_tvar(%p)\n", trec, s);
- do {
- do {
- result = s -> current_value;
- } while (GET_INFO(result) == &stg_TREC_HEADER_info);
- } while (cas(&(s -> current_value), result, trec) != result);
- return result;
-}
-
-static void unlock_tvar(StgTRecHeader *trec STG_UNUSED,
- StgTVar *s,
- StgClosure *c,
- StgBool force_update STG_UNUSED) {
- TRACE("%p : unlock_tvar(%p, %p)\n", trec, s, c);
- ASSERT(s -> current_value == trec);
- s -> current_value = c;
-}
-
-static StgBool cond_lock_tvar(StgTRecHeader *trec,
- StgTVar *s,
- StgClosure *expected) {
- StgClosure *result;
- TRACE("%p : cond_lock_tvar(%p, %p)\n", trec, s, expected);
- result = cas(&(s -> current_value), expected, trec);
- TRACE("%p : %s\n", trec, result ? "success" : "failure");
- return (result == expected);
-}
-#endif
-
-/*......................................................................*/
-
-// Helper functions for thread blocking and unblocking
-
-static void park_tso(StgTSO *tso) {
- ASSERT(tso -> why_blocked == NotBlocked);
- tso -> why_blocked = BlockedOnSTM;
- tso -> block_info.closure = (StgClosure *) END_TSO_QUEUE;
- TRACE("park_tso on tso=%p\n", tso);
-}
-
-static void unpark_tso(Capability *cap, StgTSO *tso) {
- // We will continue unparking threads while they remain on one of the wait
- // queues: it's up to the thread itself to remove it from the wait queues
- // if it decides to do so when it is scheduled.
- if (tso -> why_blocked == BlockedOnSTM) {
- TRACE("unpark_tso on tso=%p\n", tso);
- unblockOne(cap,tso);
- } else {
- TRACE("spurious unpark_tso on tso=%p\n", tso);
- }
-}
-
-static void unpark_waiters_on(Capability *cap, StgTVar *s) {
- StgTVarWaitQueue *q;
- TRACE("unpark_waiters_on tvar=%p\n", s);
- for (q = s -> first_wait_queue_entry;
- q != END_STM_WAIT_QUEUE;
- q = q -> next_queue_entry) {
- unpark_tso(cap, q -> waiting_tso);
- }
-}
-
-/*......................................................................*/
-
-// Helper functions for downstream allocation and initialization
-
-static StgTVarWaitQueue *new_stg_tvar_wait_queue(Capability *cap,
- StgTSO *waiting_tso) {
- StgTVarWaitQueue *result;
- result = (StgTVarWaitQueue *)allocateLocal(cap, sizeofW(StgTVarWaitQueue));
- SET_HDR (result, &stg_TVAR_WAIT_QUEUE_info, CCS_SYSTEM);
- result -> waiting_tso = waiting_tso;
- return result;
-}
-
-static StgTRecChunk *new_stg_trec_chunk(Capability *cap) {
- StgTRecChunk *result;
- result = (StgTRecChunk *)allocateLocal(cap, sizeofW(StgTRecChunk));
- SET_HDR (result, &stg_TREC_CHUNK_info, CCS_SYSTEM);
- result -> prev_chunk = END_STM_CHUNK_LIST;
- result -> next_entry_idx = 0;
- return result;
-}
-
-static StgTRecHeader *new_stg_trec_header(Capability *cap,
- StgTRecHeader *enclosing_trec) {
- StgTRecHeader *result;
- result = (StgTRecHeader *) allocateLocal(cap, sizeofW(StgTRecHeader));
- SET_HDR (result, &stg_TREC_HEADER_info, CCS_SYSTEM);
-
- result -> enclosing_trec = enclosing_trec;
- result -> current_chunk = new_stg_trec_chunk(cap);
-
- if (enclosing_trec == NO_TREC) {
- result -> state = TREC_ACTIVE;
- } else {
- ASSERT(enclosing_trec -> state == TREC_ACTIVE ||
- enclosing_trec -> state == TREC_CONDEMNED);
- result -> state = enclosing_trec -> state;
- }
-
- return result;
-}
-
-/*......................................................................*/
-
-// Allocation / deallocation functions that retain per-capability lists
-// of closures that can be re-used
-
-static StgTVarWaitQueue *alloc_stg_tvar_wait_queue(Capability *cap,
- StgTSO *waiting_tso) {
- StgTVarWaitQueue *result = NULL;
- if (cap -> free_tvar_wait_queues == END_STM_WAIT_QUEUE) {
- result = new_stg_tvar_wait_queue(cap, waiting_tso);
- } else {
- result = cap -> free_tvar_wait_queues;
- result -> waiting_tso = waiting_tso;
- cap -> free_tvar_wait_queues = result -> next_queue_entry;
- }
- return result;
-}
-
-static void free_stg_tvar_wait_queue(Capability *cap,
- StgTVarWaitQueue *wq) {
-#if defined(REUSE_MEMORY)
- wq -> next_queue_entry = cap -> free_tvar_wait_queues;
- cap -> free_tvar_wait_queues = wq;
-#endif
-}
-
-static StgTRecChunk *alloc_stg_trec_chunk(Capability *cap) {
- StgTRecChunk *result = NULL;
- if (cap -> free_trec_chunks == END_STM_CHUNK_LIST) {
- result = new_stg_trec_chunk(cap);
- } else {
- result = cap -> free_trec_chunks;
- cap -> free_trec_chunks = result -> prev_chunk;
- result -> prev_chunk = END_STM_CHUNK_LIST;
- result -> next_entry_idx = 0;
- }
- return result;
-}
-
-static void free_stg_trec_chunk(Capability *cap,
- StgTRecChunk *c) {
-#if defined(REUSE_MEMORY)
- c -> prev_chunk = cap -> free_trec_chunks;
- cap -> free_trec_chunks = c;
-#endif
-}
-
-static StgTRecHeader *alloc_stg_trec_header(Capability *cap,
- StgTRecHeader *enclosing_trec) {
- StgTRecHeader *result = NULL;
- if (cap -> free_trec_headers == NO_TREC) {
- result = new_stg_trec_header(cap, enclosing_trec);
- } else {
- result = cap -> free_trec_headers;
- cap -> free_trec_headers = result -> enclosing_trec;
- result -> enclosing_trec = enclosing_trec;
- result -> current_chunk -> next_entry_idx = 0;
- if (enclosing_trec == NO_TREC) {
- result -> state = TREC_ACTIVE;
- } else {
- ASSERT(enclosing_trec -> state == TREC_ACTIVE ||
- enclosing_trec -> state == TREC_CONDEMNED);
- result -> state = enclosing_trec -> state;
- }
- }
- return result;
-}
-
-static void free_stg_trec_header(Capability *cap,
- StgTRecHeader *trec) {
-#if defined(REUSE_MEMORY)
- StgTRecChunk *chunk = trec -> current_chunk -> prev_chunk;
- while (chunk != END_STM_CHUNK_LIST) {
- StgTRecChunk *prev_chunk = chunk -> prev_chunk;
- free_stg_trec_chunk(cap, chunk);
- chunk = prev_chunk;
- }
- trec -> current_chunk -> prev_chunk = END_STM_CHUNK_LIST;
- trec -> enclosing_trec = cap -> free_trec_headers;
- cap -> free_trec_headers = trec;
-#endif
-}
-
-/*......................................................................*/
-
-// Helper functions for managing waiting lists
-
-static void build_wait_queue_entries_for_trec(Capability *cap,
- StgTSO *tso,
- StgTRecHeader *trec) {
- ASSERT(trec != NO_TREC);
- ASSERT(trec -> enclosing_trec == NO_TREC);
- ASSERT(trec -> state == TREC_ACTIVE);
-
- TRACE("%p : build_wait_queue_entries_for_trec()\n", trec);
-
- FOR_EACH_ENTRY(trec, e, {
- StgTVar *s;
- StgTVarWaitQueue *q;
- StgTVarWaitQueue *fq;
- s = e -> tvar;
- TRACE("%p : adding tso=%p to wait queue for tvar=%p\n", trec, tso, s);
- ACQ_ASSERT(s -> current_value == trec);
- NACQ_ASSERT(s -> current_value == e -> expected_value);
- fq = s -> first_wait_queue_entry;
- q = alloc_stg_tvar_wait_queue(cap, tso);
- q -> next_queue_entry = fq;
- q -> prev_queue_entry = END_STM_WAIT_QUEUE;
- if (fq != END_STM_WAIT_QUEUE) {
- fq -> prev_queue_entry = q;
- }
- s -> first_wait_queue_entry = q;
- e -> new_value = (StgClosure *) q;
- });
-}
-
-static void remove_wait_queue_entries_for_trec(Capability *cap,
- StgTRecHeader *trec) {
- ASSERT(trec != NO_TREC);
- ASSERT(trec -> enclosing_trec == NO_TREC);
- ASSERT(trec -> state == TREC_WAITING ||
- trec -> state == TREC_CONDEMNED);
-
- TRACE("%p : remove_wait_queue_entries_for_trec()\n", trec);
-
- FOR_EACH_ENTRY(trec, e, {
- StgTVar *s;
- StgTVarWaitQueue *pq;
- StgTVarWaitQueue *nq;
- StgTVarWaitQueue *q;
- s = e -> tvar;
- StgClosure *saw = lock_tvar(trec, s);
- q = (StgTVarWaitQueue *) (e -> new_value);
- TRACE("%p : removing tso=%p from wait queue for tvar=%p\n", trec, q -> waiting_tso, s);
- ACQ_ASSERT(s -> current_value == trec);
- nq = q -> next_queue_entry;
- pq = q -> prev_queue_entry;
- if (nq != END_STM_WAIT_QUEUE) {
- nq -> prev_queue_entry = pq;
- }
- if (pq != END_STM_WAIT_QUEUE) {
- pq -> next_queue_entry = nq;
- } else {
- ASSERT (s -> first_wait_queue_entry == q);
- s -> first_wait_queue_entry = nq;
- }
- free_stg_tvar_wait_queue(cap, q);
- unlock_tvar(trec, s, saw, FALSE);
- });
-}
-
-/*......................................................................*/
-
-static TRecEntry *get_new_entry(Capability *cap,
- StgTRecHeader *t) {
- TRecEntry *result;
- StgTRecChunk *c;
- int i;
-
- c = t -> current_chunk;
- i = c -> next_entry_idx;
- ASSERT(c != END_STM_CHUNK_LIST);
-
- if (i < TREC_CHUNK_NUM_ENTRIES) {
- // Continue to use current chunk
- result = &(c -> entries[i]);
- c -> next_entry_idx ++;
- } else {
- // Current chunk is full: allocate a fresh one
- StgTRecChunk *nc;
- nc = alloc_stg_trec_chunk(cap);
- nc -> prev_chunk = c;
- nc -> next_entry_idx = 1;
- t -> current_chunk = nc;
- result = &(nc -> entries[0]);
- }
-
- return result;
-}
-
-/*......................................................................*/
-
-static void merge_update_into(Capability *cap,
- StgTRecHeader *t,
- StgTVar *tvar,
- StgClosure *expected_value,
- StgClosure *new_value) {
- int found;
-
- // Look for an entry in this trec
- found = FALSE;
- FOR_EACH_ENTRY(t, e, {
- StgTVar *s;
- s = e -> tvar;
- if (s == tvar) {
- found = TRUE;
- if (e -> expected_value != expected_value) {
- // Must abort if the two entries start from different values
- TRACE("%p : entries inconsistent at %p (%p vs %p)\n",
- t, tvar, e -> expected_value, expected_value);
- t -> state = TREC_CONDEMNED;
- }
- e -> new_value = new_value;
- BREAK_FOR_EACH;
- }
- });
-
- if (!found) {
- // No entry so far in this trec
- TRecEntry *ne;
- ne = get_new_entry(cap, t);
- ne -> tvar = tvar;
- ne -> expected_value = expected_value;
- ne -> new_value = new_value;
- }
-}
-
-/*......................................................................*/
-
-static StgBool entry_is_update(TRecEntry *e) {
- StgBool result;
- result = (e -> expected_value != e -> new_value);
- return result;
-}
-
-#if defined(STM_FG_LOCKS)
-static StgBool entry_is_read_only(TRecEntry *e) {
- StgBool result;
- result = (e -> expected_value == e -> new_value);
- return result;
-}
-
-static StgBool tvar_is_locked(StgTVar *s, StgTRecHeader *h) {
- StgClosure *c;
- StgBool result;
- c = s -> current_value;
- result = (c == (StgClosure *) h);
- return result;
-}
-#endif
-
-// revert_ownership : release a lock on a TVar, storing back
-// the value that it held when the lock was acquired. "revert_all"
-// is set in stmWait and stmReWait when we acquired locks on all of
-// the TVars involved. "revert_all" is not set in commit operations
-// where we don't lock TVars that have been read from but not updated.
-
-static void revert_ownership(StgTRecHeader *trec STG_UNUSED,
- StgBool revert_all STG_UNUSED) {
-#if defined(STM_FG_LOCKS)
- FOR_EACH_ENTRY(trec, e, {
- if (revert_all || entry_is_update(e)) {
- StgTVar *s;
- s = e -> tvar;
- if (tvar_is_locked(s, trec)) {
- unlock_tvar(trec, s, e -> expected_value, TRUE);
- }
- }
- });
-#endif
-}
-
-/*......................................................................*/
-
-// validate_and_acquire_ownership : this performs the twin functions
-// of checking that the TVars referred to by entries in trec hold the
-// expected values and:
-//
-// - locking the TVar (on updated TVars during commit, or all TVars
-// during wait)
-//
-// - recording the identity of the TRec who wrote the value seen in the
-// TVar (on non-updated TVars during commit). These values are
-// stashed in the TRec entries and are then checked in check_read_only
-// to ensure that an atomic snapshot of all of these locations has been
-// seen.
-
-static StgBool validate_and_acquire_ownership (StgTRecHeader *trec,
- int acquire_all,
- int retain_ownership) {
- StgBool result;
-
- if (shake()) {
- TRACE("%p : shake, pretending trec is invalid when it may not be\n", trec);
- return FALSE;
- }
-
- ASSERT ((trec -> state == TREC_ACTIVE) ||
- (trec -> state == TREC_WAITING) ||
- (trec -> state == TREC_CONDEMNED));
- result = !((trec -> state) == TREC_CONDEMNED);
- if (result) {
- FOR_EACH_ENTRY(trec, e, {
- StgTVar *s;
- s = e -> tvar;
- if (acquire_all || entry_is_update(e)) {
- TRACE("%p : trying to acquire %p\n", trec, s);
- if (!cond_lock_tvar(trec, s, e -> expected_value)) {
- TRACE("%p : failed to acquire %p\n", trec, s);
- result = FALSE;
- BREAK_FOR_EACH;
- }
- } else {
- ASSERT(use_read_phase);
- IF_STM_FG_LOCKS({
- TRACE("%p : will need to check %p\n", trec, s);
- if (s -> current_value != e -> expected_value) {
- TRACE("%p : doesn't match\n", trec);
- result = FALSE;
- BREAK_FOR_EACH;
- }
- e -> num_updates = s -> num_updates;
- if (s -> current_value != e -> expected_value) {
- TRACE("%p : doesn't match (race)\n", trec);
- result = FALSE;
- BREAK_FOR_EACH;
- } else {
- TRACE("%p : need to check version %d\n", trec, e -> num_updates);
- }
- });
- }
- });
- }
-
- if ((!result) || (!retain_ownership)) {
- revert_ownership(trec, acquire_all);
- }
-
- return result;
-}
-
-// check_read_only : check that we've seen an atomic snapshot of the
-// non-updated TVars accessed by a trec. This checks that the last TRec to
-// commit an update to the TVar is unchanged since the value was stashed in
-// validate_and_acquire_ownership. If no udpate is seen to any TVar than
-// all of them contained their expected values at the start of the call to
-// check_read_only.
-//
-// The paper "Concurrent programming without locks" (under submission), or
-// Keir Fraser's PhD dissertation "Practical lock-free programming" discuss
-// this kind of algorithm.
-
-static StgBool check_read_only(StgTRecHeader *trec STG_UNUSED) {
- StgBool result = TRUE;
-
- ASSERT (use_read_phase);
- IF_STM_FG_LOCKS({
- FOR_EACH_ENTRY(trec, e, {
- StgTVar *s;
- s = e -> tvar;
- if (entry_is_read_only(e)) {
- TRACE("%p : check_read_only for TVar %p, saw %d\n", trec, s, e -> num_updates);
- if (s -> num_updates != e -> num_updates) {
- // ||s -> current_value != e -> expected_value) {
- TRACE("%p : mismatch\n", trec);
- result = FALSE;
- BREAK_FOR_EACH;
- }
- }
- });
- });
-
- return result;
-}
-
-
-/************************************************************************/
-
-void stmPreGCHook() {
- nat i;
-
- lock_stm(NO_TREC);
- TRACE("stmPreGCHook\n");
- for (i = 0; i < n_capabilities; i ++) {
- Capability *cap = &capabilities[i];
- cap -> free_tvar_wait_queues = END_STM_WAIT_QUEUE;
- cap -> free_trec_chunks = END_STM_CHUNK_LIST;
- cap -> free_trec_headers = NO_TREC;
- }
- unlock_stm(NO_TREC);
-}
-
-/************************************************************************/
-
-// check_read_only relies on version numbers held in TVars' "num_updates"
-// fields not wrapping around while a transaction is committed. The version
-// number is incremented each time an update is committed to the TVar
-// This is unlikely to wrap around when 32-bit integers are used for the counts,
-// but to ensure correctness we maintain a shared count on the maximum
-// number of commit operations that may occur and check that this has
-// not increased by more than 2^32 during a commit.
-
-#define TOKEN_BATCH_SIZE 1024
-
-static volatile StgInt64 max_commits = 0;
-
-static volatile StgBool token_locked = FALSE;
-
-#if defined(THREADED_RTS)
-static void getTokenBatch(Capability *cap) {
- while (cas(&token_locked, FALSE, TRUE) == TRUE) { /* nothing */ }
- max_commits += TOKEN_BATCH_SIZE;
- cap -> transaction_tokens = TOKEN_BATCH_SIZE;
- token_locked = FALSE;
-}
-
-static void getToken(Capability *cap) {
- if (cap -> transaction_tokens == 0) {
- getTokenBatch(cap);
- }
- cap -> transaction_tokens --;
-}
-#else
-static void getToken(Capability *cap STG_UNUSED) {
- // Nothing
-}
-#endif
-
-/*......................................................................*/
-
-StgTRecHeader *stmStartTransaction(Capability *cap,
- StgTRecHeader *outer) {
- StgTRecHeader *t;
- TRACE("%p : stmStartTransaction with %d tokens\n",
- outer,
- cap -> transaction_tokens);
-
- getToken(cap);
-
- t = alloc_stg_trec_header(cap, outer);
- TRACE("%p : stmStartTransaction()=%p\n", outer, t);
- return t;
-}
-
-/*......................................................................*/
-
-void stmAbortTransaction(Capability *cap,
- StgTRecHeader *trec) {
- TRACE("%p : stmAbortTransaction\n", trec);
- ASSERT (trec != NO_TREC);
- ASSERT ((trec -> state == TREC_ACTIVE) ||
- (trec -> state == TREC_WAITING) ||
- (trec -> state == TREC_CONDEMNED));
-
- lock_stm(trec);
- if (trec -> state == TREC_WAITING) {
- ASSERT (trec -> enclosing_trec == NO_TREC);
- TRACE("%p : stmAbortTransaction aborting waiting transaction\n", trec);
- remove_wait_queue_entries_for_trec(cap, trec);
- }
- trec -> state = TREC_ABORTED;
- unlock_stm(trec);
-
- free_stg_trec_header(cap, trec);
-
- TRACE("%p : stmAbortTransaction done\n", trec);
-}
-
-/*......................................................................*/
-
-void stmCondemnTransaction(Capability *cap,
- StgTRecHeader *trec) {
- TRACE("%p : stmCondemnTransaction\n", trec);
- ASSERT (trec != NO_TREC);
- ASSERT ((trec -> state == TREC_ACTIVE) ||
- (trec -> state == TREC_WAITING) ||
- (trec -> state == TREC_CONDEMNED));
-
- lock_stm(trec);
- if (trec -> state == TREC_WAITING) {
- ASSERT (trec -> enclosing_trec == NO_TREC);
- TRACE("%p : stmCondemnTransaction condemning waiting transaction\n", trec);
- remove_wait_queue_entries_for_trec(cap, trec);
- }
- trec -> state = TREC_CONDEMNED;
- unlock_stm(trec);
-
- TRACE("%p : stmCondemnTransaction done\n", trec);
-}
-
-/*......................................................................*/
-
-StgTRecHeader *stmGetEnclosingTRec(StgTRecHeader *trec) {
- StgTRecHeader *outer;
- TRACE("%p : stmGetEnclosingTRec\n", trec);
- outer = trec -> enclosing_trec;
- TRACE("%p : stmGetEnclosingTRec()=%p\n", trec, outer);
- return outer;
-}
-
-/*......................................................................*/
-
-StgBool stmValidateNestOfTransactions(StgTRecHeader *trec) {
- StgTRecHeader *t;
- StgBool result;
-
- TRACE("%p : stmValidateNestOfTransactions\n", trec);
- ASSERT(trec != NO_TREC);
- ASSERT((trec -> state == TREC_ACTIVE) ||
- (trec -> state == TREC_WAITING) ||
- (trec -> state == TREC_CONDEMNED));
-
- lock_stm(trec);
-
- t = trec;
- result = TRUE;
- while (t != NO_TREC) {
- result &= validate_and_acquire_ownership(t, TRUE, FALSE);
- t = t -> enclosing_trec;
- }
-
- if (!result && trec -> state != TREC_WAITING) {
- trec -> state = TREC_CONDEMNED;
- }
-
- unlock_stm(trec);
-
- TRACE("%p : stmValidateNestOfTransactions()=%d\n", trec, result);
- return result;
-}
-
-/*......................................................................*/
-
-StgBool stmCommitTransaction(Capability *cap, StgTRecHeader *trec) {
- int result;
- StgInt64 max_commits_at_start = max_commits;
-
- TRACE("%p : stmCommitTransaction()\n", trec);
- ASSERT (trec != NO_TREC);
-
- lock_stm(trec);
-
- ASSERT (trec -> enclosing_trec == NO_TREC);
- ASSERT ((trec -> state == TREC_ACTIVE) ||
- (trec -> state == TREC_CONDEMNED));
-
- result = validate_and_acquire_ownership(trec, (!use_read_phase), TRUE);
- if (result) {
- // We now know that all the updated locations hold their expected values.
- ASSERT (trec -> state == TREC_ACTIVE);
-
- if (use_read_phase) {
- TRACE("%p : doing read check\n", trec);
- result = check_read_only(trec);
- TRACE("%p : read-check %s\n", trec, result ? "succeeded" : "failed");
-
- StgInt64 max_commits_at_end = max_commits;
- StgInt64 max_concurrent_commits;
- max_concurrent_commits = ((max_commits_at_end - max_commits_at_start) +
- (n_capabilities * TOKEN_BATCH_SIZE));
- if (((max_concurrent_commits >> 32) > 0) || shake()) {
- result = FALSE;
- }
- }
-
- if (result) {
- // We now know that all of the read-only locations held their exepcted values
- // at the end of the call to validate_and_acquire_ownership. This forms the
- // linearization point of the commit.
-
- FOR_EACH_ENTRY(trec, e, {
- StgTVar *s;
- s = e -> tvar;
- if (e -> new_value != e -> expected_value) {
- // Entry is an update: write the value back to the TVar, unlocking it if
- // necessary.
-
- ACQ_ASSERT(tvar_is_locked(s, trec));
- TRACE("%p : writing %p to %p, waking waiters\n", trec, e -> new_value, s);
- unpark_waiters_on(cap,s);
- IF_STM_FG_LOCKS({
- s -> num_updates ++;
- });
- unlock_tvar(trec, s, e -> new_value, TRUE);
- }
- ACQ_ASSERT(!tvar_is_locked(s, trec));
- });
- } else {
- revert_ownership(trec, FALSE);
- }
- }
-
- unlock_stm(trec);
-
- free_stg_trec_header(cap, trec);
-
- TRACE("%p : stmCommitTransaction()=%d\n", trec, result);
-
- return result;
-}
-
-/*......................................................................*/
-
-StgBool stmCommitNestedTransaction(Capability *cap, StgTRecHeader *trec) {
- StgTRecHeader *et;
- int result;
- ASSERT (trec != NO_TREC && trec -> enclosing_trec != NO_TREC);
- TRACE("%p : stmCommitNestedTransaction() into %p\n", trec, trec -> enclosing_trec);
- ASSERT ((trec -> state == TREC_ACTIVE) || (trec -> state == TREC_CONDEMNED));
-
- lock_stm(trec);
-
- et = trec -> enclosing_trec;
- result = validate_and_acquire_ownership(trec, (!use_read_phase), TRUE);
- if (result) {
- // We now know that all the updated locations hold their expected values.
-
- if (use_read_phase) {
- TRACE("%p : doing read check\n", trec);
- result = check_read_only(trec);
- }
- if (result) {
- // We now know that all of the read-only locations held their exepcted values
- // at the end of the call to validate_and_acquire_ownership. This forms the
- // linearization point of the commit.
-
- if (result) {
- TRACE("%p : read-check succeeded\n", trec);
- FOR_EACH_ENTRY(trec, e, {
- // Merge each entry into the enclosing transaction record, release all
- // locks.
-
- StgTVar *s;
- s = e -> tvar;
- if (entry_is_update(e)) {
- unlock_tvar(trec, s, e -> expected_value, FALSE);
- }
- merge_update_into(cap, et, s, e -> expected_value, e -> new_value);
- ACQ_ASSERT(s -> current_value != trec);
- });
- } else {
- revert_ownership(trec, FALSE);
- }
- }
- }
-
- unlock_stm(trec);
-
- free_stg_trec_header(cap, trec);
-
- TRACE("%p : stmCommitNestedTransaction()=%d\n", trec, result);
-
- return result;
-}
-
-/*......................................................................*/
-
-StgBool stmWait(Capability *cap, StgTSO *tso, StgTRecHeader *trec) {
- int result;
- TRACE("%p : stmWait(%p)\n", trec, tso);
- ASSERT (trec != NO_TREC);
- ASSERT (trec -> enclosing_trec == NO_TREC);
- ASSERT ((trec -> state == TREC_ACTIVE) ||
- (trec -> state == TREC_CONDEMNED));
-
- lock_stm(trec);
- result = validate_and_acquire_ownership(trec, TRUE, TRUE);
- if (result) {
- // The transaction is valid so far so we can actually start waiting.
- // (Otherwise the transaction was not valid and the thread will have to
- // retry it).
-
- // Put ourselves to sleep. We retain locks on all the TVars involved
- // until we are sound asleep : (a) on the wait queues, (b) BlockedOnSTM
- // in the TSO, (c) TREC_WAITING in the Trec.
- build_wait_queue_entries_for_trec(cap, tso, trec);
- park_tso(tso);
- trec -> state = TREC_WAITING;
-
- // We haven't released ownership of the transaction yet. The TSO
- // has been put on the wait queue for the TVars it is waiting for,
- // but we haven't yet tidied up the TSO's stack and made it safe
- // to wake up the TSO. Therefore, we must wait until the TSO is
- // safe to wake up before we release ownership - when all is well,
- // the runtime will call stmWaitUnlock() below, with the same
- // TRec.
-
- } else {
- unlock_stm(trec);
- free_stg_trec_header(cap, trec);
- }
-
- TRACE("%p : stmWait(%p)=%d\n", trec, tso, result);
- return result;
-}
-
-
-void
-stmWaitUnlock(Capability *cap STG_UNUSED, StgTRecHeader *trec) {
- revert_ownership(trec, TRUE);
- unlock_stm(trec);
-}
-
-/*......................................................................*/
-
-StgBool stmReWait(Capability *cap, StgTSO *tso) {
- int result;
- StgTRecHeader *trec = tso->trec;
-
- TRACE("%p : stmReWait\n", trec);
- ASSERT (trec != NO_TREC);
- ASSERT (trec -> enclosing_trec == NO_TREC);
- ASSERT ((trec -> state == TREC_WAITING) ||
- (trec -> state == TREC_CONDEMNED));
-
- lock_stm(trec);
- result = validate_and_acquire_ownership(trec, TRUE, TRUE);
- TRACE("%p : validation %s\n", trec, result ? "succeeded" : "failed");
- if (result) {
- // The transaction remains valid -- do nothing because it is already on
- // the wait queues
- ASSERT (trec -> state == TREC_WAITING);
- park_tso(tso);
- revert_ownership(trec, TRUE);
- } else {
- // The transcation has become invalid. We can now remove it from the wait
- // queues.
- if (trec -> state != TREC_CONDEMNED) {
- remove_wait_queue_entries_for_trec (cap, trec);
- }
- free_stg_trec_header(cap, trec);
- }
- unlock_stm(trec);
-
- TRACE("%p : stmReWait()=%d\n", trec, result);
- return result;
-}
-
-/*......................................................................*/
-
-static TRecEntry *get_entry_for(StgTRecHeader *trec, StgTVar *tvar, StgTRecHeader **in) {
- TRecEntry *result = NULL;
-
- TRACE("%p : get_entry_for TVar %p\n", trec, tvar);
- ASSERT(trec != NO_TREC);
-
- do {
- FOR_EACH_ENTRY(trec, e, {
- if (e -> tvar == tvar) {
- result = e;
- if (in != NULL) {
- *in = trec;
- }
- BREAK_FOR_EACH;
- }
- });
- trec = trec -> enclosing_trec;
- } while (result == NULL && trec != NO_TREC);
-
- return result;
-}
-
-static StgClosure *read_current_value(StgTRecHeader *trec STG_UNUSED, StgTVar *tvar) {
- StgClosure *result;
- result = tvar -> current_value;
-
-#if defined(STM_FG_LOCKS)
- while (GET_INFO(result) == &stg_TREC_HEADER_info) {
- TRACE("%p : read_current_value(%p) saw %p\n", trec, tvar, result);
- result = tvar -> current_value;
- }
-#endif
-
- TRACE("%p : read_current_value(%p)=%p\n", trec, tvar, result);
- return result;
-}
-
-/*......................................................................*/
-
-StgClosure *stmReadTVar(Capability *cap,
- StgTRecHeader *trec,
- StgTVar *tvar) {
- StgTRecHeader *entry_in;
- StgClosure *result = NULL;
- TRecEntry *entry = NULL;
- TRACE("%p : stmReadTVar(%p)\n", trec, tvar);
- ASSERT (trec != NO_TREC);
- ASSERT (trec -> state == TREC_ACTIVE ||
- trec -> state == TREC_CONDEMNED);
-
- entry = get_entry_for(trec, tvar, &entry_in);
-
- if (entry != NULL) {
- if (entry_in == trec) {
- // Entry found in our trec
- result = entry -> new_value;
- } else {
- // Entry found in another trec
- TRecEntry *new_entry = get_new_entry(cap, trec);
- new_entry -> tvar = tvar;
- new_entry -> expected_value = entry -> expected_value;
- new_entry -> new_value = entry -> new_value;
- result = new_entry -> new_value;
- }
- } else {
- // No entry found
- StgClosure *current_value = read_current_value(trec, tvar);
- TRecEntry *new_entry = get_new_entry(cap, trec);
- new_entry -> tvar = tvar;
- new_entry -> expected_value = current_value;
- new_entry -> new_value = current_value;
- result = current_value;
- }
-
- TRACE("%p : stmReadTVar(%p)=%p\n", trec, tvar, result);
- return result;
-}
-
-/*......................................................................*/
-
-void stmWriteTVar(Capability *cap,
- StgTRecHeader *trec,
- StgTVar *tvar,
- StgClosure *new_value) {
-
- StgTRecHeader *entry_in;
- TRecEntry *entry = NULL;
- TRACE("%p : stmWriteTVar(%p, %p)\n", trec, tvar, new_value);
- ASSERT (trec != NO_TREC);
- ASSERT (trec -> state == TREC_ACTIVE ||
- trec -> state == TREC_CONDEMNED);
-
- entry = get_entry_for(trec, tvar, &entry_in);
-
- if (entry != NULL) {
- if (entry_in == trec) {
- // Entry found in our trec
- entry -> new_value = new_value;
- } else {
- // Entry found in another trec
- TRecEntry *new_entry = get_new_entry(cap, trec);
- new_entry -> tvar = tvar;
- new_entry -> expected_value = entry -> expected_value;
- new_entry -> new_value = new_value;
- }
- } else {
- // No entry found
- StgClosure *current_value = read_current_value(trec, tvar);
- TRecEntry *new_entry = get_new_entry(cap, trec);
- new_entry -> tvar = tvar;
- new_entry -> expected_value = current_value;
- new_entry -> new_value = new_value;
- }
-
- TRACE("%p : stmWriteTVar done\n", trec);
-}
-
-/*......................................................................*/
-
-StgTVar *stmNewTVar(Capability *cap,
- StgClosure *new_value) {
- StgTVar *result;
- result = (StgTVar *)allocateLocal(cap, sizeofW(StgTVar));
- SET_HDR (result, &stg_TVAR_info, CCS_SYSTEM);
- result -> current_value = new_value;
- result -> first_wait_queue_entry = END_STM_WAIT_QUEUE;
-#if defined(THREADED_RTS)
- result -> num_updates = 0;
-#endif
- return result;
-}
-
-/*......................................................................*/