基于 shell 使用 expect 实现 SSH 自动登录远程服务器

末蓝、 2022-01-15 08:17 373阅读 0赞

[为什么80%的码农都做不了架构师？>>> ][80_] ![hot3.png][]

### 相关知识点 ###

*  expect
    
    > expect 是一个免费的编程工具，用来实现自动的交互式任务，而无需人为干预。
    > 
    > 说白了，expect 就是一套用来实现自动交互功能的软件。
    > 
    > 在实际工作中，我们运行命令、脚本或程序时，这些命令、脚本或程序都需要从终端输入某些继续运行的指令，而这些输入都需要人为的手工进行。而利用expect，则可以根据程序的提示，模拟标准输入提供给程序，从而实现自动化交互执行。这就是expect！！！
 *  IFS
    
    > IFS是存储定界符的环境变量，是Shell环境中的默认定界符字符串，默认值为空白字符（换行符、制表符、空格）

### 效果预览 ###

![14fe2e5251ec63f0db372ed6dc438e63145.jpg][]

### 主要文件 ###

#### goto.sh ####

#!/bin/bash
    ################################################
    # 基于 shell 脚本，实现 ssh 自动登录操作
    # Usage:
    # 		1. 修改 goto.conf ，追加服务器列表
    # 		2. $ chmod u+x goto.sh goto.ex
    # 		3. $ ./goto.sh
    #
    # 		或使用如下方式将 goto 加入当前用户全局使用：
    # 		$ echo "alias goto=\"$PWD/goto.sh\"" >> ~/.zshrc
    # 		$ source ~/.zshrc
    # 		$ goto
    # Author: whoru.S.Q <whoru@sqiang.net>
    # Version: 1.5
    ################################################
    
    # 服务器列表文件
    BASE_PATH=$(cd "$(dirname "$0")";pwd)
    FILE_SERVER_LIST=$BASE_PATH"/goto.conf"
    
    # 暂存服务器列表，用于登录操作
    CONFIG_ARR=()
    
    # 记录默认分隔符，用于修改后的还原
    IFS_OLD=$IFS
    
    # 初始化
    function menu {
    
    	# 检查配置文件
    	if [ ! -f $FILE_SERVER_LIST ]; then
    		echo "Config file not found."
    		exit 1
    	fi
    
    	# 读取配置文件，显示待操作服务器列表
    	clear
    	echo "-------------------------------------"
    	local serverNum=1 # 服务器列表序号
    	local config=()
    	while read line || [ -n "$line" ]
    	do
    		if [[ ${line} != \#* && "$line" != "" ]] ; then
    			IFS=, # 定义读取配置文件时的分隔符
    			config=($line)
    			CONFIG_ARR[$serverNum]=$line
    			# serverName=$(echo $line | awk  -F::: '{print $1}')
    			# serverIp=$(echo $line | awk  -F::: '{print $3}')
                spacenum=`expr 16 - ${#config[2]}`
                spaces=$(seq -s ' ' $spacenum | sed 's/[0-9]//g')
    			echo -e " ${config[2]}$spaces- \033[32m$serverNum\033[0m.${config[0]}"
    			serverNum=$(($serverNum+1))
    		fi
    	done < $FILE_SERVER_LIST
    	IFS=$IFS_OLD # 还原分隔符
    	echo "-------------------------------------"
    	echo -en "请输入\033[32m序号\033[0m选择要登录的服务器: "
    	handleChoice ;
    }
    
    # 处理用户输入
    function handleChoice {
    	read -n 1 choice
    	local serverListLength=${#CONFIG_ARR[@]}
    	if [[ "$choice" -lt 1 || "$choice" -gt serverListLength ]]; then
    		echo -en "\n\033[31m无效的序号[ $choice ], 是否重新输入( y 是 | n 否 ):\033[0m"
    		read -n 1 retry
    		if [[ -n "$retry" && "$retry" = "y" ]]; then
    			clear
    			menu ;
    		else
    			echo ""
    			exit 1
    		fi
    	else
    		sshLogin $choice;
    	fi
    }
    
    # 执行 ssh 登录
    function sshLogin {
    
    	IFS=, # 定义读取分隔符
    	local config=(${CONFIG_ARR[$1]})
    
    	# 默认用户 root
    	local user=${config[1]}
    	if [[ $user == "" ]]; then
    		user="root"
    	fi
    
    	# 默认端口号 22
    	local port=${config[3]}
    	if [[ $port == "" ]]; then
    		port="22"
    	fi
    
    	# 开始登录
    	echo -e "\n\n\033[32m==>\033[0m 正在登录【\033[32m${config[0]}\033[0m】，请稍等...\n"
    	sleep 1
    	$(which expect) $BASE_PATH/goto.ex ${config[0]} ${config[2]} $port $user ${config[4]}
    	echo -e "\n\033[32m==>\033[0m 您已退出【\033[32m${config[0]}\033[0m】\n"
    }
    
    # 执行初始化
    menu ;

#### goto.ex ####

#!/usr/bin/expect
    
    set SERVER_NAME [lindex $argv 0]
    set IP [lindex $argv 1]
    set PORT [lindex $argv 2]
    set USER_NAME [lindex $argv 3]
    set PASSWORD [lindex $argv 4]
    
    
    spawn ssh -p $PORT $USER_NAME@$IP
    
    expect {
        -timeout 300
        "*assword" { send "$PASSWORD\r\n"; exp_continue ; sleep 3; }
        "yes/no" { send \"yes\n\"; exp_continue; }
        "Last*" {
            puts "\n登录成功\n";
            send "PROMPT_COMMAND='echo -ne \"\\033]0;$SERVER_NAME \\007\"' \r";
            send "clear\r";
        }
        timeout { puts "Expect was timeout."; return }
    }
    interact

#### goto.conf ####

# 格式：hostname,user,ip,port,pwd,commands
    # 参数说明：
    #   hostname 列表中显示的服务器名
    #   user 登录用户，默认 root
    #   ip 服务器 IP 地址
    #   port 登录端口号，默认 22
    #   pwd 登录密码
    #   commands 登录成功后，默认执行的命令，比如切换到指定目录
    # 备注: 以 # 开头的行是注释，不会被读取
    #
    
    测试服务器,,192.168.1.56,,123456,cd /alidata/www && ls -l

--------------------

更新地址：[https://github.com/whorusq/ssh-autologin][https_github.com_whorusq_ssh-autologin]

转载于:https://my.oschina.net/antsky/blog/905336

[80_]: https://my.oschina.net/u/2663968/blog/3051541
[hot3.png]: /images/20220115/8d033b0f0e1840c792df3e09b1be7536.png
[14fe2e5251ec63f0db372ed6dc438e63145.jpg]: /images/20220115/1404f615a90248ba96ef5983fc00111e.png
[https_github.com_whorusq_ssh-autologin]: https://github.com/whorusq/ssh-autologin