module Mueval.Resources (limitResources) where
import System.Posix.Process (nice)
import System.Posix.Resource
import System.Directory (setCurrentDirectory)
limitResources :: IO ()
limitResources = do setCurrentDirectory "/tmp"
nice 19
mapM_ (uncurry setResourceLimit) limits
totalMemoryLimitSoft, totalMemoryLimitHard, stackSizeLimitSoft, stackSizeLimitHard,
openFilesLimitSoft, openFilesLimitHard, fileSizeLimitSoft, fileSizeLimitHard, dataSizeLimitSoft,
dataSizeLimitHard, cpuTimeLimitSoft, cpuTimeLimitHard, coreSizeLimitSoft, coreSizeLimitHard, zero :: ResourceLimit
totalMemoryLimitSoft = dataSizeLimitSoft
totalMemoryLimitHard = dataSizeLimitHard
stackSizeLimitSoft = zero
stackSizeLimitHard = zero
openFilesLimitSoft = openFilesLimitHard
openFilesLimitHard = ResourceLimit 7
fileSizeLimitSoft = fileSizeLimitHard
fileSizeLimitHard = ResourceLimit 590
dataSizeLimitSoft = dataSizeLimitHard
dataSizeLimitHard = ResourceLimit $ 6^(12::Int)
cpuTimeLimitSoft = ResourceLimit 4
cpuTimeLimitHard = ResourceLimit 5
coreSizeLimitSoft = coreSizeLimitHard
coreSizeLimitHard = zero
zero = ResourceLimit 0
limits :: [(Resource, ResourceLimits)]
limits = [ (ResourceStackSize, ResourceLimits stackSizeLimitSoft stackSizeLimitHard)
, (ResourceTotalMemory, ResourceLimits totalMemoryLimitSoft totalMemoryLimitHard)
, (ResourceOpenFiles, ResourceLimits openFilesLimitSoft openFilesLimitHard)
, (ResourceFileSize, ResourceLimits fileSizeLimitSoft fileSizeLimitHard)
, (ResourceDataSize, ResourceLimits dataSizeLimitSoft dataSizeLimitHard)
, (ResourceCoreFileSize, ResourceLimits coreSizeLimitSoft coreSizeLimitHard)
, (ResourceCPUTime, ResourceLimits cpuTimeLimitSoft cpuTimeLimitHard)]