Shell 编程 免交互 expect

本篇主要写一些shell脚本免交互expect的使用。

---

概述

Expect是建立在tcl基础上的一个工具，Expect 是用来进行自动化控制和测试的工具。主要解决shell脚本中不可交互的问题。

安装

• 使用此工具前需先安装

yum install -y expect

基本命令

send

• 向进程发送字符串，用于模拟用户的输入

• 该命令不能自动回车换行，一般要加\r (回车)

expect

• expect的一个内部命令，判断上次输出结果里是否包含指定的字符串，如果有则立即返回，否则就等待超时时间后返回。

• 只能捕捉由spawn启动的进程的输出

spawn

• 启动进程，并跟踪后续交互信息

interact

• 执行完成后保持交互状态，把控制权交给控制台

timeout

• 指定超时时间，过期则继续执行后续指令

• 单位是:秒

• timeout -1永不超时

• 默认情况下，timeout是10秒

exp_continue

• 允许expect继续向下执行指令

send_user

• 回显命令，相当于echo

$argv 参数数组

• expect脚本可以接受从bash传递的参数.可以使用[lindex $argv n]获得，n从0开始，分别表示第一个，第二个，第三个...参数

expect 脚本

• expect脚本必须以interact或expect eof结束，执行自动化任务通常expect eof就够了

• expect eof是在等待结束标志。由spawn启动的命令在结束时会产生一个eof标记，expect eof即在等待这个标记

expect 语法

• 单分支

expect "password:" {send "mypassword\r";}

• 多分支

expect "aaa" {send"AAA\r"}
expect "aaa" {send"AAA\r"}
expect "aaa" {send"AAA\r"}

send命令不具备回车换行功能，一般要加\r或\n

expect {
    "aaa" {send "AAA\r"}
    "bbb" {send "BBB\r"}
    "ccc" {send "CCC\r"}
}

只要配置aaa或bbb或ccc中的任何一个，执行相应的send语句后退出该expect语句

expect {
    "aaa" {send "AAA";exp_continue}
    "bbb" {send "BBB";exp_continue}
    "ccc" {send "CCC"}
}

exp_continue表示继续后面的匹配，如果匹配了aaa，执行完send语句后还要继续向下匹配bbb

执行方式

• 基本语法结构

spawn 命令
expect "提示信息"
send "代替人工输入的字符串\r"

• 直接执行

#!/usr/bin/expect
# 超时时间
set timeout 20
log_file test.log
log_user 1
# 参数传入
set hostname [lindex $argv 0]
set password [lindex $argv 1]
# 追踪命令
spawn ssh root@$hostname
# 捕捉信息并匹配，免交互执行
expect {
    "(yes/no)" {send "yes\r";exp_continue}
    "*password" {send "$password\r"}
}
# 控制权交给控制台执行
interact

[root@host01 ~]# yum install expect -y
[root@host01 ~]# vim ssh.sh
[root@host01 ~]# chmod +x ssh.sh
[root@host01 ~]# ./ssh.sh 192.168.28.129 000000
spawn ssh root@192.168.28.129
The authenticity of host '192.168.28.129 (192.168.28.129)' can't be established.
ECDSA key fingerprint is SHA256:QmZtJT0piBUSkF9P3GfYf3uEogzBWs08sI7j0eBE/cI.
ECDSA key fingerprint is MD5:ef:e6:06:22:8a:0f:24:00:f8:af:a5:59:5b:a2:b8:b1.
Are you sure you want to continue connecting (yes/no)? yes
Warning: Permanently added '192.168.28.129' (ECDSA) to the list of known hosts.
root@192.168.28.129's password:
Last login: Thu Oct 17 09:35:35 2019
[root@host02 ~]#

• 嵌入执行

#!/bin/bash
hostname=$1
password=$2
/usr/bin/expect <<-EOF
spawn ssh root@$hostname
expect {
    "(yes/no)" {send "yes\r";exp_continue}
    "*password" {send "$password\r"}
}
expect "*]#"
send "exit\r"
expect eof
EOF

-EOF只能容错制表符tab

[root@host01 ~]# vim ssh.sh
[root@host01 ~]# ./ssh.sh 192.168.28.129 000000
spawn ssh root@192.168.28.129
root@192.168.28.129's password:
Last login: Thu Oct 17 09:38:23 2019 from 192.168.28.128
[root@host02 ~]# exit
logout
Connection to 192.168.28.129 closed.
[root@host01 ~]#

案例1 useradd

#!/bin/bash
username=$1
password=$2
useradd $username
/usr/bin/expect << EOF
spawn passwd $username
expect "New password:"
send "$password\r"
expect "Retype new password:"
send "$password\r"
expect eof
EOF

[root@host01 ~]# vim useradd.sh
[root@host01 ~]# chmod +x useradd.sh
[root@host01 ~]# ./useradd.sh zhangsan 000000
spawn passwd zhangsan
Changing password for user zhangsan.
New password:
BAD PASSWORD: The password is a palindrome
Retype new password:
passwd: all authentication tokens updated successfully.

案例2 ssh

#!/usr/bin/expect
# 超时时间
set timeout 20
log_file test.log
log_user 1
# 参数传入
set hostname [lindex $argv 0]
set password [lindex $argv 1]
# 追踪命令
spawn ssh root@$hostname
# 捕捉信息并匹配，免交互执行
expect {
    "Connection refused" exit
    "Name or service not known" exit
    "(yes/no)" {send "yes\r";exp_continue}
    "*password" {send "$password\r"}
}
# 控制权交给控制台执行
interact
exit

[root@host02 ~]# systemctl stop sshd

[root@host01 ~]# ./ssh.sh 192.168.28.129 000000
spawn ssh root@192.168.28.129
ssh: connect to host 192.168.28.129 port 22: Connection refused
[root@host01 ~]#

[root@host02 ~]# systemctl start sshd

[root@host01 ~]# ./ssh.sh host02 000000
spawn ssh root@host02
ssh: Could not resolve hostname host02: Name or service not known
[root@host01 ~]#

[root@host01 ~]# ./ssh.sh 192.168.28.129 000000
spawn ssh root@192.168.28.129
root@192.168.28.129's password:
Last login: Thu Oct 17 09:49:38 2019
[root@host02 ~]#