这里包括Expect , array
目录
简介
先抛出一个问题 现在有两台Linux主机A和B,如何从A主机ssh到B主机,然后在B主机上执行命令,如何使这个过程实现全程自动化?你可能会使用这种方法:
ssh admin@10.222.20.15 “ls”
但是这种方式比较笨拙,每次都要输入密码,同时并不能执行一些复杂的逻辑或命令。那么如何实现全程自动化呢?这就要用到今天这篇文章总结的expect了。
expect是什么?
expect是一个免费的编程工具,用来实现自动的交互式任务,而无需人为干预。说白了,expect就是一套用来实现自动交互功能的软件。
在实际工作中,我们运行命令、脚本或程序时,这些命令、脚本或程序都需要从终端输入某些继续运行的指令,而这些输入都需要人为的手工进行。而利用expect,则可以根据程序的提示,模拟标准输入提供给程序,从而实现自动化交互执行。这就是expect!!!
expect基础
在使用expect时,基本上都是和以下四个命令打交道:
命令 作用
- send 用于向进程发送字符串
- expect 从进程接收字符串
- spawn 启动新的进程
- interact 允许用户交互
send命令接收一个字符串参数,并将该参数发送到进程。
expect命令和send命令相反,expect通常用来等待一个进程的反馈,我们根据进程的反馈,再发送对应的交互命令。
spawn命令用来启动新的进程,spawn后的send和expect命令都是和使用spawn打开的进程进行交互。
interact命令用的其实不是很多,一般情况下使用spawn、send和expect命令就可以很好的完成我们的任务;但在一些特殊场合下还是需要使用interact命令的,interact命令主要用于退出自动化,进入人工交互。比如我们使用spawn、send和expect命令完成了ftp登陆主机,执行下载文件任务,但是我们希望在文件下载结束以后,仍然可以停留在ftp命令行状态,以便手动的执行后续命令,此时使用interact命令就可以很好的完成这个任务。
实用代码分析
上面对expect进行了总结,特别是对一些常用的命令进行了详细的说明。下面就通过一些常用的expect脚本来具体的说明如何使用expect来完成日常的一些工作。
#!/usr/bin/expect
#显示赋值
set timeout 30
set host "101.200.241.109"
set username "root"
set password "123456"
spawn ssh $username@$host #启动一个会话,spawn:启动
spawn scp -r /etc/file $username$ip:/tmp #启动一个会话,并且赋值文件spawn:启动
expect {
"yes/no" { send "yes\r";exp_continue } #没有继续执行
"*password*" {send "$password\r"}
}
interact #停在对方那边
这是一段非常简单的expect示例代码,演示了expect的基本使用方法。
\#!/usr/tcl/bin/expect:
使用expect来解释该脚本;
set timeout 30:
设置超时时间,单位为秒,默认情况下是10秒;
set host "101.200.241.109":
设置变量;
spawn ssh \$username@\$host:spawn
是进入expect环境后才可以执行的expect内部命令,如果没有装expect或者直接在默认的SHELL下执行是找不到spawn命令的。它主要的功能是给ssh运行进程加个壳,用来传递交互指令;
expect "*password*":
这里的expect也是expect的一个内部命令,这个命令的意思是判断上次输出结果里是否包含“password”的字符串,如果有则立即返回;否则就等待一段时间后返回,这里等待时长就是前面设置的30秒;
send "\$password\r":
当匹配到对应的输出结果时,就发送密码到打开的ssh进程,执行交互动作;
interact:
执行完成后保持交互状态,把控制权交给控制台,这个时候就可以手工操作了。如果没有这一句登录完成后会退出,而不是留在远程终端上。
这就是对上述这段简单简单脚本的分析,在上述的示例中,涉及到expect中一个非常重要的概念——模式-动作;即上述expect “password” {send “$password\r”}这句代码表达出来的含义。
模式-动作
结合着expect “*password*” {send “$password\r”}这句代码来说说“模式-动作”。简单的说就是匹配到一个模式,就执行对应的动作;匹配到password字符串,就输入密码。你可能也会看到这样的代码:
expect {
"password" {
send "$password\r"
exp_continue
}
eof
{
send "eof"
}
}
#!/usr/bin/bash
# 读取user.txt文件,创建账号,其中line是读的一行,因为read是以换行符为结束
for i in {2...254} #or for i in `cat ip.txt` #这里要执行cat ip.txt,所以要执行``或者${
do
{
ip=192.168.0.$i
ping -c1 -W.1 $ip&> /dev/null #-c1 ping一次,-W.1等待0.1s
if [ $? -eq 0 ]
then
echo "$ip is up" >> -a ip.txt #usr tee type creat a file to record ping history
#这里千万不要有空格,否则是找死
/usr/bin/expect <<-EOF
set timeout 10 #设置执行超时
expect {
"yes/no" { send "yes\r";exp_continue } #没有继续执行
"*password*" {send "$password\r"}
}
expect eof
EOF
else
echo "$ip is down"
fi
}& # 在(子shell)后台执行,相当于并发执行 多进程
done
wait #等待前面所有后台进程结束
echo "finish"
其中exp_continue表示循环式匹配,通常匹配之后都会退出语句,但如果有exp_continue则可以不断循环匹配,输入多条命令,简化写法。
传参
很多时候,我们需要传递参数到脚本中,现在通过下面这段代码来看看如何在expect中使用参数:
#!/usr/tcl/bin/expect
if {$argc < 3} {
puts "Usage:cmd <host> <username> <password>"
exit 1
}
set timeout -1
set host [lindex $argv 0]
set username [lindex $argv 1]
set password [lindex $argv 2]
spawn ssh $username@$host
expect "*password*" {send "$password\r"}
#interact #不停留在那边
expect "#"
send "useradd yangyang\r"
expect eof #结束expect
在expect中,$argc表示参数个数,而参数值存放在$argv中,比如取第一个参数就是[lindex $argv 0],以此类推。
总结1
能够在工作中熟练的使用Shell脚本就可以很大程度的提高工作效率,如果再搭配上expect,那么很多工作都可以自动化进行,对工作的展开如虎添翼。如果你会Python的话,你的视野将会更加开阔,那个时候你又会“嫌弃”expect了。
数组操作
一般 name=qwe 变量赋值,索引0:q,1:w,2:e
arr=(“one” “two” [123]=456) 普通数组,相当于python中的list,从0开始,其中从123跳着赋值
索引下标: 0:”one”
arr=([name]=qwe [sex]=male [age]=36)关联数组,相当于python中的dict(字典)
索引下标 name:qwe sex: male age: 36
查看数组
declare -a 查看所有数组
- 赋值
arr[0]=10 #重新覆盖
arr[0]=(10 [20] 1) #重新覆盖
echo ${arr[@]} #查看所有数组 # @等价于*
echo #{arr[@]} #查看数组个数 # @等价于*
echo ${!arr[@]} #获得数组的索引 2. 取值
oneTemp=${arr[0]}
- 获得数组的所有元素,然后for循环遍历
for x in ${arr[]} # @等价于
do
echo number $x
done
- 获得数组长度
length=${#arr[]} # @等价于
- 取得数组单个元素的长度(其实就是字符串长度)
lengthn=${#arr[n]}
str=“i am rrr”
echo ${#str}
- 数组删除
unset arr[1] # 删除数组中第一个元素
unset arr # 删除整个数组
- 数组截取
${arr[*]:1} # two three four,第一个元素后所有元素
${arr[*]:0}表示所有元素
${arr[*]:0:2} # one two
${arr[*]:1:2} # two 123
- 使用IFS将字符串转化为数组
*** IFS是内部字段分隔符(internal field separator)***
#!/bin/bash
# 情况1、符合默认IFS
ips="ip1 ip2 ip3 ip6"
# 直接加上()即可,因为默认的IFS就是以' ' 切分的
ipArr=($ips)
for ip in ${ipArr[*]}
do
echo $ip
done
# 情况2、其他分隔符
ips="ip1,ip2,ip3,ip6"
# 方法1、临时修改IFS,最后还原IFS,避免影响
oldIFS=$IFS
IFS=","
ipArr=($ips)
IFS=$oldIFS
for ip in ${ipArr[*]}
do
echo $ip
done
# 方法2、使用字符串替换功能,将字符串分隔符替换为' '。 //表示全部替换
ipArr=(${ips//,/' '})
for ip in ${ipArr[*]}
do
echo $ip
done
实例
hosts文件
127.0.0.1 localhost
::1 localhose
192.168.122.47 localhost
shell1.sh
#!/usr/bin/bash
while read line
do
host[++i]=$line
done </etc/hosts
echo "hosts first: ${hosts[1]}"
echo
for i in ${!hosts[i]}
do
echo "$i: ${hosts[i]}"
done
shell2.sh
#!/usr/bin/bash
#同上方法差不多,但是因为for的分隔符与while不通,则输出不通,这里改进分隔符
#OLD_IFS=$IFS
#IFS=$'\n'
#IFS=$OLD_IFS #在最后可以恢复原有换行符
for line in `cat /etc/hosts`
do
host[++i]=$line
done </etc/hosts
echo "hosts first: ${hosts[1]}"
echo
for i in ${!hosts[@]}
do
echo "$i: ${hosts[i]}"
done
shell3.sh 统计元素-性别
sex.txt 文件
wang m
wu m
tom f
#!/usr/bin/bash
# 声明一个数组
declare -A sex
while read line
do
#读行取第二列,取
type=`echo $line |awk '{print $2}'`
#host[++i]=$sex
#sex+=([f]=1)#是元素加一,添加一个新的元素
#是某一个索引对应的值加一
let sex[$type]++
done <sex.txt
echo
# 无论什么数组,都可通过索引遍历,要统计对象作为数组的索引,然后++,就可以打印出来
for i in ${!sex[@]}
do
echo "$i: ${sex[$i]}"
done
#输出
#m:2
#f:1
shell4.sh 统计对象作为索引,统计shell的数量
#!/usr/bin/bash
# awk -F":" '{print$NF}' /etc/passwd |sort |uniq -c
# 可以直接通过以上命令进行统计,或者通过以下命令
# 声明一个数组
declare -A shells
while read line
do
#读行,截取,-F 冒号分割,为最后一列 7或者NF
type=`echo $line |awk -F ":" '{print $7}'`
#host[++i]=$sex
#sex+=([f]=1)#是元素加一,添加一个新的元素
#是某一个索引对应的值加一
let shells[$type]++
done </etc/passwd
echo
# 无论什么数组,都可通过索引遍历,要统计对象作为数组的索引,然后++,就可以打印出来
for i in ${!shells[@]}
do
echo "$i: ${shells[$i]}"
done
#输出
#shell:2
#bash:1
shell5.sh 统计对象作为索引,统计tcp连接的数量
#!/usr/bin/bash
# count tcp status
# 通过 watch -n1 ./shell5.sh脚本 #循环一秒钟执行一次
# 声明一个数组
#
#***这里如果这样写,要对数组从新赋值才行,要不然数组会一直累加的,所以放到里面,每次执行完都清零***
#unset status
#declare -A status
while :
do
#为了防止累加,所以只能这样写
unset status
declare -A status
type=`ss -an |grep :80 |awk '{print$2}'`
for j in $type
do
#读行,截取,-F 冒号分割,为最后一列 7或者NF
#type=`echo $line |awk -F ":" '{print $7}'`
#host[++i]=$sex
#sex+=([f]=1)#是元素加一,添加一个新的元素
#是某一个索引对应的值加一
let status[$j]++
done
echo
# 无论什么数组,都可通过索引遍历,要统计对象作为数组的索引,然后++,就可以打印出来
for i in ${!status[@]}
do
echo "$i: ${status[$i]}"
done
sleep 1
clear
done
#输出
#wating:2
#listen:1