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自动化交互)