PowerShell收發TCP消息包

在上篇文章中，咱們在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()
    $Writer = New-Object System.IO.StreamWriter($Stream)
    $Writer.AutoFlush = $true
    $Writer.NewLine = $true
    $Writer.Write($Message)
    $Socket.Close()
}

將代碼保存到指定目錄的對應文件後,分別啓動兩個PowerShell進程，分別導入PSNet Module：

Import-Module $env:PSSpace\PSNet

或者在啓動PowerShell時指定參數，或者將下列語句建立批處理腳本啓動加載指定Module的PowerShell進程。

start %windir%\System32\WindowsPowerShell\v1.0\powershell.exe -noExit -Command "Import-Module '%PSSpace%\PSNet' "

首先在其中一個PowerShell窗口中，指定端口用Receive-TCPMessage函數在制定端口監聽，等待接受數據

Receive-TCPMessage 8080

在另一個窗口中向上面的端口發送數據：

Send-TCPMessage 127.0.0.1 8080 「This a Message Send from PSNet!」

會發現這個進程發送消息後，前一個進程能收到相應的消息，消息中若是有空格須要用雙引號包括確保PowerShell解釋引擎知道這是一個完整的參數。這兩個小函數雖然功能簡單，並且其中的消息使用ASCII的方式發送的，這是爲了給你們演示發送和接收的效果，在真正的實際環境中仍是有問題的，可是這兩個函數將會在後續的功能中起到很重要的做用，使用PowerShell發送TCP和接收TCP報文將會用到，也將會在後續的文章中進行改進。

做者: 付海軍
出處：http://fuhj02.blog.51cto.com
版權：本文版權歸做者和51cto共有
轉載：歡迎轉載，爲了保存做者的創做熱情，請按要求【轉載】，謝謝
要求：未經做者贊成，必須保留此段聲明；必須在文章中給出原文鏈接且保證內容完整!不然必究法律責任!
我的網站: http://txj.shell.tor.hu/