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报文将会用到，也将会在后续的文章中进行改进。