blob: 97005eb770c08f08f01e26800d6e063d2ccaeb38 (
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
61
62
63
64
65
66
|
{-# LANGUAGE Trustworthy #-}
{-# LANGUAGE NoImplicitPrelude #-}
{-# LANGUAGE CPP #-}
module GHC.Environment (getFullArgs) where
import Foreign
import Foreign.C
import GHC.Base
import GHC.Real ( fromIntegral )
#ifdef mingw32_HOST_OS
import GHC.IO (finally)
import GHC.Windows
# if defined(i386_HOST_ARCH)
# define WINDOWS_CCONV stdcall
# elif defined(x86_64_HOST_ARCH)
# define WINDOWS_CCONV ccall
# else
# error Unknown mingw32 arch
# endif
#else
import GHC.IO.Encoding
import qualified GHC.Foreign as GHC
#endif
-- | Computation 'getFullArgs' is the "raw" version of 'getArgs', similar
-- to @argv@ in other languages. It returns a list of the program's
-- command line arguments, starting with the program name, and
-- including those normally eaten by the RTS (+RTS ... -RTS).
getFullArgs :: IO [String]
#ifdef mingw32_HOST_OS
-- Ignore the arguments to hs_init on Windows for the sake of Unicode compat
getFullArgs = do
p_arg_string <- c_GetCommandLine
alloca $ \p_argc -> do
p_argv <- c_CommandLineToArgv p_arg_string p_argc
if p_argv == nullPtr
then throwGetLastError "getFullArgs"
else flip finally (c_LocalFree p_argv) $ do
argc <- peek p_argc
p_argvs <- peekArray (fromIntegral argc) p_argv
mapM peekCWString p_argvs
foreign import WINDOWS_CCONV unsafe "windows.h GetCommandLineW"
c_GetCommandLine :: IO (Ptr CWString)
foreign import WINDOWS_CCONV unsafe "windows.h CommandLineToArgvW"
c_CommandLineToArgv :: Ptr CWString -> Ptr CInt -> IO (Ptr CWString)
foreign import WINDOWS_CCONV unsafe "Windows.h LocalFree"
c_LocalFree :: Ptr a -> IO (Ptr a)
#else
getFullArgs =
alloca $ \ p_argc ->
alloca $ \ p_argv -> do
getFullProgArgv p_argc p_argv
p <- fromIntegral `liftM` peek p_argc
argv <- peek p_argv
enc <- getFileSystemEncoding
peekArray p argv >>= mapM (GHC.peekCString enc)
foreign import ccall unsafe "getFullProgArgv"
getFullProgArgv :: Ptr CInt -> Ptr (Ptr CString) -> IO ()
#endif
|
