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/