526243d6f5
This fixes a crash when using the linux_standalone build in termux on android. This commit was sponsored by Jeff Goeke-Smith on Patreon.
69 lines
1.6 KiB
Haskell
69 lines
1.6 KiB
Haskell
{- user info
|
|
-
|
|
- Copyright 2012 Joey Hess <id@joeyh.name>
|
|
-
|
|
- License: BSD-2-clause
|
|
-}
|
|
|
|
{-# LANGUAGE CPP #-}
|
|
{-# OPTIONS_GHC -fno-warn-tabs #-}
|
|
|
|
module Utility.UserInfo (
|
|
myHomeDir,
|
|
myUserName,
|
|
myUserGecos,
|
|
) where
|
|
|
|
import Utility.Env.Basic
|
|
import Utility.Exception
|
|
#ifndef mingw32_HOST_OS
|
|
import Utility.Data
|
|
import Control.Applicative
|
|
#endif
|
|
|
|
import System.PosixCompat
|
|
import Prelude
|
|
|
|
{- Current user's home directory.
|
|
-
|
|
- getpwent will fail on LDAP or NIS, so use HOME if set. -}
|
|
myHomeDir :: IO FilePath
|
|
myHomeDir = either giveup return =<< myVal env homeDirectory
|
|
where
|
|
#ifndef mingw32_HOST_OS
|
|
env = ["HOME"]
|
|
#else
|
|
env = ["USERPROFILE", "HOME"] -- HOME is used in Cygwin
|
|
#endif
|
|
|
|
{- Current user's user name. -}
|
|
myUserName :: IO (Either String String)
|
|
myUserName = myVal env userName
|
|
where
|
|
#ifndef mingw32_HOST_OS
|
|
env = ["USER", "LOGNAME"]
|
|
#else
|
|
env = ["USERNAME", "USER", "LOGNAME"]
|
|
#endif
|
|
|
|
myUserGecos :: IO (Maybe String)
|
|
-- userGecos crashes on Android and is not available on Windows.
|
|
#if defined(__ANDROID__) || defined(mingw32_HOST_OS)
|
|
myUserGecos = return Nothing
|
|
#else
|
|
myUserGecos = eitherToMaybe <$> myVal [] userGecos
|
|
#endif
|
|
|
|
myVal :: [String] -> (UserEntry -> String) -> IO (Either String String)
|
|
myVal envvars extract = go envvars
|
|
where
|
|
go [] = either (const $ envnotset) (Right . extract) <$> get
|
|
go (v:vs) = maybe (go vs) (return . Right) =<< getEnv v
|
|
#ifndef mingw32_HOST_OS
|
|
-- This may throw an exception if the system doesn't have a
|
|
-- passwd file etc; don't let it crash.
|
|
get = tryNonAsync $ getUserEntryForID =<< getEffectiveUserID
|
|
#else
|
|
get = return envnotset
|
|
#endif
|
|
envnotset = Left ("environment not set: " ++ show envvars)
|