Shell编程之免交互

一、Here Document免交互

1. 定义

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

2. 语法格式

命令 <<标记  
...
内容              #标记之间是传入内容
...
标记

注意事项：

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

3. 实现对行数的统计

wc -l <<EOF
>Line1
>Line2
>EOF

4. 通过read命令接收输入并打印两个EOF标记之间的内容

read i <<EOF
>Hello
>EOF
>echo $i

5. 通过passwd给用户设置密码

passwd zs <<EOF
>abc123               #这两行是输入密码和确认密码
>abc123
>>EOF

6. 支持变量替换

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

#!/bin/bash
file="EOF1.txt"
i="Home"
cat > $file <<EOF
I am going $i
EOF

7. 整体赋值给变量，通过echo命令加印出来

#!/bin/bash
var="Great! NG is coming to pc"
myvar=$(cat <<EOF
Today is Monday
$var
EOF
)

8. 关闭变量替换功能，原样输出字符

#!/bin/bash
var="Great! NG is coming to pc"
myvar=$(cat <<'EOF'
Today is Monday
$var
EOF
)

9. 去掉每行之前的TAB字符

-EOF        #抑制TAB字符输出

10. 多行注释

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

:<<EOF             #注释

二、Expect

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

rpm -q expect
rpm -q tcl
yum install -y expect

1. 基本命令

1.1 脚本编辑器

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

#!/usr/bin/expect

1.2 spawn

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

1.3 expect

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

2. 模拟用户输入字符串

2.1 方法一

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

2.2 方法二

expect "密码" 
send "abc123\r"                                换行时不需要加{}

2.3 方法三

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

3. 结束符

3.1 expect eof

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

3.2 interact

• 执行完成后保持交互状态，把控制权交给控制台，会停留在目标终端而不会退回到原终端，这个时候就可以手工操作了，interact后的命令不起作用，比如interact后添加exit，并不会退出root用户。而如果没有interact则登录完成后会退出，而不是留在远程终端上
• 使用interact会保持在终端而不会退回到原终端，比如切换到root用户，会一直在root用户状态下；比如ssh到另一服务器，会一直在目标服务器终端，而不会切回的原服务器

3.3 set

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

set timeout 20

3.4 exp_continue

• exp_continue 附加于某个 expect 判断项之后，可以使该项被匹配后，还能继续匹配该 expect 判断语句内的其他项。exp_continue 类似于控制语句中的 continue 语句。表示允许 expect 继续向下执行指令

expect {
    "(yes/no)" {send "yes\r"; exp_continue;}          若匹配到yes/no字符，则输出yes及回车键确定
    "*password" {send "abc123\r";}                    若匹配到password字符，则输出密码abc123及回车键确定
}

3.5 接收参数

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

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

三、操作示例

1. expect脚本直接执行

#!/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 "su success!"
expect eof

2. 嵌入执行模式

#!/bin/bash
user=$1
password=$2
useradd $user
/usr/bin/expect <<-EOF
spawn passwd $user
expect "新的*"
send "${password}\r"
expect "重新*"
send "${password}\r"
expect eof
EOF

3. 实现ssh自动登录

3.1 expect直接执行

#!/usr/bin/expect   
set host [lindex $argv 0]
set passwd [lindex $argv 1]
spawn ssh $host
expect {
"Connection refused" exit     
"No route to host" exit       
"(yes/no)" {send "yes\r"; exp_continue}   
"*password:" {send "${passwd}\r"}        
}
interact

3.2 内嵌式

#!/bin/bash
username=$1
passwd=$2
/usr/bin/expect<<-EOF
spawn ssh $username
expect {
"Connection refused" exit
"No route to host" exit
"(yes/no)" {send "yes\r";exp_contiune}
"password:" {send "$passwd\r"}
}
expect eof
EOF

4. 免交互创建磁盘

#!/bin/bash
disk=$1
/usr/bin/expect <<-EOF
spawn fdisk $disk
expect "命令" {send "n\r"}
expect "Select" {send "\r"}
expect "分区" {send "\r"}
expect "起始" {send "\r"}
expect "Last" {send "\r"}
expect "命令(输入 m 获取帮助)：" {send "w\r"}
expect eof
EOF
partprobe $disk
mkfs.xfs $disk -f &> /dev/null
if [ $? -eq 0 ]
then
echo "finsh"
mount $disk /mnt
else
echo "fail"
fi