Shell 脚本之免交互操作

文章目录

  • 1. Here Document 免交互
    • 1.1 概述
    • 1.2 范例
  • 2. expect 免交互
    • 2.1 概述
    • 2.2 基本命令
      • (1) 脚本解释器
      • (2) spawn
      • (3) expect
      • (4) send
      • (5) 结束符
      • (6) set
      • (7) exp_continue
      • (8) send_user
      • (9) 接收参数
    • 2.3 范例
      • (1) 免交互创建磁盘
      • (2) 免交互登录远程主机
      • (3) 免交互登录远程主机执行命令并回退
      • (4) 变量参数实现远程登录
      • (5) 利用 expect 完成 ftp 登录

1. Here Document 免交互

1.1 概述

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

格式:

命令 <<标记
......
传入的内容
......
标记

注意事项:

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

1.2 范例

在屏幕上输出信息,不保存在文本中:

[root@c7-1 ~]#cat <
> this is a test
> haha
> EOF
this is a test
haha

保存到文件中,不输出到屏幕:

[root@c7-1 ~]#cat > test.txt <
> this is a test
> hello
> EOF
[root@c7-1 ~]#cat test.txt 
this is a test
hello

免交互方式实现对行数的统计,将要统计的内容置于标记 EOF 之间:

[root@c7-1 ~]#wc -l <
> 1
> 12
> 123
> EOF
3

通过 read 命令接收输入作为变量的值:
注意:如果有多行的话,只有第一行赋值给变量

[root@c7-1 ~]#read i <
> ABC
> abc
> EOF
[root@c7-1 ~]#echo $i
ABC

通过 passwd 给用户设置密码:

[root@c7-1 ~]#useradd zhangsan
[root@c7-1 ~]#passwd zhangsan <
> 123456
> 123456
> EOF
更改用户 zhangsan 的密码 。
新的 密码:无效的密码: 密码少于 8 个字符
重新输入新的 密码:passwd:所有的身份验证令牌已经成功更新。
#passwd 需要输入两次密码确认

支持变量替换:

[root@c7-1 ~]#cat test.sh 
#!/bin/bash
touch /root/file.txt
testfile="/root/file.txt"
I="school"
cat >> $testfile <<EOF
I'm go to $I
EOF
[root@c7-1 ~]#bash test.sh
[root@c7-1 ~]#cat file.txt 
I'm go to school

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

[root@c7-1 ~]#cat test.sh 
#!/bin/bash
var="这是天堂"
myvar=$(cat <<EOF
这是哪里?
$var
EOF
)
echo "$myvar"
[root@c7-1 ~]#bash test.sh 
这是哪里?
这是天堂

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

[root@c7-1 ~]#cat test.sh 
#!/bin/bash
var="这是天堂"
myvar=$(cat <<'EOF'		//标记加上单引号即为关闭变量替换
这是哪里?
$var
EOF
)
echo "$myvar"
[root@c7-1 ~]#bash test.sh 
这是哪里?
$var

去掉每行之前的 TAB 字符(去掉缩进,顶格输出):

[root@c7-1 ~]#cat test.sh 
#!/bin/bash
var="这是天堂"
myvar=$(cat <<-'EOF'	//标记前加 - 符号
	这是哪里?		//这里使用 TAB 缩进
        $var		//这里使用空格缩进
EOF
)
echo "$myvar"
[root@c7-1 ~]#bash test.sh 
这是哪里?			//可以看出 TAB 缩进失效了
        $var		//空格缩进仍然生效,只能禁止 TAB 缩进

多行注释:

  • bash 的默认注释是 “#”,该注释方法只支持单行注释: Here Document 的引入解决了多行注释的问题
  • “:” 代表什么都不做的空命令。中间标记区域的内容不会被执行,会被 bash 忽略掉,因此可达到批量注释的效果
[root@c7-1 ~]#cat test.sh 
#!/bin/bash
var="这是天堂"
cat <<EOF
这是哪里?
$var
EOF
echo "hello world"
[root@c7-1 ~]#bash test.sh
这是哪里?
这是天堂
hello world

[root@c7-1 ~]#vim test.sh 
[root@c7-1 ~]#cat test.sh 
#!/bin/bash
var="这是天堂"
: cat <<EOF		//加上 : 后,该标记区域内容会被忽略
这是哪里?
$var
EOF
echo "hello world"		//不在标记区域的内容可以输出
[root@c7-1 ~]#bash test.sh
hello world

2. expect 免交互

2.1 概述

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

expect 的安装:expect 是一个程序,所以它也是需要提前安装才可以使用的

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

2.2 基本命令

(1) 脚本解释器

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

#!/usr/bin/expect

(2) spawn

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

spawn passwd root
spawn ssh $user@$ip

(3) expect

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

(4) send

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

示例:

方式一
expect "密码" {send "abc123\r"}		//同一行 send 部分要有{}

方式二
expect "密码"
send "abc123\r"		//换行 send 部分不需要有{}

方式三
// expect 支持多个分支
expect {         //只要匹配了其中一个情况,执行相应的 send 语句后退出该 expect 语句
"密码1"{ send "abc123\r"}
"密码2"{ send "123456\r"}
"密码3"{ send "123123\r"}
}

(5) 结束符

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

(6) set

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

set timeout 30

(7) exp_continue

  • exp_continue 附加于某个 expect 判断项之后,可以使该项被匹配后,还能继续匹配该 expect 判断语句内的其他项
  • exp_continue 类似于控制语句中的 continue 语句,表示允许 expect 继续向下执行指令
以下示例将判断交互输出中是否存在 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

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

(9) 接收参数

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

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

2.3 范例

(1) 免交互创建磁盘

[root@c7-1 ~]#cat test.sh
#!/usr/bin/expect
set dev [lindex $argv 0]
spawn fdisk $dev
 
expect "(输入m获取帮助):"
send "n\r"
expect "(设置分区):"
send "p\r"
expect "(1-4,默认为1):"
send "\r"
expect "起始扇区"
send "\r"
expect "Last 扇区"
send "+10G\r"
expect "已设置为Linux 类型"
send "w\r"
 
expect eof

执行

chmod +x test.sh
./test.sh /dev/sdb
/usr/bin/expect/test.sh /dev/sdb

(2) 免交互登录远程主机

[root@c7-2 ~]#hostname -I
192.168.10.30 
[root@c7-2 ~]#cat test.sh 
#!/usr/bin/expect
spawn ssh [email protected]

expect {
	"password:"
	{ send "120604\r"}
}

interact	//登录后停留在目标终端,不回退
[root@c7-2 ~]#chmod +x test.sh && ./test.sh 
spawn ssh [email protected]
[email protected]'s password: 
Last login: Wed Sep 15 22:11:41 2021 from 192.168.10.30
[root@c7-1 ~]#hostname -I
192.168.10.20 

(3) 免交互登录远程主机执行命令并回退

[root@c7-2 ~]#cat test.sh 
#!/usr/bin/expect
spawn ssh [email protected]

expect {
	"password:"
	{ send "120604\r"}
}

expect "#"		//检测到该信号执行下列命令
send "df -h|grep sd*\r"
send "hostname -I\r"
send "exit\r"

expect eof		//该终止符表示回退原主机

[root@c7-2 ~]#./test.sh 
spawn ssh [email protected]
[email protected]'s password: 
Last login: Wed Sep 15 22:17:35 2021 from 192.168.10.30
[root@c7-1 ~]#df -h|grep sd*
Filesystem      Size  Used Avail Use% Mounted on
devtmpfs        1.9G     0  1.9G   0% /dev
tmpfs           1.9G     0  1.9G   0% /dev/shm
tmpfs           1.9G   13M  1.9G   1% /run
tmpfs           1.9G     0  1.9G   0% /sys/fs/cgroup
/dev/sda2        50G  5.3G   45G  11% /
/dev/sda5        44G   33M   44G   1% /data
/dev/sda1      1014M  179M  836M  18% /boot
tmpfs           378M  4.0K  378M   1% /run/user/42
tmpfs           378M   40K  378M   1% /run/user/0
/dev/sr0        4.4G  4.4G     0 100% /run/media/root/CentOS 7 x86_64
[root@c7-1 ~]#hostname -I
192.168.10.20 
[root@c7-1 ~]#exit
logout
Connection to 192.168.10.20 closed.
[root@c7-2 ~]#

(4) 变量参数实现远程登录

[root@c7-2 ~]#cat test.sh 
#!/usr/bin/expect
set user root
set ip [lindex $argv 0]
set pass [lindex $argv 1]

spawn ssh $user@$ip

expect {
	"password:"
	{ send "$pass\r";}
}

expect "#"
send "hostname -I\r"
send "exit\r"

expect eof
[root@c7-2 ~]#./test.sh 192.168.10.20 120604
spawn ssh [email protected]
[email protected]'s password: 
Last login: Wed Sep 15 22:18:04 2021 from 192.168.10.30
[root@c7-1 ~]#hostname -I
192.168.10.20 
[root@c7-1 ~]#exit
logout
Connection to 192.168.10.20 closed.

(5) 利用 expect 完成 ftp 登录

准备工作:

客户端
yum -y install ftp
服务端
yum -y install vsftpd
systemctl start vsftpd && systemctl enable vsftpd 

客户端脚本内容:

[root@c7-1 ~]#cat expect_ftp.sh 
#!/usr/bin/expect 
set timeout 10
spawn ftp 192.168.10.30		//spawn 表示追踪进程
expect "Name"
send "ftp\r"
expect "Password:*"			//当你不确定的时候就用 * 表示任意
send "\r"
expect "ftp>*"

interact
expect eof

测试登录:

[root@c7-1 ~]#./expect_ftp.sh 
spawn ftp 192.168.10.30
Connected to 192.168.10.30 (192.168.10.30).
220 (vsFTPd 3.0.2)
Name (192.168.10.30:root): ftp
331 Please specify the password.
Password:
230 Login successful.
Remote system type is UNIX.
Using binary mode to transfer files.
ftp> ls
227 Entering Passive Mode (192,168,10,30,90,79).
150 Here comes the directory listing.
drwxr-xr-x    2 0        0               6 Jun 09 16:15 pub
226 Directory send OK.
ftp> exit
221 Goodbye.
expect: spawn id exp6 not open
    while executing
"expect eof"
    (file "./expect_ftp.sh" line 11)

———————————————————————————————————————————————————
参考:
shell 编程之免交互
《云计算》-shell脚本编程-expect预期交互
shell编程之——免交互操作

你可能感兴趣的:(shell,云计算,linux运维,linux,shell,eof,云计算,运维)