expect使用詳解

linux expect的使用詳解

 

1、概述

　　咱們經過Shell能夠實現簡單的控制流功能，如：循環、判斷等。可是對於須要交互的場合則必須經過人工來干預，有時候咱們可能會須要實現和交互程序如telnet服務器等進行交互的功能。而Expect就使用來實現這種功能的工具。

　　Expect是一個免費的編程工具語言，用來實現自動和交互式任務進行通訊，而無需人的干預。Expect的做者Don Libes在1990年 開始編寫Expect時對Expect作有以下定義：Expect是一個用來實現自動交互功能的軟件套件 (Expect [is a] software suite for automating interactive tools)。使用它系統管理員 的能夠建立腳本用來實現對命令或程序提供輸入，而這些命令和程序是指望從終端（terminal）獲得輸入，通常來講這些輸入都須要手工輸入進行的。 Expect則能夠根據程序的提示模擬標準輸入提供給程序須要的輸入來實現交互程序執行。甚至能夠實現實現簡單的BBS聊天機器人。 :)

　　Expect是不斷髮展的，隨着時間的流逝，其功能愈來愈強大，已經成爲系統管理員的的一個強大助手。Expect須要Tcl編程語言的支持，要在系統上運行Expect必須首先安裝Tcl。

　　2、Expect工做原理

　　從最簡單的層次來講，Expect的工做方式象一個通用化的Chat腳本工具。Chat腳本最先用於UUCP網絡內，以用來實現計算機之間須要創建鏈接時進行特定的登陸會話的自動化。

　　Chat腳本由一系列expect-send對組成：expect等待輸出中輸出特定的字符，一般是一個提示符，而後發送特定的響應。例以下面的 Chat腳本實現等待標準輸出出現Login:字符串，而後發送somebody做爲用戶名；而後等待Password:提示符，併發出響應 sillyme。

　　引用：Login: somebody Password: sillyme

　　這個腳本用來實現一個登陸過程，並用特定的用戶名和密碼實現登陸。

　　Expect最簡單的腳本操做模式本質上和Chat腳本工做模式是同樣的。

　　例子：

　　一、實現功能

　　下面咱們分析一個響應chsh命令的腳本。咱們首先回顧一下這個交互命令的格式。假設咱們要爲用戶chavez改變登陸腳本，要求實現的命令交互過程以下：

　　引用：# chsh chavez

　　Changing the login shell for chavez

　　Enter the new value, or press return for the default

　　Login Shell [/bin/bash]: /bin/tcsh

　　#

　　能夠看到該命令首先輸出若干行提示信息而且提示輸入用戶新的登陸shell。咱們必須在提示信息後面輸入用戶的登陸shell或者直接回車不修改登陸shell。

　　二、下面是一個能用來實現自動執行該命令的Expect腳本：

　　#!/usr/bin/expect

　　# Change a login shell to tcsh

　　set user [lindex $argv 0]

　　spawn chsh $user

　　expect "]:"

　　send "/bin/tcsh "

　　expect eof

　　exit

　　這個簡單的腳本能夠解釋不少Expect程序的特性。和其餘腳本同樣首行指定用來執行該腳本的命令程序，這裏是/usr/bin/expect。程序第一行用來得到腳本的執行參數(其保存在數組$argv中，從0號開始是參數)，並將其保存到變量user中。

　　第二個參數使用Expect的spawn命令來啓動腳本和命令的會話，這裏啓動的是chsh命令，實際上命令是以衍生子進程的方式來運行的。

　　隨後的expect和send命令用來實現交互過程。腳本首先等待輸出中出現]:字符串，一旦在輸出中出現chsh輸出到的特徵字符串(通常特徵 字符串每每是等待輸入的最後的提示符的特徵信息)。對於其餘不匹配的信息則會徹底忽略。當腳本獲得特徵字符串時，expect將發送/bin/tcsh和 一個回車符給chsh命令。最後腳本等待命令退出(chsh結束)，一旦接收到標識子進程已經結束的eof字符，expect腳本也就退出結束。

　　三、決定如何響應

　　管理員每每有這樣的需求，但願根據當前的具體狀況來以不一樣的方式對一個命令進行響應。咱們能夠經過後面的例子看到expect能夠實現很是複雜的條件響應，而僅僅經過簡單的修改預處理腳本就能夠實現。下面的例子是一個更復雜的expect-send例子：

　　expect -re "\[(.*)]:"

　　if {$expect_out(1,string)!="/bin/tcsh"} {

　　send "/bin/tcsh" }

　　send " "

　　expect eof

　　在這個例子中，第一個expect命令如今使用了-re參數，這個參數表示指定的的字符串是一個正則表達式，而不是一個普通的字符串。對於上面這 個例子裏是查找一個左方括號字符(其必須進行三次逃逸(escape)，所以有三個符號，由於它對於expect和正則表達時來講都是特殊字符)後面跟有 零個或多個字符，最後是一個右方括號字符。這裏.*表示表示一個或多個任意字符，將其存放在()中是由於將匹配結果存放在一個變量中以實現隨後的對匹配結 果的訪問。

　　當發現一個匹配則檢查包含在[]中的字符串，查看是否爲/bin/tcsh。若是不是則發送/bin/tcsh給chsh命令做爲輸入，若是是則僅僅發送一個回車符。這個簡單的針對具體狀況發出不一樣相響應的小例子說明了expect的強大功能。

　　在一個正則表達時中，能夠在()中包含若干個部分並經過expect_out數組訪問它們。各個部分在表達式中從左到右進行編碼，從1開始(0包含有整個匹配輸出)。()可能會出現嵌套狀況，這這種狀況下編碼從最內層到最外層來進行的。

　　四、使用超時

　　下一個expect例子中將闡述具備超時功能的提示符函數。這個腳本提示用戶輸入，若是在給定的時間內沒有輸入，則會超時並返回一個默認的響應。這個腳本接收三個參數：提示符字串，默認響應和超時時間(秒)。

　　#!/usr/bin/expect

　　# Prompt function with timeout and default.

　　set prompt [lindex $argv 0]

　　set def [lindex $argv 1]

　　set response $def

　　set tout [lindex $argv 2]

　　腳本的第一部分首先是獲得運行參數並將其保存到內部變量中。

　　send_tty "$prompt: "

　　set timeout $tout

　　expect " " {

　　set raw $expect_out(buffer)

　　# remove final carriage return

　　set response [string trimright "$raw" " "]

　　}

　　if {"$response" == "} {set response $def}

　　send "$response

　　這是腳本其他的內容。能夠看到send_tty命令用來實如今終端上顯示提示符字串和一個冒號及空格。set timeout命令設置後面全部的expect命令的等待響應的超時時間爲$tout(-l參數用來關閉任何超時設置)。

 

 

 

　　而後expect命令就等待輸出中出現回車字符。若是在超時以前獲得回車符，那麼set命令就會將用戶輸入的內容賦值給變臉raw。隨後的命令將用戶輸入內容最後的回車符號去除之後賦值給變量response。

　　而後，若是response中內容爲空則將response值置爲默認值(若是用戶在超時之後沒有輸入或者用戶僅僅輸入了回車符)。最後send命令將response變量的值加上回車符發送給標準輸出。

　　一個有趣的事情是該腳本沒有使用spawn命令。 該expect腳本會與任何調用該腳本的進程交互。

　　若是該腳本名爲prompt，那麼它能夠用在任何C風格的shell中。

　　% set a='prompt "Enter an answer" silence 10'

　　Enter an answer: test

　　% echo Answer was "$a"

　　Answer was test

　　prompt設定的超時爲10秒。若是超時或者用戶僅僅輸入了回車符號，echo命令將輸出

　　Answer was "silence"

　　五、一個更復雜的例子

　　下面咱們將討論一個更加複雜的expect腳本例子，這個腳本使用了一些更復雜的控制結構和不少複雜的交互過程。這個例子用來實現發送write命令給任意的用戶，發送的消息來自於一個文件或者來自於鍵盤輸入。

　　#!/usr/bin/expect

　　# Write to multiple users from a prepared file

　　# or a message input interactively

　　if {$argc<2} {

　　send_user "usage: $argv0 file user1 user2 ... "

　　exit

　　}

　　send_user命令用來顯示使用幫助信息到父進程(通常爲用戶的shell)的標準輸出。

　　set nofile 0

　　# get filename via the Tcl lindex function

　　set file [lindex $argv 0]

　　if {$file=="i"} {

　　set nofile 1

　　} else {

　　# make sure message file exists

　　if {[file isfile $file]!=1} {

　　send_user "$argv0: file $file not found. "

　　exit }}

　　這部分實現處理腳本啓動參數，其必須是一個儲存要發送的消息的文件名或表示使用交互輸入獲得發送消的內容的"i"命令。

　　變量file被設置爲腳本的第一個參數的值，是經過一個Tcl函數lindex來實現的，該函數從列表/數組獲得一個特定的元素。[]用來實現將函數lindex的返回值做爲set命令的參數。

　　若是腳本的第一個參數是小寫的"i"，那麼變量nofile被設置爲1，不然經過調用Tcl的函數isfile來驗證參數指定的文件存在，若是不存在就報錯退出。

　　能夠看到這裏使用了if命令來實現邏輯判斷功能。該命令後面直接跟判斷條件，而且執行在判斷條件後的{}內的命令。if條件爲false時則運行else後的程序塊。

　　set procs {}

　　# start write processes

　　for {set i 1} {$i<$argc}

　　{incr i} {

　　spawn -noecho write

　　[lindex $argv $i]

　　lappend procs $spawn_id

　　}

　　最後一部分使用spawn命令來啓動write進程實現向用戶發送消息。這裏使用了for命令來實現循環控制功能，循環變量首先設置爲1，而後因 此遞增。循環體是最後的{}的內容。這裏咱們是用腳本的第二個和隨後的參數來spawn一個write命令，並將每一個參數做爲發送消息的用戶名。 lappend命令使用保存每一個spawn的進程的進程ID號的內部變量$spawn_id在變量procs中構造了一個進程ID號列表。

　　if {$nofile==0} {

　　setmesg [open "$file" "r"]

　　} else {

　　send_user "enter message,

　　ending with ^D: " }

　　最後腳本根據變量nofile的值實現打開消息文件或者提示用戶輸入要發送的消息。

　　set timeout -1

　　while 1 {

　　if {$nofile==0} {

　　if {[gets $mesg chars] == -1} break

　　set line "$chars "

　　} else {

　　expect_user {

　　-re " " {}

　　eof break }

　　set line $expect_out(buffer) }

　　foreach spawn_id $procs {

　　send $line }

　　sleep 1}

　　exit

　　上面這段代碼說明了實際的消息文本是如何經過無限循環while被髮送的。while循環中的 if判斷消息是如何獲得的。在非交互模式下，下一行內容從消息文件中讀出，當文件內容結束時while循環也就結束了。(break命令實現終止循環) 。

　　在交互模式下，expect_user命令從用戶接收消息，當用戶輸入ctrl+D時結束輸入，循環同時結束。 兩種狀況下變量$line都被用來保存下一行消息內容。當是消息文件時，回車會被附加到消息的尾部。

　　foreach循環遍歷spawn的全部進程，這些進程的ID號都保存在列表變量$procs中，實現分別和各個進程通訊。send命令組成了 foreach的循環體，發送一行消息到當前的write進程。while循環的最後是一個sleep命令，主要是用於處理非交互模式狀況下，以確保消息 不會太快的發送給各個write進程。當while循環退出時，expect腳本結束。

其它資料：

在Linux下靈活expect腳本的小竅門