dejafu/Test/DejaFu/Deterministic.hs

328 lines
12 KiB
Haskell
Raw Normal View History

2015-01-25 19:15:23 +03:00
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE ImpredicativeTypes #-}
2015-02-16 06:16:55 +03:00
{-# LANGUAGE Rank2Types #-}
{-# LANGUAGE TypeFamilies #-}
2015-01-25 19:15:23 +03:00
-- | Deterministic traced execution of concurrent computations which
-- don't do @IO@.
--
-- This works by executing the computation on a single thread, calling
-- out to the supplied scheduler after each step to determine which
-- thread runs next.
2015-01-31 18:50:54 +03:00
module Test.DejaFu.Deterministic
( -- * The @Conc@ Monad
2015-01-25 19:15:23 +03:00
Conc
, Failure(..)
2015-01-25 19:15:23 +03:00
, runConc
, fork
, forkFinally
, forkWithUnmask
2015-02-18 03:13:12 +03:00
, forkOn
, getNumCapabilities
2015-02-13 03:39:27 +03:00
, myThreadId
, spawn
, atomically
2015-02-12 16:42:32 +03:00
, throw
2015-02-13 20:13:00 +03:00
, throwTo
, killThread
, Test.DejaFu.Deterministic.catch
, mask
, uninterruptibleMask
2015-01-25 19:15:23 +03:00
2015-02-20 19:14:50 +03:00
-- * @CVar@s
2015-01-25 19:15:23 +03:00
, CVar
, newEmptyCVar
, putCVar
, tryPutCVar
, readCVar
, takeCVar
, tryTakeCVar
2015-02-20 19:14:50 +03:00
-- * @CRef@s
, CRef
, newCRef
, readCRef
, writeCRef
, modifyCRef
-- * Testing
, _concNoTest
, _concKnowsAbout
, _concForgets
, _concAllKnown
-- * Execution traces
, Trace
, Decision(..)
, ThreadAction(..)
2015-07-08 22:05:58 +03:00
, ThreadAction'(..)
2015-02-03 17:14:36 +03:00
, CVarId
2015-02-20 19:14:50 +03:00
, CRefId
, MaskingState(..)
, showTrace
-- * Scheduling
, module Test.DejaFu.Deterministic.Schedule
) where
2015-01-25 19:15:23 +03:00
import Control.Applicative (Applicative(..), (<$>))
2015-02-13 21:30:21 +03:00
import Control.Exception (Exception, MaskingState(..), SomeException(..))
2015-01-25 19:15:23 +03:00
import Control.Monad.Cont (cont, runCont)
import Control.Monad.ST (ST, runST)
2015-02-10 01:04:28 +03:00
import Control.State (Wrapper(..), refST)
import Data.STRef (STRef, newSTRef)
2015-01-31 18:50:54 +03:00
import Test.DejaFu.Deterministic.Internal
import Test.DejaFu.Deterministic.Schedule
import Test.DejaFu.STM (STMLike, runTransactionST)
import Test.DejaFu.STM.Internal (CTVar(..))
2015-01-25 19:15:23 +03:00
2015-02-12 22:15:07 +03:00
import qualified Control.Monad.Catch as Ca
2015-01-25 19:15:23 +03:00
import qualified Control.Monad.Conc.Class as C
-- | The @Conc@ monad itself. This uses the same
-- universally-quantified indexing state trick as used by 'ST' and
-- 'STRef's to prevent mutable references from leaking out of the
-- monad.
newtype Conc t a = C { unC :: M (ST t) (STRef t) (STMLike t) a } deriving (Functor, Applicative, Monad)
2015-01-25 19:15:23 +03:00
2015-02-15 00:16:44 +03:00
wrap :: (M (ST t) (STRef t) (STMLike t) a -> M (ST t) (STRef t) (STMLike t) a) -> (Conc t a -> Conc t a)
wrap f = C . f . unC
2015-02-12 22:15:07 +03:00
instance Ca.MonadCatch (Conc t) where
catch = Test.DejaFu.Deterministic.catch
2015-02-12 22:15:07 +03:00
instance Ca.MonadThrow (Conc t) where
2015-02-12 16:42:32 +03:00
throwM = throw
instance Ca.MonadMask (Conc t) where
mask = mask
uninterruptibleMask = uninterruptibleMask
instance C.MonadConc (Conc t) where
2015-02-13 03:39:27 +03:00
type CVar (Conc t) = CVar t
2015-02-20 19:14:50 +03:00
type CRef (Conc t) = CRef t
2015-02-13 03:39:27 +03:00
type STMLike (Conc t) = STMLike t (ST t) (STRef t)
type ThreadId (Conc t) = Int
2015-01-25 19:15:23 +03:00
fork = fork
forkWithUnmask = forkWithUnmask
2015-02-18 03:13:12 +03:00
forkOn = forkOn
getNumCapabilities = getNumCapabilities
myThreadId = myThreadId
throwTo = throwTo
newEmptyCVar = newEmptyCVar
putCVar = putCVar
tryPutCVar = tryPutCVar
readCVar = readCVar
takeCVar = takeCVar
tryTakeCVar = tryTakeCVar
2015-02-20 19:14:50 +03:00
newCRef = newCRef
readCRef = readCRef
writeCRef = writeCRef
modifyCRef = modifyCRef
atomically = atomically
_concNoTest = _concNoTest
_concKnowsAbout = _concKnowsAbout
_concForgets = _concForgets
_concAllKnown = _concAllKnown
2015-01-25 19:15:23 +03:00
fixed :: Fixed (ST t) (STRef t) (STMLike t)
2015-02-10 01:04:28 +03:00
fixed = Wrapper refST $ \ma -> cont (\c -> ALift $ c <$> ma)
2015-01-25 19:15:23 +03:00
-- | The concurrent variable type used with the 'Conc' monad. One
-- notable difference between these and 'MVar's is that 'MVar's are
-- single-wakeup, and wake up in a FIFO order. Writing to a @CVar@
-- wakes up all threads blocked on reading it, and it is up to the
-- scheduler which one runs next. Taking from a @CVar@ behaves
-- analogously.
2015-02-20 19:14:50 +03:00
newtype CVar t a = Var { unV :: V (STRef t) a } deriving Eq
-- | The mutable non-blocking reference type. These are like 'IORef's,
-- but don't have the potential re-ordering problem mentioned in
-- Data.IORef.
newtype CRef t a = Ref { unR :: R (STRef t) a } deriving Eq
2015-01-25 19:15:23 +03:00
-- | Run the provided computation concurrently, returning the result.
spawn :: Conc t a -> Conc t (CVar t a)
spawn = C.spawn
2015-01-25 19:15:23 +03:00
-- | Block on a 'CVar' until it is full, then read from it (without
-- emptying).
readCVar :: CVar t a -> Conc t a
readCVar cvar = C $ cont $ AGet $ unV cvar
-- | Run the provided computation concurrently.
2015-02-13 03:39:27 +03:00
fork :: Conc t () -> Conc t ThreadId
2015-02-15 00:16:44 +03:00
fork (C ma) = C $ cont $ AFork (const $ runCont ma $ const AStop)
2015-02-13 03:39:27 +03:00
-- | Get the 'ThreadId' of the current thread.
myThreadId :: Conc t ThreadId
myThreadId = C $ cont AMyTId
2015-01-25 19:15:23 +03:00
-- | Run the provided 'MonadSTM' transaction atomically. If 'retry' is
-- called, it will be blocked until any of the touched 'CTVar's have
-- been written to.
atomically :: STMLike t (ST t) (STRef t) a -> Conc t a
atomically stm = C $ cont $ AAtom stm
2015-01-25 19:15:23 +03:00
-- | Create a new empty 'CVar'.
newEmptyCVar :: Conc t (CVar t a)
newEmptyCVar = C $ cont lifted where
2015-02-03 17:14:36 +03:00
lifted c = ANew $ \cvid -> c <$> newEmptyCVar' cvid
2015-02-20 19:14:50 +03:00
newEmptyCVar' cvid = (\ref -> Var (cvid, ref)) <$> newSTRef Nothing
2015-01-25 19:15:23 +03:00
-- | Block on a 'CVar' until it is empty, then write to it.
putCVar :: CVar t a -> a -> Conc t ()
putCVar cvar a = C $ cont $ \c -> APut (unV cvar) a $ c ()
-- | Put a value into a 'CVar' if there isn't one, without blocking.
tryPutCVar :: CVar t a -> a -> Conc t Bool
tryPutCVar cvar a = C $ cont $ ATryPut (unV cvar) a
-- | Block on a 'CVar' until it is full, then read from it (with
-- emptying).
takeCVar :: CVar t a -> Conc t a
takeCVar cvar = C $ cont $ ATake $ unV cvar
-- | Read a value from a 'CVar' if there is one, without blocking.
tryTakeCVar :: CVar t a -> Conc t (Maybe a)
tryTakeCVar cvar = C $ cont $ ATryTake $ unV cvar
2015-02-20 19:14:50 +03:00
-- | Create a new 'CRef'.
newCRef :: a -> Conc t (CRef t a)
newCRef a = C $ cont lifted where
lifted c = ANewRef $ \crid -> c <$> newCRef' crid
newCRef' crid = (\ref -> Ref (crid, ref)) <$> newSTRef a
-- | Read the value from a 'CRef'.
readCRef :: CRef t a -> Conc t a
readCRef ref = C $ cont $ AReadRef $ unR ref
-- | Atomically modify the value inside a 'CRef'.
modifyCRef :: CRef t a -> (a -> (a, b)) -> Conc t b
modifyCRef ref f = C $ cont $ AModRef (unR ref) f
-- | Replace the value stored inside a 'CRef'.
writeCRef :: CRef t a -> a -> Conc t ()
writeCRef ref a = modifyCRef ref $ const (a, ())
2015-02-12 16:42:32 +03:00
-- | Raise an exception in the 'Conc' monad. The exception is raised
-- when the action is run, not when it is applied. It short-citcuits
-- the rest of the computation:
--
-- > throw e >> x == throw e
throw :: Exception e => e -> Conc t a
throw e = C $ cont $ \_ -> AThrow (SomeException e)
2015-02-13 20:13:00 +03:00
-- | Throw an exception to the target thread. This blocks until the
-- exception is delivered, and it is just as if the target thread had
-- raised it with 'throw'. This can interrupt a blocked action.
throwTo :: Exception e => ThreadId -> e -> Conc t ()
throwTo tid e = C $ cont $ \c -> AThrowTo tid (SomeException e) $ c ()
-- | Raise the 'ThreadKilled' exception in the target thread. Note
-- that if the thread is prepared to catch this exception, it won't
-- actually kill it.
killThread :: ThreadId -> Conc t ()
2015-02-13 20:13:00 +03:00
killThread = C.killThread
2015-02-12 22:15:07 +03:00
-- | Catch an exception raised by 'throw'. This __cannot__ catch
-- errors, such as evaluating 'undefined', or division by zero. If you
-- need that, use Control.Exception.catch and 'ConcIO'.
catch :: Exception e => Conc t a -> (e -> Conc t a) -> Conc t a
2015-02-13 03:56:45 +03:00
catch ma h = C $ cont $ ACatching (unC . h) (unC ma)
2015-02-12 22:15:07 +03:00
-- | Fork a thread and call the supplied function when the thread is
-- about to terminate, with an exception or a returned value. The
-- function is called with asynchronous exceptions masked.
--
-- This function is useful for informing the parent when a child
-- terminates, for example.
forkFinally :: Conc t a -> (Either SomeException a -> Conc t ()) -> Conc t ThreadId
forkFinally action and_then =
mask $ \restore ->
fork $ Ca.try (restore action) >>= and_then
-- | Like 'fork', but the child thread is passed a function that can
2015-02-15 00:16:44 +03:00
-- be used to unmask asynchronous exceptions. This function should not
-- be used within a 'mask' or 'uninterruptibleMask'.
forkWithUnmask :: ((forall a. Conc t a -> Conc t a) -> Conc t ()) -> Conc t ThreadId
2015-02-15 00:16:44 +03:00
forkWithUnmask ma = C $ cont $ AFork (\umask -> runCont (unC $ ma $ wrap umask) $ const AStop)
-- | Executes a computation with asynchronous exceptions
-- /masked/. That is, any thread which attempts to raise an exception
-- in the current thread with 'throwTo' will be blocked until
-- asynchronous exceptions are unmasked again.
--
-- The argument passed to mask is a function that takes as its
-- argument another function, which can be used to restore the
-- prevailing masking state within the context of the masked
2015-02-15 00:16:44 +03:00
-- computation. This function should not be used within an
-- 'uninterruptibleMask'.
mask :: ((forall a. Conc t a -> Conc t a) -> Conc t b) -> Conc t b
2015-02-13 21:30:21 +03:00
-- Can't avoid the lambda here (and in uninterruptibleMask and in
2015-02-16 06:16:55 +03:00
-- ConcIO) because higher-ranked type inference is scary.
2015-02-15 00:16:44 +03:00
mask mb = C $ cont $ AMasking MaskedInterruptible (\f -> unC $ mb $ wrap f)
-- | Like 'mask', but the masked computation is not
-- interruptible. THIS SHOULD BE USED WITH GREAT CARE, because if a
-- thread executing in 'uninterruptibleMask' blocks for any reason,
-- then the thread (and possibly the program, if this is the main
-- thread) will be unresponsive and unkillable. This function should
-- only be necessary if you need to mask exceptions around an
-- interruptible operation, and you can guarantee that the
2015-02-15 00:16:44 +03:00
-- interruptible operation will only block for a short period of
-- time. The supplied unmasking function should not be used within a
-- 'mask'.
uninterruptibleMask :: ((forall a. Conc t a -> Conc t a) -> Conc t b) -> Conc t b
2015-02-15 00:16:44 +03:00
uninterruptibleMask mb = C $ cont $ AMasking MaskedUninterruptible (\f -> unC $ mb $ wrap f)
2015-02-18 03:13:12 +03:00
-- | Fork a computation to happen on a specific processor. This
-- implementation only has a single processor.
forkOn :: Int -> Conc t () -> Conc t ThreadId
forkOn _ = fork
-- | Get the number of Haskell threads that can run
-- simultaneously. This implementation lies and always returns
-- 2. There is no way to verify in the computation that this is a lie,
-- and will potentially avoid special-case behaviour for 1 capability,
-- so it seems a sane choice.
2015-02-18 03:13:12 +03:00
getNumCapabilities :: Conc t Int
getNumCapabilities = return 2
2015-02-18 03:13:12 +03:00
-- | Run the argument in one step. If the argument fails, the whole
-- computation will fail.
_concNoTest :: Conc t a -> Conc t a
_concNoTest ma = C $ cont $ \c -> ANoTest (unC ma) c
-- | Record that the referenced variable is known by the current thread.
_concKnowsAbout :: Either (CVar t a) (CTVar t (STRef t) a) -> Conc t ()
_concKnowsAbout (Left (Var (cvarid, _))) = C $ cont $ \c -> AKnowsAbout (Left cvarid) (c ())
_concKnowsAbout (Right (V (ctvarid, _))) = C $ cont $ \c -> AKnowsAbout (Right ctvarid) (c ())
-- | Record that the referenced variable will never be touched by the
-- current thread.
_concForgets :: Either (CVar t a) (CTVar t (STRef t) a) -> Conc t ()
_concForgets (Left (Var (cvarid, _))) = C $ cont $ \c -> AForgets (Left cvarid) (c ())
_concForgets (Right (V (ctvarid, _))) = C $ cont $ \c -> AForgets (Right ctvarid) (c ())
-- | Record that all 'CVar's and 'CTVar's known by the current thread
-- have been passed to '_concKnowsAbout'.
_concAllKnown :: Conc t ()
_concAllKnown = C $ cont $ \c -> AAllKnown (c ())
2015-01-25 19:15:23 +03:00
-- | Run a concurrent computation with a given 'Scheduler' and initial
-- state, returning a failure reason on error. Also returned is the
-- final state of the scheduler, and an execution trace.
2015-01-25 19:15:23 +03:00
--
-- Note how the @t@ in 'Conc' is universally quantified, what this
-- means in practice is that you can't do something like this:
--
-- > runConc (\s _ (x:_) -> (x, s)) () newEmptyCVar
2015-01-25 19:15:23 +03:00
--
-- So 'CVar's cannot leak out of the 'Conc' computation. If this is
-- making your head hurt, check out the \"How @runST@ works\" section
-- of <https://ocharles.org.uk/blog/guest-posts/2014-12-18-rank-n-types.html>
runConc :: Scheduler s -> s -> (forall t. Conc t a) -> (Either Failure a, s, Trace)
runConc sched s ma = runST $ runFixed fixed runTransactionST sched s $ unC ma