PowerShell腳本開(kāi)發(fā)之收發(fā)UDP消息包

更新時(shí)間：2014年10月13日 09:51:37 投稿：hebedich

上篇文章我們介紹了使用PowerShell收發(fā)TCP消息包，今天我們來(lái)介紹下使用Powershell收發(fā)UDP小細(xì)胞的方法

在上篇文章中，在PSNet工具集中創(chuàng)建了Send-TCPMessage和Receive-TCPMessage兩個(gè)函數(shù)實(shí)現(xiàn)了通過(guò)PowerShell收發(fā)TCP消息包的功能，有了TCP包的發(fā)送和接收，自然少不了UDP消息包的發(fā)送和接收，本文將會(huì)介紹通過(guò)PowerShell發(fā)送和接收UDP消息包的方法。

為了能跟之前的PSNet程序集匹配，繼續(xù)基于此程序集進(jìn)行擴(kuò)展，在$env:PSSpace\PSNet下創(chuàng)建UDPOp目錄，在其中創(chuàng)建Receive-UDPMessage.ps1和Send-UDPMessage.ps1兩個(gè)文件，代碼稍后插入。

在$env:PSSpace\PSNet\PSNet.psm1中添加對(duì)上述兩個(gè)腳本文件的引用，代碼如下：

復(fù)制代碼代碼如下:

. $env:PSSpace/PSNet/TCPOp/Receive-UDPMessage.ps1 

. $env:PSSpace/PSNet/TCPOp/Send-UDPMessage.ps1

需要注意的是這兩行代碼需要在PSNet.psm1中已經(jīng)存在的Export-ModuleMember -Function *代碼之前，以便能保證被引入的腳本文件中包含的函數(shù)均可以作為該程序集模塊的成員注冊(cè)，以便能夠被PowerShell進(jìn)程識(shí)別。

下面附上代碼：

復(fù)制代碼代碼如下:

=====文件名：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{}
}

復(fù)制代碼代碼如下:

        =====文件名：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()
}

啟動(dòng)兩個(gè)PowerShell進(jìn)程分別導(dǎo)入PSNet模塊：

復(fù)制代碼代碼如下:

Import-Module $env:PSSpace\PSNet

首先運(yùn)行Receive-UDPMessage對(duì)端口進(jìn)行監(jiān)聽(tīng)：

復(fù)制代碼代碼如下:

Receive-UDPMessage 8080

然后通過(guò)Receive-UDPMessage函數(shù)對(duì)上面指定的端口發(fā)送消息：

復(fù)制代碼代碼如下:

Send-UDPMessage 127.0.0.1 8080 "This is a UDP Message Send from PSNet!"

發(fā)送之后會(huì)在前面的窗口中顯示消息內(nèi)容。

本文中的實(shí)例中為了顯示發(fā)送效果，字符串傳輸過(guò)程中是ASCII字符的形式傳送的，但是實(shí)際情況中通常是十六進(jìn)制數(shù)據(jù)形式，在后續(xù)的文章中我們將會(huì)對(duì)本文的實(shí)例進(jìn)行改進(jìn)以便適應(yīng)實(shí)際工作中的需要。

您可能感興趣的文章: