Shell 脚本攻略—Here Document 免交互与 Expect 自动化交互

前言

在实际生产环境中，撰写shell脚本难免会遇到一些会交互的界面，比如做raid和扩容等等这些都是需要交互的，那我们就需要用到以下免交互的操作，从而实现真正的shell脚本自动化。

一、Here Document免交互

1.概述

• 使用 I/O 重定向的方式将命令列表提供给交互式程序或命令，比如 ftp、cat或read 命令

• 是标准输入的一种替代品，可以帮助脚本开发人员不必使用临时文件来构建输入信息，而是直接就地生成出一个“文件”并用作“命令”的标准输入

• Here Document 也可以与非交互式程序和命令一起使用

2.语法格式

命令 <<标记
...
内容    #标记直接是传入内容
...
标记

3.注意事项

• 标记可以使用任意合法字符 （通常使用EOF）

• 结尾的标记一定要顶格写，前后都不能有任何字符（包括空格）

• 开头的标记前后的空格会被省略

4.示例

①免交互方式实现对行数的统计

[root@localhost ~]# wc -l <<EOF
> xjj1
> xjj2
> EOF
2

②通过read命令接收输入并打印

• 在写入文件时会先将变量替换成实际值
• 然后再结合 cat 命令完成写入

[root@localhost ~]# vim xjj1.sh

#!/bin/bash

file="xjj3.txt"
i="money"
cat > $file <<EOF
I'd like some $i
EOF

[root@localhost ~]# . xjj1.sh          #先执行脚本
[root@localhost ~]# ls
 xjj1.sh    xjj3.txt   
[root@localhost ~]# cat xjj3.txt
I'd like some money

③整体赋值给变量，然后通过echo命令打印出变量值

[root@localhost ~]# vim xhh1.sh

#!/bin/bash

name="OMG! xjj come!"
myname=$(cat <<EOF
It's a beautiful day
today is sunday
school starts again tomorrow
$name
EOF
)
echo $myname

[root@localhost ~]# . xhh1.sh 
It's a beautiful day today is sunday school starts again tomorrow OMG! xjj come!

④关闭变量替换功能

• 这样会按照字符原本的样子输出，且没有任何修改或替换

[root@localhost ~]# . xhh1.sh 
It's a beautiful day today is sunday school starts again tomorrow $name

⑤创建yum源

[root@localhost ~]# cat > test.repo <<EOF
> [test]
> name=test
> baseurl=file:///mnt
> enabled=1
> gpgcheck=0
> EOF
------
写法2：cat <<EOF > local.repo
写法3： tee test1.txt <<EOF

⑥多行注释

• Bash 的默认注释是“#”，该注释方法只支持单行注释
• Here Document 的引入解决了多行注释的问题
• ":"代表什么都不做的空命令
• 中间标记区域的内容不会被执行，会被bash忽略掉，因此可达到批量注释的效果

[root@localhost ~]# . xhh1.sh 
OMG! xjj come! 

二、Expect自动化交互

1.概述

• expect 是一个免费的编程工具语言，常用来实现自动和交互式任务进行通信，而无需人的干预
• expect 需要 Tcl 编程语言的支持，要在系统上运行 expect 必须首先安装 Tcl

rpm -q expect
rpm -q tcl
yum -y install tcl expect 			#yum或者光盘安装都可以

2.基础命令

①脚本解释器

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

#!/usr/bin/expect

②spawn

• spawn 后面通常跟一个Linux执行命令,表示开启一个会话、启动进程,并跟踪后续交互信息

例: spawn passwd root          ##跟踪启动更改用户密码的进程

③expect

• 判断上次输入结果中是否包含指定的字符串，如果有则立即返回，否则就等待超时时间后返回
• 只能捕捉由spawn启动的进程的输出
• 用于接收命令执行后的输出，然后和期望的字符串匹配

④send

• 向进程发送字符串,用于模拟用户的输入
• 该命令不能自动回车换行,一般要加 \r (回车)或者 \n
• 方式一

expect “密码” {send “abc123\r”}			#同一行send部分要有{}

• 方式二

expect “密码”
send “$abc123\r”						#换行send部分不需要有{}

• 方式三

expect “支持多个分支

expect  {        	#只要匹配了其中一个情况，就执行相应的send语句后退出该expect语句
"密码1 {send "abc123\r"}"
"密码2 {send "123123\r"}"
"密码3 {send "123123\r"}"
}

⑤结束符

expect eof

• 表示交互结束，等待执行结束，退回到原用户，与spawn对应
• 比如:切换到root用户，expect脚本默认的是等待10s，当执行完命令后，默认停留10s后，就会自动切回原用户

interact

• 执行完成后保持交互状态，把控制权交给控制台，会停留在目标终端，这个时候就可以手工操作了，interact后的命令不起作用，比如interact会保持在终端而不会退回到原终端，比如切换到root用户，会一直在root用户状态下
• 比如ssh到另一服务器，会一直在目标服务器终端，而不会切回的原服务器
• 注：expect eof 与 interact 只能二选一

⑥set

• expect 默认的超时时间是10秒，通过set命令可以设置会话超时时间，若不限制超时时间则应设置为-1

#例：
set timeout 30

⑦exp_continue

• exp_continue 附加于某个 expect 判断项之后，可以使该项匹配后，还能继续匹配该 expect 判断语句内的其他项
• exp_continue类似于控制语句中continue语句,表示允许expect继续向下执行指令
• 下例将判断交互输出中是否存在 yes/no 或*password , 如果匹配 yes/no 则输出 yes 并再次执行判断;如果匹配 *assword 则输出 123123 并结束该段 expect 语句

expect
"(yes/no)" {send "yes\r"; exp_ continue; }
"*password" {set timeout 300; send "123123\r";}

⑧send_user

• send_ user 表示回显命令，相当于 echo

⑨接收参数

• expect 脚本可以接受从 bash 命令行传递的参数，使用 [lindex $argv n] 获得，其中 n 从 0 开始，分别表示第一个,第二个，第三个…
• 示例：

set hostname [lindex $argv 0]
##相当于hostname=$1

set password [lindex $argv 1]
##相当于password=$2

3.expect 直接执行

• ssh五交互登录到远程服务器

[root@localhost shell]# vim expect.sh 

[root@localhost /home]#vim expect.sh
	
#!/usr/bin/expect                     	 #需要用expect自己的解释器，不要写成bash否则无法识别
spawn ssh root@192.168.126.11       	 #开启一个程序，这个程序是ssh远程登录
expect {                             	 #捕获内容，当出现password的时候，就会向程序发送密码
         "password:"
        { send "123456\r"; }
}
interact                           				 #交互，否则会直接退出远程服务器

[root@localhost /home]#chmod +x expect.sh              #需要加执行权限
[root@localhost /home]#./expect.sh  
spawn ssh root@192.168.126.11
root@192.168.126.11's password: 
Last login: Wed Sep 15 22:39:40 2021 from 192.168.126.15

• 在对方服务器上进行一下操作后再退出可执行以下脚本

[root@localhost /home]#vim expect.sh

#!/usr/bin/expect
spawn ssh root@192.168.126.11
expect {
         "password:"
        { send "123456\r"; }
}
expect "#"                      #当捕获到#的时候
send "ls\r"                     #执行ls命令
send "ifconfig ens33\r"         #执行ifconfig ens33命令
send "exit\r"                   #执行完exit退出登录
expect eof                      #不用进行交互，意味着结束expect程序
 
[root@localhost /home]#chmod +x expect.sh 
[root@localhost /home]#./expect.sh 
spawn ssh root@192.168.8.129
root@192.168.126.11's password: 
Last login: Wed Sep 15 22:55:06 2021 from 192.168.126.15
[root@localhost ~]#ls
[root@localhost ~]#ifconfig ens33
ens33: flags=4163<UP,BROADCAST,RUNNING,MULTICAST>  mtu 1500
        inet 192.168.126.11  netmask 255.255.255.0  broadcast 192.168.126.255
        inet6 fe80::2161:befa:6ffd:c44b  prefixlen 64  scopeid 0x20<link>
        ether 00:0c:29:01:e6:0a  txqueuelen 1000  (Ethernet)
    ......
[root@localhost ~]#exit
登出
Connection to 192.168.126.11 closed.

• 引用位置变量

[root@localhost /home]#vim expect1.sh 
#!/usr/bin/expect
set user root
set ip [lindex $argv 0]                #设置第一个位置变量为ip
set pass [lindex $argv 1]              #设置第二个位置变量为登陆密码
spawn ssh $user@$ip
expect {
        "password:"
        { send "$pass\r"; }
}
expect "#"
send "ls\r"
send "exit\r"
expect eof

[root@localhost /home]#chmod +x expect1.sh 
[root@localhost /home]#./expect1.sh 192.168.126.11 123456
spawn ssh root@192.168.126.11
root@192.168.126.11's password: 
Last login: Wed Sep 15 22:58:16 2021 from 192.168.126.15
[root@localhost ~]#ls
[root@localhost ~]#exit
登出
Connection to 192.168.126.11 closed.

• 创建用户并设置用户密码

[root@localhost /home]#vim user.sh

#!/bin/bash
username=$1
useradd $username
/usr/bin/expect <<-EOF
spawn passwd $username
expect {                                  #获取的内容和发送的内容不能在同一行否则执行不成功
        "新的 密码"
        { send "123456\r";exp_continue }
        "重新输入新的 密码"
        { send "123456\r"; }
}
EOF

[root@localhost /home]#chmod +x user.sh 
[root@localhost /home]#./user.sh xjj
spawn passwd xjj
更改用户 xjj 的密码 。
新的 密码：
无效的密码： 密码少于 8 个字符
重新输入新的 密码：
passwd：所有的身份验证令牌已经成功更新。

总结

借助 expect 处理交互的命令，可以将交互过程如 ssh 登录，ftp 登录等写在一个脚本上，使之自动化完成，适用于需要对多台服务器执行相同操作的环境中，可以大大提高系统管理人员的工作效率，使用它，你甚至可以仅仅执行一个脚本来完成分布式应用系统的搭建与维护。