{-# LINE 1 "libraries/unix/System/Posix/Process/Common.hsc" #-}
{-# LANGUAGE CApiFFI #-}
{-# LANGUAGE InterruptibleFFI, RankNTypes #-}
{-# LANGUAGE Trustworthy #-}
module System.Posix.Process.Common (
forkProcess,
forkProcessWithUnmask,
exitImmediately,
getProcessID,
getParentProcessID,
getProcessGroupID,
getProcessGroupIDOf,
createProcessGroupFor,
joinProcessGroup,
setProcessGroupIDOf,
createSession,
ProcessTimes(..),
getProcessTimes,
nice,
getProcessPriority,
getProcessGroupPriority,
getUserPriority,
setProcessPriority,
setProcessGroupPriority,
setUserPriority,
ProcessStatus(..),
getProcessStatus,
getAnyProcessStatus,
getGroupProcessStatus,
createProcessGroup,
setProcessGroupID,
) where
import Foreign.C.Error
import Foreign.C.Types
import Foreign.Marshal.Alloc ( alloca, allocaBytes )
import Foreign.Ptr ( Ptr )
import Foreign.StablePtr ( StablePtr, newStablePtr, freeStablePtr )
import Foreign.Storable ( Storable(..) )
import System.Exit
import System.Posix.Process.Internals
import System.Posix.Types
import Control.Monad
import Control.Exception.Base ( bracket, getMaskingState, MaskingState(..) )
import GHC.TopHandler ( runIO )
import GHC.IO ( unsafeUnmask, uninterruptibleMask_ )
getProcessID :: IO ProcessID
getProcessID :: IO ProcessID
getProcessID = IO ProcessID
c_getpid
foreign import ccall unsafe "getpid"
c_getpid :: IO CPid
getParentProcessID :: IO ProcessID
getParentProcessID :: IO ProcessID
getParentProcessID = IO ProcessID
c_getppid
foreign import ccall unsafe "getppid"
c_getppid :: IO CPid
getProcessGroupID :: IO ProcessGroupID
getProcessGroupID :: IO ProcessID
getProcessGroupID = IO ProcessID
c_getpgrp
foreign import ccall unsafe "getpgrp"
c_getpgrp :: IO CPid
getProcessGroupIDOf :: ProcessID -> IO ProcessGroupID
getProcessGroupIDOf :: ProcessID -> IO ProcessID
getProcessGroupIDOf ProcessID
pid =
String -> IO ProcessID -> IO ProcessID
forall a. (Eq a, Num a) => String -> IO a -> IO a
throwErrnoIfMinus1 String
"getProcessGroupIDOf" (ProcessID -> IO ProcessID
c_getpgid ProcessID
pid)
foreign import ccall unsafe "getpgid"
c_getpgid :: CPid -> IO CPid
createProcessGroupFor :: ProcessID -> IO ProcessGroupID
createProcessGroupFor :: ProcessID -> IO ProcessID
createProcessGroupFor ProcessID
pid = do
String -> IO CInt -> IO ()
forall a. (Eq a, Num a) => String -> IO a -> IO ()
throwErrnoIfMinus1_ String
"createProcessGroupFor" (ProcessID -> ProcessID -> IO CInt
c_setpgid ProcessID
pid ProcessID
0)
ProcessID -> IO ProcessID
forall (m :: * -> *) a. Monad m => a -> m a
return ProcessID
pid
joinProcessGroup :: ProcessGroupID -> IO ()
joinProcessGroup :: ProcessID -> IO ()
joinProcessGroup ProcessID
pgid =
String -> IO CInt -> IO ()
forall a. (Eq a, Num a) => String -> IO a -> IO ()
throwErrnoIfMinus1_ String
"joinProcessGroup" (ProcessID -> ProcessID -> IO CInt
c_setpgid ProcessID
0 ProcessID
pgid)
setProcessGroupIDOf :: ProcessID -> ProcessGroupID -> IO ()
setProcessGroupIDOf :: ProcessID -> ProcessID -> IO ()
setProcessGroupIDOf ProcessID
pid ProcessID
pgid =
String -> IO CInt -> IO ()
forall a. (Eq a, Num a) => String -> IO a -> IO ()
throwErrnoIfMinus1_ String
"setProcessGroupIDOf" (ProcessID -> ProcessID -> IO CInt
c_setpgid ProcessID
pid ProcessID
pgid)
foreign import ccall unsafe "setpgid"
c_setpgid :: CPid -> CPid -> IO CInt
createSession :: IO ProcessGroupID
createSession :: IO ProcessID
createSession = String -> IO ProcessID -> IO ProcessID
forall a. (Eq a, Num a) => String -> IO a -> IO a
throwErrnoIfMinus1 String
"createSession" IO ProcessID
c_setsid
foreign import ccall unsafe "setsid"
c_setsid :: IO CPid
data ProcessTimes
= ProcessTimes { ProcessTimes -> ClockTick
elapsedTime :: ClockTick
, ProcessTimes -> ClockTick
userTime :: ClockTick
, ProcessTimes -> ClockTick
systemTime :: ClockTick
, ProcessTimes -> ClockTick
childUserTime :: ClockTick
, ProcessTimes -> ClockTick
childSystemTime :: ClockTick
}
getProcessTimes :: IO ProcessTimes
getProcessTimes :: IO ProcessTimes
getProcessTimes = do
Int -> (Ptr CTms -> IO ProcessTimes) -> IO ProcessTimes
forall a b. Int -> (Ptr a -> IO b) -> IO b
allocaBytes (Int
32) ((Ptr CTms -> IO ProcessTimes) -> IO ProcessTimes)
-> (Ptr CTms -> IO ProcessTimes) -> IO ProcessTimes
forall a b. (a -> b) -> a -> b
$ \Ptr CTms
p_tms -> do
{-# LINE 194 "libraries/unix/System/Posix/Process/Common.hsc" #-}
elapsed <- throwErrnoIfMinus1 "getProcessTimes" (c_times p_tms)
ut <- ((\hsc_ptr -> peekByteOff hsc_ptr 0)) p_tms
{-# LINE 196 "libraries/unix/System/Posix/Process/Common.hsc" #-}
st <- ((\hsc_ptr -> peekByteOff hsc_ptr 8)) p_tms
{-# LINE 197 "libraries/unix/System/Posix/Process/Common.hsc" #-}
cut <- ((\hsc_ptr -> peekByteOff hsc_ptr 16)) p_tms
{-# LINE 198 "libraries/unix/System/Posix/Process/Common.hsc" #-}
cst <- ((\hsc_ptr -> peekByteOff hsc_ptr 24)) p_tms
{-# LINE 199 "libraries/unix/System/Posix/Process/Common.hsc" #-}
return (ProcessTimes{ elapsedTime = elapsed,
userTime = ut,
systemTime = st,
childUserTime = cut,
childSystemTime = cst
})
data {-# CTYPE "struct tms" #-} CTms
foreign import capi unsafe "HsUnix.h times"
c_times :: Ptr CTms -> IO CClock
nice :: Int -> IO ()
nice :: Int -> IO ()
nice Int
prio = do
IO ()
resetErrno
CInt
res <- CInt -> IO CInt
c_nice (Int -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
prio)
Bool -> IO () -> IO ()
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (CInt
res CInt -> CInt -> Bool
forall a. Eq a => a -> a -> Bool
== -CInt
1) (IO () -> IO ()) -> IO () -> IO ()
forall a b. (a -> b) -> a -> b
$ do
Errno
err <- IO Errno
getErrno
Bool -> IO () -> IO ()
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (Errno
err Errno -> Errno -> Bool
forall a. Eq a => a -> a -> Bool
/= Errno
eOK) (String -> IO ()
forall a. String -> IO a
throwErrno String
"nice")
foreign import ccall unsafe "nice"
c_nice :: CInt -> IO CInt
getProcessPriority :: ProcessID -> IO Int
getProcessGroupPriority :: ProcessGroupID -> IO Int
getUserPriority :: UserID -> IO Int
getProcessPriority :: ProcessID -> IO Int
getProcessPriority ProcessID
pid = do
CInt
r <- String -> IO CInt -> IO CInt
forall a. (Eq a, Num a) => String -> IO a -> IO a
throwErrnoIfMinus1 String
"getProcessPriority" (IO CInt -> IO CInt) -> IO CInt -> IO CInt
forall a b. (a -> b) -> a -> b
$
CInt -> CInt -> IO CInt
c_getpriority (CInt
0) (ProcessID -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral ProcessID
pid)
{-# LINE 232 "libraries/unix/System/Posix/Process/Common.hsc" #-}
return (fromIntegral r)
getProcessGroupPriority :: ProcessID -> IO Int
getProcessGroupPriority ProcessID
pid = do
CInt
r <- String -> IO CInt -> IO CInt
forall a. (Eq a, Num a) => String -> IO a -> IO a
throwErrnoIfMinus1 String
"getProcessPriority" (IO CInt -> IO CInt) -> IO CInt -> IO CInt
forall a b. (a -> b) -> a -> b
$
CInt -> CInt -> IO CInt
c_getpriority (CInt
1) (ProcessID -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral ProcessID
pid)
{-# LINE 237 "libraries/unix/System/Posix/Process/Common.hsc" #-}
return (fromIntegral r)
getUserPriority :: UserID -> IO Int
getUserPriority UserID
uid = do
CInt
r <- String -> IO CInt -> IO CInt
forall a. (Eq a, Num a) => String -> IO a -> IO a
throwErrnoIfMinus1 String
"getUserPriority" (IO CInt -> IO CInt) -> IO CInt -> IO CInt
forall a b. (a -> b) -> a -> b
$
CInt -> CInt -> IO CInt
c_getpriority (CInt
2) (UserID -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral UserID
uid)
{-# LINE 242 "libraries/unix/System/Posix/Process/Common.hsc" #-}
return (fromIntegral r)
foreign import ccall unsafe "getpriority"
c_getpriority :: CInt -> CInt -> IO CInt
setProcessPriority :: ProcessID -> Int -> IO ()
setProcessGroupPriority :: ProcessGroupID -> Int -> IO ()
setUserPriority :: UserID -> Int -> IO ()
setProcessPriority :: ProcessID -> Int -> IO ()
setProcessPriority ProcessID
pid Int
val =
String -> IO CInt -> IO ()
forall a. (Eq a, Num a) => String -> IO a -> IO ()
throwErrnoIfMinus1_ String
"setProcessPriority" (IO CInt -> IO ()) -> IO CInt -> IO ()
forall a b. (a -> b) -> a -> b
$
CInt -> CInt -> CInt -> IO CInt
c_setpriority (CInt
0) (ProcessID -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral ProcessID
pid) (Int -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
val)
{-# LINE 254 "libraries/unix/System/Posix/Process/Common.hsc" #-}
setProcessGroupPriority :: ProcessID -> Int -> IO ()
setProcessGroupPriority ProcessID
pid Int
val =
String -> IO CInt -> IO ()
forall a. (Eq a, Num a) => String -> IO a -> IO ()
throwErrnoIfMinus1_ String
"setProcessPriority" (IO CInt -> IO ()) -> IO CInt -> IO ()
forall a b. (a -> b) -> a -> b
$
CInt -> CInt -> CInt -> IO CInt
c_setpriority (CInt
1) (ProcessID -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral ProcessID
pid) (Int -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
val)
{-# LINE 258 "libraries/unix/System/Posix/Process/Common.hsc" #-}
setUserPriority :: UserID -> Int -> IO ()
setUserPriority UserID
uid Int
val =
String -> IO CInt -> IO ()
forall a. (Eq a, Num a) => String -> IO a -> IO ()
throwErrnoIfMinus1_ String
"setUserPriority" (IO CInt -> IO ()) -> IO CInt -> IO ()
forall a b. (a -> b) -> a -> b
$
CInt -> CInt -> CInt -> IO CInt
c_setpriority (CInt
2) (UserID -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral UserID
uid) (Int -> CInt
forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
val)
{-# LINE 262 "libraries/unix/System/Posix/Process/Common.hsc" #-}
foreign import ccall unsafe "setpriority"
c_setpriority :: CInt -> CInt -> CInt -> IO CInt
forkProcess :: IO () -> IO ProcessID
forkProcess :: IO () -> IO ProcessID
forkProcess IO ()
action = do
MaskingState
mstate <- IO MaskingState
getMaskingState
let action' :: IO ()
action' = case MaskingState
mstate of
MaskingState
Unmasked -> IO () -> IO ()
forall a. IO a -> IO a
unsafeUnmask IO ()
action
MaskingState
MaskedInterruptible -> IO ()
action
MaskingState
MaskedUninterruptible -> IO () -> IO ()
forall a. IO a -> IO a
uninterruptibleMask_ IO ()
action
IO (StablePtr (IO ()))
-> (StablePtr (IO ()) -> IO ())
-> (StablePtr (IO ()) -> IO ProcessID)
-> IO ProcessID
forall a b c. IO a -> (a -> IO b) -> (a -> IO c) -> IO c
bracket
(IO () -> IO (StablePtr (IO ()))
forall a. a -> IO (StablePtr a)
newStablePtr (IO () -> IO ()
forall a. IO a -> IO a
runIO IO ()
action'))
StablePtr (IO ()) -> IO ()
forall a. StablePtr a -> IO ()
freeStablePtr
(\StablePtr (IO ())
stable -> String -> IO ProcessID -> IO ProcessID
forall a. (Eq a, Num a) => String -> IO a -> IO a
throwErrnoIfMinus1 String
"forkProcess" (StablePtr (IO ()) -> IO ProcessID
forkProcessPrim StablePtr (IO ())
stable))
foreign import ccall "forkProcess" forkProcessPrim :: StablePtr (IO ()) -> IO CPid
forkProcessWithUnmask :: ((forall a . IO a -> IO a) -> IO ()) -> IO ProcessID
forkProcessWithUnmask :: ((forall a. IO a -> IO a) -> IO ()) -> IO ProcessID
forkProcessWithUnmask (forall a. IO a -> IO a) -> IO ()
action = IO () -> IO ProcessID
forkProcess ((forall a. IO a -> IO a) -> IO ()
action forall a. IO a -> IO a
unsafeUnmask)
getProcessStatus :: Bool -> Bool -> ProcessID -> IO (Maybe ProcessStatus)
getProcessStatus :: Bool -> Bool -> ProcessID -> IO (Maybe ProcessStatus)
getProcessStatus Bool
block Bool
stopped ProcessID
pid =
(Ptr CInt -> IO (Maybe ProcessStatus)) -> IO (Maybe ProcessStatus)
forall a b. Storable a => (Ptr a -> IO b) -> IO b
alloca ((Ptr CInt -> IO (Maybe ProcessStatus))
-> IO (Maybe ProcessStatus))
-> (Ptr CInt -> IO (Maybe ProcessStatus))
-> IO (Maybe ProcessStatus)
forall a b. (a -> b) -> a -> b
$ \Ptr CInt
wstatp -> do
ProcessID
pid' <- String -> IO ProcessID -> IO ProcessID
forall a. (Eq a, Num a) => String -> IO a -> IO a
throwErrnoIfMinus1Retry String
"getProcessStatus"
(ProcessID -> Ptr CInt -> CInt -> IO ProcessID
c_waitpid ProcessID
pid Ptr CInt
wstatp (Bool -> Bool -> CInt
waitOptions Bool
block Bool
stopped))
case ProcessID
pid' of
ProcessID
0 -> Maybe ProcessStatus -> IO (Maybe ProcessStatus)
forall (m :: * -> *) a. Monad m => a -> m a
return Maybe ProcessStatus
forall a. Maybe a
Nothing
ProcessID
_ -> do ProcessStatus
ps <- Ptr CInt -> IO ProcessStatus
readWaitStatus Ptr CInt
wstatp
Maybe ProcessStatus -> IO (Maybe ProcessStatus)
forall (m :: * -> *) a. Monad m => a -> m a
return (ProcessStatus -> Maybe ProcessStatus
forall a. a -> Maybe a
Just ProcessStatus
ps)
foreign import ccall interruptible "waitpid"
c_waitpid :: CPid -> Ptr CInt -> CInt -> IO CPid
getGroupProcessStatus :: Bool
-> Bool
-> ProcessGroupID
-> IO (Maybe (ProcessID, ProcessStatus))
getGroupProcessStatus :: Bool -> Bool -> ProcessID -> IO (Maybe (ProcessID, ProcessStatus))
getGroupProcessStatus Bool
block Bool
stopped ProcessID
pgid =
(Ptr CInt -> IO (Maybe (ProcessID, ProcessStatus)))
-> IO (Maybe (ProcessID, ProcessStatus))
forall a b. Storable a => (Ptr a -> IO b) -> IO b
alloca ((Ptr CInt -> IO (Maybe (ProcessID, ProcessStatus)))
-> IO (Maybe (ProcessID, ProcessStatus)))
-> (Ptr CInt -> IO (Maybe (ProcessID, ProcessStatus)))
-> IO (Maybe (ProcessID, ProcessStatus))
forall a b. (a -> b) -> a -> b
$ \Ptr CInt
wstatp -> do
ProcessID
pid <- String -> IO ProcessID -> IO ProcessID
forall a. (Eq a, Num a) => String -> IO a -> IO a
throwErrnoIfMinus1Retry String
"getGroupProcessStatus"
(ProcessID -> Ptr CInt -> CInt -> IO ProcessID
c_waitpid (-ProcessID
pgid) Ptr CInt
wstatp (Bool -> Bool -> CInt
waitOptions Bool
block Bool
stopped))
case ProcessID
pid of
ProcessID
0 -> Maybe (ProcessID, ProcessStatus)
-> IO (Maybe (ProcessID, ProcessStatus))
forall (m :: * -> *) a. Monad m => a -> m a
return Maybe (ProcessID, ProcessStatus)
forall a. Maybe a
Nothing
ProcessID
_ -> do ProcessStatus
ps <- Ptr CInt -> IO ProcessStatus
readWaitStatus Ptr CInt
wstatp
Maybe (ProcessID, ProcessStatus)
-> IO (Maybe (ProcessID, ProcessStatus))
forall (m :: * -> *) a. Monad m => a -> m a
return ((ProcessID, ProcessStatus) -> Maybe (ProcessID, ProcessStatus)
forall a. a -> Maybe a
Just (ProcessID
pid, ProcessStatus
ps))
getAnyProcessStatus :: Bool -> Bool -> IO (Maybe (ProcessID, ProcessStatus))
getAnyProcessStatus :: Bool -> Bool -> IO (Maybe (ProcessID, ProcessStatus))
getAnyProcessStatus Bool
block Bool
stopped = Bool -> Bool -> ProcessID -> IO (Maybe (ProcessID, ProcessStatus))
getGroupProcessStatus Bool
block Bool
stopped ProcessID
1
waitOptions :: Bool -> Bool -> CInt
waitOptions :: Bool -> Bool -> CInt
waitOptions Bool
False Bool
False = (CInt
1)
{-# LINE 370 "libraries/unix/System/Posix/Process/Common.hsc" #-}
waitOptions False True = (3)
{-# LINE 371 "libraries/unix/System/Posix/Process/Common.hsc" #-}
waitOptions True False = 0
waitOptions Bool
True Bool
True = (CInt
2)
{-# LINE 373 "libraries/unix/System/Posix/Process/Common.hsc" #-}
readWaitStatus :: Ptr CInt -> IO ProcessStatus
readWaitStatus :: Ptr CInt -> IO ProcessStatus
readWaitStatus Ptr CInt
wstatp = do
CInt
wstat <- Ptr CInt -> IO CInt
forall a. Storable a => Ptr a -> IO a
peek Ptr CInt
wstatp
CInt -> IO ProcessStatus
decipherWaitStatus CInt
wstat
exitImmediately :: ExitCode -> IO ()
exitImmediately :: ExitCode -> IO ()
exitImmediately ExitCode
exitcode = CInt -> IO ()
c_exit (ExitCode -> CInt
forall p. Num p => ExitCode -> p
exitcode2Int ExitCode
exitcode)
where
exitcode2Int :: ExitCode -> p
exitcode2Int ExitCode
ExitSuccess = p
0
exitcode2Int (ExitFailure Int
n) = Int -> p
forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
n
foreign import ccall unsafe "exit"
c_exit :: CInt -> IO ()
{-# DEPRECATED createProcessGroup "This function is scheduled to be replaced by something different in the future, we therefore recommend that you do not use this version and use 'createProcessGroupFor' instead." #-}
createProcessGroup :: ProcessID -> IO ProcessGroupID
createProcessGroup :: ProcessID -> IO ProcessID
createProcessGroup ProcessID
pid = do
String -> IO CInt -> IO ()
forall a. (Eq a, Num a) => String -> IO a -> IO ()
throwErrnoIfMinus1_ String
"createProcessGroup" (ProcessID -> ProcessID -> IO CInt
c_setpgid ProcessID
pid ProcessID
0)
ProcessID -> IO ProcessID
forall (m :: * -> *) a. Monad m => a -> m a
return ProcessID
pid
{-# DEPRECATED setProcessGroupID "This function is scheduled to be replaced by something different in the future, we therefore recommend that you do not use this version and use 'setProcessGroupIDOf' instead." #-}
setProcessGroupID :: ProcessID -> ProcessGroupID -> IO ()
setProcessGroupID :: ProcessID -> ProcessID -> IO ()
setProcessGroupID ProcessID
pid ProcessID
pgid =
String -> IO CInt -> IO ()
forall a. (Eq a, Num a) => String -> IO a -> IO ()
throwErrnoIfMinus1_ String
"setProcessGroupID" (ProcessID -> ProcessID -> IO CInt
c_setpgid ProcessID
pid ProcessID
pgid)