mirror of https://github.com/einverne/dotfiles.git
4 changed files with 201 additions and 23 deletions
@ -0,0 +1,182 @@
|
||||
#!/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 |
||||
Loading…
Reference in new issue