Shell脚本之免交互（EOFExpect）

Posted 2022-01-04 热心群众

@[toc](目录

一、Here Document 免交互

1、相关概念

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

2、语法格式

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

3、注意事项

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

4、示例

（1）免交互方式实现对行数的统计

• 将要统计的内容置于标记“EOF”之间，直接将内容传给 wc -l 来统计

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

  

（2）通过 read 命令接收输入并打印

• 输入值是两个 EOF 标记之间的部分，作为变量 i 的值

  read i <<EOF
  >Hi
  >EOF
  echo $i

  

  （3）通过 passwd 给用户设置密码

  passwd lisi <<EOF
  >abc1234    #这两行是输入的密码和确认密码
  >abc1234
  >EOF

  

（4）支持变量替换

vim ti.sh
#!/bin/bash
file="ti.txt"
i="zhangsan"
cat > $file <<EOF
hello $i
EOF

sh ti.sh
cat ti.txt

（5）整体赋值给变量

#!/bin/bash
var="day!"
bian=$(cat <<EOF
Today is Monday.
$var
EOF
)
echo $bian

（6）关闭变量替换的功能,按照字符原本的样子输出，不做任何修改或替换

#!/bin/bash
var="Great! I am going to school!"
bian=$(cat <<EOF
Today is Monday.
$var
EOF
)
echo $bian

（7）去掉每行之前的 TAB 字符

#!/bin/bash
var="day day up!"
bian=$(cat <<-EOF       #对标记前加“-”，即可抑制各行首TAB或空格          
    Today is Monday.
  $var
EOF
)
echo "$bian"

（8）多行注释

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

  #!/bin/bash
  var="day day up!"
  : bian=$(cat <<EOF
      Today is Monday.
  $var
  EOF
  )
  echo "$bian"
  echo "$var"

  

  二、expect免交互

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

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

  1、基本命令

  （1）脚本解释器

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

  #!/usr/bin/expect

  （2）spawn

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

  例：spawn passwd root

  （3）expect

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

（4）send

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

  例如：发送密码
  方式一：
  expect "密码" send "123456\\r"       #同一行send部分要有
  方式二：
  expect "密码" 
  send "123456\\r"                         #换行send部分不需要有
  方式三：
  expect                  #只要匹配了其中一个情况，执行相应的send语句后退出该expect语句
  
  "密码1" send "123456\\r"
  "密码2" send "123123\\r"
  "密码3" send "abc123\\r"
  

  （5）结束符

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

  expect eof

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

  interact

  注意：expect eof 与 interact 只能二选一。

  （6）set

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

  例：设置超时时间为30秒
  set timeout 30

  （7）exp_continue

• exp_continue 附加于某个 expect 判断项之后，可以使该项被匹配后，还能继续匹配该 expect 判断语句内的其他项。exp_continue 类似于控制语句中的 continue 语句。
• 例如：下例将判断交互输出中是否存在 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

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

  send_user

  （9）接收参数

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

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

  2、示例

  （1）expect直接执行

• 执行不能使用之前的方式执行，需要使用- expect命令去执行脚本
• 例：su切换用户

  #!/usr/bin/expect
  set timeout 3
  set username [lindex $argv 0]
  set password [lindex $argv 1]
  spawn su $username
  expect "密码" send "$password\\r"
  interact         #或者expect eof

  

  
  这里要先确保创建zhangsan 和lisi用户
  cd /home/  #进入家目录
  ls

useradd lisi
echo 123456 | passwd --stadin lisi
#lisi的密码要和后面的一样
chmod 777 /home/zhangsan

给zhangsan 权限加满

su zhangsan
cd #cd到zhangsan家目录
在zhangsan家目录里创建并执行文件

### （2）嵌入执行
- 将 expect 过程融入 Shell 当中，方便执行和处理
- 例：创建用户并设置密码

#!/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

![image.png](https://s2.51cto.com/images/20211230/1640860136282102.png?x-oss-process=image/watermark,size_14,text_QDUxQ1RP5Y2a5a6i,color_FFFFFF,t_100,g_se,x_10,y_10,shadow_20,type_ZmFuZ3poZW5naGVpdGk=)
==这里不需要用户已存在，可直接通过脚本创建新的用户，顺便更新密码==

### （3）实现 ssh 自动登录
- 直接用expect执行

#!/usr/bin/expect
set timeout 5
set hostname [lindex $argv 0]
set password [lindex $argv 1]

spawn ssh $hostname
expect
"Connection refused" exit
"Name or service not known" exit
"to continue" send "yes\\r";exp_continue
"password:" send "$password\\r"

interact
exit #interact后的命令不起作用

![image.png](https://s2.51cto.com/images/20211230/1640861349590949.png?x-oss-process=image/watermark,size_14,text_QDUxQ1RP5Y2a5a6i,color_FFFFFF,t_100,g_se,x_10,y_10,shadow_20,type_ZmFuZ3poZW5naGVpdGk=)
![image.png](https://s2.51cto.com/images/20211230/1640861370714885.png?x-oss-process=image/watermark,size_14,text_QDUxQ1RP5Y2a5a6i,color_FFFFFF,t_100,g_se,x_10,y_10,shadow_20,type_ZmFuZ3poZW5naGVpdGk=)

- 嵌入式

#!/bin/bash
hostname=$1
password=$2
/usr/bin/expect <<EOF
spawn ssh $hostname
expect
"(yes/no)" send "yes\\r";exp_continue
"password" send "$password\\r"

expect "]#" send "ifconfig\\r"
expect eof
EOF

![image.png](https://s2.51cto.com/images/20211230/1640862055172689.png?x-oss-process=image/watermark,size_14,text_QDUxQ1RP5Y2a5a6i,color_FFFFFF,t_100,g_se,x_10,y_10,shadow_20,type_ZmFuZ3poZW5naGVpdGk=)
![image.png](5)