blob: f10c2e7c47824fd04e7cc9b7a826db8bbd031916 (
plain)
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
module Main where
import Foreign
import Foreign.C
import Text.Printf
import System.Exit
import Control.Monad
foreign import ccall "getesp" getesp :: IO CInt
main = do
checkSpAlignment
wrap checkSpAlignment >>= run
wrap1 args1 >>= \f -> run1 f 3
wrap2 args2 >>= \f -> run2 f 3 4
wrap3 args3 >>= \f -> run3 f 3 4 5
wrap4 args4 >>= \f -> run4 f 3 4 5 6
foreign import ccall "wrapper" wrap :: IO () -> IO (FunPtr (IO ()))
foreign import ccall "dynamic" run :: FunPtr (IO ()) -> IO ()
type Args1 = Int -> IO ()
foreign import ccall "wrapper" wrap1 :: Args1 -> IO (FunPtr Args1)
foreign import ccall "dynamic" run1 :: FunPtr Args1 -> Args1
args1 :: Args1
args1 _ = checkSpAlignment
type Args2 = Int -> Int -> IO ()
foreign import ccall "wrapper" wrap2 :: Args2 -> IO (FunPtr Args2)
foreign import ccall "dynamic" run2 :: FunPtr Args2 -> Args2
args2 :: Args2
args2 _ _ = checkSpAlignment
type Args3 = Int -> Int -> Int -> IO ()
foreign import ccall "wrapper" wrap3 :: Args3 -> IO (FunPtr Args3)
foreign import ccall "dynamic" run3 :: FunPtr Args3 -> Args3
args3 :: Args3
args3 _ _ _ = checkSpAlignment
type Args4 = Int -> Int -> Int -> Int -> IO ()
foreign import ccall "wrapper" wrap4 :: Args4 -> IO (FunPtr Args4)
foreign import ccall "dynamic" run4 :: FunPtr Args4 -> Args4
args4 :: Args4
args4 _ _ _ _ = checkSpAlignment
checkSpAlignment :: IO ()
checkSpAlignment = do
esp <- getesp
when (((esp + fromIntegral (sizeOf (undefined :: Ptr ()))) .&. 15) /= 0) $ do
printf "esp not aligned correctly: %x\n" (fromIntegral esp :: Word32)
exitWith (ExitFailure 1)
|
