expect免交互

shell编程中可能碰到需要交互的场景，一般有两种方式：

1、 <<EOF  XXXX    EOF 方式

2、expect

第一种方式适合于简单的固定输入方式，比如输入 y 或 n。

 

localhost:/test # cat a
a
localhost:/test # cat /root/a
b
localhost:/test # cp a /root/a
localhost:/test # echo b > /root/a
localhost:/test # cat /root/a
b
localhost:/test # cat a
a
localhost:/test # cp -i a /root/a
cp: overwrite `/root/a'?
localhost:/test # cat /root/a
b
localhost:/test # cp -i a /root/a <<EOF
> y
> EOF
cp: overwrite `/root/a'? localhost:/test # cat /root/a
a
localhost:/test #

 

第二种方式功能比较强大，但是需要 expect 软件支持。

下面主要说第二种方式：

 

Expect是一个免费的编程工具语言，用来实现自动和交互式任务进行通信，而无需人的干预。
CentOS安装：yum install expect
Ubuntu安装：sudo apt-get install expect
1>.免交互登陆，查看远程主机磁盘分区
#!/usr/bin/expect
set ip 192.168.1.156
set pass 123.com
set timeout 30
spawn ssh root@$ip
expect {
"(yes/no)" {send "yes\r"; exp_continue}
"password:" {send "$pass\r"}
}
expect "root@*" {send "df -h\r"}
expect "root@*" {send "exit\r"}
expect eof

2>.在Shell脚本中嵌入Expect语法
方法1：使用EOF，将内容段让expect执行
#!/bin/bash
user=root
pass='123'
ip='192.168.1.154'
/usr/bin/expect << EOF
set timeout 30
spawn ssh $user@$ip
expect {
"(yes/no)" {send "yes\r"; exp_continue}
"password:" {send "$pass\r"}
}
expect "root@*" {send "df -h\r"}
expect "root@*" {send "exit\r"}
expect eof
EOF

#!/bin/bash
user=root
pass='123'
ip='192.168.1.154'
expect -c "
spawn ssh $user@$ip
expect {
\"(yes/no)\" {send \"yes\r\"; exp_continue}
\"password exp_continue}
\"root@*\" {send \"df -h\r exit\r\"; exp_continue}
}"

方法2：将expect脚本独立出来
# vi login.exp #免交互登陆脚本
#!/usr/bin/expect
set ipaddress [lindex $argv 0]
set username [lindex $argv 1]
set password [lindex $argv 2]
if { $argc != 3 } {
puts "Usage: expect login.exp ipaddress username password"
exit 1
}
set timeout 30
spawn ssh $username@$ipaddress
expect {
"(yes/no)" {send "yes\r"; exp_continue}
"password:" {send "$password\r"}
}
expect "$username@*" {send "df -h\r"}
expect "$username@*" {send "exit\r"}
expect eof

# vi user_info #用户信息文件
192.168.1.156 user user
192.168.1.154 root 123.com
# vi expect.sh #读取用户信息并赋值到变量
#!/bin/bash
for ip in `awk '{print $1}' user_info`
do
user=`awk -v I="$ip" '{if(I==$1)print $2}' user_info`
pass=`awk -v I="$ip" '{if(I==$1)print $3}' user_info`
expect login.exp $ip $user $pass
done

普通自动切换到root脚本，并且在shell中嵌入expect
#!/bin/bash
/usr/bin/expect << EOF
spawn /bin/su -
expect "Password: "
send "123\r"
expect "*#"
interact
expect eof
EOF
或者单独写一个脚本，实现普通用户自动切换到root用户，去掉上面脚本中的#!/bin/bash，和EOF，然后把这个脚本放在/usr/bin目录中，把脚本名改为root，然后执行root命令就可以实现切换。

参数说明：
set：可以设置超时，也可以设置变量
timeout：expect超时等待时间，默认10S
spawn：执行一个命令
expect ""：匹配输出的内容
exp_continue：继续执行下面匹配
\r：可以理解为回车
$argc：统计位置参数数量
[lindex $argv 0]：脚本后第一个参数，类似于shell中$1，以此类推
puts：打印字符串，类似于echo
awk -v I="$ip"：赋值变量
expect{...}：输入多行记录

其他参数说明：
timeout -1：永不超时退出
log_file /var/log/expect.log：记录交互信息，一般crontab时使用
interact：交互后不退出远程终端，如果加要把expect "root@*" {send "exit\r"}注释掉，如果不加，就直接退出

将spawn ssh root@$ip换成spawn ssh -o StrictHostKeyChecking=no root@ip既不会再提示是否将服务器计算机密钥加入本地known_hosts