mirror of https://github.com/einverne/dotfiles.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
182 lines
4.4 KiB
182 lines
4.4 KiB
#!/bin/bash |
|
# Feature : 使用 except 自动模拟输入密码，登录服务器，适用于跳板机情况自动登录ssh |
|
# 对于自己的服务器可以通过配置 ssh key 来直接免密码登录，但是对于一些 |
|
# 跳板机，则每次都需要手动输入比较麻烦，那么可以直接使用该脚本一键登录 |
|
# Notice : 注意系统需要安装 expect 程序 |
|
# Original Author : Jiangxianli |
|
# Modified : Ein Verne |
|
# Date : 2019/06/10 |
|
# Github : https://github.com/jiangxianli/SSHAutoLogin |
|
# Update : Xiongwilee 2018/06/18 https://github.com/xiongwilee/SSHAutoLogin |
|
# Update : |
|
|
|
#默认服务器配置项 |
|
# "别名 服务器名称 端口号 IP地址 登录用户名 登录密码/秘钥文件Key 秘钥文件地址" |
|
CONFIGS=( |
|
"tencent tencent 22 118.xxxxxxxxxx ubuntu 1234" |
|
"ds ds 22 118.xxxxxxxxxx ubuntu 1234" |
|
) |
|
|
|
#读取自定义服务器配置文件（`~/.sshloginrc`）列表，合并服务器配置列表 |
|
sshloginrc_path=~/.sshloginrc |
|
if [ -f ${sshloginrc_path} ]; then |
|
CONFIGS=() |
|
while read line |
|
do |
|
CONFIGS+=("$line") |
|
done < ${sshloginrc_path} |
|
fi |
|
|
|
#服务器配置数 |
|
CONFIG_LENGTH=${#CONFIGS[*]} #配置站点个数 |
|
|
|
if [[ $CONFIG_LENGTH -le 0 ]] ; |
|
then |
|
echo "未检测到服务器配置项!" |
|
echo "请在脚本CONFIGS变量中配置或单独创建一个 ~/.sshloginrc 文件并配置" |
|
exit ; |
|
fi |
|
|
|
## |
|
# 绿色输出 |
|
## |
|
function GreenEcho() { |
|
echo -e "\033[32m ${1} \033[0m"; |
|
} |
|
|
|
## |
|
# 服务器配置菜单 |
|
## |
|
function ConfigList(){ |
|
echo "- 序号 IP 别名" |
|
for ((i=0;i<${CONFIG_LENGTH};i++)); |
|
do |
|
CONFIG=(${CONFIGS[$i]}) #将一维sites字符串赋值到数组 |
|
serverNum=$(($i+1)) |
|
echo "- [${serverNum}] ${CONFIG[3]} ${CONFIG[0]}" |
|
done |
|
} |
|
|
|
## |
|
# 登录菜单 |
|
## |
|
function LoginMenu(){ |
|
if [ ! -n $1 ]; then |
|
AutoLogin $1 |
|
else |
|
echo "-------请输入登录的服务器序号或别名---------" |
|
ConfigList |
|
echo "请输入您选择登录的服务器序号或别名: " |
|
fi |
|
} |
|
|
|
## |
|
# 选择登录的服务器 |
|
## |
|
function ChooseServer(){ |
|
read serverNum; |
|
|
|
# 是否重新选择 |
|
needChooseServer=1; |
|
|
|
if [ -z $serverNum ]; then |
|
echo "请输入序号或者别名" |
|
reChooseServer $needChooseServer; |
|
fi |
|
|
|
AutoLogin $serverNum $needChooseServer; |
|
} |
|
|
|
## |
|
# 是退出还是重新选择Server |
|
# @param $1 是否重新选择server 1: 重新选择server |
|
## |
|
function reChooseServer(){ |
|
if [ "$1"x = "1"x ]; then |
|
ChooseServer; |
|
else |
|
exit; |
|
fi |
|
} |
|
|
|
## |
|
# 自动登录 |
|
# @param $1 序号或者别名 |
|
# @param $2 是否重新选择server 1: 重新选择server |
|
## |
|
function AutoLogin(){ |
|
num=$(GetServer $1) |
|
|
|
if [ -z $num ]; then |
|
echo "您输入的别名【$1】不存在，请重试" |
|
reChooseServer $2; |
|
fi |
|
|
|
CONFIG=(${CONFIGS[$num]}) |
|
|
|
if [ -z $CONFIG ]; then |
|
echo "您输入的序号【$1】不存在，请重试" |
|
reChooseServer $2; |
|
else |
|
echo "正在登录【${CONFIG[1]}】" |
|
fi |
|
|
|
export PASSWORD=${CONFIG[5]}; |
|
|
|
command=" |
|
expect { |
|
\"*assword\" {set timeout 6000; send \$env(PASSWORD)\r; exp_continue ; sleep 3; } |
|
\"*passphrase\" {set timeout 6000; send \$env(PASSWORD)\n\r; exp_continue ; sleep 3; } |
|
\"yes/no\" {send \"yes\n\"; exp_continue;} |
|
\"Last*\" { send_user \"\n成功登录【${CONFIG[1]}】\n\";} |
|
} |
|
interact |
|
"; |
|
pem=${CONFIG[6]} |
|
if [ -n "$pem" ] ;then |
|
expect -c " |
|
spawn ssh -p ${CONFIG[2]} -i ${CONFIG[6]} ${CONFIG[4]}@${CONFIG[3]} |
|
${command} |
|
" |
|
else |
|
expect -c " |
|
spawn ssh -p ${CONFIG[2]} ${CONFIG[4]}@${CONFIG[3]} |
|
${command} |
|
" |
|
fi |
|
GreenEcho "您已退出【${CONFIG[1]}】" |
|
exit; |
|
|
|
} |
|
|
|
## |
|
# 通过输入定位选择那个服务器配置 |
|
## |
|
function GetServer(){ |
|
# 判断输入是否为数字 |
|
if [ "$1" -gt 0 ] 2>/dev/null ;then |
|
echo $(($1-1)) |
|
else |
|
for key in ${!CONFIGS[*]} ; do |
|
item=(${CONFIGS[$key]}) |
|
if [ ${item[0]} == $1 ]; then |
|
echo $key |
|
break; |
|
fi |
|
done |
|
fi |
|
} |
|
|
|
## |
|
# 程序入口 |
|
## |
|
if [ 1 == $# ]; then |
|
if [ 'list' == $1 ]; then |
|
ConfigList |
|
else |
|
AutoLogin $1 |
|
fi |
|
else |
|
LoginMenu |
|
ChooseServer |
|
fi
|
|
|