У меня просто была такая же проблема в Linux с использованием Bash. Сначала я использовал переменную окружения SSH_CONNECTION, но потом понял, что она не установлена, если вы su -.
Решение lastlog выше не сработало ни после, suни su -.
Наконец, я использую who am i, который показывает удаленный IP (или имя хоста) в конце, если это соединение SSH. Это также работает после su.
Используя регулярные выражения Bash, это работает:
if [[ $(who am i) =~ \([-a-zA-Z0-9\.]+\)$ ]] ; then echo SSH; else echo no; fi
Если zsh не поддерживает регулярные выражения, то же самое может быть достигнуто разными способами с помощью grep, cut, sed или чего-либо еще.
Для любопытных, вот что я использую в root'е .bashrc:
# We don't allow root login over ssh.
# To enable root X forwarding if we are logged in over SSH,
# use the .Xauthority file of the user who did su
w=$(who am i)
if [[ $w =~ \([-a-zA-Z0-9\.]+\)$ ]] ; then
olduser=${w/ .*/}
oldhome=$(getent passwd $olduser | cut -d: -f 6)
[ -f "$oldhome/.Xauthority" ] \
&& export XAUTHORITY=$oldhome/.Xauthority
fi
Альтернативой, которая также работает, suбудет рекурсивный поиск sshdпо родительским процессам:
#!/bin/bash
function is_ssh() {
p=${1:-$PPID}
read pid name x ppid y < <( cat /proc/$p/stat )
# or: read pid name ppid < <(ps -o pid= -o comm= -o ppid= -p $p)
[[ "$name" =~ sshd ]] && { echo "Is SSH : $pid $name"; return 0; }
[ "$ppid" -le 1 ] && { echo "Adam is $pid $name"; return 1; }
is_ssh $ppid
}
is_ssh $PPID
exit $?
Если функция добавлена в .bashrc, ее можно использовать как if is_ssh; then ...