linux expect的使用詳解,實例

一、概述

  我們通過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。

  二、Expect工作原理

  從最簡單的層次來說,Expect的工作方式象一個通用化的Chat腳本工具。Chat腳本最早用於UUCP網絡內,以用來實現計算機之間需要建立連接時進行特定的登錄會話的自動化。

  Chat腳本由一系列expect-send對組成:expect等待輸出中輸出特定的字符,通常是一個提示符,然後發送特定的響應。例如下面的 Chat腳本實現等待標準輸出出現Login:字符串,然後發送somebody作爲用戶名;然後等待Password:提示符,併發出響應 sillyme。

  引用:Login: somebody Password: sillyme

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

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

  例子:

  1、實現功能

  下面我們分析一個響應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。

  2、下面是一個能用來實現自動執行該命令的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腳本也就退出結束。

  3、決定如何響應

  管理員往往有這樣的需求,希望根據當前的具體情況來以不同的方式對一個命令進行響應。我們可以通過後面的例子看到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包含有整個匹配輸出)。()可能會出現嵌套情況,這這種情況下編碼從最內層到最外層來進行的。

  4、使用超時

  下一個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"

  5、一個更復雜的例子

  下面我們將討論一個更加複雜的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腳本結束。



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章