shell之免交互

CAN
346 阅读5分钟

1. Here Document 免交互

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

是标准输入的一种替代品可以帮助脚本开发人员不必使用临时文件来构建输入信息，而是直接就地生产出一个“文件”并用作“命令”的标准输入。HereDocument

也可以与非交互式程序和命令一 起使用。

1.1 语法格式

命令 <<标记

内容 #标记之间是传入内容

标记

1.2 注意事项

  1. 标记可以使用任意合法字符(通常为EOF )
  2. 结尾的标记一定要顶格写，前面不能有任何字符
  3. 结尾的标记后面也不能有任何字符(包括空格)
  4. 开头标记前后的空格会被省略掉
  5. 在起始标记字符前加-会抑制每行行首tab键的输出
  6. 起始标记字符用单引号括起来的话，里面的$将被视为普通字符，不能转换变量的值

1.3 示例

  • 免交互方式实现对行数的统计，将要统计的内容置于标记“EOF"之间，直接将内容传给wc -1来统计
[root@localhost ~]# wc -l <<EOF
>Linel
>Line2
>EOF

image.png

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

image.png

  • 通过passwd给用户设置密码
[root@localhost ~]# passwd lisi <<EOF
>abc1234
>abc1234   //这两行是输入的密码和确认密码
>EOF

image.png

1.4 支持变量替换

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

image.png

  • 整体赋值给变量，然后通过echo 命令将变量值打印出来
#!/bin/bash
var="Great! I am going to school!"
myvar=$(cat <<EOF
This is Line 1.
Today is Monday.
$var
EOF
)
echo "$myvar"  //输出字符串变量的时候，要加双引号，不然可能会不换行输出

image.png

  • 多行注释

Bash的默认注释是#， 该注释方法只支持单行注释; Here Document 的引入解决了移行注释的问题。

:代表什么都不做的空命令。中间标记区域的内容不会被执行，会被bash忽略掉，因此可达到批量注释的效果。


#!/bin/bash
var="Great! I am going to school!"
: <<-EOF  //多行注释，“:"开头的Here Document 标记内容不会被执行
This is Line 1.
Today is Monday.
file="can"
EOF
echo "abcd"
echo $file

image.png

2. Expect

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

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

2.1 基本命令

2.1.1 脚本解释器

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

#!/usr/bin/expect

2.1.2 spawn

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

  • 例
spawn passwd root

2.1.3 expect

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

  2. 只能捕捉由spawn启动的进程的输出;

  3. 用于接收命令执行后的输出，然后和期望的字符串匹配

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

2.1.5 结束符

  • expect eof

表示交互结束，等待执行结束，退回到原用户，与spawn对应。

比如切换到root用户，expect脚本默认的是等待10s， 当执行完命令后，默认停留10s后，自动切回了原用户

  • interact

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

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

2.1.6 set

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

例: set timeout 30

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

2.1.8 send_user

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

2.1.9 接收参数

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

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

expect直接执行，需要使用expect命令去执行脚本

su切换用户

#!/usr/bin/expect
#设置超时时间
set t imeout 5
#参数传入
set username [lindex $argv 0]
set password [lindex $argv 1]
#开始追踪命令

3. 综合示例

3.1 fdisk分区免交互

#!/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 "命令" {send "w\r"}
expect eof
EOF
mkfs.xfs $disk -f &> /dev/null
if [ $? -eq 0 ];then
  mount $disk /opt
  df -h
else
  echo "格式化失败，脚本有问题"
fi

image.png

3.2 ssh登录主机免交互

#!/usr/bin/expect
spawn ssh 172.16.100.100
expect {
 "(yes/no)" {send "yes\n";exp_continue}
 "password" {send "123456\r";exp_continue}
}
interact

image.png

3.3 su切换用户免交互

#!/usr/bin/expect
set username [lindex $argv 0]  //设置变量名，变量值来自位置变量1
set password [lindex $argv 1]
spawn su $username
expect "密码："
send "$password\r"
expect "*]#"
send_user "切换${username}成功"
interact  //interact结束符，会把控制权交给控制台
exit
#expect eof  //expect eof结束时，会默认等待10秒，然后会自动切回原用户
[can@localhost ~]$ ./su.sh root 123456

image.png

3.4 passwd修改密码

#!/usr/bin/expect
set timeout 5  //设置expect等待的超时时间，默认是10s
spawn passwd zhangsan  //先用spawn开启对一个有交互过程命令进程的跟踪
expect "新的 密码" {send "abc1234\r"}  //用expect捕捉系统的交互提示信息。第一次匹配
expect "重新输入新的 密码" {send "abc1234\r"}
expect eof

image.png

avatar
文章
阅读
粉丝