Linux批量部署工具Expect

前言*随着IT企业的迅猛发展，Linux运维逐渐趋向于自动化，所谓的自动化运维，常常表现为：从程序打包-代码管理-上传-更新部署-代码回滚等等一系列工作。实现自动化运维后将减去人工的重复工作、避免人工的误操作等等。

目前主流的自动化工具有puppet、Expect、pssh等等，今天我们来研究一下expect如何来实现自动部署和日常管理维护。

一、Expect简介

expect是一种能够按照脚本内容里面设定的方式与交互式程序进行“会话”的程序。根据脚本内容，Expect可以知道程序会提示或反馈什么内容以及什么是正确的应答。它是一种可以提供“分支和嵌套结构”来引导程序流程的解释型脚本语言。

我们熟知的shell编程功能虽然很强大,但是不能实现有交互功能的多机器之前的操作,例如ssh和scp等。而expect可以帮助我们来实现。

二、Expect安装

yum install expect -y

三、Expect使用

使用Expect批量管理和部署服务器大致分为两个步骤，使用for循环读取服务器IP、密码列表并取值，远程执行命令。如下需求，在两台服务器上执行mkdir /tmp/`date +%Y%m%d`命令，如何实现？

首先定义expect 登陆脚本：

1、login.exp，内容如下（详细的参数就不解释了）：

#!/usr/bin/expect -f
set ip [lindex $argv 0 ]
set passwd [lindex $argv 1 ]
set command [lindex $argv 2]
set timeout 1
spawn ssh root@$ip
expect {
"yes/no" { send "yes\r";exp_continue }
"password:" { send "$passwd\r" }
}
expect "*#*" { send "$command\r" }
expect eof

2、创建批量执行脚本auto_exec.sh

#!/bin/sh
#auto exec command
#wugk 20130712
CMD="$*"
for i in `awk '{print $1}' passwd.txt`
do
    j=`awk -v I="$i" '{if(I==$1)print $2}' passwd.txt`
    expect /data/sh/login.exp  $i  $j  "$CMD"
done

3、建立批量IP、密码文件

cat  passwd.txt内容如下：(第一列为IP，第二列为密码)

192.168.1.100  abc_123
192.168.1.101  abc_456

四、测试脚本

直接执行：

/bin/sh auto_exec.sh "mkdir -p /tmp/`date +%Y%m%d`"

然后登陆两台服务器查看是否在/tmp/下生产一个以当前系统日期为名称的目录。

五、SCP远程拷贝

如果需要远程推送文件，重新建立文件login.scp相关参数和auto_exec.sh变量：

1、login.scp内容如下：

#!/usr/bin/expect -f
set ip [lindex $argv 0 ]
set passwd [lindex $argv 1 ]
set src_file [lindex $argv 2]
set des_dir [lindex $argv 3]
set timeout 1
spawn scp -r $src_file root@$ip:$des_dir
expect {
"yes/no" { send "yes\r";exp_continue }
"password:" { send "$passwd\r" }
}
expect "#*"
expect eof

2、auto_exec_scp.sh脚本内容如下

#!/bin/sh
#auto exec command
#wugk 20130712
read -p "Please Enter insert Source File or DIR: " src_file
echo ======================================================
sleep 1
read -p "Please Enter insert Destination DIR: " des_dir
for i in `awk '{print $1}' passwd.txt`
do
    j=`awk -v I="$i" '{if(I==$1)print $2}' passwd.txt`
    expect login.scp  $i  $j $src_file $des_dir
done

密码文件保持不变即可。

看上面的脚本可能有点晕，直接上下面的脚本，拷贝即可使用，一个是用于远程执行执行命令的，一个是用于远程拷贝文件的。

六、一键安装expect、ssh批量执行auto_exec_ssh.sh脚本：（远程执行命令）

#!/bin/sh
#auto exec expect shell scripts
#wugk 20130712
if
      [ ! -e /usr/bin/expect ];then
      yum  install expect -y
fi
#Judge passwd.txt exist
if
      [ ! -e ./passwd.txt ];then
      echo -e "The passwd.txt is not exist......Please touch ./passwd.txt ,Content Example:\n192.168.1.11 passwd1\n192.168.1.12 passwd2"
      sleep 2 &&exit 0
fi
#Auto Tuoch login.exp File
cat>login.exp <<EOF
#!/usr/bin/expect -f
set ip [lindex \$argv 0 ]
set passwd [lindex \$argv 1 ]
set command [lindex \$argv 2]
set timeout -1
spawn ssh root@\$ip
expect {
"yes/no" { send "yes\r";exp_continue }
"password:" { send "\$passwd\r" }
}
expect "*#*" { send "\$command\r" }
expect  "#*"  { send "exit\r" }
expect eof
EOF
##Auto exec shell scripts
CMD="$*"
if
    [ "$1" == "" ];then
    echo ========================================================
    echo "Please insert your command ,Example {/bin/sh  $0 'mkdir -p /tmp'} ,waiting exit ........... "
    sleep 2
    exit 1
fi
for i in `awk '{print $1}' passwd.txt`
do
    j=`awk -v I="$i" '{if(I==$1)print $2}' passwd.txt`
    expect ./login.exp  $i  $j  "$CMD"
done

七、一键安装expect、scp批量拷贝auto_exec_scp.sh脚本：（批量SCP拷贝）

#!/bin/sh
#auto exec expect shell scripts
#wugk 20130712
if
      [ ! -e /usr/bin/expect ];then
      yum  install expect -y
fi
#Judge passwd.txt exist
if
      [ ! -e ./passwd.txt ];then
      echo -e "The passwd.txt is not exist......Please touch ./passwd.txt ,Content Example:\n192.168.1.11 passwd1\n192.168.1.12 passwd2"
      sleep 2 &&exit 0
fi
#Auto Tuoch login.exp File
cat>login.exp <<EOF
#!/usr/bin/expect -f
set ip [lindex \$argv 0]
set passwd [lindex \$argv 1]
set src_file [lindex \$argv 2]
set des_dir [lindex \$argv 3]
set timeout -1
spawn scp -r \$src_file root@\$ip:\$des_dir
expect {
"yes/no"     { send "yes\r"; exp_continue }
"password:"  { send "\$passwd\r" }
}
expect "100%"
expect eof
EOF
##Auto exec shell scripts
if
    [ "$1" == "" ];then
    echo ========================================================
    echo "Please insert your are command ,Example {/bin/sh  $0 /src  /des } ,waiting exit ........... "
    sleep 2
    exit 1
fi
for i in `awk '{print $1}'  passwd.txt`
do
      j=`awk -v I="$i" '{if(I==$1)print $2}' passwd.txt`
      expect ./login.exp  $i  $j $1 $2
done

脚本已在真实环境使用，附上两个脚本执行过程图：