自動交互式腳本--expect

   咱們常常會遇到一些須要與服務器程序打交道的場景,好比,從登錄某個服務器,而後進行某項工做。這很日常,可是若是把這個工做自動化進行,你就須要一個程序能自動作你要告訴機器的事情,這樣,咱們的expect就能大顯身手了。html

   首先,expect是一個簡單的工具語言,如要工做就是進行自動化的人機交互。它的做者對Expect的定義:是一個實現自動交互功能的軟件套件(a software suite for automating interactive tools),使用expect,它能幫助咱們在合適的情景下進行合適的交互。linux

      例子:咱們要在凌晨登錄到一個linux服務器:192.168.1.1,而後執行一個命令作/home/test/a.sh,咱們的思路是:寫一個expect腳本,而後用crontab在凌晨運行。web

    #!/usr/bin/expect -f
    set timeout 30
    spawn ssh -l test 192.168.1.1
    expect "password:"
    send "mypassword\r"
    expect "~$*"
    send "/home/test/a.sh\r"
    send "exit\n"
    expect eof
    exit

 expect 的核心功能,對於設定好的特定匹配形式,以相匹配的動做以應對。每個expect後所跟的字符串(或者正則表達式)就是腳本所等待的匹配模式,每個send 所作的工做就是對於各類的模式串,實施相應的動做。正則表達式

    第一行設定了腳本執行的程序,-f選項指的是expect執行一個文件shell

    第二行,設定了本腳本全部的超時時間,單位是秒(s),若是超時,腳本將繼續向下進行(好比在等待某個模式出現,超時之後,會進行下一語句)。windows

    第三行,expect使用spawn命令來啓動腳本和命令會話,這裏啓動的是ssh命令,這裏的ssh命令將會以子進程的方式產生。數組

    下面就是交互的過程:ssh -l 登錄之後,會給要求客戶寫入密碼,因此等待出現「password:」,出現password:之後,須要寫入密碼,注意這裏須要送去回車或者換行符,不然遠端主機不會收到ssh請求的。登錄上系統以後,會出現命令提示符:~$,即系統已經登錄到了遠端主機的shell,而後送去要執行的命令。完畢後推出遠程機器(這個send "exit\r"前也能夠有上一個命令的輸出,也能夠沒有,由於上一個命令執行完畢後會順序執行下一條)。bash

    最後是等待標示子進程已結束的標示符eof,而後退出。(注:這個等待eof必需要有,若是沒有eof,極可能在子進程沒有結束前就退出,形成問題。)服務器

    下面的例子,介紹expect其餘重要的命令,先看以下例子:ssh

    #!/usr/bin/expect -f
    set timeout 10
    set myname [lindex $argv 0]
    spawn ./talk.sh
    expect "Name:"
    send "$myname\r"
    expect eof

 這個例子引入了輸入參數的概念,就是在執行這個expect腳本時候,須要帶入參數,好比這個腳本名字若是爲test.ep

    執行時須要鍵入參數:./test.ep "John"

    set myname [lindex $argv 0] 這句獲取外部傳入的第一個參數(argv 0)並傳給變量myname,若是獲取多個參數則使用argv 1,argv 2,以此類推。

    另外,expect支持通常語言所經常使用的if,for等流程控制語句,這個能夠參看expect高級介紹

小結:

    expect是一款很是好用的自動化交互工具

核心命令:

    spawn: 啓動一個命令或程序,並由expect程序開始監聽

    set :設置變量值和名稱

    set .. lindex:從程序輸入參數出獲取變量值並賦給變量

    expect ...send對:expect等待但願出現的匹配串,對於匹配到的串,send發送命令進行執行。

如何向expect腳本里面傳遞參數

好比下面腳本用來作ssh無密碼登錄,自動輸入確認yes和密碼信息,用戶名,密碼,hostname經過參數來傳遞
 
ssh.exp
Python代碼  
#!/usr/bin/expect  
set timeout 10  
set username [lindex $argv 0]  
set password [lindex $argv 1]  
set hostname [lindex $argv 2]  
spawn ssh-copy-id -i .ssh/id_rsa.pub $username@$hostname  
expect "yes/no"  
send "yes\r"  
expect "password:"  
send "$password\r"   
expect eof   
執行腳本./ssh.exp root pasword hostname1
expect接收參數的方式和bash腳本的方式不太同樣,bash是經過$0 ... $n 這種方式,而expect是經過set <變量名稱> [lindex $argv <param index>],例如set username [lindex $argv 0]
 

expect用法

1. [#!/usr/bin/expect] 

這一行告訴操做系統腳本里的代碼使用那一個shell來執行。這裏的expect其實和linux下的bash、windows下的cmd是一類東西。 

注意:這一行須要在腳本的第一行。 

2. [set timeout 30] 

基本上認識英文的都知道這是設置超時時間的,如今你只要記住他的計時單位是:秒   。timeout -1 爲永不超時

3. [spawn ssh -l username 192.168.1.1] 

spawn是進入expect環境後才能夠執行的expect內部命令,若是沒有裝expect或者直接在默認的SHELL下執行是找 不到spawn命令的。因此不要用 「which spawn「之類的命令去找spawn命令。比如windows裏的dir就是一個內部命令,這個命令由shell自帶,你沒法找到一個dir.com 或 dir.exe 的可執行文件。 

它主要的功能是給ssh運行進程加個殼,用來傳遞交互指令。 

4. [expect "password:"] 

這裏的expect也是expect的一個內部命令,有點暈吧,expect的shell命令和內部命令是同樣的,但不是一個功能,習 慣就行了。這個命令的意思是判斷上次輸出結果裏是否包含「password:」的字符串,若是有則當即返回,不然就等待一段時間後返回,這裏等待時長就是 前面設置的30秒 

5. [send "ispass\r"] 

這裏就是執行交互動做,與手工輸入密碼的動做等效。 

舒適提示: 命令字符串結尾別忘記加上「\r」,若是出現異常等待的狀態能夠覈查一下。 

6. [interact] 

執行完成後保持交互狀態,把控制權交給控制檯,這個時候就能夠手工操做了。若是沒有這一句登陸完成後會退出,而不是留在遠程終端上。若是你只是登陸過去執行 

7.$argv 參數數組

expect腳本能夠接受從bash傳遞過來的參數.可使用[lindex $argv n]得到,n從0開始,分別表示第一個,第二個,第三個....參數

 

下面的expect腳本的例子

執行這個文件./launch.exp 1 2 3

屏幕上就會分別打印出參數

send_user用來發送內容給用戶。

 

參數運用方面還有不少技巧

好比$argc 存儲了參數個數,args被結構化成一個列表存在argv。$argv0 被初始化爲腳本名字。

除此以外,若是你在第一行(#!那行)使用-d (debug參數),能夠在運行的時候輸出一些頗有用的信息

好比你會看見

 

argv[0] = /usr/bin/expect argv[1] = -d argv[2] = ./launch.exp argv[3] = 1 argv[4] = 2 argv[5] = 3

使用這些也能夠完成參數傳遞

8.

expect的命令行參數參考了c語言的,與bash shell有點不同。其中,$argc爲命令行參數的個數,$argv0爲腳本名字自己,$argv爲命令行參數。[lrange $argv 0 0]表示第1個參數,[lrange $argv 0 4]爲第一個到第五個參數。與c語言不同的地方在於,$argv不包含腳本名字自己。

 

9. exp_continue的用法

#!/usr/bin/expect -f

set ipaddr "localhost"

set passwd "iforgot"

spawn ssh root@$ipaddr              #spawn   意思是執行命令,expect內命令,shell中不存在

expect {

"yes/no" { send "yes\r"; exp_continue}

"password:" { send "$passwd\r" }

}

expect "]# "

send "touch a.txt\r"                       #意思爲發送命令

send "exit\r"

expect eof

exit

exp_continue能夠繼續執行下面的匹配,簡單了許多。還有一點,讓我認識到匹配不見得要匹配最後幾個字符。

10.拿來小例子   

設置變量     set PASSWD   abcd123

#!/usr/bin/expect -f

# Expect script to supply root/admin password for remote ssh server

# and execute command.

# This script needs three argument to(s) connect to remote server:

# password = Password of remote UNIX server, for root user.

# ipaddr = IP Addreess of remote UNIX server, no hostname

# scriptname = Path to remote script which will execute on remote server

# If you username and passwd has not pass the rsa trust, your login will fail.

# Usage For example:

#  ./sshlogin.exp password 192.168.1.11 who

# ------------------------------------------------------------------------

# Copyright (c) 2004 nixCraft project <http://cyberciti.biz/fb/>

# This script is licensed under GNU GPL version 2.0 or above

# -------------------------------------------------------------------------

# This script is part of nixCraft shell script collection (NSSC)

# Visit http://bash.cyberciti.biz/ for more information.

# ----------------------------------------------------------------------

# set Variables

set password [lrange $argv 0 0]

set ipaddr [lrange $argv 1 1]

set scriptname [lrange $argv 2 2]

set arg1 [lrange $argv 3 3]

set timeout -1

# now connect to remote UNIX box (ipaddr) with given script to execute

spawn ssh yourusername@$ipaddr $scriptname $arg1

match_max 100000

# Look for passwod prompt

expect "*?assword:*"

# Send password aka $password

send -- "$password\r"

# send blank line (\r) to make sure we get back to gui

send -- "\r"

expect eof
 
#!/usr/bin/expect 

 # 設置超時時間爲 60 秒

 set timeout  60                                         

 # 設置要登陸的主機 IP 地址

 set host 192.168.1.46

 # 設置以什麼名字的用戶登陸

 set name root 

 # 設置用戶名的登陸密碼

 set password 123456 

 

 #spawn 一個 ssh 登陸進程

 spawn  ssh $host -l $name 

 # 等待響應,第一次登陸每每會提示是否永久保存 RSA 到本機的 know hosts 列表中;等到回答後,在提示輸出密碼;以後就直接提示輸入密碼

 expect { 

    "(yes/no)?" { 

        send "yes\n"

        expect "assword:"

        send "$pasword\n"

    } 

        "assword:" { 

        send "$password\n"

    } 

 } 

 expect "#"

 # 下面測試是否登陸到 $host 

 send "uname\n"

 expect "Linux"

 send_user  "Now you can do some operation on this terminal\n"

 # 這裏使用了 interact 命令,使執行完程序後,用戶能夠在 $host 終端進行交互操做。

 Interact 

用expect實現ssh自動登陸對服務器進行批量管理

1.實現ssh自動登陸完成任務的expect腳本

#!/usr/bin/expect -f

set ipaddress [lindex $argv 0]

set passwd [lindex $argv 1]

set timeout 30

spawn ssh shellqun@$ipaddress

expect {

"yes/no" { send "yes\r";exp_continue }

"password:" { send "$passwd\r" }

}

expect "*from*"

send "mkdir -p ./tmp/testfile\r"

#send "exit\r"

expect "#"  命令運行完, 你要期待一個結果, 結果就是返回shell提示符了(是# 或者$)

#最後一句第13行的解釋:

 

其實寫成 interact 的最大好處是登陸後不會退出,而會一直保持會話鏈接,能夠後續手動處理其它任務,請根據實際狀況自行選擇了。

 2.調用login.exp完成批量管理

#!/bin/bash

for i in `awk '{print $1}' passwd.txt`

do

j=`awk -v I="$i" '{if(I==$1)print $2}' passwd.txt`

expect /root/shell/login.exp $i $j

done

linux下的expect的簡單用法及舉例

 一、使用expect前,須要先安裝兩個rpm包,下載:http://download.csdn.net/detail/wang7dao/4416172

# rpm -ihv expect-5.43.0-8.el5.i386.rpm
# rpm -ihv expect-devel-5.43.0-8.el5.i386.rpm

 二、使用腳本文件的例子--實現自動輸密碼

#!/usr/bin/expect -f
set password 123456
#download
spawn scp root@192.168.1.218:/root/a.wmv /home/yangyz/
set timeout 300 
expect "root@192.168.1.218's password:"
set timeout 300 
send "$password\r"
set timeout 300 
send "exit\r"
expect eof   www.2cto.com  

 三、在sh腳本中嵌入expect的例子--經過連上一個公網的服務器再轉跳到一個內網的服務器上,用腳本實現不用輸密碼,直接使用./goto.sh servername

#!/bin/bash
passmsmallq10="a"
passzhsh="a"
passfcwr="b"
passwapfx="c"
passadfx="d"
 
ip1="200.100.10.10"
ip2="10.100.100.70"
ip3="10.100.100.60"
ip4="10.100.100.10"
ip5="10.100.100.20"
 
case $1 in 
"zhsh") passstr=$passzhsh ipstr=$ip2 ;;
"fcwr") passstr=$passfcwr ipstr=$ip3 ;;
"wapfx") passstr=$passwapfx ipstr=$ip4 ;;
"adfx") passstr=$passadfx ipstr=$ip5 ;;
*) echo "The parameter $1 isn't exist"
exit 0 ;;  www.2cto.com  
esac
 

 

command1="ssh -l m_smallq -p 36000 $ip1"
command2="ssh -l mqq -p 36000 $ipstr"
 
expect -c "
        set timeout 60;
        spawn $command1;
        expect {
                \"221.130.15.10's password:\" {send \"$passmsmallq10\r\"; exp_continue}
                \"m_smallq\" {send \"$command2\r\"; exp_continue}
                \"mqq's password:\" {send \"$passstr\r\";interact}
                }
        "
對上面例子的expect的解說
expect -c "..."  --裏面輸入命令
expect {...}     --裏面的多行記錄,從上向下掃描匹配,誰先匹配誰先處理。
  www.2cto.com 

四、ssh到另外一臺機子執行df -h後退出,要點是send後面能夠跟多個命令,經過\r來分行成多個命令

#!/bin/bash
ip1="183.62.178.191"
command1="ssh -l root -p 14322 $ip1"
 
expect -c "
        spawn $command1;
        expect {
                \"183.62.178.191's password:\" {send \"aa\r\"; exp_continue}
                \"root@\" {send \"df -h\r exit\r\"; exp_continue}
                }
        "
相關文章
相關標籤/搜索