Linux：shell之编程免交互
- 一、Here Document 免交互
- 1.1 Here Document 免交互概述
- 1.2 语法格式
- 1.3 操作
- 二、Expect 命令
- 2.1 Expect 概述
- 2.2 基本命令
- 2.3 操作
一、Here Document 免交互
1.1 Here Document 免交互概述
- 使用I/O重定向的方式将命令列表提供给交互式程序或命令，比如 ftp、cat 或 read 命令。
- 是标准输入的一种替代品可以帮助脚本开发人员不必使用临时文件来构建输入信息，而是直接就地生产出一个“文件”并用作“命令”的标准输入。Here Document 也可以与非交互式程序和命令一起使用。
1.2 语法格式
命令 <<标记
...
标记
- 注意事项：
1、标记可以使用任意合法字符（通常为EOF）
2、结尾的标记一定要顶格写，前面不能有任何字符
3、结尾的标记后面也不能有任何字符（包括空格）
4、开头标记前后的空格会被省略掉
1.3 操作
- 免交互方式实现对行数的统计，将要统计的内容置于标记“EOF”之间，直接将内容传给 wc -l 来统计
wc -l <<EOF
>Line1
>Line2
>EOF
- 通过 read 命令接收输入并打印，输入值是两个 EOF 标记之间的部分，作为变量 i 的值
read i <<EOF
>Hi
>EOF
echo $i
- 通过 passwd 给用户设置密码
passwd lisi <<EOF
>abc1234 #这两行是输入的密码和确认密码
>abc1234
>EOF
- 支持变量替换
在写入文件时会先将变量替换成实际值，再结合 cat 命令完成写入
#!/bin/bash
file="EOF1.txt"
i="school"
cat > $file <<EOF #给前一个EOF加上单引号，那么他中间的内容则不会引用变量。
I am going to $i #这一行的输出将不会是 I am going to school ，而是 I am going to $i
EOF
cat EOF1.txt
- tee 命令的使用
cat <<EOF | tee a,txt
>abc
>def
>123
>EOF
abc
def
123
- 整体赋值给变量，然后通过 echo 命令将变量值打印出来
#!/bin/bash
var="Great! I am going to school!"
myvar=$(cat <<EOF
This is Line 1.
Today is Monday.
$var
EOF
)
echo "$myvar" #双引号引起来则保留换行符，否则一行输出
- 多行注释
Bash 的默认注释是“#”，该注释方法只支持单行注释；Here Document 的引入解决了多行注释的问题。 “:”代表什么都不做的空命令。中间标记区域的内容不会被执行，会被bash 忽略掉，因此可达到批量注释的效果。
#!/bin/bash
var="Great! I am going to school!"
: <<-EOF #多行注释，“:”开头的 Here Document 标记内容不会被执行;EOF前面个加“-”，则会忽略EOF内容当中的制表符
This is Line 1.
Today is Monday.
$var
EOF
echo "abcd"
二、Expect 命令
2.1 Expect 概述
- Expect：建立在tcl语言基础上的一个工具，常被用于进行自动化控制和测试，解决shell脚本中交互相关的问题。
- Expect 安装与使用环境
rpm -q expect
rpm -q tcl
yum install -y expect #安装expect即可，tcl会作为依赖包一并安装
which expect
/usr/bin/expect #expect执行环境
2.2 基本命令
（1）脚本解释器
expect 脚本中首先引入文件，表明使用的是哪一个 shell。
#!/usr/bin/expect
（2）spawn
spawn 后面通常跟一个Linux执行命令，表示开启一个会话、启动进程，并跟踪后续交互信息。
（3）expect
判断上次输出结果中是否包含指定的字符串，如果有则立即返回，否则就等待超时时间后返回；
只能捕捉由spawn启动的进程的输出；
用于接收命令执行后的输出，然后和期望的字符串匹配
（4）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"}
}
（5）结束符
expect eof
表示交互结束，等待执行结束，退回到原用户，与spawn对应。
比如切换到root用户，expect脚本默认的是等待10s，当执行完命令后，默认停留10s后，自动切回了原用户
interact
执行完成后保持交互状态，把控制权交给控制台，会停留在目标终端而不会退回到原终端，这个时候就可以手工操作了，interact后的命令不起作用，比如interact后添加exit，并不会退出root用户。而如果没有interact则登录完成后会退出，而不是留在远程终端上。
使用interact会保持在终端而不会退回到原终端，比如切换到root用户，会一直在root用户状态下；比如ssh到另一服务器，会一直在目标服务器终端，而不会切回的原服务器。
注意：expect eof 与 interact 只能二选一。
（6）set
expect 默认的超时时间是 10 秒，通过 set 命令可以设置会话超时时间, 若不限制超时时间则应设置为-1。
例：set timeout 30
（7）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 执行报错
（8）send_user
send_user 表示回显命令，相当于 echo
（9）接收参数
expect 脚本可以接受从bash命令行传递的参数，使用[lindex $argv n]获得。其中n从0开始，分别表示第一个,第二个,第三个....参数。
例：
set hostname [lindex $argv 0] 相当于 hostname=$1
set password [lindex $argv 1] 相当于 password=$2
2.3 操作
- expect直接执行，需要使用 expect 命令去执行脚本，su切换用户
#!/usr/bin/expect
#设置超时时间
set timeout 5
#参数传入
set username [lindex $argv 0]
set password [lindex $argv 1]
#开始追踪命令
spawn su $username
#免交互执行，捕捉信息并匹配
expect "密码"
send "$password\r"
expect "*]#"
send_user "ok"
#把控制权交给控制台
interact
#expect eof
- 实现 ssh 自动登录
#!/usr/bin/expect
set timeout 5
set hostname [lindex $argv 0]
set password [lindex $argv 1]
spawn ssh $hostname
expect {
"Connection refused" {send_user "ssh访问被拒绝\n"} #连接失败情况，比如对方ssh服务关闭
"No route to host" {send_user "主机名/IP有误\n"} #找不到服务器，比如输入的IP地址不正确
"(yes/no)" {send "yes\r";exp_continue}
"password:" {send "$password\r"}
}
interact
exit #interact后的命令不起作用
