Shell编程之Here Document 免交互及 expect自动化交互

1 Here Document 免交互

• 使用I/O重定同的方式将命令列表提供给交互式程序或命令，比如ftp、cat或 read 命令。
• 是标准输入的一种替代品可以帮助脚本开发人员不必使用临时文件来构建输入信息，而是直接就地生产出一个"文件"并用作"命令"的标准输入。Here Document也可以与非交互式程序和命令一起使用。

语法格式：
命令  <<标记
...
内容         #标记之间是传入内容
...
标记

注意事项：

• 标记可以使用任意合法字符（通常为EOF）
• 结尾的标记一定要顶格写，前面不能有任何字符
• 结尾的标记后面也不能有任何字符（包括空格）
• 开头标记前后的空格会被省

2 Here Document 常规用法

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

#输入值是两个 EOE 标记之间的部分，作为变量 i 的值
[root@localhost awk]# read i <<EOF
> helloworld
> EOF
[root@localhost awk]# echo $i

[root@localhost awk]# read a <<EOF  #read只能读取一行内容，一行内容遇到换行符或者回车符是结束
> hello
> world
> EOF
[root@localhost awk]# echo $a

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

#将要统计的内容置于标记“EOF”之间，直接将内容传给 wc -l 来统计
[root@localhost awk]# wc -l <<abc   #开头和结尾的标记要保持一致
> line1
> line2
> line3
> abc

2.3 免交互方式实现对文本内容查看

cat >file.txt <<EOF    #只能导入到文件当中
LINE1
LINE2
....     #输入内容
EOF

tee file.txt <<EoF   #tee可以同时实现标准输出和重定向输出
LINE1
LINE2
....
EOF

例如

[root@localhost ~]# cat <<EOF #把文本内容查看出来
> today is good day
> I want to go to school
> hello world!
> EOF

[root@localhost ~]# cat <<EOF > hello.txt
重定向可以放在前面也可以放在后面
> today is good day
> I want to go to school
> hello world!
> EOF
[root@localhost ~]# cat hello.txt

2.4 Here Document 变量设定

[root@localhost ~]# vim demo1.sh

#!bin/bash
var="Great! I am going to school!"
#EOF若加上''表示不接受变量的替换
#若想忽略每行文本前面的tab键，要在EOF前面加上-
myvar=$(cat <<EOF
today is good day!
hello world!
$var
EOF
)
#加上上引号才可以换行，不然就会在同一行
echo "$myvar"

2.5 Here Document 多行注释

[root@localhost demo]# vim demo1.sh

#!bin/bash
var="Great! I am going to school!"
#用 : 为开头的Here Document，里面所标记的内容不会被执行
: <<'EOF'
echo 'today is good day!
hello world!
$var'
EOF

echo "myvar"

3 Expect

建立在 tcl 语言基础上的一个工具，常被用于进行自动化控制和测试，解决 shell 脚本中交互相关的问题

rpm -q expect
rpm -q tcl 
yum install -y expect  #安装expect

3.1 Expect 基本命令

• 脚本解释器
  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 "123456\r"}
"密码3" {send "123123\r"}
}

• 结束符
  （1）expect eof：表示交互结束，等待执行结束，退回到原用户，与 spawn 对应

例如：切换到 root 用户，expect 脚本默认的是等待10s，当执行完命令后，默认停留 10s 后，自动切回了原用户

（2）interact：执行完成后保持交互状态，把控制权交给控制台，会停留在目标终端而不会退回到原终端，这个时候就可以手工操作了，interact 后的命令不起作用，比如 interact 后添加 exit，并不会退出 root 用户。而如果没有 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 并再次执行判断；如果匹配 *password 则输出 abc123 并结束该段 expect 语句
expect {
"(yes/no)" {send "yes\r";exp_continue;}
"*password" {set timeout 300;send "abc123\r";}

注意：使用exp_continue时，如果跟踪像 passwd 这样的输入密码后就结束进程的命令，expect{}外不要再加上expect eof 因为spawn进程结束后会默认向expect发送eof，会导致后面的expect eof 执行报错

• send_user
  send_user 表示回显命令，相当于echo
• 接收参数
  expect 脚本可以接受从bash命令行传递的参数，使用 [lindex $argv n] 获得。其中 n 从0开始，分别表示第一个，第二个，第三个.…参数

例：
set hostname [lindex Sargv 0]    相当于hostname=$1
set password [lindex $argv 1]   相当于password=$2

4 Expect 案例

expect执行方式：串行执行和并行执行

案例一：免交互 su 切换用户

[root@localhost ~]# vim demo1.sh

#!/usr/bin/expect   #先定义解释器

#设置expect等待的超时时间，默认是10s 
set timeout5

#先用spawn开启对一个有交互过程命令进程的跟踪
spawn passwd zhangsan

#用expect捕捉系统的交互提示信息
expect "新的密码"

#当expect匹配成功则会模拟用户发送字符串给交互进程，字符串结尾记得加\r或者\n
send "abc1234\r"    #第一次匹配
expect "重新输入新的密码" {send "abc1234\r"}   #第二次匹配
expect eof    #输入结束符

expect 直接执行，需要使用expect 命令去执行脚本
[root@localhost opt]#vim expect.sh 
 
#!/usr/bin/expect
 
#设置超时时间
set timeout 5

#参数传入
set hostname [lindex $argv 0]
#hostname=$1
set password [lindex $argv 1]
#password=$2
 
#开始追踪命令
spawn su $hostname
#免交互执行，用export匹配系统提示信息，再用send模拟用户发送内容
expect "密码：" {send "123123\n"}
expect "]#"
send_user "hello"

#把控制权交给控制台，免交互结束后把控制权交给用户，并不会切回原用户
interact
#免交互结束后在超时等待时间过后退回原用户
#expect eof

案例二：免交互远程 ssh 脚本

[root@localhost opt]#vim ssh.sh 
 
#!/usr/bin/expect
#开启ssh命令，ssh后面跟要远程控制的IP地址
spawn ssh 192.168.16.16
expect {
        "yes/no" { send "yes\n";exp_continue }
        "password" { send "123123\n"}
 
}
interact

案例三：免交互 scp 脚本

[root@localhost opt]#vim scp.sh 
 
#!/usr/bin/expect
spawn scp /etc/passwd 192.168.16.16:/data
expect {
   "yes/no" { send "yes\n";exp_continue }
   "password: " { send "123123\n" }
}
expect eof