在PSNet程序集中繼續(xù)進(jìn)行擴(kuò)展,在$env:PSSpace/PSNet/TCPOp/下創(chuàng)建Invoke-SSHLogin.ps1腳本用于在傳入指定ssh地址、用戶名和密碼后返回是否登錄成功。
在$env:PSSpace/PSNet/下創(chuàng)建Lib目錄用于存放在腳本中需要使用的第三方庫(kù),由于在本文中需要引用Renci.SshNet.dll用戶可以從http://sshnet.codeplex.com/releases/view/72214中下載程序,解壓后將文件放置在$env:PSSpace/PSNet/Lib中用于后續(xù)的調(diào)用。
在$env:PSSpace/PSNet/PSNet.psm1添加對(duì)剛下載的程序集的引用,并添加對(duì)Invoke-SSHLogin.ps1腳本的引用,在PowerShell啟動(dòng)時(shí)同時(shí)加載并初始化函數(shù)。
能看到例子中,選擇了正確的主機(jī)名,用戶名和密碼后登錄成功了,而輸入錯(cuò)誤的用戶名和密碼以及輸出不存在監(jiān)聽的主機(jī)名時(shí)均報(bào)錯(cuò)了.為了保證服務(wù)器和用戶名密碼的安全,對(duì)于存在的主機(jī)已經(jīng)通過(guò)處理抹去了主機(jī)名用戶名和密碼了.
源代碼如下:
=====文件名:Invoke-SSHLogin.ps1=====
Function Invoke-SSHLogin{
Param(
[parameter(Mandatory = $true)]
[string]$Site = "localhost",
[parameter(Mandatory = $true)]
[string]$User = "Anonymous",
[parameter(Mandatory = $true)]
[string]$Pass = "hello@world",
[int]$Port=22
)
Write-Host "Try to Login ssh..."
# Do directory listing
$SSHreq = New-Object Renci.SshNet.SshClient($Site,$Port,$User,$Pass)
try
{
$SSHreq.Connect()
if ($SSHreq.IsConnected) {
$cmd = $SSHreq.RunCommand("ps aux")
$output = $cmd.Result.Split("`n")
$success = $true
Write-Host "Login Succed" -BackgroundColor green -ForegroundColor yellow
}
$cmd.Dispose()
$SSHreq.Disconnect()
$SSHreq.Dispose()
}
catch
{
Write-Host "FAILED: $_" -BackgroundColor red -ForegroundColor black
$success = $false
}
}