使用expect编写脚本-白红宇

expect简介

expect是一个自动化交互式应用程序的工具，所以expect可以用来处理交互的命令。借助expect，我们可以将交互过程写在一个脚本上，使之自动化完成。形象的说，ssh登录，ftp登录等都符合交互的定义。expect含有利用正则表达式进行模式匹配以及通用的编程功能，允许简单的脚本智能地管理如下工具：telnet,ftp和ssh（这些工具都缺少编程的功能），宏以及其它程序。expect脚本的出现使得这些老的软件工具有了新的功能和更多的灵活性。

expect脚本远程登录

以上简介中也提到了expect和shell类似，可以进行编程，接下来就实际的编写一些expect脚本来学习expect简单的使用方式。

如果你的系统没有安装expect，需要先安装expect，安装命令如下：

yum install -y expect

expect示例：编写一个自动远程登录脚本，expect编写的脚本文件后缀名为expect，这是为了好区分：

[root@localhost ~]# mkdir expectFiles[root@localhost ~]# cd expectFiles/[root@localhost ~/expectFiles]# vim telnet_ept.expect#! /usr/bin/expect# 需要登录的IPset host "192.168.77.128"# 密码set passwd "123456"# 登录的命令spawn ssh root@$host# expect用于执行一些命令expect {# 初次登录会询问"yes/no"，所以如果截取的是"yes/no"就 send 发送一个yes，\r是回车的意思，exp_continue表示继续"yes/no" { send "yes\r"; exp_continue}# 如果截取的是"password:"就 send 发送密码"password:" { send "$passwd\r" }}# 结束标识，但是不会退出登录，不加interact的话就会马上退出登录interact

set是用来定义一个变量的，变量名后面跟的就是变量的值，不需要使用等于号来赋值，调用变量的方式和shell一样，也是使用$符号。

与interact相对的就是expect eof ，expect eof会在登录后，停留登录状态1-2秒后才退出登录。

执行这个脚本需要加一个执行权限，然后直接执行就可以了：

[root@localhost ~/expectFiles]# chmod a+x telnet_ept.expect [root@localhost ~/expectFiles]# ./telnet_ept.expect spawn ssh root@192.168.77.128The authenticity of host '192.168.77.128 (192.168.77.128)' can't be established.ECDSA key fingerprint is 4d:5a:9d:31:65:75:30:47:a3:9c:f5:56:63:c4:0f:6a.Are you sure you want to continue connecting (yes/no)? yesWarning: Permanently added '192.168.77.128' (ECDSA) to the list of known hosts.root@192.168.77.128's password: Last login: Wed Nov 29 14:07:45 2017 from 192.168.77.1[root@localhost ~]#

运行结果如上，成功登录了192.168.77.128。

expect脚本远程执行命令

除了能够实现远程登录外，还能实现远程执行命令，例如可以写一个自动远程登录后，执行命令并退出的脚本：

[root@localhost ~/expectFiles]# vim telnet_2.expect#!/usr/bin/expectset user "root"set passwd "123456"spawn ssh $user@192.168.77.128expect {"yes/no" { send "yes\r"; exp_continue}"password:" { send "$passwd\r" }}# 当遇到 "]#" 就执行sebd命令，我们登录后不就是：[root@localhost ~]#，末尾就是 ]# ，而send是发送touch创建文件的命令expect "]#"send "touch /tmp/12.txt\r"# 同样的当遇到 "]#" 就执行sebd里的命令expect "]#"send "echo 1212 > /tmp/12.txt\r"# 最后退出系统expect "]#"send "exit\r"

运行脚本：

[root@localhost ~/expectFiles]# chmod a+x telnet_2.expect [root@localhost ~/expectFiles]# ./telnet_2.expectspawn ssh root@192.168.77.128root@192.168.77.128's password: Last login: Wed Nov 29 14:54:34 2017 from 192.168.77.130[root@localhost ~]# touch /tmp/12.txt[root@localhost ~]# echo 1212 > /tmp/12.txt[root@localhost ~]# [root@localhost ~/expectFiles]#

然后再来看看192.168.77.128机器上有没有/tmp/12.txt 文件：

[root@localhost ~]# ll /tmp/12.txt -rw-r--r-- 1 root root 5 11月 29 15:10 /tmp/12.txt[root@localhost ~]# cat !$cat /tmp/12.txt1212[root@localhost ~]#

可以看到文件和内容与脚本中执行的命令一致。

expect脚本传递参数

expect脚本也是可以像shell脚本一样传递参数的，以下例子演示如何拿到脚本的参数：

[root@localhost ~/expectFiles]# vim telnet_3.expect#!/usr/bin/expect# expect脚本拿参数的方式有点像是在数组中拿元素一样，也是从0开始数，$argv是expect的一个内置变量set user [lindex $argv 0] # 第一个参数set host [lindex $argv 1]  # 第二个参数set passwd "123456"# 需要远程执行的命令set cm [lindex $argv 2]  # 第三个参数spawn ssh $user@$hostexpect {"yes/no" { send "yes\r"}"password:" { send "$passwd\r" }}# 执行命令然后退出登录expect "]#"send "$cm\r"expect "]#"send "exit\r"

以上脚本运行结果：

[root@localhost ~/expectFiles]# ./telnet_3.expect root 192.168.77.128 lsspawn ssh root@192.168.77.128root@192.168.77.128's password: Last login: Wed Nov 29 15:10:48 2017 from 192.168.77.130[root@localhost ~]# ls\                                           stop.sh     zabbix-release-3.2-1.el7.noarch.rpmanaconda-ks.cfg                             Test.classmysql57-community-release-el7-8.noarch.rpm  Test.java[root@localhost ~]# [root@localhost ~/expectFiles]#

执行多条命令，使用引号将命令括起来即可：

[root@localhost ~/expectFiles]# ./telnet_3.expect root 192.168.77.128 "ls;w;who"spawn ssh root@192.168.77.128root@192.168.77.128's password: Last login: Wed Nov 29 15:29:30 2017 from 192.168.77.130[root@localhost ~]# ls;w;who\                                           stop.sh     zabbix-release-3.2-1.el7.noarch.rpmanaconda-ks.cfg                             Test.classmysql57-community-release-el7-8.noarch.rpm  Test.java 15:29:48 up  5:13,  2 users,  load average: 0.00, 0.01, 0.05USER     TTY      FROM             LOGIN@   IDLE   JCPU   PCPU WHATroot     pts/0    192.168.77.1     14:07   15:00   0.06s  0.06s -bashroot     pts/1    192.168.77.130   15:29    0.00s  0.02s  0.00s wroot     pts/0        2017-11-29 14:07 (192.168.77.1)root     pts/1        2017-11-29 15:29 (192.168.77.130)[root@localhost ~]# [root@localhost ~/expectFiles]#

关于这种远程执行命令的操作，会存在一个超时时间，也就是说，如果你需要远程执行的那条命令的执行时间，超过了默认的超时时间，命令的执行就会被终止。这个默认的超时时间是10秒，我们可以通过执行一条vmstat命令来测试出这个默认的超时时间：

[root@localhost ~/expectFiles]# ./telnet_3.expect root 192.168.77.128 "vmstat 1"spawn ssh root@192.168.77.128root@192.168.77.128's password: Last login: Wed Nov 29 17:25:48 2017 from 192.168.77.130[root@localhost ~]# vmstat 1procs -----------memory---------- ---swap-- -----io---- -system-- ------cpu----- r  b   swpd   free   buff  cache   si   so    bi    bo   in   cs us sy id wa st 1  0      0 3271556   1376 259272    0    0     4     1   35   64  0  0 100  0  0 0  0      0 3271712   1376 259364    0    0     0     0   92  157  0  0 100  0  0 0  0      0 3271712   1376 259364    0    0     0     0   76  128  0  0 100  0  0 0  0      0 3271744   1376 259364    0    0     0     0   91  146  0  1 100  0  0 0  0      0 3271744   1376 259364    0    0     0     0   64  120  0  0 100  0  0 0  0      0 3271712   1376 259364    0    0     0     0   78  127  0  0 100  0  0 0  0      0 3271712   1376 259364    0    0     0     0   57  104  0  0 100  0  0 0  0      0 3271712   1376 259364    0    0     0     0   81  128  0  1 100  0  0 0  0      0 3271712   1376 259364    0    0     0     0   75  127  0  0 100  0  0 0  0      0 3271200   1376 259400    0    0     0     0   91  153  0  1 100  0  0 0  0      0 3270936   1376 259400    0    0     0     0   92  152  0  0 100  0  0[root@localhost ~/expectFiles]#

以上运行结果可以看到vmstat设置的间隔时间为1秒，只打印了11行数据就退出登录了，第一行打印不算在超时时间内，从第二行开始数，刚好是10行，每行打印的间隔时间是1秒，所以就可以知道默认的超时时间是10秒。

那么要如何修改这个默认的超时时间呢？只需要在脚本中加上一句set timeout即可，设置的时间单位为秒，示例：

#!/usr/bin/expectset user [lindex $argv 0]set host [lindex $argv 1]set passwd "Zero-One1."set cm [lindex $argv 2]spawn ssh $user@$host# 设置超时时间为3秒set timeout 3expect {"yes/no" { send "yes\r"}"password:" { send "$passwd\r" }}expect "]#"send "$cm\r"expect "]#"send "exit\r"

运行结果：

[root@localhost ~/expectFiles]# ./telnet_3.expect root 192.168.77.128 "vmstat 1"spawn ssh root@192.168.77.128root@192.168.77.128's password: Last login: Wed Nov 29 17:25:54 2017 from 192.168.77.130[root@localhost ~]# vmstat 1procs -----------memory---------- ---swap-- -----io---- -system-- ------cpu----- r  b   swpd   free   buff  cache   si   so    bi    bo   in   cs us sy id wa st 1  0      0 3271172   1376 259420    0    0     4     1   35   64  0  0 100  0  0 0  0      0 3270968   1376 259508    0    0     0     0  104  174  0  0 100  0  0 0  0      0 3270936   1376 259560    0    0     0     0   90  152  0  1 100  0  0 0  0      0 3270780   1376 259596    0    0     0     0   83  137  0  0 100  0  0[root@localhost ~/expectFiles]#

提示：如果你将set timeout的值设置为-1的话，就表示没有超时时间，命令会一直执行。

expect脚本同步文件

expect结合rsync 可以实现自动同步文件，代码示例：

[root@localhost ~/expectFiles]# vim synFile.expect#!/usr/bin/expectset passwd "123456"spawn rsync -av root@192.168.77.128:/tmp/12.txt /tmp/expect {"yes/no" { send "yes\r"}"password:" { send "$passwd\r" }}expect eof

以上脚本运行结果：

[root@localhost ~/expectFiles]# chmod a+x synFile.expect[root@localhost ~/expectFiles]# ./synFile.expect spawn rsync -av root@192.168.77.128:/tmp/12.txt /tmp/root@192.168.77.128's password: receiving incremental file list12.txtsent 30 bytes  received 84 bytes  228.00 bytes/sectotal size is 5  speedup is 0.04[root@localhost ~/expectFiles]# cat /tmp/12.txt 1212[root@localhost ~/expectFiles]#

如果把脚本中的expect eof命令给注释掉，看看会发生什么：

[root@localhost ~/expectFiles]# ./synFile.expect spawn rsync -av root@192.168.77.128:/tmp/12.txt /tmp/root@192.168.77.128's password: [root@localhost ~/expectFiles]#

从结果可以看到，还没来得及执行rsync命令就退出来了，所以在这种登录后需要执行命令的expect脚本中一定要加上expect eof或者interact，不然你需要远程执行的命令就会来不及被执行，这就是加不加expect eof或者interact的区别。

可能有些人会想到使用set timeout来设置一个超时时间避免马上退出登录，实际上是无效的，避免这种马上退出登录的方式只能用expect eof或者interact。

expect脚本指定host和要同步的文件

想要指定host和要同步的文件就把它们作为参数就好了：

#!/usr/bin/expectset passwd "123456"set host [lindex $argv 0]set file [lindex $argv 1]spawn rsync -av $file root@$host:$fileexpect {"yes/no" { send "yes\r"}"password:" { send "$passwd\r" }}expect eof

运行结果：

[root@localhost ~/expectFiles]# chmod a+x hostFile.expect [root@localhost ~/expectFiles]# ./hostFile.expect 192.168.77.128 "/tmp/12.txt"spawn rsync -av /tmp/12.txt root@192.168.77.128:/tmp/12.txtroot@192.168.77.128's password: sending incremental file listsent 31 bytes  received 12 bytes  86.00 bytes/sectotal size is 5  speedup is 0.12[root@localhost ~/expectFiles]#

文件的路径要写绝对路径，然后需要使用双引号引起来。

本文转自 ZeroOne01 51CTO博客，原文链接：http://blog.51cto.com/zero01/2045745，如需转载请自行联系原作者