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