開發中有沒有碰到過輸入一條命令後, 需要互動式輸入其他指令, 才能真正執行完第一條命令? 比如遠程ssh登錄時, 需要輸入yes, 然後輸入密碼的場景. 在自動化腳本中如何實現這種自動交互呢? 本篇博文通過對expect套件的介紹及演示, 解決這個問題. ...
目錄
1 安裝expect工具
expect是建立在tcl基礎上的一個自動化交互套件, 在一些需要交互輸入指令的場景下, 可通過腳本設置自動進行交互通信. 其交互流程是:
spawn啟動指定進程 -> expect獲取指定關鍵字 -> send想指定進程發送指定指令 -> 執行完成, 退出.
由於expect是基於tcl的, 所以需要確保系統中安裝了tcl:
# 檢查是否安裝了tcl:
[root@localhost ~]# whereis tcl
tcl: /usr/lib64/tcl8.5 /usr/include/tcl.h /usr/share/tcl8.5
# 如果沒有安裝, 使用yum安裝tcl和expect:
[root@localhost ~]# yum install -y tcl
[root@localhost ~]# yum install -y expect
# 查看expect的安裝路徑:
[root@localhost ~]# command -v expect
/usr/bin/expect2 expect的常用命令
| 命 令 | 說 明 |
|---|---|
| spawn | 啟動新的交互進程, 後面跟命令或者指定程式 |
| expect | 從進程中接收信息, 如果匹配成功, 就執行expect後的動作 |
| send | 向進程發送字元串 |
| send exp_send | 用於發送指定的字元串信息 |
| exp_continue | 在expect中多次匹配就需要用到 |
| send_user | 用來列印輸出 相當於shell中的echo |
| interact | 允許用戶交互 |
| exit | 退出expect腳本 |
| eof | expect執行結束, 退出 |
| set | 定義變數 |
| puts | 輸出變數 |
| set timeout | 設置超時時間 |
3 作用原理簡介
3.1 示例腳本
這裡以ssh遠程登錄某台伺服器的腳本為例進行說明, 假設此腳本名稱為remote_login.sh:
#!/usr/bin/expect
set timeout 30
spawn ssh -l root 172.16.22.131
expect "password*"
send "123456\r"
interact3.2 腳本功能解讀
(1) #!/usr/bin/expect
上述內容必須位於腳本文件的第一行, 用來告訴操作系統, 此腳本需要使用系統的哪個腳本解析引擎來執行.
具體路徑可通過command -v expect命令查看.
註意:
這裡的expect和Linux的bash、Windows的cmd等程式一樣, 都是一種腳本執行引擎.
腳本需要有可執行許可權(
chmod +x remote_login.sh, 或chmod 755 auto_login.sh), 然後通過命令./remote_login.sh運行即可;如果輸入
sh remote_login.sh, 意義就不一樣了: 明確調用sh引擎去執行此腳本, 此時首行的#!/usr/bin/expect就失效了.
(2) set timeout 30
設置連接的超時時間為30秒.
(3) spawn ssh -l root 172.16.22.131
spawn、send等命令是expect工具中的內部命令, 如果沒有安裝expect工具, 就會出現"spawn not found"等錯誤.
不要用
which spawn之類的命令去找spawn, 因為並沒有這樣的程式.
(4) expect "password*"
這個命令用來判斷上次輸出結果里是否包含"password*"的字元串, 如果有則立即返回, 否則就等待一段時間後返回. 這裡的等待時長就是前面設置的timeout, 也就是30秒.
(5) send "123456\r"
這裡就是執行交互動作, 作用等同於手工輸入密碼.
提示: 命令字元串結尾加上\r, 這樣的話, 如果出現異常等待的狀態就能夠停留下來, 作進一步的核查.
(6) interact
expect執行完成後保持用戶的交互狀態, 這個時候用戶就可以手工操作了.
如果沒有這一句, expect執行完成後就會退出腳本剛剛遠程登錄過去的終端, 用戶也就不能繼續操作了.
4 其他腳本使用示例
4.1 直接通過expect執行多條命令
註意首行內容, 這種情況下就只能通過./script.sh來執行這類腳本了:
#!/usr/bin/expect -f
set timeout 10
# 切換到root用戶, 然後執行ls和df命令:
spawn su - root
expect "Password*"
send "123456\r"
expect "]*" # 通配符
send "ls\r"
expect "#*" # 通配符的另一種形式
send "df -Th\r"
send "exit\r" # 退出spawn開啟的進程
expect eof # 退出此expect交互程式4.2 通過shell調用expect執行多條命令
註意首行內容, 這種情況下可通過sh script.sh、bash script.sh 或./script.sh, 都可以執行這類腳本:
#!/bin/bash
ip="172.16.22.131"
username="root"
password="123456"
# 指定執行引擎
/usr/bin/expect <<EOF
set time 30
spawn ssh $username@$ip df -Th
expect {
"*yes/no" { send "yes\r"; exp_continue }
"*password:" { send "$password\r" }
}
expect eof
EOF5 spawn not found 的解決
出現這個錯誤的基本上都是初學者: Linux 執行shell腳本有兩種方式:
一種是將腳本作為sh的命令行參數, 如
sh remote_login.sh, 或sh /data/remote_login.sh;一種是將腳本作為具有執行許可權的可執行腳本, 如
./remote_login.sh, 或/data/remote_login.sh.
而作為sh命令行參數來運行, 就會導致腳本第一行的#!/usr/bin/expect失效, 也就出現了spawn not found、send not found等錯誤.
要解決這個問題, 只需要通過以下命令運行腳本即可: ./automate_expect.sh
參考資料
版權聲明
