Shell编程之免交互

Posted 2021-05-24 Zmac111

Shell编程之免交互

• 一、Here Document
• • 概述
  • 格式
  • 注意事项
  • 几个实例
  • • 免交互方式实现对行数的统计
    • 通过 read 命令接收输入并打印
    • 通过 passwd 给用户设置密码
    • 支持变量替换
    • 整体赋值给变量
    • 关闭变量替换的功能
    • 去掉每行之前的 TAB 字符
    • 多行注释
• 二、Expect
• • 基本命令
  • • • expect eof
      • interact
  • 使用实例
  • • 嵌入执行（创建新用户，密码）
    • ssh自动登录
  • 磁盘分区
• 总结

一、Here Document

概述

（1）使用I/O重定向的方式将命令列表提供给交互式程序，比如ftp、cat或read命令。

（2）Here Document是标准输入的一种替代品可以帮助脚本开发人员不必使用临时文件来构建输入信息，而是直接就生产出一个“文件”并用作“命令”的标准输入。Here Document 也可以与非交互式程序和命令一起使用。

格式

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

注意事项

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

几个实例

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

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

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

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

read i <<EOF
>NIHAO
>EOF
echo $i

通过 passwd 给用户设置密码

passwd lolo <<EOF
>abc1234 	   #输入的密码
>abc1234       #确认密码
>EOF

支持变量替换

vim ren.sh
#!/bin/bash
file="ren.txt"
i="boy"
cat > $file <<EOF
I am a good $i
EOF

sh ren.sh
cat ren.txt

整体赋值给变量

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

关闭变量替换的功能

#!/bin/bash
var="Great! I am going to school!"
xx=$(cat <<'EOF'   #加' '表示关闭替换功能
Today is Monday.
$var
EOF
)
echo $xx

去掉每行之前的 TAB 字符

#!/bin/bash
var="Great! I am going to school!"
xx=$(cat <<EOF
		Today is Monday.    #前面为TAB字符
        $var                #前面为空格
EOF
)
echo "$xx"       # 加 ""可以换行输出，避免挤在同一行

多行注释

Bash的默认注释是“#”，该注释方法只支持单行注释。

Here Document 的引入解决了多行注释的问题。**":"**代表什么都不做的空命令。中间标记区域的内容不会被执行，会被bash忽略掉，因此可达到批量注释的效果。

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

二、Expect

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

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

基本命令

脚本解释器

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

spawn

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

expect

（1）判断上次输出结果中是否包含指定的字符串，如果有则立即返回，否则就等待超时时间后返回。

（2）只能捕捉由spawn启动的进程的输出，用于接收命令执行后的输出，然后和期望的字符串匹配

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"}
}

结束符

expect eof

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

interact

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

注：expect eof 与 interact 只能二选一

set

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

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

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 执行报错

send_user

表示回显命令，相当于 echo

接收参数

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

举例：
set hostname [lindex $argv 0]   	相当于 hostname=$1
set password [lindex $argv 1]		相当于 password=$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

ssh自动登录

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

磁盘分区

方法一

#!/usr/bin/expect

spawn fdisk /dev/sdb

expect "命令" {send "n\\r"}

expect "Select" {send "p\\r"}

expect "分区号" {send "1\\r"}

expect "起始" {send " \\r"}

expect "Last 扇区" {send "+10G\\r"}

expect "命令" {send "w\\r"}

expect eof

方法二

[root@localhost ~]# fdisk /dev/sdb <<-EOF
> n
> p
> 2
> 
> 
> 
> w
> EOF

总结

（1）掌握Here Document 和Expect 两种免交互的方法
（2）exp_continue 的用法，并行结构时使用