expect實現Linux自動登陸遠端機器指令碼範例

#!/usr/local/bin/bash
# @Version 0.3.1
# @filename to
# 修復等不需要要設定跳板機的時候執行命令，在設定跳板機位置預設填 no 即可
# @Author pemakoa@gmail.com
# Bash version >= 4.0 使用關聯陣列

# Usage: host user passwd port jump_host command 
# 四種情況如下：
# 1. 直接登入目標機器 如 A 
# 2. 需要中間機器登陸到目標機器 如 C， 其中 B 為中間機器，會先登入 B在從 B登陸到 C然後執行 command
# 3. 直接登入目標機器並執行相應的命令 如 D

declare -A _server_config

_server_config['A']="a_host a_user a_passwd a_port"
_server_config['B']="b_host b_user b_passwd b_port"
_server_config['C']="c_host c_user c_passwd c_port B '(command eg) ls .'"
_server_config['D']="d_host d_user d_passwd d_port no 'cd /home && ll'"

_config_keys=(${!_server_config[@]})
_length=${#_server_config[@]}
_login_server=$1
_config_status=false

# 是否輸入登陸機器
if [ "$_login_server" == "" ];then
    echo -e "33[40m33[31m Please input login server, you can choose one follows list 33[0m"
    for i in "${_config_keys[@]}";do
        echo -e "33[41;37m $i 33[0m "
    done
    exit
fi

# 檢查登陸的機器是否設定
for i in "${_config_keys[@]}";do
    if [ "$_login_server" == "$i" ];then
        _config_status=true
    fi
done

if [ "${_config_status}" == "false" ];then
    echo -ne "33[40m33[31m
        Not config server info ...
        Please config in _server_config like
        Host User Passwd Port Jump Command33[0m"
    exit
fi

# 登陸 如果設定跳板機，先登陸跳板機在登陸到目標機器
_host=$(echo ${_server_config["${_login_server}"]} | awk '{print $1}')
_user=$(echo ${_server_config["${_login_server}"]} | awk '{print $2}')
_passwd=$(echo ${_server_config["${_login_server}"]} | awk '{print $3}')
_port=$(echo ${_server_config["${_login_server}"]} | awk '{print $4}')
_jump=$(echo ${_server_config["${_login_server}"]} | awk '{print $5}')
_command=$(echo ${_server_config["${_login_server}"]} | awk -F"'" '{print $2}')

if [ "${_command}" != "" ]; then
    _command="expect "*]*"
    send "${_command}r""
fi

if [ "${_jump}" != "" ] && [ "${_jump}" != "no" ]; then
    _jump_host=$(echo ${_server_config["${_jump}"]} | awk '{print $1}')
    _jump_user=$(echo ${_server_config["${_jump}"]} | awk '{print $2}')
    _jump_passwd=$(echo ${_server_config["${_jump}"]} | awk '{print $3}')
    _jump_port=$(echo ${_server_config["${_jump}"]} | awk '{print $4}')

    expect -c "
    set timeout 30
    spawn ssh -p${_jump_port} ${_jump_user}@${_jump_host}
    expect {
        "yes/no" { send "yesr"; exp_continue }
        "assword" { send "${_jump_passwd}r" }
    }

    expect "*]*" 
    send "ssh -p${_port} ${_user}@${_host}r"
    expect "assword:" 
    send "${_passwd}r"
    ${_command}
    interact"
else
    expect -c "
    set timeout 30
    spawn ssh -p${_port} ${_user}@${_host}
    expect {
        "yes/no" {send "yesr"; exp_continue }
        "*assword:" { send "$_passwdr" }
    }
    ${_command}
    interact
    "
fi