每次进步一点点——linux expect 使用

Myth丶恋晨 2022-07-15 08:22 177阅读 0赞

# 1. 介绍 #

expect是建立在tcl（参见：[Tcl/Tk快速入门][Tcl_Tk] ）基础上的一个工具，它可以让一些需要交互的任务自动化地完成。相当于模拟了用户和命令行的交互操作。  
**一个具体的场景**：远程登陆服务器，并执行命令  
登录时输入密码需要交互，bash脚本无法完成，可以使用expect来完成。

# 2. 安装 #

下面介绍两种安装方式

## yum 安装 ##

yum install -y expect

## 源码编译安装 ##

expect 依赖于 tcl, 所以需要首先安装 tcl。可以使用rpm检查是否已经安装tcl:

rpm -qa | grep tcl

如果已安装，则会打印出tcl软件信息

安装过程参考：linux expect 安装 [http://www.cnblogs.com/daojian/archive/2012/10/10/2718390.html][http_www.cnblogs.com_daojian_archive_2012_10_10_2718390.html]  
tcl 地址：[https://sourceforge.net/projects/tcl/files/Tcl/][https_sourceforge.net_projects_tcl_files_Tcl] 选择一个版本  
expect 地址：[https://sourceforge.net/projects/expect/files/Expect/][https_sourceforge.net_projects_expect_files_Expect] 选择一个版本  
**注意**：wget 下载的时候需要加上 –no-check-certificate, 不检查网站证书

# 3. 原理与工作机制 #

首先使用 spawn 开启一个会话，然后使用 expect-send 对来执行交互式操作。  
spawn 后面跟上一个命令操作，表示开启一个会话。expect 等待输出特定的字符串(通常是提示符)，然后使用send 发送交互字符串。比如：

spawn ssh username@host # 远程登录
    
    expect "*assword" # 提示为："username@host's password:", 等待用户输入密码
    send "${password}\r" # 这时使用send模拟用户输入密码的字符串，完成登录验证

# 4. 基本语法介绍 #

## 脚本解释器 ##

脚本中首先引入文件，表明使用的是哪一个shell

#!/usr/bin/expect

## set ##

设置会话超时时间为30s, 若不限制超时时间则应设置为-1

set timeout 30

set 还可以设置变量

# 使用变量语句：$param 或者 ${param}({}用来避免param和后面的字符串拼接起来导致错误)
    set param "param_str"
    set param 1

## spawn ##

spawn 后面跟一个命令，开启一个会话

spawn ${cmd} # for example : spawn su root

## expect - send ##

expect 接收命令执行后的输出，然后和期望字符串匹配，若对应这执行相应的send来发送交互信息。

expect "$case1" {send "$respond1\r"} # 这一行等同于下面两行
    
    expect "$case1"
    send "$response1\r"

expect 可以有多个分支，就像switch语句一样。

expect 
    {
        "$case1" {send "$response1\r"}
        "$case2" {send "$response2\r"}
        "$case3" {send "$response3\r"}
    }

## 结束符 ##

expect eof ：等待执行结束，若没有这一句，可能导致命令还没执行，脚本就结束了  
interact ： 执行完成后保持交互状态, 这时可以手动输入信息  
**注**：expect eof 与 interact 二选一即可

## 接收参数 ##

参数存在argv中，使用第一个参数如下：

set param0 [lindex $argv 0]

$argc表示参数个数,判断语句如下:

if {$argc < 1} {
        #do something
        send_user "usage: $argv0 <param1> <param2> ... "
        exit
    }

注：$argv0 是脚本名，但\[lindex $argv 0\]是第一个参数 param1, \[lindex $argv 1\]是第二个参数 param2, 以此类推  
send\_user 用来显示信息到父进程(一般为用户的shell)的标准输出。

# 5. 实例 #

实现远程登录服务器，并切换到root用户下执行关闭防火墙的命令，然后退出

#!/usr/bin/expect
    
    if {$argc < 4} {
        #do something
        send_user "usage: $argv0 <remote_user> <remote_host> <remote_pwd> <remote_root_pwd>"
        exit
    }
    
    set timeout -1
    set remote_user [lindex $argv 0] # 远程服务器用户名
    set remote_host [lindex $argv 1] # 远程服务器域名
    set remote_pwd [lindex $argv 2] # 远程服务器密码
    set remote_root_pwd [lindex $argv 3] # 远程服务器根用户密码
    
    # 远程登录
    spawn ssh ${remote_user}@${remote_host}
    expect "*assword:" {send "${remote_pwd}\r"}
    expect "Last login:"
    
    # 切换到 root
    send "su\r"
    expect "*assword:" {send "${remote_root_pwd}\r"}
    
    # 执行关闭防火墙命令
    send "service iptables stop\r"
    send "exit\r"
    send "exit\r"
    expect eof

将代码保存到 remot\_root\_command.exp 中，权限改为755，然后执行下面这条命令即可：

./remote_root_command.exp <remote_user> <remote_host> <remote_pwd> <remote_root_pwd>

[Tcl_Tk]: http://blog.csdn.net/chinalinuxzend/article/details/2430429
[http_www.cnblogs.com_daojian_archive_2012_10_10_2718390.html]: http://www.cnblogs.com/daojian/archive/2012/10/10/2718390.html
[https_sourceforge.net_projects_tcl_files_Tcl]: https://sourceforge.net/projects/tcl/files/Tcl/
[https_sourceforge.net_projects_expect_files_Expect]: https://sourceforge.net/projects/expect/files/Expect/