Expect详解
expect命令简单说就是通过其内置的各种命令实现在交互式软件中自动交互的工具。
用在ssh时,可以配合spawn命令实现ssh的自动登录。又因为可以在脚本中编写判断,赋值等逻辑,具有很高的灵活性。
问题
如何从机器A上ssh到机器B上,然后执行机器B上的命令?如何使之自动化完成?
四个命令
Expect中最关键的四个命令是send,expect,spawn,interact。
send:用于向进程发送字符串
expect:从进程接收字符串
spawn:启动新的进程
interact:允许用户交互
1. send命令
send命令接收一个字符串参数,并将该参数发送到进程。
expect1.1> send "hello world\n"
hello world
2. expect命令
(1)基础知识
expect命令和send命令正好相反,expect通常是用来等待一个进程的反馈。expect可以接收一个字符串参数,也可以接收正则表达式参数。和上文的send命令结合,现在我们可以看一个最简单的交互式的例子:
expect "hi\n"
send "hello there!\n"
这两行代码的意思是:从标准输入中等到hi和换行键后,向标准输出输出hello there。
tips: $expect_out(buffer)存储了所有对expect的输入,<$expect_out(0,string)>存储了匹配到expect参数的输入。
比如如下程序:
expect "hi\n"
send "you typed <$expect_out(buffer)>"
send "but I only expected <$expect_out(0,string)>"
当在标准输入中输入
test
hi
是,运行结果如下
you typed: test
hi
I only expect: hi
(2)模式-动作
expect最常用的语法是来自tcl语言的模式-动作。这种语法极其灵活,下面我们就各种语法分别说明。
单一分支模式语法:
expect "hi" {send "You said hi"}
匹配到hi后,会输出"you said hi"
多分支模式语法:
expect "hi" { send "You said hi\n" } \
"hello" { send "Hello yourself\n" } \
"bye" { send "That was unexpected\n" }
匹配到hi,hello,bye任意一个字符串时,执行相应的输出。等同于如下写法:
expect {
"hi" { send "You said hi\n"}
"hello" { send "Hello yourself\n"}
"bye" { send "That was unexpected\n"}
}
3. spawn命令
上文的所有demo都是和标准输入输出进行交互,但是我们跟希望他可以和某一个进程进行交互。spawm命令就是用来启动新的进程的。spawn后的send和expect命令都是和spawn打开的进程进行交互的。结合上文的send和expect命令我们可以看一下更复杂的程序段了。
set timeout -1
spawn ftp ftp.test.com //打开新的进程,该进程用户连接远程ftp服务器
expect "Name" //进程返回Name时
send "user\r" //向进程输入anonymous\r
expect "Password:" //进程返回Password:时
send "123456\r" //向进程输入don@libes.com\r
expect "ftp> " //进程返回ftp>时
send "binary\r" //向进程输入binary\r
expect "ftp> " //进程返回ftp>时
send "get test.tar.gz\r" //向进程输入get test.tar.gz\r
这段代码的作用是登录到ftp服务器ftp ftp.uu.net上,并以二进制的方式下载服务器上的文件test.tar.gz。程序中有详细的注释。
4.interact
到现在为止,我们已经可以结合spawn、expect、send自动化的完成很多任务了。但是,如何让人在适当的时候干预这个过程了。比如下载完ftp文件时,仍然可以停留在ftp命令行状态,以便手动的执行后续命令。interact可以达到这些目的。下面的demo在自动登录ftp后,允许用户交互。
spawn ftp ftp.test.com
expect "Name"
send "user\r"
expect "Password:"
send "123456\r"
interact
解决方法
上文中提到:
如何从机器A上ssh到机器B上,然后执行机器B上的命令?如何使之自动化完成?
下面一段脚本实现了从机器A登录到机器B,然后执行机器B上的pwd命令,并停留在B机器上,等待用户交互。具体含义请参考上文。
#!/home/tools/bin/64/expect -f
set timeout -1
spawn ssh $BUser@$BHost
expect "*password:" { send "$password\r" }
expect "$*" { send "pwd\r" }
interact
实例
前言
前段时间将工作用的机器换成了 Ubuntu + 128G SSD + 8G内存的台式机,终于走上了用linux办公的道路,总的说来,Ubuntu用起来还是比我想象的顺畅得多,其“apt-get式”的包管理方式,安装软件非常方便,apt-get install xxx 即可安装。唯一的缺点就是银行的“U盾”等不支持linux,所以,使用网银时略有不便。
言归正传,在换台式后公司还做了一件事就是统一了公司内的测试环境,将所有的测试环境机器统一到机房内用虚拟机的方式统一管理(皆大欢喜)。为了以后开发测试方便,项目组申请了多套测试环境,比如:开发环境:用于开发人员每次迭代自测时的环境。测试环境:稳定的和线上代码同步的测试环境,通常用于演示或者对测试环境其他应用提供服务。除此以外,还有一套性能测试环境。这么多环境的问题就导致,每次需要更新环境:update 代码,重启tomcat时都需要ssh到某台机器上。这其实是一个浪费时间的重复性劳动,通常情况,这种事情都可以使用代码解决。这就是我们今天的主角:expect。
demo实例如下:
#!/usr/bin/expect -f ##声明使用哪种解析器解析该脚本,如果没有安装expect的话,使用apt-get install expect安装
if { [llength $argv] < 1 } { ##$argv为执行该脚本是的参数数组,判断长度,以决定是否继续
puts "Usage: $argv0 need ssh ip" //expect中的echo,System.out.println()
exit 1
}
set envs [lindex $argv 0] ##将数组中的第一个参数赋值给 envs
set timeout 30 //设置等待终端响应的超时时间为30秒
if { $envs == 61 } { ##注意 if和{间有个空格,{和$envs间有个空格,等号前后有空格,61后有空格,}{间有空格!!!
set ips xxx.xx.xxx.61
}
if { $envs == 141 } {
set ips xxx.xx.xxx.141
}
if { $envs == 136 } {
set ips xxx.xx.xxx.136
}
spawn ssh root@$ips ##执行ssh命令实现登陆
expect {
##第一次登陆的时候,会询问是否有在本地保存该密钥,仔细的你应该发现,这里的匹配是用正则的,简答说,只匹配connecting也是可以的。
"Are you sure you want to continue connecting (yes/no)?" {send "yes\r"}
"password:" {send "cdyanfa\r"}##保存过一次后,就会直接返回需要密码的阶段了。使用send 命令发送密码。
}
interact ##是Expect用来打开用户与产生进程之间通信的命令,简单说就是登陆以后将远程服务器的终端保持在当前终端,而不是将远程终端关掉。
短短几行命令就将每天重复N次的操作简化成一行了,对于提高效率很有用。