shell 脚本免交互 Here Document 和 expect

shell 脚本之免交互

一： Here Document 免交互

1.1 概述

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

1.2 语法格式

命令 << 标记

传入的内容

标记

注意

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

1.3 示例

1.3.1 免交互方式实现对行数的统计，将要统计的内容放在标记之间

1.3.2 通过read 命令接受输入并打印

1.3.3 通过passwd 给用户设置密码

1.3.4 将内容输入到文件中

1.3.5支持变量替换

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

复制#!/bin/bash
file="EOF.txt"
i="school"

cat > $file <<EOF
this is my $i
EOF

cat EOF.txt

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

复制#!/bin/bash
var="great! i am going to school!"
myvar=$(cat <<EOF
        this is line 1.
        today is monday.
        $var
EOF
) #) 要另起一行写，不可以写在EOF 后面


#变量要加上双引号“” ，否则内容将会输出在1行中
echo "$myvar"

1.3.7 关闭变量替换功能

复制#!/bin/bash
var="great! i am going to school!"
#在第一个标记两边加上单引号'' 即可关闭变量的替换功能
myvar=$(cat <<'EOF'
        this is line 1.
        today is monday.
        $var
EOF
) #")" 要另起一行写，不可以写在EOF 后面

echo "$myvar" 

1.3.8 去掉每行之前的 TAB字符（无法去掉空格造成的空）

复制#!/bin/bash
var="great! i am going to school!"

#在第一个标记前面加上"-" ，既可以抑制各行行首的TAB
#注意，只是抑制TAB，而不是空格
myvar=$(cat <<-EOF
        this is line 1.
        today is monday.
        $var
EOF
) #) 要另起一行写，不可以写在EOF 后面

echo "$myvar" 
~             

1.3.9 多行注释

":"代表什么都不做的空命令。中间标记区域的内容不会被执行，会被bash忽略掉，因此可达到批量注释的效果

复制#!/bin/bash

# ":" 开头的Here Document 标记内容不会被执行
: <<EOF
this is line 1.
today is monday.
EOF

echo "abcd"

二：expect 工具

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

所以，需要先安装软件包 expect 和 tcl

yum -y install expect

2.1 基本命令

2.1.1 脚本解释器

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

#!/usr/bin/expect

2.1.2 spawn

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

spawn passwd root

2.1.3 expect

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

2.1.4 send

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

1. 复制expect "密码" {send "1234567\n"}      #同一行send 部分要有 { }
   

2. 复制expect "密码"
   
   send  "1234567\n"          #换行send 部分不需要有 { }
   

3. 复制#expect 支持多分枝。只要匹配了其中一个情况，执行相应的send 语句后，退出相应的expect语句
   expect 
   {
   	"密码"  {send "1234567\n}
   	"password"  {send "1234567"}
   	}
   

2.1.5 结束符

1. expect eof

   表示交互结束，等待执行结束，退回到原用户，与spawn 对应

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

2. interact

   执行完成后保持交互状态，吧控制权交给控制台，会停留在目标终端而不会退回到原终端。这个时候就可以手工操作。

   interact 后的命令不起作用。比如interact 后面添加exit ，并不会退出root用户。

   而如果没有interact ，则登录完成后会退出，而不是留在远程终端上

   使用interact会保持在终端而不会退回到原终端，比如切换到root用户，会一直在root用户状态下； 比如ssh到另一服务器，会一直在目标服务器终端，而不会切回的原服务器。

# expect eof 和 interact 只能二选一

2.1.6 set

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

set timeout 10

2.1.7 exp_continue

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

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

复制expect {
  "(yes/no)" {send "yes\n";exp_continue;}
	"*password"  {set timeout 300;send "1234567";}
	}
#判断交互输出中是否存在yes/no 或者*password 。如果匹配 yes/no 则输出yes，并再次执行判断；如果匹配*password 则输出1234567，并结束该段expect6语句

2.1.8 send_user

send_user 表示回显命令，相当于echo

2.2 接收参数

expect 脚本口语i接受从bash 命令行传递的参数，使用 [lindex $argv n] 获得。n 从0开始，分别表示第一个，第二个...... 参数

复制set hostname [lindex $argv 0]    
set password  [lindex $argv 1]

2.2.1免交互切换用户

复制#!/usr/bin/expect

#设置超时时间为5s
set timeout 5

#设置第一个位置参数
set username [lindex $argv 0]

#设置第二个位置参数
set password [lindex $argv 1]

#开启跟踪命令
spawn su - $username

#免交互执行，捕捉信息并匹配
expect "密码："
send "${password}\n"

#结束符
#如果设置成expect eof 则会切换用户后，过5s，然后自动退回到原用户
interact

2.2.2 免交互远程登录

复制#!/usr/bin/expect

set timeout 5

set hostname [lindex $argv 0]
set password [lindex $argv 1]

spawn ssh $hostname

#设置多分支匹配
#ssh远程，首次登录需要保存登录主机信息。
expect {
		"NO route to host" exit
		"Connection refused" exit
		"(yes/no)" {send "yes\n" ;exp_continue}
		"password:" {send "$password\n"}
}
interact

2.3 嵌入执行模式

将expect 过程融入到 shell 当中，方便执行和处理

例：创建用户并设置密码

复制#!/bin/bash

#设置位置参数传入
#设置第一个位置参数
username=$1
#设置第二个位置参数
password=$2

#注意，/usr/bin/expect解释器不支持useradd
useradd $username


#使用Here Document 将expect 的输入直接交给 /usr/bin/expect 执行
/usr/bin/expect <<EOF
#开启命令跟踪
spawn passwd $username

#免交互执行，捕捉信息并匹配
expect "新的 密码"
send "${password}\n"

expect "重新输入新的 密码"
send "${password}\n"

#结束免交互
expect eof
EOF
#结尾标记