Windows一鍵執行SSH指令：AutoIT、Pietty搭配使用

5月 02, 2011 程式寫作 , 電腦軟體 , 網路管理 , DSpace 11 Comments Edit Copy Download

Linux的命令列畫面讓很多人惟恐避之不及，這對有許多指令需要依賴SSH連線、並在命令列輸入才能執行的DSpace來說，無疑是一大門檻。在此我提出一種可以做到「一鍵指令」的方案：利用AutoIt自動執行機器人，透過PieTTY的命令列模式，用ini設定檔來設定連線參數，來執行捷徑所指定的指令的作法。

這種作法對資安來說非常不理想，僅是一個簡便利用的應急方案而已。

一鍵指令的背景與目的

我現在要做的事情是：

1. 在Windows的環境下
2. 透過SSH連線到Linux伺服器
3. 進行登入動作（以及金鑰確認）
4. 然後執行一個指令。

而這個方案的目的是將使用者的操作簡化到極限，我希望使用者能夠以滑鼠執行一個檔案，就自動做完上述的所有動作。

一鍵指令方案概述

簡單來說，這個方案的作法是：

1. 利用PieTTY的命令列模式來輸入登入資訊與要執行的指令
2. 以ini檔案儲存登入資訊
3. 在Linux伺服器端配置參數與接收命令的sh腳本
4. 利用AutoIt來處理連線的細節
5. 利用Windows的捷徑與捷徑後輸入的參數，將參數輸入給AutoIt

Linux環境配置

要做到這些事情，需要Linux伺服器端與Windows使用者端的環境一起配置才行。這邊先敘述Linux的部份，大致上是要做以下三件事情：

• 建立帳號、密碼
• 開啟ssh連線
• 建立ssh-cmd.sh腳本

Linux伺服器建立帳號、密碼、開啟ssh連線等基本操作請參考鳥哥的Linux思房菜。在此最重要的是要配置接收來自客戶端指令的執行腳本「ssh-cmd.sh」。

在本篇的例子中，我以普通使用者「dspace」跟管理者「root」作為帳號、「password」作為該帳號的密碼來說明。

ssh-cmd.sh：執行指令腳本

因為帳號是使用「dspace」，所以我在「/home/dspace/ssh-cmd.sh」（SkyDrive備份）這個位置建立起腳本。腳本的內容如下：

#!/bin/bash
# Program:
#       Excute command and pause at finish.
## History:
## 2009/11/12    Pudding   First release
PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:~/bin
export JAVA_HOME=/usr/java/jdk1.6.0_06
export PATH
export LC_ALL=en_US
export JAVA_OPTS="-Xmx128M -Xms64M -Dfile.encoding=UTF-8"

ANT_HOME=/opt/apache-ant-1.7.0
JAVA_HOME=/usr/java/jdk1.6.0_06
CATALINA_HOME=/opt/apache-tomcat-6.0.16
PATH=$JAVA_HOME/bin:$JAVA_HOME/jre/bin:$ANT_HOME/bin:$PATH:/opt/apache-maven-2.0.9/bin
export PATH USER LOGNAME MAIL HOSTNAME HISTSIZE INPUTRC JAVA_HOME ANT_HOME CATALINA_HOME


if [ "$1" == "/home/dspace/tomcat_restart.sh" ]; then
        at now + 0 minutes < $1
        echo "Tomcat will restart now."
else
        $1
fi

echo ""
echo "====================================="
echo ""
read -p "Please press ENTER to exit. " doExit

這個腳本中設置了相當多參數，請依照你伺服器的實際情況來決定要設置的參數內容。

此外，由於Tomcat的重新啟動無法直接透過這樣的指令來執行，所以我改成以「at」執行一次的方式來執行。必須說明的是，這並不是很理想的作法，有待以後改善。

tomcat_restart.sh：重新啟動Tomcat伺服器腳本

如果你沒有需要重新啟動Tomcat的話，這個步驟是可以省略的。「/home/dspace/tomcat_restart.sh」（SkyDrive備份）是用來重新啟動Tomcat伺服器的腳本，內容如下：

export JAVA_HOME=/usr/java/jdk1.6.0_06
export JAVA_OPTS="-Xmx128M -Xms64M -Dfile.encoding=UTF-8"

/opt/apache-tomcat-6.0.16/bin/shutdown.sh 
sleep 1
/opt/apache-tomcat-6.0.16/bin/startup.sh 

要記得將這些腳本的執行權限開啟，他們才能正常地執行。

Windows使用者端配置

你現在需要在Windows中建立一個資料夾，例如「c:\ssh-cmd\」，然後把以下的檔案放進去：

• pietty0327.exe（SkyDrive備份）：SSH連線工具
• ssh.cmd.exe (AutoIt原始腳本檔案)：執行連線的自動化程式
• config.ini：連線配置
• Windows捷徑：配置要執行的身分與指令，稍後敘述

PieTTY的命令列執行方式

雖然很多人都知道PieTTY是用來作SSH連線的，但是知道PieTTY可以透過命令列與指令檔案來進行連線並執行指令的人就不多了，其實這是基於PieTTY改編自PuTTY原始就有的功能喔。

具體來說，利用命令列來透過PieTTY連線的作法如下：

pietty0327.exe -l [ID] -P [PORT] -pw [PASSWORD] -m [COMMAND] -t [HOST]

裡面的參數中：

• [ID]是要登入的帳號
• [PORT]是連接埠
• [PASSWORD]是密碼
• [COMMAND]是「存放指令的檔案位置」，注意，不是指令，只是檔案位置喔
• [HOST]要連線的主機IP

config.ini：連線配置

連線的配置資訊，我將它存放在「config.ini」中。「config.ini」的內容如下：

[ssh-dspace]
login=dspace
pw=password

[ssh-root]
login=root
pw=password

[ssh]
port=22
host=127.0.0.1

在此安排了兩種不同的身分，一種是以普通使用者「dspace」來執行，另一種是以管理者「root」來執行。除了重新啟動伺服器等重大指令之外，一般指令最好是以普通使用者「dspace」的身分來執行會比較安全。

ssh-cmd.exe：執行連線腳本

由於要執行指令必須要以存放在檔案中的形式來執行，所以建立這個檔案、並指引PieTTY來連線的工作，是交給AutoIt的自動腳本ssh-cmd.exe來執行。

ssh-cmd.exe是由ssh-cmd.au3編譯而來的，這是AutoIt的程式語言。內容如下：

Dim $user = "ssh-dspace"
Dim $script = ""
If $CmdLine[0] == 2 Then
	$user = "ssh-" & $CmdLine[1]
	$script = $CmdLine[2]
ElseIf $CmdLine[0] == 1 Then
	$script = $CmdLine[1]
EndIf

Dim $login = IniRead("config.ini",$user, "login", "root")
Dim $pw = IniRead("config.ini",$user, "pw", "password")
Dim $port = IniRead("config.ini","ssh", "port", "22")
Dim $host = IniRead("config.ini","ssh", "host", "127.0.0.1")

If $script == "" Then
	MsgBox(0, "錯誤", "請指定要執行的指令")
	Exit
EndIf

Dim $doExcute = MsgBox(1, "提示","您確定要在伺服器中執行以下命令嗎: " & $script)
If $doExcute == 2 Then
    Exit
EndIf

Dim $sh = "cmd.sh"
Dim $scriptHeader = "~/ssh-cmd.sh "

$file = FileOpen($sh, 2)

; Check if file opened for writing OK
If $file = -1 Then
    MsgBox(0, "錯誤", "無法開啟檔案")
    Exit
EndIf

FileWrite($file, $scriptHeader & $script)
FileClose($file)

Dim $cmd = "pietty0327.exe -l " & $login & " -P " & $port & " -pw " & $pw & " -m " & $sh & " -t " & $host
Run($cmd)
Sleep(1000)
If WinActivate("PieTTY Security Alert") Then
	WinActive("PieTTY Security Alert")
	Send("{ALTDOWN}Y{ALTUP}")
EndIf

這個腳本做了幾件事情：

1. 接收來自外部的「身分」與「指令」參數（稍後建立Windows捷徑的時候，我會再說明怎麼輸入這邊的參數）
2. 讀取來自config.ini中的連線資訊，包括「帳號」、「密碼」、「伺服器位置」、「連接埠」
3. 過濾指令、顯示提示
4. 將指令寫入「cmd.sh」檔案中
5. 以命令列執行PieTTY連線

建立Windows捷徑

接著要建立到「ssh-cmd.exe」的捷徑，並且在捷徑之後加上「參數」。建立好ssh-cmd.exe並改好名字之後，你可以打開捷徑的「內容」，在「目標」的欄位中設定參數。

舉例來說，我現在要在伺服器上以「dspace」身分執行「/home/dspace/tomcat_restart.sh」的指令，那麼目標裡面的設定如下：

"C:\ssh-cmd\ssh-cmd.exe" dspace /home/dspace/tomcat_restart.sh

目標的值除了指定要執行的程式位置之外，還指定了要執行的身分「dspace」跟要執行的指令「/home/dspace/tomcat_restart.sh」。如此配置之後，就能夠讓使用者直接點選此捷徑，讓他自動執行這些動作了。

你可以利用這個例子舉一反三，建立執行不同身分、不同指令的捷徑，並組成一個操作列表，讓使用者易於選擇。也可以將此捷徑放到「開始」程式集裡面，讓使用者更容易執行各種操作。這些動作我是用InnoSetup在安裝時自動配置而成，不過那又是另一番長篇大論了。

結語

儘管一鍵指令的方案我研究了很久，但是仍有很多不完美之處，希望高手能多多指教。

其實這個是DSpace-DLLL寫書的一部分技術內容。之前寫完書之後就匆忙地投入其他事情中，到最近在使用AutoIt時，又把之前寫的檔案挖出來看。

這時候就會覺得這個Blog對我來說也是很重要的，如果沒有這些記錄，我恐怕很快就會忘記自己之前做過的事情了吧。