115 lines
3.1 KiB
Haskell
115 lines
3.1 KiB
Haskell
module Propellor.Property.Ssh (
|
|
setSshdConfig,
|
|
permitRootLogin,
|
|
passwordAuthentication,
|
|
hasAuthorizedKeys,
|
|
restartSshd,
|
|
uniqueHostKeys,
|
|
keyImported,
|
|
knownHost,
|
|
) where
|
|
|
|
import Propellor
|
|
import qualified Propellor.Property.File as File
|
|
import Propellor.Property.User
|
|
import Utility.SafeCommand
|
|
import Utility.FileMode
|
|
|
|
import System.PosixCompat
|
|
|
|
sshBool :: Bool -> String
|
|
sshBool True = "yes"
|
|
sshBool False = "no"
|
|
|
|
sshdConfig :: FilePath
|
|
sshdConfig = "/etc/ssh/sshd_config"
|
|
|
|
setSshdConfig :: String -> Bool -> Property
|
|
setSshdConfig setting allowed = combineProperties "sshd config"
|
|
[ sshdConfig `File.lacksLine` (sshline $ not allowed)
|
|
, sshdConfig `File.containsLine` (sshline allowed)
|
|
]
|
|
`onChange` restartSshd
|
|
`describe` unwords [ "ssh config:", setting, sshBool allowed ]
|
|
where
|
|
sshline v = setting ++ " " ++ sshBool v
|
|
|
|
permitRootLogin :: Bool -> Property
|
|
permitRootLogin = setSshdConfig "PermitRootLogin"
|
|
|
|
passwordAuthentication :: Bool -> Property
|
|
passwordAuthentication = setSshdConfig "PasswordAuthentication"
|
|
|
|
dotDir :: UserName -> IO FilePath
|
|
dotDir user = do
|
|
h <- homedir user
|
|
return $ h </> ".ssh"
|
|
|
|
dotFile :: FilePath -> UserName -> IO FilePath
|
|
dotFile f user = do
|
|
d <- dotDir user
|
|
return $ d </> f
|
|
|
|
hasAuthorizedKeys :: UserName -> IO Bool
|
|
hasAuthorizedKeys = go <=< dotFile "authorized_keys"
|
|
where
|
|
go f = not . null <$> catchDefaultIO "" (readFile f)
|
|
|
|
restartSshd :: Property
|
|
restartSshd = cmdProperty "service" ["ssh", "restart"]
|
|
|
|
-- | Blows away existing host keys and make new ones.
|
|
-- Useful for systems installed from an image that might reuse host keys.
|
|
-- A flag file is used to only ever do this once.
|
|
uniqueHostKeys :: Property
|
|
uniqueHostKeys = flagFile prop "/etc/ssh/.unique_host_keys"
|
|
`onChange` restartSshd
|
|
where
|
|
prop = Property "ssh unique host keys" $ do
|
|
void $ liftIO $ boolSystem "sh"
|
|
[ Param "-c"
|
|
, Param "rm -f /etc/ssh/ssh_host_*"
|
|
]
|
|
ensureProperty $
|
|
cmdProperty "/var/lib/dpkg/info/openssh-server.postinst"
|
|
["configure"]
|
|
|
|
-- | Sets up a user with a ssh private key from the site's privdata.
|
|
keyImported :: SshKeyType -> UserName -> Property
|
|
keyImported keytype user = propertyList desc
|
|
[ Property desc (install (SshPubKey keytype user) ".pub")
|
|
, Property desc (install (SshPrivKey keytype user) "")
|
|
]
|
|
where
|
|
desc = user ++ " has ssh key"
|
|
install p ext = do
|
|
f <- liftIO $ keyfile ext
|
|
ifM (liftIO $ doesFileExist f)
|
|
( noChange
|
|
, withPrivData p $ \key -> makeChange $
|
|
writeFileProtected f key
|
|
)
|
|
keyfile ext = do
|
|
home <- homeDirectory <$> getUserEntryForName user
|
|
return $ home </> ".ssh" </> "id_"
|
|
++ case keytype of
|
|
SshRsa -> "rsa"
|
|
SshDsa -> "dsa"
|
|
++ ext
|
|
|
|
-- | Puts some host's ssh public key into the known_hosts file for a user.
|
|
knownHost :: [Host] -> HostName -> UserName -> Property
|
|
knownHost hosts hn user = Property desc $
|
|
go =<< fromHost hosts hn getSshPubKey
|
|
where
|
|
desc = user ++ " knows ssh key for " ++ hn
|
|
go (Just (Just k)) = do
|
|
f <- liftIO $ dotFile "known_hosts" user
|
|
ensureProperty $ propertyList desc
|
|
[ File.dirExists (takeDirectory f)
|
|
, f `File.containsLine` (hn ++ " " ++ k)
|
|
]
|
|
go _ = do
|
|
warningMessage $ "no configred sshPubKey for " ++ hn
|
|
return FailedChange
|