在上篇文章中,在PSNet工具集中創建了Send-TCPMessage和Receive-TCPMessage兩個函數實現了通過PowerShell收發TCP消息包的功能,有了TCP包的發送和接收,自然少不了UDP消息包的發送和接收,本文將會介紹通過PowerShell發送和接收UDP消息包的方法。
為了能跟之前的PSNet程序集匹配,繼續基于此程序集進行擴展,在$env:PSSpace/PSNet下創建UDPOp目錄,在其中創建Receive-UDPMessage.ps1和Send-UDPMessage.ps1兩個文件,代碼稍后插入。
在$env:PSSpace/PSNet/PSNet.psm1中添加對上述兩個腳本文件的引用,代碼如下:
代碼如下:
. $env:PSSpace/PSNet/TCPOp/Receive-UDPMessage.ps1
. $env:PSSpace/PSNet/TCPOp/Send-UDPMessage.ps1
需要注意的是這兩行代碼需要在PSNet.psm1中已經存在的Export-ModuleMember -Function *代碼之前,以便能保證被引入的腳本文件中包含的函數均可以作為該程序集模塊的成員注冊,以便能夠被PowerShell進程識別。
下面附上代碼:
代碼如下:
=====文件名:Receive-UDPMessage.ps1=====
Function Receive-UDPMessage
{
param ( [ValidateNotNullOrEmpty()]
[int] $Port )
try
{
$EndPoint = New-Object System.Net.IPEndPoint([System.Net.IPAddress]::Loopback,$Port)
$UDPClient = New-Object System.Net.Sockets.UDPClient($Port)
$ReceiveMessage = $UDPClient.Receive([System.Management.Automation.PSReference]$EndPoint)
[System.Text.Encoding]::ASCII.GetString($ReceiveMessage)
}
catch{}
}
代碼如下:
=====文件名:Send-UDPMessage.ps1=====
Function Send-UDPMessage
{
param ( [ValidateNotNullOrEmpty()]
[string] $EndPoint,
[int] $Port,
[string] $Message )
$IP = [System.Net.Dns]::GetHostAddresses($EndPoint)
$Address = [System.Net.IPAddress]::Parse($IP)
$EndPoints = New-Object System.Net.IPEndPoint($Address,$Port)
$Socket = New-Object System.Net.Sockets.UDPClient
$EncodedText = [Text.Encoding]::ASCII.GetBytes($Message)
$SendMessage = $Socket.Send($EncodedText,$EncodedText.Length,$EndPoints)
$Socket.Close()
}
啟動兩個PowerShell進程分別導入PSNet模塊:
代碼如下:
Import-Module $env:PSSpace/PSNet
首先運行Receive-UDPMessage對端口進行監聽:
代碼如下:
Receive-UDPMessage 8080
然后通過Receive-UDPMessage函數對上面指定的端口發送消息:
新聞熱點
疑難解答