forked from haskell-unordered-containers/hashable
-
Notifications
You must be signed in to change notification settings - Fork 0
/
RandomSource.hs
29 lines (24 loc) · 833 Bytes
/
RandomSource.hs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
{-# LANGUAGE CPP, ForeignFunctionInterface #-}
module Data.Hashable.RandomSource
(
getRandomBytes
, getRandomBytes_
) where
import Data.ByteString as B
import Data.ByteString.Internal (create)
import Foreign.C.Error (throwErrnoIfMinus1_)
#if MIN_VERSION_base(4,5,0)
import Foreign.C.Types (CInt(CInt))
#else
import Foreign.C.Types (CInt)
#endif
import Foreign.Ptr (Ptr)
getRandomBytes :: Int -> IO ByteString
getRandomBytes nbytes
| nbytes <= 0 = return B.empty
| otherwise = create nbytes $ flip (getRandomBytes_ "getRandomBytes") nbytes
getRandomBytes_ :: String -> Ptr a -> Int -> IO ()
getRandomBytes_ what ptr nbytes = do
throwErrnoIfMinus1_ what $ c_getRandomBytes ptr (fromIntegral nbytes)
foreign import ccall unsafe "hashable_getRandomBytes" c_getRandomBytes
:: Ptr a -> CInt -> IO CInt