#!/bin/bash
# vim: dict+=/usr/share/beakerlib/dictionary.vim cpt=.,w,b,u,t,i,k
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#
#   runtest.sh of /CoreOS/openssh/Sanity/pam_ssh_agent_auth
#   Description: This is a basic sanity test for pam_ssh_agent_auth
#   Author: Jakub Jelen <jjelen@redhat.com>
#
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#
#   Copyright (c) 2015 Red Hat, Inc.
#
#   This program is free software: you can redistribute it and/or
#   modify it under the terms of the GNU General Public License as
#   published by the Free Software Foundation, either version 2 of
#   the License, or (at your option) any later version.
#
#   This program is distributed in the hope that it will be
#   useful, but WITHOUT ANY WARRANTY; without even the implied
#   warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
#   PURPOSE.  See the GNU General Public License for more details.
#
#   You should have received a copy of the GNU General Public License
#   along with this program. If not, see http://www.gnu.org/licenses/.
#
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

# Include Beaker environment
. /usr/bin/rhts-environment.sh || exit 1
. /usr/share/beakerlib/beakerlib.sh || exit 1

PACKAGE="openssh"
PAM_SUDO="/etc/pam.d/sudo"
PAM_SSHD="/etc/pam.d/sshd"
PAM_MODULE="pam_save_ssh_var"
SUDOERS_CFG="/etc/sudoers.d/01_pam_ssh_auth"
SSHD_CFG="/etc/ssh/sshd_config"
USER="testuser$RANDOM"
PASS="testpassxy4re.3298fhdsaf"
AUTH_KEYS="/etc/security/authorized_keys"
AK_COMMAND_BIN="/root/ak.sh"
AK_COMMAND_KEYS="/root/akeys"
declare -a KEYS=("rsa" "ecdsa")

rlJournalStart
    rlPhaseStartSetup
        rlAssertRpm $PACKAGE
        rlAssertRpm pam_ssh_agent_auth
        rlImport distribution/fips
        rlServiceStart sshd
        rlRun "TmpDir=\$(mktemp -d)" 0 "Creating tmp directory"
        rlRun "cp ${PAM_MODULE}.c $TmpDir/"
        rlRun "pushd $TmpDir"
        rlFileBackup --clean $PAM_SUDO /etc/sudoers /etc/sudoers.d/ /etc/security/ $AUTH_KEYS
        rlRun "sed -i '1 a\
auth       sufficient   pam_ssh_agent_auth.so file=$AUTH_KEYS' $PAM_SUDO"
        rlRun "echo 'Defaults    env_keep += \"SSH_AUTH_SOCK\"' > $SUDOERS_CFG"
        rlRun "echo 'Defaults    !requiretty' >> $SUDOERS_CFG"
        grep '^%wheel' /etc/sudoers || \
           rlRun "echo '%wheel        ALL=(ALL)       ALL' >> $SUDOERS_CFG"
        rlRun "useradd $USER -G wheel"
        rlRun "echo $PASS |passwd --stdin $USER"
    rlPhaseEnd

    if ! fipsIsEnabled; then
        KEYS+=("dsa")
    fi

    for KEY in "${KEYS[@]}"; do
        rlPhaseStartTest "Test with key type $KEY"
            rlRun "su $USER -c 'ssh-keygen -t $KEY -f ~/.ssh/my_id_$KEY -N \"\"'" 0

            # Without authorized_keys, the authentication should fail
            rlRun -s "su $USER -c 'eval \`ssh-agent\`; sudo id; ssh-agent -k'" 0
            rlAssertNotGrep "uid=0(root) gid=0(root)" $rlRun_LOG

            # Append the keys only to make sure we can match also the non-first line
            rlRun "cat ~$USER/.ssh/my_id_${KEY}.pub >> $AUTH_KEYS"
            rlRun -s "su $USER -c 'eval \`ssh-agent\`; ssh-add ~/.ssh/my_id_$KEY; sudo id; ssh-agent -k'"
            rlAssertGrep "uid=0(root) gid=0(root)" $rlRun_LOG
        rlPhaseEnd
    done

    if rlIsRHEL '<6.8' || ( rlIsRHEL '<7.3' && rlIsRHEL 7 ) ; then
        : # not available
    else
        rlPhaseStartSetup "Setup for authorized_keys_command"
            rlFileBackup --namespace ak_command $PAM_SUDO
            rlRun "rm -f $AUTH_KEYS"
            cat >$AK_COMMAND_BIN <<_EOF
#!/bin/bash
cat $AK_COMMAND_KEYS
_EOF
            rlRun "chmod +x $AK_COMMAND_BIN"
            rlRun "sed -i 's|.*pam_ssh_agent_auth.*|auth sufficient pam_ssh_agent_auth.so authorized_keys_command=$AK_COMMAND_BIN authorized_keys_command_user=root|' $PAM_SUDO"
            rlRun "cat $PAM_SUDO"
        rlPhaseEnd

        for KEY in "${KEYS[@]}"; do
            rlPhaseStartTest "Test authorized_keys_command with key type $KEY (bz1299555, bz1317858)"
                rlRun "cat ~$USER/.ssh/my_id_${KEY}.pub >$AK_COMMAND_KEYS"
                rlRun -s "su $USER -c 'eval \`ssh-agent\`; ssh-add ~/.ssh/my_id_$KEY; sudo id; ssh-agent -k'"
                rlAssertGrep "uid=0(root) gid=0(root)" $rlRun_LOG
            rlPhaseEnd
        done

        rlPhaseStartCleanup "Cleanup for authorized_keys_command"
            rlFileRestore --namespace ak_command
            rlRun "rm -f $AK_COMMAND_BIN $AK_COMMAND_KEYS"
        rlPhaseEnd
    fi

    if rlIsRHEL '>=7.3'; then # not in Fedora anymore
        rlPhaseStartTest "bz1312304 - Exposing information about succesful auth"
            rlRun "rlFileBackup --namespace exposing $PAM_SSHD"
            rlRun "rlFileBackup --namespace exposing $SSHD_CFG"
            rlRun "rlFileBackup --namespace exposing /root/.ssh/"
            rlRun "rm -f ~/.ssh/id_rsa*"
            rlRun "ssh-keygen -f ~/.ssh/id_rsa -N \"\"" 0
            rlRun "ssh-keyscan localhost >~/.ssh/known_hosts" 0
            USER_AK_FILE=~$USER/.ssh/authorized_keys
            rlRun "cat ~/.ssh/id_rsa.pub >$USER_AK_FILE"
            rlRun "chown $USER:$USER $USER_AK_FILE"
            rlRun "chmod 0600 $USER_AK_FILE"
            rlRun "gcc -fPIC -DPIC -shared -rdynamic -o $PAM_MODULE.o $PAM_MODULE.c"
            rlRun "test -d /lib64/security && cp $PAM_MODULE.o /lib64/security/" 0,1
            rlRun "test -d /lib/security && cp $PAM_MODULE.o /lib/security/" 0,1
            rlRun "sed -i '1 i auth       optional         $PAM_MODULE.o' $PAM_SSHD"

            # pam-and-env should expose information to both PAM and environmental variable;
            # we will be testing only env variable here for the time being,
            rlRun "echo 'ExposeAuthenticationMethods pam-and-env' >>$SSHD_CFG"
            rlRun "sed -i '/^ChallengeResponseAuthentication/ d' $SSHD_CFG"
            rlRun "service sshd restart"
            rlWaitForSocket 22 -t 5
            rlRun -s "ssh -i ~/.ssh/id_rsa $USER@localhost \"env|grep SSH_USER_AUTH\"" 0 \
                "Environment variable SSH_USER_AUTH is set"
            rlAssertGrep "^SSH_USER_AUTH=publickey:" $rlRun_LOG
            rlRun "rm -f $rlRun_LOG"

            # pam-only should expose information only to PAM and not to environment variable
            rlRun "sed -i 's/pam-and-env/pam-only/' $SSHD_CFG"
            rlRun "echo 'AuthenticationMethods publickey,keyboard-interactive:pam' >>$SSHD_CFG"
            rlRun "service sshd restart"
            rlWaitForSocket 22 -t 5
ssh_with_pass() {
    ssh_args=("-i /root/.ssh/id_rsa")
    ssh_args+=("$USER@localhost")
    cat >ssh.exp <<_EOF
#!/usr/bin/expect -f

set timeout 5
spawn ssh ${ssh_args[*]} "echo CONNECTED; env|grep SSH_USER_AUTH"
expect {
    -re {.*[Pp]assword.*} { send -- "$PASS\r"; exp_continue }
    timeout { exit 1 }
    eof { exit 0 }
}
_EOF
    rlRun -s "expect -f ssh.exp"
}
            #rlRun -s "ssh ${ssh_args[*]} \"echo CONNECTED; env|grep SSH_USER_AUTH\"" 1 \
                #"Environment variable SSH_USER_AUTH is NOT set"
            rlRun "ssh_with_pass"
            rlRun "grep -q CONNECTED $rlRun_LOG" 0 "Connection was successful"
            rlAssertGrep "^SSH_USER_AUTH: 'publickey:" /tmp/SSH_USER_AUTH
            rlRun "cat /tmp/SSH_USER_AUTH"
            rlRun "rm -f $rlRun_LOG /tmp/SSH_USER_AUTH"
            for pm in /lib64/security/$PAM_MODULE.o /lib/security/$PAM_MODULE.o; do
                rlRun "test -e $pm && rm -f $pm" 0,1
            done
            rlRun "rlFileRestore --namespace exposing"
        rlPhaseEnd
    fi

    rlPhaseStartCleanup
        rlRun "popd"
        rlRun "rm -r $TmpDir" 0 "Removing tmp directory"
        rlRun "userdel -fr $USER"
        rlFileRestore
        rlServiceRestore sshd
    rlPhaseEnd
rlJournalPrintText
rlJournalEnd