Here Document 免交互与 Expect 自动化交互

Posted 他和晚风一样温柔

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Here Document 免交互与 Expect 自动化交互相关的知识,希望对你有一定的参考价值。

文章目录

Here Document 免交互

概述

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

语法格式

命令 <<标记
...
内容    #标记直接是传入内容
...
标记

注意事项

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

示例

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

  • 将要统计的内容置于标记“EOF”之间
  • 直接将内容传给wc -l 来统计
[root@localhost ~]# wc -l <<EOF
> xcf1
> xcf2
> EOF
2


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

  • 输入值是两个EOF标记之间的部分
  • 直接为变量i赋值
[root@localhost ~]# read i <<EOF
> hello xcf~
> EOF
[root@localhost ~]# echo $i
hello xcf~


通过passwd给用户设置密码

[root@localhost ~]# useradd xcf123
[root@localhost ~]# passwd xcf123 << EOF
123123
123123
EOF
更改用户 xcf123 的密码 。
新的 密码:无效的密码: 密码少于 8 个字符
重新输入新的 密码:passwd:所有的身份验证令牌已经成功更新。


支持变量替换

  • 在写入文件时会先将变量替换成实际值
  • 然后再结合 cat 命令完成写入
[root@localhost shell]# vim zxc1.sh

#!/bin/bash

file="xcf1.txt"
i="chicken"
cat > $file <<EOF
I'd like some $i
EOF
[root@localhost shell]# . zxc1.sh 
[root@localhost shell]# ls
xcf1.txt  zxc1.sh
[root@localhost shell]# cat xcf1.txt 
I'd like some chicken



整体赋值给变量,然后通过echo命令打印出变量值

[root@localhost shell]# vim zxc2.sh

#!/bin/bash

name="OMG! xucheng come!"
myname=$(cat <<EOF
It's a beautiful day
Today is Sunday
School starts again tomorrow
$name ~
EOF
)
echo $myname
[root@localhost shell]# . zxc2.sh 
It's a beautiful day Today is Sunday School starts again tomorrow OMG! xucheng come! ~



关闭变量替换的功能

  • 这样会按照字符原本的样子输出
  • 且没有任何修改或替换


去掉每行之前的TAB字符

  • 用处不大,知道这种用法能去掉各行的tab制表符即可


多行注释

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


Expect 自动化交互

概述

  • expect 是一个免费的编程工具语言,常用来实现自动和交互式任务进行通信,而无需人的干预
  • expect 需要 Tcl 编程语言的支持,要在系统上运行 expect 必须首先安装 Tcl
yum -y install tcl

yum -y install expect

基本命令

脚本解释器

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

spawn

  • spawn 后面通常跟一个Linux执行命令,表示开启一个会话、启动进程,并跟踪后续交互信息
例: spawn passwd root          ##跟踪启动更改用户密码的进程

expect

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

send

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

方式一

expect “密码” send “abc123\\r”
##同一行send部分要有

方式二

expect “密码”
send “$abc123\\r”
##换行send部分不需要有

方式三

expect “支持多个分支

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

结束符

expect eof
  • 表示交互结束,等待执行结束,退回到原用户,与spawn对应
  • 比如:切换到root用户,expect脚本默认的是等待10s,当执行完命令后,默认停留10s后,就会自动切回原用户
interact
  • 执行完成后保持交互状态,把控制权交给控制台,会停留在目标终端,这个时候就可以手工操作了,interact后的命令不起作用,比如interact会保持在终端而不会退回到原终端,比如切换到root用户,会一直在root用户状态下
  • 比如ssh到另一服务器,会一直在目标服务器终端,而不会切回的原服务器
  • 注:expect eof 与 interact 只能二选一

set

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

exp_continue

  • exp_continue 附加于某个 expect 判断项之后,可以使该项匹配后,还能继续匹配该 expect 判断语句内的其他项
  • exp_continue类似于控制语句中continue语句,表示允许expect继续向下执行指令
  • 下例将判断交互输出中是否存在 yes/no 或*password , 如果匹配 yes/no 则输出 yes 并再次执行判断;如果匹配 *assword 则输出 123123 并结束该段 expect 语句
expect
"(yes/no)" send "yes\\r"; exp_ continue; 
"*password" set timeout 300; send "123123\\r";

send_user

  • send_ user 表示回显命令,相当于 echo

接收参数

  • expect 脚本可以接受从 bash 命令行传递的参数,使用 [lindex $argv n] 获得,其中 n 从 0 开始,分别表示第一个,第二个,第三个…
  • 示例:
set hostname [lindex $argv 0]
##相当于hostname=$1
set password [lindex $argv 1]
##相当于password=$2

expect 直接执行

  • 以SSH为例进行示例
  • $argv 0 代表位置变量$1
  • $argv 1 代表位置变量$2
  • #!/usr/bin/expect 是 Expect 二进制文件的路径
[root@localhost shell]# vim zxc11.sh 

#!/usr/bin/expect
#设置超时时间
set timeout 20
#开启日志
log_file test.log
#显示信息
log_user 1
#定义变量
set hostname [lindex $argv 0]
set password [lindex $argv 1]
#追踪指令
spawn ssh root@$hostname
#捕捉提示信息
expect 
        "connecting (yes/no)"
        send "yes\\r";exp_continue
        "*password:"
        send "$password\\r";

#转交控制权
interact

expect 嵌入执行

  • 将 expect 过程融入 Shell 当中,方便执行和处理
  • 创建用户并设置密码
#! /bin/bash
user=$1
password=$2
#非交互命令放在expect外面
useradd $user
#开始免交换执行
/usr/bin/expect <<-EOF
#expect开始标志,-去掉制表符
spawn passwd $user
#开启-一个进程跟踪passwd命令,expect只能捕捉该进程信息
expect "新的*"
send "$ password\\r" .
expect "重新*"
send "$ password \\r"
expect eof
EOF
  • 以SSH为例进行示例:
#!/bin/bash
  hostname=$1
  password=$2
  /usr/bin/expect<<-EOF
  spawn ssh root@$hostname
  expect 
      "(yes/no)"
      send "yes\\r";exp_continue
      "*password"
      send "$password\\r"
  
  expect "*]#"
  send "exit\\r"
  expect eof
EOF		'Expect结束标志,EOF前后不能有空格'

实现 SSH 自动登录

[root@localhost shell]# vim zxc12.sh 

#!/usr/bin/expect
set timeout 15
set hostname [lindex $argv 0]
set password [lindex $argv 1]
spawn ssh root@$hostname
expect 
        "connection refused" exit
        #连接失败情况,比如对方ssh服务关闭
        "Name or service not known" exit
        #找不到服务器,比如输入的IP地址不正确
        "to continue"send "yes\\r";exp_continue
        "password:"send "$password\\r";

interact
#携带interact参数表示登录成功后将控制台交给用户,否则登录完成后将退出
exit

免交互创建磁盘

  • 首先在虚拟机关闭状态手动添加一块磁盘,然后开机

  • 编写脚本
[root@localhost ~]# vim disk.sh


#!/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
mkfs.xfs $disk -f &> /dev/null

if [ $? -eq 0 ]
    then
    echo -e "\\033[31m 磁盘格式化完成 \\033[0m"
    mkdir $disk.1
    mount $disk $disk.1
    df -h
    else
    echo "格式化失败,脚本有bug"
fi
  • 检查一下,本文开头提到的软件包安装好了没
[root@localhost ~]# rpm -q tcl
tcl-8.5.13-8.el7.x86_64
[root@localhost ~]# rpm -q expect
expect-5.45-14.el7_1.x86_64
  • 执行脚本
[root@localhost ~]# sh disk.sh /dev/sdb$1
spawn fdisk /dev/sdb
欢迎使用 fdisk (util-linux 2.23.2)。

更改将停留在内存中,直到您决定将更改写入磁盘。
使用写入命令前请三思。

Device does not contain a recognized partition table
使用磁盘标识符 0x1f0887af 创建新的 DOS 磁盘标签。

命令(输入 m 获取帮助):n
Partition type:
   p   primary (0 primary, 0 extended, 4 free)
   e   extended
Select (default p): 
Using default response p
分区号 (1-4,默认 1):
起始 扇区 (2048-41943039,默认为 2048):
将使用默认值 2048
Last 扇区, +扇区 or +sizeK,M,G (2048-41943039,默认为 41943039):
将使用默认值 41943039
分区 1 已设置为 Linux 类型,大小设为 20 GiB

命令(输入 m 获取帮助):w
The partition table has been altered!

Calling ioctl() to re-read partition table.
正在同步磁盘。
Warning: 无法以读写方式打开 /dev/sr0 (只读文件系统)/dev/sr0 已按照只读方式打开。
 磁盘格式化完成 
文件系统        容量  已用  可用 已用% 挂载点
/dev/sda2        20G  3.8G   17G   19% /
devtmpfs        1.9G     0  1.9G    0% /dev
tmpfs           1.9G     0  1.9G    0% /dev/shm
tmpfs           1.9G  9.0M  1.9G    1% /run
tmpfs           1.9G     0  1.9G    0% /sys/fs/cgroup
/dev/sr0        4.3G  4.3G     0  100% /mnt
/dev/sda1       6.0G  174M  5.9G    3% /boot
/dev/sda3        10G   37M   10G    1% /home
tmpfs           378M  8.0K  378M    1% /run/user/42
tmpfs           378M     0  378M    0% /run/user/0
/dev/sdb         20G   33M   20G    1% /dev/sdb.1

以上是关于Here Document 免交互与 Expect 自动化交互的主要内容,如果未能解决你的问题,请参考以下文章

Here Document 免交互与 Expect 自动化交互

Shell脚本------免交互(Here Document免交互,Expect自动化交互)

Expect 自动化控制和测试 Here Document 免交互 Linux- shell编程之免交互

Linux Document免交互与Expect自动化交互

shell 实现免交互 EOF 与 expect

Shell脚本之Expect免交互