6.1 信号捕捉 trap

trap '触发指令' 信号

进程收到系统发出的指定信号后，将执行自定义指令，而不会执行原操作

trap '' 信号

忽略信号的操作

trap '-' 信号

恢复原信号的操作

trap -p

列出自定义信号操作

trap fifinish EXIT

当脚本退出时，执行fifinish函数

范例：

#!/bin/bash
trap 'echo "Press ctrl+c"' int quit
trap -p
for((i=0;i<=10;i++))
do
       sleep 1
       echo $i
done
trap '' int 
trap -p
for((i=11;i<=20;i++))
do
       sleep 1
       echo $i
done
trap '-' int 
trap -p
for((i=21;i<=30;i++))
do
       sleep 1
       echo $i
done  

范例:

[root@centos8 script40]#cat trap_exit.sh 
#!/bin/bash
#
#********************************************************************
#Author: wangxiaochun
#QQ: 29308620
#Date: 2020-01-06
#FileName： trap_exit.sh
#URL: http://www.magedu.com
#Description： The test script
#Copyright (C): 2020 All rights reserved
#********************************************************************
finish(){
    echo finish| tee -a /root/finish.log
}
trap finish exit
while true ;do
    echo running
    sleep 1
done

6.2 创建临时文件 mktemp

mktemp 命令用于创建并显示临时文件，可避免冲突

格式

mktemp [OPTION]... [TEMPLATE]

说明：TEMPLATE: fifilenameXXX，X至少要出现三个

常见选项：

-d 创建临时目录

-p DIR或--tmpdir=DIR 指明临时文件所存放目录位置

范例：

mktemp /tmp/testXXX
tmpdir=`mktemp –d /tmp/testdirXXX`
mktemp --tmpdir=/testdir testXXXXXX

6.3 安装复制文件 install

install命令格式：

install [OPTION]... [-T] SOURCE DEST 单文件
install [OPTION]... SOURCE... DIRECTORY
install [OPTION]... -t DIRECTORY SOURCE...
install [OPTION]... -d DIRECTORY...创建空目录

选项：

-m MODE，默认755

-o OWNER

-g GROUP

范例：

install -m 700 -o wang -g admins srcfile desfile
install –m 770 –d /testdir/installdir

6.4 交互式转化批处理工具 expect

expect 是由Don Libes基于Tcl（ Tool Command Language ）语言开发的，主要应用于自动化交互式操作的场景，借助 expect 处理交互的命令，可以将交互过程如：ssh登录，ftp登录等写在一个脚本上，使之自动化完成。尤其适用于需要对多台服务器执行相同操作的环境中，可以大大提高系统管理人员的工作效率

expect 语法：

expect [选项] [ -c cmds ] [ [ -[f|b] ] cmdfile ] [ args ]

常见选项：

-c：从命令行执行expect脚本，默认expect是交互地执行的

-d：可以输出输出调试信息

示例：

expect -c 'expect "\n" {send "pressed enter\n"}'
expect  -d ssh.exp

expect中相关命令

• spawn 启动新的进程
• expect 从进程接收字符串
• send 用于向进程发送字符串
• interact 允许用户交互
• exp_continue 匹配多个字符串在执行动作后加此命令

expect最常用的语法(tcl语言:模式-动作)

单一分支模式语法：

[root@centos8 test]#expect
expect1.1> expect "hi" {send "You said hi\n"}
hahahixixi
You said hi

匹配到hi后，会输出“you said hi”，并换行

多分支模式语法：

[root@centos8 test]#expect
expect1.1> expect "hi" { send "You said hi\n" } "hehe" { send "Hehe yourself\n" 
} "bye" { send "Good bye\n" }
hehe
Hehe yourself
expect1.2> expect "hi" { send "You said hi\n" } "hehe" { send "Hehe yourself\n" 
} "bye" { send "Good bye\n" }
bye
Good bye
expect1.3> expect "hi" { send "You said hi\n" } "hehe" { send "Hehe yourself\n" 
} "bye" { send "Good bye\n" }
hi
You said hi
expect1.4>

匹配hi,hello,bye任意字符串时，执行相应输出。等同如下：

expect {
     "hi" { send "You said hi\n"}
     "hehe" { send "Hehe yourself\n"}
     "bye" { send " Good bye\n"}
}

范例1：

#!/usr/bin/expect
spawn scp /etc/fstab 10.0.0.7:/data
expect {
       "yes/no" { send "yes\n";exp_continue }
       "password" { send "magedu\n" }
}
expect eof

范例2：

#!/usr/bin/expect
spawn ssh 10.0.0.7
expect {
       "yes/no" { send "yes\n";exp_continue }
       "password" { send "magedu\n" }
}
interact

范例3：expect 变量

#!/usr/bin/expect
set ip 10.0.0.7
set user root
set password magedu
set timeout 10
spawn ssh $user@$ip
expect {
       "yes/no" { send "yes\n";exp_continue }
       "password" { send "$password\n" }
}
interact

范例4：expect 位置参数

#!/usr/bin/expect
set ip [lindex $argv 0] 
set user [lindex $argv 1]
set password [lindex $argv 2]
spawn ssh $user@$ip
expect {
       "yes/no" { send "yes\n";exp_continue }
       "password" { send "$password\n" }
}
interact
#./ssh3.exp 192.168.8.10 root magedu

范例5：expect 执行多个命令

#!/usr/bin/expect
set ip [lindex $argv 0] 
set user [lindex $argv 1]
set password [lindex $argv 2]
set timeout 10
spawn ssh $user@$ip
expect {
       "yes/no" { send "yes\n";exp_continue }
       "password" { send "$password\n" }
}
expect "]#" { send "useradd haha\n" }
expect "]#" { send "echo magedu |passwd --stdin haha\n" }
send "exit\n"
expect eof
#./ssh4.exp 10.0.0.7 root magedu

范例6：shell脚本调用expect

#!/bin/bash
ip=$1 
user=$2
password=$3
expect <<EOF
set timeout 20
spawn ssh $user@$ip
expect {
       "yes/no" { send "yes\n";exp_continue }
       "password" { send "$password\n" }
}
expect "]#" { send "useradd hehe\n" }
expect "]#" { send "echo magedu |passwd --stdin hehe\n" }
expect "]#" { send "exit\n" }
expect eof
EOF 
#./ssh5.sh 192.168.8.10 root magedu  

范例7: shell脚本利用循环调用expect在CentOS和Ubuntu上批量创建用户

#!/bin/bash
#
#********************************************************************
#Author: wangxiaochun
#QQ: 29308620
#Date: 2020-01-06
#FileName： expect6.sh
#URL: http://www.magedu.com
#Description： The test script
#Copyright (C): 2020 All rights reserved
#********************************************************************
NET=10.0.0
user=root
password=magedu
for ID in 6 7 111;do
ip=$NET.$ID
expect <<EOF
set timeout 20
spawn ssh $user@$ip
expect {
        "yes/no" { send "yes\n";exp_continue }
        "password" { send "$password\n" }
}
expect "#" { send "useradd test\n" }
expect "#" { send "exit\n" }
expect eof
EOF
done

本文为《企业级shell脚本编程实战》系列教程第六部分，作者为Linux段子手王晓春老师，上一篇见文末扩展链接，后续会分享更多shell脚本编程知识，感兴趣的朋友可以关注下！

【电子版领取见下图！！】