在上篇文章中,我們在PSNet包中創建了Test-TCPPort函數用于探測指定IP的指定端口是否開放,檢測端口之后大多數人想到的可能就是需要通過PowerShell收發TCP消息包了,這篇文章里將會描述如何在PSNet包中創建針對TCP消息包的函數Receive-和TCPMessageSend-TCPMessage。為了承接上篇中我們創建的PSNet工具集的思路,在確定了函數的命名之后,創建上述兩個函數對應的.ps1文件放置在$env:PSSpace/PSNet/TCPOp/中。
接下來在$env:PSSpace/PSNet/PSNet.psm1中添加以下兩條語句用于在工具集中引入上述兩個函數文件:
代碼如下:
. $env:PSSpace/PSNet/TCPOp/Receive-TCPMessage.ps1
. $env:PSSpace/PSNet/TCPOp/Send-TCPMessage.ps1
然后分別在創建的.ps1文件中添加以下代碼:
代碼如下:
=====文件名:Receive-TCPMessage.ps1=====
Function Receive-TCPMessage
{
param ( [ValidateNotNullOrEmpty()]
[int] $Port )
try
{
$EndPoint = New-Object System.Net.IPEndPoint([System.Net.IPAddress]::Loopback,$Port)
$Socket = New-Object System.Net.Sockets.TCPListener($EndPoint)
$Socket.Start()
$Socket = $Socket.AcceptTCPClient()
$EncodedText = New-Object System.Text.ASCIIEncoding
$Stream = $Socket.GetStream()
$Buffer = New-Object System.Byte[] $Socket.ReceiveBufferSize
while( $Bytes = $Stream.Read($Buffer,0,$Buffer.Length) )
{
$Stream.Write($Buffer,0,$Bytes)
Write-Output $EncodedText.GetString($Buffer,0,$Bytes)
}
$Socket.Close()
$Socket.Stop()
}
catch{}
}
代碼如下:
=====文件名:Send-TCPMessage.ps1=====
Function Send-TCPMessage
{
param ( [ValidateNotNullOrEmpty()]
[string] $EndPoint,
[int] $Port,
[string] $Message )
$IP = [System.Net.Dns]::GetHostAddresses($EndPoint)
$Address = [System.Net.IPAddress]::Parse($IP)
$Socket = New-Object System.Net.Sockets.TCPClient($Address,$Port)
$Stream = $Socket.GetStream()
新聞熱點
疑難解答