shell编程之免交互

 

交互的概念 与Linux中的运用
交互：当计算机播放某多媒体程序的时候，编程人员可以发出指令控制该程序的运行，而不是程序单方面执行下去，程序在接受到编程人员相应的指令后而相应地做出反应。

对于Linux操作系统中，有许多操作都会触及到交互（根据系统的指示做出相对应的操作满足操作者的需求），对于shell脚本的自动化运维，就要实现免交互来达到自动化运维的效果
常用的交互程序：read，ftp，passwd，su，sudo，fdisk等等 
cat也可配合免交互的方式重定向输出到文件。

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

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

#免交互方式实现对行数的统计，将要统计的内容置于标记“EOF”之间，直接将内容传给 wc -l 来统计
[root@zy2 ~]# wc -l <<EOF
> 123
> 456
> 789
> EOF           #已什么为标记，就以什么为结束
3

#通过 read 命令接收输入并打印，输入值是两个 EOF 标记之间的部分，作为变量 i 的值
[root@zy2 ~]# read i <<EOF
> hi
> EOF
[root@zy2 ~]# echo $i
hi

#通过 passwd 给用户设置密码
[root@zy2 ~]# passwd aa <<EOF
> 123456      #输入密码   
> 123456      #确认密码
> EOF
Changing password for user aa.
New password: BAD PASSWORD: The password is shorter than 8 characters
Retype new password: passwd: all authentication tokens updated successfully

#通过cat实现查看
[root@zy2 ~]# cat <<EOF
> my name is zhangsan
> hello world
> EOF
my name is zhangsan
hello world

#重定向输出到指定文件
[root@zy2 ~]# cat <<EOF> test.txt
> hello world
> I am zhangsan
> EOF
[root@zy2 ~]# cat test.txt
hello world
I am zhangsan

#tee命令重定向输出加标准输出
[root@zy2 ~]# tee test2.txt <<EOF
> this is tee test
> EOF
this is tee test

#tee命令有标准输出的作用，可以直接重定向输入到文件中
[root@zy2 ~]# cat test2.txt 
this is tee test

#支持变量替换，在写入文件时会先将变量替换成实际值，再结合 cat 命令完成写入
#!/bin/bash
file="test1.txt"
i="school"
cat > $file <<EOF
I am going to $i
EOF
cat test1.txt

[root@zy2 ~]# sh mjh.sh
I am going to school

#整体赋值给变量，然后通过 echo 命令将变量值打印出来
#!/bin/bash
var="Great! I am going to school!"
myvar=$(cat <<EOF			 
This is Line 1.
Today is Monday.
$var
EOF
)
echo "$myvar"

[root@zy2 ~]# sh mjh.sh
this is line 1.
today is Monday.
Great! I am going to school!

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

#!/bin/bash
var="Great! I am going to school!"
myvar=$(cat <<'EOF'   #对标记加单引号，即可关闭变量替换
this is line 1.
today is Monday.
$var
EOF    #EOF作为结束 不用加单引号
)
echo "$myvar"

[root@zy2 ~]# sh mjh.sh
this is line 1.
today is Monday.
$var

去掉每行之前的 TAB 字符     #在标记前加“-”，即可抑制各行首TAB，对空格无效

多行注释

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

#!/bin/bash
var="Great! I am going to school!"
: cat <<EOF   #多行注释以":"开头的标记内容不会被执行
this is line 1.
today is Monday.
$var
EOF
echo "$var"   


[root@zy2 ~]# sh mjh.sh
Great! I am going to school!

[root@zy2 ~]# : <<EOF
> 123
> 456
> 789
> EOF
[root@zy2 ~]#    #没有任何结果，还是用'#'注释直观

Expect

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

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

常见的转义字符及其用法：
\n：表示换行符
\t：表示制表符
\r：表示回车符
\b：表示退格符

基本命令:

1）脚本解释器
expect 脚本中首先引入文件，表明使用的是哪一个 shell。
#!/usr/bin/expect
chmod 777 xxx.sh #赋权  
./   #执行脚本

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

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后添加exit，并不会退出root用户。
而如果没有interact则登录完成后会退出，而不是留在远程终端上。
使用interact会保持在终端而不会退回到原终端，
比如切换到root用户，会一直在root用户状态下；
比如ssh到另一服务器，会一直在目标服务器终端，而不会切回的原服务器。
#一句话：也就是留在当前终端用户

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

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

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 $argv 1]        相当于 password=$2

注意：expect脚本不能通过bash、source、. 来执行（因为这三种方式是调用shell解释器），只能通过绝对路径或相对路径来执行

实例操作一：免交互预设值修改用户密码

#!/usr/bin/expect
set timeout 5
spawn passwd aa
expect "New*"
send "123456\r"
expect "Retype*"
send "123456\r"
expect eof


[root@zy2 ~]# chmod 777 passwd.sh 
[root@zy2 ~]# ./passwd.sh 
spawn passwd aa
Changing password for user aa.
New password: 
BAD PASSWORD: The password is shorter than 8 characters
Retype new password: 
passwd: all authentication tokens updated successfully.

su切换用户

#!/usr/bin/expect
#设置超时时间
set timeout 5
#参数传入
set username [lindex $argv 0]  #相当于username=$1
set password [lindex $argv 1]  #相当于password=$2
#开始追踪命令
spawn su - $username
#免交互执行，捕捉信息并匹配
expect "密码"
send "$password\r"
expect "*]#"
send_user "ok"
#把控制权交给控制台
interact

嵌入执行模式，将 expect 过程融入 Shell 当中，方便执行和处理。
像ssh、su这种会切换环境的交互式命令一般不建议使用嵌入执行模式。
 

创建用户并设置密码

#!/bin/bash
user=$1
password=$2
useradd $user
/usr/bin/expect <<-EOF
spawn passwd $user
expect "New*"
send "${password}\r"
expect "Retype*"
send "${password}\r"
expect eof
EOF

[root@zy2 ~]# sh passwd.sh ff 123456
useradd: user 'ff' already exists
spawn passwd ff
Changing password for user ff.
New password: 
BAD PASSWORD: The password is shorter than 8 characters
Retype new password: 
passwd: all authentication tokens updated successfully.

实现 ssh 自动登录

#!/usr/bin/expect
set ip 192.168.126.23
set user root
set password 123456
set timeout 5
spawn ssh $user@$ip
expect {
"yes/no" {send "yes\r";exp_continue}
"password" {send "${password}\r"}
}
expect eof

[root@zy1 ~]# ./ssh.sh 192.168.126.23 123456
spawn ssh root@192.168.126.23
root@192.168.126.23's password: 
Last login: Tue Mar 28 19:42:18 2023 from 192.168.126.21
[root@zy3 ~]# [root@zy1 ~]# 
[root@zy1 ~]#

#传参的方式：

#!/usr/bin/expect
set timeout 5
set hostname [lindex $argv 0]
set password [lindex $argv 1]
spawn ssh $hostname
expect {
	"Connection refused" {send_user "ssh访问被拒绝\n"}     #连接失败情况，比如对方ssh服务关闭
	"No route to host" {send_user "主机名/IP有误\n"}		#找不到服务器，比如输入的IP地址不正确
	"(yes/no)" {send "yes\r";exp_continue}
	"password:" {send "$password\r"}
}

interact

用免交互的方式给硬盘分区，格式化，挂载

#!/bin/bash
disk=$1  
/usr/bin/expect <<-EOF
spawn fdisk $disk
expect "Command" {send "n\r"}
expect "Select" {send "\r"}
expect "Part*" {send "\r"}
expect "First" {send "\r"}
expect "Last" {send "\r"}
expect "Command：" {send "w\r"}
expect eof
EOF
partprobe
mkfs.xfs ${disk}1 -f &> /dev/null
if [ $? -eq 0 ]
then
echo "磁盘格式化完成"
mkdir /opt/disk
mount ${disk}1 /opt/disk
df -h
else
echo "格式化失败，脚本出错！"
fi