1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

#!/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 "\033[40m\033[31m Please input login server, you can choose one follows list \033[0m"

    for i in "${_config_keys[@]}";do

        echo -e "\033[41;37m $i \033[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 "\033[40m\033[31m

        Not config server info ...

        Please config in _server_config like

        Host User Passwd Port Jump Command\033[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 \"yes\r\"; 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 \"yes\r\"; exp_continue }

        \"*assword:\" { send \"$_passwd\r\" }

    }

    ${_command}

    interact

    "

fi